Worked on OSS-Fuzz targets

Author: joachimmetz

ossfuzz/partition_fuzzer.cc

@@ -47,10 +47,18 @@ int LLVMFuzzerTestOneInput(
      const uint8_t *data,
      size_t size )
 {
+	uint8_t buffer[ 512 ];
+	char string[ 64 ];
+
 	libbfio_handle_t *file_io_handle = NULL;
-	libvsbsdl_partition_t *partition  = NULL;
-	libvsbsdl_volume_t *volume        = NULL;
+	libvsbsdl_partition_t *partition = NULL;
+	libvsbsdl_volume_t *volume       = NULL;
+	off64_t partition_offset         = 0;
+	off64_t volume_offset            = 0;
+	size64_t partition_size          = 0;
+	uint16_t value_16bit             = 0;
 	int number_of_partitions         = 0;
+	int read_iterator                = 0;
 
 	if( libbfio_memory_range_initialize(
 	     &file_io_handle,
@@ -93,12 +101,62 @@ int LLVMFuzzerTestOneInput(
 		     volume,
 		     0,
 		     &partition,
-		     NULL ) == 1 )
+		     NULL ) != 1 )
+		{
+			goto on_error_libvsbsdl_volume;
+		}
+		if( libvsbsdl_partition_get_entry_index(
+		     partition,
+		     &value_16bit,
+		     NULL ) != 1 )
+		{
+			goto on_error_libvsbsdl_partition;
+		}
+		if( libvsbsdl_partition_get_name_string(
+		     partition,
+		     string,
+		     64,
+		     NULL ) != 1 )
+		{
+			goto on_error_libvsbsdl_partition;
+		}
+		if( libvsbsdl_partition_get_volume_offset(
+		     partition,
+		     &volume_offset,
+		     NULL ) != 1 )
+		{
+			goto on_error_libvsbsdl_partition;
+		}
+		if( libvsbsdl_partition_get_size(
+		     partition,
+		     &partition_size,
+		     NULL ) != 1 )
+		{
+			goto on_error_libvsbsdl_partition;
+		}
+		for( read_iterator = 0;
+		     read_iterator < 128;
+		     read_iterator++ )
 		{
-			libvsbsdl_partition_free(
-			 &partition,
-			 NULL );
+			if( partition_offset >= partition_size )
+			{
+				break;
+			}
+			if( libvsbsdl_partition_read_buffer_at_offset(
+			     partition,
+			     buffer,
+			     497,
+			     partition_offset,
+			     NULL ) == -1 )
+			{
+				goto on_error_libvsbsdl_partition;
+			}
+			partition_offset += 497;
 		}
+on_error_libvsbsdl_partition:
+		libvsbsdl_partition_free(
+		 &partition,
+		 NULL );
 	}
 	libvsbsdl_volume_close(
 	 volume,

ossfuzz/volume_fuzzer.cc

@@ -49,6 +49,8 @@ int LLVMFuzzerTestOneInput(
 {
 	libbfio_handle_t *file_io_handle = NULL;
 	libvsbsdl_volume_t *volume       = NULL;
+	uint32_t value_32bit             = 0;
+	int number_of_partitions         = 0;
 
 	if( libbfio_memory_range_initialize(
 	     &file_io_handle,
@@ -78,6 +80,20 @@ int LLVMFuzzerTestOneInput(
 	{
 		goto on_error_libvsbsdl;
 	}
+	if( libvsbsdl_volume_get_bytes_per_sector(
+	     volume,
+	     &value_32bit,
+	     NULL ) != 1 )
+	{
+		goto on_error_libvsbsdl;
+	}
+	if( libvsbsdl_volume_get_number_of_partitions(
+	     volume,
+	     &number_of_partitions,
+	     NULL ) != 1 )
+	{
+		goto on_error_libvsbsdl;
+	}
 	libvsbsdl_volume_close(
 	 volume,
 	 NULL );