Rubocop
This commit is contained in:
Patrick J Cherry
@@ -7,25 +7,22 @@ require 'rexml/streamlistener'
|
||||
|
||||
Thread.abort_on_exception = true
|
||||
|
||||
|
||||
class Executor
|
||||
attr_reader :pid
|
||||
|
||||
def run( cmd )
|
||||
@pid = fork do exec cmd end
|
||||
def run(cmd)
|
||||
@pid = fork { exec cmd }
|
||||
end
|
||||
end # class Executor
|
||||
|
||||
|
||||
class ValgrindExecutor
|
||||
attr_reader :pid
|
||||
|
||||
def run( cmd )
|
||||
@pid = fork do exec "valgrind --track-origins=yes --suppressions=custom.supp #{cmd}" end
|
||||
def run(cmd)
|
||||
@pid = fork { exec "valgrind --track-origins=yes --suppressions=custom.supp #{cmd}" }
|
||||
end
|
||||
end # class ValgrindExecutor
|
||||
|
||||
|
||||
class ValgrindKillingExecutor
|
||||
attr_reader :pid
|
||||
|
||||
@@ -33,10 +30,10 @@ class ValgrindKillingExecutor
|
||||
attr_accessor :what, :kind, :pid
|
||||
attr_reader :backtrace
|
||||
def initialize
|
||||
@backtrace=[]
|
||||
@what = ""
|
||||
@kind = ""
|
||||
@pid = ""
|
||||
@backtrace = []
|
||||
@what = ''
|
||||
@kind = ''
|
||||
@pid = ''
|
||||
end
|
||||
|
||||
def add_frame
|
||||
@@ -56,115 +53,104 @@ class ValgrindKillingExecutor
|
||||
end
|
||||
|
||||
def to_s
|
||||
([@what + " (#{@kind}) in #{@pid}"] + @backtrace.map{|h| "#{h[:file]}:#{h[:line]} #{h[:fn]}" }).join("\n")
|
||||
([@what + " (#{@kind}) in #{@pid}"] + @backtrace.map { |h| "#{h[:file]}:#{h[:line]} #{h[:fn]}" }).join("\n")
|
||||
end
|
||||
|
||||
end # class Error
|
||||
|
||||
|
||||
class ErrorListener
|
||||
include REXML::StreamListener
|
||||
def initialize( killer )
|
||||
def initialize(killer)
|
||||
@killer = killer
|
||||
@error = Error.new
|
||||
@found = false
|
||||
end
|
||||
|
||||
def text( text )
|
||||
def text(text)
|
||||
@text = text
|
||||
end
|
||||
|
||||
def tag_start(tag, attrs)
|
||||
def tag_start(tag, _attrs)
|
||||
case tag.to_s
|
||||
when "error"
|
||||
when 'error'
|
||||
@found = true
|
||||
when "frame"
|
||||
when 'frame'
|
||||
@error.add_frame
|
||||
end
|
||||
end
|
||||
|
||||
def tag_end(tag)
|
||||
case tag.to_s
|
||||
when "what"
|
||||
when 'what'
|
||||
@error.what = @text if @found
|
||||
@text = ""
|
||||
when "kind"
|
||||
@text = ''
|
||||
when 'kind'
|
||||
@error.kind = @text if @found
|
||||
when "file"
|
||||
@error.add_file( @text ) if @found
|
||||
when "fn"
|
||||
@error.add_fn( @text ) if @found
|
||||
when "line"
|
||||
@error.add_line( @text ) if @found
|
||||
when "error", "stack"
|
||||
if @found
|
||||
@killer.call( @error )
|
||||
end
|
||||
when "pid"
|
||||
@error.pid=@text
|
||||
when 'file'
|
||||
@error.add_file(@text) if @found
|
||||
when 'fn'
|
||||
@error.add_fn(@text) if @found
|
||||
when 'line'
|
||||
@error.add_line(@text) if @found
|
||||
when 'error', 'stack'
|
||||
@killer.call(@error) if @found
|
||||
when 'pid'
|
||||
@error.pid = @text
|
||||
end
|
||||
end
|
||||
end # class ErrorListener
|
||||
|
||||
|
||||
class DebugErrorListener < ErrorListener
|
||||
def text( txt )
|
||||
def text(txt)
|
||||
print txt
|
||||
super( txt )
|
||||
super(txt)
|
||||
end
|
||||
|
||||
def tag_start( tag, attrs )
|
||||
def tag_start(tag, attrs)
|
||||
print "<#{tag}>"
|
||||
super( tag, attrs )
|
||||
super(tag, attrs)
|
||||
end
|
||||
|
||||
def tag_end( tag )
|
||||
def tag_end(tag)
|
||||
print "</#{tag}>"
|
||||
super( tag )
|
||||
super(tag)
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
def initialize
|
||||
@pid = nil
|
||||
end
|
||||
|
||||
def run( cmd )
|
||||
def run(cmd)
|
||||
@io_r, io_w = IO.pipe
|
||||
@pid = fork do exec( "valgrind --suppressions=custom.supp --xml=yes --xml-fd=#{io_w.fileno} " + cmd ) end
|
||||
launch_watch_thread( @pid, @io_r )
|
||||
@pid = fork { exec("valgrind --suppressions=custom.supp --xml=yes --xml-fd=#{io_w.fileno} " + cmd) }
|
||||
launch_watch_thread(@pid, @io_r)
|
||||
@pid
|
||||
end
|
||||
|
||||
|
||||
def call( err )
|
||||
$stderr.puts "*"*72
|
||||
$stderr.puts "* Valgrind error spotted:"
|
||||
$stderr.puts err.to_s.split("\n").map{|s| " #{s}"}
|
||||
$stderr.puts "*"*72
|
||||
Process.kill( "KILL", @pid )
|
||||
def call(err)
|
||||
warn '*' * 72
|
||||
warn '* Valgrind error spotted:'
|
||||
warn err.to_s.split("\n").map { |s| " #{s}" }
|
||||
warn '*' * 72
|
||||
Process.kill('KILL', @pid)
|
||||
exit(1)
|
||||
end
|
||||
|
||||
|
||||
private
|
||||
|
||||
def pick_listener
|
||||
ENV['DEBUG'] ? DebugErrorListener : ErrorListener
|
||||
end
|
||||
|
||||
def launch_watch_thread(pid, io_r)
|
||||
def launch_watch_thread(_pid, io_r)
|
||||
Thread.start do
|
||||
io_source = REXML::IOSource.new( io_r )
|
||||
listener = pick_listener.new( self )
|
||||
REXML::Document.parse_stream( io_source, listener )
|
||||
io_source = REXML::IOSource.new(io_r)
|
||||
listener = pick_listener.new(self)
|
||||
REXML::Document.parse_stream(io_source, listener)
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
end # class ValgrindExecutor
|
||||
|
||||
|
||||
module FlexNBD
|
||||
# Noddy test class to exercise FlexNBD from the outside for testing.
|
||||
#
|
||||
@@ -173,7 +159,7 @@ module FlexNBD
|
||||
|
||||
class << self
|
||||
def counter
|
||||
Dir['tmp/*'].select{|f| File.file?(f)}.length + 1
|
||||
Dir['tmp/*'].select { |f| File.file?(f) }.length + 1
|
||||
end
|
||||
end
|
||||
|
||||
@@ -189,19 +175,18 @@ module FlexNBD
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
def build_debug_opt
|
||||
if @do_debug
|
||||
"--verbose"
|
||||
'--verbose'
|
||||
else
|
||||
"--quiet"
|
||||
'--quiet'
|
||||
end
|
||||
end
|
||||
|
||||
attr_accessor :prefetch_proxy
|
||||
|
||||
def initialize( bin, ip, port )
|
||||
@bin = bin
|
||||
def initialize(bin, ip, port)
|
||||
@bin = bin
|
||||
@do_debug = ENV['DEBUG']
|
||||
@debug = build_debug_opt
|
||||
raise "#{bin} not executable" unless File.executable?(bin)
|
||||
@@ -213,17 +198,15 @@ module FlexNBD
|
||||
@prefetch_proxy = false
|
||||
end
|
||||
|
||||
|
||||
def debug?
|
||||
!!@do_debug
|
||||
end
|
||||
|
||||
def debug( msg )
|
||||
$stderr.puts msg if debug?
|
||||
def debug(msg)
|
||||
warn msg if debug?
|
||||
end
|
||||
|
||||
|
||||
def serve_cmd( file, acl )
|
||||
def serve_cmd(file, acl)
|
||||
"#{bin} serve "\
|
||||
"--addr #{ip} "\
|
||||
"--port #{port} "\
|
||||
@@ -233,8 +216,7 @@ module FlexNBD
|
||||
"#{acl.join(' ')}"
|
||||
end
|
||||
|
||||
|
||||
def listen_cmd( file, acl )
|
||||
def listen_cmd(file, acl)
|
||||
"#{bin} listen "\
|
||||
"--addr #{ip} "\
|
||||
"--port #{port} "\
|
||||
@@ -244,18 +226,17 @@ module FlexNBD
|
||||
"#{acl.join(' ')}"
|
||||
end
|
||||
|
||||
def proxy_cmd( connect_ip, connect_port )
|
||||
def proxy_cmd(connect_ip, connect_port)
|
||||
"#{bin}-proxy "\
|
||||
"--addr #{ip} "\
|
||||
"--port #{port} "\
|
||||
"--conn-addr #{connect_ip} "\
|
||||
"--conn-port #{connect_port} "\
|
||||
"#{prefetch_proxy ? "--cache " : ""}"\
|
||||
"#{prefetch_proxy ? '--cache ' : ''}"\
|
||||
"#{@debug}"
|
||||
end
|
||||
|
||||
|
||||
def read_cmd( offset, length )
|
||||
def read_cmd(offset, length)
|
||||
"#{bin} read "\
|
||||
"--addr #{ip} "\
|
||||
"--port #{port} "\
|
||||
@@ -264,8 +245,7 @@ module FlexNBD
|
||||
"--size #{length}"
|
||||
end
|
||||
|
||||
|
||||
def write_cmd( offset, data )
|
||||
def write_cmd(offset, data)
|
||||
"#{bin} write "\
|
||||
"--addr #{ip} "\
|
||||
"--port #{port} "\
|
||||
@@ -274,30 +254,29 @@ module FlexNBD
|
||||
"--size #{data.length}"
|
||||
end
|
||||
|
||||
|
||||
def base_mirror_opts( dest_ip, dest_port )
|
||||
def base_mirror_opts(dest_ip, dest_port)
|
||||
"--addr #{dest_ip} "\
|
||||
"--port #{dest_port} "\
|
||||
"--sock #{ctrl} "\
|
||||
end
|
||||
|
||||
def unlink_mirror_opts( dest_ip, dest_port )
|
||||
"#{base_mirror_opts( dest_ip, dest_port )} "\
|
||||
"--unlink "
|
||||
def unlink_mirror_opts(dest_ip, dest_port)
|
||||
"#{base_mirror_opts(dest_ip, dest_port)} "\
|
||||
'--unlink '
|
||||
end
|
||||
|
||||
def base_mirror_cmd( opts )
|
||||
def base_mirror_cmd(opts)
|
||||
"#{@bin} mirror "\
|
||||
"#{opts} "\
|
||||
"#{@debug}"
|
||||
end
|
||||
|
||||
def mirror_cmd(dest_ip, dest_port)
|
||||
base_mirror_cmd( base_mirror_opts( dest_ip, dest_port ) )
|
||||
base_mirror_cmd(base_mirror_opts(dest_ip, dest_port))
|
||||
end
|
||||
|
||||
def mirror_unlink_cmd( dest_ip, dest_port )
|
||||
base_mirror_cmd( unlink_mirror_opts( dest_ip, dest_port ) )
|
||||
def mirror_unlink_cmd(dest_ip, dest_port)
|
||||
base_mirror_cmd(unlink_mirror_opts(dest_ip, dest_port))
|
||||
end
|
||||
|
||||
def break_cmd
|
||||
@@ -312,58 +291,64 @@ module FlexNBD
|
||||
"#{@debug}"
|
||||
end
|
||||
|
||||
def acl_cmd( *acl )
|
||||
def acl_cmd(*acl)
|
||||
"#{@bin} acl " \
|
||||
"--sock #{ctrl} "\
|
||||
"#{@debug} "\
|
||||
"#{acl.join " "}"
|
||||
"#{acl.join ' '}"
|
||||
end
|
||||
|
||||
|
||||
def run_serve_cmd(cmd)
|
||||
File.unlink(ctrl) if File.exists?(ctrl)
|
||||
debug( cmd )
|
||||
File.unlink(ctrl) if File.exist?(ctrl)
|
||||
debug(cmd)
|
||||
|
||||
@pid = @executor.run( cmd )
|
||||
@pid = @executor.run(cmd)
|
||||
|
||||
while !File.socket?(ctrl)
|
||||
until File.socket?(ctrl)
|
||||
pid, status = Process.wait2(@pid, Process::WNOHANG)
|
||||
raise "server did not start (#{cmd})" if pid
|
||||
sleep 0.1
|
||||
end
|
||||
|
||||
|
||||
start_wait_thread( @pid )
|
||||
start_wait_thread(@pid)
|
||||
at_exit { kill }
|
||||
end
|
||||
private :run_serve_cmd
|
||||
|
||||
|
||||
def serve( file, *acl)
|
||||
cmd = serve_cmd( file, acl )
|
||||
run_serve_cmd( cmd )
|
||||
sleep( 0.2 ) until File.exists?( ctrl )
|
||||
def serve(file, *acl)
|
||||
cmd = serve_cmd(file, acl)
|
||||
run_serve_cmd(cmd)
|
||||
sleep(0.2) until File.exist?(ctrl)
|
||||
end
|
||||
|
||||
|
||||
def listen(file, *acl)
|
||||
run_serve_cmd( listen_cmd( file, acl ) )
|
||||
run_serve_cmd(listen_cmd(file, acl))
|
||||
end
|
||||
|
||||
def tcp_server_open?
|
||||
# raises if the other side doesn't accept()
|
||||
sock = TCPSocket.new(ip, port) rescue nil
|
||||
sock = begin
|
||||
TCPSocket.new(ip, port)
|
||||
rescue StandardError
|
||||
nil
|
||||
end
|
||||
|
||||
success = !!sock
|
||||
( sock.close rescue nil) if sock
|
||||
if sock
|
||||
(begin
|
||||
sock.close
|
||||
rescue StandardError
|
||||
nil
|
||||
end)
|
||||
end
|
||||
success
|
||||
end
|
||||
|
||||
def proxy( connect_ip, connect_port )
|
||||
cmd = proxy_cmd( connect_ip, connect_port )
|
||||
debug( cmd )
|
||||
def proxy(connect_ip, connect_port)
|
||||
cmd = proxy_cmd(connect_ip, connect_port)
|
||||
debug(cmd)
|
||||
|
||||
@pid = @executor.run( cmd )
|
||||
@pid = @executor.run(cmd)
|
||||
|
||||
until tcp_server_open?
|
||||
pid, status = Process.wait2(@pid, Process::WNOHANG)
|
||||
@@ -371,31 +356,29 @@ module FlexNBD
|
||||
sleep 0.1
|
||||
end
|
||||
|
||||
start_wait_thread( @pid )
|
||||
start_wait_thread(@pid)
|
||||
at_exit { kill }
|
||||
end
|
||||
|
||||
|
||||
def start_wait_thread( pid )
|
||||
def start_wait_thread(pid)
|
||||
@wait_thread = Thread.start do
|
||||
_, status = Process.waitpid2( pid )
|
||||
_, status = Process.waitpid2(pid)
|
||||
|
||||
if @kill
|
||||
if status.signaled?
|
||||
fail "flexnbd quit with a bad signal: #{status.inspect}" unless
|
||||
raise "flexnbd quit with a bad signal: #{status.inspect}" unless
|
||||
@kill.include? status.termsig
|
||||
else
|
||||
fail "flexnbd quit with a bad status: #{status.inspect}" unless
|
||||
raise "flexnbd quit with a bad status: #{status.inspect}" unless
|
||||
@kill.include? status.exitstatus
|
||||
end
|
||||
else
|
||||
$stderr.puts "flexnbd #{self.pid} quit"
|
||||
fail "flexnbd #{self.pid} quit early with status #{status.to_i}"
|
||||
warn "flexnbd #{self.pid} quit"
|
||||
raise "flexnbd #{self.pid} quit early with status #{status.to_i}"
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
def can_die(*status)
|
||||
status = [0] if status.empty?
|
||||
@kill += status
|
||||
@@ -407,7 +390,7 @@ module FlexNBD
|
||||
can_die(1)
|
||||
if @pid
|
||||
begin
|
||||
Process.kill("INT", @pid)
|
||||
Process.kill('INT', @pid)
|
||||
rescue Errno::ESRCH => e
|
||||
# already dead. Presumably this means it went away after a
|
||||
# can_die() call.
|
||||
@@ -417,63 +400,60 @@ module FlexNBD
|
||||
end
|
||||
|
||||
def read(offset, length)
|
||||
cmd = read_cmd( offset, length )
|
||||
debug( cmd )
|
||||
cmd = read_cmd(offset, length)
|
||||
debug(cmd)
|
||||
|
||||
IO.popen(cmd) do |fh|
|
||||
return fh.read
|
||||
end
|
||||
raise IOError.new "NBD read failed" unless $?.success?
|
||||
raise IOError, 'NBD read failed' unless $CHILD_STATUS.success?
|
||||
out
|
||||
end
|
||||
|
||||
def write(offset, data)
|
||||
cmd = write_cmd( offset, data )
|
||||
debug( cmd )
|
||||
cmd = write_cmd(offset, data)
|
||||
debug(cmd)
|
||||
|
||||
IO.popen(cmd, "w") do |fh|
|
||||
IO.popen(cmd, 'w') do |fh|
|
||||
fh.write(data)
|
||||
end
|
||||
raise IOError.new "NBD write failed" unless $?.success?
|
||||
raise IOError, 'NBD write failed' unless $CHILD_STATUS.success?
|
||||
nil
|
||||
end
|
||||
|
||||
|
||||
def join
|
||||
@wait_thread.join
|
||||
end
|
||||
|
||||
def mirror_unchecked(dest_ip, dest_port, _bandwidth = nil, _action = nil, timeout = nil)
|
||||
cmd = mirror_cmd(dest_ip, dest_port)
|
||||
debug(cmd)
|
||||
|
||||
def mirror_unchecked( dest_ip, dest_port, bandwidth=nil, action=nil, timeout=nil )
|
||||
cmd = mirror_cmd( dest_ip, dest_port)
|
||||
debug( cmd )
|
||||
|
||||
maybe_timeout( cmd, timeout )
|
||||
maybe_timeout(cmd, timeout)
|
||||
end
|
||||
|
||||
def mirror_unlink(dest_ip, dest_port, timeout = nil)
|
||||
cmd = mirror_unlink_cmd(dest_ip, dest_port)
|
||||
debug(cmd)
|
||||
|
||||
def mirror_unlink( dest_ip, dest_port, timeout=nil )
|
||||
cmd = mirror_unlink_cmd( dest_ip, dest_port )
|
||||
debug( cmd )
|
||||
|
||||
maybe_timeout( cmd, timeout )
|
||||
maybe_timeout(cmd, timeout)
|
||||
end
|
||||
|
||||
|
||||
def maybe_timeout(cmd, timeout=nil )
|
||||
stdout, stderr = "",""
|
||||
def maybe_timeout(cmd, timeout = nil)
|
||||
stdout = ''
|
||||
stderr = ''
|
||||
stat = nil
|
||||
run = Proc.new do
|
||||
run = proc do
|
||||
# Ruby 1.9 changed the popen3 api. instead of 3 args, the block
|
||||
# gets 4. Not only that, but it no longer sets $?, so we have to
|
||||
# go elsewhere for the process' exit status.
|
||||
Open3.popen3( cmd ) do |io_in, io_out, io_err, maybe_thr|
|
||||
Open3.popen3(cmd) do |io_in, io_out, io_err, maybe_thr|
|
||||
io_in.close
|
||||
stdout.replace io_out.read
|
||||
stderr.replace io_err.read
|
||||
stat = maybe_thr.value if maybe_thr
|
||||
end
|
||||
stat ||= $?
|
||||
stat ||= $CHILD_STATUS
|
||||
end
|
||||
|
||||
if timeout
|
||||
@@ -485,85 +465,73 @@ module FlexNBD
|
||||
[stdout, stderr, stat]
|
||||
end
|
||||
|
||||
|
||||
def mirror(dest_ip, dest_port, bandwidth=nil, action=nil)
|
||||
stdout, stderr, status = mirror_unchecked( dest_ip, dest_port, bandwidth, action )
|
||||
raise IOError.new( "Migrate command failed\n" + stderr) unless status.success?
|
||||
def mirror(dest_ip, dest_port, bandwidth = nil, action = nil)
|
||||
stdout, stderr, status = mirror_unchecked(dest_ip, dest_port, bandwidth, action)
|
||||
raise IOError, "Migrate command failed\n" + stderr unless status.success?
|
||||
|
||||
stdout
|
||||
end
|
||||
|
||||
|
||||
|
||||
def break(timeout=nil)
|
||||
def break(timeout = nil)
|
||||
cmd = break_cmd
|
||||
debug( cmd )
|
||||
debug(cmd)
|
||||
|
||||
maybe_timeout( cmd, timeout )
|
||||
maybe_timeout(cmd, timeout)
|
||||
end
|
||||
|
||||
|
||||
def acl(*acl)
|
||||
cmd = acl_cmd( *acl )
|
||||
debug( cmd )
|
||||
cmd = acl_cmd(*acl)
|
||||
debug(cmd)
|
||||
|
||||
maybe_timeout( cmd, 2 )
|
||||
maybe_timeout(cmd, 2)
|
||||
end
|
||||
|
||||
def status(timeout = nil)
|
||||
cmd = status_cmd
|
||||
debug(cmd)
|
||||
|
||||
def status( timeout = nil )
|
||||
cmd = status_cmd()
|
||||
debug( cmd )
|
||||
|
||||
o,e = maybe_timeout( cmd, timeout )
|
||||
o, e = maybe_timeout(cmd, timeout)
|
||||
|
||||
[parse_status(o), e]
|
||||
end
|
||||
|
||||
|
||||
def launched?
|
||||
!!@pid
|
||||
end
|
||||
|
||||
|
||||
def paused
|
||||
Process.kill( "STOP", @pid )
|
||||
Process.kill('STOP', @pid)
|
||||
yield
|
||||
ensure
|
||||
Process.kill( "CONT", @pid )
|
||||
Process.kill('CONT', @pid)
|
||||
end
|
||||
|
||||
|
||||
protected
|
||||
|
||||
def control_command(*args)
|
||||
raise "Server not running" unless @pid
|
||||
raise 'Server not running' unless @pid
|
||||
args = args.compact
|
||||
UNIXSocket.open(@ctrl) do |u|
|
||||
u.write(args.join("\n") + "\n")
|
||||
code, message = u.readline.split(": ", 2)
|
||||
code, message = u.readline.split(': ', 2)
|
||||
return [code, message]
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
def parse_status( status )
|
||||
def parse_status(status)
|
||||
hsh = {}
|
||||
|
||||
status.split(" ").each do |part|
|
||||
status.split(' ').each do |part|
|
||||
next if part.strip.empty?
|
||||
a,b = part.split("=")
|
||||
a, b = part.split('=')
|
||||
b.strip!
|
||||
b = true if b == "true"
|
||||
b = false if b == "false"
|
||||
b = true if b == 'true'
|
||||
b = false if b == 'false'
|
||||
|
||||
hsh[a.strip] = b
|
||||
end
|
||||
|
||||
hsh
|
||||
end
|
||||
|
||||
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
|
||||
Reference in New Issue
Block a user