fs/io/BufferedReader: new class to replace class TextFile
The new class is pluggable, to prepare for gzipped database files. For now, the TextFile class remains, and will be refactored away later.
This commit is contained in:
parent
0ea66a1275
commit
aa2e4d92e0
|
@ -508,6 +508,7 @@ FS_LIBS = libfs.a
|
||||||
libfs_a_SOURCES = \
|
libfs_a_SOURCES = \
|
||||||
src/fs/io/Reader.hxx \
|
src/fs/io/Reader.hxx \
|
||||||
src/fs/io/FileReader.cxx src/fs/io/FileReader.hxx \
|
src/fs/io/FileReader.cxx src/fs/io/FileReader.hxx \
|
||||||
|
src/fs/io/BufferedReader.cxx src/fs/io/BufferedReader.hxx \
|
||||||
src/fs/io/TextFile.cxx src/fs/io/TextFile.hxx \
|
src/fs/io/TextFile.cxx src/fs/io/TextFile.hxx \
|
||||||
src/fs/io/OutputStream.hxx \
|
src/fs/io/OutputStream.hxx \
|
||||||
src/fs/io/StdoutOutputStream.hxx \
|
src/fs/io/StdoutOutputStream.hxx \
|
||||||
|
|
|
@ -116,6 +116,29 @@ spl_map_to_fs(const char *name_utf8, Error &error)
|
||||||
return path_fs;
|
return path_fs;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
gcc_pure
|
||||||
|
static bool
|
||||||
|
IsNotFoundError(const Error &error)
|
||||||
|
{
|
||||||
|
#ifdef WIN32
|
||||||
|
return error.IsDomain(win32_domain) &&
|
||||||
|
error.GetCode() == ERROR_FILE_NOT_FOUND;
|
||||||
|
#else
|
||||||
|
return error.IsDomain(errno_domain) &&
|
||||||
|
error.GetCode() == ENOENT;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
TranslatePlaylistError(Error &error)
|
||||||
|
{
|
||||||
|
if (IsNotFoundError(error)) {
|
||||||
|
error.Clear();
|
||||||
|
error.Set(playlist_domain, int(PlaylistResult::NO_SUCH_LIST),
|
||||||
|
"No such playlist");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Create an #Error for the current errno.
|
* Create an #Error for the current errno.
|
||||||
*/
|
*/
|
||||||
|
@ -228,9 +251,9 @@ LoadPlaylistFile(const char *utf8path, Error &error)
|
||||||
if (path_fs.IsNull())
|
if (path_fs.IsNull())
|
||||||
return contents;
|
return contents;
|
||||||
|
|
||||||
TextFile file(path_fs);
|
TextFile file(path_fs, error);
|
||||||
if (file.HasFailed()) {
|
if (file.HasFailed()) {
|
||||||
playlist_errno(error);
|
TranslatePlaylistError(error);
|
||||||
return contents;
|
return contents;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -103,10 +103,10 @@ StateFile::Read()
|
||||||
|
|
||||||
FormatDebug(state_file_domain, "Loading state file %s", path_utf8.c_str());
|
FormatDebug(state_file_domain, "Loading state file %s", path_utf8.c_str());
|
||||||
|
|
||||||
TextFile file(path);
|
Error error;
|
||||||
|
TextFile file(path, error);
|
||||||
if (file.HasFailed()) {
|
if (file.HasFailed()) {
|
||||||
FormatErrno(state_file_domain, "failed to open %s",
|
LogError(error);
|
||||||
path_utf8.c_str());
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -168,12 +168,9 @@ SimpleDatabase::Load(Error &error)
|
||||||
assert(!path.IsNull());
|
assert(!path.IsNull());
|
||||||
assert(root != nullptr);
|
assert(root != nullptr);
|
||||||
|
|
||||||
TextFile file(path);
|
TextFile file(path, error);
|
||||||
if (file.HasFailed()) {
|
if (file.HasFailed())
|
||||||
error.FormatErrno("Failed to open database file \"%s\"",
|
|
||||||
path_utf8.c_str());
|
|
||||||
return false;
|
return false;
|
||||||
}
|
|
||||||
|
|
||||||
if (!db_load_internal(file, *root, error))
|
if (!db_load_internal(file, *root, error))
|
||||||
return false;
|
return false;
|
||||||
|
|
|
@ -40,6 +40,7 @@
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifdef USE_XDG
|
#ifdef USE_XDG
|
||||||
|
#include "util/Error.hxx"
|
||||||
#include "util/StringUtil.hxx"
|
#include "util/StringUtil.hxx"
|
||||||
#include "io/TextFile.hxx"
|
#include "io/TextFile.hxx"
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
@ -204,7 +205,7 @@ static AllocatedPath GetUserDir(const char *name)
|
||||||
if (config_dir.IsNull())
|
if (config_dir.IsNull())
|
||||||
return result;
|
return result;
|
||||||
auto dirs_file = AllocatedPath::Build(config_dir, "user-dirs.dirs");
|
auto dirs_file = AllocatedPath::Build(config_dir, "user-dirs.dirs");
|
||||||
TextFile input(dirs_file);
|
TextFile input(dirs_file, IgnoreError());
|
||||||
if (input.HasFailed())
|
if (input.HasFailed())
|
||||||
return result;
|
return result;
|
||||||
char *line;
|
char *line;
|
||||||
|
|
|
@ -0,0 +1,82 @@
|
||||||
|
/*
|
||||||
|
* Copyright (C) 2003-2014 The Music Player Daemon Project
|
||||||
|
* http://www.musicpd.org
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License along
|
||||||
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
||||||
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "config.h"
|
||||||
|
#include "BufferedReader.hxx"
|
||||||
|
#include "Reader.hxx"
|
||||||
|
#include "util/TextFile.hxx"
|
||||||
|
|
||||||
|
bool
|
||||||
|
BufferedReader::Fill(bool need_more)
|
||||||
|
{
|
||||||
|
if (gcc_unlikely(last_error.IsDefined()))
|
||||||
|
return false;
|
||||||
|
|
||||||
|
if (eof)
|
||||||
|
return !need_more;
|
||||||
|
|
||||||
|
auto w = buffer.Write();
|
||||||
|
if (w.IsEmpty()) {
|
||||||
|
if (buffer.GetCapacity() >= MAX_SIZE)
|
||||||
|
return !need_more;
|
||||||
|
|
||||||
|
buffer.Grow(buffer.GetCapacity() * 2);
|
||||||
|
w = buffer.Write();
|
||||||
|
assert(!w.IsEmpty());
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t nbytes = reader.Read(w.data, w.size, last_error);
|
||||||
|
if (nbytes == 0) {
|
||||||
|
if (gcc_unlikely(last_error.IsDefined()))
|
||||||
|
return false;
|
||||||
|
|
||||||
|
eof = true;
|
||||||
|
return !need_more;
|
||||||
|
}
|
||||||
|
|
||||||
|
buffer.Append(nbytes);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
char *
|
||||||
|
BufferedReader::ReadLine()
|
||||||
|
{
|
||||||
|
do {
|
||||||
|
char *line = ReadBufferedLine(buffer);
|
||||||
|
if (line != nullptr)
|
||||||
|
return line;
|
||||||
|
} while (Fill(true));
|
||||||
|
|
||||||
|
if (last_error.IsDefined() || !eof || buffer.IsEmpty())
|
||||||
|
return nullptr;
|
||||||
|
|
||||||
|
auto w = buffer.Write();
|
||||||
|
if (w.IsEmpty()) {
|
||||||
|
buffer.Grow(buffer.GetCapacity() + 1);
|
||||||
|
w = buffer.Write();
|
||||||
|
assert(!w.IsEmpty());
|
||||||
|
}
|
||||||
|
|
||||||
|
/* terminate the last line */
|
||||||
|
w[0] = 0;
|
||||||
|
|
||||||
|
char *line = buffer.Read().data;
|
||||||
|
buffer.Clear();
|
||||||
|
return line;
|
||||||
|
}
|
|
@ -0,0 +1,75 @@
|
||||||
|
/*
|
||||||
|
* Copyright (C) 2003-2014 The Music Player Daemon Project
|
||||||
|
* http://www.musicpd.org
|
||||||
|
*
|
||||||
|
* This program is free software; you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation; either version 2 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This program is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License along
|
||||||
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
||||||
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef MPD_BUFFERED_READER_HXX
|
||||||
|
#define MPD_BUFFERED_READER_HXX
|
||||||
|
|
||||||
|
#include "check.h"
|
||||||
|
#include "Compiler.h"
|
||||||
|
#include "util/DynamicFifoBuffer.hxx"
|
||||||
|
#include "util/Error.hxx"
|
||||||
|
|
||||||
|
#include <stddef.h>
|
||||||
|
|
||||||
|
class Reader;
|
||||||
|
class Error;
|
||||||
|
|
||||||
|
class BufferedReader {
|
||||||
|
static constexpr size_t MAX_SIZE = 512 * 1024;
|
||||||
|
|
||||||
|
Reader &reader;
|
||||||
|
|
||||||
|
DynamicFifoBuffer<char> buffer;
|
||||||
|
|
||||||
|
Error last_error;
|
||||||
|
|
||||||
|
bool eof;
|
||||||
|
|
||||||
|
public:
|
||||||
|
BufferedReader(Reader &_reader)
|
||||||
|
:reader(_reader), buffer(4096), eof(false) {}
|
||||||
|
|
||||||
|
gcc_pure
|
||||||
|
bool Check() const {
|
||||||
|
return !last_error.IsDefined();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool Check(Error &error) const {
|
||||||
|
if (last_error.IsDefined()) {
|
||||||
|
error.Set(last_error);
|
||||||
|
return false;
|
||||||
|
} else
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool Fill(bool need_more);
|
||||||
|
|
||||||
|
gcc_pure
|
||||||
|
WritableBuffer<void> Read() const {
|
||||||
|
return buffer.Read().ToVoid();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Consume(size_t n) {
|
||||||
|
buffer.Consume(n);
|
||||||
|
}
|
||||||
|
|
||||||
|
char *ReadLine();
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
|
@ -19,63 +19,30 @@
|
||||||
|
|
||||||
#include "config.h"
|
#include "config.h"
|
||||||
#include "TextFile.hxx"
|
#include "TextFile.hxx"
|
||||||
#include "util/Alloc.hxx"
|
#include "FileReader.hxx"
|
||||||
|
#include "BufferedReader.hxx"
|
||||||
#include "fs/Path.hxx"
|
#include "fs/Path.hxx"
|
||||||
#include "fs/FileSystem.hxx"
|
|
||||||
|
|
||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
#include <string.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
|
|
||||||
TextFile::TextFile(Path path_fs)
|
TextFile::TextFile(Path path_fs, Error &error)
|
||||||
:file(FOpen(path_fs, FOpenMode::ReadText)),
|
:file_reader(new FileReader(path_fs, error)),
|
||||||
buffer((char *)xalloc(step)), capacity(step), length(0) {}
|
buffered_reader(file_reader->IsDefined()
|
||||||
|
? new BufferedReader(*file_reader)
|
||||||
|
: nullptr)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
TextFile::~TextFile()
|
TextFile::~TextFile()
|
||||||
{
|
{
|
||||||
free(buffer);
|
delete buffered_reader;
|
||||||
|
delete file_reader;
|
||||||
if (file != nullptr)
|
|
||||||
fclose(file);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
char *
|
char *
|
||||||
TextFile::ReadLine()
|
TextFile::ReadLine()
|
||||||
{
|
{
|
||||||
assert(file != nullptr);
|
assert(buffered_reader != nullptr);
|
||||||
|
|
||||||
while (true) {
|
return buffered_reader->ReadLine();
|
||||||
if (length >= capacity) {
|
|
||||||
if (capacity >= max_length)
|
|
||||||
/* too large already - bail out */
|
|
||||||
return nullptr;
|
|
||||||
|
|
||||||
capacity <<= 1;
|
|
||||||
char *new_buffer = (char *)realloc(buffer, capacity);
|
|
||||||
if (new_buffer == nullptr)
|
|
||||||
/* out of memory - bail out */
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
char *p = fgets(buffer + length, capacity - length, file);
|
|
||||||
if (p == nullptr) {
|
|
||||||
if (length == 0 || ferror(file))
|
|
||||||
return nullptr;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
length += strlen(buffer + length);
|
|
||||||
if (buffer[length - 1] == '\n')
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* remove the newline characters */
|
|
||||||
if (buffer[length - 1] == '\n')
|
|
||||||
--length;
|
|
||||||
if (buffer[length - 1] == '\r')
|
|
||||||
--length;
|
|
||||||
|
|
||||||
buffer[length] = 0;
|
|
||||||
length = 0;
|
|
||||||
return buffer;
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -22,29 +22,26 @@
|
||||||
|
|
||||||
#include "Compiler.h"
|
#include "Compiler.h"
|
||||||
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <stddef.h>
|
#include <stddef.h>
|
||||||
|
|
||||||
class Path;
|
class Path;
|
||||||
|
class Error;
|
||||||
|
class FileReader;
|
||||||
|
class BufferedReader;
|
||||||
|
|
||||||
class TextFile {
|
class TextFile {
|
||||||
static constexpr size_t max_length = 512 * 1024;
|
FileReader *const file_reader;
|
||||||
static constexpr size_t step = 1024;
|
BufferedReader *const buffered_reader;
|
||||||
|
|
||||||
FILE *const file;
|
|
||||||
|
|
||||||
char *buffer;
|
|
||||||
size_t capacity, length;
|
|
||||||
|
|
||||||
public:
|
public:
|
||||||
TextFile(Path path_fs);
|
TextFile(Path path_fs, Error &error);
|
||||||
|
|
||||||
TextFile(const TextFile &other) = delete;
|
TextFile(const TextFile &other) = delete;
|
||||||
|
|
||||||
~TextFile();
|
~TextFile();
|
||||||
|
|
||||||
bool HasFailed() const {
|
bool HasFailed() const {
|
||||||
return gcc_unlikely(file == nullptr);
|
return gcc_unlikely(buffered_reader == nullptr);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -53,7 +50,6 @@ public:
|
||||||
* prevent denial of service.
|
* prevent denial of service.
|
||||||
*
|
*
|
||||||
* @param file the source file, opened in text mode
|
* @param file the source file, opened in text mode
|
||||||
* @param buffer an allocator for the buffer
|
|
||||||
* @return a pointer to the line, or nullptr on end-of-file or error
|
* @return a pointer to the line, or nullptr on end-of-file or error
|
||||||
*/
|
*/
|
||||||
char *ReadLine();
|
char *ReadLine();
|
||||||
|
|
Loading…
Reference in New Issue