arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-22 15:38:11 +00:00
Code Issues Releases Wiki Activity

core/monero: add gc.collect before large allocations in bulletproof code

Browse Source
This commit is contained in:
Pavol Rusnak 2019-09-29 17:42:18 +00:00
parent 5f980b50a0
commit ed0336c0a9
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
core/src/apps/monero/xmr/bulletproof.py
View File

@ -344,10 +344,12 @@ class KeyV(KeyVBase):
def _set_d(self, elems): def _set_d(self, elems):
if elems > 64 and elems % 64 == 0: if elems > 64 and elems % 64 == 0:
self.chunked = True self.chunked = True
gc.collect()
self.d = [bytearray(32 * 64) for _ in range(elems // 64)] self.d = [bytearray(32 * 64) for _ in range(elems // 64)]
else: else:
self.chunked = False self.chunked = False
gc.collect()
self.d = bytearray(32 * elems) self.d = bytearray(32 * elems)
def _set_mv(self): def _set_mv(self):
@ -401,10 +403,13 @@ class KeyV(KeyVBase):
if self.chunked and nsize <= 64: if self.chunked and nsize <= 64:
self.chunked = False # de-chunk self.chunked = False # de-chunk
if self.size > nsize and realloc: if self.size > nsize and realloc:
gc.collect()
self.d = bytearray(self.d[0][: nsize << 5]) self.d = bytearray(self.d[0][: nsize << 5])
elif self.size > nsize and not chop: elif self.size > nsize and not chop:
gc.collect()
self.d = self.d[0][: nsize << 5] self.d = self.d[0][: nsize << 5]
else: else:
gc.collect()
self.d = bytearray(nsize << 5) self.d = bytearray(nsize << 5)
elif self.chunked: elif self.chunked:
@ -412,10 +417,13 @@ class KeyV(KeyVBase):
else: else:
if self.size > nsize and realloc: if self.size > nsize and realloc:
gc.collect()
self.d = bytearray(self.d[: nsize << 5]) self.d = bytearray(self.d[: nsize << 5])
elif self.size > nsize and not chop: elif self.size > nsize and not chop:
gc.collect()
self.d = self.d[: nsize << 5] self.d = self.d[: nsize << 5]
else: else:
gc.collect()
self.d = bytearray(nsize << 5) self.d = bytearray(nsize << 5)
self.size = nsize self.size = nsize
@ -1005,6 +1013,7 @@ class BulletProofBuilder:
) )
def sX_gen(self, ln=BP_N): def sX_gen(self, ln=BP_N):
gc.collect()
buff = bytearray(ln * 32) buff = bytearray(ln * 32)
buff_mv = memoryview(buff) buff_mv = memoryview(buff)
sc = crypto.new_scalar() sc = crypto.new_scalar()
@ -1029,6 +1038,7 @@ class BulletProofBuilder:
self.proof_sec = crypto.random_bytes(64) self.proof_sec = crypto.random_bytes(64)
self._det_mask_init() self._det_mask_init()
gc.collect()
sv = [crypto.encodeint(x) for x in sv] sv = [crypto.encodeint(x) for x in sv]
gamma = [crypto.encodeint(x) for x in gamma] gamma = [crypto.encodeint(x) for x in gamma]