snarkvm_console_program/data/record/
to_fields.rs

1// Copyright 2024 Aleo Network Foundation
2// This file is part of the snarkVM library.
3
4// Licensed under the Apache License, Version 2.0 (the "License");
5// you may not use this file except in compliance with the License.
6// You may obtain a copy of the License at:
7
8// http://www.apache.org/licenses/LICENSE-2.0
9
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13// See the License for the specific language governing permissions and
14// limitations under the License.
15
16use super::*;
17
18impl<N: Network> ToFields for Record<N, Plaintext<N>> {
19    type Field = Field<N>;
20
21    /// Returns this record as a list of field elements.
22    fn to_fields(&self) -> Result<Vec<Self::Field>> {
23        // Encode the data as little-endian bits.
24        let mut bits_le = self.to_bits_le();
25        // Adds one final bit to the data, to serve as a terminus indicator.
26        // During decryption, this final bit ensures we've reached the end.
27        bits_le.push(true);
28        // Pack the bits into field elements.
29        let fields = bits_le
30            .chunks(Field::<N>::size_in_data_bits())
31            .map(Field::<N>::from_bits_le)
32            .collect::<Result<Vec<_>>>()?;
33        // Ensure the number of field elements does not exceed the maximum allowed size.
34        match fields.len() <= N::MAX_DATA_SIZE_IN_FIELDS as usize {
35            true => Ok(fields),
36            false => bail!("Record<Plaintext> exceeds maximum allowed size"),
37        }
38    }
39}
40
41impl<N: Network> ToFields for Record<N, Ciphertext<N>> {
42    type Field = Field<N>;
43
44    /// Returns this record as a list of field elements.
45    fn to_fields(&self) -> Result<Vec<Self::Field>> {
46        // Encode the data as little-endian bits.
47        let mut bits_le = self.to_bits_le();
48        // Adds one final bit to the data, to serve as a terminus indicator.
49        // During decryption, this final bit ensures we've reached the end.
50        bits_le.push(true);
51        // Pack the bits into field elements.
52        let fields = bits_le
53            .chunks(Field::<N>::size_in_data_bits())
54            .map(Field::<N>::from_bits_le)
55            .collect::<Result<Vec<_>>>()?;
56        // Ensure the number of field elements does not exceed the maximum allowed size.
57        match fields.len() <= N::MAX_DATA_SIZE_IN_FIELDS as usize {
58            true => Ok(fields),
59            false => bail!("Record<Ciphertext> exceeds maximum allowed size"),
60        }
61    }
62}