45 lines
882 B
Ruby
45 lines
882 B
Ruby
|
#!/usr/bin/env ruby
|
||
|
|
|
||
|
|
# Simulate a server which has a disc of the wrong size attached: send
|
||
|
|
# a valid NBD hello with a random size, then check that we have see an
|
||
|
|
# EOF on read.
|
||
|
|
|
||
|
|
addr, port = *ARGV
|
||
|
|
|
||
|
|
require 'socket'
|
||
|
|
require 'timeout'
|
||
|
|
require 'flexnbd/constants'
|
||
|
|
|
||
|
|
sock = TCPServer.new( addr, port )
|
||
|
|
client_sock = nil
|
||
|
|
|
||
|
|
begin
|
||
|
|
Timeout.timeout(2) do
|
||
|
|
client_sock = sock.accept
|
||
|
|
end
|
||
|
|
rescue Timeout::Error
|
||
|
|
$stderr.puts "Timed out waiting for a connection"
|
||
|
|
exit 1
|
||
|
|
end
|
||
|
|
|
||
|
|
t = Thread.new do
|
||
|
|
begin
|
||
|
|
Timeout.timeout(FlexNBD::MS_RETRY_DELAY_SECS + 1) do
|
||
|
|
client_sock2 = sock.accept
|
||
|
|
end
|
||
|
|
rescue Timeout::Error
|
||
|
|
$stderr.puts "Timed out waiting for a reconnection"
|
||
|
|
exit 1
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
client_sock.write( "NBDMAGIC" )
|
||
|
|
client_sock.write( "\x00\x00\x42\x02\x81\x86\x12\x53" )
|
||
|
|
8.times do client_sock.write rand(256).chr end
|
||
|
|
client_sock.write( "\x00" * 128 )
|
||
|
|
|
||
|
|
|
||
|
|
t.join
|
||
|
|
|
||
|
|
exit 0
|