/* * Library volume type test program * * Copyright (C) 2010-2019, Joachim Metz * * Refer to AUTHORS for acknowledgements. * * This software is free software: you can redistribute it and/or modify * it under the terms of the GNU Lesser General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This software 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 Lesser General Public License * along with this software. If not, see . */ #include #include #include #include #include #include #if defined( HAVE_STDLIB_H ) || defined( WINAPI ) #include #endif #include "fsntfs_test_functions.h" #include "fsntfs_test_getopt.h" #include "fsntfs_test_libbfio.h" #include "fsntfs_test_libcerror.h" #include "fsntfs_test_libfsntfs.h" #include "fsntfs_test_macros.h" #include "fsntfs_test_memory.h" #include "../libfsntfs/libfsntfs_volume.h" #if defined( HAVE_WIDE_SYSTEM_CHARACTER ) && SIZEOF_WCHAR_T != 2 && SIZEOF_WCHAR_T != 4 #error Unsupported size of wchar_t #endif /* Define to make fsntfs_test_volume generate verbose output #define FSNTFS_TEST_VOLUME_VERBOSE */ #if !defined( LIBFSNTFS_HAVE_BFIO ) LIBFSNTFS_EXTERN \ int libfsntfs_check_volume_signature_file_io_handle( libbfio_handle_t *file_io_handle, libcerror_error_t **error ); LIBFSNTFS_EXTERN \ int libfsntfs_volume_open_file_io_handle( libfsntfs_volume_t *volume, libbfio_handle_t *file_io_handle, int access_flags, libfsntfs_error_t **error ); #endif /* !defined( LIBFSNTFS_HAVE_BFIO ) */ /* Creates and opens a source volume * Returns 1 if successful or -1 on error */ int fsntfs_test_volume_open_source( libfsntfs_volume_t **volume, libbfio_handle_t *file_io_handle, libcerror_error_t **error ) { static char *function = "fsntfs_test_volume_open_source"; int result = 0; if( volume == NULL ) { libcerror_error_set( error, LIBCERROR_ERROR_DOMAIN_ARGUMENTS, LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE, "%s: invalid volume.", function ); return( -1 ); } if( file_io_handle == NULL ) { libcerror_error_set( error, LIBCERROR_ERROR_DOMAIN_ARGUMENTS, LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE, "%s: invalid file IO handle.", function ); return( -1 ); } if( libfsntfs_volume_initialize( volume, error ) != 1 ) { libcerror_error_set( error, LIBCERROR_ERROR_DOMAIN_RUNTIME, LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED, "%s: unable to initialize volume.", function ); goto on_error; } result = libfsntfs_volume_open_file_io_handle( *volume, file_io_handle, LIBFSNTFS_OPEN_READ, error ); if( result != 1 ) { libcerror_error_set( error, LIBCERROR_ERROR_DOMAIN_IO, LIBCERROR_IO_ERROR_OPEN_FAILED, "%s: unable to open volume.", function ); goto on_error; } return( 1 ); on_error: if( *volume != NULL ) { libfsntfs_volume_free( volume, NULL ); } return( -1 ); } /* Closes and frees a source volume * Returns 1 if successful or -1 on error */ int fsntfs_test_volume_close_source( libfsntfs_volume_t **volume, libcerror_error_t **error ) { static char *function = "fsntfs_test_volume_close_source"; int result = 0; if( volume == NULL ) { libcerror_error_set( error, LIBCERROR_ERROR_DOMAIN_ARGUMENTS, LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE, "%s: invalid volume.", function ); return( -1 ); } if( libfsntfs_volume_close( *volume, error ) != 0 ) { libcerror_error_set( error, LIBCERROR_ERROR_DOMAIN_IO, LIBCERROR_IO_ERROR_CLOSE_FAILED, "%s: unable to close volume.", function ); result = -1; } if( libfsntfs_volume_free( volume, error ) != 1 ) { libcerror_error_set( error, LIBCERROR_ERROR_DOMAIN_RUNTIME, LIBCERROR_RUNTIME_ERROR_FINALIZE_FAILED, "%s: unable to free volume.", function ); result = -1; } return( result ); } /* Tests the libfsntfs_volume_initialize function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_initialize( void ) { libcerror_error_t *error = NULL; libfsntfs_volume_t *volume = NULL; int result = 0; #if defined( HAVE_FSNTFS_TEST_MEMORY ) int number_of_malloc_fail_tests = 3; int number_of_memset_fail_tests = 1; int test_number = 0; #endif /* Test regular cases */ result = libfsntfs_volume_initialize( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); result = libfsntfs_volume_free( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libfsntfs_volume_initialize( NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); volume = (libfsntfs_volume_t *) 0x12345678UL; result = libfsntfs_volume_initialize( &volume, &error ); volume = NULL; FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); #if defined( HAVE_FSNTFS_TEST_MEMORY ) for( test_number = 0; test_number < number_of_malloc_fail_tests; test_number++ ) { /* Test libfsntfs_volume_initialize with malloc failing */ fsntfs_test_malloc_attempts_before_fail = test_number; result = libfsntfs_volume_initialize( &volume, &error ); if( fsntfs_test_malloc_attempts_before_fail != -1 ) { fsntfs_test_malloc_attempts_before_fail = -1; if( volume != NULL ) { libfsntfs_volume_free( &volume, NULL ); } } else { FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } } for( test_number = 0; test_number < number_of_memset_fail_tests; test_number++ ) { /* Test libfsntfs_volume_initialize with memset failing */ fsntfs_test_memset_attempts_before_fail = test_number; result = libfsntfs_volume_initialize( &volume, &error ); if( fsntfs_test_memset_attempts_before_fail != -1 ) { fsntfs_test_memset_attempts_before_fail = -1; if( volume != NULL ) { libfsntfs_volume_free( &volume, NULL ); } } else { FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } } #endif /* defined( HAVE_FSNTFS_TEST_MEMORY ) */ return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( volume != NULL ) { libfsntfs_volume_free( &volume, NULL ); } return( 0 ); } /* Tests the libfsntfs_volume_free function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_free( void ) { libcerror_error_t *error = NULL; int result = 0; /* Test error cases */ result = libfsntfs_volume_free( NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_open function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_open( const system_character_t *source ) { char narrow_source[ 256 ]; libcerror_error_t *error = NULL; libfsntfs_volume_t *volume = NULL; int result = 0; /* Initialize test */ result = fsntfs_test_get_narrow_source( source, narrow_source, 256, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); result = libfsntfs_volume_initialize( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test open */ result = libfsntfs_volume_open( volume, narrow_source, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libfsntfs_volume_open( NULL, narrow_source, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_open( volume, NULL, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_open( volume, narrow_source, -1, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); /* Test open when already opened */ result = libfsntfs_volume_open( volume, narrow_source, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); /* Clean up */ result = libfsntfs_volume_free( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( volume != NULL ) { libfsntfs_volume_free( &volume, NULL ); } return( 0 ); } #if defined( HAVE_WIDE_CHARACTER_TYPE ) /* Tests the libfsntfs_volume_open_wide function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_open_wide( const system_character_t *source ) { wchar_t wide_source[ 256 ]; libcerror_error_t *error = NULL; libfsntfs_volume_t *volume = NULL; int result = 0; /* Initialize test */ result = fsntfs_test_get_wide_source( source, wide_source, 256, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); result = libfsntfs_volume_initialize( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test open */ result = libfsntfs_volume_open_wide( volume, wide_source, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libfsntfs_volume_open_wide( NULL, wide_source, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_open_wide( volume, NULL, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_open_wide( volume, wide_source, -1, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); /* Test open when already opened */ result = libfsntfs_volume_open_wide( volume, wide_source, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); /* Clean up */ result = libfsntfs_volume_free( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( volume != NULL ) { libfsntfs_volume_free( &volume, NULL ); } return( 0 ); } #endif /* defined( HAVE_WIDE_CHARACTER_TYPE ) */ /* Tests the libfsntfs_volume_open_file_io_handle function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_open_file_io_handle( const system_character_t *source ) { libbfio_handle_t *file_io_handle = NULL; libcerror_error_t *error = NULL; libfsntfs_volume_t *volume = NULL; size_t string_length = 0; int result = 0; /* Initialize test */ result = libbfio_file_initialize( &file_io_handle, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "file_io_handle", file_io_handle ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); string_length = system_string_length( source ); #if defined( HAVE_WIDE_SYSTEM_CHARACTER ) result = libbfio_file_set_name_wide( file_io_handle, source, string_length, &error ); #else result = libbfio_file_set_name( file_io_handle, source, string_length, &error ); #endif FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); result = libfsntfs_volume_initialize( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test open */ result = libfsntfs_volume_open_file_io_handle( volume, file_io_handle, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libfsntfs_volume_open_file_io_handle( NULL, file_io_handle, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_open_file_io_handle( volume, NULL, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_open_file_io_handle( volume, file_io_handle, -1, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); /* Test open when already opened */ result = libfsntfs_volume_open_file_io_handle( volume, file_io_handle, LIBFSNTFS_OPEN_READ, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); /* Clean up */ result = libfsntfs_volume_free( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); result = libbfio_handle_free( &file_io_handle, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "file_io_handle", file_io_handle ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( volume != NULL ) { libfsntfs_volume_free( &volume, NULL ); } if( file_io_handle != NULL ) { libbfio_handle_free( &file_io_handle, NULL ); } return( 0 ); } /* Tests the libfsntfs_volume_close function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_close( void ) { libcerror_error_t *error = NULL; int result = 0; /* Test error cases */ result = libfsntfs_volume_close( NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_open and libfsntfs_volume_close functions * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_open_close( const system_character_t *source ) { libcerror_error_t *error = NULL; libfsntfs_volume_t *volume = NULL; int result = 0; /* Initialize test */ result = libfsntfs_volume_initialize( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test open and close */ #if defined( HAVE_WIDE_SYSTEM_CHARACTER ) result = libfsntfs_volume_open_wide( volume, source, LIBFSNTFS_OPEN_READ, &error ); #else result = libfsntfs_volume_open( volume, source, LIBFSNTFS_OPEN_READ, &error ); #endif FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); result = libfsntfs_volume_close( volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 0 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test open and close a second time to validate clean up on close */ #if defined( HAVE_WIDE_SYSTEM_CHARACTER ) result = libfsntfs_volume_open_wide( volume, source, LIBFSNTFS_OPEN_READ, &error ); #else result = libfsntfs_volume_open( volume, source, LIBFSNTFS_OPEN_READ, &error ); #endif FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); result = libfsntfs_volume_close( volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 0 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Clean up */ result = libfsntfs_volume_free( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( volume != NULL ) { libfsntfs_volume_free( &volume, NULL ); } return( 0 ); } /* Tests the libfsntfs_volume_signal_abort function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_signal_abort( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; int result = 0; /* Test regular cases */ result = libfsntfs_volume_signal_abort( volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libfsntfs_volume_signal_abort( NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_cluster_block_size function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_cluster_block_size( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; size_t cluster_block_size = 0; int cluster_block_size_is_set = 0; int result = 0; /* Test regular cases */ result = libfsntfs_volume_get_cluster_block_size( volume, &cluster_block_size, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); cluster_block_size_is_set = result; /* Test error cases */ result = libfsntfs_volume_get_cluster_block_size( NULL, &cluster_block_size, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( cluster_block_size_is_set != 0 ) { result = libfsntfs_volume_get_cluster_block_size( volume, NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_mft_entry_size function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_mft_entry_size( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; size32_t mft_entry_size = 0; int mft_entry_size_is_set = 0; int result = 0; /* Test regular cases */ result = libfsntfs_volume_get_mft_entry_size( volume, &mft_entry_size, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); mft_entry_size_is_set = result; /* Test error cases */ result = libfsntfs_volume_get_mft_entry_size( NULL, &mft_entry_size, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( mft_entry_size_is_set != 0 ) { result = libfsntfs_volume_get_mft_entry_size( volume, NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_index_entry_size function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_index_entry_size( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; size32_t index_entry_size = 0; int index_entry_size_is_set = 0; int result = 0; /* Test regular cases */ result = libfsntfs_volume_get_index_entry_size( volume, &index_entry_size, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); index_entry_size_is_set = result; /* Test error cases */ result = libfsntfs_volume_get_index_entry_size( NULL, &index_entry_size, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( index_entry_size_is_set != 0 ) { result = libfsntfs_volume_get_index_entry_size( volume, NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_utf8_name_size function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_utf8_name_size( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; size_t utf8_name_size = 0; int result = 0; int utf8_name_size_is_set = 0; /* Test regular cases */ result = libfsntfs_volume_get_utf8_name_size( volume, &utf8_name_size, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); utf8_name_size_is_set = result; /* Test error cases */ result = libfsntfs_volume_get_utf8_name_size( NULL, &utf8_name_size, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( utf8_name_size_is_set != 0 ) { result = libfsntfs_volume_get_utf8_name_size( volume, NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_utf8_name function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_utf8_name( libfsntfs_volume_t *volume ) { uint8_t utf8_name[ 512 ]; libcerror_error_t *error = NULL; int result = 0; int utf8_name_is_set = 0; /* Test regular cases */ result = libfsntfs_volume_get_utf8_name( volume, utf8_name, 512, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); utf8_name_is_set = result; /* Test error cases */ result = libfsntfs_volume_get_utf8_name( NULL, utf8_name, 512, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( utf8_name_is_set != 0 ) { result = libfsntfs_volume_get_utf8_name( volume, NULL, 512, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_get_utf8_name( volume, utf8_name, 0, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_get_utf8_name( volume, utf8_name, (size_t) SSIZE_MAX + 1, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_utf16_name_size function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_utf16_name_size( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; size_t utf16_name_size = 0; int result = 0; int utf16_name_size_is_set = 0; /* Test regular cases */ result = libfsntfs_volume_get_utf16_name_size( volume, &utf16_name_size, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); utf16_name_size_is_set = result; /* Test error cases */ result = libfsntfs_volume_get_utf16_name_size( NULL, &utf16_name_size, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( utf16_name_size_is_set != 0 ) { result = libfsntfs_volume_get_utf16_name_size( volume, NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_utf16_name function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_utf16_name( libfsntfs_volume_t *volume ) { uint16_t utf16_name[ 512 ]; libcerror_error_t *error = NULL; int result = 0; int utf16_name_is_set = 0; /* Test regular cases */ result = libfsntfs_volume_get_utf16_name( volume, utf16_name, 512, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); utf16_name_is_set = result; /* Test error cases */ result = libfsntfs_volume_get_utf16_name( NULL, utf16_name, 512, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( utf16_name_is_set != 0 ) { result = libfsntfs_volume_get_utf16_name( volume, NULL, 512, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_get_utf16_name( volume, utf16_name, 0, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libfsntfs_volume_get_utf16_name( volume, utf16_name, (size_t) SSIZE_MAX + 1, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_serial_number function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_serial_number( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; uint64_t serial_number = 0; int result = 0; int serial_number_is_set = 0; /* Test regular cases */ result = libfsntfs_volume_get_serial_number( volume, &serial_number, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); serial_number_is_set = result; /* Test error cases */ result = libfsntfs_volume_get_serial_number( NULL, &serial_number, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( serial_number_is_set != 0 ) { result = libfsntfs_volume_get_serial_number( volume, NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_number_of_file_entries function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_number_of_file_entries( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; uint64_t number_of_file_entries = 0; int number_of_file_entries_is_set = 0; int result = 0; /* Test regular cases */ result = libfsntfs_volume_get_number_of_file_entries( volume, &number_of_file_entries, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); number_of_file_entries_is_set = result; /* Test error cases */ result = libfsntfs_volume_get_number_of_file_entries( NULL, &number_of_file_entries, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( number_of_file_entries_is_set != 0 ) { result = libfsntfs_volume_get_number_of_file_entries( volume, NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } /* Tests the libfsntfs_volume_get_root_directory function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_root_directory( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; libfsntfs_file_entry_t *root_directory = 0; int result = 0; int root_directory_is_set = 0; /* Test regular cases */ result = libfsntfs_volume_get_root_directory( volume, &root_directory, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); root_directory_is_set = result; if( root_directory_is_set != 0 ) { FSNTFS_TEST_ASSERT_IS_NOT_NULL( "root_directory", root_directory ); result = libfsntfs_file_entry_free( &root_directory, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); } /* Test error cases */ result = libfsntfs_volume_get_root_directory( NULL, &root_directory, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "root_directory", root_directory ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( root_directory_is_set != 0 ) { result = libfsntfs_volume_get_root_directory( volume, NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "root_directory", root_directory ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( root_directory != NULL ) { libfsntfs_file_entry_free( &root_directory, NULL ); } return( 0 ); } /* Tests the libfsntfs_volume_get_usn_change_journal function * Returns 1 if successful or 0 if not */ int fsntfs_test_volume_get_usn_change_journal( libfsntfs_volume_t *volume ) { libcerror_error_t *error = NULL; libfsntfs_usn_change_journal_t *usn_change_journal = 0; int result = 0; int usn_change_journal_is_set = 0; /* Test regular cases */ result = libfsntfs_volume_get_usn_change_journal( volume, &usn_change_journal, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); usn_change_journal_is_set = result; if( usn_change_journal_is_set != 0 ) { FSNTFS_TEST_ASSERT_IS_NOT_NULL( "usn_change_journal", usn_change_journal ); result = libfsntfs_usn_change_journal_free( &usn_change_journal, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); } /* Test error cases */ result = libfsntfs_volume_get_usn_change_journal( NULL, &usn_change_journal, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "usn_change_journal", usn_change_journal ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); if( usn_change_journal_is_set != 0 ) { result = libfsntfs_volume_get_usn_change_journal( volume, NULL, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "usn_change_journal", usn_change_journal ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( usn_change_journal != NULL ) { libfsntfs_usn_change_journal_free( &usn_change_journal, NULL ); } return( 0 ); } /* The main program */ #if defined( HAVE_WIDE_SYSTEM_CHARACTER ) int wmain( int argc, wchar_t * const argv[] ) #else int main( int argc, char * const argv[] ) #endif { libbfio_handle_t *file_io_handle = NULL; libcerror_error_t *error = NULL; libfsntfs_volume_t *volume = NULL; system_character_t *option_offset = NULL; system_character_t *source = NULL; system_integer_t option = 0; size_t string_length = 0; off64_t volume_offset = 0; int result = 0; while( ( option = fsntfs_test_getopt( argc, argv, _SYSTEM_STRING( "o:" ) ) ) != (system_integer_t) -1 ) { switch( option ) { case (system_integer_t) '?': default: fprintf( stderr, "Invalid argument: %" PRIs_SYSTEM ".\n", argv[ optind - 1 ] ); return( EXIT_FAILURE ); case (system_integer_t) 'o': option_offset = optarg; break; } } if( optind < argc ) { source = argv[ optind ]; } if( option_offset != NULL ) { string_length = system_string_length( option_offset ); result = fsntfs_test_system_string_copy_from_64_bit_in_decimal( option_offset, string_length + 1, (uint64_t *) &volume_offset, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); } #if defined( HAVE_DEBUG_OUTPUT ) && defined( FSNTFS_TEST_VOLUME_VERBOSE ) libfsntfs_notify_set_verbose( 1 ); libfsntfs_notify_set_stream( stderr, NULL ); #endif FSNTFS_TEST_RUN( "libfsntfs_volume_initialize", fsntfs_test_volume_initialize ); FSNTFS_TEST_RUN( "libfsntfs_volume_free", fsntfs_test_volume_free ); #if !defined( __BORLANDC__ ) || ( __BORLANDC__ >= 0x0560 ) if( source != NULL ) { result = libbfio_file_range_initialize( &file_io_handle, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "file_io_handle", file_io_handle ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); string_length = system_string_length( source ); #if defined( HAVE_WIDE_SYSTEM_CHARACTER ) result = libbfio_file_range_set_name_wide( file_io_handle, source, string_length, &error ); #else result = libbfio_file_range_set_name( file_io_handle, source, string_length, &error ); #endif FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); result = libbfio_file_range_set( file_io_handle, volume_offset, 0, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); result = libfsntfs_check_volume_signature_file_io_handle( file_io_handle, &error ); FSNTFS_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); } if( ( result != 0 ) && ( volume_offset == 0 ) ) { FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_open", fsntfs_test_volume_open, source ); #if defined( HAVE_WIDE_CHARACTER_TYPE ) FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_open_wide", fsntfs_test_volume_open_wide, source ); #endif /* defined( HAVE_WIDE_CHARACTER_TYPE ) */ FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_open_file_io_handle", fsntfs_test_volume_open_file_io_handle, source ); FSNTFS_TEST_RUN( "libfsntfs_volume_close", fsntfs_test_volume_close ); FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_open_close", fsntfs_test_volume_open_close, source ); } if( result != 0 ) { /* Initialize volume for tests */ result = fsntfs_test_volume_open_source( &volume, file_io_handle, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NOT_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_signal_abort", fsntfs_test_volume_signal_abort, volume ); /* TODO: add tests for libfsntfs_volume_has_bitlocker_drive_encryption */ /* TODO: add tests for libfsntfs_volume_has_volume_shadow_snapshots */ FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_cluster_block_size", fsntfs_test_volume_get_cluster_block_size, volume ); FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_mft_entry_size", fsntfs_test_volume_get_mft_entry_size, volume ); FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_index_entry_size", fsntfs_test_volume_get_index_entry_size, volume ); FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_utf8_name_size", fsntfs_test_volume_get_utf8_name_size, volume ); FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_utf8_name", fsntfs_test_volume_get_utf8_name, volume ); FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_utf16_name_size", fsntfs_test_volume_get_utf16_name_size, volume ); FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_utf16_name", fsntfs_test_volume_get_utf16_name, volume ); /* TODO: add tests for libfsntfs_volume_get_version */ FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_serial_number", fsntfs_test_volume_get_serial_number, volume ); FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_number_of_file_entries", fsntfs_test_volume_get_number_of_file_entries, volume ); /* TODO: add tests for libfsntfs_volume_get_file_entry_by_index */ FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_root_directory", fsntfs_test_volume_get_root_directory, volume ); /* TODO: add tests for libfsntfs_volume_get_file_entry_by_utf8_path */ /* TODO: add tests for libfsntfs_volume_get_file_entry_by_utf16_path */ FSNTFS_TEST_RUN_WITH_ARGS( "libfsntfs_volume_get_usn_change_journal", fsntfs_test_volume_get_usn_change_journal, volume ); /* Clean up */ result = fsntfs_test_volume_close_source( &volume, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 0 ); FSNTFS_TEST_ASSERT_IS_NULL( "volume", volume ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); } if( file_io_handle != NULL ) { result = libbfio_handle_free( &file_io_handle, &error ); FSNTFS_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); FSNTFS_TEST_ASSERT_IS_NULL( "file_io_handle", file_io_handle ); FSNTFS_TEST_ASSERT_IS_NULL( "error", error ); } #endif /* !defined( __BORLANDC__ ) || ( __BORLANDC__ >= 0x0560 ) */ return( EXIT_SUCCESS ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( volume != NULL ) { libfsntfs_volume_free( &volume, NULL ); } if( file_io_handle != NULL ) { libbfio_handle_free( &file_io_handle, NULL ); } return( EXIT_FAILURE ); }