2246 lines
84 KiB
C
2246 lines
84 KiB
C
/* Unit test suite for Rtl* API functions
|
|
*
|
|
* Copyright 2003 Thomas Mertes
|
|
*
|
|
* 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
|
|
*
|
|
* NOTES
|
|
* We use function pointers here as there is no import library for NTDLL on
|
|
* windows.
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include "ntdll_test.h"
|
|
#include "inaddr.h"
|
|
|
|
#ifndef __WINE_WINTERNL_H
|
|
|
|
typedef struct _RTL_HANDLE
|
|
{
|
|
struct _RTL_HANDLE * Next;
|
|
} RTL_HANDLE;
|
|
|
|
typedef struct _RTL_HANDLE_TABLE
|
|
{
|
|
ULONG MaxHandleCount;
|
|
ULONG HandleSize;
|
|
ULONG Unused[2];
|
|
PVOID NextFree;
|
|
PVOID FirstHandle;
|
|
PVOID ReservedMemory;
|
|
PVOID MaxHandle;
|
|
} RTL_HANDLE_TABLE;
|
|
|
|
#endif
|
|
|
|
/* avoid #include <winsock2.h> */
|
|
#undef htons
|
|
#ifdef WORDS_BIGENDIAN
|
|
#define htons(s) ((USHORT)(s))
|
|
#else /* WORDS_BIGENDIAN */
|
|
static inline USHORT __my_ushort_swap(USHORT s)
|
|
{
|
|
return (s >> 8) | (s << 8);
|
|
}
|
|
#define htons(s) __my_ushort_swap(s)
|
|
#endif /* WORDS_BIGENDIAN */
|
|
|
|
|
|
|
|
/* Function ptrs for ntdll calls */
|
|
static HMODULE hntdll = 0;
|
|
static SIZE_T (WINAPI *pRtlCompareMemory)(LPCVOID,LPCVOID,SIZE_T);
|
|
static SIZE_T (WINAPI *pRtlCompareMemoryUlong)(PULONG, SIZE_T, ULONG);
|
|
static NTSTATUS (WINAPI *pRtlDeleteTimer)(HANDLE, HANDLE, HANDLE);
|
|
static VOID (WINAPI *pRtlMoveMemory)(LPVOID,LPCVOID,SIZE_T);
|
|
static VOID (WINAPI *pRtlFillMemory)(LPVOID,SIZE_T,BYTE);
|
|
static VOID (WINAPI *pRtlFillMemoryUlong)(LPVOID,SIZE_T,ULONG);
|
|
static VOID (WINAPI *pRtlZeroMemory)(LPVOID,SIZE_T);
|
|
static ULONGLONG (WINAPIV *pRtlUlonglongByteSwap)(ULONGLONG source);
|
|
static ULONG (WINAPI *pRtlUniform)(PULONG);
|
|
static ULONG (WINAPI *pRtlRandom)(PULONG);
|
|
static BOOLEAN (WINAPI *pRtlAreAllAccessesGranted)(ACCESS_MASK, ACCESS_MASK);
|
|
static BOOLEAN (WINAPI *pRtlAreAnyAccessesGranted)(ACCESS_MASK, ACCESS_MASK);
|
|
static DWORD (WINAPI *pRtlComputeCrc32)(DWORD,const BYTE*,INT);
|
|
static void (WINAPI * pRtlInitializeHandleTable)(ULONG, ULONG, RTL_HANDLE_TABLE *);
|
|
static BOOLEAN (WINAPI * pRtlIsValidIndexHandle)(const RTL_HANDLE_TABLE *, ULONG, RTL_HANDLE **);
|
|
static NTSTATUS (WINAPI * pRtlDestroyHandleTable)(RTL_HANDLE_TABLE *);
|
|
static RTL_HANDLE * (WINAPI * pRtlAllocateHandle)(RTL_HANDLE_TABLE *, ULONG *);
|
|
static BOOLEAN (WINAPI * pRtlFreeHandle)(RTL_HANDLE_TABLE *, RTL_HANDLE *);
|
|
static NTSTATUS (WINAPI *pRtlAllocateAndInitializeSid)(PSID_IDENTIFIER_AUTHORITY,BYTE,DWORD,DWORD,DWORD,DWORD,DWORD,DWORD,DWORD,DWORD,PSID*);
|
|
static NTSTATUS (WINAPI *pRtlFreeSid)(PSID);
|
|
static DWORD (WINAPI *pRtlGetThreadErrorMode)(void);
|
|
static NTSTATUS (WINAPI *pRtlSetThreadErrorMode)(DWORD, LPDWORD);
|
|
static IMAGE_BASE_RELOCATION *(WINAPI *pLdrProcessRelocationBlock)(void*,UINT,USHORT*,INT_PTR);
|
|
static CHAR * (WINAPI *pRtlIpv4AddressToStringA)(const IN_ADDR *, LPSTR);
|
|
static NTSTATUS (WINAPI *pRtlIpv4AddressToStringExA)(const IN_ADDR *, USHORT, LPSTR, PULONG);
|
|
static NTSTATUS (WINAPI *pRtlIpv4StringToAddressA)(PCSTR, BOOLEAN, PCSTR *, IN_ADDR *);
|
|
static NTSTATUS (WINAPI *pLdrAddRefDll)(ULONG, HMODULE);
|
|
static NTSTATUS (WINAPI *pLdrLockLoaderLock)(ULONG, ULONG*, ULONG_PTR*);
|
|
static NTSTATUS (WINAPI *pLdrUnlockLoaderLock)(ULONG, ULONG_PTR);
|
|
static NTSTATUS (WINAPI *pRtlGetCompressionWorkSpaceSize)(USHORT, PULONG, PULONG);
|
|
static NTSTATUS (WINAPI *pRtlDecompressBuffer)(USHORT, PUCHAR, ULONG, const UCHAR*, ULONG, PULONG);
|
|
static NTSTATUS (WINAPI *pRtlDecompressFragment)(USHORT, PUCHAR, ULONG, const UCHAR*, ULONG, ULONG, PULONG, PVOID);
|
|
static NTSTATUS (WINAPI *pRtlCompressBuffer)(USHORT, const UCHAR*, ULONG, PUCHAR, ULONG, ULONG, PULONG, PVOID);
|
|
static BOOL (WINAPI *pRtlIsCriticalSectionLocked)(CRITICAL_SECTION *);
|
|
static BOOL (WINAPI *pRtlIsCriticalSectionLockedByThread)(CRITICAL_SECTION *);
|
|
static NTSTATUS (WINAPI *pRtlInitializeCriticalSectionEx)(CRITICAL_SECTION *, ULONG, ULONG);
|
|
static NTSTATUS (WINAPI *pLdrEnumerateLoadedModules)(void *, void *, void *);
|
|
|
|
static HMODULE hkernel32 = 0;
|
|
static BOOL (WINAPI *pIsWow64Process)(HANDLE, PBOOL);
|
|
|
|
|
|
#define LEN 16
|
|
static const char* src_src = "This is a test!"; /* 16 bytes long, incl NUL */
|
|
static ULONG src_aligned_block[4];
|
|
static ULONG dest_aligned_block[32];
|
|
static const char *src = (const char*)src_aligned_block;
|
|
static char* dest = (char*)dest_aligned_block;
|
|
|
|
static void InitFunctionPtrs(void)
|
|
{
|
|
hntdll = LoadLibraryA("ntdll.dll");
|
|
ok(hntdll != 0, "LoadLibrary failed\n");
|
|
if (hntdll) {
|
|
pRtlCompareMemory = (void *)GetProcAddress(hntdll, "RtlCompareMemory");
|
|
pRtlCompareMemoryUlong = (void *)GetProcAddress(hntdll, "RtlCompareMemoryUlong");
|
|
pRtlDeleteTimer = (void *)GetProcAddress(hntdll, "RtlDeleteTimer");
|
|
pRtlMoveMemory = (void *)GetProcAddress(hntdll, "RtlMoveMemory");
|
|
pRtlFillMemory = (void *)GetProcAddress(hntdll, "RtlFillMemory");
|
|
pRtlFillMemoryUlong = (void *)GetProcAddress(hntdll, "RtlFillMemoryUlong");
|
|
pRtlZeroMemory = (void *)GetProcAddress(hntdll, "RtlZeroMemory");
|
|
pRtlUlonglongByteSwap = (void *)GetProcAddress(hntdll, "RtlUlonglongByteSwap");
|
|
pRtlUniform = (void *)GetProcAddress(hntdll, "RtlUniform");
|
|
pRtlRandom = (void *)GetProcAddress(hntdll, "RtlRandom");
|
|
pRtlAreAllAccessesGranted = (void *)GetProcAddress(hntdll, "RtlAreAllAccessesGranted");
|
|
pRtlAreAnyAccessesGranted = (void *)GetProcAddress(hntdll, "RtlAreAnyAccessesGranted");
|
|
pRtlComputeCrc32 = (void *)GetProcAddress(hntdll, "RtlComputeCrc32");
|
|
pRtlInitializeHandleTable = (void *)GetProcAddress(hntdll, "RtlInitializeHandleTable");
|
|
pRtlIsValidIndexHandle = (void *)GetProcAddress(hntdll, "RtlIsValidIndexHandle");
|
|
pRtlDestroyHandleTable = (void *)GetProcAddress(hntdll, "RtlDestroyHandleTable");
|
|
pRtlAllocateHandle = (void *)GetProcAddress(hntdll, "RtlAllocateHandle");
|
|
pRtlFreeHandle = (void *)GetProcAddress(hntdll, "RtlFreeHandle");
|
|
pRtlAllocateAndInitializeSid = (void *)GetProcAddress(hntdll, "RtlAllocateAndInitializeSid");
|
|
pRtlFreeSid = (void *)GetProcAddress(hntdll, "RtlFreeSid");
|
|
pRtlGetThreadErrorMode = (void *)GetProcAddress(hntdll, "RtlGetThreadErrorMode");
|
|
pRtlSetThreadErrorMode = (void *)GetProcAddress(hntdll, "RtlSetThreadErrorMode");
|
|
pLdrProcessRelocationBlock = (void *)GetProcAddress(hntdll, "LdrProcessRelocationBlock");
|
|
pRtlIpv4AddressToStringA = (void *)GetProcAddress(hntdll, "RtlIpv4AddressToStringA");
|
|
pRtlIpv4AddressToStringExA = (void *)GetProcAddress(hntdll, "RtlIpv4AddressToStringExA");
|
|
pRtlIpv4StringToAddressA = (void *)GetProcAddress(hntdll, "RtlIpv4StringToAddressA");
|
|
pLdrAddRefDll = (void *)GetProcAddress(hntdll, "LdrAddRefDll");
|
|
pLdrLockLoaderLock = (void *)GetProcAddress(hntdll, "LdrLockLoaderLock");
|
|
pLdrUnlockLoaderLock = (void *)GetProcAddress(hntdll, "LdrUnlockLoaderLock");
|
|
pRtlGetCompressionWorkSpaceSize = (void *)GetProcAddress(hntdll, "RtlGetCompressionWorkSpaceSize");
|
|
pRtlDecompressBuffer = (void *)GetProcAddress(hntdll, "RtlDecompressBuffer");
|
|
pRtlDecompressFragment = (void *)GetProcAddress(hntdll, "RtlDecompressFragment");
|
|
pRtlCompressBuffer = (void *)GetProcAddress(hntdll, "RtlCompressBuffer");
|
|
pRtlIsCriticalSectionLocked = (void *)GetProcAddress(hntdll, "RtlIsCriticalSectionLocked");
|
|
pRtlIsCriticalSectionLockedByThread = (void *)GetProcAddress(hntdll, "RtlIsCriticalSectionLockedByThread");
|
|
pRtlInitializeCriticalSectionEx = (void *)GetProcAddress(hntdll, "RtlInitializeCriticalSectionEx");
|
|
pLdrEnumerateLoadedModules = (void *)GetProcAddress(hntdll, "LdrEnumerateLoadedModules");
|
|
}
|
|
hkernel32 = LoadLibraryA("kernel32.dll");
|
|
ok(hkernel32 != 0, "LoadLibrary failed\n");
|
|
if (hkernel32) {
|
|
pIsWow64Process = (void *)GetProcAddress(hkernel32, "IsWow64Process");
|
|
}
|
|
strcpy((char*)src_aligned_block, src_src);
|
|
ok(strlen(src) == 15, "Source must be 16 bytes long!\n");
|
|
}
|
|
|
|
#define COMP(str1,str2,cmplen,len) size = pRtlCompareMemory(str1, str2, cmplen); \
|
|
ok(size == len, "Expected %ld, got %ld\n", size, (SIZE_T)len)
|
|
|
|
static void test_RtlCompareMemory(void)
|
|
{
|
|
SIZE_T size;
|
|
|
|
if (!pRtlCompareMemory)
|
|
{
|
|
win_skip("RtlCompareMemory is not available\n");
|
|
return;
|
|
}
|
|
|
|
strcpy(dest, src);
|
|
|
|
COMP(src,src,0,0);
|
|
COMP(src,src,LEN,LEN);
|
|
dest[0] = 'x';
|
|
COMP(src,dest,LEN,0);
|
|
}
|
|
|
|
static void test_RtlCompareMemoryUlong(void)
|
|
{
|
|
ULONG a[10];
|
|
ULONG result;
|
|
|
|
if (!pRtlCompareMemoryUlong)
|
|
{
|
|
win_skip("RtlCompareMemoryUlong is not available\n");
|
|
return;
|
|
}
|
|
|
|
a[0]= 0x0123;
|
|
a[1]= 0x4567;
|
|
a[2]= 0x89ab;
|
|
a[3]= 0xcdef;
|
|
result = pRtlCompareMemoryUlong(a, 0, 0x0123);
|
|
ok(result == 0, "RtlCompareMemoryUlong(%p, 0, 0x0123) returns %u, expected 0\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 3, 0x0123);
|
|
ok(result == 0, "RtlCompareMemoryUlong(%p, 3, 0x0123) returns %u, expected 0\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 4, 0x0123);
|
|
ok(result == 4, "RtlCompareMemoryUlong(%p, 4, 0x0123) returns %u, expected 4\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 5, 0x0123);
|
|
ok(result == 4, "RtlCompareMemoryUlong(%p, 5, 0x0123) returns %u, expected 4\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 7, 0x0123);
|
|
ok(result == 4, "RtlCompareMemoryUlong(%p, 7, 0x0123) returns %u, expected 4\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 8, 0x0123);
|
|
ok(result == 4, "RtlCompareMemoryUlong(%p, 8, 0x0123) returns %u, expected 4\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 9, 0x0123);
|
|
ok(result == 4, "RtlCompareMemoryUlong(%p, 9, 0x0123) returns %u, expected 4\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 4, 0x0127);
|
|
ok(result == 0, "RtlCompareMemoryUlong(%p, 4, 0x0127) returns %u, expected 0\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 4, 0x7123);
|
|
ok(result == 0, "RtlCompareMemoryUlong(%p, 4, 0x7123) returns %u, expected 0\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 16, 0x4567);
|
|
ok(result == 0, "RtlCompareMemoryUlong(%p, 16, 0x4567) returns %u, expected 0\n", a, result);
|
|
|
|
a[1]= 0x0123;
|
|
result = pRtlCompareMemoryUlong(a, 3, 0x0123);
|
|
ok(result == 0, "RtlCompareMemoryUlong(%p, 3, 0x0123) returns %u, expected 0\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 4, 0x0123);
|
|
ok(result == 4, "RtlCompareMemoryUlong(%p, 4, 0x0123) returns %u, expected 4\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 5, 0x0123);
|
|
ok(result == 4, "RtlCompareMemoryUlong(%p, 5, 0x0123) returns %u, expected 4\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 7, 0x0123);
|
|
ok(result == 4, "RtlCompareMemoryUlong(%p, 7, 0x0123) returns %u, expected 4\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 8, 0x0123);
|
|
ok(result == 8, "RtlCompareMemoryUlong(%p, 8, 0x0123) returns %u, expected 8\n", a, result);
|
|
result = pRtlCompareMemoryUlong(a, 9, 0x0123);
|
|
ok(result == 8, "RtlCompareMemoryUlong(%p, 9, 0x0123) returns %u, expected 8\n", a, result);
|
|
}
|
|
|
|
#define COPY(len) memset(dest,0,sizeof(dest_aligned_block)); pRtlMoveMemory(dest, src, len)
|
|
#define CMP(str) ok(strcmp(dest,str) == 0, "Expected '%s', got '%s'\n", str, dest)
|
|
|
|
static void test_RtlMoveMemory(void)
|
|
{
|
|
if (!pRtlMoveMemory)
|
|
{
|
|
win_skip("RtlMoveMemory is not available\n");
|
|
return;
|
|
}
|
|
|
|
/* Length should be in bytes and not rounded. Use strcmp to ensure we
|
|
* didn't write past the end (it checks for the final NUL left by memset)
|
|
*/
|
|
COPY(0); CMP("");
|
|
COPY(1); CMP("T");
|
|
COPY(2); CMP("Th");
|
|
COPY(3); CMP("Thi");
|
|
COPY(4); CMP("This");
|
|
COPY(5); CMP("This ");
|
|
COPY(6); CMP("This i");
|
|
COPY(7); CMP("This is");
|
|
COPY(8); CMP("This is ");
|
|
COPY(9); CMP("This is a");
|
|
|
|
/* Overlapping */
|
|
strcpy(dest, src); pRtlMoveMemory(dest, dest + 1, strlen(src) - 1);
|
|
CMP("his is a test!!");
|
|
strcpy(dest, src); pRtlMoveMemory(dest + 1, dest, strlen(src));
|
|
CMP("TThis is a test!");
|
|
}
|
|
|
|
#define FILL(len) memset(dest,0,sizeof(dest_aligned_block)); strcpy(dest, src); pRtlFillMemory(dest,len,'x')
|
|
|
|
static void test_RtlFillMemory(void)
|
|
{
|
|
if (!pRtlFillMemory)
|
|
{
|
|
win_skip("RtlFillMemory is not available\n");
|
|
return;
|
|
}
|
|
|
|
/* Length should be in bytes and not rounded. Use strcmp to ensure we
|
|
* didn't write past the end (the remainder of the string should match)
|
|
*/
|
|
FILL(0); CMP("This is a test!");
|
|
FILL(1); CMP("xhis is a test!");
|
|
FILL(2); CMP("xxis is a test!");
|
|
FILL(3); CMP("xxxs is a test!");
|
|
FILL(4); CMP("xxxx is a test!");
|
|
FILL(5); CMP("xxxxxis a test!");
|
|
FILL(6); CMP("xxxxxxs a test!");
|
|
FILL(7); CMP("xxxxxxx a test!");
|
|
FILL(8); CMP("xxxxxxxxa test!");
|
|
FILL(9); CMP("xxxxxxxxx test!");
|
|
}
|
|
|
|
#define LFILL(len) memset(dest,0,sizeof(dest_aligned_block)); strcpy(dest, src); pRtlFillMemoryUlong(dest,len,val)
|
|
|
|
static void test_RtlFillMemoryUlong(void)
|
|
{
|
|
ULONG val = ('x' << 24) | ('x' << 16) | ('x' << 8) | 'x';
|
|
if (!pRtlFillMemoryUlong)
|
|
{
|
|
win_skip("RtlFillMemoryUlong is not available\n");
|
|
return;
|
|
}
|
|
|
|
/* Length should be in bytes and not rounded. Use strcmp to ensure we
|
|
* didn't write past the end (the remainder of the string should match)
|
|
*/
|
|
LFILL(0); CMP("This is a test!");
|
|
LFILL(1); CMP("This is a test!");
|
|
LFILL(2); CMP("This is a test!");
|
|
LFILL(3); CMP("This is a test!");
|
|
LFILL(4); CMP("xxxx is a test!");
|
|
LFILL(5); CMP("xxxx is a test!");
|
|
LFILL(6); CMP("xxxx is a test!");
|
|
LFILL(7); CMP("xxxx is a test!");
|
|
LFILL(8); CMP("xxxxxxxxa test!");
|
|
LFILL(9); CMP("xxxxxxxxa test!");
|
|
}
|
|
|
|
#define ZERO(len) memset(dest,0,sizeof(dest_aligned_block)); strcpy(dest, src); pRtlZeroMemory(dest,len)
|
|
#define MCMP(str) ok(memcmp(dest,str,LEN) == 0, "Memcmp failed\n")
|
|
|
|
static void test_RtlZeroMemory(void)
|
|
{
|
|
if (!pRtlZeroMemory)
|
|
{
|
|
win_skip("RtlZeroMemory is not available\n");
|
|
return;
|
|
}
|
|
|
|
/* Length should be in bytes and not rounded. */
|
|
ZERO(0); MCMP("This is a test!");
|
|
ZERO(1); MCMP("\0his is a test!");
|
|
ZERO(2); MCMP("\0\0is is a test!");
|
|
ZERO(3); MCMP("\0\0\0s is a test!");
|
|
ZERO(4); MCMP("\0\0\0\0 is a test!");
|
|
ZERO(5); MCMP("\0\0\0\0\0is a test!");
|
|
ZERO(6); MCMP("\0\0\0\0\0\0s a test!");
|
|
ZERO(7); MCMP("\0\0\0\0\0\0\0 a test!");
|
|
ZERO(8); MCMP("\0\0\0\0\0\0\0\0a test!");
|
|
ZERO(9); MCMP("\0\0\0\0\0\0\0\0\0 test!");
|
|
}
|
|
|
|
static void test_RtlUlonglongByteSwap(void)
|
|
{
|
|
ULONGLONG result;
|
|
|
|
if ( !pRtlUlonglongByteSwap )
|
|
{
|
|
win_skip("RtlUlonglongByteSwap is not available\n");
|
|
return;
|
|
}
|
|
|
|
if ( pRtlUlonglongByteSwap( 0 ) != 0 )
|
|
{
|
|
win_skip("Broken RtlUlonglongByteSwap in win2k\n");
|
|
return;
|
|
}
|
|
|
|
result = pRtlUlonglongByteSwap( ((ULONGLONG)0x76543210 << 32) | 0x87654321 );
|
|
ok( (((ULONGLONG)0x21436587 << 32) | 0x10325476) == result,
|
|
"RtlUlonglongByteSwap(0x7654321087654321) returns 0x%s, expected 0x2143658710325476\n",
|
|
wine_dbgstr_longlong(result));
|
|
}
|
|
|
|
|
|
static void test_RtlUniform(void)
|
|
{
|
|
ULONGLONG num;
|
|
ULONG seed;
|
|
ULONG seed_bak;
|
|
ULONG expected;
|
|
ULONG result;
|
|
|
|
if (!pRtlUniform)
|
|
{
|
|
win_skip("RtlUniform is not available\n");
|
|
return;
|
|
}
|
|
|
|
/*
|
|
* According to the documentation RtlUniform is using D.H. Lehmer's 1948
|
|
* algorithm. This algorithm is:
|
|
*
|
|
* seed = (seed * const_1 + const_2) % const_3;
|
|
*
|
|
* According to the documentation the random number is distributed over
|
|
* [0..MAXLONG]. Therefore const_3 is MAXLONG + 1:
|
|
*
|
|
* seed = (seed * const_1 + const_2) % (MAXLONG + 1);
|
|
*
|
|
* Because MAXLONG is 0x7fffffff (and MAXLONG + 1 is 0x80000000) the
|
|
* algorithm can be expressed without division as:
|
|
*
|
|
* seed = (seed * const_1 + const_2) & MAXLONG;
|
|
*
|
|
* To find out const_2 we just call RtlUniform with seed set to 0:
|
|
*/
|
|
seed = 0;
|
|
expected = 0x7fffffc3;
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"RtlUniform(&seed (seed == 0)) returns %x, expected %x\n",
|
|
result, expected);
|
|
/*
|
|
* The algorithm is now:
|
|
*
|
|
* seed = (seed * const_1 + 0x7fffffc3) & MAXLONG;
|
|
*
|
|
* To find out const_1 we can use:
|
|
*
|
|
* const_1 = RtlUniform(1) - 0x7fffffc3;
|
|
*
|
|
* If that does not work a search loop can try all possible values of
|
|
* const_1 and compare to the result to RtlUniform(1).
|
|
* This way we find out that const_1 is 0xffffffed.
|
|
*
|
|
* For seed = 1 the const_2 is 0x7fffffc4:
|
|
*/
|
|
seed = 1;
|
|
expected = seed * 0xffffffed + 0x7fffffc3 + 1;
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"RtlUniform(&seed (seed == 1)) returns %x, expected %x\n",
|
|
result, expected);
|
|
/*
|
|
* For seed = 2 the const_2 is 0x7fffffc3:
|
|
*/
|
|
seed = 2;
|
|
expected = seed * 0xffffffed + 0x7fffffc3;
|
|
result = pRtlUniform(&seed);
|
|
|
|
/*
|
|
* Windows Vista uses different algorithms, so skip the rest of the tests
|
|
* until that is figured out. Trace output for the failures is about 10.5 MB!
|
|
*/
|
|
|
|
if (result == 0x7fffff9f) {
|
|
skip("Most likely running on Windows Vista which uses a different algorithm\n");
|
|
return;
|
|
}
|
|
|
|
ok(result == expected,
|
|
"RtlUniform(&seed (seed == 2)) returns %x, expected %x\n",
|
|
result, expected);
|
|
|
|
/*
|
|
* More tests show that if seed is odd the result must be incremented by 1:
|
|
*/
|
|
seed = 3;
|
|
expected = seed * 0xffffffed + 0x7fffffc3 + (seed & 1);
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"RtlUniform(&seed (seed == 3)) returns %x, expected %x\n",
|
|
result, expected);
|
|
|
|
seed = 0x6bca1aa;
|
|
expected = seed * 0xffffffed + 0x7fffffc3;
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"RtlUniform(&seed (seed == 0x6bca1aa)) returns %x, expected %x\n",
|
|
result, expected);
|
|
|
|
seed = 0x6bca1ab;
|
|
expected = seed * 0xffffffed + 0x7fffffc3 + 1;
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"RtlUniform(&seed (seed == 0x6bca1ab)) returns %x, expected %x\n",
|
|
result, expected);
|
|
/*
|
|
* When seed is 0x6bca1ac there is an exception:
|
|
*/
|
|
seed = 0x6bca1ac;
|
|
expected = seed * 0xffffffed + 0x7fffffc3 + 2;
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"RtlUniform(&seed (seed == 0x6bca1ac)) returns %x, expected %x\n",
|
|
result, expected);
|
|
/*
|
|
* Note that up to here const_3 is not used
|
|
* (the highest bit of the result is not set).
|
|
*
|
|
* Starting with 0x6bca1ad: If seed is even the result must be incremented by 1:
|
|
*/
|
|
seed = 0x6bca1ad;
|
|
expected = (seed * 0xffffffed + 0x7fffffc3) & MAXLONG;
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"RtlUniform(&seed (seed == 0x6bca1ad)) returns %x, expected %x\n",
|
|
result, expected);
|
|
|
|
seed = 0x6bca1ae;
|
|
expected = (seed * 0xffffffed + 0x7fffffc3 + 1) & MAXLONG;
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"RtlUniform(&seed (seed == 0x6bca1ae)) returns %x, expected %x\n",
|
|
result, expected);
|
|
/*
|
|
* There are several ranges where for odd or even seed the result must be
|
|
* incremented by 1. You can see this ranges in the following test.
|
|
*
|
|
* For a full test use one of the following loop heads:
|
|
*
|
|
* for (num = 0; num <= 0xffffffff; num++) {
|
|
* seed = num;
|
|
* ...
|
|
*
|
|
* seed = 0;
|
|
* for (num = 0; num <= 0xffffffff; num++) {
|
|
* ...
|
|
*/
|
|
seed = 0;
|
|
for (num = 0; num <= 100000; num++) {
|
|
|
|
expected = seed * 0xffffffed + 0x7fffffc3;
|
|
if (seed < 0x6bca1ac) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed == 0x6bca1ac) {
|
|
expected = (expected + 2) & MAXLONG;
|
|
} else if (seed < 0xd79435c) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x1435e50b) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x1af286ba) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x21af2869) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x286bca18) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x2f286bc7) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x35e50d77) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x3ca1af26) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x435e50d5) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x4a1af284) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x50d79433) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x579435e2) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x5e50d792) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x650d7941) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x6bca1af0) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x7286bc9f) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x79435e4e) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x7ffffffd) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x86bca1ac) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed == 0x86bca1ac) {
|
|
expected = (expected + 1) & MAXLONG;
|
|
} else if (seed < 0x8d79435c) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0x9435e50b) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0x9af286ba) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0xa1af2869) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0xa86bca18) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0xaf286bc7) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed == 0xaf286bc7) {
|
|
expected = (expected + 2) & MAXLONG;
|
|
} else if (seed < 0xb5e50d77) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0xbca1af26) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0xc35e50d5) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0xca1af284) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0xd0d79433) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0xd79435e2) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0xde50d792) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0xe50d7941) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0xebca1af0) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0xf286bc9f) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else if (seed < 0xf9435e4e) {
|
|
expected = expected + (seed & 1);
|
|
} else if (seed < 0xfffffffd) {
|
|
expected = (expected + (~seed & 1)) & MAXLONG;
|
|
} else {
|
|
expected = expected + (seed & 1);
|
|
} /* if */
|
|
seed_bak = seed;
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"test: 0x%s RtlUniform(&seed (seed == %x)) returns %x, expected %x\n",
|
|
wine_dbgstr_longlong(num), seed_bak, result, expected);
|
|
ok(seed == expected,
|
|
"test: 0x%s RtlUniform(&seed (seed == %x)) sets seed to %x, expected %x\n",
|
|
wine_dbgstr_longlong(num), seed_bak, result, expected);
|
|
} /* for */
|
|
/*
|
|
* Further investigation shows: In the different regions the highest bit
|
|
* is set or cleared when even or odd seeds need an increment by 1.
|
|
* This leads to a simplified algorithm:
|
|
*
|
|
* seed = seed * 0xffffffed + 0x7fffffc3;
|
|
* if (seed == 0xffffffff || seed == 0x7ffffffe) {
|
|
* seed = (seed + 2) & MAXLONG;
|
|
* } else if (seed == 0x7fffffff) {
|
|
* seed = 0;
|
|
* } else if ((seed & 0x80000000) == 0) {
|
|
* seed = seed + (~seed & 1);
|
|
* } else {
|
|
* seed = (seed + (seed & 1)) & MAXLONG;
|
|
* }
|
|
*
|
|
* This is also the algorithm used for RtlUniform of wine (see dlls/ntdll/rtl.c).
|
|
*
|
|
* Now comes the funny part:
|
|
* It took me one weekend, to find the complicated algorithm and one day more,
|
|
* to find the simplified algorithm. Several weeks later I found out: The value
|
|
* MAXLONG (=0x7fffffff) is never returned, neither with the native function
|
|
* nor with the simplified algorithm. In reality the native function and our
|
|
* function return a random number distributed over [0..MAXLONG-1]. Note
|
|
* that this is different from what native documentation states [0..MAXLONG].
|
|
* Expressed with D.H. Lehmer's 1948 algorithm it looks like:
|
|
*
|
|
* seed = (seed * const_1 + const_2) % MAXLONG;
|
|
*
|
|
* Further investigations show that the real algorithm is:
|
|
*
|
|
* seed = (seed * 0x7fffffed + 0x7fffffc3) % MAXLONG;
|
|
*
|
|
* This is checked with the test below:
|
|
*/
|
|
seed = 0;
|
|
for (num = 0; num <= 100000; num++) {
|
|
expected = (seed * 0x7fffffed + 0x7fffffc3) % 0x7fffffff;
|
|
seed_bak = seed;
|
|
result = pRtlUniform(&seed);
|
|
ok(result == expected,
|
|
"test: 0x%s RtlUniform(&seed (seed == %x)) returns %x, expected %x\n",
|
|
wine_dbgstr_longlong(num), seed_bak, result, expected);
|
|
ok(seed == expected,
|
|
"test: 0x%s RtlUniform(&seed (seed == %x)) sets seed to %x, expected %x\n",
|
|
wine_dbgstr_longlong(num), seed_bak, result, expected);
|
|
} /* for */
|
|
/*
|
|
* More tests show that RtlUniform does not return 0x7ffffffd for seed values
|
|
* in the range [0..MAXLONG-1]. Additionally 2 is returned twice. This shows
|
|
* that there is more than one cycle of generated randon numbers ...
|
|
*/
|
|
}
|
|
|
|
|
|
static void test_RtlRandom(void)
|
|
{
|
|
int i, j;
|
|
ULONG seed;
|
|
ULONG res[512];
|
|
|
|
if (!pRtlRandom)
|
|
{
|
|
win_skip("RtlRandom is not available\n");
|
|
return;
|
|
}
|
|
|
|
seed = 0;
|
|
for (i = 0; i < sizeof(res) / sizeof(res[0]); i++)
|
|
{
|
|
res[i] = pRtlRandom(&seed);
|
|
ok(seed != res[i], "%i: seed is same as res %x\n", i, seed);
|
|
for (j = 0; j < i; j++)
|
|
ok(res[i] != res[j], "res[%i] (%x) is same as res[%i] (%x)\n", j, res[j], i, res[i]);
|
|
}
|
|
}
|
|
|
|
|
|
typedef struct {
|
|
ACCESS_MASK GrantedAccess;
|
|
ACCESS_MASK DesiredAccess;
|
|
BOOLEAN result;
|
|
} all_accesses_t;
|
|
|
|
static const all_accesses_t all_accesses[] = {
|
|
{0xFEDCBA76, 0xFEDCBA76, 1},
|
|
{0x00000000, 0xFEDCBA76, 0},
|
|
{0xFEDCBA76, 0x00000000, 1},
|
|
{0x00000000, 0x00000000, 1},
|
|
{0xFEDCBA76, 0xFEDCBA70, 1},
|
|
{0xFEDCBA70, 0xFEDCBA76, 0},
|
|
{0xFEDCBA76, 0xFEDC8A76, 1},
|
|
{0xFEDC8A76, 0xFEDCBA76, 0},
|
|
{0xFEDCBA76, 0xC8C4B242, 1},
|
|
{0xC8C4B242, 0xFEDCBA76, 0},
|
|
};
|
|
#define NB_ALL_ACCESSES (sizeof(all_accesses)/sizeof(*all_accesses))
|
|
|
|
|
|
static void test_RtlAreAllAccessesGranted(void)
|
|
{
|
|
unsigned int test_num;
|
|
BOOLEAN result;
|
|
|
|
if (!pRtlAreAllAccessesGranted)
|
|
{
|
|
win_skip("RtlAreAllAccessesGranted is not available\n");
|
|
return;
|
|
}
|
|
|
|
for (test_num = 0; test_num < NB_ALL_ACCESSES; test_num++) {
|
|
result = pRtlAreAllAccessesGranted(all_accesses[test_num].GrantedAccess,
|
|
all_accesses[test_num].DesiredAccess);
|
|
ok(all_accesses[test_num].result == result,
|
|
"(test %d): RtlAreAllAccessesGranted(%08x, %08x) returns %d, expected %d\n",
|
|
test_num, all_accesses[test_num].GrantedAccess,
|
|
all_accesses[test_num].DesiredAccess,
|
|
result, all_accesses[test_num].result);
|
|
} /* for */
|
|
}
|
|
|
|
|
|
typedef struct {
|
|
ACCESS_MASK GrantedAccess;
|
|
ACCESS_MASK DesiredAccess;
|
|
BOOLEAN result;
|
|
} any_accesses_t;
|
|
|
|
static const any_accesses_t any_accesses[] = {
|
|
{0xFEDCBA76, 0xFEDCBA76, 1},
|
|
{0x00000000, 0xFEDCBA76, 0},
|
|
{0xFEDCBA76, 0x00000000, 0},
|
|
{0x00000000, 0x00000000, 0},
|
|
{0xFEDCBA76, 0x01234589, 0},
|
|
{0x00040000, 0xFEDCBA76, 1},
|
|
{0x00040000, 0xFED8BA76, 0},
|
|
{0xFEDCBA76, 0x00040000, 1},
|
|
{0xFED8BA76, 0x00040000, 0},
|
|
};
|
|
#define NB_ANY_ACCESSES (sizeof(any_accesses)/sizeof(*any_accesses))
|
|
|
|
|
|
static void test_RtlAreAnyAccessesGranted(void)
|
|
{
|
|
unsigned int test_num;
|
|
BOOLEAN result;
|
|
|
|
if (!pRtlAreAnyAccessesGranted)
|
|
{
|
|
win_skip("RtlAreAnyAccessesGranted is not available\n");
|
|
return;
|
|
}
|
|
|
|
for (test_num = 0; test_num < NB_ANY_ACCESSES; test_num++) {
|
|
result = pRtlAreAnyAccessesGranted(any_accesses[test_num].GrantedAccess,
|
|
any_accesses[test_num].DesiredAccess);
|
|
ok(any_accesses[test_num].result == result,
|
|
"(test %d): RtlAreAnyAccessesGranted(%08x, %08x) returns %d, expected %d\n",
|
|
test_num, any_accesses[test_num].GrantedAccess,
|
|
any_accesses[test_num].DesiredAccess,
|
|
result, any_accesses[test_num].result);
|
|
} /* for */
|
|
}
|
|
|
|
static void test_RtlComputeCrc32(void)
|
|
{
|
|
DWORD crc = 0;
|
|
|
|
if (!pRtlComputeCrc32)
|
|
{
|
|
win_skip("RtlComputeCrc32 is not available\n");
|
|
return;
|
|
}
|
|
|
|
crc = pRtlComputeCrc32(crc, (const BYTE *)src, LEN);
|
|
ok(crc == 0x40861dc2,"Expected 0x40861dc2, got %8x\n", crc);
|
|
}
|
|
|
|
|
|
typedef struct MY_HANDLE
|
|
{
|
|
RTL_HANDLE RtlHandle;
|
|
void * MyValue;
|
|
} MY_HANDLE;
|
|
|
|
static inline void RtlpMakeHandleAllocated(RTL_HANDLE * Handle)
|
|
{
|
|
ULONG_PTR *AllocatedBit = (ULONG_PTR *)(&Handle->Next);
|
|
*AllocatedBit = *AllocatedBit | 1;
|
|
}
|
|
|
|
static void test_HandleTables(void)
|
|
{
|
|
BOOLEAN result;
|
|
NTSTATUS status;
|
|
ULONG Index;
|
|
MY_HANDLE * MyHandle;
|
|
RTL_HANDLE_TABLE HandleTable;
|
|
|
|
if (!pRtlInitializeHandleTable)
|
|
{
|
|
win_skip("RtlInitializeHandleTable is not available\n");
|
|
return;
|
|
}
|
|
|
|
pRtlInitializeHandleTable(0x3FFF, sizeof(MY_HANDLE), &HandleTable);
|
|
MyHandle = (MY_HANDLE *)pRtlAllocateHandle(&HandleTable, &Index);
|
|
ok(MyHandle != NULL, "RtlAllocateHandle failed\n");
|
|
RtlpMakeHandleAllocated(&MyHandle->RtlHandle);
|
|
MyHandle = NULL;
|
|
result = pRtlIsValidIndexHandle(&HandleTable, Index, (RTL_HANDLE **)&MyHandle);
|
|
ok(result, "Handle %p wasn't valid\n", MyHandle);
|
|
result = pRtlFreeHandle(&HandleTable, &MyHandle->RtlHandle);
|
|
ok(result, "Couldn't free handle %p\n", MyHandle);
|
|
status = pRtlDestroyHandleTable(&HandleTable);
|
|
ok(status == STATUS_SUCCESS, "RtlDestroyHandleTable failed with error 0x%08x\n", status);
|
|
}
|
|
|
|
static void test_RtlAllocateAndInitializeSid(void)
|
|
{
|
|
NTSTATUS ret;
|
|
SID_IDENTIFIER_AUTHORITY sia = {{ 1, 2, 3, 4, 5, 6 }};
|
|
PSID psid;
|
|
|
|
if (!pRtlAllocateAndInitializeSid)
|
|
{
|
|
win_skip("RtlAllocateAndInitializeSid is not available\n");
|
|
return;
|
|
}
|
|
|
|
ret = pRtlAllocateAndInitializeSid(&sia, 0, 1, 2, 3, 4, 5, 6, 7, 8, &psid);
|
|
ok(!ret, "RtlAllocateAndInitializeSid error %08x\n", ret);
|
|
ret = pRtlFreeSid(psid);
|
|
ok(!ret, "RtlFreeSid error %08x\n", ret);
|
|
|
|
/* these tests crash on XP */
|
|
if (0)
|
|
{
|
|
pRtlAllocateAndInitializeSid(NULL, 0, 1, 2, 3, 4, 5, 6, 7, 8, &psid);
|
|
pRtlAllocateAndInitializeSid(&sia, 0, 1, 2, 3, 4, 5, 6, 7, 8, NULL);
|
|
}
|
|
|
|
ret = pRtlAllocateAndInitializeSid(&sia, 9, 1, 2, 3, 4, 5, 6, 7, 8, &psid);
|
|
ok(ret == STATUS_INVALID_SID, "wrong error %08x\n", ret);
|
|
}
|
|
|
|
static void test_RtlDeleteTimer(void)
|
|
{
|
|
NTSTATUS ret;
|
|
|
|
if (!pRtlDeleteTimer)
|
|
{
|
|
win_skip("RtlDeleteTimer is not available\n");
|
|
return;
|
|
}
|
|
|
|
ret = pRtlDeleteTimer(NULL, NULL, NULL);
|
|
ok(ret == STATUS_INVALID_PARAMETER_1 ||
|
|
ret == STATUS_INVALID_PARAMETER, /* W2K */
|
|
"expected STATUS_INVALID_PARAMETER_1 or STATUS_INVALID_PARAMETER, got %x\n", ret);
|
|
}
|
|
|
|
static void test_RtlThreadErrorMode(void)
|
|
{
|
|
DWORD oldmode;
|
|
BOOL is_wow64;
|
|
DWORD mode;
|
|
NTSTATUS status;
|
|
|
|
if (!pRtlGetThreadErrorMode || !pRtlSetThreadErrorMode)
|
|
{
|
|
win_skip("RtlGetThreadErrorMode and/or RtlSetThreadErrorMode not available\n");
|
|
return;
|
|
}
|
|
|
|
if (!pIsWow64Process || !pIsWow64Process(GetCurrentProcess(), &is_wow64))
|
|
is_wow64 = FALSE;
|
|
|
|
oldmode = pRtlGetThreadErrorMode();
|
|
|
|
status = pRtlSetThreadErrorMode(0x70, &mode);
|
|
ok(status == STATUS_SUCCESS ||
|
|
status == STATUS_WAIT_1, /* Vista */
|
|
"RtlSetThreadErrorMode failed with error 0x%08x\n", status);
|
|
ok(mode == oldmode,
|
|
"RtlSetThreadErrorMode returned mode 0x%x, expected 0x%x\n",
|
|
mode, oldmode);
|
|
ok(pRtlGetThreadErrorMode() == 0x70,
|
|
"RtlGetThreadErrorMode returned 0x%x, expected 0x%x\n", mode, 0x70);
|
|
if (!is_wow64)
|
|
{
|
|
ok(NtCurrentTeb()->HardErrorDisabled == 0x70,
|
|
"The TEB contains 0x%x, expected 0x%x\n",
|
|
NtCurrentTeb()->HardErrorDisabled, 0x70);
|
|
}
|
|
|
|
status = pRtlSetThreadErrorMode(0, &mode);
|
|
ok(status == STATUS_SUCCESS ||
|
|
status == STATUS_WAIT_1, /* Vista */
|
|
"RtlSetThreadErrorMode failed with error 0x%08x\n", status);
|
|
ok(mode == 0x70,
|
|
"RtlSetThreadErrorMode returned mode 0x%x, expected 0x%x\n",
|
|
mode, 0x70);
|
|
ok(pRtlGetThreadErrorMode() == 0,
|
|
"RtlGetThreadErrorMode returned 0x%x, expected 0x%x\n", mode, 0);
|
|
if (!is_wow64)
|
|
{
|
|
ok(NtCurrentTeb()->HardErrorDisabled == 0,
|
|
"The TEB contains 0x%x, expected 0x%x\n",
|
|
NtCurrentTeb()->HardErrorDisabled, 0);
|
|
}
|
|
|
|
for (mode = 1; mode; mode <<= 1)
|
|
{
|
|
status = pRtlSetThreadErrorMode(mode, NULL);
|
|
if (mode & 0x70)
|
|
ok(status == STATUS_SUCCESS ||
|
|
status == STATUS_WAIT_1, /* Vista */
|
|
"RtlSetThreadErrorMode(%x,NULL) failed with error 0x%08x\n",
|
|
mode, status);
|
|
else
|
|
ok(status == STATUS_INVALID_PARAMETER_1,
|
|
"RtlSetThreadErrorMode(%x,NULL) returns 0x%08x, "
|
|
"expected STATUS_INVALID_PARAMETER_1\n",
|
|
mode, status);
|
|
}
|
|
|
|
pRtlSetThreadErrorMode(oldmode, NULL);
|
|
}
|
|
|
|
static void test_LdrProcessRelocationBlock(void)
|
|
{
|
|
IMAGE_BASE_RELOCATION *ret;
|
|
USHORT reloc;
|
|
DWORD addr32;
|
|
SHORT addr16;
|
|
|
|
if(!pLdrProcessRelocationBlock) {
|
|
win_skip("LdrProcessRelocationBlock not available\n");
|
|
return;
|
|
}
|
|
|
|
addr32 = 0x50005;
|
|
reloc = IMAGE_REL_BASED_HIGHLOW<<12;
|
|
ret = pLdrProcessRelocationBlock(&addr32, 1, &reloc, 0x500050);
|
|
ok((USHORT*)ret == &reloc+1, "ret = %p, expected %p\n", ret, &reloc+1);
|
|
ok(addr32 == 0x550055, "addr32 = %x, expected 0x550055\n", addr32);
|
|
|
|
addr16 = 0x505;
|
|
reloc = IMAGE_REL_BASED_HIGH<<12;
|
|
ret = pLdrProcessRelocationBlock(&addr16, 1, &reloc, 0x500060);
|
|
ok((USHORT*)ret == &reloc+1, "ret = %p, expected %p\n", ret, &reloc+1);
|
|
ok(addr16 == 0x555, "addr16 = %x, expected 0x555\n", addr16);
|
|
|
|
addr16 = 0x505;
|
|
reloc = IMAGE_REL_BASED_LOW<<12;
|
|
ret = pLdrProcessRelocationBlock(&addr16, 1, &reloc, 0x500060);
|
|
ok((USHORT*)ret == &reloc+1, "ret = %p, expected %p\n", ret, &reloc+1);
|
|
ok(addr16 == 0x565, "addr16 = %x, expected 0x565\n", addr16);
|
|
}
|
|
|
|
static void test_RtlIpv4AddressToString(void)
|
|
{
|
|
CHAR buffer[20];
|
|
CHAR *res;
|
|
IN_ADDR ip;
|
|
DWORD_PTR len;
|
|
|
|
if (!pRtlIpv4AddressToStringA)
|
|
{
|
|
win_skip("RtlIpv4AddressToStringA not available\n");
|
|
return;
|
|
}
|
|
|
|
ip.S_un.S_un_b.s_b1 = 1;
|
|
ip.S_un.S_un_b.s_b2 = 2;
|
|
ip.S_un.S_un_b.s_b3 = 3;
|
|
ip.S_un.S_un_b.s_b4 = 4;
|
|
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringA(&ip, buffer);
|
|
len = strlen(buffer);
|
|
ok(res == (buffer + len), "got %p with '%s' (expected %p)\n", res, buffer, buffer + len);
|
|
|
|
res = pRtlIpv4AddressToStringA(&ip, NULL);
|
|
ok( (res == (char *)~0) ||
|
|
broken(res == (char *)len), /* XP and w2003 */
|
|
"got %p (expected ~0)\n", res);
|
|
|
|
if (0) {
|
|
/* this crashes in windows */
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringA(NULL, buffer);
|
|
trace("got %p with '%s'\n", res, buffer);
|
|
}
|
|
|
|
if (0) {
|
|
/* this crashes in windows */
|
|
res = pRtlIpv4AddressToStringA(NULL, NULL);
|
|
trace("got %p\n", res);
|
|
}
|
|
}
|
|
|
|
static void test_RtlIpv4AddressToStringEx(void)
|
|
{
|
|
CHAR ip_1234[] = "1.2.3.4";
|
|
CHAR ip_1234_80[] = "1.2.3.4:80";
|
|
LPSTR expect;
|
|
CHAR buffer[30];
|
|
NTSTATUS res;
|
|
IN_ADDR ip;
|
|
ULONG size;
|
|
DWORD used;
|
|
USHORT port;
|
|
|
|
if (!pRtlIpv4AddressToStringExA)
|
|
{
|
|
win_skip("RtlIpv4AddressToStringExA not available\n");
|
|
return;
|
|
}
|
|
|
|
ip.S_un.S_un_b.s_b1 = 1;
|
|
ip.S_un.S_un_b.s_b2 = 2;
|
|
ip.S_un.S_un_b.s_b3 = 3;
|
|
ip.S_un.S_un_b.s_b4 = 4;
|
|
|
|
port = htons(80);
|
|
expect = ip_1234_80;
|
|
|
|
size = sizeof(buffer);
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(&ip, port, buffer, &size);
|
|
used = strlen(buffer);
|
|
ok( (res == STATUS_SUCCESS) &&
|
|
(size == strlen(expect) + 1) && !strcmp(buffer, expect),
|
|
"got 0x%x and size %d with '%s'\n", res, size, buffer);
|
|
|
|
size = used + 1;
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(&ip, port, buffer, &size);
|
|
ok( (res == STATUS_SUCCESS) &&
|
|
(size == strlen(expect) + 1) && !strcmp(buffer, expect),
|
|
"got 0x%x and size %d with '%s'\n", res, size, buffer);
|
|
|
|
size = used;
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(&ip, port, buffer, &size);
|
|
ok( (res == STATUS_INVALID_PARAMETER) && (size == used + 1),
|
|
"got 0x%x and %d with '%s' (expected STATUS_INVALID_PARAMETER and %d)\n",
|
|
res, size, buffer, used + 1);
|
|
|
|
size = used - 1;
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(&ip, port, buffer, &size);
|
|
ok( (res == STATUS_INVALID_PARAMETER) && (size == used + 1),
|
|
"got 0x%x and %d with '%s' (expected STATUS_INVALID_PARAMETER and %d)\n",
|
|
res, size, buffer, used + 1);
|
|
|
|
|
|
/* to get only the ip, use 0 as port */
|
|
port = 0;
|
|
expect = ip_1234;
|
|
|
|
size = sizeof(buffer);
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(&ip, port, buffer, &size);
|
|
used = strlen(buffer);
|
|
ok( (res == STATUS_SUCCESS) &&
|
|
(size == strlen(expect) + 1) && !strcmp(buffer, expect),
|
|
"got 0x%x and size %d with '%s'\n", res, size, buffer);
|
|
|
|
size = used + 1;
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(&ip, port, buffer, &size);
|
|
ok( (res == STATUS_SUCCESS) &&
|
|
(size == strlen(expect) + 1) && !strcmp(buffer, expect),
|
|
"got 0x%x and size %d with '%s'\n", res, size, buffer);
|
|
|
|
size = used;
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(&ip, port, buffer, &size);
|
|
ok( (res == STATUS_INVALID_PARAMETER) && (size == used + 1),
|
|
"got 0x%x and %d with '%s' (expected STATUS_INVALID_PARAMETER and %d)\n",
|
|
res, size, buffer, used + 1);
|
|
|
|
size = used - 1;
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(&ip, port, buffer, &size);
|
|
ok( (res == STATUS_INVALID_PARAMETER) && (size == used + 1),
|
|
"got 0x%x and %d with '%s' (expected STATUS_INVALID_PARAMETER and %d)\n",
|
|
res, size, buffer, used + 1);
|
|
|
|
|
|
/* parameters are checked */
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(&ip, 0, buffer, NULL);
|
|
ok(res == STATUS_INVALID_PARAMETER,
|
|
"got 0x%x with '%s' (expected STATUS_INVALID_PARAMETER)\n", res, buffer);
|
|
|
|
size = sizeof(buffer);
|
|
res = pRtlIpv4AddressToStringExA(&ip, 0, NULL, &size);
|
|
ok( res == STATUS_INVALID_PARAMETER,
|
|
"got 0x%x and size %d (expected STATUS_INVALID_PARAMETER)\n", res, size);
|
|
|
|
size = sizeof(buffer);
|
|
memset(buffer, '#', sizeof(buffer) - 1);
|
|
buffer[sizeof(buffer) -1] = 0;
|
|
res = pRtlIpv4AddressToStringExA(NULL, 0, buffer, &size);
|
|
ok( res == STATUS_INVALID_PARAMETER,
|
|
"got 0x%x and size %d with '%s' (expected STATUS_INVALID_PARAMETER)\n",
|
|
res, size, buffer);
|
|
}
|
|
|
|
static void test_RtlIpv4StringToAddress(void)
|
|
{
|
|
NTSTATUS res;
|
|
IN_ADDR ip, expected_ip;
|
|
PCSTR terminator;
|
|
CHAR dummy;
|
|
struct
|
|
{
|
|
PCSTR address;
|
|
NTSTATUS res;
|
|
int terminator_offset;
|
|
int ip[4];
|
|
BOOL strict_is_different;
|
|
NTSTATUS res_strict;
|
|
int terminator_offset_strict;
|
|
int ip_strict[4];
|
|
} tests[] =
|
|
{
|
|
{ "", STATUS_INVALID_PARAMETER, 0, { -1 } },
|
|
{ " ", STATUS_INVALID_PARAMETER, 0, { -1 } },
|
|
{ "1.1.1.1", STATUS_SUCCESS, 7, { 1, 1, 1, 1 } },
|
|
{ "0.0.0.0", STATUS_SUCCESS, 7, { 0, 0, 0, 0 } },
|
|
{ "255.255.255.255", STATUS_SUCCESS, 15, { 255, 255, 255, 255 } },
|
|
{ "255.255.255.255:123",
|
|
STATUS_SUCCESS, 15, { 255, 255, 255, 255 } },
|
|
{ "255.255.255.256", STATUS_INVALID_PARAMETER, 15, { -1 } },
|
|
{ "255.255.255.4294967295",
|
|
STATUS_INVALID_PARAMETER, 22, { -1 } },
|
|
{ "255.255.255.4294967296",
|
|
STATUS_INVALID_PARAMETER, 21, { -1 } },
|
|
{ "255.255.255.4294967297",
|
|
STATUS_INVALID_PARAMETER, 21, { -1 } },
|
|
{ "a", STATUS_INVALID_PARAMETER, 0, { -1 } },
|
|
{ "1.1.1.0xaA", STATUS_SUCCESS, 10, { 1, 1, 1, 170 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 8, { -1 } },
|
|
{ "1.1.1.0XaA", STATUS_SUCCESS, 10, { 1, 1, 1, 170 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 8, { -1 } },
|
|
{ "1.1.1.0x", STATUS_INVALID_PARAMETER, 8, { -1 } },
|
|
{ "1.1.1.0xff", STATUS_SUCCESS, 10, { 1, 1, 1, 255 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 8, { -1 } },
|
|
{ "1.1.1.0x100", STATUS_INVALID_PARAMETER, 11, { -1 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 8, { -1 } },
|
|
{ "1.1.1.0xffffffff",STATUS_INVALID_PARAMETER, 16, { -1 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 8, { -1 } },
|
|
{ "1.1.1.0x100000000",
|
|
STATUS_INVALID_PARAMETER, 16, { -1, 0, 0, 0 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 8, { -1 } },
|
|
{ "1.1.1.010", STATUS_SUCCESS, 9, { 1, 1, 1, 8 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 7, { -1 } },
|
|
{ "1.1.1.00", STATUS_SUCCESS, 8, { 1, 1, 1, 0 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 7, { -1 } },
|
|
{ "1.1.1.007", STATUS_SUCCESS, 9, { 1, 1, 1, 7 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 7, { -1 } },
|
|
{ "1.1.1.08", STATUS_INVALID_PARAMETER, 7, { -1 } },
|
|
{ "1.1.1.008", STATUS_SUCCESS, 8, { 1, 1, 1, 0 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 7, { -1 } },
|
|
{ "1.1.1.0a", STATUS_SUCCESS, 7, { 1, 1, 1, 0 } },
|
|
{ "1.1.1.0o10", STATUS_SUCCESS, 7, { 1, 1, 1, 0 } },
|
|
{ "1.1.1.0b10", STATUS_SUCCESS, 7, { 1, 1, 1, 0 } },
|
|
{ "1.1.1.-2", STATUS_INVALID_PARAMETER, 6, { -1 } },
|
|
{ "1", STATUS_SUCCESS, 1, { 0, 0, 0, 1 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 1, { -1 } },
|
|
{ "-1", STATUS_INVALID_PARAMETER, 0, { -1 } },
|
|
{ "203569230", STATUS_SUCCESS, 9, { 12, 34, 56, 78 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 9, { -1 } },
|
|
{ "1.223756", STATUS_SUCCESS, 8, { 1, 3, 106, 12 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 8, { -1 } },
|
|
{ "3.4.756", STATUS_SUCCESS, 7, { 3, 4, 2, 244 },
|
|
TRUE, STATUS_INVALID_PARAMETER, 7, { -1 } },
|
|
{ "3.4.756.1", STATUS_INVALID_PARAMETER, 9, { -1 } },
|
|
{ "3.4.65536", STATUS_INVALID_PARAMETER, 9, { -1 } },
|
|
{ "3.4.5.6.7", STATUS_INVALID_PARAMETER, 7, { -1 } },
|
|
{ "3.4.5.+6", STATUS_INVALID_PARAMETER, 6, { -1 } },
|
|
{ " 3.4.5.6", STATUS_INVALID_PARAMETER, 0, { -1 } },
|
|
{ "\t3.4.5.6", STATUS_INVALID_PARAMETER, 0, { -1 } },
|
|
{ "3.4.5.6 ", STATUS_SUCCESS, 7, { 3, 4, 5, 6 } },
|
|
{ "3. 4.5.6", STATUS_INVALID_PARAMETER, 2, { -1 } },
|
|
{ ".", STATUS_INVALID_PARAMETER, 1, { -1 } },
|
|
{ "..", STATUS_INVALID_PARAMETER, 1, { -1 } },
|
|
{ "1.", STATUS_INVALID_PARAMETER, 2, { -1 } },
|
|
{ "1..", STATUS_INVALID_PARAMETER, 3, { -1 } },
|
|
{ ".1", STATUS_INVALID_PARAMETER, 1, { -1 } },
|
|
{ ".1.", STATUS_INVALID_PARAMETER, 1, { -1 } },
|
|
{ ".1.2.3", STATUS_INVALID_PARAMETER, 1, { -1 } },
|
|
{ "0.1.2.3", STATUS_SUCCESS, 7, { 0, 1, 2, 3 } },
|
|
{ "0.1.2.3.", STATUS_INVALID_PARAMETER, 7, { -1 } },
|
|
{ "[0.1.2.3]", STATUS_INVALID_PARAMETER, 0, { -1 } },
|
|
{ "::1", STATUS_INVALID_PARAMETER, 0, { -1 } },
|
|
{ ":1", STATUS_INVALID_PARAMETER, 0, { -1 } },
|
|
};
|
|
const int testcount = sizeof(tests) / sizeof(tests[0]);
|
|
int i;
|
|
|
|
if (!pRtlIpv4StringToAddressA)
|
|
{
|
|
skip("RtlIpv4StringToAddress not available\n");
|
|
return;
|
|
}
|
|
|
|
if (0)
|
|
{
|
|
/* leaving either parameter NULL crashes on Windows */
|
|
res = pRtlIpv4StringToAddressA(NULL, FALSE, &terminator, &ip);
|
|
res = pRtlIpv4StringToAddressA("1.1.1.1", FALSE, NULL, &ip);
|
|
res = pRtlIpv4StringToAddressA("1.1.1.1", FALSE, &terminator, NULL);
|
|
/* same for the wide char version */
|
|
/*
|
|
res = pRtlIpv4StringToAddressW(NULL, FALSE, &terminatorW, &ip);
|
|
res = pRtlIpv4StringToAddressW(L"1.1.1.1", FALSE, NULL, &ip);
|
|
res = pRtlIpv4StringToAddressW(L"1.1.1.1", FALSE, &terminatorW, NULL);
|
|
*/
|
|
}
|
|
|
|
for (i = 0; i < testcount; i++)
|
|
{
|
|
/* non-strict */
|
|
terminator = &dummy;
|
|
ip.S_un.S_addr = 0xabababab;
|
|
res = pRtlIpv4StringToAddressA(tests[i].address, FALSE, &terminator, &ip);
|
|
ok(res == tests[i].res,
|
|
"[%s] res = 0x%08x, expected 0x%08x\n",
|
|
tests[i].address, res, tests[i].res);
|
|
ok(terminator == tests[i].address + tests[i].terminator_offset,
|
|
"[%s] terminator = %p, expected %p\n",
|
|
tests[i].address, terminator, tests[i].address + tests[i].terminator_offset);
|
|
if (tests[i].ip[0] == -1)
|
|
expected_ip.S_un.S_addr = 0xabababab;
|
|
else
|
|
{
|
|
expected_ip.S_un.S_un_b.s_b1 = tests[i].ip[0];
|
|
expected_ip.S_un.S_un_b.s_b2 = tests[i].ip[1];
|
|
expected_ip.S_un.S_un_b.s_b3 = tests[i].ip[2];
|
|
expected_ip.S_un.S_un_b.s_b4 = tests[i].ip[3];
|
|
}
|
|
ok(ip.S_un.S_addr == expected_ip.S_un.S_addr,
|
|
"[%s] ip = %08x, expected %08x\n",
|
|
tests[i].address, ip.S_un.S_addr, expected_ip.S_un.S_addr);
|
|
|
|
if (!tests[i].strict_is_different)
|
|
{
|
|
tests[i].res_strict = tests[i].res;
|
|
tests[i].terminator_offset_strict = tests[i].terminator_offset;
|
|
tests[i].ip_strict[0] = tests[i].ip[0];
|
|
tests[i].ip_strict[1] = tests[i].ip[1];
|
|
tests[i].ip_strict[2] = tests[i].ip[2];
|
|
tests[i].ip_strict[3] = tests[i].ip[3];
|
|
}
|
|
/* strict */
|
|
terminator = &dummy;
|
|
ip.S_un.S_addr = 0xabababab;
|
|
res = pRtlIpv4StringToAddressA(tests[i].address, TRUE, &terminator, &ip);
|
|
ok(res == tests[i].res_strict,
|
|
"[%s] res = 0x%08x, expected 0x%08x\n",
|
|
tests[i].address, res, tests[i].res_strict);
|
|
ok(terminator == tests[i].address + tests[i].terminator_offset_strict,
|
|
"[%s] terminator = %p, expected %p\n",
|
|
tests[i].address, terminator, tests[i].address + tests[i].terminator_offset_strict);
|
|
if (tests[i].ip_strict[0] == -1)
|
|
expected_ip.S_un.S_addr = 0xabababab;
|
|
else
|
|
{
|
|
expected_ip.S_un.S_un_b.s_b1 = tests[i].ip_strict[0];
|
|
expected_ip.S_un.S_un_b.s_b2 = tests[i].ip_strict[1];
|
|
expected_ip.S_un.S_un_b.s_b3 = tests[i].ip_strict[2];
|
|
expected_ip.S_un.S_un_b.s_b4 = tests[i].ip_strict[3];
|
|
}
|
|
ok(ip.S_un.S_addr == expected_ip.S_un.S_addr,
|
|
"[%s] ip = %08x, expected %08x\n",
|
|
tests[i].address, ip.S_un.S_addr, expected_ip.S_un.S_addr);
|
|
}
|
|
}
|
|
|
|
static void test_LdrAddRefDll(void)
|
|
{
|
|
HMODULE mod, mod2;
|
|
NTSTATUS status;
|
|
BOOL ret;
|
|
|
|
if (!pLdrAddRefDll)
|
|
{
|
|
win_skip( "LdrAddRefDll not supported\n" );
|
|
return;
|
|
}
|
|
|
|
mod = LoadLibraryA("comctl32.dll");
|
|
ok(mod != NULL, "got %p\n", mod);
|
|
ret = FreeLibrary(mod);
|
|
ok(ret, "got %d\n", ret);
|
|
|
|
mod2 = GetModuleHandleA("comctl32.dll");
|
|
ok(mod2 == NULL, "got %p\n", mod2);
|
|
|
|
/* load, addref and release 2 times */
|
|
mod = LoadLibraryA("comctl32.dll");
|
|
ok(mod != NULL, "got %p\n", mod);
|
|
status = pLdrAddRefDll(0, mod);
|
|
ok(status == STATUS_SUCCESS, "got 0x%08x\n", status);
|
|
ret = FreeLibrary(mod);
|
|
ok(ret, "got %d\n", ret);
|
|
|
|
mod2 = GetModuleHandleA("comctl32.dll");
|
|
ok(mod2 != NULL, "got %p\n", mod2);
|
|
ret = FreeLibrary(mod);
|
|
ok(ret, "got %d\n", ret);
|
|
|
|
mod2 = GetModuleHandleA("comctl32.dll");
|
|
ok(mod2 == NULL, "got %p\n", mod2);
|
|
|
|
/* pin refcount */
|
|
mod = LoadLibraryA("comctl32.dll");
|
|
ok(mod != NULL, "got %p\n", mod);
|
|
status = pLdrAddRefDll(LDR_ADDREF_DLL_PIN, mod);
|
|
ok(status == STATUS_SUCCESS, "got 0x%08x\n", status);
|
|
|
|
ret = FreeLibrary(mod);
|
|
ok(ret, "got %d\n", ret);
|
|
ret = FreeLibrary(mod);
|
|
ok(ret, "got %d\n", ret);
|
|
ret = FreeLibrary(mod);
|
|
ok(ret, "got %d\n", ret);
|
|
ret = FreeLibrary(mod);
|
|
ok(ret, "got %d\n", ret);
|
|
|
|
mod2 = GetModuleHandleA("comctl32.dll");
|
|
ok(mod2 != NULL, "got %p\n", mod2);
|
|
}
|
|
|
|
static void test_LdrLockLoaderLock(void)
|
|
{
|
|
ULONG_PTR magic;
|
|
ULONG result;
|
|
NTSTATUS status;
|
|
|
|
if (!pLdrLockLoaderLock)
|
|
{
|
|
win_skip("LdrLockLoaderLock() is not available\n");
|
|
return;
|
|
}
|
|
|
|
/* invalid flags */
|
|
result = 10;
|
|
magic = 0xdeadbeef;
|
|
status = pLdrLockLoaderLock(0x10, &result, &magic);
|
|
ok(status == STATUS_INVALID_PARAMETER_1, "got 0x%08x\n", status);
|
|
ok(result == 0, "got %d\n", result);
|
|
ok(magic == 0, "got %lx\n", magic);
|
|
|
|
magic = 0xdeadbeef;
|
|
status = pLdrLockLoaderLock(0x10, NULL, &magic);
|
|
ok(status == STATUS_INVALID_PARAMETER_1, "got 0x%08x\n", status);
|
|
ok(magic == 0, "got %lx\n", magic);
|
|
|
|
result = 10;
|
|
status = pLdrLockLoaderLock(0x10, &result, NULL);
|
|
ok(status == STATUS_INVALID_PARAMETER_1, "got 0x%08x\n", status);
|
|
ok(result == 0, "got %d\n", result);
|
|
|
|
/* non-blocking mode, result is null */
|
|
magic = 0xdeadbeef;
|
|
status = pLdrLockLoaderLock(0x2, NULL, &magic);
|
|
ok(status == STATUS_INVALID_PARAMETER_2, "got 0x%08x\n", status);
|
|
ok(magic == 0, "got %lx\n", magic);
|
|
|
|
/* magic pointer is null */
|
|
result = 10;
|
|
status = pLdrLockLoaderLock(0, &result, NULL);
|
|
ok(status == STATUS_INVALID_PARAMETER_3, "got 0x%08x\n", status);
|
|
ok(result == 0, "got %d\n", result);
|
|
|
|
/* lock in non-blocking mode */
|
|
result = 0;
|
|
magic = 0;
|
|
status = pLdrLockLoaderLock(0x2, &result, &magic);
|
|
ok(status == STATUS_SUCCESS, "got 0x%08x\n", status);
|
|
ok(result == 1, "got %d\n", result);
|
|
ok(magic != 0, "got %lx\n", magic);
|
|
pLdrUnlockLoaderLock(0, magic);
|
|
}
|
|
|
|
static void test_RtlCompressBuffer(void)
|
|
{
|
|
ULONG compress_workspace, decompress_workspace;
|
|
static const UCHAR test_buffer[] = "WineWineWine";
|
|
static UCHAR buf1[0x1000], buf2[0x1000];
|
|
ULONG final_size, buf_size;
|
|
UCHAR *workspace = NULL;
|
|
NTSTATUS status;
|
|
|
|
if (!pRtlCompressBuffer || !pRtlDecompressBuffer || !pRtlGetCompressionWorkSpaceSize)
|
|
{
|
|
win_skip("skipping RtlCompressBuffer tests, required functions not available\n");
|
|
return;
|
|
}
|
|
|
|
compress_workspace = decompress_workspace = 0xdeadbeef;
|
|
status = pRtlGetCompressionWorkSpaceSize(COMPRESSION_FORMAT_LZNT1, &compress_workspace,
|
|
&decompress_workspace);
|
|
ok(status == STATUS_SUCCESS, "got wrong status 0x%08x\n", status);
|
|
ok(compress_workspace != 0, "got wrong compress_workspace %u\n", compress_workspace);
|
|
workspace = HeapAlloc(GetProcessHeap(), 0, compress_workspace);
|
|
ok(workspace != NULL, "HeapAlloc failed %d\n", GetLastError());
|
|
|
|
/* test compression format / engine */
|
|
final_size = 0xdeadbeef;
|
|
status = pRtlCompressBuffer(COMPRESSION_FORMAT_NONE, test_buffer, sizeof(test_buffer),
|
|
buf1, sizeof(buf1) - 1, 4096, &final_size, workspace);
|
|
ok(status == STATUS_INVALID_PARAMETER, "got wrong status 0x%08x\n", status);
|
|
ok(final_size == 0xdeadbeef, "got wrong final_size %u\n", final_size);
|
|
|
|
final_size = 0xdeadbeef;
|
|
status = pRtlCompressBuffer(COMPRESSION_FORMAT_DEFAULT, test_buffer, sizeof(test_buffer),
|
|
buf1, sizeof(buf1) - 1, 4096, &final_size, workspace);
|
|
ok(status == STATUS_INVALID_PARAMETER, "got wrong status 0x%08x\n", status);
|
|
ok(final_size == 0xdeadbeef, "got wrong final_size %u\n", final_size);
|
|
|
|
final_size = 0xdeadbeef;
|
|
status = pRtlCompressBuffer(0xFF, test_buffer, sizeof(test_buffer),
|
|
buf1, sizeof(buf1) - 1, 4096, &final_size, workspace);
|
|
ok(status == STATUS_UNSUPPORTED_COMPRESSION, "got wrong status 0x%08x\n", status);
|
|
ok(final_size == 0xdeadbeef, "got wrong final_size %u\n", final_size);
|
|
|
|
/* test compression */
|
|
final_size = 0xdeadbeef;
|
|
memset(buf1, 0x11, sizeof(buf1));
|
|
status = pRtlCompressBuffer(COMPRESSION_FORMAT_LZNT1, test_buffer, sizeof(test_buffer),
|
|
buf1, sizeof(buf1), 4096, &final_size, workspace);
|
|
ok(status == STATUS_SUCCESS, "got wrong status 0x%08x\n", status);
|
|
ok((*(WORD *)buf1 & 0x7000) == 0x3000, "no chunk signature found %04x\n", *(WORD *)buf1);
|
|
todo_wine
|
|
ok(final_size < sizeof(test_buffer), "got wrong final_size %u\n", final_size);
|
|
|
|
/* test decompression */
|
|
buf_size = final_size;
|
|
final_size = 0xdeadbeef;
|
|
memset(buf2, 0x11, sizeof(buf2));
|
|
status = pRtlDecompressBuffer(COMPRESSION_FORMAT_LZNT1, buf2, sizeof(buf2),
|
|
buf1, buf_size, &final_size);
|
|
ok(status == STATUS_SUCCESS, "got wrong status 0x%08x\n", status);
|
|
ok(final_size == sizeof(test_buffer), "got wrong final_size %u\n", final_size);
|
|
ok(!memcmp(buf2, test_buffer, sizeof(test_buffer)), "got wrong decoded data\n");
|
|
ok(buf2[sizeof(test_buffer)] == 0x11, "too many bytes written\n");
|
|
|
|
/* buffer too small */
|
|
final_size = 0xdeadbeef;
|
|
memset(buf1, 0x11, sizeof(buf1));
|
|
status = pRtlCompressBuffer(COMPRESSION_FORMAT_LZNT1, test_buffer, sizeof(test_buffer),
|
|
buf1, 4, 4096, &final_size, workspace);
|
|
ok(status == STATUS_BUFFER_TOO_SMALL, "got wrong status 0x%08x\n", status);
|
|
|
|
HeapFree(GetProcessHeap(), 0, workspace);
|
|
}
|
|
|
|
static void test_RtlGetCompressionWorkSpaceSize(void)
|
|
{
|
|
ULONG compress_workspace, decompress_workspace;
|
|
NTSTATUS status;
|
|
|
|
if (!pRtlGetCompressionWorkSpaceSize)
|
|
{
|
|
win_skip("RtlGetCompressionWorkSpaceSize is not available\n");
|
|
return;
|
|
}
|
|
|
|
/* test invalid format / engine */
|
|
status = pRtlGetCompressionWorkSpaceSize(COMPRESSION_FORMAT_NONE, &compress_workspace,
|
|
&decompress_workspace);
|
|
ok(status == STATUS_INVALID_PARAMETER, "got wrong status 0x%08x\n", status);
|
|
|
|
status = pRtlGetCompressionWorkSpaceSize(COMPRESSION_FORMAT_DEFAULT, &compress_workspace,
|
|
&decompress_workspace);
|
|
ok(status == STATUS_INVALID_PARAMETER, "got wrong status 0x%08x\n", status);
|
|
|
|
status = pRtlGetCompressionWorkSpaceSize(0xFF, &compress_workspace, &decompress_workspace);
|
|
ok(status == STATUS_UNSUPPORTED_COMPRESSION, "got wrong status 0x%08x\n", status);
|
|
|
|
/* test LZNT1 with normal and maximum compression */
|
|
compress_workspace = decompress_workspace = 0xdeadbeef;
|
|
status = pRtlGetCompressionWorkSpaceSize(COMPRESSION_FORMAT_LZNT1, &compress_workspace,
|
|
&decompress_workspace);
|
|
ok(status == STATUS_SUCCESS, "got wrong status 0x%08x\n", status);
|
|
ok(compress_workspace != 0, "got wrong compress_workspace %u\n", compress_workspace);
|
|
ok(decompress_workspace == 0x1000, "got wrong decompress_workspace %u\n", decompress_workspace);
|
|
|
|
compress_workspace = decompress_workspace = 0xdeadbeef;
|
|
status = pRtlGetCompressionWorkSpaceSize(COMPRESSION_FORMAT_LZNT1 | COMPRESSION_ENGINE_MAXIMUM,
|
|
&compress_workspace, &decompress_workspace);
|
|
ok(status == STATUS_SUCCESS, "got wrong status 0x%08x\n", status);
|
|
ok(compress_workspace != 0, "got wrong compress_workspace %u\n", compress_workspace);
|
|
ok(decompress_workspace == 0x1000, "got wrong decompress_workspace %u\n", decompress_workspace);
|
|
}
|
|
|
|
/* helper for test_RtlDecompressBuffer, checks if a chunk is incomplete */
|
|
static BOOL is_incomplete_chunk(const UCHAR *compressed, ULONG compressed_size, BOOL check_all)
|
|
{
|
|
ULONG chunk_size;
|
|
|
|
if (compressed_size <= sizeof(WORD))
|
|
return TRUE;
|
|
|
|
while (compressed_size >= sizeof(WORD))
|
|
{
|
|
chunk_size = (*(WORD *)compressed & 0xFFF) + 1;
|
|
if (compressed_size < sizeof(WORD) + chunk_size)
|
|
return TRUE;
|
|
if (!check_all)
|
|
break;
|
|
compressed += sizeof(WORD) + chunk_size;
|
|
compressed_size -= sizeof(WORD) + chunk_size;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
#define DECOMPRESS_BROKEN_FRAGMENT 1 /* < Win 7 */
|
|
#define DECOMPRESS_BROKEN_TRUNCATED 2 /* broken on all machines */
|
|
|
|
static void test_RtlDecompressBuffer(void)
|
|
{
|
|
static const struct
|
|
{
|
|
UCHAR compressed[32];
|
|
ULONG compressed_size;
|
|
NTSTATUS status;
|
|
UCHAR uncompressed[32];
|
|
ULONG uncompressed_size;
|
|
DWORD broken_flags;
|
|
}
|
|
test_lznt[] =
|
|
{
|
|
/* 4 byte uncompressed chunk */
|
|
{
|
|
{0x03, 0x30, 'W', 'i', 'n', 'e'},
|
|
6,
|
|
STATUS_SUCCESS,
|
|
"Wine",
|
|
4,
|
|
DECOMPRESS_BROKEN_FRAGMENT
|
|
},
|
|
/* 8 byte uncompressed chunk */
|
|
{
|
|
{0x07, 0x30, 'W', 'i', 'n', 'e', 'W', 'i', 'n', 'e'},
|
|
10,
|
|
STATUS_SUCCESS,
|
|
"WineWine",
|
|
8,
|
|
DECOMPRESS_BROKEN_FRAGMENT
|
|
},
|
|
/* 4 byte compressed chunk */
|
|
{
|
|
{0x04, 0xB0, 0x00, 'W', 'i', 'n', 'e'},
|
|
7,
|
|
STATUS_SUCCESS,
|
|
"Wine",
|
|
4
|
|
},
|
|
/* 8 byte compressed chunk */
|
|
{
|
|
{0x08, 0xB0, 0x00, 'W', 'i', 'n', 'e', 'W', 'i', 'n', 'e'},
|
|
11,
|
|
STATUS_SUCCESS,
|
|
"WineWine",
|
|
8
|
|
},
|
|
/* compressed chunk using backwards reference */
|
|
{
|
|
{0x06, 0xB0, 0x10, 'W', 'i', 'n', 'e', 0x01, 0x30},
|
|
9,
|
|
STATUS_SUCCESS,
|
|
"WineWine",
|
|
8,
|
|
DECOMPRESS_BROKEN_TRUNCATED
|
|
},
|
|
/* compressed chunk using backwards reference with length > bytes_read */
|
|
{
|
|
{0x06, 0xB0, 0x10, 'W', 'i', 'n', 'e', 0x05, 0x30},
|
|
9,
|
|
STATUS_SUCCESS,
|
|
"WineWineWine",
|
|
12,
|
|
DECOMPRESS_BROKEN_TRUNCATED
|
|
},
|
|
/* same as above, but unused bits != 0 */
|
|
{
|
|
{0x06, 0xB0, 0x30, 'W', 'i', 'n', 'e', 0x01, 0x30},
|
|
9,
|
|
STATUS_SUCCESS,
|
|
"WineWine",
|
|
8,
|
|
DECOMPRESS_BROKEN_TRUNCATED
|
|
},
|
|
/* compressed chunk without backwards reference and unused bits != 0 */
|
|
{
|
|
{0x01, 0xB0, 0x02, 'W'},
|
|
4,
|
|
STATUS_SUCCESS,
|
|
"W",
|
|
1
|
|
},
|
|
/* termination sequence after first chunk */
|
|
{
|
|
{0x03, 0x30, 'W', 'i', 'n', 'e', 0x00, 0x00, 0x03, 0x30, 'W', 'i', 'n', 'e'},
|
|
14,
|
|
STATUS_SUCCESS,
|
|
"Wine",
|
|
4,
|
|
DECOMPRESS_BROKEN_FRAGMENT
|
|
},
|
|
/* compressed chunk using backwards reference with 4 bit offset, 12 bit length */
|
|
{
|
|
{0x14, 0xB0, 0x00, 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H',
|
|
0x00, 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P',
|
|
0x01, 0x01, 0xF0},
|
|
23,
|
|
STATUS_SUCCESS,
|
|
"ABCDEFGHIJKLMNOPABCD",
|
|
20,
|
|
DECOMPRESS_BROKEN_TRUNCATED
|
|
},
|
|
/* compressed chunk using backwards reference with 5 bit offset, 11 bit length */
|
|
{
|
|
{0x15, 0xB0, 0x00, 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H',
|
|
0x00, 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P',
|
|
0x02, 'A', 0x00, 0x78},
|
|
24,
|
|
STATUS_SUCCESS,
|
|
"ABCDEFGHIJKLMNOPABCD",
|
|
20,
|
|
DECOMPRESS_BROKEN_TRUNCATED
|
|
},
|
|
/* uncompressed chunk with invalid magic */
|
|
{
|
|
{0x03, 0x20, 'W', 'i', 'n', 'e'},
|
|
6,
|
|
STATUS_SUCCESS,
|
|
"Wine",
|
|
4,
|
|
DECOMPRESS_BROKEN_FRAGMENT
|
|
},
|
|
/* compressed chunk with invalid magic */
|
|
{
|
|
{0x04, 0xA0, 0x00, 'W', 'i', 'n', 'e'},
|
|
7,
|
|
STATUS_SUCCESS,
|
|
"Wine",
|
|
4
|
|
},
|
|
/* garbage byte after end of buffer */
|
|
{
|
|
{0x00, 0xB0, 0x02, 0x01},
|
|
4,
|
|
STATUS_SUCCESS,
|
|
"",
|
|
0
|
|
},
|
|
/* empty compressed chunk */
|
|
{
|
|
{0x00, 0xB0, 0x00},
|
|
3,
|
|
STATUS_SUCCESS,
|
|
"",
|
|
0
|
|
},
|
|
/* empty compressed chunk with unused bits != 0 */
|
|
{
|
|
{0x00, 0xB0, 0x01},
|
|
3,
|
|
STATUS_SUCCESS,
|
|
"",
|
|
0
|
|
},
|
|
/* empty input buffer */
|
|
{
|
|
{},
|
|
0,
|
|
STATUS_BAD_COMPRESSION_BUFFER,
|
|
},
|
|
/* incomplete chunk header */
|
|
{
|
|
{0x01},
|
|
1,
|
|
STATUS_BAD_COMPRESSION_BUFFER
|
|
},
|
|
/* incomplete chunk header */
|
|
{
|
|
{0x00, 0x30},
|
|
2,
|
|
STATUS_BAD_COMPRESSION_BUFFER
|
|
},
|
|
/* compressed chunk with invalid backwards reference */
|
|
{
|
|
{0x06, 0xB0, 0x10, 'W', 'i', 'n', 'e', 0x05, 0x40},
|
|
9,
|
|
STATUS_BAD_COMPRESSION_BUFFER
|
|
},
|
|
/* compressed chunk with incomplete backwards reference */
|
|
{
|
|
{0x05, 0xB0, 0x10, 'W', 'i', 'n', 'e', 0x05},
|
|
8,
|
|
STATUS_BAD_COMPRESSION_BUFFER
|
|
},
|
|
/* incomplete uncompressed chunk */
|
|
{
|
|
{0x07, 0x30, 'W', 'i', 'n', 'e'},
|
|
6,
|
|
STATUS_BAD_COMPRESSION_BUFFER
|
|
},
|
|
/* incomplete compressed chunk */
|
|
{
|
|
{0x08, 0xB0, 0x00, 'W', 'i', 'n', 'e'},
|
|
7,
|
|
STATUS_BAD_COMPRESSION_BUFFER
|
|
},
|
|
/* two compressed chunks, the second one incomplete */
|
|
{
|
|
{0x00, 0xB0, 0x02, 0x00, 0xB0},
|
|
5,
|
|
STATUS_BAD_COMPRESSION_BUFFER,
|
|
}
|
|
};
|
|
|
|
static UCHAR buf[0x2000], workspace[0x1000];
|
|
NTSTATUS status, expected_status;
|
|
ULONG final_size;
|
|
int i;
|
|
|
|
if (!pRtlDecompressBuffer || !pRtlDecompressFragment)
|
|
{
|
|
win_skip("RtlDecompressBuffer or RtlDecompressFragment is not available\n");
|
|
return;
|
|
}
|
|
|
|
/* test compression format / engine */
|
|
final_size = 0xdeadbeef;
|
|
status = pRtlDecompressBuffer(COMPRESSION_FORMAT_NONE, buf, sizeof(buf), test_lznt[0].compressed,
|
|
test_lznt[0].compressed_size, &final_size);
|
|
ok(status == STATUS_INVALID_PARAMETER, "got wrong status 0x%08x\n", status);
|
|
ok(final_size == 0xdeadbeef, "got wrong final_size %u\n", final_size);
|
|
|
|
final_size = 0xdeadbeef;
|
|
status = pRtlDecompressBuffer(COMPRESSION_FORMAT_DEFAULT, buf, sizeof(buf), test_lznt[0].compressed,
|
|
test_lznt[0].compressed_size, &final_size);
|
|
ok(status == STATUS_INVALID_PARAMETER, "got wrong status 0x%08x\n", status);
|
|
ok(final_size == 0xdeadbeef, "got wrong final_size %u\n", final_size);
|
|
|
|
final_size = 0xdeadbeef;
|
|
status = pRtlDecompressBuffer(0xFF, buf, sizeof(buf), test_lznt[0].compressed,
|
|
test_lznt[0].compressed_size, &final_size);
|
|
ok(status == STATUS_UNSUPPORTED_COMPRESSION, "got wrong status 0x%08x\n", status);
|
|
ok(final_size == 0xdeadbeef, "got wrong final_size %u\n", final_size);
|
|
|
|
/* regular tests for RtlDecompressBuffer */
|
|
for (i = 0; i < sizeof(test_lznt) / sizeof(test_lznt[0]); i++)
|
|
{
|
|
trace("Running test %d (compressed_size=%u, uncompressed_size=%u, status=0x%08x)\n",
|
|
i, test_lznt[i].compressed_size, test_lznt[i].uncompressed_size, test_lznt[i].status);
|
|
|
|
/* test with very big buffer */
|
|
final_size = 0xdeadbeef;
|
|
memset(buf, 0x11, sizeof(buf));
|
|
status = pRtlDecompressBuffer(COMPRESSION_FORMAT_LZNT1, buf, sizeof(buf), test_lznt[i].compressed,
|
|
test_lznt[i].compressed_size, &final_size);
|
|
ok(status == test_lznt[i].status || broken(status == STATUS_BAD_COMPRESSION_BUFFER &&
|
|
(test_lznt[i].broken_flags & DECOMPRESS_BROKEN_FRAGMENT)), "%d: got wrong status 0x%08x\n", i, status);
|
|
if (!status)
|
|
{
|
|
ok(final_size == test_lznt[i].uncompressed_size,
|
|
"%d: got wrong final_size %u\n", i, final_size);
|
|
ok(!memcmp(buf, test_lznt[i].uncompressed, test_lznt[i].uncompressed_size),
|
|
"%d: got wrong decoded data\n", i);
|
|
ok(buf[test_lznt[i].uncompressed_size] == 0x11,
|
|
"%d: buf[%u] was modified\n", i, test_lznt[i].uncompressed_size);
|
|
}
|
|
|
|
/* test that modifier for compression engine is ignored */
|
|
final_size = 0xdeadbeef;
|
|
memset(buf, 0x11, sizeof(buf));
|
|
status = pRtlDecompressBuffer(COMPRESSION_FORMAT_LZNT1 | COMPRESSION_ENGINE_MAXIMUM, buf, sizeof(buf),
|
|
test_lznt[i].compressed, test_lznt[i].compressed_size, &final_size);
|
|
ok(status == test_lznt[i].status || broken(status == STATUS_BAD_COMPRESSION_BUFFER &&
|
|
(test_lznt[i].broken_flags & DECOMPRESS_BROKEN_FRAGMENT)), "%d: got wrong status 0x%08x\n", i, status);
|
|
if (!status)
|
|
{
|
|
ok(final_size == test_lznt[i].uncompressed_size,
|
|
"%d: got wrong final_size %u\n", i, final_size);
|
|
ok(!memcmp(buf, test_lznt[i].uncompressed, test_lznt[i].uncompressed_size),
|
|
"%d: got wrong decoded data\n", i);
|
|
ok(buf[test_lznt[i].uncompressed_size] == 0x11,
|
|
"%d: buf[%u] was modified\n", i, test_lznt[i].uncompressed_size);
|
|
}
|
|
|
|
/* test with expected output size */
|
|
if (test_lznt[i].uncompressed_size > 0)
|
|
{
|
|
final_size = 0xdeadbeef;
|
|
memset(buf, 0x11, sizeof(buf));
|
|
status = pRtlDecompressBuffer(COMPRESSION_FORMAT_LZNT1, buf, test_lznt[i].uncompressed_size,
|
|
test_lznt[i].compressed, test_lznt[i].compressed_size, &final_size);
|
|
ok(status == test_lznt[i].status, "%d: got wrong status 0x%08x\n", i, status);
|
|
if (!status)
|
|
{
|
|
ok(final_size == test_lznt[i].uncompressed_size,
|
|
"%d: got wrong final_size %u\n", i, final_size);
|
|
ok(!memcmp(buf, test_lznt[i].uncompressed, test_lznt[i].uncompressed_size),
|
|
"%d: got wrong decoded data\n", i);
|
|
ok(buf[test_lznt[i].uncompressed_size] == 0x11,
|
|
"%d: buf[%u] was modified\n", i, test_lznt[i].uncompressed_size);
|
|
}
|
|
}
|
|
|
|
/* test with smaller output size */
|
|
if (test_lznt[i].uncompressed_size > 1)
|
|
{
|
|
final_size = 0xdeadbeef;
|
|
memset(buf, 0x11, sizeof(buf));
|
|
status = pRtlDecompressBuffer(COMPRESSION_FORMAT_LZNT1, buf, test_lznt[i].uncompressed_size - 1,
|
|
test_lznt[i].compressed, test_lznt[i].compressed_size, &final_size);
|
|
if (test_lznt[i].broken_flags & DECOMPRESS_BROKEN_TRUNCATED)
|
|
todo_wine
|
|
ok(status == STATUS_BAD_COMPRESSION_BUFFER, "%d: got wrong status 0x%08x\n", i, status);
|
|
else
|
|
ok(status == test_lznt[i].status, "%d: got wrong status 0x%08x\n", i, status);
|
|
if (!status)
|
|
{
|
|
ok(final_size == test_lznt[i].uncompressed_size - 1,
|
|
"%d: got wrong final_size %u\n", i, final_size);
|
|
ok(!memcmp(buf, test_lznt[i].uncompressed, test_lznt[i].uncompressed_size - 1),
|
|
"%d: got wrong decoded data\n", i);
|
|
ok(buf[test_lznt[i].uncompressed_size - 1] == 0x11,
|
|
"%d: buf[%u] was modified\n", i, test_lznt[i].uncompressed_size - 1);
|
|
}
|
|
}
|
|
|
|
/* test with zero output size */
|
|
final_size = 0xdeadbeef;
|
|
memset(buf, 0x11, sizeof(buf));
|
|
status = pRtlDecompressBuffer(COMPRESSION_FORMAT_LZNT1, buf, 0, test_lznt[i].compressed,
|
|
test_lznt[i].compressed_size, &final_size);
|
|
if (is_incomplete_chunk(test_lznt[i].compressed, test_lznt[i].compressed_size, FALSE))
|
|
ok(status == STATUS_BAD_COMPRESSION_BUFFER, "%d: got wrong status 0x%08x\n", i, status);
|
|
else
|
|
{
|
|
ok(status == STATUS_SUCCESS, "%d: got wrong status 0x%08x\n", i, status);
|
|
ok(final_size == 0, "%d: got wrong final_size %u\n", i, final_size);
|
|
ok(buf[0] == 0x11, "%d: buf[0] was modified\n", i);
|
|
}
|
|
|
|
/* test RtlDecompressFragment with offset = 0 */
|
|
final_size = 0xdeadbeef;
|
|
memset(buf, 0x11, sizeof(buf));
|
|
status = pRtlDecompressFragment(COMPRESSION_FORMAT_LZNT1, buf, sizeof(buf), test_lznt[i].compressed,
|
|
test_lznt[i].compressed_size, 0, &final_size, workspace);
|
|
if (test_lznt[i].broken_flags & DECOMPRESS_BROKEN_FRAGMENT)
|
|
todo_wine
|
|
ok(status == STATUS_BAD_COMPRESSION_BUFFER, "%d: got wrong status 0x%08x\n", i, status);
|
|
else
|
|
ok(status == test_lznt[i].status, "%d: got wrong status 0x%08x\n", i, status);
|
|
if (!status)
|
|
{
|
|
ok(final_size == test_lznt[i].uncompressed_size,
|
|
"%d: got wrong final_size %u\n", i, final_size);
|
|
ok(!memcmp(buf, test_lznt[i].uncompressed, test_lznt[i].uncompressed_size),
|
|
"%d: got wrong decoded data\n", i);
|
|
ok(buf[test_lznt[i].uncompressed_size] == 0x11,
|
|
"%d: buf[%u] was modified\n", i, test_lznt[i].uncompressed_size);
|
|
}
|
|
|
|
/* test RtlDecompressFragment with offset = 1 */
|
|
final_size = 0xdeadbeef;
|
|
memset(buf, 0x11, sizeof(buf));
|
|
status = pRtlDecompressFragment(COMPRESSION_FORMAT_LZNT1, buf, sizeof(buf), test_lznt[i].compressed,
|
|
test_lznt[i].compressed_size, 1, &final_size, workspace);
|
|
if (test_lznt[i].broken_flags & DECOMPRESS_BROKEN_FRAGMENT)
|
|
todo_wine
|
|
ok(status == STATUS_BAD_COMPRESSION_BUFFER, "%d: got wrong status 0x%08x\n", i, status);
|
|
else
|
|
ok(status == test_lznt[i].status, "%d: got wrong status 0x%08x\n", i, status);
|
|
if (!status)
|
|
{
|
|
if (test_lznt[i].uncompressed_size == 0)
|
|
{
|
|
todo_wine
|
|
ok(final_size == 4095, "%d: got wrong final_size %u\n", i, final_size);
|
|
/* Buffer doesn't contain any useful value on Windows */
|
|
ok(buf[4095] == 0x11, "%d: buf[4095] was modified\n", i);
|
|
}
|
|
else
|
|
{
|
|
ok(final_size == test_lznt[i].uncompressed_size - 1,
|
|
"%d: got wrong final_size %u\n", i, final_size);
|
|
ok(!memcmp(buf, test_lznt[i].uncompressed + 1, test_lznt[i].uncompressed_size - 1),
|
|
"%d: got wrong decoded data\n", i);
|
|
ok(buf[test_lznt[i].uncompressed_size - 1] == 0x11,
|
|
"%d: buf[%u] was modified\n", i, test_lznt[i].uncompressed_size - 1);
|
|
}
|
|
}
|
|
|
|
/* test RtlDecompressFragment with offset = 4095 */
|
|
final_size = 0xdeadbeef;
|
|
memset(buf, 0x11, sizeof(buf));
|
|
status = pRtlDecompressFragment(COMPRESSION_FORMAT_LZNT1, buf, sizeof(buf), test_lznt[i].compressed,
|
|
test_lznt[i].compressed_size, 4095, &final_size, workspace);
|
|
if (test_lznt[i].broken_flags & DECOMPRESS_BROKEN_FRAGMENT)
|
|
todo_wine
|
|
ok(status == STATUS_BAD_COMPRESSION_BUFFER, "%d: got wrong status 0x%08x\n", i, status);
|
|
else
|
|
ok(status == test_lznt[i].status, "%d: got wrong status 0x%08x\n", i, status);
|
|
if (!status)
|
|
{
|
|
todo_wine
|
|
ok(final_size == 1, "%d: got wrong final_size %u\n", i, final_size);
|
|
todo_wine
|
|
ok(buf[0] == 0, "%d: padding is not zero\n", i);
|
|
ok(buf[1] == 0x11, "%d: buf[1] was modified\n", i);
|
|
}
|
|
|
|
/* test RtlDecompressFragment with offset = 4096 */
|
|
final_size = 0xdeadbeef;
|
|
memset(buf, 0x11, sizeof(buf));
|
|
status = pRtlDecompressFragment(COMPRESSION_FORMAT_LZNT1, buf, sizeof(buf), test_lznt[i].compressed,
|
|
test_lznt[i].compressed_size, 4096, &final_size, workspace);
|
|
expected_status = is_incomplete_chunk(test_lznt[i].compressed, test_lznt[i].compressed_size, TRUE) ?
|
|
test_lznt[i].status : STATUS_SUCCESS;
|
|
ok(status == expected_status, "%d: got wrong status 0x%08x, expected 0x%08x\n", i, status, expected_status);
|
|
if (!status)
|
|
{
|
|
ok(final_size == 0, "%d: got wrong final_size %u\n", i, final_size);
|
|
ok(buf[0] == 0x11, "%d: buf[4096] was modified\n", i);
|
|
}
|
|
}
|
|
}
|
|
|
|
#undef DECOMPRESS_BROKEN_FRAGMENT
|
|
#undef DECOMPRESS_BROKEN_TRUNCATED
|
|
|
|
struct critsect_locked_info
|
|
{
|
|
CRITICAL_SECTION crit;
|
|
HANDLE semaphores[2];
|
|
};
|
|
|
|
static DWORD WINAPI critsect_locked_thread(void *param)
|
|
{
|
|
struct critsect_locked_info *info = param;
|
|
DWORD ret;
|
|
|
|
ret = pRtlIsCriticalSectionLocked(&info->crit);
|
|
ok(ret == TRUE, "expected TRUE, got %u\n", ret);
|
|
ret = pRtlIsCriticalSectionLockedByThread(&info->crit);
|
|
ok(ret == FALSE, "expected FALSE, got %u\n", ret);
|
|
|
|
ReleaseSemaphore(info->semaphores[0], 1, NULL);
|
|
ret = WaitForSingleObject(info->semaphores[1], 1000);
|
|
ok(ret == WAIT_OBJECT_0, "expected WAIT_OBJECT_0, got %u\n", ret);
|
|
|
|
ret = pRtlIsCriticalSectionLocked(&info->crit);
|
|
ok(ret == FALSE, "expected FALSE, got %u\n", ret);
|
|
ret = pRtlIsCriticalSectionLockedByThread(&info->crit);
|
|
ok(ret == FALSE, "expected FALSE, got %u\n", ret);
|
|
|
|
EnterCriticalSection(&info->crit);
|
|
|
|
ret = pRtlIsCriticalSectionLocked(&info->crit);
|
|
ok(ret == TRUE, "expected TRUE, got %u\n", ret);
|
|
ret = pRtlIsCriticalSectionLockedByThread(&info->crit);
|
|
ok(ret == TRUE, "expected TRUE, got %u\n", ret);
|
|
|
|
ReleaseSemaphore(info->semaphores[0], 1, NULL);
|
|
ret = WaitForSingleObject(info->semaphores[1], 1000);
|
|
ok(ret == WAIT_OBJECT_0, "expected WAIT_OBJECT_0, got %u\n", ret);
|
|
|
|
LeaveCriticalSection(&info->crit);
|
|
return 0;
|
|
}
|
|
|
|
static void test_RtlIsCriticalSectionLocked(void)
|
|
{
|
|
struct critsect_locked_info info;
|
|
HANDLE thread;
|
|
BOOL ret;
|
|
|
|
if (!pRtlIsCriticalSectionLocked || !pRtlIsCriticalSectionLockedByThread)
|
|
{
|
|
win_skip("skipping RtlIsCriticalSectionLocked tests, required functions not available\n");
|
|
return;
|
|
}
|
|
|
|
InitializeCriticalSection(&info.crit);
|
|
info.semaphores[0] = CreateSemaphoreW(NULL, 0, 1, NULL);
|
|
ok(info.semaphores[0] != NULL, "CreateSemaphore failed with %u\n", GetLastError());
|
|
info.semaphores[1] = CreateSemaphoreW(NULL, 0, 1, NULL);
|
|
ok(info.semaphores[1] != NULL, "CreateSemaphore failed with %u\n", GetLastError());
|
|
|
|
ret = pRtlIsCriticalSectionLocked(&info.crit);
|
|
ok(ret == FALSE, "expected FALSE, got %u\n", ret);
|
|
ret = pRtlIsCriticalSectionLockedByThread(&info.crit);
|
|
ok(ret == FALSE, "expected FALSE, got %u\n", ret);
|
|
|
|
EnterCriticalSection(&info.crit);
|
|
|
|
ret = pRtlIsCriticalSectionLocked(&info.crit);
|
|
ok(ret == TRUE, "expected TRUE, got %u\n", ret);
|
|
ret = pRtlIsCriticalSectionLockedByThread(&info.crit);
|
|
ok(ret == TRUE, "expected TRUE, got %u\n", ret);
|
|
|
|
thread = CreateThread(NULL, 0, critsect_locked_thread, &info, 0, NULL);
|
|
ok(thread != NULL, "CreateThread failed with %u\n", GetLastError());
|
|
ret = WaitForSingleObject(info.semaphores[0], 1000);
|
|
ok(ret == WAIT_OBJECT_0, "expected WAIT_OBJECT_0, got %u\n", ret);
|
|
|
|
LeaveCriticalSection(&info.crit);
|
|
|
|
ReleaseSemaphore(info.semaphores[1], 1, NULL);
|
|
ret = WaitForSingleObject(info.semaphores[0], 1000);
|
|
ok(ret == WAIT_OBJECT_0, "expected WAIT_OBJECT_0, got %u\n", ret);
|
|
|
|
ret = pRtlIsCriticalSectionLocked(&info.crit);
|
|
ok(ret == TRUE, "expected TRUE, got %u\n", ret);
|
|
ret = pRtlIsCriticalSectionLockedByThread(&info.crit);
|
|
ok(ret == FALSE, "expected FALSE, got %u\n", ret);
|
|
|
|
ReleaseSemaphore(info.semaphores[1], 1, NULL);
|
|
ret = WaitForSingleObject(thread, 1000);
|
|
ok(ret == WAIT_OBJECT_0, "expected WAIT_OBJECT_0, got %u\n", ret);
|
|
|
|
CloseHandle(thread);
|
|
CloseHandle(info.semaphores[0]);
|
|
CloseHandle(info.semaphores[1]);
|
|
DeleteCriticalSection(&info.crit);
|
|
}
|
|
|
|
static void test_RtlInitializeCriticalSectionEx(void)
|
|
{
|
|
static const CRITICAL_SECTION_DEBUG *no_debug = (void *)~(ULONG_PTR)0;
|
|
CRITICAL_SECTION cs;
|
|
|
|
if (!pRtlInitializeCriticalSectionEx)
|
|
{
|
|
win_skip("RtlInitializeCriticalSectionEx is not available\n");
|
|
return;
|
|
}
|
|
|
|
memset(&cs, 0x11, sizeof(cs));
|
|
pRtlInitializeCriticalSectionEx(&cs, 0, 0);
|
|
ok((cs.DebugInfo != NULL && cs.DebugInfo != no_debug) || broken(cs.DebugInfo == no_debug) /* >= Win 8 */,
|
|
"expected DebugInfo != NULL and DebugInfo != ~0, got %p\n", cs.DebugInfo);
|
|
ok(cs.LockCount == -1, "expected LockCount == -1, got %d\n", cs.LockCount);
|
|
ok(cs.RecursionCount == 0, "expected RecursionCount == 0, got %d\n", cs.RecursionCount);
|
|
ok(cs.LockSemaphore == NULL, "expected LockSemaphore == NULL, got %p\n", cs.LockSemaphore);
|
|
ok(cs.SpinCount == 0 || broken(cs.SpinCount != 0) /* >= Win 8 */,
|
|
"expected SpinCount == 0, got %ld\n", cs.SpinCount);
|
|
RtlDeleteCriticalSection(&cs);
|
|
|
|
memset(&cs, 0x11, sizeof(cs));
|
|
pRtlInitializeCriticalSectionEx(&cs, 0, RTL_CRITICAL_SECTION_FLAG_NO_DEBUG_INFO);
|
|
todo_wine
|
|
ok(cs.DebugInfo == no_debug, "expected DebugInfo == ~0, got %p\n", cs.DebugInfo);
|
|
ok(cs.LockCount == -1, "expected LockCount == -1, got %d\n", cs.LockCount);
|
|
ok(cs.RecursionCount == 0, "expected RecursionCount == 0, got %d\n", cs.RecursionCount);
|
|
ok(cs.LockSemaphore == NULL, "expected LockSemaphore == NULL, got %p\n", cs.LockSemaphore);
|
|
ok(cs.SpinCount == 0 || broken(cs.SpinCount != 0) /* >= Win 8 */,
|
|
"expected SpinCount == 0, got %ld\n", cs.SpinCount);
|
|
RtlDeleteCriticalSection(&cs);
|
|
}
|
|
|
|
static void test_RtlLeaveCriticalSection(void)
|
|
{
|
|
RTL_CRITICAL_SECTION cs;
|
|
NTSTATUS status;
|
|
|
|
if (!pRtlInitializeCriticalSectionEx)
|
|
return; /* Skip winxp */
|
|
|
|
status = RtlInitializeCriticalSection(&cs);
|
|
ok(!status, "RtlInitializeCriticalSection failed: %x\n", status);
|
|
|
|
status = RtlEnterCriticalSection(&cs);
|
|
ok(!status, "RtlEnterCriticalSection failed: %x\n", status);
|
|
todo_wine
|
|
ok(cs.LockCount == -2, "expected LockCount == -2, got %d\n", cs.LockCount);
|
|
ok(cs.RecursionCount == 1, "expected RecursionCount == 1, got %d\n", cs.RecursionCount);
|
|
ok(cs.OwningThread == ULongToHandle(GetCurrentThreadId()), "unexpected OwningThread\n");
|
|
|
|
status = RtlLeaveCriticalSection(&cs);
|
|
ok(!status, "RtlLeaveCriticalSection failed: %x\n", status);
|
|
ok(cs.LockCount == -1, "expected LockCount == -1, got %d\n", cs.LockCount);
|
|
ok(cs.RecursionCount == 0, "expected RecursionCount == 0, got %d\n", cs.RecursionCount);
|
|
ok(!cs.OwningThread, "unexpected OwningThread %p\n", cs.OwningThread);
|
|
|
|
/*
|
|
* Trying to leave a section that wasn't acquired modifies RecursionCount to an invalid value,
|
|
* but doesn't modify LockCount so that an attempt to enter the section later will work.
|
|
*/
|
|
status = RtlLeaveCriticalSection(&cs);
|
|
ok(!status, "RtlLeaveCriticalSection failed: %x\n", status);
|
|
ok(cs.LockCount == -1, "expected LockCount == -1, got %d\n", cs.LockCount);
|
|
ok(cs.RecursionCount == -1, "expected RecursionCount == -1, got %d\n", cs.RecursionCount);
|
|
ok(!cs.OwningThread, "unexpected OwningThread %p\n", cs.OwningThread);
|
|
|
|
/* and again */
|
|
status = RtlLeaveCriticalSection(&cs);
|
|
ok(!status, "RtlLeaveCriticalSection failed: %x\n", status);
|
|
ok(cs.LockCount == -1, "expected LockCount == -1, got %d\n", cs.LockCount);
|
|
ok(cs.RecursionCount == -2, "expected RecursionCount == -2, got %d\n", cs.RecursionCount);
|
|
ok(!cs.OwningThread, "unexpected OwningThread %p\n", cs.OwningThread);
|
|
|
|
/* entering section fixes RecursionCount */
|
|
status = RtlEnterCriticalSection(&cs);
|
|
ok(!status, "RtlEnterCriticalSection failed: %x\n", status);
|
|
todo_wine
|
|
ok(cs.LockCount == -2, "expected LockCount == -2, got %d\n", cs.LockCount);
|
|
ok(cs.RecursionCount == 1, "expected RecursionCount == 1, got %d\n", cs.RecursionCount);
|
|
ok(cs.OwningThread == ULongToHandle(GetCurrentThreadId()), "unexpected OwningThread\n");
|
|
|
|
status = RtlLeaveCriticalSection(&cs);
|
|
ok(!status, "RtlLeaveCriticalSection failed: %x\n", status);
|
|
ok(cs.LockCount == -1, "expected LockCount == -1, got %d\n", cs.LockCount);
|
|
ok(cs.RecursionCount == 0, "expected RecursionCount == 0, got %d\n", cs.RecursionCount);
|
|
ok(!cs.OwningThread, "unexpected OwningThread %p\n", cs.OwningThread);
|
|
|
|
status = RtlDeleteCriticalSection(&cs);
|
|
ok(!status, "RtlDeleteCriticalSection failed: %x\n", status);
|
|
}
|
|
|
|
struct ldr_enum_context
|
|
{
|
|
BOOL abort;
|
|
BOOL found;
|
|
int count;
|
|
};
|
|
|
|
static void WINAPI ldr_enum_callback(LDR_MODULE *module, void *context, BOOLEAN *stop)
|
|
{
|
|
static const WCHAR ntdllW[] = {'n','t','d','l','l','.','d','l','l',0};
|
|
struct ldr_enum_context *ctx = context;
|
|
|
|
if (!lstrcmpiW(module->BaseDllName.Buffer, ntdllW))
|
|
ctx->found = TRUE;
|
|
|
|
ctx->count++;
|
|
*stop = ctx->abort;
|
|
}
|
|
|
|
static void test_LdrEnumerateLoadedModules(void)
|
|
{
|
|
struct ldr_enum_context ctx;
|
|
NTSTATUS status;
|
|
|
|
if (!pLdrEnumerateLoadedModules)
|
|
{
|
|
win_skip("LdrEnumerateLoadedModules not available\n");
|
|
return;
|
|
}
|
|
|
|
ctx.abort = FALSE;
|
|
ctx.found = FALSE;
|
|
ctx.count = 0;
|
|
status = pLdrEnumerateLoadedModules(NULL, ldr_enum_callback, &ctx);
|
|
ok(status == STATUS_SUCCESS, "LdrEnumerateLoadedModules failed with %08x\n", status);
|
|
ok(ctx.count > 1, "Expected more than one module, got %d\n", ctx.count);
|
|
ok(ctx.found, "Could not find ntdll in list of modules\n");
|
|
|
|
ctx.abort = TRUE;
|
|
ctx.count = 0;
|
|
status = pLdrEnumerateLoadedModules(NULL, ldr_enum_callback, &ctx);
|
|
ok(status == STATUS_SUCCESS, "LdrEnumerateLoadedModules failed with %08x\n", status);
|
|
ok(ctx.count == 1, "Expected exactly one module, got %d\n", ctx.count);
|
|
|
|
status = pLdrEnumerateLoadedModules((void *)0x1, ldr_enum_callback, (void *)0xdeadbeef);
|
|
ok(status == STATUS_INVALID_PARAMETER, "expected STATUS_INVALID_PARAMETER, got 0x%08x\n", status);
|
|
|
|
status = pLdrEnumerateLoadedModules((void *)0xdeadbeef, ldr_enum_callback, (void *)0xdeadbeef);
|
|
ok(status == STATUS_INVALID_PARAMETER, "expected STATUS_INVALID_PARAMETER, got 0x%08x\n", status);
|
|
|
|
status = pLdrEnumerateLoadedModules(NULL, NULL, (void *)0xdeadbeef);
|
|
ok(status == STATUS_INVALID_PARAMETER, "expected STATUS_INVALID_PARAMETER, got 0x%08x\n", status);
|
|
}
|
|
|
|
START_TEST(rtl)
|
|
{
|
|
InitFunctionPtrs();
|
|
|
|
test_RtlCompareMemory();
|
|
test_RtlCompareMemoryUlong();
|
|
test_RtlMoveMemory();
|
|
test_RtlFillMemory();
|
|
test_RtlFillMemoryUlong();
|
|
test_RtlZeroMemory();
|
|
test_RtlUlonglongByteSwap();
|
|
test_RtlUniform();
|
|
test_RtlRandom();
|
|
test_RtlAreAllAccessesGranted();
|
|
test_RtlAreAnyAccessesGranted();
|
|
test_RtlComputeCrc32();
|
|
test_HandleTables();
|
|
test_RtlAllocateAndInitializeSid();
|
|
test_RtlDeleteTimer();
|
|
test_RtlThreadErrorMode();
|
|
test_LdrProcessRelocationBlock();
|
|
test_RtlIpv4AddressToString();
|
|
test_RtlIpv4AddressToStringEx();
|
|
test_RtlIpv4StringToAddress();
|
|
test_LdrAddRefDll();
|
|
test_LdrLockLoaderLock();
|
|
test_RtlCompressBuffer();
|
|
test_RtlGetCompressionWorkSpaceSize();
|
|
test_RtlDecompressBuffer();
|
|
test_RtlIsCriticalSectionLocked();
|
|
test_RtlInitializeCriticalSectionEx();
|
|
test_RtlLeaveCriticalSection();
|
|
test_LdrEnumerateLoadedModules();
|
|
}
|