bytemark/flexnbd-c
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Factor common code out of fake destinations

Browse Source
This commit is contained in:
Alex Young
2012-06-28 11:34:36 +01:00
parent 192471ee82
commit 9b717d6391
6 changed files with 85 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
tests/fakes/dest/close_after_hello.rb
View File

@@ -6,40 +6,18 @@
# command-line has gone away, and can't feed an error back to the
# user, we have to keep trying.
require 'flexnbd/fake_dest'
include FlexNBD::FakeDest
addr, port = *ARGV
require 'socket'
require 'timeout'
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
client_sock.write( "NBDMAGIC" )
client_sock.write( "\x00\x00\x42\x02\x81\x86\x12\x53" )
client_sock.write( "\x00\x00\x00\x00\x00\x00\x10\x00" )
client_sock.write( "\x00" * 128 )
sock = serve( addr, port )
client_sock = accept( sock, "Timed out waiting for a connection" )
write_hello( client_sock )
client_sock.close
new_sock = nil
begin
Timeout.timeout(60) do
new_sock = sock.accept
end
rescue Timeout::Error
$stderr.puts "Timed out waiting for a reconnection"
exit 1
end
new_sock = accept( sock, "Timed out waiting for a reconnection" )
new_sock.close
sock.close

19
tests/fakes/dest/hang_after_connect.rb
View File

@@ -7,24 +7,13 @@
# This allows the test runner to check that the command-line sees the
# right error message after the timeout time.
require 'socket'
require 'timeout'
require 'flexnbd/constants'
require 'flexnbd/fake_dest'
include FlexNBD::FakeDest
addr, port = *ARGV
serve_sock = TCPServer.new( addr, port )
client_sock = nil
begin
# A failure here means a more serious issue with flexnbd
Timeout.timeout( 2 ) do
client_sock = serve_sock.accept
end
rescue Timeout::Error
$stderr.puts "Client didn't make connection"
exit 1
end
serve_sock = serve( addr, port )
client_sock = accept( serve_sock, "Client didn't make a connection" )
# Sleep for one second past the timeout (a bit of slop in case ruby
# doesn't launch things quickly)

43
tests/fakes/dest/hello_wrong_magic.rb
View File

@@ -3,43 +3,22 @@
# Simulate a destination which sends the wrong magic.
# We expect the sender to disconnect and reconnect.
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
require 'flexnbd/fake_dest'
include FlexNBD::FakeDest
sock = serve( *ARGV )
client_sock = accept( sock, "Timed out waiting for a connection" )
# Launch a second thread so that we can spot the reconnection attempt
# as soon as it happens, or alternatively die a flaming death on
# timeout.
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
client_sock2 = accept( sock, "Timed out waiting for a reconnection",
FlexNBD::MS_RETRY_DELAY_SECS + 1 )
client_sock2.close
end
client_sock.write( "NBDMAGIC" )
# We're off in the last byte, should be \x53
client_sock.write( "\x00\x00\x42\x02\x81\x86\x12\x52" )
# 4096 is the right size; this is defined in nbd_scenarios
client_sock.write( "\x00\x00\x00\x00\x00\x00\x10\x00" )
client_sock.write( "\x00" * 128 )
write_hello( client_sock, :magic => :wrong )
t.join

36
tests/fakes/dest/hello_wrong_size.rb
View File

@@ -4,40 +4,20 @@
# a valid NBD hello with a random size, then check that we have see an
# EOF on read.
addr, port = *ARGV
require 'flexnbd/fake_dest'
include FlexNBD::FakeDest
require 'socket'
require 'timeout'
require 'flexnbd/constants'
sock = serve( *ARGV )
client_sock = accept( sock, "Timed out waiting for a connection" )
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
client_sock2 = accept( sock, "Timed out waiting for a reconnection",
FlexNBD::MS_RETRY_DELAY_SECS + 1 )
client_sock2.close
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 )
write_hello( client_sock, :size => :wrong )
t.join

17
tests/fakes/dest/reject_acl.rb
View File

@@ -2,20 +2,11 @@
# Accept a connection, then immediately close it. This simulates an ACL rejection.
addr, port = *ARGV
require 'socket'
require 'timeout'
require 'flexnbd/fake_dest'
include FlexNBD::FakeDest
serve_sock = TCPServer.open( addr, port )
begin
Timeout.timeout( 2 ) do
serve_sock.accept.close
end
rescue Timeout::Error
$stderr.puts "Timed out waiting for a connection"
exit 1
end
serve_sock = serve( *ARGV )
accept( serve_sock, "Timed out waiting for a connection" ).close
serve_sock.close

51
tests/flexnbd/fake_dest.rb Normal file
View File

@@ -0,0 +1,51 @@
# encoding: utf-8
require 'socket'
require 'timeout'
require 'flexnbd/constants'
module FlexNBD
module FakeDest
def serve( addr, port )
TCPServer.new( addr, port )
end
def accept( sock, err_msg, timeout=2 )
client_sock = nil
begin
Timeout.timeout(timeout) do
client_sock = sock.accept
end
rescue Timeout::Error
$stderr.puts err_msg
exit 1
end
client_sock
end
def write_hello( client_sock, opts={} )
client_sock.write( "NBDMAGIC" )
if opts[:magic] == :wrong
client_sock.write( "\x00\x00\x42\x02\x81\x86\x12\x52" )
else
client_sock.write( "\x00\x00\x42\x02\x81\x86\x12\x53" )
end
if opts[:size] == :wrong
8.times do client_sock.write rand(256).chr end
else
client_sock.write( "\x00\x00\x00\x00\x00\x00\x10\x00" )
end
client_sock.write( "\x00" * 128 )
end
end
end # module FlexNBD