The Tor Browser: b2g/gaia/run-b2g.cpp@4ab42b5ab56c

     1 #include <Windows.h>

     2 #include <Shlwapi.h>

     3 #include <strsafe.h>

     5 // Linker options

     6 #pragma comment(lib, "User32.lib")

     7 #pragma comment(lib, "shlwapi.lib")

     8 #pragma comment(linker, "/SUBSYSTEM:windows /ENTRY:wmainCRTStartup")

    10 #define NUM_MAX_PATH_BYTES sizeof(wchar_t) * MAX_PATH + 1

    11 #define PROFILE_ARG L" -profile "

    13 // Options that can be overridden at build time with -D

    14 #ifndef B2G_NAME

    15 #define B2G_NAME L"b2g.exe"

    16 #endif

    17 #ifndef GAIA_PATH

    18 #define GAIA_PATH L"gaia\\profile"

    19 #endif

    21 void error(wchar_t* msg){

    22     MessageBoxW(nullptr, msg, L"Error starting program", MB_OK | MB_ICONERROR);

    23 }

    25 /* This function takes a string which represents a windows path, orig.

    26  * The file portion of the path is stripped off and replaced with the

    27  * path component, file.  This function returns a string which represents

    28  * a windows path with the modifications requested and needs to be freed

    29  * explicitly by the calling function.

    30  */

    31 wchar_t* make_path_with_leaf_file_name(wchar_t* orig, wchar_t* file){

    32     wchar_t* buffer = (wchar_t*) malloc(NUM_MAX_PATH_BYTES);

    33     if (!buffer) {

    34         return nullptr;

    35     }

    36     if (FAILED(StringCchCopyW(buffer, NUM_MAX_PATH_BYTES, orig))) {

    37         error(L"Error copying string");

    38         free(buffer);

    39         buffer = nullptr;

    40     }

    41     PathRemoveFileSpecW(buffer);

    42     if (!PathAppendW(buffer, file)) {

    43         error(L"Unable to append file to directory");

    44         free(buffer);

    45         buffer = nullptr;

    46     }

    47     return buffer;

    48 }

    50 BOOL execute(wchar_t* binary_path, wchar_t* args, int cp_flags) {

    51     STARTUPINFOW si;

    52     PROCESS_INFORMATION pi;

    54     ZeroMemory(&si, sizeof(si));

    55     si.cb = sizeof(si);

    56     ZeroMemory(&pi, sizeof(pi));

    58     if (!CreateProcessW(

    59         binary_path,

    60         args,

    61         nullptr,

    62         nullptr,

    63         FALSE,

    64         cp_flags,

    65         nullptr,

    66         nullptr,

    67         &si,

    68         &pi)){

    69             error(L"Could not execute program");

    70             return FALSE;

    71     }

    73     WaitForInputIdle(pi.hProcess, 0);

    74     CloseHandle(pi.hProcess);

    75     CloseHandle(pi.hThread);

    76     return true;

    77 }

    79 int wmain(int argc, wchar_t *argv[], wchar_t *envp[]){

    80     int cp_flags;

    81     wchar_t* b2g_path = make_path_with_leaf_file_name(argv[0], B2G_NAME);

    82     wchar_t* profile_path = make_path_with_leaf_file_name(argv[0], GAIA_PATH);

    83 	// 10 chars for the ' -profile ' portion of the argument

    84     wchar_t* args = (wchar_t*) malloc(2 * NUM_MAX_PATH_BYTES + wcslen(PROFILE_ARG));

    85     if (FAILED(StringCchPrintfW(args, NUM_MAX_PATH_BYTES, L"\"%ws\"%ws\"%ws\"", b2g_path, PROFILE_ARG, profile_path))) {

    86         error(L"Could not create argument string");

    87         ExitProcess(1);

    88     }

    89 #ifdef SHOW_CONSOLE

    90     cp_flags = 0;

    91 #else

    92     cp_flags = DETACHED_PROCESS;

    93 #endif

    94     if (!execute(b2g_path, args, cp_flags)) {

    95         error(L"Failed to launch program");

    96     }

    97     free(profile_path);

    98     free(b2g_path);

    99     free(args);

   100     profile_path = b2g_path = args = nullptr;

   102 }

The Tor Browser / file revision