Python parser Examples

Example #1

File: test_protocols.py Project: 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

Example #2

File: test_protocols.py Project: 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