Use global rsa module

This commit is contained in:
shortcutme 2019-03-16 02:39:11 +01:00
parent ee631730c7
commit 0e2f7fb122
No known key found for this signature in database
GPG Key ID: 5B63BAE6CB9613AE
1 changed files with 6 additions and 6 deletions

View File

@ -2,8 +2,8 @@ import base64
import hashlib
def sign(data, privatekey):
from lib import rsa
from lib.rsa import pkcs1
import rsa
from rsa import pkcs1
if "BEGIN RSA PRIVATE KEY" not in privatekey:
privatekey = "-----BEGIN RSA PRIVATE KEY-----\n%s\n-----END RSA PRIVATE KEY-----" % privatekey
@ -13,8 +13,8 @@ def sign(data, privatekey):
return sign
def verify(data, publickey, sign):
from lib import rsa
from lib.rsa import pkcs1
import rsa
from rsa import pkcs1
pub = rsa.PublicKey.load_pkcs1(publickey, format="DER")
try:
@ -24,8 +24,8 @@ def verify(data, publickey, sign):
return valid
def privatekeyToPublickey(privatekey):
from lib import rsa
from lib.rsa import pkcs1
import rsa
from rsa import pkcs1
if "BEGIN RSA PRIVATE KEY" not in privatekey:
privatekey = "-----BEGIN RSA PRIVATE KEY-----\n%s\n-----END RSA PRIVATE KEY-----" % privatekey