hosts.go 22 KB

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