fuzz/fuzz_filecfg_categories.c


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

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

int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
  struct ndpi_detection_module_struct *ndpi_struct;
  FILE *fd;
  NDPI_PROTOCOL_BITMASK all;

  /* To allow memory allocation failures */
  fuzz_set_alloc_callbacks_and_seed(size);

  ndpi_struct = ndpi_init_detection_module(NULL);
  NDPI_BITMASK_SET_ALL(all);
  ndpi_set_protocol_detection_bitmask2(ndpi_struct, &all);

  ndpi_set_config_u64(ndpi_struct, NULL, "log.level", 3);
  ndpi_set_config_u64(ndpi_struct, "all", "log", 1);

  fd = buffer_to_file(data, size);
  load_categories_file_fd(ndpi_struct, fd, NULL);
  if(fd)
    fclose(fd);

  ndpi_exit_detection_module(ndpi_struct);
  return 0;
}