tests/testcipher.py

   1 from ctypescrypto.oid import Oid
   2 from ctypescrypto import cipher
   3 import unittest
   4 import sys
   5
   6
   7 class TestCipherType(unittest.TestCase):
   8     def test_ciphdescbc(self):
   9         ct=cipher.CipherType("des-cbc")
  10         self.assertEqual(ct.block_size(),8)
  11         self.assertEqual(ct.key_length(),8)
  12         self.assertEqual(ct.iv_length(),8)
  13         self.assertEqual(ct.algo(),'DES-CBC')
  14         self.assertEqual(ct.oid().shortname(),"DES-CBC")
  15         self.assertEqual(ct.mode(),"CBC")
  16     def test_ciphaesofb(self):
  17         ct=cipher.CipherType("aes-256-ofb")
  18         self.assertEqual(ct.block_size(),1)
  19         self.assertEqual(ct.key_length(),32)
  20         self.assertEqual(ct.iv_length(),16)
  21         self.assertEqual(ct.algo(),'AES-256-OFB')
  22         self.assertEqual(ct.oid().shortname(),"AES-256-OFB")
  23         self.assertEqual(ct.mode(),"OFB")
  24     def test_unknowncipher(self):
  25         with self.assertRaises(cipher.CipherError):
  26             ct=cipher.CipherType("no-such-cipher")
  27
  28 class TestEncryptDecrypt(unittest.TestCase):
  29     def test_cons_nokey(self):
  30         ct=cipher.CipherType("DES-CBC")
  31         with self.assertRaises(ValueError):
  32             c=cipher.Cipher(ct,None,None)
  33     def test_blockcipher(self):
  34         data="sdfdsddf"
  35         key='abcdabcd'
  36         c=cipher.new("DES-CBC",key)
  37         enc=c.update(data)+c.finish()
  38         # See if padding is added by default
  39         self.assertEqual(len(enc),16)
  40         d=cipher.new("DES-CBC",key,encrypt=False)
  41         dec=d.update(enc)+d.finish()
  42         self.assertEqual(data,dec)
  43     def test_blockcipher_nopadding(self):
  44         data="sdfdsddf"
  45         key='abcdabcd'
  46         c=cipher.new("DES-CBC",key)
  47         c.padding(False)
  48         enc=c.update(data)+c.finish()
  49         # See if padding is added by default
  50         self.assertEqual(len(enc),8)
  51         d=cipher.new("DES-CBC",key,encrypt=False)
  52         d.padding(False)
  53         dec=d.update(enc)+d.finish()
  54         self.assertEqual(data,dec)
  55     def test_ofb_cipher(self):
  56         data="sdfdsddfxx"
  57         key='abcdabcd'
  58         iv='abcdabcd'
  59         c=cipher.new("DES-OFB",key,iv=iv)
  60         enc=c.update(data)+c.finish()
  61         # See if padding is added by default
  62         self.assertEqual(len(enc),len(data))
  63         d=cipher.new("DES-OFB",key,encrypt=False,iv=iv)
  64         dec=d.update(enc)+d.finish()
  65         self.assertEqual(data,dec)
  66
  67     def test_ofb_noiv(self):
  68         data="sdfdsddfxx"
  69         encryptkey='abcdabcd'*4
  70         decryptkey=encryptkey[0:5]+encryptkey[5:]
  71
  72
  73         c=cipher.new("AES-256-OFB",encryptkey)
  74         enc=c.update(data)+c.finish()
  75         # See if padding is added by default
  76         self.assertEqual(len(enc),len(data))
  77         d=cipher.new("AES-256-OFB",decryptkey,encrypt=False)
  78         dec=d.update(enc)+d.finish()
  79         self.assertEqual(data,dec)
  80     def test_wrong_keylength(self):
  81         data="sdfsdfxxx"
  82         key="abcdabcd"
  83         with self.assertRaises(ValueError):
  84             c=cipher.new("AES-128-OFB",key)
  85     def test_wrong_ivlength(self):
  86         key="abcdabcdabcdabcd"
  87         iv="xxxxx"
  88         with self.assertRaises(ValueError):
  89             c=cipher.new("AES-128-OFB",key,iv=iv)
  90     def test_variable_keylength(self):
  91         encryptkey="abcdefabcdefghtlgvasdgdgsdgdsg"
  92         data="asdfsdfsdfsdfsdfsdfsdfsdf"
  93         iv="abcdefgh"
  94         c=cipher.new("bf-ofb",encryptkey,iv=iv)
  95         ciphertext=c.update(data)+c.finish()
  96         decryptkey=encryptkey[0:5]+encryptkey[5:]
  97         d=cipher.new("bf-ofb",decryptkey,encrypt=False,iv=iv)
  98         deciph=d.update(ciphertext)+d.finish()
  99         self.assertEqual(deciph,data)
 100
 101 if __name__ == '__main__':
 102     unittest.main()