112 lines
3.0 KiB
C
112 lines
3.0 KiB
C
#include <ruby/ruby.h>
|
|
#include <ruby/encoding.h>
|
|
#include "blake2.h"
|
|
|
|
typedef struct {
|
|
size_t key_length;
|
|
uint8_t *key_bytes;
|
|
|
|
size_t output_length;
|
|
uint8_t *output;
|
|
|
|
VALUE to_hex;
|
|
VALUE to_bytes;
|
|
} Blake2;
|
|
|
|
VALUE mDigest;
|
|
VALUE mDigest_cBlake2;
|
|
|
|
static void blake2_free(Blake2 *blake2) {
|
|
free(blake2->key_bytes);
|
|
free(blake2->output);
|
|
|
|
rb_gc_mark(blake2->to_hex);
|
|
rb_gc_mark(blake2->to_bytes);
|
|
|
|
ruby_xfree(blake2);
|
|
}
|
|
|
|
static VALUE blake2_alloc(VALUE klass) {
|
|
Blake2 *blake2_obj = (Blake2 *)ruby_xmalloc(sizeof(Blake2));
|
|
|
|
return Data_Wrap_Struct(klass, NULL, blake2_free, blake2_obj);
|
|
}
|
|
|
|
VALUE m_blake2_initialize(VALUE self, VALUE _len, VALUE _key) {
|
|
Blake2 *blake2;
|
|
Data_Get_Struct(self, Blake2, blake2);
|
|
unsigned int i;
|
|
|
|
ID bytes_method = rb_intern("bytes");
|
|
blake2->to_hex = ID2SYM(rb_intern("to_hex"));
|
|
blake2->to_bytes = ID2SYM(rb_intern("to_bytes"));
|
|
|
|
VALUE key_bytes_ary = rb_funcall(_key, bytes_method, 0);
|
|
blake2->key_length = RARRAY_LEN(key_bytes_ary);
|
|
blake2->key_bytes = (uint8_t*)malloc(blake2->key_length * sizeof(uint8_t));
|
|
|
|
for(i = 0; i < blake2->key_length; i++) {
|
|
VALUE byte = rb_ary_entry(key_bytes_ary, i);
|
|
blake2->key_bytes[i] = NUM2INT(byte);
|
|
}
|
|
|
|
blake2->output_length = NUM2INT(_len);
|
|
blake2->output = (uint8_t*)malloc(blake2->output_length * sizeof(uint8_t));
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
|
|
VALUE m_blake2_digest(VALUE self, VALUE _input, VALUE _representation) {
|
|
Blake2 *blake2;
|
|
|
|
char *input = RSTRING_PTR(_input);
|
|
uint64_t input_length = RSTRING_LEN(_input);
|
|
unsigned int i;
|
|
|
|
Data_Get_Struct(self, Blake2, blake2);
|
|
|
|
blake2b(blake2->output, blake2->output_length, input, input_length,
|
|
blake2->key_bytes, blake2->key_length);
|
|
|
|
VALUE result;
|
|
|
|
if(_representation == blake2->to_bytes) {
|
|
result = rb_ary_new2(blake2->output_length);
|
|
|
|
for(i = 0; i < blake2->output_length; i++) {
|
|
rb_ary_push(result, INT2NUM(blake2->output[i]));
|
|
}
|
|
} else if(_representation == blake2->to_hex) {
|
|
unsigned long ary_len = blake2->output_length * (unsigned)sizeof(char) * 2;
|
|
char *c_str = (char*)malloc(ary_len + 1);
|
|
|
|
for(i = 0; i < blake2->output_length; i++) {
|
|
sprintf(c_str + (i * 2), "%02x", blake2->output[i]);
|
|
}
|
|
c_str[ary_len] = 0;
|
|
|
|
result = rb_str_new(c_str, ary_len);
|
|
|
|
if((unsigned long)RSTRING_LEN(result) != ary_len) {
|
|
rb_raise(rb_eRuntimeError, "m_blake2_digest: sizes don't match. Ary: %lu != %lu", RSTRING_LEN(result), ary_len);
|
|
}
|
|
|
|
free(c_str);
|
|
} else {
|
|
rb_raise(rb_eArgError, "unknown representation :%"PRIsVALUE"", _representation);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
void Init_blake2b_ext() {
|
|
mDigest = rb_define_module("Digest");
|
|
|
|
mDigest_cBlake2 = rb_define_class_under(mDigest, "Blake2b", rb_cObject);
|
|
rb_define_alloc_func(mDigest_cBlake2, blake2_alloc);
|
|
|
|
rb_define_private_method(mDigest_cBlake2, "initialize", RUBY_METHOD_FUNC(m_blake2_initialize), 2);
|
|
rb_define_method(mDigest_cBlake2, "digest", RUBY_METHOD_FUNC(m_blake2_digest), 2);
|
|
}
|