ServiceArgs.c File Reference

#include "precomp.h"

Include dependency graph for ServiceArgs.c:

Go to the source code of this file.

Functions
static void	send_msg (const char *type, const char *msg)
static void	service_ok (int cnd, const char *msg,...)
static VOID WINAPI	service_handler (DWORD ctrl)
static void	service_main_common (void)
static void WINAPI	service_mainA (DWORD service_argc, char **service_argv)
static void WINAPI	service_mainW (DWORD service_argc, WCHAR **service_argv)
static void	service_process (BOOLEAN unicode)
static SC_HANDLE	register_service (PCWSTR extra_args)
static DWORD WINAPI	pipe_thread (void *arg)
static void	test_startA (SC_HANDLE service_handle, int service_argc, const char **service_argv)
static void	test_startW (SC_HANDLE service_handle, int service_argc, const WCHAR **service_argv)
static void	test_runner (BOOLEAN unicode, PCWSTR extra_args, int service_argc, void *service_argv)
	START_TEST (ServiceArgs)

Variables
static char **	argv
static int	argc
static HANDLE	pipe_handle = INVALID_HANDLE_VALUE
static char	service_nameA [100]
static WCHAR	service_nameW [100]
static WCHAR	named_pipe_name [100]
static SC_HANDLE	scm_handle
static SERVICE_STATUS_HANDLE	service_handle

Function Documentation

pipe_thread()

static DWORD WINAPI pipe_thread ( void *  arg )

static

Definition at line 252 of file ServiceArgs.c.

{
    char buf[512];
    DWORD read;
    BOOL res;
 
    res = ConnectNamedPipe(pipe_handle, NULL);
    ok(res || GetLastError() == ERROR_PIPE_CONNECTED, "ConnectNamedPipe failed: %lu\n", GetLastError());
 
    while (1)
    {
        res = ReadFile(pipe_handle, buf, sizeof(buf), &read, NULL);
        if (!res)
        {
            ok(GetLastError() == ERROR_BROKEN_PIPE || GetLastError() == ERROR_INVALID_HANDLE,
               "ReadFile failed: %lu\n", GetLastError());
            break;
        }
 
        if (!strncmp(buf, "TRACE:", 6))
        {
            trace("service trace: %s\n", buf+6);
        }
        else if (!strncmp(buf, "OK:", 3))
        {
            ok(1, "service: %s\n", buf+3);
        }
        else if (!strncmp(buf, "FAIL:", 5))
        {
            ok(0, "service: %s\n", buf+5);
        }
        else
        {
            ok(0, "malformed service message: %s\n", buf);
        }
    }
 
    DisconnectNamedPipe(pipe_handle);
    //trace("pipe disconnected\n");
    return 0;
}

Referenced by test_runner().

register_service()

static SC_HANDLE register_service ( PCWSTR  extra_args )

static

Definition at line 226 of file ServiceArgs.c.

{
    WCHAR service_cmd[MAX_PATH+150];
    SC_HANDLE service;
 
    GetModuleFileNameW(NULL, service_cmd, MAX_PATH);
 
    StringCbCatW(service_cmd, sizeof(service_cmd), L" ServiceArgs ");
    StringCbCatW(service_cmd, sizeof(service_cmd), service_nameW);
    StringCbCatW(service_cmd, sizeof(service_cmd), extra_args);
 
    trace("service_cmd \"%ls\"\n", service_cmd);
 
    service = CreateServiceW(scm_handle, service_nameW, service_nameW, SERVICE_ALL_ACCESS,
                             SERVICE_WIN32_OWN_PROCESS, SERVICE_DEMAND_START, SERVICE_ERROR_IGNORE,
                             service_cmd, NULL, NULL, NULL, NULL, NULL);
    if (!service && GetLastError() == ERROR_ACCESS_DENIED)
    {
        skip("Not enough access right to create service\n");
        return NULL;
    }
 
    ok(service != NULL, "CreateService failed: %lu\n", GetLastError());
    return service;
}

send_msg()

static void send_msg ( const char *  type, const char *  msg  )

static

Definition at line 23 of file ServiceArgs.c.

{
    DWORD written = 0;
    char buf[512];
 
    StringCbPrintfA(buf, sizeof(buf), "%s:%s", type, msg);
    WriteFile(pipe_handle, buf, lstrlenA(buf)+1, &written, NULL);
}

Referenced by service_ok().

service_handler()

static VOID WINAPI service_handler ( DWORD  ctrl )

static

Definition at line 58 of file ServiceArgs.c.

{
    SERVICE_STATUS status;
 
    status.dwServiceType             = SERVICE_WIN32;
    status.dwControlsAccepted        = SERVICE_ACCEPT_STOP;
    status.dwWin32ExitCode           = 0;
    status.dwServiceSpecificExitCode = 0;
    status.dwCheckPoint              = 0;
    status.dwWaitHint                = 0;
 
    switch(ctrl)
    {
    case SERVICE_CONTROL_STOP:
        status.dwCurrentState     = SERVICE_STOP_PENDING;
        status.dwControlsAccepted = 0;
        SetServiceStatus(service_handle, &status);
    default:
        status.dwCurrentState = SERVICE_RUNNING;
        SetServiceStatus(service_handle, &status);
    }
}

Referenced by service_main_common().

service_main_common()

static void service_main_common ( void  )

static

Definition at line 81 of file ServiceArgs.c.

{
    SERVICE_STATUS status;
    BOOL res;
 
    service_handle = RegisterServiceCtrlHandlerW(service_nameW, service_handler);
    service_ok(service_handle != NULL, "RegisterServiceCtrlHandler failed: %lu\n", GetLastError());
    if (!service_handle)
        return;
 
    status.dwServiceType             = SERVICE_WIN32;
    status.dwCurrentState            = SERVICE_RUNNING;
    status.dwControlsAccepted        = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN;
    status.dwWin32ExitCode           = 0;
    status.dwServiceSpecificExitCode = 0;
    status.dwCheckPoint              = 0;
    status.dwWaitHint                = 10000;
    res = SetServiceStatus(service_handle, &status);
    service_ok(res, "SetServiceStatus(SERVICE_RUNNING) failed: %lu", GetLastError());
 
    Sleep(100);
 
    status.dwCurrentState     = SERVICE_STOPPED;
    status.dwControlsAccepted = 0;
    res = SetServiceStatus(service_handle, &status);
    service_ok(res, "SetServiceStatus(SERVICE_STOPPED) failed: %lu", GetLastError());
}

Referenced by service_mainA(), and service_mainW().

service_mainA()

static void WINAPI service_mainA ( DWORD  service_argc, char **  service_argv  )

static

Definition at line 124 of file ServiceArgs.c.

{
    int i;
    char *next_arg;
    char *next_arg_aligned;
 
    service_ok(service_argc == argc - 3, "service_argc = %d, expected %d", service_argc, argc - 3);
    if (service_argc == argc - 3)
    {
        service_ok(!strcmp(service_argv[0], service_nameA), "service_argv[0] = %s, expected %s", service_argv[0], service_nameA);
        service_ok(service_argv[0] == (char *)&service_argv[service_argc] ||
                   service_argv[1] == (char *)&service_argv[service_argc] ||
                   service_argv[1] == (char *)(((ULONG_PTR)&service_argv[service_argc] + 7) & ~7), "service_argv[0] = %p, [1] = %p, expected one of them to be %p", service_argv[0], service_argv[1], &service_argv[service_argc]);
        //service_trace("service_argv[0] = %p", service_argv[0]);
        next_arg_aligned = next_arg = NULL;
        for (i = 1; i < service_argc; i++)
        {
            //service_trace("service_argv[%d] = %p", i, service_argv[i]);
            service_ok(!strcmp(service_argv[i], argv[i + 3]), "service_argv[%d] = %s, expected %s", i, service_argv[i], argv[i + 3]);
            service_ok(next_arg == NULL ||
                       service_argv[i] == next_arg ||
                       service_argv[i] == next_arg_aligned, "service_argv[%d] = %p, expected %p or %p", i, service_argv[i], next_arg, next_arg_aligned);
            next_arg = service_argv[i];
            next_arg += 2 * (strlen(next_arg) + 1);
            next_arg_aligned = (char *)(((ULONG_PTR)next_arg + 3) & ~3);
        }
    }
 
    service_main_common();
}

Referenced by service_process().

service_mainW()

static void WINAPI service_mainW ( DWORD  service_argc, WCHAR **  service_argv  )

static

Definition at line 155 of file ServiceArgs.c.

{
    int i;
    WCHAR argW[32];
    WCHAR *next_arg;
    WCHAR *next_arg_aligned;
 
    service_ok(service_argc == argc - 3, "service_argc = %d, expected %d", service_argc, argc - 3);
    if (service_argc == argc - 3)
    {
        service_ok(!wcscmp(service_argv[0], service_nameW), "service_argv[0] = %ls, expected %ls", service_argv[0], service_nameW);
        service_ok(service_argv[0] == (WCHAR *)&service_argv[service_argc] ||
                   service_argv[1] == (WCHAR *)&service_argv[service_argc] ||
                   service_argv[1] == (WCHAR *)(((ULONG_PTR)&service_argv[service_argc] + 7) & ~7), "service_argv[0] = %p, [1] = %p, expected one of them to be %p", service_argv[0], service_argv[1], &service_argv[service_argc]);
        //service_trace("service_argv[0] = %p", service_argv[0]);
        next_arg_aligned = next_arg = NULL;
        for (i = 1; i < service_argc; i++)
        {
            //service_trace("service_argv[%d] = %p", i, service_argv[i]);
            MultiByteToWideChar(CP_ACP, 0, argv[i + 3], -1, argW, _countof(argW));
            service_ok(!wcscmp(service_argv[i], argW), "service_argv[%d] = %ls, expected %ls", i, service_argv[i], argW);
            service_ok(next_arg == NULL ||
                       service_argv[i] == next_arg ||
                       service_argv[i] == next_arg_aligned, "service_argv[%d] = %p, expected %p or %p", i, service_argv[i], next_arg, next_arg_aligned);
            next_arg = service_argv[i];
            next_arg += wcslen(next_arg) + 1;
            next_arg_aligned = (WCHAR *)(((ULONG_PTR)next_arg + 3) & ~3);
        }
    }
 
    service_main_common();
}

Referenced by service_process().

service_ok()

static void service_ok ( int  cnd, const char *  msg,   ...  )

static

Definition at line 46 of file ServiceArgs.c.

{
   va_list valist;
   char buf[512];
 
    va_start(valist, msg);
    StringCbVPrintfA(buf, sizeof(buf), msg, valist);
    va_end(valist);
 
    send_msg(cnd ? "OK" : "FAIL", buf);
}

Referenced by report_service_status(), service_main(), service_main_common(), service_mainA(), service_mainW(), service_process(), start_service(), test_tcp(), and test_udp().

service_process()

static void service_process ( BOOLEAN  unicode )

static

Definition at line 188 of file ServiceArgs.c.

{
    BOOL res;
 
    SERVICE_TABLE_ENTRYA servtblA[] =
    {
        { service_nameA, service_mainA },
        { NULL, NULL }
    };
    SERVICE_TABLE_ENTRYW servtblW[] =
    {
        { service_nameW, service_mainW },
        { NULL, NULL }
    };
 
    res = WaitNamedPipeW(named_pipe_name, NMPWAIT_USE_DEFAULT_WAIT);
    if (!res)
        return;
 
    pipe_handle = CreateFileW(named_pipe_name, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    if (pipe_handle == INVALID_HANDLE_VALUE)
        return;
 
    //service_trace("Starting...");
    if (unicode)
    {
        res = StartServiceCtrlDispatcherW(servtblW);
        service_ok(res, "StartServiceCtrlDispatcherW failed: %lu\n", GetLastError());
    }
    else
    {
        res = StartServiceCtrlDispatcherA(servtblA);
        service_ok(res, "StartServiceCtrlDispatcherA failed: %lu\n", GetLastError());
    }
 
    CloseHandle(pipe_handle);
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

ServiceArgs

)

Definition at line 382 of file ServiceArgs.c.

{
    argc = winetest_get_mainargs(&argv);
 
    scm_handle = OpenSCManagerW(NULL, NULL, SC_MANAGER_ALL_ACCESS);
    ok(scm_handle != NULL, "OpenSCManager failed: %lu\n", GetLastError());
    if (!scm_handle)
    {
        skip("Failed to open service control manager. Skipping test\n");
        return;
    }
 
    if (argc < 3)
    {
        char *service_argvA[10];
        WCHAR *service_argvW[10];
 
        test_runner(FALSE, L" A", 0, NULL);
        test_runner(FALSE, L" W", 0, NULL);
        test_runner(TRUE, L" A", 0, NULL);
        test_runner(TRUE, L" W", 0, NULL);
 
        service_argvA[0] = "x";
        service_argvW[0] = L"x";
        test_runner(FALSE, L" A x", 1, service_argvA);
        test_runner(FALSE, L" W x", 1, service_argvA);
        test_runner(TRUE, L" A x", 1, service_argvW);
        test_runner(TRUE, L" W x", 1, service_argvW);
 
        service_argvA[1] = "y";
        service_argvW[1] = L"y";
        test_runner(FALSE, L" A x y", 2, service_argvA);
        test_runner(FALSE, L" W x y", 2, service_argvA);
        test_runner(TRUE, L" A x y", 2, service_argvW);
        test_runner(TRUE, L" W x y", 2, service_argvW);
 
        service_argvA[0] = "ab";
        service_argvW[0] = L"ab";
        test_runner(FALSE, L" A ab y", 2, service_argvA);
        test_runner(FALSE, L" W ab y", 2, service_argvA);
        test_runner(TRUE, L" A ab y", 2, service_argvW);
        test_runner(TRUE, L" W ab y", 2, service_argvW);
 
        service_argvA[0] = "abc";
        service_argvW[0] = L"abc";
        test_runner(FALSE, L" A abc y", 2, service_argvA);
        test_runner(FALSE, L" W abc y", 2, service_argvA);
        test_runner(TRUE, L" A abc y", 2, service_argvW);
        test_runner(TRUE, L" W abc y", 2, service_argvW);
 
        service_argvA[0] = "abcd";
        service_argvW[0] = L"abcd";
        test_runner(FALSE, L" A abcd y", 2, service_argvA);
        test_runner(FALSE, L" W abcd y", 2, service_argvA);
        test_runner(TRUE, L" A abcd y", 2, service_argvW);
        test_runner(TRUE, L" W abcd y", 2, service_argvW);
 
        service_argvA[0] = "abcde";
        service_argvW[0] = L"abcde";
        test_runner(FALSE, L" A abcde y", 2, service_argvA);
        test_runner(FALSE, L" W abcde y", 2, service_argvA);
        test_runner(TRUE, L" A abcde y", 2, service_argvW);
        test_runner(TRUE, L" W abcde y", 2, service_argvW);
 
        service_argvA[0] = "abcdef";
        service_argvW[0] = L"abcdef";
        test_runner(FALSE, L" A abcdef y", 2, service_argvA);
        test_runner(FALSE, L" W abcdef y", 2, service_argvA);
        test_runner(TRUE, L" A abcdef y", 2, service_argvW);
        test_runner(TRUE, L" W abcdef y", 2, service_argvW);
 
        service_argvA[0] = "abcdefg";
        service_argvW[0] = L"abcdefg";
        test_runner(FALSE, L" A abcdefg y", 2, service_argvA);
        test_runner(FALSE, L" W abcdefg y", 2, service_argvA);
        test_runner(TRUE, L" A abcdefg y", 2, service_argvW);
        test_runner(TRUE, L" W abcdefg y", 2, service_argvW);
 
        service_argvA[0] = "";
        service_argvW[0] = L"";
        test_runner(FALSE, L" A \"\" y", 2, service_argvA);
        test_runner(FALSE, L" W \"\" y", 2, service_argvA);
        test_runner(TRUE, L" A \"\" y", 2, service_argvW);
        test_runner(TRUE, L" W \"\" y", 2, service_argvW);
    }
    else
    {
        strcpy(service_nameA, argv[2]);
        MultiByteToWideChar(CP_ACP, 0, service_nameA, -1, service_nameW, _countof(service_nameW));
        StringCbPrintfW(named_pipe_name, sizeof(named_pipe_name), L"\\\\.\\pipe\\%ls_pipe", service_nameW);
        if (!strcmp(argv[3], "A"))
            service_process(FALSE);
        else
            service_process(TRUE);
    }
 
    CloseServiceHandle(scm_handle);
}

test_runner()

static void test_runner ( BOOLEAN  unicode, PCWSTR  extra_args, int  service_argc, void *  service_argv  )

static

Definition at line 334 of file ServiceArgs.c.

{
    HANDLE thread;
    SC_HANDLE service_handle;
    BOOL res;
 
    StringCbPrintfW(service_nameW, sizeof(service_nameW), L"WineTestService%lu", GetTickCount());
    WideCharToMultiByte(CP_ACP, 0, service_nameW, -1, service_nameA, _countof(service_nameA), NULL, NULL);
    //trace("service_name: %ls\n", service_nameW);
    StringCbPrintfW(named_pipe_name, sizeof(named_pipe_name), L"\\\\.\\pipe\\%ls_pipe", service_nameW);
 
    pipe_handle = CreateNamedPipeW(named_pipe_name, PIPE_ACCESS_INBOUND,
                                   PIPE_TYPE_MESSAGE|PIPE_READMODE_MESSAGE|PIPE_WAIT, 10, 2048, 2048, 10000, NULL);
    ok(pipe_handle != INVALID_HANDLE_VALUE, "CreateNamedPipe failed: %lu\n", GetLastError());
    if (pipe_handle == INVALID_HANDLE_VALUE)
        return;
 
    thread = CreateThread(NULL, 0, pipe_thread, NULL, 0, NULL);
    ok(thread != NULL, "CreateThread failed: %lu\n", GetLastError());
    if (!thread)
        goto Quit;
 
    service_handle = register_service(extra_args);
    if (!service_handle)
        goto Quit;
 
    //trace("starting...\n");
 
    if (unicode)
        test_startW(service_handle, service_argc, service_argv);
    else
        test_startA(service_handle, service_argc, service_argv);
 
    res = DeleteService(service_handle);
    ok(res, "DeleteService failed: %lu\n", GetLastError());
 
    CloseServiceHandle(service_handle);
 
    ok(WaitForSingleObject(thread, 10000) == WAIT_OBJECT_0, "Timeout waiting for thread\n");
 
Quit:
    if (thread)
        CloseHandle(thread);
 
    if (pipe_handle != INVALID_HANDLE_VALUE)
        CloseHandle(pipe_handle);
}

Referenced by START_TEST().

test_startA()

static void test_startA ( SC_HANDLE  service_handle, int  service_argc, const char **  service_argv  )

static

Definition at line 294 of file ServiceArgs.c.

{
    SERVICE_STATUS status;
    BOOL res;
 
    res = StartServiceA(service_handle, service_argc, service_argv);
    ok(res, "StartService failed: %lu\n", GetLastError());
    if (!res)
        return;
 
    do
    {
        Sleep(100);
        ZeroMemory(&status, sizeof(status));
        res = QueryServiceStatus(service_handle, &status);
    } while (res && status.dwCurrentState != SERVICE_STOPPED);
    ok(res, "QueryServiceStatus failed: %lu\n", GetLastError());
    ok(status.dwCurrentState == SERVICE_STOPPED, "status.dwCurrentState = %lx\n", status.dwCurrentState);
}

Referenced by test_runner().

test_startW()

static void test_startW ( SC_HANDLE  service_handle, int  service_argc, const WCHAR **  service_argv  )

static

Definition at line 314 of file ServiceArgs.c.

{
    SERVICE_STATUS status;
    BOOL res;
 
    res = StartServiceW(service_handle, service_argc, service_argv);
    ok(res, "StartService failed: %lu\n", GetLastError());
    if (!res)
        return;
 
    do
    {
        Sleep(100);
        ZeroMemory(&status, sizeof(status));
        res = QueryServiceStatus(service_handle, &status);
    } while (res && status.dwCurrentState != SERVICE_STOPPED);
    ok(res, "QueryServiceStatus failed: %lu\n", GetLastError());
    ok(status.dwCurrentState == SERVICE_STOPPED, "status.dwCurrentState = %lx\n", status.dwCurrentState);
}

Referenced by test_runner().

Variable Documentation

argc

int argc

static

Definition at line 12 of file ServiceArgs.c.

Referenced by service_mainA(), service_mainW(), and START_TEST().

argv

char** argv

static

Definition at line 11 of file ServiceArgs.c.

Referenced by service_mainA(), service_mainW(), and START_TEST().

named_pipe_name

WCHAR named_pipe_name[100]

static

Definition at line 17 of file ServiceArgs.c.

Referenced by service_process(), START_TEST(), and test_runner().

pipe_handle

HANDLE pipe_handle = INVALID_HANDLE_VALUE

static

Definition at line 14 of file ServiceArgs.c.

Referenced by pipe_thread(), send_msg(), service_process(), and test_runner().

scm_handle

SC_HANDLE scm_handle

static

Definition at line 20 of file ServiceArgs.c.

Referenced by InstallService(), register_service(), register_service_exA(), register_service_exW(), register_serviceA(), register_serviceW(), RunTest(), START_TEST(), test_create_delete_svc(), test_driver_install(), test_enum_svc(), test_enum_svc_ex(), test_EventLog(), test_get_displayname(), test_get_servicekeyname(), test_install_svc_from(), test_open_scm(), test_open_svc(), test_query_svc(), test_queryconfig2(), test_refcount(), test_sequence(), test_servicenotify(), test_SetupDiInstallClassExA(), test_SetupInstallServicesFromInfSectionExA(), test_SetupInstallServicesFromInfSectionExW(), and test_start_stop().

service_handle

SERVICE_STATUS_HANDLE service_handle

static

Definition at line 21 of file ServiceArgs.c.

Referenced by service_handler(), service_main_common(), test_runner(), test_startA(), and test_startW().

service_nameA

char service_nameA[100]

static

Definition at line 15 of file ServiceArgs.c.

Referenced by service_mainA(), service_process(), START_TEST(), and test_runner().

service_nameW

WCHAR service_nameW[100]

static

Definition at line 16 of file ServiceArgs.c.

Referenced by my_test_server(), register_service(), service_main_common(), service_mainW(), service_process(), start_service(), START_TEST(), and test_runner().