hosts.go 29 KB

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