From 9a986c2fc6c52ae444a773078587fe0bb6974a64 Mon Sep 17 00:00:00 2001
From: Brendan Taylor <bct@diffeq.com>
Date: Sun, 16 Sep 2007 21:56:42 +0000
Subject: [PATCH] fixed imports

---
 src/common/crypto.py | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/src/common/crypto.py b/src/common/crypto.py
index 5dc7c7cfa..190685f11 100644
--- a/src/common/crypto.py
+++ b/src/common/crypto.py
@@ -1,5 +1,10 @@
 # common crypto functions (mostly specific to XEP-0116, but useful elsewhere)
 
+import os
+import math
+
+from Crypto.Hash import SHA256
+
 # convert a large integer to a big-endian bitstring
 def encode_mpi(n):
 	if n >= 256:
@@ -8,15 +13,23 @@ def encode_mpi(n):
 		return chr(n)
 
 # convert a large integer to a big-endian bitstring, padded with \x00s to
-# 16 bytes
+# a multiple of 16 bytes
 def encode_mpi_with_padding(n):
-	ret = encode_mpi(n)
+	return pad_to_multiple(encode_mpi(n), 16, '\x00', True)
 
-	mod = len(ret) % 16
-	if mod != 0:
-		ret = ((16 - mod) * '\x00') + ret
+# pad 'string' to a multiple of 'multiple_of' with 'char'.
+# pad on the left if 'left', otherwise pad on the right.
+def pad_to_multiple(string, multiple_of, char, left):
+	mod = len(string) % multiple_of
+	if mod == 0:
+		return string
+	else:
+		padding = (multiple_of - mod) * char
 
-	return ret
+	if left:
+		return padding + string
+	else:
+		return string + padding
 
 # convert a big-endian bitstring to an integer
 def decode_mpi(s):