hosts.go 24 KB

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