/* * Library catalog type test program * * Copyright (C) 2009-2023, Joachim Metz * * Refer to AUTHORS for acknowledgements. * * This program 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 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 Lesser General Public License * along with this program. If not, see . */ #include #include #include #if defined( HAVE_STDLIB_H ) || defined( WINAPI ) #include #endif #include "esedb_test_functions.h" #include "esedb_test_libbfio.h" #include "esedb_test_libcdata.h" #include "esedb_test_libcerror.h" #include "esedb_test_libesedb.h" #include "esedb_test_macros.h" #include "esedb_test_memory.h" #include "esedb_test_unused.h" #include "../libesedb/libesedb_catalog.h" #include "../libesedb/libesedb_io_handle.h" #if defined( __GNUC__ ) && !defined( LIBESEDB_DLL_IMPORT ) /* Tests the libesedb_catalog_initialize function * Returns 1 if successful or 0 if not */ int esedb_test_catalog_initialize( void ) { libcerror_error_t *error = NULL; libesedb_catalog_t *catalog = NULL; libesedb_io_handle_t *io_handle = NULL; int result = 0; #if defined( HAVE_ESEDB_TEST_MEMORY ) int number_of_malloc_fail_tests = 4; int number_of_memset_fail_tests = 1; int test_number = 0; #endif /* Initialize test */ result = libesedb_io_handle_initialize( &io_handle, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "io_handle", io_handle ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); io_handle->format_revision = 0x0000000c; io_handle->file_size = 16777216; io_handle->page_size = 4096; /* Test regular cases */ result = libesedb_catalog_initialize( &catalog, io_handle, 1, NULL, NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "catalog", catalog ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); result = libesedb_catalog_free( &catalog, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NULL( "catalog", catalog ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libesedb_catalog_initialize( NULL, io_handle, 1, NULL, NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); catalog = (libesedb_catalog_t *) 0x12345678UL; result = libesedb_catalog_initialize( &catalog, io_handle, 1, NULL, NULL, &error ); catalog = NULL; ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); #if defined( HAVE_ESEDB_TEST_MEMORY ) for( test_number = 0; test_number < number_of_malloc_fail_tests; test_number++ ) { /* Test libesedb_catalog_initialize with malloc failing * Test libesedb_catalog_initialize with malloc failing in libcdata_array_initialize */ esedb_test_malloc_attempts_before_fail = test_number; result = libesedb_catalog_initialize( &catalog, io_handle, 1, NULL, NULL, &error ); if( esedb_test_malloc_attempts_before_fail != -1 ) { esedb_test_malloc_attempts_before_fail = -1; if( catalog != NULL ) { libesedb_catalog_free( &catalog, NULL ); } } else { ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NULL( "catalog", catalog ); ESEDB_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 libesedb_catalog_initialize with memset failing */ esedb_test_memset_attempts_before_fail = test_number; result = libesedb_catalog_initialize( &catalog, io_handle, 1, NULL, NULL, &error ); if( esedb_test_memset_attempts_before_fail != -1 ) { esedb_test_memset_attempts_before_fail = -1; if( catalog != NULL ) { libesedb_catalog_free( &catalog, NULL ); } } else { ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NULL( "catalog", catalog ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); } } #endif /* defined( HAVE_ESEDB_TEST_MEMORY ) */ /* Clean up */ result = libesedb_io_handle_free( &io_handle, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NULL( "io_handle", io_handle ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( catalog != NULL ) { libesedb_catalog_free( &catalog, NULL ); } if( io_handle != NULL ) { libesedb_io_handle_free( &io_handle, NULL ); } return( 0 ); } /* Tests the libesedb_catalog_free function * Returns 1 if successful or 0 if not */ int esedb_test_catalog_free( void ) { libcerror_error_t *error = NULL; int result = 0; /* Test error cases */ result = libesedb_catalog_free( NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_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 libesedb_catalog_read_file_io_handle function * Returns 1 if successful or 0 if not */ int esedb_test_catalog_read_file_io_handle( void ) { uint8_t data[ 4096 ]; libbfio_handle_t *file_io_handle = NULL; libcerror_error_t *error = NULL; libesedb_catalog_t *catalog = NULL; libesedb_io_handle_t *io_handle = NULL; int result = 0; /* Initialize test */ result = libesedb_io_handle_initialize( &io_handle, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "io_handle", io_handle ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); io_handle->format_revision = 0x0000000c; io_handle->file_size = 16777216; io_handle->page_size = 4096; result = libesedb_catalog_initialize( &catalog, io_handle, 1, NULL, NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "catalog", catalog ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); /* Initialize file IO handle */ result = esedb_test_open_file_io_handle( &file_io_handle, data, 4096, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "file_io_handle", file_io_handle ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); /* Test regular cases */ /* TODO implement */ /* Test error cases */ result = libesedb_catalog_read_file_io_handle( NULL, file_io_handle, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); /* Clean up file IO handle */ result = esedb_test_close_file_io_handle( &file_io_handle, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 0 ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); /* Clean up */ result = libesedb_catalog_free( &catalog, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NULL( "catalog", catalog ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); result = libesedb_io_handle_free( &io_handle, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NULL( "io_handle", io_handle ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( file_io_handle != NULL ) { libbfio_handle_free( &file_io_handle, NULL ); } if( catalog != NULL ) { libesedb_catalog_free( &catalog, NULL ); } if( io_handle != NULL ) { libesedb_io_handle_free( &io_handle, NULL ); } return( 0 ); } /* Tests the libesedb_catalog_get_number_of_table_definitions function * Returns 1 if successful or 0 if not */ int esedb_test_catalog_get_number_of_table_definitions( libesedb_catalog_t *catalog ) { libcerror_error_t *error = NULL; int number_of_table_definitions = 0; int result = 0; /* Test regular cases */ result = libesedb_catalog_get_number_of_table_definitions( catalog, &number_of_table_definitions, &error ); ESEDB_TEST_ASSERT_NOT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libesedb_catalog_get_number_of_table_definitions( NULL, &number_of_table_definitions, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_number_of_table_definitions( catalog, NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_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 libesedb_catalog_get_table_definition_by_index function * Returns 1 if successful or 0 if not */ int esedb_test_catalog_get_table_definition_by_index( libesedb_catalog_t *catalog ) { libcerror_error_t *error = NULL; libesedb_table_definition_t *table_definition = NULL; int result = 0; /* Test regular cases */ /* TODO implement */ /* Test error cases */ result = libesedb_catalog_get_table_definition_by_index( NULL, 0, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_index( catalog, 0, NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_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 libesedb_catalog_get_table_definition_by_name function * Returns 1 if successful or 0 if not */ int esedb_test_catalog_get_table_definition_by_name( libesedb_catalog_t *catalog ) { libcdata_array_t *table_definition_array = NULL; libcerror_error_t *error = NULL; libesedb_table_definition_t *table_definition = NULL; char *name = "test"; int result = 0; /* Test regular cases */ result = libesedb_catalog_get_table_definition_by_name( catalog, (uint8_t *) name, 4, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 0 ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libesedb_catalog_get_table_definition_by_name( NULL, (uint8_t *) name, 4, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); table_definition_array = catalog->table_definition_array; catalog->table_definition_array = NULL; result = libesedb_catalog_get_table_definition_by_name( catalog, (uint8_t *) name, 4, &table_definition, &error ); catalog->table_definition_array = table_definition_array; ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_name( catalog, NULL, 4, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_name( catalog, (uint8_t *) name, (size_t) SSIZE_MAX + 1, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_name( catalog, (uint8_t *) name, 4, NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_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 libesedb_catalog_get_table_definition_by_utf8_name function * Returns 1 if successful or 0 if not */ int esedb_test_catalog_get_table_definition_by_utf8_name( libesedb_catalog_t *catalog ) { uint8_t utf8_name[ 5 ] = { 't', 'e', 's', 't', 0 }; libcdata_array_t *table_definition_array = NULL; libcerror_error_t *error = NULL; libesedb_table_definition_t *table_definition = NULL; int result = 0; /* Test regular cases */ result = libesedb_catalog_get_table_definition_by_utf8_name( catalog, utf8_name, 4, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 0 ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libesedb_catalog_get_table_definition_by_utf8_name( NULL, utf8_name, 4, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); table_definition_array = catalog->table_definition_array; catalog->table_definition_array = NULL; result = libesedb_catalog_get_table_definition_by_utf8_name( catalog, utf8_name, 4, &table_definition, &error ); catalog->table_definition_array = table_definition_array; ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_utf8_name( catalog, NULL, 4, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_utf8_name( catalog, utf8_name, (size_t) SSIZE_MAX + 1, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_utf8_name( catalog, utf8_name, 4, NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_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 libesedb_catalog_get_table_definition_by_utf16_name function * Returns 1 if successful or 0 if not */ int esedb_test_catalog_get_table_definition_by_utf16_name( libesedb_catalog_t *catalog ) { uint16_t utf16_name[ 5 ] = { 't', 'e', 's', 't', 0 }; libcdata_array_t *table_definition_array = NULL; libcerror_error_t *error = NULL; libesedb_table_definition_t *table_definition = NULL; int result = 0; /* Test regular cases */ result = libesedb_catalog_get_table_definition_by_utf16_name( catalog, utf16_name, 4, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 0 ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); /* Test error cases */ result = libesedb_catalog_get_table_definition_by_utf16_name( NULL, utf16_name, 4, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); table_definition_array = catalog->table_definition_array; catalog->table_definition_array = NULL; result = libesedb_catalog_get_table_definition_by_utf16_name( catalog, utf16_name, 4, &table_definition, &error ); catalog->table_definition_array = table_definition_array; ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_utf16_name( catalog, NULL, 4, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_utf16_name( catalog, utf16_name, (size_t) SSIZE_MAX + 1, &table_definition, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); result = libesedb_catalog_get_table_definition_by_utf16_name( catalog, utf16_name, 4, NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, -1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "error", error ); libcerror_error_free( &error ); return( 1 ); on_error: if( error != NULL ) { libcerror_error_free( &error ); } return( 0 ); } #endif /* defined( __GNUC__ ) && !defined( LIBESEDB_DLL_IMPORT ) */ /* The main program */ #if defined( HAVE_WIDE_SYSTEM_CHARACTER ) int wmain( int argc ESEDB_TEST_ATTRIBUTE_UNUSED, wchar_t * const argv[] ESEDB_TEST_ATTRIBUTE_UNUSED ) #else int main( int argc ESEDB_TEST_ATTRIBUTE_UNUSED, char * const argv[] ESEDB_TEST_ATTRIBUTE_UNUSED ) #endif { #if defined( __GNUC__ ) && !defined( LIBESEDB_DLL_IMPORT ) libcerror_error_t *error = NULL; libesedb_catalog_t *catalog = NULL; libesedb_io_handle_t *io_handle = NULL; int result = 0; #endif /* defined( __GNUC__ ) && !defined( LIBESEDB_DLL_IMPORT ) */ ESEDB_TEST_UNREFERENCED_PARAMETER( argc ) ESEDB_TEST_UNREFERENCED_PARAMETER( argv ) #if defined( __GNUC__ ) && !defined( LIBESEDB_DLL_IMPORT ) ESEDB_TEST_RUN( "libesedb_catalog_initialize", esedb_test_catalog_initialize ); ESEDB_TEST_RUN( "libesedb_catalog_free", esedb_test_catalog_free ); /* TODO add test for libesedb_catalog_read_value_data */ /* TODO add test for libesedb_catalog_read_values_from_page */ ESEDB_TEST_RUN( "libesedb_catalog_read_file_io_handle", esedb_test_catalog_read_file_io_handle ); #if !defined( __BORLANDC__ ) || ( __BORLANDC__ >= 0x0560 ) /* Initialize test */ result = libesedb_io_handle_initialize( &io_handle, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "io_handle", io_handle ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); io_handle->format_revision = 0x0000000c; io_handle->file_size = 16777216; io_handle->page_size = 4096; result = libesedb_catalog_initialize( &catalog, io_handle, 1, NULL, NULL, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NOT_NULL( "catalog", catalog ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); /* Run tests */ ESEDB_TEST_RUN_WITH_ARGS( "libesedb_catalog_get_number_of_table_definitions", esedb_test_catalog_get_number_of_table_definitions, catalog ); ESEDB_TEST_RUN_WITH_ARGS( "libesedb_catalog_get_table_definition_by_index", esedb_test_catalog_get_table_definition_by_index, catalog ); ESEDB_TEST_RUN_WITH_ARGS( "libesedb_catalog_get_table_definition_by_name", esedb_test_catalog_get_table_definition_by_name, catalog ); ESEDB_TEST_RUN_WITH_ARGS( "libesedb_catalog_get_table_definition_by_utf8_name", esedb_test_catalog_get_table_definition_by_utf8_name, catalog ); ESEDB_TEST_RUN_WITH_ARGS( "libesedb_catalog_get_table_definition_by_utf16_name", esedb_test_catalog_get_table_definition_by_utf16_name, catalog ); /* Clean up */ result = libesedb_catalog_free( &catalog, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NULL( "catalog", catalog ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); result = libesedb_io_handle_free( &io_handle, &error ); ESEDB_TEST_ASSERT_EQUAL_INT( "result", result, 1 ); ESEDB_TEST_ASSERT_IS_NULL( "io_handle", io_handle ); ESEDB_TEST_ASSERT_IS_NULL( "error", error ); #endif /* !defined( __BORLANDC__ ) || ( __BORLANDC__ >= 0x0560 ) */ #endif /* defined( __GNUC__ ) && !defined( LIBESEDB_DLL_IMPORT ) */ return( EXIT_SUCCESS ); #if defined( __GNUC__ ) && !defined( LIBESEDB_DLL_IMPORT ) on_error: if( error != NULL ) { libcerror_error_free( &error ); } if( catalog != NULL ) { libesedb_catalog_free( &catalog, NULL ); } if( io_handle != NULL ) { libesedb_io_handle_free( &io_handle, NULL ); } return( EXIT_FAILURE ); #endif /* defined( __GNUC__ ) && !defined( LIBESEDB_DLL_IMPORT ) */ }