/* Tests single open and close of a file
* Returns 1 if successful, 0 if not or -1 on error
*/
int pff_test_single_open_close_file(
libcstring_system_character_t *filename,
int access_flags,
int expected_result )
{
libcerror_error_t *error = NULL;
libpff_file_t *file = NULL;
static char *function = "pff_test_single_open_close_file";
int result = 0;
if( libpff_file_initialize(
&file,
&error ) != 1 )
{
libcerror_error_set(
&error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,
"%s: unable to create file.",
function );
return( -1 );
}
#if defined( LIBCSTRING_HAVE_WIDE_SYSTEM_CHARACTER )
result = libpff_file_open_wide(
file,
filename,
access_flags,
&error );
#else
result = libpff_file_open(
file,
filename,
access_flags,
&error );
#endif
if( result == 1 )
{
if( libpff_file_close(
file,
&error ) != 0 )
{
libcerror_error_set(
&error,
LIBCERROR_ERROR_DOMAIN_IO,
LIBCERROR_IO_ERROR_CLOSE_FAILED,
"%s: unable to close file.",
function );
result = -1;
}
}
if( libpff_file_free(
&file,
&error ) != 1 )
{
libcerror_error_set(
&error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_FINALIZE_FAILED,
"%s: unable to free file.",
function );
result = -1;
}
result = ( expected_result == result );
if( result == 1 )
{
fprintf(
stdout,
"(PASS)" );
}
else
{
fprintf(
stdout,
"(FAIL)" );
}
fprintf(
stdout,
"\n" );
if( error != NULL )
{
if( result != 1 )
{
libcerror_error_backtrace_fprint(
error,
stderr );
}
libcerror_error_free(
&error );
}
return( result );
}
/* Creates an info handle
* Make sure the value info_handle is referencing, is set to NULL
* Returns 1 if successful or -1 on error
*/
int info_handle_initialize(
info_handle_t **info_handle,
libcerror_error_t **error )
{
static char *function = "info_handle_initialize";
if( info_handle == NULL )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
"%s: invalid info handle.",
function );
return( -1 );
}
if( *info_handle != NULL )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_VALUE_ALREADY_SET,
"%s: invalid info handle value already set.",
function );
return( -1 );
}
*info_handle = memory_allocate_structure(
info_handle_t );
if( *info_handle == NULL )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_MEMORY,
LIBCERROR_MEMORY_ERROR_INSUFFICIENT,
"%s: unable to create info handle.",
function );
goto on_error;
}
if( memory_set(
*info_handle,
0,
sizeof( info_handle_t ) ) == NULL )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_MEMORY,
LIBCERROR_MEMORY_ERROR_SET_FAILED,
"%s: unable to clear info handle.",
function );
goto on_error;
}
if( libpff_file_initialize(
&( ( *info_handle )->input_file ),
error ) != 1 )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,
"%s: unable to initialize input file.",
function );
goto on_error;
}
( *info_handle )->notify_stream = INFO_HANDLE_NOTIFY_STREAM;
return( 1 );
on_error:
if( *info_handle != NULL )
{
memory_free(
*info_handle );
*info_handle = NULL;
}
return( -1 );
}
/* Tests single open and close of a file
* Returns 1 if successful, 0 if not or -1 on error
*/
int pff_test_single_open_close_file(
libcstring_system_character_t *filename,
int access_flags,
int expected_result )
{
libcerror_error_t *error = NULL;
libpff_file_t *file = NULL;
static char *function = "pff_test_single_open_close_file";
char *access_string = NULL;
int result = 0;
if( access_flags == LIBPFF_OPEN_READ )
{
access_string = "read";
}
else if( access_flags == LIBPFF_OPEN_WRITE )
{
access_string = "write";
}
else
{
access_string = "UNKNOWN";
}
fprintf(
stdout,
"Testing single open close of: " );
if( filename != NULL )
{
fprintf(
stdout,
"%" PRIs_LIBCSTRING_SYSTEM "",
filename );
}
else
{
fprintf(
stdout,
"NULL" );
}
fprintf(
stdout,
" with access: %s\t",
access_string );
if( libpff_file_initialize(
&file,
&error ) != 1 )
{
libcerror_error_set(
&error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,
"%s: unable to create file.",
function );
goto on_error;
}
#if defined( LIBCSTRING_HAVE_WIDE_SYSTEM_CHARACTER )
result = libpff_file_open_wide(
file,
filename,
access_flags,
&error );
#else
result = libpff_file_open(
file,
filename,
access_flags,
&error );
#endif
if( result == 1 )
{
if( libpff_file_close(
file,
&error ) != 0 )
{
libcerror_error_set(
&error,
LIBCERROR_ERROR_DOMAIN_IO,
LIBCERROR_IO_ERROR_CLOSE_FAILED,
"%s: unable to close file.",
function );
goto on_error;
}
}
if( libpff_file_free(
&file,
&error ) != 1 )
{
libcerror_error_set(
&error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_FINALIZE_FAILED,
"%s: unable to free file.",
function );
goto on_error;
}
result = ( expected_result == result );
if( result == 1 )
{
fprintf(
stdout,
"(PASS)" );
}
else
{
fprintf(
stdout,
"(FAIL)" );
}
fprintf(
stdout,
"\n" );
if( error != NULL )
{
libcerror_error_backtrace_fprint(
error,
stderr );
libcerror_error_free(
&error );
}
return( result );
on_error:
if( error != NULL )
{
libcerror_error_backtrace_fprint(
error,
stderr );
libcerror_error_free(
&error );
}
if( file != NULL )
{
libpff_file_free(
&file,
NULL);
}
return( -1 );
}
int main( int argc, char * const argv[] )
#endif
{
libcerror_error_t *error = NULL;
libpff_file_t *file = NULL;
size_t string_length = 0;
uint32_t feature_flags = 0;
int ascii_codepage = 0;
int result = 0;
if( argc != 2 )
{
fprintf(
stderr,
"Unsupported number of arguments.\n" );
return( EXIT_FAILURE );
}
feature_flags = LIBCLOCALE_CODEPAGE_FEATURE_FLAG_HAVE_ISO_8859
| LIBCLOCALE_CODEPAGE_FEATURE_FLAG_HAVE_KOI8
| LIBCLOCALE_CODEPAGE_FEATURE_FLAG_HAVE_WINDOWS;
string_length = libcstring_system_string_length(
argv[ 1 ] );
#if defined( LIBCSTRING_HAVE_WIDE_SYSTEM_CHARACTER )
result = libclocale_codepage_copy_from_string_wide(
&ascii_codepage,
argv[ 1 ],
string_length,
feature_flags,
&error );
#else
result = libclocale_codepage_copy_from_string(
&ascii_codepage,
argv[ 1 ],
string_length,
feature_flags,
&error );
#endif
if( result == -1 )
{
fprintf(
stderr,
"Unable to determine ASCII codepage from: %" PRIs_LIBCSTRING_SYSTEM ".\n",
argv[ 1 ] );
goto on_error;
}
if( libpff_file_initialize(
&file,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to create file.\n" );
goto on_error;
}
if( libpff_file_set_ascii_codepage(
file,
ascii_codepage,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to set codepage: %" PRIs_LIBCSTRING_SYSTEM ".\n",
argv[ 1 ] );
goto on_error;
}
if( libpff_file_free(
&file,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to free file.\n" );
goto on_error;
}
return( EXIT_SUCCESS );
on_error:
if( error != NULL )
{
libcerror_error_backtrace_fprint(
error,
stdout );
libcerror_error_free(
&error );
}
if( file != NULL )
{
libpff_file_free(
&file,
NULL );
}
return( -1 );
}
int main( int argc, char * const argv[] )
#endif
{
libcerror_error_t *error = NULL;
log_handle_t *log_handle = NULL;
libcstring_system_character_t *log_filename = NULL;
libcstring_system_character_t *option_ascii_codepage = NULL;
libcstring_system_character_t *option_export_mode = NULL;
libcstring_system_character_t *option_preferred_export_format = NULL;
libcstring_system_character_t *option_target_path = NULL;
libcstring_system_character_t *path_separator = NULL;
libcstring_system_character_t *source = NULL;
char *program = "pffexport";
size_t source_length = 0;
libcstring_system_integer_t option = 0;
uint8_t dump_item_values = 0;
uint8_t print_status_information = 1;
int result = 0;
int verbose = 0;
libcnotify_stream_set(
stderr,
NULL );
libcnotify_verbose_set(
1 );
if( libclocale_initialize(
"pfftools",
&error ) != 1 )
{
fprintf(
stderr,
"Unable to initialize locale values.\n" );
goto on_error;
}
if( libcsystem_initialize(
_IONBF,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to initialize system values.\n" );
goto on_error;
}
pffoutput_version_fprint(
stdout,
program );
while( ( option = libcsystem_getopt(
argc,
argv,
_LIBCSTRING_SYSTEM_STRING( "c:df:hl:m:qt:vV" ) ) ) != (libcstring_system_integer_t) -1 )
{
switch( option )
{
case (libcstring_system_integer_t) '?':
default:
fprintf(
stderr,
"Invalid argument: %" PRIs_LIBCSTRING_SYSTEM "\n",
argv[ optind - 1 ] );
usage_fprint(
stdout );
return( EXIT_FAILURE );
case (libcstring_system_integer_t) 'c':
option_ascii_codepage = optarg;
break;
case (libcstring_system_integer_t) 'd':
dump_item_values = 1;
break;
case (libcstring_system_integer_t) 'f':
option_preferred_export_format = optarg;
break;
case (libcstring_system_integer_t) 'h':
usage_fprint(
stdout );
return( EXIT_SUCCESS );
case (libcstring_system_integer_t) 'l':
log_filename = optarg;
break;
case (libcstring_system_integer_t) 'm':
option_export_mode = optarg;
break;
case (libcstring_system_integer_t) 'q':
print_status_information = 0;
break;
case (libcstring_system_integer_t) 't':
option_target_path = optarg;
break;
case (libcstring_system_integer_t) 'v':
verbose = 1;
break;
case (libcstring_system_integer_t) 'V':
pffoutput_copyright_fprint(
stdout );
return( EXIT_SUCCESS );
}
}
if( optind == argc )
{
fprintf(
stderr,
"Missing source file.\n" );
usage_fprint(
stdout );
return( EXIT_FAILURE );
}
source = argv[ optind ];
if( option_target_path == NULL )
{
source_length = libcstring_system_string_length(
source );
path_separator = libcstring_system_string_search_character_reverse(
source,
(libcstring_system_character_t) LIBCPATH_SEPARATOR,
source_length );
if( path_separator == NULL )
{
path_separator = source;
}
else
{
path_separator++;
}
option_target_path = path_separator;
}
libcnotify_verbose_set(
verbose );
libpff_notify_set_stream(
stderr,
NULL );
libpff_notify_set_verbose(
verbose );
if( export_handle_initialize(
&pffexport_export_handle,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to create export handle.\n" );
goto on_error;
}
pffexport_export_handle->print_status_information = print_status_information;
if( option_export_mode != NULL )
{
result = export_handle_set_export_mode(
pffexport_export_handle,
option_export_mode,
&error );
if( result == -1 )
{
fprintf(
stderr,
"Unable to set export mode.\n" );
goto on_error;
}
else if( result == 0 )
{
fprintf(
stderr,
"Unsupported export mode defaulting to: items.\n" );
}
}
pffexport_export_handle->dump_item_values = dump_item_values;
if( option_preferred_export_format != NULL )
{
result = export_handle_set_preferred_export_format(
pffexport_export_handle,
option_preferred_export_format,
&error );
if( result == -1 )
{
fprintf(
stderr,
"Unable to set preferred export format.\n" );
goto on_error;
}
else if( result == 0 )
{
fprintf(
stderr,
"Unsupported preferred export format defaulting to: text.\n" );
}
}
if( option_ascii_codepage != NULL )
{
result = export_handle_set_ascii_codepage(
pffexport_export_handle,
option_ascii_codepage,
&error );
if( result == -1 )
{
fprintf(
stderr,
"Unable to set ASCII codepage in export handle.\n" );
goto on_error;
}
else if( result == 0 )
{
fprintf(
stderr,
"Unsupported ASCII codepage defaulting to: windows-1252.\n" );
}
}
if( export_handle_set_target_path(
pffexport_export_handle,
option_target_path,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to set target path.\n" );
goto on_error;
}
result = export_handle_create_items_export_path(
pffexport_export_handle,
&error );
if( result == -1 )
{
fprintf(
stderr,
"Unable to create items export path.\n" );
goto on_error;
}
else if( result == 0 )
{
fprintf(
stderr,
"%" PRIs_LIBCSTRING_SYSTEM " already exists.\n",
pffexport_export_handle->items_export_path );
goto on_error;
}
result = export_handle_create_orphans_export_path(
pffexport_export_handle,
&error );
if( result == -1 )
{
fprintf(
stderr,
"Unable to create orphans export path.\n" );
goto on_error;
}
else if( result == 0 )
{
fprintf(
stderr,
"%" PRIs_LIBCSTRING_SYSTEM " already exists.\n",
pffexport_export_handle->orphans_export_path );
goto on_error;
}
result = export_handle_create_recovered_export_path(
pffexport_export_handle,
&error );
if( result == -1 )
{
fprintf(
stderr,
"Unable to create recovered export path.\n" );
goto on_error;
}
else if( result == 0 )
{
fprintf(
stderr,
"%" PRIs_LIBCSTRING_SYSTEM " already exists.\n",
pffexport_export_handle->recovered_export_path );
goto on_error;
}
if( log_handle_initialize(
&log_handle,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to create log handle.\n" );
goto on_error;
}
if( log_handle_open(
log_handle,
log_filename,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to open log file: %" PRIs_LIBCSTRING_SYSTEM ".\n",
log_filename );
goto on_error;
}
if( libpff_file_initialize(
&pffexport_file,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to create file.\n" );
goto on_error;
}
if( libpff_file_set_ascii_codepage(
pffexport_file,
pffexport_export_handle->ascii_codepage,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to set ASCII codepage.\n" );
goto on_error;
}
if( libcsystem_signal_attach(
pffexport_signal_handler,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to attach signal handler.\n" );
goto on_error;
}
fprintf(
stdout,
"Opening file.\n" );
#if defined( LIBCSTRING_HAVE_WIDE_SYSTEM_CHARACTER )
if( libpff_file_open_wide(
pffexport_file,
source,
LIBPFF_OPEN_READ,
&error ) != 1 )
#else
if( libpff_file_open(
pffexport_file,
source,
LIBPFF_OPEN_READ,
&error ) != 1 )
#endif
{
fprintf(
stderr,
"Error opening file: %" PRIs_LIBCSTRING_SYSTEM ".\n",
source );
goto on_error;
}
if( export_handle_export_file(
pffexport_export_handle,
pffexport_file,
log_handle,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to export file.\n" );
goto on_error;
}
if( libpff_file_close(
pffexport_file,
&error ) != 0 )
{
fprintf(
stderr,
"Unable to close file.\n" );
goto on_error;
}
if( libcsystem_signal_detach(
&error ) != 1 )
{
fprintf(
stderr,
"Unable to detach signal handler.\n" );
goto on_error;
}
if( libpff_file_free(
&pffexport_file,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to free file.\n" );
goto on_error;
}
if( log_handle_close(
log_handle,
&error ) != 0 )
{
fprintf(
stderr,
"Unable to close log file.\n" );
goto on_error;
}
if( log_handle_free(
&log_handle,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to free log handle.\n" );
goto on_error;
}
if( export_handle_free(
&pffexport_export_handle,
&error ) != 1 )
{
fprintf(
stderr,
"Unable to free export handle.\n" );
goto on_error;
}
if( pffexport_abort != 0 )
{
fprintf(
stdout,
"Export aborted.\n" );
return( EXIT_FAILURE );
}
fprintf(
stdout,
"Export completed.\n" );
return( EXIT_SUCCESS );
on_error:
if( error != NULL )
{
libcnotify_print_error_backtrace(
error );
libcerror_error_free(
&error );
}
if( pffexport_file != NULL )
{
libpff_file_close(
pffexport_file,
NULL );
libpff_file_free(
&pffexport_file,
NULL );
}
if( log_handle != NULL )
{
log_handle_close(
log_handle,
NULL );
log_handle_free(
&log_handle,
NULL );
}
if( pffexport_export_handle != NULL )
{
export_handle_free(
&pffexport_export_handle,
NULL );
}
return( EXIT_FAILURE );
}