From 7644ea4fe97bffd1f94b30f1c5ffb3a6240c1a35 Mon Sep 17 00:00:00 2001
From: "Hongli Lai (Phusion)" <hongli@phusion.nl>
Date: Thu, 4 Oct 2012 18:33:35 +0200
Subject: [PATCH] Make 512 bits the default

---
 ext/digest/sha3.c | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/ext/digest/sha3.c b/ext/digest/sha3.c
index 6d03c65..9372a3a 100644
--- a/ext/digest/sha3.c
+++ b/ext/digest/sha3.c
@@ -11,19 +11,27 @@ typedef struct {
 } RbSHA3;
 
 static VALUE
-rb_sha3_new(VALUE klass, VALUE bitlen) {
+rb_sha3_new(int argc, VALUE *argv, VALUE klass) {
 	RbSHA3 *ctx;
 	VALUE obj;
+	VALUE hashlen;
+	int i_hashlen;
+
+	if (rb_scan_args(argc, argv, "01", &hashlen) == 0) {
+		i_hashlen = 512;
+	} else {
+		i_hashlen = NUM2INT(hashlen);
+	}
 
 	ctx = (RbSHA3 *) xmalloc(sizeof(RbSHA3));
 	obj = Data_Wrap_Struct(klass, 0, xfree, ctx);
-	ctx->bitlen = NUM2INT(bitlen);
+	ctx->bitlen = i_hashlen;
 
 	if (ctx->bitlen == 0) {
 		rb_raise(rb_eRuntimeError, "Unsupported hash length");
 	}
 
-	switch (Init(&ctx->state, ctx->bitlen)) {
+	switch (Init(&ctx->state, i_hashlen)) {
 	case SUCCESS:
 		return obj;
 	case FAIL:
@@ -115,7 +123,7 @@ void
 Init_sha3() {
 	mDigest = rb_define_module("Digest");
 	cSHA3 = rb_define_class_under(mDigest, "SHA3", rb_cObject);
-	rb_define_singleton_method(cSHA3, "new", rb_sha3_new, 1);
+	rb_define_singleton_method(cSHA3, "new", rb_sha3_new, -1);
 	rb_define_method(cSHA3, "initialize_copy",  rb_sha3_copy, 1);
 	rb_define_method(cSHA3, "reset",  rb_sha3_reset, 0);
 	rb_define_method(cSHA3, "update",  rb_sha3_update, 1);