mirror of https://github.com/PurpleI2P/i2pd.git
orignal
9 years ago
30 changed files with 745 additions and 671 deletions
@ -0,0 +1,158 @@ |
|||||||
|
/*
|
||||||
|
* Copyright (c) 2013-2016, The PurpleI2P Project |
||||||
|
* |
||||||
|
* This file is part of Purple i2pd project and licensed under BSD3 |
||||||
|
* |
||||||
|
* See full license text in LICENSE file at top of project tree |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <algorithm> |
||||||
|
#include <boost/filesystem.hpp> |
||||||
|
|
||||||
|
#ifdef WIN32 |
||||||
|
#include <shlobj.h> |
||||||
|
#endif |
||||||
|
|
||||||
|
#include "Base.h" |
||||||
|
#include "FS.h" |
||||||
|
#include "Log.h" |
||||||
|
|
||||||
|
namespace i2p { |
||||||
|
namespace fs { |
||||||
|
std::string appName = "i2pd"; |
||||||
|
std::string dataDir = ""; |
||||||
|
#ifdef _WIN32 |
||||||
|
std::string dirSep = "\\"; |
||||||
|
#else |
||||||
|
std::string dirSep = "/"; |
||||||
|
#endif |
||||||
|
|
||||||
|
const std::string & GetAppName () { |
||||||
|
return appName; |
||||||
|
} |
||||||
|
|
||||||
|
void SetAppName (const std::string& name) { |
||||||
|
appName = name; |
||||||
|
} |
||||||
|
|
||||||
|
const std::string & GetDataDir () { |
||||||
|
return dataDir; |
||||||
|
} |
||||||
|
|
||||||
|
void DetectDataDir(const std::string & cmdline_param, bool isService) { |
||||||
|
if (cmdline_param != "") { |
||||||
|
dataDir = cmdline_param; |
||||||
|
return; |
||||||
|
} |
||||||
|
#if defined(WIN32) || defined(_WIN32) |
||||||
|
char localAppData[MAX_PATH]; |
||||||
|
SHGetFolderPath(NULL, CSIDL_APPDATA, 0, NULL, localAppData); |
||||||
|
dataDir = std::string(localAppData) + "\\" + appName; |
||||||
|
return; |
||||||
|
#elif defined(MAC_OSX) |
||||||
|
char *home = getenv("HOME"); |
||||||
|
dataDir = (home != NULL && strlen(home) > 0) ? home : ""; |
||||||
|
dataDir += "/Library/Application Support/" + appName; |
||||||
|
return; |
||||||
|
#else /* other unix */ |
||||||
|
char *home = getenv("HOME"); |
||||||
|
if (isService) { |
||||||
|
dataDir = "/var/lib/" + appName; |
||||||
|
} else if (home != NULL && strlen(home) > 0) { |
||||||
|
dataDir = std::string(home) + "/." + appName; |
||||||
|
} else { |
||||||
|
dataDir = "/tmp/" + appName; |
||||||
|
} |
||||||
|
return; |
||||||
|
#endif |
||||||
|
} |
||||||
|
|
||||||
|
bool Init() { |
||||||
|
if (boost::filesystem::exists(dataDir)) |
||||||
|
boost::filesystem::create_directory(dataDir); |
||||||
|
std::string destinations = DataDirPath("destinations"); |
||||||
|
if (boost::filesystem::exists(destinations)) |
||||||
|
boost::filesystem::create_directory(destinations); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
bool ReadDir(const std::string & path, std::vector<std::string> & files) { |
||||||
|
if (!boost::filesystem::exists(path)) |
||||||
|
return false; |
||||||
|
boost::filesystem::directory_iterator it(path); |
||||||
|
boost::filesystem::directory_iterator end; |
||||||
|
|
||||||
|
for ( ; it != end; it++) { |
||||||
|
if (!boost::filesystem::is_regular_file(it->status())) |
||||||
|
continue; |
||||||
|
files.push_back(it->path().string()); |
||||||
|
} |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
bool Exists(const std::string & path) { |
||||||
|
return boost::filesystem::exists(path); |
||||||
|
} |
||||||
|
|
||||||
|
bool Remove(const std::string & path) { |
||||||
|
if (!boost::filesystem::exists(path)) |
||||||
|
return false; |
||||||
|
return boost::filesystem::remove(path); |
||||||
|
} |
||||||
|
|
||||||
|
void HashedStorage::SetPlace(const std::string &path) { |
||||||
|
root = path + i2p::fs::dirSep + name; |
||||||
|
} |
||||||
|
|
||||||
|
bool HashedStorage::Init(const char * chars, size_t count) { |
||||||
|
if (!boost::filesystem::exists(root)) { |
||||||
|
boost::filesystem::create_directory(root); |
||||||
|
} |
||||||
|
|
||||||
|
for (size_t i = 0; i < count; i++) { |
||||||
|
auto p = root + i2p::fs::dirSep + prefix1 + chars[i]; |
||||||
|
if (boost::filesystem::exists(p)) |
||||||
|
continue; |
||||||
|
if (boost::filesystem::create_directory(p)) |
||||||
|
continue; /* ^ throws exception on failure */ |
||||||
|
return false; |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
std::string HashedStorage::Path(const std::string & ident) const { |
||||||
|
std::string safe_ident = ident; |
||||||
|
std::replace(safe_ident.begin(), safe_ident.end(), '/', '-'); |
||||||
|
std::replace(safe_ident.begin(), safe_ident.end(), '\\', '-'); |
||||||
|
|
||||||
|
std::stringstream t(""); |
||||||
|
t << this->root << i2p::fs::dirSep; |
||||||
|
t << prefix1 << safe_ident[0] << i2p::fs::dirSep; |
||||||
|
t << prefix2 << safe_ident << "." << suffix; |
||||||
|
|
||||||
|
return t.str(); |
||||||
|
} |
||||||
|
|
||||||
|
void HashedStorage::Remove(const std::string & ident) { |
||||||
|
std::string path = Path(ident); |
||||||
|
if (!boost::filesystem::exists(path)) |
||||||
|
return; |
||||||
|
boost::filesystem::remove(path); |
||||||
|
} |
||||||
|
|
||||||
|
void HashedStorage::Traverse(std::vector<std::string> & files) { |
||||||
|
boost::filesystem::path p(root); |
||||||
|
boost::filesystem::recursive_directory_iterator it(p); |
||||||
|
boost::filesystem::recursive_directory_iterator end; |
||||||
|
|
||||||
|
for ( ; it != end; it++) { |
||||||
|
if (!boost::filesystem::is_regular_file( it->status() )) |
||||||
|
continue; |
||||||
|
const std::string & t = it->path().string(); |
||||||
|
files.push_back(t); |
||||||
|
} |
||||||
|
} |
||||||
|
} // fs
|
||||||
|
} // i2p
|
||||||
@ -0,0 +1,142 @@ |
|||||||
|
/*
|
||||||
|
* Copyright (c) 2013-2016, The PurpleI2P Project |
||||||
|
* |
||||||
|
* This file is part of Purple i2pd project and licensed under BSD3 |
||||||
|
* |
||||||
|
* See full license text in LICENSE file at top of project tree |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef FS_H__ |
||||||
|
#define FS_H__ |
||||||
|
|
||||||
|
#include <vector> |
||||||
|
#include <string> |
||||||
|
#include <iostream> |
||||||
|
#include <sstream> |
||||||
|
|
||||||
|
namespace i2p { |
||||||
|
namespace fs { |
||||||
|
extern std::string dirSep; |
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Class to work with NetDb & Router profiles |
||||||
|
* |
||||||
|
* Usage: |
||||||
|
* |
||||||
|
* const char alphabet[8] = {'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'}; |
||||||
|
* auto h = HashedStorage("name", "y", "z-", ".txt"); |
||||||
|
* h.SetPlace("/tmp/hs-test"); |
||||||
|
* h.GetName() -> gives "name" |
||||||
|
* h.GetRoot() -> gives "/tmp/hs-test/name" |
||||||
|
* h.Init(alphabet, 8); <- creates needed dirs, 8 is size of alphabet |
||||||
|
* h.Path("abcd"); <- returns /tmp/hs-test/name/ya/z-abcd.txt |
||||||
|
* h.Remove("abcd"); <- removes /tmp/hs-test/name/ya/z-abcd.txt, if it exists |
||||||
|
* std::vector<std::string> files; |
||||||
|
* h.Traverse(files); <- finds all files in storage and saves in given vector |
||||||
|
*/ |
||||||
|
class HashedStorage { |
||||||
|
protected: |
||||||
|
std::string root; /**< path to storage with it's name included */ |
||||||
|
std::string name; /**< name of the storage */ |
||||||
|
std::string prefix1; /**< hashed directory prefix */ |
||||||
|
std::string prefix2; /**< prefix of file in storage */ |
||||||
|
std::string suffix; /**< suffix of file in storage (extension) */ |
||||||
|
|
||||||
|
public: |
||||||
|
HashedStorage(const char *n, const char *p1, const char *p2, const char *s): |
||||||
|
name(n), prefix1(p1), prefix2(p2), suffix(s) {}; |
||||||
|
|
||||||
|
/** create subdirs in storage */ |
||||||
|
bool Init(const char* chars, size_t cnt); |
||||||
|
const std::string & GetRoot() const { return this->root; } |
||||||
|
const std::string & GetName() const { return this->name; } |
||||||
|
/** set directory where to place storage directory */ |
||||||
|
void SetPlace(const std::string & path); |
||||||
|
/** path to file with given ident */ |
||||||
|
std::string Path(const std::string & ident) const; |
||||||
|
/** remove file by ident */ |
||||||
|
void Remove(const std::string & ident); |
||||||
|
/** find all files in storage and store list in provided vector */ |
||||||
|
void Traverse(std::vector<std::string> & files); |
||||||
|
}; |
||||||
|
|
||||||
|
/** @brief Returns current application name, default 'i2pd' */ |
||||||
|
const std::string & GetAppName (); |
||||||
|
/** @brief Set applicaton name, affects autodetection of datadir */ |
||||||
|
void SetAppName (const std::string& name); |
||||||
|
|
||||||
|
/** @brief Returns datadir path */ |
||||||
|
const std::string & GetDataDir(); |
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Set datadir either from cmdline option or using autodetection |
||||||
|
* @param cmdline_param Value of cmdline parameter --datadir=<something> |
||||||
|
* @param isService Value of cmdline parameter --service |
||||||
|
* |
||||||
|
* Examples of autodetected paths: |
||||||
|
* |
||||||
|
* Windows < Vista: C:\Documents and Settings\Username\Application Data\i2pd\ |
||||||
|
* Windows >= Vista: C:\Users\Username\AppData\Roaming\i2pd\ |
||||||
|
* Mac: /Library/Application Support/i2pd/ or ~/Library/Application Support/i2pd/ |
||||||
|
* Unix: /var/lib/i2pd/ (system=1) >> ~/.i2pd/ or /tmp/i2pd/ |
||||||
|
*/ |
||||||
|
void DetectDataDir(const std::string & cmdline_datadir, bool isService = false); |
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Create subdirectories inside datadir |
||||||
|
*/ |
||||||
|
bool Init(); |
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Get list of files in directory |
||||||
|
* @param path Path to directory |
||||||
|
* @param files Vector to store found files |
||||||
|
* @return true on success and false if directory not exists |
||||||
|
*/ |
||||||
|
bool ReadDir(const std::string & path, std::vector<std::string> & files); |
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Remove file with given path |
||||||
|
* @param path Absolute path to file |
||||||
|
* @return true on success, false if file not exists, throws exception on error |
||||||
|
*/ |
||||||
|
bool Remove(const std::string & path); |
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Check existence of file |
||||||
|
* @param path Absolute path to file |
||||||
|
* @return true if file exists, false otherwise |
||||||
|
*/ |
||||||
|
bool Exists(const std::string & path); |
||||||
|
|
||||||
|
template<typename T> |
||||||
|
void _ExpandPath(std::stringstream & path, T c) { |
||||||
|
path << i2p::fs::dirSep << c; |
||||||
|
} |
||||||
|
|
||||||
|
template<typename T, typename ... Other> |
||||||
|
void _ExpandPath(std::stringstream & path, T c, Other ... other) { |
||||||
|
_ExpandPath(path, c); |
||||||
|
_ExpandPath(path, other ...); |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Get path relative to datadir |
||||||
|
* |
||||||
|
* Examples (with datadir = "/tmp/i2pd"): |
||||||
|
* |
||||||
|
* i2p::fs::Path("test") -> '/tmp/i2pd/test' |
||||||
|
* i2p::fs::Path("test", "file.txt") -> '/tmp/i2pd/test/file.txt' |
||||||
|
*/ |
||||||
|
template<typename ... Other> |
||||||
|
std::string DataDirPath(Other ... components) { |
||||||
|
std::stringstream s(""); |
||||||
|
s << i2p::fs::GetDataDir(); |
||||||
|
_ExpandPath(s, components ...); |
||||||
|
|
||||||
|
return s.str(); |
||||||
|
} |
||||||
|
} // fs
|
||||||
|
} // i2p
|
||||||
|
|
||||||
|
#endif // /* FS_H__ */
|
||||||
Loading…
Reference in new issue