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, 173 insertions, 0 deletions

diff --git a/libktorrent/net/socketmonitor.cpp b/libktorrent/net/socketmonitor.cpp
new file mode 100644
index 0000000..38225ab
--- /dev/null
+++ b/libktorrent/net/socketmonitor.cpp
@@ -0,0 +1,173 @@
+/***************************************************************************
+ *   Copyright (C) 2005 by Joris Guisson                                   *
+ *   [email protected]                                               *
+ *                                                                         *
+ *   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 <math.h>
+#include <unistd.h>
+#include <util/functions.h>
+#include <util/log.h>
+#include <torrent/globals.h>
+#include "socketmonitor.h"
+#include "bufferedsocket.h"
+#include "uploadthread.h"
+#include "downloadthread.h"
+
+using namespace bt;
+
+namespace net
+{
+	SocketMonitor SocketMonitor::self;
+
+	SocketMonitor::SocketMonitor() : ut(0),dt(0),next_group_id(1)
+	{
+		dt = new DownloadThread(this);
+		ut = new UploadThread(this);
+	}
+
+
+	SocketMonitor::~SocketMonitor()
+	{
+		if (ut && ut->isRunning())
+		{
+			ut->stop();
+			ut->signalDataReady(); // kick it in the nuts, if the thread is waiting for data
+			if (!ut->wait(250))
+			{
+				ut->terminate();
+				ut->wait();
+			}
+		}
+			
+		
+		if (dt && dt->isRunning())
+		{
+			dt->stop();
+			if (!dt->wait(250))
+			{
+				dt->terminate();
+				dt->wait();
+			}
+		}
+		
+		delete ut;
+		delete dt;
+	}
+	
+	void SocketMonitor::lock()
+	{
+		mutex.lock();
+	}
+	
+	void SocketMonitor::unlock()
+	{
+		mutex.unlock();
+	}
+	
+	void SocketMonitor::setDownloadCap(Uint32 bytes_per_sec)
+	{
+		DownloadThread::setCap(bytes_per_sec);
+	}
+	
+	void SocketMonitor::setUploadCap(Uint32 bytes_per_sec)
+	{
+		UploadThread::setCap(bytes_per_sec);
+	}
+	
+	void SocketMonitor::setSleepTime(Uint32 sleep_time)
+	{
+		DownloadThread::setSleepTime(sleep_time);
+		UploadThread::setSleepTime(sleep_time);
+	}
+	
+	void SocketMonitor::add(BufferedSocket* sock)
+	{
+		QMutexLocker lock(&mutex);
+		
+		bool start_threads = smap.count() == 0;
+		smap.append(sock);
+		
+		if (start_threads)
+		{
+			Out(SYS_CON|LOG_DEBUG) << "Starting socketmonitor threads" << endl;
+				
+			if (!dt->isRunning())
+				dt->start(QThread::IdlePriority);
+			if (!ut->isRunning())
+				ut->start(QThread::IdlePriority);
+		}
+	}
+	
+	void SocketMonitor::remove(BufferedSocket* sock)
+	{
+		QMutexLocker lock(&mutex);
+		if (smap.count() == 0)
+			return;
+		
+		smap.remove(sock);
+		if (smap.count() == 0)
+		{
+			Out(SYS_CON|LOG_DEBUG) << "Stopping socketmonitor threads" << endl;
+			if (dt && dt->isRunning())
+				dt->stop();
+			if (ut && ut->isRunning())
+			{
+				ut->stop();
+				ut->signalDataReady();
+			}
+		}
+	}
+	
+	void SocketMonitor::signalPacketReady()
+	{
+		if (ut)
+			ut->signalDataReady();
+	}
+	
+	Uint32 SocketMonitor::newGroup(GroupType type,Uint32 limit)
+	{
+		lock();
+		Uint32 gid = next_group_id++;
+		if (type == UPLOAD_GROUP)
+			ut->addGroup(gid,limit);
+		else
+			dt->addGroup(gid,limit);
+		unlock();
+		return gid;
+	}
+		
+	void SocketMonitor::setGroupLimit(GroupType type,Uint32 gid,Uint32 limit)
+	{
+		lock();
+		if (type == UPLOAD_GROUP)
+			ut->setGroupLimit(gid,limit);
+		else
+			dt->setGroupLimit(gid,limit);
+		unlock();
+	}
+		
+	void SocketMonitor::removeGroup(GroupType type,Uint32 gid)
+	{
+		lock();
+		if (type == UPLOAD_GROUP)
+			ut->removeGroup(gid);
+		else
+			dt->removeGroup(gid);
+		unlock();
+	}
+
+}