* Enable custom throttles, both per-download or per-IP. See http://libtorrent.rakshasa.no/ticket/20 for info and instructions. Patch by Josef Drexler.

git-svn-id: svn://rakshasa.no/libtorrent/trunk/rtorrent@1091 e378c898-3ddf-0310-93e7-cc216c733640

src/core/manager.h

@@ -40,8 +40,11 @@
 #include <iosfwd>
 #include <vector>
 
+#include <torrent/connection_manager.h>
+
 #include "download_list.h"
 #include "poll_manager.h"
+#include "range_map.h"
 #include "log.h"
 
 namespace torrent {
@@ -57,6 +60,8 @@ namespace core {
 class DownloadStore;
 class HttpQueue;
 
+typedef std::map<std::string, torrent::ThrottlePair> ThrottleMap;
+
 class View;
 
 class Manager {
@@ -84,6 +89,13 @@ public:
   Log&                get_log_important()                 { return m_logImportant; }
   Log&                get_log_complete()                  { return m_logComplete; }
 
+  ThrottleMap&          throttles()                       { return m_throttles; }
+  torrent::ThrottlePair get_throttle(const std::string& name);
+
+  // Use custom throttle for the given range of IP addresses.
+  void                  set_address_throttle(uint32_t begin, uint32_t end, torrent::ThrottlePair throttles);
+  torrent::ThrottlePair get_address_throttle(const sockaddr* addr);
+
   // Really should find a more descriptive name.
   void                initialize_first();
   void                initialize_second();
@@ -120,6 +132,8 @@ public:
   void                try_create_download_expand(const std::string& uri, int flags, command_list_type commands = command_list_type());
 
 private:
+  typedef RangeMap<uint32_t, torrent::ThrottlePair> AddressThrottleMap;
+
   void                create_http(const std::string& uri);
   void                create_final(std::istream* s);
 
@@ -136,6 +150,9 @@ private:
 
   View*               m_hashingView;
 
+  ThrottleMap         m_throttles;
+  AddressThrottleMap  m_addressThrottles;
+
   PollManager*        m_pollManager;
   Log                 m_logImportant;
   Log                 m_logComplete;