node.go 36 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171
  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. // Delete Any Existing Client with this ID.
  612. event := mq.MqDynsecPayload{
  613. Commands: []mq.MqDynSecCmd{
  614. {
  615. Command: mq.DeleteClientCmd,
  616. Username: node.ID,
  617. },
  618. },
  619. }
  620. if err := mq.PublishEventToDynSecTopic(event); err != nil {
  621. logger.Log(0, fmt.Sprintf("failed to send DynSec command [%v]: %v",
  622. event.Commands, err.Error()))
  623. }
  624. // Create client for this node in Mq
  625. event = mq.MqDynsecPayload{
  626. Commands: []mq.MqDynSecCmd{
  627. {
  628. Command: mq.CreateRoleCmd,
  629. RoleName: fmt.Sprintf("%s-%s", "Node", node.ID),
  630. Acls: mq.FetchNodeAcls(node.ID),
  631. Textname: "Role for node " + node.Name,
  632. },
  633. {
  634. Command: mq.CreateClientCmd,
  635. Username: node.ID,
  636. Password: nodePassword,
  637. Textname: node.Name,
  638. Roles: []mq.MqDynSecRole{
  639. {
  640. Rolename: fmt.Sprintf("%s-%s", "Node", node.ID),
  641. Priority: -1,
  642. },
  643. {
  644. Rolename: node.Network,
  645. Priority: -1,
  646. },
  647. },
  648. Groups: make([]mq.MqDynSecGroup, 0),
  649. },
  650. },
  651. }
  652. if err := mq.PublishEventToDynSecTopic(event); err != nil {
  653. logger.Log(0, fmt.Sprintf("failed to send DynSec command [%v]: %v",
  654. event.Commands, err.Error()))
  655. }
  656. response := models.NodeGet{
  657. Node: node,
  658. Peers: peerUpdate.Peers,
  659. ServerConfig: servercfg.GetServerInfo(),
  660. PeerIDs: peerUpdate.PeerIDs,
  661. }
  662. logger.Log(1, r.Header.Get("user"), "created new node", node.Name, "on network", node.Network)
  663. w.WriteHeader(http.StatusOK)
  664. json.NewEncoder(w).Encode(response)
  665. runForceServerUpdate(&node, true)
  666. }
  667. // swagger:route POST /api/nodes/{network}/{nodeid}/approve nodes uncordonNode
  668. //
  669. // Takes a node out of pending state.
  670. //
  671. // Schemes: https
  672. //
  673. // Security:
  674. // oauth
  675. //
  676. // Responses:
  677. // 200: nodeResponse
  678. // Takes node out of pending state
  679. // TODO: May want to use cordon/uncordon terminology instead of "ispending".
  680. func uncordonNode(w http.ResponseWriter, r *http.Request) {
  681. var params = mux.Vars(r)
  682. w.Header().Set("Content-Type", "application/json")
  683. var nodeid = params["nodeid"]
  684. node, err := logic.UncordonNode(nodeid)
  685. if err != nil {
  686. logger.Log(0, r.Header.Get("user"),
  687. fmt.Sprintf("failed to uncordon node [%s]: %v", node.Name, err))
  688. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  689. return
  690. }
  691. logger.Log(1, r.Header.Get("user"), "uncordoned node", node.Name)
  692. w.WriteHeader(http.StatusOK)
  693. json.NewEncoder(w).Encode("SUCCESS")
  694. runUpdates(&node, false)
  695. }
  696. // == EGRESS ==
  697. // swagger:route POST /api/nodes/{network}/{nodeid}/creategateway nodes createEgressGateway
  698. //
  699. // Create an egress gateway.
  700. //
  701. // Schemes: https
  702. //
  703. // Security:
  704. // oauth
  705. //
  706. // Responses:
  707. // 200: nodeResponse
  708. func createEgressGateway(w http.ResponseWriter, r *http.Request) {
  709. var gateway models.EgressGatewayRequest
  710. var params = mux.Vars(r)
  711. w.Header().Set("Content-Type", "application/json")
  712. err := json.NewDecoder(r.Body).Decode(&gateway)
  713. if err != nil {
  714. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  715. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  716. return
  717. }
  718. gateway.NetID = params["network"]
  719. gateway.NodeID = params["nodeid"]
  720. node, err := logic.CreateEgressGateway(gateway)
  721. if err != nil {
  722. logger.Log(0, r.Header.Get("user"),
  723. fmt.Sprintf("failed to create egress gateway on node [%s] on network [%s]: %v",
  724. gateway.NodeID, gateway.NetID, err))
  725. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  726. return
  727. }
  728. logger.Log(1, r.Header.Get("user"), "created egress gateway on node", gateway.NodeID, "on network", gateway.NetID)
  729. w.WriteHeader(http.StatusOK)
  730. json.NewEncoder(w).Encode(node)
  731. runUpdates(&node, true)
  732. }
  733. // swagger:route DELETE /api/nodes/{network}/{nodeid}/deletegateway nodes deleteEgressGateway
  734. //
  735. // Delete an egress gateway.
  736. //
  737. // Schemes: https
  738. //
  739. // Security:
  740. // oauth
  741. //
  742. // Responses:
  743. // 200: nodeResponse
  744. func deleteEgressGateway(w http.ResponseWriter, r *http.Request) {
  745. w.Header().Set("Content-Type", "application/json")
  746. var params = mux.Vars(r)
  747. nodeid := params["nodeid"]
  748. netid := params["network"]
  749. node, err := logic.DeleteEgressGateway(netid, nodeid)
  750. if err != nil {
  751. logger.Log(0, r.Header.Get("user"),
  752. fmt.Sprintf("failed to delete egress gateway on node [%s] on network [%s]: %v",
  753. nodeid, netid, err))
  754. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  755. return
  756. }
  757. logger.Log(1, r.Header.Get("user"), "deleted egress gateway on node", nodeid, "on network", netid)
  758. w.WriteHeader(http.StatusOK)
  759. json.NewEncoder(w).Encode(node)
  760. runUpdates(&node, true)
  761. }
  762. // == INGRESS ==
  763. // swagger:route POST /api/nodes/{network}/{nodeid}/createingress nodes createIngressGateway
  764. //
  765. // Create an ingress gateway.
  766. //
  767. // Schemes: https
  768. //
  769. // Security:
  770. // oauth
  771. //
  772. // Responses:
  773. // 200: nodeResponse
  774. func createIngressGateway(w http.ResponseWriter, r *http.Request) {
  775. var params = mux.Vars(r)
  776. w.Header().Set("Content-Type", "application/json")
  777. nodeid := params["nodeid"]
  778. netid := params["network"]
  779. node, err := logic.CreateIngressGateway(netid, nodeid)
  780. if err != nil {
  781. logger.Log(0, r.Header.Get("user"),
  782. fmt.Sprintf("failed to create ingress gateway on node [%s] on network [%s]: %v",
  783. nodeid, netid, err))
  784. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  785. return
  786. }
  787. logger.Log(1, r.Header.Get("user"), "created ingress gateway on node", nodeid, "on network", netid)
  788. w.WriteHeader(http.StatusOK)
  789. json.NewEncoder(w).Encode(node)
  790. runUpdates(&node, true)
  791. }
  792. // swagger:route DELETE /api/nodes/{network}/{nodeid}/deleteingress nodes deleteIngressGateway
  793. //
  794. // Delete an ingress gateway.
  795. //
  796. // Schemes: https
  797. //
  798. // Security:
  799. // oauth
  800. //
  801. // Responses:
  802. // 200: nodeResponse
  803. func deleteIngressGateway(w http.ResponseWriter, r *http.Request) {
  804. w.Header().Set("Content-Type", "application/json")
  805. var params = mux.Vars(r)
  806. nodeid := params["nodeid"]
  807. netid := params["network"]
  808. node, err := logic.DeleteIngressGateway(netid, nodeid)
  809. if err != nil {
  810. logger.Log(0, r.Header.Get("user"),
  811. fmt.Sprintf("failed to delete ingress gateway on node [%s] on network [%s]: %v",
  812. nodeid, netid, err))
  813. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  814. return
  815. }
  816. logger.Log(1, r.Header.Get("user"), "deleted ingress gateway", nodeid)
  817. w.WriteHeader(http.StatusOK)
  818. json.NewEncoder(w).Encode(node)
  819. runUpdates(&node, true)
  820. }
  821. // swagger:route PUT /api/nodes/{network}/{nodeid} nodes updateNode
  822. //
  823. // Update an individual node.
  824. //
  825. // Schemes: https
  826. //
  827. // Security:
  828. // oauth
  829. //
  830. // Responses:
  831. // 200: nodeResponse
  832. func updateNode(w http.ResponseWriter, r *http.Request) {
  833. w.Header().Set("Content-Type", "application/json")
  834. var params = mux.Vars(r)
  835. var node models.Node
  836. //start here
  837. nodeid := params["nodeid"]
  838. 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. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  843. return
  844. }
  845. var newNode models.Node
  846. // we decode our body request params
  847. err = json.NewDecoder(r.Body).Decode(&newNode)
  848. if err != nil {
  849. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  850. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  851. return
  852. }
  853. relayupdate := false
  854. if node.IsRelay == "yes" && len(newNode.RelayAddrs) > 0 {
  855. if len(newNode.RelayAddrs) != len(node.RelayAddrs) {
  856. relayupdate = true
  857. } else {
  858. for i, addr := range newNode.RelayAddrs {
  859. if addr != node.RelayAddrs[i] {
  860. relayupdate = true
  861. }
  862. }
  863. }
  864. }
  865. relayedUpdate := false
  866. if node.IsRelayed == "yes" && (node.Address != newNode.Address || node.Address6 != newNode.Address6) {
  867. relayedUpdate = true
  868. }
  869. if !servercfg.GetRce() {
  870. newNode.PostDown = node.PostDown
  871. newNode.PostUp = node.PostUp
  872. }
  873. ifaceDelta := logic.IfaceDelta(&node, &newNode)
  874. // for a hub change also need to update the existing hub
  875. if newNode.IsHub == "yes" && node.IsHub != "yes" {
  876. nodeToUpdate, err := logic.UnsetHub(newNode.Network)
  877. if err != nil {
  878. logger.Log(2, "failed to unset hubs", err.Error())
  879. }
  880. if err := mq.NodeUpdate(nodeToUpdate); err != nil {
  881. logger.Log(2, "failed to update hub node", nodeToUpdate.Name, err.Error())
  882. }
  883. if nodeToUpdate.IsServer == "yes" {
  884. // set ifacdelta true to force server to update peeers
  885. if err := logic.ServerUpdate(nodeToUpdate, true); err != nil {
  886. logger.Log(2, "failed to update server node on hub change", err.Error())
  887. }
  888. }
  889. }
  890. err = logic.UpdateNode(&node, &newNode)
  891. if err != nil {
  892. logger.Log(0, r.Header.Get("user"),
  893. fmt.Sprintf("failed to update node info [ %s ] info: %v", nodeid, err))
  894. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  895. return
  896. }
  897. if relayupdate {
  898. updatenodes := logic.UpdateRelay(node.Network, node.RelayAddrs, newNode.RelayAddrs)
  899. if len(updatenodes) > 0 {
  900. for _, relayedNode := range updatenodes {
  901. runUpdates(&relayedNode, false)
  902. }
  903. }
  904. }
  905. if relayedUpdate {
  906. updateRelay(&node, &newNode)
  907. }
  908. if servercfg.IsDNSMode() {
  909. logic.SetDNS()
  910. }
  911. logger.Log(1, r.Header.Get("user"), "updated node", node.ID, "on network", node.Network)
  912. w.WriteHeader(http.StatusOK)
  913. json.NewEncoder(w).Encode(newNode)
  914. runUpdates(&newNode, ifaceDelta)
  915. }
  916. // swagger:route DELETE /api/nodes/{network}/{nodeid} nodes deleteNode
  917. //
  918. // Delete an individual node.
  919. //
  920. // Schemes: https
  921. //
  922. // Security:
  923. // oauth
  924. //
  925. // Responses:
  926. // 200: nodeResponse
  927. func deleteNode(w http.ResponseWriter, r *http.Request) {
  928. // Set header
  929. w.Header().Set("Content-Type", "application/json")
  930. // get params
  931. var params = mux.Vars(r)
  932. var nodeid = params["nodeid"]
  933. var node, err = logic.GetNodeByID(nodeid)
  934. if err != nil {
  935. logger.Log(0, r.Header.Get("user"),
  936. fmt.Sprintf("error fetching node [ %s ] info: %v", nodeid, err))
  937. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  938. return
  939. }
  940. if isServer(&node) {
  941. err := fmt.Errorf("cannot delete server node")
  942. logger.Log(0, r.Header.Get("user"),
  943. fmt.Sprintf("failed to delete node [ %s ]: %v", nodeid, err))
  944. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
  945. return
  946. }
  947. if r.Header.Get("ismaster") != "yes" {
  948. username := r.Header.Get("user")
  949. if username != "" && !doesUserOwnNode(username, params["network"], nodeid) {
  950. logic.ReturnErrorResponse(w, r, logic.FormatError(fmt.Errorf("user not permitted"), "badrequest"))
  951. return
  952. }
  953. }
  954. //send update to node to be deleted before deleting on server otherwise message cannot be sent
  955. node.Action = models.NODE_DELETE
  956. err = logic.DeleteNodeByID(&node, false)
  957. if err != nil {
  958. logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
  959. return
  960. }
  961. // deletes node related role and client
  962. event := mq.MqDynsecPayload{
  963. Commands: []mq.MqDynSecCmd{
  964. {
  965. Command: mq.DeleteRoleCmd,
  966. RoleName: fmt.Sprintf("%s-%s", "Node", nodeid),
  967. },
  968. {
  969. Command: mq.DeleteClientCmd,
  970. Username: nodeid,
  971. },
  972. },
  973. }
  974. if err := mq.PublishEventToDynSecTopic(event); err != nil {
  975. logger.Log(0, fmt.Sprintf("failed to send DynSec command [%v]: %v",
  976. event.Commands, err.Error()))
  977. }
  978. logic.ReturnSuccessResponse(w, r, nodeid+" deleted.")
  979. logger.Log(1, r.Header.Get("user"), "Deleted node", nodeid, "from network", params["network"])
  980. runUpdates(&node, false)
  981. runForceServerUpdate(&node, false)
  982. }
  983. func runUpdates(node *models.Node, ifaceDelta bool) {
  984. go func() { // don't block http response
  985. // publish node update if not server
  986. if err := mq.NodeUpdate(node); err != nil {
  987. logger.Log(1, "error publishing node update to node", node.Name, node.ID, err.Error())
  988. }
  989. if err := runServerUpdate(node, ifaceDelta); err != nil {
  990. logger.Log(1, "error running server update", err.Error())
  991. }
  992. }()
  993. }
  994. // updates local peers for a server on a given node's network
  995. func runServerUpdate(node *models.Node, ifaceDelta bool) error {
  996. if servercfg.IsClientMode() != "on" || !isServer(node) {
  997. return nil
  998. }
  999. currentServerNode, err := logic.GetNetworkServerLocal(node.Network)
  1000. if err != nil {
  1001. return err
  1002. }
  1003. if ifaceDelta && logic.IsLeader(&currentServerNode) {
  1004. if err := mq.PublishPeerUpdate(&currentServerNode, false); err != nil {
  1005. logger.Log(1, "failed to publish peer update "+err.Error())
  1006. }
  1007. }
  1008. if err := logic.ServerUpdate(&currentServerNode, ifaceDelta); err != nil {
  1009. logger.Log(1, "server node:", currentServerNode.ID, "failed update")
  1010. return err
  1011. }
  1012. return nil
  1013. }
  1014. func runForceServerUpdate(node *models.Node, publishPeerUpdateToNode bool) {
  1015. go func() {
  1016. if err := mq.PublishPeerUpdate(node, publishPeerUpdateToNode); err != nil {
  1017. logger.Log(1, "failed a peer update after creation of node", node.Name)
  1018. }
  1019. var currentServerNode, getErr = logic.GetNetworkServerLeader(node.Network)
  1020. if getErr == nil {
  1021. if err := logic.ServerUpdate(&currentServerNode, false); err != nil {
  1022. logger.Log(1, "server node:", currentServerNode.ID, "failed update")
  1023. }
  1024. }
  1025. }()
  1026. }
  1027. func isServer(node *models.Node) bool {
  1028. return node.IsServer == "yes"
  1029. }
  1030. func updateRelay(oldnode, newnode *models.Node) {
  1031. relay := logic.FindRelay(oldnode)
  1032. newrelay := relay
  1033. //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
  1034. if oldnode.Address != newnode.Address {
  1035. for i, ip := range newrelay.RelayAddrs {
  1036. if ip == oldnode.Address {
  1037. newrelay.RelayAddrs = append(newrelay.RelayAddrs[:i], relay.RelayAddrs[i+1:]...)
  1038. newrelay.RelayAddrs = append(newrelay.RelayAddrs, newnode.Address)
  1039. }
  1040. }
  1041. }
  1042. //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
  1043. if oldnode.Address6 != newnode.Address6 {
  1044. for i, ip := range newrelay.RelayAddrs {
  1045. if ip == oldnode.Address {
  1046. newrelay.RelayAddrs = append(newrelay.RelayAddrs[:i], newrelay.RelayAddrs[i+1:]...)
  1047. newrelay.RelayAddrs = append(newrelay.RelayAddrs, newnode.Address6)
  1048. }
  1049. }
  1050. }
  1051. logic.UpdateNode(relay, newrelay)
  1052. }
  1053. func doesUserOwnNode(username, network, nodeID string) bool {
  1054. u, err := logic.GetUser(username)
  1055. if err != nil {
  1056. return false
  1057. }
  1058. if u.IsAdmin {
  1059. return true
  1060. }
  1061. netUser, err := pro.GetNetworkUser(network, promodels.NetworkUserID(u.UserName))
  1062. if err != nil {
  1063. return false
  1064. }
  1065. if netUser.AccessLevel == pro.NET_ADMIN {
  1066. return true
  1067. }
  1068. return logic.StringSliceContains(netUser.Nodes, nodeID)
  1069. }