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
|
#include <ruby.h>
#include <ow-crypt.h>
static VALUE mBCrypt;
static VALUE cBCryptEngine;
#ifdef RUBY_VM
# define RUBY_1_9
#endif
#ifdef RUBY_1_9
/* When on Ruby 1.9+, we will want to unlock the GIL while performing
* expensive calculations, for greater concurrency. Do not do this for
* cheap calculations because locking/unlocking the GIL incurs some overhead as well.
*/
#define GIL_UNLOCK_COST_THRESHOLD 9
typedef struct {
char *output;
const char *key;
const char *salt;
} BCryptArguments;
static VALUE bcrypt_wrapper(void *_args) {
BCryptArguments *args = (BCryptArguments *)_args;
return (VALUE)ruby_bcrypt(args->output, args->key, args->salt);
}
#endif /* RUBY_1_9 */
/* Given a logarithmic cost parameter, generates a salt for use with +bc_crypt+.
*/
static VALUE bc_salt(VALUE self, VALUE prefix, VALUE count, VALUE input) {
char * salt;
VALUE str_salt;
salt = crypt_gensalt_ra(
StringValuePtr(prefix),
NUM2ULONG(count),
NIL_P(input) ? NULL : StringValuePtr(input),
NIL_P(input) ? 0 : RSTRING_LEN(input));
if(!salt) return Qnil;
str_salt = rb_str_new2(salt);
xfree(salt);
return str_salt;
}
/* Given a secret and a salt, generates a salted hash (which you can then store safely).
*/
static VALUE bc_crypt(VALUE self, VALUE key, VALUE setting) {
char * value;
void * data;
int size;
VALUE out;
data = NULL;
size = 0xDEADBEEF;
if(NIL_P(key) || NIL_P(setting)) return Qnil;
value = crypt_ra(
NIL_P(key) ? NULL : StringValuePtr(key),
NIL_P(setting) ? NULL : StringValuePtr(setting),
&data,
&size);
if(!value) return Qnil;
out = rb_str_new(data, size - 1);
xfree(data);
return out;
}
/* Create the BCrypt and BCrypt::Engine modules, and populate them with methods. */
void Init_bcrypt_ext(){
mBCrypt = rb_define_module("BCrypt");
cBCryptEngine = rb_define_class_under(mBCrypt, "Engine", rb_cObject);
rb_define_singleton_method(cBCryptEngine, "__bc_salt", bc_salt, 3);
rb_define_singleton_method(cBCryptEngine, "__bc_crypt", bc_crypt, 2);
}
/* vim: set noet sws=4 sw=4: */
|
