123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557 |
- // Copyright 2016 6WIND S.A. <quentin.monnet@6wind.com>
- //
- // Licensed under the Apache License, Version 2.0 <http://www.apache.org/licenses/LICENSE-2.0> or
- // the MIT license <http://opensource.org/licenses/MIT>, at your option. This file may not be
- // copied, modified, or distributed except according to those terms.
- // There are unused mut warnings due to unsafe code.
- #![allow(unused_mut)]
- #![cfg_attr(feature = "cargo-clippy", allow(unreadable_literal))]
- // These crates would be needed to load bytecode from a BPF-compiled object file. Since the crates
- // are not used anywhere else in the library, it is deactivated: we do not want to load and compile
- // them just for the tests. If you want to use them, do not forget to add the following
- // dependencies to your Cargo.toml file:
- //
- // ---
- // byteorder = "0.5.3"
- // elf = "0.0.10"
- // ---
- //
- // extern crate byteorder;
- // extern crate elf;
- // use std::path::PathBuf;
- extern crate rbpf;
- use rbpf::helpers;
- // The following two examples have been compiled from C with the following command:
- //
- // ```bash
- // clang -O2 -emit-llvm -c <file.c> -o - | llc -march=bpf -filetype=obj -o <file.o>
- // ```
- //
- // The C source code was the following:
- //
- // ```c
- // #include <linux/ip.h>
- // #include <linux/in.h>
- // #include <linux/tcp.h>
- // #include <linux/bpf.h>
- //
- // #define ETH_ALEN 6
- // #define ETH_P_IP 0x0008 /* htons(0x0800) */
- // #define TCP_HDR_LEN 20
- //
- // #define BLOCKED_TCP_PORT 0x9999
- //
- // struct eth_hdr {
- // unsigned char h_dest[ETH_ALEN];
- // unsigned char h_source[ETH_ALEN];
- // unsigned short h_proto;
- // };
- //
- // #define SEC(NAME) __attribute__((section(NAME), used))
- // SEC(".classifier")
- // int handle_ingress(struct __sk_buff *skb)
- // {
- // void *data = (void *)(long)skb->data;
- // void *data_end = (void *)(long)skb->data_end;
- // struct eth_hdr *eth = data;
- // struct iphdr *iph = data + sizeof(*eth);
- // struct tcphdr *tcp = data + sizeof(*eth) + sizeof(*iph);
- //
- // /* single length check */
- // if (data + sizeof(*eth) + sizeof(*iph) + sizeof(*tcp) > data_end)
- // return 0;
- // if (eth->h_proto != ETH_P_IP)
- // return 0;
- // if (iph->protocol != IPPROTO_TCP)
- // return 0;
- // if (tcp->source == BLOCKED_TCP_PORT || tcp->dest == BLOCKED_TCP_PORT)
- // return -1;
- // return 0;
- // }
- // char _license[] SEC(".license") = "GPL";
- // ```
- //
- // This program, once compiled, can be injected into Linux kernel, with tc for instance. Sadly, we
- // need to bring some modifications to the generated bytecode in order to run it: the three
- // instructions with opcode 0x61 load data from a packet area as 4-byte words, where we need to
- // load it as 8-bytes double words (0x79). The kernel does the same kind of translation before
- // running the program, but rbpf does not implement this.
- //
- // In addition, the offset at which the pointer to the packet data is stored must be changed: since
- // we use 8 bytes instead of 4 for the start and end addresses of the data packet, we cannot use
- // the offsets produced by clang (0x4c and 0x50), the addresses would overlap. Instead we can use,
- // for example, 0x40 and 0x50. See comments on the bytecode below to see the modifications.
- //
- // Once the bytecode has been (manually, in our case) edited, we can load the bytecode directly
- // from the ELF object file. This is easy to do, but requires the addition of two crates in the
- // Cargo.toml file (see comments above), so here we use just the hardcoded bytecode instructions
- // instead.
- #[test]
- fn test_vm_block_port() {
- // To load the bytecode from an object file instead of using the hardcoded instructions,
- // use the additional crates commented at the beginning of this file (and also add them to your
- // Cargo.toml). See comments above.
- //
- // ---
- // let filename = "my_ebpf_object_file.o";
- //
- // let path = PathBuf::from(filename);
- // let file = match elf::File::open_path(&path) {
- // Ok(f) => f,
- // Err(e) => panic!("Error: {:?}", e),
- // };
- //
- // let text_scn = match file.get_section(".classifier") {
- // Some(s) => s,
- // None => panic!("Failed to look up .classifier section"),
- // };
- //
- // let prog = &text_scn.data;
- // ---
- let prog = &[
- 0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x79, 0x12, 0x50, 0x00, 0x00, 0x00, 0x00, 0x00, // 0x79 instead of 0x61
- 0x79, 0x11, 0x40, 0x00, 0x00, 0x00, 0x00, 0x00, // 0x79 instead of 0x61, 0x40 i.o. 0x4c
- 0xbf, 0x13, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x07, 0x03, 0x00, 0x00, 0x36, 0x00, 0x00, 0x00,
- 0x2d, 0x23, 0x12, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x69, 0x12, 0x0c, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x55, 0x02, 0x10, 0x00, 0x08, 0x00, 0x00, 0x00,
- 0x71, 0x12, 0x17, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x55, 0x02, 0x0e, 0x00, 0x06, 0x00, 0x00, 0x00,
- 0x18, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x79, 0x11, 0x22, 0x00, 0x00, 0x00, 0x00, 0x00, // 0x79 instead of 0x61
- 0xbf, 0x12, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x57, 0x02, 0x00, 0x00, 0xff, 0xff, 0x00, 0x00,
- 0x15, 0x02, 0x08, 0x00, 0x99, 0x99, 0x00, 0x00,
- 0x18, 0x02, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x5f, 0x21, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0xb7, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff,
- 0x18, 0x02, 0x00, 0x00, 0x00, 0x00, 0x99, 0x99,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x1d, 0x21, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let packet = &mut [
- 0x01, 0x23, 0x45, 0x67, 0x89, 0xab,
- 0xfe, 0xdc, 0xba, 0x98, 0x76, 0x54,
- 0x08, 0x00, // ethertype
- 0x45, 0x00, 0x00, 0x3b, // start ip_hdr
- 0xa6, 0xab, 0x40, 0x00,
- 0x40, 0x06, 0x96, 0x0f,
- 0x7f, 0x00, 0x00, 0x01,
- 0x7f, 0x00, 0x00, 0x01,
- // Program matches the next two bytes: 0x9999 returns 0xffffffff, else return 0.
- 0x99, 0x99, 0xc6, 0xcc, // start tcp_hdr
- 0xd1, 0xe5, 0xc4, 0x9d,
- 0xd4, 0x30, 0xb5, 0xd2,
- 0x80, 0x18, 0x01, 0x56,
- 0xfe, 0x2f, 0x00, 0x00,
- 0x01, 0x01, 0x08, 0x0a, // start data
- 0x00, 0x23, 0x75, 0x89,
- 0x00, 0x23, 0x63, 0x2d,
- 0x71, 0x64, 0x66, 0x73,
- 0x64, 0x66, 0x0au8
- ];
- let mut vm = rbpf::EbpfVmFixedMbuff::new(prog, 0x40, 0x50);
- vm.register_helper(helpers::BPF_TRACE_PRINTK_IDX, helpers::bpf_trace_printf);
- let res = vm.prog_exec(packet);
- println!("Program returned: {:?} ({:#x})", res, res);
- assert_eq!(res, 0xffffffff);
- }
- #[test]
- fn test_jit_block_port() {
- // To load the bytecode from an object file instead of using the hardcoded instructions,
- // use the additional crates commented at the beginning of this file (and also add them to your
- // Cargo.toml). See comments above.
- //
- // ---
- // let filename = "my_ebpf_object_file.o";
- //
- // let path = PathBuf::from(filename);
- // let file = match elf::File::open_path(&path) {
- // Ok(f) => f,
- // Err(e) => panic!("Error: {:?}", e),
- // };
- //
- // let text_scn = match file.get_section(".classifier") {
- // Some(s) => s,
- // None => panic!("Failed to look up .classifier section"),
- // };
- //
- // let prog = &text_scn.data;
- // ---
- let prog = &[
- 0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x79, 0x12, 0x50, 0x00, 0x00, 0x00, 0x00, 0x00, // 0x79 instead of 0x61
- 0x79, 0x11, 0x40, 0x00, 0x00, 0x00, 0x00, 0x00, // 0x79 instead of 0x61, 0x40 i.o. 0x4c
- 0xbf, 0x13, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x07, 0x03, 0x00, 0x00, 0x36, 0x00, 0x00, 0x00,
- 0x2d, 0x23, 0x12, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x69, 0x12, 0x0c, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x55, 0x02, 0x10, 0x00, 0x08, 0x00, 0x00, 0x00,
- 0x71, 0x12, 0x17, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x55, 0x02, 0x0e, 0x00, 0x06, 0x00, 0x00, 0x00,
- 0x18, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x79, 0x11, 0x22, 0x00, 0x00, 0x00, 0x00, 0x00, // 0x79 instead of 0x61
- 0xbf, 0x12, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x57, 0x02, 0x00, 0x00, 0xff, 0xff, 0x00, 0x00,
- 0x15, 0x02, 0x08, 0x00, 0x99, 0x99, 0x00, 0x00,
- 0x18, 0x02, 0x00, 0x00, 0x00, 0x00, 0xff, 0xff,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x5f, 0x21, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0xb7, 0x00, 0x00, 0x00, 0xff, 0xff, 0xff, 0xff,
- 0x18, 0x02, 0x00, 0x00, 0x00, 0x00, 0x99, 0x99,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x1d, 0x21, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let packet = &mut [
- 0x01, 0x23, 0x45, 0x67, 0x89, 0xab,
- 0xfe, 0xdc, 0xba, 0x98, 0x76, 0x54,
- 0x08, 0x00, // ethertype
- 0x45, 0x00, 0x00, 0x3b, // start ip_hdr
- 0xa6, 0xab, 0x40, 0x00,
- 0x40, 0x06, 0x96, 0x0f,
- 0x7f, 0x00, 0x00, 0x01,
- 0x7f, 0x00, 0x00, 0x01,
- // Program matches the next two bytes: 0x9999 returns 0xffffffff, else return 0.
- 0x99, 0x99, 0xc6, 0xcc, // start tcp_hdr
- 0xd1, 0xe5, 0xc4, 0x9d,
- 0xd4, 0x30, 0xb5, 0xd2,
- 0x80, 0x18, 0x01, 0x56,
- 0xfe, 0x2f, 0x00, 0x00,
- 0x01, 0x01, 0x08, 0x0a, // start data
- 0x00, 0x23, 0x75, 0x89,
- 0x00, 0x23, 0x63, 0x2d,
- 0x71, 0x64, 0x66, 0x73,
- 0x64, 0x66, 0x0au8
- ];
- let mut vm = rbpf::EbpfVmFixedMbuff::new(prog, 0x40, 0x50);
- vm.register_helper(helpers::BPF_TRACE_PRINTK_IDX, helpers::bpf_trace_printf);
- vm.jit_compile();
- unsafe {
- let res = vm.prog_exec_jit(packet);
- println!("Program returned: {:?} ({:#x})", res, res);
- assert_eq!(res, 0xffffffff);
- }
- }
- // Program and memory come from uBPF test ldxh.
- #[test]
- fn test_vm_mbuff() {
- let prog = &[
- // Load mem from mbuff into R1
- 0x79, 0x11, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00,
- // ldhx r1[2], r0
- 0x69, 0x10, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &[
- 0xaa, 0xbb, 0x11, 0x22, 0xcc, 0xdd
- ];
- let mbuff = [0u8; 32];
- unsafe {
- let mut data = mbuff.as_ptr().offset(8) as *mut u64;
- let mut data_end = mbuff.as_ptr().offset(24) as *mut u64;
- *data = mem.as_ptr() as u64;
- *data_end = mem.as_ptr() as u64 + mem.len() as u64;
- }
- let vm = rbpf::EbpfVmMbuff::new(prog);
- assert_eq!(vm.prog_exec(mem, &mbuff), 0x2211);
- }
- // Program and memory come from uBPF test ldxh.
- #[test]
- fn test_vm_mbuff_with_rust_api() {
- use rbpf::insn_builder::*;
- let mut program = BpfCode::new();
- program
- .load_x(MemSize::DoubleWord).set_dst(0x01).set_src(0x01).set_off(0x00_08).push()
- .load_x(MemSize::HalfWord).set_dst(0x00).set_src(0x01).set_off(0x00_02).push()
- .exit().push();
- let mem = &[
- 0xaa, 0xbb, 0x11, 0x22, 0xcc, 0xdd
- ];
- let mbuff = [0u8; 32];
- unsafe {
- let mut data = mbuff.as_ptr().offset(8) as *mut u64;
- let mut data_end = mbuff.as_ptr().offset(24) as *mut u64;
- *data = mem.as_ptr() as u64;
- *data_end = mem.as_ptr() as u64 + mem.len() as u64;
- }
- let vm = rbpf::EbpfVmMbuff::new(program.into_bytes());
- assert_eq!(vm.prog_exec(mem, &mbuff), 0x2211);
- }
- // Program and memory come from uBPF test ldxh.
- #[test]
- fn test_jit_mbuff() {
- let prog = &[
- // Load mem from mbuff into R1
- 0x79, 0x11, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00,
- // ldhx r1[2], r0
- 0x69, 0x10, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0xaa, 0xbb, 0x11, 0x22, 0xcc, 0xdd
- ];
- let mut mbuff = [0u8; 32];
- unsafe {
- let mut data = mbuff.as_ptr().offset(8) as *mut u64;
- let mut data_end = mbuff.as_ptr().offset(24) as *mut u64;
- *data = mem.as_ptr() as u64;
- *data_end = mem.as_ptr() as u64 + mem.len() as u64;
- }
- unsafe {
- let mut vm = rbpf::EbpfVmMbuff::new(prog);
- vm.jit_compile();
- assert_eq!(vm.prog_exec_jit(mem, &mut mbuff), 0x2211);
- }
- }
- #[test]
- fn test_vm_jit_ldabsb() {
- let prog = &[
- 0x30, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let mut vm = rbpf::EbpfVmRaw::new(prog);
- assert_eq!(vm.prog_exec(mem), 0x33);
- vm.jit_compile();
- unsafe {
- assert_eq!(vm.prog_exec_jit(mem), 0x33);
- };
- }
- #[test]
- fn test_vm_jit_ldabsh() {
- let prog = &[
- 0x28, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let mut vm = rbpf::EbpfVmRaw::new(prog);
- assert_eq!(vm.prog_exec(mem), 0x4433);
- vm.jit_compile();
- unsafe {
- assert_eq!(vm.prog_exec_jit(mem), 0x4433);
- };
- }
- #[test]
- fn test_vm_jit_ldabsw() {
- let prog = &[
- 0x20, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let mut vm = rbpf::EbpfVmRaw::new(prog);
- assert_eq!(vm.prog_exec(mem), 0x66554433);
- vm.jit_compile();
- unsafe {
- assert_eq!(vm.prog_exec_jit(mem), 0x66554433);
- };
- }
- #[test]
- fn test_vm_jit_ldabsdw() {
- let prog = &[
- 0x38, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let mut vm = rbpf::EbpfVmRaw::new(prog);
- assert_eq!(vm.prog_exec(mem), 0xaa99887766554433);
- vm.jit_compile();
- unsafe {
- assert_eq!(vm.prog_exec_jit(mem), 0xaa99887766554433);
- };
- }
- #[test]
- #[should_panic(expected = "Error: out of bounds memory load (insn #1),")]
- fn test_vm_err_ldabsb_oob() {
- let prog = &[
- 0x38, 0x00, 0x00, 0x00, 0x33, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let vm = rbpf::EbpfVmRaw::new(prog);
- vm.prog_exec(mem);
- // Memory check not implemented for JIT yet.
- }
- #[test]
- #[should_panic(expected = "Error: out of bounds memory load (insn #1),")]
- fn test_vm_err_ldabsb_nomem() {
- let prog = &[
- 0x38, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let vm = rbpf::EbpfVmNoData::new(prog);
- vm.prog_exec();
- // Memory check not implemented for JIT yet.
- }
- #[test]
- fn test_vm_jit_ldindb() {
- let prog = &[
- 0xb7, 0x01, 0x00, 0x00, 0x05, 0x00, 0x00, 0x00,
- 0x50, 0x10, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let mut vm = rbpf::EbpfVmRaw::new(prog);
- assert_eq!(vm.prog_exec(mem), 0x88);
- vm.jit_compile();
- unsafe {
- assert_eq!(vm.prog_exec_jit(mem), 0x88);
- };
- }
- #[test]
- fn test_vm_jit_ldindh() {
- let prog = &[
- 0xb7, 0x01, 0x00, 0x00, 0x05, 0x00, 0x00, 0x00,
- 0x48, 0x10, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let mut vm = rbpf::EbpfVmRaw::new(prog);
- assert_eq!(vm.prog_exec(mem), 0x9988);
- vm.jit_compile();
- unsafe {
- assert_eq!(vm.prog_exec_jit(mem), 0x9988);
- };
- }
- #[test]
- fn test_vm_jit_ldindw() {
- let prog = &[
- 0xb7, 0x01, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00,
- 0x40, 0x10, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let mut vm = rbpf::EbpfVmRaw::new(prog);
- assert_eq!(vm.prog_exec(mem), 0x88776655);
- vm.jit_compile();
- unsafe {
- assert_eq!(vm.prog_exec_jit(mem), 0x88776655);
- };
- }
- #[test]
- fn test_vm_jit_ldinddw() {
- let prog = &[
- 0xb7, 0x01, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00,
- 0x58, 0x10, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let mut vm = rbpf::EbpfVmRaw::new(prog);
- assert_eq!(vm.prog_exec(mem), 0xccbbaa9988776655);
- vm.jit_compile();
- unsafe {
- assert_eq!(vm.prog_exec_jit(mem), 0xccbbaa9988776655);
- };
- }
- #[test]
- #[should_panic(expected = "Error: out of bounds memory load (insn #2),")]
- fn test_vm_err_ldindb_oob() {
- let prog = &[
- 0xb7, 0x01, 0x00, 0x00, 0x05, 0x00, 0x00, 0x00,
- 0x38, 0x10, 0x00, 0x00, 0x33, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let mem = &mut [
- 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
- 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
- ];
- let vm = rbpf::EbpfVmRaw::new(prog);
- vm.prog_exec(mem);
- // Memory check not implemented for JIT yet.
- }
- #[test]
- #[should_panic(expected = "Error: out of bounds memory load (insn #2),")]
- fn test_vm_err_ldindb_nomem() {
- let prog = &[
- 0xb7, 0x01, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x38, 0x10, 0x00, 0x00, 0x03, 0x00, 0x00, 0x00,
- 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
- ];
- let vm = rbpf::EbpfVmNoData::new(prog);
- vm.prog_exec();
- // Memory check not implemented for JIT yet.
- }
|