snarkvm_synthesizer_program/closure/
mod.rs

1// Copyright (c) 2019-2026 Provable Inc.
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 crate::Instruction;
17
18mod input;
19use input::*;
20
21mod output;
22use output::*;
23
24mod bytes;
25mod parse;
26
27use console::{
28    network::{error, prelude::*},
29    program::{Identifier, Register, RegisterType},
30};
31
32use indexmap::IndexSet;
33
34#[derive(Clone, PartialEq, Eq)]
35pub struct ClosureCore<N: Network> {
36    /// The name of the closure.
37    name: Identifier<N>,
38    /// The input statements, added in order of the input registers.
39    /// Input assignments are ensured to match the ordering of the input statements.
40    inputs: IndexSet<Input<N>>,
41    /// The instructions, in order of execution.
42    instructions: Vec<Instruction<N>>,
43    /// The output statements, in order of the desired output.
44    outputs: IndexSet<Output<N>>,
45}
46
47impl<N: Network> ClosureCore<N> {
48    /// Initializes a new closure with the given name.
49    pub fn new(name: Identifier<N>) -> Self {
50        Self { name, inputs: IndexSet::new(), instructions: Vec::new(), outputs: IndexSet::new() }
51    }
52
53    /// Returns the name of the closure.
54    pub const fn name(&self) -> &Identifier<N> {
55        &self.name
56    }
57
58    /// Returns the closure inputs.
59    pub const fn inputs(&self) -> &IndexSet<Input<N>> {
60        &self.inputs
61    }
62
63    /// Returns the closure instructions.
64    pub fn instructions(&self) -> &[Instruction<N>] {
65        &self.instructions
66    }
67
68    /// Returns the closure outputs.
69    pub const fn outputs(&self) -> &IndexSet<Output<N>> {
70        &self.outputs
71    }
72
73    /// Returns the closure output types.
74    pub fn output_types(&self) -> Vec<RegisterType<N>> {
75        self.outputs.iter().map(|output| output.register_type()).cloned().collect()
76    }
77
78    /// Returns whether the closure refers to an external struct.
79    pub fn contains_external_struct(&self) -> bool {
80        self.inputs.iter().any(|input| input.register_type().contains_external_struct())
81            || self.outputs.iter().any(|output| output.register_type().contains_external_struct())
82            || self.instructions.iter().any(|instruction| instruction.contains_external_struct())
83    }
84
85    /// Returns `true` if the closure instructions contain a string type.
86    /// Note that input and output types don't have to be checked if we are sure the broader function doesn't contain a string type.
87    pub fn contains_string_type(&self) -> bool {
88        self.instructions.iter().any(|instruction| instruction.contains_string_type())
89    }
90
91    /// Returns `true` if the closure contains an identifier type in its inputs, outputs, or instructions.
92    pub fn contains_identifier_type(&self) -> Result<bool> {
93        for input in &self.inputs {
94            if input.register_type().contains_identifier_type()? {
95                return Ok(true);
96            }
97        }
98        for output in &self.outputs {
99            if output.register_type().contains_identifier_type()? {
100                return Ok(true);
101            }
102        }
103        // Check instruction-level types (e.g., cast destination types).
104        for instruction in &self.instructions {
105            if instruction.contains_identifier_type()? {
106                return Ok(true);
107            }
108        }
109        Ok(false)
110    }
111
112    /// Returns `true` if the closure contains an array type with a size that exceeds the given maximum.
113    pub fn exceeds_max_array_size(&self, max_array_size: u32) -> bool {
114        self.inputs.iter().any(|input| input.register_type().exceeds_max_array_size(max_array_size))
115            || self.outputs.iter().any(|output| output.register_type().exceeds_max_array_size(max_array_size))
116            || self.instructions.iter().any(|instruction| instruction.exceeds_max_array_size(max_array_size))
117    }
118}
119
120impl<N: Network> ClosureCore<N> {
121    /// Adds the input statement to the closure.
122    ///
123    /// # Errors
124    /// This method will halt if there are instructions or output statements already.
125    /// This method will halt if the maximum number of inputs has been reached.
126    /// This method will halt if the input statement was previously added.
127    #[inline]
128    fn add_input(&mut self, input: Input<N>) -> Result<()> {
129        // Ensure there are no instructions or output statements in memory.
130        ensure!(self.instructions.is_empty(), "Cannot add inputs after instructions have been added");
131        ensure!(self.outputs.is_empty(), "Cannot add inputs after outputs have been added");
132
133        // Ensure the maximum number of inputs has not been exceeded.
134        ensure!(self.inputs.len() < N::MAX_INPUTS, "Cannot add more than {} inputs", N::MAX_INPUTS);
135        // Ensure the input statement was not previously added.
136        ensure!(!self.inputs.contains(&input), "Cannot add duplicate input statement");
137
138        // Ensure the input register is a locator.
139        ensure!(matches!(input.register(), Register::Locator(..)), "Input register must be a locator");
140
141        // Insert the input statement.
142        self.inputs.insert(input);
143        Ok(())
144    }
145
146    /// Adds the given instruction to the closure.
147    ///
148    /// # Errors
149    /// This method will halt if there are output statements already.
150    /// This method will halt if the maximum number of instructions has been reached.
151    #[inline]
152    pub fn add_instruction(&mut self, instruction: Instruction<N>) -> Result<()> {
153        // Ensure that there are no outputs in memory.
154        ensure!(self.outputs.is_empty(), "Cannot add instructions after outputs have been added");
155
156        // Ensure the maximum number of instructions has not been exceeded.
157        ensure!(
158            self.instructions.len() < N::MAX_INSTRUCTIONS,
159            "Cannot add more than {} instructions",
160            N::MAX_INSTRUCTIONS
161        );
162
163        // Ensure the destination register is a locator.
164        for register in instruction.destinations() {
165            ensure!(matches!(register, Register::Locator(..)), "Destination register must be a locator");
166        }
167
168        // Insert the instruction.
169        self.instructions.push(instruction);
170        Ok(())
171    }
172
173    /// Adds the output statement to the closure.
174    ///
175    /// # Errors
176    /// This method will halt if the maximum number of outputs has been reached.
177    #[inline]
178    fn add_output(&mut self, output: Output<N>) -> Result<()> {
179        // Ensure the maximum number of outputs has not been exceeded.
180        ensure!(self.outputs.len() < N::MAX_OUTPUTS, "Cannot add more than {} outputs", N::MAX_OUTPUTS);
181
182        // Ensure the closure output register is not a static record.
183        // ExternalRecord and DynamicRecord are checked at V15+ deployment time (see `VM::check_transaction`).
184        ensure!(!matches!(output.register_type(), RegisterType::Record(..)), "Closure outputs do not support records");
185
186        // Insert the output statement.
187        self.outputs.insert(output);
188        Ok(())
189    }
190}
191
192impl<N: Network> TypeName for ClosureCore<N> {
193    /// Returns the type name as a string.
194    #[inline]
195    fn type_name() -> &'static str {
196        "closure"
197    }
198}
199
200#[cfg(test)]
201mod tests {
202    use super::*;
203
204    use crate::{Closure, Instruction};
205
206    type CurrentNetwork = console::network::MainnetV0;
207
208    #[test]
209    fn test_add_input() {
210        // Initialize a new closure instance.
211        let name = Identifier::from_str("closure_core_test").unwrap();
212        let mut closure = Closure::<CurrentNetwork>::new(name);
213
214        // Ensure that an input can be added.
215        let input = Input::<CurrentNetwork>::from_str("input r0 as field;").unwrap();
216        assert!(closure.add_input(input.clone()).is_ok());
217
218        // Ensure that adding a duplicate input will fail.
219        assert!(closure.add_input(input).is_err());
220
221        // Ensure that adding more than the maximum number of inputs will fail.
222        for i in 1..CurrentNetwork::MAX_INPUTS * 2 {
223            let input = Input::<CurrentNetwork>::from_str(&format!("input r{i} as field;")).unwrap();
224
225            match closure.inputs.len() < CurrentNetwork::MAX_INPUTS {
226                true => assert!(closure.add_input(input).is_ok()),
227                false => assert!(closure.add_input(input).is_err()),
228            }
229        }
230    }
231
232    #[test]
233    fn test_add_instruction() {
234        // Initialize a new closure instance.
235        let name = Identifier::from_str("closure_core_test").unwrap();
236        let mut closure = Closure::<CurrentNetwork>::new(name);
237
238        // Ensure that an instruction can be added.
239        let instruction = Instruction::<CurrentNetwork>::from_str("add r0 r1 into r2;").unwrap();
240        assert!(closure.add_instruction(instruction).is_ok());
241
242        // Ensure that adding more than the maximum number of instructions will fail.
243        for i in 3..CurrentNetwork::MAX_INSTRUCTIONS * 2 {
244            let instruction = Instruction::<CurrentNetwork>::from_str(&format!("add r0 r1 into r{i};")).unwrap();
245
246            match closure.instructions.len() < CurrentNetwork::MAX_INSTRUCTIONS {
247                true => assert!(closure.add_instruction(instruction).is_ok()),
248                false => assert!(closure.add_instruction(instruction).is_err()),
249            }
250        }
251    }
252
253    #[test]
254    fn test_add_output() {
255        // Initialize a new closure instance.
256        let name = Identifier::from_str("closure_core_test").unwrap();
257        let mut closure = Closure::<CurrentNetwork>::new(name);
258
259        // Ensure that an output can be added.
260        let output = Output::<CurrentNetwork>::from_str("output r0 as field;").unwrap();
261        assert!(closure.add_output(output).is_ok());
262
263        // Ensure that adding more than the maximum number of outputs will fail.
264        for i in 1..CurrentNetwork::MAX_OUTPUTS * 2 {
265            let output = Output::<CurrentNetwork>::from_str(&format!("output r{i} as field;")).unwrap();
266
267            match closure.outputs.len() < CurrentNetwork::MAX_OUTPUTS {
268                true => assert!(closure.add_output(output).is_ok()),
269                false => assert!(closure.add_output(output).is_err()),
270            }
271        }
272    }
273
274    #[test]
275    fn test_add_output_record_types_rejected() {
276        let name = Identifier::from_str("closure_core_test").unwrap();
277
278        // DynamicRecord output is allowed at parse time (gated at V15+ deployment time).
279        let mut closure = Closure::<CurrentNetwork>::new(name);
280        let output = Output::<CurrentNetwork>::from_str("output r0 as dynamic.record;").unwrap();
281        assert!(closure.add_output(output).is_ok());
282
283        // ExternalRecord output is allowed at parse time (gated at V15+ deployment time).
284        let mut closure = Closure::<CurrentNetwork>::new(name);
285        let output = Output::<CurrentNetwork>::from_str("output r0 as test_prog.aleo/token.record;").unwrap();
286        assert!(closure.add_output(output).is_ok());
287
288        // Static Record output is rejected at parse time.
289        let mut closure = Closure::<CurrentNetwork>::new(name);
290        let output = Output::<CurrentNetwork>::from_str("output r0 as token.record;").unwrap();
291        let result = closure.add_output(output);
292        assert!(result.is_err());
293        assert!(result.unwrap_err().to_string().contains("Closure outputs do not support records"));
294    }
295}
snarkvm_synthesizer_program/closure/mod.rs

snarkvm_synthesizer_program/closure/
mod.rs
