-rw-r--r-- | lib/consumer.cc | 25 |
1 files changed, 5 insertions, 20 deletions
diff --git a/lib/consumer.cc b/lib/consumer.cc index 7881f5f..20f4174 100644 --- a/lib/consumer.cc +++ b/lib/consumer.cc | |||
@@ -1,19 +1,18 @@ | |||
1 | #include <algorithm> | 1 | #include <algorithm> |
2 | #include <cassert> | 2 | #include <cassert> |
3 | #include <cstring> | 3 | #include <cstring> |
4 | #include <opkele/util.h> | 4 | #include <opkele/util.h> |
5 | #include <opkele/curl.h> | ||
5 | #include <opkele/exception.h> | 6 | #include <opkele/exception.h> |
6 | #include <opkele/data.h> | 7 | #include <opkele/data.h> |
7 | #include <opkele/consumer.h> | 8 | #include <opkele/consumer.h> |
8 | #include <openssl/sha.h> | 9 | #include <openssl/sha.h> |
9 | #include <openssl/hmac.h> | 10 | #include <openssl/hmac.h> |
10 | #include <curl/curl.h> | ||
11 | |||
12 | #include <iostream> | 11 | #include <iostream> |
13 | 12 | ||
14 | #include "config.h" | 13 | #include "config.h" |
15 | 14 | ||
16 | #include <pcre.h> | 15 | #include <pcre.h> |
17 | 16 | ||
18 | namespace opkele { | 17 | namespace opkele { |
19 | using namespace std; | 18 | using namespace std; |
@@ -57,30 +56,16 @@ namespace opkele { | |||
57 | 56 | ||
58 | int exec(const string& s,pcre_matches_t& m) { | 57 | int exec(const string& s,pcre_matches_t& m) { |
59 | if(!_p) | 58 | if(!_p) |
60 | throw internal_error(OPKELE_CP_ "Trying to execute absent regexp"); | 59 | throw internal_error(OPKELE_CP_ "Trying to execute absent regexp"); |
61 | return pcre_exec(_p,NULL,s.c_str(),s.length(),0,0,m._ov,m._s); | 60 | return pcre_exec(_p,NULL,s.c_str(),s.length(),0,0,m._ov,m._s); |
62 | } | 61 | } |
63 | }; | 62 | }; |
64 | 63 | ||
65 | class curl_t { | ||
66 | public: | ||
67 | CURL *_c; | ||
68 | |||
69 | curl_t() : _c(0) { } | ||
70 | curl_t(CURL *c) : _c(c) { } | ||
71 | ~curl_t() throw() { if(_c) curl_easy_cleanup(_c); } | ||
72 | |||
73 | curl_t& operator=(CURL *c) { if(_c) curl_easy_cleanup(_c); _c=c; return *this; } | ||
74 | |||
75 | operator const CURL*(void) const { return _c; } | ||
76 | operator CURL*(void) { return _c; } | ||
77 | }; | ||
78 | |||
79 | static CURLcode curl_misc_sets(CURL* c) { | 64 | static CURLcode curl_misc_sets(CURL* c) { |
80 | CURLcode r; | 65 | CURLcode r; |
81 | (r=curl_easy_setopt(c,CURLOPT_FOLLOWLOCATION,1)) | 66 | (r=curl_easy_setopt(c,CURLOPT_FOLLOWLOCATION,1)) |
82 | || (r=curl_easy_setopt(c,CURLOPT_MAXREDIRS,5)) | 67 | || (r=curl_easy_setopt(c,CURLOPT_MAXREDIRS,5)) |
83 | || (r=curl_easy_setopt(c,CURLOPT_DNS_CACHE_TIMEOUT,120)) | 68 | || (r=curl_easy_setopt(c,CURLOPT_DNS_CACHE_TIMEOUT,120)) |
84 | || (r=curl_easy_setopt(c,CURLOPT_DNS_USE_GLOBAL_CACHE,1)) | 69 | || (r=curl_easy_setopt(c,CURLOPT_DNS_USE_GLOBAL_CACHE,1)) |
85 | || (r=curl_easy_setopt(c,CURLOPT_USERAGENT,PACKAGE_NAME"/"PACKAGE_SRC_VERSION)) | 70 | || (r=curl_easy_setopt(c,CURLOPT_USERAGENT,PACKAGE_NAME"/"PACKAGE_SRC_VERSION)) |
86 | || (r=curl_easy_setopt(c,CURLOPT_TIMEOUT,20)) | 71 | || (r=curl_easy_setopt(c,CURLOPT_TIMEOUT,20)) |
@@ -111,17 +96,17 @@ namespace opkele { | |||
111 | if(!DH_generate_key(dh)) | 96 | if(!DH_generate_key(dh)) |
112 | throw exception_openssl(OPKELE_CP_ "failed to DH_generate_key()"); | 97 | throw exception_openssl(OPKELE_CP_ "failed to DH_generate_key()"); |
113 | string request = | 98 | string request = |
114 | "openid.mode=associate" | 99 | "openid.mode=associate" |
115 | "&openid.assoc_type=HMAC-SHA1" | 100 | "&openid.assoc_type=HMAC-SHA1" |
116 | "&openid.session_type=DH-SHA1" | 101 | "&openid.session_type=DH-SHA1" |
117 | "&openid.dh_consumer_public="; | 102 | "&openid.dh_consumer_public="; |
118 | request += util::url_encode(util::bignum_to_base64(dh->pub_key)); | 103 | request += util::url_encode(util::bignum_to_base64(dh->pub_key)); |
119 | curl_t curl = curl_easy_init(); | 104 | util::curl_t curl = curl_easy_init(); |
120 | if(!curl) | 105 | if(!curl) |
121 | throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); | 106 | throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); |
122 | string response; | 107 | string response; |
123 | CURLcode r; | 108 | CURLcode r; |
124 | (r=curl_misc_sets(curl)) | 109 | (r=curl_misc_sets(curl)) |
125 | || (r=curl_easy_setopt(curl,CURLOPT_URL,server.c_str())) | 110 | || (r=curl_easy_setopt(curl,CURLOPT_URL,server.c_str())) |
126 | || (r=curl_easy_setopt(curl,CURLOPT_POST,1)) | 111 | || (r=curl_easy_setopt(curl,CURLOPT_POST,1)) |
127 | || (r=curl_easy_setopt(curl,CURLOPT_POSTFIELDS,request.data())) | 112 | || (r=curl_easy_setopt(curl,CURLOPT_POSTFIELDS,request.data())) |
@@ -271,17 +256,17 @@ namespace opkele { | |||
271 | for(params_t::const_iterator i=p.begin();i!=p.end();++i) { | 256 | for(params_t::const_iterator i=p.begin();i!=p.end();++i) { |
272 | if(i->first!="openid.mode") { | 257 | if(i->first!="openid.mode") { |
273 | request += '&'; | 258 | request += '&'; |
274 | request += i->first; | 259 | request += i->first; |
275 | request += '='; | 260 | request += '='; |
276 | request += util::url_encode(i->second); | 261 | request += util::url_encode(i->second); |
277 | } | 262 | } |
278 | } | 263 | } |
279 | curl_t curl = curl_easy_init(); | 264 | util::curl_t curl = curl_easy_init(); |
280 | if(!curl) | 265 | if(!curl) |
281 | throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); | 266 | throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); |
282 | string response; | 267 | string response; |
283 | CURLcode r; | 268 | CURLcode r; |
284 | (r=curl_misc_sets(curl)) | 269 | (r=curl_misc_sets(curl)) |
285 | || (r=curl_easy_setopt(curl,CURLOPT_URL,server.c_str())) | 270 | || (r=curl_easy_setopt(curl,CURLOPT_URL,server.c_str())) |
286 | || (r=curl_easy_setopt(curl,CURLOPT_POST,1)) | 271 | || (r=curl_easy_setopt(curl,CURLOPT_POST,1)) |
287 | || (r=curl_easy_setopt(curl,CURLOPT_POSTFIELDS,request.data())) | 272 | || (r=curl_easy_setopt(curl,CURLOPT_POSTFIELDS,request.data())) |
@@ -304,17 +289,17 @@ namespace opkele { | |||
304 | return; | 289 | return; |
305 | } | 290 | } |
306 | throw failed_check_authentication(OPKELE_CP_ "failed to verify response"); | 291 | throw failed_check_authentication(OPKELE_CP_ "failed to verify response"); |
307 | } | 292 | } |
308 | 293 | ||
309 | void consumer_t::retrieve_links(const string& url,string& server,string& delegate) { | 294 | void consumer_t::retrieve_links(const string& url,string& server,string& delegate) { |
310 | server.erase(); | 295 | server.erase(); |
311 | delegate.erase(); | 296 | delegate.erase(); |
312 | curl_t curl = curl_easy_init(); | 297 | util::curl_t curl = curl_easy_init(); |
313 | if(!curl) | 298 | if(!curl) |
314 | throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); | 299 | throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); |
315 | string html; | 300 | string html; |
316 | CURLcode r; | 301 | CURLcode r; |
317 | (r=curl_misc_sets(curl)) | 302 | (r=curl_misc_sets(curl)) |
318 | || (r=curl_easy_setopt(curl,CURLOPT_URL,url.c_str())) | 303 | || (r=curl_easy_setopt(curl,CURLOPT_URL,url.c_str())) |
319 | || (r=curl_easy_setopt(curl,CURLOPT_WRITEFUNCTION,_curl_tostring)) | 304 | || (r=curl_easy_setopt(curl,CURLOPT_WRITEFUNCTION,_curl_tostring)) |
320 | || (r=curl_easy_setopt(curl,CURLOPT_WRITEDATA,&html)) | 305 | || (r=curl_easy_setopt(curl,CURLOPT_WRITEDATA,&html)) |
@@ -410,17 +395,17 @@ namespace opkele { | |||
410 | if(sl==string::npos) | 395 | if(sl==string::npos) |
411 | rv += '/'; | 396 | rv += '/'; |
412 | } | 397 | } |
413 | return rv; | 398 | return rv; |
414 | } | 399 | } |
415 | 400 | ||
416 | string consumer_t::canonicalize(const string& url) { | 401 | string consumer_t::canonicalize(const string& url) { |
417 | string rv = normalize(url); | 402 | string rv = normalize(url); |
418 | curl_t curl = curl_easy_init(); | 403 | util::curl_t curl = curl_easy_init(); |
419 | if(!curl) | 404 | if(!curl) |
420 | throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); | 405 | throw exception_curl(OPKELE_CP_ "failed to curl_easy_init()"); |
421 | string html; | 406 | string html; |
422 | CURLcode r; | 407 | CURLcode r; |
423 | (r=curl_misc_sets(curl)) | 408 | (r=curl_misc_sets(curl)) |
424 | || (r=curl_easy_setopt(curl,CURLOPT_URL,rv.c_str())) | 409 | || (r=curl_easy_setopt(curl,CURLOPT_URL,rv.c_str())) |
425 | || (r=curl_easy_setopt(curl,CURLOPT_NOBODY,1)) | 410 | || (r=curl_easy_setopt(curl,CURLOPT_NOBODY,1)) |
426 | ; | 411 | ; |