input/file: use class FileReader

This commit is contained in:
Max Kellermann 2015-03-03 18:10:02 +01:00
parent f677f42be7
commit 57ad3aca8e
2 changed files with 23 additions and 30 deletions

View File

@ -78,6 +78,12 @@ public:
#endif #endif
} }
#ifndef WIN32
FileDescriptor GetFD() const {
return fd;
}
#endif
void Close(); void Close();
bool GetFileInfo(FileInfo &info, Error &error) const; bool GetFileInfo(FileInfo &info, Error &error) const;

View File

@ -24,6 +24,8 @@
#include "util/Error.hxx" #include "util/Error.hxx"
#include "util/Domain.hxx" #include "util/Domain.hxx"
#include "fs/Path.hxx" #include "fs/Path.hxx"
#include "fs/FileInfo.hxx"
#include "fs/io/FileReader.hxx"
#include "system/FileDescriptor.hxx" #include "system/FileDescriptor.hxx"
#include <sys/stat.h> #include <sys/stat.h>
@ -33,22 +35,18 @@
static constexpr Domain file_domain("file"); static constexpr Domain file_domain("file");
class FileInputStream final : public InputStream { class FileInputStream final : public InputStream {
FileDescriptor fd; FileReader reader;
public: public:
FileInputStream(const char *path, FileDescriptor _fd, off_t _size, FileInputStream(const char *path, FileReader &&_reader, off_t _size,
Mutex &_mutex, Cond &_cond) Mutex &_mutex, Cond &_cond)
:InputStream(path, _mutex, _cond), :InputStream(path, _mutex, _cond),
fd(_fd) { reader(std::move(_reader)) {
size = _size; size = _size;
seekable = true; seekable = true;
SetReady(); SetReady();
} }
~FileInputStream() {
fd.Close();
}
/* virtual methods from InputStream */ /* virtual methods from InputStream */
bool IsEOF() override { bool IsEOF() override {
@ -64,33 +62,27 @@ OpenFileInputStream(Path path,
Mutex &mutex, Cond &cond, Mutex &mutex, Cond &cond,
Error &error) Error &error)
{ {
FileDescriptor fd; FileReader reader(path, error);
if (!fd.OpenReadOnly(path.c_str())) { if (!reader.IsDefined())
error.FormatErrno("Failed to open \"%s\"",
path.c_str());
return nullptr; return nullptr;
}
struct stat st; FileInfo info;
if (fstat(fd.Get(), &st) < 0) { if (!reader.GetFileInfo(info, error))
error.FormatErrno("Failed to stat \"%s\"", path.c_str());
fd.Close();
return nullptr; return nullptr;
}
if (!S_ISREG(st.st_mode)) { if (!info.IsRegular()) {
error.Format(file_domain, "Not a regular file: %s", error.Format(file_domain, "Not a regular file: %s",
path.c_str()); path.c_str());
fd.Close();
return nullptr; return nullptr;
} }
#ifdef POSIX_FADV_SEQUENTIAL #ifdef POSIX_FADV_SEQUENTIAL
posix_fadvise(fd.Get(), (off_t)0, st.st_size, POSIX_FADV_SEQUENTIAL); posix_fadvise(reader.GetFD().Get(), (off_t)0, info.GetSize(),
POSIX_FADV_SEQUENTIAL);
#endif #endif
return new FileInputStream(path.ToUTF8().c_str(), return new FileInputStream(path.ToUTF8().c_str(),
fd, st.st_size, std::move(reader), info.GetSize(),
mutex, cond); mutex, cond);
} }
@ -107,24 +99,19 @@ input_file_open(gcc_unused const char *filename,
bool bool
FileInputStream::Seek(offset_type new_offset, Error &error) FileInputStream::Seek(offset_type new_offset, Error &error)
{ {
auto result = fd.Seek((off_t)new_offset); if (!reader.Seek((off_t)new_offset, error))
if (result < 0) {
error.SetErrno("Failed to seek");
return false; return false;
}
offset = (offset_type)result; offset = new_offset;
return true; return true;
} }
size_t size_t
FileInputStream::Read(void *ptr, size_t read_size, Error &error) FileInputStream::Read(void *ptr, size_t read_size, Error &error)
{ {
ssize_t nbytes = fd.Read(ptr, read_size); ssize_t nbytes = reader.Read(ptr, read_size, error);
if (nbytes < 0) { if (nbytes < 0)
error.SetErrno("Failed to read");
return 0; return 0;
}
offset += nbytes; offset += nbytes;
return (size_t)nbytes; return (size_t)nbytes;