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