зеркало из https://github.com/github/ruby.git
853 строки
19 KiB
C
853 строки
19 KiB
C
/*
|
|
* $Id$
|
|
* 'OpenSSL for Ruby' project
|
|
* Copyright (C) 2001-2002 Technorama team <oss-ruby@technorama.net>
|
|
* All rights reserved.
|
|
*/
|
|
/*
|
|
* This program is licenced under the same licence as Ruby.
|
|
* (See the file 'LICENCE'.)
|
|
*/
|
|
/* modified by Michal Rokos <m.rokos@sh.cvut.cz> */
|
|
#include "ossl.h"
|
|
|
|
#define WrapBN(klass, obj, bn) do { \
|
|
if (!bn) { \
|
|
ossl_raise(rb_eRuntimeError, "BN wasn't initialized!"); \
|
|
} \
|
|
obj = Data_Wrap_Struct(klass, 0, BN_clear_free, bn); \
|
|
} while (0)
|
|
|
|
#define GetBN(obj, bn) do { \
|
|
Data_Get_Struct(obj, BIGNUM, bn); \
|
|
if (!bn) { \
|
|
ossl_raise(rb_eRuntimeError, "BN wasn't initialized!"); \
|
|
} \
|
|
} while (0)
|
|
|
|
#define SafeGetBN(obj, bn) do { \
|
|
OSSL_Check_Kind(obj, cBN); \
|
|
GetBN(obj, bn); \
|
|
} while (0)
|
|
|
|
/*
|
|
* Classes
|
|
*/
|
|
VALUE cBN;
|
|
VALUE eBNError;
|
|
|
|
/*
|
|
* Public
|
|
*/
|
|
VALUE
|
|
ossl_bn_new(const BIGNUM *bn)
|
|
{
|
|
BIGNUM *newbn;
|
|
VALUE obj;
|
|
|
|
newbn = bn ? BN_dup(bn) : BN_new();
|
|
if (!newbn) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
WrapBN(cBN, obj, newbn);
|
|
|
|
return obj;
|
|
}
|
|
|
|
BIGNUM *
|
|
GetBNPtr(VALUE obj)
|
|
{
|
|
BIGNUM *bn = NULL;
|
|
|
|
if (RTEST(rb_obj_is_kind_of(obj, cBN))) {
|
|
GetBN(obj, bn);
|
|
} else switch (TYPE(obj)) {
|
|
case T_FIXNUM:
|
|
case T_BIGNUM:
|
|
obj = rb_String(obj);
|
|
if (!BN_dec2bn(&bn, StringValuePtr(obj))) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
WrapBN(cBN, obj, bn); /* Handle potencial mem leaks */
|
|
break;
|
|
default:
|
|
ossl_raise(rb_eTypeError, "Cannot convert into OpenSSL::BN");
|
|
}
|
|
return bn;
|
|
}
|
|
|
|
/*
|
|
* Private
|
|
*/
|
|
/*
|
|
* BN_CTX - is used in more difficult math. ops
|
|
* (Why just 1? Because Ruby itself isn't thread safe,
|
|
* we don't need to care about threads)
|
|
*/
|
|
BN_CTX *ossl_bn_ctx;
|
|
|
|
static VALUE
|
|
ossl_bn_alloc(VALUE klass)
|
|
{
|
|
BIGNUM *bn;
|
|
VALUE obj;
|
|
|
|
if (!(bn = BN_new())) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
WrapBN(klass, obj, bn);
|
|
|
|
return obj;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* BN.new => aBN
|
|
* BN.new(bn) => aBN
|
|
* BN.new(string) => aBN
|
|
* BN.new(string, 0 | 2 | 10 | 16) => aBN
|
|
*/
|
|
static VALUE
|
|
ossl_bn_initialize(int argc, VALUE *argv, VALUE self)
|
|
{
|
|
BIGNUM *bn;
|
|
VALUE str, bs;
|
|
int base = 10;
|
|
|
|
if (rb_scan_args(argc, argv, "11", &str, &bs) == 2) {
|
|
base = NUM2INT(bs);
|
|
}
|
|
StringValue(str);
|
|
GetBN(self, bn);
|
|
if (RTEST(rb_obj_is_kind_of(str, cBN))) {
|
|
BIGNUM *other;
|
|
|
|
GetBN(str, other); /* Safe - we checked kind_of? above */
|
|
if (!BN_copy(bn, other)) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
return self;
|
|
}
|
|
|
|
switch (base) {
|
|
case 0:
|
|
if (!BN_mpi2bn((unsigned char *)RSTRING_PTR(str), RSTRING_LEN(str), bn)) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
break;
|
|
case 2:
|
|
if (!BN_bin2bn((unsigned char *)RSTRING_PTR(str), RSTRING_LEN(str), bn)) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
break;
|
|
case 10:
|
|
if (!BN_dec2bn(&bn, RSTRING_PTR(str))) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
break;
|
|
case 16:
|
|
if (!BN_hex2bn(&bn, RSTRING_PTR(str))) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
break;
|
|
default:
|
|
ossl_raise(rb_eArgError, "invalid radix %d", base);
|
|
}
|
|
return self;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* bn.to_s => string
|
|
* bn.to_s(base) => string
|
|
*
|
|
* === Parameters
|
|
* * +base+ - integer
|
|
* * * Valid values:
|
|
* * * * 0 - MPI
|
|
* * * * 2 - binary
|
|
* * * * 10 - the default
|
|
* * * * 16 - hex
|
|
*/
|
|
static VALUE
|
|
ossl_bn_to_s(int argc, VALUE *argv, VALUE self)
|
|
{
|
|
BIGNUM *bn;
|
|
VALUE str, bs;
|
|
int base = 10, len;
|
|
char *buf;
|
|
|
|
if (rb_scan_args(argc, argv, "01", &bs) == 1) {
|
|
base = NUM2INT(bs);
|
|
}
|
|
GetBN(self, bn);
|
|
switch (base) {
|
|
case 0:
|
|
len = BN_bn2mpi(bn, NULL);
|
|
str = rb_str_new(0, len);
|
|
if (BN_bn2mpi(bn, (unsigned char *)RSTRING_PTR(str)) != len)
|
|
ossl_raise(eBNError, NULL);
|
|
break;
|
|
case 2:
|
|
len = BN_num_bytes(bn);
|
|
str = rb_str_new(0, len);
|
|
if (BN_bn2bin(bn, (unsigned char *)RSTRING_PTR(str)) != len)
|
|
ossl_raise(eBNError, NULL);
|
|
break;
|
|
case 10:
|
|
if (!(buf = BN_bn2dec(bn))) ossl_raise(eBNError, NULL);
|
|
str = ossl_buf2str(buf, strlen(buf));
|
|
break;
|
|
case 16:
|
|
if (!(buf = BN_bn2hex(bn))) ossl_raise(eBNError, NULL);
|
|
str = ossl_buf2str(buf, strlen(buf));
|
|
break;
|
|
default:
|
|
ossl_raise(rb_eArgError, "invalid radix %d", base);
|
|
}
|
|
|
|
return str;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* bn.to_i => integer
|
|
*/
|
|
static VALUE
|
|
ossl_bn_to_i(VALUE self)
|
|
{
|
|
BIGNUM *bn;
|
|
char *txt;
|
|
VALUE num;
|
|
|
|
GetBN(self, bn);
|
|
|
|
if (!(txt = BN_bn2dec(bn))) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
num = rb_cstr_to_inum(txt, 10, Qtrue);
|
|
OPENSSL_free(txt);
|
|
|
|
return num;
|
|
}
|
|
|
|
static VALUE
|
|
ossl_bn_to_bn(VALUE self)
|
|
{
|
|
return self;
|
|
}
|
|
|
|
static VALUE
|
|
ossl_bn_coerce(VALUE self, VALUE other)
|
|
{
|
|
switch(TYPE(other)) {
|
|
case T_STRING:
|
|
self = ossl_bn_to_s(0, NULL, self);
|
|
break;
|
|
case T_FIXNUM:
|
|
case T_BIGNUM:
|
|
self = ossl_bn_to_i(self);
|
|
break;
|
|
default:
|
|
if (!RTEST(rb_obj_is_kind_of(other, cBN))) {
|
|
ossl_raise(rb_eTypeError, "Don't know how to coerce");
|
|
}
|
|
}
|
|
return rb_assoc_new(other, self);
|
|
}
|
|
|
|
#define BIGNUM_BOOL1(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func -> true | false \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_##func(VALUE self) \
|
|
{ \
|
|
BIGNUM *bn; \
|
|
GetBN(self, bn); \
|
|
if (BN_##func(bn)) { \
|
|
return Qtrue; \
|
|
} \
|
|
return Qfalse; \
|
|
}
|
|
BIGNUM_BOOL1(is_zero)
|
|
BIGNUM_BOOL1(is_one)
|
|
BIGNUM_BOOL1(is_odd)
|
|
|
|
#define BIGNUM_1c(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func -> aBN \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_##func(VALUE self) \
|
|
{ \
|
|
BIGNUM *bn, *result; \
|
|
VALUE obj; \
|
|
GetBN(self, bn); \
|
|
if (!(result = BN_new())) { \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
if (!BN_##func(result, bn, ossl_bn_ctx)) { \
|
|
BN_free(result); \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
WrapBN(CLASS_OF(self), obj, result); \
|
|
return obj; \
|
|
}
|
|
BIGNUM_1c(sqr)
|
|
|
|
#define BIGNUM_2(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func(bn2) -> aBN \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_##func(VALUE self, VALUE other) \
|
|
{ \
|
|
BIGNUM *bn1, *bn2 = GetBNPtr(other), *result; \
|
|
VALUE obj; \
|
|
GetBN(self, bn1); \
|
|
if (!(result = BN_new())) { \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
if (!BN_##func(result, bn1, bn2)) { \
|
|
BN_free(result); \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
WrapBN(CLASS_OF(self), obj, result); \
|
|
return obj; \
|
|
}
|
|
BIGNUM_2(add)
|
|
BIGNUM_2(sub)
|
|
|
|
#define BIGNUM_2c(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func(bn2) -> aBN \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_##func(VALUE self, VALUE other) \
|
|
{ \
|
|
BIGNUM *bn1, *bn2 = GetBNPtr(other), *result; \
|
|
VALUE obj; \
|
|
GetBN(self, bn1); \
|
|
if (!(result = BN_new())) { \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
if (!BN_##func(result, bn1, bn2, ossl_bn_ctx)) { \
|
|
BN_free(result); \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
WrapBN(CLASS_OF(self), obj, result); \
|
|
return obj; \
|
|
}
|
|
BIGNUM_2c(mul)
|
|
BIGNUM_2c(mod)
|
|
BIGNUM_2c(exp)
|
|
BIGNUM_2c(gcd)
|
|
BIGNUM_2c(mod_sqr)
|
|
BIGNUM_2c(mod_inverse)
|
|
|
|
/*
|
|
* call-seq:
|
|
* bn1 / bn2 => [result, remainder]
|
|
*/
|
|
static VALUE
|
|
ossl_bn_div(VALUE self, VALUE other)
|
|
{
|
|
BIGNUM *bn1, *bn2 = GetBNPtr(other), *r1, *r2;
|
|
VALUE obj1, obj2;
|
|
|
|
GetBN(self, bn1);
|
|
|
|
if (!(r1 = BN_new())) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
if (!(r2 = BN_new())) {
|
|
BN_free(r1);
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
if (!BN_div(r1, r2, bn1, bn2, ossl_bn_ctx)) {
|
|
BN_free(r1);
|
|
BN_free(r2);
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
WrapBN(CLASS_OF(self), obj1, r1);
|
|
WrapBN(CLASS_OF(self), obj2, r2);
|
|
|
|
return rb_ary_new3(2, obj1, obj2);
|
|
}
|
|
|
|
#define BIGNUM_3c(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func(bn1, bn2) -> aBN \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_##func(VALUE self, VALUE other1, VALUE other2) \
|
|
{ \
|
|
BIGNUM *bn1, *bn2 = GetBNPtr(other1); \
|
|
BIGNUM *bn3 = GetBNPtr(other2), *result; \
|
|
VALUE obj; \
|
|
GetBN(self, bn1); \
|
|
if (!(result = BN_new())) { \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
if (!BN_##func(result, bn1, bn2, bn3, ossl_bn_ctx)) { \
|
|
BN_free(result); \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
WrapBN(CLASS_OF(self), obj, result); \
|
|
return obj; \
|
|
}
|
|
BIGNUM_3c(mod_add)
|
|
BIGNUM_3c(mod_sub)
|
|
BIGNUM_3c(mod_mul)
|
|
BIGNUM_3c(mod_exp)
|
|
|
|
#define BIGNUM_BIT(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func(bit) -> self \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_##func(VALUE self, VALUE bit) \
|
|
{ \
|
|
BIGNUM *bn; \
|
|
GetBN(self, bn); \
|
|
if (!BN_##func(bn, NUM2INT(bit))) { \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
return self; \
|
|
}
|
|
BIGNUM_BIT(set_bit)
|
|
BIGNUM_BIT(clear_bit)
|
|
BIGNUM_BIT(mask_bits)
|
|
|
|
/*
|
|
* call-seq:
|
|
* bn.bit_set?(bit) => true | false
|
|
*/
|
|
static VALUE
|
|
ossl_bn_is_bit_set(VALUE self, VALUE bit)
|
|
{
|
|
int b;
|
|
BIGNUM *bn;
|
|
|
|
b = NUM2INT(bit);
|
|
GetBN(self, bn);
|
|
if (BN_is_bit_set(bn, b)) {
|
|
return Qtrue;
|
|
}
|
|
return Qfalse;
|
|
}
|
|
|
|
#define BIGNUM_SHIFT(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func(bits) -> aBN \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_##func(VALUE self, VALUE bits) \
|
|
{ \
|
|
BIGNUM *bn, *result; \
|
|
int b; \
|
|
VALUE obj; \
|
|
b = NUM2INT(bits); \
|
|
GetBN(self, bn); \
|
|
if (!(result = BN_new())) { \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
if (!BN_##func(result, bn, b)) { \
|
|
BN_free(result); \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
WrapBN(CLASS_OF(self), obj, result); \
|
|
return obj; \
|
|
}
|
|
BIGNUM_SHIFT(lshift)
|
|
BIGNUM_SHIFT(rshift)
|
|
|
|
#define BIGNUM_SELF_SHIFT(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func!(bits) -> self \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_self_##func(VALUE self, VALUE bits) \
|
|
{ \
|
|
BIGNUM *bn; \
|
|
int b; \
|
|
b = NUM2INT(bits); \
|
|
GetBN(self, bn); \
|
|
if (!BN_##func(bn, bn, b)) \
|
|
ossl_raise(eBNError, NULL); \
|
|
return self; \
|
|
}
|
|
BIGNUM_SELF_SHIFT(lshift)
|
|
BIGNUM_SELF_SHIFT(rshift)
|
|
|
|
#define BIGNUM_RAND(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* BN.##func(bits [, fill [, odd]]) -> aBN \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_s_##func(int argc, VALUE *argv, VALUE klass) \
|
|
{ \
|
|
BIGNUM *result; \
|
|
int bottom = 0, top = 0, b; \
|
|
VALUE bits, fill, odd, obj; \
|
|
\
|
|
switch (rb_scan_args(argc, argv, "12", &bits, &fill, &odd)) { \
|
|
case 3: \
|
|
bottom = (odd == Qtrue) ? 1 : 0; \
|
|
/* FALLTHROUGH */ \
|
|
case 2: \
|
|
top = NUM2INT(fill); \
|
|
} \
|
|
b = NUM2INT(bits); \
|
|
if (!(result = BN_new())) { \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
if (!BN_##func(result, b, top, bottom)) { \
|
|
BN_free(result); \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
WrapBN(klass, obj, result); \
|
|
return obj; \
|
|
}
|
|
BIGNUM_RAND(rand)
|
|
BIGNUM_RAND(pseudo_rand)
|
|
|
|
#define BIGNUM_RAND_RANGE(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* BN.##func(range) -> aBN \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_s_##func##_range(VALUE klass, VALUE range) \
|
|
{ \
|
|
BIGNUM *bn = GetBNPtr(range), *result; \
|
|
VALUE obj; \
|
|
if (!(result = BN_new())) { \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
if (!BN_##func##_range(result, bn)) { \
|
|
BN_free(result); \
|
|
ossl_raise(eBNError, NULL); \
|
|
} \
|
|
WrapBN(klass, obj, result); \
|
|
return obj; \
|
|
}
|
|
BIGNUM_RAND_RANGE(rand)
|
|
BIGNUM_RAND_RANGE(pseudo_rand)
|
|
|
|
/*
|
|
* call-seq:
|
|
* BN.generate_prime(bits, [, safe [, add [, rem]]]) => bn
|
|
*
|
|
* === Parameters
|
|
* * +bits+ - integer
|
|
* * +safe+ - boolean
|
|
* * +add+ - BN
|
|
* * +rem+ - BN
|
|
*/
|
|
static VALUE
|
|
ossl_bn_s_generate_prime(int argc, VALUE *argv, VALUE klass)
|
|
{
|
|
BIGNUM *add = NULL, *rem = NULL, *result;
|
|
int safe = 1, num;
|
|
VALUE vnum, vsafe, vadd, vrem, obj;
|
|
|
|
rb_scan_args(argc, argv, "13", &vnum, &vsafe, &vadd, &vrem);
|
|
|
|
num = NUM2INT(vnum);
|
|
|
|
if (vsafe == Qfalse) {
|
|
safe = 0;
|
|
}
|
|
if (!NIL_P(vadd)) {
|
|
add = GetBNPtr(vadd);
|
|
rem = NIL_P(vrem) ? NULL : GetBNPtr(vrem);
|
|
}
|
|
if (!(result = BN_new())) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
if (!BN_generate_prime(result, num, safe, add, rem, NULL, NULL)) {
|
|
BN_free(result);
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
WrapBN(klass, obj, result);
|
|
|
|
return obj;
|
|
}
|
|
|
|
#define BIGNUM_NUM(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func -> integer \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_##func(VALUE self) \
|
|
{ \
|
|
BIGNUM *bn; \
|
|
GetBN(self, bn); \
|
|
return INT2FIX(BN_##func(bn)); \
|
|
}
|
|
BIGNUM_NUM(num_bytes)
|
|
BIGNUM_NUM(num_bits)
|
|
|
|
static VALUE
|
|
ossl_bn_copy(VALUE self, VALUE other)
|
|
{
|
|
BIGNUM *bn1, *bn2;
|
|
|
|
rb_check_frozen(self);
|
|
|
|
if (self == other) return self;
|
|
|
|
GetBN(self, bn1);
|
|
bn2 = GetBNPtr(other);
|
|
|
|
if (!BN_copy(bn1, bn2)) {
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
return self;
|
|
}
|
|
|
|
#define BIGNUM_CMP(func) \
|
|
/* \
|
|
* call-seq: \
|
|
* bn.##func(bn2) -> integer \
|
|
* \
|
|
*/ \
|
|
static VALUE \
|
|
ossl_bn_##func(VALUE self, VALUE other) \
|
|
{ \
|
|
BIGNUM *bn1, *bn2 = GetBNPtr(other); \
|
|
GetBN(self, bn1); \
|
|
return INT2FIX(BN_##func(bn1, bn2)); \
|
|
}
|
|
BIGNUM_CMP(cmp)
|
|
BIGNUM_CMP(ucmp)
|
|
|
|
static VALUE
|
|
ossl_bn_eql(VALUE self, VALUE other)
|
|
{
|
|
if (ossl_bn_cmp(self, other) == INT2FIX(0)) {
|
|
return Qtrue;
|
|
}
|
|
return Qfalse;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* bn.prime? => true | false
|
|
* bn.prime?(checks) => true | false
|
|
*
|
|
* === Parameters
|
|
* * +checks+ - integer
|
|
*/
|
|
static VALUE
|
|
ossl_bn_is_prime(int argc, VALUE *argv, VALUE self)
|
|
{
|
|
BIGNUM *bn;
|
|
VALUE vchecks;
|
|
int checks = BN_prime_checks;
|
|
|
|
if (rb_scan_args(argc, argv, "01", &vchecks) == 1) {
|
|
checks = NUM2INT(vchecks);
|
|
}
|
|
GetBN(self, bn);
|
|
switch (BN_is_prime(bn, checks, NULL, ossl_bn_ctx, NULL)) {
|
|
case 1:
|
|
return Qtrue;
|
|
case 0:
|
|
return Qfalse;
|
|
default:
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
/* not reachable */
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* bn.prime_fasttest? => true | false
|
|
* bn.prime_fasttest?(checks) => true | false
|
|
* bn.prime_fasttest?(checks, trial_div) => true | false
|
|
*
|
|
* === Parameters
|
|
* * +checks+ - integer
|
|
* * +trial_div+ - boolean
|
|
*/
|
|
static VALUE
|
|
ossl_bn_is_prime_fasttest(int argc, VALUE *argv, VALUE self)
|
|
{
|
|
BIGNUM *bn;
|
|
VALUE vchecks, vtrivdiv;
|
|
int checks = BN_prime_checks, do_trial_division = 1;
|
|
|
|
rb_scan_args(argc, argv, "02", &vchecks, &vtrivdiv);
|
|
|
|
if (!NIL_P(vchecks)) {
|
|
checks = NUM2INT(vchecks);
|
|
}
|
|
GetBN(self, bn);
|
|
/* handle true/false */
|
|
if (vtrivdiv == Qfalse) {
|
|
do_trial_division = 0;
|
|
}
|
|
switch (BN_is_prime_fasttest(bn, checks, NULL, ossl_bn_ctx, NULL, do_trial_division)) {
|
|
case 1:
|
|
return Qtrue;
|
|
case 0:
|
|
return Qfalse;
|
|
default:
|
|
ossl_raise(eBNError, NULL);
|
|
}
|
|
/* not reachable */
|
|
return Qnil;
|
|
}
|
|
|
|
/*
|
|
* INIT
|
|
* (NOTE: ordering of methods is the same as in 'man bn')
|
|
*/
|
|
void
|
|
Init_ossl_bn()
|
|
{
|
|
#if 0 /* let rdoc know about mOSSL */
|
|
mOSSL = rb_define_module("OpenSSL");
|
|
#endif
|
|
|
|
if (!(ossl_bn_ctx = BN_CTX_new())) {
|
|
ossl_raise(rb_eRuntimeError, "Cannot init BN_CTX");
|
|
}
|
|
|
|
eBNError = rb_define_class_under(mOSSL, "BNError", eOSSLError);
|
|
|
|
cBN = rb_define_class_under(mOSSL, "BN", rb_cObject);
|
|
|
|
rb_define_alloc_func(cBN, ossl_bn_alloc);
|
|
rb_define_method(cBN, "initialize", ossl_bn_initialize, -1);
|
|
|
|
rb_define_copy_func(cBN, ossl_bn_copy);
|
|
rb_define_method(cBN, "copy", ossl_bn_copy, 1);
|
|
|
|
/* swap (=coerce?) */
|
|
|
|
rb_define_method(cBN, "num_bytes", ossl_bn_num_bytes, 0);
|
|
rb_define_method(cBN, "num_bits", ossl_bn_num_bits, 0);
|
|
/* num_bits_word */
|
|
|
|
rb_define_method(cBN, "+", ossl_bn_add, 1);
|
|
rb_define_method(cBN, "-", ossl_bn_sub, 1);
|
|
rb_define_method(cBN, "*", ossl_bn_mul, 1);
|
|
rb_define_method(cBN, "sqr", ossl_bn_sqr, 0);
|
|
rb_define_method(cBN, "/", ossl_bn_div, 1);
|
|
rb_define_method(cBN, "%", ossl_bn_mod, 1);
|
|
/* nnmod */
|
|
|
|
rb_define_method(cBN, "mod_add", ossl_bn_mod_add, 2);
|
|
rb_define_method(cBN, "mod_sub", ossl_bn_mod_sub, 2);
|
|
rb_define_method(cBN, "mod_mul", ossl_bn_mod_mul, 2);
|
|
rb_define_method(cBN, "mod_sqr", ossl_bn_mod_sqr, 1);
|
|
rb_define_method(cBN, "**", ossl_bn_exp, 1);
|
|
rb_define_method(cBN, "mod_exp", ossl_bn_mod_exp, 2);
|
|
rb_define_method(cBN, "gcd", ossl_bn_gcd, 1);
|
|
|
|
/* add_word
|
|
* sub_word
|
|
* mul_word
|
|
* div_word
|
|
* mod_word */
|
|
|
|
rb_define_method(cBN, "cmp", ossl_bn_cmp, 1);
|
|
rb_define_alias(cBN, "<=>", "cmp");
|
|
rb_define_method(cBN, "ucmp", ossl_bn_ucmp, 1);
|
|
rb_define_method(cBN, "eql?", ossl_bn_eql, 1);
|
|
rb_define_alias(cBN, "==", "eql?");
|
|
rb_define_alias(cBN, "===", "eql?");
|
|
rb_define_method(cBN, "zero?", ossl_bn_is_zero, 0);
|
|
rb_define_method(cBN, "one?", ossl_bn_is_one, 0);
|
|
/* is_word */
|
|
rb_define_method(cBN, "odd?", ossl_bn_is_odd, 0);
|
|
|
|
/* zero
|
|
* one
|
|
* value_one - DON'T IMPL.
|
|
* set_word
|
|
* get_word */
|
|
|
|
rb_define_singleton_method(cBN, "rand", ossl_bn_s_rand, -1);
|
|
rb_define_singleton_method(cBN, "pseudo_rand", ossl_bn_s_pseudo_rand, -1);
|
|
rb_define_singleton_method(cBN, "rand_range", ossl_bn_s_rand_range, 1);
|
|
rb_define_singleton_method(cBN, "pseudo_rand_range", ossl_bn_s_pseudo_rand_range, 1);
|
|
|
|
rb_define_singleton_method(cBN, "generate_prime", ossl_bn_s_generate_prime, -1);
|
|
rb_define_method(cBN, "prime?", ossl_bn_is_prime, -1);
|
|
|
|
rb_define_method(cBN, "set_bit!", ossl_bn_set_bit, 1);
|
|
rb_define_method(cBN, "clear_bit!", ossl_bn_clear_bit, 1);
|
|
rb_define_method(cBN, "bit_set?", ossl_bn_is_bit_set, 1);
|
|
rb_define_method(cBN, "mask_bits!", ossl_bn_mask_bits, 1);
|
|
rb_define_method(cBN, "<<", ossl_bn_lshift, 1);
|
|
rb_define_method(cBN, ">>", ossl_bn_rshift, 1);
|
|
rb_define_method(cBN, "lshift!", ossl_bn_self_lshift, 1);
|
|
rb_define_method(cBN, "rshift!", ossl_bn_self_rshift, 1);
|
|
/* lshift1 - DON'T IMPL. */
|
|
/* rshift1 - DON'T IMPL. */
|
|
|
|
/*
|
|
* bn2bin
|
|
* bin2bn
|
|
* bn2hex
|
|
* bn2dec
|
|
* hex2bn
|
|
* dec2bn - all these are implemented in ossl_bn_initialize, and ossl_bn_to_s
|
|
* print - NOT IMPL.
|
|
* print_fp - NOT IMPL.
|
|
* bn2mpi
|
|
* mpi2bn
|
|
*/
|
|
rb_define_method(cBN, "to_s", ossl_bn_to_s, -1);
|
|
rb_define_method(cBN, "to_i", ossl_bn_to_i, 0);
|
|
rb_define_alias(cBN, "to_int", "to_i");
|
|
rb_define_method(cBN, "to_bn", ossl_bn_to_bn, 0);
|
|
rb_define_method(cBN, "coerce", ossl_bn_coerce, 1);
|
|
|
|
/*
|
|
* TODO:
|
|
* But how to: from_bin, from_mpi? PACK?
|
|
* to_bin
|
|
* to_mpi
|
|
*/
|
|
|
|
rb_define_method(cBN, "mod_inverse", ossl_bn_mod_inverse, 1);
|
|
|
|
/* RECiProcal
|
|
* MONTgomery */
|
|
|
|
/*
|
|
* TODO:
|
|
* Where to belong these?
|
|
*/
|
|
rb_define_method(cBN, "prime_fasttest?", ossl_bn_is_prime_fasttest, -1);
|
|
}
|
|
|