mirror of
https://gitlab.com/pulsechaincom/staking-deposit-cli.git
synced 2025-01-10 13:01:22 +00:00
353c40215f
* dev: (22 commits) Remove BLS incompatability warning Adds preproduction warning Adds typing & Moves global variables into pytest function args Linting fixes Test key_info != b'' import pytest into test_tree Apply suggestions from @hwwhww's code review Fix lint Add Altona setting Remove extra newline (lint) Update tests for new EIP2333 (bls v2 compliant HKDF_mod_r) Rename json file fields Create uuid at execution time instead of class attribute Add uuid test Fix key path Add Witti gitignore dist/ Revert to DepositMessage & DepositData as per spec Add fork_version adds build/ to .gitignore ...
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from typing import Any
|
|
|
|
from Crypto.Hash import (
|
|
SHA256 as _sha256,
|
|
SHA512 as _sha512,
|
|
)
|
|
from Crypto.Protocol.KDF import (
|
|
scrypt as _scrypt,
|
|
HKDF as _HKDF,
|
|
PBKDF2 as _PBKDF2,
|
|
)
|
|
from Crypto.Cipher import (
|
|
AES as _AES
|
|
)
|
|
|
|
|
|
def SHA256(x: bytes) -> bytes:
|
|
return _sha256.new(x).digest()
|
|
|
|
|
|
def scrypt(*, password: str, salt: str, n: int, r: int, p: int, dklen: int) -> bytes:
|
|
assert(n < 2**(128 * r / 8))
|
|
res = _scrypt(password=password, salt=salt, key_len=dklen, N=n, r=r, p=p)
|
|
return res if isinstance(res, bytes) else res[0] # PyCryptodome can return Tuple[bytes]
|
|
|
|
|
|
def PBKDF2(*, password: bytes, salt: bytes, dklen: int, c: int, prf: str) -> bytes:
|
|
assert('sha' in prf)
|
|
_hash = _sha256 if 'sha256' in prf else _sha512
|
|
res = _PBKDF2(password=password, salt=salt, dkLen=dklen, count=c, hmac_hash_module=_hash) # type: ignore
|
|
return res if isinstance(res, bytes) else res[0] # PyCryptodome can return Tuple[bytes]
|
|
|
|
|
|
def HKDF(*, salt: bytes, IKM: bytes, L: int, info: bytes=b'') -> bytes:
|
|
res = _HKDF(master=IKM, key_len=L, salt=salt, hashmod=_sha256, context=info)
|
|
return res if isinstance(res, bytes) else res[0] # PyCryptodome can return Tuple[bytes]
|
|
|
|
|
|
def AES_128_CTR(*, key: bytes, iv: bytes) -> Any:
|
|
assert len(key) == 16
|
|
return _AES.new(key=key, mode=_AES.MODE_CTR, initial_value=iv, nonce=b'')
|