sq

Signed-off-by: Egor Lazarchuk <[email protected]>

Author: ShadowCurse

tests/data/msr/msr_reader.py

@@ -1,16 +1,16 @@
-// Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved.
-// SPDX-License-Identifier: Apache-2.0
+# Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+# SPDX-License-Identifier: Apache-2.0
 
-// Helper script used to read MSR values from ranges known to contain MSRs.
+# Helper script used to read MSR values from ranges known to contain MSRs.
 
+import os
 import mmap
 
 msr_fd = open("/dev/cpu/0/msr", "r+b")
-msrs = mmap.mmap(msr_fd.fileno(), 0)
 
-def print_msr(msr):
-    msr_bytes = msrs[msr:msr+8]
-    if len(msr_bytes) == 8:
+def print_msr(msr: int):
+    msr_bytes = os.pread(msr_fd, 8, msr)
+    if msr_bytes:
         msr_value = int.from_bytes(msr_bytes)
         print("{msr:#x},{msr_value:#x}")