[PATCH 0/2] MR11097: bcrypt: Use SymCrypt for keys.
From: Hans Leidekker <hans@codeweavers.com> --- dlls/bcrypt/bcrypt_main.c | 3213 ++++++++++++++++++++++++------------ dlls/bcrypt/tests/bcrypt.c | 18 +- include/bcrypt.h | 3 + 3 files changed, 2208 insertions(+), 1026 deletions(-) diff --git a/dlls/bcrypt/bcrypt_main.c b/dlls/bcrypt/bcrypt_main.c index d57db526aa1..74e3d99b5e7 100644 --- a/dlls/bcrypt/bcrypt_main.c +++ b/dlls/bcrypt/bcrypt_main.c @@ -31,13 +31,63 @@ #include "symcrypt.h" #include "wine/debug.h" -#include "bcrypt_internal.h" WINE_DEFAULT_DEBUG_CHANNEL(bcrypt); SYMCRYPT_ENVIRONMENT_DEFS( WindowsUsermodeWin8_1nLater ); -#define UNIX_CALL( func, params ) WINE_UNIX_CALL( unix_ ## func, params ) +#define MAGIC_DSS1 ('D' | ('S' << 8) | ('S' << 16) | ('1' << 24)) +#define MAGIC_DSS2 ('D' | ('S' << 8) | ('S' << 16) | ('2' << 24)) + +#define MAGIC_ALG (('A' << 24) | ('L' << 16) | ('G' << 8) | '0') +#define MAGIC_HASH (('H' << 24) | ('A' << 16) | ('S' << 8) | 'H') +#define MAGIC_KEY (('K' << 24) | ('E' << 16) | ('Y' << 8) | '0') +#define MAGIC_SECRET (('S' << 24) | ('C' << 16) | ('R' << 8) | 'T') +struct object +{ + ULONG magic; +}; + +enum alg_id +{ + /* cipher */ + ALG_ID_3DES, + ALG_ID_AES, + ALG_ID_RC4, + + /* hash */ + ALG_ID_SHA256, + ALG_ID_SHA384, + ALG_ID_SHA512, + ALG_ID_SHA1, + ALG_ID_MD5, + ALG_ID_MD4, + ALG_ID_MD2, + + /* asymmetric encryption */ + ALG_ID_RSA, + + /* secret agreement */ + ALG_ID_DH, + ALG_ID_ECDH, + ALG_ID_ECDH_P256, + ALG_ID_ECDH_P384, + ALG_ID_ECDH_P521, + + /* signature */ + ALG_ID_RSA_SIGN, + ALG_ID_ECDSA, + ALG_ID_ECDSA_P256, + ALG_ID_ECDSA_P384, + ALG_ID_ECDSA_P521, + ALG_ID_DSA, + + /* rng */ + ALG_ID_RNG, + + /* key derivation */ + ALG_ID_PBKDF2, +}; #define HASH_FLAG_HMAC 0x01 #define HASH_FLAG_REUSABLE 0x02 @@ -56,6 +106,178 @@ struct hash SYMCRYPT_HMAC_STATE hmac_state; }; +enum chain_mode +{ + CHAIN_MODE_CBC, + CHAIN_MODE_ECB, + CHAIN_MODE_CFB, + CHAIN_MODE_CCM, + CHAIN_MODE_GCM, +}; + +enum ecc_curve_id +{ + ECC_CURVE_NONE, + ECC_CURVE_25519, + ECC_CURVE_P256R1, + ECC_CURVE_P384R1, + ECC_CURVE_P521R1, +}; + +struct algorithm +{ + struct object hdr; + enum alg_id id; + enum chain_mode mode; + ULONG flags; + enum ecc_curve_id curve_id; +}; + +struct aes_key +{ + SYMCRYPT_AES_EXPANDED_KEY handle; +}; + +struct gcm_key +{ + SYMCRYPT_GCM_EXPANDED_KEY handle; + SYMCRYPT_GCM_STATE state; +}; + +#define BLOCK_LENGTH_RC4 1 +#define BLOCK_LENGTH_3DES 8 +#define BLOCK_LENGTH_AES 16 + +struct symmetric_key +{ + enum chain_mode mode; + ULONG block_size; + UCHAR *secret; + ULONG secret_len; + UCHAR vector[BLOCK_LENGTH_AES]; + CRITICAL_SECTION cs; + union + { + struct aes_key aes; + struct gcm_key gcm; + }; +}; + +struct rsa_key +{ + SYMCRYPT_RSAKEY *handle; +}; + +struct ecc_key +{ + SYMCRYPT_ECKEY *handle; + SYMCRYPT_ECURVE *curve; + enum ecc_curve_id curve_id; +}; + +struct dsa_key +{ + SYMCRYPT_DLKEY *handle; + SYMCRYPT_DLGROUP *group; + UCHAR seed[20]; + UINT32 count; +}; + +struct asymmetric_key +{ + ULONG bitlen; /* key strength for ECC keys */ + union + { + struct rsa_key rsa; + struct ecc_key ecc; + struct dsa_key dsa; + }; +}; + +#define KEY_FLAG_PRIVATE 0x00000001 +#define KEY_FLAG_FINALIZED 0x00000002 + +struct key +{ + struct object hdr; + enum alg_id alg_id; + ULONG flags; + union + { + struct symmetric_key s; + struct asymmetric_key a; + }; +}; + +#define KEY_EXPORT_FLAG_PRIVATE 0x00000001 +#define KEY_EXPORT_FLAG_RSA_FULL 0x00000002 + +struct secret +{ + struct object hdr; + UCHAR *derived_key; + ULONG derived_key_len; + struct key *privkey; + struct key *pubkey; +}; + +void * SYMCRYPT_CALL SymCryptCallbackAlloc( SIZE_T size ) +{ + return _aligned_malloc( size, SYMCRYPT_ASYM_ALIGN_VALUE ); +} + +void SYMCRYPT_CALL SymCryptCallbackFree( void *ptr ) +{ + _aligned_free( ptr ); +} + +SYMCRYPT_ERROR SYMCRYPT_CALL SymCryptCallbackRandom( BYTE *buf, SIZE_T size ) +{ + return RtlGenRandom( buf, size ) ? SYMCRYPT_NO_ERROR : SYMCRYPT_EXTERNAL_FAILURE; +} + +SYMCRYPT_ERROR SYMCRYPT_CALL SymCryptRsaSignVerifyPct( const SYMCRYPT_RSAKEY *key ) +{ + return SYMCRYPT_NO_ERROR; +} + +SYMCRYPT_ERROR SYMCRYPT_CALL SymCryptDsaPct( const SYMCRYPT_DLKEY *key ) +{ + return SYMCRYPT_NO_ERROR; +} + +void SYMCRYPT_CALL SymCryptRsaSelftest( void ) +{ +} + +SYMCRYPT_ERROR SYMCRYPT_CALL SymCryptEcDsaPct( const SYMCRYPT_ECKEY *key ) +{ + return SYMCRYPT_NO_ERROR; +} + +void SYMCRYPT_CALL SymCryptDsaSelftest( void ) +{ +} + +void SYMCRYPT_CALL SymCryptEcDsaSelftest( void ) +{ +} + +void SYMCRYPT_CALL SymCryptDhSecretAgreementSelftest( void ) +{ +} + +void SYMCRYPT_CALL SymCryptEcDhSecretAgreementSelftest( void ) +{ +} + +UINT32 g_SymCryptFipsSelftestsPerformed; + +static inline ULONG len_from_bitlen( ULONG bitlen ) +{ + return (bitlen + 7) / 8; +} + NTSTATUS WINAPI BCryptAddContextFunction( ULONG table, const WCHAR *ctx, ULONG iface, const WCHAR *func, ULONG pos ) { FIXME( "%#lx, %s, %#lx, %s, %lu: stub\n", table, debugstr_w(ctx), iface, debugstr_w(func), pos ); @@ -118,6 +340,7 @@ static const struct ULONG object_length; ULONG hash_length; ULONG block_bits; + enum ecc_curve_id curve_id; } builtin_algorithms[] = { @@ -134,38 +357,29 @@ builtin_algorithms[] = { BCRYPT_RSA_ALGORITHM, BCRYPT_ASYMMETRIC_ENCRYPTION_INTERFACE, 0, 0, 0 }, { BCRYPT_DH_ALGORITHM, BCRYPT_SECRET_AGREEMENT_INTERFACE, 0, 0, 0 }, { BCRYPT_ECDH_ALGORITHM, BCRYPT_SECRET_AGREEMENT_INTERFACE, 0, 0, 0 }, - { BCRYPT_ECDH_P256_ALGORITHM, BCRYPT_SECRET_AGREEMENT_INTERFACE, 0, 0, 0 }, - { BCRYPT_ECDH_P384_ALGORITHM, BCRYPT_SECRET_AGREEMENT_INTERFACE, 0, 0, 0 }, - { BCRYPT_ECDH_P521_ALGORITHM, BCRYPT_SECRET_AGREEMENT_INTERFACE, 0, 0, 0 }, + { BCRYPT_ECDH_P256_ALGORITHM, BCRYPT_SECRET_AGREEMENT_INTERFACE, 0, 0, 0, ECC_CURVE_P256R1 }, + { BCRYPT_ECDH_P384_ALGORITHM, BCRYPT_SECRET_AGREEMENT_INTERFACE, 0, 0, 0, ECC_CURVE_P384R1 }, + { BCRYPT_ECDH_P521_ALGORITHM, BCRYPT_SECRET_AGREEMENT_INTERFACE, 0, 0, 0, ECC_CURVE_P521R1 }, { BCRYPT_RSA_SIGN_ALGORITHM, BCRYPT_SIGNATURE_INTERFACE, 0, 0, 0 }, { BCRYPT_ECDSA_ALGORITHM, BCRYPT_SIGNATURE_INTERFACE, 0, 0, 0 }, - { BCRYPT_ECDSA_P256_ALGORITHM, BCRYPT_SIGNATURE_INTERFACE, 0, 0, 0 }, - { BCRYPT_ECDSA_P384_ALGORITHM, BCRYPT_SIGNATURE_INTERFACE, 0, 0, 0 }, - { BCRYPT_ECDSA_P521_ALGORITHM, BCRYPT_SIGNATURE_INTERFACE, 0, 0, 0 }, + { BCRYPT_ECDSA_P256_ALGORITHM, BCRYPT_SIGNATURE_INTERFACE, 0, 0, 0, ECC_CURVE_P256R1 }, + { BCRYPT_ECDSA_P384_ALGORITHM, BCRYPT_SIGNATURE_INTERFACE, 0, 0, 0, ECC_CURVE_P384R1 }, + { BCRYPT_ECDSA_P521_ALGORITHM, BCRYPT_SIGNATURE_INTERFACE, 0, 0, 0, ECC_CURVE_P521R1 }, { BCRYPT_DSA_ALGORITHM, BCRYPT_SIGNATURE_INTERFACE, 0, 0, 0 }, { BCRYPT_RNG_ALGORITHM, BCRYPT_RNG_INTERFACE, 0, 0, 0 }, { BCRYPT_PBKDF2_ALGORITHM, BCRYPT_KEY_DERIVATION_INTERFACE, 618, 0, 0 }, }; -static inline BOOL is_symmetric_key( const struct key *key ) -{ - return builtin_algorithms[key->alg_id].class == BCRYPT_CIPHER_INTERFACE - || builtin_algorithms[key->alg_id].class == BCRYPT_KEY_DERIVATION_INTERFACE; -} - -static inline BOOL is_asymmetric_encryption_key( struct key *key ) -{ - return builtin_algorithms[key->alg_id].class == BCRYPT_ASYMMETRIC_ENCRYPTION_INTERFACE; -} - -static inline BOOL is_agreement_key( struct key *key ) +static inline BOOL is_symmetric_alg( const struct algorithm *alg ) { - return builtin_algorithms[key->alg_id].class == BCRYPT_SECRET_AGREEMENT_INTERFACE; + return builtin_algorithms[alg->id].class == BCRYPT_CIPHER_INTERFACE + || builtin_algorithms[alg->id].class == BCRYPT_KEY_DERIVATION_INTERFACE; } -static inline BOOL is_signature_key( struct key *key ) +static inline BOOL is_symmetric_key( const struct key *key ) { - return builtin_algorithms[key->alg_id].class == BCRYPT_SIGNATURE_INTERFACE || key->alg_id == ALG_ID_RSA; + return builtin_algorithms[key->alg_id].class == BCRYPT_CIPHER_INTERFACE + || builtin_algorithms[key->alg_id].class == BCRYPT_KEY_DERIVATION_INTERFACE; } static BOOL match_operation_type( ULONG type, ULONG class ) @@ -331,7 +545,7 @@ static inline struct secret *get_secret_object( BCRYPT_SECRET_HANDLE handle ) NTSTATUS WINAPI BCryptGenRandom( BCRYPT_ALG_HANDLE handle, UCHAR *buffer, ULONG count, ULONG flags ) { const DWORD supported_flags = BCRYPT_USE_SYSTEM_PREFERRED_RNG; - struct algorithm *alg = get_alg_object( handle ); + const struct algorithm *alg = get_alg_object( handle ); TRACE("%p, %p, %lu, %#lx - semi-stub\n", handle, buffer, count, flags); @@ -360,9 +574,7 @@ NTSTATUS WINAPI BCryptGenRandom( BCRYPT_ALG_HANDLE handle, UCHAR *buffer, ULONG if (alg) FIXME("ignoring selected algorithm\n"); - /* When zero bytes are requested the function returns success too. */ - if (!count) - return STATUS_SUCCESS; + if (!count) return STATUS_SUCCESS; if (alg || (flags & BCRYPT_USE_SYSTEM_PREFERRED_RNG)) { @@ -373,14 +585,14 @@ NTSTATUS WINAPI BCryptGenRandom( BCRYPT_ALG_HANDLE handle, UCHAR *buffer, ULONG return STATUS_NOT_IMPLEMENTED; } -static struct algorithm *create_algorithm( enum alg_id id, enum chain_mode mode, DWORD flags ) +static struct algorithm *create_algorithm( enum alg_id id, DWORD flags ) { struct algorithm *ret; if (!(ret = calloc( 1, sizeof(*ret) ))) return NULL; ret->hdr.magic = MAGIC_ALG; ret->id = id; - ret->mode = mode; ret->flags = flags; + ret->curve_id = builtin_algorithms[id].curve_id; return ret; } @@ -421,15 +633,21 @@ NTSTATUS WINAPI BCryptOpenAlgorithmProvider( BCRYPT_ALG_HANDLE *handle, const WC return STATUS_NOT_IMPLEMENTED; } - if (!(alg = create_algorithm( alg_id, 0, flags ))) return STATUS_NO_MEMORY; + if (!(alg = create_algorithm( alg_id, flags ))) return STATUS_NO_MEMORY; *handle = alg; + TRACE( "returning handle %p\n", *handle ); return STATUS_SUCCESS; } +static void secure_zero( void *ptr, SIZE_T size ) +{ + if (ptr) SecureZeroMemory( ptr, size ); +} + static void destroy_object( struct object *obj ) { - SecureZeroMemory( &obj->magic, sizeof(obj->magic) ); + secure_zero( &obj->magic, sizeof(obj->magic) ); free( obj ); } @@ -440,6 +658,7 @@ NTSTATUS WINAPI BCryptCloseAlgorithmProvider( BCRYPT_ALG_HANDLE handle, DWORD fl TRACE( "%p, %#lx\n", handle, flags ); if (!handle || is_alg_pseudo_handle( handle ) || alg->hdr.magic != MAGIC_ALG) return STATUS_INVALID_HANDLE; + destroy_object( &alg->hdr ); return STATUS_SUCCESS; } @@ -455,43 +674,31 @@ NTSTATUS WINAPI BCryptGetFipsAlgorithmMode(BOOLEAN *enabled) return STATUS_SUCCESS; } -#define BLOCK_LENGTH_RC4 1 -#define BLOCK_LENGTH_3DES 8 -#define BLOCK_LENGTH_AES 16 - -static NTSTATUS generic_alg_property( enum alg_id id, const WCHAR *prop, UCHAR *buf, ULONG size, ULONG *ret_size ) +static NTSTATUS get_generic_alg_property( enum alg_id id, const WCHAR *prop, UCHAR *buf, ULONG size, ULONG *ret_size ) { if (!wcscmp( prop, BCRYPT_OBJECT_LENGTH )) { - if (!builtin_algorithms[id].object_length) - return STATUS_NOT_SUPPORTED; + if (!builtin_algorithms[id].object_length) return STATUS_NOT_SUPPORTED; *ret_size = sizeof(ULONG); - if (size < sizeof(ULONG)) - return STATUS_BUFFER_TOO_SMALL; - if (buf) - *(ULONG *)buf = builtin_algorithms[id].object_length; + if (size < sizeof(ULONG)) return STATUS_BUFFER_TOO_SMALL; + if (buf) *(ULONG *)buf = builtin_algorithms[id].object_length; return STATUS_SUCCESS; } if (!wcscmp( prop, BCRYPT_HASH_LENGTH )) { - if (!builtin_algorithms[id].hash_length) - return STATUS_NOT_SUPPORTED; + if (!builtin_algorithms[id].hash_length) return STATUS_NOT_SUPPORTED; *ret_size = sizeof(ULONG); - if (size < sizeof(ULONG)) - return STATUS_BUFFER_TOO_SMALL; - if(buf) - *(ULONG*)buf = builtin_algorithms[id].hash_length; + if (size < sizeof(ULONG)) return STATUS_BUFFER_TOO_SMALL; + if (buf) *(ULONG*)buf = builtin_algorithms[id].hash_length; return STATUS_SUCCESS; } if (!wcscmp( prop, BCRYPT_ALGORITHM_NAME )) { *ret_size = (lstrlenW(builtin_algorithms[id].name) + 1) * sizeof(WCHAR); - if (size < *ret_size) - return STATUS_BUFFER_TOO_SMALL; - if(buf) - memcpy(buf, builtin_algorithms[id].name, *ret_size); + if (size < *ret_size) return STATUS_BUFFER_TOO_SMALL; + if(buf) memcpy(buf, builtin_algorithms[id].name, *ret_size); return STATUS_SUCCESS; } @@ -512,13 +719,15 @@ static NTSTATUS get_3des_property( enum chain_mode mode, const WCHAR *prop, UCHA const WCHAR *str; switch (mode) { - case CHAIN_MODE_CBC: str = BCRYPT_CHAIN_MODE_CBC; break; - default: return STATUS_NOT_IMPLEMENTED; + case CHAIN_MODE_CBC: + str = BCRYPT_CHAIN_MODE_CBC; break; + default: + return STATUS_NOT_IMPLEMENTED; } - *ret_size = 64; + *ret_size = (wcslen(str) + 1) * sizeof(WCHAR) * 2; /* quirk */ if (size < *ret_size) return STATUS_BUFFER_TOO_SMALL; - memcpy( buf, str, (lstrlenW(str) + 1) * sizeof(WCHAR) ); + memcpy( buf, str, (wcslen(str) + 1) * sizeof(WCHAR) ); return STATUS_SUCCESS; } if (!wcscmp( prop, BCRYPT_KEY_LENGTHS )) @@ -559,9 +768,9 @@ static NTSTATUS get_aes_property( enum chain_mode mode, const WCHAR *prop, UCHAR default: return STATUS_NOT_IMPLEMENTED; } - *ret_size = 64; + *ret_size = (wcslen(str) + 1) * sizeof(WCHAR) * 2; /* quirk */ if (size < *ret_size) return STATUS_BUFFER_TOO_SMALL; - memcpy( buf, str, (lstrlenW(str) + 1) * sizeof(WCHAR) ); + memcpy( buf, str, (wcslen(str) + 1) * sizeof(WCHAR) ); return STATUS_SUCCESS; } if (!wcscmp( prop, BCRYPT_KEY_LENGTHS )) @@ -656,8 +865,7 @@ static NTSTATUS get_alg_property( const struct algorithm *alg, const WCHAR *prop { NTSTATUS status; - status = generic_alg_property( alg->id, prop, buf, size, ret_size ); - if (status != STATUS_NOT_IMPLEMENTED) + if ((status = get_generic_alg_property( alg->id, prop, buf, size, ret_size )) != STATUS_NOT_IMPLEMENTED) return status; switch (alg->id) @@ -678,7 +886,7 @@ static NTSTATUS get_alg_property( const struct algorithm *alg, const WCHAR *prop return get_dsa_property( alg->mode, prop, buf, size, ret_size ); case ALG_ID_PBKDF2: - return get_pbkdf2_property( alg->mode, prop, buf, size, ret_size ); + return get_pbkdf2_property( alg->mode, prop, buf, size, ret_size ); default: break; @@ -695,6 +903,7 @@ static NTSTATUS set_alg_property( struct algorithm *alg, const WCHAR *prop, UCHA case ALG_ID_3DES: if (!wcscmp( prop, BCRYPT_CHAINING_MODE )) { + TRACE( "mode %s\n", debugstr_w((WCHAR *)value) ); if (!wcscmp( (WCHAR *)value, BCRYPT_CHAIN_MODE_CBC )) { alg->mode = CHAIN_MODE_CBC; @@ -712,6 +921,7 @@ static NTSTATUS set_alg_property( struct algorithm *alg, const WCHAR *prop, UCHA case ALG_ID_AES: if (!wcscmp( prop, BCRYPT_CHAINING_MODE )) { + TRACE( "mode %s\n", debugstr_w((WCHAR *)value) ); if (!wcscmp( (WCHAR *)value, BCRYPT_CHAIN_MODE_ECB )) { alg->mode = CHAIN_MODE_ECB; @@ -744,6 +954,7 @@ static NTSTATUS set_alg_property( struct algorithm *alg, const WCHAR *prop, UCHA case ALG_ID_RC4: if (!wcscmp( prop, BCRYPT_CHAINING_MODE )) { + TRACE( "mode %s\n", debugstr_w((WCHAR *)value) ); if (!wcscmp( (WCHAR *)value, BCRYPT_CHAIN_MODE_NA )) return STATUS_SUCCESS; FIXME( "unsupported mode %s\n", debugstr_w((WCHAR *)value) ); @@ -790,28 +1001,52 @@ static NTSTATUS set_alg_property( struct algorithm *alg, const WCHAR *prop, UCHA } } +static NTSTATUS set_dh_parameters( struct key *key, const UCHAR *buf, ULONG size ) +{ + BCRYPT_DH_PARAMETER_HEADER *hdr = (BCRYPT_DH_PARAMETER_HEADER *)buf; + UCHAR *prime, *generator; + ULONG key_size = len_from_bitlen( key->a.bitlen ); + + if (key->alg_id != ALG_ID_DH || (key->flags & KEY_FLAG_FINALIZED)) return STATUS_INVALID_HANDLE; + + if (size < sizeof(*hdr) || hdr->dwMagic != BCRYPT_DH_PARAMETERS_MAGIC || hdr->cbKeyLength != key_size || + hdr->cbLength != sizeof(*hdr) + key_size * 2 || size != hdr->cbLength) return STATUS_INVALID_PARAMETER; + + if (!key->a.dsa.group && !(key->a.dsa.group = SymCryptDlgroupAllocate( key_size * 8, 0 ))) + return STATUS_NO_MEMORY; + + prime = (UCHAR *)(hdr + 1); + generator = prime + key_size; + + if (SymCryptDlgroupSetValue( prime, key_size, NULL, 0, generator, key_size, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, + NULL, NULL, 0, key->a.dsa.count, SYMCRYPT_DLGROUP_FIPS_NONE, key->a.dsa.group )) + return STATUS_INTERNAL_ERROR; + + return STATUS_SUCCESS; +} + static NTSTATUS set_key_property( struct key *key, const WCHAR *prop, UCHAR *value, ULONG size, ULONG flags ) { if (!wcscmp( prop, BCRYPT_CHAINING_MODE )) { if (!wcscmp( (WCHAR *)value, BCRYPT_CHAIN_MODE_ECB )) { - key->u.s.mode = CHAIN_MODE_ECB; + key->s.mode = CHAIN_MODE_ECB; return STATUS_SUCCESS; } else if (!wcscmp( (WCHAR *)value, BCRYPT_CHAIN_MODE_CBC )) { - key->u.s.mode = CHAIN_MODE_CBC; + key->s.mode = CHAIN_MODE_CBC; return STATUS_SUCCESS; } else if (!wcscmp( (WCHAR *)value, BCRYPT_CHAIN_MODE_GCM )) { - key->u.s.mode = CHAIN_MODE_GCM; + key->s.mode = CHAIN_MODE_GCM; return STATUS_SUCCESS; } else if (!wcscmp( (WCHAR *)value, BCRYPT_CHAIN_MODE_CFB )) { - key->u.s.mode = CHAIN_MODE_CFB; + key->s.mode = CHAIN_MODE_CFB; return STATUS_SUCCESS; } else @@ -823,53 +1058,78 @@ static NTSTATUS set_key_property( struct key *key, const WCHAR *prop, UCHAR *val else if (!wcscmp( prop, BCRYPT_KEY_LENGTH )) { if (size < sizeof(DWORD)) return STATUS_INVALID_PARAMETER; - key->u.a.bitlen = *(DWORD*)value; + key->a.bitlen = *(DWORD*)value; return STATUS_SUCCESS; } else if (!wcscmp( prop, BCRYPT_DH_PARAMETERS )) { - BCRYPT_DH_PARAMETER_HEADER *hdr = (BCRYPT_DH_PARAMETER_HEADER *)value; - struct key_asymmetric_import_params params; - - if (key->u.a.flags & KEY_FLAG_FINALIZED) return STATUS_INVALID_HANDLE; - if (key->alg_id != ALG_ID_DH || size < sizeof(*hdr) || hdr->cbLength != size || - hdr->dwMagic != BCRYPT_DH_PARAMETERS_MAGIC || hdr->cbKeyLength != len_from_bitlen( key->u.a.bitlen )) - return STATUS_INVALID_PARAMETER; - - params.key = key; - params.flags = KEY_IMPORT_FLAG_DH_PARAMETERS; - params.buf = value; - params.len = size; - return UNIX_CALL( key_asymmetric_import, ¶ms ); + return set_dh_parameters( key, value, size ); } FIXME( "unsupported key property %s\n", debugstr_w(prop) ); return STATUS_NOT_IMPLEMENTED; } +NTSTATUS WINAPI BCryptSetProperty( BCRYPT_HANDLE handle, const WCHAR *prop, UCHAR *value, ULONG size, ULONG flags ) +{ + struct object *object = get_object( handle, 0 ); + + TRACE( "%p, %s, %p, %lu, %#lx\n", handle, debugstr_w(prop), value, size, flags ); + + if (!handle) return STATUS_INVALID_HANDLE; + if (is_alg_pseudo_handle( handle )) return STATUS_ACCESS_DENIED; + + switch (object->magic) + { + case MAGIC_ALG: + { + struct algorithm *alg = (struct algorithm *)object; + return set_alg_property( alg, prop, value, size, flags ); + } + case MAGIC_KEY: + { + struct key *key = (struct key *)object; + return set_key_property( key, prop, value, size, flags ); + } + default: + WARN( "unknown magic %#lx\n", object->magic ); + return STATUS_INVALID_HANDLE; + } +} + static NTSTATUS get_hash_property( const struct hash *hash, const WCHAR *prop, UCHAR *buf, ULONG size, ULONG *ret_size ) { NTSTATUS status; - status = generic_alg_property( hash->alg_id, prop, buf, size, ret_size ); - if (status == STATUS_NOT_IMPLEMENTED) + if ((status = get_generic_alg_property( hash->alg_id, prop, buf, size, ret_size )) == STATUS_NOT_IMPLEMENTED) FIXME( "unsupported property %s\n", debugstr_w(prop) ); return status; } -static NTSTATUS get_dh_property( const struct key *key, const WCHAR *prop, UCHAR *buf, ULONG size, ULONG *ret_size ) +static NTSTATUS get_dh_parameters( const struct key *key, UCHAR *buf, ULONG size, ULONG *ret_size ) { - struct key_asymmetric_export_params params; + BCRYPT_DH_PARAMETER_HEADER *hdr = (BCRYPT_DH_PARAMETER_HEADER *)buf; + UCHAR *prime, *generator; + UINT32 key_size = len_from_bitlen( key->a.bitlen ), required_size = sizeof(*hdr) + key_size * 2; + + if (!(key->flags & KEY_FLAG_FINALIZED)) return STATUS_INVALID_HANDLE; - if (wcscmp( prop, BCRYPT_DH_PARAMETERS )) return STATUS_NOT_SUPPORTED; - if (!(key->u.a.flags & KEY_FLAG_FINALIZED)) return STATUS_INVALID_HANDLE; + *ret_size = required_size; + if (!buf) return STATUS_SUCCESS; + else if (size < required_size) return STATUS_BUFFER_TOO_SMALL; - params.key = (struct key *)key; - params.flags = KEY_EXPORT_FLAG_DH_PARAMETERS; - params.buf = buf; - params.len = size; - params.ret_len = ret_size; - return UNIX_CALL( key_asymmetric_export, ¶ms ); + prime = (UCHAR *)(hdr + 1); + generator = prime + key_size; + + if (SymCryptDlgroupGetValue( key->a.dsa.group, prime, key_size, NULL, 0, generator, key_size, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, NULL, (BYTE *)key->a.dsa.seed, + sizeof(key->a.dsa.seed), (UINT32 *)&key->a.dsa.count )) return STATUS_INTERNAL_ERROR; + + hdr->cbLength = sizeof(*hdr) + key_size * 2; + hdr->dwMagic = BCRYPT_DH_PARAMETERS_MAGIC; + hdr->cbKeyLength = key_size; + + return STATUS_SUCCESS; } static NTSTATUS get_key_property( const struct key *key, const WCHAR *prop, UCHAR *buf, ULONG size, ULONG *ret_size ) @@ -880,8 +1140,8 @@ static NTSTATUS get_key_property( const struct key *key, const WCHAR *prop, UCHA if (size < sizeof(DWORD)) return STATUS_BUFFER_TOO_SMALL; if (buf) { - if (is_symmetric_key(key)) *(DWORD *)buf = key->u.s.block_size * 8; - else *(DWORD *)buf = key->u.a.bitlen; + if (is_symmetric_key(key)) *(DWORD *)buf = key->s.block_size * 8; + else *(DWORD *)buf = key->a.bitlen; } return STATUS_SUCCESS; } @@ -889,14 +1149,15 @@ static NTSTATUS get_key_property( const struct key *key, const WCHAR *prop, UCHA switch (key->alg_id) { case ALG_ID_3DES: - return get_3des_property( key->u.s.mode, prop, buf, size, ret_size ); + return get_3des_property( key->s.mode, prop, buf, size, ret_size ); case ALG_ID_AES: if (!wcscmp( prop, BCRYPT_AUTH_TAG_LENGTH )) return STATUS_NOT_SUPPORTED; - return get_aes_property( key->u.s.mode, prop, buf, size, ret_size ); + return get_aes_property( key->s.mode, prop, buf, size, ret_size ); case ALG_ID_DH: - return get_dh_property( key, prop, buf, size, ret_size ); + if (wcscmp( prop, BCRYPT_DH_PARAMETERS )) return STATUS_NOT_SUPPORTED; + return get_dh_parameters( key, buf, size, ret_size ); default: FIXME( "unsupported algorithm %u\n", key->alg_id ); @@ -907,7 +1168,7 @@ static NTSTATUS get_key_property( const struct key *key, const WCHAR *prop, UCHA NTSTATUS WINAPI BCryptGetProperty( BCRYPT_HANDLE handle, const WCHAR *prop, UCHAR *buffer, ULONG count, ULONG *res, ULONG flags ) { - struct object *object = get_object( handle, 0 ); + const struct object *object = get_object( handle, 0 ); TRACE( "%p, %s, %p, %lu, %p, %#lx\n", handle, wine_dbgstr_w(prop), buffer, count, res, flags ); @@ -1145,270 +1406,434 @@ NTSTATUS WINAPI BCryptHash( BCRYPT_ALG_HANDLE handle, UCHAR *secret, ULONG secre return hash_single( alg, secret, secret_len, input, input_len, output ); } -static ULONG curve_strength( enum ecc_curve_id curve_id ) +static void destroy_key( struct key *key ) { - switch (curve_id) + if (!key) return; + if (is_symmetric_key( key )) { - case ECC_CURVE_25519: return 253; - case ECC_CURVE_P256R1: return 256; - case ECC_CURVE_P384R1: return 384; - case ECC_CURVE_P521R1: return 521; - default: - FIXME( "unsupported curve %u\n", curve_id ); - return 0; + secure_zero( key->s.secret, key->s.secret_len ); + free( key->s.secret ); + secure_zero( key->s.vector, key->s.block_size ); + DeleteCriticalSection( &key->s.cs ); + } + else + { + switch (key->alg_id) + { + case ALG_ID_RSA: + case ALG_ID_RSA_SIGN: + if (key->a.rsa.handle) SymCryptRsakeyFree( key->a.rsa.handle ); + break; + case ALG_ID_ECDH: + case ALG_ID_ECDH_P256: + case ALG_ID_ECDH_P384: + case ALG_ID_ECDH_P521: + case ALG_ID_ECDSA: + case ALG_ID_ECDSA_P256: + case ALG_ID_ECDSA_P384: + case ALG_ID_ECDSA_P521: + if (key->a.ecc.handle) SymCryptEckeyFree( key->a.ecc.handle ); + if (key->a.ecc.curve) SymCryptEcurveFree( key->a.ecc.curve ); + break; + case ALG_ID_DH: + case ALG_ID_DSA: + if (key->a.dsa.handle) SymCryptDlkeyFree( key->a.dsa.handle ); + if (key->a.dsa.group) SymCryptDlgroupFree( key->a.dsa.group ); + break; + default: + FIXME( "algorithm %u not handled\n", key->alg_id ); + break; + } } + + secure_zero( &key->hdr.magic, sizeof(key->hdr.magic) ); + _aligned_free( key ); } -static NTSTATUS key_asymmetric_create( enum alg_id alg_id, enum ecc_curve_id curve_id, ULONG bitlen, struct key **ret_key ) +static NTSTATUS alloc_aes_key( struct key *key, enum chain_mode mode, ULONG block_size, const UCHAR *secret, + ULONG secret_len ) { - struct key *key; + key->s.mode = mode; + key->s.block_size = block_size; + + if (!(key->s.secret = malloc( secret_len ))) return STATUS_NO_MEMORY; + memcpy( key->s.secret, secret, secret_len ); + key->s.secret_len = secret_len; + + InitializeCriticalSection( &key->s.cs ); - if (!__wine_unixlib_handle) + switch (mode) { - ERR( "no encryption support\n" ); - return STATUS_NOT_IMPLEMENTED; - } + case CHAIN_MODE_ECB: + case CHAIN_MODE_CBC: + case CHAIN_MODE_CFB: + if (SymCryptAesExpandKey( &key->s.aes.handle, secret, secret_len )) return STATUS_INTERNAL_ERROR; + break; - if (alg_id == ALG_ID_DH && bitlen < 512) return STATUS_INVALID_PARAMETER; + case CHAIN_MODE_GCM: + if (SymCryptGcmExpandKey( &key->s.gcm.handle, SymCryptAesBlockCipher, secret, secret_len )) + return STATUS_INTERNAL_ERROR; + break; - if (!(key = calloc( 1, sizeof(*key) ))) return STATUS_NO_MEMORY; - key->hdr.magic = MAGIC_KEY; - key->alg_id = alg_id; - key->u.a.bitlen = curve_id ? curve_strength( curve_id ) : bitlen; - key->u.a.curve_id = curve_id; + default: + FIXME( "unhandled mode %u\n", mode ); + return STATUS_NOT_SUPPORTED; + } - *ret_key = key; return STATUS_SUCCESS; } -static BOOL is_equal_vector( const UCHAR *vector, ULONG len, const UCHAR *vector2, ULONG len2 ) +static NTSTATUS alloc_key( enum alg_id alg_id, ULONG flags, struct key **ret_key ) { - if (!vector && !vector2) return TRUE; - if (len != len2) return FALSE; - return !memcmp( vector, vector2, len ); -} + struct key *key; -static NTSTATUS key_symmetric_set_vector( struct key *key, UCHAR *vector, ULONG vector_len, BOOL force_reset ) -{ - BOOL needs_reset = force_reset || !is_equal_vector( key->u.s.vector, key->u.s.vector_len, vector, vector_len ); - if (vector) - { - free( key->u.s.vector ); - key->u.s.vector = NULL; - key->u.s.vector_len = 0; + if (!(key = _aligned_malloc( sizeof(struct key), TYPE_ALIGNMENT(struct key) ))) return STATUS_NO_MEMORY; + memset( key, 0, sizeof(*key) ); + key->hdr.magic = MAGIC_KEY; + key->alg_id = alg_id; + key->flags = flags; - if (!(key->u.s.vector = malloc( vector_len ))) return STATUS_NO_MEMORY; - memcpy( key->u.s.vector, vector, vector_len ); - key->u.s.vector_len = vector_len; - } - if (needs_reset) UNIX_CALL( key_symmetric_vector_reset, key ); + *ret_key = key; return STATUS_SUCCESS; } -static struct key *key_symmetric_create( enum alg_id alg, enum chain_mode mode, ULONG block_size, const UCHAR *secret, - ULONG secret_len ) +static NTSTATUS validate_len_aes( const BCRYPT_KEY_LENGTHS_STRUCT *key_lengths, ULONG secret_len, ULONG *ret_len ) { - struct key *ret; - - if (!(ret = calloc( 1, sizeof(*ret) ))) return NULL; - InitializeCriticalSection( &ret->u.s.cs ); - ret->hdr.magic = MAGIC_KEY; - ret->alg_id = alg; - ret->u.s.mode = mode; - ret->u.s.block_size = block_size; + ULONG size; - if (!(ret->u.s.secret = malloc( secret_len ))) + if (secret_len > (size = key_lengths->dwMaxLength / 8)) { - DeleteCriticalSection( &ret->u.s.cs ); - free( ret ); - return NULL; + WARN( "secret_len %lu exceeds key max length %lu, setting to maximum\n", secret_len, size ); + *ret_len = size; + return STATUS_SUCCESS; } - memcpy( ret->u.s.secret, secret, secret_len ); - ret->u.s.secret_len = secret_len; - - return ret; -} - -static void key_destroy( struct key *key ) -{ - if (is_symmetric_key( key )) + else if (secret_len < (size = key_lengths->dwMinLength / 8)) { - UNIX_CALL( key_symmetric_destroy, key ); - free( key->u.s.vector ); - free( key->u.s.secret ); - DeleteCriticalSection( &key->u.s.cs ); + WARN( "secret_len %lu is less than minimum key length %lu\n", secret_len, size ); + return STATUS_INVALID_PARAMETER; + } + else if (key_lengths->dwIncrement && (secret_len * 8 - key_lengths->dwMinLength) % key_lengths->dwIncrement) + { + WARN( "secret_len %lu is not a valid key length\n", secret_len ); + return STATUS_INVALID_PARAMETER; } - else - UNIX_CALL( key_asymmetric_destroy, key ); - - destroy_object( &key->hdr ); -} -static ULONG get_block_size( struct algorithm *alg ) -{ - ULONG ret = 0, size = sizeof(ret); - get_alg_property( alg, BCRYPT_BLOCK_LENGTH, (UCHAR *)&ret, sizeof(ret), &size ); - return ret; + *ret_len = secret_len; + return STATUS_SUCCESS; } -static NTSTATUS key_symmetric_generate( struct algorithm *alg, BCRYPT_KEY_HANDLE *ret_handle, const UCHAR *secret, - ULONG secret_len ) +static NTSTATUS generate_symmetric_key( const struct algorithm *alg, const UCHAR *secret, ULONG secret_len, + struct key **ret_key ) { BCRYPT_KEY_LENGTHS_STRUCT key_lengths; - ULONG block_size, size; struct key *key; NTSTATUS status; + ULONG size; - if (alg->id == ALG_ID_PBKDF2 && - !get_alg_property( alg, BCRYPT_KEY_LENGTHS, (UCHAR *)&key_lengths, sizeof(key_lengths), &size )) - { - if (secret_len > key_lengths.dwMaxLength / 8 || secret_len < key_lengths.dwMinLength / 8) - return STATUS_INVALID_PARAMETER; - block_size = secret_len; - } - else if (!(block_size = get_block_size( alg ))) return STATUS_INVALID_PARAMETER; - else if (!get_alg_property( alg, BCRYPT_KEY_LENGTHS, (UCHAR *)&key_lengths, sizeof(key_lengths), &size )) + if ((status = get_alg_property( alg, BCRYPT_KEY_LENGTHS, (UCHAR *)&key_lengths, sizeof(key_lengths), &size ))) + return status; + + if ((status = alloc_key( alg->id, 0, &key ))) return status; + + switch (alg->id) { - if (secret_len > (size = key_lengths.dwMaxLength / 8)) + case ALG_ID_AES: + if ((status = validate_len_aes( &key_lengths, secret_len, &secret_len )) || + (status = alloc_aes_key( key, alg->mode, BLOCK_LENGTH_AES, secret, secret_len ))) { - WARN( "secret_len %lu exceeds key max length %lu, setting to maximum\n", secret_len, size ); - secret_len = size; + destroy_key( key ); + return status; } - else if (secret_len < (size = key_lengths.dwMinLength / 8)) + break; + + case ALG_ID_PBKDF2: + if (secret_len > key_lengths.dwMaxLength / 8 || secret_len < key_lengths.dwMinLength / 8) { - WARN( "secret_len %lu is less than minimum key length %lu\n", secret_len, size ); + destroy_key( key ); return STATUS_INVALID_PARAMETER; } - else if (key_lengths.dwIncrement && (secret_len * 8 - key_lengths.dwMinLength) % key_lengths.dwIncrement) + if (!(key->s.secret = malloc( secret_len ))) { - WARN( "secret_len %lu is not a valid key length\n", secret_len ); - return STATUS_INVALID_PARAMETER; + destroy_key( key ); + return STATUS_NO_MEMORY; } - } + memcpy( key->s.secret, secret, secret_len ); + key->s.block_size = key->s.secret_len = secret_len; + break; - if (!(key = key_symmetric_create( alg->id, alg->mode, block_size, secret, secret_len ))) status = STATUS_NO_MEMORY; - else - { - *ret_handle = key; - status = STATUS_SUCCESS; + default: + FIXME( "unhandled algorithm %u\n", alg->id ); + destroy_key( key ); + return STATUS_NOT_SUPPORTED; } + *ret_key = key; return status; } -static NTSTATUS key_symmetric_decrypt( struct key *key, UCHAR *input, ULONG input_len, void *padding, UCHAR *iv, - ULONG iv_len, UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +NTSTATUS WINAPI BCryptGenerateSymmetricKey( BCRYPT_ALG_HANDLE handle, BCRYPT_KEY_HANDLE *ret_handle, UCHAR *object, + ULONG object_len, UCHAR *secret, ULONG secret_len, ULONG flags ) { - struct key_symmetric_set_auth_data_params auth_params; - struct key_symmetric_decrypt_params decrypt_params; - struct key_symmetric_get_tag_params tag_params; - ULONG bytes_left = input_len; - NTSTATUS status; + const struct algorithm *alg = get_alg_object( handle ); + struct key *key; + NTSTATUS status; - if (key->u.s.mode == CHAIN_MODE_GCM) - { - BCRYPT_AUTHENTICATED_CIPHER_MODE_INFO *auth_info = padding; - UCHAR tag[16]; + TRACE( "%p, %p, %p, %lu, %p, %lu, %#lx\n", handle, ret_handle, object, object_len, secret, secret_len, flags ); + if (object) FIXME( "ignoring object buffer\n" ); - if (!auth_info) return STATUS_INVALID_PARAMETER; - if (!auth_info->pbNonce) return STATUS_INVALID_PARAMETER; - if (!auth_info->pbTag) return STATUS_INVALID_PARAMETER; - if (auth_info->cbTag < 12 || auth_info->cbTag > 16) return STATUS_INVALID_PARAMETER; + if (!alg) return STATUS_INVALID_HANDLE; - if ((status = key_symmetric_set_vector( key, auth_info->pbNonce, auth_info->cbNonce, TRUE ))) - return status; + if ((status = generate_symmetric_key( alg, secret, secret_len, &key ))) return status; + *ret_handle = key; + + TRACE( "returning handle %p\n", *ret_handle ); + return STATUS_SUCCESS; +} - *ret_len = input_len; - if (flags & BCRYPT_BLOCK_PADDING) return STATUS_INVALID_PARAMETER; - if (!output) return STATUS_SUCCESS; - if (output_len < *ret_len) return STATUS_BUFFER_TOO_SMALL; - - auth_params.key = key; - auth_params.auth_data = auth_info->pbAuthData; - auth_params.len = auth_info->cbAuthData; - if ((status = UNIX_CALL( key_symmetric_set_auth_data, &auth_params ))) return status; - - decrypt_params.key = key; - decrypt_params.input = input; - decrypt_params.input_len = input_len; - decrypt_params.output = output; - decrypt_params.output_len = output_len; - if ((status = UNIX_CALL( key_symmetric_decrypt, &decrypt_params ))) return status; - - tag_params.key = key; - tag_params.tag = tag; - tag_params.len = sizeof(tag); - if ((status = UNIX_CALL( key_symmetric_get_tag, &tag_params ))) return status; - if (memcmp( tag, auth_info->pbTag, auth_info->cbTag )) return STATUS_AUTH_TAG_MISMATCH; +static NTSTATUS decrypt_aes_ecb( const struct key *key, const UCHAR *input, ULONG input_len, UCHAR *output, + ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + ULONG offset = input_len, block_size = key->s.block_size; + UCHAR buf[BLOCK_LENGTH_AES]; + NTSTATUS status = STATUS_SUCCESS; - return STATUS_SUCCESS; + if (input_len & (block_size - 1)) return STATUS_INVALID_BUFFER_SIZE; + if (!output) return STATUS_SUCCESS; + + if (flags & BCRYPT_BLOCK_PADDING) + { + if (output_len + block_size < *ret_len || input_len < block_size) return STATUS_BUFFER_TOO_SMALL; + offset -= block_size; } + else if (output_len < *ret_len) return STATUS_BUFFER_TOO_SMALL; - *ret_len = input_len; + SymCryptEcbDecrypt( SymCryptAesBlockCipher, &key->s.aes.handle, input, output, offset ); + + if (flags & BCRYPT_BLOCK_PADDING) + { + SymCryptEcbDecrypt( SymCryptAesBlockCipher, &key->s.aes.handle, input + offset, buf, block_size ); + + if (buf[block_size - 1] <= block_size) + { + *ret_len -= buf[block_size - 1]; + if (output_len < *ret_len) status = STATUS_BUFFER_TOO_SMALL; + else memcpy( output + offset, buf, block_size - buf[block_size - 1] ); + } + else status = STATUS_UNSUCCESSFUL; /* FIXME: invalid padding */ + } + + return status; +} + +static NTSTATUS decrypt_aes_gcm( struct key *key, const UCHAR *input, ULONG input_len, + const BCRYPT_AUTHENTICATED_CIPHER_MODE_INFO *info, UCHAR *output ) +{ + SYMCRYPT_ERROR error; + + if (!info || !info->pbNonce || !info->pbTag || info->cbTag < 12 || info->cbTag > 16) + return STATUS_INVALID_PARAMETER; + if (info->dwFlags & BCRYPT_AUTH_MODE_CHAIN_CALLS_FLAG) FIXME( "call chaining not implemented\n" ); + + EnterCriticalSection( &key->s.cs ); + + SymCryptGcmInit( &key->s.gcm.state, &key->s.gcm.handle, info->pbNonce, info->cbNonce ); + SymCryptGcmAuthPart( &key->s.gcm.state, info->pbAuthData, info->cbAuthData ); + SymCryptGcmDecryptPart( &key->s.gcm.state, input, output, input_len ); + error = SymCryptGcmDecryptFinal( &key->s.gcm.state, info->pbTag, info->cbTag ); - if (input_len & (key->u.s.block_size - 1)) return STATUS_INVALID_BUFFER_SIZE; + LeaveCriticalSection( &key->s.cs ); + + if (error == SYMCRYPT_AUTHENTICATION_FAILURE) return STATUS_AUTH_TAG_MISMATCH; + if (error) return STATUS_INTERNAL_ERROR; + return STATUS_SUCCESS; +} + +static NTSTATUS decrypt_aes_vector( struct key *key, const UCHAR *input, ULONG input_len, UCHAR *iv, UCHAR *output, + ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + UCHAR buf[BLOCK_LENGTH_AES], *out; + const UCHAR *in; + ULONG bytes_left = input_len, block_size = key->s.block_size; + NTSTATUS status = STATUS_SUCCESS; + + if (input_len & (block_size - 1)) return STATUS_INVALID_BUFFER_SIZE; if (!output) return STATUS_SUCCESS; + if (flags & BCRYPT_BLOCK_PADDING) { - if (output_len + key->u.s.block_size < *ret_len) return STATUS_BUFFER_TOO_SMALL; - if (input_len < key->u.s.block_size) return STATUS_BUFFER_TOO_SMALL; - bytes_left -= key->u.s.block_size; + if (output_len + block_size < *ret_len || input_len < block_size) return STATUS_BUFFER_TOO_SMALL; + bytes_left -= block_size; } else if (output_len < *ret_len) return STATUS_BUFFER_TOO_SMALL; - if (key->u.s.mode == CHAIN_MODE_ECB && iv) return STATUS_INVALID_PARAMETER; - if ((status = key_symmetric_set_vector( key, iv, iv_len, flags & BCRYPT_BLOCK_PADDING ))) return status; + EnterCriticalSection( &key->s.cs ); + + if (iv) memcpy( key->s.vector, iv, block_size ); - decrypt_params.key = key; - decrypt_params.input = input; - decrypt_params.input_len = key->u.s.block_size; - decrypt_params.output = output; - decrypt_params.output_len = key->u.s.block_size; - while (bytes_left >= key->u.s.block_size) + in = input; + out = output; + while (bytes_left >= block_size) { - if ((status = UNIX_CALL( key_symmetric_decrypt, &decrypt_params ))) return status; - if (key->u.s.mode == CHAIN_MODE_ECB && (status = key_symmetric_set_vector( key, NULL, 0, TRUE ))) - return status; - bytes_left -= key->u.s.block_size; - decrypt_params.input += key->u.s.block_size; - decrypt_params.output += key->u.s.block_size; + if (key->s.mode == CHAIN_MODE_CFB) + SymCryptCfbDecrypt( SymCryptAesBlockCipher, 1, &key->s.aes.handle, key->s.vector, in, out, block_size ); + else + SymCryptCbcDecrypt( SymCryptAesBlockCipher, &key->s.aes.handle, key->s.vector, in, out, block_size ); + + bytes_left -= block_size; + in += block_size; + out += block_size; } if (flags & BCRYPT_BLOCK_PADDING) { - UCHAR *buf, *dst = decrypt_params.output; - if (!(buf = malloc( key->u.s.block_size ))) return STATUS_NO_MEMORY; - decrypt_params.output = buf; - status = UNIX_CALL( key_symmetric_decrypt, &decrypt_params ); - if (!status && buf[ key->u.s.block_size - 1 ] <= key->u.s.block_size) + if (key->s.mode == CHAIN_MODE_CFB) + SymCryptCfbDecrypt( SymCryptAesBlockCipher, 1, &key->s.aes.handle, key->s.vector, in, buf, block_size ); + else + SymCryptCbcDecrypt( SymCryptAesBlockCipher, &key->s.aes.handle, key->s.vector, in, buf, block_size ); + + if (buf[block_size - 1] <= block_size) { - *ret_len -= buf[ key->u.s.block_size - 1 ]; + *ret_len -= buf[block_size - 1]; if (output_len < *ret_len) status = STATUS_BUFFER_TOO_SMALL; - else memcpy( dst, buf, key->u.s.block_size - buf[ key->u.s.block_size - 1 ] ); + else memcpy( out, buf, block_size - buf[block_size - 1] ); } else status = STATUS_UNSUCCESSFUL; /* FIXME: invalid padding */ - free( buf ); } - if (!status) + if (iv) memcpy( iv, key->s.vector, block_size ); + + LeaveCriticalSection( &key->s.cs ); + return status; +} + +static NTSTATUS decrypt_symmetric( struct key *key, const UCHAR *input, ULONG input_len, + const BCRYPT_AUTHENTICATED_CIPHER_MODE_INFO *info, UCHAR *iv, ULONG iv_len, + UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + *ret_len = input_len; + + if (key->s.mode == CHAIN_MODE_GCM && (flags & BCRYPT_BLOCK_PADDING)) return STATUS_INVALID_PARAMETER; + if (key->s.mode == CHAIN_MODE_ECB && iv) return STATUS_INVALID_PARAMETER; + + switch (key->alg_id) + { + case ALG_ID_AES: + switch (key->s.mode) + { + case CHAIN_MODE_ECB: + return decrypt_aes_ecb( key, input, input_len, output, output_len, ret_len, flags ); + + case CHAIN_MODE_GCM: + return decrypt_aes_gcm( key, input, input_len, info, output ); + + default: + if (iv && iv_len != key->s.block_size) return STATUS_INVALID_PARAMETER; + return decrypt_aes_vector( key, input, input_len, iv, output, output_len, ret_len, flags ); + } + default: + FIXME( "unhandled algorithm %u\n", key->alg_id ); + break; + } + + return STATUS_NOT_SUPPORTED; +} + +static const SYMCRYPT_HASH *get_hash_from_str( const WCHAR *str ) +{ + if (!str) return NULL; + if (!wcscmp( str, BCRYPT_MD5_ALGORITHM )) return SymCryptMd5Algorithm; + if (!wcscmp( str, BCRYPT_SHA1_ALGORITHM )) return SymCryptSha1Algorithm; + if (!wcscmp( str, BCRYPT_SHA256_ALGORITHM )) return SymCryptSha256Algorithm; + if (!wcscmp( str, BCRYPT_SHA384_ALGORITHM )) return SymCryptSha384Algorithm; + if (!wcscmp( str, BCRYPT_SHA512_ALGORITHM )) return SymCryptSha512Algorithm; + + FIXME( "unhandled algorithm %s\n", debugstr_w(str) ); + return NULL; +} + +static NTSTATUS decrypt_rsa( const struct key *key, const UCHAR *input, ULONG input_len, const void *padding, + UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + SIZE_T size = len_from_bitlen( key->a.bitlen ); + + if (!flags || flags == BCRYPT_PAD_NONE) + { + if (input_len % size) return STATUS_INVALID_PARAMETER; + + *ret_len = size; + if (output_len < size) + { + if (output) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; + } + if (SymCryptRsaRawDecrypt( key->a.rsa.handle, input, input_len, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0, output, + output_len )) return STATUS_INTERNAL_ERROR; + } + else if (flags == BCRYPT_PAD_PKCS1) + { + if (SymCryptRsaPkcs1Decrypt( key->a.rsa.handle, input, input_len, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0, output, + output_len, &size )) return STATUS_INTERNAL_ERROR; + *ret_len = size; + } + else if (flags == BCRYPT_PAD_OAEP) + { + const BCRYPT_OAEP_PADDING_INFO *pad = padding; + const SYMCRYPT_HASH *hash_desc; + + if (!pad) return STATUS_INVALID_PARAMETER; + hash_desc = get_hash_from_str( pad->pszAlgId ); + + if (SymCryptRsaOaepDecrypt( key->a.rsa.handle, input, input_len, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, hash_desc, + pad->pbLabel, pad->cbLabel, 0, output, output_len, &size )) + return STATUS_INTERNAL_ERROR; + *ret_len = size; + } + else + { + FIXME( "unsupported flags %#lx", flags ); + return STATUS_NOT_SUPPORTED; + } + + return STATUS_SUCCESS; +} + +NTSTATUS WINAPI BCryptDecrypt( BCRYPT_KEY_HANDLE handle, UCHAR *input, ULONG input_len, void *padding, UCHAR *iv, + ULONG iv_len, UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + struct key *key = get_key_object( handle ); + NTSTATUS status; + + TRACE( "%p, %p, %lu, %p, %p, %lu, %p, %lu, %p, %#lx\n", handle, input, input_len, padding, iv, iv_len, output, + output_len, ret_len, flags ); + + if (!key) return STATUS_INVALID_HANDLE; + + if (is_symmetric_key( key )) { - if (key->u.s.vector && input_len >= key->u.s.vector_len) + if (flags & ~BCRYPT_BLOCK_PADDING) { - memcpy( key->u.s.vector, input + input_len - key->u.s.vector_len, key->u.s.vector_len ); - if (iv) memcpy( iv, key->u.s.vector, min( iv_len, key->u.s.vector_len )); + FIXME( "flags %#lx not supported\n", flags ); + return STATUS_NOT_IMPLEMENTED; } - else if (key->u.s.vector) - FIXME( "Unexpected vector len %lu, *ret_len %lu.\n", key->u.s.vector_len, *ret_len ); + + status = decrypt_symmetric( key, input, input_len, padding, iv, iv_len, output, output_len, ret_len, flags ); } + else + { + if (key->alg_id != ALG_ID_RSA) return STATUS_NOT_SUPPORTED; + if (!(key->flags & KEY_FLAG_FINALIZED)) return STATUS_INVALID_HANDLE; + status = decrypt_rsa( key, input, input_len, padding, output, output_len, ret_len, flags ); + } return status; } /* AES Key Wrap Algorithm (RFC3394) */ -static NTSTATUS aes_unwrap( const UCHAR *secret, ULONG secret_len, const UCHAR *cipher, ULONG cipher_len, UCHAR *plain ) +static NTSTATUS unwrap_aes( const UCHAR *secret, ULONG secret_len, const UCHAR *cipher, ULONG cipher_len, UCHAR *plain ) { UCHAR a[8], *r, b[16]; - ULONG len, t, i, n = cipher_len / 8; + ULONG len = 16, t, i, n = cipher_len / 8; + NTSTATUS status; int j; struct key *key; @@ -1416,7 +1841,12 @@ static NTSTATUS aes_unwrap( const UCHAR *secret, ULONG secret_len, const UCHAR * r = plain; memcpy( r, cipher + 8, 8 * n ); - if (!(key = key_symmetric_create( ALG_ID_AES, CHAIN_MODE_ECB, 16, secret, secret_len ))) return STATUS_NO_MEMORY; + if ((status = alloc_key( ALG_ID_AES, 0, &key ))) return status; + if ((status = alloc_aes_key( key, CHAIN_MODE_ECB, BLOCK_LENGTH_AES, secret, secret_len ))) + { + destroy_key( key ); + return status; + } for (j = 5; j >= 0; j--) { @@ -1431,21 +1861,21 @@ static NTSTATUS aes_unwrap( const UCHAR *secret, ULONG secret_len, const UCHAR * b[4] ^= t >> 24; memcpy( b + 8, r, 8 ); - key_symmetric_decrypt( key, b, 16, NULL, NULL, 0, b, 16, &len, 0 ); + decrypt_aes_ecb( key, b, 16, b, 16, &len, 0 ); memcpy( a, b, 8 ); memcpy( r, b + 8, 8 ); r -= 8; } } - key_destroy( key ); + destroy_key( key ); for (i = 0; i < 8; i++) if (a[i] != 0xa6) return STATUS_UNSUCCESSFUL; return STATUS_SUCCESS; } -static NTSTATUS key_import( struct algorithm *alg, struct key *decrypt_key, const WCHAR *type, BCRYPT_KEY_HANDLE *key, - UCHAR *object, ULONG object_len, UCHAR *input, ULONG input_len ) +static NTSTATUS import_key( const struct algorithm *alg, const struct key *decrypt_key, const WCHAR *type, + const UCHAR *input, ULONG input_len, struct key **ret_key ) { ULONG len; NTSTATUS status; @@ -1470,7 +1900,7 @@ static NTSTATUS key_import( struct algorithm *alg, struct key *decrypt_key, cons len = header->cbKeyData; if (len + sizeof(BCRYPT_KEY_DATA_BLOB_HEADER) > input_len) return STATUS_INVALID_PARAMETER; - return key_symmetric_generate( alg, key, (UCHAR *)&header[1], len ); + return generate_symmetric_key( alg, (UCHAR *)&header[1], len, ret_key ); } else if (!wcscmp( type, BCRYPT_OPAQUE_KEY_BLOB )) { @@ -1478,7 +1908,7 @@ static NTSTATUS key_import( struct algorithm *alg, struct key *decrypt_key, cons len = *(ULONG *)input; if (len + sizeof(len) > input_len) return STATUS_INVALID_PARAMETER; - return key_symmetric_generate( alg, key, input + sizeof(len), len ); + return generate_symmetric_key( alg, input + sizeof(len), len, ret_key); } else if (!wcscmp( type, BCRYPT_AES_WRAP_KEY_BLOB )) { @@ -1489,120 +1919,256 @@ static NTSTATUS key_import( struct algorithm *alg, struct key *decrypt_key, cons len = input_len - 8; if (len < BLOCK_LENGTH_AES || len & (BLOCK_LENGTH_AES - 1)) return STATUS_INVALID_PARAMETER; - if ((status = aes_unwrap( decrypt_key->u.s.secret, decrypt_key->u.s.secret_len, input, len, output ))) + if ((status = unwrap_aes( decrypt_key->s.secret, decrypt_key->s.secret_len, input, len, output ))) return status; - return key_symmetric_generate( alg, key, output, len ); + return generate_symmetric_key( alg, output, len, ret_key ); } - FIXME( "unsupported key type %s\n", debugstr_w(type) ); + FIXME( "unsupported blob type %s\n", debugstr_w(type) ); return STATUS_NOT_IMPLEMENTED; } -static NTSTATUS key_symmetric_encrypt( struct key *key, UCHAR *input, ULONG input_len, void *padding, UCHAR *iv, - ULONG iv_len, UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +NTSTATUS WINAPI BCryptImportKey( BCRYPT_ALG_HANDLE handle, BCRYPT_KEY_HANDLE decrypt_key_handle, const WCHAR *type, + BCRYPT_KEY_HANDLE *ret_handle, UCHAR *object, ULONG object_len, UCHAR *input, + ULONG input_len, ULONG flags ) { - struct key_symmetric_set_auth_data_params auth_params; - struct key_symmetric_encrypt_params encrypt_params; - struct key_symmetric_get_tag_params tag_params; - ULONG bytes_left = input_len; - UCHAR *buf; + const struct algorithm *alg = get_alg_object( handle ); + struct key *key; + const struct key *decrypt_key = NULL; NTSTATUS status; - if (key->u.s.mode == CHAIN_MODE_GCM) - { - BCRYPT_AUTHENTICATED_CIPHER_MODE_INFO *auth_info = padding; + TRACE( "%p, %p, %s, %p, %p, %lu, %p, %lu, %#lx\n", handle, decrypt_key_handle, debugstr_w(type), ret_handle, + object, object_len, input, input_len, flags ); + + if (!alg || !is_symmetric_alg( alg )) return STATUS_INVALID_HANDLE; + if (!ret_handle || !type || !input) return STATUS_INVALID_PARAMETER; + if (decrypt_key_handle && !(decrypt_key = get_key_object( decrypt_key_handle ))) return STATUS_INVALID_HANDLE; - if (!auth_info) return STATUS_INVALID_PARAMETER; - if (!auth_info->pbNonce) return STATUS_INVALID_PARAMETER; - if (!auth_info->pbTag) return STATUS_INVALID_PARAMETER; - if (auth_info->cbTag < 12 || auth_info->cbTag > 16) return STATUS_INVALID_PARAMETER; - if (auth_info->dwFlags & BCRYPT_AUTH_MODE_CHAIN_CALLS_FLAG) - FIXME( "call chaining not implemented\n" ); + if ((status = import_key( alg, decrypt_key, type, input, input_len, &key ))) return status; + *ret_handle = key; - if ((status = key_symmetric_set_vector( key, auth_info->pbNonce, auth_info->cbNonce, TRUE ))) - return status; + TRACE( "returning handle %p\n", *ret_handle ); + return STATUS_SUCCESS; +} + +static NTSTATUS encrypt_aes_ecb( const struct key *key, const UCHAR *input, ULONG input_len, UCHAR *output, + ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + ULONG bytes_left, offset, block_size = key->s.block_size; + UCHAR buf[BLOCK_LENGTH_AES]; + + if (flags & BCRYPT_BLOCK_PADDING) *ret_len = (input_len + block_size) & ~(block_size - 1); + else if (input_len & (block_size - 1)) return STATUS_INVALID_BUFFER_SIZE; + + if (!output) return STATUS_SUCCESS; + if (output_len < *ret_len) return STATUS_BUFFER_TOO_SMALL; - *ret_len = input_len; - if (flags & BCRYPT_BLOCK_PADDING) return STATUS_INVALID_PARAMETER; - if (input && !output) return STATUS_SUCCESS; - if (output_len < *ret_len) return STATUS_BUFFER_TOO_SMALL; + bytes_left = input_len % block_size; + offset = input_len - bytes_left; - auth_params.key = key; - auth_params.auth_data = auth_info->pbAuthData; - auth_params.len = auth_info->cbAuthData; - if ((status = UNIX_CALL( key_symmetric_set_auth_data, &auth_params ))) return status; + SymCryptEcbEncrypt( SymCryptAesBlockCipher, &key->s.aes.handle, input, output, offset ); - encrypt_params.key = key; - encrypt_params.input = input; - encrypt_params.input_len = input_len; - encrypt_params.output = output; - encrypt_params.output_len = output_len; - if ((status = UNIX_CALL( key_symmetric_encrypt, &encrypt_params ))) return status; + if (flags & BCRYPT_BLOCK_PADDING) + { + memcpy( buf, input + offset, bytes_left ); + memset( buf + bytes_left, block_size - bytes_left, block_size - bytes_left ); - tag_params.key = key; - tag_params.tag = auth_info->pbTag; - tag_params.len = auth_info->cbTag; - return UNIX_CALL( key_symmetric_get_tag, &tag_params ); + SymCryptEcbEncrypt( SymCryptAesBlockCipher, &key->s.aes.handle, buf, output + offset, block_size ); } - *ret_len = input_len; + return STATUS_SUCCESS; +} - if (flags & BCRYPT_BLOCK_PADDING) - *ret_len = (input_len + key->u.s.block_size) & ~(key->u.s.block_size - 1); - else if (input_len & (key->u.s.block_size - 1)) - return STATUS_INVALID_BUFFER_SIZE; +static NTSTATUS encrypt_aes_gcm( struct key *key, const UCHAR *input, ULONG input_len, + const BCRYPT_AUTHENTICATED_CIPHER_MODE_INFO *info, UCHAR *output ) +{ + if (!info || !info->pbNonce || !info->pbTag || info->cbTag < 12 || info->cbTag > 16) + return STATUS_INVALID_PARAMETER; + if (info->dwFlags & BCRYPT_AUTH_MODE_CHAIN_CALLS_FLAG) FIXME( "call chaining not implemented\n" ); + + if (input && !input_len) return STATUS_SUCCESS; + + EnterCriticalSection( &key->s.cs ); + + SymCryptGcmInit( &key->s.gcm.state, &key->s.gcm.handle, info->pbNonce, info->cbNonce ); + SymCryptGcmAuthPart( &key->s.gcm.state, info->pbAuthData, info->cbAuthData ); + SymCryptGcmEncryptPart( &key->s.gcm.state, input, output, input_len ); + SymCryptGcmEncryptFinal( &key->s.gcm.state, info->pbTag, info->cbTag ); + + LeaveCriticalSection( &key->s.cs ); + return STATUS_SUCCESS; +} + +static NTSTATUS encrypt_aes_vector( struct key *key, const UCHAR *input, ULONG input_len, UCHAR *iv, UCHAR *output, + ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + ULONG bytes_left = input_len, block_size = key->s.block_size; + UCHAR buf[BLOCK_LENGTH_AES], *out; + const UCHAR *in; + + if (flags & BCRYPT_BLOCK_PADDING) *ret_len = (input_len + block_size) & ~(block_size - 1); + else if (input_len & (block_size - 1)) return STATUS_INVALID_BUFFER_SIZE; if (!output) return STATUS_SUCCESS; if (output_len < *ret_len) return STATUS_BUFFER_TOO_SMALL; - if (key->u.s.mode == CHAIN_MODE_ECB && iv) return STATUS_INVALID_PARAMETER; - if ((status = key_symmetric_set_vector( key, iv, iv_len, flags & BCRYPT_BLOCK_PADDING ))) return status; - encrypt_params.key = key; - encrypt_params.input = input; - encrypt_params.input_len = key->u.s.block_size; - encrypt_params.output = output; - encrypt_params.output_len = key->u.s.block_size; - while (bytes_left >= key->u.s.block_size) + EnterCriticalSection( &key->s.cs ); + + if (iv) memcpy( key->s.vector, iv, block_size ); + + in = input; + out = output; + while (bytes_left >= block_size) { - if ((status = UNIX_CALL( key_symmetric_encrypt, &encrypt_params ))) - return status; - if (key->u.s.mode == CHAIN_MODE_ECB && (status = key_symmetric_set_vector( key, NULL, 0, TRUE ))) - return status; - bytes_left -= key->u.s.block_size; - encrypt_params.input += key->u.s.block_size; - encrypt_params.output += key->u.s.block_size; + if (key->s.mode == CHAIN_MODE_CFB) + SymCryptCfbEncrypt( SymCryptAesBlockCipher, 1, &key->s.aes.handle, key->s.vector, in, out, block_size ); + else + SymCryptCbcEncrypt( SymCryptAesBlockCipher, &key->s.aes.handle, key->s.vector, in, out, block_size ); + + bytes_left -= block_size; + in += block_size; + out += block_size; } if (flags & BCRYPT_BLOCK_PADDING) { - if (!(buf = malloc( key->u.s.block_size ))) return STATUS_NO_MEMORY; - memcpy( buf, encrypt_params.input, bytes_left ); - memset( buf + bytes_left, key->u.s.block_size - bytes_left, key->u.s.block_size - bytes_left ); - encrypt_params.input = buf; - status = UNIX_CALL( key_symmetric_encrypt, &encrypt_params ); - free( buf ); + memcpy( buf, in, bytes_left ); + memset( buf + bytes_left, block_size - bytes_left, block_size - bytes_left ); + + if (key->s.mode == CHAIN_MODE_CFB) + SymCryptCfbEncrypt( SymCryptAesBlockCipher, 1, &key->s.aes.handle, key->s.vector, buf, out, block_size ); + else + SymCryptCbcEncrypt( SymCryptAesBlockCipher, &key->s.aes.handle, key->s.vector, buf, out, block_size ); } - if (!status) + if (iv) memcpy( iv, key->s.vector, block_size ); + + LeaveCriticalSection( &key->s.cs ); + return STATUS_SUCCESS; +} + +static NTSTATUS encrypt_symmetric( struct key *key, const UCHAR *input, ULONG input_len, + const BCRYPT_AUTHENTICATED_CIPHER_MODE_INFO *info, UCHAR *iv, ULONG iv_len, + UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + *ret_len = input_len; + + if (key->s.mode == CHAIN_MODE_ECB && iv) return STATUS_INVALID_PARAMETER; + if (key->s.mode == CHAIN_MODE_GCM && (flags & BCRYPT_BLOCK_PADDING)) return STATUS_INVALID_PARAMETER; + + switch (key->alg_id) + { + case ALG_ID_AES: + switch (key->s.mode) + { + case CHAIN_MODE_ECB: + return encrypt_aes_ecb( key, input, input_len, output, output_len, ret_len, flags ); + + case CHAIN_MODE_GCM: + return encrypt_aes_gcm( key, input, input_len, info, output ); + + default: + if (iv && iv_len != key->s.block_size) return STATUS_INVALID_PARAMETER; + return encrypt_aes_vector( key, input, input_len, iv, output, output_len, ret_len, flags ); + } + default: + FIXME( "unhandled algorithm %u\n", key->alg_id ); + break; + } + + return STATUS_NOT_SUPPORTED; +} + +static NTSTATUS encrypt_rsa( const struct key *key, const UCHAR *input, ULONG input_len, const void *padding, + UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + SIZE_T size = len_from_bitlen( key->a.bitlen ); + + *ret_len = size; + if (output_len < size) + { + if (output) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; + } + + if (!flags || flags == BCRYPT_PAD_NONE) + { + if (input_len % size) return STATUS_INVALID_PARAMETER; + + if (SymCryptRsaRawEncrypt( key->a.rsa.handle, input, input_len, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0, output, + output_len )) return STATUS_INTERNAL_ERROR; + } + else if (flags == BCRYPT_PAD_PKCS1) + { + if (SymCryptRsaPkcs1Encrypt( key->a.rsa.handle, input, input_len, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0, output, + output_len, &size )) return STATUS_INTERNAL_ERROR; + *ret_len = size; + } + else if (flags == BCRYPT_PAD_OAEP) + { + const BCRYPT_OAEP_PADDING_INFO *pad = padding; + const SYMCRYPT_HASH *hash_desc = SymCryptSha1Algorithm; + const BYTE *label = NULL; + SIZE_T label_len = 0; + + if (pad) + { + hash_desc = get_hash_from_str( pad->pszAlgId ); + label = pad->pbLabel; + label_len = pad->cbLabel; + } + if (SymCryptRsaOaepEncrypt( key->a.rsa.handle, input, input_len, hash_desc, label, label_len, 0, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, output, output_len, &size )) + return STATUS_INTERNAL_ERROR; + *ret_len = size; + } + else + { + FIXME( "unsupported flags %#lx", flags ); + return STATUS_NOT_SUPPORTED; + } + + return STATUS_SUCCESS; +} + +NTSTATUS WINAPI BCryptEncrypt( BCRYPT_KEY_HANDLE handle, UCHAR *input, ULONG input_len, void *padding, UCHAR *iv, + ULONG iv_len, UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +{ + struct key *key = get_key_object( handle ); + NTSTATUS status; + + TRACE( "%p, %p, %lu, %p, %p, %lu, %p, %lu, %p, %#lx\n", handle, input, input_len, padding, iv, iv_len, output, + output_len, ret_len, flags ); + + if (!key) return STATUS_INVALID_HANDLE; + + if (is_symmetric_key( key )) { - if (key->u.s.vector && *ret_len >= key->u.s.vector_len) + if (flags & ~BCRYPT_BLOCK_PADDING) { - memcpy( key->u.s.vector, output + *ret_len - key->u.s.vector_len, key->u.s.vector_len ); - if (iv) memcpy( iv, key->u.s.vector, min( iv_len, key->u.s.vector_len )); + FIXME( "flags %#lx not implemented\n", flags ); + return STATUS_NOT_IMPLEMENTED; } - else if (key->u.s.vector) - FIXME( "Unexpected vector len %lu, *ret_len %lu.\n", key->u.s.vector_len, *ret_len ); + + status = encrypt_symmetric( key, input, input_len, padding, iv, iv_len, output, output_len, ret_len, flags ); } + else + { + if (key->alg_id != ALG_ID_RSA) return STATUS_NOT_SUPPORTED; + if (!(key->flags & KEY_FLAG_FINALIZED)) return STATUS_INVALID_HANDLE; + status = encrypt_rsa( key, input, input_len, padding, output, output_len, ret_len, flags ); + } return status; } /* AES Key Wrap Algorithm (RFC3394) */ -static NTSTATUS aes_wrap( const UCHAR *secret, ULONG secret_len, const UCHAR *plain, ULONG plain_len, UCHAR *cipher ) +static NTSTATUS wrap_aes( const UCHAR *secret, ULONG secret_len, const UCHAR *plain, ULONG plain_len, UCHAR *cipher ) { UCHAR *a, *r, b[16]; - ULONG len, t, i, j, n = plain_len / 8; + ULONG len = 16, t, i, j, n = plain_len / 8; + NTSTATUS status; struct key *key; a = cipher; @@ -1611,7 +2177,12 @@ static NTSTATUS aes_wrap( const UCHAR *secret, ULONG secret_len, const UCHAR *pl memset( a, 0xa6, 8 ); memcpy( r, plain, 8 * n ); - if (!(key = key_symmetric_create( ALG_ID_AES, CHAIN_MODE_ECB, 16, secret, secret_len ))) return STATUS_NO_MEMORY; + if ((status = alloc_key( ALG_ID_AES, 0, &key ))) return status; + if ((status = alloc_aes_key( key, CHAIN_MODE_ECB, BLOCK_LENGTH_AES, secret, secret_len ))) + { + destroy_key( key ); + return status; + } for (j = 0; j <= 5; j++) { @@ -1620,7 +2191,7 @@ static NTSTATUS aes_wrap( const UCHAR *secret, ULONG secret_len, const UCHAR *pl { memcpy( b, a, 8 ); memcpy( b + 8, r, 8 ); - key_symmetric_encrypt( key, b, 16, NULL, NULL, 0, b, 16, &len, 0 ); + encrypt_aes_ecb( key, b, 16, b, 16, &len, 0 ); memcpy( a, b, 8 ); t = n * j + i; a[7] ^= t; @@ -1632,14 +2203,13 @@ static NTSTATUS aes_wrap( const UCHAR *secret, ULONG secret_len, const UCHAR *pl } } - key_destroy( key ); + destroy_key( key ); return STATUS_SUCCESS; } -static NTSTATUS key_export( struct key *key, struct key *encrypt_key, const WCHAR *type, UCHAR *output, - ULONG output_len, ULONG *size ) +static NTSTATUS export_symmetric_key( const struct key *key, const struct key *encrypt_key, const WCHAR *type, + UCHAR *output, ULONG output_len, ULONG *ret_len ) { - struct key_asymmetric_export_params params; NTSTATUS status; if (encrypt_key && wcscmp( type, BCRYPT_AES_WRAP_KEY_BLOB )) @@ -1651,615 +2221,1157 @@ static NTSTATUS key_export( struct key *key, struct key *encrypt_key, const WCHA if (!wcscmp( type, BCRYPT_KEY_DATA_BLOB )) { BCRYPT_KEY_DATA_BLOB_HEADER *header = (BCRYPT_KEY_DATA_BLOB_HEADER *)output; - ULONG req_size = sizeof(BCRYPT_KEY_DATA_BLOB_HEADER) + key->u.s.secret_len; + ULONG size = sizeof(BCRYPT_KEY_DATA_BLOB_HEADER) + key->s.secret_len; - *size = req_size; - if (output_len < req_size) return STATUS_BUFFER_TOO_SMALL; + *ret_len = size; if (output) { + if (output_len < size) return STATUS_BUFFER_TOO_SMALL; header->dwMagic = BCRYPT_KEY_DATA_BLOB_MAGIC; header->dwVersion = BCRYPT_KEY_DATA_BLOB_VERSION1; - header->cbKeyData = key->u.s.secret_len; - memcpy( &header[1], key->u.s.secret, key->u.s.secret_len ); + header->cbKeyData = key->s.secret_len; + memcpy( &header[1], key->s.secret, key->s.secret_len ); } return STATUS_SUCCESS; } else if (!wcscmp( type, BCRYPT_OPAQUE_KEY_BLOB )) { - ULONG len, req_size = sizeof(len) + key->u.s.secret_len; + ULONG size = sizeof(ULONG) + key->s.secret_len; - *size = req_size; - if (output_len < req_size) return STATUS_BUFFER_TOO_SMALL; + *ret_len = size; if (output) { - *(ULONG *)output = key->u.s.secret_len; - memcpy( output + sizeof(len), key->u.s.secret, key->u.s.secret_len ); + if (output_len < size) return STATUS_BUFFER_TOO_SMALL; + *(ULONG *)output = key->s.secret_len; + memcpy( output + sizeof(ULONG), key->s.secret, key->s.secret_len ); } return STATUS_SUCCESS; } - else if (!wcscmp( type, BCRYPT_DSA_PRIVATE_BLOB ) || !wcscmp( type, LEGACY_DSA_V2_PRIVATE_BLOB ) || - !wcscmp( type, BCRYPT_ECCPRIVATE_BLOB ) || !wcscmp( type, BCRYPT_DH_PRIVATE_BLOB )) - { - params.key = key; - params.flags = 0; - params.buf = output; - params.len = output_len; - params.ret_len = size; - return UNIX_CALL( key_asymmetric_export, ¶ms ); - } - else if (!wcscmp( type, BCRYPT_RSAPRIVATE_BLOB ) || !wcscmp( type, BCRYPT_RSAFULLPRIVATE_BLOB )) - { - params.key = key; - params.flags = (wcscmp( type, BCRYPT_RSAPRIVATE_BLOB )) ? KEY_EXPORT_FLAG_RSA_FULL : 0; - params.buf = output; - params.len = output_len; - params.ret_len = size; - return UNIX_CALL( key_asymmetric_export, ¶ms ); - } - else if (!wcscmp( type, BCRYPT_DSA_PUBLIC_BLOB ) || !wcscmp( type, LEGACY_DSA_V2_PUBLIC_BLOB ) || - !wcscmp( type, BCRYPT_ECCPUBLIC_BLOB ) || !wcscmp( type, BCRYPT_RSAPUBLIC_BLOB ) || - !wcscmp( type, BCRYPT_DH_PUBLIC_BLOB )) - { - params.key = key; - params.flags = KEY_EXPORT_FLAG_PUBLIC; - params.buf = output; - params.len = output_len; - params.ret_len = size; - return UNIX_CALL( key_asymmetric_export, ¶ms ); - } else if (!wcscmp( type, BCRYPT_AES_WRAP_KEY_BLOB )) { - ULONG req_size = key->u.s.secret_len + 8; + ULONG size = key->s.secret_len + 8; if (!encrypt_key) return STATUS_INVALID_PARAMETER; - *size = req_size; + *ret_len = size; if (output) { - if (output_len < req_size) return STATUS_BUFFER_TOO_SMALL; - if ((status = aes_wrap( encrypt_key->u.s.secret, encrypt_key->u.s.secret_len, key->u.s.secret, key->u.s.secret_len, output ))) - return status; + if (output_len < size) return STATUS_BUFFER_TOO_SMALL; + if ((status = wrap_aes( encrypt_key->s.secret, encrypt_key->s.secret_len, key->s.secret, + key->s.secret_len, output ))) return status; } return STATUS_SUCCESS; } - FIXME( "unsupported key type %s\n", debugstr_w(type) ); + FIXME( "unsupported blob type %s\n", debugstr_w(type) ); return STATUS_NOT_IMPLEMENTED; } -static NTSTATUS convert_legacy_rsaprivate_blob( struct algorithm *alg, BCRYPT_RSAKEY_BLOB **rsa_data, - ULONG *rsa_len, UCHAR *input, ULONG input_len ) -{ - struct { - PUBLICKEYSTRUC header; - RSAPUBKEY rsapubkey; - } *blob = (void *)input; - ULONG len, pos; - UCHAR *conv; - int i; - - if (alg->id != ALG_ID_RSA) return STATUS_NOT_SUPPORTED; - if (input_len < sizeof(*blob)) return NTE_BAD_DATA; - if (blob->header.bType != PRIVATEKEYBLOB || blob->header.bVersion != CUR_BLOB_VERSION || - blob->header.aiKeyAlg != CALG_RSA_KEYX || - blob->rsapubkey.magic != BCRYPT_RSAPRIVATE_MAGIC || - input_len != sizeof(*blob) + blob->rsapubkey.bitlen / 16 * 9) return NTE_BAD_DATA; - if (blob->rsapubkey.bitlen & 0xf) return STATUS_INVALID_PARAMETER; - - len = sizeof(**rsa_data) + sizeof(blob->rsapubkey.pubexp) + blob->rsapubkey.bitlen / 4; - if (!(conv = malloc( len ))) return STATUS_NO_MEMORY; - - *rsa_data = (BCRYPT_RSAKEY_BLOB *)conv; - (*rsa_data)->Magic = blob->rsapubkey.magic; - (*rsa_data)->BitLength = blob->rsapubkey.bitlen; - (*rsa_data)->cbPublicExp = sizeof(blob->rsapubkey.pubexp); - (*rsa_data)->cbModulus = blob->rsapubkey.bitlen / 8; - (*rsa_data)->cbPrime1 = blob->rsapubkey.bitlen / 16; - (*rsa_data)->cbPrime2 = blob->rsapubkey.bitlen / 16; - len = sizeof(**rsa_data); - - for(i = 0; i < (*rsa_data)->cbPublicExp; i++) - conv[len++] = ((UCHAR *)&blob->rsapubkey.pubexp)[(*rsa_data)->cbPublicExp - i - 1]; - pos = sizeof(*blob); - for(i = 0; i < (*rsa_data)->cbModulus; i++) - conv[len++] = input[pos + (*rsa_data)->cbModulus - i - 1]; - pos += (*rsa_data)->cbModulus; - for(i = 0; i < (*rsa_data)->cbPrime1; i++) - conv[len++] = input[pos + (*rsa_data)->cbPrime1 - i - 1]; - pos += (*rsa_data)->cbPrime1; - for(i = 0; i < (*rsa_data)->cbPrime2; i++) - conv[len++] = input[pos + (*rsa_data)->cbPrime2 - i - 1]; - - *rsa_len = len; - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_pair( struct algorithm *alg, const WCHAR *type, BCRYPT_KEY_HANDLE *ret_key, UCHAR *input, - ULONG input_len ) +static NTSTATUS export_rsa_key( const struct key *key, UCHAR *output, ULONG output_len, ULONG flags, ULONG *ret_len ) { - struct key_asymmetric_import_params params; - struct key *key; - NTSTATUS status; + BCRYPT_RSAKEY_BLOB *blob = (BCRYPT_RSAKEY_BLOB *)output; + BYTE *mod, *primes[2] = {}, *exponents[2], *coefficient, *private_exp, *ptr = (BYTE *)(blob + 1); + UINT64 pubexp; + SIZE_T mod_size, pubexp_size = sizeof(pubexp), primes_sizes[2] = {}; + ULONG size, magic, num_primes = 0; + + pubexp_size = SymCryptRsakeySizeofPublicExponent( key->a.rsa.handle, 0 ); + mod_size = SymCryptRsakeySizeofModulus( key->a.rsa.handle ); + + if (flags & KEY_EXPORT_FLAG_PRIVATE) + { + if (!(key->flags & KEY_FLAG_PRIVATE)) return STATUS_INVALID_PARAMETER; + primes_sizes[0] = SymCryptRsakeySizeofPrime( key->a.rsa.handle, 0 ); + primes_sizes[1] = SymCryptRsakeySizeofPrime( key->a.rsa.handle, 1 ); + num_primes = 2; + } + else primes_sizes[0] = primes_sizes[1] = 0; + + size = sizeof(*blob) + pubexp_size + mod_size + primes_sizes[0] + primes_sizes[1]; + if (flags & KEY_EXPORT_FLAG_RSA_FULL) size += primes_sizes[0] * 2 + primes_sizes[1] + mod_size; + if (output_len < size || !output) + { + *ret_len = size; + if (output) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; + } + + if (flags & KEY_EXPORT_FLAG_RSA_FULL) magic = BCRYPT_RSAFULLPRIVATE_MAGIC; + else if (flags & KEY_EXPORT_FLAG_PRIVATE) magic = BCRYPT_RSAPRIVATE_MAGIC; + else magic = BCRYPT_RSAPUBLIC_MAGIC; + + blob->Magic = magic; + blob->BitLength = key->a.bitlen; + blob->cbPublicExp = pubexp_size; + blob->cbModulus = mod_size; + blob->cbPrime1 = primes_sizes[0]; + blob->cbPrime2 = primes_sizes[1]; + + mod = ptr + pubexp_size; + primes[0] = mod + mod_size; + primes[1] = mod + mod_size + primes_sizes[0]; + + if (SymCryptRsakeyGetValue( key->a.rsa.handle, mod, mod_size, &pubexp, 1, primes, primes_sizes, num_primes, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0 )) return STATUS_INTERNAL_ERROR; + + memcpy( ptr, &pubexp, pubexp_size ); + + if (flags & KEY_EXPORT_FLAG_RSA_FULL) + { + exponents[0] = primes[1] + primes_sizes[1]; + exponents[1] = exponents[0] + primes_sizes[0]; + coefficient = exponents[1] + primes_sizes[1]; + private_exp = coefficient + primes_sizes[0]; + + if (SymCryptRsakeyGetCrtValue( key->a.rsa.handle, exponents, primes_sizes, 2, coefficient, primes_sizes[0], + private_exp, mod_size, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0 )) + return STATUS_INTERNAL_ERROR; + } + + *ret_len = size; + return STATUS_SUCCESS; +} + +static DWORD get_ecc_magic( enum alg_id alg, BOOL private ) +{ + switch (alg) + { + case ALG_ID_ECDH: return private ? BCRYPT_ECDH_PRIVATE_GENERIC_MAGIC : BCRYPT_ECDH_PUBLIC_GENERIC_MAGIC; + case ALG_ID_ECDH_P256: return private ? BCRYPT_ECDH_PRIVATE_P256_MAGIC : BCRYPT_ECDH_PUBLIC_P256_MAGIC; + case ALG_ID_ECDH_P384: return private ? BCRYPT_ECDH_PRIVATE_P384_MAGIC : BCRYPT_ECDH_PUBLIC_P384_MAGIC; + case ALG_ID_ECDH_P521: return private ? BCRYPT_ECDH_PRIVATE_P521_MAGIC : BCRYPT_ECDH_PUBLIC_P521_MAGIC; + + case ALG_ID_ECDSA: return private ? BCRYPT_ECDSA_PRIVATE_GENERIC_MAGIC : BCRYPT_ECDSA_PUBLIC_GENERIC_MAGIC; + case ALG_ID_ECDSA_P256: return private ? BCRYPT_ECDSA_PRIVATE_P256_MAGIC : BCRYPT_ECDSA_PUBLIC_P256_MAGIC; + case ALG_ID_ECDSA_P384: return private ? BCRYPT_ECDSA_PRIVATE_P384_MAGIC : BCRYPT_ECDSA_PUBLIC_P384_MAGIC; + case ALG_ID_ECDSA_P521: return private ? BCRYPT_ECDSA_PRIVATE_P521_MAGIC : BCRYPT_ECDSA_PUBLIC_P521_MAGIC; + default: + FIXME( "unhandled algorithm %u\n", alg ); + return 0; + } +} + +static NTSTATUS export_ecc_key( const struct key *key, UCHAR *output, ULONG output_len, ULONG flags, ULONG *ret_len ) +{ + BCRYPT_ECCKEY_BLOB *blob = (BCRYPT_ECCKEY_BLOB *)output; + UINT32 pub_size = SymCryptEckeySizeofPublicKey( key->a.ecc.handle, SYMCRYPT_ECPOINT_FORMAT_XY ), priv_size = 0; + UCHAR *pub = (UCHAR *)(blob + 1), *priv = NULL; + ULONG size, key_size = len_from_bitlen( key->a.bitlen ); + + size = sizeof(*blob) + key_size * 2; + if (flags & KEY_EXPORT_FLAG_PRIVATE) + { + if (!(key->flags & KEY_FLAG_PRIVATE)) return STATUS_INVALID_PARAMETER; + priv_size = SymCryptEckeySizeofPrivateKey( key->a.ecc.handle ); + priv = pub + key_size * 2; + size += key_size; + } + if (output_len < size || !output) + { + *ret_len = size; + if (output) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; + } + + blob->dwMagic = get_ecc_magic( key->alg_id, flags & KEY_EXPORT_FLAG_PRIVATE ); + blob->cbKey = key_size; + + if (SymCryptEckeyGetValue( key->a.ecc.handle, priv, priv_size, pub, pub_size, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, + SYMCRYPT_ECPOINT_FORMAT_XY, 0 )) return STATUS_INTERNAL_ERROR; + *ret_len = size; + return STATUS_SUCCESS; +} + +static NTSTATUS export_dsa_key( const struct key *key, UCHAR *output, ULONG output_len, ULONG flags, ULONG *ret_len ) +{ + BCRYPT_DSA_KEY_BLOB *blob = (BCRYPT_DSA_KEY_BLOB *)output; + UINT32 key_size = len_from_bitlen( key->a.bitlen ), priv_size = 0; + UCHAR *prime, *generator, *pub, *priv = NULL; ULONG size; - if (!wcscmp( type, BCRYPT_ECCPUBLIC_BLOB )) + size = sizeof(*blob) + key_size * 3; + if (flags & KEY_EXPORT_FLAG_PRIVATE) + { + if (!(key->flags & KEY_FLAG_PRIVATE)) return STATUS_INVALID_PARAMETER; + priv_size = 20; + size += priv_size; + } + if (output_len < size || !output) { - BCRYPT_ECCKEY_BLOB *ecc_blob = (BCRYPT_ECCKEY_BLOB *)input; - DWORD bitlen, magic, magic_priv; - enum ecc_curve_id curve; + *ret_len = size; + if (output) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; + } - if (input_len < sizeof(*ecc_blob)) return STATUS_INVALID_PARAMETER; + blob->dwMagic = (flags & KEY_EXPORT_FLAG_PRIVATE) ? BCRYPT_DSA_PRIVATE_MAGIC : BCRYPT_DSA_PUBLIC_MAGIC; + blob->cbKey = key_size; - switch (alg->id) - { - case ALG_ID_ECDH: - bitlen = ecc_blob->cbKey * 8; - curve = alg->curve_id; - magic = BCRYPT_ECDH_PUBLIC_GENERIC_MAGIC; - magic_priv = BCRYPT_ECDH_PRIVATE_GENERIC_MAGIC; - break; + prime = (UCHAR *)(blob + 1); + generator = prime + key_size; - case ALG_ID_ECDH_P256: - bitlen = 256; - curve = ECC_CURVE_P256R1; - magic = BCRYPT_ECDH_PUBLIC_P256_MAGIC; - magic_priv = BCRYPT_ECDH_PRIVATE_P256_MAGIC; - break; + if (SymCryptDlgroupGetValue( key->a.dsa.group, prime, key_size, blob->q, sizeof(blob->q), generator, key_size, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, NULL, blob->Seed, sizeof(blob->Seed), + (UINT32 *)blob->Count )) return STATUS_INTERNAL_ERROR; + pub = generator + key_size; + if (flags & KEY_EXPORT_FLAG_PRIVATE) priv = pub + key_size; + + if (SymCryptDlkeyGetValue( key->a.dsa.handle, priv, priv_size, pub, key_size, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, + 0 )) return STATUS_INTERNAL_ERROR; + *ret_len = size; + return STATUS_SUCCESS; +} + +static NTSTATUS export_legacy_dsa_key( const struct key *key, UCHAR *output, ULONG output_len, ULONG flags, + ULONG *ret_len ) +{ + BLOBHEADER *hdr = (BLOBHEADER *)output; + DSSPUBKEY *dsskey = (DSSPUBKEY *)(hdr + 1); + UINT32 key_size = len_from_bitlen( key->a.bitlen ), pub_size = 0, priv_size = 0; + UCHAR *prime_p, *prime_q, *generator, *pub = NULL, *priv = NULL, *seed, *count; + ULONG size; + + if ((flags & KEY_EXPORT_FLAG_PRIVATE) && !(key->flags & KEY_FLAG_PRIVATE)) return STATUS_INVALID_PARAMETER; + + size = sizeof(*hdr) + sizeof(*dsskey) + key_size * 2 + 20 /* q */ + 20 /* x or y */ + 20 /* seed */ + 4 /* count */; + if (output_len < size || !output) + { + *ret_len = size; + if (output) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; + } + + hdr->bType = (flags & KEY_EXPORT_FLAG_PRIVATE) ? PRIVATEKEYBLOB : PUBLICKEYBLOB; + hdr->bVersion = 2; + hdr->reserved = 0; + hdr->aiKeyAlg = CALG_DSS_SIGN; + + dsskey->magic = (flags & KEY_EXPORT_FLAG_PRIVATE) ? MAGIC_DSS2 : MAGIC_DSS1; + dsskey->bitlen = key->a.bitlen; + + prime_p = (UCHAR *)(dsskey + 1); + prime_q = prime_p + key_size; + generator = prime_q + 20; + seed = generator + key_size + 20; + count = seed + 20; + + if (SymCryptDlgroupGetValue( key->a.dsa.group, prime_p, key_size, prime_q, 20, generator, key_size, + SYMCRYPT_NUMBER_FORMAT_LSB_FIRST, NULL, seed, 20, (UINT32 *)count )) + return STATUS_INTERNAL_ERROR; + + if (flags & KEY_EXPORT_FLAG_PRIVATE) + { + priv = generator + key_size; + priv_size = 20; + } + else + { + pub = generator + key_size; + pub_size = 20; + } + + if (SymCryptDlkeyGetValue( key->a.dsa.handle, priv, priv_size, pub, pub_size, SYMCRYPT_NUMBER_FORMAT_LSB_FIRST, + 0 )) return STATUS_INTERNAL_ERROR; + *ret_len = size; + return STATUS_SUCCESS; +} + +static NTSTATUS export_dh_key( const struct key *key, UCHAR *output, ULONG output_len, ULONG flags, ULONG *ret_len ) +{ + BCRYPT_DH_KEY_BLOB *blob = (BCRYPT_DH_KEY_BLOB *)output; + UCHAR *modulus, *generator, *pub, *priv = NULL; + ULONG size, priv_size = 0, key_size = len_from_bitlen( key->a.bitlen ); + + size = sizeof(*blob) + key_size * 3; + if (flags & KEY_EXPORT_FLAG_PRIVATE) + { + if (!(key->flags & KEY_FLAG_PRIVATE)) return STATUS_INVALID_PARAMETER; + priv_size = key_size; + size += key_size; + } + if (output_len < size || !output) + { + *ret_len = size; + if (output) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; + } + + blob->dwMagic = (flags & KEY_EXPORT_FLAG_PRIVATE) ? BCRYPT_DH_PRIVATE_MAGIC : BCRYPT_DH_PUBLIC_MAGIC; + blob->cbKey = key_size; + + modulus = (UCHAR *)(blob + 1); + generator = modulus + key_size; + + if (SymCryptDlgroupGetValue( key->a.dsa.group, modulus, key_size, NULL, 0, generator, key_size, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, NULL, NULL, 0, 0 )) return STATUS_INTERNAL_ERROR; + pub = generator + key_size; + if (flags & KEY_EXPORT_FLAG_PRIVATE) priv = pub + key_size; + + if (SymCryptDlkeyGetValue( key->a.dsa.handle, priv, priv_size, pub, key_size, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0 )) return STATUS_INTERNAL_ERROR; + *ret_len = size; + return STATUS_SUCCESS; +} + +static NTSTATUS export_asymmetric_key( const struct key *key, const WCHAR *type, UCHAR *output, ULONG output_len, + ULONG *ret_len ) +{ + if (!(key->flags & KEY_FLAG_FINALIZED)) return STATUS_INVALID_HANDLE; + + if (!wcscmp( type, BCRYPT_ECCPRIVATE_BLOB )) + { + return export_ecc_key( key, output, output_len, KEY_EXPORT_FLAG_PRIVATE, ret_len ); + } + else if (!wcscmp( type, BCRYPT_ECCPUBLIC_BLOB )) + { + return export_ecc_key( key, output, output_len, 0, ret_len ); + } + else if (!wcscmp( type, BCRYPT_DSA_PRIVATE_BLOB )) + { + return export_dsa_key( key, output, output_len, KEY_EXPORT_FLAG_PRIVATE, ret_len ); + } + else if (!wcscmp( type, BCRYPT_DSA_PUBLIC_BLOB )) + { + return export_dsa_key( key, output, output_len, 0, ret_len ); + } + else if (!wcscmp( type, BCRYPT_DH_PRIVATE_BLOB )) + { + return export_dh_key( key, output, output_len, KEY_EXPORT_FLAG_PRIVATE, ret_len ); + } + else if (!wcscmp( type, BCRYPT_DH_PUBLIC_BLOB )) + { + return export_dh_key( key, output, output_len, 0, ret_len ); + } + else if (!wcscmp( type, BCRYPT_RSAPRIVATE_BLOB )) + { + return export_rsa_key( key, output, output_len, KEY_EXPORT_FLAG_PRIVATE, ret_len ); + } + else if (!wcscmp( type, BCRYPT_RSAFULLPRIVATE_BLOB )) + { + return export_rsa_key( key, output, output_len, KEY_EXPORT_FLAG_PRIVATE | KEY_EXPORT_FLAG_RSA_FULL, ret_len ); + } + else if (!wcscmp( type, BCRYPT_RSAPUBLIC_BLOB )) + { + return export_rsa_key( key, output, output_len, 0, ret_len ); + } + else if (!wcscmp( type, LEGACY_DSA_V2_PRIVATE_BLOB )) + { + return export_legacy_dsa_key( key, output, output_len, KEY_EXPORT_FLAG_PRIVATE, ret_len ); + } + else if (!wcscmp( type, LEGACY_DSA_V2_PUBLIC_BLOB )) + { + return export_legacy_dsa_key( key, output, output_len, 0, ret_len ); + } + + FIXME( "unsupported blob type %s\n", debugstr_w(type) ); + return STATUS_NOT_IMPLEMENTED; +} + +NTSTATUS WINAPI BCryptExportKey( BCRYPT_KEY_HANDLE export_key_handle, BCRYPT_KEY_HANDLE encrypt_key_handle, + const WCHAR *type, UCHAR *output, ULONG output_len, ULONG *size, ULONG flags ) +{ + const struct key *key = get_key_object( export_key_handle ), *encrypt_key = NULL; + + TRACE( "%p, %p, %s, %p, %lu, %p, %#lx\n", export_key_handle, encrypt_key_handle, debugstr_w(type), output, + output_len, size, flags ); + + if (!key) return STATUS_INVALID_HANDLE; + if (!type || !size) return STATUS_INVALID_PARAMETER; + if (encrypt_key_handle && !(encrypt_key = get_key_object( encrypt_key_handle ))) return STATUS_INVALID_HANDLE; + + if (is_symmetric_key( key )) return export_symmetric_key( key, encrypt_key, type, output, output_len, size ); + return export_asymmetric_key( key, type, output, output_len, size ); +} + +static NTSTATUS alloc_rsa_key( struct key *key, ULONG bitlen ) +{ + SYMCRYPT_RSA_PARAMS params; + + params.version = 1; + params.nBitsOfModulus = bitlen; + params.nPrimes = (key->flags & KEY_FLAG_PRIVATE) ? 2 : 0; + params.nPubExp = 1; + if (!(key->a.rsa.handle = SymCryptRsakeyAllocate( ¶ms, 0 ))) return STATUS_NO_MEMORY; + + key->a.bitlen = bitlen; + return STATUS_SUCCESS; +} + +static NTSTATUS import_rsa_key( enum alg_id alg, const UCHAR *input, ULONG input_len, struct key **ret_key ) +{ + const BCRYPT_RSAKEY_BLOB *blob = (const BCRYPT_RSAKEY_BLOB *)input; + const UCHAR *mod, *primes[2] = {}, *exp = (const UCHAR *)(blob + 1); + SIZE_T primes_sizes[2] = {}; + UINT64 exp64 = 0; + ULONG key_flags = 0, set_flags = SYMCRYPT_FLAG_KEY_NO_FIPS, size, i, num_primes = 0; + NTSTATUS status; + struct key *key = NULL; + + if (input_len < sizeof(*blob)) return STATUS_INVALID_PARAMETER; + + if (alg != ALG_ID_RSA && alg != ALG_ID_RSA_SIGN) return STATUS_NOT_SUPPORTED; + if (blob->Magic == BCRYPT_RSAPRIVATE_MAGIC || blob->Magic == BCRYPT_RSAFULLPRIVATE_MAGIC) + key_flags = KEY_FLAG_PRIVATE; + else if (blob->Magic != BCRYPT_RSAPUBLIC_MAGIC) return STATUS_NOT_SUPPORTED; + + if (blob->cbPublicExp > sizeof(exp64)) return NTE_BAD_DATA; + + size = sizeof(*blob) + blob->cbPublicExp + blob->cbModulus; + if (key_flags & KEY_FLAG_PRIVATE) size += blob->cbPrime1 + blob->cbPrime2; + if (blob->Magic == BCRYPT_RSAFULLPRIVATE_MAGIC) size += blob->cbPrime1 * 2 + blob->cbPrime2 + blob->cbModulus; + if (size != input_len) return STATUS_INVALID_PARAMETER; + + if ((status = alloc_key( alg, key_flags, &key )) || (status = alloc_rsa_key( key, blob->BitLength ))) + { + destroy_key( key ); + return status; + } + + if (alg == ALG_ID_RSA) set_flags |= SYMCRYPT_FLAG_RSAKEY_ENCRYPT | SYMCRYPT_FLAG_RSAKEY_SIGN; + else if (alg == ALG_ID_RSA_SIGN) set_flags |= SYMCRYPT_FLAG_RSAKEY_SIGN; + + mod = exp + blob->cbPublicExp; + for (i = 0; i < blob->cbPublicExp; i++) exp64 += exp[i] << ((blob->cbPublicExp - i - 1) * 8); + + if (key_flags & KEY_FLAG_PRIVATE) + { + num_primes = 2; + + primes[0] = mod + blob->cbModulus; + primes[1] = primes[0] + blob->cbPrime1; + + primes_sizes[0] = blob->cbPrime1; + primes_sizes[1] = blob->cbPrime2; + } + + if (SymCryptRsakeySetValue( mod, blob->cbModulus, &exp64, 1, primes, primes_sizes, num_primes, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, set_flags, key->a.rsa.handle )) + { + destroy_key( key ); + return STATUS_INTERNAL_ERROR; + } + + *ret_key = key; + return STATUS_SUCCESS; +} + +static NTSTATUS import_legacy_rsa_key( enum alg_id alg, const UCHAR *input, ULONG input_len, struct key **ret_key ) +{ + const BLOBHEADER *hdr = (const BLOBHEADER *)input; + const RSAPUBKEY *rsakey = (RSAPUBKEY *)(hdr + 1); + const UCHAR *mod, *primes[2] = {}; + SIZE_T primes_sizes[2] = {}; + UINT64 exp64; + ULONG key_size, key_flags = 0, set_flags = SYMCRYPT_FLAG_KEY_NO_FIPS, size, num_primes = 0; + NTSTATUS status; + struct key *key = NULL; + + if (alg != ALG_ID_RSA && alg != ALG_ID_RSA_SIGN) return STATUS_NOT_SUPPORTED; + if (input_len < sizeof(*hdr)) return STATUS_INVALID_PARAMETER; + + if (hdr->bType == PRIVATEKEYBLOB) key_flags |= KEY_FLAG_PRIVATE; + else if (hdr->bType != PUBLICKEYBLOB) return STATUS_NOT_SUPPORTED; + if (hdr->bVersion != 2 || (hdr->aiKeyAlg != CALG_RSA_KEYX && hdr->aiKeyAlg != CALG_RSA_SIGN)) + return STATUS_NOT_SUPPORTED; + + size = sizeof(*hdr) + sizeof(*rsakey); + if (input_len < size || rsakey->bitlen & 0xf) return STATUS_INVALID_PARAMETER; + + key_size = len_from_bitlen( rsakey->bitlen ); + size += key_size; + if (key_flags & KEY_FLAG_PRIVATE) size += key_size; /* prime1 + prime2 */ + if (input_len < size) return STATUS_INVALID_PARAMETER; + + if ((status = alloc_key( alg, key_flags, &key )) || (status = alloc_rsa_key( key, rsakey->bitlen ))) + { + destroy_key( key ); + return status; + } + + if (alg == ALG_ID_RSA) set_flags |= SYMCRYPT_FLAG_RSAKEY_ENCRYPT | SYMCRYPT_FLAG_RSAKEY_SIGN; + else if (alg == ALG_ID_RSA_SIGN) set_flags |= SYMCRYPT_FLAG_RSAKEY_SIGN; + + exp64 = rsakey->pubexp; + mod = (const UCHAR *)(rsakey + 1); + if (key_flags & KEY_FLAG_PRIVATE) + { + num_primes = 2; + + primes[0] = mod + key_size; + primes[1] = primes[0] + key_size / 2; + + primes_sizes[0] = key_size / 2; + primes_sizes[1] = key_size / 2; + } + + if (SymCryptRsakeySetValue( mod, key_size, &exp64, 1, primes, primes_sizes, num_primes, + SYMCRYPT_NUMBER_FORMAT_LSB_FIRST, set_flags, key->a.rsa.handle )) + { + destroy_key( key ); + return STATUS_INTERNAL_ERROR; + } + + *ret_key = key; + return STATUS_SUCCESS; +} + +static enum ecc_curve_id get_ecc_blob_curve( enum alg_id alg, const BCRYPT_ECCKEY_BLOB *blob, ULONG *size, + ULONG *flags ) +{ + enum ecc_curve_id curve_id = 0; + + *flags = 0; + *size = blob->cbKey; + + switch (alg) + { + case ALG_ID_ECDH: + if (blob->dwMagic == BCRYPT_ECDH_PRIVATE_GENERIC_MAGIC) *flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_ECDH_PUBLIC_GENERIC_MAGIC) return FALSE; + break; + + case ALG_ID_ECDH_P256: + if (blob->dwMagic == BCRYPT_ECDH_PRIVATE_P256_MAGIC) *flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_ECDH_PUBLIC_P256_MAGIC) return STATUS_INVALID_PARAMETER; + curve_id = ECC_CURVE_P256R1; + *size = 32; + break; + + case ALG_ID_ECDH_P384: + if (blob->dwMagic == BCRYPT_ECDH_PRIVATE_P384_MAGIC) *flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_ECDH_PUBLIC_P384_MAGIC) return STATUS_INVALID_PARAMETER; + curve_id = ECC_CURVE_P384R1; + *size = 48; + break; + + case ALG_ID_ECDH_P521: + if (blob->dwMagic == BCRYPT_ECDH_PRIVATE_P521_MAGIC) *flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_ECDH_PUBLIC_P521_MAGIC) return STATUS_INVALID_PARAMETER; + curve_id = ECC_CURVE_P521R1; + *size = 66; + break; + + case ALG_ID_ECDSA: + if (blob->dwMagic == BCRYPT_ECDSA_PRIVATE_GENERIC_MAGIC) *flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_ECDSA_PUBLIC_GENERIC_MAGIC) return STATUS_INVALID_PARAMETER; + break; + + case ALG_ID_ECDSA_P256: + if (blob->dwMagic == BCRYPT_ECDSA_PRIVATE_P256_MAGIC) *flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_ECDSA_PUBLIC_P256_MAGIC) return STATUS_INVALID_PARAMETER; + curve_id = ECC_CURVE_P256R1; + *size = 32; + break; + + case ALG_ID_ECDSA_P384: + if (blob->dwMagic == BCRYPT_ECDSA_PRIVATE_P384_MAGIC) *flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_ECDSA_PUBLIC_P384_MAGIC) return STATUS_INVALID_PARAMETER; + curve_id = ECC_CURVE_P384R1; + *size = 48; + break; + + case ALG_ID_ECDSA_P521: + if (blob->dwMagic == BCRYPT_ECDSA_PRIVATE_P521_MAGIC) *flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_ECDSA_PUBLIC_P521_MAGIC) return STATUS_INVALID_PARAMETER; + curve_id = ECC_CURVE_P521R1; + *size = 66; + break; + + default: + ERR( "unhandled algorithm %u\n", alg ); + return 0; + } + return curve_id; +} + +static UINT32 get_ecc_import_flags( enum alg_id alg ) +{ + UINT32 flags = SYMCRYPT_FLAG_KEY_NO_FIPS | SYMCRYPT_FLAG_KEY_MINIMAL_VALIDATION; + + switch (alg) + { + case ALG_ID_ECDH: + case ALG_ID_ECDH_P256: + case ALG_ID_ECDH_P384: + case ALG_ID_ECDH_P521: + flags |= SYMCRYPT_FLAG_ECKEY_ECDH; + break; + + case ALG_ID_ECDSA: + case ALG_ID_ECDSA_P256: + case ALG_ID_ECDSA_P384: + case ALG_ID_ECDSA_P521: + flags |= SYMCRYPT_FLAG_ECKEY_ECDSA; + break; + + default: + FIXME( "unhandled algorithm %u\n", alg ); + break; + } + return flags; +} + +static SYMCRYPT_ECURVE *alloc_ecc_curve( enum ecc_curve_id curve_id ) +{ + const SYMCRYPT_ECURVE_PARAMS *params; + + switch (curve_id) + { + case ECC_CURVE_25519: + params = SymCryptEcurveParamsCurve25519; + break; + case ECC_CURVE_P256R1: + params = SymCryptEcurveParamsNistP256; + break; + case ECC_CURVE_P384R1: + params = SymCryptEcurveParamsNistP384; + break; + case ECC_CURVE_P521R1: + params = SymCryptEcurveParamsNistP521; + break; + default: + FIXME( "unhandled curve %u\n", curve_id ); + return NULL; + } + return SymCryptEcurveAllocate( params, 0 ); +} + +static ULONG curve_strength( enum ecc_curve_id curve_id ) +{ + switch (curve_id) + { + case ECC_CURVE_25519: return 253; + case ECC_CURVE_P256R1: return 256; + case ECC_CURVE_P384R1: return 384; + case ECC_CURVE_P521R1: return 521; + default: + FIXME( "unsupported curve %u\n", curve_id ); + return 0; + } +} + +static NTSTATUS alloc_ecc_key( struct key *key, enum ecc_curve_id curve_id ) +{ + if (!(key->a.ecc.curve = alloc_ecc_curve( curve_id ))) return STATUS_NO_MEMORY; + if (!(key->a.ecc.handle = SymCryptEckeyAllocate( key->a.ecc.curve ))) return STATUS_NO_MEMORY; + key->a.ecc.curve_id = curve_id; + key->a.bitlen = curve_strength( curve_id ); + return STATUS_SUCCESS; +} + +static NTSTATUS import_ecc_key( enum alg_id alg, enum ecc_curve_id curve, const UCHAR *input, ULONG input_len, + struct key **ret_key ) +{ + const BCRYPT_ECCKEY_BLOB *blob = (const BCRYPT_ECCKEY_BLOB *)input; + UINT32 pub_size, priv_size = 0, set_flags = get_ecc_import_flags( alg ); + ULONG size, key_size, key_flags = 0; + const UCHAR *pub = (UCHAR *)(blob + 1), *priv = NULL; + enum ecc_curve_id blob_curve; + NTSTATUS status; + struct key *key = NULL; + + if (input_len < sizeof(*blob)) return STATUS_INVALID_PARAMETER; + if (!(blob_curve = get_ecc_blob_curve( alg, blob, &key_size, &key_flags ))) blob_curve = curve; + if (blob_curve != curve) return STATUS_INVALID_PARAMETER; + + size = sizeof(*blob) + blob->cbKey * 2; + if (key_flags & KEY_FLAG_PRIVATE) + { + priv = pub + blob->cbKey * 2; + size += blob->cbKey; + } + if (blob->cbKey != key_size || input_len != size) return STATUS_INVALID_PARAMETER; + + if ((status = alloc_key( alg, key_flags, &key )) || (status = alloc_ecc_key( key, curve ))) + { + destroy_key( key ); + return status; + } + + pub_size = SymCryptEckeySizeofPublicKey( key->a.ecc.handle, SYMCRYPT_ECPOINT_FORMAT_XY ); + if (key_flags & KEY_FLAG_PRIVATE) priv_size = SymCryptEckeySizeofPrivateKey( key->a.ecc.handle ); + + if (SymCryptEckeySetValue( priv, priv_size, pub, pub_size, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, + SYMCRYPT_ECPOINT_FORMAT_XY, set_flags, key->a.ecc.handle )) + { + destroy_key( key ); + return STATUS_INTERNAL_ERROR; + } + + *ret_key = key; + return STATUS_SUCCESS; +} - case ALG_ID_ECDH_P384: - bitlen = 384; - curve = ECC_CURVE_P384R1; - magic = BCRYPT_ECDH_PUBLIC_P384_MAGIC; - magic_priv = BCRYPT_ECDH_PRIVATE_P384_MAGIC; - break; +static NTSTATUS alloc_dsa_key( struct key *key, ULONG bitlen ) +{ + if (!(key->a.dsa.group = SymCryptDlgroupAllocate( bitlen, 0 ))) return STATUS_NO_MEMORY; + if (!(key->a.dsa.handle = SymCryptDlkeyAllocate( key->a.dsa.group ))) return STATUS_NO_MEMORY; + key->a.bitlen = bitlen; + return STATUS_SUCCESS; +} - case ALG_ID_ECDH_P521: - bitlen = 521; - curve = ECC_CURVE_P521R1; - magic = BCRYPT_ECDH_PUBLIC_P521_MAGIC; - magic_priv = BCRYPT_ECDH_PRIVATE_P521_MAGIC; - break; +static NTSTATUS import_dsa_key( enum alg_id alg, const UCHAR *input, ULONG input_len, struct key **ret_key ) +{ + const BCRYPT_DSA_KEY_BLOB *blob = (const BCRYPT_DSA_KEY_BLOB *)input; + ULONG size, count, priv_size = 0, key_flags = 0, set_flags = SYMCRYPT_FLAG_KEY_NO_FIPS | SYMCRYPT_FLAG_DLKEY_DSA; + const UCHAR *prime, *generator, *pub, *priv = NULL; + struct key *key = NULL; + NTSTATUS status; - case ALG_ID_ECDSA_P256: - bitlen = 256; - curve = ECC_CURVE_P256R1; - magic = BCRYPT_ECDSA_PUBLIC_P256_MAGIC; - magic_priv = BCRYPT_ECDSA_PRIVATE_P256_MAGIC; - break; + if (input_len < sizeof(*blob)) return STATUS_INVALID_PARAMETER; - case ALG_ID_ECDSA_P384: - bitlen = 384; - curve = ECC_CURVE_P384R1; - magic = BCRYPT_ECDSA_PUBLIC_P384_MAGIC; - magic_priv = BCRYPT_ECDSA_PRIVATE_P384_MAGIC; - break; + if (alg != ALG_ID_DSA) return STATUS_NOT_SUPPORTED; + if (blob->dwMagic == BCRYPT_DSA_PRIVATE_MAGIC) key_flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_DSA_PUBLIC_MAGIC) return STATUS_NOT_SUPPORTED; - case ALG_ID_ECDSA_P521: - bitlen = 521; - curve = ECC_CURVE_P521R1; - magic = BCRYPT_ECDSA_PUBLIC_P521_MAGIC; - magic_priv = BCRYPT_ECDSA_PRIVATE_P521_MAGIC; - break; + size = sizeof(*blob) + blob->cbKey * 3; + if (key_flags & KEY_FLAG_PRIVATE) size += 20; + if (input_len != size) return STATUS_INVALID_PARAMETER; - default: - FIXME( "algorithm %u does not yet support importing blob of type %s\n", alg->id, debugstr_w(type) ); - return STATUS_NOT_SUPPORTED; - } + if ((status = alloc_key( alg, key_flags, &key )) || (status = alloc_dsa_key( key, blob->cbKey * 8 ))) + { + destroy_key( key ); + return status; + } - if (ecc_blob->dwMagic != magic && ecc_blob->dwMagic != magic_priv) return STATUS_INVALID_PARAMETER; - if (ecc_blob->cbKey != len_from_bitlen( bitlen ) || input_len < sizeof(*ecc_blob) + ecc_blob->cbKey * 2) - return STATUS_INVALID_PARAMETER; + prime = (const UCHAR *)(blob + 1); + generator = prime + blob->cbKey; + count = *(UINT32 *)blob->Count; - if ((status = key_asymmetric_create( alg->id, curve, bitlen, &key ))) return status; - params.key = key; - params.flags = KEY_IMPORT_FLAG_PUBLIC; - params.buf = input; - params.len = input_len; - if ((status = UNIX_CALL( key_asymmetric_import, ¶ms ))) - { - key_destroy( key ); - return status; - } - } - else if (!wcscmp( type, BCRYPT_ECCPRIVATE_BLOB )) + if (SymCryptDlgroupSetValue( prime, blob->cbKey, blob->q, sizeof(blob->q), generator, blob->cbKey, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, NULL, blob->Seed, sizeof(blob->Seed), + count, SYMCRYPT_DLGROUP_FIPS_NONE, key->a.dsa.group )) { - BCRYPT_ECCKEY_BLOB *ecc_blob = (BCRYPT_ECCKEY_BLOB *)input; - DWORD bitlen, magic; - enum ecc_curve_id curve; + destroy_key( key ); + return STATUS_INTERNAL_ERROR; + } - if (input_len < sizeof(*ecc_blob)) return STATUS_INVALID_PARAMETER; + pub = generator + blob->cbKey; + if (key_flags & KEY_FLAG_PRIVATE) + { + priv = pub + blob->cbKey; + priv_size = 20; + } - switch (alg->id) - { - case ALG_ID_ECDH: - bitlen = ecc_blob->cbKey * 8; - curve = alg->curve_id; - magic = BCRYPT_ECDH_PRIVATE_GENERIC_MAGIC; - break; + if (alg == ALG_ID_DH) set_flags |= SYMCRYPT_FLAG_DLKEY_DH; + else if (alg == ALG_ID_DSA) set_flags |= SYMCRYPT_FLAG_DLKEY_DSA; - case ALG_ID_ECDH_P256: - bitlen = 256; - curve = ECC_CURVE_P256R1; - magic = BCRYPT_ECDH_PRIVATE_P256_MAGIC; - break; + if (SymCryptDlkeySetValue( priv, priv_size, pub, blob->cbKey, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, set_flags, + key->a.dsa.handle )) + { + destroy_key( key ); + return STATUS_INTERNAL_ERROR; + } - case ALG_ID_ECDH_P384: - bitlen = 384; - curve = ECC_CURVE_P384R1; - magic = BCRYPT_ECDH_PRIVATE_P384_MAGIC; - break; + memcpy( key->a.dsa.seed, blob->Seed, sizeof(blob->Seed) ); + memcpy( &key->a.dsa.count, blob->Count, sizeof(blob->Count) ); - case ALG_ID_ECDH_P521: - bitlen = 521; - curve = ECC_CURVE_P521R1; - magic = BCRYPT_ECDH_PRIVATE_P521_MAGIC; - break; + *ret_key = key; + return STATUS_SUCCESS; +} - case ALG_ID_ECDSA_P256: - bitlen = 256; - curve = ECC_CURVE_P256R1; - magic = BCRYPT_ECDSA_PRIVATE_P256_MAGIC; - break; +static NTSTATUS import_legacy_dsa_key( enum alg_id alg, const UCHAR *input, ULONG input_len, struct key **ret_key ) +{ + const BLOBHEADER *hdr = (const BLOBHEADER *)input; + const DSSPUBKEY *dsskey = (DSSPUBKEY *)(hdr + 1); + UINT32 priv_size = 0, pub_size = 0, set_flags = SYMCRYPT_FLAG_KEY_NO_FIPS | SYMCRYPT_FLAG_DLKEY_DSA; + const UCHAR *prime_p, *prime_q, *generator, *pub = NULL, *priv = NULL, *seed, *count; + ULONG size, key_size, key_flags = 0; + struct key *key = NULL; + NTSTATUS status; - case ALG_ID_ECDSA_P384: - bitlen = 384; - curve = ECC_CURVE_P384R1; - magic = BCRYPT_ECDSA_PRIVATE_P384_MAGIC; - break; + if (alg != ALG_ID_DSA) return STATUS_NOT_SUPPORTED; + if (input_len < sizeof(*hdr)) return STATUS_INVALID_PARAMETER; - case ALG_ID_ECDSA_P521: - bitlen = 521; - curve = ECC_CURVE_P521R1; - magic = BCRYPT_ECDSA_PRIVATE_P521_MAGIC; - break; + if (hdr->bType == PRIVATEKEYBLOB) key_flags |= KEY_FLAG_PRIVATE; + else if (hdr->bType != PUBLICKEYBLOB) return STATUS_NOT_SUPPORTED; + if (hdr->bVersion != 2 || hdr->aiKeyAlg != CALG_DSS_SIGN) return STATUS_NOT_SUPPORTED; - default: - FIXME( "algorithm %u does not yet support importing blob of type %s\n", alg->id, debugstr_w(type) ); - return STATUS_NOT_SUPPORTED; - } + size = sizeof(*hdr) + sizeof(*dsskey); + if (input_len < size || dsskey->bitlen & 0xf) return STATUS_INVALID_PARAMETER; - if (ecc_blob->dwMagic != magic) return STATUS_INVALID_PARAMETER; - if (ecc_blob->cbKey != len_from_bitlen( bitlen ) || input_len < sizeof(*ecc_blob) + ecc_blob->cbKey * 3) - return STATUS_INVALID_PARAMETER; + key_size = len_from_bitlen( dsskey->bitlen ); + size += key_size * 2 + 20 /* q */ + 20 /* x or y */ + 20 /* seed */ + 4 /* count */; + if (input_len != size) return STATUS_INVALID_PARAMETER; - if ((status = key_asymmetric_create( alg->id, curve, bitlen, &key ))) return status; - params.key = key; - params.flags = 0; - params.buf = input; - params.len = input_len; - if ((status = UNIX_CALL( key_asymmetric_import, ¶ms ))) - { - key_destroy( key ); - return status; - } - } - else if (!wcscmp( type, BCRYPT_RSAPUBLIC_BLOB )) + if ((status = alloc_key( alg, key_flags, &key )) || (status = alloc_dsa_key( key, dsskey->bitlen ))) { - BCRYPT_RSAKEY_BLOB *rsa_blob = (BCRYPT_RSAKEY_BLOB *)input; - - if (input_len < sizeof(*rsa_blob)) return STATUS_INVALID_PARAMETER; - if ((alg->id != ALG_ID_RSA && alg->id != ALG_ID_RSA_SIGN) || rsa_blob->Magic != BCRYPT_RSAPUBLIC_MAGIC) - return STATUS_NOT_SUPPORTED; + destroy_key( key ); + return status; + } - size = sizeof(*rsa_blob) + rsa_blob->cbPublicExp + rsa_blob->cbModulus; - if (size != input_len) return NTE_BAD_DATA; + prime_p = (const UCHAR *)(dsskey + 1); + prime_q = prime_p + key_size; + generator = prime_q + 20; + seed = generator + key_size + 20; + count = seed + 20; - if ((status = key_asymmetric_create( alg->id, 0, rsa_blob->BitLength, &key ))) return status; - params.key = key; - params.flags = KEY_IMPORT_FLAG_PUBLIC; - params.buf = input; - params.len = input_len; - if ((status = UNIX_CALL( key_asymmetric_import, ¶ms ))) - { - key_destroy( key ); - return status; - } - } - else if (!wcscmp( type, BCRYPT_RSAPRIVATE_BLOB ) || !wcscmp( type, BCRYPT_RSAFULLPRIVATE_BLOB )) + if (SymCryptDlgroupSetValue( prime_p, key_size, prime_q, 20, generator, key_size, + SYMCRYPT_NUMBER_FORMAT_LSB_FIRST, NULL, seed, 20, *(UINT32 *)count, + SYMCRYPT_DLGROUP_FIPS_NONE, key->a.dsa.group )) { - BCRYPT_RSAKEY_BLOB *rsa_blob = (BCRYPT_RSAKEY_BLOB *)input; - - if (input_len < sizeof(*rsa_blob)) return STATUS_INVALID_PARAMETER; - if (alg->id != ALG_ID_RSA || (rsa_blob->Magic != BCRYPT_RSAPRIVATE_MAGIC && - rsa_blob->Magic != BCRYPT_RSAFULLPRIVATE_MAGIC)) return STATUS_NOT_SUPPORTED; + destroy_key( key ); + return STATUS_INTERNAL_ERROR; + } - if ((status = key_asymmetric_create( alg->id, 0, rsa_blob->BitLength, &key ))) return status; - params.key = key; - params.flags = 0; - params.buf = input; - params.len = input_len; - if ((status = UNIX_CALL( key_asymmetric_import, ¶ms ))) - { - key_destroy( key ); - return status; - } + if (key_flags & KEY_FLAG_PRIVATE) + { + priv = generator + key_size; + priv_size = 20; } - else if (!wcscmp( type, LEGACY_RSAPRIVATE_BLOB )) + else { - BCRYPT_RSAKEY_BLOB *rsa_blob; - - status = convert_legacy_rsaprivate_blob( alg, &rsa_blob, &input_len, input, input_len ); - if (status != STATUS_SUCCESS) - return status; - - status = key_import_pair( alg, BCRYPT_RSAPRIVATE_BLOB, ret_key, (UCHAR *)rsa_blob, input_len ); - SecureZeroMemory( rsa_blob, input_len ); - free( rsa_blob ); - return status; + pub = generator + key_size; + pub_size = 20; } - else if (!wcscmp( type, BCRYPT_DSA_PUBLIC_BLOB )) + + if (SymCryptDlkeySetValue( priv, priv_size, pub, pub_size, SYMCRYPT_NUMBER_FORMAT_LSB_FIRST, set_flags, + key->a.dsa.handle )) { - BCRYPT_DSA_KEY_BLOB *dsa_blob = (BCRYPT_DSA_KEY_BLOB *)input; + destroy_key( key ); + return STATUS_INTERNAL_ERROR; + } - if (input_len < sizeof(*dsa_blob)) return STATUS_INVALID_PARAMETER; - if (alg->id != ALG_ID_DSA || dsa_blob->dwMagic != BCRYPT_DSA_PUBLIC_MAGIC) - return STATUS_NOT_SUPPORTED; + memcpy( key->a.dsa.seed, seed, 20 ); - if ((status = key_asymmetric_create( alg->id, 0, dsa_blob->cbKey * 8, &key ))) return status; - params.key = key; - params.flags = KEY_IMPORT_FLAG_PUBLIC; - params.buf = input; - params.len = input_len; - if ((status = UNIX_CALL( key_asymmetric_import, ¶ms ))) - { - key_destroy( key ); - return status; - } - } - else if (!wcscmp( type, LEGACY_DSA_V2_PRIVATE_BLOB )) - { - BLOBHEADER *hdr = (BLOBHEADER *)input; - DSSPUBKEY *pubkey; + *ret_key = key; + return STATUS_SUCCESS; +} - if (input_len < sizeof(*hdr)) return STATUS_INVALID_PARAMETER; +static NTSTATUS import_dh_key( enum alg_id alg, const UCHAR *input, ULONG input_len, struct key **ret_key ) +{ + const BCRYPT_DH_KEY_BLOB *blob = (const BCRYPT_DH_KEY_BLOB *)input; + ULONG size, priv_size = 0, key_flags = 0, set_flags = SYMCRYPT_FLAG_KEY_NO_FIPS | SYMCRYPT_FLAG_DLKEY_DH; + const UCHAR *modulus, *generator, *pub, *priv = NULL; + struct key *key = NULL; + NTSTATUS status; - if (hdr->bType != PRIVATEKEYBLOB && hdr->bVersion != 2 && hdr->aiKeyAlg != CALG_DSS_SIGN) - { - FIXME( "blob type %u version %u alg id %u not supported\n", hdr->bType, hdr->bVersion, hdr->aiKeyAlg ); - return STATUS_NOT_SUPPORTED; - } - if (alg->id != ALG_ID_DSA) - { - FIXME( "algorithm %u does not support importing blob of type %s\n", alg->id, debugstr_w(type) ); - return STATUS_NOT_SUPPORTED; - } + if (input_len < sizeof(*blob)) return STATUS_INVALID_PARAMETER; - if (input_len < sizeof(*hdr) + sizeof(*pubkey)) return STATUS_INVALID_PARAMETER; - pubkey = (DSSPUBKEY *)(hdr + 1); - if (pubkey->magic != MAGIC_DSS2) return STATUS_NOT_SUPPORTED; + if (alg != ALG_ID_DH) return STATUS_NOT_SUPPORTED; + if (blob->dwMagic == BCRYPT_DH_PRIVATE_MAGIC) key_flags = KEY_FLAG_PRIVATE; + else if (blob->dwMagic != BCRYPT_DH_PUBLIC_MAGIC) return STATUS_NOT_SUPPORTED; - if (input_len < sizeof(*hdr) + sizeof(*pubkey) + len_from_bitlen( pubkey->bitlen ) * 2 + 40 + sizeof(DSSSEED)) - return STATUS_INVALID_PARAMETER; + size = sizeof(*blob) + blob->cbKey * 3; + if (key_flags & KEY_FLAG_PRIVATE) size += blob->cbKey; + if (input_len != size) return STATUS_INVALID_PARAMETER; - if ((status = key_asymmetric_create( alg->id, 0, pubkey->bitlen, &key ))) return status; - key->u.a.flags |= KEY_FLAG_LEGACY_DSA_V2; - params.key = key; - params.flags = 0; - params.buf = input; - params.len = input_len; - if ((status = UNIX_CALL( key_asymmetric_import, ¶ms ))) - { - key_destroy( key ); - return status; - } + if ((status = alloc_key( alg, key_flags, &key )) || (status = alloc_dsa_key( key, blob->cbKey * 8 ))) + { + destroy_key( key ); + return status; } - else if (!wcscmp( type, LEGACY_DSA_V2_PUBLIC_BLOB )) /* not supported on native */ + + modulus = (const UCHAR *)(blob + 1); + generator = modulus + blob->cbKey; + + if (SymCryptDlgroupSetValue( modulus, blob->cbKey, NULL, 0, generator, blob->cbKey, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, NULL, NULL, 0, 0, + SYMCRYPT_DLGROUP_FIPS_NONE, key->a.dsa.group )) { - BLOBHEADER *hdr = (BLOBHEADER *)input; - DSSPUBKEY *pubkey; + destroy_key( key ); + return STATUS_INTERNAL_ERROR; + } - if (alg->id != ALG_ID_DSA) return STATUS_NOT_SUPPORTED; - if (input_len < sizeof(*hdr)) return STATUS_INVALID_PARAMETER; + pub = generator + blob->cbKey; + if (key_flags & KEY_FLAG_PRIVATE) + { + priv = pub + blob->cbKey; + priv_size = blob->cbKey; + } - if (hdr->bType != PUBLICKEYBLOB && hdr->bVersion != 2 && hdr->aiKeyAlg != CALG_DSS_SIGN) - { - FIXME( "blob type %u version %u alg id %u not supported\n", hdr->bType, hdr->bVersion, hdr->aiKeyAlg ); - return STATUS_NOT_SUPPORTED; - } + if (SymCryptDlkeySetValue( priv, priv_size, pub, blob->cbKey, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, set_flags, + key->a.dsa.handle )) + { + destroy_key( key ); + return STATUS_INTERNAL_ERROR; + } - if (input_len < sizeof(*hdr) + sizeof(*pubkey)) return STATUS_INVALID_PARAMETER; - pubkey = (DSSPUBKEY *)(hdr + 1); - if (pubkey->magic != MAGIC_DSS1) return STATUS_NOT_SUPPORTED; + *ret_key = key; + return STATUS_SUCCESS; +} - size = sizeof(*hdr) + sizeof(*pubkey) + len_from_bitlen( pubkey->bitlen ) * 3 + 20 + sizeof(DSSSEED); - if (input_len < size) return STATUS_INVALID_PARAMETER; +static NTSTATUS import_key_pair( const struct algorithm *alg, const WCHAR *type, const UCHAR *input, ULONG input_len, + struct key **ret_key ) +{ + struct key *key; + NTSTATUS status; - if ((status = key_asymmetric_create( alg->id, 0, pubkey->bitlen, &key ))) return status; - key->u.a.flags |= KEY_FLAG_LEGACY_DSA_V2; - params.key = key; - params.flags = KEY_IMPORT_FLAG_PUBLIC; - params.buf = input; - params.len = input_len; - if ((status = UNIX_CALL( key_asymmetric_import, ¶ms ))) - { - key_destroy( key ); - return status; - } + if (!wcscmp( type, BCRYPT_ECCPUBLIC_BLOB ) || !wcscmp( type, BCRYPT_ECCPRIVATE_BLOB )) + { + status = import_ecc_key( alg->id, alg->curve_id, input, input_len, &key ); } - else if (!wcscmp( type, BCRYPT_DH_PRIVATE_BLOB )) + else if (!wcscmp( type, BCRYPT_RSAPUBLIC_BLOB ) || !wcscmp( type, BCRYPT_RSAPRIVATE_BLOB ) || + !wcscmp( type, BCRYPT_RSAFULLPRIVATE_BLOB )) { - BCRYPT_DH_KEY_BLOB *dh_blob = (BCRYPT_DH_KEY_BLOB *)input; - - if (input_len != sizeof(*dh_blob) + dh_blob->cbKey * 4) return STATUS_INVALID_PARAMETER; - if (alg->id != ALG_ID_DH || dh_blob->dwMagic != BCRYPT_DH_PRIVATE_MAGIC) - return STATUS_NOT_SUPPORTED; - - if ((status = key_asymmetric_create( alg->id, 0, dh_blob->cbKey * 8, &key ))) return status; - params.key = key; - params.flags = 0; - params.buf = input; - params.len = input_len; - if ((status = UNIX_CALL( key_asymmetric_import, ¶ms ))) - { - key_destroy( key ); - return status; - } + status = import_rsa_key( alg->id, input, input_len, &key ); } - else if (!wcscmp( type, BCRYPT_DH_PUBLIC_BLOB )) + else if (!wcscmp( type, LEGACY_RSAPRIVATE_BLOB )) { - BCRYPT_DH_KEY_BLOB *dh_blob = (BCRYPT_DH_KEY_BLOB *)input; - - if (input_len != sizeof(*dh_blob) + dh_blob->cbKey * 3) return STATUS_INVALID_PARAMETER; - if (alg->id != ALG_ID_DH || dh_blob->dwMagic != BCRYPT_DH_PUBLIC_MAGIC) - return STATUS_NOT_SUPPORTED; - - if ((status = key_asymmetric_create( alg->id, 0, dh_blob->cbKey * 8, &key ))) return status; - params.key = key; - params.flags = KEY_IMPORT_FLAG_PUBLIC; - params.buf = input; - params.len = input_len; - if ((status = UNIX_CALL( key_asymmetric_import, ¶ms ))) - { - key_destroy( key ); - return status; - } + status = import_legacy_rsa_key( alg->id, input, input_len, &key ); + } + else if (!wcscmp( type, BCRYPT_DSA_PUBLIC_BLOB ) || !wcscmp( type, BCRYPT_DSA_PRIVATE_BLOB )) + { + status = import_dsa_key( alg->id, input, input_len, &key ); + } + else if (!wcscmp( type, BCRYPT_DH_PUBLIC_BLOB ) || !wcscmp( type, BCRYPT_DH_PRIVATE_BLOB )) + { + status = import_dh_key( alg->id, input, input_len, &key ); + } + else if (!wcscmp( type, LEGACY_DSA_V2_PRIVATE_BLOB ) || !wcscmp( type, LEGACY_DSA_V2_PUBLIC_BLOB )) + { + status = import_legacy_dsa_key( alg->id, input, input_len, &key ); } else { - FIXME( "unsupported key type %s\n", debugstr_w(type) ); + FIXME( "unsupported blob type %s\n", debugstr_w(type) ); return STATUS_NOT_SUPPORTED; } if (!status) { - key->u.a.flags |= KEY_FLAG_FINALIZED; + key->flags |= KEY_FLAG_FINALIZED; *ret_key = key; } return status; } -NTSTATUS WINAPI BCryptGenerateSymmetricKey( BCRYPT_ALG_HANDLE handle, BCRYPT_KEY_HANDLE *ret_handle, - UCHAR *object, ULONG object_len, UCHAR *secret, ULONG secret_len, - ULONG flags ) +static const WCHAR *resolve_blob_type( const WCHAR *type, const UCHAR *input, ULONG input_len ) { - struct algorithm *alg = get_alg_object( handle ); - NTSTATUS status; + const BCRYPT_KEY_BLOB *blob = (const BCRYPT_KEY_BLOB *)input; - TRACE( "%p, %p, %p, %lu, %p, %lu, %#lx\n", handle, ret_handle, object, object_len, secret, secret_len, flags ); - if (object) FIXME( "ignoring object buffer\n" ); - if (!__wine_unixlib_handle) + if (!type) return NULL; + if (wcscmp( type, BCRYPT_PUBLIC_KEY_BLOB )) return type; + if (input_len < sizeof(*blob)) return NULL; + + switch (blob->Magic) { - ERR( "no encryption support\n" ); - return STATUS_NOT_IMPLEMENTED; - } + case BCRYPT_ECDH_PUBLIC_P256_MAGIC: + case BCRYPT_ECDH_PUBLIC_P384_MAGIC: + case BCRYPT_ECDH_PUBLIC_P521_MAGIC: + case BCRYPT_ECDSA_PUBLIC_P256_MAGIC: + case BCRYPT_ECDSA_PUBLIC_P384_MAGIC: + case BCRYPT_ECDSA_PUBLIC_P521_MAGIC: + return BCRYPT_ECCPUBLIC_BLOB; - if (!alg) return STATUS_INVALID_HANDLE; - if ((status = key_symmetric_generate( alg, ret_handle, secret, secret_len ))) return status; - TRACE( "returning handle %p\n", *ret_handle ); - return STATUS_SUCCESS; + case BCRYPT_RSAPUBLIC_MAGIC: + return BCRYPT_RSAPUBLIC_BLOB; + + case BCRYPT_DSA_PUBLIC_MAGIC: + return BCRYPT_DSA_PUBLIC_BLOB; + + default: + FIXME( "unsupported key magic %#lx\n", blob->Magic ); + return NULL; + } } -NTSTATUS WINAPI BCryptGenerateKeyPair( BCRYPT_ALG_HANDLE handle, BCRYPT_KEY_HANDLE *ret_handle, ULONG key_len, - ULONG flags ) +NTSTATUS WINAPI BCryptImportKeyPair( BCRYPT_ALG_HANDLE handle, BCRYPT_KEY_HANDLE decrypt_key_handle, const WCHAR *type, + BCRYPT_KEY_HANDLE *ret_handle, UCHAR *input, ULONG input_len, ULONG flags ) { - struct algorithm *alg = get_alg_object( handle ); + const struct algorithm *alg = get_alg_object( handle ); struct key *key; NTSTATUS status; - TRACE( "%p, %p, %lu, %#lx\n", handle, ret_handle, key_len, flags ); + TRACE( "%p, %p, %s, %p, %p, %lu, %#lx\n", handle, decrypt_key_handle, debugstr_w(type), ret_handle, input, + input_len, flags ); if (!alg) return STATUS_INVALID_HANDLE; - if (!ret_handle) return STATUS_INVALID_PARAMETER; + if (!ret_handle || !input || !(type = resolve_blob_type( type, input, input_len ))) return STATUS_INVALID_PARAMETER; + if (decrypt_key_handle) + { + FIXME( "decryption of key not yet supported\n" ); + return STATUS_NOT_IMPLEMENTED; + } - if ((status = key_asymmetric_create( alg->id, alg->curve_id, key_len, &key ))) return status; + if ((status = import_key_pair( alg, type, input, input_len, &key ))) return status; *ret_handle = key; + TRACE( "returning handle %p\n", *ret_handle ); return STATUS_SUCCESS; } -NTSTATUS WINAPI BCryptFinalizeKeyPair( BCRYPT_KEY_HANDLE handle, ULONG flags ) +static NTSTATUS generate_key_pair( const struct algorithm *alg, ULONG bitlen, ULONG flags, struct key **ret_key ) { - struct key *key = get_key_object( handle ); - NTSTATUS ret; + NTSTATUS status; + struct key *key; - TRACE( "%p, %#lx\n", key, flags ); + if ((status = alloc_key( alg->id, KEY_FLAG_PRIVATE, &key ))) return status; - if (!key || key->u.a.flags & KEY_FLAG_FINALIZED) return STATUS_INVALID_HANDLE; + switch (alg->id) + { + case ALG_ID_DH: + if (bitlen < 512) + { + destroy_key( key ); + return STATUS_INVALID_PARAMETER; + } + /* fall through */ + case ALG_ID_DSA: + case ALG_ID_RSA: + case ALG_ID_RSA_SIGN: + key->a.bitlen = bitlen; + break; - if (!(ret = UNIX_CALL( key_asymmetric_generate, key ))) key->u.a.flags |= KEY_FLAG_FINALIZED; - return ret; + case ALG_ID_ECDH: + case ALG_ID_ECDH_P256: + case ALG_ID_ECDH_P384: + case ALG_ID_ECDH_P521: + case ALG_ID_ECDSA: + case ALG_ID_ECDSA_P256: + case ALG_ID_ECDSA_P384: + case ALG_ID_ECDSA_P521: + if (bitlen && bitlen != curve_strength( alg->curve_id )) + { + destroy_key( key ); + return STATUS_INVALID_PARAMETER; + } + key->a.ecc.curve_id = alg->curve_id; + key->a.bitlen = curve_strength( key->a.ecc.curve_id ); + break; + + default: + FIXME( "unhandled algorithm %u\n", alg->id ); + destroy_key( key ); + return STATUS_NOT_SUPPORTED; + } + + *ret_key = key; + return STATUS_SUCCESS; } -NTSTATUS WINAPI BCryptImportKey( BCRYPT_ALG_HANDLE handle, BCRYPT_KEY_HANDLE decrypt_key_handle, const WCHAR *type, - BCRYPT_KEY_HANDLE *ret_handle, UCHAR *object, ULONG object_len, UCHAR *input, - ULONG input_len, ULONG flags ) +NTSTATUS WINAPI BCryptGenerateKeyPair( BCRYPT_ALG_HANDLE handle, BCRYPT_KEY_HANDLE *ret_handle, ULONG bitlen, + ULONG flags ) { - struct algorithm *alg = get_alg_object( handle ); - struct key *decrypt_key = NULL; + const struct algorithm *alg = get_alg_object( handle ); + struct key *key; NTSTATUS status; - TRACE( "%p, %p, %s, %p, %p, %lu, %p, %lu, %#lx\n", handle, decrypt_key_handle, debugstr_w(type), ret_handle, - object, object_len, input, input_len, flags ); + TRACE( "%p, %p, %lu, %#lx\n", handle, ret_handle, bitlen, flags ); if (!alg) return STATUS_INVALID_HANDLE; - if (!ret_handle || !type || !input) return STATUS_INVALID_PARAMETER; - if (decrypt_key_handle && !(decrypt_key = get_key_object( decrypt_key_handle ))) return STATUS_INVALID_HANDLE; + if (!ret_handle) return STATUS_INVALID_PARAMETER; + + if ((status = generate_key_pair( alg, bitlen, flags, &key ))) return status; + + *ret_handle = key; + TRACE( "returning handle %p\n", *ret_handle ); + return STATUS_SUCCESS; +} + +static NTSTATUS finalize_key_pair( struct key *key ) +{ + UINT32 flags = SYMCRYPT_FLAG_KEY_NO_FIPS; + NTSTATUS status; + + switch (key->alg_id) + { + case ALG_ID_RSA: + case ALG_ID_RSA_SIGN: + { + UINT64 exp = 65537; + + if (key->alg_id == ALG_ID_RSA) flags |= SYMCRYPT_FLAG_RSAKEY_ENCRYPT | SYMCRYPT_FLAG_RSAKEY_SIGN; + else flags |= SYMCRYPT_FLAG_RSAKEY_SIGN; + + if ((status = alloc_rsa_key( key, key->a.bitlen ))) return status; + if (SymCryptRsakeyGenerate( key->a.rsa.handle, &exp, 1, flags )) return STATUS_INTERNAL_ERROR; + break; + } + case ALG_ID_ECDH: + if (!key->a.ecc.curve_id) return STATUS_INVALID_PARAMETER; + /* fall through */ + case ALG_ID_ECDH_P256: + case ALG_ID_ECDH_P384: + case ALG_ID_ECDH_P521: + { + flags |= SYMCRYPT_FLAG_ECKEY_ECDH; + if ((status = alloc_ecc_key( key, key->a.ecc.curve_id ))) return status; + if (SymCryptEckeySetRandom( flags, key->a.ecc.handle )) return STATUS_INTERNAL_ERROR; + break; + } + case ALG_ID_ECDSA: + if (!key->a.ecc.curve_id) return STATUS_INVALID_PARAMETER; + /* fall through */ + case ALG_ID_ECDSA_P256: + case ALG_ID_ECDSA_P384: + case ALG_ID_ECDSA_P521: + { + flags |= SYMCRYPT_FLAG_ECKEY_ECDSA; + if ((status = alloc_ecc_key( key, key->a.ecc.curve_id ))) return status; + if (SymCryptEckeySetRandom( flags, key->a.ecc.handle )) return STATUS_INTERNAL_ERROR; + break; + } + case ALG_ID_DH: + case ALG_ID_DSA: + { + if (key->alg_id == ALG_ID_DSA) flags |= SYMCRYPT_FLAG_DLKEY_DSA; + else flags |= SYMCRYPT_FLAG_DLKEY_DH; - if ((status = key_import( alg, decrypt_key, type, ret_handle, object, object_len, input, input_len ))) return status; - TRACE( "returning handle %p\n", *ret_handle ); + if (!(key->a.dsa.group)) /* no parameters have been imported */ + { + if (!(key->a.dsa.group = SymCryptDlgroupAllocate( key->a.bitlen, 0 ))) return STATUS_NO_MEMORY; + if (SymCryptDlgroupGenerate( NULL, SYMCRYPT_DLGROUP_FIPS_186_2, key->a.dsa.group )) + return STATUS_INTERNAL_ERROR; + } + if (!(key->a.dsa.handle = SymCryptDlkeyAllocate( key->a.dsa.group ))) return STATUS_NO_MEMORY; + if (SymCryptDlkeyGenerate( flags, key->a.dsa.handle )) return STATUS_INTERNAL_ERROR; + break; + } + default: + FIXME( "unhandled algorithm %u\n", key->alg_id ); + return STATUS_NOT_SUPPORTED; + } + + key->flags |= KEY_FLAG_FINALIZED; return STATUS_SUCCESS; } -NTSTATUS WINAPI BCryptExportKey( BCRYPT_KEY_HANDLE export_key_handle, BCRYPT_KEY_HANDLE encrypt_key_handle, - const WCHAR *type, UCHAR *output, ULONG output_len, ULONG *size, ULONG flags ) +NTSTATUS WINAPI BCryptFinalizeKeyPair( BCRYPT_KEY_HANDLE handle, ULONG flags ) { - struct key *key = get_key_object( export_key_handle ); - struct key *encrypt_key = NULL; + struct key *key = get_key_object( handle ); - TRACE( "%p, %p, %s, %p, %lu, %p, %#lx\n", export_key_handle, encrypt_key_handle, debugstr_w(type), output, - output_len, size, flags ); + TRACE( "%p, %#lx\n", key, flags ); - if (!key) return STATUS_INVALID_HANDLE; - if (!type || !size) return STATUS_INVALID_PARAMETER; - if (encrypt_key_handle && !(encrypt_key = get_key_object( encrypt_key_handle ))) return STATUS_INVALID_HANDLE; + if (!key || key->flags & KEY_FLAG_FINALIZED) return STATUS_INVALID_HANDLE; - return key_export( key, encrypt_key, type, output, output_len, size ); + return finalize_key_pair( key ); } -static NTSTATUS key_duplicate( struct key *key_orig, struct key **ret_key ) +static NTSTATUS duplicate_key( const struct key *src, struct key **ret_key ) { - struct key_asymmetric_duplicate_params params; - struct key *key_copy; + struct key *dst; NTSTATUS status; - UCHAR *buffer; - if (!(key_copy = calloc( 1, sizeof(*key_copy) ))) return STATUS_NO_MEMORY; - key_copy->hdr = key_orig->hdr; - key_copy->alg_id = key_orig->alg_id; - - if (is_symmetric_key( key_orig )) + switch (src->alg_id) { - if (!(buffer = malloc( key_orig->u.s.secret_len ))) + case ALG_ID_AES: + if ((status = alloc_key( src->alg_id, src->flags, &dst ))) return status; + if ((status = alloc_aes_key( dst, src->s.mode, src->s.block_size, src->s.secret, src->s.secret_len ))) { - free( key_copy ); - return STATUS_NO_MEMORY; + destroy_key( dst ); + return status; } - memcpy( buffer, key_orig->u.s.secret, key_orig->u.s.secret_len ); - - key_copy->u.s.mode = key_orig->u.s.mode; - key_copy->u.s.block_size = key_orig->u.s.block_size; - key_copy->u.s.secret = buffer; - key_copy->u.s.secret_len = key_orig->u.s.secret_len; - InitializeCriticalSection( &key_copy->u.s.cs ); - *ret_key = key_copy; - return STATUS_SUCCESS; - } + if (src->s.mode == CHAIN_MODE_GCM) + { + SymCryptGcmKeyCopy( &src->s.gcm.handle, &dst->s.gcm.handle ); + SymCryptGcmStateCopy( &src->s.gcm.state, &src->s.gcm.handle, &dst->s.gcm.state ); + } + else SymCryptAesKeyCopy( &src->s.aes.handle, &dst->s.aes.handle ); + break; - key_copy->u.a.bitlen = key_orig->u.a.bitlen; - key_copy->u.a.flags = key_orig->u.a.flags; - key_copy->u.a.dss_seed = key_orig->u.a.dss_seed; + default: + FIXME( "algorithm %u not supported\n", src->alg_id ); + return STATUS_NOT_SUPPORTED; + } - params.key_orig = key_orig; - params.key_copy = key_copy; - if (!(status = UNIX_CALL( key_asymmetric_duplicate, ¶ms ))) *ret_key = key_copy; - else free( key_copy ); - return status; + *ret_key = dst; + return STATUS_SUCCESS; } NTSTATUS WINAPI BCryptDuplicateKey( BCRYPT_KEY_HANDLE handle, BCRYPT_KEY_HANDLE *handle_copy, UCHAR *object, ULONG object_len, ULONG flags ) { - struct key *key_orig = get_key_object( handle ); + const struct key *key_orig = get_key_object( handle ); struct key *key_copy; NTSTATUS status; @@ -2269,240 +3381,294 @@ NTSTATUS WINAPI BCryptDuplicateKey( BCRYPT_KEY_HANDLE handle, BCRYPT_KEY_HANDLE if (!key_orig) return STATUS_INVALID_HANDLE; if (!handle_copy) return STATUS_INVALID_PARAMETER; - if ((status = key_duplicate( key_orig, &key_copy ))) return status; - + if ((status = duplicate_key( key_orig, &key_copy ))) return status; *handle_copy = key_copy; + TRACE( "returning handle %p\n", *handle_copy ); return STATUS_SUCCESS; } -static const WCHAR *resolve_blob_type( const WCHAR *type, UCHAR *input, ULONG input_len ) +static NTSTATUS get_hash_oid( const WCHAR *str, const SYMCRYPT_OID **oid, UINT32 *oid_count ) { - BCRYPT_KEY_BLOB *blob = (BCRYPT_KEY_BLOB *)input; + if (!str) + { + *oid = NULL; + *oid_count = 0; + return STATUS_SUCCESS; + } + if (!wcscmp( str, BCRYPT_MD5_ALGORITHM )) + { + *oid = SymCryptMd5OidList; + *oid_count = ARRAY_SIZE( SymCryptMd5OidList ); + return STATUS_SUCCESS; + } + if (!wcscmp( str, BCRYPT_SHA1_ALGORITHM )) + { + *oid = SymCryptSha1OidList; + *oid_count = ARRAY_SIZE( SymCryptSha1OidList ); + return STATUS_SUCCESS; + } + if (!wcscmp( str, BCRYPT_SHA256_ALGORITHM )) + { + *oid = SymCryptSha256OidList; + *oid_count = ARRAY_SIZE( SymCryptSha256OidList ); + return STATUS_SUCCESS; + } + if (!wcscmp( str, BCRYPT_SHA384_ALGORITHM )) + { + *oid = SymCryptSha384OidList; + *oid_count = ARRAY_SIZE( SymCryptSha384OidList ); + return STATUS_SUCCESS; + } + if (!wcscmp( str, BCRYPT_SHA512_ALGORITHM )) + { + *oid = SymCryptSha512OidList; + *oid_count = ARRAY_SIZE( SymCryptSha512OidList ); + return STATUS_SUCCESS; + } + FIXME( "unhandled algorithm %s\n", debugstr_w(str) ); + return STATUS_NOT_SUPPORTED; +} - if (!type) return NULL; - if (wcscmp( type, BCRYPT_PUBLIC_KEY_BLOB )) return type; - if (input_len < sizeof(*blob)) return NULL; +static NTSTATUS sign_hash_rsa( const struct key *key, const void *padding, const UCHAR *hash, ULONG hash_len, + UCHAR *signature, ULONG signature_len, ULONG *ret_len, ULONG flags ) +{ + ULONG required_len = len_from_bitlen( key->a.bitlen ); + NTSTATUS status; - switch (blob->Magic) + if (!padding) return STATUS_INVALID_PARAMETER; + if (signature_len < required_len) { - case BCRYPT_ECDH_PUBLIC_P256_MAGIC: - case BCRYPT_ECDH_PUBLIC_P384_MAGIC: - case BCRYPT_ECDH_PUBLIC_P521_MAGIC: - case BCRYPT_ECDSA_PUBLIC_P256_MAGIC: - case BCRYPT_ECDSA_PUBLIC_P384_MAGIC: - case BCRYPT_ECDSA_PUBLIC_P521_MAGIC: - return BCRYPT_ECCPUBLIC_BLOB; + *ret_len = required_len; + if (signature) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; + } + if (!(key->flags & KEY_FLAG_PRIVATE)) return STATUS_INVALID_PARAMETER; - case BCRYPT_RSAPUBLIC_MAGIC: - return BCRYPT_RSAPUBLIC_BLOB; + if (flags == BCRYPT_PAD_PKCS1) + { + const BCRYPT_PKCS1_PADDING_INFO *pad = padding; + UINT32 hash_oid_count, flags = 0; + const SYMCRYPT_OID *hash_oid; + SIZE_T size; - case BCRYPT_DSA_PUBLIC_MAGIC: - return BCRYPT_DSA_PUBLIC_BLOB; + if ((status = get_hash_oid( pad->pszAlgId, &hash_oid, &hash_oid_count ))) return status; + if (!hash_oid) flags = SYMCRYPT_FLAG_RSA_PKCS1_NO_ASN1; - default: - FIXME( "unsupported key magic %#lx\n", blob->Magic ); - return NULL; + if (SymCryptRsaPkcs1Sign( key->a.rsa.handle, hash, hash_len, hash_oid, hash_oid_count, flags, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, signature, required_len, &size )) + return STATUS_INTERNAL_ERROR; + *ret_len = size; + } + else if (flags == BCRYPT_PAD_PSS) + { + const BCRYPT_PSS_PADDING_INFO *pad = padding; + const SYMCRYPT_HASH *hash_desc = get_hash_from_str( pad->pszAlgId ); + SIZE_T size; + + if (SymCryptRsaPssSign( key->a.rsa.handle, hash, hash_len, hash_desc, pad->cbSalt, 0, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, signature, required_len, &size )) + return STATUS_INTERNAL_ERROR; + *ret_len = size; + } + else + { + FIXME( "unsupported flags %#lx", flags ); + if (padding) return STATUS_INVALID_PARAMETER; + return STATUS_NOT_SUPPORTED; } + + return STATUS_SUCCESS; } -NTSTATUS WINAPI BCryptImportKeyPair( BCRYPT_ALG_HANDLE handle, BCRYPT_KEY_HANDLE decrypt_key_handle, const WCHAR *type, - BCRYPT_KEY_HANDLE *ret_handle, UCHAR *input, ULONG input_len, ULONG flags ) +static NTSTATUS sign_hash_ecc( const struct key *key, const UCHAR *hash, ULONG hash_len, UCHAR *signature, + ULONG signature_len, ULONG *ret_len ) { - struct algorithm *alg = get_alg_object( handle ); - NTSTATUS status; + ULONG required_len = len_from_bitlen( key->a.bitlen ) * 2; - TRACE( "%p, %p, %s, %p, %p, %lu, %#lx\n", handle, decrypt_key_handle, debugstr_w(type), ret_handle, input, - input_len, flags ); - - if (!alg) return STATUS_INVALID_HANDLE; - if (!ret_handle || !input || !(type = resolve_blob_type( type, input, input_len ))) - return STATUS_INVALID_PARAMETER; - if (decrypt_key_handle) + if (signature_len < required_len) { - FIXME( "decryption of key not yet supported\n" ); - return STATUS_NOT_IMPLEMENTED; + *ret_len = required_len; + if (signature) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; } + if (!(key->flags & KEY_FLAG_PRIVATE)) return STATUS_INVALID_PARAMETER; - if ((status = key_import_pair( alg, type, ret_handle, input, input_len ))) return status; - TRACE( "returning handle %p\n", *ret_handle ); + if (SymCryptEcDsaSign( key->a.ecc.handle, hash, hash_len, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0, signature, + required_len )) return STATUS_INTERNAL_ERROR; + + *ret_len = required_len; return STATUS_SUCCESS; } -NTSTATUS WINAPI BCryptSignHash( BCRYPT_KEY_HANDLE handle, void *padding, UCHAR *input, ULONG input_len, - UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +static NTSTATUS sign_hash_dsa( const struct key *key, const UCHAR *hash, ULONG hash_len, UCHAR *signature, + ULONG signature_len, ULONG *ret_len ) { - struct key_asymmetric_sign_params params; - struct key *key = get_key_object( handle ); + ULONG required_len = hash_len * 2; - TRACE( "%p, %p, %p, %lu, %p, %lu, %p, %#lx\n", handle, padding, input, input_len, output, output_len, - ret_len, flags ); + if (signature_len < required_len) + { + *ret_len = required_len; + if (signature) return STATUS_BUFFER_TOO_SMALL; + return STATUS_SUCCESS; + } + if (!(key->flags & KEY_FLAG_PRIVATE)) return STATUS_INVALID_PARAMETER; - if (!key) return STATUS_INVALID_HANDLE; - if (!is_signature_key( key )) return STATUS_NOT_SUPPORTED; + if (SymCryptDsaSign( key->a.dsa.handle, hash, hash_len, SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0, signature, + required_len )) return STATUS_INTERNAL_ERROR; - params.key = key; - params.padding = padding; - params.input = input; - params.input_len = input_len; - params.output = output; - params.output_len = output_len; - params.ret_len = ret_len; - params.flags = flags; - return UNIX_CALL( key_asymmetric_sign, ¶ms ); + *ret_len = required_len; + return STATUS_SUCCESS; } -NTSTATUS WINAPI BCryptVerifySignature( BCRYPT_KEY_HANDLE handle, void *padding, UCHAR *hash, ULONG hash_len, - UCHAR *signature, ULONG signature_len, ULONG flags ) +NTSTATUS WINAPI BCryptSignHash( BCRYPT_KEY_HANDLE handle, void *padding, UCHAR *hash, ULONG hash_len, + UCHAR *signature, ULONG signature_len, ULONG *ret_len, ULONG flags ) { - struct key_asymmetric_verify_params params; - struct key *key = get_key_object( handle ); + const struct key *key = get_key_object( handle ); - TRACE( "%p, %p, %p, %lu, %p, %lu, %#lx\n", handle, padding, hash, hash_len, signature, signature_len, flags ); + TRACE( "%p, %p, %p, %lu, %p, %lu, %p, %#lx\n", handle, padding, hash, hash_len, signature, signature_len, + ret_len, flags ); if (!key) return STATUS_INVALID_HANDLE; - if (!is_signature_key( key )) return STATUS_NOT_SUPPORTED; - if (!hash || !hash_len || !signature || !signature_len) return STATUS_INVALID_PARAMETER; - params.key = key; - params.padding = padding; - params.hash = hash; - params.hash_len = hash_len; - params.signature = signature; - params.signature_len = signature_len; - params.flags = flags; - return UNIX_CALL( key_asymmetric_verify, ¶ms ); -} + switch (key->alg_id) + { + case ALG_ID_RSA: + case ALG_ID_RSA_SIGN: + return sign_hash_rsa( key, padding, hash, hash_len, signature, signature_len, ret_len, flags ); -NTSTATUS WINAPI BCryptDestroyKey( BCRYPT_KEY_HANDLE handle ) -{ - struct key *key = get_key_object( handle ); + case ALG_ID_ECDSA: + case ALG_ID_ECDSA_P256: + case ALG_ID_ECDSA_P384: + case ALG_ID_ECDSA_P521: + if (padding) return STATUS_INVALID_PARAMETER; + return sign_hash_ecc( key, hash, hash_len, signature, signature_len, ret_len ); - TRACE( "%p\n", handle ); + case ALG_ID_DSA: + return sign_hash_dsa( key, hash, hash_len, signature, signature_len, ret_len ); - if (!key) return STATUS_INVALID_HANDLE; - key_destroy( key ); - return STATUS_SUCCESS; + default: + FIXME( "unhandled algorithm %u\n", key->alg_id ); + break; + } + + return STATUS_NOT_SUPPORTED; } -NTSTATUS WINAPI BCryptEncrypt( BCRYPT_KEY_HANDLE handle, UCHAR *input, ULONG input_len, void *padding, UCHAR *iv, - ULONG iv_len, UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +static NTSTATUS verify_signature_rsa( const struct key *key, const void *padding, const UCHAR *hash, ULONG hash_len, + const UCHAR *signature, ULONG signature_len, ULONG flags ) { - struct key *key = get_key_object( handle ); - struct key_asymmetric_encrypt_params asymmetric_params; - NTSTATUS ret; + NTSTATUS status; + SYMCRYPT_ERROR error; - TRACE( "%p, %p, %lu, %p, %p, %lu, %p, %lu, %p, %#lx\n", handle, input, input_len, padding, iv, iv_len, output, - output_len, ret_len, flags ); + if (!padding || signature_len != len_from_bitlen( key->a.bitlen )) return STATUS_INVALID_PARAMETER; - if (!key) return STATUS_INVALID_HANDLE; + if (flags == BCRYPT_PAD_PKCS1) + { + const BCRYPT_PKCS1_PADDING_INFO *pad = padding; + UINT32 flags = 0, hash_oid_count; + const SYMCRYPT_OID *hash_oid; - if (is_symmetric_key( key )) + if (!pad) return STATUS_INVALID_PARAMETER; + if ((status = get_hash_oid( pad->pszAlgId, &hash_oid, &hash_oid_count ))) return status; + if (!hash_oid) flags = SYMCRYPT_FLAG_RSA_PKCS1_OPTIONAL_HASH_OID; + + error = SymCryptRsaPkcs1Verify( key->a.rsa.handle, hash, hash_len, signature, signature_len, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, hash_oid, hash_oid_count, flags ); + } + else if (flags == BCRYPT_PAD_PSS) { - if (flags & ~BCRYPT_BLOCK_PADDING) - { - FIXME( "flags %#lx not implemented\n", flags ); - return STATUS_NOT_IMPLEMENTED; - } - EnterCriticalSection( &key->u.s.cs ); - ret = key_symmetric_encrypt( key, input, input_len, padding, iv, iv_len, output, output_len, ret_len, flags ); - LeaveCriticalSection( &key->u.s.cs ); + const BCRYPT_PSS_PADDING_INFO *pad = padding; + const SYMCRYPT_HASH *hash_desc = get_hash_from_str( pad->pszAlgId ); + + error = SymCryptRsaPssVerify( key->a.rsa.handle, hash, hash_len, signature, signature_len, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, hash_desc, pad->cbSalt, 0 ); } else { - if (flags & BCRYPT_PAD_NONE) - { - FIXME( "flags %#lx not implemented\n", flags ); - return STATUS_NOT_IMPLEMENTED; - } - if (!is_asymmetric_encryption_key( key )) return STATUS_NOT_SUPPORTED; - - asymmetric_params.input = input; - asymmetric_params.input_len = input_len; - asymmetric_params.padding = padding; - asymmetric_params.key = key; - asymmetric_params.output = output; - asymmetric_params.output_len = output_len; - asymmetric_params.ret_len = ret_len; - asymmetric_params.flags = flags; - ret = UNIX_CALL(key_asymmetric_encrypt, &asymmetric_params); + FIXME( "unsupported flags %#lx\n", flags ); + if (padding) return STATUS_INVALID_PARAMETER; + return STATUS_NOT_SUPPORTED; } - return ret; + if (error == SYMCRYPT_SIGNATURE_VERIFICATION_FAILURE) return STATUS_INVALID_SIGNATURE; + if (error) return STATUS_INTERNAL_ERROR; + return STATUS_SUCCESS; } -NTSTATUS WINAPI BCryptDecrypt( BCRYPT_KEY_HANDLE handle, UCHAR *input, ULONG input_len, void *padding, UCHAR *iv, - ULONG iv_len, UCHAR *output, ULONG output_len, ULONG *ret_len, ULONG flags ) +static NTSTATUS verify_signature_ecc( const struct key *key, const UCHAR *hash, ULONG hash_len, const UCHAR *signature, + ULONG signature_len, ULONG flags ) { - struct key *key = get_key_object( handle ); - struct key_asymmetric_decrypt_params params; - NTSTATUS ret; + SYMCRYPT_ERROR error; - TRACE( "%p, %p, %lu, %p, %p, %lu, %p, %lu, %p, %#lx\n", handle, input, input_len, padding, iv, iv_len, output, - output_len, ret_len, flags ); + if (signature_len != len_from_bitlen( key->a.bitlen ) * 2) return STATUS_INVALID_PARAMETER; - if (!key) return STATUS_INVALID_HANDLE; + error = SymCryptEcDsaVerify( key->a.ecc.handle, hash, hash_len, signature, signature_len, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0 ); - if (is_symmetric_key( key )) - { - if (flags & ~BCRYPT_BLOCK_PADDING) - { - FIXME( "flags %#lx not supported\n", flags ); - return STATUS_NOT_IMPLEMENTED; - } + if (error == SYMCRYPT_SIGNATURE_VERIFICATION_FAILURE) return STATUS_INVALID_SIGNATURE; + if (error) return STATUS_INTERNAL_ERROR; + return STATUS_SUCCESS; +} - EnterCriticalSection( &key->u.s.cs ); - ret = key_symmetric_decrypt( key, input, input_len, padding, iv, iv_len, output, output_len, ret_len, flags ); - LeaveCriticalSection( &key->u.s.cs ); - } - else - { - if (flags & BCRYPT_PAD_NONE) - { - FIXME( "flags %#lx not implemented\n", flags ); - return STATUS_NOT_IMPLEMENTED; - } - if (!is_asymmetric_encryption_key( key )) return STATUS_NOT_SUPPORTED; +static NTSTATUS verify_signature_dsa( const struct key *key, const UCHAR *hash, ULONG hash_len, const UCHAR *signature, + ULONG signature_len, ULONG flags ) +{ + SYMCRYPT_ERROR error; - params.key = key; - params.input = input; - params.input_len = input_len; - params.padding = padding; - params.output = output; - params.output_len = output_len; - params.ret_len = ret_len; - params.flags = flags; - ret = UNIX_CALL(key_asymmetric_decrypt, ¶ms); - } + if (signature_len != hash_len * 2) return STATUS_INVALID_PARAMETER; - return ret; + error = SymCryptDsaVerify( key->a.dsa.handle, hash, hash_len, signature, signature_len, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0 ); + + if (error == SYMCRYPT_SIGNATURE_VERIFICATION_FAILURE) return STATUS_INVALID_SIGNATURE; + if (error) return STATUS_INTERNAL_ERROR; + return STATUS_SUCCESS; } -NTSTATUS WINAPI BCryptSetProperty( BCRYPT_HANDLE handle, const WCHAR *prop, UCHAR *value, ULONG size, ULONG flags ) +NTSTATUS WINAPI BCryptVerifySignature( BCRYPT_KEY_HANDLE handle, void *padding, UCHAR *hash, ULONG hash_len, + UCHAR *signature, ULONG signature_len, ULONG flags ) { - struct object *object = get_object( handle, 0 ); + const struct key *key = get_key_object( handle ); - TRACE( "%p, %s, %p, %lu, %#lx\n", handle, debugstr_w(prop), value, size, flags ); + TRACE( "%p, %p, %p, %lu, %p, %lu, %#lx\n", handle, padding, hash, hash_len, signature, signature_len, flags ); - if (!handle) return STATUS_INVALID_HANDLE; - if (is_alg_pseudo_handle( handle )) return STATUS_ACCESS_DENIED; + if (!key) return STATUS_INVALID_HANDLE; + if (!hash || !hash_len || !signature || !signature_len) return STATUS_INVALID_PARAMETER; - switch (object->magic) - { - case MAGIC_ALG: - { - struct algorithm *alg = (struct algorithm *)object; - return set_alg_property( alg, prop, value, size, flags ); - } - case MAGIC_KEY: + switch (key->alg_id) { - struct key *key = (struct key *)object; - return set_key_property( key, prop, value, size, flags ); - } + case ALG_ID_RSA: + case ALG_ID_RSA_SIGN: + return verify_signature_rsa( key, padding, hash, hash_len, signature, signature_len, flags ); + + case ALG_ID_ECDSA: + case ALG_ID_ECDSA_P256: + case ALG_ID_ECDSA_P384: + case ALG_ID_ECDSA_P521: + return verify_signature_ecc( key, hash, hash_len, signature, signature_len, flags ); + + case ALG_ID_DSA: + return verify_signature_dsa( key, hash, hash_len, signature, signature_len, flags ); + default: - WARN( "unknown magic %#lx\n", object->magic ); - return STATUS_INVALID_HANDLE; + FIXME( "unhandled algorithm %u\n", key->alg_id ); + break; } + + return STATUS_NOT_SUPPORTED; +} + +NTSTATUS WINAPI BCryptDestroyKey( BCRYPT_KEY_HANDLE handle ) +{ + struct key *key = get_key_object( handle ); + + TRACE( "%p\n", handle ); + + if (!key || key->hdr.magic != MAGIC_KEY) return STATUS_INVALID_HANDLE; + + destroy_key( key ); + return STATUS_SUCCESS; } #define HMAC_PAD_LEN 64 @@ -2548,43 +3714,64 @@ NTSTATUS WINAPI BCryptDeriveKeyCapi( BCRYPT_HASH_HANDLE handle, BCRYPT_ALG_HANDL return STATUS_SUCCESS; } +static void destroy_secret( struct secret *secret ) +{ + if (!secret) return; + secure_zero( secret->derived_key, secret->derived_key_len ); + free( secret->derived_key ); + destroy_object( &secret->hdr ); +} + NTSTATUS WINAPI BCryptSecretAgreement( BCRYPT_KEY_HANDLE privkey_handle, BCRYPT_KEY_HANDLE pubkey_handle, BCRYPT_SECRET_HANDLE *ret_handle, ULONG flags ) { - struct key_asymmetric_derive_key_params params; - struct key *privkey = get_key_object( privkey_handle ); - struct key *pubkey = get_key_object( pubkey_handle ); + const struct key *privkey = get_key_object( privkey_handle ); + const struct key *pubkey = get_key_object( pubkey_handle ); struct secret *secret; - ULONG len; - NTSTATUS status; TRACE( "%p, %p, %p, %#lx\n", privkey_handle, pubkey_handle, ret_handle, flags ); - if (!privkey || !pubkey) return STATUS_INVALID_HANDLE; - if (!is_agreement_key( privkey ) || !is_agreement_key( pubkey )) return STATUS_NOT_SUPPORTED; + if (!privkey || !pubkey || privkey->alg_id != pubkey->alg_id || !(privkey->flags & KEY_FLAG_PRIVATE)) + return STATUS_INVALID_HANDLE; if (!ret_handle) return STATUS_INVALID_PARAMETER; if (!(secret = calloc( 1, sizeof(*secret) ))) return STATUS_NO_MEMORY; secret->hdr.magic = MAGIC_SECRET; - secret->derived_key_len = len_from_bitlen( privkey->u.a.bitlen ); + secret->derived_key_len = len_from_bitlen( privkey->a.bitlen ); if (!(secret->derived_key = malloc( secret->derived_key_len ))) { free( secret ); return STATUS_NO_MEMORY; } - params.privkey = privkey; - params.pubkey = pubkey; - params.output = secret->derived_key; - params.output_len = secret->derived_key_len; - params.ret_len = &len; - if ((status = UNIX_CALL( key_asymmetric_derive_key, ¶ms ))) + switch (privkey->alg_id) { - free( secret->derived_key ); - free( secret ); - return status; + case ALG_ID_DH: + if (SymCryptDhSecretAgreement( privkey->a.dsa.handle, pubkey->a.dsa.handle, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0, secret->derived_key, + secret->derived_key_len )) + { + destroy_secret( secret ); + return STATUS_INTERNAL_ERROR; + } + break; + case ALG_ID_ECDH: + case ALG_ID_ECDH_P256: + case ALG_ID_ECDH_P384: + case ALG_ID_ECDH_P521: + if (SymCryptEcDhSecretAgreement( privkey->a.ecc.handle, pubkey->a.ecc.handle, + SYMCRYPT_NUMBER_FORMAT_MSB_FIRST, 0, secret->derived_key, + secret->derived_key_len )) + { + destroy_secret( secret ); + return STATUS_INTERNAL_ERROR; + } + break; + default: + FIXME( "algorithm %u not supported\n", privkey->alg_id ); + destroy_secret( secret ); + return STATUS_NOT_SUPPORTED; } - secret->derived_key_len = len; *ret_handle = secret; TRACE( "returning handle %p\n", *ret_handle ); @@ -2597,10 +3784,9 @@ NTSTATUS WINAPI BCryptDestroySecret( BCRYPT_SECRET_HANDLE handle ) TRACE( "%p\n", handle ); - if (!secret) return STATUS_INVALID_HANDLE; - SecureZeroMemory( secret->derived_key, secret->derived_key_len ); - free( secret->derived_key ); - destroy_object( &secret->hdr ); + if (!secret || secret->hdr.magic != MAGIC_SECRET) return STATUS_INVALID_HANDLE; + + destroy_secret( secret ); return STATUS_SUCCESS; } @@ -2783,7 +3969,7 @@ NTSTATUS WINAPI BCryptKeyDerivation( BCRYPT_KEY_HANDLE handle, BCryptBufferDesc } if (!mac) return STATUS_INVALID_PARAMETER; - if (SymCryptPbkdf2( mac, key->u.s.secret, key->u.s.secret_len, salt, salt_len, iterations, output, output_len )) + if (SymCryptPbkdf2( mac, key->s.secret, key->s.secret_len, salt, salt_len, iterations, output, output_len )) return STATUS_INTERNAL_ERROR; *ret_len = output_len; @@ -2797,14 +3983,9 @@ BOOL WINAPI DllMain( HINSTANCE hinst, DWORD reason, LPVOID reserved ) case DLL_PROCESS_ATTACH: DisableThreadLibraryCalls( hinst ); SymCryptInit(); - if (!__wine_init_unix_call()) - { - if (UNIX_CALL( process_attach, NULL)) __wine_unixlib_handle = 0; - } break; case DLL_PROCESS_DETACH: if (reserved) break; - if (__wine_unixlib_handle) UNIX_CALL( process_detach, NULL ); break; } return TRUE; diff --git a/dlls/bcrypt/tests/bcrypt.c b/dlls/bcrypt/tests/bcrypt.c index a1eb766a03e..b7c7fddbffa 100644 --- a/dlls/bcrypt/tests/bcrypt.c +++ b/dlls/bcrypt/tests/bcrypt.c @@ -2059,8 +2059,10 @@ static void test_BCryptDecrypt(void) ret = BCryptDestroyKey(key); ok(ret == STATUS_SUCCESS, "got %#lx\n", ret); + if (0) { ret = BCryptDestroyKey(key); ok(ret == STATUS_INVALID_HANDLE, "got %#lx\n", ret); + } free(buf); ret = BCryptDestroyKey(NULL); @@ -2440,7 +2442,7 @@ static void test_ECDSA(void) ok(status == STATUS_SUCCESS, "got %#lx\n", status); status = BCryptGenerateKeyPair(alg, &key, 255, 0); - todo_wine ok(status == STATUS_INVALID_PARAMETER, "got %#lx\n", status); + ok(status == STATUS_INVALID_PARAMETER, "got %#lx\n", status); status = BCryptGenerateKeyPair(alg, &key, 0, 0); ok(status == STATUS_SUCCESS, "got %#lx\n", status); @@ -2658,7 +2660,6 @@ static void test_rsa_encrypt(void) ok(ret == STATUS_SUCCESS, "got %#lx\n", ret); /* No padding */ - todo_wine { memset(input_no_padding, 0, sizeof(input_no_padding)); encrypted_size = 0; @@ -2690,11 +2691,9 @@ static void test_rsa_encrypt(void) ret = BCryptEncrypt(key, input_no_padding, sizeof(input_no_padding), NULL, NULL, 0, encrypted_b, encrypted_size, &encrypted_size, BCRYPT_PAD_NONE); ok(ret == STATUS_SUCCESS, "got %lx\n", ret); - } ok(!memcmp(encrypted_a, encrypted_b, encrypted_size), "Both outputs should be the same\n"); ok(!memcmp(encrypted_b, rsa_encrypted_no_padding, encrypted_size), "Data mismatch.\n"); - todo_wine { decrypted_size = 0; ret = BCryptDecrypt(key, encrypted_a, encrypted_size, NULL, NULL, 0, NULL, 0, &decrypted_size, BCRYPT_PAD_NONE); ok(ret == STATUS_SUCCESS, "got %lx\n", ret); @@ -2704,7 +2703,6 @@ static void test_rsa_encrypt(void) ok(ret == STATUS_SUCCESS, "got %lx\n", ret); ok(decrypted_size == sizeof(input_no_padding), "got %lu\n", decrypted_size); ok(!memcmp(decrypted, input_no_padding, sizeof(input_no_padding)), "unexpected output\n"); - } /* PKCS1 Padding */ encrypted_size = 0; @@ -2849,7 +2847,7 @@ static void test_rsa_encrypt(void) encrypted_null = malloc(encrypted_null_size); ret = BCryptEncrypt(key, input, sizeof(input), NULL, NULL, 0, encrypted_null, encrypted_null_size, &encrypted_null_size, BCRYPT_PAD_OAEP); - ok(ret == STATUS_INVALID_PARAMETER || broken(ret == STATUS_SUCCESS), + todo_wine ok(ret == STATUS_INVALID_PARAMETER || broken(ret == STATUS_SUCCESS), "unexpected OAEP(NULL) encrypt status %lx\n", ret); free(encrypted_null); @@ -3550,7 +3548,7 @@ static void test_ECDH(void) ok(status == STATUS_SUCCESS, "got %#lx\n", status); status = BCryptGenerateKeyPair(alg, &key, 255, 0); - todo_wine ok(status == STATUS_INVALID_PARAMETER, "got %#lx\n", status); + ok(status == STATUS_INVALID_PARAMETER, "got %#lx\n", status); status = BCryptGenerateKeyPair(alg, &key, 0, 0); ok(status == STATUS_SUCCESS, "got %#lx\n", status); @@ -3974,9 +3972,9 @@ static void test_BCryptSignHash(void) ok(!ret, "got %#lx\n", ret); ret = BCryptGenerateKeyPair(alg, &key, 256, 0); - todo_wine ok(ret == STATUS_INVALID_PARAMETER, "got %#lx\n", ret); + ok(ret == STATUS_INVALID_PARAMETER, "got %#lx\n", ret); ret = BCryptGenerateKeyPair(alg, &key, 522, 0); - todo_wine ok(ret == STATUS_INVALID_PARAMETER, "got %#lx\n", ret); + ok(ret == STATUS_INVALID_PARAMETER, "got %#lx\n", ret); ret = BCryptGenerateKeyPair(alg, &key, 521, 0); ok(ret == STATUS_SUCCESS, "got %#lx\n", ret); @@ -4257,7 +4255,7 @@ static void test_DSA(void) ok(size == sizeof(*dsablob) + dsablob->cbKey * 3, "got %lu\n", size); ret = BCryptExportKey(key, NULL, BCRYPT_DSA_PRIVATE_BLOB, buf2, sizeof(buf2), &size, 0); - todo_wine ok(ret == STATUS_INVALID_PARAMETER, "got %#lx\n", ret); + ok(ret == STATUS_INVALID_PARAMETER, "got %#lx\n", ret); ret = BCryptVerifySignature(key, NULL, dsaHash, sizeof(dsaHash), dsaSignature, sizeof(dsaSignature), 0); ok(!ret, "got %#lx\n", ret); diff --git a/include/bcrypt.h b/include/bcrypt.h index bf22f6dfb9b..a3441fbf72f 100644 --- a/include/bcrypt.h +++ b/include/bcrypt.h @@ -253,6 +253,9 @@ static const WCHAR BCRYPT_DH_PARAMETERS[] = {'D','H','P','a','r','a','m','e','t' #define BCRYPT_ECDH_PUBLIC_GENERIC_MAGIC 0x504b4345 #define BCRYPT_ECDH_PRIVATE_GENERIC_MAGIC 0x564b4345 +#define BCRYPT_ECDSA_PUBLIC_GENERIC_MAGIC 0x50444345 +#define BCRYPT_ECDSA_PRIVATE_GENERIC_MAGIC 0x56444345 + #define BCRYPT_CIPHER_OPERATION 0x00000001 #define BCRYPT_HASH_OPERATION 0x00000002 #define BCRYPT_ASYMMETRIC_ENCRYPTION_OPERATION 0x00000004 -- GitLab https://gitlab.winehq.org/wine/wine/-/merge_requests/11097
From: Hans Leidekker <hans@codeweavers.com> --- dlls/bcrypt/Makefile.in | 5 +- dlls/bcrypt/bcrypt_internal.h | 300 --- dlls/bcrypt/gnutls.c | 3594 --------------------------------- 3 files changed, 1 insertion(+), 3898 deletions(-) delete mode 100644 dlls/bcrypt/bcrypt_internal.h delete mode 100644 dlls/bcrypt/gnutls.c diff --git a/dlls/bcrypt/Makefile.in b/dlls/bcrypt/Makefile.in index 0e5692cb3b1..9cbbc529104 100644 --- a/dlls/bcrypt/Makefile.in +++ b/dlls/bcrypt/Makefile.in @@ -2,12 +2,9 @@ MODULE = bcrypt.dll IMPORTLIB = bcrypt IMPORTS = $(SYMCRYPT_PE_LIBS) advapi32 EXTRAINCL = $(SYMCRYPT_PE_CFLAGS) -UNIXLIB = bcrypt.so -UNIX_CFLAGS = $(GNUTLS_CFLAGS) VER_FILEDESCRIPTION_STR = "Wine Crypto Library" VER_PRODUCTVERSION = 6,0,6000,16386 SOURCES = \ - bcrypt_main.c \ - gnutls.c + bcrypt_main.c diff --git a/dlls/bcrypt/bcrypt_internal.h b/dlls/bcrypt/bcrypt_internal.h deleted file mode 100644 index 49c2ede6903..00000000000 --- a/dlls/bcrypt/bcrypt_internal.h +++ /dev/null @@ -1,300 +0,0 @@ -/* - * Copyright 2016 Michael Müller - * - * This library is free software; you can redistribute it and/or - * modify it under the terms of the GNU Lesser General Public - * License as published by the Free Software Foundation; either - * version 2.1 of the License, or (at your option) any later version. - * - * This library is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU - * Lesser General Public License for more details. - * - * You should have received a copy of the GNU Lesser General Public - * License along with this library; if not, write to the Free Software - * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA - * - */ - -#ifndef __BCRYPT_INTERNAL_H -#define __BCRYPT_INTERNAL_H - -#include <stdarg.h> - -#include "windef.h" -#include "winbase.h" -#include "winternl.h" -#include "wincrypt.h" -#include "bcrypt.h" -#include "wine/unixlib.h" - -#define MAGIC_DSS1 ('D' | ('S' << 8) | ('S' << 16) | ('1' << 24)) -#define MAGIC_DSS2 ('D' | ('S' << 8) | ('S' << 16) | ('2' << 24)) - -#define MAGIC_ALG (('A' << 24) | ('L' << 16) | ('G' << 8) | '0') -#define MAGIC_HASH (('H' << 24) | ('A' << 16) | ('S' << 8) | 'H') -#define MAGIC_KEY (('K' << 24) | ('E' << 16) | ('Y' << 8) | '0') -#define MAGIC_SECRET (('S' << 24) | ('C' << 16) | ('R' << 8) | 'T') -struct object -{ - ULONG magic; -}; - -enum alg_id -{ - /* cipher */ - ALG_ID_3DES, - ALG_ID_AES, - ALG_ID_RC4, - - /* hash */ - ALG_ID_SHA256, - ALG_ID_SHA384, - ALG_ID_SHA512, - ALG_ID_SHA1, - ALG_ID_MD5, - ALG_ID_MD4, - ALG_ID_MD2, - - /* asymmetric encryption */ - ALG_ID_RSA, - - /* secret agreement */ - ALG_ID_DH, - ALG_ID_ECDH, - ALG_ID_ECDH_P256, - ALG_ID_ECDH_P384, - ALG_ID_ECDH_P521, - - /* signature */ - ALG_ID_RSA_SIGN, - ALG_ID_ECDSA, - ALG_ID_ECDSA_P256, - ALG_ID_ECDSA_P384, - ALG_ID_ECDSA_P521, - ALG_ID_DSA, - - /* rng */ - ALG_ID_RNG, - - /* key derivation */ - ALG_ID_PBKDF2, -}; - -enum chain_mode -{ - CHAIN_MODE_CBC, - CHAIN_MODE_ECB, - CHAIN_MODE_CFB, - CHAIN_MODE_CCM, - CHAIN_MODE_GCM, -}; - -enum ecc_curve_id -{ - ECC_CURVE_NONE, - ECC_CURVE_25519, - ECC_CURVE_P256R1, - ECC_CURVE_P384R1, - ECC_CURVE_P521R1, -}; - -struct algorithm -{ - struct object hdr; - enum alg_id id; - enum chain_mode mode; - unsigned flags; - enum ecc_curve_id curve_id; -}; - -struct key_symmetric -{ - enum chain_mode mode; - ULONG block_size; - UCHAR *vector; - ULONG vector_len; - UCHAR *secret; - unsigned secret_len; - CRITICAL_SECTION cs; -}; - -#define KEY_FLAG_LEGACY_DSA_V2 0x00000001 -#define KEY_FLAG_FINALIZED 0x00000002 - -struct key_asymmetric -{ - ULONG bitlen; /* key strength for ECC keys */ - enum ecc_curve_id curve_id; - unsigned flags; - DSSSEED dss_seed; -}; - -#define PRIVATE_DATA_SIZE 3 -struct key -{ - struct object hdr; - enum alg_id alg_id; - UINT64 private[PRIVATE_DATA_SIZE]; /* private data for backend */ - union - { - struct key_symmetric s; - struct key_asymmetric a; - } u; -}; - -struct secret -{ - struct object hdr; - UCHAR *derived_key; - ULONG derived_key_len; -}; - -struct key_symmetric_set_auth_data_params -{ - struct key *key; - UCHAR *auth_data; - ULONG len; -}; - -struct key_symmetric_encrypt_params -{ - struct key *key; - const UCHAR *input; - unsigned input_len; - UCHAR *output; - ULONG output_len; -}; - -struct key_symmetric_decrypt_params -{ - struct key *key; - const UCHAR *input; - unsigned input_len; - UCHAR *output; - ULONG output_len; -}; - -struct key_symmetric_get_tag_params -{ - struct key *key; - UCHAR *tag; - ULONG len; -}; - -struct key_asymmetric_decrypt_params -{ - struct key *key; - UCHAR *input; - unsigned input_len; - void *padding; - UCHAR *output; - ULONG output_len; - ULONG *ret_len; - ULONG flags; -}; - -struct key_asymmetric_encrypt_params -{ - struct key *key; - UCHAR *input; - unsigned input_len; - void *padding; - UCHAR *output; - ULONG output_len; - ULONG *ret_len; - ULONG flags; -}; - -struct key_asymmetric_duplicate_params -{ - struct key *key_orig; - struct key *key_copy; -}; - -struct key_asymmetric_sign_params -{ - struct key *key; - void *padding; - UCHAR *input; - unsigned input_len; - UCHAR *output; - ULONG output_len; - ULONG *ret_len; - unsigned flags; -}; - -struct key_asymmetric_verify_params -{ - struct key *key; - void *padding; - UCHAR *hash; - unsigned hash_len; - UCHAR *signature; - ULONG signature_len; - unsigned flags; -}; - -#define KEY_EXPORT_FLAG_PUBLIC 0x00000001 -#define KEY_EXPORT_FLAG_RSA_FULL 0x00000002 -#define KEY_EXPORT_FLAG_DH_PARAMETERS 0x00000004 - -struct key_asymmetric_export_params -{ - struct key *key; - ULONG flags; - UCHAR *buf; - ULONG len; - ULONG *ret_len; -}; - -#define KEY_IMPORT_FLAG_PUBLIC 0x00000001 -#define KEY_IMPORT_FLAG_DH_PARAMETERS 0x00000002 - -struct key_asymmetric_import_params -{ - struct key *key; - ULONG flags; - UCHAR *buf; - ULONG len; -}; - -struct key_asymmetric_derive_key_params -{ - struct key *privkey; - struct key *pubkey; - UCHAR *output; - ULONG output_len; - ULONG *ret_len; -}; - -enum key_funcs -{ - unix_process_attach, - unix_process_detach, - unix_key_symmetric_vector_reset, - unix_key_symmetric_set_auth_data, - unix_key_symmetric_encrypt, - unix_key_symmetric_decrypt, - unix_key_symmetric_get_tag, - unix_key_symmetric_destroy, - unix_key_asymmetric_generate, - unix_key_asymmetric_decrypt, - unix_key_asymmetric_encrypt, - unix_key_asymmetric_duplicate, - unix_key_asymmetric_sign, - unix_key_asymmetric_verify, - unix_key_asymmetric_destroy, - unix_key_asymmetric_export, - unix_key_asymmetric_import, - unix_key_asymmetric_derive_key, - unix_funcs_count, -}; - -static inline ULONG len_from_bitlen( ULONG bitlen ) -{ - return (bitlen + 7) / 8; -} - -#endif /* __BCRYPT_INTERNAL_H */ diff --git a/dlls/bcrypt/gnutls.c b/dlls/bcrypt/gnutls.c deleted file mode 100644 index e8c6b6dd802..00000000000 --- a/dlls/bcrypt/gnutls.c +++ /dev/null @@ -1,3594 +0,0 @@ -/* - * Copyright 2009 Henri Verbeet for CodeWeavers - * Copyright 2018 Hans Leidekker for CodeWeavers - * - * This library is free software; you can redistribute it and/or - * modify it under the terms of the GNU Lesser General Public - * License as published by the Free Software Foundation; either - * version 2.1 of the License, or (at your option) any later version. - * - * This library is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU - * Lesser General Public License for more details. - * - * You should have received a copy of the GNU Lesser General Public - * License along with this library; if not, write to the Free Software - * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA - * - */ - -#if 0 -#pragma makedep unix -#endif - -#include "config.h" - -#ifdef HAVE_GNUTLS_CIPHER_INIT - -#include <stdarg.h> -#include <stdlib.h> -#include <assert.h> -#include <sys/types.h> -#include <dlfcn.h> -#include <gnutls/gnutls.h> -#include <gnutls/crypto.h> -#include <gnutls/abstract.h> - -#include "ntstatus.h" -#include "windef.h" -#include "winbase.h" -#include "winternl.h" -#include "ntsecapi.h" -#include "wincrypt.h" -#include "bcrypt.h" - -#include "bcrypt_internal.h" - -#include "wine/debug.h" - -WINE_DEFAULT_DEBUG_CHANNEL(bcrypt); -WINE_DECLARE_DEBUG_CHANNEL(winediag); - -#if GNUTLS_VERSION_MAJOR < 3 -#define GNUTLS_CIPHER_AES_192_CBC 92 -#define GNUTLS_CIPHER_AES_128_GCM 93 -#define GNUTLS_CIPHER_AES_256_GCM 94 -#define GNUTLS_PK_ECC 4 - -#define GNUTLS_CURVE_TO_BITS(curve) (unsigned int)(((unsigned int)1<<31)|((unsigned int)(curve))) - -typedef enum -{ - GNUTLS_ECC_CURVE_INVALID, - GNUTLS_ECC_CURVE_SECP224R1, - GNUTLS_ECC_CURVE_SECP256R1, - GNUTLS_ECC_CURVE_SECP384R1, - GNUTLS_ECC_CURVE_SECP521R1, -} gnutls_ecc_curve_t; -#endif - -#if GNUTLS_VERSION_MAJOR < 3 || (GNUTLS_VERSION_MAJOR == 3 && GNUTLS_VERSION_MINOR < 6) -#define GNUTLS_CIPHER_AES_128_CFB8 29 -#define GNUTLS_CIPHER_AES_192_CFB8 30 -#define GNUTLS_CIPHER_AES_256_CFB8 31 - -#define GNUTLS_PK_RSA_PSS 6 -#define GNUTLS_PRIVKEY_SIGN_FLAG_RSA_PSS (1 << 7) -typedef struct gnutls_x509_spki_st *gnutls_x509_spki_t; -#endif - -#if GUTLS_VERSION_MAJOR < 3 || (GNUTLS_VERSION_MAJOR == 3 && GNUTLS_VERSION_MINOR < 8) -#define GNUTLS_KEYGEN_DH 4 -#endif - -union key_data -{ - gnutls_cipher_hd_t cipher; - struct - { - gnutls_privkey_t privkey; - gnutls_pubkey_t pubkey; - gnutls_dh_params_t dh_params; - } a; -}; -C_ASSERT( sizeof(union key_data) <= sizeof(((struct key *)0)->private) ); - -static union key_data *key_data( struct key *key ) -{ - return (union key_data *)key->private; -} - -/* Not present in gnutls version < 3.0 */ -static int (*pgnutls_cipher_tag)(gnutls_cipher_hd_t, void *, size_t); -static int (*pgnutls_cipher_add_auth)(gnutls_cipher_hd_t, const void *, size_t); -static gnutls_sign_algorithm_t (*pgnutls_pk_to_sign)(gnutls_pk_algorithm_t, gnutls_digest_algorithm_t); -static int (*pgnutls_pubkey_import_ecc_raw)(gnutls_pubkey_t, gnutls_ecc_curve_t, - const gnutls_datum_t *, const gnutls_datum_t *); -static int (*pgnutls_pubkey_export_ecc_raw)(gnutls_pubkey_t key, gnutls_ecc_curve_t *curve, - gnutls_datum_t *x, gnutls_datum_t *y); -static int (*pgnutls_privkey_import_ecc_raw)(gnutls_privkey_t, gnutls_ecc_curve_t, const gnutls_datum_t *, - const gnutls_datum_t *, const gnutls_datum_t *); -static int (*pgnutls_pubkey_verify_hash2)(gnutls_pubkey_t, gnutls_sign_algorithm_t, unsigned int, - const gnutls_datum_t *, const gnutls_datum_t *); -static int (*pgnutls_pubkey_encrypt_data)(gnutls_pubkey_t, unsigned int flags, const gnutls_datum_t *, - gnutls_datum_t *); - -/* Not present in gnutls version < 2.11.0 */ -static int (*pgnutls_pubkey_import_rsa_raw)(gnutls_pubkey_t, const gnutls_datum_t *, const gnutls_datum_t *); - -/* Not present in gnutls version < 2.12.0 */ -static int (*pgnutls_pubkey_import_dsa_raw)(gnutls_pubkey_t, const gnutls_datum_t *, const gnutls_datum_t *, - const gnutls_datum_t *, const gnutls_datum_t *); -static int (*pgnutls_pubkey_import_privkey)(gnutls_pubkey_t, gnutls_privkey_t, unsigned int, unsigned int); -static int (*pgnutls_privkey_decrypt_data)(gnutls_privkey_t, unsigned int flags, const gnutls_datum_t *, - gnutls_datum_t *); - -/* Not present in gnutls version < 3.3.0 */ -static int (*pgnutls_pubkey_export_dsa_raw)(gnutls_pubkey_t, gnutls_datum_t *, gnutls_datum_t *, gnutls_datum_t *, - gnutls_datum_t *); -static int (*pgnutls_pubkey_export_rsa_raw)(gnutls_pubkey_t, gnutls_datum_t *, gnutls_datum_t *); -static int (*pgnutls_privkey_export_ecc_raw)(gnutls_privkey_t, gnutls_ecc_curve_t *, - gnutls_datum_t *, gnutls_datum_t *, gnutls_datum_t *); -static int (*pgnutls_privkey_export_rsa_raw)(gnutls_privkey_t, gnutls_datum_t *, gnutls_datum_t *, gnutls_datum_t *, - gnutls_datum_t *, gnutls_datum_t *, gnutls_datum_t *, gnutls_datum_t *, - gnutls_datum_t *); -static int (*pgnutls_privkey_export_dsa_raw)(gnutls_privkey_t, gnutls_datum_t *, gnutls_datum_t *, gnutls_datum_t *, - gnutls_datum_t *, gnutls_datum_t *); -static int (*pgnutls_privkey_import_rsa_raw)(gnutls_privkey_t, const gnutls_datum_t *, const gnutls_datum_t *, - const gnutls_datum_t *, const gnutls_datum_t *, const gnutls_datum_t *, - const gnutls_datum_t *, const gnutls_datum_t *, const gnutls_datum_t *); - -/* Not present in gnutls version < 3.5.0 */ -static int (*pgnutls_privkey_generate2)(gnutls_privkey_t, gnutls_pk_algorithm_t, unsigned int, unsigned int, - const gnutls_keygen_data_st *, unsigned); - -/* Not present in gnutls version < 3.6.0 */ -static int (*pgnutls_decode_rs_value)(const gnutls_datum_t *, gnutls_datum_t *, gnutls_datum_t *); -static int (*pgnutls_x509_spki_init)(gnutls_x509_spki_t *); -static void (*pgnutls_x509_spki_deinit)(gnutls_x509_spki_t); -static void (*pgnutls_x509_spki_set_rsa_pss_params)(gnutls_x509_spki_t, gnutls_digest_algorithm_t, unsigned int); -static int (*pgnutls_pubkey_set_spki)(gnutls_pubkey_t, const gnutls_x509_spki_t, unsigned int); -static int (*pgnutls_privkey_set_spki)(gnutls_privkey_t, const gnutls_x509_spki_t, unsigned int); - -/* Not present in gnutls version < 3.8.2 */ -static int (*pgnutls_privkey_derive_secret)(gnutls_privkey_t, gnutls_pubkey_t, const gnutls_datum_t *, - gnutls_datum_t *, unsigned int); -static int (*pgnutls_privkey_export_dh_raw)(gnutls_privkey_t, gnutls_dh_params_t, gnutls_datum_t *, gnutls_datum_t *, - unsigned int); -static int (*pgnutls_pubkey_export_dh_raw)(gnutls_pubkey_t, gnutls_dh_params_t, gnutls_datum_t *, unsigned); -static int (*pgnutls_privkey_import_dh_raw)(gnutls_privkey_t, const gnutls_dh_params_t, const gnutls_datum_t *, - const gnutls_datum_t *); -static int (*pgnutls_pubkey_import_dh_raw)(gnutls_pubkey_t, const gnutls_dh_params_t, const gnutls_datum_t *); - -/* Not present in gnutls version < 3.8.4 */ -static int (*pgnutls_x509_spki_set_rsa_oaep_params)(gnutls_x509_spki_t, gnutls_digest_algorithm_t, gnutls_datum_t *); - -static void *libgnutls_handle; -#define MAKE_FUNCPTR(f) static typeof(f) * p##f -MAKE_FUNCPTR(gnutls_cipher_decrypt2); -MAKE_FUNCPTR(gnutls_cipher_deinit); -MAKE_FUNCPTR(gnutls_cipher_encrypt2); -MAKE_FUNCPTR(gnutls_cipher_init); -MAKE_FUNCPTR(gnutls_dh_params_deinit); -MAKE_FUNCPTR(gnutls_dh_params_export_raw); -MAKE_FUNCPTR(gnutls_dh_params_import_raw); -MAKE_FUNCPTR(gnutls_dh_params_init); -MAKE_FUNCPTR(gnutls_global_deinit); -MAKE_FUNCPTR(gnutls_global_init); -MAKE_FUNCPTR(gnutls_global_set_log_function); -MAKE_FUNCPTR(gnutls_global_set_log_level); -MAKE_FUNCPTR(gnutls_perror); -MAKE_FUNCPTR(gnutls_privkey_decrypt_data); -MAKE_FUNCPTR(gnutls_privkey_deinit); -MAKE_FUNCPTR(gnutls_privkey_import_dsa_raw); -MAKE_FUNCPTR(gnutls_privkey_init); -MAKE_FUNCPTR(gnutls_privkey_sign_hash); -MAKE_FUNCPTR(gnutls_pubkey_deinit); -MAKE_FUNCPTR(gnutls_pubkey_encrypt_data); -MAKE_FUNCPTR(gnutls_pubkey_import_privkey); -MAKE_FUNCPTR(gnutls_pubkey_init); -#undef MAKE_FUNCPTR - -static int compat_gnutls_cipher_tag(gnutls_cipher_hd_t handle, void *tag, size_t tag_size) -{ - return GNUTLS_E_UNKNOWN_CIPHER_TYPE; -} - -static int compat_gnutls_cipher_add_auth(gnutls_cipher_hd_t handle, const void *ptext, size_t ptext_size) -{ - return GNUTLS_E_UNKNOWN_CIPHER_TYPE; -} - -static int compat_gnutls_pubkey_import_ecc_raw(gnutls_pubkey_t key, gnutls_ecc_curve_t curve, - const gnutls_datum_t *x, const gnutls_datum_t *y) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_pubkey_export_ecc_raw(gnutls_pubkey_t key, gnutls_ecc_curve_t *curve, - gnutls_datum_t *x, gnutls_datum_t *y) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_pubkey_export_dsa_raw(gnutls_pubkey_t key, gnutls_datum_t *p, gnutls_datum_t *q, - gnutls_datum_t *g, gnutls_datum_t *y) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_pubkey_export_rsa_raw(gnutls_pubkey_t key, gnutls_datum_t *m, gnutls_datum_t *e) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_export_rsa_raw(gnutls_privkey_t key, gnutls_datum_t *m, gnutls_datum_t *e, - gnutls_datum_t *d, gnutls_datum_t *p, gnutls_datum_t *q, - gnutls_datum_t *u, gnutls_datum_t *e1, gnutls_datum_t *e2) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_export_ecc_raw(gnutls_privkey_t key, gnutls_ecc_curve_t *curve, - gnutls_datum_t *x, gnutls_datum_t *y, gnutls_datum_t *k) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_import_ecc_raw(gnutls_privkey_t key, gnutls_ecc_curve_t curve, - const gnutls_datum_t *x, const gnutls_datum_t *y, - const gnutls_datum_t *k) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_export_dsa_raw(gnutls_privkey_t key, gnutls_datum_t *p, gnutls_datum_t *q, - gnutls_datum_t *g, gnutls_datum_t *y, gnutls_datum_t *x) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static gnutls_sign_algorithm_t compat_gnutls_pk_to_sign(gnutls_pk_algorithm_t pk, gnutls_digest_algorithm_t hash) -{ - return GNUTLS_SIGN_UNKNOWN; -} - -static int compat_gnutls_pubkey_verify_hash2(gnutls_pubkey_t key, gnutls_sign_algorithm_t algo, - unsigned int flags, const gnutls_datum_t *hash, - const gnutls_datum_t *signature) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_pubkey_import_rsa_raw(gnutls_pubkey_t key, const gnutls_datum_t *m, const gnutls_datum_t *e) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_pubkey_import_dsa_raw(gnutls_pubkey_t key, const gnutls_datum_t *p, const gnutls_datum_t *q, - const gnutls_datum_t *g, const gnutls_datum_t *y) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_decode_rs_value(const gnutls_datum_t * sig_value, gnutls_datum_t * r, gnutls_datum_t * s) -{ - return GNUTLS_E_INTERNAL_ERROR; -} - -static int compat_gnutls_privkey_import_rsa_raw(gnutls_privkey_t key, const gnutls_datum_t *m, const gnutls_datum_t *e, - const gnutls_datum_t *d, const gnutls_datum_t *p, const gnutls_datum_t *q, - const gnutls_datum_t *u, const gnutls_datum_t *e1, const gnutls_datum_t *e2) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_decrypt_data(gnutls_privkey_t key, unsigned int flags, const gnutls_datum_t *cipher_text, - gnutls_datum_t *plain_text) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_pubkey_encrypt_data(gnutls_pubkey_t key, unsigned int flags, const gnutls_datum_t *cipher_text, - gnutls_datum_t *plain_text) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_x509_spki_init(gnutls_x509_spki_t *spki) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static void compat_gnutls_x509_spki_deinit(gnutls_x509_spki_t spki) -{ -} - -static int compat_gnutls_x509_spki_set_rsa_oaep_params(gnutls_x509_spki_t spki, gnutls_digest_algorithm_t dig, - gnutls_datum_t *label) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static void compat_gnutls_x509_spki_set_rsa_pss_params(gnutls_x509_spki_t spki, gnutls_digest_algorithm_t dig, - unsigned int salt_size) -{ -} - -static int compat_gnutls_pubkey_set_spki(gnutls_pubkey_t key, const gnutls_x509_spki_t spki, unsigned int flags) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_set_spki(gnutls_privkey_t key, const gnutls_x509_spki_t spki, unsigned int flags) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_derive_secret(gnutls_privkey_t privkey, gnutls_pubkey_t pubkey, const gnutls_datum_t *nonce, - gnutls_datum_t *secret, unsigned int flags) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_export_dh_raw(gnutls_privkey_t privkey, gnutls_dh_params_t params, gnutls_datum_t *y, - gnutls_datum_t *x, unsigned int flags ) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_pubkey_export_dh_raw(gnutls_pubkey_t pubkey, gnutls_dh_params_t params, gnutls_datum_t *y, - unsigned flags) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_import_dh_raw(gnutls_privkey_t privkey, const gnutls_dh_params_t params, - const gnutls_datum_t *y, const gnutls_datum_t *x) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_pubkey_import_dh_raw(gnutls_pubkey_t pubkey, const gnutls_dh_params_t params, - const gnutls_datum_t *y) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static int compat_gnutls_privkey_generate2(gnutls_privkey_t privkey, gnutls_pk_algorithm_t alg, unsigned int bits, - unsigned int flags, const gnutls_keygen_data_st *data, unsigned data_size) -{ - return GNUTLS_E_UNKNOWN_PK_ALGORITHM; -} - -static void gnutls_log( int level, const char *msg ) -{ - TRACE( "<%d> %s", level, msg ); -} - -static NTSTATUS gnutls_process_attach( void *args ) -{ - const char *env_str; - int ret; - - if ((env_str = getenv("GNUTLS_SYSTEM_PRIORITY_FILE"))) - { - WARN("GNUTLS_SYSTEM_PRIORITY_FILE is %s.\n", debugstr_a(env_str)); - } - else - { - WARN("Setting GNUTLS_SYSTEM_PRIORITY_FILE to \"/dev/null\".\n"); - setenv("GNUTLS_SYSTEM_PRIORITY_FILE", "/dev/null", 0); - } - - if (!(libgnutls_handle = dlopen( SONAME_LIBGNUTLS, RTLD_NOW ))) - { - ERR_(winediag)( "failed to load libgnutls, no support for encryption\n" ); - return STATUS_DLL_NOT_FOUND; - } - -#define LOAD_FUNCPTR(f) \ - if (!(p##f = dlsym( libgnutls_handle, #f ))) \ - { \ - ERR( "failed to load %s\n", #f ); \ - goto fail; \ - } - - LOAD_FUNCPTR(gnutls_cipher_decrypt2) - LOAD_FUNCPTR(gnutls_cipher_deinit) - LOAD_FUNCPTR(gnutls_cipher_encrypt2) - LOAD_FUNCPTR(gnutls_cipher_init) - LOAD_FUNCPTR(gnutls_dh_params_deinit) - LOAD_FUNCPTR(gnutls_dh_params_export_raw) - LOAD_FUNCPTR(gnutls_dh_params_import_raw) - LOAD_FUNCPTR(gnutls_dh_params_init) - LOAD_FUNCPTR(gnutls_global_deinit) - LOAD_FUNCPTR(gnutls_global_init) - LOAD_FUNCPTR(gnutls_global_set_log_function) - LOAD_FUNCPTR(gnutls_global_set_log_level) - LOAD_FUNCPTR(gnutls_perror) - LOAD_FUNCPTR(gnutls_privkey_deinit); - LOAD_FUNCPTR(gnutls_privkey_import_dsa_raw); - LOAD_FUNCPTR(gnutls_privkey_init); - LOAD_FUNCPTR(gnutls_privkey_sign_hash); - LOAD_FUNCPTR(gnutls_pubkey_deinit); - LOAD_FUNCPTR(gnutls_pubkey_import_privkey); - LOAD_FUNCPTR(gnutls_pubkey_init); -#undef LOAD_FUNCPTR - -#define LOAD_FUNCPTR_OPT(f) \ - if (!(p##f = dlsym( libgnutls_handle, #f ))) \ - { \ - WARN( "failed to load %s\n", #f ); \ - p##f = compat_##f; \ - } - - LOAD_FUNCPTR_OPT(gnutls_cipher_tag) - LOAD_FUNCPTR_OPT(gnutls_cipher_add_auth) - LOAD_FUNCPTR_OPT(gnutls_decode_rs_value) - LOAD_FUNCPTR_OPT(gnutls_pk_to_sign) - LOAD_FUNCPTR_OPT(gnutls_privkey_decrypt_data) - LOAD_FUNCPTR_OPT(gnutls_privkey_derive_secret) - LOAD_FUNCPTR_OPT(gnutls_privkey_export_dh_raw) - LOAD_FUNCPTR_OPT(gnutls_privkey_export_dsa_raw) - LOAD_FUNCPTR_OPT(gnutls_privkey_export_ecc_raw) - LOAD_FUNCPTR_OPT(gnutls_privkey_export_rsa_raw) - LOAD_FUNCPTR_OPT(gnutls_privkey_generate2) - LOAD_FUNCPTR_OPT(gnutls_privkey_import_dh_raw) - LOAD_FUNCPTR_OPT(gnutls_privkey_import_ecc_raw) - LOAD_FUNCPTR_OPT(gnutls_privkey_import_rsa_raw) - LOAD_FUNCPTR_OPT(gnutls_privkey_set_spki) - LOAD_FUNCPTR_OPT(gnutls_pubkey_encrypt_data) - LOAD_FUNCPTR_OPT(gnutls_pubkey_export_dh_raw) - LOAD_FUNCPTR_OPT(gnutls_pubkey_export_dsa_raw) - LOAD_FUNCPTR_OPT(gnutls_pubkey_export_ecc_raw) - LOAD_FUNCPTR_OPT(gnutls_pubkey_export_rsa_raw) - LOAD_FUNCPTR_OPT(gnutls_pubkey_import_dh_raw) - LOAD_FUNCPTR_OPT(gnutls_pubkey_import_dsa_raw) - LOAD_FUNCPTR_OPT(gnutls_pubkey_import_ecc_raw) - LOAD_FUNCPTR_OPT(gnutls_pubkey_import_rsa_raw) - LOAD_FUNCPTR_OPT(gnutls_pubkey_set_spki) - LOAD_FUNCPTR_OPT(gnutls_pubkey_verify_hash2) - LOAD_FUNCPTR_OPT(gnutls_x509_spki_deinit) - LOAD_FUNCPTR_OPT(gnutls_x509_spki_init) - LOAD_FUNCPTR_OPT(gnutls_x509_spki_set_rsa_oaep_params) - LOAD_FUNCPTR_OPT(gnutls_x509_spki_set_rsa_pss_params) - -#undef LOAD_FUNCPTR_OPT - - if ((ret = pgnutls_global_init()) != GNUTLS_E_SUCCESS) - { - pgnutls_perror( ret ); - goto fail; - } - - if (TRACE_ON( bcrypt )) - { - char *env = getenv("GNUTLS_DEBUG_LEVEL"); - int level = env ? atoi(env) : 4; - pgnutls_global_set_log_level(level); - pgnutls_global_set_log_function( gnutls_log ); - } - - return STATUS_SUCCESS; - -fail: - dlclose( libgnutls_handle ); - libgnutls_handle = NULL; - return STATUS_DLL_NOT_FOUND; -} - -static NTSTATUS gnutls_process_detach( void *args ) -{ - if (libgnutls_handle) - { - if (TRACE_ON( bcrypt )) - pgnutls_global_set_log_function( NULL ); - pgnutls_global_deinit(); - dlclose( libgnutls_handle ); - libgnutls_handle = NULL; - } - return STATUS_SUCCESS; -} - -struct buffer -{ - BYTE *buffer; - DWORD length; - DWORD pos; - BOOL error; -}; - -static void buffer_init( struct buffer *buffer ) -{ - buffer->buffer = NULL; - buffer->length = 0; - buffer->pos = 0; - buffer->error = FALSE; -} - -static void buffer_free( struct buffer *buffer ) -{ - free( buffer->buffer ); -} - -static void buffer_append( struct buffer *buffer, BYTE *data, DWORD len ) -{ - if (!len) return; - - if (buffer->pos + len > buffer->length) - { - DWORD new_length = max( max( buffer->pos + len, buffer->length * 2 ), 64 ); - BYTE *new_buffer; - - if (!(new_buffer = realloc( buffer->buffer, new_length ))) - { - ERR( "out of memory\n" ); - buffer->error = TRUE; - return; - } - - buffer->buffer = new_buffer; - buffer->length = new_length; - } - - memcpy( &buffer->buffer[buffer->pos], data, len ); - buffer->pos += len; -} - -static void buffer_append_byte( struct buffer *buffer, BYTE value ) -{ - buffer_append( buffer, &value, sizeof(value) ); -} - -static void buffer_append_asn1_length( struct buffer *buffer, DWORD length ) -{ - DWORD num_bytes; - - if (length < 128) - { - buffer_append_byte( buffer, length ); - return; - } - - if (length <= 0xff) num_bytes = 1; - else if (length <= 0xffff) num_bytes = 2; - else if (length <= 0xffffff) num_bytes = 3; - else num_bytes = 4; - - buffer_append_byte( buffer, 0x80 | num_bytes ); - while (num_bytes--) buffer_append_byte( buffer, length >> (num_bytes * 8) ); -} - -static void buffer_append_asn1_integer( struct buffer *buffer, BYTE *data, DWORD len ) -{ - DWORD leading_zero = (*data & 0x80) != 0; - - buffer_append_byte( buffer, 0x02 ); /* tag */ - buffer_append_asn1_length( buffer, len + leading_zero ); - if (leading_zero) buffer_append_byte( buffer, 0 ); - buffer_append( buffer, data, len ); -} - -static void buffer_append_asn1_sequence( struct buffer *buffer, struct buffer *content ) -{ - if (content->error) - { - buffer->error = TRUE; - return; - } - - buffer_append_byte( buffer, 0x30 ); /* tag */ - buffer_append_asn1_length( buffer, content->pos ); - buffer_append( buffer, content->buffer, content->pos ); -} - -static void buffer_append_asn1_r_s( struct buffer *buffer, BYTE *r, DWORD r_len, BYTE *s, DWORD s_len ) -{ - struct buffer value; - - buffer_init( &value ); - buffer_append_asn1_integer( &value, r, r_len ); - buffer_append_asn1_integer( &value, s, s_len ); - buffer_append_asn1_sequence( buffer, &value ); - buffer_free( &value ); -} - -static gnutls_cipher_algorithm_t get_gnutls_cipher( const struct key *key ) -{ - switch (key->alg_id) - { - case ALG_ID_3DES: - WARN( "handle block size\n" ); - switch (key->u.s.mode) - { - case CHAIN_MODE_CBC: - return GNUTLS_CIPHER_3DES_CBC; - default: - break; - } - FIXME( "3DES mode %u with key length %u not supported\n", key->u.s.mode, key->u.s.secret_len ); - return GNUTLS_CIPHER_UNKNOWN; - - case ALG_ID_AES: - WARN( "handle block size\n" ); - switch (key->u.s.mode) - { - case CHAIN_MODE_GCM: - if (key->u.s.secret_len == 16) return GNUTLS_CIPHER_AES_128_GCM; - if (key->u.s.secret_len == 32) return GNUTLS_CIPHER_AES_256_GCM; - break; - case CHAIN_MODE_ECB: /* can be emulated with CBC + empty IV */ - case CHAIN_MODE_CBC: - if (key->u.s.secret_len == 16) return GNUTLS_CIPHER_AES_128_CBC; - if (key->u.s.secret_len == 24) return GNUTLS_CIPHER_AES_192_CBC; - if (key->u.s.secret_len == 32) return GNUTLS_CIPHER_AES_256_CBC; - break; - case CHAIN_MODE_CFB: - if (key->u.s.secret_len == 16) return GNUTLS_CIPHER_AES_128_CFB8; - if (key->u.s.secret_len == 24) return GNUTLS_CIPHER_AES_192_CFB8; - if (key->u.s.secret_len == 32) return GNUTLS_CIPHER_AES_256_CFB8; - break; - default: - break; - } - FIXME( "AES mode %u with key length %u not supported\n", key->u.s.mode, key->u.s.secret_len ); - return GNUTLS_CIPHER_UNKNOWN; - - default: - FIXME( "algorithm %u not supported\n", key->alg_id ); - return GNUTLS_CIPHER_UNKNOWN; - } -} - -static NTSTATUS key_symmetric_vector_reset( void *args ) -{ - struct key *key = args; - - if (!key_data(key)->cipher) return STATUS_SUCCESS; - TRACE( "invalidating cipher handle\n" ); - pgnutls_cipher_deinit( key_data(key)->cipher ); - key_data(key)->cipher = NULL; - return STATUS_SUCCESS; -} - -static NTSTATUS init_cipher_handle( struct key *key ) -{ - gnutls_cipher_algorithm_t cipher; - gnutls_datum_t secret, vector; - int ret; - - if (key_data(key)->cipher) return STATUS_SUCCESS; - if ((cipher = get_gnutls_cipher( key )) == GNUTLS_CIPHER_UNKNOWN) return STATUS_NOT_SUPPORTED; - - secret.data = key->u.s.secret; - secret.size = key->u.s.secret_len; - - vector.data = key->u.s.vector; - vector.size = key->u.s.vector_len; - - if ((ret = pgnutls_cipher_init( &key_data(key)->cipher, cipher, &secret, key->u.s.vector ? &vector : NULL ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - return STATUS_SUCCESS; -} - -static NTSTATUS key_symmetric_set_auth_data( void *args ) -{ - const struct key_symmetric_set_auth_data_params *params = args; - NTSTATUS status; - int ret; - - if (!params->auth_data) return STATUS_SUCCESS; - if ((status = init_cipher_handle( params->key ))) return status; - - if ((ret = pgnutls_cipher_add_auth( key_data(params->key)->cipher, params->auth_data, params->len ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - return STATUS_SUCCESS; -} - -static NTSTATUS key_symmetric_encrypt( void *args ) -{ - const struct key_symmetric_encrypt_params *params = args; - NTSTATUS status; - int ret; - - if ((status = init_cipher_handle( params->key ))) return status; - - if ((ret = pgnutls_cipher_encrypt2( key_data(params->key)->cipher, params->input, params->input_len, - params->output, params->output_len ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - return STATUS_SUCCESS; -} - -static NTSTATUS key_symmetric_decrypt( void *args ) -{ - const struct key_symmetric_decrypt_params *params = args; - NTSTATUS status; - int ret; - - if ((status = init_cipher_handle( params->key ))) return status; - - if ((ret = pgnutls_cipher_decrypt2( key_data(params->key)->cipher, params->input, params->input_len, - params->output, params->output_len ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - return STATUS_SUCCESS; -} - -static NTSTATUS key_symmetric_get_tag( void *args ) -{ - const struct key_symmetric_get_tag_params *params = args; - NTSTATUS status; - int ret; - - if ((status = init_cipher_handle( params->key ))) return status; - - if ((ret = pgnutls_cipher_tag( key_data(params->key)->cipher, params->tag, params->len ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - return STATUS_SUCCESS; -} - -static NTSTATUS key_symmetric_destroy( void *args ) -{ - struct key *key = args; - - if (key_data(key)->cipher) pgnutls_cipher_deinit( key_data(key)->cipher ); - return STATUS_SUCCESS; -} - -static ULONG export_gnutls_datum( UCHAR *buffer, ULONG buflen, gnutls_datum_t *d, BOOL zero_pad ) -{ - ULONG size = d->size; - UCHAR *src = d->data; - ULONG offset = 0; - - assert( size <= buflen + 1 ); - if (size == buflen + 1) - { - assert( !src[0] ); - src++; - size--; - } - if (zero_pad) - { - offset = buflen - size; - if (buffer) memset( buffer, 0, offset ); - size = buflen; - } - - if (buffer) memcpy( buffer + offset, src, size - offset ); - return size; -} - -#define EXPORT_SIZE(d,l,p) export_gnutls_datum( NULL, l, &d, p ) -static NTSTATUS key_export_rsa_public( struct key *key, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BCRYPT_RSAKEY_BLOB *rsa_blob = (BCRYPT_RSAKEY_BLOB *)buf; - gnutls_datum_t m, e; - ULONG size = len_from_bitlen( key->u.a.bitlen ); - UCHAR *dst; - int ret; - - if (key_data(key)->a.pubkey) - ret = pgnutls_pubkey_export_rsa_raw( key_data(key)->a.pubkey, &m, &e ); - else - return STATUS_INVALID_PARAMETER; - - if (ret) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - *ret_len = sizeof(*rsa_blob) + EXPORT_SIZE( e, size, 0 ) + EXPORT_SIZE( m, size, 1 ); - if (len >= *ret_len && buf) - { - dst = (UCHAR *)(rsa_blob + 1); - rsa_blob->cbPublicExp = export_gnutls_datum( dst, size, &e, 0 ); - - dst += rsa_blob->cbPublicExp; - rsa_blob->cbModulus = export_gnutls_datum( dst, size, &m, 1 ); - - rsa_blob->Magic = BCRYPT_RSAPUBLIC_MAGIC; - rsa_blob->BitLength = key->u.a.bitlen; - rsa_blob->cbPrime1 = 0; - rsa_blob->cbPrime2 = 0; - } - - free( e.data ); free( m.data ); - return STATUS_SUCCESS; -} - -static NTSTATUS key_export_ecc_public( struct key *key, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BCRYPT_ECCKEY_BLOB *ecc_blob = (BCRYPT_ECCKEY_BLOB *)buf; - gnutls_ecc_curve_t curve; - gnutls_datum_t x, y; - DWORD magic, size; - UCHAR *dst; - int ret; - - switch (key->alg_id) - { - case ALG_ID_ECDH: - switch (key->u.a.curve_id) - { - case ECC_CURVE_25519: - magic = BCRYPT_ECDH_PUBLIC_GENERIC_MAGIC; - size = 32; - break; - case ECC_CURVE_P256R1: - magic = BCRYPT_ECDH_PUBLIC_P256_MAGIC; - size = 32; - break; - case ECC_CURVE_P384R1: - magic = BCRYPT_ECDH_PUBLIC_P384_MAGIC; - size = 48; - break; - case ECC_CURVE_P521R1: - magic = BCRYPT_ECDH_PUBLIC_P521_MAGIC; - size = 66; - break; - default: - FIXME( "unsupported curve %u\n", key->u.a.curve_id ); - return STATUS_NOT_IMPLEMENTED; - } - break; - - case ALG_ID_ECDH_P256: - magic = BCRYPT_ECDH_PUBLIC_P256_MAGIC; - size = 32; - break; - - case ALG_ID_ECDH_P384: - magic = BCRYPT_ECDH_PUBLIC_P384_MAGIC; - size = 48; - break; - - case ALG_ID_ECDH_P521: - magic = BCRYPT_ECDH_PUBLIC_P521_MAGIC; - size = 66; - break; - - case ALG_ID_ECDSA: - switch (key->u.a.curve_id) - { - case ECC_CURVE_P256R1: - magic = BCRYPT_ECDSA_PUBLIC_P256_MAGIC; - size = 32; - break; - case ECC_CURVE_P384R1: - magic = BCRYPT_ECDSA_PUBLIC_P384_MAGIC; - size = 48; - break; - case ECC_CURVE_P521R1: - magic = BCRYPT_ECDSA_PUBLIC_P521_MAGIC; - size = 66; - break; - default: - FIXME( "unsupported curve %u\n", key->u.a.curve_id ); - return STATUS_NOT_IMPLEMENTED; - } - break; - - case ALG_ID_ECDSA_P256: - magic = BCRYPT_ECDSA_PUBLIC_P256_MAGIC; - size = 32; - break; - - case ALG_ID_ECDSA_P384: - magic = BCRYPT_ECDSA_PUBLIC_P384_MAGIC; - size = 48; - break; - - case ALG_ID_ECDSA_P521: - magic = BCRYPT_ECDSA_PUBLIC_P521_MAGIC; - size = 66; - break; - - default: - FIXME( "algorithm %u not supported\n", key->alg_id ); - return STATUS_NOT_IMPLEMENTED; - } - - if (key_data(key)->a.pubkey) - ret = pgnutls_pubkey_export_ecc_raw( key_data(key)->a.pubkey, &curve, &x, &y ); - else - return STATUS_INVALID_PARAMETER; - - if (ret) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if (curve != GNUTLS_ECC_CURVE_X25519 && curve != GNUTLS_ECC_CURVE_SECP256R1 && - curve != GNUTLS_ECC_CURVE_SECP384R1 && curve != GNUTLS_ECC_CURVE_SECP521R1) - { - FIXME( "curve %u not supported\n", curve ); - free( x.data ); free( y.data ); - return STATUS_NOT_IMPLEMENTED; - } - - *ret_len = sizeof(*ecc_blob) + EXPORT_SIZE( x, size, 1 ) + EXPORT_SIZE( y, size, 1 ); - if (len >= *ret_len && buf) - { - ecc_blob->dwMagic = magic; - ecc_blob->cbKey = size; - - dst = (UCHAR *)(ecc_blob + 1); - dst += export_gnutls_datum( dst, size, &x, 1 ); - export_gnutls_datum( dst, size, &y, 1 ); - } - - free( x.data ); free( y.data ); - return STATUS_SUCCESS; -} - -static NTSTATUS key_export_dsa_public( struct key *key, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BCRYPT_DSA_KEY_BLOB *dsa_blob = (BCRYPT_DSA_KEY_BLOB *)buf; - gnutls_datum_t p, q, g, y; - ULONG size = len_from_bitlen( key->u.a.bitlen ); - NTSTATUS status = STATUS_SUCCESS; - UCHAR *dst; - int ret; - - if (key->u.a.bitlen > 1024) - { - FIXME( "bitlen > 1024 not supported\n" ); - return STATUS_NOT_IMPLEMENTED; - } - - if (key_data(key)->a.pubkey) - ret = pgnutls_pubkey_export_dsa_raw( key_data(key)->a.pubkey, &p, &q, &g, &y ); - else - return STATUS_INVALID_PARAMETER; - - if (ret) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if (EXPORT_SIZE( q, sizeof(dsa_blob->q), 1 ) > sizeof(dsa_blob->q)) - { - status = STATUS_INVALID_PARAMETER; - goto done; - } - - *ret_len = sizeof(*dsa_blob) + EXPORT_SIZE( p, size, 1 ) + EXPORT_SIZE( g, size, 1 ) + EXPORT_SIZE( y, size, 1 ); - if (len >= *ret_len && buf) - { - dst = (UCHAR *)(dsa_blob + 1); - dst += export_gnutls_datum( dst, size, &p, 1 ); - dst += export_gnutls_datum( dst, size, &g, 1 ); - export_gnutls_datum( dst, size, &y, 1 ); - - dst = dsa_blob->q; - export_gnutls_datum( dst, sizeof(dsa_blob->q), &q, 1 ); - - dsa_blob->dwMagic = BCRYPT_DSA_PUBLIC_MAGIC; - dsa_blob->cbKey = size; - memset( dsa_blob->Count, 0, sizeof(dsa_blob->Count) ); /* FIXME */ - memset( dsa_blob->Seed, 0, sizeof(dsa_blob->Seed) ); /* FIXME */ - } - -done: - free( p.data ); free( q.data ); free( g.data ); free( y.data ); - return status; -} - -static void reverse_bytes( UCHAR *buf, ULONG len ) -{ - unsigned int i; - UCHAR tmp; - - for (i = 0; i < len / 2; ++i) - { - tmp = buf[i]; - buf[i] = buf[len - i - 1]; - buf[len - i - 1] = tmp; - } -} - -#define Q_SIZE 20 -static NTSTATUS key_export_dsa_capi_public( struct key *key, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BLOBHEADER *hdr = (BLOBHEADER *)buf; - DSSPUBKEY *dsskey; - gnutls_datum_t p, q, g, y; - ULONG size = len_from_bitlen( key->u.a.bitlen ); - NTSTATUS status = STATUS_SUCCESS; - UCHAR *dst; - int ret; - - if (key->u.a.bitlen > 1024) - { - FIXME( "bitlen > 1024 not supported\n" ); - return STATUS_NOT_IMPLEMENTED; - } - - if (key_data(key)->a.pubkey) - ret = pgnutls_pubkey_export_dsa_raw( key_data(key)->a.pubkey, &p, &q, &g, &y ); - else if (key_data(key)->a.privkey) - ret = pgnutls_privkey_export_dsa_raw( key_data(key)->a.privkey, &p, &q, &g, &y, NULL ); - else - return STATUS_INVALID_PARAMETER; - - if (ret) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if (EXPORT_SIZE( q, Q_SIZE, 1 ) > Q_SIZE) - { - status = STATUS_INVALID_PARAMETER; - goto done; - } - - *ret_len = sizeof(*hdr) + sizeof(*dsskey) + sizeof(key->u.a.dss_seed) + - EXPORT_SIZE( p, size, 1 ) + Q_SIZE + EXPORT_SIZE( g, size, 1 ) + EXPORT_SIZE( y, size, 1 ); - if (len >= *ret_len && buf) - { - hdr->bType = PUBLICKEYBLOB; - hdr->bVersion = 2; - hdr->reserved = 0; - hdr->aiKeyAlg = CALG_DSS_SIGN; - - dsskey = (DSSPUBKEY *)(hdr + 1); - dsskey->magic = MAGIC_DSS1; - dsskey->bitlen = key->u.a.bitlen; - - dst = (UCHAR *)(dsskey + 1); - export_gnutls_datum( dst, size, &p, 1 ); - reverse_bytes( dst, size ); - dst += size; - - export_gnutls_datum( dst, Q_SIZE, &q, 1 ); - reverse_bytes( dst, Q_SIZE ); - dst += Q_SIZE; - - export_gnutls_datum( dst, size, &g, 1 ); - reverse_bytes( dst, size ); - dst += size; - - export_gnutls_datum( dst, size, &y, 1 ); - reverse_bytes( dst, size ); - dst += size; - - memcpy( dst, &key->u.a.dss_seed, sizeof(key->u.a.dss_seed) ); - } - -done: - free( p.data ); free( q.data ); free( g.data ); free( y.data ); - return status; -} - -static gnutls_privkey_t create_privkey( gnutls_pk_algorithm_t pk_alg, unsigned int bitlen, - const gnutls_keygen_data_st *data, unsigned int data_size ) -{ - gnutls_privkey_t privkey; - int ret; - - if ((ret = pgnutls_privkey_init( &privkey ))) - { - pgnutls_perror( ret ); - return NULL; - } - - if ((ret = pgnutls_privkey_generate2( privkey, pk_alg, bitlen, 0, data, data_size ))) - { - pgnutls_perror( ret ); - pgnutls_privkey_deinit( privkey ); - return NULL; - } - - return privkey; -} - -static gnutls_pubkey_t create_pubkey_from_privkey( gnutls_privkey_t privkey ) -{ - gnutls_pubkey_t pubkey; - int ret; - - if ((ret = pgnutls_pubkey_init( &pubkey ))) - { - pgnutls_perror( ret ); - return NULL; - } - - if ((ret = pgnutls_pubkey_import_privkey( pubkey, privkey, 0, 0 ))) - { - pgnutls_perror( ret ); - pgnutls_pubkey_deinit( pubkey ); - return NULL; - } - - return pubkey; -} - -static gnutls_dh_params_t get_dh_params( gnutls_privkey_t privkey ) -{ - gnutls_dh_params_t params; - gnutls_datum_t x; - int ret; - - if ((ret = pgnutls_dh_params_init( ¶ms ))) - { - pgnutls_perror( ret ); - return NULL; - } - - if ((ret = pgnutls_privkey_export_dh_raw( privkey, params, NULL, &x, 0 ))) - { - pgnutls_perror( ret ); - pgnutls_dh_params_deinit( params ); - return NULL; - } - - free( x.data ); - return params; -} - -static NTSTATUS key_asymmetric_generate( void *args ) -{ - struct key *key = args; - gnutls_pk_algorithm_t pk_alg; - gnutls_privkey_t privkey; - gnutls_pubkey_t pubkey; - unsigned int bitlen; - - if (!libgnutls_handle) return STATUS_INTERNAL_ERROR; - if (key_data(key)->a.privkey) return STATUS_INVALID_HANDLE; - - switch (key->alg_id) - { - case ALG_ID_RSA: - case ALG_ID_RSA_SIGN: - pk_alg = GNUTLS_PK_RSA; - bitlen = key->u.a.bitlen; - break; - - case ALG_ID_DH: - pk_alg = GNUTLS_PK_DH; - bitlen = key->u.a.bitlen; - break; - - case ALG_ID_DSA: - pk_alg = GNUTLS_PK_DSA; - bitlen = key->u.a.bitlen; - break; - - case ALG_ID_ECDH: - case ALG_ID_ECDSA: - pk_alg = GNUTLS_PK_ECC; - switch (key->u.a.curve_id) - { - case ECC_CURVE_25519: - assert( key->alg_id == ALG_ID_ECDH ); - pk_alg = GNUTLS_PK_ECDH_X25519; - bitlen = GNUTLS_CURVE_TO_BITS( GNUTLS_ECC_CURVE_X25519 ); - break; - case ECC_CURVE_P256R1: - bitlen = GNUTLS_CURVE_TO_BITS( GNUTLS_ECC_CURVE_SECP256R1 ); - break; - case ECC_CURVE_P384R1: - bitlen = GNUTLS_CURVE_TO_BITS( GNUTLS_ECC_CURVE_SECP384R1 ); - break; - case ECC_CURVE_P521R1: - bitlen = GNUTLS_CURVE_TO_BITS( GNUTLS_ECC_CURVE_SECP521R1 ); - break; - default: - FIXME( "unsupported ECDH/ECDSA curve %u\n", key->u.a.curve_id ); - return STATUS_INVALID_PARAMETER; - } - break; - - case ALG_ID_ECDH_P256: - case ALG_ID_ECDSA_P256: - pk_alg = GNUTLS_PK_ECC; /* compatible with ECDSA and ECDH */ - bitlen = GNUTLS_CURVE_TO_BITS( GNUTLS_ECC_CURVE_SECP256R1 ); - break; - - case ALG_ID_ECDH_P384: - case ALG_ID_ECDSA_P384: - pk_alg = GNUTLS_PK_ECC; /* compatible with ECDSA and ECDH */ - bitlen = GNUTLS_CURVE_TO_BITS( GNUTLS_ECC_CURVE_SECP384R1 ); - break; - - case ALG_ID_ECDH_P521: - case ALG_ID_ECDSA_P521: - pk_alg = GNUTLS_PK_ECC; /* compatible with ECDSA and ECDH */ - bitlen = GNUTLS_CURVE_TO_BITS( GNUTLS_ECC_CURVE_SECP521R1 ); - break; - - default: - FIXME( "algorithm %u not supported\n", key->alg_id ); - return STATUS_NOT_SUPPORTED; - } - - if (key->alg_id == ALG_ID_DH && key_data(key)->a.dh_params) - { - gnutls_keygen_data_st data; - - data.type = GNUTLS_KEYGEN_DH; - data.data = (unsigned char *)key_data(key)->a.dh_params; - data.size = 0; - if (!(privkey = create_privkey( pk_alg, bitlen, &data, 1 ))) return STATUS_INTERNAL_ERROR; - } - else if (!(privkey = create_privkey( pk_alg, bitlen, NULL, 0 ))) return STATUS_INTERNAL_ERROR; - - if (key->alg_id == ALG_ID_DH && !key_data(key)->a.dh_params && - !(key_data(key)->a.dh_params = get_dh_params( privkey ))) - { - pgnutls_privkey_deinit( privkey ); - return STATUS_INTERNAL_ERROR; - } - - if (!(pubkey = create_pubkey_from_privkey( privkey ))) - { - pgnutls_privkey_deinit( privkey ); - return STATUS_INTERNAL_ERROR; - } - - key_data(key)->a.privkey = privkey; - key_data(key)->a.pubkey = pubkey; - return STATUS_SUCCESS; -} - -static NTSTATUS key_export_ecc( struct key *key, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BCRYPT_ECCKEY_BLOB *ecc_blob; - gnutls_ecc_curve_t curve; - gnutls_datum_t x, y, d; - DWORD magic, size; - UCHAR *dst; - int ret; - - switch (key->alg_id) - { - case ALG_ID_ECDH: - switch (key->u.a.curve_id) - { - case ECC_CURVE_25519: - magic = BCRYPT_ECDH_PRIVATE_GENERIC_MAGIC; - size = 32; - break; - case ECC_CURVE_P256R1: - magic = BCRYPT_ECDH_PRIVATE_P256_MAGIC; - size = 32; - break; - case ECC_CURVE_P384R1: - magic = BCRYPT_ECDH_PRIVATE_P384_MAGIC; - size = 48; - break; - case ECC_CURVE_P521R1: - magic = BCRYPT_ECDH_PRIVATE_P521_MAGIC; - size = 66; - break; - default: - FIXME( "unsupported curve %u\n", key->u.a.curve_id ); - return STATUS_NOT_IMPLEMENTED; - } - break; - - case ALG_ID_ECDH_P256: - magic = BCRYPT_ECDH_PRIVATE_P256_MAGIC; - size = 32; - break; - - case ALG_ID_ECDH_P384: - magic = BCRYPT_ECDH_PRIVATE_P384_MAGIC; - size = 48; - break; - - case ALG_ID_ECDH_P521: - magic = BCRYPT_ECDH_PRIVATE_P521_MAGIC; - size = 66; - break; - - case ALG_ID_ECDSA: - switch (key->u.a.curve_id) - { - case ECC_CURVE_P256R1: - magic = BCRYPT_ECDSA_PRIVATE_P256_MAGIC; - size = 32; - break; - case ECC_CURVE_P384R1: - magic = BCRYPT_ECDSA_PRIVATE_P384_MAGIC; - size = 48; - break; - case ECC_CURVE_P521R1: - magic = BCRYPT_ECDSA_PRIVATE_P521_MAGIC; - size = 66; - break; - default: - FIXME( "unsupported curve %u\n", key->u.a.curve_id ); - return STATUS_NOT_IMPLEMENTED; - } - break; - - case ALG_ID_ECDSA_P256: - magic = BCRYPT_ECDSA_PRIVATE_P256_MAGIC; - size = 32; - break; - - case ALG_ID_ECDSA_P384: - magic = BCRYPT_ECDSA_PRIVATE_P384_MAGIC; - size = 48; - break; - - case ALG_ID_ECDSA_P521: - magic = BCRYPT_ECDSA_PRIVATE_P521_MAGIC; - size = 66; - break; - - default: - FIXME( "algorithm %u does not yet support exporting ecc blob\n", key->alg_id ); - return STATUS_NOT_IMPLEMENTED; - } - - if (!key_data(key)->a.privkey) return STATUS_INVALID_PARAMETER; - - if ((ret = pgnutls_privkey_export_ecc_raw( key_data(key)->a.privkey, &curve, &x, &y, &d ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if (curve != GNUTLS_ECC_CURVE_X25519 && curve != GNUTLS_ECC_CURVE_SECP256R1 && - curve != GNUTLS_ECC_CURVE_SECP384R1 && curve != GNUTLS_ECC_CURVE_SECP521R1) - { - FIXME( "curve %u not supported\n", curve ); - free( x.data ); free( y.data ); free( d.data ); - return STATUS_NOT_IMPLEMENTED; - } - - *ret_len = sizeof(*ecc_blob) + EXPORT_SIZE( x, size, 1 ) + EXPORT_SIZE( y, size, 1 ) + EXPORT_SIZE( d, size, 1 ); - if (len >= *ret_len && buf) - { - ecc_blob = (BCRYPT_ECCKEY_BLOB *)buf; - ecc_blob->dwMagic = magic; - ecc_blob->cbKey = size; - - dst = (UCHAR *)(ecc_blob + 1); - dst += export_gnutls_datum( dst, size, &x, 1 ); - dst += export_gnutls_datum( dst, size, &y, 1 ); - export_gnutls_datum( dst, size, &d, 1 ); - } - - free( x.data ); free( y.data ); free( d.data ); - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_ecc( struct key *key, UCHAR *buf, ULONG len ) -{ - BCRYPT_ECCKEY_BLOB *ecc_blob; - gnutls_ecc_curve_t curve; - gnutls_privkey_t handle; - gnutls_datum_t x, y, k; - int ret; - - switch (key->alg_id) - { - case ALG_ID_ECDH: - switch (key->u.a.curve_id) - { - case ECC_CURVE_25519: curve = GNUTLS_ECC_CURVE_X25519; break; - case ECC_CURVE_P256R1: curve = GNUTLS_ECC_CURVE_SECP256R1; break; - case ECC_CURVE_P384R1: curve = GNUTLS_ECC_CURVE_SECP384R1; break; - case ECC_CURVE_P521R1: curve = GNUTLS_ECC_CURVE_SECP521R1; break; - default: - FIXME( "curve %u not supported\n", key->u.a.curve_id ); - return STATUS_NOT_IMPLEMENTED; - } - break; - - case ALG_ID_ECDH_P256: - case ALG_ID_ECDSA_P256: - curve = GNUTLS_ECC_CURVE_SECP256R1; - break; - - case ALG_ID_ECDH_P384: - case ALG_ID_ECDSA_P384: - curve = GNUTLS_ECC_CURVE_SECP384R1; - break; - - case ALG_ID_ECDH_P521: - case ALG_ID_ECDSA_P521: - curve = GNUTLS_ECC_CURVE_SECP521R1; - break; - - default: - FIXME( "algorithm %u not yet supported\n", key->alg_id ); - return STATUS_NOT_IMPLEMENTED; - } - - if ((ret = pgnutls_privkey_init( &handle ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - ecc_blob = (BCRYPT_ECCKEY_BLOB *)buf; - x.data = (unsigned char *)(ecc_blob + 1); - x.size = ecc_blob->cbKey; - y.data = x.data + ecc_blob->cbKey; - y.size = ecc_blob->cbKey; - k.data = y.data + ecc_blob->cbKey; - k.size = ecc_blob->cbKey; - - if ((ret = pgnutls_privkey_import_ecc_raw( handle, curve, &x, &y, &k ))) - { - pgnutls_perror( ret ); - pgnutls_privkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if (key_data(key)->a.privkey) pgnutls_privkey_deinit( key_data(key)->a.privkey ); - key_data(key)->a.privkey = handle; - return STATUS_SUCCESS; -} - -static NTSTATUS key_export_rsa( struct key *key, ULONG flags, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BCRYPT_RSAKEY_BLOB *rsa_blob; - gnutls_datum_t m, e, d, p, q, u, e1, e2; - ULONG size = len_from_bitlen( key->u.a.bitlen ); - BOOL full = (flags & KEY_EXPORT_FLAG_RSA_FULL); - UCHAR *dst; - int ret; - - if (!key_data(key)->a.privkey) return STATUS_INVALID_PARAMETER; - - if ((ret = pgnutls_privkey_export_rsa_raw( key_data(key)->a.privkey, &m, &e, &d, &p, &q, &u, &e1, &e2 ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - *ret_len = sizeof(*rsa_blob) + EXPORT_SIZE( e, size, 0 ) + EXPORT_SIZE( m, size, 1 ) + - EXPORT_SIZE( p, size / 2, 1 ) + EXPORT_SIZE( q, size / 2, 1 ); - - if (full) *ret_len += EXPORT_SIZE( e1, size / 2, 1 ) + EXPORT_SIZE( e2, size / 2, 1 ) + - EXPORT_SIZE( u, size / 2, 1 ) + EXPORT_SIZE( d, size, 1 ); - - if (len >= *ret_len && buf) - { - rsa_blob = (BCRYPT_RSAKEY_BLOB *)buf; - rsa_blob->Magic = full ? BCRYPT_RSAFULLPRIVATE_MAGIC : BCRYPT_RSAPRIVATE_MAGIC; - rsa_blob->BitLength = key->u.a.bitlen; - - dst = (UCHAR *)(rsa_blob + 1); - rsa_blob->cbPublicExp = export_gnutls_datum( dst, size, &e, 0 ); - - dst += rsa_blob->cbPublicExp; - rsa_blob->cbModulus = export_gnutls_datum( dst, size, &m, 1 ); - - dst += rsa_blob->cbModulus; - rsa_blob->cbPrime1 = export_gnutls_datum( dst, size / 2, &p, 1 ); - - dst += rsa_blob->cbPrime1; - rsa_blob->cbPrime2 = export_gnutls_datum( dst, size / 2, &q, 1 ); - - if (full) - { - dst += rsa_blob->cbPrime2; - export_gnutls_datum( dst, size / 2, &e1, 1 ); - - dst += rsa_blob->cbPrime1; - export_gnutls_datum( dst, size / 2, &e2, 1 ); - - dst += rsa_blob->cbPrime2; - export_gnutls_datum( dst, size / 2, &u, 1 ); - - dst += rsa_blob->cbPrime1; - export_gnutls_datum( dst, size, &d, 1 ); - } - } - - free( m.data ); free( e.data ); free( d.data ); free( p.data ); free( q.data ); free( u.data ); - free( e1.data ); free( e2.data ); - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_rsa( struct key *key, UCHAR *buf, ULONG len ) -{ - BCRYPT_RSAKEY_BLOB *rsa_blob = (BCRYPT_RSAKEY_BLOB *)buf; - gnutls_datum_t m, e, p, q; - gnutls_privkey_t handle; - int ret; - - if ((ret = pgnutls_privkey_init( &handle ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - e.data = (unsigned char *)(rsa_blob + 1); - e.size = rsa_blob->cbPublicExp; - m.data = e.data + e.size; - m.size = rsa_blob->cbModulus; - p.data = m.data + m.size; - p.size = rsa_blob->cbPrime1; - q.data = p.data + p.size; - q.size = rsa_blob->cbPrime2; - - if ((ret = pgnutls_privkey_import_rsa_raw( handle, &m, &e, NULL, &p, &q, NULL, NULL, NULL ))) - { - pgnutls_perror( ret ); - pgnutls_privkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if (key_data(key)->a.privkey) pgnutls_privkey_deinit( key_data(key)->a.privkey ); - key_data(key)->a.privkey = handle; - return STATUS_SUCCESS; -} - -static NTSTATUS key_export_dsa_capi( struct key *key, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BLOBHEADER *hdr; - DSSPUBKEY *pubkey; - gnutls_datum_t p, q, g, y, x; - ULONG size = len_from_bitlen( key->u.a.bitlen ); - UCHAR *dst; - int ret; - - if (!key_data(key)->a.privkey) return STATUS_INVALID_PARAMETER; - - if ((ret = pgnutls_privkey_export_dsa_raw( key_data(key)->a.privkey, &p, &q, &g, &y, &x ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if (q.size > 21 || x.size > 21) - { - ERR( "can't export key in this format\n" ); - free( p.data ); free( q.data ); free( g.data ); free( y.data ); free( x.data ); - return STATUS_NOT_SUPPORTED; - } - - *ret_len = sizeof(*hdr) + sizeof(*pubkey) + sizeof(key->u.a.dss_seed) + - EXPORT_SIZE( p, size, 1 ) + 20 + EXPORT_SIZE( g, size, 1 ) + 20; - if (len >= *ret_len && buf) - { - hdr = (BLOBHEADER *)buf; - hdr->bType = PRIVATEKEYBLOB; - hdr->bVersion = 2; - hdr->reserved = 0; - hdr->aiKeyAlg = CALG_DSS_SIGN; - - pubkey = (DSSPUBKEY *)(hdr + 1); - pubkey->magic = MAGIC_DSS2; - pubkey->bitlen = key->u.a.bitlen; - - dst = (UCHAR *)(pubkey + 1); - export_gnutls_datum( dst, size, &p, 1 ); - reverse_bytes( dst, size ); - dst += size; - - export_gnutls_datum( dst, 20, &q, 1 ); - reverse_bytes( dst, 20 ); - dst += 20; - - export_gnutls_datum( dst, size, &g, 1 ); - reverse_bytes( dst, size ); - dst += size; - - export_gnutls_datum( dst, 20, &x, 1 ); - reverse_bytes( dst, 20 ); - dst += 20; - - memcpy( dst, &key->u.a.dss_seed, sizeof(key->u.a.dss_seed) ); - } - - free( p.data ); free( q.data ); free( g.data ); free( y.data ); free( x.data ); - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_dsa_capi( struct key *key, UCHAR *buf, ULONG len ) -{ - BLOBHEADER *hdr = (BLOBHEADER *)buf; - DSSPUBKEY *pubkey; - gnutls_privkey_t handle; - gnutls_datum_t p, q, g, x; - unsigned char *data, p_data[128], q_data[20], g_data[128], x_data[20]; - int i, ret, size; - - if ((ret = pgnutls_privkey_init( &handle ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - pubkey = (DSSPUBKEY *)(hdr + 1); - if ((size = len_from_bitlen( pubkey->bitlen )) > sizeof(p_data)) - { - FIXME( "size %u not supported\n", size ); - pgnutls_privkey_deinit( handle ); - return STATUS_NOT_SUPPORTED; - } - data = (unsigned char *)(pubkey + 1); - - p.data = p_data; - p.size = size; - for (i = 0; i < p.size; i++) p.data[i] = data[p.size - i - 1]; - data += p.size; - - q.data = q_data; - q.size = sizeof(q_data); - for (i = 0; i < q.size; i++) q.data[i] = data[q.size - i - 1]; - data += q.size; - - g.data = g_data; - g.size = size; - for (i = 0; i < g.size; i++) g.data[i] = data[g.size - i - 1]; - data += g.size; - - x.data = x_data; - x.size = sizeof(x_data); - for (i = 0; i < x.size; i++) x.data[i] = data[x.size - i - 1]; - data += x.size; - - if ((ret = pgnutls_privkey_import_dsa_raw( handle, &p, &q, &g, NULL, &x ))) - { - pgnutls_perror( ret ); - pgnutls_privkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - memcpy( &key->u.a.dss_seed, data, sizeof(key->u.a.dss_seed) ); - - if (key_data(key)->a.privkey) pgnutls_privkey_deinit( key_data(key)->a.privkey ); - key_data(key)->a.privkey = handle; - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_ecc_public( struct key *key, UCHAR *buf, ULONG len ) -{ - BCRYPT_ECCKEY_BLOB *ecc_blob; - gnutls_ecc_curve_t curve; - gnutls_datum_t x, y; - gnutls_pubkey_t handle; - int ret; - - switch (key->alg_id) - { - case ALG_ID_ECDH: - switch (key->u.a.curve_id) - { - case ECC_CURVE_25519: curve = GNUTLS_ECC_CURVE_X25519; break; - case ECC_CURVE_P256R1: curve = GNUTLS_ECC_CURVE_SECP256R1; break; - case ECC_CURVE_P384R1: curve = GNUTLS_ECC_CURVE_SECP384R1; break; - case ECC_CURVE_P521R1: curve = GNUTLS_ECC_CURVE_SECP521R1; break; - default: - FIXME( "curve %u not supported\n", key->u.a.curve_id ); - return STATUS_NOT_IMPLEMENTED; - } - break; - - case ALG_ID_ECDH_P256: - case ALG_ID_ECDSA_P256: - curve = GNUTLS_ECC_CURVE_SECP256R1; break; - - case ALG_ID_ECDH_P384: - case ALG_ID_ECDSA_P384: - curve = GNUTLS_ECC_CURVE_SECP384R1; break; - - case ALG_ID_ECDH_P521: - case ALG_ID_ECDSA_P521: - curve = GNUTLS_ECC_CURVE_SECP521R1; break; - - default: - FIXME( "algorithm %u not yet supported\n", key->alg_id ); - return STATUS_NOT_IMPLEMENTED; - } - - if ((ret = pgnutls_pubkey_init( &handle ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - ecc_blob = (BCRYPT_ECCKEY_BLOB *)buf; - x.data = buf + sizeof(*ecc_blob); - x.size = ecc_blob->cbKey; - y.data = buf + sizeof(*ecc_blob) + ecc_blob->cbKey; - y.size = ecc_blob->cbKey; - - if ((ret = pgnutls_pubkey_import_ecc_raw( handle, curve, &x, &y ))) - { - pgnutls_perror( ret ); - pgnutls_pubkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if (key_data(key)->a.pubkey) pgnutls_pubkey_deinit( key_data(key)->a.pubkey ); - key_data(key)->a.pubkey = handle; - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_rsa_public( struct key *key, UCHAR *buf, ULONG len ) -{ - BCRYPT_RSAKEY_BLOB *rsa_blob; - gnutls_pubkey_t handle; - gnutls_datum_t m, e; - int ret; - - if ((ret = pgnutls_pubkey_init( &handle ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - rsa_blob = (BCRYPT_RSAKEY_BLOB *)buf; - e.data = buf + sizeof(*rsa_blob); - e.size = rsa_blob->cbPublicExp; - m.data = buf + sizeof(*rsa_blob) + rsa_blob->cbPublicExp; - m.size = rsa_blob->cbModulus; - - if ((ret = pgnutls_pubkey_import_rsa_raw( handle, &m, &e ))) - { - pgnutls_perror( ret ); - pgnutls_pubkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if (key_data(key)->a.pubkey) pgnutls_pubkey_deinit( key_data(key)->a.pubkey ); - key_data(key)->a.pubkey = handle; - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_dsa_public( struct key *key, UCHAR *buf, ULONG len ) -{ - BCRYPT_DSA_KEY_BLOB *dsa_blob; - gnutls_datum_t p, q, g, y; - gnutls_pubkey_t handle; - int ret; - - if ((ret = pgnutls_pubkey_init( &handle ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - dsa_blob = (BCRYPT_DSA_KEY_BLOB *)buf; - p.data = buf + sizeof(*dsa_blob); - p.size = dsa_blob->cbKey; - q.data = dsa_blob->q; - q.size = sizeof(dsa_blob->q); - g.data = buf + sizeof(*dsa_blob) + dsa_blob->cbKey; - g.size = dsa_blob->cbKey; - y.data = buf + sizeof(*dsa_blob) + dsa_blob->cbKey * 2; - y.size = dsa_blob->cbKey; - - if ((ret = pgnutls_pubkey_import_dsa_raw( handle, &p, &q, &g, &y ))) - { - pgnutls_perror( ret ); - pgnutls_pubkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if (key_data(key)->a.pubkey) pgnutls_pubkey_deinit( key_data(key)->a.pubkey ); - key_data(key)->a.pubkey = handle; - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_dsa_capi_public( struct key *key, UCHAR *buf, ULONG len ) -{ - BLOBHEADER *hdr; - DSSPUBKEY *pubkey; - gnutls_datum_t p, q, g, y; - gnutls_pubkey_t handle; - unsigned char *data, p_data[128], q_data[20], g_data[128], y_data[128]; - int i, ret, size; - - if ((ret = pgnutls_pubkey_init( &handle ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - hdr = (BLOBHEADER *)buf; - pubkey = (DSSPUBKEY *)(hdr + 1); - size = len_from_bitlen( pubkey->bitlen ); - data = (unsigned char *)(pubkey + 1); - - p.data = p_data; - p.size = size; - for (i = 0; i < p.size; i++) p.data[i] = data[p.size - i - 1]; - data += p.size; - - q.data = q_data; - q.size = sizeof(q_data); - for (i = 0; i < q.size; i++) q.data[i] = data[q.size - i - 1]; - data += q.size; - - g.data = g_data; - g.size = size; - for (i = 0; i < g.size; i++) g.data[i] = data[g.size - i - 1]; - data += g.size; - - y.data = y_data; - y.size = sizeof(y_data); - for (i = 0; i < y.size; i++) y.data[i] = data[y.size - i - 1]; - - if ((ret = pgnutls_pubkey_import_dsa_raw( handle, &p, &q, &g, &y ))) - { - pgnutls_perror( ret ); - pgnutls_pubkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if (key_data(key)->a.pubkey) pgnutls_pubkey_deinit( key_data(key)->a.pubkey ); - key_data(key)->a.pubkey = handle; - return STATUS_SUCCESS; -} - -static NTSTATUS key_export_dh_public( struct key *key, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BCRYPT_DH_KEY_BLOB *dh_blob = (BCRYPT_DH_KEY_BLOB *)buf; - ULONG size = len_from_bitlen( key->u.a.bitlen ); - gnutls_dh_params_t params; - gnutls_datum_t p, g, y; - UCHAR *dst; - int ret = GNUTLS_E_INVALID_REQUEST; - - if ((ret = pgnutls_dh_params_init( ¶ms )) < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if ((ret = pgnutls_pubkey_export_dh_raw( key_data(key)->a.pubkey, params, &y, 0 ))) - { - pgnutls_perror( ret ); - pgnutls_dh_params_deinit( params ); - return STATUS_INTERNAL_ERROR; - } - - if ((ret = pgnutls_dh_params_export_raw( params, &p, &g, NULL )) < 0) - { - pgnutls_perror( ret ); - free( y.data ); - pgnutls_dh_params_deinit( params ); - return STATUS_INTERNAL_ERROR; - } - - *ret_len = sizeof(*dh_blob) + EXPORT_SIZE(p, size, 1) + EXPORT_SIZE(g, size, 1) + EXPORT_SIZE(y, size, 1); - if (len >= *ret_len && buf) - { - dst = (UCHAR *)(dh_blob + 1); - dst += export_gnutls_datum( dst, size, &p, 1 ); - dst += export_gnutls_datum( dst, size, &g, 1 ); - dst += export_gnutls_datum( dst, size, &y, 1 ); - - dh_blob->dwMagic = BCRYPT_DH_PUBLIC_MAGIC; - dh_blob->cbKey = size; - } - - free( p.data ); free( g.data ); free( y.data ); - return STATUS_SUCCESS; -} - -static NTSTATUS key_export_dh( struct key *key, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BCRYPT_DH_KEY_BLOB *dh_blob = (BCRYPT_DH_KEY_BLOB *)buf; - gnutls_datum_t p, g, y, x; - gnutls_dh_params_t params; - ULONG size = len_from_bitlen( key->u.a.bitlen ); - UCHAR *dst; - int ret; - - if (!key_data(key)->a.privkey) return STATUS_INVALID_PARAMETER; - - if ((ret = pgnutls_dh_params_init( ¶ms )) < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if ((ret = pgnutls_privkey_export_dh_raw( key_data(key)->a.privkey, params, &y, &x, 0 ))) - { - pgnutls_perror( ret ); - pgnutls_dh_params_deinit( params ); - return STATUS_INTERNAL_ERROR; - } - - if ((ret = pgnutls_dh_params_export_raw( params, &p, &g, NULL )) < 0) - { - pgnutls_perror( ret ); - free( y.data ); free( x.data ); - pgnutls_dh_params_deinit( params ); - return STATUS_INTERNAL_ERROR; - } - - *ret_len = sizeof(*dh_blob) + EXPORT_SIZE(p, size, 1) + EXPORT_SIZE(g, size, 1) + - EXPORT_SIZE(y, size, 1) + EXPORT_SIZE(x, size, 1); - if (len >= *ret_len && buf) - { - dst = (UCHAR *)(dh_blob + 1); - dst += export_gnutls_datum( dst, size, &p, 1 ); - dst += export_gnutls_datum( dst, size, &g, 1 ); - dst += export_gnutls_datum( dst, size, &y, 1 ); - dst += export_gnutls_datum( dst, size, &x, 1 ); - - dh_blob->dwMagic = BCRYPT_DH_PRIVATE_MAGIC; - dh_blob->cbKey = size; - } - - free( p.data ); free( g.data ); free( y.data ); free( x.data ); - pgnutls_dh_params_deinit( params ); - return STATUS_SUCCESS; -} - -static NTSTATUS key_export_dh_params( struct key *key, UCHAR *buf, ULONG len, ULONG *ret_len ) -{ - BCRYPT_DH_PARAMETER_HEADER *hdr = (BCRYPT_DH_PARAMETER_HEADER *)buf; - unsigned int size = sizeof(*hdr) + len_from_bitlen( key->u.a.bitlen ) * 2; - gnutls_datum_t p, g; - NTSTATUS status = STATUS_SUCCESS; - UCHAR *dst; - int ret; - - if (!key_data(key)->a.dh_params) return STATUS_INVALID_PARAMETER; - - if ((ret = pgnutls_dh_params_export_raw( key_data(key)->a.dh_params, &p, &g, NULL ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - *ret_len = size; - if (len < size) status = STATUS_BUFFER_TOO_SMALL; - else if (buf) - { - hdr->cbLength = size; - hdr->dwMagic = BCRYPT_DH_PARAMETERS_MAGIC; - hdr->cbKeyLength = len_from_bitlen( key->u.a.bitlen ); - - dst = (UCHAR *)(hdr + 1); - dst += export_gnutls_datum( dst, hdr->cbKeyLength, &p, 1 ); - dst += export_gnutls_datum( dst, hdr->cbKeyLength, &g, 1 ); - } - - free( p.data ); free( g.data ); - return status; -} - -static NTSTATUS key_asymmetric_export( void *args ) -{ - const struct key_asymmetric_export_params *params = args; - struct key *key = params->key; - unsigned flags = params->flags; - - if (!(key->u.a.flags & KEY_FLAG_FINALIZED)) return STATUS_INVALID_HANDLE; - - switch (key->alg_id) - { - case ALG_ID_ECDH: - case ALG_ID_ECDH_P256: - case ALG_ID_ECDH_P384: - case ALG_ID_ECDH_P521: - case ALG_ID_ECDSA: - case ALG_ID_ECDSA_P256: - case ALG_ID_ECDSA_P384: - case ALG_ID_ECDSA_P521: - if (flags & KEY_EXPORT_FLAG_PUBLIC) - return key_export_ecc_public( key, params->buf, params->len, params->ret_len ); - return key_export_ecc( key, params->buf, params->len, params->ret_len ); - - case ALG_ID_RSA: - case ALG_ID_RSA_SIGN: - if (flags & KEY_EXPORT_FLAG_PUBLIC) - return key_export_rsa_public( key, params->buf, params->len, params->ret_len ); - return key_export_rsa( key, flags, params->buf, params->len, params->ret_len ); - - case ALG_ID_DSA: - if (flags & KEY_EXPORT_FLAG_PUBLIC) - { - if (key->u.a.flags & KEY_FLAG_LEGACY_DSA_V2) - return key_export_dsa_capi_public( key, params->buf, params->len, params->ret_len ); - return key_export_dsa_public( key, params->buf, params->len, params->ret_len ); - } - if (key->u.a.flags & KEY_FLAG_LEGACY_DSA_V2) - return key_export_dsa_capi( key, params->buf, params->len, params->ret_len ); - return STATUS_NOT_IMPLEMENTED; - - case ALG_ID_DH: - if (flags & KEY_EXPORT_FLAG_DH_PARAMETERS) - return key_export_dh_params( key, params->buf, params->len, params->ret_len ); - if (flags & KEY_EXPORT_FLAG_PUBLIC) - return key_export_dh_public( key, params->buf, params->len, params->ret_len ); - return key_export_dh( key, params->buf, params->len, params->ret_len ); - - default: - FIXME( "algorithm %u not yet supported\n", key->alg_id ); - return STATUS_NOT_IMPLEMENTED; - } -} - -static NTSTATUS key_import_dh_public( struct key *key, UCHAR *buf, ULONG len ) -{ - BCRYPT_DH_KEY_BLOB *dh_blob; - gnutls_dh_params_t params; - gnutls_datum_t p, g, y; - gnutls_pubkey_t handle; - int ret; - - if ((ret = pgnutls_pubkey_init( &handle ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if ((ret = pgnutls_dh_params_init( ¶ms )) < 0) - { - pgnutls_perror( ret ); - pgnutls_pubkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - dh_blob = (BCRYPT_DH_KEY_BLOB *)buf; - p.data = buf + sizeof(*dh_blob); - p.size = dh_blob->cbKey; - g.data = buf + sizeof(*dh_blob) + dh_blob->cbKey; - g.size = dh_blob->cbKey; - y.data = buf + sizeof(*dh_blob) + dh_blob->cbKey * 2; - y.size = dh_blob->cbKey; - - if ((ret = pgnutls_dh_params_import_raw( params, &p, &g )) < 0) - { - pgnutls_perror( ret ); - pgnutls_dh_params_deinit( params ); - pgnutls_pubkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if ((ret = pgnutls_pubkey_import_dh_raw( handle, params, &y ))) - { - pgnutls_perror( ret ); - pgnutls_dh_params_deinit( params ); - pgnutls_pubkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if (key_data(key)->a.pubkey) pgnutls_pubkey_deinit( key_data(key)->a.pubkey ); - key_data(key)->a.pubkey = handle; - - if (key_data(key)->a.dh_params) pgnutls_dh_params_deinit( key_data(key)->a.dh_params ); - key_data(key)->a.dh_params = params; - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_dh( struct key *key, UCHAR *buf, ULONG len ) -{ - BCRYPT_DH_KEY_BLOB *dh_blob; - gnutls_dh_params_t params; - gnutls_datum_t p, g, y, x; - gnutls_privkey_t handle; - int ret; - - if ((ret = pgnutls_privkey_init( &handle ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if ((ret = pgnutls_dh_params_init( ¶ms )) < 0) - { - pgnutls_perror( ret ); - pgnutls_privkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - dh_blob = (BCRYPT_DH_KEY_BLOB *)buf; - p.data = buf + sizeof(*dh_blob); - p.size = dh_blob->cbKey; - g.data = buf + sizeof(*dh_blob) + dh_blob->cbKey; - g.size = dh_blob->cbKey; - y.data = buf + sizeof(*dh_blob) + dh_blob->cbKey * 2; - y.size = dh_blob->cbKey; - x.data = buf + sizeof(*dh_blob) + dh_blob->cbKey * 3; - x.size = dh_blob->cbKey; - - if ((ret = pgnutls_dh_params_import_raw( params, &p, &g )) < 0) - { - pgnutls_perror( ret ); - pgnutls_dh_params_deinit( params ); - pgnutls_privkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if ((ret = pgnutls_privkey_import_dh_raw( handle, params, &y, &x ))) - { - pgnutls_perror( ret ); - pgnutls_dh_params_deinit( params ); - pgnutls_privkey_deinit( handle ); - return STATUS_INTERNAL_ERROR; - } - - if (key_data(key)->a.privkey) pgnutls_privkey_deinit( key_data(key)->a.privkey ); - key_data(key)->a.privkey = handle; - - if (key_data(key)->a.dh_params) pgnutls_dh_params_deinit( key_data(key)->a.dh_params ); - key_data(key)->a.dh_params = params; - return STATUS_SUCCESS; -} - -static NTSTATUS key_import_dh_params( struct key *key, UCHAR *buf, ULONG len ) -{ - BCRYPT_DH_PARAMETER_HEADER *dh_header = (BCRYPT_DH_PARAMETER_HEADER *)buf; - gnutls_dh_params_t params; - gnutls_datum_t p, g; - int ret; - - if ((ret = pgnutls_dh_params_init( ¶ms ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - p.data = (unsigned char *)(dh_header + 1); - p.size = dh_header->cbKeyLength; - g.data = p.data + dh_header->cbKeyLength; - g.size = dh_header->cbKeyLength; - - if ((ret = pgnutls_dh_params_import_raw( params, &p, &g ))) - { - pgnutls_perror( ret ); - pgnutls_dh_params_deinit( params ); - return STATUS_INTERNAL_ERROR; - } - - if (key_data(key)->a.dh_params) pgnutls_dh_params_deinit( key_data(key)->a.dh_params ); - key_data(key)->a.dh_params = params; - return STATUS_SUCCESS; -} - -static NTSTATUS key_asymmetric_import( void *args ) -{ - const struct key_asymmetric_import_params *params = args; - struct key *key = params->key; - unsigned flags = params->flags; - gnutls_pubkey_t pubkey; - NTSTATUS ret; - - switch (key->alg_id) - { - case ALG_ID_ECDH: - case ALG_ID_ECDH_P256: - case ALG_ID_ECDH_P384: - case ALG_ID_ECDH_P521: - case ALG_ID_ECDSA: - case ALG_ID_ECDSA_P256: - case ALG_ID_ECDSA_P384: - case ALG_ID_ECDSA_P521: - if (flags & KEY_IMPORT_FLAG_PUBLIC) - return key_import_ecc_public( key, params->buf, params->len ); - ret = key_import_ecc( key, params->buf, params->len ); - break; - - case ALG_ID_RSA: - case ALG_ID_RSA_SIGN: - if (flags & KEY_IMPORT_FLAG_PUBLIC) - return key_import_rsa_public( key, params->buf, params->len ); - ret = key_import_rsa( key, params->buf, params->len ); - break; - - case ALG_ID_DSA: - if (flags & KEY_IMPORT_FLAG_PUBLIC) - { - if (key->u.a.flags & KEY_FLAG_LEGACY_DSA_V2) - return key_import_dsa_capi_public( key, params->buf, params->len ); - return key_import_dsa_public( key, params->buf, params->len ); - } - if (key->u.a.flags & KEY_FLAG_LEGACY_DSA_V2) - { - ret = key_import_dsa_capi( key, params->buf, params->len ); - break; - } - FIXME( "DSA private key not supported\n" ); - return STATUS_NOT_IMPLEMENTED; - - case ALG_ID_DH: - if (flags & KEY_IMPORT_FLAG_DH_PARAMETERS) - return key_import_dh_params( key, params->buf, params->len ); - if (flags & KEY_IMPORT_FLAG_PUBLIC) - return key_import_dh_public( key, params->buf, params->len ); - ret = key_import_dh( key, params->buf, params->len ); - break; - - default: - FIXME( "algorithm %u not yet supported\n", key->alg_id ); - return STATUS_NOT_IMPLEMENTED; - } - - if (ret) return ret; - - if ((ret = pgnutls_pubkey_init( &pubkey ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - if (pgnutls_pubkey_import_privkey( pubkey, key_data(params->key)->a.privkey, 0, 0 )) - { - /* Imported private key may be legitimately missing public key, so ignore the failure here. */ - pgnutls_pubkey_deinit( pubkey ); - } - else - { - if (key_data(key)->a.pubkey) pgnutls_pubkey_deinit( key_data(key)->a.pubkey ); - key_data(key)->a.pubkey = pubkey; - } - return STATUS_SUCCESS; -} - -static NTSTATUS prepare_gnutls_signature_dsa( struct key *key, UCHAR *signature, ULONG signature_len, - gnutls_datum_t *gnutls_signature ) -{ - struct buffer buffer; - DWORD r_len = signature_len / 2; - DWORD s_len = r_len; - BYTE *r = signature; - BYTE *s = signature + r_len; - - buffer_init( &buffer ); - buffer_append_asn1_r_s( &buffer, r, r_len, s, s_len ); - if (buffer.error) - { - buffer_free( &buffer ); - return STATUS_NO_MEMORY; - } - - gnutls_signature->data = buffer.buffer; - gnutls_signature->size = buffer.pos; - return STATUS_SUCCESS; -} - -static NTSTATUS prepare_gnutls_signature_rsa( struct key *key, UCHAR *signature, ULONG signature_len, - gnutls_datum_t *gnutls_signature ) -{ - gnutls_signature->data = signature; - gnutls_signature->size = signature_len; - return STATUS_SUCCESS; -} - -static NTSTATUS prepare_gnutls_signature( struct key *key, UCHAR *signature, ULONG signature_len, - gnutls_datum_t *gnutls_signature ) -{ - switch (key->alg_id) - { - case ALG_ID_ECDSA_P256: - case ALG_ID_ECDSA_P384: - case ALG_ID_ECDSA_P521: - case ALG_ID_DSA: - return prepare_gnutls_signature_dsa( key, signature, signature_len, gnutls_signature ); - - case ALG_ID_RSA: - case ALG_ID_RSA_SIGN: - return prepare_gnutls_signature_rsa( key, signature, signature_len, gnutls_signature ); - - default: - FIXME( "algorithm %u not yet supported\n", key->alg_id ); - return STATUS_NOT_IMPLEMENTED; - } -} - -static gnutls_digest_algorithm_t get_digest_from_id( const WCHAR *alg_id ) -{ - if (!wcscmp( alg_id, BCRYPT_SHA1_ALGORITHM )) return GNUTLS_DIG_SHA1; - if (!wcscmp( alg_id, BCRYPT_SHA256_ALGORITHM )) return GNUTLS_DIG_SHA256; - if (!wcscmp( alg_id, BCRYPT_SHA384_ALGORITHM )) return GNUTLS_DIG_SHA384; - if (!wcscmp( alg_id, BCRYPT_SHA512_ALGORITHM )) return GNUTLS_DIG_SHA512; - if (!wcscmp( alg_id, BCRYPT_MD2_ALGORITHM )) return GNUTLS_DIG_MD2; - if (!wcscmp( alg_id, BCRYPT_MD5_ALGORITHM )) return GNUTLS_DIG_MD5; - return GNUTLS_DIG_UNKNOWN; -} - -static NTSTATUS pubkey_set_rsa_pss_params( gnutls_pubkey_t key, gnutls_digest_algorithm_t dig, unsigned int salt_size ) -{ - gnutls_x509_spki_t spki; - int ret; - - if ((ret = pgnutls_x509_spki_init( &spki )) < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - pgnutls_x509_spki_set_rsa_pss_params( spki, dig, salt_size ); - ret = pgnutls_pubkey_set_spki( key, spki, 0 ); - pgnutls_x509_spki_deinit( spki ); - if (ret < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - return STATUS_SUCCESS; -} - -static NTSTATUS key_asymmetric_verify( void *args ) -{ -#ifdef GNUTLS_VERIFY_ALLOW_BROKEN - unsigned int verify_flags = GNUTLS_VERIFY_ALLOW_BROKEN; -#else - unsigned int verify_flags = 0; -#endif - const struct key_asymmetric_verify_params *params = args; - struct key *key = params->key; - unsigned flags = params->flags; - gnutls_digest_algorithm_t hash_alg; - gnutls_sign_algorithm_t sign_alg; - gnutls_datum_t gnutls_hash, gnutls_signature; - gnutls_pk_algorithm_t pk_alg; - NTSTATUS status; - int ret; - - switch (key->alg_id) - { - case ALG_ID_ECDSA_P256: - case ALG_ID_ECDSA_P384: - case ALG_ID_ECDSA_P521: - { - if (flags) FIXME( "flags %#x not supported\n", flags ); - - /* only the hash size must match, not the actual hash function */ - switch (params->hash_len) - { - case 20: hash_alg = GNUTLS_DIG_SHA1; break; - case 32: hash_alg = GNUTLS_DIG_SHA256; break; - case 48: hash_alg = GNUTLS_DIG_SHA384; break; - case 64: hash_alg = GNUTLS_DIG_SHA512; break; - - default: - FIXME( "hash size %u not yet supported\n", params->hash_len ); - return STATUS_INVALID_SIGNATURE; - } - pk_alg = GNUTLS_PK_ECC; - break; - } - case ALG_ID_RSA: - case ALG_ID_RSA_SIGN: - { - if (flags & BCRYPT_PAD_PKCS1) - { - BCRYPT_PKCS1_PADDING_INFO *info = params->padding; - - if (!info) return STATUS_INVALID_PARAMETER; - if (!info->pszAlgId) - { - hash_alg = GNUTLS_DIG_UNKNOWN; - verify_flags |= GNUTLS_VERIFY_USE_TLS1_RSA; - } - else if ((hash_alg = get_digest_from_id(info->pszAlgId)) == GNUTLS_DIG_UNKNOWN) - { - FIXME( "hash algorithm %s not supported\n", debugstr_w(info->pszAlgId) ); - return STATUS_NOT_SUPPORTED; - } - pk_alg = GNUTLS_PK_RSA; - } - else if (flags & BCRYPT_PAD_PSS) - { - BCRYPT_PSS_PADDING_INFO *info = params->padding; - - if (!info) return STATUS_INVALID_PARAMETER; - if (!info->pszAlgId) return STATUS_INVALID_SIGNATURE; - if ((hash_alg = get_digest_from_id(info->pszAlgId)) == GNUTLS_DIG_UNKNOWN) - { - FIXME( "hash algorithm %s not supported\n", debugstr_w(info->pszAlgId) ); - return STATUS_NOT_SUPPORTED; - } - if ((status = pubkey_set_rsa_pss_params( key_data(key)->a.pubkey, hash_alg, info->cbSalt ))) return status; - pk_alg = GNUTLS_PK_RSA_PSS; - } - else return STATUS_INVALID_PARAMETER; - break; - } - case ALG_ID_DSA: - { - if (flags) FIXME( "flags %#x not supported\n", flags ); - if (params->hash_len != 20) - { - FIXME( "hash size %u not supported\n", params->hash_len ); - return STATUS_INVALID_PARAMETER; - } - hash_alg = GNUTLS_DIG_SHA1; - pk_alg = GNUTLS_PK_DSA; - break; - } - default: - FIXME( "algorithm %u not yet supported\n", key->alg_id ); - return STATUS_NOT_IMPLEMENTED; - } - - if ((sign_alg = pgnutls_pk_to_sign( pk_alg, hash_alg )) == GNUTLS_SIGN_UNKNOWN) - { - FIXME("GnuTLS does not support algorithm %u with hash len %u\n", key->alg_id, params->hash_len ); - return STATUS_NOT_IMPLEMENTED; - } - - if ((status = prepare_gnutls_signature( key, params->signature, params->signature_len, &gnutls_signature ))) - return status; - - gnutls_hash.data = params->hash; - gnutls_hash.size = params->hash_len; - - ret = pgnutls_pubkey_verify_hash2( key_data(key)->a.pubkey, sign_alg, verify_flags, &gnutls_hash, &gnutls_signature ); - if (gnutls_signature.data != params->signature) free( gnutls_signature.data ); - return (ret < 0) ? STATUS_INVALID_SIGNATURE : STATUS_SUCCESS; -} - -static unsigned int get_signature_length( enum alg_id id ) -{ - switch (id) - { - case ALG_ID_ECDSA_P256: return 64; - case ALG_ID_ECDSA_P384: return 96; - case ALG_ID_ECDSA_P521: return 132; - case ALG_ID_DSA: return 40; - default: - FIXME( "unhandled algorithm %u\n", id ); - return 0; - } -} - -static NTSTATUS format_gnutls_signature( enum alg_id type, gnutls_datum_t signature, - UCHAR *output, ULONG output_len, ULONG *ret_len ) -{ - switch (type) - { - case ALG_ID_RSA: - case ALG_ID_RSA_SIGN: - { - *ret_len = signature.size; - if (output_len < signature.size) return STATUS_BUFFER_TOO_SMALL; - if (output) memcpy( output, signature.data, signature.size ); - return STATUS_SUCCESS; - } - case ALG_ID_ECDSA_P256: - case ALG_ID_ECDSA_P384: - case ALG_ID_ECDSA_P521: - case ALG_ID_DSA: - { - int err; - unsigned int sig_len = get_signature_length( type ); - gnutls_datum_t r, s; /* format as r||s */ - - if ((err = pgnutls_decode_rs_value( &signature, &r, &s ))) - { - pgnutls_perror( err ); - return STATUS_INTERNAL_ERROR; - } - - *ret_len = sig_len; - if (output_len < sig_len) return STATUS_BUFFER_TOO_SMALL; - - if (r.size > sig_len / 2 + 1 || s.size > sig_len / 2 + 1) - { - ERR( "we didn't get a correct signature\n" ); - return STATUS_INTERNAL_ERROR; - } - - if (output) - { - export_gnutls_datum( output, sig_len / 2, &r, 1 ); - export_gnutls_datum( output + sig_len / 2, sig_len / 2, &s, 1 ); - } - - free( r.data ); free( s.data ); - return STATUS_SUCCESS; - } - default: - return STATUS_INTERNAL_ERROR; - } -} - -static NTSTATUS privkey_set_rsa_pss_params( gnutls_privkey_t key, gnutls_digest_algorithm_t dig, unsigned int salt_size ) -{ - gnutls_x509_spki_t spki; - int ret; - - if ((ret = pgnutls_x509_spki_init( &spki )) < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - pgnutls_x509_spki_set_rsa_pss_params( spki, dig, salt_size ); - ret = pgnutls_privkey_set_spki( key, spki, 0 ); - pgnutls_x509_spki_deinit( spki ); - if (ret < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - return STATUS_SUCCESS; -} - -static NTSTATUS key_asymmetric_sign( void *args ) -{ - const struct key_asymmetric_sign_params *params = args; - struct key *key = params->key; - unsigned int flags = params->flags, gnutls_flags = 0; - gnutls_datum_t hash, signature; - gnutls_digest_algorithm_t hash_alg; - NTSTATUS status; - int ret; - - if (key->alg_id == ALG_ID_ECDSA_P256 || key->alg_id == ALG_ID_ECDSA_P384 || key->alg_id == ALG_ID_ECDSA_P521) - { - /* With ECDSA, we find the digest algorithm from the hash length, and verify it */ - switch (params->input_len) - { - case 20: hash_alg = GNUTLS_DIG_SHA1; break; - case 32: hash_alg = GNUTLS_DIG_SHA256; break; - case 48: hash_alg = GNUTLS_DIG_SHA384; break; - case 64: hash_alg = GNUTLS_DIG_SHA512; break; - - default: - FIXME( "hash size %u not yet supported\n", params->input_len ); - return STATUS_INVALID_PARAMETER; - } - - if (flags == BCRYPT_PAD_PKCS1) - { - BCRYPT_PKCS1_PADDING_INFO *pad = params->padding; - if (pad && pad->pszAlgId && get_digest_from_id( pad->pszAlgId ) != hash_alg) - { - WARN( "incorrect hashing algorithm %s, expected %u\n", debugstr_w(pad->pszAlgId), hash_alg ); - return STATUS_INVALID_PARAMETER; - } - } - } - else if (key->alg_id == ALG_ID_DSA) - { - if (flags) FIXME( "flags %#x not supported\n", flags ); - if (params->input_len != 20) - { - FIXME( "hash size %u not supported\n", params->input_len ); - return STATUS_INVALID_PARAMETER; - } - hash_alg = GNUTLS_DIG_SHA1; - } - else if (flags == BCRYPT_PAD_PKCS1) - { - BCRYPT_PKCS1_PADDING_INFO *pad = params->padding; - - if (!pad) - { - WARN( "padding info not found\n" ); - return STATUS_INVALID_PARAMETER; - } - if (!pad->pszAlgId) hash_alg = GNUTLS_DIG_UNKNOWN; - else if ((hash_alg = get_digest_from_id( pad->pszAlgId )) == GNUTLS_DIG_UNKNOWN) - { - FIXME( "hash algorithm %s not recognized\n", debugstr_w(pad->pszAlgId) ); - return STATUS_NOT_SUPPORTED; - } - } - else if (flags == BCRYPT_PAD_PSS) - { - BCRYPT_PSS_PADDING_INFO *pad = params->padding; - - if (!pad || !pad->pszAlgId) - { - WARN( "padding info not found\n" ); - return STATUS_INVALID_PARAMETER; - } - if (key->alg_id != ALG_ID_RSA && key->alg_id != ALG_ID_RSA_SIGN) - { - FIXME( "BCRYPT_PAD_PSS not supported for key algorithm %u\n", key->alg_id ); - return STATUS_NOT_SUPPORTED; - } - if ((hash_alg = get_digest_from_id( pad->pszAlgId )) == GNUTLS_DIG_UNKNOWN) - { - FIXME( "hash algorithm %s not recognized\n", debugstr_w(pad->pszAlgId) ); - return STATUS_NOT_SUPPORTED; - } - - if ((status = privkey_set_rsa_pss_params( key_data(key)->a.privkey, hash_alg, pad->cbSalt ))) return status; - gnutls_flags = GNUTLS_PRIVKEY_SIGN_FLAG_RSA_PSS; - } - else if (!flags) - { - WARN( "invalid flags %#x\n", flags ); - return STATUS_INVALID_PARAMETER; - } - else - { - FIXME( "flags %#x not implemented\n", flags ); - return STATUS_NOT_IMPLEMENTED; - } - - if (!params->output) - { - *params->ret_len = len_from_bitlen( key->u.a.bitlen ); - return STATUS_SUCCESS; - } - if (!key_data(key)->a.privkey) return STATUS_INVALID_PARAMETER; - - hash.data = params->input; - hash.size = params->input_len; - - signature.data = NULL; - signature.size = 0; - - if ((ret = pgnutls_privkey_sign_hash( key_data(key)->a.privkey, hash_alg, gnutls_flags, &hash, &signature ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - status = format_gnutls_signature( key->alg_id, signature, params->output, params->output_len, params->ret_len ); - free( signature.data ); - return status; -} - -static NTSTATUS key_asymmetric_destroy( void *args ) -{ - struct key *key = args; - - if (key_data(key)->a.privkey) pgnutls_privkey_deinit( key_data(key)->a.privkey ); - if (key_data(key)->a.pubkey) pgnutls_pubkey_deinit( key_data(key)->a.pubkey ); - if (key_data(key)->a.dh_params) pgnutls_dh_params_deinit( key_data(key)->a.dh_params ); - return STATUS_SUCCESS; -} - -static NTSTATUS dup_privkey( struct key *key_orig, struct key *key_copy ) -{ - gnutls_privkey_t privkey; - int ret; - - if ((ret = pgnutls_privkey_init( &privkey ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - switch (key_orig->alg_id) - { - case ALG_ID_RSA: - case ALG_ID_RSA_SIGN: - { - gnutls_datum_t m, e, d, p, q, u, e1, e2; - - if ((ret = pgnutls_privkey_export_rsa_raw( key_data(key_orig)->a.privkey, &m, &e, &d, &p, &q, &u, &e1, &e2 ))) - break; - ret = pgnutls_privkey_import_rsa_raw( privkey, &m, &e, &d, &p, &q, &u, &e1, &e2 ); - free( m.data ); free( e.data ); free( d.data ); free( p.data ); free( q.data ); free( u.data ); - free( e1.data ); free( e2.data ); - break; - } - case ALG_ID_DSA: - { - gnutls_datum_t p, q, g, y, x; - - if ((ret = pgnutls_privkey_export_dsa_raw( key_data(key_orig)->a.privkey, &p, &q, &g, &y, &x ))) break; - ret = pgnutls_privkey_import_dsa_raw( privkey, &p, &q, &g, &y, &x ); - free( p.data ); free( q.data ); free( g.data ); free( y.data ); free( x.data ); - if (!ret) key_copy->u.a.dss_seed = key_orig->u.a.dss_seed; - break; - } - case ALG_ID_ECDH: - case ALG_ID_ECDH_P256: - case ALG_ID_ECDH_P384: - case ALG_ID_ECDH_P521: - case ALG_ID_ECDSA: - case ALG_ID_ECDSA_P256: - case ALG_ID_ECDSA_P384: - case ALG_ID_ECDSA_P521: - { - gnutls_ecc_curve_t curve; - gnutls_datum_t x, y, k; - - if ((ret = pgnutls_privkey_export_ecc_raw( key_data(key_orig)->a.privkey, &curve, &x, &y, &k ))) break; - ret = pgnutls_privkey_import_ecc_raw( privkey, curve, &x, &y, &k ); - free( x.data ); free( y.data ); free( k.data ); - break; - } - case ALG_ID_DH: - { - gnutls_dh_params_t params; - gnutls_datum_t y, x; - - if ((ret = pgnutls_dh_params_init( ¶ms )) < 0) break; - if ((ret = pgnutls_privkey_export_dh_raw( key_data(key_orig)->a.privkey, params, &y, &x, 0 )) < 0) - { - pgnutls_dh_params_deinit( params ); - break; - } - ret = pgnutls_privkey_import_dh_raw( privkey, params, &y, &x ); - pgnutls_dh_params_deinit( params ); - free( x.data ); free( y.data ); - break; - } - default: - ERR( "unhandled algorithm %u\n", key_orig->alg_id ); - pgnutls_privkey_deinit( privkey ); - return STATUS_INTERNAL_ERROR; - } - - if (ret < 0) - { - pgnutls_perror( ret ); - pgnutls_privkey_deinit( privkey ); - return STATUS_INTERNAL_ERROR; - } - - key_data(key_copy)->a.privkey = privkey; - return STATUS_SUCCESS; -} - -static NTSTATUS dup_pubkey( struct key *key_orig, struct key *key_copy ) -{ - gnutls_pubkey_t pubkey; - int ret; - - if ((ret = pgnutls_pubkey_init( &pubkey ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - switch (key_orig->alg_id) - { - case ALG_ID_RSA: - case ALG_ID_RSA_SIGN: - { - gnutls_datum_t m, e; - - if ((ret = pgnutls_pubkey_export_rsa_raw( key_data(key_orig)->a.pubkey, &m, &e ))) break; - ret = pgnutls_pubkey_import_rsa_raw( pubkey, &m, &e ); - free( m.data ); free( e.data ); - break; - } - case ALG_ID_DSA: - { - gnutls_datum_t p, q, g, y; - - if ((ret = pgnutls_pubkey_export_dsa_raw( key_data(key_orig)->a.pubkey, &p, &q, &g, &y ))) break; - ret = pgnutls_pubkey_import_dsa_raw( pubkey, &p, &q, &g, &y ); - free( p.data ); free( q.data ); free( g.data ); free( y.data ); - if (!ret) key_copy->u.a.dss_seed = key_orig->u.a.dss_seed; - break; - } - case ALG_ID_ECDH: - case ALG_ID_ECDH_P256: - case ALG_ID_ECDH_P384: - case ALG_ID_ECDH_P521: - case ALG_ID_ECDSA: - case ALG_ID_ECDSA_P256: - case ALG_ID_ECDSA_P384: - case ALG_ID_ECDSA_P521: - { - gnutls_ecc_curve_t curve; - gnutls_datum_t x, y; - - if ((ret = pgnutls_pubkey_export_ecc_raw( key_data(key_orig)->a.pubkey, &curve, &x, &y ))) break; - ret = pgnutls_pubkey_import_ecc_raw( pubkey, curve, &x, &y ); - free( x.data ); free( y.data ); - break; - } - case ALG_ID_DH: - { - gnutls_dh_params_t params; - gnutls_datum_t y; - - if ((ret = pgnutls_dh_params_init( ¶ms )) < 0) break; - if ((ret = pgnutls_pubkey_export_dh_raw( key_data(key_orig)->a.pubkey, params, &y, 0 )) < 0) - { - pgnutls_dh_params_deinit( params ); - break; - } - ret = pgnutls_pubkey_import_dh_raw( pubkey, params, &y ); - pgnutls_dh_params_deinit( params ); - free( y.data ); - break; - } - default: - ERR( "unhandled algorithm %u\n", key_orig->alg_id ); - pgnutls_pubkey_deinit( pubkey ); - return STATUS_INTERNAL_ERROR; - } - - if (ret < 0) - { - pgnutls_perror( ret ); - pgnutls_pubkey_deinit( pubkey ); - return STATUS_INTERNAL_ERROR; - } - - key_data(key_copy)->a.pubkey = pubkey; - return STATUS_SUCCESS; -} - -static NTSTATUS key_asymmetric_duplicate( void *args ) -{ - const struct key_asymmetric_duplicate_params *params = args; - NTSTATUS status; - - if (key_data(params->key_orig)->a.privkey && (status = dup_privkey( params->key_orig, params->key_copy ))) - return status; - - if (key_data(params->key_orig)->a.pubkey && (status = dup_pubkey( params->key_orig, params->key_copy ))) - return status; - - return STATUS_SUCCESS; -} - -static NTSTATUS privkey_set_rsa_oaep_params( gnutls_privkey_t key, gnutls_digest_algorithm_t dig, gnutls_datum_t *label ) -{ - gnutls_x509_spki_t spki; - int ret; - - if ((ret = pgnutls_x509_spki_init( &spki )) < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - if ((ret = pgnutls_x509_spki_set_rsa_oaep_params( spki, dig, label )) < 0) - { - pgnutls_x509_spki_deinit( spki ); - if (ret == GNUTLS_E_UNKNOWN_PK_ALGORITHM) return STATUS_NOT_SUPPORTED; - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - ret = pgnutls_privkey_set_spki( key, spki, 0 ); - pgnutls_x509_spki_deinit( spki ); - if (ret < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - return STATUS_SUCCESS; -} - -static NTSTATUS key_asymmetric_decrypt( void *args ) -{ - const struct key_asymmetric_decrypt_params *params = args; - gnutls_datum_t e, d = { 0 }; - NTSTATUS status = STATUS_SUCCESS; - int ret; - - if (params->key->alg_id == ALG_ID_RSA && (params->flags & BCRYPT_PAD_OAEP)) - { - BCRYPT_OAEP_PADDING_INFO *pad = params->padding; - gnutls_digest_algorithm_t dig; - gnutls_datum_t label; - - if (!pad || !pad->pszAlgId) - { - WARN( "padding info not found\n" ); - return STATUS_INVALID_PARAMETER; - } - - if ((dig = get_digest_from_id( pad->pszAlgId )) == GNUTLS_DIG_UNKNOWN) - { - FIXME( "hash algorithm %s not recognized\n", debugstr_w(pad->pszAlgId) ); - return STATUS_NOT_SUPPORTED; - } - - if (pad->pbLabel && pad->cbLabel) - { - label.data = pad->pbLabel; - label.size = pad->cbLabel; - status = privkey_set_rsa_oaep_params( key_data(params->key)->a.privkey, dig, &label ); - if (status == STATUS_NOT_SUPPORTED) status = STATUS_SUCCESS; - if (status) return status; - } - } - - e.data = params->input; - e.size = params->input_len; - if ((ret = pgnutls_privkey_decrypt_data( key_data(params->key)->a.privkey, 0, &e, &d ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - *params->ret_len = d.size; - if (params->output_len >= d.size) memcpy( params->output, d.data, *params->ret_len ); - else if (params->output) status = STATUS_BUFFER_TOO_SMALL; - - free( d.data ); - return status; -} - -static NTSTATUS pubkey_set_rsa_oaep_params( gnutls_pubkey_t key, gnutls_digest_algorithm_t dig, gnutls_datum_t *label ) -{ - gnutls_x509_spki_t spki; - int ret; - - if ((ret = pgnutls_x509_spki_init( &spki )) < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - if ((ret = pgnutls_x509_spki_set_rsa_oaep_params( spki, dig, label )) < 0) - { - pgnutls_x509_spki_deinit( spki ); - if (ret == GNUTLS_E_UNKNOWN_PK_ALGORITHM) return STATUS_NOT_SUPPORTED; - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - ret = pgnutls_pubkey_set_spki( key, spki, 0 ); - pgnutls_x509_spki_deinit( spki ); - if (ret < 0) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - return STATUS_SUCCESS; -} - -static NTSTATUS key_asymmetric_encrypt( void *args ) -{ - const struct key_asymmetric_encrypt_params *params = args; - gnutls_datum_t d, e = { 0 }; - NTSTATUS status = STATUS_SUCCESS; - int ret; - - if (!key_data(params->key)->a.pubkey) return STATUS_INVALID_HANDLE; - - if (params->key->alg_id == ALG_ID_RSA - && (!params->output || len_from_bitlen( params->key->u.a.bitlen ) > params->output_len)) - { - *params->ret_len = len_from_bitlen( params->key->u.a.bitlen ); - return !params->output ? STATUS_SUCCESS : STATUS_BUFFER_TOO_SMALL; - } - - if (params->key->alg_id == ALG_ID_RSA && (params->flags & BCRYPT_PAD_OAEP)) - { - BCRYPT_OAEP_PADDING_INFO *pad = params->padding; - gnutls_digest_algorithm_t dig; - gnutls_datum_t label; - - if (!pad || !pad->pszAlgId) - { - WARN( "padding info not found\n" ); - return STATUS_INVALID_PARAMETER; - } - - if ((dig = get_digest_from_id( pad->pszAlgId )) == GNUTLS_DIG_UNKNOWN) - { - FIXME( "hash algorithm %s not recognized\n", debugstr_w(pad->pszAlgId) ); - return STATUS_NOT_SUPPORTED; - } - - if (pad->pbLabel && pad->cbLabel) - { - label.data = pad->pbLabel; - label.size = pad->cbLabel; - status = pubkey_set_rsa_oaep_params( key_data(params->key)->a.pubkey, dig, &label ); - if (status == STATUS_NOT_SUPPORTED) status = STATUS_SUCCESS; - if (status) return status; - } - } - - d.data = params->input; - d.size = params->input_len; - if ((ret = pgnutls_pubkey_encrypt_data(key_data(params->key)->a.pubkey, 0, &d, &e))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - *params->ret_len = e.size; - if (params->output_len >= e.size) memcpy( params->output, e.data, *params->ret_len ); - else if (params->output_len == 0) status = STATUS_SUCCESS; - else status = STATUS_BUFFER_TOO_SMALL; - - free( e.data ); - return status; -} - -static NTSTATUS key_asymmetric_derive_key( void *args ) -{ - const struct key_asymmetric_derive_key_params *params = args; - gnutls_datum_t s; - NTSTATUS status = STATUS_SUCCESS; - int ret; - - if ((ret = pgnutls_privkey_derive_secret( key_data(params->privkey)->a.privkey, - key_data(params->pubkey)->a.pubkey, NULL, &s, 0 ))) - { - pgnutls_perror( ret ); - return STATUS_INTERNAL_ERROR; - } - - *params->ret_len = EXPORT_SIZE( s, len_from_bitlen( params->privkey->u.a.bitlen ), 1 ); - if (params->output) - { - if (params->output_len < *params->ret_len) status = STATUS_BUFFER_TOO_SMALL; - else export_gnutls_datum( params->output, *params->ret_len, &s, 1 ); - } - - free( s.data ); - return status; -} - -const unixlib_entry_t __wine_unix_call_funcs[] = -{ - gnutls_process_attach, - gnutls_process_detach, - key_symmetric_vector_reset, - key_symmetric_set_auth_data, - key_symmetric_encrypt, - key_symmetric_decrypt, - key_symmetric_get_tag, - key_symmetric_destroy, - key_asymmetric_generate, - key_asymmetric_decrypt, - key_asymmetric_encrypt, - key_asymmetric_duplicate, - key_asymmetric_sign, - key_asymmetric_verify, - key_asymmetric_destroy, - key_asymmetric_export, - key_asymmetric_import, - key_asymmetric_derive_key, -}; - -C_ASSERT( ARRAYSIZE(__wine_unix_call_funcs) == unix_funcs_count ); - -#ifdef _WIN64 - -typedef ULONG PTR32; - -struct key_symmetric32 -{ - enum chain_mode mode; - ULONG block_size; - PTR32 vector; - ULONG vector_len; - PTR32 secret; - ULONG secret_len; - ULONG __cs[6]; -}; - -struct key_asymmetric32 -{ - ULONG bitlen; /* ignored for ECC keys */ - enum ecc_curve_id curve_id; - ULONG flags; - DSSSEED dss_seed; -}; - -struct key32 -{ - struct object hdr; - enum alg_id alg_id; - UINT64 private[PRIVATE_DATA_SIZE]; /* private data for backend */ - union - { - struct key_symmetric32 s; - struct key_asymmetric32 a; - } u; -}; - -union padding -{ - BCRYPT_OAEP_PADDING_INFO oaep; - BCRYPT_PKCS1_PADDING_INFO pkcs1; - BCRYPT_PSS_PADDING_INFO pss; -}; - -union padding32 -{ - struct - { - PTR32 pszAlgId; - PTR32 pbLabel; - ULONG cbLabel; - } oaep; - struct - { - PTR32 pszAlgId; - } pkcs1; - struct - { - PTR32 pszAlgId; - ULONG cbSalt; - } pss; -}; - -static union padding *get_padding( union padding32 *padding32, union padding *padding, ULONG flags) -{ - if (!padding32) return NULL; - - switch (flags) - { - case BCRYPT_PAD_OAEP: - padding->oaep.pszAlgId = ULongToPtr( padding32->oaep.pszAlgId ); - padding->oaep.pbLabel = ULongToPtr( padding32->oaep.pbLabel ); - padding->oaep.cbLabel = padding32->oaep.cbLabel; - return padding; - - case BCRYPT_PAD_PKCS1: - padding->pkcs1.pszAlgId = ULongToPtr( padding32->pkcs1.pszAlgId ); - return padding; - - case BCRYPT_PAD_PSS: - padding->pss.pszAlgId = ULongToPtr( padding32->pss.pszAlgId ); - padding->pss.cbSalt = padding32->pss.cbSalt; - return padding; - - default: - break; - } - return NULL; -} - -static struct key *get_symmetric_key( struct key32 *key32, struct key *key ) -{ - key->hdr = key32->hdr; - key->alg_id = key32->alg_id; - memcpy( key->private, key32->private, sizeof(key->private) ); - key->u.s.mode = key32->u.s.mode; - key->u.s.block_size = key32->u.s.block_size; - key->u.s.vector = ULongToPtr(key32->u.s.vector); - key->u.s.vector_len = key32->u.s.vector_len; - key->u.s.secret = ULongToPtr(key32->u.s.secret); - key->u.s.secret_len = key32->u.s.secret_len; - return key; -} - -static struct key *get_asymmetric_key( struct key32 *key32, struct key *key ) -{ - key->hdr = key32->hdr; - key->alg_id = key32->alg_id; - memcpy( key->private, key32->private, sizeof(key->private) ); - key->u.a.bitlen = key32->u.a.bitlen; - key->u.a.curve_id = key32->u.a.curve_id; - key->u.a.flags = key32->u.a.flags; - key->u.a.dss_seed = key32->u.a.dss_seed; - return key; -} - -static void put_symmetric_key32( struct key *key, struct key32 *key32 ) -{ - memcpy( key32->private, key->private, sizeof(key32->private) ); -} - -static void put_asymmetric_key32( struct key *key, struct key32 *key32 ) -{ - memcpy( key32->private, key->private, sizeof(key32->private) ); - key32->u.a.curve_id = key->u.a.curve_id; - key32->u.a.flags = key->u.a.flags; - key32->u.a.dss_seed = key->u.a.dss_seed; -} - -static NTSTATUS wow64_key_symmetric_vector_reset( void *args ) -{ - NTSTATUS ret; - struct key key; - struct key32 *key32 = args; - - ret = key_symmetric_vector_reset( get_symmetric_key( key32, &key )); - put_symmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_symmetric_set_auth_data( void *args ) -{ - struct - { - PTR32 key; - PTR32 auth_data; - ULONG len; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_symmetric_set_auth_data_params params = - { - get_symmetric_key( key32, &key ), - ULongToPtr(params32->auth_data), - params32->len - }; - - ret = key_symmetric_set_auth_data( ¶ms ); - put_symmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_symmetric_encrypt( void *args ) -{ - struct - { - PTR32 key; - PTR32 input; - ULONG input_len; - PTR32 output; - ULONG output_len; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_symmetric_encrypt_params params = - { - get_symmetric_key( key32, &key ), - ULongToPtr(params32->input), - params32->input_len, - ULongToPtr(params32->output), - params32->output_len - }; - - ret = key_symmetric_encrypt( ¶ms ); - put_symmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_symmetric_decrypt( void *args ) -{ - struct - { - PTR32 key; - PTR32 input; - ULONG input_len; - PTR32 output; - ULONG output_len; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_symmetric_decrypt_params params = - { - get_symmetric_key( key32, &key ), - ULongToPtr(params32->input), - params32->input_len, - ULongToPtr(params32->output), - params32->output_len - }; - - ret = key_symmetric_decrypt( ¶ms ); - put_symmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_symmetric_get_tag( void *args ) -{ - struct - { - PTR32 key; - PTR32 tag; - ULONG len; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_symmetric_get_tag_params params = - { - get_symmetric_key( key32, &key ), - ULongToPtr(params32->tag), - params32->len - }; - - ret = key_symmetric_get_tag( ¶ms ); - put_symmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_symmetric_destroy( void *args ) -{ - struct key32 *key32 = args; - struct key key; - - return key_symmetric_destroy( get_symmetric_key( key32, &key )); -} - -static NTSTATUS wow64_key_asymmetric_generate( void *args ) -{ - struct key32 *key32 = args; - struct key key; - NTSTATUS ret; - - ret = key_asymmetric_generate( get_asymmetric_key( key32, &key )); - put_asymmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_asymmetric_decrypt( void *args ) -{ - struct - { - PTR32 key; - PTR32 input; - ULONG input_len; - PTR32 padding; - PTR32 output; - ULONG output_len; - PTR32 ret_len; - ULONG flags; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - union padding padding; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_asymmetric_decrypt_params params = - { - get_asymmetric_key( key32, &key ), - ULongToPtr(params32->input), - params32->input_len, - get_padding( ULongToPtr(params32->padding), &padding, params32->flags ), - ULongToPtr(params32->output), - params32->output_len, - ULongToPtr(params32->ret_len), - params32->flags - }; - - ret = key_asymmetric_decrypt( ¶ms ); - put_asymmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_asymmetric_encrypt( void *args ) -{ - struct - { - PTR32 key; - PTR32 input; - ULONG input_len; - PTR32 padding; - PTR32 output; - ULONG output_len; - PTR32 ret_len; - ULONG flags; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - union padding padding; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_asymmetric_encrypt_params params = - { - get_asymmetric_key( key32, &key ), - ULongToPtr(params32->input), - params32->input_len, - get_padding( ULongToPtr(params32->padding), &padding, params32->flags ), - ULongToPtr(params32->output), - params32->output_len, - ULongToPtr(params32->ret_len), - params32->flags - }; - - ret = key_asymmetric_encrypt( ¶ms ); - put_asymmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_asymmetric_duplicate( void *args ) -{ - struct - { - PTR32 key_orig; - PTR32 key_copy; - } const *params32 = args; - - NTSTATUS ret; - struct key key_orig, key_copy; - struct key32 *key_orig32 = ULongToPtr( params32->key_orig ); - struct key32 *key_copy32 = ULongToPtr( params32->key_copy ); - struct key_asymmetric_duplicate_params params = - { - get_asymmetric_key( key_orig32, &key_orig ), - get_asymmetric_key( key_copy32, &key_copy ) - }; - - ret = key_asymmetric_duplicate( ¶ms ); - put_asymmetric_key32( &key_copy, key_copy32 ); - return ret; -} - -static NTSTATUS wow64_key_asymmetric_sign( void *args ) -{ - struct - { - PTR32 key; - PTR32 padding; - PTR32 input; - ULONG input_len; - PTR32 output; - ULONG output_len; - PTR32 ret_len; - ULONG flags; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - union padding padding; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_asymmetric_sign_params params = - { - get_asymmetric_key( key32, &key ), - get_padding(ULongToPtr( params32->padding ), &padding, params32->flags), - ULongToPtr(params32->input), - params32->input_len, - ULongToPtr(params32->output), - params32->output_len, - ULongToPtr(params32->ret_len), - params32->flags - }; - - ret = key_asymmetric_sign( ¶ms ); - put_asymmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_asymmetric_verify( void *args ) -{ - struct - { - PTR32 key; - PTR32 padding; - PTR32 hash; - ULONG hash_len; - PTR32 signature; - ULONG signature_len; - ULONG flags; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - union padding padding; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_asymmetric_verify_params params = - { - get_asymmetric_key( key32, &key ), - get_padding(ULongToPtr( params32->padding ), &padding, params32->flags), - ULongToPtr(params32->hash), - params32->hash_len, - ULongToPtr(params32->signature), - params32->signature_len, - params32->flags - }; - - ret = key_asymmetric_verify( ¶ms ); - put_asymmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_asymmetric_destroy( void *args ) -{ - struct key32 *key32 = args; - struct key key; - - return key_asymmetric_destroy( get_asymmetric_key( key32, &key )); -} - -static NTSTATUS wow64_key_asymmetric_export( void *args ) -{ - struct - { - PTR32 key; - ULONG flags; - PTR32 buf; - ULONG len; - PTR32 ret_len; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_asymmetric_export_params params = - { - get_asymmetric_key( key32, &key ), - params32->flags, - ULongToPtr(params32->buf), - params32->len, - ULongToPtr(params32->ret_len), - }; - - ret = key_asymmetric_export( ¶ms ); - put_asymmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_asymmetric_import( void *args ) -{ - struct - { - PTR32 key; - ULONG flags; - PTR32 buf; - ULONG len; - } const *params32 = args; - - NTSTATUS ret; - struct key key; - struct key32 *key32 = ULongToPtr( params32->key ); - struct key_asymmetric_import_params params = - { - get_asymmetric_key( key32, &key ), - params32->flags, - ULongToPtr(params32->buf), - params32->len - }; - - ret = key_asymmetric_import( ¶ms ); - put_asymmetric_key32( &key, key32 ); - return ret; -} - -static NTSTATUS wow64_key_asymmetric_derive_key( void *args ) -{ - struct - { - PTR32 privkey; - PTR32 pubkey; - PTR32 output; - ULONG output_len; - PTR32 ret_len; - } const *params32 = args; - - NTSTATUS ret; - struct key privkey, pubkey; - struct key32 *privkey32 = ULongToPtr( params32->privkey ); - struct key32 *pubkey32 = ULongToPtr( params32->pubkey ); - struct key_asymmetric_derive_key_params params = - { - get_asymmetric_key( privkey32, &privkey ), - get_asymmetric_key( pubkey32, &pubkey ), - ULongToPtr(params32->output), - params32->output_len, - ULongToPtr(params32->ret_len), - }; - - ret = key_asymmetric_derive_key( ¶ms ); - put_asymmetric_key32( &privkey, privkey32 ); - put_asymmetric_key32( &pubkey, pubkey32 ); - return ret; -} - -const unixlib_entry_t __wine_unix_call_wow64_funcs[] = -{ - gnutls_process_attach, - gnutls_process_detach, - wow64_key_symmetric_vector_reset, - wow64_key_symmetric_set_auth_data, - wow64_key_symmetric_encrypt, - wow64_key_symmetric_decrypt, - wow64_key_symmetric_get_tag, - wow64_key_symmetric_destroy, - wow64_key_asymmetric_generate, - wow64_key_asymmetric_decrypt, - wow64_key_asymmetric_encrypt, - wow64_key_asymmetric_duplicate, - wow64_key_asymmetric_sign, - wow64_key_asymmetric_verify, - wow64_key_asymmetric_destroy, - wow64_key_asymmetric_export, - wow64_key_asymmetric_import, - wow64_key_asymmetric_derive_key, -}; - -C_ASSERT( ARRAYSIZE(__wine_unix_call_wow64_funcs) == unix_funcs_count ); - -#endif /* _WIN64 */ - -#endif /* HAVE_GNUTLS_CIPHER_INIT */ -- GitLab https://gitlab.winehq.org/wine/wine/-/merge_requests/11097
participants (2)
-
Hans Leidekker -
Hans Leidekker (＠hans)