hosts.go 27 KB

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