| /* Unit test suite for Ntdll atom API functions |
| * |
| * Copyright 2003 Gyorgy 'Nog' Jeney |
| * |
| * 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 <stdio.h> |
| #include <stdarg.h> |
| |
| #include "ntstatus.h" |
| /* Define WIN32_NO_STATUS so MSVC does not give us duplicate macro |
| * definition errors when we get to winnt.h |
| */ |
| #define WIN32_NO_STATUS |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winreg.h" |
| #include "winnls.h" |
| #include "winuser.h" |
| #include "wine/test.h" |
| #include "winternl.h" |
| |
| #ifndef __WINE_WINTERNL_H |
| typedef unsigned short RTL_ATOM, *PRTL_ATOM; |
| typedef struct atom_table *RTL_ATOM_TABLE, **PRTL_ATOM_TABLE; |
| #endif |
| |
| /* Function pointers for ntdll calls */ |
| static HMODULE hntdll = 0; |
| static NTSTATUS (WINAPI *pRtlCreateAtomTable)(ULONG,PRTL_ATOM_TABLE); |
| static NTSTATUS (WINAPI *pRtlDestroyAtomTable)(RTL_ATOM_TABLE); |
| static NTSTATUS (WINAPI *pRtlEmptyAtomTable)(RTL_ATOM_TABLE,BOOLEAN); |
| static NTSTATUS (WINAPI *pRtlAddAtomToAtomTable)(RTL_ATOM_TABLE,PCWSTR,PRTL_ATOM); |
| static NTSTATUS (WINAPI *pRtlDeleteAtomFromAtomTable)(RTL_ATOM_TABLE,RTL_ATOM); |
| static NTSTATUS (WINAPI *pRtlLookupAtomInAtomTable)(RTL_ATOM_TABLE,PCWSTR,PRTL_ATOM); |
| static NTSTATUS (WINAPI *pRtlPinAtomInAtomTable)(RTL_ATOM_TABLE,RTL_ATOM); |
| static NTSTATUS (WINAPI *pRtlQueryAtomInAtomTable)(RTL_ATOM_TABLE,RTL_ATOM,PULONG,PULONG,PWSTR,PULONG); |
| |
| static NTSTATUS (WINAPI* pNtAddAtom)(LPCWSTR,ULONG,RTL_ATOM*); |
| static NTSTATUS (WINAPI* pNtAddAtomNT4)(LPCWSTR,RTL_ATOM*); |
| static NTSTATUS (WINAPI* pNtQueryInformationAtom)(RTL_ATOM,DWORD,void*,ULONG,PULONG); |
| |
| static const WCHAR EmptyAtom[] = {0}; |
| static const WCHAR testAtom1[] = {'H','e','l','l','o',' ','W','o','r','l','d',0}; |
| static const WCHAR testAtom2[] = {'H','e','l','l','o',' ','W','o','r','l','d','2',0}; |
| static const WCHAR testAtom3[] = {'H','e','l','l','o',' ','W','o','r','l','d','3',0}; |
| |
| static const WCHAR testAtom1Cap[] = {'H','E','L','L','O',' ','W','O','R','L','D',0}; |
| static const WCHAR testAtom1Low[] = {'h','e','l','l','o',' ','w','o','r','l','d',0}; |
| |
| static const WCHAR testAtomInt[] = {'#','1','3','2',0}; |
| static const WCHAR testAtomIntInv[] = {'#','2','3','4','z',0}; |
| static const WCHAR testAtomOTT[] = {'#','1','2','3',0}; |
| |
| static void InitFunctionPtr(void) |
| { |
| hntdll = LoadLibraryA("ntdll.dll"); |
| ok(hntdll != 0, "Unable to load ntdll.dll\n"); |
| |
| if (hntdll) |
| { |
| pRtlCreateAtomTable = (void *)GetProcAddress(hntdll, "RtlCreateAtomTable"); |
| pRtlDestroyAtomTable = (void *)GetProcAddress(hntdll, "RtlDestroyAtomTable"); |
| pRtlEmptyAtomTable = (void *)GetProcAddress(hntdll, "RtlEmptyAtomTable"); |
| pRtlAddAtomToAtomTable = (void *)GetProcAddress(hntdll, "RtlAddAtomToAtomTable"); |
| pRtlDeleteAtomFromAtomTable = (void *)GetProcAddress(hntdll, "RtlDeleteAtomFromAtomTable"); |
| pRtlLookupAtomInAtomTable = (void *)GetProcAddress(hntdll, "RtlLookupAtomInAtomTable"); |
| pRtlPinAtomInAtomTable = (void *)GetProcAddress(hntdll, "RtlPinAtomInAtomTable"); |
| pRtlQueryAtomInAtomTable = (void *)GetProcAddress(hntdll, "RtlQueryAtomInAtomTable"); |
| |
| pNtAddAtom = (void *)GetProcAddress(hntdll, "NtAddAtom"); |
| pNtQueryInformationAtom = (void *)GetProcAddress(hntdll, "NtQueryInformationAtom"); |
| } |
| } |
| |
| static DWORD WINAPI RtlAtomTestThread(LPVOID Table) |
| { |
| RTL_ATOM_TABLE AtomTable = *(PRTL_ATOM_TABLE)Table; |
| RTL_ATOM Atom; |
| NTSTATUS res; |
| ULONG RefCount = 0, PinCount = 0, Len = 0; |
| WCHAR Name[64]; |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1, &Atom); |
| ok(!res, "Unable to find atom from another thread, retval: %x\n", res); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom2, &Atom); |
| ok(!res, "Unable to lookup pinned atom in table, retval: %x\n", res); |
| |
| res = pRtlQueryAtomInAtomTable(AtomTable, Atom, &RefCount, &PinCount, Name, &Len); |
| ok(res == STATUS_BUFFER_TOO_SMALL, "We got wrong retval: %x\n", res); |
| |
| Len = 64; |
| res = pRtlQueryAtomInAtomTable(AtomTable, Atom, &RefCount, &PinCount, Name, &Len); |
| ok(!res, "Failed with long enough buffer, retval: %x\n", res); |
| ok(RefCount == 1, "Refcount was not 1 but %x\n", RefCount); |
| ok(PinCount == 1, "Pincount was not 1 but %x\n", PinCount); |
| ok(!lstrcmpW(Name, testAtom2), "We found wrong atom!!\n"); |
| ok((lstrlenW(testAtom2) * sizeof(WCHAR)) == Len, "Returned wrong length %d\n", Len); |
| |
| Len = 64; |
| res = pRtlQueryAtomInAtomTable(AtomTable, Atom, NULL, NULL, Name, &Len); |
| ok(!res, "RtlQueryAtomInAtomTable with optional args invalid failed, retval: %x\n", res); |
| ok(!lstrcmpW(Name, testAtom2), "Found Wrong atom!\n"); |
| ok((lstrlenW(testAtom2) * sizeof(WCHAR)) == Len, "Returned wrong length %d\n", Len); |
| |
| res = pRtlPinAtomInAtomTable(AtomTable, Atom); |
| ok(!res, "Unable to pin atom in atom table, retval: %x\n", res); |
| |
| return 0; |
| } |
| |
| static void test_NtAtom(void) |
| { |
| RTL_ATOM_TABLE AtomTable = NULL; |
| NTSTATUS res; |
| RTL_ATOM Atom1, Atom2, Atom3, testEAtom, testAtom; |
| HANDLE testThread; |
| ULONG RefCount = 0, PinCount = 0, Len = 0; |
| WCHAR Name[64]; |
| |
| /* If we pass a non-null string to create atom table, then it thinks that we |
| * have passed it an already allocated atom table */ |
| res = pRtlCreateAtomTable(0, &AtomTable); |
| ok(!res, "RtlCreateAtomTable should succeed with an atom table size of 0\n"); |
| |
| if (!res) |
| { |
| res = pRtlDestroyAtomTable(AtomTable); |
| ok(!res, "We could create the atom table, but we couldn't destroy it! retval: %x\n", res); |
| } |
| |
| AtomTable = NULL; |
| res = pRtlCreateAtomTable(37, &AtomTable); |
| ok(!res, "We're unable to create an atom table with a valid table size retval: %x\n", res); |
| if (!res) |
| { |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtom1, &Atom1); |
| ok(!res, "We were unable to add a simple atom to the atom table, retval: %x\n", res); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1Cap, &testAtom); |
| ok(!res, "We were unable to find capital version of the atom, retval: %x\n", res); |
| ok(Atom1 == testAtom, "Found wrong atom in table when querying capital atom\n"); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1Low, &testAtom); |
| ok(!res, "Unable to find lowercase version of the atom, retval: %x\n", res); |
| ok(testAtom == Atom1, "Found wrong atom when querying lowercase atom\n"); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, EmptyAtom, &testEAtom); |
| ok(res == STATUS_OBJECT_NAME_INVALID, "Got wrong retval, retval: %x\n", res); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1, &testAtom); |
| ok(!res, "Failed to find totally legitimate atom, retval: %x\n", res); |
| ok(testAtom == Atom1, "Found wrong atom!\n"); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtom2, &Atom2); |
| ok(!res, "Unable to add other legitimate atom to table, retval: %x\n", res); |
| |
| res = pRtlPinAtomInAtomTable(AtomTable, Atom2); |
| ok(!res, "Unable to pin atom in atom table, retval: %x\n", res); |
| |
| testThread = CreateThread(NULL, 0, RtlAtomTestThread, &AtomTable, 0, NULL); |
| WaitForSingleObject(testThread, INFINITE); |
| |
| Len = 64; |
| res = pRtlQueryAtomInAtomTable(AtomTable, Atom2, &RefCount, &PinCount, Name, &Len); |
| ok(!res, "Unable to query atom in atom table, retval: %x\n", res); |
| ok(RefCount == 1, "RefCount is not 1 but %x\n", RefCount); |
| ok(PinCount == 1, "PinCount is not 1 but %x\n", PinCount); |
| ok(!lstrcmpW(Name, testAtom2), "We found wrong atom\n"); |
| ok((lstrlenW(testAtom2) * sizeof(WCHAR)) == Len, "Returned wrong length %d\n", Len); |
| |
| res = pRtlEmptyAtomTable(AtomTable, FALSE); |
| ok(!res, "Unable to empty atom table, retval %x\n", res); |
| |
| Len = 64; |
| res = pRtlQueryAtomInAtomTable(AtomTable, Atom2, &RefCount, &PinCount, Name, &Len); |
| ok(!res, "It seems RtlEmptyAtomTable deleted our pinned atom eaven though we asked it not to, retval: %x\n", res); |
| ok(RefCount == 1, "RefCount is not 1 but %x\n", RefCount); |
| ok(PinCount == 1, "PinCount is not 1 but %x\n", PinCount); |
| ok(!lstrcmpW(Name, testAtom2), "We found wrong atom\n"); |
| ok((lstrlenW(testAtom2) * sizeof(WCHAR)) == Len, "Returned wrong length %d\n", Len); |
| |
| Len = 8; |
| Name[0] = Name[1] = Name[2] = Name[3] = Name[4] = 0x1337; |
| res = pRtlQueryAtomInAtomTable(AtomTable, Atom2, NULL, NULL, Name, &Len); |
| ok(!res, "query atom %x\n", res); |
| ok(Len == 6, "wrong length %u\n", Len); |
| ok(!memcmp(Name, testAtom2, Len), "wrong atom string\n"); |
| ok(!Name[3], "wrong string termination\n"); |
| ok(Name[4] == 0x1337, "buffer overwrite\n"); |
| |
| Len = lstrlenW(testAtom2) * sizeof(WCHAR); |
| memset(Name, '.', sizeof(Name)); |
| res = pRtlQueryAtomInAtomTable( AtomTable, Atom2, NULL, NULL, Name, &Len ); |
| ok(!res, "query atom %x\n", res); |
| ok(Len == (lstrlenW(testAtom2) - 1) * sizeof(WCHAR), "wrong length %u\n", Len); |
| ok(!memcmp(testAtom2, Name, (lstrlenW(testAtom2) - 1) * sizeof(WCHAR)), "wrong atom name\n"); |
| ok(Name[lstrlenW(testAtom2) - 1] == '\0', "wrong char\n"); |
| ok(Name[lstrlenW(testAtom2)] == ('.' << 8) + '.', "wrong char\n"); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom2, &testAtom); |
| ok(!res, "We can't find our pinned atom!! retval: %x\n", res); |
| ok(testAtom == Atom2, "We found wrong atom!!!\n"); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1, &testAtom); |
| ok(res == STATUS_OBJECT_NAME_NOT_FOUND, "We found the atom in our table eaven though we asked RtlEmptyAtomTable to remove it, retval: %x\n", res); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtom3, &Atom3); |
| ok(!res, "Unable to add atom to table, retval: %x\n", res); |
| |
| res = pRtlEmptyAtomTable(AtomTable, TRUE); |
| ok(!res, "Unable to empty atom table, retval: %x\n", res); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom2, &testAtom); |
| ok(res == STATUS_OBJECT_NAME_NOT_FOUND, "The pinned atom should be removed, retval: %x\n", res); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom3, &testAtom); |
| ok(res == STATUS_OBJECT_NAME_NOT_FOUND, "Non pinned atom should also be removed, retval: %x\n", res); |
| |
| res = pRtlDestroyAtomTable(AtomTable); |
| ok(!res, "Can't destroy atom table, retval: %x\n", res); |
| } |
| |
| AtomTable = NULL; |
| res = pRtlCreateAtomTable(37, &AtomTable); |
| ok(!res, "Unable to create atom table, retval: %x\n", res); |
| |
| if (!res) |
| { |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1, &testAtom); |
| ok(res == STATUS_OBJECT_NAME_NOT_FOUND, "Didn't get expected retval with querying an empty atom table, retval: %x\n", res); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtom1, &Atom1); |
| ok(!res, "Unable to add atom to atom table, retval %x\n", res); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1, &testAtom); |
| ok(!res, "Can't find previously added atom in table, retval: %x\n", res); |
| ok(testAtom == Atom1, "Found wrong atom! retval: %x\n", res); |
| |
| res = pRtlDeleteAtomFromAtomTable(AtomTable, Atom1); |
| ok(!res, "Unable to delete atom from table, retval: %x\n", res); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1, &testAtom); |
| ok(res == STATUS_OBJECT_NAME_NOT_FOUND, "Able to find previously deleted atom in table, retval: %x\n", res); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtom1, &Atom1); |
| ok(!res, "Unable to add atom to atom table, retval: %x\n", res); |
| |
| Len = 0; |
| res = pRtlQueryAtomInAtomTable(AtomTable, Atom1, NULL, NULL, Name, &Len); |
| ok(res == STATUS_BUFFER_TOO_SMALL, "Got wrong retval, retval: %x\n", res); |
| ok((lstrlenW(testAtom1) * sizeof(WCHAR)) == Len || broken(!Len) /* nt4 */, "Got wrong length %x\n", Len); |
| if (!Len) pNtAddAtomNT4 = (void *)pNtAddAtom; |
| |
| res = pRtlPinAtomInAtomTable(AtomTable, Atom1); |
| ok(!res, "Unable to pin atom in atom table, retval: %x\n", res); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1, &testAtom); |
| ok(!res, "Unable to find atom in atom table, retval: %x\n", res); |
| ok(testAtom == Atom1, "Wrong atom found\n"); |
| |
| res = pRtlDeleteAtomFromAtomTable(AtomTable, Atom1); |
| ok(res == STATUS_WAS_LOCKED, "Unable to delete atom from table, retval: %x\n", res); |
| |
| res = pRtlLookupAtomInAtomTable(AtomTable, testAtom1, &testAtom); |
| ok(!res, "Able to find deleted atom in table\n"); |
| |
| res = pRtlDestroyAtomTable(AtomTable); |
| ok(!res, "Unable to destroy atom table\n"); |
| } |
| } |
| |
| /* Test Adding integer atoms to atom table */ |
| static void test_NtIntAtom(void) |
| { |
| NTSTATUS res; |
| RTL_ATOM_TABLE AtomTable; |
| RTL_ATOM testAtom; |
| ULONG RefCount = 0, PinCount = 0; |
| INT_PTR i; |
| WCHAR Name[64]; |
| ULONG Len; |
| |
| AtomTable = NULL; |
| res = pRtlCreateAtomTable(37, &AtomTable); |
| ok(!res, "Unable to create atom table, %x\n", res); |
| |
| if (!res) |
| { |
| /* According to the kernel32 functions, integer atoms are only allowed from |
| * 0x0001 to 0xbfff and not 0xc000 to 0xffff, which is correct */ |
| res = pRtlAddAtomToAtomTable(AtomTable, NULL, &testAtom); |
| ok(res == STATUS_INVALID_PARAMETER, "Didn't get expected result from adding 0 int atom, retval: %x\n", res); |
| for (i = 1; i <= 0xbfff; i++) |
| { |
| res = pRtlAddAtomToAtomTable(AtomTable, (LPWSTR)i, &testAtom); |
| ok(!res, "Unable to add valid integer atom %li, retval: %x\n", i, res); |
| } |
| |
| for (i = 1; i <= 0xbfff; i++) |
| { |
| res = pRtlLookupAtomInAtomTable(AtomTable, (LPWSTR)i, &testAtom); |
| ok(!res, "Unable to find int atom %li, retval: %x\n", i, res); |
| if (!res) |
| { |
| res = pRtlPinAtomInAtomTable(AtomTable, testAtom); |
| ok(!res, "Unable to pin int atom %li, retval: %x\n", i, res); |
| } |
| } |
| |
| for (i = 0xc000; i <= 0xffff; i++) |
| { |
| res = pRtlAddAtomToAtomTable(AtomTable, (LPWSTR)i, &testAtom); |
| ok(res, "Able to illeageal integer atom %li, retval: %x\n", i, res); |
| } |
| |
| res = pRtlDestroyAtomTable(AtomTable); |
| ok(!res, "Unable to destroy atom table, retval: %x\n", res); |
| } |
| |
| AtomTable = NULL; |
| res = pRtlCreateAtomTable(37, &AtomTable); |
| ok(!res, "Unable to create atom table, %x\n", res); |
| if (!res) |
| { |
| res = pRtlLookupAtomInAtomTable(AtomTable, (PWSTR)123, &testAtom); |
| ok(!res, "Unable to query atom in atom table, retval: %x\n", res); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtomInt, &testAtom); |
| ok(!res, "Unable to add int atom to table, retval: %x\n", res); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtomIntInv, &testAtom); |
| ok(!res, "Unable to add int atom to table, retval: %x\n", res); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, (PWSTR)123, &testAtom); |
| ok(!res, "Unable to add int atom to table, retval: %x\n", res); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, (PWSTR)123, &testAtom); |
| ok(!res, "Unable to re-add int atom to table, retval: %x\n", res); |
| |
| Len = 64; |
| res = pRtlQueryAtomInAtomTable(AtomTable, testAtom, &RefCount, &PinCount, Name, &Len); |
| ok(!res, "Unable to query atom in atom table, retval: %x\n", res); |
| ok(PinCount == 1, "Expected pincount 1 but got %x\n", PinCount); |
| ok(RefCount == 1, "Expected refcount 1 but got %x\n", RefCount); |
| ok(!lstrcmpW(testAtomOTT, Name), "Got wrong atom name\n"); |
| ok((lstrlenW(testAtomOTT) * sizeof(WCHAR)) == Len, "Got wrong len %d\n", Len); |
| |
| res = pRtlPinAtomInAtomTable(AtomTable, testAtom); |
| ok(!res, "Unable to pin int atom, retval: %x\n", res); |
| |
| res = pRtlPinAtomInAtomTable(AtomTable, testAtom); |
| ok(!res, "Unable to pin int atom, retval: %x\n", res); |
| |
| res = pRtlQueryAtomInAtomTable(AtomTable, testAtom, &RefCount, &PinCount, NULL, NULL); |
| ok(!res, "Unable to query atom in atom table, retval: %x\n", res); |
| ok(PinCount == 1, "Expected pincount 1 but got %x\n", PinCount); |
| ok(RefCount == 1, "Expected refcount 1 but got %x\n", RefCount); |
| |
| res = pRtlDestroyAtomTable(AtomTable); |
| ok(!res, "Unable to destroy atom table, retval: %x\n", res); |
| } |
| } |
| |
| /* Tests to see how the pincount and refcount actually works */ |
| static void test_NtRefPinAtom(void) |
| { |
| RTL_ATOM_TABLE AtomTable; |
| RTL_ATOM Atom; |
| ULONG PinCount = 0, RefCount = 0; |
| NTSTATUS res; |
| |
| AtomTable = NULL; |
| res = pRtlCreateAtomTable(37, &AtomTable); |
| ok(!res, "Unable to create atom table, %x\n", res); |
| |
| if (!res) |
| { |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtom1, &Atom); |
| ok(!res, "Unable to add our atom to the atom table, retval: %x\n", res); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtom1, &Atom); |
| ok(!res, "Unable to add our atom to the atom table, retval: %x\n", res); |
| |
| res = pRtlAddAtomToAtomTable(AtomTable, testAtom1, &Atom); |
| ok(!res, "Unable to add our atom to the atom table, retval: %x\n", res); |
| |
| res = pRtlQueryAtomInAtomTable(AtomTable, Atom, &RefCount, &PinCount, NULL, NULL); |
| ok(!res, "Unable to query atom in atom table, retval: %x\n", res); |
| ok(PinCount == 0, "Expected pincount 0 but got %x\n", PinCount); |
| ok(RefCount == 3, "Expected refcount 3 but got %x\n", RefCount); |
| |
| res = pRtlPinAtomInAtomTable(AtomTable, Atom); |
| ok(!res, "Unable to pin atom in atom table, retval: %x\n", res); |
| |
| res = pRtlPinAtomInAtomTable(AtomTable, Atom); |
| ok(!res, "Unable to pin atom in atom table, retval: %x\n", res); |
| |
| res = pRtlPinAtomInAtomTable(AtomTable, Atom); |
| ok(!res, "Unable to pin atom in atom table, retval: %x\n", res); |
| |
| res = pRtlQueryAtomInAtomTable(AtomTable, Atom, &RefCount, &PinCount, NULL, NULL); |
| ok(!res, "Unable to query atom in atom table, retval: %x\n", res); |
| ok(PinCount == 1, "Expected pincount 1 but got %x\n", PinCount); |
| ok(RefCount == 3, "Expected refcount 3 but got %x\n", RefCount); |
| |
| res = pRtlDestroyAtomTable(AtomTable); |
| ok(!res, "Unable to destroy atom table, retval: %x\n", res); |
| } |
| } |
| |
| static void test_Global(void) |
| { |
| NTSTATUS res; |
| RTL_ATOM atom; |
| ULONG ptr[(sizeof(ATOM_BASIC_INFORMATION) + 255 * sizeof(WCHAR)) / sizeof(ULONG)]; |
| ATOM_BASIC_INFORMATION* abi = (ATOM_BASIC_INFORMATION*)ptr; |
| ULONG ptr_size = sizeof(ptr); |
| |
| if (pNtAddAtomNT4) |
| res = pNtAddAtomNT4(testAtom1, &atom); |
| else |
| res = pNtAddAtom(testAtom1, lstrlenW(testAtom1) * sizeof(WCHAR), &atom); |
| |
| ok(!res, "Added atom (%x)\n", res); |
| |
| memset( ptr, 0xcc, sizeof(ptr) ); |
| res = pNtQueryInformationAtom( atom, AtomBasicInformation, (void*)ptr, ptr_size, NULL ); |
| ok(!res, "atom lookup\n"); |
| ok(!lstrcmpW(abi->Name, testAtom1), "ok strings\n"); |
| ok(abi->NameLength == lstrlenW(testAtom1) * sizeof(WCHAR), "wrong string length\n"); |
| ok(abi->Name[lstrlenW(testAtom1)] == 0, "wrong string termination %x\n", abi->Name[lstrlenW(testAtom1)]); |
| ok(abi->Name[lstrlenW(testAtom1) + 1] == 0xcccc, "buffer overwrite %x\n", abi->Name[lstrlenW(testAtom1) + 1]); |
| |
| ptr_size = sizeof(ATOM_BASIC_INFORMATION); |
| res = pNtQueryInformationAtom( atom, AtomBasicInformation, (void*)ptr, ptr_size, NULL ); |
| ok(res == STATUS_BUFFER_TOO_SMALL, "wrong return status (%x)\n", res); |
| ok(abi->NameLength == lstrlenW(testAtom1) * sizeof(WCHAR) || broken(abi->NameLength == sizeof(WCHAR)), /* nt4 */ |
| "string length %u\n",abi->NameLength); |
| |
| memset( ptr, 0xcc, sizeof(ptr) ); |
| ptr_size = sizeof(ATOM_BASIC_INFORMATION) + lstrlenW(testAtom1) * sizeof(WCHAR); |
| res = pNtQueryInformationAtom( atom, AtomBasicInformation, (void*)ptr, ptr_size, NULL ); |
| ok(!res, "atom lookup %x\n", res); |
| ok(!lstrcmpW(abi->Name, testAtom1), "strings don't match\n"); |
| ok(abi->NameLength == lstrlenW(testAtom1) * sizeof(WCHAR), "wrong string length\n"); |
| ok(abi->Name[lstrlenW(testAtom1)] == 0, "buffer overwrite %x\n", abi->Name[lstrlenW(testAtom1)]); |
| ok(abi->Name[lstrlenW(testAtom1) + 1] == 0xcccc, "buffer overwrite %x\n", abi->Name[lstrlenW(testAtom1) + 1]); |
| |
| memset( ptr, 0xcc, sizeof(ptr) ); |
| ptr_size = sizeof(ATOM_BASIC_INFORMATION) + 4 * sizeof(WCHAR); |
| res = pNtQueryInformationAtom( atom, AtomBasicInformation, (void*)ptr, ptr_size, NULL ); |
| ok(!res, "couldn't find atom\n"); |
| ok(abi->NameLength == 8, "wrong string length %u\n", abi->NameLength); |
| ok(!memcmp(abi->Name, testAtom1, 8), "strings don't match\n"); |
| } |
| |
| START_TEST(atom) |
| { |
| InitFunctionPtr(); |
| if (pRtlCreateAtomTable) |
| { |
| /* Global atom table seems to be available to GUI apps only in |
| Win7, so let's turn this app into a GUI app */ |
| GetDesktopWindow(); |
| |
| test_NtAtom(); |
| test_NtIntAtom(); |
| test_NtRefPinAtom(); |
| test_Global(); |
| } |
| else |
| win_skip("Needed atom functions are not available\n"); |
| |
| FreeLibrary(hntdll); |
| } |
