blob: 1047e5f153a7431eebfaf4fab1fd3b775490c48a [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 Klein88544fb2019-03-20 10:50:33 -05008#include "CommandLineFlags.h"
mtklein65e58242016-01-13 12:57:57 -08009#include "Fuzz.h"
kjlubickdba57342016-01-21 05:03:28 -080010#include "SkCanvas.h"
11#include "SkCodec.h"
kjlubickdba57342016-01-21 05:03:28 -080012#include "SkData.h"
kjlubickdba57342016-01-21 05:03:28 -080013#include "SkImage.h"
14#include "SkImageEncoder.h"
15#include "SkMallocPixelRef.h"
Kevin Lubickf80f1152017-01-06 08:26:56 -050016#include "SkOSFile.h"
17#include "SkOSPath.h"
Herb Derbya839fc02017-03-16 12:30:43 -040018#include "SkPaint.h"
Hal Canary62176db2017-02-27 16:42:03 -050019#include "SkPath.h"
Cary Clarkefd99cc2018-06-11 16:25:43 -040020#include "SkPicturePriv.h"
Mike Reedfadbfcd2017-12-06 16:09:20 -050021#include "SkReadBuffer.h"
Hal Canary62176db2017-02-27 16:42:03 -050022#include "SkStream.h"
23#include "SkSurface.h"
Kevin Lubickd46e85f2017-11-21 17:13:35 -050024#include "SkTextBlob.h"
Hal Canary62176db2017-02-27 16:42:03 -050025
Mike Kleinea3f0142019-03-20 11:12:10 -050026#include "ToolUtils.h"
Hal Canary8a001442018-09-19 11:31:27 -040027
Hal Canary62176db2017-02-27 16:42:03 -050028#include <iostream>
Kevin Lubickacd456a2018-04-24 13:58:16 -040029#include <map>
Kevin Lubickfffa6412018-04-23 16:44:55 -040030#include <regex>
mtkleina1159422016-01-15 05:46:54 -080031#include <signal.h>
Kevin Lubick2541edf2018-01-11 10:27:14 -050032
Mike Klein84836b72019-03-21 11:31:36 -050033static DEFINE_string2(bytes, b, "", "A path to a file or a directory. If a file, the "
34 "contents will be used as the fuzz bytes. If a directory, all files "
35 "in the directory will be used as fuzz bytes for the fuzzer, one at a "
36 "time.");
37static DEFINE_string2(name, n, "", "If --type is 'api', fuzz the API with this name.");
38static DEFINE_string2(dump, d, "", "If not empty, dump 'image*' or 'skp' types as a "
39 "PNG with this name.");
Kevin Lubick9ff5dc92018-01-09 12:47:33 -050040DEFINE_bool2(verbose, v, false, "Print more information while fuzzing.");
Florin Malita7796f002018-06-08 12:25:38 -040041
42// This cannot be inlined in DEFINE_string2 due to interleaved ifdefs
43static constexpr char g_type_message[] = "How to interpret --bytes, one of:\n"
Kevin Lubick0f3d2a62018-10-17 10:24:44 -040044 "android_codec\n"
Florin Malita7796f002018-06-08 12:25:38 -040045 "animated_image_decode\n"
46 "api\n"
47 "color_deserialize\n"
48 "filter_fuzz (equivalent to Chrome's filter_fuzz_stub)\n"
49 "image_decode\n"
Kevin Lubick0f3d2a62018-10-17 10:24:44 -040050 "image_decode_incremental\n"
Florin Malita7796f002018-06-08 12:25:38 -040051 "image_mode\n"
52 "image_scale\n"
Florin Malita80452be2018-06-19 11:27:20 -040053 "json\n"
Florin Malita7796f002018-06-08 12:25:38 -040054 "path_deserialize\n"
Florin Malita7796f002018-06-08 12:25:38 -040055 "region_deserialize\n"
56 "region_set_path\n"
Florin Malita7796f002018-06-08 12:25:38 -040057 "skp\n"
58 "sksl2glsl\n"
Kevin Lubicke9c1ce82019-03-11 11:09:40 -040059 "sksl2metal\n"
Kevin Lubick0f0a7102019-03-18 16:20:55 -040060 "sksl2pipeline\n"
Kevin Lubicke9c1ce82019-03-11 11:09:40 -040061 "sksl2spirv\n"
Florin Malita7796f002018-06-08 12:25:38 -040062#if defined(SK_ENABLE_SKOTTIE)
63 "skottie_json\n"
64#endif
65 "textblob";
66
Mike Klein84836b72019-03-21 11:31:36 -050067static DEFINE_string2(type, t, "", g_type_message);
kjlubickdba57342016-01-21 05:03:28 -080068
Kevin Lubickfffa6412018-04-23 16:44:55 -040069static int fuzz_file(SkString path, SkString type);
kjlubick2a42f482016-02-16 16:14:23 -080070static uint8_t calculate_option(SkData*);
Kevin Lubickfffa6412018-04-23 16:44:55 -040071static SkString try_auto_detect(SkString path, SkString* name);
kjlubickdba57342016-01-21 05:03:28 -080072
Kevin Lubick0f3d2a62018-10-17 10:24:44 -040073static void fuzz_android_codec(sk_sp<SkData>);
74static void fuzz_animated_img(sk_sp<SkData>);
Kevin Lubickfffa6412018-04-23 16:44:55 -040075static void fuzz_api(sk_sp<SkData> bytes, SkString name);
Kevin Lubickf80f1152017-01-06 08:26:56 -050076static void fuzz_color_deserialize(sk_sp<SkData>);
Kevin Lubickd46e85f2017-11-21 17:13:35 -050077static void fuzz_filter_fuzz(sk_sp<SkData>);
Kevin Lubick0f3d2a62018-10-17 10:24:44 -040078static void fuzz_image_decode(sk_sp<SkData>);
79static void fuzz_image_decode_incremental(sk_sp<SkData>);
Kevin Lubick0168e042017-02-14 13:12:37 -050080static void fuzz_img(sk_sp<SkData>, uint8_t, uint8_t);
Florin Malita80452be2018-06-19 11:27:20 -040081static void fuzz_json(sk_sp<SkData>);
Kevin Lubickf04c50a2017-01-06 13:48:19 -050082static void fuzz_path_deserialize(sk_sp<SkData>);
Kevin Lubickedee1ae2017-02-20 17:47:18 -050083static void fuzz_region_deserialize(sk_sp<SkData>);
Kevin Lubick2541edf2018-01-11 10:27:14 -050084static void fuzz_region_set_path(sk_sp<SkData>);
Kevin Lubick0168e042017-02-14 13:12:37 -050085static void fuzz_skp(sk_sp<SkData>);
Kevin Lubicke9c1ce82019-03-11 11:09:40 -040086static void fuzz_sksl2glsl(sk_sp<SkData>);
87static void fuzz_sksl2metal(sk_sp<SkData>);
88static void fuzz_sksl2spirv(sk_sp<SkData>);
Kevin Lubick0f0a7102019-03-18 16:20:55 -040089static void fuzz_sksl2pipeline(sk_sp<SkData>);
Kevin Lubickd46e85f2017-11-21 17:13:35 -050090static void fuzz_textblob_deserialize(sk_sp<SkData>);
Herb Derbya839fc02017-03-16 12:30:43 -040091
Kevin Lubickfffa6412018-04-23 16:44:55 -040092static void print_api_names();
93
Florin Malita0b0d93d2018-05-04 15:01:03 -040094#if defined(SK_ENABLE_SKOTTIE)
95static void fuzz_skottie_json(sk_sp<SkData>);
96#endif
97
mtklein65e58242016-01-13 12:57:57 -080098int main(int argc, char** argv) {
Mike Klein88544fb2019-03-20 10:50:33 -050099 CommandLineFlags::SetUsage(
100 "Usage: fuzz -t <type> -b <path/to/file> [-n api-to-fuzz]\n"
101 " fuzz -b <path/to/file>\n"
102 "--help lists the valid types. If type is not specified,\n"
103 "fuzz will make a guess based on the name of the file.\n");
104 CommandLineFlags::Parse(argc, argv);
mtkleinf5e97822016-01-15 06:19:53 -0800105
Kevin Lubickfffa6412018-04-23 16:44:55 -0400106 SkString path = SkString(FLAGS_bytes.isEmpty() ? argv[0] : FLAGS_bytes[0]);
107 SkString type = SkString(FLAGS_type.isEmpty() ? "" : FLAGS_type[0]);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500108
Kevin Lubickfffa6412018-04-23 16:44:55 -0400109 if (!sk_isdir(path.c_str())) {
110 return fuzz_file(path, type);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500111 }
112
Kevin Lubickfffa6412018-04-23 16:44:55 -0400113 SkOSFile::Iter it(path.c_str());
Kevin Lubickf80f1152017-01-06 08:26:56 -0500114 for (SkString file; it.next(&file); ) {
Kevin Lubickfffa6412018-04-23 16:44:55 -0400115 SkString p = SkOSPath::Join(path.c_str(), file.c_str());
Kevin Lubickf80f1152017-01-06 08:26:56 -0500116 SkDebugf("Fuzzing %s\n", p.c_str());
Kevin Lubickfffa6412018-04-23 16:44:55 -0400117 int rv = fuzz_file(p, type);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500118 if (rv != 0) {
119 return rv;
120 }
121 }
122 return 0;
123}
124
Kevin Lubickfffa6412018-04-23 16:44:55 -0400125static int fuzz_file(SkString path, SkString type) {
126 sk_sp<SkData> bytes(SkData::MakeFromFileName(path.c_str()));
kjlubickdba57342016-01-21 05:03:28 -0800127 if (!bytes) {
Kevin Lubickfffa6412018-04-23 16:44:55 -0400128 SkDebugf("Could not read %s\n", path.c_str());
Kevin Lubickf80f1152017-01-06 08:26:56 -0500129 return 1;
kjlubickdba57342016-01-21 05:03:28 -0800130 }
mtklein65e58242016-01-13 12:57:57 -0800131
Kevin Lubickfffa6412018-04-23 16:44:55 -0400132 SkString name = SkString(FLAGS_name.isEmpty() ? "" : FLAGS_name[0]);
133
134 if (type.isEmpty()) {
135 type = try_auto_detect(path, &name);
kjlubickdba57342016-01-21 05:03:28 -0800136 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400137
138 if (type.isEmpty()) {
139 SkDebugf("Could not autodetect type of %s\n", path.c_str());
140 return 1;
141 }
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400142 if (type.equals("android_codec")) {
143 fuzz_android_codec(bytes);
144 return 0;
145 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400146 if (type.equals("animated_image_decode")) {
147 fuzz_animated_img(bytes);
148 return 0;
149 }
150 if (type.equals("api")) {
151 fuzz_api(bytes, name);
152 return 0;
153 }
154 if (type.equals("color_deserialize")) {
155 fuzz_color_deserialize(bytes);
156 return 0;
157 }
Florin Malita80452be2018-06-19 11:27:20 -0400158 if (type.equals("filter_fuzz")) {
159 fuzz_filter_fuzz(bytes);
160 return 0;
161 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400162 if (type.equals("image_decode")) {
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400163 fuzz_image_decode(bytes);
164 return 0;
165 }
166 if (type.equals("image_decode_incremental")) {
167 fuzz_image_decode_incremental(bytes);
Kevin Lubickfffa6412018-04-23 16:44:55 -0400168 return 0;
169 }
170 if (type.equals("image_scale")) {
171 uint8_t option = calculate_option(bytes.get());
172 fuzz_img(bytes, option, 0);
173 return 0;
174 }
175 if (type.equals("image_mode")) {
176 uint8_t option = calculate_option(bytes.get());
177 fuzz_img(bytes, 0, option);
178 return 0;
179 }
Florin Malita80452be2018-06-19 11:27:20 -0400180 if (type.equals("json")) {
181 fuzz_json(bytes);
Kevin Lubick9eeede22018-05-03 16:26:10 -0400182 return 0;
183 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400184 if (type.equals("path_deserialize")) {
185 fuzz_path_deserialize(bytes);
186 return 0;
187 }
188 if (type.equals("region_deserialize")) {
189 fuzz_region_deserialize(bytes);
190 return 0;
191 }
192 if (type.equals("region_set_path")) {
193 fuzz_region_set_path(bytes);
194 return 0;
195 }
196 if (type.equals("pipe")) {
Mike Klein60900b52018-09-21 11:19:45 -0400197 SkDebugf("I would prefer not to.\n");
Kevin Lubickfffa6412018-04-23 16:44:55 -0400198 return 0;
199 }
Florin Malita0b0d93d2018-05-04 15:01:03 -0400200#if defined(SK_ENABLE_SKOTTIE)
Kevin Lubick9eeede22018-05-03 16:26:10 -0400201 if (type.equals("skottie_json")) {
202 fuzz_skottie_json(bytes);
Kevin Lubickfffa6412018-04-23 16:44:55 -0400203 return 0;
204 }
Florin Malita0b0d93d2018-05-04 15:01:03 -0400205#endif
Kevin Lubick9eeede22018-05-03 16:26:10 -0400206 if (type.equals("skp")) {
207 fuzz_skp(bytes);
Kevin Lubickfffa6412018-04-23 16:44:55 -0400208 return 0;
209 }
Kevin Lubickfffa6412018-04-23 16:44:55 -0400210 if (type.equals("sksl2glsl")) {
211 fuzz_sksl2glsl(bytes);
212 return 0;
213 }
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400214 if (type.equals("sksl2metal")) {
215 fuzz_sksl2metal(bytes);
216 return 0;
217 }
218 if (type.equals("sksl2spirv")) {
219 fuzz_sksl2spirv(bytes);
220 return 0;
221 }
Kevin Lubick0f0a7102019-03-18 16:20:55 -0400222 if (type.equals("sksl2pipeline")) {
223 fuzz_sksl2pipeline(bytes);
224 return 0;
225 }
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400226 if (type.equals("textblob")) {
227 fuzz_textblob_deserialize(bytes);
228 return 0;
229 }
Kevin Lubick457fa972018-05-29 09:22:06 -0400230 SkDebugf("Unknown type %s\n", type.c_str());
Mike Klein88544fb2019-03-20 10:50:33 -0500231 CommandLineFlags::PrintUsage();
Kevin Lubick9ff5dc92018-01-09 12:47:33 -0500232 return 1;
kjlubickdba57342016-01-21 05:03:28 -0800233}
234
Kevin Lubickfffa6412018-04-23 16:44:55 -0400235static std::map<std::string, std::string> cf_api_map = {
236 {"api_draw_functions", "DrawFunctions"},
237 {"api_gradients", "Gradients"},
238 {"api_image_filter", "ImageFilter"},
239 {"api_mock_gpu_canvas", "MockGPUCanvas"},
240 {"api_null_canvas", "NullCanvas"},
241 {"api_path_measure", "PathMeasure"},
Kevin Lubickde54d7c2018-09-24 08:48:19 -0400242 {"api_pathop", "Pathop"},
Kevin Lubick57507f12018-10-11 09:35:15 -0400243 {"api_polyutils", "PolyUtils"},
Kevin Lubickfffa6412018-04-23 16:44:55 -0400244 {"api_raster_n32_canvas", "RasterN32Canvas"},
245 {"jpeg_encoder", "JPEGEncoder"},
246 {"png_encoder", "PNGEncoder"},
Kevin Lubickf84ded22018-10-23 09:28:48 -0400247 {"skia_pathop_fuzzer", "LegacyChromiumPathop"},
Kevin Lubickfffa6412018-04-23 16:44:55 -0400248 {"webp_encoder", "WEBPEncoder"}
249};
250
Kevin Lubick6c560552018-06-20 09:29:23 -0400251// maps clusterfuzz/oss-fuzz -> Skia's name
Kevin Lubickfffa6412018-04-23 16:44:55 -0400252static std::map<std::string, std::string> cf_map = {
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400253 {"android_codec", "android_codec"},
Kevin Lubickfffa6412018-04-23 16:44:55 -0400254 {"animated_image_decode", "animated_image_decode"},
255 {"image_decode", "image_decode"},
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400256 {"image_decode_incremental", "image_decode_incremental"},
Kevin Lubickfffa6412018-04-23 16:44:55 -0400257 {"image_filter_deserialize", "filter_fuzz"},
258 {"image_filter_deserialize_width", "filter_fuzz"},
259 {"path_deserialize", "path_deserialize"},
260 {"region_deserialize", "region_deserialize"},
261 {"region_set_path", "region_set_path"},
Kevin Lubick6c560552018-06-20 09:29:23 -0400262 {"skjson", "json"},
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400263 {"sksl2glsl", "sksl2glsl"},
264 {"sksl2metal", "sksl2metal"},
265 {"sksl2spirv", "sksl2spirv"},
Kevin Lubick0f0a7102019-03-18 16:20:55 -0400266 {"sksl2pipeline", "sksl2pipeline"},
Kevin Lubickde54d7c2018-09-24 08:48:19 -0400267#if defined(SK_ENABLE_SKOTTIE)
268 {"skottie_json", "skottie_json"},
269#endif
Kevin Lubickfffa6412018-04-23 16:44:55 -0400270 {"textblob_deserialize", "textblob"}
271};
272
273static SkString try_auto_detect(SkString path, SkString* name) {
274 std::cmatch m;
275 std::regex clusterfuzz("clusterfuzz-testcase(-minimized)?-([a-z0-9_]+)-[\\d]+");
276 std::regex skiafuzzer("(api-)?(\\w+)-[a-f0-9]+");
277
278 if (std::regex_search(path.c_str(), m, clusterfuzz)) {
279 std::string type = m.str(2);
Kevin Lubick6c560552018-06-20 09:29:23 -0400280
281 if (cf_api_map.find(type) != cf_api_map.end()) {
282 *name = SkString(cf_api_map[type].c_str());
283 return SkString("api");
Kevin Lubickfffa6412018-04-23 16:44:55 -0400284 } else {
285 if (cf_map.find(type) != cf_map.end()) {
286 return SkString(cf_map[type].c_str());
287 }
288 }
289 } else if (std::regex_search(path.c_str(), m, skiafuzzer)) {
290 std::string a1 = m.str(1);
291 std::string typeOrName = m.str(2);
292 if (a1.length() > 0) {
293 // it's an api fuzzer
294 *name = SkString(typeOrName.c_str());
295 return SkString("api");
296 } else {
297 return SkString(typeOrName.c_str());
298 }
299 }
300
301 return SkString("");
302}
303
Florin Malita80452be2018-06-19 11:27:20 -0400304void FuzzJSON(sk_sp<SkData> bytes);
Florin Malita7796f002018-06-08 12:25:38 -0400305
Florin Malita80452be2018-06-19 11:27:20 -0400306static void fuzz_json(sk_sp<SkData> bytes){
307 FuzzJSON(bytes);
Florin Malita7796f002018-06-08 12:25:38 -0400308 SkDebugf("[terminated] Done parsing!\n");
309}
Florin Malita7796f002018-06-08 12:25:38 -0400310
Florin Malita0b0d93d2018-05-04 15:01:03 -0400311#if defined(SK_ENABLE_SKOTTIE)
Kevin Lubick9eeede22018-05-03 16:26:10 -0400312void FuzzSkottieJSON(sk_sp<SkData> bytes);
313
314static void fuzz_skottie_json(sk_sp<SkData> bytes){
315 FuzzSkottieJSON(bytes);
316 SkDebugf("[terminated] Done animating!\n");
317}
Florin Malita0b0d93d2018-05-04 15:01:03 -0400318#endif
Kevin Lubick9eeede22018-05-03 16:26:10 -0400319
kjlubick2a42f482016-02-16 16:14:23 -0800320// This adds up the first 1024 bytes and returns it as an 8 bit integer. This allows afl-fuzz to
321// deterministically excercise different paths, or *options* (such as different scaling sizes or
322// different image modes) without needing to introduce a parameter. This way we don't need a
323// image_scale1, image_scale2, image_scale4, etc fuzzer, we can just have a image_scale fuzzer.
324// Clients are expected to transform this number into a different range, e.g. with modulo (%).
325static uint8_t calculate_option(SkData* bytes) {
326 uint8_t total = 0;
327 const uint8_t* data = bytes->bytes();
328 for (size_t i = 0; i < 1024 && i < bytes->size(); i++) {
329 total += data[i];
330 }
331 return total;
332}
333
Kevin Lubickfffa6412018-04-23 16:44:55 -0400334static void print_api_names(){
335 SkDebugf("When using --type api, please choose an API to fuzz with --name/-n:\n");
Hal Canary972eba32018-07-30 17:07:07 -0400336 for (const Fuzzable& fuzzable : sk_tools::Registry<Fuzzable>::Range()) {
Kevin Lubickfffa6412018-04-23 16:44:55 -0400337 SkDebugf("\t%s\n", fuzzable.name);
338 }
339}
340
341static void fuzz_api(sk_sp<SkData> bytes, SkString name) {
Hal Canary972eba32018-07-30 17:07:07 -0400342 for (const Fuzzable& fuzzable : sk_tools::Registry<Fuzzable>::Range()) {
Kevin Lubickfffa6412018-04-23 16:44:55 -0400343 if (name.equals(fuzzable.name)) {
mtkleinf5e97822016-01-15 06:19:53 -0800344 SkDebugf("Fuzzing %s...\n", fuzzable.name);
Kevin Lubick1ac8fd22017-03-01 10:42:45 -0500345 Fuzz fuzz(std::move(bytes));
mtklein65e58242016-01-13 12:57:57 -0800346 fuzzable.fn(&fuzz);
kjlubick47d158e2016-02-01 08:23:50 -0800347 SkDebugf("[terminated] Success!\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500348 return;
mtklein65e58242016-01-13 12:57:57 -0800349 }
350 }
mtkleind4387ea2016-01-21 06:13:52 -0800351
Kevin Lubickfffa6412018-04-23 16:44:55 -0400352 print_api_names();
kjlubickdba57342016-01-21 05:03:28 -0800353}
354
355static void dump_png(SkBitmap bitmap) {
356 if (!FLAGS_dump.isEmpty()) {
Mike Kleinea3f0142019-03-20 11:12:10 -0500357 ToolUtils::EncodeImageToFile(FLAGS_dump[0], bitmap, SkEncodedImageFormat::kPNG, 100);
kjlubickdba57342016-01-21 05:03:28 -0800358 SkDebugf("Dumped to %s\n", FLAGS_dump[0]);
359 }
360}
361
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400362bool FuzzAnimatedImage(sk_sp<SkData> bytes);
Kevin Lubick2416f962018-02-12 08:26:39 -0500363
364static void fuzz_animated_img(sk_sp<SkData> bytes) {
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400365 if (FuzzAnimatedImage(bytes)) {
366 SkDebugf("[terminated] Success from decoding/drawing animated image!\n");
367 return;
368 }
369 SkDebugf("[terminated] Could not decode or draw animated image.\n");
Kevin Lubick2416f962018-02-12 08:26:39 -0500370}
371
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400372bool FuzzImageDecode(sk_sp<SkData> bytes);
Kevin Lubick2416f962018-02-12 08:26:39 -0500373
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400374static void fuzz_image_decode(sk_sp<SkData> bytes) {
375 if (FuzzImageDecode(bytes)) {
376 SkDebugf("[terminated] Success from decoding/drawing image!\n");
377 return;
378 }
379 SkDebugf("[terminated] Could not decode or draw image.\n");
Kevin Lubick2416f962018-02-12 08:26:39 -0500380}
381
Kevin Lubick0f3d2a62018-10-17 10:24:44 -0400382bool FuzzIncrementalImageDecode(sk_sp<SkData> bytes);
383
384static void fuzz_image_decode_incremental(sk_sp<SkData> bytes) {
385 if (FuzzIncrementalImageDecode(bytes)) {
386 SkDebugf("[terminated] Success using incremental decode!\n");
387 return;
388 }
389 SkDebugf("[terminated] Could not incrementally decode and image.\n");
390}
391
392bool FuzzAndroidCodec(sk_sp<SkData> bytes, uint8_t sampleSize);
393
394static void fuzz_android_codec(sk_sp<SkData> bytes) {
395 Fuzz fuzz(bytes);
396 uint8_t sampleSize;
397 fuzz.nextRange(&sampleSize, 1, 64);
398 bytes = SkData::MakeSubset(bytes.get(), 1, bytes->size() - 1);
399 if (FuzzAndroidCodec(bytes, sampleSize)) {
400 SkDebugf("[terminated] Success on Android Codec sampleSize=%u!\n", sampleSize);
401 return;
402 }
403 SkDebugf("[terminated] Could not use Android Codec sampleSize=%u!\n", sampleSize);
404}
405
406// This is a "legacy" fuzzer that likely does too much. It was based off of how
407// DM reads in images. image_decode, image_decode_incremental and android_codec
408// are more targeted fuzzers that do a subset of what this one does.
Kevin Lubickf80f1152017-01-06 08:26:56 -0500409static void fuzz_img(sk_sp<SkData> bytes, uint8_t scale, uint8_t mode) {
kjlubick2a42f482016-02-16 16:14:23 -0800410 // We can scale 1x, 2x, 4x, 8x, 16x
411 scale = scale % 5;
kjlubick5bd98a22016-02-18 06:27:38 -0800412 float fscale = (float)pow(2.0f, scale);
413 SkDebugf("Scaling factor: %f\n", fscale);
kjlubick2a42f482016-02-16 16:14:23 -0800414
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500415 // We have 5 different modes of decoding.
416 mode = mode % 5;
kjlubick2a42f482016-02-16 16:14:23 -0800417 SkDebugf("Mode: %d\n", mode);
418
419 // This is mostly copied from DMSrcSink's CodecSrc::draw method.
kjlubick47d158e2016-02-01 08:23:50 -0800420 SkDebugf("Decoding\n");
Mike Reedede7bac2017-07-23 15:30:02 -0400421 std::unique_ptr<SkCodec> codec(SkCodec::MakeFromData(bytes));
kjlubickdba57342016-01-21 05:03:28 -0800422 if (nullptr == codec.get()) {
kjlubick47d158e2016-02-01 08:23:50 -0800423 SkDebugf("[terminated] Couldn't create codec.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500424 return;
kjlubickdba57342016-01-21 05:03:28 -0800425 }
426
427 SkImageInfo decodeInfo = codec->getInfo();
kjlubick2a42f482016-02-16 16:14:23 -0800428 SkISize size = codec->getScaledDimensions(fscale);
429 decodeInfo = decodeInfo.makeWH(size.width(), size.height());
430
kjlubickdba57342016-01-21 05:03:28 -0800431 SkBitmap bitmap;
kjlubickdba57342016-01-21 05:03:28 -0800432 SkCodec::Options options;
433 options.fZeroInitialized = SkCodec::kYes_ZeroInitialized;
434
Mike Reed086a4272017-07-18 10:53:11 -0400435 if (!bitmap.tryAllocPixelsFlags(decodeInfo, SkBitmap::kZeroPixels_AllocFlag)) {
kjlubick47d158e2016-02-01 08:23:50 -0800436 SkDebugf("[terminated] Could not allocate memory. Image might be too large (%d x %d)",
kjlubick2a42f482016-02-16 16:14:23 -0800437 decodeInfo.width(), decodeInfo.height());
Kevin Lubickf80f1152017-01-06 08:26:56 -0500438 return;
kjlubickdba57342016-01-21 05:03:28 -0800439 }
440
kjlubick2a42f482016-02-16 16:14:23 -0800441 switch (mode) {
442 case 0: {//kCodecZeroInit_Mode, kCodec_Mode
Leon Scroggins571b30f2017-07-11 17:35:31 +0000443 switch (codec->getPixels(decodeInfo, bitmap.getPixels(), bitmap.rowBytes(), &options)) {
kjlubick2a42f482016-02-16 16:14:23 -0800444 case SkCodec::kSuccess:
445 SkDebugf("[terminated] Success!\n");
446 break;
447 case SkCodec::kIncompleteInput:
448 SkDebugf("[terminated] Partial Success\n");
449 break;
Leon Scroggins III674a1842017-07-06 12:26:09 -0400450 case SkCodec::kErrorInInput:
451 SkDebugf("[terminated] Partial Success with error\n");
452 break;
kjlubick2a42f482016-02-16 16:14:23 -0800453 case SkCodec::kInvalidConversion:
454 SkDebugf("Incompatible colortype conversion\n");
455 // Crash to allow afl-fuzz to know this was a bug.
456 raise(SIGSEGV);
457 default:
458 SkDebugf("[terminated] Couldn't getPixels.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500459 return;
kjlubick2a42f482016-02-16 16:14:23 -0800460 }
461 break;
462 }
463 case 1: {//kScanline_Mode
Leon Scroggins571b30f2017-07-11 17:35:31 +0000464 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo)) {
465 SkDebugf("[terminated] Could not start scanline decoder\n");
466 return;
467 }
kjlubick2a42f482016-02-16 16:14:23 -0800468
469 void* dst = bitmap.getAddr(0, 0);
470 size_t rowBytes = bitmap.rowBytes();
471 uint32_t height = decodeInfo.height();
Brian Osmanea176c62018-04-06 15:28:23 -0400472 // We do not need to check the return value. On an incomplete
473 // image, memory will be filled with a default value.
474 codec->getScanlines(dst, height, rowBytes);
kjlubick47d158e2016-02-01 08:23:50 -0800475 SkDebugf("[terminated] Success!\n");
kjlubickdba57342016-01-21 05:03:28 -0800476 break;
kjlubick2a42f482016-02-16 16:14:23 -0800477 }
478 case 2: { //kStripe_Mode
479 const int height = decodeInfo.height();
480 // This value is chosen arbitrarily. We exercise more cases by choosing a value that
481 // does not align with image blocks.
482 const int stripeHeight = 37;
483 const int numStripes = (height + stripeHeight - 1) / stripeHeight;
484
485 // Decode odd stripes
Leon Scroggins571b30f2017-07-11 17:35:31 +0000486 if (SkCodec::kSuccess != codec->startScanlineDecode(decodeInfo)
kjlubick2a42f482016-02-16 16:14:23 -0800487 || SkCodec::kTopDown_SkScanlineOrder != codec->getScanlineOrder()) {
488 // This mode was designed to test the new skip scanlines API in libjpeg-turbo.
489 // Jpegs have kTopDown_SkScanlineOrder, and at this time, it is not interesting
490 // to run this test for image types that do not have this scanline ordering.
491 SkDebugf("[terminated] Could not start top-down scanline decoder\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500492 return;
kjlubick2a42f482016-02-16 16:14:23 -0800493 }
494
495 for (int i = 0; i < numStripes; i += 2) {
496 // Skip a stripe
497 const int linesToSkip = SkTMin(stripeHeight, height - i * stripeHeight);
498 codec->skipScanlines(linesToSkip);
499
500 // Read a stripe
501 const int startY = (i + 1) * stripeHeight;
502 const int linesToRead = SkTMin(stripeHeight, height - startY);
503 if (linesToRead > 0) {
504 codec->getScanlines(bitmap.getAddr(0, startY), linesToRead, bitmap.rowBytes());
505 }
506 }
507
508 // Decode even stripes
Leon Scroggins571b30f2017-07-11 17:35:31 +0000509 const SkCodec::Result startResult = codec->startScanlineDecode(decodeInfo);
kjlubick2a42f482016-02-16 16:14:23 -0800510 if (SkCodec::kSuccess != startResult) {
511 SkDebugf("[terminated] Failed to restart scanline decoder with same parameters.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500512 return;
kjlubick2a42f482016-02-16 16:14:23 -0800513 }
514 for (int i = 0; i < numStripes; i += 2) {
515 // Read a stripe
516 const int startY = i * stripeHeight;
517 const int linesToRead = SkTMin(stripeHeight, height - startY);
518 codec->getScanlines(bitmap.getAddr(0, startY), linesToRead, bitmap.rowBytes());
519
520 // Skip a stripe
521 const int linesToSkip = SkTMin(stripeHeight, height - (i + 1) * stripeHeight);
522 if (linesToSkip > 0) {
523 codec->skipScanlines(linesToSkip);
524 }
525 }
526 SkDebugf("[terminated] Success!\n");
kjlubickdba57342016-01-21 05:03:28 -0800527 break;
kjlubick2a42f482016-02-16 16:14:23 -0800528 }
529 case 3: { //kSubset_Mode
530 // Arbitrarily choose a divisor.
531 int divisor = 2;
532 // Total width/height of the image.
533 const int W = codec->getInfo().width();
534 const int H = codec->getInfo().height();
535 if (divisor > W || divisor > H) {
536 SkDebugf("[terminated] Cannot codec subset: divisor %d is too big "
537 "with dimensions (%d x %d)\n", divisor, W, H);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500538 return;
kjlubick2a42f482016-02-16 16:14:23 -0800539 }
540 // subset dimensions
541 // SkWebpCodec, the only one that supports subsets, requires even top/left boundaries.
542 const int w = SkAlign2(W / divisor);
543 const int h = SkAlign2(H / divisor);
544 SkIRect subset;
545 SkCodec::Options opts;
546 opts.fSubset = &subset;
547 SkBitmap subsetBm;
548 // We will reuse pixel memory from bitmap.
549 void* pixels = bitmap.getPixels();
550 // Keep track of left and top (for drawing subsetBm into canvas). We could use
551 // fscale * x and fscale * y, but we want integers such that the next subset will start
552 // where the last one ended. So we'll add decodeInfo.width() and height().
553 int left = 0;
554 for (int x = 0; x < W; x += w) {
555 int top = 0;
556 for (int y = 0; y < H; y+= h) {
557 // Do not make the subset go off the edge of the image.
558 const int preScaleW = SkTMin(w, W - x);
559 const int preScaleH = SkTMin(h, H - y);
560 subset.setXYWH(x, y, preScaleW, preScaleH);
561 // And fscale
562 // FIXME: Should we have a version of getScaledDimensions that takes a subset
563 // into account?
564 decodeInfo = decodeInfo.makeWH(
565 SkTMax(1, SkScalarRoundToInt(preScaleW * fscale)),
566 SkTMax(1, SkScalarRoundToInt(preScaleH * fscale)));
567 size_t rowBytes = decodeInfo.minRowBytes();
Leon Scroggins571b30f2017-07-11 17:35:31 +0000568 if (!subsetBm.installPixels(decodeInfo, pixels, rowBytes)) {
kjlubick2a42f482016-02-16 16:14:23 -0800569 SkDebugf("[terminated] Could not install pixels.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500570 return;
kjlubick2a42f482016-02-16 16:14:23 -0800571 }
572 const SkCodec::Result result = codec->getPixels(decodeInfo, pixels, rowBytes,
Leon Scroggins571b30f2017-07-11 17:35:31 +0000573 &opts);
kjlubick2a42f482016-02-16 16:14:23 -0800574 switch (result) {
575 case SkCodec::kSuccess:
576 case SkCodec::kIncompleteInput:
Leon Scroggins III674a1842017-07-06 12:26:09 -0400577 case SkCodec::kErrorInInput:
kjlubick2a42f482016-02-16 16:14:23 -0800578 SkDebugf("okay\n");
579 break;
580 case SkCodec::kInvalidConversion:
581 if (0 == (x|y)) {
582 // First subset is okay to return unimplemented.
583 SkDebugf("[terminated] Incompatible colortype conversion\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500584 return;
kjlubick2a42f482016-02-16 16:14:23 -0800585 }
586 // If the first subset succeeded, a later one should not fail.
587 // fall through to failure
588 case SkCodec::kUnimplemented:
589 if (0 == (x|y)) {
590 // First subset is okay to return unimplemented.
591 SkDebugf("[terminated] subset codec not supported\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500592 return;
kjlubick2a42f482016-02-16 16:14:23 -0800593 }
594 // If the first subset succeeded, why would a later one fail?
595 // fall through to failure
596 default:
597 SkDebugf("[terminated] subset codec failed to decode (%d, %d, %d, %d) "
598 "with dimensions (%d x %d)\t error %d\n",
599 x, y, decodeInfo.width(), decodeInfo.height(),
600 W, H, result);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500601 return;
kjlubick2a42f482016-02-16 16:14:23 -0800602 }
603 // translate by the scaled height.
604 top += decodeInfo.height();
605 }
606 // translate by the scaled width.
607 left += decodeInfo.width();
608 }
609 SkDebugf("[terminated] Success!\n");
610 break;
611 }
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500612 case 4: { //kAnimated_Mode
613 std::vector<SkCodec::FrameInfo> frameInfos = codec->getFrameInfo();
614 if (frameInfos.size() == 0) {
615 SkDebugf("[terminated] Not an animated image\n");
616 break;
617 }
618
619 for (size_t i = 0; i < frameInfos.size(); i++) {
620 options.fFrameIndex = i;
621 auto result = codec->startIncrementalDecode(decodeInfo, bitmap.getPixels(),
622 bitmap.rowBytes(), &options);
623 if (SkCodec::kSuccess != result) {
624 SkDebugf("[terminated] failed to start incremental decode "
625 "in frame %d with error %d\n", i, result);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500626 return;
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500627 }
628
629 result = codec->incrementalDecode();
Leon Scroggins III674a1842017-07-06 12:26:09 -0400630 if (result == SkCodec::kIncompleteInput || result == SkCodec::kErrorInInput) {
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500631 SkDebugf("okay\n");
632 // Frames beyond this one will not decode.
633 break;
634 }
635 if (result == SkCodec::kSuccess) {
636 SkDebugf("okay - decoded frame %d\n", i);
637 } else {
638 SkDebugf("[terminated] incremental decode failed with "
639 "error %d\n", result);
Kevin Lubickf80f1152017-01-06 08:26:56 -0500640 return;
Leon Scroggins IIIc5a83662016-12-08 09:07:56 -0500641 }
642 }
643 SkDebugf("[terminated] Success!\n");
644 break;
645 }
kjlubickdba57342016-01-21 05:03:28 -0800646 default:
kjlubick2a42f482016-02-16 16:14:23 -0800647 SkDebugf("[terminated] Mode not implemented yet\n");
kjlubickdba57342016-01-21 05:03:28 -0800648 }
649
650 dump_png(bitmap);
mtklein65e58242016-01-13 12:57:57 -0800651}
652
Kevin Lubickf80f1152017-01-06 08:26:56 -0500653static void fuzz_skp(sk_sp<SkData> bytes) {
Kevin Lubick09757b22017-12-12 12:52:39 -0500654 SkReadBuffer buf(bytes->data(), bytes->size());
kjlubickdba57342016-01-21 05:03:28 -0800655 SkDebugf("Decoding\n");
Cary Clarkefd99cc2018-06-11 16:25:43 -0400656 sk_sp<SkPicture> pic(SkPicturePriv::MakeFromBuffer(buf));
kjlubickdba57342016-01-21 05:03:28 -0800657 if (!pic) {
kjlubick47d158e2016-02-01 08:23:50 -0800658 SkDebugf("[terminated] Couldn't decode as a picture.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500659 return;
kjlubickdba57342016-01-21 05:03:28 -0800660 }
661 SkDebugf("Rendering\n");
662 SkBitmap bitmap;
663 if (!FLAGS_dump.isEmpty()) {
664 SkIRect size = pic->cullRect().roundOut();
665 bitmap.allocN32Pixels(size.width(), size.height());
666 }
667 SkCanvas canvas(bitmap);
668 canvas.drawPicture(pic);
kjlubick47d158e2016-02-01 08:23:50 -0800669 SkDebugf("[terminated] Success! Decoded and rendered an SkPicture!\n");
kjlubickdba57342016-01-21 05:03:28 -0800670 dump_png(bitmap);
kjlubickdba57342016-01-21 05:03:28 -0800671}
mtklein65e58242016-01-13 12:57:57 -0800672
Kevin Lubickf80f1152017-01-06 08:26:56 -0500673static void fuzz_color_deserialize(sk_sp<SkData> bytes) {
kjlubick3e3c1a52016-06-23 10:49:27 -0700674 sk_sp<SkColorSpace> space(SkColorSpace::Deserialize(bytes->data(), bytes->size()));
675 if (!space) {
676 SkDebugf("[terminated] Couldn't deserialize Colorspace.\n");
Kevin Lubickf80f1152017-01-06 08:26:56 -0500677 return;
kjlubick3e3c1a52016-06-23 10:49:27 -0700678 }
679 SkDebugf("[terminated] Success! deserialized Colorspace.\n");
kjlubick3e3c1a52016-06-23 10:49:27 -0700680}
681
Kevin Lubickf034d112018-02-08 14:31:24 -0500682void FuzzPathDeserialize(SkReadBuffer& buf);
Kevin Lubickc9d28292017-11-09 08:12:56 -0500683
Kevin Lubickf034d112018-02-08 14:31:24 -0500684static void fuzz_path_deserialize(sk_sp<SkData> bytes) {
685 SkReadBuffer buf(bytes->data(), bytes->size());
686 FuzzPathDeserialize(buf);
687 SkDebugf("[terminated] path_deserialize didn't crash!\n");
Kevin Lubickf04c50a2017-01-06 13:48:19 -0500688}
689
Kevin Lubick2541edf2018-01-11 10:27:14 -0500690bool FuzzRegionDeserialize(sk_sp<SkData> bytes);
691
Kevin Lubickedee1ae2017-02-20 17:47:18 -0500692static void fuzz_region_deserialize(sk_sp<SkData> bytes) {
Kevin Lubick2541edf2018-01-11 10:27:14 -0500693 if (!FuzzRegionDeserialize(bytes)) {
Kevin Lubickedee1ae2017-02-20 17:47:18 -0500694 SkDebugf("[terminated] Couldn't initialize SkRegion.\n");
695 return;
696 }
Kevin Lubickedee1ae2017-02-20 17:47:18 -0500697 SkDebugf("[terminated] Success! Initialized SkRegion.\n");
698}
699
Kevin Lubickf034d112018-02-08 14:31:24 -0500700void FuzzTextBlobDeserialize(SkReadBuffer& buf);
701
Kevin Lubickd46e85f2017-11-21 17:13:35 -0500702static void fuzz_textblob_deserialize(sk_sp<SkData> bytes) {
Mike Reedfadbfcd2017-12-06 16:09:20 -0500703 SkReadBuffer buf(bytes->data(), bytes->size());
Kevin Lubickf034d112018-02-08 14:31:24 -0500704 FuzzTextBlobDeserialize(buf);
705 SkDebugf("[terminated] textblob didn't crash!\n");
Kevin Lubickd46e85f2017-11-21 17:13:35 -0500706}
707
Kevin Lubick2541edf2018-01-11 10:27:14 -0500708void FuzzRegionSetPath(Fuzz* fuzz);
709
710static void fuzz_region_set_path(sk_sp<SkData> bytes) {
711 Fuzz fuzz(bytes);
712 FuzzRegionSetPath(&fuzz);
713 SkDebugf("[terminated] region_set_path didn't crash!\n");
714}
715
Kevin Lubickf034d112018-02-08 14:31:24 -0500716void FuzzImageFilterDeserialize(sk_sp<SkData> bytes);
717
Herb Derbya839fc02017-03-16 12:30:43 -0400718static void fuzz_filter_fuzz(sk_sp<SkData> bytes) {
Kevin Lubickf034d112018-02-08 14:31:24 -0500719 FuzzImageFilterDeserialize(bytes);
720 SkDebugf("[terminated] filter_fuzz didn't crash!\n");
Herb Derbya839fc02017-03-16 12:30:43 -0400721}
722
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400723bool FuzzSKSL2GLSL(sk_sp<SkData> bytes);
724
Kevin Lubickf80f1152017-01-06 08:26:56 -0500725static void fuzz_sksl2glsl(sk_sp<SkData> bytes) {
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400726 if (FuzzSKSL2GLSL(bytes)) {
727 SkDebugf("[terminated] Success! Compiled input to GLSL.\n");
728 } else {
729 SkDebugf("[terminated] Could not compile input to GLSL.\n");
kjlubicke7195772016-10-18 10:06:24 -0700730 }
kjlubicke7195772016-10-18 10:06:24 -0700731}
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400732
733bool FuzzSKSL2SPIRV(sk_sp<SkData> bytes);
734
735static void fuzz_sksl2spirv(sk_sp<SkData> bytes) {
736 if (FuzzSKSL2SPIRV(bytes)) {
737 SkDebugf("[terminated] Success! Compiled input to SPIRV.\n");
738 } else {
739 SkDebugf("[terminated] Could not compile input to SPIRV.\n");
740 }
741}
742
743bool FuzzSKSL2Metal(sk_sp<SkData> bytes);
744
745static void fuzz_sksl2metal(sk_sp<SkData> bytes) {
746 if (FuzzSKSL2Metal(bytes)) {
Kevin Lubick39cbe462019-03-11 15:38:00 -0400747 SkDebugf("[terminated] Success! Compiled input to Metal.\n");
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400748 } else {
Kevin Lubick39cbe462019-03-11 15:38:00 -0400749 SkDebugf("[terminated] Could not compile input to Metal.\n");
Kevin Lubicke9c1ce82019-03-11 11:09:40 -0400750 }
751}
Kevin Lubick0f0a7102019-03-18 16:20:55 -0400752
753bool FuzzSKSL2Pipeline(sk_sp<SkData> bytes);
754
755static void fuzz_sksl2pipeline(sk_sp<SkData> bytes) {
756 if (FuzzSKSL2Pipeline(bytes)) {
757 SkDebugf("[terminated] Success! Compiled input to pipeline stage.\n");
758 } else {
759 SkDebugf("[terminated] Could not compile input to pipeline stage.\n");
760 }
761}