diff options
Diffstat (limited to 'plugins/zeroconf/avahiservice.cpp')
| -rw-r--r-- | plugins/zeroconf/avahiservice.cpp | 341 |
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" |