1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
|
require_relative 'common'
require_relative 'mitm_server'
require 'fileutils'
require 'tmpdir'
require 'net/ssh'
require 'timeout'
# see Vagrantfile,playbook for env.
# we're running as net_ssh_1 user password foo
# and usually connecting to net_ssh_2 user password foo2pwd
class TestHMacEtm < NetSSHTest
include IntegrationTestHelpers
variants = {
etm256: "hmac-sha2-256-etm@openssh.com",
etm512: "hmac-sha2-512-etm@openssh.com"
}
def config_with_macs(macs)
config_lines = File.read('/etc/ssh/sshd_config').split("\n")
config_lines = config_lines.map do |line|
if line =~ /^MACs/
"##{line}"
else
line
end
end
config_lines.push("MACs #{macs}")
end
variants.each do |key,variant|
define_method "test_with_only_hmac_etm#{key}" do
start_sshd_7_or_later(config: config_with_macs(variant)) do |_pid, port|
Timeout.timeout(4) do
begin
# We have our own sshd, give it a chance to come up before
# listening.
ret = Net::SSH.start(
"localhost",
"net_ssh_1",
password: 'foopwd',
port: port,
hmac: [variant]
) do |ssh|
assert_equal ssh.transport.algorithms.hmac_client, variant
assert_equal ssh.transport.algorithms.hmac_server, variant
ssh.exec! "echo 'foo123'"
end
assert_equal "foo123\n", ret
rescue SocketError, Errno::ECONNREFUSED, Errno::EHOSTUNREACH
sleep 0.25
retry
end
end
end
end
end
variants.each do |key, variant|
define_method "test_hmac_through_proxy_#{key}" do
mitm = MitmServer.new('localhost', 22)
mitm.local_read_size = 2 * 1024
mitm.target_read_size = 19
mitm.run
# host, port = 'localhost', 22
host = mitm.host
port = mitm.port
hmac_options = { hmac: variant }
ssh = Net::SSH.start(host, "net_ssh_1", hmac_options.merge(port: port, password: 'foopwd'))
(1...20).each do |i|
ret = ssh.exec!("echo \"#{'f' * i}\"")
assert_includes ret, 'f' * i
end
ssh.close
end
end
end
|