173 lines
		
	
	
	
		
			4.4 KiB
			
		
	
	
	
		
			C
		
	
	
	
	
	
		
		
			
		
	
	
			173 lines
		
	
	
	
		
			4.4 KiB
			
		
	
	
	
		
			C
		
	
	
	
	
	
| 
								 | 
							
								/* eBPF mini library */
							 | 
						||
| 
								 | 
							
								#ifndef __LIBBPF_H
							 | 
						||
| 
								 | 
							
								#define __LIBBPF_H
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								struct bpf_insn;
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								int bpf_create_map(enum bpf_map_type map_type, int key_size, int value_size,
							 | 
						||
| 
								 | 
							
										   int max_entries);
							 | 
						||
| 
								 | 
							
								int bpf_update_elem(int fd, void *key, void *value);
							 | 
						||
| 
								 | 
							
								int bpf_lookup_elem(int fd, void *key, void *value);
							 | 
						||
| 
								 | 
							
								int bpf_delete_elem(int fd, void *key);
							 | 
						||
| 
								 | 
							
								int bpf_get_next_key(int fd, void *key, void *next_key);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								int bpf_prog_load(enum bpf_prog_type prog_type,
							 | 
						||
| 
								 | 
							
										  const struct bpf_insn *insns, int insn_len,
							 | 
						||
| 
								 | 
							
										  const char *license);
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define LOG_BUF_SIZE 8192
							 | 
						||
| 
								 | 
							
								extern char bpf_log_buf[LOG_BUF_SIZE];
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* ALU ops on registers, bpf_add|sub|...: dst_reg += src_reg */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_ALU64_REG(OP, DST, SRC)				\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_ALU64 | BPF_OP(OP) | BPF_X,	\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = SRC,					\
							 | 
						||
| 
								 | 
							
										.off   = 0,					\
							 | 
						||
| 
								 | 
							
										.imm   = 0 })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_ALU32_REG(OP, DST, SRC)				\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_ALU | BPF_OP(OP) | BPF_X,		\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = SRC,					\
							 | 
						||
| 
								 | 
							
										.off   = 0,					\
							 | 
						||
| 
								 | 
							
										.imm   = 0 })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* ALU ops on immediates, bpf_add|sub|...: dst_reg += imm32 */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_ALU64_IMM(OP, DST, IMM)				\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_ALU64 | BPF_OP(OP) | BPF_K,	\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = 0,					\
							 | 
						||
| 
								 | 
							
										.off   = 0,					\
							 | 
						||
| 
								 | 
							
										.imm   = IMM })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_ALU32_IMM(OP, DST, IMM)				\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_ALU | BPF_OP(OP) | BPF_K,		\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = 0,					\
							 | 
						||
| 
								 | 
							
										.off   = 0,					\
							 | 
						||
| 
								 | 
							
										.imm   = IMM })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* Short form of mov, dst_reg = src_reg */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_MOV64_REG(DST, SRC)					\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_ALU64 | BPF_MOV | BPF_X,		\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = SRC,					\
							 | 
						||
| 
								 | 
							
										.off   = 0,					\
							 | 
						||
| 
								 | 
							
										.imm   = 0 })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* Short form of mov, dst_reg = imm32 */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_MOV64_IMM(DST, IMM)					\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_ALU64 | BPF_MOV | BPF_K,		\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = 0,					\
							 | 
						||
| 
								 | 
							
										.off   = 0,					\
							 | 
						||
| 
								 | 
							
										.imm   = IMM })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* BPF_LD_IMM64 macro encodes single 'load 64-bit immediate' insn */
							 | 
						||
| 
								 | 
							
								#define BPF_LD_IMM64(DST, IMM)					\
							 | 
						||
| 
								 | 
							
									BPF_LD_IMM64_RAW(DST, 0, IMM)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_LD_IMM64_RAW(DST, SRC, IMM)				\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_LD | BPF_DW | BPF_IMM,		\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = SRC,					\
							 | 
						||
| 
								 | 
							
										.off   = 0,					\
							 | 
						||
| 
								 | 
							
										.imm   = (__u32) (IMM) }),			\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = 0, /* zero is reserved opcode */	\
							 | 
						||
| 
								 | 
							
										.dst_reg = 0,					\
							 | 
						||
| 
								 | 
							
										.src_reg = 0,					\
							 | 
						||
| 
								 | 
							
										.off   = 0,					\
							 | 
						||
| 
								 | 
							
										.imm   = ((__u64) (IMM)) >> 32 })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_PSEUDO_MAP_FD	1
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* pseudo BPF_LD_IMM64 insn used to refer to process-local map_fd */
							 | 
						||
| 
								 | 
							
								#define BPF_LD_MAP_FD(DST, MAP_FD)				\
							 | 
						||
| 
								 | 
							
									BPF_LD_IMM64_RAW(DST, BPF_PSEUDO_MAP_FD, MAP_FD)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* Memory load, dst_reg = *(uint *) (src_reg + off16) */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_LDX_MEM(SIZE, DST, SRC, OFF)			\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_LDX | BPF_SIZE(SIZE) | BPF_MEM,	\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = SRC,					\
							 | 
						||
| 
								 | 
							
										.off   = OFF,					\
							 | 
						||
| 
								 | 
							
										.imm   = 0 })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* Memory store, *(uint *) (dst_reg + off16) = src_reg */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_STX_MEM(SIZE, DST, SRC, OFF)			\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_STX | BPF_SIZE(SIZE) | BPF_MEM,	\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = SRC,					\
							 | 
						||
| 
								 | 
							
										.off   = OFF,					\
							 | 
						||
| 
								 | 
							
										.imm   = 0 })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* Memory store, *(uint *) (dst_reg + off16) = imm32 */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_ST_MEM(SIZE, DST, OFF, IMM)				\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_ST | BPF_SIZE(SIZE) | BPF_MEM,	\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = 0,					\
							 | 
						||
| 
								 | 
							
										.off   = OFF,					\
							 | 
						||
| 
								 | 
							
										.imm   = IMM })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* Conditional jumps against registers, if (dst_reg 'op' src_reg) goto pc + off16 */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_JMP_REG(OP, DST, SRC, OFF)				\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_JMP | BPF_OP(OP) | BPF_X,		\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = SRC,					\
							 | 
						||
| 
								 | 
							
										.off   = OFF,					\
							 | 
						||
| 
								 | 
							
										.imm   = 0 })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* Conditional jumps against immediates, if (dst_reg 'op' imm32) goto pc + off16 */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_JMP_IMM(OP, DST, IMM, OFF)				\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_JMP | BPF_OP(OP) | BPF_K,		\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = 0,					\
							 | 
						||
| 
								 | 
							
										.off   = OFF,					\
							 | 
						||
| 
								 | 
							
										.imm   = IMM })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* Raw code statement block */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_RAW_INSN(CODE, DST, SRC, OFF, IMM)			\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = CODE,					\
							 | 
						||
| 
								 | 
							
										.dst_reg = DST,					\
							 | 
						||
| 
								 | 
							
										.src_reg = SRC,					\
							 | 
						||
| 
								 | 
							
										.off   = OFF,					\
							 | 
						||
| 
								 | 
							
										.imm   = IMM })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								/* Program exit */
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#define BPF_EXIT_INSN()						\
							 | 
						||
| 
								 | 
							
									((struct bpf_insn) {					\
							 | 
						||
| 
								 | 
							
										.code  = BPF_JMP | BPF_EXIT,			\
							 | 
						||
| 
								 | 
							
										.dst_reg = 0,					\
							 | 
						||
| 
								 | 
							
										.src_reg = 0,					\
							 | 
						||
| 
								 | 
							
										.off   = 0,					\
							 | 
						||
| 
								 | 
							
										.imm   = 0 })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								#endif
							 |