summaryrefslogtreecommitdiffstats
path: root/ztest.cc
diff options
context:
space:
mode:
Diffstat (limited to 'ztest.cc')
-rw-r--r--ztest.cc78
1 files changed, 44 insertions, 34 deletions
diff --git a/ztest.cc b/ztest.cc
index b9971ba..b731191 100644
--- a/ztest.cc
+++ b/ztest.cc
@@ -1,5 +1,5 @@
/* Ztest - verify integrity of compressed files
- Copyright (C) 2010, 2011 Antonio Diaz Diaz.
+ Copyright (C) 2010, 2011, 2012, 2013 Antonio Diaz Diaz.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
@@ -15,32 +15,39 @@
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
-void show_ztest_help() throw()
+void show_ztest_help()
{
- std::printf( "Ztest verifies the integrity of the specified compressed files.\n" );
- std::printf( "Non-compressed files are ignored. If no files are specified, the\n" );
- std::printf( "integrity of compressed data read from standard input is verified. Data\n" );
- std::printf( "read from standard input must be all compressed with the same compressor.\n" );
- std::printf( "The supported compressors are bzip2, gzip, lzip and xz.\n" );
- std::printf( "\nUsage: ztest [options] [files]\n" );
- std::printf( "\nExit status is 2 if any compressed file is corrupt, 0 otherwise.\n" );
- std::printf( "\nOptions:\n" );
- std::printf( " -h, --help display this help and exit\n" );
- std::printf( " -V, --version output version information and exit\n" );
- std::printf( " -q, --quiet suppress all messages\n" );
- std::printf( " -r, --recursive operate recursively on directories\n" );
- std::printf( " -v, --verbose be verbose (a 2nd -v gives more)\n" );
+ std::printf( "Ztest verifies the integrity of the specified compressed files.\n"
+ "Non-compressed files are ignored. If no files are specified, the\n"
+ "integrity of compressed data read from standard input is verified. Data\n"
+ "read from standard input must be all in the same compression format.\n"
+ "\nThe supported formats are bzip2, gzip, lzip and xz.\n"
+ "\nNote that some xz files lack integrity information, and therefore can't\n"
+ "be verified as reliably as the other formats can.\n"
+ "\nUsage: ztest [options] [files]\n"
+ "\nExit status is 0 if all compressed files verify OK, 1 if environmental\n"
+ "problems (file not found, invalid flags, I/O errors, etc), 2 if any\n"
+ "compressed file is corrupt or invalid.\n"
+ "\nOptions:\n"
+ " -h, --help display this help and exit\n"
+ " -V, --version output version information and exit\n"
+ " --format=<fmt> force given format (bz2, gz, lz, xz)\n"
+ " -q, --quiet suppress all messages\n"
+ " -r, --recursive operate recursively on directories\n"
+ " -v, --verbose be verbose (a 2nd -v gives more)\n" );
show_help_addr();
}
-int ztest_stdin( const int infd,
+int ztest_stdin( const int infd, const int format_type,
const std::vector< const char * > & ztest_args )
{
- std::string file_type;
- const uint8_t * magic_data;
+ const uint8_t * magic_data = 0;
int magic_size = 0;
- if( !test_format( infd, file_type, &magic_data, &magic_size ) )
+ const char * const decompressor_name = ( format_type >= 0 ) ?
+ decompressor_names[format_type] :
+ test_format( infd, &magic_data, &magic_size );
+ if( !decompressor_name )
{ show_error( "Unknown data format read from stdin." ); return 2; }
int fda[2]; // pipe from feeder
if( pipe( fda ) < 0 )
@@ -53,19 +60,19 @@ int ztest_stdin( const int infd,
close( fda[0] ) == 0 && close( fda[1] ) == 0 )
{
const char ** const argv = new const char *[ztest_args.size()+3];
- argv[0] = file_type.c_str();
- for( unsigned int i = 0; i < ztest_args.size(); ++i )
+ argv[0] = decompressor_name;
+ for( unsigned i = 0; i < ztest_args.size(); ++i )
argv[i+1] = ztest_args[i];
argv[ztest_args.size()+1] = "-t";
argv[ztest_args.size()+2] = 0;
execvp( argv[0], (char **)argv );
}
- show_exec_error( file_type.c_str() );
+ show_exec_error( decompressor_name );
_exit( 1 );
}
// parent
if( pid < 0 )
- { show_fork_error( file_type.c_str() ); return 1; }
+ { show_fork_error( decompressor_name ); return 1; }
const pid_t pid2 = fork();
if( pid2 == 0 ) // child2 (decompressor feeder)
@@ -82,40 +89,43 @@ int ztest_stdin( const int infd,
{ show_fork_error( "decompressor feeder" ); return 1; }
close( fda[0] ); close( fda[1] );
- int retval = wait_for_child( pid, file_type.c_str(), 1 );
+ int retval = wait_for_child( pid, decompressor_name, 1 );
if( retval == 0 && wait_for_child( pid2, "decompressor feeder" ) != 0 )
retval = 1;
return retval;
}
-int ztest_file( const int infd, const std::string & input_filename,
+int ztest_file( const int infd, const int format_type,
+ const std::string & input_filename,
const std::vector< const char * > & ztest_args )
{
- std::string file_type;
- const uint8_t * magic_data;
+ const uint8_t * magic_data = 0;
int magic_size = 0;
- if( !test_format( infd, file_type, &magic_data, &magic_size ) )
- return 0; // ignored
+ const char * const decompressor_name = ( format_type >= 0 ) ?
+ decompressor_names[format_type] :
+ test_format( infd, &magic_data, &magic_size );
+ if( !decompressor_name )
+ return 0; // skip this file
const pid_t pid = fork();
if( pid == 0 ) // child (decompressor)
{
const char ** const argv = new const char *[ztest_args.size()+5];
- argv[0] = file_type.c_str();
- for( unsigned int i = 0; i < ztest_args.size(); ++i )
+ argv[0] = decompressor_name;
+ for( unsigned i = 0; i < ztest_args.size(); ++i )
argv[i+1] = ztest_args[i];
argv[ztest_args.size()+1] = "-t";
argv[ztest_args.size()+2] = "--";
argv[ztest_args.size()+3] = input_filename.c_str();
argv[ztest_args.size()+4] = 0;
execvp( argv[0], (char **)argv );
- show_exec_error( file_type.c_str() );
+ show_exec_error( decompressor_name );
_exit( 1 );
}
// parent
if( pid < 0 )
- { show_fork_error( file_type.c_str() ); return 1; }
+ { show_fork_error( decompressor_name ); return 1; }
- return wait_for_child( pid, file_type.c_str(), 1 );
+ return wait_for_child( pid, decompressor_name, 1 );
}