hosts.go 21 KB

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