mirror of
https://github.com/libyal/libexe
synced 2026-06-08 15:29:55 +00:00
Joachim Metz
421 lines
7.6 KiB
C
421 lines
7.6 KiB
C
/*
|
|
* Library export_table type test program
|
|
*
|
|
* Copyright (C) 2011-2026, Joachim Metz <joachim.metz@gmail.com>
|
|
*
|
|
* 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 <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <common.h>
|
|
#include <file_stream.h>
|
|
#include <types.h>
|
|
|
|
#if defined( HAVE_STDLIB_H ) || defined( WINAPI )
|
|
#include <stdlib.h>
|
|
#endif
|
|
|
|
#include "exe_test_libcerror.h"
|
|
#include "exe_test_libexe.h"
|
|
#include "exe_test_macros.h"
|
|
#include "exe_test_memory.h"
|
|
#include "exe_test_unused.h"
|
|
|
|
#include "../libexe/libexe_export_table.h"
|
|
|
|
#if defined( __GNUC__ ) && !defined( LIBEXE_DLL_IMPORT )
|
|
|
|
/* Tests the libexe_export_table_initialize function
|
|
* Returns 1 if successful or 0 if not
|
|
*/
|
|
int exe_test_export_table_initialize(
|
|
void )
|
|
{
|
|
libcerror_error_t *error = NULL;
|
|
libexe_export_table_t *export_table = NULL;
|
|
int result = 0;
|
|
|
|
#if defined( HAVE_EXE_TEST_MEMORY )
|
|
int number_of_malloc_fail_tests = 1;
|
|
int number_of_memset_fail_tests = 1;
|
|
int test_number = 0;
|
|
#endif
|
|
|
|
/* Test regular cases
|
|
*/
|
|
result = libexe_export_table_initialize(
|
|
&export_table,
|
|
&error );
|
|
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NOT_NULL(
|
|
"export_table",
|
|
export_table );
|
|
|
|
EXE_TEST_ASSERT_IS_NULL(
|
|
"error",
|
|
error );
|
|
|
|
result = libexe_export_table_free(
|
|
&export_table,
|
|
&error );
|
|
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NULL(
|
|
"export_table",
|
|
export_table );
|
|
|
|
EXE_TEST_ASSERT_IS_NULL(
|
|
"error",
|
|
error );
|
|
|
|
/* Test error cases
|
|
*/
|
|
result = libexe_export_table_initialize(
|
|
NULL,
|
|
&error );
|
|
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
-1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NOT_NULL(
|
|
"error",
|
|
error );
|
|
|
|
libcerror_error_free(
|
|
&error );
|
|
|
|
export_table = (libexe_export_table_t *) 0x12345678UL;
|
|
|
|
result = libexe_export_table_initialize(
|
|
&export_table,
|
|
&error );
|
|
|
|
export_table = NULL;
|
|
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
-1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NOT_NULL(
|
|
"error",
|
|
error );
|
|
|
|
libcerror_error_free(
|
|
&error );
|
|
|
|
#if defined( HAVE_EXE_TEST_MEMORY )
|
|
|
|
for( test_number = 0;
|
|
test_number < number_of_malloc_fail_tests;
|
|
test_number++ )
|
|
{
|
|
/* Test libexe_export_table_initialize with malloc failing
|
|
*/
|
|
exe_test_malloc_attempts_before_fail = test_number;
|
|
|
|
result = libexe_export_table_initialize(
|
|
&export_table,
|
|
&error );
|
|
|
|
if( exe_test_malloc_attempts_before_fail != -1 )
|
|
{
|
|
exe_test_malloc_attempts_before_fail = -1;
|
|
|
|
if( export_table != NULL )
|
|
{
|
|
libexe_export_table_free(
|
|
&export_table,
|
|
NULL );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
-1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NULL(
|
|
"export_table",
|
|
export_table );
|
|
|
|
EXE_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 libexe_export_table_initialize with memset failing
|
|
*/
|
|
exe_test_memset_attempts_before_fail = test_number;
|
|
|
|
result = libexe_export_table_initialize(
|
|
&export_table,
|
|
&error );
|
|
|
|
if( exe_test_memset_attempts_before_fail != -1 )
|
|
{
|
|
exe_test_memset_attempts_before_fail = -1;
|
|
|
|
if( export_table != NULL )
|
|
{
|
|
libexe_export_table_free(
|
|
&export_table,
|
|
NULL );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
-1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NULL(
|
|
"export_table",
|
|
export_table );
|
|
|
|
EXE_TEST_ASSERT_IS_NOT_NULL(
|
|
"error",
|
|
error );
|
|
|
|
libcerror_error_free(
|
|
&error );
|
|
}
|
|
}
|
|
#endif /* defined( HAVE_EXE_TEST_MEMORY ) */
|
|
|
|
return( 1 );
|
|
|
|
on_error:
|
|
if( error != NULL )
|
|
{
|
|
libcerror_error_free(
|
|
&error );
|
|
}
|
|
if( export_table != NULL )
|
|
{
|
|
libexe_export_table_free(
|
|
&export_table,
|
|
NULL );
|
|
}
|
|
return( 0 );
|
|
}
|
|
|
|
/* Tests the libexe_export_table_free function
|
|
* Returns 1 if successful or 0 if not
|
|
*/
|
|
int exe_test_export_table_free(
|
|
void )
|
|
{
|
|
libcerror_error_t *error = NULL;
|
|
int result = 0;
|
|
|
|
/* Test error cases
|
|
*/
|
|
result = libexe_export_table_free(
|
|
NULL,
|
|
&error );
|
|
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
-1 );
|
|
|
|
EXE_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 libexe_export_table_read function
|
|
* Returns 1 if successful or 0 if not
|
|
*/
|
|
int exe_test_export_table_read(
|
|
void )
|
|
{
|
|
libcerror_error_t *error = NULL;
|
|
libexe_export_table_t *export_table = NULL;
|
|
int result = 0;
|
|
|
|
/* Initialize test
|
|
*/
|
|
result = libexe_export_table_initialize(
|
|
&export_table,
|
|
&error );
|
|
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NOT_NULL(
|
|
"export_table",
|
|
export_table );
|
|
|
|
EXE_TEST_ASSERT_IS_NULL(
|
|
"error",
|
|
error );
|
|
|
|
/* Test error cases
|
|
*/
|
|
result = libexe_export_table_read(
|
|
NULL,
|
|
NULL,
|
|
0,
|
|
0,
|
|
&error );
|
|
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
-1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NOT_NULL(
|
|
"error",
|
|
error );
|
|
|
|
libcerror_error_free(
|
|
&error );
|
|
|
|
result = libexe_export_table_read(
|
|
export_table,
|
|
NULL,
|
|
0,
|
|
0,
|
|
&error );
|
|
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
-1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NOT_NULL(
|
|
"error",
|
|
error );
|
|
|
|
libcerror_error_free(
|
|
&error );
|
|
|
|
/* Clean up
|
|
*/
|
|
result = libexe_export_table_free(
|
|
&export_table,
|
|
&error );
|
|
|
|
EXE_TEST_ASSERT_EQUAL_INT(
|
|
"result",
|
|
result,
|
|
1 );
|
|
|
|
EXE_TEST_ASSERT_IS_NULL(
|
|
"export_table",
|
|
export_table );
|
|
|
|
EXE_TEST_ASSERT_IS_NULL(
|
|
"error",
|
|
error );
|
|
|
|
return( 1 );
|
|
|
|
on_error:
|
|
if( error != NULL )
|
|
{
|
|
libcerror_error_free(
|
|
&error );
|
|
}
|
|
if( export_table != NULL )
|
|
{
|
|
libexe_export_table_free(
|
|
&export_table,
|
|
NULL );
|
|
}
|
|
return( 0 );
|
|
}
|
|
|
|
#endif /* defined( __GNUC__ ) && !defined( LIBEXE_DLL_IMPORT ) */
|
|
|
|
/* The main program
|
|
*/
|
|
#if defined( HAVE_WIDE_SYSTEM_CHARACTER )
|
|
int wmain(
|
|
int argc EXE_TEST_ATTRIBUTE_UNUSED,
|
|
wchar_t * const argv[] EXE_TEST_ATTRIBUTE_UNUSED )
|
|
#else
|
|
int main(
|
|
int argc EXE_TEST_ATTRIBUTE_UNUSED,
|
|
char * const argv[] EXE_TEST_ATTRIBUTE_UNUSED )
|
|
#endif
|
|
{
|
|
EXE_TEST_UNREFERENCED_PARAMETER( argc )
|
|
EXE_TEST_UNREFERENCED_PARAMETER( argv )
|
|
|
|
#if defined( __GNUC__ ) && !defined( LIBEXE_DLL_IMPORT )
|
|
|
|
EXE_TEST_RUN(
|
|
"libexe_export_table_initialize",
|
|
exe_test_export_table_initialize );
|
|
|
|
EXE_TEST_RUN(
|
|
"libexe_export_table_free",
|
|
exe_test_export_table_free );
|
|
|
|
EXE_TEST_RUN(
|
|
"libexe_export_table_read",
|
|
exe_test_export_table_read );
|
|
|
|
#endif /* defined( __GNUC__ ) && !defined( LIBEXE_DLL_IMPORT ) */
|
|
|
|
return( EXIT_SUCCESS );
|
|
|
|
#if defined( __GNUC__ ) && !defined( LIBEXE_DLL_IMPORT )
|
|
|
|
on_error:
|
|
return( EXIT_FAILURE );
|
|
|
|
#endif /* defined( __GNUC__ ) && !defined( LIBEXE_DLL_IMPORT ) */
|
|
}
|
|
|