kopia lustrzana https://dev.funkwhale.audio/funkwhale/funkwhale
49 wiersze
1.4 KiB
Python
49 wiersze
1.4 KiB
Python
from cryptography.hazmat.primitives import serialization as crypto_serialization
|
|
from cryptography.hazmat.primitives.asymmetric import rsa
|
|
from cryptography.hazmat.backends import default_backend as crypto_default_backend
|
|
|
|
import re
|
|
import urllib.parse
|
|
|
|
from . import exceptions
|
|
|
|
KEY_ID_REGEX = re.compile(r'keyId=\"(?P<id>.*)\"')
|
|
|
|
|
|
def get_key_pair(size=2048):
|
|
key = rsa.generate_private_key(
|
|
backend=crypto_default_backend(),
|
|
public_exponent=65537,
|
|
key_size=size
|
|
)
|
|
private_key = key.private_bytes(
|
|
crypto_serialization.Encoding.PEM,
|
|
crypto_serialization.PrivateFormat.PKCS8,
|
|
crypto_serialization.NoEncryption())
|
|
public_key = key.public_key().public_bytes(
|
|
crypto_serialization.Encoding.PEM,
|
|
crypto_serialization.PublicFormat.PKCS1
|
|
)
|
|
|
|
return private_key, public_key
|
|
|
|
|
|
def get_key_id_from_signature_header(header_string):
|
|
parts = header_string.split(',')
|
|
try:
|
|
raw_key_id = [p for p in parts if p.startswith('keyId="')][0]
|
|
except IndexError:
|
|
raise ValueError('Missing key id')
|
|
|
|
match = KEY_ID_REGEX.match(raw_key_id)
|
|
if not match:
|
|
raise ValueError('Invalid key id')
|
|
|
|
key_id = match.groups()[0]
|
|
url = urllib.parse.urlparse(key_id)
|
|
if not url.scheme or not url.netloc:
|
|
raise ValueError('Invalid url')
|
|
if url.scheme not in ['http', 'https']:
|
|
raise ValueError('Invalid shceme')
|
|
return key_id
|