source: trunk/libtransmission/web.c @ 5705

Last change on this file since 5705 was 5705, checked in by charles, 14 years ago

steal some curl configuration options from rtorrent :)

  • Property svn:keywords set to Date Rev Author Id
File size: 10.7 KB
Line 
1/*
2 * This file Copyright (C) 2008 Charles Kerr <charles@rebelbase.com>
3 *
4 * This file is licensed by the GPL version 2.  Works owned by the
5 * Transmission project are granted a special exemption to clause 2(b)
6 * so that the bulk of its code can remain under the MIT license.
7 * This exemption does not extend to derived works not owned by
8 * the Transmission project.
9 *
10 * $Id: web.c 5705 2008-04-25 23:54:02Z charles $
11 */
12
13#include <stdlib.h> /* bsearch */
14
15#include <event.h>
16#include <curl/curl.h>
17
18#include "transmission.h"
19#include "trevent.h"
20#include "utils.h"
21#include "web.h"
22
23#define CURL_CHECK_VERSION(major,minor,micro)    \
24    (LIBCURL_VERSION_MAJOR > (major) || \
25     (LIBCURL_VERSION_MAJOR == (major) && LIBCURL_VERSION_MINOR > (minor)) || \
26     (LIBCURL_VERSION_MAJOR == (major) && LIBCURL_VERSION_MINOR == (minor) && \
27      LIBCURL_VERSION_PATCH >= (micro)))
28
29#if CURL_CHECK_VERSION(7,16,0)
30#define USE_CURL_MULTI_SOCKET
31#else
32#define PULSE_MSEC 200
33#endif
34
35#define dbgmsg(fmt...) tr_deepLog( __FILE__, __LINE__, "web", ##fmt )
36
37struct tr_web
38{
39    CURLM * cm;
40    tr_session * session;
41    int remain;
42    struct event timer;
43};
44
45struct tr_web_task
46{
47    unsigned long tag;
48    struct evbuffer * response;
49    char * url;
50    tr_session * session;
51    tr_web_done_func * done_func;
52    void * done_func_user_data;
53};
54
55static void
56processCompletedTasks( tr_web * web )
57{
58    CURL * easy;
59    CURLMsg * msg;
60    CURLcode res;
61
62    do {
63        /* this convoluted loop is from the "hiperinfo.c" sample which
64         * hints that removing an easy handle in curl_multi_info_read's
65         * loop may be unsafe */
66        int more;
67        easy = NULL;
68        while(( msg = curl_multi_info_read( web->cm, &more ))) {
69            if( msg->msg == CURLMSG_DONE ) {
70                easy = msg->easy_handle;
71                res = msg->data.result;
72                break;
73            }
74        }
75        if( easy ) {
76            struct tr_web_task * task;
77            int response_code;
78            curl_easy_getinfo( easy, CURLINFO_PRIVATE, &task );
79            curl_easy_getinfo( easy, CURLINFO_RESPONSE_CODE, &response_code );
80            --web->remain;
81            dbgmsg( "task #%lu done (%d remain)", task->tag, web->remain );
82            task->done_func( web->session,
83                             response_code,
84                             EVBUFFER_DATA(task->response),
85                             EVBUFFER_LENGTH(task->response),
86                             task->done_func_user_data );
87
88            curl_multi_remove_handle( web->cm, easy );
89            curl_easy_cleanup( easy );
90            evbuffer_free( task->response );
91            tr_free( task->url );
92            tr_free( task );
93        }
94    } while( easy );
95}
96
97static void
98pump( tr_web * web )
99{
100    int unused;
101    CURLMcode rc;
102    do {
103#ifdef USE_CURL_MULTI_SOCKET
104        rc = curl_multi_socket_all( web->cm, &unused );
105#else
106        rc = curl_multi_perform( web->cm, &unused );
107#endif
108    } while( rc == CURLM_CALL_MULTI_PERFORM );
109    if ( rc == CURLM_OK  )
110        processCompletedTasks( web );
111    else
112        tr_err( "%s", curl_multi_strerror(rc) );
113}
114
115static size_t
116writeFunc( void * ptr, size_t size, size_t nmemb, void * task )
117{
118    const size_t byteCount = size * nmemb;
119    evbuffer_add( ((struct tr_web_task*)task)->response, ptr, byteCount );
120    return byteCount;
121}
122
123static void
124addTask( void * vtask )
125{
126    struct tr_web_task * task = vtask;
127    struct tr_web * web = task->session->web;
128    CURL * ch;
129
130    dbgmsg( "adding task #%lu [%s]", task->tag, task->url );
131    ++web->remain;
132
133    ch = curl_easy_init( );
134    curl_easy_setopt( ch, CURLOPT_PRIVATE, task );
135    curl_easy_setopt( ch, CURLOPT_URL, task->url );
136    curl_easy_setopt( ch, CURLOPT_WRITEFUNCTION, writeFunc );
137    curl_easy_setopt( ch, CURLOPT_WRITEDATA, task );
138    curl_easy_setopt( ch, CURLOPT_USERAGENT, TR_NAME "/" LONG_VERSION_STRING );
139    curl_easy_setopt( ch, CURLOPT_SSL_VERIFYPEER, 0 );
140    curl_easy_setopt( ch, CURLOPT_FORBID_REUSE, 1 );
141    curl_easy_setopt( ch, CURLOPT_NOSIGNAL, 1 );
142    curl_easy_setopt( ch, CURLOPT_FOLLOWLOCATION, 1 );
143    curl_easy_setopt( ch, CURLOPT_MAXREDIRS, 5 );
144    curl_easy_setopt( ch, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4 );
145    curl_easy_setopt( ch, CURLOPT_ENCODING, "" );
146    curl_multi_add_handle( web->cm, ch );
147
148    pump( web );
149}
150
151void
152tr_webRun( tr_session         * session,
153           const char         * url,
154           tr_web_done_func   * done_func,
155           void               * done_func_user_data )
156{
157    static unsigned long tag = 0;
158    struct tr_web_task * task;
159
160    task = tr_new0( struct tr_web_task, 1 );
161    task->session = session;
162    task->url = tr_strdup( url );
163    task->done_func = done_func;
164    task->done_func_user_data = done_func_user_data;
165    task->tag = ++tag;
166    task->response = evbuffer_new( );
167
168    tr_runInEventThread( session, addTask, task );
169}
170
171#ifdef USE_CURL_MULTI_SOCKET
172
173/* libevent says that sock is ready to be processed, so tell libcurl */
174static void
175ev_sock_cb( int sock, short action, void * vweb )
176{
177    tr_web * web = vweb;
178    CURLMcode rc;
179    int mask, unused;
180
181    switch (action & (EV_READ|EV_WRITE)) {
182        case EV_READ: mask = CURL_CSELECT_IN; break;
183        case EV_WRITE: mask = CURL_CSELECT_OUT; break;
184        case EV_READ|EV_WRITE: mask = CURL_CSELECT_IN|CURL_CSELECT_OUT; break;
185        default: tr_err( "Unknown event %hd\n", action ); return;
186    }
187
188    do {
189        rc = curl_multi_socket_action( web->cm, sock, mask, &unused );
190    } while( rc == CURLM_CALL_MULTI_PERFORM );
191    if ( rc == CURLM_OK  )
192        processCompletedTasks( web );
193    else
194        tr_err( "%s (%d)", curl_multi_strerror(rc), (int)sock );
195
196}
197
198/* CURLMPOPT_SOCKETFUNCTION */
199/* libcurl wants us to tell it when sock is ready to be processed */
200static void
201multi_sock_cb( CURL            * easy UNUSED,
202               curl_socket_t     sock,
203               int               action,
204               void            * vweb,
205               void            * assigndata )
206{
207    tr_web * web = vweb;
208    struct event * ev = assigndata;
209
210    if( action == CURL_POLL_REMOVE ) {
211        if( ev ) {
212            event_del( ev );
213            tr_free( ev );
214            curl_multi_assign( web->cm, sock, NULL );
215        }
216    } else {
217        int kind;
218        if( ev ) {
219            event_del( ev );
220        } else {
221            ev = tr_new0( struct event, 1 );
222            curl_multi_assign( web->cm, sock, ev );
223        }
224        kind = EV_PERSIST;
225        if( action & CURL_POLL_IN ) kind |= EV_READ;
226        if( action & CURL_POLL_OUT ) kind |= EV_WRITE;
227        event_set( ev, sock, kind, ev_sock_cb, web );
228        event_add( ev, NULL );
229    }
230}
231
232/* libevent says that timeout_ms have passed, so tell libcurl */
233static void
234event_timer_cb( int socket UNUSED, short action UNUSED, void * vweb )
235{
236    int unused;
237    CURLMcode rc;
238    tr_web * web = vweb;
239
240    do {
241        rc = curl_multi_socket( web->cm, CURL_SOCKET_TIMEOUT, &unused );
242    } while( rc == CURLM_CALL_MULTI_PERFORM );
243    if ( rc == CURLM_OK  )
244        processCompletedTasks( web );
245    else
246        tr_err( "%s", curl_multi_strerror(rc) );
247}
248
249/* CURLMPOPT_TIMERFUNCTION */
250static void
251multi_timer_cb( CURLM *multi UNUSED, long timeout_ms, void * vweb )
252{
253    tr_web * web = vweb;
254    struct timeval tv = tr_timevalMsec( timeout_ms );
255    evtimer_add( &web->timer, &tv );
256}
257
258#else
259
260static void
261pulse( int socket UNUSED, short action UNUSED, void * vweb )
262{
263    tr_web * web = vweb;
264    struct timeval tv = tr_timevalMsec( PULSE_MSEC );
265
266    pump( web );
267
268    evtimer_del( &web->timer );
269    evtimer_add( &web->timer, &tv );
270}
271
272#endif
273
274tr_web*
275tr_webInit( tr_session * session )
276{
277#ifndef USE_CURL_MULTI_SOCKET
278    struct timeval tv = tr_timevalMsec( PULSE_MSEC );
279#endif
280    static int curlInited = FALSE;
281    tr_web * web;
282
283    /* call curl_global_init if we haven't done it already.
284     * try to enable ssl for https support; but if that fails,
285     * try a plain vanilla init */ 
286    if( curlInited == FALSE ) {
287        curlInited = TRUE;
288        if( curl_global_init( CURL_GLOBAL_SSL ) )
289            curl_global_init( 0 );
290    }
291   
292    web = tr_new0( struct tr_web, 1 );
293    web->cm = curl_multi_init( );
294    web->session = session;
295
296#ifdef USE_CURL_MULTI_SOCKET
297    evtimer_set( &web->timer, event_timer_cb, web );
298    curl_multi_setopt( web->cm, CURLMOPT_SOCKETDATA, web );
299    curl_multi_setopt( web->cm, CURLMOPT_SOCKETFUNCTION, multi_sock_cb );
300    curl_multi_setopt( web->cm, CURLMOPT_TIMERDATA, web );
301    curl_multi_setopt( web->cm, CURLMOPT_TIMERFUNCTION, multi_timer_cb );
302#else
303    evtimer_set( &web->timer, pulse, web );
304    evtimer_add( &web->timer, &tv );
305#endif
306#if CURL_CHECK_VERSION(7,16,3)
307    curl_multi_setopt( web->cm, CURLMOPT_MAXCONNECTS, 10 );
308#endif
309#if CURL_CHECK_VERSION(7,16,0)
310    curl_multi_setopt( web->cm, CURLMOPT_PIPELINING, 1 );
311#endif
312    pump( web );
313
314    return web;
315}
316
317void
318tr_webClose( tr_web * web )
319{
320    evtimer_del( &web->timer );
321    curl_multi_cleanup( web->cm );
322    tr_free( web );
323}
324
325/***
326****
327***/
328
329static struct http_msg {
330    long code;
331    const char * text;
332} http_msg[] = {
333    { 101, "Switching Protocols" },
334    { 200, "OK" },
335    { 201, "Created" },
336    { 202, "Accepted" },
337    { 203, "Non-Authoritative Information" },
338    { 204, "No Content" },
339    { 205, "Reset Content" },
340    { 206, "Partial Content" },
341    { 300, "Multiple Choices" },
342    { 301, "Moved Permanently" },
343    { 302, "Found" },
344    { 303, "See Other" },
345    { 304, "Not Modified" },
346    { 305, "Use Proxy" },
347    { 306, "(Unused)" },
348    { 307, "Temporary Redirect" },
349    { 400, "Bad Request" },
350    { 401, "Unauthorized" },
351    { 402, "Payment Required" },
352    { 403, "Forbidden" },
353    { 404, "Not Found" },
354    { 405, "Method Not Allowed" },
355    { 406, "Not Acceptable" },
356    { 407, "Proxy Authentication Required" },
357    { 408, "Request Timeout" },
358    { 409, "Conflict" },
359    { 410, "Gone" },
360    { 411, "Length Required" },
361    { 412, "Precondition Failed" },
362    { 413, "Request Entity Too Large" },
363    { 414, "Request-URI Too Long" },
364    { 415, "Unsupported Media Type" },
365    { 416, "Requested Range Not Satisfiable" },
366    { 417, "Expectation Failed" },
367    { 500, "Internal Server Error" },
368    { 501, "Not Implemented" },
369    { 502, "Bad Gateway" },
370    { 503, "Service Unavailable" },
371    { 504, "Gateway Timeout" },
372    { 505, "HTTP Version Not Supported" },
373    { 0, NULL }
374};
375
376static int
377compareResponseCodes( const void * va, const void * vb )
378{
379    const long a = *(const long*) va;
380    const struct http_msg * b = vb;
381    return a - b->code;
382}
383
384const char *
385tr_webGetResponseStr( long code )
386{
387    struct http_msg * msg = bsearch( &code,
388                                     http_msg, 
389                                     sizeof( http_msg ) / sizeof( http_msg[0] ),
390                                     sizeof( http_msg[0] ),
391                                     compareResponseCodes );
392    return msg ? msg->text : "Unknown Error";
393}
Note: See TracBrowser for help on using the repository browser.