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

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"