main.c File Reference

#include <windef.h>
#include <winbase.h>
#include <strsafe.h>

Include dependency graph for main.c:

Go to the source code of this file.

Functions
int	run_rapps (LPWSTR cmdline)
int	wmain (int argc, wchar_t *argv[])

Function Documentation

run_rapps()

int run_rapps

(

LPWSTR

cmdline

)

Definition at line 12 of file main.c.

{
    DWORD dwExit;
    STARTUPINFOW si = { sizeof(si) };
    PROCESS_INFORMATION pi = { 0 };

    SetLastError(0);
    if (!CreateProcessW(NULL, cmdline, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
    {
        fprintf(stderr, "Unable to create rapps.exe process...\n");
        return -1;
    }
    CloseHandle(pi.hThread);
    WaitForSingleObject(pi.hProcess, INFINITE);
    GetExitCodeProcess(pi.hProcess, &dwExit);
    CloseHandle(pi.hProcess);
    return (int)dwExit;
}

Referenced by wmain().

wmain()

int wmain	(	int	argc,
		wchar_t *	argv[]
	)

Definition at line 31 of file main.c.

{
    int iRet;
    int n;
    size_t arglen;
    WCHAR RappsExe[MAX_PATH] = { 0 };
    wchar_t* cmdline;

    GetModuleFileNameW(NULL, RappsExe, ARRAYSIZE(RappsExe));
    arglen = wcslen(RappsExe);
    if (arglen > 4 && !wcsicmp(RappsExe + arglen - 4, L".com"))
    {
        wcscpy(RappsExe + arglen - 4, L".exe");
    }
    else
    {
        fprintf(stderr, "Unable to build rapps.exe path...\n");
        return -1;
    }

    arglen += (1 + 2);  // NULL terminator + 2 quotes

    for (n = 1; n < argc; ++n)
    {
        arglen += wcslen(argv[n]);
        arglen += 3;    // Surrounding quotes + space
    }

    cmdline = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, arglen * sizeof(WCHAR));
    if (cmdline)
    {
        wchar_t* ptr = cmdline;
        size_t cchRemaining = arglen;

        StringCchPrintfExW(ptr, cchRemaining, &ptr, &cchRemaining, 0, L"\"%s\"", RappsExe);

        for (n = 1; n < argc; ++n)
        {
            StringCchPrintfExW(ptr, cchRemaining, &ptr, &cchRemaining, 0, L" \"%s\"", argv[n]);
        }
    }

    iRet = run_rapps(cmdline);
    if (cmdline)
        HeapFree(GetProcessHeap(), 0, cmdline);
    return iRet;
}