-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_linux_ipv6.rb
158 lines (137 loc) · 4.59 KB
/
test_linux_ipv6.rb
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
# -*- encoding: binary -*-
require 'test/unit'
require 'tempfile'
require 'raindrops'
require 'socket'
require 'pp'
require "./test/ipv6_enabled"
$stderr.sync = $stdout.sync = true
class TestLinuxIPv6 < Test::Unit::TestCase
include Raindrops::Linux
TEST_ADDR = ENV["TEST_HOST6"] || "::1"
def test_tcp
s = TCPServer.new(TEST_ADDR, 0)
port = s.addr[1]
addr = "[#{TEST_ADDR}]:#{port}"
addrs = [ addr ]
stats = tcp_listener_stats(addrs)
assert_equal 1, stats.size
assert_equal 0, stats[addr].queued
assert_equal 0, stats[addr].active
@to_close << TCPSocket.new(TEST_ADDR, port)
stats = tcp_listener_stats(addrs)
assert_equal 1, stats.size
assert_equal 1, stats[addr].queued
assert_equal 0, stats[addr].active
@to_close << s.accept
stats = tcp_listener_stats(addrs)
assert_equal 1, stats.size
assert_equal 0, stats[addr].queued
assert_equal 1, stats[addr].active
end
def test_tcp_multi
s1 = TCPServer.new(TEST_ADDR, 0)
s2 = TCPServer.new(TEST_ADDR, 0)
port1, port2 = s1.addr[1], s2.addr[1]
addr1, addr2 = "[#{TEST_ADDR}]:#{port1}", "[#{TEST_ADDR}]:#{port2}"
addrs = [ addr1, addr2 ]
stats = tcp_listener_stats(addrs)
assert_equal 2, stats.size
assert_equal 0, stats[addr1].queued
assert_equal 0, stats[addr1].active
assert_equal 0, stats[addr2].queued
assert_equal 0, stats[addr2].active
@to_close << TCPSocket.new(TEST_ADDR, port1)
stats = tcp_listener_stats(addrs)
assert_equal 2, stats.size
assert_equal 1, stats[addr1].queued
assert_equal 0, stats[addr1].active
assert_equal 0, stats[addr2].queued
assert_equal 0, stats[addr2].active
sc1 = s1.accept
stats = tcp_listener_stats(addrs)
assert_equal 2, stats.size
assert_equal 0, stats[addr1].queued
assert_equal 1, stats[addr1].active
assert_equal 0, stats[addr2].queued
assert_equal 0, stats[addr2].active
@to_close << TCPSocket.new(TEST_ADDR, port2)
stats = tcp_listener_stats(addrs)
assert_equal 2, stats.size
assert_equal 0, stats[addr1].queued
assert_equal 1, stats[addr1].active
assert_equal 1, stats[addr2].queued
assert_equal 0, stats[addr2].active
@to_close << TCPSocket.new(TEST_ADDR, port2)
stats = tcp_listener_stats(addrs)
assert_equal 2, stats.size
assert_equal 0, stats[addr1].queued
assert_equal 1, stats[addr1].active
assert_equal 2, stats[addr2].queued
assert_equal 0, stats[addr2].active
@to_close << s2.accept
stats = tcp_listener_stats(addrs)
assert_equal 2, stats.size
assert_equal 0, stats[addr1].queued
assert_equal 1, stats[addr1].active
assert_equal 1, stats[addr2].queued
assert_equal 1, stats[addr2].active
sc1.close
stats = tcp_listener_stats(addrs)
assert_equal 0, stats[addr1].queued
assert_equal 0, stats[addr1].active
assert_equal 1, stats[addr2].queued
assert_equal 1, stats[addr2].active
end
def test_invalid_addresses
assert_raises(ArgumentError) { tcp_listener_stats(%w([1:::5)) }
assert_raises(ArgumentError) { tcp_listener_stats(%w([1:::]5)) }
end
# tries to overflow buffers
def test_tcp_stress_test
nr_proc = 32
nr_sock = 500
s = TCPServer.new(TEST_ADDR, 0)
port = s.addr[1]
addr = "[#{TEST_ADDR}]:#{port}"
addrs = [ addr ]
rda, wra = IO.pipe
rdb, wrb = IO.pipe
nr_proc.times do
fork do
rda.close
wrb.close
@to_close.concat((1..nr_sock).map { s.accept })
wra.syswrite('.')
wra.close
rdb.sysread(1) # wait for parent to nuke us
end
end
nr_proc.times do
fork do
rda.close
wrb.close
@to_close.concat((1..nr_sock).map { TCPSocket.new(TEST_ADDR, port) })
wra.syswrite('.')
wra.close
rdb.sysread(1) # wait for parent to nuke us
end
end
assert_equal('.' * (nr_proc * 2), rda.read(nr_proc * 2))
rda.close
stats = tcp_listener_stats(addrs)
expect = { addr => Raindrops::ListenStats[nr_sock * nr_proc, 0] }
assert_equal expect, stats
@to_close << TCPSocket.new(TEST_ADDR, port)
stats = tcp_listener_stats(addrs)
expect = { addr => Raindrops::ListenStats[nr_sock * nr_proc, 1] }
assert_equal expect, stats
if ENV["BENCHMARK"].to_i != 0
require 'benchmark'
puts(Benchmark.measure{1000.times { tcp_listener_stats(addrs) }})
end
wrb.syswrite('.' * (nr_proc * 2)) # broadcast a wakeup
statuses = Process.waitall
statuses.each { |(_,status)| assert status.success?, status.inspect }
end if ENV["STRESS"].to_i != 0
end if RUBY_PLATFORM =~ /linux/ && ipv6_enabled?