From 94fe2125da8619a76c5d38829543cb5818b73731 Mon Sep 17 00:00:00 2001
From: "Nicolas \"Pixel\" Noble" <pixel@nobis-crew.org>
Date: Wed, 18 Jun 2014 08:30:53 -0700
Subject: Starting to integrate cURL into Balau.
---
src/TaskMan.cc | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 59 insertions(+)
(limited to 'src')
diff --git a/src/TaskMan.cc b/src/TaskMan.cc
index 8e4e941..c50e57a 100644
--- a/src/TaskMan.cc
+++ b/src/TaskMan.cc
@@ -9,7 +9,10 @@
#include <Windows.h>
#endif
+#include <curl/curl.h>
+
static Balau::AsyncManager s_async;
+static CURLSH * s_curlshared = NULL;
namespace {
@@ -33,9 +36,49 @@ class Stopper : public Balau::Task {
int m_code;
};
+class CurlSharedManager : public Balau::AtStart, Balau::AtExit {
+ public:
+ CurlSharedManager() : AtStart(0), AtExit(0) { }
+ struct SharedLocks {
+ Balau::RWLock cookie, dns, ssl_session;
+ };
+ static void lock_function(CURL *handle, curl_lock_data data, curl_lock_access access, void * userptr) {
+ SharedLocks * locks = (SharedLocks *) userptr;
+ Balau::RWLock * lock = NULL;
+ switch (data) {
+ case CURL_LOCK_DATA_COOKIE: lock = &locks->cookie; break;
+ case CURL_LOCK_DATA_DNS: lock = &locks->dns; break;
+ case CURL_LOCK_DATA_SSL_SESSION: lock = &locks->ssl_session; break;
+ default: Failure("Unknown lock");
+ }
+ switch (access) {
+ case CURL_LOCK_ACCESS_SHARED: lock->enterR(); break;
+ case CURL_LOCK_ACCESS_SINGLE: lock->enterW(); break;
+ default: Failure("Unknown access");
+ }
+ }
+ static void unlock_function(CURL *handle, curl_lock_data data, void * userptr) {
+ SharedLocks * locks = (SharedLocks *) userptr;
+ }
+ void doStart() {
+ static SharedLocks locks;
+ s_curlshared = curl_share_init();
+ curl_share_setopt(s_curlshared, CURLSHOPT_SHARE, CURL_LOCK_DATA_COOKIE);
+ curl_share_setopt(s_curlshared, CURLSHOPT_SHARE, CURL_LOCK_DATA_DNS);
+ curl_share_setopt(s_curlshared, CURLSHOPT_SHARE, CURL_LOCK_DATA_SSL_SESSION);
+ curl_share_setopt(s_curlshared, CURLSHOPT_USERDATA, &locks);
+ curl_share_setopt(s_curlshared, CURLSHOPT_LOCKFUNC, lock_function);
+ curl_share_setopt(s_curlshared, CURLSHOPT_UNLOCKFUNC, lock_function);
+ }
+ void doExit() {
+ curl_share_cleanup(s_curlshared);
+ }
+};
+
};
static AsyncStarter s_asyncStarter;
+static CurlSharedManager s_curlSharedmManager;
void Stopper::Do() {
getTaskMan()->stopMe(m_code);
@@ -178,6 +221,8 @@ Balau::TaskMan::TaskMan() {
s_scheduler.registerTaskMan(this);
m_nStacks = 0;
+
+ m_curlMulti = curl_multi_init();
}
#ifdef _WIN32
@@ -210,6 +255,20 @@ Balau::TaskMan::~TaskMan() {
// probably way more work to do here in order to clean up tasks from that thread
m_evt.stop();
ev_loop_destroy(m_loop);
+ curl_multi_cleanup(m_curlMulti);
+ curl_multi_setopt(m_curlMulti, CURLMOPT_SOCKETFUNCTION, reinterpret_cast<curl_socket_callback>(curlSocketCallback));
+ curl_multi_setopt(m_curlMulti, CURLMOPT_SOCKETDATA, this);
+ curl_multi_setopt(m_curlMulti, CURLMOPT_TIMERFUNCTION, reinterpret_cast <curl_multi_timer_callback>(curlMultiTimerCallback));
+ curl_multi_setopt(m_curlMulti, CURLMOPT_TIMERDATA, this);
+ curl_multi_setopt(m_curlMulti, CURLMOPT_PIPELINING, 1L);
+}
+
+int Balau::TaskMan::curlSocketCallback(CURL * easy, curl_socket_t s, int what, void * userp, void * socketp) {
+ return 0;
+}
+
+int Balau::TaskMan::curlMultiTimerCallback(CURLM * multi, long timeout_ms, void * userp) {
+ return 0;
}
void * Balau::TaskMan::getStack() {
--
cgit v1.2.3