diff --git a/vmsdk/python/cc_event_log_cli.py b/vmsdk/python/cc_event_log_cli.py index e0a40f74..08f5cf82 100644 --- a/vmsdk/python/cc_event_log_cli.py +++ b/vmsdk/python/cc_event_log_cli.py @@ -3,6 +3,7 @@ """ import logging import argparse +import os from cctrusted_vm import CCTrustedVmSdk @@ -11,7 +12,11 @@ logging.basicConfig(level=logging.NOTSET, format='%(name)s %(levelname)-8s %(message)s') def main(): - """example cc event log fetching utility""" + """Example cc event log fetching utility.""" + if os.geteuid() != 0: + LOG.error("Please run as root which is required for this example!") + return + parser = argparse.ArgumentParser( description="The example utility to fetch CC event logs") parser.add_argument('-s', type=int, diff --git a/vmsdk/python/cc_imr_cli.py b/vmsdk/python/cc_imr_cli.py index a7d7fbde..da49a33d 100644 --- a/vmsdk/python/cc_imr_cli.py +++ b/vmsdk/python/cc_imr_cli.py @@ -3,21 +3,32 @@ Command line to dump the integrated measurement register """ import logging +import os from cctrusted_vm import CCTrustedVmSdk LOG = logging.getLogger(__name__) logging.basicConfig(level=logging.NOTSET, format='%(name)s %(levelname)-8s %(message)s') -count = CCTrustedVmSdk.inst().get_measurement_count() -for index in range(CCTrustedVmSdk.inst().get_measurement_count()): - alg = CCTrustedVmSdk.inst().get_default_algorithms() - imr = CCTrustedVmSdk.inst().get_cc_measurement([index, alg.alg_id]) - digest_obj = imr.digest(alg.alg_id) +def main(): + """Example to call get_cc_measurement and dump the result to stdout.""" + if os.geteuid() != 0: + LOG.error("Please run as root which is required for this example!") + return - hash_str = "" - for hash_item in digest_obj.hash: - hash_str += "".join([f"{hash_item:02x}", " "]) + count = CCTrustedVmSdk.inst().get_measurement_count() + LOG.info("Measurement Count: %d", count) + for index in range(CCTrustedVmSdk.inst().get_measurement_count()): + alg = CCTrustedVmSdk.inst().get_default_algorithms() + imr = CCTrustedVmSdk.inst().get_cc_measurement([index, alg.alg_id]) + digest_obj = imr.digest(alg.alg_id) - LOG.info("Algorithms: %s", str(alg)) - LOG.info("HASH: %s", hash_str) + hash_str = "" + for hash_item in digest_obj.hash: + hash_str += "".join([f"{hash_item:02x}", " "]) + + LOG.info("Algorithms: %s", str(alg)) + LOG.info("HASH: %s", hash_str) + +if __name__ == "__main__": + main() diff --git a/vmsdk/python/cc_quote_cli.py b/vmsdk/python/cc_quote_cli.py index 5153ef80..4225ab42 100644 --- a/vmsdk/python/cc_quote_cli.py +++ b/vmsdk/python/cc_quote_cli.py @@ -4,6 +4,7 @@ import argparse import base64 import logging +import os import random from cctrusted_vm import CCTrustedVmSdk @@ -49,6 +50,10 @@ def make_userdata(): def main(): """Example to call get_cc_report and dump the result to stdout.""" + if os.geteuid() != 0: + LOG.error("Please run as root which is required for this example!") + return + parser = argparse.ArgumentParser() parser.add_argument( "--out-format", diff --git a/vmsdk/python/td_report_cli.py b/vmsdk/python/td_report_cli.py index d36c1908..59b293d9 100644 --- a/vmsdk/python/td_report_cli.py +++ b/vmsdk/python/td_report_cli.py @@ -3,11 +3,21 @@ Command line to dump the integrated measurement register """ import logging +import os from cctrusted_vm import CCTrustedTdvmSdk LOG = logging.getLogger(__name__) logging.basicConfig(level=logging.NOTSET, format='%(message)s') -tdreport = CCTrustedTdvmSdk.inst().get_tdreport() -tdreport.dump() +def main(): + """Example to call get_tdreport and dump the result to stdout.""" + if os.geteuid() != 0: + LOG.error("Please run as root which is required for this example!") + return + + tdreport = CCTrustedTdvmSdk.inst().get_tdreport() + tdreport.dump() + +if __name__ == "__main__": + main()