summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorusa <usa@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2018-08-27 14:08:56 +0000
committerusa <usa@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2018-08-27 14:08:56 +0000
commit95abe79e04c3238ea2dde68ea1ab567e354aad21 (patch)
treeeab1b9438935992bdd33675534c0372a0c6feff0
parent1bbf778962743171693a9a279914bf9dae431405 (diff)
merge revision(s) 64007,64019,64020: [Backport #14929]
thread.c (do_select): fix leak on exception When do_select is interrupted and raise happens from RUBY_VM_CHECK_INTS_BLOCKING, the original FD sets we copied do not get freed, leading to a memory leak. Wrap up all the FD sets into a Ruby object to ensure the GC can release an allocations made for rb_fdset_t. This leak existed since Ruby 2.0.0 (r36430) [Bug #14929] increase timeout seconds. * test/ruby/test_io.rb (test_select_leak): increase timeout seconds to pass this test on a high-load machine. 60 sec is not enough at all git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_2_4@64561 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
-rw-r--r--test/ruby/test_io.rb16
-rw-r--r--thread.c55
-rw-r--r--version.h2
3 files changed, 59 insertions, 14 deletions
diff --git a/test/ruby/test_io.rb b/test/ruby/test_io.rb
index 034dac570c..dc99a8c463 100644
--- a/test/ruby/test_io.rb
+++ b/test/ruby/test_io.rb
@@ -3549,4 +3549,20 @@ __END__
end
end
end
+
+ def test_select_leak
+ assert_no_memory_leak([], <<-"end;", <<-"end;", rss: true, timeout: 60)
+ r, w = IO.pipe
+ rset = [r]
+ wset = [w]
+ Thread.new { IO.select(rset, wset, nil, 0) }.join
+ end;
+ 20_000.times do
+ th = Thread.new { IO.select(rset, wset) }
+ Thread.pass until th.stop?
+ th.kill
+ th.join
+ end
+ end;
+ end
end
diff --git a/thread.c b/thread.c
index d74c731a82..3992753304 100644
--- a/thread.c
+++ b/thread.c
@@ -3704,23 +3704,53 @@ update_timeval(struct timeval *timeout, double limit)
}
}
+struct select_set {
+ rb_fdset_t read;
+ rb_fdset_t write;
+ rb_fdset_t except;
+};
+
+static size_t
+select_set_memsize(const void *p)
+{
+ return sizeof(struct select_set);
+}
+
+static void
+select_set_free(void *p)
+{
+ struct select_set *orig = p;
+
+ rb_fd_term(&orig->read);
+ rb_fd_term(&orig->write);
+ rb_fd_term(&orig->except);
+ xfree(orig);
+}
+
+static const rb_data_type_t select_set_type = {
+ "select_set",
+ {NULL, select_set_free, select_set_memsize,},
+ 0, 0, RUBY_TYPED_FREE_IMMEDIATELY
+};
+
static int
do_select(int n, rb_fdset_t *readfds, rb_fdset_t *writefds,
rb_fdset_t *exceptfds, struct timeval *timeout)
{
int MAYBE_UNUSED(result);
int lerrno;
- rb_fdset_t MAYBE_UNUSED(orig_read);
- rb_fdset_t MAYBE_UNUSED(orig_write);
- rb_fdset_t MAYBE_UNUSED(orig_except);
double limit = 0;
struct timeval wait_rest;
rb_thread_t *th = GET_THREAD();
+ VALUE o;
+ struct select_set *orig;
+
+ o = TypedData_Make_Struct(0, struct select_set, &select_set_type, orig);
#define do_select_update() \
- (restore_fdset(readfds, &orig_read), \
- restore_fdset(writefds, &orig_write), \
- restore_fdset(exceptfds, &orig_except), \
+ (restore_fdset(readfds, &orig->read), \
+ restore_fdset(writefds, &orig->write), \
+ restore_fdset(exceptfds, &orig->except), \
update_timeval(timeout, limit), \
TRUE)
@@ -3732,7 +3762,7 @@ do_select(int n, rb_fdset_t *readfds, rb_fdset_t *writefds,
}
#define fd_init_copy(f) \
- (f##fds) ? rb_fd_init_copy(&orig_##f, f##fds) : rb_fd_no_init(&orig_##f)
+ (f##fds) ? rb_fd_init_copy(&orig->f, f##fds) : rb_fd_no_init(&orig->f)
fd_init_copy(read);
fd_init_copy(write);
fd_init_copy(except);
@@ -3747,14 +3777,13 @@ do_select(int n, rb_fdset_t *readfds, rb_fdset_t *writefds,
if (result < 0) lerrno = errno;
}, ubf_select, th, FALSE);
- RUBY_VM_CHECK_INTS_BLOCKING(th);
+ RUBY_VM_CHECK_INTS_BLOCKING(th); /* may raise */
} while (result < 0 && retryable(errno = lerrno) && do_select_update());
-#define fd_term(f) if (f##fds) rb_fd_term(&orig_##f)
- fd_term(read);
- fd_term(write);
- fd_term(except);
-#undef fd_term
+ /* didn't raise, perform cleanup ourselves */
+ select_set_free(orig);
+ rb_gc_force_recycle(o);
+
return result;
}
diff --git a/version.h b/version.h
index 4dc154a458..5058df988b 100644
--- a/version.h
+++ b/version.h
@@ -1,6 +1,6 @@
#define RUBY_VERSION "2.4.5"
#define RUBY_RELEASE_DATE "2018-08-27"
-#define RUBY_PATCHLEVEL 318
+#define RUBY_PATCHLEVEL 319
#define RUBY_RELEASE_YEAR 2018
#define RUBY_RELEASE_MONTH 8