Add fuzzer_metadata

This commit is contained in:
Martijn van Beurden 2022-06-16 16:59:50 +02:00
parent 7e0a0e5723
commit da7b9a0ced
3 changed files with 494 additions and 1 deletions

1
.gitignore vendored
View File

@ -78,6 +78,7 @@ microbench/benchmark_residual
/ogg/
oss-fuzz/fuzzer_decoder
oss-fuzz/fuzzer_seek
oss-fuzz/fuzzer_metadata
oss-fuzz/fuzzer_encoder
oss-fuzz/fuzzer_encoder_v2

View File

@ -31,7 +31,7 @@ EXTRA_DIST = \
noinst_PROGRAMS =
if USE_OSSFUZZERS
noinst_PROGRAMS += fuzzer_encoder fuzzer_encoder_v2 fuzzer_decoder fuzzer_seek
noinst_PROGRAMS += fuzzer_encoder fuzzer_encoder_v2 fuzzer_decoder fuzzer_seek fuzzer_metadata
endif
fuzzer_encoder_SOURCES = fuzzer_encoder.cc
@ -54,6 +54,11 @@ fuzzer_seek_CXXFLAGS = $(AM_CXXFLAGS) $(LIB_FUZZING_ENGINE)
fuzzer_seek_LDFLAGS = $(AM_LDFLAGS)
fuzzer_seek_LDADD = $(flac_libs)
fuzzer_metadata_SOURCES = fuzzer_metadata.cc
fuzzer_metadata_CXXFLAGS = $(AM_CXXFLAGS) $(LIB_FUZZING_ENGINE)
fuzzer_metadata_LDFLAGS = $(AM_LDFLAGS)
fuzzer_metadata_LDADD = $(flac_libs)
flac_libs = \
$(top_builddir)/src/libFLAC/libFLAC-static.la \

487
oss-fuzz/fuzzer_metadata.cc Normal file
View File

@ -0,0 +1,487 @@
/* fuzzer_metadata
* Copyright (C) 2022 Xiph.Org Foundation
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* - Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* - Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* - Neither the name of the Xiph.org Foundation nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include <cstdlib>
#include <cstdio>
#include <cstring> /* for memcpy */
#include "FLAC++/metadata.h"
#define CONFIG_LENGTH 1
#define min(x,y) (x<y?x:y)
static void run_tests_with_level_0_interface();
static void run_tests_with_level_1_interface(bool readonly, bool preservestats, const uint8_t *data, size_t size);
static void run_tests_with_level_2_interface(bool ogg, bool use_padding, const uint8_t *data, size_t size);
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
{
uint8_t command_length;
FLAC__bool init_bools[4];
/* Use first byte for configuration, leave at least one byte of input */
if(size < 1 + CONFIG_LENGTH){
return 0;
}
/* First 4 bits for configuration bools, next 4 for length of command section */
for(int i = 0; i < 4; i++)
init_bools[i] = data[i/8] & (1 << (i % 8));
command_length = data[0] >> 4;
/* Leave at least one byte as input */
if(command_length >= size - 1 - CONFIG_LENGTH)
command_length = size - 1 - CONFIG_LENGTH;
/* Dump input to file */
{
FILE * file_to_fuzz = fopen("/tmp/tmp.flac","w");
fwrite(data+CONFIG_LENGTH+command_length,1,size-CONFIG_LENGTH-command_length,file_to_fuzz);
fclose(file_to_fuzz);
}
run_tests_with_level_0_interface();
run_tests_with_level_1_interface(init_bools[1], init_bools[2], data+CONFIG_LENGTH, command_length/2);
/* Dump input to file, to start fresh for level 2 */
if(!init_bools[1]){
FILE * file_to_fuzz = fopen("/tmp/tmp.flac","w");
fwrite(data+CONFIG_LENGTH+command_length,1,size-CONFIG_LENGTH-command_length,file_to_fuzz);
fclose(file_to_fuzz);
}
run_tests_with_level_2_interface(init_bools[0], init_bools[3], data+command_length/2+CONFIG_LENGTH, command_length/2);
return 0;
}
static void run_tests_with_level_0_interface() {
FLAC::Metadata::StreamInfo streaminfo;
FLAC::Metadata::VorbisComment vorbis_comment;
FLAC::Metadata::CueSheet cue_sheet;
FLAC::Metadata::Picture picture;
FLAC::Metadata::get_streaminfo("/tmp/tmp.flac",streaminfo);
FLAC::Metadata::get_tags("/tmp/tmp.flac",vorbis_comment);
FLAC::Metadata::get_cuesheet("/tmp/tmp.flac",cue_sheet);
FLAC::Metadata::get_picture("/tmp/tmp.flac",picture, (FLAC__StreamMetadata_Picture_Type)(1), NULL, NULL, -1, -1, -1, -1);
}
static void run_tests_with_level_1_interface(bool readonly, bool preservestats, const uint8_t *data, size_t size) {
FLAC::Metadata::SimpleIterator iterator;
FLAC::Metadata::Prototype *metadata_block = nullptr;
uint8_t id[4] = {0};
if(!iterator.is_valid())
return;
if(!iterator.init("/tmp/tmp.flac",readonly,preservestats))
return;
for(size_t i = 0; i < size; i++) {
switch(data[i] & 7) {
case 0:
iterator.get_block_type();
iterator.get_block_offset();
iterator.get_block_length();
iterator.get_application_id(id);
break;
case 1:
iterator.next();
break;
case 2:
iterator.prev();
break;
case 3:
iterator.delete_block(data[i] & 8);
break;
case 4:
if(metadata_block != 0) {
delete metadata_block;
metadata_block = nullptr;
}
metadata_block = iterator.get_block();
break;
case 5:
if(metadata_block != 0)
iterator.set_block(metadata_block,data[i] & 8);
break;
case 6:
if(metadata_block != 0)
iterator.insert_block_after(metadata_block, data[i] & 8);
break;
case 7:
iterator.status();
iterator.is_last();
iterator.is_writable();
break;
}
}
if(metadata_block != 0) {
delete metadata_block;
metadata_block = nullptr;
}
}
static void run_tests_with_level_2_interface(bool ogg, bool use_padding, const uint8_t *data, size_t size) {
FLAC::Metadata::Chain chain;
FLAC::Metadata::Iterator iterator;
FLAC::Metadata::Prototype *metadata_block_get = nullptr;
FLAC::Metadata::Prototype *metadata_block_transfer = nullptr;
FLAC::Metadata::Prototype *metadata_block_put = nullptr;
if(!chain.is_valid())
return;
if(!chain.read("/tmp/tmp.flac", ogg))
return;
iterator.init(chain);
for(size_t i = 0; i < size; i++) {
switch(data[i] & 15) {
case 0:
iterator.get_block_type();
break;
case 1:
iterator.next();
break;
case 2:
iterator.prev();
break;
case 3:
iterator.delete_block(data[i] & 16);
break;
case 4:
metadata_block_get = iterator.get_block();
if(metadata_block_get != 0 && metadata_block_get->is_valid()) {
if(metadata_block_transfer != 0 && metadata_block_transfer->is_valid()) {
if(metadata_block_transfer != metadata_block_get) {
delete metadata_block_transfer;
metadata_block_transfer = nullptr;
metadata_block_transfer = FLAC::Metadata::clone(metadata_block_get);
}
}
else {
metadata_block_transfer = FLAC::Metadata::clone(metadata_block_get);
}
}
delete metadata_block_get;
break;
case 5:
if(metadata_block_transfer != 0 && metadata_block_transfer->is_valid()) {
metadata_block_put = FLAC::Metadata::clone(metadata_block_transfer);
if(!iterator.insert_block_before(metadata_block_put))
delete metadata_block_put;
}
break;
case 6:
if(metadata_block_transfer != 0 && metadata_block_transfer->is_valid()) {
metadata_block_put = FLAC::Metadata::clone(metadata_block_transfer);
if(!iterator.insert_block_after(metadata_block_put))
delete metadata_block_put;
}
break;
case 7:
if(metadata_block_transfer != 0 && metadata_block_transfer->is_valid()) {
metadata_block_put = FLAC::Metadata::clone(metadata_block_transfer);
if(!iterator.set_block(metadata_block_put))
delete metadata_block_put;
}
break;
case 8: /* Examine block */
if(metadata_block_transfer != 0 && metadata_block_transfer->is_valid()) {
switch(iterator.get_block_type()) {
case FLAC__METADATA_TYPE_VORBIS_COMMENT:
{
uint32_t num_comments;
FLAC::Metadata::VorbisComment::Entry entry;
FLAC::Metadata::VorbisComment * vorbiscomment = dynamic_cast<FLAC::Metadata::VorbisComment *>(metadata_block_transfer);
if(vorbiscomment == 0)
break;
vorbiscomment->get_vendor_string();
num_comments = vorbiscomment->get_num_comments();
if(num_comments > 0) {
entry = vorbiscomment->get_comment(min(data[i]>>4,num_comments-1));
vorbiscomment->find_entry_from(0,"TEST");
}
}
break;
case FLAC__METADATA_TYPE_CUESHEET:
{
uint32_t num_tracks, num_indices;
FLAC::Metadata::CueSheet * cuesheet = dynamic_cast<FLAC::Metadata::CueSheet *>(metadata_block_transfer);
if(cuesheet == 0 || !cuesheet->is_legal())
break;
cuesheet->is_legal(true); /* check CDDA subset */
cuesheet->calculate_cddb_id();
cuesheet->get_media_catalog_number();
cuesheet->get_lead_in();
cuesheet->get_is_cd();
num_tracks = cuesheet->get_num_tracks();
if(num_tracks > 0) {
FLAC::Metadata::CueSheet::Track track = cuesheet->get_track(min(data[i]>>4,num_tracks-1));
track.get_offset();
track.get_number();
track.get_isrc();
track.get_pre_emphasis();
num_indices = track.get_num_indices();
if(num_indices > 0) {
FLAC__StreamMetadata_CueSheet_Index index = track.get_index(min(data[i]>>4,num_indices-1));
(void)index;
}
}
}
break;
case FLAC__METADATA_TYPE_PICTURE:
{
char * violation = nullptr;
FLAC::Metadata::Picture * picture = dynamic_cast<FLAC::Metadata::Picture *>(metadata_block_transfer);
if(picture == 0 || !picture->is_legal((const char **)&violation))
break;
picture->get_data();
}
break;
default:
break;
}
}
break;
case 9: /* Replace or add in block */
if(metadata_block_transfer != 0 && metadata_block_transfer->is_valid()) {
switch(iterator.get_block_type()) {
case FLAC__METADATA_TYPE_SEEKTABLE:
{
uint32_t num_seekpoints;
FLAC__StreamMetadata_SeekPoint seekpoint;
FLAC::Metadata::SeekTable * seektable = dynamic_cast<FLAC::Metadata::SeekTable *>(metadata_block_transfer);
if(seektable == 0)
break;
if(seektable->is_valid() && seektable->is_legal()) {
num_seekpoints = seektable->get_num_points();
if(num_seekpoints > 0) {
seekpoint = seektable->get_point(min(data[i]>>5,num_seekpoints-1));
seektable->set_point(0,seekpoint);
seektable->insert_point(min(data[i]>>5,num_seekpoints-1),seekpoint);
}
seektable->template_append_placeholders(4);
seektable->template_append_point(111111);
seektable->template_append_points((FLAC__uint64[]){222222, 333333, 444444}, 3);
seektable->template_append_spaced_points(data[i]>>5, 1234567);
seektable->template_append_spaced_points_by_samples(data[i]>>5, 2468000);
seektable->template_sort(data[i] & 16);
}
}
case FLAC__METADATA_TYPE_VORBIS_COMMENT:
{
uint32_t num_comments;
FLAC::Metadata::VorbisComment::Entry entry;
FLAC::Metadata::VorbisComment * vorbiscomment = dynamic_cast<FLAC::Metadata::VorbisComment *>(metadata_block_transfer);
if(vorbiscomment == 0)
break;
num_comments = vorbiscomment->get_num_comments();
if(num_comments > 0) {
entry = vorbiscomment->get_comment(min(data[i]>>5,num_comments-1));
if(entry.is_valid()) {
vorbiscomment->replace_comment(entry,data[i] & 16);
vorbiscomment->set_comment(0,entry);
vorbiscomment->append_comment(entry);
vorbiscomment->insert_comment(0,entry);
}
}
}
break;
case FLAC__METADATA_TYPE_CUESHEET:
{
uint32_t num_tracks, num_indices;
FLAC::Metadata::CueSheet * cuesheet = dynamic_cast<FLAC::Metadata::CueSheet *>(metadata_block_transfer);
if(cuesheet == 0 || !cuesheet->is_legal())
break;
num_tracks = cuesheet->get_num_tracks();
if(num_tracks > 0) {
FLAC::Metadata::CueSheet::Track track = cuesheet->get_track(min(data[i]>>4,num_tracks-1));
num_indices = track.get_num_indices();
if(num_indices > 0) {
FLAC__StreamMetadata_CueSheet_Index index = track.get_index(min(data[i]>>4,num_indices-1));
track.set_index(0,index);
cuesheet->insert_index(0,0,index);
cuesheet->insert_blank_index(0,0);
}
cuesheet->insert_blank_track(0);
cuesheet->insert_track(0,track);
cuesheet->resize_indices(min(data[i]>>4,num_tracks-1),data[i]>>4);
}
}
break;
case FLAC__METADATA_TYPE_PICTURE:
{
FLAC::Metadata::Picture * picture = dynamic_cast<FLAC::Metadata::Picture *>(metadata_block_transfer);
const char testtext[] = "TEST";
if(picture == 0 || !picture->is_legal(NULL))
break;
picture->set_description((FLAC__byte *)&testtext);
picture->set_mime_type((const char *)&testtext);
picture->set_data((FLAC__byte *)&testtext,4);
}
break;
default:
break;
}
}
break;
case 10: /* Delete from block */
if(metadata_block_transfer != 0 && metadata_block_transfer->is_valid()) {
switch(iterator.get_block_type()) {
case FLAC__METADATA_TYPE_SEEKTABLE:
{
uint32_t num_seekpoints;
FLAC::Metadata::SeekTable * seektable = dynamic_cast<FLAC::Metadata::SeekTable *>(metadata_block_transfer);
if(seektable == 0)
break;
if(seektable->is_valid() && seektable->is_legal()) {
num_seekpoints = seektable->get_num_points();
if(num_seekpoints > 0)
seektable->delete_point(min(data[i]>>4,num_seekpoints-1));
}
}
case FLAC__METADATA_TYPE_VORBIS_COMMENT:
{
uint32_t num_comments;
FLAC::Metadata::VorbisComment * vorbiscomment = dynamic_cast<FLAC::Metadata::VorbisComment *>(metadata_block_transfer);
if(vorbiscomment == 0)
break;
num_comments = vorbiscomment->get_num_comments();
if(num_comments > 0)
vorbiscomment->delete_comment(min(data[i]>>4,num_comments-1));
vorbiscomment->remove_entry_matching("TEST");
vorbiscomment->remove_entries_matching("TEST");
}
break;
case FLAC__METADATA_TYPE_CUESHEET:
{
uint32_t num_tracks;
FLAC::Metadata::CueSheet * cuesheet = dynamic_cast<FLAC::Metadata::CueSheet *>(metadata_block_transfer);
if(cuesheet == 0 || !cuesheet->is_legal())
break;
num_tracks = cuesheet->get_num_tracks();
if(num_tracks > 0) {
FLAC::Metadata::CueSheet::Track track = cuesheet->get_track(min(data[i]>>4,num_tracks-1));
if(track.get_num_indices() > 0)
cuesheet->delete_index(min(data[i]>>4,num_tracks-1),0);
cuesheet->delete_track(0);
}
}
break;
default:
break;
}
}
break;
case 11: /* Resize block */
if(metadata_block_transfer != 0 && metadata_block_transfer->is_valid()) {
switch(iterator.get_block_type()) {
case FLAC__METADATA_TYPE_PADDING:
{
FLAC::Metadata::Padding * padding = dynamic_cast<FLAC::Metadata::Padding *>(metadata_block_transfer);
if(padding == 0)
break;
padding->set_length(data[i]>>4);
}
break;
case FLAC__METADATA_TYPE_SEEKTABLE:
{
FLAC::Metadata::SeekTable * seektable = dynamic_cast<FLAC::Metadata::SeekTable *>(metadata_block_transfer);
if(seektable == 0)
break;
seektable->resize_points(data[i]>>4);
}
break;
case FLAC__METADATA_TYPE_VORBIS_COMMENT:
{
FLAC::Metadata::VorbisComment * vorbiscomment = dynamic_cast<FLAC::Metadata::VorbisComment *>(metadata_block_transfer);
if(vorbiscomment == 0)
break;
vorbiscomment->resize_comments(data[i]>>4);
}
break;
case FLAC__METADATA_TYPE_CUESHEET:
{
uint32_t num_tracks;
FLAC::Metadata::CueSheet * cuesheet = dynamic_cast<FLAC::Metadata::CueSheet *>(metadata_block_transfer);
if(cuesheet == 0 || !cuesheet->is_legal())
break;
num_tracks = cuesheet->get_num_tracks();
if(num_tracks > 0) {
cuesheet->resize_indices(min(data[i]>>4,num_tracks-1),data[i]>>4);
}
cuesheet->resize_tracks(data[i]<<4);
}
break;
default:
break;
}
}
break;
case 12: /* Prototype functions */
if(metadata_block_transfer != 0 && metadata_block_transfer->is_valid()) {
const ::FLAC__StreamMetadata * metadata_compare = *metadata_block_transfer;
metadata_block_transfer->get_is_last();
metadata_block_transfer->get_length();
metadata_block_transfer->set_is_last(data[i] & 16);
FLAC__metadata_object_is_equal(metadata_compare, metadata_compare);
}
break;
}
}
if(metadata_block_transfer != 0) {
delete metadata_block_transfer;
metadata_block_transfer = nullptr;
}
chain.status();
chain.sort_padding();
chain.merge_padding();
chain.check_if_tempfile_needed(!use_padding);
chain.write(use_padding);
}