f1eb43e880
Change-Id: I60d60e9b9ea0b7b6544a36bef7f4e263bb9de532 Reviewed-on: https://skia-review.googlesource.com/c/skia/+/296416 Reviewed-by: Kevin Lubick <kjlubick@google.com> Commit-Queue: Zepeng Hu <zepenghu@google.com>
67 lines
1.8 KiB
C++
67 lines
1.8 KiB
C++
/*
|
|
* Copyright 2018 Google, LLC
|
|
*
|
|
* Use of this source code is governed by a BSD-style license that can be
|
|
* found in the LICENSE file.
|
|
*/
|
|
|
|
#include "include/codec/SkAndroidCodec.h"
|
|
#include "include/core/SkBitmap.h"
|
|
#include "include/core/SkCanvas.h"
|
|
#include "include/core/SkData.h"
|
|
#include "include/core/SkSurface.h"
|
|
|
|
#include "fuzz/Fuzz.h"
|
|
|
|
bool FuzzAndroidCodec(sk_sp<SkData> bytes, uint8_t sampleSize) {
|
|
auto codec = SkAndroidCodec::MakeFromData(bytes);
|
|
if (!codec) {
|
|
return false;
|
|
}
|
|
|
|
auto size = codec->getSampledDimensions(sampleSize);
|
|
auto info = SkImageInfo::MakeN32Premul(size);
|
|
SkBitmap bm;
|
|
if (!bm.tryAllocPixels(info)) {
|
|
// May fail in memory-constrained fuzzing environments
|
|
return false;
|
|
}
|
|
|
|
SkAndroidCodec::AndroidOptions options;
|
|
options.fSampleSize = sampleSize;
|
|
|
|
auto result = codec->getAndroidPixels(bm.info(), bm.getPixels(), bm.rowBytes(), &options);
|
|
switch (result) {
|
|
case SkCodec::kSuccess:
|
|
case SkCodec::kIncompleteInput:
|
|
case SkCodec::kErrorInInput:
|
|
break;
|
|
default:
|
|
return false;
|
|
}
|
|
|
|
auto surface = SkSurface::MakeRasterN32Premul(size.width(), size.height());
|
|
if (!surface) {
|
|
// May return nullptr in memory-constrained fuzzing environments
|
|
return false;
|
|
}
|
|
|
|
surface->getCanvas()->drawBitmap(bm, 0, 0);
|
|
return true;
|
|
}
|
|
|
|
#if defined(IS_FUZZING_WITH_LIBFUZZER)
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
|
|
if (size > 10240) {
|
|
return 0;
|
|
}
|
|
auto bytes = SkData::MakeWithoutCopy(data, size);
|
|
Fuzz fuzz(bytes);
|
|
uint8_t sampleSize;
|
|
fuzz.nextRange(&sampleSize, 1, 64);
|
|
bytes = SkData::MakeSubset(bytes.get(), 1, size - 1);
|
|
FuzzAndroidCodec(bytes, sampleSize);
|
|
return 0;
|
|
}
|
|
#endif
|