diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2019-01-05 20:07:41 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2019-01-05 20:07:41 +0000 |
commit | 33c0b568b11710608255b337a45aff0d147de2b4 (patch) | |
tree | 0d2d18685bef4b160194b2980174b2ebd62d97e1 /file_index.cc | |
parent | Adding upstream version 1.20. (diff) | |
download | lziprecover-upstream/1.21.tar.xz lziprecover-upstream/1.21.zip |
Adding upstream version 1.21.upstream/1.21
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'file_index.cc')
-rw-r--r-- | file_index.cc | 288 |
1 files changed, 0 insertions, 288 deletions
diff --git a/file_index.cc b/file_index.cc deleted file mode 100644 index b3d7d70..0000000 --- a/file_index.cc +++ /dev/null @@ -1,288 +0,0 @@ -/* Lziprecover - Data recovery tool for the lzip format - Copyright (C) 2009-2018 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 - the Free Software Foundation, either version 2 of the License, or - (at your option) any later version. - - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU General Public License for more details. - - You should have received a copy of the GNU General Public License - along with this program. If not, see <http://www.gnu.org/licenses/>. -*/ - -#define _FILE_OFFSET_BITS 64 - -#include <algorithm> -#include <cerrno> -#include <cstdio> -#include <cstring> -#include <string> -#include <vector> -#include <stdint.h> -#include <unistd.h> - -#include "lzip.h" -#include "block.h" -#include "file_index.h" - - -int seek_read( const int fd, uint8_t * const buf, const int size, - const long long pos ) - { - if( lseek( fd, pos, SEEK_SET ) == pos ) - return readblock( fd, buf, size ); - return 0; - } - - -void File_index::set_errno_error( const char * const msg ) - { - error_ = msg; error_ += std::strerror( errno ); - retval_ = 1; - } - -void File_index::set_num_error( const char * const msg, unsigned long long num ) - { - char buf[80]; - snprintf( buf, sizeof buf, "%s%llu", msg, num ); - error_ = buf; - retval_ = 2; - } - - -// If successful, push last member and set pos to member header. -bool File_index::skip_trailing_data( const int fd, long long & pos, - const bool ignore_bad_ds, - const bool ignore_trailing, const bool loose_trailing ) - { - enum { block_size = 16384, - buffer_size = block_size + File_trailer::size - 1 + File_header::size }; - uint8_t buffer[buffer_size]; - if( pos < min_member_size ) return false; - int bsize = pos % block_size; // total bytes in buffer - if( bsize <= buffer_size - block_size ) bsize += block_size; - int search_size = bsize; // bytes to search for trailer - int rd_size = bsize; // bytes to read from file - unsigned long long ipos = pos - rd_size; // aligned to block_size - - while( true ) - { - if( seek_read( fd, buffer, rd_size, ipos ) != rd_size ) - { set_errno_error( "Error seeking member trailer: " ); return false; } - const uint8_t max_msb = ( ipos + search_size ) >> 56; - for( int i = search_size; i >= File_trailer::size; --i ) - if( buffer[i-1] <= max_msb ) // most significant byte of member_size - { - File_trailer & trailer = - *(File_trailer *)( buffer + i - File_trailer::size ); - const unsigned long long member_size = trailer.member_size(); - if( member_size == 0 ) - { while( i > File_trailer::size && buffer[i-9] == 0 ) --i; continue; } - if( member_size < min_member_size || member_size > ipos + i ) - continue; - File_header header; - if( seek_read( fd, header.data, File_header::size, - ipos + i - member_size ) != File_header::size ) - { set_errno_error( "Error reading member header: " ); return false; } - const unsigned dictionary_size = header.dictionary_size(); - if( !header.verify_magic() || !header.verify_version() || - ( !ignore_bad_ds && !isvalid_ds( dictionary_size ) ) ) continue; - if( (*(File_header *)( buffer + i )).verify_prefix( bsize - i ) ) - { error_ = "Last member in input file is truncated or corrupt."; - retval_ = 2; return false; } - if( !loose_trailing && bsize - i >= File_header::size && - (*(File_header *)( buffer + i )).verify_corrupt() ) - { error_ = corrupt_mm_msg; retval_ = 2; return false; } - if( !ignore_trailing ) - { error_ = trailing_msg; retval_ = 2; return false; } - pos = ipos + i - member_size; - member_vector.push_back( Member( 0, trailer.data_size(), pos, - member_size, dictionary_size ) ); - return true; - } - if( ipos <= 0 ) - { set_num_error( "Member size in trailer is corrupt at pos ", pos - 8 ); - return false; } - bsize = buffer_size; - search_size = bsize - File_header::size; - rd_size = block_size; - ipos -= rd_size; - std::memcpy( buffer + rd_size, buffer, buffer_size - rd_size ); - } - } - - -File_index::File_index( const int infd, const bool ignore_bad_ds, - const bool ignore_trailing, const bool loose_trailing ) - : isize( lseek( infd, 0, SEEK_END ) ), retval_( 0 ) - { - if( isize < 0 ) - { set_errno_error( "Input file is not seekable: " ); return; } - if( isize < min_member_size ) - { error_ = "Input file is too short."; retval_ = 2; return; } - if( isize > INT64_MAX ) - { error_ = "Input file is too long (2^63 bytes or more)."; - retval_ = 2; return; } - - File_header header; - if( seek_read( infd, header.data, File_header::size, 0 ) != File_header::size ) - { set_errno_error( "Error reading member header: " ); return; } - if( !header.verify_magic() ) - { error_ = bad_magic_msg; retval_ = 2; return; } - if( !header.verify_version() ) - { error_ = bad_version( header.version() ); retval_ = 2; return; } - if( !ignore_bad_ds && !isvalid_ds( header.dictionary_size() ) ) - { error_ = bad_dict_msg; retval_ = 2; return; } - - long long pos = isize; // always points to a header or to EOF - while( pos >= min_member_size ) - { - File_trailer trailer; - if( seek_read( infd, trailer.data, File_trailer::size, - pos - File_trailer::size ) != File_trailer::size ) - { set_errno_error( "Error reading member trailer: " ); break; } - const unsigned long long member_size = trailer.member_size(); - if( member_size < min_member_size || member_size > (unsigned long long)pos ) - { - if( member_vector.empty() ) - { if( skip_trailing_data( infd, pos, ignore_bad_ds, ignore_trailing, - loose_trailing ) ) continue; else return; } - set_num_error( "Member size in trailer is corrupt at pos ", pos - 8 ); - break; - } - if( seek_read( infd, header.data, File_header::size, - pos - member_size ) != File_header::size ) - { set_errno_error( "Error reading member header: " ); break; } - const unsigned dictionary_size = header.dictionary_size(); - if( !header.verify_magic() || !header.verify_version() || - ( !ignore_bad_ds && !isvalid_ds( dictionary_size ) ) ) - { - if( member_vector.empty() ) - { if( skip_trailing_data( infd, pos, ignore_bad_ds, ignore_trailing, - loose_trailing ) ) continue; else return; } - set_num_error( "Bad header at pos ", pos - member_size ); - break; - } - pos -= member_size; - member_vector.push_back( Member( 0, trailer.data_size(), pos, - member_size, dictionary_size ) ); - } - if( pos != 0 || member_vector.empty() ) - { - member_vector.clear(); - if( retval_ == 0 ) { error_ = "Can't create file index."; retval_ = 2; } - return; - } - std::reverse( member_vector.begin(), member_vector.end() ); - for( unsigned long i = 0; i < member_vector.size() - 1; ++i ) - { - const long long end = member_vector[i].dblock.end(); - if( end < 0 || end > INT64_MAX ) - { - member_vector.clear(); - error_ = "Data in input file is too long (2^63 bytes or more)."; - retval_ = 2; return; - } - member_vector[i+1].dblock.pos( end ); - } - } - - -// All files in 'infd_vector' must be at least 'fsize' bytes long. -File_index::File_index( const std::vector< int > & infd_vector, - const long long fsize ) - : isize( fsize ), retval_( 0 ) - { - if( isize < 0 ) - { set_errno_error( "Input file is not seekable: " ); return; } - if( isize < min_member_size ) - { error_ = "Input file is too short."; retval_ = 2; return; } - if( isize > INT64_MAX ) - { error_ = "Input file is too long (2^63 bytes or more)."; - retval_ = 2; return; } - - const int files = infd_vector.size(); - File_header header; - bool done = false; - for( int i = 0; i < files && !done; ++i ) - { - const int infd = infd_vector[i]; - if( seek_read( infd, header.data, File_header::size, 0 ) != File_header::size ) - { set_errno_error( "Error reading member header: " ); return; } - if( header.verify_magic() && header.verify_version() ) done = true; - } - if( !done ) - { error_ = bad_magic_msg; retval_ = 2; return; } - - long long pos = isize; // always points to a header or to EOF - while( pos >= min_member_size ) - { - unsigned long long member_size; - File_trailer trailer; - done = false; - for( int it = 0; it < files && !done; ++it ) - { - const int tfd = infd_vector[it]; - if( seek_read( tfd, trailer.data, File_trailer::size, - pos - File_trailer::size ) != File_trailer::size ) - { set_errno_error( "Error reading member trailer: " ); goto error; } - member_size = trailer.member_size(); - if( member_size >= min_member_size && member_size <= (unsigned long long)pos ) - for( int ih = 0; ih < files && !done; ++ih ) - { - const int hfd = infd_vector[ih]; - if( seek_read( hfd, header.data, File_header::size, - pos - member_size ) != File_header::size ) - { set_errno_error( "Error reading member header: " ); goto error; } - if( header.verify_magic() && header.verify_version() ) done = true; - } - } - if( !done ) - { - if( member_vector.empty() ) { --pos; continue; } // maybe trailing data - set_num_error( "Member size in trailer may be corrupt at pos ", pos - 8 ); - break; - } - if( member_vector.empty() && isize > pos ) - { - const int size = std::min( (long long)File_header::size, isize - pos ); - for( int i = 0; i < files; ++i ) - { - const int infd = infd_vector[i]; - if( seek_read( infd, header.data, size, pos ) == size && - header.verify_prefix( size ) ) - { - error_ = "Last member in input file is truncated or corrupt."; - retval_ = 2; goto error; - } - } - } - pos -= member_size; - member_vector.push_back( Member( 0, trailer.data_size(), pos, - member_size, 0 ) ); - } -error: - if( pos != 0 || member_vector.empty() ) - { - member_vector.clear(); - if( retval_ == 0 ) { error_ = "Can't create file index."; retval_ = 2; } - return; - } - std::reverse( member_vector.begin(), member_vector.end() ); - for( unsigned long i = 0; i < member_vector.size() - 1; ++i ) - { - const long long end = member_vector[i].dblock.end(); - if( end < 0 || end > INT64_MAX ) - { - member_vector.clear(); - error_ = "Data in input file is too long (2^63 bytes or more)."; - retval_ = 2; return; - } - member_vector[i+1].dblock.pos( end ); - } - } |