node.go 35 KB

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