diff options
Diffstat (limited to 'libktorrent/net/socketmonitor.cpp')
| -rw-r--r-- | libktorrent/net/socketmonitor.cpp | 173 |
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 * + * joris.guisson@gmail.com * + * * + * 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(); + } + +} |