From 3753c661e523439a2a9f54b2830092806478c117 Mon Sep 17 00:00:00 2001 From: Yesudeep Mangalapilly Date: Wed, 24 Aug 2011 16:26:39 +0530 Subject: Allow fill_size=None, chunk_size=None in int2bytes. --- rsa/transform.py | 8 ++++---- 1 file 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 -- cgit v1.2.1