From 51d1c5e73bf44baecf598b8997a0d291518da351 Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sat, 1 Aug 2009 09:51:05 +0000
Subject: * random.c (random_float): rejects Infinity and NaN.  
 [ruby-core:24651]

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@24347 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 random.c | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'random.c')

diff --git a/random.c b/random.c
index 7aefb4d4fb..4a62993864 100644
--- a/random.c
+++ b/random.c
@@ -196,6 +196,8 @@ genrand_real(struct MT *mt)
 #ifdef HAVE_FCNTL_H
 #include <fcntl.h>
 #endif
+#include <math.h>
+#include <errno.h>
 
 typedef struct {
     VALUE seed;
@@ -939,6 +941,10 @@ random_float(int argc, VALUE *argv, VALUE obj)
 	    !NIL_P(vmax = rb_to_float(vmax)) ||
 	    (vmax = range_values(vmax, &beg)) != Qfalse) {
 	    max = RFLOAT_VALUE(vmax);
+	    if (isinf(max) || isnan(max)) {
+		VALUE error = INT2FIX(EDOM);
+		rb_exc_raise(rb_class_new_instance(1, &error, rb_eSystemCallError));
+	    }
 	}
 	else {
 	    beg = Qundef;
-- 
cgit v1.2.3