node.go 32 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046
  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. // Delete Any Existing Client with this ID.
  547. mq.DynSecChan <- mq.DynSecAction{
  548. ActionType: mq.DeleteClient,
  549. Payload: mq.MqDynsecPayload{
  550. Commands: []mq.MqDynSecCmd{
  551. {
  552. Command: mq.DeleteClientCmd,
  553. Username: node.ID,
  554. },
  555. },
  556. },
  557. }
  558. // Create client for this node in Mq
  559. mq.DynSecChan <- mq.DynSecAction{
  560. ActionType: mq.CreateClient,
  561. Payload: mq.MqDynsecPayload{
  562. Commands: []mq.MqDynSecCmd{
  563. {
  564. Command: mq.CreateClientCmd,
  565. Username: node.ID,
  566. Password: node.Password,
  567. Roles: make([]mq.MqDynSecRole, 0),
  568. Groups: make([]mq.MqDynSecGroup, 0),
  569. },
  570. },
  571. },
  572. }
  573. response := models.NodeGet{
  574. Node: node,
  575. Peers: peerUpdate.Peers,
  576. ServerConfig: servercfg.GetServerInfo(),
  577. }
  578. logger.Log(1, r.Header.Get("user"), "created new node", node.Name, "on network", node.Network)
  579. w.WriteHeader(http.StatusOK)
  580. json.NewEncoder(w).Encode(response)
  581. runForceServerUpdate(&node, true)
  582. }
  583. // swagger:route POST /api/nodes/{network}/{nodeid}/approve nodes uncordonNode
  584. //
  585. // Takes a node out of pending state.
  586. //
  587. // Schemes: https
  588. //
  589. // Security:
  590. // oauth
  591. //
  592. // Responses:
  593. // 200: nodeResponse
  594. // Takes node out of pending state
  595. // TODO: May want to use cordon/uncordon terminology instead of "ispending".
  596. func uncordonNode(w http.ResponseWriter, r *http.Request) {
  597. var params = mux.Vars(r)
  598. w.Header().Set("Content-Type", "application/json")
  599. var nodeid = params["nodeid"]
  600. node, err := logic.UncordonNode(nodeid)
  601. if err != nil {
  602. logger.Log(0, r.Header.Get("user"),
  603. fmt.Sprintf("failed to uncordon node [%s]: %v", node.Name, err))
  604. returnErrorResponse(w, r, formatError(err, "internal"))
  605. return
  606. }
  607. logger.Log(1, r.Header.Get("user"), "uncordoned node", node.Name)
  608. w.WriteHeader(http.StatusOK)
  609. json.NewEncoder(w).Encode("SUCCESS")
  610. runUpdates(&node, false)
  611. }
  612. // == EGRESS ==
  613. // swagger:route POST /api/nodes/{network}/{nodeid}/creategateway nodes createEgressGateway
  614. //
  615. // Create an egress gateway.
  616. //
  617. // Schemes: https
  618. //
  619. // Security:
  620. // oauth
  621. //
  622. // Responses:
  623. // 200: nodeResponse
  624. func createEgressGateway(w http.ResponseWriter, r *http.Request) {
  625. var gateway models.EgressGatewayRequest
  626. var params = mux.Vars(r)
  627. w.Header().Set("Content-Type", "application/json")
  628. err := json.NewDecoder(r.Body).Decode(&gateway)
  629. if err != nil {
  630. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  631. returnErrorResponse(w, r, formatError(err, "badrequest"))
  632. return
  633. }
  634. gateway.NetID = params["network"]
  635. gateway.NodeID = params["nodeid"]
  636. node, err := logic.CreateEgressGateway(gateway)
  637. if err != nil {
  638. logger.Log(0, r.Header.Get("user"),
  639. fmt.Sprintf("failed to create egress gateway on node [%s] on network [%s]: %v",
  640. gateway.NodeID, gateway.NetID, err))
  641. returnErrorResponse(w, r, formatError(err, "internal"))
  642. return
  643. }
  644. logger.Log(1, r.Header.Get("user"), "created egress gateway on node", gateway.NodeID, "on network", gateway.NetID)
  645. w.WriteHeader(http.StatusOK)
  646. json.NewEncoder(w).Encode(node)
  647. runUpdates(&node, true)
  648. }
  649. // swagger:route DELETE /api/nodes/{network}/{nodeid}/deletegateway nodes deleteEgressGateway
  650. //
  651. // Delete an egress gateway.
  652. //
  653. // Schemes: https
  654. //
  655. // Security:
  656. // oauth
  657. //
  658. // Responses:
  659. // 200: nodeResponse
  660. func deleteEgressGateway(w http.ResponseWriter, r *http.Request) {
  661. w.Header().Set("Content-Type", "application/json")
  662. var params = mux.Vars(r)
  663. nodeid := params["nodeid"]
  664. netid := params["network"]
  665. node, err := logic.DeleteEgressGateway(netid, nodeid)
  666. if err != nil {
  667. logger.Log(0, r.Header.Get("user"),
  668. fmt.Sprintf("failed to delete egress gateway on node [%s] on network [%s]: %v",
  669. nodeid, netid, err))
  670. returnErrorResponse(w, r, formatError(err, "internal"))
  671. return
  672. }
  673. logger.Log(1, r.Header.Get("user"), "deleted egress gateway on node", nodeid, "on network", netid)
  674. w.WriteHeader(http.StatusOK)
  675. json.NewEncoder(w).Encode(node)
  676. runUpdates(&node, true)
  677. }
  678. // == INGRESS ==
  679. // swagger:route POST /api/nodes/{network}/{nodeid}/createingress nodes createIngressGateway
  680. //
  681. // Create an ingress gateway.
  682. //
  683. // Schemes: https
  684. //
  685. // Security:
  686. // oauth
  687. //
  688. // Responses:
  689. // 200: nodeResponse
  690. func createIngressGateway(w http.ResponseWriter, r *http.Request) {
  691. var params = mux.Vars(r)
  692. w.Header().Set("Content-Type", "application/json")
  693. nodeid := params["nodeid"]
  694. netid := params["network"]
  695. node, err := logic.CreateIngressGateway(netid, nodeid)
  696. if err != nil {
  697. logger.Log(0, r.Header.Get("user"),
  698. fmt.Sprintf("failed to create ingress gateway on node [%s] on network [%s]: %v",
  699. nodeid, netid, err))
  700. returnErrorResponse(w, r, formatError(err, "internal"))
  701. return
  702. }
  703. logger.Log(1, r.Header.Get("user"), "created ingress gateway on node", nodeid, "on network", netid)
  704. w.WriteHeader(http.StatusOK)
  705. json.NewEncoder(w).Encode(node)
  706. runUpdates(&node, true)
  707. }
  708. // swagger:route DELETE /api/nodes/{network}/{nodeid}/deleteingress nodes deleteIngressGateway
  709. //
  710. // Delete an ingress gateway.
  711. //
  712. // Schemes: https
  713. //
  714. // Security:
  715. // oauth
  716. //
  717. // Responses:
  718. // 200: nodeResponse
  719. func deleteIngressGateway(w http.ResponseWriter, r *http.Request) {
  720. w.Header().Set("Content-Type", "application/json")
  721. var params = mux.Vars(r)
  722. nodeid := params["nodeid"]
  723. netid := params["network"]
  724. node, err := logic.DeleteIngressGateway(netid, nodeid)
  725. if err != nil {
  726. logger.Log(0, r.Header.Get("user"),
  727. fmt.Sprintf("failed to delete ingress gateway on node [%s] on network [%s]: %v",
  728. nodeid, netid, err))
  729. returnErrorResponse(w, r, formatError(err, "internal"))
  730. return
  731. }
  732. logger.Log(1, r.Header.Get("user"), "deleted ingress gateway", nodeid)
  733. w.WriteHeader(http.StatusOK)
  734. json.NewEncoder(w).Encode(node)
  735. runUpdates(&node, true)
  736. }
  737. // swagger:route PUT /api/nodes/{network}/{nodeid} nodes updateNode
  738. //
  739. // Update an individual node.
  740. //
  741. // Schemes: https
  742. //
  743. // Security:
  744. // oauth
  745. //
  746. // Responses:
  747. // 200: nodeResponse
  748. func updateNode(w http.ResponseWriter, r *http.Request) {
  749. w.Header().Set("Content-Type", "application/json")
  750. var params = mux.Vars(r)
  751. var node models.Node
  752. //start here
  753. nodeid := params["nodeid"]
  754. node, err := logic.GetNodeByID(nodeid)
  755. if err != nil {
  756. logger.Log(0, r.Header.Get("user"),
  757. fmt.Sprintf("error fetching node [ %s ] info: %v", nodeid, err))
  758. returnErrorResponse(w, r, formatError(err, "internal"))
  759. return
  760. }
  761. var newNode models.Node
  762. // we decode our body request params
  763. err = json.NewDecoder(r.Body).Decode(&newNode)
  764. if err != nil {
  765. logger.Log(0, r.Header.Get("user"), "error decoding request body: ", err.Error())
  766. returnErrorResponse(w, r, formatError(err, "badrequest"))
  767. return
  768. }
  769. relayupdate := false
  770. if node.IsRelay == "yes" && len(newNode.RelayAddrs) > 0 {
  771. if len(newNode.RelayAddrs) != len(node.RelayAddrs) {
  772. relayupdate = true
  773. } else {
  774. for i, addr := range newNode.RelayAddrs {
  775. if addr != node.RelayAddrs[i] {
  776. relayupdate = true
  777. }
  778. }
  779. }
  780. }
  781. relayedUpdate := false
  782. if node.IsRelayed == "yes" && (node.Address != newNode.Address || node.Address6 != newNode.Address6) {
  783. relayedUpdate = true
  784. }
  785. if !servercfg.GetRce() {
  786. newNode.PostDown = node.PostDown
  787. newNode.PostUp = node.PostUp
  788. }
  789. ifaceDelta := logic.IfaceDelta(&node, &newNode)
  790. // for a hub change also need to update the existing hub
  791. if newNode.IsHub == "yes" && node.IsHub != "yes" {
  792. nodeToUpdate, err := logic.UnsetHub(newNode.Network)
  793. if err != nil {
  794. logger.Log(2, "failed to unset hubs", err.Error())
  795. }
  796. if err := mq.NodeUpdate(nodeToUpdate); err != nil {
  797. logger.Log(2, "failed to update hub node", nodeToUpdate.Name, err.Error())
  798. }
  799. if nodeToUpdate.IsServer == "yes" {
  800. // set ifacdelta true to force server to update peeers
  801. if err := logic.ServerUpdate(nodeToUpdate, true); err != nil {
  802. logger.Log(2, "failed to update server node on hub change", err.Error())
  803. }
  804. }
  805. }
  806. err = logic.UpdateNode(&node, &newNode)
  807. if err != nil {
  808. logger.Log(0, r.Header.Get("user"),
  809. fmt.Sprintf("failed to update node info [ %s ] info: %v", nodeid, err))
  810. returnErrorResponse(w, r, formatError(err, "internal"))
  811. return
  812. }
  813. if relayupdate {
  814. updatenodes := logic.UpdateRelay(node.Network, node.RelayAddrs, newNode.RelayAddrs)
  815. if len(updatenodes) > 0 {
  816. for _, relayedNode := range updatenodes {
  817. runUpdates(&relayedNode, false)
  818. }
  819. }
  820. }
  821. if relayedUpdate {
  822. updateRelay(&node, &newNode)
  823. }
  824. if servercfg.IsDNSMode() {
  825. logic.SetDNS()
  826. }
  827. logger.Log(1, r.Header.Get("user"), "updated node", node.ID, "on network", node.Network)
  828. w.WriteHeader(http.StatusOK)
  829. json.NewEncoder(w).Encode(newNode)
  830. runUpdates(&newNode, ifaceDelta)
  831. }
  832. // swagger:route DELETE /api/nodes/{network}/{nodeid} nodes deleteNode
  833. //
  834. // Delete an individual node.
  835. //
  836. // Schemes: https
  837. //
  838. // Security:
  839. // oauth
  840. //
  841. // Responses:
  842. // 200: nodeResponse
  843. func deleteNode(w http.ResponseWriter, r *http.Request) {
  844. // Set header
  845. w.Header().Set("Content-Type", "application/json")
  846. // get params
  847. var params = mux.Vars(r)
  848. var nodeid = params["nodeid"]
  849. var node, err = logic.GetNodeByID(nodeid)
  850. if err != nil {
  851. logger.Log(0, r.Header.Get("user"),
  852. fmt.Sprintf("error fetching node [ %s ] info: %v", nodeid, err))
  853. returnErrorResponse(w, r, formatError(err, "badrequest"))
  854. return
  855. }
  856. if isServer(&node) {
  857. err := fmt.Errorf("cannot delete server node")
  858. logger.Log(0, r.Header.Get("user"),
  859. fmt.Sprintf("failed to delete node [ %s ]: %v", nodeid, err))
  860. returnErrorResponse(w, r, formatError(err, "badrequest"))
  861. return
  862. }
  863. //send update to node to be deleted before deleting on server otherwise message cannot be sent
  864. node.Action = models.NODE_DELETE
  865. err = logic.DeleteNodeByID(&node, false)
  866. if err != nil {
  867. returnErrorResponse(w, r, formatError(err, "internal"))
  868. return
  869. }
  870. mq.DynSecChan <- mq.DynSecAction{
  871. ActionType: mq.DeleteClient,
  872. Payload: mq.MqDynsecPayload{
  873. Commands: []mq.MqDynSecCmd{
  874. {
  875. Command: mq.DeleteClientCmd,
  876. Username: nodeid,
  877. },
  878. },
  879. },
  880. }
  881. returnSuccessResponse(w, r, nodeid+" deleted.")
  882. logger.Log(1, r.Header.Get("user"), "Deleted node", nodeid, "from network", params["network"])
  883. runUpdates(&node, false)
  884. runForceServerUpdate(&node, false)
  885. }
  886. func runUpdates(node *models.Node, ifaceDelta bool) {
  887. go func() { // don't block http response
  888. // publish node update if not server
  889. if err := mq.NodeUpdate(node); err != nil {
  890. logger.Log(1, "error publishing node update to node", node.Name, node.ID, err.Error())
  891. }
  892. if err := runServerUpdate(node, ifaceDelta); err != nil {
  893. logger.Log(1, "error running server update", err.Error())
  894. }
  895. }()
  896. }
  897. // updates local peers for a server on a given node's network
  898. func runServerUpdate(node *models.Node, ifaceDelta bool) error {
  899. if servercfg.IsClientMode() != "on" || !isServer(node) {
  900. return nil
  901. }
  902. currentServerNode, err := logic.GetNetworkServerLocal(node.Network)
  903. if err != nil {
  904. return err
  905. }
  906. if ifaceDelta && logic.IsLeader(&currentServerNode) {
  907. if err := mq.PublishPeerUpdate(&currentServerNode, false); err != nil {
  908. logger.Log(1, "failed to publish peer update "+err.Error())
  909. }
  910. }
  911. if err := logic.ServerUpdate(&currentServerNode, ifaceDelta); err != nil {
  912. logger.Log(1, "server node:", currentServerNode.ID, "failed update")
  913. return err
  914. }
  915. return nil
  916. }
  917. func runForceServerUpdate(node *models.Node, publishPeerUpdateToNode bool) {
  918. go func() {
  919. if err := mq.PublishPeerUpdate(node, publishPeerUpdateToNode); err != nil {
  920. logger.Log(1, "failed a peer update after creation of node", node.Name)
  921. }
  922. var currentServerNode, getErr = logic.GetNetworkServerLeader(node.Network)
  923. if getErr == nil {
  924. if err := logic.ServerUpdate(&currentServerNode, false); err != nil {
  925. logger.Log(1, "server node:", currentServerNode.ID, "failed update")
  926. }
  927. }
  928. }()
  929. }
  930. func isServer(node *models.Node) bool {
  931. return node.IsServer == "yes"
  932. }
  933. func updateRelay(oldnode, newnode *models.Node) {
  934. relay := logic.FindRelay(oldnode)
  935. newrelay := relay
  936. //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
  937. if oldnode.Address != newnode.Address {
  938. for i, ip := range newrelay.RelayAddrs {
  939. if ip == oldnode.Address {
  940. newrelay.RelayAddrs = append(newrelay.RelayAddrs[:i], relay.RelayAddrs[i+1:]...)
  941. newrelay.RelayAddrs = append(newrelay.RelayAddrs, newnode.Address)
  942. }
  943. }
  944. }
  945. //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
  946. if oldnode.Address6 != newnode.Address6 {
  947. for i, ip := range newrelay.RelayAddrs {
  948. if ip == oldnode.Address {
  949. newrelay.RelayAddrs = append(newrelay.RelayAddrs[:i], newrelay.RelayAddrs[i+1:]...)
  950. newrelay.RelayAddrs = append(newrelay.RelayAddrs, newnode.Address6)
  951. }
  952. }
  953. }
  954. logic.UpdateNode(relay, newrelay)
  955. }