mirror of
https://github.com/HelloZeroNet/ZeroNet.git
synced 2023-12-14 04:33:03 +01:00
296e4aab57
* Use sslcrypto instead of pyelliptic and pybitcointools * Fix CryptMessage * Support Python 3.4 * Fix user creation * Get rid of pyelliptic and pybitcointools * Fix typo * Delete test file * Add sslcrypto to tree * Update sslcrypto * Add pyaes to src/lib * Fix typo in tests * Update sslcrypto version * Use privatekey_bin instead of privatekey for bytes objects * Fix sslcrypto * Fix Benchmark plugin * Don't calculate the same thing twice * Only import sslcrypto once * Handle fallback sslcrypto implementation during tests * Fix sslcrypto fallback implementation selection * Fix thread safety * Add derivation * Bring split back * Fix typo * v3.3 * Fix custom OpenSSL discovery
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
# pylint: disable=import-outside-toplevel
|
|
|
|
class AES:
|
|
def __init__(self, backend, fallback=None):
|
|
self._backend = backend
|
|
self._fallback = fallback
|
|
|
|
|
|
def get_algo_key_length(self, algo):
|
|
if algo.count("-") != 2:
|
|
raise ValueError("Invalid algorithm name")
|
|
try:
|
|
return int(algo.split("-")[1]) // 8
|
|
except ValueError:
|
|
raise ValueError("Invalid algorithm name") from None
|
|
|
|
|
|
def new_key(self, algo="aes-256-cbc"):
|
|
if not self._backend.is_algo_supported(algo):
|
|
if self._fallback is None:
|
|
raise ValueError("This algorithm is not supported")
|
|
return self._fallback.new_key(algo)
|
|
return self._backend.random(self.get_algo_key_length(algo))
|
|
|
|
|
|
def encrypt(self, data, key, algo="aes-256-cbc"):
|
|
if not self._backend.is_algo_supported(algo):
|
|
if self._fallback is None:
|
|
raise ValueError("This algorithm is not supported")
|
|
return self._fallback.encrypt(data, key, algo)
|
|
|
|
key_length = self.get_algo_key_length(algo)
|
|
if len(key) != key_length:
|
|
raise ValueError("Expected key to be {} bytes, got {} bytes".format(key_length, len(key)))
|
|
|
|
return self._backend.encrypt(data, key, algo)
|
|
|
|
|
|
def decrypt(self, ciphertext, iv, key, algo="aes-256-cbc"):
|
|
if not self._backend.is_algo_supported(algo):
|
|
if self._fallback is None:
|
|
raise ValueError("This algorithm is not supported")
|
|
return self._fallback.decrypt(ciphertext, iv, key, algo)
|
|
|
|
key_length = self.get_algo_key_length(algo)
|
|
if len(key) != key_length:
|
|
raise ValueError("Expected key to be {} bytes, got {} bytes".format(key_length, len(key)))
|
|
|
|
return self._backend.decrypt(ciphertext, iv, key, algo)
|
|
|
|
|
|
def get_backend(self):
|
|
return self._backend.get_backend()
|