#include "ndpi_api.h"
#include "ndpi_private.h"
#include "fuzz_common_code.h"

static struct ndpi_detection_module_struct *ndpi_struct = NULL;
static struct ndpi_flow_struct ndpi_flow;
struct ndpi_iphdr iph;
#ifdef STUN_TCP
struct ndpi_tcphdr tcph;
#else
struct ndpi_udphdr udph;
#endif

extern int is_stun(struct ndpi_detection_module_struct *ndpi_struct,
                   struct ndpi_flow_struct *flow,
                   u_int16_t *app_proto);


int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
  u_int16_t app_proto; /* unused */
  struct ndpi_packet_struct *packet;

  if (ndpi_struct == NULL) {
    fuzz_init_detection_module(&ndpi_struct, NULL);
  }

  packet = &ndpi_struct->packet;
  packet->payload = data;
  packet->payload_packet_len = size;
#ifndef STUN_TCP
  packet->udp = &udph;
#else
  packet->tcp = &tcph;
#endif
  packet->iph = &iph; /* IPv4 only */

  is_stun(ndpi_struct, &ndpi_flow, &app_proto);
  return 0;
}