64 lines
1.7 KiB
C++
64 lines
1.7 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 "SkAndroidCodec.h"
|
||
|
|
#include "SkBitmap.h"
|
||
|
|
#include "SkCanvas.h"
|
||
|
|
#include "SkData.h"
|
||
|
|
#include "SkSurface.h"
|
||
|
|
|
||
|
|
#include "../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) {
|
||
|
|
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
|