1/*	$NetBSD: isc_lex_getmastertoken.c,v 1.2 2024/02/21 22:51:58 christos Exp $	*/
2
3/*
4 * Copyright (C) Internet Systems Consortium, Inc. ("ISC")
5 *
6 * SPDX-License-Identifier: MPL-2.0
7 *
8 * This Source Code Form is subject to the terms of the Mozilla Public
9 * License, v. 2.0. If a copy of the MPL was not distributed with this
10 * file, you can obtain one at https://mozilla.org/MPL/2.0/.
11 *
12 * See the COPYRIGHT file distributed with this work for additional
13 * information regarding copyright ownership.
14 */
15
16#include <stddef.h>
17#include <stdint.h>
18
19#include <isc/buffer.h>
20#include <isc/lex.h>
21#include <isc/mem.h>
22#include <isc/string.h>
23#include <isc/util.h>
24
25#include "fuzz.h"
26
27bool debug = false;
28
29int
30LLVMFuzzerInitialize(int *argc __attribute__((unused)),
31		     char ***argv __attribute__((unused)));
32
33int
34LLVMFuzzerTestOneInput(const uint8_t *data, size_t size);
35
36static isc_mem_t *mctx = NULL;
37static isc_lex_t *lex = NULL;
38
39int
40LLVMFuzzerInitialize(int *argc __attribute__((unused)),
41		     char ***argv __attribute__((unused))) {
42	isc_result_t result;
43
44	isc_mem_create(&mctx);
45
46	result = isc_lex_create(mctx, 1024, &lex);
47	REQUIRE(result == ISC_R_SUCCESS);
48
49	return (0);
50}
51
52int
53LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
54	isc_buffer_t buf;
55	isc_result_t result;
56	isc_token_t token;
57	isc_tokentype_t expect;
58	bool eol;
59
60	if (size < sizeof(expect) + sizeof(eol)) {
61		return (0);
62	}
63
64	(void)memmove(&expect, data, sizeof(expect));
65	data += sizeof(expect);
66	size -= sizeof(expect);
67
68	eol = *data != 0;
69	data += 1;
70	size -= 1;
71
72	isc_buffer_constinit(&buf, data, size);
73	isc_buffer_add(&buf, size);
74	isc_buffer_setactive(&buf, size);
75
76	CHECK(isc_lex_openbuffer(lex, &buf));
77
78	do {
79		result = isc_lex_getmastertoken(lex, &token, expect, eol);
80	} while (result == ISC_R_SUCCESS && token.type != isc_tokentype_eof);
81
82	return (0);
83}
84