vbatts/shadowsocks
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

update run_method()

Browse source
This commit is contained in:
mengskysama 2018-02-18 23:21:20 +08:00
parent 2ab8c6bf5d
commit d8a5d9a132
1 changed files with 3 additions and 6 deletions

9
shadowsocks/crypto/mbedtls.py
View file

@ -408,22 +408,20 @@ ciphers = {
def run_method(method): def run_method(method):
from shadowsocks.crypto import openssl
print(method, ': [stream]', 32) print(method, ': [stream]', 32)
cipher = MbedTLSStreamCrypto(method, b'k' * 32, b'i' * 16, 1) cipher = MbedTLSStreamCrypto(method, b'k' * 32, b'i' * 16, 1)
decipher = openssl.OpenSSLStreamCrypto(method, b'k' * 32, b'i' * 16, 0) decipher = MbedTLSStreamCrypto(method, b'k' * 32, b'i' * 16, 0)
util.run_cipher(cipher, decipher) util.run_cipher(cipher, decipher)
def run_aead_method(method, key_len=16): def run_aead_method(method, key_len=16):
from shadowsocks.crypto import openssl
print(method, ': [payload][tag]', key_len) print(method, ': [payload][tag]', key_len)
key_len = int(key_len) key_len = int(key_len)
cipher = MbedTLSAeadCrypto(method, b'k' * key_len, b'i' * key_len, 1) cipher = MbedTLSAeadCrypto(method, b'k' * key_len, b'i' * key_len, 1)
decipher = openssl.OpenSSLAeadCrypto( decipher = MbedTLSStreamCrypto(
method, method,
b'k' * key_len, b'i' * key_len, 0 b'k' * key_len, b'i' * key_len, 0
) )
@ -432,12 +430,11 @@ def run_aead_method(method, key_len=16):
def run_aead_method_chunk(method, key_len=16): def run_aead_method_chunk(method, key_len=16):
from shadowsocks.crypto import openssl
print(method, ': chunk([size][tag][payload][tag]', key_len) print(method, ': chunk([size][tag][payload][tag]', key_len)
key_len = int(key_len) key_len = int(key_len)
cipher = MbedTLSAeadCrypto(method, b'k' * key_len, b'i' * key_len, 1) cipher = MbedTLSAeadCrypto(method, b'k' * key_len, b'i' * key_len, 1)
decipher = openssl.OpenSSLAeadCrypto( decipher = MbedTLSStreamCrypto(
method, method,
b'k' * key_len, b'i' * key_len, 0 b'k' * key_len, b'i' * key_len, 0
) )