oj 2.18.5 → 3.16.11

data/ext/oj/compat.c

@@ -1,155 +1,239 @@
-/* compat.c
- * Copyright (c) 2012, Peter Ohler
- * All rights reserved.
- * 
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are met:
- * 
- *  - Redistributions of source code must retain the above copyright notice, this
- *    list of conditions and the following disclaimer.
- * 
- *  - Redistributions in binary form must reproduce the above copyright notice,
- *    this list of conditions and the following disclaimer in the documentation
- *    and/or other materials provided with the distribution.
- * 
- *  - Neither the name of Peter Ohler nor the names of its contributors may be
- *    used to endorse or promote products derived from this software without
- *    specific prior written permission.
- * 
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
- * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
- * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
- * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
+// Copyright (c) 2012 Peter Ohler. All rights reserved.
+// Licensed under the MIT License. See LICENSE file in the project root for license details.
 
 #include <stdio.h>
 
-#include "oj.h"
+#include "encode.h"
 #include "err.h"
+#include "intern.h"
+#include "mem.h"
+#include "oj.h"
 #include "parse.h"
 #include "resolve.h"
-#include "hash.h"
-#include "encode.h"
+#include "trace.h"
+
+static void hash_set_cstr(ParseInfo pi, Val kval, const char *str, size_t len, const char *orig) {
+    const char *key    = kval->key;
+    int         klen   = kval->klen;
+    Val         parent = stack_peek(&pi->stack);
+
+    if (Qundef == kval->key_val && Yes == pi->options.create_ok && NULL != pi->options.create_id &&
+        *pi->options.create_id == *key && (int)pi->options.create_id_len == klen &&
+        0 == strncmp(pi->options.create_id, key, klen)) {
+        parent->classname = oj_strndup(str, len);
+        parent->clen      = len;
+    } else {
+        volatile VALUE rstr = oj_cstr_to_value(str, len, (size_t)pi->options.cache_str);
+        volatile VALUE rkey = oj_calc_hash_key(pi, kval);
+
+        if (Yes == pi->options.create_ok && NULL != pi->options.str_rx.head) {
+            VALUE clas = oj_rxclass_match(&pi->options.str_rx, str, (int)len);
+
+            if (Qnil != clas) {
+                rstr = rb_funcall(clas, oj_json_create_id, 1, rstr);
+            }
+        }
+        if (rb_cHash != rb_obj_class(parent->val)) {
+            // The rb_hash_set would still work but the unit tests for the
+            // json gem require the less efficient []= method be called to set
+            // values. Even using the store method to set the values will fail
+            // the unit tests.
+            rb_funcall(parent->val, rb_intern("[]="), 2, rkey, rstr);
+        } else {
+            rb_hash_aset(parent->val, rkey, rstr);
+        }
+        TRACE_PARSE_CALL(pi->options.trace, "set_string", pi, rstr);
+    }
+}
 
-static void
-hash_set_cstr(ParseInfo pi, Val kval, const char *str, size_t len, const char *orig) {
-    const char		*key = kval->key;
-    int			klen = kval->klen;
-    Val			parent = stack_peek(&pi->stack);
-    volatile VALUE	rkey = kval->key_val;
-
-    if (Qundef == rkey &&
-	0 != pi->options.create_id &&
-	*pi->options.create_id == *key &&
-	(int)pi->options.create_id_len == klen &&
-	0 == strncmp(pi->options.create_id, key, klen)) {
-	parent->classname = oj_strndup(str, len);
-	parent->clen = len;
+static VALUE start_hash(ParseInfo pi) {
+    volatile VALUE h;
+
+    if (Qnil != pi->options.hash_class) {
+        h = rb_class_new_instance(0, NULL, pi->options.hash_class);
     } else {
-	volatile VALUE	rstr = rb_str_new(str, len);
-
-	if (Qundef == rkey) {
-	    rkey = rb_str_new(key, klen);
-	    rstr = oj_encode(rstr);
-	    rkey = oj_encode(rkey);
-	    if (Yes == pi->options.sym_key) {
-		rkey = rb_str_intern(rkey);
-	    }
-	}
-	rb_hash_aset(parent->val, rkey, rstr);
+        h = rb_hash_new();
     }
+    TRACE_PARSE_IN(pi->options.trace, "start_hash", pi);
+    return h;
 }
 
-static void
-end_hash(struct _ParseInfo *pi) {
-    Val	parent = stack_peek(&pi->stack);
+static void end_hash(struct _parseInfo *pi) {
+    Val parent = stack_peek(&pi->stack);
 
     if (0 != parent->classname) {
-	VALUE	clas;
-
-	clas = oj_name2class(pi, parent->classname, parent->clen, 0);
-	if (Qundef != clas) { // else an error
-	    parent->val = rb_funcall(clas, oj_json_create_id, 1, parent->val);
-	}
-	if (0 != parent->classname) {
-	    xfree((char*)parent->classname);
-	    parent->classname = 0;
-	}
+        volatile VALUE clas;
+
+        clas = oj_name2class(pi, parent->classname, parent->clen, 0, rb_eArgError);
+        if (Qundef != clas) {  // else an error
+            ID creatable = rb_intern("json_creatable?");
+
+            if (!rb_respond_to(clas, creatable) || Qtrue == rb_funcall(clas, creatable, 0)) {
+                parent->val = rb_funcall(clas, oj_json_create_id, 1, parent->val);
+            }
+        }
+        if (0 != parent->classname) {
+            OJ_R_FREE((char *)parent->classname);
+            parent->classname = 0;
+        }
     }
+    TRACE_PARSE_HASH_END(pi->options.trace, pi);
 }
 
-static VALUE
-calc_hash_key(ParseInfo pi, Val parent) {
-    volatile VALUE	rkey = parent->key_val;
+static void add_cstr(ParseInfo pi, const char *str, size_t len, const char *orig) {
+    volatile VALUE rstr = oj_cstr_to_value(str, len, (size_t)pi->options.cache_str);
 
-    if (Qundef == rkey) {
-	rkey = rb_str_new(parent->key, parent->klen);
-    }
-    rkey = oj_encode(rkey);
-    if (Yes == pi->options.sym_key) {
-	rkey = rb_str_intern(rkey);
+    if (Yes == pi->options.create_ok && NULL != pi->options.str_rx.head) {
+        VALUE clas = oj_rxclass_match(&pi->options.str_rx, str, (int)len);
+
+        if (Qnil != clas) {
+            pi->stack.head->val = rb_funcall(clas, oj_json_create_id, 1, rstr);
+            return;
+        }
     }
-    return rkey;
+    pi->stack.head->val = rstr;
+    TRACE_PARSE_CALL(pi->options.trace, "add_string", pi, rstr);
 }
 
-static void
-add_num(ParseInfo pi, NumInfo ni) {
+static void add_num(ParseInfo pi, NumInfo ni) {
     pi->stack.head->val = oj_num_as_value(ni);
+    TRACE_PARSE_CALL(pi->options.trace, "add_number", pi, pi->stack.head->val);
+}
+
+static void hash_set_num(struct _parseInfo *pi, Val parent, NumInfo ni) {
+    volatile VALUE rval = oj_num_as_value(ni);
+
+    if (rb_cHash != rb_obj_class(parent->val)) {
+        // The rb_hash_set would still work but the unit tests for the
+        // json gem require the less efficient []= method be called to set
+        // values. Even using the store method to set the values will fail
+        // the unit tests.
+        rb_funcall(stack_peek(&pi->stack)->val, rb_intern("[]="), 2, oj_calc_hash_key(pi, parent), rval);
+    } else {
+        rb_hash_aset(stack_peek(&pi->stack)->val, oj_calc_hash_key(pi, parent), rval);
+    }
+    TRACE_PARSE_CALL(pi->options.trace, "set_number", pi, rval);
+}
+
+static void hash_set_value(ParseInfo pi, Val parent, VALUE value) {
+    if (rb_cHash != rb_obj_class(parent->val)) {
+        // The rb_hash_set would still work but the unit tests for the
+        // json gem require the less efficient []= method be called to set
+        // values. Even using the store method to set the values will fail
+        // the unit tests.
+        rb_funcall(stack_peek(&pi->stack)->val, rb_intern("[]="), 2, oj_calc_hash_key(pi, parent), value);
+    } else {
+        rb_hash_aset(stack_peek(&pi->stack)->val, oj_calc_hash_key(pi, parent), value);
+    }
+    TRACE_PARSE_CALL(pi->options.trace, "set_value", pi, value);
 }
 
-static void
-hash_set_num(struct _ParseInfo *pi, Val parent, NumInfo ni) {
-    rb_hash_aset(stack_peek(&pi->stack)->val, calc_hash_key(pi, parent), oj_num_as_value(ni));
+static VALUE start_array(ParseInfo pi) {
+    if (Qnil != pi->options.array_class) {
+        return rb_class_new_instance(0, NULL, pi->options.array_class);
+    }
+    TRACE_PARSE_IN(pi->options.trace, "start_array", pi);
+    return rb_ary_new();
 }
 
-static void
-array_append_num(ParseInfo pi, NumInfo ni) {
-    rb_ary_push(stack_peek(&pi->stack)->val, oj_num_as_value(ni));
+static void array_append_num(ParseInfo pi, NumInfo ni) {
+    Val            parent = stack_peek(&pi->stack);
+    volatile VALUE rval   = oj_num_as_value(ni);
+
+    if (!oj_use_array_alt && rb_cArray != rb_obj_class(parent->val)) {
+        // The rb_ary_push would still work but the unit tests for the json
+        // gem require the less efficient << method be called to push the
+        // values.
+        rb_funcall(parent->val, rb_intern("<<"), 1, rval);
+    } else {
+        rb_ary_push(parent->val, rval);
+    }
+    TRACE_PARSE_CALL(pi->options.trace, "append_number", pi, rval);
 }
 
+static void array_append_cstr(ParseInfo pi, const char *str, size_t len, const char *orig) {
+    volatile VALUE rstr = oj_cstr_to_value(str, len, (size_t)pi->options.cache_str);
+
+    if (Yes == pi->options.create_ok && NULL != pi->options.str_rx.head) {
+        VALUE clas = oj_rxclass_match(&pi->options.str_rx, str, (int)len);
+
+        if (Qnil != clas) {
+            rb_ary_push(stack_peek(&pi->stack)->val, rb_funcall(clas, oj_json_create_id, 1, rstr));
+            return;
+        }
+    }
+    rb_ary_push(stack_peek(&pi->stack)->val, rstr);
+    TRACE_PARSE_CALL(pi->options.trace, "append_string", pi, rstr);
+}
 
-void
-oj_set_compat_callbacks(ParseInfo pi) {
+void oj_set_compat_callbacks(ParseInfo pi) {
     oj_set_strict_callbacks(pi);
-    pi->end_hash = end_hash;
-    pi->hash_set_cstr = hash_set_cstr;
-    pi->add_num = add_num;
-    pi->hash_set_num = hash_set_num;
-    pi->array_append_num = array_append_num;
+    pi->start_hash        = start_hash;
+    pi->end_hash          = end_hash;
+    pi->hash_set_cstr     = hash_set_cstr;
+    pi->hash_set_num      = hash_set_num;
+    pi->hash_set_value    = hash_set_value;
+    pi->add_num           = add_num;
+    pi->add_cstr          = add_cstr;
+    pi->array_append_cstr = array_append_cstr;
+    pi->start_array       = start_array;
+    pi->array_append_num  = array_append_num;
 }
 
 VALUE
 oj_compat_parse(int argc, VALUE *argv, VALUE self) {
-    struct _ParseInfo	pi;
+    struct _parseInfo pi;
+
+    parse_info_init(&pi);
+    pi.options              = oj_default_options;
+    pi.handler              = Qnil;
+    pi.err_class            = Qnil;
+    pi.max_depth            = 0;
+    pi.options.allow_nan    = Yes;
+    pi.options.nilnil       = Yes;
+    pi.options.empty_string = No;
+    oj_set_compat_callbacks(&pi);
 
-    pi.options = oj_default_options;
-    pi.handler = Qnil;
-    pi.err_class = Qnil;
+    if (T_STRING == rb_type(*argv)) {
+        return oj_pi_parse(argc, argv, &pi, 0, 0, false);
+    } else {
+        return oj_pi_sparse(argc, argv, &pi, 0);
+    }
+}
+
+VALUE
+oj_compat_load(int argc, VALUE *argv, VALUE self) {
+    struct _parseInfo pi;
+
+    parse_info_init(&pi);
+    pi.options              = oj_default_options;
+    pi.handler              = Qnil;
+    pi.err_class            = Qnil;
+    pi.max_depth            = 0;
+    pi.options.allow_nan    = Yes;
+    pi.options.nilnil       = Yes;
+    pi.options.empty_string = Yes;
     oj_set_compat_callbacks(&pi);
 
     if (T_STRING == rb_type(*argv)) {
-	return oj_pi_parse(argc, argv, &pi, 0, 0, 1);
+        return oj_pi_parse(argc, argv, &pi, 0, 0, false);
     } else {
-	return oj_pi_sparse(argc, argv, &pi, 0);
+        return oj_pi_sparse(argc, argv, &pi, 0);
     }
 }
 
 VALUE
 oj_compat_parse_cstr(int argc, VALUE *argv, char *json, size_t len) {
-    struct _ParseInfo	pi;
-
-    pi.options = oj_default_options;
-    pi.handler = Qnil;
-    pi.err_class = Qnil;
-    oj_set_strict_callbacks(&pi);
-    pi.end_hash = end_hash;
-    pi.hash_set_cstr = hash_set_cstr;
+    struct _parseInfo pi;
+
+    parse_info_init(&pi);
+    pi.options           = oj_default_options;
+    pi.handler           = Qnil;
+    pi.err_class         = Qnil;
+    pi.max_depth         = 0;
+    pi.options.allow_nan = Yes;
+    pi.options.nilnil    = Yes;
+    oj_set_compat_callbacks(&pi);
 
-    return oj_pi_parse(argc, argv, &pi, json, len, 1);
+    return oj_pi_parse(argc, argv, &pi, json, len, false);
 }