Backup_Repos/calibre
1
0
Fork 0
mirror of git://github.com/kovidgoyal/calibre.git synced 2026-05-08 20:14:33 +02:00
Code Issues Projects Releases Wiki Activity

Make the number of bytes used by the counter configurable

Browse source
This commit is contained in:
Kovid Goyal 2016-03-22 09:34:01 +05:30
parent cae0e38a97
commit 48f80f2a81
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/pyj/aes.pyj
View file

@ -309,18 +309,18 @@ def decrypt(self, output_from_encrypt, tag):
class Counter:
def __init__(self, initial_value):
self.bytes = Uint8Array(16)
if not initial_value:
self.set_value(1)
elif type(initial_value) is 'number':
def __init__(self, initial_value, number_of_bytes=16):
self.number_of_bytes = number_of_bytes
self.bytes = Uint8Array(number_of_bytes)
initial_value = initial_value or 1
if type(initial_value) is 'number':
self.set_value(initial_value)
else:
self.bytes.set(initial_value)
def set_value(self, value):
c = self.bytes
for v'var index = 15; index >= 0; index--':
for v'var index = self.number_of_bytes; index >= 0; index--':
c[index] = value % 256
value >>= 8
@ -329,7 +329,7 @@ def set_bytes(self, bytes):
def increment(self):
c = self.bytes
for v'var i = 15; i >= 0; i--':
for v'var i = self.number_of_bytes; i >= 0; i--':
if c[i] is 255:
c[i] = 0
else: