1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/test/socket/test_socket.rb
Naohisa Goto f8f2885bd0 Revert "test/socket/test_socket.rb: skip on Solaris"
This reverts commit 27fb9d272d.

The test failure on Solaris 10 is due to incomplete IPv6 configuration
on the CI server, that have already been fixed.

Reference for the fix: https://centrify.force.com/support/Article/KB-1179-X11-Forwarding-fails-with-Centrify-OpenSSH-5-0-Solaris/
2021-11-30 00:07:30 +09:00

765 lines
22 KiB
Ruby

# frozen_string_literal: true
begin
require "socket"
require "tmpdir"
require "fcntl"
require "etc"
require "test/unit"
rescue LoadError
end
class TestSocket < Test::Unit::TestCase
def test_socket_new
begin
s = Socket.new(:INET, :STREAM)
assert_kind_of(Socket, s)
ensure
s.close
end
end
def test_socket_new_cloexec
return unless defined? Fcntl::FD_CLOEXEC
begin
s = Socket.new(:INET, :STREAM)
assert(s.close_on_exec?)
ensure
s.close
end
end
def test_unpack_sockaddr
sockaddr_in = Socket.sockaddr_in(80, "")
assert_raise(ArgumentError) { Socket.unpack_sockaddr_un(sockaddr_in) }
sockaddr_un = Socket.sockaddr_un("/testdir/s")
assert_raise(ArgumentError) { Socket.unpack_sockaddr_in(sockaddr_un) }
assert_raise(ArgumentError) { Socket.unpack_sockaddr_in("") }
assert_raise(ArgumentError) { Socket.unpack_sockaddr_un("") }
end if Socket.respond_to?(:sockaddr_un)
def test_sysaccept
serv = Socket.new(Socket::AF_INET, Socket::SOCK_STREAM, 0)
serv.bind(Socket.sockaddr_in(0, "127.0.0.1"))
serv.listen 5
c = Socket.new(Socket::AF_INET, Socket::SOCK_STREAM, 0)
c.connect(serv.getsockname)
fd, peeraddr = serv.sysaccept
assert_equal(c.getsockname, peeraddr.to_sockaddr)
ensure
serv.close if serv
c.close if c
IO.for_fd(fd).close if fd
end
def test_initialize
Socket.open(Socket::AF_INET, Socket::SOCK_STREAM, 0) {|s|
s.bind(Socket.sockaddr_in(0, "127.0.0.1"))
addr = s.getsockname
assert_nothing_raised { Socket.unpack_sockaddr_in(addr) }
assert_raise(ArgumentError, NoMethodError) { Socket.unpack_sockaddr_un(addr) }
}
Socket.open("AF_INET", "SOCK_STREAM", 0) {|s|
s.bind(Socket.sockaddr_in(0, "127.0.0.1"))
addr = s.getsockname
assert_nothing_raised { Socket.unpack_sockaddr_in(addr) }
assert_raise(ArgumentError, NoMethodError) { Socket.unpack_sockaddr_un(addr) }
}
Socket.open(:AF_INET, :SOCK_STREAM, 0) {|s|
s.bind(Socket.sockaddr_in(0, "127.0.0.1"))
addr = s.getsockname
assert_nothing_raised { Socket.unpack_sockaddr_in(addr) }
assert_raise(ArgumentError, NoMethodError) { Socket.unpack_sockaddr_un(addr) }
}
end
def test_bind
Socket.open(Socket::AF_INET, Socket::SOCK_STREAM, 0) {|bound|
bound.bind(Socket.sockaddr_in(0, "127.0.0.1"))
addr = bound.getsockname
port, = Socket.unpack_sockaddr_in(addr)
Socket.open(Socket::AF_INET, Socket::SOCK_STREAM, 0) {|s|
e = assert_raise(Errno::EADDRINUSE) do
s.bind(Socket.sockaddr_in(port, "127.0.0.1"))
end
assert_match "bind(2) for 127.0.0.1:#{port}", e.message
}
}
end
def test_getaddrinfo
# This should not send a DNS query because AF_UNIX.
assert_raise(SocketError) { Socket.getaddrinfo("www.kame.net", 80, "AF_UNIX") }
end
def test_getaddrinfo_raises_no_errors_on_port_argument_of_0 # [ruby-core:29427]
assert_nothing_raised('[ruby-core:29427]'){ Socket.getaddrinfo('localhost', 0, Socket::AF_INET, Socket::SOCK_STREAM, nil, Socket::AI_CANONNAME) }
assert_nothing_raised('[ruby-core:29427]'){ Socket.getaddrinfo('localhost', '0', Socket::AF_INET, Socket::SOCK_STREAM, nil, Socket::AI_CANONNAME) }
assert_nothing_raised('[ruby-core:29427]'){ Socket.getaddrinfo('localhost', '00', Socket::AF_INET, Socket::SOCK_STREAM, nil, Socket::AI_CANONNAME) }
assert_raise(SocketError, '[ruby-core:29427]'){ Socket.getaddrinfo(nil, nil, Socket::AF_INET, Socket::SOCK_STREAM, nil, Socket::AI_CANONNAME) }
assert_nothing_raised('[ruby-core:29427]'){ TCPServer.open('localhost', 0) {} }
end
def test_getnameinfo
assert_raise(SocketError) { Socket.getnameinfo(["AF_UNIX", 80, "0.0.0.0"]) }
assert_raise(ArgumentError) {Socket.getnameinfo(["AF_INET", "http\0", "example.net"])}
assert_raise(ArgumentError) {Socket.getnameinfo(["AF_INET", "http", "example.net\0"])}
end
def test_ip_address_list
begin
list = Socket.ip_address_list
rescue NotImplementedError
return
end
list.each {|ai|
assert_instance_of(Addrinfo, ai)
assert(ai.ip?)
}
end
def test_ip_address_list_include_localhost
begin
list = Socket.ip_address_list
rescue NotImplementedError
return
end
assert_includes list.map(&:ip_address), Addrinfo.tcp("localhost", 0).ip_address
end
def test_tcp
TCPServer.open(0) {|serv|
addr = serv.connect_address
addr.connect {|s1|
s2 = serv.accept
begin
assert_equal(s2.remote_address.ip_unpack, s1.local_address.ip_unpack)
ensure
s2.close
end
}
}
end
def test_tcp_cloexec
return unless defined? Fcntl::FD_CLOEXEC
TCPServer.open(0) {|serv|
addr = serv.connect_address
addr.connect {|s1|
s2 = serv.accept
begin
assert(s2.close_on_exec?)
ensure
s2.close
end
}
}
end
def random_port
# IANA suggests dynamic port for 49152 to 65535
# http://www.iana.org/assignments/port-numbers
case RUBY_PLATFORM
when /mingw|mswin/
rand(50000..65535)
else
rand(49152..65535)
end
end
def errors_addrinuse
errs = [Errno::EADDRINUSE]
# MinGW fails with "Errno::EACCES: Permission denied - bind(2) for 0.0.0.0:49721"
errs << Errno::EACCES if /mingw/ =~ RUBY_PLATFORM
errs
end
def test_tcp_server_sockets
port = random_port
begin
sockets = Socket.tcp_server_sockets(port)
rescue *errors_addrinuse
return # not test failure
end
begin
sockets.each {|s|
assert_equal(port, s.local_address.ip_port)
}
ensure
sockets.each {|s|
s.close
}
end
end
def test_tcp_server_sockets_port0
sockets = Socket.tcp_server_sockets(0)
ports = sockets.map {|s| s.local_address.ip_port }
the_port = ports.first
ports.each {|port|
assert_equal(the_port, port)
}
ensure
if sockets
sockets.each {|s|
s.close
}
end
end
if defined? UNIXSocket
def test_unix
Dir.mktmpdir {|tmpdir|
path = "#{tmpdir}/sock"
UNIXServer.open(path) {|serv|
Socket.unix(path) {|s1|
s2 = serv.accept
begin
s2raddr = s2.remote_address
s1laddr = s1.local_address
assert(s2raddr.to_sockaddr.empty? ||
s1laddr.to_sockaddr.empty? ||
s2raddr.unix_path == s1laddr.unix_path)
assert(s2.close_on_exec?)
ensure
s2.close
end
}
}
}
end
def test_unix_server_socket
Dir.mktmpdir {|tmpdir|
path = "#{tmpdir}/sock"
2.times {
serv = Socket.unix_server_socket(path)
begin
assert_kind_of(Socket, serv)
assert(File.socket?(path))
assert_equal(path, serv.local_address.unix_path)
ensure
serv.close
end
}
}
end
def test_accept_loop_with_unix
Dir.mktmpdir {|tmpdir|
tcp_servers = []
clients = []
accepted = []
begin
tcp_servers = Socket.tcp_server_sockets(0)
unix_server = Socket.unix_server_socket("#{tmpdir}/sock")
tcp_servers.each {|s|
addr = s.connect_address
begin
clients << addr.connect
rescue
# allow failure if the address is IPv6
raise unless addr.ipv6?
end
}
addr = unix_server.connect_address
assert_nothing_raised("connect to #{addr.inspect}") {
clients << addr.connect
}
Socket.accept_loop(tcp_servers, unix_server) {|s|
accepted << s
break if clients.length == accepted.length
}
assert_equal(clients.length, accepted.length)
ensure
tcp_servers.each {|s| s.close if !s.closed? }
unix_server.close if unix_server && !unix_server.closed?
clients.each {|s| s.close if !s.closed? }
accepted.each {|s| s.close if !s.closed? }
end
}
end
end
def test_accept_loop
servers = []
begin
servers = Socket.tcp_server_sockets(0)
port = servers[0].local_address.ip_port
Socket.tcp("localhost", port) {|s1|
Socket.accept_loop(servers) {|s2, client_ai|
begin
assert_equal(s1.local_address.ip_unpack, client_ai.ip_unpack)
ensure
s2.close
end
break
}
}
ensure
servers.each {|s| s.close if !s.closed? }
end
end
def test_accept_loop_multi_port
servers = []
begin
servers = Socket.tcp_server_sockets(0)
port = servers[0].local_address.ip_port
servers2 = Socket.tcp_server_sockets(0)
servers.concat servers2
port2 = servers2[0].local_address.ip_port
Socket.tcp("localhost", port) {|s1|
Socket.accept_loop(servers) {|s2, client_ai|
begin
assert_equal(s1.local_address.ip_unpack, client_ai.ip_unpack)
ensure
s2.close
end
break
}
}
Socket.tcp("localhost", port2) {|s1|
Socket.accept_loop(servers) {|s2, client_ai|
begin
assert_equal(s1.local_address.ip_unpack, client_ai.ip_unpack)
ensure
s2.close
end
break
}
}
ensure
servers.each {|s| s.close if !s.closed? }
end
end
def test_udp_server
begin
ifaddrs = Socket.getifaddrs
rescue NotImplementedError
skip "Socket.getifaddrs not implemented"
end
ifconfig = nil
Socket.udp_server_sockets(0) {|sockets|
famlies = {}
sockets.each {|s| famlies[s.local_address.afamily] = s }
nd6 = {}
ifaddrs.reject! {|ifa|
ai = ifa.addr
next true unless ai
s = famlies[ai.afamily]
next true unless s
next true if ai.ipv6_linklocal? # IPv6 link-local address is too troublesome in this test.
case RUBY_PLATFORM
when /linux/
if ai.ip_address.include?('%') and
(Etc.uname[:release][/[0-9.]+/].split('.').map(&:to_i) <=> [2,6,18]) <= 0
# Cent OS 5.6 (2.6.18-238.19.1.el5xen) doesn't correctly work
# sendmsg with pktinfo for link-local ipv6 addresses
next true
end
when /freebsd/
if ifa.addr.ipv6_linklocal?
# FreeBSD 9.0 with default setting (ipv6_activate_all_interfaces
# is not YES) sets IFDISABLED to interfaces which don't have
# global IPv6 address.
# Link-local IPv6 addresses on those interfaces don't work.
ulSIOCGIFINFO_IN6 = 3225971052
ulND6_IFF_IFDISABLED = 8
in6_ondireq = ifa.name
s.ioctl(ulSIOCGIFINFO_IN6, in6_ondireq)
flag = in6_ondireq.unpack('A16L6').last
next true if flag & ulND6_IFF_IFDISABLED != 0
nd6[ai] = flag
end
when /darwin/
if !ai.ipv6?
elsif ai.ipv6_unique_local? && /darwin1[01]\./ =~ RUBY_PLATFORM
next true # iCloud addresses do not work, see Bug #6692
elsif ifr_name = ai.ip_address[/%(.*)/, 1]
# Mac OS X may sets IFDISABLED as FreeBSD does
ulSIOCGIFFLAGS = 3223349521
ulSIOCGIFINFO_IN6 = 3224398156
ulIFF_POINTOPOINT = 0x10
ulND6_IFF_IFDISABLED = 8
in6_ondireq = ifr_name
s.ioctl(ulSIOCGIFINFO_IN6, in6_ondireq)
flag = in6_ondireq.unpack('A16L6').last
next true if (flag & ulND6_IFF_IFDISABLED) != 0
nd6[ai] = flag
in6_ifreq = [ifr_name,ai.to_sockaddr].pack('a16A*')
s.ioctl(ulSIOCGIFFLAGS, in6_ifreq)
next true if in6_ifreq.unpack('A16L1').last & ulIFF_POINTOPOINT != 0
end
ifconfig ||= `/sbin/ifconfig`
next true if ifconfig.scan(/^(\w+):(.*(?:\n\t.*)*)/).find do |_ifname, value|
value.include?(ai.ip_address) && value.include?('POINTOPOINT')
end
end
false
}
skipped = false
begin
port = sockets.first.local_address.ip_port
ping_p = false
th = Thread.new {
Socket.udp_server_loop_on(sockets) {|msg, msg_src|
break if msg == "exit"
rmsg = Marshal.dump([msg, msg_src.remote_address, msg_src.local_address])
ping_p = true
msg_src.reply rmsg
}
}
ifaddrs.each {|ifa|
ai = ifa.addr
Addrinfo.udp(ai.ip_address, port).connect {|s|
ping_p = false
msg1 = "<<<#{ai.inspect}>>>"
s.sendmsg msg1
unless IO.select([s], nil, nil, 10)
nd6options = nd6.key?(ai) ? "nd6=%x " % nd6[ai] : ''
raise "no response from #{ifa.inspect} #{nd6options}ping=#{ping_p}"
end
msg2, addr = s.recvmsg
msg2, _, _ = Marshal.load(msg2)
assert_equal(msg1, msg2)
assert_equal(ai.ip_address, addr.ip_address)
}
}
rescue NotImplementedError, Errno::ENOSYS
skipped = true
skip "need sendmsg and recvmsg: #{$!}"
rescue RuntimeError
skipped = true
skip "UDP server is no response: #{$!}"
ensure
if th
if skipped
Thread.kill th unless th.join(10)
else
Addrinfo.udp("127.0.0.1", port).connect {|s| s.sendmsg "exit" }
unless th.join(10)
Thread.kill th
th.join(10)
raise "thread killed"
end
end
end
end
}
end
def test_linger
opt = Socket::Option.linger(true, 0)
assert_equal([true, 0], opt.linger)
Addrinfo.tcp("127.0.0.1", 0).listen {|serv|
serv.local_address.connect {|s1|
s2, _ = serv.accept
begin
s1.setsockopt(opt)
s1.close
assert_raise(Errno::ECONNRESET) { s2.read }
ensure
s2.close
end
}
}
end
def timestamp_retry_rw(s1, s2, t1, type)
IO.pipe do |r,w|
# UDP may not be reliable, keep sending until recvmsg returns:
th = Thread.new do
n = 0
begin
s2.send("a", 0, s1.local_address)
n += 1
end while IO.select([r], nil, nil, 0.1).nil?
n
end
timeout = (defined?(RubyVM::JIT) && RubyVM::JIT.enabled? ? 120 : 30) # for --jit-wait
assert_equal([[s1],[],[]], IO.select([s1], nil, nil, timeout))
msg, _, _, stamp = s1.recvmsg
assert_equal("a", msg)
assert(stamp.cmsg_is?(:SOCKET, type))
w.close # stop th
n = th.value
n > 1 and
warn "UDP packet loss for #{type} over loopback, #{n} tries needed"
t2 = Time.now.strftime("%Y-%m-%d")
pat = Regexp.union([t1, t2].uniq)
assert_match(pat, stamp.inspect)
t = stamp.timestamp
assert_match(pat, t.strftime("%Y-%m-%d"))
stamp
end
end
def test_timestamp
return if /linux|freebsd|netbsd|openbsd|solaris|darwin/ !~ RUBY_PLATFORM
return if !defined?(Socket::AncillaryData) || !defined?(Socket::SO_TIMESTAMP)
t1 = Time.now.strftime("%Y-%m-%d")
stamp = nil
Addrinfo.udp("127.0.0.1", 0).bind {|s1|
Addrinfo.udp("127.0.0.1", 0).bind {|s2|
s1.setsockopt(:SOCKET, :TIMESTAMP, true)
stamp = timestamp_retry_rw(s1, s2, t1, :TIMESTAMP)
}
}
t = stamp.timestamp
pat = /\.#{"%06d" % t.usec}/
assert_match(pat, stamp.inspect)
end
def test_timestampns
return if /linux/ !~ RUBY_PLATFORM || !defined?(Socket::SO_TIMESTAMPNS)
t1 = Time.now.strftime("%Y-%m-%d")
stamp = nil
Addrinfo.udp("127.0.0.1", 0).bind {|s1|
Addrinfo.udp("127.0.0.1", 0).bind {|s2|
begin
s1.setsockopt(:SOCKET, :TIMESTAMPNS, true)
rescue Errno::ENOPROTOOPT
# SO_TIMESTAMPNS is available since Linux 2.6.22
return
end
stamp = timestamp_retry_rw(s1, s2, t1, :TIMESTAMPNS)
}
}
t = stamp.timestamp
pat = /\.#{"%09d" % t.nsec}/
assert_match(pat, stamp.inspect)
end
def test_bintime
return if /freebsd/ !~ RUBY_PLATFORM
t1 = Time.now.strftime("%Y-%m-%d")
stamp = nil
Addrinfo.udp("127.0.0.1", 0).bind {|s1|
Addrinfo.udp("127.0.0.1", 0).bind {|s2|
s1.setsockopt(:SOCKET, :BINTIME, true)
s2.send "a", 0, s1.local_address
msg, _, _, stamp = s1.recvmsg
assert_equal("a", msg)
assert(stamp.cmsg_is?(:SOCKET, :BINTIME))
}
}
t2 = Time.now.strftime("%Y-%m-%d")
pat = Regexp.union([t1, t2].uniq)
assert_match(pat, stamp.inspect)
t = stamp.timestamp
assert_match(pat, t.strftime("%Y-%m-%d"))
assert_equal(stamp.data[-8,8].unpack("Q")[0], t.subsec * 2**64)
end
def test_closed_read
require 'timeout'
require 'socket'
bug4390 = '[ruby-core:35203]'
server = TCPServer.new("localhost", 0)
serv_thread = Thread.new {server.accept}
begin sleep(0.1) end until serv_thread.stop?
sock = TCPSocket.new("localhost", server.addr[1])
client_thread = Thread.new do
assert_raise(IOError, bug4390) {
sock.readline
}
end
begin sleep(0.1) end until client_thread.stop?
Timeout.timeout(1) do
sock.close
sock = nil
client_thread.join
end
ensure
serv_thread.value.close
server.close
end
def test_connect_timeout
host = "127.0.0.1"
server = TCPServer.new(host, 0)
port = server.addr[1]
serv_thread = Thread.new {server.accept}
sock = Socket.tcp(host, port, :connect_timeout => 30)
accepted = serv_thread.value
assert_kind_of TCPSocket, accepted
assert_equal sock, IO.select(nil, [ sock ])[1][0], "not writable"
sock.close
# some platforms may not timeout when the listener queue overflows,
# but we know Linux does with the default listen backlog of SOMAXCONN for
# TCPServer.
assert_raise(Errno::ETIMEDOUT) do
(Socket::SOMAXCONN*2).times do |i|
sock = Socket.tcp(host, port, :connect_timeout => 0)
assert_equal sock, IO.select(nil, [ sock ])[1][0],
"not writable (#{i})"
sock.close
end
end if RUBY_PLATFORM =~ /linux/
ensure
server.close
accepted.close if accepted
sock.close if sock && ! sock.closed?
end
def test_getifaddrs
begin
list = Socket.getifaddrs
rescue NotImplementedError
return
end
list.each {|ifaddr|
assert_instance_of(Socket::Ifaddr, ifaddr)
}
end
def test_connect_in_rescue
serv = Addrinfo.tcp(nil, 0).listen
addr = serv.connect_address
begin
raise "dummy error"
rescue
s = addr.connect
assert(!s.closed?)
end
ensure
serv.close if serv && !serv.closed?
s.close if s && !s.closed?
end
def test_bind_in_rescue
begin
raise "dummy error"
rescue
s = Addrinfo.tcp(nil, 0).bind
assert(!s.closed?)
end
ensure
s.close if s && !s.closed?
end
def test_listen_in_rescue
begin
raise "dummy error"
rescue
s = Addrinfo.tcp(nil, 0).listen
assert(!s.closed?)
end
ensure
s.close if s && !s.closed?
end
def test_udp_server_sockets_in_rescue
begin
raise "dummy error"
rescue
ss = Socket.udp_server_sockets(0)
ss.each {|s|
assert(!s.closed?)
}
end
ensure
if ss
ss.each {|s|
s.close if !s.closed?
}
end
end
def test_tcp_server_sockets_in_rescue
begin
raise "dummy error"
rescue
ss = Socket.tcp_server_sockets(0)
ss.each {|s|
assert(!s.closed?)
}
end
ensure
if ss
ss.each {|s|
s.close if !s.closed?
}
end
end
def test_recvmsg_udp_no_arg
n = 4097
s1 = Addrinfo.udp("127.0.0.1", 0).bind
s2 = s1.connect_address.connect
s2.send("a" * n, 0)
ret = s1.recvmsg
assert_equal n, ret[0].bytesize, '[ruby-core:71517] [Bug #11701]'
s2.send("a" * n, 0)
IO.select([s1])
ret = s1.recvmsg_nonblock
assert_equal n, ret[0].bytesize, 'non-blocking should also grow'
ensure
s1.close
s2.close
end
def test_udp_read_truncation
s1 = Addrinfo.udp("127.0.0.1", 0).bind
s2 = s1.connect_address.connect
s2.send("a" * 100, 0)
ret = s1.read(10)
assert_equal "a" * 10, ret
s2.send("b" * 100, 0)
ret = s1.read(10)
assert_equal "b" * 10, ret
ensure
s1.close
s2.close
end
def test_udp_recv_truncation
s1 = Addrinfo.udp("127.0.0.1", 0).bind
s2 = s1.connect_address.connect
s2.send("a" * 100, 0)
ret = s1.recv(10, Socket::MSG_PEEK)
assert_equal "a" * 10, ret
ret = s1.recv(10, 0)
assert_equal "a" * 10, ret
s2.send("b" * 100, 0)
ret = s1.recv(10, 0)
assert_equal "b" * 10, ret
ensure
s1.close
s2.close
end
def test_udp_recvmsg_truncation
s1 = Addrinfo.udp("127.0.0.1", 0).bind
s2 = s1.connect_address.connect
s2.send("a" * 100, 0)
ret, addr, rflags = s1.recvmsg(10, Socket::MSG_PEEK)
assert_equal "a" * 10, ret
# AIX does not set MSG_TRUNC for a message partially read with MSG_PEEK.
assert_equal Socket::MSG_TRUNC, rflags & Socket::MSG_TRUNC if !rflags.nil? && /aix/ !~ RUBY_PLATFORM
ret, addr, rflags = s1.recvmsg(10, 0)
assert_equal "a" * 10, ret
assert_equal Socket::MSG_TRUNC, rflags & Socket::MSG_TRUNC if !rflags.nil?
s2.send("b" * 100, 0)
ret, addr, rflags = s1.recvmsg(10, 0)
assert_equal "b" * 10, ret
assert_equal Socket::MSG_TRUNC, rflags & Socket::MSG_TRUNC if !rflags.nil?
addr
ensure
s1.close
s2.close
end
end if defined?(Socket)