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
|
require_relative '../spec_helper'
require_relative '../fixtures/classes'
describe "Socket::IPSocket#recvfrom" do
before :each do
@server = TCPServer.new("127.0.0.1", 0)
@port = @server.addr[1]
@client = TCPSocket.new("127.0.0.1", @port)
end
after :each do
@server.close unless @server.closed?
@client.close unless @client.closed?
end
it "reads data from the connection" do
data = nil
t = Thread.new do
client = @server.accept
begin
data = client.recvfrom(6)
ensure
client.close
end
end
@client.send('hello', 0)
@client.shutdown rescue nil
# shutdown may raise Errno::ENOTCONN when sent data is pending.
t.join
data.first.should == 'hello'
end
it "reads up to len bytes" do
data = nil
t = Thread.new do
client = @server.accept
begin
data = client.recvfrom(3)
ensure
client.close
end
end
@client.send('hello', 0)
@client.shutdown rescue nil
t.join
data.first.should == 'hel'
end
it "returns an array with the data and connection info" do
data = nil
t = Thread.new do
client = @server.accept
data = client.recvfrom(3)
client.close
end
@client.send('hello', 0)
@client.shutdown rescue nil
t.join
data.size.should == 2
data.first.should == "hel"
# This does not apply to every platform, dependent on recvfrom(2)
# data.last.should == nil
end
end
describe 'Socket::IPSocket#recvfrom' do
SocketSpecs.each_ip_protocol do |family, ip_address, family_name|
before do
@server = UDPSocket.new(family)
@client = UDPSocket.new(family)
@server.bind(ip_address, 0)
@client.connect(ip_address, @server.connect_address.ip_port)
@hostname = Socket.getaddrinfo(ip_address, nil)[0][2]
end
after do
@client.close
@server.close
end
it 'returns an Array containing up to N bytes and address information' do
@client.write('hello')
port = @client.local_address.ip_port
ret = @server.recvfrom(2)
ret.should == ['he', [family_name, port, @hostname, ip_address]]
end
it 'allows specifying of flags when receiving data' do
@client.write('hello')
@server.recvfrom(2, Socket::MSG_PEEK)[0].should == 'he'
@server.recvfrom(2)[0].should == 'he'
end
describe 'using reverse lookups' do
before do
@server.do_not_reverse_lookup = false
@hostname = Socket.getaddrinfo(ip_address, nil, 0, 0, 0, 0, true)[0][2]
end
it 'includes the hostname in the address Array' do
@client.write('hello')
port = @client.local_address.ip_port
ret = @server.recvfrom(2)
ret.should == ['he', [family_name, port, @hostname, ip_address]]
end
end
end
end
|