forked from ethereum/aleth
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathFileSystem.cpp
More file actions
118 lines (101 loc) · 2.98 KB
/
FileSystem.cpp
File metadata and controls
118 lines (101 loc) · 2.98 KB
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
/*
This file is part of cpp-ethereum.
cpp-ethereum is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
cpp-ethereum is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
*/
/** @file FileSystem.cpp
* @authors
* Eric Lombrozo <elombrozo@gmail.com>
* Gav Wood <i@gavwood.com>
* @date 2014
*/
#include "FileSystem.h"
#include "Common.h"
#include "Log.h"
#if defined(_WIN32)
#include <shlobj.h>
#else
#include <stdlib.h>
#include <stdio.h>
#include <pwd.h>
#include <unistd.h>
#endif
#include <boost/filesystem.hpp>
using namespace std;
using namespace dev;
namespace fs = boost::filesystem;
static_assert(BOOST_VERSION >= 106400, "Wrong boost headers version");
// Should be written to only once during startup
static fs::path s_ethereumDatadir;
static fs::path s_ethereumIpcPath;
void dev::setDataDir(fs::path const& _dataDir)
{
s_ethereumDatadir = _dataDir;
}
void dev::setIpcPath(fs::path const& _ipcDir)
{
s_ethereumIpcPath = _ipcDir;
}
fs::path dev::getIpcPath()
{
// Strip "geth.ipc" suffix if provided.
if (s_ethereumIpcPath.filename() == "geth.ipc")
return s_ethereumIpcPath.parent_path();
else
return s_ethereumIpcPath;
}
fs::path dev::getDataDir(string _prefix)
{
if (_prefix.empty())
_prefix = "ethereum";
if (_prefix == "ethereum" && !s_ethereumDatadir.empty())
return s_ethereumDatadir;
return getDefaultDataDir(_prefix);
}
fs::path dev::getDefaultDataDir(string _prefix)
{
if (_prefix.empty())
_prefix = "ethereum";
#if defined(_WIN32)
_prefix[0] = toupper(_prefix[0]);
char path[1024] = "";
if (SHGetSpecialFolderPathA(NULL, path, CSIDL_APPDATA, true))
return fs::path(path) / _prefix;
else
{
#ifndef _MSC_VER // todo?
cwarn << "getDataDir(): SHGetSpecialFolderPathA() failed.";
#endif
BOOST_THROW_EXCEPTION(std::runtime_error("getDataDir() - SHGetSpecialFolderPathA() failed."));
}
#else
fs::path dataDirPath;
char const* homeDir = getenv("HOME");
if (!homeDir || strlen(homeDir) == 0)
{
struct passwd* pwd = getpwuid(getuid());
if (pwd)
homeDir = pwd->pw_dir;
}
if (!homeDir || strlen(homeDir) == 0)
dataDirPath = fs::path("/");
else
dataDirPath = fs::path(homeDir);
return dataDirPath / ("." + _prefix);
#endif
}
fs::path dev::appendToFilename(fs::path const& _orig, string const& _suffix)
{
if (_orig.filename() == fs::path(".") || _orig.filename() == fs::path(".."))
return _orig / fs::path(_suffix);
else
return _orig.parent_path() / fs::path( _orig.filename().string() + _suffix);
}