diff options
Diffstat (limited to 'plugins/zeroconf/avahiservice.cpp')
| -rw-r--r-- | plugins/zeroconf/avahiservice.cpp | 341 |
1 files changed, 0 insertions, 341 deletions
diff --git a/plugins/zeroconf/avahiservice.cpp b/plugins/zeroconf/avahiservice.cpp deleted file mode 100644 index aebaa00..0000000 --- a/plugins/zeroconf/avahiservice.cpp +++ /dev/null @@ -1,341 +0,0 @@ -/*************************************************************************** - * 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-tqt/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(TQString("%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(TQString("_" + 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) << TQString("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) << TQString("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); - - TQString realname = TQString(name); - realname.truncate(realname.length() - 5); - - if (service->id != TQString(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: - { - TQString realname = TQString(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(TQString("_" + 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" |