nerdcruft/libcruft-util
1
0
Fork 0
Code Issues Releases Activity

io: find the exact storage for the FILE overload

Browse Source
This commit is contained in:
Danny Robson 2016-03-17 18:07:08 +11:00
parent ecb97143cf
commit b864ffa00b
1 changed files with 17 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
io.cpp
View File

@ -73,25 +73,26 @@ util::slurp (const boost::filesystem::path& path) {
std::vector<char> std::vector<char>
util::slurp (FILE *stream) util::slurp (FILE *stream)
{ {
std::vector<char> data; // find how much data is in this file
data.resize (16); const int desc = fileno (stream);
if (desc < 0)
for (size_t chunk = 0, size = data.size ();
!ferror (stream) && !feof (stream);
chunk = size, size *= 2u)
{
auto found = fread (data.data () + chunk, 1, chunk, stream);
if (found != chunk)
break;
data.resize (data.size () * 2u);
}
if (ferror (stream))
errno_error::throw_code (); errno_error::throw_code ();
struct stat meta;
if (fstat (desc, &meta) < 0)
errno_error::throw_code ();
// allocate a buffer we think is the correct size
std::vector<char> buf;
buf.resize (meta.st_size);
// read as much as possible, then resize to the actual length
auto res = fread (buf.data (), 1, meta.st_size, stream);
if (!ferror (stream))
buf.resize (res);
CHECK (feof (stream)); CHECK (feof (stream));
return data; return buf;
} }