mirror of
https://github.com/HelloZeroNet/ZeroNet.git
synced 2023-12-14 04:33:03 +01:00
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
import base64
|
||
|
|
import hashlib
|
||
|
|
|
||
|
|
def sign(data, privatekey):
|
||
|
|
from lib import rsa
|
||
|
|
from lib.rsa import pkcs1
|
||
|
|
|
||
|
|
if "BEGIN RSA PRIVATE KEY" not in privatekey:
|
||
|
|
privatekey = "-----BEGIN RSA PRIVATE KEY-----\n%s\n-----END RSA PRIVATE KEY-----" % privatekey
|
||
|
|
|
||
|
|
priv = rsa.PrivateKey.load_pkcs1(privatekey)
|
||
|
|
sign = rsa.pkcs1.sign(data, priv, 'SHA-256')
|
||
|
|
return sign
|
||
|
|
|
||
|
|
def verify(data, publickey, sign):
|
||
|
|
from lib import rsa
|
||
|
|
from lib.rsa import pkcs1
|
||
|
|
|
||
|
|
pub = rsa.PublicKey.load_pkcs1(publickey, format="DER")
|
||
|
|
try:
|
||
|
|
valid = rsa.pkcs1.verify(data, sign, pub)
|
||
|
|
except pkcs1.VerificationError:
|
||
|
|
valid = False
|
||
|
|
return valid
|
||
|
|
|
||
|
|
def privatekeyToPublickey(privatekey):
|
||
|
|
from lib import rsa
|
||
|
|
from lib.rsa import pkcs1
|
||
|
|
|
||
|
|
if "BEGIN RSA PRIVATE KEY" not in privatekey:
|
||
|
|
privatekey = "-----BEGIN RSA PRIVATE KEY-----\n%s\n-----END RSA PRIVATE KEY-----" % privatekey
|
||
|
|
|
||
|
|
priv = rsa.PrivateKey.load_pkcs1(privatekey)
|
||
|
|
pub = rsa.PublicKey(priv.n, priv.e)
|
||
|
|
return pub.save_pkcs1("DER")
|
||
|
|
|
||
|
|
def publickeyToOnion(publickey):
|
||
|
|
return base64.b32encode(hashlib.sha1(publickey).digest()[:10]).lower()
|