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