use super::*;
impl<N: Network, Private: Visibility> FromBytes for Record<N, Private> {
fn read_le<R: Read>(mut reader: R) -> IoResult<Self> {
let owner = Owner::read_le(&mut reader)?;
let gates = Balance::read_le(&mut reader)?;
let num_entries = u8::read_le(&mut reader)?;
let mut data = IndexMap::with_capacity(num_entries as usize);
for _ in 0..num_entries {
let identifier = Identifier::<N>::read_le(&mut reader)?;
let num_bytes = u16::read_le(&mut reader)?;
let bytes = (0..num_bytes).map(|_| u8::read_le(&mut reader)).collect::<Result<Vec<_>, _>>()?;
let entry = Entry::read_le(&mut bytes.as_slice())?;
data.insert(identifier, entry);
}
let nonce = Group::read_le(&mut reader)?;
let reserved = [
Identifier::from_str("owner").map_err(|e| error(e.to_string()))?,
Identifier::from_str("gates").map_err(|e| error(e.to_string()))?,
];
if has_duplicates(data.keys().chain(reserved.iter())) {
return Err(error("Duplicate entry type found in record"));
}
if data.len() > N::MAX_DATA_ENTRIES {
return Err(error("Failed to parse record: too many entries"));
}
Ok(Self { owner, gates, data, nonce })
}
}
impl<N: Network, Private: Visibility> ToBytes for Record<N, Private> {
fn write_le<W: Write>(&self, mut writer: W) -> IoResult<()> {
self.owner.write_le(&mut writer)?;
self.gates.write_le(&mut writer)?;
u8::try_from(self.data.len()).or_halt_with::<N>("Record length exceeds u8::MAX").write_le(&mut writer)?;
for (entry_name, entry_value) in &self.data {
entry_name.write_le(&mut writer)?;
let bytes = entry_value.to_bytes_le().map_err(|e| error(e.to_string()))?;
u16::try_from(bytes.len())
.or_halt_with::<N>("Record entry exceeds u16::MAX bytes")
.write_le(&mut writer)?;
bytes.write_le(&mut writer)?;
}
self.nonce.write_le(&mut writer)
}
}
#[cfg(test)]
mod tests {
use super::*;
use snarkvm_console_network::Testnet3;
type CurrentNetwork = Testnet3;
#[test]
fn test_bytes() -> Result<()> {
let expected = Record::<CurrentNetwork, Plaintext<CurrentNetwork>>::from_str(
"{ owner: aleo1d5hg2z3ma00382pngntdp68e74zv54jdxy249qhaujhks9c72yrs33ddah.private, gates: 5u64.private, token_amount: 100u64.private, _nonce: 0group.public }",
)?;
let expected_bytes = expected.to_bytes_le()?;
assert_eq!(expected, Record::read_le(&expected_bytes[..])?);
assert!(Record::<CurrentNetwork, Plaintext<CurrentNetwork>>::read_le(&expected_bytes[1..]).is_err());
Ok(())
}
}