nodeHttpController.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798
  1. package controller
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "net/http"
  6. "strings"
  7. "time"
  8. "github.com/gorilla/mux"
  9. "github.com/gravitl/netmaker/database"
  10. "github.com/gravitl/netmaker/functions"
  11. "github.com/gravitl/netmaker/logger"
  12. "github.com/gravitl/netmaker/logic"
  13. "github.com/gravitl/netmaker/models"
  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, "user", http.HandlerFunc(getAllNodes))).Methods("GET")
  19. r.HandleFunc("/api/nodes/{network}", authorize(true, "network", http.HandlerFunc(getNetworkNodes))).Methods("GET")
  20. r.HandleFunc("/api/nodes/{network}/{macaddress}", authorize(true, "node", http.HandlerFunc(getNode))).Methods("GET")
  21. r.HandleFunc("/api/nodes/{network}/{macaddress}", authorize(true, "node", http.HandlerFunc(updateNode))).Methods("PUT")
  22. r.HandleFunc("/api/nodes/{network}/{macaddress}", authorize(true, "node", http.HandlerFunc(deleteNode))).Methods("DELETE")
  23. r.HandleFunc("/api/nodes/{network}/{macaddress}/createrelay", authorize(true, "user", http.HandlerFunc(createRelay))).Methods("POST")
  24. r.HandleFunc("/api/nodes/{network}/{macaddress}/deleterelay", authorize(true, "user", http.HandlerFunc(deleteRelay))).Methods("DELETE")
  25. r.HandleFunc("/api/nodes/{network}/{macaddress}/creategateway", authorize(true, "user", http.HandlerFunc(createEgressGateway))).Methods("POST")
  26. r.HandleFunc("/api/nodes/{network}/{macaddress}/deletegateway", authorize(true, "user", http.HandlerFunc(deleteEgressGateway))).Methods("DELETE")
  27. r.HandleFunc("/api/nodes/{network}/{macaddress}/createingress", securityCheck(false, http.HandlerFunc(createIngressGateway))).Methods("POST")
  28. r.HandleFunc("/api/nodes/{network}/{macaddress}/deleteingress", securityCheck(false, http.HandlerFunc(deleteIngressGateway))).Methods("DELETE")
  29. r.HandleFunc("/api/nodes/{network}/{macaddress}/approve", authorize(true, "user", http.HandlerFunc(uncordonNode))).Methods("POST")
  30. r.HandleFunc("/api/nodes/{network}", createNode).Methods("POST")
  31. r.HandleFunc("/api/nodes/adm/{network}/lastmodified", authorize(true, "network", http.HandlerFunc(getLastModified))).Methods("GET")
  32. r.HandleFunc("/api/nodes/adm/{network}/authenticate", authenticate).Methods("POST")
  33. }
  34. //Node authenticates using its password and retrieves a JWT for authorization.
  35. func authenticate(response http.ResponseWriter, request *http.Request) {
  36. var params = mux.Vars(request)
  37. networkname := params["network"]
  38. //Auth request consists of Mac Address and Password (from node that is authorizing
  39. //in case of Master, auth is ignored and mac is set to "mastermac"
  40. var authRequest models.AuthParams
  41. var result models.Node
  42. var errorResponse = models.ErrorResponse{
  43. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  44. }
  45. //Get password fnd mac rom request
  46. decoder := json.NewDecoder(request.Body)
  47. decoderErr := decoder.Decode(&authRequest)
  48. defer request.Body.Close()
  49. if decoderErr != nil {
  50. errorResponse.Code = http.StatusBadRequest
  51. errorResponse.Message = decoderErr.Error()
  52. returnErrorResponse(response, request, errorResponse)
  53. return
  54. } else {
  55. errorResponse.Code = http.StatusBadRequest
  56. if authRequest.MacAddress == "" {
  57. errorResponse.Message = "W1R3: MacAddress can't be empty"
  58. returnErrorResponse(response, request, errorResponse)
  59. return
  60. } else if authRequest.Password == "" {
  61. errorResponse.Message = "W1R3: Password can't be empty"
  62. returnErrorResponse(response, request, errorResponse)
  63. return
  64. } else {
  65. //Search DB for node with Mac Address. Ignore pending nodes (they should not be able to authenticate with API until approved).
  66. collection, err := database.FetchRecords(database.NODES_TABLE_NAME)
  67. if err != nil {
  68. errorResponse.Code = http.StatusBadRequest
  69. errorResponse.Message = err.Error()
  70. returnErrorResponse(response, request, errorResponse)
  71. return
  72. }
  73. for _, value := range collection {
  74. if err := json.Unmarshal([]byte(value), &result); err != nil {
  75. continue
  76. }
  77. if result.MacAddress == authRequest.MacAddress && result.IsPending != "yes" && result.Network == networkname {
  78. break
  79. }
  80. }
  81. if err != nil {
  82. errorResponse.Code = http.StatusBadRequest
  83. errorResponse.Message = err.Error()
  84. returnErrorResponse(response, request, errorResponse)
  85. return
  86. }
  87. //compare password from request to stored password in database
  88. //might be able to have a common hash (certificates?) and compare those so that a password isn't passed in in plain text...
  89. //TODO: Consider a way of hashing the password client side before sending, or using certificates
  90. err = bcrypt.CompareHashAndPassword([]byte(result.Password), []byte(authRequest.Password))
  91. if err != nil {
  92. errorResponse.Code = http.StatusBadRequest
  93. errorResponse.Message = err.Error()
  94. returnErrorResponse(response, request, errorResponse)
  95. return
  96. } else {
  97. //Create a new JWT for the node
  98. tokenString, _ := logic.CreateJWT(authRequest.MacAddress, result.Network)
  99. if tokenString == "" {
  100. errorResponse.Code = http.StatusBadRequest
  101. errorResponse.Message = "Could not create Token"
  102. returnErrorResponse(response, request, errorResponse)
  103. return
  104. }
  105. var successResponse = models.SuccessResponse{
  106. Code: http.StatusOK,
  107. Message: "W1R3: Device " + authRequest.MacAddress + " Authorized",
  108. Response: models.SuccessfulLoginResponse{
  109. AuthToken: tokenString,
  110. MacAddress: authRequest.MacAddress,
  111. },
  112. }
  113. //Send back the JWT
  114. successJSONResponse, jsonError := json.Marshal(successResponse)
  115. if jsonError != nil {
  116. errorResponse.Code = http.StatusBadRequest
  117. errorResponse.Message = err.Error()
  118. returnErrorResponse(response, request, errorResponse)
  119. return
  120. }
  121. response.WriteHeader(http.StatusOK)
  122. response.Header().Set("Content-Type", "application/json")
  123. response.Write(successJSONResponse)
  124. }
  125. }
  126. }
  127. }
  128. //The middleware for most requests to the API
  129. //They all pass through here first
  130. //This will validate the JWT (or check for master token)
  131. //This will also check against the authNetwork and make sure the node should be accessing that endpoint,
  132. //even if it's technically ok
  133. //This is kind of a poor man's RBAC. There's probably a better/smarter way.
  134. //TODO: Consider better RBAC implementations
  135. func authorize(networkCheck bool, authNetwork string, next http.Handler) http.HandlerFunc {
  136. return func(w http.ResponseWriter, r *http.Request) {
  137. var errorResponse = models.ErrorResponse{
  138. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  139. }
  140. var params = mux.Vars(r)
  141. networkexists, _ := functions.NetworkExists(params["network"])
  142. //check that the request is for a valid network
  143. //if (networkCheck && !networkexists) || err != nil {
  144. if networkCheck && !networkexists {
  145. errorResponse = models.ErrorResponse{
  146. Code: http.StatusNotFound, Message: "W1R3: This network does not exist. ",
  147. }
  148. returnErrorResponse(w, r, errorResponse)
  149. return
  150. } else {
  151. w.Header().Set("Content-Type", "application/json")
  152. //get the auth token
  153. bearerToken := r.Header.Get("Authorization")
  154. var tokenSplit = strings.Split(bearerToken, " ")
  155. //I put this in in case the user doesn't put in a token at all (in which case it's empty)
  156. //There's probably a smarter way of handling this.
  157. var authToken = "928rt238tghgwe@TY@$Y@#WQAEGB2FC#@HG#@$Hddd"
  158. if len(tokenSplit) > 1 {
  159. authToken = tokenSplit[1]
  160. } else {
  161. errorResponse = models.ErrorResponse{
  162. Code: http.StatusUnauthorized, Message: "W1R3: Missing Auth Token.",
  163. }
  164. returnErrorResponse(w, r, errorResponse)
  165. return
  166. }
  167. //This checks if
  168. //A: the token is the master password
  169. //B: the token corresponds to a mac address, and if so, which one
  170. //TODO: There's probably a better way of dealing with the "master token"/master password. Plz Help.
  171. var isAuthorized = false
  172. var macaddress = ""
  173. username, networks, isadmin, errN := logic.VerifyUserToken(authToken)
  174. isnetadmin := isadmin
  175. if errN == nil && isadmin {
  176. macaddress = "mastermac"
  177. isAuthorized = true
  178. r.Header.Set("ismasterkey", "yes")
  179. }
  180. if !isadmin && params["network"] != "" {
  181. if functions.SliceContains(networks, params["network"]) {
  182. isnetadmin = true
  183. }
  184. }
  185. //The mastermac (login with masterkey from config) can do everything!! May be dangerous.
  186. if macaddress == "mastermac" {
  187. isAuthorized = true
  188. r.Header.Set("ismasterkey", "yes")
  189. //for everyone else, there's poor man's RBAC. The "cases" are defined in the routes in the handlers
  190. //So each route defines which access network should be allowed to access it
  191. } else {
  192. switch authNetwork {
  193. case "all":
  194. isAuthorized = true
  195. case "nodes":
  196. isAuthorized = (macaddress != "") || isnetadmin
  197. case "network":
  198. if isnetadmin {
  199. isAuthorized = true
  200. } else {
  201. node, err := logic.GetNodeByMacAddress(params["network"], macaddress)
  202. if err != nil {
  203. errorResponse = models.ErrorResponse{
  204. Code: http.StatusUnauthorized, Message: "W1R3: Missing Auth Token.",
  205. }
  206. returnErrorResponse(w, r, errorResponse)
  207. return
  208. }
  209. isAuthorized = (node.Network == params["network"])
  210. }
  211. case "node":
  212. if isnetadmin {
  213. isAuthorized = true
  214. } else {
  215. isAuthorized = (macaddress == params["macaddress"])
  216. }
  217. case "user":
  218. isAuthorized = true
  219. default:
  220. isAuthorized = false
  221. }
  222. }
  223. if !isAuthorized {
  224. errorResponse = models.ErrorResponse{
  225. Code: http.StatusUnauthorized, Message: "W1R3: You are unauthorized to access this endpoint.",
  226. }
  227. returnErrorResponse(w, r, errorResponse)
  228. return
  229. } else {
  230. //If authorized, this function passes along it's request and output to the appropriate route function.
  231. if username == "" {
  232. username = "(user not found)"
  233. }
  234. r.Header.Set("user", username)
  235. next.ServeHTTP(w, r)
  236. }
  237. }
  238. }
  239. }
  240. //Gets all nodes associated with network, including pending nodes
  241. func getNetworkNodes(w http.ResponseWriter, r *http.Request) {
  242. w.Header().Set("Content-Type", "application/json")
  243. var nodes []models.Node
  244. var params = mux.Vars(r)
  245. networkName := params["network"]
  246. nodes, err := logic.GetNetworkNodes(networkName)
  247. if err != nil {
  248. returnErrorResponse(w, r, formatError(err, "internal"))
  249. return
  250. }
  251. //Returns all the nodes in JSON format
  252. logger.Log(2, r.Header.Get("user"), "fetched nodes on network", networkName)
  253. w.WriteHeader(http.StatusOK)
  254. json.NewEncoder(w).Encode(nodes)
  255. }
  256. //A separate function to get all nodes, not just nodes for a particular network.
  257. //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
  258. func getAllNodes(w http.ResponseWriter, r *http.Request) {
  259. w.Header().Set("Content-Type", "application/json")
  260. user, err := logic.GetUser(r.Header.Get("user"))
  261. if err != nil && r.Header.Get("ismasterkey") != "yes" {
  262. returnErrorResponse(w, r, formatError(err, "internal"))
  263. return
  264. }
  265. var nodes []models.Node
  266. if user.IsAdmin || r.Header.Get("ismasterkey") == "yes" {
  267. nodes, err = logic.GetAllNodes()
  268. if err != nil {
  269. returnErrorResponse(w, r, formatError(err, "internal"))
  270. return
  271. }
  272. } else {
  273. nodes, err = getUsersNodes(user)
  274. if err != nil {
  275. returnErrorResponse(w, r, formatError(err, "internal"))
  276. return
  277. }
  278. }
  279. //Return all the nodes in JSON format
  280. logger.Log(2, r.Header.Get("user"), "fetched nodes")
  281. w.WriteHeader(http.StatusOK)
  282. json.NewEncoder(w).Encode(nodes)
  283. }
  284. func getUsersNodes(user models.User) ([]models.Node, error) {
  285. var nodes []models.Node
  286. var err error
  287. for _, networkName := range user.Networks {
  288. tmpNodes, err := logic.GetNetworkNodes(networkName)
  289. if err != nil {
  290. continue
  291. }
  292. nodes = append(nodes, tmpNodes...)
  293. }
  294. return nodes, err
  295. }
  296. //Get an individual node. Nothin fancy here folks.
  297. func getNode(w http.ResponseWriter, r *http.Request) {
  298. // set header.
  299. w.Header().Set("Content-Type", "application/json")
  300. var params = mux.Vars(r)
  301. node, err := GetNode(params["macaddress"], params["network"])
  302. if err != nil {
  303. returnErrorResponse(w, r, formatError(err, "internal"))
  304. return
  305. }
  306. logger.Log(2, r.Header.Get("user"), "fetched node", params["macaddress"])
  307. w.WriteHeader(http.StatusOK)
  308. json.NewEncoder(w).Encode(node)
  309. }
  310. //Get the time that a network of nodes was last modified.
  311. //TODO: This needs to be refactored
  312. //Potential way to do this: On UpdateNode, set a new field for "LastModified"
  313. //If we go with the existing way, we need to at least set network.NodesLastModified on UpdateNode
  314. func getLastModified(w http.ResponseWriter, r *http.Request) {
  315. // set header.
  316. w.Header().Set("Content-Type", "application/json")
  317. var params = mux.Vars(r)
  318. network, err := GetNetwork(params["network"])
  319. if err != nil {
  320. returnErrorResponse(w, r, formatError(err, "internal"))
  321. return
  322. }
  323. logger.Log(2, r.Header.Get("user"), "called last modified")
  324. w.WriteHeader(http.StatusOK)
  325. json.NewEncoder(w).Encode(network.NodesLastModified)
  326. }
  327. func createNode(w http.ResponseWriter, r *http.Request) {
  328. w.Header().Set("Content-Type", "application/json")
  329. var params = mux.Vars(r)
  330. var errorResponse = models.ErrorResponse{
  331. Code: http.StatusInternalServerError, Message: "W1R3: It's not you it's me.",
  332. }
  333. networkName := params["network"]
  334. networkexists, err := functions.NetworkExists(networkName)
  335. if err != nil {
  336. returnErrorResponse(w, r, formatError(err, "internal"))
  337. return
  338. } else if !networkexists {
  339. errorResponse = models.ErrorResponse{
  340. Code: http.StatusNotFound, Message: "W1R3: Network does not exist! ",
  341. }
  342. returnErrorResponse(w, r, errorResponse)
  343. return
  344. }
  345. var node models.Node
  346. //get node from body of request
  347. err = json.NewDecoder(r.Body).Decode(&node)
  348. if err != nil {
  349. returnErrorResponse(w, r, formatError(err, "internal"))
  350. return
  351. }
  352. node.Network = networkName
  353. network, err := logic.GetNetworkByNode(&node)
  354. if err != nil {
  355. returnErrorResponse(w, r, formatError(err, "internal"))
  356. return
  357. }
  358. //Check to see if key is valid
  359. //TODO: Triple inefficient!!! This is the third call to the DB we make for networks
  360. validKey := logic.IsKeyValid(networkName, node.AccessKey)
  361. if !validKey {
  362. //Check to see if network will allow manual sign up
  363. //may want to switch this up with the valid key check and avoid a DB call that way.
  364. if network.AllowManualSignUp == "yes" {
  365. node.IsPending = "yes"
  366. } else {
  367. errorResponse = models.ErrorResponse{
  368. Code: http.StatusUnauthorized, Message: "W1R3: Key invalid, or none provided.",
  369. }
  370. returnErrorResponse(w, r, errorResponse)
  371. return
  372. }
  373. }
  374. node, err = logic.CreateNode(node, networkName)
  375. if err != nil {
  376. returnErrorResponse(w, r, formatError(err, "internal"))
  377. return
  378. }
  379. logger.Log(1, r.Header.Get("user"), "created new node", node.Name, "on network", node.Network)
  380. w.WriteHeader(http.StatusOK)
  381. json.NewEncoder(w).Encode(node)
  382. }
  383. //Takes node out of pending state
  384. //TODO: May want to use cordon/uncordon terminology instead of "ispending".
  385. func uncordonNode(w http.ResponseWriter, r *http.Request) {
  386. var params = mux.Vars(r)
  387. w.Header().Set("Content-Type", "application/json")
  388. node, err := UncordonNode(params["network"], params["macaddress"])
  389. if err != nil {
  390. returnErrorResponse(w, r, formatError(err, "internal"))
  391. return
  392. }
  393. logger.Log(1, r.Header.Get("user"), "uncordoned node", node.Name)
  394. w.WriteHeader(http.StatusOK)
  395. json.NewEncoder(w).Encode("SUCCESS")
  396. }
  397. // UncordonNode - approves a node to join a network
  398. func UncordonNode(network, macaddress string) (models.Node, error) {
  399. node, err := logic.GetNodeByMacAddress(network, macaddress)
  400. if err != nil {
  401. return models.Node{}, err
  402. }
  403. node.SetLastModified()
  404. node.IsPending = "no"
  405. node.PullChanges = "yes"
  406. data, err := json.Marshal(&node)
  407. if err != nil {
  408. return node, err
  409. }
  410. key, err := logic.GetRecordKey(node.MacAddress, node.Network)
  411. if err != nil {
  412. return node, err
  413. }
  414. err = database.Insert(key, string(data), database.NODES_TABLE_NAME)
  415. return node, err
  416. }
  417. func createEgressGateway(w http.ResponseWriter, r *http.Request) {
  418. var gateway models.EgressGatewayRequest
  419. var params = mux.Vars(r)
  420. w.Header().Set("Content-Type", "application/json")
  421. err := json.NewDecoder(r.Body).Decode(&gateway)
  422. if err != nil {
  423. returnErrorResponse(w, r, formatError(err, "internal"))
  424. return
  425. }
  426. gateway.NetID = params["network"]
  427. gateway.NodeID = params["macaddress"]
  428. node, err := CreateEgressGateway(gateway)
  429. if err != nil {
  430. returnErrorResponse(w, r, formatError(err, "internal"))
  431. return
  432. }
  433. logger.Log(1, r.Header.Get("user"), "created egress gateway on node", gateway.NodeID, "on network", gateway.NetID)
  434. w.WriteHeader(http.StatusOK)
  435. json.NewEncoder(w).Encode(node)
  436. }
  437. // CreateEgressGateway - creates an egress gateway
  438. func CreateEgressGateway(gateway models.EgressGatewayRequest) (models.Node, error) {
  439. node, err := logic.GetNodeByMacAddress(gateway.NetID, gateway.NodeID)
  440. if node.OS == "windows" || node.OS == "macos" { // add in darwin later
  441. return models.Node{}, errors.New(node.OS + " is unsupported for egress gateways")
  442. }
  443. if err != nil {
  444. return models.Node{}, err
  445. }
  446. err = ValidateEgressGateway(gateway)
  447. if err != nil {
  448. return models.Node{}, err
  449. }
  450. node.IsEgressGateway = "yes"
  451. node.EgressGatewayRanges = gateway.Ranges
  452. postUpCmd := "iptables -A FORWARD -i " + node.Interface + " -j ACCEPT; iptables -t nat -A POSTROUTING -o " + gateway.Interface + " -j MASQUERADE"
  453. postDownCmd := "iptables -D FORWARD -i " + node.Interface + " -j ACCEPT; iptables -t nat -D POSTROUTING -o " + gateway.Interface + " -j MASQUERADE"
  454. if gateway.PostUp != "" {
  455. postUpCmd = gateway.PostUp
  456. }
  457. if gateway.PostDown != "" {
  458. postDownCmd = gateway.PostDown
  459. }
  460. if node.PostUp != "" {
  461. if !strings.Contains(node.PostUp, postUpCmd) {
  462. postUpCmd = node.PostUp + "; " + postUpCmd
  463. }
  464. }
  465. if node.PostDown != "" {
  466. if !strings.Contains(node.PostDown, postDownCmd) {
  467. postDownCmd = node.PostDown + "; " + postDownCmd
  468. }
  469. }
  470. key, err := logic.GetRecordKey(gateway.NodeID, gateway.NetID)
  471. if err != nil {
  472. return node, err
  473. }
  474. node.PostUp = postUpCmd
  475. node.PostDown = postDownCmd
  476. node.SetLastModified()
  477. node.PullChanges = "yes"
  478. nodeData, err := json.Marshal(&node)
  479. if err != nil {
  480. return node, err
  481. }
  482. if err = database.Insert(key, string(nodeData), database.NODES_TABLE_NAME); err != nil {
  483. return models.Node{}, err
  484. }
  485. if err = functions.NetworkNodesUpdatePullChanges(node.Network); err != nil {
  486. return models.Node{}, err
  487. }
  488. return node, nil
  489. }
  490. func ValidateEgressGateway(gateway models.EgressGatewayRequest) error {
  491. var err error
  492. //isIp := functions.IsIpCIDR(gateway.RangeString)
  493. empty := len(gateway.Ranges) == 0
  494. if empty {
  495. err = errors.New("IP Ranges Cannot Be Empty")
  496. }
  497. empty = gateway.Interface == ""
  498. if empty {
  499. err = errors.New("Interface cannot be empty")
  500. }
  501. return err
  502. }
  503. func deleteEgressGateway(w http.ResponseWriter, r *http.Request) {
  504. w.Header().Set("Content-Type", "application/json")
  505. var params = mux.Vars(r)
  506. nodeMac := params["macaddress"]
  507. netid := params["network"]
  508. node, err := DeleteEgressGateway(netid, nodeMac)
  509. if err != nil {
  510. returnErrorResponse(w, r, formatError(err, "internal"))
  511. return
  512. }
  513. logger.Log(1, r.Header.Get("user"), "deleted egress gateway", nodeMac, "on network", netid)
  514. w.WriteHeader(http.StatusOK)
  515. json.NewEncoder(w).Encode(node)
  516. }
  517. // DeleteEgressGateway - deletes egress from node
  518. func DeleteEgressGateway(network, macaddress string) (models.Node, error) {
  519. node, err := logic.GetNodeByMacAddress(network, macaddress)
  520. if err != nil {
  521. return models.Node{}, err
  522. }
  523. node.IsEgressGateway = "no"
  524. node.EgressGatewayRanges = []string{}
  525. node.PostUp = ""
  526. node.PostDown = ""
  527. if node.IsIngressGateway == "yes" { // check if node is still an ingress gateway before completely deleting postdown/up rules
  528. node.PostUp = "iptables -A FORWARD -i " + node.Interface + " -j ACCEPT; iptables -t nat -A POSTROUTING -o " + node.Interface + " -j MASQUERADE"
  529. node.PostDown = "iptables -D FORWARD -i " + node.Interface + " -j ACCEPT; iptables -t nat -D POSTROUTING -o " + node.Interface + " -j MASQUERADE"
  530. }
  531. node.SetLastModified()
  532. node.PullChanges = "yes"
  533. key, err := logic.GetRecordKey(node.MacAddress, node.Network)
  534. if err != nil {
  535. return models.Node{}, err
  536. }
  537. data, err := json.Marshal(&node)
  538. if err != nil {
  539. return models.Node{}, err
  540. }
  541. if err = database.Insert(key, string(data), database.NODES_TABLE_NAME); err != nil {
  542. return models.Node{}, err
  543. }
  544. if err = functions.NetworkNodesUpdatePullChanges(network); err != nil {
  545. return models.Node{}, err
  546. }
  547. return node, nil
  548. }
  549. // == INGRESS ==
  550. func createIngressGateway(w http.ResponseWriter, r *http.Request) {
  551. var params = mux.Vars(r)
  552. w.Header().Set("Content-Type", "application/json")
  553. nodeMac := params["macaddress"]
  554. netid := params["network"]
  555. node, err := CreateIngressGateway(netid, nodeMac)
  556. if err != nil {
  557. returnErrorResponse(w, r, formatError(err, "internal"))
  558. return
  559. }
  560. logger.Log(1, r.Header.Get("user"), "created ingress gateway on node", nodeMac, "on network", netid)
  561. w.WriteHeader(http.StatusOK)
  562. json.NewEncoder(w).Encode(node)
  563. }
  564. // CreateIngressGateway - creates an ingress gateway
  565. func CreateIngressGateway(netid string, macaddress string) (models.Node, error) {
  566. node, err := logic.GetNodeByMacAddress(netid, macaddress)
  567. if node.OS == "windows" || node.OS == "macos" { // add in darwin later
  568. return models.Node{}, errors.New(node.OS + " is unsupported for ingress gateways")
  569. }
  570. if err != nil {
  571. return models.Node{}, err
  572. }
  573. network, err := logic.GetParentNetwork(netid)
  574. if err != nil {
  575. return models.Node{}, err
  576. }
  577. node.IsIngressGateway = "yes"
  578. node.IngressGatewayRange = network.AddressRange
  579. postUpCmd := "iptables -A FORWARD -i " + node.Interface + " -j ACCEPT; iptables -t nat -A POSTROUTING -o " + node.Interface + " -j MASQUERADE"
  580. postDownCmd := "iptables -D FORWARD -i " + node.Interface + " -j ACCEPT; iptables -t nat -D POSTROUTING -o " + node.Interface + " -j MASQUERADE"
  581. if node.PostUp != "" {
  582. if !strings.Contains(node.PostUp, postUpCmd) {
  583. postUpCmd = node.PostUp + "; " + postUpCmd
  584. }
  585. }
  586. if node.PostDown != "" {
  587. if !strings.Contains(node.PostDown, postDownCmd) {
  588. postDownCmd = node.PostDown + "; " + postDownCmd
  589. }
  590. }
  591. node.SetLastModified()
  592. node.PostUp = postUpCmd
  593. node.PostDown = postDownCmd
  594. node.PullChanges = "yes"
  595. node.UDPHolePunch = "no"
  596. key, err := logic.GetRecordKey(node.MacAddress, node.Network)
  597. if err != nil {
  598. return models.Node{}, err
  599. }
  600. data, err := json.Marshal(&node)
  601. if err != nil {
  602. return models.Node{}, err
  603. }
  604. err = database.Insert(key, string(data), database.NODES_TABLE_NAME)
  605. if err != nil {
  606. return models.Node{}, err
  607. }
  608. err = logic.SetNetworkNodesLastModified(netid)
  609. return node, err
  610. }
  611. func deleteIngressGateway(w http.ResponseWriter, r *http.Request) {
  612. w.Header().Set("Content-Type", "application/json")
  613. var params = mux.Vars(r)
  614. nodeMac := params["macaddress"]
  615. node, err := DeleteIngressGateway(params["network"], nodeMac)
  616. if err != nil {
  617. returnErrorResponse(w, r, formatError(err, "internal"))
  618. return
  619. }
  620. logger.Log(1, r.Header.Get("user"), "deleted ingress gateway", nodeMac)
  621. w.WriteHeader(http.StatusOK)
  622. json.NewEncoder(w).Encode(node)
  623. }
  624. // DeleteIngressGateway - deletes an ingress gateway
  625. func DeleteIngressGateway(networkName string, macaddress string) (models.Node, error) {
  626. node, err := logic.GetNodeByMacAddress(networkName, macaddress)
  627. if err != nil {
  628. return models.Node{}, err
  629. }
  630. network, err := logic.GetParentNetwork(networkName)
  631. if err != nil {
  632. return models.Node{}, err
  633. }
  634. // delete ext clients belonging to ingress gateway
  635. if err = DeleteGatewayExtClients(macaddress, networkName); err != nil {
  636. return models.Node{}, err
  637. }
  638. node.UDPHolePunch = network.DefaultUDPHolePunch
  639. node.LastModified = time.Now().Unix()
  640. node.IsIngressGateway = "no"
  641. node.IngressGatewayRange = ""
  642. node.PullChanges = "yes"
  643. key, err := logic.GetRecordKey(node.MacAddress, node.Network)
  644. if err != nil {
  645. return models.Node{}, err
  646. }
  647. data, err := json.Marshal(&node)
  648. if err != nil {
  649. return models.Node{}, err
  650. }
  651. err = database.Insert(key, string(data), database.NODES_TABLE_NAME)
  652. if err != nil {
  653. return models.Node{}, err
  654. }
  655. err = logic.SetNetworkNodesLastModified(networkName)
  656. return node, err
  657. }
  658. func updateNode(w http.ResponseWriter, r *http.Request) {
  659. w.Header().Set("Content-Type", "application/json")
  660. var params = mux.Vars(r)
  661. var node models.Node
  662. //start here
  663. node, err := logic.GetNodeByMacAddress(params["network"], params["macaddress"])
  664. if err != nil {
  665. returnErrorResponse(w, r, formatError(err, "internal"))
  666. return
  667. }
  668. var newNode models.Node
  669. // we decode our body request params
  670. err = json.NewDecoder(r.Body).Decode(&newNode)
  671. if err != nil {
  672. returnErrorResponse(w, r, formatError(err, "badrequest"))
  673. return
  674. }
  675. newNode.PullChanges = "yes"
  676. relayupdate := false
  677. if node.IsRelay == "yes" && len(newNode.RelayAddrs) > 0 {
  678. if len(newNode.RelayAddrs) != len(node.RelayAddrs) {
  679. relayupdate = true
  680. } else {
  681. for i, addr := range newNode.RelayAddrs {
  682. if addr != node.RelayAddrs[i] {
  683. relayupdate = true
  684. }
  685. }
  686. }
  687. }
  688. err = logic.UpdateNode(&node, &newNode)
  689. if err != nil {
  690. returnErrorResponse(w, r, formatError(err, "internal"))
  691. return
  692. }
  693. if relayupdate {
  694. UpdateRelay(node.Network, node.RelayAddrs, newNode.RelayAddrs)
  695. if err = functions.NetworkNodesUpdatePullChanges(node.Network); err != nil {
  696. logger.Log(1, "error setting relay updates:", err.Error())
  697. }
  698. }
  699. if servercfg.IsDNSMode() {
  700. err = logic.SetDNS()
  701. }
  702. if err != nil {
  703. returnErrorResponse(w, r, formatError(err, "internal"))
  704. return
  705. }
  706. logger.Log(1, r.Header.Get("user"), "updated node", node.MacAddress, "on network", node.Network)
  707. w.WriteHeader(http.StatusOK)
  708. json.NewEncoder(w).Encode(newNode)
  709. }
  710. //Delete a node
  711. //Pretty straightforward
  712. func deleteNode(w http.ResponseWriter, r *http.Request) {
  713. // Set header
  714. w.Header().Set("Content-Type", "application/json")
  715. // get params
  716. var params = mux.Vars(r)
  717. err := DeleteNode(params["macaddress"]+"###"+params["network"], false)
  718. if err != nil {
  719. returnErrorResponse(w, r, formatError(err, "internal"))
  720. return
  721. }
  722. logger.Log(1, r.Header.Get("user"), "Deleted node", params["macaddress"], "from network", params["network"])
  723. returnSuccessResponse(w, r, params["macaddress"]+" deleted.")
  724. }