node.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807
  1. package controller
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "net/http"
  6. "strings"
  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. var hostIDHeader = "host-id"
  18. func nodeHandlers(r *mux.Router) {
  19. r.HandleFunc("/api/nodes", logic.SecurityCheck(true, http.HandlerFunc(getAllNodes))).Methods(http.MethodGet)
  20. r.HandleFunc("/api/nodes/{network}", logic.SecurityCheck(true, http.HandlerFunc(getNetworkNodes))).Methods(http.MethodGet)
  21. r.HandleFunc("/api/nodes/{network}/{nodeid}", Authorize(true, true, "node", http.HandlerFunc(getNode))).Methods(http.MethodGet)
  22. r.HandleFunc("/api/nodes/{network}/{nodeid}", logic.SecurityCheck(true, http.HandlerFunc(updateNode))).Methods(http.MethodPut)
  23. r.HandleFunc("/api/nodes/{network}/{nodeid}", Authorize(true, true, "node", http.HandlerFunc(deleteNode))).Methods(http.MethodDelete)
  24. r.HandleFunc("/api/nodes/{network}/{nodeid}/creategateway", logic.SecurityCheck(true, checkFreeTierLimits(limitChoiceEgress, http.HandlerFunc(createEgressGateway)))).Methods(http.MethodPost)
  25. r.HandleFunc("/api/nodes/{network}/{nodeid}/deletegateway", logic.SecurityCheck(true, http.HandlerFunc(deleteEgressGateway))).Methods(http.MethodDelete)
  26. r.HandleFunc("/api/nodes/{network}/{nodeid}/createingress", logic.SecurityCheck(true, checkFreeTierLimits(limitChoiceIngress, http.HandlerFunc(createIngressGateway)))).Methods(http.MethodPost)
  27. r.HandleFunc("/api/nodes/{network}/{nodeid}/deleteingress", logic.SecurityCheck(true, http.HandlerFunc(deleteIngressGateway))).Methods(http.MethodDelete)
  28. r.HandleFunc("/api/nodes/adm/{network}/authenticate", authenticate).Methods(http.MethodPost)
  29. r.HandleFunc("/api/v1/nodes/migrate", migrate).Methods(http.MethodPost)
  30. }
  31. func authenticate(response http.ResponseWriter, request *http.Request) {
  32. var authRequest models.AuthParams
  33. var result models.Node
  34. var errorResponse = models.ErrorResponse{
  35. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  36. }
  37. decoder := json.NewDecoder(request.Body)
  38. decoderErr := decoder.Decode(&authRequest)
  39. defer request.Body.Close()
  40. if decoderErr != nil {
  41. errorResponse.Code = http.StatusBadRequest
  42. errorResponse.Message = decoderErr.Error()
  43. logger.Log(0, request.Header.Get("user"), "error decoding request body: ",
  44. decoderErr.Error())
  45. logic.ReturnErrorResponse(response, request, errorResponse)
  46. return
  47. }
  48. errorResponse.Code = http.StatusBadRequest
  49. if authRequest.ID == "" {
  50. errorResponse.Message = "W1R3: ID can't be empty"
  51. logger.Log(0, request.Header.Get("user"), errorResponse.Message)
  52. logic.ReturnErrorResponse(response, request, errorResponse)
  53. return
  54. } else if authRequest.Password == "" {
  55. errorResponse.Message = "W1R3: Password can't be empty"
  56. logger.Log(0, request.Header.Get("user"), errorResponse.Message)
  57. logic.ReturnErrorResponse(response, request, errorResponse)
  58. return
  59. }
  60. var err error
  61. result, err = logic.GetNodeByID(authRequest.ID)
  62. if err != nil {
  63. result, err = logic.GetDeletedNodeByID(authRequest.ID)
  64. if err != nil {
  65. errorResponse.Code = http.StatusBadRequest
  66. errorResponse.Message = err.Error()
  67. logger.Log(0, request.Header.Get("user"),
  68. fmt.Sprintf("failed to get node info [%s]: %v", authRequest.ID, err))
  69. logic.ReturnErrorResponse(response, request, errorResponse)
  70. return
  71. }
  72. }
  73. host, err := logic.GetHost(result.HostID.String())
  74. if err != nil {
  75. errorResponse.Code = http.StatusBadRequest
  76. errorResponse.Message = err.Error()
  77. logger.Log(0, request.Header.Get("user"),
  78. "error retrieving host: ", result.HostID.String(), err.Error())
  79. logic.ReturnErrorResponse(response, request, errorResponse)
  80. return
  81. }
  82. err = bcrypt.CompareHashAndPassword([]byte(host.HostPass), []byte(authRequest.Password))
  83. if err != nil {
  84. errorResponse.Code = http.StatusBadRequest
  85. errorResponse.Message = err.Error()
  86. logger.Log(0, request.Header.Get("user"),
  87. "error validating user password: ", err.Error())
  88. logic.ReturnErrorResponse(response, request, errorResponse)
  89. return
  90. }
  91. tokenString, err := logic.CreateJWT(authRequest.ID, authRequest.MacAddress, result.Network)
  92. if tokenString == "" {
  93. errorResponse.Code = http.StatusBadRequest
  94. errorResponse.Message = "Could not create Token"
  95. logger.Log(0, request.Header.Get("user"),
  96. fmt.Sprintf("%s: %v", errorResponse.Message, err))
  97. logic.ReturnErrorResponse(response, request, errorResponse)
  98. return
  99. }
  100. var successResponse = models.SuccessResponse{
  101. Code: http.StatusOK,
  102. Message: "W1R3: Device " + authRequest.ID + " Authorized",
  103. Response: models.SuccessfulLoginResponse{
  104. AuthToken: tokenString,
  105. ID: authRequest.ID,
  106. },
  107. }
  108. successJSONResponse, jsonError := json.Marshal(successResponse)
  109. if jsonError != nil {
  110. errorResponse.Code = http.StatusBadRequest
  111. errorResponse.Message = err.Error()
  112. logger.Log(0, request.Header.Get("user"),
  113. "error marshalling resp: ", err.Error())
  114. logic.ReturnErrorResponse(response, request, errorResponse)
  115. return
  116. }
  117. response.WriteHeader(http.StatusOK)
  118. response.Header().Set("Content-Type", "application/json")
  119. response.Write(successJSONResponse)
  120. }
  121. // The middleware for most requests to the API
  122. // They all pass through here first
  123. // This will validate the JWT (or check for master token)
  124. // This will also check against the authNetwork and make sure the node should be accessing that endpoint,
  125. // even if it's technically ok
  126. // This is kind of a poor man's RBAC. There's probably a better/smarter way.
  127. // TODO: Consider better RBAC implementations
  128. func Authorize(
  129. hostAllowed, networkCheck bool,
  130. authNetwork string,
  131. next http.Handler,
  132. ) http.HandlerFunc {
  133. return func(w http.ResponseWriter, r *http.Request) {
  134. var errorResponse = models.ErrorResponse{
  135. Code: http.StatusForbidden, Message: logic.Forbidden_Msg,
  136. }
  137. var params = mux.Vars(r)
  138. networkexists, _ := logic.NetworkExists(params["network"])
  139. //check that the request is for a valid network
  140. //if (networkCheck && !networkexists) || err != nil {
  141. if networkCheck && !networkexists {
  142. logic.ReturnErrorResponse(w, r, errorResponse)
  143. return
  144. } else {
  145. w.Header().Set("Content-Type", "application/json")
  146. //get the auth token
  147. bearerToken := r.Header.Get("Authorization")
  148. var tokenSplit = strings.Split(bearerToken, " ")
  149. //I put this in in case the user doesn't put in a token at all (in which case it's empty)
  150. //There's probably a smarter way of handling this.
  151. var authToken = "928rt238tghgwe@TY@$Y@#WQAEGB2FC#@HG#@$Hddd"
  152. if len(tokenSplit) > 1 {
  153. authToken = tokenSplit[1]
  154. } else {
  155. logic.ReturnErrorResponse(w, r, errorResponse)
  156. return
  157. }
  158. // check if host instead of user
  159. if hostAllowed {
  160. // TODO --- should ensure that node is only operating on itself
  161. if hostID, _, _, err := logic.VerifyHostToken(authToken); err == nil {
  162. r.Header.Set(hostIDHeader, hostID)
  163. // this indicates request is from a node
  164. // used for failover - if a getNode comes from node, this will trigger a metrics wipe
  165. next.ServeHTTP(w, r)
  166. return
  167. }
  168. }
  169. var isAuthorized = false
  170. var nodeID = ""
  171. username, issuperadmin, isadmin, errN := logic.VerifyUserToken(authToken)
  172. if errN != nil {
  173. logic.ReturnErrorResponse(w, r, logic.FormatError(errN, logic.Unauthorized_Msg))
  174. return
  175. }
  176. isnetadmin := issuperadmin || isadmin
  177. if issuperadmin || isadmin {
  178. nodeID = "mastermac"
  179. isAuthorized = true
  180. r.Header.Set("ismasterkey", "yes")
  181. }
  182. //The mastermac (login with masterkey from config) can do everything!! May be dangerous.
  183. if nodeID == "mastermac" {
  184. isAuthorized = true
  185. r.Header.Set("ismasterkey", "yes")
  186. //for everyone else, there's poor man's RBAC. The "cases" are defined in the routes in the handlers
  187. //So each route defines which access network should be allowed to access it
  188. } else {
  189. switch authNetwork {
  190. case "all":
  191. isAuthorized = true
  192. case "nodes":
  193. isAuthorized = (nodeID != "") || isnetadmin
  194. case "network":
  195. if isnetadmin {
  196. isAuthorized = true
  197. } else {
  198. node, err := logic.GetNodeByID(nodeID)
  199. if err != nil {
  200. logic.ReturnErrorResponse(w, r, errorResponse)
  201. return
  202. }
  203. isAuthorized = (node.Network == params["network"])
  204. }
  205. case "node":
  206. if isnetadmin {
  207. isAuthorized = true
  208. } else {
  209. isAuthorized = (nodeID == params["netid"])
  210. }
  211. case "host":
  212. case "user":
  213. isAuthorized = true
  214. default:
  215. isAuthorized = false
  216. }
  217. }
  218. if !isAuthorized {
  219. logic.ReturnErrorResponse(w, r, errorResponse)
  220. return
  221. } else {
  222. //If authorized, this function passes along it's request and output to the appropriate route function.
  223. if username == "" {
  224. username = "(user not found)"
  225. }
  226. r.Header.Set("user", username)
  227. next.ServeHTTP(w, r)
  228. }
  229. }
  230. }
  231. }
  232. // @Summary Gets all nodes associated with network including pending nodes
  233. // @Router /api/nodes/adm/{network} [get]
  234. // @Securitydefinitions.oauth2.application OAuth2Application
  235. // @Tags Nodes
  236. // @Success 200 {array} models.Node
  237. // @Failure 500 {object} models.ErrorResponse
  238. func getNetworkNodes(w http.ResponseWriter, r *http.Request) {
  239. w.Header().Set("Content-Type", "application/json")
  240. var params = mux.Vars(r)
  241. networkName := params["network"]
  242. nodes, err := logic.GetNetworkNodes(networkName)
  243. if err != nil {
  244. logger.Log(0, r.Header.Get("user"),
  245. fmt.Sprintf("error fetching nodes on network %s: %v", networkName, err))
  246. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  247. return
  248. }
  249. filteredNodes := []models.Node{}
  250. if r.Header.Get("ismaster") != "yes" {
  251. username := r.Header.Get("user")
  252. user, err := logic.GetUser(username)
  253. if err != nil {
  254. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  255. return
  256. }
  257. userPlatformRole, err := logic.GetRole(user.PlatformRoleID)
  258. if err != nil {
  259. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  260. return
  261. }
  262. if !userPlatformRole.FullAccess {
  263. nodesMap := make(map[string]struct{})
  264. networkRoles := user.NetworkRoles[models.NetworkID(networkName)]
  265. for networkRoleID := range networkRoles {
  266. userPermTemplate, err := logic.GetRole(networkRoleID)
  267. if err != nil {
  268. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  269. return
  270. }
  271. if userPermTemplate.FullAccess {
  272. break
  273. }
  274. if rsrcPerms, ok := userPermTemplate.NetworkLevelAccess[models.RemoteAccessGwRsrc]; ok {
  275. if _, ok := rsrcPerms[models.AllRemoteAccessGwRsrcID]; ok {
  276. for _, node := range nodes {
  277. if _, ok := nodesMap[node.ID.String()]; ok {
  278. continue
  279. }
  280. if node.IsIngressGateway {
  281. nodesMap[node.ID.String()] = struct{}{}
  282. filteredNodes = append(filteredNodes, node)
  283. }
  284. }
  285. } else {
  286. for gwID, scope := range rsrcPerms {
  287. if _, ok := nodesMap[gwID.String()]; ok {
  288. continue
  289. }
  290. if scope.Read {
  291. gwNode, err := logic.GetNodeByID(gwID.String())
  292. if err == nil && gwNode.IsIngressGateway {
  293. filteredNodes = append(filteredNodes, gwNode)
  294. }
  295. }
  296. }
  297. }
  298. }
  299. }
  300. }
  301. }
  302. if len(filteredNodes) > 0 {
  303. nodes = filteredNodes
  304. }
  305. nodes = logic.AddStaticNodestoList(nodes)
  306. nodes = logic.AddStatusToNodes(nodes)
  307. // returns all the nodes in JSON/API format
  308. apiNodes := logic.GetAllNodesAPI(nodes[:])
  309. logger.Log(2, r.Header.Get("user"), "fetched nodes on network", networkName)
  310. w.WriteHeader(http.StatusOK)
  311. json.NewEncoder(w).Encode(apiNodes)
  312. }
  313. // @Summary Get all nodes across all networks
  314. // @Router /api/nodes [get]
  315. // @Tags Nodes
  316. // @Securitydefinitions.oauth2.application OAuth2Application
  317. // @Success 200 {array} models.ApiNode
  318. // @Failure 500 {object} models.ErrorResponse
  319. // Not quite sure if this is necessary. Probably necessary based on front end but may want to review after iteration 1 if it's being used or not
  320. func getAllNodes(w http.ResponseWriter, r *http.Request) {
  321. w.Header().Set("Content-Type", "application/json")
  322. var nodes []models.Node
  323. nodes, err := logic.GetAllNodes()
  324. if err != nil {
  325. logger.Log(0, "error fetching all nodes info: ", err.Error())
  326. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  327. return
  328. }
  329. username := r.Header.Get("user")
  330. if r.Header.Get("ismaster") == "no" {
  331. user, err := logic.GetUser(username)
  332. if err != nil {
  333. return
  334. }
  335. userPlatformRole, err := logic.GetRole(user.PlatformRoleID)
  336. if err != nil {
  337. return
  338. }
  339. if !userPlatformRole.FullAccess {
  340. nodes = logic.GetFilteredNodesByUserAccess(*user, nodes)
  341. }
  342. }
  343. nodes = logic.AddStaticNodestoList(nodes)
  344. nodes = logic.AddStatusToNodes(nodes)
  345. // return all the nodes in JSON/API format
  346. apiNodes := logic.GetAllNodesAPI(nodes[:])
  347. logger.Log(3, r.Header.Get("user"), "fetched all nodes they have access to")
  348. logic.SortApiNodes(apiNodes[:])
  349. w.WriteHeader(http.StatusOK)
  350. json.NewEncoder(w).Encode(apiNodes)
  351. }
  352. // @Summary Get an individual node
  353. // @Router /api/nodes/{network}/{nodeid} [get]
  354. // @Tags Nodes
  355. // @Security oauth2
  356. // @Success 200 {object} models.NodeGet
  357. // @Failure 500 {object} models.ErrorResponse
  358. func getNode(w http.ResponseWriter, r *http.Request) {
  359. // set header.
  360. w.Header().Set("Content-Type", "application/json")
  361. var params = mux.Vars(r)
  362. nodeid := params["nodeid"]
  363. node, err := logic.ValidateParams(nodeid, params["network"])
  364. if err != nil {
  365. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  366. return
  367. }
  368. host, err := logic.GetHost(node.HostID.String())
  369. if err != nil {
  370. logger.Log(0, r.Header.Get("user"),
  371. fmt.Sprintf("error fetching host for node [ %s ] info: %v", nodeid, err))
  372. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  373. return
  374. }
  375. allNodes, err := logic.GetAllNodes()
  376. if err != nil {
  377. logger.Log(
  378. 0,
  379. r.Header.Get("user"),
  380. fmt.Sprintf(
  381. "error fetching wg peers config for host [ %s ]: %v",
  382. host.ID.String(),
  383. err,
  384. ),
  385. )
  386. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  387. return
  388. }
  389. hostPeerUpdate, err := logic.GetPeerUpdateForHost(node.Network, host, allNodes, nil, nil)
  390. if err != nil && !database.IsEmptyRecord(err) {
  391. logger.Log(
  392. 0,
  393. r.Header.Get("user"),
  394. fmt.Sprintf(
  395. "error fetching wg peers config for host [ %s ]: %v",
  396. host.ID.String(),
  397. err,
  398. ),
  399. )
  400. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  401. return
  402. }
  403. net, _ := logic.GetNetwork(node.Network)
  404. node.NetworkName = net.Name
  405. server := servercfg.GetServerInfo()
  406. response := models.NodeGet{
  407. Node: node,
  408. Host: *host,
  409. HostPeers: hostPeerUpdate.Peers,
  410. Peers: hostPeerUpdate.NodePeers,
  411. ServerConfig: server,
  412. PeerIDs: hostPeerUpdate.PeerIDs,
  413. }
  414. logger.Log(2, r.Header.Get("user"), "fetched node", params["nodeid"])
  415. w.WriteHeader(http.StatusOK)
  416. json.NewEncoder(w).Encode(response)
  417. }
  418. // == EGRESS ==
  419. // @Summary Create an egress gateway
  420. // @Router /api/nodes/{network}/{nodeid}/creategateway [post]
  421. // @Tags Nodes
  422. // @Security oauth2
  423. // @Success 200 {object} models.ApiNode
  424. // @Failure 500 {object} models.ErrorResponse
  425. func createEgressGateway(w http.ResponseWriter, r *http.Request) {
  426. var gateway models.EgressGatewayRequest
  427. var params = mux.Vars(r)
  428. node, err := logic.ValidateParams(params["nodeid"], params["network"])
  429. if err != nil {
  430. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  431. return
  432. }
  433. w.Header().Set("Content-Type", "application/json")
  434. if err := json.NewDecoder(r.Body).Decode(&gateway); err != nil {
  435. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  436. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  437. return
  438. }
  439. gateway.NetID = params["network"]
  440. gateway.NodeID = params["nodeid"]
  441. err = logic.ValidateEgressRange(gateway)
  442. if err != nil {
  443. logger.Log(0, r.Header.Get("user"), "error validating egress range: ", err.Error())
  444. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  445. return
  446. }
  447. node, err = logic.CreateEgressGateway(gateway)
  448. if err != nil {
  449. logger.Log(0, r.Header.Get("user"),
  450. fmt.Sprintf("failed to create egress gateway on node [%s] on network [%s]: %v",
  451. gateway.NodeID, gateway.NetID, err))
  452. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  453. return
  454. }
  455. apiNode := node.ConvertToAPINode()
  456. logger.Log(
  457. 1,
  458. r.Header.Get("user"),
  459. "created egress gateway on node",
  460. gateway.NodeID,
  461. "on network",
  462. gateway.NetID,
  463. )
  464. w.WriteHeader(http.StatusOK)
  465. json.NewEncoder(w).Encode(apiNode)
  466. go func() {
  467. if err := mq.NodeUpdate(&node); err != nil {
  468. slog.Error("error publishing node update to node", "node", node.ID, "error", err)
  469. }
  470. mq.PublishPeerUpdate(false)
  471. }()
  472. }
  473. // @Summary Delete an egress gateway
  474. // @Router /api/nodes/{network}/{nodeid}/deletegateway [delete]
  475. // @Tags Nodes
  476. // @Security oauth2
  477. // @Success 200 {object} models.ApiNode
  478. // @Failure 500 {object} models.ErrorResponse
  479. func deleteEgressGateway(w http.ResponseWriter, r *http.Request) {
  480. w.Header().Set("Content-Type", "application/json")
  481. var params = mux.Vars(r)
  482. nodeid := params["nodeid"]
  483. netid := params["network"]
  484. node, err := logic.ValidateParams(nodeid, netid)
  485. if err != nil {
  486. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  487. return
  488. }
  489. node, err = logic.DeleteEgressGateway(netid, nodeid)
  490. if err != nil {
  491. logger.Log(0, r.Header.Get("user"),
  492. fmt.Sprintf("failed to delete egress gateway on node [%s] on network [%s]: %v",
  493. nodeid, netid, err))
  494. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  495. return
  496. }
  497. apiNode := node.ConvertToAPINode()
  498. logger.Log(
  499. 1,
  500. r.Header.Get("user"),
  501. "deleted egress gateway on node",
  502. nodeid,
  503. "on network",
  504. netid,
  505. )
  506. w.WriteHeader(http.StatusOK)
  507. json.NewEncoder(w).Encode(apiNode)
  508. go func() {
  509. if err := mq.NodeUpdate(&node); err != nil {
  510. slog.Error("error publishing node update to node", "node", node.ID, "error", err)
  511. }
  512. mq.PublishPeerUpdate(false)
  513. }()
  514. }
  515. // == INGRESS ==
  516. // @Summary Create an remote access gateway
  517. // @Router /api/nodes/{network}/{nodeid}/createingress [post]
  518. // @Tags Nodes
  519. // @Security oauth2
  520. // @Success 200 {object} models.ApiNode
  521. // @Failure 500 {object} models.ErrorResponse
  522. func createIngressGateway(w http.ResponseWriter, r *http.Request) {
  523. var params = mux.Vars(r)
  524. w.Header().Set("Content-Type", "application/json")
  525. nodeid := params["nodeid"]
  526. netid := params["network"]
  527. node, err := logic.ValidateParams(nodeid, netid)
  528. if err != nil {
  529. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  530. return
  531. }
  532. var request models.IngressRequest
  533. json.NewDecoder(r.Body).Decode(&request)
  534. node, err = logic.CreateIngressGateway(netid, nodeid, request)
  535. if err != nil {
  536. logger.Log(0, r.Header.Get("user"),
  537. fmt.Sprintf("failed to create ingress gateway on node [%s] on network [%s]: %v",
  538. nodeid, netid, err))
  539. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  540. return
  541. }
  542. apiNode := node.ConvertToAPINode()
  543. logger.Log(
  544. 1,
  545. r.Header.Get("user"),
  546. "created ingress gateway on node",
  547. nodeid,
  548. "on network",
  549. netid,
  550. )
  551. w.WriteHeader(http.StatusOK)
  552. json.NewEncoder(w).Encode(apiNode)
  553. go func() {
  554. if err := mq.NodeUpdate(&node); err != nil {
  555. slog.Error("error publishing node update to node", "node", node.ID, "error", err)
  556. }
  557. mq.PublishPeerUpdate(false)
  558. }()
  559. }
  560. // @Summary Delete an remote access gateway
  561. // @Router /api/nodes/{network}/{nodeid}/deleteingress [delete]
  562. // @Tags Nodes
  563. // @Security oauth2
  564. // @Success 200 {object} models.ApiNode
  565. // @Failure 500 {object} models.ErrorResponse
  566. func deleteIngressGateway(w http.ResponseWriter, r *http.Request) {
  567. w.Header().Set("Content-Type", "application/json")
  568. var params = mux.Vars(r)
  569. nodeid := params["nodeid"]
  570. netid := params["network"]
  571. node, err := logic.ValidateParams(nodeid, netid)
  572. if err != nil {
  573. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  574. return
  575. }
  576. node, removedClients, err := logic.DeleteIngressGateway(nodeid)
  577. if err != nil {
  578. logger.Log(0, r.Header.Get("user"),
  579. fmt.Sprintf("failed to delete ingress gateway on node [%s] on network [%s]: %v",
  580. nodeid, netid, err))
  581. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  582. return
  583. }
  584. apiNode := node.ConvertToAPINode()
  585. logger.Log(1, r.Header.Get("user"), "deleted ingress gateway", nodeid)
  586. w.WriteHeader(http.StatusOK)
  587. json.NewEncoder(w).Encode(apiNode)
  588. if len(removedClients) > 0 {
  589. host, err := logic.GetHost(node.HostID.String())
  590. if err == nil {
  591. allNodes, err := logic.GetAllNodes()
  592. if err != nil {
  593. return
  594. }
  595. go func() {
  596. if err := mq.PublishSingleHostPeerUpdate(host, allNodes, nil, removedClients[:], false, nil); err != nil {
  597. slog.Error("publishSingleHostUpdate", "host", host.Name, "error", err)
  598. }
  599. mq.PublishPeerUpdate(false)
  600. if err := mq.NodeUpdate(&node); err != nil {
  601. slog.Error(
  602. "error publishing node update to node",
  603. "node",
  604. node.ID,
  605. "error",
  606. err,
  607. )
  608. }
  609. if servercfg.IsDNSMode() {
  610. logic.SetDNS()
  611. }
  612. }()
  613. }
  614. }
  615. }
  616. // @Summary Update an individual node
  617. // @Router /api/nodes/{network}/{nodeid} [put]
  618. // @Tags Nodes
  619. // @Security oauth2
  620. // @Success 200 {object} models.ApiNode
  621. // @Failure 500 {object} models.ErrorResponse
  622. func updateNode(w http.ResponseWriter, r *http.Request) {
  623. w.Header().Set("Content-Type", "application/json")
  624. var params = mux.Vars(r)
  625. //start here
  626. nodeid := params["nodeid"]
  627. currentNode, err := logic.ValidateParams(nodeid, params["network"])
  628. if err != nil {
  629. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  630. return
  631. }
  632. var newData models.ApiNode
  633. // we decode our body request params
  634. err = json.NewDecoder(r.Body).Decode(&newData)
  635. if err != nil {
  636. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  637. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  638. return
  639. }
  640. err = logic.ValidateNodeIp(&currentNode, &newData)
  641. if err != nil {
  642. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  643. return
  644. }
  645. if !servercfg.IsPro {
  646. newData.AdditionalRagIps = []string{}
  647. }
  648. newNode := newData.ConvertToServerNode(&currentNode)
  649. if newNode == nil {
  650. logic.ReturnErrorResponse(
  651. w,
  652. r,
  653. logic.FormatError(fmt.Errorf("error converting node"), "badrequest"),
  654. )
  655. return
  656. }
  657. if newNode.IsInternetGateway != currentNode.IsInternetGateway {
  658. if newNode.IsInternetGateway {
  659. logic.SetInternetGw(newNode, models.InetNodeReq{})
  660. } else {
  661. logic.UnsetInternetGw(newNode)
  662. }
  663. }
  664. relayUpdate := logic.RelayUpdates(&currentNode, newNode)
  665. if relayUpdate && newNode.IsRelay {
  666. err = logic.ValidateRelay(models.RelayRequest{
  667. NodeID: newNode.ID.String(),
  668. NetID: newNode.Network,
  669. RelayedNodes: newNode.RelayedNodes,
  670. }, true)
  671. if err != nil {
  672. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  673. return
  674. }
  675. }
  676. _, err = logic.GetHost(newNode.HostID.String())
  677. if err != nil {
  678. logger.Log(0, r.Header.Get("user"),
  679. fmt.Sprintf("failed to get host for node [ %s ] info: %v", nodeid, err))
  680. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  681. return
  682. }
  683. ifaceDelta := logic.IfaceDelta(&currentNode, newNode)
  684. aclUpdate := currentNode.DefaultACL != newNode.DefaultACL
  685. err = logic.UpdateNode(&currentNode, newNode)
  686. if err != nil {
  687. logger.Log(0, r.Header.Get("user"),
  688. fmt.Sprintf("failed to update node info [ %s ] info: %v", nodeid, err))
  689. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  690. return
  691. }
  692. if relayUpdate {
  693. logic.UpdateRelayed(&currentNode, newNode)
  694. }
  695. apiNode := newNode.ConvertToAPINode()
  696. logger.Log(
  697. 1,
  698. r.Header.Get("user"),
  699. "updated node",
  700. currentNode.ID.String(),
  701. "on network",
  702. currentNode.Network,
  703. )
  704. w.WriteHeader(http.StatusOK)
  705. json.NewEncoder(w).Encode(apiNode)
  706. go func(aclUpdate, relayupdate bool, newNode *models.Node) {
  707. if err := mq.NodeUpdate(newNode); err != nil {
  708. slog.Error("error publishing node update to node", "node", newNode.ID, "error", err)
  709. }
  710. if aclUpdate || relayupdate || ifaceDelta {
  711. if err := mq.PublishPeerUpdate(false); err != nil {
  712. logger.Log(0, "error during node ACL update for node", newNode.ID.String())
  713. }
  714. }
  715. mq.PublishPeerUpdate(false)
  716. if servercfg.IsDNSMode() {
  717. logic.SetDNS()
  718. }
  719. }(aclUpdate, relayUpdate, newNode)
  720. }
  721. // @Summary Delete an individual node
  722. // @Router /api/nodes/{network}/{nodeid} [delete]
  723. // @Tags Nodes
  724. // @Security oauth2
  725. // @Success 200 {string} string "Node deleted."
  726. // @Failure 500 {object} models.ErrorResponse
  727. func deleteNode(w http.ResponseWriter, r *http.Request) {
  728. // Set header
  729. w.Header().Set("Content-Type", "application/json")
  730. // get params
  731. var params = mux.Vars(r)
  732. var nodeid = params["nodeid"]
  733. node, err := logic.ValidateParams(nodeid, params["network"])
  734. if err != nil {
  735. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  736. return
  737. }
  738. forceDelete := r.URL.Query().Get("force") == "true"
  739. fromNode := r.Header.Get("requestfrom") == "node"
  740. var gwClients []models.ExtClient
  741. if node.IsIngressGateway {
  742. gwClients = logic.GetGwExtclients(node.ID.String(), node.Network)
  743. }
  744. purge := forceDelete || fromNode
  745. if err := logic.DeleteNode(&node, purge); err != nil {
  746. logic.ReturnErrorResponse(
  747. w,
  748. r,
  749. logic.FormatError(fmt.Errorf("failed to delete node"), "internal"),
  750. )
  751. return
  752. }
  753. logic.ReturnSuccessResponse(w, r, nodeid+" deleted.")
  754. logger.Log(1, r.Header.Get("user"), "Deleted node", nodeid, "from network", params["network"])
  755. go mq.PublishMqUpdatesForDeletedNode(node, !fromNode, gwClients)
  756. }