diff options
Diffstat (limited to 'plugins/zeroconf/zeroconfplugin.cpp')
| -rw-r--r-- | plugins/zeroconf/zeroconfplugin.cpp | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/plugins/zeroconf/zeroconfplugin.cpp b/plugins/zeroconf/zeroconfplugin.cpp new file mode 100644 index 0000000..b906843 --- /dev/null +++ b/plugins/zeroconf/zeroconfplugin.cpp @@ -0,0 +1,135 @@ +/*************************************************************************** + * Copyright (C) 2005 by Joris Guisson * + * joris.guisson@gmail.com * + * * + * 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 <kgenericfactory.h> +#include <util/log.h> +#include <interfaces/coreinterface.h> +#include <interfaces/torrentinterface.h> +#include <torrent/globals.h> +#include <torrent/server.h> +#include "zeroconfplugin.h" +#include "avahiservice.h" + + +#define NAME "Zeroconf" +#define AUTHOR "Lesly Weyts and Kevin Andre" + +K_EXPORT_COMPONENT_FACTORY(ktzeroconfplugin,KGenericFactory<kt::ZeroConfPlugin>("ktzeroconfplugin")) + +using namespace bt; + +namespace kt +{ + + ZeroConfPlugin::ZeroConfPlugin(QObject* parent, const char* name, const QStringList& args) + : Plugin(parent, name,args,NAME,i18n("Zeroconf"),AUTHOR,QString::null,i18n("Finds peers running ktorrent on the local network to share torrents with"),"ktplugins") + { + services.setAutoDelete(true); + } + + + ZeroConfPlugin::~ZeroConfPlugin() + {} + + void ZeroConfPlugin::load() + { + CoreInterface* core = getCore(); + connect(core,SIGNAL(torrentAdded( kt::TorrentInterface* )), + this,SLOT(torrentAdded( kt::TorrentInterface* ))); + connect(core,SIGNAL(torrentRemoved( kt::TorrentInterface* )), + this,SLOT(torrentRemoved( kt::TorrentInterface* ))); + + // go over existing torrents and add them + bt::QueueManager* qman = core->getQueueManager(); + for (QPtrList<kt::TorrentInterface>::iterator i = qman->begin();i != qman->end();i++) + { + torrentAdded(*i); + } + } + + void ZeroConfPlugin::unload() + { + CoreInterface* core = getCore(); + disconnect(core,SIGNAL(torrentAdded( kt::TorrentInterface* )), + this,SLOT(torrentAdded( kt::TorrentInterface* ))); + disconnect(core,SIGNAL(torrentRemoved( kt::TorrentInterface* )), + this,SLOT(torrentRemoved( kt::TorrentInterface*))); + + bt::PtrMap<kt::TorrentInterface*,AvahiService>::iterator i = services.begin(); + while (i != services.end()) + { + AvahiService* av = i->second; + kt::TorrentInterface* ti = i->first; + ti->removePeerSource(av); + i++; + } + services.clear(); + } + + void ZeroConfPlugin::torrentAdded(kt::TorrentInterface* tc) + { + if (services.contains(tc)) + return; + + bt::Uint16 port = bt::Globals::instance().getServer().getPortInUse(); + AvahiService* av = new AvahiService(tc->getOwnPeerID(),port,tc->getInfoHash()); + services.insert(tc,av); + tc->addPeerSource(av); + Out(SYS_ZCO|LOG_NOTICE) << "ZeroConf service added for " + << tc->getStats().torrent_name << endl; + connect(av,SIGNAL(serviceDestroyed( AvahiService* )), + this,SLOT(avahiServiceDestroyed( AvahiService* ))); + } + + + void ZeroConfPlugin::torrentRemoved(kt::TorrentInterface* tc) + { + AvahiService* av = services.find(tc); + if (!av) + return; + Out(SYS_ZCO|LOG_NOTICE) << "ZeroConf service removed for " + << tc->getStats().torrent_name << endl; + tc->removePeerSource(av); + services.erase(tc); + } + + void ZeroConfPlugin::avahiServiceDestroyed(AvahiService* av) + { + services.setAutoDelete(false); + + Out(SYS_ZCO|LOG_NOTICE) << "ZeroConf service destroyed " << endl; + bt::PtrMap<kt::TorrentInterface*,AvahiService>::iterator i = services.begin(); + while (i != services.end()) + { + if (i->second == av) + { + services.erase(i->first); + break; + } + i++; + } + services.setAutoDelete(true); + } + + bool ZeroConfPlugin::versionCheck(const QString & version) const + { + return version == KT_VERSION_MACRO; + } +} +#include "zeroconfplugin.moc" |