[pentest] Reformat cryptolib pentest functions

Source format the python files in host penetrationtests.

Signed-off-by: Siemen Dhooghe <[email protected]>

Author: siemen11

sw/host/penetrationtests/python/fi/communication/fi_asym_cryptolib_commands.py

@@ -5,6 +5,7 @@
 
 Communication with OpenTitan happens over the uJSON command interface.
 """
+
 import json
 import time
 from sw.host.penetrationtests.python.util import common_library

sw/host/penetrationtests/python/fi/communication/fi_sym_cryptolib_commands.py

@@ -5,6 +5,7 @@
 
 Communication with OpenTitan happens over the uJSON command interface.
 """
+
 import json
 import time
 from sw.host.penetrationtests.python.util import common_library
@@ -94,9 +95,7 @@ def handle_aes(
         }
         self.target.write(json.dumps(input_data).encode("ascii"))
 
-    def handle_cmac(
-        self, data, data_len, key, key_len, iv, cfg, trigger
-    ) -> None:
+    def handle_cmac(self, data, data_len, key, key_len, iv, cfg, trigger) -> None:
         """Call the cryptolib CMAC.
 
         Args:
@@ -186,7 +185,15 @@ def handle_hmac(
         self.target.write(json.dumps(input_data).encode("ascii"))
 
     def handle_drbg_reseed(
-        self, entropy, entropy_len, nonce, nonce_len, reseed_interval, mode, cfg, trigger
+        self,
+        entropy,
+        entropy_len,
+        nonce,
+        nonce_len,
+        reseed_interval,
+        mode,
+        cfg,
+        trigger,
     ) -> None:
         """Call the cryptolib DRBG to reseed.
 
@@ -240,9 +247,7 @@ def handle_drbg_generate(
         }
         self.target.write(json.dumps(input_data).encode("ascii"))
 
-    def handle_trng_init(
-        self, mode, cfg, trigger
-    ) -> None:
+    def handle_trng_init(self, mode, cfg, trigger) -> None:
         """Call the cryptolib TRNG to init.
 
         Args:
@@ -259,9 +264,7 @@ def handle_trng_init(
         }
         self.target.write(json.dumps(input_data).encode("ascii"))
 
-    def handle_trng_generate(
-        self, cfg, trigger
-    ) -> None:
+    def handle_trng_generate(self, cfg, trigger) -> None:
         """Call the cryptolib TRNG to generate randomness.
 
         Args:

sw/host/penetrationtests/python/fi/test_scripts/fi_asym_cryptolib_python_test.py

@@ -38,7 +38,6 @@
 
 
 class SymCryptolibFiTest(unittest.TestCase):
-
     def test_init(self):
         asymfi = OTFIAsymCrypto(target)
         device_id, sensors, alerts, owner_page, boot_log, boot_measurements, version = (

sw/host/penetrationtests/python/fi/test_scripts/fi_sym_cryptolib_python_test.py

@@ -37,7 +37,6 @@
 
 
 class SymCryptolibFiTest(unittest.TestCase):
-
     def test_init(self):
         symfi = OTFISymCrypto(target)
         device_id, sensors, alerts, owner_page, boot_log, boot_measurements, version = (

sw/host/penetrationtests/python/sca/communication/sca_asym_cryptolib_commands.py

@@ -6,6 +6,7 @@
 Communication with OpenTitan happens over the uJson
 command interface.
 """
+
 import json
 import time
 from sw.host.penetrationtests.python.util import common_library
@@ -94,17 +95,7 @@ def handle_rsa_dec(
         self.target.write(json.dumps(input_data).encode("ascii"))
 
     def handle_rsa_sign(
-        self,
-        data,
-        data_len,
-        e,
-        n,
-        n_len,
-        d,
-        padding,
-        hashing,
-        cfg,
-        trigger
+        self, data, data_len, e, n, n_len, d, padding, hashing, cfg, trigger
     ) -> None:
         """Call the cryptolib RSA to sign.
 
@@ -168,7 +159,7 @@ def handle_p256_base_mult_fvsr(self, scalar, cfg, trigger, num_iterations) -> No
             "scalar": scalar,
             "cfg": cfg,
             "trigger": trigger,
-            "num_iterations": num_iterations
+            "num_iterations": num_iterations,
         }
         self.target.write(json.dumps(input_data).encode("ascii"))
 
@@ -187,7 +178,7 @@ def handle_p256_base_mult_daisy(self, scalar, cfg, trigger, num_iterations) -> N
             "scalar": scalar,
             "cfg": cfg,
             "trigger": trigger,
-            "num_iterations": num_iterations
+            "num_iterations": num_iterations,
         }
         self.target.write(json.dumps(input_data).encode("ascii"))
 
@@ -269,7 +260,7 @@ def handle_p384_base_mult_fvsr(self, scalar, cfg, trigger, num_iterations) -> No
             "scalar": scalar,
             "cfg": cfg,
             "trigger": trigger,
-            "num_iterations": num_iterations
+            "num_iterations": num_iterations,
         }
         self.target.write(json.dumps(input_data).encode("ascii"))
 
@@ -288,7 +279,7 @@ def handle_p384_base_mult_daisy(self, scalar, cfg, trigger, num_iterations) -> N
             "scalar": scalar,
             "cfg": cfg,
             "trigger": trigger,
-            "num_iterations": num_iterations
+            "num_iterations": num_iterations,
         }
         self.target.write(json.dumps(input_data).encode("ascii"))
 

sw/host/penetrationtests/python/sca/communication/sca_sym_cryptolib_commands.py

@@ -6,6 +6,7 @@
 Communication with OpenTitan happens over the uJson
 command interface.
 """
+
 import json
 import time
 from sw.host.penetrationtests.python.util import common_library