/* * This file Copyright (C) Mnemosyne LLC * * This file is licensed by the GPL version 2. Works owned by the * Transmission project are granted a special exemption to clause 2 (b) * so that the bulk of its code can remain under the MIT license. * This exemption does not extend to derived works not owned by * the Transmission project. * * $Id$ */ #include #include /* isdigit */ #include #include /* strtol */ #include /* strcmp */ #include /* unlink */ #ifdef HAVE_ZLIB #include #endif #include #include "transmission.h" #include "bencode.h" #include "completion.h" #include "fdlimit.h" #include "json.h" #include "rpcimpl.h" #include "session.h" #include "torrent.h" #include "utils.h" #include "version.h" #include "web.h" #define RPC_VERSION 14 #define RPC_VERSION_MIN 1 #define RECENTLY_ACTIVE_SECONDS 60 #define TR_N_ELEMENTS(ary)(sizeof (ary) / sizeof (*ary)) #if 0 #define dbgmsg(fmt, ...) \ do { \ fprintf (stderr, "%s:%d"#fmt, __FILE__, __LINE__, __VA_ARGS__); \ fprintf (stderr, "\n"); \ } while (0) #else #define dbgmsg(...) \ do { \ if (tr_deepLoggingIsActive ()) \ tr_deepLog (__FILE__, __LINE__, "RPC", __VA_ARGS__); \ } while (0) #endif /*** **** ***/ static tr_rpc_callback_status notify (tr_session * session, int type, tr_torrent * tor) { tr_rpc_callback_status status = 0; if (session->rpc_func) status = session->rpc_func (session, type, tor, session->rpc_func_user_data); return status; } /*** **** ***/ /* For functions that can't be immediately executed, like torrentAdd, * this is the callback data used to pass a response to the caller * when the task is complete */ struct tr_rpc_idle_data { tr_session * session; tr_benc * response; tr_benc * args_out; tr_rpc_response_func callback; void * callback_user_data; }; static void tr_idle_function_done (struct tr_rpc_idle_data * data, const char * result) { struct evbuffer * buf; if (result == NULL) result = "success"; tr_bencDictAddStr (data->response, "result", result); buf = tr_bencToBuf (data->response, TR_FMT_JSON_LEAN); (*data->callback)(data->session, buf, data->callback_user_data); evbuffer_free (buf); tr_bencFree (data->response); tr_free (data->response); tr_free (data); } /*** **** ***/ static tr_torrent ** getTorrents (tr_session * session, tr_benc * args, int * setmeCount) { int torrentCount = 0; int64_t id; tr_torrent ** torrents = NULL; tr_benc * ids; const char * str; if (tr_bencDictFindList (args, "ids", &ids)) { int i; const int n = tr_bencListSize (ids); torrents = tr_new0 (tr_torrent *, n); for (i = 0; i < n; ++i) { tr_torrent * tor = NULL; tr_benc * node = tr_bencListChild (ids, i); const char * str; if (tr_bencGetInt (node, &id)) tor = tr_torrentFindFromId (session, id); else if (tr_bencGetStr (node, &str)) tor = tr_torrentFindFromHashString (session, str); if (tor) torrents[torrentCount++] = tor; } } else if (tr_bencDictFindInt (args, "ids", &id) || tr_bencDictFindInt (args, "id", &id)) { tr_torrent * tor; torrents = tr_new0 (tr_torrent *, 1); if ((tor = tr_torrentFindFromId (session, id))) torrents[torrentCount++] = tor; } else if (tr_bencDictFindStr (args, "ids", &str)) { if (!strcmp (str, "recently-active")) { tr_torrent * tor = NULL; const time_t now = tr_time (); const time_t window = RECENTLY_ACTIVE_SECONDS; const int n = tr_sessionCountTorrents (session); torrents = tr_new0 (tr_torrent *, n); while ((tor = tr_torrentNext (session, tor))) if (tor->anyDate >= now - window) torrents[torrentCount++] = tor; } else { tr_torrent * tor; torrents = tr_new0 (tr_torrent *, 1); if ((tor = tr_torrentFindFromHashString (session, str))) torrents[torrentCount++] = tor; } } else /* all of them */ { tr_torrent * tor = NULL; const int n = tr_sessionCountTorrents (session); torrents = tr_new0 (tr_torrent *, n); while ((tor = tr_torrentNext (session, tor))) torrents[torrentCount++] = tor; } *setmeCount = torrentCount; return torrents; } static void notifyBatchQueueChange (tr_session * session, tr_torrent ** torrents, int n) { int i; for (i=0; iisRunning) { tr_torrentStart (tor); notify (session, TR_RPC_TORRENT_STARTED, tor); } } tr_free (torrents); return NULL; } static const char* torrentStartNow (tr_session * session, tr_benc * args_in, tr_benc * args_out UNUSED, struct tr_rpc_idle_data * idle_data UNUSED) { int i, torrentCount; tr_torrent ** torrents = getTorrents (session, args_in, &torrentCount); assert (idle_data == NULL); for (i = 0; i < torrentCount; ++i) { tr_torrent * tor = torrents[i]; if (!tor->isRunning) { tr_torrentStartNow (tor); notify (session, TR_RPC_TORRENT_STARTED, tor); } } tr_free (torrents); return NULL; } static const char* torrentStop (tr_session * session, tr_benc * args_in, tr_benc * args_out UNUSED, struct tr_rpc_idle_data * idle_data UNUSED) { int i, torrentCount; tr_torrent ** torrents = getTorrents (session, args_in, &torrentCount); assert (idle_data == NULL); for (i = 0; i < torrentCount; ++i) { tr_torrent * tor = torrents[i]; if (tor->isRunning || tr_torrentIsQueued (tor)) { tor->isStopping = true; notify (session, TR_RPC_TORRENT_STOPPED, tor); } } tr_free (torrents); return NULL; } static const char* torrentRemove (tr_session * session, tr_benc * args_in, tr_benc * args_out UNUSED, struct tr_rpc_idle_data * idle_data UNUSED) { int i; int torrentCount; tr_rpc_callback_type type; bool deleteFlag = false; tr_torrent ** torrents = getTorrents (session, args_in, &torrentCount); assert (idle_data == NULL); tr_bencDictFindBool (args_in, "delete-local-data", &deleteFlag); type = deleteFlag ? TR_RPC_TORRENT_TRASHING : TR_RPC_TORRENT_REMOVING; for (i=0; ifileCount; ++i) { const tr_file * file = &info->files[i]; tr_benc * d = tr_bencListAddDict (list, 3); tr_bencDictAddInt (d, "bytesCompleted", files[i].bytesCompleted); tr_bencDictAddInt (d, "priority", file->priority); tr_bencDictAddBool (d, "wanted", !file->dnd); } tr_torrentFilesFree (files, n); } static void addFiles (const tr_torrent * tor, tr_benc * list) { tr_file_index_t i; tr_file_index_t n; const tr_info * info = tr_torrentInfo (tor); tr_file_stat * files = tr_torrentFiles (tor, &n); for (i = 0; i < info->fileCount; ++i) { const tr_file * file = &info->files[i]; tr_benc * d = tr_bencListAddDict (list, 3); tr_bencDictAddInt (d, "bytesCompleted", files[i].bytesCompleted); tr_bencDictAddInt (d, "length", file->length); tr_bencDictAddStr (d, "name", file->name); } tr_torrentFilesFree (files, n); } static void addWebseeds (const tr_info * info, tr_benc * webseeds) { int i; for (i = 0; i < info->webseedCount; ++i) tr_bencListAddStr (webseeds, info->webseeds[i]); } static void addTrackers (const tr_info * info, tr_benc * trackers) { int i; for (i = 0; i < info->trackerCount; ++i) { const tr_tracker_info * t = &info->trackers[i]; tr_benc * d = tr_bencListAddDict (trackers, 4); tr_bencDictAddStr (d, "announce", t->announce); tr_bencDictAddInt (d, "id", t->id); tr_bencDictAddStr (d, "scrape", t->scrape); tr_bencDictAddInt (d, "tier", t->tier); } } static void addTrackerStats (const tr_tracker_stat * st, int n, tr_benc * list) { int i; for (i=0; iannounce); tr_bencDictAddInt (d, "announceState", s->announceState); tr_bencDictAddInt (d, "downloadCount", s->downloadCount); tr_bencDictAddBool (d, "hasAnnounced", s->hasAnnounced); tr_bencDictAddBool (d, "hasScraped", s->hasScraped); tr_bencDictAddStr (d, "host", s->host); tr_bencDictAddInt (d, "id", s->id); tr_bencDictAddBool (d, "isBackup", s->isBackup); tr_bencDictAddInt (d, "lastAnnouncePeerCount", s->lastAnnouncePeerCount); tr_bencDictAddStr (d, "lastAnnounceResult", s->lastAnnounceResult); tr_bencDictAddInt (d, "lastAnnounceStartTime", s->lastAnnounceStartTime); tr_bencDictAddBool (d, "lastAnnounceSucceeded", s->lastAnnounceSucceeded); tr_bencDictAddInt (d, "lastAnnounceTime", s->lastAnnounceTime); tr_bencDictAddBool (d, "lastAnnounceTimedOut", s->lastAnnounceTimedOut); tr_bencDictAddStr (d, "lastScrapeResult", s->lastScrapeResult); tr_bencDictAddInt (d, "lastScrapeStartTime", s->lastScrapeStartTime); tr_bencDictAddBool (d, "lastScrapeSucceeded", s->lastScrapeSucceeded); tr_bencDictAddInt (d, "lastScrapeTime", s->lastScrapeTime); tr_bencDictAddInt (d, "lastScrapeTimedOut", s->lastScrapeTimedOut); tr_bencDictAddInt (d, "leecherCount", s->leecherCount); tr_bencDictAddInt (d, "nextAnnounceTime", s->nextAnnounceTime); tr_bencDictAddInt (d, "nextScrapeTime", s->nextScrapeTime); tr_bencDictAddStr (d, "scrape", s->scrape); tr_bencDictAddInt (d, "scrapeState", s->scrapeState); tr_bencDictAddInt (d, "seederCount", s->seederCount); tr_bencDictAddInt (d, "tier", s->tier); } } static void addPeers (const tr_torrent * tor, tr_benc * list) { int i; int peerCount; tr_peer_stat * peers = tr_torrentPeers (tor, &peerCount); tr_bencInitList (list, peerCount); for (i = 0; i < peerCount; ++i) { tr_benc * d = tr_bencListAddDict (list, 16); const tr_peer_stat * peer = peers + i; tr_bencDictAddStr (d, "address", peer->addr); tr_bencDictAddStr (d, "clientName", peer->client); tr_bencDictAddBool (d, "clientIsChoked", peer->clientIsChoked); tr_bencDictAddBool (d, "clientIsInterested", peer->clientIsInterested); tr_bencDictAddStr (d, "flagStr", peer->flagStr); tr_bencDictAddBool (d, "isDownloadingFrom", peer->isDownloadingFrom); tr_bencDictAddBool (d, "isEncrypted", peer->isEncrypted); tr_bencDictAddBool (d, "isIncoming", peer->isIncoming); tr_bencDictAddBool (d, "isUploadingTo", peer->isUploadingTo); tr_bencDictAddBool (d, "isUTP", peer->isUTP); tr_bencDictAddBool (d, "peerIsChoked", peer->peerIsChoked); tr_bencDictAddBool (d, "peerIsInterested", peer->peerIsInterested); tr_bencDictAddInt (d, "port", peer->port); tr_bencDictAddReal (d, "progress", peer->progress); tr_bencDictAddInt (d, "rateToClient", toSpeedBytes (peer->rateToClient_KBps)); tr_bencDictAddInt (d, "rateToPeer", toSpeedBytes (peer->rateToPeer_KBps)); } tr_torrentPeersFree (peers, peerCount); } /* faster-than-strcmp () optimization. This is kind of clumsy, but addField () was in the profiler's top 10 list, and this makes it 4x faster... */ #define tr_streq(a,alen,b)((alen+1==sizeof (b)) && !memcmp (a,b,alen)) static void addField (const tr_torrent * const tor, const tr_info * const inf, const tr_stat * const st, tr_benc * const d, const char * const key) { const size_t keylen = strlen (key); if (tr_streq (key, keylen, "activityDate")) tr_bencDictAddInt (d, key, st->activityDate); else if (tr_streq (key, keylen, "addedDate")) tr_bencDictAddInt (d, key, st->addedDate); else if (tr_streq (key, keylen, "bandwidthPriority")) tr_bencDictAddInt (d, key, tr_torrentGetPriority (tor)); else if (tr_streq (key, keylen, "comment")) tr_bencDictAddStr (d, key, inf->comment ? inf->comment : ""); else if (tr_streq (key, keylen, "corruptEver")) tr_bencDictAddInt (d, key, st->corruptEver); else if (tr_streq (key, keylen, "creator")) tr_bencDictAddStr (d, key, inf->creator ? inf->creator : ""); else if (tr_streq (key, keylen, "dateCreated")) tr_bencDictAddInt (d, key, inf->dateCreated); else if (tr_streq (key, keylen, "desiredAvailable")) tr_bencDictAddInt (d, key, st->desiredAvailable); else if (tr_streq (key, keylen, "doneDate")) tr_bencDictAddInt (d, key, st->doneDate); else if (tr_streq (key, keylen, "downloadDir")) tr_bencDictAddStr (d, key, tr_torrentGetDownloadDir (tor)); else if (tr_streq (key, keylen, "downloadedEver")) tr_bencDictAddInt (d, key, st->downloadedEver); else if (tr_streq (key, keylen, "downloadLimit")) tr_bencDictAddInt (d, key, tr_torrentGetSpeedLimit_KBps (tor, TR_DOWN)); else if (tr_streq (key, keylen, "downloadLimited")) tr_bencDictAddBool (d, key, tr_torrentUsesSpeedLimit (tor, TR_DOWN)); else if (tr_streq (key, keylen, "error")) tr_bencDictAddInt (d, key, st->error); else if (tr_streq (key, keylen, "errorString")) tr_bencDictAddStr (d, key, st->errorString); else if (tr_streq (key, keylen, "eta")) tr_bencDictAddInt (d, key, st->eta); else if (tr_streq (key, keylen, "files")) addFiles (tor, tr_bencDictAddList (d, key, inf->fileCount)); else if (tr_streq (key, keylen, "fileStats")) addFileStats (tor, tr_bencDictAddList (d, key, inf->fileCount)); else if (tr_streq (key, keylen, "hashString")) tr_bencDictAddStr (d, key, tor->info.hashString); else if (tr_streq (key, keylen, "haveUnchecked")) tr_bencDictAddInt (d, key, st->haveUnchecked); else if (tr_streq (key, keylen, "haveValid")) tr_bencDictAddInt (d, key, st->haveValid); else if (tr_streq (key, keylen, "honorsSessionLimits")) tr_bencDictAddBool (d, key, tr_torrentUsesSessionLimits (tor)); else if (tr_streq (key, keylen, "id")) tr_bencDictAddInt (d, key, st->id); else if (tr_streq (key, keylen, "isFinished")) tr_bencDictAddBool (d, key, st->finished); else if (tr_streq (key, keylen, "isPrivate")) tr_bencDictAddBool (d, key, tr_torrentIsPrivate (tor)); else if (tr_streq (key, keylen, "isStalled")) tr_bencDictAddBool (d, key, st->isStalled); else if (tr_streq (key, keylen, "leftUntilDone")) tr_bencDictAddInt (d, key, st->leftUntilDone); else if (tr_streq (key, keylen, "manualAnnounceTime")) tr_bencDictAddInt (d, key, st->manualAnnounceTime); else if (tr_streq (key, keylen, "maxConnectedPeers")) tr_bencDictAddInt (d, key, tr_torrentGetPeerLimit (tor)); else if (tr_streq (key, keylen, "magnetLink")) { char * str = tr_torrentGetMagnetLink (tor); tr_bencDictAddStr (d, key, str); tr_free (str); } else if (tr_streq (key, keylen, "metadataPercentComplete")) tr_bencDictAddReal (d, key, st->metadataPercentComplete); else if (tr_streq (key, keylen, "name")) tr_bencDictAddStr (d, key, tr_torrentName (tor)); else if (tr_streq (key, keylen, "percentDone")) tr_bencDictAddReal (d, key, st->percentDone); else if (tr_streq (key, keylen, "peer-limit")) tr_bencDictAddInt (d, key, tr_torrentGetPeerLimit (tor)); else if (tr_streq (key, keylen, "peers")) addPeers (tor, tr_bencDictAdd (d, key)); else if (tr_streq (key, keylen, "peersConnected")) tr_bencDictAddInt (d, key, st->peersConnected); else if (tr_streq (key, keylen, "peersFrom")) { tr_benc * tmp = tr_bencDictAddDict (d, key, 7); const int * f = st->peersFrom; tr_bencDictAddInt (tmp, "fromCache", f[TR_PEER_FROM_RESUME]); tr_bencDictAddInt (tmp, "fromDht", f[TR_PEER_FROM_DHT]); tr_bencDictAddInt (tmp, "fromIncoming", f[TR_PEER_FROM_INCOMING]); tr_bencDictAddInt (tmp, "fromLpd", f[TR_PEER_FROM_LPD]); tr_bencDictAddInt (tmp, "fromLtep", f[TR_PEER_FROM_LTEP]); tr_bencDictAddInt (tmp, "fromPex", f[TR_PEER_FROM_PEX]); tr_bencDictAddInt (tmp, "fromTracker", f[TR_PEER_FROM_TRACKER]); } else if (tr_streq (key, keylen, "peersGettingFromUs")) tr_bencDictAddInt (d, key, st->peersGettingFromUs); else if (tr_streq (key, keylen, "peersSendingToUs")) tr_bencDictAddInt (d, key, st->peersSendingToUs); else if (tr_streq (key, keylen, "pieces")) { if (tr_torrentHasMetadata (tor)) { size_t byte_count = 0; void * bytes = tr_cpCreatePieceBitfield (&tor->completion, &byte_count); char * str = tr_base64_encode (bytes, byte_count, NULL); tr_bencDictAddStr (d, key, str!=NULL ? str : ""); tr_free (str); tr_free (bytes); } else { tr_bencDictAddStr (d, key, ""); } } else if (tr_streq (key, keylen, "pieceCount")) tr_bencDictAddInt (d, key, inf->pieceCount); else if (tr_streq (key, keylen, "pieceSize")) tr_bencDictAddInt (d, key, inf->pieceSize); else if (tr_streq (key, keylen, "priorities")) { tr_file_index_t i; tr_benc * p = tr_bencDictAddList (d, key, inf->fileCount); for (i = 0; i < inf->fileCount; ++i) tr_bencListAddInt (p, inf->files[i].priority); } else if (tr_streq (key, keylen, "queuePosition")) tr_bencDictAddInt (d, key, st->queuePosition); else if (tr_streq (key, keylen, "rateDownload")) tr_bencDictAddInt (d, key, toSpeedBytes (st->pieceDownloadSpeed_KBps)); else if (tr_streq (key, keylen, "rateUpload")) tr_bencDictAddInt (d, key, toSpeedBytes (st->pieceUploadSpeed_KBps)); else if (tr_streq (key, keylen, "recheckProgress")) tr_bencDictAddReal (d, key, st->recheckProgress); else if (tr_streq (key, keylen, "seedIdleLimit")) tr_bencDictAddInt (d, key, tr_torrentGetIdleLimit (tor)); else if (tr_streq (key, keylen, "seedIdleMode")) tr_bencDictAddInt (d, key, tr_torrentGetIdleMode (tor)); else if (tr_streq (key, keylen, "seedRatioLimit")) tr_bencDictAddReal (d, key, tr_torrentGetRatioLimit (tor)); else if (tr_streq (key, keylen, "seedRatioMode")) tr_bencDictAddInt (d, key, tr_torrentGetRatioMode (tor)); else if (tr_streq (key, keylen, "sizeWhenDone")) tr_bencDictAddInt (d, key, st->sizeWhenDone); else if (tr_streq (key, keylen, "startDate")) tr_bencDictAddInt (d, key, st->startDate); else if (tr_streq (key, keylen, "status")) tr_bencDictAddInt (d, key, st->activity); else if (tr_streq (key, keylen, "secondsDownloading")) tr_bencDictAddInt (d, key, st->secondsDownloading); else if (tr_streq (key, keylen, "secondsSeeding")) tr_bencDictAddInt (d, key, st->secondsSeeding); else if (tr_streq (key, keylen, "trackers")) addTrackers (inf, tr_bencDictAddList (d, key, inf->trackerCount)); else if (tr_streq (key, keylen, "trackerStats")) { int n; tr_tracker_stat * s = tr_torrentTrackers (tor, &n); addTrackerStats (s, n, tr_bencDictAddList (d, key, n)); tr_torrentTrackersFree (s, n); } else if (tr_streq (key, keylen, "torrentFile")) tr_bencDictAddStr (d, key, inf->torrent); else if (tr_streq (key, keylen, "totalSize")) tr_bencDictAddInt (d, key, inf->totalSize); else if (tr_streq (key, keylen, "uploadedEver")) tr_bencDictAddInt (d, key, st->uploadedEver); else if (tr_streq (key, keylen, "uploadLimit")) tr_bencDictAddInt (d, key, tr_torrentGetSpeedLimit_KBps (tor, TR_UP)); else if (tr_streq (key, keylen, "uploadLimited")) tr_bencDictAddBool (d, key, tr_torrentUsesSpeedLimit (tor, TR_UP)); else if (tr_streq (key, keylen, "uploadRatio")) tr_bencDictAddReal (d, key, st->ratio); else if (tr_streq (key, keylen, "wanted")) { tr_file_index_t i; tr_benc * w = tr_bencDictAddList (d, key, inf->fileCount); for (i = 0; i < inf->fileCount; ++i) tr_bencListAddInt (w, inf->files[i].dnd ? 0 : 1); } else if (tr_streq (key, keylen, "webseeds")) addWebseeds (inf, tr_bencDictAddList (d, key, inf->webseedCount)); else if (tr_streq (key, keylen, "webseedsSendingToUs")) tr_bencDictAddInt (d, key, st->webseedsSendingToUs); } static void addInfo (const tr_torrent * tor, tr_benc * d, tr_benc * fields) { const char * str; const int n = tr_bencListSize (fields); tr_bencInitDict (d, n); if (n > 0) { int i; const tr_info const * inf = tr_torrentInfo (tor); const tr_stat const * st = tr_torrentStat ((tr_torrent*)tor); for (i=0; iremovedTorrents, n++))) { int64_t intVal; if (tr_bencDictFindInt (d, "date", &intVal) && (intVal >= now - interval)) { tr_bencDictFindInt (d, "id", &intVal); tr_bencListAddInt (removed_out, intVal); } } } if (!tr_bencDictFindList (args_in, "fields", &fields)) msg = "no fields specified"; else for (i = 0; i < torrentCount; ++i) addInfo (torrents[i], tr_bencListAdd (list), fields); tr_free (torrents); return msg; } /*** **** ***/ static const char* setFilePriorities (tr_torrent * tor, int priority, tr_benc * list) { int i; int64_t tmp; int fileCount = 0; const int n = tr_bencListSize (list); const char * errmsg = NULL; tr_file_index_t * files = tr_new0 (tr_file_index_t, tor->info.fileCount); if (n) { for (i = 0; i < n; ++i) { if (tr_bencGetInt (tr_bencListChild (list, i), &tmp)) { if (0 <= tmp && tmp < tor->info.fileCount) { files[fileCount++] = tmp; } else { errmsg = "file index out of range"; } } } } else /* if empty set, apply to all */ { tr_file_index_t t; for (t = 0; t < tor->info.fileCount; ++t) files[fileCount++] = t; } if (fileCount) tr_torrentSetFilePriorities (tor, files, fileCount, priority); tr_free (files); return errmsg; } static const char* setFileDLs (tr_torrent * tor, int do_download, tr_benc * list) { int i; int64_t tmp; int fileCount = 0; const int n = tr_bencListSize (list); const char * errmsg = NULL; tr_file_index_t * files = tr_new0 (tr_file_index_t, tor->info.fileCount); if (n) /* if argument list, process them */ { for (i = 0; i < n; ++i) { if (tr_bencGetInt (tr_bencListChild (list, i), &tmp)) { if (0 <= tmp && tmp < tor->info.fileCount) { files[fileCount++] = tmp; } else { errmsg = "file index out of range"; } } } } else /* if empty set, apply to all */ { tr_file_index_t t; for (t = 0; t < tor->info.fileCount; ++t) files[fileCount++] = t; } if (fileCount) tr_torrentSetFileDLs (tor, files, fileCount, do_download); tr_free (files); return errmsg; } static bool findAnnounceUrl (const tr_tracker_info * t, int n, const char * url, int * pos) { int i; bool found = false; for (i=0; itrackerCount; trackers = tr_new0 (tr_tracker_info, n + tr_bencListSize (urls)); tier = copyTrackers (trackers, inf->trackers, n); /* and add the new ones */ i = 0; while ((val = tr_bencListChild (urls, i++))) { const char * announce = NULL; if ( tr_bencGetStr (val, &announce) && tr_urlIsValidTracker (announce) && !findAnnounceUrl (trackers, n, announce, NULL)) { trackers[n].tier = ++tier; /* add a new tier */ trackers[n].announce = tr_strdup (announce); ++n; changed = true; } } if (!changed) errmsg = "invalid argument"; else if (!tr_torrentSetAnnounceList (tor, trackers, n)) errmsg = "error setting announce list"; freeTrackers (trackers, n); return errmsg; } static const char* replaceTrackers (tr_torrent * tor, tr_benc * urls) { int i; tr_benc * pair[2]; tr_tracker_info * trackers; bool changed = false; const tr_info * inf = tr_torrentInfo (tor); const int n = inf->trackerCount; const char * errmsg = NULL; /* make a working copy of the existing announce list */ trackers = tr_new0 (tr_tracker_info, n); copyTrackers (trackers, inf->trackers, n); /* make the substitutions... */ i = 0; while (((pair[0] = tr_bencListChild (urls,i))) && ((pair[1] = tr_bencListChild (urls,i+1)))) { int64_t pos; const char * newval; if ( tr_bencGetInt (pair[0], &pos) && tr_bencGetStr (pair[1], &newval) && tr_urlIsValidTracker (newval) && pos < n && pos >= 0) { tr_free (trackers[pos].announce); trackers[pos].announce = tr_strdup (newval); changed = true; } i += 2; } if (!changed) errmsg = "invalid argument"; else if (!tr_torrentSetAnnounceList (tor, trackers, n)) errmsg = "error setting announce list"; freeTrackers (trackers, n); return errmsg; } static const char* removeTrackers (tr_torrent * tor, tr_benc * ids) { int i; int n; int t = 0; int dup = -1; int * tids; tr_benc * val; tr_tracker_info * trackers; bool changed = false; const tr_info * inf = tr_torrentInfo (tor); const char * errmsg = NULL; /* make a working copy of the existing announce list */ n = inf->trackerCount; tids = tr_new0 (int, n); trackers = tr_new0 (tr_tracker_info, n); copyTrackers (trackers, inf->trackers, n); /* remove the ones specified in the urls list */ i = 0; while ((val = tr_bencListChild (ids, i++))) { int64_t pos; if ( tr_bencGetInt (val, &pos) && pos < n && pos >= 0) tids[t++] = pos; } /* sort trackerIds and remove from largest to smallest so there is no need to recacluate array indicies */ qsort (tids, t, sizeof (int), compareInt); while (t--) { /* check for duplicates */ if (tids[t] == dup) continue; tr_removeElementFromArray (trackers, tids[t], sizeof (tr_tracker_info), n--); dup = tids[t]; changed = true; } if (!changed) errmsg = "invalid argument"; else if (!tr_torrentSetAnnounceList (tor, trackers, n)) errmsg = "error setting announce list"; freeTrackers (trackers, n); tr_free (tids); return errmsg; } static const char* torrentSet (tr_session * session, tr_benc * args_in, tr_benc * args_out UNUSED, struct tr_rpc_idle_data * idle_data UNUSED) { const char * errmsg = NULL; int i, torrentCount; tr_torrent ** torrents = getTorrents (session, args_in, &torrentCount); assert (idle_data == NULL); for (i = 0; i < torrentCount; ++i) { int64_t tmp; double d; tr_benc * files; tr_benc * trackers; bool boolVal; tr_torrent * tor = torrents[i]; if (tr_bencDictFindInt (args_in, "bandwidthPriority", &tmp)) if (tr_isPriority (tmp)) tr_torrentSetPriority (tor, tmp); if (!errmsg && tr_bencDictFindList (args_in, "files-unwanted", &files)) errmsg = setFileDLs (tor, false, files); if (!errmsg && tr_bencDictFindList (args_in, "files-wanted", &files)) errmsg = setFileDLs (tor, true, files); if (tr_bencDictFindInt (args_in, "peer-limit", &tmp)) tr_torrentSetPeerLimit (tor, tmp); if (!errmsg && tr_bencDictFindList (args_in, "priority-high", &files)) errmsg = setFilePriorities (tor, TR_PRI_HIGH, files); if (!errmsg && tr_bencDictFindList (args_in, "priority-low", &files)) errmsg = setFilePriorities (tor, TR_PRI_LOW, files); if (!errmsg && tr_bencDictFindList (args_in, "priority-normal", &files)) errmsg = setFilePriorities (tor, TR_PRI_NORMAL, files); if (tr_bencDictFindInt (args_in, "downloadLimit", &tmp)) tr_torrentSetSpeedLimit_KBps (tor, TR_DOWN, tmp); if (tr_bencDictFindBool (args_in, "downloadLimited", &boolVal)) tr_torrentUseSpeedLimit (tor, TR_DOWN, boolVal); if (tr_bencDictFindBool (args_in, "honorsSessionLimits", &boolVal)) tr_torrentUseSessionLimits (tor, boolVal); if (tr_bencDictFindInt (args_in, "uploadLimit", &tmp)) tr_torrentSetSpeedLimit_KBps (tor, TR_UP, tmp); if (tr_bencDictFindBool (args_in, "uploadLimited", &boolVal)) tr_torrentUseSpeedLimit (tor, TR_UP, boolVal); if (tr_bencDictFindInt (args_in, "seedIdleLimit", &tmp)) tr_torrentSetIdleLimit (tor, tmp); if (tr_bencDictFindInt (args_in, "seedIdleMode", &tmp)) tr_torrentSetIdleMode (tor, tmp); if (tr_bencDictFindReal (args_in, "seedRatioLimit", &d)) tr_torrentSetRatioLimit (tor, d); if (tr_bencDictFindInt (args_in, "seedRatioMode", &tmp)) tr_torrentSetRatioMode (tor, tmp); if (tr_bencDictFindInt (args_in, "queuePosition", &tmp)) tr_torrentSetQueuePosition (tor, tmp); if (!errmsg && tr_bencDictFindList (args_in, "trackerAdd", &trackers)) errmsg = addTrackerUrls (tor, trackers); if (!errmsg && tr_bencDictFindList (args_in, "trackerRemove", &trackers)) errmsg = removeTrackers (tor, trackers); if (!errmsg && tr_bencDictFindList (args_in, "trackerReplace", &trackers)) errmsg = replaceTrackers (tor, trackers); notify (session, TR_RPC_TORRENT_CHANGED, tor); } tr_free (torrents); return errmsg; } static const char* torrentSetLocation (tr_session * session, tr_benc * args_in, tr_benc * args_out UNUSED, struct tr_rpc_idle_data * idle_data UNUSED) { const char * errmsg = NULL; const char * location = NULL; assert (idle_data == NULL); if (!tr_bencDictFindStr (args_in, "location", &location)) { errmsg = "no location"; } else { bool move = false; int i, torrentCount; tr_torrent ** torrents = getTorrents (session, args_in, &torrentCount); tr_bencDictFindBool (args_in, "move", &move); for (i=0; iargs_out, "port-is-open", isOpen); tr_snprintf (result, sizeof (result), "success"); } tr_idle_function_done (data, result); } static const char* portTest (tr_session * session, tr_benc * args_in UNUSED, tr_benc * args_out UNUSED, struct tr_rpc_idle_data * idle_data) { const int port = tr_sessionGetPeerPort (session); char * url = tr_strdup_printf ("http://portcheck.transmissionbt.com/%d", port); tr_webRun (session, url, NULL, NULL, portTested, idle_data); tr_free (url); return NULL; } /*** **** ***/ static void gotNewBlocklist (tr_session * session, bool did_connect UNUSED, bool did_timeout UNUSED, long response_code, const void * response, size_t response_byte_count, void * user_data) { char result[1024]; struct tr_rpc_idle_data * data = user_data; *result = '\0'; if (response_code != 200) { tr_snprintf (result, sizeof (result), "gotNewBlocklist: http error %ld: %s", response_code, tr_webGetResponseStr (response_code)); } else /* successfully fetched the blocklist... */ { int fd; int err; char * filename; z_stream stream; const char * configDir = tr_sessionGetConfigDir (session); const size_t buflen = 1024 * 128; /* 128 KiB buffer */ uint8_t * buf = tr_valloc (buflen); /* this is an odd Magic Number required by zlib to enable gz support. See zlib's inflateInit2 () documentation for a full description */ const int windowBits = 15 + 32; stream.zalloc = (alloc_func) Z_NULL; stream.zfree = (free_func) Z_NULL; stream.opaque = (voidpf) Z_NULL; stream.next_in = (void*) response; stream.avail_in = response_byte_count; inflateInit2 (&stream, windowBits); filename = tr_buildPath (configDir, "blocklist.tmp", NULL); fd = tr_open_file_for_writing (filename); if (fd < 0) tr_snprintf (result, sizeof (result), _("Couldn't save file \"%1$s\": %2$s"), filename, tr_strerror (errno)); for (;;) { stream.next_out = (void*) buf; stream.avail_out = buflen; err = inflate (&stream, Z_NO_FLUSH); if (stream.avail_out < buflen) { const int e = write (fd, buf, buflen - stream.avail_out); if (e < 0) { tr_snprintf (result, sizeof (result), _("Couldn't save file \"%1$s\": %2$s"), filename, tr_strerror (errno)); break; } } if (err != Z_OK) { if ((err != Z_STREAM_END) && (err != Z_DATA_ERROR)) tr_snprintf (result, sizeof (result), _("Error uncompressing blocklist: %s (%d)"), zError (err), err); break; } } inflateEnd (&stream); if (err == Z_DATA_ERROR) /* couldn't inflate it... it's probably already uncompressed */ if (write (fd, response, response_byte_count) < 0) tr_snprintf (result, sizeof (result), _("Couldn't save file \"%1$s\": %2$s"), filename, tr_strerror (errno)); if (*result) tr_err ("%s", result); else { /* feed it to the session and give the client a response */ const int rule_count = tr_blocklistSetContent (session, filename); tr_bencDictAddInt (data->args_out, "blocklist-size", rule_count); tr_snprintf (result, sizeof (result), "success"); } unlink (filename); tr_free (filename); tr_free (buf); } tr_idle_function_done (data, result); } static const char* blocklistUpdate (tr_session * session, tr_benc * args_in UNUSED, tr_benc * args_out UNUSED, struct tr_rpc_idle_data * idle_data) { tr_webRun (session, session->blocklist_url, NULL, NULL, gotNewBlocklist, idle_data); return NULL; } /*** **** ***/ static void addTorrentImpl (struct tr_rpc_idle_data * data, tr_ctor * ctor) { int err = 0; const char * result = NULL; tr_torrent * tor = tr_torrentNew (ctor, &err); tr_ctorFree (ctor); if (tor) { tr_benc fields; tr_bencInitList (&fields, 3); tr_bencListAddStr (&fields, "id"); tr_bencListAddStr (&fields, "name"); tr_bencListAddStr (&fields, "hashString"); addInfo (tor, tr_bencDictAdd (data->args_out, "torrent-added"), &fields); notify (data->session, TR_RPC_TORRENT_ADDED, tor); tr_bencFree (&fields); } else if (err == TR_PARSE_DUPLICATE) { result = "duplicate torrent"; } else if (err == TR_PARSE_ERR) { result = "invalid or corrupt torrent file"; } tr_idle_function_done (data, result); } struct add_torrent_idle_data { struct tr_rpc_idle_data * data; tr_ctor * ctor; }; static void gotMetadataFromURL (tr_session * session UNUSED, bool did_connect UNUSED, bool did_timeout UNUSED, long response_code, const void * response, size_t response_byte_count, void * user_data) { struct add_torrent_idle_data * data = user_data; dbgmsg ("torrentAdd: HTTP response code was %ld (%s); response length was %zu bytes", response_code, tr_webGetResponseStr (response_code), response_byte_count); if (response_code==200 || response_code==221) /* http or ftp success.. */ { tr_ctorSetMetainfo (data->ctor, response, response_byte_count); addTorrentImpl (data->data, data->ctor); } else { char result[1024]; tr_snprintf (result, sizeof (result), "gotMetadataFromURL: http error %ld: %s", response_code, tr_webGetResponseStr (response_code)); tr_idle_function_done (data->data, result); } tr_free (data); } static bool isCurlURL (const char * filename) { if (filename == NULL) return false; return !strncmp (filename, "ftp://", 6) || !strncmp (filename, "http://", 7) || !strncmp (filename, "https://", 8); } static tr_file_index_t* fileListFromList (tr_benc * list, tr_file_index_t * setmeCount) { size_t i; const size_t childCount = tr_bencListSize (list); tr_file_index_t n = 0; tr_file_index_t * files = tr_new0 (tr_file_index_t, childCount); for (i=0; idata = idle_data; d->ctor = ctor; tr_webRun (session, filename, NULL, cookies, gotMetadataFromURL, d); } else { char * fname = tr_strstrip (tr_strdup (filename)); if (fname == NULL) { int len; char * metainfo = tr_base64_decode (metainfo_base64, -1, &len); tr_ctorSetMetainfo (ctor, (uint8_t*)metainfo, len); tr_free (metainfo); } else if (!strncmp (fname, "magnet:?", 8)) { tr_ctorSetMetainfoFromMagnetLink (ctor, fname); } else { tr_ctorSetMetainfoFromFile (ctor, fname); } addTorrentImpl (idle_data, ctor); tr_free (fname); } } return NULL; } /*** **** ***/ static const char* sessionSet (tr_session * session, tr_benc * args_in, tr_benc * args_out UNUSED, struct tr_rpc_idle_data * idle_data UNUSED) { int64_t i; double d; bool boolVal; const char * str; assert (idle_data == NULL); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_MAX_CACHE_SIZE_MB, &i)) tr_sessionSetCacheLimit_MB (session, i); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_ALT_SPEED_UP_KBps, &i)) tr_sessionSetAltSpeed_KBps (session, TR_UP, i); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_ALT_SPEED_DOWN_KBps, &i)) tr_sessionSetAltSpeed_KBps (session, TR_DOWN, i); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_ALT_SPEED_ENABLED, &boolVal)) tr_sessionUseAltSpeed (session, boolVal); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_ALT_SPEED_TIME_BEGIN, &i)) tr_sessionSetAltSpeedBegin (session, i); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_ALT_SPEED_TIME_END, &i)) tr_sessionSetAltSpeedEnd (session, i); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_ALT_SPEED_TIME_DAY, &i)) tr_sessionSetAltSpeedDay (session, i); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_ALT_SPEED_TIME_ENABLED, &boolVal)) tr_sessionUseAltSpeedTime (session, boolVal); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_BLOCKLIST_ENABLED, &boolVal)) tr_blocklistSetEnabled (session, boolVal); if (tr_bencDictFindStr (args_in, TR_PREFS_KEY_BLOCKLIST_URL, &str)) tr_blocklistSetURL (session, str); if (tr_bencDictFindStr (args_in, TR_PREFS_KEY_DOWNLOAD_DIR, &str)) tr_sessionSetDownloadDir (session, str); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_QUEUE_STALLED_MINUTES, &i)) tr_sessionSetQueueStalledMinutes (session, i); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_QUEUE_STALLED_ENABLED, &boolVal)) tr_sessionSetQueueStalledEnabled (session, boolVal); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_DOWNLOAD_QUEUE_SIZE, &i)) tr_sessionSetQueueSize (session, TR_DOWN, i); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_DOWNLOAD_QUEUE_ENABLED, &boolVal)) tr_sessionSetQueueEnabled (session, TR_DOWN, boolVal); if (tr_bencDictFindStr (args_in, TR_PREFS_KEY_INCOMPLETE_DIR, &str)) tr_sessionSetIncompleteDir (session, str); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_INCOMPLETE_DIR_ENABLED, &boolVal)) tr_sessionSetIncompleteDirEnabled (session, boolVal); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_PEER_LIMIT_GLOBAL, &i)) tr_sessionSetPeerLimit (session, i); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_PEER_LIMIT_TORRENT, &i)) tr_sessionSetPeerLimitPerTorrent (session, i); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_PEX_ENABLED, &boolVal)) tr_sessionSetPexEnabled (session, boolVal); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_DHT_ENABLED, &boolVal)) tr_sessionSetDHTEnabled (session, boolVal); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_UTP_ENABLED, &boolVal)) tr_sessionSetUTPEnabled (session, boolVal); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_LPD_ENABLED, &boolVal)) tr_sessionSetLPDEnabled (session, boolVal); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_PEER_PORT_RANDOM_ON_START, &boolVal)) tr_sessionSetPeerPortRandomOnStart (session, boolVal); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_PEER_PORT, &i)) tr_sessionSetPeerPort (session, i); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_PORT_FORWARDING, &boolVal)) tr_sessionSetPortForwardingEnabled (session, boolVal); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_RENAME_PARTIAL_FILES, &boolVal)) tr_sessionSetIncompleteFileNamingEnabled (session, boolVal); if (tr_bencDictFindReal (args_in, "seedRatioLimit", &d)) tr_sessionSetRatioLimit (session, d); if (tr_bencDictFindBool (args_in, "seedRatioLimited", &boolVal)) tr_sessionSetRatioLimited (session, boolVal); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_IDLE_LIMIT, &i)) tr_sessionSetIdleLimit (session, i); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_IDLE_LIMIT_ENABLED, &boolVal)) tr_sessionSetIdleLimited (session, boolVal); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_START, &boolVal)) tr_sessionSetPaused (session, !boolVal); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_SEED_QUEUE_ENABLED, &boolVal)) tr_sessionSetQueueEnabled (session, TR_UP, boolVal); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_SEED_QUEUE_SIZE, &i)) tr_sessionSetQueueSize (session, TR_UP, i); if (tr_bencDictFindStr (args_in, TR_PREFS_KEY_SCRIPT_TORRENT_DONE_FILENAME, &str)) tr_sessionSetTorrentDoneScript (session, str); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_SCRIPT_TORRENT_DONE_ENABLED, &boolVal)) tr_sessionSetTorrentDoneScriptEnabled (session, boolVal); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_TRASH_ORIGINAL, &boolVal)) tr_sessionSetDeleteSource (session, boolVal); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_DSPEED_KBps, &i)) tr_sessionSetSpeedLimit_KBps (session, TR_DOWN, i); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_DSPEED_ENABLED, &boolVal)) tr_sessionLimitSpeed (session, TR_DOWN, boolVal); if (tr_bencDictFindInt (args_in, TR_PREFS_KEY_USPEED_KBps, &i)) tr_sessionSetSpeedLimit_KBps (session, TR_UP, i); if (tr_bencDictFindBool (args_in, TR_PREFS_KEY_USPEED_ENABLED, &boolVal)) tr_sessionLimitSpeed (session, TR_UP, boolVal); if (tr_bencDictFindStr (args_in, TR_PREFS_KEY_ENCRYPTION, &str)) { if (!strcmp (str, "required")) tr_sessionSetEncryption (session, TR_ENCRYPTION_REQUIRED); else if (!strcmp (str, "tolerated")) tr_sessionSetEncryption (session, TR_CLEAR_PREFERRED); else tr_sessionSetEncryption (session, TR_ENCRYPTION_PREFERRED); } notify (session, TR_RPC_SESSION_CHANGED, NULL); return NULL; } static const char* sessionStats (tr_session * session, tr_benc * args_in UNUSED, tr_benc * args_out, struct tr_rpc_idle_data * idle_data UNUSED) { int running = 0; int total = 0; tr_benc * d; tr_session_stats currentStats = { 0.0f, 0, 0, 0, 0, 0 }; tr_session_stats cumulativeStats = { 0.0f, 0, 0, 0, 0, 0 }; tr_torrent * tor = NULL; assert (idle_data == NULL); while ((tor = tr_torrentNext (session, tor))) { ++total; if (tor->isRunning) ++running; } tr_sessionGetStats (session, ¤tStats); tr_sessionGetCumulativeStats (session, &cumulativeStats); tr_bencDictAddInt (args_out, "activeTorrentCount", running); tr_bencDictAddReal (args_out, "downloadSpeed", tr_sessionGetPieceSpeed_Bps (session, TR_DOWN)); tr_bencDictAddInt (args_out, "pausedTorrentCount", total - running); tr_bencDictAddInt (args_out, "torrentCount", total); tr_bencDictAddReal (args_out, "uploadSpeed", tr_sessionGetPieceSpeed_Bps (session, TR_UP)); d = tr_bencDictAddDict (args_out, "cumulative-stats", 5); tr_bencDictAddInt (d, "downloadedBytes", cumulativeStats.downloadedBytes); tr_bencDictAddInt (d, "filesAdded", cumulativeStats.filesAdded); tr_bencDictAddInt (d, "secondsActive", cumulativeStats.secondsActive); tr_bencDictAddInt (d, "sessionCount", cumulativeStats.sessionCount); tr_bencDictAddInt (d, "uploadedBytes", cumulativeStats.uploadedBytes); d = tr_bencDictAddDict (args_out, "current-stats", 5); tr_bencDictAddInt (d, "downloadedBytes", currentStats.downloadedBytes); tr_bencDictAddInt (d, "filesAdded", currentStats.filesAdded); tr_bencDictAddInt (d, "secondsActive", currentStats.secondsActive); tr_bencDictAddInt (d, "sessionCount", currentStats.sessionCount); tr_bencDictAddInt (d, "uploadedBytes", currentStats.uploadedBytes); return NULL; } static const char* sessionGet (tr_session * s, tr_benc * args_in UNUSED, tr_benc * args_out, struct tr_rpc_idle_data * idle_data UNUSED) { const char * str; tr_benc * d = args_out; assert (idle_data == NULL); tr_bencDictAddInt (d, TR_PREFS_KEY_ALT_SPEED_UP_KBps, tr_sessionGetAltSpeed_KBps (s,TR_UP)); tr_bencDictAddInt (d, TR_PREFS_KEY_ALT_SPEED_DOWN_KBps, tr_sessionGetAltSpeed_KBps (s,TR_DOWN)); tr_bencDictAddBool (d, TR_PREFS_KEY_ALT_SPEED_ENABLED, tr_sessionUsesAltSpeed (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_ALT_SPEED_TIME_BEGIN, tr_sessionGetAltSpeedBegin (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_ALT_SPEED_TIME_END,tr_sessionGetAltSpeedEnd (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_ALT_SPEED_TIME_DAY,tr_sessionGetAltSpeedDay (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_ALT_SPEED_TIME_ENABLED, tr_sessionUsesAltSpeedTime (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_BLOCKLIST_ENABLED, tr_blocklistIsEnabled (s)); tr_bencDictAddStr (d, TR_PREFS_KEY_BLOCKLIST_URL, tr_blocklistGetURL (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_MAX_CACHE_SIZE_MB, tr_sessionGetCacheLimit_MB (s)); tr_bencDictAddInt (d, "blocklist-size", tr_blocklistGetRuleCount (s)); tr_bencDictAddStr (d, "config-dir", tr_sessionGetConfigDir (s)); tr_bencDictAddStr (d, TR_PREFS_KEY_DOWNLOAD_DIR, tr_sessionGetDownloadDir (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_DOWNLOAD_QUEUE_ENABLED, tr_sessionGetQueueEnabled (s, TR_DOWN)); tr_bencDictAddInt (d, TR_PREFS_KEY_DOWNLOAD_QUEUE_SIZE, tr_sessionGetQueueSize (s, TR_DOWN)); tr_bencDictAddInt (d, "download-dir-free-space", tr_sessionGetDownloadDirFreeSpace (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_PEER_LIMIT_GLOBAL, tr_sessionGetPeerLimit (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_PEER_LIMIT_TORRENT, tr_sessionGetPeerLimitPerTorrent (s)); tr_bencDictAddStr (d, TR_PREFS_KEY_INCOMPLETE_DIR, tr_sessionGetIncompleteDir (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_INCOMPLETE_DIR_ENABLED, tr_sessionIsIncompleteDirEnabled (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_PEX_ENABLED, tr_sessionIsPexEnabled (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_UTP_ENABLED, tr_sessionIsUTPEnabled (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_DHT_ENABLED, tr_sessionIsDHTEnabled (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_LPD_ENABLED, tr_sessionIsLPDEnabled (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_PEER_PORT, tr_sessionGetPeerPort (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_PEER_PORT_RANDOM_ON_START, tr_sessionGetPeerPortRandomOnStart (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_PORT_FORWARDING, tr_sessionIsPortForwardingEnabled (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_RENAME_PARTIAL_FILES, tr_sessionIsIncompleteFileNamingEnabled (s)); tr_bencDictAddInt (d, "rpc-version", RPC_VERSION); tr_bencDictAddInt (d, "rpc-version-minimum", RPC_VERSION_MIN); tr_bencDictAddReal (d, "seedRatioLimit", tr_sessionGetRatioLimit (s)); tr_bencDictAddBool (d, "seedRatioLimited", tr_sessionIsRatioLimited (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_IDLE_LIMIT, tr_sessionGetIdleLimit (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_IDLE_LIMIT_ENABLED, tr_sessionIsIdleLimited (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_SEED_QUEUE_ENABLED, tr_sessionGetQueueEnabled (s, TR_UP)); tr_bencDictAddInt (d, TR_PREFS_KEY_SEED_QUEUE_SIZE, tr_sessionGetQueueSize (s, TR_UP)); tr_bencDictAddBool (d, TR_PREFS_KEY_START, !tr_sessionGetPaused (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_TRASH_ORIGINAL, tr_sessionGetDeleteSource (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_USPEED_KBps, tr_sessionGetSpeedLimit_KBps (s, TR_UP)); tr_bencDictAddBool (d, TR_PREFS_KEY_USPEED_ENABLED, tr_sessionIsSpeedLimited (s, TR_UP)); tr_bencDictAddInt (d, TR_PREFS_KEY_DSPEED_KBps, tr_sessionGetSpeedLimit_KBps (s, TR_DOWN)); tr_bencDictAddBool (d, TR_PREFS_KEY_DSPEED_ENABLED, tr_sessionIsSpeedLimited (s, TR_DOWN)); tr_bencDictAddStr (d, TR_PREFS_KEY_SCRIPT_TORRENT_DONE_FILENAME, tr_sessionGetTorrentDoneScript (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_SCRIPT_TORRENT_DONE_ENABLED, tr_sessionIsTorrentDoneScriptEnabled (s)); tr_bencDictAddInt (d, TR_PREFS_KEY_QUEUE_STALLED_MINUTES, tr_sessionGetQueueStalledMinutes (s)); tr_bencDictAddBool (d, TR_PREFS_KEY_QUEUE_STALLED_ENABLED, tr_sessionGetQueueStalledEnabled (s)); tr_formatter_get_units (tr_bencDictAddDict (d, "units", 0)); tr_bencDictAddStr (d, "version", LONG_VERSION_STRING); switch (tr_sessionGetEncryption (s)) { case TR_CLEAR_PREFERRED: str = "tolerated"; break; case TR_ENCRYPTION_REQUIRED: str = "required"; break; default: str = "preferred"; break; } tr_bencDictAddStr (d, TR_PREFS_KEY_ENCRYPTION, str); return NULL; } /*** **** ***/ static const char* sessionClose (tr_session * session, tr_benc * args_in UNUSED, tr_benc * args_out UNUSED, struct tr_rpc_idle_data * idle_data UNUSED) { notify (session, TR_RPC_SESSION_CLOSE, NULL); return NULL; } /*** **** ***/ typedef const char* (*handler)(tr_session*, tr_benc*, tr_benc*, struct tr_rpc_idle_data *); static struct method { const char * name; bool immediate; handler func; } methods[] = { { "port-test", false, portTest }, { "blocklist-update", false, blocklistUpdate }, { "session-close", true, sessionClose }, { "session-get", true, sessionGet }, { "session-set", true, sessionSet }, { "session-stats", true, sessionStats }, { "torrent-add", false, torrentAdd }, { "torrent-get", true, torrentGet }, { "torrent-remove", true, torrentRemove }, { "torrent-set", true, torrentSet }, { "torrent-set-location", true, torrentSetLocation }, { "torrent-start", true, torrentStart }, { "torrent-start-now", true, torrentStartNow }, { "torrent-stop", true, torrentStop }, { "torrent-verify", true, torrentVerify }, { "torrent-reannounce", true, torrentReannounce }, { "queue-move-top", true, queueMoveTop }, { "queue-move-up", true, queueMoveUp }, { "queue-move-down", true, queueMoveDown }, { "queue-move-bottom", true, queueMoveBottom } }; static void noop_response_callback (tr_session * session UNUSED, struct evbuffer * response UNUSED, void * user_data UNUSED) { } static void request_exec (tr_session * session, tr_benc * request, tr_rpc_response_func callback, void * callback_user_data) { int i; const char * str; tr_benc * args_in = tr_bencDictFind (request, "arguments"); const char * result = NULL; if (callback == NULL) callback = noop_response_callback; /* parse the request */ if (!tr_bencDictFindStr (request, "method", &str)) result = "no method name"; else { const int n = TR_N_ELEMENTS (methods); for (i = 0; i < n; ++i) if (!strcmp (str, methods[i].name)) break; if (i ==n) result = "method name not recognized"; } /* if we couldn't figure out which method to use, return an error */ if (result != NULL) { int64_t tag; tr_benc response; struct evbuffer * buf; tr_bencInitDict (&response, 3); tr_bencDictAddDict (&response, "arguments", 0); tr_bencDictAddStr (&response, "result", result); if (tr_bencDictFindInt (request, "tag", &tag)) tr_bencDictAddInt (&response, "tag", tag); buf = tr_bencToBuf (&response, TR_FMT_JSON_LEAN); (*callback)(session, buf, callback_user_data); evbuffer_free (buf); tr_bencFree (&response); } else if (methods[i].immediate) { int64_t tag; tr_benc response; tr_benc * args_out; struct evbuffer * buf; tr_bencInitDict (&response, 3); args_out = tr_bencDictAddDict (&response, "arguments", 0); result = (*methods[i].func)(session, args_in, args_out, NULL); if (result == NULL) result = "success"; tr_bencDictAddStr (&response, "result", result); if (tr_bencDictFindInt (request, "tag", &tag)) tr_bencDictAddInt (&response, "tag", tag); buf = tr_bencToBuf (&response, TR_FMT_JSON_LEAN); (*callback)(session, buf, callback_user_data); evbuffer_free (buf); tr_bencFree (&response); } else { int64_t tag; struct tr_rpc_idle_data * data = tr_new0 (struct tr_rpc_idle_data, 1); data->session = session; data->response = tr_new0 (tr_benc, 1); tr_bencInitDict (data->response, 3); if (tr_bencDictFindInt (request, "tag", &tag)) tr_bencDictAddInt (data->response, "tag", tag); data->args_out = tr_bencDictAddDict (data->response, "arguments", 0); data->callback = callback; data->callback_user_data = callback_user_data; (*methods[i].func)(session, args_in, data->args_out, data); } } void tr_rpc_request_exec_json (tr_session * session, const void * request_json, int request_len, tr_rpc_response_func callback, void * callback_user_data) { tr_benc top; int have_content; if (request_len < 0) request_len = strlen (request_json); have_content = !tr_jsonParse ("rpc", request_json, request_len, &top, NULL); request_exec (session, have_content ? &top : NULL, callback, callback_user_data); if (have_content) tr_bencFree (&top); } /** * Munge the URI into a usable form. * * We have very loose typing on this to make the URIs as simple as possible: * - anything not a 'tag' or 'method' is automatically in 'arguments' * - values that are all-digits are numbers * - values that are all-digits or commas are number lists * - all other values are strings */ void tr_rpc_parse_list_str (tr_benc * setme, const char * str, int len) { int valueCount; int * values = tr_parseNumberRange (str, len, &valueCount); if (valueCount == 0) tr_bencInitStr (setme, str, len); else if (valueCount == 1) tr_bencInitInt (setme, values[0]); else { int i; tr_bencInitList (setme, valueCount); for (i=0; i