ubpf_vm.rs 58 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358
  1. // SPDX-License-Identifier: (Apache-2.0 OR MIT)
  2. // Converted from the tests for uBPF <https://github.com/iovisor/ubpf>
  3. // Copyright 2015 Big Switch Networks, Inc
  4. // Copyright 2016 6WIND S.A. <quentin.monnet@6wind.com>
  5. // The tests contained in this file are extracted from the unit tests of uBPF software. Each test
  6. // in this file has a name in the form `test_vm_<name>`, and corresponds to the (human-readable)
  7. // code in `ubpf/tree/master/tests/<name>`, available at
  8. // <https://github.com/iovisor/ubpf/tree/master/tests> (hyphen had to be replaced with underscores
  9. // as Rust will not accept them in function names). It is strongly advised to refer to the uBPF
  10. // version to understand what these program do.
  11. //
  12. // Each program was assembled from the uBPF version with the assembler provided by uBPF itself, and
  13. // available at <https://github.com/iovisor/ubpf/tree/master/ubpf>.
  14. // The very few modifications that have been realized should be indicated.
  15. // These are unit tests for the eBPF interpreter.
  16. #![cfg_attr(feature = "cargo-clippy", allow(clippy::unreadable_literal))]
  17. extern crate rbpf;
  18. mod common;
  19. use rbpf::helpers;
  20. use rbpf::assembler::assemble;
  21. use common::{TCP_SACK_ASM, TCP_SACK_MATCH, TCP_SACK_NOMATCH};
  22. #[test]
  23. fn test_vm_add() {
  24. let prog = assemble("
  25. mov32 r0, 0
  26. mov32 r1, 2
  27. add32 r0, 1
  28. add32 r0, r1
  29. exit").unwrap();
  30. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  31. assert_eq!(vm.execute_program().unwrap(), 0x3);
  32. }
  33. #[test]
  34. fn test_vm_alu64_arith() {
  35. let prog = assemble("
  36. mov r0, 0
  37. mov r1, 1
  38. mov r2, 2
  39. mov r3, 3
  40. mov r4, 4
  41. mov r5, 5
  42. mov r6, 6
  43. mov r7, 7
  44. mov r8, 8
  45. mov r9, 9
  46. add r0, 23
  47. add r0, r7
  48. sub r0, 13
  49. sub r0, r1
  50. mul r0, 7
  51. mul r0, r3
  52. div r0, 2
  53. div r0, r4
  54. exit").unwrap();
  55. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  56. assert_eq!(vm.execute_program().unwrap(), 0x2a);
  57. }
  58. #[test]
  59. fn test_vm_alu64_bit() {
  60. let prog = assemble("
  61. mov r0, 0
  62. mov r1, 1
  63. mov r2, 2
  64. mov r3, 3
  65. mov r4, 4
  66. mov r5, 5
  67. mov r6, 6
  68. mov r7, 7
  69. mov r8, 8
  70. or r0, r5
  71. or r0, 0xa0
  72. and r0, 0xa3
  73. mov r9, 0x91
  74. and r0, r9
  75. lsh r0, 32
  76. lsh r0, 22
  77. lsh r0, r8
  78. rsh r0, 32
  79. rsh r0, 19
  80. rsh r0, r7
  81. xor r0, 0x03
  82. xor r0, r2
  83. exit").unwrap();
  84. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  85. assert_eq!(vm.execute_program().unwrap(), 0x11);
  86. }
  87. #[test]
  88. fn test_vm_alu_arith() {
  89. let prog = assemble("
  90. mov32 r0, 0
  91. mov32 r1, 1
  92. mov32 r2, 2
  93. mov32 r3, 3
  94. mov32 r4, 4
  95. mov32 r5, 5
  96. mov32 r6, 6
  97. mov32 r7, 7
  98. mov32 r8, 8
  99. mov32 r9, 9
  100. add32 r0, 23
  101. add32 r0, r7
  102. sub32 r0, 13
  103. sub32 r0, r1
  104. mul32 r0, 7
  105. mul32 r0, r3
  106. div32 r0, 2
  107. div32 r0, r4
  108. exit").unwrap();
  109. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  110. assert_eq!(vm.execute_program().unwrap(), 0x2a);
  111. }
  112. #[test]
  113. fn test_vm_alu_bit() {
  114. let prog = assemble("
  115. mov32 r0, 0
  116. mov32 r1, 1
  117. mov32 r2, 2
  118. mov32 r3, 3
  119. mov32 r4, 4
  120. mov32 r5, 5
  121. mov32 r6, 6
  122. mov32 r7, 7
  123. mov32 r8, 8
  124. or32 r0, r5
  125. or32 r0, 0xa0
  126. and32 r0, 0xa3
  127. mov32 r9, 0x91
  128. and32 r0, r9
  129. lsh32 r0, 22
  130. lsh32 r0, r8
  131. rsh32 r0, 19
  132. rsh32 r0, r7
  133. xor32 r0, 0x03
  134. xor32 r0, r2
  135. exit").unwrap();
  136. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  137. assert_eq!(vm.execute_program().unwrap(), 0x11);
  138. }
  139. #[test]
  140. fn test_vm_arsh32_high_shift() {
  141. let prog = assemble("
  142. mov r0, 8
  143. lddw r1, 0x100000001
  144. arsh32 r0, r1
  145. exit").unwrap();
  146. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  147. assert_eq!(vm.execute_program().unwrap(), 0x4);
  148. }
  149. #[test]
  150. fn test_vm_arsh() {
  151. let prog = assemble("
  152. mov32 r0, 0xf8
  153. lsh32 r0, 28
  154. arsh32 r0, 16
  155. exit").unwrap();
  156. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  157. assert_eq!(vm.execute_program().unwrap(), 0xffff8000);
  158. }
  159. #[test]
  160. fn test_vm_arsh64() {
  161. let prog = assemble("
  162. mov32 r0, 1
  163. lsh r0, 63
  164. arsh r0, 55
  165. mov32 r1, 5
  166. arsh r0, r1
  167. exit").unwrap();
  168. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  169. assert_eq!(vm.execute_program().unwrap(), 0xfffffffffffffff8);
  170. }
  171. #[test]
  172. fn test_vm_arsh_reg() {
  173. let prog = assemble("
  174. mov32 r0, 0xf8
  175. mov32 r1, 16
  176. lsh32 r0, 28
  177. arsh32 r0, r1
  178. exit").unwrap();
  179. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  180. assert_eq!(vm.execute_program().unwrap(), 0xffff8000);
  181. }
  182. #[test]
  183. fn test_vm_arsh_imm_overflow() {
  184. let prog = assemble("
  185. mov r0, 1
  186. lsh r0, 63
  187. arsh r0, 0xff20
  188. exit").unwrap();
  189. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  190. assert_eq!(vm.execute_program().unwrap(), 0xffffffff80000000);
  191. }
  192. #[test]
  193. fn test_vm_arsh_reg_overflow() {
  194. let prog = assemble("
  195. mov r0, 1
  196. lsh r0, 63
  197. mov r1, 0xff04
  198. arsh r0, r1
  199. exit").unwrap();
  200. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  201. assert_eq!(vm.execute_program().unwrap(), 0xf800000000000000);
  202. }
  203. #[test]
  204. fn test_vm_arsh32_imm_overflow() {
  205. let prog = assemble("
  206. mov32 r0, 1
  207. lsh32 r0, 31
  208. arsh32 r0, 0xff10
  209. exit").unwrap();
  210. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  211. assert_eq!(vm.execute_program().unwrap(), 0xffff8000);
  212. }
  213. #[test]
  214. fn test_vm_arsh32_reg_overflow() {
  215. let prog = assemble("
  216. mov32 r0, 1
  217. lsh32 r0, 31
  218. mov32 r1, 32
  219. arsh32 r0, r1
  220. exit").unwrap();
  221. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  222. assert_eq!(vm.execute_program().unwrap(), 0x80000000);
  223. }
  224. #[test]
  225. fn test_vm_be16() {
  226. let prog = assemble("
  227. ldxh r0, [r1]
  228. be16 r0
  229. exit").unwrap();
  230. let mem = &mut [
  231. 0x11, 0x22
  232. ];
  233. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  234. assert_eq!(vm.execute_program(mem).unwrap(), 0x1122);
  235. }
  236. #[test]
  237. fn test_vm_be16_high() {
  238. let prog = assemble("
  239. ldxdw r0, [r1]
  240. be16 r0
  241. exit").unwrap();
  242. let mem = &mut [
  243. 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88
  244. ];
  245. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  246. assert_eq!(vm.execute_program(mem).unwrap(), 0x1122);
  247. }
  248. #[test]
  249. fn test_vm_be32() {
  250. let prog = assemble("
  251. ldxw r0, [r1]
  252. be32 r0
  253. exit").unwrap();
  254. let mem = &mut [
  255. 0x11, 0x22, 0x33, 0x44
  256. ];
  257. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  258. assert_eq!(vm.execute_program(mem).unwrap(), 0x11223344);
  259. }
  260. #[test]
  261. fn test_vm_be32_high() {
  262. let prog = assemble("
  263. ldxdw r0, [r1]
  264. be32 r0
  265. exit").unwrap();
  266. let mem = &mut [
  267. 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88
  268. ];
  269. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  270. assert_eq!(vm.execute_program(mem).unwrap(), 0x11223344);
  271. }
  272. #[test]
  273. fn test_vm_be64() {
  274. let prog = assemble("
  275. ldxdw r0, [r1]
  276. be64 r0
  277. exit").unwrap();
  278. let mem = &mut [
  279. 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88
  280. ];
  281. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  282. assert_eq!(vm.execute_program(mem).unwrap(), 0x1122334455667788);
  283. }
  284. #[test]
  285. fn test_vm_call() {
  286. let prog = assemble("
  287. mov r1, 1
  288. mov r2, 2
  289. mov r3, 3
  290. mov r4, 4
  291. mov r5, 5
  292. call 0
  293. exit").unwrap();
  294. let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  295. vm.register_helper(0, helpers::gather_bytes).unwrap();
  296. assert_eq!(vm.execute_program().unwrap(), 0x0102030405);
  297. }
  298. #[test]
  299. fn test_vm_call_memfrob() {
  300. let prog = assemble("
  301. mov r6, r1
  302. add r1, 2
  303. mov r2, 4
  304. call 1
  305. ldxdw r0, [r6]
  306. be64 r0
  307. exit").unwrap();
  308. let mem = &mut [
  309. 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08
  310. ];
  311. let mut vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  312. vm.register_helper(1, helpers::memfrob).unwrap();
  313. assert_eq!(vm.execute_program(mem).unwrap(), 0x102292e2f2c0708);
  314. }
  315. // TODO: helpers::trash_registers needs asm!().
  316. // Try this again once asm!() is available in stable.
  317. //#[test]
  318. //fn test_vm_call_save() {
  319. //let prog = &[
  320. //0xb7, 0x06, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
  321. //0xb7, 0x07, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00,
  322. //0xb7, 0x08, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00,
  323. //0xb7, 0x09, 0x00, 0x00, 0x00, 0x40, 0x00, 0x00,
  324. //0x85, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00,
  325. //0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  326. //0x4f, 0x60, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  327. //0x4f, 0x70, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  328. //0x4f, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  329. //0x4f, 0x90, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  330. //0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
  331. //];
  332. //let mut vm = rbpf::EbpfVmNoData::new(Some(prog)).unwrap();
  333. //vm.register_helper(2, helpers::trash_registers);
  334. //assert_eq!(vm.execute_program().unwrap(), 0x4321);
  335. //}
  336. #[test]
  337. fn test_vm_div32_high_divisor() {
  338. let prog = assemble("
  339. mov r0, 12
  340. lddw r1, 0x100000004
  341. div32 r0, r1
  342. exit").unwrap();
  343. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  344. assert_eq!(vm.execute_program().unwrap(), 0x3);
  345. }
  346. #[test]
  347. fn test_vm_div32_imm() {
  348. let prog = assemble("
  349. lddw r0, 0x10000000c
  350. div32 r0, 4
  351. exit").unwrap();
  352. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  353. assert_eq!(vm.execute_program().unwrap(), 0x3);
  354. }
  355. #[test]
  356. fn test_vm_div32_reg() {
  357. let prog = assemble("
  358. lddw r0, 0x10000000c
  359. mov r1, 4
  360. div32 r0, r1
  361. exit").unwrap();
  362. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  363. assert_eq!(vm.execute_program().unwrap(), 0x3);
  364. }
  365. #[test]
  366. fn test_vm_div64_imm() {
  367. let prog = assemble("
  368. mov r0, 0xc
  369. lsh r0, 32
  370. div r0, 4
  371. exit").unwrap();
  372. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  373. assert_eq!(vm.execute_program().unwrap(), 0x300000000);
  374. }
  375. #[test]
  376. fn test_vm_div64_reg() {
  377. let prog = assemble("
  378. mov r0, 0xc
  379. lsh r0, 32
  380. mov r1, 4
  381. div r0, r1
  382. exit").unwrap();
  383. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  384. assert_eq!(vm.execute_program().unwrap(), 0x300000000);
  385. }
  386. #[test]
  387. fn test_vm_early_exit() {
  388. let prog = assemble("
  389. mov r0, 3
  390. exit
  391. mov r0, 4
  392. exit").unwrap();
  393. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  394. assert_eq!(vm.execute_program().unwrap(), 0x3);
  395. }
  396. // uBPF limits the number of user functions at 64. We don't.
  397. //#[test]
  398. //fn test_vm_err_call_bad_imm() {
  399. //}
  400. #[test]
  401. #[should_panic(expected = "Error: unknown helper function (id: 0x3f)")]
  402. fn test_vm_err_call_unreg() {
  403. let prog = assemble("
  404. mov r1, 1
  405. mov r2, 2
  406. mov r3, 3
  407. mov r4, 4
  408. mov r5, 5
  409. call 63
  410. exit").unwrap();
  411. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  412. vm.execute_program().unwrap();
  413. }
  414. #[test]
  415. fn test_vm_div64_by_zero_imm() {
  416. let prog = assemble("
  417. mov32 r0, 1
  418. div r0, 0
  419. exit").unwrap();
  420. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  421. assert_eq!(vm.execute_program().unwrap(), 0x0);
  422. }
  423. #[test]
  424. fn test_vm_div_by_zero_imm() {
  425. let prog = assemble("
  426. mov32 r0, 1
  427. div32 r0, 0
  428. exit").unwrap();
  429. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  430. assert_eq!(vm.execute_program().unwrap(), 0x0);
  431. }
  432. #[test]
  433. fn test_vm_mod64_by_zero_imm() {
  434. let prog = assemble("
  435. mov32 r0, 1
  436. mod r0, 0
  437. exit").unwrap();
  438. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  439. assert_eq!(vm.execute_program().unwrap(), 0x1);
  440. }
  441. #[test]
  442. fn test_vm_mod_by_zero_imm() {
  443. let prog = assemble("
  444. mov32 r0, 1
  445. mod32 r0, 0
  446. exit").unwrap();
  447. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  448. assert_eq!(vm.execute_program().unwrap(), 0x1);
  449. }
  450. // Make sure we only consider the last 32 bits of the divisor.
  451. #[test]
  452. fn test_vm_mod_by_zero_reg_long() {
  453. let prog = assemble("
  454. lddw r1, 0x100000000
  455. mod32 r0, r1
  456. exit").unwrap();
  457. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  458. assert_eq!(vm.execute_program().unwrap(), 0x0);
  459. }
  460. #[test]
  461. fn test_vm_div64_by_zero_reg() {
  462. let prog = assemble("
  463. mov32 r0, 1
  464. mov32 r1, 0
  465. div r0, r1
  466. exit").unwrap();
  467. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  468. assert_eq!(vm.execute_program().unwrap(), 0x0);
  469. }
  470. #[test]
  471. fn test_vm_div_by_zero_reg() {
  472. let prog = assemble("
  473. mov32 r0, 1
  474. mov32 r1, 0
  475. div32 r0, r1
  476. exit").unwrap();
  477. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  478. assert_eq!(vm.execute_program().unwrap(), 0x0);
  479. }
  480. // Make sure we only consider the last 32 bits of the divisor.
  481. #[test]
  482. fn test_vm_div_by_zero_reg_long() {
  483. let prog = assemble("
  484. lddw r1, 0x100000000
  485. div32 r0, r1
  486. exit").unwrap();
  487. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  488. assert_eq!(vm.execute_program().unwrap(), 0x0);
  489. }
  490. #[test]
  491. fn test_vm_mod64_by_zero_reg() {
  492. let prog = assemble("
  493. mov32 r0, 1
  494. mov32 r1, 0
  495. mod r0, r1
  496. exit").unwrap();
  497. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  498. assert_eq!(vm.execute_program().unwrap(), 0x1);
  499. }
  500. #[test]
  501. fn test_vm_mod_by_zero_reg() {
  502. let prog = assemble("
  503. mov32 r0, 1
  504. mov32 r1, 0
  505. mod32 r0, r1
  506. exit").unwrap();
  507. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  508. assert_eq!(vm.execute_program().unwrap(), 0x1);
  509. }
  510. #[test]
  511. #[should_panic(expected = "Error: out of bounds memory store (insn #1)")]
  512. fn test_vm_err_stack_out_of_bound() {
  513. let prog = assemble("
  514. stb [r10], 0
  515. exit").unwrap();
  516. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  517. vm.execute_program().unwrap();
  518. }
  519. #[test]
  520. fn test_vm_exit() {
  521. let prog = assemble("
  522. mov r0, 0
  523. exit").unwrap();
  524. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  525. assert_eq!(vm.execute_program().unwrap(), 0x0);
  526. }
  527. #[test]
  528. fn test_vm_ja() {
  529. let prog = assemble("
  530. mov r0, 1
  531. ja +1
  532. mov r0, 2
  533. exit").unwrap();
  534. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  535. assert_eq!(vm.execute_program().unwrap(), 0x1);
  536. }
  537. #[test]
  538. fn test_vm_jeq_imm() {
  539. let prog = assemble("
  540. mov32 r0, 0
  541. mov32 r1, 0xa
  542. jeq r1, 0xb, +4
  543. mov32 r0, 1
  544. mov32 r1, 0xb
  545. jeq r1, 0xb, +1
  546. mov32 r0, 2
  547. exit").unwrap();
  548. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  549. assert_eq!(vm.execute_program().unwrap(), 0x1);
  550. }
  551. #[test]
  552. fn test_vm_jeq_reg() {
  553. let prog = assemble("
  554. mov32 r0, 0
  555. mov32 r1, 0xa
  556. mov32 r2, 0xb
  557. jeq r1, r2, +4
  558. mov32 r0, 1
  559. mov32 r1, 0xb
  560. jeq r1, r2, +1
  561. mov32 r0, 2
  562. exit").unwrap();
  563. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  564. assert_eq!(vm.execute_program().unwrap(), 0x1);
  565. }
  566. #[test]
  567. fn test_vm_jge_imm() {
  568. let prog = assemble("
  569. mov32 r0, 0
  570. mov32 r1, 0xa
  571. jge r1, 0xb, +4
  572. mov32 r0, 1
  573. mov32 r1, 0xc
  574. jge r1, 0xb, +1
  575. mov32 r0, 2
  576. exit").unwrap();
  577. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  578. assert_eq!(vm.execute_program().unwrap(), 0x1);
  579. }
  580. #[test]
  581. fn test_vm_jle_imm() {
  582. let prog = assemble("
  583. mov32 r0, 0
  584. mov32 r1, 5
  585. jle r1, 4, +1
  586. jle r1, 6, +1
  587. exit
  588. jle r1, 5, +1
  589. exit
  590. mov32 r0, 1
  591. exit").unwrap();
  592. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  593. assert_eq!(vm.execute_program().unwrap(), 0x1);
  594. }
  595. #[test]
  596. fn test_vm_jle_reg() {
  597. let prog = assemble("
  598. mov r0, 0
  599. mov r1, 5
  600. mov r2, 4
  601. mov r3, 6
  602. jle r1, r2, +2
  603. jle r1, r1, +1
  604. exit
  605. jle r1, r3, +1
  606. exit
  607. mov r0, 1
  608. exit").unwrap();
  609. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  610. assert_eq!(vm.execute_program().unwrap(), 0x1);
  611. }
  612. #[test]
  613. fn test_vm_jgt_imm() {
  614. let prog = assemble("
  615. mov32 r0, 0
  616. mov32 r1, 5
  617. jgt r1, 6, +2
  618. jgt r1, 5, +1
  619. jgt r1, 4, +1
  620. exit
  621. mov32 r0, 1
  622. exit").unwrap();
  623. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  624. assert_eq!(vm.execute_program().unwrap(), 0x1);
  625. }
  626. #[test]
  627. fn test_vm_jgt_reg() {
  628. let prog = assemble("
  629. mov r0, 0
  630. mov r1, 5
  631. mov r2, 6
  632. mov r3, 4
  633. jgt r1, r2, +2
  634. jgt r1, r1, +1
  635. jgt r1, r3, +1
  636. exit
  637. mov r0, 1
  638. exit").unwrap();
  639. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  640. assert_eq!(vm.execute_program().unwrap(), 0x1);
  641. }
  642. #[test]
  643. fn test_vm_jlt_imm() {
  644. let prog = assemble("
  645. mov32 r0, 0
  646. mov32 r1, 5
  647. jlt r1, 4, +2
  648. jlt r1, 5, +1
  649. jlt r1, 6, +1
  650. exit
  651. mov32 r0, 1
  652. exit").unwrap();
  653. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  654. assert_eq!(vm.execute_program().unwrap(), 0x1);
  655. }
  656. #[test]
  657. fn test_vm_jlt_reg() {
  658. let prog = assemble("
  659. mov r0, 0
  660. mov r1, 5
  661. mov r2, 4
  662. mov r3, 6
  663. jlt r1, r2, +2
  664. jlt r1, r1, +1
  665. jlt r1, r3, +1
  666. exit
  667. mov r0, 1
  668. exit").unwrap();
  669. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  670. assert_eq!(vm.execute_program().unwrap(), 0x1);
  671. }
  672. #[test]
  673. fn test_vm_jit_bounce() {
  674. let prog = assemble("
  675. mov r0, 1
  676. mov r6, r0
  677. mov r7, r6
  678. mov r8, r7
  679. mov r9, r8
  680. mov r0, r9
  681. exit").unwrap();
  682. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  683. assert_eq!(vm.execute_program().unwrap(), 0x1);
  684. }
  685. #[test]
  686. fn test_vm_jne_reg() {
  687. let prog = assemble("
  688. mov32 r0, 0
  689. mov32 r1, 0xb
  690. mov32 r2, 0xb
  691. jne r1, r2, +4
  692. mov32 r0, 1
  693. mov32 r1, 0xa
  694. jne r1, r2, +1
  695. mov32 r0, 2
  696. exit").unwrap();
  697. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  698. assert_eq!(vm.execute_program().unwrap(), 0x1);
  699. }
  700. #[test]
  701. fn test_vm_jset_imm() {
  702. let prog = assemble("
  703. mov32 r0, 0
  704. mov32 r1, 0x7
  705. jset r1, 0x8, +4
  706. mov32 r0, 1
  707. mov32 r1, 0x9
  708. jset r1, 0x8, +1
  709. mov32 r0, 2
  710. exit").unwrap();
  711. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  712. assert_eq!(vm.execute_program().unwrap(), 0x1);
  713. }
  714. #[test]
  715. fn test_vm_jset_reg() {
  716. let prog = assemble("
  717. mov32 r0, 0
  718. mov32 r1, 0x7
  719. mov32 r2, 0x8
  720. jset r1, r2, +4
  721. mov32 r0, 1
  722. mov32 r1, 0x9
  723. jset r1, r2, +1
  724. mov32 r0, 2
  725. exit").unwrap();
  726. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  727. assert_eq!(vm.execute_program().unwrap(), 0x1);
  728. }
  729. #[test]
  730. fn test_vm_jsge_imm() {
  731. let prog = assemble("
  732. mov32 r0, 0
  733. mov r1, -2
  734. jsge r1, -1, +5
  735. jsge r1, 0, +4
  736. mov32 r0, 1
  737. mov r1, -1
  738. jsge r1, -1, +1
  739. mov32 r0, 2
  740. exit").unwrap();
  741. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  742. assert_eq!(vm.execute_program().unwrap(), 0x1);
  743. }
  744. #[test]
  745. fn test_vm_jsge_reg() {
  746. let prog = assemble("
  747. mov32 r0, 0
  748. mov r1, -2
  749. mov r2, -1
  750. mov32 r3, 0
  751. jsge r1, r2, +5
  752. jsge r1, r3, +4
  753. mov32 r0, 1
  754. mov r1, r2
  755. jsge r1, r2, +1
  756. mov32 r0, 2
  757. exit").unwrap();
  758. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  759. assert_eq!(vm.execute_program().unwrap(), 0x1);
  760. }
  761. #[test]
  762. fn test_vm_jsle_imm() {
  763. let prog = assemble("
  764. mov32 r0, 0
  765. mov r1, -2
  766. jsle r1, -3, +1
  767. jsle r1, -1, +1
  768. exit
  769. mov32 r0, 1
  770. jsle r1, -2, +1
  771. mov32 r0, 2
  772. exit").unwrap();
  773. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  774. assert_eq!(vm.execute_program().unwrap(), 0x1);
  775. }
  776. #[test]
  777. fn test_vm_jsle_reg() {
  778. let prog = assemble("
  779. mov32 r0, 0
  780. mov r1, -1
  781. mov r2, -2
  782. mov32 r3, 0
  783. jsle r1, r2, +1
  784. jsle r1, r3, +1
  785. exit
  786. mov32 r0, 1
  787. mov r1, r2
  788. jsle r1, r2, +1
  789. mov32 r0, 2
  790. exit").unwrap();
  791. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  792. assert_eq!(vm.execute_program().unwrap(), 0x1);
  793. }
  794. #[test]
  795. fn test_vm_jsgt_imm() {
  796. let prog = assemble("
  797. mov32 r0, 0
  798. mov r1, -2
  799. jsgt r1, -1, +4
  800. mov32 r0, 1
  801. mov32 r1, 0
  802. jsgt r1, -1, +1
  803. mov32 r0, 2
  804. exit").unwrap();
  805. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  806. assert_eq!(vm.execute_program().unwrap(), 0x1);
  807. }
  808. #[test]
  809. fn test_vm_jsgt_reg() {
  810. let prog = assemble("
  811. mov32 r0, 0
  812. mov r1, -2
  813. mov r2, -1
  814. jsgt r1, r2, +4
  815. mov32 r0, 1
  816. mov32 r1, 0
  817. jsgt r1, r2, +1
  818. mov32 r0, 2
  819. exit").unwrap();
  820. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  821. assert_eq!(vm.execute_program().unwrap(), 0x1);
  822. }
  823. #[test]
  824. fn test_vm_jslt_imm() {
  825. let prog = assemble("
  826. mov32 r0, 0
  827. mov r1, -2
  828. jslt r1, -3, +2
  829. jslt r1, -2, +1
  830. jslt r1, -1, +1
  831. exit
  832. mov32 r0, 1
  833. exit").unwrap();
  834. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  835. assert_eq!(vm.execute_program().unwrap(), 0x1);
  836. }
  837. #[test]
  838. fn test_vm_jslt_reg() {
  839. let prog = assemble("
  840. mov32 r0, 0
  841. mov r1, -2
  842. mov r2, -3
  843. mov r3, -1
  844. jslt r1, r1, +2
  845. jslt r1, r2, +1
  846. jslt r1, r3, +1
  847. exit
  848. mov32 r0, 1
  849. exit").unwrap();
  850. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  851. assert_eq!(vm.execute_program().unwrap(), 0x1);
  852. }
  853. #[test]
  854. fn test_vm_jeq32_imm() {
  855. let prog = assemble("
  856. mov r9, 1
  857. lsh r9, 32
  858. mov32 r0, 0x0
  859. mov32 r1, 0xa
  860. jeq32 r1, 0xb, +5
  861. mov32 r0, 1
  862. mov r1, 0xb
  863. or r1, r9
  864. jeq32 r1, 0xb, +1
  865. mov32 r0, 2
  866. exit").unwrap();
  867. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  868. assert_eq!(vm.execute_program().unwrap(), 0x1);
  869. }
  870. #[test]
  871. fn test_vm_jeq32_reg() {
  872. let prog = assemble("
  873. mov r9, 1
  874. lsh r9, 32
  875. mov32 r0, 0
  876. mov32 r1, 0xa
  877. mov32 r2, 0xb
  878. jeq32 r1, r2, +5
  879. mov32 r0, 1
  880. mov32 r1, 0xb
  881. or r1, r9
  882. jeq32 r1, r2, +1
  883. mov32 r0, 2
  884. exit").unwrap();
  885. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  886. assert_eq!(vm.execute_program().unwrap(), 0x1);
  887. }
  888. #[test]
  889. fn test_vm_jge32_imm() {
  890. let prog = assemble("
  891. mov r9, 1
  892. lsh r9, 32
  893. mov32 r0, 0
  894. mov32 r1, 0xa
  895. jge32 r1, 0xb, +5
  896. mov32 r0, 1
  897. or r1, r9
  898. mov32 r1, 0xc
  899. jge32 r1, 0xb, +1
  900. mov32 r0, 2
  901. exit").unwrap();
  902. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  903. assert_eq!(vm.execute_program().unwrap(), 0x1);
  904. }
  905. #[test]
  906. fn test_vm_jge32_reg() {
  907. let prog = assemble("
  908. mov r9, 1
  909. lsh r9, 32
  910. mov32 r0, 0
  911. mov32 r1, 0xa
  912. mov32 r2, 0xb
  913. jge32 r1, r2, +5
  914. mov32 r0, 1
  915. or r1, r9
  916. mov32 r1, 0xc
  917. jge32 r1, r2, +1
  918. mov32 r0, 2
  919. exit").unwrap();
  920. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  921. assert_eq!(vm.execute_program().unwrap(), 0x1);
  922. }
  923. #[test]
  924. fn test_vm_jgt32_imm() {
  925. let prog = assemble("
  926. mov r9, 1
  927. lsh r9, 32
  928. mov32 r0, 0
  929. mov32 r1, 5
  930. or r1, r9
  931. jgt32 r1, 6, +4
  932. jgt32 r1, 5, +3
  933. jgt32 r1, 4, +1
  934. exit
  935. mov32 r0, 1
  936. exit").unwrap();
  937. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  938. assert_eq!(vm.execute_program().unwrap(), 0x1);
  939. }
  940. #[test]
  941. fn test_vm_jgt32_reg() {
  942. let prog = assemble("
  943. mov r9, 1
  944. lsh r9, 32
  945. mov r0, 0
  946. mov r1, 5
  947. mov32 r1, 5
  948. or r1, r9
  949. mov r2, 6
  950. mov r3, 4
  951. jgt32 r1, r2, +4
  952. jgt32 r1, r1, +3
  953. jgt32 r1, r3, +1
  954. exit
  955. mov r0, 1
  956. exit").unwrap();
  957. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  958. assert_eq!(vm.execute_program().unwrap(), 0x1);
  959. }
  960. #[test]
  961. fn test_vm_jle32_imm() {
  962. let prog = assemble("
  963. mov r9, 1
  964. lsh r9, 32
  965. mov32 r0, 0
  966. mov32 r1, 5
  967. or r1, r9
  968. jle32 r1, 4, +5
  969. jle32 r1, 6, +1
  970. exit
  971. jle32 r1, 5, +1
  972. exit
  973. mov32 r0, 1
  974. exit").unwrap();
  975. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  976. assert_eq!(vm.execute_program().unwrap(), 0x1);
  977. }
  978. #[test]
  979. fn test_vm_jle32_reg() {
  980. let prog = assemble("
  981. mov r9, 1
  982. lsh r9, 32
  983. mov r0, 0
  984. mov r1, 5
  985. mov r2, 4
  986. mov r3, 6
  987. or r1, r9
  988. jle32 r1, r2, +5
  989. jle32 r1, r1, +1
  990. exit
  991. jle32 r1, r3, +1
  992. exit
  993. mov r0, 1
  994. exit").unwrap();
  995. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  996. assert_eq!(vm.execute_program().unwrap(), 0x1);
  997. }
  998. #[test]
  999. fn test_vm_jlt32_imm() {
  1000. let prog = assemble("
  1001. mov r9, 1
  1002. lsh r9, 32
  1003. mov32 r0, 0
  1004. mov32 r1, 5
  1005. or r1, r9
  1006. jlt32 r1, 4, +4
  1007. jlt32 r1, 5, +3
  1008. jlt32 r1, 6, +1
  1009. exit
  1010. mov32 r0, 1
  1011. exit").unwrap();
  1012. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1013. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1014. }
  1015. #[test]
  1016. fn test_vm_jlt32_reg() {
  1017. let prog = assemble("
  1018. mov r9, 1
  1019. lsh r9, 32
  1020. mov r0, 0
  1021. mov r1, 5
  1022. mov r2, 4
  1023. mov r3, 6
  1024. or r1, r9
  1025. jlt32 r1, r2, +4
  1026. jlt32 r1, r1, +3
  1027. jlt32 r1, r3, +1
  1028. exit
  1029. mov r0, 1
  1030. exit").unwrap();
  1031. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1032. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1033. }
  1034. #[test]
  1035. fn test_vm_jne32_imm() {
  1036. let prog = assemble("
  1037. mov r9, 1
  1038. lsh r9, 32
  1039. mov32 r0, 0
  1040. mov32 r1, 0xb
  1041. or r1, r9
  1042. jne32 r1, 0xb, +4
  1043. mov32 r0, 1
  1044. mov32 r1, 0xa
  1045. or r1, r9
  1046. jne32 r1, 0xb, +1
  1047. mov32 r0, 2
  1048. exit").unwrap();
  1049. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1050. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1051. }
  1052. #[test]
  1053. fn test_vm_jne32_reg() {
  1054. let prog = assemble("
  1055. mov r9, 1
  1056. lsh r9, 32
  1057. mov32 r0, 0
  1058. mov32 r1, 0xb
  1059. or r1, r9
  1060. mov32 r2, 0xb
  1061. jne32 r1, r2, +4
  1062. mov32 r0, 1
  1063. mov32 r1, 0xa
  1064. or r1, r9
  1065. jne32 r1, r2, +1
  1066. mov32 r0, 2
  1067. exit").unwrap();
  1068. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1069. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1070. }
  1071. #[test]
  1072. fn test_vm_jset32_imm() {
  1073. let prog = assemble("
  1074. mov r9, 1
  1075. lsh r9, 32
  1076. mov32 r0, 0
  1077. mov32 r1, 0x7
  1078. or r1, r9
  1079. jset32 r1, 0x8, +4
  1080. mov32 r0, 1
  1081. mov32 r1, 0x9
  1082. jset32 r1, 0x8, +1
  1083. mov32 r0, 2
  1084. exit").unwrap();
  1085. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1086. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1087. }
  1088. #[test]
  1089. fn test_vm_jset32_reg() {
  1090. let prog = assemble("
  1091. mov r9, 1
  1092. lsh r9, 32
  1093. mov32 r0, 0
  1094. mov32 r1, 0x7
  1095. or r1, r9
  1096. mov32 r2, 0x8
  1097. jset32 r1, r2, +4
  1098. mov32 r0, 1
  1099. mov32 r1, 0x9
  1100. jset32 r1, r2, +1
  1101. mov32 r0, 2
  1102. exit").unwrap();
  1103. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1104. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1105. }
  1106. #[test]
  1107. fn test_vm_jsge32_imm() {
  1108. let prog = assemble("
  1109. mov r9, 1
  1110. lsh r9, 32
  1111. mov32 r0, 0
  1112. mov32 r1, -2
  1113. or r1, r9
  1114. jsge32 r1, -1, +5
  1115. jsge32 r1, 0, +4
  1116. mov32 r0, 1
  1117. mov r1, -1
  1118. jsge32 r1, -1, +1
  1119. mov32 r0, 2
  1120. exit").unwrap();
  1121. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1122. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1123. }
  1124. #[test]
  1125. fn test_vm_jsge32_reg() {
  1126. let prog = assemble("
  1127. mov r9, 1
  1128. lsh r9, 32
  1129. mov32 r0, 0
  1130. mov32 r1, -2
  1131. or r1, r9
  1132. mov r2, -1
  1133. mov32 r3, 0
  1134. jsge32 r1, r2, +5
  1135. jsge32 r1, r3, +4
  1136. mov32 r0, 1
  1137. mov r1, r2
  1138. jsge32 r1, r2, +1
  1139. mov32 r0, 2
  1140. exit").unwrap();
  1141. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1142. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1143. }
  1144. #[test]
  1145. fn test_vm_jsgt32_imm() {
  1146. let prog = assemble("
  1147. mov r9, 1
  1148. lsh r9, 32
  1149. mov32 r0, 0
  1150. mov32 r1, -2
  1151. or r1, r9
  1152. jsgt32 r1, -1, +4
  1153. mov32 r0, 1
  1154. mov32 r1, 0
  1155. jsgt32 r1, -1, +1
  1156. mov32 r0, 2
  1157. exit").unwrap();
  1158. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1159. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1160. }
  1161. #[test]
  1162. fn test_vm_jsgt32_reg() {
  1163. let prog = assemble("
  1164. mov r9, 1
  1165. lsh r9, 32
  1166. mov32 r0, 0
  1167. mov32 r1, -2
  1168. or r1, r9
  1169. mov r2, -1
  1170. jsgt32 r1, r2, +4
  1171. mov32 r0, 1
  1172. mov32 r1, 0
  1173. jsgt32 r1, r2, +1
  1174. mov32 r0, 2
  1175. exit").unwrap();
  1176. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1177. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1178. }
  1179. #[test]
  1180. fn test_vm_jsle32_imm() {
  1181. let prog = assemble("
  1182. mov r9, 1
  1183. lsh r9, 32
  1184. mov32 r0, 0
  1185. mov32 r1, -2
  1186. or r1, r9
  1187. jsle32 r1, -3, +5
  1188. jsle32 r1, -1, +1
  1189. exit
  1190. mov32 r0, 1
  1191. jsle32 r1, -2, +1
  1192. mov32 r0, 2
  1193. exit").unwrap();
  1194. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1195. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1196. }
  1197. #[test]
  1198. fn test_vm_jsle32_reg() {
  1199. let prog = assemble("
  1200. mov r9, 1
  1201. lsh r9, 32
  1202. mov32 r0, 0
  1203. mov32 r1, -2
  1204. or r1, r9
  1205. mov r2, -3
  1206. mov32 r3, 0
  1207. jsle32 r1, r2, +6
  1208. jsle32 r1, r3, +1
  1209. exit
  1210. mov32 r0, 1
  1211. mov r1, r2
  1212. jsle32 r1, r2, +1
  1213. mov32 r0, 2
  1214. exit").unwrap();
  1215. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1216. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1217. }
  1218. #[test]
  1219. fn test_vm_jslt32_imm() {
  1220. let prog = assemble("
  1221. mov r9, 1
  1222. lsh r9, 32
  1223. mov32 r0, 0
  1224. mov32 r1, -2
  1225. or r1, r9
  1226. jslt32 r1, -3, +4
  1227. jslt32 r1, -2, +3
  1228. jslt32 r1, -1, +1
  1229. exit
  1230. mov32 r0, 1
  1231. exit").unwrap();
  1232. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1233. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1234. }
  1235. #[test]
  1236. fn test_vm_jslt32_reg() {
  1237. let prog = assemble("
  1238. mov r9, 1
  1239. lsh r9, 32
  1240. mov32 r0, 0
  1241. mov32 r1, -2
  1242. or r1, r9
  1243. mov r2, -3
  1244. mov r3, -1
  1245. jslt32 r1, r1, +4
  1246. jslt32 r1, r2, +3
  1247. jslt32 r1, r3, +1
  1248. exit
  1249. mov32 r0, 1
  1250. exit").unwrap();
  1251. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1252. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1253. }
  1254. #[test]
  1255. fn test_vm_lddw() {
  1256. let prog = assemble("lddw r0, 0x1122334455667788
  1257. exit").unwrap();
  1258. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1259. assert_eq!(vm.execute_program().unwrap(), 0x1122334455667788);
  1260. }
  1261. #[test]
  1262. fn test_vm_lddw2() {
  1263. let prog = assemble("
  1264. lddw r0, 0x0000000080000000
  1265. exit").unwrap();
  1266. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1267. assert_eq!(vm.execute_program().unwrap(), 0x80000000);
  1268. }
  1269. #[test]
  1270. fn test_vm_ldxb_all() {
  1271. let prog = assemble("
  1272. mov r0, r1
  1273. ldxb r9, [r0+0]
  1274. lsh r9, 0
  1275. ldxb r8, [r0+1]
  1276. lsh r8, 4
  1277. ldxb r7, [r0+2]
  1278. lsh r7, 8
  1279. ldxb r6, [r0+3]
  1280. lsh r6, 12
  1281. ldxb r5, [r0+4]
  1282. lsh r5, 16
  1283. ldxb r4, [r0+5]
  1284. lsh r4, 20
  1285. ldxb r3, [r0+6]
  1286. lsh r3, 24
  1287. ldxb r2, [r0+7]
  1288. lsh r2, 28
  1289. ldxb r1, [r0+8]
  1290. lsh r1, 32
  1291. ldxb r0, [r0+9]
  1292. lsh r0, 36
  1293. or r0, r1
  1294. or r0, r2
  1295. or r0, r3
  1296. or r0, r4
  1297. or r0, r5
  1298. or r0, r6
  1299. or r0, r7
  1300. or r0, r8
  1301. or r0, r9
  1302. exit").unwrap();
  1303. let mem = &mut [
  1304. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  1305. 0x08, 0x09
  1306. ];
  1307. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1308. assert_eq!(vm.execute_program(mem).unwrap(), 0x9876543210);
  1309. }
  1310. #[test]
  1311. fn test_vm_ldxb() {
  1312. let prog = assemble("
  1313. ldxb r0, [r1+2]
  1314. exit").unwrap();
  1315. let mem = &mut [
  1316. 0xaa, 0xbb, 0x11, 0xcc, 0xdd
  1317. ];
  1318. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1319. assert_eq!(vm.execute_program(mem).unwrap(), 0x11);
  1320. }
  1321. #[test]
  1322. fn test_vm_ldxdw() {
  1323. let prog = assemble("
  1324. ldxdw r0, [r1+2]
  1325. exit").unwrap();
  1326. let mem = &mut [
  1327. 0xaa, 0xbb, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66,
  1328. 0x77, 0x88, 0xcc, 0xdd
  1329. ];
  1330. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1331. assert_eq!(vm.execute_program(mem).unwrap(), 0x8877665544332211);
  1332. }
  1333. #[test]
  1334. fn test_vm_ldxh_all() {
  1335. let prog = assemble("
  1336. mov r0, r1
  1337. ldxh r9, [r0+0]
  1338. be16 r9
  1339. lsh r9, 0
  1340. ldxh r8, [r0+2]
  1341. be16 r8
  1342. lsh r8, 4
  1343. ldxh r7, [r0+4]
  1344. be16 r7
  1345. lsh r7, 8
  1346. ldxh r6, [r0+6]
  1347. be16 r6
  1348. lsh r6, 12
  1349. ldxh r5, [r0+8]
  1350. be16 r5
  1351. lsh r5, 16
  1352. ldxh r4, [r0+10]
  1353. be16 r4
  1354. lsh r4, 20
  1355. ldxh r3, [r0+12]
  1356. be16 r3
  1357. lsh r3, 24
  1358. ldxh r2, [r0+14]
  1359. be16 r2
  1360. lsh r2, 28
  1361. ldxh r1, [r0+16]
  1362. be16 r1
  1363. lsh r1, 32
  1364. ldxh r0, [r0+18]
  1365. be16 r0
  1366. lsh r0, 36
  1367. or r0, r1
  1368. or r0, r2
  1369. or r0, r3
  1370. or r0, r4
  1371. or r0, r5
  1372. or r0, r6
  1373. or r0, r7
  1374. or r0, r8
  1375. or r0, r9
  1376. exit").unwrap();
  1377. let mem = &mut [
  1378. 0x00, 0x00, 0x00, 0x01, 0x00, 0x02, 0x00, 0x03,
  1379. 0x00, 0x04, 0x00, 0x05, 0x00, 0x06, 0x00, 0x07,
  1380. 0x00, 0x08, 0x00, 0x09
  1381. ];
  1382. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1383. assert_eq!(vm.execute_program(mem).unwrap(), 0x9876543210);
  1384. }
  1385. #[test]
  1386. fn test_vm_ldxh_all2() {
  1387. let prog = assemble("
  1388. mov r0, r1
  1389. ldxh r9, [r0+0]
  1390. be16 r9
  1391. ldxh r8, [r0+2]
  1392. be16 r8
  1393. ldxh r7, [r0+4]
  1394. be16 r7
  1395. ldxh r6, [r0+6]
  1396. be16 r6
  1397. ldxh r5, [r0+8]
  1398. be16 r5
  1399. ldxh r4, [r0+10]
  1400. be16 r4
  1401. ldxh r3, [r0+12]
  1402. be16 r3
  1403. ldxh r2, [r0+14]
  1404. be16 r2
  1405. ldxh r1, [r0+16]
  1406. be16 r1
  1407. ldxh r0, [r0+18]
  1408. be16 r0
  1409. or r0, r1
  1410. or r0, r2
  1411. or r0, r3
  1412. or r0, r4
  1413. or r0, r5
  1414. or r0, r6
  1415. or r0, r7
  1416. or r0, r8
  1417. or r0, r9
  1418. exit").unwrap();
  1419. let mem = &mut [
  1420. 0x00, 0x01, 0x00, 0x02, 0x00, 0x04, 0x00, 0x08,
  1421. 0x00, 0x10, 0x00, 0x20, 0x00, 0x40, 0x00, 0x80,
  1422. 0x01, 0x00, 0x02, 0x00
  1423. ];
  1424. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1425. assert_eq!(vm.execute_program(mem).unwrap(), 0x3ff);
  1426. }
  1427. #[test]
  1428. fn test_vm_ldxh() {
  1429. let prog = assemble("
  1430. ldxh r0, [r1+2]
  1431. exit").unwrap();
  1432. let mem = &mut [
  1433. 0xaa, 0xbb, 0x11, 0x22, 0xcc, 0xdd
  1434. ];
  1435. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1436. assert_eq!(vm.execute_program(mem).unwrap(), 0x2211);
  1437. }
  1438. #[test]
  1439. fn test_vm_ldxh_same_reg() {
  1440. let prog = assemble("
  1441. mov r0, r1
  1442. sth [r0], 0x1234
  1443. ldxh r0, [r0]
  1444. exit").unwrap();
  1445. let mem = &mut [
  1446. 0xff, 0xff
  1447. ];
  1448. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1449. assert_eq!(vm.execute_program(mem).unwrap(), 0x1234);
  1450. }
  1451. #[test]
  1452. fn test_vm_ldxw_all() {
  1453. let prog = assemble("
  1454. mov r0, r1
  1455. ldxw r9, [r0+0]
  1456. be32 r9
  1457. ldxw r8, [r0+4]
  1458. be32 r8
  1459. ldxw r7, [r0+8]
  1460. be32 r7
  1461. ldxw r6, [r0+12]
  1462. be32 r6
  1463. ldxw r5, [r0+16]
  1464. be32 r5
  1465. ldxw r4, [r0+20]
  1466. be32 r4
  1467. ldxw r3, [r0+24]
  1468. be32 r3
  1469. ldxw r2, [r0+28]
  1470. be32 r2
  1471. ldxw r1, [r0+32]
  1472. be32 r1
  1473. ldxw r0, [r0+36]
  1474. be32 r0
  1475. or r0, r1
  1476. or r0, r2
  1477. or r0, r3
  1478. or r0, r4
  1479. or r0, r5
  1480. or r0, r6
  1481. or r0, r7
  1482. or r0, r8
  1483. or r0, r9
  1484. exit").unwrap();
  1485. let mem = &mut [
  1486. 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02,
  1487. 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x08,
  1488. 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00,
  1489. 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x08, 0x00,
  1490. 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00
  1491. ];
  1492. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1493. assert_eq!(vm.execute_program(mem).unwrap(), 0x030f0f);
  1494. }
  1495. #[test]
  1496. fn test_vm_ldxw() {
  1497. let prog = assemble("
  1498. ldxw r0, [r1+2]
  1499. exit").unwrap();
  1500. let mem = &mut [
  1501. 0xaa, 0xbb, 0x11, 0x22, 0x33, 0x44, 0xcc, 0xdd
  1502. ];
  1503. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1504. assert_eq!(vm.execute_program(mem).unwrap(), 0x44332211);
  1505. }
  1506. #[test]
  1507. fn test_vm_le16() {
  1508. let prog = assemble("
  1509. ldxh r0, [r1]
  1510. le16 r0
  1511. exit").unwrap();
  1512. let mem = &mut [
  1513. 0x22, 0x11
  1514. ];
  1515. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1516. assert_eq!(vm.execute_program(mem).unwrap(), 0x1122);
  1517. }
  1518. #[test]
  1519. fn test_vm_le32() {
  1520. let prog = assemble("
  1521. ldxw r0, [r1]
  1522. le32 r0
  1523. exit").unwrap();
  1524. let mem = &mut [
  1525. 0x44, 0x33, 0x22, 0x11
  1526. ];
  1527. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1528. assert_eq!(vm.execute_program(mem).unwrap(), 0x11223344);
  1529. }
  1530. #[test]
  1531. fn test_vm_le64() {
  1532. let prog = assemble("
  1533. ldxdw r0, [r1]
  1534. le64 r0
  1535. exit").unwrap();
  1536. let mem = &mut [
  1537. 0x88, 0x77, 0x66, 0x55, 0x44, 0x33, 0x22, 0x11
  1538. ];
  1539. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1540. assert_eq!(vm.execute_program(mem).unwrap(), 0x1122334455667788);
  1541. }
  1542. #[test]
  1543. fn test_vm_lsh_imm() {
  1544. let prog = assemble("
  1545. mov r0, 1
  1546. lsh r0, 4
  1547. exit").unwrap();
  1548. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1549. assert_eq!(vm.execute_program().unwrap(), 0x10);
  1550. }
  1551. #[test]
  1552. fn test_vm_lsh_reg() {
  1553. let prog = assemble("
  1554. mov r0, 1
  1555. mov r7, 4
  1556. lsh r0, r7
  1557. exit").unwrap();
  1558. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1559. assert_eq!(vm.execute_program().unwrap(), 0x10);
  1560. }
  1561. #[test]
  1562. fn test_vm_lsh32_imm() {
  1563. let prog = assemble("
  1564. mov32 r0, 1
  1565. lsh32 r0, 4
  1566. exit").unwrap();
  1567. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1568. assert_eq!(vm.execute_program().unwrap(), 0x10);
  1569. }
  1570. #[test]
  1571. fn test_vm_lsh32_reg() {
  1572. let prog = assemble("
  1573. mov32 r0, 1
  1574. mov32 r7, 4
  1575. lsh32 r0, r7
  1576. exit").unwrap();
  1577. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1578. assert_eq!(vm.execute_program().unwrap(), 0x10);
  1579. }
  1580. #[test]
  1581. fn test_vm_lsh_imm_overflow() {
  1582. let prog = assemble("
  1583. mov r0, 1
  1584. lsh r0, 64
  1585. exit").unwrap();
  1586. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1587. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1588. }
  1589. #[test]
  1590. fn test_vm_lsh_reg_overflow() {
  1591. let prog = assemble("
  1592. mov r0, 1
  1593. mov r7, 64
  1594. lsh r0, r7
  1595. exit").unwrap();
  1596. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1597. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1598. }
  1599. #[test]
  1600. fn test_vm_lsh32_imm_overflow() {
  1601. let prog = assemble("
  1602. mov32 r0, 1
  1603. lsh32 r0, 32
  1604. exit").unwrap();
  1605. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1606. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1607. }
  1608. #[test]
  1609. fn test_vm_lsh32_reg_overflow() {
  1610. let prog = assemble("
  1611. mov32 r0, 1
  1612. mov32 r7, 32
  1613. lsh32 r0, r7
  1614. exit").unwrap();
  1615. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1616. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1617. }
  1618. #[test]
  1619. fn test_vm_mod() {
  1620. let prog = assemble("
  1621. mov32 r0, 5748
  1622. mod32 r0, 92
  1623. mov32 r1, 13
  1624. mod32 r0, r1
  1625. exit").unwrap();
  1626. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1627. assert_eq!(vm.execute_program().unwrap(), 0x5);
  1628. }
  1629. #[test]
  1630. fn test_vm_mod32() {
  1631. let prog = assemble("
  1632. lddw r0, 0x100000003
  1633. mod32 r0, 3
  1634. exit").unwrap();
  1635. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1636. assert_eq!(vm.execute_program().unwrap(), 0x0);
  1637. }
  1638. #[test]
  1639. fn test_vm_mod64() {
  1640. let prog = assemble("
  1641. mov32 r0, -1316649930
  1642. lsh r0, 32
  1643. or r0, 0x100dc5c8
  1644. mov32 r1, 0xdde263e
  1645. lsh r1, 32
  1646. or r1, 0x3cbef7f3
  1647. mod r0, r1
  1648. mod r0, 0x658f1778
  1649. exit").unwrap();
  1650. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1651. assert_eq!(vm.execute_program().unwrap(), 0x30ba5a04);
  1652. }
  1653. #[test]
  1654. fn test_vm_mov() {
  1655. let prog = assemble("
  1656. mov32 r1, 1
  1657. mov32 r0, r1
  1658. exit").unwrap();
  1659. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1660. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1661. }
  1662. #[test]
  1663. fn test_vm_mul32_imm() {
  1664. let prog = assemble("
  1665. mov r0, 3
  1666. mul32 r0, 4
  1667. exit").unwrap();
  1668. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1669. assert_eq!(vm.execute_program().unwrap(), 0xc);
  1670. }
  1671. #[test]
  1672. fn test_vm_mul32_reg() {
  1673. let prog = assemble("
  1674. mov r0, 3
  1675. mov r1, 4
  1676. mul32 r0, r1
  1677. exit").unwrap();
  1678. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1679. assert_eq!(vm.execute_program().unwrap(), 0xc);
  1680. }
  1681. #[test]
  1682. fn test_vm_mul32_reg_overflow() {
  1683. let prog = assemble("
  1684. mov r0, 0x40000001
  1685. mov r1, 4
  1686. mul32 r0, r1
  1687. exit").unwrap();
  1688. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1689. assert_eq!(vm.execute_program().unwrap(), 0x4);
  1690. }
  1691. #[test]
  1692. fn test_vm_mul64_imm() {
  1693. let prog = assemble("
  1694. mov r0, 0x40000001
  1695. mul r0, 4
  1696. exit").unwrap();
  1697. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1698. assert_eq!(vm.execute_program().unwrap(), 0x100000004);
  1699. }
  1700. #[test]
  1701. fn test_vm_mul64_reg() {
  1702. let prog = assemble("
  1703. mov r0, 0x40000001
  1704. mov r1, 4
  1705. mul r0, r1
  1706. exit").unwrap();
  1707. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1708. assert_eq!(vm.execute_program().unwrap(), 0x100000004);
  1709. }
  1710. #[test]
  1711. fn test_vm_mul_loop() {
  1712. let prog = assemble("
  1713. mov r0, 0x7
  1714. add r1, 0xa
  1715. lsh r1, 0x20
  1716. rsh r1, 0x20
  1717. jeq r1, 0x0, +4
  1718. mov r0, 0x7
  1719. mul r0, 0x7
  1720. add r1, -1
  1721. jne r1, 0x0, -3
  1722. exit").unwrap();
  1723. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1724. assert_eq!(vm.execute_program().unwrap(), 0x75db9c97);
  1725. }
  1726. #[test]
  1727. fn test_vm_neg64() {
  1728. let prog = assemble("
  1729. mov32 r0, 2
  1730. neg r0
  1731. exit").unwrap();
  1732. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1733. assert_eq!(vm.execute_program().unwrap(), 0xfffffffffffffffe);
  1734. }
  1735. #[test]
  1736. fn test_vm_neg() {
  1737. let prog = assemble("
  1738. mov32 r0, 2
  1739. neg32 r0
  1740. exit").unwrap();
  1741. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1742. assert_eq!(vm.execute_program().unwrap(), 0xfffffffe);
  1743. }
  1744. #[test]
  1745. fn test_vm_prime() {
  1746. let prog = assemble("
  1747. mov r1, 67
  1748. mov r0, 0x1
  1749. mov r2, 0x2
  1750. jgt r1, 0x2, +4
  1751. ja +10
  1752. add r2, 0x1
  1753. mov r0, 0x1
  1754. jge r2, r1, +7
  1755. mov r3, r1
  1756. div r3, r2
  1757. mul r3, r2
  1758. mov r4, r1
  1759. sub r4, r3
  1760. mov r0, 0x0
  1761. jne r4, 0x0, -10
  1762. exit").unwrap();
  1763. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1764. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1765. }
  1766. #[test]
  1767. fn test_vm_rhs32() {
  1768. let prog = assemble("
  1769. xor r0, r0
  1770. sub r0, 1
  1771. rsh32 r0, 8
  1772. exit").unwrap();
  1773. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1774. assert_eq!(vm.execute_program().unwrap(), 0x00ffffff);
  1775. }
  1776. #[test]
  1777. fn test_vm_rsh_reg() {
  1778. let prog = assemble("
  1779. mov r0, 0x10
  1780. mov r7, 4
  1781. rsh r0, r7
  1782. exit").unwrap();
  1783. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1784. assert_eq!(vm.execute_program().unwrap(), 0x1);
  1785. }
  1786. #[test]
  1787. fn test_vm_stack() {
  1788. let prog = assemble("
  1789. mov r1, 51
  1790. stdw [r10-16], 0xab
  1791. stdw [r10-8], 0xcd
  1792. and r1, 1
  1793. lsh r1, 3
  1794. mov r2, r10
  1795. add r2, r1
  1796. ldxdw r0, [r2-16]
  1797. exit").unwrap();
  1798. let vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1799. assert_eq!(vm.execute_program().unwrap(), 0xcd);
  1800. }
  1801. #[test]
  1802. fn test_vm_stack2() {
  1803. let prog = assemble("
  1804. stb [r10-4], 0x01
  1805. stb [r10-3], 0x02
  1806. stb [r10-2], 0x03
  1807. stb [r10-1], 0x04
  1808. mov r1, r10
  1809. mov r2, 0x4
  1810. sub r1, r2
  1811. call 1
  1812. mov r1, 0
  1813. ldxb r2, [r10-4]
  1814. ldxb r3, [r10-3]
  1815. ldxb r4, [r10-2]
  1816. ldxb r5, [r10-1]
  1817. call 0
  1818. xor r0, 0x2a2a2a2a
  1819. exit").unwrap();
  1820. let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1821. vm.register_helper(0, helpers::gather_bytes).unwrap();
  1822. vm.register_helper(1, helpers::memfrob).unwrap();
  1823. assert_eq!(vm.execute_program().unwrap(), 0x01020304);
  1824. }
  1825. #[test]
  1826. fn test_vm_stb() {
  1827. let prog = assemble("
  1828. stb [r1+2], 0x11
  1829. ldxb r0, [r1+2]
  1830. exit").unwrap();
  1831. let mem = &mut [
  1832. 0xaa, 0xbb, 0xff, 0xcc, 0xdd
  1833. ];
  1834. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1835. assert_eq!(vm.execute_program(mem).unwrap(), 0x11);
  1836. }
  1837. #[test]
  1838. fn test_vm_stdw() {
  1839. let prog = assemble("
  1840. stdw [r1+2], 0x44332211
  1841. ldxdw r0, [r1+2]
  1842. exit").unwrap();
  1843. let mem = &mut [
  1844. 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
  1845. 0xff, 0xff, 0xcc, 0xdd
  1846. ];
  1847. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1848. assert_eq!(vm.execute_program(mem).unwrap(), 0x44332211);
  1849. }
  1850. // If this case is not handled properly in check_mem(), then we may overflow when adding the
  1851. // context address and the offset, and make the thread panic with "attempt to add with overflow".
  1852. // Check that we panic with the expected out-of-bounds error.
  1853. #[test]
  1854. #[should_panic(expected = "Error: out of bounds memory store (insn #1)")]
  1855. fn test_vm_stdw_add_overflow() {
  1856. let prog = assemble("
  1857. stdw [r2-0x1], 0x44332211
  1858. ldxw r0, [r1+2]
  1859. exit").unwrap();
  1860. let mem = &mut [
  1861. 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
  1862. 0xff, 0xff, 0xcc, 0xdd
  1863. ];
  1864. let mut vm = rbpf::EbpfVmFixedMbuff::new(Some(&prog), 0x00, 0x10).unwrap();
  1865. _ = vm.execute_program(mem).unwrap();
  1866. }
  1867. #[test]
  1868. fn test_vm_sth() {
  1869. let prog = assemble("
  1870. sth [r1+2], 0x2211
  1871. ldxh r0, [r1+2]
  1872. exit").unwrap();
  1873. let mem = &mut [
  1874. 0xaa, 0xbb, 0xff, 0xff, 0xcc, 0xdd
  1875. ];
  1876. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1877. assert_eq!(vm.execute_program(mem).unwrap(), 0x2211);
  1878. }
  1879. #[test]
  1880. fn test_vm_string_stack() {
  1881. let prog = assemble("
  1882. mov r1, 0x78636261
  1883. stxw [r10-8], r1
  1884. mov r6, 0x0
  1885. stxb [r10-4], r6
  1886. stxb [r10-12], r6
  1887. mov r1, 0x79636261
  1888. stxw [r10-16], r1
  1889. mov r1, r10
  1890. add r1, -8
  1891. mov r2, r1
  1892. call 0x4
  1893. mov r1, r0
  1894. mov r0, 0x1
  1895. lsh r1, 0x20
  1896. rsh r1, 0x20
  1897. jne r1, 0x0, +11
  1898. mov r1, r10
  1899. add r1, -8
  1900. mov r2, r10
  1901. add r2, -16
  1902. call 0x4
  1903. mov r1, r0
  1904. lsh r1, 0x20
  1905. rsh r1, 0x20
  1906. mov r0, 0x1
  1907. jeq r1, r6, +1
  1908. mov r0, 0x0
  1909. exit").unwrap();
  1910. let mut vm = rbpf::EbpfVmNoData::new(Some(&prog)).unwrap();
  1911. vm.register_helper(4, helpers::strcmp).unwrap();
  1912. assert_eq!(vm.execute_program().unwrap(), 0x0);
  1913. }
  1914. #[test]
  1915. fn test_vm_stw() {
  1916. let prog = assemble("
  1917. stw [r1+2], 0x44332211
  1918. ldxw r0, [r1+2]
  1919. exit").unwrap();
  1920. let mem = &mut [
  1921. 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd
  1922. ];
  1923. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1924. assert_eq!(vm.execute_program(mem).unwrap(), 0x44332211);
  1925. }
  1926. #[test]
  1927. fn test_vm_stxb() {
  1928. let prog = assemble("
  1929. mov32 r2, 0x11
  1930. stxb [r1+2], r2
  1931. ldxb r0, [r1+2]
  1932. exit").unwrap();
  1933. let mem = &mut [
  1934. 0xaa, 0xbb, 0xff, 0xcc, 0xdd
  1935. ];
  1936. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1937. assert_eq!(vm.execute_program(mem).unwrap(), 0x11);
  1938. }
  1939. #[test]
  1940. fn test_vm_stxb_all() {
  1941. let prog = assemble("
  1942. mov r0, 0xf0
  1943. mov r2, 0xf2
  1944. mov r3, 0xf3
  1945. mov r4, 0xf4
  1946. mov r5, 0xf5
  1947. mov r6, 0xf6
  1948. mov r7, 0xf7
  1949. mov r8, 0xf8
  1950. stxb [r1], r0
  1951. stxb [r1+1], r2
  1952. stxb [r1+2], r3
  1953. stxb [r1+3], r4
  1954. stxb [r1+4], r5
  1955. stxb [r1+5], r6
  1956. stxb [r1+6], r7
  1957. stxb [r1+7], r8
  1958. ldxdw r0, [r1]
  1959. be64 r0
  1960. exit").unwrap();
  1961. let mem = &mut [
  1962. 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff
  1963. ];
  1964. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1965. assert_eq!(vm.execute_program(mem).unwrap(), 0xf0f2f3f4f5f6f7f8);
  1966. }
  1967. #[test]
  1968. fn test_vm_stxb_all2() {
  1969. let prog = assemble("
  1970. mov r0, r1
  1971. mov r1, 0xf1
  1972. mov r9, 0xf9
  1973. stxb [r0], r1
  1974. stxb [r0+1], r9
  1975. ldxh r0, [r0]
  1976. be16 r0
  1977. exit").unwrap();
  1978. let mem = &mut [
  1979. 0xff, 0xff
  1980. ];
  1981. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  1982. assert_eq!(vm.execute_program(mem).unwrap(), 0xf1f9);
  1983. }
  1984. #[test]
  1985. fn test_vm_stxb_chain() {
  1986. let prog = assemble("
  1987. mov r0, r1
  1988. ldxb r9, [r0+0]
  1989. stxb [r0+1], r9
  1990. ldxb r8, [r0+1]
  1991. stxb [r0+2], r8
  1992. ldxb r7, [r0+2]
  1993. stxb [r0+3], r7
  1994. ldxb r6, [r0+3]
  1995. stxb [r0+4], r6
  1996. ldxb r5, [r0+4]
  1997. stxb [r0+5], r5
  1998. ldxb r4, [r0+5]
  1999. stxb [r0+6], r4
  2000. ldxb r3, [r0+6]
  2001. stxb [r0+7], r3
  2002. ldxb r2, [r0+7]
  2003. stxb [r0+8], r2
  2004. ldxb r1, [r0+8]
  2005. stxb [r0+9], r1
  2006. ldxb r0, [r0+9]
  2007. exit").unwrap();
  2008. let mem = &mut [
  2009. 0x2a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  2010. 0x00, 0x00
  2011. ];
  2012. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  2013. assert_eq!(vm.execute_program(mem).unwrap(), 0x2a);
  2014. }
  2015. #[test]
  2016. fn test_vm_stxdw() {
  2017. let prog = assemble("
  2018. mov r2, -2005440939
  2019. lsh r2, 32
  2020. or r2, 0x44332211
  2021. stxdw [r1+2], r2
  2022. ldxdw r0, [r1+2]
  2023. exit").unwrap();
  2024. let mem = &mut [
  2025. 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
  2026. 0xff, 0xff, 0xcc, 0xdd
  2027. ];
  2028. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  2029. assert_eq!(vm.execute_program(mem).unwrap(), 0x8877665544332211);
  2030. }
  2031. #[test]
  2032. fn test_vm_stxh() {
  2033. let prog = assemble("
  2034. mov32 r2, 0x2211
  2035. stxh [r1+2], r2
  2036. ldxh r0, [r1+2]
  2037. exit").unwrap();
  2038. let mem = &mut [
  2039. 0xaa, 0xbb, 0xff, 0xff, 0xcc, 0xdd
  2040. ];
  2041. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  2042. assert_eq!(vm.execute_program(mem).unwrap(), 0x2211);
  2043. }
  2044. #[test]
  2045. fn test_vm_stxw() {
  2046. let prog = assemble("
  2047. mov32 r2, 0x44332211
  2048. stxw [r1+2], r2
  2049. ldxw r0, [r1+2]
  2050. exit").unwrap();
  2051. let mem = &mut [
  2052. 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd
  2053. ];
  2054. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  2055. assert_eq!(vm.execute_program(mem).unwrap(), 0x44332211);
  2056. }
  2057. #[test]
  2058. fn test_vm_subnet() {
  2059. let prog = assemble("
  2060. mov r2, 0xe
  2061. ldxh r3, [r1+12]
  2062. jne r3, 0x81, +2
  2063. mov r2, 0x12
  2064. ldxh r3, [r1+16]
  2065. and r3, 0xffff
  2066. jne r3, 0x8, +5
  2067. add r1, r2
  2068. mov r0, 0x1
  2069. ldxw r1, [r1+16]
  2070. and r1, 0xffffff
  2071. jeq r1, 0x1a8c0, +1
  2072. mov r0, 0x0
  2073. exit").unwrap();
  2074. let mem = &mut [
  2075. 0x00, 0x00, 0xc0, 0x9f, 0xa0, 0x97, 0x00, 0xa0,
  2076. 0xcc, 0x3b, 0xbf, 0xfa, 0x08, 0x00, 0x45, 0x10,
  2077. 0x00, 0x3c, 0x46, 0x3c, 0x40, 0x00, 0x40, 0x06,
  2078. 0x73, 0x1c, 0xc0, 0xa8, 0x01, 0x02, 0xc0, 0xa8,
  2079. 0x01, 0x01, 0x06, 0x0e, 0x00, 0x17, 0x99, 0xc5,
  2080. 0xa0, 0xec, 0x00, 0x00, 0x00, 0x00, 0xa0, 0x02,
  2081. 0x7d, 0x78, 0xe0, 0xa3, 0x00, 0x00, 0x02, 0x04,
  2082. 0x05, 0xb4, 0x04, 0x02, 0x08, 0x0a, 0x00, 0x9c,
  2083. 0x27, 0x24, 0x00, 0x00, 0x00, 0x00, 0x01, 0x03,
  2084. 0x03, 0x00
  2085. ];
  2086. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  2087. assert_eq!(vm.execute_program(mem).unwrap(), 0x1);
  2088. }
  2089. const PROG_TCP_PORT_80: [u8;152] = [
  2090. 0x71, 0x12, 0x0c, 0x00, 0x00, 0x00, 0x00, 0x00,
  2091. 0x71, 0x13, 0x0d, 0x00, 0x00, 0x00, 0x00, 0x00,
  2092. 0x67, 0x03, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00,
  2093. 0x4f, 0x23, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  2094. 0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  2095. 0x55, 0x03, 0x0c, 0x00, 0x08, 0x00, 0x00, 0x00,
  2096. 0x71, 0x12, 0x17, 0x00, 0x00, 0x00, 0x00, 0x00,
  2097. 0x55, 0x02, 0x0a, 0x00, 0x06, 0x00, 0x00, 0x00,
  2098. 0x71, 0x12, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00,
  2099. 0x07, 0x01, 0x00, 0x00, 0x0e, 0x00, 0x00, 0x00,
  2100. 0x57, 0x02, 0x00, 0x00, 0x0f, 0x00, 0x00, 0x00,
  2101. 0x67, 0x02, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00,
  2102. 0x0f, 0x21, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  2103. 0x69, 0x12, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00,
  2104. 0x15, 0x02, 0x02, 0x00, 0x00, 0x50, 0x00, 0x00,
  2105. 0x69, 0x11, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  2106. 0x55, 0x01, 0x01, 0x00, 0x00, 0x50, 0x00, 0x00,
  2107. 0xb7, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
  2108. 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
  2109. ];
  2110. #[test]
  2111. fn test_vm_tcp_port80_match() {
  2112. let mem = &mut [
  2113. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06,
  2114. 0x07, 0x08, 0x09, 0x0a, 0x08, 0x00, 0x45, 0x00,
  2115. 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06,
  2116. 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8,
  2117. 0x00, 0x02, 0x27, 0x10, 0x00, 0x50, 0x00, 0x00,
  2118. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02,
  2119. 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44,
  2120. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2121. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2122. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2123. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2124. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2125. 0x44, 0x44, 0x44, 0x44
  2126. ];
  2127. let prog = &PROG_TCP_PORT_80;
  2128. let vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
  2129. assert_eq!(vm.execute_program(mem).unwrap(), 0x1);
  2130. }
  2131. #[test]
  2132. fn test_vm_tcp_port80_nomatch() {
  2133. let mem = &mut [
  2134. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06,
  2135. 0x07, 0x08, 0x09, 0x0a, 0x08, 0x00, 0x45, 0x00,
  2136. 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06,
  2137. 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8,
  2138. 0x00, 0x02, 0x00, 0x16, 0x27, 0x10, 0x00, 0x00,
  2139. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x51, 0x02,
  2140. 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44,
  2141. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2142. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2143. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2144. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2145. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2146. 0x44, 0x44, 0x44, 0x44
  2147. ];
  2148. let prog = &PROG_TCP_PORT_80;
  2149. let vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
  2150. assert_eq!(vm.execute_program(mem).unwrap(), 0x0);
  2151. }
  2152. #[test]
  2153. fn test_vm_tcp_port80_nomatch_ethertype() {
  2154. let mem = &mut [
  2155. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06,
  2156. 0x07, 0x08, 0x09, 0x0a, 0x08, 0x01, 0x45, 0x00,
  2157. 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06,
  2158. 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8,
  2159. 0x00, 0x02, 0x27, 0x10, 0x00, 0x50, 0x00, 0x00,
  2160. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02,
  2161. 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44,
  2162. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2163. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2164. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2165. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2166. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2167. 0x44, 0x44, 0x44, 0x44
  2168. ];
  2169. let prog = &PROG_TCP_PORT_80;
  2170. let vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
  2171. assert_eq!(vm.execute_program(mem).unwrap(), 0x0);
  2172. }
  2173. #[test]
  2174. fn test_vm_tcp_port80_nomatch_proto() {
  2175. let mem = &mut [
  2176. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06,
  2177. 0x07, 0x08, 0x09, 0x0a, 0x08, 0x00, 0x45, 0x00,
  2178. 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x11,
  2179. 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8,
  2180. 0x00, 0x02, 0x27, 0x10, 0x00, 0x50, 0x00, 0x00,
  2181. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02,
  2182. 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44,
  2183. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2184. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2185. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2186. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2187. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  2188. 0x44, 0x44, 0x44, 0x44
  2189. ];
  2190. let prog = &PROG_TCP_PORT_80;
  2191. let vm = rbpf::EbpfVmRaw::new(Some(prog)).unwrap();
  2192. assert_eq!(vm.execute_program(mem).unwrap(), 0x0);
  2193. }
  2194. #[test]
  2195. fn test_vm_tcp_sack_match() {
  2196. let mut mem = TCP_SACK_MATCH.to_vec();
  2197. let prog = assemble(TCP_SACK_ASM).unwrap();
  2198. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  2199. assert_eq!(vm.execute_program(mem.as_mut_slice()).unwrap(), 0x1);
  2200. }
  2201. #[test]
  2202. fn test_vm_tcp_sack_nomatch() {
  2203. let mut mem = TCP_SACK_NOMATCH.to_vec();
  2204. let prog = assemble(TCP_SACK_ASM).unwrap();
  2205. let vm = rbpf::EbpfVmRaw::new(Some(&prog)).unwrap();
  2206. assert_eq!(vm.execute_program(mem.as_mut_slice()).unwrap(), 0x0);
  2207. }