2
0

hosts.go 21 KB

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