source: trunk/cli/transmissioncli.c @ 310

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

Add support to libtransmission and the MacOS X GUI for saving private copies of

torrent files.

  • Property svn:keywords set to Date Rev Author Id
File size: 8.8 KB
Line 
1/******************************************************************************
2 * $Id: transmissioncli.c 310 2006-06-09 19:53:35Z 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"  -h, --help           Print this help and exit\n" \
41"  -i, --info           Print metainfo and exit\n" \
42"  -s, --scrape         Print counts of seeders/leechers and exit\n" \
43"  -v, --verbose <int>  Verbose level (0 to 2, default = 0)\n" \
44"  -p, --port <int>     Port we should listen on (default = %d)\n" \
45"  -u, --upload <int>   Maximum upload rate (-1 = no limit, default = 20)\n" \
46"  -d, --download <int> Maximum download rate (-1 = no limit, default = -1)\n" \
47"  -f, --finish <shell script> Command you wish to run on completion\n"
48
49static int           showHelp      = 0;
50static int           showInfo      = 0;
51static int           showScrape    = 0;
52static int           verboseLevel  = 0;
53static int           bindPort      = TR_DEFAULT_PORT;
54static int           uploadLimit   = 20;
55static int           downloadLimit = -1;
56static char          * torrentPath = NULL;
57static volatile char mustDie       = 0;
58
59static char          * finishCall   = NULL;
60
61static int  parseCommandLine ( int argc, char ** argv );
62static void sigHandler       ( int signal );
63
64int main( int argc, char ** argv )
65{
66    int i, error;
67    tr_handle_t  * h;
68    tr_torrent_t * tor;
69    tr_stat_t    * s;
70
71    printf( "Transmission %s (%d) - http://transmission.m0k.org/\n\n",
72            VERSION_STRING, VERSION_REVISION );
73
74    /* Get options */
75    if( parseCommandLine( argc, argv ) )
76    {
77        printf( USAGE, argv[0], TR_DEFAULT_PORT );
78        return 1;
79    }
80
81    if( showHelp )
82    {
83        printf( USAGE, argv[0], TR_DEFAULT_PORT );
84        return 0;
85    }
86
87    if( verboseLevel < 0 )
88    {
89        verboseLevel = 0;
90    }
91    else if( verboseLevel > 9 )
92    {
93        verboseLevel = 9;
94    }
95    if( verboseLevel )
96    {
97        static char env[11];
98        sprintf( env, "TR_DEBUG=%d", verboseLevel );
99        putenv( env );
100    }
101
102    if( bindPort < 1 || bindPort > 65535 )
103    {
104        printf( "Invalid port '%d'\n", bindPort );
105        return 1;
106    }
107
108    /* Initialize libtransmission */
109    h = tr_init();
110
111    /* Open and parse torrent file */
112    if( !( tor = tr_torrentInit( h, torrentPath, 0, &error ) ) )
113    {
114        printf( "Failed opening torrent file `%s'\n", torrentPath );
115        goto failed;
116    }
117
118    if( showInfo )
119    {
120        tr_info_t * info = tr_torrentInfo( tor );
121
122        /* Print torrent info (quite à la btshowmetainfo) */
123        printf( "hash:     " );
124        for( i = 0; i < SHA_DIGEST_LENGTH; i++ )
125        {
126            printf( "%02x", info->hash[i] );
127        }
128        printf( "\n" );
129        printf( "tracker:  %s:%d\n",
130                info->trackerAddress, info->trackerPort );
131        printf( "announce: %s\n", info->trackerAnnounce );
132        printf( "size:     %"PRIu64" (%"PRIu64" * %d + %"PRIu64")\n",
133                info->totalSize, info->totalSize / info->pieceSize,
134                info->pieceSize, info->totalSize % info->pieceSize );
135        printf( "file(s):\n" );
136        for( i = 0; i < info->fileCount; i++ )
137        {
138            printf( " %s (%"PRIu64")\n", info->files[i].name,
139                    info->files[i].length );
140        }
141
142        goto cleanup;
143    }
144
145    if( showScrape )
146    {
147        int seeders, leechers;
148
149        if( tr_torrentScrape( tor, &seeders, &leechers ) )
150        {
151            printf( "Scrape failed.\n" );
152        }
153        else
154        {
155            printf( "%d seeder(s), %d leecher(s).\n", seeders, leechers );
156        }
157
158        goto cleanup;
159    }
160
161    signal( SIGINT, sigHandler );
162
163    tr_setBindPort( h, bindPort );
164    tr_setUploadLimit( h, uploadLimit );
165    tr_setDownloadLimit( h, downloadLimit );
166   
167    tr_torrentSetFolder( tor, "." );
168    tr_torrentStart( tor );
169
170    while( !mustDie )
171    {
172        char string[80];
173        int  chars = 0;
174        int result;
175
176        sleep( 1 );
177
178        s = tr_torrentStat( tor );
179
180        if( s->status & TR_STATUS_CHECK )
181        {
182            chars = snprintf( string, 80,
183                "Checking files... %.2f %%", 100.0 * s->progress );
184        }
185        else if( s->status & TR_STATUS_DOWNLOAD )
186        {
187            chars = snprintf( string, 80,
188                "Progress: %.2f %%, %d peer%s, dl from %d (%.2f KB/s), "
189                "ul to %d (%.2f KB/s)", 100.0 * s->progress,
190                s->peersTotal, ( s->peersTotal == 1 ) ? "" : "s",
191                s->peersUploading, s->rateDownload,
192                s->peersDownloading, s->rateUpload );
193        }
194        else if( s->status & TR_STATUS_SEED )
195        {
196            chars = snprintf( string, 80,
197                "Seeding, uploading to %d of %d peer(s), %.2f KB/s",
198                s->peersDownloading, s->peersTotal,
199                s->rateUpload );
200        }
201        memset( &string[chars], ' ', 79 - chars );
202        string[79] = '\0';
203        fprintf( stderr, "\r%s", string );
204
205        if( s->error & TR_ETRACKER )
206        {
207            fprintf( stderr, "\n%s\n", s->trackerError );
208        }
209        else if( verboseLevel > 0 )
210        {
211            fprintf( stderr, "\n" );
212        }
213       
214        if( tr_getFinished( tor ) )
215        {
216            result = system(finishCall);
217        }
218    }
219    fprintf( stderr, "\n" );
220
221    /* Try for 5 seconds to notice the tracker that we are leaving */
222    tr_torrentStop( tor );
223    for( i = 0; i < 10; i++ )
224    {
225        s = tr_torrentStat( tor );
226        if( s->status & TR_STATUS_PAUSE )
227        {
228            /* The 'stopped' message was sent */
229            break;
230        }
231        usleep( 500000 );
232    }
233   
234cleanup:
235    tr_torrentClose( h, tor );
236
237failed:
238    tr_close( h );
239
240    return 0;
241}
242
243static int parseCommandLine( int argc, char ** argv )
244{
245    for( ;; )
246    {
247        static struct option long_options[] =
248          { { "help",     no_argument,       NULL, 'h' },
249            { "info",     no_argument,       NULL, 'i' },
250            { "scrape",   no_argument,       NULL, 's' },
251            { "verbose",  required_argument, NULL, 'v' },
252            { "port",     required_argument, NULL, 'p' },
253            { "upload",   required_argument, NULL, 'u' },
254            { "download", required_argument, NULL, 'd' },
255            { "finish",   required_argument, NULL, 'f' },
256            { 0, 0, 0, 0} };
257
258        int c, optind = 0;
259        c = getopt_long( argc, argv, "hisv:p:u:d:f:", long_options, &optind );
260        if( c < 0 )
261        {
262            break;
263        }
264        switch( c )
265        {
266            case 'h':
267                showHelp = 1;
268                break;
269            case 'i':
270                showInfo = 1;
271                break;
272            case 's':
273                showScrape = 1;
274                break;
275            case 'v':
276                verboseLevel = atoi( optarg );
277                break;
278            case 'p':
279                bindPort = atoi( optarg );
280                break;
281            case 'u':
282                uploadLimit = atoi( optarg );
283                break;
284            case 'd':
285                downloadLimit = atoi( optarg );
286                break;
287            case 'f':
288                finishCall = optarg;
289                break;
290            default:
291                return 1;
292        }
293    }
294
295    if( optind > argc - 1  )
296    {
297        return !showHelp;
298    }
299
300    torrentPath = argv[optind];
301
302    return 0;
303}
304
305static void sigHandler( int signal )
306{
307    switch( signal )
308    {
309        case SIGINT:
310            mustDie = 1;
311            break;
312
313        default:
314            break;
315    }
316}
Note: See TracBrowser for help on using the repository browser.