Python parserの例

コード例 #1

ファイル: test_protocols.py プロジェクト: ownermz/shadowproxy

def test_ss():
    cipher = AES256CFB(secrets.token_urlsafe(20))
    iv, encrypt = cipher.make_encrypter()
    length = len(iv) // 2
    parser = ss_reader.parser(cipher)
    parser.send(iv[:length])
    assert parser.read_output_bytes() == b""
    data = os.urandom(20)
    parser.send(iv[length:] + encrypt(data))
    assert parser.read_output_bytes() == data

コード例 #2

ファイル: test_protocols.py プロジェクト: ownermz/shadowproxy

def test_ss2():
    cipher = AES256CFB(secrets.token_urlsafe(20))
    iv, encrypt = cipher.make_encrypter()
    parser = ss_reader.parser(cipher)
    parser.send(iv)
    assert parser.read_output_bytes() == b""
    assert parser.read_output_bytes() == b""
    data = os.urandom(20)
    parser.send(encrypt(data))
    assert parser.read_output_bytes() == data