File: ossl_rand.c

package info (click to toggle)
libopenssl-ruby 0.1.4a-1sarge1
  • links: PTS
  • area: main
  • in suites: sarge
  • size: 556 kB
  • ctags: 701
  • sloc: ansic: 6,200; ruby: 1,538; makefile: 45
file content (126 lines) | stat: -rw-r--r-- 2,349 bytes parent folder | download
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);
}