File: mosaic_fuzzer.cc

package info (click to toggle)
vips 8.17.3-2
  • links: PTS
  • area: main
  • in suites: forky, sid
  • size: 52,228 kB
  • sloc: ansic: 169,684; cpp: 12,156; python: 4,887; sh: 733; perl: 40; makefile: 25; javascript: 6
file content (76 lines) | stat: -rw-r--r-- 1,422 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#include <cstring>
#include <vips/vips.h>

#ifdef __GNUC__
#define PACK(...) __VA_ARGS__ __attribute__((__packed__))
#elif defined(_MSC_VER)
#define PACK(...) __pragma(pack(push, 1)) __VA_ARGS__ __pragma(pack(pop))
#else
#define PACK(...) __VA_ARGS__
#endif

PACK(struct mosaic_opt {
	guint8 dir : 1;
	guint16 xref;
	guint16 yref;
	guint16 xsec;
	guint16 ysec;
});

extern "C" int
LLVMFuzzerInitialize(int *argc, char ***argv)
{
	if (VIPS_INIT(*argv[0]))
		return -1;

	vips_concurrency_set(1);
	return 0;
}

extern "C" int
LLVMFuzzerTestOneInput(const guint8 *data, size_t size)
{
	VipsImage *ref, *sec, *out;
	mosaic_opt opt = {};
	double d;

	if (size < sizeof(mosaic_opt))
		return 0;

	/* The tail of `data` is treated as mosaic configuration
	 */
	size -= sizeof(mosaic_opt);
	memcpy(&opt, data + size, sizeof(mosaic_opt));

	/* Remainder of input is the image
	 */
	if (!(ref = vips_image_new_from_buffer(data, size, "", nullptr)))
		return 0;

	if (ref->Xsize > 100 ||
		ref->Ysize > 100 ||
		ref->Bands > 4) {
		g_object_unref(ref);
		return 0;
	}

	if (vips_rot180(ref, &sec, nullptr)) {
		g_object_unref(ref);
		return 0;
	}

	if (vips_mosaic(ref, sec, &out, (VipsDirection) opt.dir,
			opt.xref, opt.yref, opt.xsec, opt.ysec, nullptr)) {
		g_object_unref(sec);
		g_object_unref(ref);
		return 0;
	}

	vips_max(out, &d, nullptr);

	g_object_unref(out);
	g_object_unref(sec);
	g_object_unref(ref);

	return 0;
}