valitydev/osquery-1
14
0
Fork 0
mirror of https://github.com/valitydev/osquery-1.git synced 2024-11-07 18:08:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

read the file directly into a stringstream buffer

Browse Source
This commit is contained in:
Gabriele Carrettoni 2014-11-09 16:57:12 +01:00
parent 848bd4d96e
commit 77b521ce7b
1 changed files with 8 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
osquery/filesystem/filesystem.cpp
View File

@ -59,27 +59,21 @@ Status readFile(const boost::filesystem::path& path, std::string& content) {
int statusCode = 0;
std::string statusMessage = "OK";
std::unique_ptr<char> buffer;
std::stringstream buffer;
fs::ifstream file_h(path);
if (file_h) {
file_h.seekg(0, file_h.end);
int len = file_h.tellg();
file_h.seekg(0, file_h.beg);
buffer = std::unique_ptr<char>(new char[len]);
file_h.read(buffer.get(), len);
if (!file_h) {
if (file_h.is_open()) {
buffer << file_h.rdbuf();
if (file_h.bad()) {
statusCode = 1;
statusMessage = "Could not read file";
}
content.assign(buffer.get(), len);
} else
content.assign(std::move(buffer.str()));
} else {
statusCode = 1;
statusMessage = "Could not open file for reading";
if (file_h) {
file_h.close();
}
}
return Status(statusCode, statusMessage);
}