// SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)

/*
 * Common eBPF ELF object loading operations.
 *
 * Copyright (C) 2013-2015 Alexei Starovoitov <ast@kernel.org>
 * Copyright (C) 2015 Wang Nan <wangnan0@huawei.com>
 * Copyright (C) 2015 Huawei Inc.
 * Copyright (C) 2017 Nicira, Inc.
 * Copyright (C) 2019 Isovalent, Inc.
 */

#ifndef _GNU_SOURCE
#define _GNU_SOURCE
#endif
#include <stdlib.h>
#include <stdio.h>
#include <stdarg.h>
#include <libgen.h>
#include <inttypes.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <asm/unistd.h>
#include <linux/err.h>
#include <linux/kernel.h>
#include <linux/bpf.h>
#include <linux/btf.h>
#include <linux/filter.h>
#include <linux/list.h>
#include <linux/limits.h>
#include <linux/perf_event.h>
#include <linux/ring_buffer.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/vfs.h>
#include <tools/libc_compat.h>
#include <libelf.h>
#include <gelf.h>

#include "libbpf.h"
#include "bpf.h"
#include "btf.h"
#include "str_error.h"
#include "libbpf_internal.h"

#ifndef EM_BPF
#define EM_BPF 247
#endif

#ifndef BPF_FS_MAGIC
#define BPF_FS_MAGIC            0xcafe4a11
#endif

/* vsprintf() in __base_pr() uses nonliteral format string. It may break
 * compilation if user enables corresponding warning. Disable it explicitly.
 */
#pragma GCC diagnostic ignored "-Wformat-nonliteral"

#define __printf(a, b)  __attribute__((format(printf, a, b)))

static int __base_pr(enum libbpf_print_level level, const char *format,
                     va_list args)
{
        if (level == LIBBPF_DEBUG)
                return 0;

        return vfprintf(stderr, format, args);
}

static libbpf_print_fn_t __libbpf_pr = __base_pr;

void libbpf_set_print(libbpf_print_fn_t fn)
{
        __libbpf_pr = fn;
}

__printf(2, 3)
void libbpf_print(enum libbpf_print_level level, const char *format, ...)
{
        va_list args;

        if (!__libbpf_pr)
                return;

        va_start(args, format);
        __libbpf_pr(level, format, args);
        va_end(args);
}

#define STRERR_BUFSIZE  128

#define CHECK_ERR(action, err, out) do {        \
        err = action;                   \
        if (err)                        \
                goto out;               \
} while(0)


/* Copied from tools/perf/util/util.h */
#ifndef zfree
# define zfree(ptr) ({ free(*ptr); *ptr = NULL; })
#endif

#ifndef zclose
# define zclose(fd) ({                  \
        int ___err = 0;                 \
        if ((fd) >= 0)                  \
                ___err = close((fd));   \
        fd = -1;                        \
        ___err; })
#endif

#ifdef HAVE_LIBELF_MMAP_SUPPORT
# define LIBBPF_ELF_C_READ_MMAP ELF_C_READ_MMAP
#else
# define LIBBPF_ELF_C_READ_MMAP ELF_C_READ
#endif

static inline __u64 ptr_to_u64(const void *ptr)
{
        return (__u64) (unsigned long) ptr;
}

struct bpf_capabilities {
        /* v4.14: kernel support for program & map names. */
        __u32 name:1;
        /* v5.2: kernel support for global data sections. */
        __u32 global_data:1;
        /* BTF_KIND_FUNC and BTF_KIND_FUNC_PROTO support */
        __u32 btf_func:1;
        /* BTF_KIND_VAR and BTF_KIND_DATASEC support */
        __u32 btf_datasec:1;
};

/*
 * bpf_prog should be a better name but it has been used in
 * linux/filter.h.
 */
struct bpf_program {
        /* Index in elf obj file, for relocation use. */
        int idx;
        char *name;
        int prog_ifindex;
        char *section_name;
        /* section_name with / replaced by _; makes recursive pinning
         * in bpf_object__pin_programs easier
         */
        char *pin_name;
        struct bpf_insn *insns;
        size_t insns_cnt, main_prog_cnt;
        enum bpf_prog_type type;

        struct reloc_desc {
                enum {
                        RELO_LD64,
                        RELO_CALL,
                        RELO_DATA,
                } type;
                int insn_idx;
                union {
                        int map_idx;
                        int text_off;
                };
        } *reloc_desc;
        int nr_reloc;
        int log_level;

        struct {
                int nr;
                int *fds;
        } instances;
        bpf_program_prep_t preprocessor;

        struct bpf_object *obj;
        void *priv;
        bpf_program_clear_priv_t clear_priv;

        enum bpf_attach_type expected_attach_type;
        int btf_fd;
        void *func_info;
        __u32 func_info_rec_size;
        __u32 func_info_cnt;

        struct bpf_capabilities *caps;

        void *line_info;
        __u32 line_info_rec_size;
        __u32 line_info_cnt;
};

enum libbpf_map_type {
        LIBBPF_MAP_UNSPEC,
        LIBBPF_MAP_DATA,
        LIBBPF_MAP_BSS,
        LIBBPF_MAP_RODATA,
};

static const char * const libbpf_type_to_btf_name[] = {
        [LIBBPF_MAP_DATA]       = ".data",
        [LIBBPF_MAP_BSS]        = ".bss",
        [LIBBPF_MAP_RODATA]     = ".rodata",
};

struct bpf_map {
        int fd;
        char *name;
        size_t offset;
        int map_ifindex;
        int inner_map_fd;
        struct bpf_map_def def;
        __u32 btf_key_type_id;
        __u32 btf_value_type_id;
        void *priv;
        bpf_map_clear_priv_t clear_priv;
        enum libbpf_map_type libbpf_type;
};

struct bpf_secdata {
        void *rodata;
        void *data;
};

static LIST_HEAD(bpf_objects_list);

struct bpf_object {
        char name[BPF_OBJ_NAME_LEN];
        char license[64];
        __u32 kern_version;

        struct bpf_program *programs;
        size_t nr_programs;
        struct bpf_map *maps;
        size_t nr_maps;
        struct bpf_secdata sections;

        bool loaded;
        bool has_pseudo_calls;

        /*
         * Information when doing elf related work. Only valid if fd
         * is valid.
         */
        struct {
                int fd;
                void *obj_buf;
                size_t obj_buf_sz;
                Elf *elf;
                GElf_Ehdr ehdr;
                Elf_Data *symbols;
                Elf_Data *data;
                Elf_Data *rodata;
                Elf_Data *bss;
                size_t strtabidx;
                struct {
                        GElf_Shdr shdr;
                        Elf_Data *data;
                } *reloc;
                int nr_reloc;
                int maps_shndx;
                int text_shndx;
                int data_shndx;
                int rodata_shndx;
                int bss_shndx;
        } efile;
        /*
         * All loaded bpf_object is linked in a list, which is
         * hidden to caller. bpf_objects__<func> handlers deal with
         * all objects.
         */
        struct list_head list;

        struct btf *btf;
        struct btf_ext *btf_ext;

        void *priv;
        bpf_object_clear_priv_t clear_priv;

        struct bpf_capabilities caps;

        char path[];
};
#define obj_elf_valid(o)        ((o)->efile.elf)

void bpf_program__unload(struct bpf_program *prog)
{
        int i;

        if (!prog)
                return;

        /*
         * If the object is opened but the program was never loaded,
         * it is possible that prog->instances.nr == -1.
         */
        if (prog->instances.nr > 0) {
                for (i = 0; i < prog->instances.nr; i++)
                        zclose(prog->instances.fds[i]);
        } else if (prog->instances.nr != -1) {
                pr_warning("Internal error: instances.nr is %d\n",
                           prog->instances.nr);
        }

        prog->instances.nr = -1;
        zfree(&prog->instances.fds);

        zclose(prog->btf_fd);
        zfree(&prog->func_info);
        zfree(&prog->line_info);
}

static void bpf_program__exit(struct bpf_program *prog)
{
        if (!prog)
                return;

        if (prog->clear_priv)
                prog->clear_priv(prog, prog->priv);

        prog->priv = NULL;
        prog->clear_priv = NULL;

        bpf_program__unload(prog);
        zfree(&prog->name);
        zfree(&prog->section_name);
        zfree(&prog->pin_name);
        zfree(&prog->insns);
        zfree(&prog->reloc_desc);

        prog->nr_reloc = 0;
        prog->insns_cnt = 0;
        prog->idx = -1;
}

static char *__bpf_program__pin_name(struct bpf_program *prog)
{
        char *name, *p;

        name = p = strdup(prog->section_name);
        while ((p = strchr(p, '/')))
                *p = '_';

        return name;
}

static int
bpf_program__init(void *data, size_t size, char *section_name, int idx,
                  struct bpf_program *prog)
{
        if (size < sizeof(struct bpf_insn)) {
                pr_warning("corrupted section '%s'\n", section_name);
                return -EINVAL;
        }

        memset(prog, 0, sizeof(*prog));

        prog->section_name = strdup(section_name);
        if (!prog->section_name) {
                pr_warning("failed to alloc name for prog under section(%d) %s\n",
                           idx, section_name);
                goto errout;
        }

        prog->pin_name = __bpf_program__pin_name(prog);
        if (!prog->pin_name) {
                pr_warning("failed to alloc pin name for prog under section(%d) %s\n",
                           idx, section_name);
                goto errout;
        }

        prog->insns = malloc(size);
        if (!prog->insns) {
                pr_warning("failed to alloc insns for prog under section %s\n",
                           section_name);
                goto errout;
        }
        prog->insns_cnt = size / sizeof(struct bpf_insn);
        memcpy(prog->insns, data,
               prog->insns_cnt * sizeof(struct bpf_insn));
        prog->idx = idx;
        prog->instances.fds = NULL;
        prog->instances.nr = -1;
        prog->type = BPF_PROG_TYPE_UNSPEC;
        prog->btf_fd = -1;

        return 0;
errout:
        bpf_program__exit(prog);
        return -ENOMEM;
}

static int
bpf_object__add_program(struct bpf_object *obj, void *data, size_t size,
                        char *section_name, int idx)
{
        struct bpf_program prog, *progs;
        int nr_progs, err;

        err = bpf_program__init(data, size, section_name, idx, &prog);
        if (err)
                return err;

        prog.caps = &obj->caps;
        progs = obj->programs;
        nr_progs = obj->nr_programs;

        progs = reallocarray(progs, nr_progs + 1, sizeof(progs[0]));
        if (!progs) {
                /*
                 * In this case the original obj->programs
                 * is still valid, so don't need special treat for
                 * bpf_close_object().
                 */
                pr_warning("failed to alloc a new program under section '%s'\n",
                           section_name);
                bpf_program__exit(&prog);
                return -ENOMEM;
        }

        pr_debug("found program %s\n", prog.section_name);
        obj->programs = progs;
        obj->nr_programs = nr_progs + 1;
        prog.obj = obj;
        progs[nr_progs] = prog;
        return 0;
}

static int
bpf_object__init_prog_names(struct bpf_object *obj)
{
        Elf_Data *symbols = obj->efile.symbols;
        struct bpf_program *prog;
        size_t pi, si;

        for (pi = 0; pi < obj->nr_programs; pi++) {
                const char *name = NULL;

                prog = &obj->programs[pi];

                for (si = 0; si < symbols->d_size / sizeof(GElf_Sym) && !name;
                     si++) {
                        GElf_Sym sym;

                        if (!gelf_getsym(symbols, si, &sym))
                                continue;
                        if (sym.st_shndx != prog->idx)
                                continue;
                        if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL)
                                continue;

                        name = elf_strptr(obj->efile.elf,
                                          obj->efile.strtabidx,
                                          sym.st_name);
                        if (!name) {
                                pr_warning("failed to get sym name string for prog %s\n",
                                           prog->section_name);
                                return -LIBBPF_ERRNO__LIBELF;
                        }
                }

                if (!name && prog->idx == obj->efile.text_shndx)
                        name = ".text";

                if (!name) {
                        pr_warning("failed to find sym for prog %s\n",
                                   prog->section_name);
                        return -EINVAL;
                }

                prog->name = strdup(name);
                if (!prog->name) {
                        pr_warning("failed to allocate memory for prog sym %s\n",
                                   name);
                        return -ENOMEM;
                }
        }

        return 0;
}

static struct bpf_object *bpf_object__new(const char *path,
                                          void *obj_buf,
                                          size_t obj_buf_sz)
{
        struct bpf_object *obj;
        char *end;

        obj = calloc(1, sizeof(struct bpf_object) + strlen(path) + 1);
        if (!obj) {
                pr_warning("alloc memory failed for %s\n", path);
                return ERR_PTR(-ENOMEM);
        }

        strcpy(obj->path, path);
        /* Using basename() GNU version which doesn't modify arg. */
        strncpy(obj->name, basename((void *)path),
                sizeof(obj->name) - 1);
        end = strchr(obj->name, '.');
        if (end)
                *end = 0;

        obj->efile.fd = -1;
        /*
         * Caller of this function should also calls
         * bpf_object__elf_finish() after data collection to return
         * obj_buf to user. If not, we should duplicate the buffer to
         * avoid user freeing them before elf finish.
         */
        obj->efile.obj_buf = obj_buf;
        obj->efile.obj_buf_sz = obj_buf_sz;
        obj->efile.maps_shndx = -1;
        obj->efile.data_shndx = -1;
        obj->efile.rodata_shndx = -1;
        obj->efile.bss_shndx = -1;

        obj->loaded = false;

        INIT_LIST_HEAD(&obj->list);
        list_add(&obj->list, &bpf_objects_list);
        return obj;
}

static void bpf_object__elf_finish(struct bpf_object *obj)
{
        if (!obj_elf_valid(obj))
                return;

        if (obj->efile.elf) {
                elf_end(obj->efile.elf);
                obj->efile.elf = NULL;
        }
        obj->efile.symbols = NULL;
        obj->efile.data = NULL;
        obj->efile.rodata = NULL;
        obj->efile.bss = NULL;

        zfree(&obj->efile.reloc);
        obj->efile.nr_reloc = 0;
        zclose(obj->efile.fd);
        obj->efile.obj_buf = NULL;
        obj->efile.obj_buf_sz = 0;
}

static int bpf_object__elf_init(struct bpf_object *obj)
{
        int err = 0;
        GElf_Ehdr *ep;

        if (obj_elf_valid(obj)) {
                pr_warning("elf init: internal error\n");
                return -LIBBPF_ERRNO__LIBELF;
        }

        if (obj->efile.obj_buf_sz > 0) {
                /*
                 * obj_buf should have been validated by
                 * bpf_object__open_buffer().
                 */
                obj->efile.elf = elf_memory(obj->efile.obj_buf,
                                            obj->efile.obj_buf_sz);
        } else {
                obj->efile.fd = open(obj->path, O_RDONLY);
                if (obj->efile.fd < 0) {
                        char errmsg[STRERR_BUFSIZE];
                        char *cp = libbpf_strerror_r(errno, errmsg,
                                                     sizeof(errmsg));

                        pr_warning("failed to open %s: %s\n", obj->path, cp);
                        return -errno;
                }

                obj->efile.elf = elf_begin(obj->efile.fd,
                                LIBBPF_ELF_C_READ_MMAP,
                                NULL);
        }

        if (!obj->efile.elf) {
                pr_warning("failed to open %s as ELF file\n",
                                obj->path);
                err = -LIBBPF_ERRNO__LIBELF;
                goto errout;
        }

        if (!gelf_getehdr(obj->efile.elf, &obj->efile.ehdr)) {
                pr_warning("failed to get EHDR from %s\n",
                                obj->path);
                err = -LIBBPF_ERRNO__FORMAT;
                goto errout;
        }
        ep = &obj->efile.ehdr;

        /* Old LLVM set e_machine to EM_NONE */
        if ((ep->e_type != ET_REL) || (ep->e_machine && (ep->e_machine != EM_BPF))) {
                pr_warning("%s is not an eBPF object file\n",
                        obj->path);
                err = -LIBBPF_ERRNO__FORMAT;
                goto errout;
        }

        return 0;
errout:
        bpf_object__elf_finish(obj);
        return err;
}

static int
bpf_object__check_endianness(struct bpf_object *obj)
{
        static unsigned int const endian = 1;

        switch (obj->efile.ehdr.e_ident[EI_DATA]) {
        case ELFDATA2LSB:
                /* We are big endian, BPF obj is little endian. */
                if (*(unsigned char const *)&endian != 1)
                        goto mismatch;
                break;

        case ELFDATA2MSB:
                /* We are little endian, BPF obj is big endian. */
                if (*(unsigned char const *)&endian != 0)
                        goto mismatch;
                break;
        default:
                return -LIBBPF_ERRNO__ENDIAN;
        }

        return 0;

mismatch:
        pr_warning("Error: endianness mismatch.\n");
        return -LIBBPF_ERRNO__ENDIAN;
}

static int
bpf_object__init_license(struct bpf_object *obj,
                         void *data, size_t size)
{
        memcpy(obj->license, data,
               min(size, sizeof(obj->license) - 1));
        pr_debug("license of %s is %s\n", obj->path, obj->license);
        return 0;
}

static int
bpf_object__init_kversion(struct bpf_object *obj,
                          void *data, size_t size)
{
        __u32 kver;

        if (size != sizeof(kver)) {
                pr_warning("invalid kver section in %s\n", obj->path);
                return -LIBBPF_ERRNO__FORMAT;
        }
        memcpy(&kver, data, sizeof(kver));
        obj->kern_version = kver;
        pr_debug("kernel version of %s is %x\n", obj->path,
                 obj->kern_version);
        return 0;
}

static int compare_bpf_map(const void *_a, const void *_b)
{
        const struct bpf_map *a = _a;
        const struct bpf_map *b = _b;

        return a->offset - b->offset;
}

static bool bpf_map_type__is_map_in_map(enum bpf_map_type type)
{
        if (type == BPF_MAP_TYPE_ARRAY_OF_MAPS ||
            type == BPF_MAP_TYPE_HASH_OF_MAPS)
                return true;
        return false;
}

static int bpf_object_search_section_size(const struct bpf_object *obj,
                                          const char *name, size_t *d_size)
{
        const GElf_Ehdr *ep = &obj->efile.ehdr;
        Elf *elf = obj->efile.elf;
        Elf_Scn *scn = NULL;
        int idx = 0;

        while ((scn = elf_nextscn(elf, scn)) != NULL) {
                const char *sec_name;
                Elf_Data *data;
                GElf_Shdr sh;

                idx++;
                if (gelf_getshdr(scn, &sh) != &sh) {
                        pr_warning("failed to get section(%d) header from %s\n",
                                   idx, obj->path);
                        return -EIO;
                }

                sec_name = elf_strptr(elf, ep->e_shstrndx, sh.sh_name);
                if (!sec_name) {
                        pr_warning("failed to get section(%d) name from %s\n",
                                   idx, obj->path);
                        return -EIO;
                }

                if (strcmp(name, sec_name))
                        continue;

                data = elf_getdata(scn, 0);
                if (!data) {
                        pr_warning("failed to get section(%d) data from %s(%s)\n",
                                   idx, name, obj->path);
                        return -EIO;
                }

                *d_size = data->d_size;
                return 0;
        }

        return -ENOENT;
}

int bpf_object__section_size(const struct bpf_object *obj, const char *name,
                             __u32 *size)
{
        int ret = -ENOENT;
        size_t d_size;

        *size = 0;
        if (!name) {
                return -EINVAL;
        } else if (!strcmp(name, ".data")) {
                if (obj->efile.data)
                        *size = obj->efile.data->d_size;
        } else if (!strcmp(name, ".bss")) {
                if (obj->efile.bss)
                        *size = obj->efile.bss->d_size;
        } else if (!strcmp(name, ".rodata")) {
                if (obj->efile.rodata)
                        *size = obj->efile.rodata->d_size;
        } else {
                ret = bpf_object_search_section_size(obj, name, &d_size);
                if (!ret)
                        *size = d_size;
        }

        return *size ? 0 : ret;
}

int bpf_object__variable_offset(const struct bpf_object *obj, const char *name,
                                __u32 *off)
{
        Elf_Data *symbols = obj->efile.symbols;
        const char *sname;
        size_t si;

        if (!name || !off)
                return -EINVAL;

        for (si = 0; si < symbols->d_size / sizeof(GElf_Sym); si++) {
                GElf_Sym sym;

                if (!gelf_getsym(symbols, si, &sym))
                        continue;
                if (GELF_ST_BIND(sym.st_info) != STB_GLOBAL ||
                    GELF_ST_TYPE(sym.st_info) != STT_OBJECT)
                        continue;

                sname = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
                                   sym.st_name);
                if (!sname) {
                        pr_warning("failed to get sym name string for var %s\n",
                                   name);
                        return -EIO;
                }
                if (strcmp(name, sname) == 0) {
                        *off = sym.st_value;
                        return 0;
                }
        }

        return -ENOENT;
}

static bool bpf_object__has_maps(const struct bpf_object *obj)
{
        return obj->efile.maps_shndx >= 0 ||
               obj->efile.data_shndx >= 0 ||
               obj->efile.rodata_shndx >= 0 ||
               obj->efile.bss_shndx >= 0;
}

static int
bpf_object__init_internal_map(struct bpf_object *obj, struct bpf_map *map,
                              enum libbpf_map_type type, Elf_Data *data,
                              void **data_buff)
{
        struct bpf_map_def *def = &map->def;
        char map_name[BPF_OBJ_NAME_LEN];

        map->libbpf_type = type;
        map->offset = ~(typeof(map->offset))0;
        snprintf(map_name, sizeof(map_name), "%.8s%.7s", obj->name,
                 libbpf_type_to_btf_name[type]);
        map->name = strdup(map_name);
        if (!map->name) {
                pr_warning("failed to alloc map name\n");
                return -ENOMEM;
        }

        def->type = BPF_MAP_TYPE_ARRAY;
        def->key_size = sizeof(int);
        def->value_size = data->d_size;
        def->max_entries = 1;
        def->map_flags = type == LIBBPF_MAP_RODATA ?
                         BPF_F_RDONLY_PROG : 0;
        if (data_buff) {
                *data_buff = malloc(data->d_size);
                if (!*data_buff) {
                        zfree(&map->name);
                        pr_warning("failed to alloc map content buffer\n");
                        return -ENOMEM;
                }
                memcpy(*data_buff, data->d_buf, data->d_size);
        }

        pr_debug("map %td is \"%s\"\n", map - obj->maps, map->name);
        return 0;
}

static int
bpf_object__init_maps(struct bpf_object *obj, int flags)
{
        int i, map_idx, map_def_sz = 0, nr_syms, nr_maps = 0, nr_maps_glob = 0;
        bool strict = !(flags & MAPS_RELAX_COMPAT);
        Elf_Data *symbols = obj->efile.symbols;
        Elf_Data *data = NULL;
        int ret = 0;

        if (!symbols)
                return -EINVAL;
        nr_syms = symbols->d_size / sizeof(GElf_Sym);

        if (obj->efile.maps_shndx >= 0) {
                Elf_Scn *scn = elf_getscn(obj->efile.elf,
                                          obj->efile.maps_shndx);

                if (scn)
                        data = elf_getdata(scn, NULL);
                if (!scn || !data) {
                        pr_warning("failed to get Elf_Data from map section %d\n",
                                   obj->efile.maps_shndx);
                        return -EINVAL;
                }
        }

        /*
         * Count number of maps. Each map has a name.
         * Array of maps is not supported: only the first element is
         * considered.
         *
         * TODO: Detect array of map and report error.
         */
        if (obj->caps.global_data) {
                if (obj->efile.data_shndx >= 0)
                        nr_maps_glob++;
                if (obj->efile.rodata_shndx >= 0)
                        nr_maps_glob++;
                if (obj->efile.bss_shndx >= 0)
                        nr_maps_glob++;
        }

        for (i = 0; data && i < nr_syms; i++) {
                GElf_Sym sym;

                if (!gelf_getsym(symbols, i, &sym))
                        continue;
                if (sym.st_shndx != obj->efile.maps_shndx)
                        continue;
                nr_maps++;
        }

        if (!nr_maps && !nr_maps_glob)
                return 0;

        /* Assume equally sized map definitions */
        if (data) {
                pr_debug("maps in %s: %d maps in %zd bytes\n", obj->path,
                         nr_maps, data->d_size);

                map_def_sz = data->d_size / nr_maps;
                if (!data->d_size || (data->d_size % nr_maps) != 0) {
                        pr_warning("unable to determine map definition size "
                                   "section %s, %d maps in %zd bytes\n",
                                   obj->path, nr_maps, data->d_size);
                        return -EINVAL;
                }
        }

        nr_maps += nr_maps_glob;
        obj->maps = calloc(nr_maps, sizeof(obj->maps[0]));
        if (!obj->maps) {
                pr_warning("alloc maps for object failed\n");
                return -ENOMEM;
        }
        obj->nr_maps = nr_maps;

        for (i = 0; i < nr_maps; i++) {
                /*
                 * fill all fd with -1 so won't close incorrect
                 * fd (fd=0 is stdin) when failure (zclose won't close
                 * negative fd)).
                 */
                obj->maps[i].fd = -1;
                obj->maps[i].inner_map_fd = -1;
        }

        /*
         * Fill obj->maps using data in "maps" section.
         */
        for (i = 0, map_idx = 0; data && i < nr_syms; i++) {
                GElf_Sym sym;
                const char *map_name;
                struct bpf_map_def *def;

                if (!gelf_getsym(symbols, i, &sym))
                        continue;
                if (sym.st_shndx != obj->efile.maps_shndx)
                        continue;

                map_name = elf_strptr(obj->efile.elf,
                                      obj->efile.strtabidx,
                                      sym.st_name);

                obj->maps[map_idx].libbpf_type = LIBBPF_MAP_UNSPEC;
                obj->maps[map_idx].offset = sym.st_value;
                if (sym.st_value + map_def_sz > data->d_size) {
                        pr_warning("corrupted maps section in %s: last map \"%s\" too small\n",
                                   obj->path, map_name);
                        return -EINVAL;
                }

                obj->maps[map_idx].name = strdup(map_name);
                if (!obj->maps[map_idx].name) {
                        pr_warning("failed to alloc map name\n");
                        return -ENOMEM;
                }
                pr_debug("map %d is \"%s\"\n", map_idx,
                         obj->maps[map_idx].name);
                def = (struct bpf_map_def *)(data->d_buf + sym.st_value);
                /*
                 * If the definition of the map in the object file fits in
                 * bpf_map_def, copy it.  Any extra fields in our version
                 * of bpf_map_def will default to zero as a result of the
                 * calloc above.
                 */
                if (map_def_sz <= sizeof(struct bpf_map_def)) {
                        memcpy(&obj->maps[map_idx].def, def, map_def_sz);
                } else {
                        /*
                         * Here the map structure being read is bigger than what
                         * we expect, truncate if the excess bits are all zero.
                         * If they are not zero, reject this map as
                         * incompatible.
                         */
                        char *b;
                        for (b = ((char *)def) + sizeof(struct bpf_map_def);
                             b < ((char *)def) + map_def_sz; b++) {
                                if (*b != 0) {
                                        pr_warning("maps section in %s: \"%s\" "
                                                   "has unrecognized, non-zero "
                                                   "options\n",
                                                   obj->path, map_name);
                                        if (strict)
                                                return -EINVAL;
                                }
                        }
                        memcpy(&obj->maps[map_idx].def, def,
                               sizeof(struct bpf_map_def));
                }
                map_idx++;
        }

        if (!obj->caps.global_data)
                goto finalize;

        /*
         * Populate rest of obj->maps with libbpf internal maps.
         */
        if (obj->efile.data_shndx >= 0)
                ret = bpf_object__init_internal_map(obj, &obj->maps[map_idx++],
                                                    LIBBPF_MAP_DATA,
                                                    obj->efile.data,
                                                    &obj->sections.data);
        if (!ret && obj->efile.rodata_shndx >= 0)
                ret = bpf_object__init_internal_map(obj, &obj->maps[map_idx++],
                                                    LIBBPF_MAP_RODATA,
                                                    obj->efile.rodata,
                                                    &obj->sections.rodata);
        if (!ret && obj->efile.bss_shndx >= 0)
                ret = bpf_object__init_internal_map(obj, &obj->maps[map_idx++],

                                                    LIBBPF_MAP_BSS,
                                                    obj->efile.bss, NULL);
finalize:
        if (!ret)
                qsort(obj->maps, obj->nr_maps, sizeof(obj->maps[0]),
                      compare_bpf_map);
        return ret;
}

static bool section_have_execinstr(struct bpf_object *obj, int idx)
{
        Elf_Scn *scn;
        GElf_Shdr sh;

        scn = elf_getscn(obj->efile.elf, idx);
        if (!scn)
                return false;

        if (gelf_getshdr(scn, &sh) != &sh)
                return false;

        if (sh.sh_flags & SHF_EXECINSTR)
                return true;

        return false;
}

static void bpf_object__sanitize_btf(struct bpf_object *obj)
{
        bool has_datasec = obj->caps.btf_datasec;
        bool has_func = obj->caps.btf_func;
        struct btf *btf = obj->btf;
        struct btf_type *t;
        int i, j, vlen;
        __u16 kind;

        if (!obj->btf || (has_func && has_datasec))
                return;

        for (i = 1; i <= btf__get_nr_types(btf); i++) {
                t = (struct btf_type *)btf__type_by_id(btf, i);
                kind = BTF_INFO_KIND(t->info);

                if (!has_datasec && kind == BTF_KIND_VAR) {
                        /* replace VAR with INT */
                        t->info = BTF_INFO_ENC(BTF_KIND_INT, 0, 0);
                        t->size = sizeof(int);
                        *(int *)(t+1) = BTF_INT_ENC(0, 0, 32);
                } else if (!has_datasec && kind == BTF_KIND_DATASEC) {
                        /* replace DATASEC with STRUCT */
                        struct btf_var_secinfo *v = (void *)(t + 1);
                        struct btf_member *m = (void *)(t + 1);
                        struct btf_type *vt;
                        char *name;

                        name = (char *)btf__name_by_offset(btf, t->name_off);
                        while (*name) {
                                if (*name == '.')
                                        *name = '_';
                                name++;
                        }

                        vlen = BTF_INFO_VLEN(t->info);
                        t->info = BTF_INFO_ENC(BTF_KIND_STRUCT, 0, vlen);
                        for (j = 0; j < vlen; j++, v++, m++) {
                                /* order of field assignments is important */
                                m->offset = v->offset * 8;
                                m->type = v->type;
                                /* preserve variable name as member name */
                                vt = (void *)btf__type_by_id(btf, v->type);
                                m->name_off = vt->name_off;
                        }
                } else if (!has_func && kind == BTF_KIND_FUNC_PROTO) {
                        /* replace FUNC_PROTO with ENUM */
                        vlen = BTF_INFO_VLEN(t->info);
                        t->info = BTF_INFO_ENC(BTF_KIND_ENUM, 0, vlen);
                        t->size = sizeof(__u32); /* kernel enforced */
                } else if (!has_func && kind == BTF_KIND_FUNC) {
                        /* replace FUNC with TYPEDEF */
                        t->info = BTF_INFO_ENC(BTF_KIND_TYPEDEF, 0, 0);
                }
        }
}

static void bpf_object__sanitize_btf_ext(struct bpf_object *obj)
{
        if (!obj->btf_ext)
                return;

        if (!obj->caps.btf_func) {
                btf_ext__free(obj->btf_ext);
                obj->btf_ext = NULL;
        }
}

static int bpf_object__elf_collect(struct bpf_object *obj, int flags)
{
        Elf *elf = obj->efile.elf;
        GElf_Ehdr *ep = &obj->efile.ehdr;
        Elf_Data *btf_ext_data = NULL;
        Elf_Data *btf_data = NULL;
        Elf_Scn *scn = NULL;
        int idx = 0, err = 0;

        /* Elf is corrupted/truncated, avoid calling elf_strptr. */
        if (!elf_rawdata(elf_getscn(elf, ep->e_shstrndx), NULL)) {
                pr_warning("failed to get e_shstrndx from %s\n",
                           obj->path);
                return -LIBBPF_ERRNO__FORMAT;
        }

        while ((scn = elf_nextscn(elf, scn)) != NULL) {
                char *name;
                GElf_Shdr sh;
                Elf_Data *data;

                idx++;
                if (gelf_getshdr(scn, &sh) != &sh) {
                        pr_warning("failed to get section(%d) header from %s\n",
                                   idx, obj->path);
                        err = -LIBBPF_ERRNO__FORMAT;
                        goto out;
                }

                name = elf_strptr(elf, ep->e_shstrndx, sh.sh_name);
                if (!name) {
                        pr_warning("failed to get section(%d) name from %s\n",
                                   idx, obj->path);
                        err = -LIBBPF_ERRNO__FORMAT;
                        goto out;
                }

                data = elf_getdata(scn, 0);
                if (!data) {
                        pr_warning("failed to get section(%d) data from %s(%s)\n",
                                   idx, name, obj->path);
                        err = -LIBBPF_ERRNO__FORMAT;
                        goto out;
                }
                pr_debug("section(%d) %s, size %ld, link %d, flags %lx, type=%d\n",
                         idx, name, (unsigned long)data->d_size,
                         (int)sh.sh_link, (unsigned long)sh.sh_flags,
                         (int)sh.sh_type);

                if (strcmp(name, "license") == 0) {
                        err = bpf_object__init_license(obj,
                                                       data->d_buf,
                                                       data->d_size);
                } else if (strcmp(name, "version") == 0) {
                        err = bpf_object__init_kversion(obj,
                                                        data->d_buf,
                                                        data->d_size);
                } else if (strcmp(name, "maps") == 0) {
                        obj->efile.maps_shndx = idx;
                } else if (strcmp(name, BTF_ELF_SEC) == 0) {
                        btf_data = data;
                } else if (strcmp(name, BTF_EXT_ELF_SEC) == 0) {
                        btf_ext_data = data;
                } else if (sh.sh_type == SHT_SYMTAB) {
                        if (obj->efile.symbols) {
                                pr_warning("bpf: multiple SYMTAB in %s\n",
                                           obj->path);
                                err = -LIBBPF_ERRNO__FORMAT;
                        } else {
                                obj->efile.symbols = data;
                                obj->efile.strtabidx = sh.sh_link;
                        }
                } else if (sh.sh_type == SHT_PROGBITS && data->d_size > 0) {
                        if (sh.sh_flags & SHF_EXECINSTR) {
                                if (strcmp(name, ".text") == 0)
                                        obj->efile.text_shndx = idx;
                                err = bpf_object__add_program(obj, data->d_buf,
                                                              data->d_size, name, idx);
                                if (err) {
                                        char errmsg[STRERR_BUFSIZE];
                                        char *cp = libbpf_strerror_r(-err, errmsg,
                                                                     sizeof(errmsg));

                                        pr_warning("failed to alloc program %s (%s): %s",
                                                   name, obj->path, cp);
                                }
                        } else if (strcmp(name, ".data") == 0) {
                                obj->efile.data = data;
                                obj->efile.data_shndx = idx;
                        } else if (strcmp(name, ".rodata") == 0) {
                                obj->efile.rodata = data;
                                obj->efile.rodata_shndx = idx;
                        } else {
                                pr_debug("skip section(%d) %s\n", idx, name);
                        }
                } else if (sh.sh_type == SHT_REL) {
                        void *reloc = obj->efile.reloc;
                        int nr_reloc = obj->efile.nr_reloc + 1;
                        int sec = sh.sh_info; /* points to other section */

                        /* Only do relo for section with exec instructions */
                        if (!section_have_execinstr(obj, sec)) {
                                pr_debug("skip relo %s(%d) for section(%d)\n",
                                         name, idx, sec);
                                continue;
                        }

                        reloc = reallocarray(reloc, nr_reloc,
                                             sizeof(*obj->efile.reloc));
                        if (!reloc) {
                                pr_warning("realloc failed\n");
                                err = -ENOMEM;
                        } else {
                                int n = nr_reloc - 1;

                                obj->efile.reloc = reloc;
                                obj->efile.nr_reloc = nr_reloc;

                                obj->efile.reloc[n].shdr = sh;
                                obj->efile.reloc[n].data = data;
                        }
                } else if (sh.sh_type == SHT_NOBITS && strcmp(name, ".bss") == 0) {
                        obj->efile.bss = data;
                        obj->efile.bss_shndx = idx;
                } else {
                        pr_debug("skip section(%d) %s\n", idx, name);
                }
                if (err)
                        goto out;
        }

        if (!obj->efile.strtabidx || obj->efile.strtabidx >= idx) {
                pr_warning("Corrupted ELF file: index of strtab invalid\n");
                return LIBBPF_ERRNO__FORMAT;
        }
        if (btf_data) {
                obj->btf = btf__new(btf_data->d_buf, btf_data->d_size);
                if (IS_ERR(obj->btf)) {
                        pr_warning("Error loading ELF section %s: %ld. Ignored and continue.\n",
                                   BTF_ELF_SEC, PTR_ERR(obj->btf));
                        obj->btf = NULL;
                } else {
                        err = btf__finalize_data(obj, obj->btf);
                        if (!err) {
                                bpf_object__sanitize_btf(obj);
                                err = btf__load(obj->btf);
                        }
                        if (err) {
                                pr_warning("Error finalizing and loading %s into kernel: %d. Ignored and continue.\n",
                                           BTF_ELF_SEC, err);
                                btf__free(obj->btf);
                                obj->btf = NULL;
                                err = 0;
                        }
                }
        }
        if (btf_ext_data) {
                if (!obj->btf) {
                        pr_debug("Ignore ELF section %s because its depending ELF section %s is not found.\n",
                                 BTF_EXT_ELF_SEC, BTF_ELF_SEC);
                } else {
                        obj->btf_ext = btf_ext__new(btf_ext_data->d_buf,
                                                    btf_ext_data->d_size);
                        if (IS_ERR(obj->btf_ext)) {
                                pr_warning("Error loading ELF section %s: %ld. Ignored and continue.\n",
                                           BTF_EXT_ELF_SEC,
                                           PTR_ERR(obj->btf_ext));
                                obj->btf_ext = NULL;
                        } else {
                                bpf_object__sanitize_btf_ext(obj);
                        }
                }
        }
        if (bpf_object__has_maps(obj)) {
                err = bpf_object__init_maps(obj, flags);
                if (err)
                        goto out;
        }
        err = bpf_object__init_prog_names(obj);
out:
        return err;
}

static struct bpf_program *
bpf_object__find_prog_by_idx(struct bpf_object *obj, int idx)
{
        struct bpf_program *prog;
        size_t i;

        for (i = 0; i < obj->nr_programs; i++) {
                prog = &obj->programs[i];
                if (prog->idx == idx)
                        return prog;
        }
        return NULL;
}

struct bpf_program *
bpf_object__find_program_by_title(struct bpf_object *obj, const char *title)
{
        struct bpf_program *pos;

        bpf_object__for_each_program(pos, obj) {
                if (pos->section_name && !strcmp(pos->section_name, title))
                        return pos;
        }
        return NULL;
}

static bool bpf_object__shndx_is_data(const struct bpf_object *obj,
                                      int shndx)
{
        return shndx == obj->efile.data_shndx ||
               shndx == obj->efile.bss_shndx ||
               shndx == obj->efile.rodata_shndx;
}

static bool bpf_object__shndx_is_maps(const struct bpf_object *obj,
                                      int shndx)
{
        return shndx == obj->efile.maps_shndx;
}

static bool bpf_object__relo_in_known_section(const struct bpf_object *obj,
                                              int shndx)
{
        return shndx == obj->efile.text_shndx ||
               bpf_object__shndx_is_maps(obj, shndx) ||
               bpf_object__shndx_is_data(obj, shndx);
}

static enum libbpf_map_type
bpf_object__section_to_libbpf_map_type(const struct bpf_object *obj, int shndx)
{
        if (shndx == obj->efile.data_shndx)
                return LIBBPF_MAP_DATA;
        else if (shndx == obj->efile.bss_shndx)
                return LIBBPF_MAP_BSS;
        else if (shndx == obj->efile.rodata_shndx)
                return LIBBPF_MAP_RODATA;
        else
                return LIBBPF_MAP_UNSPEC;
}

static int
bpf_program__collect_reloc(struct bpf_program *prog, GElf_Shdr *shdr,
                           Elf_Data *data, struct bpf_object *obj)
{
        Elf_Data *symbols = obj->efile.symbols;
        struct bpf_map *maps = obj->maps;
        size_t nr_maps = obj->nr_maps;
        int i, nrels;

        pr_debug("collecting relocating info for: '%s'\n",
                 prog->section_name);
        nrels = shdr->sh_size / shdr->sh_entsize;

        prog->reloc_desc = malloc(sizeof(*prog->reloc_desc) * nrels);
        if (!prog->reloc_desc) {
                pr_warning("failed to alloc memory in relocation\n");
                return -ENOMEM;
        }
        prog->nr_reloc = nrels;

        for (i = 0; i < nrels; i++) {
                GElf_Sym sym;
                GElf_Rel rel;
                unsigned int insn_idx;
                unsigned int shdr_idx;
                struct bpf_insn *insns = prog->insns;
                enum libbpf_map_type type;
                const char *name;
                size_t map_idx;

                if (!gelf_getrel(data, i, &rel)) {
                        pr_warning("relocation: failed to get %d reloc\n", i);
                        return -LIBBPF_ERRNO__FORMAT;
                }

                if (!gelf_getsym(symbols,
                                 GELF_R_SYM(rel.r_info),
                                 &sym)) {
                        pr_warning("relocation: symbol %"PRIx64" not found\n",
                                   GELF_R_SYM(rel.r_info));
                        return -LIBBPF_ERRNO__FORMAT;
                }

                name = elf_strptr(obj->efile.elf, obj->efile.strtabidx,
                                  sym.st_name) ? : "<?>";

                pr_debug("relo for %lld value %lld name %d (\'%s\')\n",
                         (long long) (rel.r_info >> 32),
                         (long long) sym.st_value, sym.st_name, name);

                shdr_idx = sym.st_shndx;
                if (!bpf_object__relo_in_known_section(obj, shdr_idx)) {
                        pr_warning("Program '%s' contains unrecognized relo data pointing to section %u\n",
                                   prog->section_name, shdr_idx);
                        return -LIBBPF_ERRNO__RELOC;
                }

                insn_idx = rel.r_offset / sizeof(struct bpf_insn);
                pr_debug("relocation: insn_idx=%u\n", insn_idx);

                if (insns[insn_idx].code == (BPF_JMP | BPF_CALL)) {
                        if (insns[insn_idx].src_reg != BPF_PSEUDO_CALL) {
                                pr_warning("incorrect bpf_call opcode\n");
                                return -LIBBPF_ERRNO__RELOC;
                        }
                        prog->reloc_desc[i].type = RELO_CALL;
                        prog->reloc_desc[i].insn_idx = insn_idx;
                        prog->reloc_desc[i].text_off = sym.st_value;
                        obj->has_pseudo_calls = true;
                        continue;
                }

                if (insns[insn_idx].code != (BPF_LD | BPF_IMM | BPF_DW)) {
                        pr_warning("bpf: relocation: invalid relo for insns[%d].code 0x%x\n",
                                   insn_idx, insns[insn_idx].code);
                        return -LIBBPF_ERRNO__RELOC;
                }

                if (bpf_object__shndx_is_maps(obj, shdr_idx) ||
                    bpf_object__shndx_is_data(obj, shdr_idx)) {
                        type = bpf_object__section_to_libbpf_map_type(obj, shdr_idx);
                        if (type != LIBBPF_MAP_UNSPEC) {
                                if (GELF_ST_BIND(sym.st_info) == STB_GLOBAL) {
                                        pr_warning("bpf: relocation: not yet supported relo for non-static global \'%s\' variable found in insns[%d].code 0x%x\n",
                                                   name, insn_idx, insns[insn_idx].code);
                                        return -LIBBPF_ERRNO__RELOC;
                                }
                                if (!obj->caps.global_data) {
                                        pr_warning("bpf: relocation: kernel does not support global \'%s\' variable access in insns[%d]\n",
                                                   name, insn_idx);
                                        return -LIBBPF_ERRNO__RELOC;
                                }
                        }

                        for (map_idx = 0; map_idx < nr_maps; map_idx++) {
                                if (maps[map_idx].libbpf_type != type)
                                        continue;
                                if (type != LIBBPF_MAP_UNSPEC ||
                                    (type == LIBBPF_MAP_UNSPEC &&
                                     maps[map_idx].offset == sym.st_value)) {
                                        pr_debug("relocation: find map %zd (%s) for insn %u\n",
                                                 map_idx, maps[map_idx].name, insn_idx);
                                        break;
                                }
                        }

                        if (map_idx >= nr_maps) {
                                pr_warning("bpf relocation: map_idx %d large than %d\n",
                                           (int)map_idx, (int)nr_maps - 1);
                                return -LIBBPF_ERRNO__RELOC;
                        }

                        prog->reloc_desc[i].type = type != LIBBPF_MAP_UNSPEC ?
                                                   RELO_DATA : RELO_LD64;
                        prog->reloc_desc[i].insn_idx = insn_idx;
                        prog->reloc_desc[i].map_idx = map_idx;
                }
        }
        return 0;
}

static int bpf_map_find_btf_info(struct bpf_map *map, const struct btf *btf)
{
        struct bpf_map_def *def = &map->def;
        __u32 key_type_id = 0, value_type_id = 0;
        int ret;

        if (!bpf_map__is_internal(map)) {
                ret = btf__get_map_kv_tids(btf, map->name, def->key_size,
                                           def->value_size, &key_type_id,
                                           &value_type_id);
        } else {
                /*
                 * LLVM annotates global data differently in BTF, that is,
                 * only as '.data', '.bss' or '.rodata'.
                 */
                ret = btf__find_by_name(btf,
                                libbpf_type_to_btf_name[map->libbpf_type]);
        }
        if (ret < 0)
                return ret;

        map->btf_key_type_id = key_type_id;
        map->btf_value_type_id = bpf_map__is_internal(map) ?
                                 ret : value_type_id;
        return 0;
}

int bpf_map__reuse_fd(struct bpf_map *map, int fd)
{
        struct bpf_map_info info = {};
        __u32 len = sizeof(info);
        int new_fd, err;
        char *new_name;

        err = bpf_obj_get_info_by_fd(fd, &info, &len);
        if (err)
                return err;

        new_name = strdup(info.name);
        if (!new_name)
                return -errno;

        new_fd = open("/", O_RDONLY | O_CLOEXEC);
        if (new_fd < 0)
                goto err_free_new_name;

        new_fd = dup3(fd, new_fd, O_CLOEXEC);
        if (new_fd < 0)
                goto err_close_new_fd;

        err = zclose(map->fd);
        if (err)
                goto err_close_new_fd;
        free(map->name);

        map->fd = new_fd;
        map->name = new_name;
        map->def.type = info.type;
        map->def.key_size = info.key_size;
        map->def.value_size = info.value_size;
        map->def.max_entries = info.max_entries;
        map->def.map_flags = info.map_flags;
        map->btf_key_type_id = info.btf_key_type_id;
        map->btf_value_type_id = info.btf_value_type_id;

        return 0;

err_close_new_fd:
        close(new_fd);
err_free_new_name:
        free(new_name);
        return -errno;
}

int bpf_map__resize(struct bpf_map *map, __u32 max_entries)
{
        if (!map || !max_entries)
                return -EINVAL;

        /* If map already created, its attributes can't be changed. */
        if (map->fd >= 0)
                return -EBUSY;

        map->def.max_entries = max_entries;

        return 0;
}

static int
bpf_object__probe_name(struct bpf_object *obj)
{
        struct bpf_load_program_attr attr;
        char *cp, errmsg[STRERR_BUFSIZE];
        struct bpf_insn insns[] = {
                BPF_MOV64_IMM(BPF_REG_0, 0),
                BPF_EXIT_INSN(),
        };
        int ret;

        /* make sure basic loading works */

        memset(&attr, 0, sizeof(attr));
        attr.prog_type = BPF_PROG_TYPE_SOCKET_FILTER;
        attr.insns = insns;
        attr.insns_cnt = ARRAY_SIZE(insns);
        attr.license = "GPL";

        ret = bpf_load_program_xattr(&attr, NULL, 0);
        if (ret < 0) {
                cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
                pr_warning("Error in %s():%s(%d). Couldn't load basic 'r0 = 0' BPF program.\n",
                           __func__, cp, errno);
                return -errno;
        }
        close(ret);

        /* now try the same program, but with the name */

        attr.name = "test";
        ret = bpf_load_program_xattr(&attr, NULL, 0);
        if (ret >= 0) {
                obj->caps.name = 1;
                close(ret);
        }

        return 0;
}

static int
bpf_object__probe_global_data(struct bpf_object *obj)
{
        struct bpf_load_program_attr prg_attr;
        struct bpf_create_map_attr map_attr;
        char *cp, errmsg[STRERR_BUFSIZE];
        struct bpf_insn insns[] = {
                BPF_LD_MAP_VALUE(BPF_REG_1, 0, 16),
                BPF_ST_MEM(BPF_DW, BPF_REG_1, 0, 42),
                BPF_MOV64_IMM(BPF_REG_0, 0),
                BPF_EXIT_INSN(),
        };
        int ret, map;

        memset(&map_attr, 0, sizeof(map_attr));
        map_attr.map_type = BPF_MAP_TYPE_ARRAY;
        map_attr.key_size = sizeof(int);
        map_attr.value_size = 32;
        map_attr.max_entries = 1;

        map = bpf_create_map_xattr(&map_attr);
        if (map < 0) {
                cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
                pr_warning("Error in %s():%s(%d). Couldn't create simple array map.\n",
                           __func__, cp, errno);
                return -errno;
        }

        insns[0].imm = map;

        memset(&prg_attr, 0, sizeof(prg_attr));
        prg_attr.prog_type = BPF_PROG_TYPE_SOCKET_FILTER;
        prg_attr.insns = insns;
        prg_attr.insns_cnt = ARRAY_SIZE(insns);
        prg_attr.license = "GPL";

        ret = bpf_load_program_xattr(&prg_attr, NULL, 0);
        if (ret >= 0) {
                obj->caps.global_data = 1;
                close(ret);
        }

        close(map);
        return 0;
}

static int bpf_object__probe_btf_func(struct bpf_object *obj)
{
        const char strs[] = "\0int\0x\0a";
        /* void x(int a) {} */
        __u32 types[] = {
                /* int */
                BTF_TYPE_INT_ENC(1, BTF_INT_SIGNED, 0, 32, 4),  /* [1] */
                /* FUNC_PROTO */                                /* [2] */
                BTF_TYPE_ENC(0, BTF_INFO_ENC(BTF_KIND_FUNC_PROTO, 0, 1), 0),
                BTF_PARAM_ENC(7, 1),
                /* FUNC x */                                    /* [3] */
                BTF_TYPE_ENC(5, BTF_INFO_ENC(BTF_KIND_FUNC, 0, 0), 2),
        };
        int btf_fd;

        btf_fd = libbpf__load_raw_btf((char *)types, sizeof(types),
                                      strs, sizeof(strs));
        if (btf_fd >= 0) {
                obj->caps.btf_func = 1;
                close(btf_fd);
                return 1;
        }

        return 0;
}

static int bpf_object__probe_btf_datasec(struct bpf_object *obj)
{
        const char strs[] = "\0x\0.data";
        /* static int a; */
        __u32 types[] = {
                /* int */
                BTF_TYPE_INT_ENC(0, BTF_INT_SIGNED, 0, 32, 4),  /* [1] */
                /* VAR x */                                     /* [2] */
                BTF_TYPE_ENC(1, BTF_INFO_ENC(BTF_KIND_VAR, 0, 0), 1),
                BTF_VAR_STATIC,
                /* DATASEC val */                               /* [3] */
                BTF_TYPE_ENC(3, BTF_INFO_ENC(BTF_KIND_DATASEC, 0, 1), 4),
                BTF_VAR_SECINFO_ENC(2, 0, 4),
        };
        int btf_fd;

        btf_fd = libbpf__load_raw_btf((char *)types, sizeof(types),
                                      strs, sizeof(strs));
        if (btf_fd >= 0) {
                obj->caps.btf_datasec = 1;
                close(btf_fd);
                return 1;
        }

        return 0;
}

static int
bpf_object__probe_caps(struct bpf_object *obj)
{
        int (*probe_fn[])(struct bpf_object *obj) = {
                bpf_object__probe_name,
                bpf_object__probe_global_data,
                bpf_object__probe_btf_func,
                bpf_object__probe_btf_datasec,
        };
        int i, ret;

        for (i = 0; i < ARRAY_SIZE(probe_fn); i++) {
                ret = probe_fn[i](obj);
                if (ret < 0)
                        pr_debug("Probe #%d failed with %d.\n", i, ret);
        }

        return 0;
}

static int
bpf_object__populate_internal_map(struct bpf_object *obj, struct bpf_map *map)
{
        char *cp, errmsg[STRERR_BUFSIZE];
        int err, zero = 0;
        __u8 *data;

        /* Nothing to do here since kernel already zero-initializes .bss map. */
        if (map->libbpf_type == LIBBPF_MAP_BSS)
                return 0;

        data = map->libbpf_type == LIBBPF_MAP_DATA ?
               obj->sections.data : obj->sections.rodata;

        err = bpf_map_update_elem(map->fd, &zero, data, 0);
        /* Freeze .rodata map as read-only from syscall side. */
        if (!err && map->libbpf_type == LIBBPF_MAP_RODATA) {
                err = bpf_map_freeze(map->fd);
                if (err) {
                        cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
                        pr_warning("Error freezing map(%s) as read-only: %s\n",
                                   map->name, cp);
                        err = 0;
                }
        }
        return err;
}

static int
bpf_object__create_maps(struct bpf_object *obj)
{
        struct bpf_create_map_attr create_attr = {};
        unsigned int i;
        int err;

        for (i = 0; i < obj->nr_maps; i++) {
                struct bpf_map *map = &obj->maps[i];
                struct bpf_map_def *def = &map->def;
                char *cp, errmsg[STRERR_BUFSIZE];
                int *pfd = &map->fd;

                if (map->fd >= 0) {
                        pr_debug("skip map create (preset) %s: fd=%d\n",
                                 map->name, map->fd);
                        continue;
                }

                if (obj->caps.name)
                        create_attr.name = map->name;
                create_attr.map_ifindex = map->map_ifindex;
                create_attr.map_type = def->type;
                create_attr.map_flags = def->map_flags;
                create_attr.key_size = def->key_size;
                create_attr.value_size = def->value_size;
                create_attr.max_entries = def->max_entries;
                create_attr.btf_fd = 0;
                create_attr.btf_key_type_id = 0;
                create_attr.btf_value_type_id = 0;
                if (bpf_map_type__is_map_in_map(def->type) &&
                    map->inner_map_fd >= 0)
                        create_attr.inner_map_fd = map->inner_map_fd;

                if (obj->btf && !bpf_map_find_btf_info(map, obj->btf)) {
                        create_attr.btf_fd = btf__fd(obj->btf);
                        create_attr.btf_key_type_id = map->btf_key_type_id;
                        create_attr.btf_value_type_id = map->btf_value_type_id;
                }

                *pfd = bpf_create_map_xattr(&create_attr);
                if (*pfd < 0 && create_attr.btf_key_type_id) {
                        cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
                        pr_warning("Error in bpf_create_map_xattr(%s):%s(%d). Retrying without BTF.\n",
                                   map->name, cp, errno);
                        create_attr.btf_fd = 0;
                        create_attr.btf_key_type_id = 0;
                        create_attr.btf_value_type_id = 0;
                        map->btf_key_type_id = 0;
                        map->btf_value_type_id = 0;
                        *pfd = bpf_create_map_xattr(&create_attr);
                }

                if (*pfd < 0) {
                        size_t j;

                        err = *pfd;
err_out:
                        cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
                        pr_warning("failed to create map (name: '%s'): %s\n",
                                   map->name, cp);
                        for (j = 0; j < i; j++)
                                zclose(obj->maps[j].fd);
                        return err;
                }

                if (bpf_map__is_internal(map)) {
                        err = bpf_object__populate_internal_map(obj, map);
                        if (err < 0) {
                                zclose(*pfd);
                                goto err_out;
                        }
                }

                pr_debug("create map %s: fd=%d\n", map->name, *pfd);
        }

        return 0;
}

static int
check_btf_ext_reloc_err(struct bpf_program *prog, int err,
                        void *btf_prog_info, const char *info_name)
{
        if (err != -ENOENT) {
                pr_warning("Error in loading %s for sec %s.\n",
                           info_name, prog->section_name);
                return err;
        }

        /* err == -ENOENT (i.e. prog->section_name not found in btf_ext) */

        if (btf_prog_info) {
                /*
                 * Some info has already been found but has problem
                 * in the last btf_ext reloc.  Must have to error
                 * out.
                 */
                pr_warning("Error in relocating %s for sec %s.\n",
                           info_name, prog->section_name);
                return err;
        }

        /*
         * Have problem loading the very first info.  Ignore
         * the rest.
         */
        pr_warning("Cannot find %s for main program sec %s. Ignore all %s.\n",
                   info_name, prog->section_name, info_name);
        return 0;
}

static int
bpf_program_reloc_btf_ext(struct bpf_program *prog, struct bpf_object *obj,
                          const char *section_name,  __u32 insn_offset)
{
        int err;

        if (!insn_offset || prog->func_info) {
                /*
                 * !insn_offset => main program
                 *
                 * For sub prog, the main program's func_info has to
                 * be loaded first (i.e. prog->func_info != NULL)
                 */
                err = btf_ext__reloc_func_info(obj->btf, obj->btf_ext,
                                               section_name, insn_offset,
                                               &prog->func_info,
                                               &prog->func_info_cnt);
                if (err)
                        return check_btf_ext_reloc_err(prog, err,
                                                       prog->func_info,
                                                       "bpf_func_info");

                prog->func_info_rec_size = btf_ext__func_info_rec_size(obj->btf_ext);
        }

        if (!insn_offset || prog->line_info) {
                err = btf_ext__reloc_line_info(obj->btf, obj->btf_ext,
                                               section_name, insn_offset,
                                               &prog->line_info,
                                               &prog->line_info_cnt);
                if (err)
                        return check_btf_ext_reloc_err(prog, err,
                                                       prog->line_info,
                                                       "bpf_line_info");

                prog->line_info_rec_size = btf_ext__line_info_rec_size(obj->btf_ext);
        }

        if (!insn_offset)
                prog->btf_fd = btf__fd(obj->btf);

        return 0;
}

static int
bpf_program__reloc_text(struct bpf_program *prog, struct bpf_object *obj,
                        struct reloc_desc *relo)
{
        struct bpf_insn *insn, *new_insn;
        struct bpf_program *text;
        size_t new_cnt;
        int err;

        if (relo->type != RELO_CALL)
                return -LIBBPF_ERRNO__RELOC;

        if (prog->idx == obj->efile.text_shndx) {
                pr_warning("relo in .text insn %d into off %d\n",
                           relo->insn_idx, relo->text_off);
                return -LIBBPF_ERRNO__RELOC;
        }

        if (prog->main_prog_cnt == 0) {
                text = bpf_object__find_prog_by_idx(obj, obj->efile.text_shndx);
                if (!text) {
                        pr_warning("no .text section found yet relo into text exist\n");
                        return -LIBBPF_ERRNO__RELOC;
                }
                new_cnt = prog->insns_cnt + text->insns_cnt;
                new_insn = reallocarray(prog->insns, new_cnt, sizeof(*insn));
                if (!new_insn) {
                        pr_warning("oom in prog realloc\n");
                        return -ENOMEM;
                }

                if (obj->btf_ext) {
                        err = bpf_program_reloc_btf_ext(prog, obj,
                                                        text->section_name,
                                                        prog->insns_cnt);
                        if (err)
                                return err;
                }

                memcpy(new_insn + prog->insns_cnt, text->insns,
                       text->insns_cnt * sizeof(*insn));
                prog->insns = new_insn;
                prog->main_prog_cnt = prog->insns_cnt;
                prog->insns_cnt = new_cnt;
                pr_debug("added %zd insn from %s to prog %s\n",
                         text->insns_cnt, text->section_name,
                         prog->section_name);
        }
        insn = &prog->insns[relo->insn_idx];
        insn->imm += prog->main_prog_cnt - relo->insn_idx;
        return 0;
}

static int
bpf_program__relocate(struct bpf_program *prog, struct bpf_object *obj)
{
        int i, err;

        if (!prog)
                return 0;

        if (obj->btf_ext) {
                err = bpf_program_reloc_btf_ext(prog, obj,
                                                prog->section_name, 0);
                if (err)
                        return err;
        }

        if (!prog->reloc_desc)
                return 0;

        for (i = 0; i < prog->nr_reloc; i++) {
                if (prog->reloc_desc[i].type == RELO_LD64 ||
                    prog->reloc_desc[i].type == RELO_DATA) {
                        bool relo_data = prog->reloc_desc[i].type == RELO_DATA;
                        struct bpf_insn *insns = prog->insns;
                        int insn_idx, map_idx;

                        insn_idx = prog->reloc_desc[i].insn_idx;
                        map_idx = prog->reloc_desc[i].map_idx;

                        if (insn_idx + 1 >= (int)prog->insns_cnt) {
                                pr_warning("relocation out of range: '%s'\n",
                                           prog->section_name);
                                return -LIBBPF_ERRNO__RELOC;
                        }

                        if (!relo_data) {
                                insns[insn_idx].src_reg = BPF_PSEUDO_MAP_FD;
                        } else {
                                insns[insn_idx].src_reg = BPF_PSEUDO_MAP_VALUE;
                                insns[insn_idx + 1].imm = insns[insn_idx].imm;
                        }
                        insns[insn_idx].imm = obj->maps[map_idx].fd;
                } else if (prog->reloc_desc[i].type == RELO_CALL) {
                        err = bpf_program__reloc_text(prog, obj,
                                                      &prog->reloc_desc[i]);
                        if (err)
                                return err;
                }
        }

        zfree(&prog->reloc_desc);
        prog->nr_reloc = 0;
        return 0;
}


static int

bpf_object__relocate(struct bpf_object *obj)
{
        struct bpf_program *prog;
        size_t i;
        int err;

        for (i = 0; i < obj->nr_programs; i++) {
                prog = &obj->programs[i];

                err = bpf_program__relocate(prog, obj);
                if (err) {
                        pr_warning("failed to relocate '%s'\n",
                                   prog->section_name);
                        return err;
                }
        }
        return 0;
}

static int bpf_object__collect_reloc(struct bpf_object *obj)
{
        int i, err;

        if (!obj_elf_valid(obj)) {
                pr_warning("Internal error: elf object is closed\n");
                return -LIBBPF_ERRNO__INTERNAL;
        }

        for (i = 0; i < obj->efile.nr_reloc; i++) {
                GElf_Shdr *shdr = &obj->efile.reloc[i].shdr;
                Elf_Data *data = obj->efile.reloc[i].data;
                int idx = shdr->sh_info;
                struct bpf_program *prog;

                if (shdr->sh_type != SHT_REL) {
                        pr_warning("internal error at %d\n", __LINE__);
                        return -LIBBPF_ERRNO__INTERNAL;
                }

                prog = bpf_object__find_prog_by_idx(obj, idx);
                if (!prog) {
                        pr_warning("relocation failed: no section(%d)\n", idx);
                        return -LIBBPF_ERRNO__RELOC;
                }

                err = bpf_program__collect_reloc(prog,
                                                 shdr, data,
                                                 obj);
                if (err)
                        return err;
        }
        return 0;
}

static int
load_program(struct bpf_program *prog, struct bpf_insn *insns, int insns_cnt,
             char *license, __u32 kern_version, int *pfd)
{
        struct bpf_load_program_attr load_attr;
        char *cp, errmsg[STRERR_BUFSIZE];
        int log_buf_size = BPF_LOG_BUF_SIZE;
        char *log_buf;
        int ret;

        memset(&load_attr, 0, sizeof(struct bpf_load_program_attr));
        load_attr.prog_type = prog->type;
        load_attr.expected_attach_type = prog->expected_attach_type;
        if (prog->caps->name)
                load_attr.name = prog->name;
        load_attr.insns = insns;
        load_attr.insns_cnt = insns_cnt;
        load_attr.license = license;
        load_attr.kern_version = kern_version;
        load_attr.prog_ifindex = prog->prog_ifindex;
        load_attr.prog_btf_fd = prog->btf_fd >= 0 ? prog->btf_fd : 0;
        load_attr.func_info = prog->func_info;
        load_attr.func_info_rec_size = prog->func_info_rec_size;
        load_attr.func_info_cnt = prog->func_info_cnt;
        load_attr.line_info = prog->line_info;
        load_attr.line_info_rec_size = prog->line_info_rec_size;
        load_attr.line_info_cnt = prog->line_info_cnt;
        load_attr.log_level = prog->log_level;
        if (!load_attr.insns || !load_attr.insns_cnt)
                return -EINVAL;

retry_load:
        log_buf = malloc(log_buf_size);
        if (!log_buf)
                pr_warning("Alloc log buffer for bpf loader error, continue without log\n");

        ret = bpf_load_program_xattr(&load_attr, log_buf, log_buf_size);

        if (ret >= 0) {
                if (load_attr.log_level)
                        pr_debug("verifier log:\n%s", log_buf);
                *pfd = ret;
                ret = 0;
                goto out;
        }

        if (errno == ENOSPC) {
                log_buf_size <<= 1;
                free(log_buf);
                goto retry_load;
        }
        ret = -LIBBPF_ERRNO__LOAD;
        cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
        pr_warning("load bpf program failed: %s\n", cp);

        if (log_buf && log_buf[0] != '\0') {
                ret = -LIBBPF_ERRNO__VERIFY;
                pr_warning("-- BEGIN DUMP LOG ---\n");
                pr_warning("\n%s\n", log_buf);
                pr_warning("-- END LOG --\n");
        } else if (load_attr.insns_cnt >= BPF_MAXINSNS) {
                pr_warning("Program too large (%zu insns), at most %d insns\n",
                           load_attr.insns_cnt, BPF_MAXINSNS);
                ret = -LIBBPF_ERRNO__PROG2BIG;
        } else {
                /* Wrong program type? */
                if (load_attr.prog_type != BPF_PROG_TYPE_KPROBE) {
                        int fd;

                        load_attr.prog_type = BPF_PROG_TYPE_KPROBE;
                        load_attr.expected_attach_type = 0;
                        fd = bpf_load_program_xattr(&load_attr, NULL, 0);
                        if (fd >= 0) {
                                close(fd);
                                ret = -LIBBPF_ERRNO__PROGTYPE;
                                goto out;
                        }
                }

                if (log_buf)
                        ret = -LIBBPF_ERRNO__KVER;
        }

out:
        free(log_buf);
        return ret;
}

int
bpf_program__load(struct bpf_program *prog,
                  char *license, __u32 kern_version)
{
        int err = 0, fd, i;

        if (prog->instances.nr < 0 || !prog->instances.fds) {
                if (prog->preprocessor) {
                        pr_warning("Internal error: can't load program '%s'\n",
                                   prog->section_name);
                        return -LIBBPF_ERRNO__INTERNAL;
                }

                prog->instances.fds = malloc(sizeof(int));
                if (!prog->instances.fds) {
                        pr_warning("Not enough memory for BPF fds\n");
                        return -ENOMEM;
                }
                prog->instances.nr = 1;
                prog->instances.fds[0] = -1;
        }

        if (!prog->preprocessor) {
                if (prog->instances.nr != 1) {
                        pr_warning("Program '%s' is inconsistent: nr(%d) != 1\n",
                                   prog->section_name, prog->instances.nr);
                }
                err = load_program(prog, prog->insns, prog->insns_cnt,
                                   license, kern_version, &fd);
                if (!err)
                        prog->instances.fds[0] = fd;
                goto out;
        }

        for (i = 0; i < prog->instances.nr; i++) {
                struct bpf_prog_prep_result result;
                bpf_program_prep_t preprocessor = prog->preprocessor;

                memset(&result, 0, sizeof(result));
                err = preprocessor(prog, i, prog->insns,
                                   prog->insns_cnt, &result);
                if (err) {
                        pr_warning("Preprocessing the %dth instance of program '%s' failed\n",
                                   i, prog->section_name);
                        goto out;
                }

                if (!result.new_insn_ptr || !result.new_insn_cnt) {
                        pr_debug("Skip loading the %dth instance of program '%s'\n",
                                 i, prog->section_name);
                        prog->instances.fds[i] = -1;
                        if (result.pfd)
                                *result.pfd = -1;
                        continue;
                }

                err = load_program(prog, result.new_insn_ptr,
                                   result.new_insn_cnt,
                                   license, kern_version, &fd);

                if (err) {
                        pr_warning("Loading the %dth instance of program '%s' failed\n",
                                        i, prog->section_name);
                        goto out;
                }

                if (result.pfd)
                        *result.pfd = fd;
                prog->instances.fds[i] = fd;
        }
out:
        if (err)
                pr_warning("failed to load program '%s'\n",
                           prog->section_name);
        zfree(&prog->insns);
        prog->insns_cnt = 0;
        return err;
}

static bool bpf_program__is_function_storage(struct bpf_program *prog,
                                             struct bpf_object *obj)
{
        return prog->idx == obj->efile.text_shndx && obj->has_pseudo_calls;
}

static int
bpf_object__load_progs(struct bpf_object *obj)
{
        size_t i;
        int err;

        for (i = 0; i < obj->nr_programs; i++) {
                if (bpf_program__is_function_storage(&obj->programs[i], obj))
                        continue;
                err = bpf_program__load(&obj->programs[i],
                                        obj->license,
                                        obj->kern_version);
                if (err)
                        return err;
        }
        return 0;
}

static bool bpf_prog_type__needs_kver(enum bpf_prog_type type)
{
        switch (type) {
        case BPF_PROG_TYPE_SOCKET_FILTER:
        case BPF_PROG_TYPE_SCHED_CLS:
        case BPF_PROG_TYPE_SCHED_ACT:
        case BPF_PROG_TYPE_XDP:
        case BPF_PROG_TYPE_CGROUP_SKB:
        case BPF_PROG_TYPE_CGROUP_SOCK:
        case BPF_PROG_TYPE_LWT_IN:
        case BPF_PROG_TYPE_LWT_OUT:
        case BPF_PROG_TYPE_LWT_XMIT:
        case BPF_PROG_TYPE_LWT_SEG6LOCAL:
        case BPF_PROG_TYPE_SOCK_OPS:
        case BPF_PROG_TYPE_SK_SKB:
        case BPF_PROG_TYPE_CGROUP_DEVICE:
        case BPF_PROG_TYPE_SK_MSG:
        case BPF_PROG_TYPE_CGROUP_SOCK_ADDR:
        case BPF_PROG_TYPE_LIRC_MODE2:
        case BPF_PROG_TYPE_SK_REUSEPORT:
        case BPF_PROG_TYPE_FLOW_DISSECTOR:
        case BPF_PROG_TYPE_UNSPEC:
        case BPF_PROG_TYPE_TRACEPOINT:
        case BPF_PROG_TYPE_RAW_TRACEPOINT:
        case BPF_PROG_TYPE_RAW_TRACEPOINT_WRITABLE:
        case BPF_PROG_TYPE_PERF_EVENT:
        case BPF_PROG_TYPE_CGROUP_SYSCTL:
                return false;
        case BPF_PROG_TYPE_KPROBE:
        default:
                return true;
        }
}

static int bpf_object__validate(struct bpf_object *obj, bool needs_kver)
{
        if (needs_kver && obj->kern_version == 0) {
                pr_warning("%s doesn't provide kernel version\n",
                           obj->path);
                return -LIBBPF_ERRNO__KVERSION;
        }
        return 0;
}

static struct bpf_object *
__bpf_object__open(const char *path, void *obj_buf, size_t obj_buf_sz,
                   bool needs_kver, int flags)
{
        struct bpf_object *obj;
        int err;

        if (elf_version(EV_CURRENT) == EV_NONE) {
                pr_warning("failed to init libelf for %s\n", path);
                return ERR_PTR(-LIBBPF_ERRNO__LIBELF);
        }

        obj = bpf_object__new(path, obj_buf, obj_buf_sz);
        if (IS_ERR(obj))
                return obj;

        CHECK_ERR(bpf_object__elf_init(obj), err, out);
        CHECK_ERR(bpf_object__check_endianness(obj), err, out);
        CHECK_ERR(bpf_object__probe_caps(obj), err, out);
        CHECK_ERR(bpf_object__elf_collect(obj, flags), err, out);
        CHECK_ERR(bpf_object__collect_reloc(obj), err, out);
        CHECK_ERR(bpf_object__validate(obj, needs_kver), err, out);

        bpf_object__elf_finish(obj);
        return obj;
out:
        bpf_object__close(obj);
        return ERR_PTR(err);
}

struct bpf_object *__bpf_object__open_xattr(struct bpf_object_open_attr *attr,
                                            int flags)
{
        /* param validation */
        if (!attr->file)
                return NULL;

        pr_debug("loading %s\n", attr->file);

        return __bpf_object__open(attr->file, NULL, 0,
                                  bpf_prog_type__needs_kver(attr->prog_type),
                                  flags);
}

struct bpf_object *bpf_object__open_xattr(struct bpf_object_open_attr *attr)
{
        return __bpf_object__open_xattr(attr, 0);
}

struct bpf_object *bpf_object__open(const char *path)
{
        struct bpf_object_open_attr attr = {
                .file           = path,
                .prog_type      = BPF_PROG_TYPE_UNSPEC,
        };

        return bpf_object__open_xattr(&attr);
}

struct bpf_object *bpf_object__open_buffer(void *obj_buf,
                                           size_t obj_buf_sz,
                                           const char *name)
{
        char tmp_name[64];

        /* param validation */
        if (!obj_buf || obj_buf_sz <= 0)
                return NULL;

        if (!name) {
                snprintf(tmp_name, sizeof(tmp_name), "%lx-%lx",
                         (unsigned long)obj_buf,
                         (unsigned long)obj_buf_sz);
                tmp_name[sizeof(tmp_name) - 1] = '\0';
                name = tmp_name;
        }
        pr_debug("loading object '%s' from buffer\n",
                 name);

        return __bpf_object__open(name, obj_buf, obj_buf_sz, true, true);
}

int bpf_object__unload(struct bpf_object *obj)
{
        size_t i;

        if (!obj)
                return -EINVAL;

        for (i = 0; i < obj->nr_maps; i++)
                zclose(obj->maps[i].fd);

        for (i = 0; i < obj->nr_programs; i++)
                bpf_program__unload(&obj->programs[i]);

        return 0;
}

int bpf_object__load(struct bpf_object *obj)
{
        int err;

        if (!obj)
                return -EINVAL;

        if (obj->loaded) {
                pr_warning("object should not be loaded twice\n");
                return -EINVAL;
        }

        obj->loaded = true;

        CHECK_ERR(bpf_object__create_maps(obj), err, out);
        CHECK_ERR(bpf_object__relocate(obj), err, out);
        CHECK_ERR(bpf_object__load_progs(obj), err, out);

        return 0;
out:
        bpf_object__unload(obj);
        pr_warning("failed to load object '%s'\n", obj->path);
        return err;
}

static int check_path(const char *path)
{
        char *cp, errmsg[STRERR_BUFSIZE];
        struct statfs st_fs;
        char *dname, *dir;
        int err = 0;

        if (path == NULL)
                return -EINVAL;

        dname = strdup(path);
        if (dname == NULL)
                return -ENOMEM;

        dir = dirname(dname);
        if (statfs(dir, &st_fs)) {
                cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
                pr_warning("failed to statfs %s: %s\n", dir, cp);
                err = -errno;
        }
        free(dname);

        if (!err && st_fs.f_type != BPF_FS_MAGIC) {
                pr_warning("specified path %s is not on BPF FS\n", path);
                err = -EINVAL;
        }

        return err;
}

int bpf_program__pin_instance(struct bpf_program *prog, const char *path,
                              int instance)
{
        char *cp, errmsg[STRERR_BUFSIZE];
        int err;

        err = check_path(path);
        if (err)
                return err;

        if (prog == NULL) {
                pr_warning("invalid program pointer\n");
                return -EINVAL;
        }

        if (instance < 0 || instance >= prog->instances.nr) {
                pr_warning("invalid prog instance %d of prog %s (max %d)\n",
                           instance, prog->section_name, prog->instances.nr);
                return -EINVAL;
        }

        if (bpf_obj_pin(prog->instances.fds[instance], path)) {
                cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
                pr_warning("failed to pin program: %s\n", cp);
                return -errno;
        }
        pr_debug("pinned program '%s'\n", path);

        return 0;
}

int bpf_program__unpin_instance(struct bpf_program *prog, const char *path,
                                int instance)
{
        int err;

        err = check_path(path);
        if (err)
                return err;

        if (prog == NULL) {
                pr_warning("invalid program pointer\n");
                return -EINVAL;
        }

        if (instance < 0 || instance >= prog->instances.nr) {
                pr_warning("invalid prog instance %d of prog %s (max %d)\n",
                           instance, prog->section_name, prog->instances.nr);
                return -EINVAL;
        }

        err = unlink(path);
        if (err != 0)
                return -errno;
        pr_debug("unpinned program '%s'\n", path);

        return 0;
}

static int make_dir(const char *path)
{
        char *cp, errmsg[STRERR_BUFSIZE];
        int err = 0;

        if (mkdir(path, 0700) && errno != EEXIST)
                err = -errno;

        if (err) {
                cp = libbpf_strerror_r(-err, errmsg, sizeof(errmsg));
                pr_warning("failed to mkdir %s: %s\n", path, cp);
        }
        return err;
}

int bpf_program__pin(struct bpf_program *prog, const char *path)
{
        int i, err;

        err = check_path(path);
        if (err)
                return err;

        if (prog == NULL) {
                pr_warning("invalid program pointer\n");
                return -EINVAL;
        }

        if (prog->instances.nr <= 0) {
                pr_warning("no instances of prog %s to pin\n",
                           prog->section_name);
                return -EINVAL;
        }

        if (prog->instances.nr == 1) {
                /* don't create subdirs when pinning single instance */
                return bpf_program__pin_instance(prog, path, 0);
        }

        err = make_dir(path);
        if (err)
                return err;

        for (i = 0; i < prog->instances.nr; i++) {
                char buf[PATH_MAX];
                int len;

                len = snprintf(buf, PATH_MAX, "%s/%d", path, i);
                if (len < 0) {
                        err = -EINVAL;
                        goto err_unpin;
                } else if (len >= PATH_MAX) {
                        err = -ENAMETOOLONG;
                        goto err_unpin;
                }

                err = bpf_program__pin_instance(prog, buf, i);
                if (err)
                        goto err_unpin;
        }

        return 0;

err_unpin:
        for (i = i - 1; i >= 0; i--) {
                char buf[PATH_MAX];
                int len;

                len = snprintf(buf, PATH_MAX, "%s/%d", path, i);
                if (len < 0)
                        continue;
                else if (len >= PATH_MAX)
                        continue;

                bpf_program__unpin_instance(prog, buf, i);
        }

        rmdir(path);

        return err;
}

int bpf_program__unpin(struct bpf_program *prog, const char *path)
{
        int i, err;

        err = check_path(path);
        if (err)
                return err;

        if (prog == NULL) {
                pr_warning("invalid program pointer\n");
                return -EINVAL;
        }

        if (prog->instances.nr <= 0) {
                pr_warning("no instances of prog %s to pin\n",
                           prog->section_name);
                return -EINVAL;
        }

        if (prog->instances.nr == 1) {
                /* don't create subdirs when pinning single instance */
                return bpf_program__unpin_instance(prog, path, 0);
        }

        for (i = 0; i < prog->instances.nr; i++) {
                char buf[PATH_MAX];
                int len;

                len = snprintf(buf, PATH_MAX, "%s/%d", path, i);
                if (len < 0)
                        return -EINVAL;
                else if (len >= PATH_MAX)
                        return -ENAMETOOLONG;

                err = bpf_program__unpin_instance(prog, buf, i);
                if (err)
                        return err;
        }

        err = rmdir(path);
        if (err)
                return -errno;

        return 0;
}

int bpf_map__pin(struct bpf_map *map, const char *path)
{
        char *cp, errmsg[STRERR_BUFSIZE];
        int err;

        err = check_path(path);
        if (err)
                return err;

        if (map == NULL) {
                pr_warning("invalid map pointer\n");
                return -EINVAL;
        }

        if (bpf_obj_pin(map->fd, path)) {
                cp = libbpf_strerror_r(errno, errmsg, sizeof(errmsg));
                pr_warning("failed to pin map: %s\n", cp);
                return -errno;
        }

        pr_debug("pinned map '%s'\n", path);

        return 0;
}

int bpf_map__unpin(struct bpf_map *map, const char *path)
{
        int err;

        err = check_path(path);
        if (err)
                return err;

        if (map == NULL) {
                pr_warning("invalid map pointer\n");
                return -EINVAL;
        }

        err = unlink(path);
        if (err != 0)
                return -errno;
        pr_debug("unpinned map '%s'\n", path);

        return 0;
}

int bpf_object__pin_maps(struct bpf_object *obj, const char *path)
{
        struct bpf_map *map;
        int err;

        if (!obj)
                return -ENOENT;

        if (!obj->loaded) {
                pr_warning("object not yet loaded; load it first\n");
                return -ENOENT;
        }

        err = make_dir(path);
        if (err)
                return err;

        bpf_object__for_each_map(map, obj) {
                char buf[PATH_MAX];
                int len;

                len = snprintf(buf, PATH_MAX, "%s/%s", path,
                               bpf_map__name(map));
                if (len < 0) {
                        err = -EINVAL;
                        goto err_unpin_maps;
                } else if (len >= PATH_MAX) {
                        err = -ENAMETOOLONG;
                        goto err_unpin_maps;
                }

                err = bpf_map__pin(map, buf);
                if (err)
                        goto err_unpin_maps;
        }

        return 0;

err_unpin_maps:
        while ((map = bpf_map__prev(map, obj))) {
                char buf[PATH_MAX];
                int len;

                len = snprintf(buf, PATH_MAX, "%s/%s", path,
                               bpf_map__name(map));
                if (len < 0)
                        continue;
                else if (len >= PATH_MAX)
                        continue;

                bpf_map__unpin(map, buf);
        }

        return err;
}

int bpf_object__unpin_maps(struct bpf_object *obj, const char *path)
{
        struct bpf_map *map;
        int err;

        if (!obj)
                return -ENOENT;

        bpf_object__for_each_map(map, obj) {
                char buf[PATH_MAX];
                int len;

                len = snprintf(buf, PATH_MAX, "%s/%s", path,
                               bpf_map__name(map));
                if (len < 0)
                        return -EINVAL;
                else if (len >= PATH_MAX)
                        return -ENAMETOOLONG;

                err = bpf_map__unpin(map, buf);
                if (err)
                        return err;
        }

        return 0;
}

int bpf_object__pin_programs(struct bpf_object *obj, const char *path)
{
        struct bpf_program *prog;
        int err;

        if (!obj)
                return -ENOENT;

        if (!obj->loaded) {
                pr_warning("object not yet loaded; load it first\n");
                return -ENOENT;
        }

        err = make_dir(path);
        if (err)
                return err;

        bpf_object__for_each_program(prog, obj) {
                char buf[PATH_MAX];
                int len;

                len = snprintf(buf, PATH_MAX, "%s/%s", path,
                               prog->pin_name);
                if (len < 0) {
                        err = -EINVAL;
                        goto err_unpin_programs;
                } else if (len >= PATH_MAX) {
                        err = -ENAMETOOLONG;
                        goto err_unpin_programs;
                }

                err = bpf_program__pin(prog, buf);
                if (err)
                        goto err_unpin_programs;
        }

        return 0;

err_unpin_programs:
        while ((prog = bpf_program__prev(prog, obj))) {
                char buf[PATH_MAX];
                int len;

                len = snprintf(buf, PATH_MAX, "%s/%s", path,
                               prog->pin_name);
                if (len < 0)
                        continue;
                else if (len >= PATH_MAX)
                        continue;

                bpf_program__unpin(prog, buf);
        }

        return err;
}

int bpf_object__unpin_programs(struct bpf_object *obj, const char *path)
{
        struct bpf_program *prog;
        int err;

        if (!obj)
                return -ENOENT;

        bpf_object__for_each_program(prog, obj) {
                char buf[PATH_MAX];
                int len;

                len = snprintf(buf, PATH_MAX, "%s/%s", path,
                               prog->pin_name);
                if (len < 0)
                        return -EINVAL;
                else if (len >= PATH_MAX)
                        return -ENAMETOOLONG;

                err = bpf_program__unpin(prog, buf);
                if (err)
                        return err;
        }

        return 0;
}

int bpf_object__pin(struct bpf_object *obj, const char *path)
{
        int err;

        err = bpf_object__pin_maps(obj, path);
        if (err)
                return err;

        err = bpf_object__pin_programs(obj, path);
        if (err) {
                bpf_object__unpin_maps(obj, path);
                return err;
        }

        return 0;
}

void bpf_object__close(struct bpf_object *obj)
{
        size_t i;

        if (!obj)
                return;

        if (obj->clear_priv)
                obj->clear_priv(obj, obj->priv);

        bpf_object__elf_finish(obj);
        bpf_object__unload(obj);
        btf__free(obj->btf);
        btf_ext__free(obj->btf_ext);

        for (i = 0; i < obj->nr_maps; i++) {
                zfree(&obj->maps[i].name);
                if (obj->maps[i].clear_priv)
                        obj->maps[i].clear_priv(&obj->maps[i],
                                                obj->maps[i].priv);
                obj->maps[i].priv = NULL;
                obj->maps[i].clear_priv = NULL;
        }

        zfree(&obj->sections.rodata);
        zfree(&obj->sections.data);
        zfree(&obj->maps);
        obj->nr_maps = 0;

        if (obj->programs && obj->nr_programs) {
                for (i = 0; i < obj->nr_programs; i++)
                        bpf_program__exit(&obj->programs[i]);
        }
        zfree(&obj->programs);

        list_del(&obj->list);
        free(obj);
}

struct bpf_object *
bpf_object__next(struct bpf_object *prev)
{
        struct bpf_object *next;

        if (!prev)
                next = list_first_entry(&bpf_objects_list,
                                        struct bpf_object,
                                        list);
        else
                next = list_next_entry(prev, list);

        /* Empty list is noticed here so don't need checking on entry. */
        if (&next->list == &bpf_objects_list)
                return NULL;

        return next;
}

const char *bpf_object__name(struct bpf_object *obj)
{
        return obj ? obj->path : ERR_PTR(-EINVAL);
}

unsigned int bpf_object__kversion(struct bpf_object *obj)
{
        return obj ? obj->kern_version : 0;
}

struct btf *bpf_object__btf(struct bpf_object *obj)
{
        return obj ? obj->btf : NULL;
}

int bpf_object__btf_fd(const struct bpf_object *obj)
{
        return obj->btf ? btf__fd(obj->btf) : -1;
}

int bpf_object__set_priv(struct bpf_object *obj, void *priv,
                         bpf_object_clear_priv_t clear_priv)
{
        if (obj->priv && obj->clear_priv)
                obj->clear_priv(obj, obj->priv);

        obj->priv = priv;
        obj->clear_priv = clear_priv;
        return 0;
}

void *bpf_object__priv(struct bpf_object *obj)
{
        return obj ? obj->priv : ERR_PTR(-EINVAL);
}

static struct bpf_program *
__bpf_program__iter(struct bpf_program *p, struct bpf_object *obj, bool forward)
{
        size_t nr_programs = obj->nr_programs;
        ssize_t idx;

        if (!nr_programs)
                return NULL;

        if (!p)
                /* Iter from the beginning */
                return forward ? &obj->programs[0] :
                        &obj->programs[nr_programs - 1];

        if (p->obj != obj) {
                pr_warning("error: program handler doesn't match object\n");
                return NULL;
        }

        idx = (p - obj->programs) + (forward ? 1 : -1);
        if (idx >= obj->nr_programs || idx < 0)
                return NULL;
        return &obj->programs[idx];
}

struct bpf_program *
bpf_program__next(struct bpf_program *prev, struct bpf_object *obj)
{
        struct bpf_program *prog = prev;

        do {
                prog = __bpf_program__iter(prog, obj, true);
        } while (prog && bpf_program__is_function_storage(prog, obj));

        return prog;
}

struct bpf_program *
bpf_program__prev(struct bpf_program *next, struct bpf_object *obj)
{
        struct bpf_program *prog = next;

        do {
                prog = __bpf_program__iter(prog, obj, false);
        } while (prog && bpf_program__is_function_storage(prog, obj));

        return prog;
}