| /* |
| * Copyright 2014 Alistair Leslie-Hughes |
| * |
| * 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_LEAN_AND_MEAN |
| #include <stdio.h> |
| |
| #include <dplay8.h> |
| #define COBJMACROS |
| #include <netfw.h> |
| #include "wine/test.h" |
| |
| #include "dpnet_test.h" |
| |
| /* {CD0C3D4B-E15E-4CF2-9EA8-6E1D6548C5A5} */ |
| static const GUID appguid = { 0xcd0c3d4b, 0xe15e, 0x4cf2, { 0x9e, 0xa8, 0x6e, 0x1d, 0x65, 0x48, 0xc5, 0xa5 } }; |
| static WCHAR sessionname[] = {'w','i','n','e','g','a','m','e','s','s','e','r','v','e','r',0}; |
| |
| static BOOL nCreatePlayer; |
| static BOOL nDestroyPlayer; |
| |
| static HRESULT WINAPI DirectPlayMessageHandler(PVOID pvUserContext, DWORD dwMessageId, PVOID pMsgBuffer) |
| { |
| trace("msgid: 0x%08x\n", dwMessageId); |
| |
| switch(dwMessageId) |
| { |
| case DPN_MSGID_CREATE_PLAYER: |
| nCreatePlayer = TRUE; |
| break; |
| case DPN_MSGID_DESTROY_PLAYER: |
| nDestroyPlayer = TRUE; |
| break; |
| } |
| |
| return S_OK; |
| } |
| |
| static void create_server(void) |
| { |
| HRESULT hr; |
| IDirectPlay8Server *server = NULL; |
| |
| hr = CoCreateInstance( &CLSID_DirectPlay8Server, NULL, CLSCTX_ALL, &IID_IDirectPlay8Server, (LPVOID*)&server); |
| ok(hr == S_OK, "Failed to create IDirectPlay8Server object\n"); |
| if( SUCCEEDED(hr) ) |
| { |
| hr = IDirectPlay8Server_Close(server, 0); |
| todo_wine ok(hr == DPNERR_UNINITIALIZED, "got 0x%08x\n", hr); |
| |
| hr = IDirectPlay8Server_Initialize(server, NULL, NULL, 0); |
| ok(hr == DPNERR_INVALIDPARAM, "got 0x%08x\n", hr); |
| |
| hr = IDirectPlay8Server_Initialize(server, NULL, DirectPlayMessageHandler, 0); |
| ok(hr == S_OK, "got 0x%08x\n", hr); |
| if(hr == S_OK) |
| { |
| IDirectPlay8Address *localaddr = NULL; |
| DPN_APPLICATION_DESC appdesc; |
| |
| hr = CoCreateInstance( &CLSID_DirectPlay8Address, NULL, CLSCTX_ALL, &IID_IDirectPlay8Address, (LPVOID*)&localaddr); |
| ok(hr == S_OK, "Failed to create IDirectPlay8Address object\n"); |
| |
| hr = IDirectPlay8Address_SetSP(localaddr, &CLSID_DP8SP_TCPIP); |
| ok(hr == S_OK, "got 0x%08x\n", hr); |
| |
| memset( &appdesc, 0, sizeof(DPN_APPLICATION_DESC) ); |
| appdesc.dwSize = sizeof( DPN_APPLICATION_DESC ); |
| appdesc.dwFlags = DPNSESSION_CLIENT_SERVER; |
| appdesc.guidApplication = appguid; |
| appdesc.pwszSessionName = sessionname; |
| |
| hr = IDirectPlay8Server_Host(server, &appdesc, &localaddr, 1, NULL, NULL, NULL, 0); |
| todo_wine ok(hr == S_OK, "got 0x%08x\n", hr); |
| |
| todo_wine ok(nCreatePlayer, "No DPN_MSGID_CREATE_PLAYER Message\n"); |
| ok(!nDestroyPlayer, "Received DPN_MSGID_DESTROY_PLAYER Message\n"); |
| |
| hr = IDirectPlay8Server_Close(server, 0); |
| todo_wine ok(hr == S_OK, "got 0x%08x\n", hr); |
| |
| todo_wine ok(nDestroyPlayer, "No DPN_MSGID_DESTROY_PLAYER Message\n"); |
| |
| IDirectPlay8Address_Release(localaddr); |
| } |
| |
| IDirectPlay8Server_Release(server); |
| } |
| } |
| |
| static void test_server_info(void) |
| { |
| HRESULT hr; |
| DPN_PLAYER_INFO info; |
| WCHAR name[] = {'w','i','n','e',0}; |
| WCHAR name2[] = {'w','i','n','e','2',0}; |
| WCHAR data[] = {'X','X','X','X',0}; |
| IDirectPlay8Server *server = NULL; |
| |
| hr = CoCreateInstance( &CLSID_DirectPlay8Server, NULL, CLSCTX_ALL, &IID_IDirectPlay8Server, (LPVOID*)&server); |
| ok(hr == S_OK, "Failed to create IDirectPlay8Server object\n"); |
| if( SUCCEEDED(hr) ) |
| { |
| ZeroMemory( &info, sizeof(DPN_PLAYER_INFO) ); |
| info.dwSize = sizeof(DPN_PLAYER_INFO); |
| info.dwInfoFlags = DPNINFO_NAME; |
| |
| hr = IDirectPlay8Server_SetServerInfo(server, NULL, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == E_POINTER, "got %x\n", hr); |
| |
| info.pwszName = name; |
| hr = IDirectPlay8Server_SetServerInfo(server, &info, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == DPNERR_UNINITIALIZED, "got %x\n", hr); |
| |
| hr = IDirectPlay8Server_Initialize(server, NULL, DirectPlayMessageHandler, 0); |
| ok(hr == S_OK, "got 0x%08x\n", hr); |
| |
| hr = IDirectPlay8Server_SetServerInfo(server, NULL, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == E_POINTER, "got %x\n", hr); |
| |
| info.pwszName = NULL; |
| hr = IDirectPlay8Server_SetServerInfo(server, &info, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == S_OK, "got %x\n", hr); |
| |
| info.pwszName = name; |
| hr = IDirectPlay8Server_SetServerInfo(server, &info, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == S_OK, "got %x\n", hr); |
| |
| info.dwInfoFlags = DPNINFO_NAME; |
| info.pwszName = name2; |
| hr = IDirectPlay8Server_SetServerInfo(server, &info, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == S_OK, "got %x\n", hr); |
| |
| info.dwInfoFlags = DPNINFO_DATA; |
| info.pwszName = NULL; |
| info.pvData = NULL; |
| info.dwDataSize = sizeof(data); |
| hr = IDirectPlay8Server_SetServerInfo(server, &info, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == E_POINTER, "got %x\n", hr); |
| |
| info.dwInfoFlags = DPNINFO_DATA; |
| info.pwszName = NULL; |
| info.pvData = data; |
| info.dwDataSize = 0; |
| hr = IDirectPlay8Server_SetServerInfo(server, &info, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == S_OK, "got %x\n", hr); |
| |
| info.dwInfoFlags = DPNINFO_DATA; |
| info.pwszName = NULL; |
| info.pvData = data; |
| info.dwDataSize = sizeof(data); |
| hr = IDirectPlay8Server_SetServerInfo(server, &info, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == S_OK, "got %x\n", hr); |
| |
| info.dwInfoFlags = DPNINFO_DATA | DPNINFO_NAME; |
| info.pwszName = name; |
| info.pvData = data; |
| info.dwDataSize = sizeof(data); |
| hr = IDirectPlay8Server_SetServerInfo(server, &info, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == S_OK, "got %x\n", hr); |
| |
| info.dwInfoFlags = DPNINFO_DATA | DPNINFO_NAME; |
| info.pwszName = name; |
| info.pvData = NULL; |
| info.dwDataSize = 0; |
| hr = IDirectPlay8Server_SetServerInfo(server, &info, NULL, NULL, DPNSETSERVERINFO_SYNC); |
| ok(hr == S_OK, "got %x\n", hr); |
| |
| IDirectPlay8Server_Release(server); |
| } |
| } |
| |
| BOOL is_process_elevated(void) |
| { |
| HANDLE token; |
| if (OpenProcessToken( GetCurrentProcess(), TOKEN_QUERY, &token )) |
| { |
| TOKEN_ELEVATION_TYPE type; |
| DWORD size; |
| BOOL ret; |
| |
| ret = GetTokenInformation( token, TokenElevationType, &type, sizeof(type), &size ); |
| CloseHandle( token ); |
| return (ret && type == TokenElevationTypeFull); |
| } |
| return FALSE; |
| } |
| |
| BOOL is_firewall_enabled(void) |
| { |
| HRESULT hr, init; |
| INetFwMgr *mgr = NULL; |
| INetFwPolicy *policy = NULL; |
| INetFwProfile *profile = NULL; |
| VARIANT_BOOL enabled = VARIANT_FALSE; |
| |
| init = CoInitializeEx( 0, COINIT_APARTMENTTHREADED ); |
| |
| hr = CoCreateInstance( &CLSID_NetFwMgr, NULL, CLSCTX_INPROC_SERVER, &IID_INetFwMgr, |
| (void **)&mgr ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| if (hr != S_OK) goto done; |
| |
| hr = INetFwMgr_get_LocalPolicy( mgr, &policy ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| if (hr != S_OK) goto done; |
| |
| hr = INetFwPolicy_get_CurrentProfile( policy, &profile ); |
| if (hr != S_OK) goto done; |
| |
| hr = INetFwProfile_get_FirewallEnabled( profile, &enabled ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| |
| done: |
| if (policy) INetFwPolicy_Release( policy ); |
| if (profile) INetFwProfile_Release( profile ); |
| if (mgr) INetFwMgr_Release( mgr ); |
| if (SUCCEEDED( init )) CoUninitialize(); |
| return (enabled == VARIANT_TRUE); |
| } |
| |
| HRESULT set_firewall( enum firewall_op op ) |
| { |
| static const WCHAR dpnsvrW[] = |
| {'d','p','n','s','v','r','.','e','x','e',0}; |
| static const WCHAR separator[] = {'\\',0}; |
| static const WCHAR clientW[] = |
| {'d','p','n','e','t','_','c','l','i','e','n','t',0}; |
| static const WCHAR serverW[] = |
| {'d','p','n','e','t','_','s','e','r','v','e','r',0}; |
| HRESULT hr, init; |
| INetFwMgr *mgr = NULL; |
| INetFwPolicy *policy = NULL; |
| INetFwProfile *profile = NULL; |
| INetFwAuthorizedApplication *app = NULL; |
| INetFwAuthorizedApplications *apps = NULL; |
| BSTR name, image = SysAllocStringLen( NULL, MAX_PATH ); |
| WCHAR path[MAX_PATH]; |
| |
| if (!GetModuleFileNameW( NULL, image, MAX_PATH )) |
| { |
| SysFreeString( image ); |
| return E_FAIL; |
| } |
| |
| if(!GetSystemDirectoryW(path, MAX_PATH)) |
| { |
| SysFreeString( image ); |
| return E_FAIL; |
| } |
| lstrcatW(path, separator); |
| lstrcatW(path, dpnsvrW); |
| |
| init = CoInitializeEx( 0, COINIT_APARTMENTTHREADED ); |
| |
| hr = CoCreateInstance( &CLSID_NetFwMgr, NULL, CLSCTX_INPROC_SERVER, &IID_INetFwMgr, |
| (void **)&mgr ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| if (hr != S_OK) goto done; |
| |
| hr = INetFwMgr_get_LocalPolicy( mgr, &policy ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| if (hr != S_OK) goto done; |
| |
| hr = INetFwPolicy_get_CurrentProfile( policy, &profile ); |
| if (hr != S_OK) goto done; |
| |
| hr = INetFwProfile_get_AuthorizedApplications( profile, &apps ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| if (hr != S_OK) goto done; |
| |
| hr = CoCreateInstance( &CLSID_NetFwAuthorizedApplication, NULL, CLSCTX_INPROC_SERVER, |
| &IID_INetFwAuthorizedApplication, (void **)&app ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| if (hr != S_OK) goto done; |
| |
| hr = INetFwAuthorizedApplication_put_ProcessImageFileName( app, image ); |
| if (hr != S_OK) goto done; |
| |
| name = SysAllocString( clientW ); |
| hr = INetFwAuthorizedApplication_put_Name( app, name ); |
| SysFreeString( name ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| if (hr != S_OK) goto done; |
| |
| if (op == APP_ADD) |
| hr = INetFwAuthorizedApplications_Add( apps, app ); |
| else if (op == APP_REMOVE) |
| hr = INetFwAuthorizedApplications_Remove( apps, image ); |
| else |
| hr = E_INVALIDARG; |
| if (hr != S_OK) goto done; |
| |
| INetFwAuthorizedApplication_Release( app ); |
| hr = CoCreateInstance( &CLSID_NetFwAuthorizedApplication, NULL, CLSCTX_INPROC_SERVER, |
| &IID_INetFwAuthorizedApplication, (void **)&app ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| if (hr != S_OK) goto done; |
| |
| SysFreeString( image ); |
| image = SysAllocString( path ); |
| hr = INetFwAuthorizedApplication_put_ProcessImageFileName( app, image ); |
| if (hr != S_OK) goto done; |
| |
| name = SysAllocString( serverW ); |
| hr = INetFwAuthorizedApplication_put_Name( app, name ); |
| SysFreeString( name ); |
| ok( hr == S_OK, "got %08x\n", hr ); |
| if (hr != S_OK) goto done; |
| |
| if (op == APP_ADD) |
| hr = INetFwAuthorizedApplications_Add( apps, app ); |
| else if (op == APP_REMOVE) |
| hr = INetFwAuthorizedApplications_Remove( apps, image ); |
| else |
| hr = E_INVALIDARG; |
| |
| done: |
| if (app) INetFwAuthorizedApplication_Release( app ); |
| if (apps) INetFwAuthorizedApplications_Release( apps ); |
| if (policy) INetFwPolicy_Release( policy ); |
| if (profile) INetFwProfile_Release( profile ); |
| if (mgr) INetFwMgr_Release( mgr ); |
| if (SUCCEEDED( init )) CoUninitialize(); |
| SysFreeString( image ); |
| return hr; |
| } |
| |
| /* taken from programs/winetest/main.c */ |
| BOOL is_stub_dll(const char *filename) |
| { |
| DWORD size, ver; |
| BOOL isstub = FALSE; |
| char *p, *data; |
| |
| size = GetFileVersionInfoSizeA(filename, &ver); |
| if (!size) return FALSE; |
| |
| data = HeapAlloc(GetProcessHeap(), 0, size); |
| if (!data) return FALSE; |
| |
| if (GetFileVersionInfoA(filename, ver, size, data)) |
| { |
| char buf[256]; |
| |
| sprintf(buf, "\\StringFileInfo\\%04x%04x\\OriginalFilename", MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US), 1200); |
| if (VerQueryValueA(data, buf, (void**)&p, &size)) |
| isstub = !lstrcmpiA("wcodstub.dll", p); |
| } |
| HeapFree(GetProcessHeap(), 0, data); |
| |
| return isstub; |
| } |
| |
| START_TEST(server) |
| { |
| HRESULT hr; |
| BOOL firewall_enabled; |
| char path[MAX_PATH]; |
| |
| if(!GetSystemDirectoryA(path, MAX_PATH)) |
| { |
| skip("Failed to get systems directory\n"); |
| return; |
| } |
| strcat(path, "\\dpnet.dll"); |
| |
| if (!winetest_interactive && is_stub_dll(path)) |
| { |
| win_skip("dpnet is a stub dll, skipping tests\n"); |
| return; |
| } |
| |
| if ((firewall_enabled = is_firewall_enabled()) && !is_process_elevated()) |
| { |
| skip("no privileges, skipping tests to avoid firewall dialog\n"); |
| return; |
| } |
| |
| if (firewall_enabled) |
| { |
| HRESULT hr = set_firewall(APP_ADD); |
| if (hr != S_OK) |
| { |
| skip("can't authorize app in firewall %08x\n", hr); |
| return; |
| } |
| } |
| |
| hr = CoInitialize(0); |
| ok( hr == S_OK, "failed to init com\n"); |
| if (hr != S_OK) |
| goto done; |
| |
| create_server(); |
| test_server_info(); |
| |
| CoUninitialize(); |
| |
| done: |
| if (firewall_enabled) set_firewall(APP_REMOVE); |
| } |
