diff options
author | Maximilian Hils <git@maximilianhils.com> | 2016-03-16 12:16:52 +0100 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2016-03-16 12:16:52 +0100 |
commit | 2a20fc59b28778657984d774cc006cfe7314586e (patch) | |
tree | 8575b4387ef1ceaa6a871807a0543c2e8d76c9e9 /test/netlib | |
parent | 0ab9e60168a2b0ee305ebff47b0532ea7acfecee (diff) | |
parent | ba933dff2c342ed0095f28da2e6f90a0d12bcd3d (diff) | |
download | mitmproxy-2a20fc59b28778657984d774cc006cfe7314586e.tar.gz mitmproxy-2a20fc59b28778657984d774cc006cfe7314586e.tar.bz2 mitmproxy-2a20fc59b28778657984d774cc006cfe7314586e.zip |
Merge pull request #1030 from xhy940801/socks
Add 'UsernamePasswordAuth' 'UsernamePasswordAuthResponse' to SOCKS
Diffstat (limited to 'test/netlib')
-rw-r--r-- | test/netlib/test_socks.py | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/test/netlib/test_socks.py b/test/netlib/test_socks.py index d95dee41..486b975b 100644 --- a/test/netlib/test_socks.py +++ b/test/netlib/test_socks.py @@ -85,6 +85,45 @@ def test_server_greeting_assert_socks5(): assert False +def test_username_password_auth(): + raw = tutils.treader(b"\x01\x03usr\x03psd\xBE\xEF") + out = BytesIO() + auth = socks.UsernamePasswordAuth.from_file(raw) + auth.assert_authver1() + assert raw.read(2) == b"\xBE\xEF" + auth.to_file(out) + + assert out.getvalue() == raw.getvalue()[:-2] + assert auth.ver == socks.USERNAME_PASSWORD_VERSION.DEFAULT + assert auth.username == "usr" + assert auth.password == "psd" + + +def test_username_password_auth_assert_ver1(): + raw = tutils.treader(b"\x02\x03usr\x03psd\xBE\xEF") + auth = socks.UsernamePasswordAuth.from_file(raw) + tutils.raises(socks.SocksError, auth.assert_authver1) + + +def test_username_password_auth_response(): + raw = tutils.treader(b"\x01\x00\xBE\xEF") + out = BytesIO() + auth = socks.UsernamePasswordAuthResponse.from_file(raw) + auth.assert_authver1() + assert raw.read(2) == b"\xBE\xEF" + auth.to_file(out) + + assert out.getvalue() == raw.getvalue()[:-2] + assert auth.ver == socks.USERNAME_PASSWORD_VERSION.DEFAULT + assert auth.status == 0 + + +def test_username_password_auth_response_auth_assert_ver1(): + raw = tutils.treader(b"\x02\x00\xBE\xEF") + auth = socks.UsernamePasswordAuthResponse.from_file(raw) + tutils.raises(socks.SocksError, auth.assert_authver1) + + def test_message(): raw = tutils.treader(b"\x05\x01\x00\x03\x0bexample.com\xDE\xAD\xBE\xEF") out = BytesIO() |