hack serialisation for now

Author: Trivo25

plonk-napi/src/types.rs

@@ -0,0 +1,79 @@
+use kimchi::{linearization::expr_linearization, prover_index::ProverIndex};
+use mina_curves::pasta::{Vesta as GAffine, VestaParameters};
+use mina_poseidon::{constants::PlonkSpongeConstantsKimchi, sponge::DefaultFqSponge};
+use napi::bindgen_prelude::{Error, External, Result as NapiResult, Status, Uint8Array};
+use napi_derive::napi;
+use poly_commitment::ipa::{OpeningProof, SRS};
+use serde::{Deserialize, Serialize};
+use std::{io::Cursor, sync::Arc};
+
+pub struct WasmPastaFpPlonkIndex(pub Box<ProverIndex<GAffine, OpeningProof<GAffine>>>);
+
+// TOOD: remove incl all dependencies when no longer needed and we only pass napi objects around
+#[derive(Serialize, Deserialize)]
+struct SerializedProverIndex {
+    prover_index: Vec<u8>,
+    srs: Vec<u8>,
+}
+
+// TOOD: remove incl all dependencies when no longer needed and we only pass napi objects around
+impl WasmPastaFpPlonkIndex {
+    fn serialize_inner(&self) -> Result<Vec<u8>, String> {
+        let prover_index = rmp_serde::to_vec(self.0.as_ref()).map_err(|e| e.to_string())?;
+
+        let mut srs = Vec::new();
+        self.0
+            .srs
+            .serialize(&mut rmp_serde::Serializer::new(&mut srs))
+            .map_err(|e| e.to_string())?;
+
+        let serialized = SerializedProverIndex { prover_index, srs };
+
+        rmp_serde::to_vec(&serialized).map_err(|e| e.to_string())
+    }
+
+    fn deserialize_inner(bytes: &[u8]) -> Result<Self, String> {
+        let serialized: SerializedProverIndex =
+            rmp_serde::from_slice(bytes).map_err(|e| e.to_string())?;
+
+        let mut index: ProverIndex<GAffine, OpeningProof<GAffine>> = ProverIndex::deserialize(
+            &mut rmp_serde::Deserializer::new(Cursor::new(serialized.prover_index)),
+        )
+        .map_err(|e| e.to_string())?;
+
+        let srs = SRS::<GAffine>::deserialize(&mut rmp_serde::Deserializer::new(Cursor::new(
+            serialized.srs,
+        )))
+        .map_err(|e| e.to_string())?;
+
+        index.srs = Arc::new(srs);
+
+        let (linearization, powers_of_alpha) =
+            expr_linearization(Some(&index.cs.feature_flags), true);
+        index.linearization = linearization;
+        index.powers_of_alpha = powers_of_alpha;
+
+        index.compute_verifier_index_digest::<
+            DefaultFqSponge<VestaParameters, PlonkSpongeConstantsKimchi>,
+        >();
+
+        Ok(WasmPastaFpPlonkIndex(Box::new(index)))
+    }
+}
+
+// TOOD: remove incl all dependencies when no longer needed and we only pass napi objects around
+#[napi]
+pub fn prover_index_from_bytes(bytes: Uint8Array) -> NapiResult<External<WasmPastaFpPlonkIndex>> {
+    let index = WasmPastaFpPlonkIndex::deserialize_inner(bytes.as_ref())
+        .map_err(|e| Error::new(Status::InvalidArg, e))?;
+    Ok(External::new(index))
+}
+
+// TOOD: remove incl all dependencies when no longer needed and we only pass napi objects around
+#[napi]
+pub fn prover_index_to_bytes(index: External<WasmPastaFpPlonkIndex>) -> NapiResult<Uint8Array> {
+    let bytes = index
+        .serialize_inner()
+        .map_err(|e| Error::new(Status::GenericFailure, e))?;
+    Ok(Uint8Array::from(bytes))
+}

plonk-wasm/src/pasta_fp_plonk_index.rs

@@ -20,7 +20,8 @@ use mina_poseidon::{constants::PlonkSpongeConstantsKimchi, sponge::DefaultFqSpon
 use serde::{Deserialize, Serialize};
 use std::{
     fs::{File, OpenOptions},
-    io::{BufReader, BufWriter, Seek, SeekFrom::Start},
+    io::{BufReader, BufWriter, Cursor, Seek, SeekFrom::Start},
+    sync::Arc,
 };
 use wasm_bindgen::prelude::*;
 use wasm_types::FlatVector as WasmFlatVector;
@@ -35,6 +36,71 @@ pub struct WasmPastaFpPlonkIndex(
     #[wasm_bindgen(skip)] pub Box<ProverIndex<GAffine, OpeningProof<GAffine>>>,
 );
 
+// TOOD: remove incl all dependencies when no longer needed and we only pass napi objects around
+#[derive(Serialize, Deserialize)]
+struct SerializedProverIndex {
+    prover_index: Vec<u8>,
+    srs: Vec<u8>,
+}
+
+// TOOD: remove incl all dependencies when no longer needed and we only pass napi objects around
+#[wasm_bindgen]
+impl WasmPastaFpPlonkIndex {
+    #[wasm_bindgen(js_name = "serialize")]
+    pub fn serialize(&self) -> Result<Vec<u8>, JsError> {
+        serialize_prover_index(self.0.as_ref())
+            .map_err(|e| JsError::new(&format!("WasmPastaFpPlonkIndex::serialize: {e}")))
+    }
+
+    #[wasm_bindgen(js_name = "deserialize")]
+    pub fn deserialize(bytes: &[u8]) -> Result<WasmPastaFpPlonkIndex, JsError> {
+        deserialize_prover_index(bytes)
+            .map(WasmPastaFpPlonkIndex)
+            .map_err(|e| JsError::new(&format!("WasmPastaFpPlonkIndex::deserialize: {e}")))
+    }
+}
+
+fn serialize_prover_index(
+    index: &ProverIndex<GAffine, OpeningProof<GAffine>>,
+) -> Result<Vec<u8>, String> {
+    let prover_index = rmp_serde::to_vec(index).map_err(|e| e.to_string())?;
+
+    let mut srs = Vec::new();
+    index
+        .srs
+        .serialize(&mut rmp_serde::Serializer::new(&mut srs))
+        .map_err(|e| e.to_string())?;
+
+    let serialized = SerializedProverIndex { prover_index, srs };
+
+    rmp_serde::to_vec(&serialized).map_err(|e| e.to_string())
+}
+
+fn deserialize_prover_index(
+    bytes: &[u8],
+) -> Result<Box<ProverIndex<GAffine, OpeningProof<GAffine>>>, String> {
+    let serialized: SerializedProverIndex =
+        rmp_serde::from_slice(bytes).map_err(|e| e.to_string())?;
+
+    let mut index: ProverIndex<GAffine, OpeningProof<GAffine>> = ProverIndex::deserialize(
+        &mut rmp_serde::Deserializer::new(Cursor::new(serialized.prover_index)),
+    )
+    .map_err(|e| e.to_string())?;
+
+    let srs = poly_commitment::ipa::SRS::<GAffine>::deserialize(&mut rmp_serde::Deserializer::new(
+        Cursor::new(serialized.srs),
+    ))
+    .map_err(|e| e.to_string())?;
+
+    index.srs = Arc::new(srs);
+
+    let (linearization, powers_of_alpha) = expr_linearization(Some(&index.cs.feature_flags), true);
+    index.linearization = linearization;
+    index.powers_of_alpha = powers_of_alpha;
+
+    Ok(Box::new(index))
+}
+
 // This should mimic LookupTable structure
 #[wasm_bindgen]
 pub struct WasmPastaFpLookupTable {