|
@@ -8,6 +8,7 @@ import (
|
|
"net/http"
|
|
"net/http"
|
|
"strings"
|
|
"strings"
|
|
|
|
|
|
|
|
+ "github.com/google/uuid"
|
|
"github.com/gorilla/mux"
|
|
"github.com/gorilla/mux"
|
|
"golang.org/x/exp/slog"
|
|
"golang.org/x/exp/slog"
|
|
|
|
|
|
@@ -17,6 +18,7 @@ import (
|
|
"github.com/gravitl/netmaker/logic/acls"
|
|
"github.com/gravitl/netmaker/logic/acls"
|
|
"github.com/gravitl/netmaker/models"
|
|
"github.com/gravitl/netmaker/models"
|
|
"github.com/gravitl/netmaker/mq"
|
|
"github.com/gravitl/netmaker/mq"
|
|
|
|
+ "github.com/gravitl/netmaker/servercfg"
|
|
)
|
|
)
|
|
|
|
|
|
func networkHandlers(r *mux.Router) {
|
|
func networkHandlers(r *mux.Router) {
|
|
@@ -27,6 +29,7 @@ func networkHandlers(r *mux.Router) {
|
|
r.HandleFunc("/api/networks/{networkname}", logic.SecurityCheck(true, http.HandlerFunc(updateNetwork))).Methods(http.MethodPut)
|
|
r.HandleFunc("/api/networks/{networkname}", logic.SecurityCheck(true, http.HandlerFunc(updateNetwork))).Methods(http.MethodPut)
|
|
// ACLs
|
|
// ACLs
|
|
r.HandleFunc("/api/networks/{networkname}/acls", logic.SecurityCheck(true, http.HandlerFunc(updateNetworkACL))).Methods(http.MethodPut)
|
|
r.HandleFunc("/api/networks/{networkname}/acls", logic.SecurityCheck(true, http.HandlerFunc(updateNetworkACL))).Methods(http.MethodPut)
|
|
|
|
+ r.HandleFunc("/api/networks/{networkname}/acls/v2", logic.SecurityCheck(true, http.HandlerFunc(updateNetworkACLv2))).Methods(http.MethodPut)
|
|
r.HandleFunc("/api/networks/{networkname}/acls", logic.SecurityCheck(true, http.HandlerFunc(getNetworkACL))).Methods(http.MethodGet)
|
|
r.HandleFunc("/api/networks/{networkname}/acls", logic.SecurityCheck(true, http.HandlerFunc(getNetworkACL))).Methods(http.MethodGet)
|
|
}
|
|
}
|
|
|
|
|
|
@@ -129,7 +132,7 @@ func updateNetworkACL(w http.ResponseWriter, r *http.Request) {
|
|
// send peer updates
|
|
// send peer updates
|
|
go func() {
|
|
go func() {
|
|
if err = mq.PublishPeerUpdate(false); err != nil {
|
|
if err = mq.PublishPeerUpdate(false); err != nil {
|
|
- logger.Log(0, "failed to publish peer update after ACL update on", netname)
|
|
|
|
|
|
+ logger.Log(0, "failed to publish peer update after ACL update on network:", netname)
|
|
}
|
|
}
|
|
}()
|
|
}()
|
|
|
|
|
|
@@ -137,6 +140,196 @@ func updateNetworkACL(w http.ResponseWriter, r *http.Request) {
|
|
json.NewEncoder(w).Encode(newNetACL)
|
|
json.NewEncoder(w).Encode(newNetACL)
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+// swagger:route PUT /api/networks/{networkname}/acls/v2 networks updateNetworkACL
|
|
|
|
+//
|
|
|
|
+// Update a network ACL (Access Control List).
|
|
|
|
+//
|
|
|
|
+// Schemes: https
|
|
|
|
+//
|
|
|
|
+// Security:
|
|
|
|
+// oauth
|
|
|
|
+//
|
|
|
|
+// Responses:
|
|
|
|
+// 200: aclContainerResponse
|
|
|
|
+func updateNetworkACLv2(w http.ResponseWriter, r *http.Request) {
|
|
|
|
+ w.Header().Set("Content-Type", "application/json")
|
|
|
|
+ var params = mux.Vars(r)
|
|
|
|
+ netname := params["networkname"]
|
|
|
|
+ var networkACLChange acls.ACLContainer
|
|
|
|
+ networkACLChange, err := networkACLChange.Get(acls.ContainerID(netname))
|
|
|
|
+ if err != nil {
|
|
|
|
+ logger.Log(0, r.Header.Get("user"),
|
|
|
|
+ fmt.Sprintf("failed to fetch ACLs for network [%s]: %v", netname, err))
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ err = json.NewDecoder(r.Body).Decode(&networkACLChange)
|
|
|
|
+ if err != nil {
|
|
|
|
+ logger.Log(0, r.Header.Get("user"), "error decoding request body: ",
|
|
|
|
+ err.Error())
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ // clone req body to use as return data successful update
|
|
|
|
+ retData := make(acls.ACLContainer)
|
|
|
|
+ data, err := json.Marshal(networkACLChange)
|
|
|
|
+ if err != nil {
|
|
|
|
+ slog.Error("failed to marshal networkACLChange whiles cloning", "error", err.Error())
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ err = json.Unmarshal(data, &retData)
|
|
|
|
+ if err != nil {
|
|
|
|
+ slog.Error("failed to unmarshal networkACLChange whiles cloning", "error", err.Error())
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ allNodes, err := logic.GetAllNodes()
|
|
|
|
+ if err != nil {
|
|
|
|
+ slog.Error("failed to fetch all nodes", "error", err.Error())
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ networkNodes := make([]models.Node, 0)
|
|
|
|
+ for _, node := range allNodes {
|
|
|
|
+ if node.Network == netname {
|
|
|
|
+ networkNodes = append(networkNodes, node)
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ networkNodesIdMap := make(map[string]models.Node)
|
|
|
|
+ for _, node := range networkNodes {
|
|
|
|
+ networkNodesIdMap[node.ID.String()] = node
|
|
|
|
+ }
|
|
|
|
+ networkClients, err := logic.GetNetworkExtClients(netname)
|
|
|
|
+ if err != nil {
|
|
|
|
+ slog.Error("failed to fetch network clients", "error", err.Error())
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ networkClientsMap := make(map[string]models.ExtClient)
|
|
|
|
+ for _, client := range networkClients {
|
|
|
|
+ networkClientsMap[client.ClientID] = client
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ // keep track of ingress gateways to disconnect from their clients
|
|
|
|
+ // this is required because PublishPeerUpdate only somehow does not stop communication
|
|
|
|
+ // between blocked clients and their ingress
|
|
|
|
+ assocClientsToDisconnectPerHost := make(map[uuid.UUID][]models.ExtClient)
|
|
|
|
+
|
|
|
|
+ // update client acls and then, remove client acls from req data to pass to existing functions
|
|
|
|
+ for id, acl := range networkACLChange {
|
|
|
|
+ // for node acls
|
|
|
|
+ if _, ok := networkNodesIdMap[string(id)]; ok {
|
|
|
|
+ nodeId := string(id)
|
|
|
|
+ // check acl update, then remove client entries
|
|
|
|
+ for id2 := range acl {
|
|
|
|
+ if _, ok := networkNodesIdMap[string(id2)]; !ok {
|
|
|
|
+ // update client acl
|
|
|
|
+ clientId := string(id2)
|
|
|
|
+ if client, ok := networkClientsMap[clientId]; ok {
|
|
|
|
+ if client.DeniedACLs == nil {
|
|
|
|
+ client.DeniedACLs = make(map[string]struct{})
|
|
|
|
+ }
|
|
|
|
+ if acl[acls.AclID(clientId)] == acls.NotAllowed {
|
|
|
|
+ client.DeniedACLs[nodeId] = struct{}{}
|
|
|
|
+ } else {
|
|
|
|
+ delete(client.DeniedACLs, string(nodeId))
|
|
|
|
+ }
|
|
|
|
+ networkClientsMap[clientId] = client
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ } else {
|
|
|
|
+ // for client acls
|
|
|
|
+ clientId := string(id)
|
|
|
|
+ for id2 := range acl {
|
|
|
|
+ if _, ok := networkNodesIdMap[string(id2)]; !ok {
|
|
|
|
+ // update client acl
|
|
|
|
+ clientId2 := string(id2)
|
|
|
|
+ if client, ok := networkClientsMap[clientId]; ok {
|
|
|
|
+ if client.DeniedACLs == nil {
|
|
|
|
+ client.DeniedACLs = make(map[string]struct{})
|
|
|
|
+ }
|
|
|
|
+ {
|
|
|
|
+ // TODO: review this when client-to-client acls are supported
|
|
|
|
+ // if acl[acls.AclID(clientId2)] == acls.NotAllowed {
|
|
|
|
+ // client.DeniedACLs[clientId2] = struct{}{}
|
|
|
|
+ // } else {
|
|
|
|
+ // delete(client.DeniedACLs, clientId2)
|
|
|
|
+ // }
|
|
|
|
+ delete(client.DeniedACLs, clientId2)
|
|
|
|
+ }
|
|
|
|
+ networkClientsMap[clientId] = client
|
|
|
|
+ }
|
|
|
|
+ } else {
|
|
|
|
+ nodeId2 := string(id2)
|
|
|
|
+ if networkClientsMap[clientId].IngressGatewayID == nodeId2 && acl[acls.AclID(nodeId2)] == acls.NotAllowed {
|
|
|
|
+ assocClientsToDisconnectPerHost[networkNodesIdMap[nodeId2].HostID] = append(assocClientsToDisconnectPerHost[networkNodesIdMap[nodeId2].HostID], networkClientsMap[clientId])
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ // update each client in db for pro servers
|
|
|
|
+ if servercfg.IsPro {
|
|
|
|
+ for _, client := range networkClientsMap {
|
|
|
|
+ client := client
|
|
|
|
+ err := logic.DeleteExtClient(client.Network, client.ClientID)
|
|
|
|
+ if err != nil {
|
|
|
|
+ slog.Error("failed to delete client during update", "client", client.ClientID, "error", err.Error())
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ err = logic.SaveExtClient(&client)
|
|
|
|
+ if err != nil {
|
|
|
|
+ slog.Error("failed to save client during update", "client", client.ClientID, "error", err.Error())
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ _, err = networkACLChange.Save(acls.ContainerID(netname))
|
|
|
|
+ if err != nil {
|
|
|
|
+ logger.Log(0, r.Header.Get("user"),
|
|
|
|
+ fmt.Sprintf("failed to update ACLs for network [%s]: %v", netname, err))
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ logger.Log(1, r.Header.Get("user"), "updated ACLs for network", netname)
|
|
|
|
+
|
|
|
|
+ // send peer updates
|
|
|
|
+ go func() {
|
|
|
|
+ if err = mq.PublishPeerUpdate(false); err != nil {
|
|
|
|
+ logger.Log(0, "failed to publish peer update after ACL update on network:", netname)
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ // update ingress gateways of associated clients
|
|
|
|
+ hosts, err := logic.GetAllHosts()
|
|
|
|
+ if err != nil {
|
|
|
|
+ slog.Error("failed to fetch hosts after network ACL update. skipping publish extclients ACL", "network", netname)
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ hostsMap := make(map[uuid.UUID]models.Host)
|
|
|
|
+ for _, host := range hosts {
|
|
|
|
+ hostsMap[host.ID] = host
|
|
|
|
+ }
|
|
|
|
+ for hostId, clients := range assocClientsToDisconnectPerHost {
|
|
|
|
+ if host, ok := hostsMap[hostId]; ok {
|
|
|
|
+ if err = mq.PublishSingleHostPeerUpdate(&host, allNodes, nil, clients, false); err != nil {
|
|
|
|
+ slog.Error("failed to publish peer update to ingress after ACL update on network", "network", netname, "host", hostId)
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }()
|
|
|
|
+
|
|
|
|
+ w.WriteHeader(http.StatusOK)
|
|
|
|
+ json.NewEncoder(w).Encode(networkACLChange)
|
|
|
|
+}
|
|
|
|
+
|
|
// swagger:route GET /api/networks/{networkname}/acls networks getNetworkACL
|
|
// swagger:route GET /api/networks/{networkname}/acls networks getNetworkACL
|
|
//
|
|
//
|
|
// Get a network ACL (Access Control List).
|
|
// Get a network ACL (Access Control List).
|
|
@@ -288,25 +481,30 @@ func createNetwork(w http.ResponseWriter, r *http.Request) {
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
|
|
return
|
|
return
|
|
}
|
|
}
|
|
-
|
|
|
|
- defaultHosts := logic.GetDefaultHosts()
|
|
|
|
- for i := range defaultHosts {
|
|
|
|
- currHost := &defaultHosts[i]
|
|
|
|
- newNode, err := logic.UpdateHostNetwork(currHost, network.NetID, true)
|
|
|
|
- if err != nil {
|
|
|
|
- logger.Log(0, r.Header.Get("user"), "failed to add host to network:", currHost.ID.String(), network.NetID, err.Error())
|
|
|
|
- logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
|
- return
|
|
|
|
- }
|
|
|
|
- logger.Log(1, "added new node", newNode.ID.String(), "to host", currHost.Name)
|
|
|
|
- if err = mq.HostUpdate(&models.HostUpdate{
|
|
|
|
- Action: models.JoinHostToNetwork,
|
|
|
|
- Host: *currHost,
|
|
|
|
- Node: *newNode,
|
|
|
|
- }); err != nil {
|
|
|
|
- logger.Log(0, r.Header.Get("user"), "failed to add host to network:", currHost.ID.String(), network.NetID, err.Error())
|
|
|
|
|
|
+ go func() {
|
|
|
|
+ defaultHosts := logic.GetDefaultHosts()
|
|
|
|
+ for i := range defaultHosts {
|
|
|
|
+ currHost := &defaultHosts[i]
|
|
|
|
+ newNode, err := logic.UpdateHostNetwork(currHost, network.NetID, true)
|
|
|
|
+ if err != nil {
|
|
|
|
+ logger.Log(0, r.Header.Get("user"), "failed to add host to network:", currHost.ID.String(), network.NetID, err.Error())
|
|
|
|
+ logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
|
|
|
+ return
|
|
|
|
+ }
|
|
|
|
+ logger.Log(1, "added new node", newNode.ID.String(), "to host", currHost.Name)
|
|
|
|
+ if err = mq.HostUpdate(&models.HostUpdate{
|
|
|
|
+ Action: models.JoinHostToNetwork,
|
|
|
|
+ Host: *currHost,
|
|
|
|
+ Node: *newNode,
|
|
|
|
+ }); err != nil {
|
|
|
|
+ logger.Log(0, r.Header.Get("user"), "failed to add host to network:", currHost.ID.String(), network.NetID, err.Error())
|
|
|
|
+ }
|
|
|
|
+ // make host failover
|
|
|
|
+ logic.CreateFailOver(*newNode)
|
|
|
|
+ // make host remote access gateway
|
|
|
|
+ logic.CreateIngressGateway(network.NetID, newNode.ID.String(), models.IngressRequest{})
|
|
}
|
|
}
|
|
- }
|
|
|
|
|
|
+ }()
|
|
|
|
|
|
logger.Log(1, r.Header.Get("user"), "created network", network.NetID)
|
|
logger.Log(1, r.Header.Get("user"), "created network", network.NetID)
|
|
w.WriteHeader(http.StatusOK)
|
|
w.WriteHeader(http.StatusOK)
|