From patchwork Thu Oct 8 23:39:53 2020 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Andrii Nakryiko X-Patchwork-Id: 11824745 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org X-Spam-Level: X-Spam-Status: No, score=-13.1 required=3.0 tests=BAYES_00,DKIMWL_WL_HIGH, DKIM_SIGNED,DKIM_VALID,DKIM_VALID_AU,INCLUDES_PATCH,MAILING_LIST_MULTI, SIGNED_OFF_BY,SPF_HELO_NONE,SPF_PASS,URIBL_BLOCKED,USER_AGENT_GIT autolearn=ham autolearn_force=no version=3.4.0 Received: from mail.kernel.org (mail.kernel.org [198.145.29.99]) by smtp.lore.kernel.org (Postfix) with ESMTP id 78349C433E7 for ; Thu, 8 Oct 2020 23:40:15 +0000 (UTC) Received: from vger.kernel.org (vger.kernel.org [23.128.96.18]) by mail.kernel.org (Postfix) with ESMTP id 072A82224E for ; Thu, 8 Oct 2020 23:40:14 +0000 (UTC) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=kernel.org; s=default; t=1602200415; bh=kCyrmyD4PjmMxJYyQwUOXpHfAt+gT8Tljrpn9SpYh1U=; h=From:To:CC:Subject:Date:In-Reply-To:References:List-ID:From; b=W09qiJQJqMSw05cZjfdxs3832XsLzFWoUmL+bqNtzaOLkjffH4hI+Nzc1VUucaqHI zpOPeOzg+/kCwyjaJRpZhfuwtxTrEHxnLVNUnjV7h8eO06rxUiRmGT0UJYJ2O5rT4F fcSZaIfWGd4Wu22tBEBPlQ9Zmeu5Kh6sLafqgdqs= Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S1730538AbgJHXkO convert rfc822-to-8bit (ORCPT ); Thu, 8 Oct 2020 19:40:14 -0400 Received: from mx0b-00082601.pphosted.com ([67.231.153.30]:18428 "EHLO mx0b-00082601.pphosted.com" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S1725909AbgJHXkO (ORCPT ); Thu, 8 Oct 2020 19:40:14 -0400 Received: from pps.filterd (m0109332.ppops.net [127.0.0.1]) by mx0a-00082601.pphosted.com (8.16.0.42/8.16.0.42) with SMTP id 098NZ5JN017591 for ; Thu, 8 Oct 2020 16:40:09 -0700 Received: from maileast.thefacebook.com ([163.114.130.16]) by mx0a-00082601.pphosted.com with ESMTP id 3429gp8w68-3 (version=TLSv1.2 cipher=ECDHE-RSA-AES128-GCM-SHA256 bits=128 verify=NOT) for ; Thu, 08 Oct 2020 16:40:09 -0700 Received: from intmgw002.08.frc2.facebook.com (2620:10d:c0a8:1b::d) by mail.thefacebook.com (2620:10d:c0a8:83::4) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256) id 15.1.1979.3; Thu, 8 Oct 2020 16:40:07 -0700 Received: by devbig012.ftw2.facebook.com (Postfix, from userid 137359) id C5BB52EC7C76; Thu, 8 Oct 2020 16:40:05 -0700 (PDT) From: Andrii Nakryiko To: CC: , , , Andrii Nakryiko , Arnaldo Carvalho de Melo , Andrii Nakryiko Subject: [PATCH v2 dwarves 1/8] btf_loader: use libbpf to load BTF Date: Thu, 8 Oct 2020 16:39:53 -0700 Message-ID: <20201008234000.740660-2-andrii@kernel.org> X-Mailer: git-send-email 2.24.1 In-Reply-To: <20201008234000.740660-1-andrii@kernel.org> References: <20201008234000.740660-1-andrii@kernel.org> MIME-Version: 1.0 X-FB-Internal: Safe X-Proofpoint-Virus-Version: vendor=fsecure engine=2.50.10434:6.0.235,18.0.687 definitions=2020-10-08_15:2020-10-08,2020-10-08 signatures=0 X-Proofpoint-Spam-Details: rule=fb_default_notspam policy=fb_default score=0 lowpriorityscore=0 spamscore=0 clxscore=1015 bulkscore=0 mlxscore=0 adultscore=0 mlxlogscore=999 suspectscore=38 impostorscore=0 phishscore=0 malwarescore=0 priorityscore=1501 classifier=spam adjust=0 reason=mlx scancount=1 engine=8.12.0-2009150000 definitions=main-2010080166 X-FB-Internal: deliver Precedence: bulk List-ID: X-Mailing-List: bpf@vger.kernel.org From: Andrii Nakryiko Switch BTF loading to completely use libbpf's own struct btf and related APIs. BTF encoding is still happening with pahole's own code, so these two code paths are not sharing anything now. String fetching is happening based on whether btfe->strings were set to non-NULL pointer by btf_encoder. Signed-off-by: Andrii Nakryiko --- btf_loader.c | 244 +++++++++++++++++++-------------------------------- libbtf.c | 116 +++++------------------- libbtf.h | 11 +-- 3 files changed, 113 insertions(+), 258 deletions(-) diff --git a/btf_loader.c b/btf_loader.c index 8fc6c3c15f25..9b5da3a4997a 100644 --- a/btf_loader.c +++ b/btf_loader.c @@ -46,21 +46,17 @@ static void *tag__alloc(const size_t size) } static int btf_elf__load_ftype(struct btf_elf *btfe, struct ftype *proto, uint32_t tag, - uint32_t type, uint16_t vlen, struct btf_param *args, uint32_t id) + const struct btf_type *tp, uint32_t id) { - int i; + const struct btf_param *param = btf_params(tp); + int i, vlen = btf_vlen(tp); proto->tag.tag = tag; - proto->tag.type = type; + proto->tag.type = tp->type; INIT_LIST_HEAD(&proto->parms); - for (i = 0; i < vlen; ++i) { - struct btf_param param = { - .name_off = btf_elf__get32(btfe, &args[i].name_off), - .type = btf_elf__get32(btfe, &args[i].type), - }; - - if (param.type == 0) + for (i = 0; i < vlen; ++i, param++) { + if (param->type == 0) proto->unspec_parms = 1; else { struct parameter *p = tag__alloc(sizeof(*p)); @@ -68,25 +64,22 @@ static int btf_elf__load_ftype(struct btf_elf *btfe, struct ftype *proto, uint32 if (p == NULL) goto out_free_parameters; p->tag.tag = DW_TAG_formal_parameter; - p->tag.type = param.type; - p->name = param.name_off; + p->tag.type = param->type; + p->name = param->name_off; ftype__add_parameter(proto, p); } } - vlen *= sizeof(*args); cu__add_tag_with_id(btfe->priv, &proto->tag, id); - return vlen; + return 0; out_free_parameters: ftype__delete(proto, btfe->priv); return -ENOMEM; } -static int create_new_function(struct btf_elf *btfe, struct btf_type *tp, uint64_t size, uint32_t id) +static int create_new_function(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - strings_t name = btf_elf__get32(btfe, &tp->name_off); - unsigned int type_id = btf_elf__get32(btfe, &tp->type); struct function *func = tag__alloc(sizeof(*func)); if (func == NULL) @@ -96,9 +89,9 @@ static int create_new_function(struct btf_elf *btfe, struct btf_type *tp, uint64 // but the prototype, the return type is the one in type_id func->btf = 1; func->proto.tag.tag = DW_TAG_subprogram; - func->proto.tag.type = type_id; + func->proto.tag.type = tp->type; + func->name = tp->name_off; INIT_LIST_HEAD(&func->lexblock.tags); - func->name = name; cu__add_tag_with_id(btfe->priv, &func->proto.tag, id); return 0; @@ -166,26 +159,24 @@ static struct variable *variable__new(strings_t name, uint32_t linkage) return var; } -static int create_new_base_type(struct btf_elf *btfe, void *ptr, struct btf_type *tp, uint32_t id) +static int create_new_base_type(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - uint32_t *enc = ptr; - uint32_t eval = btf_elf__get32(btfe, enc); - uint32_t attrs = BTF_INT_ENCODING(eval); - strings_t name = btf_elf__get32(btfe, &tp->name_off); - struct base_type *base = base_type__new(name, attrs, 0, - BTF_INT_BITS(eval)); + uint32_t attrs = btf_int_encoding(tp); + strings_t name = tp->name_off; + struct base_type *base = base_type__new(name, attrs, 0, btf_int_bits(tp)); + if (base == NULL) return -ENOMEM; base->tag.tag = DW_TAG_base_type; cu__add_tag_with_id(btfe->priv, &base->tag, id); - return sizeof(*enc); + return 0; } -static int create_new_array(struct btf_elf *btfe, void *ptr, uint32_t id) +static int create_new_array(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - struct btf_array *ap = ptr; + struct btf_array *ap = btf_array(tp); struct array_type *array = tag__alloc(sizeof(*array)); if (array == NULL) @@ -201,81 +192,67 @@ static int create_new_array(struct btf_elf *btfe, void *ptr, uint32_t id) return -ENOMEM; } - array->nr_entries[0] = btf_elf__get32(btfe, &ap->nelems); + array->nr_entries[0] = ap->nelems; array->tag.tag = DW_TAG_array_type; - array->tag.type = btf_elf__get32(btfe, &ap->type); + array->tag.type = ap->type; cu__add_tag_with_id(btfe->priv, &array->tag, id); - return sizeof(*ap); + return 0; } -static int create_members(struct btf_elf *btfe, void *ptr, int vlen, struct type *class, - bool kflag) +static int create_members(struct btf_elf *btfe, const struct btf_type *tp, + struct type *class) { - struct btf_member *mp = ptr; - int i; + struct btf_member *mp = btf_members(tp); + int i, vlen = btf_vlen(tp); for (i = 0; i < vlen; i++) { struct class_member *member = zalloc(sizeof(*member)); - uint32_t offset; if (member == NULL) return -ENOMEM; member->tag.tag = DW_TAG_member; - member->tag.type = btf_elf__get32(btfe, &mp[i].type); - member->name = btf_elf__get32(btfe, &mp[i].name_off); - offset = btf_elf__get32(btfe, &mp[i].offset); - if (kflag) { - member->bit_offset = BTF_MEMBER_BIT_OFFSET(offset); - member->bitfield_size = BTF_MEMBER_BITFIELD_SIZE(offset); - } else { - member->bit_offset = offset; - member->bitfield_size = 0; - } + member->tag.type = mp[i].type; + member->name = mp[i].name_off; + member->bit_offset = btf_member_bit_offset(tp, i); + member->bitfield_size = btf_member_bitfield_size(tp, i); member->byte_offset = member->bit_offset / 8; /* sizes and offsets will be corrected at class__fixup_btf_bitfields */ type__add_member(class, member); } - return sizeof(*mp); + return 0; } -static int create_new_class(struct btf_elf *btfe, void *ptr, int vlen, - struct btf_type *tp, uint64_t size, uint32_t id, - bool kflag) +static int create_new_class(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - strings_t name = btf_elf__get32(btfe, &tp->name_off); - struct class *class = class__new(name, size); - int member_size = create_members(btfe, ptr, vlen, &class->type, kflag); + struct class *class = class__new(tp->name_off, tp->size); + int member_size = create_members(btfe, tp, &class->type); if (member_size < 0) goto out_free; cu__add_tag_with_id(btfe->priv, &class->type.namespace.tag, id); - return (vlen * member_size); + return 0; out_free: class__delete(class, btfe->priv); return -ENOMEM; } -static int create_new_union(struct btf_elf *btfe, void *ptr, - int vlen, struct btf_type *tp, - uint64_t size, uint32_t id, - bool kflag) +static int create_new_union(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - strings_t name = btf_elf__get32(btfe, &tp->name_off); - struct type *un = type__new(DW_TAG_union_type, name, size); - int member_size = create_members(btfe, ptr, vlen, un, kflag); + struct type *un = type__new(DW_TAG_union_type, tp->name_off, tp->size); + int member_size = create_members(btfe, tp, un); if (member_size < 0) goto out_free; cu__add_tag_with_id(btfe->priv, &un->namespace.tag, id); - return (vlen * member_size); + return 0; out_free: type__delete(un, btfe->priv); return -ENOMEM; @@ -294,22 +271,20 @@ static struct enumerator *enumerator__new(strings_t name, uint32_t value) return en; } -static int create_new_enumeration(struct btf_elf *btfe, void *ptr, - int vlen, struct btf_type *tp, - uint16_t size, uint32_t id) +static int create_new_enumeration(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - struct btf_enum *ep = ptr; - uint16_t i; + struct btf_enum *ep = btf_enum(tp); + uint16_t i, vlen = btf_vlen(tp); struct type *enumeration = type__new(DW_TAG_enumeration_type, - btf_elf__get32(btfe, &tp->name_off), - size ? size * 8 : (sizeof(int) * 8)); + tp->name_off, + tp->size ? tp->size * 8 : (sizeof(int) * 8)); if (enumeration == NULL) return -ENOMEM; for (i = 0; i < vlen; i++) { - strings_t name = btf_elf__get32(btfe, &ep[i].name_off); - uint32_t value = btf_elf__get32(btfe, (uint32_t *)&ep[i].val); + strings_t name = ep[i].name_off; + uint32_t value = ep[i].val; struct enumerator *enumerator = enumerator__new(name, value); if (enumerator == NULL) @@ -320,32 +295,25 @@ static int create_new_enumeration(struct btf_elf *btfe, void *ptr, cu__add_tag_with_id(btfe->priv, &enumeration->namespace.tag, id); - return (vlen * sizeof(*ep)); + return 0; out_free: enumeration__delete(enumeration, btfe->priv); return -ENOMEM; } -static int create_new_subroutine_type(struct btf_elf *btfe, void *ptr, - int vlen, struct btf_type *tp, - uint32_t id) +static int create_new_subroutine_type(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - struct btf_param *args = ptr; - unsigned int type = btf_elf__get32(btfe, &tp->type); struct ftype *proto = tag__alloc(sizeof(*proto)); if (proto == NULL) return -ENOMEM; - vlen = btf_elf__load_ftype(btfe, proto, DW_TAG_subroutine_type, type, vlen, args, id); - return vlen < 0 ? -ENOMEM : vlen; + return btf_elf__load_ftype(btfe, proto, DW_TAG_subroutine_type, tp, id); } -static int create_new_forward_decl(struct btf_elf *btfe, struct btf_type *tp, - uint64_t size, uint32_t id) +static int create_new_forward_decl(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - strings_t name = btf_elf__get32(btfe, &tp->name_off); - struct class *fwd = class__new(name, size); + struct class *fwd = class__new(tp->name_off, 0); if (fwd == NULL) return -ENOMEM; @@ -354,41 +322,33 @@ static int create_new_forward_decl(struct btf_elf *btfe, struct btf_type *tp, return 0; } -static int create_new_typedef(struct btf_elf *btfe, struct btf_type *tp, uint64_t size, uint32_t id) +static int create_new_typedef(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - strings_t name = btf_elf__get32(btfe, &tp->name_off); - unsigned int type_id = btf_elf__get32(btfe, &tp->type); - struct type *type = type__new(DW_TAG_typedef, name, size); + struct type *type = type__new(DW_TAG_typedef, tp->name_off, 0); if (type == NULL) return -ENOMEM; - type->namespace.tag.type = type_id; + type->namespace.tag.type = tp->type; cu__add_tag_with_id(btfe->priv, &type->namespace.tag, id); return 0; } -static int create_new_variable(struct btf_elf *btfe, void *ptr, struct btf_type *tp, - uint64_t size, uint32_t id) +static int create_new_variable(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { - strings_t name = btf_elf__get32(btfe, &tp->name_off); - unsigned int type_id = btf_elf__get32(btfe, &tp->type); - struct btf_var *bvar = ptr; - uint32_t linkage = btf_elf__get32(btfe, &bvar->linkage); - struct variable *var = variable__new(name, linkage); + struct btf_var *bvar = btf_var(tp); + struct variable *var = variable__new(tp->name_off, bvar->linkage); if (var == NULL) return -ENOMEM; - var->ip.tag.type = type_id; + var->ip.tag.type = tp->type; cu__add_tag_with_id(btfe->priv, &var->ip.tag, id); - return sizeof(*bvar); + return 0; } -static int create_new_datasec(struct btf_elf *btfe, void *ptr, int vlen, - struct btf_type *tp, uint64_t size, uint32_t id, - bool kflag) +static int create_new_datasec(struct btf_elf *btfe, const struct btf_type *tp, uint32_t id) { //strings_t name = btf_elf__get32(btfe, &tp->name_off); @@ -398,12 +358,11 @@ static int create_new_datasec(struct btf_elf *btfe, void *ptr, int vlen, * FIXME: this will not be used to reconstruct some original C code, * its about runtime placement of variables so just ignore this for now */ - return vlen * sizeof(struct btf_var_secinfo); + return 0; } -static int create_new_tag(struct btf_elf *btfe, int type, struct btf_type *tp, uint32_t id) +static int create_new_tag(struct btf_elf *btfe, int type, const struct btf_type *tp, uint32_t id) { - unsigned int type_id = btf_elf__get32(btfe, &tp->type); struct tag *tag = zalloc(sizeof(*tag)); if (tag == NULL) @@ -420,104 +379,77 @@ static int create_new_tag(struct btf_elf *btfe, int type, struct btf_type *tp, u return 0; } - tag->type = type_id; + tag->type = tp->type; cu__add_tag_with_id(btfe->priv, tag, id); return 0; } -void *btf_elf__get_buffer(struct btf_elf *btfe) -{ - return btfe->data; -} - -size_t btf_elf__get_size(struct btf_elf *btfe) -{ - return btfe->size; -} - static int btf_elf__load_types(struct btf_elf *btfe) { - void *btf_buffer = btf_elf__get_buffer(btfe); - struct btf_header *hp = btf_buffer; - void *btf_contents = btf_buffer + sizeof(*hp), - *type_section = (btf_contents + btf_elf__get32(btfe, &hp->type_off)), - *strings_section = (btf_contents + btf_elf__get32(btfe, &hp->str_off)); - struct btf_type *type_ptr = type_section, - *end = strings_section; - uint32_t type_index = 0x0001; - - while (type_ptr < end) { - uint32_t val = btf_elf__get32(btfe, &type_ptr->info); - uint32_t type = BTF_INFO_KIND(val); - int vlen = BTF_INFO_VLEN(val); - void *ptr = type_ptr; - uint32_t size = btf_elf__get32(btfe, &type_ptr->size); - bool kflag = BTF_INFO_KFLAG(val); - - ptr += sizeof(struct btf_type); + uint32_t type_index; + int err; + + for (type_index = 1; type_index <= btf__get_nr_types(btfe->btf); type_index++) { + const struct btf_type *type_ptr = btf__type_by_id(btfe->btf, type_index); + uint32_t type = btf_kind(type_ptr); switch (type) { case BTF_KIND_INT: - vlen = create_new_base_type(btfe, ptr, type_ptr, type_index); + err = create_new_base_type(btfe, type_ptr, type_index); break; case BTF_KIND_ARRAY: - vlen = create_new_array(btfe, ptr, type_index); + err = create_new_array(btfe, type_ptr, type_index); break; case BTF_KIND_STRUCT: - vlen = create_new_class(btfe, ptr, vlen, type_ptr, size, type_index, kflag); + err = create_new_class(btfe, type_ptr, type_index); break; case BTF_KIND_UNION: - vlen = create_new_union(btfe, ptr, vlen, type_ptr, size, type_index, kflag); + err = create_new_union(btfe, type_ptr, type_index); break; case BTF_KIND_ENUM: - vlen = create_new_enumeration(btfe, ptr, vlen, type_ptr, size, type_index); + err = create_new_enumeration(btfe, type_ptr, type_index); break; case BTF_KIND_FWD: - vlen = create_new_forward_decl(btfe, type_ptr, size, type_index); + err = create_new_forward_decl(btfe, type_ptr, type_index); break; case BTF_KIND_TYPEDEF: - vlen = create_new_typedef(btfe, type_ptr, size, type_index); + err = create_new_typedef(btfe, type_ptr, type_index); break; case BTF_KIND_VAR: - vlen = create_new_variable(btfe, ptr, type_ptr, size, type_index); + err = create_new_variable(btfe, type_ptr, type_index); break; case BTF_KIND_DATASEC: - vlen = create_new_datasec(btfe, ptr, vlen, type_ptr, size, type_index, kflag); + err = create_new_datasec(btfe, type_ptr, type_index); break; case BTF_KIND_VOLATILE: case BTF_KIND_PTR: case BTF_KIND_CONST: case BTF_KIND_RESTRICT: - vlen = create_new_tag(btfe, type, type_ptr, type_index); + err = create_new_tag(btfe, type, type_ptr, type_index); break; case BTF_KIND_UNKN: cu__table_nullify_type_entry(btfe->priv, type_index); - fprintf(stderr, "BTF: idx: %d, off: %zd, Unknown kind %d\n", - type_index, ((void *)type_ptr) - type_section, type); + fprintf(stderr, "BTF: idx: %d, Unknown kind %d\n", type_index, type); fflush(stderr); - vlen = 0; + err = 0; break; case BTF_KIND_FUNC_PROTO: - vlen = create_new_subroutine_type(btfe, ptr, vlen, type_ptr, type_index); + err = create_new_subroutine_type(btfe, type_ptr, type_index); break; case BTF_KIND_FUNC: // BTF_KIND_FUNC corresponding to a defined subprogram. - vlen = create_new_function(btfe, type_ptr, size, type_index); + err = create_new_function(btfe, type_ptr, type_index); break; default: - fprintf(stderr, "BTF: idx: %d, off: %zd, Unknown kind %d\n", - type_index, ((void *)type_ptr) - type_section, type); + fprintf(stderr, "BTF: idx: %d, Unknown kind %d\n", type_index, type); fflush(stderr); - vlen = 0; + err = 0; break; } - if (vlen < 0) - return vlen; - - type_ptr = ptr + vlen; - type_index++; + if (err < 0) + return err; } return 0; } diff --git a/libbtf.c b/libbtf.c index b80c482f6ccf..20f9074ad68b 100644 --- a/libbtf.c +++ b/libbtf.c @@ -62,89 +62,29 @@ static int btf_var_secinfo_cmp(const void *a, const void *b) return av->offset - bv->offset; } -uint32_t btf_elf__get32(struct btf_elf *btfe, uint32_t *p) -{ - uint32_t val = *p; - - if (btfe->swapped) - val = ((val >> 24) | - ((val >> 8) & 0x0000ff00) | - ((val << 8) & 0x00ff0000) | - (val << 24)); - return val; -} - -static int btf_raw__load(struct btf_elf *btfe) +static int libbpf_log(enum libbpf_print_level level, const char *format, va_list args) { - size_t read_cnt; - struct stat st; - void *data; - FILE *fp; - - if (stat(btfe->filename, &st)) - return -1; - - data = malloc(st.st_size); - if (!data) - return -1; - - fp = fopen(btfe->filename, "rb"); - if (!fp) - goto cleanup; - - read_cnt = fread(data, 1, st.st_size, fp); - fclose(fp); - if (read_cnt < st.st_size) - goto cleanup; - - btfe->swapped = 0; - btfe->data = data; - btfe->size = read_cnt; - return 0; -cleanup: - free(data); - return -1; + return vfprintf(stderr, format, args); } int btf_elf__load(struct btf_elf *btfe) { - if (btfe->raw_btf) - return btf_raw__load(btfe); - - int err = -ENOTSUP; - GElf_Shdr shdr; - Elf_Scn *sec = elf_section_by_name(btfe->elf, &btfe->ehdr, &shdr, ".BTF", NULL); + int err; - if (sec == NULL) - return -ESRCH; - - Elf_Data *data = elf_getdata(sec, NULL); - if (data == NULL) { - fprintf(stderr, "%s: cannot get data of BTF section.\n", __func__); - return -1; - } - - struct btf_header *hp = data->d_buf; - size_t orig_size = data->d_size; - - if (hp->version != BTF_VERSION) - goto out; + libbpf_set_print(libbpf_log); - err = -EINVAL; - if (hp->magic == BTF_MAGIC) - btfe->swapped = 0; + /* free initial empty BTF */ + btf__free(btfe->btf); + if (btfe->raw_btf) + btfe->btf = btf__parse_raw(btfe->filename); else - goto out; + btfe->btf = btf__parse_elf(btfe->filename, NULL); - err = -ENOMEM; - btfe->data = malloc(orig_size); - if (btfe->data != NULL) { - memcpy(btfe->data, hp, orig_size); - btfe->size = orig_size; - err = 0; - } -out: - return err; + err = libbpf_get_error(btfe->btf); + if (err) + return err; + + return 0; } @@ -252,26 +192,17 @@ void btf_elf__delete(struct btf_elf *btfe) __gobuffer__delete(&btfe->types); __gobuffer__delete(&btfe->percpu_secinfo); + btf__free(btfe->btf); free(btfe->filename); free(btfe->data); free(btfe); } -char *btf_elf__string(struct btf_elf *btfe, uint32_t ref) +const char *btf_elf__string(struct btf_elf *btfe, uint32_t ref) { - struct btf_header *hp = btfe->hdr; - uint32_t off = ref; - char *name; - - if (off >= btf_elf__get32(btfe, &hp->str_len)) - return "(ref out-of-bounds)"; - - if ((off + btf_elf__get32(btfe, &hp->str_off)) >= btfe->size) - return "(string table truncated)"; + const char *s = btf__str_by_offset(btfe->btf, ref); - name = ((char *)(hp + 1) + btf_elf__get32(btfe, &hp->str_off) + off); - - return name[0] == '\0' ? NULL : name; + return s && s[0] == '\0' ? NULL : s; } static void *btf_elf__nohdr_data(struct btf_elf *btfe) @@ -313,8 +244,10 @@ static const char *btf_elf__name_in_gobuf(const struct btf_elf *btfe, uint32_t o { if (!offset) return "(anon)"; - else + else if (btfe->strings) return &btfe->strings->entries[offset]; + else + return btf__str_by_offset(btfe->btf, offset); } static const char * btf_elf__int_encoding_str(uint8_t encoding) @@ -839,11 +772,6 @@ out: return err; } -static int libbpf_log(enum libbpf_print_level level, const char *format, va_list args) -{ - return vfprintf(stderr, format, args); -} - int btf_elf__encode(struct btf_elf *btfe, uint8_t flags) { struct btf_header *hdr; @@ -889,7 +817,7 @@ int btf_elf__encode(struct btf_elf *btfe, uint8_t flags) return -1; } if (btf__dedup(btf, NULL, NULL)) { - fprintf(stderr, "%s: btf__dedup failed!", __func__); + fprintf(stderr, "%s: btf__dedup failed!\n", __func__); return -1; } diff --git a/libbtf.h b/libbtf.h index be06480bf854..5f29b427c4fd 100644 --- a/libbtf.h +++ b/libbtf.h @@ -11,6 +11,7 @@ #include #include +#include "lib/bpf/src/btf.h" struct btf_elf { union { @@ -26,7 +27,6 @@ struct btf_elf { struct gobuffer percpu_secinfo; char *filename; size_t size; - int swapped; int in_fd; uint8_t wordsize; bool is_big_endian; @@ -34,6 +34,7 @@ struct btf_elf { uint32_t type_index; uint32_t percpu_shndx; uint64_t percpu_base_addr; + struct btf *btf; }; extern uint8_t btf_elf__verbose; @@ -70,13 +71,7 @@ int32_t btf_elf__add_datasec_type(struct btf_elf *btfe, const char *section_name void btf_elf__set_strings(struct btf_elf *btf, struct gobuffer *strings); int btf_elf__encode(struct btf_elf *btf, uint8_t flags); -char *btf_elf__string(struct btf_elf *btf, uint32_t ref); +const char *btf_elf__string(struct btf_elf *btf, uint32_t ref); int btf_elf__load(struct btf_elf *btf); -uint32_t btf_elf__get32(struct btf_elf *btf, uint32_t *p); - -void *btf_elf__get_buffer(struct btf_elf *btf); - -size_t btf_elf__get_size(struct btf_elf *btf); - #endif /* _LIBBTF_H */