codegen: tidy up

Move some code out of a loop, where it appears to be nonsense.
aya-rs · Feb 4, 2025 · 9aa08b4 · 9aa08b4
1 parent 2056e7d
commit 9aa08b4
Show file tree

Hide file tree

Showing 5 changed files with 200 additions and 188 deletions.
diff --git a/aya-obj/include/linux_wrapper.h b/aya-obj/include/linux_wrapper.h
@@ -1,4 +1,3 @@
-#include <asm-generic/socket.h>
 #include <linux/bpf.h>
 #include <linux/btf.h>
 #include <linux/if_link.h>
@@ -7,6 +6,7 @@
 #include <linux/pkt_cls.h>
 #include <linux/pkt_sched.h>
 #include <linux/rtnetlink.h>
+#include <sys/socket.h>
 
 /* workaround the fact that bindgen can't parse the IOC macros */
 int AYA_PERF_EVENT_IOC_ENABLE = PERF_EVENT_IOC_ENABLE;

diff --git a/ebpf/aya-ebpf-bindings/include/bindings.h b/ebpf/aya-ebpf-bindings/include/bindings.h
@@ -1,9 +1,10 @@
 #include <linux/types.h>
-// __wsum is missing from types.h, see
-// https://elixir.bootlin.com/linux/v5.13/source/include/uapi/linux/types.h
+// __wsum is missing from types.h, compare:
+// https://github.com/torvalds/linux/blob/v5.13/include/uapi/linux/types.h
+// https://github.com/libbpf/libbpf/blob/v1.5.0/include/linux/types.h
 typedef __u32 __bitwise __wsum;
 
-#include "bpf_helpers.h"
+#include <bpf/bpf_helpers.h>
 #include <linux/bpf.h>
 // needed for TC_ACT_*
 #include <linux/pkt_cls.h>

diff --git a/xtask/src/codegen/aya.rs b/xtask/src/codegen/aya.rs
@@ -1,16 +1,16 @@
 use std::path::{Path, PathBuf};
 
-use anyhow::anyhow;
+use anyhow::{Context as _, Result};
 use aya_tool::{bindgen, write_to_file};
 
 use crate::codegen::{Architecture, SysrootOptions};
 
-pub fn codegen(opts: &SysrootOptions, libbpf_dir: &Path) -> Result<(), anyhow::Error> {
+pub fn codegen(opts: &SysrootOptions, libbpf_dir: &Path) -> Result<()> {
     codegen_internal_btf_bindings(libbpf_dir)?;
     codegen_bindings(opts, libbpf_dir)
 }
 
-fn codegen_internal_btf_bindings(libbpf_dir: &Path) -> Result<(), anyhow::Error> {
+fn codegen_internal_btf_bindings(libbpf_dir: &Path) -> Result<()> {
     let dir = PathBuf::from("aya-obj");
     let generated = dir.join("src/generated");
 
@@ -26,18 +26,15 @@ fn codegen_internal_btf_bindings(libbpf_dir: &Path) -> Result<(), anyhow::Error>
         bindgen = bindgen.allowlist_type(x);
     }
 
-    let bindings = bindgen
-        .generate()
-        .map_err(|op| anyhow!("bindgen failed - {op}"))?
-        .to_string();
+    let bindings = bindgen.generate().context("bindgen failed")?.to_string();
 
     // write the bindings, with the original helpers removed
     write_to_file(generated.join("btf_internal_bindings.rs"), &bindings)?;
 
     Ok(())
 }
 
-fn codegen_bindings(opts: &SysrootOptions, libbpf_dir: &Path) -> Result<(), anyhow::Error> {
+fn codegen_bindings(opts: &SysrootOptions, libbpf_dir: &Path) -> Result<()> {
     let SysrootOptions {
         x86_64_sysroot,
         aarch64_sysroot,
@@ -47,132 +44,154 @@ fn codegen_bindings(opts: &SysrootOptions, libbpf_dir: &Path) -> Result<(), anyh
         s390x_sysroot,
         mips_sysroot,
     } = opts;
-    let types = [
-        // BPF
-        "BPF_TYPES",
-        "bpf_cmd",
-        "bpf_insn",
-        "bpf_attr",
-        "bpf_map_type",
-        "bpf_prog_type",
-        "bpf_attach_type",
-        "bpf_prog_info",
-        "bpf_map_info",
-        "bpf_link_info",
-        "bpf_link_type",
-        "bpf_btf_info",
-        "bpf_func_id",
-        "bpf_func_info",
-        "bpf_line_info",
-        "bpf_lpm_trie_key",
-        "bpf_cpumap_val",
-        "bpf_devmap_val",
-        "bpf_stats_type",
-        "bpf_perf_event_type",
-        "bpf_task_fd_type",
-        // BTF
-        "btf_header",
-        "btf_ext_info",
-        "btf_ext_info_sec",
-        "btf_type",
-        "btf_enum",
-        "btf_array",
-        "btf_member",
-        "btf_param",
-        "btf_var",
-        "btf_var_secinfo",
-        "btf_func_linkage",
-        "btf_decl_tag",
-        // PERF
-        "perf_event_attr",
-        "perf_sw_ids",
-        "perf_hw_id",
-        "perf_hw_cache_id",
-        "perf_hw_cache_op_id",
-        "perf_hw_cache_op_result_id",
-        "perf_event_sample_format",
-        "perf_event_mmap_page",
-        "perf_event_header",
-        "perf_type_id",
-        "perf_event_type",
-        // NETLINK
-        "ifinfomsg",
-        "tcmsg",
-        "nlmsgerr_attrs",
-        // ITER
-        "bpf_cgroup_iter_order",
-        // NETFILTER
-        "nf_inet_hooks",
-    ];
-
-    let vars = [
-        // BPF
-        "BPF_PSEUDO_.*",
-        "BPF_ALU",
-        "BPF_ALU64",
-        "BPF_LDX",
-        "BPF_ST",
-        "BPF_STX",
-        "BPF_LD",
-        "BPF_K",
-        "BPF_X",
-        "BPF_DW",
-        "BPF_W",
-        "BPF_H",
-        "BPF_B",
-        "BPF_IMM",
-        "BPF_MEM",
-        "BPF_SUB",
-        "BPF_MOV",
-        "BPF_F_.*",
-        "BPF_JMP",
-        "BPF_CALL",
-        "BPF_EXIT",
-        "SO_ATTACH_BPF",
-        "SO_DETACH_BPF",
-        // BTF
-        "BTF_INT_.*",
-        "BTF_KIND_.*",
-        "BTF_VAR_.*",
-        // PERF
-        "PERF_FLAG_.*",
-        "PERF_EVENT_.*",
-        "PERF_MAX_.*",
-        // see linux_wrapper.h, these are to workaround the IOC macros
-        "AYA_PERF_EVENT_.*",
-        // NETLINK
-        "NLMSG_ALIGNTO",
-        "IFLA_XDP_FD",
-        "TCA_KIND",
-        "TCA_OPTIONS",
-        "TCA_BPF_FD",
-        "TCA_BPF_NAME",
-        "TCA_BPF_FLAGS",
-        "TCA_BPF_FLAG_ACT_DIRECT",
-        "XDP_FLAGS_.*",
-        "TC_H_MAJ_MASK",
-        "TC_H_MIN_MASK",
-        "TC_H_UNSPEC",
-        "TC_H_ROOT",
-        "TC_H_INGRESS",
-        "TC_H_CLSACT",
-        "TC_H_MIN_PRIORITY",
-        "TC_H_MIN_INGRESS",
-        "TC_H_MIN_EGRESS",
-        // Ringbuf
-        "BPF_RINGBUF_.*",
-        // NETFILTER
-        "NFPROTO_.*",
-    ];
-
     let dir = PathBuf::from("aya-obj");
     let generated = dir.join("src/generated");
 
     let builder = || {
-        bindgen::user_builder()
+        let mut bindgen = bindgen::user_builder()
             .header(dir.join("include/linux_wrapper.h").to_str().unwrap())
             .clang_args(["-I", libbpf_dir.join("include/uapi").to_str().unwrap()])
             .clang_args(["-I", libbpf_dir.join("include").to_str().unwrap()])
+            // BPF_F_LINK is defined twice. Once in an anonymous enum
+            // which bindgen will constify, and once via #define macro
+            // which generates a duplicate const.
+            .blocklist_var("BPF_F_LINK")
+            .constified_enum("BPF_F_.*")
+            .constified_enum("BTF_KIND_.*")
+            .constified_enum("BTF_VAR_.*")
+            .constified_enum("IFLA_.*")
+            .constified_enum("TCA_.*")
+            .constified_enum("BPF_RINGBUF_.*")
+            // NETFILTER
+            .constified_enum("NFPROTO_.*");
+
+        let types = [
+            // BPF
+            "bpf_cmd",
+            "bpf_insn",
+            "bpf_attr",
+            "bpf_map_type",
+            "bpf_prog_type",
+            "bpf_attach_type",
+            "bpf_prog_info",
+            "bpf_map_info",
+            "bpf_link_info",
+            "bpf_link_type",
+            "bpf_btf_info",
+            "bpf_func_id",
+            "bpf_func_info",
+            "bpf_line_info",
+            "bpf_lpm_trie_key",
+            "bpf_cpumap_val",
+            "bpf_devmap_val",
+            "bpf_stats_type",
+            "bpf_perf_event_type",
+            "bpf_task_fd_type",
+            // BTF
+            "btf_header",
+            "btf_ext_info",
+            "btf_ext_info_sec",
+            "btf_type",
+            "btf_enum",
+            "btf_array",
+            "btf_member",
+            "btf_param",
+            "btf_var",
+            "btf_var_secinfo",
+            "btf_func_linkage",
+            "btf_decl_tag",
+            // PERF
+            "perf_event_attr",
+            "perf_sw_ids",
+            "perf_hw_id",
+            "perf_hw_cache_id",
+            "perf_hw_cache_op_id",
+            "perf_hw_cache_op_result_id",
+            "perf_event_sample_format",
+            "perf_event_mmap_page",
+            "perf_event_header",
+            "perf_type_id",
+            "perf_event_type",
+            // NETLINK
+            "ifinfomsg",
+            "tcmsg",
+            "nlmsgerr_attrs",
+            // ITER
+            "bpf_cgroup_iter_order",
+            // NETFILTER
+            "nf_inet_hooks",
+        ];
+
+        let vars = [
+            // BPF
+            "BPF_FUNC_.*",
+            "BPF_PSEUDO_.*",
+            "BPF_ALU",
+            "BPF_ALU64",
+            "BPF_LDX",
+            "BPF_ST",
+            "BPF_STX",
+            "BPF_LD",
+            "BPF_K",
+            "BPF_X",
+            "BPF_DW",
+            "BPF_W",
+            "BPF_H",
+            "BPF_B",
+            "BPF_IMM",
+            "BPF_MEM",
+            "BPF_SUB",
+            "BPF_MOV",
+            "BPF_F_.*",
+            "BPF_JMP",
+            "BPF_CALL",
+            "BPF_EXIT",
+            "SO_ATTACH_BPF",
+            "SO_DETACH_BPF",
+            // BTF
+            "BTF_INT_.*",
+            "BTF_KIND_.*",
+            "BTF_VAR_.*",
+            // PERF
+            "PERF_FLAG_.*",
+            "PERF_EVENT_.*",
+            "PERF_MAX_.*",
+            // see linux_wrapper.h, these are to workaround the IOC macros
+            "AYA_PERF_EVENT_.*",
+            // NETLINK
+            "NLMSG_ALIGNTO",
+            "IFLA_XDP_FD",
+            "TCA_KIND",
+            "TCA_OPTIONS",
+            "TCA_BPF_FD",
+            "TCA_BPF_NAME",
+            "TCA_BPF_FLAGS",
+            "TCA_BPF_FLAG_ACT_DIRECT",
+            "XDP_FLAGS_.*",
+            "TC_H_MAJ_MASK",
+            "TC_H_MIN_MASK",
+            "TC_H_UNSPEC",
+            "TC_H_ROOT",
+            "TC_H_INGRESS",
+            "TC_H_CLSACT",
+            "TC_H_MIN_PRIORITY",
+            "TC_H_MIN_INGRESS",
+            "TC_H_MIN_EGRESS",
+            // Ringbuf
+            "BPF_RINGBUF_.*",
+            // NETFILTER
+            "NFPROTO_.*",
+        ];
+
+        for x in &types {
+            bindgen = bindgen.allowlist_type(x);
+        }
+
+        for x in &vars {
+            bindgen = bindgen.allowlist_var(x);
+        }
+
+        bindgen
     };
 
     for arch in Architecture::supported() {
@@ -204,38 +223,7 @@ fn codegen_bindings(opts: &SysrootOptions, libbpf_dir: &Path) -> Result<(), anyh
         };
         bindgen = bindgen.clang_args(["-I", sysroot.to_str().unwrap()]);
 
-        for x in &types {
-            bindgen = bindgen.allowlist_type(x);
-        }
-        for x in &vars {
-            bindgen = bindgen
-                .allowlist_var(x)
-                // BPF_F_LINK is defined twice. Once in an anonymous enum
-                // which bindgen will constify, and once via #define macro
-                // which generates a duplicate const.
-                .blocklist_var("BPF_F_LINK")
-                .constified_enum("BPF_F_.*")
-                .constified_enum("BTF_KIND_.*")
-                .constified_enum("BTF_VAR_.*")
-                .constified_enum("IFLA_.*")
-                .constified_enum("TCA_.*")
-                .constified_enum("BPF_RINGBUF_.*")
-                // NETFILTER
-                .constified_enum("NFPROTO_.*");
-        }
-
-        for x in &types {
-            bindgen = bindgen.allowlist_type(x);
-        }
-
-        for x in &vars {
-            bindgen = bindgen.allowlist_var(x);
-        }
-
-        let bindings = bindgen
-            .generate()
-            .map_err(|op| anyhow!("bindgen failed - {op}"))?
-            .to_string();
+        let bindings = bindgen.generate().context("bindgen failed")?.to_string();
 
         // write the bindings, with the original helpers removed
         write_to_file(