@super-protocol/sp-nvtrust-wrapper 2.0.2 → 2.0.3
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/package.json +1 -1
- package/py/nvtrust_wrapper.py +50 -0
package/package.json
CHANGED
package/py/nvtrust_wrapper.py
CHANGED
|
@@ -2,10 +2,59 @@ import sys
|
|
|
2
2
|
import os
|
|
3
3
|
import glob
|
|
4
4
|
import traceback
|
|
5
|
+
import logging
|
|
5
6
|
import proto.NvtrustWrapperTypes_pb2 as proto
|
|
6
7
|
from nv_attestation_sdk import attestation
|
|
7
8
|
from nvml_helper import NvmlHelper
|
|
8
9
|
|
|
10
|
+
|
|
11
|
+
def configure_sdk_loggers():
|
|
12
|
+
log_file_path = os.path.join(os.getcwd(), "attestation_sdk.log")
|
|
13
|
+
verifier_log_file_path = os.path.join(os.getcwd(), "verifier.log")
|
|
14
|
+
formatter = logging.Formatter("%(asctime)s:%(levelname)s: %(message)s", "%m-%d-%Y %H:%M:%S")
|
|
15
|
+
|
|
16
|
+
sdk_file_logger = logging.getLogger("sdk-file")
|
|
17
|
+
sdk_file_logger.setLevel(logging.DEBUG)
|
|
18
|
+
sdk_file_logger.propagate = False
|
|
19
|
+
has_sdk_file_handler = any(
|
|
20
|
+
isinstance(handler, logging.FileHandler)
|
|
21
|
+
and getattr(handler, "baseFilename", None) == log_file_path
|
|
22
|
+
for handler in sdk_file_logger.handlers
|
|
23
|
+
)
|
|
24
|
+
if not has_sdk_file_handler:
|
|
25
|
+
file_handler = logging.FileHandler(log_file_path)
|
|
26
|
+
file_handler.setLevel(logging.DEBUG)
|
|
27
|
+
file_handler.setFormatter(formatter)
|
|
28
|
+
sdk_file_logger.addHandler(file_handler)
|
|
29
|
+
|
|
30
|
+
sdk_console_logger = logging.getLogger("sdk-console")
|
|
31
|
+
sdk_console_logger.setLevel(logging.INFO)
|
|
32
|
+
sdk_console_logger.propagate = False
|
|
33
|
+
has_sdk_console_handler = any(
|
|
34
|
+
isinstance(handler, logging.FileHandler)
|
|
35
|
+
and getattr(handler, "baseFilename", None) == log_file_path
|
|
36
|
+
for handler in sdk_console_logger.handlers
|
|
37
|
+
)
|
|
38
|
+
if not has_sdk_console_handler:
|
|
39
|
+
console_file_handler = logging.FileHandler(log_file_path)
|
|
40
|
+
console_file_handler.setLevel(logging.INFO)
|
|
41
|
+
console_file_handler.setFormatter(formatter)
|
|
42
|
+
sdk_console_logger.addHandler(console_file_handler)
|
|
43
|
+
|
|
44
|
+
verifier_event_logger = logging.getLogger("gpu-verifier-event")
|
|
45
|
+
verifier_event_logger.setLevel(logging.DEBUG)
|
|
46
|
+
verifier_event_logger.propagate = False
|
|
47
|
+
has_verifier_file_handler = any(
|
|
48
|
+
isinstance(handler, logging.FileHandler)
|
|
49
|
+
and getattr(handler, "baseFilename", None) == verifier_log_file_path
|
|
50
|
+
for handler in verifier_event_logger.handlers
|
|
51
|
+
)
|
|
52
|
+
if not has_verifier_file_handler:
|
|
53
|
+
verifier_file_handler = logging.FileHandler(verifier_log_file_path)
|
|
54
|
+
verifier_file_handler.setLevel(logging.DEBUG)
|
|
55
|
+
verifier_file_handler.setFormatter(formatter)
|
|
56
|
+
verifier_event_logger.addHandler(verifier_file_handler)
|
|
57
|
+
|
|
9
58
|
def get_nvidia_topology():
|
|
10
59
|
result = proto.GPUTopologyResult()
|
|
11
60
|
result.gpuCount = 0
|
|
@@ -84,6 +133,7 @@ def handle_request(serialized_data):
|
|
|
84
133
|
|
|
85
134
|
if __name__ == "__main__":
|
|
86
135
|
_, tmpDir = sys.argv
|
|
136
|
+
configure_sdk_loggers()
|
|
87
137
|
|
|
88
138
|
exitcode = 0
|
|
89
139
|
serialized_response = b""
|