#include "Common.h" // "targetver.h", <Windows.h>, и др.
#include <boost/smart_ptr.hpp>
#include "Procedures.h"



namespace {

BOOL
_stdcall
RunInteractiveProcess(
    __in    wchar_t const     * pCmdLine,
    __in    DWORD               IdSession
    )
{
    // Получаем маркер безопасности текущего процесса.

    HANDLE hProcessToken;

    if (FALSE == OpenProcessToken(GetCurrentProcess(),
        TOKEN_ALL_ACCESS,
        &hProcessToken))
    {
        return FALSE;
    }   

    // Создаем производный маркер.

    HANDLE hNewToken;

    if (FALSE == DuplicateTokenEx(hProcessToken,
        MAXIMUM_ALLOWED,
        NULL,
        SecurityDelegation,
        TokenPrimary,
        &hNewToken))
    {
        CloseHandle(hProcessToken);
        return FALSE;
    }

    CloseHandle(hProcessToken);

    // Устанавливаем в маркере id сессии пользователя, в
    // которой будет создан новый процесс.

    if (FALSE == SetTokenInformation(hNewToken,
        TokenSessionId,
        &IdSession,
        sizeof (IdSession)))
    {
        CloseHandle(hNewToken);
        return FALSE;
    }

    // Выделяем временный буфер для командной строки.
    // Зачем это надо - смотрите в описании третьего параметра
    // функции CreateProcessAsUser (MSDN).

    int LengthOfCmdLine = lstrlenW(pCmdLine);
    boost::scoped_array<wchar_t> CmdLineBuffer(new wchar_t[LengthOfCmdLine + 1]);
    lstrcpyW(CmdLineBuffer.get(), pCmdLine);

    // Заполняем структуру STARTUPINFO.

    STARTUPINFOW si;
    
    GetStartupInfoW(&si);

    si.dwFlags      = STARTF_USESHOWWINDOW;
    si.wShowWindow  = SW_SHOWNORMAL;
    si.lpDesktop    = L"WinSta0\\Default";

    // Запускаем процесс с новым маркером безопасности.

    PROCESS_INFORMATION pi;

    if (FALSE == CreateProcessAsUserW(hNewToken,
        NULL,
        CmdLineBuffer.get(),
        NULL,
        NULL,
        FALSE,
        NORMAL_PRIORITY_CLASS,
        NULL,
        NULL,
        &si,
        &pi))
    {
        CloseHandle(hNewToken);
        return FALSE;
    }

    // Очистка ресурсов и выход.

    CloseHandle(pi.hProcess);
    CloseHandle(pi.hThread);
    CloseHandle(hNewToken);
    return TRUE;
}

} // namespace (anonymous)



VOID
_stdcall
ServiceMain(
    __in    DWORD           /* nArgs */,
    __in    wchar_t      ** /* pArgs */
    )
{
    // Id сессий можно посмотреть в диспетчере задач.

    // Если залогиниться двумя пользователями, то они
    // будут иметь номера сессий 0 и 1 в Windows XP, и
    // 1 и 2 на Vista и более поздних системах (ввиду
    // "Session 0 isolation").

    DWORD AdminSessionId = 0; // 1 for Vista and later.
    DWORD GuestSessionId = 1; // 2 for Vista and later.

    RunInteractiveProcess(L"cmd.exe", AdminSessionId);
    RunInteractiveProcess(L"calc.exe", AdminSessionId);

    RunInteractiveProcess(L"cmd.exe", GuestSessionId);
    RunInteractiveProcess(L"calc.exe", GuestSessionId);

    // Необходимо сообщить менеджеру служб об остановке,
    // иначе служба "повиснет".

    SERVICE_STATUS_HANDLE hStatus = RegisterServiceCtrlHandlerExW(L"my_service",
        HandlerEx,
        NULL);

    SERVICE_STATUS ss;
    ZeroMemory(&ss, sizeof (ss));

    ss.dwServiceType        = SERVICE_WIN32_OWN_PROCESS;
    ss.dwControlsAccepted   = SERVICE_ACCEPT_SHUTDOWN |
        SERVICE_ACCEPT_SESSIONCHANGE | SERVICE_ACCEPT_STOP;
    ss.dwCurrentState       = SERVICE_STOPPED;
    SetServiceStatus(hStatus, &ss);
}