hosts.go 29 KB

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