← Back to branch summary

~ubuntu-branches/ubuntu/wily/libtorrent/wily-proposed

« back to all changes in this revision

Viewing changes to src/resource_manager.h

  • Committer: Bazaar Package Importer
  • Author(s): Rogério Brito
  • Date: 2011-03-20 01:06:18 UTC
  • mfrom: (1.1.13 upstream) (4.1.9 sid)
  • Revision ID: james.westby@ubuntu.com-20110320010618-g3wyylccqzqko73c
Tags: 0.12.7-5
http://bugs.debian.org/617791
* Use Steinar's "real" patch for IPv6. Addresses #490277, #618275,
  and Closes: #617791.
* Adapt libtorrent-0.12.6-ipv6-07.patch. It FTBFS otherwise.
* Add proper attibution to the IPv6 patch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/resource_manager.h
 
1
// libTorrent - BitTorrent library
 
2
// Copyright (C) 2005-2007, Jari Sundell
 
3
//
 
4
// This program is free software; you can redistribute it and/or modify
 
5
// it under the terms of the GNU General Public License as published by
 
6
// the Free Software Foundation; either version 2 of the License, or
 
7
// (at your option) any later version.
 
8
// 
 
9
// This program is distributed in the hope that it will be useful,
 
10
// but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
// GNU General Public License for more details.
 
13
// 
 
14
// You should have received a copy of the GNU General Public License
 
15
// along with this program; if not, write to the Free Software
 
16
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
17
//
 
18
// In addition, as a special exception, the copyright holders give
 
19
// permission to link the code of portions of this program with the
 
20
// OpenSSL library under certain conditions as described in each
 
21
// individual source file, and distribute linked combinations
 
22
// including the two.
 
23
//
 
24
// You must obey the GNU General Public License in all respects for
 
25
// all of the code used other than OpenSSL.  If you modify file(s)
 
26
// with this exception, you may extend this exception to your version
 
27
// of the file(s), but you are not obligated to do so.  If you do not
 
28
// wish to do so, delete this exception statement from your version.
 
29
// If you delete this exception statement from all source files in the
 
30
// program, then also delete it here.
 
31
//
 
32
// Contact:  Jari Sundell <jaris@ifi.uio.no>
 
33
//
 
34
//           Skomakerveien 33
 
35
//           3185 Skoppum, NORWAY
 
36
 
 
37
#ifndef LIBTORRENT_RESOURCE_MANAGER_H
 
38
#define LIBTORRENT_RESOURCE_MANAGER_H
 
39
 
 
40
#include <vector>
 
41
#include <inttypes.h>
 
42
 
 
43
namespace torrent {
 
44
 
 
45
// This class will handle the division of various resources like
 
46
// uploads. For now the weight is equal to the value of the priority.
 
47
//
 
48
// Add unlimited handling later.
 
49
 
 
50
class DownloadMain;
 
51
 
 
52
class ResourceManager : public std::vector<std::pair<uint16_t, DownloadMain*> > {
 
53
public:
 
54
  typedef std::vector<std::pair<uint16_t, DownloadMain*> > base_type;
 
55
  typedef base_type::value_type                            value_type;
 
56
 
 
57
  using base_type::begin;
 
58
  using base_type::end;
 
59
 
 
60
  ResourceManager() :
 
61
    m_currentlyUploadUnchoked(0),
 
62
    m_currentlyDownloadUnchoked(0),
 
63
    m_maxUploadUnchoked(0),
 
64
    m_maxDownloadUnchoked(0) {}
 
65
  ~ResourceManager();
 
66
 
 
67
  void                insert(DownloadMain* d, uint16_t priority);
 
68
  void                erase(DownloadMain* d);
 
69
 
 
70
  iterator            find(DownloadMain* d);
 
71
  void                set_priority(iterator itr, uint16_t pri);
 
72
 
 
73
  // When setting this, make sure you choke peers, else change
 
74
  // receive_can_unchoke.
 
75
  unsigned int        currently_upload_unchoked() const         { return m_currentlyUploadUnchoked; }
 
76
  unsigned int        currently_download_unchoked() const       { return m_currentlyDownloadUnchoked; }
 
77
 
 
78
  unsigned int        max_upload_unchoked() const               { return m_maxUploadUnchoked; }
 
79
  unsigned int        max_download_unchoked() const             { return m_maxDownloadUnchoked; }
 
80
 
 
81
  void                set_max_upload_unchoked(unsigned int m)   { m_maxUploadUnchoked = m; }
 
82
  void                set_max_download_unchoked(unsigned int m) { m_maxDownloadUnchoked = m; }
 
83
 
 
84
  void                receive_upload_unchoke(int num);
 
85
  void                receive_download_unchoke(int num);
 
86
 
 
87
  unsigned int        retrieve_upload_can_unchoke();
 
88
  unsigned int        retrieve_download_can_unchoke();
 
89
 
 
90
  void                receive_tick();
 
91
 
 
92
private:
 
93
  unsigned int        total_weight() const;
 
94
 
 
95
  void                balance_upload_unchoked(unsigned int weight);
 
96
  void                balance_download_unchoked(unsigned int weight);
 
97
 
 
98
  unsigned int        m_currentlyUploadUnchoked;
 
99
  unsigned int        m_currentlyDownloadUnchoked;
 
100
 
 
101
  unsigned int        m_maxUploadUnchoked;
 
102
  unsigned int        m_maxDownloadUnchoked;
 
103
};
 
104
 
 
105
}
 
106
 
 
107
#endif