@super-protocol/sp-nvtrust-wrapper 1.0.0 → 1.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/nvtrust_wrapper.js
CHANGED
|
@@ -66,7 +66,7 @@ class NvtrustWrapper {
|
|
|
66
66
|
timeoutMs;
|
|
67
67
|
constructor(tmpDir, timeoutMs) {
|
|
68
68
|
this.tmpDir = tmpDir ?? '/tmp';
|
|
69
|
-
this.timeoutMs = timeoutMs ??
|
|
69
|
+
this.timeoutMs = timeoutMs ?? 120000;
|
|
70
70
|
}
|
|
71
71
|
async gramineCompatibleRmDir(dirPath) {
|
|
72
72
|
try {
|
|
@@ -195,22 +195,4 @@ class NvtrustWrapper {
|
|
|
195
195
|
}
|
|
196
196
|
}
|
|
197
197
|
exports.NvtrustWrapper = NvtrustWrapper;
|
|
198
|
-
(async () => {
|
|
199
|
-
const wrapper = new NvtrustWrapper();
|
|
200
|
-
//const topology = await wrapper.getGPUTopology();
|
|
201
|
-
//console.log(topology);
|
|
202
|
-
const token = await wrapper.getNvidiaToken({
|
|
203
|
-
device: proto.NvidiaDeviceType.GPU,
|
|
204
|
-
nonce: '3bb86d2df2736dac5576772e4bef6c25a38a2c1fd3d1bda37cd3e7980a34475d',
|
|
205
|
-
ppcieMode: false,
|
|
206
|
-
});
|
|
207
|
-
const res = await wrapper.validateToken({
|
|
208
|
-
device: proto.NvidiaDeviceType.GPU,
|
|
209
|
-
token,
|
|
210
|
-
policy: exports.defaultGPUpolicy,
|
|
211
|
-
});
|
|
212
|
-
console.log(res);
|
|
213
|
-
const gpu = wrapper.extractGPuInfoFromToken(token);
|
|
214
|
-
console.log(gpu);
|
|
215
|
-
})();
|
|
216
198
|
//# sourceMappingURL=nvtrust_wrapper.js.map
|
|
@@ -1 +1 @@
|
|
|
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,
|
|
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,MAAM,CAAC;IACvC,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"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@super-protocol/sp-nvtrust-wrapper",
|
|
3
|
-
"version": "1.0.
|
|
3
|
+
"version": "1.0.1",
|
|
4
4
|
"description": "NodeJS wrapper for nvtrust",
|
|
5
5
|
"main": "./dist/index.js",
|
|
6
6
|
"scripts": {
|
|
@@ -38,7 +38,8 @@
|
|
|
38
38
|
},
|
|
39
39
|
"files": [
|
|
40
40
|
"dist/**/*",
|
|
41
|
-
"py
|
|
41
|
+
"py/**/*.py",
|
|
42
|
+
"py/**/*.txt"
|
|
42
43
|
],
|
|
43
44
|
"homepage": "https://github.com/Super-Protocol/sp-nvtrust-wrapper#",
|
|
44
45
|
"bugs": {
|
|
Binary file
|
|
Binary file
|
|
@@ -1,65 +0,0 @@
|
|
|
1
|
-
from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
|
|
2
|
-
from google.protobuf import descriptor as _descriptor
|
|
3
|
-
from google.protobuf import message as _message
|
|
4
|
-
from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
|
|
5
|
-
|
|
6
|
-
DESCRIPTOR: _descriptor.FileDescriptor
|
|
7
|
-
GPU: NvidiaDeviceType
|
|
8
|
-
SWITCH: NvidiaDeviceType
|
|
9
|
-
|
|
10
|
-
class GPUTopologyResult(_message.Message):
|
|
11
|
-
__slots__ = ["gpuCount", "nvswitchCount"]
|
|
12
|
-
GPUCOUNT_FIELD_NUMBER: _ClassVar[int]
|
|
13
|
-
NVSWITCHCOUNT_FIELD_NUMBER: _ClassVar[int]
|
|
14
|
-
gpuCount: int
|
|
15
|
-
nvswitchCount: int
|
|
16
|
-
def __init__(self, gpuCount: _Optional[int] = ..., nvswitchCount: _Optional[int] = ...) -> None: ...
|
|
17
|
-
|
|
18
|
-
class GetGPUTopology(_message.Message):
|
|
19
|
-
__slots__ = []
|
|
20
|
-
def __init__(self) -> None: ...
|
|
21
|
-
|
|
22
|
-
class GetNRASToken(_message.Message):
|
|
23
|
-
__slots__ = ["device", "nonce", "ppcieMode"]
|
|
24
|
-
DEVICE_FIELD_NUMBER: _ClassVar[int]
|
|
25
|
-
NONCE_FIELD_NUMBER: _ClassVar[int]
|
|
26
|
-
PPCIEMODE_FIELD_NUMBER: _ClassVar[int]
|
|
27
|
-
device: NvidiaDeviceType
|
|
28
|
-
nonce: str
|
|
29
|
-
ppcieMode: bool
|
|
30
|
-
def __init__(self, device: _Optional[_Union[NvidiaDeviceType, str]] = ..., nonce: _Optional[str] = ..., ppcieMode: bool = ...) -> None: ...
|
|
31
|
-
|
|
32
|
-
class MethodCallRequest(_message.Message):
|
|
33
|
-
__slots__ = ["nrasToken", "topology", "validateToken"]
|
|
34
|
-
NRASTOKEN_FIELD_NUMBER: _ClassVar[int]
|
|
35
|
-
TOPOLOGY_FIELD_NUMBER: _ClassVar[int]
|
|
36
|
-
VALIDATETOKEN_FIELD_NUMBER: _ClassVar[int]
|
|
37
|
-
nrasToken: GetNRASToken
|
|
38
|
-
topology: GetGPUTopology
|
|
39
|
-
validateToken: ValidateToken
|
|
40
|
-
def __init__(self, topology: _Optional[_Union[GetGPUTopology, _Mapping]] = ..., nrasToken: _Optional[_Union[GetNRASToken, _Mapping]] = ..., validateToken: _Optional[_Union[ValidateToken, _Mapping]] = ...) -> None: ...
|
|
41
|
-
|
|
42
|
-
class NvidiaToken(_message.Message):
|
|
43
|
-
__slots__ = ["jwt"]
|
|
44
|
-
JWT_FIELD_NUMBER: _ClassVar[int]
|
|
45
|
-
jwt: str
|
|
46
|
-
def __init__(self, jwt: _Optional[str] = ...) -> None: ...
|
|
47
|
-
|
|
48
|
-
class ValidateToken(_message.Message):
|
|
49
|
-
__slots__ = ["device", "policy", "token"]
|
|
50
|
-
DEVICE_FIELD_NUMBER: _ClassVar[int]
|
|
51
|
-
POLICY_FIELD_NUMBER: _ClassVar[int]
|
|
52
|
-
TOKEN_FIELD_NUMBER: _ClassVar[int]
|
|
53
|
-
device: NvidiaDeviceType
|
|
54
|
-
policy: str
|
|
55
|
-
token: NvidiaToken
|
|
56
|
-
def __init__(self, token: _Optional[_Union[NvidiaToken, _Mapping]] = ..., policy: _Optional[str] = ..., device: _Optional[_Union[NvidiaDeviceType, str]] = ...) -> None: ...
|
|
57
|
-
|
|
58
|
-
class VerificationResult(_message.Message):
|
|
59
|
-
__slots__ = ["success"]
|
|
60
|
-
SUCCESS_FIELD_NUMBER: _ClassVar[int]
|
|
61
|
-
success: bool
|
|
62
|
-
def __init__(self, success: bool = ...) -> None: ...
|
|
63
|
-
|
|
64
|
-
class NvidiaDeviceType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
|
|
65
|
-
__slots__ = []
|