1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00

* ext/digest: Merge from trunk; apply all changes since the

initial import, except for the removal of compatibility stub
  libraries (md5.rb and sha1.rb).


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/branches/ruby_1_8@11117 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
knu 2006-10-10 05:01:33 +00:00
parent 6d83e40c32
commit 9b2fa26be4
33 changed files with 298 additions and 751 deletions

View file

@ -6,49 +6,30 @@
created at: Fri May 25 08:57:27 JST 2001
Copyright (C) 1995-2001 Yukihiro Matsumoto
Copyright (C) 2001 Akinori MUSHA
Copyright (C) 2001-2006 Akinori MUSHA
$RoughId: digest.c,v 1.16 2001/07/13 15:38:27 knu Exp $
$Id$
************************************************/
/*
* This module provides an interface to the following hash algorithms:
*
* - the MD5 Message-Digest Algorithm by the RSA Data Security,
* Inc., described in RFC 1321
*
* - the SHA-1 Secure Hash Algorithm by NIST (the US' National
* Institute of Standards and Technology), described in FIPS PUB
* 180-1.
*
* - the SHA-256/384/512 Secure Hash Algorithm by NIST (the US'
* National Institute of Standards and Technology), described in
* FIPS PUB 180-2.
*
* - the RIPEMD-160 cryptographic hash function, designed by Hans
* Dobbertin, Antoon Bosselaers, and Bart Preneel.
*/
#include "digest.h"
static VALUE mDigest, cDigest_Base;
static ID id_metadata;
static ID id_metadata, id_new, id_update, id_digest;
/*
* Digest::Base
*/
static algo_t *
get_digest_base_metadata(klass)
VALUE klass;
get_digest_base_metadata(VALUE klass)
{
VALUE obj;
algo_t *algo;
if (rb_cvar_defined(klass, id_metadata) == Qfalse) {
rb_notimplement();
return NULL;
}
obj = rb_cvar_get(klass, id_metadata);
@ -58,10 +39,95 @@ get_digest_base_metadata(klass)
return algo;
}
static VALUE rb_digest_base_alloc _((VALUE));
static VALUE
rb_digest_base_alloc(klass)
VALUE klass;
hexdigest_str_new(VALUE str_digest)
{
char *digest = RSTRING_PTR(str_digest);
size_t digest_len = RSTRING_LEN(str_digest);
int i;
VALUE str;
char *p;
static const char hex[] = {
'0', '1', '2', '3', '4', '5', '6', '7', '8', '9',
'a', 'b', 'c', 'd', 'e', 'f'
};
if (LONG_MAX / 2 < digest_len) {
rb_raise(rb_eRuntimeError, "digest string too long");
}
str = rb_str_new(0, digest_len * 2);
for (i = 0, p = RSTRING_PTR(str); i < digest_len; i++) {
unsigned char byte = digest[i];
p[i + i] = hex[byte >> 4];
p[i + i + 1] = hex[byte & 0x0f];
}
return str;
}
static VALUE
bubblebabble_str_new(VALUE str_digest)
{
char *digest = RSTRING_PTR(str_digest);
size_t digest_len = RSTRING_LEN(str_digest);
VALUE str;
char *p;
int i, j, seed = 1;
static const char vowels[] = {
'a', 'e', 'i', 'o', 'u', 'y'
};
static const char consonants[] = {
'b', 'c', 'd', 'f', 'g', 'h', 'k', 'l', 'm', 'n',
'p', 'r', 's', 't', 'v', 'z', 'x'
};
if ((LONG_MAX - 2) / 3 < (digest_len | 1)) {
rb_raise(rb_eRuntimeError, "digest string too long");
}
str = rb_str_new(0, (digest_len | 1) * 3 + 2);
p = RSTRING_PTR(str);
i = j = 0;
p[j++] = 'x';
for (;;) {
unsigned char byte1, byte2;
if (i >= digest_len) {
p[j++] = vowels[seed % 6];
p[j++] = consonants[16];
p[j++] = vowels[seed / 6];
break;
}
byte1 = digest[i++];
p[j++] = vowels[(((byte1 >> 6) & 3) + seed) % 6];
p[j++] = consonants[(byte1 >> 2) & 15];
p[j++] = vowels[((byte1 & 3) + (seed / 6)) % 6];
if (i >= digest_len) {
break;
}
byte2 = digest[i++];
p[j++] = consonants[(byte2 >> 4) & 15];
p[j++] = '-';
p[j++] = consonants[byte2 & 15];
seed = (seed * 5 + byte1 * 7 + byte2) % 36;
}
p[j] = 'x';
return str;
}
static VALUE
rb_digest_base_alloc(VALUE klass)
{
algo_t *algo;
VALUE obj;
@ -73,7 +139,11 @@ rb_digest_base_alloc(klass)
algo = get_digest_base_metadata(klass);
/* XXX: An uninitialized buffer leads ALGO_Equal() to fail */
if (algo == NULL) {
return Data_Wrap_Struct(klass, 0, free, 0);
}
/* XXX: An uninitialized buffer may lead ALGO_Equal() to fail */
pctx = xcalloc(algo->ctx_size, 1);
algo->init_func(pctx);
@ -83,66 +153,44 @@ rb_digest_base_alloc(klass)
}
static VALUE
rb_digest_base_s_digest(klass, str)
VALUE klass;
VALUE str;
rb_digest_base_s_digest(VALUE klass, VALUE str)
{
algo_t *algo;
algo_t *algo = get_digest_base_metadata(klass);
void *pctx;
size_t len;
unsigned char *digest;
volatile VALUE obj = rb_digest_base_alloc(klass);
volatile VALUE obj;
algo = get_digest_base_metadata(klass);
if (algo == NULL) {
VALUE obj = rb_funcall(klass, id_new, 0);
rb_funcall(obj, id_update, 1, str);
return rb_funcall(obj, id_digest, 0);
}
obj = rb_digest_base_alloc(klass);
Data_Get_Struct(obj, void, pctx);
StringValue(str);
algo->update_func(pctx, RSTRING(str)->ptr, RSTRING(str)->len);
len = algo->digest_len;
str = rb_str_new(0, algo->digest_len);
algo->finish_func(pctx, RSTRING(str)->ptr);
digest = xmalloc(len);
algo->final_func(digest, pctx);
obj = rb_str_new(digest, len);
free(digest);
return obj;
return str;
}
static VALUE
rb_digest_base_s_hexdigest(klass, str)
VALUE klass;
VALUE str;
rb_digest_base_s_hexdigest(VALUE klass, VALUE str)
{
algo_t *algo;
void *pctx;
size_t len;
unsigned char *hexdigest;
volatile VALUE obj = rb_digest_base_alloc(klass);
algo = get_digest_base_metadata(klass);
Data_Get_Struct(obj, void, pctx);
StringValue(str);
algo->update_func(pctx, RSTRING(str)->ptr, RSTRING(str)->len);
len = algo->digest_len * 2;
hexdigest = xmalloc(len + 1); /* +1 is for '\0' */
algo->end_func(pctx, hexdigest);
obj = rb_str_new(hexdigest, len);
free(hexdigest);
return obj;
return hexdigest_str_new(rb_funcall(klass, id_digest, 1, str));
}
static VALUE
rb_digest_base_copy(copy, obj)
VALUE copy, obj;
rb_digest_base_s_bubblebabble(VALUE klass, VALUE str)
{
return bubblebabble_str_new(rb_funcall(klass, id_digest, 1, str));
}
static VALUE
rb_digest_base_copy(VALUE copy, VALUE obj)
{
algo_t *algo;
void *pctx1, *pctx2;
@ -150,6 +198,11 @@ rb_digest_base_copy(copy, obj)
if (copy == obj) return copy;
rb_check_frozen(copy);
algo = get_digest_base_metadata(rb_obj_class(copy));
if (algo == NULL)
rb_notimplement();
/* get_digest_base_metadata() may return a NULL */
if (algo != get_digest_base_metadata(rb_obj_class(obj))) {
rb_raise(rb_eTypeError, "wrong argument class");
}
@ -161,26 +214,51 @@ rb_digest_base_copy(copy, obj)
}
static VALUE
rb_digest_base_update(self, str)
VALUE self, str;
rb_digest_base_update(VALUE self, VALUE str)
{
algo_t *algo;
void *pctx;
StringValue(str);
algo = get_digest_base_metadata(rb_obj_class(self));
if (algo == NULL) {
/* subclasses must define update() */
rb_notimplement();
}
Data_Get_Struct(self, void, pctx);
StringValue(str);
algo->update_func(pctx, RSTRING(str)->ptr, RSTRING(str)->len);
return self;
}
static VALUE
rb_digest_base_init(argc, argv, self)
int argc;
VALUE* argv;
VALUE self;
rb_digest_base_lshift(VALUE self, VALUE str)
{
algo_t *algo;
void *pctx;
algo = get_digest_base_metadata(rb_obj_class(self));
if (algo == NULL) {
/* subclasses just need to define update(), not << */
rb_funcall(self, id_update, 1, str);
return self;
}
Data_Get_Struct(self, void, pctx);
StringValue(str);
algo->update_func(pctx, RSTRING(str)->ptr, RSTRING(str)->len);
return self;
}
static VALUE
rb_digest_base_init(int argc, VALUE *argv, VALUE self)
{
VALUE arg;
@ -192,70 +270,71 @@ rb_digest_base_init(argc, argv, self)
}
static VALUE
rb_digest_base_digest(self)
VALUE self;
rb_digest_base_digest(VALUE self)
{
algo_t *algo;
void *pctx1, *pctx2;
unsigned char *digest;
size_t len;
size_t ctx_size;
VALUE str;
algo = get_digest_base_metadata(rb_obj_class(self));
if (algo == NULL)
rb_notimplement();
Data_Get_Struct(self, void, pctx1);
len = algo->ctx_size;
ctx_size = algo->ctx_size;
pctx2 = xmalloc(ctx_size);
memcpy(pctx2, pctx1, ctx_size);
pctx2 = xmalloc(len);
memcpy(pctx2, pctx1, len);
len = algo->digest_len;
digest = xmalloc(len);
algo->final_func(digest, pctx2);
str = rb_str_new(digest, len);
free(digest);
str = rb_str_new(0, algo->digest_len);
algo->finish_func(pctx2, RSTRING(str)->ptr);
free(pctx2);
return str;
}
static VALUE
rb_digest_base_hexdigest(self)
VALUE self;
rb_digest_base_hexdigest(VALUE self)
{
return hexdigest_str_new(rb_funcall(self, id_digest, 0));
}
static VALUE
rb_digest_base_bubblebabble(VALUE self)
{
return bubblebabble_str_new(rb_funcall(self, id_digest, 0));
}
static VALUE
rb_digest_base_inspect(VALUE self)
{
algo_t *algo;
void *pctx1, *pctx2;
unsigned char *hexdigest;
size_t len;
VALUE str;
VALUE klass, str;
size_t digest_len = 32; /* no need to be just the right size */
char *cname;
algo = get_digest_base_metadata(rb_obj_class(self));
Data_Get_Struct(self, void, pctx1);
klass = rb_obj_class(self);
algo = get_digest_base_metadata(klass);
len = algo->ctx_size;
if (algo != NULL)
digest_len = algo->digest_len;
pctx2 = xmalloc(len);
memcpy(pctx2, pctx1, len);
len = algo->digest_len * 2;
hexdigest = xmalloc(len + 1); /* +1 is for '\0' */
algo->end_func(pctx2, hexdigest);
str = rb_str_new(hexdigest, len);
free(hexdigest);
free(pctx2);
cname = rb_obj_classname(self);
/* #<Digest::Alg: xxxxx...xxxx> */
str = rb_str_buf_new(2 + strlen(cname) + 2 + digest_len * 2 + 1);
rb_str_buf_cat2(str, "#<");
rb_str_buf_cat2(str, cname);
rb_str_buf_cat2(str, ": ");
rb_str_buf_append(str, rb_digest_base_hexdigest(self));
rb_str_buf_cat2(str, ">");
return str;
}
static VALUE
rb_digest_base_equal(self, other)
VALUE self, other;
rb_digest_base_equal(VALUE self, VALUE other)
{
algo_t *algo;
VALUE klass;
@ -289,11 +368,25 @@ rb_digest_base_equal(self, other)
}
/*
* Init
* This module provides an interface to the following hash algorithms:
*
* - the MD5 Message-Digest Algorithm by the RSA Data Security,
* Inc., described in RFC 1321
*
* - the SHA-1 Secure Hash Algorithm by NIST (the US' National
* Institute of Standards and Technology), described in FIPS PUB
* 180-1.
*
* - the SHA-256/384/512 Secure Hash Algorithm by NIST (the US'
* National Institute of Standards and Technology), described in
* FIPS PUB 180-2.
*
* - the RIPEMD-160 cryptographic hash function, designed by Hans
* Dobbertin, Antoon Bosselaers, and Bart Preneel.
*/
void
Init_digest()
Init_digest(void)
{
mDigest = rb_define_module("Digest");
@ -302,15 +395,21 @@ Init_digest()
rb_define_alloc_func(cDigest_Base, rb_digest_base_alloc);
rb_define_singleton_method(cDigest_Base, "digest", rb_digest_base_s_digest, 1);
rb_define_singleton_method(cDigest_Base, "hexdigest", rb_digest_base_s_hexdigest, 1);
rb_define_singleton_method(cDigest_Base, "bubblebabble", rb_digest_base_s_bubblebabble, 1);
rb_define_method(cDigest_Base, "initialize", rb_digest_base_init, -1);
rb_define_method(cDigest_Base, "initialize_copy", rb_digest_base_copy, 1);
rb_define_method(cDigest_Base, "update", rb_digest_base_update, 1);
rb_define_method(cDigest_Base, "<<", rb_digest_base_update, 1);
rb_define_method(cDigest_Base, "<<", rb_digest_base_lshift, 1);
rb_define_method(cDigest_Base, "digest", rb_digest_base_digest, 0);
rb_define_method(cDigest_Base, "hexdigest", rb_digest_base_hexdigest, 0);
rb_define_method(cDigest_Base, "bubblebabble", rb_digest_base_bubblebabble, 0);
rb_define_method(cDigest_Base, "to_s", rb_digest_base_hexdigest, 0);
rb_define_method(cDigest_Base, "inspect", rb_digest_base_inspect, 0);
rb_define_method(cDigest_Base, "==", rb_digest_base_equal, 1);
id_metadata = rb_intern("metadata");
id_new = rb_intern("new");
id_update = rb_intern("update");
id_digest = rb_intern("digest");
}