Home | History | Annotate | Line # | Download | only in net
bpfjit.c revision 1.36.2.1
      1  1.36.2.1     skrll /*	$NetBSD: bpfjit.c,v 1.36.2.1 2015/04/06 15:18:22 skrll Exp $	*/
      2       1.3     rmind 
      3       1.1     alnsn /*-
      4  1.36.2.1     skrll  * Copyright (c) 2011-2015 Alexander Nasonov.
      5       1.1     alnsn  * All rights reserved.
      6       1.1     alnsn  *
      7       1.1     alnsn  * Redistribution and use in source and binary forms, with or without
      8       1.1     alnsn  * modification, are permitted provided that the following conditions
      9       1.1     alnsn  * are met:
     10       1.1     alnsn  *
     11       1.1     alnsn  * 1. Redistributions of source code must retain the above copyright
     12       1.1     alnsn  *    notice, this list of conditions and the following disclaimer.
     13       1.1     alnsn  * 2. Redistributions in binary form must reproduce the above copyright
     14       1.1     alnsn  *    notice, this list of conditions and the following disclaimer in
     15       1.1     alnsn  *    the documentation and/or other materials provided with the
     16       1.1     alnsn  *    distribution.
     17       1.1     alnsn  *
     18       1.1     alnsn  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     19       1.1     alnsn  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     20       1.1     alnsn  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
     21       1.1     alnsn  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
     22       1.1     alnsn  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
     23       1.1     alnsn  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
     24       1.1     alnsn  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     25       1.1     alnsn  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
     26       1.1     alnsn  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
     27       1.1     alnsn  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
     28       1.1     alnsn  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29       1.1     alnsn  * SUCH DAMAGE.
     30       1.1     alnsn  */
     31       1.1     alnsn 
     32       1.2     alnsn #include <sys/cdefs.h>
     33       1.2     alnsn #ifdef _KERNEL
     34  1.36.2.1     skrll __KERNEL_RCSID(0, "$NetBSD: bpfjit.c,v 1.36.2.1 2015/04/06 15:18:22 skrll Exp $");
     35       1.2     alnsn #else
     36  1.36.2.1     skrll __RCSID("$NetBSD: bpfjit.c,v 1.36.2.1 2015/04/06 15:18:22 skrll Exp $");
     37       1.2     alnsn #endif
     38       1.2     alnsn 
     39       1.3     rmind #include <sys/types.h>
     40       1.3     rmind #include <sys/queue.h>
     41       1.1     alnsn 
     42       1.1     alnsn #ifndef _KERNEL
     43       1.7     alnsn #include <assert.h>
     44       1.7     alnsn #define BJ_ASSERT(c) assert(c)
     45       1.7     alnsn #else
     46       1.7     alnsn #define BJ_ASSERT(c) KASSERT(c)
     47       1.7     alnsn #endif
     48       1.7     alnsn 
     49       1.7     alnsn #ifndef _KERNEL
     50       1.3     rmind #include <stdlib.h>
     51       1.7     alnsn #define BJ_ALLOC(sz) malloc(sz)
     52       1.7     alnsn #define BJ_FREE(p, sz) free(p)
     53       1.1     alnsn #else
     54       1.3     rmind #include <sys/kmem.h>
     55       1.7     alnsn #define BJ_ALLOC(sz) kmem_alloc(sz, KM_SLEEP)
     56       1.7     alnsn #define BJ_FREE(p, sz) kmem_free(p, sz)
     57       1.1     alnsn #endif
     58       1.1     alnsn 
     59       1.1     alnsn #ifndef _KERNEL
     60       1.1     alnsn #include <limits.h>
     61       1.1     alnsn #include <stdbool.h>
     62       1.1     alnsn #include <stddef.h>
     63       1.1     alnsn #include <stdint.h>
     64       1.1     alnsn #else
     65       1.1     alnsn #include <sys/atomic.h>
     66       1.1     alnsn #include <sys/module.h>
     67       1.1     alnsn #endif
     68       1.1     alnsn 
     69       1.5     rmind #define	__BPF_PRIVATE
     70       1.5     rmind #include <net/bpf.h>
     71       1.3     rmind #include <net/bpfjit.h>
     72       1.1     alnsn #include <sljitLir.h>
     73       1.1     alnsn 
     74       1.7     alnsn #if !defined(_KERNEL) && defined(SLJIT_VERBOSE) && SLJIT_VERBOSE
     75       1.7     alnsn #include <stdio.h> /* for stderr */
     76       1.7     alnsn #endif
     77       1.7     alnsn 
     78       1.7     alnsn /*
     79      1.13     alnsn  * Arguments of generated bpfjit_func_t.
     80      1.13     alnsn  * The first argument is reassigned upon entry
     81      1.13     alnsn  * to a more frequently used buf argument.
     82      1.13     alnsn  */
     83      1.13     alnsn #define BJ_CTX_ARG	SLJIT_SAVED_REG1
     84      1.13     alnsn #define BJ_ARGS		SLJIT_SAVED_REG2
     85      1.13     alnsn 
     86      1.13     alnsn /*
     87       1.7     alnsn  * Permanent register assignments.
     88       1.7     alnsn  */
     89       1.7     alnsn #define BJ_BUF		SLJIT_SAVED_REG1
     90      1.13     alnsn //#define BJ_ARGS	SLJIT_SAVED_REG2
     91       1.7     alnsn #define BJ_BUFLEN	SLJIT_SAVED_REG3
     92      1.12     alnsn #define BJ_AREG		SLJIT_SCRATCH_REG1
     93      1.12     alnsn #define BJ_TMP1REG	SLJIT_SCRATCH_REG2
     94      1.12     alnsn #define BJ_TMP2REG	SLJIT_SCRATCH_REG3
     95      1.32     alnsn #define BJ_XREG		SLJIT_TEMPORARY_EREG1
     96      1.32     alnsn #define BJ_TMP3REG	SLJIT_TEMPORARY_EREG2
     97       1.7     alnsn 
     98      1.13     alnsn #ifdef _KERNEL
     99      1.13     alnsn #define MAX_MEMWORDS BPF_MAX_MEMWORDS
    100      1.13     alnsn #else
    101      1.13     alnsn #define MAX_MEMWORDS BPF_MEMWORDS
    102      1.13     alnsn #endif
    103      1.13     alnsn 
    104      1.13     alnsn #define BJ_INIT_NOBITS  ((bpf_memword_init_t)0)
    105      1.13     alnsn #define BJ_INIT_MBIT(k) BPF_MEMWORD_INIT(k)
    106      1.13     alnsn #define BJ_INIT_ABIT    BJ_INIT_MBIT(MAX_MEMWORDS)
    107      1.13     alnsn #define BJ_INIT_XBIT    BJ_INIT_MBIT(MAX_MEMWORDS + 1)
    108       1.1     alnsn 
    109       1.9     alnsn /*
    110      1.19     alnsn  * Get a number of memwords and external memwords from a bpf_ctx object.
    111      1.19     alnsn  */
    112      1.19     alnsn #define GET_EXTWORDS(bc) ((bc) ? (bc)->extwords : 0)
    113      1.19     alnsn #define GET_MEMWORDS(bc) (GET_EXTWORDS(bc) ? GET_EXTWORDS(bc) : BPF_MEMWORDS)
    114      1.19     alnsn 
    115      1.19     alnsn /*
    116      1.20     alnsn  * Optimization hints.
    117      1.20     alnsn  */
    118      1.20     alnsn typedef unsigned int bpfjit_hint_t;
    119      1.28     alnsn #define BJ_HINT_ABS  0x01 /* packet read at absolute offset   */
    120      1.28     alnsn #define BJ_HINT_IND  0x02 /* packet read at variable offset   */
    121      1.29     alnsn #define BJ_HINT_MSH  0x04 /* BPF_MSH instruction              */
    122      1.29     alnsn #define BJ_HINT_COP  0x08 /* BPF_COP or BPF_COPX instruction  */
    123      1.29     alnsn #define BJ_HINT_COPX 0x10 /* BPF_COPX instruction             */
    124      1.29     alnsn #define BJ_HINT_XREG 0x20 /* BJ_XREG is needed                */
    125      1.29     alnsn #define BJ_HINT_LDX  0x40 /* BPF_LDX instruction              */
    126      1.29     alnsn #define BJ_HINT_PKT  (BJ_HINT_ABS|BJ_HINT_IND|BJ_HINT_MSH)
    127      1.20     alnsn 
    128      1.20     alnsn /*
    129       1.9     alnsn  * Datatype for Array Bounds Check Elimination (ABC) pass.
    130       1.9     alnsn  */
    131       1.9     alnsn typedef uint64_t bpfjit_abc_length_t;
    132       1.9     alnsn #define MAX_ABC_LENGTH (UINT32_MAX + UINT64_C(4)) /* max. width is 4 */
    133       1.8     alnsn 
    134       1.7     alnsn struct bpfjit_stack
    135       1.7     alnsn {
    136      1.13     alnsn 	bpf_ctx_t *ctx;
    137      1.13     alnsn 	uint32_t *extmem; /* pointer to external memory store */
    138      1.32     alnsn 	uint32_t reg; /* saved A or X register */
    139       1.7     alnsn #ifdef _KERNEL
    140      1.21     alnsn 	int err; /* 3rd argument for m_xword/m_xhalf/m_xbyte function call */
    141       1.7     alnsn #endif
    142      1.13     alnsn 	uint32_t mem[BPF_MEMWORDS]; /* internal memory store */
    143       1.7     alnsn };
    144       1.7     alnsn 
    145       1.7     alnsn /*
    146       1.7     alnsn  * Data for BPF_JMP instruction.
    147       1.7     alnsn  * Forward declaration for struct bpfjit_jump.
    148       1.1     alnsn  */
    149       1.7     alnsn struct bpfjit_jump_data;
    150       1.1     alnsn 
    151       1.1     alnsn /*
    152       1.7     alnsn  * Node of bjumps list.
    153       1.1     alnsn  */
    154       1.3     rmind struct bpfjit_jump {
    155       1.7     alnsn 	struct sljit_jump *sjump;
    156       1.7     alnsn 	SLIST_ENTRY(bpfjit_jump) entries;
    157       1.7     alnsn 	struct bpfjit_jump_data *jdata;
    158       1.1     alnsn };
    159       1.1     alnsn 
    160       1.1     alnsn /*
    161       1.1     alnsn  * Data for BPF_JMP instruction.
    162       1.1     alnsn  */
    163       1.3     rmind struct bpfjit_jump_data {
    164       1.1     alnsn 	/*
    165       1.7     alnsn 	 * These entries make up bjumps list:
    166       1.7     alnsn 	 * jtf[0] - when coming from jt path,
    167       1.7     alnsn 	 * jtf[1] - when coming from jf path.
    168       1.1     alnsn 	 */
    169       1.7     alnsn 	struct bpfjit_jump jtf[2];
    170       1.7     alnsn 	/*
    171       1.7     alnsn 	 * Length calculated by Array Bounds Check Elimination (ABC) pass.
    172       1.7     alnsn 	 */
    173       1.8     alnsn 	bpfjit_abc_length_t abc_length;
    174       1.7     alnsn 	/*
    175       1.7     alnsn 	 * Length checked by the last out-of-bounds check.
    176       1.7     alnsn 	 */
    177       1.8     alnsn 	bpfjit_abc_length_t checked_length;
    178       1.1     alnsn };
    179       1.1     alnsn 
    180       1.1     alnsn /*
    181       1.1     alnsn  * Data for "read from packet" instructions.
    182       1.1     alnsn  * See also read_pkt_insn() function below.
    183       1.1     alnsn  */
    184       1.3     rmind struct bpfjit_read_pkt_data {
    185       1.1     alnsn 	/*
    186       1.7     alnsn 	 * Length calculated by Array Bounds Check Elimination (ABC) pass.
    187       1.7     alnsn 	 */
    188       1.8     alnsn 	bpfjit_abc_length_t abc_length;
    189       1.7     alnsn 	/*
    190       1.7     alnsn 	 * If positive, emit "if (buflen < check_length) return 0"
    191       1.7     alnsn 	 * out-of-bounds check.
    192       1.9     alnsn 	 * Values greater than UINT32_MAX generate unconditional "return 0".
    193       1.1     alnsn 	 */
    194       1.8     alnsn 	bpfjit_abc_length_t check_length;
    195       1.1     alnsn };
    196       1.1     alnsn 
    197       1.1     alnsn /*
    198       1.1     alnsn  * Additional (optimization-related) data for bpf_insn.
    199       1.1     alnsn  */
    200       1.3     rmind struct bpfjit_insn_data {
    201       1.1     alnsn 	/* List of jumps to this insn. */
    202       1.7     alnsn 	SLIST_HEAD(, bpfjit_jump) bjumps;
    203       1.1     alnsn 
    204       1.1     alnsn 	union {
    205       1.7     alnsn 		struct bpfjit_jump_data     jdata;
    206       1.7     alnsn 		struct bpfjit_read_pkt_data rdata;
    207       1.7     alnsn 	} u;
    208       1.1     alnsn 
    209      1.13     alnsn 	bpf_memword_init_t invalid;
    210       1.7     alnsn 	bool unreachable;
    211       1.1     alnsn };
    212       1.1     alnsn 
    213       1.1     alnsn #ifdef _KERNEL
    214       1.1     alnsn 
    215       1.1     alnsn uint32_t m_xword(const struct mbuf *, uint32_t, int *);
    216       1.1     alnsn uint32_t m_xhalf(const struct mbuf *, uint32_t, int *);
    217       1.1     alnsn uint32_t m_xbyte(const struct mbuf *, uint32_t, int *);
    218       1.1     alnsn 
    219       1.1     alnsn MODULE(MODULE_CLASS_MISC, bpfjit, "sljit")
    220       1.1     alnsn 
    221       1.1     alnsn static int
    222       1.1     alnsn bpfjit_modcmd(modcmd_t cmd, void *arg)
    223       1.1     alnsn {
    224       1.1     alnsn 
    225       1.1     alnsn 	switch (cmd) {
    226       1.1     alnsn 	case MODULE_CMD_INIT:
    227       1.1     alnsn 		bpfjit_module_ops.bj_free_code = &bpfjit_free_code;
    228       1.1     alnsn 		membar_producer();
    229       1.1     alnsn 		bpfjit_module_ops.bj_generate_code = &bpfjit_generate_code;
    230       1.1     alnsn 		membar_producer();
    231       1.1     alnsn 		return 0;
    232       1.1     alnsn 
    233       1.1     alnsn 	case MODULE_CMD_FINI:
    234       1.1     alnsn 		return EOPNOTSUPP;
    235       1.1     alnsn 
    236       1.1     alnsn 	default:
    237       1.1     alnsn 		return ENOTTY;
    238       1.1     alnsn 	}
    239       1.1     alnsn }
    240       1.1     alnsn #endif
    241       1.1     alnsn 
    242      1.20     alnsn /*
    243      1.21     alnsn  * Return a number of scratch registers to pass
    244      1.20     alnsn  * to sljit_emit_enter() function.
    245      1.20     alnsn  */
    246      1.20     alnsn static sljit_si
    247      1.20     alnsn nscratches(bpfjit_hint_t hints)
    248      1.20     alnsn {
    249      1.20     alnsn 	sljit_si rv = 2;
    250      1.20     alnsn 
    251      1.22     alnsn #ifdef _KERNEL
    252      1.24     alnsn 	if (hints & BJ_HINT_PKT)
    253      1.24     alnsn 		rv = 3; /* xcall with three arguments */
    254      1.22     alnsn #endif
    255      1.22     alnsn 
    256      1.27     alnsn 	if (hints & BJ_HINT_IND)
    257      1.20     alnsn 		rv = 3; /* uses BJ_TMP2REG */
    258      1.20     alnsn 
    259      1.20     alnsn 	if (hints & BJ_HINT_COP)
    260      1.20     alnsn 		rv = 3; /* calls copfunc with three arguments */
    261      1.20     alnsn 
    262      1.32     alnsn 	if (hints & BJ_HINT_XREG)
    263      1.32     alnsn 		rv = 4; /* uses BJ_XREG */
    264      1.32     alnsn 
    265      1.32     alnsn #ifdef _KERNEL
    266      1.32     alnsn 	if (hints & BJ_HINT_LDX)
    267      1.32     alnsn 		rv = 5; /* uses BJ_TMP3REG */
    268      1.32     alnsn #endif
    269      1.32     alnsn 
    270      1.29     alnsn 	if (hints & BJ_HINT_COPX)
    271      1.32     alnsn 		rv = 5; /* uses BJ_TMP3REG */
    272      1.29     alnsn 
    273      1.29     alnsn 	return rv;
    274      1.29     alnsn }
    275      1.29     alnsn 
    276      1.29     alnsn /*
    277      1.29     alnsn  * Return a number of saved registers to pass
    278      1.29     alnsn  * to sljit_emit_enter() function.
    279      1.29     alnsn  */
    280      1.29     alnsn static sljit_si
    281      1.29     alnsn nsaveds(bpfjit_hint_t hints)
    282      1.29     alnsn {
    283      1.29     alnsn 	sljit_si rv = 3;
    284      1.29     alnsn 
    285      1.20     alnsn 	return rv;
    286      1.20     alnsn }
    287      1.20     alnsn 
    288       1.1     alnsn static uint32_t
    289       1.7     alnsn read_width(const struct bpf_insn *pc)
    290       1.1     alnsn {
    291       1.1     alnsn 
    292       1.1     alnsn 	switch (BPF_SIZE(pc->code)) {
    293  1.36.2.1     skrll 	case BPF_W: return 4;
    294  1.36.2.1     skrll 	case BPF_H: return 2;
    295  1.36.2.1     skrll 	case BPF_B: return 1;
    296  1.36.2.1     skrll 	default:    return 0;
    297       1.1     alnsn 	}
    298       1.1     alnsn }
    299       1.1     alnsn 
    300      1.13     alnsn /*
    301      1.13     alnsn  * Copy buf and buflen members of bpf_args from BJ_ARGS
    302      1.13     alnsn  * pointer to BJ_BUF and BJ_BUFLEN registers.
    303      1.13     alnsn  */
    304      1.13     alnsn static int
    305      1.13     alnsn load_buf_buflen(struct sljit_compiler *compiler)
    306      1.13     alnsn {
    307      1.13     alnsn 	int status;
    308      1.13     alnsn 
    309      1.13     alnsn 	status = sljit_emit_op1(compiler,
    310      1.13     alnsn 	    SLJIT_MOV_P,
    311      1.13     alnsn 	    BJ_BUF, 0,
    312      1.13     alnsn 	    SLJIT_MEM1(BJ_ARGS),
    313      1.13     alnsn 	    offsetof(struct bpf_args, pkt));
    314      1.13     alnsn 	if (status != SLJIT_SUCCESS)
    315      1.13     alnsn 		return status;
    316      1.13     alnsn 
    317      1.13     alnsn 	status = sljit_emit_op1(compiler,
    318      1.21     alnsn 	    SLJIT_MOV, /* size_t source */
    319      1.13     alnsn 	    BJ_BUFLEN, 0,
    320      1.13     alnsn 	    SLJIT_MEM1(BJ_ARGS),
    321      1.13     alnsn 	    offsetof(struct bpf_args, buflen));
    322      1.13     alnsn 
    323      1.13     alnsn 	return status;
    324      1.13     alnsn }
    325      1.13     alnsn 
    326       1.7     alnsn static bool
    327       1.7     alnsn grow_jumps(struct sljit_jump ***jumps, size_t *size)
    328       1.7     alnsn {
    329       1.7     alnsn 	struct sljit_jump **newptr;
    330       1.7     alnsn 	const size_t elemsz = sizeof(struct sljit_jump *);
    331       1.7     alnsn 	size_t old_size = *size;
    332       1.7     alnsn 	size_t new_size = 2 * old_size;
    333       1.7     alnsn 
    334       1.7     alnsn 	if (new_size < old_size || new_size > SIZE_MAX / elemsz)
    335       1.7     alnsn 		return false;
    336       1.7     alnsn 
    337       1.7     alnsn 	newptr = BJ_ALLOC(new_size * elemsz);
    338       1.7     alnsn 	if (newptr == NULL)
    339       1.7     alnsn 		return false;
    340       1.7     alnsn 
    341       1.7     alnsn 	memcpy(newptr, *jumps, old_size * elemsz);
    342       1.7     alnsn 	BJ_FREE(*jumps, old_size * elemsz);
    343       1.7     alnsn 
    344       1.7     alnsn 	*jumps = newptr;
    345       1.7     alnsn 	*size = new_size;
    346       1.7     alnsn 	return true;
    347       1.7     alnsn }
    348       1.7     alnsn 
    349       1.7     alnsn static bool
    350       1.7     alnsn append_jump(struct sljit_jump *jump, struct sljit_jump ***jumps,
    351       1.7     alnsn     size_t *size, size_t *max_size)
    352       1.1     alnsn {
    353       1.7     alnsn 	if (*size == *max_size && !grow_jumps(jumps, max_size))
    354       1.7     alnsn 		return false;
    355       1.1     alnsn 
    356       1.7     alnsn 	(*jumps)[(*size)++] = jump;
    357       1.7     alnsn 	return true;
    358       1.1     alnsn }
    359       1.1     alnsn 
    360       1.1     alnsn /*
    361      1.24     alnsn  * Emit code for BPF_LD+BPF_B+BPF_ABS    A <- P[k:1].
    362       1.1     alnsn  */
    363       1.1     alnsn static int
    364      1.27     alnsn emit_read8(struct sljit_compiler *compiler, sljit_si src, uint32_t k)
    365       1.1     alnsn {
    366       1.1     alnsn 
    367       1.1     alnsn 	return sljit_emit_op1(compiler,
    368       1.1     alnsn 	    SLJIT_MOV_UB,
    369       1.7     alnsn 	    BJ_AREG, 0,
    370      1.27     alnsn 	    SLJIT_MEM1(src), k);
    371       1.1     alnsn }
    372       1.1     alnsn 
    373       1.1     alnsn /*
    374      1.24     alnsn  * Emit code for BPF_LD+BPF_H+BPF_ABS    A <- P[k:2].
    375       1.1     alnsn  */
    376       1.1     alnsn static int
    377      1.27     alnsn emit_read16(struct sljit_compiler *compiler, sljit_si src, uint32_t k)
    378       1.1     alnsn {
    379       1.1     alnsn 	int status;
    380       1.1     alnsn 
    381      1.27     alnsn 	BJ_ASSERT(k <= UINT32_MAX - 1);
    382      1.27     alnsn 
    383      1.27     alnsn 	/* A = buf[k]; */
    384       1.1     alnsn 	status = sljit_emit_op1(compiler,
    385       1.1     alnsn 	    SLJIT_MOV_UB,
    386      1.27     alnsn 	    BJ_AREG, 0,
    387      1.27     alnsn 	    SLJIT_MEM1(src), k);
    388       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    389       1.1     alnsn 		return status;
    390       1.1     alnsn 
    391      1.27     alnsn 	/* tmp1 = buf[k+1]; */
    392       1.1     alnsn 	status = sljit_emit_op1(compiler,
    393       1.1     alnsn 	    SLJIT_MOV_UB,
    394      1.27     alnsn 	    BJ_TMP1REG, 0,
    395      1.27     alnsn 	    SLJIT_MEM1(src), k+1);
    396       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    397       1.1     alnsn 		return status;
    398       1.1     alnsn 
    399      1.27     alnsn 	/* A = A << 8; */
    400       1.1     alnsn 	status = sljit_emit_op2(compiler,
    401       1.1     alnsn 	    SLJIT_SHL,
    402      1.27     alnsn 	    BJ_AREG, 0,
    403      1.27     alnsn 	    BJ_AREG, 0,
    404       1.1     alnsn 	    SLJIT_IMM, 8);
    405       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    406       1.1     alnsn 		return status;
    407       1.1     alnsn 
    408       1.1     alnsn 	/* A = A + tmp1; */
    409       1.1     alnsn 	status = sljit_emit_op2(compiler,
    410       1.1     alnsn 	    SLJIT_ADD,
    411       1.7     alnsn 	    BJ_AREG, 0,
    412       1.7     alnsn 	    BJ_AREG, 0,
    413       1.7     alnsn 	    BJ_TMP1REG, 0);
    414       1.1     alnsn 	return status;
    415       1.1     alnsn }
    416       1.1     alnsn 
    417       1.1     alnsn /*
    418      1.24     alnsn  * Emit code for BPF_LD+BPF_W+BPF_ABS    A <- P[k:4].
    419       1.1     alnsn  */
    420       1.1     alnsn static int
    421      1.27     alnsn emit_read32(struct sljit_compiler *compiler, sljit_si src, uint32_t k)
    422       1.1     alnsn {
    423       1.1     alnsn 	int status;
    424       1.1     alnsn 
    425      1.27     alnsn 	BJ_ASSERT(k <= UINT32_MAX - 3);
    426       1.1     alnsn 
    427      1.27     alnsn 	/* A = buf[k]; */
    428       1.1     alnsn 	status = sljit_emit_op1(compiler,
    429       1.1     alnsn 	    SLJIT_MOV_UB,
    430      1.27     alnsn 	    BJ_AREG, 0,
    431      1.27     alnsn 	    SLJIT_MEM1(src), k);
    432       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    433       1.1     alnsn 		return status;
    434       1.1     alnsn 
    435      1.27     alnsn 	/* tmp1 = buf[k+1]; */
    436       1.1     alnsn 	status = sljit_emit_op1(compiler,
    437       1.1     alnsn 	    SLJIT_MOV_UB,
    438      1.27     alnsn 	    BJ_TMP1REG, 0,
    439      1.27     alnsn 	    SLJIT_MEM1(src), k+1);
    440       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    441       1.1     alnsn 		return status;
    442       1.1     alnsn 
    443      1.27     alnsn 	/* A = A << 8; */
    444       1.1     alnsn 	status = sljit_emit_op2(compiler,
    445       1.1     alnsn 	    SLJIT_SHL,
    446      1.27     alnsn 	    BJ_AREG, 0,
    447      1.27     alnsn 	    BJ_AREG, 0,
    448      1.27     alnsn 	    SLJIT_IMM, 8);
    449       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    450       1.1     alnsn 		return status;
    451       1.1     alnsn 
    452       1.1     alnsn 	/* A = A + tmp1; */
    453       1.1     alnsn 	status = sljit_emit_op2(compiler,
    454       1.1     alnsn 	    SLJIT_ADD,
    455       1.7     alnsn 	    BJ_AREG, 0,
    456       1.7     alnsn 	    BJ_AREG, 0,
    457       1.7     alnsn 	    BJ_TMP1REG, 0);
    458       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    459       1.1     alnsn 		return status;
    460       1.1     alnsn 
    461       1.1     alnsn 	/* tmp1 = buf[k+2]; */
    462       1.1     alnsn 	status = sljit_emit_op1(compiler,
    463       1.1     alnsn 	    SLJIT_MOV_UB,
    464       1.7     alnsn 	    BJ_TMP1REG, 0,
    465      1.27     alnsn 	    SLJIT_MEM1(src), k+2);
    466       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    467       1.1     alnsn 		return status;
    468       1.1     alnsn 
    469      1.27     alnsn 	/* A = A << 8; */
    470       1.1     alnsn 	status = sljit_emit_op2(compiler,
    471       1.1     alnsn 	    SLJIT_SHL,
    472      1.27     alnsn 	    BJ_AREG, 0,
    473      1.27     alnsn 	    BJ_AREG, 0,
    474      1.27     alnsn 	    SLJIT_IMM, 8);
    475       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    476       1.1     alnsn 		return status;
    477       1.1     alnsn 
    478      1.27     alnsn 	/* A = A + tmp1; */
    479       1.1     alnsn 	status = sljit_emit_op2(compiler,
    480       1.1     alnsn 	    SLJIT_ADD,
    481       1.7     alnsn 	    BJ_AREG, 0,
    482       1.7     alnsn 	    BJ_AREG, 0,
    483      1.27     alnsn 	    BJ_TMP1REG, 0);
    484      1.27     alnsn 	if (status != SLJIT_SUCCESS)
    485      1.27     alnsn 		return status;
    486      1.27     alnsn 
    487      1.27     alnsn 	/* tmp1 = buf[k+3]; */
    488      1.27     alnsn 	status = sljit_emit_op1(compiler,
    489      1.27     alnsn 	    SLJIT_MOV_UB,
    490      1.27     alnsn 	    BJ_TMP1REG, 0,
    491      1.27     alnsn 	    SLJIT_MEM1(src), k+3);
    492       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    493       1.1     alnsn 		return status;
    494       1.1     alnsn 
    495      1.27     alnsn 	/* A = A << 8; */
    496       1.1     alnsn 	status = sljit_emit_op2(compiler,
    497       1.1     alnsn 	    SLJIT_SHL,
    498      1.27     alnsn 	    BJ_AREG, 0,
    499      1.27     alnsn 	    BJ_AREG, 0,
    500       1.1     alnsn 	    SLJIT_IMM, 8);
    501       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    502       1.1     alnsn 		return status;
    503       1.1     alnsn 
    504       1.1     alnsn 	/* A = A + tmp1; */
    505       1.1     alnsn 	status = sljit_emit_op2(compiler,
    506       1.1     alnsn 	    SLJIT_ADD,
    507       1.7     alnsn 	    BJ_AREG, 0,
    508       1.7     alnsn 	    BJ_AREG, 0,
    509       1.7     alnsn 	    BJ_TMP1REG, 0);
    510       1.1     alnsn 	return status;
    511       1.1     alnsn }
    512       1.1     alnsn 
    513       1.1     alnsn #ifdef _KERNEL
    514       1.1     alnsn /*
    515      1.24     alnsn  * Emit code for m_xword/m_xhalf/m_xbyte call.
    516       1.1     alnsn  *
    517      1.24     alnsn  * @pc BPF_LD+BPF_W+BPF_ABS    A <- P[k:4]
    518      1.24     alnsn  *     BPF_LD+BPF_H+BPF_ABS    A <- P[k:2]
    519      1.24     alnsn  *     BPF_LD+BPF_B+BPF_ABS    A <- P[k:1]
    520      1.24     alnsn  *     BPF_LD+BPF_W+BPF_IND    A <- P[X+k:4]
    521      1.24     alnsn  *     BPF_LD+BPF_H+BPF_IND    A <- P[X+k:2]
    522      1.24     alnsn  *     BPF_LD+BPF_B+BPF_IND    A <- P[X+k:1]
    523      1.24     alnsn  *     BPF_LDX+BPF_B+BPF_MSH   X <- 4*(P[k:1]&0xf)
    524       1.1     alnsn  */
    525       1.1     alnsn static int
    526      1.32     alnsn emit_xcall(struct sljit_compiler *compiler, bpfjit_hint_t hints,
    527      1.32     alnsn     const struct bpf_insn *pc, int dst, struct sljit_jump ***ret0,
    528      1.32     alnsn     size_t *ret0_size, size_t *ret0_maxsize,
    529       1.1     alnsn     uint32_t (*fn)(const struct mbuf *, uint32_t, int *))
    530       1.1     alnsn {
    531      1.32     alnsn #if BJ_XREG == SLJIT_RETURN_REG   || \
    532      1.32     alnsn     BJ_XREG == SLJIT_SCRATCH_REG1 || \
    533      1.32     alnsn     BJ_XREG == SLJIT_SCRATCH_REG2 || \
    534      1.32     alnsn     BJ_XREG == SLJIT_SCRATCH_REG3
    535      1.32     alnsn #error "Not supported assignment of registers."
    536      1.32     alnsn #endif
    537      1.23     alnsn 	struct sljit_jump *jump;
    538      1.32     alnsn 	sljit_si save_reg;
    539       1.1     alnsn 	int status;
    540       1.1     alnsn 
    541      1.32     alnsn 	save_reg = (BPF_CLASS(pc->code) == BPF_LDX) ? BJ_AREG : BJ_XREG;
    542      1.23     alnsn 
    543      1.32     alnsn 	if (save_reg == BJ_AREG || (hints & BJ_HINT_XREG)) {
    544      1.32     alnsn 		/* save A or X */
    545       1.1     alnsn 		status = sljit_emit_op1(compiler,
    546      1.32     alnsn 		    SLJIT_MOV_UI, /* uint32_t destination */
    547      1.32     alnsn 		    SLJIT_MEM1(SLJIT_LOCALS_REG),
    548      1.32     alnsn 		    offsetof(struct bpfjit_stack, reg),
    549      1.32     alnsn 		    save_reg, 0);
    550       1.1     alnsn 		if (status != SLJIT_SUCCESS)
    551       1.1     alnsn 			return status;
    552       1.1     alnsn 	}
    553       1.1     alnsn 
    554       1.1     alnsn 	/*
    555      1.23     alnsn 	 * Prepare registers for fn(mbuf, k, &err) call.
    556       1.1     alnsn 	 */
    557       1.1     alnsn 	status = sljit_emit_op1(compiler,
    558       1.1     alnsn 	    SLJIT_MOV,
    559      1.12     alnsn 	    SLJIT_SCRATCH_REG1, 0,
    560       1.7     alnsn 	    BJ_BUF, 0);
    561       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    562       1.1     alnsn 		return status;
    563       1.1     alnsn 
    564       1.1     alnsn 	if (BPF_CLASS(pc->code) == BPF_LD && BPF_MODE(pc->code) == BPF_IND) {
    565      1.31     alnsn 		if (pc->k == 0) {
    566      1.31     alnsn 			/* k = X; */
    567      1.31     alnsn 			status = sljit_emit_op1(compiler,
    568      1.31     alnsn 			    SLJIT_MOV,
    569      1.31     alnsn 			    SLJIT_SCRATCH_REG2, 0,
    570      1.31     alnsn 			    BJ_XREG, 0);
    571      1.31     alnsn 			if (status != SLJIT_SUCCESS)
    572      1.31     alnsn 				return status;
    573      1.31     alnsn 		} else {
    574      1.31     alnsn 			/* if (X > UINT32_MAX - pc->k) return 0; */
    575      1.31     alnsn 			jump = sljit_emit_cmp(compiler,
    576      1.31     alnsn 			    SLJIT_C_GREATER,
    577      1.31     alnsn 			    BJ_XREG, 0,
    578      1.31     alnsn 			    SLJIT_IMM, UINT32_MAX - pc->k);
    579      1.31     alnsn 			if (jump == NULL)
    580      1.31     alnsn 				return SLJIT_ERR_ALLOC_FAILED;
    581      1.31     alnsn 			if (!append_jump(jump, ret0, ret0_size, ret0_maxsize))
    582      1.31     alnsn 				return SLJIT_ERR_ALLOC_FAILED;
    583      1.31     alnsn 
    584      1.31     alnsn 			/* k = X + pc->k; */
    585      1.31     alnsn 			status = sljit_emit_op2(compiler,
    586      1.31     alnsn 			    SLJIT_ADD,
    587      1.31     alnsn 			    SLJIT_SCRATCH_REG2, 0,
    588      1.31     alnsn 			    BJ_XREG, 0,
    589      1.31     alnsn 			    SLJIT_IMM, (uint32_t)pc->k);
    590      1.31     alnsn 			if (status != SLJIT_SUCCESS)
    591      1.31     alnsn 				return status;
    592      1.31     alnsn 		}
    593       1.1     alnsn 	} else {
    594      1.23     alnsn 		/* k = pc->k */
    595       1.1     alnsn 		status = sljit_emit_op1(compiler,
    596       1.1     alnsn 		    SLJIT_MOV,
    597      1.12     alnsn 		    SLJIT_SCRATCH_REG2, 0,
    598       1.1     alnsn 		    SLJIT_IMM, (uint32_t)pc->k);
    599      1.24     alnsn 		if (status != SLJIT_SUCCESS)
    600      1.24     alnsn 			return status;
    601       1.1     alnsn 	}
    602       1.1     alnsn 
    603      1.21     alnsn 	/*
    604      1.21     alnsn 	 * The third argument of fn is an address on stack.
    605      1.21     alnsn 	 */
    606       1.1     alnsn 	status = sljit_get_local_base(compiler,
    607      1.21     alnsn 	    SLJIT_SCRATCH_REG3, 0,
    608      1.21     alnsn 	    offsetof(struct bpfjit_stack, err));
    609       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    610       1.1     alnsn 		return status;
    611       1.1     alnsn 
    612       1.1     alnsn 	/* fn(buf, k, &err); */
    613       1.1     alnsn 	status = sljit_emit_ijump(compiler,
    614       1.1     alnsn 	    SLJIT_CALL3,
    615       1.1     alnsn 	    SLJIT_IMM, SLJIT_FUNC_OFFSET(fn));
    616      1.24     alnsn 	if (status != SLJIT_SUCCESS)
    617      1.24     alnsn 		return status;
    618       1.1     alnsn 
    619       1.7     alnsn 	if (dst != SLJIT_RETURN_REG) {
    620       1.1     alnsn 		/* move return value to dst */
    621       1.1     alnsn 		status = sljit_emit_op1(compiler,
    622       1.1     alnsn 		    SLJIT_MOV,
    623      1.23     alnsn 		    dst, 0,
    624       1.1     alnsn 		    SLJIT_RETURN_REG, 0);
    625       1.1     alnsn 		if (status != SLJIT_SUCCESS)
    626       1.1     alnsn 			return status;
    627       1.7     alnsn 	}
    628       1.1     alnsn 
    629      1.30     alnsn 	/* if (*err != 0) return 0; */
    630      1.30     alnsn 	jump = sljit_emit_cmp(compiler,
    631      1.30     alnsn 	    SLJIT_C_NOT_EQUAL|SLJIT_INT_OP,
    632      1.21     alnsn 	    SLJIT_MEM1(SLJIT_LOCALS_REG),
    633      1.30     alnsn 	    offsetof(struct bpfjit_stack, err),
    634       1.1     alnsn 	    SLJIT_IMM, 0);
    635      1.23     alnsn 	if (jump == NULL)
    636      1.23     alnsn 		return SLJIT_ERR_ALLOC_FAILED;
    637      1.23     alnsn 
    638      1.23     alnsn 	if (!append_jump(jump, ret0, ret0_size, ret0_maxsize))
    639       1.1     alnsn 		return SLJIT_ERR_ALLOC_FAILED;
    640       1.1     alnsn 
    641      1.32     alnsn 	if (save_reg == BJ_AREG || (hints & BJ_HINT_XREG)) {
    642      1.32     alnsn 		/* restore A or X */
    643      1.26     alnsn 		status = sljit_emit_op1(compiler,
    644      1.32     alnsn 		    SLJIT_MOV_UI, /* uint32_t source */
    645      1.32     alnsn 		    save_reg, 0,
    646      1.32     alnsn 		    SLJIT_MEM1(SLJIT_LOCALS_REG),
    647      1.32     alnsn 		    offsetof(struct bpfjit_stack, reg));
    648      1.26     alnsn 		if (status != SLJIT_SUCCESS)
    649      1.26     alnsn 			return status;
    650      1.26     alnsn 	}
    651      1.26     alnsn 
    652      1.24     alnsn 	return SLJIT_SUCCESS;
    653       1.1     alnsn }
    654       1.1     alnsn #endif
    655       1.1     alnsn 
    656       1.1     alnsn /*
    657      1.13     alnsn  * Emit code for BPF_COP and BPF_COPX instructions.
    658      1.13     alnsn  */
    659      1.13     alnsn static int
    660      1.32     alnsn emit_cop(struct sljit_compiler *compiler, bpfjit_hint_t hints,
    661      1.28     alnsn     const bpf_ctx_t *bc, const struct bpf_insn *pc,
    662      1.28     alnsn     struct sljit_jump ***ret0, size_t *ret0_size, size_t *ret0_maxsize)
    663      1.13     alnsn {
    664      1.32     alnsn #if BJ_XREG    == SLJIT_RETURN_REG   || \
    665      1.32     alnsn     BJ_XREG    == SLJIT_SCRATCH_REG1 || \
    666      1.32     alnsn     BJ_XREG    == SLJIT_SCRATCH_REG2 || \
    667      1.32     alnsn     BJ_XREG    == SLJIT_SCRATCH_REG3 || \
    668      1.32     alnsn     BJ_TMP3REG == SLJIT_SCRATCH_REG1 || \
    669      1.28     alnsn     BJ_TMP3REG == SLJIT_SCRATCH_REG2 || \
    670      1.28     alnsn     BJ_TMP3REG == SLJIT_SCRATCH_REG3
    671      1.13     alnsn #error "Not supported assignment of registers."
    672      1.13     alnsn #endif
    673      1.13     alnsn 
    674      1.13     alnsn 	struct sljit_jump *jump;
    675      1.28     alnsn 	sljit_si call_reg;
    676      1.28     alnsn 	sljit_sw call_off;
    677      1.13     alnsn 	int status;
    678      1.13     alnsn 
    679      1.13     alnsn 	BJ_ASSERT(bc != NULL && bc->copfuncs != NULL);
    680      1.13     alnsn 
    681      1.32     alnsn 	if (hints & BJ_HINT_LDX) {
    682      1.32     alnsn 		/* save X */
    683      1.32     alnsn 		status = sljit_emit_op1(compiler,
    684      1.32     alnsn 		    SLJIT_MOV_UI, /* uint32_t destination */
    685      1.32     alnsn 		    SLJIT_MEM1(SLJIT_LOCALS_REG),
    686      1.32     alnsn 		    offsetof(struct bpfjit_stack, reg),
    687      1.32     alnsn 		    BJ_XREG, 0);
    688      1.32     alnsn 		if (status != SLJIT_SUCCESS)
    689      1.32     alnsn 			return status;
    690      1.32     alnsn 	}
    691      1.32     alnsn 
    692      1.28     alnsn 	if (BPF_MISCOP(pc->code) == BPF_COP) {
    693      1.28     alnsn 		call_reg = SLJIT_IMM;
    694      1.28     alnsn 		call_off = SLJIT_FUNC_OFFSET(bc->copfuncs[pc->k]);
    695      1.28     alnsn 	} else {
    696      1.13     alnsn 		/* if (X >= bc->nfuncs) return 0; */
    697      1.13     alnsn 		jump = sljit_emit_cmp(compiler,
    698      1.13     alnsn 		    SLJIT_C_GREATER_EQUAL,
    699      1.13     alnsn 		    BJ_XREG, 0,
    700      1.13     alnsn 		    SLJIT_IMM, bc->nfuncs);
    701      1.13     alnsn 		if (jump == NULL)
    702      1.13     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
    703      1.28     alnsn 		if (!append_jump(jump, ret0, ret0_size, ret0_maxsize))
    704      1.28     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
    705      1.28     alnsn 
    706      1.28     alnsn 		/* tmp1 = ctx; */
    707      1.28     alnsn 		status = sljit_emit_op1(compiler,
    708      1.28     alnsn 		    SLJIT_MOV_P,
    709      1.28     alnsn 		    BJ_TMP1REG, 0,
    710      1.28     alnsn 		    SLJIT_MEM1(SLJIT_LOCALS_REG),
    711      1.28     alnsn 		    offsetof(struct bpfjit_stack, ctx));
    712      1.28     alnsn 		if (status != SLJIT_SUCCESS)
    713      1.28     alnsn 			return status;
    714      1.28     alnsn 
    715      1.28     alnsn 		/* tmp1 = ctx->copfuncs; */
    716      1.28     alnsn 		status = sljit_emit_op1(compiler,
    717      1.28     alnsn 		    SLJIT_MOV_P,
    718      1.28     alnsn 		    BJ_TMP1REG, 0,
    719      1.28     alnsn 		    SLJIT_MEM1(BJ_TMP1REG),
    720      1.28     alnsn 		    offsetof(struct bpf_ctx, copfuncs));
    721      1.28     alnsn 		if (status != SLJIT_SUCCESS)
    722      1.28     alnsn 			return status;
    723      1.28     alnsn 
    724      1.28     alnsn 		/* tmp2 = X; */
    725      1.28     alnsn 		status = sljit_emit_op1(compiler,
    726      1.28     alnsn 		    SLJIT_MOV,
    727      1.28     alnsn 		    BJ_TMP2REG, 0,
    728      1.28     alnsn 		    BJ_XREG, 0);
    729      1.28     alnsn 		if (status != SLJIT_SUCCESS)
    730      1.28     alnsn 			return status;
    731      1.28     alnsn 
    732      1.28     alnsn 		/* tmp3 = ctx->copfuncs[tmp2]; */
    733      1.28     alnsn 		call_reg = BJ_TMP3REG;
    734      1.28     alnsn 		call_off = 0;
    735      1.28     alnsn 		status = sljit_emit_op1(compiler,
    736      1.28     alnsn 		    SLJIT_MOV_P,
    737      1.28     alnsn 		    call_reg, call_off,
    738      1.28     alnsn 		    SLJIT_MEM2(BJ_TMP1REG, BJ_TMP2REG),
    739      1.28     alnsn 		    SLJIT_WORD_SHIFT);
    740      1.28     alnsn 		if (status != SLJIT_SUCCESS)
    741      1.28     alnsn 			return status;
    742      1.13     alnsn 	}
    743      1.13     alnsn 
    744      1.13     alnsn 	/*
    745      1.13     alnsn 	 * Copy bpf_copfunc_t arguments to registers.
    746      1.13     alnsn 	 */
    747      1.13     alnsn #if BJ_AREG != SLJIT_SCRATCH_REG3
    748      1.13     alnsn 	status = sljit_emit_op1(compiler,
    749      1.13     alnsn 	    SLJIT_MOV_UI,
    750      1.13     alnsn 	    SLJIT_SCRATCH_REG3, 0,
    751      1.13     alnsn 	    BJ_AREG, 0);
    752      1.13     alnsn 	if (status != SLJIT_SUCCESS)
    753      1.13     alnsn 		return status;
    754      1.13     alnsn #endif
    755      1.13     alnsn 
    756      1.13     alnsn 	status = sljit_emit_op1(compiler,
    757      1.13     alnsn 	    SLJIT_MOV_P,
    758      1.13     alnsn 	    SLJIT_SCRATCH_REG1, 0,
    759      1.13     alnsn 	    SLJIT_MEM1(SLJIT_LOCALS_REG),
    760      1.13     alnsn 	    offsetof(struct bpfjit_stack, ctx));
    761      1.13     alnsn 	if (status != SLJIT_SUCCESS)
    762      1.13     alnsn 		return status;
    763      1.13     alnsn 
    764      1.13     alnsn 	status = sljit_emit_op1(compiler,
    765      1.13     alnsn 	    SLJIT_MOV_P,
    766      1.13     alnsn 	    SLJIT_SCRATCH_REG2, 0,
    767      1.13     alnsn 	    BJ_ARGS, 0);
    768      1.13     alnsn 	if (status != SLJIT_SUCCESS)
    769      1.13     alnsn 		return status;
    770      1.13     alnsn 
    771      1.28     alnsn 	status = sljit_emit_ijump(compiler,
    772      1.28     alnsn 	    SLJIT_CALL3, call_reg, call_off);
    773      1.28     alnsn 	if (status != SLJIT_SUCCESS)
    774      1.28     alnsn 		return status;
    775      1.13     alnsn 
    776      1.13     alnsn #if BJ_AREG != SLJIT_RETURN_REG
    777      1.13     alnsn 	status = sljit_emit_op1(compiler,
    778      1.13     alnsn 	    SLJIT_MOV,
    779      1.13     alnsn 	    BJ_AREG, 0,
    780      1.13     alnsn 	    SLJIT_RETURN_REG, 0);
    781      1.13     alnsn 	if (status != SLJIT_SUCCESS)
    782      1.13     alnsn 		return status;
    783      1.13     alnsn #endif
    784      1.13     alnsn 
    785      1.32     alnsn 	if (hints & BJ_HINT_LDX) {
    786      1.32     alnsn 		/* restore X */
    787      1.32     alnsn 		status = sljit_emit_op1(compiler,
    788      1.32     alnsn 		    SLJIT_MOV_UI, /* uint32_t source */
    789      1.32     alnsn 		    BJ_XREG, 0,
    790      1.32     alnsn 		    SLJIT_MEM1(SLJIT_LOCALS_REG),
    791      1.32     alnsn 		    offsetof(struct bpfjit_stack, reg));
    792      1.32     alnsn 		if (status != SLJIT_SUCCESS)
    793      1.32     alnsn 			return status;
    794      1.32     alnsn 	}
    795      1.32     alnsn 
    796      1.24     alnsn 	return SLJIT_SUCCESS;
    797      1.13     alnsn }
    798      1.13     alnsn 
    799      1.13     alnsn /*
    800       1.1     alnsn  * Generate code for
    801       1.1     alnsn  * BPF_LD+BPF_W+BPF_ABS    A <- P[k:4]
    802       1.1     alnsn  * BPF_LD+BPF_H+BPF_ABS    A <- P[k:2]
    803       1.1     alnsn  * BPF_LD+BPF_B+BPF_ABS    A <- P[k:1]
    804       1.1     alnsn  * BPF_LD+BPF_W+BPF_IND    A <- P[X+k:4]
    805       1.1     alnsn  * BPF_LD+BPF_H+BPF_IND    A <- P[X+k:2]
    806       1.1     alnsn  * BPF_LD+BPF_B+BPF_IND    A <- P[X+k:1]
    807       1.1     alnsn  */
    808       1.1     alnsn static int
    809      1.32     alnsn emit_pkt_read(struct sljit_compiler *compiler, bpfjit_hint_t hints,
    810       1.7     alnsn     const struct bpf_insn *pc, struct sljit_jump *to_mchain_jump,
    811       1.7     alnsn     struct sljit_jump ***ret0, size_t *ret0_size, size_t *ret0_maxsize)
    812       1.1     alnsn {
    813      1.25     alnsn 	int status = SLJIT_ERR_ALLOC_FAILED;
    814       1.1     alnsn 	uint32_t width;
    815      1.27     alnsn 	sljit_si ld_reg;
    816       1.1     alnsn 	struct sljit_jump *jump;
    817       1.1     alnsn #ifdef _KERNEL
    818       1.1     alnsn 	struct sljit_label *label;
    819       1.1     alnsn 	struct sljit_jump *over_mchain_jump;
    820       1.1     alnsn 	const bool check_zero_buflen = (to_mchain_jump != NULL);
    821       1.1     alnsn #endif
    822       1.1     alnsn 	const uint32_t k = pc->k;
    823       1.1     alnsn 
    824       1.1     alnsn #ifdef _KERNEL
    825       1.1     alnsn 	if (to_mchain_jump == NULL) {
    826       1.1     alnsn 		to_mchain_jump = sljit_emit_cmp(compiler,
    827       1.1     alnsn 		    SLJIT_C_EQUAL,
    828       1.7     alnsn 		    BJ_BUFLEN, 0,
    829       1.1     alnsn 		    SLJIT_IMM, 0);
    830       1.1     alnsn 		if (to_mchain_jump == NULL)
    831       1.7     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
    832       1.1     alnsn 	}
    833       1.1     alnsn #endif
    834       1.1     alnsn 
    835      1.27     alnsn 	ld_reg = BJ_BUF;
    836       1.1     alnsn 	width = read_width(pc);
    837  1.36.2.1     skrll 	if (width == 0)
    838  1.36.2.1     skrll 		return SLJIT_ERR_ALLOC_FAILED;
    839       1.1     alnsn 
    840       1.1     alnsn 	if (BPF_MODE(pc->code) == BPF_IND) {
    841       1.1     alnsn 		/* tmp1 = buflen - (pc->k + width); */
    842       1.1     alnsn 		status = sljit_emit_op2(compiler,
    843       1.1     alnsn 		    SLJIT_SUB,
    844       1.7     alnsn 		    BJ_TMP1REG, 0,
    845       1.7     alnsn 		    BJ_BUFLEN, 0,
    846       1.1     alnsn 		    SLJIT_IMM, k + width);
    847       1.1     alnsn 		if (status != SLJIT_SUCCESS)
    848       1.1     alnsn 			return status;
    849       1.1     alnsn 
    850      1.27     alnsn 		/* ld_reg = buf + X; */
    851      1.27     alnsn 		ld_reg = BJ_TMP2REG;
    852       1.1     alnsn 		status = sljit_emit_op2(compiler,
    853       1.1     alnsn 		    SLJIT_ADD,
    854      1.27     alnsn 		    ld_reg, 0,
    855       1.7     alnsn 		    BJ_BUF, 0,
    856       1.7     alnsn 		    BJ_XREG, 0);
    857       1.1     alnsn 		if (status != SLJIT_SUCCESS)
    858       1.1     alnsn 			return status;
    859       1.1     alnsn 
    860       1.1     alnsn 		/* if (tmp1 < X) return 0; */
    861       1.1     alnsn 		jump = sljit_emit_cmp(compiler,
    862       1.1     alnsn 		    SLJIT_C_LESS,
    863       1.7     alnsn 		    BJ_TMP1REG, 0,
    864       1.7     alnsn 		    BJ_XREG, 0);
    865       1.1     alnsn 		if (jump == NULL)
    866       1.7     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
    867       1.7     alnsn 		if (!append_jump(jump, ret0, ret0_size, ret0_maxsize))
    868       1.7     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
    869       1.1     alnsn 	}
    870       1.1     alnsn 
    871  1.36.2.1     skrll 	/*
    872  1.36.2.1     skrll 	 * Don't emit wrapped-around reads. They're dead code but
    873  1.36.2.1     skrll 	 * dead code elimination logic isn't smart enough to figure
    874  1.36.2.1     skrll 	 * it out.
    875  1.36.2.1     skrll 	 */
    876  1.36.2.1     skrll 	if (k <= UINT32_MAX - width + 1) {
    877  1.36.2.1     skrll 		switch (width) {
    878  1.36.2.1     skrll 		case 4:
    879  1.36.2.1     skrll 			status = emit_read32(compiler, ld_reg, k);
    880  1.36.2.1     skrll 			break;
    881  1.36.2.1     skrll 		case 2:
    882  1.36.2.1     skrll 			status = emit_read16(compiler, ld_reg, k);
    883  1.36.2.1     skrll 			break;
    884  1.36.2.1     skrll 		case 1:
    885  1.36.2.1     skrll 			status = emit_read8(compiler, ld_reg, k);
    886  1.36.2.1     skrll 			break;
    887  1.36.2.1     skrll 		}
    888       1.1     alnsn 
    889  1.36.2.1     skrll 		if (status != SLJIT_SUCCESS)
    890  1.36.2.1     skrll 			return status;
    891  1.36.2.1     skrll 	}
    892       1.1     alnsn 
    893       1.1     alnsn #ifdef _KERNEL
    894       1.1     alnsn 	over_mchain_jump = sljit_emit_jump(compiler, SLJIT_JUMP);
    895       1.1     alnsn 	if (over_mchain_jump == NULL)
    896       1.7     alnsn 		return SLJIT_ERR_ALLOC_FAILED;
    897       1.1     alnsn 
    898       1.1     alnsn 	/* entry point to mchain handler */
    899       1.1     alnsn 	label = sljit_emit_label(compiler);
    900       1.1     alnsn 	if (label == NULL)
    901       1.7     alnsn 		return SLJIT_ERR_ALLOC_FAILED;
    902       1.1     alnsn 	sljit_set_label(to_mchain_jump, label);
    903       1.1     alnsn 
    904       1.1     alnsn 	if (check_zero_buflen) {
    905       1.1     alnsn 		/* if (buflen != 0) return 0; */
    906       1.1     alnsn 		jump = sljit_emit_cmp(compiler,
    907       1.1     alnsn 		    SLJIT_C_NOT_EQUAL,
    908       1.7     alnsn 		    BJ_BUFLEN, 0,
    909       1.1     alnsn 		    SLJIT_IMM, 0);
    910       1.1     alnsn 		if (jump == NULL)
    911       1.1     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
    912       1.7     alnsn 		if (!append_jump(jump, ret0, ret0_size, ret0_maxsize))
    913       1.7     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
    914       1.1     alnsn 	}
    915       1.1     alnsn 
    916       1.1     alnsn 	switch (width) {
    917       1.1     alnsn 	case 4:
    918      1.32     alnsn 		status = emit_xcall(compiler, hints, pc, BJ_AREG,
    919      1.23     alnsn 		    ret0, ret0_size, ret0_maxsize, &m_xword);
    920       1.1     alnsn 		break;
    921       1.1     alnsn 	case 2:
    922      1.32     alnsn 		status = emit_xcall(compiler, hints, pc, BJ_AREG,
    923      1.23     alnsn 		    ret0, ret0_size, ret0_maxsize, &m_xhalf);
    924       1.1     alnsn 		break;
    925       1.1     alnsn 	case 1:
    926      1.32     alnsn 		status = emit_xcall(compiler, hints, pc, BJ_AREG,
    927      1.23     alnsn 		    ret0, ret0_size, ret0_maxsize, &m_xbyte);
    928       1.1     alnsn 		break;
    929       1.1     alnsn 	}
    930       1.1     alnsn 
    931       1.1     alnsn 	if (status != SLJIT_SUCCESS)
    932       1.1     alnsn 		return status;
    933       1.1     alnsn 
    934       1.1     alnsn 	label = sljit_emit_label(compiler);
    935       1.1     alnsn 	if (label == NULL)
    936       1.1     alnsn 		return SLJIT_ERR_ALLOC_FAILED;
    937       1.1     alnsn 	sljit_set_label(over_mchain_jump, label);
    938       1.1     alnsn #endif
    939       1.1     alnsn 
    940      1.24     alnsn 	return SLJIT_SUCCESS;
    941       1.1     alnsn }
    942       1.1     alnsn 
    943      1.13     alnsn static int
    944      1.19     alnsn emit_memload(struct sljit_compiler *compiler,
    945      1.13     alnsn     sljit_si dst, uint32_t k, size_t extwords)
    946      1.13     alnsn {
    947      1.13     alnsn 	int status;
    948      1.13     alnsn 	sljit_si src;
    949      1.13     alnsn 	sljit_sw srcw;
    950      1.13     alnsn 
    951      1.13     alnsn 	srcw = k * sizeof(uint32_t);
    952      1.13     alnsn 
    953      1.13     alnsn 	if (extwords == 0) {
    954      1.13     alnsn 		src = SLJIT_MEM1(SLJIT_LOCALS_REG);
    955      1.13     alnsn 		srcw += offsetof(struct bpfjit_stack, mem);
    956      1.13     alnsn 	} else {
    957      1.13     alnsn 		/* copy extmem pointer to the tmp1 register */
    958      1.13     alnsn 		status = sljit_emit_op1(compiler,
    959      1.16     alnsn 		    SLJIT_MOV_P,
    960      1.13     alnsn 		    BJ_TMP1REG, 0,
    961      1.13     alnsn 		    SLJIT_MEM1(SLJIT_LOCALS_REG),
    962      1.13     alnsn 		    offsetof(struct bpfjit_stack, extmem));
    963      1.13     alnsn 		if (status != SLJIT_SUCCESS)
    964      1.13     alnsn 			return status;
    965      1.13     alnsn 		src = SLJIT_MEM1(BJ_TMP1REG);
    966      1.13     alnsn 	}
    967      1.13     alnsn 
    968      1.13     alnsn 	return sljit_emit_op1(compiler, SLJIT_MOV_UI, dst, 0, src, srcw);
    969      1.13     alnsn }
    970      1.13     alnsn 
    971      1.13     alnsn static int
    972      1.19     alnsn emit_memstore(struct sljit_compiler *compiler,
    973      1.13     alnsn     sljit_si src, uint32_t k, size_t extwords)
    974      1.13     alnsn {
    975      1.13     alnsn 	int status;
    976      1.13     alnsn 	sljit_si dst;
    977      1.13     alnsn 	sljit_sw dstw;
    978      1.13     alnsn 
    979      1.13     alnsn 	dstw = k * sizeof(uint32_t);
    980      1.13     alnsn 
    981      1.13     alnsn 	if (extwords == 0) {
    982      1.13     alnsn 		dst = SLJIT_MEM1(SLJIT_LOCALS_REG);
    983      1.13     alnsn 		dstw += offsetof(struct bpfjit_stack, mem);
    984      1.13     alnsn 	} else {
    985      1.13     alnsn 		/* copy extmem pointer to the tmp1 register */
    986      1.13     alnsn 		status = sljit_emit_op1(compiler,
    987      1.16     alnsn 		    SLJIT_MOV_P,
    988      1.13     alnsn 		    BJ_TMP1REG, 0,
    989      1.13     alnsn 		    SLJIT_MEM1(SLJIT_LOCALS_REG),
    990      1.13     alnsn 		    offsetof(struct bpfjit_stack, extmem));
    991      1.13     alnsn 		if (status != SLJIT_SUCCESS)
    992      1.13     alnsn 			return status;
    993      1.13     alnsn 		dst = SLJIT_MEM1(BJ_TMP1REG);
    994      1.13     alnsn 	}
    995      1.13     alnsn 
    996      1.13     alnsn 	return sljit_emit_op1(compiler, SLJIT_MOV_UI, dst, dstw, src, 0);
    997      1.13     alnsn }
    998      1.13     alnsn 
    999       1.1     alnsn /*
   1000      1.24     alnsn  * Emit code for BPF_LDX+BPF_B+BPF_MSH    X <- 4*(P[k:1]&0xf).
   1001       1.1     alnsn  */
   1002       1.1     alnsn static int
   1003      1.32     alnsn emit_msh(struct sljit_compiler *compiler, bpfjit_hint_t hints,
   1004       1.7     alnsn     const struct bpf_insn *pc, struct sljit_jump *to_mchain_jump,
   1005       1.7     alnsn     struct sljit_jump ***ret0, size_t *ret0_size, size_t *ret0_maxsize)
   1006       1.1     alnsn {
   1007       1.1     alnsn 	int status;
   1008       1.1     alnsn #ifdef _KERNEL
   1009       1.1     alnsn 	struct sljit_label *label;
   1010       1.1     alnsn 	struct sljit_jump *jump, *over_mchain_jump;
   1011       1.1     alnsn 	const bool check_zero_buflen = (to_mchain_jump != NULL);
   1012       1.1     alnsn #endif
   1013       1.1     alnsn 	const uint32_t k = pc->k;
   1014       1.1     alnsn 
   1015       1.1     alnsn #ifdef _KERNEL
   1016       1.1     alnsn 	if (to_mchain_jump == NULL) {
   1017       1.1     alnsn 		to_mchain_jump = sljit_emit_cmp(compiler,
   1018       1.1     alnsn 		    SLJIT_C_EQUAL,
   1019       1.7     alnsn 		    BJ_BUFLEN, 0,
   1020       1.1     alnsn 		    SLJIT_IMM, 0);
   1021       1.1     alnsn 		if (to_mchain_jump == NULL)
   1022       1.7     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
   1023       1.1     alnsn 	}
   1024       1.1     alnsn #endif
   1025       1.1     alnsn 
   1026       1.1     alnsn 	/* tmp1 = buf[k] */
   1027       1.1     alnsn 	status = sljit_emit_op1(compiler,
   1028       1.1     alnsn 	    SLJIT_MOV_UB,
   1029       1.7     alnsn 	    BJ_TMP1REG, 0,
   1030       1.7     alnsn 	    SLJIT_MEM1(BJ_BUF), k);
   1031       1.1     alnsn 	if (status != SLJIT_SUCCESS)
   1032       1.1     alnsn 		return status;
   1033       1.1     alnsn 
   1034       1.1     alnsn #ifdef _KERNEL
   1035       1.1     alnsn 	over_mchain_jump = sljit_emit_jump(compiler, SLJIT_JUMP);
   1036       1.1     alnsn 	if (over_mchain_jump == NULL)
   1037       1.1     alnsn 		return SLJIT_ERR_ALLOC_FAILED;
   1038       1.1     alnsn 
   1039       1.1     alnsn 	/* entry point to mchain handler */
   1040       1.1     alnsn 	label = sljit_emit_label(compiler);
   1041       1.1     alnsn 	if (label == NULL)
   1042       1.1     alnsn 		return SLJIT_ERR_ALLOC_FAILED;
   1043       1.1     alnsn 	sljit_set_label(to_mchain_jump, label);
   1044       1.1     alnsn 
   1045       1.1     alnsn 	if (check_zero_buflen) {
   1046       1.1     alnsn 		/* if (buflen != 0) return 0; */
   1047       1.1     alnsn 		jump = sljit_emit_cmp(compiler,
   1048       1.1     alnsn 		    SLJIT_C_NOT_EQUAL,
   1049       1.7     alnsn 		    BJ_BUFLEN, 0,
   1050       1.1     alnsn 		    SLJIT_IMM, 0);
   1051       1.1     alnsn 		if (jump == NULL)
   1052       1.7     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
   1053       1.7     alnsn 		if (!append_jump(jump, ret0, ret0_size, ret0_maxsize))
   1054       1.7     alnsn 			return SLJIT_ERR_ALLOC_FAILED;
   1055       1.1     alnsn 	}
   1056       1.1     alnsn 
   1057      1.32     alnsn 	status = emit_xcall(compiler, hints, pc, BJ_TMP1REG,
   1058      1.23     alnsn 	    ret0, ret0_size, ret0_maxsize, &m_xbyte);
   1059       1.1     alnsn 	if (status != SLJIT_SUCCESS)
   1060       1.1     alnsn 		return status;
   1061       1.7     alnsn 
   1062      1.30     alnsn 	label = sljit_emit_label(compiler);
   1063      1.30     alnsn 	if (label == NULL)
   1064      1.30     alnsn 		return SLJIT_ERR_ALLOC_FAILED;
   1065      1.30     alnsn 	sljit_set_label(over_mchain_jump, label);
   1066      1.30     alnsn #endif
   1067      1.30     alnsn 
   1068       1.1     alnsn 	/* tmp1 &= 0xf */
   1069       1.1     alnsn 	status = sljit_emit_op2(compiler,
   1070       1.1     alnsn 	    SLJIT_AND,
   1071       1.7     alnsn 	    BJ_TMP1REG, 0,
   1072       1.7     alnsn 	    BJ_TMP1REG, 0,
   1073       1.1     alnsn 	    SLJIT_IMM, 0xf);
   1074       1.1     alnsn 	if (status != SLJIT_SUCCESS)
   1075       1.1     alnsn 		return status;
   1076       1.1     alnsn 
   1077      1.30     alnsn 	/* X = tmp1 << 2 */
   1078       1.1     alnsn 	status = sljit_emit_op2(compiler,
   1079       1.1     alnsn 	    SLJIT_SHL,
   1080       1.7     alnsn 	    BJ_XREG, 0,
   1081       1.7     alnsn 	    BJ_TMP1REG, 0,
   1082       1.1     alnsn 	    SLJIT_IMM, 2);
   1083       1.1     alnsn 	if (status != SLJIT_SUCCESS)
   1084       1.1     alnsn 		return status;
   1085       1.1     alnsn 
   1086      1.24     alnsn 	return SLJIT_SUCCESS;
   1087       1.1     alnsn }
   1088       1.1     alnsn 
   1089      1.35     alnsn /*
   1090      1.35     alnsn  * Emit code for A = A / k or A = A % k when k is a power of 2.
   1091      1.35     alnsn  * @pc BPF_DIV or BPF_MOD instruction.
   1092      1.35     alnsn  */
   1093       1.1     alnsn static int
   1094      1.35     alnsn emit_pow2_moddiv(struct sljit_compiler *compiler, const struct bpf_insn *pc)
   1095       1.1     alnsn {
   1096      1.35     alnsn 	uint32_t k = pc->k;
   1097       1.1     alnsn 	int status = SLJIT_SUCCESS;
   1098       1.1     alnsn 
   1099      1.35     alnsn 	BJ_ASSERT(k != 0 && (k & (k - 1)) == 0);
   1100       1.1     alnsn 
   1101      1.35     alnsn 	if (BPF_OP(pc->code) == BPF_MOD) {
   1102       1.1     alnsn 		status = sljit_emit_op2(compiler,
   1103      1.35     alnsn 		    SLJIT_AND,
   1104       1.7     alnsn 		    BJ_AREG, 0,
   1105       1.7     alnsn 		    BJ_AREG, 0,
   1106      1.35     alnsn 		    SLJIT_IMM, k - 1);
   1107      1.35     alnsn 	} else {
   1108      1.35     alnsn 		int shift = 0;
   1109      1.35     alnsn 
   1110      1.35     alnsn 		/*
   1111      1.35     alnsn 		 * Do shift = __builtin_ctz(k).
   1112      1.35     alnsn 		 * The loop is slower, but that's ok.
   1113      1.35     alnsn 		 */
   1114      1.35     alnsn 		while (k > 1) {
   1115      1.35     alnsn 			k >>= 1;
   1116      1.35     alnsn 			shift++;
   1117      1.35     alnsn 		}
   1118      1.35     alnsn 
   1119      1.35     alnsn 		if (shift != 0) {
   1120      1.35     alnsn 			status = sljit_emit_op2(compiler,
   1121      1.35     alnsn 			    SLJIT_LSHR|SLJIT_INT_OP,
   1122      1.35     alnsn 			    BJ_AREG, 0,
   1123      1.35     alnsn 			    BJ_AREG, 0,
   1124      1.35     alnsn 			    SLJIT_IMM, shift);
   1125      1.35     alnsn 		}
   1126       1.1     alnsn 	}
   1127       1.1     alnsn 
   1128       1.1     alnsn 	return status;
   1129       1.1     alnsn }
   1130       1.1     alnsn 
   1131       1.1     alnsn #if !defined(BPFJIT_USE_UDIV)
   1132       1.1     alnsn static sljit_uw
   1133       1.1     alnsn divide(sljit_uw x, sljit_uw y)
   1134       1.1     alnsn {
   1135       1.1     alnsn 
   1136       1.1     alnsn 	return (uint32_t)x / (uint32_t)y;
   1137       1.1     alnsn }
   1138      1.33  christos 
   1139      1.33  christos static sljit_uw
   1140      1.33  christos modulus(sljit_uw x, sljit_uw y)
   1141      1.33  christos {
   1142      1.33  christos 
   1143      1.33  christos 	return (uint32_t)x % (uint32_t)y;
   1144      1.33  christos }
   1145       1.1     alnsn #endif
   1146       1.1     alnsn 
   1147       1.1     alnsn /*
   1148      1.35     alnsn  * Emit code for A = A / div or A = A % div.
   1149      1.35     alnsn  * @pc BPF_DIV or BPF_MOD instruction.
   1150       1.1     alnsn  */
   1151       1.1     alnsn static int
   1152      1.35     alnsn emit_moddiv(struct sljit_compiler *compiler, const struct bpf_insn *pc)
   1153       1.1     alnsn {
   1154       1.1     alnsn 	int status;
   1155  1.36.2.1     skrll 	const bool xdiv = BPF_OP(pc->code) == BPF_DIV;
   1156      1.35     alnsn 	const bool xreg = BPF_SRC(pc->code) == BPF_X;
   1157       1.1     alnsn 
   1158      1.32     alnsn #if BJ_XREG == SLJIT_RETURN_REG   || \
   1159      1.32     alnsn     BJ_XREG == SLJIT_SCRATCH_REG1 || \
   1160      1.32     alnsn     BJ_XREG == SLJIT_SCRATCH_REG2 || \
   1161      1.32     alnsn     BJ_AREG == SLJIT_SCRATCH_REG2
   1162      1.32     alnsn #error "Not supported assignment of registers."
   1163      1.32     alnsn #endif
   1164      1.32     alnsn 
   1165      1.12     alnsn #if BJ_AREG != SLJIT_SCRATCH_REG1
   1166       1.1     alnsn 	status = sljit_emit_op1(compiler,
   1167       1.1     alnsn 	    SLJIT_MOV,
   1168      1.12     alnsn 	    SLJIT_SCRATCH_REG1, 0,
   1169       1.7     alnsn 	    BJ_AREG, 0);
   1170       1.1     alnsn 	if (status != SLJIT_SUCCESS)
   1171       1.1     alnsn 		return status;
   1172       1.1     alnsn #endif
   1173       1.1     alnsn 
   1174       1.1     alnsn 	status = sljit_emit_op1(compiler,
   1175       1.1     alnsn 	    SLJIT_MOV,
   1176      1.12     alnsn 	    SLJIT_SCRATCH_REG2, 0,
   1177      1.35     alnsn 	    xreg ? BJ_XREG : SLJIT_IMM,
   1178      1.35     alnsn 	    xreg ? 0 : (uint32_t)pc->k);
   1179       1.1     alnsn 	if (status != SLJIT_SUCCESS)
   1180       1.1     alnsn 		return status;
   1181       1.1     alnsn 
   1182       1.1     alnsn #if defined(BPFJIT_USE_UDIV)
   1183       1.1     alnsn 	status = sljit_emit_op0(compiler, SLJIT_UDIV|SLJIT_INT_OP);
   1184       1.1     alnsn 
   1185      1.36     alnsn 	if (BPF_OP(pc->code) == BPF_DIV) {
   1186      1.12     alnsn #if BJ_AREG != SLJIT_SCRATCH_REG1
   1187      1.36     alnsn 		status = sljit_emit_op1(compiler,
   1188      1.36     alnsn 		    SLJIT_MOV,
   1189      1.36     alnsn 		    BJ_AREG, 0,
   1190      1.36     alnsn 		    SLJIT_SCRATCH_REG1, 0);
   1191      1.36     alnsn #endif
   1192      1.36     alnsn 	} else {
   1193      1.36     alnsn #if BJ_AREG != SLJIT_SCRATCH_REG2
   1194      1.36     alnsn 		/* Remainder is in SLJIT_SCRATCH_REG2. */
   1195      1.36     alnsn 		status = sljit_emit_op1(compiler,
   1196      1.36     alnsn 		    SLJIT_MOV,
   1197      1.36     alnsn 		    BJ_AREG, 0,
   1198      1.36     alnsn 		    SLJIT_SCRATCH_REG2, 0);
   1199      1.36     alnsn #endif
   1200      1.36     alnsn 	}
   1201      1.36     alnsn 
   1202       1.1     alnsn 	if (status != SLJIT_SUCCESS)
   1203       1.1     alnsn 		return status;
   1204       1.1     alnsn #else
   1205       1.1     alnsn 	status = sljit_emit_ijump(compiler,
   1206       1.1     alnsn 	    SLJIT_CALL2,
   1207  1.36.2.1     skrll 	    SLJIT_IMM, xdiv ? SLJIT_FUNC_OFFSET(divide) :
   1208      1.33  christos 		SLJIT_FUNC_OFFSET(modulus));
   1209       1.1     alnsn 
   1210       1.7     alnsn #if BJ_AREG != SLJIT_RETURN_REG
   1211       1.1     alnsn 	status = sljit_emit_op1(compiler,
   1212       1.1     alnsn 	    SLJIT_MOV,
   1213       1.7     alnsn 	    BJ_AREG, 0,
   1214       1.1     alnsn 	    SLJIT_RETURN_REG, 0);
   1215       1.1     alnsn 	if (status != SLJIT_SUCCESS)
   1216       1.1     alnsn 		return status;
   1217       1.1     alnsn #endif
   1218       1.1     alnsn #endif
   1219       1.1     alnsn 
   1220       1.1     alnsn 	return status;
   1221       1.1     alnsn }
   1222       1.1     alnsn 
   1223       1.1     alnsn /*
   1224       1.1     alnsn  * Return true if pc is a "read from packet" instruction.
   1225       1.1     alnsn  * If length is not NULL and return value is true, *length will
   1226       1.1     alnsn  * be set to a safe length required to read a packet.
   1227       1.1     alnsn  */
   1228       1.1     alnsn static bool
   1229       1.8     alnsn read_pkt_insn(const struct bpf_insn *pc, bpfjit_abc_length_t *length)
   1230       1.1     alnsn {
   1231       1.1     alnsn 	bool rv;
   1232  1.36.2.1     skrll 	bpfjit_abc_length_t width = 0; /* XXXuninit */
   1233       1.1     alnsn 
   1234       1.1     alnsn 	switch (BPF_CLASS(pc->code)) {
   1235       1.1     alnsn 	default:
   1236       1.1     alnsn 		rv = false;
   1237       1.1     alnsn 		break;
   1238       1.1     alnsn 
   1239       1.1     alnsn 	case BPF_LD:
   1240       1.1     alnsn 		rv = BPF_MODE(pc->code) == BPF_ABS ||
   1241       1.1     alnsn 		     BPF_MODE(pc->code) == BPF_IND;
   1242  1.36.2.1     skrll 		if (rv) {
   1243       1.1     alnsn 			width = read_width(pc);
   1244  1.36.2.1     skrll 			rv = (width != 0);
   1245  1.36.2.1     skrll 		}
   1246       1.1     alnsn 		break;
   1247       1.1     alnsn 
   1248       1.1     alnsn 	case BPF_LDX:
   1249  1.36.2.1     skrll 		rv = BPF_MODE(pc->code) == BPF_MSH &&
   1250  1.36.2.1     skrll 		     BPF_SIZE(pc->code) == BPF_B;
   1251       1.1     alnsn 		width = 1;
   1252       1.1     alnsn 		break;
   1253       1.1     alnsn 	}
   1254       1.1     alnsn 
   1255       1.1     alnsn 	if (rv && length != NULL) {
   1256       1.9     alnsn 		/*
   1257       1.9     alnsn 		 * Values greater than UINT32_MAX will generate
   1258       1.9     alnsn 		 * unconditional "return 0".
   1259       1.9     alnsn 		 */
   1260       1.9     alnsn 		*length = (uint32_t)pc->k + width;
   1261       1.1     alnsn 	}
   1262       1.1     alnsn 
   1263       1.1     alnsn 	return rv;
   1264       1.1     alnsn }
   1265       1.1     alnsn 
   1266       1.1     alnsn static void
   1267       1.7     alnsn optimize_init(struct bpfjit_insn_data *insn_dat, size_t insn_count)
   1268       1.1     alnsn {
   1269       1.7     alnsn 	size_t i;
   1270       1.1     alnsn 
   1271       1.7     alnsn 	for (i = 0; i < insn_count; i++) {
   1272       1.7     alnsn 		SLIST_INIT(&insn_dat[i].bjumps);
   1273       1.7     alnsn 		insn_dat[i].invalid = BJ_INIT_NOBITS;
   1274       1.1     alnsn 	}
   1275       1.1     alnsn }
   1276       1.1     alnsn 
   1277       1.1     alnsn /*
   1278       1.1     alnsn  * The function divides instructions into blocks. Destination of a jump
   1279       1.1     alnsn  * instruction starts a new block. BPF_RET and BPF_JMP instructions
   1280       1.1     alnsn  * terminate a block. Blocks are linear, that is, there are no jumps out
   1281       1.1     alnsn  * from the middle of a block and there are no jumps in to the middle of
   1282       1.1     alnsn  * a block.
   1283       1.7     alnsn  *
   1284       1.7     alnsn  * The function also sets bits in *initmask for memwords that
   1285       1.7     alnsn  * need to be initialized to zero. Note that this set should be empty
   1286       1.7     alnsn  * for any valid kernel filter program.
   1287       1.1     alnsn  */
   1288       1.7     alnsn static bool
   1289      1.19     alnsn optimize_pass1(const bpf_ctx_t *bc, const struct bpf_insn *insns,
   1290      1.19     alnsn     struct bpfjit_insn_data *insn_dat, size_t insn_count,
   1291      1.20     alnsn     bpf_memword_init_t *initmask, bpfjit_hint_t *hints)
   1292       1.1     alnsn {
   1293       1.7     alnsn 	struct bpfjit_jump *jtf;
   1294       1.1     alnsn 	size_t i;
   1295       1.7     alnsn 	uint32_t jt, jf;
   1296      1.10     alnsn 	bpfjit_abc_length_t length;
   1297      1.13     alnsn 	bpf_memword_init_t invalid; /* borrowed from bpf_filter() */
   1298       1.1     alnsn 	bool unreachable;
   1299       1.1     alnsn 
   1300      1.19     alnsn 	const size_t memwords = GET_MEMWORDS(bc);
   1301      1.13     alnsn 
   1302      1.20     alnsn 	*hints = 0;
   1303       1.7     alnsn 	*initmask = BJ_INIT_NOBITS;
   1304       1.1     alnsn 
   1305       1.1     alnsn 	unreachable = false;
   1306       1.7     alnsn 	invalid = ~BJ_INIT_NOBITS;
   1307       1.1     alnsn 
   1308       1.1     alnsn 	for (i = 0; i < insn_count; i++) {
   1309       1.7     alnsn 		if (!SLIST_EMPTY(&insn_dat[i].bjumps))
   1310       1.1     alnsn 			unreachable = false;
   1311       1.7     alnsn 		insn_dat[i].unreachable = unreachable;
   1312       1.1     alnsn 
   1313       1.1     alnsn 		if (unreachable)
   1314       1.1     alnsn 			continue;
   1315       1.1     alnsn 
   1316       1.7     alnsn 		invalid |= insn_dat[i].invalid;
   1317       1.1     alnsn 
   1318      1.10     alnsn 		if (read_pkt_insn(&insns[i], &length) && length > UINT32_MAX)
   1319      1.10     alnsn 			unreachable = true;
   1320      1.10     alnsn 
   1321       1.1     alnsn 		switch (BPF_CLASS(insns[i].code)) {
   1322       1.1     alnsn 		case BPF_RET:
   1323       1.7     alnsn 			if (BPF_RVAL(insns[i].code) == BPF_A)
   1324       1.7     alnsn 				*initmask |= invalid & BJ_INIT_ABIT;
   1325       1.7     alnsn 
   1326       1.1     alnsn 			unreachable = true;
   1327       1.1     alnsn 			continue;
   1328       1.1     alnsn 
   1329       1.7     alnsn 		case BPF_LD:
   1330      1.27     alnsn 			if (BPF_MODE(insns[i].code) == BPF_ABS)
   1331      1.27     alnsn 				*hints |= BJ_HINT_ABS;
   1332       1.7     alnsn 
   1333      1.20     alnsn 			if (BPF_MODE(insns[i].code) == BPF_IND) {
   1334      1.27     alnsn 				*hints |= BJ_HINT_IND | BJ_HINT_XREG;
   1335       1.7     alnsn 				*initmask |= invalid & BJ_INIT_XBIT;
   1336      1.20     alnsn 			}
   1337       1.7     alnsn 
   1338       1.7     alnsn 			if (BPF_MODE(insns[i].code) == BPF_MEM &&
   1339      1.13     alnsn 			    (uint32_t)insns[i].k < memwords) {
   1340       1.7     alnsn 				*initmask |= invalid & BJ_INIT_MBIT(insns[i].k);
   1341       1.7     alnsn 			}
   1342       1.7     alnsn 
   1343       1.7     alnsn 			invalid &= ~BJ_INIT_ABIT;
   1344       1.7     alnsn 			continue;
   1345       1.7     alnsn 
   1346       1.7     alnsn 		case BPF_LDX:
   1347      1.20     alnsn 			*hints |= BJ_HINT_XREG | BJ_HINT_LDX;
   1348       1.7     alnsn 
   1349       1.7     alnsn 			if (BPF_MODE(insns[i].code) == BPF_MEM &&
   1350      1.13     alnsn 			    (uint32_t)insns[i].k < memwords) {
   1351       1.7     alnsn 				*initmask |= invalid & BJ_INIT_MBIT(insns[i].k);
   1352       1.7     alnsn 			}
   1353       1.7     alnsn 
   1354      1.29     alnsn 			if (BPF_MODE(insns[i].code) == BPF_MSH &&
   1355      1.29     alnsn 			    BPF_SIZE(insns[i].code) == BPF_B) {
   1356      1.29     alnsn 				*hints |= BJ_HINT_MSH;
   1357      1.29     alnsn 			}
   1358      1.29     alnsn 
   1359       1.7     alnsn 			invalid &= ~BJ_INIT_XBIT;
   1360       1.7     alnsn 			continue;
   1361       1.7     alnsn 
   1362       1.7     alnsn 		case BPF_ST:
   1363       1.7     alnsn 			*initmask |= invalid & BJ_INIT_ABIT;
   1364       1.7     alnsn 
   1365      1.13     alnsn 			if ((uint32_t)insns[i].k < memwords)
   1366       1.7     alnsn 				invalid &= ~BJ_INIT_MBIT(insns[i].k);
   1367       1.7     alnsn 
   1368       1.7     alnsn 			continue;
   1369       1.7     alnsn 
   1370       1.7     alnsn 		case BPF_STX:
   1371      1.20     alnsn 			*hints |= BJ_HINT_XREG;
   1372       1.7     alnsn 			*initmask |= invalid & BJ_INIT_XBIT;
   1373       1.7     alnsn 
   1374      1.13     alnsn 			if ((uint32_t)insns[i].k < memwords)
   1375       1.7     alnsn 				invalid &= ~BJ_INIT_MBIT(insns[i].k);
   1376       1.7     alnsn 
   1377       1.7     alnsn 			continue;
   1378       1.7     alnsn 
   1379       1.7     alnsn 		case BPF_ALU:
   1380       1.7     alnsn 			*initmask |= invalid & BJ_INIT_ABIT;
   1381       1.7     alnsn 
   1382       1.7     alnsn 			if (insns[i].code != (BPF_ALU|BPF_NEG) &&
   1383       1.7     alnsn 			    BPF_SRC(insns[i].code) == BPF_X) {
   1384      1.20     alnsn 				*hints |= BJ_HINT_XREG;
   1385       1.7     alnsn 				*initmask |= invalid & BJ_INIT_XBIT;
   1386       1.7     alnsn 			}
   1387       1.7     alnsn 
   1388       1.7     alnsn 			invalid &= ~BJ_INIT_ABIT;
   1389       1.7     alnsn 			continue;
   1390       1.7     alnsn 
   1391       1.7     alnsn 		case BPF_MISC:
   1392       1.7     alnsn 			switch (BPF_MISCOP(insns[i].code)) {
   1393       1.7     alnsn 			case BPF_TAX: // X <- A
   1394      1.20     alnsn 				*hints |= BJ_HINT_XREG;
   1395       1.7     alnsn 				*initmask |= invalid & BJ_INIT_ABIT;
   1396       1.7     alnsn 				invalid &= ~BJ_INIT_XBIT;
   1397       1.7     alnsn 				continue;
   1398       1.7     alnsn 
   1399       1.7     alnsn 			case BPF_TXA: // A <- X
   1400      1.20     alnsn 				*hints |= BJ_HINT_XREG;
   1401       1.7     alnsn 				*initmask |= invalid & BJ_INIT_XBIT;
   1402       1.7     alnsn 				invalid &= ~BJ_INIT_ABIT;
   1403       1.7     alnsn 				continue;
   1404      1.13     alnsn 
   1405      1.13     alnsn 			case BPF_COPX:
   1406      1.28     alnsn 				*hints |= BJ_HINT_XREG | BJ_HINT_COPX;
   1407      1.13     alnsn 				/* FALLTHROUGH */
   1408      1.13     alnsn 
   1409      1.13     alnsn 			case BPF_COP:
   1410      1.20     alnsn 				*hints |= BJ_HINT_COP;
   1411      1.13     alnsn 				*initmask |= invalid & BJ_INIT_ABIT;
   1412      1.13     alnsn 				invalid &= ~BJ_INIT_ABIT;
   1413      1.13     alnsn 				continue;
   1414       1.7     alnsn 			}
   1415       1.7     alnsn 
   1416       1.7     alnsn 			continue;
   1417       1.7     alnsn 
   1418       1.1     alnsn 		case BPF_JMP:
   1419       1.7     alnsn 			/* Initialize abc_length for ABC pass. */
   1420       1.8     alnsn 			insn_dat[i].u.jdata.abc_length = MAX_ABC_LENGTH;
   1421       1.7     alnsn 
   1422  1.36.2.1     skrll 			*initmask |= invalid & BJ_INIT_ABIT;
   1423  1.36.2.1     skrll 
   1424  1.36.2.1     skrll 			if (BPF_SRC(insns[i].code) == BPF_X) {
   1425  1.36.2.1     skrll 				*hints |= BJ_HINT_XREG;
   1426  1.36.2.1     skrll 				*initmask |= invalid & BJ_INIT_XBIT;
   1427  1.36.2.1     skrll 			}
   1428  1.36.2.1     skrll 
   1429       1.7     alnsn 			if (BPF_OP(insns[i].code) == BPF_JA) {
   1430       1.1     alnsn 				jt = jf = insns[i].k;
   1431       1.1     alnsn 			} else {
   1432       1.1     alnsn 				jt = insns[i].jt;
   1433       1.1     alnsn 				jf = insns[i].jf;
   1434       1.1     alnsn 			}
   1435       1.1     alnsn 
   1436       1.1     alnsn 			if (jt >= insn_count - (i + 1) ||
   1437       1.1     alnsn 			    jf >= insn_count - (i + 1)) {
   1438       1.7     alnsn 				return false;
   1439       1.1     alnsn 			}
   1440       1.1     alnsn 
   1441       1.1     alnsn 			if (jt > 0 && jf > 0)
   1442       1.1     alnsn 				unreachable = true;
   1443       1.1     alnsn 
   1444       1.7     alnsn 			jt += i + 1;
   1445       1.7     alnsn 			jf += i + 1;
   1446       1.7     alnsn 
   1447       1.7     alnsn 			jtf = insn_dat[i].u.jdata.jtf;
   1448       1.1     alnsn 
   1449       1.7     alnsn 			jtf[0].jdata = &insn_dat[i].u.jdata;
   1450       1.7     alnsn 			SLIST_INSERT_HEAD(&insn_dat[jt].bjumps,
   1451       1.7     alnsn 			    &jtf[0], entries);
   1452       1.1     alnsn 
   1453       1.1     alnsn 			if (jf != jt) {
   1454       1.7     alnsn 				jtf[1].jdata = &insn_dat[i].u.jdata;
   1455       1.7     alnsn 				SLIST_INSERT_HEAD(&insn_dat[jf].bjumps,
   1456       1.7     alnsn 				    &jtf[1], entries);
   1457       1.1     alnsn 			}
   1458       1.1     alnsn 
   1459       1.7     alnsn 			insn_dat[jf].invalid |= invalid;
   1460       1.7     alnsn 			insn_dat[jt].invalid |= invalid;
   1461       1.7     alnsn 			invalid = 0;
   1462       1.7     alnsn 
   1463       1.1     alnsn 			continue;
   1464       1.1     alnsn 		}
   1465       1.1     alnsn 	}
   1466       1.1     alnsn 
   1467       1.7     alnsn 	return true;
   1468       1.1     alnsn }
   1469       1.1     alnsn 
   1470       1.1     alnsn /*
   1471       1.7     alnsn  * Array Bounds Check Elimination (ABC) pass.
   1472       1.1     alnsn  */
   1473       1.7     alnsn static void
   1474      1.19     alnsn optimize_pass2(const bpf_ctx_t *bc, const struct bpf_insn *insns,
   1475      1.19     alnsn     struct bpfjit_insn_data *insn_dat, size_t insn_count)
   1476       1.7     alnsn {
   1477       1.7     alnsn 	struct bpfjit_jump *jmp;
   1478       1.7     alnsn 	const struct bpf_insn *pc;
   1479       1.7     alnsn 	struct bpfjit_insn_data *pd;
   1480       1.7     alnsn 	size_t i;
   1481       1.8     alnsn 	bpfjit_abc_length_t length, abc_length = 0;
   1482       1.7     alnsn 
   1483      1.19     alnsn 	const size_t extwords = GET_EXTWORDS(bc);
   1484      1.19     alnsn 
   1485       1.7     alnsn 	for (i = insn_count; i != 0; i--) {
   1486       1.7     alnsn 		pc = &insns[i-1];
   1487       1.7     alnsn 		pd = &insn_dat[i-1];
   1488       1.7     alnsn 
   1489       1.7     alnsn 		if (pd->unreachable)
   1490       1.7     alnsn 			continue;
   1491       1.7     alnsn 
   1492       1.7     alnsn 		switch (BPF_CLASS(pc->code)) {
   1493       1.7     alnsn 		case BPF_RET:
   1494      1.11     alnsn 			/*
   1495      1.11     alnsn 			 * It's quite common for bpf programs to
   1496      1.11     alnsn 			 * check packet bytes in increasing order
   1497      1.11     alnsn 			 * and return zero if bytes don't match
   1498      1.11     alnsn 			 * specified critetion. Such programs disable
   1499      1.11     alnsn 			 * ABC optimization completely because for
   1500      1.11     alnsn 			 * every jump there is a branch with no read
   1501      1.11     alnsn 			 * instruction.
   1502      1.13     alnsn 			 * With no side effects, BPF_STMT(BPF_RET+BPF_K, 0)
   1503      1.13     alnsn 			 * is indistinguishable from out-of-bound load.
   1504      1.11     alnsn 			 * Therefore, abc_length can be set to
   1505      1.11     alnsn 			 * MAX_ABC_LENGTH and enable ABC for many
   1506      1.11     alnsn 			 * bpf programs.
   1507      1.13     alnsn 			 * If this optimization encounters any
   1508      1.11     alnsn 			 * instruction with a side effect, it will
   1509      1.11     alnsn 			 * reset abc_length.
   1510      1.11     alnsn 			 */
   1511      1.11     alnsn 			if (BPF_RVAL(pc->code) == BPF_K && pc->k == 0)
   1512      1.11     alnsn 				abc_length = MAX_ABC_LENGTH;
   1513      1.11     alnsn 			else
   1514      1.11     alnsn 				abc_length = 0;
   1515       1.7     alnsn 			break;
   1516       1.7     alnsn 
   1517      1.13     alnsn 		case BPF_MISC:
   1518      1.13     alnsn 			if (BPF_MISCOP(pc->code) == BPF_COP ||
   1519      1.13     alnsn 			    BPF_MISCOP(pc->code) == BPF_COPX) {
   1520      1.13     alnsn 				/* COP instructions can have side effects. */
   1521      1.13     alnsn 				abc_length = 0;
   1522      1.13     alnsn 			}
   1523      1.13     alnsn 			break;
   1524      1.13     alnsn 
   1525      1.13     alnsn 		case BPF_ST:
   1526      1.13     alnsn 		case BPF_STX:
   1527      1.13     alnsn 			if (extwords != 0) {
   1528      1.13     alnsn 				/* Write to memory is visible after a call. */
   1529      1.13     alnsn 				abc_length = 0;
   1530      1.13     alnsn 			}
   1531      1.13     alnsn 			break;
   1532      1.13     alnsn 
   1533       1.7     alnsn 		case BPF_JMP:
   1534       1.7     alnsn 			abc_length = pd->u.jdata.abc_length;
   1535       1.7     alnsn 			break;
   1536       1.7     alnsn 
   1537       1.7     alnsn 		default:
   1538       1.7     alnsn 			if (read_pkt_insn(pc, &length)) {
   1539       1.7     alnsn 				if (abc_length < length)
   1540       1.7     alnsn 					abc_length = length;
   1541       1.7     alnsn 				pd->u.rdata.abc_length = abc_length;
   1542       1.7     alnsn 			}
   1543       1.7     alnsn 			break;
   1544       1.7     alnsn 		}
   1545       1.7     alnsn 
   1546       1.7     alnsn 		SLIST_FOREACH(jmp, &pd->bjumps, entries) {
   1547       1.7     alnsn 			if (jmp->jdata->abc_length > abc_length)
   1548       1.7     alnsn 				jmp->jdata->abc_length = abc_length;
   1549       1.7     alnsn 		}
   1550       1.7     alnsn 	}
   1551       1.7     alnsn }
   1552       1.7     alnsn 
   1553       1.7     alnsn static void
   1554       1.7     alnsn optimize_pass3(const struct bpf_insn *insns,
   1555       1.7     alnsn     struct bpfjit_insn_data *insn_dat, size_t insn_count)
   1556       1.1     alnsn {
   1557       1.7     alnsn 	struct bpfjit_jump *jmp;
   1558       1.1     alnsn 	size_t i;
   1559       1.8     alnsn 	bpfjit_abc_length_t checked_length = 0;
   1560       1.1     alnsn 
   1561       1.1     alnsn 	for (i = 0; i < insn_count; i++) {
   1562       1.7     alnsn 		if (insn_dat[i].unreachable)
   1563       1.7     alnsn 			continue;
   1564       1.1     alnsn 
   1565       1.7     alnsn 		SLIST_FOREACH(jmp, &insn_dat[i].bjumps, entries) {
   1566       1.7     alnsn 			if (jmp->jdata->checked_length < checked_length)
   1567       1.7     alnsn 				checked_length = jmp->jdata->checked_length;
   1568       1.1     alnsn 		}
   1569       1.1     alnsn 
   1570       1.7     alnsn 		if (BPF_CLASS(insns[i].code) == BPF_JMP) {
   1571       1.7     alnsn 			insn_dat[i].u.jdata.checked_length = checked_length;
   1572       1.8     alnsn 		} else if (read_pkt_insn(&insns[i], NULL)) {
   1573       1.7     alnsn 			struct bpfjit_read_pkt_data *rdata =
   1574       1.7     alnsn 			    &insn_dat[i].u.rdata;
   1575       1.7     alnsn 			rdata->check_length = 0;
   1576       1.7     alnsn 			if (checked_length < rdata->abc_length) {
   1577       1.7     alnsn 				checked_length = rdata->abc_length;
   1578       1.7     alnsn 				rdata->check_length = checked_length;
   1579       1.7     alnsn 			}
   1580       1.1     alnsn 		}
   1581       1.7     alnsn 	}
   1582       1.7     alnsn }
   1583       1.1     alnsn 
   1584       1.7     alnsn static bool
   1585      1.19     alnsn optimize(const bpf_ctx_t *bc, const struct bpf_insn *insns,
   1586       1.7     alnsn     struct bpfjit_insn_data *insn_dat, size_t insn_count,
   1587      1.20     alnsn     bpf_memword_init_t *initmask, bpfjit_hint_t *hints)
   1588       1.7     alnsn {
   1589       1.1     alnsn 
   1590       1.7     alnsn 	optimize_init(insn_dat, insn_count);
   1591       1.7     alnsn 
   1592      1.20     alnsn 	if (!optimize_pass1(bc, insns, insn_dat, insn_count, initmask, hints))
   1593       1.7     alnsn 		return false;
   1594       1.1     alnsn 
   1595      1.19     alnsn 	optimize_pass2(bc, insns, insn_dat, insn_count);
   1596       1.7     alnsn 	optimize_pass3(insns, insn_dat, insn_count);
   1597       1.7     alnsn 
   1598       1.7     alnsn 	return true;
   1599       1.1     alnsn }
   1600       1.1     alnsn 
   1601       1.1     alnsn /*
   1602       1.1     alnsn  * Convert BPF_ALU operations except BPF_NEG and BPF_DIV to sljit operation.
   1603       1.1     alnsn  */
   1604       1.1     alnsn static int
   1605       1.7     alnsn bpf_alu_to_sljit_op(const struct bpf_insn *pc)
   1606       1.1     alnsn {
   1607  1.36.2.1     skrll 	const int bad = SLJIT_UNUSED;
   1608  1.36.2.1     skrll 	const uint32_t k = pc->k;
   1609       1.1     alnsn 
   1610       1.1     alnsn 	/*
   1611       1.1     alnsn 	 * Note: all supported 64bit arches have 32bit multiply
   1612       1.1     alnsn 	 * instruction so SLJIT_INT_OP doesn't have any overhead.
   1613       1.1     alnsn 	 */
   1614       1.1     alnsn 	switch (BPF_OP(pc->code)) {
   1615       1.1     alnsn 	case BPF_ADD: return SLJIT_ADD;
   1616       1.1     alnsn 	case BPF_SUB: return SLJIT_SUB;
   1617       1.1     alnsn 	case BPF_MUL: return SLJIT_MUL|SLJIT_INT_OP;
   1618       1.1     alnsn 	case BPF_OR:  return SLJIT_OR;
   1619      1.33  christos 	case BPF_XOR: return SLJIT_XOR;
   1620       1.1     alnsn 	case BPF_AND: return SLJIT_AND;
   1621  1.36.2.1     skrll 	case BPF_LSH: return (k > 31) ? bad : SLJIT_SHL;
   1622  1.36.2.1     skrll 	case BPF_RSH: return (k > 31) ? bad : SLJIT_LSHR|SLJIT_INT_OP;
   1623       1.1     alnsn 	default:
   1624  1.36.2.1     skrll 		return bad;
   1625       1.1     alnsn 	}
   1626       1.1     alnsn }
   1627       1.1     alnsn 
   1628       1.1     alnsn /*
   1629       1.1     alnsn  * Convert BPF_JMP operations except BPF_JA to sljit condition.
   1630       1.1     alnsn  */
   1631       1.1     alnsn static int
   1632       1.7     alnsn bpf_jmp_to_sljit_cond(const struct bpf_insn *pc, bool negate)
   1633       1.1     alnsn {
   1634       1.1     alnsn 	/*
   1635       1.1     alnsn 	 * Note: all supported 64bit arches have 32bit comparison
   1636       1.1     alnsn 	 * instructions so SLJIT_INT_OP doesn't have any overhead.
   1637       1.1     alnsn 	 */
   1638       1.1     alnsn 	int rv = SLJIT_INT_OP;
   1639       1.1     alnsn 
   1640       1.1     alnsn 	switch (BPF_OP(pc->code)) {
   1641       1.1     alnsn 	case BPF_JGT:
   1642       1.1     alnsn 		rv |= negate ? SLJIT_C_LESS_EQUAL : SLJIT_C_GREATER;
   1643       1.1     alnsn 		break;
   1644       1.1     alnsn 	case BPF_JGE:
   1645       1.1     alnsn 		rv |= negate ? SLJIT_C_LESS : SLJIT_C_GREATER_EQUAL;
   1646       1.1     alnsn 		break;
   1647       1.1     alnsn 	case BPF_JEQ:
   1648       1.1     alnsn 		rv |= negate ? SLJIT_C_NOT_EQUAL : SLJIT_C_EQUAL;
   1649       1.1     alnsn 		break;
   1650       1.1     alnsn 	case BPF_JSET:
   1651       1.1     alnsn 		rv |= negate ? SLJIT_C_EQUAL : SLJIT_C_NOT_EQUAL;
   1652       1.1     alnsn 		break;
   1653       1.1     alnsn 	default:
   1654       1.7     alnsn 		BJ_ASSERT(false);
   1655       1.1     alnsn 	}
   1656       1.1     alnsn 
   1657       1.1     alnsn 	return rv;
   1658       1.1     alnsn }
   1659       1.1     alnsn 
   1660       1.1     alnsn /*
   1661       1.1     alnsn  * Convert BPF_K and BPF_X to sljit register.
   1662       1.1     alnsn  */
   1663       1.1     alnsn static int
   1664       1.7     alnsn kx_to_reg(const struct bpf_insn *pc)
   1665       1.1     alnsn {
   1666       1.1     alnsn 
   1667       1.1     alnsn 	switch (BPF_SRC(pc->code)) {
   1668       1.1     alnsn 	case BPF_K: return SLJIT_IMM;
   1669       1.7     alnsn 	case BPF_X: return BJ_XREG;
   1670       1.1     alnsn 	default:
   1671       1.7     alnsn 		BJ_ASSERT(false);
   1672       1.1     alnsn 		return 0;
   1673       1.1     alnsn 	}
   1674       1.1     alnsn }
   1675       1.1     alnsn 
   1676      1.12     alnsn static sljit_sw
   1677       1.7     alnsn kx_to_reg_arg(const struct bpf_insn *pc)
   1678       1.1     alnsn {
   1679       1.1     alnsn 
   1680       1.1     alnsn 	switch (BPF_SRC(pc->code)) {
   1681       1.1     alnsn 	case BPF_K: return (uint32_t)pc->k; /* SLJIT_IMM, pc->k, */
   1682       1.7     alnsn 	case BPF_X: return 0;               /* BJ_XREG, 0,      */
   1683       1.1     alnsn 	default:
   1684       1.7     alnsn 		BJ_ASSERT(false);
   1685       1.1     alnsn 		return 0;
   1686       1.1     alnsn 	}
   1687       1.1     alnsn }
   1688       1.1     alnsn 
   1689      1.19     alnsn static bool
   1690      1.32     alnsn generate_insn_code(struct sljit_compiler *compiler, bpfjit_hint_t hints,
   1691      1.32     alnsn     const bpf_ctx_t *bc, const struct bpf_insn *insns,
   1692      1.32     alnsn     struct bpfjit_insn_data *insn_dat, size_t insn_count)
   1693       1.1     alnsn {
   1694       1.1     alnsn 	/* a list of jumps to out-of-bound return from a generated function */
   1695       1.1     alnsn 	struct sljit_jump **ret0;
   1696       1.7     alnsn 	size_t ret0_size, ret0_maxsize;
   1697       1.1     alnsn 
   1698      1.19     alnsn 	struct sljit_jump *jump;
   1699      1.19     alnsn 	struct sljit_label *label;
   1700       1.7     alnsn 	const struct bpf_insn *pc;
   1701       1.1     alnsn 	struct bpfjit_jump *bjump, *jtf;
   1702       1.1     alnsn 	struct sljit_jump *to_mchain_jump;
   1703       1.1     alnsn 
   1704      1.19     alnsn 	size_t i;
   1705      1.19     alnsn 	int status;
   1706      1.19     alnsn 	int branching, negate;
   1707      1.33  christos 	unsigned int rval, mode, src, op;
   1708       1.1     alnsn 	uint32_t jt, jf;
   1709       1.1     alnsn 
   1710      1.19     alnsn 	bool unconditional_ret;
   1711      1.19     alnsn 	bool rv;
   1712      1.19     alnsn 
   1713      1.19     alnsn 	const size_t extwords = GET_EXTWORDS(bc);
   1714      1.19     alnsn 	const size_t memwords = GET_MEMWORDS(bc);
   1715      1.13     alnsn 
   1716      1.13     alnsn 	ret0 = NULL;
   1717      1.19     alnsn 	rv = false;
   1718       1.7     alnsn 
   1719       1.1     alnsn 	ret0_size = 0;
   1720       1.7     alnsn 	ret0_maxsize = 64;
   1721       1.7     alnsn 	ret0 = BJ_ALLOC(ret0_maxsize * sizeof(ret0[0]));
   1722       1.7     alnsn 	if (ret0 == NULL)
   1723       1.1     alnsn 		goto fail;
   1724       1.1     alnsn 
   1725      1.24     alnsn 	/* reset sjump members of jdata */
   1726      1.24     alnsn 	for (i = 0; i < insn_count; i++) {
   1727      1.24     alnsn 		if (insn_dat[i].unreachable ||
   1728      1.24     alnsn 		    BPF_CLASS(insns[i].code) != BPF_JMP) {
   1729      1.24     alnsn 			continue;
   1730      1.24     alnsn 		}
   1731      1.24     alnsn 
   1732      1.24     alnsn 		jtf = insn_dat[i].u.jdata.jtf;
   1733      1.24     alnsn 		jtf[0].sjump = jtf[1].sjump = NULL;
   1734      1.24     alnsn 	}
   1735      1.24     alnsn 
   1736      1.24     alnsn 	/* main loop */
   1737       1.1     alnsn 	for (i = 0; i < insn_count; i++) {
   1738       1.7     alnsn 		if (insn_dat[i].unreachable)
   1739       1.1     alnsn 			continue;
   1740       1.1     alnsn 
   1741       1.1     alnsn 		/*
   1742       1.1     alnsn 		 * Resolve jumps to the current insn.
   1743       1.1     alnsn 		 */
   1744       1.1     alnsn 		label = NULL;
   1745       1.7     alnsn 		SLIST_FOREACH(bjump, &insn_dat[i].bjumps, entries) {
   1746       1.7     alnsn 			if (bjump->sjump != NULL) {
   1747       1.1     alnsn 				if (label == NULL)
   1748       1.1     alnsn 					label = sljit_emit_label(compiler);
   1749       1.1     alnsn 				if (label == NULL)
   1750       1.1     alnsn 					goto fail;
   1751       1.7     alnsn 				sljit_set_label(bjump->sjump, label);
   1752       1.1     alnsn 			}
   1753       1.1     alnsn 		}
   1754       1.1     alnsn 
   1755       1.9     alnsn 		to_mchain_jump = NULL;
   1756       1.9     alnsn 		unconditional_ret = false;
   1757       1.9     alnsn 
   1758       1.9     alnsn 		if (read_pkt_insn(&insns[i], NULL)) {
   1759       1.9     alnsn 			if (insn_dat[i].u.rdata.check_length > UINT32_MAX) {
   1760       1.9     alnsn 				/* Jump to "return 0" unconditionally. */
   1761       1.9     alnsn 				unconditional_ret = true;
   1762       1.9     alnsn 				jump = sljit_emit_jump(compiler, SLJIT_JUMP);
   1763       1.9     alnsn 				if (jump == NULL)
   1764       1.9     alnsn 					goto fail;
   1765       1.9     alnsn 				if (!append_jump(jump, &ret0,
   1766       1.9     alnsn 				    &ret0_size, &ret0_maxsize))
   1767       1.9     alnsn 					goto fail;
   1768       1.9     alnsn 			} else if (insn_dat[i].u.rdata.check_length > 0) {
   1769       1.9     alnsn 				/* if (buflen < check_length) return 0; */
   1770       1.9     alnsn 				jump = sljit_emit_cmp(compiler,
   1771       1.9     alnsn 				    SLJIT_C_LESS,
   1772       1.9     alnsn 				    BJ_BUFLEN, 0,
   1773       1.9     alnsn 				    SLJIT_IMM,
   1774       1.9     alnsn 				    insn_dat[i].u.rdata.check_length);
   1775       1.9     alnsn 				if (jump == NULL)
   1776       1.9     alnsn 					goto fail;
   1777       1.1     alnsn #ifdef _KERNEL
   1778       1.9     alnsn 				to_mchain_jump = jump;
   1779       1.1     alnsn #else
   1780       1.9     alnsn 				if (!append_jump(jump, &ret0,
   1781       1.9     alnsn 				    &ret0_size, &ret0_maxsize))
   1782       1.9     alnsn 					goto fail;
   1783       1.1     alnsn #endif
   1784       1.9     alnsn 			}
   1785       1.1     alnsn 		}
   1786       1.1     alnsn 
   1787       1.1     alnsn 		pc = &insns[i];
   1788       1.1     alnsn 		switch (BPF_CLASS(pc->code)) {
   1789       1.1     alnsn 
   1790       1.1     alnsn 		default:
   1791       1.1     alnsn 			goto fail;
   1792       1.1     alnsn 
   1793       1.1     alnsn 		case BPF_LD:
   1794       1.1     alnsn 			/* BPF_LD+BPF_IMM          A <- k */
   1795       1.1     alnsn 			if (pc->code == (BPF_LD|BPF_IMM)) {
   1796       1.1     alnsn 				status = sljit_emit_op1(compiler,
   1797       1.1     alnsn 				    SLJIT_MOV,
   1798       1.7     alnsn 				    BJ_AREG, 0,
   1799       1.1     alnsn 				    SLJIT_IMM, (uint32_t)pc->k);
   1800       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1801       1.1     alnsn 					goto fail;
   1802       1.1     alnsn 
   1803       1.1     alnsn 				continue;
   1804       1.1     alnsn 			}
   1805       1.1     alnsn 
   1806       1.1     alnsn 			/* BPF_LD+BPF_MEM          A <- M[k] */
   1807       1.1     alnsn 			if (pc->code == (BPF_LD|BPF_MEM)) {
   1808      1.13     alnsn 				if ((uint32_t)pc->k >= memwords)
   1809       1.1     alnsn 					goto fail;
   1810      1.13     alnsn 				status = emit_memload(compiler,
   1811      1.13     alnsn 				    BJ_AREG, pc->k, extwords);
   1812       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1813       1.1     alnsn 					goto fail;
   1814       1.1     alnsn 
   1815       1.1     alnsn 				continue;
   1816       1.1     alnsn 			}
   1817       1.1     alnsn 
   1818       1.1     alnsn 			/* BPF_LD+BPF_W+BPF_LEN    A <- len */
   1819       1.1     alnsn 			if (pc->code == (BPF_LD|BPF_W|BPF_LEN)) {
   1820       1.1     alnsn 				status = sljit_emit_op1(compiler,
   1821      1.21     alnsn 				    SLJIT_MOV, /* size_t source */
   1822       1.7     alnsn 				    BJ_AREG, 0,
   1823      1.13     alnsn 				    SLJIT_MEM1(BJ_ARGS),
   1824      1.13     alnsn 				    offsetof(struct bpf_args, wirelen));
   1825       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1826       1.1     alnsn 					goto fail;
   1827       1.1     alnsn 
   1828       1.1     alnsn 				continue;
   1829       1.1     alnsn 			}
   1830       1.1     alnsn 
   1831       1.1     alnsn 			mode = BPF_MODE(pc->code);
   1832       1.1     alnsn 			if (mode != BPF_ABS && mode != BPF_IND)
   1833       1.1     alnsn 				goto fail;
   1834       1.1     alnsn 
   1835       1.9     alnsn 			if (unconditional_ret)
   1836       1.9     alnsn 				continue;
   1837       1.9     alnsn 
   1838      1.32     alnsn 			status = emit_pkt_read(compiler, hints, pc,
   1839       1.7     alnsn 			    to_mchain_jump, &ret0, &ret0_size, &ret0_maxsize);
   1840       1.1     alnsn 			if (status != SLJIT_SUCCESS)
   1841       1.1     alnsn 				goto fail;
   1842       1.1     alnsn 
   1843       1.1     alnsn 			continue;
   1844       1.1     alnsn 
   1845       1.1     alnsn 		case BPF_LDX:
   1846       1.1     alnsn 			mode = BPF_MODE(pc->code);
   1847       1.1     alnsn 
   1848       1.1     alnsn 			/* BPF_LDX+BPF_W+BPF_IMM    X <- k */
   1849       1.1     alnsn 			if (mode == BPF_IMM) {
   1850       1.1     alnsn 				if (BPF_SIZE(pc->code) != BPF_W)
   1851       1.1     alnsn 					goto fail;
   1852       1.1     alnsn 				status = sljit_emit_op1(compiler,
   1853       1.1     alnsn 				    SLJIT_MOV,
   1854       1.7     alnsn 				    BJ_XREG, 0,
   1855       1.1     alnsn 				    SLJIT_IMM, (uint32_t)pc->k);
   1856       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1857       1.1     alnsn 					goto fail;
   1858       1.1     alnsn 
   1859       1.1     alnsn 				continue;
   1860       1.1     alnsn 			}
   1861       1.1     alnsn 
   1862       1.1     alnsn 			/* BPF_LDX+BPF_W+BPF_LEN    X <- len */
   1863       1.1     alnsn 			if (mode == BPF_LEN) {
   1864       1.1     alnsn 				if (BPF_SIZE(pc->code) != BPF_W)
   1865       1.1     alnsn 					goto fail;
   1866       1.1     alnsn 				status = sljit_emit_op1(compiler,
   1867      1.21     alnsn 				    SLJIT_MOV, /* size_t source */
   1868       1.7     alnsn 				    BJ_XREG, 0,
   1869      1.13     alnsn 				    SLJIT_MEM1(BJ_ARGS),
   1870      1.13     alnsn 				    offsetof(struct bpf_args, wirelen));
   1871       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1872       1.1     alnsn 					goto fail;
   1873       1.1     alnsn 
   1874       1.1     alnsn 				continue;
   1875       1.1     alnsn 			}
   1876       1.1     alnsn 
   1877       1.1     alnsn 			/* BPF_LDX+BPF_W+BPF_MEM    X <- M[k] */
   1878       1.1     alnsn 			if (mode == BPF_MEM) {
   1879       1.1     alnsn 				if (BPF_SIZE(pc->code) != BPF_W)
   1880       1.1     alnsn 					goto fail;
   1881      1.13     alnsn 				if ((uint32_t)pc->k >= memwords)
   1882       1.1     alnsn 					goto fail;
   1883      1.13     alnsn 				status = emit_memload(compiler,
   1884      1.13     alnsn 				    BJ_XREG, pc->k, extwords);
   1885       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1886       1.1     alnsn 					goto fail;
   1887       1.1     alnsn 
   1888       1.1     alnsn 				continue;
   1889       1.1     alnsn 			}
   1890       1.1     alnsn 
   1891       1.1     alnsn 			/* BPF_LDX+BPF_B+BPF_MSH    X <- 4*(P[k:1]&0xf) */
   1892       1.1     alnsn 			if (mode != BPF_MSH || BPF_SIZE(pc->code) != BPF_B)
   1893       1.1     alnsn 				goto fail;
   1894       1.1     alnsn 
   1895       1.9     alnsn 			if (unconditional_ret)
   1896       1.9     alnsn 				continue;
   1897       1.9     alnsn 
   1898      1.32     alnsn 			status = emit_msh(compiler, hints, pc,
   1899       1.7     alnsn 			    to_mchain_jump, &ret0, &ret0_size, &ret0_maxsize);
   1900       1.1     alnsn 			if (status != SLJIT_SUCCESS)
   1901       1.1     alnsn 				goto fail;
   1902       1.1     alnsn 
   1903       1.1     alnsn 			continue;
   1904       1.1     alnsn 
   1905       1.1     alnsn 		case BPF_ST:
   1906       1.8     alnsn 			if (pc->code != BPF_ST ||
   1907      1.13     alnsn 			    (uint32_t)pc->k >= memwords) {
   1908       1.1     alnsn 				goto fail;
   1909       1.8     alnsn 			}
   1910       1.1     alnsn 
   1911      1.13     alnsn 			status = emit_memstore(compiler,
   1912      1.13     alnsn 			    BJ_AREG, pc->k, extwords);
   1913       1.1     alnsn 			if (status != SLJIT_SUCCESS)
   1914       1.1     alnsn 				goto fail;
   1915       1.1     alnsn 
   1916       1.1     alnsn 			continue;
   1917       1.1     alnsn 
   1918       1.1     alnsn 		case BPF_STX:
   1919       1.8     alnsn 			if (pc->code != BPF_STX ||
   1920      1.13     alnsn 			    (uint32_t)pc->k >= memwords) {
   1921       1.1     alnsn 				goto fail;
   1922       1.8     alnsn 			}
   1923       1.1     alnsn 
   1924      1.13     alnsn 			status = emit_memstore(compiler,
   1925      1.13     alnsn 			    BJ_XREG, pc->k, extwords);
   1926       1.1     alnsn 			if (status != SLJIT_SUCCESS)
   1927       1.1     alnsn 				goto fail;
   1928       1.1     alnsn 
   1929       1.1     alnsn 			continue;
   1930       1.1     alnsn 
   1931       1.1     alnsn 		case BPF_ALU:
   1932       1.1     alnsn 			if (pc->code == (BPF_ALU|BPF_NEG)) {
   1933       1.1     alnsn 				status = sljit_emit_op1(compiler,
   1934       1.1     alnsn 				    SLJIT_NEG,
   1935       1.7     alnsn 				    BJ_AREG, 0,
   1936       1.7     alnsn 				    BJ_AREG, 0);
   1937       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1938       1.1     alnsn 					goto fail;
   1939       1.1     alnsn 
   1940       1.1     alnsn 				continue;
   1941       1.1     alnsn 			}
   1942       1.1     alnsn 
   1943      1.33  christos 			op = BPF_OP(pc->code);
   1944      1.33  christos 			if (op != BPF_DIV && op != BPF_MOD) {
   1945  1.36.2.1     skrll 				const int op2 = bpf_alu_to_sljit_op(pc);
   1946  1.36.2.1     skrll 
   1947  1.36.2.1     skrll 				if (op2 == SLJIT_UNUSED)
   1948  1.36.2.1     skrll 					goto fail;
   1949       1.1     alnsn 				status = sljit_emit_op2(compiler,
   1950  1.36.2.1     skrll 				    op2, BJ_AREG, 0, BJ_AREG, 0,
   1951       1.1     alnsn 				    kx_to_reg(pc), kx_to_reg_arg(pc));
   1952       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1953       1.1     alnsn 					goto fail;
   1954       1.1     alnsn 
   1955       1.1     alnsn 				continue;
   1956       1.1     alnsn 			}
   1957       1.1     alnsn 
   1958      1.33  christos 			/* BPF_DIV/BPF_MOD */
   1959       1.1     alnsn 
   1960       1.1     alnsn 			src = BPF_SRC(pc->code);
   1961       1.1     alnsn 			if (src != BPF_X && src != BPF_K)
   1962       1.1     alnsn 				goto fail;
   1963       1.1     alnsn 
   1964       1.1     alnsn 			/* division by zero? */
   1965       1.1     alnsn 			if (src == BPF_X) {
   1966       1.1     alnsn 				jump = sljit_emit_cmp(compiler,
   1967       1.1     alnsn 				    SLJIT_C_EQUAL|SLJIT_INT_OP,
   1968       1.8     alnsn 				    BJ_XREG, 0,
   1969       1.1     alnsn 				    SLJIT_IMM, 0);
   1970       1.1     alnsn 				if (jump == NULL)
   1971       1.1     alnsn 					goto fail;
   1972       1.7     alnsn 				if (!append_jump(jump, &ret0,
   1973       1.7     alnsn 				    &ret0_size, &ret0_maxsize))
   1974       1.7     alnsn 					goto fail;
   1975       1.1     alnsn 			} else if (pc->k == 0) {
   1976       1.1     alnsn 				jump = sljit_emit_jump(compiler, SLJIT_JUMP);
   1977       1.1     alnsn 				if (jump == NULL)
   1978       1.1     alnsn 					goto fail;
   1979       1.7     alnsn 				if (!append_jump(jump, &ret0,
   1980       1.7     alnsn 				    &ret0_size, &ret0_maxsize))
   1981       1.7     alnsn 					goto fail;
   1982       1.1     alnsn 			}
   1983       1.1     alnsn 
   1984       1.1     alnsn 			if (src == BPF_X) {
   1985      1.35     alnsn 				status = emit_moddiv(compiler, pc);
   1986       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1987       1.1     alnsn 					goto fail;
   1988       1.1     alnsn 			} else if (pc->k != 0) {
   1989      1.35     alnsn 				if (pc->k & (pc->k - 1)) {
   1990      1.35     alnsn 					status = emit_moddiv(compiler, pc);
   1991       1.1     alnsn 				} else {
   1992      1.35     alnsn 					status = emit_pow2_moddiv(compiler, pc);
   1993       1.1     alnsn 				}
   1994       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   1995       1.1     alnsn 					goto fail;
   1996       1.1     alnsn 			}
   1997       1.1     alnsn 
   1998       1.1     alnsn 			continue;
   1999       1.1     alnsn 
   2000       1.1     alnsn 		case BPF_JMP:
   2001      1.33  christos 			op = BPF_OP(pc->code);
   2002      1.33  christos 			if (op == BPF_JA) {
   2003       1.1     alnsn 				jt = jf = pc->k;
   2004       1.1     alnsn 			} else {
   2005       1.1     alnsn 				jt = pc->jt;
   2006       1.1     alnsn 				jf = pc->jf;
   2007       1.1     alnsn 			}
   2008       1.1     alnsn 
   2009       1.1     alnsn 			negate = (jt == 0) ? 1 : 0;
   2010       1.1     alnsn 			branching = (jt == jf) ? 0 : 1;
   2011       1.7     alnsn 			jtf = insn_dat[i].u.jdata.jtf;
   2012       1.1     alnsn 
   2013       1.1     alnsn 			if (branching) {
   2014      1.33  christos 				if (op != BPF_JSET) {
   2015       1.1     alnsn 					jump = sljit_emit_cmp(compiler,
   2016       1.1     alnsn 					    bpf_jmp_to_sljit_cond(pc, negate),
   2017       1.7     alnsn 					    BJ_AREG, 0,
   2018       1.1     alnsn 					    kx_to_reg(pc), kx_to_reg_arg(pc));
   2019       1.1     alnsn 				} else {
   2020       1.1     alnsn 					status = sljit_emit_op2(compiler,
   2021       1.1     alnsn 					    SLJIT_AND,
   2022       1.7     alnsn 					    BJ_TMP1REG, 0,
   2023       1.7     alnsn 					    BJ_AREG, 0,
   2024       1.1     alnsn 					    kx_to_reg(pc), kx_to_reg_arg(pc));
   2025       1.1     alnsn 					if (status != SLJIT_SUCCESS)
   2026       1.1     alnsn 						goto fail;
   2027       1.1     alnsn 
   2028       1.1     alnsn 					jump = sljit_emit_cmp(compiler,
   2029       1.1     alnsn 					    bpf_jmp_to_sljit_cond(pc, negate),
   2030       1.7     alnsn 					    BJ_TMP1REG, 0,
   2031       1.1     alnsn 					    SLJIT_IMM, 0);
   2032       1.1     alnsn 				}
   2033       1.1     alnsn 
   2034       1.1     alnsn 				if (jump == NULL)
   2035       1.1     alnsn 					goto fail;
   2036       1.1     alnsn 
   2037       1.7     alnsn 				BJ_ASSERT(jtf[negate].sjump == NULL);
   2038       1.7     alnsn 				jtf[negate].sjump = jump;
   2039       1.1     alnsn 			}
   2040       1.1     alnsn 
   2041       1.1     alnsn 			if (!branching || (jt != 0 && jf != 0)) {
   2042       1.1     alnsn 				jump = sljit_emit_jump(compiler, SLJIT_JUMP);
   2043       1.1     alnsn 				if (jump == NULL)
   2044       1.1     alnsn 					goto fail;
   2045       1.1     alnsn 
   2046       1.7     alnsn 				BJ_ASSERT(jtf[branching].sjump == NULL);
   2047       1.7     alnsn 				jtf[branching].sjump = jump;
   2048       1.1     alnsn 			}
   2049       1.1     alnsn 
   2050       1.1     alnsn 			continue;
   2051       1.1     alnsn 
   2052       1.1     alnsn 		case BPF_RET:
   2053       1.1     alnsn 			rval = BPF_RVAL(pc->code);
   2054       1.1     alnsn 			if (rval == BPF_X)
   2055       1.1     alnsn 				goto fail;
   2056       1.1     alnsn 
   2057       1.1     alnsn 			/* BPF_RET+BPF_K    accept k bytes */
   2058       1.1     alnsn 			if (rval == BPF_K) {
   2059       1.7     alnsn 				status = sljit_emit_return(compiler,
   2060       1.7     alnsn 				    SLJIT_MOV_UI,
   2061       1.1     alnsn 				    SLJIT_IMM, (uint32_t)pc->k);
   2062       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   2063       1.1     alnsn 					goto fail;
   2064       1.1     alnsn 			}
   2065       1.1     alnsn 
   2066       1.1     alnsn 			/* BPF_RET+BPF_A    accept A bytes */
   2067       1.1     alnsn 			if (rval == BPF_A) {
   2068       1.7     alnsn 				status = sljit_emit_return(compiler,
   2069       1.7     alnsn 				    SLJIT_MOV_UI,
   2070       1.7     alnsn 				    BJ_AREG, 0);
   2071       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   2072       1.1     alnsn 					goto fail;
   2073       1.1     alnsn 			}
   2074       1.1     alnsn 
   2075       1.1     alnsn 			continue;
   2076       1.1     alnsn 
   2077       1.1     alnsn 		case BPF_MISC:
   2078       1.7     alnsn 			switch (BPF_MISCOP(pc->code)) {
   2079       1.7     alnsn 			case BPF_TAX:
   2080       1.1     alnsn 				status = sljit_emit_op1(compiler,
   2081       1.1     alnsn 				    SLJIT_MOV_UI,
   2082       1.7     alnsn 				    BJ_XREG, 0,
   2083       1.7     alnsn 				    BJ_AREG, 0);
   2084       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   2085       1.1     alnsn 					goto fail;
   2086       1.1     alnsn 
   2087       1.1     alnsn 				continue;
   2088       1.1     alnsn 
   2089       1.7     alnsn 			case BPF_TXA:
   2090       1.1     alnsn 				status = sljit_emit_op1(compiler,
   2091       1.1     alnsn 				    SLJIT_MOV,
   2092       1.7     alnsn 				    BJ_AREG, 0,
   2093       1.7     alnsn 				    BJ_XREG, 0);
   2094       1.1     alnsn 				if (status != SLJIT_SUCCESS)
   2095       1.1     alnsn 					goto fail;
   2096       1.1     alnsn 
   2097       1.1     alnsn 				continue;
   2098      1.13     alnsn 
   2099      1.13     alnsn 			case BPF_COP:
   2100      1.13     alnsn 			case BPF_COPX:
   2101      1.13     alnsn 				if (bc == NULL || bc->copfuncs == NULL)
   2102      1.13     alnsn 					goto fail;
   2103      1.13     alnsn 				if (BPF_MISCOP(pc->code) == BPF_COP &&
   2104      1.13     alnsn 				    (uint32_t)pc->k >= bc->nfuncs) {
   2105      1.13     alnsn 					goto fail;
   2106      1.13     alnsn 				}
   2107      1.13     alnsn 
   2108      1.32     alnsn 				status = emit_cop(compiler, hints, bc, pc,
   2109      1.28     alnsn 				    &ret0, &ret0_size, &ret0_maxsize);
   2110      1.13     alnsn 				if (status != SLJIT_SUCCESS)
   2111      1.13     alnsn 					goto fail;
   2112      1.13     alnsn 
   2113      1.13     alnsn 				continue;
   2114       1.1     alnsn 			}
   2115       1.1     alnsn 
   2116       1.1     alnsn 			goto fail;
   2117       1.1     alnsn 		} /* switch */
   2118       1.1     alnsn 	} /* main loop */
   2119       1.1     alnsn 
   2120       1.7     alnsn 	BJ_ASSERT(ret0_size <= ret0_maxsize);
   2121       1.1     alnsn 
   2122       1.7     alnsn 	if (ret0_size > 0) {
   2123       1.1     alnsn 		label = sljit_emit_label(compiler);
   2124       1.1     alnsn 		if (label == NULL)
   2125       1.1     alnsn 			goto fail;
   2126       1.7     alnsn 		for (i = 0; i < ret0_size; i++)
   2127       1.7     alnsn 			sljit_set_label(ret0[i], label);
   2128       1.1     alnsn 	}
   2129       1.1     alnsn 
   2130      1.23     alnsn 	status = sljit_emit_return(compiler,
   2131      1.23     alnsn 	    SLJIT_MOV_UI,
   2132      1.23     alnsn 	    SLJIT_IMM, 0);
   2133      1.23     alnsn 	if (status != SLJIT_SUCCESS)
   2134      1.23     alnsn 		goto fail;
   2135      1.23     alnsn 
   2136      1.19     alnsn 	rv = true;
   2137      1.19     alnsn 
   2138      1.19     alnsn fail:
   2139      1.19     alnsn 	if (ret0 != NULL)
   2140      1.19     alnsn 		BJ_FREE(ret0, ret0_maxsize * sizeof(ret0[0]));
   2141      1.19     alnsn 
   2142      1.19     alnsn 	return rv;
   2143      1.19     alnsn }
   2144      1.19     alnsn 
   2145      1.19     alnsn bpfjit_func_t
   2146      1.19     alnsn bpfjit_generate_code(const bpf_ctx_t *bc,
   2147      1.19     alnsn     const struct bpf_insn *insns, size_t insn_count)
   2148      1.19     alnsn {
   2149      1.19     alnsn 	void *rv;
   2150      1.19     alnsn 	struct sljit_compiler *compiler;
   2151      1.19     alnsn 
   2152      1.19     alnsn 	size_t i;
   2153      1.19     alnsn 	int status;
   2154      1.19     alnsn 
   2155      1.19     alnsn 	/* optimization related */
   2156      1.19     alnsn 	bpf_memword_init_t initmask;
   2157      1.20     alnsn 	bpfjit_hint_t hints;
   2158      1.19     alnsn 
   2159      1.19     alnsn 	/* memory store location for initial zero initialization */
   2160      1.19     alnsn 	sljit_si mem_reg;
   2161      1.19     alnsn 	sljit_sw mem_off;
   2162      1.19     alnsn 
   2163      1.19     alnsn 	struct bpfjit_insn_data *insn_dat;
   2164      1.19     alnsn 
   2165      1.19     alnsn 	const size_t extwords = GET_EXTWORDS(bc);
   2166      1.19     alnsn 	const size_t memwords = GET_MEMWORDS(bc);
   2167      1.19     alnsn 	const bpf_memword_init_t preinited = extwords ? bc->preinited : 0;
   2168      1.19     alnsn 
   2169      1.19     alnsn 	rv = NULL;
   2170      1.19     alnsn 	compiler = NULL;
   2171      1.19     alnsn 	insn_dat = NULL;
   2172      1.19     alnsn 
   2173      1.19     alnsn 	if (memwords > MAX_MEMWORDS)
   2174      1.19     alnsn 		goto fail;
   2175      1.19     alnsn 
   2176      1.19     alnsn 	if (insn_count == 0 || insn_count > SIZE_MAX / sizeof(insn_dat[0]))
   2177      1.19     alnsn 		goto fail;
   2178      1.19     alnsn 
   2179      1.19     alnsn 	insn_dat = BJ_ALLOC(insn_count * sizeof(insn_dat[0]));
   2180      1.19     alnsn 	if (insn_dat == NULL)
   2181      1.19     alnsn 		goto fail;
   2182      1.19     alnsn 
   2183      1.20     alnsn 	if (!optimize(bc, insns, insn_dat, insn_count, &initmask, &hints))
   2184      1.19     alnsn 		goto fail;
   2185      1.19     alnsn 
   2186      1.19     alnsn 	compiler = sljit_create_compiler();
   2187      1.19     alnsn 	if (compiler == NULL)
   2188      1.19     alnsn 		goto fail;
   2189      1.19     alnsn 
   2190      1.19     alnsn #if !defined(_KERNEL) && defined(SLJIT_VERBOSE) && SLJIT_VERBOSE
   2191      1.19     alnsn 	sljit_compiler_verbose(compiler, stderr);
   2192      1.19     alnsn #endif
   2193      1.19     alnsn 
   2194      1.30     alnsn 	status = sljit_emit_enter(compiler,
   2195      1.30     alnsn 	    2, nscratches(hints), nsaveds(hints), sizeof(struct bpfjit_stack));
   2196      1.19     alnsn 	if (status != SLJIT_SUCCESS)
   2197      1.19     alnsn 		goto fail;
   2198      1.19     alnsn 
   2199      1.20     alnsn 	if (hints & BJ_HINT_COP) {
   2200      1.19     alnsn 		/* save ctx argument */
   2201      1.19     alnsn 		status = sljit_emit_op1(compiler,
   2202      1.19     alnsn 		    SLJIT_MOV_P,
   2203      1.19     alnsn 		    SLJIT_MEM1(SLJIT_LOCALS_REG),
   2204      1.19     alnsn 		    offsetof(struct bpfjit_stack, ctx),
   2205      1.19     alnsn 		    BJ_CTX_ARG, 0);
   2206      1.19     alnsn 		if (status != SLJIT_SUCCESS)
   2207      1.19     alnsn 			goto fail;
   2208      1.19     alnsn 	}
   2209      1.19     alnsn 
   2210      1.19     alnsn 	if (extwords == 0) {
   2211      1.19     alnsn 		mem_reg = SLJIT_MEM1(SLJIT_LOCALS_REG);
   2212      1.19     alnsn 		mem_off = offsetof(struct bpfjit_stack, mem);
   2213      1.19     alnsn 	} else {
   2214      1.19     alnsn 		/* copy "mem" argument from bpf_args to bpfjit_stack */
   2215      1.19     alnsn 		status = sljit_emit_op1(compiler,
   2216      1.19     alnsn 		    SLJIT_MOV_P,
   2217      1.19     alnsn 		    BJ_TMP1REG, 0,
   2218      1.19     alnsn 		    SLJIT_MEM1(BJ_ARGS), offsetof(struct bpf_args, mem));
   2219      1.19     alnsn 		if (status != SLJIT_SUCCESS)
   2220      1.19     alnsn 			goto fail;
   2221      1.19     alnsn 
   2222      1.19     alnsn 		status = sljit_emit_op1(compiler,
   2223      1.19     alnsn 		    SLJIT_MOV_P,
   2224      1.19     alnsn 		    SLJIT_MEM1(SLJIT_LOCALS_REG),
   2225      1.19     alnsn 		    offsetof(struct bpfjit_stack, extmem),
   2226      1.19     alnsn 		    BJ_TMP1REG, 0);
   2227      1.19     alnsn 		if (status != SLJIT_SUCCESS)
   2228      1.19     alnsn 			goto fail;
   2229      1.19     alnsn 
   2230      1.19     alnsn 		mem_reg = SLJIT_MEM1(BJ_TMP1REG);
   2231      1.19     alnsn 		mem_off = 0;
   2232      1.19     alnsn 	}
   2233      1.19     alnsn 
   2234      1.19     alnsn 	/*
   2235      1.19     alnsn 	 * Exclude pre-initialised external memory words but keep
   2236      1.19     alnsn 	 * initialization statuses of A and X registers in case
   2237      1.19     alnsn 	 * bc->preinited wrongly sets those two bits.
   2238      1.19     alnsn 	 */
   2239      1.19     alnsn 	initmask &= ~preinited | BJ_INIT_ABIT | BJ_INIT_XBIT;
   2240      1.19     alnsn 
   2241      1.19     alnsn #if defined(_KERNEL)
   2242      1.19     alnsn 	/* bpf_filter() checks initialization of memwords. */
   2243      1.19     alnsn 	BJ_ASSERT((initmask & (BJ_INIT_MBIT(memwords) - 1)) == 0);
   2244      1.19     alnsn #endif
   2245      1.19     alnsn 	for (i = 0; i < memwords; i++) {
   2246      1.19     alnsn 		if (initmask & BJ_INIT_MBIT(i)) {
   2247      1.19     alnsn 			/* M[i] = 0; */
   2248      1.19     alnsn 			status = sljit_emit_op1(compiler,
   2249      1.19     alnsn 			    SLJIT_MOV_UI,
   2250      1.19     alnsn 			    mem_reg, mem_off + i * sizeof(uint32_t),
   2251      1.19     alnsn 			    SLJIT_IMM, 0);
   2252      1.19     alnsn 			if (status != SLJIT_SUCCESS)
   2253      1.19     alnsn 				goto fail;
   2254      1.19     alnsn 		}
   2255      1.19     alnsn 	}
   2256      1.19     alnsn 
   2257      1.19     alnsn 	if (initmask & BJ_INIT_ABIT) {
   2258      1.19     alnsn 		/* A = 0; */
   2259      1.19     alnsn 		status = sljit_emit_op1(compiler,
   2260      1.19     alnsn 		    SLJIT_MOV,
   2261      1.19     alnsn 		    BJ_AREG, 0,
   2262      1.19     alnsn 		    SLJIT_IMM, 0);
   2263      1.19     alnsn 		if (status != SLJIT_SUCCESS)
   2264      1.19     alnsn 			goto fail;
   2265      1.19     alnsn 	}
   2266      1.19     alnsn 
   2267      1.19     alnsn 	if (initmask & BJ_INIT_XBIT) {
   2268      1.19     alnsn 		/* X = 0; */
   2269      1.19     alnsn 		status = sljit_emit_op1(compiler,
   2270      1.19     alnsn 		    SLJIT_MOV,
   2271      1.19     alnsn 		    BJ_XREG, 0,
   2272      1.19     alnsn 		    SLJIT_IMM, 0);
   2273      1.19     alnsn 		if (status != SLJIT_SUCCESS)
   2274      1.19     alnsn 			goto fail;
   2275      1.19     alnsn 	}
   2276      1.19     alnsn 
   2277      1.19     alnsn 	status = load_buf_buflen(compiler);
   2278      1.19     alnsn 	if (status != SLJIT_SUCCESS)
   2279      1.19     alnsn 		goto fail;
   2280      1.19     alnsn 
   2281      1.32     alnsn 	if (!generate_insn_code(compiler, hints,
   2282      1.32     alnsn 	    bc, insns, insn_dat, insn_count)) {
   2283      1.19     alnsn 		goto fail;
   2284      1.32     alnsn 	}
   2285      1.19     alnsn 
   2286       1.1     alnsn 	rv = sljit_generate_code(compiler);
   2287       1.1     alnsn 
   2288       1.1     alnsn fail:
   2289       1.1     alnsn 	if (compiler != NULL)
   2290       1.1     alnsn 		sljit_free_compiler(compiler);
   2291       1.1     alnsn 
   2292       1.1     alnsn 	if (insn_dat != NULL)
   2293       1.7     alnsn 		BJ_FREE(insn_dat, insn_count * sizeof(insn_dat[0]));
   2294       1.1     alnsn 
   2295       1.4     rmind 	return (bpfjit_func_t)rv;
   2296       1.1     alnsn }
   2297       1.1     alnsn 
   2298       1.1     alnsn void
   2299       1.4     rmind bpfjit_free_code(bpfjit_func_t code)
   2300       1.1     alnsn {
   2301       1.7     alnsn 
   2302       1.1     alnsn 	sljit_free_code((void *)code);
   2303       1.1     alnsn }
   2304