blob: 634071bc9001e739c99f2e0df23f696ca5ff5f18 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "nsCOMPtr.h"
#include "nsIFile.h"
#include "nsDependentString.h"
#include "prenv.h"
#ifdef XP_WIN
# include <windows.h>
#endif
namespace mozilla {
// Load the path of a file saved with SaveFileToEnv
already_AddRefed<nsIFile> GetFileFromEnv(const char* name) {
nsresult rv;
nsCOMPtr<nsIFile> file;
#ifdef XP_WIN
WCHAR path[_MAX_PATH];
if (!GetEnvironmentVariableW(NS_ConvertASCIItoUTF16(name).get(), path,
_MAX_PATH))
return nullptr;
rv = NS_NewLocalFile(nsDependentString(path), true, getter_AddRefs(file));
if (NS_FAILED(rv)) return nullptr;
return file.forget();
#else
const char* arg = PR_GetEnv(name);
if (!arg || !*arg) {
return nullptr;
}
rv = NS_NewNativeLocalFile(nsDependentCString(arg), true,
getter_AddRefs(file));
if (NS_FAILED(rv)) {
return nullptr;
}
return file.forget();
#endif
}
} // namespace mozilla
|
