hosts.go 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141
  1. package controller
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "net/http"
  7. "time"
  8. "github.com/google/uuid"
  9. "github.com/gorilla/mux"
  10. "github.com/gravitl/netmaker/database"
  11. "github.com/gravitl/netmaker/logger"
  12. "github.com/gravitl/netmaker/logic"
  13. "github.com/gravitl/netmaker/models"
  14. "github.com/gravitl/netmaker/mq"
  15. "github.com/gravitl/netmaker/servercfg"
  16. "golang.org/x/crypto/bcrypt"
  17. "golang.org/x/exp/slog"
  18. )
  19. func hostHandlers(r *mux.Router) {
  20. r.HandleFunc("/api/hosts", logic.SecurityCheck(true, http.HandlerFunc(getHosts))).
  21. Methods(http.MethodGet)
  22. r.HandleFunc("/api/hosts/keys", logic.SecurityCheck(true, http.HandlerFunc(updateAllKeys))).
  23. Methods(http.MethodPut)
  24. r.HandleFunc("/api/hosts/sync", logic.SecurityCheck(true, http.HandlerFunc(syncHosts))).
  25. Methods(http.MethodPost)
  26. r.HandleFunc("/api/hosts/upgrade", logic.SecurityCheck(true, http.HandlerFunc(upgradeHosts))).
  27. Methods(http.MethodPost)
  28. r.HandleFunc("/api/hosts/{hostid}/keys", logic.SecurityCheck(true, http.HandlerFunc(updateKeys))).
  29. Methods(http.MethodPut)
  30. r.HandleFunc("/api/hosts/{hostid}/sync", logic.SecurityCheck(true, http.HandlerFunc(syncHost))).
  31. Methods(http.MethodPost)
  32. r.HandleFunc("/api/hosts/{hostid}", logic.SecurityCheck(true, http.HandlerFunc(updateHost))).
  33. Methods(http.MethodPut)
  34. r.HandleFunc("/api/hosts/{hostid}", Authorize(true, false, "all", http.HandlerFunc(deleteHost))).
  35. Methods(http.MethodDelete)
  36. r.HandleFunc("/api/hosts/{hostid}/upgrade", logic.SecurityCheck(true, http.HandlerFunc(upgradeHost))).
  37. Methods(http.MethodPut)
  38. r.HandleFunc("/api/hosts/{hostid}/networks/{network}", logic.SecurityCheck(true, http.HandlerFunc(addHostToNetwork))).
  39. Methods(http.MethodPost)
  40. r.HandleFunc("/api/hosts/{hostid}/networks/{network}", logic.SecurityCheck(true, http.HandlerFunc(deleteHostFromNetwork))).
  41. Methods(http.MethodDelete)
  42. r.HandleFunc("/api/hosts/adm/authenticate", authenticateHost).Methods(http.MethodPost)
  43. r.HandleFunc("/api/v1/host", Authorize(true, false, "host", http.HandlerFunc(pull))).
  44. Methods(http.MethodGet)
  45. r.HandleFunc("/api/v1/host/{hostid}/signalpeer", Authorize(true, false, "host", http.HandlerFunc(signalPeer))).
  46. Methods(http.MethodPost)
  47. r.HandleFunc("/api/v1/fallback/host/{hostid}", Authorize(true, false, "host", http.HandlerFunc(hostUpdateFallback))).
  48. Methods(http.MethodPut)
  49. r.HandleFunc("/api/v1/host/{hostid}/peer_info", Authorize(true, false, "host", http.HandlerFunc(getHostPeerInfo))).
  50. Methods(http.MethodGet)
  51. r.HandleFunc("/api/emqx/hosts", logic.SecurityCheck(true, http.HandlerFunc(delEmqxHosts))).
  52. Methods(http.MethodDelete)
  53. r.HandleFunc("/api/v1/auth-register/host", socketHandler)
  54. }
  55. // @Summary Requests all the hosts to upgrade their version
  56. // @Router /api/hosts/upgrade [post]
  57. // @Tags Hosts
  58. // @Security oauth
  59. // @Param force query bool false "Force upgrade"
  60. // @Success 200 {string} string "upgrade all hosts request received"
  61. func upgradeHosts(w http.ResponseWriter, r *http.Request) {
  62. w.Header().Set("Content-Type", "application/json")
  63. action := models.Upgrade
  64. if r.URL.Query().Get("force") == "true" {
  65. action = models.ForceUpgrade
  66. }
  67. user := r.Header.Get("user")
  68. go func() {
  69. slog.Info("requesting all hosts to upgrade", "user", user)
  70. hosts, err := logic.GetAllHosts()
  71. if err != nil {
  72. slog.Error("failed to retrieve all hosts", "user", user, "error", err)
  73. return
  74. }
  75. for _, host := range hosts {
  76. go func(host models.Host) {
  77. hostUpdate := models.HostUpdate{
  78. Action: action,
  79. Host: host,
  80. }
  81. if err = mq.HostUpdate(&hostUpdate); err != nil {
  82. slog.Error("failed to request host to upgrade", "user", user, "host", host.ID.String(), "error", err)
  83. } else {
  84. slog.Info("host upgrade requested", "user", user, "host", host.ID.String())
  85. }
  86. }(host)
  87. }
  88. }()
  89. logic.LogEvent(&models.Event{
  90. Action: models.UpgradeAll,
  91. Source: models.Subject{
  92. ID: r.Header.Get("user"),
  93. Name: r.Header.Get("user"),
  94. Type: models.UserSub,
  95. },
  96. TriggeredBy: r.Header.Get("user"),
  97. Target: models.Subject{
  98. ID: "All Hosts",
  99. Name: "All Hosts",
  100. Type: models.DeviceSub,
  101. },
  102. Origin: models.Dashboard,
  103. })
  104. slog.Info("upgrade all hosts request received", "user", user)
  105. logic.ReturnSuccessResponse(w, r, "upgrade all hosts request received")
  106. }
  107. // @Summary Upgrade a host
  108. // @Router /api/hosts/{hostid}/upgrade [put]
  109. // @Tags Hosts
  110. // @Security oauth
  111. // @Param hostid path string true "Host ID"
  112. // @Param force query bool false "Force upgrade"
  113. // @Success 200 {string} string "passed message to upgrade host"
  114. // @Failure 500 {object} models.ErrorResponse
  115. // upgrade host is a handler to send upgrade message to a host
  116. func upgradeHost(w http.ResponseWriter, r *http.Request) {
  117. host, err := logic.GetHost(mux.Vars(r)["hostid"])
  118. if err != nil {
  119. slog.Error("failed to find host", "error", err)
  120. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "notfound"))
  121. return
  122. }
  123. action := models.Upgrade
  124. if r.URL.Query().Get("force") == "true" {
  125. action = models.ForceUpgrade
  126. }
  127. if err := mq.HostUpdate(&models.HostUpdate{Action: action, Host: *host}); err != nil {
  128. slog.Error("failed to upgrade host", "error", err)
  129. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  130. return
  131. }
  132. logic.ReturnSuccessResponse(w, r, "passed message to upgrade host")
  133. }
  134. // @Summary List all hosts
  135. // @Router /api/hosts [get]
  136. // @Tags Hosts
  137. // @Security oauth
  138. // @Success 200 {array} models.ApiHost
  139. // @Failure 500 {object} models.ErrorResponse
  140. func getHosts(w http.ResponseWriter, r *http.Request) {
  141. w.Header().Set("Content-Type", "application/json")
  142. currentHosts, err := logic.GetAllHosts()
  143. if err != nil {
  144. logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
  145. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  146. return
  147. }
  148. apiHosts := logic.GetAllHostsAPI(currentHosts[:])
  149. logger.Log(2, r.Header.Get("user"), "fetched all hosts")
  150. logic.SortApiHosts(apiHosts[:])
  151. w.WriteHeader(http.StatusOK)
  152. json.NewEncoder(w).Encode(apiHosts)
  153. }
  154. // @Summary Used by clients for "pull" command
  155. // @Router /api/v1/host [get]
  156. // @Tags Hosts
  157. // @Security oauth
  158. // @Success 200 {object} models.HostPull
  159. // @Failure 500 {object} models.ErrorResponse
  160. func pull(w http.ResponseWriter, r *http.Request) {
  161. hostID := r.Header.Get(hostIDHeader) // return JSON/API formatted keys
  162. if len(hostID) == 0 {
  163. logger.Log(0, "no host authorized to pull")
  164. logic.ReturnErrorResponse(
  165. w,
  166. r,
  167. logic.FormatError(fmt.Errorf("no host authorized to pull"), "internal"),
  168. )
  169. return
  170. }
  171. host, err := logic.GetHost(hostID)
  172. if err != nil {
  173. logger.Log(0, "no host found during pull", hostID)
  174. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  175. return
  176. }
  177. sendPeerUpdate := false
  178. for _, nodeID := range host.Nodes {
  179. node, err := logic.GetNodeByID(nodeID)
  180. if err != nil {
  181. slog.Error("failed to get node:", "id", node.ID, "error", err)
  182. continue
  183. }
  184. if node.FailedOverBy != uuid.Nil && r.URL.Query().Get("reset_failovered") == "true" {
  185. logic.ResetFailedOverPeer(&node)
  186. sendPeerUpdate = true
  187. }
  188. }
  189. if sendPeerUpdate {
  190. if err := mq.PublishPeerUpdate(false); err != nil {
  191. logger.Log(0, "fail to publish peer update: ", err.Error())
  192. }
  193. }
  194. allNodes, err := logic.GetAllNodes()
  195. if err != nil {
  196. logger.Log(0, "failed to get nodes: ", hostID)
  197. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  198. return
  199. }
  200. hPU, err := logic.GetPeerUpdateForHost("", host, allNodes, nil, nil)
  201. if err != nil {
  202. logger.Log(0, "could not pull peers for host", hostID, err.Error())
  203. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  204. return
  205. }
  206. serverConf := logic.GetServerInfo()
  207. key, keyErr := logic.RetrievePublicTrafficKey()
  208. if keyErr != nil {
  209. logger.Log(0, "error retrieving key:", keyErr.Error())
  210. logic.ReturnErrorResponse(w, r, logic.FormatError(keyErr, "internal"))
  211. return
  212. }
  213. _ = logic.CheckHostPorts(host)
  214. serverConf.TrafficKey = key
  215. response := models.HostPull{
  216. Host: *host,
  217. Nodes: logic.GetHostNodes(host),
  218. ServerConfig: serverConf,
  219. Peers: hPU.Peers,
  220. PeerIDs: hPU.PeerIDs,
  221. HostNetworkInfo: hPU.HostNetworkInfo,
  222. EgressRoutes: hPU.EgressRoutes,
  223. FwUpdate: hPU.FwUpdate,
  224. ChangeDefaultGw: hPU.ChangeDefaultGw,
  225. DefaultGwIp: hPU.DefaultGwIp,
  226. IsInternetGw: hPU.IsInternetGw,
  227. EndpointDetection: logic.IsEndpointDetectionEnabled(),
  228. }
  229. logger.Log(1, hostID, "completed a pull")
  230. w.WriteHeader(http.StatusOK)
  231. json.NewEncoder(w).Encode(&response)
  232. }
  233. // @Summary Updates a Netclient host on Netmaker server
  234. // @Router /api/hosts/{hostid} [put]
  235. // @Tags Hosts
  236. // @Security oauth
  237. // @Param hostid path string true "Host ID"
  238. // @Param body body models.ApiHost true "New host data"
  239. // @Success 200 {object} models.ApiHost
  240. // @Failure 500 {object} models.ErrorResponse
  241. func updateHost(w http.ResponseWriter, r *http.Request) {
  242. var newHostData models.ApiHost
  243. err := json.NewDecoder(r.Body).Decode(&newHostData)
  244. if err != nil {
  245. logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
  246. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  247. return
  248. }
  249. // confirm host exists
  250. currHost, err := logic.GetHost(newHostData.ID)
  251. if err != nil {
  252. logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
  253. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  254. return
  255. }
  256. newHost := newHostData.ConvertAPIHostToNMHost(currHost)
  257. logic.UpdateHost(newHost, currHost) // update the in memory struct values
  258. if err = logic.UpsertHost(newHost); err != nil {
  259. logger.Log(0, r.Header.Get("user"), "failed to update a host:", err.Error())
  260. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  261. return
  262. }
  263. // publish host update through MQ
  264. if err := mq.HostUpdate(&models.HostUpdate{
  265. Action: models.UpdateHost,
  266. Host: *newHost,
  267. }); err != nil {
  268. logger.Log(
  269. 0,
  270. r.Header.Get("user"),
  271. "failed to send host update: ",
  272. currHost.ID.String(),
  273. err.Error(),
  274. )
  275. }
  276. go func() {
  277. if err := mq.PublishPeerUpdate(false); err != nil {
  278. logger.Log(0, "fail to publish peer update: ", err.Error())
  279. }
  280. if newHost.Name != currHost.Name {
  281. if servercfg.IsDNSMode() {
  282. logic.SetDNS()
  283. }
  284. }
  285. }()
  286. logic.LogEvent(&models.Event{
  287. Action: models.Update,
  288. Source: models.Subject{
  289. ID: r.Header.Get("user"),
  290. Name: r.Header.Get("user"),
  291. Type: models.UserSub,
  292. },
  293. TriggeredBy: r.Header.Get("user"),
  294. Target: models.Subject{
  295. ID: currHost.ID.String(),
  296. Name: newHost.Name,
  297. Type: models.DeviceSub,
  298. },
  299. Diff: models.Diff{
  300. Old: currHost,
  301. New: newHost,
  302. },
  303. Origin: models.Dashboard,
  304. })
  305. apiHostData := newHost.ConvertNMHostToAPI()
  306. logger.Log(2, r.Header.Get("user"), "updated host", newHost.ID.String())
  307. w.WriteHeader(http.StatusOK)
  308. json.NewEncoder(w).Encode(apiHostData)
  309. }
  310. // @Summary Updates a Netclient host on Netmaker server
  311. // @Router /api/v1/fallback/host/{hostid} [put]
  312. // @Tags Hosts
  313. // @Security oauth
  314. // @Param hostid path string true "Host ID"
  315. // @Param body body models.HostUpdate true "Host update data"
  316. // @Success 200 {string} string "updated host data"
  317. // @Failure 500 {object} models.ErrorResponse
  318. func hostUpdateFallback(w http.ResponseWriter, r *http.Request) {
  319. var params = mux.Vars(r)
  320. hostid := params["hostid"]
  321. currentHost, err := logic.GetHost(hostid)
  322. if err != nil {
  323. slog.Error("error getting host", "id", hostid, "error", err)
  324. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  325. return
  326. }
  327. var sendPeerUpdate bool
  328. var replacePeers bool
  329. var hostUpdate models.HostUpdate
  330. err = json.NewDecoder(r.Body).Decode(&hostUpdate)
  331. if err != nil {
  332. slog.Error("failed to update a host:", "user", r.Header.Get("user"), "error", err.Error(), "host", currentHost.Name)
  333. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  334. return
  335. }
  336. slog.Info("recieved host update", "name", hostUpdate.Host.Name, "id", hostUpdate.Host.ID, "action", hostUpdate.Action)
  337. switch hostUpdate.Action {
  338. case models.CheckIn:
  339. sendPeerUpdate = mq.HandleHostCheckin(&hostUpdate.Host, currentHost)
  340. case models.UpdateHost:
  341. if hostUpdate.Host.PublicKey != currentHost.PublicKey {
  342. //remove old peer entry
  343. replacePeers = true
  344. }
  345. sendPeerUpdate = logic.UpdateHostFromClient(&hostUpdate.Host, currentHost)
  346. err := logic.UpsertHost(currentHost)
  347. if err != nil {
  348. slog.Error("failed to update host", "id", currentHost.ID, "error", err)
  349. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  350. return
  351. }
  352. case models.UpdateMetrics:
  353. mq.UpdateMetricsFallBack(hostUpdate.Node.ID.String(), hostUpdate.NewMetrics)
  354. }
  355. if sendPeerUpdate {
  356. err := mq.PublishPeerUpdate(replacePeers)
  357. if err != nil {
  358. slog.Error("failed to publish peer update", "error", err)
  359. }
  360. }
  361. logic.ReturnSuccessResponse(w, r, "updated host data")
  362. }
  363. // @Summary Deletes a Netclient host from Netmaker server
  364. // @Router /api/hosts/{hostid} [delete]
  365. // @Tags Hosts
  366. // @Security oauth
  367. // @Param hostid path string true "Host ID"
  368. // @Param force query bool false "Force delete"
  369. // @Success 200 {object} models.ApiHost
  370. // @Failure 500 {object} models.ErrorResponse
  371. func deleteHost(w http.ResponseWriter, r *http.Request) {
  372. var params = mux.Vars(r)
  373. hostid := params["hostid"]
  374. forceDelete := r.URL.Query().Get("force") == "true"
  375. // confirm host exists
  376. currHost, err := logic.GetHost(hostid)
  377. if err != nil {
  378. logger.Log(0, r.Header.Get("user"), "failed to delete a host:", err.Error())
  379. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  380. return
  381. }
  382. for _, nodeID := range currHost.Nodes {
  383. node, err := logic.GetNodeByID(nodeID)
  384. if err != nil {
  385. slog.Error("failed to get node", "nodeid", nodeID, "error", err)
  386. continue
  387. }
  388. var gwClients []models.ExtClient
  389. if node.IsIngressGateway {
  390. gwClients = logic.GetGwExtclients(node.ID.String(), node.Network)
  391. }
  392. go mq.PublishMqUpdatesForDeletedNode(node, false, gwClients)
  393. }
  394. if servercfg.GetBrokerType() == servercfg.EmqxBrokerType {
  395. // delete EMQX credentials for host
  396. if err := mq.GetEmqxHandler().DeleteEmqxUser(currHost.ID.String()); err != nil {
  397. slog.Error(
  398. "failed to remove host credentials from EMQX",
  399. "id",
  400. currHost.ID,
  401. "error",
  402. err,
  403. )
  404. }
  405. }
  406. if err = mq.HostUpdate(&models.HostUpdate{
  407. Action: models.DeleteHost,
  408. Host: *currHost,
  409. }); err != nil {
  410. logger.Log(
  411. 0,
  412. r.Header.Get("user"),
  413. "failed to send delete host update: ",
  414. currHost.ID.String(),
  415. err.Error(),
  416. )
  417. }
  418. if err = logic.RemoveHost(currHost, forceDelete); err != nil {
  419. logger.Log(0, r.Header.Get("user"), "failed to delete a host:", err.Error())
  420. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  421. return
  422. }
  423. logic.LogEvent(&models.Event{
  424. Action: models.Delete,
  425. Source: models.Subject{
  426. ID: r.Header.Get("user"),
  427. Name: r.Header.Get("user"),
  428. Type: models.UserSub,
  429. },
  430. TriggeredBy: r.Header.Get("user"),
  431. Target: models.Subject{
  432. ID: currHost.ID.String(),
  433. Name: currHost.Name,
  434. Type: models.DeviceSub,
  435. },
  436. Origin: models.Dashboard,
  437. })
  438. apiHostData := currHost.ConvertNMHostToAPI()
  439. logger.Log(2, r.Header.Get("user"), "removed host", currHost.Name)
  440. w.WriteHeader(http.StatusOK)
  441. json.NewEncoder(w).Encode(apiHostData)
  442. }
  443. // @Summary To Add Host To Network
  444. // @Router /api/hosts/{hostid}/networks/{network} [post]
  445. // @Tags Hosts
  446. // @Security oauth
  447. // @Param hostid path string true "Host ID"
  448. // @Param network path string true "Network name"
  449. // @Success 200 {string} string "OK"
  450. // @Failure 500 {object} models.ErrorResponse
  451. func addHostToNetwork(w http.ResponseWriter, r *http.Request) {
  452. var params = mux.Vars(r)
  453. hostid := params["hostid"]
  454. network := params["network"]
  455. if hostid == "" || network == "" {
  456. logic.ReturnErrorResponse(
  457. w,
  458. r,
  459. logic.FormatError(errors.New("hostid or network cannot be empty"), "badrequest"),
  460. )
  461. return
  462. }
  463. // confirm host exists
  464. currHost, err := logic.GetHost(hostid)
  465. if err != nil {
  466. logger.Log(0, r.Header.Get("user"), "failed to find host:", hostid, err.Error())
  467. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  468. return
  469. }
  470. newNode, err := logic.UpdateHostNetwork(currHost, network, true)
  471. if err != nil {
  472. logger.Log(
  473. 0,
  474. r.Header.Get("user"),
  475. "failed to add host to network:",
  476. hostid,
  477. network,
  478. err.Error(),
  479. )
  480. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  481. return
  482. }
  483. logger.Log(1, "added new node", newNode.ID.String(), "to host", currHost.Name)
  484. if currHost.IsDefault {
  485. // make host failover
  486. logic.CreateFailOver(*newNode)
  487. // make host remote access gateway
  488. logic.CreateIngressGateway(network, newNode.ID.String(), models.IngressRequest{})
  489. }
  490. go func() {
  491. mq.HostUpdate(&models.HostUpdate{
  492. Action: models.JoinHostToNetwork,
  493. Host: *currHost,
  494. Node: *newNode,
  495. })
  496. mq.PublishPeerUpdate(false)
  497. if servercfg.IsDNSMode() {
  498. logic.SetDNS()
  499. }
  500. }()
  501. logger.Log(
  502. 2,
  503. r.Header.Get("user"),
  504. fmt.Sprintf("added host %s to network %s", currHost.Name, network),
  505. )
  506. logic.LogEvent(&models.Event{
  507. Action: models.JoinHostToNet,
  508. Source: models.Subject{
  509. ID: r.Header.Get("user"),
  510. Name: r.Header.Get("user"),
  511. Type: models.UserSub,
  512. },
  513. TriggeredBy: r.Header.Get("user"),
  514. Target: models.Subject{
  515. ID: currHost.ID.String(),
  516. Name: currHost.Name,
  517. Type: models.DeviceSub,
  518. },
  519. NetworkID: models.NetworkID(network),
  520. Origin: models.Dashboard,
  521. })
  522. w.WriteHeader(http.StatusOK)
  523. }
  524. // @Summary To Remove Host from Network
  525. // @Router /api/hosts/{hostid}/networks/{network} [delete]
  526. // @Tags Hosts
  527. // @Security oauth
  528. // @Param hostid path string true "Host ID"
  529. // @Param network path string true "Network name"
  530. // @Param force query bool false "Force delete"
  531. // @Success 200 {string} string "OK"
  532. // @Failure 500 {object} models.ErrorResponse
  533. func deleteHostFromNetwork(w http.ResponseWriter, r *http.Request) {
  534. var params = mux.Vars(r)
  535. hostid := params["hostid"]
  536. network := params["network"]
  537. forceDelete := r.URL.Query().Get("force") == "true"
  538. if hostid == "" || network == "" {
  539. logic.ReturnErrorResponse(
  540. w,
  541. r,
  542. logic.FormatError(errors.New("hostid or network cannot be empty"), "badrequest"),
  543. )
  544. return
  545. }
  546. // confirm host exists
  547. currHost, err := logic.GetHost(hostid)
  548. if err != nil {
  549. if database.IsEmptyRecord(err) {
  550. // check if there is any daemon nodes that needs to be deleted
  551. node, err := logic.GetNodeByHostRef(hostid, network)
  552. if err != nil {
  553. slog.Error(
  554. "couldn't get node for host",
  555. "hostid",
  556. hostid,
  557. "network",
  558. network,
  559. "error",
  560. err,
  561. )
  562. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  563. return
  564. }
  565. if err = logic.DeleteNodeByID(&node); err != nil {
  566. slog.Error("failed to force delete daemon node",
  567. "nodeid", node.ID.String(), "hostid", hostid, "network", network, "error", err)
  568. logic.ReturnErrorResponse(
  569. w,
  570. r,
  571. logic.FormatError(
  572. fmt.Errorf("failed to force delete daemon node: %s", err.Error()),
  573. "internal",
  574. ),
  575. )
  576. return
  577. }
  578. logic.ReturnSuccessResponse(w, r, "force deleted daemon node successfully")
  579. return
  580. }
  581. logger.Log(0, r.Header.Get("user"), "failed to find host:", err.Error())
  582. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  583. return
  584. }
  585. node, err := logic.UpdateHostNetwork(currHost, network, false)
  586. if err != nil {
  587. if node == nil && forceDelete {
  588. // force cleanup the node
  589. node, err := logic.GetNodeByHostRef(hostid, network)
  590. if err != nil {
  591. slog.Error(
  592. "couldn't get node for host",
  593. "hostid",
  594. hostid,
  595. "network",
  596. network,
  597. "error",
  598. err,
  599. )
  600. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  601. return
  602. }
  603. if err = logic.DeleteNodeByID(&node); err != nil {
  604. slog.Error("failed to force delete daemon node",
  605. "nodeid", node.ID.String(), "hostid", hostid, "network", network, "error", err)
  606. logic.ReturnErrorResponse(
  607. w,
  608. r,
  609. logic.FormatError(
  610. fmt.Errorf("failed to force delete daemon node: %s", err.Error()),
  611. "internal",
  612. ),
  613. )
  614. return
  615. }
  616. logic.ReturnSuccessResponse(w, r, "force deleted daemon node successfully")
  617. return
  618. }
  619. logger.Log(
  620. 0,
  621. r.Header.Get("user"),
  622. "failed to remove host from network:",
  623. hostid,
  624. network,
  625. err.Error(),
  626. )
  627. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  628. return
  629. }
  630. var gwClients []models.ExtClient
  631. if node.IsIngressGateway {
  632. gwClients = logic.GetGwExtclients(node.ID.String(), node.Network)
  633. }
  634. logger.Log(1, "deleting node", node.ID.String(), "from host", currHost.Name)
  635. if err := logic.DeleteNode(node, forceDelete); err != nil {
  636. logic.ReturnErrorResponse(
  637. w,
  638. r,
  639. logic.FormatError(fmt.Errorf("failed to delete node"), "internal"),
  640. )
  641. return
  642. }
  643. go func() {
  644. mq.PublishMqUpdatesForDeletedNode(*node, true, gwClients)
  645. if servercfg.IsDNSMode() {
  646. logic.SetDNS()
  647. }
  648. }()
  649. logic.LogEvent(&models.Event{
  650. Action: models.RemoveHostFromNet,
  651. Source: models.Subject{
  652. ID: r.Header.Get("user"),
  653. Name: r.Header.Get("user"),
  654. Type: models.UserSub,
  655. },
  656. TriggeredBy: r.Header.Get("user"),
  657. Target: models.Subject{
  658. ID: currHost.ID.String(),
  659. Name: currHost.Name,
  660. Type: models.DeviceSub,
  661. },
  662. NetworkID: models.NetworkID(network),
  663. Origin: models.Dashboard,
  664. })
  665. logger.Log(
  666. 2,
  667. r.Header.Get("user"),
  668. fmt.Sprintf("removed host %s from network %s", currHost.Name, network),
  669. )
  670. w.WriteHeader(http.StatusOK)
  671. }
  672. // @Summary To Fetch Auth Token for a Host
  673. // @Router /api/hosts/adm/authenticate [post]
  674. // @Tags Auth
  675. // @Accept json
  676. // @Param body body models.AuthParams true "Authentication parameters"
  677. // @Success 200 {object} models.SuccessResponse
  678. // @Failure 400 {object} models.ErrorResponse
  679. // @Failure 401 {object} models.ErrorResponse
  680. // @Failure 500 {object} models.ErrorResponse
  681. func authenticateHost(response http.ResponseWriter, request *http.Request) {
  682. var authRequest models.AuthParams
  683. var errorResponse = models.ErrorResponse{
  684. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  685. }
  686. decoder := json.NewDecoder(request.Body)
  687. decoderErr := decoder.Decode(&authRequest)
  688. defer request.Body.Close()
  689. if decoderErr != nil {
  690. errorResponse.Code = http.StatusBadRequest
  691. errorResponse.Message = decoderErr.Error()
  692. logger.Log(0, request.Header.Get("user"), "error decoding request body: ",
  693. decoderErr.Error())
  694. logic.ReturnErrorResponse(response, request, errorResponse)
  695. return
  696. }
  697. errorResponse.Code = http.StatusBadRequest
  698. if authRequest.ID == "" {
  699. errorResponse.Message = "W1R3: ID can't be empty"
  700. logger.Log(0, request.Header.Get("user"), errorResponse.Message)
  701. logic.ReturnErrorResponse(response, request, errorResponse)
  702. return
  703. } else if authRequest.Password == "" {
  704. errorResponse.Message = "W1R3: Password can't be empty"
  705. logger.Log(0, request.Header.Get("user"), errorResponse.Message)
  706. logic.ReturnErrorResponse(response, request, errorResponse)
  707. return
  708. }
  709. host, err := logic.GetHost(authRequest.ID)
  710. if err != nil {
  711. errorResponse.Code = http.StatusBadRequest
  712. errorResponse.Message = err.Error()
  713. logger.Log(0, request.Header.Get("user"),
  714. "error retrieving host: ", authRequest.ID, err.Error())
  715. logic.ReturnErrorResponse(response, request, errorResponse)
  716. return
  717. }
  718. err = bcrypt.CompareHashAndPassword([]byte(host.HostPass), []byte(authRequest.Password))
  719. if err != nil {
  720. errorResponse.Code = http.StatusUnauthorized
  721. errorResponse.Message = "unauthorized"
  722. logger.Log(0, request.Header.Get("user"),
  723. "error validating user password: ", err.Error())
  724. logic.ReturnErrorResponse(response, request, errorResponse)
  725. return
  726. }
  727. tokenString, err := logic.CreateJWT(authRequest.ID, authRequest.MacAddress, "")
  728. if tokenString == "" {
  729. errorResponse.Code = http.StatusUnauthorized
  730. errorResponse.Message = "unauthorized"
  731. logger.Log(0, request.Header.Get("user"),
  732. fmt.Sprintf("%s: %v", errorResponse.Message, err))
  733. logic.ReturnErrorResponse(response, request, errorResponse)
  734. return
  735. }
  736. var successResponse = models.SuccessResponse{
  737. Code: http.StatusOK,
  738. Message: "W1R3: Host " + authRequest.ID + " Authorized",
  739. Response: models.SuccessfulLoginResponse{
  740. AuthToken: tokenString,
  741. ID: authRequest.ID,
  742. },
  743. }
  744. successJSONResponse, jsonError := json.Marshal(successResponse)
  745. if jsonError != nil {
  746. errorResponse.Code = http.StatusBadRequest
  747. errorResponse.Message = err.Error()
  748. logger.Log(0, request.Header.Get("user"),
  749. "error marshalling resp: ", err.Error())
  750. logic.ReturnErrorResponse(response, request, errorResponse)
  751. return
  752. }
  753. go func() {
  754. // Create EMQX creds
  755. if servercfg.GetBrokerType() == servercfg.EmqxBrokerType {
  756. if err := mq.GetEmqxHandler().CreateEmqxUser(host.ID.String(), authRequest.Password); err != nil {
  757. slog.Error("failed to create host credentials for EMQX: ", err.Error())
  758. }
  759. }
  760. }()
  761. response.WriteHeader(http.StatusOK)
  762. response.Header().Set("Content-Type", "application/json")
  763. response.Write(successJSONResponse)
  764. }
  765. // @Summary Send signal to peer
  766. // @Router /api/v1/host/{hostid}/signalpeer [post]
  767. // @Tags Hosts
  768. // @Security oauth
  769. // @Param hostid path string true "Host ID"
  770. // @Param body body models.Signal true "Signal data"
  771. // @Success 200 {object} models.Signal
  772. // @Failure 400 {object} models.ErrorResponse
  773. func signalPeer(w http.ResponseWriter, r *http.Request) {
  774. var params = mux.Vars(r)
  775. hostid := params["hostid"]
  776. // confirm host exists
  777. _, err := logic.GetHost(hostid)
  778. if err != nil {
  779. logger.Log(0, r.Header.Get("user"), "failed to get host:", err.Error())
  780. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  781. return
  782. }
  783. var signal models.Signal
  784. w.Header().Set("Content-Type", "application/json")
  785. err = json.NewDecoder(r.Body).Decode(&signal)
  786. if err != nil {
  787. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  788. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  789. return
  790. }
  791. if signal.ToHostPubKey == "" {
  792. msg := "insufficient data to signal peer"
  793. logger.Log(0, r.Header.Get("user"), msg)
  794. logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New(msg), "badrequest"))
  795. return
  796. }
  797. signal.IsPro = servercfg.IsPro
  798. peerHost, err := logic.GetHost(signal.ToHostID)
  799. if err != nil {
  800. logic.ReturnErrorResponse(
  801. w,
  802. r,
  803. logic.FormatError(errors.New("failed to signal, peer not found"), "badrequest"),
  804. )
  805. return
  806. }
  807. err = mq.HostUpdate(&models.HostUpdate{
  808. Action: models.SignalHost,
  809. Host: *peerHost,
  810. Signal: signal,
  811. })
  812. if err != nil {
  813. logic.ReturnErrorResponse(
  814. w,
  815. r,
  816. logic.FormatError(
  817. errors.New("failed to publish signal to peer: "+err.Error()),
  818. "badrequest",
  819. ),
  820. )
  821. return
  822. }
  823. w.WriteHeader(http.StatusOK)
  824. json.NewEncoder(w).Encode(signal)
  825. }
  826. // @Summary Update keys for all hosts
  827. // @Router /api/hosts/keys [put]
  828. // @Tags Hosts
  829. // @Security oauth
  830. // @Success 200 {string} string "OK"
  831. // @Failure 400 {object} models.ErrorResponse
  832. func updateAllKeys(w http.ResponseWriter, r *http.Request) {
  833. var errorResponse = models.ErrorResponse{}
  834. w.Header().Set("Content-Type", "application/json")
  835. hosts, err := logic.GetAllHosts()
  836. if err != nil {
  837. errorResponse.Code = http.StatusBadRequest
  838. errorResponse.Message = err.Error()
  839. logger.Log(0, r.Header.Get("user"),
  840. "error retrieving hosts ", err.Error())
  841. logic.ReturnErrorResponse(w, r, errorResponse)
  842. return
  843. }
  844. go func() {
  845. hostUpdate := models.HostUpdate{}
  846. hostUpdate.Action = models.UpdateKeys
  847. for _, host := range hosts {
  848. hostUpdate.Host = host
  849. logger.Log(2, "updating host", host.ID.String(), " for a key update")
  850. if err = mq.HostUpdate(&hostUpdate); err != nil {
  851. logger.Log(
  852. 0,
  853. "failed to send update to node during a network wide key update",
  854. host.ID.String(),
  855. err.Error(),
  856. )
  857. }
  858. }
  859. }()
  860. logic.LogEvent(&models.Event{
  861. Action: models.RefreshAllKeys,
  862. Source: models.Subject{
  863. ID: r.Header.Get("user"),
  864. Name: r.Header.Get("user"),
  865. Type: models.UserSub,
  866. },
  867. TriggeredBy: r.Header.Get("user"),
  868. Target: models.Subject{
  869. ID: "All Devices",
  870. Name: "All Devices",
  871. Type: models.DeviceSub,
  872. },
  873. Origin: models.Dashboard,
  874. })
  875. logger.Log(2, r.Header.Get("user"), "updated keys for all hosts")
  876. w.WriteHeader(http.StatusOK)
  877. }
  878. // @Summary Update keys for a host
  879. // @Router /api/hosts/{hostid}/keys [put]
  880. // @Tags Hosts
  881. // @Security oauth
  882. // @Param hostid path string true "Host ID"
  883. // @Success 200 {string} string "OK"
  884. // @Failure 400 {object} models.ErrorResponse
  885. func updateKeys(w http.ResponseWriter, r *http.Request) {
  886. var errorResponse = models.ErrorResponse{}
  887. w.Header().Set("Content-Type", "application/json")
  888. var params = mux.Vars(r)
  889. hostid := params["hostid"]
  890. host, err := logic.GetHost(hostid)
  891. if err != nil {
  892. logger.Log(0, "failed to retrieve host", hostid, err.Error())
  893. errorResponse.Code = http.StatusBadRequest
  894. errorResponse.Message = err.Error()
  895. logger.Log(0, r.Header.Get("user"),
  896. "error retrieving hosts ", err.Error())
  897. logic.ReturnErrorResponse(w, r, errorResponse)
  898. return
  899. }
  900. go func() {
  901. hostUpdate := models.HostUpdate{
  902. Action: models.UpdateKeys,
  903. Host: *host,
  904. }
  905. if err = mq.HostUpdate(&hostUpdate); err != nil {
  906. logger.Log(0, "failed to send host key update", host.ID.String(), err.Error())
  907. }
  908. }()
  909. logic.LogEvent(&models.Event{
  910. Action: models.RefreshKey,
  911. Source: models.Subject{
  912. ID: r.Header.Get("user"),
  913. Name: r.Header.Get("user"),
  914. Type: models.UserSub,
  915. },
  916. TriggeredBy: r.Header.Get("user"),
  917. Target: models.Subject{
  918. ID: host.ID.String(),
  919. Name: host.Name,
  920. Type: models.DeviceSub,
  921. },
  922. Origin: models.Dashboard,
  923. })
  924. logger.Log(2, r.Header.Get("user"), "updated key on host", host.Name)
  925. w.WriteHeader(http.StatusOK)
  926. }
  927. // @Summary Requests all the hosts to pull
  928. // @Router /api/hosts/sync [post]
  929. // @Tags Hosts
  930. // @Security oauth
  931. // @Success 200 {string} string "sync all hosts request received"
  932. func syncHosts(w http.ResponseWriter, r *http.Request) {
  933. w.Header().Set("Content-Type", "application/json")
  934. user := r.Header.Get("user")
  935. go func() {
  936. slog.Info("requesting all hosts to sync", "user", user)
  937. hosts, err := logic.GetAllHosts()
  938. if err != nil {
  939. slog.Error("failed to retrieve all hosts", "user", user, "error", err)
  940. return
  941. }
  942. for _, host := range hosts {
  943. go func(host models.Host) {
  944. hostUpdate := models.HostUpdate{
  945. Action: models.RequestPull,
  946. Host: host,
  947. }
  948. if err = mq.HostUpdate(&hostUpdate); err != nil {
  949. slog.Error("failed to request host to sync", "user", user, "host", host.ID.String(), "error", err)
  950. } else {
  951. slog.Info("host sync requested", "user", user, "host", host.ID.String())
  952. }
  953. }(host)
  954. time.Sleep(time.Millisecond * 100)
  955. }
  956. }()
  957. logic.LogEvent(&models.Event{
  958. Action: models.SyncAll,
  959. Source: models.Subject{
  960. ID: r.Header.Get("user"),
  961. Name: r.Header.Get("user"),
  962. Type: models.UserSub,
  963. },
  964. TriggeredBy: r.Header.Get("user"),
  965. Target: models.Subject{
  966. ID: "All Devices",
  967. Name: "All Devices",
  968. Type: models.DeviceSub,
  969. },
  970. Origin: models.Dashboard,
  971. })
  972. slog.Info("sync all hosts request received", "user", user)
  973. logic.ReturnSuccessResponse(w, r, "sync all hosts request received")
  974. }
  975. // @Summary Requests a host to pull
  976. // @Router /api/hosts/{hostid}/sync [post]
  977. // @Tags Hosts
  978. // @Security oauth
  979. // @Param hostid path string true "Host ID"
  980. // @Success 200 {string} string "OK"
  981. // @Failure 400 {object} models.ErrorResponse
  982. func syncHost(w http.ResponseWriter, r *http.Request) {
  983. hostId := mux.Vars(r)["hostid"]
  984. var errorResponse = models.ErrorResponse{}
  985. w.Header().Set("Content-Type", "application/json")
  986. host, err := logic.GetHost(hostId)
  987. if err != nil {
  988. slog.Error("failed to retrieve host", "user", r.Header.Get("user"), "error", err)
  989. errorResponse.Code = http.StatusBadRequest
  990. errorResponse.Message = err.Error()
  991. logic.ReturnErrorResponse(w, r, errorResponse)
  992. return
  993. }
  994. go func() {
  995. hostUpdate := models.HostUpdate{
  996. Action: models.RequestPull,
  997. Host: *host,
  998. }
  999. if err = mq.HostUpdate(&hostUpdate); err != nil {
  1000. slog.Error("failed to send host pull request", "host", host.ID.String(), "error", err)
  1001. }
  1002. }()
  1003. logic.LogEvent(&models.Event{
  1004. Action: models.Sync,
  1005. Source: models.Subject{
  1006. ID: r.Header.Get("user"),
  1007. Name: r.Header.Get("user"),
  1008. Type: models.UserSub,
  1009. },
  1010. TriggeredBy: r.Header.Get("user"),
  1011. Target: models.Subject{
  1012. ID: host.ID.String(),
  1013. Name: host.Name,
  1014. Type: models.DeviceSub,
  1015. },
  1016. Origin: models.Dashboard,
  1017. })
  1018. slog.Info("requested host pull", "user", r.Header.Get("user"), "host", host.ID.String())
  1019. w.WriteHeader(http.StatusOK)
  1020. }
  1021. // @Summary Deletes all EMQX hosts
  1022. // @Router /api/emqx/hosts [delete]
  1023. // @Tags Hosts
  1024. // @Security oauth
  1025. // @Success 200 {string} string "deleted hosts data on emqx"
  1026. // @Failure 500 {object} models.ErrorResponse
  1027. func delEmqxHosts(w http.ResponseWriter, r *http.Request) {
  1028. currentHosts, err := logic.GetAllHosts()
  1029. if err != nil {
  1030. logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
  1031. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  1032. return
  1033. }
  1034. for _, host := range currentHosts {
  1035. // delete EMQX credentials for host
  1036. if err := mq.GetEmqxHandler().DeleteEmqxUser(host.ID.String()); err != nil {
  1037. slog.Error("failed to remove host credentials from EMQX", "id", host.ID, "error", err)
  1038. }
  1039. }
  1040. err = mq.GetEmqxHandler().DeleteEmqxUser(servercfg.GetMqUserName())
  1041. if err != nil {
  1042. slog.Error(
  1043. "failed to remove server credentials from EMQX",
  1044. "user",
  1045. servercfg.GetMqUserName(),
  1046. "error",
  1047. err,
  1048. )
  1049. }
  1050. logic.ReturnSuccessResponse(w, r, "deleted hosts data on emqx")
  1051. }
  1052. // @Summary Fetches host peerinfo
  1053. // @Router /api/host/{hostid}/peer_info [get]
  1054. // @Tags Hosts
  1055. // @Security oauth
  1056. // @Param hostid path string true "Host ID"
  1057. // @Success 200 {object} models.SuccessResponse
  1058. // @Failure 500 {object} models.ErrorResponse
  1059. func getHostPeerInfo(w http.ResponseWriter, r *http.Request) {
  1060. hostId := mux.Vars(r)["hostid"]
  1061. var errorResponse = models.ErrorResponse{}
  1062. host, err := logic.GetHost(hostId)
  1063. if err != nil {
  1064. slog.Error("failed to retrieve host", "error", err)
  1065. errorResponse.Code = http.StatusBadRequest
  1066. errorResponse.Message = err.Error()
  1067. logic.ReturnErrorResponse(w, r, errorResponse)
  1068. return
  1069. }
  1070. peerInfo, err := logic.GetHostPeerInfo(host)
  1071. if err != nil {
  1072. slog.Error("failed to retrieve host peerinfo", "error", err)
  1073. errorResponse.Code = http.StatusBadRequest
  1074. errorResponse.Message = err.Error()
  1075. logic.ReturnErrorResponse(w, r, errorResponse)
  1076. return
  1077. }
  1078. logic.ReturnSuccessResponseWithJson(w, r, peerInfo, "fetched host peer info")
  1079. }