Home | History | Annotate | Line # | Download | only in net
bpfjit.c revision 1.43
      1  1.43     alnsn /*	$NetBSD: bpfjit.c,v 1.43 2015/02/14 21:32:46 alnsn Exp $	*/
      2   1.3     rmind 
      3   1.1     alnsn /*-
      4  1.43     alnsn  * 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.43     alnsn __KERNEL_RCSID(0, "$NetBSD: bpfjit.c,v 1.43 2015/02/14 21:32:46 alnsn Exp $");
     35   1.2     alnsn #else
     36  1.43     alnsn __RCSID("$NetBSD: bpfjit.c,v 1.43 2015/02/14 21:32:46 alnsn 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.39     alnsn 	case BPF_W: return 4;
    294  1.39     alnsn 	case BPF_H: return 2;
    295  1.39     alnsn 	case BPF_B: return 1;
    296  1.39     alnsn 	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.39     alnsn 	if (width == 0)
    838  1.39     alnsn 		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.40     alnsn 	/*
    872  1.40     alnsn 	 * Don't emit wrapped-around reads. They're dead code but
    873  1.40     alnsn 	 * dead code elimination logic isn't smart enough to figure
    874  1.40     alnsn 	 * it out.
    875  1.40     alnsn 	 */
    876  1.40     alnsn 	if (k <= UINT32_MAX - width + 1) {
    877  1.40     alnsn 		switch (width) {
    878  1.40     alnsn 		case 4:
    879  1.40     alnsn 			status = emit_read32(compiler, ld_reg, k);
    880  1.40     alnsn 			break;
    881  1.40     alnsn 		case 2:
    882  1.40     alnsn 			status = emit_read16(compiler, ld_reg, k);
    883  1.40     alnsn 			break;
    884  1.40     alnsn 		case 1:
    885  1.40     alnsn 			status = emit_read8(compiler, ld_reg, k);
    886  1.40     alnsn 			break;
    887  1.40     alnsn 		}
    888  1.40     alnsn 
    889  1.40     alnsn 		if (status != SLJIT_SUCCESS)
    890  1.40     alnsn 			return status;
    891   1.1     alnsn 	}
    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.38  christos 	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.38  christos 	    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.37    justin 	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.39     alnsn 		if (rv) {
   1243   1.1     alnsn 			width = read_width(pc);
   1244  1.39     alnsn 			rv = (width != 0);
   1245  1.39     alnsn 		}
   1246   1.1     alnsn 		break;
   1247   1.1     alnsn 
   1248   1.1     alnsn 	case BPF_LDX:
   1249  1.39     alnsn 		rv = BPF_MODE(pc->code) == BPF_MSH &&
   1250  1.39     alnsn 		     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.41     alnsn 			*initmask |= invalid & BJ_INIT_ABIT;
   1423  1.41     alnsn 
   1424  1.41     alnsn 			if (BPF_SRC(insns[i].code) == BPF_X) {
   1425  1.39     alnsn 				*hints |= BJ_HINT_XREG;
   1426  1.41     alnsn 				*initmask |= invalid & BJ_INIT_XBIT;
   1427  1.41     alnsn 			}
   1428  1.39     alnsn 
   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.39     alnsn 	const int bad = SLJIT_UNUSED;
   1608  1.42     alnsn 	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.42     alnsn 	case BPF_LSH: return (k > 31) ? bad : SLJIT_SHL;
   1622  1.42     alnsn 	case BPF_RSH: return (k > 31) ? bad : SLJIT_LSHR|SLJIT_INT_OP;
   1623   1.1     alnsn 	default:
   1624  1.39     alnsn 		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.39     alnsn 				const int op2 = bpf_alu_to_sljit_op(pc);
   1946  1.39     alnsn 
   1947  1.39     alnsn 				if (op2 == SLJIT_UNUSED)
   1948  1.39     alnsn 					goto fail;
   1949   1.1     alnsn 				status = sljit_emit_op2(compiler,
   1950  1.39     alnsn 				    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