/*
   * Copyright (C) 2015, Wang Nan <wangnan0@huawei.com>
   * Copyright (C) 2015, Huawei Inc.
   */
  #ifndef __BPF_LOADER_H
  #define __BPF_LOADER_H
  
  #include <linux/compiler.h>
  #include <linux/err.h>
  #include <string.h>

  #include <bpf/libbpf.h>

  #include "probe-event.h"

  #include "evlist.h"

  #include "debug.h"

  enum bpf_loader_errno {
  	__BPF_LOADER_ERRNO__START = __LIBBPF_ERRNO__START - 100,
  	/* Invalid config string */
  	BPF_LOADER_ERRNO__CONFIG = __BPF_LOADER_ERRNO__START,
  	BPF_LOADER_ERRNO__GROUP,	/* Invalid group name */
  	BPF_LOADER_ERRNO__EVENTNAME,	/* Event name is missing */
  	BPF_LOADER_ERRNO__INTERNAL,	/* BPF loader internal error */
  	BPF_LOADER_ERRNO__COMPILE,	/* Error when compiling BPF scriptlet */

  	BPF_LOADER_ERRNO__PROGCONF_TERM,/* Invalid program config term in config string */

  	BPF_LOADER_ERRNO__PROLOGUE,	/* Failed to generate prologue */
  	BPF_LOADER_ERRNO__PROLOGUE2BIG,	/* Prologue too big for program */
  	BPF_LOADER_ERRNO__PROLOGUEOOB,	/* Offset out of bound for prologue */

  	BPF_LOADER_ERRNO__OBJCONF_OPT,	/* Invalid object config option */
  	BPF_LOADER_ERRNO__OBJCONF_CONF,	/* Config value not set (lost '=')) */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_OPT,	/* Invalid object map config option */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_NOTEXIST,	/* Target map not exist */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_VALUE,	/* Incorrect value type for map */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_TYPE,	/* Incorrect map type */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_KEYSIZE,	/* Incorrect map key size */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_VALUESIZE,/* Incorrect map value size */

  	BPF_LOADER_ERRNO__OBJCONF_MAP_NOEVT,	/* Event not found for map setting */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_MAPSIZE,	/* Invalid map size for event setting */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_EVTDIM,	/* Event dimension too large */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_EVTINH,	/* Doesn't support inherit event */
  	BPF_LOADER_ERRNO__OBJCONF_MAP_EVTTYPE,	/* Wrong event type for map */

  	BPF_LOADER_ERRNO__OBJCONF_MAP_IDX2BIG,	/* Index too large */

  	__BPF_LOADER_ERRNO__END,
  };

  struct bpf_object;

  struct parse_events_term;

  #define PERF_BPF_PROBE_GROUP "perf_bpf_probe"

  typedef int (*bpf_prog_iter_callback_t)(const char *group, const char *event,

  					int fd, void *arg);

  #ifdef HAVE_LIBBPF_SUPPORT

  struct bpf_object *bpf__prepare_load(const char *filename, bool source);

  int bpf__strerror_prepare_load(const char *filename, bool source,
  			       int err, char *buf, size_t size);

  struct bpf_object *bpf__prepare_load_buffer(void *obj_buf, size_t obj_buf_sz,
  					    const char *name);

  void bpf__clear(void);

  
  int bpf__probe(struct bpf_object *obj);
  int bpf__unprobe(struct bpf_object *obj);
  int bpf__strerror_probe(struct bpf_object *obj, int err,
  			char *buf, size_t size);

  int bpf__load(struct bpf_object *obj);
  int bpf__strerror_load(struct bpf_object *obj, int err,
  		       char *buf, size_t size);

  int bpf__foreach_event(struct bpf_object *obj,
  		       bpf_prog_iter_callback_t func, void *arg);

  
  int bpf__config_obj(struct bpf_object *obj, struct parse_events_term *term,
  		    struct perf_evlist *evlist, int *error_pos);
  int bpf__strerror_config_obj(struct bpf_object *obj,
  			     struct parse_events_term *term,
  			     struct perf_evlist *evlist,
  			     int *error_pos, int err, char *buf,
  			     size_t size);

  int bpf__apply_obj_config(void);
  int bpf__strerror_apply_obj_config(int err, char *buf, size_t size);

  
  int bpf__setup_stdout(struct perf_evlist *evlist);
  int bpf__strerror_setup_stdout(struct perf_evlist *evlist, int err,
  			       char *buf, size_t size);

  #else

  #include <errno.h>

  static inline struct bpf_object *

  bpf__prepare_load(const char *filename __maybe_unused,
  		  bool source __maybe_unused)

  {
  	pr_debug("ERROR: eBPF object loading is disabled during compiling.
  ");
  	return ERR_PTR(-ENOTSUP);
  }

  static inline struct bpf_object *
  bpf__prepare_load_buffer(void *obj_buf __maybe_unused,
  					   size_t obj_buf_sz __maybe_unused)
  {
  	return ERR_PTR(-ENOTSUP);
  }

  static inline void bpf__clear(void) { }

  
  static inline int bpf__probe(struct bpf_object *obj __maybe_unused) { return 0;}
  static inline int bpf__unprobe(struct bpf_object *obj __maybe_unused) { return 0;}

  static inline int bpf__load(struct bpf_object *obj __maybe_unused) { return 0; }

  
  static inline int

  bpf__foreach_event(struct bpf_object *obj __maybe_unused,
  		   bpf_prog_iter_callback_t func __maybe_unused,
  		   void *arg __maybe_unused)

  {
  	return 0;
  }
  
  static inline int

  bpf__config_obj(struct bpf_object *obj __maybe_unused,
  		struct parse_events_term *term __maybe_unused,
  		struct perf_evlist *evlist __maybe_unused,
  		int *error_pos __maybe_unused)
  {
  	return 0;
  }
  
  static inline int

  bpf__apply_obj_config(void)
  {
  	return 0;
  }
  
  static inline int

  bpf__setup_stdout(struct perf_evlist *evlist __maybe_unused)
  {
  	return 0;
  }
  
  static inline int

  __bpf_strerror(char *buf, size_t size)
  {
  	if (!size)
  		return 0;
  	strncpy(buf,
  		"ERROR: eBPF object loading is disabled during compiling.
  ",
  		size);
  	buf[size - 1] = '\0';
  	return 0;
  }

  static inline
  int bpf__strerror_prepare_load(const char *filename __maybe_unused,
  			       bool source __maybe_unused,
  			       int err __maybe_unused,
  			       char *buf, size_t size)
  {
  	return __bpf_strerror(buf, size);
  }

  static inline int
  bpf__strerror_probe(struct bpf_object *obj __maybe_unused,
  		    int err __maybe_unused,
  		    char *buf, size_t size)
  {
  	return __bpf_strerror(buf, size);
  }

  
  static inline int bpf__strerror_load(struct bpf_object *obj __maybe_unused,
  				     int err __maybe_unused,
  				     char *buf, size_t size)
  {
  	return __bpf_strerror(buf, size);
  }

  
  static inline int
  bpf__strerror_config_obj(struct bpf_object *obj __maybe_unused,
  			 struct parse_events_term *term __maybe_unused,
  			 struct perf_evlist *evlist __maybe_unused,
  			 int *error_pos __maybe_unused,
  			 int err __maybe_unused,
  			 char *buf, size_t size)
  {
  	return __bpf_strerror(buf, size);
  }

  
  static inline int
  bpf__strerror_apply_obj_config(int err __maybe_unused,
  			       char *buf, size_t size)
  {
  	return __bpf_strerror(buf, size);
  }

  
  static inline int
  bpf__strerror_setup_stdout(struct perf_evlist *evlist __maybe_unused,
  			   int err __maybe_unused, char *buf,
  			   size_t size)
  {
  	return __bpf_strerror(buf, size);
  }

  #endif
  #endif