core.rs 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796
  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(&format!("invalid args to unpack-bits found \n {:?}", a).to_string()),
  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, bls12_381::Scalar::from_string(&a1));
  341. s0.mul_assign(s1);
  342. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  343. }
  344. (Str(a0), ZKScalar(a1)) => {
  345. let (mut s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  346. s0.mul_assign(s1);
  347. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  348. }
  349. _ => error(&format!("scalar mul expect (zkscalar, zkscalar) \n {:?}", a).to_string()),
  350. }
  351. }
  352. fn div_scalar(a: MalArgs) -> MalRet {
  353. match (a[0].clone(), a[1].clone()) {
  354. (ZKScalar(s0), ZKScalar(s1)) => {
  355. let ret = s1.invert().map(|other| *&s0 * other);
  356. if bool::from(ret.is_some()) {
  357. Ok(Str(
  358. std::string::ToString::to_string(&ret.unwrap())[2..].to_string()
  359. ))
  360. } else {
  361. error("DivisionByZero")
  362. }
  363. }
  364. (Str(a0), ZKScalar(a1)) => {
  365. let (s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  366. let ret = s1.invert().map(|other| *&s0 * other);
  367. if bool::from(ret.is_some()) {
  368. Ok(Str(
  369. std::string::ToString::to_string(&ret.unwrap())[2..].to_string()
  370. ))
  371. } else {
  372. error("DivisionByZero")
  373. }
  374. }
  375. (Str(a0), Str(a1)) => {
  376. let (s0, s1) = (
  377. bls12_381::Scalar::from_string(&a0),
  378. bls12_381::Scalar::from_string(&a1),
  379. );
  380. let ret = s1.invert().map(|other| *&s0 * other);
  381. if bool::from(ret.is_some()) {
  382. Ok(Str(
  383. std::string::ToString::to_string(&ret.unwrap())[2..].to_string()
  384. ))
  385. } else {
  386. error("DivisionByZero")
  387. }
  388. }
  389. _ => error(&format!("scalar div expect (zkscalar, zkscalar) \n {:?}", a).to_string()),
  390. }
  391. }
  392. fn range(a: MalArgs) -> MalRet {
  393. let mut result = vec![];
  394. match (a[0].clone(), a[1].clone()) {
  395. (Int(a0), Int(a1)) => {
  396. for n in a0..a1 {
  397. result.push(n);
  398. }
  399. Ok(list!(result.iter().map(|_a| Nil).collect::<Vec<MalVal>>()))
  400. }
  401. _ => error("expected int int"),
  402. }
  403. }
  404. fn scalar_zero(a: MalArgs) -> MalRet {
  405. match a.len() {
  406. 0 => Ok(vector![vec![ZKScalar(bls12_381::Scalar::zero())]]),
  407. _ => Ok(vector![vec![
  408. ZKScalar(bls12_381::Scalar::zero()),
  409. a[0].clone()
  410. ]]),
  411. }
  412. }
  413. fn scalar_one(a: MalArgs) -> MalRet {
  414. match a.len() {
  415. 0 => Ok(vector![vec![ZKScalar(bls12_381::Scalar::one())]]),
  416. _ => Ok(vector![vec![
  417. ZKScalar(bls12_381::Scalar::one()),
  418. a[0].clone()
  419. ]]),
  420. }
  421. }
  422. fn cs_one(_a: MalArgs) -> MalRet {
  423. Ok(vector![vec![Sym("cs::one".to_string())]])
  424. }
  425. fn negate_from(a: MalArgs) -> MalRet {
  426. match a[0].clone() {
  427. ZKScalar(a0) => Ok(ZKScalar(a0.neg())),
  428. _ => match a[0].apply(vec![])? {
  429. List(v, _) | Vector(v, _) => match v[0] {
  430. ZKScalar(val) => Ok(vector![vec![ZKScalar(val.neg())]]),
  431. _ => error("not scalar"),
  432. },
  433. _ => return error("non zkscalar passed to negate"),
  434. },
  435. }
  436. }
  437. fn scalar_from(a: MalArgs) -> MalRet {
  438. match a[0].clone() {
  439. ZKScalar(s0) => Ok(ZKScalar(s0)),
  440. Str(a0) => {
  441. let s0 = bls12_381::Scalar::from_string(&a0.to_string());
  442. Ok(ZKScalar(s0))
  443. }
  444. Int(a0) => {
  445. let s0 = bls12_381::Scalar::from(a0 as u64);
  446. Ok(ZKScalar(s0))
  447. }
  448. _ => error("scalar from expected (string or int)"),
  449. }
  450. }
  451. fn scalar_square(a: MalArgs) -> MalRet {
  452. match a[0].clone() {
  453. ZKScalar(a0) => {
  454. let z0 = a0.clone();
  455. Ok(ZKScalar(z0.square()))
  456. }
  457. Str(a0) => {
  458. let s0 = bls12_381::Scalar::from_string(&a0);
  459. Ok(ZKScalar(s0.square()))
  460. }
  461. _ => error(
  462. &format!("scalar square expect (zkscalar or string) found \n {:?}", a).to_string(),
  463. ),
  464. }
  465. }
  466. fn scalar_double(a: MalArgs) -> MalRet {
  467. match a[0].clone() {
  468. Func(_, _) => {
  469. if let Vector(ref values, _) = r oua[0].apply(vec![]).unwrap() {
  470. if let ZKScalar(a0) = values[0] {
  471. a0.double();
  472. Ok(ZKScalar(a0))
  473. } else {
  474. error(&format!("scalar double expect (zkscalar or string) found \n {:?}", a).to_string())
  475. }
  476. }
  477. }
  478. ZKScalar(a0) => {
  479. let z0 = a0.clone();
  480. Ok(ZKScalar(z0.double()))
  481. }
  482. Str(a0) => {
  483. let s0 = bls12_381::Scalar::from_string(&a0);
  484. Ok(ZKScalar(s0.double()))
  485. }
  486. _ => error(
  487. &format!("scalar double expect (zkscalar or string) found \n {:?}", a).to_string(),
  488. ),
  489. }
  490. }
  491. fn scalar_invert(a: MalArgs) -> MalRet {
  492. match a[0].clone() {
  493. Func(_, _) => {
  494. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  495. if let ZKScalar(a0) = values[0] {
  496. if a0.is_zero() {
  497. error(&format!("scalar invert divizion by zero \n {:?}", a0).to_string())
  498. } else {
  499. Ok(ZKScalar(a0.invert().unwrap()))
  500. }
  501. } else {
  502. error(
  503. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a)
  504. .to_string(),
  505. )
  506. }
  507. } else {
  508. error(
  509. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a)
  510. .to_string(),
  511. )
  512. }
  513. }
  514. ZKScalar(a0) => {
  515. let z0 = a0.clone();
  516. Ok(ZKScalar(z0.invert().unwrap()))
  517. }
  518. Str(a0) => {
  519. let s0 = bls12_381::Scalar::from_string(&a0);
  520. Ok(ZKScalar(s0.invert().unwrap()))
  521. }
  522. _ => error(
  523. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a).to_string(),
  524. ),
  525. }
  526. }
  527. fn scalar_is_zero(a: MalArgs) -> MalRet {
  528. match a[0].clone() {
  529. Func(_, _) => {
  530. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  531. if let ZKScalar(a0) = values[0] {
  532. Ok(Bool(a0.is_zero()))
  533. } else {
  534. error(
  535. &format!(
  536. "scalar is zero expect (zkscalar or string) found \n {:?}",
  537. a
  538. )
  539. .to_string(),
  540. )
  541. }
  542. } else {
  543. error(
  544. &format!(
  545. "scalar is zero expect (zkscalar or string) found \n {:?}",
  546. a
  547. )
  548. .to_string(),
  549. )
  550. }
  551. }
  552. ZKScalar(a0) => {
  553. let z0 = a0.clone();
  554. Ok(Bool(z0.is_zero()))
  555. }
  556. Str(a0) => {
  557. let s0 = bls12_381::Scalar::from_string(&a0);
  558. Ok(Bool(s0.is_zero()))
  559. }
  560. _ => error(
  561. &format!(
  562. "scalar is zero expect (zkscalar or string) found \n {:?}",
  563. a
  564. )
  565. .to_string(),
  566. ),
  567. }
  568. }
  569. fn add_scalar(a: MalArgs) -> MalRet {
  570. match (a[0].clone(), a[1].clone()) {
  571. (Func(_, _), ZKScalar(a1)) => {
  572. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  573. if let ZKScalar(mut a0) = values[0] {
  574. a0.add_assign(a1);
  575. Ok(ZKScalar(a0))
  576. } else {
  577. error("scalar add expect (zkscalar, zkscalar) found (func, zkscalar)")
  578. }
  579. } else {
  580. error("scalar add expect (zkscalar, zkscalar)")
  581. }
  582. }
  583. (Func(_, _), Str(a1)) => {
  584. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  585. if let ZKScalar(mut a0) = values[0] {
  586. let s1 = bls12_381::Scalar::from_string(&a1);
  587. a0.add_assign(s1);
  588. Ok(ZKScalar(a0))
  589. } else {
  590. error("scalar add expect (zkscalar, zkscalar) found (func, zkscalar)")
  591. }
  592. } else {
  593. error("scalar add expect (zkscalar, zkscalar)")
  594. }
  595. }
  596. (ZKScalar(a0), ZKScalar(a1)) => {
  597. let (mut z0, z1) = (a0.clone(), a1.clone());
  598. z0.add_assign(z1);
  599. Ok(ZKScalar(z0))
  600. }
  601. (Str(a0), Str(a1)) => {
  602. let (mut s0, s1) = (
  603. bls12_381::Scalar::from_string(&a0),
  604. bls12_381::Scalar::from_string(&a1),
  605. );
  606. s0.add_assign(s1);
  607. Ok(ZKScalar(s0))
  608. }
  609. (Str(a0), ZKScalar(a1)) => {
  610. let (mut s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  611. s0.add_assign(s1);
  612. Ok(ZKScalar(s0))
  613. }
  614. _ => error(&format!("scalar add expect (zkscalar, zkscalar) found \n {:?}", a).to_string()),
  615. }
  616. }
  617. fn seq(a: MalArgs) -> MalRet {
  618. match a[0] {
  619. List(ref v, _) | Vector(ref v, _) if v.len() == 0 => Ok(Nil),
  620. List(ref v, _) | Vector(ref v, _) => Ok(list!(v.to_vec())),
  621. Str(ref s) if s.len() == 0 => Ok(Nil),
  622. Str(ref s) if !a[0].keyword_q() => {
  623. Ok(list!(s.chars().map(|c| { Str(c.to_string()) }).collect()))
  624. }
  625. Nil => Ok(Nil),
  626. _ => error("seq: called with non-seq"),
  627. }
  628. }
  629. fn gen_rand(a: MalArgs) -> MalRet {
  630. let mut rng = rand::thread_rng();
  631. Ok(MalVal::Int(rng.gen::<i64>()))
  632. }
  633. fn scalar_rnd(a: MalArgs) -> MalRet {
  634. let randomness_value: jubjub::Fr = jubjub::Fr::random(&mut OsRng);
  635. let value = bls12_381::Scalar::from_bytes(&randomness_value.to_bytes());
  636. Ok(MalVal::ZKScalar(value.unwrap()))
  637. }
  638. pub fn ns() -> Vec<(&'static str, MalVal)> {
  639. vec![
  640. ("=", func(|a| Ok(Bool(a[0] == a[1])))),
  641. ("throw", func(|a| Err(ErrMalVal(a[0].clone())))),
  642. ("nil?", func(fn_is_type!(Nil))),
  643. ("true?", func(fn_is_type!(Bool(true)))),
  644. ("false?", func(fn_is_type!(Bool(false)))),
  645. ("symbol", func(symbol)),
  646. ("symbol?", func(fn_is_type!(Sym(_)))),
  647. (
  648. "string?",
  649. func(fn_is_type!(Str(ref s) if !s.starts_with("\u{29e}"))),
  650. ),
  651. ("keyword", func(|a| a[0].keyword())),
  652. (
  653. "keyword?",
  654. func(fn_is_type!(Str(ref s) if s.starts_with("\u{29e}"))),
  655. ),
  656. ("number?", func(fn_is_type!(Int(_)))),
  657. (
  658. "fn?",
  659. func(fn_is_type!(MalFunc{is_macro,..} if !is_macro,Func(_,_))),
  660. ),
  661. (
  662. "macro?",
  663. func(fn_is_type!(MalFunc{is_macro,..} if is_macro)),
  664. ),
  665. ("pr-str", func(|a| Ok(Str(pr_seq(&a, true, "", "", " "))))),
  666. ("str", func(|a| Ok(Str(pr_seq(&a, false, "", "", ""))))),
  667. (
  668. "prn",
  669. func(|a| {
  670. println!("{}", pr_seq(&a, true, "", "", " "));
  671. Ok(Nil)
  672. }),
  673. ),
  674. (
  675. "println",
  676. func(|a| {
  677. println!("{}", pr_seq(&a, false, "", "", " "));
  678. Ok(Nil)
  679. }),
  680. ),
  681. ("read-string", func(fn_str!(|s| { read_str(s) }))),
  682. ("slurp", func(fn_str!(|f| { slurp(f) }))),
  683. ("<", func(fn_t_int_int!(Bool, |i, j| { i < j }))),
  684. ("<=", func(fn_t_int_int!(Bool, |i, j| { i <= j }))),
  685. (">", func(fn_t_int_int!(Bool, |i, j| { i > j }))),
  686. (">=", func(fn_t_int_int!(Bool, |i, j| { i >= j }))),
  687. ("+", func(add_scalar)),
  688. ("-", func(sub_scalar)),
  689. ("*", func(mul_scalar)),
  690. ("/", func(div_scalar)),
  691. ("time-ms", func(time_ms)),
  692. ("i+", func(fn_t_int_int!(Int, |i, j| { i + j }))),
  693. ("i-", func(fn_t_int_int!(Int, |i, j| { i - j }))),
  694. ("i*", func(fn_t_int_int!(Int, |i, j| { i * j }))),
  695. ("i/", func(fn_t_int_int!(Int, |i, j| { i / j }))),
  696. ("i<", func(fn_t_int_int!(Bool, |i, j| { i < j }))),
  697. ("i<=", func(fn_t_int_int!(Bool, |i, j| { i <= j }))),
  698. ("i>", func(fn_t_int_int!(Bool, |i, j| { i > j }))),
  699. ("i>=", func(fn_t_int_int!(Bool, |i, j| { i >= j }))),
  700. ("time-ms", func(time_ms)),
  701. ("sequential?", func(fn_is_type!(List(_, _), Vector(_, _)))),
  702. ("list", func(|a| Ok(list!(a)))),
  703. ("list?", func(fn_is_type!(List(_, _)))),
  704. ("vector", func(|a| Ok(vector!(a)))),
  705. ("vector?", func(fn_is_type!(Vector(_, _)))),
  706. ("hash-map", func(|a| hash_map(a))),
  707. ("map?", func(fn_is_type!(Hash(_, _)))),
  708. ("assoc", func(assoc)),
  709. ("dissoc", func(dissoc)),
  710. ("get", func(get)),
  711. ("contains?", func(contains_q)),
  712. ("keys", func(keys)),
  713. ("vals", func(vals)),
  714. ("vec", func(vec)),
  715. ("cons", func(cons)),
  716. ("concat", func(concat)),
  717. ("empty?", func(|a| a[0].empty_q())),
  718. ("nth", func(nth)),
  719. ("first", func(first)),
  720. ("last", func(last)),
  721. ("rest", func(rest)),
  722. ("count", func(|a| a[0].count())),
  723. ("apply", func(apply)),
  724. ("map", func(map)),
  725. ("conj", func(conj)),
  726. ("seq", func(seq)),
  727. ("meta", func(|a| a[0].get_meta())),
  728. ("with-meta", func(|a| a[0].clone().with_meta(&a[1]))),
  729. ("atom", func(|a| Ok(atom(&a[0])))),
  730. ("atom?", func(fn_is_type!(Atom(_)))),
  731. ("deref", func(|a| a[0].deref())),
  732. ("reset!", func(|a| a[0].reset_bang(&a[1]))),
  733. ("swap!", func(|a| a[0].swap_bang(&a[1..].to_vec()))),
  734. ("unpack-bits", func(unpack_bits)),
  735. ("range", func(range)),
  736. ("scalar::one", func(scalar_one)),
  737. ("neg", func(negate_from)),
  738. ("scalar::zero", func(scalar_zero)),
  739. ("scalar", func(scalar_from)),
  740. ("square", func(scalar_square)),
  741. ("cs::one", func(cs_one)),
  742. ("second", func(second)),
  743. ("genrand", func(gen_rand)),
  744. ("double", func(scalar_double)),
  745. ("invert", func(scalar_invert)),
  746. ("zero?", func(scalar_is_zero)),
  747. ("rnd-scalar", func(scalar_rnd)),
  748. ]
  749. }