mirror of
https://github.com/rakshasa/rtorrent.git
synced 2026-08-06 18:22:32 +00:00
rakshasa
ddf619f934
Seperated the algorithm extensions from functional. git-svn-id: svn://rakshasa.no/libtorrent/trunk/rtorrent@326 e378c898-3ddf-0310-93e7-cc216c733640
98 lines
2.1 KiB
C++
98 lines
2.1 KiB
C++
#include "config.h"
|
|
|
|
#include <algorithm>
|
|
#include <curl/multi.h>
|
|
#include <torrent/exceptions.h>
|
|
|
|
#include "utils/functional.h"
|
|
#include "curl_get.h"
|
|
#include "curl_stack.h"
|
|
|
|
namespace core {
|
|
|
|
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(),
|
|
utils::equal(msg->easy_handle, std::mem_fun(&CurlGet::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) {
|
|
if (curl_multi_fdset((CURLM*)m_handle, readfds, writefds, exceptfds, maxFd) > 0)
|
|
throw torrent::local_error("Error calling curl_multi_fdset");
|
|
}
|
|
|
|
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();
|
|
}
|
|
|
|
}
|