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

« back to all changes in this revision

Viewing changes to src/tracker/tracker_dht.cc

  • 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
* 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:
 
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
#include "config.h"
 
38
 
 
39
#include <sstream>
 
40
#include <cstdio>
 
41
 
 
42
#include "dht/dht_router.h"
 
43
#include "torrent/connection_manager.h"
 
44
#include "torrent/download_info.h"
 
45
#include "torrent/dht_manager.h"
 
46
#include "torrent/exceptions.h"
 
47
#include "torrent/tracker_list.h"
 
48
 
 
49
#include "tracker_dht.h"
 
50
 
 
51
#include "globals.h"
 
52
#include "manager.h"
 
53
 
 
54
namespace torrent {
 
55
 
 
56
const char* TrackerDht::states[] = { "Idle", "Searching", "Announcing" };
 
57
 
 
58
TrackerDht::TrackerDht(TrackerList* parent, const std::string& url) :
 
59
  Tracker(parent, url),
 
60
  m_state(state_idle) {
 
61
 
 
62
  if (!manager->dht_manager()->is_valid())
 
63
    throw internal_error("Trying to add DHT tracker with no DHT manager.");
 
64
}
 
65
 
 
66
TrackerDht::~TrackerDht() {
 
67
  if (is_busy())
 
68
    manager->dht_manager()->router()->cancel_announce(NULL, this);
 
69
}
 
70
 
 
71
bool
 
72
TrackerDht::is_busy() const {
 
73
  return m_state != state_idle;
 
74
}
 
75
 
 
76
bool
 
77
TrackerDht::is_usable() const {
 
78
  return m_enabled && manager->dht_manager()->is_active();
 
79
}
 
80
 
 
81
void
 
82
TrackerDht::send_state(int state) {
 
83
  if (m_parent == NULL)
 
84
    throw internal_error("TrackerDht::send_state(...) does not have a valid m_parent.");
 
85
 
 
86
  if (is_busy()) {
 
87
    manager->dht_manager()->router()->cancel_announce(m_parent->info(), this);
 
88
 
 
89
    if (is_busy())
 
90
      throw internal_error("TrackerDht::send_state cancel_announce did not cancel announce.");
 
91
  }
 
92
 
 
93
  if (state == DownloadInfo::STOPPED)
 
94
    return;
 
95
 
 
96
  m_state = state_searching;
 
97
 
 
98
  if (!manager->dht_manager()->is_active())
 
99
    return receive_failed("DHT server not active.");
 
100
 
 
101
  manager->dht_manager()->router()->announce(m_parent->info(), this);
 
102
 
 
103
  set_normal_interval(20 * 60);
 
104
  set_min_interval(0);
 
105
}
 
106
 
 
107
void
 
108
TrackerDht::close() {
 
109
  if (is_busy())
 
110
    manager->dht_manager()->router()->cancel_announce(m_parent->info(), this);
 
111
}
 
112
 
 
113
TrackerDht::Type
 
114
TrackerDht::type() const {
 
115
  return TRACKER_DHT;
 
116
}
 
117
 
 
118
void
 
119
TrackerDht::receive_peers(raw_list peers) {
 
120
  if (!is_busy())
 
121
    throw internal_error("TrackerDht::receive_peers called while not busy.");
 
122
 
 
123
  m_peers.parse_address_bencode(peers);
 
124
}
 
125
 
 
126
void
 
127
TrackerDht::receive_success() {
 
128
  if (!is_busy())
 
129
    throw internal_error("TrackerDht::receive_success called while not busy.");
 
130
 
 
131
  m_state = state_idle;
 
132
  m_parent->receive_success(this, &m_peers);
 
133
  m_peers.clear();
 
134
}
 
135
 
 
136
void
 
137
TrackerDht::receive_failed(const char* msg) {
 
138
  if (!is_busy())
 
139
    throw internal_error("TrackerDht::receive_failed called while not busy.");
 
140
 
 
141
  m_state = state_idle;
 
142
  m_parent->receive_failed(this, msg);
 
143
  m_peers.clear();
 
144
}
 
145
 
 
146
void
 
147
TrackerDht::receive_progress(int replied, int contacted) {
 
148
  if (!is_busy())
 
149
    throw internal_error("TrackerDht::receive_status called while not busy.");
 
150
 
 
151
  m_replied = replied;
 
152
  m_contacted = contacted;
 
153
}
 
154
 
 
155
void
 
156
TrackerDht::get_status(char* buffer, int length) {
 
157
  if (!is_busy())
 
158
    throw internal_error("TrackerDht::get_status called while not busy.");
 
159
  
 
160
  snprintf(buffer, length, "[%s: %d/%d nodes replied]", states[m_state], m_replied, m_contacted);
 
161
}
 
162
 
 
163
}