def test5_AES(self):
plain = "this is a test"
key = os.urandom(16)
iv = os.urandom(8)
crypter1 = mycrypto.PayloadCrypter()
crypter1.setSessionKey(key, iv)
crypter2 = mycrypto.PayloadCrypter()
crypter2.setSessionKey(key, iv)
cipher = crypter1.encrypt(plain)
self.failIf(cipher == plain)
self.failUnless(crypter2.decrypt(cipher) == plain)
def test2_getPadding(self):
pm = packetmorpher.new()
sendCrypter = mycrypto.PayloadCrypter()
sendCrypter.setSessionKey("A" * 32, "A" * 8)
sendHMAC = "A" * 32
for i in range(0, const.MTU + 2):
padLen = len(pm.getPadding(sendCrypter, sendHMAC, i))
self.assertTrue(const.HDR_LENGTH <= padLen < const.MTU + \
const.HDR_LENGTH)