source: trunk/cli/transmissioncli.c @ 1517

Last change on this file since 1517 was 1517, checked in by joshe, 15 years ago

Use a tr_tracker_info_t * directly in tr_info_t and tr_tracker_t.

  • Property svn:keywords set to Date Rev Author Id
File size: 10.3 KB
Line 
1/******************************************************************************
2 * $Id: transmissioncli.c 1517 2007-02-27 04:00:38Z joshe $
3 *
4 * Copyright (c) 2005-2006 Transmission authors and contributors
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the "Software"),
8 * to deal in the Software without restriction, including without limitation
9 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10 * and/or sell copies of the Software, and to permit persons to whom the
11 * Software is furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22 * DEALINGS IN THE SOFTWARE.
23 *****************************************************************************/
24
25#include <stdio.h>
26#include <stdlib.h>
27#include <string.h>
28#include <unistd.h>
29#include <getopt.h>
30#include <signal.h>
31#include <transmission.h>
32#ifdef SYS_BEOS
33#include <kernel/OS.h>
34#define usleep snooze
35#endif
36
37#define USAGE \
38"Usage: %s [options] file.torrent [options]\n\n" \
39"Options:\n" \
40"  -d, --download <int> Maximum download rate (-1 = no limit, default = -1)\n"\
41"  -f, --finish <shell script> Command you wish to run on completion\n" \
42"  -h, --help           Print this help and exit\n" \
43"  -i, --info           Print metainfo and exit\n" \
44"  -n  --nat-traversal  Attempt NAT traversal using NAT-PMP or UPnP IGD\n" \
45"  -p, --port <int>     Port we should listen on (default = %d)\n" \
46"  -s, --scrape         Print counts of seeders/leechers and exit\n" \
47"  -u, --upload <int>   Maximum upload rate (-1 = no limit, default = 20)\n" \
48"  -v, --verbose <int>  Verbose level (0 to 2, default = 0)\n"
49
50static int           showHelp      = 0;
51static int           showInfo      = 0;
52static int           showScrape    = 0;
53static int           verboseLevel  = 0;
54static int           bindPort      = TR_DEFAULT_PORT;
55static int           uploadLimit   = 20;
56static int           downloadLimit = -1;
57static char          * torrentPath = NULL;
58static int           natTraversal  = 0;
59static sig_atomic_t  gotsig        = 0;
60static tr_torrent_t  * tor;
61
62static char          * finishCall   = NULL;
63
64static int  parseCommandLine ( int argc, char ** argv );
65static void sigHandler       ( int signal );
66
67char * getStringRatio( float ratio )
68{
69    static char string[20];
70
71    if( ratio == TR_RATIO_NA )
72        return "n/a";
73    if( ratio == TR_RATIO_INF )
74        return "inf";
75    snprintf( string, sizeof string, "%.3f", ratio );
76    return string;
77}
78
79#define LINEWIDTH 80
80
81int main( int argc, char ** argv )
82{
83    int i, error;
84    tr_handle_t  * h;
85    tr_stat_t    * s;
86    tr_handle_status_t * hstat;
87
88    printf( "Transmission %s (%d) - http://transmission.m0k.org/\n\n",
89            VERSION_STRING, VERSION_REVISION );
90
91    /* Get options */
92    if( parseCommandLine( argc, argv ) )
93    {
94        printf( USAGE, argv[0], TR_DEFAULT_PORT );
95        return EXIT_FAILURE;
96    }
97
98    if( showHelp )
99    {
100        printf( USAGE, argv[0], TR_DEFAULT_PORT );
101        return EXIT_SUCCESS;
102    }
103
104    if( verboseLevel < 0 )
105    {
106        verboseLevel = 0;
107    }
108    else if( verboseLevel > 9 )
109    {
110        verboseLevel = 9;
111    }
112    if( verboseLevel )
113    {
114        static char env[11];
115        sprintf( env, "TR_DEBUG=%d", verboseLevel );
116        putenv( env );
117    }
118
119    if( bindPort < 1 || bindPort > 65535 )
120    {
121        printf( "Invalid port '%d'\n", bindPort );
122        return EXIT_FAILURE;
123    }
124
125    /* Initialize libtransmission */
126    h = tr_init();
127
128    /* Open and parse torrent file */
129    if( !( tor = tr_torrentInit( h, torrentPath, 0, &error ) ) )
130    {
131        printf( "Failed opening torrent file `%s'\n", torrentPath );
132        tr_close( h );
133        return EXIT_FAILURE;
134    }
135
136    if( showInfo )
137    {
138        tr_info_t * info = tr_torrentInfo( tor );
139
140        s = tr_torrentStat( tor );
141
142        /* Print torrent info (quite à la btshowmetainfo) */
143        printf( "hash:     " );
144        for( i = 0; i < SHA_DIGEST_LENGTH; i++ )
145        {
146            printf( "%02x", info->hash[i] );
147        }
148        printf( "\n" );
149        printf( "tracker:  %s:%d\n",
150                s->tracker->address, s->tracker->port );
151        printf( "announce: %s\n", s->tracker->announce );
152        printf( "size:     %"PRIu64" (%"PRIu64" * %d + %"PRIu64")\n",
153                info->totalSize, info->totalSize / info->pieceSize,
154                info->pieceSize, info->totalSize % info->pieceSize );
155        if( info->comment[0] )
156        {
157            printf( "comment:  %s\n", info->comment );
158        }
159        if( info->creator[0] )
160        {
161            printf( "creator:  %s\n", info->creator );
162        }
163        printf( "file(s):\n" );
164        for( i = 0; i < info->fileCount; i++ )
165        {
166            printf( " %s (%"PRIu64")\n", info->files[i].name,
167                    info->files[i].length );
168        }
169
170        goto cleanup;
171    }
172
173    if( showScrape )
174    {
175        int seeders, leechers, downloaded;
176
177        if( tr_torrentScrape( tor, &seeders, &leechers, &downloaded ) )
178        {
179            printf( "Scrape failed.\n" );
180        }
181        else
182        {
183            printf( "%d seeder(s), %d leecher(s), %d download(s).\n",
184                    seeders, leechers, downloaded );
185        }
186
187        goto cleanup;
188    }
189
190    signal( SIGINT, sigHandler );
191
192    tr_setBindPort( h, bindPort );
193    tr_setGlobalUploadLimit( h, uploadLimit );
194    tr_setGlobalDownloadLimit( h, downloadLimit );
195
196    tr_natTraversalEnable( h, natTraversal );
197   
198    tr_torrentSetFolder( tor, "." );
199    tr_torrentStart( tor );
200
201    for( ;; )
202    {
203        char string[LINEWIDTH];
204        int  chars = 0;
205        int result;
206
207        sleep( 1 );
208
209        if( gotsig )
210        {
211            gotsig = 0;
212            tr_torrentStop( tor );
213        }
214
215        s = tr_torrentStat( tor );
216
217        if( s->status & TR_STATUS_PAUSE )
218        {
219            break;
220        }
221        else if( s->status & TR_STATUS_CHECK )
222        {
223            chars = snprintf( string, sizeof string,
224                "Checking files... %.2f %%", 100.0 * s->progress );
225        }
226        else if( s->status & TR_STATUS_DOWNLOAD )
227        {
228            chars = snprintf( string, sizeof string,
229                "Progress: %.2f %%, %d peer%s, dl from %d (%.2f KB/s), "
230                "ul to %d (%.2f KB/s) [%s]", 100.0 * s->progress,
231                s->peersTotal, ( s->peersTotal == 1 ) ? "" : "s",
232                s->peersUploading, s->rateDownload,
233                s->peersDownloading, s->rateUpload,
234                getStringRatio(s->ratio) );
235        }
236        else if( s->status & TR_STATUS_SEED )
237        {
238            chars = snprintf( string, sizeof string,
239                "Seeding, uploading to %d of %d peer(s), %.2f KB/s [%s]",
240                s->peersDownloading, s->peersTotal,
241                s->rateUpload, getStringRatio(s->ratio) );
242        }
243        else if( s->status & TR_STATUS_STOPPING )
244        {
245            chars = snprintf( string, sizeof string, "Stopping..." );
246        }
247        if( ( signed )sizeof string > chars )
248        {
249            memset( &string[chars], ' ', sizeof string - 1 - chars );
250        }
251        string[sizeof string - 1] = '\0';
252        fprintf( stderr, "\r%s", string );
253
254        if( s->error )
255        {
256            fprintf( stderr, "\n%s\n", s->errorString );
257        }
258        else if( verboseLevel > 0 )
259        {
260            fprintf( stderr, "\n" );
261        }
262       
263        if( tr_getFinished( tor ) )
264        {
265            result = system(finishCall);
266        }
267    }
268    fprintf( stderr, "\n" );
269
270    /* Try for 5 seconds to delete any port mappings for nat traversal */
271    tr_natTraversalEnable( h, 0 );
272    for( i = 0; i < 10; i++ )
273    {
274        hstat = tr_handleStatus( h );
275        if( TR_NAT_TRAVERSAL_DISABLED == hstat->natTraversalStatus )
276        {
277            /* Port mappings were deleted */
278            break;
279        }
280        usleep( 500000 );
281    }
282   
283cleanup:
284    tr_torrentClose( h, tor );
285    tr_close( h );
286
287    return EXIT_SUCCESS;
288}
289
290static int parseCommandLine( int argc, char ** argv )
291{
292    for( ;; )
293    {
294        static struct option long_options[] =
295          { { "help",     no_argument,       NULL, 'h' },
296            { "info",     no_argument,       NULL, 'i' },
297            { "scrape",   no_argument,       NULL, 's' },
298            { "verbose",  required_argument, NULL, 'v' },
299            { "port",     required_argument, NULL, 'p' },
300            { "upload",   required_argument, NULL, 'u' },
301            { "download", required_argument, NULL, 'd' },
302            { "finish",   required_argument, NULL, 'f' },
303            { "nat-traversal", no_argument,  NULL, 'n' },
304            { 0, 0, 0, 0} };
305
306        int c, optind = 0;
307        c = getopt_long( argc, argv, "hisv:p:u:d:f:n", long_options, &optind );
308        if( c < 0 )
309        {
310            break;
311        }
312        switch( c )
313        {
314            case 'h':
315                showHelp = 1;
316                break;
317            case 'i':
318                showInfo = 1;
319                break;
320            case 's':
321                showScrape = 1;
322                break;
323            case 'v':
324                verboseLevel = atoi( optarg );
325                break;
326            case 'p':
327                bindPort = atoi( optarg );
328                break;
329            case 'u':
330                uploadLimit = atoi( optarg );
331                break;
332            case 'd':
333                downloadLimit = atoi( optarg );
334                break;
335            case 'f':
336                finishCall = optarg;
337                break;
338            case 'n':
339                natTraversal = 1;
340                break;
341            default:
342                return 1;
343        }
344    }
345
346    if( optind > argc - 1  )
347    {
348        return !showHelp;
349    }
350
351    torrentPath = argv[optind];
352
353    return 0;
354}
355
356static void sigHandler( int signal )
357{
358    switch( signal )
359    {
360        case SIGINT:
361            gotsig = 1;
362            break;
363
364        default:
365            break;
366    }
367}
Note: See TracBrowser for help on using the repository browser.