hosts.go 26 KB

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