From f0367359d6f55de6caf62fd1dbb72d3661d4ae12 Mon Sep 17 00:00:00 2001 From: matz Date: Tue, 4 Jul 2000 04:17:26 +0000 Subject: matz git-svn-id: http://svn.ruby-lang.org/repos/ruby/trunk@810 b2dd03c8-39d4-4d8f-98ff-823fe69b080e --- random.c | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) (limited to 'random.c') diff --git a/random.c b/random.c index 65fc69ed3..32cb525a3 100644 --- a/random.c +++ b/random.c @@ -130,11 +130,15 @@ rb_f_srand(argc, argv, obj) } static VALUE -rb_f_rand(obj, vmax) - VALUE obj, vmax; +rb_f_rand(argc, argv, obj) + int argc; + VALUE *argv; + VALUE obj; { + VALUE vmax; long val, max; + rb_scan_args(argc, argv, "01", &vmax); if (first) { struct timeval tv; @@ -148,9 +152,14 @@ rb_f_rand(obj, vmax) /* fall through */ case T_BIGNUM: return rb_big_rand(vmax, RANDOM_NUMBER); + case T_NIL: + max = 0; + break; + default: + max = NUM2LONG(vmax); + break; } - max = NUM2LONG(vmax); if (max == 0) { return rb_float_new(RANDOM_NUMBER); } @@ -164,5 +173,5 @@ void Init_Random() { rb_define_global_function("srand", rb_f_srand, -1); - rb_define_global_function("rand", rb_f_rand, 1); + rb_define_global_function("rand", rb_f_rand, -1); } -- cgit