Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450/* * PROJECT: ReactOS api tests * LICENSE: GPLv2+ - See COPYING in the top level directory * PURPOSE: Test for service networking * PROGRAMMER: Pierre Schweitzer */
#include "precomp.h"
#include "svchlp.h"
#define WIN32_NO_STATUS#include <iphlpapi.h>#include <winsock2.h>
/*** Service part of the test ***/
static SERVICE_STATUS_HANDLE status_handle;
static voidreport_service_status(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwWaitHint){ BOOL res; SERVICE_STATUS status;
status.dwServiceType = SERVICE_WIN32_OWN_PROCESS; status.dwCurrentState = dwCurrentState; status.dwWin32ExitCode = dwWin32ExitCode; status.dwWaitHint = dwWaitHint;
status.dwServiceSpecificExitCode = 0; status.dwCheckPoint = 0;
if ( (dwCurrentState == SERVICE_START_PENDING) || (dwCurrentState == SERVICE_STOP_PENDING) || (dwCurrentState == SERVICE_STOPPED) ) { status.dwControlsAccepted = 0; } else { status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN; }
#if 0 if ( (dwCurrentState == SERVICE_RUNNING) || (dwCurrentState == SERVICE_STOPPED) ) status.dwCheckPoint = 0; else status.dwCheckPoint = dwCheckPoint++;#endif
res = SetServiceStatus(status_handle, &status); service_ok(res, "SetServiceStatus(%d) failed: %lu\n", dwCurrentState, GetLastError());}
static VOID WINAPI service_handler(DWORD ctrl){ switch(ctrl) { case SERVICE_CONTROL_STOP: case SERVICE_CONTROL_SHUTDOWN: report_service_status(SERVICE_STOP_PENDING, NO_ERROR, 0); default: report_service_status(SERVICE_RUNNING, NO_ERROR, 0); }}
static DWORD GetExtendedTcpTableWithAlloc(PVOID *TcpTable, BOOL Order, DWORD Family, TCP_TABLE_CLASS Class){ DWORD ret; DWORD Size = 0;
*TcpTable = NULL;
ret = GetExtendedTcpTable(*TcpTable, &Size, Order, Family, Class, 0); if (ret == ERROR_INSUFFICIENT_BUFFER) { *TcpTable = HeapAlloc(GetProcessHeap(), 0, Size); if (*TcpTable == NULL) { return ERROR_OUTOFMEMORY; }
ret = GetExtendedTcpTable(*TcpTable, &Size, Order, Family, Class, 0); if (ret != NO_ERROR) { HeapFree(GetProcessHeap(), 0, *TcpTable); *TcpTable = NULL; } }
return ret;}
static DWORD GetExtendedUdpTableWithAlloc(PVOID *UdpTable, BOOL Order, DWORD Family, UDP_TABLE_CLASS Class){ DWORD ret; DWORD Size = 0;
*UdpTable = NULL;
ret = GetExtendedUdpTable(*UdpTable, &Size, Order, Family, Class, 0); if (ret == ERROR_INSUFFICIENT_BUFFER) { *UdpTable = HeapAlloc(GetProcessHeap(), 0, Size); if (*UdpTable == NULL) { return ERROR_OUTOFMEMORY; }
ret = GetExtendedUdpTable(*UdpTable, &Size, Order, Family, Class, 0); if (ret != NO_ERROR) { HeapFree(GetProcessHeap(), 0, *UdpTable); *UdpTable = NULL; } }
return ret;}
static voidtest_tcp(LPWSTR svc_name, DWORD service_tag){ SOCKET sock; SOCKADDR_IN server; PMIB_TCPTABLE_OWNER_MODULE TcpTableOwnerMod; DWORD i, ret; BOOLEAN Found; DWORD Pid = GetCurrentProcessId();
sock = socket(AF_INET, SOCK_STREAM, 0); service_ok(sock != INVALID_SOCKET, "Socket creation failed!\n");
ZeroMemory(&server, sizeof(SOCKADDR_IN)); server.sin_family = AF_INET; server.sin_addr.s_addr = htonl(INADDR_ANY); server.sin_port = htons(9876);
ret = bind(sock, (SOCKADDR*)&server, sizeof(SOCKADDR_IN)); service_ok(ret != SOCKET_ERROR, "binding failed\n");
ret = listen(sock, SOMAXCONN); service_ok(ret != SOCKET_ERROR, "listening failed\n");
ret = GetExtendedTcpTableWithAlloc((PVOID *)&TcpTableOwnerMod, TRUE, AF_INET, TCP_TABLE_OWNER_MODULE_LISTENER); service_ok(ret == ERROR_SUCCESS, "GetExtendedTcpTableWithAlloc failed: %x\n", ret); if (ret == ERROR_SUCCESS) { service_ok(TcpTableOwnerMod->dwNumEntries > 0, "No TCP connections?!\n");
Found = FALSE; for (i = 0; i < TcpTableOwnerMod->dwNumEntries; ++i) { if (TcpTableOwnerMod->table[i].dwState == MIB_TCP_STATE_LISTEN && TcpTableOwnerMod->table[i].dwLocalAddr == 0 && TcpTableOwnerMod->table[i].dwLocalPort == htons(9876) && TcpTableOwnerMod->table[i].dwRemoteAddr == 0) { Found = TRUE; break; } }
service_ok(Found, "Our socket wasn't found!\n"); if (Found) { DWORD Size = 0; PTCPIP_OWNER_MODULE_BASIC_INFO BasicInfo = NULL;
service_ok(TcpTableOwnerMod->table[i].dwOwningPid == Pid, "Invalid owner\n"); service_ok((DWORD)(TcpTableOwnerMod->table[i].OwningModuleInfo[0]) == service_tag, "Invalid tag: %x - %x\n", (DWORD)TcpTableOwnerMod->table[i].OwningModuleInfo[0], service_tag);
ret = GetOwnerModuleFromTcpEntry(&TcpTableOwnerMod->table[i], TCPIP_OWNER_MODULE_INFO_BASIC, BasicInfo, &Size); service_ok(ret == ERROR_INSUFFICIENT_BUFFER, "GetOwnerModuleFromTcpEntry failed with: %x\n", ret); if (ERROR_INSUFFICIENT_BUFFER == ERROR_INSUFFICIENT_BUFFER) { BasicInfo = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, Size); service_ok(BasicInfo != NULL, "HeapAlloc failed\n");
ret = GetOwnerModuleFromTcpEntry(&TcpTableOwnerMod->table[i], TCPIP_OWNER_MODULE_INFO_BASIC, BasicInfo, &Size); service_ok(ret == ERROR_SUCCESS, "GetOwnerModuleFromTcpEntry failed with: %x\n", ret); service_ok(_wcsicmp(svc_name, BasicInfo->pModulePath) == 0, "Mismatching names (%S, %S)\n", svc_name, BasicInfo->pModulePath); service_ok(_wcsicmp(svc_name, BasicInfo->pModuleName) == 0, "Mismatching names (%S, %S)\n", svc_name, BasicInfo->pModuleName); } }
HeapFree(GetProcessHeap(), 0, TcpTableOwnerMod); }
closesocket(sock);}
static voidtest_udp(LPWSTR svc_name, DWORD service_tag){ SOCKET sock; SOCKADDR_IN server; PMIB_UDPTABLE_OWNER_MODULE UdpTableOwnerMod; DWORD i, ret; BOOLEAN Found; DWORD Pid = GetCurrentProcessId();
sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); service_ok(sock != INVALID_SOCKET, "Socket creation failed!\n");
ZeroMemory(&server, sizeof(SOCKADDR_IN)); server.sin_family = AF_INET; server.sin_addr.s_addr = htonl(INADDR_ANY); server.sin_port = htons(9876);
ret = bind(sock, (SOCKADDR*)&server, sizeof(SOCKADDR_IN)); service_ok(ret != SOCKET_ERROR, "binding failed\n");
ret = GetExtendedUdpTableWithAlloc((PVOID *)&UdpTableOwnerMod, TRUE, AF_INET, UDP_TABLE_OWNER_MODULE); service_ok(ret == ERROR_SUCCESS, "GetExtendedUdpTableWithAlloc failed: %x\n", ret); if (ret == ERROR_SUCCESS) { service_ok(UdpTableOwnerMod->dwNumEntries > 0, "No TCP connections?!\n");
Found = FALSE; for (i = 0; i < UdpTableOwnerMod->dwNumEntries; ++i) { if (UdpTableOwnerMod->table[i].dwLocalAddr == 0 && UdpTableOwnerMod->table[i].dwLocalPort == htons(9876)) { Found = TRUE; break; } }
service_ok(Found, "Our socket wasn't found!\n"); if (Found) { DWORD Size = 0; PTCPIP_OWNER_MODULE_BASIC_INFO BasicInfo = NULL;
service_ok(UdpTableOwnerMod->table[i].dwOwningPid == Pid, "Invalid owner\n"); service_ok((DWORD)(UdpTableOwnerMod->table[i].OwningModuleInfo[0]) == service_tag, "Invalid tag: %x - %x\n", (DWORD)UdpTableOwnerMod->table[i].OwningModuleInfo[0], service_tag);
ret = GetOwnerModuleFromUdpEntry(&UdpTableOwnerMod->table[i], TCPIP_OWNER_MODULE_INFO_BASIC, BasicInfo, &Size); service_ok(ret == ERROR_INSUFFICIENT_BUFFER, "GetOwnerModuleFromUdpEntry failed with: %x\n", ret); if (ret == ERROR_INSUFFICIENT_BUFFER) { BasicInfo = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, Size); service_ok(BasicInfo != NULL, "HeapAlloc failed\n");
ret = GetOwnerModuleFromUdpEntry(&UdpTableOwnerMod->table[i], TCPIP_OWNER_MODULE_INFO_BASIC, BasicInfo, &Size); service_ok(ret == ERROR_SUCCESS, "GetOwnerModuleFromUdpEntry failed with: %x\n", ret); service_ok(_wcsicmp(svc_name, BasicInfo->pModulePath) == 0, "Mismatching names (%S, %S)\n", svc_name, BasicInfo->pModulePath); service_ok(_wcsicmp(svc_name, BasicInfo->pModuleName) == 0, "Mismatching names (%S, %S)\n", svc_name, BasicInfo->pModuleName); } }
HeapFree(GetProcessHeap(), 0, UdpTableOwnerMod); }
closesocket(sock);}
static void WINAPIservice_main(DWORD dwArgc, LPWSTR* lpszArgv){ // SERVICE_STATUS_HANDLE status_handle; PTEB Teb; WSADATA wsaData;
UNREFERENCED_PARAMETER(dwArgc);
/* Register our service for control (lpszArgv[0] holds the service name) */ status_handle = RegisterServiceCtrlHandlerW(lpszArgv[0], service_handler); service_ok(status_handle != NULL, "RegisterServiceCtrlHandler failed: %lu\n", GetLastError()); if (!status_handle) return;
/* Report SERVICE_RUNNING status */ report_service_status(SERVICE_RUNNING, NO_ERROR, 4000);
/* Check our tag is not 0 */ Teb = NtCurrentTeb(); service_ok(Teb->SubProcessTag != 0, "SubProcessTag is not defined!\n");
if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) { skip("Failed to init WS2\n"); goto quit; }
test_tcp(lpszArgv[0], PtrToUlong(Teb->SubProcessTag)); test_udp(lpszArgv[0], PtrToUlong(Teb->SubProcessTag));
WSACleanup();quit: /* Work is done */ report_service_status(SERVICE_STOPPED, NO_ERROR, 0);}
static BOOL start_service(PCSTR service_nameA, PCWSTR service_nameW){ BOOL res;
SERVICE_TABLE_ENTRYW servtbl[] = { { (PWSTR)service_nameW, service_main }, { NULL, NULL } };
res = StartServiceCtrlDispatcherW(servtbl); service_ok(res, "StartServiceCtrlDispatcherW failed: %lu\n", GetLastError()); return res;}
/*** Tester part of the test ***/
static voidmy_test_server(PCSTR service_nameA, PCWSTR service_nameW, void *param){ BOOL res; SC_HANDLE hSC = NULL; SC_HANDLE hService = NULL; SERVICE_STATUS ServiceStatus;
/* Open the SCM */ hSC = OpenSCManagerW(NULL, NULL, SC_MANAGER_ALL_ACCESS); if (!hSC) { skip("OpenSCManagerW failed with error %lu!\n", GetLastError()); return; }
/* First create ourselves as a service running in the default LocalSystem account */ hService = register_service_exW(hSC, L"ServiceNetwork", service_nameW, NULL, SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS, SERVICE_DEMAND_START, SERVICE_ERROR_IGNORE, NULL, NULL, NULL, NULL, NULL); if (!hService) { skip("CreateServiceW failed with error %lu!\n", GetLastError()); goto Cleanup; }
/* Start it */ if (!StartServiceW(hService, 0, NULL)) { skip("StartServiceW failed with error %lu!\n", GetLastError()); goto Cleanup; }
/* Wait for the service to stop by itself */ do { Sleep(100); ZeroMemory(&ServiceStatus, sizeof(ServiceStatus)); res = QueryServiceStatus(hService, &ServiceStatus); } while (res && ServiceStatus.dwCurrentState != SERVICE_STOPPED); ok(res, "QueryServiceStatus failed: %lu\n", GetLastError()); ok(ServiceStatus.dwCurrentState == SERVICE_STOPPED, "ServiceStatus.dwCurrentState = %lx\n", ServiceStatus.dwCurrentState);
/* Be sure the service is really stopped */ res = ControlService(hService, SERVICE_CONTROL_STOP, &ServiceStatus); if (!res && ServiceStatus.dwCurrentState != SERVICE_STOPPED && ServiceStatus.dwCurrentState != SERVICE_STOP_PENDING && GetLastError() != ERROR_SERVICE_NOT_ACTIVE) { skip("ControlService failed with error %lu!\n", GetLastError()); goto Cleanup; }
#if 0 trace("Service stopped. Going to restart it...\n");
/* Now change the service configuration to make it start under the NetworkService account */ if (!ChangeServiceConfigW(hService, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, SERVICE_NO_CHANGE, NULL, NULL, NULL, NULL, L"NT AUTHORITY\\NetworkService", L"", NULL)) { skip("ChangeServiceConfigW failed with error %lu!\n", GetLastError()); goto Cleanup; }
/* Start it */ if (!StartServiceW(hService, 0, NULL)) { skip("StartServiceW failed with error %lu!\n", GetLastError()); goto Cleanup; }
/* Wait for the service to stop by itself */ do { Sleep(100); ZeroMemory(&ServiceStatus, sizeof(ServiceStatus)); res = QueryServiceStatus(hService, &ServiceStatus); } while (res && ServiceStatus.dwCurrentState != SERVICE_STOPPED); ok(res, "QueryServiceStatus failed: %lu\n", GetLastError()); ok(ServiceStatus.dwCurrentState == SERVICE_STOPPED, "ServiceStatus.dwCurrentState = %lx\n", ServiceStatus.dwCurrentState);
/* Be sure the service is really stopped */ res = ControlService(hService, SERVICE_CONTROL_STOP, &ServiceStatus); if (!res && ServiceStatus.dwCurrentState != SERVICE_STOPPED && ServiceStatus.dwCurrentState != SERVICE_STOP_PENDING && GetLastError() != ERROR_SERVICE_NOT_ACTIVE) { skip("ControlService failed with error %lu!\n", GetLastError()); goto Cleanup; }#endif
Cleanup: if (hService) { res = DeleteService(hService); ok(res, "DeleteService failed: %lu\n", GetLastError()); CloseServiceHandle(hService); }
if (hSC) CloseServiceHandle(hSC);}
START_TEST(ServiceNetwork){ int argc; char** argv;
/* Check whether this test is started as a separated service process */ argc = winetest_get_mainargs(&argv); if (argc >= 3) { service_process(start_service, argc, argv); return; }
/* We are started as the real test */ test_runner(my_test_server, NULL); // trace("Returned from test_runner\n");}