source: trunk/libtransmission/rpcimpl.c @ 12918

Last change on this file since 12918 was 12772, checked in by livings124, 10 years ago

add an RPC session callback for when queue order changes

  • Property svn:keywords set to Date Rev Author Id
File size: 70.6 KB
Line 
1/*
2 * This file Copyright (C) Mnemosyne LLC
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: rpcimpl.c 12772 2011-08-27 23:54:10Z jordan $
11 */
12
13#include <assert.h>
14#include <ctype.h> /* isdigit */
15#include <errno.h>
16#include <stdlib.h> /* strtol */
17#include <string.h> /* strcmp */
18#include <unistd.h> /* unlink */
19
20#ifdef HAVE_ZLIB
21 #include <zlib.h>
22#endif
23
24#include <event2/buffer.h>
25
26#include "transmission.h"
27#include "bencode.h"
28#include "completion.h"
29#include "fdlimit.h"
30#include "json.h"
31#include "rpcimpl.h"
32#include "session.h"
33#include "torrent.h"
34#include "utils.h"
35#include "version.h"
36#include "web.h"
37
38#define RPC_VERSION     14
39#define RPC_VERSION_MIN 1
40
41#define RECENTLY_ACTIVE_SECONDS 60
42
43#define TR_N_ELEMENTS( ary ) ( sizeof( ary ) / sizeof( *ary ) )
44
45#if 0
46#define dbgmsg(fmt, ...) \
47    do { \
48        fprintf( stderr, "%s:%d"#fmt, __FILE__, __LINE__, __VA_ARGS__ ); \
49        fprintf( stderr, "\n" ); \
50    } while( 0 )
51#else
52#define dbgmsg( ... ) \
53    do { \
54        if( tr_deepLoggingIsActive( ) ) \
55            tr_deepLog( __FILE__, __LINE__, "RPC", __VA_ARGS__ ); \
56    } while( 0 )
57#endif
58
59
60/***
61****
62***/
63
64static tr_rpc_callback_status
65notify( tr_session * session,
66        int          type,
67        tr_torrent * tor )
68{
69    tr_rpc_callback_status status = 0;
70
71    if( session->rpc_func )
72        status = session->rpc_func( session, type, tor,
73                                    session->rpc_func_user_data );
74
75    return status;
76}
77
78/***
79****
80***/
81
82/* For functions that can't be immediately executed, like torrentAdd,
83 * this is the callback data used to pass a response to the caller
84 * when the task is complete */
85struct tr_rpc_idle_data
86{
87    tr_session            * session;
88    tr_benc               * response;
89    tr_benc               * args_out;
90    tr_rpc_response_func    callback;
91    void                  * callback_user_data;
92};
93
94static void
95tr_idle_function_done( struct tr_rpc_idle_data * data, const char * result )
96{
97    struct evbuffer * buf;
98
99    if( result == NULL )
100        result = "success";
101    tr_bencDictAddStr( data->response, "result", result );
102
103    buf = tr_bencToBuf( data->response, TR_FMT_JSON_LEAN );
104    (*data->callback)( data->session, buf, data->callback_user_data );
105    evbuffer_free( buf );
106
107    tr_bencFree( data->response );
108    tr_free( data->response );
109    tr_free( data );
110}
111
112/***
113****
114***/
115
116static tr_torrent **
117getTorrents( tr_session * session,
118             tr_benc    * args,
119             int        * setmeCount )
120{
121    int           torrentCount = 0;
122    int64_t       id;
123    tr_torrent ** torrents = NULL;
124    tr_benc *     ids;
125    const char * str;
126
127    if( tr_bencDictFindList( args, "ids", &ids ) )
128    {
129        int       i;
130        const int n = tr_bencListSize( ids );
131
132        torrents = tr_new0( tr_torrent *, n );
133
134        for( i = 0; i < n; ++i )
135        {
136            tr_torrent * tor = NULL;
137            tr_benc *    node = tr_bencListChild( ids, i );
138            const char * str;
139            if( tr_bencGetInt( node, &id ) )
140                tor = tr_torrentFindFromId( session, id );
141            else if( tr_bencGetStr( node, &str ) )
142                tor = tr_torrentFindFromHashString( session, str );
143            if( tor )
144                torrents[torrentCount++] = tor;
145        }
146    }
147    else if( tr_bencDictFindInt( args, "ids", &id )
148           || tr_bencDictFindInt( args, "id", &id ) )
149    {
150        tr_torrent * tor;
151        torrents = tr_new0( tr_torrent *, 1 );
152        if( ( tor = tr_torrentFindFromId( session, id ) ) )
153            torrents[torrentCount++] = tor;
154    }
155    else if( tr_bencDictFindStr( args, "ids", &str ) )
156    {
157        if( !strcmp( str, "recently-active" ) )
158        {
159            tr_torrent * tor = NULL;
160            const time_t now = tr_time( );
161            const time_t window = RECENTLY_ACTIVE_SECONDS;
162            const int n = tr_sessionCountTorrents( session );
163            torrents = tr_new0( tr_torrent *, n );
164            while( ( tor = tr_torrentNext( session, tor ) ) )
165                if( tor->anyDate >= now - window )
166                    torrents[torrentCount++] = tor;
167        }
168        else
169        {
170            tr_torrent * tor;
171            torrents = tr_new0( tr_torrent *, 1 );
172            if(( tor = tr_torrentFindFromHashString( session, str )))
173                torrents[torrentCount++] = tor;
174        }
175    }
176    else /* all of them */
177    {
178        tr_torrent * tor = NULL;
179        const int n = tr_sessionCountTorrents( session );
180        torrents = tr_new0( tr_torrent *, n );
181        while( ( tor = tr_torrentNext( session, tor ) ) )
182            torrents[torrentCount++] = tor;
183    }
184
185    *setmeCount = torrentCount;
186    return torrents;
187}
188
189static void
190notifyBatchQueueChange( tr_session * session, tr_torrent ** torrents, int n )
191{
192    int i;
193    for( i=0; i<n; ++i )
194        notify( session, TR_RPC_TORRENT_CHANGED, torrents[i] );
195    notify( session, TR_RPC_SESSION_QUEUE_POSITIONS_CHANGED, NULL );
196}
197
198static const char*
199queueMoveTop( tr_session               * session,
200              tr_benc                  * args_in,
201              tr_benc                  * args_out UNUSED,
202              struct tr_rpc_idle_data  * idle_data UNUSED )
203{
204    int n;
205    tr_torrent ** torrents = getTorrents( session, args_in, &n );
206    tr_torrentsQueueMoveTop( torrents, n );
207    notifyBatchQueueChange( session, torrents, n );
208    tr_free( torrents );
209    return NULL;
210}
211
212static const char*
213queueMoveUp( tr_session               * session,
214             tr_benc                  * args_in,
215             tr_benc                  * args_out UNUSED,
216             struct tr_rpc_idle_data  * idle_data UNUSED )
217{
218    int n;
219    tr_torrent ** torrents = getTorrents( session, args_in, &n );
220    tr_torrentsQueueMoveUp( torrents, n );
221    notifyBatchQueueChange( session, torrents, n );
222    tr_free( torrents );
223    return NULL;
224}
225
226static const char*
227queueMoveDown( tr_session               * session,
228               tr_benc                  * args_in,
229               tr_benc                  * args_out UNUSED,
230               struct tr_rpc_idle_data  * idle_data UNUSED )
231{
232    int n;
233    tr_torrent ** torrents = getTorrents( session, args_in, &n );
234    tr_torrentsQueueMoveDown( torrents, n );
235    notifyBatchQueueChange( session, torrents, n );
236    tr_free( torrents );
237    return NULL;
238}
239
240static const char*
241queueMoveBottom( tr_session               * session,
242                 tr_benc                  * args_in,
243                 tr_benc                  * args_out UNUSED,
244                 struct tr_rpc_idle_data  * idle_data UNUSED )
245{
246    int n;
247    tr_torrent ** torrents = getTorrents( session, args_in, &n );
248    tr_torrentsQueueMoveBottom( torrents, n );
249    notifyBatchQueueChange( session, torrents, n );
250    tr_free( torrents );
251    return NULL;
252}
253
254static const char*
255torrentStart( tr_session               * session,
256              tr_benc                  * args_in,
257              tr_benc                  * args_out UNUSED,
258              struct tr_rpc_idle_data  * idle_data UNUSED )
259{
260    int           i, torrentCount;
261    tr_torrent ** torrents = getTorrents( session, args_in, &torrentCount );
262
263    assert( idle_data == NULL );
264
265    for( i = 0; i < torrentCount; ++i )
266    {
267        tr_torrent * tor = torrents[i];
268        if( !tor->isRunning )
269        {
270            tr_torrentStart( tor );
271            notify( session, TR_RPC_TORRENT_STARTED, tor );
272        }
273    }
274    tr_free( torrents );
275    return NULL;
276}
277
278static const char*
279torrentStartNow( tr_session               * session,
280                 tr_benc                  * args_in,
281                 tr_benc                  * args_out UNUSED,
282                 struct tr_rpc_idle_data  * idle_data UNUSED )
283{
284    int           i, torrentCount;
285    tr_torrent ** torrents = getTorrents( session, args_in, &torrentCount );
286
287    assert( idle_data == NULL );
288
289    for( i = 0; i < torrentCount; ++i )
290    {
291        tr_torrent * tor = torrents[i];
292        if( !tor->isRunning )
293        {
294            tr_torrentStartNow( tor );
295            notify( session, TR_RPC_TORRENT_STARTED, tor );
296        }
297    }
298    tr_free( torrents );
299    return NULL;
300}
301
302static const char*
303torrentStop( tr_session               * session,
304             tr_benc                  * args_in,
305             tr_benc                  * args_out UNUSED,
306             struct tr_rpc_idle_data  * idle_data UNUSED )
307{
308    int           i, torrentCount;
309    tr_torrent ** torrents = getTorrents( session, args_in, &torrentCount );
310
311    assert( idle_data == NULL );
312
313    for( i = 0; i < torrentCount; ++i )
314    {
315        tr_torrent * tor = torrents[i];
316
317        if( tor->isRunning || tr_torrentIsQueued( tor ) )
318        {
319            tor->isStopping = true;
320            notify( session, TR_RPC_TORRENT_STOPPED, tor );
321        }
322    }
323    tr_free( torrents );
324    return NULL;
325}
326
327static const char*
328torrentRemove( tr_session               * session,
329               tr_benc                  * args_in,
330               tr_benc                  * args_out UNUSED,
331               struct tr_rpc_idle_data  * idle_data UNUSED )
332{
333    int i;
334    int torrentCount;
335    tr_rpc_callback_type type;
336    bool deleteFlag = false;
337    tr_torrent ** torrents = getTorrents( session, args_in, &torrentCount );
338
339    assert( idle_data == NULL );
340
341    tr_bencDictFindBool( args_in, "delete-local-data", &deleteFlag );
342    type = deleteFlag ? TR_RPC_TORRENT_TRASHING
343                      : TR_RPC_TORRENT_REMOVING;
344
345    for( i=0; i<torrentCount; ++i )
346    {
347        tr_torrent * tor = torrents[i];
348        const tr_rpc_callback_status status = notify( session, type, tor );
349        if( !( status & TR_RPC_NOREMOVE ) )
350            tr_torrentRemove( tor, deleteFlag, NULL );
351    }
352
353    tr_free( torrents );
354    return NULL;
355}
356
357static const char*
358torrentReannounce( tr_session               * session,
359                   tr_benc                  * args_in,
360                   tr_benc                  * args_out UNUSED,
361                   struct tr_rpc_idle_data  * idle_data UNUSED )
362{
363    int i, torrentCount;
364    tr_torrent ** torrents = getTorrents( session, args_in, &torrentCount );
365
366    assert( idle_data == NULL );
367
368    for( i=0; i<torrentCount; ++i )
369    {
370        tr_torrent * tor = torrents[i];
371        if( tr_torrentCanManualUpdate( tor ) )
372        {
373            tr_torrentManualUpdate( tor );
374            notify( session, TR_RPC_TORRENT_CHANGED, tor );
375        }
376    }
377
378    tr_free( torrents );
379    return NULL;
380}
381
382static const char*
383torrentVerify( tr_session               * session,
384               tr_benc                  * args_in,
385               tr_benc                  * args_out UNUSED,
386               struct tr_rpc_idle_data  * idle_data UNUSED )
387{
388    int           i, torrentCount;
389    tr_torrent ** torrents = getTorrents( session, args_in, &torrentCount );
390
391    assert( idle_data == NULL );
392
393    for( i = 0; i < torrentCount; ++i )
394    {
395        tr_torrent * tor = torrents[i];
396        tr_torrentVerify( tor );
397        notify( session, TR_RPC_TORRENT_CHANGED, tor );
398    }
399
400    tr_free( torrents );
401    return NULL;
402}
403
404/***
405****
406***/
407
408static void
409addFileStats( const tr_torrent * tor, tr_benc * list )
410{
411    tr_file_index_t i;
412    tr_file_index_t n;
413    const tr_info * info = tr_torrentInfo( tor );
414    tr_file_stat * files = tr_torrentFiles( tor, &n );
415
416    for( i = 0; i < info->fileCount; ++i )
417    {
418        const tr_file * file = &info->files[i];
419        tr_benc * d = tr_bencListAddDict( list, 3 );
420        tr_bencDictAddInt( d, "bytesCompleted", files[i].bytesCompleted );
421        tr_bencDictAddInt( d, "priority", file->priority );
422        tr_bencDictAddBool( d, "wanted", !file->dnd );
423    }
424
425    tr_torrentFilesFree( files, n );
426}
427
428static void
429addFiles( const tr_torrent * tor,
430          tr_benc *          list )
431{
432    tr_file_index_t i;
433    tr_file_index_t n;
434    const tr_info * info = tr_torrentInfo( tor );
435    tr_file_stat *  files = tr_torrentFiles( tor, &n );
436
437    for( i = 0; i < info->fileCount; ++i )
438    {
439        const tr_file * file = &info->files[i];
440        tr_benc *       d = tr_bencListAddDict( list, 3 );
441        tr_bencDictAddInt( d, "bytesCompleted", files[i].bytesCompleted );
442        tr_bencDictAddInt( d, "length", file->length );
443        tr_bencDictAddStr( d, "name", file->name );
444    }
445
446    tr_torrentFilesFree( files, n );
447}
448
449static void
450addWebseeds( const tr_info * info,
451             tr_benc *       webseeds )
452{
453    int i;
454
455    for( i = 0; i < info->webseedCount; ++i )
456        tr_bencListAddStr( webseeds, info->webseeds[i] );
457}
458
459static void
460addTrackers( const tr_info * info,
461             tr_benc *       trackers )
462{
463    int i;
464
465    for( i = 0; i < info->trackerCount; ++i )
466    {
467        const tr_tracker_info * t = &info->trackers[i];
468        tr_benc *               d = tr_bencListAddDict( trackers, 4 );
469        tr_bencDictAddStr( d, "announce", t->announce );
470        tr_bencDictAddInt( d, "id", t->id );
471        tr_bencDictAddStr( d, "scrape", t->scrape );
472        tr_bencDictAddInt( d, "tier", t->tier );
473    }
474}
475
476static void
477addTrackerStats( const tr_tracker_stat * st, int n, tr_benc * list )
478{
479    int i;
480
481    for( i=0; i<n; ++i )
482    {
483        const tr_tracker_stat * s = &st[i];
484        tr_benc * d = tr_bencListAddDict( list, 26 );
485        tr_bencDictAddStr ( d, "announce", s->announce );
486        tr_bencDictAddInt ( d, "announceState", s->announceState );
487        tr_bencDictAddInt ( d, "downloadCount", s->downloadCount );
488        tr_bencDictAddBool( d, "hasAnnounced", s->hasAnnounced );
489        tr_bencDictAddBool( d, "hasScraped", s->hasScraped );
490        tr_bencDictAddStr ( d, "host", s->host );
491        tr_bencDictAddInt ( d, "id", s->id );
492        tr_bencDictAddBool( d, "isBackup", s->isBackup );
493        tr_bencDictAddInt ( d, "lastAnnouncePeerCount", s->lastAnnouncePeerCount );
494        tr_bencDictAddStr ( d, "lastAnnounceResult", s->lastAnnounceResult );
495        tr_bencDictAddInt ( d, "lastAnnounceStartTime", s->lastAnnounceStartTime );
496        tr_bencDictAddBool( d, "lastAnnounceSucceeded", s->lastAnnounceSucceeded );
497        tr_bencDictAddInt ( d, "lastAnnounceTime", s->lastAnnounceTime );
498        tr_bencDictAddBool( d, "lastAnnounceTimedOut", s->lastAnnounceTimedOut );
499        tr_bencDictAddStr ( d, "lastScrapeResult", s->lastScrapeResult );
500        tr_bencDictAddInt ( d, "lastScrapeStartTime", s->lastScrapeStartTime );
501        tr_bencDictAddBool( d, "lastScrapeSucceeded", s->lastScrapeSucceeded );
502        tr_bencDictAddInt ( d, "lastScrapeTime", s->lastScrapeTime );
503        tr_bencDictAddInt ( d, "lastScrapeTimedOut", s->lastScrapeTimedOut );
504        tr_bencDictAddInt ( d, "leecherCount", s->leecherCount );
505        tr_bencDictAddInt ( d, "nextAnnounceTime", s->nextAnnounceTime );
506        tr_bencDictAddInt ( d, "nextScrapeTime", s->nextScrapeTime );
507        tr_bencDictAddStr ( d, "scrape", s->scrape );
508        tr_bencDictAddInt ( d, "scrapeState", s->scrapeState );
509        tr_bencDictAddInt ( d, "seederCount", s->seederCount );
510        tr_bencDictAddInt ( d, "tier", s->tier );
511    }
512}
513
514static void
515addPeers( const tr_torrent * tor,
516          tr_benc *          list )
517{
518    int            i;
519    int            peerCount;
520    tr_peer_stat * peers = tr_torrentPeers( tor, &peerCount );
521
522    tr_bencInitList( list, peerCount );
523
524    for( i = 0; i < peerCount; ++i )
525    {
526        tr_benc *            d = tr_bencListAddDict( list, 14 );
527        const tr_peer_stat * peer = peers + i;
528        tr_bencDictAddStr ( d, "address", peer->addr );
529        tr_bencDictAddStr ( d, "clientName", peer->client );
530        tr_bencDictAddBool( d, "clientIsChoked", peer->clientIsChoked );
531        tr_bencDictAddBool( d, "clientIsInterested", peer->clientIsInterested );
532        tr_bencDictAddStr ( d, "flagStr", peer->flagStr );
533        tr_bencDictAddBool( d, "isDownloadingFrom", peer->isDownloadingFrom );
534        tr_bencDictAddBool( d, "isEncrypted", peer->isEncrypted );
535        tr_bencDictAddBool( d, "isIncoming", peer->isIncoming );
536        tr_bencDictAddBool( d, "isUploadingTo", peer->isUploadingTo );
537        tr_bencDictAddBool( d, "isUTP", peer->isUTP );
538        tr_bencDictAddBool( d, "peerIsChoked", peer->peerIsChoked );
539        tr_bencDictAddBool( d, "peerIsInterested", peer->peerIsInterested );
540        tr_bencDictAddInt ( d, "port", peer->port );
541        tr_bencDictAddReal( d, "progress", peer->progress );
542        tr_bencDictAddInt ( d, "rateToClient", toSpeedBytes( peer->rateToClient_KBps ) );
543        tr_bencDictAddInt ( d, "rateToPeer", toSpeedBytes( peer->rateToPeer_KBps ) );
544    }
545
546    tr_torrentPeersFree( peers, peerCount );
547}
548
549/* faster-than-strcmp() optimization. This is kind of clumsy,
550   but addField() was in the profiler's top 10 list, and this
551   makes it 4x faster... */
552#define tr_streq(a,alen,b) ((alen+1==sizeof(b)) && !memcmp(a,b,alen))
553
554static void
555addField( const tr_torrent * const tor,
556          const tr_info    * const inf,
557          const tr_stat    * const st,
558          tr_benc          * const d,
559          const char       * const key )
560{
561    const size_t keylen = strlen( key );
562
563    if( tr_streq( key, keylen, "activityDate" ) )
564        tr_bencDictAddInt( d, key, st->activityDate );
565    else if( tr_streq( key, keylen, "addedDate" ) )
566        tr_bencDictAddInt( d, key, st->addedDate );
567    else if( tr_streq( key, keylen, "bandwidthPriority" ) )
568        tr_bencDictAddInt( d, key, tr_torrentGetPriority( tor ) );
569    else if( tr_streq( key, keylen, "comment" ) )
570        tr_bencDictAddStr( d, key, inf->comment ? inf->comment : "" );
571    else if( tr_streq( key, keylen, "corruptEver" ) )
572        tr_bencDictAddInt( d, key, st->corruptEver );
573    else if( tr_streq( key, keylen, "creator" ) )
574        tr_bencDictAddStr( d, key, inf->creator ? inf->creator : "" );
575    else if( tr_streq( key, keylen, "dateCreated" ) )
576        tr_bencDictAddInt( d, key, inf->dateCreated );
577    else if( tr_streq( key, keylen, "desiredAvailable" ) )
578        tr_bencDictAddInt( d, key, st->desiredAvailable );
579    else if( tr_streq( key, keylen, "doneDate" ) )
580        tr_bencDictAddInt( d, key, st->doneDate );
581    else if( tr_streq( key, keylen, "downloadDir" ) )
582        tr_bencDictAddStr( d, key, tr_torrentGetDownloadDir( tor ) );
583    else if( tr_streq( key, keylen, "downloadedEver" ) )
584        tr_bencDictAddInt( d, key, st->downloadedEver );
585    else if( tr_streq( key, keylen, "downloadLimit" ) )
586        tr_bencDictAddInt( d, key, tr_torrentGetSpeedLimit_KBps( tor, TR_DOWN ) );
587    else if( tr_streq( key, keylen, "downloadLimited" ) )
588        tr_bencDictAddBool( d, key, tr_torrentUsesSpeedLimit( tor, TR_DOWN ) );
589    else if( tr_streq( key, keylen, "error" ) )
590        tr_bencDictAddInt( d, key, st->error );
591    else if( tr_streq( key, keylen, "errorString" ) )
592        tr_bencDictAddStr( d, key, st->errorString );
593    else if( tr_streq( key, keylen, "eta" ) )
594        tr_bencDictAddInt( d, key, st->eta );
595    else if( tr_streq( key, keylen, "files" ) )
596        addFiles( tor, tr_bencDictAddList( d, key, inf->fileCount ) );
597    else if( tr_streq( key, keylen, "fileStats" ) )
598        addFileStats( tor, tr_bencDictAddList( d, key, inf->fileCount ) );
599    else if( tr_streq( key, keylen, "hashString" ) )
600        tr_bencDictAddStr( d, key, tor->info.hashString );
601    else if( tr_streq( key, keylen, "haveUnchecked" ) )
602        tr_bencDictAddInt( d, key, st->haveUnchecked );
603    else if( tr_streq( key, keylen, "haveValid" ) )
604        tr_bencDictAddInt( d, key, st->haveValid );
605    else if( tr_streq( key, keylen, "honorsSessionLimits" ) )
606        tr_bencDictAddBool( d, key, tr_torrentUsesSessionLimits( tor ) );
607    else if( tr_streq( key, keylen, "id" ) )
608        tr_bencDictAddInt( d, key, st->id );
609    else if( tr_streq( key, keylen, "isFinished" ) )
610        tr_bencDictAddBool( d, key, st->finished );
611    else if( tr_streq( key, keylen, "isPrivate" ) )
612        tr_bencDictAddBool( d, key, tr_torrentIsPrivate( tor ) );
613    else if( tr_streq( key, keylen, "isStalled" ) )
614        tr_bencDictAddBool( d, key, st->isStalled );
615    else if( tr_streq( key, keylen, "leftUntilDone" ) )
616        tr_bencDictAddInt( d, key, st->leftUntilDone );
617    else if( tr_streq( key, keylen, "manualAnnounceTime" ) )
618        tr_bencDictAddInt( d, key, st->manualAnnounceTime );
619    else if( tr_streq( key, keylen, "maxConnectedPeers" ) )
620        tr_bencDictAddInt( d, key,  tr_torrentGetPeerLimit( tor ) );
621    else if( tr_streq( key, keylen, "magnetLink" ) ) {
622        char * str = tr_torrentGetMagnetLink( tor );
623        tr_bencDictAddStr( d, key, str );
624        tr_free( str );
625    }
626    else if( tr_streq( key, keylen, "metadataPercentComplete" ) )
627        tr_bencDictAddReal( d, key, st->metadataPercentComplete );
628    else if( tr_streq( key, keylen, "name" ) )
629        tr_bencDictAddStr( d, key, tr_torrentName( tor ) );
630    else if( tr_streq( key, keylen, "percentDone" ) )
631        tr_bencDictAddReal( d, key, st->percentDone );
632    else if( tr_streq( key, keylen, "peer-limit" ) )
633        tr_bencDictAddInt( d, key, tr_torrentGetPeerLimit( tor ) );
634    else if( tr_streq( key, keylen, "peers" ) )
635        addPeers( tor, tr_bencDictAdd( d, key ) );
636    else if( tr_streq( key, keylen, "peersConnected" ) )
637        tr_bencDictAddInt( d, key, st->peersConnected );
638    else if( tr_streq( key, keylen, "peersFrom" ) )
639    {
640        tr_benc *   tmp = tr_bencDictAddDict( d, key, 6 );
641        const int * f = st->peersFrom;
642        tr_bencDictAddInt( tmp, "fromCache",    f[TR_PEER_FROM_RESUME] );
643        tr_bencDictAddInt( tmp, "fromDht",      f[TR_PEER_FROM_DHT] );
644        tr_bencDictAddInt( tmp, "fromIncoming", f[TR_PEER_FROM_INCOMING] );
645        tr_bencDictAddInt( tmp, "fromLtep",     f[TR_PEER_FROM_LTEP] );
646        tr_bencDictAddInt( tmp, "fromPex",      f[TR_PEER_FROM_PEX] );
647        tr_bencDictAddInt( tmp, "fromTracker",  f[TR_PEER_FROM_TRACKER] );
648    }
649    else if( tr_streq( key, keylen, "peersGettingFromUs" ) )
650        tr_bencDictAddInt( d, key, st->peersGettingFromUs );
651    else if( tr_streq( key, keylen, "peersSendingToUs" ) )
652        tr_bencDictAddInt( d, key, st->peersSendingToUs );
653    else if( tr_streq( key, keylen, "pieces" ) ) {
654        size_t byte_count = 0;
655        void * bytes = tr_cpCreatePieceBitfield( &tor->completion, &byte_count );
656        char * str = tr_base64_encode( bytes, byte_count, NULL );
657        tr_bencDictAddStr( d, key, str!=NULL ? str : "" );
658        tr_free( str );
659        tr_free( bytes );
660    }
661    else if( tr_streq( key, keylen, "pieceCount" ) )
662        tr_bencDictAddInt( d, key, inf->pieceCount );
663    else if( tr_streq( key, keylen, "pieceSize" ) )
664        tr_bencDictAddInt( d, key, inf->pieceSize );
665    else if( tr_streq( key, keylen, "priorities" ) )
666    {
667        tr_file_index_t i;
668        tr_benc *       p = tr_bencDictAddList( d, key, inf->fileCount );
669        for( i = 0; i < inf->fileCount; ++i )
670            tr_bencListAddInt( p, inf->files[i].priority );
671    }
672    else if( tr_streq( key, keylen, "queuePosition" ) )
673        tr_bencDictAddInt( d, key, st->queuePosition );
674    else if( tr_streq( key, keylen, "rateDownload" ) )
675        tr_bencDictAddInt( d, key, toSpeedBytes( st->pieceDownloadSpeed_KBps ) );
676    else if( tr_streq( key, keylen, "rateUpload" ) )
677        tr_bencDictAddInt( d, key, toSpeedBytes( st->pieceUploadSpeed_KBps ) );
678    else if( tr_streq( key, keylen, "recheckProgress" ) )
679        tr_bencDictAddReal( d, key, st->recheckProgress );
680    else if( tr_streq( key, keylen, "seedIdleLimit" ) )
681        tr_bencDictAddInt( d, key, tr_torrentGetIdleLimit( tor ) );
682    else if( tr_streq( key, keylen, "seedIdleMode" ) )
683        tr_bencDictAddInt( d, key, tr_torrentGetIdleMode( tor ) );
684    else if( tr_streq( key, keylen, "seedRatioLimit" ) )
685        tr_bencDictAddReal( d, key, tr_torrentGetRatioLimit( tor ) );
686    else if( tr_streq( key, keylen, "seedRatioMode" ) )
687        tr_bencDictAddInt( d, key, tr_torrentGetRatioMode( tor ) );
688    else if( tr_streq( key, keylen, "sizeWhenDone" ) )
689        tr_bencDictAddInt( d, key, st->sizeWhenDone );
690    else if( tr_streq( key, keylen, "startDate" ) )
691        tr_bencDictAddInt( d, key, st->startDate );
692    else if( tr_streq( key, keylen, "status" ) )
693        tr_bencDictAddInt( d, key, st->activity );
694    else if( tr_streq( key, keylen, "secondsDownloading" ) )
695        tr_bencDictAddInt( d, key, st->secondsDownloading );
696    else if( tr_streq( key, keylen, "secondsSeeding" ) )
697        tr_bencDictAddInt( d, key, st->secondsSeeding );
698    else if( tr_streq( key, keylen, "trackers" ) )
699        addTrackers( inf, tr_bencDictAddList( d, key, inf->trackerCount ) );
700    else if( tr_streq( key, keylen, "trackerStats" ) ) {
701        int n;
702        tr_tracker_stat * s = tr_torrentTrackers( tor, &n );
703        addTrackerStats( s, n, tr_bencDictAddList( d, key, n ) );
704        tr_torrentTrackersFree( s, n );
705    }
706    else if( tr_streq( key, keylen, "torrentFile" ) )
707        tr_bencDictAddStr( d, key, inf->torrent );
708    else if( tr_streq( key, keylen, "totalSize" ) )
709        tr_bencDictAddInt( d, key, inf->totalSize );
710    else if( tr_streq( key, keylen, "uploadedEver" ) )
711        tr_bencDictAddInt( d, key, st->uploadedEver );
712    else if( tr_streq( key, keylen, "uploadLimit" ) )
713        tr_bencDictAddInt( d, key, tr_torrentGetSpeedLimit_KBps( tor, TR_UP ) );
714    else if( tr_streq( key, keylen, "uploadLimited" ) )
715        tr_bencDictAddBool( d, key, tr_torrentUsesSpeedLimit( tor, TR_UP ) );
716    else if( tr_streq( key, keylen, "uploadRatio" ) )
717        tr_bencDictAddReal( d, key, st->ratio );
718    else if( tr_streq( key, keylen, "wanted" ) )
719    {
720        tr_file_index_t i;
721        tr_benc *       w = tr_bencDictAddList( d, key, inf->fileCount );
722        for( i = 0; i < inf->fileCount; ++i )
723            tr_bencListAddInt( w, inf->files[i].dnd ? 0 : 1 );
724    }
725    else if( tr_streq( key, keylen, "webseeds" ) )
726        addWebseeds( inf, tr_bencDictAddList( d, key, inf->webseedCount ) );
727    else if( tr_streq( key, keylen, "webseedsSendingToUs" ) )
728        tr_bencDictAddInt( d, key, st->webseedsSendingToUs );
729}
730
731static void
732addInfo( const tr_torrent * tor, tr_benc * d, tr_benc * fields )
733{
734    const char * str;
735    const int n = tr_bencListSize( fields );
736
737    tr_bencInitDict( d, n );
738
739    if( n > 0 )
740    {
741        int i;
742        const tr_info const * inf = tr_torrentInfo( tor );
743        const tr_stat const * st = tr_torrentStat( (tr_torrent*)tor );
744
745        for( i=0; i<n; ++i )
746            if( tr_bencGetStr( tr_bencListChild( fields, i ), &str ) )
747                addField( tor, inf, st, d, str );
748    }
749}
750
751static const char*
752torrentGet( tr_session               * session,
753            tr_benc                  * args_in,
754            tr_benc                  * args_out,
755            struct tr_rpc_idle_data  * idle_data UNUSED )
756{
757    int           i, torrentCount;
758    tr_torrent ** torrents = getTorrents( session, args_in, &torrentCount );
759    tr_benc *     list = tr_bencDictAddList( args_out, "torrents", torrentCount );
760    tr_benc *     fields;
761    const char *  msg = NULL;
762    const char *  strVal;
763
764    assert( idle_data == NULL );
765
766    if( tr_bencDictFindStr( args_in, "ids", &strVal ) && !strcmp( strVal, "recently-active" ) ) {
767        int n = 0;
768        tr_benc * d;
769        const time_t now = tr_time( );
770        const int interval = RECENTLY_ACTIVE_SECONDS;
771        tr_benc * removed_out = tr_bencDictAddList( args_out, "removed", 0 );
772        while(( d = tr_bencListChild( &session->removedTorrents, n++ ))) {
773            int64_t intVal;
774            if( tr_bencDictFindInt( d, "date", &intVal ) && ( intVal >= now - interval ) ) {
775                tr_bencDictFindInt( d, "id", &intVal );
776                tr_bencListAddInt( removed_out, intVal );
777            }
778        }
779    }
780
781    if( !tr_bencDictFindList( args_in, "fields", &fields ) )
782        msg = "no fields specified";
783    else for( i = 0; i < torrentCount; ++i )
784        addInfo( torrents[i], tr_bencListAdd( list ), fields );
785
786    tr_free( torrents );
787    return msg;
788}
789
790/***
791****
792***/
793
794static const char*
795setFilePriorities( tr_torrent * tor,
796                   int          priority,
797                   tr_benc *    list )
798{
799    int i;
800    int64_t tmp;
801    int fileCount = 0;
802    const int n = tr_bencListSize( list );
803    const char * errmsg = NULL;
804    tr_file_index_t * files = tr_new0( tr_file_index_t, tor->info.fileCount );
805
806    if( n )
807    {
808        for( i = 0; i < n; ++i ) {
809            if( tr_bencGetInt( tr_bencListChild( list, i ), &tmp ) ) {
810                if( 0 <= tmp && tmp < tor->info.fileCount ) {
811                    files[fileCount++] = tmp;
812                } else {
813                    errmsg = "file index out of range";
814                }
815            }
816        }
817    }
818    else /* if empty set, apply to all */
819    {
820        tr_file_index_t t;
821        for( t = 0; t < tor->info.fileCount; ++t )
822            files[fileCount++] = t;
823    }
824
825    if( fileCount )
826        tr_torrentSetFilePriorities( tor, files, fileCount, priority );
827
828    tr_free( files );
829    return errmsg;
830}
831
832static const char*
833setFileDLs( tr_torrent * tor,
834            int          do_download,
835            tr_benc *    list )
836{
837    int i;
838    int64_t tmp;
839    int fileCount = 0;
840    const int n = tr_bencListSize( list );
841    const char * errmsg = NULL;
842    tr_file_index_t * files = tr_new0( tr_file_index_t, tor->info.fileCount );
843
844    if( n ) /* if argument list, process them */
845    {
846        for( i = 0; i < n; ++i ) {
847            if( tr_bencGetInt( tr_bencListChild( list, i ), &tmp ) ) {
848                if( 0 <= tmp && tmp < tor->info.fileCount ) {
849                    files[fileCount++] = tmp;
850                } else {
851                    errmsg = "file index out of range";
852                }
853            }
854        }
855    }
856    else /* if empty set, apply to all */
857    {
858        tr_file_index_t t;
859        for( t = 0; t < tor->info.fileCount; ++t )
860            files[fileCount++] = t;
861    }
862
863    if( fileCount )
864        tr_torrentSetFileDLs( tor, files, fileCount, do_download );
865
866    tr_free( files );
867    return errmsg;
868}
869
870static bool
871findAnnounceUrl( const tr_tracker_info * t, int n, const char * url, int * pos )
872{
873    int i;
874    bool found = false;
875
876    for( i=0; i<n; ++i )
877    {
878        if( !strcmp( t[i].announce, url ) )
879        {
880            found = true;
881            if( pos ) *pos = i;
882            break;
883        }
884    }
885
886    return found;
887}
888
889static int
890copyTrackers( tr_tracker_info * tgt, const tr_tracker_info * src, int n )
891{
892    int i;
893    int maxTier = -1;
894
895    for( i=0; i<n; ++i )
896    {
897        tgt[i].tier = src[i].tier;
898        tgt[i].announce = tr_strdup( src[i].announce );
899        maxTier = MAX( maxTier, src[i].tier );
900    }
901
902    return maxTier;
903}
904
905static void
906freeTrackers( tr_tracker_info * trackers, int n )
907{
908    int i;
909
910    for( i=0; i<n; ++i )
911        tr_free( trackers[i].announce );
912
913    tr_free( trackers );
914}
915
916static const char*
917addTrackerUrls( tr_torrent * tor, tr_benc * urls )
918{
919    int i;
920    int n;
921    int tier;
922    tr_benc * val;
923    tr_tracker_info * trackers;
924    bool changed = false;
925    const tr_info * inf = tr_torrentInfo( tor );
926    const char * errmsg = NULL;
927
928    /* make a working copy of the existing announce list */
929    n = inf->trackerCount;
930    trackers = tr_new0( tr_tracker_info, n + tr_bencListSize( urls ) );
931    tier = copyTrackers( trackers, inf->trackers, n );
932
933    /* and add the new ones */
934    i = 0;
935    while(( val = tr_bencListChild( urls, i++ ) ))
936    {
937        const char * announce = NULL;
938
939        if(    tr_bencGetStr( val, &announce )
940            && tr_urlIsValidTracker( announce )
941            && !findAnnounceUrl( trackers, n, announce, NULL ) )
942        {
943            trackers[n].tier = ++tier; /* add a new tier */
944            trackers[n].announce = tr_strdup( announce );
945            ++n;
946            changed = true;
947        }
948    }
949
950    if( !changed )
951        errmsg = "invalid argument";
952    else if( !tr_torrentSetAnnounceList( tor, trackers, n ) )
953        errmsg = "error setting announce list";
954
955    freeTrackers( trackers, n );
956    return errmsg;
957}
958
959static const char*
960replaceTrackers( tr_torrent * tor, tr_benc * urls )
961{
962    int i;
963    tr_benc * pair[2];
964    tr_tracker_info * trackers;
965    bool changed = false;
966    const tr_info * inf = tr_torrentInfo( tor );
967    const int n = inf->trackerCount;
968    const char * errmsg = NULL;
969
970    /* make a working copy of the existing announce list */
971    trackers = tr_new0( tr_tracker_info, n );
972    copyTrackers( trackers, inf->trackers, n );
973
974    /* make the substitutions... */
975    i = 0;
976    while(((pair[0] = tr_bencListChild(urls,i))) &&
977          ((pair[1] = tr_bencListChild(urls,i+1))))
978    {
979        int64_t pos;
980        const char * newval;
981
982        if(    tr_bencGetInt( pair[0], &pos )
983            && tr_bencGetStr( pair[1], &newval )
984            && tr_urlIsValid( newval, -1 )
985            && pos < n
986            && pos >= 0 )
987        {
988            tr_free( trackers[pos].announce );
989            trackers[pos].announce = tr_strdup( newval );
990            changed = true;
991        }
992
993        i += 2;
994    }
995
996    if( !changed )
997        errmsg = "invalid argument";
998    else if( !tr_torrentSetAnnounceList( tor, trackers, n ) )
999        errmsg = "error setting announce list";
1000
1001    freeTrackers( trackers, n );
1002    return errmsg;
1003}
1004
1005static const char*
1006removeTrackers( tr_torrent * tor, tr_benc * ids )
1007{
1008    int i;
1009    int n;
1010    int t = 0;
1011    int dup = -1;
1012    int * tids;
1013    tr_benc * val;
1014    tr_tracker_info * trackers;
1015    bool changed = false;
1016    const tr_info * inf = tr_torrentInfo( tor );
1017    const char * errmsg = NULL;
1018
1019    /* make a working copy of the existing announce list */
1020    n = inf->trackerCount;
1021    tids = tr_new0( int, n );
1022    trackers = tr_new0( tr_tracker_info, n );
1023    copyTrackers( trackers, inf->trackers, n );
1024
1025    /* remove the ones specified in the urls list */
1026    i = 0;
1027    while(( val = tr_bencListChild( ids, i++ )))
1028    {
1029        int64_t pos;
1030
1031        if(    tr_bencGetInt( val, &pos )
1032            && pos < n
1033            && pos >= 0 )
1034            tids[t++] = pos;
1035    }
1036
1037    /* sort trackerIds and remove from largest to smallest so there is no need to recacluate array indicies */
1038    qsort( tids, t, sizeof(int), compareInt );
1039    while( t-- )
1040    {
1041        /* check for duplicates */
1042        if( tids[t] == dup )
1043            continue;
1044        tr_removeElementFromArray( trackers, tids[t], sizeof( tr_tracker_info ), n-- );
1045        dup = tids[t];
1046        changed = true;
1047    }
1048
1049    if( !changed )
1050        errmsg = "invalid argument";
1051    else if( !tr_torrentSetAnnounceList( tor, trackers, n ) )
1052        errmsg = "error setting announce list";
1053
1054    freeTrackers( trackers, n );
1055    tr_free( tids );
1056    return errmsg;
1057}
1058
1059static const char*
1060torrentSet( tr_session               * session,
1061            tr_benc                  * args_in,
1062            tr_benc                  * args_out UNUSED,
1063            struct tr_rpc_idle_data  * idle_data UNUSED )
1064{
1065    const char * errmsg = NULL;
1066    int i, torrentCount;
1067    tr_torrent ** torrents = getTorrents( session, args_in, &torrentCount );
1068
1069    assert( idle_data == NULL );
1070
1071    for( i = 0; i < torrentCount; ++i )
1072    {
1073        int64_t      tmp;
1074        double       d;
1075        tr_benc *    files;
1076        tr_benc *    trackers;
1077        bool         boolVal;
1078        tr_torrent * tor = torrents[i];
1079
1080        if( tr_bencDictFindInt( args_in, "bandwidthPriority", &tmp ) )
1081            if( tr_isPriority( tmp ) )
1082                tr_torrentSetPriority( tor, tmp );
1083        if( !errmsg && tr_bencDictFindList( args_in, "files-unwanted", &files ) )
1084            errmsg = setFileDLs( tor, false, files );
1085        if( !errmsg && tr_bencDictFindList( args_in, "files-wanted", &files ) )
1086            errmsg = setFileDLs( tor, true, files );
1087        if( tr_bencDictFindInt( args_in, "peer-limit", &tmp ) )
1088            tr_torrentSetPeerLimit( tor, tmp );
1089        if( !errmsg &&  tr_bencDictFindList( args_in, "priority-high", &files ) )
1090            errmsg = setFilePriorities( tor, TR_PRI_HIGH, files );
1091        if( !errmsg && tr_bencDictFindList( args_in, "priority-low", &files ) )
1092            errmsg = setFilePriorities( tor, TR_PRI_LOW, files );
1093        if( !errmsg && tr_bencDictFindList( args_in, "priority-normal", &files ) )
1094            errmsg = setFilePriorities( tor, TR_PRI_NORMAL, files );
1095        if( tr_bencDictFindInt( args_in, "downloadLimit", &tmp ) )
1096            tr_torrentSetSpeedLimit_KBps( tor, TR_DOWN, tmp );
1097        if( tr_bencDictFindBool( args_in, "downloadLimited", &boolVal ) )
1098            tr_torrentUseSpeedLimit( tor, TR_DOWN, boolVal );
1099        if( tr_bencDictFindBool( args_in, "honorsSessionLimits", &boolVal ) )
1100            tr_torrentUseSessionLimits( tor, boolVal );
1101        if( tr_bencDictFindInt( args_in, "uploadLimit", &tmp ) )
1102            tr_torrentSetSpeedLimit_KBps( tor, TR_UP, tmp );
1103        if( tr_bencDictFindBool( args_in, "uploadLimited", &boolVal ) )
1104            tr_torrentUseSpeedLimit( tor, TR_UP, boolVal );
1105        if( tr_bencDictFindInt( args_in, "seedIdleLimit", &tmp ) )
1106            tr_torrentSetIdleLimit( tor, tmp );
1107        if( tr_bencDictFindInt( args_in, "seedIdleMode", &tmp ) )
1108            tr_torrentSetIdleMode( tor, tmp );
1109        if( tr_bencDictFindReal( args_in, "seedRatioLimit", &d ) )
1110            tr_torrentSetRatioLimit( tor, d );
1111        if( tr_bencDictFindInt( args_in, "seedRatioMode", &tmp ) )
1112            tr_torrentSetRatioMode( tor, tmp );
1113        if( tr_bencDictFindInt( args_in, "queuePosition", &tmp ) )
1114            tr_torrentSetQueuePosition( tor, tmp );
1115        if( !errmsg && tr_bencDictFindList( args_in, "trackerAdd", &trackers ) )
1116            errmsg = addTrackerUrls( tor, trackers );
1117        if( !errmsg && tr_bencDictFindList( args_in, "trackerRemove", &trackers ) )
1118            errmsg = removeTrackers( tor, trackers );
1119        if( !errmsg && tr_bencDictFindList( args_in, "trackerReplace", &trackers ) )
1120            errmsg = replaceTrackers( tor, trackers );
1121        notify( session, TR_RPC_TORRENT_CHANGED, tor );
1122    }
1123
1124    tr_free( torrents );
1125    return errmsg;
1126}
1127
1128static const char*
1129torrentSetLocation( tr_session               * session,
1130                    tr_benc                  * args_in,
1131                    tr_benc                  * args_out UNUSED,
1132                    struct tr_rpc_idle_data  * idle_data UNUSED )
1133{
1134    const char * errmsg = NULL;
1135    const char * location = NULL;
1136
1137    assert( idle_data == NULL );
1138
1139    if( !tr_bencDictFindStr( args_in, "location", &location ) )
1140    {
1141        errmsg = "no location";
1142    }
1143    else
1144    {
1145        bool move = false;
1146        int i, torrentCount;
1147        tr_torrent ** torrents = getTorrents( session, args_in, &torrentCount );
1148
1149        tr_bencDictFindBool( args_in, "move", &move );
1150
1151        for( i=0; i<torrentCount; ++i )
1152        {
1153            tr_torrent * tor = torrents[i];
1154            tr_torrentSetLocation( tor, location, move, NULL, NULL );
1155            notify( session, TR_RPC_TORRENT_MOVED, tor );
1156        }
1157
1158        tr_free( torrents );
1159    }
1160
1161    return errmsg;
1162}
1163
1164/***
1165****
1166***/
1167
1168static void
1169portTested( tr_session       * session UNUSED,
1170            bool               did_connect UNUSED,
1171            bool               did_timeout UNUSED,
1172            long               response_code,
1173            const void       * response,
1174            size_t             response_byte_count,
1175            void             * user_data )
1176{
1177    char result[1024];
1178    struct tr_rpc_idle_data * data = user_data;
1179
1180    if( response_code != 200 )
1181    {
1182        tr_snprintf( result, sizeof( result ), "portTested: http error %ld: %s",
1183                     response_code, tr_webGetResponseStr( response_code ) );
1184    }
1185    else /* success */
1186    {
1187        const bool isOpen = response_byte_count && *(char*)response == '1';
1188        tr_bencDictAddBool( data->args_out, "port-is-open", isOpen );
1189        tr_snprintf( result, sizeof( result ), "success" );
1190    }
1191
1192    tr_idle_function_done( data, result );
1193}
1194
1195static const char*
1196portTest( tr_session               * session,
1197          tr_benc                  * args_in UNUSED,
1198          tr_benc                  * args_out UNUSED,
1199          struct tr_rpc_idle_data  * idle_data )
1200{
1201    const int port = tr_sessionGetPeerPort( session );
1202    char * url = tr_strdup_printf( "http://portcheck.transmissionbt.com/%d", port );
1203    tr_webRun( session, url, NULL, NULL, portTested, idle_data );
1204    tr_free( url );
1205    return NULL;
1206}
1207
1208/***
1209****
1210***/
1211
1212static void
1213gotNewBlocklist( tr_session       * session,
1214                 bool               did_connect UNUSED,
1215                 bool               did_timeout UNUSED,
1216                 long               response_code,
1217                 const void       * response,
1218                 size_t             response_byte_count,
1219                 void             * user_data )
1220{
1221    char result[1024];
1222    struct tr_rpc_idle_data * data = user_data;
1223
1224    *result = '\0';
1225
1226    if( response_code != 200 )
1227    {
1228        tr_snprintf( result, sizeof( result ), "gotNewBlocklist: http error %ld: %s",
1229                     response_code, tr_webGetResponseStr( response_code ) );
1230    }
1231    else /* successfully fetched the blocklist... */
1232    {
1233        int fd;
1234        int err;
1235        char * filename;
1236        z_stream stream;
1237        const char * configDir = tr_sessionGetConfigDir( session );
1238        const size_t buflen = 1024 * 128; /* 128 KiB buffer */
1239        uint8_t * buf = tr_valloc( buflen );
1240
1241        /* this is an odd Magic Number required by zlib to enable gz support.
1242           See zlib's inflateInit2() documentation for a full description */
1243        const int windowBits = 15 + 32;
1244
1245        stream.zalloc = (alloc_func) Z_NULL;
1246        stream.zfree = (free_func) Z_NULL;
1247        stream.opaque = (voidpf) Z_NULL;
1248        stream.next_in = (void*) response;
1249        stream.avail_in = response_byte_count;
1250        inflateInit2( &stream, windowBits );
1251
1252        filename = tr_buildPath( configDir, "blocklist.tmp", NULL );
1253        fd = tr_open_file_for_writing( filename );
1254        if( fd < 0 )
1255            tr_snprintf( result, sizeof( result ), _( "Couldn't save file \"%1$s\": %2$s" ), filename, tr_strerror( errno ) );
1256
1257        for( ;; )
1258        {
1259            stream.next_out = (void*) buf;
1260            stream.avail_out = buflen;
1261            err = inflate( &stream, Z_NO_FLUSH );
1262
1263            if( stream.avail_out < buflen ) {
1264                const int e = write( fd, buf, buflen - stream.avail_out );
1265                if( e < 0 ) {
1266                    tr_snprintf( result, sizeof( result ), _( "Couldn't save file \"%1$s\": %2$s" ), filename, tr_strerror( errno ) );
1267                    break;
1268                }
1269            }
1270
1271            if( err != Z_OK ) {
1272                if( ( err != Z_STREAM_END ) && ( err != Z_DATA_ERROR ) )
1273                    tr_snprintf( result, sizeof( result ), _( "Error uncompressing blocklist: %s (%d)" ), zError( err ), err );
1274                break;
1275            }
1276        }
1277
1278        inflateEnd( &stream );
1279
1280        if( err == Z_DATA_ERROR ) /* couldn't inflate it... it's probably already uncompressed */
1281            if( write( fd, response, response_byte_count ) < 0 )
1282                tr_snprintf( result, sizeof( result ), _( "Couldn't save file \"%1$s\": %2$s" ), filename, tr_strerror( errno ) );
1283
1284        if( *result )
1285            tr_err( "%s", result );
1286        else {
1287            /* feed it to the session and give the client a response */
1288            const int rule_count = tr_blocklistSetContent( session, filename );
1289            tr_bencDictAddInt( data->args_out, "blocklist-size", rule_count );
1290            tr_snprintf( result, sizeof( result ), "success" );
1291        }
1292
1293        unlink( filename );
1294        tr_free( filename );
1295        tr_free( buf );
1296    }
1297
1298    tr_idle_function_done( data, result );
1299}
1300
1301static const char*
1302blocklistUpdate( tr_session               * session,
1303                 tr_benc                  * args_in UNUSED,
1304                 tr_benc                  * args_out UNUSED,
1305                 struct tr_rpc_idle_data  * idle_data )
1306{
1307    tr_webRun( session, session->blocklist_url, NULL, NULL, gotNewBlocklist, idle_data );
1308    return NULL;
1309}
1310
1311/***
1312****
1313***/
1314
1315static void
1316addTorrentImpl( struct tr_rpc_idle_data * data, tr_ctor * ctor )
1317{
1318    int err = 0;
1319    const char * result = NULL;
1320    tr_torrent * tor = tr_torrentNew( ctor, &err );
1321
1322    tr_ctorFree( ctor );
1323
1324    if( tor )
1325    {
1326        tr_benc fields;
1327        tr_bencInitList( &fields, 3 );
1328        tr_bencListAddStr( &fields, "id" );
1329        tr_bencListAddStr( &fields, "name" );
1330        tr_bencListAddStr( &fields, "hashString" );
1331        addInfo( tor, tr_bencDictAdd( data->args_out, "torrent-added" ), &fields );
1332        notify( data->session, TR_RPC_TORRENT_ADDED, tor );
1333        tr_bencFree( &fields );
1334    }
1335    else if( err == TR_PARSE_DUPLICATE )
1336    {
1337        result = "duplicate torrent";
1338    }
1339    else if( err == TR_PARSE_ERR )
1340    {
1341        result = "invalid or corrupt torrent file";
1342    }
1343
1344    tr_idle_function_done( data, result );
1345}
1346
1347
1348struct add_torrent_idle_data
1349{
1350    struct tr_rpc_idle_data * data;
1351    tr_ctor * ctor;
1352};
1353
1354static void
1355gotMetadataFromURL( tr_session       * session UNUSED,
1356                    bool               did_connect UNUSED,
1357                    bool               did_timeout UNUSED,
1358                    long               response_code,
1359                    const void       * response,
1360                    size_t             response_byte_count,
1361                    void             * user_data )
1362{
1363    struct add_torrent_idle_data * data = user_data;
1364
1365    dbgmsg( "torrentAdd: HTTP response code was %ld (%s); response length was %zu bytes",
1366            response_code, tr_webGetResponseStr( response_code ), response_byte_count );
1367
1368    if( response_code==200 || response_code==221 ) /* http or ftp success.. */
1369    {
1370        tr_ctorSetMetainfo( data->ctor, response, response_byte_count );
1371        addTorrentImpl( data->data, data->ctor );
1372    }
1373    else
1374    {
1375        char result[1024];
1376        tr_snprintf( result, sizeof( result ), "gotMetadataFromURL: http error %ld: %s",
1377                     response_code, tr_webGetResponseStr( response_code ) );
1378        tr_idle_function_done( data->data, result );
1379    }
1380
1381    tr_free( data );
1382}
1383
1384static bool
1385isCurlURL( const char * filename )
1386{
1387    if( filename == NULL )
1388        return false;
1389
1390    return !strncmp( filename, "ftp://", 6 ) ||
1391           !strncmp( filename, "http://", 7 ) ||
1392           !strncmp( filename, "https://", 8 );
1393}
1394
1395static tr_file_index_t*
1396fileListFromList( tr_benc * list, tr_file_index_t * setmeCount )
1397{
1398    size_t i;
1399    const size_t childCount = tr_bencListSize( list );
1400    tr_file_index_t n = 0;
1401    tr_file_index_t * files = tr_new0( tr_file_index_t, childCount );
1402
1403    for( i=0; i<childCount; ++i ) {
1404        int64_t intVal;
1405        if( tr_bencGetInt( tr_bencListChild( list, i ), &intVal ) )
1406            files[n++] = (tr_file_index_t)intVal;
1407    }
1408
1409    *setmeCount = n;
1410    return files;
1411}
1412
1413static const char*
1414torrentAdd( tr_session               * session,
1415            tr_benc                  * args_in,
1416            tr_benc                  * args_out UNUSED,
1417            struct tr_rpc_idle_data  * idle_data )
1418{
1419    const char * filename = NULL;
1420    const char * metainfo_base64 = NULL;
1421
1422    assert( idle_data != NULL );
1423
1424    tr_bencDictFindStr( args_in, "filename", &filename );
1425    tr_bencDictFindStr( args_in, "metainfo", &metainfo_base64 );
1426    if( !filename && !metainfo_base64 )
1427        return "no filename or metainfo specified";
1428    else
1429    {
1430        int64_t      i;
1431        bool         boolVal;
1432        tr_benc    * l;
1433        const char * str;
1434        const char * cookies = NULL;
1435        tr_ctor    * ctor = tr_ctorNew( session );
1436
1437        /* set the optional arguments */
1438
1439        tr_bencDictFindStr( args_in, "cookies", &cookies );
1440
1441        if( tr_bencDictFindStr( args_in, TR_PREFS_KEY_DOWNLOAD_DIR, &str ) )
1442            tr_ctorSetDownloadDir( ctor, TR_FORCE, str );
1443
1444        if( tr_bencDictFindBool( args_in, "paused", &boolVal ) )
1445            tr_ctorSetPaused( ctor, TR_FORCE, boolVal );
1446
1447        if( tr_bencDictFindInt( args_in, "peer-limit", &i ) )
1448            tr_ctorSetPeerLimit( ctor, TR_FORCE, i );
1449
1450        if( tr_bencDictFindInt( args_in, "bandwidthPriority", &i ) )
1451            tr_ctorSetBandwidthPriority( ctor, i );
1452
1453        if( tr_bencDictFindList( args_in, "files-unwanted", &l ) ) {
1454            tr_file_index_t fileCount;
1455            tr_file_index_t * files = fileListFromList( l, &fileCount );
1456            tr_ctorSetFilesWanted( ctor, files, fileCount, false );
1457            tr_free( files );
1458        }
1459        if( tr_bencDictFindList( args_in, "files-wanted", &l ) ) {
1460            tr_file_index_t fileCount;
1461            tr_file_index_t * files = fileListFromList( l, &fileCount );
1462            tr_ctorSetFilesWanted( ctor, files, fileCount, true );
1463            tr_free( files );
1464        }
1465
1466        if( tr_bencDictFindList( args_in, "priority-low", &l ) ) {
1467            tr_file_index_t fileCount;
1468            tr_file_index_t * files = fileListFromList( l, &fileCount );
1469            tr_ctorSetFilePriorities( ctor, files, fileCount, TR_PRI_LOW );
1470            tr_free( files );
1471        }
1472        if( tr_bencDictFindList( args_in, "priority-normal", &l ) ) {
1473            tr_file_index_t fileCount;
1474            tr_file_index_t * files = fileListFromList( l, &fileCount );
1475            tr_ctorSetFilePriorities( ctor, files, fileCount, TR_PRI_NORMAL );
1476            tr_free( files );
1477        }
1478        if( tr_bencDictFindList( args_in, "priority-high", &l ) ) {
1479            tr_file_index_t fileCount;
1480            tr_file_index_t * files = fileListFromList( l, &fileCount );
1481            tr_ctorSetFilePriorities( ctor, files, fileCount, TR_PRI_HIGH );
1482            tr_free( files );
1483        }
1484
1485        dbgmsg( "torrentAdd: filename is \"%s\"", filename ? filename : "(null)" );
1486
1487        if( isCurlURL( filename ) )
1488        {
1489            struct add_torrent_idle_data * d = tr_new0( struct add_torrent_idle_data, 1 );
1490            d->data = idle_data;
1491            d->ctor = ctor;
1492            tr_webRun( session, filename, NULL, cookies, gotMetadataFromURL, d );
1493        }
1494        else
1495        {
1496            char * fname = tr_strstrip( tr_strdup( filename ) );
1497
1498            if( fname == NULL )
1499            {
1500                int len;
1501                char * metainfo = tr_base64_decode( metainfo_base64, -1, &len );
1502                tr_ctorSetMetainfo( ctor, (uint8_t*)metainfo, len );
1503                tr_free( metainfo );
1504            }
1505            else if( !strncmp( fname, "magnet:?", 8 ) )
1506            {
1507                tr_ctorSetMetainfoFromMagnetLink( ctor, fname );
1508            }
1509            else
1510            {
1511                tr_ctorSetMetainfoFromFile( ctor, fname );
1512            }
1513
1514            addTorrentImpl( idle_data, ctor );
1515
1516            tr_free( fname );
1517        }
1518
1519    }
1520
1521    return NULL;
1522}
1523
1524/***
1525****
1526***/
1527
1528static const char*
1529sessionSet( tr_session               * session,
1530            tr_benc                  * args_in,
1531            tr_benc                  * args_out UNUSED,
1532            struct tr_rpc_idle_data  * idle_data UNUSED )
1533{
1534    int64_t      i;
1535    double       d;
1536    bool         boolVal;
1537    const char * str;
1538
1539    assert( idle_data == NULL );
1540
1541    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_MAX_CACHE_SIZE_MB, &i ) )
1542        tr_sessionSetCacheLimit_MB( session, i );
1543    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_ALT_SPEED_UP_KBps, &i ) )
1544        tr_sessionSetAltSpeed_KBps( session, TR_UP, i );
1545    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_ALT_SPEED_DOWN_KBps, &i ) )
1546        tr_sessionSetAltSpeed_KBps( session, TR_DOWN, i );
1547    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_ALT_SPEED_ENABLED, &boolVal ) )
1548        tr_sessionUseAltSpeed( session, boolVal );
1549    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_ALT_SPEED_TIME_BEGIN, &i ) )
1550        tr_sessionSetAltSpeedBegin( session, i );
1551    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_ALT_SPEED_TIME_END, &i ) )
1552        tr_sessionSetAltSpeedEnd( session, i );
1553    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_ALT_SPEED_TIME_DAY, &i ) )
1554        tr_sessionSetAltSpeedDay( session, i );
1555    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_ALT_SPEED_TIME_ENABLED, &boolVal ) )
1556        tr_sessionUseAltSpeedTime( session, boolVal );
1557    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_BLOCKLIST_ENABLED, &boolVal ) )
1558        tr_blocklistSetEnabled( session, boolVal );
1559    if( tr_bencDictFindStr( args_in, TR_PREFS_KEY_BLOCKLIST_URL, &str ) )
1560        tr_blocklistSetURL( session, str );
1561    if( tr_bencDictFindStr( args_in, TR_PREFS_KEY_DOWNLOAD_DIR, &str ) )
1562        tr_sessionSetDownloadDir( session, str );
1563    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_QUEUE_STALLED_MINUTES, &i ) )
1564        tr_sessionSetQueueStalledMinutes( session, i );
1565    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_QUEUE_STALLED_ENABLED, &boolVal ) )
1566        tr_sessionSetQueueStalledEnabled( session, boolVal );
1567    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_DOWNLOAD_QUEUE_SIZE, &i ) )
1568        tr_sessionSetQueueSize( session, TR_DOWN, i );
1569    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_DOWNLOAD_QUEUE_ENABLED, &boolVal ) )
1570        tr_sessionSetQueueEnabled ( session, TR_DOWN, boolVal );
1571    if( tr_bencDictFindStr( args_in, TR_PREFS_KEY_INCOMPLETE_DIR, &str ) )
1572        tr_sessionSetIncompleteDir( session, str );
1573    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_INCOMPLETE_DIR_ENABLED, &boolVal ) )
1574        tr_sessionSetIncompleteDirEnabled( session, boolVal );
1575    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_PEER_LIMIT_GLOBAL, &i ) )
1576        tr_sessionSetPeerLimit( session, i );
1577    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_PEER_LIMIT_TORRENT, &i ) )
1578        tr_sessionSetPeerLimitPerTorrent( session, i );
1579    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_PEX_ENABLED, &boolVal ) )
1580        tr_sessionSetPexEnabled( session, boolVal );
1581    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_DHT_ENABLED, &boolVal ) )
1582        tr_sessionSetDHTEnabled( session, boolVal );
1583    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_UTP_ENABLED, &boolVal ) )
1584        tr_sessionSetUTPEnabled( session, boolVal );
1585    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_LPD_ENABLED, &boolVal ) )
1586        tr_sessionSetLPDEnabled( session, boolVal );
1587    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_PEER_PORT_RANDOM_ON_START, &boolVal ) )
1588        tr_sessionSetPeerPortRandomOnStart( session, boolVal );
1589    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_PEER_PORT, &i ) )
1590        tr_sessionSetPeerPort( session, i );
1591    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_PORT_FORWARDING, &boolVal ) )
1592        tr_sessionSetPortForwardingEnabled( session, boolVal );
1593    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_RENAME_PARTIAL_FILES, &boolVal ) )
1594        tr_sessionSetIncompleteFileNamingEnabled( session, boolVal );
1595    if( tr_bencDictFindReal( args_in, "seedRatioLimit", &d ) )
1596        tr_sessionSetRatioLimit( session, d );
1597    if( tr_bencDictFindBool( args_in, "seedRatioLimited", &boolVal ) )
1598        tr_sessionSetRatioLimited( session, boolVal );
1599    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_IDLE_LIMIT, &i ) )
1600        tr_sessionSetIdleLimit( session, i );
1601    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_IDLE_LIMIT_ENABLED, &boolVal ) )
1602        tr_sessionSetIdleLimited( session, boolVal );
1603    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_START, &boolVal ) )
1604        tr_sessionSetPaused( session, !boolVal );
1605    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_SEED_QUEUE_ENABLED, &boolVal ) )
1606        tr_sessionSetQueueEnabled ( session, TR_UP, boolVal );
1607    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_SEED_QUEUE_SIZE, &i ) )
1608        tr_sessionSetQueueSize( session, TR_UP, i );
1609    if( tr_bencDictFindStr( args_in, TR_PREFS_KEY_SCRIPT_TORRENT_DONE_FILENAME, &str ) )
1610        tr_sessionSetTorrentDoneScript( session, str );
1611    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_SCRIPT_TORRENT_DONE_ENABLED, &boolVal ) )
1612        tr_sessionSetTorrentDoneScriptEnabled( session, boolVal );
1613    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_TRASH_ORIGINAL, &boolVal ) )
1614        tr_sessionSetDeleteSource( session, boolVal );
1615    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_DSPEED_KBps, &i ) )
1616        tr_sessionSetSpeedLimit_KBps( session, TR_DOWN, i );
1617    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_DSPEED_ENABLED, &boolVal ) )
1618        tr_sessionLimitSpeed( session, TR_DOWN, boolVal );
1619    if( tr_bencDictFindInt( args_in, TR_PREFS_KEY_USPEED_KBps, &i ) )
1620        tr_sessionSetSpeedLimit_KBps( session, TR_UP, i );
1621    if( tr_bencDictFindBool( args_in, TR_PREFS_KEY_USPEED_ENABLED, &boolVal ) )
1622        tr_sessionLimitSpeed( session, TR_UP, boolVal );
1623    if( tr_bencDictFindStr( args_in, TR_PREFS_KEY_ENCRYPTION, &str ) ) {
1624        if( !strcmp( str, "required" ) )
1625            tr_sessionSetEncryption( session, TR_ENCRYPTION_REQUIRED );
1626        else if( !strcmp( str, "tolerated" ) )
1627            tr_sessionSetEncryption( session, TR_CLEAR_PREFERRED );
1628        else
1629            tr_sessionSetEncryption( session, TR_ENCRYPTION_PREFERRED );
1630    }
1631
1632    notify( session, TR_RPC_SESSION_CHANGED, NULL );
1633
1634    return NULL;
1635}
1636
1637static const char*
1638sessionStats( tr_session               * session,
1639              tr_benc                  * args_in UNUSED,
1640              tr_benc                  * args_out,
1641              struct tr_rpc_idle_data  * idle_data UNUSED )
1642{
1643    int running = 0;
1644    int total = 0;
1645    tr_benc * d;
1646    tr_session_stats currentStats = { 0.0f, 0, 0, 0, 0, 0 };
1647    tr_session_stats cumulativeStats = { 0.0f, 0, 0, 0, 0, 0 };
1648    tr_torrent * tor = NULL;
1649
1650    assert( idle_data == NULL );
1651
1652    while(( tor = tr_torrentNext( session, tor ))) {
1653        ++total;
1654        if( tor->isRunning )
1655            ++running;
1656    }
1657
1658    tr_sessionGetStats( session, &currentStats );
1659    tr_sessionGetCumulativeStats( session, &cumulativeStats );
1660
1661    tr_bencDictAddInt ( args_out, "activeTorrentCount", running );
1662    tr_bencDictAddReal( args_out, "downloadSpeed", tr_sessionGetPieceSpeed_Bps( session, TR_DOWN ) );
1663    tr_bencDictAddInt ( args_out, "pausedTorrentCount", total - running );
1664    tr_bencDictAddInt ( args_out, "torrentCount", total );
1665    tr_bencDictAddReal( args_out, "uploadSpeed", tr_sessionGetPieceSpeed_Bps( session, TR_UP ) );
1666
1667    d = tr_bencDictAddDict( args_out, "cumulative-stats", 5 );
1668    tr_bencDictAddInt( d, "downloadedBytes", cumulativeStats.downloadedBytes );
1669    tr_bencDictAddInt( d, "filesAdded", cumulativeStats.filesAdded );
1670    tr_bencDictAddInt( d, "secondsActive", cumulativeStats.secondsActive );
1671    tr_bencDictAddInt( d, "sessionCount", cumulativeStats.sessionCount );
1672    tr_bencDictAddInt( d, "uploadedBytes", cumulativeStats.uploadedBytes );
1673
1674    d = tr_bencDictAddDict( args_out, "current-stats", 5 );
1675    tr_bencDictAddInt( d, "downloadedBytes", currentStats.downloadedBytes );
1676    tr_bencDictAddInt( d, "filesAdded", currentStats.filesAdded );
1677    tr_bencDictAddInt( d, "secondsActive", currentStats.secondsActive );
1678    tr_bencDictAddInt( d, "sessionCount", currentStats.sessionCount );
1679    tr_bencDictAddInt( d, "uploadedBytes", currentStats.uploadedBytes );
1680
1681    return NULL;
1682}
1683
1684static const char*
1685sessionGet( tr_session               * s,
1686            tr_benc                  * args_in UNUSED,
1687            tr_benc                  * args_out,
1688            struct tr_rpc_idle_data  * idle_data UNUSED )
1689{
1690    const char * str;
1691    tr_benc * d = args_out;
1692
1693    assert( idle_data == NULL );
1694    tr_bencDictAddInt ( d, TR_PREFS_KEY_ALT_SPEED_UP_KBps, tr_sessionGetAltSpeed_KBps(s,TR_UP) );
1695    tr_bencDictAddInt ( d, TR_PREFS_KEY_ALT_SPEED_DOWN_KBps, tr_sessionGetAltSpeed_KBps(s,TR_DOWN) );
1696    tr_bencDictAddBool( d, TR_PREFS_KEY_ALT_SPEED_ENABLED, tr_sessionUsesAltSpeed(s) );
1697    tr_bencDictAddInt ( d, TR_PREFS_KEY_ALT_SPEED_TIME_BEGIN, tr_sessionGetAltSpeedBegin(s) );
1698    tr_bencDictAddInt ( d, TR_PREFS_KEY_ALT_SPEED_TIME_END,tr_sessionGetAltSpeedEnd(s) );
1699    tr_bencDictAddInt ( d, TR_PREFS_KEY_ALT_SPEED_TIME_DAY,tr_sessionGetAltSpeedDay(s) );
1700    tr_bencDictAddBool( d, TR_PREFS_KEY_ALT_SPEED_TIME_ENABLED, tr_sessionUsesAltSpeedTime(s) );
1701    tr_bencDictAddBool( d, TR_PREFS_KEY_BLOCKLIST_ENABLED, tr_blocklistIsEnabled( s ) );
1702    tr_bencDictAddStr ( d, TR_PREFS_KEY_BLOCKLIST_URL, tr_blocklistGetURL( s ) );
1703    tr_bencDictAddInt ( d, TR_PREFS_KEY_MAX_CACHE_SIZE_MB, tr_sessionGetCacheLimit_MB( s ) );
1704    tr_bencDictAddInt ( d, "blocklist-size", tr_blocklistGetRuleCount( s ) );
1705    tr_bencDictAddStr ( d, "config-dir", tr_sessionGetConfigDir( s ) );
1706    tr_bencDictAddStr ( d, TR_PREFS_KEY_DOWNLOAD_DIR, tr_sessionGetDownloadDir( s ) );
1707    tr_bencDictAddBool( d, TR_PREFS_KEY_DOWNLOAD_QUEUE_ENABLED, tr_sessionGetQueueEnabled( s, TR_DOWN ) );
1708    tr_bencDictAddInt ( d, TR_PREFS_KEY_DOWNLOAD_QUEUE_SIZE, tr_sessionGetQueueSize( s, TR_DOWN ) );
1709    tr_bencDictAddInt ( d, "download-dir-free-space",  tr_sessionGetDownloadDirFreeSpace( s ) );
1710    tr_bencDictAddInt ( d, TR_PREFS_KEY_PEER_LIMIT_GLOBAL, tr_sessionGetPeerLimit( s ) );
1711    tr_bencDictAddInt ( d, TR_PREFS_KEY_PEER_LIMIT_TORRENT, tr_sessionGetPeerLimitPerTorrent( s ) );
1712    tr_bencDictAddStr ( d, TR_PREFS_KEY_INCOMPLETE_DIR, tr_sessionGetIncompleteDir( s ) );
1713    tr_bencDictAddBool( d, TR_PREFS_KEY_INCOMPLETE_DIR_ENABLED, tr_sessionIsIncompleteDirEnabled( s ) );
1714    tr_bencDictAddBool( d, TR_PREFS_KEY_PEX_ENABLED, tr_sessionIsPexEnabled( s ) );
1715    tr_bencDictAddBool( d, TR_PREFS_KEY_UTP_ENABLED, tr_sessionIsUTPEnabled( s ) );
1716    tr_bencDictAddBool( d, TR_PREFS_KEY_DHT_ENABLED, tr_sessionIsDHTEnabled( s ) );
1717    tr_bencDictAddBool( d, TR_PREFS_KEY_LPD_ENABLED, tr_sessionIsLPDEnabled( s ) );
1718    tr_bencDictAddInt ( d, TR_PREFS_KEY_PEER_PORT, tr_sessionGetPeerPort( s ) );
1719    tr_bencDictAddBool( d, TR_PREFS_KEY_PEER_PORT_RANDOM_ON_START, tr_sessionGetPeerPortRandomOnStart( s ) );
1720    tr_bencDictAddBool( d, TR_PREFS_KEY_PORT_FORWARDING, tr_sessionIsPortForwardingEnabled( s ) );
1721    tr_bencDictAddBool( d, TR_PREFS_KEY_RENAME_PARTIAL_FILES, tr_sessionIsIncompleteFileNamingEnabled( s ) );
1722    tr_bencDictAddInt ( d, "rpc-version", RPC_VERSION );
1723    tr_bencDictAddInt ( d, "rpc-version-minimum", RPC_VERSION_MIN );
1724    tr_bencDictAddReal( d, "seedRatioLimit", tr_sessionGetRatioLimit( s ) );
1725    tr_bencDictAddBool( d, "seedRatioLimited", tr_sessionIsRatioLimited( s ) );
1726    tr_bencDictAddInt ( d, TR_PREFS_KEY_IDLE_LIMIT, tr_sessionGetIdleLimit( s ) );
1727    tr_bencDictAddBool( d, TR_PREFS_KEY_IDLE_LIMIT_ENABLED, tr_sessionIsIdleLimited( s ) );
1728    tr_bencDictAddBool( d, TR_PREFS_KEY_SEED_QUEUE_ENABLED, tr_sessionGetQueueEnabled( s, TR_UP ) );
1729    tr_bencDictAddInt ( d, TR_PREFS_KEY_SEED_QUEUE_SIZE, tr_sessionGetQueueSize( s, TR_UP ) );
1730    tr_bencDictAddBool( d, TR_PREFS_KEY_START, !tr_sessionGetPaused( s ) );
1731    tr_bencDictAddBool( d, TR_PREFS_KEY_TRASH_ORIGINAL, tr_sessionGetDeleteSource( s ) );
1732    tr_bencDictAddInt ( d, TR_PREFS_KEY_USPEED_KBps, tr_sessionGetSpeedLimit_KBps( s, TR_UP ) );
1733    tr_bencDictAddBool( d, TR_PREFS_KEY_USPEED_ENABLED, tr_sessionIsSpeedLimited( s, TR_UP ) );
1734    tr_bencDictAddInt ( d, TR_PREFS_KEY_DSPEED_KBps, tr_sessionGetSpeedLimit_KBps( s, TR_DOWN ) );
1735    tr_bencDictAddBool( d, TR_PREFS_KEY_DSPEED_ENABLED, tr_sessionIsSpeedLimited( s, TR_DOWN ) );
1736    tr_bencDictAddStr ( d, TR_PREFS_KEY_SCRIPT_TORRENT_DONE_FILENAME, tr_sessionGetTorrentDoneScript( s ) );
1737    tr_bencDictAddBool( d, TR_PREFS_KEY_SCRIPT_TORRENT_DONE_ENABLED, tr_sessionIsTorrentDoneScriptEnabled( s ) );
1738    tr_bencDictAddInt ( d, TR_PREFS_KEY_QUEUE_STALLED_MINUTES, tr_sessionGetQueueStalledMinutes( s ) );
1739    tr_bencDictAddBool( d, TR_PREFS_KEY_QUEUE_STALLED_ENABLED, tr_sessionGetQueueStalledEnabled( s ) );
1740    tr_formatter_get_units( tr_bencDictAddDict( d, "units", 0 ) );
1741    tr_bencDictAddStr ( d, "version", LONG_VERSION_STRING );
1742    switch( tr_sessionGetEncryption( s ) ) {
1743        case TR_CLEAR_PREFERRED: str = "tolerated"; break;
1744        case TR_ENCRYPTION_REQUIRED: str = "required"; break;
1745        default: str = "preferred"; break;
1746    }
1747    tr_bencDictAddStr( d, TR_PREFS_KEY_ENCRYPTION, str );
1748
1749    return NULL;
1750}
1751
1752/***
1753****
1754***/
1755
1756static const char*
1757sessionClose( tr_session               * session,
1758              tr_benc                  * args_in UNUSED,
1759              tr_benc                  * args_out UNUSED,
1760              struct tr_rpc_idle_data  * idle_data UNUSED )
1761{
1762    notify( session, TR_RPC_SESSION_CLOSE, NULL );
1763    return NULL;
1764}
1765
1766/***
1767****
1768***/
1769
1770typedef const char* ( *handler )( tr_session*, tr_benc*, tr_benc*, struct tr_rpc_idle_data * );
1771
1772static struct method
1773{
1774    const char *  name;
1775    bool          immediate;
1776    handler       func;
1777}
1778methods[] =
1779{
1780    { "port-test",             false, portTest            },
1781    { "blocklist-update",      false, blocklistUpdate     },
1782    { "session-close",         true,  sessionClose        },
1783    { "session-get",           true,  sessionGet          },
1784    { "session-set",           true,  sessionSet          },
1785    { "session-stats",         true,  sessionStats        },
1786    { "torrent-add",           false, torrentAdd          },
1787    { "torrent-get",           true,  torrentGet          },
1788    { "torrent-remove",        true,  torrentRemove       },
1789    { "torrent-set",           true,  torrentSet          },
1790    { "torrent-set-location",  true,  torrentSetLocation  },
1791    { "torrent-start",         true,  torrentStart        },
1792    { "torrent-start-now",     true,  torrentStartNow     },
1793    { "torrent-stop",          true,  torrentStop         },
1794    { "torrent-verify",        true,  torrentVerify       },
1795    { "torrent-reannounce",    true,  torrentReannounce   },
1796    { "queue-move-top",        true,  queueMoveTop        },
1797    { "queue-move-up",         true,  queueMoveUp         },
1798    { "queue-move-down",       true,  queueMoveDown       },
1799    { "queue-move-bottom",     true,  queueMoveBottom     }
1800};
1801
1802static void
1803noop_response_callback( tr_session       * session UNUSED,
1804                        struct evbuffer  * response UNUSED,
1805                        void             * user_data UNUSED )
1806{
1807}
1808
1809static void
1810request_exec( tr_session             * session,
1811              tr_benc                * request,
1812              tr_rpc_response_func     callback,
1813              void                   * callback_user_data )
1814{
1815    int i;
1816    const char * str;
1817    tr_benc * args_in = tr_bencDictFind( request, "arguments" );
1818    const char * result = NULL;
1819
1820    if( callback == NULL )
1821        callback = noop_response_callback;
1822
1823    /* parse the request */
1824    if( !tr_bencDictFindStr( request, "method", &str ) )
1825        result = "no method name";
1826    else {
1827        const int n = TR_N_ELEMENTS( methods );
1828        for( i = 0; i < n; ++i )
1829            if( !strcmp( str, methods[i].name ) )
1830                break;
1831        if( i ==n )
1832            result = "method name not recognized";
1833    }
1834
1835    /* if we couldn't figure out which method to use, return an error */
1836    if( result != NULL )
1837    {
1838        int64_t tag;
1839        tr_benc response;
1840        struct evbuffer * buf;
1841
1842        tr_bencInitDict( &response, 3 );
1843        tr_bencDictAddDict( &response, "arguments", 0 );
1844        tr_bencDictAddStr( &response, "result", result );
1845        if( tr_bencDictFindInt( request, "tag", &tag ) )
1846            tr_bencDictAddInt( &response, "tag", tag );
1847
1848        buf = tr_bencToBuf( &response, TR_FMT_JSON_LEAN );
1849        (*callback)( session, buf, callback_user_data );
1850        evbuffer_free( buf );
1851
1852        tr_bencFree( &response );
1853    }
1854    else if( methods[i].immediate )
1855    {
1856        int64_t tag;
1857        tr_benc response;
1858        tr_benc * args_out;
1859        struct evbuffer * buf;
1860
1861        tr_bencInitDict( &response, 3 );
1862        args_out = tr_bencDictAddDict( &response, "arguments", 0 );
1863        result = (*methods[i].func)( session, args_in, args_out, NULL );
1864        if( result == NULL )
1865            result = "success";
1866        tr_bencDictAddStr( &response, "result", result );
1867        if( tr_bencDictFindInt( request, "tag", &tag ) )
1868            tr_bencDictAddInt( &response, "tag", tag );
1869
1870        buf = tr_bencToBuf( &response, TR_FMT_JSON_LEAN );
1871        (*callback)( session, buf, callback_user_data );
1872        evbuffer_free( buf );
1873
1874        tr_bencFree( &response );
1875    }
1876    else
1877    {
1878        int64_t tag;
1879        struct tr_rpc_idle_data * data = tr_new0( struct tr_rpc_idle_data, 1 );
1880        data->session = session;
1881        data->response = tr_new0( tr_benc, 1 );
1882        tr_bencInitDict( data->response, 3 );
1883        if( tr_bencDictFindInt( request, "tag", &tag ) )
1884            tr_bencDictAddInt( data->response, "tag", tag );
1885        data->args_out = tr_bencDictAddDict( data->response, "arguments", 0 );
1886        data->callback = callback;
1887        data->callback_user_data = callback_user_data;
1888        (*methods[i].func)( session, args_in, data->args_out, data );
1889    }
1890}
1891
1892void
1893tr_rpc_request_exec_json( tr_session            * session,
1894                          const void            * request_json,
1895                          int                     request_len,
1896                          tr_rpc_response_func    callback,
1897                          void                  * callback_user_data )
1898{
1899    tr_benc top;
1900    int have_content;
1901
1902    if( request_len < 0 )
1903        request_len = strlen( request_json );
1904
1905    have_content = !tr_jsonParse( "rpc", request_json, request_len, &top, NULL );
1906    request_exec( session, have_content ? &top : NULL, callback, callback_user_data );
1907
1908    if( have_content )
1909        tr_bencFree( &top );
1910}
1911
1912/**
1913 * Munge the URI into a usable form.
1914 *
1915 * We have very loose typing on this to make the URIs as simple as possible:
1916 * - anything not a 'tag' or 'method' is automatically in 'arguments'
1917 * - values that are all-digits are numbers
1918 * - values that are all-digits or commas are number lists
1919 * - all other values are strings
1920 */
1921void
1922tr_rpc_parse_list_str( tr_benc     * setme,
1923                       const char  * str,
1924                       int           len )
1925
1926{
1927    int valueCount;
1928    int * values = tr_parseNumberRange( str, len, &valueCount );
1929
1930    if( valueCount == 0 )
1931        tr_bencInitStr( setme, str, len );
1932    else if( valueCount == 1 )
1933        tr_bencInitInt( setme, values[0] );
1934    else {
1935        int i;
1936        tr_bencInitList( setme, valueCount );
1937        for( i=0; i<valueCount; ++i )
1938            tr_bencListAddInt( setme, values[i] );
1939    }
1940
1941    tr_free( values );
1942}
1943
1944void
1945tr_rpc_request_exec_uri( tr_session           * session,
1946                         const void           * request_uri,
1947                         int                    request_len,
1948                         tr_rpc_response_func   callback,
1949                         void                 * callback_user_data )
1950{
1951    tr_benc      top, * args;
1952    char *       request = tr_strndup( request_uri, request_len );
1953    const char * pch;
1954
1955    tr_bencInitDict( &top, 3 );
1956    args = tr_bencDictAddDict( &top, "arguments", 0 );
1957
1958    pch = strchr( request, '?' );
1959    if( !pch ) pch = request;
1960    while( pch )
1961    {
1962        const char * delim = strchr( pch, '=' );
1963        const char * next = strchr( pch, '&' );
1964        if( delim )
1965        {
1966            char *    key = tr_strndup( pch, delim - pch );
1967            int       isArg = strcmp( key, "method" ) && strcmp( key, "tag" );
1968            tr_benc * parent = isArg ? args : &top;
1969            tr_rpc_parse_list_str( tr_bencDictAdd( parent, key ),
1970                                  delim + 1,
1971                                  next ? (size_t)(
1972                                       next -
1973                                      ( delim + 1 ) ) : strlen( delim + 1 ) );
1974            tr_free( key );
1975        }
1976        pch = next ? next + 1 : NULL;
1977    }
1978
1979    request_exec( session, &top, callback, callback_user_data );
1980
1981    /* cleanup */
1982    tr_bencFree( &top );
1983    tr_free( request );
1984}
Note: See TracBrowser for help on using the repository browser.