diff options
author | tpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da> | 2010-01-20 02:37:40 +0000 |
---|---|---|
committer | tpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da> | 2010-01-20 02:37:40 +0000 |
commit | 9ad5c7b5e23b4940e7a3ea3ca3a6fb77e6a8fab0 (patch) | |
tree | d088b5210e77d9fa91d954d8550e00e372b47378 /libktorrent/net/socketmonitor.h | |
download | ktorrent-9ad5c7b5e23b4940e7a3ea3ca3a6fb77e6a8fab0.tar.gz ktorrent-9ad5c7b5e23b4940e7a3ea3ca3a6fb77e6a8fab0.zip |
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
Diffstat (limited to 'libktorrent/net/socketmonitor.h')
-rw-r--r-- | libktorrent/net/socketmonitor.h | 118 |
1 files changed, 118 insertions, 0 deletions
diff --git a/libktorrent/net/socketmonitor.h b/libktorrent/net/socketmonitor.h new file mode 100644 index 0000000..79e4a2e --- /dev/null +++ b/libktorrent/net/socketmonitor.h @@ -0,0 +1,118 @@ +/*************************************************************************** + * 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. * + ***************************************************************************/ +#ifndef NETSOCKETMONITOR_H +#define NETSOCKETMONITOR_H + + +#include <qmutex.h> +#include <qptrlist.h> +#include <util/constants.h> + + +namespace net +{ + using bt::Uint32; + + class BufferedSocket; + class UploadThread; + class DownloadThread; + + /** + * @author Joris Guisson <joris.guisson@gmail.com> + * + * Monitors all sockets for upload and download traffic. + * It uses two threads to do this. + */ + class SocketMonitor + { + static SocketMonitor self; + + QMutex mutex; + UploadThread* ut; + DownloadThread* dt; + QPtrList<BufferedSocket> smap; + Uint32 next_group_id; + + SocketMonitor(); + public: + virtual ~SocketMonitor(); + + /// Add a new socket, will start the threads if necessary + void add(BufferedSocket* sock); + + /// Remove a socket, will stop threads if no more sockets are left + void remove(BufferedSocket* sock); + + enum GroupType + { + UPLOAD_GROUP, + DOWNLOAD_GROUP + }; + + + /** + * Creata a new upload or download group + * @param type Wether it is an upload or download group + * @param limit Limit of group in bytes/s + * @return The group ID + */ + Uint32 newGroup(GroupType type,Uint32 limit); + + /** + * Change the group limit + * @param type The group type + * @param gid The group id + * @param limit The limit + */ + void setGroupLimit(GroupType type,Uint32 gid,Uint32 limit); + + /** + * Remove a group + * @param type The group type + * @param gid The group id + */ + void removeGroup(GroupType type,Uint32 gid); + + typedef QPtrList<BufferedSocket>::iterator Itr; + + /// Get the begin of the list of sockets + Itr begin() {return smap.begin();} + + /// Get the end of the list of sockets + Itr end() {return smap.end();} + + /// lock the monitor + void lock(); + + /// unlock the monitor + void unlock(); + + /// Tell upload thread a packet is ready + void signalPacketReady(); + + static void setDownloadCap(Uint32 bytes_per_sec); + static void setUploadCap(Uint32 bytes_per_sec); + static void setSleepTime(Uint32 sleep_time); + static SocketMonitor & instance() {return self;} + }; + +} + +#endif |