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 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
|
/*
* $Id: ossl_rand.c,v 1.10 2002/03/05 15:05:29 majkl Exp $
* 'OpenSSL for Ruby' project
* Copyright (C) 2001-2002 Michal Rokos <m.rokos@sh.cvut.cz>
* All rights reserved.
*/
/*
* This program is licenced under the same licence as Ruby.
* (See the file 'LICENCE'.)
*/
#include "ossl.h"
/*
* Classes
*/
VALUE eRandomError;
/*
* Struct
*/
/*
* Public
*/
/*
* Private
*/
static VALUE
ossl_rand_seed(VALUE self, VALUE str)
{
str = rb_obj_as_string(str);
RAND_seed(RSTRING(str)->ptr, RSTRING(str)->len);
return str;
}
static VALUE
ossl_rand_load_file(VALUE self, VALUE filename)
{
filename = rb_str_to_str(filename);
Check_SafeStr(filename);
if(!RAND_load_file(RSTRING(filename)->ptr, -1)) {
OSSL_Raise(eRandomError, "");
}
return Qtrue;
}
static VALUE
ossl_rand_write_file(VALUE self, VALUE filename)
{
filename = rb_str_to_str(filename);
Check_SafeStr(filename);
if (RAND_write_file(RSTRING(filename)->ptr) == -1) {
OSSL_Raise(eRandomError, "");
}
return Qtrue;
}
static VALUE
ossl_rand_bytes(VALUE self, VALUE len)
{
unsigned char *buffer = NULL;
VALUE str;
Check_Type(len, T_FIXNUM);
if (!(buffer = OPENSSL_malloc(FIX2INT(len)+1))) {
OSSL_Raise(eRandomError, "");
}
if (!RAND_bytes(buffer, FIX2INT(len))) {
OPENSSL_free(buffer);
OSSL_Raise(eRandomError, "");
}
str = rb_str_new(buffer, FIX2INT(len));
OPENSSL_free(buffer);
return str;
}
static VALUE
ossl_rand_egd(VALUE self, VALUE filename)
{
Check_SafeStr(filename);
if(!RAND_egd(RSTRING(filename)->ptr)) {
OSSL_Raise(eRandomError, "");
}
return Qtrue;
}
static VALUE
ossl_rand_egd_bytes(VALUE self, VALUE filename, VALUE len)
{
Check_SafeStr(filename);
Check_Type(len, T_FIXNUM);
if (!RAND_egd_bytes(RSTRING(filename)->ptr, FIX2INT(len))) {
OSSL_Raise(eRandomError, "");
}
return Qtrue;
}
/*
* INIT
*/
void
Init_ossl_rand(VALUE module)
{
rb_define_method(module, "seed", ossl_rand_seed, 1);
rb_define_method(module, "load_random_file", ossl_rand_load_file, 1);
rb_define_method(module, "write_random_file", ossl_rand_write_file, 1);
rb_define_method(module, "random_bytes", ossl_rand_bytes, 1);
rb_define_method(module, "egd", ossl_rand_egd, 1);
rb_define_method(module, "egd_bytes", ossl_rand_egd_bytes, 2);
eRandomError = rb_define_class_under(module, "RandomError", rb_eStandardError);
}
|