node.go 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200
  1. package controller
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "fmt"
  6. "net/http"
  7. "strings"
  8. "github.com/gorilla/mux"
  9. "github.com/gravitl/netclient/nmproxy/manager"
  10. "github.com/gravitl/netmaker/database"
  11. "github.com/gravitl/netmaker/logger"
  12. "github.com/gravitl/netmaker/logic"
  13. "github.com/gravitl/netmaker/logic/pro"
  14. "github.com/gravitl/netmaker/models"
  15. "github.com/gravitl/netmaker/models/promodels"
  16. "github.com/gravitl/netmaker/mq"
  17. "github.com/gravitl/netmaker/servercfg"
  18. "golang.org/x/crypto/bcrypt"
  19. )
  20. func nodeHandlers(r *mux.Router) {
  21. r.HandleFunc("/api/nodes", authorize(false, false, "user", http.HandlerFunc(getAllNodes))).Methods("GET")
  22. r.HandleFunc("/api/nodes/{network}", authorize(false, true, "network", http.HandlerFunc(getNetworkNodes))).Methods("GET")
  23. r.HandleFunc("/api/nodes/{network}/{nodeid}", authorize(true, true, "node", http.HandlerFunc(getNode))).Methods("GET")
  24. r.HandleFunc("/api/nodes/{network}/{nodeid}", authorize(false, true, "node", http.HandlerFunc(updateNode))).Methods("PUT")
  25. r.HandleFunc("/api/nodes/{network}/{nodeid}/migrate", authorize(true, true, "node", http.HandlerFunc(nodeNodeUpdate))).Methods("PUT")
  26. r.HandleFunc("/api/nodes/{network}/{nodeid}", authorize(true, true, "node", http.HandlerFunc(deleteNode))).Methods("DELETE")
  27. r.HandleFunc("/api/nodes/{network}/{nodeid}/createrelay", authorize(false, true, "user", http.HandlerFunc(createRelay))).Methods("POST")
  28. r.HandleFunc("/api/nodes/{network}/{nodeid}/deleterelay", authorize(false, true, "user", http.HandlerFunc(deleteRelay))).Methods("DELETE")
  29. r.HandleFunc("/api/nodes/{network}/{nodeid}/creategateway", authorize(false, true, "user", http.HandlerFunc(createEgressGateway))).Methods("POST")
  30. r.HandleFunc("/api/nodes/{network}/{nodeid}/deletegateway", authorize(false, true, "user", http.HandlerFunc(deleteEgressGateway))).Methods("DELETE")
  31. r.HandleFunc("/api/nodes/{network}/{nodeid}/createingress", logic.SecurityCheck(false, http.HandlerFunc(createIngressGateway))).Methods("POST")
  32. r.HandleFunc("/api/nodes/{network}/{nodeid}/deleteingress", logic.SecurityCheck(false, http.HandlerFunc(deleteIngressGateway))).Methods("DELETE")
  33. r.HandleFunc("/api/nodes/{network}/{nodeid}", authorize(true, true, "node", http.HandlerFunc(updateNode))).Methods("POST")
  34. r.HandleFunc("/api/nodes/{network}", nodeauth(checkFreeTierLimits(node_l, http.HandlerFunc(createNode)))).Methods("POST")
  35. r.HandleFunc("/api/nodes/adm/{network}/authenticate", authenticate).Methods("POST")
  36. }
  37. // swagger:route POST /api/nodes/adm/{network}/authenticate nodes authenticate
  38. //
  39. // Authenticate to make further API calls related to a network.
  40. //
  41. // Schemes: https
  42. //
  43. // Security:
  44. // oauth
  45. //
  46. // Responses:
  47. // 200: successResponse
  48. func authenticate(response http.ResponseWriter, request *http.Request) {
  49. var authRequest models.AuthParams
  50. var result models.Node
  51. var errorResponse = models.ErrorResponse{
  52. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  53. }
  54. decoder := json.NewDecoder(request.Body)
  55. decoderErr := decoder.Decode(&authRequest)
  56. defer request.Body.Close()
  57. if decoderErr != nil {
  58. errorResponse.Code = http.StatusBadRequest
  59. errorResponse.Message = decoderErr.Error()
  60. logger.Log(0, request.Header.Get("user"), "error decoding request body: ",
  61. decoderErr.Error())
  62. logic.ReturnErrorResponse(response, request, errorResponse)
  63. return
  64. }
  65. errorResponse.Code = http.StatusBadRequest
  66. if authRequest.ID == "" {
  67. errorResponse.Message = "W1R3: ID can't be empty"
  68. logger.Log(0, request.Header.Get("user"), errorResponse.Message)
  69. logic.ReturnErrorResponse(response, request, errorResponse)
  70. return
  71. } else if authRequest.Password == "" {
  72. errorResponse.Message = "W1R3: Password can't be empty"
  73. logger.Log(0, request.Header.Get("user"), errorResponse.Message)
  74. logic.ReturnErrorResponse(response, request, errorResponse)
  75. return
  76. }
  77. var err error
  78. result, err = logic.GetNodeByID(authRequest.ID)
  79. if err != nil {
  80. result, err = logic.GetDeletedNodeByID(authRequest.ID)
  81. if err != nil {
  82. errorResponse.Code = http.StatusBadRequest
  83. errorResponse.Message = err.Error()
  84. logger.Log(0, request.Header.Get("user"),
  85. fmt.Sprintf("failed to get node info [%s]: %v", authRequest.ID, err))
  86. logic.ReturnErrorResponse(response, request, errorResponse)
  87. return
  88. }
  89. }
  90. host, err := logic.GetHost(result.HostID.String())
  91. if err != nil {
  92. errorResponse.Code = http.StatusBadRequest
  93. errorResponse.Message = err.Error()
  94. logger.Log(0, request.Header.Get("user"),
  95. "error retrieving host: ", err.Error())
  96. logic.ReturnErrorResponse(response, request, errorResponse)
  97. return
  98. }
  99. err = bcrypt.CompareHashAndPassword([]byte(host.HostPass), []byte(authRequest.Password))
  100. if err != nil {
  101. errorResponse.Code = http.StatusBadRequest
  102. errorResponse.Message = err.Error()
  103. logger.Log(0, request.Header.Get("user"),
  104. "error validating user password: ", err.Error())
  105. logic.ReturnErrorResponse(response, request, errorResponse)
  106. return
  107. }
  108. // creates network role,node client (added here to resolve any missing configuration in MQ)
  109. event := mq.MqDynsecPayload{
  110. Commands: []mq.MqDynSecCmd{
  111. {
  112. Command: mq.CreateRoleCmd,
  113. RoleName: result.Network,
  114. Textname: "Network wide role with Acls for nodes",
  115. Acls: mq.FetchNetworkAcls(result.Network),
  116. },
  117. {
  118. Command: mq.CreateClientCmd,
  119. Username: result.HostID.String(),
  120. Password: authRequest.Password,
  121. Textname: host.Name,
  122. Roles: []mq.MqDynSecRole{
  123. {
  124. Rolename: mq.NodeRole,
  125. Priority: -1,
  126. },
  127. {
  128. Rolename: result.Network,
  129. Priority: -1,
  130. },
  131. },
  132. Groups: make([]mq.MqDynSecGroup, 0),
  133. },
  134. },
  135. }
  136. if err := mq.PublishEventToDynSecTopic(event); err != nil {
  137. logger.Log(0, fmt.Sprintf("failed to send DynSec command [%v]: %v",
  138. event.Commands, err.Error()))
  139. errorResponse.Code = http.StatusInternalServerError
  140. errorResponse.Message = fmt.Sprintf("could not create mq client for node [%s]: %v", result.ID, err)
  141. return
  142. }
  143. tokenString, err := logic.CreateJWT(authRequest.ID, authRequest.MacAddress, result.Network)
  144. if tokenString == "" {
  145. errorResponse.Code = http.StatusBadRequest
  146. errorResponse.Message = "Could not create Token"
  147. logger.Log(0, request.Header.Get("user"),
  148. fmt.Sprintf("%s: %v", errorResponse.Message, err))
  149. logic.ReturnErrorResponse(response, request, errorResponse)
  150. return
  151. }
  152. var successResponse = models.SuccessResponse{
  153. Code: http.StatusOK,
  154. Message: "W1R3: Device " + authRequest.ID + " Authorized",
  155. Response: models.SuccessfulLoginResponse{
  156. AuthToken: tokenString,
  157. ID: authRequest.ID,
  158. },
  159. }
  160. successJSONResponse, jsonError := json.Marshal(successResponse)
  161. if jsonError != nil {
  162. errorResponse.Code = http.StatusBadRequest
  163. errorResponse.Message = err.Error()
  164. logger.Log(0, request.Header.Get("user"),
  165. "error marshalling resp: ", err.Error())
  166. logic.ReturnErrorResponse(response, request, errorResponse)
  167. return
  168. }
  169. response.WriteHeader(http.StatusOK)
  170. response.Header().Set("Content-Type", "application/json")
  171. response.Write(successJSONResponse)
  172. }
  173. // auth middleware for api calls from nodes where node is has not yet joined the server (register, join)
  174. func nodeauth(next http.Handler) http.HandlerFunc {
  175. return func(w http.ResponseWriter, r *http.Request) {
  176. bearerToken := r.Header.Get("Authorization")
  177. var tokenSplit = strings.Split(bearerToken, " ")
  178. var token = ""
  179. if len(tokenSplit) < 2 {
  180. errorResponse := models.ErrorResponse{
  181. Code: http.StatusUnauthorized, Message: "W1R3: You are unauthorized to access this endpoint.",
  182. }
  183. logic.ReturnErrorResponse(w, r, errorResponse)
  184. return
  185. } else {
  186. token = tokenSplit[1]
  187. }
  188. found := false
  189. networks, err := logic.GetNetworks()
  190. if err != nil {
  191. logger.Log(0, "no networks", err.Error())
  192. errorResponse := models.ErrorResponse{
  193. Code: http.StatusNotFound, Message: "no networks",
  194. }
  195. logic.ReturnErrorResponse(w, r, errorResponse)
  196. return
  197. }
  198. for _, network := range networks {
  199. for _, key := range network.AccessKeys {
  200. if key.Value == token {
  201. found = true
  202. break
  203. }
  204. }
  205. }
  206. if !found {
  207. logger.Log(0, "valid access key not found")
  208. errorResponse := models.ErrorResponse{
  209. Code: http.StatusUnauthorized, Message: "You are unauthorized to access this endpoint.",
  210. }
  211. logic.ReturnErrorResponse(w, r, errorResponse)
  212. return
  213. }
  214. next.ServeHTTP(w, r)
  215. }
  216. }
  217. // The middleware for most requests to the API
  218. // They all pass through here first
  219. // This will validate the JWT (or check for master token)
  220. // This will also check against the authNetwork and make sure the node should be accessing that endpoint,
  221. // even if it's technically ok
  222. // This is kind of a poor man's RBAC. There's probably a better/smarter way.
  223. // TODO: Consider better RBAC implementations
  224. func authorize(nodesAllowed, networkCheck bool, authNetwork string, next http.Handler) http.HandlerFunc {
  225. return func(w http.ResponseWriter, r *http.Request) {
  226. var errorResponse = models.ErrorResponse{
  227. Code: http.StatusUnauthorized, Message: logic.Unauthorized_Msg,
  228. }
  229. var params = mux.Vars(r)
  230. networkexists, _ := logic.NetworkExists(params["network"])
  231. //check that the request is for a valid network
  232. //if (networkCheck && !networkexists) || err != nil {
  233. if networkCheck && !networkexists {
  234. logic.ReturnErrorResponse(w, r, errorResponse)
  235. return
  236. } else {
  237. w.Header().Set("Content-Type", "application/json")
  238. //get the auth token
  239. bearerToken := r.Header.Get("Authorization")
  240. var tokenSplit = strings.Split(bearerToken, " ")
  241. //I put this in in case the user doesn't put in a token at all (in which case it's empty)
  242. //There's probably a smarter way of handling this.
  243. var authToken = "928rt238tghgwe@TY@$Y@#WQAEGB2FC#@HG#@$Hddd"
  244. if len(tokenSplit) > 1 {
  245. authToken = tokenSplit[1]
  246. } else {
  247. logic.ReturnErrorResponse(w, r, errorResponse)
  248. return
  249. }
  250. //check if node instead of user
  251. if nodesAllowed {
  252. // TODO --- should ensure that node is only operating on itself
  253. if _, _, _, err := logic.VerifyToken(authToken); err == nil {
  254. // this indicates request is from a node
  255. // used for failover - if a getNode comes from node, this will trigger a metrics wipe
  256. next.ServeHTTP(w, r)
  257. return
  258. }
  259. }
  260. var isAuthorized = false
  261. var nodeID = ""
  262. username, networks, isadmin, errN := logic.VerifyUserToken(authToken)
  263. if errN != nil {
  264. logic.ReturnErrorResponse(w, r, errorResponse)
  265. return
  266. }
  267. isnetadmin := isadmin
  268. if errN == nil && isadmin {
  269. nodeID = "mastermac"
  270. isAuthorized = true
  271. r.Header.Set("ismasterkey", "yes")
  272. }
  273. if !isadmin && params["network"] != "" {
  274. if logic.StringSliceContains(networks, params["network"]) && pro.IsUserNetAdmin(params["network"], username) {
  275. isnetadmin = true
  276. }
  277. }
  278. //The mastermac (login with masterkey from config) can do everything!! May be dangerous.
  279. if nodeID == "mastermac" {
  280. isAuthorized = true
  281. r.Header.Set("ismasterkey", "yes")
  282. //for everyone else, there's poor man's RBAC. The "cases" are defined in the routes in the handlers
  283. //So each route defines which access network should be allowed to access it
  284. } else {
  285. switch authNetwork {
  286. case "all":
  287. isAuthorized = true
  288. case "nodes":
  289. isAuthorized = (nodeID != "") || isnetadmin
  290. case "network":
  291. if isnetadmin {
  292. isAuthorized = true
  293. } else {
  294. node, err := logic.GetNodeByID(nodeID)
  295. if err != nil {
  296. logic.ReturnErrorResponse(w, r, errorResponse)
  297. return
  298. }
  299. isAuthorized = (node.Network == params["network"])
  300. }
  301. case "node":
  302. if isnetadmin {
  303. isAuthorized = true
  304. } else {
  305. isAuthorized = (nodeID == params["netid"])
  306. }
  307. case "user":
  308. isAuthorized = true
  309. default:
  310. isAuthorized = false
  311. }
  312. }
  313. if !isAuthorized {
  314. logic.ReturnErrorResponse(w, r, errorResponse)
  315. return
  316. } else {
  317. //If authorized, this function passes along it's request and output to the appropriate route function.
  318. if username == "" {
  319. username = "(user not found)"
  320. }
  321. r.Header.Set("user", username)
  322. next.ServeHTTP(w, r)
  323. }
  324. }
  325. }
  326. }
  327. // swagger:route GET /api/nodes/{network} nodes getNetworkNodes
  328. //
  329. // Gets all nodes associated with network including pending nodes.
  330. //
  331. // Schemes: https
  332. //
  333. // Security:
  334. // oauth
  335. //
  336. // Responses:
  337. // 200: nodeSliceResponse
  338. func getNetworkNodes(w http.ResponseWriter, r *http.Request) {
  339. w.Header().Set("Content-Type", "application/json")
  340. var params = mux.Vars(r)
  341. networkName := params["network"]
  342. nodes, err := logic.GetNetworkNodes(networkName)
  343. if err != nil {
  344. logger.Log(0, r.Header.Get("user"),
  345. fmt.Sprintf("error fetching nodes on network %s: %v", networkName, err))
  346. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  347. return
  348. }
  349. // returns all the nodes in JSON/API format
  350. apiNodes := logic.GetAllNodesAPI(nodes[:])
  351. logger.Log(2, r.Header.Get("user"), "fetched nodes on network", networkName)
  352. w.WriteHeader(http.StatusOK)
  353. json.NewEncoder(w).Encode(apiNodes)
  354. }
  355. // swagger:route GET /api/nodes nodes getAllNodes
  356. //
  357. // Get all nodes across all networks.
  358. //
  359. // Schemes: https
  360. //
  361. // Security:
  362. // oauth
  363. //
  364. // Responses:
  365. // 200: nodeSliceResponse
  366. //
  367. // 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
  368. func getAllNodes(w http.ResponseWriter, r *http.Request) {
  369. w.Header().Set("Content-Type", "application/json")
  370. user, err := logic.GetUser(r.Header.Get("user"))
  371. if err != nil && r.Header.Get("ismasterkey") != "yes" {
  372. logger.Log(0, r.Header.Get("user"),
  373. "error fetching user info: ", err.Error())
  374. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  375. return
  376. }
  377. var nodes []models.Node
  378. if user.IsAdmin || r.Header.Get("ismasterkey") == "yes" {
  379. nodes, err = logic.GetAllNodes()
  380. if err != nil {
  381. logger.Log(0, "error fetching all nodes info: ", err.Error())
  382. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  383. return
  384. }
  385. } else {
  386. nodes, err = getUsersNodes(user)
  387. if err != nil {
  388. logger.Log(0, r.Header.Get("user"),
  389. "error fetching nodes: ", err.Error())
  390. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  391. return
  392. }
  393. }
  394. // return all the nodes in JSON/API format
  395. apiNodes := logic.GetAllNodesAPI(nodes[:])
  396. logger.Log(3, r.Header.Get("user"), "fetched all nodes they have access to")
  397. w.WriteHeader(http.StatusOK)
  398. json.NewEncoder(w).Encode(apiNodes)
  399. }
  400. func getUsersNodes(user models.User) ([]models.Node, error) {
  401. var nodes []models.Node
  402. var err error
  403. for _, networkName := range user.Networks {
  404. tmpNodes, err := logic.GetNetworkNodes(networkName)
  405. if err != nil {
  406. continue
  407. }
  408. nodes = append(nodes, tmpNodes...)
  409. }
  410. return nodes, err
  411. }
  412. // swagger:route GET /api/nodes/{network}/{nodeid} nodes getNode
  413. //
  414. // Get an individual node.
  415. //
  416. // Schemes: https
  417. //
  418. // Security:
  419. // oauth
  420. //
  421. // Responses:
  422. // 200: nodeResponse
  423. func getNode(w http.ResponseWriter, r *http.Request) {
  424. // set header.
  425. w.Header().Set("Content-Type", "application/json")
  426. nodeRequest := r.Header.Get("requestfrom") == "node"
  427. var params = mux.Vars(r)
  428. nodeid := params["nodeid"]
  429. node, err := logic.GetNodeByID(nodeid)
  430. if err != nil {
  431. logger.Log(0, r.Header.Get("user"),
  432. fmt.Sprintf("error fetching node [ %s ] info: %v", nodeid, err))
  433. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  434. return
  435. }
  436. host, err := logic.GetHost(node.HostID.String())
  437. if err != nil {
  438. logger.Log(0, r.Header.Get("user"),
  439. fmt.Sprintf("error fetching host for node [ %s ] info: %v", nodeid, err))
  440. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  441. return
  442. }
  443. peerUpdate, err := logic.GetPeerUpdate(&node, host)
  444. if err != nil && !database.IsEmptyRecord(err) {
  445. logger.Log(0, r.Header.Get("user"),
  446. fmt.Sprintf("error fetching wg peers config for node [ %s ]: %v", nodeid, err))
  447. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  448. return
  449. }
  450. server := servercfg.GetServerInfo()
  451. network, err := logic.GetNetwork(node.Network)
  452. if err != nil {
  453. logger.Log(0, r.Header.Get("user"),
  454. fmt.Sprintf("error fetching network for node [ %s ] info: %v", nodeid, err))
  455. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  456. return
  457. }
  458. legacy := node.Legacy(host, &server, &network)
  459. response := models.NodeGet{
  460. Node: *legacy,
  461. Peers: peerUpdate.Peers,
  462. ServerConfig: server,
  463. PeerIDs: peerUpdate.PeerIDs,
  464. }
  465. if host.ProxyEnabled {
  466. proxyPayload, err := logic.GetPeersForProxy(&node, false)
  467. if err == nil {
  468. response.ProxyUpdate = proxyPayload
  469. } else {
  470. logger.Log(0, "failed to get proxy update: ", err.Error())
  471. }
  472. }
  473. if servercfg.Is_EE && nodeRequest {
  474. if err = logic.EnterpriseResetAllPeersFailovers(node.ID.String(), node.Network); err != nil {
  475. logger.Log(1, "failed to reset failover list during node config pull", node.ID.String(), node.Network)
  476. }
  477. }
  478. logger.Log(2, r.Header.Get("user"), "fetched node", params["nodeid"])
  479. w.WriteHeader(http.StatusOK)
  480. json.NewEncoder(w).Encode(response)
  481. }
  482. // swagger:route POST /api/nodes/{network} nodes createNode
  483. //
  484. // Create a node on a network.
  485. //
  486. // Schemes: https
  487. //
  488. // Security:
  489. // oauth
  490. //
  491. // Responses:
  492. // 200: nodeGetResponse
  493. func createNode(w http.ResponseWriter, r *http.Request) {
  494. w.Header().Set("Content-Type", "application/json")
  495. var params = mux.Vars(r)
  496. var errorResponse = models.ErrorResponse{
  497. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  498. }
  499. networkName := params["network"]
  500. networkexists, err := logic.NetworkExists(networkName)
  501. if err != nil {
  502. logger.Log(0, r.Header.Get("user"),
  503. fmt.Sprintf("failed to fetch network [%s] info: %v", networkName, err))
  504. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  505. return
  506. } else if !networkexists {
  507. errorResponse = models.ErrorResponse{
  508. Code: http.StatusNotFound, Message: "W1R3: Network does not exist! ",
  509. }
  510. logger.Log(0, r.Header.Get("user"),
  511. fmt.Sprintf("network [%s] does not exist", networkName))
  512. logic.ReturnErrorResponse(w, r, errorResponse)
  513. return
  514. }
  515. //get data from body of request
  516. data := models.JoinData{}
  517. err = json.NewDecoder(r.Body).Decode(&data)
  518. if err != nil {
  519. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  520. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  521. return
  522. }
  523. if !logic.IsVersionComptatible(data.Host.Version) {
  524. err := errors.New("incomatible netclient version")
  525. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  526. return
  527. }
  528. data.Node.Network = networkName
  529. networkSettings, err := logic.GetNetworkSettings(networkName)
  530. if err != nil {
  531. logger.Log(0, r.Header.Get("user"),
  532. fmt.Sprintf("failed to get network [%s] settings: %v", networkName, err))
  533. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  534. return
  535. }
  536. data.Node.NetworkSettings(networkSettings)
  537. keyName, validKey := logic.IsKeyValid(networkName, data.Key)
  538. if !validKey {
  539. errorResponse = models.ErrorResponse{
  540. Code: http.StatusUnauthorized, Message: "W1R3: Key invalid, or none provided.",
  541. }
  542. logger.Log(0, r.Header.Get("user"),
  543. fmt.Sprintf("failed to create node on network [%s]: %s",
  544. data.Node.Network, errorResponse.Message))
  545. logic.ReturnErrorResponse(w, r, errorResponse)
  546. return
  547. }
  548. user, err := pro.GetNetworkUser(networkName, promodels.NetworkUserID(keyName))
  549. if err == nil {
  550. if user.ID != "" {
  551. logger.Log(1, "associating new node with user", keyName)
  552. data.Node.OwnerID = string(user.ID)
  553. }
  554. }
  555. key, keyErr := logic.RetrievePublicTrafficKey()
  556. if keyErr != nil {
  557. logger.Log(0, "error retrieving key: ", keyErr.Error())
  558. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  559. return
  560. }
  561. if key == nil {
  562. logger.Log(0, "error: server traffic key is nil")
  563. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  564. return
  565. }
  566. if data.Host.TrafficKeyPublic == nil {
  567. logger.Log(0, "error: node traffic key is nil")
  568. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  569. return
  570. }
  571. server := servercfg.GetServerInfo()
  572. server.TrafficKey = key
  573. // consume password before hashing for mq client creation
  574. nodePassword := data.Host.HostPass
  575. data.Node.Server = servercfg.GetServer()
  576. if err := logic.CreateHost(&data.Host); err != nil {
  577. if errors.Is(err, logic.ErrHostExists) {
  578. logger.Log(3, "host exists .. no need to create")
  579. } else {
  580. logger.Log(0, "error creating host", err.Error())
  581. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  582. return
  583. }
  584. }
  585. err = logic.AssociateNodeToHost(&data.Node, &data.Host)
  586. if err != nil {
  587. logger.Log(0, r.Header.Get("user"),
  588. fmt.Sprintf("failed to create node on network [%s]: %s",
  589. networkName, err))
  590. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  591. return
  592. }
  593. // check if key belongs to a user
  594. // if so add to their netuser data
  595. // if it fails remove the node and fail request
  596. if user != nil {
  597. var updatedUserNode bool
  598. user.Nodes = append(user.Nodes, data.Node.ID.String()) // add new node to user
  599. if err = pro.UpdateNetworkUser(networkName, user); err == nil {
  600. logger.Log(1, "added node", data.Node.ID.String(), data.Host.Name, "to user", string(user.ID))
  601. updatedUserNode = true
  602. }
  603. if !updatedUserNode { // user was found but not updated, so delete node
  604. logger.Log(0, "failed to add node to user", keyName)
  605. logic.DeleteNode(&data.Node, true)
  606. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  607. return
  608. }
  609. }
  610. peerUpdate, err := logic.GetPeerUpdate(&data.Node, &data.Host)
  611. if err != nil && !database.IsEmptyRecord(err) {
  612. logger.Log(0, r.Header.Get("user"),
  613. fmt.Sprintf("error fetching wg peers config for node [ %s ]: %v", data.Node.ID.String(), err))
  614. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  615. return
  616. }
  617. data.Node.Peers = peerUpdate.Peers
  618. // Create client for this host in Mq
  619. event := mq.MqDynsecPayload{
  620. Commands: []mq.MqDynSecCmd{
  621. { // delete if any client exists already
  622. Command: mq.DeleteClientCmd,
  623. Username: data.Host.ID.String(),
  624. },
  625. {
  626. Command: mq.CreateRoleCmd,
  627. RoleName: networkName,
  628. Textname: "Network wide role with Acls for nodes",
  629. Acls: mq.FetchNetworkAcls(networkName),
  630. },
  631. {
  632. Command: mq.CreateClientCmd,
  633. Username: data.Host.ID.String(),
  634. Password: nodePassword,
  635. Textname: data.Host.Name,
  636. Roles: []mq.MqDynSecRole{
  637. {
  638. Rolename: mq.NodeRole,
  639. Priority: -1,
  640. },
  641. {
  642. Rolename: networkName,
  643. Priority: -1,
  644. },
  645. },
  646. Groups: make([]mq.MqDynSecGroup, 0),
  647. },
  648. },
  649. }
  650. if err := mq.PublishEventToDynSecTopic(event); err != nil {
  651. logger.Log(0, fmt.Sprintf("failed to send DynSec command [%v]: %v",
  652. event.Commands, err.Error()))
  653. }
  654. response := models.NodeJoinResponse{
  655. Node: data.Node,
  656. ServerConfig: server,
  657. PeerIDs: peerUpdate.PeerIDs,
  658. }
  659. logger.Log(1, r.Header.Get("user"), "created new node", data.Host.Name, "on network", networkName)
  660. w.WriteHeader(http.StatusOK)
  661. json.NewEncoder(w).Encode(response)
  662. go func() {
  663. if err := mq.PublishPeerUpdate(data.Node.Network, true); err != nil {
  664. logger.Log(1, "failed a peer update after creation of node", data.Host.Name)
  665. }
  666. }()
  667. //runForceServerUpdate(&data.Node, true)
  668. }
  669. // == EGRESS ==
  670. // swagger:route POST /api/nodes/{network}/{nodeid}/creategateway nodes createEgressGateway
  671. //
  672. // Create an egress gateway.
  673. //
  674. // Schemes: https
  675. //
  676. // Security:
  677. // oauth
  678. //
  679. // Responses:
  680. // 200: nodeResponse
  681. func createEgressGateway(w http.ResponseWriter, r *http.Request) {
  682. var gateway models.EgressGatewayRequest
  683. var params = mux.Vars(r)
  684. w.Header().Set("Content-Type", "application/json")
  685. err := json.NewDecoder(r.Body).Decode(&gateway)
  686. if err != nil {
  687. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  688. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  689. return
  690. }
  691. gateway.NetID = params["network"]
  692. gateway.NodeID = params["nodeid"]
  693. node, err := logic.CreateEgressGateway(gateway)
  694. if err != nil {
  695. logger.Log(0, r.Header.Get("user"),
  696. fmt.Sprintf("failed to create egress gateway on node [%s] on network [%s]: %v",
  697. gateway.NodeID, gateway.NetID, err))
  698. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  699. return
  700. }
  701. apiNode := node.ConvertToAPINode()
  702. logger.Log(1, r.Header.Get("user"), "created egress gateway on node", gateway.NodeID, "on network", gateway.NetID)
  703. w.WriteHeader(http.StatusOK)
  704. json.NewEncoder(w).Encode(apiNode)
  705. runUpdates(&node, true)
  706. }
  707. // swagger:route DELETE /api/nodes/{network}/{nodeid}/deletegateway nodes deleteEgressGateway
  708. //
  709. // Delete an egress gateway.
  710. //
  711. // Schemes: https
  712. //
  713. // Security:
  714. // oauth
  715. //
  716. // Responses:
  717. // 200: nodeResponse
  718. func deleteEgressGateway(w http.ResponseWriter, r *http.Request) {
  719. w.Header().Set("Content-Type", "application/json")
  720. var params = mux.Vars(r)
  721. nodeid := params["nodeid"]
  722. netid := params["network"]
  723. node, err := logic.DeleteEgressGateway(netid, nodeid)
  724. if err != nil {
  725. logger.Log(0, r.Header.Get("user"),
  726. fmt.Sprintf("failed to delete egress gateway on node [%s] on network [%s]: %v",
  727. nodeid, netid, err))
  728. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  729. return
  730. }
  731. apiNode := node.ConvertToAPINode()
  732. logger.Log(1, r.Header.Get("user"), "deleted egress gateway on node", nodeid, "on network", netid)
  733. w.WriteHeader(http.StatusOK)
  734. json.NewEncoder(w).Encode(apiNode)
  735. runUpdates(&node, true)
  736. }
  737. // == INGRESS ==
  738. // swagger:route POST /api/nodes/{network}/{nodeid}/createingress nodes createIngressGateway
  739. //
  740. // Create an ingress gateway.
  741. //
  742. // Schemes: https
  743. //
  744. // Security:
  745. // oauth
  746. //
  747. // Responses:
  748. // 200: nodeResponse
  749. func createIngressGateway(w http.ResponseWriter, r *http.Request) {
  750. var params = mux.Vars(r)
  751. w.Header().Set("Content-Type", "application/json")
  752. nodeid := params["nodeid"]
  753. netid := params["network"]
  754. type failoverData struct {
  755. Failover bool `json:"failover"`
  756. }
  757. var failoverReqBody failoverData
  758. json.NewDecoder(r.Body).Decode(&failoverReqBody)
  759. node, err := logic.CreateIngressGateway(netid, nodeid, failoverReqBody.Failover)
  760. if err != nil {
  761. logger.Log(0, r.Header.Get("user"),
  762. fmt.Sprintf("failed to create ingress gateway on node [%s] on network [%s]: %v",
  763. nodeid, netid, err))
  764. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  765. return
  766. }
  767. if servercfg.Is_EE && failoverReqBody.Failover {
  768. if err = logic.EnterpriseResetFailoverFunc(node.Network); err != nil {
  769. logger.Log(1, "failed to reset failover list during failover create", node.ID.String(), node.Network)
  770. }
  771. }
  772. apiNode := node.ConvertToAPINode()
  773. logger.Log(1, r.Header.Get("user"), "created ingress gateway on node", nodeid, "on network", netid)
  774. w.WriteHeader(http.StatusOK)
  775. json.NewEncoder(w).Encode(apiNode)
  776. runUpdates(&node, true)
  777. }
  778. // swagger:route DELETE /api/nodes/{network}/{nodeid}/deleteingress nodes deleteIngressGateway
  779. //
  780. // Delete an ingress gateway.
  781. //
  782. // Schemes: https
  783. //
  784. // Security:
  785. // oauth
  786. //
  787. // Responses:
  788. // 200: nodeResponse
  789. func deleteIngressGateway(w http.ResponseWriter, r *http.Request) {
  790. w.Header().Set("Content-Type", "application/json")
  791. var params = mux.Vars(r)
  792. nodeid := params["nodeid"]
  793. netid := params["network"]
  794. node, wasFailover, err := logic.DeleteIngressGateway(netid, nodeid)
  795. if err != nil {
  796. logger.Log(0, r.Header.Get("user"),
  797. fmt.Sprintf("failed to delete ingress gateway on node [%s] on network [%s]: %v",
  798. nodeid, netid, err))
  799. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  800. return
  801. }
  802. if servercfg.Is_EE && wasFailover {
  803. if err = logic.EnterpriseResetFailoverFunc(node.Network); err != nil {
  804. logger.Log(1, "failed to reset failover list during failover create", node.ID.String(), node.Network)
  805. }
  806. }
  807. apiNode := node.ConvertToAPINode()
  808. logger.Log(1, r.Header.Get("user"), "deleted ingress gateway", nodeid)
  809. w.WriteHeader(http.StatusOK)
  810. json.NewEncoder(w).Encode(apiNode)
  811. runUpdates(&node, true)
  812. }
  813. // swagger:route PUT /api/nodes/{network}/{nodeid}/migrate nodes migrateNode
  814. //
  815. // Used to migrate a legacy node.
  816. //
  817. // Schemes: https
  818. //
  819. // Security:
  820. // oauth
  821. //
  822. // Responses:
  823. // 200: nodeResponse
  824. func nodeNodeUpdate(w http.ResponseWriter, r *http.Request) {
  825. // should only be used by nodes
  826. w.Header().Set("Content-Type", "application/json")
  827. var params = mux.Vars(r)
  828. //start here
  829. nodeid := params["nodeid"]
  830. currentNode, err := logic.GetNodeByID(nodeid)
  831. if err != nil {
  832. logger.Log(0,
  833. fmt.Sprintf("error fetching node [ %s ] info: %v during migrate", nodeid, err))
  834. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  835. return
  836. }
  837. var newNode models.Node
  838. // we decode our body request params
  839. err = json.NewDecoder(r.Body).Decode(&newNode)
  840. if err != nil {
  841. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  842. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  843. return
  844. }
  845. relayupdate := false
  846. if currentNode.IsRelay && len(newNode.RelayAddrs) > 0 {
  847. if len(newNode.RelayAddrs) != len(currentNode.RelayAddrs) {
  848. relayupdate = true
  849. } else {
  850. for i, addr := range newNode.RelayAddrs {
  851. if addr != currentNode.RelayAddrs[i] {
  852. relayupdate = true
  853. }
  854. }
  855. }
  856. }
  857. relayedUpdate := false
  858. if currentNode.IsRelayed && (currentNode.Address.String() != newNode.Address.String() || currentNode.Address6.String() != newNode.Address6.String()) {
  859. relayedUpdate = true
  860. }
  861. if !servercfg.GetRce() {
  862. newNode.PostDown = currentNode.PostDown
  863. newNode.PostUp = currentNode.PostUp
  864. }
  865. ifaceDelta := logic.IfaceDelta(&currentNode, &newNode)
  866. if ifaceDelta && servercfg.Is_EE {
  867. if err = logic.EnterpriseResetAllPeersFailovers(currentNode.ID.String(), currentNode.Network); err != nil {
  868. logger.Log(0, "failed to reset failover lists during node update for node", currentNode.ID.String(), currentNode.Network)
  869. }
  870. }
  871. err = logic.UpdateNode(&currentNode, &newNode)
  872. if err != nil {
  873. logger.Log(0, r.Header.Get("user"),
  874. fmt.Sprintf("failed to update node info [ %s ] info: %v", nodeid, err))
  875. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  876. return
  877. }
  878. if relayupdate {
  879. updatenodes := logic.UpdateRelay(currentNode.Network, currentNode.RelayAddrs, newNode.RelayAddrs)
  880. if len(updatenodes) > 0 {
  881. for _, relayedNode := range updatenodes {
  882. runUpdates(&relayedNode, false)
  883. }
  884. }
  885. }
  886. if relayedUpdate {
  887. updateRelay(&currentNode, &newNode)
  888. }
  889. if servercfg.IsDNSMode() {
  890. logic.SetDNS()
  891. }
  892. logger.Log(1, r.Header.Get("user"), "updated node", currentNode.ID.String(), "on network", currentNode.Network)
  893. w.WriteHeader(http.StatusOK)
  894. json.NewEncoder(w).Encode(newNode)
  895. runUpdates(&newNode, ifaceDelta)
  896. }
  897. // swagger:route PUT /api/nodes/{network}/{nodeid} nodes updateNode
  898. //
  899. // Update an individual node.
  900. //
  901. // Schemes: https
  902. //
  903. // Security:
  904. // oauth
  905. //
  906. // Responses:
  907. // 200: nodeResponse
  908. func updateNode(w http.ResponseWriter, r *http.Request) {
  909. w.Header().Set("Content-Type", "application/json")
  910. var params = mux.Vars(r)
  911. //start here
  912. nodeid := params["nodeid"]
  913. currentNode, err := logic.GetNodeByID(nodeid)
  914. if err != nil {
  915. logger.Log(0, r.Header.Get("user"),
  916. fmt.Sprintf("error fetching node [ %s ] info: %v", nodeid, err))
  917. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  918. return
  919. }
  920. var newData models.ApiNode
  921. // we decode our body request params
  922. err = json.NewDecoder(r.Body).Decode(&newData)
  923. if err != nil {
  924. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  925. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  926. return
  927. }
  928. newNode := newData.ConvertToServerNode(&currentNode)
  929. relayupdate := false
  930. if currentNode.IsRelay && len(newNode.RelayAddrs) > 0 {
  931. if len(newNode.RelayAddrs) != len(currentNode.RelayAddrs) {
  932. relayupdate = true
  933. } else {
  934. for i, addr := range newNode.RelayAddrs {
  935. if addr != currentNode.RelayAddrs[i] {
  936. relayupdate = true
  937. }
  938. }
  939. }
  940. }
  941. relayedUpdate := false
  942. if currentNode.IsRelayed && (currentNode.Address.String() != newNode.Address.String() || currentNode.Address6.String() != newNode.Address6.String()) {
  943. relayedUpdate = true
  944. }
  945. if !servercfg.GetRce() {
  946. newNode.PostDown = currentNode.PostDown
  947. newNode.PostUp = currentNode.PostUp
  948. }
  949. ifaceDelta := logic.IfaceDelta(&currentNode, newNode)
  950. if ifaceDelta && servercfg.Is_EE {
  951. if err = logic.EnterpriseResetAllPeersFailovers(currentNode.ID.String(), currentNode.Network); err != nil {
  952. logger.Log(0, "failed to reset failover lists during node update for node", currentNode.ID.String(), currentNode.Network)
  953. }
  954. }
  955. err = logic.UpdateNode(&currentNode, newNode)
  956. if err != nil {
  957. logger.Log(0, r.Header.Get("user"),
  958. fmt.Sprintf("failed to update node info [ %s ] info: %v", nodeid, err))
  959. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  960. return
  961. }
  962. if relayupdate {
  963. updatenodes := logic.UpdateRelay(currentNode.Network, currentNode.RelayAddrs, newNode.RelayAddrs)
  964. if len(updatenodes) > 0 {
  965. for _, relayedNode := range updatenodes {
  966. runUpdates(&relayedNode, false)
  967. }
  968. }
  969. }
  970. if relayedUpdate {
  971. updateRelay(&currentNode, newNode)
  972. }
  973. if servercfg.IsDNSMode() {
  974. logic.SetDNS()
  975. }
  976. apiNode := newNode.ConvertToAPINode()
  977. logger.Log(1, r.Header.Get("user"), "updated node", currentNode.ID.String(), "on network", currentNode.Network)
  978. w.WriteHeader(http.StatusOK)
  979. json.NewEncoder(w).Encode(apiNode)
  980. runUpdates(newNode, ifaceDelta)
  981. }
  982. // swagger:route DELETE /api/nodes/{network}/{nodeid} nodes deleteNode
  983. //
  984. // Delete an individual node.
  985. //
  986. // Schemes: https
  987. //
  988. // Security:
  989. // oauth
  990. //
  991. // Responses:
  992. // 200: nodeResponse
  993. func deleteNode(w http.ResponseWriter, r *http.Request) {
  994. // Set header
  995. w.Header().Set("Content-Type", "application/json")
  996. // get params
  997. var params = mux.Vars(r)
  998. var nodeid = params["nodeid"]
  999. fromNode := r.Header.Get("requestfrom") == "node"
  1000. node, err := logic.GetNodeByID(nodeid)
  1001. if err != nil {
  1002. logger.Log(0, "error retrieving node to delete", err.Error())
  1003. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1004. return
  1005. }
  1006. host, err := logic.GetHost(node.HostID.String())
  1007. if err != nil {
  1008. logger.Log(0, "error retrieving host for node", node.ID.String(), err.Error())
  1009. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  1010. return
  1011. }
  1012. if r.Header.Get("ismaster") != "yes" {
  1013. username := r.Header.Get("user")
  1014. if username != "" && !doesUserOwnNode(username, params["network"], nodeid) {
  1015. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("user not permitted"), "badrequest"))
  1016. return
  1017. }
  1018. }
  1019. if err := logic.DeleteNode(&node, fromNode); err != nil {
  1020. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("failed to delete node"), "internal"))
  1021. return
  1022. }
  1023. if host.ProxyEnabled {
  1024. mq.ProxyUpdate(&manager.ProxyManagerPayload{
  1025. Action: manager.DeleteNetwork,
  1026. Network: node.Network,
  1027. }, &node)
  1028. }
  1029. if fromNode {
  1030. //check if server should be removed from mq
  1031. found := false
  1032. // err is irrelevent
  1033. nodes, _ := logic.GetAllNodes()
  1034. for _, nodetocheck := range nodes {
  1035. if nodetocheck.HostID == node.HostID {
  1036. found = true
  1037. break
  1038. }
  1039. }
  1040. // TODO: Address how to remove host
  1041. if !found {
  1042. // deletes node related role and client
  1043. event := mq.MqDynsecPayload{
  1044. Commands: []mq.MqDynSecCmd{
  1045. {
  1046. Command: mq.DeleteClientCmd,
  1047. Username: node.HostID.String(),
  1048. },
  1049. },
  1050. }
  1051. if err := mq.PublishEventToDynSecTopic(event); err != nil {
  1052. logger.Log(0, fmt.Sprintf("failed to send DynSec command [%v]: %v",
  1053. event.Commands, err.Error()))
  1054. }
  1055. }
  1056. }
  1057. logic.ReturnSuccessResponse(w, r, nodeid+" deleted.")
  1058. logger.Log(1, r.Header.Get("user"), "Deleted node", nodeid, "from network", params["network"])
  1059. runUpdates(&node, false)
  1060. }
  1061. func runUpdates(node *models.Node, ifaceDelta bool) {
  1062. go func() { // don't block http response
  1063. // publish node update if not server
  1064. if err := mq.NodeUpdate(node); err != nil {
  1065. logger.Log(1, "error publishing node update to node", node.ID.String(), err.Error())
  1066. }
  1067. }()
  1068. }
  1069. func updateRelay(oldnode, newnode *models.Node) {
  1070. relay := logic.FindRelay(oldnode)
  1071. newrelay := relay
  1072. //check if node's address has been updated and if so, update the relayAddrs of the relay node with the updated address of the relayed node
  1073. if oldnode.Address.String() != newnode.Address.String() {
  1074. for i, ip := range newrelay.RelayAddrs {
  1075. if ip == oldnode.Address.IP.String() {
  1076. newrelay.RelayAddrs = append(newrelay.RelayAddrs[:i], relay.RelayAddrs[i+1:]...)
  1077. newrelay.RelayAddrs = append(newrelay.RelayAddrs, newnode.Address.IP.String())
  1078. }
  1079. }
  1080. }
  1081. //check if node's address(v6) has been updated and if so, update the relayAddrs of the relay node with the updated address(v6) of the relayed node
  1082. if oldnode.Address6.String() != newnode.Address6.String() {
  1083. for i, ip := range newrelay.RelayAddrs {
  1084. if ip == oldnode.Address.IP.String() {
  1085. newrelay.RelayAddrs = append(newrelay.RelayAddrs[:i], newrelay.RelayAddrs[i+1:]...)
  1086. newrelay.RelayAddrs = append(newrelay.RelayAddrs, newnode.Address6.IP.String())
  1087. }
  1088. }
  1089. }
  1090. logic.UpdateNode(relay, newrelay)
  1091. }
  1092. func doesUserOwnNode(username, network, nodeID string) bool {
  1093. u, err := logic.GetUser(username)
  1094. if err != nil {
  1095. return false
  1096. }
  1097. if u.IsAdmin {
  1098. return true
  1099. }
  1100. netUser, err := pro.GetNetworkUser(network, promodels.NetworkUserID(u.UserName))
  1101. if err != nil {
  1102. return false
  1103. }
  1104. if netUser.AccessLevel == pro.NET_ADMIN {
  1105. return true
  1106. }
  1107. return logic.StringSliceContains(netUser.Nodes, nodeID)
  1108. }