xmr: bp - KeyVEval fix

trezor · Aug 17, 2018 · d7e9dab · d7e9dab
1 parent 1523f40
commit d7e9dab
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/apps/monero/xmr/bulletproof.py b/src/apps/monero/xmr/bulletproof.py
@@ -591,10 +591,10 @@ def aR(self, i, dst=None):
         return dst
 
     def aL_vct(self):
-        return KeyVEval(64, lambda x: self.aL(x))
+        return KeyVEval(64, lambda x, r: self.aL(x, r))
 
     def aR_vct(self):
-        return KeyVEval(64, lambda x: self.aR(x))
+        return KeyVEval(64, lambda x, r: self.aR(x, r))
 
     def _det_mask(self, i, is_sL=True, dst=None):
         dst = _ensure_dst_key(dst)
@@ -615,12 +615,16 @@ def sR(self, i, dst=None):
 
     def sL_vct(self):
         return (
-            KeyVEval(64, lambda x: self.sL(x)) if self.use_det_masks else self.sX_gen()
+            KeyVEval(64, lambda x, r: self.sL(x, r))
+            if self.use_det_masks
+            else self.sX_gen()
         )
 
     def sR_vct(self):
         return (
-            KeyVEval(64, lambda x: self.sR(x)) if self.use_det_masks else self.sX_gen()
+            KeyVEval(64, lambda x, r: self.sR(x, r))
+            if self.use_det_masks
+            else self.sX_gen()
         )
 
     def sX_gen(self):