summaryrefslogtreecommitdiffstats
path: root/plugins/zeroconf/avahiservice.cpp
diff options
context:
space:
mode:
authortpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da>2010-01-20 02:37:40 +0000
committertpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da>2010-01-20 02:37:40 +0000
commit9ad5c7b5e23b4940e7a3ea3ca3a6fb77e6a8fab0 (patch)
treed088b5210e77d9fa91d954d8550e00e372b47378 /plugins/zeroconf/avahiservice.cpp
downloadktorrent-9ad5c7b5e23b4940e7a3ea3ca3a6fb77e6a8fab0.tar.gz
ktorrent-9ad5c7b5e23b4940e7a3ea3ca3a6fb77e6a8fab0.zip
Updated to final KDE3 ktorrent release (2.2.6)
git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/applications/ktorrent@1077377 283d02a7-25f6-0310-bc7c-ecb5cbfe19da
Diffstat (limited to 'plugins/zeroconf/avahiservice.cpp')
-rw-r--r--plugins/zeroconf/avahiservice.cpp341
1 files changed, 341 insertions, 0 deletions
diff --git a/plugins/zeroconf/avahiservice.cpp b/plugins/zeroconf/avahiservice.cpp
new file mode 100644
index 0000000..a190d8e
--- /dev/null
+++ b/plugins/zeroconf/avahiservice.cpp
@@ -0,0 +1,341 @@
+/***************************************************************************
+ * Copyright (C) 2006 by Lesly Weyts and Kevin Andre *
+ * *
+ * 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 <util/log.h>
+#include <torrent/peerid.h>
+#include <avahi-common/watch.h>
+#include <avahi-qt3/qt-watch.h>
+#include "localbrowser.h"
+#include "avahiservice.h"
+
+using namespace bt;
+
+namespace kt
+{
+
+ void group_callback(AvahiEntryGroup* g, AvahiEntryGroupState state, void* userdata)
+ {
+ AvahiService* service = reinterpret_cast<AvahiService*>(userdata);
+
+ if (g == service->group)
+ {
+ switch (state)
+ {
+ case AVAHI_ENTRY_GROUP_ESTABLISHED:
+ break;
+ case AVAHI_ENTRY_GROUP_COLLISION:
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Entry group collision." << endl;
+ break;
+ case AVAHI_ENTRY_GROUP_FAILURE:
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Entry group failure." << endl;
+ break;
+ case AVAHI_ENTRY_GROUP_UNCOMMITED:
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Entry group uncommited." << endl;
+ break;
+ case AVAHI_ENTRY_GROUP_REGISTERING:
+ ;
+ }
+ }
+ }
+
+ void publish_service(AvahiService* service, AvahiClient *c)
+ {
+ assert(c);
+
+ if (!service->group)
+ {
+ if (!(service->group = avahi_entry_group_new(c, group_callback, service)))
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: avahi_entry_group_new failed." << endl;
+ return;
+ }
+ }
+
+ const char* name = avahi_strdup(QString("%1__%2%3").arg(service->id).arg((rand() % 26) + 65).arg((rand() % 26) + 65).ascii());
+ const char* type = avahi_strdup("_bittorrent._tcp");
+ const char* subtype = avahi_strdup(QString("_" + service->infoHash + "._sub._bittorrent._tcp").ascii());
+
+ if (avahi_entry_group_add_service(
+ service->group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC,
+ (AvahiPublishFlags)0, name, type, NULL, NULL, service->port, NULL))
+ {
+ if (avahi_client_errno(c) != -8)
+ Out(SYS_ZCO|LOG_DEBUG) << QString("ZC: Failed to add the service (%i).").arg(avahi_client_errno(c)) << endl;
+ else
+ publish_service(service, c);
+ return;
+ }
+
+ if (avahi_entry_group_add_service_subtype(
+ service->group, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC,
+ (AvahiPublishFlags)0, name, type, NULL, subtype))
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << QString("ZC: Failed to add the service subtype (%i).").arg( avahi_client_errno(c)) << endl;
+ return;
+ }
+
+ if (avahi_entry_group_commit(service->group))
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Failed to commit the entry group." << endl;
+ return;
+ }
+ }
+
+ void publisher_callback(AvahiClient* c, AvahiClientState state, void* userdata)
+ {
+ if (!c)
+ return;
+
+ AvahiService* service = reinterpret_cast<AvahiService*>(userdata);
+
+ switch (state) {
+ case AVAHI_CLIENT_S_RUNNING:
+ {
+ if (!service->group)
+ publish_service(service, c);
+ break;
+ }
+ case AVAHI_CLIENT_FAILURE:
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "Failure when publishing." << endl;
+ break;
+ }
+ case AVAHI_CLIENT_S_COLLISION:
+ case AVAHI_CLIENT_S_REGISTERING:
+ {
+ if (service->group)
+ avahi_entry_group_reset(service->group);
+ break;
+ }
+ case AVAHI_CLIENT_CONNECTING:
+ ;
+ }
+ }
+
+ void listener_callback(AvahiClient* c, AvahiClientState state, void* userdata)
+ {
+ assert(c);
+
+ AvahiService* service = reinterpret_cast<AvahiService*>(userdata);
+
+ if (state == AVAHI_CLIENT_FAILURE)
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Server connection failure." << endl;
+ }
+ }
+
+ void resolve_callback(
+ AvahiServiceResolver* r,
+ AVAHI_GCC_UNUSED AvahiIfIndex interface,
+ AVAHI_GCC_UNUSED AvahiProtocol protocol,
+ AvahiResolverEvent event,
+ const char* name,
+ AVAHI_GCC_UNUSED const char* type,
+ AVAHI_GCC_UNUSED const char* domain,
+ AVAHI_GCC_UNUSED const char* host_name,
+ const AvahiAddress* address,
+ uint16_t port,
+ AVAHI_GCC_UNUSED AvahiStringList* txt,
+ AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
+ void* userdata)
+ {
+ assert(r);
+
+ switch (event) {
+ case AVAHI_RESOLVER_FAILURE:
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Resolver failed." << endl;
+ break;
+ }
+ case AVAHI_RESOLVER_FOUND:
+ {
+ AvahiService* service = reinterpret_cast<AvahiService*>(userdata);
+
+ QString realname = QString(name);
+ realname.truncate(realname.length() - 5);
+
+ if (service->id != QString(realname))
+ {
+ char a[AVAHI_ADDRESS_STR_MAX];
+ avahi_address_snprint(a, sizeof(a), address);
+ const char* ip = a;
+ LocalBrowser::insert(bt::PeerID(realname.ascii()));
+
+ Out(SYS_ZCO|LOG_NOTICE) << "ZC: found local peer " << ip << ":" << port << endl;
+ service->addPeer(ip,port,true);
+ service->emitPeersReady();
+ }
+ }
+ }
+ avahi_service_resolver_free(r);
+ }
+
+ void browser_callback(
+ AvahiServiceBrowser* b,
+ AvahiIfIndex interface,
+ AvahiProtocol protocol,
+ AvahiBrowserEvent event,
+ const char* name,
+ const char* type,
+ const char* domain,
+ AVAHI_GCC_UNUSED AvahiLookupResultFlags flags,
+ void* userdata)
+ {
+ assert(b);
+
+ AvahiService* service = reinterpret_cast<AvahiService*>(userdata);
+
+ switch (event) {
+ case AVAHI_BROWSER_FAILURE:
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Browser failure." << endl;
+ break;
+ }
+ case AVAHI_BROWSER_NEW:
+ {
+ if (!(avahi_service_resolver_new(service->listener, interface, protocol, name, type, domain, AVAHI_PROTO_UNSPEC, (AvahiLookupFlags)0, resolve_callback, userdata)))
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Failed to resolve the service." << endl;
+ break;
+ }
+ case AVAHI_BROWSER_REMOVE:
+ {
+ QString realname = QString(name);
+ realname.truncate(realname.length() - 5);
+
+ LocalBrowser::remove(bt::PeerID(realname.ascii()));
+
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Browser removed." << endl;
+ break;
+ }
+ case AVAHI_BROWSER_ALL_FOR_NOW:
+ case AVAHI_BROWSER_CACHE_EXHAUSTED:
+ ;
+ }
+ }
+
+ AvahiService::AvahiService(const bt::PeerID& id,bt::Uint16 port, const bt::SHA1Hash & infoHash)
+ : group(0), publisher_poll(0), listener_poll(0),publisher(0), listener(0), browser(0)
+ {
+ started = false;
+
+ this->id = id.toString();
+ this->port = port;
+ this->infoHash = infoHash.toString();
+ }
+
+ AvahiService::~AvahiService()
+ {
+ stop(0);
+ }
+
+ void AvahiService::stop(bt::WaitJob*)
+ {
+ if (started)
+ {
+ started = false;
+
+ publisher_poll = 0;
+ if (publisher)
+ {
+ avahi_client_free(publisher);
+ publisher = 0;
+ }
+
+ listener_poll = 0;
+ if (listener)
+ {
+ avahi_client_free(listener);
+ listener = 0;
+ }
+ }
+ }
+
+ void AvahiService::start()
+ {
+ started = startPublishing() && startBrowsing();
+ }
+
+ bool AvahiService::startPublishing()
+ {
+ group = NULL;
+ publisher_poll = NULL;
+ publisher = NULL;
+
+ if (!(publisher_poll = avahi_qt_poll_get()))
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Failed to create a poll for publishing." << endl;
+ stop();
+ return false;
+ }
+
+ publisher = avahi_client_new(publisher_poll, AVAHI_CLIENT_NO_FAIL, publisher_callback, this, NULL);
+
+ if (!(publisher))
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Failed to create a client for publishing." << endl;
+ stop();
+ return false;
+ }
+
+ return true;
+ }
+
+ bool AvahiService::startBrowsing()
+ {
+ listener_poll = NULL;
+ listener = NULL;
+ browser = NULL;
+
+ if (!(listener_poll = avahi_qt_poll_get()))
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Failed to create a poll for browsing." << endl;
+ stop();
+ return false;
+ }
+
+ listener = avahi_client_new(listener_poll,AVAHI_CLIENT_NO_FAIL, listener_callback, this, NULL);
+
+ if (!listener)
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Failed to create a client for browsing." << endl;
+ stop();
+ return false;
+ }
+
+ if (!(browser = avahi_service_browser_new(listener, AVAHI_IF_UNSPEC, AVAHI_PROTO_UNSPEC, avahi_strdup(QString("_" + infoHash + "._sub._bittorrent._tcp").ascii()), NULL, (AvahiLookupFlags)0, browser_callback, this)))
+ {
+ Out(SYS_ZCO|LOG_DEBUG) << "ZC: Failed to create a service browser." << endl;
+ stop();
+ return false;
+ }
+
+ return true;
+ }
+
+ void AvahiService::emitPeersReady()
+ {
+ peersReady(this);
+ }
+
+ void AvahiService::aboutToBeDestroyed()
+ {
+ serviceDestroyed(this);
+ }
+}
+
+#include "avahiservice.moc"