blob: 772df2d09e7b556e309d2aa57ed00f3127422c62 (
plain)
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, @behavior = argv
@args = Marshal.load [args].pack("H*")
@no_at_exit = false
end
def run
send @behavior
# 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"
write ""
end
def keep_stdio_open_false_stderr
Process.daemon(*@args)
$stderr.write "writing to stderr"
write ""
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
|