Fix bug when data structures are indexed by MemValue.

data61 · Dec 5, 2023 · 1f8f784 · 1f8f784
1 parent 3e31341
commit 1f8f784
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Compiler/types.py b/Compiler/types.py
@@ -5638,6 +5638,7 @@ def address(self):
     def address(self, address):
         self._address = address
 
+    @read_mem_value
     def get_address(self, index, size=None):
         if isinstance(index, (_secret, _single)):
             raise CompilerError('need cleartext index')
@@ -6146,6 +6147,7 @@ def __init__(self, sizes, value_type, address, index, debug=None):
         if debug:
             library.print_ln_if(self.address + reduce(operator.mul, self.sizes) * self.value_type.n_elements() > program.allocated_mem[self.value_type.reg_type], 'AOF%d:' % len(self.sizes) + self.debug)
 
+    @read_mem_value
     def __getitem__(self, index):
         """ Part access.