core.rs 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773
  1. use rand::Rng;
  2. use std::fs::File;
  3. use std::io::Read;
  4. use std::rc::Rc;
  5. use std::time::{SystemTime, UNIX_EPOCH};
  6. use crate::printer::pr_seq;
  7. use crate::reader::read_str;
  8. use crate::types::MalErr::ErrMalVal;
  9. use crate::types::MalVal::{
  10. Atom, Bool, Func, Hash, Int, List, MalFunc, Nil, Str, Sym, Vector, ZKScalar,
  11. };
  12. use crate::types::{MalArgs, MalRet, MalVal, _assoc, _dissoc, atom, error, func, hash_map};
  13. use bls12_381;
  14. use ff::{Field, PrimeField};
  15. use rand::rngs::OsRng;
  16. use sapvi::bls_extensions::BlsStringConversion;
  17. use std::ops::{AddAssign, MulAssign, SubAssign};
  18. macro_rules! fn_t_int_int {
  19. ($ret:ident, $fn:expr) => {{
  20. |a: MalArgs| match (a[0].clone(), a[1].clone()) {
  21. (Int(a0), Int(a1)) => Ok($ret($fn(a0, a1))),
  22. _ => error("expecting (int,int) args"),
  23. }
  24. }};
  25. }
  26. macro_rules! fn_is_type {
  27. ($($ps:pat),*) => {{
  28. |a:MalArgs| { Ok(Bool(match a[0] { $($ps => true,)* _ => false})) }
  29. }};
  30. ($p:pat if $e:expr) => {{
  31. |a:MalArgs| { Ok(Bool(match a[0] { $p if $e => true, _ => false})) }
  32. }};
  33. ($p:pat if $e:expr,$($ps:pat),*) => {{
  34. |a:MalArgs| { Ok(Bool(match a[0] { $p if $e => true, $($ps => true,)* _ => false})) }
  35. }};
  36. }
  37. macro_rules! fn_str {
  38. ($fn:expr) => {{
  39. |a: MalArgs| match a[0].clone() {
  40. Str(a0) => $fn(a0),
  41. _ => error("expecting (str) arg"),
  42. }
  43. }};
  44. }
  45. fn symbol(a: MalArgs) -> MalRet {
  46. match a[0] {
  47. Str(ref s) => Ok(Sym(s.to_string())),
  48. _ => error("illegal symbol call"),
  49. }
  50. }
  51. fn slurp(f: String) -> MalRet {
  52. let mut s = String::new();
  53. match File::open(f).and_then(|mut f| f.read_to_string(&mut s)) {
  54. Ok(_) => Ok(Str(s)),
  55. Err(e) => error(&e.to_string()),
  56. }
  57. }
  58. fn time_ms(_a: MalArgs) -> MalRet {
  59. let ms_e = match SystemTime::now().duration_since(UNIX_EPOCH) {
  60. Ok(d) => d,
  61. Err(e) => return error(&format!("{:?}", e)),
  62. };
  63. Ok(Int(
  64. ms_e.as_secs() as i64 * 1000 + ms_e.subsec_nanos() as i64 / 1_000_000
  65. ))
  66. }
  67. fn get(a: MalArgs) -> MalRet {
  68. match (a[0].clone(), a[1].clone()) {
  69. (Nil, _) => Ok(Nil),
  70. (Hash(ref hm, _), Str(ref s)) => match hm.get(s) {
  71. Some(mv) => Ok(mv.clone()),
  72. None => Ok(Nil),
  73. },
  74. _ => error("illegal get args"),
  75. }
  76. }
  77. fn assoc(a: MalArgs) -> MalRet {
  78. match a[0] {
  79. Hash(ref hm, _) => _assoc((**hm).clone(), a[1..].to_vec()),
  80. _ => error("assoc on non-Hash Map"),
  81. }
  82. }
  83. fn dissoc(a: MalArgs) -> MalRet {
  84. match a[0] {
  85. Hash(ref hm, _) => _dissoc((**hm).clone(), a[1..].to_vec()),
  86. _ => error("dissoc on non-Hash Map"),
  87. }
  88. }
  89. fn contains_q(a: MalArgs) -> MalRet {
  90. match (a[0].clone(), a[1].clone()) {
  91. (Hash(ref hm, _), Str(ref s)) => Ok(Bool(hm.contains_key(s))),
  92. _ => error("illegal get args"),
  93. }
  94. }
  95. fn keys(a: MalArgs) -> MalRet {
  96. match a[0] {
  97. Hash(ref hm, _) => Ok(list!(hm.keys().map(|k| { Str(k.to_string()) }).collect())),
  98. _ => error("keys requires Hash Map"),
  99. }
  100. }
  101. fn vals(a: MalArgs) -> MalRet {
  102. match a[0] {
  103. Hash(ref hm, _) => Ok(list!(hm.values().map(|v| { v.clone() }).collect())),
  104. _ => error("keys requires Hash Map"),
  105. }
  106. }
  107. fn vec(a: MalArgs) -> MalRet {
  108. match a[0] {
  109. List(ref v, _) | Vector(ref v, _) => Ok(vector!(v.to_vec())),
  110. _ => error("non-seq passed to vec"),
  111. }
  112. }
  113. fn cons(a: MalArgs) -> MalRet {
  114. match a[1].clone() {
  115. List(v, _) | Vector(v, _) => {
  116. let mut new_v = vec![a[0].clone()];
  117. new_v.extend_from_slice(&v);
  118. Ok(list!(new_v.to_vec()))
  119. }
  120. _ => error("cons expects seq as second arg"),
  121. }
  122. }
  123. fn concat(a: MalArgs) -> MalRet {
  124. let mut new_v = vec![];
  125. for seq in a.iter() {
  126. match seq {
  127. List(v, _) | Vector(v, _) => new_v.extend_from_slice(v),
  128. _ => return error("non-seq passed to concat"),
  129. }
  130. }
  131. Ok(list!(new_v.to_vec()))
  132. }
  133. fn nth(a: MalArgs) -> MalRet {
  134. match (a[0].clone(), a[1].clone()) {
  135. (List(seq, _), Int(idx)) | (Vector(seq, _), Int(idx)) => {
  136. if seq.len() <= idx as usize {
  137. return error("nth: index out of range");
  138. }
  139. Ok(seq[idx as usize].clone())
  140. }
  141. _ => error("invalid args to nth"),
  142. }
  143. }
  144. fn unpack_bits(a: MalArgs) -> MalRet {
  145. let mut result = vec![];
  146. match a[0].clone() {
  147. Str(ref s) => {
  148. let value = bls12_381::Scalar::from_string(s);
  149. for (_, bit) in value.to_le_bits().into_iter().enumerate() {
  150. match bit {
  151. true => result.push(bls12_381::Scalar::one()),
  152. false => result.push(bls12_381::Scalar::zero()),
  153. }
  154. }
  155. Ok(list!(result
  156. .iter()
  157. .map(|a| Str(std::string::ToString::to_string(&a)[2..].to_string()))
  158. .collect::<Vec<MalVal>>()))
  159. }
  160. ZKScalar(ref s) => {
  161. for (_, bit) in s.to_le_bits().into_iter().enumerate() {
  162. match bit {
  163. true => result.push(bls12_381::Scalar::one()),
  164. false => result.push(bls12_381::Scalar::zero()),
  165. }
  166. }
  167. Ok(list!(result
  168. .iter()
  169. .map(|a| Str(std::string::ToString::to_string(&a)[2..].to_string()))
  170. .collect::<Vec<MalVal>>()))
  171. }
  172. _ => error("invalid args to unpack-bits"),
  173. }
  174. }
  175. fn last(a: MalArgs) -> MalRet {
  176. match a[0].clone() {
  177. List(ref seq, _) | Vector(ref seq, _) if seq.len() == 0 => Ok(Nil),
  178. List(ref seq, _) | Vector(ref seq, _) => Ok(seq[seq.len() - 1].clone()),
  179. Nil => Ok(Nil),
  180. _ => error("invalid args to first"),
  181. }
  182. }
  183. fn first(a: MalArgs) -> MalRet {
  184. match a[0].clone() {
  185. List(ref seq, _) | Vector(ref seq, _) if seq.len() == 0 => Ok(Nil),
  186. List(ref seq, _) | Vector(ref seq, _) => Ok(seq[0].clone()),
  187. Nil => Ok(Nil),
  188. _ => error("invalid args to first"),
  189. }
  190. }
  191. fn second(a: MalArgs) -> MalRet {
  192. match a[0].clone() {
  193. List(ref seq, _) | Vector(ref seq, _) if seq.len() < 2 => Ok(Nil),
  194. List(ref seq, _) | Vector(ref seq, _) => Ok(seq[1].clone()),
  195. Nil => Ok(Nil),
  196. _ => error("invalid args to first"),
  197. }
  198. }
  199. fn rest(a: MalArgs) -> MalRet {
  200. match a[0].clone() {
  201. List(ref seq, _) | Vector(ref seq, _) => {
  202. if seq.len() > 1 {
  203. Ok(list!(seq[1..].to_vec()))
  204. } else {
  205. Ok(list![])
  206. }
  207. }
  208. Nil => Ok(list![]),
  209. _ => error("invalid args to first"),
  210. }
  211. }
  212. fn apply(a: MalArgs) -> MalRet {
  213. match a[a.len() - 1] {
  214. List(ref v, _) | Vector(ref v, _) => {
  215. let f = &a[0];
  216. let mut fargs = a[1..a.len() - 1].to_vec();
  217. fargs.extend_from_slice(&v);
  218. f.apply(fargs)
  219. }
  220. _ => error("apply called with non-seq"),
  221. }
  222. }
  223. fn map(a: MalArgs) -> MalRet {
  224. match a[1] {
  225. List(ref v, _) | Vector(ref v, _) => {
  226. let mut res = vec![];
  227. for mv in v.iter() {
  228. res.push(a[0].apply(vec![mv.clone()])?)
  229. }
  230. Ok(list!(res))
  231. }
  232. _ => error("map called with non-seq"),
  233. }
  234. }
  235. fn conj(a: MalArgs) -> MalRet {
  236. match a[0] {
  237. List(ref v, _) => {
  238. let sl = a[1..]
  239. .iter()
  240. .rev()
  241. .map(|a| a.clone())
  242. .collect::<Vec<MalVal>>();
  243. Ok(list!([&sl[..], v].concat()))
  244. }
  245. Vector(ref v, _) => Ok(vector!([v, &a[1..]].concat())),
  246. _ => error("conj: called with non-seq"),
  247. }
  248. }
  249. fn sub_scalar(a: MalArgs) -> MalRet {
  250. match (a[0].clone(), a[1].clone()) {
  251. (Func(_, _), ZKScalar(a1)) => {
  252. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  253. if let ZKScalar(mut a0) = values[0] {
  254. a0.sub_assign(a1);
  255. Ok(ZKScalar(a0))
  256. } else {
  257. error("scalar sub expect (zkscalar, zkscalar) found (func, zkscalar)")
  258. }
  259. } else {
  260. error("scalar sub expect (zkscalar, zkscalar)")
  261. }
  262. }
  263. (Func(_, _), Str(a1)) => {
  264. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  265. let s1 = bls12_381::Scalar::from_string(&a1);
  266. if let ZKScalar(mut a0) = values[0] {
  267. a0.sub_assign(s1);
  268. Ok(ZKScalar(a0))
  269. } else {
  270. error("scalar sub expect (zkscalar, zkscalar) found (func, zkscalar)")
  271. }
  272. } else {
  273. error("scalar sub expect (zkscalar, zkscalar)")
  274. }
  275. }
  276. (ZKScalar(mut a0), ZKScalar(a1)) => {
  277. a0.sub_assign(a1);
  278. Ok(ZKScalar(a0))
  279. }
  280. (Str(a0), ZKScalar(a1)) => {
  281. let (mut s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  282. s0.sub_assign(s1);
  283. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  284. }
  285. (ZKScalar(a0), Str(a1)) => {
  286. let (mut s0, s1) = (a0, bls12_381::Scalar::from_string(&a1));
  287. s0.sub_assign(s1);
  288. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  289. }
  290. (Str(a0), Str(a1)) => {
  291. let (mut s0, s1) = (
  292. bls12_381::Scalar::from_string(&a0),
  293. bls12_381::Scalar::from_string(&a1),
  294. );
  295. s0.sub_assign(s1);
  296. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  297. }
  298. _ => error(&format!("scalar sub expect (zkscalar, zkscalar) found \n {:?}", a).to_string()),
  299. }
  300. }
  301. fn mul_scalar(a: MalArgs) -> MalRet {
  302. match (a[0].clone(), a[1].clone()) {
  303. (Func(_, _), ZKScalar(a1)) => {
  304. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  305. if let ZKScalar(mut a0) = values[0] {
  306. a0.mul_assign(a1);
  307. Ok(ZKScalar(a0))
  308. } else {
  309. error("scalar mul expect (zkscalar, zkscalar) found (func, zkscalar)")
  310. }
  311. } else {
  312. error("scalar mul expect (zkscalar, zkscalar)")
  313. }
  314. }
  315. (ZKScalar(a1), Func(_, _)) => {
  316. if let Vector(ref values, _) = a[1].apply(vec![]).unwrap() {
  317. if let ZKScalar(mut a0) = values[0] {
  318. a0.mul_assign(a1);
  319. Ok(ZKScalar(a0))
  320. } else {
  321. error("scalar mul expect (zkscalar, zkscalar) found (func, zkscalar)")
  322. }
  323. } else {
  324. error("scalar mul expect (zkscalar, zkscalar)")
  325. }
  326. }
  327. (ZKScalar(mut a0), ZKScalar(a1)) => {
  328. a0.mul_assign(a1);
  329. Ok(ZKScalar(a0))
  330. }
  331. (Str(a0), Str(a1)) => {
  332. let (mut s0, s1) = (
  333. bls12_381::Scalar::from_string(&a0),
  334. bls12_381::Scalar::from_string(&a1),
  335. );
  336. s0.mul_assign(s1);
  337. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  338. }
  339. (ZKScalar(a0), Str(a1)) => {
  340. let (mut s0, s1) = (a0,
  341. bls12_381::Scalar::from_string(&a1),
  342. );
  343. s0.mul_assign(s1);
  344. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  345. }
  346. (Str(a0), ZKScalar(a1)) => {
  347. let (mut s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  348. s0.mul_assign(s1);
  349. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  350. }
  351. _ => error(
  352. &format!("scalar mul expect (zkscalar, zkscalar) \n {:?}", a).to_string())
  353. }
  354. }
  355. fn div_scalar(a: MalArgs) -> MalRet {
  356. match (a[0].clone(), a[1].clone()) {
  357. (ZKScalar(s0), ZKScalar(s1)) => {
  358. let ret = s1.invert().map(|other| *&s0 * other);
  359. if bool::from(ret.is_some()) {
  360. Ok(Str(
  361. std::string::ToString::to_string(&ret.unwrap())[2..].to_string()
  362. ))
  363. } else {
  364. error("DivisionByZero")
  365. }
  366. }
  367. (Str(a0), ZKScalar(a1)) => {
  368. let (s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  369. let ret = s1.invert().map(|other| *&s0 * other);
  370. if bool::from(ret.is_some()) {
  371. Ok(Str(
  372. std::string::ToString::to_string(&ret.unwrap())[2..].to_string()
  373. ))
  374. } else {
  375. error("DivisionByZero")
  376. }
  377. }
  378. (Str(a0), Str(a1)) => {
  379. let (s0, s1) = (
  380. bls12_381::Scalar::from_string(&a0),
  381. bls12_381::Scalar::from_string(&a1),
  382. );
  383. let ret = s1.invert().map(|other| *&s0 * other);
  384. if bool::from(ret.is_some()) {
  385. Ok(Str(
  386. std::string::ToString::to_string(&ret.unwrap())[2..].to_string()
  387. ))
  388. } else {
  389. error("DivisionByZero")
  390. }
  391. }
  392. _ => error(
  393. &format!("scalar div expect (zkscalar, zkscalar) \n {:?}", a).to_string())
  394. }
  395. }
  396. fn range(a: MalArgs) -> MalRet {
  397. let mut result = vec![];
  398. match (a[0].clone(), a[1].clone()) {
  399. (Int(a0), Int(a1)) => {
  400. for n in a0..a1 {
  401. result.push(n);
  402. }
  403. Ok(list!(result.iter().map(|_a| Nil).collect::<Vec<MalVal>>()))
  404. }
  405. _ => error("expected int int"),
  406. }
  407. }
  408. fn scalar_zero(a: MalArgs) -> MalRet {
  409. match a.len() {
  410. 0 => Ok(vector![vec![ZKScalar(bls12_381::Scalar::zero())]]),
  411. _ => Ok(vector![vec![
  412. ZKScalar(bls12_381::Scalar::zero()),
  413. a[0].clone()
  414. ]]),
  415. }
  416. }
  417. fn scalar_one(a: MalArgs) -> MalRet {
  418. match a.len() {
  419. 0 => Ok(vector![vec![ZKScalar(bls12_381::Scalar::one())]]),
  420. _ => Ok(vector![vec![
  421. ZKScalar(bls12_381::Scalar::one()),
  422. a[0].clone()
  423. ]]),
  424. }
  425. }
  426. fn cs_one(_a: MalArgs) -> MalRet {
  427. Ok(vector![vec![Sym("cs::one".to_string())]])
  428. }
  429. fn negate_from(a: MalArgs) -> MalRet {
  430. match a[0].clone() {
  431. ZKScalar(a0) => Ok(ZKScalar(a0.neg())),
  432. _ => match a[0].apply(vec![])? {
  433. List(v, _) | Vector(v, _) => match v[0] {
  434. ZKScalar(val) => Ok(vector![vec![ZKScalar(val.neg())]]),
  435. _ => error("not scalar"),
  436. },
  437. _ => return error("non zkscalar passed to negate"),
  438. },
  439. }
  440. }
  441. fn scalar_from(a: MalArgs) -> MalRet {
  442. match a[0].clone() {
  443. ZKScalar(s0) => Ok(ZKScalar(s0)),
  444. Str(a0) => {
  445. let s0 = bls12_381::Scalar::from_string(&a0.to_string());
  446. Ok(ZKScalar(s0))
  447. }
  448. Int(a0) => {
  449. let s0 = bls12_381::Scalar::from(a0 as u64);
  450. Ok(ZKScalar(s0))
  451. }
  452. _ => error("scalar from expected (string or int)"),
  453. }
  454. }
  455. fn scalar_square(a: MalArgs) -> MalRet {
  456. match a[0].clone() {
  457. ZKScalar(a0) => {
  458. let z0 = a0.clone();
  459. Ok(ZKScalar(z0.square()))
  460. }
  461. Str(a0) => {
  462. let s0 = bls12_381::Scalar::from_string(&a0);
  463. Ok(ZKScalar(s0.square()))
  464. }
  465. _ => error(
  466. &format!("scalar square expect (zkscalar or string) found \n {:?}", a).to_string(),
  467. ),
  468. }
  469. }
  470. fn scalar_double(a: MalArgs) -> MalRet {
  471. match a[0].clone() {
  472. ZKScalar(a0) => {
  473. let z0 = a0.clone();
  474. Ok(ZKScalar(z0.double()))
  475. }
  476. Str(a0) => {
  477. let s0 = bls12_381::Scalar::from_string(&a0);
  478. Ok(ZKScalar(s0.double()))
  479. }
  480. _ => error(
  481. &format!("scalar double expect (zkscalar or string) found \n {:?}", a).to_string(),
  482. ),
  483. }
  484. }
  485. fn scalar_invert(a: MalArgs) -> MalRet {
  486. match a[0].clone() {
  487. Func(_, _) => {
  488. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  489. if let ZKScalar(a0) = values[0] {
  490. if a0.is_zero() {
  491. error(
  492. &format!("scalar invert divizion by zero \n {:?}", a0).to_string())
  493. } else {
  494. Ok(ZKScalar(a0.invert().unwrap()))
  495. }
  496. } else {
  497. error(
  498. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a).to_string())
  499. }
  500. } else {
  501. error(
  502. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a).to_string())
  503. }
  504. }
  505. ZKScalar(a0) => {
  506. let z0 = a0.clone();
  507. Ok(ZKScalar(z0.invert().unwrap()))
  508. }
  509. Str(a0) => {
  510. let s0 = bls12_381::Scalar::from_string(&a0);
  511. Ok(ZKScalar(s0.invert().unwrap()))
  512. }
  513. _ => error(
  514. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a).to_string(),
  515. ),
  516. }
  517. }
  518. fn scalar_is_zero(a: MalArgs) -> MalRet {
  519. match a[0].clone() {
  520. Func(_, _) => {
  521. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  522. if let ZKScalar(a0) = values[0] {
  523. Ok(Bool(a0.is_zero()))
  524. } else {
  525. error(
  526. &format!("scalar is zero expect (zkscalar or string) found \n {:?}", a).to_string())
  527. }
  528. } else {
  529. error(
  530. &format!("scalar is zero expect (zkscalar or string) found \n {:?}", a).to_string())
  531. }
  532. }
  533. ZKScalar(a0) => {
  534. let z0 = a0.clone();
  535. Ok(Bool(z0.is_zero()))
  536. }
  537. Str(a0) => {
  538. let s0 = bls12_381::Scalar::from_string(&a0);
  539. Ok(Bool(s0.is_zero()))
  540. }
  541. _ => error(
  542. &format!("scalar is zero expect (zkscalar or string) found \n {:?}", a).to_string(),
  543. ),
  544. }
  545. }
  546. fn add_scalar(a: MalArgs) -> MalRet {
  547. match (a[0].clone(), a[1].clone()) {
  548. (Func(_, _), ZKScalar(a1)) => {
  549. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  550. if let ZKScalar(mut a0) = values[0] {
  551. a0.add_assign(a1);
  552. Ok(ZKScalar(a0))
  553. } else {
  554. error("scalar add expect (zkscalar, zkscalar) found (func, zkscalar)")
  555. }
  556. } else {
  557. error("scalar add expect (zkscalar, zkscalar)")
  558. }
  559. }
  560. (Func(_, _), Str(a1)) => {
  561. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  562. if let ZKScalar(mut a0) = values[0] {
  563. let s1 = bls12_381::Scalar::from_string(&a1);
  564. a0.add_assign(s1);
  565. Ok(ZKScalar(a0))
  566. } else {
  567. error("scalar add expect (zkscalar, zkscalar) found (func, zkscalar)")
  568. }
  569. } else {
  570. error("scalar add expect (zkscalar, zkscalar)")
  571. }
  572. }
  573. (ZKScalar(a0), ZKScalar(a1)) => {
  574. let (mut z0, z1) = (a0.clone(), a1.clone());
  575. z0.add_assign(z1);
  576. Ok(ZKScalar(z0))
  577. }
  578. (Str(a0), Str(a1)) => {
  579. let (mut s0, s1) = (
  580. bls12_381::Scalar::from_string(&a0),
  581. bls12_381::Scalar::from_string(&a1),
  582. );
  583. s0.add_assign(s1);
  584. Ok(ZKScalar(s0))
  585. }
  586. (Str(a0), ZKScalar(a1)) => {
  587. let (mut s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  588. s0.add_assign(s1);
  589. Ok(ZKScalar(s0))
  590. }
  591. _ => error(&format!("scalar add expect (zkscalar, zkscalar) found \n {:?}", a).to_string()),
  592. }
  593. }
  594. fn seq(a: MalArgs) -> MalRet {
  595. match a[0] {
  596. List(ref v, _) | Vector(ref v, _) if v.len() == 0 => Ok(Nil),
  597. List(ref v, _) | Vector(ref v, _) => Ok(list!(v.to_vec())),
  598. Str(ref s) if s.len() == 0 => Ok(Nil),
  599. Str(ref s) if !a[0].keyword_q() => {
  600. Ok(list!(s.chars().map(|c| { Str(c.to_string()) }).collect()))
  601. }
  602. Nil => Ok(Nil),
  603. _ => error("seq: called with non-seq"),
  604. }
  605. }
  606. fn gen_rand(a: MalArgs) -> MalRet {
  607. let mut rng = rand::thread_rng();
  608. Ok(MalVal::Int(rng.gen::<i64>()))
  609. }
  610. fn scalar_rnd(a: MalArgs) -> MalRet {
  611. let randomness_value: jubjub::Fr = jubjub::Fr::random(&mut OsRng);
  612. let value = bls12_381::Scalar::from_bytes(&randomness_value.to_bytes());
  613. Ok(MalVal::ZKScalar(value.unwrap()))
  614. }
  615. pub fn ns() -> Vec<(&'static str, MalVal)> {
  616. vec![
  617. ("=", func(|a| Ok(Bool(a[0] == a[1])))),
  618. ("throw", func(|a| Err(ErrMalVal(a[0].clone())))),
  619. ("nil?", func(fn_is_type!(Nil))),
  620. ("true?", func(fn_is_type!(Bool(true)))),
  621. ("false?", func(fn_is_type!(Bool(false)))),
  622. ("symbol", func(symbol)),
  623. ("symbol?", func(fn_is_type!(Sym(_)))),
  624. (
  625. "string?",
  626. func(fn_is_type!(Str(ref s) if !s.starts_with("\u{29e}"))),
  627. ),
  628. ("keyword", func(|a| a[0].keyword())),
  629. (
  630. "keyword?",
  631. func(fn_is_type!(Str(ref s) if s.starts_with("\u{29e}"))),
  632. ),
  633. ("number?", func(fn_is_type!(Int(_)))),
  634. (
  635. "fn?",
  636. func(fn_is_type!(MalFunc{is_macro,..} if !is_macro,Func(_,_))),
  637. ),
  638. (
  639. "macro?",
  640. func(fn_is_type!(MalFunc{is_macro,..} if is_macro)),
  641. ),
  642. ("pr-str", func(|a| Ok(Str(pr_seq(&a, true, "", "", " "))))),
  643. ("str", func(|a| Ok(Str(pr_seq(&a, false, "", "", ""))))),
  644. (
  645. "prn",
  646. func(|a| {
  647. println!("{}", pr_seq(&a, true, "", "", " "));
  648. Ok(Nil)
  649. }),
  650. ),
  651. (
  652. "println",
  653. func(|a| {
  654. println!("{}", pr_seq(&a, false, "", "", " "));
  655. Ok(Nil)
  656. }),
  657. ),
  658. ("read-string", func(fn_str!(|s| { read_str(s) }))),
  659. ("slurp", func(fn_str!(|f| { slurp(f) }))),
  660. ("<", func(fn_t_int_int!(Bool, |i, j| { i < j }))),
  661. ("<=", func(fn_t_int_int!(Bool, |i, j| { i <= j }))),
  662. (">", func(fn_t_int_int!(Bool, |i, j| { i > j }))),
  663. (">=", func(fn_t_int_int!(Bool, |i, j| { i >= j }))),
  664. ("+", func(add_scalar)),
  665. ("-", func(sub_scalar)),
  666. ("*", func(mul_scalar)),
  667. ("/", func(div_scalar)),
  668. ("time-ms", func(time_ms)),
  669. ("i+", func(fn_t_int_int!(Int, |i, j| { i + j }))),
  670. ("i-", func(fn_t_int_int!(Int, |i, j| { i - j }))),
  671. ("i*", func(fn_t_int_int!(Int, |i, j| { i * j }))),
  672. ("i/", func(fn_t_int_int!(Int, |i, j| { i / j }))),
  673. ("i<", func(fn_t_int_int!(Bool, |i, j| { i < j }))),
  674. ("i<=", func(fn_t_int_int!(Bool, |i, j| { i <= j }))),
  675. ("i>", func(fn_t_int_int!(Bool, |i, j| { i > j }))),
  676. ("i>=", func(fn_t_int_int!(Bool, |i, j| { i >= j }))),
  677. ("time-ms", func(time_ms)),
  678. ("sequential?", func(fn_is_type!(List(_, _), Vector(_, _)))),
  679. ("list", func(|a| Ok(list!(a)))),
  680. ("list?", func(fn_is_type!(List(_, _)))),
  681. ("vector", func(|a| Ok(vector!(a)))),
  682. ("vector?", func(fn_is_type!(Vector(_, _)))),
  683. ("hash-map", func(|a| hash_map(a))),
  684. ("map?", func(fn_is_type!(Hash(_, _)))),
  685. ("assoc", func(assoc)),
  686. ("dissoc", func(dissoc)),
  687. ("get", func(get)),
  688. ("contains?", func(contains_q)),
  689. ("keys", func(keys)),
  690. ("vals", func(vals)),
  691. ("vec", func(vec)),
  692. ("cons", func(cons)),
  693. ("concat", func(concat)),
  694. ("empty?", func(|a| a[0].empty_q())),
  695. ("nth", func(nth)),
  696. ("first", func(first)),
  697. ("last", func(last)),
  698. ("rest", func(rest)),
  699. ("count", func(|a| a[0].count())),
  700. ("apply", func(apply)),
  701. ("map", func(map)),
  702. ("conj", func(conj)),
  703. ("seq", func(seq)),
  704. ("meta", func(|a| a[0].get_meta())),
  705. ("with-meta", func(|a| a[0].clone().with_meta(&a[1]))),
  706. ("atom", func(|a| Ok(atom(&a[0])))),
  707. ("atom?", func(fn_is_type!(Atom(_)))),
  708. ("deref", func(|a| a[0].deref())),
  709. ("reset!", func(|a| a[0].reset_bang(&a[1]))),
  710. ("swap!", func(|a| a[0].swap_bang(&a[1..].to_vec()))),
  711. ("unpack-bits", func(unpack_bits)),
  712. ("range", func(range)),
  713. ("scalar::one", func(scalar_one)),
  714. ("neg", func(negate_from)),
  715. ("scalar::zero", func(scalar_zero)),
  716. ("scalar", func(scalar_from)),
  717. ("square", func(scalar_square)),
  718. ("cs::one", func(cs_one)),
  719. ("second", func(second)),
  720. ("genrand", func(gen_rand)),
  721. ("double", func(scalar_double)),
  722. ("invert", func(scalar_invert)),
  723. ("zero?", func(scalar_is_zero)),
  724. ("rnd-scalar", func(scalar_rnd)),
  725. ]
  726. }