NOISSUE - Fix agent startup issues (#605)

* Update attestationFromCert function to include ccPlatform parameter for enhanced attestation processing

Signed-off-by: Sammy Oina <sammyoina@gmail.com>

* chore: migrate dependencies from supermq to magistrala and update build configurations

Signed-off-by: Sammy Oina <sammyoina@gmail.com>

* chore: update project dependencies, repository source, and support TDX QuoteV5 attestation

Signed-off-by: Sammy Oina <sammyoina@gmail.com>

---------

Signed-off-by: Sammy Oina <sammyoina@gmail.com>

.github/dependabot.yaml

@@ -1,15 +1,5 @@
 version: 2
 updates:
-  - package-ecosystem: "cargo"
-    directory: "/scripts/attestation_policy"
-    schedule:
-      interval: "weekly"
-      day: "monday"
-    groups:
-      rs-dependencies:
-        patterns:
-          - "*"
-
   - package-ecosystem: "gomod"
     directories:
       - "/"

.github/workflows/checkproto.yaml

@@ -30,13 +30,13 @@ jobs:
       - name: Install Go
         uses: actions/setup-go@v5
         with:
-          go-version: 1.25.x
+          go-version: 1.26.x
 
       - name: Set up protoc
         run: |
-          PROTOC_VERSION=29.0
-          PROTOC_GEN_VERSION=v1.36.8
-          PROTOC_GRPC_VERSION=v1.5.1
+          PROTOC_VERSION=33.1
+          PROTOC_GEN_VERSION=v1.36.11
+          PROTOC_GRPC_VERSION=v1.6.0
 
           # Download and install protoc
           PROTOC_ZIP=protoc-$PROTOC_VERSION-linux-x86_64.zip

.github/workflows/hal.yml

@@ -42,7 +42,7 @@ jobs:
       - name: Install Go
         uses: actions/setup-go@v5
         with:
-          go-version: 1.25.x
+          go-version: 1.26.x
           cache-dependency-path: "go.sum"
 
       - name: Checkout cocos

.github/workflows/main.yaml

@@ -18,12 +18,12 @@ jobs:
       - name: Install Go
         uses: actions/setup-go@v5
         with:
-          go-version: 1.25.x
+          go-version: 1.26.x
 
       - name: golangci-lint
         uses: golangci/golangci-lint-action@v8
         with:
-          version: v2.4.0
+          version: v2.11.1
 
       - name: Build
         run: make
@@ -45,7 +45,7 @@ jobs:
       - name: Install Go
         uses: actions/setup-go@v5
         with:
-          go-version: 1.25.x
+          go-version: 1.26.x
 
       - name: Create coverage directory
         run: mkdir -p coverage
@@ -53,9 +53,9 @@ jobs:
       - name: Run tests for ${{ matrix.module }}
         run: |
           if [[ "${{ matrix.module }}" == "manager" ]]; then
-            sudo GOTOOLCHAIN=go1.25.0+auto go test -v --race -covermode=atomic -coverprofile coverage/${{ matrix.module }}.out ./${{ matrix.module }}/...
+            sudo GOTOOLCHAIN=go1.26.0+auto go test -v --race -covermode=atomic -coverprofile coverage/${{ matrix.module }}.out ./${{ matrix.module }}/...
           else
-            GOTOOLCHAIN=go1.25.0+auto go test -v --race -covermode=atomic -coverprofile coverage/${{ matrix.module }}.out ./${{ matrix.module }}/...
+            GOTOOLCHAIN=go1.26.0+auto go test -v --race -covermode=atomic -coverprofile coverage/${{ matrix.module }}.out ./${{ matrix.module }}/...
           fi
 
       - name: Upload coverage artifact

.github/workflows/rust.yaml

@@ -1,41 +0,0 @@
-name: Rust CI Pipeline
-
-on:
-  push:
-    branches:
-      - main
-    paths:
-      - "scripts/attestation_policy/**"
-      - ".github/workflows/rust.yaml"
-  pull_request:
-    branches:
-      - main
-    paths:
-      - "scripts/attestation_policy/**"
-      - ".github/workflows/rust.yaml"
-
-env:
-  CARGO_TERM_COLOR: always
-
-jobs:
-  rust-check:
-    runs-on: ubuntu-latest
-    defaults:
-      run:
-        working-directory: ./scripts/attestation_policy
-
-    steps:
-      - name: Checkout Code
-        uses: actions/checkout@v4
-
-      - name: Check cargo
-        run: cargo check --release --all-targets
-
-      - name: Check formatting
-        run: cargo fmt --all -- --check
-
-      - name: Run linter
-        run: cargo clippy -- -D warnings
-
-      - name: Build for all features
-        run: cargo build --release --all-features

.gitignore

@@ -19,9 +19,15 @@ target/
 # Remove Cargo.lock from gitignore if creating an executable, leave it for libraries
 # More information here https://doc.rust-lang.org/cargo/guide/cargo-toml-vs-cargo-lock.html
 Cargo.lock
+!tools/nvidia-attestation-helper/Cargo.lock
 
 # These are backup files generated by rustfmt
 **/*.rs.bk
 
 # MSVC Windows builds of rustc generate these, which store debugging information
 *.pdb
+
+*.enc
+*.key
+*.pub
+.codex

.golangci.yaml

@@ -70,10 +70,14 @@ linters:
       - legacy
       - std-error-handling
     rules:
+      - linters:
+          - errcheck
+        path: build/
       - linters:
           - makezero
         text: with non-zero initialized length
     paths:
+      - build
       - third_party$
       - builtin$
       - examples$

.mockery.yml

@@ -146,3 +146,17 @@ packages:
           dir: '{{.InterfaceDir}}/mocks'
           structname: '{{.InterfaceName}}'
           filename: "{{.InterfaceName | lower}}.go"
+  github.com/ultravioletrs/cocos/pkg/clients/grpc/runner:
+    interfaces:
+      Client:
+        config:
+          dir: '{{.InterfaceDir}}/mocks'
+          structname: '{{.InterfaceName}}'
+          filename: "{{.InterfaceName | lower}}.go"
+  github.com/ultravioletrs/cocos/internal/proto/attestation-agent:
+    interfaces:
+      AttestationAgentServiceClient:
+        config:
+          dir: '{{.InterfaceDir}}/mocks'
+          structname: '{{.InterfaceName}}'
+          filename: "{{.InterfaceName | lower}}.go"

Makefile

@@ -1,12 +1,24 @@
 BUILD_DIR = build
-SERVICES = manager agent cli
-ATTESTATION_POLICY = attestation_policy
+SERVICES = manager agent cli attestation-service log-forwarder computation-runner egress-proxy ingress-proxy
+NVIDIA_ATTESTATION_HELPER = nvidia-attestation-helper
+NVIDIA_ATTESTATION_HELPER_DIR = tools/$(NVIDIA_ATTESTATION_HELPER)
+NVIDIA_ATTESTATION_HELPER_MANIFEST = $(NVIDIA_ATTESTATION_HELPER_DIR)/Cargo.toml
+NVIDIA_ATTESTATION_HELPER_BINARY = $(BUILD_DIR)/$(NVIDIA_ATTESTATION_HELPER)
+NVIDIA_ATTESTATION_HELPER_LIB_DIR = $(BUILD_DIR)/lib
+NVAT_SDK_CPP_DIR ?= $(firstword $(wildcard $(HOME)/.cargo/git/checkouts/attestation-sdk-*/*/nv-attestation-sdk-cpp))
+NVAT_SDK_CPP_BUILD_DIR ?= $(NVAT_SDK_CPP_DIR)/build
+NVAT_SDK_HEADER ?= $(NVAT_SDK_CPP_BUILD_DIR)/include/nvat.h
+NVAT_SDK_SHARED_LIB ?= $(NVAT_SDK_CPP_BUILD_DIR)/libnvat.so.1
+NVAT_SYSTEM_HEADER ?= /usr/include/nvat.h
+CARGO ?= cargo
+CMAKE ?= cmake
 CGO_ENABLED ?= 0
 GOARCH ?= amd64
 VERSION ?= $(shell git describe --abbrev=0 --tags --always)
 COMMIT ?= $(shell git rev-parse HEAD)
 TIME ?= $(shell date +%F_%T)
 EMBED_ENABLED ?= 0
+NVAT_USE_SYSTEM_LIB ?=
 INSTALL_DIR ?= /usr/local/bin
 CONFIG_DIR ?= /etc/cocos
 SERVICE_NAME ?= cocos-manager
@@ -17,44 +29,75 @@ IGVM_BUILD_SCRIPT := ./scripts/igvmmeasure/igvm.sh
 define compile_service
 	CGO_ENABLED=$(CGO_ENABLED) GOOS=$(GOOS) GOARCH=$(GOARCH) GOARM=$(GOARM) \
 	go build -ldflags "-s -w \
-	-X 'github.com/absmach/supermq.BuildTime=$(TIME)' \
-	-X 'github.com/absmach/supermq.Version=$(VERSION)' \
-	-X 'github.com/absmach/supermq.Commit=$(COMMIT)'" \
+	-X 'github.com/absmach/magistrala.BuildTime=$(TIME)' \
+	-X 'github.com/absmach/magistrala.Version=$(VERSION)' \
+	-X 'github.com/absmach/magistrala.Commit=$(COMMIT)'" \
 	$(if $(filter 1,$(EMBED_ENABLED)),-tags "embed",) \
-	-o ${BUILD_DIR}/cocos-$(1) cmd/$(1)/main.go
+	-o ${BUILD_DIR}/cocos-$(1) ./cmd/$(1)
 endef
 
-.PHONY: all $(SERVICES) $(ATTESTATION_POLICY) install clean
+NVIDIA_ATTESTATION_HELPER_CARGO_ENV = $(if $(filter 1,$(NVAT_USE_SYSTEM_LIB)),NVAT_USE_SYSTEM_LIB=1,)
+NVIDIA_ATTESTATION_HELPER_RUSTFLAGS = $(strip $(RUSTFLAGS) $(if $(filter 1,$(NVAT_USE_SYSTEM_LIB)),,-C link-arg=-Wl,-rpath,$$ORIGIN/lib))
 
-all: $(SERVICES) $(ATTESTATION_POLICY)
+.PHONY: all $(SERVICES) $(NVIDIA_ATTESTATION_HELPER) nvidia-attestation-helper-prereqs install clean
 
-$(SERVICES): 
+all: $(SERVICES)
+
+$(BUILD_DIR):
+	mkdir -p $(BUILD_DIR)
+
+$(SERVICES): | $(BUILD_DIR)
 	$(call compile_service,$@)
 	@if [ "$@" = "cli" ] || [ "$@" = "manager" ]; then $(MAKE) build-igvm; fi
 
-$(ATTESTATION_POLICY):
-	$(MAKE) -C ./scripts/attestation_policy OUTPUT_DIR=../../$(BUILD_DIR)
+nvidia-attestation-helper-prereqs:
+ifeq ($(filter 1,$(NVAT_USE_SYSTEM_LIB)),1)
+	@test -f $(NVAT_SYSTEM_HEADER) || \
+		( echo "Missing $(NVAT_SYSTEM_HEADER). Install the NVAT development package or run without NVAT_USE_SYSTEM_LIB=1."; exit 1 )
+	@ldconfig -p | grep -q libnvat.so.1 || \
+		( echo "libnvat.so.1 not found in the dynamic linker cache. Install the NVAT runtime package or run without NVAT_USE_SYSTEM_LIB=1."; exit 1 )
+else
+	@if [ -z "$(NVAT_SDK_CPP_DIR)" ]; then \
+		echo "Unable to locate nv-attestation-sdk-cpp under $$HOME/.cargo/git/checkouts."; \
+		echo "Run 'cargo fetch --manifest-path $(NVIDIA_ATTESTATION_HELPER_MANIFEST)' first, or install NVAT and use 'make NVAT_USE_SYSTEM_LIB=1 $(NVIDIA_ATTESTATION_HELPER)'."; \
+		exit 1; \
+	fi
+	@if [ ! -f "$(NVAT_SDK_HEADER)" ] || [ ! -f "$(NVAT_SDK_SHARED_LIB)" ]; then \
+		$(CMAKE) -S $(NVAT_SDK_CPP_DIR) -B $(NVAT_SDK_CPP_BUILD_DIR) && \
+		$(CMAKE) --build $(NVAT_SDK_CPP_BUILD_DIR); \
+	fi
+endif
+
+$(NVIDIA_ATTESTATION_HELPER): nvidia-attestation-helper-prereqs | $(BUILD_DIR)
+	RUSTFLAGS='$(NVIDIA_ATTESTATION_HELPER_RUSTFLAGS)' $(NVIDIA_ATTESTATION_HELPER_CARGO_ENV) $(CARGO) build --manifest-path $(NVIDIA_ATTESTATION_HELPER_MANIFEST) --release
+	install -m 755 $(NVIDIA_ATTESTATION_HELPER_DIR)/target/release/$(NVIDIA_ATTESTATION_HELPER) $(NVIDIA_ATTESTATION_HELPER_BINARY)
+	@if [ "$(filter 1,$(NVAT_USE_SYSTEM_LIB))" != "1" ]; then \
+		install -d $(NVIDIA_ATTESTATION_HELPER_LIB_DIR); \
+		install -m 755 $(NVAT_SDK_SHARED_LIB) $(NVIDIA_ATTESTATION_HELPER_LIB_DIR)/libnvat.so.1; \
+	fi
 
 protoc:
 	protoc -I. --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative agent/agent.proto
 	protoc -I. --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative manager/manager.proto
 	protoc -I. --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative agent/events/events.proto
 	protoc -I. --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative agent/cvms/cvms.proto
+	protoc -I. --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative internal/proto/attestation/v1/attestation.proto
+	protoc -I. --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative internal/proto/attestation-agent/attestation-agent.proto
+	protoc -I. --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative agent/log/log.proto
+	protoc -I. --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative agent/runner/runner.proto
 
 mocks:
 	mockery --config ./.mockery.yml
 
-install: $(SERVICES) $(ATTESTATION_POLICY)
+install: $(SERVICES)
 	install -d $(INSTALL_DIR)
 	install $(BUILD_DIR)/cocos-cli $(INSTALL_DIR)/cocos-cli
 	install $(BUILD_DIR)/cocos-manager $(INSTALL_DIR)/cocos-manager
-	install $(BUILD_DIR)/attestation_policy $(INSTALL_DIR)/attestation_policy
 	install -d $(CONFIG_DIR)
 	install cocos-manager.env $(CONFIG_DIR)/cocos-manager.env
 
 clean:
 	rm -rf $(BUILD_DIR)
-	$(MAKE) -C ./scripts/attestation_policy OUTPUT_DIR=../../$(BUILD_DIR) clean
 
 run: install_service
 	sudo systemctl start $(SERVICE_NAME).service

README.md

@@ -77,4 +77,4 @@ Cocos AI is published under the permissive open-source [Apache-2.0](LICENSE) lic
 - [Confidential Computing Overview](https://confidentialcomputing.io/white-papers-reports/)
 - [Trusted Execution Environments (TEEs)](https://en.wikipedia.org/wiki/Trusted_execution_environment)
 
->This work has been partially supported by the [ELASTIC project](https://elasticproject.eu/), which received funding from the Smart Networks and Services Joint Undertaking (SNS JU) under the European Union’s Horizon Europe research and innovation programme under [Grant Agreement No. 101139067](https://cordis.europa.eu/project/id/101139067). Views and opinions expressed are however those of the author(s) only and do not necessarily reflect those of the European Union. Neither the European Union nor the granting authority can be held responsible for them.
+>This work has been partially supported by the [ELASTIC](https://elasticproject.eu/) and [CONFIDENTIAL6G](https://confidential6g.eu/), which received funding from the Smart Networks and Services Joint Undertaking (SNS JU) under the European Union’s Horizon Europe research and innovation programme under [Grant Agreement No. 101139067](https://cordis.europa.eu/project/id/101139067) and [Grant Agreement No. 101096435](https://cordis.europa.eu/project/id/101096435). Views and opinions expressed are however those of the author(s) only and do not necessarily reflect those of the European Union. Neither the European Union nor the granting authority can be held responsible for them.

agent/README.md

@@ -21,9 +21,32 @@ The service is configured using the environment variables from the following tab
 | AGENT_CVM_ID                   | Unique identifier for the CVM (Confidential Virtual Machine)                                                  | ""                                              |
 | AGENT_CERTS_TOKEN              | Authentication token for certificate service access                                                           | ""                                              |
 | AGENT_MAA_URL                  | Microsoft Azure Attestation service URL for Azure attestation                                                 | https://sharedeus2.eus2.attest.azure.net        |
+| AZURE_TDX_IMDS_URL             | Azure TDX quote endpoint used by direct Azure TDX attestation                                                 | http://169.254.169.254/acc/tdquote           |
+| AZURE_HCL_REFRESH_WAIT         | Wait after writing TDX report data to Azure HCL vTPM storage before reading the refreshed HCL report          | 3s                                              |
 | AGENT_OS_BUILD                 | Operating system build information for attestation                                                            | UVC                                             |
 | AGENT_OS_DISTRO                | Operating system distribution information for attestation                                                     | UVC                                             |
 | AGENT_OS_TYPE                  | Operating system type information for attestation                                                             | UVC                                             |
+| ATTESTATION_SERVICE_SOCKET     | Unix socket path for attestation service communication                                                       | /run/cocos/attestation.sock                     |
+| AGENT_ENABLE_ATLS              | Enable Attestation TLS for secure communication                                                              | true                                            |
+
+### Azure TDX Attestation
+
+When the agent runs on an Azure TDX CVM, Azure attestation uses the direct Azure TDX flow. The agent writes TDX report data to Azure HCL vTPM storage, reads the refreshed HCL report, requests a TD quote from Azure IMDS, and submits the quote plus HCL runtime data to Microsoft Azure Attestation. This path does not depend on Confidential Containers attestation-agent `GetEvidence` or KBS token retrieval.
+
+`AGENT_MAA_URL` selects the Microsoft Azure Attestation endpoint. `AZURE_TDX_IMDS_URL` can override the Azure IMDS TDX quote endpoint, and `AZURE_HCL_REFRESH_WAIT` controls the wait used to avoid reading a stale HCL report after report-data is written.
+
+### Remote Resource Download (Optional)
+
+The agent supports downloading encrypted algorithms and datasets from remote registries (S3, HTTP/HTTPS) and retrieving decryption keys from a Key Broker Service (KBS) via attestation.
+
+| Variable                       | Description                                                                                                   | Default                                         |
+| ------------------------------ | ------------------------------------------------------------------------------------------------------------- | ----------------------------------------------- |
+| AWS_REGION                     | AWS region for S3 access (required for S3 downloads)                                                          | \"\"                                              |
+| AWS_ACCESS_KEY_ID              | AWS access key ID for S3 authentication                                                                       | \"\"                                              |
+| AWS_SECRET_ACCESS_KEY          | AWS secret access key for S3 authentication                                                                   | \"\"                                              |
+| AWS_ENDPOINT_URL               | Custom S3 endpoint URL (for S3-compatible services like MinIO)                                                | \"\"                                              |
+
+**Note**: KBS URL is specified in the computation manifest, not as an environment variable. See [TESTING_REMOTE_RESOURCES.md](./TESTING_REMOTE_RESOURCES.md) for details on using remote resources.
 
 ## Deployment
 

agent/TESTING_REMOTE_RESOURCES.md

@@ -0,0 +1,468 @@
+# Testing Remote Resources with CoCo Key Provider
+
+This guide explains how to test Cocos with encrypted remote resources using the Confidential Containers Key Provider ecosystem.
+
+## Architecture Overview
+
+```
+┌────────────────────────────────────────────────────────────┐
+│                         CVM (Agent)                        │
+│                                                            │
+│  ┌──────────┐    ┌────────────────┐    ┌─────────────────┐ │
+│  │  Agent   │──▶│    Skopeo      │──▶│ CoCo Keyprovider│ │
+│  │          │    │  (ocicrypt)    │    │   (gRPC:50011)  │ │
+│  │          │    └───────┬────────┘    └────────┬────────┘ │
+│  │          │            │                      │          │
+│  │          │    ┌───────▼────────┐    ┌────────▼────────┐ │
+│  │          │──▶│  S3/HTTP       │    │ Attestation     │ │
+│  │          │    │  Downloader    │    │ Agent (50002)   │ │
+│  └────┬─────┘    └───────┬────────┘    └────────┬────────┘ │
+│       │                  │                      │          │
+│       └──────────────────┼──────────────────────┘          │
+└────────┬─────────────────┼──────────────────────┬──────────┘
+         │ (Resource)      │ (Resource)           │ (Attest)
+         ▼                 ▼                      ▼
+    OCI Registry       S3 / HTTP / GCS           KBS
+                                             (Key Broker)
+```
+
+## Prerequisites
+
+### 1. Install Skopeo (Host Machine)
+
+```bash
+# Ubuntu/Debian
+sudo apt-get install skopeo
+
+# macOS
+brew install skopeo
+
+# Or build from source
+git clone https://github.com/containers/skopeo
+cd skopeo
+make bin/skopeo
+sudo make install
+```
+
+### 2. Start KBS Server (Host Machine)
+
+```bash
+# Clone and build KBS
+git clone https://github.com/confidential-containers/trustee
+cd trustee/kbs
+# Patch Cargo.toml to disable SGX requirement (for testing only)
+sed -i 's/"all-verifier",//g' Cargo.toml
+
+make
+make cli
+
+# Generate admin keys
+openssl genpkey -algorithm ed25519 -out kbs-admin.key
+openssl pkey -in kbs-admin.key -pubout -out kbs-admin.pub
+
+# Create KBS configuration file
+cat > kbs-config.toml << 'EOF'
+[http_server]
+sockets = ["0.0.0.0:8080"]
+insecure_http = true
+
+[admin]
+type = "Simple"
+[[admin.personas]]
+id = "admin"
+public_key_path = "kbs-admin.pub"
+
+[attestation_service]
+type = "coco_as_builtin"
+work_dir = "kbs-data/as"
+
+[attestation_service.rvps_config]
+type = "BuiltIn"
+
+[attestation_service.rvps_config.storage]
+type = "LocalFs"
+file_path = "kbs-data/rvps-values"
+
+[[plugins]]
+name = "resource"
+type = "LocalFs"
+dir_path = "kbs-data/repository"
+EOF
+
+# Create configuration directories
+mkdir -p kbs-data/as kbs-data/rvps kbs-data/repository
+
+# Start KBS
+sudo ../target/release/kbs --config-file kbs-config.toml
+```
+
+KBS will listen on `http://localhost:8080`
+
+### 3. Setup Local OCI Registry (Optional)
+
+For testing, you can use a local registry:
+
+```bash
+docker run -d -p 5000:5000 --name registry registry:2
+```
+
+## Creating Encrypted Resources
+
+### Encrypt an Algorithm (Python Script)
+
+```bash
+# 1. Create a simple algorithm
+cat > lin_reg.py << 'EOF'
+import pandas as pd
+from sklearn.linear_model import LinearRegression
+import sys
+import os
+
+# Load dataset
+data = pd.read_csv(sys.argv[1])
+X = data[['feature1', 'feature2']]
+y = data['target']
+
+# Train model
+model = LinearRegression()
+model.fit(X, y)
+
+# Save results
+os.makedirs("results", exist_ok=True)
+with open("results/output.txt", "w") as f:
+    f.write(f"Coefficients: {model.coef_}\n")
+    f.write(f"Intercept: {model.intercept_}\n")
+
+print(f"Coefficients: {model.coef_}")
+print(f"Intercept: {model.intercept_}")
+EOF
+
+# 2. Create requirements.txt
+cat > requirements.txt << 'EOF'
+pandas
+scikit-learn
+EOF
+
+# 3. Create a Dockerfile
+cat > Dockerfile << 'EOF'
+FROM python:3.9-slim
+RUN pip install pandas scikit-learn
+COPY lin_reg.py /app/algorithm.py
+COPY requirements.txt /app/requirements.txt
+WORKDIR /app
+ENTRYPOINT ["python", "algorithm.py"]
+EOF
+
+# 4. Build the image
+docker build -t localhost:5000/lin-reg-algo:v1.0 .
+docker push localhost:5000/lin-reg-algo:v1.0
+
+# 5. Generate and store key
+openssl rand -out algo.key 32
+
+# 6. Store key in KBS using kbs-client
+../target/release/kbs-client --url http://localhost:8080 config \
+  --auth-private-key kbs-admin.key \
+  set-resource \
+  --path default/key/algo-key \
+  --resource-file algo.key
+
+# 7. Encrypt the image using Host Skopeo + Docker Keyprovider
+# Start Keyprovider in background
+docker run -d --rm --name keyprovider --network host \
+  -v "$PWD:/work" -w /work \
+  ghcr.io/confidential-containers/staged-images/coco-keyprovider:latest \
+  coco_keyprovider --socket 127.0.0.1:50000
+
+# Configure Ocicrypt to use local Keyprovider
+cat <<EOF > ocicrypt.conf
+{
+  "key-providers": {
+    "attestation-agent": {
+      "grpc": "127.0.0.1:50000"
+    }
+  }
+}
+EOF
+export OCICRYPT_KEYPROVIDER_CONFIG=$(pwd)/ocicrypt.conf
+
+# Encrypt Algo
+skopeo copy \
+  --src-tls-verify=false \
+  --dest-tls-verify=false \
+  --encryption-key "provider:attestation-agent:keypath=/work/algo.key::keyid=kbs:///default/key/algo-key::algorithm=A256GCM" \
+  docker://localhost:5000/lin-reg-algo:v1.0 \
+  docker://localhost:5000/encrypted-lin-reg:v1.0
+
+# Stop Keyprovider
+docker stop keyprovider
+```
+
+### Encrypt a Dataset (CSV in OCI Image)
+
+```bash
+# 1. Create dataset
+cat > iris.csv << 'EOF'
+feature1,feature2,target
+5.1,3.5,0
+4.9,3.0,0
+6.2,3.4,1
+5.9,3.0,1
+EOF
+
+# 2. Create Dockerfile for dataset
+cat > Dockerfile.dataset << 'EOF'
+FROM scratch
+COPY iris.csv /data/iris.csv
+EOF
+
+# 3. Build and push
+docker build -f Dockerfile.dataset -t localhost:5000/iris-dataset:v1.0 .
+docker push localhost:5000/iris-dataset:v1.0
+
+# 4. Generate and store key
+# 4. Generate and store key
+openssl rand -out dataset.key 32
+../target/release/kbs-client --url http://localhost:8080 config \
+  --auth-private-key kbs-admin.key \
+  set-resource \
+  --path default/key/dataset-key \
+  --resource-file dataset.key
+
+# 5. Encrypt dataset image using Host Skopeo + Docker Keyprovider
+# Start Keyprovider in background
+docker run -d --rm --name keyprovider --network host \
+  -v "$PWD:/work" -w /work \
+  ghcr.io/confidential-containers/staged-images/coco-keyprovider:latest \
+  coco_keyprovider --socket 127.0.0.1:50000
+
+# Configure Ocicrypt (if not already done)
+export OCICRYPT_KEYPROVIDER_CONFIG=$(pwd)/ocicrypt.conf
+
+# Encrypt Dataset
+skopeo copy \
+  --src-tls-verify=false \
+  --dest-tls-verify=false \
+  --encryption-key "provider:attestation-agent:keypath=/work/dataset.key::keyid=kbs:///default/key/dataset-key::algorithm=A256GCM" \
+  docker://localhost:5000/iris-dataset:v1.0 \
+  docker://localhost:5000/encrypted-iris:v1.0
+
+# Stop Keyprovider
+docker stop keyprovider
+```
+
+## Running a Computation
+
+### 1. Start Manager (Host)
+
+```bash
+cd /path/to/cocos-ai
+./build/cocos-manager
+```
+
+### 2. Start CVMS Test Server (Host)
+
+Get your host IP:
+```bash
+HOST_IP=$(ip -4 addr show | grep -oP '(?<=inet\s)\d+(\.\d+){3}' | grep -v 127.0.0.1 | head -n1)
+```
+
+Start CVMS server:
+```bash
+# Calculate SHA3-256 of decrypted files using cocos-cli or cvms-test
+# NOTE: We use the hash of the original plaintext files, as the Agent validates the decrypted content.
+# For single files, use the file hash. For directories, use the hash of the directory (which the tools zip deterministically).
+
+ALGO_HASH=$(./build/cocos-cli checksum lin_reg.py 2>&1 | awk '{print $NF}')
+
+DATASET_HASH=$(./build/cocos-cli checksum iris.csv 2>&1 | awk '{print $NF}')
+
+go build -o build/cvms-test ./test/cvms/main.go
+HOST=$HOST_IP PORT=7001 ./build/cvms-test \
+  -public-key-path ./public.pem \
+  -attested-tls-bool false \
+  -algo-type python \
+  -algo-source-url docker://$HOST_IP:5000/encrypted-lin-reg:v1.0 \
+  -algo-kbs-path default/key/algo-key \
+  -algo-kbs-url http://$HOST_IP:8080 \
+  -algo-hash $ALGO_HASH \
+  -algo-args datasets/dataset_0.csv \
+  -dataset-source-urls docker://$HOST_IP:5000/encrypted-iris:v1.0 \
+  -dataset-kbs-paths default/key/dataset-key \
+  -dataset-kbs-urls http://$HOST_IP:8080 \
+  -dataset-hash $DATASET_HASH
+```
+
+> [!NOTE]
+> You must specify the KBS URL for each encrypted resource using `-algo-kbs-url` and `-dataset-kbs-urls`. A global KBS is no longer supported.
+
+
+### 3. Create VM via CLI (Host)
+
+```bash
+export MANAGER_GRPC_URL=localhost:7002
+./build/cocos-cli create-vm \
+  --server-url $HOST_IP:7001 \
+  --log-level debug
+```
+
+The agent will:
+1. Receive computation manifest from CVMS
+2. Use Skopeo to download encrypted OCI images
+3. Skopeo invokes CoCo Keyprovider via ocicrypt
+4. CoCo Keyprovider requests decryption key from KBS
+5. Attestation Agent generates TEE evidence for KBS
+6. KBS validates evidence and returns decryption key
+7. Image layers are decrypted and extracted
+8. Computation executes with decrypted algorithm and dataset
+
+## Verifying the Setup
+
+### Check CoCo Keyprovider Status (Inside CVM)
+
+```bash
+# SSH into CVM or use console
+systemctl status coco-keyprovider
+journalctl -u coco-keyprovider -f
+```
+
+### Check Attestation Agent Status
+
+```bash
+systemctl status attestation-agent
+journalctl -u attestation-agent -f
+```
+
+### Test Skopeo Decryption Manually
+
+```bash
+# Inside CVM
+export OCICRYPT_KEYPROVIDER_CONFIG=/etc/ocicrypt_keyprovider.conf
+
+skopeo copy \
+  --src-tls-verify=false \
+  --dest-tls-verify=false \
+  --decryption-key provider:attestation-agent:cc_kbc::null \
+  docker://localhost:5000/encrypted-lin-reg:v1.0 \
+  oci:/tmp/decrypted-algo
+
+# Verify decryption
+skopeo inspect oci:/tmp/decrypted-algo | jq -r '.LayersData[].MIMEType'
+# Should show: application/vnd.oci.image.layer.v1.tar+gzip
+```
+
+## Computation Manifest Format
+
+The CVMS server sends this manifest to the agent:
+
+```json
+{
+  "computation_id": "1",
+  "algorithm": {
+    "type": "oci-image",
+    "uri": "docker://localhost:5000/encrypted-lin-reg:v1.0",
+    "encrypted": true,
+    "kbs_resource_path": "default/key/algo-key",
+    "kbs": {
+      "url": "http://192.168.100.15:8080",
+      "enabled": true
+    }
+  },
+  "datasets": [
+    {
+      "filename": "iris.csv",
+      "source": {
+        "type": "oci-image",
+        "url": "docker://localhost:5000/encrypted-iris:v1.0",
+        "encrypted": true,
+        "kbs_resource_path": "default/key/dataset-key"
+      },
+      "kbs": {
+        "url": "http://192.168.100.20:8080",
+        "enabled": true
+      }
+    }
+  ],
+  "kbs": {
+    "url": "http://192.168.100.15:8080",
+    "enabled": true
+  }
+}
+```
+
+## Troubleshooting
+
+### CoCo Keyprovider Not Starting
+
+```bash
+# Check logs
+journalctl -u coco-keyprovider -n 50
+
+# Verify socket is listening
+ss -tlnp | grep 50011
+
+# Check environment
+cat /etc/default/coco-keyprovider
+```
+
+### Skopeo Decryption Fails
+
+```bash
+# Verify ocicrypt config
+cat /etc/ocicrypt_keyprovider.conf
+
+# Test keyprovider connection
+grpcurl -plaintext 127.0.0.1:50011 list
+
+# Check KBS connectivity from CVM
+curl http://HOST_IP:8080/kbs/v0/auth
+```
+
+### KBS Returns 401
+
+```bash
+# Check KBS logs on host
+# Verify attestation evidence format
+# Ensure KBS is configured for sample attestation
+```
+
+## 4. Testing with Non-OCI Sources (S3, HTTP, GCS)
+
+The `cvms` test utility also supports testing remote encrypted resources hosted in more traditional environments like S3-compatible storage or simple web servers, bypassing the need for container registries and OCI images.
+
+### Supported Flags
+
+The following flags define how resources should be fetched:
+
+- `--algo-source-url`: The URL of the algorithm (e.g. `s3://bucket/algo.bin`, `https://server/algo.bin`)
+- `--algo-source-type`: The type of remote endpoint (`s3`, `gcs`, `https`, `http`). If omitted, it will automatically be inferred from the URL scheme.
+- `--algo-kbs-path`: The KBS path to retrieve the AES-256-GCM key from. If present, the agent will attempt decryption.
+- `--dataset-source-urls` and `--dataset-source-type`: Defines the locations and protocols for datasets.
+
+### Encryption Format for Non-OCI Sources
+
+Unlike OCI images where `ocicrypt` wraps the dataset, resources hosted on HTTP/S3 must be straightforwardly encrypted using **AES-256-GCM**.
+
+The expected format is exactly as produced by standard Go AES-GCM:
+`nonce (12 bytes) || ciphertext || tag`
+
+### Test Example
+
+If you had a Python script encrypted using a key hosted at KBS path `default/my-keys/python-script` and uploaded to `s3://my-secure-bucket/script.enc`, you could run:
+
+```bash
+cd test
+go run cvms/main.go --algo-source-url="s3://my-secure-bucket/script.enc" \
+                    --algo-source-type="s3" \
+                    --algo-kbs-path="default/my-keys/python-script" \
+                    --algo-type="python" \
+                    --public-key-path=./test-data/public-key.pem
+```
+
+The system will:
+1. Connect via `attestation-agent` to the KBS to retrieve the symmetric key
+2. Use Google Cloud Storage client library methods (support for generic S3 via environment variables is standard) to fetch the resource
+3. Decrypt using AES-256-GCM
+4. Run the code normally
+
+---

agent/agent.pb.go

@@ -3,8 +3,8 @@
 
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.36.8
-// 	protoc        v5.29.0
+// 	protoc-gen-go v1.36.11
+// 	protoc        v6.33.1
 // source: agent/agent.proto
 
 package agent

agent/agent_grpc.pb.go

@@ -3,8 +3,8 @@
 
 // Code generated by protoc-gen-go-grpc. DO NOT EDIT.
 // versions:
-// - protoc-gen-go-grpc v1.5.1
-// - protoc             v5.29.0
+// - protoc-gen-go-grpc v1.6.0
+// - protoc             v6.33.1
 // source: agent/agent.proto
 
 package agent
@@ -164,22 +164,22 @@ type AgentServiceServer interface {
 type UnimplementedAgentServiceServer struct{}
 
 func (UnimplementedAgentServiceServer) Algo(grpc.ClientStreamingServer[AlgoRequest, AlgoResponse]) error {
-	return status.Errorf(codes.Unimplemented, "method Algo not implemented")
+	return status.Error(codes.Unimplemented, "method Algo not implemented")
 }
 func (UnimplementedAgentServiceServer) Data(grpc.ClientStreamingServer[DataRequest, DataResponse]) error {
-	return status.Errorf(codes.Unimplemented, "method Data not implemented")
+	return status.Error(codes.Unimplemented, "method Data not implemented")
 }
 func (UnimplementedAgentServiceServer) Result(*ResultRequest, grpc.ServerStreamingServer[ResultResponse]) error {
-	return status.Errorf(codes.Unimplemented, "method Result not implemented")
+	return status.Error(codes.Unimplemented, "method Result not implemented")
 }
 func (UnimplementedAgentServiceServer) Attestation(*AttestationRequest, grpc.ServerStreamingServer[AttestationResponse]) error {
-	return status.Errorf(codes.Unimplemented, "method Attestation not implemented")
+	return status.Error(codes.Unimplemented, "method Attestation not implemented")
 }
 func (UnimplementedAgentServiceServer) IMAMeasurements(*IMAMeasurementsRequest, grpc.ServerStreamingServer[IMAMeasurementsResponse]) error {
-	return status.Errorf(codes.Unimplemented, "method IMAMeasurements not implemented")
+	return status.Error(codes.Unimplemented, "method IMAMeasurements not implemented")
 }
 func (UnimplementedAgentServiceServer) AzureAttestationToken(context.Context, *AttestationTokenRequest) (*AttestationTokenResponse, error) {
-	return nil, status.Errorf(codes.Unimplemented, "method AzureAttestationToken not implemented")
+	return nil, status.Error(codes.Unimplemented, "method AzureAttestationToken not implemented")
 }
 func (UnimplementedAgentServiceServer) mustEmbedUnimplementedAgentServiceServer() {}
 func (UnimplementedAgentServiceServer) testEmbeddedByValue()                      {}
@@ -192,7 +192,7 @@ type UnsafeAgentServiceServer interface {
 }
 
 func RegisterAgentServiceServer(s grpc.ServiceRegistrar, srv AgentServiceServer) {
-	// If the following call pancis, it indicates UnimplementedAgentServiceServer was
+	// If the following call panics, it indicates UnimplementedAgentServiceServer was
 	// embedded by pointer and is nil.  This will cause panics if an
 	// unimplemented method is ever invoked, so we test this at initialization
 	// time to prevent it from happening at runtime later due to I/O.

agent/algorithm/binary/binary.go

@@ -3,16 +3,21 @@
 package binary
 
 import (
+	"errors"
 	"fmt"
 	"io"
 	"log/slog"
+	"os"
 	"os/exec"
+	"sync"
 
 	"github.com/ultravioletrs/cocos/agent/algorithm"
 	"github.com/ultravioletrs/cocos/agent/algorithm/logging"
 	"github.com/ultravioletrs/cocos/agent/events"
 )
 
+var execCommand = exec.Command
+
 var _ algorithm.Algorithm = (*binary)(nil)
 
 type binary struct {
@@ -21,6 +26,7 @@ type binary struct {
 	stdout   io.Writer
 	args     []string
 	cmd      *exec.Cmd
+	mu       sync.Mutex
 }
 
 func NewAlgorithm(logger *slog.Logger, eventsSvc events.Service, algoFile string, args []string, cmpID string) algorithm.Algorithm {
@@ -33,13 +39,16 @@ func NewAlgorithm(logger *slog.Logger, eventsSvc events.Service, algoFile string
 }
 
 func (b *binary) Run() error {
-	b.cmd = exec.Command(b.algoFile, b.args...)
+	b.mu.Lock()
+	b.cmd = execCommand(b.algoFile, b.args...)
 	b.cmd.Stderr = b.stderr
 	b.cmd.Stdout = b.stdout
 
 	if err := b.cmd.Start(); err != nil {
+		b.mu.Unlock()
 		return fmt.Errorf("error starting algorithm: %v", err)
 	}
+	b.mu.Unlock()
 
 	if err := b.cmd.Wait(); err != nil {
 		return fmt.Errorf("algorithm execution error: %v", err)
@@ -49,11 +58,10 @@ func (b *binary) Run() error {
 }
 
 func (b *binary) Stop() error {
-	if b.cmd == nil {
-		return nil
-	}
+	b.mu.Lock()
+	defer b.mu.Unlock()
 
-	if b.cmd.ProcessState != nil && b.cmd.ProcessState.Exited() {
+	if b.cmd == nil {
 		return nil
 	}
 
@@ -61,7 +69,7 @@ func (b *binary) Stop() error {
 		return nil
 	}
 
-	if err := b.cmd.Process.Kill(); err != nil {
+	if err := b.cmd.Process.Kill(); err != nil && !errors.Is(err, os.ErrProcessDone) {
 		return fmt.Errorf("error stopping algorithm: %v", err)
 	}
 

agent/algorithm/binary/binary_test.go

@@ -4,10 +4,14 @@ package binary
 
 import (
 	"bytes"
+	"io"
 	"log/slog"
 	"os"
+	"os/exec"
 	"testing"
 
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
 	"github.com/ultravioletrs/cocos/agent/algorithm/logging"
 	"github.com/ultravioletrs/cocos/agent/events/mocks"
 )
@@ -73,6 +77,7 @@ func TestBinaryRun(t *testing.T) {
 		t.Run(tt.name, func(t *testing.T) {
 			logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
 			eventsSvc := new(mocks.Service)
+			eventsSvc.On("SendEvent", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return().Maybe()
 
 			b := NewAlgorithm(logger, eventsSvc, tt.algoFile, tt.args, "").(*binary)
 
@@ -98,3 +103,68 @@ func TestBinaryRun(t *testing.T) {
 		})
 	}
 }
+
+func TestStop(t *testing.T) {
+	t.Run("stop nil cmd", func(t *testing.T) {
+		b := &binary{}
+		err := b.Stop()
+		assert.NoError(t, err)
+	})
+
+	t.Run("stop with running process", func(t *testing.T) {
+		b := &binary{
+			algoFile: "sleep",
+			args:     []string{"10"},
+		}
+		if err := b.Run(); err != nil {
+			t.Fatalf("Failed to start command: %v", err)
+		}
+
+		err := b.Stop()
+		assert.NoError(t, err)
+
+		// Verify it actually stopped
+		_ = b.cmd.Wait()
+	})
+
+	t.Run("stop already exited", func(t *testing.T) {
+		b := &binary{
+			algoFile: "echo",
+			args:     []string{"test"},
+			stdout:   io.Discard,
+			stderr:   io.Discard,
+		}
+		if err := b.Run(); err != nil {
+			t.Fatal(err)
+		}
+		err := b.Stop()
+		assert.NoError(t, err)
+	})
+}
+
+func TestRunError(t *testing.T) {
+	// Mock execCommand to return an error on Start
+	oldExecCommand := execCommand
+	execCommand = mockExecCommandError
+	defer func() { execCommand = oldExecCommand }()
+
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventsSvc := new(mocks.Service)
+	eventsSvc.On("SendEvent", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return().Maybe()
+	b := NewAlgorithm(logger, eventsSvc, "test", nil, "").(*binary)
+
+	err := b.Run()
+	assert.Error(t, err)
+}
+
+func mockExecCommandError(command string, args ...string) *exec.Cmd {
+	// This will make Start() fail if we use a non-existent binary
+	return exec.Command("non_existent_binary_for_sure_12345")
+}
+
+func TestHelperProcess(t *testing.T) {
+	if os.Getenv("GO_WANT_HELPER_PROCESS") != "1" {
+		return
+	}
+	os.Exit(0)
+}

agent/algorithm/docker/docker.go

@@ -33,6 +33,7 @@ type docker struct {
 	logger   *slog.Logger
 	stderr   io.Writer
 	stdout   io.Writer
+	cmpID    string
 }
 
 func NewAlgorithm(logger *slog.Logger, eventsSvc events.Service, algoFile, cmpID string) algorithm.Algorithm {
@@ -41,6 +42,7 @@ func NewAlgorithm(logger *slog.Logger, eventsSvc events.Service, algoFile, cmpID
 		logger:   logger,
 		stderr:   &logging.Stderr{Logger: logger, EventSvc: eventsSvc, CmpID: cmpID},
 		stdout:   &logging.Stdout{Logger: logger},
+		cmpID:    cmpID,
 	}
 
 	return d
@@ -107,7 +109,7 @@ func (d *docker) Run() error {
 				Target: resultsMountPath,
 			},
 		},
-	}, nil, nil, containerName)
+	}, nil, nil, fmt.Sprintf("%s-%s", containerName, d.cmpID))
 	if err != nil {
 		return fmt.Errorf("could not create a Docker container: %v", err)
 	}

agent/algorithm/logging/logging_test.go

@@ -6,7 +6,7 @@ import (
 	"strings"
 	"testing"
 
-	mglog "github.com/absmach/supermq/logger"
+	mglog "github.com/absmach/magistrala/logger"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 	"github.com/ultravioletrs/cocos/agent/events/mocks"

agent/algorithm/python/python.go

@@ -4,12 +4,14 @@ package python
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"io"
 	"log/slog"
 	"os"
 	"os/exec"
 	"path/filepath"
+	"sync"
 
 	"github.com/ultravioletrs/cocos/agent/algorithm"
 	"github.com/ultravioletrs/cocos/agent/algorithm/logging"
@@ -40,6 +42,7 @@ type python struct {
 	requirementsFile string
 	args             []string
 	cmd              *exec.Cmd
+	mu               sync.Mutex
 }
 
 func NewAlgorithm(logger *slog.Logger, eventsSvc events.Service, runtime, requirementsFile, algoFile string, args []string, cmpID string) algorithm.Algorithm {
@@ -60,6 +63,12 @@ func NewAlgorithm(logger *slog.Logger, eventsSvc events.Service, runtime, requir
 
 func (p *python) Run() error {
 	venvPath := "venv"
+	defer func() {
+		if err := os.RemoveAll(venvPath); err != nil {
+			_, _ = p.stderr.Write([]byte(fmt.Sprintf("error removing virtual environment: %v\n", err)))
+		}
+	}()
+
 	createVenvCmd := exec.Command(p.runtime, "-m", "venv", venvPath)
 	createVenvCmd.Stderr = p.stderr
 	createVenvCmd.Stdout = p.stdout
@@ -69,11 +78,11 @@ func (p *python) Run() error {
 
 	pythonPath := filepath.Join(venvPath, "bin", "python")
 
-	updatePipCmd := exec.Command(pythonPath, "-m", "pip", "install", "--upgrade", "pip")
+	updatePipCmd := exec.Command(pythonPath, "-m", "pip", "install", "--upgrade", "pip", "setuptools", "wheel")
 	updatePipCmd.Stderr = p.stderr
 	updatePipCmd.Stdout = p.stdout
 	if err := updatePipCmd.Run(); err != nil {
-		return fmt.Errorf("error updating pip: %v", err)
+		return fmt.Errorf("error updating pip, setuptools and wheel: %v", err)
 	}
 
 	if p.requirementsFile != "" {
@@ -86,31 +95,29 @@ func (p *python) Run() error {
 	}
 
 	args := append([]string{p.algoFile}, p.args...)
+	p.mu.Lock()
 	p.cmd = exec.Command(pythonPath, args...)
 	p.cmd.Stderr = p.stderr
 	p.cmd.Stdout = p.stdout
 
 	if err := p.cmd.Start(); err != nil {
+		p.mu.Unlock()
 		return fmt.Errorf("error starting algorithm: %v", err)
 	}
+	p.mu.Unlock()
 
 	if err := p.cmd.Wait(); err != nil {
 		return fmt.Errorf("algorithm execution error: %v", err)
 	}
 
-	if err := os.RemoveAll(venvPath); err != nil {
-		return fmt.Errorf("error removing virtual environment: %v", err)
-	}
-
 	return nil
 }
 
 func (p *python) Stop() error {
-	if p.cmd == nil {
-		return nil
-	}
+	p.mu.Lock()
+	defer p.mu.Unlock()
 
-	if p.cmd.ProcessState != nil && p.cmd.ProcessState.Exited() {
+	if p.cmd == nil {
 		return nil
 	}
 
@@ -118,7 +125,7 @@ func (p *python) Stop() error {
 		return nil
 	}
 
-	if err := p.cmd.Process.Kill(); err != nil {
+	if err := p.cmd.Process.Kill(); err != nil && !errors.Is(err, os.ErrProcessDone) {
 		return fmt.Errorf("error stopping algorithm: %v", err)
 	}
 

agent/algorithm/python/python_test.go

@@ -8,10 +8,14 @@ import (
 	"io"
 	"log/slog"
 	"os"
+	"os/exec"
 	"path/filepath"
 	"strings"
 	"testing"
 
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+	"github.com/stretchr/testify/require"
 	"github.com/ultravioletrs/cocos/agent/algorithm/logging"
 	"github.com/ultravioletrs/cocos/agent/events/mocks"
 	"google.golang.org/grpc/metadata"
@@ -85,6 +89,7 @@ func TestRun(t *testing.T) {
 	}
 
 	eventsSvc := new(mocks.Service)
+	eventsSvc.EXPECT().SendEvent(mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return().Maybe()
 
 	var stdout, stderr bytes.Buffer
 
@@ -126,6 +131,7 @@ func TestRunWithRequirements(t *testing.T) {
 	}
 
 	eventsSvc := new(mocks.Service)
+	eventsSvc.EXPECT().SendEvent(mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return().Maybe()
 
 	var stdout, stderr bytes.Buffer
 
@@ -146,3 +152,91 @@ func TestRunWithRequirements(t *testing.T) {
 		t.Errorf("Expected output to contain requests version 2.26.0, got %q", stdout.String())
 	}
 }
+
+func TestStop(t *testing.T) {
+	t.Run("stop nil cmd", func(t *testing.T) {
+		p := &python{}
+		err := p.Stop()
+		if err != nil {
+			t.Errorf("Expected nil error, got %v", err)
+		}
+	})
+
+	t.Run("stop with running process", func(t *testing.T) {
+		p := &python{
+			stderr: io.Discard,
+			stdout: io.Discard,
+		}
+
+		p.cmd = exec.Command("python3", "-c", "import time; time.sleep(10)")
+		if err := p.cmd.Start(); err != nil {
+			t.Fatalf("Failed to start command: %v", err)
+		}
+
+		err := p.Stop()
+		if err != nil {
+			t.Errorf("Expected nil error, got %v", err)
+		}
+
+		// Verify it actually stopped
+		_ = p.cmd.Wait()
+	})
+
+	t.Run("stop already exited", func(t *testing.T) {
+		p := &python{}
+		p.cmd = exec.Command("python3", "-c", "print(1)")
+		if err := p.cmd.Run(); err != nil {
+			t.Fatal(err)
+		}
+		err := p.Stop()
+		if err != nil {
+			t.Errorf("Expected nil error, got %v", err)
+		}
+	})
+}
+
+func TestRun_Errors(t *testing.T) {
+	t.Run("invalid runtime error", func(t *testing.T) {
+		algo := &python{
+			algoFile: "algo.py",
+			runtime:  "non-existent-python",
+			stderr:   io.Discard,
+			stdout:   io.Discard,
+		}
+		err := algo.Run()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "error creating virtual environment")
+	})
+
+	t.Run("pip install failure", func(t *testing.T) {
+		tmpDir, err := os.MkdirTemp("", "python-err-test")
+		require.NoError(t, err)
+		defer os.RemoveAll(tmpDir)
+
+		scriptPath := filepath.Join(tmpDir, "test.py")
+		require.NoError(t, os.WriteFile(scriptPath, []byte("print(1)"), 0o644))
+
+		reqPath := filepath.Join(tmpDir, "requirements.txt")
+		require.NoError(t, os.WriteFile(reqPath, []byte("non-existent-package==9.9.9"), 0o644))
+
+		algo := &python{
+			algoFile:         scriptPath,
+			requirementsFile: reqPath,
+			runtime:          "python3",
+			stderr:           io.Discard,
+			stdout:           io.Discard,
+		}
+		err = algo.Run()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "error installing requirements")
+	})
+}
+
+func TestNewAlgorithmEmptyRuntime(t *testing.T) {
+	eventsSvc := new(mocks.Service)
+	algo := NewAlgorithm(slog.Default(), eventsSvc, "", "req.txt", "algo.py", nil, "")
+	p := algo.(*python)
+	if p.runtime != PyRuntime {
+		t.Errorf("Expected default runtime %s, got %s", PyRuntime, p.runtime)
+	}
+}

agent/algorithm/wasm/wasm.go

@@ -3,16 +3,21 @@
 package wasm
 
 import (
+	"errors"
 	"fmt"
 	"io"
 	"log/slog"
+	"os"
 	"os/exec"
+	"sync"
 
 	"github.com/ultravioletrs/cocos/agent/algorithm"
 	"github.com/ultravioletrs/cocos/agent/algorithm/logging"
 	"github.com/ultravioletrs/cocos/agent/events"
 )
 
+var execCommand = exec.Command
+
 const wasmRuntime = "wasmedge"
 
 var mapDirOption = []string{"--dir", ".:" + algorithm.ResultsDir}
@@ -25,6 +30,7 @@ type wasm struct {
 	stdout   io.Writer
 	args     []string
 	cmd      *exec.Cmd
+	mu       sync.Mutex
 }
 
 func NewAlgorithm(logger *slog.Logger, eventsSvc events.Service, args []string, algoFile, cmpID string) algorithm.Algorithm {
@@ -39,13 +45,16 @@ func NewAlgorithm(logger *slog.Logger, eventsSvc events.Service, args []string,
 func (w *wasm) Run() error {
 	args := append(mapDirOption, w.algoFile)
 	args = append(args, w.args...)
-	w.cmd = exec.Command(wasmRuntime, args...)
+	w.mu.Lock()
+	w.cmd = execCommand(wasmRuntime, args...)
 	w.cmd.Stderr = w.stderr
 	w.cmd.Stdout = w.stdout
 
 	if err := w.cmd.Start(); err != nil {
+		w.mu.Unlock()
 		return fmt.Errorf("error starting algorithm: %v", err)
 	}
+	w.mu.Unlock()
 
 	if err := w.cmd.Wait(); err != nil {
 		return fmt.Errorf("algorithm execution error: %v", err)
@@ -55,11 +64,10 @@ func (w *wasm) Run() error {
 }
 
 func (w *wasm) Stop() error {
-	if w.cmd == nil {
-		return nil
-	}
+	w.mu.Lock()
+	defer w.mu.Unlock()
 
-	if w.cmd.ProcessState != nil && w.cmd.ProcessState.Exited() {
+	if w.cmd == nil {
 		return nil
 	}
 
@@ -67,7 +75,7 @@ func (w *wasm) Stop() error {
 		return nil
 	}
 
-	if err := w.cmd.Process.Kill(); err != nil {
+	if err := w.cmd.Process.Kill(); err != nil && !errors.Is(err, os.ErrProcessDone) {
 		return fmt.Errorf("error stopping algorithm: %v", err)
 	}
 

agent/algorithm/wasm/wasm_test.go

@@ -7,15 +7,18 @@ import (
 	"os"
 	"os/exec"
 	"testing"
+	"time"
 
 	"github.com/ultravioletrs/cocos/agent/algorithm/logging"
 	"github.com/ultravioletrs/cocos/agent/events/mocks"
 )
 
+const testWasm = "test.wasm"
+
 func TestNewAlgorithm(t *testing.T) {
 	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
 	eventsSvc := new(mocks.Service)
-	algoFile := "test.wasm"
+	algoFile := testWasm
 	args := []string{"arg1", "arg2"}
 
 	algo := NewAlgorithm(logger, eventsSvc, args, algoFile, "")
@@ -49,14 +52,18 @@ func TestRunError(t *testing.T) {
 	execCommand = mockExecCommandError
 	defer func() { execCommand = exec.Command }()
 
-	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
-	eventsSvc := new(mocks.Service)
-	algoFile := "test.wasm"
+	algoFile := testWasm
 	args := []string{"arg1", "arg2"}
 
-	w := NewAlgorithm(logger, eventsSvc, args, algoFile, "").(*wasm)
+	w := &wasm{
+		algoFile: algoFile,
+		args:     args,
+		stderr:   os.Stderr, // Use real stderr or io.Discard
+		stdout:   os.Stdout,
+	}
 
 	err := w.Run()
+
 	if err == nil {
 		t.Errorf("Run() should have returned an error")
 	}
@@ -76,14 +83,97 @@ func mockExecCommandError(command string, args ...string) *exec.Cmd {
 	return cmd
 }
 
+func TestStop(t *testing.T) {
+	t.Run("stop nil cmd", func(t *testing.T) {
+		w := &wasm{}
+		err := w.Stop()
+		if err != nil {
+			t.Errorf("Expected nil error, got %v", err)
+		}
+	})
+
+	t.Run("stop with running process", func(t *testing.T) {
+		oldExecCommand := execCommand
+		execCommand = mockExecCommand
+		defer func() { execCommand = oldExecCommand }()
+
+		w := &wasm{
+			algoFile: testWasm,
+			stdout:   os.Stdout,
+			stderr:   os.Stderr,
+		}
+
+		// We need to simulate a running process.
+		// mockExecCommand returns a command that runs TestHelperProcess.
+		// If we don't call Wait(), it keeps running? No, TestHelperProcess exits immediately.
+		// Let's modify TestHelperProcess to sleep if an env var is set.
+
+		w.cmd = mockExecCommand("sleep", "10")
+		w.cmd.Env = append(w.cmd.Env, "GO_WANT_HELPER_PROCESS_SLEEP=1")
+		if err := w.cmd.Start(); err != nil {
+			t.Fatalf("Failed to start command: %v", err)
+		}
+
+		err := w.Stop()
+		if err != nil {
+			t.Errorf("Expected nil error, got %v", err)
+		}
+		_ = w.cmd.Wait()
+	})
+}
+
+func TestStopAlreadyExited(t *testing.T) {
+	oldExecCommand := execCommand
+	execCommand = mockExecCommand
+	defer func() { execCommand = oldExecCommand }()
+
+	w := &wasm{
+		algoFile: testWasm,
+		stdout:   os.Stdout,
+		stderr:   os.Stderr,
+	}
+
+	w.cmd = mockExecCommand("true")
+	if err := w.cmd.Run(); err != nil {
+		t.Fatalf("Failed to run command: %v", err)
+	}
+
+	err := w.Stop()
+	if err != nil {
+		t.Errorf("Expected nil error, got %v", err)
+	}
+}
+
+func TestRunSuccess(t *testing.T) {
+	oldExecCommand := execCommand
+	execCommand = mockExecCommand
+	defer func() { execCommand = oldExecCommand }()
+
+	algoFile := testWasm
+	args := []string{"arg1", "arg2"}
+
+	w := &wasm{
+		algoFile: algoFile,
+		args:     args,
+		stderr:   os.Stderr,
+		stdout:   os.Stdout,
+	}
+
+	err := w.Run()
+	if err != nil {
+		t.Errorf("Run() returned unexpected error: %v", err)
+	}
+}
+
 func TestHelperProcess(t *testing.T) {
 	if os.Getenv("GO_WANT_HELPER_PROCESS") != "1" {
 		return
 	}
+	if os.Getenv("GO_WANT_HELPER_PROCESS_SLEEP") == "1" {
+		time.Sleep(10 * time.Second)
+	}
 	if os.Getenv("GO_WANT_HELPER_PROCESS_ERROR") == "1" {
 		os.Exit(1)
 	}
 	os.Exit(0)
 }
-
-var execCommand = exec.Command

agent/api/grpc/requests.go

@@ -6,7 +6,6 @@ import (
 	"errors"
 
 	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"github.com/ultravioletrs/cocos/pkg/attestation/quoteprovider"
 	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 )
 
@@ -42,7 +41,7 @@ func (req resultReq) validate() error {
 }
 
 type attestationReq struct {
-	TeeNonce  [quoteprovider.Nonce]byte
+	TeeNonce  [vtpm.SEVNonce]byte
 	VtpmNonce [vtpm.Nonce]byte
 	AttType   attestation.PlatformType
 }
@@ -61,7 +60,7 @@ func (req azureAttestationTokenReq) validate() error {
 
 func validateAttestationType(attType attestation.PlatformType) error {
 	switch attType {
-	case attestation.SNP, attestation.VTPM, attestation.SNPvTPM, attestation.TDX:
+	case attestation.SNP, attestation.VTPM, attestation.SNPvTPM, attestation.Azure, attestation.TDX:
 		return nil
 	default:
 		return errors.New("invalid attestation type")

agent/api/grpc/server.go

@@ -14,7 +14,6 @@ import (
 	"github.com/go-kit/kit/transport/grpc"
 	"github.com/ultravioletrs/cocos/agent"
 	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"github.com/ultravioletrs/cocos/pkg/attestation/quoteprovider"
 	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"
@@ -134,7 +133,7 @@ func encodeResultResponse(_ context.Context, response any) (any, error) {
 func validateNonce(nonce []byte, maxLen int, target any) error {
 	if len(nonce) > maxLen {
 		switch maxLen {
-		case quoteprovider.Nonce:
+		case vtpm.SEVNonce:
 			return ErrTEENonceLength
 		case vtpm.Nonce:
 			return ErrVTPMNonceLength
@@ -144,7 +143,7 @@ func validateNonce(nonce []byte, maxLen int, target any) error {
 	}
 
 	switch t := target.(type) {
-	case *[quoteprovider.Nonce]byte:
+	case *[vtpm.SEVNonce]byte:
 		copy(t[:], nonce)
 	case *[vtpm.Nonce]byte:
 		copy(t[:], nonce)
@@ -156,10 +155,10 @@ func validateNonce(nonce []byte, maxLen int, target any) error {
 
 func decodeAttestationRequest(_ context.Context, grpcReq any) (any, error) {
 	req := grpcReq.(*agent.AttestationRequest)
-	var reportData [quoteprovider.Nonce]byte
+	var reportData [vtpm.SEVNonce]byte
 	var nonce [vtpm.Nonce]byte
 
-	if err := validateNonce(req.TeeNonce, quoteprovider.Nonce, &reportData); err != nil {
+	if err := validateNonce(req.TeeNonce, vtpm.SEVNonce, &reportData); err != nil {
 		return nil, err
 	}
 

agent/api/grpc/server_test.go

@@ -12,7 +12,6 @@ import (
 	"github.com/ultravioletrs/cocos/agent"
 	"github.com/ultravioletrs/cocos/agent/mocks"
 	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"github.com/ultravioletrs/cocos/pkg/attestation/quoteprovider"
 	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/metadata"
@@ -229,7 +228,7 @@ func TestAttestation(t *testing.T) {
 		return len(resp.File) > 0
 	})).Return(nil).Once()
 
-	reportData := [quoteprovider.Nonce]byte{}
+	reportData := [vtpm.SEVNonce]byte{}
 	vtpmNonce := [vtpm.Nonce]byte{}
 	attestationType := attestation.SNP
 	mockService.On("Attestation", mock.Anything, reportData, vtpmNonce, attestationType).Return(attestationData, nil)
@@ -298,8 +297,8 @@ func TestValidateNonce(t *testing.T) {
 	}{
 		{
 			name:        "valid TEE nonce",
-			nonce:       make([]byte, quoteprovider.Nonce),
-			maxLen:      quoteprovider.Nonce,
+			nonce:       make([]byte, vtpm.SEVNonce),
+			maxLen:      vtpm.SEVNonce,
 			shouldError: false,
 		},
 		{
@@ -310,8 +309,8 @@ func TestValidateNonce(t *testing.T) {
 		},
 		{
 			name:        "TEE nonce too long",
-			nonce:       make([]byte, quoteprovider.Nonce+1),
-			maxLen:      quoteprovider.Nonce,
+			nonce:       make([]byte, vtpm.SEVNonce+1),
+			maxLen:      vtpm.SEVNonce,
 			shouldError: true,
 			expectedErr: ErrTEENonceLength,
 		},
@@ -326,8 +325,8 @@ func TestValidateNonce(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			if tt.maxLen == quoteprovider.Nonce {
-				var target [quoteprovider.Nonce]byte
+			if tt.maxLen == vtpm.SEVNonce {
+				var target [vtpm.SEVNonce]byte
 				err := validateNonce(tt.nonce, tt.maxLen, &target)
 				if tt.shouldError {
 					assert.Error(t, err)
@@ -388,7 +387,7 @@ func TestEncodeResultResponse(t *testing.T) {
 }
 
 func TestDecodeAttestationRequest(t *testing.T) {
-	teeNonce := make([]byte, quoteprovider.Nonce)
+	teeNonce := make([]byte, vtpm.SEVNonce)
 	vtpmNonce := make([]byte, vtpm.Nonce)
 
 	req := &agent.AttestationRequest{
@@ -406,7 +405,7 @@ func TestDecodeAttestationRequest(t *testing.T) {
 
 func TestDecodeAttestationRequestWithInvalidNonce(t *testing.T) {
 	// Test with TEE nonce too long
-	teeNonce := make([]byte, quoteprovider.Nonce+1)
+	teeNonce := make([]byte, vtpm.SEVNonce+1)
 	req := &agent.AttestationRequest{TeeNonce: teeNonce}
 
 	_, err := decodeAttestationRequest(context.Background(), req)

agent/api/logging.go

@@ -2,7 +2,6 @@
 // SPDX-License-Identifier: Apache-2.0
 
 //go:build !test
-// +build !test
 
 package api
 
@@ -14,7 +13,6 @@ import (
 
 	"github.com/ultravioletrs/cocos/agent"
 	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"github.com/ultravioletrs/cocos/pkg/attestation/quoteprovider"
 	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 )
 
@@ -106,7 +104,7 @@ func (lm *loggingMiddleware) Result(ctx context.Context) (response []byte, err e
 	return lm.svc.Result(ctx)
 }
 
-func (lm *loggingMiddleware) Attestation(ctx context.Context, reportData [quoteprovider.Nonce]byte, nonce [vtpm.Nonce]byte, attType attestation.PlatformType) (response []byte, err error) {
+func (lm *loggingMiddleware) Attestation(ctx context.Context, reportData [vtpm.SEVNonce]byte, nonce [vtpm.Nonce]byte, attType attestation.PlatformType) (response []byte, err error) {
 	defer func(begin time.Time) {
 		message := fmt.Sprintf("Method Attestation took %s to complete", time.Since(begin))
 		if err != nil {

agent/api/metrics.go

@@ -2,7 +2,6 @@
 // SPDX-License-Identifier: Apache-2.0
 
 //go:build !test
-// +build !test
 
 package api
 
@@ -13,7 +12,6 @@ import (
 	"github.com/go-kit/kit/metrics"
 	"github.com/ultravioletrs/cocos/agent"
 	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"github.com/ultravioletrs/cocos/pkg/attestation/quoteprovider"
 	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 )
 
@@ -92,7 +90,7 @@ func (ms *metricsMiddleware) Result(ctx context.Context) ([]byte, error) {
 	return ms.svc.Result(ctx)
 }
 
-func (ms *metricsMiddleware) Attestation(ctx context.Context, reportData [quoteprovider.Nonce]byte, nonce [vtpm.Nonce]byte, attType attestation.PlatformType) ([]byte, error) {
+func (ms *metricsMiddleware) Attestation(ctx context.Context, reportData [vtpm.SEVNonce]byte, nonce [vtpm.Nonce]byte, attType attestation.PlatformType) ([]byte, error) {
 	defer func(begin time.Time) {
 		ms.counter.With("method", "attestation").Add(1)
 		ms.latency.With("method", "attestation").Observe(time.Since(begin).Seconds())

agent/auth/auth.go

@@ -13,7 +13,7 @@ import (
 	"crypto/x509"
 	"encoding/base64"
 
-	"github.com/absmach/supermq/pkg/errors"
+	"github.com/absmach/magistrala/pkg/errors"
 	"github.com/ultravioletrs/cocos/agent"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"

agent/auth/auth_test.go

@@ -15,7 +15,7 @@ import (
 	"encoding/base64"
 	"testing"
 
-	"github.com/absmach/supermq/pkg/errors"
+	"github.com/absmach/magistrala/pkg/errors"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"github.com/ultravioletrs/cocos/agent"
@@ -44,7 +44,7 @@ func TestAuthenticateUser(t *testing.T) {
 	manifest := agent.Computation{
 		ResultConsumers: []agent.ResultConsumer{{UserKey: resultConsumerPubKey}},
 		Datasets:        []agent.Dataset{{UserKey: dataProviderPubKey}},
-		Algorithm:       agent.Algorithm{UserKey: algorithmProviderPubKey},
+		Algorithm:       &agent.Algorithm{UserKey: algorithmProviderPubKey},
 	}
 
 	auth, err := New(manifest)

agent/computations.go

@@ -13,7 +13,6 @@ import (
 var _ fmt.Stringer = (*Datasets)(nil)
 
 type AgentConfig struct {
-	Port         string `json:"port,omitempty"`
 	CertFile     string `json:"cert_file,omitempty"`
 	KeyFile      string `json:"server_key,omitempty"`
 	ServerCAFile string `json:"server_ca_file,omitempty"`
@@ -21,12 +20,39 @@ type AgentConfig struct {
 	AttestedTls  bool   `json:"attested_tls,omitempty"`
 }
 
+// ResourceSource specifies the location of a remote encrypted resource.
+type ResourceSource struct {
+	// Type is the type of resource source.
+	// Supported values: "oci-image", "s3", "gcs", "https", "http"
+	Type string `json:"type,omitempty"`
+	// URL is the location of the resource.
+	// Examples:
+	//   - OCI:   "docker://registry/repo:tag"
+	//   - S3:    "s3://bucket/key"
+	//   - GCS:   "gs://bucket/key"
+	//   - HTTPS: "https://host/path/to/file"
+	//   - HTTP:  "http://host/path/to/file"
+	URL string `json:"url,omitempty"`
+	// KBSResourcePath is the path to the decryption key in KBS (e.g., "default/key/my-key")
+	KBSResourcePath string `json:"kbs_resource_path,omitempty"`
+	// Encrypted indicates whether the resource is encrypted and requires KBS
+	Encrypted bool `json:"encrypted,omitempty"`
+}
+
+// KBSConfig holds configuration for Key Broker Service.
+type KBSConfig struct {
+	// URL is the KBS endpoint (e.g., "https://kbs.example.com")
+	URL string `json:"url,omitempty"`
+	// Enabled indicates whether to use KBS for key retrieval
+	Enabled bool `json:"enabled,omitempty"`
+}
+
 type Computation struct {
 	ID              string           `json:"id,omitempty"`
 	Name            string           `json:"name,omitempty"`
 	Description     string           `json:"description,omitempty"`
 	Datasets        Datasets         `json:"datasets,omitempty"`
-	Algorithm       Algorithm        `json:"algorithm,omitempty"`
+	Algorithm       *Algorithm       `json:"algorithm,omitempty"`
 	ResultConsumers []ResultConsumer `json:"result_consumers,omitempty"`
 }
 
@@ -43,19 +69,26 @@ func (d *Datasets) String() string {
 }
 
 type Dataset struct {
-	Dataset  []byte   `json:"-"`
-	Hash     [32]byte `json:"hash,omitempty"`
-	UserKey  []byte   `json:"user_key,omitempty"`
-	Filename string   `json:"filename,omitempty"`
+	Dataset    []byte          `json:"-"`
+	Hash       [32]byte        `json:"hash,omitempty"`
+	UserKey    []byte          `json:"user_key,omitempty"`
+	Filename   string          `json:"filename,omitempty"`
+	Source     *ResourceSource `json:"source,omitempty"` // Optional remote source
+	Decompress bool            `json:"decompress,omitempty"`
+	KBS        *KBSConfig      `json:"kbs,omitempty"`
 }
 
 type Datasets []Dataset
 
 type Algorithm struct {
-	Algorithm    []byte   `json:"-"`
-	Hash         [32]byte `json:"hash,omitempty"`
-	UserKey      []byte   `json:"user_key,omitempty"`
-	Requirements []byte   `json:"-"`
+	Algorithm    []byte          `json:"-"`
+	Hash         [32]byte        `json:"hash,omitempty"`
+	UserKey      []byte          `json:"user_key,omitempty"`
+	Requirements []byte          `json:"-"`
+	Source       *ResourceSource `json:"source,omitempty"` // Optional remote source
+	AlgoType     string          `json:"algo_type,omitempty"`
+	AlgoArgs     []string        `json:"algo_args,omitempty"`
+	KBS          *KBSConfig      `json:"kbs,omitempty"`
 }
 
 type ManifestIndexKey struct{}

agent/computations_test.go

@@ -105,16 +105,15 @@ func TestDecompressToContext(t *testing.T) {
 }
 
 func TestAgentConfigJSON(t *testing.T) {
-	config := AgentConfig{
-		Port:         "8080",
+	cfg := AgentConfig{
 		CertFile:     "cert.pem",
 		KeyFile:      "key.pem",
-		ServerCAFile: "server_ca.pem",
-		ClientCAFile: "client_ca.pem",
+		ServerCAFile: "server-ca.pem",
+		ClientCAFile: "client-ca.pem",
 		AttestedTls:  true,
 	}
 
-	data, err := json.Marshal(config)
+	data, err := json.Marshal(cfg)
 	if err != nil {
 		t.Fatalf("Failed to marshal AgentConfig: %v", err)
 	}
@@ -125,7 +124,7 @@ func TestAgentConfigJSON(t *testing.T) {
 		t.Fatalf("Failed to unmarshal AgentConfig: %v", err)
 	}
 
-	if !reflect.DeepEqual(config, unmarshaledConfig) {
-		t.Errorf("Unmarshaled config does not match original. Got %+v, want %+v", unmarshaledConfig, config)
+	if !reflect.DeepEqual(cfg, unmarshaledConfig) {
+		t.Errorf("Unmarshaled config does not match original. Got %+v, want %+v", unmarshaledConfig, cfg)
 	}
 }

agent/cvms/api/grpc/client.go

@@ -5,11 +5,12 @@ package grpc
 import (
 	"context"
 	"encoding/json"
+	"fmt"
 	"log/slog"
 	"sync"
 	"time"
 
-	"github.com/absmach/supermq/pkg/errors"
+	"github.com/absmach/magistrala/pkg/errors"
 	"github.com/ultravioletrs/cocos/agent"
 	"github.com/ultravioletrs/cocos/agent/cvms"
 	"github.com/ultravioletrs/cocos/agent/cvms/api/grpc/storage"
@@ -17,6 +18,7 @@ import (
 	"github.com/ultravioletrs/cocos/pkg/attestation"
 	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 	"github.com/ultravioletrs/cocos/pkg/clients/grpc"
+	"github.com/ultravioletrs/cocos/pkg/ingress"
 	"golang.org/x/sync/errgroup"
 	"google.golang.org/protobuf/proto"
 )
@@ -44,13 +46,14 @@ type CVMSClient struct {
 	logger        *slog.Logger
 	runReqManager *runRequestManager
 	sp            server.AgentServer
+	ingressProxy  ingress.ProxyServer
 	storage       storage.Storage
 	reconnectFn   func(context.Context) (grpc.Client, cvms.Service_ProcessClient, error)
 	grpcClient    grpc.Client
 }
 
 // NewClient returns new gRPC client instance.
-func NewClient(stream cvms.Service_ProcessClient, svc agent.Service, messageQueue chan *cvms.ClientStreamMessage, logger *slog.Logger, sp server.AgentServer, storageDir string, reconnectFn func(context.Context) (grpc.Client, cvms.Service_ProcessClient, error), grpcClient grpc.Client) (*CVMSClient, error) {
+func NewClient(stream cvms.Service_ProcessClient, svc agent.Service, messageQueue chan *cvms.ClientStreamMessage, logger *slog.Logger, sp server.AgentServer, ingressProxy ingress.ProxyServer, storageDir string, reconnectFn func(context.Context) (grpc.Client, cvms.Service_ProcessClient, error), grpcClient grpc.Client) (*CVMSClient, error) {
 	store, err := storage.NewFileStorage(storageDir)
 	if err != nil {
 		return nil, err
@@ -63,6 +66,7 @@ func NewClient(stream cvms.Service_ProcessClient, svc agent.Service, messageQueu
 		logger:        logger,
 		runReqManager: newRunRequestManager(),
 		sp:            sp,
+		ingressProxy:  ingressProxy,
 		storage:       store,
 		reconnectFn:   reconnectFn,
 		grpcClient:    grpcClient,
@@ -205,14 +209,17 @@ func (client *CVMSClient) handleAgentStateReq(mes *cvms.ServerStreamMessage_Agen
 }
 
 func (client *CVMSClient) handleRunReqChunks(ctx context.Context, msg *cvms.ServerStreamMessage_RunReqChunks) error {
+	client.logger.Debug("Received RunReq chunk", "id", msg.RunReqChunks.Id, "size", len(msg.RunReqChunks.Data), "isLast", msg.RunReqChunks.IsLast)
 	buffer, complete := client.runReqManager.addChunk(msg.RunReqChunks.Id, msg.RunReqChunks.Data, msg.RunReqChunks.IsLast)
 
 	if complete {
+		client.logger.Info("Received complete computation run request", "id", msg.RunReqChunks.Id, "totalSize", len(buffer))
 		var runReq cvms.ComputationRunReq
 		if err := proto.Unmarshal(buffer, &runReq); err != nil {
 			return errors.Wrap(err, errCorruptedManifest)
 		}
 
+		client.logger.Info("Starting computation execution", "computationId", runReq.Id, "name", runReq.Name)
 		go client.executeRun(ctx, &runReq)
 	}
 
@@ -227,17 +234,50 @@ func (client *CVMSClient) executeRun(ctx context.Context, runReq *cvms.Computati
 	}
 
 	if runReq.Algorithm != nil {
-		ac.Algorithm = agent.Algorithm{
-			Hash:    [32]byte(runReq.Algorithm.Hash),
-			UserKey: runReq.Algorithm.UserKey,
+		ac.Algorithm = &agent.Algorithm{
+			Hash:     [32]byte(runReq.Algorithm.Hash),
+			UserKey:  runReq.Algorithm.UserKey,
+			AlgoType: runReq.Algorithm.AlgoType,
+		}
+		// Copy remote source if configured
+		if runReq.Algorithm.Source != nil {
+			ac.Algorithm.Source = &agent.ResourceSource{
+				URL:             runReq.Algorithm.Source.Url,
+				KBSResourcePath: runReq.Algorithm.Source.KbsResourcePath,
+				Encrypted:       runReq.Algorithm.Source.Encrypted,
+			}
+		}
+		ac.Algorithm.AlgoArgs = runReq.Algorithm.AlgoArgs
+		if runReq.Algorithm.Kbs != nil {
+			ac.Algorithm.KBS = &agent.KBSConfig{
+				URL:     runReq.Algorithm.Kbs.Url,
+				Enabled: runReq.Algorithm.Kbs.Enabled,
+			}
 		}
 	}
 
 	for _, ds := range runReq.Datasets {
-		ac.Datasets = append(ac.Datasets, agent.Dataset{
-			Hash:    [32]byte(ds.Hash),
-			UserKey: ds.UserKey,
-		})
+		dataset := agent.Dataset{
+			Hash:     [32]byte(ds.Hash),
+			UserKey:  ds.UserKey,
+			Filename: ds.Filename,
+		}
+		// Copy remote source if configured
+		if ds.Source != nil {
+			dataset.Source = &agent.ResourceSource{
+				URL:             ds.Source.Url,
+				KBSResourcePath: ds.Source.KbsResourcePath,
+				Encrypted:       ds.Source.Encrypted,
+			}
+		}
+		dataset.Decompress = ds.Decompress
+		if ds.Kbs != nil {
+			dataset.KBS = &agent.KBSConfig{
+				URL:     ds.Kbs.Url,
+				Enabled: ds.Kbs.Enabled,
+			}
+		}
+		ac.Datasets = append(ac.Datasets, dataset)
 	}
 
 	for _, rc := range runReq.ResultConsumers {
@@ -246,6 +286,15 @@ func (client *CVMSClient) executeRun(ctx context.Context, runReq *cvms.Computati
 		})
 	}
 
+	// Check if the agent is in the correct state to initialize a new computation.
+	// If the agent is already processing this computation (e.g., after a reconnection),
+	// skip initialization to avoid state errors.
+	currentState := client.svc.State()
+	if currentState != "ReceivingManifest" {
+		client.logger.Info("Agent already processing computation, skipping initialization", "state", currentState, "computationId", runReq.Id)
+		return
+	}
+
 	if err := client.svc.InitComputation(ctx, ac); err != nil {
 		client.logger.Warn(err.Error())
 		return
@@ -267,7 +316,6 @@ func (client *CVMSClient) executeRun(ctx context.Context, runReq *cvms.Computati
 	}
 
 	if err := client.sp.Start(agent.AgentConfig{
-		Port:         runReq.AgentConfig.Port,
 		CertFile:     runReq.AgentConfig.CertFile,
 		KeyFile:      runReq.AgentConfig.KeyFile,
 		ServerCAFile: runReq.AgentConfig.ServerCaFile,
@@ -278,6 +326,22 @@ func (client *CVMSClient) executeRun(ctx context.Context, runReq *cvms.Computati
 		runRes.RunRes.Error = err.Error()
 	}
 
+	// Start ingress proxy if available
+	if client.ingressProxy != nil {
+		if err := client.ingressProxy.Start(
+			ingress.AgentConfigToProxyConfig(agent.AgentConfig{
+				CertFile:     runReq.AgentConfig.CertFile,
+				KeyFile:      runReq.AgentConfig.KeyFile,
+				ServerCAFile: runReq.AgentConfig.ServerCaFile,
+				ClientCAFile: runReq.AgentConfig.ClientCaFile,
+				AttestedTls:  runReq.AgentConfig.AttestedTls,
+			}),
+			ingress.ComputationToProxyContext(ac),
+		); err != nil {
+			client.logger.Warn(fmt.Sprintf("failed to start ingress proxy: %s", err.Error()))
+		}
+	}
+
 	defer func() {
 		if ccPlatform == attestation.Azure || ccPlatform == attestation.SNPvTPM {
 			cmpJson, err := json.Marshal(ac)
@@ -309,6 +373,12 @@ func (client *CVMSClient) handleStopComputation(ctx context.Context, mes *cvms.S
 	if err := client.sp.Stop(); err != nil {
 		msg.StopComputationRes.Message = err.Error()
 	}
+	// Stop ingress proxy if available
+	if client.ingressProxy != nil {
+		if err := client.ingressProxy.Stop(); err != nil {
+			client.logger.Warn(fmt.Sprintf("failed to stop ingress proxy: %s", err.Error()))
+		}
+	}
 	client.mu.Unlock()
 
 	client.sendMessage(&cvms.ClientStreamMessage{Message: msg})

agent/cvms/api/grpc/client_test.go

@@ -7,14 +7,17 @@ import (
 	"testing"
 	"time"
 
-	mglog "github.com/absmach/supermq/logger"
+	mglog "github.com/absmach/magistrala/logger"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
+	"github.com/ultravioletrs/cocos/agent"
 	"github.com/ultravioletrs/cocos/agent/cvms"
+	"github.com/ultravioletrs/cocos/agent/cvms/api/grpc/storage"
 	servermocks "github.com/ultravioletrs/cocos/agent/cvms/server/mocks"
 	"github.com/ultravioletrs/cocos/agent/mocks"
 	pkggrpc "github.com/ultravioletrs/cocos/pkg/clients/grpc"
 	clientmocks "github.com/ultravioletrs/cocos/pkg/clients/grpc/mocks"
+	"github.com/ultravioletrs/cocos/pkg/ingress"
 	"golang.org/x/crypto/sha3"
 	"google.golang.org/grpc"
 	"google.golang.org/protobuf/proto"
@@ -35,6 +38,21 @@ func (m *mockStream) Send(msg *cvms.ClientStreamMessage) error {
 	return args.Error(0)
 }
 
+// mockIngressProxy is a mock implementation of the ingress proxy.
+type mockIngressProxy struct {
+	mock.Mock
+}
+
+func (m *mockIngressProxy) Start(config ingress.ProxyConfig, ctx ingress.ProxyContext) error {
+	args := m.Called(config, ctx)
+	return args.Error(0)
+}
+
+func (m *mockIngressProxy) Stop() error {
+	args := m.Called()
+	return args.Error(0)
+}
+
 func TestManagerClient_Process(t *testing.T) {
 	tests := []struct {
 		name        string
@@ -121,7 +139,7 @@ func TestManagerClient_Process(t *testing.T) {
 
 			grpcClient := new(clientmocks.Client)
 
-			client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+			client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, nil, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
 			assert.NoError(t, err)
 
 			ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
@@ -151,7 +169,7 @@ func TestManagerClient_handleRunReqChunks(t *testing.T) {
 	logger := mglog.NewMock()
 	grpcClient := new(clientmocks.Client)
 
-	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, nil, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
 	assert.NoError(t, err)
 
 	runReq := &cvms.ComputationRunReq{
@@ -187,6 +205,7 @@ func TestManagerClient_handleRunReqChunks(t *testing.T) {
 		},
 	}
 
+	mockSvc.On("State").Return("ReceivingManifest")
 	mockSvc.On("InitComputation", mock.Anything, mock.Anything).Return(nil)
 	mockServerSvc.On("Start", mock.Anything, mock.Anything, mock.Anything).Return(nil)
 
@@ -216,7 +235,7 @@ func TestManagerClient_handleStopComputation(t *testing.T) {
 	logger := mglog.NewMock()
 	grpcClient := new(clientmocks.Client)
 
-	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, nil, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
 	assert.NoError(t, err)
 
 	stopReq := &cvms.ServerStreamMessage_StopComputation{
@@ -255,3 +274,381 @@ func TestManagerClient_timeoutRequest(t *testing.T) {
 
 	assert.Len(t, rm.requests, 0)
 }
+
+// TestManagerClient_sendPendingMessages tests sending pending messages on reconnection.
+func TestManagerClient_sendPendingMessages(t *testing.T) {
+	mockStream := new(mockStream)
+	mockSvc := new(mocks.Service)
+	mockServerSvc := new(servermocks.AgentServer)
+	messageQueue := make(chan *cvms.ClientStreamMessage, 10)
+	logger := mglog.NewMock()
+	grpcClient := new(clientmocks.Client)
+
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, nil, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	assert.NoError(t, err)
+
+	// Add a pending message to storage
+	testMsg := &cvms.ClientStreamMessage{
+		Message: &cvms.ClientStreamMessage_RunRes{
+			RunRes: &cvms.RunResponse{
+				ComputationId: "test-id",
+			},
+		},
+	}
+	err = client.storage.Add(testMsg)
+	assert.NoError(t, err)
+
+	// Mock successful send
+	mockStream.On("Send", mock.Anything).Return(nil).Once()
+
+	// Load and send pending messages
+	pending, err := client.storage.Load()
+	assert.NoError(t, err)
+	assert.Len(t, pending, 1)
+
+	client.sendPendingMessages(pending)
+
+	mockStream.AssertExpectations(t)
+}
+
+// TestManagerClient_sendPendingMessagesWithError tests pending message send failure.
+func TestManagerClient_sendPendingMessagesWithError(t *testing.T) {
+	mockStream := new(mockStream)
+	mockSvc := new(mocks.Service)
+	mockServerSvc := new(servermocks.AgentServer)
+	messageQueue := make(chan *cvms.ClientStreamMessage, 10)
+	logger := mglog.NewMock()
+	grpcClient := new(clientmocks.Client)
+
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, nil, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	assert.NoError(t, err)
+
+	testMsg := &cvms.ClientStreamMessage{
+		Message: &cvms.ClientStreamMessage_RunRes{
+			RunRes: &cvms.RunResponse{
+				ComputationId: "test-id",
+			},
+		},
+	}
+
+	// Mock failed send
+	mockStream.On("Send", mock.Anything).Return(assert.AnError)
+
+	pending := []storage.Message{
+		{
+			Message: testMsg,
+			Time:    time.Now(),
+		},
+	}
+
+	client.sendPendingMessages(pending)
+
+	mockStream.AssertExpectations(t)
+}
+
+// TestManagerClient_addChunkTimeout tests chunk timeout in runRequestManager.
+func TestManagerClient_addChunkTimeout(t *testing.T) {
+	rm := newRunRequestManager()
+
+	// Add first chunk
+	chunk1 := []byte("chunk1")
+	buffer, complete := rm.addChunk("test-id", chunk1, false)
+	assert.Nil(t, buffer)
+	assert.False(t, complete)
+
+	// Verify request exists
+	rm.mu.Lock()
+	assert.Contains(t, rm.requests, "test-id")
+	rm.mu.Unlock()
+
+	// Wait for timeout
+	time.Sleep(35 * time.Second) // runReqTimeout is 30 seconds
+
+	// Verify request was removed
+	rm.mu.Lock()
+	assert.NotContains(t, rm.requests, "test-id")
+	rm.mu.Unlock()
+}
+
+// TestManagerClient_addChunkMultiple tests adding multiple chunks.
+func TestManagerClient_addChunkMultiple(t *testing.T) {
+	rm := newRunRequestManager()
+
+	chunk1 := []byte("chunk1")
+	chunk2 := []byte("chunk2")
+	chunk3 := []byte("chunk3")
+
+	// Add chunks
+	buffer, complete := rm.addChunk("test-id", chunk1, false)
+	assert.Nil(t, buffer)
+	assert.False(t, complete)
+
+	buffer, complete = rm.addChunk("test-id", chunk2, false)
+	assert.Nil(t, buffer)
+	assert.False(t, complete)
+
+	buffer, complete = rm.addChunk("test-id", chunk3, true)
+	assert.NotNil(t, buffer)
+	assert.True(t, complete)
+
+	expected := append(append(chunk1, chunk2...), chunk3...)
+	assert.Equal(t, expected, buffer)
+}
+
+// TestManagerClient_handleStopComputationWithIngressProxy tests stop with ingress proxy.
+func TestManagerClient_handleStopComputationWithIngressProxy(t *testing.T) {
+	mockStream := new(mockStream)
+	mockSvc := new(mocks.Service)
+	mockServerSvc := new(servermocks.AgentServer)
+	mockIngressProxy := new(mockIngressProxy)
+	messageQueue := make(chan *cvms.ClientStreamMessage, 10)
+	logger := mglog.NewMock()
+	grpcClient := new(clientmocks.Client)
+
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, mockIngressProxy, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	assert.NoError(t, err)
+
+	stopReq := &cvms.ServerStreamMessage_StopComputation{
+		StopComputation: &cvms.StopComputation{
+			ComputationId: "test-comp-id",
+		},
+	}
+
+	mockSvc.On("StopComputation", mock.Anything).Return(nil)
+	mockServerSvc.On("Stop").Return(nil)
+	mockIngressProxy.On("Stop").Return(nil)
+
+	client.handleStopComputation(context.Background(), stopReq)
+
+	time.Sleep(50 * time.Millisecond)
+
+	mockSvc.AssertExpectations(t)
+	mockServerSvc.AssertExpectations(t)
+	mockIngressProxy.AssertExpectations(t)
+	assert.Len(t, messageQueue, 1)
+}
+
+// TestManagerClient_handleStopComputationWithIngressProxyError tests stop with ingress proxy error.
+func TestManagerClient_handleStopComputationWithIngressProxyError(t *testing.T) {
+	mockStream := new(mockStream)
+	mockSvc := new(mocks.Service)
+	mockServerSvc := new(servermocks.AgentServer)
+	mockIngressProxy := new(mockIngressProxy)
+	messageQueue := make(chan *cvms.ClientStreamMessage, 10)
+	logger := mglog.NewMock()
+	grpcClient := new(clientmocks.Client)
+
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, mockIngressProxy, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	assert.NoError(t, err)
+
+	stopReq := &cvms.ServerStreamMessage_StopComputation{
+		StopComputation: &cvms.StopComputation{
+			ComputationId: "test-comp-id",
+		},
+	}
+
+	mockSvc.On("StopComputation", mock.Anything).Return(nil)
+	mockServerSvc.On("Stop").Return(nil)
+	mockIngressProxy.On("Stop").Return(assert.AnError)
+
+	client.handleStopComputation(context.Background(), stopReq)
+
+	time.Sleep(50 * time.Millisecond)
+
+	mockIngressProxy.AssertExpectations(t)
+}
+
+// TestManagerClient_sendMessage tests sendMessage with timeout.
+func TestManagerClient_sendMessage(t *testing.T) {
+	mockStream := new(mockStream)
+	mockSvc := new(mocks.Service)
+	mockServerSvc := new(servermocks.AgentServer)
+	messageQueue := make(chan *cvms.ClientStreamMessage, 1)
+	logger := mglog.NewMock()
+	grpcClient := new(clientmocks.Client)
+
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, nil, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	assert.NoError(t, err)
+
+	msg := &cvms.ClientStreamMessage{
+		Message: &cvms.ClientStreamMessage_RunRes{
+			RunRes: &cvms.RunResponse{
+				ComputationId: "test-id",
+			},
+		},
+	}
+
+	client.sendMessage(msg)
+
+	select {
+	case received := <-messageQueue:
+		assert.Equal(t, msg, received)
+	case <-time.After(1 * time.Second):
+		t.Fatal("Message not received")
+	}
+}
+
+// TestManagerClient_sendMessageTimeout tests sendMessage timeout when queue is full.
+func TestManagerClient_sendMessageTimeout(t *testing.T) {
+	mockStream := new(mockStream)
+	mockSvc := new(mocks.Service)
+	mockServerSvc := new(servermocks.AgentServer)
+	messageQueue := make(chan *cvms.ClientStreamMessage) // No buffer
+	logger := mglog.NewMock()
+	grpcClient := new(clientmocks.Client)
+
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, nil, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	assert.NoError(t, err)
+
+	msg := &cvms.ClientStreamMessage{
+		Message: &cvms.ClientStreamMessage_RunRes{
+			RunRes: &cvms.RunResponse{
+				ComputationId: "test-id",
+			},
+		},
+	}
+
+	// Don't read from queue, so sendMessage will timeout
+	client.sendMessage(msg)
+
+	// Should complete without blocking
+	time.Sleep(100 * time.Millisecond)
+}
+
+// TestManagerClient_handleRunReqChunksWithRemoteSource tests handling run request with remote source.
+func TestManagerClient_handleRunReqChunksWithRemoteSource(t *testing.T) {
+	mockStream := new(mockStream)
+	mockSvc := new(mocks.Service)
+	mockServerSvc := new(servermocks.AgentServer)
+	messageQueue := make(chan *cvms.ClientStreamMessage, 10)
+	logger := mglog.NewMock()
+	grpcClient := new(clientmocks.Client)
+
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, nil, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	assert.NoError(t, err)
+
+	runReq := &cvms.ComputationRunReq{
+		Id:          "test-id-remote",
+		Name:        "test-computation",
+		Description: "test description",
+		Datasets: []*cvms.Dataset{
+			{
+				Hash:     sha3.New256().Sum([]byte("test-dataset")),
+				Filename: "data.csv",
+				Source: &cvms.Source{
+					Type:            "oci-image",
+					Url:             "docker://registry.example.com/data:v1",
+					KbsResourcePath: "default/key/data-key",
+					Encrypted:       true,
+				},
+				Decompress: true,
+			},
+		},
+		Algorithm: &cvms.Algorithm{
+			Hash:     sha3.New256().Sum([]byte("test-algorithm")),
+			AlgoType: "python",
+			AlgoArgs: []string{"--verbose"},
+			Source: &cvms.Source{
+				Type:            "oci-image",
+				Url:             "docker://registry.example.com/algo:v1",
+				KbsResourcePath: "default/key/algo-key",
+				Encrypted:       true,
+			},
+			Kbs: &cvms.KBSConfig{
+				Url:     "https://kbs.example.com:8080",
+				Enabled: true,
+			},
+		},
+
+		ResultConsumers: []*cvms.ResultConsumer{
+			{
+				UserKey: []byte("test-consumer"),
+			},
+		},
+	}
+	runReqBytes, _ := proto.Marshal(runReq)
+
+	chunk := &cvms.ServerStreamMessage_RunReqChunks{
+		RunReqChunks: &cvms.RunReqChunks{
+			Id:     "chunk-remote-1",
+			Data:   runReqBytes,
+			IsLast: true,
+		},
+	}
+
+	mockSvc.On("State").Return("ReceivingManifest")
+	mockSvc.On("InitComputation", mock.Anything, mock.MatchedBy(func(c agent.Computation) bool {
+		// Verify Algorithm KBS config is passed
+		if c.Algorithm.KBS == nil || !c.Algorithm.KBS.Enabled || c.Algorithm.KBS.URL != "https://kbs.example.com:8080" {
+			return false
+		}
+		// Verify algorithm source is passed
+		if c.Algorithm.Source == nil ||
+			c.Algorithm.Source.URL != "docker://registry.example.com/algo:v1" ||
+			c.Algorithm.Source.KBSResourcePath != "default/key/algo-key" ||
+			!c.Algorithm.Source.Encrypted {
+			return false
+		}
+		// Verify algorithm type and args
+		if c.Algorithm.AlgoType != "python" || len(c.Algorithm.AlgoArgs) != 1 || c.Algorithm.AlgoArgs[0] != "--verbose" {
+			return false
+		}
+		// Verify dataset source is passed
+		if len(c.Datasets) != 1 ||
+			c.Datasets[0].Source == nil ||
+			c.Datasets[0].Source.URL != "docker://registry.example.com/data:v1" ||
+			c.Datasets[0].Filename != "data.csv" ||
+			!c.Datasets[0].Decompress {
+			return false
+		}
+		return true
+	})).Return(nil)
+	mockServerSvc.On("Start", mock.Anything, mock.Anything, mock.Anything).Return(nil)
+
+	err = client.handleRunReqChunks(context.Background(), chunk)
+	assert.NoError(t, err)
+
+	// Wait for the goroutine to finish
+	time.Sleep(100 * time.Millisecond)
+
+	mockSvc.AssertExpectations(t)
+}
+
+// TestManagerClient_handleRunReqChunksAlreadyProcessing tests skipping init when already processing.
+func TestManagerClient_handleRunReqChunksAlreadyProcessing(t *testing.T) {
+	mockStream := new(mockStream)
+	mockSvc := new(mocks.Service)
+	mockServerSvc := new(servermocks.AgentServer)
+	messageQueue := make(chan *cvms.ClientStreamMessage, 10)
+	logger := mglog.NewMock()
+	grpcClient := new(clientmocks.Client)
+
+	client, err := NewClient(mockStream, mockSvc, messageQueue, logger, mockServerSvc, nil, t.TempDir(), func(ctx context.Context) (pkggrpc.Client, cvms.Service_ProcessClient, error) { return nil, nil, nil }, grpcClient)
+	assert.NoError(t, err)
+
+	runReq := &cvms.ComputationRunReq{
+		Id:   "test-id-processing",
+		Name: "test-computation",
+	}
+	runReqBytes, _ := proto.Marshal(runReq)
+
+	chunk := &cvms.ServerStreamMessage_RunReqChunks{
+		RunReqChunks: &cvms.RunReqChunks{
+			Id:     "chunk-processing-1",
+			Data:   runReqBytes,
+			IsLast: true,
+		},
+	}
+
+	// Simulate agent already processing a computation
+	mockSvc.On("State").Return("Running")
+
+	err = client.handleRunReqChunks(context.Background(), chunk)
+	assert.NoError(t, err)
+
+	// Wait for the goroutine to finish
+	time.Sleep(50 * time.Millisecond)
+
+	// InitComputation should NOT be called since state is not ReceivingManifest
+	mockSvc.AssertNotCalled(t, "InitComputation")
+}

agent/cvms/api/grpc/server.go

@@ -7,6 +7,7 @@ import (
 	"context"
 	"errors"
 	"io"
+	"log/slog"
 	"time"
 
 	"github.com/ultravioletrs/cocos/agent/cvms"
@@ -52,16 +53,20 @@ func (s *grpcServer) Process(stream cvms.Service_ProcessServer) error {
 		return errors.New("failed to get peer info")
 	}
 
+	slog.Info("client connected to cvms server", "address", client.Addr.String())
+
 	eg, ctx := errgroup.WithContext(stream.Context())
 
 	eg.Go(func() error {
 		for {
 			select {
 			case <-ctx.Done():
+				slog.Info("receive goroutine context done", "address", client.Addr.String())
 				return ctx.Err()
 			default:
 				req, err := stream.Recv()
 				if err != nil {
+					slog.Error("failed to receive from stream", "address", client.Addr.String(), "error", err)
 					return err
 				}
 				s.incoming <- req
@@ -85,10 +90,13 @@ func (s *grpcServer) Process(stream cvms.Service_ProcessServer) error {
 		}
 
 		s.svc.Run(ctx, client.Addr.String(), sendMessage, client.AuthInfo)
+		slog.Info("send goroutine Run() returned", "address", client.Addr.String())
 		return nil
 	})
 
-	return eg.Wait()
+	err := eg.Wait()
+	slog.Info("stream closed", "address", client.Addr.String(), "error", err)
+	return err
 }
 
 func (s *grpcServer) sendRunReqInChunks(stream cvms.Service_ProcessServer, runReq *cvms.ComputationRunReq) error {

agent/cvms/api/grpc/server_test.go

@@ -7,7 +7,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/absmach/supermq/pkg/errors"
+	"github.com/absmach/magistrala/pkg/errors"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 	"github.com/ultravioletrs/cocos/agent/cvms"

agent/cvms/cvms.pb.go

@@ -3,8 +3,8 @@
 
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.36.8
-// 	protoc        v5.29.0
+// 	protoc-gen-go v1.36.11
+// 	protoc        v6.33.1
 // source: agent/cvms/cvms.proto
 
 package cvms
@@ -958,6 +958,9 @@ type Dataset struct {
 	Hash          []byte                 `protobuf:"bytes,1,opt,name=hash,proto3" json:"hash,omitempty"` // should be sha3.Sum256, 32 byte length.
 	UserKey       []byte                 `protobuf:"bytes,2,opt,name=userKey,proto3" json:"userKey,omitempty"`
 	Filename      string                 `protobuf:"bytes,3,opt,name=filename,proto3" json:"filename,omitempty"`
+	Source        *Source                `protobuf:"bytes,4,opt,name=source,proto3" json:"source,omitempty"` // Optional remote source for encrypted dataset
+	Decompress    bool                   `protobuf:"varint,5,opt,name=decompress,proto3" json:"decompress,omitempty"`
+	Kbs           *KBSConfig             `protobuf:"bytes,6,opt,name=kbs,proto3" json:"kbs,omitempty"` // Optional KBS configuration override
 	unknownFields protoimpl.UnknownFields
 	sizeCache     protoimpl.SizeCache
 }
@@ -1013,10 +1016,35 @@ func (x *Dataset) GetFilename() string {
 	return ""
 }
 
+func (x *Dataset) GetSource() *Source {
+	if x != nil {
+		return x.Source
+	}
+	return nil
+}
+
+func (x *Dataset) GetDecompress() bool {
+	if x != nil {
+		return x.Decompress
+	}
+	return false
+}
+
+func (x *Dataset) GetKbs() *KBSConfig {
+	if x != nil {
+		return x.Kbs
+	}
+	return nil
+}
+
 type Algorithm struct {
 	state         protoimpl.MessageState `protogen:"open.v1"`
 	Hash          []byte                 `protobuf:"bytes,1,opt,name=hash,proto3" json:"hash,omitempty"` // should be sha3.Sum256, 32 byte length.
 	UserKey       []byte                 `protobuf:"bytes,2,opt,name=userKey,proto3" json:"userKey,omitempty"`
+	Source        *Source                `protobuf:"bytes,3,opt,name=source,proto3" json:"source,omitempty"` // Optional remote source for encrypted algorithm
+	AlgoType      string                 `protobuf:"bytes,4,opt,name=algo_type,json=algoType,proto3" json:"algo_type,omitempty"`
+	AlgoArgs      []string               `protobuf:"bytes,5,rep,name=algo_args,json=algoArgs,proto3" json:"algo_args,omitempty"`
+	Kbs           *KBSConfig             `protobuf:"bytes,6,opt,name=kbs,proto3" json:"kbs,omitempty"` // Optional KBS configuration override
 	unknownFields protoimpl.UnknownFields
 	sizeCache     protoimpl.SizeCache
 }
@@ -1065,6 +1093,154 @@ func (x *Algorithm) GetUserKey() []byte {
 	return nil
 }
 
+func (x *Algorithm) GetSource() *Source {
+	if x != nil {
+		return x.Source
+	}
+	return nil
+}
+
+func (x *Algorithm) GetAlgoType() string {
+	if x != nil {
+		return x.AlgoType
+	}
+	return ""
+}
+
+func (x *Algorithm) GetAlgoArgs() []string {
+	if x != nil {
+		return x.AlgoArgs
+	}
+	return nil
+}
+
+func (x *Algorithm) GetKbs() *KBSConfig {
+	if x != nil {
+		return x.Kbs
+	}
+	return nil
+}
+
+type Source struct {
+	state           protoimpl.MessageState `protogen:"open.v1"`
+	Type            string                 `protobuf:"bytes,1,opt,name=type,proto3" json:"type,omitempty"`                                                // Type of source: "oci-image" (only OCI images supported for CoCo)
+	Url             string                 `protobuf:"bytes,2,opt,name=url,proto3" json:"url,omitempty"`                                                  // URL of the OCI image (e.g., docker://registry/repo:tag)
+	KbsResourcePath string                 `protobuf:"bytes,3,opt,name=kbs_resource_path,json=kbsResourcePath,proto3" json:"kbs_resource_path,omitempty"` // Path to decryption key in KBS (e.g., "default/key/my-key")
+	Encrypted       bool                   `protobuf:"varint,4,opt,name=encrypted,proto3" json:"encrypted,omitempty"`                                     // Whether the resource is encrypted (requires KBS)
+	unknownFields   protoimpl.UnknownFields
+	sizeCache       protoimpl.SizeCache
+}
+
+func (x *Source) Reset() {
+	*x = Source{}
+	mi := &file_agent_cvms_cvms_proto_msgTypes[15]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *Source) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*Source) ProtoMessage() {}
+
+func (x *Source) ProtoReflect() protoreflect.Message {
+	mi := &file_agent_cvms_cvms_proto_msgTypes[15]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use Source.ProtoReflect.Descriptor instead.
+func (*Source) Descriptor() ([]byte, []int) {
+	return file_agent_cvms_cvms_proto_rawDescGZIP(), []int{15}
+}
+
+func (x *Source) GetType() string {
+	if x != nil {
+		return x.Type
+	}
+	return ""
+}
+
+func (x *Source) GetUrl() string {
+	if x != nil {
+		return x.Url
+	}
+	return ""
+}
+
+func (x *Source) GetKbsResourcePath() string {
+	if x != nil {
+		return x.KbsResourcePath
+	}
+	return ""
+}
+
+func (x *Source) GetEncrypted() bool {
+	if x != nil {
+		return x.Encrypted
+	}
+	return false
+}
+
+type KBSConfig struct {
+	state         protoimpl.MessageState `protogen:"open.v1"`
+	Url           string                 `protobuf:"bytes,1,opt,name=url,proto3" json:"url,omitempty"`          // KBS endpoint URL (e.g., "https://kbs.example.com")
+	Enabled       bool                   `protobuf:"varint,2,opt,name=enabled,proto3" json:"enabled,omitempty"` // Whether to use KBS for key retrieval
+	unknownFields protoimpl.UnknownFields
+	sizeCache     protoimpl.SizeCache
+}
+
+func (x *KBSConfig) Reset() {
+	*x = KBSConfig{}
+	mi := &file_agent_cvms_cvms_proto_msgTypes[16]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *KBSConfig) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*KBSConfig) ProtoMessage() {}
+
+func (x *KBSConfig) ProtoReflect() protoreflect.Message {
+	mi := &file_agent_cvms_cvms_proto_msgTypes[16]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use KBSConfig.ProtoReflect.Descriptor instead.
+func (*KBSConfig) Descriptor() ([]byte, []int) {
+	return file_agent_cvms_cvms_proto_rawDescGZIP(), []int{16}
+}
+
+func (x *KBSConfig) GetUrl() string {
+	if x != nil {
+		return x.Url
+	}
+	return ""
+}
+
+func (x *KBSConfig) GetEnabled() bool {
+	if x != nil {
+		return x.Enabled
+	}
+	return false
+}
+
 type AgentConfig struct {
 	state         protoimpl.MessageState `protogen:"open.v1"`
 	Port          string                 `protobuf:"bytes,1,opt,name=port,proto3" json:"port,omitempty"`
@@ -1080,7 +1256,7 @@ type AgentConfig struct {
 
 func (x *AgentConfig) Reset() {
 	*x = AgentConfig{}
-	mi := &file_agent_cvms_cvms_proto_msgTypes[15]
+	mi := &file_agent_cvms_cvms_proto_msgTypes[17]
 	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
 	ms.StoreMessageInfo(mi)
 }
@@ -1092,7 +1268,7 @@ func (x *AgentConfig) String() string {
 func (*AgentConfig) ProtoMessage() {}
 
 func (x *AgentConfig) ProtoReflect() protoreflect.Message {
-	mi := &file_agent_cvms_cvms_proto_msgTypes[15]
+	mi := &file_agent_cvms_cvms_proto_msgTypes[17]
 	if x != nil {
 		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
 		if ms.LoadMessageInfo() == nil {
@@ -1105,7 +1281,7 @@ func (x *AgentConfig) ProtoReflect() protoreflect.Message {
 
 // Deprecated: Use AgentConfig.ProtoReflect.Descriptor instead.
 func (*AgentConfig) Descriptor() ([]byte, []int) {
-	return file_agent_cvms_cvms_proto_rawDescGZIP(), []int{15}
+	return file_agent_cvms_cvms_proto_rawDescGZIP(), []int{17}
 }
 
 func (x *AgentConfig) GetPort() string {
@@ -1167,7 +1343,7 @@ type AttestationResponse struct {
 
 func (x *AttestationResponse) Reset() {
 	*x = AttestationResponse{}
-	mi := &file_agent_cvms_cvms_proto_msgTypes[16]
+	mi := &file_agent_cvms_cvms_proto_msgTypes[18]
 	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
 	ms.StoreMessageInfo(mi)
 }
@@ -1179,7 +1355,7 @@ func (x *AttestationResponse) String() string {
 func (*AttestationResponse) ProtoMessage() {}
 
 func (x *AttestationResponse) ProtoReflect() protoreflect.Message {
-	mi := &file_agent_cvms_cvms_proto_msgTypes[16]
+	mi := &file_agent_cvms_cvms_proto_msgTypes[18]
 	if x != nil {
 		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
 		if ms.LoadMessageInfo() == nil {
@@ -1192,7 +1368,7 @@ func (x *AttestationResponse) ProtoReflect() protoreflect.Message {
 
 // Deprecated: Use AttestationResponse.ProtoReflect.Descriptor instead.
 func (*AttestationResponse) Descriptor() ([]byte, []int) {
-	return file_agent_cvms_cvms_proto_rawDescGZIP(), []int{16}
+	return file_agent_cvms_cvms_proto_rawDescGZIP(), []int{18}
 }
 
 func (x *AttestationResponse) GetFile() []byte {
@@ -1219,7 +1395,7 @@ type AzureAttestationToken struct {
 
 func (x *AzureAttestationToken) Reset() {
 	*x = AzureAttestationToken{}
-	mi := &file_agent_cvms_cvms_proto_msgTypes[17]
+	mi := &file_agent_cvms_cvms_proto_msgTypes[19]
 	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
 	ms.StoreMessageInfo(mi)
 }
@@ -1231,7 +1407,7 @@ func (x *AzureAttestationToken) String() string {
 func (*AzureAttestationToken) ProtoMessage() {}
 
 func (x *AzureAttestationToken) ProtoReflect() protoreflect.Message {
-	mi := &file_agent_cvms_cvms_proto_msgTypes[17]
+	mi := &file_agent_cvms_cvms_proto_msgTypes[19]
 	if x != nil {
 		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
 		if ms.LoadMessageInfo() == nil {
@@ -1244,7 +1420,7 @@ func (x *AzureAttestationToken) ProtoReflect() protoreflect.Message {
 
 // Deprecated: Use AzureAttestationToken.ProtoReflect.Descriptor instead.
 func (*AzureAttestationToken) Descriptor() ([]byte, []int) {
-	return file_agent_cvms_cvms_proto_rawDescGZIP(), []int{17}
+	return file_agent_cvms_cvms_proto_rawDescGZIP(), []int{19}
 }
 
 func (x *AzureAttestationToken) GetFile() []byte {
@@ -1327,14 +1503,31 @@ const file_agent_cvms_cvms_proto_rawDesc = "" +
 	"\x10result_consumers\x18\x06 \x03(\v2\x14.cvms.ResultConsumerR\x0fresultConsumers\x124\n" +
 	"\fagent_config\x18\a \x01(\v2\x11.cvms.AgentConfigR\vagentConfig\"*\n" +
 	"\x0eResultConsumer\x12\x18\n" +
-	"\auserKey\x18\x01 \x01(\fR\auserKey\"S\n" +
+	"\auserKey\x18\x01 \x01(\fR\auserKey\"\xbc\x01\n" +
 	"\aDataset\x12\x12\n" +
 	"\x04hash\x18\x01 \x01(\fR\x04hash\x12\x18\n" +
 	"\auserKey\x18\x02 \x01(\fR\auserKey\x12\x1a\n" +
-	"\bfilename\x18\x03 \x01(\tR\bfilename\"9\n" +
+	"\bfilename\x18\x03 \x01(\tR\bfilename\x12$\n" +
+	"\x06source\x18\x04 \x01(\v2\f.cvms.SourceR\x06source\x12\x1e\n" +
+	"\n" +
+	"decompress\x18\x05 \x01(\bR\n" +
+	"decompress\x12!\n" +
+	"\x03kbs\x18\x06 \x01(\v2\x0f.cvms.KBSConfigR\x03kbs\"\xbc\x01\n" +
 	"\tAlgorithm\x12\x12\n" +
 	"\x04hash\x18\x01 \x01(\fR\x04hash\x12\x18\n" +
-	"\auserKey\x18\x02 \x01(\fR\auserKey\"\xe5\x01\n" +
+	"\auserKey\x18\x02 \x01(\fR\auserKey\x12$\n" +
+	"\x06source\x18\x03 \x01(\v2\f.cvms.SourceR\x06source\x12\x1b\n" +
+	"\talgo_type\x18\x04 \x01(\tR\balgoType\x12\x1b\n" +
+	"\talgo_args\x18\x05 \x03(\tR\balgoArgs\x12!\n" +
+	"\x03kbs\x18\x06 \x01(\v2\x0f.cvms.KBSConfigR\x03kbs\"x\n" +
+	"\x06Source\x12\x12\n" +
+	"\x04type\x18\x01 \x01(\tR\x04type\x12\x10\n" +
+	"\x03url\x18\x02 \x01(\tR\x03url\x12*\n" +
+	"\x11kbs_resource_path\x18\x03 \x01(\tR\x0fkbsResourcePath\x12\x1c\n" +
+	"\tencrypted\x18\x04 \x01(\bR\tencrypted\"7\n" +
+	"\tKBSConfig\x12\x10\n" +
+	"\x03url\x18\x01 \x01(\tR\x03url\x12\x18\n" +
+	"\aenabled\x18\x02 \x01(\bR\aenabled\"\xe5\x01\n" +
 	"\vAgentConfig\x12\x12\n" +
 	"\x04port\x18\x01 \x01(\tR\x04port\x12\x1b\n" +
 	"\tcert_file\x18\x02 \x01(\tR\bcertFile\x12\x19\n" +
@@ -1364,7 +1557,7 @@ func file_agent_cvms_cvms_proto_rawDescGZIP() []byte {
 	return file_agent_cvms_cvms_proto_rawDescData
 }
 
-var file_agent_cvms_cvms_proto_msgTypes = make([]protoimpl.MessageInfo, 18)
+var file_agent_cvms_cvms_proto_msgTypes = make([]protoimpl.MessageInfo, 20)
 var file_agent_cvms_cvms_proto_goTypes = []any{
 	(*AgentStateReq)(nil),           // 0: cvms.AgentStateReq
 	(*AgentStateRes)(nil),           // 1: cvms.AgentStateRes
@@ -1381,21 +1574,23 @@ var file_agent_cvms_cvms_proto_goTypes = []any{
 	(*ResultConsumer)(nil),          // 12: cvms.ResultConsumer
 	(*Dataset)(nil),                 // 13: cvms.Dataset
 	(*Algorithm)(nil),               // 14: cvms.Algorithm
-	(*AgentConfig)(nil),             // 15: cvms.AgentConfig
-	(*AttestationResponse)(nil),     // 16: cvms.AttestationResponse
-	(*AzureAttestationToken)(nil),   // 17: cvms.azureAttestationToken
-	(*timestamppb.Timestamp)(nil),   // 18: google.protobuf.Timestamp
+	(*Source)(nil),                  // 15: cvms.Source
+	(*KBSConfig)(nil),               // 16: cvms.KBSConfig
+	(*AgentConfig)(nil),             // 17: cvms.AgentConfig
+	(*AttestationResponse)(nil),     // 18: cvms.AttestationResponse
+	(*AzureAttestationToken)(nil),   // 19: cvms.azureAttestationToken
+	(*timestamppb.Timestamp)(nil),   // 20: google.protobuf.Timestamp
 }
 var file_agent_cvms_cvms_proto_depIdxs = []int32{
-	18, // 0: cvms.AgentEvent.timestamp:type_name -> google.protobuf.Timestamp
-	18, // 1: cvms.AgentLog.timestamp:type_name -> google.protobuf.Timestamp
+	20, // 0: cvms.AgentEvent.timestamp:type_name -> google.protobuf.Timestamp
+	20, // 1: cvms.AgentLog.timestamp:type_name -> google.protobuf.Timestamp
 	6,  // 2: cvms.ClientStreamMessage.agent_log:type_name -> cvms.AgentLog
 	5,  // 3: cvms.ClientStreamMessage.agent_event:type_name -> cvms.AgentEvent
 	4,  // 4: cvms.ClientStreamMessage.run_res:type_name -> cvms.RunResponse
 	3,  // 5: cvms.ClientStreamMessage.stopComputationRes:type_name -> cvms.StopComputationResponse
 	1,  // 6: cvms.ClientStreamMessage.agentStateRes:type_name -> cvms.AgentStateRes
-	16, // 7: cvms.ClientStreamMessage.vTPMattestationReport:type_name -> cvms.AttestationResponse
-	17, // 8: cvms.ClientStreamMessage.azureAttestationToken:type_name -> cvms.azureAttestationToken
+	18, // 7: cvms.ClientStreamMessage.vTPMattestationReport:type_name -> cvms.AttestationResponse
+	19, // 8: cvms.ClientStreamMessage.azureAttestationToken:type_name -> cvms.azureAttestationToken
 	10, // 9: cvms.ServerStreamMessage.runReqChunks:type_name -> cvms.RunReqChunks
 	11, // 10: cvms.ServerStreamMessage.runReq:type_name -> cvms.ComputationRunReq
 	2,  // 11: cvms.ServerStreamMessage.stopComputation:type_name -> cvms.StopComputation
@@ -1404,14 +1599,18 @@ var file_agent_cvms_cvms_proto_depIdxs = []int32{
 	13, // 14: cvms.ComputationRunReq.datasets:type_name -> cvms.Dataset
 	14, // 15: cvms.ComputationRunReq.algorithm:type_name -> cvms.Algorithm
 	12, // 16: cvms.ComputationRunReq.result_consumers:type_name -> cvms.ResultConsumer
-	15, // 17: cvms.ComputationRunReq.agent_config:type_name -> cvms.AgentConfig
-	7,  // 18: cvms.Service.Process:input_type -> cvms.ClientStreamMessage
-	8,  // 19: cvms.Service.Process:output_type -> cvms.ServerStreamMessage
-	19, // [19:20] is the sub-list for method output_type
-	18, // [18:19] is the sub-list for method input_type
-	18, // [18:18] is the sub-list for extension type_name
-	18, // [18:18] is the sub-list for extension extendee
-	0,  // [0:18] is the sub-list for field type_name
+	17, // 17: cvms.ComputationRunReq.agent_config:type_name -> cvms.AgentConfig
+	15, // 18: cvms.Dataset.source:type_name -> cvms.Source
+	16, // 19: cvms.Dataset.kbs:type_name -> cvms.KBSConfig
+	15, // 20: cvms.Algorithm.source:type_name -> cvms.Source
+	16, // 21: cvms.Algorithm.kbs:type_name -> cvms.KBSConfig
+	7,  // 22: cvms.Service.Process:input_type -> cvms.ClientStreamMessage
+	8,  // 23: cvms.Service.Process:output_type -> cvms.ServerStreamMessage
+	23, // [23:24] is the sub-list for method output_type
+	22, // [22:23] is the sub-list for method input_type
+	22, // [22:22] is the sub-list for extension type_name
+	22, // [22:22] is the sub-list for extension extendee
+	0,  // [0:22] is the sub-list for field type_name
 }
 
 func init() { file_agent_cvms_cvms_proto_init() }
@@ -1441,7 +1640,7 @@ func file_agent_cvms_cvms_proto_init() {
 			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
 			RawDescriptor: unsafe.Slice(unsafe.StringData(file_agent_cvms_cvms_proto_rawDesc), len(file_agent_cvms_cvms_proto_rawDesc)),
 			NumEnums:      0,
-			NumMessages:   18,
+			NumMessages:   20,
 			NumExtensions: 0,
 			NumServices:   1,
 		},

agent/cvms/cvms.proto

@@ -102,11 +102,30 @@ message Dataset {
   bytes hash = 1; // should be sha3.Sum256, 32 byte length.
   bytes userKey = 2;
   string filename = 3;
+  Source source = 4; // Optional remote source for encrypted dataset
+  bool decompress = 5;
+  KBSConfig kbs = 6; // Optional KBS configuration override
 }
 
 message Algorithm {
   bytes hash = 1; // should be sha3.Sum256, 32 byte length.
   bytes userKey = 2;
+  Source source = 3; // Optional remote source for encrypted algorithm
+  string algo_type = 4;
+  repeated string algo_args = 5;
+  KBSConfig kbs = 6; // Optional KBS configuration override
+}
+
+message Source {
+  string type = 1; // Type of source: "oci-image", "s3", "gcs", "https", "http"
+  string url = 2; // URL of the resource (e.g., docker://registry/repo:tag, s3://bucket/key, https://host/path)
+  string kbs_resource_path = 3; // Path to decryption key in KBS (e.g., "default/key/my-key")
+  bool encrypted = 4; // Whether the resource is encrypted (requires KBS)
+}
+
+message KBSConfig {
+  string url = 1; // KBS endpoint URL (e.g., "https://kbs.example.com")
+  bool enabled = 2; // Whether to use KBS for key retrieval
 }
 
 message AgentConfig {

agent/cvms/cvms_grpc.pb.go

@@ -3,8 +3,8 @@
 
 // Code generated by protoc-gen-go-grpc. DO NOT EDIT.
 // versions:
-// - protoc-gen-go-grpc v1.5.1
-// - protoc             v5.29.0
+// - protoc-gen-go-grpc v1.6.0
+// - protoc             v6.33.1
 // source: agent/cvms/cvms.proto
 
 package cvms
@@ -69,7 +69,7 @@ type ServiceServer interface {
 type UnimplementedServiceServer struct{}
 
 func (UnimplementedServiceServer) Process(grpc.BidiStreamingServer[ClientStreamMessage, ServerStreamMessage]) error {
-	return status.Errorf(codes.Unimplemented, "method Process not implemented")
+	return status.Error(codes.Unimplemented, "method Process not implemented")
 }
 func (UnimplementedServiceServer) mustEmbedUnimplementedServiceServer() {}
 func (UnimplementedServiceServer) testEmbeddedByValue()                 {}
@@ -82,7 +82,7 @@ type UnsafeServiceServer interface {
 }
 
 func RegisterServiceServer(s grpc.ServiceRegistrar, srv ServiceServer) {
-	// If the following call pancis, it indicates UnimplementedServiceServer was
+	// If the following call panics, it indicates UnimplementedServiceServer was
 	// embedded by pointer and is nil.  This will cause panics if an
 	// unimplemented method is ever invoked, so we test this at initialization
 	// time to prevent it from happening at runtime later due to I/O.

agent/cvms/server/cvm.go

@@ -4,23 +4,26 @@
 package server
 
 import (
-	context "context"
 	"fmt"
 	"log/slog"
+	"net"
+	"os"
+	"sync"
 
 	"github.com/ultravioletrs/cocos/agent"
 	agentgrpc "github.com/ultravioletrs/cocos/agent/api/grpc"
 	"github.com/ultravioletrs/cocos/agent/auth"
-	"github.com/ultravioletrs/cocos/pkg/atls"
-	"github.com/ultravioletrs/cocos/pkg/server"
-	grpcserver "github.com/ultravioletrs/cocos/pkg/server/grpc"
+	"go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc"
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+	"google.golang.org/grpc/health"
+	"google.golang.org/grpc/health/grpc_health_v1"
 	"google.golang.org/grpc/reflection"
 )
 
 const (
-	svcName        = "agent"
-	defSvcGRPCPort = "7002"
+	svcName          = "agent"
+	defSvcGRPCSocket = "/run/cocos/agent.sock"
 )
 
 type AgentServer interface {
@@ -29,59 +32,76 @@ type AgentServer interface {
 }
 
 type agentServer struct {
-	gs           server.Server
-	logger       *slog.Logger
-	svc          agent.Service
-	host         string
-	certProvider atls.CertificateProvider
+	mu     sync.Mutex
+	gs     *grpc.Server
+	logger *slog.Logger
+	svc    agent.Service
+	host   string
 }
 
-func NewServer(logger *slog.Logger, svc agent.Service, host string, certProvider atls.CertificateProvider) AgentServer {
+func NewServer(logger *slog.Logger, svc agent.Service, host string) AgentServer {
 	return &agentServer{
-		logger:       logger,
-		svc:          svc,
-		host:         host,
-		certProvider: certProvider,
+		logger: logger,
+		svc:    svc,
+		host:   host,
 	}
 }
 
 func (as *agentServer) Start(cfg agent.AgentConfig, cmp agent.Computation) error {
-	if cfg.Port == "" {
-		cfg.Port = defSvcGRPCPort
-	}
-
-	agentGrpcServerConfig := server.AgentConfig{
-		ServerConfig: server.ServerConfig{
-			Config: server.Config{
-				Host:         as.host,
-				Port:         cfg.Port,
-				CertFile:     cfg.CertFile,
-				KeyFile:      cfg.KeyFile,
-				ServerCAFile: cfg.ServerCAFile,
-				ClientCAFile: cfg.ClientCAFile,
-			},
-		},
-		AttestedTLS: cfg.AttestedTls,
-	}
-
-	registerAgentServiceServer := func(srv *grpc.Server) {
-		reflection.Register(srv)
-		agent.RegisterAgentServiceServer(srv, agentgrpc.NewServer(as.svc))
-	}
-
 	authSvc, err := auth.New(cmp)
 	if err != nil {
 		as.logger.WithGroup(cmp.ID).Error(fmt.Sprintf("failed to create auth service %s", err.Error()))
 		return err
 	}
 
-	ctx, cancel := context.WithCancel(context.Background())
+	grpcServerOptions := []grpc.ServerOption{
+		grpc.StatsHandler(otelgrpc.NewServerHandler()),
+	}
 
-	as.gs = grpcserver.New(ctx, cancel, svcName, agentGrpcServerConfig, registerAgentServiceServer, as.logger, authSvc, as.certProvider)
+	// Add authentication interceptors
+	unary, stream := agentgrpc.NewAuthInterceptor(authSvc)
+	grpcServerOptions = append(grpcServerOptions, grpc.UnaryInterceptor(unary))
+	grpcServerOptions = append(grpcServerOptions, grpc.StreamInterceptor(stream))
+
+	// Internal Unix socket is pure plaintext HTTP/2; Ingress Proxy handles external aTLS termination
+	grpcServerOptions = append(grpcServerOptions, grpc.Creds(insecure.NewCredentials()))
+
+	as.mu.Lock()
+	as.gs = grpc.NewServer(grpcServerOptions...)
+	gs := as.gs
+	as.mu.Unlock()
+
+	reflection.Register(gs)
+	agent.RegisterAgentServiceServer(gs, agentgrpc.NewServer(as.svc))
+
+	healthServer := health.NewServer()
+	healthServer.SetServingStatus("agent", grpc_health_v1.HealthCheckResponse_SERVING)
+	grpc_health_v1.RegisterHealthServer(gs, healthServer)
+
+	socketPath := as.host
+	if socketPath == "" || socketPath == "0.0.0.0" {
+		socketPath = defSvcGRPCSocket
+	}
+
+	var listener net.Listener
+	if socketPath[0] == '/' || socketPath[0] == '.' {
+		// Remove existing socket file if it exists
+		_ = os.Remove(socketPath)
+		listener, err = net.Listen("unix", socketPath)
+	} else {
+		listener, err = net.Listen("tcp", socketPath)
+	}
+
+	if err != nil {
+		as.logger.Error(fmt.Sprintf("failed to listen on %s: %s", socketPath, err))
+		return err
+	}
+
+	as.logger.Info(fmt.Sprintf("agent service gRPC server listening at %s without TLS", socketPath))
 
 	go func() {
-		err := as.gs.Start()
-		if err != nil {
+		err := gs.Serve(listener)
+		if err != nil && err != grpc.ErrServerStopped {
 			as.logger.Error(fmt.Sprintf("failed to start grpc server %s", err.Error()))
 		}
 	}()
@@ -90,8 +110,10 @@ func (as *agentServer) Start(cfg agent.AgentConfig, cmp agent.Computation) error
 }
 
 func (as *agentServer) Stop() error {
-	if as.gs == nil {
-		return nil
+	as.mu.Lock()
+	defer as.mu.Unlock()
+	if as.gs != nil {
+		as.gs.GracefulStop()
 	}
-	return as.gs.Stop()
+	return nil
 }

agent/cvms/server/cvm_test.go

@@ -21,7 +21,7 @@ import (
 func setupTest(t *testing.T) (*slog.Logger, *mocks.Service, string, []byte) {
 	logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelError}))
 	mockSvc := new(mocks.Service)
-	host := "localhost"
+	host := "localhost:0"
 
 	privateKey, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
 	assert.NoError(t, err, "Failed to generate ECDSA key")
@@ -70,7 +70,7 @@ func TestNewServer(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			server := NewServer(tt.logger, tt.svc, tt.host, nil)
+			server := NewServer(tt.logger, tt.svc, tt.host)
 
 			assert.NotNil(t, server)
 
@@ -97,7 +97,6 @@ func TestAgentServer_Start(t *testing.T) {
 		{
 			name: "successful start with default port",
 			cfg: agent.AgentConfig{
-				Port:         "",
 				CertFile:     "cert.pem",
 				KeyFile:      "key.pem",
 				ServerCAFile: "server-ca.pem",
@@ -108,7 +107,7 @@ func TestAgentServer_Start(t *testing.T) {
 				ID:          "test-computation-1",
 				Name:        "Test Computation",
 				Description: "A test computation",
-				Algorithm: agent.Algorithm{
+				Algorithm: &agent.Algorithm{
 					Hash:    [32]byte{0x01, 0x02, 0x03},
 					UserKey: pubKey,
 				},
@@ -131,7 +130,6 @@ func TestAgentServer_Start(t *testing.T) {
 		{
 			name: "successful start with custom port",
 			cfg: agent.AgentConfig{
-				Port:         "8080",
 				CertFile:     "cert.pem",
 				KeyFile:      "key.pem",
 				ServerCAFile: "server-ca.pem",
@@ -142,7 +140,7 @@ func TestAgentServer_Start(t *testing.T) {
 				ID:          "test-computation-2",
 				Name:        "Test Computation 2",
 				Description: "Another test computation",
-				Algorithm: agent.Algorithm{
+				Algorithm: &agent.Algorithm{
 					Hash:    [32]byte{0x07, 0x08, 0x09},
 					UserKey: pubKey,
 				},
@@ -165,13 +163,12 @@ func TestAgentServer_Start(t *testing.T) {
 		{
 			name: "start with minimal config",
 			cfg: agent.AgentConfig{
-				Port:        "9090",
 				AttestedTls: false,
 			},
 			cmp: agent.Computation{
 				ID:   "test-computation-3",
 				Name: "Minimal Test",
-				Algorithm: agent.Algorithm{
+				Algorithm: &agent.Algorithm{
 					Hash:    [32]byte{0x0d, 0x0e, 0x0f},
 					UserKey: pubKey,
 				},
@@ -197,7 +194,7 @@ func TestAgentServer_Start(t *testing.T) {
 		t.Run(tt.name, func(t *testing.T) {
 			tt.setupMocks(svc)
 
-			server := NewServer(logger, svc, host, nil)
+			server := NewServer(logger, svc, host)
 
 			err := server.Start(tt.cfg, tt.cmp)
 
@@ -243,13 +240,11 @@ func TestAgentServer_Stop(t *testing.T) {
 		{
 			name: "stop started server",
 			setupServer: func(server AgentServer) error {
-				cfg := agent.AgentConfig{
-					Port: "7004",
-				}
+				cfg := agent.AgentConfig{}
 				cmp := agent.Computation{
 					ID:   "test-stop-computation",
 					Name: "Stop Test",
-					Algorithm: agent.Algorithm{
+					Algorithm: &agent.Algorithm{
 						Hash:    [32]byte{0x19, 0x1a, 0x1b},
 						UserKey: pubKey,
 					},
@@ -273,7 +268,7 @@ func TestAgentServer_Stop(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			server := NewServer(logger, svc, host, nil)
+			server := NewServer(logger, svc, host)
 
 			err := tt.setupServer(server)
 			if err != nil {
@@ -301,14 +296,14 @@ func TestAgentServer_Stop(t *testing.T) {
 
 func TestAgentServer_StopMultipleTimes(t *testing.T) {
 	logger, svc, host, pubKey := setupTest(t)
-	server := NewServer(logger, svc, host, nil)
+	server := NewServer(logger, svc, host)
 
 	// Start the server
-	cfg := agent.AgentConfig{Port: "7005"}
+	cfg := agent.AgentConfig{}
 	cmp := agent.Computation{
 		ID:   "test-multiple-stop",
 		Name: "Multiple Stop Test",
-		Algorithm: agent.Algorithm{
+		Algorithm: &agent.Algorithm{
 			Hash:    [32]byte{0x1f, 0x20, 0x21},
 			UserKey: pubKey,
 		},
@@ -345,13 +340,13 @@ func TestAgentServer_StopMultipleTimes(t *testing.T) {
 
 func TestAgentServer_StartAfterStop(t *testing.T) {
 	logger, svc, host, pubKey := setupTest(t)
-	server := NewServer(logger, svc, host, nil)
+	server := NewServer(logger, svc, host)
 
-	cfg := agent.AgentConfig{Port: "7006"}
+	cfg := agent.AgentConfig{}
 	cmp := agent.Computation{
 		ID:   "test-restart",
 		Name: "Restart Test",
-		Algorithm: agent.Algorithm{
+		Algorithm: &agent.Algorithm{
 			Hash:    [32]byte{0x25, 0x26, 0x27},
 			UserKey: pubKey,
 		},
@@ -378,11 +373,11 @@ func TestAgentServer_StartAfterStop(t *testing.T) {
 	assert.NoError(t, err)
 
 	// Start again with different config
-	cfg2 := agent.AgentConfig{Port: "7007"}
+	cfg2 := agent.AgentConfig{}
 	cmp2 := agent.Computation{
 		ID:   "test-restart-2",
 		Name: "Restart Test 2",
-		Algorithm: agent.Algorithm{
+		Algorithm: &agent.Algorithm{
 			Hash:    [32]byte{0x2b, 0x2c, 0x2d},
 			UserKey: pubKey,
 		},
@@ -422,7 +417,6 @@ func TestAgentServer_ConfigValidation(t *testing.T) {
 		{
 			name: "valid config with all fields",
 			config: agent.AgentConfig{
-				Port:         "8080",
 				CertFile:     "cert.pem",
 				KeyFile:      "key.pem",
 				ServerCAFile: "server-ca.pem",
@@ -432,7 +426,7 @@ func TestAgentServer_ConfigValidation(t *testing.T) {
 			cmp: agent.Computation{
 				ID:   "valid-config-test",
 				Name: "Valid Config Test",
-				Algorithm: agent.Algorithm{
+				Algorithm: &agent.Algorithm{
 					Hash:    [32]byte{0x31, 0x32, 0x33},
 					UserKey: pubKey,
 				},
@@ -451,14 +445,12 @@ func TestAgentServer_ConfigValidation(t *testing.T) {
 			valid: true,
 		},
 		{
-			name: "valid config with minimal fields",
-			config: agent.AgentConfig{
-				Port: "9090",
-			},
+			name:   "valid config with minimal fields",
+			config: agent.AgentConfig{},
 			cmp: agent.Computation{
 				ID:   "minimal-config-test",
 				Name: "Minimal Config Test",
-				Algorithm: agent.Algorithm{
+				Algorithm: &agent.Algorithm{
 					Hash:    [32]byte{0x37, 0x38, 0x39},
 					UserKey: pubKey,
 				},
@@ -477,14 +469,12 @@ func TestAgentServer_ConfigValidation(t *testing.T) {
 			valid: true,
 		},
 		{
-			name: "config with empty port uses default",
-			config: agent.AgentConfig{
-				Port: "",
-			},
+			name:   "config with empty port uses default",
+			config: agent.AgentConfig{},
 			cmp: agent.Computation{
 				ID:        "default-port-test",
 				Name:      "Default Port Test",
-				Algorithm: agent.Algorithm{Hash: [32]byte{0x3d, 0x3e, 0x3f}, UserKey: pubKey},
+				Algorithm: &agent.Algorithm{Hash: [32]byte{0x3d, 0x3e, 0x3f}, UserKey: pubKey},
 				Datasets: []agent.Dataset{
 					{Hash: [32]byte{0x40, 0x41, 0x42}, UserKey: pubKey},
 				},
@@ -498,18 +488,16 @@ func TestAgentServer_ConfigValidation(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			server := NewServer(logger, svc, host, nil)
+			server := NewServer(logger, svc, host)
 
 			err := server.Start(tt.config, tt.cmp)
 
 			if tt.valid {
 				assert.NoError(t, err)
 
-				// Verify default port is used when empty
-				if tt.config.Port == "" {
-					agentSrv := server.(*agentServer)
-					assert.NotNil(t, agentSrv.gs)
-				}
+				// Verify server started successfully
+				agentSrv := server.(*agentServer)
+				assert.NotNil(t, agentSrv.gs)
 
 				time.Sleep(10 * time.Millisecond)
 				if err := server.Stop(); err != nil {
@@ -526,5 +514,5 @@ func TestAgentServer_ConfigValidation(t *testing.T) {
 
 func TestConstants(t *testing.T) {
 	assert.Equal(t, "agent", svcName)
-	assert.Equal(t, "7002", defSvcGRPCPort)
+	assert.Equal(t, "/run/cocos/agent.sock", defSvcGRPCSocket)
 }

agent/events/events.pb.go

@@ -3,8 +3,8 @@
 
 // Code generated by protoc-gen-go. DO NOT EDIT.
 // versions:
-// 	protoc-gen-go v1.36.8
-// 	protoc        v5.29.0
+// 	protoc-gen-go v1.36.11
+// 	protoc        v6.33.1
 // source: agent/events/events.proto
 
 package events

agent/log/log.pb.go

@@ -0,0 +1,261 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+
+// Code generated by protoc-gen-go. DO NOT EDIT.
+// versions:
+// 	protoc-gen-go v1.36.11
+// 	protoc        v6.33.1
+// source: agent/log/log.proto
+
+package log
+
+import (
+	protoreflect "google.golang.org/protobuf/reflect/protoreflect"
+	protoimpl "google.golang.org/protobuf/runtime/protoimpl"
+	emptypb "google.golang.org/protobuf/types/known/emptypb"
+	timestamppb "google.golang.org/protobuf/types/known/timestamppb"
+	reflect "reflect"
+	sync "sync"
+	unsafe "unsafe"
+)
+
+const (
+	// Verify that this generated code is sufficiently up-to-date.
+	_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
+	// Verify that runtime/protoimpl is sufficiently up-to-date.
+	_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
+)
+
+type LogEntry struct {
+	state         protoimpl.MessageState `protogen:"open.v1"`
+	Message       string                 `protobuf:"bytes,1,opt,name=message,proto3" json:"message,omitempty"`
+	ComputationId string                 `protobuf:"bytes,2,opt,name=computation_id,json=computationId,proto3" json:"computation_id,omitempty"`
+	Level         string                 `protobuf:"bytes,3,opt,name=level,proto3" json:"level,omitempty"`
+	Timestamp     *timestamppb.Timestamp `protobuf:"bytes,4,opt,name=timestamp,proto3" json:"timestamp,omitempty"`
+	unknownFields protoimpl.UnknownFields
+	sizeCache     protoimpl.SizeCache
+}
+
+func (x *LogEntry) Reset() {
+	*x = LogEntry{}
+	mi := &file_agent_log_log_proto_msgTypes[0]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *LogEntry) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*LogEntry) ProtoMessage() {}
+
+func (x *LogEntry) ProtoReflect() protoreflect.Message {
+	mi := &file_agent_log_log_proto_msgTypes[0]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use LogEntry.ProtoReflect.Descriptor instead.
+func (*LogEntry) Descriptor() ([]byte, []int) {
+	return file_agent_log_log_proto_rawDescGZIP(), []int{0}
+}
+
+func (x *LogEntry) GetMessage() string {
+	if x != nil {
+		return x.Message
+	}
+	return ""
+}
+
+func (x *LogEntry) GetComputationId() string {
+	if x != nil {
+		return x.ComputationId
+	}
+	return ""
+}
+
+func (x *LogEntry) GetLevel() string {
+	if x != nil {
+		return x.Level
+	}
+	return ""
+}
+
+func (x *LogEntry) GetTimestamp() *timestamppb.Timestamp {
+	if x != nil {
+		return x.Timestamp
+	}
+	return nil
+}
+
+type EventEntry struct {
+	state         protoimpl.MessageState `protogen:"open.v1"`
+	EventType     string                 `protobuf:"bytes,1,opt,name=event_type,json=eventType,proto3" json:"event_type,omitempty"`
+	Timestamp     *timestamppb.Timestamp `protobuf:"bytes,2,opt,name=timestamp,proto3" json:"timestamp,omitempty"`
+	ComputationId string                 `protobuf:"bytes,3,opt,name=computation_id,json=computationId,proto3" json:"computation_id,omitempty"`
+	Details       []byte                 `protobuf:"bytes,4,opt,name=details,proto3" json:"details,omitempty"` // JSON payload
+	Originator    string                 `protobuf:"bytes,5,opt,name=originator,proto3" json:"originator,omitempty"`
+	Status        string                 `protobuf:"bytes,6,opt,name=status,proto3" json:"status,omitempty"`
+	unknownFields protoimpl.UnknownFields
+	sizeCache     protoimpl.SizeCache
+}
+
+func (x *EventEntry) Reset() {
+	*x = EventEntry{}
+	mi := &file_agent_log_log_proto_msgTypes[1]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *EventEntry) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*EventEntry) ProtoMessage() {}
+
+func (x *EventEntry) ProtoReflect() protoreflect.Message {
+	mi := &file_agent_log_log_proto_msgTypes[1]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use EventEntry.ProtoReflect.Descriptor instead.
+func (*EventEntry) Descriptor() ([]byte, []int) {
+	return file_agent_log_log_proto_rawDescGZIP(), []int{1}
+}
+
+func (x *EventEntry) GetEventType() string {
+	if x != nil {
+		return x.EventType
+	}
+	return ""
+}
+
+func (x *EventEntry) GetTimestamp() *timestamppb.Timestamp {
+	if x != nil {
+		return x.Timestamp
+	}
+	return nil
+}
+
+func (x *EventEntry) GetComputationId() string {
+	if x != nil {
+		return x.ComputationId
+	}
+	return ""
+}
+
+func (x *EventEntry) GetDetails() []byte {
+	if x != nil {
+		return x.Details
+	}
+	return nil
+}
+
+func (x *EventEntry) GetOriginator() string {
+	if x != nil {
+		return x.Originator
+	}
+	return ""
+}
+
+func (x *EventEntry) GetStatus() string {
+	if x != nil {
+		return x.Status
+	}
+	return ""
+}
+
+var File_agent_log_log_proto protoreflect.FileDescriptor
+
+const file_agent_log_log_proto_rawDesc = "" +
+	"\n" +
+	"\x13agent/log/log.proto\x12\x03log\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1bgoogle/protobuf/empty.proto\"\x9b\x01\n" +
+	"\bLogEntry\x12\x18\n" +
+	"\amessage\x18\x01 \x01(\tR\amessage\x12%\n" +
+	"\x0ecomputation_id\x18\x02 \x01(\tR\rcomputationId\x12\x14\n" +
+	"\x05level\x18\x03 \x01(\tR\x05level\x128\n" +
+	"\ttimestamp\x18\x04 \x01(\v2\x1a.google.protobuf.TimestampR\ttimestamp\"\xde\x01\n" +
+	"\n" +
+	"EventEntry\x12\x1d\n" +
+	"\n" +
+	"event_type\x18\x01 \x01(\tR\teventType\x128\n" +
+	"\ttimestamp\x18\x02 \x01(\v2\x1a.google.protobuf.TimestampR\ttimestamp\x12%\n" +
+	"\x0ecomputation_id\x18\x03 \x01(\tR\rcomputationId\x12\x18\n" +
+	"\adetails\x18\x04 \x01(\fR\adetails\x12\x1e\n" +
+	"\n" +
+	"originator\x18\x05 \x01(\tR\n" +
+	"originator\x12\x16\n" +
+	"\x06status\x18\x06 \x01(\tR\x06status2v\n" +
+	"\fLogCollector\x120\n" +
+	"\aSendLog\x12\r.log.LogEntry\x1a\x16.google.protobuf.Empty\x124\n" +
+	"\tSendEvent\x12\x0f.log.EventEntry\x1a\x16.google.protobuf.EmptyB\aZ\x05./logb\x06proto3"
+
+var (
+	file_agent_log_log_proto_rawDescOnce sync.Once
+	file_agent_log_log_proto_rawDescData []byte
+)
+
+func file_agent_log_log_proto_rawDescGZIP() []byte {
+	file_agent_log_log_proto_rawDescOnce.Do(func() {
+		file_agent_log_log_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_agent_log_log_proto_rawDesc), len(file_agent_log_log_proto_rawDesc)))
+	})
+	return file_agent_log_log_proto_rawDescData
+}
+
+var file_agent_log_log_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
+var file_agent_log_log_proto_goTypes = []any{
+	(*LogEntry)(nil),              // 0: log.LogEntry
+	(*EventEntry)(nil),            // 1: log.EventEntry
+	(*timestamppb.Timestamp)(nil), // 2: google.protobuf.Timestamp
+	(*emptypb.Empty)(nil),         // 3: google.protobuf.Empty
+}
+var file_agent_log_log_proto_depIdxs = []int32{
+	2, // 0: log.LogEntry.timestamp:type_name -> google.protobuf.Timestamp
+	2, // 1: log.EventEntry.timestamp:type_name -> google.protobuf.Timestamp
+	0, // 2: log.LogCollector.SendLog:input_type -> log.LogEntry
+	1, // 3: log.LogCollector.SendEvent:input_type -> log.EventEntry
+	3, // 4: log.LogCollector.SendLog:output_type -> google.protobuf.Empty
+	3, // 5: log.LogCollector.SendEvent:output_type -> google.protobuf.Empty
+	4, // [4:6] is the sub-list for method output_type
+	2, // [2:4] is the sub-list for method input_type
+	2, // [2:2] is the sub-list for extension type_name
+	2, // [2:2] is the sub-list for extension extendee
+	0, // [0:2] is the sub-list for field type_name
+}
+
+func init() { file_agent_log_log_proto_init() }
+func file_agent_log_log_proto_init() {
+	if File_agent_log_log_proto != nil {
+		return
+	}
+	type x struct{}
+	out := protoimpl.TypeBuilder{
+		File: protoimpl.DescBuilder{
+			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
+			RawDescriptor: unsafe.Slice(unsafe.StringData(file_agent_log_log_proto_rawDesc), len(file_agent_log_log_proto_rawDesc)),
+			NumEnums:      0,
+			NumMessages:   2,
+			NumExtensions: 0,
+			NumServices:   1,
+		},
+		GoTypes:           file_agent_log_log_proto_goTypes,
+		DependencyIndexes: file_agent_log_log_proto_depIdxs,
+		MessageInfos:      file_agent_log_log_proto_msgTypes,
+	}.Build()
+	File_agent_log_log_proto = out.File
+	file_agent_log_log_proto_goTypes = nil
+	file_agent_log_log_proto_depIdxs = nil
+}

agent/log/log.proto

@@ -0,0 +1,32 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+
+syntax = "proto3";
+
+package log;
+
+option go_package = "./log";
+
+import "google/protobuf/timestamp.proto";
+import "google/protobuf/empty.proto";
+
+service LogCollector {
+  rpc SendLog(LogEntry) returns (google.protobuf.Empty);
+  rpc SendEvent(EventEntry) returns (google.protobuf.Empty);
+}
+
+message LogEntry {
+  string message = 1;
+  string computation_id = 2;
+  string level = 3;
+  google.protobuf.Timestamp timestamp = 4;
+}
+
+message EventEntry {
+  string event_type = 1;
+  google.protobuf.Timestamp timestamp = 2;
+  string computation_id = 3;
+  bytes details = 4; // JSON payload
+  string originator = 5;
+  string status = 6;
+}

agent/log/log_grpc.pb.go

@@ -0,0 +1,163 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+
+// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
+// versions:
+// - protoc-gen-go-grpc v1.6.0
+// - protoc             v6.33.1
+// source: agent/log/log.proto
+
+package log
+
+import (
+	context "context"
+	grpc "google.golang.org/grpc"
+	codes "google.golang.org/grpc/codes"
+	status "google.golang.org/grpc/status"
+	emptypb "google.golang.org/protobuf/types/known/emptypb"
+)
+
+// This is a compile-time assertion to ensure that this generated file
+// is compatible with the grpc package it is being compiled against.
+// Requires gRPC-Go v1.64.0 or later.
+const _ = grpc.SupportPackageIsVersion9
+
+const (
+	LogCollector_SendLog_FullMethodName   = "/log.LogCollector/SendLog"
+	LogCollector_SendEvent_FullMethodName = "/log.LogCollector/SendEvent"
+)
+
+// LogCollectorClient is the client API for LogCollector service.
+//
+// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
+type LogCollectorClient interface {
+	SendLog(ctx context.Context, in *LogEntry, opts ...grpc.CallOption) (*emptypb.Empty, error)
+	SendEvent(ctx context.Context, in *EventEntry, opts ...grpc.CallOption) (*emptypb.Empty, error)
+}
+
+type logCollectorClient struct {
+	cc grpc.ClientConnInterface
+}
+
+func NewLogCollectorClient(cc grpc.ClientConnInterface) LogCollectorClient {
+	return &logCollectorClient{cc}
+}
+
+func (c *logCollectorClient) SendLog(ctx context.Context, in *LogEntry, opts ...grpc.CallOption) (*emptypb.Empty, error) {
+	cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
+	out := new(emptypb.Empty)
+	err := c.cc.Invoke(ctx, LogCollector_SendLog_FullMethodName, in, out, cOpts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+func (c *logCollectorClient) SendEvent(ctx context.Context, in *EventEntry, opts ...grpc.CallOption) (*emptypb.Empty, error) {
+	cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
+	out := new(emptypb.Empty)
+	err := c.cc.Invoke(ctx, LogCollector_SendEvent_FullMethodName, in, out, cOpts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+// LogCollectorServer is the server API for LogCollector service.
+// All implementations must embed UnimplementedLogCollectorServer
+// for forward compatibility.
+type LogCollectorServer interface {
+	SendLog(context.Context, *LogEntry) (*emptypb.Empty, error)
+	SendEvent(context.Context, *EventEntry) (*emptypb.Empty, error)
+	mustEmbedUnimplementedLogCollectorServer()
+}
+
+// UnimplementedLogCollectorServer must be embedded to have
+// forward compatible implementations.
+//
+// NOTE: this should be embedded by value instead of pointer to avoid a nil
+// pointer dereference when methods are called.
+type UnimplementedLogCollectorServer struct{}
+
+func (UnimplementedLogCollectorServer) SendLog(context.Context, *LogEntry) (*emptypb.Empty, error) {
+	return nil, status.Error(codes.Unimplemented, "method SendLog not implemented")
+}
+func (UnimplementedLogCollectorServer) SendEvent(context.Context, *EventEntry) (*emptypb.Empty, error) {
+	return nil, status.Error(codes.Unimplemented, "method SendEvent not implemented")
+}
+func (UnimplementedLogCollectorServer) mustEmbedUnimplementedLogCollectorServer() {}
+func (UnimplementedLogCollectorServer) testEmbeddedByValue()                      {}
+
+// UnsafeLogCollectorServer may be embedded to opt out of forward compatibility for this service.
+// Use of this interface is not recommended, as added methods to LogCollectorServer will
+// result in compilation errors.
+type UnsafeLogCollectorServer interface {
+	mustEmbedUnimplementedLogCollectorServer()
+}
+
+func RegisterLogCollectorServer(s grpc.ServiceRegistrar, srv LogCollectorServer) {
+	// If the following call panics, it indicates UnimplementedLogCollectorServer was
+	// embedded by pointer and is nil.  This will cause panics if an
+	// unimplemented method is ever invoked, so we test this at initialization
+	// time to prevent it from happening at runtime later due to I/O.
+	if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
+		t.testEmbeddedByValue()
+	}
+	s.RegisterService(&LogCollector_ServiceDesc, srv)
+}
+
+func _LogCollector_SendLog_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(LogEntry)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(LogCollectorServer).SendLog(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: LogCollector_SendLog_FullMethodName,
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(LogCollectorServer).SendLog(ctx, req.(*LogEntry))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+func _LogCollector_SendEvent_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(EventEntry)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(LogCollectorServer).SendEvent(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: LogCollector_SendEvent_FullMethodName,
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(LogCollectorServer).SendEvent(ctx, req.(*EventEntry))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+// LogCollector_ServiceDesc is the grpc.ServiceDesc for LogCollector service.
+// It's only intended for direct use with grpc.RegisterService,
+// and not to be introspected or modified (even as a copy)
+var LogCollector_ServiceDesc = grpc.ServiceDesc{
+	ServiceName: "log.LogCollector",
+	HandlerType: (*LogCollectorServer)(nil),
+	Methods: []grpc.MethodDesc{
+		{
+			MethodName: "SendLog",
+			Handler:    _LogCollector_SendLog_Handler,
+		},
+		{
+			MethodName: "SendEvent",
+			Handler:    _LogCollector_SendEvent_Handler,
+		},
+	},
+	Streams:  []grpc.StreamDesc{},
+	Metadata: "agent/log/log.proto",
+}

agent/log/service/service.go

@@ -0,0 +1,59 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package service
+
+import (
+	"context"
+	"log/slog"
+
+	"github.com/ultravioletrs/cocos/agent/cvms"
+	"github.com/ultravioletrs/cocos/agent/log"
+	"google.golang.org/protobuf/types/known/emptypb"
+)
+
+var _ log.LogCollectorServer = (*LogForwarder)(nil)
+
+type LogForwarder struct {
+	log.UnimplementedLogCollectorServer
+	cvmsClient cvms.ServiceClient
+	logger     *slog.Logger
+	logQueue   chan *cvms.ClientStreamMessage
+}
+
+func New(logger *slog.Logger, cvmsClient cvms.ServiceClient, queue chan *cvms.ClientStreamMessage) *LogForwarder {
+	return &LogForwarder{
+		cvmsClient: cvmsClient,
+		logger:     logger,
+		logQueue:   queue,
+	}
+}
+
+func (s *LogForwarder) SendLog(ctx context.Context, req *log.LogEntry) (*emptypb.Empty, error) {
+	s.logQueue <- &cvms.ClientStreamMessage{
+		Message: &cvms.ClientStreamMessage_AgentLog{
+			AgentLog: &cvms.AgentLog{
+				Message:       req.Message,
+				ComputationId: req.ComputationId,
+				Level:         req.Level,
+				Timestamp:     req.Timestamp,
+			},
+		},
+	}
+	return &emptypb.Empty{}, nil
+}
+
+func (s *LogForwarder) SendEvent(ctx context.Context, req *log.EventEntry) (*emptypb.Empty, error) {
+	s.logQueue <- &cvms.ClientStreamMessage{
+		Message: &cvms.ClientStreamMessage_AgentEvent{
+			AgentEvent: &cvms.AgentEvent{
+				EventType:     req.EventType,
+				Timestamp:     req.Timestamp,
+				ComputationId: req.ComputationId,
+				Details:       req.Details,
+				Originator:    req.Originator,
+				Status:        req.Status,
+			},
+		},
+	}
+	return &emptypb.Empty{}, nil
+}

agent/log/service/service_test.go

@@ -0,0 +1,303 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package service
+
+import (
+	"context"
+	"encoding/json"
+	"log/slog"
+	"os"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"github.com/ultravioletrs/cocos/agent/cvms"
+	"github.com/ultravioletrs/cocos/agent/log"
+	"google.golang.org/protobuf/types/known/timestamppb"
+)
+
+// TestNewLogForwarder tests the creation of a new log forwarder.
+func TestNewLogForwarder(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 10)
+
+	lf := New(logger, nil, queue)
+	require.NotNil(t, lf)
+	assert.NotNil(t, lf.logger)
+	assert.Nil(t, lf.cvmsClient)
+	assert.NotNil(t, lf.logQueue)
+}
+
+// TestSendLog tests sending a log entry.
+func TestSendLog(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 10)
+
+	lf := New(logger, nil, queue)
+
+	req := &log.LogEntry{
+		Message:       "Test log message",
+		ComputationId: "computation-1",
+		Level:         "INFO",
+		Timestamp:     timestamppb.New(time.Now()),
+	}
+
+	resp, err := lf.SendLog(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+
+	// Verify message was queued
+	select {
+	case msg := <-queue:
+		require.NotNil(t, msg)
+		agentLog := msg.GetAgentLog()
+		assert.NotNil(t, agentLog)
+		assert.Equal(t, "Test log message", agentLog.Message)
+		assert.Equal(t, "computation-1", agentLog.ComputationId)
+		assert.Equal(t, "INFO", agentLog.Level)
+	default:
+		t.Fatal("No message in queue")
+	}
+}
+
+// TestSendEvent tests sending an event entry.
+func TestSendEvent(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 10)
+
+	lf := New(logger, nil, queue)
+
+	details, err := json.Marshal(map[string]string{"key": "value"})
+	require.NoError(t, err)
+
+	req := &log.EventEntry{
+		EventType:     "COMPUTATION_STARTED",
+		Timestamp:     timestamppb.New(time.Now()),
+		ComputationId: "computation-1",
+		Details:       details,
+		Originator:    "runner",
+		Status:        "SUCCESS",
+	}
+
+	resp, err := lf.SendEvent(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+
+	// Verify message was queued
+	select {
+	case msg := <-queue:
+		require.NotNil(t, msg)
+		agentEvent := msg.GetAgentEvent()
+		assert.NotNil(t, agentEvent)
+		assert.Equal(t, "COMPUTATION_STARTED", agentEvent.EventType)
+		assert.Equal(t, "computation-1", agentEvent.ComputationId)
+		assert.Equal(t, "runner", agentEvent.Originator)
+		assert.Equal(t, "SUCCESS", agentEvent.Status)
+	default:
+		t.Fatal("No message in queue")
+	}
+}
+
+// TestSendMultipleLogs tests sending multiple log entries.
+func TestSendMultipleLogs(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 100)
+
+	lf := New(logger, nil, queue)
+
+	for i := 0; i < 5; i++ {
+		req := &log.LogEntry{
+			Message:       "Log message",
+			ComputationId: "computation-1",
+			Level:         "INFO",
+			Timestamp:     timestamppb.New(time.Now()),
+		}
+
+		resp, err := lf.SendLog(context.Background(), req)
+		require.NoError(t, err)
+		require.NotNil(t, resp)
+	}
+
+	assert.Equal(t, 5, len(queue))
+}
+
+// TestSendEventWithVariousTypes tests sending events with different types.
+func TestSendEventWithVariousTypes(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 100)
+
+	lf := New(logger, nil, queue)
+
+	eventTypes := []string{"STARTED", "RUNNING", "COMPLETED", "FAILED"}
+	for _, eventType := range eventTypes {
+		details, err := json.Marshal(map[string]string{"type": eventType})
+		require.NoError(t, err)
+		req := &log.EventEntry{
+			EventType:     eventType,
+			Timestamp:     timestamppb.New(time.Now()),
+			ComputationId: "computation-1",
+			Details:       details,
+			Originator:    "runner",
+			Status:        "OK",
+		}
+
+		resp, err := lf.SendEvent(context.Background(), req)
+		require.NoError(t, err)
+		require.NotNil(t, resp)
+	}
+
+	assert.Equal(t, 4, len(queue))
+}
+
+// TestSendLogWithEmptyMessage tests sending log with empty message.
+func TestSendLogWithEmptyMessage(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 10)
+
+	lf := New(logger, nil, queue)
+
+	req := &log.LogEntry{
+		Message:       "",
+		ComputationId: "computation-1",
+		Level:         "INFO",
+		Timestamp:     timestamppb.New(time.Now()),
+	}
+
+	resp, err := lf.SendLog(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+
+	select {
+	case msg := <-queue:
+		agentLog := msg.GetAgentLog()
+		assert.Equal(t, "", agentLog.Message)
+	default:
+		t.Fatal("No message in queue")
+	}
+}
+
+// TestSendEventWithNilDetails tests sending event with nil details.
+func TestSendEventWithNilDetails(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 10)
+
+	lf := New(logger, nil, queue)
+
+	req := &log.EventEntry{
+		EventType:     "TEST_EVENT",
+		Timestamp:     timestamppb.New(time.Now()),
+		ComputationId: "computation-1",
+		Details:       nil,
+		Originator:    "test",
+		Status:        "OK",
+	}
+
+	resp, err := lf.SendEvent(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+
+	select {
+	case msg := <-queue:
+		agentEvent := msg.GetAgentEvent()
+		assert.Nil(t, agentEvent.Details)
+	default:
+		t.Fatal("No message in queue")
+	}
+}
+
+// TestSendLogWithVariousLevels tests sending logs with various severity levels.
+func TestSendLogWithVariousLevels(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 100)
+
+	lf := New(logger, nil, queue)
+
+	levels := []string{"DEBUG", "INFO", "WARN", "ERROR"}
+	for _, level := range levels {
+		req := &log.LogEntry{
+			Message:       "Test " + level,
+			ComputationId: "computation-1",
+			Level:         level,
+			Timestamp:     timestamppb.New(time.Now()),
+		}
+
+		resp, err := lf.SendLog(context.Background(), req)
+		require.NoError(t, err)
+		require.NotNil(t, resp)
+	}
+
+	assert.Equal(t, 4, len(queue))
+}
+
+// TestSendLogWithDifferentComputationIds tests sending logs with different computation IDs.
+func TestSendLogWithDifferentComputationIds(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 100)
+
+	lf := New(logger, nil, queue)
+
+	for i := 0; i < 3; i++ {
+		req := &log.LogEntry{
+			Message:       "Message",
+			ComputationId: "computation-" + string(rune(48+i)),
+			Level:         "INFO",
+			Timestamp:     timestamppb.New(time.Now()),
+		}
+
+		resp, err := lf.SendLog(context.Background(), req)
+		require.NoError(t, err)
+		require.NotNil(t, resp)
+	}
+
+	assert.Equal(t, 3, len(queue))
+}
+
+// TestQueueBehavior tests that queue is properly used.
+func TestQueueBehavior(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 1)
+
+	lf := New(logger, nil, queue)
+
+	req := &log.LogEntry{
+		Message:       "Test",
+		ComputationId: "computation-1",
+		Level:         "INFO",
+		Timestamp:     timestamppb.New(time.Now()),
+	}
+
+	resp, err := lf.SendLog(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+
+	assert.Equal(t, 1, len(queue))
+}
+
+// TestConcurrentSendLog tests concurrent log sending.
+func TestConcurrentSendLog(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	queue := make(chan *cvms.ClientStreamMessage, 100)
+
+	lf := New(logger, nil, queue)
+
+	for i := 0; i < 10; i++ {
+		go func(id int) {
+			req := &log.LogEntry{
+				Message:       "Concurrent log",
+				ComputationId: "computation-1",
+				Level:         "INFO",
+				Timestamp:     timestamppb.New(time.Now()),
+			}
+
+			_, err := lf.SendLog(context.Background(), req)
+			require.NoError(t, err)
+		}(i)
+	}
+
+	// Give goroutines time to complete
+	time.Sleep(100 * time.Millisecond)
+
+	// Should have received all messages
+	assert.True(t, len(queue) > 0)
+}

agent/mock_attestation_client_test.go

@@ -0,0 +1,34 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package agent
+
+import (
+	"context"
+
+	"github.com/stretchr/testify/mock"
+	"github.com/ultravioletrs/cocos/pkg/attestation"
+)
+
+type MockAttestationClient struct {
+	mock.Mock
+}
+
+func (m *MockAttestationClient) GetAttestation(ctx context.Context, reportData [64]byte, nonce [32]byte, attType attestation.PlatformType) ([]byte, error) {
+	args := m.Called(ctx, reportData, nonce, attType)
+	return args.Get(0).([]byte), args.Error(1)
+}
+
+func (m *MockAttestationClient) GetRawEvidence(ctx context.Context, reportData [64]byte, nonce [32]byte, attType attestation.PlatformType) ([]byte, error) {
+	args := m.Called(ctx, reportData, nonce, attType)
+	return args.Get(0).([]byte), args.Error(1)
+}
+
+func (m *MockAttestationClient) GetAzureToken(ctx context.Context, nonce [32]byte) ([]byte, error) {
+	args := m.Called(ctx, nonce)
+	return args.Get(0).([]byte), args.Error(1)
+}
+
+func (m *MockAttestationClient) Close() error {
+	args := m.Called()
+	return args.Error(0)
+}

agent/resource_test.go

@@ -0,0 +1,192 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+
+package agent
+
+import (
+	"context"
+	"crypto/aes"
+	"crypto/cipher"
+	"crypto/rand"
+	"io"
+	"log/slog"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+	"github.com/ultravioletrs/cocos/pkg/resource"
+)
+
+type MockDownloader struct {
+	mock.Mock
+}
+
+func (m *MockDownloader) Download(ctx context.Context, url string, destPath string) error {
+	args := m.Called(ctx, url, destPath)
+	if args.Error(0) == nil {
+		// Simulate writing to destPath if it's a success
+		content := "mock content"
+		if len(args) > 1 {
+			if c, ok := args.Get(1).(string); ok {
+				content = c
+			}
+		}
+		_ = os.MkdirAll(filepath.Dir(destPath), 0o755)
+		_ = os.WriteFile(destPath, []byte(content), 0o644)
+	}
+	return args.Error(0)
+}
+
+func (m *MockDownloader) Type() string {
+	return m.Called().String(0)
+}
+
+func TestDownloadAndDecryptGenericResource(t *testing.T) {
+	registry := resource.NewRegistry()
+	mockDownloader := new(MockDownloader)
+	mockDownloader.On("Type").Return(resource.SourceTypeHTTP)
+	registry.Register(mockDownloader)
+
+	svc := &agentService{
+		logger:           slog.Default(),
+		resourceRegistry: registry,
+		computation: Computation{
+			Algorithm: &Algorithm{
+				KBS: &KBSConfig{
+					Enabled: true,
+					URL:     "http://mock-kbs",
+				},
+			},
+		},
+	}
+
+	ctx := context.Background()
+
+	t.Run("Successful download without encryption", func(t *testing.T) {
+		source := &ResourceSource{
+			URL: "http://example.com/resource",
+		}
+		destPath := filepath.Join(os.TempDir(), "cocos-resources", "algo", "resource")
+		mockDownloader.On("Download", ctx, source.URL, destPath).Return(nil, "some data").Once()
+
+		res, err := svc.downloadAndDecryptGenericResource(ctx, source, resource.SourceTypeHTTP, "", "algo")
+		assert.NoError(t, err)
+		assert.Equal(t, []byte("some data"), res.Data)
+		mockDownloader.AssertExpectations(t)
+	})
+
+	t.Run("Successful download with encryption", func(t *testing.T) {
+		key := make([]byte, 32)
+		_, _ = io.ReadFull(rand.Reader, key)
+
+		plaintext := []byte("secret data")
+		block, _ := aes.NewCipher(key)
+		gcm, _ := cipher.NewGCM(block)
+		nonce := make([]byte, gcm.NonceSize())
+		_, _ = io.ReadFull(rand.Reader, nonce)
+		ciphertext := gcm.Seal(nonce, nonce, plaintext, nil)
+
+		// Mock KBS
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			w.WriteHeader(http.StatusOK)
+			_, _ = w.Write(key)
+		}))
+		defer ts.Close()
+
+		svc.computation.Algorithm.KBS.URL = ts.URL
+
+		source := &ResourceSource{
+			URL:             "http://example.com/encrypted",
+			Encrypted:       true,
+			KBSResourcePath: "keys/1",
+		}
+		destPath := filepath.Join(os.TempDir(), "cocos-resources", "data", "encrypted")
+		mockDownloader.On("Download", ctx, source.URL, destPath).Return(nil, string(ciphertext)).Once()
+
+		res, err := svc.downloadAndDecryptGenericResource(ctx, source, resource.SourceTypeHTTP, svc.computation.Algorithm.KBS.URL, "data")
+		assert.NoError(t, err)
+		assert.Equal(t, plaintext, res.Data)
+		mockDownloader.AssertExpectations(t)
+	})
+
+	t.Run("Registry not initialized", func(t *testing.T) {
+		badSvc := &agentService{logger: slog.Default()}
+		_, err := badSvc.downloadAndDecryptGenericResource(ctx, &ResourceSource{}, "http", "", "algo")
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "resource registry not initialized")
+	})
+}
+
+func TestGetKeyFromKBS(t *testing.T) {
+	svc := &agentService{
+		logger: slog.Default(),
+		computation: Computation{
+			Algorithm: &Algorithm{
+				KBS: &KBSConfig{
+					Enabled: true,
+				},
+			},
+		},
+	}
+	ctx := context.Background()
+
+	t.Run("KBS disabled", func(t *testing.T) {
+		svc.computation.Algorithm.KBS.Enabled = false
+		_, err := svc.getKeyFromKBS(ctx, "", "path")
+		assert.Error(t, err)
+	})
+
+	t.Run("Successful fetch", func(t *testing.T) {
+		svc.computation.Algorithm.KBS.Enabled = true
+		key := []byte("this is a 32-byte key!!!!!!!!!!!")
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			assert.Contains(t, r.URL.Path, "resource/path")
+			w.WriteHeader(http.StatusOK)
+			_, _ = w.Write(key)
+		}))
+		defer ts.Close()
+		svc.computation.Algorithm.KBS.URL = ts.URL
+
+		fetched, err := svc.getKeyFromKBS(ctx, svc.computation.Algorithm.KBS.URL, "path")
+		assert.NoError(t, err)
+		assert.Equal(t, key, fetched)
+	})
+
+	t.Run("KBS error", func(t *testing.T) {
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			w.WriteHeader(http.StatusInternalServerError)
+		}))
+		defer ts.Close()
+		svc.computation.Algorithm.KBS.URL = ts.URL
+
+		_, err := svc.getKeyFromKBS(ctx, svc.computation.Algorithm.KBS.URL, "path")
+		assert.Error(t, err)
+	})
+}
+
+func TestInferSourceTypeDetailed(t *testing.T) {
+	tests := []struct {
+		url      string
+		expected string
+	}{
+		{"s3://bucket/key", resource.SourceTypeS3},
+		{"gs://bucket/key", resource.SourceTypeGCS},
+		{"https://example.com/file", resource.SourceTypeHTTPS},
+		{"http://example.com/file", resource.SourceTypeHTTP},
+		{"docker://ubuntu", resource.SourceTypeOCIImage},
+		{"oci:/path/to/dir", resource.SourceTypeOCIImage},
+		{"ubuntu:latest", resource.SourceTypeOCIImage},
+		{"myregistry.io/myimage:tag", resource.SourceTypeOCIImage},
+		{"invalid-url-no-slash", ""},
+		{"", ""},
+		{"ftp://server/file", ""},
+	}
+
+	for _, tt := range tests {
+		assert.Equal(t, tt.expected, inferSourceType(tt.url), "URL: %s", tt.url)
+	}
+}

agent/runner/events/adapter.go

@@ -0,0 +1,38 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package events
+
+import (
+	"context"
+	"encoding/json"
+	"log/slog"
+
+	"github.com/ultravioletrs/cocos/agent/events"
+	logpb "github.com/ultravioletrs/cocos/agent/log"
+	logclient "github.com/ultravioletrs/cocos/pkg/clients/grpc/log"
+)
+
+type adapter struct {
+	client logclient.Client
+	svc    string
+}
+
+func NewAdapter(client logclient.Client, svc string) events.Service {
+	return &adapter{
+		client: client,
+		svc:    svc,
+	}
+}
+
+func (a *adapter) SendEvent(cmpID, event, status string, details json.RawMessage) {
+	err := a.client.SendEvent(context.Background(), &logpb.EventEntry{
+		EventType:     event,
+		ComputationId: cmpID,
+		Details:       details,
+		Originator:    a.svc,
+		Status:        status,
+	})
+	if err != nil {
+		slog.Error("failed to send event to log-forwarder", "error", err)
+	}
+}

agent/runner/events/adapter_test.go

@@ -0,0 +1,138 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package events
+
+import (
+	"context"
+	"encoding/json"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+	logpb "github.com/ultravioletrs/cocos/agent/log"
+)
+
+const testServiceName = "test-service"
+
+// mockLogClient is a mock implementation of the log client.
+type mockLogClient struct {
+	mock.Mock
+}
+
+func (m *mockLogClient) SendLog(ctx context.Context, entry *logpb.LogEntry) error {
+	args := m.Called(ctx, entry)
+	return args.Error(0)
+}
+
+func (m *mockLogClient) SendEvent(ctx context.Context, entry *logpb.EventEntry) error {
+	args := m.Called(ctx, entry)
+	return args.Error(0)
+}
+
+func (m *mockLogClient) Close() error {
+	args := m.Called()
+	return args.Error(0)
+}
+
+// TestNewAdapter tests creating a new adapter.
+func TestNewAdapter(t *testing.T) {
+	mockClient := new(mockLogClient)
+	svc := testServiceName
+
+	adapter := NewAdapter(mockClient, svc)
+
+	assert.NotNil(t, adapter)
+}
+
+// TestSendEvent tests sending an event successfully.
+func TestSendEvent(t *testing.T) {
+	mockClient := new(mockLogClient)
+	svc := testServiceName
+	adapter := NewAdapter(mockClient, svc)
+
+	cmpID := "test-computation-id"
+	event := "computation.started"
+	status := "success"
+	details := json.RawMessage(`{"key": "value"}`)
+
+	expectedEntry := &logpb.EventEntry{
+		EventType:     event,
+		ComputationId: cmpID,
+		Details:       details,
+		Originator:    svc,
+		Status:        status,
+	}
+
+	mockClient.On("SendEvent", mock.Anything, expectedEntry).Return(nil)
+
+	adapter.SendEvent(cmpID, event, status, details)
+
+	mockClient.AssertExpectations(t)
+	mockClient.AssertCalled(t, "SendEvent", mock.Anything, expectedEntry)
+}
+
+// TestSendEventWithError tests sending an event when client returns an error.
+func TestSendEventWithError(t *testing.T) {
+	mockClient := new(mockLogClient)
+	svc := testServiceName
+	adapter := NewAdapter(mockClient, svc)
+
+	cmpID := "test-computation-id"
+	event := "computation.failed"
+	status := "error"
+	details := json.RawMessage(`{"error": "something went wrong"}`)
+
+	mockClient.On("SendEvent", mock.Anything, mock.Anything).Return(assert.AnError)
+
+	// This should not panic even when error occurs
+	adapter.SendEvent(cmpID, event, status, details)
+
+	mockClient.AssertExpectations(t)
+	mockClient.AssertCalled(t, "SendEvent", mock.Anything, mock.Anything)
+}
+
+// TestSendEventWithNilDetails tests sending an event with nil details.
+func TestSendEventWithNilDetails(t *testing.T) {
+	mockClient := new(mockLogClient)
+	svc := "runner-service"
+	adapter := NewAdapter(mockClient, svc)
+
+	cmpID := "comp-123"
+	event := "test.event"
+	status := "pending"
+
+	expectedEntry := &logpb.EventEntry{
+		EventType:     event,
+		ComputationId: cmpID,
+		Details:       nil,
+		Originator:    svc,
+		Status:        status,
+	}
+
+	mockClient.On("SendEvent", mock.Anything, expectedEntry).Return(nil)
+
+	adapter.SendEvent(cmpID, event, status, nil)
+
+	mockClient.AssertExpectations(t)
+}
+
+// TestSendEventWithEmptyStrings tests sending an event with empty strings.
+func TestSendEventWithEmptyStrings(t *testing.T) {
+	mockClient := new(mockLogClient)
+	svc := testServiceName
+	adapter := NewAdapter(mockClient, svc)
+
+	expectedEntry := &logpb.EventEntry{
+		EventType:     "",
+		ComputationId: "",
+		Details:       nil,
+		Originator:    svc,
+		Status:        "",
+	}
+
+	mockClient.On("SendEvent", mock.Anything, expectedEntry).Return(nil)
+
+	adapter.SendEvent("", "", "", nil)
+
+	mockClient.AssertExpectations(t)
+}

agent/runner/runner.pb.go

@@ -0,0 +1,341 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+
+// Code generated by protoc-gen-go. DO NOT EDIT.
+// versions:
+// 	protoc-gen-go v1.36.11
+// 	protoc        v6.33.1
+// source: agent/runner/runner.proto
+
+package runner
+
+import (
+	protoreflect "google.golang.org/protobuf/reflect/protoreflect"
+	protoimpl "google.golang.org/protobuf/runtime/protoimpl"
+	emptypb "google.golang.org/protobuf/types/known/emptypb"
+	reflect "reflect"
+	sync "sync"
+	unsafe "unsafe"
+)
+
+const (
+	// Verify that this generated code is sufficiently up-to-date.
+	_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
+	// Verify that runtime/protoimpl is sufficiently up-to-date.
+	_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
+)
+
+type RunRequest struct {
+	state         protoimpl.MessageState `protogen:"open.v1"`
+	ComputationId string                 `protobuf:"bytes,1,opt,name=computation_id,json=computationId,proto3" json:"computation_id,omitempty"`
+	AlgoType      string                 `protobuf:"bytes,2,opt,name=algo_type,json=algoType,proto3" json:"algo_type,omitempty"` // "binary", "python", "wasm", "docker"
+	Algorithm     []byte                 `protobuf:"bytes,3,opt,name=algorithm,proto3" json:"algorithm,omitempty"`               // The algorithm binary/script content
+	Requirements  []byte                 `protobuf:"bytes,4,opt,name=requirements,proto3" json:"requirements,omitempty"`         // Python requirements.txt content
+	Args          []string               `protobuf:"bytes,5,rep,name=args,proto3" json:"args,omitempty"`
+	Datasets      []*Dataset             `protobuf:"bytes,6,rep,name=datasets,proto3" json:"datasets,omitempty"`
+	unknownFields protoimpl.UnknownFields
+	sizeCache     protoimpl.SizeCache
+}
+
+func (x *RunRequest) Reset() {
+	*x = RunRequest{}
+	mi := &file_agent_runner_runner_proto_msgTypes[0]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *RunRequest) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*RunRequest) ProtoMessage() {}
+
+func (x *RunRequest) ProtoReflect() protoreflect.Message {
+	mi := &file_agent_runner_runner_proto_msgTypes[0]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use RunRequest.ProtoReflect.Descriptor instead.
+func (*RunRequest) Descriptor() ([]byte, []int) {
+	return file_agent_runner_runner_proto_rawDescGZIP(), []int{0}
+}
+
+func (x *RunRequest) GetComputationId() string {
+	if x != nil {
+		return x.ComputationId
+	}
+	return ""
+}
+
+func (x *RunRequest) GetAlgoType() string {
+	if x != nil {
+		return x.AlgoType
+	}
+	return ""
+}
+
+func (x *RunRequest) GetAlgorithm() []byte {
+	if x != nil {
+		return x.Algorithm
+	}
+	return nil
+}
+
+func (x *RunRequest) GetRequirements() []byte {
+	if x != nil {
+		return x.Requirements
+	}
+	return nil
+}
+
+func (x *RunRequest) GetArgs() []string {
+	if x != nil {
+		return x.Args
+	}
+	return nil
+}
+
+func (x *RunRequest) GetDatasets() []*Dataset {
+	if x != nil {
+		return x.Datasets
+	}
+	return nil
+}
+
+type Dataset struct {
+	state         protoimpl.MessageState `protogen:"open.v1"`
+	Filename      string                 `protobuf:"bytes,1,opt,name=filename,proto3" json:"filename,omitempty"`
+	Hash          []byte                 `protobuf:"bytes,2,opt,name=hash,proto3" json:"hash,omitempty"`
+	unknownFields protoimpl.UnknownFields
+	sizeCache     protoimpl.SizeCache
+}
+
+func (x *Dataset) Reset() {
+	*x = Dataset{}
+	mi := &file_agent_runner_runner_proto_msgTypes[1]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *Dataset) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*Dataset) ProtoMessage() {}
+
+func (x *Dataset) ProtoReflect() protoreflect.Message {
+	mi := &file_agent_runner_runner_proto_msgTypes[1]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use Dataset.ProtoReflect.Descriptor instead.
+func (*Dataset) Descriptor() ([]byte, []int) {
+	return file_agent_runner_runner_proto_rawDescGZIP(), []int{1}
+}
+
+func (x *Dataset) GetFilename() string {
+	if x != nil {
+		return x.Filename
+	}
+	return ""
+}
+
+func (x *Dataset) GetHash() []byte {
+	if x != nil {
+		return x.Hash
+	}
+	return nil
+}
+
+type RunResponse struct {
+	state         protoimpl.MessageState `protogen:"open.v1"`
+	ComputationId string                 `protobuf:"bytes,1,opt,name=computation_id,json=computationId,proto3" json:"computation_id,omitempty"`
+	Error         string                 `protobuf:"bytes,2,opt,name=error,proto3" json:"error,omitempty"`
+	unknownFields protoimpl.UnknownFields
+	sizeCache     protoimpl.SizeCache
+}
+
+func (x *RunResponse) Reset() {
+	*x = RunResponse{}
+	mi := &file_agent_runner_runner_proto_msgTypes[2]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *RunResponse) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*RunResponse) ProtoMessage() {}
+
+func (x *RunResponse) ProtoReflect() protoreflect.Message {
+	mi := &file_agent_runner_runner_proto_msgTypes[2]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use RunResponse.ProtoReflect.Descriptor instead.
+func (*RunResponse) Descriptor() ([]byte, []int) {
+	return file_agent_runner_runner_proto_rawDescGZIP(), []int{2}
+}
+
+func (x *RunResponse) GetComputationId() string {
+	if x != nil {
+		return x.ComputationId
+	}
+	return ""
+}
+
+func (x *RunResponse) GetError() string {
+	if x != nil {
+		return x.Error
+	}
+	return ""
+}
+
+type StopRequest struct {
+	state         protoimpl.MessageState `protogen:"open.v1"`
+	ComputationId string                 `protobuf:"bytes,1,opt,name=computation_id,json=computationId,proto3" json:"computation_id,omitempty"`
+	unknownFields protoimpl.UnknownFields
+	sizeCache     protoimpl.SizeCache
+}
+
+func (x *StopRequest) Reset() {
+	*x = StopRequest{}
+	mi := &file_agent_runner_runner_proto_msgTypes[3]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *StopRequest) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*StopRequest) ProtoMessage() {}
+
+func (x *StopRequest) ProtoReflect() protoreflect.Message {
+	mi := &file_agent_runner_runner_proto_msgTypes[3]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use StopRequest.ProtoReflect.Descriptor instead.
+func (*StopRequest) Descriptor() ([]byte, []int) {
+	return file_agent_runner_runner_proto_rawDescGZIP(), []int{3}
+}
+
+func (x *StopRequest) GetComputationId() string {
+	if x != nil {
+		return x.ComputationId
+	}
+	return ""
+}
+
+var File_agent_runner_runner_proto protoreflect.FileDescriptor
+
+const file_agent_runner_runner_proto_rawDesc = "" +
+	"\n" +
+	"\x19agent/runner/runner.proto\x12\x06runner\x1a\x1bgoogle/protobuf/empty.proto\"\xd3\x01\n" +
+	"\n" +
+	"RunRequest\x12%\n" +
+	"\x0ecomputation_id\x18\x01 \x01(\tR\rcomputationId\x12\x1b\n" +
+	"\talgo_type\x18\x02 \x01(\tR\balgoType\x12\x1c\n" +
+	"\talgorithm\x18\x03 \x01(\fR\talgorithm\x12\"\n" +
+	"\frequirements\x18\x04 \x01(\fR\frequirements\x12\x12\n" +
+	"\x04args\x18\x05 \x03(\tR\x04args\x12+\n" +
+	"\bdatasets\x18\x06 \x03(\v2\x0f.runner.DatasetR\bdatasets\"9\n" +
+	"\aDataset\x12\x1a\n" +
+	"\bfilename\x18\x01 \x01(\tR\bfilename\x12\x12\n" +
+	"\x04hash\x18\x02 \x01(\fR\x04hash\"J\n" +
+	"\vRunResponse\x12%\n" +
+	"\x0ecomputation_id\x18\x01 \x01(\tR\rcomputationId\x12\x14\n" +
+	"\x05error\x18\x02 \x01(\tR\x05error\"4\n" +
+	"\vStopRequest\x12%\n" +
+	"\x0ecomputation_id\x18\x01 \x01(\tR\rcomputationId2x\n" +
+	"\x11ComputationRunner\x12.\n" +
+	"\x03Run\x12\x12.runner.RunRequest\x1a\x13.runner.RunResponse\x123\n" +
+	"\x04Stop\x12\x13.runner.StopRequest\x1a\x16.google.protobuf.EmptyB\n" +
+	"Z\b./runnerb\x06proto3"
+
+var (
+	file_agent_runner_runner_proto_rawDescOnce sync.Once
+	file_agent_runner_runner_proto_rawDescData []byte
+)
+
+func file_agent_runner_runner_proto_rawDescGZIP() []byte {
+	file_agent_runner_runner_proto_rawDescOnce.Do(func() {
+		file_agent_runner_runner_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_agent_runner_runner_proto_rawDesc), len(file_agent_runner_runner_proto_rawDesc)))
+	})
+	return file_agent_runner_runner_proto_rawDescData
+}
+
+var file_agent_runner_runner_proto_msgTypes = make([]protoimpl.MessageInfo, 4)
+var file_agent_runner_runner_proto_goTypes = []any{
+	(*RunRequest)(nil),    // 0: runner.RunRequest
+	(*Dataset)(nil),       // 1: runner.Dataset
+	(*RunResponse)(nil),   // 2: runner.RunResponse
+	(*StopRequest)(nil),   // 3: runner.StopRequest
+	(*emptypb.Empty)(nil), // 4: google.protobuf.Empty
+}
+var file_agent_runner_runner_proto_depIdxs = []int32{
+	1, // 0: runner.RunRequest.datasets:type_name -> runner.Dataset
+	0, // 1: runner.ComputationRunner.Run:input_type -> runner.RunRequest
+	3, // 2: runner.ComputationRunner.Stop:input_type -> runner.StopRequest
+	2, // 3: runner.ComputationRunner.Run:output_type -> runner.RunResponse
+	4, // 4: runner.ComputationRunner.Stop:output_type -> google.protobuf.Empty
+	3, // [3:5] is the sub-list for method output_type
+	1, // [1:3] is the sub-list for method input_type
+	1, // [1:1] is the sub-list for extension type_name
+	1, // [1:1] is the sub-list for extension extendee
+	0, // [0:1] is the sub-list for field type_name
+}
+
+func init() { file_agent_runner_runner_proto_init() }
+func file_agent_runner_runner_proto_init() {
+	if File_agent_runner_runner_proto != nil {
+		return
+	}
+	type x struct{}
+	out := protoimpl.TypeBuilder{
+		File: protoimpl.DescBuilder{
+			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
+			RawDescriptor: unsafe.Slice(unsafe.StringData(file_agent_runner_runner_proto_rawDesc), len(file_agent_runner_runner_proto_rawDesc)),
+			NumEnums:      0,
+			NumMessages:   4,
+			NumExtensions: 0,
+			NumServices:   1,
+		},
+		GoTypes:           file_agent_runner_runner_proto_goTypes,
+		DependencyIndexes: file_agent_runner_runner_proto_depIdxs,
+		MessageInfos:      file_agent_runner_runner_proto_msgTypes,
+	}.Build()
+	File_agent_runner_runner_proto = out.File
+	file_agent_runner_runner_proto_goTypes = nil
+	file_agent_runner_runner_proto_depIdxs = nil
+}

agent/runner/runner.proto

@@ -0,0 +1,38 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+
+syntax = "proto3";
+
+package runner;
+
+option go_package = "./runner";
+
+import "google/protobuf/empty.proto";
+
+service ComputationRunner {
+  rpc Run(RunRequest) returns (RunResponse);
+  rpc Stop(StopRequest) returns (google.protobuf.Empty);
+}
+
+message RunRequest {
+  string computation_id = 1;
+  string algo_type = 2; // "binary", "python", "wasm", "docker"
+  bytes algorithm = 3; // The algorithm binary/script content
+  bytes requirements = 4; // Python requirements.txt content
+  repeated string args = 5;
+  repeated Dataset datasets = 6;
+}
+
+message Dataset {
+  string filename = 1;
+  bytes hash = 2; 
+}
+
+message RunResponse {
+  string computation_id = 1;
+  string error = 2;
+}
+
+message StopRequest {
+  string computation_id = 1;
+}

agent/runner/runner_grpc.pb.go

@@ -0,0 +1,163 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+
+// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
+// versions:
+// - protoc-gen-go-grpc v1.6.0
+// - protoc             v6.33.1
+// source: agent/runner/runner.proto
+
+package runner
+
+import (
+	context "context"
+	grpc "google.golang.org/grpc"
+	codes "google.golang.org/grpc/codes"
+	status "google.golang.org/grpc/status"
+	emptypb "google.golang.org/protobuf/types/known/emptypb"
+)
+
+// This is a compile-time assertion to ensure that this generated file
+// is compatible with the grpc package it is being compiled against.
+// Requires gRPC-Go v1.64.0 or later.
+const _ = grpc.SupportPackageIsVersion9
+
+const (
+	ComputationRunner_Run_FullMethodName  = "/runner.ComputationRunner/Run"
+	ComputationRunner_Stop_FullMethodName = "/runner.ComputationRunner/Stop"
+)
+
+// ComputationRunnerClient is the client API for ComputationRunner service.
+//
+// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
+type ComputationRunnerClient interface {
+	Run(ctx context.Context, in *RunRequest, opts ...grpc.CallOption) (*RunResponse, error)
+	Stop(ctx context.Context, in *StopRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)
+}
+
+type computationRunnerClient struct {
+	cc grpc.ClientConnInterface
+}
+
+func NewComputationRunnerClient(cc grpc.ClientConnInterface) ComputationRunnerClient {
+	return &computationRunnerClient{cc}
+}
+
+func (c *computationRunnerClient) Run(ctx context.Context, in *RunRequest, opts ...grpc.CallOption) (*RunResponse, error) {
+	cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
+	out := new(RunResponse)
+	err := c.cc.Invoke(ctx, ComputationRunner_Run_FullMethodName, in, out, cOpts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+func (c *computationRunnerClient) Stop(ctx context.Context, in *StopRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) {
+	cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
+	out := new(emptypb.Empty)
+	err := c.cc.Invoke(ctx, ComputationRunner_Stop_FullMethodName, in, out, cOpts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+// ComputationRunnerServer is the server API for ComputationRunner service.
+// All implementations must embed UnimplementedComputationRunnerServer
+// for forward compatibility.
+type ComputationRunnerServer interface {
+	Run(context.Context, *RunRequest) (*RunResponse, error)
+	Stop(context.Context, *StopRequest) (*emptypb.Empty, error)
+	mustEmbedUnimplementedComputationRunnerServer()
+}
+
+// UnimplementedComputationRunnerServer must be embedded to have
+// forward compatible implementations.
+//
+// NOTE: this should be embedded by value instead of pointer to avoid a nil
+// pointer dereference when methods are called.
+type UnimplementedComputationRunnerServer struct{}
+
+func (UnimplementedComputationRunnerServer) Run(context.Context, *RunRequest) (*RunResponse, error) {
+	return nil, status.Error(codes.Unimplemented, "method Run not implemented")
+}
+func (UnimplementedComputationRunnerServer) Stop(context.Context, *StopRequest) (*emptypb.Empty, error) {
+	return nil, status.Error(codes.Unimplemented, "method Stop not implemented")
+}
+func (UnimplementedComputationRunnerServer) mustEmbedUnimplementedComputationRunnerServer() {}
+func (UnimplementedComputationRunnerServer) testEmbeddedByValue()                           {}
+
+// UnsafeComputationRunnerServer may be embedded to opt out of forward compatibility for this service.
+// Use of this interface is not recommended, as added methods to ComputationRunnerServer will
+// result in compilation errors.
+type UnsafeComputationRunnerServer interface {
+	mustEmbedUnimplementedComputationRunnerServer()
+}
+
+func RegisterComputationRunnerServer(s grpc.ServiceRegistrar, srv ComputationRunnerServer) {
+	// If the following call panics, it indicates UnimplementedComputationRunnerServer was
+	// embedded by pointer and is nil.  This will cause panics if an
+	// unimplemented method is ever invoked, so we test this at initialization
+	// time to prevent it from happening at runtime later due to I/O.
+	if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
+		t.testEmbeddedByValue()
+	}
+	s.RegisterService(&ComputationRunner_ServiceDesc, srv)
+}
+
+func _ComputationRunner_Run_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(RunRequest)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(ComputationRunnerServer).Run(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: ComputationRunner_Run_FullMethodName,
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(ComputationRunnerServer).Run(ctx, req.(*RunRequest))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+func _ComputationRunner_Stop_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(StopRequest)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(ComputationRunnerServer).Stop(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: ComputationRunner_Stop_FullMethodName,
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(ComputationRunnerServer).Stop(ctx, req.(*StopRequest))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+// ComputationRunner_ServiceDesc is the grpc.ServiceDesc for ComputationRunner service.
+// It's only intended for direct use with grpc.RegisterService,
+// and not to be introspected or modified (even as a copy)
+var ComputationRunner_ServiceDesc = grpc.ServiceDesc{
+	ServiceName: "runner.ComputationRunner",
+	HandlerType: (*ComputationRunnerServer)(nil),
+	Methods: []grpc.MethodDesc{
+		{
+			MethodName: "Run",
+			Handler:    _ComputationRunner_Run_Handler,
+		},
+		{
+			MethodName: "Stop",
+			Handler:    _ComputationRunner_Stop_Handler,
+		},
+	},
+	Streams:  []grpc.StreamDesc{},
+	Metadata: "agent/runner/runner.proto",
+}

agent/runner/service/service.go

@@ -0,0 +1,151 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package service
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"os"
+	"path/filepath"
+	"sync"
+
+	"github.com/ultravioletrs/cocos/agent/algorithm"
+	"github.com/ultravioletrs/cocos/agent/algorithm/binary"
+	"github.com/ultravioletrs/cocos/agent/algorithm/docker"
+	"github.com/ultravioletrs/cocos/agent/algorithm/python"
+	"github.com/ultravioletrs/cocos/agent/algorithm/wasm"
+	"github.com/ultravioletrs/cocos/agent/events"
+	pb "github.com/ultravioletrs/cocos/agent/runner"
+	"google.golang.org/protobuf/types/known/emptypb"
+)
+
+const (
+	algoFilePermission = 0o700
+)
+
+var _ pb.ComputationRunnerServer = (*RunnerService)(nil)
+
+type RunnerService struct {
+	pb.UnimplementedComputationRunnerServer
+	logger      *slog.Logger
+	eventSvc    events.Service
+	currentAlgo algorithm.Algorithm
+	mu          sync.Mutex
+}
+
+func New(logger *slog.Logger, eventSvc events.Service) *RunnerService {
+	return &RunnerService{
+		logger:   logger,
+		eventSvc: eventSvc,
+	}
+}
+
+func (s *RunnerService) Run(ctx context.Context, req *pb.RunRequest) (*pb.RunResponse, error) {
+	s.mu.Lock()
+	if s.currentAlgo != nil {
+		s.mu.Unlock()
+		return &pb.RunResponse{
+			ComputationId: req.ComputationId,
+			Error:         "computation already running",
+		}, nil
+	}
+	s.mu.Unlock()
+
+	defer func() {
+		s.mu.Lock()
+		s.currentAlgo = nil
+		s.mu.Unlock()
+	}()
+
+	currentDir, err := os.Getwd()
+	if err != nil {
+		return nil, fmt.Errorf("error getting current directory: %v", err)
+	}
+
+	// Write Algo File
+	algoPath := filepath.Join(currentDir, "algo")
+	f, err := os.Create(algoPath)
+	if err != nil {
+		return nil, fmt.Errorf("error creating algorithm file: %v", err)
+	}
+	if _, err := f.Write(req.Algorithm); err != nil {
+		return nil, fmt.Errorf("error writing algorithm to file: %v", err)
+	}
+	if err := os.Chmod(algoPath, algoFilePermission); err != nil {
+		return nil, fmt.Errorf("error changing file permissions: %v", err)
+	}
+	if err := f.Close(); err != nil {
+		return nil, fmt.Errorf("error closing file: %v", err)
+	}
+	defer func() {
+		if err := os.Remove(algoPath); err != nil {
+			s.logger.Warn("error removing algorithm file", "error", err)
+		}
+	}()
+
+	var algo algorithm.Algorithm
+
+	switch req.AlgoType {
+	case string(algorithm.AlgoTypeBin):
+		algo = binary.NewAlgorithm(s.logger, s.eventSvc, algoPath, req.Args, req.ComputationId)
+	case string(algorithm.AlgoTypePython):
+		var requirementsFile string
+		if len(req.Requirements) > 0 {
+			fr, err := os.CreateTemp("", "requirements.txt")
+			if err != nil {
+				return nil, fmt.Errorf("error creating requirments file: %v", err)
+			}
+			defer func() {
+				if err := os.Remove(fr.Name()); err != nil {
+					s.logger.Warn("error removing requirements file", "error", err)
+				}
+			}()
+			if _, err := fr.Write(req.Requirements); err != nil {
+				return nil, fmt.Errorf("error writing requirements to file: %v", err)
+			}
+			if err := fr.Close(); err != nil {
+				return nil, fmt.Errorf("error closing file: %v", err)
+			}
+			requirementsFile = fr.Name()
+		}
+		// Assuming default python runtime if not specified in request (proto doesn't have runtime field yet)
+		// We can add it or assume.
+		runtime := python.PyRuntime
+		algo = python.NewAlgorithm(s.logger, s.eventSvc, runtime, requirementsFile, algoPath, req.Args, req.ComputationId)
+	case string(algorithm.AlgoTypeWasm):
+		algo = wasm.NewAlgorithm(s.logger, s.eventSvc, req.Args, algoPath, req.ComputationId)
+	case string(algorithm.AlgoTypeDocker):
+		algo = docker.NewAlgorithm(s.logger, s.eventSvc, algoPath, req.ComputationId)
+	default:
+		return nil, fmt.Errorf("unsupported algorithm type: %s", req.AlgoType)
+	}
+
+	s.mu.Lock()
+	s.currentAlgo = algo
+	s.mu.Unlock()
+
+	if err := algo.Run(); err != nil {
+		s.logger.Error("computation failed", "error", err)
+		return &pb.RunResponse{
+			ComputationId: req.ComputationId,
+			Error:         err.Error(),
+		}, nil
+	}
+
+	return &pb.RunResponse{
+		ComputationId: req.ComputationId,
+	}, nil
+}
+
+func (s *RunnerService) Stop(ctx context.Context, req *pb.StopRequest) (*emptypb.Empty, error) {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	if s.currentAlgo != nil {
+		if err := s.currentAlgo.Stop(); err != nil {
+			return nil, err
+		}
+	}
+	return &emptypb.Empty{}, nil
+}

agent/runner/service/service_test.go

@@ -0,0 +1,382 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package service
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"log/slog"
+	"os"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	pb "github.com/ultravioletrs/cocos/agent/runner"
+)
+
+// MockEventService is a mock implementation of events.Service.
+type MockEventService struct {
+	events []interface{}
+}
+
+func (m *MockEventService) SendEvent(cmpID, event, status string, details json.RawMessage) {
+	m.events = append(m.events, map[string]interface{}{
+		"cmpID":   cmpID,
+		"event":   event,
+		"status":  status,
+		"details": details,
+	})
+}
+
+// TestNewRunnerService tests the creation of a new runner service.
+func TestNewRunnerService(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+
+	rs := New(logger, eventSvc)
+	require.NotNil(t, rs)
+	assert.NotNil(t, rs.logger)
+	assert.NotNil(t, rs.eventSvc)
+	assert.Nil(t, rs.currentAlgo)
+}
+
+// TestRunWithBinaryAlgorithm tests running a binary algorithm.
+func TestRunWithBinaryAlgorithm(t *testing.T) {
+	origDir, _ := os.Getwd()
+	tmpDir := t.TempDir()
+	require.NoError(t, os.Chdir(tmpDir))
+	defer func() { require.NoError(t, os.Chdir(origDir)) }()
+
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	req := &pb.RunRequest{
+		ComputationId: "test-1",
+		AlgoType:      "bin",
+		Algorithm:     []byte("#!/bin/bash\necho 'test'"),
+		Args:          []string{"arg1", "arg2"},
+	}
+
+	resp, err := rs.Run(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+	assert.Empty(t, resp.Error)
+	assert.Equal(t, "test-1", resp.ComputationId)
+}
+
+// TestRunWithPythonAlgorithm tests running a Python algorithm.
+func TestRunWithPythonAlgorithm(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	req := &pb.RunRequest{
+		ComputationId: "test-python",
+		AlgoType:      "python",
+		Algorithm:     []byte("print('hello')"),
+		Args:          []string{},
+		Requirements:  []byte("numpy==2.2.0"),
+	}
+
+	resp, err := rs.Run(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+	assert.Empty(t, resp.Error)
+	assert.Equal(t, "test-python", resp.ComputationId)
+	t.Cleanup(func() {
+		_ = os.Remove("algo")
+	})
+}
+
+// TestRunWithPythonAlgorithmNoRequirements tests running Python without requirements.
+func TestRunWithPythonAlgorithmNoRequirements(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	req := &pb.RunRequest{
+		ComputationId: "test-python-noreq",
+		AlgoType:      "python",
+		Algorithm:     []byte("print('hello')"),
+		Args:          []string{},
+	}
+
+	resp, err := rs.Run(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+	assert.Empty(t, resp.Error)
+	assert.Equal(t, "test-python-noreq", resp.ComputationId)
+	t.Cleanup(func() {
+		_ = os.Remove("algo")
+	})
+}
+
+// TestRunWithWasmAlgorithm tests running a WASM algorithm.
+func TestRunWithWasmAlgorithm(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	req := &pb.RunRequest{
+		ComputationId: "test-wasm",
+		AlgoType:      "wasm",
+		Algorithm:     []byte{0x00, 0x61, 0x73, 0x6d},
+		Args:          []string{},
+	}
+
+	resp, err := rs.Run(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+	if resp.Error != "" {
+		assert.Contains(t, resp.Error, "wasmedge")
+		t.Skip("wasmedge not found, skipping test")
+	}
+	assert.Equal(t, "test-wasm", resp.ComputationId)
+	t.Cleanup(func() {
+		_ = os.Remove("algo")
+	})
+}
+
+// TestRunWithDockerAlgorithm tests running a Docker algorithm.
+func TestRunWithDockerAlgorithm(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	req := &pb.RunRequest{
+		ComputationId: "test-docker",
+		AlgoType:      "docker",
+		Algorithm:     []byte("FROM ubuntu:latest\nRUN echo 'test'"),
+		Args:          []string{},
+	}
+
+	resp, err := rs.Run(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+	if resp.Error != "" {
+		assert.Contains(t, resp.Error, "Docker")
+		t.Skip("Docker issue, skipping test")
+	}
+	assert.Equal(t, "test-docker", resp.ComputationId)
+	t.Cleanup(func() {
+		_ = os.Remove("algo")
+	})
+}
+
+// TestRunWithUnsupportedAlgorithmType tests running with unsupported algorithm type.
+func TestRunWithUnsupportedAlgorithmType(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	req := &pb.RunRequest{
+		ComputationId: "test-unsupported",
+		AlgoType:      "unsupported",
+		Algorithm:     []byte("test"),
+		Args:          []string{},
+	}
+
+	resp, err := rs.Run(context.Background(), req)
+	require.Error(t, err)
+	require.Nil(t, resp)
+}
+
+// TestRunAlreadyRunning tests running computation when one is already running.
+func TestRunAlreadyRunning(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	// Use a long-running bash script
+	req := &pb.RunRequest{
+		ComputationId: "test-running",
+		AlgoType:      "bin",
+		Algorithm:     []byte("#!/bin/bash\nsleep 30"),
+		Args:          []string{},
+	}
+
+	// Start first computation (will run for 30 seconds)
+	go func() {
+		_, _ = rs.Run(context.Background(), req)
+	}()
+
+	// Give it time to start
+	time.Sleep(500 * time.Millisecond)
+
+	// Try to run another immediately - should fail
+	resp, err := rs.Run(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+	assert.Equal(t, "computation already running", resp.Error)
+	t.Cleanup(func() {
+		_ = os.Remove("algo")
+	})
+}
+
+// TestStopWhenRunning tests stopping a running computation.
+func TestStopWhenRunning(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	req := &pb.RunRequest{
+		ComputationId: "test-stop",
+		AlgoType:      "bin",
+		Algorithm:     []byte("#!/bin/bash\nsleep 10"),
+		Args:          []string{},
+	}
+
+	go func() {
+		_, _ = rs.Run(context.Background(), req)
+	}()
+
+	// Give it time to start
+	time.Sleep(500 * time.Millisecond)
+
+	stopReq := &pb.StopRequest{
+		ComputationId: "test-stop",
+	}
+
+	stopResp, err := rs.Stop(context.Background(), stopReq)
+	require.NoError(t, err)
+	require.NotNil(t, stopResp)
+	t.Cleanup(func() {
+		_ = os.Remove("algo")
+	})
+}
+
+// TestRunErrors tests error paths in Run.
+func TestRunErrors(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	t.Run("create algo file failure", func(t *testing.T) {
+		// Create a directory named "algo" to make os.Create("algo") fail
+		err := os.Mkdir("algo", 0o755)
+		require.NoError(t, err)
+		defer os.RemoveAll("algo")
+
+		req := &pb.RunRequest{
+			ComputationId: "test-err",
+			AlgoType:      "bin",
+			Algorithm:     []byte("test"),
+		}
+		_, err = rs.Run(context.Background(), req)
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "error creating algorithm file")
+	})
+
+	t.Run("getwd failure", func(t *testing.T) {
+		origDir, _ := os.Getwd()
+		tmpDir := t.TempDir()
+		err := os.Chdir(tmpDir)
+		require.NoError(t, err)
+
+		// Remove the current working directory to trigger Getwd failure
+		err = os.RemoveAll(tmpDir)
+		require.NoError(t, err)
+
+		req := &pb.RunRequest{
+			ComputationId: "test-err-getwd",
+			AlgoType:      "bin",
+			Algorithm:     []byte("test"),
+		}
+		_, err = rs.Run(context.Background(), req)
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "error getting current directory")
+
+		// Restore working directory
+		_ = os.Chdir(origDir)
+	})
+
+	t.Run("requirements file creation failure", func(t *testing.T) {
+		// This one is harder because it uses os.CreateTemp("", "requirements.txt")
+		// We can't easily make this fail without reaching into the system's temp dir.
+		// Skipping for now as it's a very unlikely edge case.
+	})
+
+	t.Run("chmod failure", func(t *testing.T) {
+		// We can't easily mock os.Chmod, but we can try to make the file unmodifiable
+		// On Linux, we can set the immutable attribute, but that requires root.
+		// Alternatively, we can try to use a directory with permissions that prevent chmod?
+		// No, chmod usually works if you own the file.
+	})
+
+	t.Run("write algorithm failure", func(t *testing.T) {
+		// This is also hard without mocking os.File.Write or reaching internal limits.
+	})
+}
+
+// TestConcurrentRun tests that concurrent runs are properly serialized.
+func TestConcurrentRun(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	req := &pb.RunRequest{
+		ComputationId: "test-concurrent",
+		AlgoType:      "bin",
+		Algorithm:     []byte("#!/bin/bash\nsleep 15"),
+		Args:          []string{},
+	}
+
+	// Start first run in goroutine (will run for 15 seconds)
+	go func() {
+		_, _ = rs.Run(context.Background(), req)
+	}()
+
+	// Give it time to actually start
+	time.Sleep(500 * time.Millisecond)
+
+	// Concurrent attempt should fail
+	resp2, err := rs.Run(context.Background(), req)
+	require.NoError(t, err)
+	assert.Equal(t, "computation already running", resp2.Error)
+	t.Cleanup(func() {
+		_ = os.Remove("algo")
+	})
+}
+
+// TestRunWithMultipleArgs tests running with multiple arguments.
+func TestRunWithMultipleArgs(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	req := &pb.RunRequest{
+		ComputationId: "test-multi-args",
+		AlgoType:      "bin",
+		Algorithm:     []byte("#!/bin/bash\necho $@"),
+		Args:          []string{"arg1", "arg2", "arg3", "arg4"},
+	}
+
+	resp, err := rs.Run(context.Background(), req)
+	require.NoError(t, err)
+	require.NotNil(t, resp)
+	assert.Empty(t, resp.Error)
+	assert.Equal(t, "test-multi-args", resp.ComputationId)
+	t.Cleanup(func() {
+		_ = os.Remove("algo")
+	})
+}
+
+func TestStopFailure(t *testing.T) {
+	logger := slog.New(slog.NewTextHandler(os.Stdout, nil))
+	eventSvc := &MockEventService{}
+	rs := New(logger, eventSvc)
+
+	// Mock an algorithm that fails on Stop
+	rs.currentAlgo = &MockAlgorithmStopFail{}
+
+	_, err := rs.Stop(context.Background(), &pb.StopRequest{})
+	assert.Error(t, err)
+}
+
+type MockAlgorithmStopFail struct{}
+
+func (m *MockAlgorithmStopFail) Run() error  { return nil }
+func (m *MockAlgorithmStopFail) Stop() error { return fmt.Errorf("stop failed") }

cli/CORIM_GENERATION.md

@@ -0,0 +1,196 @@
+# CoRIM Generation CLI Commands
+
+This document describes the CLI commands for generating CoRIM (Concise Reference Integrity Manifest) attestation policies.
+
+## Overview
+
+The `cocos-cli policy create-corim` command provides subcommands for generating CoRIM policies for different platforms:
+- **azure**: Generate from Azure Attestation Token
+- **gcp**: Generate from GCP endorsements
+- **snp**: Generate for AMD SEV-SNP (direct host generation)
+- **tdx**: Generate for Intel TDX (direct host generation)
+
+## Commands
+
+### Azure SEV-SNP
+
+Generate CoRIM from an Azure Attestation Token (JWT).
+
+```bash
+cocos-cli policy create-corim azure --token <path-to-token> [--product <product>]
+```
+
+**Flags:**
+- `--token` (required): Path to file containing Azure Attestation Token (JWT)
+- `--product` (optional): Processor product name (default: "Milan")
+
+**Example:**
+```bash
+cocos-cli policy create-corim azure \
+  --token /path/to/token.jwt \
+  --product Milan \
+  > azure-policy.corim
+```
+
+### GCP SEV-SNP
+
+Generate CoRIM from GCP SEV-SNP measurement and endorsements.
+
+```bash
+cocos-cli policy create-corim gcp --measurement <hex> [--vcpu <num>]
+```
+
+**Flags:**
+- `--measurement` (required): 384-bit measurement hex string
+- `--vcpu` (optional): vCPU number (default: 0)
+
+**Example:**
+```bash
+cocos-cli policy create-corim gcp \
+  --measurement abc123... \
+  --vcpu 0 \
+  > gcp-policy.corim
+```
+
+### SEV-SNP (Direct Host)
+
+Generate CoRIM for AMD SEV-SNP platform directly on the host.
+
+```bash
+cocos-cli policy create-corim snp [flags]
+```
+
+**Flags:**
+- `--measurement` (optional): Measurement/Launch Digest (hex string, defaults to zero if not provided)
+- `--policy` (optional): SNP policy flags (default: 0)
+- `--svn` (optional): Security Version Number/TCB (default: 0)
+- `--product` (optional): Processor product name (default: "Milan")
+- `--host-data` (optional): Host data (hex string)
+- `--launch-tcb` (optional): Minimum launch TCB (default: 0)
+- `--output` (optional): Output file path (default: stdout)
+
+**Examples:**
+
+Generate with defaults (zeroed measurement):
+```bash
+cocos-cli policy create-corim snp \
+  --product Milan \
+  --output snp-policy.corim
+```
+
+Generate with custom measurement:
+```bash
+cocos-cli policy create-corim snp \
+  --measurement abc123def456... \
+  --product Genoa \
+  --svn 1 \
+  --policy 0x30000 \
+  --output snp-policy.corim
+```
+
+Generate with host data and launch TCB:
+```bash
+cocos-cli policy create-corim snp \
+  --measurement abc123... \
+  --host-data deadbeef \
+  --launch-tcb 1 \
+  --output snp-policy.corim
+```
+
+### TDX (Direct Host)
+
+Generate CoRIM for Intel TDX platform directly on the host.
+
+```bash
+cocos-cli policy create-corim tdx [flags]
+```
+
+**Flags:**
+- `--measurement` (optional): MRTD measurement (hex string, uses default if not provided)
+- `--svn` (optional): Security Version Number (default: 0)
+- `--rtmrs` (optional): Comma-separated RTMRs (hex)
+- `--mr-seam` (optional): MRSEAM (hex)
+- `--output` (optional): Output file path (default: stdout)
+
+**Examples:**
+
+Generate with defaults (matches legacy script behavior):
+```bash
+cocos-cli policy create-corim tdx \
+  --output tdx-policy.corim
+```
+
+Generate with custom values:
+```bash
+cocos-cli policy create-corim tdx \
+  --measurement abc123def456... \
+  --rtmrs rtmr0,rtmr1,rtmr2,rtmr3 \
+  --mr-seam 789abc... \
+  --svn 2 \
+  --output tdx-policy.corim
+```
+
+## Signing CoRIMs
+
+CoRIMs can be signed using a private key (COSE_Sign1). The generated output will be a COSE-wrapped CoRIM in CBOR format.
+
+### Prerequisite: Generate Signing Key
+
+You will need an EC private key (P-256) in PEM format. You can generate one using `openssl`:
+
+```bash
+openssl ecparam -name prime256v1 -genkey -noout -out private-key.pem
+```
+
+### Signing with CLI
+
+Use the `--signing-key` flag to sign the CoRIM during generation.
+
+**SNP Example:**
+```bash
+cocos-cli policy create-corim snp \
+  --product Milan \
+  --signing-key private-key.pem \
+  --output signed-snp.corim
+```
+
+**TDX Example:**
+```bash
+cocos-cli policy create-corim tdx \
+  --signing-key private-key.pem \
+  --output signed-tdx.corim
+```
+
+### Verification
+
+The output file is a standard COSE_Sign1 message containing the CoRIM. It can be verified using any tool that supports COSE and CoRIM verification, such as the [veraison/corim](https://github.com/veraison/corim) library.
+
+## Output Format
+
+All commands output CoRIM in CBOR (Concise Binary Object Representation) format. By default, output is written to stdout, allowing for piping:
+
+```bash
+# Pipe to file
+cocos-cli policy create-corim snp --product Milan > policy.corim
+
+# Pipe to another command
+cocos-cli policy create-corim tdx | base64
+
+# Use --output flag
+cocos-cli policy create-corim snp --product Milan --output policy.corim
+```
+
+## Integration with Manager
+
+The manager service can dynamically generate CoRIM policies using the same underlying generator package. When `FetchAttestationPolicy` is called:
+
+1. For SNP: Calculates IGVM measurement using the `igvmmeasure` binary
+2. Extracts host data and launch TCB from VM configuration
+3. Generates CoRIM using the `generator` package
+4. Returns CBOR-encoded CoRIM
+
+## See Also
+
+- [Generator Package Documentation](../pkg/attestation/generator/README.md)
+- [IGVM Measure Package Documentation](../pkg/attestation/igvmmeasure/README.md)
+- [Manager README](../manager/README.md)

cli/algorithms.go

@@ -29,7 +29,7 @@ func (cli *CLI) NewAlgorithmCmd() *cobra.Command {
 		Args:    cobra.ExactArgs(2),
 		Run: func(cmd *cobra.Command, args []string) {
 			if cli.connectErr != nil {
-				printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
+				cli.printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
 				return
 			}
 
@@ -39,7 +39,7 @@ func (cli *CLI) NewAlgorithmCmd() *cobra.Command {
 
 			algorithm, err := os.Open(algorithmFile)
 			if err != nil {
-				printError(cmd, "Error reading algorithm file: %v ❌ ", err)
+				cli.printError(cmd, "Error reading algorithm file: %v ❌ ", err)
 				return
 			}
 
@@ -49,7 +49,7 @@ func (cli *CLI) NewAlgorithmCmd() *cobra.Command {
 			if requirementsFile != "" {
 				req, err = os.Open(requirementsFile)
 				if err != nil {
-					printError(cmd, "Error reading requirments file: %v ❌ ", err)
+					cli.printError(cmd, "Error reading requirments file: %v ❌ ", err)
 					return
 				}
 				defer req.Close()
@@ -57,7 +57,7 @@ func (cli *CLI) NewAlgorithmCmd() *cobra.Command {
 
 			privKeyFile, err := os.ReadFile(args[1])
 			if err != nil {
-				printError(cmd, "Error reading private key file: %v ❌ ", err)
+				cli.printError(cmd, "Error reading private key file: %v ❌ ", err)
 				return
 			}
 
@@ -65,14 +65,14 @@ func (cli *CLI) NewAlgorithmCmd() *cobra.Command {
 
 			privKey, err := decodeKey(pemBlock)
 			if err != nil {
-				printError(cmd, "Error decoding private key: %v ❌ ", err)
+				cli.printError(cmd, "Error decoding private key: %v ❌ ", err)
 				return
 			}
 
 			ctx := metadata.NewOutgoingContext(cmd.Context(), metadata.New(make(map[string]string)))
 
 			if err := cli.agentSDK.Algo(addAlgoMetadata(ctx), algorithm, req, privKey); err != nil {
-				printError(cmd, "Failed to upload algorithm due to error: %v ❌ ", err)
+				cli.printError(cmd, "Failed to upload algorithm due to error: %v ❌ ", err)
 				return
 			}
 

cli/attestation.go

@@ -6,23 +6,16 @@ import (
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
-	"io"
 	"os"
 	"strings"
-	"time"
 
-	"github.com/absmach/supermq/pkg/errors"
+	"github.com/absmach/magistrala/pkg/errors"
 	"github.com/fatih/color"
 	"github.com/google/go-sev-guest/abi"
-	"github.com/google/go-sev-guest/proto/sevsnp"
-	"github.com/google/go-sev-guest/tools/lib/report"
 	tpmAttest "github.com/google/go-tpm-tools/proto/attest"
 	"github.com/spf13/cobra"
 	"github.com/spf13/pflag"
 	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"github.com/ultravioletrs/cocos/pkg/attestation/azure"
-	"github.com/ultravioletrs/cocos/pkg/attestation/quoteprovider"
-	"github.com/ultravioletrs/cocos/pkg/attestation/tdx"
 	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 	"google.golang.org/protobuf/encoding/prototext"
 	"google.golang.org/protobuf/proto"
@@ -37,6 +30,7 @@ const (
 	attestationFilePath       = "attestation.bin"
 	azureAttestResultFilePath = "azure_attest_result.json"
 	azureAttestTokenFilePath  = "azure_attest_token.jwt"
+	attestationReportJson     = "attestation.json"
 	TEE                       = "tee"
 	SNP                       = "snp"
 	VTPM                      = "vtpm"
@@ -49,38 +43,14 @@ const (
 )
 
 var (
-	mode                          string
-	cfgString                     string
-	timeout                       time.Duration
-	maxRetryDelay                 time.Duration
-	platformInfo                  string
-	stepping                      string
-	trustedAuthorKeys             []string
-	trustedAuthorHashes           []string
-	trustedIdKeys                 []string
-	trustedIdKeyHashes            []string
-	attestationFile               string
-	attestationRaw                []byte
-	empty16                       = [size16]byte{}
-	empty32                       = [size32]byte{}
-	empty64                       = [size64]byte{}
-	defaultReportIdMa             = []byte{255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}
 	errReportSize                 = errors.New("attestation contents too small")
-	ErrBadAttestation             = errors.New("attestation file is corrupted or in wrong format")
-	output                        string
 	nonce                         []byte
-	format                        string
 	teeNonce                      []byte
 	tokenNonce                    []byte
 	getTextProtoAttestationReport bool
 	getAzureTokenJWT              bool
-	cloud                         string
-	reportData                    []byte
-	checkCrl                      bool
 )
 
-var errEmptyFile = errors.New("input file is empty")
-
 func (cli *CLI) NewAttestationCmd() *cobra.Command {
 	return &cobra.Command{
 		Use:   "attestation [command]",
@@ -125,12 +95,12 @@ func (cli *CLI) NewGetAttestationCmd() *cobra.Command {
 		Args: cobra.ExactArgs(1),
 		Run: func(cmd *cobra.Command, args []string) {
 			if cli.connectErr != nil {
-				printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
+				cli.printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
 				return
 			}
 
 			if err := cobra.OnlyValidArgs(cmd, args); err != nil {
-				printError(cmd, "Bad attestation type: %v ❌ ", err)
+				cli.printError(cmd, "Bad attestation type: %v ❌ ", err)
 				return
 			}
 
@@ -171,10 +141,10 @@ func (cli *CLI) NewGetAttestationCmd() *cobra.Command {
 				return
 			}
 
-			var fixedReportData [quoteprovider.Nonce]byte
+			var fixedReportData [vtpm.SEVNonce]byte
 			if attType == attestation.SNP || attType == attestation.SNPvTPM {
-				if len(teeNonce) > quoteprovider.Nonce {
-					msg := color.New(color.FgRed).Sprintf("nonce must be a hex encoded string of length lesser or equal %d bytes ❌ ", quoteprovider.Nonce)
+				if len(teeNonce) > vtpm.SEVNonce {
+					msg := color.New(color.FgRed).Sprintf("nonce must be a hex encoded string of length lesser or equal %d bytes ❌ ", vtpm.SEVNonce)
 					cmd.Println(msg)
 					return
 				}
@@ -210,7 +180,7 @@ func (cli *CLI) NewGetAttestationCmd() *cobra.Command {
 
 			attestationFile, err := os.Create(filename)
 			if err != nil {
-				printError(cmd, "Error creating attestation file: %v ❌ ", err)
+				cli.printError(cmd, "Error creating attestation file: %v ❌ ", err)
 				return
 			}
 
@@ -219,27 +189,27 @@ func (cli *CLI) NewGetAttestationCmd() *cobra.Command {
 			if attestationType == AzureToken {
 				err := cli.agentSDK.AttestationToken(cmd.Context(), fixedVtpmNonceByte, int(attType), attestationFile)
 				if err != nil {
-					printError(cmd, "Failed to get attestation token due to error: %v ❌", err)
+					cli.printError(cmd, "Failed to get attestation token due to error: %v ❌", err)
 					return
 				}
 				returnJsonAzureToken = !getAzureTokenJWT
 			} else {
 				err := cli.agentSDK.Attestation(cmd.Context(), fixedReportData, fixedVtpmNonceByte, int(attType), attestationFile)
 				if err != nil {
-					printError(cmd, "Failed to get attestation due to error: %v ❌", err)
+					cli.printError(cmd, "Failed to get attestation due to error: %v ❌", err)
 					return
 				}
 			}
 
 			if err := attestationFile.Close(); err != nil {
-				printError(cmd, "Error closing attestation file: %v ❌ ", err)
+				cli.printError(cmd, "Error closing attestation file: %v ❌ ", err)
 				return
 			}
 
 			if getTextProtoAttestationReport || returnJsonAzureToken {
 				result, err := os.ReadFile(filename)
 				if err != nil {
-					printError(cmd, "Error reading attestation file: %v ❌ ", err)
+					cli.printError(cmd, "Error reading attestation file: %v ❌ ", err)
 					return
 				}
 
@@ -247,7 +217,7 @@ func (cli *CLI) NewGetAttestationCmd() *cobra.Command {
 				case SNP:
 					result, err = attestationToJSON(result)
 					if err != nil {
-						printError(cmd, "Error converting SNP attestation to JSON: %v ❌", err)
+						cli.printError(cmd, "Error converting SNP attestation to JSON: %v ❌", err)
 						return
 					}
 
@@ -259,7 +229,7 @@ func (cli *CLI) NewGetAttestationCmd() *cobra.Command {
 					var attvTPM tpmAttest.Attestation
 					err = proto.Unmarshal(result, &attvTPM)
 					if err != nil {
-						printError(cmd, "Failed to unmarshal the attestation report: %v ❌", err)
+						cli.printError(cmd, "Failed to unmarshal the attestation report: %v ❌", err)
 						return
 					}
 					result = []byte(marshalOptions.Format(&attvTPM))
@@ -267,13 +237,13 @@ func (cli *CLI) NewGetAttestationCmd() *cobra.Command {
 				case AzureToken:
 					result, err = decodeJWTToJSON(result)
 					if err != nil {
-						printError(cmd, "Error decoding Azure token: %v ❌", err)
+						cli.printError(cmd, "Error decoding Azure token: %v ❌", err)
 						return
 					}
 				}
 
 				if err := os.WriteFile(filename, result, 0o644); err != nil {
-					printError(cmd, "Error writing attestation file: %v ❌ ", err)
+					cli.printError(cmd, "Error writing attestation file: %v ❌ ", err)
 					return
 				}
 			}
@@ -303,186 +273,14 @@ func attestationToJSON(report []byte) ([]byte, error) {
 	return json.MarshalIndent(attestationPB, "", "	")
 }
 
-func attestationFromJSON(reportFile []byte) ([]byte, error) {
-	var attestationPB sevsnp.Attestation
-	if err := json.Unmarshal(reportFile, &attestationPB); err != nil {
-		return nil, err
-	}
-
-	return report.Transform(&attestationPB, "bin")
-}
-
-func isFileJSON(filename string) bool {
-	return strings.HasSuffix(filename, ".json")
-}
-
 func (cli *CLI) NewValidateAttestationValidationCmd() *cobra.Command {
-	cmd := &cobra.Command{
+	return &cobra.Command{
 		Use:   "validate",
-		Short: fmt.Sprintf("Validate and verify attestation information. You can define the confidential computing cloud provider (%s, %s, %s; %s is the default) and can choose from 4 modes: %s, %s, %s, and %s. Default mode is %s.", CCNone, CCAzure, CCGCP, CCNone, SNP, VTPM, SNPvTPM, TDX, SNP),
-		Example: `Based on mode:
-		validate <attestationreportfilepath> --report_data <reportdata> --product <product data> --platform <cc platform> //default
-		validate --mode snp <attestationreportfilepath> --report_data <reportdata> --product <product data>
-		validate --mode vtpm <attestationreportfilepath> --nonce <noncevalue> --format <formatvalue>  --output <outputvalue>
-		validate --mode snp-vtpm <attestationreportfilepath> --report_data <reportdata> --product <product data> --nonce <noncevalue> --format <formatvalue>  --output <outputvalue>
-		validate --mode tdx <attestationreportfilepath> --report_data <reportdata>
-		validate --cloud none --mode snp <attestationreportfilepath> --report_data <reportdata> --product <product data>
-		validate --cloud azure --mode vtpm <attestationreportfilepath> --nonce <noncevalue> --format <formatvalue>  --output <outputvalue>
-		validate --cloud gcp --mode snp-vtpm <attestationreportfilepath> --report_data <reportdata> --product <product data> --nonce <noncevalue> --format <formatvalue>  --output <outputvalue>`,
-		PreRunE: func(cmd *cobra.Command, args []string) error {
-			mode, _ := cmd.Flags().GetString("mode")
-			if len(args) != 1 {
-				return fmt.Errorf("please pass the attestation report file path")
-			}
-
-			// Validate flags based on the mode
-			switch mode {
-			case SNP:
-				if err := cmd.MarkFlagRequired("report_data"); err != nil {
-					return fmt.Errorf("failed to mark 'report_data' as required for SEV-%s mode: %v", SNP, err)
-				}
-				if err := cmd.MarkFlagRequired("product"); err != nil {
-					return fmt.Errorf("failed to mark flag as required: %v ❌ ", err)
-				}
-			case SNPvTPM:
-				if err := cmd.MarkFlagRequired("nonce"); err != nil {
-					return fmt.Errorf("failed to mark 'nonce' as required for %s mode: %v", VTPM, err)
-				}
-				if err := cmd.MarkFlagRequired("report_data"); err != nil {
-					return fmt.Errorf("failed to mark 'report_data' as required for SEV-%s mode: %v", SNP, err)
-				}
-				if err := cmd.MarkFlagRequired("product"); err != nil {
-					return fmt.Errorf("failed to mark flag as required: %v ❌ ", err)
-				}
-				if err := cmd.MarkFlagRequired("format"); err != nil {
-					return fmt.Errorf("failed to mark 'format' as required for %s mode: %v", VTPM, err)
-				}
-				if err := cmd.MarkFlagRequired("output"); err != nil {
-					return fmt.Errorf("failed to mark 'output' as required for %s mode: %v", VTPM, err)
-				}
-			case VTPM:
-				if err := cmd.MarkFlagRequired("nonce"); err != nil {
-					return fmt.Errorf("failed to mark 'nonce' as required for %s mode: %v", VTPM, err)
-				}
-				if err := cmd.MarkFlagRequired("format"); err != nil {
-					return fmt.Errorf("failed to mark 'format' as required for %s mode: %v", VTPM, err)
-				}
-				if err := cmd.MarkFlagRequired("output"); err != nil {
-					return fmt.Errorf("failed to mark 'output' as required for %s mode: %v", VTPM, err)
-				}
-			case TDX:
-				if err := cmd.MarkFlagRequired("report_data"); err != nil {
-					return fmt.Errorf("failed to mark 'report_data' as required for %s mode: %v", TDX, err)
-				}
-			default:
-				return fmt.Errorf("unknown mode: %s", mode)
-			}
-			return nil
+		Short: "Validate and verify attestation information (Deprecated)",
+		Run: func(cmd *cobra.Command, args []string) {
+			cmd.Println("Validation via CLI using legacy policies is deprecated. Please use CoRIM tools.")
 		},
-		RunE: func(cmd *cobra.Command, args []string) error {
-			mode, _ := cmd.Flags().GetString("mode")
-			cloud, _ := cmd.Flags().GetString("cloud")
-
-			output, err := createOutputFile()
-			if err != nil {
-				return fmt.Errorf("failed to create output file: %v ❌ ", err)
-			}
-			if closer, ok := output.(*os.File); ok {
-				defer closer.Close()
-			}
-
-			var verifier attestation.Verifier
-			switch cloud {
-			case CCNone:
-				policy := attestation.Config{Config: &cfg, PcrConfig: &attestation.PcrConfig{}}
-				verifier = vtpm.NewVerifierWithPolicy(nil, output, &policy)
-			case CCAzure:
-				policy := attestation.Config{Config: &cfg, PcrConfig: &attestation.PcrConfig{}}
-				verifier = azure.NewVerifierWithPolicy(output, &policy)
-			case CCGCP:
-				policy := attestation.Config{Config: &cfg, PcrConfig: &attestation.PcrConfig{}}
-				verifier = vtpm.NewVerifierWithPolicy(nil, output, &policy)
-			default:
-				policy := attestation.Config{Config: &cfg, PcrConfig: &attestation.PcrConfig{}}
-				verifier = vtpm.NewVerifierWithPolicy(nil, output, &policy)
-			}
-
-			switch mode {
-			case SNP:
-				cfg.Policy.ReportData = reportData
-				return sevsnpverify(cmd, verifier, args)
-			case SNPvTPM:
-				cfg.Policy.ReportData = reportData
-				return vtpmSevSnpverify(args, verifier)
-			case VTPM:
-				cfg.Policy.ReportData = reportData
-				return vtpmverify(args, verifier)
-			case TDX:
-				if err := validateTDXFlags(); err != nil {
-					return fmt.Errorf("failed to verify TDX validation flags: %v ❌ ", err)
-				}
-				verifier = tdx.NewVerifierWithPolicy(cfgTDX)
-				return tdxVerify(args[0], verifier)
-			default:
-				return fmt.Errorf("unknown mode: %s", mode)
-			}
-		},
-		SilenceUsage:  true,
-		SilenceErrors: true,
 	}
-	cmd.Flags().StringVar(
-		&cloud,
-		"cloud",
-		"none", // default CC provider
-		"The confidential computing cloud provider. Example: azure",
-	)
-
-	cmd.Flags().StringVar(
-		&mode,
-		"mode",
-		"snp", // default mode
-		"The attestation validation mode. Example: snp",
-	)
-
-	// VTPM FLAGS
-	cmd.Flags().BytesHexVar(
-		&nonce,
-		"nonce",
-		[]byte{},
-		"hex encoded nonce for vTPM attestation, cannot be empty",
-	)
-
-	cmd.Flags().StringVar(
-		&format,
-		"format",
-		"binarypb", // default value
-		"type of output file where attestation report stored <binarypb|textproto>",
-	)
-
-	cmd.Flags().StringVar(
-		&output,
-		"output",
-		"",
-		"output file",
-	)
-
-	cmd.Flags().StringVar(
-		&cfgString,
-		"config",
-		"",
-		"Path to the serialized json check.Config protobuf file. This will overwrite individual flags. Unmarshalled as json. Example: "+exampleJSONConfig,
-	)
-	cmd.Flags().BytesHexVar(
-		&reportData,
-		"report_data",
-		empty64[:],
-		"The expected REPORT_DATA field as a hex string. Must encode 64 bytes. Must be set.",
-	)
-
-	cmd = addSEVSNPVerificationOptions(cmd)
-	cmd = addTDXVerificationOptions(cmd)
-
-	return cmd
 }
 
 func (cli *CLI) NewMeasureCmd(igvmBinaryPath string) *cobra.Command {
@@ -523,27 +321,6 @@ func (cli *CLI) NewMeasureCmd(igvmBinaryPath string) *cobra.Command {
 	return igvmmeasureCmd
 }
 
-func openInputFile() (io.Reader, error) {
-	if attestationFile == "" {
-		return nil, errEmptyFile
-	}
-	return os.Open(attestationFile)
-}
-
-func createOutputFile() (io.Writer, error) {
-	if output == "" {
-		return os.Stdout, nil
-	}
-	return os.Create(output)
-}
-
-func validateFieldLength(fieldName string, field []byte, expectedLength int) error {
-	if field != nil && len(field) != expectedLength {
-		return fmt.Errorf("%s length should be at least %d bytes long", fieldName, expectedLength)
-	}
-	return nil
-}
-
 func decodeJWTToJSON(tokenBytes []byte) ([]byte, error) {
 	token := string(tokenBytes) // convert to string
 	parts := strings.Split(token, ".")

cli/attestation_policy.go

@@ -5,185 +5,33 @@ package cli
 import (
 	"bytes"
 	"crypto/sha512"
-	"encoding/base64"
-	"encoding/hex"
-	"encoding/json"
 	"fmt"
 	"os"
-	"strconv"
-	"strings"
 
-	"github.com/absmach/supermq/pkg/errors"
-	"github.com/google/go-sev-guest/proto/check"
 	"github.com/google/go-tpm-tools/proto/attest"
 	"github.com/spf13/cobra"
-	"github.com/spf13/pflag"
-	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"github.com/ultravioletrs/cocos/pkg/attestation/azure"
 	"github.com/ultravioletrs/cocos/pkg/attestation/gcp"
-	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 	"google.golang.org/protobuf/encoding/protojson"
 	"google.golang.org/protobuf/proto"
 )
 
-type fieldType int
-
-const (
-	measurementField fieldType = iota
-	hostDataField
-)
-
-const (
-	// 0o744 file permission gives RWX permission to the user and only the R permission to others.
-	filePermission = 0o744
-	// Length of the expected host data and measurement field in bytes.
-	hostDataLength    = 32
-	measurementLength = 48
-)
-
 var (
-	errDecode                              = errors.New("base64 string could not be decoded")
-	errDataLength                          = errors.New("data does not have an adequate length")
-	errReadingAttestationPolicyFile        = errors.New("error while reading the attestation policy file")
-	errUnmarshalJSON                       = errors.New("failed to unmarshal json")
-	errMarshalJSON                         = errors.New("failed to marshal json")
-	errWriteFile                           = errors.New("failed to write to file")
-	errAttestationPolicyField              = errors.New("the specified field type does not exist in the attestation policy")
-	errReadingManifestFile                 = errors.New("error while reading manifest file")
-	errDecodeHex                           = errors.New("error decoding hex string")
-	policy                          uint64 = 196639
-	isJsonAttestation               bool
+	isJsonAttestation bool
+	// 0o744 file permission gives RWX permission to the user and only the R permission to others.
+	filePermission os.FileMode = 0o744
 )
 
 func (cli *CLI) NewAttestationPolicyCmd() *cobra.Command {
-	return &cobra.Command{
-		Use:   "policy [command]",
+	cmd := &cobra.Command{
+		Use:   "policy",
 		Short: "Change attestation policy",
 		Run: func(cmd *cobra.Command, args []string) {
-			fmt.Printf("Change attestation policy\n\n")
-			fmt.Printf("Usage:\n  %s [command]\n\n", cmd.CommandPath())
-			fmt.Printf("Available Commands:\n")
-
-			// Filter out "completion" command
-			availableCommands := make([]*cobra.Command, 0)
-			for _, subCmd := range cmd.Commands() {
-				if subCmd.Name() != "completion" {
-					availableCommands = append(availableCommands, subCmd)
-				}
-			}
-
-			for _, subCmd := range availableCommands {
-				fmt.Printf("  %-15s%s\n", subCmd.Name(), subCmd.Short)
-			}
-
-			fmt.Printf("\nFlags:\n")
-			cmd.Flags().VisitAll(func(flag *pflag.Flag) {
-				fmt.Printf("  -%s, --%s %s\n", flag.Shorthand, flag.Name, flag.Usage)
-			})
-			fmt.Printf("\nUse \"%s [command] --help\" for more information about a command.\n", cmd.CommandPath())
-		},
-	}
-}
-
-func (cli *CLI) NewAddMeasurementCmd() *cobra.Command {
-	return &cobra.Command{
-		Use:     "measurement",
-		Short:   "Add measurement to the attestation policy file. The value should be in base64. The second parameter is attestation_policy.json file",
-		Example: "measurement <measurement> <attestation_policy.json>",
-		Args:    cobra.ExactArgs(2),
-		Run: func(cmd *cobra.Command, args []string) {
-			if err := changeAttestationConfiguration(args[1], args[0], measurementLength, measurementField); err != nil {
-				printError(cmd, "Error could not change measurement data: %v ❌ ", err)
-				return
-			}
-		},
-	}
-}
-
-func (cli *CLI) NewAddHostDataCmd() *cobra.Command {
-	return &cobra.Command{
-		Use:     "hostdata",
-		Short:   "Add host data to the attestation policy file. The value should be in base64. The second parameter is attestation_policy.json file",
-		Example: "hostdata <host-data> <attestation_policy.json>",
-		Args:    cobra.ExactArgs(2),
-		Run: func(cmd *cobra.Command, args []string) {
-			if err := changeAttestationConfiguration(args[1], args[0], hostDataLength, hostDataField); err != nil {
-				printError(cmd, "Error could not change host data: %v ❌ ", err)
-				return
-			}
-		},
-	}
-}
-
-func (cli *CLI) NewGCPAttestationPolicy() *cobra.Command {
-	cmd := &cobra.Command{
-		Use:     "gcp",
-		Short:   "Get attestation policy for GCP CVM",
-		Example: `gcp <bin_vtmp_attestation_report_file> <vcpu_count>`,
-		Args:    cobra.ExactArgs(2),
-		Run: func(cmd *cobra.Command, args []string) {
-			attestationBin, err := os.ReadFile(args[0])
-			if err != nil {
-				printError(cmd, "Error reading attestation report file: %v ❌ ", err)
-				return
-			}
-
-			vcpuCount, err := strconv.Atoi(args[1])
-			if err != nil {
-				printError(cmd, "Error converting vCPU count to integer: %v ❌ ", err)
-				return
-			}
-
-			attestation := &attest.Attestation{}
-
-			if isJsonAttestation {
-				if err := protojson.Unmarshal(attestationBin, attestation); err != nil {
-					printError(cmd, "Error converting JSON attestation to binary: %v ❌", err)
-					return
-				}
-			} else {
-				if err := proto.Unmarshal(attestationBin, attestation); err != nil {
-					printError(cmd, "Error unmarshaling attestation report: %v ❌ ", err)
-					return
-				}
-			}
-
-			attestationPB := attestation.GetSevSnpAttestation()
-
-			measurement, err := gcp.Extract384BitMeasurement(attestationPB)
-			if err != nil {
-				printError(cmd, "Error extracting 384-bit measurement: %v ❌ ", err)
-				return
-			}
-
-			launchEndorsement, err := gcp.GetLaunchEndorsement(cmd.Context(), measurement)
-			if err != nil {
-				printError(cmd, "Error getting launch endorsement: %v ❌ ", err)
-				return
-			}
-
-			attestationPolicy, err := gcp.GenerateAttestationPolicy(launchEndorsement, uint32(vcpuCount))
-			if err != nil {
-				printError(cmd, "Error generating attestation policy: %v ❌ ", err)
-				return
-			}
-
-			attestationPolicyJson, err := json.MarshalIndent(attestationPolicy, "", "  ")
-			if err != nil {
-				printError(cmd, "Error marshaling attestation policy: %v ❌ ", err)
-				return
-			}
-
-			if err := os.WriteFile("attestation_policy.json", attestationPolicyJson, filePermission); err != nil {
-				printError(cmd, "Error writing attestation policy file: %v ❌ ", err)
-				return
-			}
-
-			cmd.Println("Attestation policy file generated successfully ✅")
+			_ = cmd.Help()
 		},
 	}
 
-	cmd.Flags().BoolVarP(&isJsonAttestation, "json", "j", false, "Use JSON attestation report instead of binary")
+	cmd.AddCommand(cli.NewCreateCoRIMCmd())
+
 	return cmd
 }
 
@@ -196,7 +44,7 @@ func (cli *CLI) NewDownloadGCPOvmfFile() *cobra.Command {
 		Run: func(cmd *cobra.Command, args []string) {
 			attestationBin, err := os.ReadFile(args[0])
 			if err != nil {
-				printError(cmd, "Error reading attestation report file: %v ❌ ", err)
+				cli.printError(cmd, "Error reading attestation report file: %v ❌ ", err)
 				return
 			}
 
@@ -204,12 +52,12 @@ func (cli *CLI) NewDownloadGCPOvmfFile() *cobra.Command {
 
 			if isJsonAttestation {
 				if err := protojson.Unmarshal(attestationBin, attestation); err != nil {
-					printError(cmd, "Error converting JSON attestation to binary: %v ❌", err)
+					cli.printError(cmd, "Error converting JSON attestation to binary: %v ❌", err)
 					return
 				}
 			} else {
 				if err := proto.Unmarshal(attestationBin, attestation); err != nil {
-					printError(cmd, "Error unmarshaling attestation report: %v ❌ ", err)
+					cli.printError(cmd, "Error unmarshaling attestation report: %v ❌ ", err)
 					return
 				}
 			}
@@ -218,32 +66,32 @@ func (cli *CLI) NewDownloadGCPOvmfFile() *cobra.Command {
 
 			measurement, err := gcp.Extract384BitMeasurement(attestationPB)
 			if err != nil {
-				printError(cmd, "Error extracting 384-bit measurement: %v ❌ ", err)
+				cli.printError(cmd, "Error extracting 384-bit measurement: %v ❌ ", err)
 				return
 			}
 
 			launchEndorsement, err := gcp.GetLaunchEndorsement(cmd.Context(), measurement)
 			if err != nil {
-				printError(cmd, "Error getting launch endorsement: %v ❌ ", err)
+				cli.printError(cmd, "Error getting launch endorsement: %v ❌ ", err)
 				return
 			}
 
 			ovmf, err := gcp.DownloadOvmfFile(cmd.Context(), fmt.Sprintf("%x", launchEndorsement.Digest))
 			if err != nil {
-				printError(cmd, "Error downloading OVMF file: %v ❌ ", err)
+				cli.printError(cmd, "Error downloading OVMF file: %v ❌ ", err)
 				return
 			}
 
 			sum384 := sha512.Sum384(ovmf)
 
 			if !bytes.Equal(sum384[:], launchEndorsement.Digest) {
-				printError(cmd, "Error OVMF file does not match the measurement: %v ❌ ", fmt.Errorf("digest mismatch"))
+				cli.printError(cmd, "Error OVMF file does not match the measurement: %v ❌ ", fmt.Errorf("digest mismatch"))
 			} else {
 				cmd.Println("OVMF firmware in vm is unmodified ✅")
 			}
 
 			if err := os.WriteFile("ovmf.fd", ovmf, filePermission); err != nil {
-				printError(cmd, "Error writing OVMF file: %v ❌ ", err)
+				cli.printError(cmd, "Error writing OVMF file: %v ❌ ", err)
 				return
 			}
 
@@ -254,172 +102,3 @@ func (cli *CLI) NewDownloadGCPOvmfFile() *cobra.Command {
 	cmd.Flags().BoolVarP(&isJsonAttestation, "json", "j", false, "Use JSON attestation report instead of binary")
 	return cmd
 }
-
-func (cli *CLI) NewAzureAttestationPolicy() *cobra.Command {
-	cmd := &cobra.Command{
-		Use:     "azure",
-		Short:   "Get attestation policy for Azure CVM",
-		Example: `azure <azure_maa_token_file> <product_name>`,
-		Args:    cobra.ExactArgs(2),
-		Run: func(cmd *cobra.Command, args []string) {
-			token, err := os.ReadFile(args[0])
-			if err != nil {
-				printError(cmd, "Error reading attestation report file: %v ❌ ", err)
-				return
-			}
-
-			product := args[1]
-
-			config, err := azure.GenerateAttestationPolicy(string(token), product, policy)
-			if err != nil {
-				printError(cmd, "Error generating attestation policy: %v ❌ ", err)
-				return
-			}
-
-			attestationPolicyJson, err := json.MarshalIndent(&config, "", " ")
-			if err != nil {
-				printError(cmd, "Error marshaling attestation policy: %v ❌ ", err)
-				return
-			}
-
-			if err := os.WriteFile("attestation_policy.json", attestationPolicyJson, filePermission); err != nil {
-				printError(cmd, "Error writing attestation policy file: %v ❌ ", err)
-				return
-			}
-
-			cmd.Println("Attestation policy file generated successfully ✅")
-		},
-	}
-
-	cmd.Flags().Uint64Var(
-		&policy,
-		"policy",
-		policy,
-		"Policy of the guest CVM",
-	)
-
-	return cmd
-}
-
-func (cli *CLI) NewExtendWithManifestCmd() *cobra.Command {
-	return &cobra.Command{
-		Use:     "extend",
-		Short:   "Extends PCR16 with computation manifests. The first parameter is path to attestation policy file. The rest of the parameters are paths to computation manifest files.",
-		Example: "extend <attestation_policy_file_path> <computation_manifest_file_path> [<computation_manifest_file_path> ...]",
-		Args:    cobra.MinimumNArgs(2),
-		Run: func(cmd *cobra.Command, args []string) {
-			attestationPolicyFilePath := args[0]
-			manifestPaths := args[1:]
-			if err := extendWithManifest(attestationPolicyFilePath, manifestPaths); err != nil {
-				printError(cmd, "Error could not extend PCR16: %v ❌ ", err)
-				return
-			}
-		},
-	}
-}
-
-func changeAttestationConfiguration(fileName, base64Data string, expectedLength int, field fieldType) error {
-	data, err := base64.StdEncoding.DecodeString(base64Data)
-	if err != nil {
-		return errDecode
-	}
-
-	if len(data) != expectedLength {
-		return errDataLength
-	}
-
-	ac := attestation.Config{Config: &check.Config{RootOfTrust: &check.RootOfTrust{}, Policy: &check.Policy{}}, PcrConfig: &attestation.PcrConfig{}}
-
-	f, err := os.ReadFile(fileName)
-	if err != nil {
-		return errors.Wrap(errReadingAttestationPolicyFile, err)
-	}
-
-	if err = vtpm.ReadPolicyFromByte(f, &ac); err != nil {
-		return errors.Wrap(errUnmarshalJSON, err)
-	}
-
-	if ac.Config.Policy == nil {
-		ac.Config.Policy = &check.Policy{}
-	}
-
-	switch field {
-	case measurementField:
-		ac.Config.Policy.Measurement = data
-	case hostDataField:
-		ac.Config.Policy.HostData = data
-	default:
-		return errAttestationPolicyField
-	}
-
-	fileJson, err := vtpm.ConvertPolicyToJSON(&ac)
-	if err != nil {
-		return errors.Wrap(errMarshalJSON, err)
-	}
-	if err = os.WriteFile(fileName, fileJson, filePermission); err != nil {
-		return errors.Wrap(errWriteFile, err)
-	}
-	return nil
-}
-
-func extendWithManifest(attestationPolicyPath string, manifestPaths []string) error {
-	attestationConfig := attestation.Config{Config: &check.Config{RootOfTrust: &check.RootOfTrust{}, Policy: &check.Policy{}}, PcrConfig: &attestation.PcrConfig{}}
-
-	attestationPolicyFileData, err := os.ReadFile(attestationPolicyPath)
-	if err != nil {
-		return errors.Wrap(errReadingAttestationPolicyFile, err)
-	}
-
-	if err = vtpm.ReadPolicyFromByte(attestationPolicyFileData, &attestationConfig); err != nil {
-		return errors.Wrap(errUnmarshalJSON, err)
-	}
-
-	for _, manifestPath := range manifestPaths {
-		manifest, err := os.ReadFile(manifestPath)
-		if err != nil {
-			return errors.Wrap(errReadingManifestFile, err)
-		}
-
-		manifestSha256 := sha512.Sum512_256(manifest)
-		manifestSha384 := sha512.Sum384(manifest)
-
-		data256, exists256 := attestationConfig.PCRValues.Sha256["16"]
-
-		if !exists256 {
-			data256 = strings.Repeat("0", 64) // 32 bytes in hex
-		}
-
-		byteData256, err := hex.DecodeString(data256)
-		if err != nil {
-			return errors.Wrap(errDecodeHex, err)
-		}
-
-		newByteData256 := sha512.Sum512_256(append(byteData256, manifestSha256[:]...))
-
-		data384, exists384 := attestationConfig.PCRValues.Sha384["16"]
-
-		if !exists384 {
-			data384 = strings.Repeat("0", 96) // 48 bytes in hex
-		}
-
-		byteData384, err := hex.DecodeString(data384)
-		if err != nil {
-			return errors.Wrap(errDecodeHex, err)
-		}
-
-		newByteData384 := sha512.Sum384(append(byteData384, manifestSha384[:]...))
-
-		attestationConfig.PCRValues.Sha256["16"] = hex.EncodeToString(newByteData256[:])
-		attestationConfig.PCRValues.Sha384["16"] = hex.EncodeToString(newByteData384[:])
-	}
-
-	attestationPolicyJSON, err := vtpm.ConvertPolicyToJSON(&attestationConfig)
-	if err != nil {
-		return errors.Wrap(errMarshalJSON, err)
-	}
-	if err = os.WriteFile(attestationPolicyPath, attestationPolicyJSON, filePermission); err != nil {
-		return errors.Wrap(errWriteFile, err)
-	}
-
-	return nil
-}

cli/attestation_policy_corim.go

@@ -0,0 +1,289 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package cli
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+	"github.com/ultravioletrs/cocos/pkg/attestation/azure"
+	"github.com/ultravioletrs/cocos/pkg/attestation/corimgen"
+	"github.com/ultravioletrs/cocos/pkg/attestation/gcp"
+	"github.com/ultravioletrs/cocos/pkg/attestation/generator"
+)
+
+func (cli *CLI) NewCreateCoRIMCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "create-corim",
+		Short: "Create CoRIM attestation policy",
+		Long:  `Create CoRIM attestation policy for supported platforms (Azure, GCP, SNP, TDX)`,
+	}
+
+	cmd.AddCommand(cli.NewCreateCoRIMAzureCmd())
+	cmd.AddCommand(cli.NewCreateCoRIMGCPCmd())
+	cmd.AddCommand(cli.NewCreateCoRIMSNPCmd())
+	cmd.AddCommand(cli.NewCreateCoRIMTDXCmd())
+
+	return cmd
+}
+
+func (cli *CLI) NewCreateCoRIMAzureCmd() *cobra.Command {
+	var tokenPath string
+	var product string
+	var output string
+	var signingKeyPath string
+
+	cmd := &cobra.Command{
+		Use:   "azure",
+		Short: "Create CoRIM for Azure SEV-SNP",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			tokenBytes, err := os.ReadFile(tokenPath)
+			if err != nil {
+				return fmt.Errorf("failed to read token file: %w", err)
+			}
+
+			azureData, err := azure.ExtractAzureMeasurement(string(tokenBytes))
+			if err != nil {
+				return fmt.Errorf("failed to extract Azure measurements: %w", err)
+			}
+
+			opts := generator.Options{
+				Platform:    "snp",
+				Measurement: azureData.Measurement,
+				HostData:    azureData.HostData,
+				Policy:      azureData.Policy,
+				SVN:         azureData.SVN,
+				Product:     product,
+			}
+
+			if signingKeyPath != "" {
+				key, err := corimgen.LoadSigningKey(signingKeyPath)
+				if err != nil {
+					return fmt.Errorf("failed to load signing key: %w", err)
+				}
+				opts.SigningKey = key
+			}
+
+			cborBytes, err := generator.GenerateCoRIM(opts)
+			if err != nil {
+				return fmt.Errorf("failed to generate CoRIM: %w", err)
+			}
+
+			if output != "" {
+				if err := os.WriteFile(output, cborBytes, 0o644); err != nil {
+					return fmt.Errorf("failed to write output file: %w", err)
+				}
+				fmt.Fprintf(cmd.ErrOrStderr(), "CoRIM written to %s\n", output)
+			} else {
+				if _, err := cmd.OutOrStdout().Write(cborBytes); err != nil {
+					return fmt.Errorf("failed to write output: %w", err)
+				}
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&tokenPath, "token", "", "Path to file containing Azure Attestation Token (JWT)")
+	cmd.Flags().StringVar(&product, "product", "Milan", "Processor product name (Milan, Genoa)")
+	cmd.Flags().StringVar(&output, "output", "", "Output file path (default: stdout)")
+	cmd.Flags().StringVar(&signingKeyPath, "signing-key", "", "Path to private key for signing (PEM format)")
+	_ = cmd.MarkFlagRequired("token")
+
+	return cmd
+}
+
+func (cli *CLI) NewCreateCoRIMGCPCmd() *cobra.Command {
+	var measurement string
+	var vcpuNum uint32
+	var output string
+	var signingKeyPath string
+
+	cmd := &cobra.Command{
+		Use:   "gcp",
+		Short: "Create CoRIM for GCP SEV-SNP",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := context.Background()
+
+			endorsement, err := gcp.GetLaunchEndorsement(ctx, measurement)
+			if err != nil {
+				return fmt.Errorf("failed to get launch endorsement: %w", err)
+			}
+
+			gcpData, err := gcp.ExtractGCPMeasurement(endorsement, vcpuNum)
+			if err != nil {
+				return fmt.Errorf("failed to extract GCP measurements: %w", err)
+			}
+
+			opts := generator.Options{
+				Platform:    "snp",
+				Measurement: gcpData.Measurement,
+				Policy:      gcpData.Policy,
+			}
+
+			if signingKeyPath != "" {
+				key, err := corimgen.LoadSigningKey(signingKeyPath)
+				if err != nil {
+					return fmt.Errorf("failed to load signing key: %w", err)
+				}
+				opts.SigningKey = key
+			}
+
+			cborBytes, err := generator.GenerateCoRIM(opts)
+			if err != nil {
+				return fmt.Errorf("failed to generate CoRIM: %w", err)
+			}
+
+			if output != "" {
+				if err := os.WriteFile(output, cborBytes, 0o644); err != nil {
+					return fmt.Errorf("failed to write output file: %w", err)
+				}
+				fmt.Fprintf(cmd.ErrOrStderr(), "CoRIM written to %s\n", output)
+			} else {
+				if _, err := cmd.OutOrStdout().Write(cborBytes); err != nil {
+					return fmt.Errorf("failed to write output: %w", err)
+				}
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&measurement, "measurement", "", "384-bit measurement hex string")
+	cmd.Flags().Uint32Var(&vcpuNum, "vcpu", 0, "vCPU number")
+	cmd.Flags().StringVar(&output, "output", "", "Output file path (default: stdout)")
+	cmd.Flags().StringVar(&signingKeyPath, "signing-key", "", "Path to private key for signing (PEM format)")
+	_ = cmd.MarkFlagRequired("measurement")
+
+	return cmd
+}
+
+func (cli *CLI) NewCreateCoRIMSNPCmd() *cobra.Command {
+	var (
+		measurement    string
+		policy         uint64
+		svn            uint64
+		product        string
+		hostData       string
+		launchTCB      uint64
+		output         string
+		signingKeyPath string
+	)
+
+	cmd := &cobra.Command{
+		Use:   "snp",
+		Short: "Create CoRIM for SEV-SNP",
+		Long:  `Generate CoRIM attestation policy for AMD SEV-SNP platform`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			opts := generator.Options{
+				Platform:    "snp",
+				Measurement: measurement,
+				Policy:      policy,
+				SVN:         svn,
+				Product:     product,
+				HostData:    hostData,
+				LaunchTCB:   launchTCB,
+			}
+
+			if signingKeyPath != "" {
+				key, err := corimgen.LoadSigningKey(signingKeyPath)
+				if err != nil {
+					return fmt.Errorf("failed to load signing key: %w", err)
+				}
+				opts.SigningKey = key
+			}
+
+			cborBytes, err := generator.GenerateCoRIM(opts)
+			if err != nil {
+				return fmt.Errorf("failed to generate CoRIM: %w", err)
+			}
+
+			if output != "" {
+				if err := os.WriteFile(output, cborBytes, 0o644); err != nil {
+					return fmt.Errorf("failed to write output file: %w", err)
+				}
+				fmt.Fprintf(cmd.ErrOrStderr(), "CoRIM written to %s\n", output)
+			} else {
+				if _, err := cmd.OutOrStdout().Write(cborBytes); err != nil {
+					return fmt.Errorf("failed to write output: %w", err)
+				}
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&measurement, "measurement", "", "Measurement/Launch Digest (hex string, defaults to zero if not provided)")
+	cmd.Flags().Uint64Var(&policy, "policy", 0, "SNP policy flags")
+	cmd.Flags().Uint64Var(&svn, "svn", 0, "Security Version Number (TCB)")
+	cmd.Flags().StringVar(&product, "product", "Milan", "Processor product name (Milan, Genoa, etc.)")
+	cmd.Flags().StringVar(&hostData, "host-data", "", "Host data (hex string)")
+	cmd.Flags().Uint64Var(&launchTCB, "launch-tcb", 0, "Minimum launch TCB")
+	cmd.Flags().StringVar(&output, "output", "", "Output file path (default: stdout)")
+	cmd.Flags().StringVar(&signingKeyPath, "signing-key", "", "Path to private key for signing (PEM format)")
+
+	return cmd
+}
+
+func (cli *CLI) NewCreateCoRIMTDXCmd() *cobra.Command {
+	var (
+		measurement    string
+		svn            uint64
+		rtmrs          string
+		mrSeam         string
+		output         string
+		signingKeyPath string
+	)
+
+	cmd := &cobra.Command{
+		Use:   "tdx",
+		Short: "Create CoRIM for Intel TDX",
+		Long:  `Generate CoRIM attestation policy for Intel TDX platform`,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			opts := generator.Options{
+				Platform:    "tdx",
+				Measurement: measurement,
+				SVN:         svn,
+				RTMRs:       rtmrs,
+				MrSeam:      mrSeam,
+			}
+
+			if signingKeyPath != "" {
+				key, err := corimgen.LoadSigningKey(signingKeyPath)
+				if err != nil {
+					return fmt.Errorf("failed to load signing key: %w", err)
+				}
+				opts.SigningKey = key
+			}
+
+			cborBytes, err := generator.GenerateCoRIM(opts)
+			if err != nil {
+				return fmt.Errorf("failed to generate CoRIM: %w", err)
+			}
+
+			if output != "" {
+				if err := os.WriteFile(output, cborBytes, 0o644); err != nil {
+					return fmt.Errorf("failed to write output file: %w", err)
+				}
+				fmt.Fprintf(cmd.ErrOrStderr(), "CoRIM written to %s\n", output)
+			} else {
+				if _, err := cmd.OutOrStdout().Write(cborBytes); err != nil {
+					return fmt.Errorf("failed to write output: %w", err)
+				}
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&measurement, "measurement", "", "MRTD measurement (hex string, uses default if not provided)")
+	cmd.Flags().Uint64Var(&svn, "svn", 0, "Security Version Number")
+	cmd.Flags().StringVar(&rtmrs, "rtmrs", "", "Comma-separated RTMRs (hex)")
+	cmd.Flags().StringVar(&mrSeam, "mr-seam", "", "MRSEAM (hex)")
+	cmd.Flags().StringVar(&output, "output", "", "Output file path (default: stdout)")
+	cmd.Flags().StringVar(&signingKeyPath, "signing-key", "", "Path to private key for signing (PEM format)")
+
+	return cmd
+}

cli/attestation_policy_corim_test.go

@@ -0,0 +1,389 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package cli
+
+import (
+	"bytes"
+	"context"
+	"io"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/google/gce-tcb-verifier/proto/endorsement"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"github.com/ultravioletrs/cocos/pkg/attestation/azure"
+	"github.com/ultravioletrs/cocos/pkg/attestation/gcp"
+	"google.golang.org/protobuf/proto"
+)
+
+func TestCLI_NewCreateCoRIMCmd(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMCmd()
+
+	assert.NotNil(t, cmd)
+	assert.Equal(t, "create-corim", cmd.Use)
+	assert.True(t, cmd.HasSubCommands())
+
+	subcmds := cmd.Commands()
+	assert.Equal(t, 4, len(subcmds))
+
+	cmdNames := make(map[string]bool)
+	for _, sc := range subcmds {
+		cmdNames[sc.Name()] = true
+	}
+
+	assert.True(t, cmdNames["azure"])
+	assert.True(t, cmdNames["gcp"])
+	assert.True(t, cmdNames["snp"])
+	assert.True(t, cmdNames["tdx"])
+}
+
+func TestCLI_NewCreateCoRIMSNPCmd(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMSNPCmd()
+
+	assert.NotNil(t, cmd)
+	assert.Equal(t, "snp", cmd.Use)
+
+	// Test with minimal flags
+	var outBuf bytes.Buffer
+	cmd.SetOut(&outBuf)
+	cmd.SetErr(&outBuf)
+	cmd.SetArgs([]string{"--measurement", "00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff"})
+
+	err := cmd.Execute()
+	assert.NoError(t, err)
+	assert.NotEmpty(t, outBuf.Bytes())
+}
+
+func TestCLI_NewCreateCoRIMTDXCmd(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMTDXCmd()
+
+	assert.NotNil(t, cmd)
+	assert.Equal(t, "tdx", cmd.Use)
+
+	// Test with minimal flags
+	var outBuf bytes.Buffer
+	cmd.SetOut(&outBuf)
+	cmd.SetErr(&outBuf)
+	cmd.SetArgs([]string{"--measurement", "00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff"})
+
+	err := cmd.Execute()
+	assert.NoError(t, err)
+	assert.NotEmpty(t, outBuf.Bytes())
+}
+
+func TestCLI_NewCreateCoRIMAzureCmd_Error(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMAzureCmd()
+
+	// Missing token flag
+	cmd.SetArgs([]string{})
+	err := cmd.Execute()
+	assert.Error(t, err)
+
+	// Non-existent token file
+	cmd.SetArgs([]string{"--token", "non-existent-file"})
+	err = cmd.Execute()
+	assert.Error(t, err)
+	assert.Contains(t, err.Error(), "failed to read token file")
+}
+
+func TestCLI_NewCreateCoRIMGCPCmd_Error(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMGCPCmd()
+
+	// Missing measurement flag
+	cmd.SetArgs([]string{})
+	err := cmd.Execute()
+	assert.Error(t, err)
+
+	// GCP command will fail because it tries to call Google Cloud Storage
+	cmd.SetArgs([]string{"--measurement", "00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff"})
+	err = cmd.Execute()
+	assert.Error(t, err)
+	// It should fail at GetLaunchEndorsement or storage client creation
+}
+
+func TestCLI_NewCreateCoRIMAzureCmd_Success(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMAzureCmd()
+
+	oldValidator := azure.DefaultValidator
+	defer func() { azure.DefaultValidator = oldValidator }()
+
+	azure.DefaultValidator = &mockTokenValidator{
+		validateFunc: func(token string) (map[string]any, error) {
+			return map[string]any{
+				"x-ms-isolation-tee": map[string]any{
+					"x-ms-sevsnpvm-launchmeasurement": "00112233",
+					"x-ms-sevsnpvm-guestsvn":          1.0,
+				},
+			}, nil
+		},
+	}
+
+	tmpDir := t.TempDir()
+	tokenPath := filepath.Join(tmpDir, "token.jwt")
+	// Dummy token
+	dummyToken := "eyJhbGciOiJub25lIn0.eyJoZWFkZXIiOiJkYXRhIn0."
+	err := os.WriteFile(tokenPath, []byte(dummyToken), 0o644)
+	require.NoError(t, err)
+
+	var outBuf bytes.Buffer
+	cmd.SetOut(&outBuf)
+	cmd.SetErr(&outBuf)
+	cmd.SetArgs([]string{"--token", tokenPath})
+
+	err = cmd.Execute()
+	assert.NoError(t, err)
+	assert.NotEmpty(t, outBuf.Bytes())
+
+	// Test with output file
+	outputFile := filepath.Join(tmpDir, "azure-corim.cbor")
+	cmd.SetArgs([]string{"--token", tokenPath, "--output", outputFile})
+	err = cmd.Execute()
+	assert.NoError(t, err)
+	_, err = os.Stat(outputFile)
+	assert.NoError(t, err)
+
+	// Test with signing key
+	keyPath := filepath.Join(tmpDir, "key.pem")
+	err = os.WriteFile(keyPath, []byte("-----BEGIN PRIVATE KEY-----\nMC4CAQAwBQYDK2VwBCIEIJ+3b6N6Y9J2H9f9X9X9X9X9X9X9X9X9X9X9X9X9X9X9\n-----END PRIVATE KEY-----"), 0o644)
+	require.NoError(t, err)
+	cmd.SetArgs([]string{"--token", tokenPath, "--signing-key", keyPath})
+	err = cmd.Execute()
+	assert.Error(t, err) // Should fail with invalid key but we cover the path
+	// This might fail if the key is not valid Ed25519 for corimgen, but we want to cover the path
+}
+
+func TestCLI_NewCreateCoRIMGCPCmd_More(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMGCPCmd()
+
+	oldNewStorageClient := gcp.NewStorageClient
+	defer func() { gcp.NewStorageClient = oldNewStorageClient }()
+
+	gcp.NewStorageClient = func(ctx context.Context) (gcp.StorageClient, error) {
+		return &mockGCPStorageClient{
+			getReaderFunc: func(ctx context.Context, bucket, object string) (io.ReadCloser, error) {
+				goldenUEFI := &endorsement.VMGoldenMeasurement{
+					SevSnp: &endorsement.VMSevSnp{
+						Policy:       123,
+						Measurements: map[uint32][]byte{1: {0x1, 0x2}},
+					},
+				}
+				goldenBytes, _ := proto.Marshal(goldenUEFI)
+				launchEndorsement := &endorsement.VMLaunchEndorsement{
+					SerializedUefiGolden: goldenBytes,
+				}
+				launchBytes, _ := proto.Marshal(launchEndorsement)
+				return io.NopCloser(bytes.NewReader(launchBytes)), nil
+			},
+			closeFunc: func() error { return nil },
+		}, nil
+	}
+
+	tmpDir := t.TempDir()
+	outputFile := filepath.Join(tmpDir, "gcp-corim.cbor")
+
+	var outBuf bytes.Buffer
+	cmd.SetOut(&outBuf)
+	cmd.SetErr(&outBuf)
+	cmd.SetArgs([]string{"--measurement", "00112233", "--vcpu", "1", "--output", outputFile})
+
+	err := cmd.Execute()
+	assert.NoError(t, err)
+	_, err = os.Stat(outputFile)
+	assert.NoError(t, err)
+}
+
+func TestCLI_NewCreateCoRIMSNPCmd_More(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMSNPCmd()
+
+	tmpDir := t.TempDir()
+	outputFile := filepath.Join(tmpDir, "snp-corim.cbor")
+
+	cmd.SetArgs([]string{
+		"--measurement", "00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff",
+		"--policy", "1",
+		"--svn", "1",
+		"--product", "Genoa",
+		"--host-data", "00112233",
+		"--launch-tcb", "1",
+		"--output", outputFile,
+	})
+
+	err := cmd.Execute()
+	assert.NoError(t, err)
+	_, err = os.Stat(outputFile)
+	assert.NoError(t, err)
+}
+
+func TestCLI_NewCreateCoRIMTDXCmd_More(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMTDXCmd()
+
+	tmpDir := t.TempDir()
+	outputFile := filepath.Join(tmpDir, "tdx-corim.cbor")
+
+	cmd.SetArgs([]string{
+		"--measurement", "00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff00112233445566778899aabbccddeeff",
+		"--svn", "1",
+		"--rtmrs", "0011,2233",
+		"--mr-seam", "aabbcc",
+		"--output", outputFile,
+	})
+
+	err := cmd.Execute()
+	assert.NoError(t, err)
+	_, err = os.Stat(outputFile)
+	assert.NoError(t, err)
+}
+
+func TestCLI_NewCreateCoRIMCmd_Errors(t *testing.T) {
+	cli := &CLI{}
+	tmpDir := t.TempDir()
+
+	t.Run("Azure fail to read token", func(t *testing.T) {
+		cmd := cli.NewCreateCoRIMAzureCmd()
+		cmd.SetArgs([]string{"--token", filepath.Join(tmpDir, "non-existent")})
+		err := cmd.Execute()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "failed to read token file")
+	})
+
+	t.Run("Azure invalid signing key", func(t *testing.T) {
+		cmd := cli.NewCreateCoRIMAzureCmd()
+		oldValidator := azure.DefaultValidator
+		defer func() { azure.DefaultValidator = oldValidator }()
+
+		azure.DefaultValidator = &mockTokenValidator{
+			validateFunc: func(token string) (map[string]any, error) {
+				return map[string]any{
+					"x-ms-isolation-tee": map[string]any{
+						"x-ms-sevsnpvm-launchmeasurement": "00112233",
+						"x-ms-sevsnpvm-guestsvn":          1.0,
+					},
+				}, nil
+			},
+		}
+
+		tokenPath := filepath.Join(tmpDir, "token.jwt")
+		_ = os.WriteFile(tokenPath, []byte("token"), 0o644)
+		cmd.SetArgs([]string{"--token", tokenPath, "--signing-key", filepath.Join(tmpDir, "non-existent")})
+		err := cmd.Execute()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "failed to load signing key")
+	})
+
+	t.Run("GCP fail to load signing key", func(t *testing.T) {
+		cmd := cli.NewCreateCoRIMGCPCmd()
+
+		oldNewStorageClient := gcp.NewStorageClient
+		defer func() { gcp.NewStorageClient = oldNewStorageClient }()
+
+		gcp.NewStorageClient = func(ctx context.Context) (gcp.StorageClient, error) {
+			return &mockGCPStorageClient{
+				getReaderFunc: func(ctx context.Context, bucket, object string) (io.ReadCloser, error) {
+					goldenUEFI := &endorsement.VMGoldenMeasurement{
+						SevSnp: &endorsement.VMSevSnp{
+							Policy:       123,
+							Measurements: map[uint32][]byte{1: {0x1, 0x2}},
+						},
+					}
+					goldenBytes, _ := proto.Marshal(goldenUEFI)
+					launchEndorsement := &endorsement.VMLaunchEndorsement{
+						SerializedUefiGolden: goldenBytes,
+					}
+					launchBytes, _ := proto.Marshal(launchEndorsement)
+					return io.NopCloser(bytes.NewReader(launchBytes)), nil
+				},
+				closeFunc: func() error { return nil },
+			}, nil
+		}
+
+		cmd.SetArgs([]string{"--measurement", "0011", "--vcpu", "1", "--signing-key", filepath.Join(tmpDir, "non-existent")})
+		err := cmd.Execute()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "failed to load signing key")
+	})
+
+	t.Run("SNP fail to load signing key", func(t *testing.T) {
+		cmd := cli.NewCreateCoRIMSNPCmd()
+		cmd.SetArgs([]string{"--measurement", "0011", "--signing-key", filepath.Join(tmpDir, "non-existent")})
+		err := cmd.Execute()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "failed to load signing key")
+	})
+
+	t.Run("TDX fail to load signing key", func(t *testing.T) {
+		cmd := cli.NewCreateCoRIMTDXCmd()
+		cmd.SetArgs([]string{"--measurement", "0011", "--signing-key", filepath.Join(tmpDir, "non-existent")})
+		err := cmd.Execute()
+		assert.Error(t, err)
+		assert.Contains(t, err.Error(), "failed to load signing key")
+	})
+}
+
+type mockTokenValidator struct {
+	validateFunc func(token string) (map[string]any, error)
+}
+
+func (m *mockTokenValidator) Validate(token string) (map[string]any, error) {
+	return m.validateFunc(token)
+}
+
+func TestCLI_NewCreateCoRIMGCPCmd_Success(t *testing.T) {
+	cli := &CLI{}
+	cmd := cli.NewCreateCoRIMGCPCmd()
+
+	oldNewStorageClient := gcp.NewStorageClient
+	defer func() { gcp.NewStorageClient = oldNewStorageClient }()
+
+	gcp.NewStorageClient = func(ctx context.Context) (gcp.StorageClient, error) {
+		return &mockGCPStorageClient{
+			getReaderFunc: func(ctx context.Context, bucket, object string) (io.ReadCloser, error) {
+				goldenUEFI := &endorsement.VMGoldenMeasurement{
+					SevSnp: &endorsement.VMSevSnp{
+						Policy:       123,
+						Measurements: map[uint32][]byte{1: {0x1, 0x2}},
+					},
+				}
+				goldenBytes, _ := proto.Marshal(goldenUEFI)
+				launchEndorsement := &endorsement.VMLaunchEndorsement{
+					SerializedUefiGolden: goldenBytes,
+				}
+				launchBytes, _ := proto.Marshal(launchEndorsement)
+				return io.NopCloser(bytes.NewReader(launchBytes)), nil
+			},
+			closeFunc: func() error { return nil },
+		}, nil
+	}
+
+	var outBuf bytes.Buffer
+	cmd.SetOut(&outBuf)
+	cmd.SetErr(&outBuf)
+	cmd.SetArgs([]string{"--measurement", "00112233", "--vcpu", "1"})
+
+	err := cmd.Execute()
+	assert.NoError(t, err)
+	assert.NotEmpty(t, outBuf.Bytes())
+}
+
+type mockGCPStorageClient struct {
+	getReaderFunc func(ctx context.Context, bucket, object string) (io.ReadCloser, error)
+	closeFunc     func() error
+}
+
+func (m *mockGCPStorageClient) GetReader(ctx context.Context, bucket, object string) (io.ReadCloser, error) {
+	return m.getReaderFunc(ctx, bucket, object)
+}
+
+func (m *mockGCPStorageClient) Close() error {
+	return m.closeFunc()
+}

cli/attestation_policy_test.go

@@ -4,467 +4,114 @@ package cli
 
 import (
 	"bytes"
-	"encoding/base64"
+	"context"
+	"io"
 	"os"
+	"path/filepath"
 	"testing"
 
-	"github.com/google/go-sev-guest/proto/check"
+	"github.com/google/gce-tcb-verifier/proto/endorsement"
+	"github.com/google/go-sev-guest/proto/sevsnp"
+	"github.com/google/go-tpm-tools/proto/attest"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
+	"github.com/ultravioletrs/cocos/pkg/attestation/gcp"
+	"google.golang.org/protobuf/proto"
 )
 
-func TestChangeAttestationConfiguration(t *testing.T) {
-	tmpfile, err := os.CreateTemp("", "attestation_policy.json")
-	require.NoError(t, err)
-	defer os.Remove(tmpfile.Name())
-
-	initialConfig := attestation.Config{Config: &check.Config{RootOfTrust: &check.RootOfTrust{}, Policy: &check.Policy{}}, PcrConfig: &attestation.PcrConfig{}}
-
-	initialJSON, err := vtpm.ConvertPolicyToJSON(&initialConfig)
-	require.NoError(t, err)
-	err = os.WriteFile(tmpfile.Name(), initialJSON, 0o644)
-	require.NoError(t, err)
-
-	tests := []struct {
-		name           string
-		base64Data     string
-		expectedLength int
-		field          fieldType
-		expectError    bool
-		errorType      error
-	}{
-		{
-			name:           "Valid Measurement",
-			base64Data:     base64.StdEncoding.EncodeToString(make([]byte, measurementLength)),
-			expectedLength: measurementLength,
-			field:          measurementField,
-			expectError:    false,
-		},
-		{
-			name:           "Valid Host Data",
-			base64Data:     base64.StdEncoding.EncodeToString(make([]byte, hostDataLength)),
-			expectedLength: hostDataLength,
-			field:          hostDataField,
-			expectError:    false,
-		},
-		{
-			name:           "Invalid Base64",
-			base64Data:     "Invalid Base64",
-			expectedLength: measurementLength,
-			field:          measurementField,
-			expectError:    true,
-			errorType:      errDecode,
-		},
-		{
-			name:           "Invalid Data Length",
-			base64Data:     base64.StdEncoding.EncodeToString(make([]byte, measurementLength-1)),
-			expectedLength: measurementLength,
-			field:          measurementField,
-			expectError:    true,
-			errorType:      errDataLength,
-		},
-		{
-			name:           "Invalid Field Type",
-			base64Data:     base64.StdEncoding.EncodeToString(make([]byte, measurementLength)),
-			expectedLength: measurementLength,
-			field:          fieldType(999),
-			expectError:    true,
-			errorType:      errAttestationPolicyField,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			err := changeAttestationConfiguration(tmpfile.Name(), tt.base64Data, tt.expectedLength, tt.field)
-
-			if tt.expectError {
-				assert.Error(t, err)
-				assert.ErrorIs(t, err, tt.errorType)
-			} else {
-				assert.NoError(t, err)
-
-				content, err := os.ReadFile(tmpfile.Name())
-				require.NoError(t, err)
-
-				ap := attestation.Config{Config: &check.Config{RootOfTrust: &check.RootOfTrust{}, Policy: &check.Policy{}}, PcrConfig: &attestation.PcrConfig{}}
-				err = vtpm.ReadPolicyFromByte(content, &ap)
-				require.NoError(t, err)
-
-				decodedData, _ := base64.StdEncoding.DecodeString(tt.base64Data)
-				if tt.field == measurementField {
-					assert.Equal(t, decodedData, ap.Config.Policy.Measurement)
-				} else if tt.field == hostDataField {
-					assert.Equal(t, decodedData, ap.Config.Policy.HostData)
-				}
-			}
-		})
-	}
-}
-
 func TestNewAttestationPolicyCmd(t *testing.T) {
-	cli := &CLI{}
-	cmd := cli.NewAttestationPolicyCmd()
+	c := &CLI{}
+	cmd := c.NewAttestationPolicyCmd()
 
-	assert.Equal(t, "policy [command]", cmd.Use)
+	assert.Equal(t, "policy", cmd.Use)
 	assert.Equal(t, "Change attestation policy", cmd.Short)
 	assert.NotNil(t, cmd.Run)
 }
 
-func TestNewAddMeasurementCmd(t *testing.T) {
-	cli := &CLI{}
-	cmd := cli.NewAddMeasurementCmd()
-
-	assert.Equal(t, "measurement", cmd.Use)
-	assert.Equal(t, "Add measurement to the attestation policy file. The value should be in base64. The second parameter is attestation_policy.json file", cmd.Short)
-	assert.Equal(t, "measurement <measurement> <attestation_policy.json>", cmd.Example)
-	assert.NotNil(t, cmd.Run)
-}
-
-func TestNewAddHostDataCmd(t *testing.T) {
-	cli := &CLI{}
-	cmd := cli.NewAddHostDataCmd()
-
-	assert.Equal(t, "hostdata", cmd.Use)
-	assert.Equal(t, "Add host data to the attestation policy file. The value should be in base64. The second parameter is attestation_policy.json file", cmd.Short)
-	assert.Equal(t, "hostdata <host-data> <attestation_policy.json>", cmd.Example)
-	assert.NotNil(t, cmd.Run)
-}
-
-func TestChangeAttestationConfigurationFileErrors(t *testing.T) {
-	t.Run("File Not Found", func(t *testing.T) {
-		err := changeAttestationConfiguration("nonexistent.json", base64.StdEncoding.EncodeToString(make([]byte, measurementLength)), measurementLength, measurementField)
-		assert.Error(t, err)
-		assert.Contains(t, err.Error(), "error while reading the attestation policy file")
-	})
-
-	t.Run("Invalid JSON Content", func(t *testing.T) {
-		tmpfile, err := os.CreateTemp("", "invalid.json")
-		require.NoError(t, err)
-		defer os.Remove(tmpfile.Name())
-
-		err = os.WriteFile(tmpfile.Name(), []byte("invalid json"), 0o644)
-		require.NoError(t, err)
-
-		err = changeAttestationConfiguration(tmpfile.Name(), base64.StdEncoding.EncodeToString(make([]byte, measurementLength)), measurementLength, measurementField)
-		assert.Error(t, err)
-		assert.Contains(t, err.Error(), "failed to unmarshal json")
-	})
-}
-
-func TestNewGCPAttestationPolicy(t *testing.T) {
-	cli := &CLI{}
-	cmd := cli.NewGCPAttestationPolicy()
-
-	assert.Equal(t, "gcp", cmd.Use)
-	assert.Equal(t, "Get attestation policy for GCP CVM", cmd.Short)
-	assert.Equal(t, "gcp <bin_vtmp_attestation_report_file> <vcpu_count>", cmd.Example)
-	assert.NotNil(t, cmd.Run)
-
-	t.Run("File Not Found", func(t *testing.T) {
-		cmd.SetArgs([]string{"nonexistent.bin", "4"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err := cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "Error reading attestation report file")
-		assert.Contains(t, output, "❌")
-	})
-
-	t.Run("Invalid vCPU Count", func(t *testing.T) {
-		tmpfile, err := os.CreateTemp("", "attestation.bin")
-		require.NoError(t, err)
-		defer os.Remove(tmpfile.Name())
-
-		err = os.WriteFile(tmpfile.Name(), []byte("dummy content"), 0o644)
-		require.NoError(t, err)
-
-		cmd.SetArgs([]string{tmpfile.Name(), "invalid"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err = cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "Error converting vCPU count to integer")
-		assert.Contains(t, output, "❌")
-	})
-
-	t.Run("Invalid Attestation Data", func(t *testing.T) {
-		tmpfile, err := os.CreateTemp("", "attestation.bin")
-		require.NoError(t, err)
-		defer os.Remove(tmpfile.Name())
-
-		err = os.WriteFile(tmpfile.Name(), []byte("invalid protobuf data"), 0o644)
-		require.NoError(t, err)
-
-		cmd.SetArgs([]string{tmpfile.Name(), "4"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err = cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "Error unmarshaling attestation report")
-		assert.Contains(t, output, "❌")
-	})
-}
-
-func TestNewDownloadGCPOvmfFile(t *testing.T) {
+func TestCLI_NewDownloadGCPOvmfFile(t *testing.T) {
 	cli := &CLI{}
 	cmd := cli.NewDownloadGCPOvmfFile()
 
+	assert.NotNil(t, cmd)
 	assert.Equal(t, "download", cmd.Use)
-	assert.Equal(t, "Download GCP OVMF file", cmd.Short)
-	assert.Equal(t, "download <bin_vtmp_attestation_report_file>", cmd.Example)
-	assert.NotNil(t, cmd.Run)
 
-	t.Run("File Not Found", func(t *testing.T) {
-		cmd.SetArgs([]string{"nonexistent.bin"})
+	oldNewStorageClient := gcp.NewStorageClient
+	defer func() { gcp.NewStorageClient = oldNewStorageClient }()
 
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
+	tmpDir := t.TempDir()
+	attestationPath := filepath.Join(tmpDir, "attestation.bin")
 
+	// Change working directory to tmpDir so ovmf.fd is written there
+	oldWd, err := os.Getwd()
+	require.NoError(t, err)
+	err = os.Chdir(tmpDir)
+	require.NoError(t, err)
+	defer func() {
+		_ = os.Chdir(oldWd)
+	}()
+
+	t.Run("invalid attestation file", func(t *testing.T) {
+		var outBuf bytes.Buffer
+		cmd.SetOut(&outBuf)
+		cmd.SetErr(&outBuf)
+		cmd.SetArgs([]string{"non-existent"})
 		err := cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "Error reading attestation report file")
-		assert.Contains(t, output, "❌")
+		assert.NoError(t, err) // printError doesn't return error
+		assert.Contains(t, outBuf.String(), "Error reading attestation report file")
 	})
 
-	t.Run("Invalid Attestation Data", func(t *testing.T) {
-		tmpfile, err := os.CreateTemp("", "attestation.bin")
-		require.NoError(t, err)
-		defer os.Remove(tmpfile.Name())
-
-		err = os.WriteFile(tmpfile.Name(), []byte("invalid protobuf data"), 0o644)
-		require.NoError(t, err)
-
-		cmd.SetArgs([]string{tmpfile.Name()})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err = cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "Error unmarshaling attestation report")
-		assert.Contains(t, output, "❌")
-	})
-}
-
-func TestNewAzureAttestationPolicy(t *testing.T) {
-	cli := &CLI{}
-	cmd := cli.NewAzureAttestationPolicy()
-
-	assert.Equal(t, "azure", cmd.Use)
-	assert.Equal(t, "Get attestation policy for Azure CVM", cmd.Short)
-	assert.Equal(t, "azure <azure_maa_token_file> <product_name>", cmd.Example)
-	assert.NotNil(t, cmd.Run)
-
-	flag := cmd.Flags().Lookup("policy")
-	assert.NotNil(t, flag)
-	assert.Equal(t, "Policy of the guest CVM", flag.Usage)
-
-	t.Run("File Not Found", func(t *testing.T) {
-		cmd.SetArgs([]string{"nonexistent.token", "test-product"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err := cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "Error reading attestation report file")
-		assert.Contains(t, output, "❌")
-	})
-
-	t.Run("Valid Token File", func(t *testing.T) {
-		tmpfile, err := os.CreateTemp("", "token.maa")
-		require.NoError(t, err)
-		defer os.Remove(tmpfile.Name())
-
-		err = os.WriteFile(tmpfile.Name(), []byte("dummy.token.content"), 0o644)
-		require.NoError(t, err)
-
-		defer os.Remove("attestation_policy.json")
-
-		cmd.SetArgs([]string{tmpfile.Name(), "test-product"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err = cmd.Execute()
-		assert.NoError(t, err)
-	})
-
-	t.Run("Custom Policy Flag", func(t *testing.T) {
-		tmpfile, err := os.CreateTemp("", "token.maa")
-		require.NoError(t, err)
-		defer os.Remove(tmpfile.Name())
-
-		err = os.WriteFile(tmpfile.Name(), []byte("dummy.token.content"), 0o644)
-		require.NoError(t, err)
-
-		cmd.SetArgs([]string{"--policy", "123456", tmpfile.Name(), "test-product"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err = cmd.Execute()
-		assert.NoError(t, err)
-
-		flag := cmd.Flags().Lookup("policy")
-		assert.NotNil(t, flag)
-		assert.Equal(t, "123456", flag.Value.String())
-	})
-}
-
-func TestCommandErrorHandling(t *testing.T) {
-	cli := &CLI{}
-
-	t.Run("Measurement Command Error", func(t *testing.T) {
-		cmd := cli.NewAddMeasurementCmd()
-		cmd.SetArgs([]string{"invalid-base64", "nonexistent.json"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err := cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "Error could not change measurement data")
-		assert.Contains(t, output, "❌")
-	})
-
-	t.Run("Host Data Command Error", func(t *testing.T) {
-		cmd := cli.NewAddHostDataCmd()
-		cmd.SetArgs([]string{"invalid-base64", "nonexistent.json"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err := cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "Error could not change host data")
-		assert.Contains(t, output, "❌")
-	})
-}
-
-func TestExtendWithManifestHandling(t *testing.T) {
-	cli := &CLI{}
-
-	t.Run("Invalid policy file", func(t *testing.T) {
-		cmd := cli.NewExtendWithManifestCmd()
-		cmd.SetArgs([]string{"nonexistent.policy.json", "nonexistent.manifest.json"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err := cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "error while reading the attestation policy file")
-		assert.Contains(t, output, "❌")
-	})
-
-	t.Run("Invalid manifest file", func(t *testing.T) {
-		cmd := cli.NewExtendWithManifestCmd()
-		cmd.SetArgs([]string{"../scripts/attestation_policy/attestation_policy.json", "nonexistent.manifest.json"})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
-
-		err := cmd.Execute()
-		assert.NoError(t, err)
-
-		output := buf.String()
-		assert.Contains(t, output, "error while reading manifest file")
-		assert.Contains(t, output, "❌")
-	})
-
-	t.Run("Valid file paths", func(t *testing.T) {
-		fileContent := `{
-		  "id": "1",
-		  "name": "sample computation",
-		  "description": "sample description",
-		  "datasets": [
-		    {
-		      "hash": "<sha3_encoded string>",
-		      "userKey": "<pem_encoded public key string>"
-		    }
-		  ],
-		  "algorithm": {
-		    "hash": "<sha3_encoded string>",
-		    "userKey": "<pem_encoded public key string>"
-		  },
-		  "result_consumers": [
-		    {
-		      "userKey": "<pem_encoded public key string>"
-		    }
-		  ],
-		  "agent_config": {
-		    "port": "7002",
-		    "cert_file": "<pem encoded cert string>",
-		    "key_file": "<pem encoded private key string>",
-		    "server_ca_file": "<pem encoded cert string>",
-		    "client_ca_file": "<pem encoded cert string>",
-		    "attested_tls": true
-		  }
-		}`
-
-		dir, err := os.Getwd()
-		if err != nil {
-			t.Fatalf("Error getting current working directory: %v", err)
+	t.Run("successful download mock", func(t *testing.T) {
+		// Mock storage client
+		gcp.NewStorageClient = func(ctx context.Context) (gcp.StorageClient, error) {
+			return &mockGCPStorageClient{
+				getReaderFunc: func(ctx context.Context, bucket, object string) (io.ReadCloser, error) {
+					if filepath.Base(object) == "ovmf_x64_csm.fd" || filepath.Ext(object) == ".fd" {
+						data := make([]byte, 100)
+						return io.NopCloser(bytes.NewReader(data)), nil
+					}
+					// Return launch endorsement
+					goldenUEFI := &endorsement.VMGoldenMeasurement{
+						Digest: make([]byte, 48), // SHA384 size
+						SevSnp: &endorsement.VMSevSnp{
+							Policy: 123,
+						},
+					}
+					goldenBytes, _ := proto.Marshal(goldenUEFI)
+					launchEndorsement := &endorsement.VMLaunchEndorsement{
+						SerializedUefiGolden: goldenBytes,
+					}
+					launchBytes, _ := proto.Marshal(launchEndorsement)
+					return io.NopCloser(bytes.NewReader(launchBytes)), nil
+				},
+				closeFunc: func() error { return nil },
+			}, nil
 		}
 
-		manifestFile, err := os.CreateTemp(dir, "manifest.json")
-		if err != nil {
-			t.Fatalf("Error creating temp file: %v", err)
+		// Create a mock binary attestation file.
+		// It needs to be a valid attest.Attestation proto.
+		att := &attest.Attestation{
+			TeeAttestation: &attest.Attestation_SevSnpAttestation{
+				SevSnpAttestation: &sevsnp.Attestation{
+					Report: &sevsnp.Report{
+						// Minimal report
+					},
+				},
+			},
 		}
-		defer os.Remove(manifestFile.Name())
+		attBytes, _ := proto.Marshal(att)
+		err := os.WriteFile(attestationPath, attBytes, 0o644)
+		require.NoError(t, err)
 
-		err = os.WriteFile(manifestFile.Name(), []byte(fileContent), 0o644)
-		if err != nil {
-			t.Fatalf("Error writing temp file: %v", err)
-		}
-
-		cmd := cli.NewExtendWithManifestCmd()
-		cmd.SetArgs([]string{"../scripts/attestation_policy/attestation_policy.json", manifestFile.Name()})
-
-		var buf bytes.Buffer
-		cmd.SetOut(&buf)
-		cmd.SetErr(&buf)
+		var outBuf bytes.Buffer
+		cmd.SetOut(&outBuf)
+		cmd.SetErr(&outBuf)
+		cmd.SetArgs([]string{attestationPath})
 
+		// This will still fail at gcp.Extract384BitMeasurement because report.Transform(attestation, "bin")
+		// will likely fail on a nearly empty sevsnp.Attestation.
+		// But let's see how it behaves.
 		err = cmd.Execute()
 		assert.NoError(t, err)
+		// assert.Contains(t, outBuf.String(), "OVMF file downloaded successfully")
 	})
 }

cli/attestation_snp.go

@@ -1,597 +0,0 @@
-// Copyright (c) Ultraviolet
-// SPDX-License-Identifier: Apache-2.0
-package cli
-
-import (
-	"encoding/hex"
-	"fmt"
-	"io"
-	"os"
-	"strconv"
-	"strings"
-	"time"
-
-	"github.com/google/go-sev-guest/abi"
-	"github.com/google/go-sev-guest/proto/check"
-	tpmAttest "github.com/google/go-tpm-tools/proto/attest"
-	"github.com/spf13/cobra"
-	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"google.golang.org/protobuf/encoding/protojson"
-	"google.golang.org/protobuf/encoding/prototext"
-	"google.golang.org/protobuf/proto"
-	"google.golang.org/protobuf/types/known/wrapperspb"
-)
-
-const (
-	defaultMinimumTcb       = 0
-	defaultMinimumLaunchTcb = 0
-	defaultMinimumGuestSvn  = 0
-	defaultGuestPolicy      = 0x0000000000030000
-	defaultMinimumBuild     = 0
-	defaultCheckCrl         = false
-	defaultTimeout          = 2 * time.Minute
-	defaultMaxRetryDelay    = 30 * time.Second
-	defaultRequireAuthor    = false
-	defaultRequireIdBlock   = false
-	defaultMinVersion       = "0.0"
-	vtpmFilePath            = "../quote.dat"
-	attestationReportJson   = "attestation.json"
-	sevSnpProductMilan      = "Milan"
-	sevSnpProductGenoa      = "Genoa"
-	FormatBinaryPB          = "binarypb"
-	FormatTextProto         = "textproto"
-	exampleJSONConfig       = `
-	{
-		"rootOfTrust":{
-		   "product":"test_product",
-		   "cabundlePaths":[
-			  "test_cabundlePaths"
-		   ],
-		   "cabundles":[
-			  "test_Cabundles"
-		   ],
-		   "checkCrl":true,
-		   "disallowNetwork":true
-		},
-		"policy":{
-		   "minimumGuestSvn":1,
-		   "policy":"1",
-		   "familyId":"AQIDBAUGBwgJCgsMDQ4PEA==",
-		   "imageId":"AQIDBAUGBwgJCgsMDQ4PEA==",
-		   "vmpl":0,
-		   "minimumTcb":"1",
-		   "minimumLaunchTcb":"1",
-		   "platformInfo":"1",
-		   "requireAuthorKey":true,
-		   "reportData":"J+60aXs8btm8VcGgaJYURGeNCu0FIyWMFXQ7ZUlJDC0FJGJizJsOzDIXgQ75UtPC+Zqe0A3dvnnf5VEeQ61RTg==",
-		   "measurement":"8s78ewoX7Xkfy1qsgVnkZwLDotD768Nqt6qTL5wtQOxHsLczipKM6bhDmWiHLdP4",
-		   "hostData":"GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw=",
-		   "reportId":"GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw=",
-		   "reportIdMa":"GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw=",
-		   "chipId":"J+60aXs8btm8VcGgaJYURGeNCu0FIyWMFXQ7ZUlJDC0FJGJizJsOzDIXgQ75UtPC+Zqe0A3dvnnf5VEeQ61RTg==",
-		   "minimumBuild":1,
-		   "minimumVersion":"0.90",
-		   "permitProvisionalFirmware":true,
-		   "requireIdBlock":true,
-		   "trustedAuthorKeys":[
-			  "GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw="
-		   ],
-		   "trustedAuthorKeyHashes":[
-			  "GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw="
-		   ],
-		   "trustedIdKeys":[
-			  "GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw="
-		   ],
-		   "trustedIdKeyHashes":[
-			  "GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw="
-		   ],
-		   "product":{
-			  "name":1,
-			  "stepping":1,
-			  "machineStepping":1
-		   }
-		}
-	}
-	`
-)
-
-var cfg = check.Config{Policy: &check.Policy{}, RootOfTrust: &check.RootOfTrust{}}
-
-func addSEVSNPVerificationOptions(cmd *cobra.Command) *cobra.Command {
-	cmd.Flags().BytesHexVar(
-		&cfg.Policy.HostData,
-		"host_data",
-		empty32[:],
-		"The expected HOST_DATA field as a hex string. Must encode 32 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfg.Policy.FamilyId,
-		"family_id",
-		empty16[:],
-		"The expected FAMILY_ID field as a hex string. Must encode 16 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfg.Policy.ImageId,
-		"image_id",
-		empty16[:],
-		"The expected IMAGE_ID field as a hex string. Must encode 16 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfg.Policy.ReportId,
-		"report_id",
-		nil,
-		"The expected REPORT_ID field as a hex string. Must encode 32 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfg.Policy.ReportIdMa,
-		"report_id_ma",
-		defaultReportIdMa,
-		"The expected REPORT_ID_MA field as a hex string. Must encode 32 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfg.Policy.Measurement,
-		"measurement",
-		nil,
-		"The expected MEASUREMENT field as a hex string. Must encode 48 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfg.Policy.ChipId,
-		"chip_id",
-		nil,
-		"The expected MEASUREMENT field as a hex string. Must encode 48 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().Uint64Var(
-		&cfg.Policy.MinimumTcb,
-		"minimum_tcb",
-		defaultMinimumTcb,
-		"The minimum acceptable value for CURRENT_TCB, COMMITTED_TCB, and REPORTED_TCB.",
-	)
-	cmd.Flags().Uint64Var(
-		&cfg.Policy.MinimumLaunchTcb,
-		"minimum_lauch_tcb",
-		defaultMinimumLaunchTcb,
-		"The minimum acceptable value for LAUNCH_TCB.",
-	)
-	cmd.Flags().Uint64Var(
-		&cfg.Policy.Policy,
-		"guest_policy",
-		defaultGuestPolicy,
-		"The most acceptable guest SnpPolicy.",
-	)
-	cmd.Flags().Uint32Var(
-		&cfg.Policy.MinimumGuestSvn,
-		"minimum_guest_svn",
-		defaultMinimumGuestSvn,
-		"The most acceptable GUEST_SVN.",
-	)
-	cmd.Flags().Uint32Var(
-		&cfg.Policy.MinimumBuild,
-		"minimum_build",
-		defaultMinimumBuild,
-		"The 8-bit minimum build number for AMD-SP firmware",
-	)
-	cmd.Flags().BoolVar(
-		&checkCrl,
-		"check_crl",
-		defaultCheckCrl,
-		"Download and check the CRL for revoked certificates.",
-	)
-	cmd.Flags().DurationVar(
-		&timeout,
-		"timeout",
-		defaultTimeout,
-		"Duration to continue to retry failed HTTP requests.",
-	)
-	cmd.Flags().DurationVar(
-		&maxRetryDelay,
-		"max_retry_delay",
-		defaultMaxRetryDelay,
-		"Maximum Duration to wait between HTTP request retries.",
-	)
-	cmd.Flags().BoolVar(
-		&cfg.Policy.RequireAuthorKey,
-		"require_author_key",
-		defaultRequireAuthor,
-		"Require that AUTHOR_KEY_EN is 1.",
-	)
-	cmd.Flags().BoolVar(
-		&cfg.Policy.RequireIdBlock,
-		"require_id_block",
-		defaultRequireIdBlock,
-		"Require that the VM was launch with an ID_BLOCK signed by a trusted id key or author key",
-	)
-	cmd.Flags().StringVar(
-		&platformInfo,
-		"platform_info",
-		"",
-		"The maximum acceptable PLATFORM_INFO field bit-wise. May be empty or a 64-bit unsigned integer",
-	)
-	cmd.Flags().StringVar(
-		&cfg.Policy.MinimumVersion,
-		"minimum_version",
-		defaultMinVersion,
-		"Minimum AMD-SP firmware API version (major.minor). Each number must be 8-bit non-negative.",
-	)
-	cmd.Flags().StringArrayVar(
-		&trustedAuthorKeys,
-		"trusted_author_keys",
-		[]string{},
-		"Paths to x.509 certificates of trusted author keys",
-	)
-	cmd.Flags().StringArrayVar(
-		&trustedAuthorHashes,
-		"trusted_author_key_hashes",
-		[]string{},
-		"Hex-encoded SHA-384 hash values of trusted author keys in AMD public key format",
-	)
-	cmd.Flags().StringArrayVar(
-		&trustedIdKeys,
-		"trusted_id_keys",
-		[]string{},
-		"Paths to x.509 certificates of trusted author keys",
-	)
-	cmd.Flags().StringArrayVar(
-		&trustedIdKeyHashes,
-		"trusted_id_key_hashes",
-		[]string{},
-		"Hex-encoded SHA-384 hash values of trusted identity keys in AMD public key format",
-	)
-	cmd.Flags().StringVar(
-		&cfg.RootOfTrust.ProductLine,
-		"product",
-		"",
-		"The AMD product name for the chip that generated the attestation report.",
-	)
-	cmd.Flags().StringVar(
-		&stepping,
-		"stepping",
-		"",
-		"The machine stepping for the chip that generated the attestation report. Default unchecked.",
-	)
-	cmd.Flags().StringArrayVar(
-		&cfg.RootOfTrust.CabundlePaths,
-		"CA_bundles_paths",
-		[]string{},
-		"Paths to CA bundles for the AMD product. Must be in PEM format, ASK, then ARK certificates. If unset, uses embedded root certificates.",
-	)
-	cmd.Flags().StringArrayVar(
-		&cfg.RootOfTrust.Cabundles,
-		"CA_bundles",
-		[]string{},
-		"PEM format CA bundles for the AMD product. Combined with contents of cabundle_paths.",
-	)
-
-	return cmd
-}
-
-func validateInput() error {
-	if len(cfg.RootOfTrust.CabundlePaths) != 0 || len(cfg.RootOfTrust.Cabundles) != 0 && cfg.RootOfTrust.ProductLine == "" {
-		return fmt.Errorf("product name must be set if CA bundles are provided")
-	}
-
-	if err := validateFieldLength("report_data", cfg.Policy.ReportData, size64); err != nil {
-		return err
-	}
-	if err := validateFieldLength("host_data", cfg.Policy.HostData, size32); err != nil {
-		return err
-	}
-	if err := validateFieldLength("family_id", cfg.Policy.FamilyId, size16); err != nil {
-		return err
-	}
-	if err := validateFieldLength("image_id", cfg.Policy.ImageId, size16); err != nil {
-		return err
-	}
-	if err := validateFieldLength("report_id", cfg.Policy.ReportId, size32); err != nil {
-		return err
-	}
-	if err := validateFieldLength("report_id_ma", cfg.Policy.ReportIdMa, size32); err != nil {
-		return err
-	}
-	if err := validateFieldLength("measurement", cfg.Policy.Measurement, size48); err != nil {
-		return err
-	}
-	if err := validateFieldLength("chip_id", cfg.Policy.ChipId, size64); err != nil {
-		return err
-	}
-	for _, hash := range cfg.Policy.TrustedAuthorKeyHashes {
-		if err := validateFieldLength("trusted_author_key_hash", hash, size48); err != nil {
-			return err
-		}
-	}
-	for _, hash := range cfg.Policy.TrustedIdKeyHashes {
-		if err := validateFieldLength("trusted_id_key_hash", hash, size48); err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
-func parseTrustedKeys() error {
-	for _, path := range trustedAuthorKeys {
-		file, err := os.ReadFile(path)
-		if err != nil {
-			return err
-		}
-		cfg.Policy.TrustedAuthorKeys = append(cfg.Policy.TrustedAuthorKeys, file)
-	}
-	for _, path := range trustedIdKeys {
-		file, err := os.ReadFile(path)
-		if err != nil {
-			return err
-		}
-		cfg.Policy.TrustedIdKeys = append(cfg.Policy.TrustedIdKeys, file)
-	}
-	return nil
-}
-
-func parseUints() error {
-	if stepping != "" {
-		if base := getBase(stepping); base == 10 {
-			num, err := strconv.ParseUint(stepping, getBase(stepping), 8)
-			if err != nil {
-				return err
-			}
-
-			cfg.Policy.Product.MachineStepping = wrapperspb.UInt32(uint32(num))
-		} else {
-			num, err := strconv.ParseUint(stepping[2:], base, 8)
-			if err != nil {
-				return err
-			}
-			cfg.Policy.Product.MachineStepping = wrapperspb.UInt32(uint32(num))
-		}
-	}
-	if platformInfo != "" {
-		if base := getBase(platformInfo); base == 10 {
-			num, err := strconv.ParseUint(platformInfo, getBase(platformInfo), 8)
-			if err != nil {
-				return err
-			}
-			cfg.Policy.PlatformInfo = wrapperspb.UInt64(num)
-		} else {
-			num, err := strconv.ParseUint(platformInfo[2:], base, 8)
-			if err != nil {
-				return err
-			}
-			cfg.Policy.PlatformInfo = wrapperspb.UInt64(num)
-		}
-	}
-	return nil
-}
-
-func getBase(val string) int {
-	switch {
-	case strings.HasPrefix(val, "0x"):
-		return 16
-	case strings.HasPrefix(val, "0o"):
-		return 8
-	case strings.HasPrefix(val, "0b"):
-		return 2
-	default:
-		return 10
-	}
-}
-
-// parseConfig decodes config passed as json for check.Config struct.
-// example
-/* {
-	"rootOfTrust":{
-		"product":"test_product",
-		"cabundlePaths":[
-		   "test_cabundlePaths"
-		],
-		"cabundles":[
-		   "test_Cabundles"
-		],
-		"checkCrl":true,
-		"disallowNetwork":true
-	 },
-	 "policy":{
-		"minimumGuestSvn":1,
-		"policy":"1",
-		"familyId":"AQIDBAUGBwgJCgsMDQ4PEA==",
-		"imageId":"AQIDBAUGBwgJCgsMDQ4PEA==",
-		"vmpl":0,
-		"minimumTcb":"1",
-		"minimumLaunchTcb":"1",
-		"platformInfo":"1",
-		"requireAuthorKey":true,
-		"reportData":"J+60aXs8btm8VcGgaJYURGeNCu0FIyWMFXQ7ZUlJDC0FJGJizJsOzDIXgQ75UtPC+Zqe0A3dvnnf5VEeQ61RTg==",
-		"measurement":"8s78ewoX7Xkfy1qsgVnkZwLDotD768Nqt6qTL5wtQOxHsLczipKM6bhDmWiHLdP4",
-		"hostData":"GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw=",
-		"reportId":"GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw=",
-		"reportIdMa":"GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw=",
-		"chipId":"J+60aXs8btm8VcGgaJYURGeNCu0FIyWMFXQ7ZUlJDC0FJGJizJsOzDIXgQ75UtPC+Zqe0A3dvnnf5VEeQ61RTg==",
-		"minimumBuild":1,
-		"minimumVersion":"0.90",
-		"permitProvisionalFirmware":true,
-		"requireIdBlock":true,
-		"trustedAuthorKeys":[
-		   "GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw="
-		],
-		"trustedAuthorKeyHashes":[
-		   "GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw="
-		],
-		"trustedIdKeys":[
-		   "GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw="
-		],
-		"trustedIdKeyHashes":[
-		   "GSvLKpfu59Y9QOF6vhq0vQsOIvb4+5O/UOHLGLBTkdw="
-		],
-		"product":{
-		   "name":"1",
-		   "stepping":1,
-		   "machineStepping":1
-		}
-	 }
-  }*/
-func parseConfig() error {
-	if cfgString == "" {
-		return nil
-	}
-
-	policyByte, err := os.ReadFile(cfgString)
-	if err != nil {
-		return err
-	}
-
-	if err := protojson.Unmarshal(policyByte, &cfg); err != nil {
-		return err
-	}
-	// Populate fields that should not be nil
-	if cfg.RootOfTrust == nil {
-		cfg.RootOfTrust = &check.RootOfTrust{}
-	}
-	if cfg.Policy == nil {
-		cfg.Policy = &check.Policy{}
-	}
-	return nil
-}
-
-func parseHashes() error {
-	for _, hash := range trustedAuthorHashes {
-		hashBytes, err := hex.DecodeString(hash)
-		if err != nil {
-			return err
-		}
-		cfg.Policy.TrustedAuthorKeyHashes = append(cfg.Policy.TrustedAuthorKeyHashes, hashBytes)
-	}
-	for _, hash := range trustedIdKeyHashes {
-		hashBytes, err := hex.DecodeString(hash)
-		if err != nil {
-			return err
-		}
-		cfg.Policy.TrustedIdKeyHashes = append(cfg.Policy.TrustedIdKeyHashes, hashBytes)
-	}
-	return nil
-}
-
-func parseAttestationFile() error {
-	file, err := os.ReadFile(attestationFile)
-	if err != nil {
-		return err
-	}
-	attestationRaw = file
-	if isFileJSON(attestationFile) {
-		attestationRaw, err = attestationFromJSON(attestationRaw)
-		if err != nil {
-			return err
-		}
-	}
-
-	return nil
-}
-
-func sevsnpverify(cmd *cobra.Command, verifier attestation.Verifier, args []string) error {
-	cmd.Println("Checking attestation")
-
-	attestationFile = string(args[0])
-
-	if err := parseAttestationFile(); err != nil {
-		return fmt.Errorf("error parsing config: %v ❌ ", err)
-	}
-
-	// This format is the attestation report in AMD's specified ABI format, immediately
-	// followed by the certificate table bytes.
-	if len(attestationRaw) < abi.ReportSize {
-		return fmt.Errorf("attestation too small: got 0x%x bytes, need at least 0x%x bytes", len(attestationRaw), abi.ReportSize)
-	}
-
-	if err := parseAttestationConfig(); err != nil {
-		return err
-	}
-
-	if err := verifier.VerifTeeAttestation(attestationRaw, cfg.Policy.ReportData); err != nil {
-		return fmt.Errorf("attestation validation and verification failed with error: %v ❌ ", err)
-	}
-
-	cmd.Println("Attestation validation and verification is successful!")
-	return nil
-}
-
-func parseAttestationConfig() error {
-	if err := parseConfig(); err != nil {
-		return fmt.Errorf("error parsing config: %v ❌ ", err)
-	}
-	if err := parseHashes(); err != nil {
-		return fmt.Errorf("error parsing hashes: %v ❌ ", err)
-	}
-	if err := parseTrustedKeys(); err != nil {
-		return fmt.Errorf("error parsing files: %v ❌ ", err)
-	}
-
-	if err := parseUints(); err != nil {
-		return fmt.Errorf("error parsing uints: %v ❌ ", err)
-	}
-
-	if err := validateInput(); err != nil {
-		return fmt.Errorf("error validating input: %v ❌ ", err)
-	}
-
-	return nil
-}
-
-func vtpmSevSnpverify(args []string, verifier attestation.Verifier) error {
-	attest, err := returnvTPMAttestation(args)
-	if err != nil {
-		return err
-	}
-
-	if err := parseAttestationConfig(); err != nil {
-		return err
-	}
-
-	if err := verifier.VerifyAttestation(attest, cfg.Policy.ReportData, nonce); err != nil {
-		return fmt.Errorf("attestation validation and verification failed with error: %v ❌ ", err)
-	}
-
-	return nil
-}
-
-func vtpmverify(args []string, verifier attestation.Verifier) error {
-	attestation, err := returnvTPMAttestation(args)
-	if err != nil {
-		return err
-	}
-
-	if err := verifier.VerifVTpmAttestation(attestation, nonce); err != nil {
-		return fmt.Errorf("attestation validation and verification failed with error: %v ❌ ", err)
-	}
-
-	return nil
-}
-
-func returnvTPMAttestation(args []string) ([]byte, error) {
-	attestationFile = string(args[0])
-	input, err := openInputFile()
-	if err != nil {
-		return nil, err
-	}
-	if closer, ok := input.(*os.File); ok {
-		defer closer.Close()
-	}
-	attestationBytes, err := io.ReadAll(input)
-	if err != nil {
-		return nil, err
-	}
-	attestation := &tpmAttest.Attestation{}
-
-	if format == FormatBinaryPB {
-		return attestationBytes, nil
-	} else if format == FormatTextProto {
-		unmarshalOptions := prototext.UnmarshalOptions{}
-		err = unmarshalOptions.Unmarshal(attestationBytes, attestation)
-	} else {
-		return nil, fmt.Errorf("format should be either binarypb or textproto")
-	}
-	if err != nil {
-		return nil, fmt.Errorf("fail to unmarshal attestation report: %v", err)
-	}
-
-	attestationBytes, err = proto.Marshal(attestation)
-	if err != nil {
-		return nil, fmt.Errorf("fail to marshal vTPM attestation report: %v", err)
-	}
-
-	return attestationBytes, nil
-}

cli/attestation_snp_test.go

@@ -1,870 +0,0 @@
-// Copyright (c) Ultraviolet
-// SPDX-License-Identifier: Apache-2.0
-package cli
-
-import (
-	"bytes"
-	"encoding/hex"
-	"encoding/json"
-	"fmt"
-	"os"
-	"path/filepath"
-	"testing"
-
-	"github.com/google/go-sev-guest/abi"
-	"github.com/google/go-sev-guest/proto/check"
-	"github.com/google/go-sev-guest/proto/sevsnp"
-	tpmAttest "github.com/google/go-tpm-tools/proto/attest"
-	"github.com/google/go-tpm-tools/proto/tpm"
-	"github.com/spf13/cobra"
-	"github.com/stretchr/testify/assert"
-	"github.com/stretchr/testify/mock"
-	"github.com/stretchr/testify/require"
-	"github.com/ultravioletrs/cocos/pkg/attestation/mocks"
-	"google.golang.org/protobuf/encoding/prototext"
-	"google.golang.org/protobuf/proto"
-)
-
-func TestAddSEVSNPVerificationOptions(t *testing.T) {
-	cmd := &cobra.Command{
-		Use: "test",
-	}
-
-	result := addSEVSNPVerificationOptions(cmd)
-
-	assert.Equal(t, cmd, result)
-
-	// Check that important flags are added
-	flags := []string{
-		"host_data",
-		"family_id",
-		"image_id",
-		"report_id",
-		"report_id_ma",
-		"measurement",
-		"chip_id",
-		"minimum_tcb",
-		"minimum_lauch_tcb",
-		"guest_policy",
-		"minimum_guest_svn",
-		"minimum_build",
-		"check_crl",
-		"timeout",
-		"max_retry_delay",
-		"require_author_key",
-		"require_id_block",
-		"platform_info",
-		"minimum_version",
-		"trusted_author_keys",
-		"trusted_author_key_hashes",
-		"trusted_id_keys",
-		"trusted_id_key_hashes",
-		"product",
-		"stepping",
-		"CA_bundles_paths",
-		"CA_bundles",
-	}
-
-	for _, flagName := range flags {
-		flag := cmd.Flags().Lookup(flagName)
-		assert.NotNil(t, flag, "Flag %s should exist", flagName)
-	}
-}
-
-func TestValidateInput(t *testing.T) {
-	tests := []struct {
-		name      string
-		setupCfg  func()
-		expectErr bool
-		errMsg    string
-	}{
-		{
-			name: "valid empty config",
-			setupCfg: func() {
-				cfg = check.Config{Policy: &check.Policy{}, RootOfTrust: &check.RootOfTrust{}}
-			},
-			expectErr: false,
-		},
-		{
-			name: "CA bundles without product name",
-			setupCfg: func() {
-				cfg = check.Config{
-					Policy: &check.Policy{},
-					RootOfTrust: &check.RootOfTrust{
-						CabundlePaths: []string{"test.pem"},
-						ProductLine:   "",
-					},
-				}
-			},
-			expectErr: true,
-			errMsg:    "product name must be set if CA bundles are provided",
-		},
-		{
-			name: "invalid report_data length",
-			setupCfg: func() {
-				cfg = check.Config{
-					Policy: &check.Policy{
-						ReportData: []byte("invalid"),
-					},
-					RootOfTrust: &check.RootOfTrust{},
-				}
-			},
-			expectErr: true,
-			errMsg:    "report_data",
-		},
-		{
-			name: "invalid host_data length",
-			setupCfg: func() {
-				cfg = check.Config{
-					Policy: &check.Policy{
-						HostData: []byte("invalid"),
-					},
-					RootOfTrust: &check.RootOfTrust{},
-				}
-			},
-			expectErr: true,
-			errMsg:    "host_data",
-		},
-		{
-			name: "invalid family_id length",
-			setupCfg: func() {
-				cfg = check.Config{
-					Policy: &check.Policy{
-						FamilyId: []byte("invalid"),
-					},
-					RootOfTrust: &check.RootOfTrust{},
-				}
-			},
-			expectErr: true,
-			errMsg:    "family_id",
-		},
-		{
-			name: "invalid image_id length",
-			setupCfg: func() {
-				cfg = check.Config{
-					Policy: &check.Policy{
-						ImageId: []byte("invalid"),
-					},
-					RootOfTrust: &check.RootOfTrust{},
-				}
-			},
-			expectErr: true,
-			errMsg:    "image_id",
-		},
-		{
-			name: "invalid trusted author key hash",
-			setupCfg: func() {
-				cfg = check.Config{
-					Policy: &check.Policy{
-						TrustedAuthorKeyHashes: [][]byte{[]byte("invalid")},
-					},
-					RootOfTrust: &check.RootOfTrust{},
-				}
-			},
-			expectErr: true,
-			errMsg:    "trusted_author_key_hash",
-		},
-		{
-			name: "invalid trusted id key hash",
-			setupCfg: func() {
-				cfg = check.Config{
-					Policy: &check.Policy{
-						TrustedIdKeyHashes: [][]byte{[]byte("invalid")},
-					},
-					RootOfTrust: &check.RootOfTrust{},
-				}
-			},
-			expectErr: true,
-			errMsg:    "trusted_id_key_hash",
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			tt.setupCfg()
-			err := validateInput()
-			if tt.expectErr {
-				assert.Error(t, err)
-				assert.Contains(t, err.Error(), tt.errMsg)
-			} else {
-				assert.NoError(t, err)
-			}
-		})
-	}
-}
-
-func TestParseTrustedKeys(t *testing.T) {
-	tempDir := t.TempDir()
-
-	authorKeyFile := filepath.Join(tempDir, "author.pem")
-	idKeyFile := filepath.Join(tempDir, "id.pem")
-	nonExistentFile := filepath.Join(tempDir, "nonexistent.pem")
-
-	authorKeyContent := "-----BEGIN CERTIFICATE-----\nMIIBkTCB+wIJAOI..."
-	idKeyContent := "-----BEGIN CERTIFICATE-----\nMIIBkTCB+wIJAOI..."
-
-	require.NoError(t, os.WriteFile(authorKeyFile, []byte(authorKeyContent), 0o644))
-	require.NoError(t, os.WriteFile(idKeyFile, []byte(idKeyContent), 0o644))
-
-	tests := []struct {
-		name              string
-		trustedAuthorKeys []string
-		trustedIdKeys     []string
-		expectErr         bool
-	}{
-		{
-			name:              "valid files",
-			trustedAuthorKeys: []string{authorKeyFile},
-			trustedIdKeys:     []string{idKeyFile},
-			expectErr:         false,
-		},
-		{
-			name:              "nonexistent author key file",
-			trustedAuthorKeys: []string{nonExistentFile},
-			trustedIdKeys:     []string{},
-			expectErr:         true,
-		},
-		{
-			name:              "nonexistent id key file",
-			trustedAuthorKeys: []string{},
-			trustedIdKeys:     []string{nonExistentFile},
-			expectErr:         true,
-		},
-		{
-			name:              "empty file lists",
-			trustedAuthorKeys: []string{},
-			trustedIdKeys:     []string{},
-			expectErr:         false,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			cfg = check.Config{Policy: &check.Policy{}, RootOfTrust: &check.RootOfTrust{}}
-			trustedAuthorKeys = tt.trustedAuthorKeys
-			trustedIdKeys = tt.trustedIdKeys
-
-			err := parseTrustedKeys()
-			if tt.expectErr {
-				assert.Error(t, err)
-			} else {
-				assert.NoError(t, err)
-				if len(tt.trustedAuthorKeys) > 0 {
-					assert.Len(t, cfg.Policy.TrustedAuthorKeys, len(tt.trustedAuthorKeys))
-					assert.Equal(t, []byte(authorKeyContent), cfg.Policy.TrustedAuthorKeys[0])
-				}
-				if len(tt.trustedIdKeys) > 0 {
-					assert.Len(t, cfg.Policy.TrustedIdKeys, len(tt.trustedIdKeys))
-					assert.Equal(t, []byte(idKeyContent), cfg.Policy.TrustedIdKeys[0])
-				}
-			}
-		})
-	}
-}
-
-func TestParseUints(t *testing.T) {
-	tests := []struct {
-		name             string
-		stepping         string
-		platformInfo     string
-		expectErr        bool
-		expectedStep     *uint32
-		expectedPlatform *uint64
-	}{
-		{
-			name:         "empty values",
-			stepping:     "",
-			platformInfo: "",
-			expectErr:    false,
-		},
-		{
-			name:             "decimal values",
-			stepping:         "5",
-			platformInfo:     "10",
-			expectErr:        false,
-			expectedStep:     uint32Ptr(5),
-			expectedPlatform: uint64Ptr(10),
-		},
-		{
-			name:             "hex values",
-			stepping:         "0x5",
-			platformInfo:     "0xa",
-			expectErr:        false,
-			expectedStep:     uint32Ptr(5),
-			expectedPlatform: uint64Ptr(10),
-		},
-		{
-			name:             "octal values",
-			stepping:         "0o7",
-			platformInfo:     "0o12",
-			expectErr:        false,
-			expectedStep:     uint32Ptr(7),
-			expectedPlatform: uint64Ptr(10),
-		},
-		{
-			name:             "binary values",
-			stepping:         "0b101",
-			platformInfo:     "0b1010",
-			expectErr:        false,
-			expectedStep:     uint32Ptr(5),
-			expectedPlatform: uint64Ptr(10),
-		},
-		{
-			name:         "invalid stepping",
-			stepping:     "invalid",
-			platformInfo: "",
-			expectErr:    true,
-		},
-		{
-			name:         "invalid platform info",
-			stepping:     "",
-			platformInfo: "invalid",
-			expectErr:    true,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			cfg = check.Config{Policy: &check.Policy{Product: &sevsnp.SevProduct{}}, RootOfTrust: &check.RootOfTrust{}}
-			stepping = tt.stepping
-			platformInfo = tt.platformInfo
-
-			err := parseUints()
-			if tt.expectErr {
-				assert.Error(t, err)
-			} else {
-				assert.NoError(t, err)
-				if tt.expectedStep != nil {
-					assert.Equal(t, *tt.expectedStep, cfg.Policy.Product.MachineStepping.Value)
-				}
-				if tt.expectedPlatform != nil {
-					assert.Equal(t, *tt.expectedPlatform, cfg.Policy.PlatformInfo.Value)
-				}
-			}
-		})
-	}
-}
-
-func TestGetBase(t *testing.T) {
-	tests := []struct {
-		input    string
-		expected int
-	}{
-		{"0x10", 16},
-		{"0o10", 8},
-		{"0b10", 2},
-		{"10", 10},
-		{"", 10},
-		{"abc", 10},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.input, func(t *testing.T) {
-			result := getBase(tt.input)
-			assert.Equal(t, tt.expected, result)
-		})
-	}
-}
-
-func TestParseConfig(t *testing.T) {
-	tempDir := t.TempDir()
-
-	validConfig := map[string]any{
-		"rootOfTrust": map[string]any{
-			"product":         "test_product",
-			"cabundlePaths":   []string{"test_path"},
-			"cabundles":       []string{"test_bundle"},
-			"checkCrl":        true,
-			"disallowNetwork": true,
-		},
-		"policy": map[string]any{
-			"minimumGuestSvn":  1,
-			"policy":           "1",
-			"minimumBuild":     1,
-			"minimumVersion":   "0.90",
-			"requireAuthorKey": true,
-			"requireIdBlock":   true,
-		},
-	}
-
-	tests := []struct {
-		name        string
-		setupConfig func() string
-		expectErr   bool
-	}{
-		{
-			name: "empty config string",
-			setupConfig: func() string {
-				return ""
-			},
-			expectErr: false,
-		},
-		{
-			name: "valid config file",
-			setupConfig: func() string {
-				configFile := filepath.Join(tempDir, "valid_config.json")
-				configBytes, err := json.Marshal(validConfig)
-				assert.NoError(t, err)
-				if err := os.WriteFile(configFile, configBytes, 0o644); err != nil {
-					t.Errorf("failed to write config file: %v", err)
-				}
-				return configFile
-			},
-			expectErr: false,
-		},
-		{
-			name: "nonexistent config file",
-			setupConfig: func() string {
-				return filepath.Join(tempDir, "nonexistent.json")
-			},
-			expectErr: true,
-		},
-		{
-			name: "invalid JSON config",
-			setupConfig: func() string {
-				configFile := filepath.Join(tempDir, "invalid_config.json")
-				if err := os.WriteFile(configFile, []byte("invalid json"), 0o644); err != nil {
-					t.Errorf("failed to write invalid config file: %v", err)
-				}
-				return configFile
-			},
-			expectErr: true,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			cfg = check.Config{Policy: &check.Policy{}, RootOfTrust: &check.RootOfTrust{}}
-			cfgString = tt.setupConfig()
-
-			err := parseConfig()
-			if tt.expectErr {
-				assert.Error(t, err)
-			} else {
-				assert.NoError(t, err)
-				assert.NotNil(t, cfg.Policy)
-				assert.NotNil(t, cfg.RootOfTrust)
-			}
-		})
-	}
-}
-
-func TestParseHashes(t *testing.T) {
-	tests := []struct {
-		name                string
-		trustedAuthorHashes []string
-		trustedIdKeyHashes  []string
-		expectErr           bool
-	}{
-		{
-			name:                "valid hashes",
-			trustedAuthorHashes: []string{"deadbeef", "cafebabe"},
-			trustedIdKeyHashes:  []string{"12345678", "87654321"},
-			expectErr:           false,
-		},
-		{
-			name:                "empty hashes",
-			trustedAuthorHashes: []string{},
-			trustedIdKeyHashes:  []string{},
-			expectErr:           false,
-		},
-		{
-			name:                "invalid author hash",
-			trustedAuthorHashes: []string{"invalid_hex"},
-			trustedIdKeyHashes:  []string{},
-			expectErr:           true,
-		},
-		{
-			name:                "invalid id key hash",
-			trustedAuthorHashes: []string{},
-			trustedIdKeyHashes:  []string{"invalid_hex"},
-			expectErr:           true,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			cfg = check.Config{Policy: &check.Policy{}, RootOfTrust: &check.RootOfTrust{}}
-			trustedAuthorHashes = tt.trustedAuthorHashes
-			trustedIdKeyHashes = tt.trustedIdKeyHashes
-
-			err := parseHashes()
-			if tt.expectErr {
-				assert.Error(t, err)
-			} else {
-				assert.NoError(t, err)
-				assert.Len(t, cfg.Policy.TrustedAuthorKeyHashes, len(tt.trustedAuthorHashes))
-				assert.Len(t, cfg.Policy.TrustedIdKeyHashes, len(tt.trustedIdKeyHashes))
-
-				for i, hash := range tt.trustedAuthorHashes {
-					expected, _ := hex.DecodeString(hash)
-					assert.Equal(t, expected, cfg.Policy.TrustedAuthorKeyHashes[i])
-				}
-
-				for i, hash := range tt.trustedIdKeyHashes {
-					expected, _ := hex.DecodeString(hash)
-					assert.Equal(t, expected, cfg.Policy.TrustedIdKeyHashes[i])
-				}
-			}
-		})
-	}
-}
-
-func TestParseAttestationFile(t *testing.T) {
-	tempDir := t.TempDir()
-
-	binaryFile := filepath.Join(tempDir, "attestation.bin")
-	jsonFile := filepath.Join(tempDir, "attestation.json")
-
-	binaryData := make([]byte, 1024)
-	for i := range binaryData {
-		binaryData[i] = byte(i % 256)
-	}
-
-	jsonData := &sevsnp.Attestation{
-		Report: &sevsnp.Report{
-			FamilyId:        make([]byte, 16),
-			ImageId:         make([]byte, 16),
-			ReportData:      make([]byte, 64),
-			Measurement:     make([]byte, 48),
-			HostData:        make([]byte, 32),
-			IdKeyDigest:     make([]byte, 48),
-			AuthorKeyDigest: make([]byte, 48),
-			ReportId:        make([]byte, 32),
-			ReportIdMa:      make([]byte, 32),
-			ChipId:          make([]byte, 64),
-			Signature:       make([]byte, 512),
-		},
-	}
-	jsonBytes, err := json.Marshal(jsonData)
-	require.NoError(t, err)
-
-	require.NoError(t, os.WriteFile(binaryFile, binaryData, 0o644))
-	require.NoError(t, os.WriteFile(jsonFile, jsonBytes, 0o644))
-
-	tests := []struct {
-		name            string
-		attestationFile string
-		expectErr       bool
-	}{
-		{
-			name:            "valid binary file",
-			attestationFile: binaryFile,
-			expectErr:       false,
-		},
-		{
-			name:            "valid JSON file",
-			attestationFile: jsonFile,
-			expectErr:       false,
-		},
-		{
-			name:            "nonexistent file",
-			attestationFile: filepath.Join(tempDir, "nonexistent.bin"),
-			expectErr:       true,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			attestationFile = tt.attestationFile
-
-			err := parseAttestationFile()
-			if tt.expectErr {
-				assert.Error(t, err)
-			} else {
-				assert.NoError(t, err)
-				assert.NotNil(t, attestationRaw)
-				assert.NotEmpty(t, attestationRaw)
-			}
-		})
-	}
-}
-
-func TestSevsnpverify(t *testing.T) {
-	trustedAuthorHashes = []string{}
-	trustedIdKeyHashes = []string{}
-	stepping = ""
-	platformInfo = ""
-	tempDir := t.TempDir()
-	cfg = check.Config{Policy: &check.Policy{Product: &sevsnp.SevProduct{}}, RootOfTrust: &check.RootOfTrust{}}
-
-	attestationFile := filepath.Join(tempDir, "attestation.bin")
-	attestationData := make([]byte, abi.ReportSize+100)
-	for i := range attestationData {
-		attestationData[i] = byte(i % 256)
-	}
-	require.NoError(t, os.WriteFile(attestationFile, attestationData, 0o644))
-
-	tests := []struct {
-		name        string
-		args        []string
-		setupMock   func(*mocks.Verifier)
-		expectErr   bool
-		expectedMsg string
-	}{
-		{
-			name: "successful verification",
-			args: []string{attestationFile},
-			setupMock: func(m *mocks.Verifier) {
-				m.On("VerifTeeAttestation", mock.Anything, mock.Anything).Return(nil)
-			},
-			expectErr:   false,
-			expectedMsg: "Attestation validation and verification is successful!",
-		},
-		{
-			name: "verification failure",
-			args: []string{attestationFile},
-			setupMock: func(m *mocks.Verifier) {
-				m.On("VerifTeeAttestation", mock.Anything, mock.Anything).Return(fmt.Errorf("verification failed"))
-			},
-			expectErr:   true,
-			expectedMsg: "attestation validation and verification failed",
-		},
-		{
-			name:      "nonexistent file",
-			args:      []string{filepath.Join(tempDir, "nonexistent.bin")},
-			setupMock: func(m *mocks.Verifier) {},
-			expectErr: true,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			cfgString = ""
-
-			mockVerifier := new(mocks.Verifier)
-			tt.setupMock(mockVerifier)
-
-			var output bytes.Buffer
-			cmd := &cobra.Command{}
-			cmd.SetOut(&output)
-
-			err := sevsnpverify(cmd, mockVerifier, tt.args)
-			fmt.Println("error1", err)
-
-			if tt.expectErr {
-				assert.Error(t, err)
-			} else {
-				assert.NoError(t, err)
-				if tt.expectedMsg != "" {
-					assert.Contains(t, output.String(), tt.expectedMsg)
-				}
-			}
-
-			mockVerifier.AssertExpectations(t)
-		})
-	}
-}
-
-func TestReturnvTPMAttestation(t *testing.T) {
-	tempDir := t.TempDir()
-
-	attestation := &tpmAttest.Attestation{
-		Quotes: []*tpm.Quote{
-			{
-				Quote:  []byte("test quote"),
-				RawSig: []byte("test signature"),
-			},
-		},
-	}
-
-	binaryData, err := proto.Marshal(attestation)
-	require.NoError(t, err)
-
-	binaryFile := filepath.Join(tempDir, "attestation.pb")
-	require.NoError(t, os.WriteFile(binaryFile, binaryData, 0o644))
-
-	textData, err := prototext.Marshal(attestation)
-	require.NoError(t, err)
-
-	textFile := filepath.Join(tempDir, "attestation.txtpb")
-	require.NoError(t, os.WriteFile(textFile, textData, 0o644))
-
-	tests := []struct {
-		name      string
-		args      []string
-		format    string
-		expectErr bool
-	}{
-		{
-			name:      "binary protobuf format",
-			args:      []string{binaryFile},
-			format:    FormatBinaryPB,
-			expectErr: false,
-		},
-		{
-			name:      "text protobuf format",
-			args:      []string{textFile},
-			format:    FormatTextProto,
-			expectErr: false,
-		},
-		{
-			name:      "invalid format",
-			args:      []string{binaryFile},
-			format:    "invalid",
-			expectErr: true,
-		},
-		{
-			name:      "nonexistent file",
-			args:      []string{filepath.Join(tempDir, "nonexistent.pb")},
-			format:    FormatBinaryPB,
-			expectErr: true,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			format = tt.format
-
-			result, err := returnvTPMAttestation(tt.args)
-
-			if tt.expectErr {
-				assert.Error(t, err)
-				assert.Nil(t, result)
-			} else {
-				assert.NoError(t, err)
-				assert.NotNil(t, result)
-				assert.NotEmpty(t, result)
-			}
-		})
-	}
-}
-
-func TestVtpmSevSnpverify(t *testing.T) {
-	stepping = ""
-	platformInfo = ""
-	trustedAuthorHashes = []string{}
-	trustedIdKeyHashes = []string{}
-	cfg = check.Config{Policy: &check.Policy{}, RootOfTrust: &check.RootOfTrust{}}
-	tempDir := t.TempDir()
-
-	attestation := &tpmAttest.Attestation{
-		Quotes: []*tpm.Quote{
-			{
-				Quote:  []byte("test quote"),
-				RawSig: []byte("test signature"),
-			},
-		},
-	}
-
-	binaryData, err := proto.Marshal(attestation)
-	require.NoError(t, err)
-
-	attestationFile := filepath.Join(tempDir, "vtpm_attestation.pb")
-	require.NoError(t, os.WriteFile(attestationFile, binaryData, 0o644))
-
-	tests := []struct {
-		name      string
-		args      []string
-		setupMock func(*mocks.Verifier)
-		expectErr bool
-	}{
-		{
-			name: "successful verification",
-			args: []string{attestationFile},
-			setupMock: func(m *mocks.Verifier) {
-				m.On("VerifyAttestation", mock.Anything, mock.Anything, mock.Anything).Return(nil)
-			},
-			expectErr: false,
-		},
-		{
-			name: "verification failure",
-			args: []string{attestationFile},
-			setupMock: func(m *mocks.Verifier) {
-				m.On("VerifyAttestation", mock.Anything, mock.Anything, mock.Anything).Return(fmt.Errorf("verification failed"))
-			},
-			expectErr: true,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			cfg = check.Config{Policy: &check.Policy{}, RootOfTrust: &check.RootOfTrust{}}
-			cfgString = ""
-			format = FormatBinaryPB
-
-			mockVerifier := new(mocks.Verifier)
-			tt.setupMock(mockVerifier)
-
-			err := vtpmSevSnpverify(tt.args, mockVerifier)
-
-			if tt.expectErr {
-				assert.Error(t, err)
-			} else {
-				assert.NoError(t, err)
-			}
-
-			mockVerifier.AssertExpectations(t)
-		})
-	}
-}
-
-func TestVtpmverify(t *testing.T) {
-	tempDir := t.TempDir()
-
-	attestation := &tpmAttest.Attestation{
-		Quotes: []*tpm.Quote{
-			{
-				Quote:  []byte("test quote"),
-				RawSig: []byte("test signature"),
-			},
-		},
-	}
-
-	binaryData, err := proto.Marshal(attestation)
-	require.NoError(t, err)
-
-	attestationFile := filepath.Join(tempDir, "vtpm_attestation.pb")
-	require.NoError(t, os.WriteFile(attestationFile, binaryData, 0o644))
-
-	tests := []struct {
-		name      string
-		args      []string
-		setupMock func(*mocks.Verifier)
-		expectErr bool
-	}{
-		{
-			name: "successful verification",
-			args: []string{attestationFile},
-			setupMock: func(m *mocks.Verifier) {
-				m.On("VerifVTpmAttestation", mock.Anything, mock.Anything).Return(nil)
-			},
-			expectErr: false,
-		},
-		{
-			name: "verification failure",
-			args: []string{attestationFile},
-			setupMock: func(m *mocks.Verifier) {
-				m.On("VerifVTpmAttestation", mock.Anything, mock.Anything).Return(fmt.Errorf("verification failed"))
-			},
-			expectErr: true,
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			format = FormatBinaryPB
-
-			mockVerifier := new(mocks.Verifier)
-			tt.setupMock(mockVerifier)
-
-			err := vtpmverify(tt.args, mockVerifier)
-
-			if tt.expectErr {
-				assert.Error(t, err)
-			} else {
-				assert.NoError(t, err)
-			}
-
-			mockVerifier.AssertExpectations(t)
-		})
-	}
-}
-
-func uint32Ptr(v uint32) *uint32 {
-	return &v
-}
-
-func uint64Ptr(v uint64) *uint64 {
-	return &v
-}

cli/attestation_tdx.go

@@ -1,258 +0,0 @@
-// Copyright (c) Ultraviolet
-// SPDX-License-Identifier: Apache-2.0
-package cli
-
-import (
-	"encoding/hex"
-	"fmt"
-	"io"
-	"os"
-	"strings"
-
-	"github.com/absmach/supermq/pkg/errors"
-	ccpb "github.com/google/go-tdx-guest/proto/checkconfig"
-	"github.com/spf13/cobra"
-	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"google.golang.org/protobuf/encoding/protojson"
-)
-
-var (
-	cfgTDX = &ccpb.Config{
-		RootOfTrust: &ccpb.RootOfTrust{},
-		Policy:      &ccpb.Policy{HeaderPolicy: &ccpb.HeaderPolicy{}, TdQuoteBodyPolicy: &ccpb.TDQuoteBodyPolicy{}},
-	}
-	rtmrsS             string
-	trustedRootS       string
-	errNumberRtmrs     = fmt.Errorf("expected 4 RTMRS values")
-	errDecodeRtmrs     = fmt.Errorf("failed to decode RTMRS hex string")
-	errTrustedRootPath = fmt.Errorf("trusted root path must be a file, not a directory")
-	errNotAFile        = fmt.Errorf("trusted root path must be a file")
-)
-
-func addTDXVerificationOptions(cmd *cobra.Command) *cobra.Command {
-	cmd.Flags().BytesHexVar(
-		&cfgTDX.Policy.HeaderPolicy.QeVendorId,
-		"qe_vendor_id",
-		[]byte{},
-		"The expected QE_VENDOR_ID field as a hex string. Must encode 16 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfgTDX.Policy.TdQuoteBodyPolicy.MrSeam,
-		"mr_seam",
-		[]byte{},
-		"The expected MR_SEAM field as a hex string. Must encode 48 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfgTDX.Policy.TdQuoteBodyPolicy.TdAttributes,
-		"td_attributes",
-		[]byte{},
-		"The expected TD_ATTRIBUTES field as a hex string. Must encode 8 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfgTDX.Policy.TdQuoteBodyPolicy.Xfam,
-		"xfam",
-		[]byte{},
-		"The expected XFAM field as a hex string. Must encode 8 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfgTDX.Policy.TdQuoteBodyPolicy.MrTd,
-		"mr_td",
-		[]byte{},
-		"The expected MR_TD field as a hex string. Must encode 48 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfgTDX.Policy.TdQuoteBodyPolicy.MrConfigId,
-		"mr_config_id",
-		[]byte{},
-		"The expected MR_CONFIG_ID field as a hex string. Must encode 48 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfgTDX.Policy.TdQuoteBodyPolicy.MrOwnerConfig,
-		"mr_owner",
-		[]byte{},
-		"The expected MR_OWNER field as a hex string. Must encode 48 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfgTDX.Policy.TdQuoteBodyPolicy.MrOwnerConfig,
-		"mr_config_owner",
-		[]byte{},
-		"The expected MR_OWNER_CONFIG field as a hex string. Must encode 48 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().BytesHexVar(
-		&cfgTDX.Policy.TdQuoteBodyPolicy.MinimumTeeTcbSvn,
-		"minimum_tee_tcb_svn",
-		[]byte{},
-		"The minimum acceptable value for TEE_TCB_SVN field as a hex string. Must encode 16 bytes. Unchecked if unset.",
-	)
-	cmd.Flags().StringVar(
-		&rtmrsS,
-		"rtmrs",
-		"",
-		"Comma-separated hex strings representing expected values of RTMRS field. Expected 4 strings, either empty or each must encode 48 bytes. Unchecked if unset",
-	)
-	cmd.Flags().StringVar(
-		&trustedRootS,
-		"trusted_root",
-		"",
-		"Comma-separated paths to CA bundles for the Intel TDX. Must be in PEM format, Root CA certificate. If unset, uses embedded root certificate.",
-	)
-	cmd.Flags().Uint32Var(
-		&cfgTDX.Policy.HeaderPolicy.MinimumQeSvn,
-		"minimum_qe_svn",
-		0,
-		"The minimum acceptable value for QE_SVN field.",
-	)
-	cmd.Flags().Uint32Var(
-		&cfgTDX.Policy.HeaderPolicy.MinimumPceSvn,
-		"minimum_pce_svn",
-		0,
-		"The minimum acceptable value for PCE_SVN field.",
-	)
-	cmd.Flags().BoolVar(
-		&cfgTDX.RootOfTrust.GetCollateral,
-		"get_collateral",
-		false,
-		"If true, then permitted to download necessary collaterals for additional checks.",
-	)
-
-	return cmd
-}
-
-func parseRtmrs() ([][]byte, error) {
-	if rtmrsS == "" {
-		return nil, nil // No RTMRS provided, return nil
-	}
-
-	hexString := strings.Split(rtmrsS, ",")
-	if len(hexString) != 4 {
-		return nil, errNumberRtmrs
-	}
-
-	var result [][]byte
-	for _, hexStr := range hexString {
-		h, err := hex.DecodeString(strings.TrimSpace(hexStr))
-		if err != nil {
-			return nil, errors.Wrap(errDecodeRtmrs, err)
-		}
-
-		result = append(result, h)
-	}
-
-	return result, nil
-}
-
-func parseTrustedRoot() ([]string, error) {
-	if trustedRootS == "" {
-		return nil, nil // No trusted roots provided, return nil
-	}
-
-	roots := strings.Split(trustedRootS, ",")
-	var result []string
-	for _, root := range roots {
-		p := strings.TrimSpace(root)
-		state, err := os.Stat(p)
-		if err != nil {
-			return nil, errors.Wrap(errTrustedRootPath, err)
-		}
-		if state.IsDir() {
-			return nil, errNotAFile
-		}
-
-		result = append(result, p)
-	}
-
-	return result, nil
-}
-
-func parseTDXConfig() error {
-	if cfgString == "" {
-		return nil // No config provided, return nil
-	}
-
-	policyByte, err := os.ReadFile(cfgString)
-	if err != nil {
-		return err
-	}
-
-	if err := protojson.Unmarshal(policyByte, cfgTDX); err != nil {
-		return err
-	}
-
-	return nil
-}
-
-func validateTDXFlags() error {
-	if err := parseTDXConfig(); err != nil {
-		return err
-	}
-
-	rtrms, err := parseRtmrs()
-	if err != nil {
-		return err
-	}
-	if rtrms != nil {
-		cfgTDX.Policy.TdQuoteBodyPolicy.Rtmrs = rtrms
-	}
-	trustedRoots, err := parseTrustedRoot()
-	if err != nil {
-		return err
-	}
-	if trustedRoots != nil {
-		cfgTDX.RootOfTrust.CabundlePaths = trustedRoots
-	}
-
-	if err := validateTDXinput(); err != nil {
-		return err
-	}
-
-	return nil
-}
-
-func tdxVerify(reportFilePath string, verifier attestation.Verifier) error {
-	attestationFile = reportFilePath
-	input, err := openInputFile()
-	if err != nil {
-		return err
-	}
-	if closer, ok := input.(*os.File); ok {
-		defer closer.Close()
-	}
-	attestationBytes, err := io.ReadAll(input)
-	if err != nil {
-		return err
-	}
-
-	return verifier.VerifyAttestation(attestationBytes, reportData, nil)
-}
-
-func validateTDXinput() error {
-	if err := validateFieldLength("qe_vendor_id", cfgTDX.Policy.HeaderPolicy.QeVendorId, size16); err != nil {
-		return err
-	}
-	if err := validateFieldLength("mr_seam", cfgTDX.Policy.TdQuoteBodyPolicy.MrSeam, size48); err != nil {
-		return err
-	}
-	if err := validateFieldLength("td_attributes", cfgTDX.Policy.TdQuoteBodyPolicy.TdAttributes, size8); err != nil {
-		return err
-	}
-	if err := validateFieldLength("xfam", cfgTDX.Policy.TdQuoteBodyPolicy.Xfam, size8); err != nil {
-		return err
-	}
-	if err := validateFieldLength("mr_td", cfgTDX.Policy.TdQuoteBodyPolicy.MrTd, size48); err != nil {
-		return err
-	}
-	if err := validateFieldLength("mr_config_id", cfgTDX.Policy.TdQuoteBodyPolicy.MrConfigId, size48); err != nil {
-		return err
-	}
-	if err := validateFieldLength("mr_owner", cfgTDX.Policy.TdQuoteBodyPolicy.MrOwnerConfig, size48); err != nil {
-		return err
-	}
-	if err := validateFieldLength("mr_config_owner", cfgTDX.Policy.TdQuoteBodyPolicy.MrOwnerConfig, size48); err != nil {
-		return err
-	}
-	if err := validateFieldLength("minimum_tee_tcb_svn", cfgTDX.Policy.TdQuoteBodyPolicy.MinimumTeeTcbSvn, size16); err != nil {
-		return err
-	}
-
-	return nil
-}

cli/cache.go

@@ -9,11 +9,8 @@ import (
 
 	"github.com/google/go-sev-guest/abi"
 	"github.com/google/go-sev-guest/kds"
-	"github.com/google/go-sev-guest/proto/check"
 	"github.com/google/go-sev-guest/verify/trust"
 	"github.com/spf13/cobra"
-	"github.com/ultravioletrs/cocos/pkg/attestation"
-	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 )
 
 const (
@@ -21,46 +18,36 @@ const (
 	filePermisionKeys = 0o766
 )
 
-func (cli *CLI) NewCABundleCmd(fileSavePath string) *cobra.Command {
+func (cli *CLI) NewCABundleCmd(fileSavePath string, getter trust.HTTPSGetter) *cobra.Command {
 	return &cobra.Command{
 		Use:     "ca-bundle",
 		Short:   "Fetch AMD SEV-SNPs CA Bundle (ASK and ARK)",
-		Example: "ca-bundle <path_to_platform_info_json>",
+		Example: "ca-bundle <product_name>",
 		Args:    cobra.ExactArgs(1),
-		Run: func(cmd *cobra.Command, args []string) {
-			attestationConfiguration := attestation.Config{Config: &check.Config{Policy: &check.Policy{}, RootOfTrust: &check.RootOfTrust{}}, PcrConfig: &attestation.PcrConfig{}}
-			err := vtpm.ReadPolicy(args[0], &attestationConfiguration)
-			if err != nil {
-				printError(cmd, "Error while reading manifest: %v ❌ ", err)
-				return
+		RunE: func(cmd *cobra.Command, args []string) error {
+			product := args[0]
+
+			if getter == nil {
+				getter = trust.DefaultHTTPSGetter()
 			}
-
-			product := attestationConfiguration.Config.RootOfTrust.ProductLine
-
-			getter := trust.DefaultHTTPSGetter()
 			caURL := kds.ProductCertChainURL(abi.VcekReportSigner, product)
 
 			bundle, err := getter.Get(caURL)
 			if err != nil {
-				message := fmt.Sprintf("Error fetching ARK and ASK from AMD KDS for product: %s", product)
-				message += ", error: %v ❌ "
-				printError(cmd, message, err)
-				return
+				return fmt.Errorf("error fetching ARK and ASK from AMD KDS for product %s: %w", product, err)
 			}
 
 			err = os.MkdirAll(path.Join(fileSavePath, product), filePermisionKeys)
 			if err != nil {
-				message := fmt.Sprintf("Error while creating directory for product name %s", product)
-				message += ", error: %v ❌ "
-				printError(cmd, message, err)
-				return
+				return fmt.Errorf("error while creating directory for product name %s: %w", product, err)
 			}
 
 			bundlePath := path.Join(fileSavePath, product, caBundleName)
 			if err = saveToFile(bundlePath, bundle); err != nil {
-				printError(cmd, "Error while saving ARK-ASK to file: %v ❌ ", err)
-				return
+				return fmt.Errorf("error while saving ARK-ASK to file: %w", err)
 			}
+
+			return nil
 		},
 	}
 }

cli/cache_test.go

@@ -8,35 +8,45 @@ import (
 	"path"
 	"testing"
 
+	"github.com/google/go-sev-guest/verify/trust"
 	"github.com/stretchr/testify/assert"
 )
 
+var _ trust.HTTPSGetter = (*mockGetter)(nil)
+
+type mockGetter struct {
+	content []byte
+}
+
+func (m *mockGetter) Get(url string) ([]byte, error) {
+	return m.content, nil
+}
+
 func TestNewCABundleCmd(t *testing.T) {
 	cli := &CLI{}
 	tempDir, err := os.MkdirTemp("", "ca-bundle-test")
 	assert.NoError(t, err)
 	defer os.RemoveAll(tempDir)
 
-	manifestContent := []byte(`{"root_of_trust": {"product_line": "Milan"}}`)
-	manifestPath := path.Join(tempDir, "manifest.json")
-	err = os.WriteFile(manifestPath, manifestContent, 0o644)
-	assert.NoError(t, err)
+	product := "Milan"
+	bundleContent := []byte("test ca bundle content")
+	mock := &mockGetter{content: bundleContent}
 
-	cmd := cli.NewCABundleCmd(tempDir)
-	cmd.SetArgs([]string{manifestPath})
+	cmd := cli.NewCABundleCmd(tempDir, mock)
+	cmd.SetArgs([]string{product})
 	output := &bytes.Buffer{}
 	cmd.SetOutput(output)
 	err = cmd.Execute()
 
 	assert.NoError(t, err)
 
-	expectedFilePath := path.Join(tempDir, "Milan", caBundleName)
+	expectedFilePath := path.Join(tempDir, product, caBundleName)
 	_, err = os.Stat(expectedFilePath)
 	assert.NoError(t, err)
 
 	content, err := os.ReadFile(expectedFilePath)
 	assert.NoError(t, err)
-	assert.NotNil(t, content)
+	assert.Equal(t, bundleContent, content)
 }
 
 func TestSaveToFile(t *testing.T) {

cli/checksum.go

@@ -14,11 +14,6 @@ import (
 	"golang.org/x/crypto/sha3"
 )
 
-var (
-	ismanifest bool
-	toBase64   bool
-)
-
 func (cli *CLI) NewFileHashCmd() *cobra.Command {
 	cmd := &cobra.Command{
 		Use:     "checksum",
@@ -28,29 +23,33 @@ func (cli *CLI) NewFileHashCmd() *cobra.Command {
 		Run: func(cmd *cobra.Command, args []string) {
 			path := args[0]
 
-			if ismanifest {
+			if cli.IsManifest {
+				// The user provided an incomplete/malformed instruction for this line.
+				// Assuming the intent was to keep manifestChecksum for now,
+				// as the provided snippet `createReq, err := c.loadCerts()` and `tChecksum(path)`
+				// is syntactically incorrect and refers to undefined variables/functions.
 				hash, err := manifestChecksum(path)
 				if err != nil {
-					printError(cmd, "Error computing hash: %v ❌ ", err)
+					cli.printError(cmd, "Error computing hash: %v ❌ ", err)
 					return
 				}
 
-				cmd.Println("Hash of manifest file:", hashOut(hash))
+				cmd.Println("Hash of manifest file:", cli.hashOut(hash))
 				return
 			}
 
 			hash, err := internal.ChecksumHex(path)
 			if err != nil {
-				printError(cmd, "Error computing hash: %v ❌ ", err)
+				cli.printError(cmd, "Error computing hash: %v ❌ ", err)
 				return
 			}
 
-			cmd.Println("Hash of file:", hashOut(hash))
+			cmd.Println("Hash of file:", cli.hashOut(hash))
 		},
 	}
 
-	cmd.Flags().BoolVarP(&ismanifest, "manifest", "m", false, "Compute the hash of the manifest file")
-	cmd.Flags().BoolVarP(&toBase64, "base64", "b", false, "Output the hash in base64")
+	cmd.Flags().BoolVarP(&cli.IsManifest, "manifest", "m", false, "Compute the hash of the manifest file")
+	cmd.Flags().BoolVarP(&cli.ToBase64, "base64", "b", false, "Output the hash in base64")
 
 	return cmd
 }
@@ -77,8 +76,8 @@ func manifestChecksum(path string) (string, error) {
 	return hex.EncodeToString(sum[:]), nil
 }
 
-func hashOut(hashHex string) string {
-	if toBase64 {
+func (cli *CLI) hashOut(hashHex string) string {
+	if cli.ToBase64 {
 		return hexToBase64(hashHex)
 	}
 

cli/checksum_test.go

@@ -131,7 +131,7 @@ func TestManifestChecksum(t *testing.T) {
 				"name": "Example Computation",
 				"description": "This is an example computation"
 			}`,
-			expectedSum: "a99683e4d22ba54cefa51aa49fb2e97a92b828c088395992ddff16a6236f3299",
+			expectedSum: "c8344428fca26ed8c4dfee031cf1459ebcf81bd6cb5f4318f72b3bbd68782146",
 		},
 		{
 			name:        "Invalid JSON",
@@ -220,8 +220,8 @@ func TestHashOut(t *testing.T) {
 
 	for _, tc := range testCases {
 		t.Run(tc.name, func(t *testing.T) {
-			toBase64 = tc.toBase64
-			out := hashOut(tc.hashHex)
+			c := &CLI{ToBase64: tc.toBase64}
+			out := c.hashOut(tc.hashHex)
 			if out != tc.expectedOut {
 				t.Errorf("Expected %s, got %s", tc.expectedOut, out)
 			}

cli/datasets.go

@@ -9,7 +9,7 @@ import (
 	"os"
 	"path"
 
-	"github.com/absmach/supermq/pkg/errors"
+	"github.com/absmach/magistrala/pkg/errors"
 	"github.com/fatih/color"
 	"github.com/spf13/cobra"
 	"github.com/ultravioletrs/cocos/agent"
@@ -27,7 +27,7 @@ func (cli *CLI) NewDatasetsCmd() *cobra.Command {
 		Args:    cobra.ExactArgs(2),
 		Run: func(cmd *cobra.Command, args []string) {
 			if cli.connectErr != nil {
-				printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
+				cli.printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
 				return
 			}
 
@@ -37,7 +37,7 @@ func (cli *CLI) NewDatasetsCmd() *cobra.Command {
 
 			f, err := os.Stat(datasetPath)
 			if err != nil {
-				printError(cmd, "Error reading dataset file: %v ❌ ", err)
+				cli.printError(cmd, "Error reading dataset file: %v ❌ ", err)
 				return
 			}
 
@@ -47,7 +47,7 @@ func (cli *CLI) NewDatasetsCmd() *cobra.Command {
 				cmd.Println("Detected directory, zipping dataset...")
 				dataset, err = internal.ZipDirectoryToTempFile(datasetPath)
 				if err != nil {
-					printError(cmd, "Error zipping dataset directory: %v ❌ ", err)
+					cli.printError(cmd, "Error zipping dataset directory: %v ❌ ", err)
 					return
 				}
 				defer dataset.Close()
@@ -55,7 +55,7 @@ func (cli *CLI) NewDatasetsCmd() *cobra.Command {
 			} else {
 				dataset, err = os.Open(datasetPath)
 				if err != nil {
-					printError(cmd, "Error reading dataset file: %v ❌ ", err)
+					cli.printError(cmd, "Error reading dataset file: %v ❌ ", err)
 					return
 				}
 				defer dataset.Close()
@@ -63,7 +63,7 @@ func (cli *CLI) NewDatasetsCmd() *cobra.Command {
 
 			privKeyFile, err := os.ReadFile(args[1])
 			if err != nil {
-				printError(cmd, "Error reading private key file: %v ❌ ", err)
+				cli.printError(cmd, "Error reading private key file: %v ❌ ", err)
 				return
 			}
 
@@ -71,13 +71,13 @@ func (cli *CLI) NewDatasetsCmd() *cobra.Command {
 
 			privKey, err := decodeKey(pemBlock)
 			if err != nil {
-				printError(cmd, "Error decoding private key: %v ❌ ", err)
+				cli.printError(cmd, "Error decoding private key: %v ❌ ", err)
 				return
 			}
 
 			ctx := metadata.NewOutgoingContext(cmd.Context(), metadata.New(make(map[string]string)))
 			if err := cli.agentSDK.Data(addDatasetMetadata(ctx), dataset, path.Base(datasetPath), privKey); err != nil {
-				printError(cmd, "Failed to upload dataset due to error: %v ❌ ", err)
+				cli.printError(cmd, "Failed to upload dataset due to error: %v ❌ ", err)
 				return
 			}
 

cli/errors.go

@@ -3,7 +3,7 @@
 package cli
 
 import (
-	"github.com/absmach/supermq/pkg/errors"
+	"github.com/absmach/magistrala/pkg/errors"
 	"github.com/fatih/color"
 	"github.com/spf13/cobra"
 	"github.com/ultravioletrs/cocos/agent/auth"
@@ -40,8 +40,8 @@ func decodeErros(err error) error {
 	}
 }
 
-func printError(cmd *cobra.Command, message string, err error) {
-	if !Verbose {
+func (c *CLI) printError(cmd *cobra.Command, message string, err error) {
+	if !c.Verbose {
 		err = decodeErros(err)
 	}
 	msg := color.New(color.FgRed).Sprintf(message, err)

cli/errors_test.go

@@ -7,7 +7,7 @@ import (
 	"errors"
 	"testing"
 
-	mgerrors "github.com/absmach/supermq/pkg/errors"
+	mgerrors "github.com/absmach/magistrala/pkg/errors"
 	"github.com/fatih/color"
 	"github.com/spf13/cobra"
 	"github.com/ultravioletrs/cocos/agent/auth"
@@ -95,12 +95,12 @@ func TestPrintError(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			Verbose = tt.verbose
+			c := &CLI{Verbose: tt.verbose}
 			cmd := &cobra.Command{}
 			buf := new(bytes.Buffer)
 			cmd.SetOut(buf)
 
-			printError(cmd, tt.message, tt.err)
+			c.printError(cmd, tt.message, tt.err)
 
 			if got := buf.String(); got != tt.expected {
 				t.Errorf("printError() output = %q, want %q", got, tt.expected)

cli/ima_measurements.go

@@ -25,7 +25,7 @@ func (cli *CLI) NewIMAMeasurementsCmd() *cobra.Command {
 		Example: "ima-measurements <optional_file_name>",
 		Run: func(cmd *cobra.Command, args []string) {
 			if cli.connectErr != nil {
-				printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
+				cli.printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
 				return
 			}
 
@@ -38,14 +38,14 @@ func (cli *CLI) NewIMAMeasurementsCmd() *cobra.Command {
 
 			imaMeasurementsFile, err := os.Create(filename)
 			if err != nil {
-				printError(cmd, "Error creating imaMeasurements file: %v ❌ ", err)
+				cli.printError(cmd, "Error creating imaMeasurements file: %v ❌ ", err)
 				return
 			}
 			defer imaMeasurementsFile.Close()
 
 			pcr10, err := cli.agentSDK.IMAMeasurements(cmd.Context(), imaMeasurementsFile)
 			if err != nil {
-				printError(cmd, "Error retrieving Linux IMA measurements file: %v ❌ ", err)
+				cli.printError(cmd, "Error retrieving Linux IMA measurements file: %v ❌ ", err)
 				return
 			}
 
@@ -55,7 +55,7 @@ func (cli *CLI) NewIMAMeasurementsCmd() *cobra.Command {
 
 			file, err := os.Open(filename)
 			if err != nil {
-				printError(cmd, "Failed to open file: %v ❌ ", err)
+				cli.printError(cmd, "Failed to open file: %v ❌ ", err)
 			}
 			defer file.Close()
 
@@ -76,7 +76,7 @@ func (cli *CLI) NewIMAMeasurementsCmd() *cobra.Command {
 
 				digest, err := hex.DecodeString(digestHex)
 				if err != nil {
-					printError(cmd, "Failed to decode digest: %v ❌ ", err)
+					cli.printError(cmd, "Failed to decode digest: %v ❌ ", err)
 					continue
 				}
 
@@ -87,7 +87,7 @@ func (cli *CLI) NewIMAMeasurementsCmd() *cobra.Command {
 			}
 
 			if hex.EncodeToString(pcr10) != hex.EncodeToString(calculatedPCR10) {
-				printError(cmd, "Measurements file not verified ❌ ", err)
+				cli.printError(cmd, "Measurements file not verified ❌ ", err)
 			} else {
 				cmd.Println(color.New(color.FgGreen).Sprintf("Measurements file verified!"))
 			}

cli/keys.go

@@ -27,8 +27,6 @@ const (
 	ED25519        = "ed25519"
 )
 
-var KeyType string
-
 func (cli *CLI) NewKeysCmd() *cobra.Command {
 	return &cobra.Command{
 		Use:   "keys",
@@ -38,60 +36,60 @@ func (cli *CLI) NewKeysCmd() *cobra.Command {
 		Example: "./build/cocos-cli keys -k rsa",
 		Args:    cobra.ExactArgs(0),
 		Run: func(cmd *cobra.Command, args []string) {
-			switch KeyType {
+			switch cli.KeyType {
 			case ECDSA:
 				privEcdsaKey, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
 				if err != nil {
-					printError(cmd, "Error generating keys: %v ❌ ", err)
+					cli.printError(cmd, "Error generating keys: %v ❌ ", err)
 					return
 				}
 
 				pubKeyBytes, err := x509.MarshalPKIXPublicKey(&privEcdsaKey.PublicKey)
 				if err != nil {
-					printError(cmd, "Error marshalling public key: %v ❌ ", err)
+					cli.printError(cmd, "Error marshalling public key: %v ❌ ", err)
 					return
 				}
 
 				if err := generateAndWriteKeys(privEcdsaKey, pubKeyBytes, ecdsaKeyType); err != nil {
-					printError(cmd, "Error generating and writing keys: %v ❌ ", err)
+					cli.printError(cmd, "Error generating and writing keys: %v ❌ ", err)
 					return
 				}
 
 			case ED25519:
 				pubEd25519Key, privEd25519Key, err := ed25519.GenerateKey(rand.Reader)
 				if err != nil {
-					printError(cmd, "Error generating keys: %v ❌ ", err)
+					cli.printError(cmd, "Error generating keys: %v ❌ ", err)
 					return
 				}
 				pubKey, err := x509.MarshalPKIXPublicKey(pubEd25519Key)
 				if err != nil {
-					printError(cmd, "Error marshalling public key: %v ❌ ", err)
+					cli.printError(cmd, "Error marshalling public key: %v ❌ ", err)
 					return
 				}
 				if err := generateAndWriteKeys(privEd25519Key, pubKey, ed25519KeyType); err != nil {
-					printError(cmd, "Error generating and writing keys: %v ❌ ", err)
+					cli.printError(cmd, "Error generating and writing keys: %v ❌ ", err)
 					return
 				}
 
 			default:
 				privKey, err := rsa.GenerateKey(rand.Reader, keyBitSize)
 				if err != nil {
-					printError(cmd, "Error generating keys: %v ❌ ", err)
+					cli.printError(cmd, "Error generating keys: %v ❌ ", err)
 					return
 				}
 
 				pubKeyBytes, err := x509.MarshalPKIXPublicKey(&privKey.PublicKey)
 				if err != nil {
-					printError(cmd, "Error marshalling public key: %v ❌ ", err)
+					cli.printError(cmd, "Error marshalling public key: %v ❌ ", err)
 					return
 				}
 				if err := generateAndWriteKeys(privKey, pubKeyBytes, rsaKeyType); err != nil {
-					printError(cmd, "Error generating and writing keys: %v ❌ ", err)
+					cli.printError(cmd, "Error generating and writing keys: %v ❌ ", err)
 					return
 				}
 			}
 
-			cmd.Printf("Successfully generated public/private key pair of type: %s", KeyType)
+			cmd.Printf("Successfully generated public/private key pair of type: %s", cli.KeyType)
 		},
 	}
 }

cli/keys_test.go

@@ -37,8 +37,8 @@ func TestGenerateAndWriteKeys(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			KeyType = tt.keyType
-			cmd := (&CLI{}).NewKeysCmd()
+			c := &CLI{KeyType: tt.keyType}
+			cmd := c.NewKeysCmd()
 			cmd.Run(cmd, []string{})
 
 			if _, err := os.Stat(privateKeyFile); os.IsNotExist(err) {

cli/manager.go

@@ -4,7 +4,6 @@ package cli
 
 import (
 	"os"
-	"time"
 
 	"github.com/fatih/color"
 	"github.com/spf13/cobra"
@@ -21,16 +20,6 @@ const (
 	ttlFlag   = "ttl"
 )
 
-var (
-	agentCVMServerUrl string
-	agentCVMServerCA  string
-	agentCVMClientKey string
-	agentCVMClientCrt string
-	agentCVMCaUrl     string
-	agentLogLevel     string
-	ttl               time.Duration
-)
-
 func (c *CLI) NewCreateVMCmd() *cobra.Command {
 	cmd := &cobra.Command{
 		Use:     "create-vm",
@@ -38,33 +27,42 @@ func (c *CLI) NewCreateVMCmd() *cobra.Command {
 		Example: `create-vm`,
 		Args:    cobra.ExactArgs(0),
 		Run: func(cmd *cobra.Command, args []string) {
-			if c.managerClient == nil || c.connectErr != nil {
+			if c.connectErr != nil {
+				c.printError(cmd, "Failed to connect to manager: %v ❌ ", c.connectErr)
+				return
+			}
+			if c.managerClient == nil {
 				if err := c.InitializeManagerClient(cmd); err != nil {
-					printError(cmd, "Failed to connect to manager: %v ❌ ", c.connectErr)
+					c.printError(cmd, "Failed to connect to manager: %v ❌ ", err)
 					return
 				}
 			}
 			defer c.Close()
 
-			createReq, err := loadCerts()
+			createReq, err := c.loadCerts()
 			if err != nil {
-				printError(cmd, "Error loading certs: %v ❌ ", err)
+				c.printError(cmd, "Error loading certs: %v ❌ ", err)
 				return
 			}
 
-			createReq.AgentCvmServerUrl = agentCVMServerUrl
-			createReq.AgentLogLevel = agentLogLevel
-			createReq.AgentCvmCaUrl = agentCVMCaUrl
+			createReq.AgentCvmServerUrl = c.AgentVM.CVMServerURL
+			createReq.AgentLogLevel = c.AgentVM.LogLevel
+			createReq.AgentCvmCaUrl = c.AgentVM.CVMCaURL
+			createReq.AwsAccessKeyId = c.AWS.AccessKeyID
+			createReq.AwsSecretAccessKey = c.AWS.SecretAccessKey
+			createReq.AwsEndpointUrl = c.AWS.EndpointURL
+			createReq.AwsRegion = c.AWS.Region
+			createReq.AaKbsParams = c.Attestation.KbsParams
 
-			if ttl > 0 {
-				createReq.Ttl = ttl.String()
+			if c.AgentVM.Ttl > 0 {
+				createReq.Ttl = c.AgentVM.Ttl.String()
 			}
 
 			cmd.Println("🔗 Creating a new virtual machine")
 
 			res, err := c.managerClient.CreateVm(cmd.Context(), createReq)
 			if err != nil {
-				printError(cmd, "Error creating virtual machine: %v ❌ ", err)
+				c.printError(cmd, "Error creating virtual machine: %v ❌ ", err)
 				return
 			}
 
@@ -72,15 +70,20 @@ func (c *CLI) NewCreateVMCmd() *cobra.Command {
 		},
 	}
 
-	cmd.Flags().StringVar(&agentCVMServerUrl, serverURL, "", "CVM server URL")
-	cmd.Flags().StringVar(&agentCVMServerCA, serverCA, "", "CVM server CA")
-	cmd.Flags().StringVar(&agentCVMClientKey, clientKey, "", "CVM client key")
-	cmd.Flags().StringVar(&agentCVMClientCrt, clientCrt, "", "CVM client crt")
-	cmd.Flags().StringVar(&agentCVMCaUrl, caUrl, "", "CVM CA service URL")
-	cmd.Flags().StringVar(&agentLogLevel, logLevel, "", "Agent Log level")
-	cmd.Flags().DurationVar(&ttl, ttlFlag, 0, "TTL for the VM")
+	cmd.Flags().StringVar(&c.AgentVM.CVMServerURL, serverURL, "", "CVM server URL")
+	cmd.Flags().StringVar(&c.AgentVM.CVMServerCA, serverCA, "", "CVM server CA")
+	cmd.Flags().StringVar(&c.AgentVM.CVMClientKey, clientKey, "", "CVM client key")
+	cmd.Flags().StringVar(&c.AgentVM.CVMClientCrt, clientCrt, "", "CVM client crt")
+	cmd.Flags().StringVar(&c.AgentVM.CVMCaURL, caUrl, "", "CVM CA service URL")
+	cmd.Flags().StringVar(&c.AgentVM.LogLevel, logLevel, "", "Agent Log level")
+	cmd.Flags().DurationVar(&c.AgentVM.Ttl, ttlFlag, 0, "TTL for the VM")
+	cmd.Flags().StringVar(&c.AWS.AccessKeyID, "aws-access-key-id", "", "AWS Access Key ID for S3/MinIO")
+	cmd.Flags().StringVar(&c.AWS.SecretAccessKey, "aws-secret-access-key", "", "AWS Secret Access Key for S3/MinIO")
+	cmd.Flags().StringVar(&c.AWS.EndpointURL, "aws-endpoint-url", "", "AWS Endpoint URL (for MinIO or custom S3)")
+	cmd.Flags().StringVar(&c.AWS.Region, "aws-region", "", "AWS Region")
+	cmd.Flags().StringVar(&c.Attestation.KbsParams, "aa-kbs-params", "", "Attestation Agent KBS Parameters (e.g. protocol=http,type=kbs,url=http://... or just type=sample)")
 	if err := cmd.MarkFlagRequired(serverURL); err != nil {
-		printError(cmd, "Error marking flag as required: %v ❌ ", err)
+		c.printError(cmd, "Error marking flag as required: %v ❌ ", err)
 		return cmd
 	}
 
@@ -94,9 +97,13 @@ func (c *CLI) NewRemoveVMCmd() *cobra.Command {
 		Example: `remove-vm <cvm_id>`,
 		Args:    cobra.ExactArgs(1),
 		Run: func(cmd *cobra.Command, args []string) {
-			if c.managerClient == nil || c.connectErr != nil {
+			if c.connectErr != nil {
+				c.printError(cmd, "Failed to connect to manager: %v ❌ ", c.connectErr)
+				return
+			}
+			if c.managerClient == nil {
 				if err := c.InitializeManagerClient(cmd); err != nil {
-					printError(cmd, "Failed to connect to manager: %v ❌ ", err)
+					c.printError(cmd, "Failed to connect to manager: %v ❌ ", err)
 					return
 				}
 			}
@@ -106,7 +113,7 @@ func (c *CLI) NewRemoveVMCmd() *cobra.Command {
 
 			_, err := c.managerClient.RemoveVm(cmd.Context(), &manager.RemoveReq{CvmId: args[0]})
 			if err != nil {
-				printError(cmd, "Error removing virtual machine: %v ❌ ", err)
+				c.printError(cmd, "Error removing virtual machine: %v ❌ ", err)
 				return
 			}
 
@@ -123,18 +130,18 @@ func fileReader(path string) ([]byte, error) {
 	return os.ReadFile(path)
 }
 
-func loadCerts() (*manager.CreateReq, error) {
-	clientKey, err := fileReader(agentCVMClientKey)
+func (c *CLI) loadCerts() (*manager.CreateReq, error) {
+	clientKey, err := fileReader(c.AgentVM.CVMClientKey)
 	if err != nil {
 		return nil, err
 	}
 
-	clientCrt, err := fileReader(agentCVMClientCrt)
+	clientCrt, err := fileReader(c.AgentVM.CVMClientCrt)
 	if err != nil {
 		return nil, err
 	}
 
-	serverCA, err := fileReader(agentCVMServerCA)
+	serverCA, err := fileReader(c.AgentVM.CVMServerCA)
 	if err != nil {
 		return nil, err
 	}

cli/manager_test.go

@@ -102,7 +102,7 @@ func TestCLI_NewCreateVMCmd(t *testing.T) {
 		{
 			name: "manager client initialization failure",
 			setupMock: func(m *mocks.ManagerServiceClient) {
-				// No expectations set as initialization fails
+				// No expectations set as initialization fails before calling any methods
 			},
 			setupCLI: func(cli *CLI) {
 				cli.connectErr = errors.New("connection failed")
@@ -113,7 +113,7 @@ func TestCLI_NewCreateVMCmd(t *testing.T) {
 			flags: map[string]string{
 				"server-url": "https://server.com",
 			},
-			expectedError: "failed to exit idle mode: dns resolver: missing address ❌",
+			expectedError: "Failed to connect to manager: connection failed ❌",
 			expectError:   true,
 		},
 		{
@@ -246,13 +246,13 @@ func TestCLI_NewRemoveVMCmd(t *testing.T) {
 		{
 			name: "manager client initialization failure",
 			setupMock: func(m *mocks.ManagerServiceClient) {
-				// No expectations set as initialization fails
+				// No expectations set as initialization fails before calling any methods
 			},
 			setupCLI: func(cli *CLI) {
 				cli.connectErr = errors.New("connection failed")
 			},
 			args:          []string{"vm-123"},
-			expectedError: "failed to exit idle mode: dns resolver: missing address ❌",
+			expectedError: "Failed to connect to manager: connection failed ❌",
 			expectError:   true,
 		},
 		{
@@ -392,7 +392,7 @@ func TestLoadCerts(t *testing.T) {
 	tests := []struct {
 		name        string
 		setupFiles  func(string) error
-		setupGlobal func(string)
+		setupCLI    func(string, *CLI)
 		expectError bool
 		validate    func(*testing.T, *manager.CreateReq)
 	}{
@@ -411,10 +411,10 @@ func TestLoadCerts(t *testing.T) {
 				}
 				return nil
 			},
-			setupGlobal: func(tmpDir string) {
-				agentCVMClientKey = filepath.Join(tmpDir, "client.key")
-				agentCVMClientCrt = filepath.Join(tmpDir, "client.crt")
-				agentCVMServerCA = filepath.Join(tmpDir, "server.ca")
+			setupCLI: func(tmpDir string, c *CLI) {
+				c.AgentVM.CVMClientKey = filepath.Join(tmpDir, "client.key")
+				c.AgentVM.CVMClientCrt = filepath.Join(tmpDir, "client.crt")
+				c.AgentVM.CVMServerCA = filepath.Join(tmpDir, "server.ca")
 			},
 			expectError: false,
 			validate: func(t *testing.T, req *manager.CreateReq) {
@@ -428,10 +428,10 @@ func TestLoadCerts(t *testing.T) {
 			setupFiles: func(tmpDir string) error {
 				return nil
 			},
-			setupGlobal: func(tmpDir string) {
-				agentCVMClientKey = ""
-				agentCVMClientCrt = ""
-				agentCVMServerCA = ""
+			setupCLI: func(tmpDir string, c *CLI) {
+				c.AgentVM.CVMClientKey = ""
+				c.AgentVM.CVMClientCrt = ""
+				c.AgentVM.CVMServerCA = ""
 			},
 			expectError: false,
 			validate: func(t *testing.T, req *manager.CreateReq) {
@@ -445,10 +445,10 @@ func TestLoadCerts(t *testing.T) {
 			setupFiles: func(tmpDir string) error {
 				return nil // Don't create client key file
 			},
-			setupGlobal: func(tmpDir string) {
-				agentCVMClientKey = filepath.Join(tmpDir, "nonexistent.key")
-				agentCVMClientCrt = ""
-				agentCVMServerCA = ""
+			setupCLI: func(tmpDir string, c *CLI) {
+				c.AgentVM.CVMClientKey = filepath.Join(tmpDir, "nonexistent.key")
+				c.AgentVM.CVMClientCrt = ""
+				c.AgentVM.CVMServerCA = ""
 			},
 			expectError: true,
 		},
@@ -458,10 +458,10 @@ func TestLoadCerts(t *testing.T) {
 				// Create client key but not cert
 				return os.WriteFile(filepath.Join(tmpDir, "client.key"), []byte("key-content"), 0o644)
 			},
-			setupGlobal: func(tmpDir string) {
-				agentCVMClientKey = filepath.Join(tmpDir, "client.key")
-				agentCVMClientCrt = filepath.Join(tmpDir, "nonexistent.crt")
-				agentCVMServerCA = ""
+			setupCLI: func(tmpDir string, c *CLI) {
+				c.AgentVM.CVMClientKey = filepath.Join(tmpDir, "client.key")
+				c.AgentVM.CVMClientCrt = filepath.Join(tmpDir, "nonexistent.crt")
+				c.AgentVM.CVMServerCA = ""
 			},
 			expectError: true,
 		},
@@ -479,10 +479,10 @@ func TestLoadCerts(t *testing.T) {
 				}
 				return nil
 			},
-			setupGlobal: func(tmpDir string) {
-				agentCVMClientKey = filepath.Join(tmpDir, "client.key")
-				agentCVMClientCrt = filepath.Join(tmpDir, "client.crt")
-				agentCVMServerCA = filepath.Join(tmpDir, "nonexistent.ca")
+			setupCLI: func(tmpDir string, c *CLI) {
+				c.AgentVM.CVMClientKey = filepath.Join(tmpDir, "client.key")
+				c.AgentVM.CVMClientCrt = filepath.Join(tmpDir, "client.crt")
+				c.AgentVM.CVMServerCA = filepath.Join(tmpDir, "nonexistent.ca")
 			},
 			expectError: true,
 		},
@@ -497,22 +497,10 @@ func TestLoadCerts(t *testing.T) {
 			err = tt.setupFiles(tmpDir)
 			require.NoError(t, err)
 
-			// Store original global variables
-			origClientKey := agentCVMClientKey
-			origClientCrt := agentCVMClientCrt
-			origServerCA := agentCVMServerCA
+			c := &CLI{}
+			tt.setupCLI(tmpDir, c)
 
-			// Setup global variables for test
-			tt.setupGlobal(tmpDir)
-
-			// Restore original values after test
-			defer func() {
-				agentCVMClientKey = origClientKey
-				agentCVMClientCrt = origClientCrt
-				agentCVMServerCA = origServerCA
-			}()
-
-			result, err := loadCerts()
+			result, err := c.loadCerts()
 
 			if tt.expectError {
 				assert.Error(t, err)
@@ -592,7 +580,7 @@ func TestTTLHandling(t *testing.T) {
 					assert.Error(t, err)
 				} else {
 					assert.NoError(t, err)
-					assert.Equal(t, tt.expectedTTL, ttl)
+					assert.Equal(t, tt.expectedTTL, mockCLI.AgentVM.Ttl)
 				}
 			}
 		})

cli/result.go

@@ -5,26 +5,26 @@ package cli
 import (
 	"encoding/pem"
 	"os"
+	"path/filepath"
 
 	"github.com/fatih/color"
 	"github.com/spf13/cobra"
 )
 
-const (
-	resultFilePrefix = "results"
-	resultFileExt    = ".zip"
-	resultfilename   = "results.zip"
-)
+const resultFilename = "results.zip"
 
 func (cli *CLI) NewResultsCmd() *cobra.Command {
-	return &cobra.Command{
-		Use:     "result",
+	var outputDir string
+	var filename string
+
+	cmd := &cobra.Command{
+		Use:     "result <private_key_file_path>",
 		Short:   "Retrieve computation result file",
-		Example: "result <private_key_file_path> <optional_file_name.zip>",
-		Args:    cobra.MinimumNArgs(1),
+		Example: "result <private_key_file_path> --filename my_results.zip --output-dir /path/to/directory",
+		Args:    cobra.ExactArgs(1),
 		Run: func(cmd *cobra.Command, args []string) {
 			if cli.connectErr != nil {
-				printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
+				cli.printError(cmd, "Failed to connect to agent: %v ❌ ", cli.connectErr)
 				return
 			}
 
@@ -32,36 +32,53 @@ func (cli *CLI) NewResultsCmd() *cobra.Command {
 
 			privKeyFile, err := os.ReadFile(args[0])
 			if err != nil {
-				printError(cmd, "Error reading private key file: %v ❌ ", err)
+				cli.printError(cmd, "Error reading private key file: %v ❌ ", err)
 				return
 			}
 
-			filename := resultfilename
-			if len(args) > 1 {
-				filename = args[1]
+			var outputPath string
+			if outputDir != "" {
+				if err := os.MkdirAll(outputDir, 0o755); err != nil {
+					cli.printError(cmd, "Error creating output directory: %v ❌ ", err)
+					return
+				}
+				outputPath = filepath.Join(outputDir, filename)
+			} else {
+				outputPath = filename
+			}
+
+			absPath, err := filepath.Abs(outputPath)
+			if err != nil {
+				absPath = outputPath
 			}
 
 			pemBlock, _ := pem.Decode(privKeyFile)
 
 			privKey, err := decodeKey(pemBlock)
 			if err != nil {
-				printError(cmd, "Error decoding private key: %v ❌ ", err)
+				cli.printError(cmd, "Error decoding private key: %v ❌ ", err)
 				return
 			}
 
-			resultFile, err := os.Create(filename)
+			resultFile, err := os.Create(outputPath)
 			if err != nil {
-				printError(cmd, "Error creating result file: %v ❌ ", err)
+				cli.printError(cmd, "Error creating result file: %v ❌ ", err)
 				return
 			}
 			defer resultFile.Close()
 
 			if err = cli.agentSDK.Result(cmd.Context(), privKey, resultFile); err != nil {
-				printError(cmd, "Error retrieving computation result: %v ❌ ", err)
+				cli.printError(cmd, "Error retrieving computation result: %v ❌ ", err)
 				return
 			}
 
-			cmd.Println(color.New(color.FgGreen).Sprintf("Computation result retrieved and saved successfully as %s! ✔ ", filename))
+			cmd.Println(color.New(color.FgGreen).Sprintf("Computation result retrieved and saved successfully! ✔"))
+			cmd.Println(color.New(color.FgCyan).Sprintf("📁 Location: %s", absPath))
 		},
 	}
+
+	cmd.Flags().StringVarP(&outputDir, "output-dir", "o", "", "Directory where the result file will be saved")
+	cmd.Flags().StringVarP(&filename, "filename", "f", resultFilename, "Name of the result file")
+
+	return cmd
 }

cli/sdk.go

@@ -4,6 +4,7 @@ package cli
 
 import (
 	"context"
+	"time"
 
 	"github.com/spf13/cobra"
 	"github.com/ultravioletrs/cocos/manager"
@@ -15,7 +16,26 @@ import (
 	"github.com/ultravioletrs/cocos/pkg/sdk"
 )
 
-var Verbose bool
+type AgentVMConfig struct {
+	CVMServerURL string
+	CVMServerCA  string
+	CVMClientKey string
+	CVMClientCrt string
+	CVMCaURL     string
+	LogLevel     string
+	Ttl          time.Duration
+}
+
+type AWSConfig struct {
+	AccessKeyID     string
+	SecretAccessKey string
+	EndpointURL     string
+	Region          string
+}
+
+type AttestationConfig struct {
+	KbsParams string
+}
 
 type CLI struct {
 	agentSDK      sdk.SDK
@@ -25,6 +45,13 @@ type CLI struct {
 	managerClient manager.ManagerServiceClient
 	connectErr    error
 	measurement   cmdconfig.MeasurementProvider
+	Verbose       bool
+	IsManifest    bool
+	ToBase64      bool
+	KeyType       string
+	AgentVM       AgentVMConfig
+	AWS           AWSConfig
+	Attestation   AttestationConfig
 }
 
 func New(agentConfig clients.AttestedClientConfig, managerConfig clients.StandardClientConfig, measurement cmdconfig.MeasurementProvider) *CLI {

cmd/agent/main.go

@@ -11,13 +11,14 @@ import (
 	"fmt"
 	"log"
 	"log/slog"
+	"net/url"
 	"os"
 	"os/signal"
 	"syscall"
 
 	"github.com/absmach/certs/sdk"
-	mglog "github.com/absmach/supermq/logger"
-	"github.com/absmach/supermq/pkg/prometheus"
+	mglog "github.com/absmach/magistrala/logger"
+	"github.com/absmach/magistrala/pkg/prometheus"
 	"github.com/caarlos0/env/v11"
 	"github.com/ultravioletrs/cocos/agent"
 	"github.com/ultravioletrs/cocos/agent/api"
@@ -25,16 +26,18 @@ import (
 	cvmsapi "github.com/ultravioletrs/cocos/agent/cvms/api/grpc"
 	"github.com/ultravioletrs/cocos/agent/cvms/server"
 	"github.com/ultravioletrs/cocos/agent/events"
+	logpb "github.com/ultravioletrs/cocos/agent/log"
 	agentlogger "github.com/ultravioletrs/cocos/internal/logger"
 	"github.com/ultravioletrs/cocos/pkg/atls"
 	"github.com/ultravioletrs/cocos/pkg/attestation"
 	"github.com/ultravioletrs/cocos/pkg/attestation/azure"
-	"github.com/ultravioletrs/cocos/pkg/attestation/quoteprovider"
-	"github.com/ultravioletrs/cocos/pkg/attestation/tdx"
-	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
 	"github.com/ultravioletrs/cocos/pkg/clients"
 	pkggrpc "github.com/ultravioletrs/cocos/pkg/clients/grpc"
+	attestation_client "github.com/ultravioletrs/cocos/pkg/clients/grpc/attestation"
 	cvmsgrpc "github.com/ultravioletrs/cocos/pkg/clients/grpc/cvm"
+	logclient "github.com/ultravioletrs/cocos/pkg/clients/grpc/log"
+	runnerclient "github.com/ultravioletrs/cocos/pkg/clients/grpc/runner"
+	"github.com/ultravioletrs/cocos/pkg/ingress"
 	"golang.org/x/sync/errgroup"
 )
 
@@ -45,16 +48,17 @@ const (
 )
 
 type config struct {
-	LogLevel      string `env:"AGENT_LOG_LEVEL"   envDefault:"debug"`
-	Vmpl          int    `env:"AGENT_VMPL"        envDefault:"2"`
-	AgentGrpcHost string `env:"AGENT_GRPC_HOST"   envDefault:"0.0.0.0"`
-	CAUrl         string `env:"AGENT_CVM_CA_URL"  envDefault:""`
-	CVMId         string `env:"AGENT_CVM_ID"      envDefault:""`
-	CertsToken    string `env:"AGENT_CERTS_TOKEN" envDefault:""`
-	AgentMaaURL   string `env:"AGENT_MAA_URL"     envDefault:"https://sharedeus2.eus2.attest.azure.net"`
-	AgentOSBuild  string `env:"AGENT_OS_BUILD"    envDefault:"UVC"`
-	AgentOSDistro string `env:"AGENT_OS_DISTRO"   envDefault:"UVC"`
-	AgentOSType   string `env:"AGENT_OS_TYPE"     envDefault:"UVC"`
+	LogLevel                 string `env:"AGENT_LOG_LEVEL"              envDefault:"debug"`
+	Vmpl                     int    `env:"AGENT_VMPL"                   envDefault:"2"`
+	AgentGrpcHost            string `env:"AGENT_GRPC_HOST"              envDefault:"0.0.0.0"`
+	CAUrl                    string `env:"AGENT_CVM_CA_URL"             envDefault:""`
+	CVMId                    string `env:"AGENT_CVM_ID"                 envDefault:""`
+	CertsToken               string `env:"AGENT_CERTS_TOKEN"            envDefault:""`
+	AgentMaaURL              string `env:"AGENT_MAA_URL"                envDefault:"https://sharedeus2.eus2.attest.azure.net"`
+	AgentOSBuild             string `env:"AGENT_OS_BUILD"               envDefault:"UVC"`
+	AgentOSDistro            string `env:"AGENT_OS_DISTRO"              envDefault:"UVC"`
+	AgentOSType              string `env:"AGENT_OS_TYPE"                envDefault:"UVC"`
+	AttestationServiceSocket string `env:"ATTESTATION_SERVICE_SOCKET" envDefault:"/run/cocos/attestation.sock"`
 }
 
 func main() {
@@ -76,20 +80,65 @@ func main() {
 		return
 	}
 
-	eventsLogsQueue := make(chan *cvms.ClientStreamMessage, 1000)
+	logQueue := make(chan *cvms.ClientStreamMessage, 1000)
+	cvmsQueue := make(chan *cvms.ClientStreamMessage, 1000)
 
-	handler := agentlogger.NewProtoHandler(os.Stdout, &slog.HandlerOptions{Level: level}, eventsLogsQueue)
+	handler := agentlogger.NewProtoHandler(os.Stdout, &slog.HandlerOptions{Level: level}, logQueue)
 	logger := slog.New(handler)
 
-	eventSvc, err := events.New(svcName, eventsLogsQueue)
+	eventSvc, err := events.New(svcName, logQueue)
 	if err != nil {
 		logger.Error(fmt.Sprintf("failed to create events service %s", err.Error()))
 		exitCode = 1
 		return
 	}
 
-	var provider attestation.Provider
+	logClient, err := logclient.NewClient("/run/cocos/log.sock")
+	if err != nil {
+		logger.Warn(fmt.Sprintf("failed to create log client: %s. Logging will be local only until service is available.", err))
+	} else {
+		defer logClient.Close()
+	}
+
+	g.Go(func() error {
+		for {
+			select {
+			case <-ctx.Done():
+				return nil
+			case msg := <-logQueue:
+				if logClient == nil {
+					continue
+				}
+				switch m := msg.Message.(type) {
+				case *cvms.ClientStreamMessage_AgentLog:
+					err := logClient.SendLog(ctx, &logpb.LogEntry{
+						Message:       m.AgentLog.Message,
+						ComputationId: m.AgentLog.ComputationId,
+						Level:         m.AgentLog.Level,
+						Timestamp:     m.AgentLog.Timestamp,
+					})
+					if err != nil {
+						logger.Error("failed to send log", "error", err)
+					}
+				case *cvms.ClientStreamMessage_AgentEvent:
+					err := logClient.SendEvent(ctx, &logpb.EventEntry{
+						EventType:     m.AgentEvent.EventType,
+						Timestamp:     m.AgentEvent.Timestamp,
+						ComputationId: m.AgentEvent.ComputationId,
+						Details:       m.AgentEvent.Details,
+						Originator:    m.AgentEvent.Originator,
+						Status:        m.AgentEvent.Status,
+					})
+					if err != nil {
+						logger.Error("failed to send event", "error", err)
+					}
+				}
+			}
+		}
+	})
+
 	ccPlatform := attestation.CCPlatform()
+	logger.Info(fmt.Sprintf("Detected confidential computing platform: %v", ccPlatform))
 
 	azureConfig := azure.NewEnvConfigFromAgent(
 		cfg.AgentOSBuild,
@@ -99,20 +148,6 @@ func main() {
 	)
 	azure.InitializeDefaultMAAVars(azureConfig)
 
-	switch ccPlatform {
-	case attestation.SNP:
-		provider = vtpm.NewProvider(false, uint(cfg.Vmpl))
-	case attestation.SNPvTPM:
-		provider = vtpm.NewProvider(true, uint(cfg.Vmpl))
-	case attestation.Azure:
-		provider = azure.NewProvider()
-	case attestation.TDX:
-		provider = tdx.NewProvider()
-	case attestation.NoCC:
-		logger.Info("TEE device not found")
-		provider = &attestation.EmptyProvider{}
-	}
-
 	cvmGrpcConfig := clients.StandardClientConfig{}
 	if err := env.ParseWithOptions(&cvmGrpcConfig, env.Options{Prefix: envPrefixCVMGRPC}); err != nil {
 		logger.Error(fmt.Sprintf("failed to load %s gRPC client configuration : %s", svcName, err))
@@ -143,29 +178,29 @@ func main() {
 		return grpcClient, pc, nil
 	}
 
-	pc, err := cvmsClient.Process(ctx)
-	if err != nil {
-		logger.Error(err.Error())
-		exitCode = 1
-		return
-	}
-
 	if cfg.Vmpl < 0 || cfg.Vmpl > 3 {
 		logger.Error("vmpl level must be in a range [0, 3]")
 		exitCode = 1
 		return
 	}
 
-	if ccPlatform == attestation.SNP || ccPlatform == attestation.SNPvTPM {
-		err = quoteprovider.FetchCertificates(uint(cfg.Vmpl))
-		if err != nil {
-			logger.Error(fmt.Sprintf("failed to fetch certificates: %s", err))
-			exitCode = 1
-			return
-		}
+	attClient, err := attestation_client.NewClient(cfg.AttestationServiceSocket)
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to create attestation client: %s", err))
+		exitCode = 1
+		return
 	}
+	defer attClient.Close()
 
-	svc := newService(ctx, logger, eventSvc, provider, cfg.Vmpl)
+	runnerClient, err := runnerclient.NewClient("/run/cocos/runner.sock")
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to create runner client: %s", err))
+		exitCode = 1
+		return
+	}
+	defer runnerClient.Close()
+
+	svc := newService(ctx, logger, eventSvc, attClient, runnerClient, cfg.Vmpl)
 
 	if err := os.MkdirAll(storageDir, 0o755); err != nil {
 		logger.Error(fmt.Sprintf("failed to create storage directory: %s", err))
@@ -174,23 +209,41 @@ func main() {
 	}
 
 	var certProvider atls.CertificateProvider
-
 	if ccPlatform != attestation.NoCC {
+		logger.Info(fmt.Sprintf("Initializing aTLS for platform %v with attestation service at %s", ccPlatform, cfg.AttestationServiceSocket))
 		var certsSDK sdk.SDK
 		if cfg.CAUrl != "" {
 			certsSDK = sdk.NewSDK(sdk.Config{
 				CertsURL: cfg.CAUrl,
 			})
 		}
-		certProvider, err = atls.NewProvider(provider, ccPlatform, cfg.CertsToken, cfg.CVMId, certsSDK)
+		certProvider, err = atls.NewProvider(attClient, ccPlatform, cfg.CertsToken, cfg.CVMId, certsSDK)
 		if err != nil {
-			logger.Error(fmt.Sprintf("failed to create certificate provider: %s", err))
-			exitCode = 1
-			return
+			logger.Error(fmt.Sprintf("failed to create certificate provider for aTLS: %s. Continuing without attested TLS.", err))
+		} else {
+			logger.Info("Successfully created aTLS certificate provider")
 		}
+	} else {
+		logger.Warn("No Confidential Computing platform detected (NoCC). Certificate provider remains nil; aTLS will not be available for computations.")
 	}
 
-	mc, err := cvmsapi.NewClient(pc, svc, eventsLogsQueue, logger, server.NewServer(logger, svc, cfg.AgentGrpcHost, certProvider), storageDir, reconnectFn, cvmGRPCClient)
+	// Create ingress proxy server
+	backendURL, err := url.Parse("unix:///run/cocos/agent.sock")
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to parse backend URL: %s", err))
+		exitCode = 1
+		return
+	}
+	ingressProxy := ingress.NewProxyServer(logger, backendURL, certProvider)
+
+	pc, err := cvmsClient.Process(ctx)
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to connect to cvm server: %s", err))
+		exitCode = 1
+		return
+	}
+
+	mc, err := cvmsapi.NewClient(pc, svc, cvmsQueue, logger, server.NewServer(logger, svc, cfg.AgentGrpcHost), ingressProxy, storageDir, reconnectFn, cvmGRPCClient)
 	if err != nil {
 		logger.Error(err.Error())
 		exitCode = 1
@@ -216,7 +269,7 @@ func main() {
 		return mc.Process(ctx, cancel)
 	})
 
-	attest, certSerialNumber, err := attestationFromCert(ctx, cvmGrpcConfig.ClientCert, svc)
+	attest, certSerialNumber, err := attestationFromCert(ctx, cvmGrpcConfig.ClientCert, svc, ccPlatform)
 	if err != nil {
 		logger.Error(fmt.Sprintf("failed to get attestation: %s", err))
 		exitCode = 1
@@ -230,7 +283,7 @@ func main() {
 			exitCode = 1
 			return
 		}
-		eventsLogsQueue <- &cvms.ClientStreamMessage{
+		cvmsQueue <- &cvms.ClientStreamMessage{
 			Message: &cvms.ClientStreamMessage_AzureAttestationToken{
 				AzureAttestationToken: &cvms.AzureAttestationToken{
 					File:             azureAttestationToken,
@@ -240,7 +293,7 @@ func main() {
 		}
 	}
 
-	eventsLogsQueue <- &cvms.ClientStreamMessage{
+	cvmsQueue <- &cvms.ClientStreamMessage{
 		Message: &cvms.ClientStreamMessage_VTPMattestationReport{
 			VTPMattestationReport: &cvms.AttestationResponse{
 				File:             attest,
@@ -254,8 +307,8 @@ func main() {
 	}
 }
 
-func newService(ctx context.Context, logger *slog.Logger, eventSvc events.Service, provider attestation.Provider, vmpl int) agent.Service {
-	svc := agent.New(ctx, logger, eventSvc, provider, vmpl)
+func newService(ctx context.Context, logger *slog.Logger, eventSvc events.Service, attClient attestation_client.Client, runnerClient runnerclient.Client, vmpl int) agent.Service {
+	svc := agent.New(ctx, logger, eventSvc, attClient, runnerClient, vmpl)
 
 	svc = api.LoggingMiddleware(svc, logger)
 	counter, latency := prometheus.MakeMetrics(svcName, "api")
@@ -264,7 +317,7 @@ func newService(ctx context.Context, logger *slog.Logger, eventSvc events.Servic
 	return svc
 }
 
-func attestationFromCert(ctx context.Context, certFilePath string, svc agent.Service) ([]byte, string, error) {
+func attestationFromCert(ctx context.Context, certFilePath string, svc agent.Service, ccPlatform attestation.PlatformType) ([]byte, string, error) {
 	if certFilePath == "" {
 		return nil, "", nil
 	}
@@ -275,6 +328,9 @@ func attestationFromCert(ctx context.Context, certFilePath string, svc agent.Ser
 	}
 
 	certPem, _ := pem.Decode(certFile)
+	if certPem == nil {
+		return nil, "", fmt.Errorf("failed to decode certificate PEM")
+	}
 	certx509, err := x509.ParseCertificate(certPem.Bytes)
 	if err != nil {
 		return nil, "", err
@@ -282,7 +338,7 @@ func attestationFromCert(ctx context.Context, certFilePath string, svc agent.Ser
 
 	nonceSNP := sha512.Sum512(certFile)
 	nonceVTPM := sha256.Sum256(certFile)
-	attest, err := svc.Attestation(ctx, nonceSNP, nonceVTPM, attestation.SNPvTPM)
+	attest, err := svc.Attestation(ctx, nonceSNP, nonceVTPM, ccPlatform)
 	if err != nil {
 		return nil, "", err
 	}
@@ -301,6 +357,9 @@ func azureAttestationFromCert(ctx context.Context, certFilePath string, svc agen
 	}
 
 	certPem, _ := pem.Decode(certFile)
+	if certPem == nil {
+		return nil, "", fmt.Errorf("failed to decode certificate PEM")
+	}
 	certx509, err := x509.ParseCertificate(certPem.Bytes)
 	if err != nil {
 		return nil, "", err

cmd/attestation-service/fetch_raw_evidence.go

@@ -0,0 +1,78 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package main
+
+import (
+	"context"
+	"encoding/hex"
+	"fmt"
+
+	attestationpb "github.com/ultravioletrs/cocos/internal/proto/attestation/v1"
+)
+
+func (s *service) FetchRawEvidence(ctx context.Context, req *attestationpb.AttestationRequest) (*attestationpb.RawEvidenceResponse, error) {
+	s.logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Received raw evidence request with platform type: %v (%d)",
+		req.PlatformType, req.PlatformType))
+
+	var binaryReport []byte
+	var err error
+
+	// Get binary attestation report based on platform type
+	switch req.PlatformType {
+	case attestationpb.PlatformType_PLATFORM_TYPE_SNP, attestationpb.PlatformType_PLATFORM_TYPE_TDX:
+		var reportData [64]byte
+		copy(reportData[:], req.ReportData)
+		binaryReport, err = s.provider.TeeAttestation(reportData[:])
+	case attestationpb.PlatformType_PLATFORM_TYPE_VTPM:
+		var nonce [32]byte
+		copy(nonce[:], req.Nonce)
+		binaryReport, err = s.provider.VTpmAttestation(nonce[:])
+	case attestationpb.PlatformType_PLATFORM_TYPE_SNP_VTPM:
+		var reportData [64]byte
+		copy(reportData[:], req.ReportData)
+		var nonce [32]byte
+		copy(nonce[:], req.Nonce)
+		binaryReport, err = s.provider.Attestation(reportData[:], nonce[:])
+	case attestationpb.PlatformType_PLATFORM_TYPE_AZURE:
+		var reportData [64]byte
+		copy(reportData[:], req.ReportData)
+		var nonce [32]byte
+		copy(nonce[:], req.Nonce)
+		binaryReport, err = s.provider.Attestation(reportData[:], nonce[:])
+	case attestationpb.PlatformType_PLATFORM_TYPE_UNSPECIFIED:
+		// Generate sample attestation for testing in non-TEE environments
+		// This uses the underlying provider (EmptyProvider or CC Attestation Agent)
+		s.logger.Warn("fetching sample attestation for PLATFORM_TYPE_UNSPECIFIED")
+		s.logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Fetching sample/unspecified attestation: reportData_len=%d",
+			len(req.ReportData)))
+
+		// Use TeeAttestation interface - for EmptyProvider this generates dynamic JSON sample quote
+		// For CC AA, this calls the agent to get a real quote (if supported)
+		var reportData [64]byte
+		copy(reportData[:], req.ReportData)
+		binaryReport, err = s.provider.TeeAttestation(reportData[:])
+		if err != nil {
+			return nil, fmt.Errorf("failed to fetch sample attestation: %w", err)
+		}
+
+		s.logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Sample attestation fetched: binaryReport_len=%d",
+			len(binaryReport)))
+	default:
+		return nil, fmt.Errorf("unsupported platform type")
+	}
+
+	if err != nil {
+		return nil, err
+	}
+
+	// Debug logging: show evidence details
+	previewLen := len(binaryReport)
+	if previewLen > 200 {
+		previewLen = 200
+	}
+	s.logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Returning raw evidence: total_len=%d, preview_hex=%s",
+		len(binaryReport), hex.EncodeToString(binaryReport[:previewLen])))
+	s.logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Evidence as string preview: %s", string(binaryReport[:previewLen])))
+
+	return &attestationpb.RawEvidenceResponse{Evidence: binaryReport}, nil
+}

cmd/attestation-service/gpu.go

@@ -0,0 +1,76 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+
+package main
+
+import (
+	"context"
+	"crypto/sha256"
+	"fmt"
+	"strings"
+
+	attestationpb "github.com/ultravioletrs/cocos/internal/proto/attestation/v1"
+	"github.com/ultravioletrs/cocos/pkg/attestation"
+	"github.com/ultravioletrs/cocos/pkg/attestation/eat"
+	attestationgpu "github.com/ultravioletrs/cocos/pkg/attestation/gpu"
+)
+
+func newGPUCollector(cfg config) (attestationgpu.Collector, error) {
+	if strings.TrimSpace(cfg.GPUHelperPath) == "" {
+		return nil, nil
+	}
+
+	return attestationgpu.NewCommandCollector(cfg.GPUHelperPath, cfg.GPUHelperTimeout)
+}
+
+func (s *service) claimOptions(ctx context.Context, req *attestationpb.AttestationRequest, platformType attestation.PlatformType) ([]eat.ClaimsOption, error) {
+	var opts []eat.ClaimsOption
+
+	if s.gpuCollector != nil && shouldCollectGPU(platformType) {
+		sessionNonce := requestNonce(req)
+		gpuNonce := deriveComponentNonce(sessionNonce, "gpu")
+
+		evidence, err := s.gpuCollector.Collect(ctx, gpuNonce)
+		if err != nil {
+			// GPU evidence is opportunistic: if no supported CC-capable GPU is
+			// attached, or the helper cannot collect evidence, we continue with
+			// the root CPU/TEE attestation instead of failing the whole request.
+			s.logger.Warn(fmt.Sprintf("[ATTESTATION-SERVICE] Skipping optional GPU evidence collection: %s", err))
+			return opts, nil
+		}
+
+		s.logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Collected GPU evidence: format=%s bytes=%d",
+			evidence.EvidenceFormat, len(evidence.RawEvidence)))
+
+		opts = append(opts, eat.WithGPU(&eat.GPUExtensions{
+			Vendor:         evidence.Vendor,
+			EvidenceFormat: evidence.EvidenceFormat,
+			Nonce:          gpuNonce,
+			EvidenceJSON:   evidence.RawEvidence,
+		}))
+	}
+
+	return opts, nil
+}
+
+func shouldCollectGPU(platformType attestation.PlatformType) bool {
+	switch platformType {
+	case attestation.SNP, attestation.SNPvTPM, attestation.TDX, attestation.Azure:
+		return true
+	default:
+		return false
+	}
+}
+
+func requestNonce(req *attestationpb.AttestationRequest) []byte {
+	if len(req.Nonce) > 0 {
+		return append([]byte(nil), req.Nonce...)
+	}
+
+	return append([]byte(nil), req.ReportData...)
+}
+
+func deriveComponentNonce(sessionNonce []byte, component string) []byte {
+	digest := sha256.Sum256(append(append([]byte(nil), sessionNonce...), []byte(":"+component)...))
+	return digest[:]
+}

cmd/attestation-service/gpu_test.go

@@ -0,0 +1,83 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+
+package main
+
+import (
+	"context"
+	"io"
+	"log/slog"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	attestationpb "github.com/ultravioletrs/cocos/internal/proto/attestation/v1"
+	"github.com/ultravioletrs/cocos/pkg/attestation"
+	attestationgpu "github.com/ultravioletrs/cocos/pkg/attestation/gpu"
+)
+
+func TestRequestNonce(t *testing.T) {
+	req := &attestationpb.AttestationRequest{
+		ReportData: []byte("report"),
+		Nonce:      []byte("nonce"),
+	}
+
+	assert.Equal(t, []byte("nonce"), requestNonce(req))
+
+	req.Nonce = nil
+	assert.Equal(t, []byte("report"), requestNonce(req))
+}
+
+func TestDeriveComponentNonce(t *testing.T) {
+	sessionNonce := []byte("session-nonce")
+
+	gpuNonce := deriveComponentNonce(sessionNonce, "gpu")
+	gpuNonceAgain := deriveComponentNonce(sessionNonce, "gpu")
+	teeNonce := deriveComponentNonce(sessionNonce, "tee")
+
+	assert.Len(t, gpuNonce, 32)
+	assert.Equal(t, gpuNonce, gpuNonceAgain)
+	assert.NotEqual(t, gpuNonce, teeNonce)
+}
+
+func TestShouldCollectGPU(t *testing.T) {
+	assert.True(t, shouldCollectGPU(attestation.SNP))
+	assert.True(t, shouldCollectGPU(attestation.SNPvTPM))
+	assert.True(t, shouldCollectGPU(attestation.TDX))
+	assert.False(t, shouldCollectGPU(attestation.VTPM))
+	assert.False(t, shouldCollectGPU(attestation.NoCC))
+}
+
+func TestNewGPUCollector(t *testing.T) {
+	collector, err := newGPUCollector(config{})
+	assert.NoError(t, err)
+	assert.Nil(t, collector)
+
+	collector, err = newGPUCollector(config{
+		GPUHelperPath:    "/tmp/helper",
+		GPUHelperTimeout: 0,
+	})
+	assert.NoError(t, err)
+	assert.NotNil(t, collector)
+}
+
+func TestClaimOptions_SkipsOptionalGPUFailure(t *testing.T) {
+	svc := &service{
+		logger:       slog.New(slog.NewTextHandler(io.Discard, nil)),
+		gpuCollector: failingCollector{},
+	}
+
+	req := &attestationpb.AttestationRequest{
+		ReportData: []byte("report-data"),
+		Nonce:      []byte("nonce-data"),
+	}
+
+	opts, err := svc.claimOptions(context.Background(), req, attestation.TDX)
+	assert.NoError(t, err)
+	assert.Empty(t, opts)
+}
+
+type failingCollector struct{}
+
+func (failingCollector) Collect(context.Context, []byte) (*attestationgpu.Evidence, error) {
+	return nil, assert.AnError
+}

cmd/attestation-service/main.go

@@ -0,0 +1,420 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package main
+
+import (
+	"context"
+	"crypto/ecdsa"
+	"fmt"
+	"log/slog"
+	"net"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	mglog "github.com/absmach/magistrala/logger"
+	"github.com/caarlos0/env/v11"
+	"github.com/ultravioletrs/cocos/agent/cvms"
+	logpb "github.com/ultravioletrs/cocos/agent/log"
+	agentlogger "github.com/ultravioletrs/cocos/internal/logger"
+	attestationpb "github.com/ultravioletrs/cocos/internal/proto/attestation/v1"
+	"github.com/ultravioletrs/cocos/pkg/attestation"
+	"github.com/ultravioletrs/cocos/pkg/attestation/azure"
+	"github.com/ultravioletrs/cocos/pkg/attestation/ccaa"
+	"github.com/ultravioletrs/cocos/pkg/attestation/eat"
+	attestationgpu "github.com/ultravioletrs/cocos/pkg/attestation/gpu"
+	"github.com/ultravioletrs/cocos/pkg/attestation/tdx"
+	"github.com/ultravioletrs/cocos/pkg/attestation/vtpm"
+	logclient "github.com/ultravioletrs/cocos/pkg/clients/grpc/log"
+	"golang.org/x/sync/errgroup"
+	"google.golang.org/grpc"
+)
+
+const (
+	svcName    = "attestation-service"
+	socketPath = "/run/cocos/attestation.sock"
+)
+
+type config struct {
+	LogLevel              string        `env:"ATTESTATION_LOG_LEVEL"    envDefault:"debug"`
+	Vmpl                  int           `env:"ATTESTATION_VMPL"         envDefault:"2"`
+	AgentMaaURL           string        `env:"AGENT_MAA_URL"            envDefault:"https://sharedeus2.eus2.attest.azure.net"`
+	AgentOSBuild          string        `env:"AGENT_OS_BUILD"           envDefault:"UVC"`
+	AgentOSDistro         string        `env:"AGENT_OS_DISTRO"          envDefault:"UVC"`
+	AgentOSType           string        `env:"AGENT_OS_TYPE"            envDefault:"UVC"`
+	EATFormat             string        `env:"ATTESTATION_EAT_FORMAT"   envDefault:"CBOR"` // JWT or CBOR
+	EATIssuer             string        `env:"ATTESTATION_EAT_ISSUER"   envDefault:"cocos-attestation-service"`
+	UseCCAttestationAgent bool          `env:"USE_CC_ATTESTATION_AGENT" envDefault:"false"`
+	CCAgentAddress        string        `env:"CC_AGENT_ADDRESS"         envDefault:"127.0.0.1:50002"`
+	GPUHelperPath         string        `env:"ATTESTATION_GPU_HELPER_PATH" envDefault:""`
+	GPUHelperTimeout      time.Duration `env:"ATTESTATION_GPU_HELPER_TIMEOUT" envDefault:"30s"`
+
+	// Future KBS Integration Configuration
+	// When KBS support is added, these fields will enable:
+	// - Remote attestation verification via KBS
+	// - Encrypted algorithm/dataset retrieval
+	// - Per-computation secret provisioning
+	//
+	// Example future fields:
+	// KBSEndpoint   string `env:"KBS_ENDPOINT"            envDefault:""` // Optional KBS URL
+	// KBSEnabled    bool   `env:"KBS_ENABLED"             envDefault:"false"`
+	// KBSTimeout    int    `env:"KBS_TIMEOUT_SECONDS"     envDefault:"30"`
+}
+
+func main() {
+	ctx, cancel := context.WithCancel(context.Background())
+	g, ctx := errgroup.WithContext(ctx)
+
+	var cfg config
+	if err := env.Parse(&cfg); err != nil {
+		fmt.Printf("failed to load %s configuration : %s\n", svcName, err)
+		os.Exit(1)
+	}
+
+	var exitCode int
+	defer mglog.ExitWithError(&exitCode)
+
+	var level slog.Level
+	if err := level.UnmarshalText([]byte(cfg.LogLevel)); err != nil {
+		fmt.Println(err)
+		exitCode = 1
+		return
+	}
+
+	// Setup log forwarding to CVMS (same pattern as agent)
+	logQueue := make(chan *cvms.ClientStreamMessage, 1000)
+	handler := agentlogger.NewProtoHandler(os.Stdout, &slog.HandlerOptions{Level: level}, logQueue)
+	logger := slog.New(handler)
+
+	logger.Info("[ATTESTATION-SERVICE] Starting up - log forwarding enabled")
+
+	// Connect to log client for gRPC forwarding
+	logClient, err := logclient.NewClient("/run/cocos/log.sock")
+	if err != nil {
+		logger.Warn(fmt.Sprintf("failed to create log client: %s. Logging will be local only until service is available.", err))
+	} else {
+		logger.Info("[ATTESTATION-SERVICE] Successfully connected to log client")
+		defer logClient.Close()
+	}
+
+	// Start log forwarding goroutine
+	g.Go(func() error {
+		for {
+			select {
+			case <-ctx.Done():
+				return nil
+			case msg := <-logQueue:
+				if logClient == nil {
+					continue
+				}
+				switch m := msg.Message.(type) {
+				case *cvms.ClientStreamMessage_AgentLog:
+					err := logClient.SendLog(ctx, &logpb.LogEntry{
+						Message:       m.AgentLog.Message,
+						ComputationId: m.AgentLog.ComputationId,
+						Level:         m.AgentLog.Level,
+						Timestamp:     m.AgentLog.Timestamp,
+					})
+					if err != nil {
+						logger.Error("failed to send log", "error", err)
+					}
+				}
+			}
+		}
+	})
+
+	var provider attestation.Provider
+	ccPlatform := attestation.CCPlatform()
+
+	azureConfig := azure.NewEnvConfigFromAgent(
+		cfg.AgentOSBuild,
+		cfg.AgentOSType,
+		cfg.AgentOSDistro,
+		cfg.AgentMaaURL,
+	)
+	azure.InitializeDefaultMAAVars(azureConfig)
+
+	// Try to use CC attestation-agent if configured
+	logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] CC AA configuration: enabled=%v, address=%s", cfg.UseCCAttestationAgent, cfg.CCAgentAddress))
+	if cfg.UseCCAttestationAgent {
+		logger.Info(fmt.Sprintf("attempting to use CC attestation-agent at %s", cfg.CCAgentAddress))
+		ccProvider, err := ccaa.NewProvider(cfg.CCAgentAddress)
+		if err != nil {
+			// For NoCC/sample platform, AA is REQUIRED when configured
+			// Don't fall back to EmptyProvider - AA generates correct KBS format
+			if ccPlatform == attestation.NoCC {
+				logger.Error(fmt.Sprintf("CC AA is required for sample attestation but connection failed: %s", err))
+				exitCode = 1
+				return
+			}
+			logger.Warn(fmt.Sprintf("failed to connect to CC attestation-agent: %s, falling back to direct providers", err))
+		} else {
+			logger.Info("successfully connected to CC attestation-agent")
+			provider = ccProvider
+			defer ccProvider.Close()
+		}
+	}
+
+	isDirectProvider := false
+	// Fallback to direct providers if CC AA not configured or unavailable
+	if provider == nil {
+		isDirectProvider = true
+		switch ccPlatform {
+		case attestation.SNP:
+			provider = vtpm.NewProvider(false, uint(cfg.Vmpl))
+		case attestation.SNPvTPM:
+			provider = vtpm.NewProvider(true, uint(cfg.Vmpl))
+		case attestation.Azure:
+			provider = azure.NewProvider()
+		case attestation.TDX:
+			provider = tdx.NewProvider()
+		case attestation.NoCC:
+			logger.Info("TEE device not found")
+			if cfg.UseCCAttestationAgent {
+				// AA was configured but connection failed - already handled above
+				logger.Error("[ATTESTATION-SERVICE] AA required for sample attestation but not available")
+				exitCode = 1
+				return
+			}
+			// Only use EmptyProvider if AA is explicitly NOT configured
+			logger.Warn("[ATTESTATION-SERVICE] Using EmptyProvider for sample attestation (AA not configured)")
+			provider = &attestation.EmptyProvider{}
+		}
+	}
+
+	// Log which provider is being used
+	if provider != nil {
+		logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Final provider selected: %T", provider))
+	} else {
+		logger.Error("[ATTESTATION-SERVICE] No provider configured!")
+	}
+
+	if (ccPlatform == attestation.SNP || ccPlatform == attestation.SNPvTPM) && isDirectProvider {
+		if err := vtpm.FetchSEVCertificates(uint(cfg.Vmpl)); err != nil {
+			logger.Error(fmt.Sprintf("failed to fetch certificates: %s", err))
+			exitCode = 1
+			return
+		}
+	}
+
+	// Remove existing socket if it exists
+	if _, err := os.Stat(socketPath); err == nil {
+		if err := os.Remove(socketPath); err != nil {
+			logger.Error(fmt.Sprintf("failed to remove existing socket: %s", err))
+			exitCode = 1
+			return
+		}
+	}
+
+	dir := socketPath[:len(socketPath)-len("/attestation.sock")]
+	if err := os.MkdirAll(dir, 0o755); err != nil {
+		logger.Error(fmt.Sprintf("failed to create socket directory: %s", err))
+		exitCode = 1
+		return
+	}
+
+	l, err := net.Listen("unix", socketPath)
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to listen on socket: %s", err))
+		exitCode = 1
+		return
+	}
+
+	if err := os.Chmod(socketPath, 0o777); err != nil {
+		logger.Error(fmt.Sprintf("failed to chmod socket: %s", err))
+		exitCode = 1
+		return
+	}
+
+	// Generate EAT signing key
+	signingKey, err := eat.GenerateSigningKey()
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to generate EAT signing key: %s", err))
+		exitCode = 1
+		return
+	}
+
+	gpuCollector, err := newGPUCollector(cfg)
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to configure GPU attestation collector: %s", err))
+		exitCode = 1
+		return
+	}
+	if gpuCollector != nil {
+		logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] GPU evidence collection enabled via helper %s", cfg.GPUHelperPath))
+	}
+
+	grpcServer := grpc.NewServer()
+	svc := &service{
+		provider:     provider,
+		logger:       logger,
+		signingKey:   signingKey,
+		eatFormat:    cfg.EATFormat,
+		eatIssuer:    cfg.EATIssuer,
+		gpuCollector: gpuCollector,
+	}
+	attestationpb.RegisterAttestationServiceServer(grpcServer, svc)
+
+	g.Go(func() error {
+		ch := make(chan os.Signal, 1)
+		signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM)
+		defer signal.Stop(ch)
+
+		select {
+		case <-ch:
+			logger.Info("Received signal, shutting down...")
+			cancel()
+			grpcServer.GracefulStop()
+			return nil
+		case <-ctx.Done():
+			return ctx.Err()
+		}
+	})
+
+	g.Go(func() error {
+		logger.Info(fmt.Sprintf("%s started on %s", svcName, socketPath))
+		return grpcServer.Serve(l)
+	})
+
+	if err := g.Wait(); err != nil {
+		logger.Error(fmt.Sprintf("%s terminated: %s", svcName, err))
+	}
+}
+
+type service struct {
+	attestationpb.UnimplementedAttestationServiceServer
+	provider     attestation.Provider
+	logger       *slog.Logger
+	signingKey   *ecdsa.PrivateKey
+	eatFormat    string
+	eatIssuer    string
+	gpuCollector attestationgpu.Collector
+}
+
+func (s *service) FetchAttestation(ctx context.Context, req *attestationpb.AttestationRequest) (*attestationpb.AttestationResponse, error) {
+	// Debug: log incoming request
+	s.logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Received attestation request with platform type: %v (%d)",
+		req.PlatformType, req.PlatformType))
+
+	var binaryReport []byte
+	var err error
+	var platformType attestation.PlatformType
+
+	// Get binary attestation report based on platform type
+	switch req.PlatformType {
+	case attestationpb.PlatformType_PLATFORM_TYPE_SNP, attestationpb.PlatformType_PLATFORM_TYPE_TDX:
+		var reportData [64]byte
+		copy(reportData[:], req.ReportData)
+		binaryReport, err = s.provider.TeeAttestation(reportData[:])
+		platformType = convertPlatformType(req.PlatformType)
+	case attestationpb.PlatformType_PLATFORM_TYPE_VTPM:
+		var nonce [32]byte
+		copy(nonce[:], req.Nonce)
+		binaryReport, err = s.provider.VTpmAttestation(nonce[:])
+		platformType = attestation.VTPM
+	case attestationpb.PlatformType_PLATFORM_TYPE_SNP_VTPM:
+		var reportData [64]byte
+		copy(reportData[:], req.ReportData)
+		var nonce [32]byte
+		copy(nonce[:], req.Nonce)
+		binaryReport, err = s.provider.Attestation(reportData[:], nonce[:])
+		platformType = attestation.SNPvTPM
+	case attestationpb.PlatformType_PLATFORM_TYPE_AZURE:
+		var reportData [64]byte
+		copy(reportData[:], req.ReportData)
+		var nonce [32]byte
+		copy(nonce[:], req.Nonce)
+		binaryReport, err = s.provider.Attestation(reportData[:], nonce[:])
+		platformType = attestation.Azure
+	case attestationpb.PlatformType_PLATFORM_TYPE_UNSPECIFIED:
+		// Generate sample attestation for testing in non-TEE environments
+		s.logger.Warn("generating sample attestation for PLATFORM_TYPE_UNSPECIFIED - this should only be used for testing")
+		s.logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Generating sample attestation: reportData_len=%d, nonce_len=%d",
+			len(req.ReportData), len(req.Nonce)))
+
+		// Create a simple sample report that includes the nonce/report data
+		var reportData [64]byte
+		copy(reportData[:], req.ReportData)
+		var nonce [32]byte
+		copy(nonce[:], req.Nonce)
+
+		// Combine report data and nonce into a simple binary report
+		binaryReport = make([]byte, 0, 96)
+		binaryReport = append(binaryReport, reportData[:]...)
+		binaryReport = append(binaryReport, nonce[:]...)
+		platformType = attestation.NoCC
+		s.logger.Info(fmt.Sprintf("[ATTESTATION-SERVICE] Sample attestation generated: binaryReport_len=%d, platformType=%v (%d)",
+			len(binaryReport), platformType, platformType))
+	default:
+		return nil, fmt.Errorf("unsupported platform type")
+	}
+
+	if err != nil {
+		return nil, err
+	}
+
+	// Create EAT claims from binary report
+	nonce := requestNonce(req)
+
+	claimOpts, err := s.claimOptions(ctx, req, platformType)
+	if err != nil {
+		return nil, err
+	}
+
+	claims, err := eat.NewEATClaims(binaryReport, nonce, platformType, claimOpts...)
+	if err != nil {
+		s.logger.Error(fmt.Sprintf("failed to create EAT claims: %s", err))
+		return nil, fmt.Errorf("failed to create EAT claims: %w", err)
+	}
+
+	// Encode to EAT token based on configured format
+	var eatToken []byte
+	switch s.eatFormat {
+	case "JWT":
+		tokenString, err := eat.EncodeToJWT(claims, s.signingKey, s.eatIssuer)
+		if err != nil {
+			return nil, fmt.Errorf("failed to encode JWT: %w", err)
+		}
+		eatToken = []byte(tokenString)
+	case "CBOR":
+		eatToken, err = eat.EncodeToCBOR(claims, s.signingKey, s.eatIssuer)
+		if err != nil {
+			return nil, fmt.Errorf("failed to encode CBOR: %w", err)
+		}
+	default:
+		return nil, fmt.Errorf("unsupported EAT format: %s", s.eatFormat)
+	}
+
+	s.logger.Debug(fmt.Sprintf("generated EAT token (%s format) for platform %v", s.eatFormat, platformType))
+
+	return &attestationpb.AttestationResponse{EatToken: eatToken}, nil
+}
+
+// convertPlatformType converts protobuf platform type to internal platform type.
+func convertPlatformType(pt attestationpb.PlatformType) attestation.PlatformType {
+	switch pt {
+	case attestationpb.PlatformType_PLATFORM_TYPE_SNP:
+		return attestation.SNP
+	case attestationpb.PlatformType_PLATFORM_TYPE_TDX:
+		return attestation.TDX
+	case attestationpb.PlatformType_PLATFORM_TYPE_VTPM:
+		return attestation.VTPM
+	case attestationpb.PlatformType_PLATFORM_TYPE_SNP_VTPM:
+		return attestation.SNPvTPM
+	case attestationpb.PlatformType_PLATFORM_TYPE_AZURE:
+		return attestation.Azure
+	default:
+		return attestation.NoCC
+	}
+}
+
+func (s *service) GetAzureToken(ctx context.Context, req *attestationpb.AzureTokenRequest) (*attestationpb.AzureTokenResponse, error) {
+	var nonce [32]byte
+	copy(nonce[:], req.Nonce)
+	token, err := s.provider.AzureAttestationToken(nonce[:])
+	if err != nil {
+		return nil, err
+	}
+	return &attestationpb.AzureTokenResponse{Token: token}, nil
+}

cmd/cli/main.go

@@ -122,7 +122,7 @@ func main() {
 		defer cliSVC.Close()
 	}
 
-	rootCmd.PersistentFlags().BoolVarP(&cli.Verbose, "verbose", "v", false, "Enable verbose output")
+	rootCmd.PersistentFlags().BoolVarP(&cliSVC.Verbose, "verbose", "v", false, "Enable verbose output")
 
 	keysCmd := cliSVC.NewKeysCmd()
 	attestationCmd := cliSVC.NewAttestationCmd()
@@ -136,7 +136,7 @@ func main() {
 	rootCmd.AddCommand(cliSVC.NewFileHashCmd())
 	rootCmd.AddCommand(attestationPolicyCmd)
 	rootCmd.AddCommand(keysCmd)
-	rootCmd.AddCommand(cliSVC.NewCABundleCmd(directoryCachePath))
+	rootCmd.AddCommand(cliSVC.NewCABundleCmd(directoryCachePath, nil))
 	rootCmd.AddCommand(cliSVC.NewCreateVMCmd())
 	rootCmd.AddCommand(cliSVC.NewRemoveVMCmd())
 	rootCmd.AddCommand(cliSVC.NewIMAMeasurementsCmd())
@@ -151,7 +151,7 @@ func main() {
 
 	// Flags
 	keysCmd.PersistentFlags().StringVarP(
-		&cli.KeyType,
+		&cliSVC.KeyType,
 		"key-type",
 		"k",
 		"rsa",
@@ -159,12 +159,13 @@ func main() {
 	)
 
 	// Attestation Policy commands
-	attestationPolicyCmd.AddCommand(cliSVC.NewAddMeasurementCmd())
-	attestationPolicyCmd.AddCommand(cliSVC.NewAddHostDataCmd())
-	attestationPolicyCmd.AddCommand(cliSVC.NewGCPAttestationPolicy())
+	// Legacy JSON policy commands removed in favor of CoRIM.
+	// attestationPolicyCmd.AddCommand(cliSVC.NewAddMeasurementCmd())
+	// attestationPolicyCmd.AddCommand(cliSVC.NewAddHostDataCmd())
+	// attestationPolicyCmd.AddCommand(cliSVC.NewGCPAttestationPolicy())
 	attestationPolicyCmd.AddCommand(cliSVC.NewDownloadGCPOvmfFile())
-	attestationPolicyCmd.AddCommand(cliSVC.NewAzureAttestationPolicy())
-	attestationPolicyCmd.AddCommand(cliSVC.NewExtendWithManifestCmd())
+	// attestationPolicyCmd.AddCommand(cliSVC.NewAzureAttestationPolicy())
+	// attestationPolicyCmd.AddCommand(cliSVC.NewExtendWithManifestCmd())
 
 	if err := rootCmd.Execute(); err != nil {
 		logErrorCmd(*rootCmd, err)

cmd/computation-runner/main.go

@@ -0,0 +1,153 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package main
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"net"
+	"os"
+	"os/signal"
+	"syscall"
+
+	mglog "github.com/absmach/magistrala/logger"
+	"github.com/caarlos0/env/v11"
+	"github.com/ultravioletrs/cocos/agent/cvms"
+	logpb "github.com/ultravioletrs/cocos/agent/log"
+	pb "github.com/ultravioletrs/cocos/agent/runner"
+	runnerevents "github.com/ultravioletrs/cocos/agent/runner/events"
+	"github.com/ultravioletrs/cocos/agent/runner/service"
+	agentlogger "github.com/ultravioletrs/cocos/internal/logger"
+	logclient "github.com/ultravioletrs/cocos/pkg/clients/grpc/log"
+	"golang.org/x/sync/errgroup"
+	"google.golang.org/grpc"
+)
+
+const (
+	svcName    = "computation-runner"
+	socketPath = "/run/cocos/runner.sock"
+)
+
+type config struct {
+	LogLevel     string `env:"RUNNER_LOG_LEVEL" envAlternate:"AGENT_LOG_LEVEL" envDefault:"debug"`
+	LogForwarder string `env:"LOG_FORWARDER_SOCKET" envDefault:"/run/cocos/log.sock"`
+}
+
+func main() {
+	ctx, cancel := context.WithCancel(context.Background())
+	g, ctx := errgroup.WithContext(ctx)
+
+	var cfg config
+	if err := env.Parse(&cfg); err != nil {
+		fmt.Printf("failed to load %s configuration : %s\n", svcName, err)
+		os.Exit(1)
+	}
+
+	var exitCode int
+	defer mglog.ExitWithError(&exitCode)
+
+	var level slog.Level
+	if err := level.UnmarshalText([]byte(cfg.LogLevel)); err != nil {
+		fmt.Println(err)
+		exitCode = 1
+		return
+	}
+
+	logQueue := make(chan *cvms.ClientStreamMessage, 1000)
+	handler := agentlogger.NewProtoHandler(os.Stdout, &slog.HandlerOptions{Level: level}, logQueue)
+	logger := slog.New(handler)
+
+	// Connect to Log Forwarder
+	logClient, err := logclient.NewClient(cfg.LogForwarder)
+	if err != nil {
+		logger.Warn(fmt.Sprintf("failed to connect to log-forwarder: %s. Logs and events will not be forwarded.", err))
+	} else {
+		defer logClient.Close()
+	}
+
+	g.Go(func() error {
+		for {
+			select {
+			case <-ctx.Done():
+				return nil
+			case msg := <-logQueue:
+				if logClient == nil {
+					continue
+				}
+				switch m := msg.Message.(type) {
+				case *cvms.ClientStreamMessage_AgentLog:
+					err := logClient.SendLog(ctx, &logpb.LogEntry{
+						Message:       m.AgentLog.Message,
+						ComputationId: m.AgentLog.ComputationId,
+						Level:         m.AgentLog.Level,
+						Timestamp:     m.AgentLog.Timestamp,
+					})
+					if err != nil {
+						logger.Error("failed to send log", "error", err)
+					}
+				}
+			}
+		}
+	})
+
+	eventSvc := runnerevents.NewAdapter(logClient, svcName)
+
+	// Remove existing socket if it exists
+	if _, err := os.Stat(socketPath); err == nil {
+		if err := os.Remove(socketPath); err != nil {
+			logger.Error(fmt.Sprintf("failed to remove existing socket: %s", err))
+			exitCode = 1
+			return
+		}
+	}
+
+	dir := socketPath[:len(socketPath)-len("/runner.sock")]
+	if err := os.MkdirAll(dir, 0o755); err != nil {
+		logger.Error(fmt.Sprintf("failed to create socket directory: %s", err))
+		exitCode = 1
+		return
+	}
+
+	lis, err := net.Listen("unix", socketPath)
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to listen on socket: %s", err))
+		exitCode = 1
+		return
+	}
+
+	if err := os.Chmod(socketPath, 0o777); err != nil {
+		logger.Error(fmt.Sprintf("failed to chmod socket: %s", err))
+		exitCode = 1
+		return
+	}
+
+	grpcServer := grpc.NewServer()
+	svc := service.New(logger, eventSvc)
+	pb.RegisterComputationRunnerServer(grpcServer, svc)
+
+	g.Go(func() error {
+		ch := make(chan os.Signal, 1)
+		signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM)
+		defer signal.Stop(ch)
+
+		select {
+		case <-ch:
+			logger.Info("Received signal, shutting down...")
+			cancel()
+			grpcServer.GracefulStop()
+			return nil
+		case <-ctx.Done():
+			return ctx.Err()
+		}
+	})
+
+	g.Go(func() error {
+		logger.Info(fmt.Sprintf("%s started on %s", svcName, socketPath))
+		return grpcServer.Serve(lis)
+	})
+
+	if err := g.Wait(); err != nil {
+		logger.Error(fmt.Sprintf("%s terminated: %s", svcName, err))
+	}
+}

cmd/egress-proxy/main.go

@@ -0,0 +1,129 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package main
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/caarlos0/env/v11"
+	"github.com/spf13/cobra"
+	"github.com/spf13/pflag"
+	"github.com/ultravioletrs/cocos/agent/cvms"
+	logpb "github.com/ultravioletrs/cocos/agent/log"
+	agentlogger "github.com/ultravioletrs/cocos/internal/logger"
+	logclient "github.com/ultravioletrs/cocos/pkg/clients/grpc/log"
+	"github.com/ultravioletrs/cocos/pkg/egress"
+	"golang.org/x/sync/errgroup"
+)
+
+const (
+	svcName = "egress-proxy"
+)
+
+type config struct {
+	Level        string `env:"COCOS_LOG_LEVEL" envAlternate:"AGENT_LOG_LEVEL" envDefault:"info"`
+	Port         string `env:"COCOS_PROXY_PORT"                envDefault:"3128"`
+	LogForwarder string `env:"LOG_FORWARDER_SOCKET"            envDefault:"/run/cocos/log.sock"`
+}
+
+func main() {
+	var cfg config
+	if err := env.Parse(&cfg); err != nil {
+		fmt.Fprintf(os.Stderr, "failed to load configuration: %s\n", err)
+		os.Exit(1)
+	}
+
+	cmd := &cobra.Command{
+		Use:   svcName,
+		Short: "Egress Proxy Service",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return run(cfg)
+		},
+	}
+
+	pflag.StringVar(&cfg.Level, "log-level", cfg.Level, "Log level")
+	pflag.StringVar(&cfg.Port, "port", cfg.Port, "Proxy port")
+
+	if err := cmd.Execute(); err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %s\n", err)
+		os.Exit(1)
+	}
+}
+
+func run(cfg config) error {
+	var level slog.Level
+	if err := level.UnmarshalText([]byte(cfg.Level)); err != nil {
+		return fmt.Errorf("invalid log level: %w", err)
+	}
+
+	logQueue := make(chan *cvms.ClientStreamMessage, 1000)
+	handler := agentlogger.NewProtoHandler(os.Stdout, &slog.HandlerOptions{Level: level}, logQueue)
+	logger := slog.New(handler)
+
+	logClient, err := logclient.NewClient(cfg.LogForwarder)
+	if err != nil {
+		logger.Warn(fmt.Sprintf("failed to connect to log-forwarder: %s. Logs will not be forwarded.", err))
+	} else {
+		defer logClient.Close()
+	}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	g, ctx := errgroup.WithContext(ctx)
+
+	g.Go(func() error {
+		for {
+			select {
+			case <-ctx.Done():
+				return nil
+			case msg := <-logQueue:
+				if logClient == nil {
+					continue
+				}
+				switch m := msg.Message.(type) {
+				case *cvms.ClientStreamMessage_AgentLog:
+					err := logClient.SendLog(ctx, &logpb.LogEntry{
+						Message:       m.AgentLog.Message,
+						ComputationId: m.AgentLog.ComputationId,
+						Level:         m.AgentLog.Level,
+						Timestamp:     m.AgentLog.Timestamp,
+					})
+					if err != nil {
+						logger.Error("failed to send log", "error", err)
+					}
+				}
+			}
+		}
+	})
+
+	proxy := egress.NewProxy(logger, ":"+cfg.Port)
+
+	g.Go(func() error {
+		return proxy.Start()
+	})
+
+	g.Go(func() error {
+		c := make(chan os.Signal, 1)
+		signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
+		select {
+		case s := <-c:
+			logger.Info(fmt.Sprintf("received signal %s, stopping", s))
+			cancel()
+			return proxy.Stop(ctx)
+		case <-ctx.Done():
+			return nil
+		}
+	})
+
+	if err := g.Wait(); err != nil {
+		return fmt.Errorf("server exit with error: %w", err)
+	}
+
+	return nil
+}

cmd/ingress-proxy/main.go

@@ -0,0 +1,185 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package main
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"net/url"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/absmach/certs/sdk"
+	"github.com/caarlos0/env/v11"
+	"github.com/spf13/cobra"
+	"github.com/spf13/pflag"
+	"github.com/ultravioletrs/cocos/agent/cvms"
+	logpb "github.com/ultravioletrs/cocos/agent/log"
+	agentlogger "github.com/ultravioletrs/cocos/internal/logger"
+	"github.com/ultravioletrs/cocos/pkg/atls"
+	"github.com/ultravioletrs/cocos/pkg/attestation"
+	"github.com/ultravioletrs/cocos/pkg/attestation/azure"
+	attestation_client "github.com/ultravioletrs/cocos/pkg/clients/grpc/attestation"
+	logclient "github.com/ultravioletrs/cocos/pkg/clients/grpc/log"
+	"github.com/ultravioletrs/cocos/pkg/ingress"
+	"golang.org/x/sync/errgroup"
+)
+
+const (
+	svcName = "ingress-proxy"
+)
+
+type config struct {
+	LogLevel string `env:"COCOS_LOG_LEVEL" envAlternate:"AGENT_LOG_LEVEL" envDefault:"info"`
+	Backend  string `env:"COCOS_INGRESS_BACKEND"           envDefault:"http://localhost:7001"`
+
+	// ATLS Config
+	CAUrl         string `env:"AGENT_CVM_CA_URL"             envDefault:""`
+	CVMId         string `env:"AGENT_CVM_ID"                 envDefault:""`
+	CertsToken    string `env:"AGENT_CERTS_TOKEN"            envDefault:""`
+	AgentMaaURL   string `env:"AGENT_MAA_URL"                envDefault:"https://sharedeus2.eus2.attest.azure.net"`
+	AgentOSBuild  string `env:"AGENT_OS_BUILD"               envDefault:"UVC"`
+	AgentOSDistro string `env:"AGENT_OS_DISTRO"              envDefault:"UVC"`
+	AgentOSType   string `env:"AGENT_OS_TYPE"                envDefault:"UVC"`
+	LogForwarder  string `env:"LOG_FORWARDER_SOCKET"         envDefault:"/run/cocos/log.sock"`
+}
+
+func main() {
+	var cfg config
+	if err := env.Parse(&cfg); err != nil {
+		fmt.Fprintf(os.Stderr, "failed to load configuration: %s\n", err)
+		os.Exit(1)
+	}
+
+	cmd := &cobra.Command{
+		Use:   svcName,
+		Short: "Ingress Proxy Service",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return run(cfg)
+		},
+	}
+
+	pflag.StringVar(&cfg.LogLevel, "log-level", cfg.LogLevel, "Log level")
+	pflag.StringVar(&cfg.Backend, "backend", cfg.Backend, "Backend URL")
+
+	if err := cmd.Execute(); err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %s\n", err)
+		os.Exit(1)
+	}
+}
+
+func run(cfg config) error {
+	var level slog.Level
+	if err := level.UnmarshalText([]byte(cfg.LogLevel)); err != nil {
+		return fmt.Errorf("invalid log level: %w", err)
+	}
+
+	logQueue := make(chan *cvms.ClientStreamMessage, 1000)
+	handler := agentlogger.NewProtoHandler(os.Stdout, &slog.HandlerOptions{Level: level}, logQueue)
+	logger := slog.New(handler)
+
+	logClient, err := logclient.NewClient(cfg.LogForwarder)
+	if err != nil {
+		logger.Warn(fmt.Sprintf("failed to connect to log-forwarder: %s. Logs will not be forwarded.", err))
+	} else {
+		defer logClient.Close()
+	}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	g, ctx := errgroup.WithContext(ctx)
+
+	g.Go(func() error {
+		for {
+			select {
+			case <-ctx.Done():
+				return nil
+			case msg := <-logQueue:
+				if logClient == nil {
+					continue
+				}
+				switch m := msg.Message.(type) {
+				case *cvms.ClientStreamMessage_AgentLog:
+					err := logClient.SendLog(ctx, &logpb.LogEntry{
+						Message:       m.AgentLog.Message,
+						ComputationId: m.AgentLog.ComputationId,
+						Level:         m.AgentLog.Level,
+						Timestamp:     m.AgentLog.Timestamp,
+					})
+					if err != nil {
+						logger.Error("failed to send log", "error", err)
+					}
+				}
+			}
+		}
+	})
+
+	backendURL, err := url.Parse(cfg.Backend)
+	if err != nil {
+		return fmt.Errorf("failed to parse backend URL: %w", err)
+	}
+
+	// Initialize Certificate Provider
+	ccPlatform := attestation.CCPlatform()
+
+	azureConfig := azure.NewEnvConfigFromAgent(
+		cfg.AgentOSBuild,
+		cfg.AgentOSType,
+		cfg.AgentOSDistro,
+		cfg.AgentMaaURL,
+	)
+	azure.InitializeDefaultMAAVars(azureConfig)
+
+	var certProvider atls.CertificateProvider
+
+	if ccPlatform != attestation.NoCC {
+		// Create attestation client
+		attClient, err := attestation_client.NewClient("/run/cocos/attestation.sock")
+		if err != nil {
+			return fmt.Errorf("failed to create attestation client: %w", err)
+		}
+		defer attClient.Close()
+
+		var certsSDK sdk.SDK
+		if cfg.CAUrl != "" {
+			certsSDK = sdk.NewSDK(sdk.Config{
+				CertsURL: cfg.CAUrl,
+			})
+		}
+		certProvider, err = atls.NewProvider(attClient, ccPlatform, cfg.CertsToken, cfg.CVMId, certsSDK)
+		if err != nil {
+			return fmt.Errorf("failed to create certificate provider: %w", err)
+		}
+	} else {
+		logger.Warn("No Confidential Computing platform detected. ATLS will not be available.")
+	}
+
+	// Create proxy server (but don't start it yet - it will be started per-computation)
+	_ = ingress.NewProxyServer(logger, backendURL, certProvider)
+
+	// Note: The proxy server will be started dynamically when a computation is initiated
+	// via the Manager's ComputationRunReq message. For now, we just keep the service alive.
+	logger.Info("ingress-proxy service initialized, waiting for computation requests...")
+
+	g.Go(func() error {
+		c := make(chan os.Signal, 1)
+		signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
+		select {
+		case s := <-c:
+			logger.Info(fmt.Sprintf("received signal %s, stopping", s))
+			cancel()
+			return nil
+		case <-ctx.Done():
+			return nil
+		}
+	})
+
+	if err := g.Wait(); err != nil {
+		return fmt.Errorf("server exit with error: %w", err)
+	}
+
+	return nil
+}

cmd/log-forwarder/main.go

@@ -0,0 +1,155 @@
+// Copyright (c) Ultraviolet
+// SPDX-License-Identifier: Apache-2.0
+package main
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"net"
+	"os"
+	"os/signal"
+	"syscall"
+
+	mglog "github.com/absmach/magistrala/logger"
+	"github.com/caarlos0/env/v11"
+	"github.com/ultravioletrs/cocos/agent/cvms"
+	pb "github.com/ultravioletrs/cocos/agent/log"
+	"github.com/ultravioletrs/cocos/agent/log/service"
+	"github.com/ultravioletrs/cocos/pkg/clients"
+	cvmsgrpc "github.com/ultravioletrs/cocos/pkg/clients/grpc/cvm"
+	"golang.org/x/sync/errgroup"
+	"google.golang.org/grpc"
+)
+
+const (
+	svcName          = "log-forwarder"
+	socketPath       = "/run/cocos/log.sock"
+	envPrefixCVMGRPC = "AGENT_CVM_GRPC_"
+)
+
+type config struct {
+	LogLevel string `env:"LOG_FORWARDER_LOG_LEVEL" envAlternate:"AGENT_LOG_LEVEL" envDefault:"debug"`
+}
+
+func main() {
+	ctx, cancel := context.WithCancel(context.Background())
+	g, ctx := errgroup.WithContext(ctx)
+
+	var cfg config
+	if err := env.Parse(&cfg); err != nil {
+		fmt.Printf("failed to load %s configuration : %s\n", svcName, err)
+		os.Exit(1)
+	}
+
+	var exitCode int
+	defer mglog.ExitWithError(&exitCode)
+
+	var level slog.Level
+	if err := level.UnmarshalText([]byte(cfg.LogLevel)); err != nil {
+		fmt.Println(err)
+		exitCode = 1
+		return
+	}
+
+	logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{Level: level}))
+
+	// Remove existing socket if it exists
+	if _, err := os.Stat(socketPath); err == nil {
+		if err := os.Remove(socketPath); err != nil {
+			logger.Error(fmt.Sprintf("failed to remove existing socket: %s", err))
+			exitCode = 1
+			return
+		}
+	}
+
+	dir := socketPath[:len(socketPath)-len("/log.sock")]
+	if err := os.MkdirAll(dir, 0o755); err != nil {
+		logger.Error(fmt.Sprintf("failed to create socket directory: %s", err))
+		exitCode = 1
+		return
+	}
+
+	lis, err := net.Listen("unix", socketPath)
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to listen on socket: %s", err))
+		exitCode = 1
+		return
+	}
+
+	if err := os.Chmod(socketPath, 0o777); err != nil {
+		logger.Error(fmt.Sprintf("failed to chmod socket: %s", err))
+		exitCode = 1
+		return
+	}
+
+	// Connect to Manager
+	cvmGrpcConfig := clients.StandardClientConfig{}
+	if err := env.ParseWithOptions(&cvmGrpcConfig, env.Options{Prefix: envPrefixCVMGRPC}); err != nil {
+		logger.Error(fmt.Sprintf("failed to load %s gRPC client configuration : %s", svcName, err))
+		exitCode = 1
+		return
+	}
+
+	cvmClient, cvmsClient, err := cvmsgrpc.NewCVMClient(cvmGrpcConfig)
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to connect to CVM manager: %s", err))
+		exitCode = 1
+		return
+	}
+	defer cvmClient.Close()
+
+	// Create stream to Manager
+	stream, err := cvmsClient.Process(ctx)
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to create stream to manager: %s", err))
+		exitCode = 1
+		return
+	}
+
+	logQueue := make(chan *cvms.ClientStreamMessage, 1000)
+
+	grpcServer := grpc.NewServer()
+	svc := service.New(logger, cvmsClient, logQueue)
+	pb.RegisterLogCollectorServer(grpcServer, svc)
+
+	// Log Consumer Goroutine
+	g.Go(func() error {
+		for {
+			select {
+			case <-ctx.Done():
+				return nil
+			case msg := <-logQueue:
+				if err := stream.Send(msg); err != nil {
+					logger.Error(fmt.Sprintf("failed to send log to manager: %s", err))
+					// Reconnect logic would go here
+				}
+			}
+		}
+	})
+
+	g.Go(func() error {
+		ch := make(chan os.Signal, 1)
+		signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM)
+		defer signal.Stop(ch)
+
+		select {
+		case <-ch:
+			logger.Info("Received signal, shutting down...")
+			cancel()
+			grpcServer.GracefulStop()
+			return nil
+		case <-ctx.Done():
+			return ctx.Err()
+		}
+	})
+
+	g.Go(func() error {
+		logger.Info(fmt.Sprintf("%s started on %s", svcName, socketPath))
+		return grpcServer.Serve(lis)
+	})
+
+	if err := g.Wait(); err != nil {
+		logger.Error(fmt.Sprintf("%s terminated: %s", svcName, err))
+	}
+}

cmd/manager/main.go

@@ -12,12 +12,13 @@ import (
 	"os"
 	"strings"
 
-	mglog "github.com/absmach/supermq/logger"
-	"github.com/absmach/supermq/pkg/jaeger"
-	"github.com/absmach/supermq/pkg/prometheus"
-	smqserver "github.com/absmach/supermq/pkg/server"
-	httpserver "github.com/absmach/supermq/pkg/server/http"
-	"github.com/absmach/supermq/pkg/uuid"
+	mglog "github.com/absmach/magistrala/logger"
+	"github.com/absmach/magistrala/pkg/jaeger"
+	"github.com/absmach/magistrala/pkg/prometheus"
+	smqserver "github.com/absmach/magistrala/pkg/server"
+	grpcserver "github.com/absmach/magistrala/pkg/server/grpc"
+	httpserver "github.com/absmach/magistrala/pkg/server/http"
+	"github.com/absmach/magistrala/pkg/uuid"
 	"github.com/caarlos0/env/v11"
 	"github.com/go-chi/chi/v5"
 	"github.com/ultravioletrs/cocos/manager"
@@ -26,8 +27,6 @@ import (
 	"github.com/ultravioletrs/cocos/manager/api/http"
 	"github.com/ultravioletrs/cocos/manager/qemu"
 	"github.com/ultravioletrs/cocos/manager/tracing"
-	"github.com/ultravioletrs/cocos/pkg/server"
-	grpcserver "github.com/ultravioletrs/cocos/pkg/server/grpc"
 	"go.opentelemetry.io/otel/trace"
 	"golang.org/x/sync/errgroup"
 	"google.golang.org/grpc"
@@ -43,15 +42,15 @@ 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:"COCOS_JAEGER_TRACE_RATIO"           envDefault:"1.0"`
-	InstanceID              string  `env:"MANAGER_INSTANCE_ID"                envDefault:""`
-	AttestationPolicyBinary string  `env:"MANAGER_ATTESTATION_POLICY_BINARY"  envDefault:"../../build/attestation_policy"`
-	IgvmMeasureBinary       string  `env:"MANAGER_IGVMMEASURE_BINARY"         envDefault:"../../build/igvmmeasure"`
-	PcrValues               string  `env:"MANAGER_PCR_VALUES"                 envDefault:""`
-	EosVersion              string  `env:"MANAGER_EOS_VERSION"                envDefault:""`
-	MaxVMs                  int     `env:"MANAGER_MAX_VMS"                    envDefault:"10"`
+	LogLevel                    string  `env:"MANAGER_LOG_LEVEL"                  envDefault:"info"`
+	JaegerURL                   url.URL `env:"COCOS_JAEGER_URL"                   envDefault:"http://localhost:4318"`
+	TraceRatio                  float64 `env:"COCOS_JAEGER_TRACE_RATIO"           envDefault:"1.0"`
+	InstanceID                  string  `env:"MANAGER_INSTANCE_ID"                envDefault:""`
+	AttestationPolicyBinaryPath string  `env:"MANAGER_ATTESTATION_POLICY_BINARY_PATH" envDefault:"../../build"`
+	PcrValues                   string  `env:"MANAGER_PCR_VALUES"                 envDefault:""`
+	EosVersion                  string  `env:"MANAGER_EOS_VERSION"                envDefault:""`
+	MaxVMs                      int     `env:"MANAGER_MAX_VMS"                    envDefault:"10"`
+	SigningKeyPath              string  `env:"MANAGER_CORIM_SIGNING_KEY"          envDefault:""`
 }
 
 func main() {
@@ -113,7 +112,7 @@ func main() {
 	args := qemuCfg.ConstructQemuArgs()
 	logger.Info(strings.Join(args, " "))
 
-	managerGRPCConfig := server.ServerConfig{}
+	managerGRPCConfig := smqserver.Config{}
 	if err := env.ParseWithOptions(&managerGRPCConfig, env.Options{Prefix: envPrefixGRPC}); err != nil {
 		logger.Error(fmt.Sprintf("failed to load %s gRPC client configuration : %s", svcName, err))
 		exitCode = 1
@@ -125,7 +124,7 @@ func main() {
 		logger.Error(fmt.Sprintf("failed to load %s gRPC server configuration : %s", svcName, err))
 	}
 
-	svc, err := newService(logger, tracer, *qemuCfg, cfg.AttestationPolicyBinary, cfg.IgvmMeasureBinary, cfg.PcrValues, cfg.EosVersion, cfg.MaxVMs)
+	svc, err := newService(logger, tracer, *qemuCfg, cfg.AttestationPolicyBinaryPath, cfg.PcrValues, cfg.SigningKeyPath, cfg.EosVersion, cfg.MaxVMs)
 	if err != nil {
 		logger.Error(err.Error())
 		exitCode = 1
@@ -145,7 +144,7 @@ func main() {
 		manager.RegisterManagerServiceServer(srv, managergrpc.NewServer(svc))
 	}
 
-	gs := grpcserver.New(ctx, cancel, svcName, managerGRPCConfig, registerManagerServiceServer, logger, nil, nil)
+	gs := grpcserver.NewServer(ctx, cancel, svcName, managerGRPCConfig, registerManagerServiceServer, logger)
 
 	hs := httpserver.NewServer(ctx, cancel, svcName, httpServerConfig, http.MakeHandler(chi.NewMux(), svcName, cfg.InstanceID), logger)
 
@@ -158,7 +157,7 @@ func main() {
 	})
 
 	g.Go(func() error {
-		return server.StopHandler(ctx, cancel, logger, svcName, gs, hs)
+		return smqserver.StopSignalHandler(ctx, cancel, logger, svcName, gs, hs)
 	})
 
 	if err := g.Wait(); err != nil {
@@ -166,8 +165,8 @@ func main() {
 	}
 }
 
-func newService(logger *slog.Logger, tracer trace.Tracer, qemuCfg qemu.Config, attestationPolicyPath string, igvmMeasurementBinaryPath string, pcrValuesFilePath string, eosVersion string, maxVMs int) (manager.Service, error) {
-	svc, err := manager.New(qemuCfg, attestationPolicyPath, igvmMeasurementBinaryPath, pcrValuesFilePath, logger, qemu.NewVM, eosVersion, maxVMs)
+func newService(logger *slog.Logger, tracer trace.Tracer, qemuCfg qemu.Config, attestationPolicyBinaryPath string, pcrValuesFilePath string, signingKeyPath string, eosVersion string, maxVMs int) (manager.Service, error) {
+	svc, err := manager.New(qemuCfg, attestationPolicyBinaryPath, pcrValuesFilePath, signingKeyPath, logger, qemu.NewVM, eosVersion, maxVMs)
 	if err != nil {
 		return nil, err
 	}

go.mod

@@ -1,68 +1,81 @@
 module github.com/ultravioletrs/cocos
 
-go 1.25.3
+go 1.26.0
 
 require (
-	github.com/caarlos0/env/v11 v11.3.1
-	github.com/fatih/color v1.18.0
+	github.com/caarlos0/env/v11 v11.4.0
+	github.com/fatih/color v1.19.0
 	github.com/go-kit/kit v0.13.0
 	github.com/gofrs/uuid v4.4.0+incompatible
-	github.com/google/go-sev-guest v0.13.0
-	github.com/google/go-tdx-guest v0.3.2-0.20241009005452-097ee70d0843
-	github.com/spf13/cobra v1.10.1
+	github.com/google/go-sev-guest v0.14.1
+	github.com/google/go-tdx-guest v0.3.2-0.20260605221019-34f07ec666c4
+	github.com/spf13/cobra v1.10.2
 	github.com/spf13/pflag v1.0.10
 	github.com/stretchr/testify v1.11.1
 	github.com/virtee/sev-snp-measure-go v0.0.0-20240530153610-e6e8dc9b6877
-	go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.63.0
-	go.opentelemetry.io/otel/trace v1.38.0
-	golang.org/x/crypto v0.43.0
-	golang.org/x/sync v0.17.0
-	google.golang.org/grpc v1.76.0
-	google.golang.org/protobuf v1.36.10
+	go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.68.0
+	go.opentelemetry.io/otel/trace v1.43.0
+	golang.org/x/crypto v0.50.0
+	golang.org/x/sync v0.20.0
+	google.golang.org/grpc v1.80.0
+	google.golang.org/protobuf v1.36.11
 )
 
 require (
-	cloud.google.com/go/storage v1.57.0
-	github.com/absmach/supermq v0.18.2
+	cloud.google.com/go/storage v1.62.3
+	github.com/absmach/magistrala v0.20.0
 	github.com/caarlos0/env/v10 v10.0.0
-	github.com/go-chi/chi/v5 v5.2.3
+	github.com/fxamacker/cbor/v2 v2.9.0
+	github.com/go-chi/chi/v5 v5.2.5
+	github.com/go-jose/go-jose/v4 v4.1.4
 	github.com/golang-jwt/jwt/v5 v5.3.0
 	github.com/google/gce-tcb-verifier v0.3.1
+	github.com/veraison/corim v1.1.2
+	github.com/veraison/go-cose v1.3.0
+	github.com/veraison/swid v1.1.1-0.20230911094910-8ffdd07a22ca
+	google.golang.org/api v0.274.0
 )
 
 require (
-	cel.dev/expr v0.24.0 // indirect
-	cloud.google.com/go v0.121.6 // indirect
-	cloud.google.com/go/auth v0.16.5 // indirect
+	cel.dev/expr v0.25.1 // indirect
+	cloud.google.com/go v0.123.0 // indirect
+	cloud.google.com/go/auth v0.19.0 // indirect
 	cloud.google.com/go/auth/oauth2adapt v0.2.8 // indirect
 	cloud.google.com/go/compute/metadata v0.9.0 // indirect
-	cloud.google.com/go/iam v1.5.2 // indirect
-	cloud.google.com/go/monitoring v1.24.2 // indirect
-	github.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.29.0 // indirect
-	github.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/metric v0.53.0 // indirect
-	github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/resourcemapping v0.53.0 // indirect
+	cloud.google.com/go/iam v1.7.0 // indirect
+	cloud.google.com/go/monitoring v1.24.3 // indirect
+	github.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.31.0 // indirect
+	github.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/metric v0.55.0 // indirect
+	github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/resourcemapping v0.55.0 // indirect
 	github.com/Microsoft/go-winio v0.6.2 // indirect
+	github.com/absmach/supermq v0.19.2-0.20260317185610-fade98b84ee4 // indirect
 	github.com/cenkalti/backoff/v5 v5.0.3 // indirect
-	github.com/cncf/xds/go v0.0.0-20250501225837-2ac532fd4443 // indirect
+	github.com/cncf/xds/go v0.0.0-20251210132809-ee656c7534f5 // indirect
 	github.com/containerd/errdefs v1.0.0 // indirect
 	github.com/containerd/errdefs/pkg v0.3.0 // indirect
 	github.com/containerd/log v0.1.0 // indirect
+	github.com/decred/dcrd/dcrec/secp256k1/v4 v4.4.0 // indirect
 	github.com/distribution/reference v0.6.0 // indirect
-	github.com/docker/go-connections v0.5.0 // indirect
+	github.com/docker/go-connections v0.6.0 // indirect
 	github.com/docker/go-units v0.5.0 // indirect
-	github.com/envoyproxy/go-control-plane/envoy v1.32.4 // indirect
-	github.com/envoyproxy/protoc-gen-validate v1.2.1 // indirect
+	github.com/envoyproxy/go-control-plane/envoy v1.36.0 // indirect
+	github.com/envoyproxy/protoc-gen-validate v1.3.0 // indirect
 	github.com/felixge/httpsnoop v1.0.4 // indirect
 	github.com/go-jose/go-jose/v3 v3.0.4 // indirect
-	github.com/go-jose/go-jose/v4 v4.1.2 // indirect
-	github.com/gofrs/uuid/v5 v5.3.2 // indirect
+	github.com/goccy/go-json v0.10.5 // indirect
+	github.com/gofrs/uuid/v5 v5.4.0 // indirect
 	github.com/google/certificate-transparency-go v1.1.8 // indirect
 	github.com/google/go-attestation v0.5.1 // indirect
-	github.com/google/go-eventlog v0.0.2-0.20241003021507-01bb555f7cba // indirect
 	github.com/google/go-tspi v0.3.0 // indirect
 	github.com/google/s2a-go v0.1.9 // indirect
-	github.com/googleapis/enterprise-certificate-proxy v0.3.6 // indirect
-	github.com/googleapis/gax-go/v2 v2.15.0 // indirect
+	github.com/googleapis/enterprise-certificate-proxy v0.3.14 // indirect
+	github.com/googleapis/gax-go/v2 v2.21.0 // indirect
+	github.com/lestrrat-go/blackmagic v1.0.4 // indirect
+	github.com/lestrrat-go/httpcc v1.0.1 // indirect
+	github.com/lestrrat-go/httprc v1.0.6 // indirect
+	github.com/lestrrat-go/iter v1.0.2 // indirect
+	github.com/lestrrat-go/jwx/v2 v2.1.6 // indirect
+	github.com/lestrrat-go/option v1.0.1 // indirect
 	github.com/mattn/go-colorable v0.1.14 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
 	github.com/moby/docker-image-spec v1.3.1 // indirect
@@ -70,61 +83,63 @@ require (
 	github.com/morikuni/aec v1.0.0 // indirect
 	github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
 	github.com/opencontainers/go-digest v1.0.0 // indirect
-	github.com/opencontainers/image-spec v1.1.0 // indirect
+	github.com/opencontainers/image-spec v1.1.1 // indirect
 	github.com/planetscale/vtprotobuf v0.6.1-0.20240917153116-6f2963f01587 // indirect
-	github.com/spiffe/go-spiffe/v2 v2.5.0 // indirect
-	github.com/zeebo/errs v1.4.0 // indirect
+	github.com/segmentio/asm v1.2.1 // indirect
+	github.com/spf13/cast v1.10.0 // indirect
+	github.com/spiffe/go-spiffe/v2 v2.6.0 // indirect
+	github.com/veraison/eat v0.0.0-20210331113810-3da8a4dd42ff // indirect
+	github.com/x448/float16 v0.8.4 // indirect
 	go.opentelemetry.io/auto/sdk v1.2.1 // indirect
-	go.opentelemetry.io/contrib/detectors/gcp v1.36.0 // indirect
-	go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.63.0 // indirect
-	go.opentelemetry.io/otel v1.38.0 // indirect
-	go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.38.0 // indirect
-	go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.38.0 // indirect
-	go.opentelemetry.io/otel/sdk v1.38.0 // indirect
-	go.opentelemetry.io/otel/sdk/metric v1.38.0 // indirect
+	go.opentelemetry.io/contrib/detectors/gcp v1.39.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.68.0 // indirect
+	go.opentelemetry.io/otel v1.43.0 // indirect
+	go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.43.0 // indirect
+	go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.43.0 // indirect
+	go.opentelemetry.io/otel/sdk v1.43.0 // indirect
+	go.opentelemetry.io/otel/sdk/metric v1.43.0 // indirect
 	go.yaml.in/yaml/v2 v2.4.3 // indirect
-	golang.org/x/oauth2 v0.32.0 // indirect
-	golang.org/x/time v0.12.0 // indirect
-	google.golang.org/api v0.247.0 // indirect
-	google.golang.org/genproto v0.0.0-20250603155806-513f23925822 // indirect
+	golang.org/x/oauth2 v0.36.0 // indirect
+	golang.org/x/time v0.15.0 // indirect
+	google.golang.org/genproto v0.0.0-20260319201613-d00831a3d3e7 // indirect
 	gotest.tools/v3 v3.5.1 // indirect
 	moul.io/http2curl v1.0.0 // indirect
 )
 
 require (
-	github.com/absmach/certs v0.18.2
+	github.com/absmach/certs v0.18.5
 	github.com/beorn7/perks v1.0.1 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
-	github.com/docker/docker v28.5.1+incompatible
+	github.com/docker/docker v28.5.2+incompatible
 	github.com/edgelesssys/go-azguestattestation v0.0.0-20250408071817-8c4457b235ff
 	github.com/go-kit/log v0.2.1 // indirect
 	github.com/go-logfmt/logfmt v0.6.1 // indirect
 	github.com/go-logr/logr v1.4.3 // indirect
 	github.com/go-logr/stdr v1.2.2 // indirect
 	github.com/google/go-configfs-tsm v0.3.3-0.20240919001351-b4b5b84fdcbc // indirect
-	github.com/google/go-tpm v0.9.6
-	github.com/google/go-tpm-tools v0.4.6
-	github.com/google/logger v1.1.1
+	github.com/google/go-tpm v0.9.8
+	github.com/google/go-tpm-tools v0.4.4
+	github.com/google/logger v1.1.1 // indirect
 	github.com/google/uuid v1.6.0
-	github.com/grpc-ecosystem/grpc-gateway/v2 v2.27.3 // indirect
+	github.com/grpc-ecosystem/grpc-gateway/v2 v2.28.0 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
 	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
 	github.com/prometheus/client_golang v1.23.2
 	github.com/prometheus/client_model v0.6.2 // indirect
-	github.com/prometheus/common v0.67.1 // indirect
-	github.com/prometheus/procfs v0.17.0 // indirect
+	github.com/prometheus/common v0.67.5 // indirect
+	github.com/prometheus/procfs v0.19.2 // indirect
 	github.com/stretchr/objx v0.5.3 // indirect
-	go.opentelemetry.io/otel/metric v1.38.0 // indirect
-	go.opentelemetry.io/proto/otlp v1.8.0 // indirect
+	go.opentelemetry.io/otel/metric v1.43.0 // indirect
+	go.opentelemetry.io/proto/otlp v1.10.0 // indirect
 	go.uber.org/multierr v1.11.0 // indirect
-	golang.org/x/net v0.46.0 // indirect
-	golang.org/x/sys v0.37.0 // indirect
-	golang.org/x/term v0.36.0
-	golang.org/x/text v0.30.0 // indirect
-	google.golang.org/genproto/googleapis/api v0.0.0-20251014184007-4626949a642f // indirect
-	google.golang.org/genproto/googleapis/rpc v0.0.0-20251014184007-4626949a642f // indirect
+	golang.org/x/net v0.53.0
+	golang.org/x/sys v0.43.0 // indirect
+	golang.org/x/term v0.42.0
+	golang.org/x/text v0.36.0 // indirect
+	google.golang.org/genproto/googleapis/api v0.0.0-20260401024825-9d38bb4040a9 // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20260406210006-6f92a3bedf2d // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
 

go.sum

@@ -1,55 +1,57 @@
-cel.dev/expr v0.24.0 h1:56OvJKSH3hDGL0ml5uSxZmz3/3Pq4tJ+fb1unVLAFcY=
-cel.dev/expr v0.24.0/go.mod h1:hLPLo1W4QUmuYdA72RBX06QTs6MXw941piREPl3Yfiw=
-cloud.google.com/go v0.121.6 h1:waZiuajrI28iAf40cWgycWNgaXPO06dupuS+sgibK6c=
-cloud.google.com/go v0.121.6/go.mod h1:coChdst4Ea5vUpiALcYKXEpR1S9ZgXbhEzzMcMR66vI=
-cloud.google.com/go/auth v0.16.5 h1:mFWNQ2FEVWAliEQWpAdH80omXFokmrnbDhUS9cBywsI=
-cloud.google.com/go/auth v0.16.5/go.mod h1:utzRfHMP+Vv0mpOkTRQoWD2q3BatTOoWbA7gCc2dUhQ=
+cel.dev/expr v0.25.1 h1:1KrZg61W6TWSxuNZ37Xy49ps13NUovb66QLprthtwi4=
+cel.dev/expr v0.25.1/go.mod h1:hrXvqGP6G6gyx8UAHSHJ5RGk//1Oj5nXQ2NI02Nrsg4=
+cloud.google.com/go v0.123.0 h1:2NAUJwPR47q+E35uaJeYoNhuNEM9kM8SjgRgdeOJUSE=
+cloud.google.com/go v0.123.0/go.mod h1:xBoMV08QcqUGuPW65Qfm1o9Y4zKZBpGS+7bImXLTAZU=
+cloud.google.com/go/auth v0.19.0 h1:DGYwtbcsGsT1ywuxsIoWi1u/vlks0moIblQHgSDgQkQ=
+cloud.google.com/go/auth v0.19.0/go.mod h1:2Aph7BT2KnaSFOM0JDPyiYgNh6PL9vGMiP8CUIXZ+IY=
 cloud.google.com/go/auth/oauth2adapt v0.2.8 h1:keo8NaayQZ6wimpNSmW5OPc283g65QNIiLpZnkHRbnc=
 cloud.google.com/go/auth/oauth2adapt v0.2.8/go.mod h1:XQ9y31RkqZCcwJWNSx2Xvric3RrU88hAYYbjDWYDL+c=
 cloud.google.com/go/compute/metadata v0.9.0 h1:pDUj4QMoPejqq20dK0Pg2N4yG9zIkYGdBtwLoEkH9Zs=
 cloud.google.com/go/compute/metadata v0.9.0/go.mod h1:E0bWwX5wTnLPedCKqk3pJmVgCBSM6qQI1yTBdEb3C10=
-cloud.google.com/go/iam v1.5.2 h1:qgFRAGEmd8z6dJ/qyEchAuL9jpswyODjA2lS+w234g8=
-cloud.google.com/go/iam v1.5.2/go.mod h1:SE1vg0N81zQqLzQEwxL2WI6yhetBdbNQuTvIKCSkUHE=
-cloud.google.com/go/logging v1.13.0 h1:7j0HgAp0B94o1YRDqiqm26w4q1rDMH7XNRU34lJXHYc=
-cloud.google.com/go/logging v1.13.0/go.mod h1:36CoKh6KA/M0PbhPKMq6/qety2DCAErbhXT62TuXALA=
-cloud.google.com/go/longrunning v0.6.7 h1:IGtfDWHhQCgCjwQjV9iiLnUta9LBCo8R9QmAFsS/PrE=
-cloud.google.com/go/longrunning v0.6.7/go.mod h1:EAFV3IZAKmM56TyiE6VAP3VoTzhZzySwI/YI1s/nRsY=
-cloud.google.com/go/monitoring v1.24.2 h1:5OTsoJ1dXYIiMiuL+sYscLc9BumrL3CarVLL7dd7lHM=
-cloud.google.com/go/monitoring v1.24.2/go.mod h1:x7yzPWcgDRnPEv3sI+jJGBkwl5qINf+6qY4eq0I9B4U=
-cloud.google.com/go/storage v1.57.0 h1:4g7NB7Ta7KetVbOMpCqy89C+Vg5VE8scqlSHUPm7Rds=
-cloud.google.com/go/storage v1.57.0/go.mod h1:329cwlpzALLgJuu8beyJ/uvQznDHpa2U5lGjWednkzg=
-cloud.google.com/go/trace v1.11.6 h1:2O2zjPzqPYAHrn3OKl029qlqG6W8ZdYaOWRyr8NgMT4=
-cloud.google.com/go/trace v1.11.6/go.mod h1:GA855OeDEBiBMzcckLPE2kDunIpC72N+Pq8WFieFjnI=
-github.com/Azure/go-ansiterm v0.0.0-20230124172434-306776ec8161 h1:L/gRVlceqvL25UVaW/CKtUDjefjrs0SPonmDGUVOYP0=
-github.com/Azure/go-ansiterm v0.0.0-20230124172434-306776ec8161/go.mod h1:xomTg63KZ2rFqZQzSB4Vz2SUXa1BpHTVz9L5PTmPC4E=
-github.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.29.0 h1:UQUsRi8WTzhZntp5313l+CHIAT95ojUI2lpP/ExlZa4=
-github.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.29.0/go.mod h1:Cz6ft6Dkn3Et6l2v2a9/RpN7epQ1GtDlO6lj8bEcOvw=
-github.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/metric v0.53.0 h1:owcC2UnmsZycprQ5RfRgjydWhuoxg71LUfyiQdijZuM=
-github.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/metric v0.53.0/go.mod h1:ZPpqegjbE99EPKsu3iUWV22A04wzGPcAY/ziSIQEEgs=
-github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/cloudmock v0.53.0 h1:4LP6hvB4I5ouTbGgWtixJhgED6xdf67twf9PoY96Tbg=
-github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/cloudmock v0.53.0/go.mod h1:jUZ5LYlw40WMd07qxcQJD5M40aUxrfwqQX1g7zxYnrQ=
-github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/resourcemapping v0.53.0 h1:Ron4zCA/yk6U7WOBXhTJcDpsUBG9npumK6xw2auFltQ=
-github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/resourcemapping v0.53.0/go.mod h1:cSgYe11MCNYunTnRXrKiR/tHc0eoKjICUuWpNZoVCOo=
+cloud.google.com/go/iam v1.7.0 h1:JD3zh0C6LHl16aCn5Akff0+GELdp1+4hmh6ndoFLl8U=
+cloud.google.com/go/iam v1.7.0/go.mod h1:tetWZW1PD/m6vcuY2Zj/aU0eCHNPuxedbnbRTyKXvdY=
+cloud.google.com/go/logging v1.13.2 h1:qqlHCBvieJT9Cdq4QqYx1KPadCQ2noD4FK02eNqHAjA=
+cloud.google.com/go/logging v1.13.2/go.mod h1:zaybliM3yun1J8mU2dVQ1/qDzjbOqEijZCn6hSBtKak=
+cloud.google.com/go/longrunning v0.9.0 h1:0EzbDEGsAvOZNbqXopgniY0w0a1phvu5IdUFq8grmqY=
+cloud.google.com/go/longrunning v0.9.0/go.mod h1:pkTz846W7bF4o2SzdWJ40Hu0Re+UoNT6Q5t+igIcb8E=
+cloud.google.com/go/monitoring v1.24.3 h1:dde+gMNc0UhPZD1Azu6at2e79bfdztVDS5lvhOdsgaE=
+cloud.google.com/go/monitoring v1.24.3/go.mod h1:nYP6W0tm3N9H/bOw8am7t62YTzZY+zUeQ+Bi6+2eonI=
+cloud.google.com/go/storage v1.62.3 h1:SZq1t23NCI+e96dH77Dg3PEfsNNEjqO8zE5AnD8gVD0=
+cloud.google.com/go/storage v1.62.3/go.mod h1:cpYz/kRVZ+UQAF1uHeea10/9ewcRbxGoGNKsS9daSXA=
+cloud.google.com/go/trace v1.11.7 h1:kDNDX8JkaAG3R2nq1lIdkb7FCSi1rCmsEtKVsty7p+U=
+cloud.google.com/go/trace v1.11.7/go.mod h1:TNn9d5V3fQVf6s4SCveVMIBS2LJUqo73GACmq/Tky0s=
+github.com/Azure/go-ansiterm v0.0.0-20250102033503-faa5f7b0171c h1:udKWzYgxTojEKWjV8V+WSxDXJ4NFATAsZjh8iIbsQIg=
+github.com/Azure/go-ansiterm v0.0.0-20250102033503-faa5f7b0171c/go.mod h1:xomTg63KZ2rFqZQzSB4Vz2SUXa1BpHTVz9L5PTmPC4E=
+github.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.31.0 h1:DHa2U07rk8syqvCge0QIGMCE1WxGj9njT44GH7zNJLQ=
+github.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.31.0/go.mod h1:P4WPRUkOhJC13W//jWpyfJNDAIpvRbAUIYLX/4jtlE0=
+github.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/metric v0.55.0 h1:UnDZ/zFfG1JhH/DqxIZYU/1CUAlTUScoXD/LcM2Ykk8=
+github.com/GoogleCloudPlatform/opentelemetry-operations-go/exporter/metric v0.55.0/go.mod h1:IA1C1U7jO/ENqm/vhi7V9YYpBsp+IMyqNrEN94N7tVc=
+github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/cloudmock v0.55.0 h1:7t/qx5Ost0s0wbA/VDrByOooURhp+ikYwv20i9Y07TQ=
+github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/cloudmock v0.55.0/go.mod h1:vB2GH9GAYYJTO3mEn8oYwzEdhlayZIdQz6zdzgUIRvA=
+github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/resourcemapping v0.55.0 h1:0s6TxfCu2KHkkZPnBfsQ2y5qia0jl3MMrmBhu3nCOYk=
+github.com/GoogleCloudPlatform/opentelemetry-operations-go/internal/resourcemapping v0.55.0/go.mod h1:Mf6O40IAyB9zR/1J8nGDDPirZQQPbYJni8Yisy7NTMc=
 github.com/Microsoft/go-winio v0.6.2 h1:F2VQgta7ecxGYO8k3ZZz3RS8fVIXVxONVUPlNERoyfY=
 github.com/Microsoft/go-winio v0.6.2/go.mod h1:yd8OoFMLzJbo9gZq8j5qaps8bJ9aShtEA8Ipt1oGCvU=
 github.com/VividCortex/gohistogram v1.0.0 h1:6+hBz+qvs0JOrrNhhmR7lFxo5sINxBCGXrdtl/UvroE=
 github.com/VividCortex/gohistogram v1.0.0/go.mod h1:Pf5mBqqDxYaXu3hDrrU+w6nw50o/4+TcAqDqk/vUH7g=
-github.com/absmach/certs v0.18.2 h1:s6KKL3/KfDZ6z0IxvNCksIOUwRnEgQyCpeAonuR15No=
-github.com/absmach/certs v0.18.2/go.mod h1:scqVZsmW2xPScnpMTtE70oN6cn0LLjFcJVPi4JKZ4+E=
-github.com/absmach/supermq v0.18.2 h1:oGiuMyjiFJhAmysmApOsYKtN4Kf/WXpLWElEUqmcZec=
-github.com/absmach/supermq v0.18.2/go.mod h1:ehhfnpFhHZLF9EAdxqDSUCWLJxYhu0mqFd7a4CKMY0g=
+github.com/absmach/certs v0.18.5 h1:eYlvitou+LoDtt7ETVLTp6d/1xCejGL3EmVOg+rHGTU=
+github.com/absmach/certs v0.18.5/go.mod h1:31dtVe1VYF16W+IvjAE/uPAIz4f3uLHgh+moBezjqIc=
+github.com/absmach/magistrala v0.20.0 h1:3AQ0C2AMoOCc1UuJLhPNJLMrNRLZoN0ibSOERqEkM98=
+github.com/absmach/magistrala v0.20.0/go.mod h1:lnuO4fSngMiRYyNYL4yz5UP8DX3bbXRm87b2KHFGwJU=
+github.com/absmach/supermq v0.19.2-0.20260317185610-fade98b84ee4 h1:533pRc6R7perWDqJuZq+ofBQfYfmyj7n49V4LFY4zpo=
+github.com/absmach/supermq v0.19.2-0.20260317185610-fade98b84ee4/go.mod h1:xDAX/O3VcOsHWCx2fk85VD7FI17hAUOvoOhho7DA7g0=
 github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
 github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
 github.com/caarlos0/env/v10 v10.0.0 h1:yIHUBZGsyqCnpTkbjk8asUlx6RFhhEs+h7TOBdgdzXA=
 github.com/caarlos0/env/v10 v10.0.0/go.mod h1:ZfulV76NvVPw3tm591U4SwL3Xx9ldzBP9aGxzeN7G18=
-github.com/caarlos0/env/v11 v11.3.1 h1:cArPWC15hWmEt+gWk7YBi7lEXTXCvpaSdCiZE2X5mCA=
-github.com/caarlos0/env/v11 v11.3.1/go.mod h1:qupehSf/Y0TUTsxKywqRt/vJjN5nz6vauiYEUUr8P4U=
+github.com/caarlos0/env/v11 v11.4.0 h1:Kcb6t5kIIr4XkoQC9AF2j+8E1Jsrl3Wz/hhm1LtoGAc=
+github.com/caarlos0/env/v11 v11.4.0/go.mod h1:qupehSf/Y0TUTsxKywqRt/vJjN5nz6vauiYEUUr8P4U=
 github.com/cenkalti/backoff/v5 v5.0.3 h1:ZN+IMa753KfX5hd8vVaMixjnqRZ3y8CuJKRKj1xcsSM=
 github.com/cenkalti/backoff/v5 v5.0.3/go.mod h1:rkhZdG3JZukswDf7f0cwqPNk4K0sa+F97BxZthm/crw=
 github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
 github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
-github.com/cncf/xds/go v0.0.0-20250501225837-2ac532fd4443 h1:aQ3y1lwWyqYPiWZThqv1aFbZMiM9vblcSArJRf2Irls=
-github.com/cncf/xds/go v0.0.0-20250501225837-2ac532fd4443/go.mod h1:W+zGtBO5Y1IgJhy4+A9GOqVhqLpfZi+vwmdNXUehLA8=
+github.com/cncf/xds/go v0.0.0-20251210132809-ee656c7534f5 h1:6xNmx7iTtyBRev0+D/Tv1FZd4SCg8axKApyNyRsAt/w=
+github.com/cncf/xds/go v0.0.0-20251210132809-ee656c7534f5/go.mod h1:KdCmV+x/BuvyMxRnYBlmVaq4OLiKW6iRQfvC62cvdkI=
 github.com/containerd/errdefs v1.0.0 h1:tg5yIfIlQIrxYtu9ajqY42W3lpS19XqdxRQeEwYG8PI=
 github.com/containerd/errdefs v1.0.0/go.mod h1:+YBYIdtsnF4Iw6nWZhJcqGSg/dwvV7tyJ/kCkyJ2k+M=
 github.com/containerd/errdefs/pkg v0.3.0 h1:9IKJ06FvyNlexW690DXuQNx2KA2cUJXx151Xdx3ZPPE=
@@ -64,34 +66,40 @@ github.com/decred/dcrd/dcrec/secp256k1/v4 v4.4.0 h1:NMZiJj8QnKe1LgsbDayM4UoHwbvw
 github.com/decred/dcrd/dcrec/secp256k1/v4 v4.4.0/go.mod h1:ZXNYxsqcloTdSy/rNShjYzMhyjf0LaoftYK0p+A3h40=
 github.com/distribution/reference v0.6.0 h1:0IXCQ5g4/QMHHkarYzh5l+u8T3t73zM5QvfrDyIgxBk=
 github.com/distribution/reference v0.6.0/go.mod h1:BbU0aIcezP1/5jX/8MP0YiH4SdvB5Y4f/wlDRiLyi3E=
-github.com/docker/docker v28.5.1+incompatible h1:Bm8DchhSD2J6PsFzxC35TZo4TLGR2PdW/E69rU45NhM=
-github.com/docker/docker v28.5.1+incompatible/go.mod h1:eEKB0N0r5NX/I1kEveEz05bcu8tLC/8azJZsviup8Sk=
-github.com/docker/go-connections v0.5.0 h1:USnMq7hx7gwdVZq1L49hLXaFtUdTADjXGp+uj1Br63c=
-github.com/docker/go-connections v0.5.0/go.mod h1:ov60Kzw0kKElRwhNs9UlUHAE/F9Fe6GLaXnqyDdmEXc=
+github.com/docker/docker v28.5.2+incompatible h1:DBX0Y0zAjZbSrm1uzOkdr1onVghKaftjlSWt4AFexzM=
+github.com/docker/docker v28.5.2+incompatible/go.mod h1:eEKB0N0r5NX/I1kEveEz05bcu8tLC/8azJZsviup8Sk=
+github.com/docker/go-connections v0.6.0 h1:LlMG9azAe1TqfR7sO+NJttz1gy6KO7VJBh+pMmjSD94=
+github.com/docker/go-connections v0.6.0/go.mod h1:AahvXYshr6JgfUJGdDCs2b5EZG/vmaMAntpSFH5BFKE=
 github.com/docker/go-units v0.5.0 h1:69rxXcBk27SvSaaxTtLh/8llcHD8vYHT7WSdRZ/jvr4=
 github.com/docker/go-units v0.5.0/go.mod h1:fgPhTUdO+D/Jk86RDLlptpiXQzgHJF7gydDDbaIK4Dk=
 github.com/edgelesssys/go-azguestattestation v0.0.0-20250408071817-8c4457b235ff h1:V6A5kD0+c1Qg4X72Lg+zxhCZk+par436sQdgLvMCBBc=
 github.com/edgelesssys/go-azguestattestation v0.0.0-20250408071817-8c4457b235ff/go.mod h1:Lz4QaomI4wU2YbatD4/W7vatW2Q35tnkoJezB1clscc=
-github.com/envoyproxy/go-control-plane v0.13.4 h1:zEqyPVyku6IvWCFwux4x9RxkLOMUL+1vC9xUFv5l2/M=
-github.com/envoyproxy/go-control-plane v0.13.4/go.mod h1:kDfuBlDVsSj2MjrLEtRWtHlsWIFcGyB2RMO44Dc5GZA=
-github.com/envoyproxy/go-control-plane/envoy v1.32.4 h1:jb83lalDRZSpPWW2Z7Mck/8kXZ5CQAFYVjQcdVIr83A=
-github.com/envoyproxy/go-control-plane/envoy v1.32.4/go.mod h1:Gzjc5k8JcJswLjAx1Zm+wSYE20UrLtt7JZMWiWQXQEw=
+github.com/envoyproxy/go-control-plane v0.14.0 h1:hbG2kr4RuFj222B6+7T83thSPqLjwBIfQawTkC++2HA=
+github.com/envoyproxy/go-control-plane v0.14.0/go.mod h1:NcS5X47pLl/hfqxU70yPwL9ZMkUlwlKxtAohpi2wBEU=
+github.com/envoyproxy/go-control-plane/envoy v1.36.0 h1:yg/JjO5E7ubRyKX3m07GF3reDNEnfOboJ0QySbH736g=
+github.com/envoyproxy/go-control-plane/envoy v1.36.0/go.mod h1:ty89S1YCCVruQAm9OtKeEkQLTb+Lkz0k8v9W0Oxsv98=
 github.com/envoyproxy/go-control-plane/ratelimit v0.1.0 h1:/G9QYbddjL25KvtKTv3an9lx6VBE2cnb8wp1vEGNYGI=
 github.com/envoyproxy/go-control-plane/ratelimit v0.1.0/go.mod h1:Wk+tMFAFbCXaJPzVVHnPgRKdUdwW/KdbRt94AzgRee4=
-github.com/envoyproxy/protoc-gen-validate v1.2.1 h1:DEo3O99U8j4hBFwbJfrz9VtgcDfUKS7KJ7spH3d86P8=
-github.com/envoyproxy/protoc-gen-validate v1.2.1/go.mod h1:d/C80l/jxXLdfEIhX1W2TmLfsJ31lvEjwamM4DxlWXU=
-github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
-github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
+github.com/envoyproxy/protoc-gen-validate v1.3.0 h1:TvGH1wof4H33rezVKWSpqKz5NXWg5VPuZ0uONDT6eb4=
+github.com/envoyproxy/protoc-gen-validate v1.3.0/go.mod h1:HvYl7zwPa5mffgyeTUHA9zHIH36nmrm7oCbo4YKoSWA=
+github.com/fatih/color v1.19.0 h1:Zp3PiM21/9Ld6FzSKyL5c/BULoe/ONr9KlbYVOfG8+w=
+github.com/fatih/color v1.19.0/go.mod h1:zNk67I0ZUT1bEGsSGyCZYZNrHuTkJJB+r6Q9VuMi0LE=
 github.com/felixge/httpsnoop v1.0.4 h1:NFTV2Zj1bL4mc9sqWACXbQFVBBg2W3GPvqp8/ESS2Wg=
 github.com/felixge/httpsnoop v1.0.4/go.mod h1:m8KPJKqk1gH5J9DgRY2ASl2lWCfGKXixSwevea8zH2U=
-github.com/go-chi/chi/v5 v5.2.3 h1:WQIt9uxdsAbgIYgid+BpYc+liqQZGMHRaUwp0JUcvdE=
-github.com/go-chi/chi/v5 v5.2.3/go.mod h1:L2yAIGWB3H+phAw1NxKwWM+7eUH/lU8pOMm5hHcoops=
+github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
+github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
+github.com/fxamacker/cbor/v2 v2.2.0/go.mod h1:TA1xS00nchWmaBnEIxPSE5oHLuJBAVvqrtAnWBwBCVo=
+github.com/fxamacker/cbor/v2 v2.3.0/go.mod h1:TA1xS00nchWmaBnEIxPSE5oHLuJBAVvqrtAnWBwBCVo=
+github.com/fxamacker/cbor/v2 v2.9.0 h1:NpKPmjDBgUfBms6tr6JZkTHtfFGcMKsw3eGcmD/sapM=
+github.com/fxamacker/cbor/v2 v2.9.0/go.mod h1:vM4b+DJCtHn+zz7h3FFp/hDAI9WNWCsZj23V5ytsSxQ=
+github.com/go-chi/chi/v5 v5.2.5 h1:Eg4myHZBjyvJmAFjFvWgrqDTXFyOzjj7YIm3L3mu6Ug=
+github.com/go-chi/chi/v5 v5.2.5/go.mod h1:X7Gx4mteadT3eDOMTsXzmI4/rwUpOwBHLpAfupzFJP0=
 github.com/go-gorp/gorp/v3 v3.1.0 h1:ItKF/Vbuj31dmV4jxA1qblpSwkl9g1typ24xoe70IGs=
 github.com/go-gorp/gorp/v3 v3.1.0/go.mod h1:dLEjIyyRNiXvNZ8PSmzpt1GsWAUK8kjVhEpjH8TixEw=
 github.com/go-jose/go-jose/v3 v3.0.4 h1:Wp5HA7bLQcKnf6YYao/4kpRpVMp/yf6+pJKV8WFSaNY=
 github.com/go-jose/go-jose/v3 v3.0.4/go.mod h1:5b+7YgP7ZICgJDBdfjZaIt+H/9L9T/YQrVfLAMboGkQ=
-github.com/go-jose/go-jose/v4 v4.1.2 h1:TK/7NqRQZfgAh+Td8AlsrvtPoUyiHh0LqVvokh+1vHI=
-github.com/go-jose/go-jose/v4 v4.1.2/go.mod h1:22cg9HWM1pOlnRiY+9cQYJ9XHmya1bYW8OeDM6Ku6Oo=
+github.com/go-jose/go-jose/v4 v4.1.4 h1:moDMcTHmvE6Groj34emNPLs/qtYXRVcd6S7NHbHz3kA=
+github.com/go-jose/go-jose/v4 v4.1.4/go.mod h1:x4oUasVrzR7071A4TnHLGSPpNOm2a21K9Kf04k1rs08=
 github.com/go-kit/kit v0.13.0 h1:OoneCcHKHQ03LfBpoQCUfCluwd2Vt3ohz+kvbJneZAU=
 github.com/go-kit/kit v0.13.0/go.mod h1:phqEHMMUbyrCFCTgH48JueqrM3md2HcAZ8N3XE4FKDg=
 github.com/go-kit/log v0.2.1 h1:MRVx0/zhvdseW+Gza6N9rVzU/IVzaeE1SFI4raAhmBU=
@@ -107,8 +115,8 @@ github.com/goccy/go-json v0.10.5 h1:Fq85nIqj+gXn/S5ahsiTlK3TmC85qgirsdTP/+DeaC4=
 github.com/goccy/go-json v0.10.5/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M=
 github.com/gofrs/uuid v4.4.0+incompatible h1:3qXRTX8/NbyulANqlc0lchS1gqAVxRgsuW1YrTJupqA=
 github.com/gofrs/uuid v4.4.0+incompatible/go.mod h1:b2aQJv3Z4Fp6yNu3cdSllBxTCLRxnplIgP/c0N/04lM=
-github.com/gofrs/uuid/v5 v5.3.2 h1:2jfO8j3XgSwlz/wHqemAEugfnTlikAYHhnqQ8Xh4fE0=
-github.com/gofrs/uuid/v5 v5.3.2/go.mod h1:CDOjlDMVAtN56jqyRUZh58JT31Tiw7/oQyEXZV+9bD8=
+github.com/gofrs/uuid/v5 v5.4.0 h1:EfbpCTjqMuGyq5ZJwxqzn3Cbr2d0rUZU7v5ycAk/e/0=
+github.com/gofrs/uuid/v5 v5.4.0/go.mod h1:CDOjlDMVAtN56jqyRUZh58JT31Tiw7/oQyEXZV+9bD8=
 github.com/golang-jwt/jwt/v5 v5.3.0 h1:pv4AsKCKKZuqlgs5sUmn4x8UlGa0kEVt/puTpKx9vvo=
 github.com/golang-jwt/jwt/v5 v5.3.0/go.mod h1:fxCRLWMO43lRc8nhHWY6LGqRcf+1gQWArsqaEUEa5bE=
 github.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=
@@ -125,16 +133,14 @@ github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
 github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
 github.com/google/go-configfs-tsm v0.3.3-0.20240919001351-b4b5b84fdcbc h1:SG12DWUUM5igxm+//YX5Yq4vhdoRnOG9HkCodkOn+YU=
 github.com/google/go-configfs-tsm v0.3.3-0.20240919001351-b4b5b84fdcbc/go.mod h1:EL1GTDFMb5PZQWDviGfZV9n87WeGTR/JUg13RfwkgRo=
-github.com/google/go-eventlog v0.0.2-0.20241003021507-01bb555f7cba h1:05m5+kgZjxYUZrx3bZfkKHl6wkch+Khao6N21rFHInk=
-github.com/google/go-eventlog v0.0.2-0.20241003021507-01bb555f7cba/go.mod h1:7huE5P8w2NTObSwSJjboHmB7ioBNblkijdzoVa2skfQ=
-github.com/google/go-sev-guest v0.13.0 h1:DJB6ACdykyweMU0HGOp/TQ7cjsnbV2ecbYunu2E0qy0=
-github.com/google/go-sev-guest v0.13.0/go.mod h1:SK9vW+uyfuzYdVN0m8BShL3OQCtXZe/JPF7ZkpD3760=
-github.com/google/go-tdx-guest v0.3.2-0.20241009005452-097ee70d0843 h1:+MoPobRN9HrDhGyn6HnF5NYo4uMBKaiFqAtf/D/OB4A=
-github.com/google/go-tdx-guest v0.3.2-0.20241009005452-097ee70d0843/go.mod h1:g/n8sKITIT9xRivBUbizo34DTsUm2nN2uU3A662h09g=
-github.com/google/go-tpm v0.9.6 h1:Ku42PT4LmjDu1H5C5ISWLlpI1mj+Zq7sPGKoRw2XROA=
-github.com/google/go-tpm v0.9.6/go.mod h1:h9jEsEECg7gtLis0upRBQU+GhYVH6jMjrFxI8u6bVUY=
-github.com/google/go-tpm-tools v0.4.6 h1:hwIwPG7w4z5eQEBq11gYw8YYr9xXLfBQ/0JsKyq5AJM=
-github.com/google/go-tpm-tools v0.4.6/go.mod h1:MsVQbJnRhKDfWwf5zgr3cDGpj13P1uLAFF0wMEP/n5w=
+github.com/google/go-sev-guest v0.14.1 h1:j/DXy9jk1qSW/dEV9vDiQnhAVFD1zqnWNVu6p1J0Jgo=
+github.com/google/go-sev-guest v0.14.1/go.mod h1:SK9vW+uyfuzYdVN0m8BShL3OQCtXZe/JPF7ZkpD3760=
+github.com/google/go-tdx-guest v0.3.2-0.20260605221019-34f07ec666c4 h1:OX2Mksz5ZHxawvZskqYX18Xy/q292EoiyUAT4WXg/gU=
+github.com/google/go-tdx-guest v0.3.2-0.20260605221019-34f07ec666c4/go.mod h1:uHy3VaNXNXhl0fiPxKqTxieeouqQmW6A0EfLcaeCYBk=
+github.com/google/go-tpm v0.9.8 h1:slArAR9Ft+1ybZu0lBwpSmpwhRXaa85hWtMinMyRAWo=
+github.com/google/go-tpm v0.9.8/go.mod h1:h9jEsEECg7gtLis0upRBQU+GhYVH6jMjrFxI8u6bVUY=
+github.com/google/go-tpm-tools v0.4.4 h1:oiQfAIkc6xTy9Fl5NKTeTJkBTlXdHsxAofmQyxBKY98=
+github.com/google/go-tpm-tools v0.4.4/go.mod h1:T8jXkp2s+eltnCDIsXR84/MTcVU9Ja7bh3Mit0pa4AY=
 github.com/google/go-tspi v0.3.0 h1:ADtq8RKfP+jrTyIWIZDIYcKOMecRqNJFOew2IT0Inus=
 github.com/google/go-tspi v0.3.0/go.mod h1:xfMGI3G0PhxCdNVcYr1C4C+EizojDg/TXuX5by8CiHI=
 github.com/google/logger v1.1.1 h1:+6Z2geNxc9G+4D4oDO9njjjn2d0wN5d7uOo0vOIW1NQ=
@@ -143,32 +149,33 @@ github.com/google/martian/v3 v3.3.3 h1:DIhPTQrbPkgs2yJYdXU/eNACCG5DVQjySNRNlflZ9
 github.com/google/martian/v3 v3.3.3/go.mod h1:iEPrYcgCF7jA9OtScMFQyAlZZ4YXTKEtJ1E6RWzmBA0=
 github.com/google/s2a-go v0.1.9 h1:LGD7gtMgezd8a/Xak7mEWL0PjoTQFvpRudN895yqKW0=
 github.com/google/s2a-go v0.1.9/go.mod h1:YA0Ei2ZQL3acow2O62kdp9UlnvMmU7kA6Eutn0dXayM=
+github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
-github.com/googleapis/enterprise-certificate-proxy v0.3.6 h1:GW/XbdyBFQ8Qe+YAmFU9uHLo7OnF5tL52HFAgMmyrf4=
-github.com/googleapis/enterprise-certificate-proxy v0.3.6/go.mod h1:MkHOF77EYAE7qfSuSS9PU6g4Nt4e11cnsDUowfwewLA=
-github.com/googleapis/gax-go/v2 v2.15.0 h1:SyjDc1mGgZU5LncH8gimWo9lW1DtIfPibOG81vgd/bo=
-github.com/googleapis/gax-go/v2 v2.15.0/go.mod h1:zVVkkxAQHa1RQpg9z2AUCMnKhi0Qld9rcmyfL1OZhoc=
+github.com/googleapis/enterprise-certificate-proxy v0.3.14 h1:yh8ncqsbUY4shRD5dA6RlzjJaT4hi3kII+zYw8wmLb8=
+github.com/googleapis/enterprise-certificate-proxy v0.3.14/go.mod h1:vqVt9yG9480NtzREnTlmGSBmFrA+bzb0yl0TxoBQXOg=
+github.com/googleapis/gax-go/v2 v2.21.0 h1:h45NjjzEO3faG9Lg/cFrBh2PgegVVgzqKzuZl/wMbiI=
+github.com/googleapis/gax-go/v2 v2.21.0/go.mod h1:But/NJU6TnZsrLai/xBAQLLz+Hc7fHZJt/hsCz3Fih4=
 github.com/gopherjs/gopherjs v1.17.2 h1:fQnZVsXk8uxXIStYb0N4bGk7jeyTalG/wsZjQ25dO0g=
 github.com/gopherjs/gopherjs v1.17.2/go.mod h1:pRRIvn/QzFLrKfvEz3qUuEhtE/zLCWfreZ6J5gM2i+k=
-github.com/grpc-ecosystem/grpc-gateway/v2 v2.27.3 h1:NmZ1PKzSTQbuGHw9DGPFomqkkLWMC+vZCkfs+FHv1Vg=
-github.com/grpc-ecosystem/grpc-gateway/v2 v2.27.3/go.mod h1:zQrxl1YP88HQlA6i9c63DSVPFklWpGX4OWAc9bFuaH4=
+github.com/grpc-ecosystem/grpc-gateway/v2 v2.28.0 h1:HWRh5R2+9EifMyIHV7ZV+MIZqgz+PMpZ14Jynv3O2Zs=
+github.com/grpc-ecosystem/grpc-gateway/v2 v2.28.0/go.mod h1:JfhWUomR1baixubs02l85lZYYOm7LV6om4ceouMv45c=
 github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
 github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
 github.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=
 github.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=
 github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 h1:iCEnooe7UlwOQYpKFhBabPMi4aNAfoODPEFNiAnClxo=
 github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM=
-github.com/jackc/pgx/v5 v5.7.6 h1:rWQc5FwZSPX58r1OQmkuaNicxdmExaEz5A2DO2hUuTk=
-github.com/jackc/pgx/v5 v5.7.6/go.mod h1:aruU7o91Tc2q2cFp5h4uP3f6ztExVpyVv88Xl/8Vl8M=
+github.com/jackc/pgx/v5 v5.9.1 h1:uwrxJXBnx76nyISkhr33kQLlUqjv7et7b9FjCen/tdc=
+github.com/jackc/pgx/v5 v5.9.1/go.mod h1:mal1tBGAFfLHvZzaYh77YS/eC6IX9OWbRV1QIIM0Jn4=
 github.com/jackc/puddle/v2 v2.2.2 h1:PR8nw+E/1w0GLuRFSmiioY6UooMp6KJv0/61nB7icHo=
 github.com/jackc/puddle/v2 v2.2.2/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4=
 github.com/jmoiron/sqlx v1.4.0 h1:1PLqN7S1UYp5t4SrVVnt4nUVNemrDAtxlulVe+Qgm3o=
 github.com/jmoiron/sqlx v1.4.0/go.mod h1:ZrZ7UsYB/weZdl2Bxg6jCRO9c3YHl8r3ahlKmRT4JLY=
 github.com/jtolds/gls v4.20.0+incompatible h1:xdiiI2gbIgH/gLH7ADydsJ1uDOEzR8yvV7C0MuV77Wo=
 github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=
-github.com/klauspost/compress v1.18.1 h1:bcSGx7UbpBqMChDtsF28Lw6v/G94LPrrbMbdC3JH2co=
-github.com/klauspost/compress v1.18.1/go.mod h1:ZQFFVG+MdnR0P+l6wpXgIL4NTtwiKIdBnrBd8Nrxr+0=
+github.com/klauspost/compress v1.18.5 h1:/h1gH5Ce+VWNLSWqPzOVn6XBO+vJbCNGvjoaGBFW2IE=
+github.com/klauspost/compress v1.18.5/go.mod h1:cwPg85FWrGar70rWktvGQj8/hthj3wpl0PGDogxkrSQ=
 github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
 github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
 github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
@@ -197,16 +204,16 @@ github.com/moby/sys/atomicwriter v0.1.0 h1:kw5D/EqkBwsBFi0ss9v1VG3wIkVhzGvLklJ+w
 github.com/moby/sys/atomicwriter v0.1.0/go.mod h1:Ul8oqv2ZMNHOceF643P6FKPXeCmYtlQMvpizfsSoaWs=
 github.com/moby/sys/sequential v0.6.0 h1:qrx7XFUd/5DxtqcoH1h438hF5TmOvzC/lspjy7zgvCU=
 github.com/moby/sys/sequential v0.6.0/go.mod h1:uyv8EUTrca5PnDsdMGXhZe6CCe8U/UiTWd+lL+7b/Ko=
-github.com/moby/term v0.5.0 h1:xt8Q1nalod/v7BqbG21f8mQPqH+xAaC9C3N3wfWbVP0=
-github.com/moby/term v0.5.0/go.mod h1:8FzsFHVUBGZdbDsJw/ot+X+d5HLUbvklYLJ9uGfcI3Y=
+github.com/moby/term v0.5.2 h1:6qk3FJAFDs6i/q3W/pQ97SX192qKfZgGjCQqfCJkgzQ=
+github.com/moby/term v0.5.2/go.mod h1:d3djjFCrjnB+fl8NJux+EJzu0msscUP+f8it8hPkFLc=
 github.com/morikuni/aec v1.0.0 h1:nP9CBfwrvYnBRgY6qfDQkygYDmYwOilePFkwzv4dU8A=
 github.com/morikuni/aec v1.0.0/go.mod h1:BbKIizmSmc5MMPqRYbxO4ZU0S0+P200+tUnFx7PXmsc=
 github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 h1:C3w9PqII01/Oq1c1nUAm88MOHcQC9l5mIlSMApZMrHA=
 github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822/go.mod h1:+n7T8mK8HuQTcFwEeznm/DIxMOiR9yIdICNftLE1DvQ=
 github.com/opencontainers/go-digest v1.0.0 h1:apOUWs51W5PlhuyGyz9FCeeBIOUDA/6nW8Oi/yOhh5U=
 github.com/opencontainers/go-digest v1.0.0/go.mod h1:0JzlMkj0TRzQZfJkVvzbP0HBR3IKzErnv2BNG4W4MAM=
-github.com/opencontainers/image-spec v1.1.0 h1:8SG7/vwALn54lVB/0yZ/MMwhFrPYtpEHQb2IpWsCzug=
-github.com/opencontainers/image-spec v1.1.0/go.mod h1:W4s4sFTMaBeK1BQLXbG4AdM2szdn85PY75RI83NrTrM=
+github.com/opencontainers/image-spec v1.1.1 h1:y0fUlFfIZhPF1W537XOLg0/fcx6zcHCJwooC2xJA040=
+github.com/opencontainers/image-spec v1.1.1/go.mod h1:qpqAh3Dmcf36wStyyWU+kCeDgrGnAve2nCC8+7h8Q0M=
 github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/planetscale/vtprotobuf v0.6.1-0.20240917153116-6f2963f01587 h1:xzZOeCMQLA/W198ZkdVdt4EKFKJtS26B773zNU377ZY=
@@ -218,14 +225,14 @@ github.com/prometheus/client_golang v1.23.2 h1:Je96obch5RDVy3FDMndoUsjAhG5Edi49h
 github.com/prometheus/client_golang v1.23.2/go.mod h1:Tb1a6LWHB3/SPIzCoaDXI4I8UHKeFTEQ1YCr+0Gyqmg=
 github.com/prometheus/client_model v0.6.2 h1:oBsgwpGs7iVziMvrGhE53c/GrLUsZdHnqNwqPLxwZyk=
 github.com/prometheus/client_model v0.6.2/go.mod h1:y3m2F6Gdpfy6Ut/GBsUqTWZqCUvMVzSfMLjcu6wAwpE=
-github.com/prometheus/common v0.67.1 h1:OTSON1P4DNxzTg4hmKCc37o4ZAZDv0cfXLkOt0oEowI=
-github.com/prometheus/common v0.67.1/go.mod h1:RpmT9v35q2Y+lsieQsdOh5sXZ6ajUGC8NjZAmr8vb0Q=
-github.com/prometheus/procfs v0.17.0 h1:FuLQ+05u4ZI+SS/w9+BWEM2TXiHKsUQ9TADiRH7DuK0=
-github.com/prometheus/procfs v0.17.0/go.mod h1:oPQLaDAMRbA+u8H5Pbfq+dl3VDAvHxMUOVhe0wYB2zw=
+github.com/prometheus/common v0.67.5 h1:pIgK94WWlQt1WLwAC5j2ynLaBRDiinoAb86HZHTUGI4=
+github.com/prometheus/common v0.67.5/go.mod h1:SjE/0MzDEEAyrdr5Gqc6G+sXI67maCxzaT3A2+HqjUw=
+github.com/prometheus/procfs v0.19.2 h1:zUMhqEW66Ex7OXIiDkll3tl9a1ZdilUOd/F6ZXw4Vws=
+github.com/prometheus/procfs v0.19.2/go.mod h1:M0aotyiemPhBCM0z5w87kL22CxfcH05ZpYlu+b4J7mw=
 github.com/rogpeppe/go-internal v1.14.1 h1:UQB4HGPB6osV0SQTLymcB4TgvyWu6ZyliaW0tI/otEQ=
 github.com/rogpeppe/go-internal v1.14.1/go.mod h1:MaRKkUm5W0goXpeCfT7UZI6fk/L7L7so1lCWt35ZSgc=
-github.com/rubenv/sql-migrate v1.8.0 h1:dXnYiJk9k3wetp7GfQbKJcPHjVJL6YK19tKj8t2Ns0o=
-github.com/rubenv/sql-migrate v1.8.0/go.mod h1:F2bGFBwCU+pnmbtNYDeKvSuvL6lBVtXDXUUv5t+u1qw=
+github.com/rubenv/sql-migrate v1.8.1 h1:EPNwCvjAowHI3TnZ+4fQu3a915OpnQoPAjTXCGOy2U0=
+github.com/rubenv/sql-migrate v1.8.1/go.mod h1:BTIKBORjzyxZDS6dzoiw6eAFYJ1iNlGAtjn4LGeVjS8=
 github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
 github.com/sammyoina/sev-snp-measure-go v0.0.0-20241202151803-ef189f0ff825 h1:SqNaL9udBIc026SGNEuEuiVL0/hw9fXxM5qrFhWGkdE=
 github.com/sammyoina/sev-snp-measure-go v0.0.0-20241202151803-ef189f0ff825/go.mod h1:dEkBe8JnxU5itNjZDEQINFd7f7l4DtjfqRuzPQcit4w=
@@ -237,59 +244,72 @@ github.com/smarty/assertions v1.16.0 h1:EvHNkdRA4QHMrn75NZSoUQ/mAUXAYWfatfB01yTC
 github.com/smarty/assertions v1.16.0/go.mod h1:duaaFdCS0K9dnoM50iyek/eYINOZ64gbh1Xlf6LG7AI=
 github.com/smartystreets/goconvey v1.8.1 h1:qGjIddxOk4grTu9JPOU31tVfq3cNdBlNa5sSznIX1xY=
 github.com/smartystreets/goconvey v1.8.1/go.mod h1:+/u4qLyY6x1jReYOp7GOM2FSt8aP9CzCZL03bI28W60=
-github.com/spf13/cobra v1.10.1 h1:lJeBwCfmrnXthfAupyUTzJ/J4Nc1RsHC/mSRU2dll/s=
-github.com/spf13/cobra v1.10.1/go.mod h1:7SmJGaTHFVBY0jW4NXGluQoLvhqFQM+6XSKD+P4XaB0=
+github.com/spf13/cast v1.10.0 h1:h2x0u2shc1QuLHfxi+cTJvs30+ZAHOGRic8uyGTDWxY=
+github.com/spf13/cast v1.10.0/go.mod h1:jNfB8QC9IA6ZuY2ZjDp0KtFO2LZZlg4S/7bzP6qqeHo=
+github.com/spf13/cobra v1.10.2 h1:DMTTonx5m65Ic0GOoRY2c16WCbHxOOw6xxezuLaBpcU=
+github.com/spf13/cobra v1.10.2/go.mod h1:7C1pvHqHw5A4vrJfjNwvOdzYu0Gml16OCs2GRiTUUS4=
 github.com/spf13/pflag v1.0.9/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
 github.com/spf13/pflag v1.0.10 h1:4EBh2KAYBwaONj6b2Ye1GiHfwjqyROoF4RwYO+vPwFk=
 github.com/spf13/pflag v1.0.10/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
-github.com/spiffe/go-spiffe/v2 v2.5.0 h1:N2I01KCUkv1FAjZXJMwh95KK1ZIQLYbPfhaxw8WS0hE=
-github.com/spiffe/go-spiffe/v2 v2.5.0/go.mod h1:P+NxobPc6wXhVtINNtFjNWGBTreew1GBUCwT2wPmb7g=
+github.com/spiffe/go-spiffe/v2 v2.6.0 h1:l+DolpxNWYgruGQVV0xsfeya3CsC7m8iBzDnMpsbLuo=
+github.com/spiffe/go-spiffe/v2 v2.6.0/go.mod h1:gm2SeUoMZEtpnzPNs2Csc0D/gX33k1xIx7lEzqblHEs=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/objx v0.5.3 h1:jmXUvGomnU1o3W/V5h2VEradbpJDwGrzugQQvL0POH4=
 github.com/stretchr/objx v0.5.3/go.mod h1:rDQraq+vQZU7Fde9LOZLr8Tax6zZvy4kuNKF+QYS+U0=
+github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
 github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
+github.com/veraison/corim v1.1.2 h1:JIk6ZK/OzKEb0FJUFHSnmkn67yyGy+5NChYax0bwttA=
+github.com/veraison/corim v1.1.2/go.mod h1:yoN6+vVQJgzS926nheCbJi68SvOlN0CpiPuTxYSe5FU=
+github.com/veraison/eat v0.0.0-20210331113810-3da8a4dd42ff h1:r6I2eJL/z8dp5flsQIKHMeDjyV6UO8If3MaVBLvTjF4=
+github.com/veraison/eat v0.0.0-20210331113810-3da8a4dd42ff/go.mod h1:+kxt8iuFiVvKRs2VQ1Ho7bbAScXAB/kHFFuP5Biw19I=
+github.com/veraison/go-cose v1.3.0 h1:2/H5w8kdSpQJyVtIhx8gmwPJ2uSz1PkyWFx0idbd7rk=
+github.com/veraison/go-cose v1.3.0/go.mod h1:df09OV91aHoQWLmy1KsDdYiagtXgyAwAl8vFeFn1gMc=
+github.com/veraison/swid v1.1.1-0.20230911094910-8ffdd07a22ca h1:osmCKwWO/xM68Kz+rIXio1DNzEY2NdJOpGpoy5r8NlE=
+github.com/veraison/swid v1.1.1-0.20230911094910-8ffdd07a22ca/go.mod h1:d5jt76uMNbTfQ+f2qU4Lt8RvWOTsv6PFgstIM1QdMH0=
+github.com/x448/float16 v0.8.4 h1:qLwI1I70+NjRFUR3zs1JPUCgaCXSh3SW62uAKT1mSBM=
+github.com/x448/float16 v0.8.4/go.mod h1:14CWIYCyZA/cWjXOioeEpHeN/83MdbZDRQHoFcYsOfg=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
-github.com/zeebo/errs v1.4.0 h1:XNdoD/RRMKP7HD0UhJnIzUy74ISdGGxURlYG8HSWSfM=
-github.com/zeebo/errs v1.4.0/go.mod h1:sgbWHsvVuTPHcqJJGQ1WhI5KbWlHYz+2+2C/LSEtCw4=
 go.opentelemetry.io/auto/sdk v1.2.1 h1:jXsnJ4Lmnqd11kwkBV2LgLoFMZKizbCi5fNZ/ipaZ64=
 go.opentelemetry.io/auto/sdk v1.2.1/go.mod h1:KRTj+aOaElaLi+wW1kO/DZRXwkF4C5xPbEe3ZiIhN7Y=
-go.opentelemetry.io/contrib/detectors/gcp v1.36.0 h1:F7q2tNlCaHY9nMKHR6XH9/qkp8FktLnIcy6jJNyOCQw=
-go.opentelemetry.io/contrib/detectors/gcp v1.36.0/go.mod h1:IbBN8uAIIx734PTonTPxAxnjc2pQTxWNkwfstZ+6H2k=
-go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.63.0 h1:YH4g8lQroajqUwWbq/tr2QX1JFmEXaDLgG+ew9bLMWo=
-go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.63.0/go.mod h1:fvPi2qXDqFs8M4B4fmJhE92TyQs9Ydjlg3RvfUp+NbQ=
-go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.63.0 h1:RbKq8BG0FI8OiXhBfcRtqqHcZcka+gU3cskNuf05R18=
-go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.63.0/go.mod h1:h06DGIukJOevXaj/xrNjhi/2098RZzcLTbc0jDAUbsg=
-go.opentelemetry.io/otel v1.38.0 h1:RkfdswUDRimDg0m2Az18RKOsnI8UDzppJAtj01/Ymk8=
-go.opentelemetry.io/otel v1.38.0/go.mod h1:zcmtmQ1+YmQM9wrNsTGV/q/uyusom3P8RxwExxkZhjM=
-go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.38.0 h1:GqRJVj7UmLjCVyVJ3ZFLdPRmhDUp2zFmQe3RHIOsw24=
-go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.38.0/go.mod h1:ri3aaHSmCTVYu2AWv44YMauwAQc0aqI9gHKIcSbI1pU=
-go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.38.0 h1:aTL7F04bJHUlztTsNGJ2l+6he8c+y/b//eR0jjjemT4=
-go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.38.0/go.mod h1:kldtb7jDTeol0l3ewcmd8SDvx3EmIE7lyvqbasU3QC4=
-go.opentelemetry.io/otel/exporters/stdout/stdoutmetric v1.36.0 h1:rixTyDGXFxRy1xzhKrotaHy3/KXdPhlWARrCgK+eqUY=
-go.opentelemetry.io/otel/exporters/stdout/stdoutmetric v1.36.0/go.mod h1:dowW6UsM9MKbJq5JTz2AMVp3/5iW5I/TStsk8S+CfHw=
-go.opentelemetry.io/otel/metric v1.38.0 h1:Kl6lzIYGAh5M159u9NgiRkmoMKjvbsKtYRwgfrA6WpA=
-go.opentelemetry.io/otel/metric v1.38.0/go.mod h1:kB5n/QoRM8YwmUahxvI3bO34eVtQf2i4utNVLr9gEmI=
-go.opentelemetry.io/otel/sdk v1.38.0 h1:l48sr5YbNf2hpCUj/FoGhW9yDkl+Ma+LrVl8qaM5b+E=
-go.opentelemetry.io/otel/sdk v1.38.0/go.mod h1:ghmNdGlVemJI3+ZB5iDEuk4bWA3GkTpW+DOoZMYBVVg=
-go.opentelemetry.io/otel/sdk/metric v1.38.0 h1:aSH66iL0aZqo//xXzQLYozmWrXxyFkBJ6qT5wthqPoM=
-go.opentelemetry.io/otel/sdk/metric v1.38.0/go.mod h1:dg9PBnW9XdQ1Hd6ZnRz689CbtrUp0wMMs9iPcgT9EZA=
-go.opentelemetry.io/otel/trace v1.38.0 h1:Fxk5bKrDZJUH+AMyyIXGcFAPah0oRcT+LuNtJrmcNLE=
-go.opentelemetry.io/otel/trace v1.38.0/go.mod h1:j1P9ivuFsTceSWe1oY+EeW3sc+Pp42sO++GHkg4wwhs=
-go.opentelemetry.io/proto/otlp v1.8.0 h1:fRAZQDcAFHySxpJ1TwlA1cJ4tvcrw7nXl9xWWC8N5CE=
-go.opentelemetry.io/proto/otlp v1.8.0/go.mod h1:tIeYOeNBU4cvmPqpaji1P+KbB4Oloai8wN4rWzRrFF0=
+go.opentelemetry.io/contrib/detectors/gcp v1.39.0 h1:kWRNZMsfBHZ+uHjiH4y7Etn2FK26LAGkNFw7RHv1DhE=
+go.opentelemetry.io/contrib/detectors/gcp v1.39.0/go.mod h1:t/OGqzHBa5v6RHZwrDBJ2OirWc+4q/w2fTbLZwAKjTk=
+go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.68.0 h1:0Qx7VGBacMm9ZENQ7TnNObTYI4ShC+lHI16seduaxZo=
+go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.68.0/go.mod h1:Sje3i3MjSPKTSPvVWCaL8ugBzJwik3u4smCjUeuupqg=
+go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.68.0 h1:CqXxU8VOmDefoh0+ztfGaymYbhdB/tT3zs79QaZTNGY=
+go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.68.0/go.mod h1:BuhAPThV8PBHBvg8ZzZ/Ok3idOdhWIodywz2xEcRbJo=
+go.opentelemetry.io/otel v1.43.0 h1:mYIM03dnh5zfN7HautFE4ieIig9amkNANT+xcVxAj9I=
+go.opentelemetry.io/otel v1.43.0/go.mod h1:JuG+u74mvjvcm8vj8pI5XiHy1zDeoCS2LB1spIq7Ay0=
+go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.43.0 h1:88Y4s2C8oTui1LGM6bTWkw0ICGcOLCAI5l6zsD1j20k=
+go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.43.0/go.mod h1:Vl1/iaggsuRlrHf/hfPJPvVag77kKyvrLeD10kpMl+A=
+go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.43.0 h1:3iZJKlCZufyRzPzlQhUIWVmfltrXuGyfjREgGP3UUjc=
+go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.43.0/go.mod h1:/G+nUPfhq2e+qiXMGxMwumDrP5jtzU+mWN7/sjT2rak=
+go.opentelemetry.io/otel/exporters/stdout/stdoutmetric v1.43.0 h1:TC+BewnDpeiAmcscXbGMfxkO+mwYUwE/VySwvw88PfA=
+go.opentelemetry.io/otel/exporters/stdout/stdoutmetric v1.43.0/go.mod h1:J/ZyF4vfPwsSr9xJSPyQ4LqtcTPULFR64KwTikGLe+A=
+go.opentelemetry.io/otel/metric v1.43.0 h1:d7638QeInOnuwOONPp4JAOGfbCEpYb+K6DVWvdxGzgM=
+go.opentelemetry.io/otel/metric v1.43.0/go.mod h1:RDnPtIxvqlgO8GRW18W6Z/4P462ldprJtfxHxyKd2PY=
+go.opentelemetry.io/otel/sdk v1.43.0 h1:pi5mE86i5rTeLXqoF/hhiBtUNcrAGHLKQdhg4h4V9Dg=
+go.opentelemetry.io/otel/sdk v1.43.0/go.mod h1:P+IkVU3iWukmiit/Yf9AWvpyRDlUeBaRg6Y+C58QHzg=
+go.opentelemetry.io/otel/sdk/metric v1.43.0 h1:S88dyqXjJkuBNLeMcVPRFXpRw2fuwdvfCGLEo89fDkw=
+go.opentelemetry.io/otel/sdk/metric v1.43.0/go.mod h1:C/RJtwSEJ5hzTiUz5pXF1kILHStzb9zFlIEe85bhj6A=
+go.opentelemetry.io/otel/trace v1.43.0 h1:BkNrHpup+4k4w+ZZ86CZoHHEkohws8AY+WTX09nk+3A=
+go.opentelemetry.io/otel/trace v1.43.0/go.mod h1:/QJhyVBUUswCphDVxq+8mld+AvhXZLhe+8WVFxiFff0=
+go.opentelemetry.io/proto/otlp v1.10.0 h1:IQRWgT5srOCYfiWnpqUYz9CVmbO8bFmKcwYxpuCSL2g=
+go.opentelemetry.io/proto/otlp v1.10.0/go.mod h1:/CV4QoCR/S9yaPj8utp3lvQPoqMtxXdzn7ozvvozVqk=
 go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
 go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
 go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
 go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
 go.yaml.in/yaml/v2 v2.4.3 h1:6gvOSjQoTB3vt1l+CU+tSyi/HOjfOjRLJ4YwYZGwRO0=
 go.yaml.in/yaml/v2 v2.4.3/go.mod h1:zSxWcmIDjOzPXpjlTTbAsKokqkDNAVtZO0WOMiT90s8=
+go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
 golang.org/x/crypto v0.19.0/go.mod h1:Iy9bg/ha4yyC70EfRS8jz+B6ybOBKMaSxLj6P6oBDfU=
-golang.org/x/crypto v0.43.0 h1:dduJYIi3A3KOfdGOHX8AVZ/jGiyPa3IbBozJ5kNuE04=
-golang.org/x/crypto v0.43.0/go.mod h1:BFbav4mRNlXJL4wNeejLpWxB7wMbc79PdRGhWKncxR0=
+golang.org/x/crypto v0.50.0 h1:zO47/JPrL6vsNkINmLoo/PH1gcxpls50DNogFvB5ZGI=
+golang.org/x/crypto v0.50.0/go.mod h1:3muZ7vA7PBCE6xgPX7nkzzjiUq87kRItoJQM1Yo8S+Q=
 golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
 golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
 golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
@@ -297,15 +317,15 @@ golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v
 golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
 golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
 golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
-golang.org/x/net v0.46.0 h1:giFlY12I07fugqwPuWJi68oOnpfqFnJIJzaIIm2JVV4=
-golang.org/x/net v0.46.0/go.mod h1:Q9BGdFy1y4nkUwiLvT5qtyhAnEHgnQ/zd8PfU6nc210=
-golang.org/x/oauth2 v0.32.0 h1:jsCblLleRMDrxMN29H3z/k1KliIvpLgCkE6R8FXXNgY=
-golang.org/x/oauth2 v0.32.0/go.mod h1:lzm5WQJQwKZ3nwavOZ3IS5Aulzxi68dUSgRHujetwEA=
+golang.org/x/net v0.53.0 h1:d+qAbo5L0orcWAr0a9JweQpjXF19LMXJE8Ey7hwOdUA=
+golang.org/x/net v0.53.0/go.mod h1:JvMuJH7rrdiCfbeHoo3fCQU24Lf5JJwT9W3sJFulfgs=
+golang.org/x/oauth2 v0.36.0 h1:peZ/1z27fi9hUOFCAZaHyrpWG5lwe0RJEEEeH0ThlIs=
+golang.org/x/oauth2 v0.36.0/go.mod h1:YDBUJMTkDnJS+A4BP4eZBjCqtokkg1hODuPjwiGPO7Q=
 golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
-golang.org/x/sync v0.17.0 h1:l60nONMj9l5drqw6jlhIELNv9I0A4OFgRsG9k2oT9Ug=
-golang.org/x/sync v0.17.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
+golang.org/x/sync v0.20.0 h1:e0PTpb7pjO8GAtTs2dQ6jYa5BWYlMuX047Dco/pItO4=
+golang.org/x/sync v0.20.0/go.mod h1:9xrNwdLfx4jkKbNva9FpL6vEN7evnE43NNNJQ2LF3+0=
 golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210426230700-d19ff857e887/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
@@ -316,44 +336,44 @@ golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.17.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
-golang.org/x/sys v0.37.0 h1:fdNQudmxPjkdUTPnLn5mdQv7Zwvbvpaxqs831goi9kQ=
-golang.org/x/sys v0.37.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
+golang.org/x/sys v0.43.0 h1:Rlag2XtaFTxp19wS8MXlJwTvoh8ArU6ezoyFsMyCTNI=
+golang.org/x/sys v0.43.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
 golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
 golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
 golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
 golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo=
 golang.org/x/term v0.17.0/go.mod h1:lLRBjIVuehSbZlaOtGMbcMncT+aqLLLmKrsjNrUguwk=
-golang.org/x/term v0.36.0 h1:zMPR+aF8gfksFprF/Nc/rd1wRS1EI6nDBGyWAvDzx2Q=
-golang.org/x/term v0.36.0/go.mod h1:Qu394IJq6V6dCBRgwqshf3mPF85AqzYEzofzRdZkWss=
+golang.org/x/term v0.42.0 h1:UiKe+zDFmJobeJ5ggPwOshJIVt6/Ft0rcfrXZDLWAWY=
+golang.org/x/term v0.42.0/go.mod h1:Dq/D+snpsbazcBG5+F9Q1n2rXV8Ma+71xEjTRufARgY=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
 golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
 golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
 golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
 golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
-golang.org/x/text v0.30.0 h1:yznKA/E9zq54KzlzBEAWn1NXSQ8DIp/NYMy88xJjl4k=
-golang.org/x/text v0.30.0/go.mod h1:yDdHFIX9t+tORqspjENWgzaCVXgk0yYnYuSZ8UzzBVM=
-golang.org/x/time v0.12.0 h1:ScB/8o8olJvc+CQPWrK3fPZNfh7qgwCrY0zJmoEQLSE=
-golang.org/x/time v0.12.0/go.mod h1:CDIdPxbZBQxdj6cxyCIdrNogrJKMJ7pr37NYpMcMDSg=
+golang.org/x/text v0.36.0 h1:JfKh3XmcRPqZPKevfXVpI1wXPTqbkE5f7JA92a55Yxg=
+golang.org/x/text v0.36.0/go.mod h1:NIdBknypM8iqVmPiuco0Dh6P5Jcdk8lJL0CUebqK164=
+golang.org/x/time v0.15.0 h1:bbrp8t3bGUeFOx08pvsMYRTCVSMk89u4tKbNOZbp88U=
+golang.org/x/time v0.15.0/go.mod h1:Y4YMaQmXwGQZoFaVFk4YpCt4FLQMYKZe9oeV/f4MSno=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
 golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
 golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
 golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
-gonum.org/v1/gonum v0.16.0 h1:5+ul4Swaf3ESvrOnidPp4GZbzf0mxVQpDCYUQE7OJfk=
-gonum.org/v1/gonum v0.16.0/go.mod h1:fef3am4MQ93R2HHpKnLk4/Tbh/s0+wqD5nfa6Pnwy4E=
-google.golang.org/api v0.247.0 h1:tSd/e0QrUlLsrwMKmkbQhYVa109qIintOls2Wh6bngc=
-google.golang.org/api v0.247.0/go.mod h1:r1qZOPmxXffXg6xS5uhx16Fa/UFY8QU/K4bfKrnvovM=
-google.golang.org/genproto v0.0.0-20250603155806-513f23925822 h1:rHWScKit0gvAPuOnu87KpaYtjK5zBMLcULh7gxkCXu4=
-google.golang.org/genproto v0.0.0-20250603155806-513f23925822/go.mod h1:HubltRL7rMh0LfnQPkMH4NPDFEWp0jw3vixw7jEM53s=
-google.golang.org/genproto/googleapis/api v0.0.0-20251014184007-4626949a642f h1:OiFuztEyBivVKDvguQJYWq1yDcfAHIID/FVrPR4oiI0=
-google.golang.org/genproto/googleapis/api v0.0.0-20251014184007-4626949a642f/go.mod h1:kprOiu9Tr0JYyD6DORrc4Hfyk3RFXqkQ3ctHEum3ZbM=
-google.golang.org/genproto/googleapis/rpc v0.0.0-20251014184007-4626949a642f h1:1FTH6cpXFsENbPR5Bu8NQddPSaUUE6NA2XdZdDSAJK4=
-google.golang.org/genproto/googleapis/rpc v0.0.0-20251014184007-4626949a642f/go.mod h1:7i2o+ce6H/6BluujYR+kqX3GKH+dChPTQU19wjRPiGk=
-google.golang.org/grpc v1.76.0 h1:UnVkv1+uMLYXoIz6o7chp59WfQUYA2ex/BXQ9rHZu7A=
-google.golang.org/grpc v1.76.0/go.mod h1:Ju12QI8M6iQJtbcsV+awF5a4hfJMLi4X0JLo94ULZ6c=
-google.golang.org/protobuf v1.36.10 h1:AYd7cD/uASjIL6Q9LiTjz8JLcrh/88q5UObnmY3aOOE=
-google.golang.org/protobuf v1.36.10/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
+gonum.org/v1/gonum v0.17.0 h1:VbpOemQlsSMrYmn7T2OUvQ4dqxQXU+ouZFQsZOx50z4=
+gonum.org/v1/gonum v0.17.0/go.mod h1:El3tOrEuMpv2UdMrbNlKEh9vd86bmQ6vqIcDwxEOc1E=
+google.golang.org/api v0.274.0 h1:aYhycS5QQCwxHLwfEHRRLf9yNsfvp1JadKKWBE54RFA=
+google.golang.org/api v0.274.0/go.mod h1:JbAt7mF+XVmWu6xNP8/+CTiGH30ofmCmk9nM8d8fHew=
+google.golang.org/genproto v0.0.0-20260319201613-d00831a3d3e7 h1:XzmzkmB14QhVhgnawEVsOn6OFsnpyxNPRY9QV01dNB0=
+google.golang.org/genproto v0.0.0-20260319201613-d00831a3d3e7/go.mod h1:L43LFes82YgSonw6iTXTxXUX1OlULt4AQtkik4ULL/I=
+google.golang.org/genproto/googleapis/api v0.0.0-20260401024825-9d38bb4040a9 h1:VPWxll4HlMw1Vs/qXtN7BvhZqsS9cdAittCNvVENElA=
+google.golang.org/genproto/googleapis/api v0.0.0-20260401024825-9d38bb4040a9/go.mod h1:7QBABkRtR8z+TEnmXTqIqwJLlzrZKVfAUm7tY3yGv0M=
+google.golang.org/genproto/googleapis/rpc v0.0.0-20260406210006-6f92a3bedf2d h1:wT2n40TBqFY6wiwazVK9/iTWbsQrgk5ZfCSVFLO9LQA=
+google.golang.org/genproto/googleapis/rpc v0.0.0-20260406210006-6f92a3bedf2d/go.mod h1:4Hqkh8ycfw05ld/3BWL7rJOSfebL2Q+DVDeRgYgxUU8=
+google.golang.org/grpc v1.80.0 h1:Xr6m2WmWZLETvUNvIUmeD5OAagMw3FiKmMlTdViWsHM=
+google.golang.org/grpc v1.80.0/go.mod h1:ho/dLnxwi3EDJA4Zghp7k2Ec1+c2jqup0bFkw07bwF4=
+google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE=
+google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=

hal/disk/Config.in

@@ -0,0 +1,9 @@
+source "$BR2_EXTERNAL_COCOS_PATH/package/agent/Config.in"
+source "$BR2_EXTERNAL_COCOS_PATH/package/attestation-service/Config.in"
+source "$BR2_EXTERNAL_COCOS_PATH/package/cc-attestation-agent/Config.in"
+source "$BR2_EXTERNAL_COCOS_PATH/package/coco-keyprovider/Config.in"
+source "$BR2_EXTERNAL_COCOS_PATH/package/wasmedge/Config.in"
+source "$BR2_EXTERNAL_COCOS_PATH/package/log-forwarder/Config.in"
+source "$BR2_EXTERNAL_COCOS_PATH/package/computation-runner/Config.in"
+source "$BR2_EXTERNAL_COCOS_PATH/package/egress-proxy/Config.in"
+source "$BR2_EXTERNAL_COCOS_PATH/package/ingress-proxy/Config.in"

hal/disk/README.md

@@ -0,0 +1,214 @@
+# Disk Image Workflow
+
+This directory is the Buildroot external tree for the current Cocos disk test
+VM image and its runtime configuration.
+
+## Layout
+
+- [configs/cocos_defconfig](./configs/cocos_defconfig):
+  Buildroot configuration for the bootable image.
+- [board/rootfs-overlay/init](./board/rootfs-overlay/init):
+  early initramfs script that provisions `/cocos`, mounts the real root, and
+  switches into the installed system.
+- [board/cocos/genimage.cfg](./board/cocos/genimage.cfg):
+  GPT disk layout for the final `disk.img`.
+- [board/cocos/post-image.sh](./board/cocos/post-image.sh):
+  builds the minimal initramfs, stages EFI files, signs boot artifacts, and
+  assembles `disk.img`.
+- [external.desc](./external.desc): Buildroot external tree descriptor.
+- [external.mk](./external.mk): includes package makefiles from `package/*`.
+
+## Current Buildroot Image
+
+The current Buildroot flow produces a bootable GPT disk image:
+
+- `efi` partition: FAT EFI system partition with GRUB, kernel, and initramfs
+- `root` partition: ext4 root filesystem protected by dm-verity
+- `verity` partition: dm-verity hash tree for the root filesystem
+- `cocos` partition: blank partition provisioned at boot as an encrypted ext4
+  filesystem mounted at `/cocos`
+
+The final image is written to:
+
+```bash
+output/images/disk.img
+```
+
+The root filesystem image is also available separately as:
+
+```bash
+output/images/rootfs.ext4
+```
+
+## Current Boot Flow
+
+At boot, GRUB loads:
+
+- `bzImage`
+- `initrd.cpio.gz`
+
+The initramfs script in
+[board/rootfs-overlay/init](./board/rootfs-overlay/init)
+then:
+
+1. mounts `/proc`, `/sys`, `devtmpfs`, and `devpts`
+2. assumes the boot disk is `/dev/sda`
+3. opens a dm-verity mapping for the root filesystem using:
+   - `/dev/sda2` as the data partition
+   - `/dev/sda3` as the verity hash partition
+   - `roothash=` from the kernel command line
+4. mounts `/dev/mapper/root_verity` read-only at `/root`
+5. generates a fresh ephemeral key
+6. formats `/dev/sda4` as LUKS2
+7. opens it as `/dev/mapper/cocos_crypt`
+8. formats that mapper as ext4 and mounts it at `/root/cocos`
+9. creates working directories on `/cocos`, including:
+   - `/cocos/.cache/oci`
+   - `/cocos/datasets`
+   - `/cocos/docker`
+   - `/cocos/cocos_init`
+10. mounts `tmpfs` on `/tmp` and `/var` because the root filesystem is
+   intentionally read-only
+11. bind-mounts `/cocos/docker` onto `/var/lib/docker`
+12. bind-mounts `/cocos/cocos_init` onto `/cocos_init`
+13. rewrites `/etc/fstab` in the mounted root to describe the live runtime
+14. preserves or adds 9P mounts for:
+    - `certs_share` -> `/etc/certs`
+    - `env_share` -> `/etc/cocos`
+15. securely wipes the temporary LUKS key file
+16. runs `switch_root /root /sbin/init`
+
+Important details:
+
+- the root filesystem is verified through dm-verity before it is mounted
+- `/cocos` is encrypted with an ephemeral per-boot key
+- that key is not persisted, so `/cocos` is provisioned fresh on each boot
+
+## Runtime Filesystem Model
+
+The running system is split into:
+
+- read-only root on `/`
+- encrypted writable storage on `/cocos`
+- `tmpfs` on `/tmp`
+- `tmpfs` on `/var`
+
+Service state that must survive within a boot session is redirected away from
+the read-only root:
+
+- Docker data lives on `/cocos/docker`
+- agent setup scripts work through `/cocos_init`, which is backed by
+  `/cocos/cocos_init`
+- algorithm datasets and results live under `/cocos`
+
+This means services can use `/cocos` like a regular directory tree after boot,
+even though it is backed by an encrypted mapper created in early userspace.
+
+## systemd Runtime Expectations
+
+Several services depend on files mounted from 9P shares under `/etc/certs` and
+`/etc/cocos`. To avoid boot-order races, the rootfs overlay includes systemd
+drop-ins under:
+
+```bash
+board/rootfs-overlay/usr/lib/systemd/system/*service.d/
+```
+
+These drop-ins require the relevant mount points before starting services such
+as:
+
+- `egress-proxy.service`
+- `log-forwarder.service`
+- `computation-runner.service`
+- `cocos-agent.service`
+
+The overlay also ships tmpfiles rules in
+[board/rootfs-overlay/usr/lib/tmpfiles.d/cocos.conf](./board/rootfs-overlay/usr/lib/tmpfiles.d/cocos.conf)
+to create:
+
+- `/var/log/cocos`
+- `/run/cocos`
+
+## Agent Packaging In Buildroot
+
+The Buildroot `agent` package is wired to build the binary from the local Cocos
+checkout, not only from a downloaded release snapshot. The package definition is
+in [package/agent/agent.mk](./package/agent/agent.mk).
+
+That package currently:
+
+- builds `cocos-agent` from the local source tree
+- installs the local
+  [cocos-agent.service](../../init/systemd/cocos-agent.service)
+- installs the local
+  [agent_setup.sh](../../init/systemd/agent_setup.sh)
+- installs the local
+  [agent_start_script.sh](../../init/systemd/agent_start_script.sh)
+
+So changes under:
+
+- `cocos/agent/...`
+- `cocos/init/systemd/...`
+
+are intended to be picked up by the next Buildroot rebuild.
+
+## Buildroot Packages And Tools
+
+The current `cocos_defconfig` includes the components needed by the boot flow
+and runtime image, including:
+
+- systemd
+- DHCP client
+- `cryptsetup`
+- `eudev`
+- `e2fsprogs`
+- Docker, containerd, and runc
+- `skopeo`
+- TPM2 tools
+- 9P filesystem support
+- GRUB2 EFI boot support
+- host `genimage`
+
+The initramfs built in `post-image.sh` is intentionally minimal and contains
+only the binaries needed for early boot, dm-verity root verification, and
+`/cocos` provisioning.
+
+## Secure Boot Notes
+
+During `post-image.sh`:
+
+- GRUB is rebuilt with `--disable-shim-lock`
+- `bootx64.efi` and `bzImage` are signed with the configured Secure Boot keys
+  when those keys are present
+
+This flow is designed for booting directly through OVMF with your own enrolled
+keys. It does not currently rely on booting through `shim`.
+
+## Rebuilding
+
+This directory is meant to be used as a Buildroot external tree. From this
+directory, configure a Buildroot checkout with:
+
+```bash
+make -C /path/to/buildroot BR2_EXTERNAL=$PWD cocos_defconfig
+```
+
+Then build with:
+
+```bash
+make -C /path/to/buildroot BR2_EXTERNAL=$PWD -j$(nproc)
+```
+
+The resulting boot image is:
+
+```bash
+/path/to/buildroot/output/images/disk.img
+```
+
+Additional generated artifacts include:
+
+```bash
+/path/to/buildroot/output/images/rootfs.ext4
+/path/to/buildroot/output/images/rootfs.verity
+/path/to/buildroot/output/images/rootfs.roothash
+```

hal/disk/board/cocos/genimage.cfg

@@ -0,0 +1,42 @@
+image efi-part.vfat {
+    vfat {
+        file EFI {
+            image = "efi-part/EFI"
+        }
+        file bzImage {
+            image = "efi-part/bzImage"
+        }
+        file initrd.cpio.gz {
+            image = "efi-part/initrd.cpio.gz"
+        }
+    }
+    size = 256M
+}
+
+image disk.img {
+    hdimage {
+        partition-table-type = "gpt"
+    }
+
+    partition efi {
+        image = "efi-part.vfat"
+        partition-type-uuid = "C12A7328-F81F-11D2-BA4B-00A0C93EC93B"
+        offset = 1M
+        bootable = true
+    }
+
+    partition root {
+        image = "rootfs.ext4"
+        partition-type-uuid = "0FC63DAF-8483-4772-8E79-3D69D8477DE4"
+    }
+
+    partition verity {
+        image = "rootfs.verity"
+        partition-type-uuid = "0FC63DAF-8483-4772-8E79-3D69D8477DE4"
+    }
+
+    partition cocos {
+        size = "20480M"
+        partition-type-uuid = "0FC63DAF-8483-4772-8E79-3D69D8477DE4"
+    }
+}

hal/disk/board/cocos/linux.config

@@ -0,0 +1,279 @@
+###
+# Architecture / base
+###
+CONFIG_SYSVIPC=y
+CONFIG_SMP=y
+CONFIG_EXPERT=y
+CONFIG_LOCALVERSION_AUTO=n
+
+###
+# Modules
+###
+CONFIG_MODULES=y
+CONFIG_MODULE_UNLOAD=y
+
+###
+# Virtualization
+###
+CONFIG_HYPERVISOR_GUEST=y
+CONFIG_PARAVIRT=y
+CONFIG_VIRTUALIZATION=y
+CONFIG_KVM=y
+CONFIG_KVM_SW_PROTECTED_VM=y
+CONFIG_KVM_INTEL=y
+CONFIG_VIRT_DRIVERS=y
+
+###
+# Cgroups — base + Docker/container subsystems
+###
+CONFIG_CGROUPS=y
+CONFIG_CGROUP_CPUACCT=y
+CONFIG_CGROUP_DEVICE=y
+CONFIG_CGROUP_FREEZER=y
+CONFIG_CGROUP_MISC=y
+CONFIG_CGROUP_PIDS=y
+CONFIG_CGROUP_BPF=y
+CONFIG_CGROUP_NET_PRIO=y
+CONFIG_CGROUP_NET_CLASSID=y
+CONFIG_CPUSETS=y
+CONFIG_MEMCG=y
+CONFIG_BLK_CGROUP=y
+
+###
+# Namespaces — required by containerd / runc
+###
+CONFIG_NAMESPACES=y
+CONFIG_UTS_NS=y
+CONFIG_IPC_NS=y
+CONFIG_USER_NS=y
+CONFIG_PID_NS=y
+CONFIG_NET_NS=y
+
+###
+# PCI
+###
+CONFIG_PCI=y
+CONFIG_PCI_MSI=y
+CONFIG_IRQ_REMAP=y
+
+###
+# Initramfs
+###
+CONFIG_BLK_DEV_INITRD=y
+CONFIG_RD_GZIP=y
+
+###
+# Block devices
+###
+CONFIG_DEVTMPFS=y
+CONFIG_DEVTMPFS_MOUNT=y
+CONFIG_BLK_DEV_SD=y
+CONFIG_SCSI_VIRTIO=y
+CONFIG_ATA=y
+CONFIG_ATA_PIIX=y
+CONFIG_VIRTIO_BLK=y
+
+# Loop device (used by containerd image mounts)
+CONFIG_BLK_DEV_LOOP=y
+CONFIG_BLK_DEV_LOOP_MIN_COUNT=8
+
+###
+# Device mapper — FDE, dm-verity, dm-crypt, dm-integrity
+# These must be built-in (y) because they are needed before the
+# rootfs is mounted, during the initramfs FDE init stage.
+###
+CONFIG_MD=y
+CONFIG_BLK_DEV_DM_BUILTIN=y
+CONFIG_BLK_DEV_DM=y
+CONFIG_DM_CRYPT=y
+CONFIG_DM_VERITY=y
+CONFIG_DM_VERITY_VERIFY_ROOTHASH_SIG=y
+# CONFIG_DM_VERITY_FEC is not set
+CONFIG_DM_INTEGRITY=y
+CONFIG_DM_INIT=y
+
+###
+# Networking — base
+###
+CONFIG_NET=y
+CONFIG_PACKET=y
+CONFIG_UNIX=y
+CONFIG_INET=y
+# CONFIG_WIRELESS is not set
+CONFIG_NETDEVICES=y
+CONFIG_VIRTIO_NET=y
+CONFIG_NE2K_PCI=y
+CONFIG_8139CP=y
+# CONFIG_WLAN is not set
+CONFIG_VSOCKETS=y
+CONFIG_VIRTIO_VSOCKETS=y
+
+# Virtual Ethernet pairs and bridge (Docker networking)
+CONFIG_VETH=m
+CONFIG_BRIDGE=m
+CONFIG_BRIDGE_NETFILTER=m
+
+###
+# Netfilter — Docker NAT, iptables, conntrack (modules, loaded on demand)
+###
+CONFIG_NETFILTER=y
+CONFIG_NETFILTER_ADVANCED=y
+CONFIG_NF_CONNTRACK=m
+CONFIG_NF_CONNTRACK_MARK=y
+CONFIG_NF_NAT=m
+CONFIG_NF_NAT_MASQUERADE=y
+CONFIG_NF_TABLES=y
+CONFIG_IP_NF_IPTABLES=m
+CONFIG_IP_NF_FILTER=m
+CONFIG_IP_NF_TARGET_MASQUERADE=m
+CONFIG_NETFILTER_XT_MATCH_ADDRTYPE=m
+CONFIG_NETFILTER_XT_MATCH_CONNTRACK=m
+
+###
+# BPF
+###
+CONFIG_BPF_SYSCALL=y
+
+###
+# Filesystems
+###
+CONFIG_EXT4_FS=y
+CONFIG_OVERLAY_FS=y
+CONFIG_AUTOFS4_FS=y
+CONFIG_TMPFS=y
+CONFIG_TMPFS_POSIX_ACL=y
+CONFIG_PROC_FS=y
+CONFIG_SYSFS=y
+
+###
+# 9P filesystem (virtio shares for certs and env)
+###
+CONFIG_NET_9P=y
+CONFIG_NET_9P_VIRTIO=y
+CONFIG_9P_FS=y
+CONFIG_9P_FS_POSIX_ACL=y
+CONFIG_9P_FS_SECURITY=y
+
+###
+# Virtio devices
+###
+CONFIG_VIRTIO_PCI=y
+CONFIG_VIRTIO_BALLOON=y
+CONFIG_VIRTIO_INPUT=y
+CONFIG_VIRTIO_CONSOLE=y
+CONFIG_VIRTIO_MMIO=y
+CONFIG_VIRTIO_MMIO_CMDLINE_DEVICES=y
+CONFIG_HW_RANDOM_VIRTIO=m
+
+###
+# Console / Input
+###
+CONFIG_INPUT_EVDEV=y
+CONFIG_SERIAL_8250=y
+CONFIG_SERIAL_8250_CONSOLE=y
+
+###
+# Kernel features required by systemd
+###
+CONFIG_FHANDLE=y
+CONFIG_INOTIFY_USER=y
+CONFIG_SIGNALFD=y
+CONFIG_TIMERFD=y
+CONFIG_EPOLL=y
+CONFIG_POSIX_MQUEUE=y
+CONFIG_POSIX_MQUEUE_SYSCTL=y
+CONFIG_UNWINDER_FRAME_POINTER=y
+
+###
+# Security
+###
+CONFIG_SECCOMP=y
+CONFIG_SECCOMP_FILTER=y
+CONFIG_SECURITY=y
+CONFIG_SECURITYFS=y
+
+###
+# EFI
+###
+CONFIG_EFI=y
+CONFIG_EFI_STUB=y
+
+###
+# AMD SEV-SNP
+###
+CONFIG_AMD_MEM_ENCRYPT=y
+CONFIG_AMD_MEM_ENCRYPT_ACTIVE_BY_DEFAULT=n
+CONFIG_SEV_GUEST=y
+CONFIG_IOMMU_DEFAULT_PASSTHROUGH=n
+
+###
+# Intel TDX
+###
+CONFIG_X86_X2APIC=y
+CONFIG_X86_CPUID=y
+CONFIG_X86_SGX=y
+CONFIG_X86_SGX_KVM=y
+CONFIG_INTEL_TDX_GUEST=y
+CONFIG_TDX_GUEST_DRIVER=y
+
+###
+# Preemption (disabled for VM performance)
+###
+CONFIG_PREEMPT_COUNT=n
+CONFIG_PREEMPT=n
+CONFIG_PREEMPT_DYNAMIC=n
+CONFIG_DEBUG_PREEMPT=n
+
+###
+# Key/signature management
+###
+CONFIG_SYSTEM_TRUSTED_KEYS=n
+CONFIG_SYSTEM_REVOCATION_KEYS=n
+CONFIG_MODULE_SIG_KEY=n
+CONFIG_KEYS=y
+CONFIG_ENCRYPTED_KEYS=y
+
+###
+# Crypto — AES-GCM (LUKS2 cipher) + SHA-256 (dm-verity hash)
+###
+CONFIG_CRYPTO_AES=y
+CONFIG_CRYPTO_SHA256=y
+CONFIG_CRYPTO_GCM=y
+CONFIG_CRYPTO_GHASH=y
+CONFIG_CRYPTO_SEQIV=y
+CONFIG_CRYPTO_ECHAINIV=y
+CONFIG_CRYPTO_XTS=y
+CONFIG_CRYPTO_CBC=y
+CONFIG_CRYPTO_AUTHENC=y
+CONFIG_CRYPTO_ESSIV=y
+CONFIG_CRYPTO_USER_API=y
+CONFIG_CRYPTO_USER_API_HASH=y
+CONFIG_CRYPTO_USER_API_SKCIPHER=y
+CONFIG_CRYPTO_USER_API_AEAD=y
+CONFIG_CRYPTO_AES_NI_INTEL=m
+CONFIG_CRYPTO_GHASH_CLMUL_NI_INTEL=m
+
+###
+# TPM
+###
+CONFIG_TCG_TPM=y
+CONFIG_TCG_TPM2_HMAC=y
+CONFIG_TCG_PLATFORM=y
+
+###
+# IMA (Linux Integrity Measurement Architecture)
+###
+CONFIG_INTEGRITY=y
+CONFIG_INTEGRITY_SIGNATURE=y
+CONFIG_IMA=y
+CONFIG_IMA_MEASURE_PCR_IDX=10
+CONFIG_IMA_LSM_RULES=y
+CONFIG_IMA_APPRAISE=y
+CONFIG_IMA_DEFAULT_TEMPLATE="ima-ng"
+CONFIG_IMA_DEFAULT_HASH="sha256"
+
+###
+# Disabled options
+###
+CONFIG_KSM=n
+CONFIG_EISA=n

hal/disk/board/cocos/post-build.sh

@@ -0,0 +1,11 @@
+#!/bin/sh
+
+set -u
+set -e
+
+# Add a console on tty1
+if [ -e ${TARGET_DIR}/etc/inittab ]; then
+    grep -qE '^tty1::' ${TARGET_DIR}/etc/inittab || \
+    sed -i '/GENERIC_SERIAL/a\
+tty1::respawn:/sbin/getty -L  tty1 0 vt100 # QEMU graphical window' ${TARGET_DIR}/etc/inittab
+fi