The Tor Browser / file diff

     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/toolkit/xre/nsWindowsWMain.cpp	Wed Dec 31 06:09:35 2014 +0100
     1.3 @@ -0,0 +1,116 @@
     1.4 +/* This Source Code Form is subject to the terms of the Mozilla Public
     1.5 + * License, v. 2.0. If a copy of the MPL was not distributed with this
     1.6 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
     1.7 +
     1.8 +// This file is a .cpp file meant to be included in nsBrowserApp.cpp and other
     1.9 +// similar bootstrap code. It converts wide-character windows wmain into UTF-8
    1.10 +// narrow-character strings.
    1.11 +
    1.12 +#ifndef XP_WIN
    1.13 +#error This file only makes sense on Windows.
    1.14 +#endif
    1.15 +
    1.16 +#include "nsUTF8Utils.h"
    1.17 +
    1.18 +#ifndef XRE_DONT_PROTECT_DLL_LOAD
    1.19 +#include "nsSetDllDirectory.h"
    1.20 +#endif
    1.21 +
    1.22 +#ifdef __MINGW32__
    1.23 +
    1.24 +/* MingW currently does not implement a wide version of the
    1.25 +   startup routines.  Workaround is to implement something like
    1.26 +   it ourselves.  See bug 411826 */
    1.27 +
    1.28 +#include <shellapi.h>
    1.29 +
    1.30 +int wmain(int argc, WCHAR **argv);
    1.31 +
    1.32 +int main(int argc, char **argv)
    1.33 +{
    1.34 +  LPWSTR commandLine = GetCommandLineW();
    1.35 +  int argcw = 0;
    1.36 +  LPWSTR *argvw = CommandLineToArgvW(commandLine, &argcw);
    1.37 +  if (!argvw)
    1.38 +    return 127;
    1.39 +
    1.40 +  int result = wmain(argcw, argvw);
    1.41 +  LocalFree(argvw);
    1.42 +  return result;
    1.43 +}
    1.44 +#endif /* __MINGW32__ */
    1.45 +
    1.46 +#define main NS_internal_main
    1.47 +
    1.48 +#ifndef XRE_WANT_ENVIRON
    1.49 +int main(int argc, char **argv);
    1.50 +#else
    1.51 +int main(int argc, char **argv, char **envp);
    1.52 +#endif
    1.53 +
    1.54 +static char*
    1.55 +AllocConvertUTF16toUTF8(char16ptr_t arg)
    1.56 +{
    1.57 +  // be generous... UTF16 units can expand up to 3 UTF8 units
    1.58 +  int len = wcslen(arg);
    1.59 +  char *s = new char[len * 3 + 1];
    1.60 +  if (!s)
    1.61 +    return nullptr;
    1.62 +
    1.63 +  ConvertUTF16toUTF8 convert(s);
    1.64 +  convert.write(arg, len);
    1.65 +  convert.write_terminator();
    1.66 +  return s;
    1.67 +}
    1.68 +
    1.69 +static void
    1.70 +FreeAllocStrings(int argc, char **argv)
    1.71 +{
    1.72 +  while (argc) {
    1.73 +    --argc;
    1.74 +    delete [] argv[argc];
    1.75 +  }
    1.76 +
    1.77 +  delete [] argv;
    1.78 +}
    1.79 +
    1.80 +int wmain(int argc, WCHAR **argv)
    1.81 +{
    1.82 +#ifndef XRE_DONT_PROTECT_DLL_LOAD
    1.83 +  mozilla::SanitizeEnvironmentVariables();
    1.84 +  SetDllDirectoryW(L"");
    1.85 +#endif
    1.86 +
    1.87 +  char **argvConverted = new char*[argc + 1];
    1.88 +  if (!argvConverted)
    1.89 +    return 127;
    1.90 +
    1.91 +  for (int i = 0; i < argc; ++i) {
    1.92 +    argvConverted[i] = AllocConvertUTF16toUTF8(argv[i]);
    1.93 +    if (!argvConverted[i]) {
    1.94 +      return 127;
    1.95 +    }
    1.96 +  }
    1.97 +  argvConverted[argc] = nullptr;
    1.98 +
    1.99 +  // need to save argvConverted copy for later deletion.
   1.100 +  char **deleteUs = new char*[argc+1];
   1.101 +  if (!deleteUs) {
   1.102 +    FreeAllocStrings(argc, argvConverted);
   1.103 +    return 127;
   1.104 +  }
   1.105 +  for (int i = 0; i < argc; i++)
   1.106 +    deleteUs[i] = argvConverted[i];
   1.107 +#ifndef XRE_WANT_ENVIRON
   1.108 +  int result = main(argc, argvConverted);
   1.109 +#else
   1.110 +  // Force creation of the multibyte _environ variable.
   1.111 +  getenv("PATH");
   1.112 +  int result = main(argc, argvConverted, _environ);
   1.113 +#endif
   1.114 +
   1.115 +  delete[] argvConverted;
   1.116 +  FreeAllocStrings(argc, deleteUs);
   1.117 +
   1.118 +  return result;
   1.119 +}