/* Determines if a file contains a NTFS volume signature
* Returns 1 if true, 0 if not or -1 on error
*/
int libfsntfs_check_volume_signature_wide(
const wchar_t *filename,
libcerror_error_t **error )
{
libbfio_handle_t *file_io_handle = NULL;
static char *function = "libfsntfs_check_volume_signature_wide";
size_t filename_length = 0;
int result = 0;
if( filename == NULL )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
"%s: invalid filename.",
function );
return( -1 );
}
filename_length = libcstring_wide_string_length(
filename );
if( filename_length == 0 )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_ARGUMENTS,
LIBCERROR_ARGUMENT_ERROR_INVALID_VALUE,
"%s: invalid filename.",
function );
goto on_error;
}
if( libbfio_file_initialize(
&file_io_handle,
error ) != 1 )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,
"%s: unable to create file IO handle.",
function );
goto on_error;
}
if( libbfio_file_set_name_wide(
file_io_handle,
filename,
filename_length,
error ) != 1 )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_SET_FAILED,
"%s: unable to set filename in file IO handle.",
function );
goto on_error;
}
result = libfsntfs_check_volume_signature_file_io_handle(
file_io_handle,
error );
if( result == -1 )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_GET_FAILED,
"%s: unable to check file signature using a file handle.",
function );
goto on_error;
}
if( libbfio_handle_free(
&file_io_handle,
error ) != 1 )
{
libcerror_error_set(
error,
LIBCERROR_ERROR_DOMAIN_RUNTIME,
LIBCERROR_RUNTIME_ERROR_INITIALIZE_FAILED,
"%s: unable to free file IO handle.",
function );
goto on_error;
}
return( result );
on_error:
if( file_io_handle != NULL )
{
libbfio_handle_free(
&file_io_handle,
NULL );
}
return( -1 );
}
/* Tests the libfsntfs_check_volume_signature_file_io_handle function
* Returns 1 if successful or 0 if not
*/
int fsntfs_test_check_volume_signature_file_io_handle(
const system_character_t *source )
{
uint8_t empty_block[ 8192 ];
libbfio_handle_t *file_io_handle = NULL;
libcerror_error_t *error = NULL;
void *memset_result = NULL;
size_t source_length = 0;
int result = 0;
/* Initialize test
*/
memset_result = memory_set(
empty_block,
0,
sizeof( uint8_t ) * 8192 );
FSNTFS_TEST_ASSERT_IS_NOT_NULL(
"memset_result",
memset_result );
if( source != NULL )
{
/* 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 );
source_length = system_string_length(
source );
#if defined( HAVE_WIDE_SYSTEM_CHARACTER )
result = libbfio_file_set_name_wide(
file_io_handle,
source,
source_length,
&error );
#else
result = libbfio_file_set_name(
file_io_handle,
source,
source_length,
&error );
#endif
FSNTFS_TEST_ASSERT_EQUAL_INT(
"result",
result,
1 );
FSNTFS_TEST_ASSERT_IS_NULL(
"error",
error );
result = libbfio_handle_open(
file_io_handle,
LIBBFIO_OPEN_READ,
&error );
FSNTFS_TEST_ASSERT_EQUAL_INT(
"result",
result,
1 );
FSNTFS_TEST_ASSERT_IS_NULL(
"error",
error );
/* Test check volume signature
*/
result = libfsntfs_check_volume_signature_file_io_handle(
file_io_handle,
&error );
FSNTFS_TEST_ASSERT_EQUAL_INT(
"result",
result,
1 );
FSNTFS_TEST_ASSERT_IS_NULL(
"error",
error );
}
/* Test error cases
*/
result = libfsntfs_check_volume_signature_file_io_handle(
NULL,
&error );
FSNTFS_TEST_ASSERT_EQUAL_INT(
"result",
result,
-1 );
FSNTFS_TEST_ASSERT_IS_NOT_NULL(
"error",
error );
libcerror_error_free(
&error );
/* Clean up
*/
if( source != NULL )
{
result = libbfio_handle_close(
file_io_handle,
&error );
FSNTFS_TEST_ASSERT_EQUAL_INT(
"result",
result,
0 );
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 );
}
/* Test check volume signature with data too small
*/
result = fsntfs_test_open_file_io_handle(
&file_io_handle,
empty_block,
sizeof( uint8_t ) * 1,
&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 );
result = libfsntfs_check_volume_signature_file_io_handle(
file_io_handle,
&error );
FSNTFS_TEST_ASSERT_EQUAL_INT(
"result",
result,
-1 );
FSNTFS_TEST_ASSERT_IS_NOT_NULL(
"error",
error );
libcerror_error_free(
&error );
result = fsntfs_test_close_file_io_handle(
&file_io_handle,
&error );
FSNTFS_TEST_ASSERT_EQUAL_INT(
"result",
result,
0 );
FSNTFS_TEST_ASSERT_IS_NULL(
"error",
error );
/* Test check volume signature with empty block
*/
result = fsntfs_test_open_file_io_handle(
&file_io_handle,
empty_block,
sizeof( uint8_t ) * 8192,
&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 );
result = libfsntfs_check_volume_signature_file_io_handle(
file_io_handle,
&error );
FSNTFS_TEST_ASSERT_EQUAL_INT(
"result",
result,
0 );
FSNTFS_TEST_ASSERT_IS_NULL(
"error",
error );
result = fsntfs_test_close_file_io_handle(
&file_io_handle,
&error );
FSNTFS_TEST_ASSERT_EQUAL_INT(
"result",
result,
0 );
FSNTFS_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 );
}
return( 0 );
}
