1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674 |
- // 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. <[email protected]>
- // 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_vm_<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 interpreter.
- #![allow(clippy::unreadable_literal)]
- extern crate rbpf;
- mod common;
- use common::{TCP_SACK_ASM, TCP_SACK_MATCH, TCP_SACK_NOMATCH};
- use rbpf::{assembler::assemble, helpers};
- #[test]
- fn test_vm_add() {
- let prog = assemble(
- "
- mov32 r0, 0
- mov32 r1, 2
- add32 r0, 1
- add32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x3);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x2a);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x11);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x2a);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x11);
- }
- #[test]
- fn test_vm_arsh32_high_shift() {
- let prog = assemble(
- "
- mov r0, 8
- lddw r1, 0x100000001
- arsh32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x4);
- }
- #[test]
- fn test_vm_arsh() {
- let prog = assemble(
- "
- mov32 r0, 0xf8
- lsh32 r0, 28
- arsh32 r0, 16
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xffff8000);
- }
- #[test]
- fn test_vm_arsh64() {
- let prog = assemble(
- "
- mov32 r0, 1
- lsh r0, 63
- arsh r0, 55
- mov32 r1, 5
- arsh r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xfffffffffffffff8);
- }
- #[test]
- fn test_vm_arsh_reg() {
- let prog = assemble(
- "
- mov32 r0, 0xf8
- mov32 r1, 16
- lsh32 r0, 28
- arsh32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xffff8000);
- }
- #[test]
- fn test_vm_arsh_imm_overflow() {
- let prog = assemble(
- "
- mov r0, 1
- lsh r0, 63
- arsh r0, 0xff20
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xffffffff80000000);
- }
- #[test]
- fn test_vm_arsh_reg_overflow() {
- let prog = assemble(
- "
- mov r0, 1
- lsh r0, 63
- mov r1, 0xff04
- arsh r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xf800000000000000);
- }
- #[test]
- fn test_vm_arsh32_imm_overflow() {
- let prog = assemble(
- "
- mov32 r0, 1
- lsh32 r0, 31
- arsh32 r0, 0xff10
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xffff8000);
- }
- #[test]
- fn test_vm_arsh32_reg_overflow() {
- let prog = assemble(
- "
- mov32 r0, 1
- lsh32 r0, 31
- mov32 r1, 32
- arsh32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x80000000);
- }
- #[test]
- fn test_vm_be16() {
- let prog = assemble(
- "
- ldxh r0, [r1]
- be16 r0
- exit",
- )
- .unwrap();
- let mem = &mut [0x11, 0x22];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x1122);
- }
- #[test]
- fn test_vm_be16_high() {
- let prog = assemble(
- "
- ldxdw r0, [r1]
- be16 r0
- exit",
- )
- .unwrap();
- let mem = &mut [0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x1122);
- }
- #[test]
- fn test_vm_be32() {
- let prog = assemble(
- "
- ldxw r0, [r1]
- be32 r0
- exit",
- )
- .unwrap();
- let mem = &mut [0x11, 0x22, 0x33, 0x44];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x11223344);
- }
- #[test]
- fn test_vm_be32_high() {
- let prog = assemble(
- "
- ldxdw r0, [r1]
- be32 r0
- exit",
- )
- .unwrap();
- let mem = &mut [0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x11223344);
- }
- #[test]
- fn test_vm_be64() {
- let prog = assemble(
- "
- ldxdw r0, [r1]
- be64 r0
- exit",
- )
- .unwrap();
- let mem = &mut [0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x1122334455667788);
- }
- #[test]
- fn test_vm_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_program().unwrap(), 0x0102030405);
- }
- #[test]
- fn test_vm_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();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x102292e2f2c0708);
- }
- // TODO: helpers::trash_registers needs asm!().
- // Try this again once asm!() is available in stable.
- //#[test]
- //fn test_vm_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);
- //assert_eq!(vm.execute_program().unwrap(), 0x4321);
- //}
- #[test]
- fn test_vm_div32_high_divisor() {
- let prog = assemble(
- "
- mov r0, 12
- lddw r1, 0x100000004
- div32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x3);
- }
- #[test]
- fn test_vm_div32_imm() {
- let prog = assemble(
- "
- lddw r0, 0x10000000c
- div32 r0, 4
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x3);
- }
- #[test]
- fn test_vm_div32_reg() {
- let prog = assemble(
- "
- lddw r0, 0x10000000c
- mov r1, 4
- div32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x3);
- }
- #[test]
- fn test_vm_div64_imm() {
- let prog = assemble(
- "
- mov r0, 0xc
- lsh r0, 32
- div r0, 4
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x300000000);
- }
- #[test]
- fn test_vm_div64_reg() {
- let prog = assemble(
- "
- mov r0, 0xc
- lsh r0, 32
- mov r1, 4
- div r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x300000000);
- }
- #[test]
- fn test_vm_early_exit() {
- let prog = assemble(
- "
- mov r0, 3
- exit
- mov r0, 4
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x3);
- }
- // uBPF limits the number of user functions at 64. We don't.
- //#[test]
- //fn test_vm_err_call_bad_imm() {
- //}
- #[test]
- #[should_panic(expected = "Error: unknown helper function (id: 0x3f)")]
- fn test_vm_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_program().unwrap();
- }
- #[test]
- fn test_vm_div64_by_zero_imm() {
- let prog = assemble(
- "
- mov32 r0, 1
- div r0, 0
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x0);
- }
- #[test]
- fn test_vm_div_by_zero_imm() {
- let prog = assemble(
- "
- mov32 r0, 1
- div32 r0, 0
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x0);
- }
- #[test]
- fn test_vm_mod64_by_zero_imm() {
- let prog = assemble(
- "
- mov32 r0, 1
- mod r0, 0
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_mod_by_zero_imm() {
- let prog = assemble(
- "
- mov32 r0, 1
- mod32 r0, 0
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- // Make sure we only consider the last 32 bits of the divisor.
- #[test]
- fn test_vm_mod_by_zero_reg_long() {
- let prog = assemble(
- "
- lddw r1, 0x100000000
- mod32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x0);
- }
- #[test]
- fn test_vm_div64_by_zero_reg() {
- let prog = assemble(
- "
- mov32 r0, 1
- mov32 r1, 0
- div r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x0);
- }
- #[test]
- fn test_vm_div_by_zero_reg() {
- let prog = assemble(
- "
- mov32 r0, 1
- mov32 r1, 0
- div32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x0);
- }
- // Make sure we only consider the last 32 bits of the divisor.
- #[test]
- fn test_vm_div_by_zero_reg_long() {
- let prog = assemble(
- "
- lddw r1, 0x100000000
- div32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x0);
- }
- #[test]
- fn test_vm_mod64_by_zero_reg() {
- let prog = assemble(
- "
- mov32 r0, 1
- mov32 r1, 0
- mod r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_mod_by_zero_reg() {
- let prog = assemble(
- "
- mov32 r0, 1
- mov32 r1, 0
- mod32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- #[should_panic(expected = "Error: out of bounds memory store (insn #1)")]
- fn test_vm_err_stack_out_of_bound() {
- let prog = assemble(
- "
- stb [r10], 0
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- vm.execute_program().unwrap();
- }
- #[test]
- fn test_vm_exit() {
- let prog = assemble(
- "
- mov r0, 0
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x0);
- }
- #[test]
- fn test_vm_ja() {
- let prog = assemble(
- "
- mov r0, 1
- ja +1
- mov r0, 2
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_lddw() {
- let prog = assemble(
- "lddw r0, 0x1122334455667788
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1122334455667788);
- }
- #[test]
- fn test_vm_lddw2() {
- let prog = assemble(
- "
- lddw r0, 0x0000000080000000
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x80000000);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x9876543210);
- }
- #[test]
- fn test_vm_ldxb() {
- let prog = assemble(
- "
- ldxb r0, [r1+2]
- exit",
- )
- .unwrap();
- let mem = &mut [0xaa, 0xbb, 0x11, 0xcc, 0xdd];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x11);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x8877665544332211);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x9876543210);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x3ff);
- }
- #[test]
- fn test_vm_ldxh() {
- let prog = assemble(
- "
- ldxh r0, [r1+2]
- exit",
- )
- .unwrap();
- let mem = &mut [0xaa, 0xbb, 0x11, 0x22, 0xcc, 0xdd];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x2211);
- }
- #[test]
- fn test_vm_ldxh_same_reg() {
- let prog = assemble(
- "
- mov r0, r1
- sth [r0], 0x1234
- ldxh r0, [r0]
- exit",
- )
- .unwrap();
- let mem = &mut [0xff, 0xff];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x1234);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x030f0f);
- }
- #[test]
- fn test_vm_ldxw() {
- let prog = assemble(
- "
- ldxw r0, [r1+2]
- exit",
- )
- .unwrap();
- let mem = &mut [0xaa, 0xbb, 0x11, 0x22, 0x33, 0x44, 0xcc, 0xdd];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x44332211);
- }
- #[test]
- fn test_vm_le16() {
- let prog = assemble(
- "
- ldxh r0, [r1]
- le16 r0
- exit",
- )
- .unwrap();
- let mem = &mut [0x22, 0x11];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x1122);
- }
- #[test]
- fn test_vm_le32() {
- let prog = assemble(
- "
- ldxw r0, [r1]
- le32 r0
- exit",
- )
- .unwrap();
- let mem = &mut [0x44, 0x33, 0x22, 0x11];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x11223344);
- }
- #[test]
- fn test_vm_le64() {
- let prog = assemble(
- "
- ldxdw r0, [r1]
- le64 r0
- exit",
- )
- .unwrap();
- let mem = &mut [0x88, 0x77, 0x66, 0x55, 0x44, 0x33, 0x22, 0x11];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x1122334455667788);
- }
- #[test]
- fn test_vm_lsh_imm() {
- let prog = assemble(
- "
- mov r0, 1
- lsh r0, 4
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x10);
- }
- #[test]
- fn test_vm_lsh_reg() {
- let prog = assemble(
- "
- mov r0, 1
- mov r7, 4
- lsh r0, r7
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x10);
- }
- #[test]
- fn test_vm_lsh32_imm() {
- let prog = assemble(
- "
- mov32 r0, 1
- lsh32 r0, 4
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x10);
- }
- #[test]
- fn test_vm_lsh32_reg() {
- let prog = assemble(
- "
- mov32 r0, 1
- mov32 r7, 4
- lsh32 r0, r7
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x10);
- }
- #[test]
- fn test_vm_lsh_imm_overflow() {
- let prog = assemble(
- "
- mov r0, 1
- lsh r0, 64
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_lsh_reg_overflow() {
- let prog = assemble(
- "
- mov r0, 1
- mov r7, 64
- lsh r0, r7
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_lsh32_imm_overflow() {
- let prog = assemble(
- "
- mov32 r0, 1
- lsh32 r0, 32
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_lsh32_reg_overflow() {
- let prog = assemble(
- "
- mov32 r0, 1
- mov32 r7, 32
- lsh32 r0, r7
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_mod() {
- let prog = assemble(
- "
- mov32 r0, 5748
- mod32 r0, 92
- mov32 r1, 13
- mod32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x5);
- }
- #[test]
- fn test_vm_mod32() {
- let prog = assemble(
- "
- lddw r0, 0x100000003
- mod32 r0, 3
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x0);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x30ba5a04);
- }
- #[test]
- fn test_vm_mov() {
- let prog = assemble(
- "
- mov32 r1, 1
- mov32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_mul32_imm() {
- let prog = assemble(
- "
- mov r0, 3
- mul32 r0, 4
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xc);
- }
- #[test]
- fn test_vm_mul32_reg() {
- let prog = assemble(
- "
- mov r0, 3
- mov r1, 4
- mul32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xc);
- }
- #[test]
- fn test_vm_mul32_reg_overflow() {
- let prog = assemble(
- "
- mov r0, 0x40000001
- mov r1, 4
- mul32 r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x4);
- }
- #[test]
- fn test_vm_mul64_imm() {
- let prog = assemble(
- "
- mov r0, 0x40000001
- mul r0, 4
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x100000004);
- }
- #[test]
- fn test_vm_mul64_reg() {
- let prog = assemble(
- "
- mov r0, 0x40000001
- mov r1, 4
- mul r0, r1
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x100000004);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x75db9c97);
- }
- #[test]
- fn test_vm_neg64() {
- let prog = assemble(
- "
- mov32 r0, 2
- neg r0
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xfffffffffffffffe);
- }
- #[test]
- fn test_vm_neg() {
- let prog = assemble(
- "
- mov32 r0, 2
- neg32 r0
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xfffffffe);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_rhs32() {
- let prog = assemble(
- "
- xor r0, r0
- sub r0, 1
- rsh32 r0, 8
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x00ffffff);
- }
- #[test]
- fn test_vm_rsh_reg() {
- let prog = assemble(
- "
- mov r0, 0x10
- mov r7, 4
- rsh r0, r7
- exit",
- )
- .unwrap();
- let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program().unwrap(), 0xcd);
- }
- #[test]
- fn test_vm_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_program().unwrap(), 0x01020304);
- }
- #[test]
- fn test_vm_stb() {
- let prog = assemble(
- "
- stb [r1+2], 0x11
- ldxb r0, [r1+2]
- exit",
- )
- .unwrap();
- let mem = &mut [0xaa, 0xbb, 0xff, 0xcc, 0xdd];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x11);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x44332211);
- }
- // If this case is not handled properly in check_mem(), then we may overflow when adding the
- // context address and the offset, and make the thread panic with "attempt to add with overflow".
- // Check that we panic with the expected out-of-bounds error.
- //
- // The new toolchain introduced `assert_unsafe_precondition` which panics with a different message and can't be
- // caught by `#[should_panic]`. This is why we use `#[ignore]` here.
- #[test]
- #[should_panic(expected = "Error: out of bounds memory store (insn #1)")]
- #[ignore]
- fn test_vm_stdw_add_overflow() {
- let prog = assemble(
- "
- stdw [r2-0x1], 0x44332211
- ldxw r0, [r1+2]
- exit",
- )
- .unwrap();
- let mem = &mut [
- 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd,
- ];
- let mut vm = rbpf::EbpfVmFixedMbuff::new(Some(&prog), 0x00, 0x10).unwrap();
- _ = vm.execute_program(mem).unwrap();
- }
- #[test]
- fn test_vm_sth() {
- let prog = assemble(
- "
- sth [r1+2], 0x2211
- ldxh r0, [r1+2]
- exit",
- )
- .unwrap();
- let mem = &mut [0xaa, 0xbb, 0xff, 0xff, 0xcc, 0xdd];
- let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x2211);
- }
- #[test]
- fn test_vm_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_program().unwrap(), 0x0);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x44332211);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x11);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0xf0f2f3f4f5f6f7f8);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0xf1f9);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x2a);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x8877665544332211);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x2211);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(mem).unwrap(), 0x44332211);
- }
- #[test]
- fn test_vm_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 vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
- assert_eq!(vm.execute_program(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_vm_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_program(mem).unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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_program(mem).unwrap(), 0x0);
- }
- #[test]
- fn test_vm_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_program(mem).unwrap(), 0x0);
- }
- #[test]
- fn test_vm_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_program(mem).unwrap(), 0x0);
- }
- #[test]
- fn test_vm_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_program(mem.as_mut_slice()).unwrap(), 0x1);
- }
- #[test]
- fn test_vm_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_program(mem.as_mut_slice()).unwrap(), 0x0);
- }
|