blob: aecf19df9ccbb86d2735028408d8ac6ef522bc12 [file] [log] [blame]
mtklein65e58242016-01-13 12:57:57 -08001/*
2 * Copyright 2016 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
Mike Kleinc0bd9f92019-04-23 12:05:21 -05008#include "fuzz/Fuzz.h"
9#include "include/codec/SkCodec.h"
10#include "include/core/SkCanvas.h"
11#include "include/core/SkData.h"
12#include "include/core/SkImage.h"
13#include "include/core/SkImageEncoder.h"
14#include "include/core/SkMallocPixelRef.h"
15#include "include/core/SkPaint.h"
16#include "include/core/SkPath.h"
17#include "include/core/SkStream.h"
18#include "include/core/SkSurface.h"
19#include "include/core/SkTextBlob.h"
Kevin Lubickd7255a72019-06-03 11:59:23 -040020#include "src/core/SkFontMgrPriv.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050021#include "src/core/SkOSFile.h"
22#include "src/core/SkPicturePriv.h"
23#include "src/core/SkReadBuffer.h"
24#include "src/utils/SkOSPath.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050025#include "tools/ToolUtils.h"
Kevin Lubickd7255a72019-06-03 11:59:23 -040026#include "tools/flags/CommandLineFlags.h"
27#include "tools/fonts/TestFontMgr.h"
Hal Canary8a001442018-09-19 11:31:27 -040028
Hal Canary62176db2017-02-27 16:42:03 -050029#include <iostream>
Kevin Lubickacd456a2018-04-24 13:58:16 -040030#include <map>
Kevin Lubickfffa6412018-04-23 16:44:55 -040031#include <regex>
mtkleina1159422016-01-15 05:46:54 -080032#include <signal.h>
Kevin Lubick2541edf2018-01-11 10:27:14 -050033
Mike Klein84836b72019-03-21 11:31:36 -050034static DEFINE_string2(bytes, b, "", "A path to a file or a directory. If a file, the "
35 "contents will be used as the fuzz bytes. If a directory, all files "
36 "in the directory will be used as fuzz bytes for the fuzzer, one at a "
37 "time.");
38static DEFINE_string2(name, n, "", "If --type is 'api', fuzz the API with this name.");
39static DEFINE_string2(dump, d, "", "If not empty, dump 'image*' or 'skp' types as a "
40 "PNG with this name.");
Kevin Lubick9ff5dc92018-01-09 12:47:33 -050041DEFINE_bool2(verbose, v, false, "Print more information while fuzzing.");
Florin Malita7796f002018-06-08 12:25:38 -040042
43// This cannot be inlined in DEFINE_string2 due to interleaved ifdefs
44static constexpr char g_type_message[] = "How to interpret --bytes, one of:\n"
Kevin Lubick0f3d2a62018-10-17 10:24:44 -040045 "android_codec\n"
Florin Malita7796f002018-06-08 12:25:38 -040046 "animated_image_decode\n"
47 "api\n"
48 "color_deserialize\n"
49 "filter_fuzz (equivalent to Chrome's filter_fuzz_stub)\n"
50 "image_decode\n"
Kevin Lubick0f3d2a62018-10-17 10:24:44 -040051 "image_decode_incremental\n"
Florin Malita7796f002018-06-08 12:25:38 -040052 "image_mode\n"
53 "image_scale\n"
Florin Malita80452be2018-06-19 11:27:20 -040054 "json\n"
Florin Malita7796f002018-06-08 12:25:38 -040055 "path_deserialize\n"
Florin Malita7796f002018-06-08 12:25:38 -040056 "region_deserialize\n"
57 "region_set_path\n"
Florin Malita7796f002018-06-08 12:25:38 -040058 "skp\n"
59 "sksl2glsl\n"
Kevin Lubicke9c1ce82019-03-11 11:09:40 -040060 "sksl2metal\n"
Kevin Lubick0f0a7102019-03-18 16:20:55 -040061 "sksl2pipeline\n"
Kevin Lubicke9c1ce82019-03-11 11:09:40 -040062 "sksl2spirv\n"
Florin Malita7796f002018-06-08 12:25:38 -040063#if defined(SK_ENABLE_SKOTTIE)
64 "skottie_json\n"
65#endif
66 "textblob";
67
Mike Klein84836b72019-03-21 11:31:36 -050068static DEFINE_string2(type, t, "", g_type_message);
kjlubickdba57342016-01-21 05:03:28 -080069
Kevin Lubickfffa6412018-04-23 16:44:55 -040070static int fuzz_file(SkString path, SkString type);
kjlubick2a42f482016-02-16 16:14:23 -080071static uint8_t calculate_option(SkData*);
Kevin Lubickfffa6412018-04-23 16:44:55 -040072static SkString try_auto_detect(SkString path, SkString* name);
kjlubickdba57342016-01-21 05:03:28 -080073
Kevin Lubick0f3d2a62018-10-17 10:24:44 -040074static void fuzz_android_codec(sk_sp<SkData>);
75static void fuzz_animated_img(sk_sp<SkData>);
Kevin Lubickfffa6412018-04-23 16:44:55 -040076static void fuzz_api(sk_sp<SkData> bytes, SkString name);
Kevin Lubickf80f1152017-01-06 08:26:56 -050077static void fuzz_color_deserialize(sk_sp<SkData>);
Kevin Lubickd46e85f2017-11-21 17:13:35 -050078static void fuzz_filter_fuzz(sk_sp<SkData>);
Kevin Lubick0f3d2a62018-10-17 10:24:44 -040079static void fuzz_image_decode(sk_sp<SkData>);
80static void fuzz_image_decode_incremental(sk_sp<SkData>);
Kevin Lubick0168e042017-02-14 13:12:37 -050081static void fuzz_img(sk_sp<SkData>, uint8_t, uint8_t);
Florin Malita80452be2018-06-19 11:27:20 -040082static void fuzz_json(sk_sp<SkData>);
Kevin Lubickf04c50a2017-01-06 13:48:19 -050083static void fuzz_path_deserialize(sk_sp<SkData>);
Kevin Lubickedee1ae2017-02-20 17:47:18 -050084static void fuzz_region_deserialize(sk_sp<SkData>);
Kevin Lubick2541edf2018-01-11 10:27:14 -050085static void fuzz_region_set_path(sk_sp<SkData>);
Kevin Lubick0168e042017-02-14 13:12:37 -050086static void fuzz_skp(sk_sp<SkData>);
Kevin Lubicke9c1ce82019-03-11 11:09:40 -040087static void fuzz_sksl2glsl(sk_sp<SkData>);
88static void fuzz_sksl2metal(sk_sp<SkData>);
89static void fuzz_sksl2spirv(sk_sp<SkData>);
Kevin Lubick0f0a7102019-03-18 16:20:55 -040090static void fuzz_sksl2pipeline(sk_sp<SkData>);
Kevin Lubickd46e85f2017-11-21 17:13:35 -050091static void fuzz_textblob_deserialize(sk_sp<SkData>);
Herb Derbya839fc02017-03-16 12:30:43 -040092
Kevin Lubickfffa6412018-04-23 16:44:55 -040093static void print_api_names();
94
Florin Malita0b0d93d2018-05-04 15:01:03 -040095#if defined(SK_ENABLE_SKOTTIE)
96static void fuzz_skottie_json(sk_sp<SkData>);
97#endif
98
mtklein65e58242016-01-13 12:57:57 -080099int main(int argc, char** argv) {
Mike Klein88544fb2019-03-20 10:50:33 -0500100 CommandLineFlags::SetUsage(
101 "Usage: fuzz -t <type> -b <path/to/file> [-n api-to-fuzz]\n"
102 " fuzz -b <path/to/file>\n"
103 "--help lists the valid types. If type is not specified,\n"
104 "fuzz will make a guess based on the name of the file.\n");
105 CommandLineFlags::Parse(argc, argv);
Kevin Lubickd7255a72019-06-03 11:59:23 -0400106 gSkFontMgr_DefaultFactory = &ToolUtils::MakePortableFontMgr;
mtkleinf5e97822016-01-15 06:19:53 -0800107
Kevin Lubickfffa6412018-04-23 16:44:55 -0400108 SkString path = SkString(FLAGS_bytes.isEmpty() ? argv[0] : FLAGS_bytes[0]);
109 SkString type = SkString(FLAGS_type.isEmpty() ? "" : FLAGS_type[0]);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500110
Kevin Lubickfffa6412018-04-23 16:44:55 -0400111 if (!sk_isdir(path.c_str())) {
112 return fuzz_file(path, type);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500113 }
114
Kevin Lubickfffa6412018-04-23 16:44:55 -0400115 SkOSFile::Iter it(path.c_str());
Kevin Lubickf80f1152017-01-06 08:26:56 -0500116 for (SkString file; it.next(&file); ) {
Kevin Lubickfffa6412018-04-23 16:44:55 -0400117 SkString p = SkOSPath::Join(path.c_str(), file.c_str());
Kevin Lubickf80f1152017-01-06 08:26:56 -0500118 SkDebugf("Fuzzing %s\n", p.c_str());
Kevin Lubickfffa6412018-04-23 16:44:55 -0400119 int rv = fuzz_file(p, type);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500120 if (rv != 0) {
121 return rv;
122 }
123 }
124 return 0;
125}
126
Kevin Lubickfffa6412018-04-23 16:44:55 -0400127static int fuzz_file(SkString path, SkString type) {
128 sk_sp<SkData> bytes(SkData::MakeFromFileName(path.c_str()));
kjlubickdba57342016-01-21 05:03:28 -0800129 if (!bytes) {
Kevin Lubickfffa6412018-04-23 16:44:55 -0400130 SkDebugf("Could not read %s\n", path.c_str());
Kevin Lubickf80f1152017-01-06 08:26:56 -0500131 return 1;
kjlubickdba57342016-01-21 05:03:28 -0800132 }
mtklein65e58242016-01-13 12:57:57 -0800133
Kevin Lubickfffa6412018-04-23 16:44:55 -0400134 SkString name = SkString(FLAGS_name.isEmpty() ? "" : FLAGS_name[0]);
135
136 if (type.isEmpty()) {
137 type = try_auto_detect(path, &name);
kjlubickdba57342016-01-21 05:03:28 -0800138 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400139
140 if (type.isEmpty()) {
141 SkDebugf("Could not autodetect type of %s\n", path.c_str());
142 return 1;
143 }
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400144 if (type.equals("android_codec")) {
145 fuzz_android_codec(bytes);
146 return 0;
147 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400148 if (type.equals("animated_image_decode")) {
149 fuzz_animated_img(bytes);
150 return 0;
151 }
152 if (type.equals("api")) {
153 fuzz_api(bytes, name);
154 return 0;
155 }
156 if (type.equals("color_deserialize")) {
157 fuzz_color_deserialize(bytes);
158 return 0;
159 }
Florin Malita80452be2018-06-19 11:27:20 -0400160 if (type.equals("filter_fuzz")) {
161 fuzz_filter_fuzz(bytes);
162 return 0;
163 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400164 if (type.equals("image_decode")) {
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400165 fuzz_image_decode(bytes);
166 return 0;
167 }
168 if (type.equals("image_decode_incremental")) {
169 fuzz_image_decode_incremental(bytes);
Kevin Lubickfffa6412018-04-23 16:44:55 -0400170 return 0;
171 }
172 if (type.equals("image_scale")) {
173 uint8_t option = calculate_option(bytes.get());
174 fuzz_img(bytes, option, 0);
175 return 0;
176 }
177 if (type.equals("image_mode")) {
178 uint8_t option = calculate_option(bytes.get());
179 fuzz_img(bytes, 0, option);
180 return 0;
181 }
Florin Malita80452be2018-06-19 11:27:20 -0400182 if (type.equals("json")) {
183 fuzz_json(bytes);
Kevin Lubick9eeede22018-05-03 16:26:10 -0400184 return 0;
185 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400186 if (type.equals("path_deserialize")) {
187 fuzz_path_deserialize(bytes);
188 return 0;
189 }
190 if (type.equals("region_deserialize")) {
191 fuzz_region_deserialize(bytes);
192 return 0;
193 }
194 if (type.equals("region_set_path")) {
195 fuzz_region_set_path(bytes);
196 return 0;
197 }
198 if (type.equals("pipe")) {
Mike Klein60900b52018-09-21 11:19:45 -0400199 SkDebugf("I would prefer not to.\n");
Kevin Lubickfffa6412018-04-23 16:44:55 -0400200 return 0;
201 }
Florin Malita0b0d93d2018-05-04 15:01:03 -0400202#if defined(SK_ENABLE_SKOTTIE)
Kevin Lubick9eeede22018-05-03 16:26:10 -0400203 if (type.equals("skottie_json")) {
204 fuzz_skottie_json(bytes);
Kevin Lubickfffa6412018-04-23 16:44:55 -0400205 return 0;
206 }
Florin Malita0b0d93d2018-05-04 15:01:03 -0400207#endif
Kevin Lubick9eeede22018-05-03 16:26:10 -0400208 if (type.equals("skp")) {
209 fuzz_skp(bytes);
Kevin Lubickfffa6412018-04-23 16:44:55 -0400210 return 0;
211 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400212 if (type.equals("sksl2glsl")) {
213 fuzz_sksl2glsl(bytes);
214 return 0;
215 }
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400216 if (type.equals("sksl2metal")) {
217 fuzz_sksl2metal(bytes);
218 return 0;
219 }
220 if (type.equals("sksl2spirv")) {
221 fuzz_sksl2spirv(bytes);
222 return 0;
223 }
Kevin Lubick0f0a7102019-03-18 16:20:55 -0400224 if (type.equals("sksl2pipeline")) {
225 fuzz_sksl2pipeline(bytes);
226 return 0;
227 }
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400228 if (type.equals("textblob")) {
229 fuzz_textblob_deserialize(bytes);
230 return 0;
231 }
Kevin Lubick457fa972018-05-29 09:22:06 -0400232 SkDebugf("Unknown type %s\n", type.c_str());
Mike Klein88544fb2019-03-20 10:50:33 -0500233 CommandLineFlags::PrintUsage();
Kevin Lubick9ff5dc92018-01-09 12:47:33 -0500234 return 1;
kjlubickdba57342016-01-21 05:03:28 -0800235}
236
Kevin Lubickfffa6412018-04-23 16:44:55 -0400237static std::map<std::string, std::string> cf_api_map = {
238 {"api_draw_functions", "DrawFunctions"},
239 {"api_gradients", "Gradients"},
240 {"api_image_filter", "ImageFilter"},
241 {"api_mock_gpu_canvas", "MockGPUCanvas"},
242 {"api_null_canvas", "NullCanvas"},
243 {"api_path_measure", "PathMeasure"},
Kevin Lubickde54d7c2018-09-24 08:48:19 -0400244 {"api_pathop", "Pathop"},
Kevin Lubick57507f12018-10-11 09:35:15 -0400245 {"api_polyutils", "PolyUtils"},
Kevin Lubickfffa6412018-04-23 16:44:55 -0400246 {"api_raster_n32_canvas", "RasterN32Canvas"},
247 {"jpeg_encoder", "JPEGEncoder"},
248 {"png_encoder", "PNGEncoder"},
Kevin Lubickf84ded22018-10-23 09:28:48 -0400249 {"skia_pathop_fuzzer", "LegacyChromiumPathop"},
Kevin Lubickfffa6412018-04-23 16:44:55 -0400250 {"webp_encoder", "WEBPEncoder"}
251};
252
Kevin Lubick6c560552018-06-20 09:29:23 -0400253// maps clusterfuzz/oss-fuzz -> Skia's name
Kevin Lubickfffa6412018-04-23 16:44:55 -0400254static std::map<std::string, std::string> cf_map = {
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400255 {"android_codec", "android_codec"},
Kevin Lubickfffa6412018-04-23 16:44:55 -0400256 {"animated_image_decode", "animated_image_decode"},
257 {"image_decode", "image_decode"},
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400258 {"image_decode_incremental", "image_decode_incremental"},
Kevin Lubickfffa6412018-04-23 16:44:55 -0400259 {"image_filter_deserialize", "filter_fuzz"},
260 {"image_filter_deserialize_width", "filter_fuzz"},
261 {"path_deserialize", "path_deserialize"},
262 {"region_deserialize", "region_deserialize"},
263 {"region_set_path", "region_set_path"},
Kevin Lubick6c560552018-06-20 09:29:23 -0400264 {"skjson", "json"},
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400265 {"sksl2glsl", "sksl2glsl"},
266 {"sksl2metal", "sksl2metal"},
267 {"sksl2spirv", "sksl2spirv"},
Kevin Lubick0f0a7102019-03-18 16:20:55 -0400268 {"sksl2pipeline", "sksl2pipeline"},
Kevin Lubickde54d7c2018-09-24 08:48:19 -0400269#if defined(SK_ENABLE_SKOTTIE)
270 {"skottie_json", "skottie_json"},
271#endif
Kevin Lubickfffa6412018-04-23 16:44:55 -0400272 {"textblob_deserialize", "textblob"}
273};
274
275static SkString try_auto_detect(SkString path, SkString* name) {
276 std::cmatch m;
277 std::regex clusterfuzz("clusterfuzz-testcase(-minimized)?-([a-z0-9_]+)-[\\d]+");
278 std::regex skiafuzzer("(api-)?(\\w+)-[a-f0-9]+");
279
280 if (std::regex_search(path.c_str(), m, clusterfuzz)) {
281 std::string type = m.str(2);
Kevin Lubick6c560552018-06-20 09:29:23 -0400282
283 if (cf_api_map.find(type) != cf_api_map.end()) {
284 *name = SkString(cf_api_map[type].c_str());
285 return SkString("api");
Kevin Lubickfffa6412018-04-23 16:44:55 -0400286 } else {
287 if (cf_map.find(type) != cf_map.end()) {
288 return SkString(cf_map[type].c_str());
289 }
290 }
291 } else if (std::regex_search(path.c_str(), m, skiafuzzer)) {
292 std::string a1 = m.str(1);
293 std::string typeOrName = m.str(2);
294 if (a1.length() > 0) {
295 // it's an api fuzzer
296 *name = SkString(typeOrName.c_str());
297 return SkString("api");
298 } else {
299 return SkString(typeOrName.c_str());
300 }
301 }
302
303 return SkString("");
304}
305
Florin Malita80452be2018-06-19 11:27:20 -0400306void FuzzJSON(sk_sp<SkData> bytes);
Florin Malita7796f002018-06-08 12:25:38 -0400307
Florin Malita80452be2018-06-19 11:27:20 -0400308static void fuzz_json(sk_sp<SkData> bytes){
309 FuzzJSON(bytes);
Florin Malita7796f002018-06-08 12:25:38 -0400310 SkDebugf("[terminated] Done parsing!\n");
311}
Florin Malita7796f002018-06-08 12:25:38 -0400312
Florin Malita0b0d93d2018-05-04 15:01:03 -0400313#if defined(SK_ENABLE_SKOTTIE)
Kevin Lubick9eeede22018-05-03 16:26:10 -0400314void FuzzSkottieJSON(sk_sp<SkData> bytes);
315
316static void fuzz_skottie_json(sk_sp<SkData> bytes){
317 FuzzSkottieJSON(bytes);
318 SkDebugf("[terminated] Done animating!\n");
319}
Florin Malita0b0d93d2018-05-04 15:01:03 -0400320#endif
Kevin Lubick9eeede22018-05-03 16:26:10 -0400321
kjlubick2a42f482016-02-16 16:14:23 -0800322// This adds up the first 1024 bytes and returns it as an 8 bit integer. This allows afl-fuzz to
323// deterministically excercise different paths, or *options* (such as different scaling sizes or
324// different image modes) without needing to introduce a parameter. This way we don't need a
325// image_scale1, image_scale2, image_scale4, etc fuzzer, we can just have a image_scale fuzzer.
326// Clients are expected to transform this number into a different range, e.g. with modulo (%).
327static uint8_t calculate_option(SkData* bytes) {
328 uint8_t total = 0;
329 const uint8_t* data = bytes->bytes();
330 for (size_t i = 0; i < 1024 && i < bytes->size(); i++) {
331 total += data[i];
332 }
333 return total;
334}
335
Kevin Lubickfffa6412018-04-23 16:44:55 -0400336static void print_api_names(){
337 SkDebugf("When using --type api, please choose an API to fuzz with --name/-n:\n");
Hal Canary972eba32018-07-30 17:07:07 -0400338 for (const Fuzzable& fuzzable : sk_tools::Registry<Fuzzable>::Range()) {
Kevin Lubickfffa6412018-04-23 16:44:55 -0400339 SkDebugf("\t%s\n", fuzzable.name);
340 }
341}
342
343static void fuzz_api(sk_sp<SkData> bytes, SkString name) {
Hal Canary972eba32018-07-30 17:07:07 -0400344 for (const Fuzzable& fuzzable : sk_tools::Registry<Fuzzable>::Range()) {
Kevin Lubickfffa6412018-04-23 16:44:55 -0400345 if (name.equals(fuzzable.name)) {
mtkleinf5e97822016-01-15 06:19:53 -0800346 SkDebugf("Fuzzing %s...\n", fuzzable.name);
Kevin Lubick1ac8fd22017-03-01 10:42:45 -0500347 Fuzz fuzz(std::move(bytes));
mtklein65e58242016-01-13 12:57:57 -0800348 fuzzable.fn(&fuzz);
kjlubick47d158e2016-02-01 08:23:50 -0800349 SkDebugf("[terminated] Success!\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500350 return;
mtklein65e58242016-01-13 12:57:57 -0800351 }
352 }
mtkleind4387ea2016-01-21 06:13:52 -0800353
Kevin Lubickfffa6412018-04-23 16:44:55 -0400354 print_api_names();
kjlubickdba57342016-01-21 05:03:28 -0800355}
356
357static void dump_png(SkBitmap bitmap) {
358 if (!FLAGS_dump.isEmpty()) {
Mike Kleinea3f0142019-03-20 11:12:10 -0500359 ToolUtils::EncodeImageToFile(FLAGS_dump[0], bitmap, SkEncodedImageFormat::kPNG, 100);
kjlubickdba57342016-01-21 05:03:28 -0800360 SkDebugf("Dumped to %s\n", FLAGS_dump[0]);
361 }
362}
363
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400364bool FuzzAnimatedImage(sk_sp<SkData> bytes);
Kevin Lubick2416f962018-02-12 08:26:39 -0500365
366static void fuzz_animated_img(sk_sp<SkData> bytes) {
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400367 if (FuzzAnimatedImage(bytes)) {
368 SkDebugf("[terminated] Success from decoding/drawing animated image!\n");
369 return;
370 }
371 SkDebugf("[terminated] Could not decode or draw animated image.\n");
Kevin Lubick2416f962018-02-12 08:26:39 -0500372}
373
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400374bool FuzzImageDecode(sk_sp<SkData> bytes);
Kevin Lubick2416f962018-02-12 08:26:39 -0500375
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400376static void fuzz_image_decode(sk_sp<SkData> bytes) {
377 if (FuzzImageDecode(bytes)) {
378 SkDebugf("[terminated] Success from decoding/drawing image!\n");
379 return;
380 }
381 SkDebugf("[terminated] Could not decode or draw image.\n");
Kevin Lubick2416f962018-02-12 08:26:39 -0500382}
383
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400384bool FuzzIncrementalImageDecode(sk_sp<SkData> bytes);
385
386static void fuzz_image_decode_incremental(sk_sp<SkData> bytes) {
387 if (FuzzIncrementalImageDecode(bytes)) {
388 SkDebugf("[terminated] Success using incremental decode!\n");
389 return;
390 }
391 SkDebugf("[terminated] Could not incrementally decode and image.\n");
392}
393
394bool FuzzAndroidCodec(sk_sp<SkData> bytes, uint8_t sampleSize);
395
396static void fuzz_android_codec(sk_sp<SkData> bytes) {
397 Fuzz fuzz(bytes);
398 uint8_t sampleSize;
399 fuzz.nextRange(&sampleSize, 1, 64);
400 bytes = SkData::MakeSubset(bytes.get(), 1, bytes->size() - 1);
401 if (FuzzAndroidCodec(bytes, sampleSize)) {
402 SkDebugf("[terminated] Success on Android Codec sampleSize=%u!\n", sampleSize);
403 return;
404 }
405 SkDebugf("[terminated] Could not use Android Codec sampleSize=%u!\n", sampleSize);
406}
407
408// This is a "legacy" fuzzer that likely does too much. It was based off of how
409// DM reads in images. image_decode, image_decode_incremental and android_codec
410// are more targeted fuzzers that do a subset of what this one does.
Kevin Lubickf80f1152017-01-06 08:26:56 -0500411static void fuzz_img(sk_sp<SkData> bytes, uint8_t scale, uint8_t mode) {
kjlubick2a42f482016-02-16 16:14:23 -0800412 // We can scale 1x, 2x, 4x, 8x, 16x
413 scale = scale % 5;
kjlubick5bd98a22016-02-18 06:27:38 -0800414 float fscale = (float)pow(2.0f, scale);
415 SkDebugf("Scaling factor: %f\n", fscale);
kjlubick2a42f482016-02-16 16:14:23 -0800416
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500417 // We have 5 different modes of decoding.
418 mode = mode % 5;
kjlubick2a42f482016-02-16 16:14:23 -0800419 SkDebugf("Mode: %d\n", mode);
420
421 // This is mostly copied from DMSrcSink's CodecSrc::draw method.
kjlubick47d158e2016-02-01 08:23:50 -0800422 SkDebugf("Decoding\n");
Mike Reedede7bac2017-07-23 15:30:02 -0400423 std::unique_ptr<SkCodec> codec(SkCodec::MakeFromData(bytes));
kjlubickdba57342016-01-21 05:03:28 -0800424 if (nullptr == codec.get()) {
kjlubick47d158e2016-02-01 08:23:50 -0800425 SkDebugf("[terminated] Couldn't create codec.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500426 return;
kjlubickdba57342016-01-21 05:03:28 -0800427 }
428
429 SkImageInfo decodeInfo = codec->getInfo();
kjlubick2a42f482016-02-16 16:14:23 -0800430 SkISize size = codec->getScaledDimensions(fscale);
431 decodeInfo = decodeInfo.makeWH(size.width(), size.height());
432
kjlubickdba57342016-01-21 05:03:28 -0800433 SkBitmap bitmap;
kjlubickdba57342016-01-21 05:03:28 -0800434 SkCodec::Options options;
435 options.fZeroInitialized = SkCodec::kYes_ZeroInitialized;
436
Mike Reed086a4272017-07-18 10:53:11 -0400437 if (!bitmap.tryAllocPixelsFlags(decodeInfo, SkBitmap::kZeroPixels_AllocFlag)) {
kjlubick47d158e2016-02-01 08:23:50 -0800438 SkDebugf("[terminated] Could not allocate memory. Image might be too large (%d x %d)",
kjlubick2a42f482016-02-16 16:14:23 -0800439 decodeInfo.width(), decodeInfo.height());
Kevin Lubickf80f1152017-01-06 08:26:56 -0500440 return;
kjlubickdba57342016-01-21 05:03:28 -0800441 }
442
kjlubick2a42f482016-02-16 16:14:23 -0800443 switch (mode) {
444 case 0: {//kCodecZeroInit_Mode, kCodec_Mode
Leon Scroggins571b30f2017-07-11 17:35:31 +0000445 switch (codec->getPixels(decodeInfo, bitmap.getPixels(), bitmap.rowBytes(), &options)) {
kjlubick2a42f482016-02-16 16:14:23 -0800446 case SkCodec::kSuccess:
447 SkDebugf("[terminated] Success!\n");
448 break;
449 case SkCodec::kIncompleteInput:
450 SkDebugf("[terminated] Partial Success\n");
451 break;
Leon Scroggins III674a1842017-07-06 12:26:09 -0400452 case SkCodec::kErrorInInput:
453 SkDebugf("[terminated] Partial Success with error\n");
454 break;
kjlubick2a42f482016-02-16 16:14:23 -0800455 case SkCodec::kInvalidConversion:
456 SkDebugf("Incompatible colortype conversion\n");
457 // Crash to allow afl-fuzz to know this was a bug.
458 raise(SIGSEGV);
459 default:
460 SkDebugf("[terminated] Couldn't getPixels.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500461 return;
kjlubick2a42f482016-02-16 16:14:23 -0800462 }
463 break;
464 }
465 case 1: {//kScanline_Mode
Leon Scroggins571b30f2017-07-11 17:35:31 +0000466 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo)) {
467 SkDebugf("[terminated] Could not start scanline decoder\n");
468 return;
469 }
kjlubick2a42f482016-02-16 16:14:23 -0800470
471 void* dst = bitmap.getAddr(0, 0);
472 size_t rowBytes = bitmap.rowBytes();
473 uint32_t height = decodeInfo.height();
Brian Osmanea176c62018-04-06 15:28:23 -0400474 // We do not need to check the return value. On an incomplete
475 // image, memory will be filled with a default value.
476 codec->getScanlines(dst, height, rowBytes);
kjlubick47d158e2016-02-01 08:23:50 -0800477 SkDebugf("[terminated] Success!\n");
kjlubickdba57342016-01-21 05:03:28 -0800478 break;
kjlubick2a42f482016-02-16 16:14:23 -0800479 }
480 case 2: { //kStripe_Mode
481 const int height = decodeInfo.height();
482 // This value is chosen arbitrarily. We exercise more cases by choosing a value that
483 // does not align with image blocks.
484 const int stripeHeight = 37;
485 const int numStripes = (height + stripeHeight - 1) / stripeHeight;
486
487 // Decode odd stripes
Leon Scroggins571b30f2017-07-11 17:35:31 +0000488 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo)
kjlubick2a42f482016-02-16 16:14:23 -0800489 || SkCodec::kTopDown_SkScanlineOrder != codec->getScanlineOrder()) {
490 // This mode was designed to test the new skip scanlines API in libjpeg-turbo.
491 // Jpegs have kTopDown_SkScanlineOrder, and at this time, it is not interesting
492 // to run this test for image types that do not have this scanline ordering.
493 SkDebugf("[terminated] Could not start top-down scanline decoder\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500494 return;
kjlubick2a42f482016-02-16 16:14:23 -0800495 }
496
497 for (int i = 0; i < numStripes; i += 2) {
498 // Skip a stripe
499 const int linesToSkip = SkTMin(stripeHeight, height - i * stripeHeight);
500 codec->skipScanlines(linesToSkip);
501
502 // Read a stripe
503 const int startY = (i + 1) * stripeHeight;
504 const int linesToRead = SkTMin(stripeHeight, height - startY);
505 if (linesToRead > 0) {
506 codec->getScanlines(bitmap.getAddr(0, startY), linesToRead, bitmap.rowBytes());
507 }
508 }
509
510 // Decode even stripes
Leon Scroggins571b30f2017-07-11 17:35:31 +0000511 const SkCodec::Result startResult = codec->startScanlineDecode(decodeInfo);
kjlubick2a42f482016-02-16 16:14:23 -0800512 if (SkCodec::kSuccess != startResult) {
513 SkDebugf("[terminated] Failed to restart scanline decoder with same parameters.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500514 return;
kjlubick2a42f482016-02-16 16:14:23 -0800515 }
516 for (int i = 0; i < numStripes; i += 2) {
517 // Read a stripe
518 const int startY = i * stripeHeight;
519 const int linesToRead = SkTMin(stripeHeight, height - startY);
520 codec->getScanlines(bitmap.getAddr(0, startY), linesToRead, bitmap.rowBytes());
521
522 // Skip a stripe
523 const int linesToSkip = SkTMin(stripeHeight, height - (i + 1) * stripeHeight);
524 if (linesToSkip > 0) {
525 codec->skipScanlines(linesToSkip);
526 }
527 }
528 SkDebugf("[terminated] Success!\n");
kjlubickdba57342016-01-21 05:03:28 -0800529 break;
kjlubick2a42f482016-02-16 16:14:23 -0800530 }
531 case 3: { //kSubset_Mode
532 // Arbitrarily choose a divisor.
533 int divisor = 2;
534 // Total width/height of the image.
535 const int W = codec->getInfo().width();
536 const int H = codec->getInfo().height();
537 if (divisor > W || divisor > H) {
538 SkDebugf("[terminated] Cannot codec subset: divisor %d is too big "
539 "with dimensions (%d x %d)\n", divisor, W, H);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500540 return;
kjlubick2a42f482016-02-16 16:14:23 -0800541 }
542 // subset dimensions
543 // SkWebpCodec, the only one that supports subsets, requires even top/left boundaries.
544 const int w = SkAlign2(W / divisor);
545 const int h = SkAlign2(H / divisor);
546 SkIRect subset;
547 SkCodec::Options opts;
548 opts.fSubset = &subset;
549 SkBitmap subsetBm;
550 // We will reuse pixel memory from bitmap.
551 void* pixels = bitmap.getPixels();
552 // Keep track of left and top (for drawing subsetBm into canvas). We could use
553 // fscale * x and fscale * y, but we want integers such that the next subset will start
554 // where the last one ended. So we'll add decodeInfo.width() and height().
555 int left = 0;
556 for (int x = 0; x < W; x += w) {
557 int top = 0;
558 for (int y = 0; y < H; y+= h) {
559 // Do not make the subset go off the edge of the image.
560 const int preScaleW = SkTMin(w, W - x);
561 const int preScaleH = SkTMin(h, H - y);
562 subset.setXYWH(x, y, preScaleW, preScaleH);
563 // And fscale
564 // FIXME: Should we have a version of getScaledDimensions that takes a subset
565 // into account?
566 decodeInfo = decodeInfo.makeWH(
567 SkTMax(1, SkScalarRoundToInt(preScaleW * fscale)),
568 SkTMax(1, SkScalarRoundToInt(preScaleH * fscale)));
569 size_t rowBytes = decodeInfo.minRowBytes();
Leon Scroggins571b30f2017-07-11 17:35:31 +0000570 if (!subsetBm.installPixels(decodeInfo, pixels, rowBytes)) {
kjlubick2a42f482016-02-16 16:14:23 -0800571 SkDebugf("[terminated] Could not install pixels.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500572 return;
kjlubick2a42f482016-02-16 16:14:23 -0800573 }
574 const SkCodec::Result result = codec->getPixels(decodeInfo, pixels, rowBytes,
Leon Scroggins571b30f2017-07-11 17:35:31 +0000575 &opts);
kjlubick2a42f482016-02-16 16:14:23 -0800576 switch (result) {
577 case SkCodec::kSuccess:
578 case SkCodec::kIncompleteInput:
Leon Scroggins III674a1842017-07-06 12:26:09 -0400579 case SkCodec::kErrorInInput:
kjlubick2a42f482016-02-16 16:14:23 -0800580 SkDebugf("okay\n");
581 break;
582 case SkCodec::kInvalidConversion:
583 if (0 == (x|y)) {
584 // First subset is okay to return unimplemented.
585 SkDebugf("[terminated] Incompatible colortype conversion\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500586 return;
kjlubick2a42f482016-02-16 16:14:23 -0800587 }
588 // If the first subset succeeded, a later one should not fail.
589 // fall through to failure
590 case SkCodec::kUnimplemented:
591 if (0 == (x|y)) {
592 // First subset is okay to return unimplemented.
593 SkDebugf("[terminated] subset codec not supported\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500594 return;
kjlubick2a42f482016-02-16 16:14:23 -0800595 }
596 // If the first subset succeeded, why would a later one fail?
597 // fall through to failure
598 default:
599 SkDebugf("[terminated] subset codec failed to decode (%d, %d, %d, %d) "
600 "with dimensions (%d x %d)\t error %d\n",
601 x, y, decodeInfo.width(), decodeInfo.height(),
602 W, H, result);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500603 return;
kjlubick2a42f482016-02-16 16:14:23 -0800604 }
605 // translate by the scaled height.
606 top += decodeInfo.height();
607 }
608 // translate by the scaled width.
609 left += decodeInfo.width();
610 }
611 SkDebugf("[terminated] Success!\n");
612 break;
613 }
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500614 case 4: { //kAnimated_Mode
615 std::vector<SkCodec::FrameInfo> frameInfos = codec->getFrameInfo();
616 if (frameInfos.size() == 0) {
617 SkDebugf("[terminated] Not an animated image\n");
618 break;
619 }
620
621 for (size_t i = 0; i < frameInfos.size(); i++) {
622 options.fFrameIndex = i;
623 auto result = codec->startIncrementalDecode(decodeInfo, bitmap.getPixels(),
624 bitmap.rowBytes(), &options);
625 if (SkCodec::kSuccess != result) {
626 SkDebugf("[terminated] failed to start incremental decode "
627 "in frame %d with error %d\n", i, result);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500628 return;
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500629 }
630
631 result = codec->incrementalDecode();
Leon Scroggins III674a1842017-07-06 12:26:09 -0400632 if (result == SkCodec::kIncompleteInput || result == SkCodec::kErrorInInput) {
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500633 SkDebugf("okay\n");
634 // Frames beyond this one will not decode.
635 break;
636 }
637 if (result == SkCodec::kSuccess) {
638 SkDebugf("okay - decoded frame %d\n", i);
639 } else {
640 SkDebugf("[terminated] incremental decode failed with "
641 "error %d\n", result);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500642 return;
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500643 }
644 }
645 SkDebugf("[terminated] Success!\n");
646 break;
647 }
kjlubickdba57342016-01-21 05:03:28 -0800648 default:
kjlubick2a42f482016-02-16 16:14:23 -0800649 SkDebugf("[terminated] Mode not implemented yet\n");
kjlubickdba57342016-01-21 05:03:28 -0800650 }
651
652 dump_png(bitmap);
mtklein65e58242016-01-13 12:57:57 -0800653}
654
Kevin Lubickf80f1152017-01-06 08:26:56 -0500655static void fuzz_skp(sk_sp<SkData> bytes) {
Kevin Lubick09757b22017-12-12 12:52:39 -0500656 SkReadBuffer buf(bytes->data(), bytes->size());
kjlubickdba57342016-01-21 05:03:28 -0800657 SkDebugf("Decoding\n");
Cary Clarkefd99cc2018-06-11 16:25:43 -0400658 sk_sp<SkPicture> pic(SkPicturePriv::MakeFromBuffer(buf));
kjlubickdba57342016-01-21 05:03:28 -0800659 if (!pic) {
kjlubick47d158e2016-02-01 08:23:50 -0800660 SkDebugf("[terminated] Couldn't decode as a picture.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500661 return;
kjlubickdba57342016-01-21 05:03:28 -0800662 }
663 SkDebugf("Rendering\n");
664 SkBitmap bitmap;
665 if (!FLAGS_dump.isEmpty()) {
666 SkIRect size = pic->cullRect().roundOut();
667 bitmap.allocN32Pixels(size.width(), size.height());
668 }
669 SkCanvas canvas(bitmap);
670 canvas.drawPicture(pic);
kjlubick47d158e2016-02-01 08:23:50 -0800671 SkDebugf("[terminated] Success! Decoded and rendered an SkPicture!\n");
kjlubickdba57342016-01-21 05:03:28 -0800672 dump_png(bitmap);
kjlubickdba57342016-01-21 05:03:28 -0800673}
mtklein65e58242016-01-13 12:57:57 -0800674
Kevin Lubickf80f1152017-01-06 08:26:56 -0500675static void fuzz_color_deserialize(sk_sp<SkData> bytes) {
kjlubick3e3c1a52016-06-23 10:49:27 -0700676 sk_sp<SkColorSpace> space(SkColorSpace::Deserialize(bytes->data(), bytes->size()));
677 if (!space) {
678 SkDebugf("[terminated] Couldn't deserialize Colorspace.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500679 return;
kjlubick3e3c1a52016-06-23 10:49:27 -0700680 }
681 SkDebugf("[terminated] Success! deserialized Colorspace.\n");
kjlubick3e3c1a52016-06-23 10:49:27 -0700682}
683
Kevin Lubickf034d112018-02-08 14:31:24 -0500684void FuzzPathDeserialize(SkReadBuffer& buf);
Kevin Lubickc9d28292017-11-09 08:12:56 -0500685
Kevin Lubickf034d112018-02-08 14:31:24 -0500686static void fuzz_path_deserialize(sk_sp<SkData> bytes) {
687 SkReadBuffer buf(bytes->data(), bytes->size());
688 FuzzPathDeserialize(buf);
689 SkDebugf("[terminated] path_deserialize didn't crash!\n");
Kevin Lubickf04c50a2017-01-06 13:48:19 -0500690}
691
Kevin Lubick2541edf2018-01-11 10:27:14 -0500692bool FuzzRegionDeserialize(sk_sp<SkData> bytes);
693
Kevin Lubickedee1ae2017-02-20 17:47:18 -0500694static void fuzz_region_deserialize(sk_sp<SkData> bytes) {
Kevin Lubick2541edf2018-01-11 10:27:14 -0500695 if (!FuzzRegionDeserialize(bytes)) {
Kevin Lubickedee1ae2017-02-20 17:47:18 -0500696 SkDebugf("[terminated] Couldn't initialize SkRegion.\n");
697 return;
698 }
Kevin Lubickedee1ae2017-02-20 17:47:18 -0500699 SkDebugf("[terminated] Success! Initialized SkRegion.\n");
700}
701
Kevin Lubickf034d112018-02-08 14:31:24 -0500702void FuzzTextBlobDeserialize(SkReadBuffer& buf);
703
Kevin Lubickd46e85f2017-11-21 17:13:35 -0500704static void fuzz_textblob_deserialize(sk_sp<SkData> bytes) {
Mike Reedfadbfcd2017-12-06 16:09:20 -0500705 SkReadBuffer buf(bytes->data(), bytes->size());
Kevin Lubickf034d112018-02-08 14:31:24 -0500706 FuzzTextBlobDeserialize(buf);
707 SkDebugf("[terminated] textblob didn't crash!\n");
Kevin Lubickd46e85f2017-11-21 17:13:35 -0500708}
709
Kevin Lubick2541edf2018-01-11 10:27:14 -0500710void FuzzRegionSetPath(Fuzz* fuzz);
711
712static void fuzz_region_set_path(sk_sp<SkData> bytes) {
713 Fuzz fuzz(bytes);
714 FuzzRegionSetPath(&fuzz);
715 SkDebugf("[terminated] region_set_path didn't crash!\n");
716}
717
Kevin Lubickf034d112018-02-08 14:31:24 -0500718void FuzzImageFilterDeserialize(sk_sp<SkData> bytes);
719
Herb Derbya839fc02017-03-16 12:30:43 -0400720static void fuzz_filter_fuzz(sk_sp<SkData> bytes) {
Kevin Lubickf034d112018-02-08 14:31:24 -0500721 FuzzImageFilterDeserialize(bytes);
722 SkDebugf("[terminated] filter_fuzz didn't crash!\n");
Herb Derbya839fc02017-03-16 12:30:43 -0400723}
724
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400725bool FuzzSKSL2GLSL(sk_sp<SkData> bytes);
726
Kevin Lubickf80f1152017-01-06 08:26:56 -0500727static void fuzz_sksl2glsl(sk_sp<SkData> bytes) {
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400728 if (FuzzSKSL2GLSL(bytes)) {
729 SkDebugf("[terminated] Success! Compiled input to GLSL.\n");
730 } else {
731 SkDebugf("[terminated] Could not compile input to GLSL.\n");
kjlubicke7195772016-10-18 10:06:24 -0700732 }
kjlubicke7195772016-10-18 10:06:24 -0700733}
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400734
735bool FuzzSKSL2SPIRV(sk_sp<SkData> bytes);
736
737static void fuzz_sksl2spirv(sk_sp<SkData> bytes) {
738 if (FuzzSKSL2SPIRV(bytes)) {
739 SkDebugf("[terminated] Success! Compiled input to SPIRV.\n");
740 } else {
741 SkDebugf("[terminated] Could not compile input to SPIRV.\n");
742 }
743}
744
745bool FuzzSKSL2Metal(sk_sp<SkData> bytes);
746
747static void fuzz_sksl2metal(sk_sp<SkData> bytes) {
748 if (FuzzSKSL2Metal(bytes)) {
Kevin Lubick39cbe462019-03-11 15:38:00 -0400749 SkDebugf("[terminated] Success! Compiled input to Metal.\n");
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400750 } else {
Kevin Lubick39cbe462019-03-11 15:38:00 -0400751 SkDebugf("[terminated] Could not compile input to Metal.\n");
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400752 }
753}
Kevin Lubick0f0a7102019-03-18 16:20:55 -0400754
755bool FuzzSKSL2Pipeline(sk_sp<SkData> bytes);
756
757static void fuzz_sksl2pipeline(sk_sp<SkData> bytes) {
758 if (FuzzSKSL2Pipeline(bytes)) {
759 SkDebugf("[terminated] Success! Compiled input to pipeline stage.\n");
760 } else {
761 SkDebugf("[terminated] Could not compile input to pipeline stage.\n");
762 }
763}