diff options
author | Jaime Marquínez Ferrándiz <jaime.marquinez.ferrandiz@gmail.com> | 2015-03-21 12:07:23 +0100 |
---|---|---|
committer | Jaime Marquínez Ferrándiz <jaime.marquinez.ferrandiz@gmail.com> | 2015-03-21 12:07:23 +0100 |
commit | a7d9ded45dac709fbb9140ab3525a10a2a6d3c85 (patch) | |
tree | e2627dd65b8afcba23e5b2db48c4dd86300f2b1e /devscripts | |
parent | 531980d89c37427c663152a525fd795dc0b040e4 (diff) | |
download | youtube-dl-a7d9ded45dac709fbb9140ab3525a10a2a6d3c85.zip youtube-dl-a7d9ded45dac709fbb9140ab3525a10a2a6d3c85.tar.gz youtube-dl-a7d9ded45dac709fbb9140ab3525a10a2a6d3c85.tar.bz2 |
[test] Add tests for aes
Diffstat (limited to 'devscripts')
-rw-r--r-- | devscripts/generate_aes_testdata.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/devscripts/generate_aes_testdata.py b/devscripts/generate_aes_testdata.py new file mode 100644 index 0000000..ff66449 --- /dev/null +++ b/devscripts/generate_aes_testdata.py @@ -0,0 +1,36 @@ +from __future__ import unicode_literals + +import codecs +import subprocess + +import os +import sys +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) + +from youtube_dl.utils import intlist_to_bytes +from youtube_dl.aes import aes_encrypt, key_expansion + +secret_msg = b'Secret message goes here' + + +def hex_str(int_list): + return codecs.encode(intlist_to_bytes(int_list), 'hex') + + +def openssl_encode(algo, key, iv): + cmd = ['openssl', 'enc', '-e', '-' + algo, '-K', hex_str(key), '-iv', hex_str(iv)] + prog = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE) + out, _ = prog.communicate(secret_msg) + return out + +iv = key = [0x20, 0x15] + 14 * [0] + +r = openssl_encode('aes-128-cbc', key, iv) +print('aes_cbc_decrypt') +print(repr(r)) + +password = key +new_key = aes_encrypt(password, key_expansion(password)) +r = openssl_encode('aes-128-ctr', new_key, iv) +print('aes_decrypt_text') +print(repr(r)) |