mirror of
https://github.com/rakshasa/rtorrent.git
synced 2026-08-06 18:22:32 +00:00
rakshasa
bd7849199c
git-svn-id: svn://rakshasa.no/libtorrent/trunk/rtorrent@252 e378c898-3ddf-0310-93e7-cc216c733640
123 lines
2.6 KiB
C++
123 lines
2.6 KiB
C++
#include "config.h"
|
|
|
|
#include <algorithm>
|
|
#include <functional>
|
|
#include <curl/multi.h>
|
|
#include <torrent/exceptions.h>
|
|
|
|
#include "curl_get.h"
|
|
#include "curl_stack.h"
|
|
|
|
namespace core {
|
|
|
|
template <typename Type, typename Ftor>
|
|
struct _equal {
|
|
_equal(Type t, Ftor f) : m_t(t), m_f(f) {}
|
|
|
|
template <typename Arg>
|
|
bool operator () (Arg& a) {
|
|
return m_t == m_f(a);
|
|
}
|
|
|
|
Type m_t;
|
|
Ftor m_f;
|
|
};
|
|
|
|
template <typename Type, typename Ftor>
|
|
_equal<Type, Ftor> equal(Type t, Ftor f) {
|
|
return _equal<Type, Ftor>(t, f);
|
|
}
|
|
|
|
CurlStack::CurlStack() :
|
|
m_handle((void*)curl_multi_init()),
|
|
m_size(0) {
|
|
}
|
|
|
|
CurlStack::~CurlStack() {
|
|
while (!m_getList.empty())
|
|
m_getList.front()->close();
|
|
|
|
curl_multi_cleanup((CURLM*)m_handle);
|
|
}
|
|
|
|
void
|
|
CurlStack::perform() {
|
|
int s;
|
|
CURLMcode code;
|
|
|
|
do {
|
|
code = curl_multi_perform((CURLM*)m_handle, &s);
|
|
|
|
if (code > 0)
|
|
throw torrent::local_error("Error calling curl_multi_perform");
|
|
|
|
if (s != m_size) {
|
|
// Done with some handles.
|
|
int t;
|
|
|
|
do {
|
|
CURLMsg* msg = curl_multi_info_read((CURLM*)m_handle, &t);
|
|
|
|
CurlGetList::iterator itr = std::find_if(m_getList.begin(), m_getList.end(),
|
|
equal(msg->easy_handle, std::mem_fun(&CurlGet::handle)));
|
|
|
|
// eq(call_member(&CurlGet::handle),
|
|
// value(msg->easy_handle)));
|
|
|
|
if (itr == m_getList.end())
|
|
throw torrent::client_error("Could not find CurlGet with the right easy_handle");
|
|
|
|
(*itr)->perform(msg);
|
|
} while (t);
|
|
}
|
|
|
|
} while (code == CURLM_CALL_MULTI_PERFORM);
|
|
}
|
|
|
|
void
|
|
CurlStack::fdset(fd_set* readfds, fd_set* writefds, fd_set* exceptfds, int* maxFd) {
|
|
int f;
|
|
|
|
if (curl_multi_fdset((CURLM*)m_handle, readfds, writefds, exceptfds, &f) > 0)
|
|
throw torrent::local_error("Error calling curl_multi_fdset");
|
|
|
|
*maxFd = std::max(f, *maxFd);
|
|
}
|
|
|
|
void
|
|
CurlStack::add_get(CurlGet* get) {
|
|
CURLMcode code;
|
|
|
|
if ((code = curl_multi_add_handle((CURLM*)m_handle, get->handle())) > 0)
|
|
throw torrent::local_error("curl_multi_add_handle \"" + std::string(curl_multi_strerror(code)));
|
|
|
|
m_size++;
|
|
m_getList.push_back(get);
|
|
}
|
|
|
|
void
|
|
CurlStack::remove_get(CurlGet* get) {
|
|
if (curl_multi_remove_handle((CURLM*)m_handle, get->handle()) > 0)
|
|
throw torrent::local_error("Error calling curl_multi_remove_handle");
|
|
|
|
CurlGetList::iterator itr = std::find(m_getList.begin(), m_getList.end(), get);
|
|
|
|
if (itr == m_getList.end())
|
|
throw torrent::client_error("Could not find CurlGet when calling CurlStack::remove");
|
|
|
|
m_size--;
|
|
m_getList.erase(itr);
|
|
}
|
|
|
|
void
|
|
CurlStack::init() {
|
|
curl_global_init(CURL_GLOBAL_ALL);
|
|
}
|
|
|
|
void
|
|
CurlStack::cleanup() {
|
|
curl_global_cleanup();
|
|
}
|
|
|
|
}
|