1b2441318SGreg Kroah-Hartman /* SPDX-License-Identifier: GPL-2.0 */
2e00c7b21SDaniel Borkmann #ifndef __BPF_UTIL__
3e00c7b21SDaniel Borkmann #define __BPF_UTIL__
4e00c7b21SDaniel Borkmann
5e00c7b21SDaniel Borkmann #include <stdio.h>
6e00c7b21SDaniel Borkmann #include <stdlib.h>
7e00c7b21SDaniel Borkmann #include <string.h>
8e00c7b21SDaniel Borkmann #include <errno.h>
93e689141SToke Høiland-Jørgensen #include <bpf/libbpf.h> /* libbpf_num_possible_cpus */
10e00c7b21SDaniel Borkmann
bpf_num_possible_cpus(void)11e00c7b21SDaniel Borkmann static inline unsigned int bpf_num_possible_cpus(void)
12e00c7b21SDaniel Borkmann {
134c587c19SHechao Li int possible_cpus = libbpf_num_possible_cpus();
14e00c7b21SDaniel Borkmann
154c587c19SHechao Li if (possible_cpus < 0) {
164c587c19SHechao Li printf("Failed to get # of possible cpus: '%s'!\n",
174c587c19SHechao Li strerror(-possible_cpus));
18e00c7b21SDaniel Borkmann exit(1);
19e00c7b21SDaniel Borkmann }
20e00c7b21SDaniel Borkmann return possible_cpus;
21e00c7b21SDaniel Borkmann }
22e00c7b21SDaniel Borkmann
23*b3c09fdcSRong Tao /* Copy up to sz - 1 bytes from zero-terminated src string and ensure that dst
24*b3c09fdcSRong Tao * is zero-terminated string no matter what (unless sz == 0, in which case
25*b3c09fdcSRong Tao * it's a no-op). It's conceptually close to FreeBSD's strlcpy(), but differs
26*b3c09fdcSRong Tao * in what is returned. Given this is internal helper, it's trivial to extend
27*b3c09fdcSRong Tao * this, when necessary. Use this instead of strncpy inside libbpf source code.
28*b3c09fdcSRong Tao */
bpf_strlcpy(char * dst,const char * src,size_t sz)29*b3c09fdcSRong Tao static inline void bpf_strlcpy(char *dst, const char *src, size_t sz)
30*b3c09fdcSRong Tao {
31*b3c09fdcSRong Tao size_t i;
32*b3c09fdcSRong Tao
33*b3c09fdcSRong Tao if (sz == 0)
34*b3c09fdcSRong Tao return;
35*b3c09fdcSRong Tao
36*b3c09fdcSRong Tao sz--;
37*b3c09fdcSRong Tao for (i = 0; i < sz && src[i]; i++)
38*b3c09fdcSRong Tao dst[i] = src[i];
39*b3c09fdcSRong Tao dst[i] = '\0';
40*b3c09fdcSRong Tao }
41*b3c09fdcSRong Tao
42f3515b5dSDaniel Borkmann #define __bpf_percpu_val_align __attribute__((__aligned__(8)))
43f3515b5dSDaniel Borkmann
44f3515b5dSDaniel Borkmann #define BPF_DECLARE_PERCPU(type, name) \
45f3515b5dSDaniel Borkmann struct { type v; /* padding */ } __bpf_percpu_val_align \
46f3515b5dSDaniel Borkmann name[bpf_num_possible_cpus()]
47f3515b5dSDaniel Borkmann #define bpf_percpu(name, cpu) name[(cpu)].v
48f3515b5dSDaniel Borkmann
49aa5f0c96SMartin KaFai Lau #ifndef ARRAY_SIZE
50aa5f0c96SMartin KaFai Lau # define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0]))
51aa5f0c96SMartin KaFai Lau #endif
52aa5f0c96SMartin KaFai Lau
53fb47d1d9SMartin KaFai Lau #ifndef sizeof_field
54fb47d1d9SMartin KaFai Lau #define sizeof_field(TYPE, MEMBER) sizeof((((TYPE *)0)->MEMBER))
55fb47d1d9SMartin KaFai Lau #endif
56fb47d1d9SMartin KaFai Lau
57fb47d1d9SMartin KaFai Lau #ifndef offsetofend
58fb47d1d9SMartin KaFai Lau #define offsetofend(TYPE, MEMBER) \
59fb47d1d9SMartin KaFai Lau (offsetof(TYPE, MEMBER) + sizeof_field(TYPE, MEMBER))
60fb47d1d9SMartin KaFai Lau #endif
61fb47d1d9SMartin KaFai Lau
62e00c7b21SDaniel Borkmann #endif /* __BPF_UTIL__ */
63