blob: f41f3584cb9f87ac7d55f7d0d1661945aea03823 (
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
|
begin
require "readline"
rescue LoadError
end
if defined?(Readline)
require "test/unit"
require "tempfile"
class TestReadline < Test::Unit::TestCase
def test_readline
stdin = Tempfile.new("test_readline_stdin")
stdout = Tempfile.new("test_readline_stdout")
begin
stdin.write("hello\n")
stdin.rewind
line = replace_stdio(stdin, stdout) { Readline.readline("> ") }
assert_equal("hello", line)
assert_equal(true, line.tainted?)
assert_raises(SecurityError) do
Thread.start {
$SAFE = 1
replace_stdio(stdin, stdout) { Readline.readline("> ".taint) }
}.join
end
assert_raises(SecurityError) do
Thread.start {
$SAFE = 4
replace_stdio(stdin, stdout) { Readline.readline("> ") }
}.join
end
stdout.rewind
assert_equal("> ", stdout.read(2))
ensure
stdin.close(true)
stdout.close(true)
end
end
private
def replace_stdio(stdin, stdout)
orig_stdin = STDIN.dup
orig_stdout = STDOUT.dup
STDIN.reopen(stdin)
STDOUT.reopen(stdout)
begin
yield
ensure
STDIN.reopen(orig_stdin)
STDOUT.reopen(orig_stdout)
end
end
end
end
|