mirror of https://github.com/aya-rs/aya
aya-obj: add conversion u32 to enum type for prog, link, & attach type
Add conversion from u32 to program type, link type, and attach type. Additionally, remove duplicate match statement for u32 conversion to `BPF_MAP_TYPE_BLOOM_FILTER` & `BPF_MAP_TYPE_CGRP_STORAGE`. New error `InvalidTypeBinding<T>` is created to represent when a parsed/received value binding to a type is invalid. This is used in the new conversions added here, and also replaces `InvalidMapTypeError` in `TryFrom` for `bpf_map_type`.pull/1007/head
tyrone-wu
2 months ago
parent
cb8e478800
commit
1634fa7188
@ -0,0 +1,100 @@
|
||||
//! Link type bindings.
|
||||
|
||||
use crate::{
|
||||
generated::{bpf_attach_type, bpf_link_type},
|
||||
InvalidTypeBinding,
|
||||
};
|
||||
|
||||
impl TryFrom<u32> for bpf_link_type {
|
||||
type Error = InvalidTypeBinding<u32>;
|
||||
|
||||
fn try_from(link_type: u32) -> Result<Self, Self::Error> {
|
||||
use bpf_link_type::*;
|
||||
Ok(match link_type {
|
||||
x if x == BPF_LINK_TYPE_UNSPEC as u32 => BPF_LINK_TYPE_UNSPEC,
|
||||
x if x == BPF_LINK_TYPE_RAW_TRACEPOINT as u32 => BPF_LINK_TYPE_RAW_TRACEPOINT,
|
||||
x if x == BPF_LINK_TYPE_TRACING as u32 => BPF_LINK_TYPE_TRACING,
|
||||
x if x == BPF_LINK_TYPE_CGROUP as u32 => BPF_LINK_TYPE_CGROUP,
|
||||
x if x == BPF_LINK_TYPE_ITER as u32 => BPF_LINK_TYPE_ITER,
|
||||
x if x == BPF_LINK_TYPE_NETNS as u32 => BPF_LINK_TYPE_NETNS,
|
||||
x if x == BPF_LINK_TYPE_XDP as u32 => BPF_LINK_TYPE_XDP,
|
||||
x if x == BPF_LINK_TYPE_PERF_EVENT as u32 => BPF_LINK_TYPE_PERF_EVENT,
|
||||
x if x == BPF_LINK_TYPE_KPROBE_MULTI as u32 => BPF_LINK_TYPE_KPROBE_MULTI,
|
||||
x if x == BPF_LINK_TYPE_STRUCT_OPS as u32 => BPF_LINK_TYPE_STRUCT_OPS,
|
||||
x if x == BPF_LINK_TYPE_NETFILTER as u32 => BPF_LINK_TYPE_NETFILTER,
|
||||
x if x == BPF_LINK_TYPE_TCX as u32 => BPF_LINK_TYPE_TCX,
|
||||
x if x == BPF_LINK_TYPE_UPROBE_MULTI as u32 => BPF_LINK_TYPE_UPROBE_MULTI,
|
||||
x if x == BPF_LINK_TYPE_NETKIT as u32 => BPF_LINK_TYPE_NETKIT,
|
||||
_ => return Err(InvalidTypeBinding { value: link_type }),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
impl TryFrom<u32> for bpf_attach_type {
|
||||
type Error = InvalidTypeBinding<u32>;
|
||||
|
||||
fn try_from(attach_type: u32) -> Result<Self, Self::Error> {
|
||||
use bpf_attach_type::*;
|
||||
Ok(match attach_type {
|
||||
x if x == BPF_CGROUP_INET_INGRESS as u32 => BPF_CGROUP_INET_INGRESS,
|
||||
x if x == BPF_CGROUP_INET_EGRESS as u32 => BPF_CGROUP_INET_EGRESS,
|
||||
x if x == BPF_CGROUP_INET_SOCK_CREATE as u32 => BPF_CGROUP_INET_SOCK_CREATE,
|
||||
x if x == BPF_CGROUP_SOCK_OPS as u32 => BPF_CGROUP_SOCK_OPS,
|
||||
x if x == BPF_SK_SKB_STREAM_PARSER as u32 => BPF_SK_SKB_STREAM_PARSER,
|
||||
x if x == BPF_SK_SKB_STREAM_VERDICT as u32 => BPF_SK_SKB_STREAM_VERDICT,
|
||||
x if x == BPF_CGROUP_DEVICE as u32 => BPF_CGROUP_DEVICE,
|
||||
x if x == BPF_SK_MSG_VERDICT as u32 => BPF_SK_MSG_VERDICT,
|
||||
x if x == BPF_CGROUP_INET4_BIND as u32 => BPF_CGROUP_INET4_BIND,
|
||||
x if x == BPF_CGROUP_INET6_BIND as u32 => BPF_CGROUP_INET6_BIND,
|
||||
x if x == BPF_CGROUP_INET4_CONNECT as u32 => BPF_CGROUP_INET4_CONNECT,
|
||||
x if x == BPF_CGROUP_INET6_CONNECT as u32 => BPF_CGROUP_INET6_CONNECT,
|
||||
x if x == BPF_CGROUP_INET4_POST_BIND as u32 => BPF_CGROUP_INET4_POST_BIND,
|
||||
x if x == BPF_CGROUP_INET6_POST_BIND as u32 => BPF_CGROUP_INET6_POST_BIND,
|
||||
x if x == BPF_CGROUP_UDP4_SENDMSG as u32 => BPF_CGROUP_UDP4_SENDMSG,
|
||||
x if x == BPF_CGROUP_UDP6_SENDMSG as u32 => BPF_CGROUP_UDP6_SENDMSG,
|
||||
x if x == BPF_LIRC_MODE2 as u32 => BPF_LIRC_MODE2,
|
||||
x if x == BPF_FLOW_DISSECTOR as u32 => BPF_FLOW_DISSECTOR,
|
||||
x if x == BPF_CGROUP_SYSCTL as u32 => BPF_CGROUP_SYSCTL,
|
||||
x if x == BPF_CGROUP_UDP4_RECVMSG as u32 => BPF_CGROUP_UDP4_RECVMSG,
|
||||
x if x == BPF_CGROUP_UDP6_RECVMSG as u32 => BPF_CGROUP_UDP6_RECVMSG,
|
||||
x if x == BPF_CGROUP_GETSOCKOPT as u32 => BPF_CGROUP_GETSOCKOPT,
|
||||
x if x == BPF_CGROUP_SETSOCKOPT as u32 => BPF_CGROUP_SETSOCKOPT,
|
||||
x if x == BPF_TRACE_RAW_TP as u32 => BPF_TRACE_RAW_TP,
|
||||
x if x == BPF_TRACE_FENTRY as u32 => BPF_TRACE_FENTRY,
|
||||
x if x == BPF_TRACE_FEXIT as u32 => BPF_TRACE_FEXIT,
|
||||
x if x == BPF_MODIFY_RETURN as u32 => BPF_MODIFY_RETURN,
|
||||
x if x == BPF_LSM_MAC as u32 => BPF_LSM_MAC,
|
||||
x if x == BPF_TRACE_ITER as u32 => BPF_TRACE_ITER,
|
||||
x if x == BPF_CGROUP_INET4_GETPEERNAME as u32 => BPF_CGROUP_INET4_GETPEERNAME,
|
||||
x if x == BPF_CGROUP_INET6_GETPEERNAME as u32 => BPF_CGROUP_INET6_GETPEERNAME,
|
||||
x if x == BPF_CGROUP_INET4_GETSOCKNAME as u32 => BPF_CGROUP_INET4_GETSOCKNAME,
|
||||
x if x == BPF_CGROUP_INET6_GETSOCKNAME as u32 => BPF_CGROUP_INET6_GETSOCKNAME,
|
||||
x if x == BPF_XDP_DEVMAP as u32 => BPF_XDP_DEVMAP,
|
||||
x if x == BPF_CGROUP_INET_SOCK_RELEASE as u32 => BPF_CGROUP_INET_SOCK_RELEASE,
|
||||
x if x == BPF_XDP_CPUMAP as u32 => BPF_XDP_CPUMAP,
|
||||
x if x == BPF_SK_LOOKUP as u32 => BPF_SK_LOOKUP,
|
||||
x if x == BPF_XDP as u32 => BPF_XDP,
|
||||
x if x == BPF_SK_SKB_VERDICT as u32 => BPF_SK_SKB_VERDICT,
|
||||
x if x == BPF_SK_REUSEPORT_SELECT as u32 => BPF_SK_REUSEPORT_SELECT,
|
||||
x if x == BPF_SK_REUSEPORT_SELECT_OR_MIGRATE as u32 => {
|
||||
BPF_SK_REUSEPORT_SELECT_OR_MIGRATE
|
||||
}
|
||||
x if x == BPF_PERF_EVENT as u32 => BPF_PERF_EVENT,
|
||||
x if x == BPF_TRACE_KPROBE_MULTI as u32 => BPF_TRACE_KPROBE_MULTI,
|
||||
x if x == BPF_LSM_CGROUP as u32 => BPF_LSM_CGROUP,
|
||||
x if x == BPF_STRUCT_OPS as u32 => BPF_STRUCT_OPS,
|
||||
x if x == BPF_NETFILTER as u32 => BPF_NETFILTER,
|
||||
x if x == BPF_TCX_INGRESS as u32 => BPF_TCX_INGRESS,
|
||||
x if x == BPF_TCX_EGRESS as u32 => BPF_TCX_EGRESS,
|
||||
x if x == BPF_TRACE_UPROBE_MULTI as u32 => BPF_TRACE_UPROBE_MULTI,
|
||||
x if x == BPF_CGROUP_UNIX_CONNECT as u32 => BPF_CGROUP_UNIX_CONNECT,
|
||||
x if x == BPF_CGROUP_UNIX_SENDMSG as u32 => BPF_CGROUP_UNIX_SENDMSG,
|
||||
x if x == BPF_CGROUP_UNIX_RECVMSG as u32 => BPF_CGROUP_UNIX_RECVMSG,
|
||||
x if x == BPF_CGROUP_UNIX_GETPEERNAME as u32 => BPF_CGROUP_UNIX_GETPEERNAME,
|
||||
x if x == BPF_CGROUP_UNIX_GETSOCKNAME as u32 => BPF_CGROUP_UNIX_GETSOCKNAME,
|
||||
x if x == BPF_NETKIT_PRIMARY as u32 => BPF_NETKIT_PRIMARY,
|
||||
x if x == BPF_NETKIT_PEER as u32 => BPF_NETKIT_PEER,
|
||||
_ => return Err(InvalidTypeBinding { value: attach_type }),
|
||||
})
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,51 @@
|
||||
//! Program type bindings.
|
||||
|
||||
use crate::{
|
||||
generated::bpf_prog_type::{self, *},
|
||||
InvalidTypeBinding,
|
||||
};
|
||||
|
||||
impl TryFrom<u32> for bpf_prog_type {
|
||||
type Error = InvalidTypeBinding<u32>;
|
||||
|
||||
fn try_from(prog_type: u32) -> Result<Self, Self::Error> {
|
||||
Ok(match prog_type {
|
||||
x if x == BPF_PROG_TYPE_UNSPEC as u32 => BPF_PROG_TYPE_UNSPEC,
|
||||
x if x == BPF_PROG_TYPE_SOCKET_FILTER as u32 => BPF_PROG_TYPE_SOCKET_FILTER,
|
||||
x if x == BPF_PROG_TYPE_KPROBE as u32 => BPF_PROG_TYPE_KPROBE,
|
||||
x if x == BPF_PROG_TYPE_SCHED_CLS as u32 => BPF_PROG_TYPE_SCHED_CLS,
|
||||
x if x == BPF_PROG_TYPE_SCHED_ACT as u32 => BPF_PROG_TYPE_SCHED_ACT,
|
||||
x if x == BPF_PROG_TYPE_TRACEPOINT as u32 => BPF_PROG_TYPE_TRACEPOINT,
|
||||
x if x == BPF_PROG_TYPE_XDP as u32 => BPF_PROG_TYPE_XDP,
|
||||
x if x == BPF_PROG_TYPE_PERF_EVENT as u32 => BPF_PROG_TYPE_PERF_EVENT,
|
||||
x if x == BPF_PROG_TYPE_CGROUP_SKB as u32 => BPF_PROG_TYPE_CGROUP_SKB,
|
||||
x if x == BPF_PROG_TYPE_CGROUP_SOCK as u32 => BPF_PROG_TYPE_CGROUP_SOCK,
|
||||
x if x == BPF_PROG_TYPE_LWT_IN as u32 => BPF_PROG_TYPE_LWT_IN,
|
||||
x if x == BPF_PROG_TYPE_LWT_OUT as u32 => BPF_PROG_TYPE_LWT_OUT,
|
||||
x if x == BPF_PROG_TYPE_LWT_XMIT as u32 => BPF_PROG_TYPE_LWT_XMIT,
|
||||
x if x == BPF_PROG_TYPE_SOCK_OPS as u32 => BPF_PROG_TYPE_SOCK_OPS,
|
||||
x if x == BPF_PROG_TYPE_SK_SKB as u32 => BPF_PROG_TYPE_SK_SKB,
|
||||
x if x == BPF_PROG_TYPE_CGROUP_DEVICE as u32 => BPF_PROG_TYPE_CGROUP_DEVICE,
|
||||
x if x == BPF_PROG_TYPE_SK_MSG as u32 => BPF_PROG_TYPE_SK_MSG,
|
||||
x if x == BPF_PROG_TYPE_RAW_TRACEPOINT as u32 => BPF_PROG_TYPE_RAW_TRACEPOINT,
|
||||
x if x == BPF_PROG_TYPE_CGROUP_SOCK_ADDR as u32 => BPF_PROG_TYPE_CGROUP_SOCK_ADDR,
|
||||
x if x == BPF_PROG_TYPE_LWT_SEG6LOCAL as u32 => BPF_PROG_TYPE_LWT_SEG6LOCAL,
|
||||
x if x == BPF_PROG_TYPE_LIRC_MODE2 as u32 => BPF_PROG_TYPE_LIRC_MODE2,
|
||||
x if x == BPF_PROG_TYPE_SK_REUSEPORT as u32 => BPF_PROG_TYPE_SK_REUSEPORT,
|
||||
x if x == BPF_PROG_TYPE_FLOW_DISSECTOR as u32 => BPF_PROG_TYPE_FLOW_DISSECTOR,
|
||||
x if x == BPF_PROG_TYPE_CGROUP_SYSCTL as u32 => BPF_PROG_TYPE_CGROUP_SYSCTL,
|
||||
x if x == BPF_PROG_TYPE_RAW_TRACEPOINT_WRITABLE as u32 => {
|
||||
BPF_PROG_TYPE_RAW_TRACEPOINT_WRITABLE
|
||||
}
|
||||
x if x == BPF_PROG_TYPE_CGROUP_SOCKOPT as u32 => BPF_PROG_TYPE_CGROUP_SOCKOPT,
|
||||
x if x == BPF_PROG_TYPE_TRACING as u32 => BPF_PROG_TYPE_TRACING,
|
||||
x if x == BPF_PROG_TYPE_STRUCT_OPS as u32 => BPF_PROG_TYPE_STRUCT_OPS,
|
||||
x if x == BPF_PROG_TYPE_EXT as u32 => BPF_PROG_TYPE_EXT,
|
||||
x if x == BPF_PROG_TYPE_LSM as u32 => BPF_PROG_TYPE_LSM,
|
||||
x if x == BPF_PROG_TYPE_SK_LOOKUP as u32 => BPF_PROG_TYPE_SK_LOOKUP,
|
||||
x if x == BPF_PROG_TYPE_SYSCALL as u32 => BPF_PROG_TYPE_SYSCALL,
|
||||
x if x == BPF_PROG_TYPE_NETFILTER as u32 => BPF_PROG_TYPE_NETFILTER,
|
||||
_ => return Err(InvalidTypeBinding { value: prog_type }),
|
||||
})
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue