|
1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ |
|
2 /* This Source Code Form is subject to the terms of the Mozilla Public |
|
3 * License, v. 2.0. If a copy of the MPL was not distributed with this |
|
4 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ |
|
5 |
|
6 #ifndef nsSetDllDirectory_h |
|
7 #define nsSetDllDirectory_h |
|
8 |
|
9 #ifndef XP_WIN |
|
10 #error This file only makes sense on Windows. |
|
11 #endif |
|
12 |
|
13 #include <windows.h> |
|
14 #include <nscore.h> |
|
15 #include <stdlib.h> |
|
16 |
|
17 namespace mozilla { |
|
18 |
|
19 static void SanitizeEnvironmentVariables() |
|
20 { |
|
21 DWORD bufferSize = GetEnvironmentVariableW(L"PATH", nullptr, 0); |
|
22 if (bufferSize) { |
|
23 wchar_t* originalPath = new wchar_t[bufferSize]; |
|
24 if (bufferSize - 1 == GetEnvironmentVariableW(L"PATH", originalPath, bufferSize)) { |
|
25 bufferSize = ExpandEnvironmentStringsW(originalPath, nullptr, 0); |
|
26 if (bufferSize) { |
|
27 wchar_t* newPath = new wchar_t[bufferSize]; |
|
28 if (ExpandEnvironmentStringsW(originalPath, |
|
29 newPath, |
|
30 bufferSize)) { |
|
31 SetEnvironmentVariableW(L"PATH", newPath); |
|
32 } |
|
33 delete[] newPath; |
|
34 } |
|
35 } |
|
36 delete[] originalPath; |
|
37 } |
|
38 } |
|
39 |
|
40 } |
|
41 |
|
42 #endif |