Commit 5c4a42ca authored by Max Kellermann's avatar Max Kellermann

neighbor: new subsystem to detect file servers on the local network

This commit adds the NeighborPlugin API which can be used to detect nearby file servers that can be used by input plugins. This list of servers is exported using the new "listneighbors" command. The idle even "neighbor" notifies interested clients when a new neighbor is found or an existing one is lost. There's a lot missing currently: protocol&user documentation, and a way to "mount" remote servers into the music database. Obviously, some code from the UPnP database plugin can be moved to a neighbor plugin.
parent e8477885
......@@ -17,6 +17,7 @@ noinst_LIBRARIES = \
libtag.a \
libinput.a \
libfs.a \
libneighbor.a \
libdb_plugins.a \
libplaylist_plugins.a \
libdecoder_plugins.a \
......@@ -30,6 +31,7 @@ src_mpd_CPPFLAGS = $(AM_CPPFLAGS) \
$(LIBWRAP_CFLAGS) \
$(SQLITE_CFLAGS)
src_mpd_LDADD = \
$(NEIGHBOR_LIBS) \
$(DB_LIBS) \
$(PLAYLIST_LIBS) \
$(AVAHI_LIBS) \
......@@ -398,6 +400,37 @@ libfs_a_SOURCES = \
src/fs/StandardDirectory.cxx src/fs/StandardDirectory.hxx \
src/fs/DirectoryReader.hxx
# neighbor plugins
if ENABLE_NEIGHBOR_PLUGINS
src_mpd_SOURCES += \
src/command/NeighborCommands.cxx \
src/command/NeighborCommands.hxx
libneighbor_a_SOURCES = \
src/neighbor/Registry.cxx src/neighbor/Registry.hxx \
src/neighbor/Glue.cxx src/neighbor/Glue.hxx \
src/neighbor/Info.hxx \
src/neighbor/Listener.hxx \
src/neighbor/Explorer.hxx \
src/neighbor/NeighborPlugin.hxx
libneighbor_a_CPPFLAGS = $(AM_CPPFLAGS) \
$(SMBCLIENT_CFLAGS)
if ENABLE_SMBCLIENT
libneighbor_a_SOURCES += \
src/lib/smbclient/Init.cxx src/lib/smbclient/Init.hxx \
src/neighbor/plugins/SmbclientNeighborPlugin.cxx src/neighbor/plugins/SmbclientNeighborPlugin.hxx
endif
NEIGHBOR_LIBS = \
$(SMBCLIENT_LIBS) \
libneighbor.a
endif
# database plugins
libdb_plugins_a_SOURCES = \
......@@ -1199,6 +1232,10 @@ noinst_PROGRAMS = \
test/run_normalize \
test/software_volume
if ENABLE_NEIGHBOR_PLUGINS
noinst_PROGRAMS += test/run_neighbor_explorer
endif
if HAVE_AVAHI
noinst_PROGRAMS += test/run_avahi
endif
......@@ -1278,6 +1315,23 @@ test_run_input_SOURCES = test/run_input.cxx \
src/IOThread.cxx \
src/TagSave.cxx
if ENABLE_NEIGHBOR_PLUGINS
test_run_neighbor_explorer_SOURCES = \
src/Log.cxx src/LogBackend.cxx \
test/run_neighbor_explorer.cxx
test_run_neighbor_explorer_LDADD = \
$(GLIB_LIBS) \
$(NEIGHBOR_LIBS) \
libconf.a \
libevent.a \
libfs.a \
libsystem.a \
libthread.a \
libutil.a
endif
if ENABLE_ARCHIVE
test_visit_archive_LDADD = \
......
......@@ -2,6 +2,7 @@ ver 0.19 (not yet released)
* protocol
- new commands "addtagid", "cleartagid"
- "lsinfo" and "readcomments" allowed for remote files
- "listneighbors" lists file servers on the local network
* database
- proxy: forward "idle" events
- proxy: copy "Last-Modified" from remote directories
......
......@@ -885,6 +885,27 @@ fi
AM_CONDITIONAL(ENABLE_MMS, test x$enable_mms = xyes)
dnl ---------------------------------------------------------------------------
dnl Neighbor Plugins
dnl ---------------------------------------------------------------------------
AC_ARG_ENABLE(neighbor-plugins,
AS_HELP_STRING([--enable-neighbor-plugins],
[enable support for neighbor discovery (default: auto)]),,
[enable_neighbor_plugins=auto])
if test x$enable_neighbor_plugins = xauto; then
if test x$enable_smbclient = xyes; then
enable_neighbor_plugins=yes
fi
fi
if test x$enable_neighbor_plugins = xyes; then
AC_DEFINE(ENABLE_NEIGHBOR_PLUGINS, 1,
[Define to enable support for neighbor discovery])
fi
AM_CONDITIONAL(ENABLE_NEIGHBOR_PLUGINS, test x$enable_neighbor_plugins = xyes)
dnl ---------------------------------------------------------------------------
dnl Archive Plugins
dnl ---------------------------------------------------------------------------
......
......@@ -42,6 +42,11 @@
#include "util/OptionDef.hxx"
#include "util/OptionParser.hxx"
#ifdef ENABLE_NEIGHBOR_PLUGINS
#include "neighbor/Registry.hxx"
#include "neighbor/NeighborPlugin.hxx"
#endif
#ifdef ENABLE_ENCODER
#include "encoder/EncoderList.hxx"
#include "encoder/EncoderPlugin.hxx"
......@@ -104,6 +109,13 @@ static void version(void)
for (auto i = database_plugins; *i != nullptr; ++i)
printf(" %s", (*i)->name);
#ifdef ENABLE_NEIGHBOR_PLUGINS
puts("\n\n"
"Neighbor plugins:");
for (auto i = neighbor_plugins; *i != nullptr; ++i)
printf(" %s", (*i)->name);
#endif
puts("\n\n"
"Decoders plugins:");
......@@ -148,7 +160,7 @@ static void version(void)
#endif
puts("\n"
"input/Input plugins:");
"Input plugins:");
input_plugins_for_each(plugin)
printf(" %s", plugin->name);
......
......@@ -44,6 +44,7 @@ static const char *const idle_names[] = {
"update",
"subscription",
"message",
"neighbor",
nullptr
};
......
......@@ -59,6 +59,9 @@ static constexpr unsigned IDLE_SUBSCRIPTION = 0x200;
/** a message on the subscribed channel was received */
static constexpr unsigned IDLE_MESSAGE = 0x400;
/** a neighbor was found or lost */
static constexpr unsigned IDLE_NEIGHBOR = 0x800;
/**
* Adds idle flag (with bitwise "or") and queues notifications to all
* clients.
......
......@@ -54,3 +54,19 @@ Instance::OnDatabaseModified()
{
DatabaseModified();
}
#ifdef ENABLE_NEIGHBOR_PLUGINS
void
Instance::FoundNeighbor(gcc_unused const NeighborInfo &info)
{
idle_add(IDLE_NEIGHBOR);
}
void
Instance::LostNeighbor(gcc_unused const NeighborInfo &info)
{
idle_add(IDLE_NEIGHBOR);
}
#endif
......@@ -24,10 +24,24 @@
#include "db/DatabaseListener.hxx"
#include "Compiler.h"
#ifdef ENABLE_NEIGHBOR_PLUGINS
#include "neighbor/Listener.hxx"
class NeighborGlue;
#endif
class ClientList;
struct Partition;
struct Instance final : public DatabaseListener {
struct Instance final
: public DatabaseListener
#ifdef ENABLE_NEIGHBOR_PLUGINS
, public NeighborListener
#endif
{
#ifdef ENABLE_NEIGHBOR_PLUGINS
NeighborGlue *neighbors;
#endif
ClientList *client_list;
Partition *partition;
......@@ -53,6 +67,12 @@ struct Instance final : public DatabaseListener {
private:
virtual void OnDatabaseModified();
#ifdef ENABLE_NEIGHBOR_PLUGINS
/* virtual methods from class NeighborListener */
virtual void FoundNeighbor(const NeighborInfo &info) override;
virtual void LostNeighbor(const NeighborInfo &info) override;
#endif
};
#endif
......@@ -67,6 +67,10 @@
#include "config/ConfigOption.hxx"
#include "Stats.hxx"
#ifdef ENABLE_NEIGHBOR_PLUGINS
#include "neighbor/Glue.hxx"
#endif
#ifdef ENABLE_INOTIFY
#include "db/update/InotifyUpdate.hxx"
#endif
......@@ -394,6 +398,19 @@ int mpd_main(int argc, char *argv[])
instance = new Instance();
#ifdef ENABLE_NEIGHBOR_PLUGINS
instance->neighbors = new NeighborGlue();
if (!instance->neighbors->Init(io_thread_get(), *instance, error)) {
LogError(error);
return EXIT_FAILURE;
}
if (instance->neighbors->IsEmpty()) {
delete instance->neighbors;
instance->neighbors = nullptr;
}
#endif
const unsigned max_clients = config_get_positive(CONF_MAX_CONN, 10);
instance->client_list = new ClientList(max_clients);
......@@ -460,6 +477,12 @@ int mpd_main(int argc, char *argv[])
io_thread_start();
#ifdef ENABLE_NEIGHBOR_PLUGINS
if (instance->neighbors != nullptr &&
!instance->neighbors->Open(error))
FatalError(error);
#endif
ZeroconfInit(*main_loop);
player_create(instance->partition->pc);
......@@ -523,6 +546,13 @@ int mpd_main(int argc, char *argv[])
listen_global_finish();
delete instance->client_list;
#ifdef ENABLE_NEIGHBOR_PLUGINS
if (instance->neighbors != nullptr) {
instance->neighbors->Close();
delete instance->neighbors;
}
#endif
const clock_t start = clock();
DatabaseGlobalDeinit();
FormatDebug(main_domain,
......
......@@ -27,6 +27,7 @@
#include "FileCommands.hxx"
#include "OutputCommands.hxx"
#include "MessageCommands.hxx"
#include "NeighborCommands.hxx"
#include "OtherCommands.hxx"
#include "Permission.hxx"
#include "tag/TagType.h"
......@@ -99,6 +100,9 @@ static const struct command commands[] = {
{ "list", PERMISSION_READ, 1, -1, handle_list },
{ "listall", PERMISSION_READ, 0, 1, handle_listall },
{ "listallinfo", PERMISSION_READ, 0, 1, handle_listallinfo },
#ifdef ENABLE_NEIGHBOR_PLUGINS
{ "listneighbors", PERMISSION_READ, 0, 0, handle_listneighbors },
#endif
{ "listplaylist", PERMISSION_READ, 1, 1, handle_listplaylist },
{ "listplaylistinfo", PERMISSION_READ, 1, 1, handle_listplaylistinfo },
{ "listplaylists", PERMISSION_READ, 0, 0, handle_listplaylists },
......@@ -179,6 +183,11 @@ command_available(gcc_unused const struct command *cmd)
return sticker_enabled();
#endif
#ifdef ENABLE_NEIGHBOR_PLUGINS
if (strcmp(cmd->cmd, "listneighbors") == 0)
return neighbor_commands_available();
#endif
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.
*/
#include "config.h"
#include "NeighborCommands.hxx"
#include "client/Client.hxx"
#include "Instance.hxx"
#include "Main.hxx"
#include "protocol/Result.hxx"
#include "neighbor/Glue.hxx"
#include "neighbor/Info.hxx"
#include <set>
#include <string>
#include <assert.h>
bool
neighbor_commands_available()
{
return instance->neighbors != nullptr;
}
CommandResult
handle_listneighbors(Client &client,
gcc_unused int argc, gcc_unused char *argv[])
{
assert(instance->neighbors != nullptr);
const auto neighbors = instance->neighbors->GetList();
for (const auto &i : neighbors)
client_printf(client,
"neighbor: %s\n"
"name: %s\n",
i.uri.c_str(),
i.display_name.c_str());
return CommandResult::OK;
}
/*
* 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_NEIGHBOR_COMMANDS_HXX
#define MPD_NEIGHBOR_COMMANDS_HXX
#include "CommandResult.hxx"
#include "Compiler.h"
class Client;
gcc_pure
bool
neighbor_commands_available();
CommandResult
handle_listneighbors(Client &client, int argc, char *argv[]);
#endif
......@@ -77,6 +77,7 @@ enum ConfigOption {
CONF_DESPOTIFY_HIGH_BITRATE,
CONF_AUDIO_FILTER,
CONF_DATABASE,
CONF_NEIGHBORS,
CONF_MAX
};
......
......@@ -77,6 +77,7 @@ const ConfigTemplate config_templates[] = {
{ "despotify_high_bitrate", false, false },
{ "filter", true, true },
{ "database", false, true },
{ "neighbors", true, true },
};
static constexpr unsigned n_config_templates =
......
/*
* 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_NEIGHBOR_EXPLORER_HXX
#define MPD_NEIGHBOR_EXPLORER_HXX
#include <forward_list>
class Error;
class NeighborListener;
struct NeighborInfo;
/**
* An object that explores the neighborhood for music servers.
*
* As soon as this object is opened, it will start exploring, and
* notify the #NeighborListener when it found or lost something.
*
* The implementation is supposed to be non-blocking. This can be
* implemented either using the #EventLoop instance that was passed to
* the NeighborPlugin or by moving the blocking parts in a dedicated
* thread.
*/
class NeighborExplorer {
protected:
NeighborListener &listener;
explicit NeighborExplorer(NeighborListener &_listener)
:listener(_listener) {}
public:
typedef std::forward_list<NeighborInfo> List;
/**
* Free instance data.
*/
virtual ~NeighborExplorer() {}
/**
* Start exploring the neighborhood.
*/
virtual bool Open(Error &error) = 0;
/**
* Stop exploring.
*/
virtual void Close() = 0;
/**
* Obtain a list of currently known neighbors.
*/
virtual List GetList() const = 0;
};
#endif
/*
* 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 "Glue.hxx"
#include "Registry.hxx"
#include "Explorer.hxx"
#include "NeighborPlugin.hxx"
#include "Info.hxx"
#include "config/ConfigGlobal.hxx"
#include "config/ConfigData.hxx"
#include "config/ConfigError.hxx"
#include "util/Error.hxx"
NeighborGlue::Explorer::~Explorer()
{
delete explorer;
}
NeighborGlue::~NeighborGlue() {}
static NeighborExplorer *
CreateNeighborExplorer(EventLoop &loop, NeighborListener &listener,
const config_param &param, Error &error)
{
const char *plugin_name = param.GetBlockValue("plugin");
if (plugin_name == nullptr) {
error.Set(config_domain,
"Missing \"plugin\" configuration");
return nullptr;
}
const NeighborPlugin *plugin = GetNeighborPluginByName(plugin_name);
if (plugin == nullptr) {
error.Format(config_domain, "No such neighbor plugin: %s",
plugin_name);
return nullptr;
}
return plugin->create(loop, listener, param, error);
}
bool
NeighborGlue::Init(EventLoop &loop, NeighborListener &listener, Error &error)
{
const config_param *param = nullptr;
while ((param = config_get_next_param(CONF_NEIGHBORS, param))) {
NeighborExplorer *explorer =
CreateNeighborExplorer(loop, listener, *param, error);
if (explorer == nullptr) {
error.FormatPrefix("Line %i: ", param->line);
return false;
}
explorers.emplace_front(explorer);
}
return true;
}
bool
NeighborGlue::Open(Error &error)
{
for (auto i = explorers.begin(), end = explorers.end();
i != end; ++i) {
if (!i->explorer->Open(error)) {
/* roll back */
for (auto k = ++i; k != end; ++k)
k->explorer->Close();
return false;
}
}
return true;
}
void
NeighborGlue::Close()
{
for (auto i = explorers.begin(), end = explorers.end(); i != end; ++i)
i->explorer->Close();
}
NeighborGlue::List
NeighborGlue::GetList() const
{
List result;
for (const auto &i : explorers)
result.splice_after(result.before_begin(),
i.explorer->GetList());
return result;
}
/*
* 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_NEIGHBOR_ALL_HXX
#define MPD_NEIGHBOR_ALL_HXX
#include "check.h"
#include "Compiler.h"
#include "thread/Mutex.hxx"
#include <forward_list>
struct config_param;
class Error;
class EventLoop;
class NeighborExplorer;
class NeighborListener;
struct NeighborInfo;
/**
* A class that initializes and opens all configured neighbor plugins.
*/
class NeighborGlue {
struct Explorer {
NeighborExplorer *const explorer;
Explorer(NeighborExplorer *_explorer):explorer(_explorer) {}
Explorer(const Explorer &) = delete;
~Explorer();
};
Mutex mutex;
std::forward_list<Explorer> explorers;
public:
typedef std::forward_list<NeighborInfo> List;
NeighborGlue() = default;
NeighborGlue(const NeighborGlue &) = delete;
~NeighborGlue();
bool IsEmpty() const {
return explorers.empty();
}
bool Init(EventLoop &loop, NeighborListener &listener, Error &error);
bool Open(Error &error);
void Close();
/**
* Get the combined list of all neighbors from all active
* plugins.
*/
gcc_pure
List GetList() const;
};
#endif
/*
* 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_NEIGHBOR_INFO_HXX
#define MPD_NEIGHBOR_INFO_HXX
#include <string>
struct NeighborInfo {
std::string uri;
std::string display_name;
};
#endif
/*
* 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_NEIGHBOR_LISTENER_HXX
#define MPD_NEIGHBOR_LISTENER_HXX
struct NeighborInfo;
class NeighborExplorer;
/**
* An interface that listens on events from neighbor plugins. The
* methods must be thread-safe and non-blocking.
*/
class NeighborListener {
public:
virtual void FoundNeighbor(const NeighborInfo &info) = 0;
virtual void LostNeighbor(const NeighborInfo &info) = 0;
};
#endif
/*
* 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_NEIGHBOR_PLUGIN_HXX
#define MPD_NEIGHBOR_PLUGIN_HXX
struct config_param;
class Error;
class EventLoop;
class NeighborListener;
class NeighborExplorer;
struct NeighborPlugin {
const char *name;
/**
* Allocates and configures a #NeighborExplorer instance.
*/
NeighborExplorer *(*create)(EventLoop &loop, NeighborListener &listener,
const config_param &param,
Error &error);
};
#endif
/*
* 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 "Registry.hxx"
#include "NeighborPlugin.hxx"
#include "plugins/SmbclientNeighborPlugin.hxx"
#include <string.h>
const NeighborPlugin *const neighbor_plugins[] = {
#ifdef ENABLE_SMBCLIENT
&smbclient_neighbor_plugin,
#endif
nullptr
};
const NeighborPlugin *
GetNeighborPluginByName(const char *name)
{
for (auto i = neighbor_plugins; *i != nullptr; ++i)
if (strcmp((*i)->name, name) == 0)
return *i;
return nullptr;
}
/*
* 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_NEIGHBOR_REGISTRY_HXX
#define MPD_NEIGHBOR_REGISTRY_HXX
#include "Compiler.h"
struct NeighborPlugin;
/**
* nullptr terminated list of all neighbor plugins which were enabled at
* compile time.
*/
extern const NeighborPlugin *const neighbor_plugins[];
gcc_pure
const NeighborPlugin *
GetNeighborPluginByName(const char *name);
#endif
/*
* 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 "SmbclientNeighborPlugin.hxx"
#include "lib/smbclient/Init.hxx"
#include "neighbor/NeighborPlugin.hxx"
#include "neighbor/Explorer.hxx"
#include "neighbor/Listener.hxx"
#include "neighbor/Info.hxx"
#include "thread/Mutex.hxx"
#include "thread/Cond.hxx"
#include "thread/Thread.hxx"
#include "thread/Name.hxx"
#include "util/Macros.hxx"
#include "util/Domain.hxx"
#include "util/Error.hxx"
#include "Log.hxx"
#include <libsmbclient.h>
#include <list>
#include <algorithm>
static constexpr Domain smbclient_domain("smbclient");
class SmbclientNeighborExplorer final : public NeighborExplorer {
struct Server {
std::string name, comment;
bool alive;
Server(std::string &&_name, std::string &&_comment)
:name(std::move(_name)), comment(std::move(_comment)),
alive(true) {}
Server(const Server &) = delete;
gcc_pure
bool operator==(const Server &other) const {
return name == other.name;
}
gcc_pure
NeighborInfo Export() const {
return { "smb://" + name + "/", comment };
}
};
Thread thread;
mutable Mutex mutex;
Cond cond;
List list;
bool quit;
public:
SmbclientNeighborExplorer(NeighborListener &_listener)
:NeighborExplorer(_listener) {}
/* virtual methods from class NeighborExplorer */
virtual bool Open(Error &error) override;
virtual void Close() override;
virtual List GetList() const override;
private:
void Run();
void ThreadFunc();
static void ThreadFunc(void *ctx);
};
bool
SmbclientNeighborExplorer::Open(Error &error)
{
quit = false;
return thread.Start(ThreadFunc, this, error);
}
void
SmbclientNeighborExplorer::Close()
{
mutex.lock();
quit = true;
cond.signal();
mutex.unlock();
thread.Join();
}
NeighborExplorer::List
SmbclientNeighborExplorer::GetList() const
{
const ScopeLock protect(mutex);
/*
List list;
for (const auto &i : servers)
list.emplace_front(i.Export());
*/
return list;
}
static void
ReadServer(NeighborExplorer::List &list, const smbc_dirent &e)
{
const std::string name(e.name, e.namelen);
const std::string comment(e.comment, e.commentlen);
NeighborInfo info{
"smb://" + name,
name + " (" + comment + ")",
};
list.emplace_front(std::move(info));
}
static void
ReadServers(NeighborExplorer::List &list, const char *uri);
static void
ReadWorkgroup(NeighborExplorer::List &list, const std::string &name)
{
std::string uri = "smb://" + name;
ReadServers(list, uri.c_str());
}
static void
ReadEntry(NeighborExplorer::List &list, const smbc_dirent &e)
{
switch (e.smbc_type) {
case SMBC_WORKGROUP:
ReadWorkgroup(list, std::string(e.name, e.namelen));
break;
case SMBC_SERVER:
ReadServer(list, e);
break;
}
}
static void
ReadServers(NeighborExplorer::List &list, int fd)
{
smbc_dirent *e;
while ((e = smbc_readdir(fd)) != nullptr)
ReadEntry(list, *e);
smbc_closedir(fd);
}
static void
ReadServers(NeighborExplorer::List &list, const char *uri)
{
int fd = smbc_opendir(uri);
if (fd >= 0) {
ReadServers(list, fd);
smbc_closedir(fd);
} else
FormatErrno(smbclient_domain, "smbc_opendir('%s') failed",
uri);
}
gcc_pure
static NeighborExplorer::List
DetectServers()
{
NeighborExplorer::List list;
ReadServers(list, "smb://");
return list;
}
gcc_pure
static NeighborExplorer::List::const_iterator
FindBeforeServerByURI(NeighborExplorer::List::const_iterator prev,
NeighborExplorer::List::const_iterator end,
const std::string &uri)
{
for (auto i = std::next(prev); i != end; prev = i, i = std::next(prev))
if (i->uri == uri)
return prev;
return end;
}
inline void
SmbclientNeighborExplorer::Run()
{
List found = DetectServers(), lost;
mutex.lock();
const auto found_before_begin = found.before_begin();
const auto found_end = found.end();
for (auto prev = list.before_begin(), i = std::next(prev), end = list.end();
i != end; i = std::next(prev)) {
auto f = FindBeforeServerByURI(found_before_begin, found_end,
i->uri);
if (f != found_end) {
/* still visible: remove from "found" so we
don't believe it's a new one */
*i = std::move(*std::next(f));
found.erase_after(f);
prev = i;
} else {
/* can't see it anymore: move to "lost" */
lost.splice_after(lost.before_begin(), list, prev);
}
}
for (auto prev = found_before_begin, i = std::next(prev);
i != found_end; prev = i, i = std::next(prev))
list.push_front(*i);
mutex.unlock();
for (auto &i : lost)
listener.LostNeighbor(i);
for (auto &i : found)
listener.FoundNeighbor(i);
}
inline void
SmbclientNeighborExplorer::ThreadFunc()
{
mutex.lock();
while (!quit) {
mutex.unlock();
Run();
mutex.lock();
if (quit)
break;
// TODO: sleep for how long?
cond.timed_wait(mutex, 10000);
}
mutex.unlock();
}
void
SmbclientNeighborExplorer::ThreadFunc(void *ctx)
{
SetThreadName("smbclient");
SmbclientNeighborExplorer &e = *(SmbclientNeighborExplorer *)ctx;
e.ThreadFunc();
}
static NeighborExplorer *
smbclient_neighbor_create(gcc_unused EventLoop &loop,
NeighborListener &listener,
gcc_unused const config_param &param,
gcc_unused Error &error)
{
if (!SmbclientInit(error))
return nullptr;
return new SmbclientNeighborExplorer(listener);
}
const NeighborPlugin smbclient_neighbor_plugin = {
"smbclient",
smbclient_neighbor_create,
};
/*
* 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_NEIGHBOR_SMBCLIENT_HXX
#define MPD_NEIGHBOR_SMBCLIENT_HXX
struct NeighborPlugin;
extern const NeighborPlugin smbclient_neighbor_plugin;
#endif
/run_neighbor_explorer
/*
* 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 "config/ConfigGlobal.hxx"
#include "neighbor/Listener.hxx"
#include "neighbor/Info.hxx"
#include "neighbor/Glue.hxx"
#include "fs/Path.hxx"
#include "event/Loop.hxx"
#include "util/Error.hxx"
#include "Log.hxx"
#include <stdio.h>
#include <stdlib.h>
class MyNeighborListener final : public NeighborListener {
public:
/* virtual methods from class NeighborListener */
virtual void FoundNeighbor(const NeighborInfo &info) override {
printf("found '%s' (%s)\n",
info.display_name.c_str(), info.uri.c_str());
}
virtual void LostNeighbor(const NeighborInfo &info) override {
printf("lost '%s' (%s)\n",
info.display_name.c_str(), info.uri.c_str());
}
};
int
main(int argc, char **argv)
{
if (argc != 2) {
fprintf(stderr, "Usage: run_neighbor_explorer CONFIG\n");
return EXIT_FAILURE;
}
const Path config_path = Path::FromFS(argv[1]);
/* read configuration file (mpd.conf) */
Error error;
config_global_init();
if (!ReadConfigFile(config_path, error)) {
LogError(error);
return EXIT_FAILURE;
}
/* initialize the core */
EventLoop loop((EventLoop::Default()));
/* initialize neighbor plugins */
MyNeighborListener listener;
NeighborGlue neighbor;
if (!neighbor.Init(loop, listener, error) || !neighbor.Open(error)) {
LogError(error);
return EXIT_FAILURE;
}
/* run */
loop.Run();
neighbor.Close();
return EXIT_SUCCESS;
}
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