summaryrefslogtreecommitdiffstats
path: root/main.cc
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2023-06-16 11:13:24 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2023-06-16 11:13:24 +0000
commitd7c278fca708bc2c4badaeac041ecb136a4ef955 (patch)
tree5b163b83e6e3171a311f67187aa75c0f96f98f13 /main.cc
parentAdding upstream version 1.23. (diff)
downloadlziprecover-d7c278fca708bc2c4badaeac041ecb136a4ef955.tar.xz
lziprecover-d7c278fca708bc2c4badaeac041ecb136a4ef955.zip
Adding upstream version 1.24~pre1.upstream/1.24_pre1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r--main.cc321
1 files changed, 166 insertions, 155 deletions
diff --git a/main.cc b/main.cc
index 72f415e..ed8f1ed 100644
--- a/main.cc
+++ b/main.cc
@@ -1,5 +1,5 @@
/* Lziprecover - Data recovery tool for the lzip format
- Copyright (C) 2009-2022 Antonio Diaz Diaz.
+ Copyright (C) 2009-2023 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
@@ -16,9 +16,9 @@
*/
/*
Exit status: 0 for a normal exit, 1 for environmental problems
- (file not found, invalid flags, I/O errors, etc), 2 to indicate a
- corrupt or invalid input file, 3 for an internal consistency error
- (e.g., bug) which caused lziprecover to panic.
+ (file not found, invalid command line options, I/O errors, etc), 2 to
+ indicate a corrupt or invalid input file, 3 for an internal consistency
+ error (e.g., bug) which caused lziprecover to panic.
*/
#define _FILE_OFFSET_BITS 64
@@ -26,7 +26,7 @@
#include <algorithm>
#include <cctype>
#include <cerrno>
-#include <climits>
+#include <climits> // SSIZE_MAX
#include <csignal>
#include <cstdio>
#include <cstdlib>
@@ -35,7 +35,7 @@
#include <string>
#include <vector>
#include <fcntl.h>
-#include <stdint.h>
+#include <stdint.h> // SIZE_MAX
#include <unistd.h>
#include <utime.h>
#include <sys/stat.h>
@@ -71,11 +71,15 @@
#error "Environments where CHAR_BIT != 8 are not supported."
#endif
-#if ( defined SIZE_MAX && SIZE_MAX < UINT_MAX ) || \
- ( defined SSIZE_MAX && SSIZE_MAX < INT_MAX )
-#error "Environments where 'size_t' is narrower than 'int' are not supported."
+#if ( defined SIZE_MAX && SIZE_MAX < ULONG_MAX ) || \
+ ( defined SSIZE_MAX && SSIZE_MAX < LONG_MAX )
+#error "Environments where 'size_t' is narrower than 'long' are not supported."
#endif
+bool fits_in_size_t( const unsigned long long size )
+ { return ( sizeof (long) <= sizeof (size_t) && size <= LONG_MAX ) ||
+ ( sizeof (int) <= sizeof (size_t) && size <= INT_MAX ); }
+
int verbosity = 0;
const char * const program_name = "lziprecover";
@@ -91,11 +95,11 @@ const struct { const char * from; const char * to; } known_extensions[] = {
{ ".tlz", ".tar" },
{ 0, 0 } };
-enum Mode { m_none, m_alone_to_lz, m_debug_decompress, m_debug_delay,
- m_debug_repair, m_decompress, m_dump, m_list, m_md5sum, m_merge,
- m_nrep_stats, m_range_dec, m_remove, m_repair, m_reproduce,
- m_show_packets, m_split, m_strip, m_test, m_unzcrash_bit,
- m_unzcrash_block };
+enum Mode { m_none, m_alone_to_lz, m_byte_repair, m_clear_marking,
+ m_debug_byte_repair, m_debug_decompress, m_debug_delay,
+ m_decompress, m_dump, m_list, m_md5sum, m_merge, m_nrep_stats,
+ m_range_dec, m_remove, m_reproduce, m_show_packets, m_split,
+ m_strip, m_test, m_unzcrash_bit, m_unzcrash_block };
/* Variable used in signal handler context.
It is not declared volatile because the handler never returns. */
@@ -127,7 +131,7 @@ void show_help()
" -a, --trailing-error exit with error status if trailing data\n"
" -A, --alone-to-lz convert lzma-alone files to lzip format\n"
" -c, --stdout write to standard output, keep input files\n"
- " -d, --decompress decompress\n"
+ " -d, --decompress decompress, test compressed file integrity\n"
" -D, --range-decompress=<n-m> decompress a range of bytes to stdout\n"
" -e, --reproduce try to reproduce a zeroed sector in file\n"
" --lzip-level=N|a|m[N] reproduce one level, all, or match length\n"
@@ -140,14 +144,17 @@ void show_help()
" -m, --merge correct errors in file using several copies\n"
" -o, --output=<file> place the output into <file>\n"
" -q, --quiet suppress all messages\n"
- " -R, --repair try to repair a small error in file\n"
+ " -R, --byte-repair try to repair a corrupt byte in file\n"
" -s, --split split multimember file in single-member files\n"
" -t, --test test compressed file integrity\n"
" -v, --verbose be verbose (a 2nd -v gives more)\n"
+ " --dump=<list>:d:e:t dump members, damaged/empty, tdata to stdout\n"
+ " --remove=<list>:d:e:t remove members, tdata from files in place\n"
+ " --strip=<list>:d:e:t copy files to stdout stripping members given\n"
+ " --empty-error exit with error status if empty member in file\n"
+ " --marking-error exit with error status if 1st LZMA byte not 0\n"
" --loose-trailing allow trailing data seeming corrupt header\n"
- " --dump=<list>:d:t dump members listed/damaged, tdata to stdout\n"
- " --remove=<list>:d:t remove members, tdata from files in place\n"
- " --strip=<list>:d:t copy files to stdout stripping members given\n" );
+ " --clear-marking reset the first LZMA byte of each member\n" );
if( verbosity >= 1 )
{
std::printf( "\nDebug options for experts:\n"
@@ -158,7 +165,7 @@ void show_help()
" -W, --debug-decompress=<pos>,<val> set pos to val and decompress to stdout\n"
" -X, --show-packets[=<pos>,<val>] show in stdout the decoded LZMA packets\n"
" -Y, --debug-delay=<range> find max error detection delay in <range>\n"
- " -Z, --debug-repair=<pos>,<val> test repair one-byte error at <pos>\n" );
+ " -Z, --debug-byte-repair=<pos>,<val> test repair one-byte error at <pos>\n" );
}
std::printf( "\nIf no file names are given, or if a file is '-', lziprecover decompresses\n"
"from standard input to standard output.\n"
@@ -166,10 +173,10 @@ void show_help()
"Ki = KiB = 2^10 = 1024, M = 10^6, Mi = 2^20, G = 10^9, Gi = 2^30, etc...\n"
"\nTo extract all the files from archive 'foo.tar.lz', use the commands\n"
"'tar -xf foo.tar.lz' or 'lziprecover -cd foo.tar.lz | tar -xf -'.\n"
- "\nExit status: 0 for a normal exit, 1 for environmental problems (file\n"
- "not found, invalid flags, I/O errors, etc), 2 to indicate a corrupt or\n"
- "invalid input file, 3 for an internal consistency error (e.g., bug) which\n"
- "caused lziprecover to panic.\n"
+ "\nExit status: 0 for a normal exit, 1 for environmental problems\n"
+ "(file not found, invalid command line options, I/O errors, etc), 2 to\n"
+ "indicate a corrupt or invalid input file, 3 for an internal consistency\n"
+ "error (e.g., bug) which caused lziprecover to panic.\n"
"\nReport bugs to lzip-bug@nongnu.org\n"
"Lziprecover home page: http://www.nongnu.org/lzip/lziprecover.html\n" );
}
@@ -202,14 +209,13 @@ const char * format_ds( const unsigned dictionary_size )
{
enum { bufsize = 16, factor = 1024 };
static char buf[bufsize];
- const char * const prefix[8] =
- { "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", "Zi", "Yi" };
+ const char * const prefix[3] = { "Ki", "Mi", "Gi" };
const char * p = "";
const char * np = " ";
unsigned num = dictionary_size;
bool exact = ( num % factor == 0 );
- for( int i = 0; i < 8 && ( num > 9999 || ( exact && num >= factor ) ); ++i )
+ for( int i = 0; i < 3 && ( num > 9999 || ( exact && num >= factor ) ); ++i )
{ num /= factor; if( num % factor != 0 ) exact = false;
p = prefix[i]; np = ""; }
snprintf( buf, bufsize, "%s%4u %sB", np, num, p );
@@ -226,42 +232,47 @@ void show_header( const unsigned dictionary_size )
#include "main_common.cc"
-// Colon-separated list of "damaged", "tdata", [r][^]<list> (1 1,3-5,8)
-void Member_list::parse_ml( const char * arg, const char * const option_name )
+// Colon-separated list of "damaged", "empty", "tdata", [r][^]<list> (1 1,3-5)
+void Member_list::parse_ml( const char * const arg,
+ const char * const option_name,
+ Cl_options & cl_opts )
{
+ const char * p = arg; // points to current char
while( true )
{
- const char * tp = arg; // points to terminator (':' or '\0')
+ const char * tp = p; // points to terminator (':' or '\0')
while( *tp && *tp != ':' ) ++tp;
- const unsigned len = tp - arg;
- if( std::islower( *(const unsigned char *)arg ) )
+ const unsigned len = tp - p;
+ if( std::islower( *(const unsigned char *)p ) )
{
- if( len <= 7 && std::strncmp( "damaged", arg, len ) == 0 )
- { damaged = true; goto next; }
- if( len <= 5 && std::strncmp( "tdata", arg, len ) == 0 )
- { tdata = true; goto next; }
+ if( len <= 7 && std::strncmp( "damaged", p, len ) == 0 )
+ { damaged = true; cl_opts.ignore_errors = true; goto next; }
+ if( len <= 5 && std::strncmp( "empty", p, len ) == 0 )
+ { empty = true; cl_opts.ignore_empty = true; goto next; }
+ if( len <= 5 && std::strncmp( "tdata", p, len ) == 0 )
+ { tdata = true; cl_opts.ignore_trailing = true; goto next; }
}
{
- const bool reverse = ( *arg == 'r' );
- if( reverse ) ++arg;
- if( *arg == '^' ) { ++arg; if( reverse ) rin = false; else in = false; }
+ const bool reverse = ( *p == 'r' );
+ if( reverse ) ++p;
+ if( *p == '^' ) { ++p; if( reverse ) rin = false; else in = false; }
std::vector< Block > * rvp = reverse ? &rrange_vector : &range_vector;
- while( std::isdigit( *(const unsigned char *)arg ) )
+ while( std::isdigit( *(const unsigned char *)p ) )
{
const char * tail;
- const int pos = getnum( arg, option_name, 0, 1, INT_MAX, &tail ) - 1;
+ const long pos = getnum( p, option_name, 0, 1, LONG_MAX, &tail ) - 1;
if( rvp->size() && pos < rvp->back().end() ) break;
- const int size = (*tail == '-') ?
- getnum( tail + 1, option_name, 0, pos + 1, INT_MAX, &tail ) - pos : 1;
+ const long size = (*tail == '-') ?
+ getnum( tail + 1, option_name, 0, pos + 1, LONG_MAX, &tail ) - pos : 1;
rvp->push_back( Block( pos, size ) );
if( tail == tp ) goto next;
- if( *tail == ',' ) arg = tail + 1; else break;
+ if( *tail == ',' ) p = tail + 1; else break;
}
}
- show_error( "Invalid list of members." );
+ show_option_error( arg, "Invalid list of members in", option_name );
std::exit( 1 );
next:
- if( *(arg = tp) != 0 ) ++arg; else return;
+ if( *(p = tp) != 0 ) ++p; else return;
}
}
@@ -274,12 +285,8 @@ int parse_lzip_level( const char * const arg, const char * const option_name )
{
if( *arg == 'a' || std::isdigit( *(const unsigned char *)arg ) ) return *arg;
if( *arg != 'm' )
- {
- if( verbosity >= 0 )
- std::fprintf( stderr, "%s: Bad argument in option '%s'.\n",
- program_name, option_name );
- std::exit( 1 );
- }
+ { show_option_error( arg, "Invalid argument in", option_name );
+ std::exit( 1 ); }
if( arg[1] == 0 ) return -1;
return -getnum( arg + 1, option_name, 0, min_match_len_limit, max_match_len );
}
@@ -287,39 +294,34 @@ int parse_lzip_level( const char * const arg, const char * const option_name )
/* Recognized format: <range>[,<sector_size>]
range formats: <begin> <begin>-<end> <begin>,<size> ,<size>
+ Return a pointer to the byte following the bytes parsed.
*/
-void parse_range( const char * const arg, const char * const pn,
- Block & range, int * const sector_sizep = 0 )
+const char * parse_range( const char * const arg, const char * const pn,
+ Block & range, int * const sector_sizep = 0 )
{
const char * tail = arg;
long long value =
( arg[0] == ',' ) ? 0 : getnum( arg, pn, 0, 0, INT64_MAX - 1, &tail );
- if( tail[0] == 0 || tail[0] == ',' || tail[0] == '-' )
+ if( tail[0] == 0 || tail[0] == ',' || tail[0] == '-' || tail[0] == ':' )
{
range.pos( value );
- if( tail[0] == 0 ) { range.size( INT64_MAX - value ); return; }
+ if( tail[0] == 0 || tail[0] == ':' )
+ { range.size( INT64_MAX - value ); return tail; }
const bool is_size = ( tail[0] == ',' );
if( sector_sizep && tail[1] == ',' ) { value = INT64_MAX - value; ++tail; }
else value = getnum( tail + 1, pn, 0, 1, INT64_MAX, &tail ); // size
if( !is_size && value <= range.pos() )
- {
- if( verbosity >= 0 )
- std::fprintf( stderr, "%s: Begin must be < end in range argument "
- "of option '%s'.\n", program_name, pn );
- std::exit( 1 );
- }
- if( !is_size ) value -= range.pos();
+ { show_option_error( arg, "Begin must be < end in", pn ); std::exit( 1 ); }
+ if( !is_size ) value -= range.pos(); // size = end - pos
if( INT64_MAX - value >= range.pos() )
{
range.size( value );
if( sector_sizep && tail[0] == ',' )
- *sector_sizep = getnum( tail + 1, pn, 0, 8, INT_MAX );
- return;
+ *sector_sizep = getnum( tail + 1, pn, 0, 8, INT_MAX, &tail );
+ return tail;
}
}
- if( verbosity >= 0 )
- std::fprintf( stderr, "%s: Bad decompression range in option '%s'.\n",
- program_name, pn );
+ show_option_error( arg, "Invalid decompression range in", pn );
std::exit( 1 );
}
@@ -333,6 +335,15 @@ void one_file( const int files )
}
}
+void at_least_one_file( const int files )
+ {
+ if( files < 1 )
+ {
+ show_error( "You must specify at least 1 file.", 0, true );
+ std::exit( 1 );
+ }
+ }
+
void set_mode( Mode & program_mode, const Mode new_mode )
{
@@ -353,12 +364,8 @@ void parse_u( const char * const arg, const char * const option_name,
{ set_mode( program_mode, m_unzcrash_block );
sector_size = getnum( arg + 1, option_name, 0, 1, INT_MAX ); }
else
- {
- if( verbosity >= 0 )
- std::fprintf( stderr, "%s: Bad argument for option '%s'.\n",
- program_name, option_name );
- std::exit( 1 );
- }
+ { show_option_error( arg, "Invalid argument in", option_name );
+ std::exit( 1 ); }
}
@@ -476,15 +483,15 @@ bool open_outstream( const bool force, const bool protect,
}
-bool file_exists( const std::string & filename )
+bool output_file_exists()
{
struct stat st;
- if( stat( filename.c_str(), &st ) == 0 )
+ if( stat( output_filename.c_str(), &st ) == 0 )
{
if( verbosity >= 0 )
std::fprintf( stderr, "%s: Output file '%s' already exists."
" Use '--force' to overwrite it.\n",
- program_name, filename.c_str() );
+ program_name, output_filename.c_str() );
return true;
}
return false;
@@ -585,7 +592,7 @@ void close_and_set_permissions( const struct stat * const in_statsp )
}
-unsigned char xdigit( const unsigned value )
+unsigned char xdigit( const unsigned value ) // hex digit for 'value'
{
if( value <= 9 ) return '0' + value;
if( value <= 15 ) return 'A' + value - 10;
@@ -620,8 +627,7 @@ bool show_trailing_data( const uint8_t * const data, const int size,
int decompress( const unsigned long long cfile_size, const int infd,
- const Pretty_print & pp, const bool ignore_errors,
- const bool ignore_trailing, const bool loose_trailing,
+ const Cl_options & cl_opts, const Pretty_print & pp,
const bool testing )
{
unsigned long long partial_file_pos = 0;
@@ -632,50 +638,49 @@ int decompress( const unsigned long long cfile_size, const int infd,
{
Lzip_header header;
rdec.reset_member_position();
- const int size = rdec.read_header_carefully( header, ignore_errors );
+ const int size = rdec.read_header_carefully( header, cl_opts.ignore_errors );
if( rdec.finished() || // End Of File
( size < Lzip_header::size && !rdec.find_header( header ) ) )
{
if( first_member )
{ show_file_error( pp.name(), "File ends unexpectedly at member header." );
retval = 2; }
- else if( header.verify_prefix( size ) )
+ else if( header.check_prefix( size ) )
{ pp( "Truncated header in multimember file." );
- show_trailing_data( header.data, size, pp, true, -1 );
- retval = 2; }
- else if( size > 0 && !show_trailing_data( header.data, size, pp,
- true, ignore_trailing ) )
- retval = 2;
+ show_trailing_data( header.data, size, pp, true, -1 ); retval = 2; }
+ else if( size > 0 && !show_trailing_data( header.data, size, pp, true,
+ cl_opts.ignore_trailing ) ) retval = 2;
break;
}
- if( !header.verify_magic() )
+ if( !header.check_magic() )
{
if( first_member )
{ show_file_error( pp.name(), bad_magic_msg ); retval = 2; }
- else if( !loose_trailing && header.verify_corrupt() )
+ else if( !cl_opts.loose_trailing && header.check_corrupt() )
{ pp( corrupt_mm_msg );
- show_trailing_data( header.data, size, pp, false, -1 );
- retval = 2; }
- else if( !show_trailing_data( header.data, size, pp, false, ignore_trailing ) )
- retval = 2;
- if( ignore_errors ) { pp.reset(); continue; } else break;
+ show_trailing_data( header.data, size, pp, false, -1 ); retval = 2; }
+ else if( !show_trailing_data( header.data, size, pp, false,
+ cl_opts.ignore_trailing ) ) retval = 2;
+ if( cl_opts.ignore_errors ) { pp.reset(); continue; } else break;
}
- if( !header.verify_version() )
+ if( !header.check_version() )
{ pp( bad_version( header.version() ) ); retval = 2;
- if( ignore_errors ) { pp.reset(); continue; } else break; }
+ if( cl_opts.ignore_errors ) { pp.reset(); continue; } else break; }
const unsigned dictionary_size = header.dictionary_size();
if( !isvalid_ds( dictionary_size ) )
{ pp( bad_dict_msg ); retval = 2;
- if( ignore_errors ) { pp.reset(); continue; } else break; }
+ if( cl_opts.ignore_errors ) { pp.reset(); continue; } else break; }
if( verbosity >= 2 || ( verbosity == 1 && first_member ) ) pp();
LZ_decoder decoder( rdec, dictionary_size, outfd );
show_dprogress( cfile_size, partial_file_pos, &rdec, &pp ); // init
- const int result = decoder.decode_member( pp );
+ const int result =
+ decoder.decode_member( pp, cl_opts.ignore_empty, cl_opts.ignore_marking );
partial_file_pos += rdec.member_position();
if( result != 0 )
{
+ retval = 2;
if( verbosity >= 0 && result <= 2 )
{
pp();
@@ -683,14 +688,16 @@ int decompress( const unsigned long long cfile_size, const int infd,
"File ends unexpectedly" : "Decoder error",
partial_file_pos );
}
- retval = 2; if( ignore_errors ) { pp.reset(); continue; } else break;
+ else if( result == 5 ) { pp( empty_msg ); break; }
+ else if( result == 6 ) { pp( marking_msg ); break; }
+ if( cl_opts.ignore_errors ) { pp.reset(); continue; } else break;
}
if( verbosity >= 2 )
{ std::fputs( testing ? "ok\n" : "done\n", stderr ); pp.reset(); }
}
if( verbosity == 1 && retval == 0 )
std::fputs( testing ? "ok\n" : "done\n", stderr );
- if( retval == 2 && ignore_errors ) retval = 0;
+ if( retval == 2 && cl_opts.ignore_errors ) retval = 0;
return retval;
}
@@ -776,15 +783,14 @@ int main( const int argc, const char * const argv[] )
// '0'..'9' = level, 'a' = all levels
// -5..-273 = match length, -1 = all lengths
int repeated_byte = -1; // 0 to 255, or -1 for all values
+ Cl_options cl_opts; // command line options
bool force = false;
- bool ignore_errors = false;
- bool ignore_trailing = true;
bool keep_input_files = false;
- bool loose_trailing = false;
bool to_stdout = false;
if( argc > 0 ) invocation_name = argv[0];
- enum { opt_du = 256, opt_lt, opt_lzl, opt_lzn, opt_ref, opt_re, opt_st };
+ enum { opt_cm = 256, opt_du, opt_eer, opt_lt, opt_lzl, opt_lzn, opt_mer,
+ opt_ref, opt_rem, opt_st };
const Arg_parser::Option options[] =
{
{ 'a', "trailing-error", Arg_parser::no },
@@ -804,6 +810,7 @@ int main( const int argc, const char * const argv[] )
{ 'n', "threads", Arg_parser::yes },
{ 'o', "output", Arg_parser::yes },
{ 'q', "quiet", Arg_parser::no },
+ { 'R', "byte-repair", Arg_parser::no },
{ 'R', "repair", Arg_parser::no },
{ 's', "split", Arg_parser::no },
{ 'S', "nrep-stats", Arg_parser::maybe },
@@ -814,13 +821,16 @@ int main( const int argc, const char * const argv[] )
{ 'W', "debug-decompress", Arg_parser::yes },
{ 'X', "show-packets", Arg_parser::maybe },
{ 'Y', "debug-delay", Arg_parser::yes },
- { 'Z', "debug-repair", Arg_parser::yes },
+ { 'Z', "debug-byte-repair", Arg_parser::yes },
+ { opt_cm, "clear-marking", Arg_parser::no },
{ opt_du, "dump", Arg_parser::yes },
+ { opt_eer, "empty-error", Arg_parser::no },
+ { opt_mer, "marking-error", Arg_parser::no },
{ opt_lt, "loose-trailing", Arg_parser::no },
{ opt_lzl, "lzip-level", Arg_parser::yes },
{ opt_lzn, "lzip-name", Arg_parser::yes },
{ opt_ref, "reference-file", Arg_parser::yes },
- { opt_re, "remove", Arg_parser::yes },
+ { opt_rem, "remove", Arg_parser::yes },
{ opt_st, "strip", Arg_parser::yes },
{ 0 , 0, Arg_parser::no } };
@@ -838,7 +848,7 @@ int main( const int argc, const char * const argv[] )
const char * const arg = sarg.c_str();
switch( code )
{
- case 'a': ignore_trailing = false; break;
+ case 'a': cl_opts.ignore_trailing = false; break;
case 'A': set_mode( program_mode, m_alone_to_lz ); break;
case 'c': to_stdout = true; break;
case 'd': set_mode( program_mode, m_decompress ); break;
@@ -849,7 +859,7 @@ int main( const int argc, const char * const argv[] )
parse_range( arg, pn, range, &sector_size ); break;
case 'f': force = true; break;
case 'h': show_help(); return 0;
- case 'i': ignore_errors = true; break;
+ case 'i': cl_opts.ignore_errors = true; break;
case 'k': keep_input_files = true; break;
case 'l': set_mode( program_mode, m_list ); break;
case 'm': set_mode( program_mode, m_merge ); break;
@@ -858,7 +868,7 @@ int main( const int argc, const char * const argv[] )
case 'o': if( sarg == "-" ) to_stdout = true;
else { default_output_filename = sarg; } break;
case 'q': verbosity = -1; break;
- case 'R': set_mode( program_mode, m_repair ); break;
+ case 'R': set_mode( program_mode, m_byte_repair ); break;
case 's': set_mode( program_mode, m_split ); break;
case 'S': if( arg[0] ) repeated_byte = getnum( arg, pn, 0, 0, 255 );
set_mode( program_mode, m_nrep_stats ); break;
@@ -872,18 +882,22 @@ int main( const int argc, const char * const argv[] )
if( arg[0] ) { bad_byte.parse_bb( arg, pn ); } break;
case 'Y': set_mode( program_mode, m_debug_delay );
parse_range( arg, pn, range ); break;
- case 'Z': set_mode( program_mode, m_debug_repair );
+ case 'Z': set_mode( program_mode, m_debug_byte_repair );
bad_byte.parse_bb( arg, pn ); break;
+ case opt_cm: set_mode( program_mode, m_clear_marking );
+ cl_opts.ignore_marking = true; break;
case opt_du: set_mode( program_mode, m_dump );
- member_list.parse_ml( arg, pn ); break;
- case opt_lt: loose_trailing = true; break;
+ member_list.parse_ml( arg, pn, cl_opts ); break;
+ case opt_eer: cl_opts.ignore_empty = false; break;
+ case opt_lt: cl_opts.loose_trailing = true; break;
case opt_lzl: lzip_level = parse_lzip_level( arg, pn ); break;
case opt_lzn: lzip_name = arg; break;
+ case opt_mer: cl_opts.ignore_marking = false; break;
case opt_ref: reference_filename = arg; break;
- case opt_re: set_mode( program_mode, m_remove );
- member_list.parse_ml( arg, pn ); break;
+ case opt_rem: set_mode( program_mode, m_remove );
+ member_list.parse_ml( arg, pn, cl_opts ); break;
case opt_st: set_mode( program_mode, m_strip );
- member_list.parse_ml( arg, pn ); break;
+ member_list.parse_ml( arg, pn, cl_opts ); break;
default : internal_error( "uncaught option." );
}
} // end process options
@@ -913,67 +927,67 @@ int main( const int argc, const char * const argv[] )
{
case m_none: internal_error( "invalid operation." ); break;
case m_alone_to_lz: break;
+ case m_byte_repair:
+ one_file( filenames.size() );
+ return byte_repair( filenames[0], default_output_filename, cl_opts,
+ terminator, force );
+ case m_clear_marking:
+ at_least_one_file( filenames.size() );
+ return clear_marking( filenames, cl_opts );
+ case m_debug_byte_repair:
+ one_file( filenames.size() );
+ return debug_byte_repair( filenames[0].c_str(), cl_opts, bad_byte, terminator );
case m_debug_decompress:
one_file( filenames.size() );
- return debug_decompress( filenames[0], bad_byte, false );
+ return debug_decompress( filenames[0].c_str(), cl_opts, bad_byte, false );
case m_debug_delay:
one_file( filenames.size() );
- return debug_delay( filenames[0], range, terminator );
- case m_debug_repair:
- one_file( filenames.size() );
- return debug_repair( filenames[0], bad_byte, terminator );
+ return debug_delay( filenames[0].c_str(), cl_opts, range, terminator );
case m_decompress: break;
case m_dump:
case m_strip:
- if( filenames.size() < 1 )
- { show_error( "You must specify at least 1 file.", 0, true ); return 1; }
- return dump_members( filenames, default_output_filename, member_list,
- force, ignore_errors, ignore_trailing,
- loose_trailing, program_mode == m_strip, to_stdout );
+ at_least_one_file( filenames.size() );
+ return dump_members( filenames, default_output_filename, cl_opts,
+ member_list, force, program_mode == m_strip, to_stdout );
case m_list: break;
case m_md5sum: break;
case m_merge:
if( filenames.size() < 2 )
{ show_error( "You must specify at least 2 files.", 0, true ); return 1; }
- return merge_files( filenames, default_output_filename, terminator, force );
- case m_nrep_stats: return print_nrep_stats( filenames, repeated_byte,
- ignore_errors, ignore_trailing, loose_trailing );
+ return merge_files( filenames, default_output_filename, cl_opts,
+ terminator, force );
+ case m_nrep_stats:
+ return print_nrep_stats( filenames, cl_opts, repeated_byte );
case m_range_dec:
one_file( filenames.size() );
- return range_decompress( filenames[0], default_output_filename, range,
- force, ignore_errors, ignore_trailing,
- loose_trailing, to_stdout );
+ return range_decompress( filenames[0], default_output_filename,
+ cl_opts, range, force, to_stdout );
case m_remove:
- if( filenames.size() < 1 )
- { show_error( "You must specify at least 1 file.", 0, true ); return 1; }
- return remove_members( filenames, member_list, ignore_errors,
- ignore_trailing, loose_trailing );
- case m_repair:
- one_file( filenames.size() );
- return repair_file( filenames[0], default_output_filename, terminator, force );
+ at_least_one_file( filenames.size() );
+ return remove_members( filenames, cl_opts, member_list );
case m_reproduce:
one_file( filenames.size() );
if( !reference_filename || !reference_filename[0] )
{ show_error( "You must specify a reference file.", 0, true ); return 1; }
if( range.size() > 0 )
- return debug_reproduce_file( filenames[0], lzip_name,
- reference_filename, range, sector_size, lzip_level );
+ return debug_reproduce_file( filenames[0].c_str(), lzip_name,
+ reference_filename, cl_opts, range, sector_size, lzip_level );
else
- return reproduce_file( filenames[0], default_output_filename,
- lzip_name, reference_filename, lzip_level, terminator, force );
+ return reproduce_file( filenames[0], default_output_filename, lzip_name,
+ reference_filename, cl_opts, lzip_level, terminator, force );
case m_show_packets:
one_file( filenames.size() );
- return debug_decompress( filenames[0], bad_byte, true );
+ return debug_decompress( filenames[0].c_str(), cl_opts, bad_byte, true );
case m_split:
one_file( filenames.size() );
- return split_file( filenames[0], default_output_filename, force );
+ return split_file( filenames[0], default_output_filename, cl_opts, force );
case m_test: break;
case m_unzcrash_bit:
one_file( filenames.size() );
- return lunzcrash_bit( filenames[0].c_str() );
+ return lunzcrash_bit( filenames[0].c_str(), cl_opts );
case m_unzcrash_block:
one_file( filenames.size() );
- return lunzcrash_block( filenames[0].c_str(), sector_size );
+ return lunzcrash_block( filenames[0].c_str(), cl_opts, sector_size );
}
}
catch( std::bad_alloc & ) { show_error( mem_msg ); cleanup_and_fail( 1 ); }
@@ -981,10 +995,8 @@ int main( const int argc, const char * const argv[] )
if( filenames.empty() ) filenames.push_back("-");
- if( program_mode == m_list )
- return list_files( filenames, ignore_errors, ignore_trailing, loose_trailing );
- if( program_mode == m_md5sum )
- return md5sum_files( filenames );
+ if( program_mode == m_list ) return list_files( filenames, cl_opts );
+ if( program_mode == m_md5sum ) return md5sum_files( filenames );
if( program_mode != m_alone_to_lz && program_mode != m_decompress &&
program_mode != m_test )
@@ -1028,7 +1040,7 @@ int main( const int argc, const char * const argv[] )
infd = open_instream( input_filename.c_str(), &in_stats, one_to_one );
if( infd < 0 ) { set_retval( retval, 1 ); continue; }
if( !check_tty_in( pp.name(), infd, program_mode, retval ) ) continue;
- if( one_to_one ) // open outfd after verifying infd
+ if( one_to_one ) // open outfd after checking infd
{
if( program_mode == m_alone_to_lz ) set_a_outname( input_filename );
else set_d_outname( input_filename, extension_index( input_filename ) );
@@ -1040,7 +1052,7 @@ int main( const int argc, const char * const argv[] )
if( one_to_one && !check_tty_out( program_mode ) )
{ set_retval( retval, 1 ); return retval; } // don't delete a tty
- if( to_file && outfd < 0 ) // open outfd after verifying infd
+ if( to_file && outfd < 0 ) // open outfd after checking infd
{
output_filename = default_output_filename;
if( !open_outstream( force, false ) || !check_tty_out( program_mode ) )
@@ -1057,8 +1069,7 @@ int main( const int argc, const char * const argv[] )
if( program_mode == m_alone_to_lz )
tmp = alone_to_lz( infd, pp );
else
- tmp = decompress( cfile_size, infd, pp, ignore_errors, ignore_trailing,
- loose_trailing, program_mode == m_test );
+ tmp = decompress( cfile_size, infd, cl_opts, pp, program_mode == m_test );
}
catch( std::bad_alloc & ) { pp( mem_msg ); tmp = 1; }
catch( Error & e ) { pp(); show_error( e.msg, errno ); tmp = 1; }
@@ -1073,7 +1084,7 @@ int main( const int argc, const char * const argv[] )
if( delete_output_on_interrupt && one_to_one )
close_and_set_permissions( in_statsp );
if( input_filename.size() && !keep_input_files && one_to_one &&
- ( program_mode != m_decompress || !ignore_errors ) )
+ ( program_mode != m_decompress || !cl_opts.ignore_errors ) )
std::remove( input_filename.c_str() );
}
if( delete_output_on_interrupt ) close_and_set_permissions( 0 ); // -o