From 2fbc63b8a8a7a72bd986a37ab423d4d87f54ae12 Mon Sep 17 00:00:00 2001 From: nobu Date: Tue, 27 Dec 2016 07:18:27 +0000 Subject: [PATCH] io.c: ARGF fd leak * io.c (argf_next_argv): fix leak of fd after breaking in #each method. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@57203 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- io.c | 1 + test/ruby/test_argf.rb | 8 +++++++- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/io.c b/io.c index d36bd48d5c..37b57e019b 100644 --- a/io.c +++ b/io.c @@ -8023,6 +8023,7 @@ argf_next_argv(VALUE argf) } if (ARGF.next_p == 1) { + if (ARGF.init_p == 1) argf_close(argf); retry: if (RARRAY_LEN(ARGF.argv) > 0) { VALUE filename = rb_ary_shift(ARGF.argv); diff --git a/test/ruby/test_argf.rb b/test/ruby/test_argf.rb index fdd31c630a..d6aa8f7295 100644 --- a/test/ruby/test_argf.rb +++ b/test/ruby/test_argf.rb @@ -137,7 +137,7 @@ class TestArgf < Test::Unit::TestCase INPUT end - def test_new_lineno + def test_new_lineno_each f = ARGF.class.new(@t1.path, @t2.path, @t3.path) result = [] f.each {|line| result << [f.lineno, line]; break if result.size == 3} @@ -146,10 +146,16 @@ class TestArgf < Test::Unit::TestCase f.rewind assert_equal(2, f.lineno) + ensure + f.close + end + def test_new_lineno_each_char f = ARGF.class.new(@t1.path, @t2.path, @t3.path) f.each_char.to_a assert_equal(0, f.lineno) + ensure + f.close end def test_inplace