DragonOS-Community/DragonOS
1
1
Fork 0
mirror of https://github.com/DragonOS-Community/DragonOS.git synced 2025-06-09 11:16:47 +00:00
Code Issues Releases Wiki Activity
DragonOS/kernel/crates/rbpf/tests/ubpf_jit_x86_64.rs
linfeng fae6e9ade4
feat(ebpf):[WIP] add eBPF support (#948) 
* feat(kprobe): Add basic kprobe support for x86_64

* feat: add ebpf support (#912)

- 实现bpf()一部分命令，包括几种基本map，相关的helper函数
- 实现部分perf相关的数据结构
- 暂时为文件实现简单mmap
- 实现一个使用kprobe统计syscall 调用次数的ebpf程序

对eBPF支持程度(基本)：

- 简单的eBPF程序(没有指定特殊的Map)
- 使用内核已经实现的Map的eBPF程序
- 可以和kprobe配合使用
- 内核Map相关的接口定义已经实现，添加新的Map较为简单

不支持的功能:
- 区分不同的eBPF程序类型(Network/Cgroup)并限定可调用的helper函数集
- 与内核其它跟踪机制配合(tracepoint)
- 其它helper和Map


todo

- [ ]  修改mmap，需要讨论，因为这个和块缓存层相关
- [x]  添加文档
- [x]  修复可能的错误
- [x] 增加rbpf版本信息

* feat: add /sys/devices/system/cpu/possible file

* feat: add /sys/devices/system/cpu/online
2024-10-25 15:59:57 +08:00

2892 lines
64 KiB
Rust
Raw Permalink Blame History

// SPDX-License-Identifier: (Apache-2.0 OR MIT)
// Converted from the tests for uBPF <https://github.com/iovisor/ubpf>
// Copyright 2015 Big Switch Networks, Inc
// Copyright 2016 6WIND S.A. <quentin.monnet@6wind.com>
// The tests contained in this file are extracted from the unit tests of uBPF software. Each test
// in this file has a name in the form `test_jit_<name>`, and corresponds to the (human-readable)
// code in `ubpf/tree/master/tests/<name>`, available at
// <https://github.com/iovisor/ubpf/tree/master/tests> (hyphen had to be replaced with underscores
// as Rust will not accept them in function names). It is strongly advised to refer to the uBPF
// version to understand what these program do.
//
// Each program was assembled from the uBPF version with the assembler provided by uBPF itself, and
// available at <https://github.com/iovisor/ubpf/tree/master/ubpf>.
// The very few modifications that have been realized should be indicated.
// These are unit tests for the eBPF JIT compiler.
#![allow(clippy::unreadable_literal)]
#![cfg(all(not(windows), feature = "std"))]
extern crate rbpf;
mod common;
use common::{TCP_SACK_ASM, TCP_SACK_MATCH, TCP_SACK_NOMATCH};
use rbpf::{assembler::assemble, helpers};
#[test]
fn test_jit_add() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 2
add32 r0, 1
add32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x3);
}
}
#[test]
fn test_jit_alu64_arith() {
let prog = assemble(
"
mov r0, 0
mov r1, 1
mov r2, 2
mov r3, 3
mov r4, 4
mov r5, 5
mov r6, 6
mov r7, 7
mov r8, 8
mov r9, 9
add r0, 23
add r0, r7
sub r0, 13
sub r0, r1
mul r0, 7
mul r0, r3
div r0, 2
div r0, r4
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x2a);
}
}
#[test]
fn test_jit_alu64_bit() {
let prog = assemble(
"
mov r0, 0
mov r1, 1
mov r2, 2
mov r3, 3
mov r4, 4
mov r5, 5
mov r6, 6
mov r7, 7
mov r8, 8
or r0, r5
or r0, 0xa0
and r0, 0xa3
mov r9, 0x91
and r0, r9
lsh r0, 32
lsh r0, 22
lsh r0, r8
rsh r0, 32
rsh r0, 19
rsh r0, r7
xor r0, 0x03
xor r0, r2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x11);
}
}
#[test]
fn test_jit_alu_arith() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 1
mov32 r2, 2
mov32 r3, 3
mov32 r4, 4
mov32 r5, 5
mov32 r6, 6
mov32 r7, 7
mov32 r8, 8
mov32 r9, 9
add32 r0, 23
add32 r0, r7
sub32 r0, 13
sub32 r0, r1
mul32 r0, 7
mul32 r0, r3
div32 r0, 2
div32 r0, r4
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x2a);
}
}
#[test]
fn test_jit_alu_bit() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 1
mov32 r2, 2
mov32 r3, 3
mov32 r4, 4
mov32 r5, 5
mov32 r6, 6
mov32 r7, 7
mov32 r8, 8
or32 r0, r5
or32 r0, 0xa0
and32 r0, 0xa3
mov32 r9, 0x91
and32 r0, r9
lsh32 r0, 22
lsh32 r0, r8
rsh32 r0, 19
rsh32 r0, r7
xor32 r0, 0x03
xor32 r0, r2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x11);
}
}
#[test]
fn test_jit_arsh32_high_shift() {
let prog = assemble(
"
mov r0, 8
lddw r1, 0x100000001
arsh32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x4);
}
}
#[test]
fn test_jit_arsh() {
let prog = assemble(
"
mov32 r0, 0xf8
lsh32 r0, 28
arsh32 r0, 16
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0xffff8000);
}
}
#[test]
fn test_jit_arsh64() {
let prog = assemble(
"
mov32 r0, 1
lsh r0, 63
arsh r0, 55
mov32 r1, 5
arsh r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0xfffffffffffffff8);
}
}
#[test]
fn test_jit_arsh_reg() {
let prog = assemble(
"
mov32 r0, 0xf8
mov32 r1, 16
lsh32 r0, 28
arsh32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0xffff8000);
}
}
#[test]
fn test_jit_be16() {
let prog = assemble(
"
ldxh r0, [r1]
be16 r0
exit",
)
.unwrap();
let mem = &mut [0x11, 0x22];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x1122);
}
}
#[test]
fn test_jit_be16_high() {
let prog = assemble(
"
ldxdw r0, [r1]
be16 r0
exit",
)
.unwrap();
let mem = &mut [0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x1122);
}
}
#[test]
fn test_jit_be32() {
let prog = assemble(
"
ldxw r0, [r1]
be32 r0
exit",
)
.unwrap();
let mem = &mut [0x11, 0x22, 0x33, 0x44];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x11223344);
}
}
#[test]
fn test_jit_be32_high() {
let prog = assemble(
"
ldxdw r0, [r1]
be32 r0
exit",
)
.unwrap();
let mem = &mut [0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x11223344);
}
}
#[test]
fn test_jit_be64() {
let prog = assemble(
"
ldxdw r0, [r1]
be64 r0
exit",
)
.unwrap();
let mem = &mut [0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x1122334455667788);
}
}
#[test]
fn test_jit_call() {
let prog = assemble(
"
mov r1, 1
mov r2, 2
mov r3, 3
mov r4, 4
mov r5, 5
call 0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.register_helper(0, helpers::gather_bytes).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x0102030405);
}
}
#[test]
fn test_jit_call_memfrob() {
let prog = assemble(
"
mov r6, r1
add r1, 2
mov r2, 4
call 1
ldxdw r0, [r6]
be64 r0
exit",
)
.unwrap();
let mem = &mut [0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.register_helper(1, helpers::memfrob).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x102292e2f2c0708);
}
}
// TODO: helpers::trash_registers needs asm!().
// Try this again once asm!() is available in stable.
//#[test]
//fn test_jit_call_save() {
//let prog = &[
//0xb7, 0x06, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
//0xb7, 0x07, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00,
//0xb7, 0x08, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00,
//0xb7, 0x09, 0x00, 0x00, 0x00, 0x40, 0x00, 0x00,
//0x85, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00,
//0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
//0x4f, 0x60, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
//0x4f, 0x70, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
//0x4f, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
//0x4f, 0x90, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
//0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
//];
//let mut vm = rbpf::EbpfVmNoData::new(Some(prog)).unwrap();
//vm.register_helper(2, helpers::trash_registers);
//vm.jit_compile().unwrap();
//unsafe { assert_eq!(vm.execute_program_jit().unwrap(), 0x4321); }
//}
#[test]
fn test_jit_div32_high_divisor() {
let prog = assemble(
"
mov r0, 12
lddw r1, 0x100000004
div32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x3);
}
}
#[test]
fn test_jit_div32_imm() {
let prog = assemble(
"
lddw r0, 0x10000000c
div32 r0, 4
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x3);
}
}
#[test]
fn test_jit_div32_reg() {
let prog = assemble(
"
lddw r0, 0x10000000c
mov r1, 4
div32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x3);
}
}
#[test]
fn test_jit_div64_imm() {
let prog = assemble(
"
mov r0, 0xc
lsh r0, 32
div r0, 4
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x300000000);
}
}
#[test]
fn test_jit_div64_reg() {
let prog = assemble(
"
mov r0, 0xc
lsh r0, 32
mov r1, 4
div r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x300000000);
}
}
// For some register numbers, we don't emit the same instructions for handling divisions by zero,
// which means we don't use the same offset to skip these instructions when the divisor is not
// zero. We've had a regression because of this before, make sure we test it.
#[test]
fn test_jit_div32_highreg() {
let prog = assemble(
"
mov r0, 2
mov r7, 4
div32 r7, r0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x2);
}
}
#[test]
fn test_jit_div64_highreg() {
let prog = assemble(
"
mov r0, 2
mov r7, 4
div r7, r0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x2);
}
}
#[test]
fn test_jit_early_exit() {
let prog = assemble(
"
mov r0, 3
exit
mov r0, 4
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x3);
}
}
// uBPF limits the number of user functions at 64. We don't.
//#[test]
//fn test_jit_err_call_bad_imm() {
//}
#[test]
#[should_panic(expected = "[JIT] Error: unknown helper function (id: 0x3f)")]
fn test_jit_err_call_unreg() {
let prog = assemble(
"
mov r1, 1
mov r2, 2
mov r3, 3
mov r4, 4
mov r5, 5
call 63
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
vm.execute_program_jit().unwrap();
}
}
#[test]
fn test_jit_div64_by_zero_imm() {
let prog = assemble(
"
mov32 r0, 1
div r0, 0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x0);
}
}
#[test]
fn test_jit_div_by_zero_imm() {
let prog = assemble(
"
mov32 r0, 1
div32 r0, 0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x0);
}
}
#[test]
fn test_jit_mod64_by_zero_imm() {
let prog = assemble(
"
mov32 r0, 1
mod r0, 0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_mod_by_zero_imm() {
let prog = assemble(
"
mov32 r0, 1
mod32 r0, 0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_div64_by_zero_reg() {
let prog = assemble(
"
mov32 r0, 1
mov32 r1, 0
div r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x0);
}
}
#[test]
fn test_jit_div_by_zero_reg() {
let prog = assemble(
"
mov32 r0, 1
mov32 r1, 0
div32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x0);
}
}
#[test]
fn test_jit_mod64_by_zero_reg() {
let prog = assemble(
"
mov32 r0, 1
mov32 r1, 0
mod r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_mod_by_zero_reg() {
let prog = assemble(
"
mov32 r0, 1
mov32 r1, 0
mod32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
// TODO SKIP: JIT disabled for this testcase (stack oob check not implemented)
// #[test]
// #[should_panic(expected = "Error: out of bounds memory store (insn #1)")]
// fn test_jit_err_stack_out_of_bound() {
// let prog = &[
// 0x72, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
// 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
// ];
// let mut vm = rbpf::EbpfVmNoData::new(Some(prog)).unwrap();
// vm.jit_compile().unwrap();
// unsafe { vm.execute_program_jit().unwrap(); }
// }
#[test]
fn test_jit_exit() {
let prog = assemble(
"
mov r0, 0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x0);
}
}
#[test]
fn test_jit_ja() {
let prog = assemble(
"
mov r0, 1
ja +1
mov r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jeq_imm() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 0xa
jeq r1, 0xb, +4
mov32 r0, 1
mov32 r1, 0xb
jeq r1, 0xb, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jeq_reg() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 0xa
mov32 r2, 0xb
jeq r1, r2, +4
mov32 r0, 1
mov32 r1, 0xb
jeq r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jge_imm() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 0xa
jge r1, 0xb, +4
mov32 r0, 1
mov32 r1, 0xc
jge r1, 0xb, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jle_imm() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 5
jle r1, 4, +1
jle r1, 6, +1
exit
jle r1, 5, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jle_reg() {
let prog = assemble(
"
mov r0, 0
mov r1, 5
mov r2, 4
mov r3, 6
jle r1, r2, +2
jle r1, r1, +1
exit
jle r1, r3, +1
exit
mov r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jgt_imm() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 5
jgt r1, 6, +2
jgt r1, 5, +1
jgt r1, 4, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jgt_reg() {
let prog = assemble(
"
mov r0, 0
mov r1, 5
mov r2, 6
mov r3, 4
jgt r1, r2, +2
jgt r1, r1, +1
jgt r1, r3, +1
exit
mov r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jlt_imm() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 5
jlt r1, 4, +2
jlt r1, 5, +1
jlt r1, 6, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jlt_reg() {
let prog = assemble(
"
mov r0, 0
mov r1, 5
mov r2, 4
mov r3, 6
jlt r1, r2, +2
jlt r1, r1, +1
jlt r1, r3, +1
exit
mov r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jit_bounce() {
let prog = assemble(
"
mov r0, 1
mov r6, r0
mov r7, r6
mov r8, r7
mov r9, r8
mov r0, r9
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jne_reg() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 0xb
mov32 r2, 0xb
jne r1, r2, +4
mov32 r0, 1
mov32 r1, 0xa
jne r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jset_imm() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 0x7
jset r1, 0x8, +4
mov32 r0, 1
mov32 r1, 0x9
jset r1, 0x8, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jset_reg() {
let prog = assemble(
"
mov32 r0, 0
mov32 r1, 0x7
mov32 r2, 0x8
jset r1, r2, +4
mov32 r0, 1
mov32 r1, 0x9
jset r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsge_imm() {
let prog = assemble(
"
mov32 r0, 0
mov r1, -2
jsge r1, -1, +5
jsge r1, 0, +4
mov32 r0, 1
mov r1, -1
jsge r1, -1, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsge_reg() {
let prog = assemble(
"
mov32 r0, 0
mov r1, -2
mov r2, -1
mov32 r3, 0
jsge r1, r2, +5
jsge r1, r3, +4
mov32 r0, 1
mov r1, r2
jsge r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsle_imm() {
let prog = assemble(
"
mov32 r0, 0
mov r1, -2
jsle r1, -3, +1
jsle r1, -1, +1
exit
mov32 r0, 1
jsle r1, -2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsle_reg() {
let prog = assemble(
"
mov32 r0, 0
mov r1, -1
mov r2, -2
mov32 r3, 0
jsle r1, r2, +1
jsle r1, r3, +1
exit
mov32 r0, 1
mov r1, r2
jsle r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsgt_imm() {
let prog = assemble(
"
mov32 r0, 0
mov r1, -2
jsgt r1, -1, +4
mov32 r0, 1
mov32 r1, 0
jsgt r1, -1, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsgt_reg() {
let prog = assemble(
"
mov32 r0, 0
mov r1, -2
mov r2, -1
jsgt r1, r2, +4
mov32 r0, 1
mov32 r1, 0
jsgt r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jslt_imm() {
let prog = assemble(
"
mov32 r0, 0
mov r1, -2
jslt r1, -3, +2
jslt r1, -2, +1
jslt r1, -1, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jslt_reg() {
let prog = assemble(
"
mov32 r0, 0
mov r1, -2
mov r2, -3
mov r3, -1
jslt r1, r1, +2
jslt r1, r2, +1
jslt r1, r3, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jeq32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0x0
mov32 r1, 0xa
jeq32 r1, 0xb, +5
mov32 r0, 1
mov r1, 0xb
or r1, r9
jeq32 r1, 0xb, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jeq32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 0xa
mov32 r2, 0xb
jeq32 r1, r2, +5
mov32 r0, 1
mov32 r1, 0xb
or r1, r9
jeq32 r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jge32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 0xa
jge32 r1, 0xb, +5
mov32 r0, 1
or r1, r9
mov32 r1, 0xc
jge32 r1, 0xb, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jge32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 0xa
mov32 r2, 0xb
jge32 r1, r2, +5
mov32 r0, 1
or r1, r9
mov32 r1, 0xc
jge32 r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jgt32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 5
or r1, r9
jgt32 r1, 6, +4
jgt32 r1, 5, +3
jgt32 r1, 4, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jgt32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov r0, 0
mov r1, 5
mov32 r1, 5
or r1, r9
mov r2, 6
mov r3, 4
jgt32 r1, r2, +4
jgt32 r1, r1, +3
jgt32 r1, r3, +1
exit
mov r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jle32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 5
or r1, r9
jle32 r1, 4, +5
jle32 r1, 6, +1
exit
jle32 r1, 5, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jle32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov r0, 0
mov r1, 5
mov r2, 4
mov r3, 6
or r1, r9
jle32 r1, r2, +5
jle32 r1, r1, +1
exit
jle32 r1, r3, +1
exit
mov r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jlt32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 5
or r1, r9
jlt32 r1, 4, +4
jlt32 r1, 5, +3
jlt32 r1, 6, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jlt32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov r0, 0
mov r1, 5
mov r2, 4
mov r3, 6
or r1, r9
jlt32 r1, r2, +4
jlt32 r1, r1, +3
jlt32 r1, r3, +1
exit
mov r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jne32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 0xb
or r1, r9
jne32 r1, 0xb, +4
mov32 r0, 1
mov32 r1, 0xa
or r1, r9
jne32 r1, 0xb, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jne32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 0xb
or r1, r9
mov32 r2, 0xb
jne32 r1, r2, +4
mov32 r0, 1
mov32 r1, 0xa
or r1, r9
jne32 r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jset32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 0x7
or r1, r9
jset32 r1, 0x8, +4
mov32 r0, 1
mov32 r1, 0x9
jset32 r1, 0x8, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jset32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, 0x7
or r1, r9
mov32 r2, 0x8
jset32 r1, r2, +4
mov32 r0, 1
mov32 r1, 0x9
jset32 r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsge32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, -2
or r1, r9
jsge32 r1, -1, +5
jsge32 r1, 0, +4
mov32 r0, 1
mov r1, -1
jsge32 r1, -1, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsge32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, -2
or r1, r9
mov r2, -1
mov32 r3, 0
jsge32 r1, r2, +5
jsge32 r1, r3, +4
mov32 r0, 1
mov r1, r2
jsge32 r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsgt32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, -2
or r1, r9
jsgt32 r1, -1, +4
mov32 r0, 1
mov32 r1, 0
jsgt32 r1, -1, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsgt32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, -2
or r1, r9
mov r2, -1
jsgt32 r1, r2, +4
mov32 r0, 1
mov32 r1, 0
jsgt32 r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsle32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, -2
or r1, r9
jsle32 r1, -3, +5
jsle32 r1, -1, +1
exit
mov32 r0, 1
jsle32 r1, -2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jsle32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, -2
or r1, r9
mov r2, -3
mov32 r3, 0
jsle32 r1, r2, +6
jsle32 r1, r3, +1
exit
mov32 r0, 1
mov r1, r2
jsle32 r1, r2, +1
mov32 r0, 2
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jslt32_imm() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, -2
or r1, r9
jslt32 r1, -3, +4
jslt32 r1, -2, +3
jslt32 r1, -1, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_jslt32_reg() {
let prog = assemble(
"
mov r9, 1
lsh r9, 32
mov32 r0, 0
mov32 r1, -2
or r1, r9
mov r2, -3
mov r3, -1
jslt32 r1, r1, +4
jslt32 r1, r2, +3
jslt32 r1, r3, +1
exit
mov32 r0, 1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_lddw() {
let prog = assemble(
"
lddw r0, 0x1122334455667788
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1122334455667788);
}
}
#[test]
fn test_jit_lddw2() {
let prog = assemble(
"
lddw r0, 0x0000000080000000
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x80000000);
}
}
#[test]
fn test_jit_ldxb_all() {
let prog = assemble(
"
mov r0, r1
ldxb r9, [r0+0]
lsh r9, 0
ldxb r8, [r0+1]
lsh r8, 4
ldxb r7, [r0+2]
lsh r7, 8
ldxb r6, [r0+3]
lsh r6, 12
ldxb r5, [r0+4]
lsh r5, 16
ldxb r4, [r0+5]
lsh r4, 20
ldxb r3, [r0+6]
lsh r3, 24
ldxb r2, [r0+7]
lsh r2, 28
ldxb r1, [r0+8]
lsh r1, 32
ldxb r0, [r0+9]
lsh r0, 36
or r0, r1
or r0, r2
or r0, r3
or r0, r4
or r0, r5
or r0, r6
or r0, r7
or r0, r8
or r0, r9
exit",
)
.unwrap();
let mem = &mut [0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x9876543210);
}
}
#[test]
fn test_jit_ldxb() {
let prog = assemble(
"
ldxb r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [0xaa, 0xbb, 0x11, 0xcc, 0xdd];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x11);
}
}
#[test]
fn test_jit_ldxdw() {
let prog = assemble(
"
ldxdw r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [
0xaa, 0xbb, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0xcc, 0xdd,
];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x8877665544332211);
}
}
#[test]
fn test_jit_ldxh_all() {
let prog = assemble(
"
mov r0, r1
ldxh r9, [r0+0]
be16 r9
lsh r9, 0
ldxh r8, [r0+2]
be16 r8
lsh r8, 4
ldxh r7, [r0+4]
be16 r7
lsh r7, 8
ldxh r6, [r0+6]
be16 r6
lsh r6, 12
ldxh r5, [r0+8]
be16 r5
lsh r5, 16
ldxh r4, [r0+10]
be16 r4
lsh r4, 20
ldxh r3, [r0+12]
be16 r3
lsh r3, 24
ldxh r2, [r0+14]
be16 r2
lsh r2, 28
ldxh r1, [r0+16]
be16 r1
lsh r1, 32
ldxh r0, [r0+18]
be16 r0
lsh r0, 36
or r0, r1
or r0, r2
or r0, r3
or r0, r4
or r0, r5
or r0, r6
or r0, r7
or r0, r8
or r0, r9
exit",
)
.unwrap();
let mem = &mut [
0x00, 0x00, 0x00, 0x01, 0x00, 0x02, 0x00, 0x03, 0x00, 0x04, 0x00, 0x05, 0x00, 0x06, 0x00,
0x07, 0x00, 0x08, 0x00, 0x09,
];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x9876543210);
}
}
#[test]
fn test_jit_ldxh_all2() {
let prog = assemble(
"
mov r0, r1
ldxh r9, [r0+0]
be16 r9
ldxh r8, [r0+2]
be16 r8
ldxh r7, [r0+4]
be16 r7
ldxh r6, [r0+6]
be16 r6
ldxh r5, [r0+8]
be16 r5
ldxh r4, [r0+10]
be16 r4
ldxh r3, [r0+12]
be16 r3
ldxh r2, [r0+14]
be16 r2
ldxh r1, [r0+16]
be16 r1
ldxh r0, [r0+18]
be16 r0
or r0, r1
or r0, r2
or r0, r3
or r0, r4
or r0, r5
or r0, r6
or r0, r7
or r0, r8
or r0, r9
exit",
)
.unwrap();
let mem = &mut [
0x00, 0x01, 0x00, 0x02, 0x00, 0x04, 0x00, 0x08, 0x00, 0x10, 0x00, 0x20, 0x00, 0x40, 0x00,
0x80, 0x01, 0x00, 0x02, 0x00,
];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x3ff);
}
}
#[test]
fn test_jit_ldxh() {
let prog = assemble(
"
ldxh r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [0xaa, 0xbb, 0x11, 0x22, 0xcc, 0xdd];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x2211);
}
}
#[test]
fn test_jit_ldxh_same_reg() {
let prog = assemble(
"
mov r0, r1
sth [r0], 0x1234
ldxh r0, [r0]
exit",
)
.unwrap();
let mem = &mut [0xff, 0xff];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x1234);
}
}
#[test]
fn test_jit_ldxw_all() {
let prog = assemble(
"
mov r0, r1
ldxw r9, [r0+0]
be32 r9
ldxw r8, [r0+4]
be32 r8
ldxw r7, [r0+8]
be32 r7
ldxw r6, [r0+12]
be32 r6
ldxw r5, [r0+16]
be32 r5
ldxw r4, [r0+20]
be32 r4
ldxw r3, [r0+24]
be32 r3
ldxw r2, [r0+28]
be32 r2
ldxw r1, [r0+32]
be32 r1
ldxw r0, [r0+36]
be32 r0
or r0, r1
or r0, r2
or r0, r3
or r0, r4
or r0, r5
or r0, r6
or r0, r7
or r0, r8
or r0, r9
exit",
)
.unwrap();
let mem = &mut [
0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00,
0x08, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00,
0x08, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00,
];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x030f0f);
}
}
#[test]
fn test_jit_ldxw() {
let prog = assemble(
"
ldxw r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [0xaa, 0xbb, 0x11, 0x22, 0x33, 0x44, 0xcc, 0xdd];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x44332211);
}
}
#[test]
fn test_jit_le16() {
let prog = assemble(
"
ldxh r0, [r1]
le16 r0
exit",
)
.unwrap();
let mem = &mut [0x22, 0x11];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x1122);
}
}
#[test]
fn test_jit_le32() {
let prog = assemble(
"
ldxw r0, [r1]
le32 r0
exit",
)
.unwrap();
let mem = &mut [0x44, 0x33, 0x22, 0x11];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x11223344);
}
}
#[test]
fn test_jit_le64() {
let prog = assemble(
"
ldxdw r0, [r1]
le64 r0
exit",
)
.unwrap();
let mem = &mut [0x88, 0x77, 0x66, 0x55, 0x44, 0x33, 0x22, 0x11];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x1122334455667788);
}
}
#[test]
fn test_jit_lsh_reg() {
let prog = assemble(
"
mov r0, 0x1
mov r7, 4
lsh r0, r7
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x10);
}
}
#[test]
fn test_jit_mod() {
let prog = assemble(
"
mov32 r0, 5748
mod32 r0, 92
mov32 r1, 13
mod32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x5);
}
}
#[test]
fn test_jit_mod32() {
let prog = assemble(
"
lddw r0, 0x100000003
mod32 r0, 3
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x0);
}
}
#[test]
fn test_jit_mod64() {
let prog = assemble(
"
mov32 r0, -1316649930
lsh r0, 32
or r0, 0x100dc5c8
mov32 r1, 0xdde263e
lsh r1, 32
or r1, 0x3cbef7f3
mod r0, r1
mod r0, 0x658f1778
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x30ba5a04);
}
}
#[test]
fn test_jit_mov() {
let prog = assemble(
"
mov32 r1, 1
mov32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_mul32_imm() {
let prog = assemble(
"
mov r0, 3
mul32 r0, 4
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0xc);
}
}
#[test]
fn test_jit_mul32_reg() {
let prog = assemble(
"
mov r0, 3
mov r1, 4
mul32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0xc);
}
}
#[test]
fn test_jit_mul32_reg_overflow() {
let prog = assemble(
"
mov r0, 0x40000001
mov r1, 4
mul32 r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x4);
}
}
#[test]
fn test_jit_mul64_imm() {
let prog = assemble(
"
mov r0, 0x40000001
mul r0, 4
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x100000004);
}
}
#[test]
fn test_jit_mul64_reg() {
let prog = assemble(
"
mov r0, 0x40000001
mov r1, 4
mul r0, r1
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x100000004);
}
}
#[test]
fn test_jit_mul_loop() {
let prog = assemble(
"
mov r0, 0x7
add r1, 0xa
lsh r1, 0x20
rsh r1, 0x20
jeq r1, 0x0, +4
mov r0, 0x7
mul r0, 0x7
add r1, -1
jne r1, 0x0, -3
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x75db9c97);
}
}
#[test]
fn test_jit_neg64() {
let prog = assemble(
"
mov32 r0, 2
neg r0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0xfffffffffffffffe);
}
}
#[test]
fn test_jit_neg() {
let prog = assemble(
"
mov32 r0, 2
neg32 r0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0xfffffffe);
}
}
#[test]
fn test_jit_prime() {
let prog = assemble(
"
mov r1, 67
mov r0, 0x1
mov r2, 0x2
jgt r1, 0x2, +4
ja +10
add r2, 0x1
mov r0, 0x1
jge r2, r1, +7
mov r3, r1
div r3, r2
mul r3, r2
mov r4, r1
sub r4, r3
mov r0, 0x0
jne r4, 0x0, -10
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_rhs32() {
let prog = assemble(
"
xor r0, r0
sub r0, 1
rsh32 r0, 8
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x00ffffff);
}
}
#[test]
fn test_jit_rsh_reg() {
let prog = assemble(
"
mov r0, 0x10
mov r7, 4
rsh r0, r7
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x1);
}
}
#[test]
fn test_jit_stack() {
let prog = assemble(
"
mov r1, 51
stdw [r10-16], 0xab
stdw [r10-8], 0xcd
and r1, 1
lsh r1, 3
mov r2, r10
add r2, r1
ldxdw r0, [r2-16]
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0xcd);
}
}
#[test]
fn test_jit_stack2() {
let prog = assemble(
"
stb [r10-4], 0x01
stb [r10-3], 0x02
stb [r10-2], 0x03
stb [r10-1], 0x04
mov r1, r10
mov r2, 0x4
sub r1, r2
call 1
mov r1, 0
ldxb r2, [r10-4]
ldxb r3, [r10-3]
ldxb r4, [r10-2]
ldxb r5, [r10-1]
call 0
xor r0, 0x2a2a2a2a
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.register_helper(0, helpers::gather_bytes).unwrap();
vm.register_helper(1, helpers::memfrob).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x01020304);
}
}
#[test]
fn test_jit_stb() {
let prog = assemble(
"
stb [r1+2], 0x11
ldxb r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [0xaa, 0xbb, 0xff, 0xcc, 0xdd];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x11);
}
}
#[test]
fn test_jit_stdw() {
let prog = assemble(
"
stdw [r1+2], 0x44332211
ldxdw r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [
0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd,
];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x44332211);
}
}
#[test]
fn test_jit_sth() {
let prog = assemble(
"
sth [r1+2], 0x2211
ldxh r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [0xaa, 0xbb, 0xff, 0xff, 0xcc, 0xdd];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x2211);
}
}
#[test]
fn test_jit_string_stack() {
let prog = assemble(
"
mov r1, 0x78636261
stxw [r10-8], r1
mov r6, 0x0
stxb [r10-4], r6
stxb [r10-12], r6
mov r1, 0x79636261
stxw [r10-16], r1
mov r1, r10
add r1, -8
mov r2, r1
call 0x4
mov r1, r0
mov r0, 0x1
lsh r1, 0x20
rsh r1, 0x20
jne r1, 0x0, +11
mov r1, r10
add r1, -8
mov r2, r10
add r2, -16
call 0x4
mov r1, r0
lsh r1, 0x20
rsh r1, 0x20
mov r0, 0x1
jeq r1, r6, +1
mov r0, 0x0
exit",
)
.unwrap();
let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
vm.register_helper(4, helpers::strcmp).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit().unwrap(), 0x0);
}
}
#[test]
fn test_jit_stw() {
let prog = assemble(
"
stw [r1+2], 0x44332211
ldxw r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x44332211);
}
}
#[test]
fn test_jit_stxb() {
let prog = assemble(
"
mov32 r2, 0x11
stxb [r1+2], r2
ldxb r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [0xaa, 0xbb, 0xff, 0xcc, 0xdd];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x11);
}
}
#[test]
fn test_jit_stxb_all() {
let prog = assemble(
"
mov r0, 0xf0
mov r2, 0xf2
mov r3, 0xf3
mov r4, 0xf4
mov r5, 0xf5
mov r6, 0xf6
mov r7, 0xf7
mov r8, 0xf8
stxb [r1], r0
stxb [r1+1], r2
stxb [r1+2], r3
stxb [r1+3], r4
stxb [r1+4], r5
stxb [r1+5], r6
stxb [r1+6], r7
stxb [r1+7], r8
ldxdw r0, [r1]
be64 r0
exit",
)
.unwrap();
let mem = &mut [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0xf0f2f3f4f5f6f7f8);
}
}
#[test]
fn test_jit_stxb_all2() {
let prog = assemble(
"
mov r0, r1
mov r1, 0xf1
mov r9, 0xf9
stxb [r0], r1
stxb [r0+1], r9
ldxh r0, [r0]
be16 r0
exit",
)
.unwrap();
let mem = &mut [0xff, 0xff];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0xf1f9);
}
}
#[test]
fn test_jit_stxb_chain() {
let prog = assemble(
"
mov r0, r1
ldxb r9, [r0+0]
stxb [r0+1], r9
ldxb r8, [r0+1]
stxb [r0+2], r8
ldxb r7, [r0+2]
stxb [r0+3], r7
ldxb r6, [r0+3]
stxb [r0+4], r6
ldxb r5, [r0+4]
stxb [r0+5], r5
ldxb r4, [r0+5]
stxb [r0+6], r4
ldxb r3, [r0+6]
stxb [r0+7], r3
ldxb r2, [r0+7]
stxb [r0+8], r2
ldxb r1, [r0+8]
stxb [r0+9], r1
ldxb r0, [r0+9]
exit",
)
.unwrap();
let mem = &mut [0x2a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x2a);
}
}
#[test]
fn test_jit_stxdw() {
let prog = assemble(
"
mov r2, -2005440939
lsh r2, 32
or r2, 0x44332211
stxdw [r1+2], r2
ldxdw r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [
0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd,
];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x8877665544332211);
}
}
#[test]
fn test_jit_stxh() {
let prog = assemble(
"
mov32 r2, 0x2211
stxh [r1+2], r2
ldxh r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [0xaa, 0xbb, 0xff, 0xff, 0xcc, 0xdd];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x2211);
}
}
#[test]
fn test_jit_stxw() {
let prog = assemble(
"
mov32 r2, 0x44332211
stxw [r1+2], r2
ldxw r0, [r1+2]
exit",
)
.unwrap();
let mem = &mut [0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x44332211);
}
}
#[test]
fn test_jit_subnet() {
let prog = assemble(
"
mov r2, 0xe
ldxh r3, [r1+12]
jne r3, 0x81, +2
mov r2, 0x12
ldxh r3, [r1+16]
and r3, 0xffff
jne r3, 0x8, +5
add r1, r2
mov r0, 0x1
ldxw r1, [r1+16]
and r1, 0xffffff
jeq r1, 0x1a8c0, +1
mov r0, 0x0
exit",
)
.unwrap();
let mem = &mut [
0x00, 0x00, 0xc0, 0x9f, 0xa0, 0x97, 0x00, 0xa0, 0xcc, 0x3b, 0xbf, 0xfa, 0x08, 0x00, 0x45,
0x10, 0x00, 0x3c, 0x46, 0x3c, 0x40, 0x00, 0x40, 0x06, 0x73, 0x1c, 0xc0, 0xa8, 0x01, 0x02,
0xc0, 0xa8, 0x01, 0x01, 0x06, 0x0e, 0x00, 0x17, 0x99, 0xc5, 0xa0, 0xec, 0x00, 0x00, 0x00,
0x00, 0xa0, 0x02, 0x7d, 0x78, 0xe0, 0xa3, 0x00, 0x00, 0x02, 0x04, 0x05, 0xb4, 0x04, 0x02,
0x08, 0x0a, 0x00, 0x9c, 0x27, 0x24, 0x00, 0x00, 0x00, 0x00, 0x01, 0x03, 0x03, 0x00,
];
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x1);
}
}
const PROG_TCP_PORT_80: [u8; 152] = [
0x71, 0x12, 0x0c, 0x00, 0x00, 0x00, 0x00, 0x00, 0x71, 0x13, 0x0d, 0x00, 0x00, 0x00, 0x00, 0x00,
0x67, 0x03, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00, 0x4f, 0x23, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x55, 0x03, 0x0c, 0x00, 0x08, 0x00, 0x00, 0x00,
0x71, 0x12, 0x17, 0x00, 0x00, 0x00, 0x00, 0x00, 0x55, 0x02, 0x0a, 0x00, 0x06, 0x00, 0x00, 0x00,
0x71, 0x12, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x07, 0x01, 0x00, 0x00, 0x0e, 0x00, 0x00, 0x00,
0x57, 0x02, 0x00, 0x00, 0x0f, 0x00, 0x00, 0x00, 0x67, 0x02, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00,
0x0f, 0x21, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x69, 0x12, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00,
0x15, 0x02, 0x02, 0x00, 0x00, 0x50, 0x00, 0x00, 0x69, 0x11, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x55, 0x01, 0x01, 0x00, 0x00, 0x50, 0x00, 0x00, 0xb7, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
];
#[test]
fn test_jit_tcp_port80_match() {
let mem = &mut [
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x08, 0x00, 0x45,
0x00, 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06, 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01,
0xc0, 0xa8, 0x00, 0x02, 0x27, 0x10, 0x00, 0x50, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x50, 0x02, 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
];
let prog = &PROG_TCP_PORT_80;
let mut vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x1);
}
}
#[test]
fn test_jit_tcp_port80_nomatch() {
let mem = &mut [
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x08, 0x00, 0x45,
0x00, 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06, 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01,
0xc0, 0xa8, 0x00, 0x02, 0x00, 0x16, 0x27, 0x10, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x51, 0x02, 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
];
let prog = &PROG_TCP_PORT_80;
let mut vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x0);
}
}
#[test]
fn test_jit_tcp_port80_nomatch_ethertype() {
let mem = &mut [
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x08, 0x01, 0x45,
0x00, 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06, 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01,
0xc0, 0xa8, 0x00, 0x02, 0x27, 0x10, 0x00, 0x50, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x50, 0x02, 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
];
let prog = &PROG_TCP_PORT_80;
let mut vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x0);
}
}
#[test]
fn test_jit_tcp_port80_nomatch_proto() {
let mem = &mut [
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x08, 0x00, 0x45,
0x00, 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x11, 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01,
0xc0, 0xa8, 0x00, 0x02, 0x27, 0x10, 0x00, 0x50, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x50, 0x02, 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
];
let prog = &PROG_TCP_PORT_80;
let mut vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem).unwrap(), 0x0);
}
}
#[test]
fn test_jit_tcp_sack_match() {
let mut mem = TCP_SACK_MATCH.to_vec();
let prog = assemble(TCP_SACK_ASM).unwrap();
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem.as_mut_slice()).unwrap(), 0x1);
}
}
#[test]
fn test_jit_tcp_sack_nomatch() {
let mut mem = TCP_SACK_NOMATCH.to_vec();
let prog = assemble(TCP_SACK_ASM).unwrap();
let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
vm.jit_compile().unwrap();
unsafe {
assert_eq!(vm.execute_program_jit(mem.as_mut_slice()).unwrap(), 0x0);
}
}
View Git Blame Copy Permalink