diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-06-16 11:13:24 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2023-06-16 11:13:24 +0000 |
commit | d7c278fca708bc2c4badaeac041ecb136a4ef955 (patch) | |
tree | 5b163b83e6e3171a311f67187aa75c0f96f98f13 /lunzcrash.cc | |
parent | Adding upstream version 1.23. (diff) | |
download | lziprecover-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-- | lunzcrash.cc | 66 |
1 files changed, 35 insertions, 31 deletions
diff --git a/lunzcrash.cc b/lunzcrash.cc index 577d355..6267ce8 100644 --- a/lunzcrash.cc +++ b/lunzcrash.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 @@ -37,31 +37,31 @@ namespace { -bool verify_member( const uint8_t * const mbuffer, const long long msize, - const unsigned dictionary_size, const char * const name, - uint8_t digest[16] ) +bool check_member( const uint8_t * const mbuffer, const long msize, + const unsigned dictionary_size, const char * const name, + md5_type & digest ) { MD5SUM md5sum; LZ_mtester mtester( mbuffer, msize, dictionary_size, -1, &md5sum ); if( mtester.test_member() != 0 || !mtester.finished() ) - { show_file_error( name, "Error verifying input file." ); return false; } + { show_file_error( name, "Error checking input file." ); return false; } md5sum.md5_finish( digest ); return true; } -bool compare_member( const uint8_t * const mbuffer, const long long msize, +bool compare_member( const uint8_t * const mbuffer, const long msize, const unsigned dictionary_size, - const long long byte_pos, const uint8_t digest[16] ) + const long long byte_pos, const md5_type & digest ) { MD5SUM md5sum; LZ_mtester mtester( mbuffer, msize, dictionary_size, -1, &md5sum ); bool error = ( mtester.test_member() != 0 || !mtester.finished() ); if( !error ) { - uint8_t new_digest[16]; + md5_type new_digest; md5sum.md5_finish( new_digest ); - if( std::memcmp( digest, new_digest, 16 ) != 0 ) error = true; + if( digest != new_digest ) error = true; } if( error && verbosity >= 0 ) std::printf( "byte %llu comparison failed\n", byte_pos ); @@ -75,14 +75,14 @@ int test_member_rest( const LZ_mtester & master, uint8_t * const buffer2, { LZ_mtester mtester( master ); // tester with external buffer mtester.duplicate_buffer( buffer2 ); - int result = mtester.test_member( LLONG_MAX, LLONG_MAX, stdout, byte_pos ); + int result = mtester.test_member( LONG_MAX, LLONG_MAX, stdout, byte_pos ); if( result == 0 && !mtester.finished() ) result = -1; // false negative if( result != 0 ) *failure_posp = mtester.member_position(); return result; } -long next_pct_pos( const Lzip_index & lzip_index, const int i, const int pct, +long next_pct_pos( const Lzip_index & lzip_index, const long i, const int pct, const int sector_size = 0 ) { if( pct <= 0 ) return 0; @@ -103,13 +103,14 @@ long next_pct_pos( const Lzip_index & lzip_index, const int i, const int pct, /* Test 1-bit errors in LZMA streams in file. Unless verbosity >= 1, print only the bytes with interesting results. */ -int lunzcrash_bit( const char * const input_filename ) +int lunzcrash_bit( const char * const input_filename, + const Cl_options & cl_opts ) { struct stat in_stats; // not used const int infd = open_instream( input_filename, &in_stats, false, true ); if( infd < 0 ) return 1; - const Lzip_index lzip_index( infd, true, true ); + const Lzip_index lzip_index( infd, cl_opts ); if( lzip_index.retval() != 0 ) { show_file_error( input_filename, lzip_index.error().c_str() ); return lzip_index.retval(); } @@ -122,12 +123,12 @@ int lunzcrash_bit( const char * const input_filename ) { const long long mpos = lzip_index.mblock( i ).pos(); const long long msize = lzip_index.mblock( i ).size(); - const unsigned dictionary_size = lzip_index.dictionary_size( i ); - uint8_t * const mbuffer = read_member( infd, mpos, msize ); + uint8_t * const mbuffer = read_member( infd, mpos, msize, input_filename ); if( !mbuffer ) return 1; - uint8_t md5_orig[16]; - if( !verify_member( mbuffer, msize, dictionary_size, input_filename, - md5_orig ) ) return 2; + const unsigned dictionary_size = lzip_index.dictionary_size( i ); + md5_type md5_orig; + if( !check_member( mbuffer, msize, dictionary_size, input_filename, + md5_orig ) ) return 2; long pct_pos = next_pct_pos( lzip_index, i, pct ); long pos = Lzip_header::size + 1, printed = 0; // last pos printed const long end = msize - 20; @@ -212,7 +213,7 @@ int lunzcrash_bit( const char * const input_filename ) if( failed_comparisons > 0 ) std::printf( ", of which\n%9ld comparisons failed\n", failed_comparisons ); - else std::fputs( "\n all comparisons passed\n", stdout ); + else std::fputs( "\n all comparisons passed\n", stdout ); } else std::fputc( '\n', stdout ); } @@ -222,13 +223,14 @@ int lunzcrash_bit( const char * const input_filename ) /* Test zeroed blocks of given size in LZMA streams in file. Unless verbosity >= 1, print only the bytes with interesting results. */ -int lunzcrash_block( const char * const input_filename, const int sector_size ) +int lunzcrash_block( const char * const input_filename, + const Cl_options & cl_opts, const int sector_size ) { struct stat in_stats; // not used const int infd = open_instream( input_filename, &in_stats, false, true ); if( infd < 0 ) return 1; - const Lzip_index lzip_index( infd, true, true ); + const Lzip_index lzip_index( infd, cl_opts ); if( lzip_index.retval() != 0 ) { show_file_error( input_filename, lzip_index.error().c_str() ); return lzip_index.retval(); } @@ -242,16 +244,17 @@ int lunzcrash_block( const char * const input_filename, const int sector_size ) { const long long mpos = lzip_index.mblock( i ).pos(); const long long msize = lzip_index.mblock( i ).size(); - long pos = Lzip_header::size + 1; - const long end = msize - sector_size - 20; - if( end <= pos ) continue; // sector_size larger than LZMA stream - const unsigned dictionary_size = lzip_index.dictionary_size( i ); - uint8_t * const mbuffer = read_member( infd, mpos, msize ); + // skip members with LZMA stream smaller than sector_size + if( msize - Lzip_header::size - 1 - 20 <= sector_size ) continue; + uint8_t * const mbuffer = read_member( infd, mpos, msize, input_filename ); if( !mbuffer ) return 1; - uint8_t md5_orig[16]; - if( !verify_member( mbuffer, msize, dictionary_size, input_filename, - md5_orig ) ) return 2; + const unsigned dictionary_size = lzip_index.dictionary_size( i ); + md5_type md5_orig; + if( !check_member( mbuffer, msize, dictionary_size, input_filename, + md5_orig ) ) return 2; long pct_pos = next_pct_pos( lzip_index, i, pct, sector_size ); + long pos = Lzip_header::size + 1; + const long end = msize - sector_size - 20; if( verbosity >= 0 ) // give a clue of the range being tested std::printf( "Testing blocks of size %u from pos %llu to %llu\n", sector_size, mpos + pos, mpos + end - 1 ); @@ -324,7 +327,7 @@ int lunzcrash_block( const char * const input_filename, const int sector_size ) if( failed_comparisons > 0 ) std::printf( ", of which\n%9ld comparisons failed\n", failed_comparisons ); - else std::fputs( "\n all comparisons passed\n", stdout ); + else std::fputs( "\n all comparisons passed\n", stdout ); } else std::fputc( '\n', stdout ); } @@ -348,7 +351,8 @@ int md5sum_files( const std::vector< std::string > & filenames ) if( infd < 0 ) { set_retval( retval, 1 ); continue; } enum { buffer_size = 16384 }; - uint8_t buffer[buffer_size], md5_digest[16]; + uint8_t buffer[buffer_size]; + md5_type md5_digest; MD5SUM md5sum; while( true ) { |