ubpf_vm.rs 41 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744
  1. // Converted from the tests for uBPF <https://github.com/iovisor/ubpf>
  2. // Copyright 2015 Big Switch Networks, Inc
  3. // Copyright 2016 6WIND S.A. <quentin.monnet@6wind.com>
  4. //
  5. // Licensed under the Apache License, Version 2.0 <http://www.apache.org/licenses/LICENSE-2.0> or
  6. // the MIT license <http://opensource.org/licenses/MIT>, at your option. This file may not be
  7. // copied, modified, or distributed except according to those terms.
  8. // The tests contained in this file are extracted from the unit tests of uBPF software. Each test
  9. // in this file has a name in the form `test_vm_<name>`, and corresponds to the (human-readable)
  10. // code in `ubpf/tree/master/tests/<name>`, available at
  11. // <https://github.com/iovisor/ubpf/tree/master/tests> (hyphen had to be replaced with underscores
  12. // as Rust will not accept them in function names). It is strongly advised to refer to the uBPF
  13. // version to understand what these program do.
  14. //
  15. // Each program was assembled from the uBPF version with the assembler provided by uBPF itself, and
  16. // available at <https://github.com/iovisor/ubpf/tree/master/ubpf>.
  17. // The very few modifications that have been realized should be indicated.
  18. // These are unit tests for the eBPF interpreter.
  19. #![cfg_attr(feature = "cargo-clippy", allow(unreadable_literal))]
  20. extern crate rbpf;
  21. mod common;
  22. use rbpf::helpers;
  23. use rbpf::assembler::assemble;
  24. use common::{TCP_SACK_ASM, TCP_SACK_MATCH, TCP_SACK_NOMATCH};
  25. #[test]
  26. fn test_vm_add() {
  27. let prog = assemble("
  28. mov32 r0, 0
  29. mov32 r1, 2
  30. add32 r0, 1
  31. add32 r0, r1
  32. exit").unwrap();
  33. let vm = rbpf::EbpfVmNoData::new(&prog);
  34. assert_eq!(vm.prog_exec(), 0x3);
  35. }
  36. #[test]
  37. fn test_vm_alu64_arith() {
  38. let prog = assemble("
  39. mov r0, 0
  40. mov r1, 1
  41. mov r2, 2
  42. mov r3, 3
  43. mov r4, 4
  44. mov r5, 5
  45. mov r6, 6
  46. mov r7, 7
  47. mov r8, 8
  48. mov r9, 9
  49. add r0, 23
  50. add r0, r7
  51. sub r0, 13
  52. sub r0, r1
  53. mul r0, 7
  54. mul r0, r3
  55. div r0, 2
  56. div r0, r4
  57. exit").unwrap();
  58. let vm = rbpf::EbpfVmNoData::new(&prog);
  59. assert_eq!(vm.prog_exec(), 0x2a);
  60. }
  61. #[test]
  62. fn test_vm_alu64_bit() {
  63. let prog = assemble("
  64. mov r0, 0
  65. mov r1, 1
  66. mov r2, 2
  67. mov r3, 3
  68. mov r4, 4
  69. mov r5, 5
  70. mov r6, 6
  71. mov r7, 7
  72. mov r8, 8
  73. or r0, r5
  74. or r0, 0xa0
  75. and r0, 0xa3
  76. mov r9, 0x91
  77. and r0, r9
  78. lsh r0, 32
  79. lsh r0, 22
  80. lsh r0, r8
  81. rsh r0, 32
  82. rsh r0, 19
  83. rsh r0, r7
  84. xor r0, 0x03
  85. xor r0, r2
  86. exit").unwrap();
  87. let vm = rbpf::EbpfVmNoData::new(&prog);
  88. assert_eq!(vm.prog_exec(), 0x11);
  89. }
  90. #[test]
  91. fn test_vm_alu_arith() {
  92. let prog = assemble("
  93. mov32 r0, 0
  94. mov32 r1, 1
  95. mov32 r2, 2
  96. mov32 r3, 3
  97. mov32 r4, 4
  98. mov32 r5, 5
  99. mov32 r6, 6
  100. mov32 r7, 7
  101. mov32 r8, 8
  102. mov32 r9, 9
  103. add32 r0, 23
  104. add32 r0, r7
  105. sub32 r0, 13
  106. sub32 r0, r1
  107. mul32 r0, 7
  108. mul32 r0, r3
  109. div32 r0, 2
  110. div32 r0, r4
  111. exit").unwrap();
  112. let vm = rbpf::EbpfVmNoData::new(&prog);
  113. assert_eq!(vm.prog_exec(), 0x2a);
  114. }
  115. #[test]
  116. fn test_vm_alu_bit() {
  117. let prog = assemble("
  118. mov32 r0, 0
  119. mov32 r1, 1
  120. mov32 r2, 2
  121. mov32 r3, 3
  122. mov32 r4, 4
  123. mov32 r5, 5
  124. mov32 r6, 6
  125. mov32 r7, 7
  126. mov32 r8, 8
  127. or32 r0, r5
  128. or32 r0, 0xa0
  129. and32 r0, 0xa3
  130. mov32 r9, 0x91
  131. and32 r0, r9
  132. lsh32 r0, 22
  133. lsh32 r0, r8
  134. rsh32 r0, 19
  135. rsh32 r0, r7
  136. xor32 r0, 0x03
  137. xor32 r0, r2
  138. exit").unwrap();
  139. let vm = rbpf::EbpfVmNoData::new(&prog);
  140. assert_eq!(vm.prog_exec(), 0x11);
  141. }
  142. #[test]
  143. fn test_vm_arsh32_high_shift() {
  144. let prog = assemble("
  145. mov r0, 8
  146. lddw r1, 0x100000001
  147. arsh32 r0, r1
  148. exit").unwrap();
  149. let vm = rbpf::EbpfVmNoData::new(&prog);
  150. assert_eq!(vm.prog_exec(), 0x4);
  151. }
  152. #[test]
  153. fn test_vm_arsh() {
  154. let prog = assemble("
  155. mov32 r0, 0xf8
  156. lsh32 r0, 28
  157. arsh32 r0, 16
  158. exit").unwrap();
  159. let vm = rbpf::EbpfVmNoData::new(&prog);
  160. assert_eq!(vm.prog_exec(), 0xffff8000);
  161. }
  162. #[test]
  163. fn test_vm_arsh64() {
  164. let prog = assemble("
  165. mov32 r0, 1
  166. lsh r0, 63
  167. arsh r0, 55
  168. mov32 r1, 5
  169. arsh r0, r1
  170. exit").unwrap();
  171. let vm = rbpf::EbpfVmNoData::new(&prog);
  172. assert_eq!(vm.prog_exec(), 0xfffffffffffffff8);
  173. }
  174. #[test]
  175. fn test_vm_arsh_reg() {
  176. let prog = assemble("
  177. mov32 r0, 0xf8
  178. mov32 r1, 16
  179. lsh32 r0, 28
  180. arsh32 r0, r1
  181. exit").unwrap();
  182. let vm = rbpf::EbpfVmNoData::new(&prog);
  183. assert_eq!(vm.prog_exec(), 0xffff8000);
  184. }
  185. #[test]
  186. fn test_vm_be16() {
  187. let prog = assemble("
  188. ldxh r0, [r1]
  189. be16 r0
  190. exit").unwrap();
  191. let mem = &mut [
  192. 0x11, 0x22
  193. ];
  194. let vm = rbpf::EbpfVmRaw::new(&prog);
  195. assert_eq!(vm.prog_exec(mem), 0x1122);
  196. }
  197. #[test]
  198. fn test_vm_be16_high() {
  199. let prog = assemble("
  200. ldxdw r0, [r1]
  201. be16 r0
  202. exit").unwrap();
  203. let mem = &mut [
  204. 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88
  205. ];
  206. let vm = rbpf::EbpfVmRaw::new(&prog);
  207. assert_eq!(vm.prog_exec(mem), 0x1122);
  208. }
  209. #[test]
  210. fn test_vm_be32() {
  211. let prog = assemble("
  212. ldxw r0, [r1]
  213. be32 r0
  214. exit").unwrap();
  215. let mem = &mut [
  216. 0x11, 0x22, 0x33, 0x44
  217. ];
  218. let vm = rbpf::EbpfVmRaw::new(&prog);
  219. assert_eq!(vm.prog_exec(mem), 0x11223344);
  220. }
  221. #[test]
  222. fn test_vm_be32_high() {
  223. let prog = assemble("
  224. ldxdw r0, [r1]
  225. be32 r0
  226. exit").unwrap();
  227. let mem = &mut [
  228. 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88
  229. ];
  230. let vm = rbpf::EbpfVmRaw::new(&prog);
  231. assert_eq!(vm.prog_exec(mem), 0x11223344);
  232. }
  233. #[test]
  234. fn test_vm_be64() {
  235. let prog = assemble("
  236. ldxdw r0, [r1]
  237. be64 r0
  238. exit").unwrap();
  239. let mem = &mut [
  240. 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88
  241. ];
  242. let vm = rbpf::EbpfVmRaw::new(&prog);
  243. assert_eq!(vm.prog_exec(mem), 0x1122334455667788);
  244. }
  245. #[test]
  246. fn test_vm_call() {
  247. let prog = assemble("
  248. mov r1, 1
  249. mov r2, 2
  250. mov r3, 3
  251. mov r4, 4
  252. mov r5, 5
  253. call 0
  254. exit").unwrap();
  255. let mut vm = rbpf::EbpfVmNoData::new(&prog);
  256. vm.register_helper(0, helpers::gather_bytes);
  257. assert_eq!(vm.prog_exec(), 0x0102030405);
  258. }
  259. #[test]
  260. fn test_vm_call_memfrob() {
  261. let prog = assemble("
  262. mov r6, r1
  263. add r1, 2
  264. mov r2, 4
  265. call 1
  266. ldxdw r0, [r6]
  267. be64 r0
  268. exit").unwrap();
  269. let mem = &mut [
  270. 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08
  271. ];
  272. let mut vm = rbpf::EbpfVmRaw::new(&prog);
  273. vm.register_helper(1, helpers::memfrob);
  274. assert_eq!(vm.prog_exec(mem), 0x102292e2f2c0708);
  275. }
  276. // TODO: helpers::trash_registers needs asm!().
  277. // Try this again once asm!() is available in stable.
  278. //#[test]
  279. //fn test_vm_call_save() {
  280. //let prog = &[
  281. //0xb7, 0x06, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
  282. //0xb7, 0x07, 0x00, 0x00, 0x20, 0x00, 0x00, 0x00,
  283. //0xb7, 0x08, 0x00, 0x00, 0x00, 0x03, 0x00, 0x00,
  284. //0xb7, 0x09, 0x00, 0x00, 0x00, 0x40, 0x00, 0x00,
  285. //0x85, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00,
  286. //0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  287. //0x4f, 0x60, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  288. //0x4f, 0x70, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  289. //0x4f, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  290. //0x4f, 0x90, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  291. //0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
  292. //];
  293. //let mut vm = rbpf::EbpfVmNoData::new(prog);
  294. //vm.register_helper(2, helpers::trash_registers);
  295. //assert_eq!(vm.prog_exec(), 0x4321);
  296. //}
  297. #[test]
  298. fn test_vm_div32_high_divisor() {
  299. let prog = assemble("
  300. mov r0, 12
  301. lddw r1, 0x100000004
  302. div32 r0, r1
  303. exit").unwrap();
  304. let vm = rbpf::EbpfVmNoData::new(&prog);
  305. assert_eq!(vm.prog_exec(), 0x3);
  306. }
  307. #[test]
  308. fn test_vm_div32_imm() {
  309. let prog = assemble("
  310. lddw r0, 0x10000000c
  311. div32 r0, 4
  312. exit").unwrap();
  313. let vm = rbpf::EbpfVmNoData::new(&prog);
  314. assert_eq!(vm.prog_exec(), 0x3);
  315. }
  316. #[test]
  317. fn test_vm_div32_reg() {
  318. let prog = assemble("
  319. lddw r0, 0x10000000c
  320. mov r1, 4
  321. div32 r0, r1
  322. exit").unwrap();
  323. let vm = rbpf::EbpfVmNoData::new(&prog);
  324. assert_eq!(vm.prog_exec(), 0x3);
  325. }
  326. #[test]
  327. fn test_vm_div64_imm() {
  328. let prog = assemble("
  329. mov r0, 0xc
  330. lsh r0, 32
  331. div r0, 4
  332. exit").unwrap();
  333. let vm = rbpf::EbpfVmNoData::new(&prog);
  334. assert_eq!(vm.prog_exec(), 0x300000000);
  335. }
  336. #[test]
  337. fn test_vm_div64_reg() {
  338. let prog = assemble("
  339. mov r0, 0xc
  340. lsh r0, 32
  341. mov r1, 4
  342. div r0, r1
  343. exit").unwrap();
  344. let vm = rbpf::EbpfVmNoData::new(&prog);
  345. assert_eq!(vm.prog_exec(), 0x300000000);
  346. }
  347. #[test]
  348. fn test_vm_early_exit() {
  349. let prog = assemble("
  350. mov r0, 3
  351. exit
  352. mov r0, 4
  353. exit").unwrap();
  354. let vm = rbpf::EbpfVmNoData::new(&prog);
  355. assert_eq!(vm.prog_exec(), 0x3);
  356. }
  357. // uBPF limits the number of user functions at 64. We don't.
  358. //#[test]
  359. //fn test_vm_err_call_bad_imm() {
  360. //}
  361. #[test]
  362. #[should_panic(expected = "Error: unknown helper function (id: 0x3f)")]
  363. fn test_vm_err_call_unreg() {
  364. let prog = assemble("
  365. mov r1, 1
  366. mov r2, 2
  367. mov r3, 3
  368. mov r4, 4
  369. mov r5, 5
  370. call 63
  371. exit").unwrap();
  372. let vm = rbpf::EbpfVmNoData::new(&prog);
  373. vm.prog_exec();
  374. }
  375. #[test]
  376. #[should_panic(expected = "Error: division by 0")]
  377. fn test_vm_err_div64_by_zero_reg() {
  378. let prog = assemble("
  379. mov32 r0, 1
  380. mov32 r1, 0
  381. div r0, r1
  382. exit").unwrap();
  383. let vm = rbpf::EbpfVmNoData::new(&prog);
  384. vm.prog_exec();
  385. }
  386. #[test]
  387. #[should_panic(expected = "Error: division by 0")]
  388. fn test_vm_err_div_by_zero_reg() {
  389. let prog = assemble("
  390. mov32 r0, 1
  391. mov32 r1, 0
  392. div32 r0, r1
  393. exit").unwrap();
  394. let vm = rbpf::EbpfVmNoData::new(&prog);
  395. vm.prog_exec();
  396. }
  397. #[test]
  398. #[should_panic(expected = "Error: division by 0")]
  399. fn test_vm_err_mod64_by_zero_reg() {
  400. let prog = assemble("
  401. mov32 r0, 1
  402. mov32 r1, 0
  403. mod r0, r1
  404. exit").unwrap();
  405. let vm = rbpf::EbpfVmNoData::new(&prog);
  406. vm.prog_exec();
  407. }
  408. #[test]
  409. #[should_panic(expected = "Error: division by 0")]
  410. fn test_vm_err_mod_by_zero_reg() {
  411. let prog = assemble("
  412. mov32 r0, 1
  413. mov32 r1, 0
  414. mod32 r0, r1
  415. exit").unwrap();
  416. let vm = rbpf::EbpfVmNoData::new(&prog);
  417. vm.prog_exec();
  418. }
  419. #[test]
  420. #[should_panic(expected = "Error: out of bounds memory store (insn #1)")]
  421. fn test_vm_err_stack_out_of_bound() {
  422. let prog = assemble("
  423. stb [r10], 0
  424. exit").unwrap();
  425. let vm = rbpf::EbpfVmNoData::new(&prog);
  426. vm.prog_exec();
  427. }
  428. #[test]
  429. fn test_vm_exit() {
  430. let prog = assemble("
  431. mov r0, 0
  432. exit").unwrap();
  433. let vm = rbpf::EbpfVmNoData::new(&prog);
  434. assert_eq!(vm.prog_exec(), 0x0);
  435. }
  436. #[test]
  437. fn test_vm_ja() {
  438. let prog = assemble("
  439. mov r0, 1
  440. ja +1
  441. mov r0, 2
  442. exit").unwrap();
  443. let vm = rbpf::EbpfVmNoData::new(&prog);
  444. assert_eq!(vm.prog_exec(), 0x1);
  445. }
  446. #[test]
  447. fn test_vm_jeq_imm() {
  448. let prog = assemble("
  449. mov32 r0, 0
  450. mov32 r1, 0xa
  451. jeq r1, 0xb, +4
  452. mov32 r0, 1
  453. mov32 r1, 0xb
  454. jeq r1, 0xb, +1
  455. mov32 r0, 2
  456. exit").unwrap();
  457. let vm = rbpf::EbpfVmNoData::new(&prog);
  458. assert_eq!(vm.prog_exec(), 0x1);
  459. }
  460. #[test]
  461. fn test_vm_jeq_reg() {
  462. let prog = assemble("
  463. mov32 r0, 0
  464. mov32 r1, 0xa
  465. mov32 r2, 0xb
  466. jeq r1, r2, +4
  467. mov32 r0, 1
  468. mov32 r1, 0xb
  469. jeq r1, r2, +1
  470. mov32 r0, 2
  471. exit").unwrap();
  472. let vm = rbpf::EbpfVmNoData::new(&prog);
  473. assert_eq!(vm.prog_exec(), 0x1);
  474. }
  475. #[test]
  476. fn test_vm_jge_imm() {
  477. let prog = assemble("
  478. mov32 r0, 0
  479. mov32 r1, 0xa
  480. jge r1, 0xb, +4
  481. mov32 r0, 1
  482. mov32 r1, 0xc
  483. jge r1, 0xb, +1
  484. mov32 r0, 2
  485. exit").unwrap();
  486. let vm = rbpf::EbpfVmNoData::new(&prog);
  487. assert_eq!(vm.prog_exec(), 0x1);
  488. }
  489. #[test]
  490. fn test_vm_jle_imm() {
  491. let prog = assemble("
  492. mov32 r0, 0
  493. mov32 r1, 5
  494. jle r1, 4, +1
  495. jle r1, 6, +1
  496. exit
  497. jle r1, 5, +1
  498. exit
  499. mov32 r0, 1
  500. exit").unwrap();
  501. let vm = rbpf::EbpfVmNoData::new(&prog);
  502. assert_eq!(vm.prog_exec(), 0x1);
  503. }
  504. #[test]
  505. fn test_vm_jle_reg() {
  506. let prog = assemble("
  507. mov r0, 0
  508. mov r1, 5
  509. mov r2, 4
  510. mov r3, 6
  511. jle r1, r2, +2
  512. jle r1, r1, +1
  513. exit
  514. jle r1, r3, +1
  515. exit
  516. mov r0, 1
  517. exit").unwrap();
  518. let vm = rbpf::EbpfVmNoData::new(&prog);
  519. assert_eq!(vm.prog_exec(), 0x1);
  520. }
  521. #[test]
  522. fn test_vm_jgt_imm() {
  523. let prog = assemble("
  524. mov32 r0, 0
  525. mov32 r1, 5
  526. jgt r1, 6, +2
  527. jgt r1, 5, +1
  528. jgt r1, 4, +1
  529. exit
  530. mov32 r0, 1
  531. exit").unwrap();
  532. let vm = rbpf::EbpfVmNoData::new(&prog);
  533. assert_eq!(vm.prog_exec(), 0x1);
  534. }
  535. #[test]
  536. fn test_vm_jgt_reg() {
  537. let prog = assemble("
  538. mov r0, 0
  539. mov r1, 5
  540. mov r2, 6
  541. mov r3, 4
  542. jgt r1, r2, +2
  543. jgt r1, r1, +1
  544. jgt r1, r3, +1
  545. exit
  546. mov r0, 1
  547. exit").unwrap();
  548. let vm = rbpf::EbpfVmNoData::new(&prog);
  549. assert_eq!(vm.prog_exec(), 0x1);
  550. }
  551. #[test]
  552. fn test_vm_jlt_imm() {
  553. let prog = assemble("
  554. mov32 r0, 0
  555. mov32 r1, 5
  556. jlt r1, 4, +2
  557. jlt r1, 5, +1
  558. jlt r1, 6, +1
  559. exit
  560. mov32 r0, 1
  561. exit").unwrap();
  562. let vm = rbpf::EbpfVmNoData::new(&prog);
  563. assert_eq!(vm.prog_exec(), 0x1);
  564. }
  565. #[test]
  566. fn test_vm_jlt_reg() {
  567. let prog = assemble("
  568. mov r0, 0
  569. mov r1, 5
  570. mov r2, 4
  571. mov r3, 6
  572. jlt r1, r2, +2
  573. jlt r1, r1, +1
  574. jlt r1, r3, +1
  575. exit
  576. mov r0, 1
  577. exit").unwrap();
  578. let vm = rbpf::EbpfVmNoData::new(&prog);
  579. assert_eq!(vm.prog_exec(), 0x1);
  580. }
  581. #[test]
  582. fn test_vm_jit_bounce() {
  583. let prog = assemble("
  584. mov r0, 1
  585. mov r6, r0
  586. mov r7, r6
  587. mov r8, r7
  588. mov r9, r8
  589. mov r0, r9
  590. exit").unwrap();
  591. let vm = rbpf::EbpfVmNoData::new(&prog);
  592. assert_eq!(vm.prog_exec(), 0x1);
  593. }
  594. #[test]
  595. fn test_vm_jne_reg() {
  596. let prog = assemble("
  597. mov32 r0, 0
  598. mov32 r1, 0xb
  599. mov32 r2, 0xb
  600. jne r1, r2, +4
  601. mov32 r0, 1
  602. mov32 r1, 0xa
  603. jne r1, r2, +1
  604. mov32 r0, 2
  605. exit").unwrap();
  606. let vm = rbpf::EbpfVmNoData::new(&prog);
  607. assert_eq!(vm.prog_exec(), 0x1);
  608. }
  609. #[test]
  610. fn test_vm_jset_imm() {
  611. let prog = assemble("
  612. mov32 r0, 0
  613. mov32 r1, 0x7
  614. jset r1, 0x8, +4
  615. mov32 r0, 1
  616. mov32 r1, 0x9
  617. jset r1, 0x8, +1
  618. mov32 r0, 2
  619. exit").unwrap();
  620. let vm = rbpf::EbpfVmNoData::new(&prog);
  621. assert_eq!(vm.prog_exec(), 0x1);
  622. }
  623. #[test]
  624. fn test_vm_jset_reg() {
  625. let prog = assemble("
  626. mov32 r0, 0
  627. mov32 r1, 0x7
  628. mov32 r2, 0x8
  629. jset r1, r2, +4
  630. mov32 r0, 1
  631. mov32 r1, 0x9
  632. jset r1, r2, +1
  633. mov32 r0, 2
  634. exit").unwrap();
  635. let vm = rbpf::EbpfVmNoData::new(&prog);
  636. assert_eq!(vm.prog_exec(), 0x1);
  637. }
  638. #[test]
  639. fn test_vm_jsge_imm() {
  640. let prog = assemble("
  641. mov32 r0, 0
  642. mov r1, -2
  643. jsge r1, -1, +5
  644. jsge r1, 0, +4
  645. mov32 r0, 1
  646. mov r1, -1
  647. jsge r1, -1, +1
  648. mov32 r0, 2
  649. exit").unwrap();
  650. let vm = rbpf::EbpfVmNoData::new(&prog);
  651. assert_eq!(vm.prog_exec(), 0x1);
  652. }
  653. #[test]
  654. fn test_vm_jsge_reg() {
  655. let prog = assemble("
  656. mov32 r0, 0
  657. mov r1, -2
  658. mov r2, -1
  659. mov32 r3, 0
  660. jsge r1, r2, +5
  661. jsge r1, r3, +4
  662. mov32 r0, 1
  663. mov r1, r2
  664. jsge r1, r2, +1
  665. mov32 r0, 2
  666. exit").unwrap();
  667. let vm = rbpf::EbpfVmNoData::new(&prog);
  668. assert_eq!(vm.prog_exec(), 0x1);
  669. }
  670. #[test]
  671. fn test_vm_jsle_imm() {
  672. let prog = assemble("
  673. mov32 r0, 0
  674. mov r1, -2
  675. jsle r1, -3, +1
  676. jsle r1, -1, +1
  677. exit
  678. mov32 r0, 1
  679. jsle r1, -2, +1
  680. mov32 r0, 2
  681. exit").unwrap();
  682. let vm = rbpf::EbpfVmNoData::new(&prog);
  683. assert_eq!(vm.prog_exec(), 0x1);
  684. }
  685. #[test]
  686. fn test_vm_jsle_reg() {
  687. let prog = assemble("
  688. mov32 r0, 0
  689. mov r1, -1
  690. mov r2, -2
  691. mov32 r3, 0
  692. jsle r1, r2, +1
  693. jsle r1, r3, +1
  694. exit
  695. mov32 r0, 1
  696. mov r1, r2
  697. jsle r1, r2, +1
  698. mov32 r0, 2
  699. exit").unwrap();
  700. let vm = rbpf::EbpfVmNoData::new(&prog);
  701. assert_eq!(vm.prog_exec(), 0x1);
  702. }
  703. #[test]
  704. fn test_vm_jsgt_imm() {
  705. let prog = assemble("
  706. mov32 r0, 0
  707. mov r1, -2
  708. jsgt r1, -1, +4
  709. mov32 r0, 1
  710. mov32 r1, 0
  711. jsgt r1, -1, +1
  712. mov32 r0, 2
  713. exit").unwrap();
  714. let vm = rbpf::EbpfVmNoData::new(&prog);
  715. assert_eq!(vm.prog_exec(), 0x1);
  716. }
  717. #[test]
  718. fn test_vm_jsgt_reg() {
  719. let prog = assemble("
  720. mov32 r0, 0
  721. mov r1, -2
  722. mov r2, -1
  723. jsgt r1, r2, +4
  724. mov32 r0, 1
  725. mov32 r1, 0
  726. jsgt r1, r2, +1
  727. mov32 r0, 2
  728. exit").unwrap();
  729. let vm = rbpf::EbpfVmNoData::new(&prog);
  730. assert_eq!(vm.prog_exec(), 0x1);
  731. }
  732. #[test]
  733. fn test_vm_jslt_imm() {
  734. let prog = assemble("
  735. mov32 r0, 0
  736. mov r1, -2
  737. jslt r1, -3, +2
  738. jslt r1, -2, +1
  739. jslt r1, -1, +1
  740. exit
  741. mov32 r0, 1
  742. exit").unwrap();
  743. let vm = rbpf::EbpfVmNoData::new(&prog);
  744. assert_eq!(vm.prog_exec(), 0x1);
  745. }
  746. #[test]
  747. fn test_vm_jslt_reg() {
  748. let prog = assemble("
  749. mov32 r0, 0
  750. mov r1, -2
  751. mov r2, -3
  752. mov r3, -1
  753. jslt r1, r1, +2
  754. jslt r1, r2, +1
  755. jslt r1, r3, +1
  756. exit
  757. mov32 r0, 1
  758. exit").unwrap();
  759. let vm = rbpf::EbpfVmNoData::new(&prog);
  760. assert_eq!(vm.prog_exec(), 0x1);
  761. }
  762. #[test]
  763. fn test_vm_lddw() {
  764. let prog = assemble("lddw r0, 0x1122334455667788
  765. exit").unwrap();
  766. let vm = rbpf::EbpfVmNoData::new(&prog);
  767. assert_eq!(vm.prog_exec(), 0x1122334455667788);
  768. }
  769. #[test]
  770. fn test_vm_lddw2() {
  771. let prog = assemble("
  772. lddw r0, 0x0000000080000000
  773. exit").unwrap();
  774. let vm = rbpf::EbpfVmNoData::new(&prog);
  775. assert_eq!(vm.prog_exec(), 0x80000000);
  776. }
  777. #[test]
  778. fn test_vm_ldxb_all() {
  779. let prog = assemble("
  780. mov r0, r1
  781. ldxb r9, [r0+0]
  782. lsh r9, 0
  783. ldxb r8, [r0+1]
  784. lsh r8, 4
  785. ldxb r7, [r0+2]
  786. lsh r7, 8
  787. ldxb r6, [r0+3]
  788. lsh r6, 12
  789. ldxb r5, [r0+4]
  790. lsh r5, 16
  791. ldxb r4, [r0+5]
  792. lsh r4, 20
  793. ldxb r3, [r0+6]
  794. lsh r3, 24
  795. ldxb r2, [r0+7]
  796. lsh r2, 28
  797. ldxb r1, [r0+8]
  798. lsh r1, 32
  799. ldxb r0, [r0+9]
  800. lsh r0, 36
  801. or r0, r1
  802. or r0, r2
  803. or r0, r3
  804. or r0, r4
  805. or r0, r5
  806. or r0, r6
  807. or r0, r7
  808. or r0, r8
  809. or r0, r9
  810. exit").unwrap();
  811. let mem = &mut [
  812. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  813. 0x08, 0x09
  814. ];
  815. let vm = rbpf::EbpfVmRaw::new(&prog);
  816. assert_eq!(vm.prog_exec(mem), 0x9876543210);
  817. }
  818. #[test]
  819. fn test_vm_ldxb() {
  820. let prog = assemble("
  821. ldxb r0, [r1+2]
  822. exit").unwrap();
  823. let mem = &mut [
  824. 0xaa, 0xbb, 0x11, 0xcc, 0xdd
  825. ];
  826. let vm = rbpf::EbpfVmRaw::new(&prog);
  827. assert_eq!(vm.prog_exec(mem), 0x11);
  828. }
  829. #[test]
  830. fn test_vm_ldxdw() {
  831. let prog = assemble("
  832. ldxdw r0, [r1+2]
  833. exit").unwrap();
  834. let mem = &mut [
  835. 0xaa, 0xbb, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66,
  836. 0x77, 0x88, 0xcc, 0xdd
  837. ];
  838. let vm = rbpf::EbpfVmRaw::new(&prog);
  839. assert_eq!(vm.prog_exec(mem), 0x8877665544332211);
  840. }
  841. #[test]
  842. fn test_vm_ldxh_all() {
  843. let prog = assemble("
  844. mov r0, r1
  845. ldxh r9, [r0+0]
  846. be16 r9
  847. lsh r9, 0
  848. ldxh r8, [r0+2]
  849. be16 r8
  850. lsh r8, 4
  851. ldxh r7, [r0+4]
  852. be16 r7
  853. lsh r7, 8
  854. ldxh r6, [r0+6]
  855. be16 r6
  856. lsh r6, 12
  857. ldxh r5, [r0+8]
  858. be16 r5
  859. lsh r5, 16
  860. ldxh r4, [r0+10]
  861. be16 r4
  862. lsh r4, 20
  863. ldxh r3, [r0+12]
  864. be16 r3
  865. lsh r3, 24
  866. ldxh r2, [r0+14]
  867. be16 r2
  868. lsh r2, 28
  869. ldxh r1, [r0+16]
  870. be16 r1
  871. lsh r1, 32
  872. ldxh r0, [r0+18]
  873. be16 r0
  874. lsh r0, 36
  875. or r0, r1
  876. or r0, r2
  877. or r0, r3
  878. or r0, r4
  879. or r0, r5
  880. or r0, r6
  881. or r0, r7
  882. or r0, r8
  883. or r0, r9
  884. exit").unwrap();
  885. let mem = &mut [
  886. 0x00, 0x00, 0x00, 0x01, 0x00, 0x02, 0x00, 0x03,
  887. 0x00, 0x04, 0x00, 0x05, 0x00, 0x06, 0x00, 0x07,
  888. 0x00, 0x08, 0x00, 0x09
  889. ];
  890. let vm = rbpf::EbpfVmRaw::new(&prog);
  891. assert_eq!(vm.prog_exec(mem), 0x9876543210);
  892. }
  893. #[test]
  894. fn test_vm_ldxh_all2() {
  895. let prog = assemble("
  896. mov r0, r1
  897. ldxh r9, [r0+0]
  898. be16 r9
  899. ldxh r8, [r0+2]
  900. be16 r8
  901. ldxh r7, [r0+4]
  902. be16 r7
  903. ldxh r6, [r0+6]
  904. be16 r6
  905. ldxh r5, [r0+8]
  906. be16 r5
  907. ldxh r4, [r0+10]
  908. be16 r4
  909. ldxh r3, [r0+12]
  910. be16 r3
  911. ldxh r2, [r0+14]
  912. be16 r2
  913. ldxh r1, [r0+16]
  914. be16 r1
  915. ldxh r0, [r0+18]
  916. be16 r0
  917. or r0, r1
  918. or r0, r2
  919. or r0, r3
  920. or r0, r4
  921. or r0, r5
  922. or r0, r6
  923. or r0, r7
  924. or r0, r8
  925. or r0, r9
  926. exit").unwrap();
  927. let mem = &mut [
  928. 0x00, 0x01, 0x00, 0x02, 0x00, 0x04, 0x00, 0x08,
  929. 0x00, 0x10, 0x00, 0x20, 0x00, 0x40, 0x00, 0x80,
  930. 0x01, 0x00, 0x02, 0x00
  931. ];
  932. let vm = rbpf::EbpfVmRaw::new(&prog);
  933. assert_eq!(vm.prog_exec(mem), 0x3ff);
  934. }
  935. #[test]
  936. fn test_vm_ldxh() {
  937. let prog = assemble("
  938. ldxh r0, [r1+2]
  939. exit").unwrap();
  940. let mem = &mut [
  941. 0xaa, 0xbb, 0x11, 0x22, 0xcc, 0xdd
  942. ];
  943. let vm = rbpf::EbpfVmRaw::new(&prog);
  944. assert_eq!(vm.prog_exec(mem), 0x2211);
  945. }
  946. #[test]
  947. fn test_vm_ldxh_same_reg() {
  948. let prog = assemble("
  949. mov r0, r1
  950. sth [r0], 0x1234
  951. ldxh r0, [r0]
  952. exit").unwrap();
  953. let mem = &mut [
  954. 0xff, 0xff
  955. ];
  956. let vm = rbpf::EbpfVmRaw::new(&prog);
  957. assert_eq!(vm.prog_exec(mem), 0x1234);
  958. }
  959. #[test]
  960. fn test_vm_ldxw_all() {
  961. let prog = assemble("
  962. mov r0, r1
  963. ldxw r9, [r0+0]
  964. be32 r9
  965. ldxw r8, [r0+4]
  966. be32 r8
  967. ldxw r7, [r0+8]
  968. be32 r7
  969. ldxw r6, [r0+12]
  970. be32 r6
  971. ldxw r5, [r0+16]
  972. be32 r5
  973. ldxw r4, [r0+20]
  974. be32 r4
  975. ldxw r3, [r0+24]
  976. be32 r3
  977. ldxw r2, [r0+28]
  978. be32 r2
  979. ldxw r1, [r0+32]
  980. be32 r1
  981. ldxw r0, [r0+36]
  982. be32 r0
  983. or r0, r1
  984. or r0, r2
  985. or r0, r3
  986. or r0, r4
  987. or r0, r5
  988. or r0, r6
  989. or r0, r7
  990. or r0, r8
  991. or r0, r9
  992. exit").unwrap();
  993. let mem = &mut [
  994. 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02,
  995. 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x08,
  996. 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00,
  997. 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x08, 0x00,
  998. 0x00, 0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00
  999. ];
  1000. let vm = rbpf::EbpfVmRaw::new(&prog);
  1001. assert_eq!(vm.prog_exec(mem), 0x030f0f);
  1002. }
  1003. #[test]
  1004. fn test_vm_ldxw() {
  1005. let prog = assemble("
  1006. ldxw r0, [r1+2]
  1007. exit").unwrap();
  1008. let mem = &mut [
  1009. 0xaa, 0xbb, 0x11, 0x22, 0x33, 0x44, 0xcc, 0xdd
  1010. ];
  1011. let vm = rbpf::EbpfVmRaw::new(&prog);
  1012. assert_eq!(vm.prog_exec(mem), 0x44332211);
  1013. }
  1014. #[test]
  1015. fn test_vm_le16() {
  1016. let prog = assemble("
  1017. ldxh r0, [r1]
  1018. le16 r0
  1019. exit").unwrap();
  1020. let mem = &mut [
  1021. 0x22, 0x11
  1022. ];
  1023. let vm = rbpf::EbpfVmRaw::new(&prog);
  1024. assert_eq!(vm.prog_exec(mem), 0x1122);
  1025. }
  1026. #[test]
  1027. fn test_vm_le32() {
  1028. let prog = assemble("
  1029. ldxw r0, [r1]
  1030. le32 r0
  1031. exit").unwrap();
  1032. let mem = &mut [
  1033. 0x44, 0x33, 0x22, 0x11
  1034. ];
  1035. let vm = rbpf::EbpfVmRaw::new(&prog);
  1036. assert_eq!(vm.prog_exec(mem), 0x11223344);
  1037. }
  1038. #[test]
  1039. fn test_vm_le64() {
  1040. let prog = assemble("
  1041. ldxdw r0, [r1]
  1042. le64 r0
  1043. exit").unwrap();
  1044. let mem = &mut [
  1045. 0x88, 0x77, 0x66, 0x55, 0x44, 0x33, 0x22, 0x11
  1046. ];
  1047. let vm = rbpf::EbpfVmRaw::new(&prog);
  1048. assert_eq!(vm.prog_exec(mem), 0x1122334455667788);
  1049. }
  1050. #[test]
  1051. fn test_vm_lsh_reg() {
  1052. let prog = assemble("
  1053. mov r0, 0x1
  1054. mov r7, 4
  1055. lsh r0, r7
  1056. exit").unwrap();
  1057. let vm = rbpf::EbpfVmNoData::new(&prog);
  1058. assert_eq!(vm.prog_exec(), 0x10);
  1059. }
  1060. #[test]
  1061. fn test_vm_mod() {
  1062. let prog = assemble("
  1063. mov32 r0, 5748
  1064. mod32 r0, 92
  1065. mov32 r1, 13
  1066. mod32 r0, r1
  1067. exit").unwrap();
  1068. let vm = rbpf::EbpfVmNoData::new(&prog);
  1069. assert_eq!(vm.prog_exec(), 0x5);
  1070. }
  1071. #[test]
  1072. fn test_vm_mod32() {
  1073. let prog = assemble("
  1074. lddw r0, 0x100000003
  1075. mod32 r0, 3
  1076. exit").unwrap();
  1077. let vm = rbpf::EbpfVmNoData::new(&prog);
  1078. assert_eq!(vm.prog_exec(), 0x0);
  1079. }
  1080. #[test]
  1081. fn test_vm_mod64() {
  1082. let prog = assemble("
  1083. mov32 r0, -1316649930
  1084. lsh r0, 32
  1085. or r0, 0x100dc5c8
  1086. mov32 r1, 0xdde263e
  1087. lsh r1, 32
  1088. or r1, 0x3cbef7f3
  1089. mod r0, r1
  1090. mod r0, 0x658f1778
  1091. exit").unwrap();
  1092. let vm = rbpf::EbpfVmNoData::new(&prog);
  1093. assert_eq!(vm.prog_exec(), 0x30ba5a04);
  1094. }
  1095. #[test]
  1096. fn test_vm_mov() {
  1097. let prog = assemble("
  1098. mov32 r1, 1
  1099. mov32 r0, r1
  1100. exit").unwrap();
  1101. let vm = rbpf::EbpfVmNoData::new(&prog);
  1102. assert_eq!(vm.prog_exec(), 0x1);
  1103. }
  1104. #[test]
  1105. fn test_vm_mul32_imm() {
  1106. let prog = assemble("
  1107. mov r0, 3
  1108. mul32 r0, 4
  1109. exit").unwrap();
  1110. let vm = rbpf::EbpfVmNoData::new(&prog);
  1111. assert_eq!(vm.prog_exec(), 0xc);
  1112. }
  1113. #[test]
  1114. fn test_vm_mul32_reg() {
  1115. let prog = assemble("
  1116. mov r0, 3
  1117. mov r1, 4
  1118. mul32 r0, r1
  1119. exit").unwrap();
  1120. let vm = rbpf::EbpfVmNoData::new(&prog);
  1121. assert_eq!(vm.prog_exec(), 0xc);
  1122. }
  1123. #[test]
  1124. fn test_vm_mul32_reg_overflow() {
  1125. let prog = assemble("
  1126. mov r0, 0x40000001
  1127. mov r1, 4
  1128. mul32 r0, r1
  1129. exit").unwrap();
  1130. let vm = rbpf::EbpfVmNoData::new(&prog);
  1131. assert_eq!(vm.prog_exec(), 0x4);
  1132. }
  1133. #[test]
  1134. fn test_vm_mul64_imm() {
  1135. let prog = assemble("
  1136. mov r0, 0x40000001
  1137. mul r0, 4
  1138. exit").unwrap();
  1139. let vm = rbpf::EbpfVmNoData::new(&prog);
  1140. assert_eq!(vm.prog_exec(), 0x100000004);
  1141. }
  1142. #[test]
  1143. fn test_vm_mul64_reg() {
  1144. let prog = assemble("
  1145. mov r0, 0x40000001
  1146. mov r1, 4
  1147. mul r0, r1
  1148. exit").unwrap();
  1149. let vm = rbpf::EbpfVmNoData::new(&prog);
  1150. assert_eq!(vm.prog_exec(), 0x100000004);
  1151. }
  1152. #[test]
  1153. fn test_vm_mul_loop() {
  1154. let prog = assemble("
  1155. mov r0, 0x7
  1156. add r1, 0xa
  1157. lsh r1, 0x20
  1158. rsh r1, 0x20
  1159. jeq r1, 0x0, +4
  1160. mov r0, 0x7
  1161. mul r0, 0x7
  1162. add r1, -1
  1163. jne r1, 0x0, -3
  1164. exit").unwrap();
  1165. let vm = rbpf::EbpfVmNoData::new(&prog);
  1166. assert_eq!(vm.prog_exec(), 0x75db9c97);
  1167. }
  1168. #[test]
  1169. fn test_vm_neg64() {
  1170. let prog = assemble("
  1171. mov32 r0, 2
  1172. neg r0
  1173. exit").unwrap();
  1174. let vm = rbpf::EbpfVmNoData::new(&prog);
  1175. assert_eq!(vm.prog_exec(), 0xfffffffffffffffe);
  1176. }
  1177. #[test]
  1178. fn test_vm_neg() {
  1179. let prog = assemble("
  1180. mov32 r0, 2
  1181. neg32 r0
  1182. exit").unwrap();
  1183. let vm = rbpf::EbpfVmNoData::new(&prog);
  1184. assert_eq!(vm.prog_exec(), 0xfffffffe);
  1185. }
  1186. #[test]
  1187. fn test_vm_prime() {
  1188. let prog = assemble("
  1189. mov r1, 67
  1190. mov r0, 0x1
  1191. mov r2, 0x2
  1192. jgt r1, 0x2, +4
  1193. ja +10
  1194. add r2, 0x1
  1195. mov r0, 0x1
  1196. jge r2, r1, +7
  1197. mov r3, r1
  1198. div r3, r2
  1199. mul r3, r2
  1200. mov r4, r1
  1201. sub r4, r3
  1202. mov r0, 0x0
  1203. jne r4, 0x0, -10
  1204. exit").unwrap();
  1205. let vm = rbpf::EbpfVmNoData::new(&prog);
  1206. assert_eq!(vm.prog_exec(), 0x1);
  1207. }
  1208. #[test]
  1209. fn test_vm_rhs32() {
  1210. let prog = assemble("
  1211. xor r0, r0
  1212. sub r0, 1
  1213. rsh32 r0, 8
  1214. exit").unwrap();
  1215. let vm = rbpf::EbpfVmNoData::new(&prog);
  1216. assert_eq!(vm.prog_exec(), 0x00ffffff);
  1217. }
  1218. #[test]
  1219. fn test_vm_rsh_reg() {
  1220. let prog = assemble("
  1221. mov r0, 0x10
  1222. mov r7, 4
  1223. rsh r0, r7
  1224. exit").unwrap();
  1225. let vm = rbpf::EbpfVmNoData::new(&prog);
  1226. assert_eq!(vm.prog_exec(), 0x1);
  1227. }
  1228. #[test]
  1229. fn test_vm_stack() {
  1230. let prog = assemble("
  1231. mov r1, 51
  1232. stdw [r10-16], 0xab
  1233. stdw [r10-8], 0xcd
  1234. and r1, 1
  1235. lsh r1, 3
  1236. mov r2, r10
  1237. add r2, r1
  1238. ldxdw r0, [r2-16]
  1239. exit").unwrap();
  1240. let vm = rbpf::EbpfVmNoData::new(&prog);
  1241. assert_eq!(vm.prog_exec(), 0xcd);
  1242. }
  1243. #[test]
  1244. fn test_vm_stack2() {
  1245. let prog = assemble("
  1246. stb [r10-4], 0x01
  1247. stb [r10-3], 0x02
  1248. stb [r10-2], 0x03
  1249. stb [r10-1], 0x04
  1250. mov r1, r10
  1251. mov r2, 0x4
  1252. sub r1, r2
  1253. call 1
  1254. mov r1, 0
  1255. ldxb r2, [r10-4]
  1256. ldxb r3, [r10-3]
  1257. ldxb r4, [r10-2]
  1258. ldxb r5, [r10-1]
  1259. call 0
  1260. xor r0, 0x2a2a2a2a
  1261. exit").unwrap();
  1262. let mut vm = rbpf::EbpfVmNoData::new(&prog);
  1263. vm.register_helper(0, helpers::gather_bytes);
  1264. vm.register_helper(1, helpers::memfrob);
  1265. assert_eq!(vm.prog_exec(), 0x01020304);
  1266. }
  1267. #[test]
  1268. fn test_vm_stb() {
  1269. let prog = assemble("
  1270. stb [r1+2], 0x11
  1271. ldxb r0, [r1+2]
  1272. exit").unwrap();
  1273. let mem = &mut [
  1274. 0xaa, 0xbb, 0xff, 0xcc, 0xdd
  1275. ];
  1276. let vm = rbpf::EbpfVmRaw::new(&prog);
  1277. assert_eq!(vm.prog_exec(mem), 0x11);
  1278. }
  1279. #[test]
  1280. fn test_vm_stdw() {
  1281. let prog = assemble("
  1282. stdw [r1+2], 0x44332211
  1283. ldxdw r0, [r1+2]
  1284. exit").unwrap();
  1285. let mem = &mut [
  1286. 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
  1287. 0xff, 0xff, 0xcc, 0xdd
  1288. ];
  1289. let vm = rbpf::EbpfVmRaw::new(&prog);
  1290. assert_eq!(vm.prog_exec(mem), 0x44332211);
  1291. }
  1292. #[test]
  1293. fn test_vm_sth() {
  1294. let prog = assemble("
  1295. sth [r1+2], 0x2211
  1296. ldxh r0, [r1+2]
  1297. exit").unwrap();
  1298. let mem = &mut [
  1299. 0xaa, 0xbb, 0xff, 0xff, 0xcc, 0xdd
  1300. ];
  1301. let vm = rbpf::EbpfVmRaw::new(&prog);
  1302. assert_eq!(vm.prog_exec(mem), 0x2211);
  1303. }
  1304. #[test]
  1305. fn test_vm_string_stack() {
  1306. let prog = assemble("
  1307. mov r1, 0x78636261
  1308. stxw [r10-8], r1
  1309. mov r6, 0x0
  1310. stxb [r10-4], r6
  1311. stxb [r10-12], r6
  1312. mov r1, 0x79636261
  1313. stxw [r10-16], r1
  1314. mov r1, r10
  1315. add r1, -8
  1316. mov r2, r1
  1317. call 0x4
  1318. mov r1, r0
  1319. mov r0, 0x1
  1320. lsh r1, 0x20
  1321. rsh r1, 0x20
  1322. jne r1, 0x0, +11
  1323. mov r1, r10
  1324. add r1, -8
  1325. mov r2, r10
  1326. add r2, -16
  1327. call 0x4
  1328. mov r1, r0
  1329. lsh r1, 0x20
  1330. rsh r1, 0x20
  1331. mov r0, 0x1
  1332. jeq r1, r6, +1
  1333. mov r0, 0x0
  1334. exit").unwrap();
  1335. let mut vm = rbpf::EbpfVmNoData::new(&prog);
  1336. vm.register_helper(4, helpers::strcmp);
  1337. assert_eq!(vm.prog_exec(), 0x0);
  1338. }
  1339. #[test]
  1340. fn test_vm_stw() {
  1341. let prog = assemble("
  1342. stw [r1+2], 0x44332211
  1343. ldxw r0, [r1+2]
  1344. exit").unwrap();
  1345. let mem = &mut [
  1346. 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd
  1347. ];
  1348. let vm = rbpf::EbpfVmRaw::new(&prog);
  1349. assert_eq!(vm.prog_exec(mem), 0x44332211);
  1350. }
  1351. #[test]
  1352. fn test_vm_stxb() {
  1353. let prog = assemble("
  1354. mov32 r2, 0x11
  1355. stxb [r1+2], r2
  1356. ldxb r0, [r1+2]
  1357. exit").unwrap();
  1358. let mem = &mut [
  1359. 0xaa, 0xbb, 0xff, 0xcc, 0xdd
  1360. ];
  1361. let vm = rbpf::EbpfVmRaw::new(&prog);
  1362. assert_eq!(vm.prog_exec(mem), 0x11);
  1363. }
  1364. #[test]
  1365. fn test_vm_stxb_all() {
  1366. let prog = assemble("
  1367. mov r0, 0xf0
  1368. mov r2, 0xf2
  1369. mov r3, 0xf3
  1370. mov r4, 0xf4
  1371. mov r5, 0xf5
  1372. mov r6, 0xf6
  1373. mov r7, 0xf7
  1374. mov r8, 0xf8
  1375. stxb [r1], r0
  1376. stxb [r1+1], r2
  1377. stxb [r1+2], r3
  1378. stxb [r1+3], r4
  1379. stxb [r1+4], r5
  1380. stxb [r1+5], r6
  1381. stxb [r1+6], r7
  1382. stxb [r1+7], r8
  1383. ldxdw r0, [r1]
  1384. be64 r0
  1385. exit").unwrap();
  1386. let mem = &mut [
  1387. 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff
  1388. ];
  1389. let vm = rbpf::EbpfVmRaw::new(&prog);
  1390. assert_eq!(vm.prog_exec(mem), 0xf0f2f3f4f5f6f7f8);
  1391. }
  1392. #[test]
  1393. fn test_vm_stxb_all2() {
  1394. let prog = assemble("
  1395. mov r0, r1
  1396. mov r1, 0xf1
  1397. mov r9, 0xf9
  1398. stxb [r0], r1
  1399. stxb [r0+1], r9
  1400. ldxh r0, [r0]
  1401. be16 r0
  1402. exit").unwrap();
  1403. let mem = &mut [
  1404. 0xff, 0xff
  1405. ];
  1406. let vm = rbpf::EbpfVmRaw::new(&prog);
  1407. assert_eq!(vm.prog_exec(mem), 0xf1f9);
  1408. }
  1409. #[test]
  1410. fn test_vm_stxb_chain() {
  1411. let prog = assemble("
  1412. mov r0, r1
  1413. ldxb r9, [r0+0]
  1414. stxb [r0+1], r9
  1415. ldxb r8, [r0+1]
  1416. stxb [r0+2], r8
  1417. ldxb r7, [r0+2]
  1418. stxb [r0+3], r7
  1419. ldxb r6, [r0+3]
  1420. stxb [r0+4], r6
  1421. ldxb r5, [r0+4]
  1422. stxb [r0+5], r5
  1423. ldxb r4, [r0+5]
  1424. stxb [r0+6], r4
  1425. ldxb r3, [r0+6]
  1426. stxb [r0+7], r3
  1427. ldxb r2, [r0+7]
  1428. stxb [r0+8], r2
  1429. ldxb r1, [r0+8]
  1430. stxb [r0+9], r1
  1431. ldxb r0, [r0+9]
  1432. exit").unwrap();
  1433. let mem = &mut [
  1434. 0x2a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1435. 0x00, 0x00
  1436. ];
  1437. let vm = rbpf::EbpfVmRaw::new(&prog);
  1438. assert_eq!(vm.prog_exec(mem), 0x2a);
  1439. }
  1440. #[test]
  1441. fn test_vm_stxdw() {
  1442. let prog = assemble("
  1443. mov r2, -2005440939
  1444. lsh r2, 32
  1445. or r2, 0x44332211
  1446. stxdw [r1+2], r2
  1447. ldxdw r0, [r1+2]
  1448. exit").unwrap();
  1449. let mem = &mut [
  1450. 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
  1451. 0xff, 0xff, 0xcc, 0xdd
  1452. ];
  1453. let vm = rbpf::EbpfVmRaw::new(&prog);
  1454. assert_eq!(vm.prog_exec(mem), 0x8877665544332211);
  1455. }
  1456. #[test]
  1457. fn test_vm_stxh() {
  1458. let prog = assemble("
  1459. mov32 r2, 0x2211
  1460. stxh [r1+2], r2
  1461. ldxh r0, [r1+2]
  1462. exit").unwrap();
  1463. let mem = &mut [
  1464. 0xaa, 0xbb, 0xff, 0xff, 0xcc, 0xdd
  1465. ];
  1466. let vm = rbpf::EbpfVmRaw::new(&prog);
  1467. assert_eq!(vm.prog_exec(mem), 0x2211);
  1468. }
  1469. #[test]
  1470. fn test_vm_stxw() {
  1471. let prog = assemble("
  1472. mov32 r2, 0x44332211
  1473. stxw [r1+2], r2
  1474. ldxw r0, [r1+2]
  1475. exit").unwrap();
  1476. let mem = &mut [
  1477. 0xaa, 0xbb, 0xff, 0xff, 0xff, 0xff, 0xcc, 0xdd
  1478. ];
  1479. let vm = rbpf::EbpfVmRaw::new(&prog);
  1480. assert_eq!(vm.prog_exec(mem), 0x44332211);
  1481. }
  1482. #[test]
  1483. fn test_vm_subnet() {
  1484. let prog = assemble("
  1485. mov r2, 0xe
  1486. ldxh r3, [r1+12]
  1487. jne r3, 0x81, +2
  1488. mov r2, 0x12
  1489. ldxh r3, [r1+16]
  1490. and r3, 0xffff
  1491. jne r3, 0x8, +5
  1492. add r1, r2
  1493. mov r0, 0x1
  1494. ldxw r1, [r1+16]
  1495. and r1, 0xffffff
  1496. jeq r1, 0x1a8c0, +1
  1497. mov r0, 0x0
  1498. exit").unwrap();
  1499. let mem = &mut [
  1500. 0x00, 0x00, 0xc0, 0x9f, 0xa0, 0x97, 0x00, 0xa0,
  1501. 0xcc, 0x3b, 0xbf, 0xfa, 0x08, 0x00, 0x45, 0x10,
  1502. 0x00, 0x3c, 0x46, 0x3c, 0x40, 0x00, 0x40, 0x06,
  1503. 0x73, 0x1c, 0xc0, 0xa8, 0x01, 0x02, 0xc0, 0xa8,
  1504. 0x01, 0x01, 0x06, 0x0e, 0x00, 0x17, 0x99, 0xc5,
  1505. 0xa0, 0xec, 0x00, 0x00, 0x00, 0x00, 0xa0, 0x02,
  1506. 0x7d, 0x78, 0xe0, 0xa3, 0x00, 0x00, 0x02, 0x04,
  1507. 0x05, 0xb4, 0x04, 0x02, 0x08, 0x0a, 0x00, 0x9c,
  1508. 0x27, 0x24, 0x00, 0x00, 0x00, 0x00, 0x01, 0x03,
  1509. 0x03, 0x00
  1510. ];
  1511. let vm = rbpf::EbpfVmRaw::new(&prog);
  1512. assert_eq!(vm.prog_exec(mem), 0x1);
  1513. }
  1514. const PROG_TCP_PORT_80: [u8;152] = [
  1515. 0x71, 0x12, 0x0c, 0x00, 0x00, 0x00, 0x00, 0x00,
  1516. 0x71, 0x13, 0x0d, 0x00, 0x00, 0x00, 0x00, 0x00,
  1517. 0x67, 0x03, 0x00, 0x00, 0x08, 0x00, 0x00, 0x00,
  1518. 0x4f, 0x23, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1519. 0xb7, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1520. 0x55, 0x03, 0x0c, 0x00, 0x08, 0x00, 0x00, 0x00,
  1521. 0x71, 0x12, 0x17, 0x00, 0x00, 0x00, 0x00, 0x00,
  1522. 0x55, 0x02, 0x0a, 0x00, 0x06, 0x00, 0x00, 0x00,
  1523. 0x71, 0x12, 0x0e, 0x00, 0x00, 0x00, 0x00, 0x00,
  1524. 0x07, 0x01, 0x00, 0x00, 0x0e, 0x00, 0x00, 0x00,
  1525. 0x57, 0x02, 0x00, 0x00, 0x0f, 0x00, 0x00, 0x00,
  1526. 0x67, 0x02, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00,
  1527. 0x0f, 0x21, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1528. 0x69, 0x12, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00,
  1529. 0x15, 0x02, 0x02, 0x00, 0x00, 0x50, 0x00, 0x00,
  1530. 0x69, 0x11, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1531. 0x55, 0x01, 0x01, 0x00, 0x00, 0x50, 0x00, 0x00,
  1532. 0xb7, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
  1533. 0x95, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
  1534. ];
  1535. #[test]
  1536. fn test_vm_tcp_port80_match() {
  1537. let mem = &mut [
  1538. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06,
  1539. 0x07, 0x08, 0x09, 0x0a, 0x08, 0x00, 0x45, 0x00,
  1540. 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06,
  1541. 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8,
  1542. 0x00, 0x02, 0x27, 0x10, 0x00, 0x50, 0x00, 0x00,
  1543. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02,
  1544. 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44,
  1545. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1546. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1547. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1548. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1549. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1550. 0x44, 0x44, 0x44, 0x44
  1551. ];
  1552. let prog = &PROG_TCP_PORT_80;
  1553. let vm = rbpf::EbpfVmRaw::new(prog);
  1554. assert_eq!(vm.prog_exec(mem), 0x1);
  1555. }
  1556. #[test]
  1557. fn test_vm_tcp_port80_nomatch() {
  1558. let mem = &mut [
  1559. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06,
  1560. 0x07, 0x08, 0x09, 0x0a, 0x08, 0x00, 0x45, 0x00,
  1561. 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06,
  1562. 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8,
  1563. 0x00, 0x02, 0x00, 0x16, 0x27, 0x10, 0x00, 0x00,
  1564. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x51, 0x02,
  1565. 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44,
  1566. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1567. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1568. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1569. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1570. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1571. 0x44, 0x44, 0x44, 0x44
  1572. ];
  1573. let prog = &PROG_TCP_PORT_80;
  1574. let vm = rbpf::EbpfVmRaw::new(prog);
  1575. assert_eq!(vm.prog_exec(mem), 0x0);
  1576. }
  1577. #[test]
  1578. fn test_vm_tcp_port80_nomatch_ethertype() {
  1579. let mem = &mut [
  1580. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06,
  1581. 0x07, 0x08, 0x09, 0x0a, 0x08, 0x01, 0x45, 0x00,
  1582. 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x06,
  1583. 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8,
  1584. 0x00, 0x02, 0x27, 0x10, 0x00, 0x50, 0x00, 0x00,
  1585. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02,
  1586. 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44,
  1587. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1588. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1589. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1590. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1591. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1592. 0x44, 0x44, 0x44, 0x44
  1593. ];
  1594. let prog = &PROG_TCP_PORT_80;
  1595. let vm = rbpf::EbpfVmRaw::new(prog);
  1596. assert_eq!(vm.prog_exec(mem), 0x0);
  1597. }
  1598. #[test]
  1599. fn test_vm_tcp_port80_nomatch_proto() {
  1600. let mem = &mut [
  1601. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x00, 0x06,
  1602. 0x07, 0x08, 0x09, 0x0a, 0x08, 0x00, 0x45, 0x00,
  1603. 0x00, 0x56, 0x00, 0x01, 0x00, 0x00, 0x40, 0x11,
  1604. 0xf9, 0x4d, 0xc0, 0xa8, 0x00, 0x01, 0xc0, 0xa8,
  1605. 0x00, 0x02, 0x27, 0x10, 0x00, 0x50, 0x00, 0x00,
  1606. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02,
  1607. 0x20, 0x00, 0xc5, 0x18, 0x00, 0x00, 0x44, 0x44,
  1608. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1609. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1610. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1611. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1612. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1613. 0x44, 0x44, 0x44, 0x44
  1614. ];
  1615. let prog = &PROG_TCP_PORT_80;
  1616. let vm = rbpf::EbpfVmRaw::new(prog);
  1617. assert_eq!(vm.prog_exec(mem), 0x0);
  1618. }
  1619. #[test]
  1620. fn test_vm_tcp_sack_match() {
  1621. let mut mem = TCP_SACK_MATCH.to_vec();
  1622. let prog = assemble(TCP_SACK_ASM).unwrap();
  1623. let vm = rbpf::EbpfVmRaw::new(&prog);
  1624. assert_eq!(vm.prog_exec(mem.as_mut_slice()), 0x1);
  1625. }
  1626. #[test]
  1627. fn test_vm_tcp_sack_nomatch() {
  1628. let mut mem = TCP_SACK_NOMATCH.to_vec();
  1629. let prog = assemble(TCP_SACK_ASM).unwrap();
  1630. let vm = rbpf::EbpfVmRaw::new(&prog);
  1631. assert_eq!(vm.prog_exec(mem.as_mut_slice()), 0x0);
  1632. }