#include "ndpi_api.h"

#include "ndpi_private.h"

#include "fuzz_common_code.h"

#include <stdint.h>

#include <stdio.h>

struct ndpi_tcphdr tcph;

struct ndpi_iphdr iph;

struct ndpi_ipv6hdr iphv6;

struct ndpi_detection_module_struct *ndpi_struct = NULL;

struct ndpi_flow_struct *ndpi_flow = NULL;

static char *path = NULL;

int LLVMFuzzerInitialize(int *argc, char ***argv) {

  (void)argc;

  path = dirname(strdup(*argv[0])); /* No errors; no free! */

  return 0;

}

int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {

  struct ndpi_packet_struct *packet;

  int is_ipv6;

  if (ndpi_struct == NULL) {

    fuzz_init_detection_module(&ndpi_struct, NULL, path);

    ndpi_flow = ndpi_calloc(1, sizeof(struct ndpi_flow_struct));

  }

  if(size == 0)

    return -1;

  fuzz_set_alloc_callbacks_and_seed(size);

  packet = &ndpi_struct->packet;

  packet->payload = data;

  packet->payload_packet_len = size;

  is_ipv6 = data[size - 1] % 5 ? 1 : 0; /* "Random" ipv4 vs ipv6 */

  packet->iphv6 = is_ipv6 ? &iphv6 : NULL;

  packet->iph = is_ipv6 ? NULL : &iph;

  packet->tcp = &tcph;

  memset(ndpi_flow, 0, sizeof(struct ndpi_flow_struct));

  ndpi_hostname_sni_set(ndpi_flow, (const u_int8_t *)"doh.opendns.com",

                        strlen("doh.opendns.com"), NDPI_HOSTNAME_NORM_ALL);

  ndpi_flow->core.detected_protocol_stack[0] = NDPI_PROTOCOL_TLS;

  processCertificateElements(ndpi_struct, ndpi_flow, 0, size);

  ndpi_free_flow_data(ndpi_flow);

  return 0;

}