Add stubs for wolfcrypt

Author: Max-Sentyron

pyrightconfig.stricter.json

@@ -92,6 +92,7 @@
         "stubs/tensorflow",
         "stubs/tqdm",
         "stubs/vobject",
+        "stubs/wolfcrypt",
         "stubs/workalendar",
         "stubs/xmldiff",
     ],

stubs/wolfcrypt/METADATA.toml

@@ -0,0 +1,2 @@
+version = "5.8.*"
+upstream_repository = "https://github.com/wolfssl/wolfcrypt-py"

stubs/wolfcrypt/wolfcrypt/__init__.pyi

@@ -0,0 +1,30 @@
+from wolfcrypt._version import __version__ as __version__
+
+__all__ = [
+    "__title__",
+    "__summary__",
+    "__uri__",
+    "__version__",
+    "__author__",
+    "__email__",
+    "__license__",
+    "__copyright__",
+    "ciphers",
+    "hashes",
+    "random",
+    "pwdbased",
+]
+
+__title__: str
+__summary__: str
+__uri__: str
+__author__: str
+__email__: str
+__license__: str
+__copyright__: str
+
+# Names in __all__ with no definition:
+#   ciphers
+#   hashes
+#   pwdbased
+#   random

stubs/wolfcrypt/wolfcrypt/_ffi.pyi

@@ -0,0 +1,3 @@
+import _cffi_backend
+
+ffi: _cffi_backend.FFI

stubs/wolfcrypt/wolfcrypt/_version.pyi

@@ -0,0 +1,2 @@
+__wolfssl_version__: str
+__version__: str

stubs/wolfcrypt/wolfcrypt/asn.pyi

@@ -0,0 +1,7 @@
+from wolfcrypt.hashes import _Hash
+
+def pem_to_der(pem:bytes, pem_type:int)->bytes: ...
+def der_to_pem(der:bytes, pem_type:int)->bytes: ...
+def hash_oid_from_class(hash_cls: type[_Hash])->int: ...
+def make_signature(data: bytes, hash_cls:type[_Hash], key:str|bytes=None)->bytes: ...
+def check_signature(signature: bytes, data:bytes, hash_cls:type[_Hash], pub_key)->bool: ...

stubs/wolfcrypt/wolfcrypt/ciphers.pyi

@@ -0,0 +1,292 @@
+from _typeshed import Incomplete
+from enum import IntEnum
+from typing import Literal, overload
+
+MODE_ECB = Literal[1]
+MODE_CBC = Literal[2]
+MODE_CFB = Literal[3]
+MODE_OFB = Literal[5]
+MODE_CTR = Literal[6]
+_FEEDBACK_MODES = Literal[MODE_ECB, MODE_CBC, MODE_CFB, MODE_OFB, MODE_CTR]
+
+ECC_CURVE_INVALID: int
+ECC_CURVE_DEF: int
+ECC_SECP192R1: int
+ECC_PRIME192V2: int
+ECC_PRIME192V3: int
+ECC_PRIME239V1: int
+ECC_PRIME239V2: int
+ECC_PRIME239V3: int
+ECC_SECP256R1: int
+ECC_SECP112R1: int
+ECC_SECP112R2: int
+ECC_SECP128R1: int
+ECC_SECP128R2: int
+ECC_SECP160R1: int
+ECC_SECP160R2: int
+ECC_SECP224R1: int
+ECC_SECP384R1: int
+ECC_SECP521R1: int
+ECC_SECP160K1: int
+ECC_SECP192K1: int
+ECC_SECP224K1: int
+ECC_SECP256K1: int
+ECC_BRAINPOOLP160R1: int
+ECC_BRAINPOOLP192R1: int
+ECC_BRAINPOOLP224R1: int
+ECC_BRAINPOOLP256R1: int
+ECC_BRAINPOOLP320R1: int
+ECC_BRAINPOOLP384R1: int
+ECC_BRAINPOOLP512R1: int
+
+MGF1NONE: int
+MGF1SHA1: int
+MGF1SHA224: int
+MGF1SHA256: int
+MGF1SHA384: int
+MGF1SHA512: int
+HASH_TYPE_NONE: int
+HASH_TYPE_MD2: int
+HASH_TYPE_MD4: int
+HASH_TYPE_MD5: int
+HASH_TYPE_SHA: int
+HASH_TYPE_SHA224: int
+HASH_TYPE_SHA256: int
+HASH_TYPE_SHA384: int
+HASH_TYPE_SHA512: int
+HASH_TYPE_MD5_SHA: int
+HASH_TYPE_SHA3_224: int
+HASH_TYPE_SHA3_256: int
+HASH_TYPE_SHA3_384: int
+HASH_TYPE_SHA3_512: int
+HASH_TYPE_BLAKE2B: int
+HASH_TYPE_BLAKE2S: int
+
+
+class _Cipher:
+    mode: _FEEDBACK_MODES
+    @overload
+    def __init__(self, key:str| bytes, mode: Literal[MODE_CBC, MODE_CTR], IV:str| bytes) -> None: ...
+    @overload
+    def __init__(self, key:str| bytes, mode: _FEEDBACK_MODES, IV:str| bytes|None=None) -> None: ...
+    @overload
+    @classmethod
+    def new(cls, key:str| bytes, mode: Literal[MODE_CBC, MODE_CTR], IV:str| bytes, **kwargs): ...
+    @overload
+    @classmethod
+    def new(cls, key:str| bytes, mode: _FEEDBACK_MODES, IV:str| bytes|None=None, **kwargs): ...
+    def encrypt(self, string: str|bytes)->bytes: ...
+    def decrypt(self, string: str|bytes)->bytes: ...
+
+class Aes(_Cipher):
+    block_size: int
+    key_size: None
+
+class AesGcmStream:
+    block_size: int
+    def __init__(self, key: str|bytes, IV: str|bytes, tag_bytes: int = 16) -> None: ...
+    def set_aad(self, data: str|bytes) -> None: ...
+    def get_aad(self)->bytes: ...
+    def encrypt(self, data: str|bytes)->bytes: ...
+    def decrypt(self, data: str|bytes)->bytes: ...
+    def final(self, authTag=None): ...
+
+class ChaCha(_Cipher):
+    block_size: int
+    key_size: int
+    def __init__(self, key: str = "", size: int = 32) -> None: ...
+    def set_iv(self, nonce, counter: int = 0) -> None: ...
+
+class ChaCha20Poly1305:
+    block_size: int
+    def __init__(self, key, IV, aad, tag_bytes: int = 16) -> None: ...
+    def set_aad(self, data) -> None: ...
+    def get_aad(self): ...
+    def encrypt(self, inPlainText): ...
+    def decrypt(self, inGeneratedAuthTag, inGeneratedCipher): ...
+    def checkTag(self, authTag) -> None: ...
+    def final(self, authTag=None): ...
+
+class Des3(_Cipher):
+    block_size: int
+    key_size: int
+
+class _Rsa:
+    RSA_MIN_PAD_SIZE: int
+    native_object: Incomplete
+    def __init__(self) -> None: ...
+    def __del__(self) -> None: ...
+    def set_mgf(self, mgf) -> None: ...
+
+class RsaPublic(_Rsa):
+    output_size: Incomplete
+    size: Incomplete
+    def __init__(self, key=None, hash_type=None) -> None: ...
+    @classmethod
+    def from_pem(cls, file, hash_type=None): ...
+    def encrypt(self, plaintext): ...
+    def encrypt_oaep(self, plaintext, label: str = ""): ...
+    def verify(self, signature): ...
+    def verify_pss(self, plaintext, signature): ...
+
+class RsaPrivate(RsaPublic):
+    @classmethod
+    def make_key(cls, size, rng=..., hash_type=None): ...
+    size: Incomplete
+    output_size: Incomplete
+    def __init__(self, key=None, hash_type=None) -> None: ...
+    @classmethod
+    def from_pem(cls, file, hash_type=None): ...
+    def encode_key(self): ...
+    def decrypt(self, ciphertext): ...
+    def decrypt_oaep(self, ciphertext, label: str = ""): ...
+    def sign(self, plaintext): ...
+    def sign_pss(self, plaintext): ...
+
+class _Ecc:
+    native_object: Incomplete
+    def __init__(self) -> None: ...
+    def __del__(self) -> None: ...
+    @property
+    def size(self): ...
+    @property
+    def max_signature_size(self): ...
+
+class EccPublic(_Ecc):
+    def __init__(self, key=None) -> None: ...
+    def decode_key(self, key) -> None: ...
+    def decode_key_raw(self, qx, qy, curve_id=7) -> None: ...
+    def encode_key(self, with_curve: bool = True): ...
+    def encode_key_raw(self): ...
+    def import_x963(self, x963) -> None: ...
+    def export_x963(self): ...
+    def verify(self, signature, data): ...
+    def verify_raw(self, R, S, data): ...
+
+class EccPrivate(EccPublic):
+    @classmethod
+    def make_key(cls, size, rng=...): ...
+    def decode_key(self, key) -> None: ...
+    def decode_key_raw(self, qx, qy, d, curve_id=7) -> None: ...
+    def encode_key(self): ...
+    def encode_key_raw(self): ...
+    def shared_secret(self, peer): ...
+    def sign(self, plaintext, rng=...): ...
+    def sign_raw(self, plaintext, rng=...): ...
+
+class _Ed25519:
+    native_object: Incomplete
+    def __init__(self) -> None: ...
+    def __del__(self) -> None: ...
+    @property
+    def size(self): ...
+    @property
+    def max_signature_size(self): ...
+
+class Ed25519Public(_Ed25519):
+    def __init__(self, key=None) -> None: ...
+    def decode_key(self, key) -> None: ...
+    def encode_key(self): ...
+    def verify(self, signature, data): ...
+
+class Ed25519Private(Ed25519Public):
+    def __init__(self, key=None, pub=None) -> None: ...
+    @classmethod
+    def make_key(cls, size, rng=...): ...
+    def decode_key(self, key, pub=None) -> None: ...
+    def encode_key(self): ...
+    def sign(self, plaintext): ...
+
+class _Ed448:
+    native_object: Incomplete
+    def __init__(self) -> None: ...
+    def __del__(self) -> None: ...
+    @property
+    def size(self): ...
+    @property
+    def max_signature_size(self): ...
+
+class Ed448Public(_Ed448):
+    def __init__(self, key=None) -> None: ...
+    def decode_key(self, key) -> None: ...
+    def encode_key(self): ...
+    def verify(self, signature, data, ctx=None): ...
+
+class Ed448Private(Ed448Public):
+    def __init__(self, key=None, pub=None) -> None: ...
+    @classmethod
+    def make_key(cls, size, rng=...): ...
+    def decode_key(self, key, pub=None) -> None: ...
+    def encode_key(self): ...
+    def sign(self, plaintext, ctx=None): ...
+
+class MlKemType(IntEnum):
+    ML_KEM_512 = ...
+    ML_KEM_768 = ...
+    ML_KEM_1024 = ...
+
+class _MlKemBase:
+    INVALID_DEVID: Incomplete
+    init_done: bool
+    native_object: Incomplete
+    def __init__(self, mlkem_type) -> None: ...
+    def __del__(self) -> None: ...
+    @property
+    def ct_size(self): ...
+    @property
+    def ss_size(self): ...
+
+class MlKemPublic(_MlKemBase):
+    @property
+    def key_size(self): ...
+    def encode_key(self): ...
+    def decode_key(self, pub_key) -> None: ...
+    def encapsulate(self, rng=...): ...
+    def encapsulate_with_random(self, rand): ...
+
+class MlKemPrivate(_MlKemBase):
+    @classmethod
+    def make_key(cls, mlkem_type, rng=...): ...
+    @classmethod
+    def make_key_with_random(cls, mlkem_type, rand): ...
+    @property
+    def pub_key_size(self): ...
+    @property
+    def priv_key_size(self): ...
+    def encode_pub_key(self): ...
+    def encode_priv_key(self): ...
+    def decode_key(self, priv_key: tuple[bytes, str]): ...
+    native_object: Incomplete
+    def decapsulate(self, ct): ...
+
+class MlDsaType(IntEnum):
+    ML_DSA_44 = ...
+    ML_DSA_65 = ...
+    ML_DSA_87 = ...
+
+class _MlDsaBase:
+    INVALID_DEVID: Incomplete
+    native_object: Incomplete
+    def __init__(self, mldsa_type) -> None: ...
+    def __del__(self) -> None: ...
+    @property
+    def sig_size(self): ...
+    def verify(self, signature, message): ...
+
+class MlDsaPrivate(_MlDsaBase):
+    @classmethod
+    def make_key(cls, mldsa_type, rng=...): ...
+    @property
+    def pub_key_size(self): ...
+    @property
+    def priv_key_size(self): ...
+    def encode_pub_key(self): ...
+    def encode_priv_key(self): ...
+    def decode_key(self, priv_key, pub_key=None) -> None: ...
+    def sign(self, message, rng=...): ...
+
+class MlDsaPublic(_MlDsaBase):
+    @property
+    def key_size(self): ...
+    def decode_key(self, pub_key): ...
+    def encode_key(self): ...

stubs/wolfcrypt/wolfcrypt/exceptions.pyi

@@ -0,0 +1 @@
+class WolfCryptError(Exception): ...