feat: 支持 piico 设备国密加密

apps/common/sdk/gm/piico/device.py

@@ -0,0 +1,70 @@
+from ctypes import *
+
+from .exception import PiicoError
+from .session import Session
+from .cipher import *
+from .digest import *
+
+
+class Device:
+    _driver = None
+    __device = None
+
+    def open(self, driver_path="./libpiico_ccmu"):
+        # load driver
+        self.__load_driver(driver_path)
+        # open device
+        self.__open_device()
+
+    def close(self):
+        if self.__device is None:
+            raise Exception("device not turned on")
+        ret = self._driver.SDF_CloseDevice(self.__device)
+        if not ret == 0:
+            raise Exception("turn off device failed")
+        self.__device = None
+
+    def new_session(self):
+        session = c_void_p()
+        ret = self._driver.SDF_OpenSession(self.__device, pointer(session))
+        if not ret == 0:
+            raise Exception("create session failed")
+        return Session(self._driver, session)
+
+    def generate_ecc_key_pair(self):
+        session = self.new_session()
+        return session.generate_ecc_key_pair(alg_id=0x00020200)
+
+    def generate_random(self, length=64):
+        session = self.new_session()
+        return session.generate_random(length)
+
+    def new_sm2_ecc_cipher(self, public_key, private_key):
+        session = self.new_session()
+        return ECCCipher(session, public_key, private_key)
+
+    def new_sm4_ebc_cipher(self, key_val):
+        session = self.new_session()
+        return EBCCipher(session, key_val)
+
+    def new_sm4_cbc_cipher(self, key_val, iv):
+        session = self.new_session()
+        return CBCCipher(session, key_val, iv)
+
+    def new_digest(self, mode="sm3"):
+        session = self.new_session()
+        return Digest(session, mode)
+
+    def __load_driver(self, path):
+        # check driver status
+        if self._driver is not None:
+            raise Exception("already load driver")
+        # load driver
+        self._driver = cdll.LoadLibrary(path)
+
+    def __open_device(self):
+        device = c_void_p()
+        ret = self._driver.SDF_OpenDevice(pointer(device))
+        if not ret == 0:
+            raise PiicoError("open piico device failed", ret)
+        self.__device = device