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
|
# frozen_string_literal: true
begin
require "socket"
require "test/unit"
rescue LoadError
end
class TestSocket_UDPSocket < Test::Unit::TestCase
def test_open
assert_nothing_raised { UDPSocket.open {} }
assert_nothing_raised { UDPSocket.open(Socket::AF_INET) {} }
assert_nothing_raised { UDPSocket.open("AF_INET") {} }
assert_nothing_raised { UDPSocket.open(:AF_INET) {} }
end
def test_inspect
UDPSocket.open() {|sock|
assert_match(/AF_INET\b/, sock.inspect)
}
if Socket.const_defined?(:AF_INET6)
begin
UDPSocket.open(Socket::AF_INET6) {|sock|
assert_match(/AF_INET6\b/, sock.inspect)
}
rescue Errno::EAFNOSUPPORT
skip 'AF_INET6 not supported by kernel'
end
end
end
def test_connect
s = UDPSocket.new
host = Object.new
class << host; self end.send(:define_method, :to_str) {
s.close
"127.0.0.1"
}
assert_raise(IOError, "[ruby-dev:25045]") {
s.connect(host, 1)
}
end
def test_bind
s = UDPSocket.new
host = Object.new
class << host; self end.send(:define_method, :to_str) {
s.close
"127.0.0.1"
}
assert_raise(IOError, "[ruby-dev:25057]") {
s.bind(host, 2000)
}
ensure
s.close if s && !s.closed?
end
def test_bind_addrinuse
host = "127.0.0.1"
in_use = UDPSocket.new
in_use.bind(host, 0)
port = in_use.addr[1]
s = UDPSocket.new
e = assert_raise(Errno::EADDRINUSE) do
s.bind(host, port)
end
assert_match "bind(2) for \"#{host}\" port #{port}", e.message
ensure
in_use.close if in_use
s.close if s
end
def test_send_too_long
u = UDPSocket.new
e = assert_raise(Errno::EMSGSIZE) do
u.send "\0" * 100_000, 0, "127.0.0.1", 7 # echo
end
assert_match 'for "127.0.0.1" port 7', e.message
ensure
u.close if u
end
def test_bind_no_memory_leak
assert_no_memory_leak(["-rsocket"], <<-"end;", <<-"end;", rss: true)
s = UDPSocket.new
s.close
end;
100_000.times {begin s.bind("127.0.0.1", 1) rescue IOError; end}
end;
end
def test_connect_no_memory_leak
assert_no_memory_leak(["-rsocket"], <<-"end;", <<-"end;", rss: true)
s = UDPSocket.new
s.close
end;
100_000.times {begin s.connect("127.0.0.1", 1) rescue IOError; end}
end;
end
def test_send_no_memory_leak
assert_no_memory_leak(["-rsocket"], <<-"end;", <<-"end;", rss: true)
s = UDPSocket.new
s.close
end;
100_000.times {begin s.send("\0"*100, 0, "127.0.0.1", 1) rescue IOError; end}
end;
end
end if defined?(UDPSocket)
|
