+++ /dev/null
-/*
- * Unit tests for BroadcastSystemMessage
- *
- * Copyright 2008 Maarten Lankhorst
- *
- * 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
- */
-
-//#define _WIN32_WINNT 0x0501
-
-#include <stdarg.h>
-#include <stdio.h>
-
-#include "windef.h"
-#include "winbase.h"
-#include "wingdi.h"
-#include "winuser.h"
-#include "winnls.h"
-
-#include "wine/test.h"
-
-typedef LONG (WINAPI *PBROADCAST)( DWORD,LPDWORD,UINT,WPARAM,LPARAM );
-typedef LONG (WINAPI *PBROADCASTEX)( DWORD,LPDWORD,UINT,WPARAM,LPARAM,PBSMINFO );
-static PBROADCAST pBroadcastA;
-static PBROADCAST pBroadcastW;
-static PBROADCASTEX pBroadcastExA;
-static PBROADCASTEX pBroadcastExW;
-static HANDLE hevent;
-
-static LRESULT WINAPI main_window_procA(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
-{
- if (msg == WM_NULL)
- {
- trace("main_window_procA: Sleeping for %lu ms\n", wparam);
- if (wparam)
- {
- if (WaitForSingleObject(hevent, wparam) == WAIT_TIMEOUT)
- SetEvent(hevent);
- }
- trace("main_window_procA: Returning WM_NULL with parameter %08lx\n", lparam);
- return lparam;
- }
-
- return DefWindowProcA(hwnd, msg, wparam, lparam);
-}
-
-static BOOL init_procs(void)
-{
- WNDCLASSA cls;
- HANDLE user32 = GetModuleHandleA("user32.dll");
- pBroadcastA = (PBROADCAST)GetProcAddress(user32, "BroadcastSystemMessageA");
- if (!pBroadcastA)
- pBroadcastA = (PBROADCAST)GetProcAddress(user32, "BroadcastSystemMessage");
- ok(pBroadcastA != NULL, "No BroadcastSystemMessage found\n");
- if (!pBroadcastA)
- {
- win_skip("BroadcastA is not available\n");
- return FALSE;
- }
-
- pBroadcastW = (PBROADCAST)GetProcAddress(user32, "BroadcastSystemMessageW");
- pBroadcastExA = (PBROADCASTEX)GetProcAddress(user32, "BroadcastSystemMessageExA");
- pBroadcastExW = (PBROADCASTEX)GetProcAddress(user32, "BroadcastSystemMessageExW");
-
- hevent = CreateEventA(NULL, TRUE, FALSE, "Asynchronous checking event");
-
- cls.style = CS_DBLCLKS;
- cls.lpfnWndProc = main_window_procA;
- cls.cbClsExtra = 0;
- cls.cbWndExtra = 0;
- cls.hInstance = GetModuleHandleA(0);
- cls.hIcon = 0;
- cls.hCursor = LoadCursorA(0, (LPCSTR)IDC_ARROW);
- cls.hbrBackground = GetStockObject(WHITE_BRUSH);
- cls.lpszMenuName = NULL;
- cls.lpszClassName = "MainWindowClass";
-
- if (!RegisterClassA(&cls))
- return FALSE;
-
- if (!CreateWindowExA(0, "MainWindowClass", "Main window", WS_CAPTION | WS_SYSMENU |
- WS_MINIMIZEBOX | WS_MAXIMIZEBOX | WS_POPUP, 100, 100, 200,
- 200, 0, 0, GetModuleHandleA(NULL), NULL))
- return FALSE;
- return TRUE;
-}
-
-static void test_parameters(PBROADCAST broadcast, const char *functionname)
-{
- LONG ret;
- DWORD recips;
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcast( 0x80000000, &recips, WM_NULL, 0, 0 );
- if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
- {
- win_skip("%s is not implemented\n", functionname);
- return;
- }
- ok(!ret || broken(ret), "Returned: %d\n", ret);
- if (!ret) ok(GetLastError() == ERROR_INVALID_PARAMETER, "Last error: %08x\n", GetLastError());
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcast( 0x80000000, &recips, WM_NULL, 0, 0 );
- ok(!ret || broken(ret), "Returned: %d\n", ret);
- if (!ret) ok(GetLastError() == ERROR_INVALID_PARAMETER, "Last error: %08x\n", GetLastError());
-
-if (0) /* TODO: Check the hang flags */
-{
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcast( BSF_QUERY|(BSF_NOHANG|BSF_FORCEIFHUNG), &recips, WM_NULL, 30000, 0 );
- ok(0, "Last error: %08x\n", GetLastError());
- ok(0, "Returned: %d\n", ret);
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcast( BSF_QUERY|(BSF_NOHANG|BSF_NOTIMEOUTIFNOTHUNG), &recips, WM_NULL, 30000, 0 );
- ok(0, "Last error: %08x\n", GetLastError());
- ok(0, "Returned: %d\n", ret);
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcast( BSF_QUERY|(BSF_NOTIMEOUTIFNOTHUNG|BSF_FORCEIFHUNG), &recips, WM_NULL, 30000, 0 );
- ok(0, "Last error: %08x\n", GetLastError());
- ok(0, "Returned: %d\n", ret);
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcast( BSF_POSTMESSAGE|(BSF_NOTIMEOUTIFNOTHUNG|BSF_FORCEIFHUNG), &recips, WM_NULL, 30000, 0 );
- ok(0, "Last error: %08x\n", GetLastError());
- ok(0, "Returned: %d\n", ret);
-}
-
- recips = BSM_APPLICATIONS;
- ResetEvent(hevent);
- ret = broadcast( BSF_POSTMESSAGE|BSF_QUERY, &recips, WM_NULL, 100, 0 );
- ok(ret==1, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- PulseEvent(hevent);
-
- SetLastError( 0xdeadbeef );
- recips = BSM_APPLICATIONS;
- ret = broadcast( BSF_POSTMESSAGE|BSF_SENDNOTIFYMESSAGE, &recips, WM_NULL, 100, 0 );
- if (ret)
- {
- ok(ret==1, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_OBJECT_0, "Synchronous message sent instead\n");
- PulseEvent(hevent);
-
- recips = BSM_APPLICATIONS;
- ret = broadcast( BSF_SENDNOTIFYMESSAGE, &recips, WM_NULL, 100, BROADCAST_QUERY_DENY );
- ok(ret==1, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- PulseEvent(hevent);
-
- recips = BSM_APPLICATIONS;
- ret = broadcast( BSF_SENDNOTIFYMESSAGE|BSF_QUERY, &recips, WM_NULL, 100, BROADCAST_QUERY_DENY );
- ok(!ret, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- PulseEvent(hevent);
- }
- else /* BSF_SENDNOTIFYMESSAGE not supported on NT4 */
- ok( GetLastError() == ERROR_INVALID_PARAMETER, "failed with err %u\n", GetLastError() );
-
- recips = BSM_APPLICATIONS;
- ret = broadcast( 0, &recips, WM_NULL, 100, 0 );
- ok(ret==1, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- PulseEvent(hevent);
-}
-
-/* BSF_SENDNOTIFYMESSAGE and BSF_QUERY are both synchronous within the same process
- * However you should be able to distinguish them by sending the BROADCAST_QUERY_DENY flag
- */
-
-static void test_parametersEx(PBROADCASTEX broadcastex)
-{
- LONG ret;
- DWORD recips;
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcastex( 0x80000000, &recips, WM_NULL, 0, 0, NULL );
- ok(GetLastError() == ERROR_INVALID_PARAMETER, "Last error: %08x\n", GetLastError());
- ok(!ret, "Returned: %d\n", ret);
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcastex( 0x80000000, &recips, WM_NULL, 0, 0, NULL );
- ok(GetLastError() == ERROR_INVALID_PARAMETER, "Last error: %08x\n", GetLastError());
- ok(!ret, "Returned: %d\n", ret);
-
-if (0) /* TODO: Check the hang flags */
-{
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcastex( BSF_QUERY|(BSF_NOHANG|BSF_FORCEIFHUNG), &recips, WM_NULL, 30000, 0, NULL );
- ok(0, "Last error: %08x\n", GetLastError());
- ok(0, "Returned: %d\n", ret);
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcastex( BSF_QUERY|(BSF_NOHANG|BSF_NOTIMEOUTIFNOTHUNG), &recips, WM_NULL, 30000, 0, NULL );
- ok(0, "Last error: %08x\n", GetLastError());
- ok(0, "Returned: %d\n", ret);
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcastex( BSF_QUERY|(BSF_NOTIMEOUTIFNOTHUNG|BSF_FORCEIFHUNG), &recips, WM_NULL, 30000, 0, NULL );
- ok(0, "Last error: %08x\n", GetLastError());
- ok(0, "Returned: %d\n", ret);
-
- SetLastError(0xcafebabe);
- recips = BSM_APPLICATIONS;
- ret = broadcastex( BSF_POSTMESSAGE|(BSF_NOTIMEOUTIFNOTHUNG|BSF_FORCEIFHUNG), &recips, WM_NULL, 30000, 0, NULL );
- ok(0, "Last error: %08x\n", GetLastError());
- ok(0, "Returned: %d\n", ret);
-}
-
- recips = BSM_APPLICATIONS;
- ResetEvent(hevent);
- ret = broadcastex( BSF_POSTMESSAGE|BSF_QUERY, &recips, WM_NULL, 100, 0, NULL );
- ok(ret==1, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- PulseEvent(hevent);
-
- recips = BSM_APPLICATIONS;
- ret = broadcastex( BSF_POSTMESSAGE|BSF_SENDNOTIFYMESSAGE, &recips, WM_NULL, 100, 0, NULL );
- ok(ret==1, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_OBJECT_0, "Synchronous message sent instead\n");
- PulseEvent(hevent);
-
- recips = BSM_APPLICATIONS;
- ret = broadcastex( BSF_SENDNOTIFYMESSAGE, &recips, WM_NULL, 100, BROADCAST_QUERY_DENY, NULL );
- ok(ret==1, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- PulseEvent(hevent);
-
- recips = BSM_APPLICATIONS;
- ret = broadcastex( BSF_SENDNOTIFYMESSAGE|BSF_QUERY, &recips, WM_NULL, 100, BROADCAST_QUERY_DENY, NULL );
- ok(!ret, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- PulseEvent(hevent);
-
- recips = BSM_APPLICATIONS;
- ret = broadcastex( 0, &recips, WM_NULL, 100, 0, NULL );
- ok(ret==1, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- PulseEvent(hevent);
-}
-
-static BOOL (WINAPI *pOpenProcessToken)(HANDLE, DWORD, HANDLE*);
-static BOOL (WINAPI *pAdjustTokenPrivileges)(HANDLE, BOOL, PTOKEN_PRIVILEGES, DWORD, PTOKEN_PRIVILEGES, PDWORD);
-
-static void test_noprivileges(void)
-{
- HANDLE advapi32 = GetModuleHandleA("advapi32");
- HANDLE token;
- DWORD recips;
- BOOL ret;
-
- static const DWORD BSM_ALL_RECIPS = BSM_VXDS | BSM_NETDRIVER |
- BSM_INSTALLABLEDRIVERS | BSM_APPLICATIONS;
-
- pOpenProcessToken = (void *)GetProcAddress(advapi32, "OpenProcessToken");
- pAdjustTokenPrivileges = (void *)GetProcAddress(advapi32, "AdjustTokenPrivileges");
- if (!pOpenProcessToken || !pAdjustTokenPrivileges || !pOpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &token))
- {
- skip("Can't open security token for process\n");
- return;
- }
- if (!pAdjustTokenPrivileges(token, TRUE, NULL, 0, NULL, NULL))
- {
- skip("Can't adjust security token for process\n");
- return;
- }
-
- trace("Trying privileged edition!\n");
- SetLastError(0xcafebabe);
- recips = BSM_ALLDESKTOPS;
- ResetEvent(hevent);
- ret = pBroadcastExW( BSF_QUERY, &recips, WM_NULL, 100, 0, NULL );
- ok(ret==1, "Returned: %d error %u\n", ret, GetLastError());
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- ok(recips == BSM_ALLDESKTOPS ||
- recips == BSM_ALL_RECIPS, /* win2k3 */
- "Received by: %08x\n", recips);
- PulseEvent(hevent);
-
- SetLastError(0xcafebabe);
- recips = BSM_ALLCOMPONENTS;
- ResetEvent(hevent);
- ret = pBroadcastExW( BSF_QUERY, &recips, WM_NULL, 100, 0, NULL );
- ok(ret==1, "Returned: %d error %u\n", ret, GetLastError());
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- ok(recips == BSM_ALLCOMPONENTS ||
- recips == BSM_ALL_RECIPS, /* win2k3 */
- "Received by: %08x\n", recips);
- PulseEvent(hevent);
-
- SetLastError(0xcafebabe);
- recips = BSM_ALLDESKTOPS|BSM_APPLICATIONS;
- ResetEvent(hevent);
- ret = pBroadcastExW( BSF_QUERY, &recips, WM_NULL, 100, 0, NULL );
- ok(ret==1, "Returned: %d error %u\n", ret, GetLastError());
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- ok(recips == (BSM_ALLDESKTOPS|BSM_APPLICATIONS) ||
- recips == BSM_APPLICATIONS, /* win2k3 */
- "Received by: %08x\n", recips);
- PulseEvent(hevent);
-
- SetLastError(0xcafebabe);
- recips = BSM_ALLDESKTOPS|BSM_APPLICATIONS;
- ResetEvent(hevent);
- ret = pBroadcastExW( BSF_QUERY, &recips, WM_NULL, 100, BROADCAST_QUERY_DENY, NULL );
- ok(!ret, "Returned: %d\n", ret);
- ok(WaitForSingleObject(hevent, 0) != WAIT_TIMEOUT, "Asynchronous message sent instead\n");
- ok(recips == (BSM_ALLDESKTOPS|BSM_APPLICATIONS) ||
- recips == BSM_APPLICATIONS, /* win2k3 */
- "Received by: %08x\n", recips);
- PulseEvent(hevent);
-}
-
-START_TEST(broadcast)
-{
- if (!init_procs())
- return;
-
- trace("Running BroadcastSystemMessageA tests\n");
- test_parameters(pBroadcastA, "BroadcastSystemMessageA");
- if (pBroadcastW)
- {
- trace("Running BroadcastSystemMessageW tests\n");
- test_parameters(pBroadcastW, "BroadcastSystemMessageW");
- }
- else
- win_skip("No BroadcastSystemMessageW, skipping\n");
- if (pBroadcastExA)
- {
- trace("Running BroadcastSystemMessageExA tests\n");
- test_parametersEx(pBroadcastExA);
- }
- else
- win_skip("No BroadcastSystemMessageExA, skipping\n");
- if (pBroadcastExW)
- {
- trace("Running BroadcastSystemMessageExW tests\n");
- test_parametersEx(pBroadcastExW);
- trace("Attempting privileges checking tests\n");
- test_noprivileges();
- }
- else
- win_skip("No BroadcastSystemMessageExW, skipping\n");
-}
projects / reactos.git / blobdiff