forked from mirrors/gecko-dev
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
'''Tests block operations.'''
|
|
from rsa._compat import b
|
|
|
|
try:
|
|
from StringIO import StringIO as BytesIO
|
|
except ImportError:
|
|
from io import BytesIO
|
|
import unittest2
|
|
|
|
import rsa
|
|
from rsa import bigfile, varblock, pkcs1
|
|
|
|
class BigfileTest(unittest2.TestCase):
|
|
|
|
def test_encrypt_decrypt_bigfile(self):
|
|
|
|
# Expected block size + 11 bytes padding
|
|
pub_key, priv_key = rsa.newkeys((6 + 11) * 8)
|
|
|
|
# Encrypt the file
|
|
message = b('123456Sybren')
|
|
infile = BytesIO(message)
|
|
outfile = BytesIO()
|
|
|
|
bigfile.encrypt_bigfile(infile, outfile, pub_key)
|
|
|
|
# Test
|
|
crypto = outfile.getvalue()
|
|
|
|
cryptfile = BytesIO(crypto)
|
|
clearfile = BytesIO()
|
|
|
|
bigfile.decrypt_bigfile(cryptfile, clearfile, priv_key)
|
|
self.assertEquals(clearfile.getvalue(), message)
|
|
|
|
# We have 2x6 bytes in the message, so that should result in two
|
|
# bigfile.
|
|
cryptfile.seek(0)
|
|
varblocks = list(varblock.yield_varblocks(cryptfile))
|
|
self.assertEqual(2, len(varblocks))
|
|
|
|
|
|
def test_sign_verify_bigfile(self):
|
|
|
|
# Large enough to store MD5-sum and ASN.1 code for MD5
|
|
pub_key, priv_key = rsa.newkeys((34 + 11) * 8)
|
|
|
|
# Sign the file
|
|
msgfile = BytesIO(b('123456Sybren'))
|
|
signature = pkcs1.sign(msgfile, priv_key, 'MD5')
|
|
|
|
# Check the signature
|
|
msgfile.seek(0)
|
|
self.assertTrue(pkcs1.verify(msgfile, signature, pub_key))
|
|
|
|
# Alter the message, re-check
|
|
msgfile = BytesIO(b('123456sybren'))
|
|
self.assertRaises(pkcs1.VerificationError,
|
|
pkcs1.verify, msgfile, signature, pub_key)
|
|
|