vetto 0.2.21

Daemon-less sandbox + security layer for AI coding agents (Landlock/Seatbelt, TUI statusline, post-session audit reports)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

//! Restricted-token construction without privilege escalation.

use std::ffi::c_void;
use std::os::windows::io::{AsRawHandle, FromRawHandle, OwnedHandle};
use std::ptr::null_mut;

use anyhow::{bail, Result};

pub type RawHandle = *mut c_void;
type Handle = RawHandle;
type Dword = u32;

const DISABLE_MAX_PRIVILEGE: Dword = 0x0000_0001;
const LUA_TOKEN: Dword = 0x0000_0004;
const TOKEN_ASSIGN_PRIMARY: Dword = 0x0001;
const TOKEN_DUPLICATE: Dword = 0x0002;
const TOKEN_QUERY: Dword = 0x0008;
const TOKEN_ADJUST_DEFAULT: Dword = 0x0080;
const SECURITY_IMPERSONATION: Dword = 2;
const TOKEN_PRIMARY: Dword = 1;
const ERROR_INVALID_HANDLE: Dword = 6;

#[link(name = "advapi32")]
extern "system" {
    fn CreateRestrictedToken(
        existing_token: Handle,
        flags: Dword,
        disable_sid_count: Dword,
        sids_to_disable: *const c_void,
        delete_privilege_count: Dword,
        privileges_to_delete: *const c_void,
        restricting_sid_count: Dword,
        restricting_sids: *const c_void,
        new_token: *mut Handle,
    ) -> i32;
    fn DuplicateTokenEx(
        existing_token: Handle,
        desired_access: Dword,
        token_attributes: *mut c_void,
        impersonation_level: Dword,
        token_type: Dword,
        new_token: *mut Handle,
    ) -> i32;
}

#[link(name = "kernel32")]
extern "system" {
    fn CloseHandle(handle: Handle) -> i32;
    fn GetLastError() -> Dword;
}

/// Create a restricted primary token from a token handle owned by the caller.
/// Applies `DISABLE_MAX_PRIVILEGE | LUA_TOKEN` to drop privileges and administrative SIDs.
///
/// # Safety
///
/// `source` must be a live Windows token handle owned by the caller and remain
/// valid for the duration of the call.
pub unsafe fn create_primary(source: RawHandle) -> Result<OwnedHandle> {
    if source.is_null() || source == (-1isize as Handle) {
        bail!(
            "invalid source token handle (error {})",
            ERROR_INVALID_HANDLE
        );
    }
    let mut restricted: Handle = null_mut();
    let ok = CreateRestrictedToken(
        source,
        DISABLE_MAX_PRIVILEGE | LUA_TOKEN,
        0,
        std::ptr::null(),
        0,
        std::ptr::null(),
        0,
        std::ptr::null(),
        &mut restricted,
    );
    if ok == 0 || restricted.is_null() {
        bail!("CreateRestrictedToken failed with {}", GetLastError());
    }

    let mut primary: Handle = null_mut();
    let ok = DuplicateTokenEx(
        restricted,
        TOKEN_ASSIGN_PRIMARY | TOKEN_DUPLICATE | TOKEN_QUERY | TOKEN_ADJUST_DEFAULT,
        null_mut(),
        SECURITY_IMPERSONATION,
        TOKEN_PRIMARY,
        &mut primary,
    );
    let _ = CloseHandle(restricted);
    if ok == 0 || primary.is_null() {
        bail!(
            "DuplicateTokenEx(TokenPrimary) failed with {}",
            GetLastError()
        );
    }
    Ok(OwnedHandle::from_raw_handle(primary.cast()))
}

pub fn raw(handle: &OwnedHandle) -> RawHandle {
    handle.as_raw_handle().cast()
}

pub const fn contract() -> &'static str {
    "restricted primary token with DISABLE_MAX_PRIVILEGE | LUA_TOKEN"
}