123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670 |
- // 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.
- #[test]
- #[should_panic(expected = "Error: out of bounds memory store (insn #1)")]
- 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);
- }
|