X-Git-Url: https://mudpy.org/gitweb?p=mudpy.git;a=blobdiff_plain;f=mudpy%2Ftests%2Fselftest.py;h=055e50a84c13464c9b29fd8b12c648cc74ed3737;hp=43e93ae192f3ad46e4c4dd64128d887b9fa4ad20;hb=de5bbdac1277ea31cb7e2fda1ad0df1595dc04ad;hpb=52d18039ae407b9c30b4b852c99e187b09854b81 diff --git a/mudpy/tests/selftest.py b/mudpy/tests/selftest.py index 43e93ae..055e50a 100644 --- a/mudpy/tests/selftest.py +++ b/mudpy/tests/selftest.py @@ -184,13 +184,19 @@ test_telnet_iac = ( (2, r"Non-ASCII characters from admin: b'say argle\\xffbargle'.*> ", ""), ) -test_telnet_unknown = ( +test_telnet_unknown_command = ( # Send an unsupported negotiation command #127 which should get filtered # from the line of input (2, "> ", b"say glop\xff\x7fglyf\r\0"), (2, r'Ignored unknown command 127 from admin\..*"Glopglyf\.".*> ', ""), ) +test_telnet_unknown_option = ( + # Send an unassigned negotiation option #127 which should get logged + (2, "> ", b"\xff\xfe\x7f\r\0"), + (2, r'''Received "don't 127" from admin\..*> ''', ""), +) + test_admin_restriction = ( (0, "> ", "help halt"), (0, r"That is not an available command\.", "halt"), @@ -317,7 +323,8 @@ dialogue = ( (test_preferences, "set and show preferences"), (test_crlf_eol, "send crlf from the client as eol"), (test_telnet_iac, "escape stray telnet iac bytes"), - (test_telnet_unknown, "strip unknown telnet command"), + (test_telnet_unknown_command, "strip unknown telnet command"), + (test_telnet_unknown_option, "log unknown telnet option"), (test_admin_restriction, "restricted admin commands"), (test_admin_help, "admin help"), (test_reload, "reload"), @@ -419,6 +426,17 @@ def tlog(message, quiet=False): return True +def option_callback(telnet_socket, command, option): + if option == b'\x7f': + # We use this unassigned option value as a canary, so short-circuit + # any response to avoid endlessly looping + pass + elif command in (telnetlib.DO, telnetlib.DONT): + telnet_socket.send(b"%s%s%s" % (telnetlib.IAC, telnetlib.WONT, option)) + elif command in (telnetlib.WILL, telnetlib.WONT): + telnet_socket.send(b"%s%s%s" % (telnetlib.IAC, telnetlib.DONT, option)) + + def main(): captures = ["", "", ""] lusers = [telnetlib.Telnet(), telnetlib.Telnet(), telnetlib.Telnet()] @@ -430,6 +448,7 @@ def main(): service = start_service(sys.argv[1]) for luser in lusers: luser.open("::1", 4000) + luser.set_option_negotiation_callback(option_callback) for test, description in dialogue: tlog("\nTesting %s..." % description) test_start = time.time()