core.rs 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786
  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. ZKScalar(a0) => {
  469. let z0 = a0.clone();
  470. Ok(ZKScalar(z0.double()))
  471. }
  472. Str(a0) => {
  473. let s0 = bls12_381::Scalar::from_string(&a0);
  474. Ok(ZKScalar(s0.double()))
  475. }
  476. _ => error(
  477. &format!("scalar double expect (zkscalar or string) found \n {:?}", a).to_string(),
  478. ),
  479. }
  480. }
  481. fn scalar_invert(a: MalArgs) -> MalRet {
  482. match a[0].clone() {
  483. Func(_, _) => {
  484. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  485. if let ZKScalar(a0) = values[0] {
  486. if a0.is_zero() {
  487. error(&format!("scalar invert divizion by zero \n {:?}", a0).to_string())
  488. } else {
  489. Ok(ZKScalar(a0.invert().unwrap()))
  490. }
  491. } else {
  492. error(
  493. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a)
  494. .to_string(),
  495. )
  496. }
  497. } else {
  498. error(
  499. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a)
  500. .to_string(),
  501. )
  502. }
  503. }
  504. ZKScalar(a0) => {
  505. let z0 = a0.clone();
  506. Ok(ZKScalar(z0.invert().unwrap()))
  507. }
  508. Str(a0) => {
  509. let s0 = bls12_381::Scalar::from_string(&a0);
  510. Ok(ZKScalar(s0.invert().unwrap()))
  511. }
  512. _ => error(
  513. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a).to_string(),
  514. ),
  515. }
  516. }
  517. fn scalar_is_zero(a: MalArgs) -> MalRet {
  518. match a[0].clone() {
  519. Func(_, _) => {
  520. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  521. if let ZKScalar(a0) = values[0] {
  522. Ok(Bool(a0.is_zero()))
  523. } else {
  524. error(
  525. &format!(
  526. "scalar is zero expect (zkscalar or string) found \n {:?}",
  527. a
  528. )
  529. .to_string(),
  530. )
  531. }
  532. } else {
  533. error(
  534. &format!(
  535. "scalar is zero expect (zkscalar or string) found \n {:?}",
  536. a
  537. )
  538. .to_string(),
  539. )
  540. }
  541. }
  542. ZKScalar(a0) => {
  543. let z0 = a0.clone();
  544. Ok(Bool(z0.is_zero()))
  545. }
  546. Str(a0) => {
  547. let s0 = bls12_381::Scalar::from_string(&a0);
  548. Ok(Bool(s0.is_zero()))
  549. }
  550. _ => error(
  551. &format!(
  552. "scalar is zero expect (zkscalar or string) found \n {:?}",
  553. a
  554. )
  555. .to_string(),
  556. ),
  557. }
  558. }
  559. fn add_scalar(a: MalArgs) -> MalRet {
  560. match (a[0].clone(), a[1].clone()) {
  561. (Func(_, _), ZKScalar(a1)) => {
  562. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  563. if let ZKScalar(mut a0) = values[0] {
  564. a0.add_assign(a1);
  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. (Func(_, _), Str(a1)) => {
  574. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  575. if let ZKScalar(mut a0) = values[0] {
  576. let s1 = bls12_381::Scalar::from_string(&a1);
  577. a0.add_assign(s1);
  578. Ok(ZKScalar(a0))
  579. } else {
  580. error("scalar add expect (zkscalar, zkscalar) found (func, zkscalar)")
  581. }
  582. } else {
  583. error("scalar add expect (zkscalar, zkscalar)")
  584. }
  585. }
  586. (ZKScalar(a0), ZKScalar(a1)) => {
  587. let (mut z0, z1) = (a0.clone(), a1.clone());
  588. z0.add_assign(z1);
  589. Ok(ZKScalar(z0))
  590. }
  591. (Str(a0), Str(a1)) => {
  592. let (mut s0, s1) = (
  593. bls12_381::Scalar::from_string(&a0),
  594. bls12_381::Scalar::from_string(&a1),
  595. );
  596. s0.add_assign(s1);
  597. Ok(ZKScalar(s0))
  598. }
  599. (Str(a0), ZKScalar(a1)) => {
  600. let (mut s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  601. s0.add_assign(s1);
  602. Ok(ZKScalar(s0))
  603. }
  604. _ => error(&format!("scalar add expect (zkscalar, zkscalar) found \n {:?}", a).to_string()),
  605. }
  606. }
  607. fn seq(a: MalArgs) -> MalRet {
  608. match a[0] {
  609. List(ref v, _) | Vector(ref v, _) if v.len() == 0 => Ok(Nil),
  610. List(ref v, _) | Vector(ref v, _) => Ok(list!(v.to_vec())),
  611. Str(ref s) if s.len() == 0 => Ok(Nil),
  612. Str(ref s) if !a[0].keyword_q() => {
  613. Ok(list!(s.chars().map(|c| { Str(c.to_string()) }).collect()))
  614. }
  615. Nil => Ok(Nil),
  616. _ => error("seq: called with non-seq"),
  617. }
  618. }
  619. fn gen_rand(a: MalArgs) -> MalRet {
  620. let mut rng = rand::thread_rng();
  621. Ok(MalVal::Int(rng.gen::<i64>()))
  622. }
  623. fn scalar_rnd(a: MalArgs) -> MalRet {
  624. let randomness_value: jubjub::Fr = jubjub::Fr::random(&mut OsRng);
  625. let value = bls12_381::Scalar::from_bytes(&randomness_value.to_bytes());
  626. Ok(MalVal::ZKScalar(value.unwrap()))
  627. }
  628. pub fn ns() -> Vec<(&'static str, MalVal)> {
  629. vec![
  630. ("=", func(|a| Ok(Bool(a[0] == a[1])))),
  631. ("throw", func(|a| Err(ErrMalVal(a[0].clone())))),
  632. ("nil?", func(fn_is_type!(Nil))),
  633. ("true?", func(fn_is_type!(Bool(true)))),
  634. ("false?", func(fn_is_type!(Bool(false)))),
  635. ("symbol", func(symbol)),
  636. ("symbol?", func(fn_is_type!(Sym(_)))),
  637. (
  638. "string?",
  639. func(fn_is_type!(Str(ref s) if !s.starts_with("\u{29e}"))),
  640. ),
  641. ("keyword", func(|a| a[0].keyword())),
  642. (
  643. "keyword?",
  644. func(fn_is_type!(Str(ref s) if s.starts_with("\u{29e}"))),
  645. ),
  646. ("number?", func(fn_is_type!(Int(_)))),
  647. (
  648. "fn?",
  649. func(fn_is_type!(MalFunc{is_macro,..} if !is_macro,Func(_,_))),
  650. ),
  651. (
  652. "macro?",
  653. func(fn_is_type!(MalFunc{is_macro,..} if is_macro)),
  654. ),
  655. ("pr-str", func(|a| Ok(Str(pr_seq(&a, true, "", "", " "))))),
  656. ("str", func(|a| Ok(Str(pr_seq(&a, false, "", "", ""))))),
  657. (
  658. "prn",
  659. func(|a| {
  660. println!("{}", pr_seq(&a, true, "", "", " "));
  661. Ok(Nil)
  662. }),
  663. ),
  664. (
  665. "println",
  666. func(|a| {
  667. println!("{}", pr_seq(&a, false, "", "", " "));
  668. Ok(Nil)
  669. }),
  670. ),
  671. ("read-string", func(fn_str!(|s| { read_str(s) }))),
  672. ("slurp", func(fn_str!(|f| { slurp(f) }))),
  673. ("<", func(fn_t_int_int!(Bool, |i, j| { i < j }))),
  674. ("<=", func(fn_t_int_int!(Bool, |i, j| { i <= j }))),
  675. (">", func(fn_t_int_int!(Bool, |i, j| { i > j }))),
  676. (">=", func(fn_t_int_int!(Bool, |i, j| { i >= j }))),
  677. ("+", func(add_scalar)),
  678. ("-", func(sub_scalar)),
  679. ("*", func(mul_scalar)),
  680. ("/", func(div_scalar)),
  681. ("time-ms", func(time_ms)),
  682. ("i+", func(fn_t_int_int!(Int, |i, j| { i + j }))),
  683. ("i-", func(fn_t_int_int!(Int, |i, j| { i - j }))),
  684. ("i*", func(fn_t_int_int!(Int, |i, j| { i * j }))),
  685. ("i/", func(fn_t_int_int!(Int, |i, j| { i / j }))),
  686. ("i<", func(fn_t_int_int!(Bool, |i, j| { i < j }))),
  687. ("i<=", func(fn_t_int_int!(Bool, |i, j| { i <= j }))),
  688. ("i>", func(fn_t_int_int!(Bool, |i, j| { i > j }))),
  689. ("i>=", func(fn_t_int_int!(Bool, |i, j| { i >= j }))),
  690. ("time-ms", func(time_ms)),
  691. ("sequential?", func(fn_is_type!(List(_, _), Vector(_, _)))),
  692. ("list", func(|a| Ok(list!(a)))),
  693. ("list?", func(fn_is_type!(List(_, _)))),
  694. ("vector", func(|a| Ok(vector!(a)))),
  695. ("vector?", func(fn_is_type!(Vector(_, _)))),
  696. ("hash-map", func(|a| hash_map(a))),
  697. ("map?", func(fn_is_type!(Hash(_, _)))),
  698. ("assoc", func(assoc)),
  699. ("dissoc", func(dissoc)),
  700. ("get", func(get)),
  701. ("contains?", func(contains_q)),
  702. ("keys", func(keys)),
  703. ("vals", func(vals)),
  704. ("vec", func(vec)),
  705. ("cons", func(cons)),
  706. ("concat", func(concat)),
  707. ("empty?", func(|a| a[0].empty_q())),
  708. ("nth", func(nth)),
  709. ("first", func(first)),
  710. ("last", func(last)),
  711. ("rest", func(rest)),
  712. ("count", func(|a| a[0].count())),
  713. ("apply", func(apply)),
  714. ("map", func(map)),
  715. ("conj", func(conj)),
  716. ("seq", func(seq)),
  717. ("meta", func(|a| a[0].get_meta())),
  718. ("with-meta", func(|a| a[0].clone().with_meta(&a[1]))),
  719. ("atom", func(|a| Ok(atom(&a[0])))),
  720. ("atom?", func(fn_is_type!(Atom(_)))),
  721. ("deref", func(|a| a[0].deref())),
  722. ("reset!", func(|a| a[0].reset_bang(&a[1]))),
  723. ("swap!", func(|a| a[0].swap_bang(&a[1..].to_vec()))),
  724. ("unpack-bits", func(unpack_bits)),
  725. ("range", func(range)),
  726. ("scalar::one", func(scalar_one)),
  727. ("neg", func(negate_from)),
  728. ("scalar::zero", func(scalar_zero)),
  729. ("scalar", func(scalar_from)),
  730. ("square", func(scalar_square)),
  731. ("cs::one", func(cs_one)),
  732. ("second", func(second)),
  733. ("genrand", func(gen_rand)),
  734. ("double", func(scalar_double)),
  735. ("invert", func(scalar_invert)),
  736. ("zero?", func(scalar_is_zero)),
  737. ("rnd-scalar", func(scalar_rnd)),
  738. ]
  739. }