215 lines
7.8 KiB
C
215 lines
7.8 KiB
C
/*
|
|
* ATL test program
|
|
*
|
|
* Copyright 2010 Marcus Meissner
|
|
*
|
|
*
|
|
* 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
|
|
*/
|
|
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
|
|
#define COBJMACROS
|
|
|
|
#include <atlbase.h>
|
|
#include <atlwin.h>
|
|
|
|
#include <wine/test.h>
|
|
|
|
#define MAXSIZE 512
|
|
static void test_StructSize(void)
|
|
{
|
|
_ATL_MODULEW *tst;
|
|
HRESULT hres;
|
|
int i;
|
|
|
|
tst = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, MAXSIZE);
|
|
|
|
for (i=0;i<MAXSIZE;i++) {
|
|
tst->cbSize = i;
|
|
hres = AtlModuleInit(tst, NULL, NULL);
|
|
|
|
switch (i) {
|
|
case FIELD_OFFSET(_ATL_MODULEW, dwAtlBuildVer):
|
|
case sizeof(_ATL_MODULEW):
|
|
#ifdef _WIN64
|
|
case sizeof(_ATL_MODULEW) + sizeof(void *):
|
|
#endif
|
|
ok (hres == S_OK, "AtlModuleInit with %d failed (0x%x).\n", i, (int)hres);
|
|
break;
|
|
default:
|
|
ok (FAILED(hres), "AtlModuleInit with %d succeeded? (0x%x).\n", i, (int)hres);
|
|
break;
|
|
}
|
|
}
|
|
|
|
HeapFree (GetProcessHeap(), 0, tst);
|
|
}
|
|
|
|
static void test_winmodule(void)
|
|
{
|
|
_AtlCreateWndData create_data[3];
|
|
_ATL_MODULEW winmod;
|
|
void *p;
|
|
HRESULT hres;
|
|
|
|
winmod.cbSize = sizeof(winmod);
|
|
winmod.m_pCreateWndList = (void*)0xdeadbeef;
|
|
winmod.m_csWindowCreate.LockCount = 0xdeadbeef;
|
|
hres = AtlModuleInit(&winmod, NULL, NULL);
|
|
ok(hres == S_OK, "AtlModuleInit failed: %08x\n", hres);
|
|
ok(!winmod.m_pCreateWndList, "winmod.m_pCreateWndList = %p\n", winmod.m_pCreateWndList);
|
|
ok(winmod.m_csWindowCreate.LockCount == -1, "winmod.m_csWindowCreate.LockCount = %d\n",
|
|
winmod.m_csWindowCreate.LockCount);
|
|
|
|
AtlModuleAddCreateWndData(&winmod, create_data, (void*)0xdead0001);
|
|
ok(winmod.m_pCreateWndList == create_data, "winmod.m_pCreateWndList != create_data\n");
|
|
ok(create_data[0].m_pThis == (void*)0xdead0001, "unexpected create_data[0].m_pThis %p\n", create_data[0].m_pThis);
|
|
ok(create_data[0].m_dwThreadID == GetCurrentThreadId(), "unexpected create_data[0].m_dwThreadID %x\n",
|
|
create_data[0].m_dwThreadID);
|
|
ok(!create_data[0].m_pNext, "unexpected create_data[0].m_pNext %p\n", create_data[0].m_pNext);
|
|
|
|
AtlModuleAddCreateWndData(&winmod, create_data+1, (void*)0xdead0002);
|
|
ok(winmod.m_pCreateWndList == create_data+1, "winmod.m_pCreateWndList != create_data\n");
|
|
ok(create_data[1].m_pThis == (void*)0xdead0002, "unexpected create_data[1].m_pThis %p\n", create_data[1].m_pThis);
|
|
ok(create_data[1].m_dwThreadID == GetCurrentThreadId(), "unexpected create_data[1].m_dwThreadID %x\n",
|
|
create_data[1].m_dwThreadID);
|
|
ok(create_data[1].m_pNext == create_data, "unexpected create_data[1].m_pNext %p\n", create_data[1].m_pNext);
|
|
|
|
AtlModuleAddCreateWndData(&winmod, create_data+2, (void*)0xdead0003);
|
|
ok(winmod.m_pCreateWndList == create_data+2, "winmod.m_pCreateWndList != create_data\n");
|
|
ok(create_data[2].m_pThis == (void*)0xdead0003, "unexpected create_data[2].m_pThis %p\n", create_data[2].m_pThis);
|
|
ok(create_data[2].m_dwThreadID == GetCurrentThreadId(), "unexpected create_data[2].m_dwThreadID %x\n",
|
|
create_data[2].m_dwThreadID);
|
|
ok(create_data[2].m_pNext == create_data+1, "unexpected create_data[2].m_pNext %p\n", create_data[2].m_pNext);
|
|
|
|
p = AtlModuleExtractCreateWndData(&winmod);
|
|
ok(p == (void*)0xdead0003, "unexpected AtlModuleExtractCreateWndData result %p\n", p);
|
|
ok(winmod.m_pCreateWndList == create_data+1, "winmod.m_pCreateWndList != create_data\n");
|
|
ok(create_data[2].m_pNext == create_data+1, "unexpected create_data[2].m_pNext %p\n", create_data[2].m_pNext);
|
|
|
|
create_data[1].m_dwThreadID = 0xdeadbeef;
|
|
|
|
p = AtlModuleExtractCreateWndData(&winmod);
|
|
ok(p == (void*)0xdead0001, "unexpected AtlModuleExtractCreateWndData result %p\n", p);
|
|
ok(winmod.m_pCreateWndList == create_data+1, "winmod.m_pCreateWndList != create_data\n");
|
|
ok(!create_data[0].m_pNext, "unexpected create_data[0].m_pNext %p\n", create_data[0].m_pNext);
|
|
ok(!create_data[1].m_pNext, "unexpected create_data[1].m_pNext %p\n", create_data[1].m_pNext);
|
|
|
|
p = AtlModuleExtractCreateWndData(&winmod);
|
|
ok(!p, "unexpected AtlModuleExtractCreateWndData result %p\n", p);
|
|
ok(winmod.m_pCreateWndList == create_data+1, "winmod.m_pCreateWndList != create_data\n");
|
|
}
|
|
|
|
static void test_winclassinfo(void)
|
|
{
|
|
_ATL_MODULEW winmod;
|
|
HRESULT hres;
|
|
int len, expectedLen;
|
|
ATOM atom;
|
|
WNDPROC wndProc;
|
|
_ATL_WNDCLASSINFOW wci =
|
|
{
|
|
/* .m_wc = */
|
|
{
|
|
sizeof(WNDCLASSEXW),
|
|
CS_VREDRAW | CS_HREDRAW,
|
|
DefWindowProcW,
|
|
0,
|
|
0,
|
|
NULL,
|
|
NULL,
|
|
LoadCursorW(NULL, (LPCWSTR)IDC_ARROW),
|
|
(HBRUSH)(COLOR_BTNFACE + 1),
|
|
NULL,
|
|
NULL, /* LPCSTR lpszClassName; <-- We force ATL class name generation */
|
|
NULL
|
|
},
|
|
/* .m_lpszOrigName = */ NULL,
|
|
/* .pWndProc = */ NULL,
|
|
/* .m_lpszCursorID = */ (LPCWSTR)IDC_ARROW,
|
|
/* .m_bSystemCursor = */ TRUE,
|
|
/* .m_atom = */ 0,
|
|
/* .m_szAutoName = */ L""
|
|
};
|
|
|
|
winmod.cbSize = sizeof(winmod);
|
|
winmod.m_pCreateWndList = (void*)0xdeadbeef;
|
|
hres = AtlModuleInit(&winmod, NULL, NULL);
|
|
ok(hres == S_OK, "AtlModuleInit failed: %08x\n", hres);
|
|
ok(!winmod.m_pCreateWndList, "winmod.m_pCreateWndList = %p\n", winmod.m_pCreateWndList);
|
|
|
|
atom = AtlModuleRegisterWndClassInfoW(&winmod, &wci, &wndProc);
|
|
ok(atom, "AtlModuleRegisterWndClassInfoA failed: %08x\n", atom);
|
|
ok(atom == wci.m_atom, "(atom = %08x) is != than (wci.m_atom = %08x)\n", atom, wci.m_atom);
|
|
|
|
ok(wcsncmp(wci.m_szAutoName, L"ATL:", 4) == 0, "wci.m_szAutoName = '%ls', expected starting with 'ATL:'\n", wci.m_szAutoName);
|
|
|
|
len = wcslen(wci.m_szAutoName);
|
|
expectedLen = sizeof("ATL:") + sizeof(void *) * 2 - 1;
|
|
ok(len == expectedLen, "wci.m_szAutoName has length %d, expected length %d\n", len, expectedLen);
|
|
}
|
|
|
|
static DWORD_PTR cb_val;
|
|
|
|
static void WINAPI term_callback(DWORD_PTR dw)
|
|
{
|
|
cb_val = dw;
|
|
}
|
|
|
|
static void test_term(void)
|
|
{
|
|
_ATL_MODULEW test;
|
|
ULONG_PTR ex;
|
|
HRESULT hres;
|
|
|
|
ex = (ULONG_PTR)-37;
|
|
|
|
test.cbSize = sizeof(_ATL_MODULEW);
|
|
|
|
hres = AtlModuleInit(&test, NULL, NULL);
|
|
ok (hres == S_OK, "AtlModuleInit failed (0x%x).\n", hres);
|
|
|
|
hres = AtlModuleAddTermFunc(&test, term_callback, ex);
|
|
ok (hres == S_OK, "AtlModuleAddTermFunc failed (0x%x).\n", hres);
|
|
|
|
cb_val = 0xdeadbeef;
|
|
hres = AtlModuleTerm(&test);
|
|
ok (hres == S_OK, "AtlModuleTerm failed (0x%x).\n", hres);
|
|
ok (cb_val == ex, "wrong callback value (0x%lx).\n", cb_val);
|
|
|
|
test.cbSize = FIELD_OFFSET(_ATL_MODULEW, dwAtlBuildVer);
|
|
|
|
hres = AtlModuleInit(&test, NULL, NULL);
|
|
ok (hres == S_OK, "AtlModuleInit failed (0x%x).\n", hres);
|
|
|
|
hres = AtlModuleAddTermFunc(&test, term_callback, 0x23);
|
|
ok (hres == S_OK, "AtlModuleAddTermFunc failed (0x%x).\n", hres);
|
|
|
|
cb_val = 0xdeadbeef;
|
|
hres = AtlModuleTerm(&test);
|
|
ok (hres == S_OK, "AtlModuleTerm failed (0x%x).\n", hres);
|
|
ok (cb_val == 0xdeadbeef, "wrong callback value (0x%lx).\n", cb_val);
|
|
}
|
|
|
|
START_TEST(module)
|
|
{
|
|
test_StructSize();
|
|
test_winmodule();
|
|
test_winclassinfo();
|
|
test_term();
|
|
}
|