msgpack 0.4.0 → 0.4.1

data/ext/unpack.c

@@ -156,15 +156,27 @@ static inline int template_callback_raw(unpack_user* u, const char* b, const cha
 		rb_raise(rb_eTypeError, "instance of String needed"); \
 	}
 
+#ifdef RUBY_VM
+#define RERAISE rb_exc_raise(rb_errinfo())
+#else
+#define RERAISE rb_exc_raise(ruby_errinfo)
+#endif
 
-static VALUE template_execute_rescue(VALUE nouse)
+
+#ifdef HAVE_RUBY_ENCODING_H
+static VALUE template_execute_rescue_enc(VALUE data)
 {
 	rb_gc_enable();
-#ifdef RUBY_VM
-	rb_exc_raise(rb_errinfo());
-#else
-	rb_exc_raise(ruby_errinfo);
+	VALUE* resc = (VALUE*)data;
+	rb_enc_set_index(resc[0], (int)resc[1]);
+	RERAISE;
+}
 #endif
+
+static VALUE template_execute_rescue(VALUE nouse)
+{
+	rb_gc_enable();
+	RERAISE;
 }
 
 static VALUE template_execute_do(VALUE argv)
@@ -192,7 +204,6 @@ static int template_execute_wrap(msgpack_unpack_t* mp,
 	};
 
 #ifdef HAVE_RUBY_ENCODING_H
-	// FIXME encodingをASCII-8BITにする
 	int enc_orig = rb_enc_get_index(str);
 	rb_enc_set_index(str, s_ascii_8bit);
 #endif
@@ -202,10 +213,14 @@ static int template_execute_wrap(msgpack_unpack_t* mp,
 
 	mp->user.source = str;
 
+#ifdef HAVE_RUBY_ENCODING_H
+	VALUE resc[2] = {str, enc_orig};
+	int ret = (int)rb_rescue(template_execute_do, (VALUE)args,
+			template_execute_rescue_enc, (VALUE)resc);
+#else
 	int ret = (int)rb_rescue(template_execute_do, (VALUE)args,
 			template_execute_rescue, Qnil);
-
-	mp->user.source = Qnil;
+#endif
 
 	rb_gc_enable();
 
@@ -229,6 +244,8 @@ static int template_execute_wrap_each(msgpack_unpack_t* mp,
 	// FIXME execute実行中はmp->topが更新されないのでGC markが機能しない
 	rb_gc_disable();
 
+	mp->user.source = Qnil;
+
 	int ret = (int)rb_rescue(template_execute_do, (VALUE)args,
 			template_execute_rescue, Qnil);
 
@@ -401,8 +418,7 @@ static void feed_buffer(msgpack_unpack_t* mp, const char* ptr, size_t len)
 	struct unpack_buffer* buffer = &mp->user.buffer;
 
 	if(buffer->size == 0) {
-		char* tmp = (char*)malloc(MSGPACK_UNPACKER_BUFFER_INIT_SIZE);
-		// FIXME check tmp == NULL
+		char* tmp = ALLOC_N(char, MSGPACK_UNPACKER_BUFFER_INIT_SIZE);
 		buffer->ptr = tmp;
 		buffer->free = MSGPACK_UNPACKER_BUFFER_INIT_SIZE;
 		buffer->size = 0;
@@ -430,8 +446,7 @@ static void feed_buffer(msgpack_unpack_t* mp, const char* ptr, size_t len)
 		while(csize < buffer->size + len) {
 			csize *= 2;
 		}
-		char* tmp = (char*)realloc(buffer->ptr, csize);
-		// FIXME check tmp == NULL
+		char* tmp = REALLOC_N(buffer->ptr, char, csize);
 		memcpy(tmp + buffer->size, ptr, len);
 		buffer->ptr = tmp;
 		buffer->free = csize - buffer->size;
@@ -446,8 +461,7 @@ static void feed_buffer(msgpack_unpack_t* mp, const char* ptr, size_t len)
 		while(csize < not_parsed + len) {
 			csize *= 2;
 		}
-		char* tmp = (char*)realloc(buffer->ptr, csize);
-		// FIXME check tmp == NULL
+		char* tmp = REALLOC_N(buffer->ptr, char, csize);
 		buffer->ptr = tmp;
 	}
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: msgpack
 version: !ruby/object:Gem::Version 
-  version: 0.4.0
+  version: 0.4.1
 platform: ruby
 authors: 
 - FURUHASHI Sadayuki
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-05-26 00:00:00 +09:00
+date: 2010-05-27 00:00:00 +09:00
 default_executable: 
 dependencies: []