release/src/router/transmission/utils/edit.c

   1 /*
   2  * This file Copyright (C) Mnemosyne LLC
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License version 2
   6  * as published by the Free Software Foundation.
   7  *
   8  * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
   9  *
  10  * $Id: edit.c 11791 2011-01-30 16:40:11Z jordan $
  11  */
  12
  13 #include <stdio.h>
  14
  15 #include <event2/buffer.h>
  16
  17 #include <libtransmission/transmission.h>
  18 #include <libtransmission/bencode.h>
  19 #include <libtransmission/tr-getopt.h>
  20 #include <libtransmission/utils.h>
  21 #include <libtransmission/version.h>
  22
  23 #define MY_NAME "transmission-edit"
  24
  25 static int fileCount = 0;
  26 static tr_bool showVersion = FALSE;
  27 static const char ** files = NULL;
  28 static const char * add = NULL;
  29 static const char * deleteme = NULL;
  30 static const char * replace[2] = { NULL, NULL };
  31
  32 static tr_option options[] =
  33 {
  34   { 'a', "add", "Add a tracker's announce URL", "a", 1, "<url>" },
  35   { 'd', "delete", "Delete a tracker's announce URL", "d", 1, "<url>" },
  36   { 'r', "replace", "Search and replace a substring in the announce URLs", "r", 1, "<old> <new>" },
  37   { 'V', "version", "Show version number and exit", "V", 0, NULL },
  38   { 0, NULL, NULL, NULL, 0, NULL }
  39 };
  40
  41 static const char *
  42 getUsage( void )
  43 {
  44     return "Usage: " MY_NAME " [options] torrent-file(s)";
  45 }
  46
  47 static int
  48 parseCommandLine( int argc, const char ** argv )
  49 {
  50     int c;
  51     const char * optarg;
  52
  53     while(( c = tr_getopt( getUsage( ), argc, argv, options, &optarg )))
  54     {
  55         switch( c )
  56         {
  57             case 'a': add = optarg;
  58                       break;
  59             case 'd': deleteme = optarg;
  60                       break;
  61             case 'r': replace[0] = optarg;
  62                       c = tr_getopt( getUsage( ), argc, argv, options, &optarg );
  63                       if( c != TR_OPT_UNK ) return 1;
  64                       replace[1] = optarg;
  65                       break;
  66             case 'V': showVersion = TRUE;
  67                       break;
  68             case TR_OPT_UNK: files[fileCount++] = optarg; break;
  69             default: return 1;
  70         }
  71     }
  72
  73     return 0;
  74 }
  75
  76 static tr_bool
  77 removeURL( tr_benc * metainfo, const char * url )
  78 {
  79     const char * str;
  80     tr_benc * announce_list;
  81     tr_bool changed = FALSE;
  82
  83     if( tr_bencDictFindStr( metainfo, "announce", &str ) && !strcmp( str, url ) )
  84     {
  85         printf( "\tRemoved \"%s\" from \"announce\"\n", str );
  86         tr_bencDictRemove( metainfo, "announce" );
  87         changed = TRUE;
  88     }
  89
  90     if( tr_bencDictFindList( metainfo, "announce-list", &announce_list ) )
  91     {
  92         tr_benc * tier;
  93         int tierIndex = 0;
  94         while(( tier = tr_bencListChild( announce_list, tierIndex )))
  95         {
  96             tr_benc * node;
  97             int nodeIndex = 0;
  98             while(( node = tr_bencListChild( tier, nodeIndex )))
  99             {
 100                 if( tr_bencGetStr( node, &str ) && !strcmp( str, url ) )
 101                 {
 102                     printf( "\tRemoved \"%s\" from \"announce-list\" tier #%d\n", str, (tierIndex+1) );
 103                     tr_bencListRemove( tier, nodeIndex );
 104                     changed = TRUE;
 105                 }
 106                 else ++nodeIndex;
 107             }
 108
 109             if( tr_bencListSize( tier ) == 0 )
 110             {
 111                 printf( "\tNo URLs left in tier #%d... removing tier\n", (tierIndex+1) );
 112                 tr_bencListRemove( announce_list, tierIndex );
 113             }
 114             else ++tierIndex;
 115         }
 116
 117         if( tr_bencListSize( announce_list ) == 0 )
 118         {
 119             printf( "\tNo tiers left... removing announce-list\n" );
 120             tr_bencDictRemove( metainfo, "announce-list" );
 121         }
 122     }
 123
 124     /* if we removed the "announce" field and there's still another track left,
 125      * use it as the "announce" field */
 126     if( changed && !tr_bencDictFindStr( metainfo, "announce", &str ) )
 127     {
 128         tr_benc * tier;
 129         tr_benc * node;
 130
 131         if(( tier = tr_bencListChild( announce_list, 0 ))) {
 132             if(( node = tr_bencListChild( tier, 0 ))) {
 133                 if( tr_bencGetStr( node, &str ) ) {
 134                     tr_bencDictAddStr( metainfo, "announce", str );
 135                     printf( "\tAdded \"%s\" to announce\n", str );
 136                 }
 137             }
 138         }
 139     }
 140
 141     return changed;
 142 }
 143
 144 static char*
 145 replaceSubstr( const char * str, const char * in, const char * out )
 146 {
 147     char * walk;
 148     struct evbuffer * buf = evbuffer_new( );
 149     const size_t inlen = strlen( in );
 150     const size_t outlen = strlen( out );
 151
 152     while(( walk = strstr( str, in )))
 153     {
 154         evbuffer_add( buf, str, walk-str );
 155         evbuffer_add( buf, out, outlen );
 156         str = walk + inlen;
 157     }
 158
 159     evbuffer_add( buf, str, strlen( str ) );
 160
 161     return evbuffer_free_to_str( buf );
 162 }
 163
 164 static tr_bool
 165 replaceURL( tr_benc * metainfo, const char * in, const char * out )
 166 {
 167     const char * str;
 168     tr_benc * announce_list;
 169     tr_bool changed = FALSE;
 170
 171     if( tr_bencDictFindStr( metainfo, "announce", &str ) && strstr( str, in ) )
 172     {
 173         char * newstr = replaceSubstr( str, in, out );
 174         printf( "\tReplaced in \"announce\": \"%s\" --> \"%s\"\n", str, newstr );
 175         tr_bencDictAddStr( metainfo, "announce", newstr );
 176         tr_free( newstr );
 177         changed = TRUE;
 178     }
 179
 180     if( tr_bencDictFindList( metainfo, "announce-list", &announce_list ) )
 181     {
 182         tr_benc * tier;
 183         int tierCount = 0;
 184         while(( tier = tr_bencListChild( announce_list, tierCount++ )))
 185         {
 186             tr_benc * node;
 187             int nodeCount = 0;
 188             while(( node = tr_bencListChild( tier, nodeCount++ )))
 189             {
 190                 if( tr_bencGetStr( node, &str ) && strstr( str, in ) )
 191                 {
 192                     char * newstr = replaceSubstr( str, in, out );
 193                     printf( "\tReplaced in \"announce-list\" tier %d: \"%s\" --> \"%s\"\n", tierCount, str, newstr );
 194                     tr_bencFree( node );
 195                     tr_bencInitStr( node, newstr, -1 );
 196                     tr_free( newstr );
 197                     changed = TRUE;
 198                 }
 199             }
 200         }
 201     }
 202
 203     return changed;
 204 }
 205
 206 static tr_bool
 207 addURL( tr_benc * metainfo, const char * url )
 208 {
 209     const char * str;
 210     tr_benc * announce_list;
 211     tr_bool changed = FALSE;
 212     tr_bool match = FALSE;
 213
 214     /* maybe add it to "announce" */
 215     if( !tr_bencDictFindStr( metainfo, "announce", &str ) )
 216     {
 217         printf( "\tAdded \"%s\" in \"announce\"\n", url );
 218         tr_bencDictAddStr( metainfo, "announce", url );
 219         changed = TRUE;
 220     }
 221
 222     /* see if it's already in announce-list */
 223     if( tr_bencDictFindList( metainfo, "announce-list", &announce_list ) ) {
 224         tr_benc * tier;
 225         int tierCount = 0;
 226         while(( tier = tr_bencListChild( announce_list, tierCount++ ))) {
 227             tr_benc * node;
 228             int nodeCount = 0;
 229             while(( node = tr_bencListChild( tier, nodeCount++ )))
 230                 if( tr_bencGetStr( node, &str ) && !strcmp( str, url ) )
 231                     match = TRUE;
 232         }
 233     }
 234
 235     /* if it's not in announce-list, add it now */
 236     if( !match )
 237     {
 238         tr_benc * tier;
 239
 240         if( !tr_bencDictFindList( metainfo, "announce-list", &announce_list ) )
 241             announce_list = tr_bencDictAddList( metainfo, "announce-list", 1 );
 242
 243         tier = tr_bencListAddList( announce_list, 1 );
 244         tr_bencListAddStr( tier, url );
 245         printf( "\tAdded \"%s\" to \"announce-list\" tier %zu\n", url, tr_bencListSize( announce_list ) );
 246         changed = TRUE;
 247     }
 248
 249     return changed;
 250 }
 251
 252 int
 253 main( int argc, char * argv[] )
 254 {
 255     int i;
 256     int changedCount = 0;
 257
 258     files = tr_new0( const char*, argc );
 259
 260     tr_setMessageLevel( TR_MSG_ERR );
 261
 262     if( parseCommandLine( argc, (const char**)argv ) )
 263         return EXIT_FAILURE;
 264
 265     if( showVersion )
 266     {
 267         fprintf( stderr, MY_NAME" "LONG_VERSION_STRING"\n" );
 268         return 0;
 269     }
 270
 271     if( fileCount < 1 )
 272     {
 273         fprintf( stderr, "ERROR: No torrent files specified.\n" );
 274         tr_getopt_usage( MY_NAME, getUsage( ), options );
 275         fprintf( stderr, "\n" );
 276         return EXIT_FAILURE;
 277     }
 278
 279     if( !add && !deleteme && !replace[0] )
 280     {
 281         fprintf( stderr, "ERROR: Must specify -a, -d or -r\n" );
 282         tr_getopt_usage( MY_NAME, getUsage( ), options );
 283         fprintf( stderr, "\n" );
 284         return EXIT_FAILURE;
 285     }
 286
 287     for( i=0; i<fileCount; ++i )
 288     {
 289         tr_benc top;
 290         tr_bool changed = FALSE;
 291         const char * filename = files[i];
 292
 293         printf( "%s\n", filename );
 294
 295         if( tr_bencLoadFile( &top, TR_FMT_BENC, filename ) )
 296         {
 297             printf( "\tError reading file\n" );
 298             continue;
 299         }
 300
 301         if( deleteme != NULL )
 302             changed |= removeURL( &top, deleteme );
 303
 304         if( add != NULL )
 305             changed = addURL( &top, add );
 306
 307         if( replace[0] && replace[1] )
 308             changed |= replaceURL( &top, replace[0], replace[1] );
 309
 310         if( changed )
 311         {
 312             ++changedCount;
 313             tr_bencToFile( &top, TR_FMT_BENC, filename );
 314         }
 315
 316         tr_bencFree( &top );
 317     }
 318
 319     printf( "Changed %d files\n", changedCount );
 320
 321     tr_free( files );
 322     return 0;
 323 }