1berry/ruby
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[ruby/openssl] make bn shareable when frozen

Browse Source 
https://github.com/ruby/openssl/commit/d3c8e661e8
This commit is contained in:
HoneyryderChuck 2024-10-19 17:09:51 +01:00 committed by git
parent 0f28be6ab9
commit 2335768437
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ext/openssl/ossl_bn.c
View File

@ -37,7 +37,7 @@ static const rb_data_type_t ossl_bn_type = {
{ {
0, ossl_bn_free, 0, ossl_bn_free,
}, },
0, 0, RUBY_TYPED_FREE_IMMEDIATELY | RUBY_TYPED_WB_PROTECTED, 0, 0, RUBY_TYPED_FREE_IMMEDIATELY | RUBY_TYPED_WB_PROTECTED | RUBY_TYPED_FROZEN_SHAREABLE,
}; };
/* /*
@ -259,6 +259,7 @@ ossl_bn_initialize(int argc, VALUE *argv, VALUE self)
ossl_raise(rb_eArgError, "invalid argument"); ossl_raise(rb_eArgError, "invalid argument");
} }
rb_check_frozen(self);
if (RB_INTEGER_TYPE_P(str)) { if (RB_INTEGER_TYPE_P(str)) {
GetBN(self, bn); GetBN(self, bn);
integer_to_bnptr(str, bn); integer_to_bnptr(str, bn);
@ -689,6 +690,7 @@ BIGNUM_3c(mod_exp)
ossl_bn_##func(VALUE self, VALUE bit) \ ossl_bn_##func(VALUE self, VALUE bit) \
{ \ { \
BIGNUM *bn; \ BIGNUM *bn; \
rb_check_frozen(self); \
GetBN(self, bn); \ GetBN(self, bn); \
if (BN_##func(bn, NUM2INT(bit)) <= 0) { \ if (BN_##func(bn, NUM2INT(bit)) <= 0) { \
ossl_raise(eBNError, NULL); \ ossl_raise(eBNError, NULL); \
@ -778,6 +780,7 @@ BIGNUM_SHIFT(rshift)
{ \ { \
BIGNUM *bn; \ BIGNUM *bn; \
int b; \ int b; \
rb_check_frozen(self); \
b = NUM2INT(bits); \ b = NUM2INT(bits); \
GetBN(self, bn); \ GetBN(self, bn); \
if (BN_##func(bn, bn, b) <= 0) \ if (BN_##func(bn, bn, b) <= 0) \
@ -1187,6 +1190,7 @@ ossl_bn_set_flags(VALUE self, VALUE arg)
BIGNUM *bn; BIGNUM *bn;
GetBN(self, bn); GetBN(self, bn);
rb_check_frozen(self);
BN_set_flags(bn, NUM2INT(arg)); BN_set_flags(bn, NUM2INT(arg));
return Qnil; return Qnil;
} }

2
test/openssl/test_bn.rb
View File

@ -365,6 +365,8 @@ class OpenSSL::TestBN < OpenSSL::TestCase
assert_include(128..255, Ractor.new { OpenSSL::BN.rand(8)}.take) assert_include(128..255, Ractor.new { OpenSSL::BN.rand(8)}.take)
assert_include(0...2**32, Ractor.new { OpenSSL::BN.generate_prime(32) }.take) assert_include(0...2**32, Ractor.new { OpenSSL::BN.generate_prime(32) }.take)
assert_equal(0, Ractor.new { OpenSSL::BN.new(999).get_flags(OpenSSL::BN::CONSTTIME) }.take) assert_equal(0, Ractor.new { OpenSSL::BN.new(999).get_flags(OpenSSL::BN::CONSTTIME) }.take)
# test if shareable when frozen
assert Ractor.shareable?(@e1.freeze)
end end
end end
end end