blob: a59eb609afa59327e880dce42893e21e08739485 (
plain)
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 'net/ssh/buffer'
require 'net/ssh'
require 'timeout'
require 'tempfile'
require 'net/ssh/proxy/command'
require 'net/ssh/proxy/http'
require 'net/ssh/proxy/https'
require 'webrick'
require 'webrick/httpproxy'
require 'webrick/https'
class TestHTTPProxy < NetSSHTest
include IntegrationTestHelpers
def localhost
'localhost'
end
def user
'net_ssh_1'
end
def ssh_start_params(options)
[localhost, user, { keys: @key_id_rsa }.merge(options)]
end
def setup_ssh_env(&block)
tmpdir do |dir|
@key_id_rsa = "#{dir}/id_rsa"
sh "rm -rf #{@key_id_rsa} #{@key_id_rsa}.pub"
sh "ssh-keygen -q -f #{@key_id_rsa} -t rsa -N ''"
set_authorized_key(user, "#{@key_id_rsa}.pub")
yield
end
end
def with_http_proxy_server(&block)
proxy = WEBrick::HTTPProxyServer.new Port: 0
Thread.start { proxy.start }
begin
yield(proxy)
ensure
proxy.shutdown
end
end
def test_smoke
setup_ssh_env do
with_http_proxy_server do |http_proxy|
msg = 'echo123'
ret = Net::SSH.start(*ssh_start_params(proxy: Net::SSH::Proxy::HTTP.new('localhost', http_proxy.config[:Port]))) do |ssh|
ssh.exec! "echo \"$USER:#{msg}\""
end
assert_equal "net_ssh_1:#{msg}\n", ret
end
end
end
def with_http_tls_proxy_server(&block)
cert_name = [%w[CN localhost]]
proxy = WEBrick::HTTPProxyServer.new Port: 0, SSLEnable: true, SSLCertName: cert_name
Thread.start { proxy.start }
begin
yield(proxy)
ensure
proxy.shutdown
end
end
def test_smoke_tls
setup_ssh_env do
with_http_tls_proxy_server do |http_proxy|
msg = 'echo123'
ret = Net::SSH.start(*ssh_start_params(proxy: Net::SSH::Proxy::HTTPS.new('localhost', http_proxy.config[:Port]))) do |ssh|
ssh.exec! "echo \"$USER:#{msg}\""
end
assert_equal "net_ssh_1:#{msg}\n", ret
end
end
end
end
|