mirror/transmission
1
0
Fork 0
mirror of https://github.com/transmission/transmission synced 2024-12-23 00:04:06 +00:00
Code Issues Releases Wiki Activity
transmission/daemon/remote.c
Charles Kerr e47808c277 (trunk) #1608: Completed pieces bitfield in rpc
2009-03-02 05:48:32 +00:00

1358 lines
46 KiB
C
Raw Blame History

/*
* This file Copyright (C) 2008-2009 Charles Kerr <charles@transmissionbt.com>
*
* 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 <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h> /* strcmp */
#ifdef WIN32
#include <direct.h> /* getcwd */
#else
#include <unistd.h> /* getcwd */
#endif
#include <libevent/event.h>
#define CURL_DISABLE_TYPECHECK /* otherwise -Wunreachable-code goes insane */
#include <curl/curl.h>
#include <libtransmission/transmission.h>
#include <libtransmission/bencode.h>
#include <libtransmission/rpcimpl.h>
#include <libtransmission/json.h>
#include <libtransmission/tr-getopt.h>
#include <libtransmission/utils.h>
#include <libtransmission/version.h>
#define MY_NAME "transmission-remote"
#define DEFAULT_HOST "localhost"
#define DEFAULT_PORT atoi(TR_DEFAULT_RPC_PORT_STR)
enum { TAG_SESSION, TAG_LIST, TAG_DETAILS, TAG_FILES, TAG_PEERS };
static const char*
getUsage( void )
{
return
"Transmission " LONG_VERSION_STRING
" http://www.transmissionbt.com/\n"
"A fast and easy BitTorrent client\n"
"\n"
"Usage: " MY_NAME
" [host] [options]\n"
" "
MY_NAME " [port] [options]\n"
" "
MY_NAME " [host:port] [options]\n"
"\n"
"See the man page for detailed explanations and many examples.";
}
static tr_option opts[] =
{
{ 'a', "add", "Add torrent files by filename or URL", "a", 0, NULL },
{ 'b', "debug", "Print debugging information", "b", 0, NULL },
{ 'd', "downlimit", "Set the maximum global download speed in KB/s", "d", 1, "<speed>" },
{ 'D', "no-downlimit", "Don't limit the global download speed", "D", 0, NULL },
{ 910, "encryption-required", "Encrypt all peer connections", "er", 0, NULL },
{ 911, "encryption-preferred", "Prefer encrypted peer connections", "ep", 0, NULL },
{ 912, "encryption-tolerated", "Prefer unencrypted peer connections", "et", 0, NULL },
{ 'f', "files", "List the current torrent(s)' files", "f", 0, NULL },
{ 'g', "get", "Mark files for download", "g", 1, "<files>" },
{ 'G', "no-get", "Mark files for not downloading", "G", 1, "<files>" },
{ 'i', "info", "Show the current torrent(s)' details", "i", 0, NULL },
{ 920, "session-info", "Show the session's details", "si", 0, NULL },
{ 'l', "list", "List all torrents", "l", 0, NULL },
{ 'm', "portmap", "Enable portmapping via NAT-PMP or UPnP", "m", 0, NULL },
{ 'M', "no-portmap", "Disable portmapping", "M", 0, NULL },
{ 'n', "auth", "Set authentication info", "n", 1, "<username:password>" },
{ 'N', "netrc", "Set authentication info from a .netrc file", "N", 1, "<filename>" },
{ 'p', "port", "Port for incoming peers (Default: " TR_DEFAULT_PEER_PORT_STR ")", "p", 1, "<port>" },
{ 900, "priority-high", "Set the files' priorities as high", "ph", 1, "<files>" },
{ 901, "priority-normal", "Set the files' priorities as normal", "pn", 1, "<files>" },
{ 902, "priority-low", "Set the files' priorities as low", "pl", 1, "<files>" },
{ 'r', "remove", "Remove the current torrent(s)", "r", 0, NULL },
{ 930, "peers", "Set the current torrent(s)' maximum number of peers each", "pr", 1, "<max>" },
{ 931, "global-peers", "Set the global maximum number of peers", "gpr", 1, "<max>" },
{ 'R', "remove-and-delete", "Remove the current torrent(s) and delete local data", NULL, 0, NULL },
{ 950, "seedratio", "Let the current torrent(s) seed until a specific ratio", "sr", 1, "ratio" },
{ 951, "seedratio-default", "Let the current torrent(s) use the global seedratio settings", "srd", 0, NULL },
{ 952, "no-seedratio", "Let the current torrent(s) seed regardless of ratio", "SR", 0, NULL },
{ 953, "global-seedratio", "All torrents, unless overridden by a per-torrent setting, should seed until a specific ratio", "gsr", 1, "ratio" },
{ 954, "no-global-seedratio", "All torrents, unless overridden by a per-torrent setting, should seed regardless of ratio", "GSR", 0, NULL },
{ 's', "start", "Start the current torrent(s)", "s", 0, NULL },
{ 'S', "stop", "Stop the current torrent(s)", "S", 0, NULL },
{ 't', "torrent", "Set the current torrent(s)", "t", 1, "<torrent>" },
{ 'u', "uplimit", "Set the maximum global upload speed in KB/s", "u", 1, "<speed>" },
{ 'U', "no-uplimit", "Don't limit the global upload speed", "U", 0, NULL },
{ 'v', "verify", "Verify the current torrent(s)", "v", 0, NULL },
{ 'V', "version", "Show version number and exit", "V", 0, NULL },
{ 'w', "download-dir", "Set the default download folder", "w", 1, "<path>" },
{ 'x', "pex", "Enable peer exchange (PEX)", "x", 0, NULL },
{ 'X', "no-pex", "Disable peer exchange (PEX)", "X", 0, NULL },
{ 940, "peer-info", "List the current torrent(s)' peers", "pi", 0, NULL },
{ 0, NULL, NULL, NULL, 0, NULL }
};
static void
showUsage( void )
{
tr_getopt_usage( MY_NAME, getUsage( ), opts );
exit( 0 );
}
static int
numarg( const char * arg )
{
char * end = NULL;
const long num = strtol( arg, &end, 10 );
if( *end )
{
fprintf( stderr, "Not a number: \"%s\"\n", arg );
showUsage( );
}
return num;
}
static char * reqs[256]; /* arbitrary max */
static int reqCount = 0;
static int debug = 0;
static char * auth = NULL;
static char * netrc = NULL;
static char*
tr_getcwd( void )
{
char buf[2048];
*buf = '\0';
#ifdef WIN32
_getcwd( buf, sizeof( buf ) );
#else
getcwd( buf, sizeof( buf ) );
#endif
return tr_strdup( buf );
}
static char*
absolutify( const char * path )
{
char * buf;
if( *path == '/' )
buf = tr_strdup( path );
else {
char * cwd = tr_getcwd( );
buf = tr_buildPath( cwd, path, NULL );
tr_free( cwd );
}
return buf;
}
static char*
getEncodedMetainfo( const char * filename )
{
size_t len = 0;
char * b64 = NULL;
uint8_t * buf = tr_loadFile( filename, &len );
if( buf )
{
b64 = tr_base64_encode( buf, len, NULL );
tr_free( buf );
}
return b64;
}
static void
addIdArg( tr_benc * args,
const char * id )
{
if( !*id )
{
fprintf(
stderr,
"No torrent specified! Please use the -t option first.\n" );
id = "-1"; /* no torrent will have this ID, so should be a no-op */
}
if( strcmp( id, "all" ) )
{
tr_rpc_parse_list_str( tr_bencDictAdd( args,
"ids" ), id, strlen( id ) );
}
}
static void
addFiles( tr_benc * args,
const char * key,
const char * arg )
{
tr_benc * files = tr_bencDictAddList( args, key, 100 );
if( !*arg )
{
fprintf( stderr, "No files specified!\n" );
arg = "-1"; /* no file will have this index, so should be a no-op */
}
if( strcmp( arg, "all" ) )
{
int i;
int valueCount;
int * values = tr_parseNumberRange( arg, -1, &valueCount );
for( i=0; i<valueCount; ++i )
tr_bencListAddInt( files, values[i] );
tr_free( values );
}
}
#define TR_N_ELEMENTS( ary ) ( sizeof( ary ) / sizeof( *ary ) )
static const char * files_keys[] = {
"files",
"name",
"priorities",
"wanted"
};
static const char * details_keys[] = {
"activityDate",
"addedDate",
"announceResponse",
"announceURL",
"comment",
"corruptEver",
"creator",
"dateCreated",
"doneDate",
"downloadDir",
"downloadedEver",
"errorString",
"eta",
"hashString",
"haveUnchecked",
"haveValid",
"id",
"isPrivate",
"lastAnnounceTime",
"lastScrapeTime",
"leechers",
"leftUntilDone",
"name",
"nextAnnounceTime",
"nextScrapeTime",
"peersConnected",
"peersGettingFromUs",
"peersSendingToUs",
"pieceCount",
"pieceSize",
"rateDownload",
"rateUpload",
"recheckProgress",
"scrapeResponse",
"seeders",
"sizeWhenDone",
"startDate",
"status",
"timesCompleted",
"totalSize",
"uploadedEver",
"pieces",
"webseeds",
"webseedsSendingToUs"
};
static const char * list_keys[] = {
"error",
"errorString",
"eta",
"id",
"leftUntilDone",
"name",
"peersGettingFromUs",
"peersSendingToUs",
"rateDownload",
"rateUpload",
"sizeWhenDone",
"status",
"uploadRatio"
};
static void
readargs( int argc,
const char ** argv )
{
int c;
int addingTorrents = 0;
const char * optarg;
char id[4096];
*id = '\0';
while( ( c = tr_getopt( getUsage( ), argc, argv, opts, &optarg ) ) )
{
int i, n;
int addArg = TRUE;
tr_benc top, *args, *fields;
tr_bencInitDict( &top, 3 );
args = tr_bencDictAddDict( &top, "arguments", 0 );
switch( c )
{
case TR_OPT_UNK:
if( addingTorrents )
{
char * tmp = getEncodedMetainfo( optarg );
if( tmp )
{
tr_bencDictAddStr( &top, "method", "torrent-add" );
tr_bencDictAddStr( args, "metainfo", tmp );
tr_free( tmp );
}
else
{
tr_bencDictAddStr( &top, "method", "torrent-add" );
tr_bencDictAddStr( args, "filename", optarg );
}
}
else
{
fprintf( stderr, "Unknown option: %s\n", optarg );
addArg = FALSE;
}
break;
case 'a':
addingTorrents = 1;
addArg = FALSE;
break;
case 'b':
debug = 1;
addArg = FALSE;
break;
case 'd':
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "speed-limit-down", numarg( optarg ) );
tr_bencDictAddInt( args, "speed-limit-down-enabled", 1 );
break;
case 'D':
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "speed-limit-down-enabled", 0 );
break;
case 'f':
tr_bencDictAddStr( &top, "method", "torrent-get" );
tr_bencDictAddInt( &top, "tag", TAG_FILES );
addIdArg( args, id );
n = TR_N_ELEMENTS( files_keys );
fields = tr_bencDictAddList( args, "fields", n );
for( i = 0; i < n; ++i )
tr_bencListAddStr( fields, files_keys[i] );
break;
case 'g':
tr_bencDictAddStr( &top, "method", "torrent-set" );
addIdArg( args, id );
addFiles( args, "files-wanted", optarg );
break;
case 'G':
tr_bencDictAddStr( &top, "method", "torrent-set" );
addIdArg( args, id );
addFiles( args, "files-unwanted", optarg );
break;
case 'i':
tr_bencDictAddStr( &top, "method", "torrent-get" );
tr_bencDictAddInt( &top, "tag", TAG_DETAILS );
addIdArg( args, id );
n = TR_N_ELEMENTS( details_keys );
fields = tr_bencDictAddList( args, "fields", n );
for( i = 0; i < n; ++i )
tr_bencListAddStr( fields, details_keys[i] );
break;
case 'l':
tr_bencDictAddStr( &top, "method", "torrent-get" );
tr_bencDictAddInt( &top, "tag", TAG_LIST );
n = TR_N_ELEMENTS( list_keys );
fields = tr_bencDictAddList( args, "fields", n );
for( i = 0; i < n; ++i )
tr_bencListAddStr( fields, list_keys[i] );
break;
case 'm':
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "port-forwarding-enabled", 1 );
break;
case 'M':
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "port-forwarding-enabled", 0 );
break;
case 'n':
auth = tr_strdup( optarg );
addArg = FALSE;
break;
case 'N':
netrc = tr_strdup( optarg );
addArg = FALSE;
break;
case 'p':
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "port", numarg( optarg ) );
break;
case 'r':
tr_bencDictAddStr( &top, "method", "torrent-remove" );
addIdArg( args, id );
break;
case 'R':
tr_bencDictAddStr( &top, "method", "torrent-remove" );
addIdArg( args, id );
tr_bencDictAddInt( args, "delete-local-data", 1 );
break;
case 's':
tr_bencDictAddStr( &top, "method", "torrent-start" );
addIdArg( args, id );
break;
case 'S':
tr_bencDictAddStr( &top, "method", "torrent-stop" );
addIdArg( args, id );
break;
case 't':
tr_strlcpy( id, optarg, sizeof( id ) );
addArg = FALSE;
break;
case 'u':
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "speed-limit-up", numarg( optarg ) );
tr_bencDictAddInt( args, "speed-limit-up-enabled", 1 );
break;
case 'U':
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "speed-limit-up-enabled", 0 );
break;
case 'v':
tr_bencDictAddStr( &top, "method", "torrent-verify" );
addIdArg( args, id );
break;
case 'V':
fprintf( stderr, "Transmission %s\n", LONG_VERSION_STRING );
exit( 0 );
break;
case 'w': {
char * path = absolutify( optarg );
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddStr( args, "download-dir", path );
tr_free( path );
break;
}
case 'x':
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "pex-allowed", 1 );
break;
case 'X':
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "pex-allowed", 0 );
break;
case 900:
tr_bencDictAddStr( &top, "method", "torrent-set" );
addIdArg( args, id );
addFiles( args, "priority-high", optarg );
break;
case 901:
tr_bencDictAddStr( &top, "method", "torrent-set" );
addIdArg( args, id );
addFiles( args, "priority-normal", optarg );
break;
case 902:
tr_bencDictAddStr( &top, "method", "torrent-set" );
addIdArg( args, id );
addFiles( args, "priority-low", optarg );
break;
case 910:
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddStr( args, "encryption", "required" );
break;
case 911:
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddStr( args, "encryption", "preferred" );
break;
case 912:
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddStr( args, "encryption", "tolerated" );
break;
case 920:
tr_bencDictAddStr( &top, "method", "session-get" );
tr_bencDictAddInt( &top, "tag", TAG_SESSION );
break;
case 930:
tr_bencDictAddStr( &top, "method", "torrent-set" );
addIdArg( args, id );
tr_bencDictAddInt( args, "peer-limit", atoi(optarg) );
break;
case 931:
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "peer-limit", atoi(optarg) );
break;
case 940:
tr_bencDictAddStr( &top, "method", "torrent-get" );
tr_bencDictAddInt( &top, "tag", TAG_PEERS );
addIdArg( args, id );
fields = tr_bencDictAddList( args, "fields", 1 );
tr_bencListAddStr( fields, "peers" );
break;
case 950:
tr_bencDictAddStr( &top, "method", "torrent-set" );
tr_bencDictAddDouble( args, "ratio-limit", atof(optarg) );
tr_bencDictAddInt( args, "ratio-limit-mode", TR_RATIOLIMIT_SINGLE );
addIdArg( args, id );
break;
case 951:
tr_bencDictAddStr( &top, "method", "torrent-set" );
tr_bencDictAddInt( args, "ratio-limit-mode", TR_RATIOLIMIT_GLOBAL );
addIdArg( args, id );
break;
case 952:
tr_bencDictAddStr( &top, "method", "torrent-set" );
tr_bencDictAddInt( args, "ratio-limit-mode", TR_RATIOLIMIT_UNLIMITED );
addIdArg( args, id );
break;
case 953:
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddDouble( args, "ratio-limit", atof(optarg) );
tr_bencDictAddInt( args, "ratio-limit-enabled", 1 );
break;
case 954:
tr_bencDictAddStr( &top, "method", "session-set" );
tr_bencDictAddInt( args, "ratio-limit-enabled", 0 );
break;
case TR_OPT_ERR:
fprintf( stderr, "invalid option\n" );
showUsage( );
break;
default:
fprintf( stderr, "got opt [%d]\n", (int)c );
showUsage( );
break;
}
if( addArg )
{
struct evbuffer * buf = tr_getBuffer( );
reqs[reqCount++] = tr_strdup( tr_bencSaveAsJSON( &top, buf ) );
tr_releaseBuffer( buf );
}
tr_bencFree( &top );
}
}
/* [host:port] or [host] or [port] */
static void
getHostAndPort( int * argc,
char ** argv,
char ** host,
int * port )
{
if( *argv[1] != '-' )
{
int i;
const char * s = argv[1];
const char * delim = strchr( s, ':' );
if( delim ) /* user passed in both host and port */
{
*host = tr_strndup( s, delim - s );
*port = atoi( delim + 1 );
}
else
{
char * end;
const int i = strtol( s, &end, 10 );
if( !*end ) /* user passed in a port */
*port = i;
else /* user passed in a host */
*host = tr_strdup( s );
}
*argc -= 1;
for( i = 1; i < *argc; ++i )
argv[i] = argv[i + 1];
}
}
static size_t
writeFunc( void * ptr,
size_t size,
size_t nmemb,
void * buf )
{
const size_t byteCount = size * nmemb;
evbuffer_add( buf, ptr, byteCount );
return byteCount;
}
static void
etaToString( char * buf,
size_t buflen,
int64_t eta )
{
if( eta < 0 ) tr_snprintf( buf, buflen, "Unknown" );
else if( eta < 60 ) tr_snprintf( buf, buflen, "%" PRId64 "sec", eta );
else if( eta <
( 60 * 60 ) ) tr_snprintf( buf, buflen, "%" PRId64 " min",
eta / 60 );
else if( eta <
( 60 * 60 * 24 ) ) tr_snprintf( buf, buflen, "%" PRId64 " hrs",
eta / ( 60 * 60 ) );
else tr_snprintf( buf, buflen, "%" PRId64 " days", eta / ( 60 * 60 * 24 ) );
}
#define KILOBYTE_FACTOR 1024.0
#define MEGABYTE_FACTOR ( 1024.0 * 1024.0 )
#define GIGABYTE_FACTOR ( 1024.0 * 1024.0 * 1024.0 )
static char*
strlratio2( char * buf, double ratio, size_t buflen )
{
if( (int)ratio == TR_RATIO_NA )
tr_strlcpy( buf, "None", buflen );
else if( (int)ratio == TR_RATIO_INF )
tr_strlcpy( buf, "Inf", buflen );
else if( ratio < 10.0 )
tr_snprintf( buf, buflen, "%'.2f", ratio );
else if( ratio < 100.0 )
tr_snprintf( buf, buflen, "%'.1f", ratio );
else
tr_snprintf( buf, buflen, "%'.0f", ratio );
return buf;
}
static char*
strlratio( char * buf,
double numerator,
double denominator,
size_t buflen )
{
double ratio;
if( denominator )
ratio = numerator / denominator;
else if( numerator )
ratio = TR_RATIO_INF;
else
ratio = TR_RATIO_NA;
return strlratio2( buf, ratio, buflen );
}
static char*
strlsize( char * buf, int64_t size, size_t buflen )
{
if( !size )
tr_strlcpy( buf, "None", buflen );
else if( size < (int64_t)KILOBYTE_FACTOR )
tr_snprintf( buf, buflen, "%'" PRId64 " bytes", (int64_t)size );
else
{
double displayed_size;
if( size < (int64_t)MEGABYTE_FACTOR )
{
displayed_size = (double) size / KILOBYTE_FACTOR;
tr_snprintf( buf, buflen, "%'.1f KB", displayed_size );
}
else if( size < (int64_t)GIGABYTE_FACTOR )
{
displayed_size = (double) size / MEGABYTE_FACTOR;
tr_snprintf( buf, buflen, "%'.1f MB", displayed_size );
}
else
{
displayed_size = (double) size / GIGABYTE_FACTOR;
tr_snprintf( buf, buflen, "%'.1f GB", displayed_size );
}
}
return buf;
}
static char*
getStatusString( tr_benc * t, char * buf, size_t buflen )
{
int64_t status;
if( !tr_bencDictFindInt( t, "status", &status ) )
{
*buf = '\0';
}
else switch( status )
{
case TR_STATUS_STOPPED:
tr_strlcpy( buf, "Stopped", buflen );
break;
case TR_STATUS_CHECK_WAIT:
case TR_STATUS_CHECK: {
const char * str = status == TR_STATUS_CHECK_WAIT
? "Will Verify"
: "Verifying";
double percent;
if( tr_bencDictFindDouble( t, "recheckProgress", &percent ) )
tr_snprintf( buf, buflen, "%s (%.0f%%)", str, percent*100.0 );
else
tr_strlcpy( buf, str, buflen );
break;
}
case TR_STATUS_DOWNLOAD:
case TR_STATUS_SEED: {
int64_t fromUs = 0;
int64_t toUs = 0;
tr_bencDictFindInt( t, "peersGettingFromUs", &fromUs );
tr_bencDictFindInt( t, "peersSendingToUs", &toUs );
if( fromUs && toUs )
tr_strlcpy( buf, "Up & Down", buflen );
else if( toUs )
tr_strlcpy( buf, "Downloading", buflen );
else if( fromUs ) {
int64_t leftUntilDone = 0;
tr_bencDictFindInt( t, "leftUntilDone", &leftUntilDone );
if( leftUntilDone > 0 )
tr_strlcpy( buf, "Uploading", buflen );
else
tr_strlcpy( buf, "Seeding", buflen );
} else {
tr_strlcpy( buf, "Idle", buflen );
}
break;
}
}
return buf;
}
static const char*
getTrackerDateStr( const time_t t, tr_bool isStopped )
{
const char * str;
switch( t ) {
case 0: str = isStopped ? "None (Stopped)\n" : "None\n"; break;
case 1: str = "In Progress\n"; break;
default: str = ctime( &t ); break;
}
return str;
}
static void
printSession( tr_benc * top )
{
tr_benc *args;
if( ( tr_bencDictFindDict( top, "arguments", &args ) ) )
{
const char * str;
int64_t i;
printf( "VERSION\n" );
if( tr_bencDictFindStr( args, "version", &str ) )
printf( " Daemon version: %s\n", str );
if( tr_bencDictFindInt( args, "rpc-version", &i ) )
printf( " RPC version: %" PRId64 "\n", i );
if( tr_bencDictFindInt( args, "rpc-version-minimum", &i ) )
printf( " RPC minimum version: %" PRId64 "\n", i );
printf( "\n" );
printf( "TRANSFER\n" );
if( tr_bencDictFindStr( args, "download-dir", &str ) )
printf( " Download directory: %s\n", str );
if( tr_bencDictFindInt( args, "port", &i ) )
printf( " Listenport: %" PRId64 "\n", i );
if( tr_bencDictFindInt( args, "port-forwarding-enabled", &i ) )
printf( " Portforwarding enabled: %s\n", ( i ? "Yes" : "No" ) );
if( tr_bencDictFindInt( args, "pex-allowed", &i ) )
printf( " Peer exchange allowed: %s\n", ( i ? "Yes" : "No" ) );
if( tr_bencDictFindStr( args, "encryption", &str ) )
printf( " Encryption: %s\n", str );
printf( "\n" );
printf( "LIMITS\n" );
if( tr_bencDictFindInt( args, "peer-limit", &i ) )
printf( " Peer limit: %" PRId64 "\n", i );
if( tr_bencDictFindInt( args, "speed-limit-down-enabled", &i ) )
printf( " Downloadlimit enabled: %s\n", ( i ? "Yes" : "No" ) );
if( tr_bencDictFindInt( args, "speed-limit-down", &i ) )
printf( " Downloadlimit: %6" PRId64 " KB/sec\n", i );
if( tr_bencDictFindInt( args, "speed-limit-up-enabled", &i ) )
printf( " Uploadlimit enabled: %s\n", ( i ? "Yes" : "No" ) );
if( tr_bencDictFindInt( args, "speed-limit-up", &i ) )
printf( " Uploadlimit: %6" PRId64 " KB/sec\n", i );
}
}
static void
printDetails( tr_benc * top )
{
tr_benc *args, *torrents;
if( ( tr_bencDictFindDict( top, "arguments", &args ) )
&& ( tr_bencDictFindList( args, "torrents", &torrents ) ) )
{
int ti, tCount;
for( ti = 0, tCount = tr_bencListSize( torrents ); ti < tCount;
++ti )
{
tr_benc * t = tr_bencListChild( torrents, ti );
tr_benc * l;
const uint8_t * raw;
size_t rawlen;
const char * str;
char buf[512];
char buf2[512];
int64_t i, j, k;
tr_bool isStopped;
isStopped = tr_bencDictFindInt( t, "status", &i ) && (i==TR_STATUS_STOPPED);
printf( "NAME\n" );
if( tr_bencDictFindInt( t, "id", &i ) )
printf( " Id: %" PRId64 "\n", i );
if( tr_bencDictFindStr( t, "name", &str ) )
printf( " Name: %s\n", str );
if( tr_bencDictFindStr( t, "hashString", &str ) )
printf( " Hash: %s\n", str );
printf( "\n" );
printf( "TRANSFER\n" );
getStatusString( t, buf, sizeof( buf ) );
printf( " State: %s\n", buf );
if( tr_bencDictFindStr( t, "downloadDir", &str ) )
printf( " Location: %s\n", str );
if( tr_bencDictFindInt( t, "sizeWhenDone", &i )
&& tr_bencDictFindInt( t, "leftUntilDone", &j ) )
{
strlratio( buf, 100.0 * ( i - j ), i, sizeof( buf ) );
printf( " Percent Done: %s%%\n", buf );
}
if( tr_bencDictFindInt( t, "eta", &i ) )
{
etaToString( buf, sizeof( buf ), i );
printf( " ETA: %s\n", buf );
}
if( tr_bencDictFindInt( t, "rateDownload", &i ) )
printf( " Download Speed: %.1f KB/s\n", i / 1024.0 );
if( tr_bencDictFindInt( t, "rateUpload", &i ) )
printf( " Upload Speed: %.1f KB/s\n", i / 1024.0 );
if( tr_bencDictFindInt( t, "haveUnchecked", &i )
&& tr_bencDictFindInt( t, "haveValid", &j ) )
{
strlsize( buf, i + j, sizeof( buf ) );
strlsize( buf2, j, sizeof( buf2 ) );
printf( " Have: %s (%s verified)\n", buf, buf2 );
}
if( tr_bencDictFindInt( t, "sizeWhenDone", &i )
&& tr_bencDictFindInt( t, "totalSize", &j ) )
{
strlsize( buf, j, sizeof( buf ) );
strlsize( buf2, i, sizeof( buf2 ) );
printf( " Total size: %s (%s wanted)\n", buf, buf2 );
}
if( tr_bencDictFindInt( t, "downloadedEver", &i )
&& tr_bencDictFindInt( t, "uploadedEver", &j ) )
{
strlsize( buf, i, sizeof( buf ) );
printf( " Downloaded: %s\n", buf );
strlsize( buf, j, sizeof( buf ) );
printf( " Uploaded: %s\n", buf );
strlratio( buf, j, i, sizeof( buf ) );
printf( " Ratio: %s\n", buf );
}
if( tr_bencDictFindInt( t, "corruptEver", &i ) )
{
strlsize( buf, i, sizeof( buf ) );
printf( " Corrupt DL: %s\n", buf );
}
if( tr_bencDictFindStr( t, "errorString", &str ) && str && *str )
printf( " Error: %s\n", str );
if( tr_bencDictFindInt( t, "peersConnected", &i )
&& tr_bencDictFindInt( t, "peersGettingFromUs", &j )
&& tr_bencDictFindInt( t, "peersSendingToUs", &k ) )
{
printf(
" Peers: "
"connected to %" PRId64 ", "
"uploading to %" PRId64
", "
"downloading from %"
PRId64 "\n",
i, j, k );
}
if( tr_bencDictFindList( t, "webseeds", &l )
&& tr_bencDictFindInt( t, "webseedsSendingToUs", &i ) )
{
const int64_t n = tr_bencListSize( l );
if( n > 0 )
printf(
" Web Seeds: downloading from %" PRId64 " of %"
PRId64
" web seeds\n", i, n );
}
printf( "\n" );
printf( "HISTORY\n" );
if( tr_bencDictFindInt( t, "addedDate", &i ) && i )
{
const time_t tt = i;
printf( " Date added: %s", ctime( &tt ) );
}
if( tr_bencDictFindInt( t, "doneDate", &i ) && i )
{
const time_t tt = i;
printf( " Date finished: %s", ctime( &tt ) );
}
if( tr_bencDictFindInt( t, "startDate", &i ) && i )
{
const time_t tt = i;
printf( " Date started: %s", ctime( &tt ) );
}
if( tr_bencDictFindInt( t, "activityDate", &i ) && i )
{
const time_t tt = i;
printf( " Latest activity: %s", ctime( &tt ) );
}
printf( "\n" );
printf( "TRACKER\n" );
if( tr_bencDictFindInt( t, "lastAnnounceTime", &i ) )
printf( " Latest announce: %s", getTrackerDateStr( (time_t)i, isStopped ) );
if( tr_bencDictFindStr( t, "announceURL", &str ) )
printf( " Announce URL: %s\n", str );
if( tr_bencDictFindStr( t, "announceResponse", &str ) && str && *str )
printf( " Announce response: %s\n", str );
if( tr_bencDictFindInt( t, "nextAnnounceTime", &i ) )
printf( " Next announce: %s", getTrackerDateStr( (time_t)i, isStopped ) );
if( tr_bencDictFindInt( t, "lastScrapeTime", &i ) )
printf( " Latest scrape: %s", getTrackerDateStr( (time_t)i, isStopped ) );
if( tr_bencDictFindStr( t, "scrapeResponse", &str ) )
printf( " Scrape response: %s\n", str );
if( tr_bencDictFindInt( t, "nextScrapeTime", &i ) )
printf( " Next scrape: %s", getTrackerDateStr( (time_t)i, isStopped ) );
if( tr_bencDictFindInt( t, "seeders", &i ) && tr_bencDictFindInt( t, "leechers", &j ) )
printf( " Tracker knows of %" PRId64 " seeders and %" PRId64 " leechers\n", i, j );
if( tr_bencDictFindInt( t, "timesCompleted", &i ) )
printf( " Tracker has seen %" PRId64 " clients complete this torrent\n", i );
printf( "\n" );
printf( "ORIGINS\n" );
if( tr_bencDictFindInt( t, "dateCreated", &i ) && i )
{
const time_t tt = i;
printf( " Date created: %s", ctime( &tt ) );
}
if( tr_bencDictFindInt( t, "isPrivate", &i ) )
printf( " Public torrent: %s\n", ( i ? "No" : "Yes" ) );
if( tr_bencDictFindStr( t, "comment", &str ) && str && *str )
printf( " Comment: %s\n", str );
if( tr_bencDictFindStr( t, "creator", &str ) && str && *str )
printf( " Creator: %s\n", str );
if( tr_bencDictFindInt( t, "pieceCount", &i ) )
printf( " Piece Count: %" PRId64 "\n", i );
if( tr_bencDictFindInt( t, "pieceSize", &i ) )
printf( " Piece Size: %" PRId64 "\n", i );
printf( "\n" );
printf("PIECES\n ");
if( tr_bencDictFindRaw( t, "pieces", &raw, &rawlen ) && tr_bencDictFindInt( t, "pieceCount", &j ) ) {
int len;
char * str = tr_base64_decode( raw, rawlen, &len );
for( i=k=0; k<len; ++k ) {
int e;
for( e=0; i<j && e<8; ++e, ++i )
printf( str[k] & (1<<(7-e)) ? "1" : "0" );
printf( " " );
if( !(i%64) )
printf( "\n " );
}
tr_free( str );
}
printf( "\n" );
}
}
}
static void
printFileList( tr_benc * top )
{
tr_benc *args, *torrents;
if( ( tr_bencDictFindDict( top, "arguments", &args ) )
&& ( tr_bencDictFindList( args, "torrents", &torrents ) ) )
{
int i, in;
for( i = 0, in = tr_bencListSize( torrents ); i < in; ++i )
{
tr_benc * d = tr_bencListChild( torrents, i );
tr_benc * files, *priorities, *wanteds;
const char * name;
if( tr_bencDictFindStr( d, "name", &name )
&& tr_bencDictFindList( d, "files", &files )
&& tr_bencDictFindList( d, "priorities", &priorities )
&& tr_bencDictFindList( d, "wanted", &wanteds ) )
{
int j = 0, jn = tr_bencListSize( files );
printf( "%s (%d files):\n", name, jn );
printf( "%3s %4s %8s %3s %9s %s\n", "#", "Done",
"Priority", "Get", "Size",
"Name" );
for( j = 0, jn = tr_bencListSize( files ); j < jn; ++j )
{
int64_t have;
int64_t length;
int64_t priority;
int64_t wanted;
const char * filename;
tr_benc * file = tr_bencListChild( files, j );
if( tr_bencDictFindInt( file, "length", &length )
&& tr_bencDictFindStr( file, "name", &filename )
&& tr_bencDictFindInt( file, "bytesCompleted", &have )
&& tr_bencGetInt( tr_bencListChild( priorities,
j ), &priority )
&& tr_bencGetInt( tr_bencListChild( wanteds,
j ), &wanted ) )
{
char sizestr[64];
double percent = (double)have / length;
const char * pristr;
strlsize( sizestr, length, sizeof( sizestr ) );
switch( priority )
{
case TR_PRI_LOW:
pristr = "Low"; break;
case TR_PRI_HIGH:
pristr = "High"; break;
default:
pristr = "Normal"; break;
}
printf( "%3d: %3.0f%% %-8s %-3s %9s %s\n",
j,
( 100.0 * percent ),
pristr,
( wanted ? "Yes" : "No" ),
sizestr,
filename );
}
}
}
}
}
}
static void
printPeersImpl( tr_benc * peers )
{
int i, n;
printf( "%-20s %-12s %-5s %-6s %-6s %s\n",
"Address", "Flags", "Done", "Down", "Up", "Client" );
for( i = 0, n = tr_bencListSize( peers ); i < n; ++i )
{
double progress;
const char * address, * client, * flagstr;
int64_t rateToClient, rateToPeer;
tr_benc * d = tr_bencListChild( peers, i );
if( tr_bencDictFindStr( d, "address", &address )
&& tr_bencDictFindStr( d, "clientName", &client )
&& tr_bencDictFindDouble( d, "progress", &progress )
&& tr_bencDictFindStr( d, "flagStr", &flagstr )
&& tr_bencDictFindInt( d, "rateToClient", &rateToClient )
&& tr_bencDictFindInt( d, "rateToPeer", &rateToPeer ) )
{
printf( "%-20s %-12s %-5.1f %6.1f %6.1f %s\n",
address, flagstr, (progress*100.0),
rateToClient / 1024.0,
rateToPeer / 1024.0,
client );
}
}
}
static void
printPeers( tr_benc * top )
{
tr_benc *args, *torrents;
if( tr_bencDictFindDict( top, "arguments", &args )
&& tr_bencDictFindList( args, "torrents", &torrents ) )
{
int i, n;
for( i=0, n=tr_bencListSize( torrents ); i<n; ++i )
{
tr_benc * peers;
tr_benc * torrent = tr_bencListChild( torrents, i );
if( tr_bencDictFindList( torrent, "peers", &peers ) ) {
printPeersImpl( peers );
if( i+1<n )
printf( "\n" );
}
}
}
}
static void
printTorrentList( tr_benc * top )
{
tr_benc *args, *list;
if( ( tr_bencDictFindDict( top, "arguments", &args ) )
&& ( tr_bencDictFindList( args, "torrents", &list ) ) )
{
int i, n;
int64_t total_up = 0, total_down = 0, total_size = 0;
char haveStr[32];
printf( "%-4s %-4s %9s %-8s %6s %6s %-5s %-11s %s\n",
"ID", "Done", "Have", "ETA", "Up", "Down", "Ratio", "Status",
"Name" );
for( i = 0, n = tr_bencListSize( list ); i < n; ++i )
{
int64_t id, eta, status, up, down;
int64_t sizeWhenDone, leftUntilDone;
double ratio;
const char * name;
tr_benc * d = tr_bencListChild( list, i );
if( tr_bencDictFindInt( d, "eta", &eta )
&& tr_bencDictFindInt( d, "id", &id )
&& tr_bencDictFindInt( d, "leftUntilDone", &leftUntilDone )
&& tr_bencDictFindStr( d, "name", &name )
&& tr_bencDictFindInt( d, "rateDownload", &down )
&& tr_bencDictFindInt( d, "rateUpload", &up )
&& tr_bencDictFindInt( d, "sizeWhenDone", &sizeWhenDone )
&& tr_bencDictFindInt( d, "status", &status )
&& tr_bencDictFindDouble( d, "uploadRatio", &ratio ) )
{
char etaStr[16];
char statusStr[64];
char ratioStr[32];
char doneStr[8];
int64_t error;
char errorMark;
if( sizeWhenDone )
tr_snprintf( doneStr, sizeof( doneStr ), "%d%%", (int)( 100.0 * ( sizeWhenDone - leftUntilDone ) / sizeWhenDone ) );
else
tr_strlcpy( doneStr, "n/a", sizeof( doneStr ) );
strlsize( haveStr, sizeWhenDone - leftUntilDone, sizeof( haveStr ) );
if( leftUntilDone )
etaToString( etaStr, sizeof( etaStr ), eta );
else
tr_snprintf( etaStr, sizeof( etaStr ), "Done" );
if( tr_bencDictFindInt( d, "error", &error ) && error )
errorMark = '*';
else
errorMark = ' ';
printf(
"%4d%c %4s %9s %-8s %6.1f %6.1f %5s %-11s %s\n",
(int)id, errorMark,
doneStr,
haveStr,
etaStr,
up / 1024.0,
down / 1024.0,
strlratio2( ratioStr, ratio, sizeof( ratioStr ) ),
getStatusString( d, statusStr, sizeof( statusStr ) ),
name );
total_up += up;
total_down += down;
total_size += sizeWhenDone - leftUntilDone;
}
}
printf( "Sum: %9s %6.1f %6.1f\n",
strlsize( haveStr, total_size, sizeof( haveStr ) ),
total_up / 1024.0,
total_down / 1024.0 );
}
}
static void
processResponse( const char * host,
int port,
const void * response,
size_t len )
{
tr_benc top;
if( debug )
fprintf( stderr, "got response:\n--------\n%*.*s\n--------\n",
(int)len, (int)len, (const char*) response );
if( tr_jsonParse( response, len, &top, NULL ) )
tr_nerr( MY_NAME, "Unable to parse response \"%*.*s\"", (int)len,
(int)len, (char*)response );
else
{
int64_t tag = -1;
const char * str;
tr_bencDictFindInt( &top, "tag", &tag );
switch( tag )
{
case TAG_SESSION:
printSession( &top ); break;
case TAG_FILES:
printFileList( &top ); break;
case TAG_DETAILS:
printDetails( &top ); break;
case TAG_LIST:
printTorrentList( &top ); break;
case TAG_PEERS:
printPeers( &top ); break;
default:
if( tr_bencDictFindStr( &top, "result", &str ) )
printf( "%s:%d responded: \"%s\"\n", host, port, str );
}
tr_bencFree( &top );
}
}
static void
processRequests( const char * host,
int port,
const char ** reqs,
int reqCount )
{
int i;
CURL * curl;
struct evbuffer * buf = evbuffer_new( );
char * url = tr_strdup_printf(
"http://%s:%d/transmission/rpc", host, port );
curl = curl_easy_init( );
curl_easy_setopt( curl, CURLOPT_VERBOSE, debug );
#ifdef HAVE_LIBZ
curl_easy_setopt( curl, CURLOPT_ENCODING, "deflate" );
#endif
curl_easy_setopt( curl, CURLOPT_USERAGENT, MY_NAME "/" LONG_VERSION_STRING );
curl_easy_setopt( curl, CURLOPT_WRITEFUNCTION, writeFunc );
curl_easy_setopt( curl, CURLOPT_WRITEDATA, buf );
curl_easy_setopt( curl, CURLOPT_POST, 1 );
curl_easy_setopt( curl, CURLOPT_URL, url );
curl_easy_setopt( curl, CURLOPT_NETRC, CURL_NETRC_OPTIONAL );
curl_easy_setopt( curl, CURLOPT_HTTPAUTH, CURLAUTH_ANY );
if( netrc )
curl_easy_setopt( curl, CURLOPT_NETRC_FILE, netrc );
if( auth )
curl_easy_setopt( curl, CURLOPT_USERPWD, auth );
for( i = 0; i < reqCount; ++i )
{
CURLcode res;
curl_easy_setopt( curl, CURLOPT_POSTFIELDS, reqs[i] );
if( debug )
fprintf( stderr, "posting:\n--------\n%s\n--------\n", reqs[i] );
if( ( res = curl_easy_perform( curl ) ) )
tr_nerr( MY_NAME, "(%s:%d) %s", host, port,
curl_easy_strerror( res ) );
else
processResponse( host, port, EVBUFFER_DATA(
buf ), EVBUFFER_LENGTH( buf ) );
evbuffer_drain( buf, EVBUFFER_LENGTH( buf ) );
}
/* cleanup */
tr_free( url );
evbuffer_free( buf );
curl_easy_cleanup( curl );
}
int
main( int argc,
char ** argv )
{
int i;
int port = DEFAULT_PORT;
char * host = NULL;
if( argc < 2 )
showUsage( );
getHostAndPort( &argc, argv, &host, &port );
if( host == NULL )
host = tr_strdup( DEFAULT_HOST );
readargs( argc, (const char**)argv );
if( reqCount )
processRequests( host, port, (const char**)reqs, reqCount );
else
showUsage( );
for( i = 0; i < reqCount; ++i )
tr_free( reqs[i] );
tr_free( host );
return 0;
}
Reference in a new issue View git blame Copy permalink