node.go 31 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024
  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/functions"
  10. "github.com/gravitl/netmaker/logger"
  11. "github.com/gravitl/netmaker/logic"
  12. "github.com/gravitl/netmaker/models"
  13. "github.com/gravitl/netmaker/mq"
  14. "github.com/gravitl/netmaker/servercfg"
  15. "golang.org/x/crypto/bcrypt"
  16. )
  17. func nodeHandlers(r *mux.Router) {
  18. r.HandleFunc("/api/nodes", authorize(false, false, "user", http.HandlerFunc(getAllNodes))).Methods("GET")
  19. r.HandleFunc("/api/nodes/{network}", authorize(false, true, "network", http.HandlerFunc(getNetworkNodes))).Methods("GET")
  20. r.HandleFunc("/api/nodes/{network}/{nodeid}", authorize(true, true, "node", http.HandlerFunc(getNode))).Methods("GET")
  21. r.HandleFunc("/api/nodes/{network}/{nodeid}", authorize(false, true, "node", http.HandlerFunc(updateNode))).Methods("PUT")
  22. r.HandleFunc("/api/nodes/{network}/{nodeid}", authorize(true, true, "node", http.HandlerFunc(deleteNode))).Methods("DELETE")
  23. r.HandleFunc("/api/nodes/{network}/{nodeid}/createrelay", authorize(false, true, "user", http.HandlerFunc(createRelay))).Methods("POST")
  24. r.HandleFunc("/api/nodes/{network}/{nodeid}/deleterelay", authorize(false, true, "user", http.HandlerFunc(deleteRelay))).Methods("DELETE")
  25. r.HandleFunc("/api/nodes/{network}/{nodeid}/creategateway", authorize(false, true, "user", http.HandlerFunc(createEgressGateway))).Methods("POST")
  26. r.HandleFunc("/api/nodes/{network}/{nodeid}/deletegateway", authorize(false, true, "user", http.HandlerFunc(deleteEgressGateway))).Methods("DELETE")
  27. r.HandleFunc("/api/nodes/{network}/{nodeid}/createingress", securityCheck(false, http.HandlerFunc(createIngressGateway))).Methods("POST")
  28. r.HandleFunc("/api/nodes/{network}/{nodeid}/deleteingress", securityCheck(false, http.HandlerFunc(deleteIngressGateway))).Methods("DELETE")
  29. r.HandleFunc("/api/nodes/{network}/{nodeid}/approve", authorize(false, true, "user", http.HandlerFunc(uncordonNode))).Methods("POST")
  30. r.HandleFunc("/api/nodes/{network}", nodeauth(http.HandlerFunc(createNode))).Methods("POST")
  31. r.HandleFunc("/api/nodes/adm/{network}/lastmodified", authorize(false, true, "network", http.HandlerFunc(getLastModified))).Methods("GET")
  32. r.HandleFunc("/api/nodes/adm/{network}/authenticate", authenticate).Methods("POST")
  33. }
  34. // swagger:route POST /api/nodes/adm/{network}/authenticate nodes authenticate
  35. //
  36. // Authenticate to make further API calls related to a network.
  37. //
  38. // Schemes: https
  39. //
  40. // Security:
  41. // oauth
  42. //
  43. // Responses:
  44. // 200: successResponse
  45. func authenticate(response http.ResponseWriter, request *http.Request) {
  46. var authRequest models.AuthParams
  47. var result models.Node
  48. var errorResponse = models.ErrorResponse{
  49. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  50. }
  51. decoder := json.NewDecoder(request.Body)
  52. decoderErr := decoder.Decode(&authRequest)
  53. defer request.Body.Close()
  54. if decoderErr != nil {
  55. errorResponse.Code = http.StatusBadRequest
  56. errorResponse.Message = decoderErr.Error()
  57. logger.Log(0, request.Header.Get("user"), "error decoding request body: ",
  58. decoderErr.Error())
  59. returnErrorResponse(response, request, errorResponse)
  60. return
  61. } else {
  62. errorResponse.Code = http.StatusBadRequest
  63. if authRequest.ID == "" {
  64. errorResponse.Message = "W1R3: ID can't be empty"
  65. logger.Log(0, request.Header.Get("user"), errorResponse.Message)
  66. returnErrorResponse(response, request, errorResponse)
  67. return
  68. } else if authRequest.Password == "" {
  69. errorResponse.Message = "W1R3: Password can't be empty"
  70. logger.Log(0, request.Header.Get("user"), errorResponse.Message)
  71. returnErrorResponse(response, request, errorResponse)
  72. return
  73. } else {
  74. var err error
  75. result, err = logic.GetNodeByID(authRequest.ID)
  76. if err != nil {
  77. errorResponse.Code = http.StatusBadRequest
  78. errorResponse.Message = err.Error()
  79. logger.Log(0, request.Header.Get("user"),
  80. fmt.Sprintf("failed to get node info [%s]: %v", authRequest.ID, err))
  81. returnErrorResponse(response, request, errorResponse)
  82. return
  83. }
  84. err = bcrypt.CompareHashAndPassword([]byte(result.Password), []byte(authRequest.Password))
  85. if err != nil {
  86. errorResponse.Code = http.StatusBadRequest
  87. errorResponse.Message = err.Error()
  88. logger.Log(0, request.Header.Get("user"),
  89. "error validating user password: ", err.Error())
  90. returnErrorResponse(response, request, errorResponse)
  91. return
  92. } else {
  93. tokenString, err := logic.CreateJWT(authRequest.ID, authRequest.MacAddress, result.Network)
  94. if tokenString == "" {
  95. errorResponse.Code = http.StatusBadRequest
  96. errorResponse.Message = "Could not create Token"
  97. logger.Log(0, request.Header.Get("user"),
  98. fmt.Sprintf("%s: %v", errorResponse.Message, err))
  99. returnErrorResponse(response, request, errorResponse)
  100. return
  101. }
  102. var successResponse = models.SuccessResponse{
  103. Code: http.StatusOK,
  104. Message: "W1R3: Device " + authRequest.ID + " Authorized",
  105. Response: models.SuccessfulLoginResponse{
  106. AuthToken: tokenString,
  107. ID: authRequest.ID,
  108. },
  109. }
  110. successJSONResponse, jsonError := json.Marshal(successResponse)
  111. if jsonError != nil {
  112. errorResponse.Code = http.StatusBadRequest
  113. errorResponse.Message = err.Error()
  114. logger.Log(0, request.Header.Get("user"),
  115. "error marshalling resp: ", err.Error())
  116. returnErrorResponse(response, request, errorResponse)
  117. return
  118. }
  119. response.WriteHeader(http.StatusOK)
  120. response.Header().Set("Content-Type", "application/json")
  121. response.Write(successJSONResponse)
  122. }
  123. }
  124. }
  125. }
  126. // auth middleware for api calls from nodes where node is has not yet joined the server (register, join)
  127. func nodeauth(next http.Handler) http.HandlerFunc {
  128. return func(w http.ResponseWriter, r *http.Request) {
  129. bearerToken := r.Header.Get("Authorization")
  130. var tokenSplit = strings.Split(bearerToken, " ")
  131. var token = ""
  132. if len(tokenSplit) < 2 {
  133. errorResponse := models.ErrorResponse{
  134. Code: http.StatusUnauthorized, Message: "W1R3: You are unauthorized to access this endpoint.",
  135. }
  136. returnErrorResponse(w, r, errorResponse)
  137. return
  138. } else {
  139. token = tokenSplit[1]
  140. }
  141. found := false
  142. networks, err := logic.GetNetworks()
  143. if err != nil {
  144. logger.Log(0, "no networks", err.Error())
  145. errorResponse := models.ErrorResponse{
  146. Code: http.StatusNotFound, Message: "no networks",
  147. }
  148. returnErrorResponse(w, r, errorResponse)
  149. return
  150. }
  151. for _, network := range networks {
  152. for _, key := range network.AccessKeys {
  153. if key.Value == token {
  154. found = true
  155. break
  156. }
  157. }
  158. }
  159. if !found {
  160. logger.Log(0, "valid access key not found")
  161. errorResponse := models.ErrorResponse{
  162. Code: http.StatusUnauthorized, Message: "You are unauthorized to access this endpoint.",
  163. }
  164. returnErrorResponse(w, r, errorResponse)
  165. return
  166. }
  167. next.ServeHTTP(w, r)
  168. }
  169. }
  170. // The middleware for most requests to the API
  171. // They all pass through here first
  172. // This will validate the JWT (or check for master token)
  173. // This will also check against the authNetwork and make sure the node should be accessing that endpoint,
  174. // even if it's technically ok
  175. // This is kind of a poor man's RBAC. There's probably a better/smarter way.
  176. // TODO: Consider better RBAC implementations
  177. func authorize(nodesAllowed, networkCheck bool, authNetwork string, next http.Handler) http.HandlerFunc {
  178. return func(w http.ResponseWriter, r *http.Request) {
  179. var errorResponse = models.ErrorResponse{
  180. Code: http.StatusUnauthorized, Message: unauthorized_msg,
  181. }
  182. var params = mux.Vars(r)
  183. networkexists, _ := functions.NetworkExists(params["network"])
  184. //check that the request is for a valid network
  185. //if (networkCheck && !networkexists) || err != nil {
  186. if networkCheck && !networkexists {
  187. returnErrorResponse(w, r, errorResponse)
  188. return
  189. } else {
  190. w.Header().Set("Content-Type", "application/json")
  191. //get the auth token
  192. bearerToken := r.Header.Get("Authorization")
  193. var tokenSplit = strings.Split(bearerToken, " ")
  194. //I put this in in case the user doesn't put in a token at all (in which case it's empty)
  195. //There's probably a smarter way of handling this.
  196. var authToken = "928rt238tghgwe@TY@$Y@#WQAEGB2FC#@HG#@$Hddd"
  197. if len(tokenSplit) > 1 {
  198. authToken = tokenSplit[1]
  199. } else {
  200. returnErrorResponse(w, r, errorResponse)
  201. return
  202. }
  203. //check if node instead of user
  204. if nodesAllowed {
  205. // TODO --- should ensure that node is only operating on itself
  206. if _, _, _, err := logic.VerifyToken(authToken); err == nil {
  207. next.ServeHTTP(w, r)
  208. return
  209. }
  210. }
  211. var isAuthorized = false
  212. var nodeID = ""
  213. username, networks, isadmin, errN := logic.VerifyUserToken(authToken)
  214. if errN != nil {
  215. returnErrorResponse(w, r, errorResponse)
  216. return
  217. }
  218. isnetadmin := isadmin
  219. if errN == nil && isadmin {
  220. nodeID = "mastermac"
  221. isAuthorized = true
  222. r.Header.Set("ismasterkey", "yes")
  223. }
  224. if !isadmin && params["network"] != "" {
  225. if logic.StringSliceContains(networks, params["network"]) {
  226. isnetadmin = true
  227. }
  228. }
  229. //The mastermac (login with masterkey from config) can do everything!! May be dangerous.
  230. if nodeID == "mastermac" {
  231. isAuthorized = true
  232. r.Header.Set("ismasterkey", "yes")
  233. //for everyone else, there's poor man's RBAC. The "cases" are defined in the routes in the handlers
  234. //So each route defines which access network should be allowed to access it
  235. } else {
  236. switch authNetwork {
  237. case "all":
  238. isAuthorized = true
  239. case "nodes":
  240. isAuthorized = (nodeID != "") || isnetadmin
  241. case "network":
  242. if isnetadmin {
  243. isAuthorized = true
  244. } else {
  245. node, err := logic.GetNodeByID(nodeID)
  246. if err != nil {
  247. returnErrorResponse(w, r, errorResponse)
  248. return
  249. }
  250. isAuthorized = (node.Network == params["network"])
  251. }
  252. case "node":
  253. if isnetadmin {
  254. isAuthorized = true
  255. } else {
  256. isAuthorized = (nodeID == params["netid"])
  257. }
  258. case "user":
  259. isAuthorized = true
  260. default:
  261. isAuthorized = false
  262. }
  263. }
  264. if !isAuthorized {
  265. returnErrorResponse(w, r, errorResponse)
  266. return
  267. } else {
  268. //If authorized, this function passes along it's request and output to the appropriate route function.
  269. if username == "" {
  270. username = "(user not found)"
  271. }
  272. r.Header.Set("user", username)
  273. next.ServeHTTP(w, r)
  274. }
  275. }
  276. }
  277. }
  278. // swagger:route GET /api/nodes/{network} nodes getNetworkNodes
  279. //
  280. // Gets all nodes associated with network including pending nodes.
  281. //
  282. // Schemes: https
  283. //
  284. // Security:
  285. // oauth
  286. //
  287. // Responses:
  288. // 200: nodeSliceResponse
  289. func getNetworkNodes(w http.ResponseWriter, r *http.Request) {
  290. w.Header().Set("Content-Type", "application/json")
  291. var nodes []models.Node
  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. returnErrorResponse(w, r, formatError(err, "internal"))
  299. return
  300. }
  301. for _, node := range nodes {
  302. if len(node.NetworkSettings.AccessKeys) > 0 {
  303. node.NetworkSettings.AccessKeys = []models.AccessKey{} // not to be sent back to client; client already knows how to join the network
  304. }
  305. }
  306. //Returns all the nodes in JSON format
  307. logger.Log(2, r.Header.Get("user"), "fetched nodes on network", networkName)
  308. w.WriteHeader(http.StatusOK)
  309. json.NewEncoder(w).Encode(nodes)
  310. }
  311. // swagger:route GET /api/nodes nodes getAllNodes
  312. //
  313. // Get all nodes across all networks.
  314. //
  315. // Schemes: https
  316. //
  317. // Security:
  318. // oauth
  319. //
  320. // Responses:
  321. // 200: nodeSliceResponse
  322. // 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
  323. func getAllNodes(w http.ResponseWriter, r *http.Request) {
  324. w.Header().Set("Content-Type", "application/json")
  325. user, err := logic.GetUser(r.Header.Get("user"))
  326. if err != nil && r.Header.Get("ismasterkey") != "yes" {
  327. logger.Log(0, r.Header.Get("user"),
  328. "error fetching user info: ", err.Error())
  329. returnErrorResponse(w, r, formatError(err, "internal"))
  330. return
  331. }
  332. var nodes []models.Node
  333. if user.IsAdmin || r.Header.Get("ismasterkey") == "yes" {
  334. nodes, err = logic.GetAllNodes()
  335. if err != nil {
  336. logger.Log(0, "error fetching all nodes info: ", err.Error())
  337. returnErrorResponse(w, r, formatError(err, "internal"))
  338. return
  339. }
  340. } else {
  341. nodes, err = getUsersNodes(user)
  342. if err != nil {
  343. logger.Log(0, r.Header.Get("user"),
  344. "error fetching nodes: ", err.Error())
  345. returnErrorResponse(w, r, formatError(err, "internal"))
  346. return
  347. }
  348. }
  349. //Return all the nodes in JSON format
  350. logger.Log(3, r.Header.Get("user"), "fetched all nodes they have access to")
  351. w.WriteHeader(http.StatusOK)
  352. json.NewEncoder(w).Encode(nodes)
  353. }
  354. func getUsersNodes(user models.User) ([]models.Node, error) {
  355. var nodes []models.Node
  356. var err error
  357. for _, networkName := range user.Networks {
  358. tmpNodes, err := logic.GetNetworkNodes(networkName)
  359. if err != nil {
  360. continue
  361. }
  362. nodes = append(nodes, tmpNodes...)
  363. }
  364. return nodes, err
  365. }
  366. // swagger:route GET /api/nodes/{network}/{nodeid} nodes getNode
  367. //
  368. // Get an individual node.
  369. //
  370. // Schemes: https
  371. //
  372. // Security:
  373. // oauth
  374. //
  375. // Responses:
  376. // 200: nodeResponse
  377. func getNode(w http.ResponseWriter, r *http.Request) {
  378. // set header.
  379. w.Header().Set("Content-Type", "application/json")
  380. var params = mux.Vars(r)
  381. nodeid := params["nodeid"]
  382. node, err := logic.GetNodeByID(nodeid)
  383. if err != nil {
  384. logger.Log(0, r.Header.Get("user"),
  385. fmt.Sprintf("error fetching node [ %s ] info: %v", nodeid, err))
  386. returnErrorResponse(w, r, formatError(err, "internal"))
  387. return
  388. }
  389. peerUpdate, err := logic.GetPeerUpdate(&node)
  390. if err != nil && !database.IsEmptyRecord(err) {
  391. logger.Log(0, r.Header.Get("user"),
  392. fmt.Sprintf("error fetching wg peers config for node [ %s ]: %v", nodeid, err))
  393. returnErrorResponse(w, r, formatError(err, "internal"))
  394. return
  395. }
  396. if len(node.NetworkSettings.AccessKeys) > 0 {
  397. node.NetworkSettings.AccessKeys = []models.AccessKey{} // not to be sent back to client; client already knows how to join the network
  398. }
  399. response := models.NodeGet{
  400. Node: node,
  401. Peers: peerUpdate.Peers,
  402. ServerConfig: servercfg.GetServerInfo(),
  403. }
  404. logger.Log(2, r.Header.Get("user"), "fetched node", params["nodeid"])
  405. w.WriteHeader(http.StatusOK)
  406. json.NewEncoder(w).Encode(response)
  407. }
  408. // swagger:route GET /api/nodes/adm/{network}/lastmodified nodes getLastModified
  409. //
  410. // Get the time that a network of nodes was last modified.
  411. //
  412. // Schemes: https
  413. //
  414. // Security:
  415. // oauth
  416. //
  417. // Responses:
  418. // 200: nodeLastModifiedResponse
  419. // TODO: This needs to be refactored
  420. // Potential way to do this: On UpdateNode, set a new field for "LastModified"
  421. // If we go with the existing way, we need to at least set network.NodesLastModified on UpdateNode
  422. func getLastModified(w http.ResponseWriter, r *http.Request) {
  423. // set header.
  424. w.Header().Set("Content-Type", "application/json")
  425. var params = mux.Vars(r)
  426. networkName := params["network"]
  427. network, err := logic.GetNetwork(networkName)
  428. if err != nil {
  429. logger.Log(0, r.Header.Get("user"),
  430. fmt.Sprintf("error fetching network [%s] info: %v", networkName, err))
  431. returnErrorResponse(w, r, formatError(err, "internal"))
  432. return
  433. }
  434. logger.Log(2, r.Header.Get("user"), "called last modified")
  435. w.WriteHeader(http.StatusOK)
  436. json.NewEncoder(w).Encode(network.NodesLastModified)
  437. }
  438. // swagger:route POST /api/nodes/{network} nodes createNode
  439. //
  440. // Create a node on a network.
  441. //
  442. // Schemes: https
  443. //
  444. // Security:
  445. // oauth
  446. //
  447. // Responses:
  448. // 200: nodeGetResponse
  449. func createNode(w http.ResponseWriter, r *http.Request) {
  450. w.Header().Set("Content-Type", "application/json")
  451. var params = mux.Vars(r)
  452. var errorResponse = models.ErrorResponse{
  453. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  454. }
  455. networkName := params["network"]
  456. networkexists, err := functions.NetworkExists(networkName)
  457. if err != nil {
  458. logger.Log(0, r.Header.Get("user"),
  459. fmt.Sprintf("failed to fetch network [%s] info: %v", networkName, err))
  460. returnErrorResponse(w, r, formatError(err, "internal"))
  461. return
  462. } else if !networkexists {
  463. errorResponse = models.ErrorResponse{
  464. Code: http.StatusNotFound, Message: "W1R3: Network does not exist! ",
  465. }
  466. logger.Log(0, r.Header.Get("user"),
  467. fmt.Sprintf("network [%s] does not exist", networkName))
  468. returnErrorResponse(w, r, errorResponse)
  469. return
  470. }
  471. var node = models.Node{}
  472. //get node from body of request
  473. err = json.NewDecoder(r.Body).Decode(&node)
  474. if err != nil {
  475. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  476. returnErrorResponse(w, r, formatError(err, "badrequest"))
  477. return
  478. }
  479. node.Network = networkName
  480. network, err := logic.GetNetworkByNode(&node)
  481. if err != nil {
  482. logger.Log(0, r.Header.Get("user"),
  483. fmt.Sprintf("failed to get network [%s] info: %v", node.Network, err))
  484. returnErrorResponse(w, r, formatError(err, "internal"))
  485. return
  486. }
  487. node.NetworkSettings, err = logic.GetNetworkSettings(node.Network)
  488. if err != nil {
  489. logger.Log(0, r.Header.Get("user"),
  490. fmt.Sprintf("failed to get network [%s] settings: %v", node.Network, err))
  491. returnErrorResponse(w, r, formatError(err, "internal"))
  492. return
  493. }
  494. validKey := logic.IsKeyValid(networkName, node.AccessKey)
  495. if !validKey {
  496. // Check to see if network will allow manual sign up
  497. // may want to switch this up with the valid key check and avoid a DB call that way.
  498. if network.AllowManualSignUp == "yes" {
  499. node.IsPending = "yes"
  500. } else {
  501. errorResponse = models.ErrorResponse{
  502. Code: http.StatusUnauthorized, Message: "W1R3: Key invalid, or none provided.",
  503. }
  504. logger.Log(0, r.Header.Get("user"),
  505. fmt.Sprintf("failed to create node on network [%s]: %s",
  506. node.Network, errorResponse.Message))
  507. returnErrorResponse(w, r, errorResponse)
  508. return
  509. }
  510. }
  511. key, keyErr := logic.RetrievePublicTrafficKey()
  512. if keyErr != nil {
  513. logger.Log(0, "error retrieving key: ", keyErr.Error())
  514. returnErrorResponse(w, r, formatError(err, "internal"))
  515. return
  516. }
  517. if key == nil {
  518. logger.Log(0, "error: server traffic key is nil")
  519. returnErrorResponse(w, r, formatError(err, "internal"))
  520. return
  521. }
  522. if node.TrafficKeys.Mine == nil {
  523. logger.Log(0, "error: node traffic key is nil")
  524. returnErrorResponse(w, r, formatError(err, "internal"))
  525. return
  526. }
  527. node.TrafficKeys = models.TrafficKeys{
  528. Mine: node.TrafficKeys.Mine,
  529. Server: key,
  530. }
  531. err = logic.CreateNode(&node)
  532. if err != nil {
  533. logger.Log(0, r.Header.Get("user"),
  534. fmt.Sprintf("failed to create node on network [%s]: %s",
  535. node.Network, err))
  536. returnErrorResponse(w, r, formatError(err, "internal"))
  537. return
  538. }
  539. peerUpdate, err := logic.GetPeerUpdate(&node)
  540. if err != nil && !database.IsEmptyRecord(err) {
  541. logger.Log(0, r.Header.Get("user"),
  542. fmt.Sprintf("error fetching wg peers config for node [ %s ]: %v", node.ID, err))
  543. returnErrorResponse(w, r, formatError(err, "internal"))
  544. return
  545. }
  546. // Create client for this node in Mq
  547. // Delete Any Existing Client
  548. mq.DynSecChan <- mq.DynSecAction{
  549. ActionType: mq.CreateClient,
  550. Payload: mq.MqDynsecPayload{
  551. Commands: []mq.MqDynSecCmd{
  552. {
  553. Command: mq.CreateClientCmd,
  554. Username: node.ID,
  555. Password: node.Password,
  556. Roles: make([]mq.MqDynSecRole, 0),
  557. Groups: make([]mq.MqDynSecGroup, 0),
  558. },
  559. },
  560. },
  561. }
  562. response := models.NodeGet{
  563. Node: node,
  564. Peers: peerUpdate.Peers,
  565. ServerConfig: servercfg.GetServerInfo(),
  566. }
  567. logger.Log(1, r.Header.Get("user"), "created new node", node.Name, "on network", node.Network)
  568. w.WriteHeader(http.StatusOK)
  569. json.NewEncoder(w).Encode(response)
  570. runForceServerUpdate(&node, true)
  571. }
  572. // swagger:route POST /api/nodes/{network}/{nodeid}/approve nodes uncordonNode
  573. //
  574. // Takes a node out of pending state.
  575. //
  576. // Schemes: https
  577. //
  578. // Security:
  579. // oauth
  580. //
  581. // Responses:
  582. // 200: nodeResponse
  583. // Takes node out of pending state
  584. // TODO: May want to use cordon/uncordon terminology instead of "ispending".
  585. func uncordonNode(w http.ResponseWriter, r *http.Request) {
  586. var params = mux.Vars(r)
  587. w.Header().Set("Content-Type", "application/json")
  588. var nodeid = params["nodeid"]
  589. node, err := logic.UncordonNode(nodeid)
  590. if err != nil {
  591. logger.Log(0, r.Header.Get("user"),
  592. fmt.Sprintf("failed to uncordon node [%s]: %v", node.Name, err))
  593. returnErrorResponse(w, r, formatError(err, "internal"))
  594. return
  595. }
  596. logger.Log(1, r.Header.Get("user"), "uncordoned node", node.Name)
  597. w.WriteHeader(http.StatusOK)
  598. json.NewEncoder(w).Encode("SUCCESS")
  599. runUpdates(&node, false)
  600. }
  601. // == EGRESS ==
  602. // swagger:route POST /api/nodes/{network}/{nodeid}/creategateway nodes createEgressGateway
  603. //
  604. // Create an egress gateway.
  605. //
  606. // Schemes: https
  607. //
  608. // Security:
  609. // oauth
  610. //
  611. // Responses:
  612. // 200: nodeResponse
  613. func createEgressGateway(w http.ResponseWriter, r *http.Request) {
  614. var gateway models.EgressGatewayRequest
  615. var params = mux.Vars(r)
  616. w.Header().Set("Content-Type", "application/json")
  617. err := json.NewDecoder(r.Body).Decode(&gateway)
  618. if err != nil {
  619. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  620. returnErrorResponse(w, r, formatError(err, "badrequest"))
  621. return
  622. }
  623. gateway.NetID = params["network"]
  624. gateway.NodeID = params["nodeid"]
  625. node, err := logic.CreateEgressGateway(gateway)
  626. if err != nil {
  627. logger.Log(0, r.Header.Get("user"),
  628. fmt.Sprintf("failed to create egress gateway on node [%s] on network [%s]: %v",
  629. gateway.NodeID, gateway.NetID, err))
  630. returnErrorResponse(w, r, formatError(err, "internal"))
  631. return
  632. }
  633. logger.Log(1, r.Header.Get("user"), "created egress gateway on node", gateway.NodeID, "on network", gateway.NetID)
  634. w.WriteHeader(http.StatusOK)
  635. json.NewEncoder(w).Encode(node)
  636. runUpdates(&node, true)
  637. }
  638. // swagger:route DELETE /api/nodes/{network}/{nodeid}/deletegateway nodes deleteEgressGateway
  639. //
  640. // Delete an egress gateway.
  641. //
  642. // Schemes: https
  643. //
  644. // Security:
  645. // oauth
  646. //
  647. // Responses:
  648. // 200: nodeResponse
  649. func deleteEgressGateway(w http.ResponseWriter, r *http.Request) {
  650. w.Header().Set("Content-Type", "application/json")
  651. var params = mux.Vars(r)
  652. nodeid := params["nodeid"]
  653. netid := params["network"]
  654. node, err := logic.DeleteEgressGateway(netid, nodeid)
  655. if err != nil {
  656. logger.Log(0, r.Header.Get("user"),
  657. fmt.Sprintf("failed to delete egress gateway on node [%s] on network [%s]: %v",
  658. nodeid, netid, err))
  659. returnErrorResponse(w, r, formatError(err, "internal"))
  660. return
  661. }
  662. logger.Log(1, r.Header.Get("user"), "deleted egress gateway on node", nodeid, "on network", netid)
  663. w.WriteHeader(http.StatusOK)
  664. json.NewEncoder(w).Encode(node)
  665. runUpdates(&node, true)
  666. }
  667. // == INGRESS ==
  668. // swagger:route POST /api/nodes/{network}/{nodeid}/createingress nodes createIngressGateway
  669. //
  670. // Create an ingress gateway.
  671. //
  672. // Schemes: https
  673. //
  674. // Security:
  675. // oauth
  676. //
  677. // Responses:
  678. // 200: nodeResponse
  679. func createIngressGateway(w http.ResponseWriter, r *http.Request) {
  680. var params = mux.Vars(r)
  681. w.Header().Set("Content-Type", "application/json")
  682. nodeid := params["nodeid"]
  683. netid := params["network"]
  684. node, err := logic.CreateIngressGateway(netid, nodeid)
  685. if err != nil {
  686. logger.Log(0, r.Header.Get("user"),
  687. fmt.Sprintf("failed to create ingress gateway on node [%s] on network [%s]: %v",
  688. nodeid, netid, err))
  689. returnErrorResponse(w, r, formatError(err, "internal"))
  690. return
  691. }
  692. logger.Log(1, r.Header.Get("user"), "created ingress gateway on node", nodeid, "on network", netid)
  693. w.WriteHeader(http.StatusOK)
  694. json.NewEncoder(w).Encode(node)
  695. runUpdates(&node, true)
  696. }
  697. // swagger:route DELETE /api/nodes/{network}/{nodeid}/deleteingress nodes deleteIngressGateway
  698. //
  699. // Delete an ingress gateway.
  700. //
  701. // Schemes: https
  702. //
  703. // Security:
  704. // oauth
  705. //
  706. // Responses:
  707. // 200: nodeResponse
  708. func deleteIngressGateway(w http.ResponseWriter, r *http.Request) {
  709. w.Header().Set("Content-Type", "application/json")
  710. var params = mux.Vars(r)
  711. nodeid := params["nodeid"]
  712. netid := params["network"]
  713. node, err := logic.DeleteIngressGateway(netid, nodeid)
  714. if err != nil {
  715. logger.Log(0, r.Header.Get("user"),
  716. fmt.Sprintf("failed to delete ingress gateway on node [%s] on network [%s]: %v",
  717. nodeid, netid, err))
  718. returnErrorResponse(w, r, formatError(err, "internal"))
  719. return
  720. }
  721. logger.Log(1, r.Header.Get("user"), "deleted ingress gateway", nodeid)
  722. w.WriteHeader(http.StatusOK)
  723. json.NewEncoder(w).Encode(node)
  724. runUpdates(&node, true)
  725. }
  726. // swagger:route PUT /api/nodes/{network}/{nodeid} nodes updateNode
  727. //
  728. // Update an individual node.
  729. //
  730. // Schemes: https
  731. //
  732. // Security:
  733. // oauth
  734. //
  735. // Responses:
  736. // 200: nodeResponse
  737. func updateNode(w http.ResponseWriter, r *http.Request) {
  738. w.Header().Set("Content-Type", "application/json")
  739. var params = mux.Vars(r)
  740. var node models.Node
  741. //start here
  742. nodeid := params["nodeid"]
  743. node, err := logic.GetNodeByID(nodeid)
  744. if err != nil {
  745. logger.Log(0, r.Header.Get("user"),
  746. fmt.Sprintf("error fetching node [ %s ] info: %v", nodeid, err))
  747. returnErrorResponse(w, r, formatError(err, "internal"))
  748. return
  749. }
  750. var newNode models.Node
  751. // we decode our body request params
  752. err = json.NewDecoder(r.Body).Decode(&newNode)
  753. if err != nil {
  754. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  755. returnErrorResponse(w, r, formatError(err, "badrequest"))
  756. return
  757. }
  758. relayupdate := false
  759. if node.IsRelay == "yes" && len(newNode.RelayAddrs) > 0 {
  760. if len(newNode.RelayAddrs) != len(node.RelayAddrs) {
  761. relayupdate = true
  762. } else {
  763. for i, addr := range newNode.RelayAddrs {
  764. if addr != node.RelayAddrs[i] {
  765. relayupdate = true
  766. }
  767. }
  768. }
  769. }
  770. relayedUpdate := false
  771. if node.IsRelayed == "yes" && (node.Address != newNode.Address || node.Address6 != newNode.Address6) {
  772. relayedUpdate = true
  773. }
  774. if !servercfg.GetRce() {
  775. newNode.PostDown = node.PostDown
  776. newNode.PostUp = node.PostUp
  777. }
  778. ifaceDelta := logic.IfaceDelta(&node, &newNode)
  779. // for a hub change also need to update the existing hub
  780. if newNode.IsHub == "yes" && node.IsHub != "yes" {
  781. nodeToUpdate, err := logic.UnsetHub(newNode.Network)
  782. if err != nil {
  783. logger.Log(2, "failed to unset hubs", err.Error())
  784. }
  785. if err := mq.NodeUpdate(nodeToUpdate); err != nil {
  786. logger.Log(2, "failed to update hub node", nodeToUpdate.Name, err.Error())
  787. }
  788. if nodeToUpdate.IsServer == "yes" {
  789. // set ifacdelta true to force server to update peeers
  790. if err := logic.ServerUpdate(nodeToUpdate, true); err != nil {
  791. logger.Log(2, "failed to update server node on hub change", err.Error())
  792. }
  793. }
  794. }
  795. err = logic.UpdateNode(&node, &newNode)
  796. if err != nil {
  797. logger.Log(0, r.Header.Get("user"),
  798. fmt.Sprintf("failed to update node info [ %s ] info: %v", nodeid, err))
  799. returnErrorResponse(w, r, formatError(err, "internal"))
  800. return
  801. }
  802. if relayupdate {
  803. updatenodes := logic.UpdateRelay(node.Network, node.RelayAddrs, newNode.RelayAddrs)
  804. if len(updatenodes) > 0 {
  805. for _, relayedNode := range updatenodes {
  806. runUpdates(&relayedNode, false)
  807. }
  808. }
  809. }
  810. if relayedUpdate {
  811. updateRelay(&node, &newNode)
  812. }
  813. if servercfg.IsDNSMode() {
  814. logic.SetDNS()
  815. }
  816. logger.Log(1, r.Header.Get("user"), "updated node", node.ID, "on network", node.Network)
  817. w.WriteHeader(http.StatusOK)
  818. json.NewEncoder(w).Encode(newNode)
  819. runUpdates(&newNode, ifaceDelta)
  820. }
  821. // swagger:route DELETE /api/nodes/{network}/{nodeid} nodes deleteNode
  822. //
  823. // Delete an individual node.
  824. //
  825. // Schemes: https
  826. //
  827. // Security:
  828. // oauth
  829. //
  830. // Responses:
  831. // 200: nodeResponse
  832. func deleteNode(w http.ResponseWriter, r *http.Request) {
  833. // Set header
  834. w.Header().Set("Content-Type", "application/json")
  835. // get params
  836. var params = mux.Vars(r)
  837. var nodeid = params["nodeid"]
  838. var node, err = logic.GetNodeByID(nodeid)
  839. if err != nil {
  840. logger.Log(0, r.Header.Get("user"),
  841. fmt.Sprintf("error fetching node [ %s ] info: %v", nodeid, err))
  842. returnErrorResponse(w, r, formatError(err, "badrequest"))
  843. return
  844. }
  845. if isServer(&node) {
  846. err := fmt.Errorf("cannot delete server node")
  847. logger.Log(0, r.Header.Get("user"),
  848. fmt.Sprintf("failed to delete node [ %s ]: %v", nodeid, err))
  849. returnErrorResponse(w, r, formatError(err, "badrequest"))
  850. return
  851. }
  852. //send update to node to be deleted before deleting on server otherwise message cannot be sent
  853. node.Action = models.NODE_DELETE
  854. err = logic.DeleteNodeByID(&node, false)
  855. if err != nil {
  856. returnErrorResponse(w, r, formatError(err, "internal"))
  857. return
  858. }
  859. returnSuccessResponse(w, r, nodeid+" deleted.")
  860. logger.Log(1, r.Header.Get("user"), "Deleted node", nodeid, "from network", params["network"])
  861. runUpdates(&node, false)
  862. runForceServerUpdate(&node, false)
  863. }
  864. func runUpdates(node *models.Node, ifaceDelta bool) {
  865. go func() { // don't block http response
  866. // publish node update if not server
  867. if err := mq.NodeUpdate(node); err != nil {
  868. logger.Log(1, "error publishing node update to node", node.Name, node.ID, err.Error())
  869. }
  870. if err := runServerUpdate(node, ifaceDelta); err != nil {
  871. logger.Log(1, "error running server update", err.Error())
  872. }
  873. }()
  874. }
  875. // updates local peers for a server on a given node's network
  876. func runServerUpdate(node *models.Node, ifaceDelta bool) error {
  877. if servercfg.IsClientMode() != "on" || !isServer(node) {
  878. return nil
  879. }
  880. currentServerNode, err := logic.GetNetworkServerLocal(node.Network)
  881. if err != nil {
  882. return err
  883. }
  884. if ifaceDelta && logic.IsLeader(&currentServerNode) {
  885. if err := mq.PublishPeerUpdate(&currentServerNode, false); err != nil {
  886. logger.Log(1, "failed to publish peer update "+err.Error())
  887. }
  888. }
  889. if err := logic.ServerUpdate(&currentServerNode, ifaceDelta); err != nil {
  890. logger.Log(1, "server node:", currentServerNode.ID, "failed update")
  891. return err
  892. }
  893. return nil
  894. }
  895. func runForceServerUpdate(node *models.Node, publishPeerUpdateToNode bool) {
  896. go func() {
  897. if err := mq.PublishPeerUpdate(node, publishPeerUpdateToNode); err != nil {
  898. logger.Log(1, "failed a peer update after creation of node", node.Name)
  899. }
  900. var currentServerNode, getErr = logic.GetNetworkServerLeader(node.Network)
  901. if getErr == nil {
  902. if err := logic.ServerUpdate(&currentServerNode, false); err != nil {
  903. logger.Log(1, "server node:", currentServerNode.ID, "failed update")
  904. }
  905. }
  906. }()
  907. }
  908. func isServer(node *models.Node) bool {
  909. return node.IsServer == "yes"
  910. }
  911. func updateRelay(oldnode, newnode *models.Node) {
  912. relay := logic.FindRelay(oldnode)
  913. newrelay := relay
  914. //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
  915. if oldnode.Address != newnode.Address {
  916. for i, ip := range newrelay.RelayAddrs {
  917. if ip == oldnode.Address {
  918. newrelay.RelayAddrs = append(newrelay.RelayAddrs[:i], relay.RelayAddrs[i+1:]...)
  919. newrelay.RelayAddrs = append(newrelay.RelayAddrs, newnode.Address)
  920. }
  921. }
  922. }
  923. //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
  924. if oldnode.Address6 != newnode.Address6 {
  925. for i, ip := range newrelay.RelayAddrs {
  926. if ip == oldnode.Address {
  927. newrelay.RelayAddrs = append(newrelay.RelayAddrs[:i], newrelay.RelayAddrs[i+1:]...)
  928. newrelay.RelayAddrs = append(newrelay.RelayAddrs, newnode.Address6)
  929. }
  930. }
  931. }
  932. logic.UpdateNode(relay, newrelay)
  933. }