source: trunk/libtransmission/rpcimpl.c @ 13807

Last change on this file since 13807 was 13807, checked in by jordan, 8 years ago

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