657 строки
16 KiB
Plaintext
657 строки
16 KiB
Plaintext
/*
|
|
* BPF asm code parser
|
|
*
|
|
* This program is free software; you can distribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published
|
|
* by the Free Software Foundation; either version 2 of the License,
|
|
* or (at your option) any later version.
|
|
*
|
|
* Syntax kept close to:
|
|
*
|
|
* Steven McCanne and Van Jacobson. 1993. The BSD packet filter: a new
|
|
* architecture for user-level packet capture. In Proceedings of the
|
|
* USENIX Winter 1993 Conference Proceedings on USENIX Winter 1993
|
|
* Conference Proceedings (USENIX'93). USENIX Association, Berkeley,
|
|
* CA, USA, 2-2.
|
|
*
|
|
* Copyright 2013 Daniel Borkmann <borkmann@redhat.com>
|
|
* Licensed under the GNU General Public License, version 2.0 (GPLv2)
|
|
*/
|
|
|
|
%{
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <stdbool.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <assert.h>
|
|
#include <linux/filter.h>
|
|
|
|
#include "bpf_exp.yacc.h"
|
|
|
|
enum jmp_type { JTL, JFL, JKL };
|
|
|
|
extern FILE *yyin;
|
|
extern int yylineno;
|
|
extern int yylex(void);
|
|
extern void yyerror(const char *str);
|
|
|
|
extern void bpf_asm_compile(FILE *fp, bool cstyle);
|
|
static void bpf_set_curr_instr(uint16_t op, uint8_t jt, uint8_t jf, uint32_t k);
|
|
static void bpf_set_curr_label(char *label);
|
|
static void bpf_set_jmp_label(char *label, enum jmp_type type);
|
|
|
|
%}
|
|
|
|
%union {
|
|
char *label;
|
|
uint32_t number;
|
|
}
|
|
|
|
%token OP_LDB OP_LDH OP_LD OP_LDX OP_ST OP_STX OP_JMP OP_JEQ OP_JGT OP_JGE
|
|
%token OP_JSET OP_ADD OP_SUB OP_MUL OP_DIV OP_AND OP_OR OP_XOR OP_LSH OP_RSH
|
|
%token OP_RET OP_TAX OP_TXA OP_LDXB OP_MOD OP_NEG OP_JNEQ OP_JLT OP_JLE OP_LDI
|
|
%token OP_LDXI
|
|
|
|
%token K_PKT_LEN
|
|
|
|
%token ':' ',' '[' ']' '(' ')' 'x' 'a' '+' 'M' '*' '&' '#' '%'
|
|
|
|
%token extension number label
|
|
|
|
%type <label> label
|
|
%type <number> extension
|
|
%type <number> number
|
|
|
|
%%
|
|
|
|
prog
|
|
: line
|
|
| prog line
|
|
;
|
|
|
|
line
|
|
: instr
|
|
| labelled_instr
|
|
;
|
|
|
|
labelled_instr
|
|
: labelled instr
|
|
;
|
|
|
|
instr
|
|
: ldb
|
|
| ldh
|
|
| ld
|
|
| ldi
|
|
| ldx
|
|
| ldxi
|
|
| st
|
|
| stx
|
|
| jmp
|
|
| jeq
|
|
| jneq
|
|
| jlt
|
|
| jle
|
|
| jgt
|
|
| jge
|
|
| jset
|
|
| add
|
|
| sub
|
|
| mul
|
|
| div
|
|
| mod
|
|
| neg
|
|
| and
|
|
| or
|
|
| xor
|
|
| lsh
|
|
| rsh
|
|
| ret
|
|
| tax
|
|
| txa
|
|
;
|
|
|
|
labelled
|
|
: label ':' { bpf_set_curr_label($1); }
|
|
;
|
|
|
|
ldb
|
|
: OP_LDB '[' 'x' '+' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_B | BPF_IND, 0, 0, $5); }
|
|
| OP_LDB '[' '%' 'x' '+' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_B | BPF_IND, 0, 0, $6); }
|
|
| OP_LDB '[' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_B | BPF_ABS, 0, 0, $3); }
|
|
| OP_LDB extension {
|
|
bpf_set_curr_instr(BPF_LD | BPF_B | BPF_ABS, 0, 0,
|
|
SKF_AD_OFF + $2); }
|
|
;
|
|
|
|
ldh
|
|
: OP_LDH '[' 'x' '+' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_H | BPF_IND, 0, 0, $5); }
|
|
| OP_LDH '[' '%' 'x' '+' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_H | BPF_IND, 0, 0, $6); }
|
|
| OP_LDH '[' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_H | BPF_ABS, 0, 0, $3); }
|
|
| OP_LDH extension {
|
|
bpf_set_curr_instr(BPF_LD | BPF_H | BPF_ABS, 0, 0,
|
|
SKF_AD_OFF + $2); }
|
|
;
|
|
|
|
ldi
|
|
: OP_LDI '#' number {
|
|
bpf_set_curr_instr(BPF_LD | BPF_IMM, 0, 0, $3); }
|
|
| OP_LDI number {
|
|
bpf_set_curr_instr(BPF_LD | BPF_IMM, 0, 0, $2); }
|
|
;
|
|
|
|
ld
|
|
: OP_LD '#' number {
|
|
bpf_set_curr_instr(BPF_LD | BPF_IMM, 0, 0, $3); }
|
|
| OP_LD K_PKT_LEN {
|
|
bpf_set_curr_instr(BPF_LD | BPF_W | BPF_LEN, 0, 0, 0); }
|
|
| OP_LD extension {
|
|
bpf_set_curr_instr(BPF_LD | BPF_W | BPF_ABS, 0, 0,
|
|
SKF_AD_OFF + $2); }
|
|
| OP_LD 'M' '[' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_MEM, 0, 0, $4); }
|
|
| OP_LD '[' 'x' '+' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_W | BPF_IND, 0, 0, $5); }
|
|
| OP_LD '[' '%' 'x' '+' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_W | BPF_IND, 0, 0, $6); }
|
|
| OP_LD '[' number ']' {
|
|
bpf_set_curr_instr(BPF_LD | BPF_W | BPF_ABS, 0, 0, $3); }
|
|
;
|
|
|
|
ldxi
|
|
: OP_LDXI '#' number {
|
|
bpf_set_curr_instr(BPF_LDX | BPF_IMM, 0, 0, $3); }
|
|
| OP_LDXI number {
|
|
bpf_set_curr_instr(BPF_LDX | BPF_IMM, 0, 0, $2); }
|
|
;
|
|
|
|
ldx
|
|
: OP_LDX '#' number {
|
|
bpf_set_curr_instr(BPF_LDX | BPF_IMM, 0, 0, $3); }
|
|
| OP_LDX K_PKT_LEN {
|
|
bpf_set_curr_instr(BPF_LDX | BPF_W | BPF_LEN, 0, 0, 0); }
|
|
| OP_LDX 'M' '[' number ']' {
|
|
bpf_set_curr_instr(BPF_LDX | BPF_MEM, 0, 0, $4); }
|
|
| OP_LDXB number '*' '(' '[' number ']' '&' number ')' {
|
|
if ($2 != 4 || $9 != 0xf) {
|
|
fprintf(stderr, "ldxb offset not supported!\n");
|
|
exit(0);
|
|
} else {
|
|
bpf_set_curr_instr(BPF_LDX | BPF_MSH | BPF_B, 0, 0, $6); } }
|
|
| OP_LDX number '*' '(' '[' number ']' '&' number ')' {
|
|
if ($2 != 4 || $9 != 0xf) {
|
|
fprintf(stderr, "ldxb offset not supported!\n");
|
|
exit(0);
|
|
} else {
|
|
bpf_set_curr_instr(BPF_LDX | BPF_MSH | BPF_B, 0, 0, $6); } }
|
|
;
|
|
|
|
st
|
|
: OP_ST 'M' '[' number ']' {
|
|
bpf_set_curr_instr(BPF_ST, 0, 0, $4); }
|
|
;
|
|
|
|
stx
|
|
: OP_STX 'M' '[' number ']' {
|
|
bpf_set_curr_instr(BPF_STX, 0, 0, $4); }
|
|
;
|
|
|
|
jmp
|
|
: OP_JMP label {
|
|
bpf_set_jmp_label($2, JKL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JA, 0, 0, 0); }
|
|
;
|
|
|
|
jeq
|
|
: OP_JEQ '#' number ',' label ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_jmp_label($7, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JEQ | BPF_K, 0, 0, $3); }
|
|
| OP_JEQ 'x' ',' label ',' label {
|
|
bpf_set_jmp_label($4, JTL);
|
|
bpf_set_jmp_label($6, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JEQ | BPF_X, 0, 0, 0); }
|
|
| OP_JEQ '%' 'x' ',' label ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_jmp_label($7, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JEQ | BPF_X, 0, 0, 0); }
|
|
| OP_JEQ '#' number ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JEQ | BPF_K, 0, 0, $3); }
|
|
| OP_JEQ 'x' ',' label {
|
|
bpf_set_jmp_label($4, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JEQ | BPF_X, 0, 0, 0); }
|
|
| OP_JEQ '%' 'x' ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JEQ | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
jneq
|
|
: OP_JNEQ '#' number ',' label {
|
|
bpf_set_jmp_label($5, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JEQ | BPF_K, 0, 0, $3); }
|
|
| OP_JNEQ 'x' ',' label {
|
|
bpf_set_jmp_label($4, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JEQ | BPF_X, 0, 0, 0); }
|
|
| OP_JNEQ '%' 'x' ',' label {
|
|
bpf_set_jmp_label($5, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JEQ | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
jlt
|
|
: OP_JLT '#' number ',' label {
|
|
bpf_set_jmp_label($5, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGE | BPF_K, 0, 0, $3); }
|
|
| OP_JLT 'x' ',' label {
|
|
bpf_set_jmp_label($4, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGE | BPF_X, 0, 0, 0); }
|
|
| OP_JLT '%' 'x' ',' label {
|
|
bpf_set_jmp_label($5, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGE | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
jle
|
|
: OP_JLE '#' number ',' label {
|
|
bpf_set_jmp_label($5, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGT | BPF_K, 0, 0, $3); }
|
|
| OP_JLE 'x' ',' label {
|
|
bpf_set_jmp_label($4, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGT | BPF_X, 0, 0, 0); }
|
|
| OP_JLE '%' 'x' ',' label {
|
|
bpf_set_jmp_label($5, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGT | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
jgt
|
|
: OP_JGT '#' number ',' label ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_jmp_label($7, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGT | BPF_K, 0, 0, $3); }
|
|
| OP_JGT 'x' ',' label ',' label {
|
|
bpf_set_jmp_label($4, JTL);
|
|
bpf_set_jmp_label($6, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGT | BPF_X, 0, 0, 0); }
|
|
| OP_JGT '%' 'x' ',' label ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_jmp_label($7, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGT | BPF_X, 0, 0, 0); }
|
|
| OP_JGT '#' number ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGT | BPF_K, 0, 0, $3); }
|
|
| OP_JGT 'x' ',' label {
|
|
bpf_set_jmp_label($4, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGT | BPF_X, 0, 0, 0); }
|
|
| OP_JGT '%' 'x' ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGT | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
jge
|
|
: OP_JGE '#' number ',' label ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_jmp_label($7, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGE | BPF_K, 0, 0, $3); }
|
|
| OP_JGE 'x' ',' label ',' label {
|
|
bpf_set_jmp_label($4, JTL);
|
|
bpf_set_jmp_label($6, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGE | BPF_X, 0, 0, 0); }
|
|
| OP_JGE '%' 'x' ',' label ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_jmp_label($7, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGE | BPF_X, 0, 0, 0); }
|
|
| OP_JGE '#' number ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGE | BPF_K, 0, 0, $3); }
|
|
| OP_JGE 'x' ',' label {
|
|
bpf_set_jmp_label($4, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGE | BPF_X, 0, 0, 0); }
|
|
| OP_JGE '%' 'x' ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JGE | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
jset
|
|
: OP_JSET '#' number ',' label ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_jmp_label($7, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JSET | BPF_K, 0, 0, $3); }
|
|
| OP_JSET 'x' ',' label ',' label {
|
|
bpf_set_jmp_label($4, JTL);
|
|
bpf_set_jmp_label($6, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JSET | BPF_X, 0, 0, 0); }
|
|
| OP_JSET '%' 'x' ',' label ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_jmp_label($7, JFL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JSET | BPF_X, 0, 0, 0); }
|
|
| OP_JSET '#' number ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JSET | BPF_K, 0, 0, $3); }
|
|
| OP_JSET 'x' ',' label {
|
|
bpf_set_jmp_label($4, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JSET | BPF_X, 0, 0, 0); }
|
|
| OP_JSET '%' 'x' ',' label {
|
|
bpf_set_jmp_label($5, JTL);
|
|
bpf_set_curr_instr(BPF_JMP | BPF_JSET | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
add
|
|
: OP_ADD '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_ADD | BPF_K, 0, 0, $3); }
|
|
| OP_ADD 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_ADD | BPF_X, 0, 0, 0); }
|
|
| OP_ADD '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_ADD | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
sub
|
|
: OP_SUB '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_SUB | BPF_K, 0, 0, $3); }
|
|
| OP_SUB 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_SUB | BPF_X, 0, 0, 0); }
|
|
| OP_SUB '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_SUB | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
mul
|
|
: OP_MUL '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_MUL | BPF_K, 0, 0, $3); }
|
|
| OP_MUL 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_MUL | BPF_X, 0, 0, 0); }
|
|
| OP_MUL '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_MUL | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
div
|
|
: OP_DIV '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_DIV | BPF_K, 0, 0, $3); }
|
|
| OP_DIV 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_DIV | BPF_X, 0, 0, 0); }
|
|
| OP_DIV '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_DIV | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
mod
|
|
: OP_MOD '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_MOD | BPF_K, 0, 0, $3); }
|
|
| OP_MOD 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_MOD | BPF_X, 0, 0, 0); }
|
|
| OP_MOD '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_MOD | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
neg
|
|
: OP_NEG {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_NEG, 0, 0, 0); }
|
|
;
|
|
|
|
and
|
|
: OP_AND '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_AND | BPF_K, 0, 0, $3); }
|
|
| OP_AND 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_AND | BPF_X, 0, 0, 0); }
|
|
| OP_AND '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_AND | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
or
|
|
: OP_OR '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_OR | BPF_K, 0, 0, $3); }
|
|
| OP_OR 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_OR | BPF_X, 0, 0, 0); }
|
|
| OP_OR '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_OR | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
xor
|
|
: OP_XOR '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_XOR | BPF_K, 0, 0, $3); }
|
|
| OP_XOR 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_XOR | BPF_X, 0, 0, 0); }
|
|
| OP_XOR '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_XOR | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
lsh
|
|
: OP_LSH '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_LSH | BPF_K, 0, 0, $3); }
|
|
| OP_LSH 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_LSH | BPF_X, 0, 0, 0); }
|
|
| OP_LSH '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_LSH | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
rsh
|
|
: OP_RSH '#' number {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_RSH | BPF_K, 0, 0, $3); }
|
|
| OP_RSH 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_RSH | BPF_X, 0, 0, 0); }
|
|
| OP_RSH '%' 'x' {
|
|
bpf_set_curr_instr(BPF_ALU | BPF_RSH | BPF_X, 0, 0, 0); }
|
|
;
|
|
|
|
ret
|
|
: OP_RET 'a' {
|
|
bpf_set_curr_instr(BPF_RET | BPF_A, 0, 0, 0); }
|
|
| OP_RET '%' 'a' {
|
|
bpf_set_curr_instr(BPF_RET | BPF_A, 0, 0, 0); }
|
|
| OP_RET 'x' {
|
|
bpf_set_curr_instr(BPF_RET | BPF_X, 0, 0, 0); }
|
|
| OP_RET '%' 'x' {
|
|
bpf_set_curr_instr(BPF_RET | BPF_X, 0, 0, 0); }
|
|
| OP_RET '#' number {
|
|
bpf_set_curr_instr(BPF_RET | BPF_K, 0, 0, $3); }
|
|
;
|
|
|
|
tax
|
|
: OP_TAX {
|
|
bpf_set_curr_instr(BPF_MISC | BPF_TAX, 0, 0, 0); }
|
|
;
|
|
|
|
txa
|
|
: OP_TXA {
|
|
bpf_set_curr_instr(BPF_MISC | BPF_TXA, 0, 0, 0); }
|
|
;
|
|
|
|
%%
|
|
|
|
static int curr_instr = 0;
|
|
static struct sock_filter out[BPF_MAXINSNS];
|
|
static char **labels, **labels_jt, **labels_jf, **labels_k;
|
|
|
|
static void bpf_assert_max(void)
|
|
{
|
|
if (curr_instr >= BPF_MAXINSNS) {
|
|
fprintf(stderr, "only max %u insns allowed!\n", BPF_MAXINSNS);
|
|
exit(0);
|
|
}
|
|
}
|
|
|
|
static void bpf_set_curr_instr(uint16_t code, uint8_t jt, uint8_t jf,
|
|
uint32_t k)
|
|
{
|
|
bpf_assert_max();
|
|
out[curr_instr].code = code;
|
|
out[curr_instr].jt = jt;
|
|
out[curr_instr].jf = jf;
|
|
out[curr_instr].k = k;
|
|
curr_instr++;
|
|
}
|
|
|
|
static void bpf_set_curr_label(char *label)
|
|
{
|
|
bpf_assert_max();
|
|
labels[curr_instr] = label;
|
|
}
|
|
|
|
static void bpf_set_jmp_label(char *label, enum jmp_type type)
|
|
{
|
|
bpf_assert_max();
|
|
switch (type) {
|
|
case JTL:
|
|
labels_jt[curr_instr] = label;
|
|
break;
|
|
case JFL:
|
|
labels_jf[curr_instr] = label;
|
|
break;
|
|
case JKL:
|
|
labels_k[curr_instr] = label;
|
|
break;
|
|
}
|
|
}
|
|
|
|
static int bpf_find_insns_offset(const char *label)
|
|
{
|
|
int i, max = curr_instr, ret = -ENOENT;
|
|
|
|
for (i = 0; i < max; i++) {
|
|
if (labels[i] && !strcmp(label, labels[i])) {
|
|
ret = i;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (ret == -ENOENT) {
|
|
fprintf(stderr, "no such label \'%s\'!\n", label);
|
|
exit(0);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static void bpf_stage_1_insert_insns(void)
|
|
{
|
|
yyparse();
|
|
}
|
|
|
|
static void bpf_reduce_k_jumps(void)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < curr_instr; i++) {
|
|
if (labels_k[i]) {
|
|
int off = bpf_find_insns_offset(labels_k[i]);
|
|
out[i].k = (uint32_t) (off - i - 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void bpf_reduce_jt_jumps(void)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < curr_instr; i++) {
|
|
if (labels_jt[i]) {
|
|
int off = bpf_find_insns_offset(labels_jt[i]);
|
|
out[i].jt = (uint8_t) (off - i -1);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void bpf_reduce_jf_jumps(void)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < curr_instr; i++) {
|
|
if (labels_jf[i]) {
|
|
int off = bpf_find_insns_offset(labels_jf[i]);
|
|
out[i].jf = (uint8_t) (off - i - 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void bpf_stage_2_reduce_labels(void)
|
|
{
|
|
bpf_reduce_k_jumps();
|
|
bpf_reduce_jt_jumps();
|
|
bpf_reduce_jf_jumps();
|
|
}
|
|
|
|
static void bpf_pretty_print_c(void)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < curr_instr; i++)
|
|
printf("{ %#04x, %2u, %2u, %#010x },\n", out[i].code,
|
|
out[i].jt, out[i].jf, out[i].k);
|
|
}
|
|
|
|
static void bpf_pretty_print(void)
|
|
{
|
|
int i;
|
|
|
|
printf("%u,", curr_instr);
|
|
for (i = 0; i < curr_instr; i++)
|
|
printf("%u %u %u %u,", out[i].code,
|
|
out[i].jt, out[i].jf, out[i].k);
|
|
printf("\n");
|
|
}
|
|
|
|
static void bpf_init(void)
|
|
{
|
|
memset(out, 0, sizeof(out));
|
|
|
|
labels = calloc(BPF_MAXINSNS, sizeof(*labels));
|
|
assert(labels);
|
|
labels_jt = calloc(BPF_MAXINSNS, sizeof(*labels_jt));
|
|
assert(labels_jt);
|
|
labels_jf = calloc(BPF_MAXINSNS, sizeof(*labels_jf));
|
|
assert(labels_jf);
|
|
labels_k = calloc(BPF_MAXINSNS, sizeof(*labels_k));
|
|
assert(labels_k);
|
|
}
|
|
|
|
static void bpf_destroy_labels(void)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < curr_instr; i++) {
|
|
free(labels_jf[i]);
|
|
free(labels_jt[i]);
|
|
free(labels_k[i]);
|
|
free(labels[i]);
|
|
}
|
|
}
|
|
|
|
static void bpf_destroy(void)
|
|
{
|
|
bpf_destroy_labels();
|
|
free(labels_jt);
|
|
free(labels_jf);
|
|
free(labels_k);
|
|
free(labels);
|
|
}
|
|
|
|
void bpf_asm_compile(FILE *fp, bool cstyle)
|
|
{
|
|
yyin = fp;
|
|
|
|
bpf_init();
|
|
bpf_stage_1_insert_insns();
|
|
bpf_stage_2_reduce_labels();
|
|
bpf_destroy();
|
|
|
|
if (cstyle)
|
|
bpf_pretty_print_c();
|
|
else
|
|
bpf_pretty_print();
|
|
|
|
if (fp != stdin)
|
|
fclose(yyin);
|
|
}
|
|
|
|
void yyerror(const char *str)
|
|
{
|
|
fprintf(stderr, "error: %s at line %d\n", str, yylineno);
|
|
exit(1);
|
|
}
|