// SPDX-License-Identifier: GPL-2.0
/*
 * The ARCv2 backend of Just-In-Time compiler for eBPF bytecode.
 *
 * Copyright (c) 2024 Synopsys Inc.
 * Author: Shahab Vahedi <shahab@synopsys.com>
 */
#include <linux/bug.h>
#include "bpf_jit.h"

/* ARC core registers. */
enum {
        ARC_R_0,  ARC_R_1,  ARC_R_2,  ARC_R_3,  ARC_R_4,  ARC_R_5,
        ARC_R_6,  ARC_R_7,  ARC_R_8,  ARC_R_9,  ARC_R_10, ARC_R_11,
        ARC_R_12, ARC_R_13, ARC_R_14, ARC_R_15, ARC_R_16, ARC_R_17,
        ARC_R_18, ARC_R_19, ARC_R_20, ARC_R_21, ARC_R_22, ARC_R_23,
        ARC_R_24, ARC_R_25, ARC_R_26, ARC_R_FP, ARC_R_SP, ARC_R_ILINK,
        ARC_R_30, ARC_R_BLINK,
        /*
         * Having ARC_R_IMM encoded as source register means there is an
         * immediate that must be interpreted from the next 4 bytes. If
         * encoded as the destination register though, it implies that the
         * output of the operation is not assigned to any register. The
         * latter is helpful if we only care about updating the CPU status
         * flags.
         */
        ARC_R_IMM = 62
};

/*
 * Remarks about the rationale behind the chosen mapping:
 *
 * - BPF_REG_{1,2,3,4} are the argument registers and must be mapped to
 *   argument registers in ARCv2 ABI: r0-r7. The r7 registers is the last
 *   argument register in the ABI. Therefore BPF_REG_5, as the fifth
 *   argument, must be pushed onto the stack. This is a must for calling
 *   in-kernel functions.
 *
 * - In ARCv2 ABI, the return value is in r0 for 32-bit results and (r1,r0)
 *   for 64-bit results. However, because they're already used for BPF_REG_1,
 *   the next available scratch registers, r8 and r9, are the best candidates
 *   for BPF_REG_0. After a "call" to a(n) (in-kernel) function, the result
 *   is "mov"ed to these registers. At a BPF_EXIT, their value is "mov"ed to
 *   (r1,r0).
 *   It is worth mentioning that scratch registers are the best choice for
 *   BPF_REG_0, because it is very popular in BPF instruction encoding.
 *
 * - JIT_REG_TMP is an artifact needed to translate some BPF instructions.
 *   Its life span is one single BPF instruction. Since during the
 *   analyze_reg_usage(), it is not known if temporary registers are used,
 *   it is mapped to ARC's scratch registers: r10 and r11. Therefore, they
 *   don't matter in analysing phase and don't need saving. This temporary
 *   register is added as yet another index in the bpf2arc array, so it will
 *   unfold like the rest of registers during the code generation process.
 *
 * - Mapping of callee-saved BPF registers, BPF_REG_{6,7,8,9}, starts from
 *   (r15,r14) register pair. The (r13,r12) is not a good choice, because
 *   in ARCv2 ABI, r12 is not a callee-saved register and this can cause
 *   problem when calling an in-kernel function. Theoretically, the mapping
 *   could start from (r14,r13), but it is not a conventional ARCv2 register
 *   pair. To have a future proof design, I opted for this arrangement.
 *   If/when we decide to add ARCv2 instructions that do use register pairs,
 *   the mapping, hopefully, doesn't need to be revisited.
 */
static const u8 bpf2arc[][2] = {
        /* Return value from in-kernel function, and exit value from eBPF */
        [BPF_REG_0] = {ARC_R_8, ARC_R_9},
        /* Arguments from eBPF program to in-kernel function */
        [BPF_REG_1] = {ARC_R_0, ARC_R_1},
        [BPF_REG_2] = {ARC_R_2, ARC_R_3},
        [BPF_REG_3] = {ARC_R_4, ARC_R_5},
        [BPF_REG_4] = {ARC_R_6, ARC_R_7},
        /* Remaining arguments, to be passed on the stack per 32-bit ABI */
        [BPF_REG_5] = {ARC_R_22, ARC_R_23},
        /* Callee-saved registers that in-kernel function will preserve */
        [BPF_REG_6] = {ARC_R_14, ARC_R_15},
        [BPF_REG_7] = {ARC_R_16, ARC_R_17},
        [BPF_REG_8] = {ARC_R_18, ARC_R_19},
        [BPF_REG_9] = {ARC_R_20, ARC_R_21},
        /* Read-only frame pointer to access the eBPF stack. 32-bit only. */
        [BPF_REG_FP] = {ARC_R_FP, },
        /* Register for blinding constants */
        [BPF_REG_AX] = {ARC_R_24, ARC_R_25},
        /* Temporary registers for internal use */
        [JIT_REG_TMP] = {ARC_R_10, ARC_R_11}
};

#define ARC_CALLEE_SAVED_REG_FIRST ARC_R_13
#define ARC_CALLEE_SAVED_REG_LAST  ARC_R_25

#define REG_LO(r) (bpf2arc[(r)][0])
#define REG_HI(r) (bpf2arc[(r)][1])

/*
 * To comply with ARCv2 ABI, BPF's arg5 must be put on stack. After which,
 * the stack needs to be restored by ARG5_SIZE.
 */
#define ARG5_SIZE 8

/* Instruction lengths in bytes. */
enum {
        INSN_len_normal = 4,    /* Normal instructions length. */
        INSN_len_imm = 4        /* Length of an extra 32-bit immediate. */
};

/* ZZ defines the size of operation in encodings that it is used. */
enum {
        ZZ_1_byte = 1,
        ZZ_2_byte = 2,
        ZZ_4_byte = 0,
        ZZ_8_byte = 3
};

/*
 * AA is mostly about address write back mode. It determines if the
 * address in question should be updated before usage or after:
 *   addr += offset; data = *addr;
 *   data = *addr; addr += offset;
 *
 * In "scaling" mode, the effective address will become the sum
 * of "address" + "index"*"size". The "size" is specified by the
 * "ZZ" field. There is no write back when AA is set for scaling:
 *   data = *(addr + offset<<zz)
 */
enum {
        AA_none  = 0,
        AA_pre   = 1,   /* in assembly known as "a/aw". */
        AA_post  = 2,   /* in assembly known as "ab". */
        AA_scale = 3    /* in assembly known as "as". */
};

/* X flag determines the mode of extension. */
enum {
        X_zero = 0,
        X_sign = 1
};

/* Condition codes. */
enum {
        CC_always     = 0,      /* condition is true all the time */
        CC_equal      = 1,      /* if status32.z flag is set */
        CC_unequal    = 2,      /* if status32.z flag is clear */
        CC_positive   = 3,      /* if status32.n flag is clear */
        CC_negative   = 4,      /* if status32.n flag is set */
        CC_less_u     = 5,      /* less than (unsigned) */
        CC_less_eq_u  = 14,     /* less than or equal (unsigned) */
        CC_great_eq_u = 6,      /* greater than or equal (unsigned) */
        CC_great_u    = 13,     /* greater than (unsigned) */
        CC_less_s     = 11,     /* less than (signed) */
        CC_less_eq_s  = 12,     /* less than or equal (signed) */
        CC_great_eq_s = 10,     /* greater than or equal (signed) */
        CC_great_s    = 9       /* greater than (signed) */
};

#define IN_U6_RANGE(x)  ((x) <= (0x40      - 1) && (x) >= 0)
#define IN_S9_RANGE(x)  ((x) <= (0x100     - 1) && (x) >= -0x100)
#define IN_S12_RANGE(x) ((x) <= (0x800     - 1) && (x) >= -0x800)
#define IN_S21_RANGE(x) ((x) <= (0x100000  - 1) && (x) >= -0x100000)
#define IN_S25_RANGE(x) ((x) <= (0x1000000 - 1) && (x) >= -0x1000000)

/* Operands in most of the encodings. */
#define OP_A(x) ((x) & 0x03f)
#define OP_B(x) ((((x) & 0x07) << 24) | (((x) & 0x38) <<  9))
#define OP_C(x) (((x) & 0x03f) << 6)
#define OP_IMM  (OP_C(ARC_R_IMM))
#define COND(x) (OP_A((x) & 31))
#define FLAG(x) (((x) & 1) << 15)

/*
 * The 4-byte encoding of "mov b,c":
 *
 * 0010_0bbb 0000_1010 0BBB_cccc cc00_0000
 *
 * b:  BBBbbb           destination register
 * c:  cccccc           source register
 */
#define OPC_MOV         0x200a0000

/*
 * The 4-byte encoding of "mov b,s12" (used for moving small immediates):
 *
 * 0010_0bbb 1000_1010 0BBB_ssss ssSS_SSSS
 *
 * b:  BBBbbb           destination register
 * s:  SSSSSSssssss     source immediate (signed)
 */
#define OPC_MOVI        0x208a0000
#define MOVI_S12(x)     ((((x) & 0xfc0) >> 6) | (((x) & 0x3f) << 6))

/*
 * The 4-byte encoding of "mov[.qq] b,u6", used for conditional
 * moving of even smaller immediates:
 *
 * 0010_0bbb 1100_1010 0BBB_cccc cciq_qqqq
 *
 * qq: qqqqq            condition code
 * i:                   If set, c is considered a 6-bit immediate, else a reg.
 *
 * b:  BBBbbb           destination register
 * c:  cccccc           source
 */
#define OPC_MOV_CC      0x20ca0000
#define MOV_CC_I        BIT(5)
#define OPC_MOVU_CC     (OPC_MOV_CC | MOV_CC_I)

/*
 * The 4-byte encoding of "sexb b,c" (8-bit sign extension):
 *
 * 0010_0bbb 0010_1111 0BBB_cccc cc00_0101
 *
 * b:  BBBbbb           destination register
 * c:  cccccc           source register
 */
#define OPC_SEXB        0x202f0005

/*
 * The 4-byte encoding of "sexh b,c" (16-bit sign extension):
 *
 * 0010_0bbb 0010_1111 0BBB_cccc cc00_0110
 *
 * b:  BBBbbb           destination register
 * c:  cccccc           source register
 */
#define OPC_SEXH        0x202f0006

/*
 * The 4-byte encoding of "ld[zz][.x][.aa] c,[b,s9]":
 *
 * 0001_0bbb ssss_ssss SBBB_0aaz zxcc_cccc
 *
 * zz:                  size mode
 * aa:                  address write back mode
 * x:                   extension mode
 *
 * s9: S_ssss_ssss      9-bit signed number
 * b:  BBBbbb           source reg for address
 * c:  cccccc           destination register
 */
#define OPC_LOAD        0x10000000
#define LOAD_X(x)       ((x) << 6)
#define LOAD_ZZ(x)      ((x) << 7)
#define LOAD_AA(x)      ((x) << 9)
#define LOAD_S9(x)      ((((x) & 0x0ff) << 16) | (((x) & 0x100) <<  7))
#define LOAD_C(x)       ((x) & 0x03f)
/* Unsigned and signed loads. */
#define OPC_LDU         (OPC_LOAD | LOAD_X(X_zero))
#define OPC_LDS         (OPC_LOAD | LOAD_X(X_sign))
/* 32-bit load. */
#define OPC_LD32        (OPC_LDU | LOAD_ZZ(ZZ_4_byte))
/* "pop reg" is merely a "ld.ab reg,[sp,4]". */
#define OPC_POP         \
        (OPC_LD32 | LOAD_AA(AA_post) | LOAD_S9(4) | OP_B(ARC_R_SP))

/*
 * The 4-byte encoding of "st[zz][.aa] c,[b,s9]":
 *
 * 0001_1bbb ssss_ssss SBBB_cccc cc0a_azz0
 *
 * zz: zz               size mode
 * aa: aa               address write back mode
 *
 * s9: S_ssss_ssss      9-bit signed number
 * b:  BBBbbb           source reg for address
 * c:  cccccc           source reg to be stored
 */
#define OPC_STORE       0x18000000
#define STORE_ZZ(x)     ((x) << 1)
#define STORE_AA(x)     ((x) << 3)
#define STORE_S9(x)     ((((x) & 0x0ff) << 16) | (((x) & 0x100) <<  7))
/* 32-bit store. */
#define OPC_ST32        (OPC_STORE | STORE_ZZ(ZZ_4_byte))
/* "push reg" is merely a "st.aw reg,[sp,-4]". */
#define OPC_PUSH        \
        (OPC_ST32 | STORE_AA(AA_pre) | STORE_S9(-4) | OP_B(ARC_R_SP))

/*
 * The 4-byte encoding of "add a,b,c":
 *
 * 0010_0bbb 0i00_0000 fBBB_cccc ccaa_aaaa
 *
 * f:                   indicates if flags (carry, etc.) should be updated
 * i:                   If set, c is considered a 6-bit immediate, else a reg.
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_ADD         0x20000000
/* Addition with updating the pertinent flags in "status32" register. */
#define OPC_ADDF        (OPC_ADD | FLAG(1))
#define ADDI            BIT(22)
#define ADDI_U6(x)      OP_C(x)
#define OPC_ADDI        (OPC_ADD | ADDI)
#define OPC_ADDIF       (OPC_ADDI | FLAG(1))
#define OPC_ADD_I       (OPC_ADD | OP_IMM)

/*
 * The 4-byte encoding of "adc a,b,c" (addition with carry):
 *
 * 0010_0bbb 0i00_0001 0BBB_cccc ccaa_aaaa
 *
 * i:                   if set, c is considered a 6-bit immediate, else a reg.
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_ADC         0x20010000
#define ADCI            BIT(22)
#define ADCI_U6(x)      OP_C(x)
#define OPC_ADCI        (OPC_ADC | ADCI)

/*
 * The 4-byte encoding of "sub a,b,c":
 *
 * 0010_0bbb 0i00_0010 fBBB_cccc ccaa_aaaa
 *
 * f:                   indicates if flags (carry, etc.) should be updated
 * i:                   if set, c is considered a 6-bit immediate, else a reg.
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_SUB         0x20020000
/* Subtraction with updating the pertinent flags in "status32" register. */
#define OPC_SUBF        (OPC_SUB | FLAG(1))
#define SUBI            BIT(22)
#define SUBI_U6(x)      OP_C(x)
#define OPC_SUBI        (OPC_SUB | SUBI)
#define OPC_SUB_I       (OPC_SUB | OP_IMM)

/*
 * The 4-byte encoding of "sbc a,b,c" (subtraction with carry):
 *
 * 0010_0bbb 0000_0011 fBBB_cccc ccaa_aaaa
 *
 * f:                   indicates if flags (carry, etc.) should be updated
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_SBC         0x20030000

/*
 * The 4-byte encoding of "cmp[.qq] b,c":
 *
 * 0010_0bbb 1100_1100 1BBB_cccc cc0q_qqqq
 *
 * qq:  qqqqq           condition code
 *
 * b:  BBBbbb           the 1st operand
 * c:  cccccc           the 2nd operand
 */
#define OPC_CMP         0x20cc8000

/*
 * The 4-byte encoding of "neg a,b":
 *
 * 0010_0bbb 0100_1110 0BBB_0000 00aa_aaaa
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           input
 */
#define OPC_NEG         0x204e0000

/*
 * The 4-byte encoding of "mpy a,b,c".
 * mpy is the signed 32-bit multiplication with the lower 32-bit
 * of the product as the result.
 *
 * 0010_0bbb 0001_1010 0BBB_cccc ccaa_aaaa
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_MPY         0x201a0000
#define OPC_MPYI        (OPC_MPY | OP_IMM)

/*
 * The 4-byte encoding of "mpydu a,b,c".
 * mpydu is the unsigned 32-bit multiplication with the lower 32-bit of
 * the product in register "a" and the higher 32-bit in register "a+1".
 *
 * 0010_1bbb 0001_1001 0BBB_cccc ccaa_aaaa
 *
 * a:  aaaaaa           64-bit result in registers (R_a+1,R_a)
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_MPYDU       0x28190000
#define OPC_MPYDUI      (OPC_MPYDU | OP_IMM)

/*
 * The 4-byte encoding of "divu a,b,c" (unsigned division):
 *
 * 0010_1bbb 0000_0101 0BBB_cccc ccaa_aaaa
 *
 * a:  aaaaaa           result (quotient)
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand (divisor)
 */
#define OPC_DIVU        0x28050000
#define OPC_DIVUI       (OPC_DIVU | OP_IMM)

/*
 * The 4-byte encoding of "div a,b,c" (signed division):
 *
 * 0010_1bbb 0000_0100 0BBB_cccc ccaa_aaaa
 *
 * a:  aaaaaa           result (quotient)
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand (divisor)
 */
#define OPC_DIVS        0x28040000
#define OPC_DIVSI       (OPC_DIVS | OP_IMM)

/*
 * The 4-byte encoding of "remu a,b,c" (unsigned remainder):
 *
 * 0010_1bbb 0000_1001 0BBB_cccc ccaa_aaaa
 *
 * a:  aaaaaa           result (remainder)
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand (divisor)
 */
#define OPC_REMU        0x28090000
#define OPC_REMUI       (OPC_REMU | OP_IMM)

/*
 * The 4-byte encoding of "rem a,b,c" (signed remainder):
 *
 * 0010_1bbb 0000_1000 0BBB_cccc ccaa_aaaa
 *
 * a:  aaaaaa           result (remainder)
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand (divisor)
 */
#define OPC_REMS        0x28080000
#define OPC_REMSI       (OPC_REMS | OP_IMM)

/*
 * The 4-byte encoding of "and a,b,c":
 *
 * 0010_0bbb 0000_0100 fBBB_cccc ccaa_aaaa
 *
 * f:                   indicates if zero and negative flags should be updated
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_AND         0x20040000
#define OPC_ANDI        (OPC_AND | OP_IMM)

/*
 * The 4-byte encoding of "tst[.qq] b,c".
 * Checks if the two input operands have any bit set at the same
 * position.
 *
 * 0010_0bbb 1100_1011 1BBB_cccc cc0q_qqqq
 *
 * qq:  qqqqq           condition code
 *
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_TST         0x20cb8000

/*
 * The 4-byte encoding of "or a,b,c":
 *
 * 0010_0bbb 0000_0101 0BBB_cccc ccaa_aaaa
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_OR          0x20050000
#define OPC_ORI         (OPC_OR | OP_IMM)

/*
 * The 4-byte encoding of "xor a,b,c":
 *
 * 0010_0bbb 0000_0111 0BBB_cccc ccaa_aaaa
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           the 1st input operand
 * c:  cccccc           the 2nd input operand
 */
#define OPC_XOR         0x20070000
#define OPC_XORI        (OPC_XOR | OP_IMM)

/*
 * The 4-byte encoding of "not b,c":
 *
 * 0010_0bbb 0010_1111 0BBB_cccc cc00_1010
 *
 * b:  BBBbbb           result
 * c:  cccccc           input
 */
#define OPC_NOT         0x202f000a

/*
 * The 4-byte encoding of "btst b,u6":
 *
 * 0010_0bbb 0101_0001 1BBB_uuuu uu00_0000
 *
 * b:  BBBbbb           input number to check
 * u6: uuuuuu           6-bit unsigned number specifying bit position to check
 */
#define OPC_BTSTU6      0x20518000
#define BTST_U6(x)      (OP_C((x) & 63))

/*
 * The 4-byte encoding of "asl[.qq] b,b,c" (arithmetic shift left):
 *
 * 0010_1bbb 0i00_0000 0BBB_cccc ccaa_aaaa
 *
 * i:                   if set, c is considered a 5-bit immediate, else a reg.
 *
 * b:  BBBbbb           result and the first operand (number to be shifted)
 * c:  cccccc           amount to be shifted
 */
#define OPC_ASL         0x28000000
#define ASL_I           BIT(22)
#define ASLI_U6(x)      OP_C((x) & 31)
#define OPC_ASLI        (OPC_ASL | ASL_I)

/*
 * The 4-byte encoding of "asr a,b,c" (arithmetic shift right):
 *
 * 0010_1bbb 0i00_0010 0BBB_cccc ccaa_aaaa
 *
 * i:                   if set, c is considered a 6-bit immediate, else a reg.
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           first input:  number to be shifted
 * c:  cccccc           second input: amount to be shifted
 */
#define OPC_ASR         0x28020000
#define ASR_I           ASL_I
#define ASRI_U6(x)      ASLI_U6(x)
#define OPC_ASRI        (OPC_ASR | ASR_I)

/*
 * The 4-byte encoding of "lsr a,b,c" (logical shift right):
 *
 * 0010_1bbb 0i00_0001 0BBB_cccc ccaa_aaaa
 *
 * i:                   if set, c is considered a 6-bit immediate, else a reg.
 *
 * a:  aaaaaa           result
 * b:  BBBbbb           first input:  number to be shifted
 * c:  cccccc           second input: amount to be shifted
 */
#define OPC_LSR         0x28010000
#define LSR_I           ASL_I
#define LSRI_U6(x)      ASLI_U6(x)
#define OPC_LSRI        (OPC_LSR | LSR_I)

/*
 * The 4-byte encoding of "swape b,c":
 *
 * 0010_1bbb 0010_1111 0bbb_cccc cc00_1001
 *
 * b:  BBBbbb           destination register
 * c:  cccccc           source register
 */
#define OPC_SWAPE       0x282f0009

/*
 * Encoding for jump to an address in register:
 * j reg_c
 *
 * 0010_0000 1110_0000 0000_cccc cc00_0000
 *
 * c:  cccccc           register holding the destination address
 */
#define OPC_JMP         0x20e00000
/* Jump to "branch-and-link" register, which effectively is a "return". */
#define OPC_J_BLINK     (OPC_JMP | OP_C(ARC_R_BLINK))

/*
 * Encoding for jump-and-link to an address in register:
 * jl reg_c
 *
 * 0010_0000 0010_0010 0000_cccc cc00_0000
 *
 * c:  cccccc           register holding the destination address
 */
#define OPC_JL          0x20220000

/*
 * Encoding for (conditional) branch to an offset from the current location
 * that is word aligned: (PC & 0xffff_fffc) + s21
 * B[qq] s21
 *
 * 0000_0sss ssss_sss0 SSSS_SSSS SS0q_qqqq
 *
 * qq:  qqqqq                           condition code
 * s21: SSSS SSSS_SSss ssss_ssss        The displacement (21-bit signed)
 *
 * The displacement is supposed to be 16-bit (2-byte) aligned. Therefore,
 * it should be a multiple of 2. Hence, there is an implied '0' bit at its
 * LSB: S_SSSS SSSS_Ssss ssss_sss0
 */
#define OPC_BCC         0x00000000
#define BCC_S21(d)      ((((d) & 0x7fe) << 16) | (((d) & 0x1ff800) >> 5))

/*
 * Encoding for unconditional branch to an offset from the current location
 * that is word aligned: (PC & 0xffff_fffc) + s25
 * B s25
 *
 * 0000_0sss ssss_sss1 SSSS_SSSS SS00_TTTT
 *
 * s25: TTTT SSSS SSSS_SSss ssss_ssss   The displacement (25-bit signed)
 *
 * The displacement is supposed to be 16-bit (2-byte) aligned. Therefore,
 * it should be a multiple of 2. Hence, there is an implied '0' bit at its
 * LSB: T TTTS_SSSS SSSS_Ssss ssss_sss0
 */
#define OPC_B           0x00010000
#define B_S25(d)        ((((d) & 0x1e00000) >> 21) | BCC_S21(d))

static inline void emit_2_bytes(u8 *buf, u16 bytes)
{
        *((u16 *)buf) = bytes;
}

static inline void emit_4_bytes(u8 *buf, u32 bytes)
{
        emit_2_bytes(buf, bytes >> 16);
        emit_2_bytes(buf + 2, bytes & 0xffff);
}

static inline u8 bpf_to_arc_size(u8 size)
{
        switch (size) {
        case BPF_B:
                return ZZ_1_byte;
        case BPF_H:
                return ZZ_2_byte;
        case BPF_W:
                return ZZ_4_byte;
        case BPF_DW:
                return ZZ_8_byte;
        default:
                return ZZ_4_byte;
        }
}

/************** Encoders (Deal with ARC regs) ************/

/* Move an immediate to register with a 4-byte instruction. */
static u8 arc_movi_r(u8 *buf, u8 reg, s16 imm)
{
        const u32 insn = OPC_MOVI | OP_B(reg) | MOVI_S12(imm);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* rd <- rs */
static u8 arc_mov_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_MOV | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* The emitted code may have different sizes based on "imm". */
static u8 arc_mov_i(u8 *buf, u8 rd, s32 imm)
{
        const u32 insn = OPC_MOV | OP_B(rd) | OP_IMM;

        if (IN_S12_RANGE(imm))
                return arc_movi_r(buf, rd, imm);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* The emitted code will always have the same size (8). */
static u8 arc_mov_i_fixed(u8 *buf, u8 rd, s32 imm)
{
        const u32 insn = OPC_MOV | OP_B(rd) | OP_IMM;

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* Conditional move. */
static u8 arc_mov_cc_r(u8 *buf, u8 cc, u8 rd, u8 rs)
{
        const u32 insn = OPC_MOV_CC | OP_B(rd) | OP_C(rs) | COND(cc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* Conditional move of a small immediate to rd. */
static u8 arc_movu_cc_r(u8 *buf, u8 cc, u8 rd, u8 imm)
{
        const u32 insn = OPC_MOVU_CC | OP_B(rd) | OP_C(imm) | COND(cc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* Sign extension from a byte. */
static u8 arc_sexb_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_SEXB | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* Sign extension from two bytes. */
static u8 arc_sexh_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_SEXH | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* st reg, [reg_mem, off] */
static u8 arc_st_r(u8 *buf, u8 reg, u8 reg_mem, s16 off, u8 zz)
{
        const u32 insn = OPC_STORE | STORE_ZZ(zz) | OP_C(reg) |
                OP_B(reg_mem) | STORE_S9(off);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* st.aw reg, [sp, -4] */
static u8 arc_push_r(u8 *buf, u8 reg)
{
        const u32 insn = OPC_PUSH | OP_C(reg);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* ld reg, [reg_mem, off] (unsigned) */
static u8 arc_ld_r(u8 *buf, u8 reg, u8 reg_mem, s16 off, u8 zz)
{
        const u32 insn = OPC_LDU | LOAD_ZZ(zz) | LOAD_C(reg) |
                OP_B(reg_mem) | LOAD_S9(off);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* ld.x reg, [reg_mem, off] (sign extend) */
static u8 arc_ldx_r(u8 *buf, u8 reg, u8 reg_mem, s16 off, u8 zz)
{
        const u32 insn = OPC_LDS | LOAD_ZZ(zz) | LOAD_C(reg) |
                OP_B(reg_mem) | LOAD_S9(off);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* ld.ab reg,[sp,4] */
static u8 arc_pop_r(u8 *buf, u8 reg)
{
        const u32 insn = OPC_POP | LOAD_C(reg);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* add Ra,Ra,Rc */
static u8 arc_add_r(u8 *buf, u8 ra, u8 rc)
{
        const u32 insn = OPC_ADD | OP_A(ra) | OP_B(ra) | OP_C(rc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* add.f Ra,Ra,Rc */
static u8 arc_addf_r(u8 *buf, u8 ra, u8 rc)
{
        const u32 insn = OPC_ADDF | OP_A(ra) | OP_B(ra) | OP_C(rc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* add.f Ra,Ra,u6 */
static u8 arc_addif_r(u8 *buf, u8 ra, u8 u6)
{
        const u32 insn = OPC_ADDIF | OP_A(ra) | OP_B(ra) | ADDI_U6(u6);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* add Ra,Ra,u6 */
static u8 arc_addi_r(u8 *buf, u8 ra, u8 u6)
{
        const u32 insn = OPC_ADDI | OP_A(ra) | OP_B(ra) | ADDI_U6(u6);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* add Ra,Rb,imm */
static u8 arc_add_i(u8 *buf, u8 ra, u8 rb, s32 imm)
{
        const u32 insn = OPC_ADD_I | OP_A(ra) | OP_B(rb);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* adc Ra,Ra,Rc */
static u8 arc_adc_r(u8 *buf, u8 ra, u8 rc)
{
        const u32 insn = OPC_ADC | OP_A(ra) | OP_B(ra) | OP_C(rc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* adc Ra,Ra,u6 */
static u8 arc_adci_r(u8 *buf, u8 ra, u8 u6)
{
        const u32 insn = OPC_ADCI | OP_A(ra) | OP_B(ra) | ADCI_U6(u6);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* sub Ra,Ra,Rc */
static u8 arc_sub_r(u8 *buf, u8 ra, u8 rc)
{
        const u32 insn = OPC_SUB | OP_A(ra) | OP_B(ra) | OP_C(rc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* sub.f Ra,Ra,Rc */
static u8 arc_subf_r(u8 *buf, u8 ra, u8 rc)
{
        const u32 insn = OPC_SUBF | OP_A(ra) | OP_B(ra) | OP_C(rc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* sub Ra,Ra,u6 */
static u8 arc_subi_r(u8 *buf, u8 ra, u8 u6)
{
        const u32 insn = OPC_SUBI | OP_A(ra) | OP_B(ra) | SUBI_U6(u6);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* sub Ra,Ra,imm */
static u8 arc_sub_i(u8 *buf, u8 ra, s32 imm)
{
        const u32 insn = OPC_SUB_I | OP_A(ra) | OP_B(ra);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* sbc Ra,Ra,Rc */
static u8 arc_sbc_r(u8 *buf, u8 ra, u8 rc)
{
        const u32 insn = OPC_SBC | OP_A(ra) | OP_B(ra) | OP_C(rc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* cmp Rb,Rc */
static u8 arc_cmp_r(u8 *buf, u8 rb, u8 rc)
{
        const u32 insn = OPC_CMP | OP_B(rb) | OP_C(rc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/*
 * cmp.z Rb,Rc
 *
 * This "cmp.z" variant of compare instruction is used on lower
 * 32-bits of register pairs after "cmp"ing their upper parts. If the
 * upper parts are equal (z), then this one will proceed to check the
 * rest.
 */
static u8 arc_cmpz_r(u8 *buf, u8 rb, u8 rc)
{
        const u32 insn = OPC_CMP | OP_B(rb) | OP_C(rc) | CC_equal;

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* neg Ra,Rb */
static u8 arc_neg_r(u8 *buf, u8 ra, u8 rb)
{
        const u32 insn = OPC_NEG | OP_A(ra) | OP_B(rb);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* mpy Ra,Rb,Rc */
static u8 arc_mpy_r(u8 *buf, u8 ra, u8 rb, u8 rc)
{
        const u32 insn = OPC_MPY | OP_A(ra) | OP_B(rb) | OP_C(rc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* mpy Ra,Rb,imm */
static u8 arc_mpy_i(u8 *buf, u8 ra, u8 rb, s32 imm)
{
        const u32 insn = OPC_MPYI | OP_A(ra) | OP_B(rb);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* mpydu Ra,Ra,Rc */
static u8 arc_mpydu_r(u8 *buf, u8 ra, u8 rc)
{
        const u32 insn = OPC_MPYDU | OP_A(ra) | OP_B(ra) | OP_C(rc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* mpydu Ra,Ra,imm */
static u8 arc_mpydu_i(u8 *buf, u8 ra, s32 imm)
{
        const u32 insn = OPC_MPYDUI | OP_A(ra) | OP_B(ra);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* divu Rd,Rd,Rs */
static u8 arc_divu_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_DIVU | OP_A(rd) | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* divu Rd,Rd,imm */
static u8 arc_divu_i(u8 *buf, u8 rd, s32 imm)
{
        const u32 insn = OPC_DIVUI | OP_A(rd) | OP_B(rd);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* div Rd,Rd,Rs */
static u8 arc_divs_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_DIVS | OP_A(rd) | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* div Rd,Rd,imm */
static u8 arc_divs_i(u8 *buf, u8 rd, s32 imm)
{
        const u32 insn = OPC_DIVSI | OP_A(rd) | OP_B(rd);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* remu Rd,Rd,Rs */
static u8 arc_remu_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_REMU | OP_A(rd) | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* remu Rd,Rd,imm */
static u8 arc_remu_i(u8 *buf, u8 rd, s32 imm)
{
        const u32 insn = OPC_REMUI | OP_A(rd) | OP_B(rd);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* rem Rd,Rd,Rs */
static u8 arc_rems_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_REMS | OP_A(rd) | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* rem Rd,Rd,imm */
static u8 arc_rems_i(u8 *buf, u8 rd, s32 imm)
{
        const u32 insn = OPC_REMSI | OP_A(rd) | OP_B(rd);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* and Rd,Rd,Rs */
static u8 arc_and_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_AND | OP_A(rd) | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/* and Rd,Rd,limm */
static u8 arc_and_i(u8 *buf, u8 rd, s32 imm)
{
        const u32 insn = OPC_ANDI | OP_A(rd) | OP_B(rd);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

/* tst Rd,Rs */
static u8 arc_tst_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_TST | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/*
 * This particular version, "tst.z ...", is meant to be used after a
 * "tst" on the low 32-bit of register pairs. If that "tst" is not
 * zero, then we don't need to test the upper 32-bits lest it sets
 * the zero flag.
 */
static u8 arc_tstz_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_TST | OP_B(rd) | OP_C(rs) | CC_equal;

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_or_r(u8 *buf, u8 rd, u8 rs1, u8 rs2)
{
        const u32 insn = OPC_OR | OP_A(rd) | OP_B(rs1) | OP_C(rs2);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_or_i(u8 *buf, u8 rd, s32 imm)
{
        const u32 insn = OPC_ORI | OP_A(rd) | OP_B(rd);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

static u8 arc_xor_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_XOR | OP_A(rd) | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_xor_i(u8 *buf, u8 rd, s32 imm)
{
        const u32 insn = OPC_XORI | OP_A(rd) | OP_B(rd);

        if (buf) {
                emit_4_bytes(buf, insn);
                emit_4_bytes(buf + INSN_len_normal, imm);
        }
        return INSN_len_normal + INSN_len_imm;
}

static u8 arc_not_r(u8 *buf, u8 rd, u8 rs)
{
        const u32 insn = OPC_NOT | OP_B(rd) | OP_C(rs);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_btst_i(u8 *buf, u8 rs, u8 imm)
{
        const u32 insn = OPC_BTSTU6 | OP_B(rs) | BTST_U6(imm);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_asl_r(u8 *buf, u8 rd, u8 rs1, u8 rs2)
{
        const u32 insn = OPC_ASL | OP_A(rd) | OP_B(rs1) | OP_C(rs2);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_asli_r(u8 *buf, u8 rd, u8 rs, u8 imm)
{
        const u32 insn = OPC_ASLI | OP_A(rd) | OP_B(rs) | ASLI_U6(imm);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_asr_r(u8 *buf, u8 rd, u8 rs1, u8 rs2)
{
        const u32 insn = OPC_ASR | OP_A(rd) | OP_B(rs1) | OP_C(rs2);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_asri_r(u8 *buf, u8 rd, u8 rs, u8 imm)
{
        const u32 insn = OPC_ASRI | OP_A(rd) | OP_B(rs) | ASRI_U6(imm);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_lsr_r(u8 *buf, u8 rd, u8 rs1, u8 rs2)
{
        const u32 insn = OPC_LSR | OP_A(rd) | OP_B(rs1) | OP_C(rs2);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_lsri_r(u8 *buf, u8 rd, u8 rs, u8 imm)
{
        const u32 insn = OPC_LSRI | OP_A(rd) | OP_B(rs) | LSRI_U6(imm);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_swape_r(u8 *buf, u8 r)
{
        const u32 insn = OPC_SWAPE | OP_B(r) | OP_C(r);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

static u8 arc_jmp_return(u8 *buf)
{
        if (buf)
                emit_4_bytes(buf, OPC_J_BLINK);
        return INSN_len_normal;
}

static u8 arc_jl(u8 *buf, u8 reg)
{
        const u32 insn = OPC_JL | OP_C(reg);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/*
 * Conditional jump to an address that is max 21 bits away (signed).
 *
 * b<cc> s21
 */
static u8 arc_bcc(u8 *buf, u8 cc, int offset)
{
        const u32 insn = OPC_BCC | BCC_S21(offset) | COND(cc);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/*
 * Unconditional jump to an address that is max 25 bits away (signed).
 *
 * b     s25
 */
static u8 arc_b(u8 *buf, s32 offset)
{
        const u32 insn = OPC_B | B_S25(offset);

        if (buf)
                emit_4_bytes(buf, insn);
        return INSN_len_normal;
}

/************* Packers (Deal with BPF_REGs) **************/

u8 zext(u8 *buf, u8 rd)
{
        if (rd != BPF_REG_FP)
                return arc_movi_r(buf, REG_HI(rd), 0);
        else
                return 0;
}

u8 mov_r32(u8 *buf, u8 rd, u8 rs, u8 sign_ext)
{
        u8 len = 0;

        if (sign_ext) {
                if (sign_ext == 8)
                        len = arc_sexb_r(buf, REG_LO(rd), REG_LO(rs));
                else if (sign_ext == 16)
                        len = arc_sexh_r(buf, REG_LO(rd), REG_LO(rs));
                else if (sign_ext == 32 && rd != rs)
                        len = arc_mov_r(buf, REG_LO(rd), REG_LO(rs));

                return len;
        }

        /* Unsigned move. */

        if (rd != rs)
                len = arc_mov_r(buf, REG_LO(rd), REG_LO(rs));

        return len;
}

u8 mov_r32_i32(u8 *buf, u8 reg, s32 imm)
{
        return arc_mov_i(buf, REG_LO(reg), imm);
}

u8 mov_r64(u8 *buf, u8 rd, u8 rs, u8 sign_ext)
{
        u8 len = 0;

        if (sign_ext) {
                /* First handle the low 32-bit part. */
                len = mov_r32(buf, rd, rs, sign_ext);

                /* Now propagate the sign bit of LO to HI. */
                if (sign_ext == 8 || sign_ext == 16 || sign_ext == 32) {
                        len += arc_asri_r(BUF(buf, len),
                                          REG_HI(rd), REG_LO(rd), 31);
                }

                return len;
        }

        /* Unsigned move. */

        if (rd == rs)
                return 0;

        len = arc_mov_r(buf, REG_LO(rd), REG_LO(rs));

        if (rs != BPF_REG_FP)
                len += arc_mov_r(BUF(buf, len), REG_HI(rd), REG_HI(rs));
        /* BPF_REG_FP is mapped to 32-bit "fp" register. */
        else
                len += arc_movi_r(BUF(buf, len), REG_HI(rd), 0);

        return len;
}

/* Sign extend the 32-bit immediate into 64-bit register pair. */
u8 mov_r64_i32(u8 *buf, u8 reg, s32 imm)
{
        u8 len = 0;

        len = arc_mov_i(buf, REG_LO(reg), imm);

        /* BPF_REG_FP is mapped to 32-bit "fp" register. */
        if (reg != BPF_REG_FP) {
                if (imm >= 0)
                        len += arc_movi_r(BUF(buf, len), REG_HI(reg), 0);
                else
                        len += arc_movi_r(BUF(buf, len), REG_HI(reg), -1);
        }

        return len;
}

/*
 * This is merely used for translation of "LD R, IMM64" instructions
 * of the BPF. These sort of instructions are sometimes used for
 * relocations. If during the normal pass, the relocation value is
 * not known, the BPF instruction may look something like:
 *
 * LD R <- 0x0000_0001_0000_0001
 *
 * Which will nicely translate to two 4-byte ARC instructions:
 *
 * mov R_lo, 1               # imm is small enough to be s12
 * mov R_hi, 1               # same
 *
 * However, during the extra pass, the IMM64 will have changed
 * to the resolved address and looks something like:
 *
 * LD R <- 0x0000_0000_1234_5678
 *
 * Now, the translated code will require 12 bytes:
 *
 * mov R_lo, 0x12345678      # this is an 8-byte instruction
 * mov R_hi, 0               # still 4 bytes
 *
 * Which in practice will result in overwriting the following
 * instruction. To avoid such cases, we will always emit codes
 * with fixed sizes.
 */
u8 mov_r64_i64(u8 *buf, u8 reg, u32 lo, u32 hi)
{
        u8 len;

        len  = arc_mov_i_fixed(buf, REG_LO(reg), lo);
        len += arc_mov_i_fixed(BUF(buf, len), REG_HI(reg), hi);

        return len;
}

/*
 * If the "off"set is too big (doesn't encode as S9) for:
 *
 *   {ld,st}  r, [rm, off]
 *
 * Then emit:
 *
 *   add r10, REG_LO(rm), off
 *
 * and make sure that r10 becomes the effective address:
 *
 *   {ld,st}  r, [r10, 0]
 */
static u8 adjust_mem_access(u8 *buf, s16 *off, u8 size,
                            u8 rm, u8 *arc_reg_mem)
{
        u8 len = 0;
        *arc_reg_mem = REG_LO(rm);

        if (!IN_S9_RANGE(*off) ||
            (size == BPF_DW && !IN_S9_RANGE(*off + 4))) {
                len += arc_add_i(BUF(buf, len),
                                 REG_LO(JIT_REG_TMP), REG_LO(rm), (u32)(*off));
                *arc_reg_mem = REG_LO(JIT_REG_TMP);
                *off = 0;
        }

        return len;
}

/* store rs, [rd, off] */
u8 store_r(u8 *buf, u8 rs, u8 rd, s16 off, u8 size)
{
        u8 len, arc_reg_mem;

        len = adjust_mem_access(buf, &off, size, rd, &arc_reg_mem);

        if (size == BPF_DW) {
                len += arc_st_r(BUF(buf, len), REG_LO(rs), arc_reg_mem,
                                off, ZZ_4_byte);
                len += arc_st_r(BUF(buf, len), REG_HI(rs), arc_reg_mem,
                                off + 4, ZZ_4_byte);
        } else {
                u8 zz = bpf_to_arc_size(size);

                len += arc_st_r(BUF(buf, len), REG_LO(rs), arc_reg_mem,
                                off, zz);
        }

        return len;
}

/*
 * For {8,16,32}-bit stores:
 *   mov r21, imm
 *   st  r21, [...]
 * For 64-bit stores:
 *   mov r21, imm
 *   st  r21, [...]
 *   mov r21, {0,-1}
 *   st  r21, [...+4]
 */
u8 store_i(u8 *buf, s32 imm, u8 rd, s16 off, u8 size)
{
        u8 len, arc_reg_mem;
        /* REG_LO(JIT_REG_TMP) might be used by "adjust_mem_access()". */
        const u8 arc_rs = REG_HI(JIT_REG_TMP);

        len = adjust_mem_access(buf, &off, size, rd, &arc_reg_mem);

        if (size == BPF_DW) {
                len += arc_mov_i(BUF(buf, len), arc_rs, imm);
                len += arc_st_r(BUF(buf, len), arc_rs, arc_reg_mem,
                                off, ZZ_4_byte);
                imm = (imm >= 0 ? 0 : -1);
                len += arc_mov_i(BUF(buf, len), arc_rs, imm);
                len += arc_st_r(BUF(buf, len), arc_rs, arc_reg_mem,
                                off + 4, ZZ_4_byte);
        } else {
                u8 zz = bpf_to_arc_size(size);

                len += arc_mov_i(BUF(buf, len), arc_rs, imm);
                len += arc_st_r(BUF(buf, len), arc_rs, arc_reg_mem, off, zz);
        }

        return len;
}

/*
 * For the calling convention of a little endian machine, the LO part
 * must be on top of the stack.
 */
static u8 push_r64(u8 *buf, u8 reg)
{
        u8 len = 0;

#ifdef __LITTLE_ENDIAN
        /* BPF_REG_FP is mapped to 32-bit "fp" register. */
        if (reg != BPF_REG_FP)
                len += arc_push_r(BUF(buf, len), REG_HI(reg));
        len += arc_push_r(BUF(buf, len), REG_LO(reg));
#else
        len += arc_push_r(BUF(buf, len), REG_LO(reg));
        if (reg != BPF_REG_FP)
                len += arc_push_r(BUF(buf, len), REG_HI(reg));
#endif

        return len;
}

/* load rd, [rs, off] */
u8 load_r(u8 *buf, u8 rd, u8 rs, s16 off, u8 size, bool sign_ext)
{
        u8 len, arc_reg_mem;

        len = adjust_mem_access(buf, &off, size, rs, &arc_reg_mem);

        if (size == BPF_B || size == BPF_H || size == BPF_W) {
                const u8 zz = bpf_to_arc_size(size);

                /* Use LD.X only if the data size is less than 32-bit. */
                if (sign_ext && (zz == ZZ_1_byte || zz == ZZ_2_byte)) {
                        len += arc_ldx_r(BUF(buf, len), REG_LO(rd),
                                         arc_reg_mem, off, zz);
                } else {
                        len += arc_ld_r(BUF(buf, len), REG_LO(rd),
                                        arc_reg_mem, off, zz);
                }

                if (sign_ext) {
                        /* Propagate the sign bit to the higher reg. */
                        len += arc_asri_r(BUF(buf, len),
                                          REG_HI(rd), REG_LO(rd), 31);
                } else {
                        len += arc_movi_r(BUF(buf, len), REG_HI(rd), 0);
                }
        } else if (size == BPF_DW) {
                /*
                 * We are about to issue 2 consecutive loads:
                 *
                 *   ld rx, [rb, off+0]
                 *   ld ry, [rb, off+4]
                 *
                 * If "rx" and "rb" are the same registers, then the order
                 * should change to guarantee that "rb" remains intact
                 * during these 2 operations:
                 *
                 *   ld ry, [rb, off+4]
                 *   ld rx, [rb, off+0]
                 */
                if (REG_LO(rd) != arc_reg_mem) {
                        len += arc_ld_r(BUF(buf, len), REG_LO(rd), arc_reg_mem,
                                        off, ZZ_4_byte);
                        len += arc_ld_r(BUF(buf, len), REG_HI(rd), arc_reg_mem,
                                        off + 4, ZZ_4_byte);
                } else {
                        len += arc_ld_r(BUF(buf, len), REG_HI(rd), arc_reg_mem,
                                        off + 4, ZZ_4_byte);
                        len += arc_ld_r(BUF(buf, len), REG_LO(rd), arc_reg_mem,
                                        off, ZZ_4_byte);
                }
        }

        return len;
}

u8 add_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_add_r(buf, REG_LO(rd), REG_LO(rs));
}

u8 add_r32_i32(u8 *buf, u8 rd, s32 imm)
{
        if (IN_U6_RANGE(imm))
                return arc_addi_r(buf, REG_LO(rd), imm);
        else
                return arc_add_i(buf, REG_LO(rd), REG_LO(rd), imm);
}

u8 add_r64(u8 *buf, u8 rd, u8 rs)
{
        u8 len;

        len  = arc_addf_r(buf, REG_LO(rd), REG_LO(rs));
        len += arc_adc_r(BUF(buf, len), REG_HI(rd), REG_HI(rs));
        return len;
}

u8 add_r64_i32(u8 *buf, u8 rd, s32 imm)
{
        u8 len;

        if (IN_U6_RANGE(imm)) {
                len  = arc_addif_r(buf, REG_LO(rd), imm);
                len += arc_adci_r(BUF(buf, len), REG_HI(rd), 0);
        } else {
                len  = mov_r64_i32(buf, JIT_REG_TMP, imm);
                len += add_r64(BUF(buf, len), rd, JIT_REG_TMP);
        }
        return len;
}

u8 sub_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_sub_r(buf, REG_LO(rd), REG_LO(rs));
}

u8 sub_r32_i32(u8 *buf, u8 rd, s32 imm)
{
        if (IN_U6_RANGE(imm))
                return arc_subi_r(buf, REG_LO(rd), imm);
        else
                return arc_sub_i(buf, REG_LO(rd), imm);
}

u8 sub_r64(u8 *buf, u8 rd, u8 rs)
{
        u8 len;

        len  = arc_subf_r(buf, REG_LO(rd), REG_LO(rs));
        len += arc_sbc_r(BUF(buf, len), REG_HI(rd), REG_HI(rs));
        return len;
}

u8 sub_r64_i32(u8 *buf, u8 rd, s32 imm)
{
        u8 len;

        len  = mov_r64_i32(buf, JIT_REG_TMP, imm);
        len += sub_r64(BUF(buf, len), rd, JIT_REG_TMP);
        return len;
}

static u8 cmp_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_cmp_r(buf, REG_LO(rd), REG_LO(rs));
}

u8 neg_r32(u8 *buf, u8 r)
{
        return arc_neg_r(buf, REG_LO(r), REG_LO(r));
}

/* In a two's complement system, -r is (~r + 1). */
u8 neg_r64(u8 *buf, u8 r)
{
        u8 len;

        len  = arc_not_r(buf, REG_LO(r), REG_LO(r));
        len += arc_not_r(BUF(buf, len), REG_HI(r), REG_HI(r));
        len += add_r64_i32(BUF(buf, len), r, 1);
        return len;
}

u8 mul_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_mpy_r(buf, REG_LO(rd), REG_LO(rd), REG_LO(rs));
}

u8 mul_r32_i32(u8 *buf, u8 rd, s32 imm)
{
        return arc_mpy_i(buf, REG_LO(rd), REG_LO(rd), imm);
}

/*
 * MUL B, C
 * --------
 * mpy       t0, B_hi, C_lo
 * mpy       t1, B_lo, C_hi
 * mpydu   B_lo, B_lo, C_lo
 * add     B_hi, B_hi,   t0
 * add     B_hi, B_hi,   t1
 */
u8 mul_r64(u8 *buf, u8 rd, u8 rs)
{
        const u8 t0   = REG_LO(JIT_REG_TMP);
        const u8 t1   = REG_HI(JIT_REG_TMP);
        const u8 C_lo = REG_LO(rs);
        const u8 C_hi = REG_HI(rs);
        const u8 B_lo = REG_LO(rd);
        const u8 B_hi = REG_HI(rd);
        u8 len;

        len  = arc_mpy_r(buf, t0, B_hi, C_lo);
        len += arc_mpy_r(BUF(buf, len), t1, B_lo, C_hi);
        len += arc_mpydu_r(BUF(buf, len), B_lo, C_lo);
        len += arc_add_r(BUF(buf, len), B_hi, t0);
        len += arc_add_r(BUF(buf, len), B_hi, t1);

        return len;
}

/*
 * MUL B, imm
 * ----------
 *
 *  To get a 64-bit result from a signed 64x32 multiplication:
 *
 *         B_hi             B_lo   *
 *         sign             imm
 *  -----------------------------
 *  HI(B_lo*imm)     LO(B_lo*imm)  +
 *     B_hi*imm                    +
 *     B_lo*sign
 *  -----------------------------
 *        res_hi           res_lo
 *
 * mpy     t1, B_lo, sign(imm)
 * mpy     t0, B_hi, imm
 * mpydu B_lo, B_lo, imm
 * add   B_hi, B_hi,  t0
 * add   B_hi, B_hi,  t1
 *
 * Note: We can't use signed double multiplication, "mpyd", instead of an
 * unsigned version, "mpydu", and then get rid of the sign adjustments
 * calculated in "t1". The signed multiplication, "mpyd", will consider
 * both operands, "B_lo" and "imm", as signed inputs. However, for this
 * 64x32 multiplication, "B_lo" must be treated as an unsigned number.
 */
u8 mul_r64_i32(u8 *buf, u8 rd, s32 imm)
{
        const u8 t0   = REG_LO(JIT_REG_TMP);
        const u8 t1   = REG_HI(JIT_REG_TMP);
        const u8 B_lo = REG_LO(rd);
        const u8 B_hi = REG_HI(rd);
        u8 len = 0;

        if (imm == 1)
                return 0;

        /* Is the sign-extension of the immediate "-1"? */
        if (imm < 0)
                len += arc_neg_r(BUF(buf, len), t1, B_lo);

        len += arc_mpy_i(BUF(buf, len), t0, B_hi, imm);
        len += arc_mpydu_i(BUF(buf, len), B_lo, imm);
        len += arc_add_r(BUF(buf, len), B_hi, t0);

        /* Add the "sign*B_lo" part, if necessary. */
        if (imm < 0)
                len += arc_add_r(BUF(buf, len), B_hi, t1);

        return len;
}

u8 div_r32(u8 *buf, u8 rd, u8 rs, bool sign_ext)
{
        if (sign_ext)
                return arc_divs_r(buf, REG_LO(rd), REG_LO(rs));
        else
                return arc_divu_r(buf, REG_LO(rd), REG_LO(rs));
}

u8 div_r32_i32(u8 *buf, u8 rd, s32 imm, bool sign_ext)
{
        if (imm == 0)
                return 0;

        if (sign_ext)
                return arc_divs_i(buf, REG_LO(rd), imm);
        else
                return arc_divu_i(buf, REG_LO(rd), imm);
}

u8 mod_r32(u8 *buf, u8 rd, u8 rs, bool sign_ext)
{
        if (sign_ext)
                return arc_rems_r(buf, REG_LO(rd), REG_LO(rs));
        else
                return arc_remu_r(buf, REG_LO(rd), REG_LO(rs));
}

u8 mod_r32_i32(u8 *buf, u8 rd, s32 imm, bool sign_ext)
{
        if (imm == 0)
                return 0;

        if (sign_ext)
                return arc_rems_i(buf, REG_LO(rd), imm);
        else
                return arc_remu_i(buf, REG_LO(rd), imm);
}

u8 and_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_and_r(buf, REG_LO(rd), REG_LO(rs));
}

u8 and_r32_i32(u8 *buf, u8 rd, s32 imm)
{
        return arc_and_i(buf, REG_LO(rd), imm);
}

u8 and_r64(u8 *buf, u8 rd, u8 rs)
{
        u8 len;

        len  = arc_and_r(buf, REG_LO(rd), REG_LO(rs));
        len += arc_and_r(BUF(buf, len), REG_HI(rd), REG_HI(rs));
        return len;
}

u8 and_r64_i32(u8 *buf, u8 rd, s32 imm)
{
        u8 len;

        len  = mov_r64_i32(buf, JIT_REG_TMP, imm);
        len += and_r64(BUF(buf, len), rd, JIT_REG_TMP);
        return len;
}

static u8 tst_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_tst_r(buf, REG_LO(rd), REG_LO(rs));
}

u8 or_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_or_r(buf, REG_LO(rd), REG_LO(rd), REG_LO(rs));
}

u8 or_r32_i32(u8 *buf, u8 rd, s32 imm)
{
        return arc_or_i(buf, REG_LO(rd), imm);
}

u8 or_r64(u8 *buf, u8 rd, u8 rs)
{
        u8 len;

        len  = arc_or_r(buf, REG_LO(rd), REG_LO(rd), REG_LO(rs));
        len += arc_or_r(BUF(buf, len), REG_HI(rd), REG_HI(rd), REG_HI(rs));
        return len;
}

u8 or_r64_i32(u8 *buf, u8 rd, s32 imm)
{
        u8 len;

        len  = mov_r64_i32(buf, JIT_REG_TMP, imm);
        len += or_r64(BUF(buf, len), rd, JIT_REG_TMP);
        return len;
}

u8 xor_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_xor_r(buf, REG_LO(rd), REG_LO(rs));
}

u8 xor_r32_i32(u8 *buf, u8 rd, s32 imm)
{
        return arc_xor_i(buf, REG_LO(rd), imm);
}

u8 xor_r64(u8 *buf, u8 rd, u8 rs)
{
        u8 len;

        len  = arc_xor_r(buf, REG_LO(rd), REG_LO(rs));
        len += arc_xor_r(BUF(buf, len), REG_HI(rd), REG_HI(rs));
        return len;
}

u8 xor_r64_i32(u8 *buf, u8 rd, s32 imm)
{
        u8 len;

        len  = mov_r64_i32(buf, JIT_REG_TMP, imm);
        len += xor_r64(BUF(buf, len), rd, JIT_REG_TMP);
        return len;
}

/* "asl a,b,c" --> "a = (b << (c & 31))". */
u8 lsh_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_asl_r(buf, REG_LO(rd), REG_LO(rd), REG_LO(rs));
}

u8 lsh_r32_i32(u8 *buf, u8 rd, u8 imm)
{
        return arc_asli_r(buf, REG_LO(rd), REG_LO(rd), imm);
}

/*
 * algorithm
 * ---------
 * if (n <= 32)
 *   to_hi = lo >> (32-n)   # (32-n) is the negate of "n" in a 5-bit width.
 *   lo <<= n
 *   hi <<= n
 *   hi |= to_hi
 * else
 *   hi = lo << (n-32)
 *   lo = 0
 *
 * assembly translation for "LSH B, C"
 * (heavily influenced by ARC gcc)
 * -----------------------------------
 * not    t0, C_lo            # The first 3 lines are almost the same as:
 * lsr    t1, B_lo, 1         #   neg   t0, C_lo
 * lsr    t1, t1, t0          #   lsr   t1, B_lo, t0   --> t1 is "to_hi"
 * mov    t0, C_lo*           # with one important difference. In "neg"
 * asl    B_lo, B_lo, t0      # version, when C_lo=0, t1 becomes B_lo while
 * asl    B_hi, B_hi, t0      # it should be 0. The "not" approach instead,
 * or     B_hi, B_hi, t1      # "shift"s t1 once and 31 times, practically
 * btst   t0, 5               # setting it to 0 when C_lo=0.
 * mov.ne B_hi, B_lo**
 * mov.ne B_lo, 0
 *
 * *The "mov t0, C_lo" is necessary to cover the cases that C is the same
 * register as B.
 *
 * **ARC performs a shift in this manner: B <<= (C & 31)
 * For 32<=n<64, "n-32" and "n&31" are the same. Therefore, "B << n" and
 * "B << (n-32)" yield the same results. e.g. the results of "B << 35" and
 * "B << 3" are the same.
 *
 * The behaviour is undefined for n >= 64.
 */
u8 lsh_r64(u8 *buf, u8 rd, u8 rs)
{
        const u8 t0   = REG_LO(JIT_REG_TMP);
        const u8 t1   = REG_HI(JIT_REG_TMP);
        const u8 C_lo = REG_LO(rs);
        const u8 B_lo = REG_LO(rd);
        const u8 B_hi = REG_HI(rd);
        u8 len;

        len  = arc_not_r(buf, t0, C_lo);
        len += arc_lsri_r(BUF(buf, len), t1, B_lo, 1);
        len += arc_lsr_r(BUF(buf, len), t1, t1, t0);
        len += arc_mov_r(BUF(buf, len), t0, C_lo);
        len += arc_asl_r(BUF(buf, len), B_lo, B_lo, t0);
        len += arc_asl_r(BUF(buf, len), B_hi, B_hi, t0);
        len += arc_or_r(BUF(buf, len), B_hi, B_hi, t1);
        len += arc_btst_i(BUF(buf, len), t0, 5);
        len += arc_mov_cc_r(BUF(buf, len), CC_unequal, B_hi, B_lo);
        len += arc_movu_cc_r(BUF(buf, len), CC_unequal, B_lo, 0);

        return len;
}

/*
 * if (n < 32)
 *   to_hi = B_lo >> 32-n          # extract upper n bits
 *   lo <<= n
 *   hi <<=n
 *   hi |= to_hi
 * else if (n < 64)
 *   hi = lo << n-32
 *   lo = 0
 */
u8 lsh_r64_i32(u8 *buf, u8 rd, s32 imm)
{
        const u8 t0   = REG_LO(JIT_REG_TMP);
        const u8 B_lo = REG_LO(rd);
        const u8 B_hi = REG_HI(rd);
        const u8 n    = (u8)imm;
        u8 len = 0;

        if (n == 0) {
                return 0;
        } else if (n <= 31) {
                len  = arc_lsri_r(buf, t0, B_lo, 32 - n);
                len += arc_asli_r(BUF(buf, len), B_lo, B_lo, n);
                len += arc_asli_r(BUF(buf, len), B_hi, B_hi, n);
                len += arc_or_r(BUF(buf, len), B_hi, B_hi, t0);
        } else if (n <= 63) {
                len  = arc_asli_r(buf, B_hi, B_lo, n - 32);
                len += arc_movi_r(BUF(buf, len), B_lo, 0);
        }
        /* n >= 64 is undefined behaviour. */

        return len;
}

/* "lsr a,b,c" --> "a = (b >> (c & 31))". */
u8 rsh_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_lsr_r(buf, REG_LO(rd), REG_LO(rd), REG_LO(rs));
}

u8 rsh_r32_i32(u8 *buf, u8 rd, u8 imm)
{
        return arc_lsri_r(buf, REG_LO(rd), REG_LO(rd), imm);
}

/*
 * For better commentary, see lsh_r64().
 *
 * algorithm
 * ---------
 * if (n <= 32)
 *   to_lo = hi << (32-n)
 *   hi >>= n
 *   lo >>= n
 *   lo |= to_lo
 * else
 *   lo = hi >> (n-32)
 *   hi = 0
 *
 * RSH    B,C
 * ----------
 * not    t0, C_lo
 * asl    t1, B_hi, 1
 * asl    t1, t1, t0
 * mov    t0, C_lo
 * lsr    B_hi, B_hi, t0
 * lsr    B_lo, B_lo, t0
 * or     B_lo, B_lo, t1
 * btst   t0, 5
 * mov.ne B_lo, B_hi
 * mov.ne B_hi, 0
 */
u8 rsh_r64(u8 *buf, u8 rd, u8 rs)
{
        const u8 t0   = REG_LO(JIT_REG_TMP);
        const u8 t1   = REG_HI(JIT_REG_TMP);
        const u8 C_lo = REG_LO(rs);
        const u8 B_lo = REG_LO(rd);
        const u8 B_hi = REG_HI(rd);
        u8 len;

        len  = arc_not_r(buf, t0, C_lo);
        len += arc_asli_r(BUF(buf, len), t1, B_hi, 1);
        len += arc_asl_r(BUF(buf, len), t1, t1, t0);
        len += arc_mov_r(BUF(buf, len), t0, C_lo);
        len += arc_lsr_r(BUF(buf, len), B_hi, B_hi, t0);
        len += arc_lsr_r(BUF(buf, len), B_lo, B_lo, t0);
        len += arc_or_r(BUF(buf, len), B_lo, B_lo, t1);
        len += arc_btst_i(BUF(buf, len), t0, 5);
        len += arc_mov_cc_r(BUF(buf, len), CC_unequal, B_lo, B_hi);
        len += arc_movu_cc_r(BUF(buf, len), CC_unequal, B_hi, 0);

        return len;
}

/*
 * if (n < 32)
 *   to_lo = B_lo << 32-n     # extract lower n bits, right-padded with 32-n 0s
 *   lo >>=n
 *   hi >>=n
 *   hi |= to_lo
 * else if (n < 64)
 *   lo = hi >> n-32
 *   hi = 0
 */
u8 rsh_r64_i32(u8 *buf, u8 rd, s32 imm)
{
        const u8 t0   = REG_LO(JIT_REG_TMP);
        const u8 B_lo = REG_LO(rd);
        const u8 B_hi = REG_HI(rd);
        const u8 n    = (u8)imm;
        u8 len = 0;

        if (n == 0) {
                return 0;
        } else if (n <= 31) {
                len  = arc_asli_r(buf, t0, B_hi, 32 - n);
                len += arc_lsri_r(BUF(buf, len), B_lo, B_lo, n);
                len += arc_lsri_r(BUF(buf, len), B_hi, B_hi, n);
                len += arc_or_r(BUF(buf, len), B_lo, B_lo, t0);
        } else if (n <= 63) {
                len  = arc_lsri_r(buf, B_lo, B_hi, n - 32);
                len += arc_movi_r(BUF(buf, len), B_hi, 0);
        }
        /* n >= 64 is undefined behaviour. */

        return len;
}

/* "asr a,b,c" --> "a = (b s>> (c & 31))". */
u8 arsh_r32(u8 *buf, u8 rd, u8 rs)
{
        return arc_asr_r(buf, REG_LO(rd), REG_LO(rd), REG_LO(rs));
}

u8 arsh_r32_i32(u8 *buf, u8 rd, u8 imm)
{
        return arc_asri_r(buf, REG_LO(rd), REG_LO(rd), imm);
}

/*
 * For comparison, see rsh_r64().
 *
 * algorithm
 * ---------
 * if (n <= 32)
 *   to_lo = hi << (32-n)
 *   hi s>>= n
 *   lo  >>= n
 *   lo |= to_lo
 * else
 *   hi_sign = hi s>>31
 *   lo = hi s>> (n-32)
 *   hi = hi_sign
 *
 * ARSH   B,C
 * ----------
 * not    t0, C_lo
 * asl    t1, B_hi, 1
 * asl    t1, t1, t0
 * mov    t0, C_lo
 * asr    B_hi, B_hi, t0
 * lsr    B_lo, B_lo, t0
 * or     B_lo, B_lo, t1
 * btst   t0, 5
 * asr    t0, B_hi, 31        # now, t0 = 0 or -1 based on B_hi's sign
 * mov.ne B_lo, B_hi
 * mov.ne B_hi, t0
 */
u8 arsh_r64(u8 *buf, u8 rd, u8 rs)
{
        const u8 t0   = REG_LO(JIT_REG_TMP);
        const u8 t1   = REG_HI(JIT_REG_TMP);
        const u8 C_lo = REG_LO(rs);
        const u8 B_lo = REG_LO(rd);
        const u8 B_hi = REG_HI(rd);
        u8 len;

        len  = arc_not_r(buf, t0, C_lo);
        len += arc_asli_r(BUF(buf, len), t1, B_hi, 1);
        len += arc_asl_r(BUF(buf, len), t1, t1, t0);
        len += arc_mov_r(BUF(buf, len), t0, C_lo);
        len += arc_asr_r(BUF(buf, len), B_hi, B_hi, t0);
        len += arc_lsr_r(BUF(buf, len), B_lo, B_lo, t0);
        len += arc_or_r(BUF(buf, len), B_lo, B_lo, t1);
        len += arc_btst_i(BUF(buf, len), t0, 5);
        len += arc_asri_r(BUF(buf, len), t0, B_hi, 31);
        len += arc_mov_cc_r(BUF(buf, len), CC_unequal, B_lo, B_hi);
        len += arc_mov_cc_r(BUF(buf, len), CC_unequal, B_hi, t0);

        return len;
}

/*
 * if (n < 32)
 *   to_lo = lo << 32-n     # extract lower n bits, right-padded with 32-n 0s
 *   lo >>=n
 *   hi s>>=n
 *   hi |= to_lo
 * else if (n < 64)
 *   lo = hi s>> n-32
 *   hi = (lo[msb] ? -1 : 0)
 */
u8 arsh_r64_i32(u8 *buf, u8 rd, s32 imm)
{
        const u8 t0   = REG_LO(JIT_REG_TMP);
        const u8 B_lo = REG_LO(rd);
        const u8 B_hi = REG_HI(rd);
        const u8 n    = (u8)imm;
        u8 len = 0;

        if (n == 0) {
                return 0;
        } else if (n <= 31) {
                len  = arc_asli_r(buf, t0, B_hi, 32 - n);
                len += arc_lsri_r(BUF(buf, len), B_lo, B_lo, n);
                len += arc_asri_r(BUF(buf, len), B_hi, B_hi, n);
                len += arc_or_r(BUF(buf, len), B_lo, B_lo, t0);
        } else if (n <= 63) {
                len  = arc_asri_r(buf, B_lo, B_hi, n - 32);
                len += arc_movi_r(BUF(buf, len), B_hi, -1);
                len += arc_btst_i(BUF(buf, len), B_lo, 31);
                len += arc_movu_cc_r(BUF(buf, len), CC_equal, B_hi, 0);
        }
        /* n >= 64 is undefined behaviour. */

        return len;
}

u8 gen_swap(u8 *buf, u8 rd, u8 size, u8 endian, bool force, bool do_zext)
{
        u8 len = 0;
#ifdef __BIG_ENDIAN
        const u8 host_endian = BPF_FROM_BE;
#else
        const u8 host_endian = BPF_FROM_LE;
#endif
        if (host_endian != endian || force) {
                switch (size) {
                case 16:
                        /*
                         * r = B4B3_B2B1 << 16 --> r = B2B1_0000
                         * then, swape(r) would become the desired 0000_B1B2
                         */
                        len = arc_asli_r(buf, REG_LO(rd), REG_LO(rd), 16);
                        fallthrough;
                case 32:
                        len += arc_swape_r(BUF(buf, len), REG_LO(rd));
                        if (do_zext)
                                len += zext(BUF(buf, len), rd);
                        break;
                case 64:
                        /*
                         * swap "hi" and "lo":
                         *   hi ^= lo;
                         *   lo ^= hi;
                         *   hi ^= lo;
                         * and then swap the bytes in "hi" and "lo".
                         */
                        len  = arc_xor_r(buf, REG_HI(rd), REG_LO(rd));
                        len += arc_xor_r(BUF(buf, len), REG_LO(rd), REG_HI(rd));
                        len += arc_xor_r(BUF(buf, len), REG_HI(rd), REG_LO(rd));
                        len += arc_swape_r(BUF(buf, len), REG_LO(rd));
                        len += arc_swape_r(BUF(buf, len), REG_HI(rd));
                        break;
                default:
                        /* The caller must have handled this. */
                        break;
                }
        } else {
                /*
                 * If the same endianness, there's not much to do other
                 * than zeroing out the upper bytes based on the "size".
                 */
                switch (size) {
                case 16:
                        len = arc_and_i(buf, REG_LO(rd), 0xffff);
                        fallthrough;
                case 32:
                        if (do_zext)
                                len += zext(BUF(buf, len), rd);
                        break;
                case 64:
                        break;
                default:
                        /* The caller must have handled this. */
                        break;
                }
        }

        return len;
}

/*
 * To create a frame, all that is needed is:
 *
 *  push fp
 *  mov  fp, sp
 *  sub  sp, <frame_size>
 *
 * "push fp" is taken care of separately while saving the clobbered registers.
 * All that remains is copying SP value to FP and shrinking SP's address space
 * for any possible function call to come.
 */
static inline u8 frame_create(u8 *buf, u16 size)
{
        u8 len;

        len = arc_mov_r(buf, ARC_R_FP, ARC_R_SP);
        if (IN_U6_RANGE(size))
                len += arc_subi_r(BUF(buf, len), ARC_R_SP, size);
        else
                len += arc_sub_i(BUF(buf, len), ARC_R_SP, size);
        return len;
}

/*
 * mov sp, fp
 *
 * The value of SP upon entering was copied to FP.
 */
static inline u8 frame_restore(u8 *buf)
{
        return arc_mov_r(buf, ARC_R_SP, ARC_R_FP);
}

/*
 * Going from a JITed code to the native caller:
 *
 * mov ARC_ABI_RET_lo, BPF_REG_0_lo     # r0 <- r8
 * mov ARC_ABI_RET_hi, BPF_REG_0_hi     # r1 <- r9
 */
static u8 bpf_to_arc_return(u8 *buf)
{
        u8 len;

        len  = arc_mov_r(buf, ARC_R_0, REG_LO(BPF_REG_0));
        len += arc_mov_r(BUF(buf, len), ARC_R_1, REG_HI(BPF_REG_0));
        return len;
}

/*
 * Coming back from an external (in-kernel) function to the JITed code:
 *
 * mov ARC_ABI_RET_lo, BPF_REG_0_lo     # r8 <- r0
 * mov ARC_ABI_RET_hi, BPF_REG_0_hi     # r9 <- r1
 */
u8 arc_to_bpf_return(u8 *buf)
{
        u8 len;

        len  = arc_mov_r(buf, REG_LO(BPF_REG_0), ARC_R_0);
        len += arc_mov_r(BUF(buf, len), REG_HI(BPF_REG_0), ARC_R_1);
        return len;
}

/*
 * This translation leads to:
 *
 *   mov r10, addr                # always an 8-byte instruction
 *   jl  [r10]
 *
 * The length of the "mov" must be fixed (8), otherwise it may diverge
 * during the normal and extra passes:
 *
 *          normal pass                  extra pass
 *
 *   180:  mov     r10,0        |   180:  mov     r10,0x700578d8
 *   184:  jl      [r10]        |   188:  jl      [r10]
 *   188:  add.f   r16,r16,0x1  |   18c:  adc     r17,r17,0
 *   18c:  adc     r17,r17,0    |
 *
 * In the above example, the change from "r10 <- 0" to "r10 <- 0x700578d8"
 * has led to an increase in the length of the "mov" instruction.
 * Inadvertently, that caused the loss of the "add.f" instruction.
 */
static u8 jump_and_link(u8 *buf, u32 addr)
{
        u8 len;

        len  = arc_mov_i_fixed(buf, REG_LO(JIT_REG_TMP), addr);
        len += arc_jl(BUF(buf, len), REG_LO(JIT_REG_TMP));
        return len;
}

/*
 * This function determines which ARC registers must be saved and restored.
 * It does so by looking into:
 *
 * "bpf_reg": The clobbered (destination) BPF register
 * "is_call": Indicator if the current instruction is a call
 *
 * When a register of interest is clobbered, its corresponding bit position
 * in return value, "usage", is set to true.
 */
u32 mask_for_used_regs(u8 bpf_reg, bool is_call)
{
        u32 usage = 0;

        /* BPF registers that must be saved. */
        if (bpf_reg >= BPF_REG_6 && bpf_reg <= BPF_REG_9) {
                usage |= BIT(REG_LO(bpf_reg));
                usage |= BIT(REG_HI(bpf_reg));
        /*
         * Using the frame pointer register implies that it should
         * be saved and reinitialised with the current frame data.
         */
        } else if (bpf_reg == BPF_REG_FP) {
                usage |= BIT(REG_LO(BPF_REG_FP));
        /* Could there be some ARC registers that must to be saved? */
        } else {
                if (REG_LO(bpf_reg) >= ARC_CALLEE_SAVED_REG_FIRST &&
                    REG_LO(bpf_reg) <= ARC_CALLEE_SAVED_REG_LAST)
                        usage |= BIT(REG_LO(bpf_reg));

                if (REG_HI(bpf_reg) >= ARC_CALLEE_SAVED_REG_FIRST &&
                    REG_HI(bpf_reg) <= ARC_CALLEE_SAVED_REG_LAST)
                        usage |= BIT(REG_HI(bpf_reg));
        }

        /* A "call" indicates that ARC's "blink" reg must be saved. */
        usage |= is_call ? BIT(ARC_R_BLINK) : 0;

        return usage;
}

/*
 * push blink             # if blink is marked as clobbered
 * push r[0-n]            # if r[i] is marked as clobbered
 * push fp                # if fp is marked as clobbered
 * mov  fp, sp            # if frame_size > 0 (clobbers fp)
 * sub  sp, <frame_size>  # same as above
 */
u8 arc_prologue(u8 *buf, u32 usage, u16 frame_size)
{
        u8 len = 0;
        u32 gp_regs = 0;

        /* Deal with blink first. */
        if (usage & BIT(ARC_R_BLINK))
                len += arc_push_r(BUF(buf, len), ARC_R_BLINK);

        gp_regs = usage & ~(BIT(ARC_R_BLINK) | BIT(ARC_R_FP));
        while (gp_regs) {
                u8 reg = __builtin_ffs(gp_regs) - 1;

                len += arc_push_r(BUF(buf, len), reg);
                gp_regs &= ~BIT(reg);
        }

        /* Deal with fp last. */
        if ((usage & BIT(ARC_R_FP)) || frame_size > 0)
                len += arc_push_r(BUF(buf, len), ARC_R_FP);

        if (frame_size > 0)
                len += frame_create(BUF(buf, len), frame_size);

#ifdef ARC_BPF_JIT_DEBUG
        if ((usage & BIT(ARC_R_FP)) && frame_size == 0) {
                pr_err("FP is being saved while there is no frame.");
                BUG();
        }
#endif

        return len;
}

/*
 * mov  sp, fp            # if frame_size > 0
 * pop  fp                # if fp is marked as clobbered
 * pop  r[n-0]            # if r[i] is marked as clobbered
 * pop  blink             # if blink is marked as clobbered
 * mov  r0, r8            # always: ABI_return <- BPF_return
 * mov  r1, r9            # continuation of above
 * j    [blink]           # always
 *
 * "fp being marked as clobbered" and "frame_size > 0" are the two sides of
 * the same coin.
 */
u8 arc_epilogue(u8 *buf, u32 usage, u16 frame_size)
{
        u32 len = 0;
        u32 gp_regs = 0;

#ifdef ARC_BPF_JIT_DEBUG
        if ((usage & BIT(ARC_R_FP)) && frame_size == 0) {
                pr_err("FP is being saved while there is no frame.");
                BUG();
        }
#endif

        if (frame_size > 0)
                len += frame_restore(BUF(buf, len));

        /* Deal with fp first. */
        if ((usage & BIT(ARC_R_FP)) || frame_size > 0)
                len += arc_pop_r(BUF(buf, len), ARC_R_FP);

        gp_regs = usage & ~(BIT(ARC_R_BLINK) | BIT(ARC_R_FP));
        while (gp_regs) {
                /* "usage" is 32-bit, each bit indicating an ARC register. */
                u8 reg = 31 - __builtin_clz(gp_regs);

                len += arc_pop_r(BUF(buf, len), reg);
                gp_regs &= ~BIT(reg);
        }

        /* Deal with blink last. */
        if (usage & BIT(ARC_R_BLINK))
                len += arc_pop_r(BUF(buf, len), ARC_R_BLINK);

        /* Wrap up the return value and jump back to the caller. */
        len += bpf_to_arc_return(BUF(buf, len));
        len += arc_jmp_return(BUF(buf, len));

        return len;
}

/*
 * For details on the algorithm, see the comments of "gen_jcc_64()".
 *
 * This data structure is holding information for jump translations.
 *
 * jit_off: How many bytes into the current JIT address, "b"ranch insn. occurs
 * cond: The condition that the ARC branch instruction must use
 *
 * e.g.:
 *
 * BPF_JGE  R1, R0, @target
 * ------------------------
 *            |
 *            v
 * 0x1000: cmp  r3, r1     # 0x1000 is the JIT address for "BPF_JGE ..." insn
 * 0x1004: bhi  @target    # first jump (branch higher)
 * 0x1008: blo  @end       # second jump acting as a skip (end is 0x1014)
 * 0x100C: cmp  r2, r0     # the lower 32 bits are evaluated
 * 0x1010: bhs  @target    # third jump (branch higher or same)
 * 0x1014: ...
 *
 * The jit_off(set) of the "bhi" is 4 bytes.
 * The cond(ition) for the "bhi" is "CC_great_u".
 *
 * The jit_off(set) is necessary for calculating the exact displacement
 * to the "target" address:
 *
 * jit_address + jit_off(set) - @target
 * 0x1000      + 4            - @target
 */
#define JCC64_NR_OF_JMPS 3      /* Number of jumps in jcc64 template. */
#define JCC64_INSNS_TO_END 3    /* Number of insn. inclusive the 2nd jmp to end. */
#define JCC64_SKIP_JMP 1        /* Index of the "skip" jump to "end". */
static const struct {
        /*
         * "jit_off" is common between all "jmp[]" and is coupled with
         * "cond" of each "jmp[]" instance. e.g.:
         *
         * arcv2_64_jccs.jit_off[1]
         * arcv2_64_jccs.jmp[ARC_CC_UGT].cond[1]
         *
         * Are indicating that the second jump in JITed code of "UGT"
         * is at offset "jit_off[1]" while its condition is "cond[1]".
         */
        u8 jit_off[JCC64_NR_OF_JMPS];

        struct {
                u8 cond[JCC64_NR_OF_JMPS];
        } jmp[ARC_CC_SLE + 1];
} arcv2_64_jccs = {
        .jit_off = {
                INSN_len_normal * 1,
                INSN_len_normal * 2,
                INSN_len_normal * 4
        },
        /*
         *   cmp  rd_hi, rs_hi
         *   bhi  @target         # 1: u>
         *   blo  @end            # 2: u<
         *   cmp  rd_lo, rs_lo
         *   bhi  @target         # 3: u>
         * end:
         */
        .jmp[ARC_CC_UGT] = {
                .cond = {CC_great_u, CC_less_u, CC_great_u}
        },
        /*
         *   cmp  rd_hi, rs_hi
         *   bhi  @target         # 1: u>
         *   blo  @end            # 2: u<
         *   cmp  rd_lo, rs_lo
         *   bhs  @target         # 3: u>=
         * end:
         */
        .jmp[ARC_CC_UGE] = {
                .cond = {CC_great_u, CC_less_u, CC_great_eq_u}
        },
        /*
         *   cmp  rd_hi, rs_hi
         *   blo  @target         # 1: u<
         *   bhi  @end            # 2: u>
         *   cmp  rd_lo, rs_lo
         *   blo  @target         # 3: u<
         * end:
         */
        .jmp[ARC_CC_ULT] = {
                .cond = {CC_less_u, CC_great_u, CC_less_u}
        },
        /*
         *   cmp  rd_hi, rs_hi
         *   blo  @target         # 1: u<
         *   bhi  @end            # 2: u>
         *   cmp  rd_lo, rs_lo
         *   bls  @target         # 3: u<=
         * end:
         */
        .jmp[ARC_CC_ULE] = {
                .cond = {CC_less_u, CC_great_u, CC_less_eq_u}
        },
        /*
         *   cmp  rd_hi, rs_hi
         *   bgt  @target         # 1: s>
         *   blt  @end            # 2: s<
         *   cmp  rd_lo, rs_lo
         *   bhi  @target         # 3: u>
         * end:
         */
        .jmp[ARC_CC_SGT] = {
                .cond = {CC_great_s, CC_less_s, CC_great_u}
        },
        /*
         *   cmp  rd_hi, rs_hi
         *   bgt  @target         # 1: s>
         *   blt  @end            # 2: s<
         *   cmp  rd_lo, rs_lo
         *   bhs  @target         # 3: u>=
         * end:
         */
        .jmp[ARC_CC_SGE] = {
                .cond = {CC_great_s, CC_less_s, CC_great_eq_u}
        },
        /*
         *   cmp  rd_hi, rs_hi
         *   blt  @target         # 1: s<
         *   bgt  @end            # 2: s>
         *   cmp  rd_lo, rs_lo
         *   blo  @target         # 3: u<
         * end:
         */
        .jmp[ARC_CC_SLT] = {
                .cond = {CC_less_s, CC_great_s, CC_less_u}
        },
        /*
         *   cmp  rd_hi, rs_hi
         *   blt  @target         # 1: s<
         *   bgt  @end            # 2: s>
         *   cmp  rd_lo, rs_lo
         *   bls  @target         # 3: u<=
         * end:
         */
        .jmp[ARC_CC_SLE] = {
                .cond = {CC_less_s, CC_great_s, CC_less_eq_u}
        }
};

/*
 * The displacement (offset) for ARC's "b"ranch instruction is the distance
 * from the aligned version of _current_ instruction (PCL) to the target
 * instruction:
 *
 * DISP = TARGET - PCL          # PCL is the word aligned PC
 */
static inline s32 get_displacement(u32 curr_off, u32 targ_off)
{
        return (s32)(targ_off - (curr_off & ~3L));
}

/*
 * "disp"lacement should be:
 *
 * 1. 16-bit aligned.
 * 2. fit in S25, because no "condition code" is supposed to be encoded.
 */
static inline bool is_valid_far_disp(s32 disp)
{
        return (!(disp & 1) && IN_S25_RANGE(disp));
}

/*
 * "disp"lacement should be:
 *
 * 1. 16-bit aligned.
 * 2. fit in S21, because "condition code" is supposed to be encoded too.
 */
static inline bool is_valid_near_disp(s32 disp)
{
        return (!(disp & 1) && IN_S21_RANGE(disp));
}

/*
 * cmp        rd_hi, rs_hi
 * cmp.z      rd_lo, rs_lo
 * b{eq,ne}   @target
 *   |  |
 *   |  `-->  "eq" param is false (JNE)
 *   `----->  "eq" param is true  (JEQ)
 */
static int gen_j_eq_64(u8 *buf, u8 rd, u8 rs, bool eq,
                       u32 curr_off, u32 targ_off)
{
        s32 disp;
        u8 len = 0;

        len += arc_cmp_r(BUF(buf, len), REG_HI(rd), REG_HI(rs));
        len += arc_cmpz_r(BUF(buf, len), REG_LO(rd), REG_LO(rs));
        disp = get_displacement(curr_off + len, targ_off);
        len += arc_bcc(BUF(buf, len), eq ? CC_equal : CC_unequal, disp);

        return len;
}

/*
 * tst   rd_hi, rs_hi
 * tst.z rd_lo, rs_lo
 * bne   @target
 */
static u8 gen_jset_64(u8 *buf, u8 rd, u8 rs, u32 curr_off, u32 targ_off)
{
        u8 len = 0;
        s32 disp;

        len += arc_tst_r(BUF(buf, len), REG_HI(rd), REG_HI(rs));
        len += arc_tstz_r(BUF(buf, len), REG_LO(rd), REG_LO(rs));
        disp = get_displacement(curr_off + len, targ_off);
        len += arc_bcc(BUF(buf, len), CC_unequal, disp);

        return len;
}

/*
 * Verify if all the jumps for a JITed jcc64 operation are valid,
 * by consulting the data stored at "arcv2_64_jccs".
 */
static bool check_jcc_64(u32 curr_off, u32 targ_off, u8 cond)
{
        size_t i;

        if (cond >= ARC_CC_LAST)
                return false;

        for (i = 0; i < JCC64_NR_OF_JMPS; i++) {
                u32 from, to;

                from = curr_off + arcv2_64_jccs.jit_off[i];
                /* for the 2nd jump, we jump to the end of block. */
                if (i != JCC64_SKIP_JMP)
                        to = targ_off;
                else
                        to = from + (JCC64_INSNS_TO_END * INSN_len_normal);
                /* There is a "cc" in the instruction, so a "near" jump. */
                if (!is_valid_near_disp(get_displacement(from, to)))
                        return false;
        }

        return true;
}

/* Can the jump from "curr_off" to "targ_off" actually happen? */
bool check_jmp_64(u32 curr_off, u32 targ_off, u8 cond)
{
        s32 disp;

        switch (cond) {
        case ARC_CC_UGT:
        case ARC_CC_UGE:
        case ARC_CC_ULT:
        case ARC_CC_ULE:
        case ARC_CC_SGT:
        case ARC_CC_SGE:
        case ARC_CC_SLT:
        case ARC_CC_SLE:
                return check_jcc_64(curr_off, targ_off, cond);
        case ARC_CC_EQ:
        case ARC_CC_NE:
        case ARC_CC_SET:
                /*
                 * The "jump" for the JITed BPF_J{SET,EQ,NE} is actually the
                 * 3rd instruction. See comments of "gen_j{set,_eq}_64()".
                 */
                curr_off += 2 * INSN_len_normal;
                disp = get_displacement(curr_off, targ_off);
                /* There is a "cc" field in the issued instruction. */
                return is_valid_near_disp(disp);
        case ARC_CC_AL:
                disp = get_displacement(curr_off, targ_off);
                return is_valid_far_disp(disp);
        default:
                return false;
        }
}

/*
 * The template for the 64-bit jumps with the following BPF conditions
 *
 * u< u<= u> u>= s< s<= s> s>=
 *
 * Looks like below:
 *
 *   cmp   rd_hi, rs_hi
 *   b<c1> @target
 *   b<c2> @end
 *   cmp   rd_lo, rs_lo   # if execution reaches here, r{d,s}_hi are equal
 *   b<c3> @target
 * end:
 *
 * "c1" is the condition that JIT is handling minus the equality part.
 * For instance if we have to translate an "unsigned greater or equal",
 * then "c1" will be "unsigned greater". We won't know about equality
 * until all 64-bits of data (higeher and lower registers) are processed.
 *
 * "c2" is the counter logic of "c1". For instance, if "c1" is originated
 * from "s>", then "c2" would be "s<". Notice that equality doesn't play
 * a role here either, because the lower 32 bits are not processed yet.
 *
 * "c3" is the unsigned version of "c1", no matter if the BPF condition
 * was signed or unsigned. An unsigned version is necessary, because the
 * MSB of the lower 32 bits does not reflect a sign in the whole 64-bit
 * scheme. Otherwise, 64-bit comparisons like
 * (0x0000_0000,0x8000_0000) s>= (0x0000_0000,0x0000_0000)
 * would yield an incorrect result. Finally, if there is an equality
 * check in the BPF condition, it will be reflected in "c3".
 *
 * You can find all the instances of this template where the
 * "arcv2_64_jccs" is getting initialised.
 */
static u8 gen_jcc_64(u8 *buf, u8 rd, u8 rs, u8 cond,
                     u32 curr_off, u32 targ_off)
{
        s32 disp;
        u32 end_off;
        const u8 *cc = arcv2_64_jccs.jmp[cond].cond;
        u8 len = 0;

        /* cmp rd_hi, rs_hi */
        len += arc_cmp_r(buf, REG_HI(rd), REG_HI(rs));

        /* b<c1> @target */
        disp = get_displacement(curr_off + len, targ_off);
        len += arc_bcc(BUF(buf, len), cc[0], disp);

        /* b<c2> @end */
        end_off = curr_off + len + (JCC64_INSNS_TO_END * INSN_len_normal);
        disp = get_displacement(curr_off + len, end_off);
        len += arc_bcc(BUF(buf, len), cc[1], disp);

        /* cmp rd_lo, rs_lo */
        len += arc_cmp_r(BUF(buf, len), REG_LO(rd), REG_LO(rs));

        /* b<c3> @target */
        disp = get_displacement(curr_off + len, targ_off);
        len += arc_bcc(BUF(buf, len), cc[2], disp);

        return len;
}

/*
 * This function only applies the necessary logic to make the proper
 * translations. All the sanity checks must have already been done
 * by calling the check_jmp_64().
 */
u8 gen_jmp_64(u8 *buf, u8 rd, u8 rs, u8 cond, u32 curr_off, u32 targ_off)
{
        u8 len = 0;
        bool eq = false;
        s32 disp;

        switch (cond) {
        case ARC_CC_AL:
                disp = get_displacement(curr_off, targ_off);
                len = arc_b(buf, disp);
                break;
        case ARC_CC_UGT:
        case ARC_CC_UGE:
        case ARC_CC_ULT:
        case ARC_CC_ULE:
        case ARC_CC_SGT:
        case ARC_CC_SGE:
        case ARC_CC_SLT:
        case ARC_CC_SLE:
                len = gen_jcc_64(buf, rd, rs, cond, curr_off, targ_off);
                break;
        case ARC_CC_EQ:
                eq = true;
                fallthrough;
        case ARC_CC_NE:
                len = gen_j_eq_64(buf, rd, rs, eq, curr_off, targ_off);
                break;
        case ARC_CC_SET:
                len = gen_jset_64(buf, rd, rs, curr_off, targ_off);
                break;
        default:
#ifdef ARC_BPF_JIT_DEBUG
                pr_err("64-bit jump condition is not known.");
                BUG();
#endif
        }
        return len;
}

/*
 * The condition codes to use when generating JIT instructions
 * for 32-bit jumps.
 *
 * The "ARC_CC_AL" index is not really used by the code, but it
 * is here for the sake of completeness.
 *
 * The "ARC_CC_SET" becomes "CC_unequal" because of the "tst"
 * instruction that precedes the conditional branch.
 */
static const u8 arcv2_32_jmps[ARC_CC_LAST] = {
        [ARC_CC_UGT] = CC_great_u,
        [ARC_CC_UGE] = CC_great_eq_u,
        [ARC_CC_ULT] = CC_less_u,
        [ARC_CC_ULE] = CC_less_eq_u,
        [ARC_CC_SGT] = CC_great_s,
        [ARC_CC_SGE] = CC_great_eq_s,
        [ARC_CC_SLT] = CC_less_s,
        [ARC_CC_SLE] = CC_less_eq_s,
        [ARC_CC_AL]  = CC_always,
        [ARC_CC_EQ]  = CC_equal,
        [ARC_CC_NE]  = CC_unequal,
        [ARC_CC_SET] = CC_unequal
};

/* Can the jump from "curr_off" to "targ_off" actually happen? */
bool check_jmp_32(u32 curr_off, u32 targ_off, u8 cond)
{
        u8 addendum;
        s32 disp;

        if (cond >= ARC_CC_LAST)
                return false;

        /*
         * The unconditional jump happens immediately, while the rest
         * are either preceded by a "cmp" or "tst" instruction.
         */
        addendum = (cond == ARC_CC_AL) ? 0 : INSN_len_normal;
        disp = get_displacement(curr_off + addendum, targ_off);

        if (cond == ARC_CC_AL)
                return is_valid_far_disp(disp);
        else
                return is_valid_near_disp(disp);
}

/*
 * The JITed code for 32-bit (conditional) branches:
 *
 * ARC_CC_AL  @target
 *   b @jit_targ_addr
 *
 * ARC_CC_SET rd, rs, @target
 *   tst rd, rs
 *   bnz @jit_targ_addr
 *
 * ARC_CC_xx  rd, rs, @target
 *   cmp rd, rs
 *   b<cc> @jit_targ_addr            # cc = arcv2_32_jmps[xx]
 */
u8 gen_jmp_32(u8 *buf, u8 rd, u8 rs, u8 cond, u32 curr_off, u32 targ_off)
{
        s32 disp;
        u8 len = 0;

        /*
         * Although this must have already been checked by "check_jmp_32()",
         * we're not going to risk accessing "arcv2_32_jmps" array without
         * the boundary check.
         */
        if (cond >= ARC_CC_LAST) {
#ifdef ARC_BPF_JIT_DEBUG
                pr_err("32-bit jump condition is not known.");
                BUG();
#endif
                return 0;
        }

        /* If there is a "condition", issue the "cmp" or "tst" first. */
        if (cond != ARC_CC_AL) {
                if (cond == ARC_CC_SET)
                        len = tst_r32(buf, rd, rs);
                else
                        len = cmp_r32(buf, rd, rs);
                /*
                 * The issued instruction affects the "disp"lacement as
                 * it alters the "curr_off" by its "len"gth. The "curr_off"
                 * should always point to the jump instruction.
                 */
                disp = get_displacement(curr_off + len, targ_off);
                len += arc_bcc(BUF(buf, len), arcv2_32_jmps[cond], disp);
        } else {
                /* The straight forward unconditional jump. */
                disp = get_displacement(curr_off, targ_off);
                len = arc_b(buf, disp);
        }

        return len;
}

/*
 * Generate code for functions calls. There can be two types of calls:
 *
 * - Calling another BPF function
 * - Calling an in-kernel function which is compiled by ARC gcc
 *
 * In the later case, we must comply to ARCv2 ABI and handle arguments
 * and return values accordingly.
 */
u8 gen_func_call(u8 *buf, ARC_ADDR func_addr, bool external_func)
{
        u8 len = 0;

        /*
         * In case of an in-kernel function call, always push the 5th
         * argument onto the stack, because that's where the ABI dictates
         * it should be found. If the callee doesn't really use it, no harm
         * is done. The stack is readjusted either way after the call.
         */
        if (external_func)
                len += push_r64(BUF(buf, len), BPF_REG_5);

        len += jump_and_link(BUF(buf, len), func_addr);

        if (external_func)
                len += arc_add_i(BUF(buf, len), ARC_R_SP, ARC_R_SP, ARG5_SIZE);

        return len;
}