node.go 35 KB

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