hosts.go 25 KB

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