mirror of
https://github.com/rakshasa/rtorrent.git
synced 2026-08-09 03:32:29 +00:00
kannibalox
6e976616a0
The erase/insert pattern can be replaced by a single method that updates an existing entry's timer, or inserting if it doesn't exist. This has much less impact than the libtorrent change it's based on, but might as well be consistent.
149 lines
4.3 KiB
C++
149 lines
4.3 KiB
C++
// rTorrent - BitTorrent client
|
|
// Copyright (C) 2005-2011, Jari Sundell
|
|
//
|
|
// 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
//
|
|
// In addition, as a special exception, the copyright holders give
|
|
// permission to link the code of portions of this program with the
|
|
// OpenSSL library under certain conditions as described in each
|
|
// individual source file, and distribute linked combinations
|
|
// including the two.
|
|
//
|
|
// You must obey the GNU General Public License in all respects for
|
|
// all of the code used other than OpenSSL. If you modify file(s)
|
|
// with this exception, you may extend this exception to your version
|
|
// of the file(s), but you are not obligated to do so. If you do not
|
|
// wish to do so, delete this exception statement from your version.
|
|
// If you delete this exception statement from all source files in the
|
|
// program, then also delete it here.
|
|
//
|
|
// Contact: Jari Sundell <jaris@ifi.uio.no>
|
|
//
|
|
// Skomakerveien 33
|
|
// 3185 Skoppum, NORWAY
|
|
|
|
#include "config.h"
|
|
|
|
#include <iostream>
|
|
#include <curl/curl.h>
|
|
#include <curl/easy.h>
|
|
#include <torrent/exceptions.h>
|
|
|
|
#include "globals.h"
|
|
#include "curl_get.h"
|
|
#include "curl_stack.h"
|
|
|
|
namespace core {
|
|
|
|
size_t
|
|
curl_get_receive_write(void* data, size_t size, size_t nmemb, void* handle) {
|
|
if (!((CurlGet*)handle)->stream()->write((const char*)data, size * nmemb).fail())
|
|
return size * nmemb;
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
CurlGet::~CurlGet() {
|
|
close();
|
|
}
|
|
|
|
void
|
|
CurlGet::start() {
|
|
if (is_busy())
|
|
throw torrent::internal_error("Tried to call CurlGet::start on a busy object.");
|
|
|
|
if (m_stream == NULL)
|
|
throw torrent::internal_error("Tried to call CurlGet::start without a valid output stream.");
|
|
|
|
m_handle = curl_easy_init();
|
|
|
|
if (m_handle == NULL)
|
|
throw torrent::internal_error("Call to curl_easy_init() failed.");
|
|
|
|
curl_easy_setopt(m_handle, CURLOPT_URL, m_url.c_str());
|
|
curl_easy_setopt(m_handle, CURLOPT_WRITEFUNCTION, &curl_get_receive_write);
|
|
curl_easy_setopt(m_handle, CURLOPT_WRITEDATA, this);
|
|
|
|
if (m_timeout != 0) {
|
|
curl_easy_setopt(m_handle, CURLOPT_CONNECTTIMEOUT, (long)60);
|
|
curl_easy_setopt(m_handle, CURLOPT_TIMEOUT, (long)m_timeout);
|
|
|
|
// Normally libcurl should handle the timeout. But sometimes that doesn't
|
|
// work right so we do a fallback timeout that just aborts the transfer.
|
|
m_taskTimeout.slot() = std::bind(&CurlGet::receive_timeout, this);
|
|
priority_queue_update(&taskScheduler, &m_taskTimeout, cachedTime + rak::timer::from_seconds(m_timeout + 5));
|
|
}
|
|
|
|
curl_easy_setopt(m_handle, CURLOPT_FORBID_REUSE, (long)1);
|
|
curl_easy_setopt(m_handle, CURLOPT_NOSIGNAL, (long)1);
|
|
curl_easy_setopt(m_handle, CURLOPT_FOLLOWLOCATION, (long)1);
|
|
curl_easy_setopt(m_handle, CURLOPT_MAXREDIRS, (long)5);
|
|
|
|
curl_easy_setopt(m_handle, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_WHATEVER);
|
|
|
|
curl_easy_setopt(m_handle, CURLOPT_ENCODING, "");
|
|
|
|
m_ipv6 = false;
|
|
|
|
m_stack->add_get(this);
|
|
}
|
|
|
|
void
|
|
CurlGet::close() {
|
|
priority_queue_erase(&taskScheduler, &m_taskTimeout);
|
|
if (!is_busy())
|
|
return;
|
|
|
|
m_stack->remove_get(this);
|
|
|
|
curl_easy_cleanup(m_handle);
|
|
|
|
m_handle = NULL;
|
|
}
|
|
|
|
void
|
|
CurlGet::retry_ipv6() {
|
|
CURL* nhandle = curl_easy_duphandle(m_handle);
|
|
|
|
curl_easy_setopt(nhandle, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V6);
|
|
curl_easy_cleanup(m_handle);
|
|
|
|
m_handle = nhandle;
|
|
m_ipv6 = true;
|
|
}
|
|
|
|
void
|
|
CurlGet::receive_timeout() {
|
|
return m_stack->transfer_done(m_handle, "Timed out");
|
|
}
|
|
|
|
double
|
|
CurlGet::size_done() {
|
|
double d = 0.0;
|
|
curl_easy_getinfo(m_handle, CURLINFO_SIZE_DOWNLOAD, &d);
|
|
|
|
return d;
|
|
}
|
|
|
|
double
|
|
CurlGet::size_total() {
|
|
double d = 0.0;
|
|
curl_easy_getinfo(m_handle, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &d);
|
|
|
|
return d;
|
|
}
|
|
|
|
}
|