gboolean
guc_download_new(const char *filename,
const char *uri,
filesize_t size,
const host_addr_t addr,
guint16 port,
const struct guid *guid,
const char *hostname,
const struct sha1 *sha1,
const struct tth *tth,
time_t stamp,
fileinfo_t *fi,
gnet_host_vec_t *proxies,
guint32 flags,
const char *parq_id)
{
return download_new(filename,
uri,
size,
addr,
port,
guid,
hostname,
sha1,
tth,
stamp,
fi,
proxies,
flags,
parq_id);
}
/* main */
int main(int argc, char * argv[])
{
DownloadPrefs prefs;
int o;
int cnt;
char const * p;
char http[256] = "";
uint16_t port;
Download * download;
if(setlocale(LC_ALL, "") == NULL)
_error("setlocale", 1);
bindtextdomain(PACKAGE, LOCALEDIR);
textdomain(PACKAGE);
memset(&prefs, 0, sizeof(prefs));
if(g_thread_supported() == FALSE)
g_thread_init(NULL);
gtk_init(&argc, &argv);
while((o = getopt(argc, argv, "O:U:x")) != -1)
switch(o)
{
case 'O':
prefs.output = optarg;
break;
case 'U':
prefs.user_agent = optarg;
break;
case 'x':
prefs.embedded = 1;
break;
default:
return _usage();
}
if((cnt = argc - optind) == 0)
return _usage();
if((p = getenv("http_proxy")) != NULL
&& sscanf(p, "http://%255[^:]:%hu", http, &port) == 2)
http[sizeof(http) - 1] = '\0';
else
http[0] = '\0';
for(o = 0; o < cnt; o++)
if((download = download_new(&prefs, argv[optind + o])) != NULL)
_download_set_proxy(download, http, port);
gtk_main();
return 0;
}
int download( int argc, char *argv[] )
{
char fn[MAX_STRING] = "";
int do_search = 0;
search_t *search;
conf_t conf[1];
download_t *download;
int i, j, cur_head = 0;
char *s;
#ifdef I18N
setlocale( LC_ALL, "" );
bindtextdomain( PACKAGE, LOCALE );
textdomain( PACKAGE );
#endif
if( !conf_init( conf ) )
{
return( 1 );
}
opterr = 0;
j = -1;
while( 1 )
{
int option;
option = getopt_long( argc, argv, "s:n:o:S::NqvhVaH:U:", download_options, NULL );
if( option == -1 )
break;
switch( option )
{
case 'U':
strncpy( conf->user_agent, optarg, MAX_STRING);
break;
case 'H':
strncpy( conf->add_header[cur_head++], optarg, MAX_STRING );
break;
case 's':
if( !sscanf( optarg, "%i", &conf->max_speed ) )
{
print_help();
return( 1 );
}
break;
case 'n':
if( !sscanf( optarg, "%i", &conf->num_connections ) )
{
print_help();
return( 1 );
}
break;
case 'o':
strncpy( fn, optarg, MAX_STRING );
break;
case 'S':
do_search = 1;
if( optarg != NULL )
if( !sscanf( optarg, "%i", &conf->search_top ) )
{
print_help();
return( 1 );
}
break;
case 'a':
conf->alternate_output = 1;
break;
case 'N':
*conf->http_proxy = 0;
break;
case 'h':
print_help();
return( 0 );
case 'v':
if( j == -1 )
j = 1;
else
j ++;
break;
case 'V':
print_version();
return( 0 );
case 'q':
close( 1 );
conf->verbose = -1;
if( open( "/dev/null", O_WRONLY ) != 1 )
{
fprintf( stderr, _("Can't redirect stdout to /dev/null.\n") );
return( 1 );
}
break;
default:
print_help();
return( 1 );
}
}
conf->add_header_count = cur_head;
if( j > -1 )
conf->verbose = j;
if( argc - optind == 0 )
{
print_help();
return( 1 );
}
else if( strcmp( argv[optind], "-" ) == 0 )
{
s = (char *)malloc( MAX_STRING );
if (scanf( "%1024[^\n]s", s) != 1) {
fprintf( stderr, _("Error when trying to read URL (Too long?).\n") );
return( 1 );
}
}
else
{
s = argv[optind];
if( strlen( s ) > MAX_STRING )
{
fprintf( stderr, _("Can't handle URLs of length over %d\n" ), MAX_STRING );
return( 1 );
}
}
printf( _("Initializing download: %s\n"), s );
if( do_search )
{
search = (search_t*)malloc( sizeof( search_t ) * ( conf->search_amount + 1 ) );
memset( search, 0, sizeof( search_t ) * ( conf->search_amount + 1 ) );
search[0].conf = conf;
if( conf->verbose )
printf( _("Doing search...\n") );
i = search_makelist( search, s );
if( i < 0 )
{
fprintf( stderr, _("File not found\n" ) );
return( 1 );
}
if( conf->verbose )
printf( _("Testing speeds, this can take a while...\n") );
j = search_getspeeds( search, i );
search_sortlist( search, i );
if( conf->verbose )
{
printf( _("%i usable servers found, will use these URLs:\n"), j );
j = min( j, conf->search_top );
printf( "%-60s %15s\n", "URL", "Speed" );
for( i = 0; i < j; i ++ )
printf( "%-70.70s %5i\n", search[i].url, search[i].speed );
printf( "\n" );
}
download = download_new( conf, j, search );
free( search );
if( download->ready == -1 )
{
print_messages( download );
download_close( download );
return( 1 );
}
}
else if( argc - optind == 1 )
{
download = download_new( conf, 0, s );
if( download->ready == -1 )
{
print_messages( download );
download_close( download );
return( 1 );
}
}
else
{
search = (search_t*)malloc( sizeof( search_t ) * ( argc - optind ) );
memset( search, 0, sizeof( search_t ) * ( argc - optind ) );
for( i = 0; i < ( argc - optind ); i ++ )
strncpy( search[i].url, argv[optind+i], MAX_STRING );
download = download_new( conf, argc - optind, search );
free( search );
if( download->ready == -1 )
{
print_messages( download );
download_close( download );
return( 1 );
}
}
print_messages( download );
if( s != argv[optind] )
{
free( s );
}
if( *fn )
{
struct stat buf;
if( stat( fn, &buf ) == 0 )
{
if( S_ISDIR( buf.st_mode ) )
{
size_t fnlen = strlen(fn);
size_t downloadfnlen = strlen(download->filename);
if (fnlen + 1 + downloadfnlen + 1 > MAX_STRING) {
fprintf( stderr, _("Filename too long!\n"));
return ( 1 );
}
fn[fnlen] = '/';
memcpy(fn+fnlen+1, download->filename, downloadfnlen);
fn[fnlen + 1 + downloadfnlen] = '\0';
}
}
sprintf( string, "%s.st", fn );
if( access( fn, F_OK ) == 0 ) if( access( string, F_OK ) != 0 )
{
fprintf( stderr, _("No state file, cannot resume!\n") );
return( 1 );
}
if( access( string, F_OK ) == 0 ) if( access( fn, F_OK ) != 0 )
{
printf( _("State file found, but no downloaded data. Starting from scratch.\n" ) );
unlink( string );
}
strcpy( download->filename, fn );
}
else
{
/* Local file existence check */
i = 0;
s = download->filename + strlen( download->filename );
while( 1 )
{
sprintf( string, "%s.st", download->filename );
if( access( download->filename, F_OK ) == 0 )
{
if( download->conn[0].supported )
{
if( access( string, F_OK ) == 0 )
break;
}
}
else
{
if( access( string, F_OK ) )
break;
}
sprintf( s, ".%i", i );
i ++;
}
}
if( !download_open( download ) )
{
print_messages( download );
return( 1 );
}
print_messages( download );
download_start( download );
print_messages( download );
if( conf->alternate_output )
{
putchar('\n');
}
else
{
if( download->bytes_done > 0 ) /* Print first dots if resuming */
{
putchar( '\n' );
print_commas( download->bytes_done );
}
}
download->start_byte = download->bytes_done;
/* Install save_state signal handler for resuming support */
//signal( SIGINT, stop );
//signal( SIGTERM, stop );
while( !download->ready && run )
{
long long int prev, done;
prev = download->bytes_done;
download_do( download );
total = download->size;
if( conf->alternate_output )
{
if( !download->message && prev != download->bytes_done )
print_alternate_output( download );
}
else
{
/* The infamous wget-like 'interface'.. ;) */
done = ( download->bytes_done / 1024 ) - ( prev / 1024 );
if( done && conf->verbose > -1 )
{
for( i = 0; i < done; i ++ )
{
i += ( prev / 1024 );
if( ( i % 50 ) == 0 )
{
if( prev >= 1024 )
{
speed= download->bytes_per_second;
printf( " [%6.1fKB/s]", (double) download->bytes_per_second / 1024 );
}
if( download->size < 10240000 )
percent=min( 100, 102400 * (float)i / download->size );
//printf( "\n[%3lld%%] ", min( 100, 102400 * i / download->size ) );
else
percent=min( 100, (float)i / ( download->size / 102400 ) );
printf( "\n[%.2f%%] ", percent);
}
else if( ( i % 10 ) == 0 )
{
putchar( ' ' );
}
putchar( '.' );
i -= ( prev / 1024 );
show->setDownValue(percent,speed,total);
}
fflush( stdout );
}
}
if( download->message )
{
if(conf->alternate_output==1)
{
/* clreol-simulation */
putchar( '\r' );
for( i = 0; i < 79; i++ ) /* linewidth known? */
putchar( ' ' );
putchar( '\r' );
}
else
{
putchar( '\n' );
}
print_messages( download );
if( !download->ready )
{
if(conf->alternate_output!=1)
print_commas( download->bytes_done );
else
print_alternate_output(download);
}
}
else if( download->ready )
{
putchar( '\n' );
}
}
if(!run)
{
download_close( download );
close( 1 );
conf->verbose = -1;
return( 0 );
}
strcpy( string + MAX_STRING / 2,
size_human( download->bytes_done - download->start_byte ) );
printf( _("\nDownloaded %s in %s. (%.2f KB/s)\n"),
string + MAX_STRING / 2,
time_human( gettime() - download->start_time ),
(double) download->bytes_per_second / 1024 );
i = download->ready ? 0 : 2;
download_close( download );
return( i );
}
