@super-protocol/sp-nvtrust-wrapper 1.0.0

package/README.md

@@ -0,0 +1 @@
+# sp-nvtrust-wrapper

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+export { NvtrustWrapper } from './nvtrust_wrapper';
+export * from './proto/nvtrust_wrapper_types';

package/dist/index.js

@@ -0,0 +1,21 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NvtrustWrapper = void 0;
+var nvtrust_wrapper_1 = require("./nvtrust_wrapper");
+Object.defineProperty(exports, "NvtrustWrapper", { enumerable: true, get: function () { return nvtrust_wrapper_1.NvtrustWrapper; } });
+__exportStar(require("./proto/nvtrust_wrapper_types"), exports);
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,qDAAmD;AAA1C,iHAAA,cAAc,OAAA;AACvB,gEAA8C"}

package/dist/nvtrust_wrapper.d.ts

@@ -0,0 +1,20 @@
+import * as proto from './proto/nvtrust_wrapper_types';
+export declare const defaultGPUpolicy = "{\n  \"version\": \"3.0\",\n  \"authorization-rules\": {\n    \"type\": \"JWT\",\n    \"overall-claims\": {\n      \"x-nvidia-overall-att-result\": true,\n      \"x-nvidia-ver\": \"2.0\"\n    },\n    \"detached-claims\": {\n      \"measres\": \"success\",\n      \"x-nvidia-gpu-arch-check\": true,\n      \"x-nvidia-gpu-attestation-report-cert-chain-validated\": true,\n      \"x-nvidia-gpu-attestation-report-parsed\": true,\n      \"x-nvidia-gpu-attestation-report-nonce-match\": true,\n      \"x-nvidia-gpu-attestation-report-signature-verified\": true,\n      \"x-nvidia-gpu-driver-rim-fetched\": true,\n      \"x-nvidia-gpu-driver-rim-schema-validated\": true,\n      \"x-nvidia-gpu-driver-rim-cert-validated\": true,\n      \"x-nvidia-gpu-driver-rim-signature-verified\": true,\n      \"x-nvidia-gpu-driver-rim-measurements-available\": true,\n      \"x-nvidia-gpu-vbios-rim-fetched\": true,\n      \"x-nvidia-gpu-vbios-rim-schema-validated\": true,\n      \"x-nvidia-gpu-vbios-rim-cert-validated\": true,\n      \"x-nvidia-gpu-vbios-rim-signature-verified\": true,\n      \"x-nvidia-gpu-vbios-rim-measurements-available\": true,\n      \"x-nvidia-gpu-vbios-index-no-conflict\": true,\n      \"dbgstat\": \"disabled\"\n    }\n  }\n}";
+export interface GPUInfo {
+    model: string;
+    driverVersion: string;
+    vbios: string;
+    dbgStat: boolean;
+}
+export declare class NvtrustWrapper {
+    private readonly tmpDir;
+    private readonly timeoutMs;
+    constructor(tmpDir?: string, timeoutMs?: number);
+    gramineCompatibleRmDir(dirPath: string): Promise<void>;
+    private pythonCall;
+    getGPUTopology(): Promise<proto.GPUTopologyResult>;
+    generateNonce(): string;
+    getNvidiaToken(options: proto.GetNRASToken): Promise<proto.NvidiaToken>;
+    validateToken(options: proto.ValidateToken): Promise<proto.VerificationResult>;
+    extractGPuInfoFromToken(token: proto.NvidiaToken): GPUInfo[];
+}

package/dist/nvtrust_wrapper.js

@@ -0,0 +1,216 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NvtrustWrapper = exports.defaultGPUpolicy = void 0;
+const child_process_1 = require("child_process");
+const fs = __importStar(require("fs/promises"));
+const path = __importStar(require("path"));
+const fs_1 = require("fs");
+const proto = __importStar(require("./proto/nvtrust_wrapper_types"));
+const crypto_1 = require("crypto");
+const jwt = __importStar(require("jsonwebtoken"));
+exports.defaultGPUpolicy = `{
+  "version": "3.0",
+  "authorization-rules": {
+    "type": "JWT",
+    "overall-claims": {
+      "x-nvidia-overall-att-result": true,
+      "x-nvidia-ver": "2.0"
+    },
+    "detached-claims": {
+      "measres": "success",
+      "x-nvidia-gpu-arch-check": true,
+      "x-nvidia-gpu-attestation-report-cert-chain-validated": true,
+      "x-nvidia-gpu-attestation-report-parsed": true,
+      "x-nvidia-gpu-attestation-report-nonce-match": true,
+      "x-nvidia-gpu-attestation-report-signature-verified": true,
+      "x-nvidia-gpu-driver-rim-fetched": true,
+      "x-nvidia-gpu-driver-rim-schema-validated": true,
+      "x-nvidia-gpu-driver-rim-cert-validated": true,
+      "x-nvidia-gpu-driver-rim-signature-verified": true,
+      "x-nvidia-gpu-driver-rim-measurements-available": true,
+      "x-nvidia-gpu-vbios-rim-fetched": true,
+      "x-nvidia-gpu-vbios-rim-schema-validated": true,
+      "x-nvidia-gpu-vbios-rim-cert-validated": true,
+      "x-nvidia-gpu-vbios-rim-signature-verified": true,
+      "x-nvidia-gpu-vbios-rim-measurements-available": true,
+      "x-nvidia-gpu-vbios-index-no-conflict": true,
+      "dbgstat": "disabled"
+    }
+  }
+}`;
+class NvtrustWrapper {
+    tmpDir;
+    timeoutMs;
+    constructor(tmpDir, timeoutMs) {
+        this.tmpDir = tmpDir ?? '/tmp';
+        this.timeoutMs = timeoutMs ?? 10000;
+    }
+    async gramineCompatibleRmDir(dirPath) {
+        try {
+            await fs.access(dirPath, fs_1.constants.F_OK);
+        }
+        catch {
+            return;
+        }
+        const files = await fs.readdir(dirPath);
+        for (const file of files) {
+            const filePath = path.join(dirPath, file);
+            const stat = await fs.lstat(filePath);
+            if (stat.isDirectory()) {
+                await this.gramineCompatibleRmDir(filePath);
+            }
+            else {
+                await fs.unlink(filePath);
+            }
+        }
+        await fs.rmdir(dirPath);
+    }
+    async pythonCall(request) {
+        const serializedRequest = proto.MethodCallRequest.encode(request).finish();
+        const tempDir = await fs.mkdtemp(path.join(this.tmpDir, 'temp-'));
+        const logDir = path.join(tempDir, 'logs');
+        await fs.mkdir(logDir);
+        try {
+            await fs.writeFile(path.join(tempDir, 'request.bin'), serializedRequest);
+            return new Promise((resolve, reject) => {
+                const pythonProcess = (0, child_process_1.spawn)(`${__dirname}/../py-venv/bin/python3`, ['-B', `${__dirname}/../py/nvtrust_wrapper.py`, tempDir], {
+                    cwd: logDir,
+                });
+                let timeoutId;
+                if (this.timeoutMs && this.timeoutMs > 0) {
+                    timeoutId = setTimeout(() => {
+                        pythonProcess.kill();
+                        reject(new Error(`Process timed out after ${this.timeoutMs}ms`));
+                    }, this.timeoutMs);
+                }
+                pythonProcess.on('close', async (exitCode) => {
+                    if (timeoutId)
+                        clearTimeout(timeoutId);
+                    if (exitCode === 0) {
+                        const result = await fs.readFile(path.join(tempDir, 'response.bin'));
+                        resolve(result);
+                    }
+                    else {
+                        const error = await fs.readFile(path.join(tempDir, 'error.txt'), 'utf-8');
+                        reject(`Python error: ${error.trim()}`);
+                    }
+                });
+                pythonProcess.on('error', (error) => {
+                    if (timeoutId)
+                        clearTimeout(timeoutId);
+                    reject(new Error(`Failed to start process: ${error.message}`));
+                });
+            }).finally(async () => {
+                await this.gramineCompatibleRmDir(tempDir);
+            });
+        }
+        catch (err) {
+            await this.gramineCompatibleRmDir(tempDir);
+            throw err;
+        }
+    }
+    async getGPUTopology() {
+        const request = proto.MethodCallRequest.fromPartial({
+            topology: proto.GetGPUTopology.fromPartial({}),
+        });
+        const result = await this.pythonCall(request);
+        return proto.GPUTopologyResult.decode(result);
+    }
+    generateNonce() {
+        return (0, crypto_1.randomBytes)(32).toString('hex');
+    }
+    async getNvidiaToken(options) {
+        const request = proto.MethodCallRequest.fromPartial({
+            nrasToken: options,
+        });
+        const result = await this.pythonCall(request);
+        return proto.NvidiaToken.decode(result);
+    }
+    async validateToken(options) {
+        const request = proto.MethodCallRequest.fromPartial({
+            validateToken: options,
+        });
+        const result = await this.pythonCall(request);
+        return proto.VerificationResult.decode(result);
+    }
+    extractGPuInfoFromToken(token) {
+        const jsonArray = JSON.parse(token.jwt);
+        const gpuInfo = [];
+        if (Array.isArray(jsonArray) && jsonArray.length >= 2) {
+            for (const [key, value] of Object.entries(jsonArray[1])) {
+                if (key !== 'REMOTE_GPU_CLAIMS') {
+                    throw new Error(`The expected value does not match REMOTE_GPU_CLAIMS. ${key} found`);
+                }
+                if (Array.isArray(value)) {
+                    for (const item of value) {
+                        if (typeof item === 'object' && item !== null && !Array.isArray(item)) {
+                            for (const [, v] of Object.entries(item)) {
+                                try {
+                                    const payload = jwt.decode(v, {
+                                        complete: false,
+                                        json: false,
+                                    });
+                                    if (payload) {
+                                        gpuInfo.push({
+                                            model: payload['hwmodel'],
+                                            driverVersion: payload['x-nvidia-gpu-driver-version'],
+                                            vbios: payload['x-nvidia-gpu-vbios-version'],
+                                            dbgStat: payload['dbgstat'] === 'disabled' ? false : true,
+                                        });
+                                    }
+                                }
+                                catch (err) {
+                                    continue;
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        return gpuInfo;
+    }
+}
+exports.NvtrustWrapper = NvtrustWrapper;
+(async () => {
+    const wrapper = new NvtrustWrapper();
+    //const topology = await wrapper.getGPUTopology();
+    //console.log(topology);
+    const token = await wrapper.getNvidiaToken({
+        device: proto.NvidiaDeviceType.GPU,
+        nonce: '3bb86d2df2736dac5576772e4bef6c25a38a2c1fd3d1bda37cd3e7980a34475d',
+        ppcieMode: false,
+    });
+    const res = await wrapper.validateToken({
+        device: proto.NvidiaDeviceType.GPU,
+        token,
+        policy: exports.defaultGPUpolicy,
+    });
+    console.log(res);
+    const gpu = wrapper.extractGPuInfoFromToken(token);
+    console.log(gpu);
+})();
+//# sourceMappingURL=nvtrust_wrapper.js.map

package/dist/nvtrust_wrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"nvtrust_wrapper.js","sourceRoot":"","sources":["../src/nvtrust_wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,iDAAsC;AACtC,gDAAkC;AAClC,2CAA6B;AAC7B,2BAA+B;AAC/B,qEAAuD;AACvD,mCAAqC;AACrC,kDAAoC;AAEvB,QAAA,gBAAgB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA6B9B,CAAC;AAQH,MAAa,cAAc;IACR,MAAM,CAAS;IACf,SAAS,CAAS;IACnC,YAAY,MAAe,EAAE,SAAkB;QAC7C,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,MAAM,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,KAAK,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,OAAe;QAC1C,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,cAAS,CAAC,IAAI,CAAC,CAAC;QAC3C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAExC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC1C,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAEtC,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBACvB,MAAM,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACN,MAAM,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QACD,MAAM,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,OAAgC;QACvD,MAAM,iBAAiB,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QAC3E,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;QAClE,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC1C,MAAM,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAEvB,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,EAAE,iBAAiB,CAAC,CAAC;YAEzE,OAAO,IAAI,OAAO,CAAa,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACjD,MAAM,aAAa,GAAG,IAAA,qBAAK,EACzB,GAAG,SAAS,yBAAyB,EACrC,CAAC,IAAI,EAAE,GAAG,SAAS,2BAA2B,EAAE,OAAO,CAAC,EACxD;oBACE,GAAG,EAAE,MAAM;iBACZ,CACF,CAAC;gBAEF,IAAI,SAAqC,CAAC;gBAC1C,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC;oBACzC,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;wBAC1B,aAAa,CAAC,IAAI,EAAE,CAAC;wBACrB,MAAM,CAAC,IAAI,KAAK,CAAC,2BAA2B,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC;oBACnE,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBACrB,CAAC;gBAED,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;oBAC3C,IAAI,SAAS;wBAAE,YAAY,CAAC,SAAS,CAAC,CAAC;oBACvC,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;wBACnB,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC,CAAC;wBACrE,OAAO,CAAC,MAAM,CAAC,CAAC;oBAClB,CAAC;yBAAM,CAAC;wBACN,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,OAAO,CAAC,CAAC;wBAC1E,MAAM,CAAC,iBAAiB,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAC1C,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBAClC,IAAI,SAAS;wBAAE,YAAY,CAAC,SAAS,CAAC,CAAC;oBACvC,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBACjE,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE;gBACpB,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;YAC3C,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,MAAM,OAAO,GAAG,KAAK,CAAC,iBAAiB,CAAC,WAAW,CAAC;YAClD,QAAQ,EAAE,KAAK,CAAC,cAAc,CAAC,WAAW,CAAC,EAAE,CAAC;SAC/C,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC9C,OAAO,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAChD,CAAC;IAED,aAAa;QACX,OAAO,IAAA,oBAAW,EAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAA2B;QAC9C,MAAM,OAAO,GAAG,KAAK,CAAC,iBAAiB,CAAC,WAAW,CAAC;YAClD,SAAS,EAAE,OAAO;SACnB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC9C,OAAO,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAA4B;QAC9C,MAAM,OAAO,GAAG,KAAK,CAAC,iBAAiB,CAAC,WAAW,CAAC;YAClD,aAAa,EAAE,OAAO;SACvB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC9C,OAAO,KAAK,CAAC,kBAAkB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACjD,CAAC;IAED,uBAAuB,CAAC,KAAwB;QAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,OAAO,GAAc,EAAE,CAAC;QAC9B,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACtD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxD,IAAI,GAAG,KAAK,mBAAmB,EAAE,CAAC;oBAChC,MAAM,IAAI,KAAK,CAAC,wDAAwD,GAAG,QAAQ,CAAC,CAAC;gBACvF,CAAC;gBACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;oBACzB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACzB,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;4BACtE,KAAK,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gCACzC,IAAI,CAAC;oCACH,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,CAAW,EAAE;wCACtC,QAAQ,EAAE,KAAK;wCACf,IAAI,EAAE,KAAK;qCACZ,CAAmB,CAAC;oCACrB,IAAI,OAAO,EAAE,CAAC;wCACZ,OAAO,CAAC,IAAI,CAAC;4CACX,KAAK,EAAE,OAAO,CAAC,SAAS,CAAW;4CACnC,aAAa,EAAE,OAAO,CAAC,6BAA6B,CAAW;4CAC/D,KAAK,EAAE,OAAO,CAAC,4BAA4B,CAAW;4CACtD,OAAO,EAAG,OAAO,CAAC,SAAS,CAAY,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI;yCACtE,CAAC,CAAC;oCACL,CAAC;gCACH,CAAC;gCAAC,OAAO,GAAG,EAAE,CAAC;oCACb,SAAS;gCACX,CAAC;4BACH,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAlJD,wCAkJC;AAED,CAAC,KAAK,IAAI,EAAE;IACV,MAAM,OAAO,GAAG,IAAI,cAAc,EAAE,CAAC;IACrC,kDAAkD;IAClD,wBAAwB;IACxB,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC;QACzC,MAAM,EAAE,KAAK,CAAC,gBAAgB,CAAC,GAAG;QAClC,KAAK,EAAE,kEAAkE;QACzE,SAAS,EAAE,KAAK;KACjB,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC;QACtC,MAAM,EAAE,KAAK,CAAC,gBAAgB,CAAC,GAAG;QAClC,KAAK;QACL,MAAM,EAAE,wBAAgB;KACzB,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAEjB,MAAM,GAAG,GAAG,OAAO,CAAC,uBAAuB,CAAC,KAAK,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;AACnB,CAAC,CAAC,EAAE,CAAC"}

package/dist/proto/nvtrust_wrapper_types.d.ts

@@ -0,0 +1,254 @@
+import _m0 from "protobufjs/minimal.js";
+export declare const protobufPackage = "nvtrust_wrapper_types";
+export declare enum NvidiaDeviceType {
+    GPU = 0,
+    SWITCH = 1
+}
+export declare function nvidiaDeviceTypeFromJSON(object: any): NvidiaDeviceType;
+export declare function nvidiaDeviceTypeToJSON(object: NvidiaDeviceType): string;
+export interface GetGPUTopology {
+}
+export interface GPUTopologyResult {
+    gpuCount: number;
+    nvswitchCount: number;
+}
+export interface GetNRASToken {
+    device: NvidiaDeviceType;
+    nonce: string;
+    ppcieMode: boolean;
+}
+export interface NvidiaToken {
+    jwt: string;
+}
+export interface ValidateToken {
+    token: NvidiaToken | undefined;
+    policy: string;
+    device: NvidiaDeviceType;
+}
+export interface VerificationResult {
+    success: boolean;
+}
+export interface MethodCallRequest {
+    topology?: GetGPUTopology | undefined;
+    nrasToken?: GetNRASToken | undefined;
+    validateToken?: ValidateToken | undefined;
+}
+export declare const GetGPUTopology: {
+    encode(_: GetGPUTopology, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): GetGPUTopology;
+    fromJSON(_: any): GetGPUTopology;
+    toJSON(_: GetGPUTopology): unknown;
+    create<I extends {} & {} & { [K in Exclude<keyof I, never>]: never; }>(base?: I | undefined): GetGPUTopology;
+    fromPartial<I_1 extends {} & {} & { [K_1 in Exclude<keyof I_1, never>]: never; }>(_: I_1): GetGPUTopology;
+};
+export declare const GPUTopologyResult: {
+    encode(message: GPUTopologyResult, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): GPUTopologyResult;
+    fromJSON(object: any): GPUTopologyResult;
+    toJSON(message: GPUTopologyResult): unknown;
+    create<I extends {
+        gpuCount?: number | undefined;
+        nvswitchCount?: number | undefined;
+    } & {
+        gpuCount?: number | undefined;
+        nvswitchCount?: number | undefined;
+    } & { [K in Exclude<keyof I, keyof GPUTopologyResult>]: never; }>(base?: I | undefined): GPUTopologyResult;
+    fromPartial<I_1 extends {
+        gpuCount?: number | undefined;
+        nvswitchCount?: number | undefined;
+    } & {
+        gpuCount?: number | undefined;
+        nvswitchCount?: number | undefined;
+    } & { [K_1 in Exclude<keyof I_1, keyof GPUTopologyResult>]: never; }>(object: I_1): GPUTopologyResult;
+};
+export declare const GetNRASToken: {
+    encode(message: GetNRASToken, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): GetNRASToken;
+    fromJSON(object: any): GetNRASToken;
+    toJSON(message: GetNRASToken): unknown;
+    create<I extends {
+        device?: NvidiaDeviceType | undefined;
+        nonce?: string | undefined;
+        ppcieMode?: boolean | undefined;
+    } & {
+        device?: NvidiaDeviceType | undefined;
+        nonce?: string | undefined;
+        ppcieMode?: boolean | undefined;
+    } & { [K in Exclude<keyof I, keyof GetNRASToken>]: never; }>(base?: I | undefined): GetNRASToken;
+    fromPartial<I_1 extends {
+        device?: NvidiaDeviceType | undefined;
+        nonce?: string | undefined;
+        ppcieMode?: boolean | undefined;
+    } & {
+        device?: NvidiaDeviceType | undefined;
+        nonce?: string | undefined;
+        ppcieMode?: boolean | undefined;
+    } & { [K_1 in Exclude<keyof I_1, keyof GetNRASToken>]: never; }>(object: I_1): GetNRASToken;
+};
+export declare const NvidiaToken: {
+    encode(message: NvidiaToken, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): NvidiaToken;
+    fromJSON(object: any): NvidiaToken;
+    toJSON(message: NvidiaToken): unknown;
+    create<I extends {
+        jwt?: string | undefined;
+    } & {
+        jwt?: string | undefined;
+    } & { [K in Exclude<keyof I, "jwt">]: never; }>(base?: I | undefined): NvidiaToken;
+    fromPartial<I_1 extends {
+        jwt?: string | undefined;
+    } & {
+        jwt?: string | undefined;
+    } & { [K_1 in Exclude<keyof I_1, "jwt">]: never; }>(object: I_1): NvidiaToken;
+};
+export declare const ValidateToken: {
+    encode(message: ValidateToken, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): ValidateToken;
+    fromJSON(object: any): ValidateToken;
+    toJSON(message: ValidateToken): unknown;
+    create<I extends {
+        token?: {
+            jwt?: string | undefined;
+        } | undefined;
+        policy?: string | undefined;
+        device?: NvidiaDeviceType | undefined;
+    } & {
+        token?: ({
+            jwt?: string | undefined;
+        } & {
+            jwt?: string | undefined;
+        } & { [K in Exclude<keyof I["token"], "jwt">]: never; }) | undefined;
+        policy?: string | undefined;
+        device?: NvidiaDeviceType | undefined;
+    } & { [K_1 in Exclude<keyof I, keyof ValidateToken>]: never; }>(base?: I | undefined): ValidateToken;
+    fromPartial<I_1 extends {
+        token?: {
+            jwt?: string | undefined;
+        } | undefined;
+        policy?: string | undefined;
+        device?: NvidiaDeviceType | undefined;
+    } & {
+        token?: ({
+            jwt?: string | undefined;
+        } & {
+            jwt?: string | undefined;
+        } & { [K_2 in Exclude<keyof I_1["token"], "jwt">]: never; }) | undefined;
+        policy?: string | undefined;
+        device?: NvidiaDeviceType | undefined;
+    } & { [K_3 in Exclude<keyof I_1, keyof ValidateToken>]: never; }>(object: I_1): ValidateToken;
+};
+export declare const VerificationResult: {
+    encode(message: VerificationResult, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): VerificationResult;
+    fromJSON(object: any): VerificationResult;
+    toJSON(message: VerificationResult): unknown;
+    create<I extends {
+        success?: boolean | undefined;
+    } & {
+        success?: boolean | undefined;
+    } & { [K in Exclude<keyof I, "success">]: never; }>(base?: I | undefined): VerificationResult;
+    fromPartial<I_1 extends {
+        success?: boolean | undefined;
+    } & {
+        success?: boolean | undefined;
+    } & { [K_1 in Exclude<keyof I_1, "success">]: never; }>(object: I_1): VerificationResult;
+};
+export declare const MethodCallRequest: {
+    encode(message: MethodCallRequest, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): MethodCallRequest;
+    fromJSON(object: any): MethodCallRequest;
+    toJSON(message: MethodCallRequest): unknown;
+    create<I extends {
+        topology?: {} | undefined;
+        nrasToken?: {
+            device?: NvidiaDeviceType | undefined;
+            nonce?: string | undefined;
+            ppcieMode?: boolean | undefined;
+        } | undefined;
+        validateToken?: {
+            token?: {
+                jwt?: string | undefined;
+            } | undefined;
+            policy?: string | undefined;
+            device?: NvidiaDeviceType | undefined;
+        } | undefined;
+    } & {
+        topology?: ({} & {} & { [K in Exclude<keyof I["topology"], never>]: never; }) | undefined;
+        nrasToken?: ({
+            device?: NvidiaDeviceType | undefined;
+            nonce?: string | undefined;
+            ppcieMode?: boolean | undefined;
+        } & {
+            device?: NvidiaDeviceType | undefined;
+            nonce?: string | undefined;
+            ppcieMode?: boolean | undefined;
+        } & { [K_1 in Exclude<keyof I["nrasToken"], keyof GetNRASToken>]: never; }) | undefined;
+        validateToken?: ({
+            token?: {
+                jwt?: string | undefined;
+            } | undefined;
+            policy?: string | undefined;
+            device?: NvidiaDeviceType | undefined;
+        } & {
+            token?: ({
+                jwt?: string | undefined;
+            } & {
+                jwt?: string | undefined;
+            } & { [K_2 in Exclude<keyof I["validateToken"]["token"], "jwt">]: never; }) | undefined;
+            policy?: string | undefined;
+            device?: NvidiaDeviceType | undefined;
+        } & { [K_3 in Exclude<keyof I["validateToken"], keyof ValidateToken>]: never; }) | undefined;
+    } & { [K_4 in Exclude<keyof I, keyof MethodCallRequest>]: never; }>(base?: I | undefined): MethodCallRequest;
+    fromPartial<I_1 extends {
+        topology?: {} | undefined;
+        nrasToken?: {
+            device?: NvidiaDeviceType | undefined;
+            nonce?: string | undefined;
+            ppcieMode?: boolean | undefined;
+        } | undefined;
+        validateToken?: {
+            token?: {
+                jwt?: string | undefined;
+            } | undefined;
+            policy?: string | undefined;
+            device?: NvidiaDeviceType | undefined;
+        } | undefined;
+    } & {
+        topology?: ({} & {} & { [K_5 in Exclude<keyof I_1["topology"], never>]: never; }) | undefined;
+        nrasToken?: ({
+            device?: NvidiaDeviceType | undefined;
+            nonce?: string | undefined;
+            ppcieMode?: boolean | undefined;
+        } & {
+            device?: NvidiaDeviceType | undefined;
+            nonce?: string | undefined;
+            ppcieMode?: boolean | undefined;
+        } & { [K_6 in Exclude<keyof I_1["nrasToken"], keyof GetNRASToken>]: never; }) | undefined;
+        validateToken?: ({
+            token?: {
+                jwt?: string | undefined;
+            } | undefined;
+            policy?: string | undefined;
+            device?: NvidiaDeviceType | undefined;
+        } & {
+            token?: ({
+                jwt?: string | undefined;
+            } & {
+                jwt?: string | undefined;
+            } & { [K_7 in Exclude<keyof I_1["validateToken"]["token"], "jwt">]: never; }) | undefined;
+            policy?: string | undefined;
+            device?: NvidiaDeviceType | undefined;
+        } & { [K_8 in Exclude<keyof I_1["validateToken"], keyof ValidateToken>]: never; }) | undefined;
+    } & { [K_9 in Exclude<keyof I_1, keyof MethodCallRequest>]: never; }>(object: I_1): MethodCallRequest;
+};
+type Builtin = Date | Function | Uint8Array | string | number | boolean | undefined;
+export type DeepPartial<T> = T extends Builtin ? T : T extends globalThis.Array<infer U> ? globalThis.Array<DeepPartial<U>> : T extends ReadonlyArray<infer U> ? ReadonlyArray<DeepPartial<U>> : T extends {} ? {
+    [K in keyof T]?: DeepPartial<T[K]>;
+} : Partial<T>;
+type KeysOfUnion<T> = T extends T ? keyof T : never;
+export type Exact<P, I extends P> = P extends Builtin ? P : P & {
+    [K in keyof P]: Exact<P[K], I[K]>;
+} & {
+    [K in Exclude<keyof I, KeysOfUnion<P>>]: never;
+};
+export {};