Feature/remove ex db access

.gitignore

@@ -8,4 +8,6 @@ logs
 casbin.md
 data
 vms
-.VSCodeCounter
+.VSCodeCounter
+log.log
+pid

config/config.example.yml

@@ -13,12 +13,6 @@ file-transfer-root: /path/to/desired/filetransfer/root
 ova-dir: /path/to/desired/ova/directory
 state-path: /path/to/desired/state/directory
 
-exercise-service:
-  grpc: "localhost:port"
-  auth-key: dev-auth-key
-  sign-key: dev-sign-key
-  tls-enabled: false
-
 vpn-service:
   endpoint: vpn.localhost
   port: 5353

go.mod

@@ -4,8 +4,6 @@ go 1.18
 
 require (
 	github.com/aau-network-security/gwireguard v1.0.5-0.20230311152758-128bc2b44342
-	github.com/aau-network-security/haaukins-exercises v1.2.2
-	github.com/dgrijalva/jwt-go v3.2.0+incompatible
 	github.com/fsouza/go-dockerclient v1.8.3
 	github.com/gin-contrib/cors v1.4.0
 	github.com/gin-gonic/gin v1.8.1

internal/agent/agent.go

@@ -3,7 +3,6 @@ package agent
 import (
 	"context"
 	"errors"
-	"fmt"
 	"io/ioutil"
 	"os"
 	"path/filepath"
@@ -15,25 +14,21 @@ import (
 	env "github.com/aau-network-security/haaukins-agent/internal/environment"
 	"github.com/aau-network-security/haaukins-agent/internal/environment/lab/virtual"
 	"github.com/aau-network-security/haaukins-agent/internal/worker"
-	"github.com/aau-network-security/haaukins-agent/pkg/proto"
 	pb "github.com/aau-network-security/haaukins-agent/pkg/proto"
-	eproto "github.com/aau-network-security/haaukins-exercises/proto"
 	"github.com/rs/zerolog/log"
 	"gopkg.in/yaml.v2"
 )
 
 var configPath string
 
 type Agent struct {
-	initialized bool
-	config      *Config
-	State       *state.State
-	auth        Authenticator
-	vlib        *virtual.VboxLibrary
+	config *Config
+	State  *state.State
+	auth   Authenticator
+	vlib   *virtual.VboxLibrary
 	pb.UnimplementedAgentServer
 	workerPool worker.WorkerPool
 	newLabs    chan pb.Lab
-	ExClient   eproto.ExerciseStoreClient
 	EnvPool    *env.EnvPool `json:"envpool,omitempty"`
 }
 
@@ -119,20 +114,6 @@ func New(conf *Config) (*Agent, error) {
 		}
 	}
 
-	var initialized = true
-	var exClient eproto.ExerciseStoreClient
-	// Check if exercise service has been configured by daemon
-	if conf.ExerciseService.Grpc == "" {
-		log.Debug().Msg("exercise service not yet configured, waiting for daemon to initiliaze...")
-		initialized = false
-		exClient = nil
-	} else {
-		exClient, err = NewExerciseClientConn(conf.ExerciseService)
-		if err != nil {
-			return nil, fmt.Errorf("error connecting to exercise service: %s", err)
-		}
-	}
-
 	// Creating and starting a workerPool for lab creation
 	// This is to ensure that resources are not spent without having them
 	// Workeramount can be configured from the config
@@ -157,15 +138,13 @@ func New(conf *Config) (*Agent, error) {
 	}
 	// Creating agent struct
 	a := &Agent{
-		initialized: initialized,
-		config:      conf,
-		workerPool:  workerPool,
-		vlib:        vlib,
-		auth:        NewAuthenticator(conf.SignKey, conf.AuthKey),
-		newLabs:     make(chan pb.Lab, 1000),
-		ExClient:    exClient,
-		EnvPool:     envPool,
-		State:       &state.State{},
+		config:     conf,
+		workerPool: workerPool,
+		vlib:       vlib,
+		auth:       NewAuthenticator(conf.SignKey, conf.AuthKey),
+		newLabs:    make(chan pb.Lab, 1000),
+		EnvPool:    envPool,
+		State:      &state.State{},
 	}
 	return a, nil
 }
@@ -192,51 +171,3 @@ func (d *Agent) NewGRPCServer(opts ...grpc.ServerOption) *grpc.Server {
 	}, opts...)
 	return grpc.NewServer(opts...)
 }
-
-// Connect to exdb based on what creds sent by daemon, and write to config
-func (a *Agent) Init(ctx context.Context, req *proto.InitRequest) (*proto.StatusResponse, error) {
-	// Creating service config based on request from daemon
-	var exConf = ServiceConfig{
-		Grpc:       req.Url,
-		AuthKey:    req.AuthKey,
-		SignKey:    req.SignKey,
-		TLSEnabled: req.TlsEnabled,
-	}
-	// Creating new exercise service connection from config
-	log.Debug().Msgf("request: %v", req)
-	exClient, err := NewExerciseClientConn(exConf)
-	if err != nil {
-		log.Error().Err(err).Msg("error connecting to exercise service")
-		return nil, fmt.Errorf("error connecting to exercise service: %s", err)
-	}
-
-	// Saving the config in the agent config
-	a.config.ExerciseService = exConf
-
-	// Updating the config
-	data, err := yaml.Marshal(a.config)
-	if err != nil {
-		log.Error().Err(err).Msg("error marshalling yaml")
-		return nil, fmt.Errorf("error marshalling yaml: %s", err)
-	}
-
-	// Truncates existing file to overwrite with new data
-	f, err := os.Create(configPath)
-	if err != nil {
-		log.Error().Err(err).Msg("error creating or truncating config file")
-		return nil, fmt.Errorf("error creating or truncating config file: %s", err)
-	}
-
-	if err := f.Chmod(0600); err != nil {
-		log.Error().Err(err).Msg("error changing file perms")
-		return nil, fmt.Errorf("error changing file perms: %s", err)
-	}
-
-	if _, err := f.Write(data); err != nil {
-		log.Error().Err(err).Msg("error writing config to file")
-		return nil, fmt.Errorf("error writing config to file: %s", err)
-	}
-	a.initialized = true
-	a.ExClient = exClient
-	return &proto.StatusResponse{Message: "OK"}, nil
-}

internal/agent/config.go

@@ -15,7 +15,6 @@ type Config struct {
 	FileTransferRoot   string                           `yaml:"file-transfer-root"`
 	OvaDir             string                           `yaml:"ova-dir"`
 	StatePath          string                           `yaml:"state-path"`
-	ExerciseService    ServiceConfig                    `yaml:"exercise-service"`
 	VPNService         VPNconf                          `yaml:"vpn-service"`
 	DockerRepositories []dockerclient.AuthConfiguration `yaml:"docker-repositories"`
 	GuacSSL            bool                             `yaml:"guac-ssl"`

internal/agent/environment.go

@@ -16,7 +16,6 @@ import (
 	"github.com/aau-network-security/haaukins-agent/internal/environment/lab/virtual"
 	"github.com/aau-network-security/haaukins-agent/internal/state"
 	"github.com/aau-network-security/haaukins-agent/pkg/proto"
-	eproto "github.com/aau-network-security/haaukins-exercises/proto"
 	"github.com/rs/zerolog/log"
 )
 
@@ -30,9 +29,6 @@ var (
 // beginner events where the user would just need to press the connect button and a lab would be ready with all challenges running.
 func (a *Agent) CreateEnvironment(ctx context.Context, req *proto.CreatEnvRequest) (*proto.StatusResponse, error) {
 	// Env for event already exists, Do not start a new guac container
-	if !a.initialized {
-		return nil, errors.New("agent not yet initialized")
-	}
 	log.Debug().Msgf("got createEnv request: %v", req)
 
 	if a.EnvPool.DoesEnvExist(req.EventTag) {
@@ -47,15 +43,10 @@ func (a *Agent) CreateEnvironment(ctx context.Context, req *proto.CreatEnvReques
 	envConf.WorkerPool = a.workerPool
 	envConf.TeamSize = int(req.TeamSize)
 	log.Debug().Str("envtype", envConf.Type.String()).Msg("making environment with type")
-	// Get exercise info from exercise db
 
-	exerDbConfs, err := a.ExClient.GetExerciseByTags(ctx, &eproto.GetExerciseByTagsRequest{Tag: req.Exercises})
-	if err != nil {
-		return nil, fmt.Errorf("error getting exercises: %s", err)
-	}
 	// Unpack into exercise slice
 	var exerConfs []exercise.ExerciseConfig
-	for _, e := range exerDbConfs.Exercises {
+	for _, e := range req.ExerciseConfigs {
 		ex, err := protobufToJson(e)
 		if err != nil {
 			return nil, err
@@ -234,13 +225,9 @@ func (a *Agent) AddExercisesToEnv(ctx context.Context, req *proto.ExerciseReques
 	env.M.Lock()
 	defer env.M.Unlock()
 
-	exerDbConfs, err := a.ExClient.GetExerciseByTags(ctx, &eproto.GetExerciseByTagsRequest{Tag: req.Exercises})
-	if err != nil {
-		return nil, fmt.Errorf("error getting exercises: %s", err)
-	}
 	// Unpack into exercise slice
 	var exerConfs []exercise.ExerciseConfig
-	for _, e := range exerDbConfs.Exercises {
+	for _, e := range req.ExerciseConfigs {
 		ex, err := protobufToJson(e)
 		if err != nil {
 			return nil, err