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
|
module ProcessSpecs
class Daemon
def initialize(argv)
args, @input, @data, @signal, @behavior = argv
@args = Marshal.load [args].pack("H*")
@no_at_exit = false
end
def run
File.delete @signal if File.exists? @signal
send @behavior
File.open(@signal, "w") { }
# Exit without running any at_exit handlers
exit!(0) if @no_at_exit
end
def write(data)
File.open(@data, "wb") { |f| f.puts data }
end
def daemonizing_at_exit
at_exit do
write "running at_exit"
end
@no_at_exit = true
Process.daemon
write "not running at_exit"
end
def return_value
write Process.daemon.to_s
end
def pid
parent = Process.pid
Process.daemon
daemon = Process.pid
write "#{parent}:#{daemon}"
end
def process_group
parent = Process.getpgrp
Process.daemon
daemon = Process.getpgrp
write "#{parent}:#{daemon}"
end
def daemon_at_exit
at_exit do
write "running at_exit"
end
Process.daemon
end
def stay_in_dir
Process.daemon *@args
write Dir.pwd
end
def keep_stdio_open_false_stdout
Process.daemon *@args
$stdout.write "writing to stdout"
end
def keep_stdio_open_false_stderr
Process.daemon *@args
$stderr.write "writing to stderr"
end
def keep_stdio_open_false_stdin
Process.daemon *@args
# Reading from /dev/null will return right away. If STDIN were not
# /dev/null, reading would block and the spec would hang. This is not a
# perfect way to spec the behavior but it works.
write $stdin.read
end
def keep_stdio_open_true_stdout
$stdout.reopen @data
Process.daemon *@args
$stdout.write "writing to stdout"
end
def keep_stdio_open_true_stderr
$stderr.reopen @data
Process.daemon *@args
$stderr.write "writing to stderr"
end
def keep_stdio_open_true_stdin
File.open(@input, "w") { |f| f.puts "reading from stdin" }
$stdin.reopen @input, "r"
Process.daemon *@args
write $stdin.read
end
def keep_stdio_open_files
file = File.open @input, "w"
Process.daemon *@args
write file.closed?
end
end
end
ProcessSpecs::Daemon.new(ARGV).run
|