You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ktorrent/plugins/zeroconf/zeroconfplugin.cpp

136 lines
4.7 KiB

/***************************************************************************
* 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(TQObject* tqparent, const char* name, const TQStringList& args)
: Plugin(tqparent, name,args,NAME,i18n("Zeroconf"),AUTHOR,TQString(),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,TQT_SIGNAL(torrentAdded( kt::TorrentInterface* )),
this,TQT_SLOT(torrentAdded( kt::TorrentInterface* )));
connect(core,TQT_SIGNAL(torrentRemoved( kt::TorrentInterface* )),
this,TQT_SLOT(torrentRemoved( kt::TorrentInterface* )));
// go over existing torrents and add them
bt::QueueManager* qman = core->getQueueManager();
for (TQPtrList<kt::TorrentInterface>::iterator i = qman->begin();i != qman->end();i++)
{
torrentAdded(*i);
}
}
void ZeroConfPlugin::unload()
{
CoreInterface* core = getCore();
disconnect(core,TQT_SIGNAL(torrentAdded( kt::TorrentInterface* )),
this,TQT_SLOT(torrentAdded( kt::TorrentInterface* )));
disconnect(core,TQT_SIGNAL(torrentRemoved( kt::TorrentInterface* )),
this,TQT_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,TQT_SIGNAL(serviceDestroyed( AvahiService* )),
this,TQT_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 TQString & version) const
{
return version == KT_VERSION_MACRO;
}
}
#include "zeroconfplugin.moc"