hosts.go 22 KB

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