File: /home/django/libpff/tests/pff_test_block_tree_node.c
/*
* Library block_tree_node type test program
*
* Copyright (C) 2008-2024, 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 "pff_test_libcerror.h"
#include "pff_test_libpff.h"
#include "pff_test_macros.h"
#include "pff_test_memory.h"
#include "pff_test_unused.h"
#include "../libpff/libpff_block_tree_node.h"
#if defined( __GNUC__ ) && !defined( LIBPFF_DLL_IMPORT )
/* Tests the libpff_block_tree_node_initialize function
* Returns 1 if successful or 0 if not
*/
int pff_test_block_tree_node_initialize(
void )
{
libcerror_error_t *error = NULL;
libpff_block_tree_node_t *block_tree_node = NULL;
int result = 0;
#if defined( HAVE_PFF_TEST_MEMORY )
int number_of_malloc_fail_tests = 1;
int number_of_memset_fail_tests = 1;
int test_number = 0;
#endif
/* Test block_tree_node initialization
*/
result = libpff_block_tree_node_initialize(
&block_tree_node,
0,
1024,
256,
&error );
PFF_TEST_ASSERT_EQUAL_INT(
"result",
result,
1 );
PFF_TEST_ASSERT_IS_NOT_NULL(
"block_tree_node",
block_tree_node );
PFF_TEST_ASSERT_IS_NULL(
"error",
error );
result = libpff_block_tree_node_free(
&block_tree_node,
NULL,
&error );
PFF_TEST_ASSERT_EQUAL_INT(
"result",
result,
1 );
PFF_TEST_ASSERT_IS_NULL(
"block_tree_node",
block_tree_node );
PFF_TEST_ASSERT_IS_NULL(
"error",
error );
/* Test error cases
*/
result = libpff_block_tree_node_initialize(
NULL,
0,
1024,
256,
&error );
PFF_TEST_ASSERT_EQUAL_INT(
"result",
result,
-1 );
PFF_TEST_ASSERT_IS_NOT_NULL(
"error",
error );
libcerror_error_free(
&error );
block_tree_node = (libpff_block_tree_node_t *) 0x12345678UL;
result = libpff_block_tree_node_initialize(
&block_tree_node,
0,
1024,
256,
&error );
block_tree_node = NULL;
PFF_TEST_ASSERT_EQUAL_INT(
"result",
result,
-1 );
PFF_TEST_ASSERT_IS_NOT_NULL(
"error",
error );
libcerror_error_free(
&error );
#if defined( HAVE_PFF_TEST_MEMORY )
for( test_number = 0;
test_number < number_of_malloc_fail_tests;
test_number++ )
{
/* Test libpff_block_tree_node_initialize with malloc failing
*/
pff_test_malloc_attempts_before_fail = test_number;
result = libpff_block_tree_node_initialize(
&block_tree_node,
0,
1024,
256,
&error );
if( pff_test_malloc_attempts_before_fail != -1 )
{
pff_test_malloc_attempts_before_fail = -1;
if( block_tree_node != NULL )
{
libpff_block_tree_node_free(
&block_tree_node,
NULL,
NULL );
}
}
else
{
PFF_TEST_ASSERT_EQUAL_INT(
"result",
result,
-1 );
PFF_TEST_ASSERT_IS_NULL(
"block_tree_node",
block_tree_node );
PFF_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 libpff_block_tree_node_initialize with memset failing
*/
pff_test_memset_attempts_before_fail = test_number;
result = libpff_block_tree_node_initialize(
&block_tree_node,
0,
1024,
256,
&error );
if( pff_test_memset_attempts_before_fail != -1 )
{
pff_test_memset_attempts_before_fail = -1;
if( block_tree_node != NULL )
{
libpff_block_tree_node_free(
&block_tree_node,
NULL,
NULL );
}
}
else
{
PFF_TEST_ASSERT_EQUAL_INT(
"result",
result,
-1 );
PFF_TEST_ASSERT_IS_NULL(
"block_tree_node",
block_tree_node );
PFF_TEST_ASSERT_IS_NOT_NULL(
"error",
error );
libcerror_error_free(
&error );
}
}
#endif /* defined( HAVE_PFF_TEST_MEMORY ) */
return( 1 );
on_error:
if( error != NULL )
{
libcerror_error_free(
&error );
}
if( block_tree_node != NULL )
{
libpff_block_tree_node_free(
&block_tree_node,
NULL,
NULL );
}
return( 0 );
}
/* Tests the libpff_block_tree_node_free function
* Returns 1 if successful or 0 if not
*/
int pff_test_block_tree_node_free(
void )
{
libcerror_error_t *error = NULL;
int result = 0;
/* Test error cases
*/
result = libpff_block_tree_node_free(
NULL,
NULL,
&error );
PFF_TEST_ASSERT_EQUAL_INT(
"result",
result,
-1 );
PFF_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( LIBPFF_DLL_IMPORT ) */
/* The main program
*/
#if defined( HAVE_WIDE_SYSTEM_CHARACTER )
int wmain(
int argc PFF_TEST_ATTRIBUTE_UNUSED,
wchar_t * const argv[] PFF_TEST_ATTRIBUTE_UNUSED )
#else
int main(
int argc PFF_TEST_ATTRIBUTE_UNUSED,
char * const argv[] PFF_TEST_ATTRIBUTE_UNUSED )
#endif
{
PFF_TEST_UNREFERENCED_PARAMETER( argc )
PFF_TEST_UNREFERENCED_PARAMETER( argv )
#if defined( __GNUC__ ) && !defined( LIBPFF_DLL_IMPORT )
PFF_TEST_RUN(
"libpff_block_tree_node_initialize",
pff_test_block_tree_node_initialize );
PFF_TEST_RUN(
"libpff_block_tree_node_free",
pff_test_block_tree_node_free );
#endif /* defined( __GNUC__ ) && !defined( LIBPFF_DLL_IMPORT ) */
return( EXIT_SUCCESS );
#if defined( __GNUC__ ) && !defined( LIBPFF_DLL_IMPORT )
on_error:
return( EXIT_FAILURE );
#endif /* defined( __GNUC__ ) && !defined( LIBPFF_DLL_IMPORT ) */
}