ultravioletrs · drasko · Aug 1, 2024 · Jul 21, 2024 · Jul 29, 2024 · Jul 29, 2024
diff --git a/cmd/manager/main.go b/cmd/manager/main.go
@@ -37,10 +37,11 @@ const (
 )
 
 type config struct {
-	LogLevel   string  `env:"MANAGER_LOG_LEVEL"     envDefault:"info"`
-	JaegerURL  url.URL `env:"COCOS_JAEGER_URL"      envDefault:"http://localhost:4318"`
-	TraceRatio float64 `env:"MG_JAEGER_TRACE_RATIO" envDefault:"1.0"`
-	InstanceID string  `env:"MANAGER_INSTANCE_ID"   envDefault:""`
+	LogLevel                 string  `env:"MANAGER_LOG_LEVEL"          envDefault:"info"`
+	JaegerURL                url.URL `env:"COCOS_JAEGER_URL"           envDefault:"http://localhost:4318"`
+	TraceRatio               float64 `env:"MG_JAEGER_TRACE_RATIO"      envDefault:"1.0"`
+	InstanceID               string  `env:"MANAGER_INSTANCE_ID"        envDefault:""`
+	BackendMeasurementBinary string  `env:"BACKEND_MEASUREMENT_BINARY" envDefault:"../../build"`
 }
 
 func main() {
@@ -103,7 +104,7 @@ func main() {
 	}
 
 	eventsChan := make(chan *pkgmanager.ClientStreamMessage)
-	svc := newService(logger, tracer, qemuCfg, eventsChan)
+	svc := newService(logger, tracer, qemuCfg, eventsChan, cfg.BackendMeasurementBinary)
 
 	mc := managerapi.NewClient(pc, svc, eventsChan)
 
@@ -120,8 +121,8 @@ func main() {
 	}
 }
 
-func newService(logger *slog.Logger, tracer trace.Tracer, qemuCfg qemu.Config, eventsChan chan *pkgmanager.ClientStreamMessage) manager.Service {
-	svc := manager.New(qemuCfg, logger, eventsChan, qemu.NewVM)
+func newService(logger *slog.Logger, tracer trace.Tracer, qemuCfg qemu.Config, eventsChan chan *pkgmanager.ClientStreamMessage, backendMeasurementPath string) manager.Service {
+	svc := manager.New(qemuCfg, backendMeasurementPath, logger, eventsChan, qemu.NewVM)
 	go svc.RetrieveAgentEventsLogs()
 	svc = api.LoggingMiddleware(svc, logger)
 	counter, latency := prometheus.MakeMetrics(svcName, "api")

diff --git a/manager/api/grpc/client.go b/manager/api/grpc/client.go
@@ -54,6 +54,18 @@ func (client ManagerClient) Process(ctx context.Context, cancel context.CancelFu
 				if err := client.svc.Stop(ctx, mes.StopComputation.ComputationId); err != nil {
 					return err
 				}
+			case *pkgmanager.ServerStreamMessage_BackendInfoReq:
+				res, err := client.svc.FetchBackendInfo()
+				if err != nil {
+					return err
+				}
+				info := &pkgmanager.ClientStreamMessage_BackendInfo{BackendInfo: &pkgmanager.BackendInfo{
+					Info: res,
+					Id:   mes.BackendInfoReq.Id,
+				}}
+				if err := client.stream.Send(&pkgmanager.ClientStreamMessage{Message: info}); err != nil {
+					return err
+				}
 			}
 		}
 	})

diff --git a/manager/api/logging.go b/manager/api/logging.go
@@ -57,3 +57,12 @@ func (lm *loggingMiddleware) Stop(ctx context.Context, computationID string) (er
 func (lm *loggingMiddleware) RetrieveAgentEventsLogs() {
 	lm.svc.RetrieveAgentEventsLogs()
 }
+
+func (lm *loggingMiddleware) FetchBackendInfo() ([]byte, error) {
+	defer func(begin time.Time) {
+		message := fmt.Sprintf("Method FetchBackendInfo took %s to complete", time.Since(begin))
+		lm.logger.Info(message)
+	}(time.Now())
+
+	return lm.svc.FetchBackendInfo()
+}
diff --git a/manager/api/metrics.go b/manager/api/metrics.go
@@ -54,3 +54,12 @@ func (ms *metricsMiddleware) Stop(ctx context.Context, computationID string) err
 func (ms *metricsMiddleware) RetrieveAgentEventsLogs() {
 	ms.svc.RetrieveAgentEventsLogs()
 }
+
+func (ms *metricsMiddleware) FetchBackendInfo() ([]byte, error) {
+	defer func(begin time.Time) {
+		ms.counter.With("method", "FetchBackendInfo").Add(1)
+		ms.latency.With("method", "FetchBackendInfo").Observe(time.Since(begin).Seconds())
+	}(time.Now())
+
+	return ms.svc.FetchBackendInfo()
+}
diff --git a/manager/manager.proto b/manager/manager.proto
@@ -26,6 +26,11 @@ message RunResponse{
   string computation_id = 2;
 }
 
+message BackendInfo{
+  bytes info = 1;
+  string id = 2;
+}
+
 message AgentEvent {
   string event_type = 1;
 	google.protobuf.Timestamp timestamp = 2;
@@ -47,6 +52,7 @@ message ClientStreamMessage {
     AgentLog agent_log = 1;
     AgentEvent agent_event = 2;
     RunResponse run_res = 3;
+    BackendInfo backendInfo = 4;
   }
 }
 
@@ -55,6 +61,7 @@ message ServerStreamMessage {
     ComputationRunReq runReq = 1;
     Terminate terminateReq = 2;
     StopComputation stopComputation = 3;
+    BackendInfoReq backendInfoReq = 4;
   }
 }
 
@@ -68,6 +75,10 @@ message ComputationRunReq {
   AgentConfig agent_config = 7;
 }
 
+message BackendInfoReq {
+  string id = 1;
+}
+
 message ResultConsumer {
   bytes userKey = 1;
 }

diff --git a/manager/manager_test.go b/manager/manager_test.go
@@ -17,12 +17,12 @@ import (
 )
 
 func TestNew(t *testing.T) {
-	qemuCfg := qemu.Config{}
+	cfg := qemu.Config{}
 	logger := slog.Default()
 	eventsChan := make(chan *manager.ClientStreamMessage)
 	vmf := new(mocks.Provider)
 
-	service := New(qemuCfg, logger, eventsChan, vmf.Execute)
+	service := New(cfg, "", logger, eventsChan, vmf.Execute)
 
 	assert.NotNil(t, service)
 	assert.IsType(t, &managerService{}, service)

diff --git a/manager/service.go b/manager/service.go
@@ -9,6 +9,8 @@ import (
 	"fmt"
 	"log/slog"
 	"net"
+	"os"
+	"os/exec"
 	"strconv"
 
 	"github.com/absmach/magistrala/pkg/errors"
@@ -53,28 +55,32 @@ type Service interface {
 	Stop(ctx context.Context, computationID string) error
 	// RetrieveAgentEventsLogs Retrieve and forward agent logs and events via vsock.
 	RetrieveAgentEventsLogs()
+	// FetchBackendInfo measures and fetches the backend information.
+	FetchBackendInfo() ([]byte, error)
 }
 
 type managerService struct {
-	qemuCfg    qemu.Config
-	logger     *slog.Logger
-	agents     map[int]string // agent map of vsock cid to computationID.
-	eventsChan chan *manager.ClientStreamMessage
-	vms        map[string]vm.VM
-	vmFactory  vm.Provider
+	qemuCfg                      qemu.Config
+	backendMeasurementBinaryPath string
+	logger                       *slog.Logger
+	agents                       map[int]string // agent map of vsock cid to computationID.
+	eventsChan                   chan *manager.ClientStreamMessage
+	vms                          map[string]vm.VM
+	vmFactory                    vm.Provider
 }
 
 var _ Service = (*managerService)(nil)
 
 // New instantiates the manager service implementation.
-func New(qemuCfg qemu.Config, logger *slog.Logger, eventsChan chan *manager.ClientStreamMessage, vmFactory vm.Provider) Service {
+func New(cfg qemu.Config, backendMeasurementBinPath string, logger *slog.Logger, eventsChan chan *manager.ClientStreamMessage, vmFactory vm.Provider) Service {
 	ms := &managerService{
-		qemuCfg:    qemuCfg,
-		logger:     logger,
-		agents:     make(map[int]string),
-		vms:        make(map[string]vm.VM),
-		eventsChan: eventsChan,
-		vmFactory:  vmFactory,
+		qemuCfg:                      cfg,
+		logger:                       logger,
+		agents:                       make(map[int]string),
+		vms:                          make(map[string]vm.VM),
+		eventsChan:                   eventsChan,
+		vmFactory:                    vmFactory,
+		backendMeasurementBinaryPath: backendMeasurementBinPath,
 	}
 	return ms
 }
@@ -160,6 +166,22 @@ func (ms *managerService) Stop(ctx context.Context, computationID string) error
 	return nil
 }
 
+func (ms *managerService) FetchBackendInfo() ([]byte, error) {
+	cmd := exec.Command("sudo", fmt.Sprintf("%s/backend_info", ms.backendMeasurementBinaryPath), "--policy", "1966081")
+
+	_, err := cmd.Output()
+	if err != nil {
+		return nil, err
+	}
+
+	f, err := os.ReadFile("./backend_info.json")
+	if err != nil {
+		return nil, err
+	}
+
+	return f, nil
+}
+
 func getFreePort() (int, error) {
 	listener, err := net.Listen("tcp", "")
 	if err != nil {

diff --git a/manager/tracing/tracing.go b/manager/tracing/tracing.go
@@ -39,3 +39,10 @@ func (tm *tracingMiddleware) Stop(ctx context.Context, computationID string) err
 func (tm *tracingMiddleware) RetrieveAgentEventsLogs() {
 	tm.svc.RetrieveAgentEventsLogs()
 }
+
+func (tm *tracingMiddleware) FetchBackendInfo() ([]byte, error) {
+	_, span := tm.tracer.Start(context.Background(), "fetch_backend_info")
+	defer span.End()
+
+	return tm.svc.FetchBackendInfo()
+}