hosts.go 24 KB

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