summaryrefslogtreecommitdiff
path: root/rsa
diff options
context:
space:
mode:
authorYesudeep Mangalapilly <yesudeep@gmail.com>2011-08-24 16:26:39 +0530
committerYesudeep Mangalapilly <yesudeep@gmail.com>2011-08-24 16:26:39 +0530
commit3753c661e523439a2a9f54b2830092806478c117 (patch)
tree2845e44d9749b8eb5b8649cd61c23e2c64b64b52 /rsa
parent3f38343bf048266883124715a1e3be27e551defa (diff)
downloadrsa-3753c661e523439a2a9f54b2830092806478c117.tar.gz
Allow fill_size=None, chunk_size=None in int2bytes.
Diffstat (limited to 'rsa')
-rw-r--r--rsa/transform.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/rsa/transform.py b/rsa/transform.py
index 9abc6ae..511233b 100644
--- a/rsa/transform.py
+++ b/rsa/transform.py
@@ -121,7 +121,7 @@ def bytes_leading(raw_bytes, needle=ZERO_BYTE):
return leading
-def int2bytes(number, fill_size=0, chunk_size=0, overflow=False):
+def int2bytes(number, fill_size=None, chunk_size=None, overflow=False):
"""
Convert an unsigned integer to bytes (base-256 representation)::
@@ -162,7 +162,7 @@ def int2bytes(number, fill_size=0, chunk_size=0, overflow=False):
raise ValueError("You can either fill or pad chunks, but not both")
# Ensure these are integers.
- number & 1 and chunk_size & 1 and fill_size & 1
+ number & 1
raw_bytes = b('')
@@ -181,14 +181,14 @@ def int2bytes(number, fill_size=0, chunk_size=0, overflow=False):
raw_bytes = raw_bytes[zero_leading:]
length = len(raw_bytes)
- if fill_size > 0:
+ if fill_size and fill_size > 0:
if not overflow and length > fill_size:
raise OverflowError(
"Need %d bytes for number, but fill size is %d" %
(length, fill_size)
)
raw_bytes = raw_bytes.rjust(fill_size, ZERO_BYTE)
- elif chunk_size > 0:
+ elif chunk_size and chunk_size > 0:
remainder = length % chunk_size
if remainder:
padding_size = chunk_size - remainder