source: trunk/transmissioncli.c @ 7

Last change on this file since 7 was 7, checked in by root, 17 years ago

Update 2005-11-24

File size: 8.2 KB
Line 
1/******************************************************************************
2 * Copyright (c) 2005 Eric Petit
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
19 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
20 * DEALINGS IN THE SOFTWARE.
21 *****************************************************************************/
22
23#include <stdio.h>
24#include <stdlib.h>
25#include <string.h>
26#include <unistd.h>
27#include <getopt.h>
28#include <signal.h>
29#include <transmission.h>
30#ifdef SYS_BEOS
31#include <kernel/OS.h>
32#define usleep snooze
33#endif
34
35#define USAGE \
36"Usage: %s [options] file.torrent [options]\n\n" \
37"Options:\n" \
38"  -h, --help           Print this help and exit\n" \
39"  -i, --info           Print metainfo and exit\n" \
40"  -s, --scrape         Print counts of seeders/leechers and exit\n" \
41"  -v, --verbose <int>  Verbose level (0 to 2, default = 0)\n" \
42"  -p, --port <int>     Port we should listen on (default = %d)\n" \
43"  -u, --upload <int>   Maximum upload rate (-1 = no limit, default = 20)\n"
44
45static int             showHelp     = 0;
46static int             showInfo     = 0;
47static int             showScrape   = 0;
48static int             verboseLevel = 0;
49static int             bindPort     = TR_DEFAULT_PORT;
50static int             uploadLimit  = 20;
51static char          * torrentPath  = NULL;
52static volatile char   mustDie      = 0;
53
54static int  parseCommandLine ( int argc, char ** argv );
55static void sigHandler       ( int signal );
56
57int main( int argc, char ** argv )
58{
59    int           i, count;
60    tr_handle_t * h;
61    tr_stat_t   * s;
62
63    printf( "Transmission %s - http://transmission.m0k.org/\n\n",
64            VERSION_STRING );
65
66    /* Get options */
67    if( parseCommandLine( argc, argv ) )
68    {
69        printf( USAGE, argv[0], TR_DEFAULT_PORT );
70        return 1;
71    }
72
73    if( showHelp )
74    {
75        printf( USAGE, argv[0], TR_DEFAULT_PORT );
76        return 0;
77    }
78
79    if( verboseLevel < 0 )
80    {
81        verboseLevel = 0;
82    }
83    else if( verboseLevel > 9 )
84    {
85        verboseLevel = 9;
86    }
87    if( verboseLevel )
88    {
89        static char env[11];
90        sprintf( env, "TR_DEBUG=%d", verboseLevel );
91        putenv( env );
92    }
93
94    if( bindPort < 1 || bindPort > 65535 )
95    {
96        printf( "Invalid port '%d'\n", bindPort );
97        return 1;
98    }
99
100    /* Initialize libtransmission */
101    h = tr_init();
102
103    /* Open and parse torrent file */
104    if( tr_torrentInit( h, torrentPath ) )
105    {
106        printf( "Failed opening torrent file `%s'\n", torrentPath );
107        goto failed;
108    }
109
110    if( showInfo )
111    {
112        tr_info_t * info;
113
114        count = tr_torrentStat( h, &s );
115        info  = &s[0].info;
116
117        /* Print torrent info (quite à la btshowmetainfo) */
118        printf( "hash:     " );
119        for( i = 0; i < SHA_DIGEST_LENGTH; i++ )
120        {
121            printf( "%02x", info->hash[i] );
122        }
123        printf( "\n" );
124        printf( "tracker:  %s:%d\n",
125                info->trackerAddress, info->trackerPort );
126        printf( "announce: %s\n", info->trackerAnnounce );
127        printf( "size:     %lld (%lld * %d + %lld)\n",
128                info->totalSize, info->totalSize / info->pieceSize,
129                info->pieceSize, info->totalSize % info->pieceSize );
130        printf( "file(s):\n" );
131        for( i = 0; i < info->fileCount; i++ )
132        {
133            printf( " %s (%lld)\n", info->files[i].name,
134                    info->files[i].length );
135        }
136
137        free( s );
138        goto cleanup;
139    }
140
141    if( showScrape )
142    {
143        int seeders, leechers;
144
145        if( tr_torrentScrape( h, 0, &seeders, &leechers ) )
146        {
147            printf( "Scrape failed.\n" );
148        }
149        else
150        {
151            printf( "%d seeder(s), %d leecher(s).\n", seeders, leechers );
152        }
153
154        goto cleanup;
155    }
156
157    signal( SIGINT, sigHandler );
158
159    tr_setBindPort( h, bindPort );
160    tr_setUploadLimit( h, uploadLimit );
161   
162    tr_torrentSetFolder( h, 0, "." );
163    tr_torrentStart( h, 0 );
164
165    while( !mustDie )
166    {
167        char string[80];
168        int  chars = 0;
169
170        sleep( 1 );
171
172        count = tr_torrentStat( h, &s );
173
174        if( s[0].status & TR_STATUS_CHECK )
175        {
176            chars = snprintf( string, 80,
177                "Checking files... %.2f %%", 100.0 * s[0].progress );
178        }
179        else if( s[0].status & TR_STATUS_DOWNLOAD )
180        {
181            chars = snprintf( string, 80,
182                "Progress: %.2f %%, %d peer%s, dl from %d (%.2f kbps), "
183                "ul to %d (%.2f kbps)", 100.0 * s[0].progress,
184                s[0].peersTotal, ( s[0].peersTotal == 1 ) ? "" : "s",
185                s[0].peersUploading, s[0].rateDownload,
186                s[0].peersDownloading, s[0].rateUpload );
187        }
188        else if( s[0].status & TR_STATUS_SEED )
189        {
190            chars = snprintf( string, 80,
191                "Seeding, uploading to %d of %d peer(s), %.2f kbps",
192                s[0].peersDownloading, s[0].peersTotal,
193                s[0].rateUpload );
194        }
195        memset( &string[chars], ' ', 79 - chars );
196        string[79] = '\0';
197        fprintf( stderr, "\r%s", string );
198
199        if( s[0].status & TR_TRACKER_ERROR )
200        {
201            fprintf( stderr, "\n%s\n", s[0].error );
202        }
203        else if( verboseLevel > 0 )
204        {
205            fprintf( stderr, "\n" );
206        }
207
208        free( s );
209    }
210    fprintf( stderr, "\n" );
211
212    /* Try for 5 seconds to notice the tracker that we are leaving */
213    tr_torrentStop( h, 0 );
214    for( i = 0; i < 10; i++ )
215    {
216        count = tr_torrentStat( h, &s );
217        if( s[0].status & TR_STATUS_PAUSE )
218        {
219            /* The 'stopped' message was sent */
220            free( s );
221            break;
222        }
223        free( s );
224        usleep( 500000 );
225    }
226   
227cleanup:
228    tr_torrentClose( h, 0 );
229
230failed:
231    tr_close( h );
232
233    return 0;
234}
235
236static int parseCommandLine( int argc, char ** argv )
237{
238    for( ;; )
239    {
240        static struct option long_options[] =
241          { { "help",    no_argument,       NULL, 'h' },
242            { "info",    no_argument,       NULL, 'i' },
243            { "scrape",  no_argument,       NULL, 's' },
244            { "verbose", required_argument, NULL, 'v' },
245            { "port",    required_argument, NULL, 'p' },
246            { "upload",  required_argument, NULL, 'u' },
247            { 0, 0, 0, 0 } };
248
249        int c, optind = 0;
250        c = getopt_long( argc, argv, "hisv:p:u:", long_options, &optind );
251        if( c < 0 )
252        {
253            break;
254        }
255        switch( c )
256        {
257            case 'h':
258                showHelp = 1;
259                break;
260            case 'i':
261                showInfo = 1;
262                break;
263            case 's':
264                showScrape = 1;
265                break;
266            case 'v':
267                verboseLevel = atoi( optarg );
268                break;
269            case 'p':
270                bindPort = atoi( optarg );
271                break;
272            case 'u':
273                uploadLimit = atoi( optarg );
274                break;
275            default:
276                return 1;
277        }
278    }
279
280    if( optind > argc - 1  )
281    {
282        return !showHelp;
283    }
284
285    torrentPath = argv[optind];
286
287    return 0;
288}
289
290static void sigHandler( int signal )
291{
292    switch( signal )
293    {
294        case SIGINT:
295            mustDie = 1;
296            break;
297
298        default:
299            break;
300    }
301}
Note: See TracBrowser for help on using the repository browser.