json 0.4.3 → 1.0.0

data/ext/json/ext/parser/parser.rl

@@ -0,0 +1,515 @@
+/* vim: set cin et sw=4 ts=4: */
+
+#include "ruby.h"
+#include "re.h"
+#include "unicode.h"
+
+#ifndef swap16
+#define swap16(x)	((((x)&0xFF)<<8) | (((x)>>8)&0xFF))
+#endif
+
+#define EVIL 0x666
+
+static VALUE mJSON, mExt, cParser, eParserError;
+
+static ID i_json_creatable_p, i_json_create, i_create_id, i_chr; 
+
+typedef struct JSON_ParserStruct {
+    VALUE Vsource;
+    char *source;
+    long len;
+    char *memo;
+    VALUE create_id;
+} JSON_Parser;
+
+static char *JSON_parse_object(JSON_Parser *json, char *p, char *pe, VALUE *result);
+static char *JSON_parse_array(JSON_Parser *json, char *p, char *pe, VALUE *result);
+static char *JSON_parse_value(JSON_Parser *json, char *p, char *pe, VALUE *result);
+static char *JSON_parse_string(JSON_Parser *json, char *p, char *pe, VALUE *result);
+static char *JSON_parse_integer(JSON_Parser *json, char *p, char *pe, VALUE *result);
+static char *JSON_parse_float(JSON_Parser *json, char *p, char *pe, VALUE *result);
+
+#define GET_STRUCT                          \
+    JSON_Parser *json;                      \
+    Data_Get_Struct(self, JSON_Parser, json);
+
+%%{
+    machine JSON_common;
+
+    cr                  = '\n';
+    cr_neg              = [^\n];
+    ws                  = [ \t\r\n];
+    c_comment           = '/*' ( any* - (any* '*/' any* ) ) '*/';
+    cpp_comment         = '//' cr_neg* cr;
+    comment             = c_comment | cpp_comment;
+    ignore              = ws | comment;
+    name_separator      = ':';
+    value_separator     = ',';
+    Vnull               = 'null';
+    Vfalse              = 'false';
+    Vtrue               = 'true';
+    begin_value         = [nft"\-[{] | digit;
+    begin_object        = '{';
+    end_object          = '}';
+    begin_array         = '[';
+    end_array           = ']';
+    begin_string        = '"';
+    begin_name          = begin_string;
+    begin_number        = digit | '-';
+}%%
+
+%%{
+    machine JSON_object;
+    include JSON_common;
+
+    write data;
+
+    action parse_value {
+        VALUE v = Qnil;
+        char *np = JSON_parse_value(json, fpc, pe, &v); 
+        if (np == NULL) {
+            fbreak;
+        } else {
+            rb_hash_aset(*result, last_name, v);
+            fexec np;
+        }
+    }
+
+    action parse_name {
+        char *np = JSON_parse_string(json, fpc, pe, &last_name);
+        if (np == NULL) fbreak; else fexec np;
+    }
+
+    action exit { fbreak; }
+
+    a_pair  = ignore* begin_name >parse_name
+        ignore* name_separator ignore*
+        begin_value >parse_value;
+
+    main := begin_object
+          (a_pair (ignore* value_separator a_pair)*)?
+          ignore* end_object @exit;
+}%%
+
+static char *JSON_parse_object(JSON_Parser *json, char *p, char *pe, VALUE *result)
+{
+    int cs = EVIL;
+    VALUE last_name = Qnil;
+    *result = rb_hash_new();
+
+    %% write init;
+    %% write exec;
+
+    if (cs >= JSON_object_first_final) {
+        VALUE klassname = rb_hash_aref(*result, json->create_id);
+        if (!NIL_P(klassname)) {
+            VALUE klass = rb_path2class(StringValueCStr(klassname));
+            if RTEST(rb_funcall(klass, i_json_creatable_p, 0)) {
+                *result = rb_funcall(klass, i_json_create, 1, *result);
+            }
+        }
+        return p + 1;
+    } else {
+        return NULL;
+    }
+}
+
+%%{
+    machine JSON_value;
+    include JSON_common;
+
+    write data;
+
+    action parse_null {
+        *result = Qnil;
+    }
+    action parse_false {
+        *result = Qfalse;
+    }
+    action parse_true {
+        *result = Qtrue;
+    }
+    action parse_string {
+        char *np = JSON_parse_string(json, fpc, pe, result);
+        if (np == NULL) fbreak; else fexec np;
+    }
+
+    action parse_number {
+        char *np;
+        np = JSON_parse_float(json, fpc, pe, result);
+        if (np != NULL) fexec np;
+        np = JSON_parse_integer(json, fpc, pe, result);
+        if (np != NULL) fexec np;
+        fbreak;
+    }
+
+    action parse_array { 
+        char *np = JSON_parse_array(json, fpc, pe, result);
+        if (np == NULL) fbreak; else fexec np;
+    }
+
+    action parse_object { 
+        char *np =  JSON_parse_object(json, fpc, pe, result);
+        if (np == NULL) fbreak; else fexec np;
+    }
+
+    action exit { fbreak; }
+
+main := (
+              Vnull @parse_null |
+              Vfalse @parse_false |
+              Vtrue @parse_true |
+              begin_number >parse_number |
+              begin_string >parse_string |
+              begin_array >parse_array |
+              begin_object >parse_object
+        ) %*exit;
+}%%
+
+static char *JSON_parse_value(JSON_Parser *json, char *p, char *pe, VALUE *result)
+{
+    int cs = EVIL;
+
+    %% write init;
+    %% write exec;
+
+    if (cs >= JSON_value_first_final) {
+        return p;
+    } else {
+        return NULL;
+    }
+}
+
+%%{
+    machine JSON_integer;
+
+    write data;
+
+    action exit { fbreak; }
+
+    main := '-'? ('0' | [1-9][0-9]*) (^[0-9] @exit);
+}%%
+
+static char *JSON_parse_integer(JSON_Parser *json, char *p, char *pe, VALUE *result)
+{
+    int cs = EVIL;
+
+    %% write init;
+    json->memo = p;
+    %% write exec;
+
+    if (cs >= JSON_integer_first_final) {
+        long len = p - json->memo;
+        *result = rb_Integer(rb_str_new(json->memo, len));
+        return p + 1;
+    } else {
+        return NULL;
+    }
+}
+
+%%{
+    machine JSON_float;
+    include JSON_common;
+
+    write data;
+
+    action exit { fbreak; }
+
+    main := '-'? (
+              (('0' | [1-9][0-9]*) '.' [0-9]+ ([Ee] [+\-]?[0-9]+)?)
+              | ([1-9][0-9]* ([Ee] [+\-]?[0-9]+))
+             )  (^[0-9Ee.\-] @exit );
+}%%
+
+static char *JSON_parse_float(JSON_Parser *json, char *p, char *pe, VALUE *result)
+{
+    int cs = EVIL;
+
+    %% write init;
+    json->memo = p;
+    %% write exec;
+
+    if (cs >= JSON_float_first_final) {
+        long len = p - json->memo;
+        *result = rb_Float(rb_str_new(json->memo, len));
+        return p + 1;
+    } else {
+        return NULL;
+    }
+}
+
+
+%%{
+    machine JSON_array;
+    include JSON_common;
+
+    write data;
+
+    action parse_value {
+        VALUE v = Qnil;
+        char *np = JSON_parse_value(json, fpc, pe, &v); 
+        if (np == NULL) {
+            fbreak;
+        } else {
+            rb_ary_push(*result, v);
+            fexec np;
+        }
+    }
+
+    action exit { fbreak; }
+
+    next_element  = value_separator ignore* begin_value >parse_value;
+
+    main := begin_array ignore*
+          ((begin_value >parse_value ignore*)
+           (ignore* next_element ignore*)*)?
+          end_array @exit;
+}%%
+
+static char *JSON_parse_array(JSON_Parser *json, char *p, char *pe, VALUE *result)
+{
+    int cs = EVIL;
+    *result = rb_ary_new();
+
+    %% write init;
+    %% write exec;
+
+    if(cs >= JSON_array_first_final) {
+        return p + 1;
+    } else {
+        rb_raise(eParserError, "unexpected token at '%s'", p);
+    }
+}
+
+static VALUE json_string_escape(char *p, char *pe)
+{
+    VALUE result = rb_str_buf_new(pe - p + 1);
+
+    while (p < pe) {
+        if (*p == '\\') {
+            p++;
+            if (p >= pe) return Qnil; /* raise an exception later, \ at end */
+            switch (*p) {
+                case '"':
+                case '\\':
+                    rb_str_buf_cat(result, p, 1);
+                    p++;
+                    break;
+                case 'b':
+                    rb_str_buf_cat2(result, "\b");
+                    p++;
+                    break;
+                case 'f':
+                    rb_str_buf_cat2(result, "\f");
+                    p++;
+                    break;
+                case 'n':
+                    rb_str_buf_cat2(result, "\n");
+                    p++;
+                    break;
+                case 'r':
+                    rb_str_buf_cat2(result, "\r");
+                    p++;
+                    break;
+                case 't':
+                    rb_str_buf_cat2(result, "\t");
+                    p++;
+                    break;
+                case 'u':
+                    if (p > pe - 4) { 
+                        return Qnil;
+                    } else {
+                        p = JSON_convert_UTF16_to_UTF8(result, p, pe, strictConversion);
+                    }
+                    break;
+            }
+        } else {
+            char *q = p;
+            while (*q != '\\' && q < pe) q++;
+            rb_str_buf_cat(result, p, q - p);
+            p = q;
+        }
+    }
+    return result;
+}
+
+%%{
+    machine JSON_string;
+    include JSON_common;
+
+    write data;
+
+    action parse_string {
+        *result = json_string_escape(json->memo + 1, p);
+        if (NIL_P(*result)) fbreak; else fexec p + 1;
+    }
+
+    action exit { fbreak; }
+
+    main := '"' ((^(["\\] | 0..0x1f) | '\\'["\\/bfnrt] | '\\u'[0-9a-fA-F]{4})* %parse_string) '"' @exit;
+}%%
+
+static char *JSON_parse_string(JSON_Parser *json, char *p, char *pe, VALUE *result)
+{
+    int cs = EVIL;
+
+    *result = rb_str_new("", 0);
+    %% write init;
+    json->memo = p;
+    %% write exec;
+
+    if (cs >= JSON_string_first_final) {
+        return p + 1;
+    } else {
+        return NULL;
+    }
+}
+
+
+%%{
+    machine JSON;
+
+    write data;
+
+    include JSON_common;
+
+    action parse_object {
+        char *np = JSON_parse_object(json, fpc, pe, &result);
+        if (np == NULL) fbreak; else fexec np;
+    }
+
+    action parse_array {
+        char *np = JSON_parse_array(json, fpc, pe, &result);
+        if (np == NULL) fbreak; else fexec np;
+    }
+
+    main := ignore* (
+            begin_object >parse_object |
+            begin_array >parse_array
+            ) ignore*;
+}%%
+
+/* 
+ * Document-class: JSON::Ext::Parser
+ *
+ * This is the JSON parser implemented as a C extension. It can be configured
+ * to be used by setting
+ *
+ *  JSON.parser = JSON::Ext::Parser
+ *
+ * with the method parser= in JSON.
+ *
+ */
+
+/*
+ * call-seq: new(source)
+ *
+ * Creates a new JSON::Ext::Parser instance for the string _source_.
+ */
+static VALUE cParser_initialize(VALUE self, VALUE source)
+{
+    char *ptr;
+    long len;
+    GET_STRUCT;
+    source = StringValue(source);
+    ptr = RSTRING(source)->ptr;
+    len = RSTRING(source)->len;
+    if (len < 2) {
+        rb_raise(eParserError, "A JSON text must at least contain two octets!");
+    }
+    /*
+       Convert these?
+    if (len >= 4 &&  ptr[0] == 0 && ptr[1] == 0 && ptr[2] == 0) {
+        rb_raise(eParserError, "Only UTF8 octet streams are supported atm!");
+    } else if (len >= 4 && ptr[0] == 0 && ptr[2] == 0) {
+        rb_raise(eParserError, "Only UTF8 octet streams are supported atm!");
+    } else if (len >= 4 && ptr[1] == 0 && ptr[2] == 0 && ptr[3] == 0) {
+        rb_raise(eParserError, "Only UTF8 octet streams are supported atm!");
+    } else if (len >= 4 && ptr[1] == 0 && ptr[3] == 0) {
+        rb_raise(eParserError, "Only UTF8 octet streams are supported atm!");
+    }
+    */
+    json->len = len;
+    json->source = ptr;
+    json->Vsource = source;
+    json->create_id = rb_funcall(mJSON, i_create_id, 0);
+    return self;
+}
+
+/*
+ * call-seq: parse()
+ *
+ *  Parses the current JSON text _source_ and returns the complete data
+ *  structure as a result.
+ */
+static VALUE cParser_parse(VALUE self)
+{
+    GET_STRUCT;
+    char *p, *pe;
+    int cs = EVIL;
+    VALUE result = Qnil;
+
+    %% write init;
+    p = json->source;
+    pe = p + json->len;
+    %% write exec;
+
+    if (cs >= JSON_first_final && p == pe) {
+        return result;
+    } else {
+        rb_raise(eParserError, "unexpected token at '%s'", p);
+    }
+}
+
+static JSON_Parser *JSON_allocate()
+{
+    JSON_Parser *json = ALLOC(JSON_Parser);
+    MEMZERO(json, JSON_Parser, 1);
+    return json;
+}
+
+static void JSON_mark(JSON_Parser *json)
+{
+    rb_gc_mark_maybe(json->Vsource);
+    rb_gc_mark_maybe(json->create_id);
+}
+
+static void JSON_free(JSON_Parser *json)
+{
+    free(json);
+}
+
+static VALUE cJSON_parser_s_allocate(VALUE klass)
+{
+    JSON_Parser *json = JSON_allocate();
+    return Data_Wrap_Struct(klass, JSON_mark, JSON_free, json);
+}
+
+/*
+ * call-seq: source()
+ *
+ * Returns a copy of the current _source_ string, that was used to construct
+ * this Parser.
+ */
+static VALUE cParser_source(VALUE self)
+{
+    GET_STRUCT;
+    return rb_str_dup(json->Vsource);
+}
+
+void Init_parser()
+{
+    mJSON = rb_define_module("JSON");
+    mExt = rb_define_module_under(mJSON, "Ext");
+    cParser = rb_define_class_under(mExt, "Parser", rb_cObject);
+    eParserError = rb_path2class("JSON::ParserError");
+    rb_define_alloc_func(cParser, cJSON_parser_s_allocate);
+    rb_define_method(cParser, "initialize", cParser_initialize, 1);
+    rb_define_method(cParser, "parse", cParser_parse, 0);
+    rb_define_method(cParser, "source", cParser_source, 0);
+
+    i_json_creatable_p = rb_intern("json_creatable?");
+    i_json_create = rb_intern("json_create");
+    i_create_id = rb_intern("create_id");
+    i_chr = rb_intern("chr");
+}