github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-20 02:51:27 +00:00
Code Issues Projects Releases Wiki Activity

perf: 优化代码表达

Browse Source
This commit is contained in:
Aaron3S
2022-08-23 20:19:53 +08:00
parent 8772cd8c71
commit 51820f23bf
3 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
apps/common/sdk/gm/piico/session.py
View File

@@ -17,7 +17,7 @@ class Session(SM2Mixin, SM3Mixin, SM4Mixin):
def generate_random(self, length=64):
random_data = (c_ubyte * length)()
ret = self._driver.SDF_GenerateRandom(self._session, c_int(length), random_data)
if not ret == 0:
if ret != 0:
raise PiicoError("generate random error", ret)
return bytes(random_data)
@@ -26,11 +26,11 @@ class Session(SM2Mixin, SM3Mixin, SM4Mixin):
private_key = ECCrefPrivateKey()
ret = self._driver.SDF_GenerateKeyPair_ECC(self._session, c_int(alg_id), c_int(256), pointer(public_key),
pointer(private_key))
if not ret == 0:
if ret != 0:
raise PiicoError("generate ecc key pair failed", ret)
return ECCKeyPair(public_key.encode(), private_key.encode())
def close(self):
ret = self._driver.SDF_CloseSession(self._session)
if not ret == 0:
if ret != 0:
raise PiicoError("close session failed", ret)