FuzzEncoders.cpp File Reference

#include "fuzz/Fuzz.h"
#include "include/core/SkBitmap.h"
#include "include/core/SkImage.h"
#include "include/core/SkImageInfo.h"
#include "include/core/SkPixmap.h"
#include "include/core/SkStream.h"
#include "include/encode/SkJpegEncoder.h"
#include "include/encode/SkPngEncoder.h"
#include "include/encode/SkWebpEncoder.h"
#include "src/base/SkRandom.h"
#include "src/core/SkOSFile.h"
#include <vector>

Go to the source code of this file.

Functions
static SkBitmap	make_fuzzed_bitmap (Fuzz *fuzz)
	DEF_FUZZ (PNGEncoder, fuzz)
	DEF_FUZZ (JPEGEncoder, fuzz)
	DEF_FUZZ (WEBPEncoder, fuzz)
	DEF_FUZZ (_MakeEncoderCorpus, fuzz)

Variables
constexpr int	MAX_WIDTH = 512
constexpr int	MAX_HEIGHT = 512

Function Documentation

DEF_FUZZ() [1/4]

DEF_FUZZ	(	_MakeEncoderCorpus	,
		fuzz
	)

Definition at line 79 of file FuzzEncoders.cpp.

                                   {
    sk_sp<SkData> bytes = SkData::MakeWithoutCopy(fuzz->fData, fuzz->fSize);
    SkDebugf("bytes %zu\n", bytes->size());
    auto img = SkImages::DeferredFromEncodedData(bytes);
    if (nullptr == img.get()) {
        SkDebugf("invalid image, could not decode\n");
        return;
    }
    if (img->width() > MAX_WIDTH || img->height() > MAX_HEIGHT) {
        SkDebugf("Too big (%d x %d)\n", img->width(), img->height());
        return;
    }
    std::vector<int32_t> dstPixels;
    int rowBytes = img->width() * 4;
    dstPixels.resize(img->height() * rowBytes);
    SkPixmap pm(SkImageInfo::MakeN32Premul(img->width(), img->height()),
        &dstPixels.front(), rowBytes);
    if (!img->readPixels(nullptr, pm, 0, 0)) {
        SkDebugf("Could not read pixmap\n");
        return;
    }
 
    SkString s("./encoded_corpus/enc_");
    static SkRandom rand;
    s.appendU32(rand.nextU());
    auto file = sk_fopen(s.c_str(), SkFILE_Flags::kWrite_SkFILE_Flag);
    if (!file) {
        SkDebugf("Can't initialize file\n");
        return;
    }
    auto total = pm.info().bytesPerPixel() * pm.width() * pm.height();
    SkDebugf("Writing %d (%d x %d) bytes\n", total, pm.width(), pm.height());
    // Write out the size in two bytes since that's what the fuzzer will
    // read first.
    uint32_t w = pm.width();
    sk_fwrite(&w, sizeof(uint32_t), file);
    uint32_t h = pm.height();
    sk_fwrite(&h, sizeof(uint32_t), file);
    sk_fwrite(pm.addr(), total, file);
    sk_fclose(file);
}

DEF_FUZZ() [2/4]

DEF_FUZZ	(	JPEGEncoder	,
		fuzz
	)

Definition at line 50 of file FuzzEncoders.cpp.

                            {
    auto bm = make_fuzzed_bitmap(fuzz);
 
    auto opts = SkJpegEncoder::Options{};
    fuzz->nextRange(&opts.fQuality, 0, 100);
 
    SkDynamicMemoryWStream dest;
    (void)SkJpegEncoder::Encode(&dest, bm.pixmap(), opts);
}

DEF_FUZZ() [3/4]

DEF_FUZZ	(	PNGEncoder	,
		fuzz
	)

Definition at line 40 of file FuzzEncoders.cpp.

                           {
    auto bm = make_fuzzed_bitmap(fuzz);
 
    auto opts = SkPngEncoder::Options{};
    fuzz->nextRange(&opts.fZLibLevel, 0, 9);
 
    SkDynamicMemoryWStream dest;
    SkPngEncoder::Encode(&dest, bm.pixmap(), opts);
}

DEF_FUZZ() [4/4]

DEF_FUZZ	(	WEBPEncoder	,
		fuzz
	)

Definition at line 60 of file FuzzEncoders.cpp.

                            {
    auto bm = make_fuzzed_bitmap(fuzz);
 
    auto opts = SkWebpEncoder::Options{};
    fuzz->nextRange(&opts.fQuality, 0.0f, 100.0f);
    bool lossy;
    fuzz->next(&lossy);
    if (lossy) {
        opts.fCompression = SkWebpEncoder::Compression::kLossy;
    } else {
        opts.fCompression = SkWebpEncoder::Compression::kLossless;
    }
 
    SkDynamicMemoryWStream dest;
    (void)SkWebpEncoder::Encode(&dest, bm.pixmap(), opts);
}

make_fuzzed_bitmap()

static SkBitmap make_fuzzed_bitmap ( Fuzz *  fuzz )

static

Definition at line 27 of file FuzzEncoders.cpp.

                                               {
    SkBitmap bm;
    uint32_t w, h;
    fuzz->nextRange(&w, 1, MAX_WIDTH);
    fuzz->nextRange(&h, 1, MAX_HEIGHT);
    if (!bm.tryAllocPixels(SkImageInfo::MakeN32Premul(w, h))) {
        return bm;
    }
    uint32_t n = w * h;
    fuzz->nextN((SkPMColor*)bm.getPixels(), n);
    return bm;
}

Variable Documentation

MAX_HEIGHT

constexpr int MAX_HEIGHT = 512

constexpr

Definition at line 25 of file FuzzEncoders.cpp.

MAX_WIDTH

constexpr int MAX_WIDTH = 512

constexpr

Definition at line 24 of file FuzzEncoders.cpp.