libpsl/fuzz/libpsl_load_dafsa_fuzzer.c

/*
 * Copyright(c) 2017-2022 Tim Ruehsen
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 *
 * This file is part of libpsl.
 */

#include <config.h>

#include <assert.h> /* assert */

#ifdef HAVE_STDINT_H
#include <stdint.h> /* uint8_t */
#elif defined (_MSC_VER)
typedef unsigned __int8 uint8_t;
#endif

#include <stdlib.h> /* malloc, free */
#include <string.h> /* memcpy */
#include <stdio.h> /* fmemopen */

#include "libpsl.h"
#include "fuzzer.h"

int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
{
#ifdef HAVE_FMEMOPEN
	FILE *fp;
	psl_ctx_t *psl;
	char *in = (char *) malloc(size + 16);

	assert(in != NULL);

	/* create a valid DAFSA input file */
	memcpy(in, ".DAFSA@PSL_0   \n", 16);
	memcpy(in + 16, data, size);

	fp = fmemopen(in, size + 16, "r");
	assert(fp != NULL);

	psl = psl_load_fp(fp);

	psl_is_public_suffix(NULL, NULL);
	psl_is_public_suffix(psl, ".ü.com");
	psl_suffix_wildcard_count(psl);
	psl_suffix_exception_count(psl);
	psl_suffix_count(psl);

	psl_free(psl);
	fclose(fp);

	psl = psl_latest(NULL);
	psl_free(psl);

	free(in);
#endif

	return 0;
}