github-mig/ZoKrates
github-mig/ZoKrates
1
0
Fork 0
mirror of synced 2025-09-23 04:08:33 +00:00
Code Issues Projects Releases Wiki Activity

implement fold_program to avoid calling fold_module more than expected

Browse source
This commit is contained in:
schaeff 2021-07-25 18:31:22 +02:00
parent a1cf614f53
commit a98701309d
2 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
changelogs/unreleased/949-schaeff Normal file
View file

@ -0,0 +1 @@
Fix state corruption in the constant inliner

11
zokrates_core/src/static_analysis/constant_inliner.rs
View file

@ -67,6 +67,17 @@ impl<'ast, 'a, T: Field> ConstantInliner<'ast, T> {
}
impl<'ast, T: Field> Folder<'ast, T> for ConstantInliner<'ast, T> {
fn fold_program(&mut self, p: TypedProgram<'ast, T>) -> TypedProgram<'ast, T> {
for (id, _) in p.modules {
self.fold_module_id(id);
}
TypedProgram {
modules: std::mem::take(&mut self.modules),
..p
}
}
fn fold_module_id(&mut self, id: OwnedTypedModuleId) -> OwnedTypedModuleId {
// anytime we encounter a module id, visit the corresponding module if it hasn't been done yet
if !self.treated(&id) {