|
|
@ -54,25 +54,26 @@ int set_core_affinity(int core) { |
|
|
|
namespace util { |
|
|
|
namespace util { |
|
|
|
|
|
|
|
|
|
|
|
std::string read_file(const std::string& fn) { |
|
|
|
std::string read_file(const std::string& fn) { |
|
|
|
std::ifstream f(fn, std::ios::binary | std::ios::in); |
|
|
|
std::ifstream ifs(fn, std::ios::binary | std::ios::ate); |
|
|
|
if (f) { |
|
|
|
if (ifs) { |
|
|
|
f.seekg(0, std::ios::end); |
|
|
|
int pos = ifs.tellg(); |
|
|
|
int pos = f.tellg(); |
|
|
|
|
|
|
|
if (pos > 0) { |
|
|
|
if (pos > 0) { |
|
|
|
std::string result; |
|
|
|
std::string result; |
|
|
|
result.resize(pos); |
|
|
|
result.resize(pos); |
|
|
|
f.seekg(0, std::ios::beg); |
|
|
|
ifs.seekg(0, std::ios::beg); |
|
|
|
if (f.read(result.data(), pos)) { |
|
|
|
ifs.read(result.data(), pos); |
|
|
|
|
|
|
|
if (ifs) { |
|
|
|
return result; |
|
|
|
return result; |
|
|
|
} |
|
|
|
} |
|
|
|
} else { |
|
|
|
|
|
|
|
// fallback for files created on read, e.g. procfs
|
|
|
|
|
|
|
|
std::stringstream buffer; |
|
|
|
|
|
|
|
buffer << f.rdbuf(); |
|
|
|
|
|
|
|
return buffer.str(); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
return std::string(); |
|
|
|
ifs.close(); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// fallback for files created on read, e.g. procfs
|
|
|
|
|
|
|
|
std::ifstream f(fn); |
|
|
|
|
|
|
|
std::stringstream buffer; |
|
|
|
|
|
|
|
buffer << f.rdbuf(); |
|
|
|
|
|
|
|
return buffer.str(); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
int read_files_in_dir(const std::string &path, std::map<std::string, std::string> *contents) { |
|
|
|
int read_files_in_dir(const std::string &path, std::map<std::string, std::string> *contents) { |
|
|
|