reactos/ntoskrnl/tests/symtest.c

257 lines
8.1 KiB
C
Raw Normal View History

// ntest.cpp : Defines the entry point for the console application.
//
#include "stdafx.h"
#include "windows.h"
#include "ntndk.h"
ULONG Runs;
ULONG Failures;
#define TEST_ASSERT( exp ) \
((!(exp)) ? \
(printf("%s, %s, %d\n", #exp, __FILE__, __LINE__), Failures++, FALSE) : \
TRUE), Runs++
HANDLE
SymlinkCreateTests(OUT PHANDLE OddLink)
{
NTSTATUS Status;
HANDLE LinkHandle;
UNICODE_STRING TargetName = RTL_CONSTANT_STRING(L"\\");
UNICODE_STRING TargetName2 = RTL_CONSTANT_STRING(L"\\");
UNICODE_STRING TargetName3 = RTL_CONSTANT_STRING(L"\\");
UNICODE_STRING TargetName4 = RTL_CONSTANT_STRING(L"\\");
UNICODE_STRING TargetName5 = RTL_CONSTANT_STRING(L"\\");
UNICODE_STRING OkName = RTL_CONSTANT_STRING(L"\\OddLink");
UNICODE_STRING OkName2 = RTL_CONSTANT_STRING(L"\\TestLink");
//
// Test1: Empty Attributes
//
{
OBJECT_ATTRIBUTES Test1 = RTL_INIT_OBJECT_ATTRIBUTES(NULL, 0);
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
&Test1,
&TargetName);
TEST_ASSERT(NT_SUCCESS(Status));
}
//
// Test2: No Attributes
//
{
POBJECT_ATTRIBUTES Test2 = NULL;
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
Test2,
&TargetName);
TEST_ASSERT(NT_SUCCESS(Status));
}
//
// Test3: Attributes with an empty name
//
{
UNICODE_STRING TestName1 = {0, 0, NULL};
OBJECT_ATTRIBUTES Test3 = RTL_INIT_OBJECT_ATTRIBUTES(&TestName1, 0);
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
&Test3,
&TargetName);
TEST_ASSERT(NT_SUCCESS(Status));
}
//
// Test4: Attributes with an invalid name
//
{
UNICODE_STRING TestName2 = {10, 12, UlongToPtr(0x81000000)};
OBJECT_ATTRIBUTES Test4 = RTL_INIT_OBJECT_ATTRIBUTES(&TestName2, 0);
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
&Test4,
&TargetName);
TEST_ASSERT(Status == STATUS_ACCESS_VIOLATION);
}
//
// Test5: Target with an odd name len
//
{
UNICODE_STRING OddName = RTL_CONSTANT_STRING(L"\\TestLink");
OBJECT_ATTRIBUTES Test5 = RTL_INIT_OBJECT_ATTRIBUTES(&OkName, 0);
TargetName3.Length--;
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
&Test5,
&TargetName3);
TEST_ASSERT(Status == STATUS_INVALID_PARAMETER);
}
//
// Test6: Target with an emtpy name len
//
{
OBJECT_ATTRIBUTES Test5 = RTL_INIT_OBJECT_ATTRIBUTES(&OkName, 0);
TargetName4.MaximumLength = 0;
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
&Test5,
&TargetName4);
TEST_ASSERT(Status == STATUS_INVALID_PARAMETER);
}
//
// Test7: Target with an name len larger then maxlen
//
{
OBJECT_ATTRIBUTES Test6 = RTL_INIT_OBJECT_ATTRIBUTES(&OkName, 0);
TargetName5.Length = TargetName5.MaximumLength + 2;
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
&Test6,
&TargetName5);
TEST_ASSERT(Status == STATUS_INVALID_PARAMETER);
}
//
// Test5: Target with an odd name maxlen
//
{
OBJECT_ATTRIBUTES Test5 = RTL_INIT_OBJECT_ATTRIBUTES(&OkName, 0);
TargetName2.MaximumLength--;
TEST_ASSERT(TargetName2.MaximumLength % sizeof(WCHAR));
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
&Test5,
&TargetName2);
*OddLink = LinkHandle;
TEST_ASSERT(NT_SUCCESS(Status));
TEST_ASSERT(TargetName2.MaximumLength % sizeof(WCHAR));
}
//
// Test6: collission
//
{
OBJECT_ATTRIBUTES Test6 = RTL_INIT_OBJECT_ATTRIBUTES(&OkName, 0);
TargetName2.MaximumLength++;
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
&Test6,
&TargetName2);
TEST_ASSERT(Status == STATUS_OBJECT_NAME_COLLISION);
}
//
// Test7: OK!
//
{
OBJECT_ATTRIBUTES Test7 = RTL_INIT_OBJECT_ATTRIBUTES(&OkName2, 0);
Status = NtCreateSymbolicLinkObject(&LinkHandle,
SYMBOLIC_LINK_ALL_ACCESS,
&Test7,
&TargetName2);
TEST_ASSERT(NT_SUCCESS(Status));
}
//
// Return the last valid handle
//
return LinkHandle;
};
int _tmain(int argc, _TCHAR* argv[])
{
NTSTATUS Status;
HANDLE LinkHandle, OddHandle;
WCHAR TargetBuffer[MAX_PATH] = {0};
WCHAR TargetBuffer2[MAX_PATH] = {0};
UNICODE_STRING TargetName;
UNICODE_STRING TargetName2;
UNICODE_STRING OkName = RTL_CONSTANT_STRING(L"\\TestLink");
ULONG NameSize;
//
// Start with the create tests
//
LinkHandle = SymlinkCreateTests(&OddHandle);
//
// Setup the two empty strings. One will have a magic-char at the end
//
RtlInitEmptyUnicodeString(&TargetName, TargetBuffer, sizeof(TargetBuffer));
RtlInitEmptyUnicodeString(&TargetName2, TargetBuffer2, sizeof(TargetBuffer2));
//
// Now query the odd link
//
Status = NtQuerySymbolicLinkObject(OddHandle,
&TargetName,
&NameSize);
TEST_ASSERT(NT_SUCCESS(Status));
TEST_ASSERT(NameSize == sizeof(WCHAR));
TEST_ASSERT(TargetName.Length == NameSize);
TEST_ASSERT(TargetName.Buffer[0] == OBJ_NAME_PATH_SEPARATOR);
NtClose(OddHandle);
//
// Now query the test link
//
Status = NtQuerySymbolicLinkObject(LinkHandle,
&TargetName,
&NameSize);
TEST_ASSERT(NT_SUCCESS(Status));
TEST_ASSERT(NameSize == 2 * sizeof(WCHAR));
TEST_ASSERT(TargetName.Length == NameSize - sizeof(WCHAR));
TEST_ASSERT(TargetName.Buffer[0] == OBJ_NAME_PATH_SEPARATOR);
//
// Corrupt the characters
//
TargetName.Buffer[(NameSize - 2) / sizeof(WCHAR)] = 'v';
TargetName.Buffer[(NameSize - 3) / sizeof(WCHAR)] = 'v';
//
// Now query the test link
//
Status = NtQuerySymbolicLinkObject(LinkHandle,
&TargetName,
&NameSize);
TEST_ASSERT(NT_SUCCESS(Status));
TEST_ASSERT(TargetName.Buffer[0] == OBJ_NAME_PATH_SEPARATOR);
TEST_ASSERT(TargetName.Buffer[1] == UNICODE_NULL);
//
// Corrupt the characters
//
TargetName.Buffer[(NameSize - 2) / sizeof(WCHAR)] = 'v';
TargetName.Buffer[(NameSize - 3) / sizeof(WCHAR)] = 'v';
//
// Now query the test link
//
Status = NtQuerySymbolicLinkObject(LinkHandle,
&TargetName,
NULL);
TEST_ASSERT(NT_SUCCESS(Status));
TEST_ASSERT(TargetName.Buffer[0] == OBJ_NAME_PATH_SEPARATOR);
TEST_ASSERT(TargetName.Buffer[1] == 'v');
//
// Print out results
//
{
RTL_OSVERSIONINFOW VersionInfo;
VersionInfo.dwOSVersionInfoSize = sizeof(RTL_OSVERSIONINFOW);
RtlGetVersion(&VersionInfo);
printf("Test complete on: %d\n", VersionInfo.dwBuildNumber);
printf("Number of Failures: %d\n", Failures);
printf("Number of Runs: %d\n", Runs);
}
return 0;
}