diff --git a/crates/core/src/lib.rs b/crates/core/src/lib.rs index 849660c63..dc76fa7a9 100644 --- a/crates/core/src/lib.rs +++ b/crates/core/src/lib.rs @@ -122,6 +122,14 @@ impl<'a> ProveConfigBuilder<'a> { self.reveal(Direction::Sent, ranges) } + /// Reveals all of the sent data transcript. + pub fn reveal_sent_all(&mut self) -> Result<&mut Self, ProveConfigBuilderError> { + let len = self.transcript.len_of_direction(Direction::Sent); + let (sent, _) = self.reveal.get_or_insert_default(); + sent.union_mut(&(0..len)); + Ok(self) + } + /// Reveals the given ranges of the received data transcript. pub fn reveal_recv( &mut self, @@ -130,6 +138,14 @@ impl<'a> ProveConfigBuilder<'a> { self.reveal(Direction::Received, ranges) } + /// Reveals all of the received data transcript. + pub fn reveal_recv_all(&mut self) -> Result<&mut Self, ProveConfigBuilderError> { + let len = self.transcript.len_of_direction(Direction::Received); + let (_, recv) = self.reveal.get_or_insert_default(); + recv.union_mut(&(0..len)); + Ok(self) + } + /// Builds the configuration. pub fn build(self) -> Result { Ok(ProveConfig {