← Back to branch summary

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

« back to all changes in this revision

Viewing changes to src/net/socket_set.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/net/socket_set.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_NET_SOCKET_SET_H
 
38
#define LIBTORRENT_NET_SOCKET_SET_H
 
39
 
 
40
#include <list>
 
41
#include <vector>
 
42
#include <inttypes.h>
 
43
#include <rak/allocators.h>
 
44
 
 
45
#include "torrent/exceptions.h"
 
46
#include "torrent/event.h"
 
47
 
 
48
namespace torrent {
 
49
 
 
50
// SocketSet's Base is a vector of active SocketBase
 
51
// instances. 'm_table' is a vector with the size 'openMax', each
 
52
// element of which points to an active instance in the Base vector.
 
53
 
 
54
// Propably should rename to EventSet...
 
55
 
 
56
class SocketSet : private std::vector<Event*, rak::cacheline_allocator<> > {
 
57
public:
 
58
  typedef uint32_t    size_type;
 
59
 
 
60
  typedef std::vector<Event*, rak::cacheline_allocator<> > base_type;
 
61
  typedef std::vector<size_type, rak::cacheline_allocator<> > Table;
 
62
 
 
63
  static const size_type npos = static_cast<size_type>(-1);
 
64
 
 
65
  using base_type::value_type;
 
66
 
 
67
  using base_type::iterator;
 
68
  using base_type::reverse_iterator;
 
69
  using base_type::empty;
 
70
  using base_type::size;
 
71
 
 
72
  using base_type::begin;
 
73
  using base_type::end;
 
74
  using base_type::rbegin;
 
75
  using base_type::rend;
 
76
 
 
77
  bool                has(Event* s) const                    { return _index(s) != npos; }
 
78
 
 
79
  iterator            find(Event* s);
 
80
  void                insert(Event* s);
 
81
  void                erase(Event* s);
 
82
 
 
83
  // Remove all erased elements from the container.
 
84
  void                prepare();
 
85
  // Allocate storage for fd's with up to 'openMax' value. TODO: Remove reserve
 
86
  void                reserve(size_t openMax);
 
87
 
 
88
  size_t              max_size() const                       { return m_table.size(); }
 
89
 
 
90
private:
 
91
  size_type&          _index(Event* s)                       { return m_table[s->file_descriptor()]; }
 
92
  const size_type&    _index(Event* s) const                 { return m_table[s->file_descriptor()]; }
 
93
 
 
94
  inline void         _replace_with_last(size_type idx);
 
95
 
 
96
  // TODO: Table of indexes or iterators?
 
97
  Table               m_table;
 
98
  Table               m_erased;
 
99
};
 
100
 
 
101
inline SocketSet::iterator
 
102
SocketSet::find(Event* s) {
 
103
  if (_index(s) == npos)
 
104
    return end();
 
105
 
 
106
  return begin() + _index(s);
 
107
}
 
108
 
 
109
inline void
 
110
SocketSet::insert(Event* s) {
 
111
  if (static_cast<size_type>(s->file_descriptor()) >= m_table.size())
 
112
    throw internal_error("Tried to insert an out-of-bounds file descriptor to SocketSet");
 
113
 
 
114
  if (_index(s) != npos)
 
115
    return;
 
116
 
 
117
  _index(s) = size();
 
118
  base_type::push_back(s);
 
119
}
 
120
 
 
121
inline void
 
122
SocketSet::erase(Event* s) {
 
123
  if (static_cast<size_type>(s->file_descriptor()) >= m_table.size())
 
124
    throw internal_error("Tried to erase an out-of-bounds file descriptor from SocketSet");
 
125
 
 
126
  size_type idx = _index(s);
 
127
 
 
128
  if (idx == npos)
 
129
    return;
 
130
 
 
131
  _index(s) = npos;
 
132
 
 
133
  *(begin() + idx) = NULL;
 
134
  m_erased.push_back(idx);
 
135
}
 
136
 
 
137
}
 
138
 
 
139
#endif