123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254 |
- // SPDX-License-Identifier: (Apache-2.0 OR MIT)
- #![cfg_attr(feature = "cargo-clippy", allow(clippy::unreadable_literal))]
- #![cfg(feature = "cranelift")]
- extern crate rbpf;
- mod common;
- use rbpf::{assembler::assemble, helpers};
- use crate::common::{TCP_SACK_ASM, TCP_SACK_MATCH, TCP_SACK_NOMATCH};
- macro_rules! test_cranelift {
- ($name:ident, $prog:expr, $expected:expr) => {
- #[test]
- fn $name() {
- let prog = assemble($prog).unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_cranelift().unwrap(), $expected);
- }
- };
- ($name:ident, $prog:expr, $mem:expr, $expected:expr) => {
- #[test]
- fn $name() {
- let prog = assemble($prog).unwrap();
- let mem = &mut $mem;
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_cranelift(mem).unwrap(), $expected);
- }
- };
- }
- test_cranelift!(
- test_cranelift_add,
- "
- mov32 r0, 0
- mov32 r1, 2
- add32 r0, 1
- add32 r0, r1
- exit
- ",
- 0x3
- );
- test_cranelift!(
- test_cranelift_alu64_arith,
- "
- 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
- ",
- 0x2a
- );
- test_cranelift!(
- test_cranelift_alu64_bit,
- "
- 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
- ",
- 0x11
- );
- test_cranelift!(
- test_cranelift_alu_arith,
- "
- 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
- ",
- 0x2a
- );
- test_cranelift!(
- test_cranelift_alu_bit,
- "
- 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
- ",
- 0x11
- );
- test_cranelift!(
- test_cranelift_arsh32_high_shift,
- "
- mov r0, 8
- lddw r1, 0x100000001
- arsh32 r0, r1
- exit
- ",
- 0x4
- );
- test_cranelift!(
- test_cranelift_arsh,
- "
- mov32 r0, 0xf8
- lsh32 r0, 28
- arsh32 r0, 16
- exit
- ",
- 0xffff8000
- );
- test_cranelift!(
- test_cranelift_arsh64,
- "
- mov32 r0, 1
- lsh r0, 63
- arsh r0, 55
- mov32 r1, 5
- arsh r0, r1
- exit
- ",
- 0xfffffffffffffff8
- );
- test_cranelift!(
- test_cranelift_arsh_reg,
- "
- mov32 r0, 0xf8
- mov32 r1, 16
- lsh32 r0, 28
- arsh32 r0, r1
- exit
- ",
- 0xffff8000
- );
- test_cranelift!(
- test_cranelift_be16,
- "
- ldxh r0, [r1]
- be16 r0
- exit
- ",
- [0x11, 0x22],
- 0x1122
- );
- test_cranelift!(
- test_cranelift_be16_high,
- "
- ldxdw r0, [r1]
- be16 r0
- exit
- ",
- [0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88],
- 0x1122
- );
- test_cranelift!(
- test_cranelift_be32,
- "
- ldxw r0, [r1]
- be32 r0
- exit
- ",
- [0x11, 0x22, 0x33, 0x44],
- 0x11223344
- );
- test_cranelift!(
- test_cranelift_be32_high,
- "
- ldxdw r0, [r1]
- be32 r0
- exit
- ",
- [0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88],
- 0x11223344
- );
- test_cranelift!(
- test_cranelift_be64,
- "
- ldxdw r0, [r1]
- be64 r0
- exit
- ",
- [0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88],
- 0x1122334455667788
- );
- #[test]
- fn test_cranelift_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();
- assert_eq!(vm.execute_cranelift().unwrap(), 0x0102030405);
- }
- #[test]
- #[should_panic(expected = "[CRANELIFT] Error: unknown helper function (id: 0x3f)")]
- fn test_cranelift_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- vm.execute_cranelift().unwrap();
- }
- #[test]
- fn test_cranelift_call_memfrob() {
- let prog = assemble(
- "
- mov r6, r1
- add r1, 2
- mov r2, 4
- call 1
- ldxdw r0, [r6]
- be64 r0
- exit",
- )
- .unwrap();
- let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- vm.register_helper(1, helpers::memfrob).unwrap();
- let mem = &mut [0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08];
- assert_eq!(vm.execute_cranelift(mem).unwrap(), 0x102292e2f2c0708);
- }
- test_cranelift!(
- test_cranelift_div32_high_divisor,
- "
- mov r0, 12
- lddw r1, 0x100000004
- div32 r0, r1
- exit
- ",
- 0x3
- );
- test_cranelift!(
- test_cranelift_div32_imm,
- "
- lddw r0, 0x10000000c
- div32 r0, 4
- exit
- ",
- 0x3
- );
- test_cranelift!(
- test_cranelift_div32_reg,
- "
- lddw r0, 0x10000000c
- mov r1, 4
- div32 r0, r1
- exit
- ",
- 0x3
- );
- test_cranelift!(
- test_cranelift_div64_imm,
- "
- mov r0, 0xc
- lsh r0, 32
- div r0, 4
- exit
- ",
- 0x300000000
- );
- test_cranelift!(
- test_cranelift_div64_reg,
- "
- mov r0, 0xc
- lsh r0, 32
- mov r1, 4
- div r0, r1
- exit
- ",
- 0x300000000
- );
- test_cranelift!(
- test_cranelift_div64_by_zero_imm,
- "
- mov32 r0, 1
- div r0, 0
- exit
- ",
- 0x0
- );
- test_cranelift!(
- test_cranelift_div_by_zero_imm,
- "
- mov32 r0, 1
- div32 r0, 0
- exit
- ",
- 0x0
- );
- test_cranelift!(
- test_cranelift_mod64_by_zero_imm,
- "
- mov32 r0, 1
- mod r0, 0
- exit
- ",
- 0x1
- );
- test_cranelift!(
- test_cranelift_mod_by_zero_imm,
- "
- mov32 r0, 1
- mod32 r0, 0
- exit
- ",
- 0x1
- );
- test_cranelift!(
- test_cranelift_div64_by_zero_reg,
- "
- mov32 r0, 1
- mov32 r1, 0
- div r0, r1
- exit
- ",
- 0x0
- );
- test_cranelift!(
- test_cranelift_div_by_zero_reg,
- "
- mov32 r0, 1
- mov32 r1, 0
- div32 r0, r1
- exit
- ",
- 0x0
- );
- test_cranelift!(
- test_cranelift_mod64_by_zero_reg,
- "
- mov32 r0, 1
- mov32 r1, 0
- mod r0, r1
- exit
- ",
- 0x1
- );
- test_cranelift!(
- test_cranelift_mod_by_zero_reg,
- "
- mov32 r0, 1
- mov32 r1, 0
- mod32 r0, r1
- exit
- ",
- 0x1
- );
- #[test]
- // #[should_panic(expected = "Error: out of bounds memory store (insn #1)")]
- #[ignore = "We have stack OOB checks, but we don't yet catch the trap code and convert it into a panic"]
- fn test_cranelift_err_stack_out_of_bound() {
- let prog = [
- 0x72, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00,
- ];
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- vm.execute_cranelift().unwrap();
- }
- test_cranelift!(
- test_cranelift_exit,
- "
- mov r0, 0
- exit
- ",
- 0x0
- );
- test_cranelift!(
- test_cranelift_lddw,
- "
- lddw r0, 0x1122334455667788
- exit
- ",
- 0x1122334455667788
- );
- test_cranelift!(
- test_cranelift_lddw2,
- "
- lddw r0, 0x0000000080000000
- exit
- ",
- 0x80000000
- );
- test_cranelift!(
- test_cranelift_ldxb_all,
- "
- 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
- ",
- [0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09],
- 0x9876543210
- );
- test_cranelift!(
- test_cranelift_ldxb,
- "
- ldxb r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0x11, 0xcc, 0xdd],
- 0x11
- );
- test_cranelift!(
- test_cranelift_ldxdw,
- "
- ldxdw r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0xcc, 0xdd],
- 0x8877665544332211
- );
- test_cranelift!(
- test_cranelift_ldxh_all,
- "
- 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
- ",
- [
- 0x00, 0x00, 0x00, 0x01, 0x00, 0x02, 0x00, 0x03, 0x00, 0x04, 0x00, 0x05, 0x00, 0x06, 0x00,
- 0x07, 0x00, 0x08, 0x00, 0x09
- ],
- 0x9876543210
- );
- test_cranelift!(
- test_cranelift_ldxh_all2,
- "
- 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
- ",
- [
- 0x00, 0x01, 0x00, 0x02, 0x00, 0x04, 0x00, 0x08, 0x00, 0x10, 0x00, 0x20, 0x00, 0x40, 0x00,
- 0x80, 0x01, 0x00, 0x02, 0x00
- ],
- 0x3ff
- );
- test_cranelift!(
- test_cranelift_ldxh,
- "
- ldxh r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0x11, 0x22, 0xcc, 0xdd],
- 0x2211
- );
- test_cranelift!(
- test_cranelift_ldxh_same_reg,
- "
- mov r0, r1
- sth [r0], 0x1234
- ldxh r0, [r0]
- exit
- ",
- [0xff, 0xff],
- 0x1234
- );
- test_cranelift!(
- test_cranelift_ldxw_all,
- "
- 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
- ",
- [
- 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
- ],
- 0x030f0f
- );
- test_cranelift!(
- test_cranelift_ldxw,
- "
- ldxw r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0x11, 0x22, 0x33, 0x44, 0xcc, 0xdd],
- 0x44332211
- );
- test_cranelift!(
- test_cranelift_le16,
- "
- ldxh r0, [r1]
- le16 r0
- exit
- ",
- [0x22, 0x11],
- 0x1122
- );
- test_cranelift!(
- test_cranelift_le32,
- "
- ldxw r0, [r1]
- le32 r0
- exit
- ",
- [0x44, 0x33, 0x22, 0x11],
- 0x11223344
- );
- test_cranelift!(
- test_cranelift_le64,
- "
- ldxdw r0, [r1]
- le64 r0
- exit
- ",
- [0x88, 0x77, 0x66, 0x55, 0x44, 0x33, 0x22, 0x11],
- 0x1122334455667788
- );
- test_cranelift!(
- test_cranelift_lsh_reg,
- "
- mov r0, 0x1
- mov r7, 4
- lsh r0, r7
- exit
- ",
- 0x10
- );
- test_cranelift!(
- test_cranelift_mod,
- "
- mov32 r0, 5748
- mod32 r0, 92
- mov32 r1, 13
- mod32 r0, r1
- exit
- ",
- 0x5
- );
- test_cranelift!(
- test_cranelift_mod32,
- "
- lddw r0, 0x100000003
- mod32 r0, 3
- exit
- ",
- 0x0
- );
- test_cranelift!(
- test_cranelift_mod64,
- "
- 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
- ",
- 0x30ba5a04
- );
- test_cranelift!(
- test_cranelift_mov,
- "
- mov32 r1, 1
- mov32 r0, r1
- exit
- ",
- 0x1
- );
- test_cranelift!(
- test_cranelift_mul32_imm,
- "
- mov r0, 3
- mul32 r0, 4
- exit
- ",
- 0xc
- );
- test_cranelift!(
- test_cranelift_mul32_reg,
- "
- mov r0, 3
- mov r1, 4
- mul32 r0, r1
- exit
- ",
- 0xc
- );
- test_cranelift!(
- test_cranelift_mul32_reg_overflow,
- "
- mov r0, 0x40000001
- mov r1, 4
- mul32 r0, r1
- exit
- ",
- 0x4
- );
- test_cranelift!(
- test_cranelift_mul64_imm,
- "
- mov r0, 0x40000001
- mul r0, 4
- exit
- ",
- 0x100000004
- );
- test_cranelift!(
- test_cranelift_mul64_reg,
- "
- mov r0, 0x40000001
- mov r1, 4
- mul r0, r1
- exit
- ",
- 0x100000004
- );
- test_cranelift!(
- test_cranelift_neg64,
- "
- mov32 r0, 2
- neg r0
- exit
- ",
- 0xfffffffffffffffe
- );
- test_cranelift!(
- test_cranelift_neg,
- "
- mov32 r0, 2
- neg32 r0
- exit
- ",
- 0xfffffffe
- );
- test_cranelift!(
- test_cranelift_rhs32,
- "
- xor r0, r0
- sub r0, 1
- rsh32 r0, 8
- exit
- ",
- 0x00ffffff
- );
- test_cranelift!(
- test_cranelift_rsh_reg,
- "
- mov r0, 0x10
- mov r7, 4
- rsh r0, r7
- exit
- ",
- 0x1
- );
- test_cranelift!(
- test_cranelift_stack,
- "
- 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
- ",
- 0xcd
- );
- #[test]
- fn test_cranelift_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();
- assert_eq!(vm.execute_cranelift().unwrap(), 0x01020304);
- }
- test_cranelift!(
- test_cranelift_stb,
- "
- stb [r1+2], 0x11
- ldxb r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0xff, 0xcc, 0xdd],
- 0x11
- );
- test_cranelift!(
- test_cranelift_stdw,
- "
- stdw [r1+2], 0x44332211
- ldxdw r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd],
- 0x44332211
- );
- test_cranelift!(
- test_cranelift_sth,
- "
- sth [r1+2], 0x2211
- ldxh r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0xff, 0xff, 0xcc, 0xdd],
- 0x2211
- );
- #[test]
- #[ignore]
- fn test_cranelift_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();
- assert_eq!(vm.execute_cranelift().unwrap(), 0x0);
- }
- test_cranelift!(
- test_cranelift_stw,
- "
- stw [r1+2], 0x44332211
- ldxw r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd],
- 0x44332211
- );
- test_cranelift!(
- test_cranelift_stxb,
- "
- mov32 r2, 0x11
- stxb [r1+2], r2
- ldxb r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0xff, 0xcc, 0xdd],
- 0x11
- );
- test_cranelift!(
- test_cranelift_stxb_all,
- "
- 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
- ",
- [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff],
- 0xf0f2f3f4f5f6f7f8
- );
- test_cranelift!(
- test_cranelift_stxb_all2,
- "
- mov r0, r1
- mov r1, 0xf1
- mov r9, 0xf9
- stxb [r0], r1
- stxb [r0+1], r9
- ldxh r0, [r0]
- be16 r0
- exit
- ",
- [0xff, 0xff],
- 0xf1f9
- );
- test_cranelift!(
- test_cranelift_stxb_chain,
- "
- 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
- ",
- [0x2a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00],
- 0x2a
- );
- test_cranelift!(
- test_cranelift_stxdw,
- "
- mov r2, -2005440939
- lsh r2, 32
- or r2, 0x44332211
- stxdw [r1+2], r2
- ldxdw r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd],
- 0x8877665544332211
- );
- test_cranelift!(
- test_cranelift_stxh,
- "
- mov32 r2, 0x2211
- stxh [r1+2], r2
- ldxh r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0xff, 0xff, 0xcc, 0xdd],
- 0x2211
- );
- test_cranelift!(
- test_cranelift_stxw,
- "
- mov32 r2, 0x44332211
- stxw [r1+2], r2
- ldxw r0, [r1+2]
- exit
- ",
- [0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd],
- 0x44332211
- );
- 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]
- #[ignore]
- fn test_cranelift_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 vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
- assert_eq!(vm.execute_cranelift(mem).unwrap(), 0x1);
- }
- #[test]
- #[ignore]
- fn test_cranelift_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 vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
- assert_eq!(vm.execute_cranelift(mem).unwrap(), 0x0);
- }
- #[test]
- #[ignore]
- fn test_cranelift_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 vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
- assert_eq!(vm.execute_cranelift(mem).unwrap(), 0x0);
- }
- #[test]
- #[ignore]
- fn test_cranelift_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 vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
- assert_eq!(vm.execute_cranelift(mem).unwrap(), 0x0);
- }
- #[test]
- #[ignore]
- fn test_cranelift_tcp_sack_match() {
- let mut mem = TCP_SACK_MATCH.to_vec();
- let prog = assemble(TCP_SACK_ASM).unwrap();
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_cranelift(mem.as_mut_slice()).unwrap(), 0x1);
- }
- #[test]
- #[ignore]
- fn test_cranelift_tcp_sack_nomatch() {
- let mut mem = TCP_SACK_NOMATCH.to_vec();
- let prog = assemble(TCP_SACK_ASM).unwrap();
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_cranelift(mem.as_mut_slice()).unwrap(), 0x0);
- }
|