98 lines
4.4 KiB
Ruby
98 lines
4.4 KiB
Ruby
|
# encoding: UTF-8
|
||
|
|
|
||
|
|
require_relative "test_helper"
|
||
|
|
require_relative "../lib/meow/telnet"
|
||
|
|
|
||
|
|
module MeowTelnetTest
|
||
|
|
Telnet = MeowClient::Telnet
|
||
|
|
|
||
|
|
def self.bytes(*values)
|
||
|
|
values.pack("C*")
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.collect(parser, chunks)
|
||
|
|
chunks.each_with_object({:data => +"".b, :replies => [], :events => []}) do |chunk, result|
|
||
|
|
parsed = parser.feed(chunk)
|
||
|
|
result[:data] << parsed.data
|
||
|
|
result[:replies].concat(parsed.replies)
|
||
|
|
result[:events].concat(parsed.events)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
MeowTest.test("Telnet preserves data and unescapes IAC") do
|
||
|
|
parser = Telnet::Parser.new
|
||
|
|
result = parser.feed("look".b + bytes(Telnet::IAC, Telnet::IAC) + "!".b)
|
||
|
|
MeowTest.assert_equal("look".b + bytes(Telnet::IAC) + "!".b, result.data)
|
||
|
|
MeowTest.assert_empty(result.replies)
|
||
|
|
MeowTest.assert_empty(result.events)
|
||
|
|
end
|
||
|
|
|
||
|
|
MeowTest.test("Telnet negotiation survives every input boundary") do
|
||
|
|
packet = bytes(Telnet::IAC, Telnet::WILL, Telnet::ECHO)
|
||
|
|
whole = collect(Telnet::Parser.new, [packet])
|
||
|
|
split = collect(Telnet::Parser.new, packet.bytes.map { |byte| bytes(byte) })
|
||
|
|
MeowTest.assert_equal(whole, split)
|
||
|
|
MeowTest.assert_equal([bytes(Telnet::IAC, Telnet::DO, Telnet::ECHO)], split[:replies])
|
||
|
|
MeowTest.assert_equal([[:remote_echo, true]], split[:events])
|
||
|
|
end
|
||
|
|
|
||
|
|
MeowTest.test("Telnet tracks echo transitions without duplicate replies") do
|
||
|
|
parser = Telnet::Parser.new
|
||
|
|
enabled = parser.feed(bytes(Telnet::IAC, Telnet::WILL, Telnet::ECHO))
|
||
|
|
duplicate = parser.feed(bytes(Telnet::IAC, Telnet::WILL, Telnet::ECHO))
|
||
|
|
disabled = parser.feed(bytes(Telnet::IAC, Telnet::WONT, Telnet::ECHO))
|
||
|
|
disabled_duplicate = parser.feed(bytes(Telnet::IAC, Telnet::WONT, Telnet::ECHO))
|
||
|
|
|
||
|
|
MeowTest.assert_equal([bytes(Telnet::IAC, Telnet::DO, Telnet::ECHO)], enabled.replies)
|
||
|
|
MeowTest.assert_empty(duplicate.replies)
|
||
|
|
MeowTest.assert_equal([bytes(Telnet::IAC, Telnet::DONT, Telnet::ECHO)], disabled.replies)
|
||
|
|
MeowTest.assert_equal([[:remote_echo, false]], disabled.events)
|
||
|
|
MeowTest.assert_empty(disabled_duplicate.replies)
|
||
|
|
MeowTest.refute(parser.remote_echo)
|
||
|
|
end
|
||
|
|
|
||
|
|
MeowTest.test("Telnet accepts supported options and rejects unknown options") do
|
||
|
|
parser = Telnet::Parser.new
|
||
|
|
remote = parser.feed(bytes(Telnet::IAC, Telnet::WILL, Telnet::SUPPRESS_GO_AHEAD))
|
||
|
|
remote_unknown = parser.feed(bytes(Telnet::IAC, Telnet::WILL, 42))
|
||
|
|
local = parser.feed(bytes(Telnet::IAC, Telnet::DO, Telnet::TERMINAL_TYPE))
|
||
|
|
local_unknown = parser.feed(bytes(Telnet::IAC, Telnet::DO, 43))
|
||
|
|
|
||
|
|
MeowTest.assert_equal([bytes(Telnet::IAC, Telnet::DO, Telnet::SUPPRESS_GO_AHEAD)], remote.replies)
|
||
|
|
MeowTest.assert_equal([bytes(Telnet::IAC, Telnet::DONT, 42)], remote_unknown.replies)
|
||
|
|
MeowTest.assert_equal([bytes(Telnet::IAC, Telnet::WILL, Telnet::TERMINAL_TYPE)], local.replies)
|
||
|
|
MeowTest.assert_equal([bytes(Telnet::IAC, Telnet::WONT, 43)], local_unknown.replies)
|
||
|
|
end
|
||
|
|
|
||
|
|
MeowTest.test("Telnet sends terminal type only after accepting the option") do
|
||
|
|
request = bytes(Telnet::IAC, Telnet::SB, Telnet::TERMINAL_TYPE, Telnet::TTYPE_SEND, Telnet::IAC, Telnet::SE)
|
||
|
|
parser = Telnet::Parser.new
|
||
|
|
before = parser.feed(request)
|
||
|
|
parser.feed(bytes(Telnet::IAC, Telnet::DO, Telnet::TERMINAL_TYPE))
|
||
|
|
after = collect(parser, request.bytes.map { |byte| bytes(byte) })
|
||
|
|
expected = bytes(Telnet::IAC, Telnet::SB, Telnet::TERMINAL_TYPE, Telnet::TTYPE_IS) + "MEOW".b + bytes(Telnet::IAC, Telnet::SE)
|
||
|
|
|
||
|
|
MeowTest.assert_empty(before.replies)
|
||
|
|
MeowTest.assert_equal([[:subnegotiation, Telnet::TERMINAL_TYPE]], before.events)
|
||
|
|
MeowTest.assert_equal([expected], after[:replies])
|
||
|
|
MeowTest.assert_equal([[:terminal_type, "MEOW"]], after[:events])
|
||
|
|
end
|
||
|
|
|
||
|
|
MeowTest.test("Telnet handles escaped IAC inside subnegotiation") do
|
||
|
|
parser = Telnet::Parser.new
|
||
|
|
packet = bytes(Telnet::IAC, Telnet::SB, 77, 1, Telnet::IAC, Telnet::IAC, 2, Telnet::IAC, Telnet::SE)
|
||
|
|
result = collect(parser, packet.bytes.map { |byte| bytes(byte) })
|
||
|
|
MeowTest.assert_empty(result[:data])
|
||
|
|
MeowTest.assert_empty(result[:replies])
|
||
|
|
MeowTest.assert_equal([[:subnegotiation, 77]], result[:events])
|
||
|
|
end
|
||
|
|
|
||
|
|
MeowTest.test("Telnet ignores unknown commands and resumes ordinary data") do
|
||
|
|
parser = Telnet::Parser.new
|
||
|
|
result = parser.feed("a".b + bytes(Telnet::IAC, 241) + "b".b)
|
||
|
|
MeowTest.assert_equal("ab".b, result.data)
|
||
|
|
MeowTest.assert_empty(result.replies)
|
||
|
|
MeowTest.assert_empty(result.events)
|
||
|
|
end
|
||
|
|
end
|