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.
This commit is contained in:
Max Kellermann 2014-01-18 16:36:42 +01:00
parent e847788569
commit 5c4a42caa0
26 changed files with 1101 additions and 2 deletions

View File

@ -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 = \

1
NEWS
View File

@ -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

View File

@ -884,6 +884,27 @@ if test x$enable_mms = xyes; then
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 ---------------------------------------------------------------------------

View File

@ -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);

View File

@ -44,6 +44,7 @@ static const char *const idle_names[] = {
"update",
"subscription",
"message",
"neighbor",
nullptr
};

View File

@ -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.

View File

@ -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

View File

@ -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

View File

@ -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,

View File

@ -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;
}

View File

@ -0,0 +1,54 @@
/*
* 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;
}

View File

@ -0,0 +1,35 @@
/*
* 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

View File

@ -77,6 +77,7 @@ enum ConfigOption {
CONF_DESPOTIFY_HIGH_BITRATE,
CONF_AUDIO_FILTER,
CONF_DATABASE,
CONF_NEIGHBORS,
CONF_MAX
};

View File

@ -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 =

71
src/neighbor/Explorer.hxx Normal file
View File

@ -0,0 +1,71 @@
/*
* 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

111
src/neighbor/Glue.cxx Normal file
View File

@ -0,0 +1,111 @@
/*
* 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;
}

76
src/neighbor/Glue.hxx Normal file
View File

@ -0,0 +1,76 @@
/*
* 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

30
src/neighbor/Info.hxx Normal file
View File

@ -0,0 +1,30 @@
/*
* 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

36
src/neighbor/Listener.hxx Normal file
View File

@ -0,0 +1,36 @@
/*
* 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

View File

@ -0,0 +1,40 @@
/*
* 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

42
src/neighbor/Registry.cxx Normal file
View File

@ -0,0 +1,42 @@
/*
* 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;
}

37
src/neighbor/Registry.hxx Normal file
View File

@ -0,0 +1,37 @@
/*
* 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

View File

@ -0,0 +1,285 @@
/*
* 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,
};

View File

@ -0,0 +1,27 @@
/*
* 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

1
test/.gitignore vendored Normal file
View File

@ -0,0 +1 @@
/run_neighbor_explorer

View File

@ -0,0 +1,85 @@
/*
* 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;
}