Commit e5ff85b6 authored by Max Kellermann's avatar Max Kellermann

storage/nfs: move code to class MemoryStorageDirectoryReader

Read all directory entries into memory and close the struct nfsdir before returning the StorageDirectoryReader instance. This is what libnfs does, anyway.
parent f0bb5b84
...@@ -572,6 +572,7 @@ libstorage_a_SOURCES = \ ...@@ -572,6 +572,7 @@ libstorage_a_SOURCES = \
src/storage/Registry.cxx src/storage/Registry.hxx \ src/storage/Registry.cxx src/storage/Registry.hxx \
src/storage/StorageInterface.cxx src/storage/StorageInterface.hxx \ src/storage/StorageInterface.cxx src/storage/StorageInterface.hxx \
src/storage/CompositeStorage.cxx src/storage/CompositeStorage.hxx \ src/storage/CompositeStorage.cxx src/storage/CompositeStorage.hxx \
src/storage/MemoryDirectoryReader.cxx src/storage/MemoryDirectoryReader.hxx \
src/storage/Configured.cxx src/storage/Configured.hxx \ src/storage/Configured.cxx src/storage/Configured.hxx \
src/storage/plugins/LocalStorage.cxx src/storage/plugins/LocalStorage.hxx \ src/storage/plugins/LocalStorage.cxx src/storage/plugins/LocalStorage.hxx \
src/storage/FileInfo.hxx src/storage/FileInfo.hxx
......
/*
* 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 "MemoryDirectoryReader.hxx"
#include <assert.h>
const char *
MemoryStorageDirectoryReader::Read()
{
if (first)
first = false;
else
entries.pop_front();
if (entries.empty())
return nullptr;
return entries.front().name.c_str();
}
bool
MemoryStorageDirectoryReader::GetInfo(gcc_unused bool follow, FileInfo &info,
gcc_unused Error &error)
{
assert(!first);
assert(!entries.empty());
info = entries.front().info;
return true;
}
/*
* 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_STORAGE_MEMORY_DIRECTORY_READER_HXX
#define MPD_STORAGE_MEMORY_DIRECTORY_READER_HXX
#include "check.h"
#include "StorageInterface.hxx"
#include "FileInfo.hxx"
#include <string>
#include <forward_list>
/**
* A #StorageDirectoryReader implementation that returns directory
* entries from a memory allocation.
*/
class MemoryStorageDirectoryReader final : public StorageDirectoryReader {
public:
struct Entry {
std::string name;
FileInfo info;
template<typename N>
explicit Entry(N &&_name):name(std::forward<N>(_name)) {}
};
typedef std::forward_list<Entry> List;
private:
List entries;
bool first;
public:
MemoryStorageDirectoryReader()
:first(true) {}
MemoryStorageDirectoryReader(MemoryStorageDirectoryReader &&src)
:entries(std::move(src.entries)), first(src.first) {}
MemoryStorageDirectoryReader(List &&_entries)
:entries(std::move(_entries)), first(true) {}
/* virtual methods from class StorageDirectoryReader */
const char *Read() override;
bool GetInfo(bool follow, FileInfo &info, Error &error) override;
};
#endif
...@@ -22,6 +22,7 @@ ...@@ -22,6 +22,7 @@
#include "storage/StoragePlugin.hxx" #include "storage/StoragePlugin.hxx"
#include "storage/StorageInterface.hxx" #include "storage/StorageInterface.hxx"
#include "storage/FileInfo.hxx" #include "storage/FileInfo.hxx"
#include "storage/MemoryDirectoryReader.hxx"
#include "lib/nfs/Domain.hxx" #include "lib/nfs/Domain.hxx"
#include "lib/nfs/Base.hxx" #include "lib/nfs/Base.hxx"
#include "fs/AllocatedPath.hxx" #include "fs/AllocatedPath.hxx"
...@@ -36,31 +37,6 @@ extern "C" { ...@@ -36,31 +37,6 @@ extern "C" {
#include <sys/stat.h> #include <sys/stat.h>
#include <fcntl.h> #include <fcntl.h>
class NfsDirectoryReader final : public StorageDirectoryReader {
const std::string base;
nfs_context *const ctx;
nfsdir *const dir;
nfsdirent *ent;
/**
* Buffer for Read() which holds the current file name
* converted to UTF-8.
*/
std::string name_utf8;
public:
NfsDirectoryReader(const char *_base, nfs_context *_ctx, nfsdir *_dir)
:base(_base), ctx(_ctx), dir(_dir) {}
virtual ~NfsDirectoryReader();
/* virtual methods from class StorageDirectoryReader */
const char *Read() override;
bool GetInfo(bool follow, FileInfo &info, Error &error) override;
};
class NfsStorage final : public Storage { class NfsStorage final : public Storage {
const std::string base; const std::string base;
...@@ -156,23 +132,6 @@ NfsStorage::GetInfo(const char *uri_utf8, gcc_unused bool follow, ...@@ -156,23 +132,6 @@ NfsStorage::GetInfo(const char *uri_utf8, gcc_unused bool follow,
return ::GetInfo(ctx, path.c_str(), info, error); return ::GetInfo(ctx, path.c_str(), info, error);
} }
StorageDirectoryReader *
NfsStorage::OpenDirectory(const char *uri_utf8, Error &error)
{
const std::string path = UriToNfsPath(uri_utf8, error);
if (path.empty())
return nullptr;
nfsdir *dir;
int result = nfs_opendir(ctx, path.c_str(), &dir);
if (result < 0) {
error.SetErrno(-result, "nfs_opendir() failed");
return nullptr;
}
return new NfsDirectoryReader(uri_utf8, ctx, dir);
}
gcc_pure gcc_pure
static bool static bool
SkipNameFS(const char *name) SkipNameFS(const char *name)
...@@ -182,31 +141,6 @@ SkipNameFS(const char *name) ...@@ -182,31 +141,6 @@ SkipNameFS(const char *name)
(name[1] == '.' && name[2] == 0)); (name[1] == '.' && name[2] == 0));
} }
NfsDirectoryReader::~NfsDirectoryReader()
{
nfs_closedir(ctx, dir);
}
const char *
NfsDirectoryReader::Read()
{
while ((ent = nfs_readdir(ctx, dir)) != nullptr) {
const Path name_fs = Path::FromFS(ent->name);
if (SkipNameFS(name_fs.c_str()))
continue;
name_utf8 = name_fs.ToUTF8();
if (name_utf8.empty())
/* ignore files whose name cannot be converted
to UTF-8 */
continue;
return name_utf8.c_str();
}
return nullptr;
}
static void static void
Copy(FileInfo &info, const struct nfsdirent &ent) Copy(FileInfo &info, const struct nfsdirent &ent)
{ {
...@@ -230,14 +164,42 @@ Copy(FileInfo &info, const struct nfsdirent &ent) ...@@ -230,14 +164,42 @@ Copy(FileInfo &info, const struct nfsdirent &ent)
info.inode = ent.inode; info.inode = ent.inode;
} }
bool StorageDirectoryReader *
NfsDirectoryReader::GetInfo(gcc_unused bool follow, FileInfo &info, NfsStorage::OpenDirectory(const char *uri_utf8, Error &error)
gcc_unused Error &error)
{ {
assert(ent != nullptr); const std::string path = UriToNfsPath(uri_utf8, error);
if (path.empty())
return nullptr;
Copy(info, *ent); nfsdir *dir;
return true; int result = nfs_opendir(ctx, path.c_str(), &dir);
if (result < 0) {
error.SetErrno(-result, "nfs_opendir() failed");
return nullptr;
}
MemoryStorageDirectoryReader::List entries;
const struct nfsdirent *ent;
while ((ent = nfs_readdir(ctx, dir)) != nullptr) {
const Path name_fs = Path::FromFS(ent->name);
if (SkipNameFS(name_fs.c_str()))
continue;
std::string name_utf8 = name_fs.ToUTF8();
if (name_utf8.empty())
/* ignore files whose name cannot be converted
to UTF-8 */
continue;
entries.emplace_front(std::move(name_utf8));
Copy(entries.front().info, *ent);
}
nfs_closedir(ctx, dir);
/* don't reverse the list - order does not matter */
return new MemoryStorageDirectoryReader(std::move(entries));
} }
static Storage * static Storage *
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment