diff options
| author | Donald Stufft <donald@stufft.io> | 2013-08-10 16:01:56 -0400 |
|---|---|---|
| committer | Donald Stufft <donald@stufft.io> | 2013-08-10 16:01:56 -0400 |
| commit | ad89af11b234524e543284d3095a15905518ad70 (patch) | |
| tree | 85634a84d21455faa897f5f4f6cbd62e4209a2e8 /cryptography | |
| parent | e9d8a777f2db59b11f745eb38d44f7257b40f5ab (diff) | |
| download | cryptography-ad89af11b234524e543284d3095a15905518ad70.tar.gz cryptography-ad89af11b234524e543284d3095a15905518ad70.tar.bz2 cryptography-ad89af11b234524e543284d3095a15905518ad70.zip | |
Ensure that AES gets a proper key size
Diffstat (limited to 'cryptography')
| -rw-r--r-- | cryptography/primitives/block/ciphers.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/cryptography/primitives/block/ciphers.py b/cryptography/primitives/block/ciphers.py index f40fc2a1..2bf936ee 100644 --- a/cryptography/primitives/block/ciphers.py +++ b/cryptography/primitives/block/ciphers.py @@ -14,11 +14,17 @@ class AES(object): name = "AES" + key_sizes = {128, 192, 256} def __init__(self, key): super(AES, self).__init__() self.key = key + if self.key_size not in self.key_sizes: + raise ValueError("Invalid key size (%s) for %s".format( + self.key_size, self.name + )) + @property def key_size(self): return len(self.key) * 8 |