source: trunk/libtransmission/web.c @ 6052

Last change on this file since 6052 was 6052, checked in by charles, 13 years ago

maybe fix the tracker authentication problems reported by demio

  • Property svn:keywords set to Date Rev Author Id
File size: 8.6 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 6052 2008-06-05 16:35:33Z charles $
11 */
12
13#include <assert.h>
14#include <stdlib.h> /* bsearch */
15
16#include <event.h>
17#include <curl/curl.h>
18
19#include "transmission.h"
20#include "trevent.h"
21#include "utils.h"
22#include "web.h"
23
24#define CURL_CHECK_VERSION(major,minor,micro)    \
25    (LIBCURL_VERSION_MAJOR > (major) || \
26     (LIBCURL_VERSION_MAJOR == (major) && LIBCURL_VERSION_MINOR > (minor)) || \
27     (LIBCURL_VERSION_MAJOR == (major) && LIBCURL_VERSION_MINOR == (minor) && \
28      LIBCURL_VERSION_PATCH >= (micro)))
29
30#define PULSE_MSEC 500
31
32#define dbgmsg(fmt...) tr_deepLog( __FILE__, __LINE__, "web", ##fmt )
33
34struct tr_web
35{
36    unsigned int dying     : 1;
37    unsigned int running   : 1;
38    int remain;
39    CURLM * cm;
40    tr_session * session;
41    struct event timer;
42};
43
44struct tr_web_task
45{
46    unsigned long tag;
47    struct evbuffer * response;
48    char * url;
49    tr_session * session;
50    tr_web_done_func * done_func;
51    void * done_func_user_data;
52};
53
54static void
55processCompletedTasks( tr_web * web )
56{
57    CURL * easy;
58    CURLMsg * msg;
59    CURLcode res;
60
61    do {
62        /* this convoluted loop is from the "hiperinfo.c" sample which
63         * hints that removing an easy handle in curl_multi_info_read's
64         * loop may be unsafe */
65        int more;
66        easy = NULL;
67        while(( msg = curl_multi_info_read( web->cm, &more ))) {
68            if( msg->msg == CURLMSG_DONE ) {
69                easy = msg->easy_handle;
70                res = msg->data.result;
71                break;
72            }
73        }
74        if( easy ) {
75            struct tr_web_task * task;
76            long response_code;
77            curl_easy_getinfo( easy, CURLINFO_PRIVATE, &task );
78            curl_easy_getinfo( easy, CURLINFO_RESPONSE_CODE, &response_code );
79            --web->remain;
80            dbgmsg( "task #%lu done (%d remain)", task->tag, web->remain );
81            task->done_func( web->session,
82                             response_code,
83                             EVBUFFER_DATA(task->response),
84                             EVBUFFER_LENGTH(task->response),
85                             task->done_func_user_data );
86            curl_multi_remove_handle( web->cm, easy );
87            curl_easy_cleanup( easy );
88            evbuffer_free( task->response );
89            tr_free( task->url );
90            tr_free( task );
91        }
92    } while( easy );
93}
94
95static void
96pump( tr_web * web )
97{
98    int unused;
99    CURLMcode rc;
100    do {
101        rc = curl_multi_perform( web->cm, &unused );
102    } while( rc == CURLM_CALL_MULTI_PERFORM );
103    if ( rc == CURLM_OK  )
104        processCompletedTasks( web );
105    else
106        tr_err( "%s", curl_multi_strerror(rc) );
107}
108
109static size_t
110writeFunc( void * ptr, size_t size, size_t nmemb, void * task )
111{
112    const size_t byteCount = size * nmemb;
113    evbuffer_add( ((struct tr_web_task*)task)->response, ptr, byteCount );
114    return byteCount;
115}
116
117static void
118ensureTimerIsRunning( tr_web * web )
119{
120    if( !web->running )
121    {
122        struct timeval tv = tr_timevalMsec( PULSE_MSEC );
123        dbgmsg( "starting web timer" );
124        web->running = 1;
125        evtimer_add( &web->timer, &tv );
126    }
127}
128
129static void
130addTask( void * vtask )
131{
132    struct tr_web_task * task = vtask;
133
134    if( task->session && task->session->web )
135    {
136        struct tr_web * web = task->session->web;
137        CURL * ch;
138
139        ensureTimerIsRunning( web );
140
141        ++web->remain;
142        dbgmsg( "adding task #%lu [%s] (%d remain)", task->tag, task->url, web->remain );
143
144        ch = curl_easy_init( );
145        curl_easy_setopt( ch, CURLOPT_PRIVATE, task );
146        curl_easy_setopt( ch, CURLOPT_URL, task->url );
147        curl_easy_setopt( ch, CURLOPT_WRITEFUNCTION, writeFunc );
148        curl_easy_setopt( ch, CURLOPT_WRITEDATA, task );
149        curl_easy_setopt( ch, CURLOPT_USERAGENT, TR_NAME "/" LONG_VERSION_STRING );
150        curl_easy_setopt( ch, CURLOPT_SSL_VERIFYHOST, 0 );
151        curl_easy_setopt( ch, CURLOPT_SSL_VERIFYPEER, 0 );
152        curl_easy_setopt( ch, CURLOPT_FORBID_REUSE, 1 );
153        curl_easy_setopt( ch, CURLOPT_NOSIGNAL, 1 );
154        curl_easy_setopt( ch, CURLOPT_FOLLOWLOCATION, 1 );
155        curl_easy_setopt( ch, CURLOPT_MAXREDIRS, 5 );
156        curl_easy_setopt( ch, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4 );
157        curl_easy_setopt( ch, CURLOPT_ENCODING, "" );
158        curl_multi_add_handle( web->cm, ch );
159    }
160}
161
162void
163tr_webRun( tr_session         * session,
164           const char         * url,
165           tr_web_done_func   * done_func,
166           void               * done_func_user_data )
167{
168    if( session->web )
169    {
170        static unsigned long tag = 0;
171        struct tr_web_task * task;
172
173        task = tr_new0( struct tr_web_task, 1 );
174        task->session = session;
175        task->url = tr_strdup( url );
176        task->done_func = done_func;
177        task->done_func_user_data = done_func_user_data;
178        task->tag = ++tag;
179        task->response = evbuffer_new( );
180
181        tr_runInEventThread( session, addTask, task );
182    }
183}
184
185static void
186webDestroy( tr_web * web )
187{
188    dbgmsg( "deleting web timer" );
189    assert( !web->running );
190    evtimer_del( &web->timer );
191    curl_multi_cleanup( web->cm );
192    tr_free( web );
193}
194
195static void
196pulse( int socket UNUSED, short action UNUSED, void * vweb )
197{
198    tr_web * web = vweb;
199    assert( web->running );
200
201    pump( web );
202
203    evtimer_del( &web->timer );
204
205    web->running = web->remain > 0;
206
207    if( web->running ) {
208        struct timeval tv = tr_timevalMsec( PULSE_MSEC );
209        evtimer_add( &web->timer, &tv );
210    } else if( web->dying ) {
211        webDestroy( web );
212    } else {
213        dbgmsg( "stopping web timer" );
214    }
215}
216
217tr_web*
218tr_webInit( tr_session * session )
219{
220    static int curlInited = FALSE;
221    tr_web * web;
222
223    /* call curl_global_init if we haven't done it already.
224     * try to enable ssl for https support; but if that fails,
225     * try a plain vanilla init */ 
226    if( curlInited == FALSE ) {
227        curlInited = TRUE;
228        if( curl_global_init( CURL_GLOBAL_SSL ) )
229            curl_global_init( 0 );
230    }
231   
232    web = tr_new0( struct tr_web, 1 );
233    web->cm = curl_multi_init( );
234    web->session = session;
235
236    evtimer_set( &web->timer, pulse, web );
237#if CURL_CHECK_VERSION(7,16,3)
238    curl_multi_setopt( web->cm, CURLMOPT_MAXCONNECTS, 10 );
239#endif
240    pump( web );
241
242    return web;
243}
244
245void
246tr_webClose( tr_web ** web_in )
247{
248    tr_web * web = *web_in;
249    *web_in = NULL;
250
251    if( !web->running )
252        webDestroy( web );
253    else
254        web->dying = 1;
255}
256
257/***
258****
259***/
260
261static struct http_msg {
262    long code;
263    const char * text;
264} http_msg[] = {
265    { 101, "Switching Protocols" },
266    { 200, "OK" },
267    { 201, "Created" },
268    { 202, "Accepted" },
269    { 203, "Non-Authoritative Information" },
270    { 204, "No Content" },
271    { 205, "Reset Content" },
272    { 206, "Partial Content" },
273    { 300, "Multiple Choices" },
274    { 301, "Moved Permanently" },
275    { 302, "Found" },
276    { 303, "See Other" },
277    { 304, "Not Modified" },
278    { 305, "Use Proxy" },
279    { 306, "(Unused)" },
280    { 307, "Temporary Redirect" },
281    { 400, "Bad Request" },
282    { 401, "Unauthorized" },
283    { 402, "Payment Required" },
284    { 403, "Forbidden" },
285    { 404, "Not Found" },
286    { 405, "Method Not Allowed" },
287    { 406, "Not Acceptable" },
288    { 407, "Proxy Authentication Required" },
289    { 408, "Request Timeout" },
290    { 409, "Conflict" },
291    { 410, "Gone" },
292    { 411, "Length Required" },
293    { 412, "Precondition Failed" },
294    { 413, "Request Entity Too Large" },
295    { 414, "Request-URI Too Long" },
296    { 415, "Unsupported Media Type" },
297    { 416, "Requested Range Not Satisfiable" },
298    { 417, "Expectation Failed" },
299    { 500, "Internal Server Error" },
300    { 501, "Not Implemented" },
301    { 502, "Bad Gateway" },
302    { 503, "Service Unavailable" },
303    { 504, "Gateway Timeout" },
304    { 505, "HTTP Version Not Supported" },
305    { 0, NULL }
306};
307
308static int
309compareResponseCodes( const void * va, const void * vb )
310{
311    const long a = *(const long*) va;
312    const struct http_msg * b = vb;
313    return a - b->code;
314}
315
316const char *
317tr_webGetResponseStr( long code )
318{
319    struct http_msg * msg = bsearch( &code,
320                                     http_msg, 
321                                     sizeof( http_msg ) / sizeof( http_msg[0] ),
322                                     sizeof( http_msg[0] ),
323                                     compareResponseCodes );
324    return msg ? msg->text : "Unknown Error";
325}
Note: See TracBrowser for help on using the repository browser.