Compare commits
1 Commits
feat/nixos
...
feat/restr
| Author | SHA1 | Date | |
|---|---|---|---|
| 9459839d74 |
@@ -1,64 +0,0 @@
|
||||
name: NixOS Build & Test
|
||||
run-name: Build ${{ gitea.event_name == 'push' && gitea.ref_name || format('PR #{0}', gitea.event.pull_request.number) }}
|
||||
|
||||
on:
|
||||
push:
|
||||
branches:
|
||||
- master
|
||||
paths:
|
||||
- '**.nix'
|
||||
- 'flake.lock'
|
||||
- 'secrets/**'
|
||||
- 'hosts/**'
|
||||
- 'modules/**'
|
||||
pull_request:
|
||||
branches:
|
||||
- master
|
||||
paths:
|
||||
- '**.nix'
|
||||
- 'flake.lock'
|
||||
- 'secrets/**'
|
||||
- 'hosts/**'
|
||||
- 'modules/**'
|
||||
|
||||
jobs:
|
||||
build:
|
||||
runs-on: nixos-builder
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 0
|
||||
|
||||
- name: Set up Nix environment
|
||||
run: |
|
||||
echo "extra-experimental-features = nix-command flakes" >> ~/.config/nix/nix.conf
|
||||
cat ~/.config/nix/nix.conf
|
||||
|
||||
- name: Install nh (nix helper)
|
||||
run: |
|
||||
nix --extra-experimental-features "nix-command flakes" \
|
||||
profile add nixpkgs#nh
|
||||
nh --version
|
||||
|
||||
- name: Build NixOS configuration (lazyworkhorse)
|
||||
run: |
|
||||
nh os build .#lazyworkhorse
|
||||
env:
|
||||
NIX_CONFIG: "extra-experimental-features = nix-command flakes"
|
||||
|
||||
- name: Build NixOS configuration (cyt-pi)
|
||||
run: |
|
||||
nh os build .#cyt-pi
|
||||
env:
|
||||
NIX_CONFIG: "extra-experimental-features = nix-command flakes"
|
||||
|
||||
- name: Integration tests (placeholder)
|
||||
run: |
|
||||
echo "TODO: Add integration tests here"
|
||||
echo ""
|
||||
echo "Suggested future checks:"
|
||||
echo " - nix flake check (evaluate all NixOS configs)"
|
||||
echo " - Validate agenix secrets are decryptable"
|
||||
echo " - Check services are defined correctly"
|
||||
echo " - Run VM test if nixos-test infrastructure exists"
|
||||
@@ -59,7 +59,6 @@
|
||||
./modules/nixos/filesystem/hoardingcow-mount.nix
|
||||
./modules/nixos/services/docker_manager.nix
|
||||
./modules/nixos/services/open_code_server.nix
|
||||
./modules/nixos/services/staging-vm.nix
|
||||
./modules/nixos/services/ollama_init_custom_models.nix
|
||||
./modules/nixos/services/openclaw_node.nix
|
||||
./modules/nixos/security/ai-worker-restricted.nix
|
||||
|
||||
@@ -1,10 +1,44 @@
|
||||
# AI Worker Restricted Access
|
||||
|
||||
This module provides SSH access for the AI worker (hermes-agent) to run ollama benchmarks on the host.
|
||||
This module provides SSH access for the AI worker (hermes-agent) to run docker commands on the host.
|
||||
|
||||
## Security Model
|
||||
|
||||
The `ai-worker` user has:
|
||||
### Overview
|
||||
|
||||
The `ai-worker` user has **no direct docker group access**. All docker commands must go through `sudo`, and only specific subcommands are whitelisted:
|
||||
|
||||
- **Container lifecycle**: `docker ps`, `docker inspect`, `docker logs`, `docker images`, `docker info`, `docker version`, `docker stats`
|
||||
- **Control**: `docker start`, `docker stop`, `docker restart`, `docker rm`, `docker rmi`, `docker wait`
|
||||
- **Image management**: `docker pull`, `docker build`, `docker run`, `docker compose`
|
||||
- **Disk cleanup**: `docker system`
|
||||
- **Network/Volume**: `docker network ls`, `docker volume ls` (read-only)
|
||||
|
||||
### EXPLICITLY BLOCKED (not in sudo whitelist)
|
||||
|
||||
| Command | Risk | Result |
|
||||
|---------|------|--------|
|
||||
| `docker exec` | Execute arbitrary commands inside containers (FILE MODIFICATION) | Blocked by sudo |
|
||||
| `docker cp` | Copy files between containers and host | Blocked by sudo |
|
||||
| `docker commit` | Create images from running containers (data exfil) | Blocked by sudo |
|
||||
| `docker diff` | Inspect filesystem changes | Blocked by sudo |
|
||||
| `docker export` | Export container filesystem | Blocked by sudo |
|
||||
| `docker import` | Import filesystem archives | Blocked by sudo |
|
||||
| `docker load` | Load docker images | Blocked by sudo |
|
||||
| `docker save` | Save docker images to tar | Blocked by sudo |
|
||||
| `docker attach` | Interactive access to containers | Blocked by sudo |
|
||||
| `docker push` | Push images to registries | Blocked by sudo |
|
||||
| `docker tag` | Rename images | Blocked by sudo |
|
||||
|
||||
### Why This Approach?
|
||||
|
||||
Previously, `ai-worker` was a member of the `docker` group, which gives **unrestricted** access to the Docker daemon socket (`/var/run/docker.sock`). Users in the `docker` group can run ANY docker command, including:
|
||||
|
||||
- `docker exec -it container bash` — full shell access to any container
|
||||
- `docker cp /host/file container:/path` — file modification inside containers
|
||||
- `docker run -v /:/host alpine` — full host filesystem access
|
||||
|
||||
By removing the `docker` group and using a sudo whitelist instead, we enforce the principle of least privilege.
|
||||
|
||||
### Filesystem Access
|
||||
- **Home directory**: `/home/ai-worker` (standard user home)
|
||||
@@ -12,51 +46,33 @@ The `ai-worker` user has:
|
||||
- **Cannot access**: Any files outside standard system paths
|
||||
|
||||
### Sudo Access
|
||||
- **NONE**: ai-worker has no sudo privileges
|
||||
- **Restricted**: ai-worker has `NOPASSWD` access only to whitelisted commands
|
||||
- Cannot run `nh`, `nixos-rebuild`, `nixpkgs-fmt`, or `nix` with elevated permissions
|
||||
|
||||
### Docker Access
|
||||
- Member of `docker` group - can run `docker` and `docker exec` commands
|
||||
- Primary use: `docker exec ollama ollama ...` for benchmarking
|
||||
- Can run `docker exec --privileged ollama rocm-smi ...` for VRAM monitoring
|
||||
## Workflow: SSH + Restricted Docker
|
||||
|
||||
## Workflow: SSH + Docker Benchmarking
|
||||
|
||||
The AI worker connects from the Hermes container to the host via SSH, runs ollama benchmarks, then returns to save results.
|
||||
|
||||
### Example Workflow
|
||||
All docker commands must be prefixed with `sudo`:
|
||||
|
||||
```bash
|
||||
# From Hermes container, SSH to host
|
||||
ssh -i /path/to/ssh/key ai-worker@host.docker.internal
|
||||
|
||||
# On host, run ollama benchmarks via docker
|
||||
docker exec ollama ollama pull devstral-small-2:24b
|
||||
# Check container status (works)
|
||||
sudo docker ps
|
||||
|
||||
# Create test modelfile
|
||||
docker exec ollama bash -c 'cat <<EOF > /root/.ollama/test.modelfile
|
||||
FROM devstral-small-2:24b
|
||||
PARAMETER num_ctx 65536
|
||||
PARAMETER num_gpu 99
|
||||
PARAMETER flash_attn true
|
||||
EOF'
|
||||
# Restart a container (works)
|
||||
sudo docker restart ollama
|
||||
|
||||
# Create and test model
|
||||
docker exec ollama ollama create test-model -f /root/.ollama/test.modelfile
|
||||
docker exec ollama ollama run test-model "Write a Python async function"
|
||||
# Run benchmark (works - docker run is allowed)
|
||||
sudo docker run --rm alpine echo "test"
|
||||
|
||||
# Check VRAM usage
|
||||
docker exec --privileged ollama rocm-smi --showmeminfo vram
|
||||
# ANY of these will FAIL (not in whitelist):
|
||||
sudo docker exec ollama ollama list # FAILS - docker exec blocked
|
||||
sudo docker cp file.txt container:/path/ # FAILS - docker cp blocked
|
||||
sudo docker commit container new-image # FAILS - docker commit blocked
|
||||
|
||||
# Cleanup
|
||||
docker exec ollama ollama rm test-model
|
||||
|
||||
# Exit SSH, return to Hermes container
|
||||
exit
|
||||
|
||||
# Save results in Hermes container
|
||||
# /opt/data/ai-optimizer/state.json
|
||||
# /opt/data/ai-optimizer/results.csv
|
||||
# For ollama operations, use the HTTP API instead of docker exec:
|
||||
curl http://ollama:11434/api/tags
|
||||
```
|
||||
|
||||
## SSH Access
|
||||
@@ -74,25 +90,30 @@ Check ai-worker permissions:
|
||||
```bash
|
||||
# On the host, as root or gortium:
|
||||
sudo -u ai-worker sudo -l
|
||||
# Should show: no sudo access
|
||||
# Should show the whitelisted commands only (no docker exec/cp/commit)
|
||||
|
||||
# Check docker group membership
|
||||
# Verify NOT in docker group
|
||||
groups ai-worker
|
||||
# Should show: ai-worker docker
|
||||
# Should show: ai-worker (NO docker group)
|
||||
```
|
||||
|
||||
## Troubleshooting
|
||||
|
||||
If ai-worker cannot run docker commands:
|
||||
If docker commands fail:
|
||||
|
||||
```bash
|
||||
# Check docker group membership
|
||||
# Check sudo permissions
|
||||
sudo -u ai-worker sudo -l | grep docker
|
||||
|
||||
# Verify group membership
|
||||
groups ai-worker
|
||||
|
||||
# Verify ollama container is running
|
||||
docker ps | grep ollama
|
||||
# Test allowed command
|
||||
sudo -u ai-worker sudo docker ps
|
||||
|
||||
# Test docker access
|
||||
sudo -u ai-worker docker exec ollama ollama list
|
||||
# Test blocked command (should fail)
|
||||
sudo -u ai-worker sudo docker exec ollama ollama list
|
||||
# Expected: "Sorry, user ai-worker is not allowed to execute"
|
||||
```
|
||||
|
||||
If SSH connection fails:
|
||||
|
||||
@@ -6,12 +6,19 @@ with lib;
|
||||
options.services.aiWorkerAccess = mkOption {
|
||||
type = types.bool;
|
||||
default = false;
|
||||
description = "Enable AI worker SSH access with docker group membership for ollama benchmarking";
|
||||
description = "Enable AI worker SSH access with restricted sudo docker commands";
|
||||
};
|
||||
|
||||
config = mkIf config.services.aiWorkerAccess {
|
||||
# ai-worker is member of docker group - can run docker commands via SSH
|
||||
# No bind mounts, no sudo access - docker-only for ollama benchmarking
|
||||
users.groups.docker.members = [ "ai-worker" ];
|
||||
# SECURITY: ai-worker is NOT added to docker group.
|
||||
# Docker access is granted via sudo whitelist in users/ai-worker.nix.
|
||||
# This prevents unrestricted docker daemon access (docker exec, cp, commit, etc.)
|
||||
# Only specific docker subcommands are allowed via sudo NOPASSWD rules.
|
||||
|
||||
# The old approach (docker group membership) has been removed because:
|
||||
# - Docker group gives UNRESTRICTED access to the docker daemon socket
|
||||
# - No way to limit which docker subcommands a docker group member can run
|
||||
# - Allowed: docker exec, docker cp, docker run -v /:/host, etc.
|
||||
# users.groups.docker.members = [ "ai-worker" ]; // REMOVED
|
||||
};
|
||||
}
|
||||
|
||||
@@ -1,415 +0,0 @@
|
||||
{ config, lib, pkgs, ... }:
|
||||
|
||||
with lib;
|
||||
|
||||
let
|
||||
cfg = config.services.staging-vm;
|
||||
|
||||
# Resolve the first IP in the subnet (the gateway address for the NAT network)
|
||||
networkIp = head (splitString "/" cfg.network.subnet);
|
||||
|
||||
# ── pr-test-vm helper script ──────────────────────────────────────────
|
||||
pr-test-vm = pkgs.writeShellScriptBin "pr-test-vm" ''
|
||||
set -euo pipefail
|
||||
|
||||
LIBVIRT_URI="qemu:///system"
|
||||
STORAGE_POOL="${cfg.storagePool}"
|
||||
VM_DIR="${cfg.dataDir}"
|
||||
NETWORK="${cfg.network.name}"
|
||||
SCRIPT_NAME="$(basename "$0")"
|
||||
|
||||
usage() {
|
||||
cat <<EOF
|
||||
Usage: $SCRIPT_NAME <command> [options]
|
||||
|
||||
Commands:
|
||||
build <nixos-config> [--name <name>] Build VM image from a NixOS config
|
||||
start <vm-name> Start a VM
|
||||
stop <vm-name> Gracefully shut down a VM
|
||||
destroy <vm-name> Force-power-off and undefine a VM
|
||||
ssh [user@]<vm-name> SSH into a running VM
|
||||
console <vm-name> Connect to VM serial console
|
||||
list List all staging VMs
|
||||
status <vm-name> Show VM status
|
||||
rebuild <vm-name> Redeploy the VM (stop + start)
|
||||
|
||||
Examples:
|
||||
$SCRIPT_NAME build ./vm-config.nix --name my-test
|
||||
$SCRIPT_NAME start my-test
|
||||
$SCRIPT_NAME ssh root@my-test
|
||||
EOF
|
||||
exit 1
|
||||
}
|
||||
|
||||
# Find the VM's IP address from the DHCP lease
|
||||
vm_ip() {
|
||||
local name="$1"
|
||||
local mac
|
||||
mac=$(${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" domiflist "$name" 2>/dev/null \
|
||||
| ${pkgs.gawk}/bin/awk 'NR>2 && $1 ~ /^vnet/ {print $NF; exit}')
|
||||
[ -z "$mac" ] && { echo "error: cannot find MAC for VM '$name'"; exit 1; }
|
||||
|
||||
local ip
|
||||
ip=$(${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" net-dhcp-leases "$NETWORK" 2>/dev/null \
|
||||
| ${pkgs.gawk}/bin/awk -v mac="$mac" '$0 ~ mac {gsub(/-.*/, "", $3); print $3; exit}')
|
||||
[ -z "$ip" ] && { echo "error: no DHCP lease found for VM '$name' (MAC: $mac)"; exit 1; }
|
||||
echo "$ip"
|
||||
}
|
||||
|
||||
case "''${1:-help}" in
|
||||
build)
|
||||
shift
|
||||
CONFIG="''${1:?Missing NixOS config path}"
|
||||
VM_NAME="''${2:-}"
|
||||
[ -f "$CONFIG" ] || { echo "error: config file not found: $CONFIG"; exit 1; }
|
||||
|
||||
# Extract name from --name flag or config basename
|
||||
if [ "''${2:-}" = "--name" ] && [ -n "''${3:-}" ]; then
|
||||
VM_NAME="$3"
|
||||
elif [ -z "$VM_NAME" ] || [ "''${VM_NAME#--}" != "$VM_NAME" ]; then
|
||||
VM_NAME="$(basename "$CONFIG" .nix)"
|
||||
fi
|
||||
|
||||
BUILD_DIR="$VM_DIR/$VM_NAME"
|
||||
IMAGE="$BUILD_DIR/disk-image.qcow2"
|
||||
|
||||
echo "==> Building VM '$VM_NAME' from config: $CONFIG"
|
||||
mkdir -p "$BUILD_DIR"
|
||||
|
||||
# Build the NixOS VM derivation
|
||||
nix build --no-link -f "$CONFIG" vm 2>&1 || {
|
||||
# Fallback: try as flake output
|
||||
echo "Trying flake build..."
|
||||
nix build "''${CONFIG%/.nix}#nixosConfigurations.$VM_NAME.config.system.build.vm" --no-link 2>&1 || {
|
||||
echo "error: failed to build VM (tried both import and flake)"
|
||||
exit 1
|
||||
}
|
||||
}
|
||||
|
||||
echo "==> Build complete. Run 'pr-test-vm start $VM_NAME' to launch."
|
||||
;;
|
||||
|
||||
start)
|
||||
VM_NAME="''${1:?Missing VM name}"
|
||||
IMAGE="$VM_DIR/$VM_NAME/disk-image.qcow2"
|
||||
[ -f "$IMAGE" ] || { echo "error: no disk image found at $IMAGE. Build first."; exit 1; }
|
||||
|
||||
# Check if already running
|
||||
STATE=$(${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" domstate "$VM_NAME" 2>/dev/null || echo "undefined")
|
||||
if [ "$STATE" = "running" ]; then
|
||||
echo "VM '$VM_NAME' is already running."
|
||||
exit 0
|
||||
fi
|
||||
|
||||
echo "==> Starting VM '$VM_NAME'..."
|
||||
|
||||
# Undefine if defined but not running
|
||||
if [ "$STATE" != "undefined" ]; then
|
||||
${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" undefine "$VM_NAME" 2>/dev/null || true
|
||||
fi
|
||||
|
||||
# Define and start with virt-install
|
||||
${pkgs.virt-manager}/bin/virt-install \
|
||||
--connect "$LIBVIRT_URI" \
|
||||
--name "$VM_NAME" \
|
||||
--memory "${toString cfg.memory}" \
|
||||
--vcpus "${toString cfg.vcpus}" \
|
||||
--disk "$IMAGE",bus=virtio \
|
||||
--import \
|
||||
--network network="$NETWORK",model=virtio \
|
||||
--graphics none \
|
||||
--console pty,target_type=virtio \
|
||||
--serial pty \
|
||||
--memballoon virtio \
|
||||
--rng /dev/urandom \
|
||||
--noautoconsole \
|
||||
--os-variant detect=on,name=generic
|
||||
|
||||
echo "==> VM '$VM_NAME' started. Get IP with: pr-test-vm status $VM_NAME"
|
||||
;;
|
||||
|
||||
stop)
|
||||
VM_NAME="''${1:?Missing VM name}"
|
||||
echo "==> Stoping VM '$VM_NAME'..."
|
||||
${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" shutdown "$VM_NAME" 2>/dev/null && {
|
||||
echo "Waiting for VM to shut down..."
|
||||
for i in $(seq 1 30); do
|
||||
STATE=$(${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" domstate "$VM_NAME" 2>/dev/null || echo "undefined")
|
||||
[ "$STATE" != "running" ] && { echo "VM stopped."; exit 0; }
|
||||
sleep 2
|
||||
done
|
||||
echo "warning: VM did not shut down gracefully, use 'destroy' for force"
|
||||
} || {
|
||||
echo "VM '$VM_NAME' not running or does not exist."
|
||||
}
|
||||
;;
|
||||
|
||||
destroy)
|
||||
VM_NAME="''${1:?Missing VM name}"
|
||||
echo "==> Destroying VM '$VM_NAME'..."
|
||||
${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" destroy "$VM_NAME" 2>/dev/null || true
|
||||
${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" undefine "$VM_NAME" 2>/dev/null || true
|
||||
echo "==> VM '$VM_NAME' destroyed and undefined."
|
||||
;;
|
||||
|
||||
ssh)
|
||||
TARGET="''${1:?Usage: $SCRIPT_NAME ssh [user@]<vm-name>}"
|
||||
# Split user@hostname if present
|
||||
if echo "$TARGET" | ${pkgs.gnugrep}/bin/grep -q '@'; then
|
||||
USER="''${TARGET%@*}"
|
||||
VM_NAME="''${TARGET#*@}"
|
||||
else
|
||||
VM_NAME="$TARGET"
|
||||
USER=""
|
||||
fi
|
||||
|
||||
IP=$(vm_ip "$VM_NAME") || exit 1
|
||||
if [ -n "$USER" ]; then
|
||||
exec ${pkgs.openssh}/bin/ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null "''${USER}@''${IP}"
|
||||
else
|
||||
exec ${pkgs.openssh}/bin/ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null "$IP"
|
||||
fi
|
||||
;;
|
||||
|
||||
console)
|
||||
VM_NAME="''${1:?Missing VM name}"
|
||||
exec ${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" console "$VM_NAME"
|
||||
;;
|
||||
|
||||
list)
|
||||
echo "Staging VMs:"
|
||||
${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" list --all
|
||||
echo ""
|
||||
echo "Active networks:"
|
||||
${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" net-list
|
||||
echo ""
|
||||
echo "Storage pools:"
|
||||
${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" pool-list
|
||||
;;
|
||||
|
||||
status)
|
||||
VM_NAME="''${1:?Missing VM name}"
|
||||
echo "VM: $VM_NAME"
|
||||
STATE=$(${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" domstate "$VM_NAME" 2>/dev/null || echo "not found")
|
||||
echo "State: $STATE"
|
||||
if [ "$STATE" = "running" ]; then
|
||||
IP=$(vm_ip "$VM_NAME" 2>/dev/null || echo "N/A")
|
||||
echo "IP: $IP"
|
||||
${pkgs.libvirt}/bin/virsh -c "$LIBVIRT_URI" dommemstat "$VM_NAME" 2>/dev/null | head -3 || true
|
||||
fi
|
||||
;;
|
||||
|
||||
rebuild)
|
||||
VM_NAME="''${1:?Missing VM name}"
|
||||
"$0" destroy "$VM_NAME"
|
||||
"$0" start "$VM_NAME"
|
||||
;;
|
||||
|
||||
help|--help|-h)
|
||||
usage
|
||||
;;
|
||||
|
||||
*)
|
||||
usage
|
||||
;;
|
||||
esac
|
||||
'';
|
||||
in
|
||||
{
|
||||
options.services.staging-vm = {
|
||||
enable = mkEnableOption "Staging VM infrastructure with libvirt/KVM";
|
||||
|
||||
network = {
|
||||
name = mkOption {
|
||||
type = types.str;
|
||||
default = "staging";
|
||||
description = "Name of the libvirt NAT network for staging VMs";
|
||||
};
|
||||
|
||||
bridge = mkOption {
|
||||
type = types.str;
|
||||
default = "virbr1";
|
||||
description = "Bridge interface name for the staging network";
|
||||
};
|
||||
|
||||
subnet = mkOption {
|
||||
type = types.str;
|
||||
default = "192.168.122.0/24";
|
||||
description = "NAT network subnet in CIDR notation";
|
||||
};
|
||||
|
||||
dhcpStart = mkOption {
|
||||
type = types.str;
|
||||
default = "192.168.122.2";
|
||||
description = "Start of the DHCP range";
|
||||
};
|
||||
|
||||
dhcpEnd = mkOption {
|
||||
type = types.str;
|
||||
default = "192.168.122.254";
|
||||
description = "End of the DHCP range";
|
||||
};
|
||||
};
|
||||
|
||||
storagePool = mkOption {
|
||||
type = types.str;
|
||||
default = "/var/lib/libvirt/images";
|
||||
description = "Directory path for the libvirt storage pool";
|
||||
};
|
||||
|
||||
dataDir = mkOption {
|
||||
type = types.str;
|
||||
default = "/var/lib/staging-vm";
|
||||
description = "Directory for staging VM test data (images, cloud-init configs)";
|
||||
};
|
||||
|
||||
memory = mkOption {
|
||||
type = types.int;
|
||||
default = 2048;
|
||||
description = "Default RAM in MiB for staging VMs";
|
||||
};
|
||||
|
||||
vcpus = mkOption {
|
||||
type = types.int;
|
||||
default = 2;
|
||||
description = "Default number of vCPUs for staging VMs";
|
||||
};
|
||||
};
|
||||
|
||||
config = mkIf cfg.enable {
|
||||
# ── libvirtd with QEMU/KVM ──────────────────────────────────────────
|
||||
virtualisation.libvirtd = {
|
||||
enable = true;
|
||||
|
||||
qemu = {
|
||||
package = pkgs.qemu_kvm;
|
||||
runAsRoot = true;
|
||||
|
||||
swtpm = {
|
||||
enable = true;
|
||||
};
|
||||
|
||||
ovmf = {
|
||||
enable = true;
|
||||
packages = [ pkgs.OVMF ];
|
||||
};
|
||||
};
|
||||
|
||||
# Allow the staging bridge for guest networking
|
||||
allowedBridges = [ cfg.network.bridge ];
|
||||
};
|
||||
|
||||
# ── System packages ─────────────────────────────────────────────────
|
||||
environment.systemPackages = with pkgs; [
|
||||
libvirt # virsh, virt-admin
|
||||
qemu_kvm # QEMU/KVM
|
||||
swtpm # Software TPM
|
||||
OVMF # UEFI firmware for VMs
|
||||
virt-manager # GUI management
|
||||
virt-viewer # SPICE/VNC viewer
|
||||
libguestfs # virt-install, virt-customize, guestfish
|
||||
cdrtools # genisoimage for cloud-init ISOs
|
||||
jq # JSON parsing
|
||||
gawk # awk for DHCP lease parsing
|
||||
gnugrep # grep
|
||||
];
|
||||
|
||||
# ── User permissions ────────────────────────────────────────────────
|
||||
users.users.gortium.extraGroups = [ "libvirtd" ];
|
||||
|
||||
# ── Directories ─────────────────────────────────────────────────────
|
||||
systemd.tmpfiles.rules = [
|
||||
"d ${cfg.storagePool} 0755 root root -"
|
||||
"d ${cfg.dataDir} 0755 root root -"
|
||||
];
|
||||
|
||||
# ── Default NAT network definition ──────────────────────────────────
|
||||
systemd.services.define-staging-network = {
|
||||
description = "Define the staging libvirt NAT network";
|
||||
after = [ "libvirtd.service" ];
|
||||
requires = [ "libvirtd.service" ];
|
||||
wantedBy = [ "multi-user.target" ];
|
||||
before = [ "define-staging-pool.service" ];
|
||||
serviceConfig = {
|
||||
Type = "oneshot";
|
||||
RemainAfterExit = true;
|
||||
};
|
||||
script = let
|
||||
networkXml = pkgs.writeText "staging-network.xml" ''
|
||||
<network>
|
||||
<name>${cfg.network.name}</name>
|
||||
<forward mode='nat'/>
|
||||
<bridge name='${cfg.network.bridge}' stp='on' delay='0'/>
|
||||
<ip address='${networkIp}' netmask='255.255.255.0'>
|
||||
<dhcp>
|
||||
<range start='${cfg.network.dhcpStart}' end='${cfg.network.dhcpEnd}'/>
|
||||
</dhcp>
|
||||
</ip>
|
||||
</network>
|
||||
'';
|
||||
in ''
|
||||
set -e
|
||||
${pkgs.libvirt}/bin/virsh -c qemu:///system net-info "${cfg.network.name}" 2>/dev/null && {
|
||||
echo "Network '${cfg.network.name}' already exists"
|
||||
} || {
|
||||
echo "Defining network '${cfg.network.name}'..."
|
||||
${pkgs.libvirt}/bin/virsh -c qemu:///system net-define "${networkXml}"
|
||||
}
|
||||
${pkgs.libvirt}/bin/virsh -c qemu:///system net-autostart "${cfg.network.name}"
|
||||
# Start the network if not already active
|
||||
STATE=$(${pkgs.libvirt}/bin/virsh -c qemu:///system net-state "${cfg.network.name}" 2>/dev/null || echo "inactive")
|
||||
if [ "$STATE" != "active" ]; then
|
||||
${pkgs.libvirt}/bin/virsh -c qemu:///system net-start "${cfg.network.name}"
|
||||
fi
|
||||
echo "Network '${cfg.network.name}' is ready."
|
||||
'';
|
||||
};
|
||||
|
||||
# ── Storage pool definition ─────────────────────────────────────────
|
||||
systemd.services.define-staging-pool = {
|
||||
description = "Define the staging libvirt storage pool";
|
||||
after = [ "libvirtd.service" "define-staging-network.service" ];
|
||||
requires = [ "libvirtd.service" ];
|
||||
wantedBy = [ "multi-user.target" ];
|
||||
serviceConfig = {
|
||||
Type = "oneshot";
|
||||
RemainAfterExit = true;
|
||||
};
|
||||
script = ''
|
||||
set -e
|
||||
${pkgs.libvirt}/bin/virsh -c qemu:///system pool-info staging 2>/dev/null && {
|
||||
echo "Storage pool 'staging' already exists"
|
||||
} || {
|
||||
echo "Defining storage pool 'staging' at ${cfg.storagePool}..."
|
||||
${pkgs.libvirt}/bin/virsh -c qemu:///system pool-define-as \
|
||||
--name staging --type dir --target "${cfg.storagePool}"
|
||||
}
|
||||
${pkgs.libvirt}/bin/virsh -c qemu:///system pool-autostart staging
|
||||
STATE=$(${pkgs.libvirt}/bin/virsh -c qemu:///system pool-state staging 2>/dev/null || echo "inactive")
|
||||
if [ "$STATE" != "running" ]; then
|
||||
${pkgs.libvirt}/bin/virsh -c qemu:///system pool-build staging
|
||||
${pkgs.libvirt}/bin/virsh -c qemu:///system pool-start staging
|
||||
fi
|
||||
echo "Storage pool 'staging' is ready."
|
||||
'';
|
||||
};
|
||||
|
||||
# ── Firewall rules for libvirt guests ───────────────────────────────
|
||||
networking.firewall = {
|
||||
# Trust the bridge interface to allow guest traffic
|
||||
trustedInterfaces = [ cfg.network.bridge ];
|
||||
|
||||
extraCommands = mkAfter ''
|
||||
# Allow forwarding between the staging bridge and the outside world
|
||||
iptables -I FORWARD -i ${cfg.network.bridge} -j ACCEPT 2>/dev/null || true
|
||||
iptables -I FORWARD -o ${cfg.network.bridge} -m conntrack --ctstate RELATED,ESTABLISHED -j ACCEPT 2>/dev/null || true
|
||||
|
||||
# NAT for guest outbound traffic
|
||||
iptables -t nat -I POSTROUTING -s ${cfg.network.subnet} -j MASQUERADE 2>/dev/null || true
|
||||
'';
|
||||
};
|
||||
|
||||
# ── pr-test-vm helper script ────────────────────────────────────────
|
||||
environment.systemPackages = [ pr-test-vm ];
|
||||
};
|
||||
}
|
||||
@@ -4,7 +4,9 @@
|
||||
group = "ai-worker";
|
||||
home = "/home/ai-worker";
|
||||
createHome = true;
|
||||
extraGroups = [ "docker" ];
|
||||
# SECURITY: ai-worker is NOT in the docker group.
|
||||
# Docker access is restricted via sudo whitelist — only specific subcommands allowed.
|
||||
# extraGroups = [ "docker" ]; — REMOVED: docker group gives unrestricted docker daemon access
|
||||
shell = pkgs.bashInteractive;
|
||||
openssh.authorizedKeys.keys = [
|
||||
keys.users.ai-worker.main
|
||||
@@ -17,19 +19,134 @@
|
||||
# Enable restricted AI worker SSH access for ollama benchmarking
|
||||
# SECURITY: ai-worker can only:
|
||||
# - SSH into host from Hermes container
|
||||
# - Run docker commands (docker exec ollama ...) via docker group
|
||||
# - Run docker commands via sudo (whitelist below — no exec/cp/commit)
|
||||
# - Run specific security audit commands
|
||||
# - NO access to infra repo (no bind mount)
|
||||
# - NO sudo access (no nh, nixos-rebuild, nixpkgs-fmt, nix)
|
||||
# WORKFLOW: SSH from Hermes container, run docker benchmarks, return and save results to /opt/data/ai-optimizer/
|
||||
# - NO nix/nixos-rebuild/nh commands
|
||||
# WORKFLOW: SSH from Hermes container, run docker commands via sudo, return and save results
|
||||
services.aiWorkerAccess = true;
|
||||
|
||||
# Restricted sudo for ai-worker - security checks only
|
||||
|
||||
# Restricted sudo for ai-worker
|
||||
# IMPORTANT: ai-worker is NOT in docker group. All docker access goes through sudo.
|
||||
# Only the subcommands listed below are allowed — everything else is denied.
|
||||
# This prevents: docker exec, docker cp, docker commit, and other file-modifying operations.
|
||||
security.sudo.extraRules = [
|
||||
{
|
||||
users = [ "ai-worker" ];
|
||||
commands = [
|
||||
# Firewall checks
|
||||
# === Docker commands: lifecycle management (NO file modification) ===
|
||||
# ps/inspect/logs — read-only status checks
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker ps";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker inspect *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker logs *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker images";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker info";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker version";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker stats *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
|
||||
# start/stop/restart — container lifecycle
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker start *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker stop *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker restart *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker rm *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker rmi *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker wait *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
|
||||
# pull/build/run — image management and container creation
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker pull *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker build *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker run *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
|
||||
# compose — orchestration
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker compose *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
|
||||
# system — disk cleanup
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker system *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
|
||||
# network — list only (create/modify not needed)
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker network ls";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
|
||||
# volume — list only (create/modify not needed)
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker volume ls";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
|
||||
# === EXPLICITLY DENIED docker commands (not in whitelist — sudo rejects them) ===
|
||||
# docker exec — executes arbitrary commands inside running containers (FILE MODIFICATION)
|
||||
# docker cp — copies files between containers and host (FILE ACCESS)
|
||||
# docker commit — creates images from running containers (DATA EXFIL)
|
||||
# docker diff — inspects filesystem changes (INFO LEAK)
|
||||
# docker export — exports container filesystem (DATA EXFIL)
|
||||
# docker import — imports filesystem archives
|
||||
# docker load — loads docker images
|
||||
# docker save — saves docker images to tar (DATA EXFIL)
|
||||
# docker attach — attaches to running containers (INTERACTIVE ACCESS)
|
||||
# docker push — pushes images to registries (DATA EXFIL)
|
||||
# docker tag — renames images
|
||||
# docker create — creates containers (use 'docker run' instead)
|
||||
# docker plugin — manages plugins
|
||||
# docker network create/rm — network management
|
||||
# docker volume create/rm — volume management
|
||||
|
||||
# === Firewall checks ===
|
||||
{
|
||||
command = "/run/wrappers/bin/sudo iptables -L -n -v";
|
||||
options = [ "NOPASSWD" ];
|
||||
@@ -38,7 +155,8 @@
|
||||
command = "/run/wrappers/bin/sudo iptables -S";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
# Fail2ban status
|
||||
|
||||
# === Fail2ban status ===
|
||||
{
|
||||
command = "/run/current-system/sw/bin/fail2ban-client status";
|
||||
options = [ "NOPASSWD" ];
|
||||
@@ -51,7 +169,8 @@
|
||||
command = "/run/current-system/sw/bin/fail2ban-client get * banned";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
# Log inspection
|
||||
|
||||
# === Log inspection ===
|
||||
{
|
||||
command = "/run/current-system/sw/bin/journalctl -t kernel -n 100";
|
||||
options = [ "NOPASSWD" ];
|
||||
@@ -64,21 +183,14 @@
|
||||
command = "/run/current-system/sw/bin/journalctl -u firewall -n 50";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
# SSH config verification
|
||||
|
||||
# === SSH config verification ===
|
||||
{
|
||||
command = "/run/current-system/sw/bin/sshd -T";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
# Docker service checks
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker ps";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
{
|
||||
command = "/run/current-system/sw/bin/docker inspect *";
|
||||
options = [ "NOPASSWD" ];
|
||||
}
|
||||
# Network diagnostics
|
||||
|
||||
# === Network diagnostics ===
|
||||
{
|
||||
command = "/run/current-system/sw/bin/ss -tlnp";
|
||||
options = [ "NOPASSWD" ];
|
||||
|
||||
Reference in New Issue
Block a user