-
- - Drago -
-
-
-
-
-
--====|===================> -___ ____ ____ ____ ____ -| \ |__/ |__| | __ | | -|__/ | \ | | |__] |__| - -<===================|==== -- -
If you're seeing this message, it means that Drago's web UI was not built properly.
-In order to build it, run go generate from the project root directory.
- - -` diff --git a/air.conf b/air.conf deleted file mode 100644 index 5c1003a..0000000 --- a/air.conf +++ /dev/null @@ -1,47 +0,0 @@ -# Config file for [Air](https://github.com/cosmtrek/air) in TOML format - -# Working directory -# . or absolute path, please note that the directories following must be under root. -root = "." -tmp_dir = "tmp" - -[build] -# Just plain old shell command. You could use `make` as well. -cmd = "go build -o ./tmp/air/drago ." -# Binary file yields from `cmd`. -bin = "tmp/air/drago agent" -# Customize binary. -full_bin = "APP_ENV=dev APP_USER=air ./tmp/air/drago agent --dev" -# Watch these filename extensions. -include_ext = ["go", "tpl", "tmpl", "html"] -# Ignore these filename extensions or directories. -exclude_dir = ["assets", "tmp", "vendor", "ui", "data"] -# Watch these directories if you specified. -include_dir = [] -# Exclude files. -exclude_file = [] -# This log file places in your tmp_dir. -log = "./tmp/air.log" -# It's not necessary to trigger build each time file changes if it's too frequent. -delay = 500 # ms -# Stop running old binary when build errors occur. -stop_on_error = true -# Send Interrupt signal before killing process (windows does not support this feature) -send_interrupt = false -# Delay after sending Interrupt signal -kill_delay = 500 # ms - -[log] -# Show log time -time = false - -[color] -# Customize each part's color. If no color found, use the raw app log. -main = "magenta" -watcher = "cyan" -build = "yellow" -runner = "green" - -[misc] -# Delete tmp directory on exit -clean_on_exit = true \ No newline at end of file diff --git a/air.sh b/air.sh deleted file mode 100755 index 1b7017e..0000000 --- a/air.sh +++ /dev/null @@ -1 +0,0 @@ -go run github.com/cosmtrek/air -c "./air.conf" \ No newline at end of file diff --git a/api/README.md b/api/README.md deleted file mode 100644 index bf42483..0000000 --- a/api/README.md +++ /dev/null @@ -1,49 +0,0 @@ -## Drago API Client - -This directory contains the `api` package which aims at providing programmatic access to Drago's HTTP API. - -### Documentation - -... - -### Usage - -```go -package main - -import "github.com/seashell/drago/api" - -func main() { - // Get a new client - client, err := api.NewClient(api.DefaultConfig()) - if err != nil { - panic(err) - } - - // Get a handle to the networks API - networks := client.Networks() - - // Create a new network - n := &api.Network{ - Name: "my-new-network", - IPAddressRange: "10.1.1.0/24" - } - - id, err := networks.Create(context.Background(), n) - if err != nil { - panic(err) - } - - ... -} -``` - -To run this example, start a Drago server: - -``` -drago agent --server -``` - -Copy the code above into a file such as `main.go`, and run it. - -After running the code, you can also view the values in the Drago UI on your local machine at http://localhost:8080/ui/ diff --git a/api/acl.go b/api/acl.go deleted file mode 100644 index f0f1ef8..0000000 --- a/api/acl.go +++ /dev/null @@ -1,33 +0,0 @@ -package api - -import ( - "path" - - "github.com/seashell/drago/drago/structs" -) - -const ( - aclPath = "/api/acl" -) - -// ACL is a handle to the ACL API -type ACL struct { - client *Client -} - -// Networks returns a handle on the networks endpoints. -func (c *Client) ACL() *ACL { - return &ACL{client: c} -} - -// Boostrap : -func (a *ACL) Bootstrap() (*structs.ACLToken, error) { - - var token structs.ACLToken - err := a.client.createResource(path.Join(aclPath, "bootstrap"), nil, &token) - if err != nil { - return nil, err - } - - return &token, nil -} diff --git a/api/acl_policy.go b/api/acl_policy.go deleted file mode 100644 index 831ade9..0000000 --- a/api/acl_policy.go +++ /dev/null @@ -1,68 +0,0 @@ -package api - -import ( - "path" - - "github.com/seashell/drago/drago/structs" -) - -const ( - aclPoliciesPath = "/api/acl/policies" -) - -// ACLPolicies is a handle to the ACL policies API -type ACLPolicies struct { - client *Client -} - -// ACLPolicies returns a handle on the ACL policies endpoints. -func (c *Client) ACLPolicies() *ACLPolicies { - return &ACLPolicies{client: c} -} - -// Create : -func (p *ACLPolicies) Upsert(policy *structs.ACLPolicy) (*structs.ACLPolicy, error) { - - var rcvPolicy *structs.ACLPolicy - err := p.client.createResource(aclPoliciesPath, policy, &rcvPolicy) - if err != nil { - return nil, err - } - - return policy, nil -} - -// Delete : -func (p *ACLPolicies) Delete(name string) error { - - err := p.client.deleteResource(name, aclPoliciesPath, nil) - if err != nil { - return err - } - - return nil -} - -// Get : -func (p *ACLPolicies) Get(name string) (*structs.ACLPolicy, error) { - - out := &structs.ACLPolicy{} - err := p.client.getResource(aclPoliciesPath, name, out) - if err != nil { - return nil, err - } - - return out, nil -} - -// List : -func (p *ACLPolicies) List() ([]*structs.ACLPolicyListStub, error) { - - var items []*structs.ACLPolicyListStub - err := p.client.listResources(path.Join(aclPoliciesPath, "/"), nil, &items) - if err != nil { - return nil, err - } - - return items, nil -} diff --git a/api/acl_token.go b/api/acl_token.go deleted file mode 100644 index 70d395a..0000000 --- a/api/acl_token.go +++ /dev/null @@ -1,94 +0,0 @@ -package api - -import ( - "path" - - "github.com/seashell/drago/drago/structs" -) - -const ( - aclTokensPath = "/api/acl/tokens" -) - -// ACLTokens is a handle to the ACL tokens API -type ACLTokens struct { - client *Client -} - -// ACLTokens returns a handle on the ACL tokens endpoints. -func (c *Client) ACLTokens() *ACLTokens { - return &ACLTokens{client: c} -} - -// Create : -func (t *ACLTokens) Create(token *structs.ACLToken) (*structs.ACLToken, error) { - - out := &structs.ACLToken{} - - err := t.client.createResource(aclTokensPath, token, out) - if err != nil { - return nil, err - } - - return out, nil -} - -// Delete : -func (t *ACLTokens) Delete(id string) error { - - err := t.client.deleteResource(id, aclTokensPath, nil) - if err != nil { - return err - } - - return nil -} - -// Update : -func (t *ACLTokens) Update(token *structs.ACLToken) (*structs.ACLToken, error) { - - out := structs.ACLToken{} - - err := t.client.updateResource(token.ID, aclTokensPath, token, &out) - if err != nil { - return nil, err - } - - return &out, nil -} - -// Get : -func (t *ACLTokens) Get(id string) (*structs.ACLToken, error) { - - var token *structs.ACLToken - err := t.client.getResource(aclTokensPath, id, &token) - if err != nil { - return nil, err - } - - return token, nil -} - -// List : -func (t *ACLTokens) List() ([]*structs.ACLTokenListStub, error) { - - var items []*structs.ACLTokenListStub - err := t.client.listResources(path.Join(aclTokensPath, "/"), nil, &items) - if err != nil { - return nil, err - } - - return items, nil -} - -// Self : -func (t *ACLTokens) Self() (*structs.ACLToken, error) { - - var token *structs.ACLToken - err := t.client.getResource(aclTokensPath, "self", &token) - if err != nil { - return nil, err - } - - return token, nil -} diff --git a/api/agent.go b/api/agent.go deleted file mode 100644 index 7216bde..0000000 --- a/api/agent.go +++ /dev/null @@ -1,33 +0,0 @@ -package api - -import ( - "path" - - "github.com/seashell/drago/drago/structs" -) - -const ( - agentPath = "/api/agent" -) - -// Agent is a handle to the agent API -type Agent struct { - client *Client -} - -// Agent returns a handle on the agent endpoints. -func (c *Client) Agent() *Agent { - return &Agent{client: c} -} - -// Self : -func (t *Agent) Self() (*structs.Agent, error) { - - var agent *structs.Agent - err := t.client.getResource(path.Join(agentPath, "self"), "", &agent) - if err != nil { - return nil, err - } - - return agent, nil -} diff --git a/api/api.go b/api/api.go deleted file mode 100644 index 8f662c3..0000000 --- a/api/api.go +++ /dev/null @@ -1,185 +0,0 @@ -package api - -import ( - "bytes" - "encoding/json" - "fmt" - "io/ioutil" - "net/http" - "net/url" - "path" - - "github.com/hashicorp/go-cleanhttp" -) - -// Client provides a client to the Drago API -type Client struct { - config *Config - httpClient *http.Client -} - -// NewClient returns a new Drago API client -func NewClient(config *Config) (*Client, error) { - - config = DefaultConfig().Merge(config) - - if err := config.Validate(); err != nil { - return nil, err - } - - client := &Client{ - config: config, - httpClient: cleanhttp.DefaultClient(), - } - - return client, nil -} - -func (c *Client) getResource(p string, id string, receiver interface{}) error { - - u, err := url.Parse(c.config.Address) - if err != nil { - return err - } - - u.Path += path.Join(p, id) - - req, err := http.NewRequest("GET", u.String(), nil) - if err != nil { - return err - } - - c.addHeaders(req) - - return c.doRequest(req, receiver) -} - -func (c *Client) createResource(p string, sender interface{}, receiver interface{}) error { - - u, err := url.Parse(c.config.Address) - if err != nil { - return err - } - - u.Path += p + "/" - - b := &bytes.Buffer{} - json.NewEncoder(b).Encode(sender) - - req, err := http.NewRequest("POST", u.String(), b) - if err != nil { - return err - } - - c.addHeaders(req) - - return c.doRequest(req, receiver) -} - -func (c *Client) updateResource(id, p string, sender interface{}, receiver interface{}) error { - - u, err := url.Parse(c.config.Address) - if err != nil { - return err - } - - u.Path += path.Join(p, id) - - b := &bytes.Buffer{} - json.NewEncoder(b).Encode(sender) - - req, err := http.NewRequest("PATCH", u.String(), b) - if err != nil { - return err - } - - c.addHeaders(req) - return c.doRequest(req, receiver) -} - -func (c *Client) deleteResource(id, p string, receiver interface{}) error { - - u, err := url.Parse(c.config.Address) - if err != nil { - return err - } - - u.Path += path.Join(p, id) - - req, err := http.NewRequest("DELETE", u.String(), nil) - if err != nil { - return err - } - - c.addHeaders(req) - - return c.doRequest(req, receiver) -} - -func (c *Client) listResources(p string, filters map[string][]string, receiver interface{}) error { - - u, err := url.Parse(c.config.Address) - if err != nil { - return err - } - - u.Path += p - - req, err := http.NewRequest("GET", u.String(), nil) - if err != nil { - return err - } - - c.addQuery(filters, req) - c.addHeaders(req) - - return c.doRequest(req, receiver) -} - -func (c *Client) addHeaders(req *http.Request) { - req.Header.Set("Content-Type", "application/json") - req.Header.Add("X-Drago-Token", c.config.Token) -} - -func (c *Client) addQuery(filters map[string][]string, req *http.Request) { - - q := req.URL.Query() - - for k, values := range filters { - for _, v := range values { - q.Add(k, v) - } - } - - req.URL.RawQuery = q.Encode() -} - -func (c *Client) doRequest(req *http.Request, receiver interface{}) error { - - res, err := c.httpClient.Do(req) - if err != nil { - return err - } - - defer res.Body.Close() - - if ok := res.StatusCode >= 200 && res.StatusCode < 300; !ok { - - err := CodedError{ - Code: res.StatusCode, - } - - if err := json.NewDecoder(res.Body).Decode(&err); err != nil { - resBody, _ := ioutil.ReadAll(res.Body) - return fmt.Errorf("%v (%v)", res.Status, string(resBody)) - } - - return err - } - - if receiver != nil { - return json.NewDecoder(res.Body).Decode(receiver) - } - - return nil -} diff --git a/api/config.go b/api/config.go deleted file mode 100644 index 23eadbe..0000000 --- a/api/config.go +++ /dev/null @@ -1,64 +0,0 @@ -package api - -import ( - "net/url" - "time" -) - -const ( - // DefaultAddress is the default Drago server address. - DefaultAddress = "http://127.0.0.1:8080" - - // DefaultTimeout is the default request timeout. - DefaultTimeout = 2 * time.Second -) - -// Config contains configurations for Drago's API client. -type Config struct { - // URL of the Drago server (e.g. http://127.0.0.1:8080). - Address string - - // Token to be used for authentication. - Token string - - // Request timeout. - Timeout time.Duration -} - -// DefaultConfig returns a default configuration for Drago's API client. -func DefaultConfig() *Config { - config := &Config{ - Address: DefaultAddress, - } - return config -} - -// Validate validates the configurations contained wihin the Config struct. -func (c *Config) Validate() error { - if _, err := url.Parse(c.Address); err != nil { - return err - } - return nil -} - -// Merge merges two API client configurations. -func (c *Config) Merge(b *Config) *Config { - - if b == nil { - return c - } - - result := *c - - if b.Address != "" { - result.Address = b.Address - } - if b.Token != "" { - result.Token = b.Token - } - if b.Timeout != 0 { - result.Timeout = b.Timeout - } - - return &result -} diff --git a/api/connection.go b/api/connection.go deleted file mode 100644 index b47f09d..0000000 --- a/api/connection.go +++ /dev/null @@ -1,76 +0,0 @@ -package api - -import ( - "path" - - "github.com/seashell/drago/drago/structs" -) - -const ( - connectionsPath = "/api/connections" -) - -// Connections is a handle to the connection API -type Connections struct { - client *Client -} - -// Connections returns a handle on the connections endpoints. -func (c *Client) Connections() *Connections { - return &Connections{client: c} -} - -// Get : -func (n *Connections) Get(id string) (*structs.Connection, error) { - - var conn *structs.Connection - err := n.client.getResource(connectionsPath, id, &conn) - if err != nil { - return nil, err - } - - return conn, nil -} - -// List : -func (n *Connections) List() ([]*structs.ConnectionListStub, error) { - - var items []*structs.ConnectionListStub - err := n.client.listResources(path.Join(connectionsPath, "/"), nil, &items) - if err != nil { - return nil, err - } - - return items, nil -} - -func (n *Connections) Create(connection *structs.Connection) error { - - err := n.client.createResource(connectionsPath, connection, nil) - if err != nil { - return err - } - - return nil -} - -func (n *Connections) Update(conn *structs.Connection) (*structs.Connection, error) { - - out := &structs.Connection{} - err := n.client.createResource(connectionsPath, conn, out) - if err != nil { - return nil, err - } - - return out, nil -} - -func (n *Connections) Delete(id string) error { - - err := n.client.deleteResource(id, connectionsPath, nil) - if err != nil { - return err - } - - return nil -} diff --git a/api/error.go b/api/error.go deleted file mode 100644 index 14ebe97..0000000 --- a/api/error.go +++ /dev/null @@ -1,12 +0,0 @@ -package api - -import "fmt" - -type CodedError struct { - Message string - Code int -} - -func (e CodedError) Error() string { - return fmt.Sprintf("%d (%s)", e.Code, e.Message) -} diff --git a/api/interface.go b/api/interface.go deleted file mode 100644 index 0dfaeb4..0000000 --- a/api/interface.go +++ /dev/null @@ -1,85 +0,0 @@ -package api - -import ( - "path" - - "github.com/seashell/drago/drago/structs" -) - -const ( - interfacesPath = "/api/interfaces" -) - -// Interfaces is a handle to the interfaces API -type Interfaces struct { - client *Client -} - -// Interfaces returns a handle on the interfaces endpoints. -func (c *Client) Interfaces() *Interfaces { - return &Interfaces{client: c} -} - -// Get : -func (n *Interfaces) Get(id string) (*structs.Interface, error) { - - var iface *structs.Interface - err := n.client.getResource(interfacesPath, id, &iface) - if err != nil { - return nil, err - } - - return iface, nil -} - -// List : -func (n *Interfaces) List(filters map[string][]string) ([]*structs.InterfaceListStub, error) { - - var items []*structs.InterfaceListStub - err := n.client.listResources(path.Join(interfacesPath, "/"), filters, &items) - if err != nil { - return nil, err - } - - return items, nil -} - -// Create : -func (n *Interfaces) Create(nodeID, networkID string) (*structs.Interface, error) { - - iface := &structs.Interface{ - NodeID: nodeID, - NetworkID: networkID, - } - - out := &structs.Interface{} - err := n.client.createResource(interfacesPath, iface, out) - if err != nil { - return nil, err - } - - return out, nil -} - -// Update : -func (n *Interfaces) Update(iface *structs.Interface) (*structs.Interface, error) { - - out := &structs.Interface{} - err := n.client.createResource(interfacesPath, iface, out) - if err != nil { - return nil, err - } - - return out, err -} - -// Delete : -func (n *Interfaces) Delete(id string) error { - - err := n.client.deleteResource(id, interfacesPath, nil) - if err != nil { - return err - } - - return nil -} diff --git a/api/network.go b/api/network.go deleted file mode 100644 index 3ac6775..0000000 --- a/api/network.go +++ /dev/null @@ -1,67 +0,0 @@ -package api - -import ( - "path" - - "github.com/seashell/drago/drago/structs" -) - -const ( - networksPath = "/api/networks" -) - -// Networks is a handle to the nodes API -type Networks struct { - client *Client -} - -// Networks returns a handle on the networks endpoints. -func (c *Client) Networks() *Networks { - return &Networks{client: c} -} - -// Create : -func (n *Networks) Create(network *structs.Network) error { - - err := n.client.createResource(networksPath, network, nil) - if err != nil { - return err - } - - return err -} - -// Delete : -func (n *Networks) Delete(id string) error { - - err := n.client.deleteResource(id, networksPath, nil) - if err != nil { - return err - } - - return nil -} - -// Get : -func (n *Networks) Get(id string) (*structs.Network, error) { - - var network *structs.Network - err := n.client.getResource(networksPath, id, &network) - if err != nil { - return nil, err - } - - return network, nil -} - -// List : -func (n *Networks) List() ([]*structs.NetworkListStub, error) { - - var items []*structs.NetworkListStub - err := n.client.listResources(path.Join(networksPath, "/"), nil, &items) - if err != nil { - return nil, err - } - - return items, nil -} diff --git a/api/node.go b/api/node.go deleted file mode 100644 index 0ce22f5..0000000 --- a/api/node.go +++ /dev/null @@ -1,45 +0,0 @@ -package api - -import ( - "path" - - "github.com/seashell/drago/drago/structs" -) - -const ( - nodesPath = "/api/nodes" -) - -// Nodes is a handle to the nodes API -type Nodes struct { - client *Client -} - -// Nodes returns a handle on the nodes endpoints. -func (c *Client) Nodes() *Nodes { - return &Nodes{client: c} -} - -// Get : -func (t *Nodes) Get(id string) (*structs.Node, error) { - - var node *structs.Node - err := t.client.getResource(nodesPath, id, &node) - if err != nil { - return nil, err - } - - return node, nil -} - -// List : -func (t *Nodes) List(filters map[string][]string) ([]*structs.NodeListStub, error) { - - var items []*structs.NodeListStub - err := t.client.listResources(path.Join(nodesPath, "/"), filters, &items) - if err != nil { - return nil, err - } - - return items, nil -} diff --git a/architecture.png b/architecture.png new file mode 100644 index 0000000..ad27cc6 Binary files /dev/null and b/architecture.png differ diff --git a/architecture.svg b/architecture.svg new file mode 100644 index 0000000..54e2766 --- /dev/null +++ b/architecture.svg @@ -0,0 +1,687 @@ + + + + diff --git a/build/Dockerfile.builder b/build/Dockerfile.builder deleted file mode 100644 index 205f1cb..0000000 --- a/build/Dockerfile.builder +++ /dev/null @@ -1,22 +0,0 @@ -FROM golang:1.16.2-stretch as drago-builder - -ARG HOST_UID=${HOST_UID} -ARG HOST_USER=${HOST_USER} - -RUN curl -sS http://dl.yarnpkg.com/debian/pubkey.gpg | apt-key add - && \ - echo "deb http://dl.yarnpkg.com/debian/ stable main" | tee /etc/apt/sources.list.d/yarn.list && \ - curl -sL http://deb.nodesource.com/setup_15.x | bash - && \ - apt-get install -y nodejs && \ - apt-get update && \ - apt-get remove cmdtest && \ - apt-get install -y yarn - -RUN apt-get install -y gcc-arm-linux-gnueabihf libc6-dev-armhf-cross \ - gcc-aarch64-linux-gnu libc6-dev-arm64-cross - -RUN if [ "${HOST_USER}" != "root" ]; then \ - (adduser -q --gecos "" --home /home/${HOST_USER} --disabled-password -u ${HOST_UID} ${HOST_USER} \ - && chown -R "${HOST_UID}:${HOST_UID}" /home/${HOST_USER}); \ - fi - -USER ${HOST_USER} \ No newline at end of file diff --git a/build/Dockerfile.linux_amd64 b/build/Dockerfile.linux_amd64 deleted file mode 100644 index 502c423..0000000 --- a/build/Dockerfile.linux_amd64 +++ /dev/null @@ -1,9 +0,0 @@ -FROM alpine:3.13 - -WORKDIR /home - -RUN apk add -U wireguard-tools - -COPY ./bin/linux_amd64/drago ./drago - -ENTRYPOINT [ "./drago" ] \ No newline at end of file diff --git a/client/client.go b/client/client.go deleted file mode 100644 index 2d67c50..0000000 --- a/client/client.go +++ /dev/null @@ -1,532 +0,0 @@ -package client - -import ( - "fmt" - "io/ioutil" - "math/rand" - "os" - "path" - "path/filepath" - "sync" - "time" - - "github.com/seashell/drago/agent/conn" - nic "github.com/seashell/drago/client/nic" - state "github.com/seashell/drago/client/state" - boltdb "github.com/seashell/drago/client/state/boltdb" - structs "github.com/seashell/drago/drago/structs" - log "github.com/seashell/drago/pkg/log" - uuid "github.com/seashell/drago/pkg/uuid" -) - -var ( - defaultRegistrationRetryInterval = 5 * time.Second - defaultReconciliationRetryInterval = 5 * time.Second - defaultReconciliationInterval = 2 * time.Second - defaultFirstHeartbeatDelay = 1 * time.Second - defaultHeartbeatInterval = 5 * time.Second -) - -// Client is the Drago client -type Client struct { - config *Config - - logger log.Logger - - rpc conn.RPCConnection - - state state.Repository - - niController nic.NetworkInterfaceController - niControllerLock sync.Mutex - - node *structs.Node - nodeLock sync.Mutex - - shutdown bool - shutdownCh chan struct{} - shutdownLock sync.Mutex -} - -// New is used to create a new Drago client from the -// configuration, potentially returning an error -func New(conn conn.RPCConnection, config *Config) (*Client, error) { - - rand.Seed(time.Now().Unix()) - - config = DefaultConfig().Merge(config) - - c := &Client{ - config: config, - rpc: conn, - logger: config.Logger.WithName("client"), - shutdownCh: make(chan struct{}), - } - - if err := c.setupState(); err != nil { - return nil, fmt.Errorf("error setting up client state: %v", err) - } - - if err := c.setupNode(); err != nil { - return nil, fmt.Errorf("error setting up node: %v", err) - } - - if err := c.setupNetworkController(); err != nil { - return nil, fmt.Errorf("error setting up network controller: %v", err) - } - - if err := c.setupInterfaces(); err != nil { - return nil, fmt.Errorf("error setting up interfaces: %v", err) - } - - go c.registerAndHeartbeat() - - go c.run() - - c.logger.Infof("started client node %s", c.NodeID()) - - return c, nil -} - -func (c *Client) Node() *structs.Node { - return c.node -} - -// NodeID returns the node ID for the given client -func (c *Client) NodeID() string { - return c.node.ID -} - -// NodeSecretID returns the node secret ID for the given client -func (c *Client) NodeSecretID() string { - return c.node.ID -} - -// Stats is used to return statistics for the server -func (c *Client) Stats() map[string]map[string]string { - - stats := map[string]map[string]string{ - "client": { - "node_id": c.NodeID(), - }, - } - - return stats -} - -func (c *Client) setupNode() error { - - if c.node == nil { - c.node = &structs.Node{} - } - - id, err := c.readOrGenerateNodeID() - if err != nil { - return fmt.Errorf("could not retrieve node ID: %v", err) - } - - secret, err := c.readOrGenerateNodeSecretID() - if err != nil { - return fmt.Errorf("could not retrieve node secret ID: %v", err) - } - - c.node.ID = id - c.node.SecretID = secret - - c.node.Name = c.config.Name - c.node.Meta = c.config.Meta - - c.node.AdvertiseAddress = c.config.AdvertiseAddress - c.node.Status = structs.NodeStatusInit - - if c.node.Name == "" { - if hostname, _ := os.Hostname(); hostname != "" { - c.node.Name = hostname - } else { - c.node.Name = c.node.ID - } - } - if c.node.Meta == nil { - c.node.Meta = make(map[string]string) - } - - return nil -} - -func (c *Client) setupState() error { - - // Ensure the state dir exists. If it was not was specified, - // create a temporary directory to store the client state. - if c.config.StateDir != "" { - if err := os.MkdirAll(c.config.StateDir, 0700); err != nil { - return fmt.Errorf("failed to create state dir: %s", err) - } - } else { - tmp, err := c.createTempDir("DragoClient") - if err != nil { - return fmt.Errorf("failed to create tmp dir for storing state: %s", err) - } - c.config.StateDir = tmp - } - - c.logger.Infof("using state directory %s", c.config.StateDir) - - repo, err := boltdb.NewStateRepository(path.Join(c.config.StateDir, "client.state")) - if err != nil { - return fmt.Errorf("failed to open state database: %v", err) - } - - c.state = repo - - return nil -} - -func (c *Client) setupNetworkController() error { - - nc, err := nic.NewController(&nic.Config{ - InterfacesPrefix: c.config.InterfacesPrefix, - WireguardPath: c.config.WireguardPath, - KeyStore: c.state, // TODO: improve how we store private keys (do we really need to store them?) - }) - if err != nil { - return err - } - - c.niController = nc - - return nil -} - -func (c *Client) setupInterfaces() error { - - current, err := c.niController.Interfaces() - if err != nil { - return fmt.Errorf("could not retrieve network interfaces from network controller: %v", err) - } - - desired, err := c.state.Interfaces() - if err != nil { - return fmt.Errorf("could not retrieve network interfaces from state: %v", err) - } - - c.reconcileInterfaces(current, desired) - - return nil -} - -func (c *Client) registerAndHeartbeat() { - - c.tryToRegisterUntilSuccessful() - - heartbeatCh := time.After(defaultFirstHeartbeatDelay) - - for { - select { - case <-heartbeatCh: - case <-c.shutdownCh: - return - } - - c.logger.Debugf("heartbeating (client -> server)") - - if err := c.updateNodeStatus(); err != nil { - c.logger.Debugf("error updating node status: %v", err) - c.tryToRegisterUntilSuccessful() - heartbeatCh = time.After(defaultHeartbeatInterval) - } else { - heartbeatCh = time.After(defaultHeartbeatInterval) - } - - } -} - -func (c *Client) run() { - - c.logger.Debugf("running node") - - interfacesUpdateCh := make(chan []*structs.Interface) - go c.watchInterfaces(interfacesUpdateCh) - - go c.synchronizeInterfaces() - - for { - select { - case desired := <-interfacesUpdateCh: - c.shutdownLock.Lock() - if c.shutdown { - c.shutdownLock.Unlock() - return - } - - current, err := c.state.Interfaces() - if err != nil { - c.logger.Errorf("could not read interfaces from state repository: %v", err) - } - - c.reconcileInterfaces(current, desired) - - c.shutdownLock.Unlock() - case <-c.shutdownCh: - return - } - } -} - -func (c *Client) reconcileInterfaces(current, desired []*structs.Interface) { - - currentMap := map[string]*structs.Interface{} - for _, i := range current { - currentMap[i.ID] = i - } - - desiredMap := map[string]*structs.Interface{} - for _, i := range desired { - desiredMap[i.ID] = i - } - - diff := interfacesDiff(currentMap, desiredMap) - - c.logger.Debugf("interface updates: (created: %d, deleted: %d, updated: %d, unchanged: %d)", - len(diff.created), len(diff.deleted), len(diff.updated), len(diff.unchanged)) - - c.niControllerLock.Lock() - defer c.niControllerLock.Unlock() - - // Delete old interfaces - for _, id := range diff.deleted { - if err := c.state.DeleteInterfaces([]string{id}); err != nil { - c.logger.Warnf("could not persist interface deletion to the state: %v", err) - } - if err := c.niController.DeleteInterfaceByAlias(id); err != nil { - c.logger.Warnf("could not delete interface: %v", err) - } - } - - // Create a new interface from scratch - for _, id := range diff.created { - - iface := desiredMap[id] - - err := c.state.UpsertInterface(iface) - if err != nil { - c.logger.Warnf("could not persist interface: %v", err) - continue - } - - err = c.niController.CreateInterface(iface) - if err != nil { - c.logger.Warnf("could not create wireguard interface: %v", err) - } - } - - // Update an existing interface - for _, id := range diff.updated { - - iface := desiredMap[id] - - err := c.state.UpsertInterface(iface) - if err != nil { - c.logger.Warnf("could not persist interface: %v", err) - continue - } - - if err := c.niController.UpdateInterface(iface); err != nil { - c.logger.Warnf("could not update wireguard interface: %v", err) - } - } - -} - -func (c *Client) watchInterfaces(ch chan []*structs.Interface) { - - req := &structs.NodeSpecificRequest{ - NodeID: c.NodeID(), - } - - for { - - c.logger.Debugf("updating interface configuration (server -> client)") - - var resp structs.NodeInterfacesResponse - err := c.RPC("Node.GetInterfaces", req, &resp) - if err != nil { - c.logger.Debugf("error fetching interfaces: %v", err) - retryCh := time.After(defaultReconciliationRetryInterval) - select { - case <-retryCh: - case <-c.shutdownCh: - return - } - } else { - ch <- resp.Items - } - - retryCh := time.After(c.config.ReconcileInterval) - select { - case <-c.shutdownCh: - return - case <-retryCh: - } - } -} - -func (c *Client) synchronizeInterfaces() { - - for { - - c.logger.Debugf("updating interface status (client -> server)") - - interfaces, err := c.niController.Interfaces() - if err != nil { - c.logger.Warnf("could not retrieve interfaces with public key from controller") - } - - if interfaces == nil { - interfaces = []*structs.Interface{} - } - - req := &structs.NodeInterfaceUpdateRequest{ - NodeID: c.NodeID(), - Interfaces: interfaces, - } - - var resp structs.GenericResponse - err = c.RPC("Node.UpdateInterfaces", req, &resp) - if err != nil { - c.logger.Debugf("error updating interfaces: %v", err) - retryCh := time.After(randomDuration(defaultReconciliationInterval, 0*time.Second)) - select { - case <-retryCh: - case <-c.shutdownCh: - return - } - } - retryCh := time.After(randomDuration(defaultReconciliationInterval, 0*time.Second)) - select { - case <-c.shutdownCh: - return - case <-retryCh: - } - } -} - -func (c *Client) tryToRegisterUntilSuccessful() { - - for { - - c.logger.Debugf("registering node (client -> server)") - - req := &structs.NodeRegisterRequest{ - Node: c.Node(), - } - - var err error - var resp structs.NodeUpdateResponse - if err = c.RPC("Node.Register", req, &resp); err == nil { - - c.nodeLock.Lock() - c.node.Status = structs.NodeStatusReady - c.nodeLock.Unlock() - - return - } - - c.logger.Debugf("error registering node: %v", err) - - retryCh := time.After(randomDuration(defaultRegistrationRetryInterval, 0*time.Second)) - - select { - case <-retryCh: - case <-c.shutdownCh: - return - } - } -} - -func (c *Client) updateNodeStatus() error { - - c.logger.Debugf("updating node status (client -> server)") - - req := &structs.NodeUpdateStatusRequest{ - NodeID: c.NodeID(), - Status: structs.NodeStatusReady, - AdvertiseAddress: c.Node().AdvertiseAddress, - Meta: c.node.Meta, - } - - var err error - var resp structs.NodeUpdateResponse - if err = c.RPC("Node.UpdateStatus", req, &resp); err != nil { - return err - } - - return nil -} - -func (c *Client) RPC(method string, args interface{}, reply interface{}) error { - return c.rpc.Call(method, args, reply) -} - -// Shutdown is used to tear down the client -func (c *Client) Shutdown() error { - c.shutdownLock.Lock() - defer c.shutdownLock.Unlock() - - if c.shutdown { - c.logger.Infof("client already shutdown") - return nil - } - c.logger.Infof("shutting down") - - c.shutdown = true - close(c.shutdownCh) - - return nil -} - -func (c *Client) createTempDir(pattern string) (string, error) { - p, err := ioutil.TempDir("", pattern) - if err != nil { - return "", fmt.Errorf("could not create temporary directory: %v", err) - } - p, err = filepath.EvalSymlinks(p) - if err != nil { - return "", fmt.Errorf("could not retrieve path to StateDir: %v", err) - } - return p, nil -} - -func (c *Client) readOrGenerateNodeID() (string, error) { - - id := uuid.Generate() - if c.config.DevMode { - return id, nil - } - - path := filepath.Join(c.config.StateDir, "client-id") - - id, err := c.readFileLazy(path, id) - if err != nil { - return "", err - } - - return id, nil -} - -func (c *Client) readOrGenerateNodeSecretID() (string, error) { - - path := filepath.Join(c.config.StateDir, "secret-id") - - secret, err := c.readFileLazy(path, uuid.Generate()) - if err != nil { - return "", err - } - - return secret, nil -} - -// Generates a random duration in the interval [mean-delta, mean+delta] -func randomDuration(mean time.Duration, delta time.Duration) time.Duration { - t := mean.Milliseconds() + int64((rand.Float32()-0.5)*float32(delta.Milliseconds())) - return time.Duration(t * int64(time.Millisecond)) -} diff --git a/client/config.go b/client/config.go deleted file mode 100644 index bf7c876..0000000 --- a/client/config.go +++ /dev/null @@ -1,112 +0,0 @@ -package client - -import ( - "time" - - log "github.com/seashell/drago/pkg/log" - version "github.com/seashell/drago/version" -) - -const ( - defaultLogLevel = "DEBUG" - defaultStateDir = "/tmp/drago" - defaultWireguardPath = "" - defaultInterfacesPrefix = "drago-" -) - -// Config : Drago client configuration -type Config struct { - - // DevMode indicates whether the client is running in development mode. - DevMode bool - - // Name is used to specify the name of the client node. - Name string - - // Version is the version of the Drago client. - Version *version.VersionInfo - - // LogLevel is the level at which the client should output logs. - LogLevel string - - //Logger is the logger the client will use to log. - Logger log.Logger - - // Servers is a list of known server addresses. These are as "host:port". - Servers []string - - // Token contains the auth token used by the client. - Token string - - // StateDir is the directory to store our state in. - StateDir string - - // AdvertiseAddress is the public address advertised to other nodes. - AdvertiseAddress string - - // InterfacesPrefix is the string prepended to the name of all WireGuard - // interfaces created by Drago. - InterfacesPrefix string - - // ReconcileInterval is the interval between two reconciliation cycles. - ReconcileInterval time.Duration - - // WireguardPath is path to the WireGuard binary. - WireguardPath string - - // Meta contains client metadata - Meta map[string]string -} - -// DefaultConfig returns the default configuration. -func DefaultConfig() *Config { - return &Config{ - Name: "", - Version: version.GetVersion(), - LogLevel: defaultLogLevel, - Servers: []string{"localhost:8081"}, - StateDir: defaultStateDir, - InterfacesPrefix: defaultInterfacesPrefix, - ReconcileInterval: 5 * time.Second, - WireguardPath: defaultWireguardPath, - Meta: map[string]string{}, - } -} - -// Merge combines two config structs, returning the result. -func (c *Config) Merge(b *Config) *Config { - result := *c - - if b.Name != "" { - result.Name = b.Name - } - if b.Logger != nil { - result.Logger = b.Logger - } - if b.LogLevel != "" { - result.LogLevel = b.LogLevel - } - if b.Servers != nil { - result.Servers = b.Servers - } - if b.StateDir != "" { - result.StateDir = b.StateDir - } - if b.AdvertiseAddress != "" { - result.AdvertiseAddress = b.AdvertiseAddress - } - if b.InterfacesPrefix != "" { - result.InterfacesPrefix = b.InterfacesPrefix - } - if b.ReconcileInterval != 0 { - result.ReconcileInterval = b.ReconcileInterval - } - if b.WireguardPath != "" { - result.WireguardPath = b.WireguardPath - } - if b.Meta != nil { - result.Meta = b.Meta - } - - return &result -} diff --git a/client/nic/controller.go b/client/nic/controller.go deleted file mode 100644 index fe730f9..0000000 --- a/client/nic/controller.go +++ /dev/null @@ -1,343 +0,0 @@ -package nic - -import ( - "bytes" - "crypto/rand" - "encoding/hex" - "fmt" - "net" - "os/exec" - "strings" - "time" - - structs "github.com/seashell/drago/drago/structs" - util "github.com/seashell/drago/pkg/util" - netlink "github.com/vishvananda/netlink" - wgctrl "golang.zx2c4.com/wireguard/wgctrl" - wgtypes "golang.zx2c4.com/wireguard/wgctrl/wgtypes" -) - -const ( - linkTypeWireguard = "wireguard" -) - -// Config contains configurations for a network controller. -type Config struct { - // InterfacesPrefix defines the string prepended to each interface name. - // Example: if InterfacePrefix is "abc", interfaces will be named as - // "abc-xxxxxx", where "xxxxxx" is a random string. - InterfacesPrefix string - - // WireguardPath is the path to a userspace Wireguard binary. In case - // it is not defined, Drago will try to use the kernel module. - WireguardPath string - - // KeyStore is an implementation of the KeyStore interface, used by the - // Controller to cache private keys for each interface. - KeyStore PrivateKeyStore -} - -// Controller : network interface controller. -type Controller struct { - config *Config - wg *wgctrl.Client -} - -// NewController : -func NewController(config *Config) (*Controller, error) { - - wg, err := wgctrl.New() - if err != nil { - return nil, err - } - - if config.KeyStore == nil { - panic("must provide a key store") - } - - c := &Controller{ - config: config, - wg: wg, - } - - return c, nil -} - -// Interfaces returns a slice of all network interfaces managed by -// the controller. -func (c *Controller) Interfaces() ([]*structs.Interface, error) { - - out := []*structs.Interface{} - - links, err := linksByPrefix(c.config.InterfacesPrefix) - if err != nil { - return nil, err - } - - for _, l := range links { - - dev, err := c.wg.Device(l.Attrs().Name) - if err != nil { - return nil, err - } - - out = append(out, &structs.Interface{ - ID: l.Attrs().Alias, - Name: util.StrToPtr(l.Attrs().Name), - ListenPort: &dev.ListenPort, - PublicKey: util.StrToPtr(dev.PrivateKey.PublicKey().String()), - // TODO: capture other information that might be useful e.g. for diagnosis (see l.Attrs().Statistics) - }) - } - - return out, nil -} - -// DeleteInterfaceByName deletes a network interface and all associated routes by name. -func (c *Controller) DeleteInterfaceByName(s string) error { - err := deleteLinkAndRoutesByName(s) - if err != nil { - return err - } - return nil -} - -// DeleteInterfaceByAlias deletes a network interface and all associated routes by alias. -func (c *Controller) DeleteInterfaceByAlias(s string) error { - err := deleteLinksAndRoutesByAlias(s) - if err != nil { - return err - } - return nil -} - -// DeleteAllInterfaces deletes all network interfaces and routes. -func (c *Controller) DeleteAllInterfaces() error { - err := deleteLinksAndRoutesByPrefix(c.config.InterfacesPrefix) - if err != nil { - return err - } - return nil -} - -// CreateInterface ... -func (c *Controller) CreateInterface(iface *structs.Interface) error { - - linkName, linkAlias := c.randomInterfaceName(), iface.ID - - err := c.createLink(linkName, linkAlias) - if err != nil { - return err - } - - return c.UpdateInterface(iface) - -} - -// UpdateInterface : -func (c *Controller) UpdateInterface(iface *structs.Interface) error { - return c.configureLink(iface) -} - -func (c *Controller) createLink(name string, alias string) error { - - attrs := netlink.NewLinkAttrs() - attrs.Name, attrs.Alias = name, alias - - // Create a new WireGuard interface. If a path to a valid userspace WireGuard binary - // was specified in the configurations, use if. Otherwise, create the interface manually. - if c.config.WireguardPath != "" { - - wgt, err := wgImplementationType(c.config.WireguardPath) - if err != nil { - return err - } - - var cmd *exec.Cmd - - switch wgt { - case "wireguard-go": - cmd = exec.Command(c.config.WireguardPath, name) - case "boringtun": - cmd = exec.Command(c.config.WireguardPath, name, "--disable-drop-privileges", "root") - } - - if err := cmd.Run(); err != nil { - return fmt.Errorf("can't create network interface with specified wireguard binary: %s", err.Error()) - } - - } else { - if err := netlink.LinkAdd(&netlink.Wireguard{LinkAttrs: attrs}); err != nil { - return fmt.Errorf("can't create network interface : %s", err.Error()) - } - } - - link, err := netlink.LinkByName(name) - if err != nil { - return err - } - - err = netlink.LinkSetAlias(link, alias) - if err != nil { - return err - } - - return nil -} - -func (c *Controller) configureLink(iface *structs.Interface) error { - - link, err := netlink.LinkByAlias(iface.ID) - if err != nil { - return err - } - - err = netlink.LinkSetDown(link) - if err != nil { - return err - } - - // Create a new private key for this interface. - // If a private key has already been created, use it. - // TODO: implement an expiration/rotation strategy. - key, err := c.config.KeyStore.KeyByID(iface.ID) - if err != nil { - wgKey, err := wgtypes.GeneratePrivateKey() - if err != nil { - return fmt.Errorf("could not generate private key: %v", err) - } - key = &PrivateKey{ - ID: iface.ID, - Key: wgKey.String(), - CreatedAt: time.Now().Unix(), - } - err = c.config.KeyStore.UpsertKey(key) - if err != nil { - return fmt.Errorf("could not persist private key: %v", err) - } - } - - wgKey, err := wgtypes.ParseKey(key.Key) - if err != nil { - return fmt.Errorf("could not parse private key: %v", err) - } - - config := wgtypes.Config{ - PrivateKey: &wgKey, - ListenPort: iface.ListenPort, - Peers: []wgtypes.PeerConfig{}, - ReplacePeers: true, - } - - for _, peer := range iface.Peers { - peerConfig, err := c.newPeerConfig(peer) - if err != nil { - return err - } - config.Peers = append(config.Peers, *peerConfig) - } - - err = c.wg.ConfigureDevice(link.Attrs().Name, config) - if err != nil { - return err - } - - // Assign IP address in CIDR format to the newly created link - err = setLinkAddress(link, iface.Address) - if err != nil { - return err - } - - err = netlink.LinkSetUp(link) - if err != nil { - return err - } - - for _, peerConfig := range config.Peers { - for _, ip := range peerConfig.AllowedIPs { - if err = netlink.RouteReplace(&netlink.Route{LinkIndex: link.Attrs().Index, Dst: &ip}); err != nil { - return err - } - } - } - - return nil -} - -func (c *Controller) newPeerConfig(peer *structs.Peer) (*wgtypes.PeerConfig, error) { - - var err error - - config := &wgtypes.PeerConfig{ - Remove: false, - UpdateOnly: false, - ReplaceAllowedIPs: true, - PresharedKey: nil, - PublicKey: wgtypes.Key{}, - AllowedIPs: []net.IPNet{}, - Endpoint: nil, - PersistentKeepaliveInterval: nil, - } - - if peer.PublicKey != nil { - var key wgtypes.Key - if key, err = wgtypes.ParseKey(*peer.PublicKey); err != nil { - return nil, err - } - config.PublicKey = key - } - - for _, ip := range peer.AllowedIPs { - _, parsed, err := net.ParseCIDR(ip) - if err != nil { - return nil, err - } - config.AllowedIPs = append(config.AllowedIPs, *parsed) - } - - if peer.PersistentKeepalive != nil { - pk := time.Duration(*peer.PersistentKeepalive) * time.Second - config.PersistentKeepaliveInterval = &pk - } - - if peer.Address != nil { - port := 51820 // TODO: should we use this or the zero value by default? - if peer.Port != nil { - port = *peer.Port - } - config.Endpoint = &net.UDPAddr{ - IP: net.ParseIP(*peer.Address), - Port: port, - } - } - - return config, nil -} - -func (c *Controller) randomInterfaceName() string { - buf := make([]byte, 3) - if _, err := rand.Read(buf); err != nil { - panic(fmt.Errorf("failed to read random bytes: %v", err)) - } - return c.config.InterfacesPrefix + "-" + hex.EncodeToString(buf) -} - -func wgImplementationType(path string) (string, error) { - - var out bytes.Buffer - - cmd := exec.Command(path, "--version") - cmd.Stdout = &out - - if err := cmd.Run(); err != nil { - return "", err - } - - if strings.Contains(out.String(), "wireguard-go") { - return "wireguard-go", nil - } else if strings.Contains(out.String(), "boringtun") { - return "boringtun", nil - } - - return "", fmt.Errorf("unknown wireguard implementation") -} diff --git a/client/nic/controller_test.go b/client/nic/controller_test.go deleted file mode 100644 index 160c98f..0000000 --- a/client/nic/controller_test.go +++ /dev/null @@ -1,74 +0,0 @@ -package nic - -import ( - "fmt" - "testing" - - structs "github.com/seashell/drago/drago/structs" -) - -func TestCreateInterface(t *testing.T) { - config := &Config{ - InterfacesPrefix: "drago", - WireguardPath: "./wireguard", - } - - c, err := NewController(config) - if err != nil { - t.Fatal(err) - } - - key, err := c.GenerateKey() - if err != nil { - t.Error() - } - - err = c.CreateInterfaceWithKey(&structs.Interface{ - Name: "1234567890abcd", - Address: "192.168.2.1/24", - Peers: []*structs.Peer{}, - }, key) - if err != nil { - t.Error(err) - } -} - -func TestListInterfaces(t *testing.T) { - - config := &Config{ - InterfacesPrefix: "drago", - WireguardPath: "./wireguard", - } - - c, err := NewController(config) - if err != nil { - t.Fatal(err) - } - - ifaces, err := c.Interfaces() - if err != nil { - t.Fatal(err) - } - - for _, i := range ifaces { - fmt.Printf("%s (%s)", i.Name, i.Address) - } -} - -func TestDeleteAllInterfaces(t *testing.T) { - return - config := &Config{ - InterfacesPrefix: "drago", - WireguardPath: "./wireguard", - } - - c, err := NewController(config) - if err != nil { - t.Fatal(err) - } - - err = c.DeleteAllInterfaces() - if err != nil { - t.Fatal(err) - } -} diff --git a/client/nic/nic.go b/client/nic/nic.go deleted file mode 100644 index 331d35d..0000000 --- a/client/nic/nic.go +++ /dev/null @@ -1,27 +0,0 @@ -package nic - -import ( - structs "github.com/seashell/drago/drago/structs" -) - -// NetworkInterfaceController provides network configuration capabilities. -type NetworkInterfaceController interface { - Interfaces() ([]*structs.Interface, error) - CreateInterface(iface *structs.Interface) error - UpdateInterface(iface *structs.Interface) error - DeleteInterfaceByAlias(s string) error - DeleteInterfaceByName(s string) error - DeleteAllInterfaces() error -} - -type PrivateKeyStore interface { - KeyByID(id string) (*PrivateKey, error) - UpsertKey(key *PrivateKey) error - DeleteKey(id string) error -} - -type PrivateKey struct { - ID string - Key string - CreatedAt int64 -} diff --git a/client/nic/util.go b/client/nic/util.go deleted file mode 100644 index 36667e5..0000000 --- a/client/nic/util.go +++ /dev/null @@ -1,146 +0,0 @@ -package nic - -import ( - "strings" - - netlink "github.com/vishvananda/netlink" -) - -func linkNameByAlias(s string) (string, error) { - link, err := netlink.LinkByAlias(s) - if err != nil { - return "", err - } - return link.Attrs().Name, nil - -} - -func deleteLinkAndRoutesByName(s string) error { - - link, err := netlink.LinkByName(s) - if err != nil { - return err - } - - deleteLinkAndRoutes(link) - - return nil -} - -func deleteLinksAndRoutesByAlias(s string) error { - - links, err := linksByAlias(s) - if err != nil { - return err - } - - for _, l := range links { - err := deleteLinkAndRoutes(l) - if err != nil { - return err - } - } - - return nil -} - -func deleteLinksAndRoutesByPrefix(s string) error { - - links, err := linksByPrefix(s) - if err != nil { - return err - } - - for _, l := range links { - err := deleteLinkAndRoutes(l) - if err != nil { - return err - } - } - - return nil -} - -func deleteLinkAndRoutes(link netlink.Link) error { - - // List routes - routes, err := netlink.RouteList(&netlink.Wireguard{LinkAttrs: *link.Attrs()}, 0) - if err != nil { - return err - } - - // Delete routes - for _, r := range routes { - if err = netlink.RouteDel(&r); err != nil { - return err - } - } - - // Delete link - if err := netlink.LinkDel(&netlink.Wireguard{LinkAttrs: *link.Attrs()}); err != nil { - return err - } - - return nil -} - -func linksByPrefix(p string) ([]netlink.Link, error) { - - out := []netlink.Link{} - - links, _ := netlink.LinkList() - for _, l := range links { - if strings.HasPrefix(l.Attrs().Name, p) { - out = append(out, l) - } - } - - return out, nil -} - -func linksByAlias(s string) ([]netlink.Link, error) { - - out := []netlink.Link{} - - links, _ := netlink.LinkList() - for _, l := range links { - if l.Attrs().Alias == s { - out = append(out, l) - } - } - - return out, nil -} - -func setLinkAddress(link netlink.Link, cidr *string) error { - - var err error - var addr *netlink.Addr - - if cidr != nil && *cidr != "" { - addr, err = netlink.ParseAddr(*cidr) - if err != nil { - return err - } - } - - if addr != nil { - err = netlink.AddrReplace(link, addr) - if err != nil { - return err - } - } else { - addrs, err := netlink.AddrList(link, 0) - if err != nil { - return err - } - for _, addr := range addrs { - err = netlink.AddrDel(link, &addr) - if err != nil { - return err - } - } - } - - return err -} diff --git a/client/state/boltdb/boltdb.go b/client/state/boltdb/boltdb.go deleted file mode 100644 index d9dc0aa..0000000 --- a/client/state/boltdb/boltdb.go +++ /dev/null @@ -1,154 +0,0 @@ -package boltdb - -import ( - "encoding/json" - "fmt" - - "github.com/seashell/drago/client/nic" - "github.com/seashell/drago/drago/structs" - "go.etcd.io/bbolt" - bolt "go.etcd.io/bbolt" -) - -var ( - interfacesBucketName = []byte("interfaces") - privateKeysBucketName = []byte("keys") -) - -// StateRepository ... -type StateRepository struct { - db *bolt.DB -} - -// NewStateRepository creates a new BoltDB state repository -func NewStateRepository(path string) (*StateRepository, error) { - db, err := bolt.Open(path, 0666, nil) - if err != nil { - return nil, err - } - - err = db.Update(func(tx *bbolt.Tx) error { - - if _, err := tx.CreateBucketIfNotExists(interfacesBucketName); err != nil { - return err - } - - if _, err := tx.CreateBucketIfNotExists(privateKeysBucketName); err != nil { - return err - } - - return nil - }) - - if err != nil { - panic(err) - } - - return &StateRepository{db}, nil - -} - -// Name : -func (r *StateRepository) Name() string { - return "boltdb" -} - -// Interfaces : -func (r *StateRepository) Interfaces() ([]*structs.Interface, error) { - - ifaces := []*structs.Interface{} - - err := r.db.View(func(tx *bbolt.Tx) error { - b := tx.Bucket(interfacesBucketName) - err := b.ForEach(func(k []byte, v []byte) error { - - iface := &structs.Interface{} - - err := decode(v, iface) - if err != nil { - return err - } - - ifaces = append(ifaces, iface) - - return nil - }) - return err - }) - - return ifaces, err - -} - -// UpsertInterface : -func (r *StateRepository) UpsertInterface(iface *structs.Interface) error { - - err := r.db.Update(func(tx *bbolt.Tx) error { - b := tx.Bucket(interfacesBucketName) - return b.Put([]byte(iface.ID), encode(iface)) - }) - - return err -} - -// DeleteInterfaces : -func (r *StateRepository) DeleteInterfaces(ids []string) error { - - err := r.db.Update(func(tx *bbolt.Tx) error { - b := tx.Bucket(interfacesBucketName) - for _, id := range ids { - if err := b.Delete([]byte(id)); err != nil { - return err - } - } - return nil - }) - - return err -} - -func (r *StateRepository) KeyByID(id string) (*nic.PrivateKey, error) { - - var key *nic.PrivateKey - - err := r.db.View(func(tx *bbolt.Tx) error { - v := tx.Bucket(privateKeysBucketName).Get([]byte(id)) - if v != nil { - if err := decode(v, &key); err != nil { - return err - } - return nil - } - return fmt.Errorf("not found") - }) - - return key, err -} - -func (r *StateRepository) UpsertKey(key *nic.PrivateKey) error { - err := r.db.Update(func(tx *bbolt.Tx) error { - b := tx.Bucket(privateKeysBucketName) - return b.Put([]byte(key.ID), encode(key)) - }) - return err -} - -func (r *StateRepository) DeleteKey(id string) error { - err := r.db.View(func(tx *bbolt.Tx) error { - b := tx.Bucket(privateKeysBucketName) - return b.Delete([]byte(id)) - }) - return err -} - -func encode(in interface{}) []byte { - out, err := json.Marshal(in) - if err != nil { - panic(err) - } - return out -} - -func decode(encoded []byte, out interface{}) error { - return json.Unmarshal(encoded, out) -} diff --git a/client/state/inmem/inmem.go b/client/state/inmem/inmem.go deleted file mode 100644 index f04cf2d..0000000 --- a/client/state/inmem/inmem.go +++ /dev/null @@ -1,58 +0,0 @@ -package state - -import ( - "log" - "sync" - - "github.com/seashell/drago/drago/structs" -) - -type Repository struct { - logger log.Logger - - // interface_id -> value - interfaces map[string]*structs.Interface - - mu sync.RWMutex -} - -func NewRepository(logger log.Logger) *Repository { - return &Repository{ - interfaces: make(map[string]*structs.Interface), - logger: logger, - } -} - -func (r *Repository) Name() string { - return "inmem" -} - -func (r *Repository) Interfaces() ([]*structs.Interface, error) { - r.mu.RLock() - defer r.mu.RUnlock() - - ifaces := make([]*structs.Interface, 0, len(r.interfaces)) - for _, v := range r.interfaces { - ifaces = append(ifaces, v) - } - - return ifaces, nil -} - -func (r *Repository) UpsertInterface(iface *structs.Interface) error { - r.mu.Lock() - defer r.mu.Unlock() - r.interfaces[iface.ID] = iface - return nil -} - -func (r *Repository) DeleteInterfaces(ids []string) error { - r.mu.Lock() - defer r.mu.Unlock() - - for _, id := range ids { - delete(r.interfaces, id) - } - - return nil -} diff --git a/client/state/state.go b/client/state/state.go deleted file mode 100644 index 4ea0a39..0000000 --- a/client/state/state.go +++ /dev/null @@ -1,21 +0,0 @@ -package state - -import ( - "github.com/seashell/drago/client/nic" - "github.com/seashell/drago/drago/structs" -) - -// Repository : -type Repository interface { - Name() string - - // Client state - Interfaces() ([]*structs.Interface, error) - UpsertInterface(*structs.Interface) error - DeleteInterfaces(id []string) error - - // Key store - KeyByID(id string) (*nic.PrivateKey, error) - UpsertKey(key *nic.PrivateKey) error - DeleteKey(id string) error -} diff --git a/client/util.go b/client/util.go deleted file mode 100644 index c635d5d..0000000 --- a/client/util.go +++ /dev/null @@ -1,64 +0,0 @@ -package client - -import ( - "io/ioutil" - "os" - "reflect" - - "github.com/seashell/drago/drago/structs" -) - -type Diff struct { - created []string - deleted []string - updated []string - unchanged []string -} - -// Read file contents if they exist, or persist and return a default value otherwise. -func (c *Client) readFileLazy(path string, s string) (string, error) { - - var out string - - buf, err := ioutil.ReadFile(path) - if err != nil && !os.IsNotExist(err) { - return "", err - } - - if len(buf) != 0 { - out = string(buf) - } else { - out = s - if err := ioutil.WriteFile(path, []byte(s), 0700); err != nil { - return "", err - } - } - - return out, nil -} - -func interfacesDiff(old, new map[string]*structs.Interface) Diff { - - diff := Diff{} - - for _, vold := range old { - if _, ok := new[vold.ID]; !ok { - diff.deleted = append(diff.deleted, vold.ID) - } - } - - for _, vnew := range new { - if vold, ok := old[vnew.ID]; !ok { - diff.created = append(diff.created, vnew.ID) - } else { - vnew = vold.Merge(vnew) - if !reflect.DeepEqual(vold, vnew) { - diff.updated = append(diff.updated, vnew.ID) - } else { - diff.unchanged = append(diff.unchanged, vold.ID) - } - } - } - - return diff -} diff --git a/command/acl.go b/command/acl.go deleted file mode 100644 index 0b35643..0000000 --- a/command/acl.go +++ /dev/null @@ -1,46 +0,0 @@ -package command - -import ( - "context" - "strings" - - cli "github.com/seashell/drago/pkg/cli" -) - -// ACLCommand : -type ACLCommand struct { - UI cli.UI -} - -// Name : -func (c *ACLCommand) Name() string { - return "acl" -} - -// Synopsis : -func (c *ACLCommand) Synopsis() string { - return "Interact with ACL policies and tokens" -} - -// Run : -func (c *ACLCommand) Run(ctx context.Context, args []string) int { - return cli.CommandReturnCodeHelp -} - -// Help : -func (c *ACLCommand) Help() string { - h := ` -Usage: drago aclFor more information, please contact us.
-
-
-
-
.
- {JSON.stringify(props, null, 4)}
-
-