summaryrefslogtreecommitdiffabout
Unidiff
Diffstat (more/less context) (show whitespace changes)
-rw-r--r--include/opkele/curl.h15
-rw-r--r--lib/Makefile.am3
-rw-r--r--lib/consumer.cc105
-rw-r--r--lib/curl.cc42
4 files changed, 101 insertions, 64 deletions
diff --git a/include/opkele/curl.h b/include/opkele/curl.h
index 8794ece..6a7d084 100644
--- a/include/opkele/curl.h
+++ b/include/opkele/curl.h
@@ -3,2 +3,3 @@
3 3
4#include <cassert>
4#include <curl/curl.h> 5#include <curl/curl.h>
@@ -15,5 +16,5 @@ namespace opkele {
15 curl_t(CURL *c) : _c(c) { } 16 curl_t(CURL *c) : _c(c) { }
16 ~curl_t() throw() { if(_c) curl_easy_cleanup(_c); } 17 virtual ~curl_t() throw();
17 18
18 curl_t& operator=(CURL *c) { if(_c) curl_easy_cleanup(_c); _c=c; return *this; } 19 curl_t& operator=(CURL *c);
19 20
@@ -21,2 +22,12 @@ namespace opkele {
21 operator CURL*(void) { return _c; } 22 operator CURL*(void) { return _c; }
23
24 CURLcode misc_sets();
25
26 template<typename PT>
27 inline CURLcode easy_setopt(CURLoption o,PT p) { assert(_c); return curl_easy_setopt(_c,o,p); }
28 CURLcode easy_perform() { assert(_c); return curl_easy_perform(_c); }
29 template<typename IT>
30 inline CURLcode easy_getinfo(CURLINFO i,IT p) { assert(_c); return curl_easy_getinfo(_c,i,p); }
31
32 static inline CURL *easy_init() { return curl_easy_init(); }
22 }; 33 };
diff --git a/lib/Makefile.am b/lib/Makefile.am
index b008a52..0fe705a 100644
--- a/lib/Makefile.am
+++ b/lib/Makefile.am
@@ -24,3 +24,4 @@ libopkele_la_SOURCES = \
24 sreg.cc \ 24 sreg.cc \
25 extension_chain.cc 25 extension_chain.cc \
26 curl.cc
26libopkele_la_LDFLAGS = \ 27libopkele_la_LDFLAGS = \
diff --git a/lib/consumer.cc b/lib/consumer.cc
index 20f4174..c155157 100644
--- a/lib/consumer.cc
+++ b/lib/consumer.cc
@@ -18,2 +18,3 @@ namespace opkele {
18 using namespace std; 18 using namespace std;
19 using util::curl_t;
19 20
@@ -63,20 +64,2 @@ namespace opkele {
63 64
64 static CURLcode curl_misc_sets(CURL* c) {
65 CURLcode r;
66 (r=curl_easy_setopt(c,CURLOPT_FOLLOWLOCATION,1))
67 || (r=curl_easy_setopt(c,CURLOPT_MAXREDIRS,5))
68 || (r=curl_easy_setopt(c,CURLOPT_DNS_CACHE_TIMEOUT,120))
69 || (r=curl_easy_setopt(c,CURLOPT_DNS_USE_GLOBAL_CACHE,1))
70 || (r=curl_easy_setopt(c,CURLOPT_USERAGENT,PACKAGE_NAME"/"PACKAGE_SRC_VERSION))
71 || (r=curl_easy_setopt(c,CURLOPT_TIMEOUT,20))
72 #ifdefDISABLE_CURL_SSL_VERIFYHOST
73 || (r=curl_easy_setopt(c,CURLOPT_SSL_VERIFYHOST,0))
74#endif
75 #ifdefDISABLE_CURL_SSL_VERIFYPEER
76 || (r=curl_easy_setopt(c,CURLOPT_SSL_VERIFYPEER,0))
77#endif
78 ;
79 return r;
80 }
81
82 static size_t _curl_tostring(void *ptr,size_t size,size_t nmemb,void *stream) { 65 static size_t _curl_tostring(void *ptr,size_t size,size_t nmemb,void *stream) {
@@ -103,19 +86,19 @@ namespace opkele {
103 request += util::url_encode(util::bignum_to_base64(dh->pub_key)); 86 request += util::url_encode(util::bignum_to_base64(dh->pub_key));
104 util::curl_t curl = curl_easy_init(); 87 curl_t curl = curl_t::easy_init();
105 if(!curl) 88 if(!curl)
106 throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); 89 throw exception_curl(OPKELE_CP_ "failed to initialize curl");
107 string response; 90 string response;
108 CURLcode r; 91 CURLcode r;
109 (r=curl_misc_sets(curl)) 92 (r=curl.misc_sets())
110 || (r=curl_easy_setopt(curl,CURLOPT_URL,server.c_str())) 93 || (r=curl.easy_setopt(CURLOPT_URL,server.c_str()))
111 || (r=curl_easy_setopt(curl,CURLOPT_POST,1)) 94 || (r=curl.easy_setopt(CURLOPT_POST,1))
112 || (r=curl_easy_setopt(curl,CURLOPT_POSTFIELDS,request.data())) 95 || (r=curl.easy_setopt(CURLOPT_POSTFIELDS,request.data()))
113 || (r=curl_easy_setopt(curl,CURLOPT_POSTFIELDSIZE,request.length())) 96 || (r=curl.easy_setopt(CURLOPT_POSTFIELDSIZE,request.length()))
114 || (r=curl_easy_setopt(curl,CURLOPT_WRITEFUNCTION,_curl_tostring)) 97 || (r=curl.easy_setopt(CURLOPT_WRITEFUNCTION,_curl_tostring))
115 || (r=curl_easy_setopt(curl,CURLOPT_WRITEDATA,&response)) 98 || (r=curl.easy_setopt(CURLOPT_WRITEDATA,&response))
116 ; 99 ;
117 if(r) 100 if(r)
118 throw exception_curl(OPKELE_CP_ "failed to curl_easy_setopt()",r); 101 throw exception_curl(OPKELE_CP_ "failed to set curly options",r);
119 if( (r=curl_easy_perform(curl)) ) 102 if( (r=curl.easy_perform()) )
120 throw exception_curl(OPKELE_CP_ "failed to curl_easy_perform()",r); 103 throw exception_curl(OPKELE_CP_ "failed to perform curly request",r);
121 params_t p; p.parse_keyvalues(response); 104 params_t p; p.parse_keyvalues(response);
@@ -263,19 +246,19 @@ namespace opkele {
263 } 246 }
264 util::curl_t curl = curl_easy_init(); 247 curl_t curl = curl_t::easy_init();
265 if(!curl) 248 if(!curl)
266 throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); 249 throw exception_curl(OPKELE_CP_ "failed to initialize curl");
267 string response; 250 string response;
268 CURLcode r; 251 CURLcode r;
269 (r=curl_misc_sets(curl)) 252 (r=curl.misc_sets())
270 || (r=curl_easy_setopt(curl,CURLOPT_URL,server.c_str())) 253 || (r=curl.easy_setopt(CURLOPT_URL,server.c_str()))
271 || (r=curl_easy_setopt(curl,CURLOPT_POST,1)) 254 || (r=curl.easy_setopt(CURLOPT_POST,1))
272 || (r=curl_easy_setopt(curl,CURLOPT_POSTFIELDS,request.data())) 255 || (r=curl.easy_setopt(CURLOPT_POSTFIELDS,request.data()))
273 || (r=curl_easy_setopt(curl,CURLOPT_POSTFIELDSIZE,request.length())) 256 || (r=curl.easy_setopt(CURLOPT_POSTFIELDSIZE,request.length()))
274 || (r=curl_easy_setopt(curl,CURLOPT_WRITEFUNCTION,_curl_tostring)) 257 || (r=curl.easy_setopt(CURLOPT_WRITEFUNCTION,_curl_tostring))
275 || (r=curl_easy_setopt(curl,CURLOPT_WRITEDATA,&response)) 258 || (r=curl.easy_setopt(CURLOPT_WRITEDATA,&response))
276 ; 259 ;
277 if(r) 260 if(r)
278 throw exception_curl(OPKELE_CP_ "failed to curl_easy_setopt()",r); 261 throw exception_curl(OPKELE_CP_ "failed to set curly options",r);
279 if( (r=curl_easy_perform(curl)) ) 262 if( (r=curl.easy_perform()) )
280 throw exception_curl(OPKELE_CP_ "failed to curl_easy_perform()",r); 263 throw exception_curl(OPKELE_CP_ "failed to perform curly request",r);
281 params_t pp; pp.parse_keyvalues(response); 264 params_t pp; pp.parse_keyvalues(response);
@@ -296,17 +279,17 @@ namespace opkele {
296 delegate.erase(); 279 delegate.erase();
297 util::curl_t curl = curl_easy_init(); 280 curl_t curl = curl_t::easy_init();
298 if(!curl) 281 if(!curl)
299 throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); 282 throw exception_curl(OPKELE_CP_ "failed to initialize curl");
300 string html; 283 string html;
301 CURLcode r; 284 CURLcode r;
302 (r=curl_misc_sets(curl)) 285 (r=curl.misc_sets())
303 || (r=curl_easy_setopt(curl,CURLOPT_URL,url.c_str())) 286 || (r=curl.easy_setopt(CURLOPT_URL,url.c_str()))
304 || (r=curl_easy_setopt(curl,CURLOPT_WRITEFUNCTION,_curl_tostring)) 287 || (r=curl.easy_setopt(CURLOPT_WRITEFUNCTION,_curl_tostring))
305 || (r=curl_easy_setopt(curl,CURLOPT_WRITEDATA,&html)) 288 || (r=curl.easy_setopt(CURLOPT_WRITEDATA,&html))
306 ; 289 ;
307 if(r) 290 if(r)
308 throw exception_curl(OPKELE_CP_ "failed to curl_easy_setopt()",r); 291 throw exception_curl(OPKELE_CP_ "failed to set curly options",r);
309 r = curl_easy_perform(curl); 292 r = curl.easy_perform();
310 if(r && r!=CURLE_WRITE_ERROR) 293 if(r && r!=CURLE_WRITE_ERROR)
311 throw exception_curl(OPKELE_CP_ "failed to curl_easy_perform()",r); 294 throw exception_curl(OPKELE_CP_ "failed to perform curly request",r);
312 static const char *re_bre = "<\\s*body\\b", *re_hdre = "<\\s*head[^>]*>", 295 static const char *re_bre = "<\\s*body\\b", *re_hdre = "<\\s*head[^>]*>",
@@ -402,20 +385,20 @@ namespace opkele {
402 string rv = normalize(url); 385 string rv = normalize(url);
403 util::curl_t curl = curl_easy_init(); 386 curl_t curl = curl_t::easy_init();
404 if(!curl) 387 if(!curl)
405 throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); 388 throw exception_curl(OPKELE_CP_ "failed to initialize curl()");
406 string html; 389 string html;
407 CURLcode r; 390 CURLcode r;
408 (r=curl_misc_sets(curl)) 391 (r=curl.misc_sets())
409 || (r=curl_easy_setopt(curl,CURLOPT_URL,rv.c_str())) 392 || (r=curl.easy_setopt(CURLOPT_URL,rv.c_str()))
410 || (r=curl_easy_setopt(curl,CURLOPT_NOBODY,1)) 393 || (r=curl.easy_setopt(CURLOPT_NOBODY,1))
411 ; 394 ;
412 if(r) 395 if(r)
413 throw exception_curl(OPKELE_CP_ "failed to curl_easy_setopt()",r); 396 throw exception_curl(OPKELE_CP_ "failed to set curly options",r);
414 r = curl_easy_perform(curl); 397 r = curl.easy_perform();
415 if(r) 398 if(r)
416 throw exception_curl(OPKELE_CP_ "failed to curl_easy_perform()",r); 399 throw exception_curl(OPKELE_CP_ "failed to perform curly request",r);
417 const char *eu = 0; 400 const char *eu = 0;
418 r = curl_easy_getinfo(curl,CURLINFO_EFFECTIVE_URL,&eu); 401 r = curl.easy_getinfo(CURLINFO_EFFECTIVE_URL,&eu);
419 if(r) 402 if(r)
420 throw exception_curl(OPKELE_CP_ "failed to curl_easy_getinfo(..CURLINFO_EFFECTIVE_URL..)",r); 403 throw exception_curl(OPKELE_CP_ "failed to get CURLINFO_EFFECTIVE_URL",r);
421 rv = eu; 404 rv = eu;
diff --git a/lib/curl.cc b/lib/curl.cc
new file mode 100644
index 0000000..418aa79
--- a/dev/null
+++ b/lib/curl.cc
@@ -0,0 +1,42 @@
1#include <opkele/curl.h>
2
3#include "config.h"
4
5namespace opkele {
6
7 namespace util {
8
9 curl_t::~curl_t() throw() {
10 if(_c)
11 curl_easy_cleanup(_c);
12 }
13
14 curl_t& curl_t::operator=(CURL *c) {
15 if(_c)
16 curl_easy_cleanup(_c);
17 _c = c;
18 return *this;
19 }
20
21 CURLcode curl_t::misc_sets() {
22 assert(_c);
23 CURLcode r;
24 (r=easy_setopt(CURLOPT_FOLLOWLOCATION,1))
25 || (r=easy_setopt(CURLOPT_MAXREDIRS,5))
26 || (r=easy_setopt(CURLOPT_DNS_CACHE_TIMEOUT,120))
27 || (r=easy_setopt(CURLOPT_DNS_USE_GLOBAL_CACHE,1))
28 || (r=easy_setopt(CURLOPT_USERAGENT,PACKAGE_NAME"/"PACKAGE_SRC_VERSION))
29 || (r=easy_setopt(CURLOPT_TIMEOUT,20))
30 #ifdefDISABLE_CURL_SSL_VERIFYHOST
31 || (r=easy_setopt(CURLOPT_SSL_VERIFYHOST,0))
32#endif
33 #ifdefDISABLE_CURL_SSL_VERIFYPEER
34 || (r=easy_setopt(CURLOPT_SSL_VERIFYPEER,0))
35#endif
36 ;
37 return r;
38 }
39
40 }
41
42}