Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-11-18 13:01:40 +00:00
Code Issues Projects Releases Packages Wiki Activity

[ADVAPI32_WINETEST] Add tests for SystemFunction004 and SystemFunction005.

Browse source 
These tests show that the required output buffer size it returned in out.Length field when STATUS_BUFFER_TOO_SMALL is returned.

This patch will be sent upstream.
This commit is contained in:
Eric Kohl 2018-09-17 23:12:16 +02:00
parent 5e2c4657ca
commit ffd1293f99
1 changed files with 120 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

120
modules/rostests/winetests/advapi32/crypt.c
View file

@ -28,6 +28,12 @@
#include "wine/test.h"
struct ustring {
DWORD Length;
DWORD MaximumLength;
unsigned char *Buffer;
};
static const char szRsaBaseProv[] = MS_DEF_PROV_A;
static const char szNonExistentProv[] = "Wine Nonexistent Cryptographic Provider v11.2";
static const char szKeySet[] = "wine_test_keyset";
@ -64,6 +70,8 @@ static BOOL (WINAPI *pCryptSetHashParam)(HCRYPTKEY, DWORD, BYTE*, DWORD);
static BOOL (WINAPI *pCryptSetKeyParam)(HCRYPTKEY, DWORD, BYTE*, DWORD);
static BOOL (WINAPI *pCryptSetProvParam)(HCRYPTPROV, DWORD, BYTE*, DWORD);
static BOOL (WINAPI *pCryptVerifySignatureW)(HCRYPTHASH, BYTE*, DWORD, HCRYPTKEY, LPCWSTR, DWORD);
static NTSTATUS (WINAPI *pSystemFunction004)(struct ustring*,struct ustring*, struct ustring*);
static NTSTATUS (WINAPI *pSystemFunction005)(struct ustring*,struct ustring*, struct ustring*);
static BOOLEAN (WINAPI *pSystemFunction036)(PVOID, ULONG);
static void init_function_pointers(void)
@ -100,6 +108,8 @@ static void init_function_pointers(void)
pCryptSetKeyParam = (void*)GetProcAddress(hadvapi32, "CryptSetKeyParam");
pCryptSetProvParam = (void*)GetProcAddress(hadvapi32, "CryptSetProvParam");
pCryptVerifySignatureW = (void*)GetProcAddress(hadvapi32, "CryptVerifySignatureW");
pSystemFunction004 = (void*)GetProcAddress(hadvapi32, "SystemFunction004");
pSystemFunction005 = (void*)GetProcAddress(hadvapi32, "SystemFunction005");
pSystemFunction036 = (void*)GetProcAddress(hadvapi32, "SystemFunction036");
}
@ -1120,6 +1130,114 @@ static void test_rc2_keylen(void)
}
}
static void test_SystemFunction004(void)
{
struct ustring inData;
struct ustring keyData;
struct ustring outData;
char inString[] = "Testdata for encryption";
char keyString[] = "EncryptionKey";
unsigned char outBuffer[32];
NTSTATUS Status;
#if 0
int i;
#endif
if (!pSystemFunction004)
{
win_skip("SystemFunction004 is not available\n");
return;
}
inData.Length = strlen(inString) + 1;
inData.MaximumLength = inData.Length;
inData.Buffer = (unsigned char *)inString;
keyData.Length = strlen(keyString) + 1;
keyData.MaximumLength = keyData.Length;
keyData.Buffer = (unsigned char *)keyString;
outData.Length = 0;
outData.MaximumLength = 0;
outData.Buffer = NULL;
Status = pSystemFunction004(&inData, &keyData, &outData);
ok(Status == STATUS_BUFFER_TOO_SMALL, "Expected SystemFunction004 to return STATUS_BUFFER_TOO_SMALL, got 0x%08lx\n", Status);
ok(outData.Length == 32, "Expected outData.Length to be 32, got %lu\n", outData.Length);
ok(outData.MaximumLength == 0, "Expected outData.MaximumLength to be 0, got %lu\n", outData.MaximumLength);
ok(outData.Buffer == NULL, "Expected outData.Length to be NULL, got %p\n", outData.Buffer);
outData.Length = sizeof(outBuffer);
outData.MaximumLength = outData.Length;
outData.Buffer = outBuffer;
Status = pSystemFunction004(&inData, &keyData, &outData);
ok(Status == STATUS_SUCCESS, "Expected SystemFunction004 to return STATUS_SUCCESS, got 0x%08lx\n", Status);
ok(outData.Length == 32, "Expected outData.Length to be 32, got %lu\n", outData.Length);
ok(outData.MaximumLength == 32, "Expected outData.MaximumLength to be 32, got %lu\n", outData.MaximumLength);
ok(outData.Buffer != NULL, "Expected outData.Buffer not to be NULL, got %p\n", outData.Buffer);
#if 0
if (Status == STATUS_SUCCESS)
{
printf("outData.Buffer:\n");
for (i = 0; i < sizeof(outBuffer); i++)
printf("0x%02x ", outBuffer[i]);
printf("\n");
}
#endif
}
static void test_SystemFunction005(void)
{
struct ustring inData;
struct ustring keyData;
struct ustring outData;
unsigned char inBuffer[32] = {0xdc, 0xff, 0x05, 0x8d, 0xaf, 0xb6, 0xe2, 0x8c, 0x4f, 0xee, 0x00, 0x06, 0xac, 0x1d, 0x56, 0xf1,
0x24, 0xbd, 0x17, 0xe0, 0xf6, 0xb8, 0x6d, 0x3a, 0x69, 0x5d, 0x14, 0xf9, 0x5a, 0x54, 0x93, 0xd1};
char keyString[] = "EncryptionKey";
char outBuffer[24];
NTSTATUS Status;
if (!pSystemFunction005)
{
win_skip("SystemFunction005 is not available\n");
return;
}
inData.Length = sizeof(inBuffer);
inData.MaximumLength = inData.Length;
inData.Buffer = inBuffer;
keyData.Length = strlen(keyString) + 1;
keyData.MaximumLength = keyData.Length;
keyData.Buffer = (unsigned char *)keyString;
outData.Length = 0;
outData.MaximumLength = 0;
outData.Buffer = NULL;
Status = pSystemFunction005(&inData, &keyData, &outData);
ok(Status == STATUS_BUFFER_TOO_SMALL, "Expected SystemFunction005 to return STATUS_BUFFER_TOO_SMALL, got 0x%08lx\n", Status);
ok(outData.Length == 24, "Expected outData.Length to be 24, got %lu\n", outData.Length);
ok(outData.MaximumLength == 0, "Expected outData.MaximumLength to be 0, got %lu\n", outData.MaximumLength);
ok(outData.Buffer == NULL, "Expected outData.Buffer to be NULL, got %p\n", outData.Buffer);
outData.Length = sizeof(outBuffer);
outData.MaximumLength = outData.Length;
outData.Buffer = (unsigned char *)outBuffer;
Status = pSystemFunction005(&inData, &keyData, &outData);
ok(Status == STATUS_SUCCESS, "Expected SystemFunction005 to return STATUS_SUCCESS, got 0x%08lx\n", Status);
ok(outData.Length == 24, "Expected outData.Length to be 24, got %lu\n", outData.Length);
ok(outData.MaximumLength == 24, "Expected outData.MaximumLength to be 24, got %lu\n", outData.MaximumLength);
ok(outData.Buffer != NULL, "Expected outData.Buffer not to be NULL, got %p\n", outData.Buffer);
#if 0
if (Status == STATUS_SUCCESS)
printf("outData.Buffer: '%s'\n", outData.Buffer);
#endif
}
static void test_SystemFunction036(void)
{
BOOL ret;
@ -1200,5 +1318,7 @@ START_TEST(crypt)
test_enum_provider_types();
test_get_default_provider();
test_set_provider_ex();
test_SystemFunction004();
test_SystemFunction005();
test_SystemFunction036();
}