core.rs 27 KB

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