diff options
author | Takashi Kokubun <takashikkbn@gmail.com> | 2022-07-21 09:23:58 -0700 |
---|---|---|
committer | Takashi Kokubun <takashikkbn@gmail.com> | 2022-07-21 09:42:04 -0700 |
commit | 5b21e94bebed90180d8ff63dad03b8b948361089 (patch) | |
tree | f9f7196d84b51b7a3a8001658e4391a63b71c396 /marshal.c | |
parent | 3ff53c8e04ecc91e0190de6d5950ecce2a2ea188 (diff) |
Expand tabs [ci skip]
[Misc #18891]
Notes
Notes:
Merged: https://github.com/ruby/ruby/pull/6094
Diffstat (limited to 'marshal.c')
-rw-r--r-- | marshal.c | 1632 |
1 files changed, 816 insertions, 816 deletions
@@ -54,8 +54,8 @@ shortlen(size_t len, BDIGIT *ds) num = ds[len-1]; while (num) { - num = SHORTDN(num); - offset++; + num = SHORTDN(num); + offset++; } return (len - 1)*SIZEOF_BDIGIT/2 + offset; } @@ -184,20 +184,20 @@ check_dump_arg(VALUE ret, struct dump_arg *arg, const char *name) { if (!arg->symbols) { rb_raise(rb_eRuntimeError, "Marshal.dump reentered at %s", - name); + name); } return ret; } static VALUE check_userdump_arg(VALUE obj, ID sym, int argc, const VALUE *argv, - struct dump_arg *arg, const char *name) + struct dump_arg *arg, const char *name) { VALUE ret = rb_funcallv(obj, sym, argc, argv); VALUE klass = CLASS_OF(obj); if (CLASS_OF(ret) == klass) { rb_raise(rb_eRuntimeError, "%"PRIsVALUE"#%s returned same class instance", - klass, name); + klass, name); } return check_dump_arg(ret, arg, name); } @@ -246,13 +246,13 @@ must_not_be_anonymous(const char *type, VALUE path) char *n = RSTRING_PTR(path); if (!rb_enc_asciicompat(rb_enc_get(path))) { - /* cannot occur? */ - rb_raise(rb_eTypeError, "can't dump non-ascii %s name % "PRIsVALUE, - type, path); + /* cannot occur? */ + rb_raise(rb_eTypeError, "can't dump non-ascii %s name % "PRIsVALUE, + type, path); } if (n[0] == '#') { - rb_raise(rb_eTypeError, "can't dump anonymous %s % "PRIsVALUE, - type, path); + rb_raise(rb_eTypeError, "can't dump anonymous %s % "PRIsVALUE, + type, path); } return path; } @@ -264,7 +264,7 @@ class2path(VALUE klass) must_not_be_anonymous((RB_TYPE_P(klass, T_CLASS) ? "class" : "module"), path); if (rb_path_to_class(path) != rb_class_real(klass)) { - rb_raise(rb_eTypeError, "% "PRIsVALUE" can't be referred to", path); + rb_raise(rb_eTypeError, "% "PRIsVALUE" can't be referred to", path); } return path; } @@ -280,8 +280,8 @@ w_nbyte(const char *s, long n, struct dump_arg *arg) VALUE buf = arg->str; rb_str_buf_cat(buf, s, n); if (arg->dest && RSTRING_LEN(buf) >= BUFSIZ) { - rb_io_write(arg->dest, buf); - rb_str_resize(buf, 0); + rb_io_write(arg->dest, buf); + rb_str_resize(buf, 0); } } @@ -325,7 +325,7 @@ ruby_marshal_write_long(long x, char *buf) #if SIZEOF_LONG > 4 if (!(RSHIFT(x, 31) == 0 || RSHIFT(x, 31) == -1)) { - /* big long does not fit in 4 bytes */ + /* big long does not fit in 4 bytes */ return -1; } #endif @@ -343,16 +343,16 @@ ruby_marshal_write_long(long x, char *buf) return 1; } for (i=1;i<(int)sizeof(long)+1;i++) { - buf[i] = (char)(x & 0xff); - x = RSHIFT(x,8); - if (x == 0) { - buf[0] = i; - break; - } - if (x == -1) { - buf[0] = -i; - break; - } + buf[i] = (char)(x & 0xff); + x = RSHIFT(x,8); + if (x == 0) { + buf[0] = i; + break; + } + if (x == -1) { + buf[0] = -i; + break; + } } return i+1; } @@ -375,13 +375,13 @@ load_mantissa(double d, const char *buf, long len) { if (!len) return d; if (--len > 0 && !*buf++) { /* binary mantissa mark */ - int e, s = d < 0, dig = 0; - unsigned long m; + int e, s = d < 0, dig = 0; + unsigned long m; - modf(ldexp(frexp(fabs(d), &e), DECIMAL_MANT), &d); - do { - m = 0; - switch (len) { + modf(ldexp(frexp(fabs(d), &e), DECIMAL_MANT), &d); + do { + m = 0; + switch (len) { default: m = *buf++ & 0xff; /* fall through */ #if MANT_BITS > 24 case 3: m = (m << 8) | (*buf++ & 0xff); /* fall through */ @@ -390,14 +390,14 @@ load_mantissa(double d, const char *buf, long len) case 2: m = (m << 8) | (*buf++ & 0xff); /* fall through */ #endif #if MANT_BITS > 8 - case 1: m = (m << 8) | (*buf++ & 0xff); + case 1: m = (m << 8) | (*buf++ & 0xff); #endif - } - dig -= len < MANT_BITS / 8 ? 8 * (unsigned)len : MANT_BITS; - d += ldexp((double)m, dig); - } while ((len -= MANT_BITS / 8) > 0); - d = ldexp(d, e - DECIMAL_MANT); - if (s) d = -d; + } + dig -= len < MANT_BITS / 8 ? 8 * (unsigned)len : MANT_BITS; + d += ldexp((double)m, dig); + } while ((len -= MANT_BITS / 8) > 0); + d = ldexp(d, e - DECIMAL_MANT); + if (s) d = -d; } return d; } @@ -417,49 +417,49 @@ w_float(double d, struct dump_arg *arg) char buf[FLOAT_DIG + (DECIMAL_MANT + 7) / 8 + 10]; if (isinf(d)) { - if (d < 0) w_cstr("-inf", arg); - else w_cstr("inf", arg); + if (d < 0) w_cstr("-inf", arg); + else w_cstr("inf", arg); } else if (isnan(d)) { - w_cstr("nan", arg); + w_cstr("nan", arg); } else if (d == 0.0) { if (signbit(d)) w_cstr("-0", arg); else w_cstr("0", arg); } else { - int decpt, sign, digs, len = 0; - char *e, *p = ruby_dtoa(d, 0, 0, &decpt, &sign, &e); - if (sign) buf[len++] = '-'; - digs = (int)(e - p); - if (decpt < -3 || decpt > digs) { - buf[len++] = p[0]; - if (--digs > 0) buf[len++] = '.'; - memcpy(buf + len, p + 1, digs); - len += digs; - len += snprintf(buf + len, sizeof(buf) - len, "e%d", decpt - 1); - } - else if (decpt > 0) { - memcpy(buf + len, p, decpt); - len += decpt; - if ((digs -= decpt) > 0) { - buf[len++] = '.'; - memcpy(buf + len, p + decpt, digs); - len += digs; - } - } - else { - buf[len++] = '0'; - buf[len++] = '.'; - if (decpt) { - memset(buf + len, '0', -decpt); - len -= decpt; - } - memcpy(buf + len, p, digs); - len += digs; - } - xfree(p); - w_bytes(buf, len, arg); + int decpt, sign, digs, len = 0; + char *e, *p = ruby_dtoa(d, 0, 0, &decpt, &sign, &e); + if (sign) buf[len++] = '-'; + digs = (int)(e - p); + if (decpt < -3 || decpt > digs) { + buf[len++] = p[0]; + if (--digs > 0) buf[len++] = '.'; + memcpy(buf + len, p + 1, digs); + len += digs; + len += snprintf(buf + len, sizeof(buf) - len, "e%d", decpt - 1); + } + else if (decpt > 0) { + memcpy(buf + len, p, decpt); + len += decpt; + if ((digs -= decpt) > 0) { + buf[len++] = '.'; + memcpy(buf + len, p + decpt, digs); + len += digs; + } + } + else { + buf[len++] = '0'; + buf[len++] = '.'; + if (decpt) { + memset(buf + len, '0', -decpt); + len -= decpt; + } + memcpy(buf + len, p, digs); + len += digs; + } + xfree(p); + w_bytes(buf, len, arg); } } @@ -470,33 +470,33 @@ w_symbol(VALUE sym, struct dump_arg *arg) VALUE encname; if (st_lookup(arg->symbols, sym, &num)) { - w_byte(TYPE_SYMLINK, arg); - w_long((long)num, arg); + w_byte(TYPE_SYMLINK, arg); + w_long((long)num, arg); } else { - const VALUE orig_sym = sym; - sym = rb_sym2str(sym); - if (!sym) { - rb_raise(rb_eTypeError, "can't dump anonymous ID %"PRIdVALUE, sym); - } - encname = encoding_name(sym, arg); - if (NIL_P(encname) || - is_ascii_string(sym)) { - encname = Qnil; - } - else { - w_byte(TYPE_IVAR, arg); - } - w_byte(TYPE_SYMBOL, arg); - w_bytes(RSTRING_PTR(sym), RSTRING_LEN(sym), arg); - st_add_direct(arg->symbols, orig_sym, arg->symbols->num_entries); - if (!NIL_P(encname)) { - struct dump_call_arg c_arg; - c_arg.limit = 1; - c_arg.arg = arg; - w_long(1L, arg); - w_encoding(encname, &c_arg); - } + const VALUE orig_sym = sym; + sym = rb_sym2str(sym); + if (!sym) { + rb_raise(rb_eTypeError, "can't dump anonymous ID %"PRIdVALUE, sym); + } + encname = encoding_name(sym, arg); + if (NIL_P(encname) || + is_ascii_string(sym)) { + encname = Qnil; + } + else { + w_byte(TYPE_IVAR, arg); + } + w_byte(TYPE_SYMBOL, arg); + w_bytes(RSTRING_PTR(sym), RSTRING_LEN(sym), arg); + st_add_direct(arg->symbols, orig_sym, arg->symbols->num_entries); + if (!NIL_P(encname)) { + struct dump_call_arg c_arg; + c_arg.limit = 1; + c_arg.arg = arg; + w_long(1L, arg); + w_encoding(encname, &c_arg); + } } } @@ -526,12 +526,12 @@ static void w_extended(VALUE klass, struct dump_arg *arg, int check) { if (check && FL_TEST(klass, FL_SINGLETON)) { - VALUE origin = RCLASS_ORIGIN(klass); - if (SINGLETON_DUMP_UNABLE_P(klass) || - (origin != klass && SINGLETON_DUMP_UNABLE_P(origin))) { - rb_raise(rb_eTypeError, "singleton can't be dumped"); - } - klass = RCLASS_SUPER(klass); + VALUE origin = RCLASS_ORIGIN(klass); + if (SINGLETON_DUMP_UNABLE_P(klass) || + (origin != klass && SINGLETON_DUMP_UNABLE_P(origin))) { + rb_raise(rb_eTypeError, "singleton can't be dumped"); + } + klass = RCLASS_SUPER(klass); } while (BUILTIN_TYPE(klass) == T_ICLASS) { if (!FL_TEST(klass, RICLASS_IS_ORIGIN) || @@ -552,7 +552,7 @@ w_class(char type, VALUE obj, struct dump_arg *arg, int check) VALUE klass; if (arg->compat_tbl && - st_lookup(arg->compat_tbl, (st_data_t)obj, &real_obj)) { + st_lookup(arg->compat_tbl, (st_data_t)obj, &real_obj)) { obj = (VALUE)real_obj; } klass = CLASS_OF(obj); @@ -570,8 +570,8 @@ w_uclass(VALUE obj, VALUE super, struct dump_arg *arg) w_extended(klass, arg, TRUE); klass = rb_class_real(klass); if (klass != super) { - w_byte(TYPE_UCLASS, arg); - w_unique(class2path(klass), arg); + w_byte(TYPE_UCLASS, arg); + w_unique(class2path(klass), arg); } } @@ -682,7 +682,7 @@ w_encoding(VALUE encname, struct dump_call_arg *arg) case Qfalse: case Qtrue: w_symbol(ID2SYM(s_encoding_short), arg->arg); - w_object(encname, arg->arg, limit); + w_object(encname, arg->arg, limit); return 1; case Qnil: return 0; @@ -702,14 +702,14 @@ has_ivars(VALUE obj, VALUE encname, VALUE *ivobj) case T_OBJECT: case T_CLASS: case T_MODULE: - break; /* counted elsewhere */ + break; /* counted elsewhere */ case T_HASH: if (rb_hash_ruby2_keywords_p(obj)) ++num; /* fall through */ default: generic: - rb_ivar_foreach(obj, obj_count_ivars, (st_data_t)&num); - if (num) *ivobj = obj; + rb_ivar_foreach(obj, obj_count_ivars, (st_data_t)&num); + if (num) *ivobj = obj; } return num; @@ -736,7 +736,7 @@ w_ivar(st_index_t num, VALUE ivobj, VALUE encname, struct dump_call_arg *arg) int limit = arg->limit; if (limit >= 0) ++limit; w_symbol(ID2SYM(s_ruby2_keywords_flag), arg->arg); - w_object(Qtrue, arg->arg, limit); + w_object(Qtrue, arg->arg, limit); num--; } if (ivobj != Qundef && num) { @@ -764,7 +764,7 @@ w_object(VALUE obj, struct dump_arg *arg, int limit) VALUE encname = Qnil; if (limit == 0) { - rb_raise(rb_eArgError, "exceed depth limit"); + rb_raise(rb_eArgError, "exceed depth limit"); } if (limit > 0) limit--; @@ -773,88 +773,88 @@ w_object(VALUE obj, struct dump_arg *arg, int limit) c_arg.obj = obj; if (st_lookup(arg->data, obj, &num)) { - w_byte(TYPE_LINK, arg); - w_long((long)num, arg); - return; + w_byte(TYPE_LINK, arg); + w_long((long)num, arg); + return; } if (NIL_P(obj)) { - w_byte(TYPE_NIL, arg); + w_byte(TYPE_NIL, arg); } else if (obj == Qtrue) { - w_byte(TYPE_TRUE, arg); + w_byte(TYPE_TRUE, arg); } else if (obj == Qfalse) { - w_byte(TYPE_FALSE, arg); + w_byte(TYPE_FALSE, arg); } else if (FIXNUM_P(obj)) { #if SIZEOF_LONG <= 4 - w_byte(TYPE_FIXNUM, arg); - w_long(FIX2INT(obj), arg); + w_byte(TYPE_FIXNUM, arg); + w_long(FIX2INT(obj), arg); #else - if (RSHIFT((long)obj, 31) == 0 || RSHIFT((long)obj, 31) == -1) { - w_byte(TYPE_FIXNUM, arg); - w_long(FIX2LONG(obj), arg); - } - else { - w_object(rb_int2big(FIX2LONG(obj)), arg, limit); - } + if (RSHIFT((long)obj, 31) == 0 || RSHIFT((long)obj, 31) == -1) { + w_byte(TYPE_FIXNUM, arg); + w_long(FIX2LONG(obj), arg); + } + else { + w_object(rb_int2big(FIX2LONG(obj)), arg, limit); + } #endif } else if (SYMBOL_P(obj)) { - w_symbol(obj, arg); + w_symbol(obj, arg); } else if (FLONUM_P(obj)) { - st_add_direct(arg->data, obj, arg->data->num_entries); - w_byte(TYPE_FLOAT, arg); - w_float(RFLOAT_VALUE(obj), arg); + st_add_direct(arg->data, obj, arg->data->num_entries); + w_byte(TYPE_FLOAT, arg); + w_float(RFLOAT_VALUE(obj), arg); } else { - VALUE v; - - if (!RBASIC_CLASS(obj)) { - rb_raise(rb_eTypeError, "can't dump internal %s", - rb_builtin_type_name(BUILTIN_TYPE(obj))); - } - - if (rb_obj_respond_to(obj, s_mdump, TRUE)) { - st_add_direct(arg->data, obj, arg->data->num_entries); - - v = dump_funcall(arg, obj, s_mdump, 0, 0); - w_class(TYPE_USRMARSHAL, obj, arg, FALSE); - w_object(v, arg, limit); - return; - } - if (rb_obj_respond_to(obj, s_dump, TRUE)) { - VALUE ivobj2 = Qundef; - st_index_t hasiv2; - VALUE encname2; - - v = INT2NUM(limit); - v = dump_funcall(arg, obj, s_dump, 1, &v); - if (!RB_TYPE_P(v, T_STRING)) { - rb_raise(rb_eTypeError, "_dump() must return string"); - } - hasiv = has_ivars(obj, (encname = encoding_name(obj, arg)), &ivobj); - hasiv2 = has_ivars(v, (encname2 = encoding_name(v, arg)), &ivobj2); - if (hasiv2) { - hasiv = hasiv2; - ivobj = ivobj2; - encname = encname2; - } - if (hasiv) w_byte(TYPE_IVAR, arg); - w_class(TYPE_USERDEF, obj, arg, FALSE); - w_bytes(RSTRING_PTR(v), RSTRING_LEN(v), arg); - if (hasiv) { - w_ivar(hasiv, ivobj, encname, &c_arg); - } + VALUE v; + + if (!RBASIC_CLASS(obj)) { + rb_raise(rb_eTypeError, "can't dump internal %s", + rb_builtin_type_name(BUILTIN_TYPE(obj))); + } + + if (rb_obj_respond_to(obj, s_mdump, TRUE)) { st_add_direct(arg->data, obj, arg->data->num_entries); - return; - } + + v = dump_funcall(arg, obj, s_mdump, 0, 0); + w_class(TYPE_USRMARSHAL, obj, arg, FALSE); + w_object(v, arg, limit); + return; + } + if (rb_obj_respond_to(obj, s_dump, TRUE)) { + VALUE ivobj2 = Qundef; + st_index_t hasiv2; + VALUE encname2; + + v = INT2NUM(limit); + v = dump_funcall(arg, obj, s_dump, 1, &v); + if (!RB_TYPE_P(v, T_STRING)) { + rb_raise(rb_eTypeError, "_dump() must return string"); + } + hasiv = has_ivars(obj, (encname = encoding_name(obj, arg)), &ivobj); + hasiv2 = has_ivars(v, (encname2 = encoding_name(v, arg)), &ivobj2); + if (hasiv2) { + hasiv = hasiv2; + ivobj = ivobj2; + encname = encname2; + } + if (hasiv) w_byte(TYPE_IVAR, arg); + w_class(TYPE_USERDEF, obj, arg, FALSE); + w_bytes(RSTRING_PTR(v), RSTRING_LEN(v), arg); + if (hasiv) { + w_ivar(hasiv, ivobj, encname, &c_arg); + } + st_add_direct(arg->data, obj, arg->data->num_entries); + return; + } st_add_direct(arg->data, obj, arg->data->num_entries); - hasiv = has_ivars(obj, (encname = encoding_name(obj, arg)), &ivobj); + hasiv = has_ivars(obj, (encname = encoding_name(obj, arg)), &ivobj); { st_data_t compat_data; rb_alloc_func_t allocator = rb_get_alloc_func(RBASIC(obj)->klass); @@ -868,79 +868,79 @@ w_object(VALUE obj, struct dump_arg *arg, int limit) arg->compat_tbl = rb_init_identtable(); } st_insert(arg->compat_tbl, (st_data_t)obj, (st_data_t)real_obj); - if (obj != real_obj && ivobj == Qundef) hasiv = 0; + if (obj != real_obj && ivobj == Qundef) hasiv = 0; } } - if (hasiv) w_byte(TYPE_IVAR, arg); - - switch (BUILTIN_TYPE(obj)) { - case T_CLASS: - if (FL_TEST(obj, FL_SINGLETON)) { - rb_raise(rb_eTypeError, "singleton class can't be dumped"); - } - w_byte(TYPE_CLASS, arg); - { - VALUE path = class2path(obj); - w_bytes(RSTRING_PTR(path), RSTRING_LEN(path), arg); - RB_GC_GUARD(path); - } - break; - - case T_MODULE: - w_byte(TYPE_MODULE, arg); - { - VALUE path = class2path(obj); - w_bytes(RSTRING_PTR(path), RSTRING_LEN(path), arg); - RB_GC_GUARD(path); - } - break; - - case T_FLOAT: - w_byte(TYPE_FLOAT, arg); - w_float(RFLOAT_VALUE(obj), arg); - break; - - case T_BIGNUM: - w_byte(TYPE_BIGNUM, arg); - { - char sign = BIGNUM_SIGN(obj) ? '+' : '-'; - size_t len = BIGNUM_LEN(obj); - size_t slen; + if (hasiv) w_byte(TYPE_IVAR, arg); + + switch (BUILTIN_TYPE(obj)) { + case T_CLASS: + if (FL_TEST(obj, FL_SINGLETON)) { + rb_raise(rb_eTypeError, "singleton class can't be dumped"); + } + w_byte(TYPE_CLASS, arg); + { + VALUE path = class2path(obj); + w_bytes(RSTRING_PTR(path), RSTRING_LEN(path), arg); + RB_GC_GUARD(path); + } + break; + + case T_MODULE: + w_byte(TYPE_MODULE, arg); + { + VALUE path = class2path(obj); + w_bytes(RSTRING_PTR(path), RSTRING_LEN(path), arg); + RB_GC_GUARD(path); + } + break; + + case T_FLOAT: + w_byte(TYPE_FLOAT, arg); + w_float(RFLOAT_VALUE(obj), arg); + break; + + case T_BIGNUM: + w_byte(TYPE_BIGNUM, arg); + { + char sign = BIGNUM_SIGN(obj) ? '+' : '-'; + size_t len = BIGNUM_LEN(obj); + size_t slen; size_t j; - BDIGIT *d = BIGNUM_DIGITS(obj); + BDIGIT *d = BIGNUM_DIGITS(obj); slen = SHORTLEN(len); if (LONG_MAX < slen) { rb_raise(rb_eTypeError, "too big Bignum can't be dumped"); } - w_byte(sign, arg); - w_long((long)slen, arg); + w_byte(sign, arg); + w_long((long)slen, arg); for (j = 0; j < len; j++) { #if SIZEOF_BDIGIT > SIZEOF_SHORT - BDIGIT num = *d; - int i; + BDIGIT num = *d; + int i; - for (i=0; i<SIZEOF_BDIGIT; i+=SIZEOF_SHORT) { - w_short(num & SHORTMASK, arg); - num = SHORTDN(num); + for (i=0; i<SIZEOF_BDIGIT; i+=SIZEOF_SHORT) { + w_short(num & SHORTMASK, arg); + num = SHORTDN(num); if (j == len - 1 && num == 0) break; - } + } #else - w_short(*d, arg); + w_short(*d, arg); #endif - d++; - } - } - break; - - case T_STRING: - w_uclass(obj, rb_cString, arg); - w_byte(TYPE_STRING, arg); - w_bytes(RSTRING_PTR(obj), RSTRING_LEN(obj), arg); - break; - - case T_REGEXP: + d++; + } + } + break; + + case T_STRING: + w_uclass(obj, rb_cString, arg); + w_byte(TYPE_STRING, arg); + w_bytes(RSTRING_PTR(obj), RSTRING_LEN(obj), arg); + break; + + case T_REGEXP: w_uclass(obj, rb_cRegexp, arg); w_byte(TYPE_REGEXP, arg); { @@ -948,91 +948,91 @@ w_object(VALUE obj, struct dump_arg *arg, int limit) w_bytes(RREGEXP_SRC_PTR(obj), RREGEXP_SRC_LEN(obj), arg); w_byte((char)opts, arg); } - break; - - case T_ARRAY: - w_uclass(obj, rb_cArray, arg); - w_byte(TYPE_ARRAY, arg); - { - long i, len = RARRAY_LEN(obj); - - w_long(len, arg); - for (i=0; i<RARRAY_LEN(obj); i++) { - w_object(RARRAY_AREF(obj, i), arg, limit); - if (len != RARRAY_LEN(obj)) { - rb_raise(rb_eRuntimeError, "array modified during dump"); - } - } - } - break; - - case T_HASH: - w_uclass(obj, rb_cHash, arg); - if (rb_hash_compare_by_id_p(obj)) { - w_byte(TYPE_UCLASS, arg); - w_symbol(rb_sym_intern_ascii_cstr("Hash"), arg); - } - if (NIL_P(RHASH_IFNONE(obj))) { - w_byte(TYPE_HASH, arg); - } + break; + + case T_ARRAY: + w_uclass(obj, rb_cArray, arg); + w_byte(TYPE_ARRAY, arg); + { + long i, len = RARRAY_LEN(obj); + + w_long(len, arg); + for (i=0; i<RARRAY_LEN(obj); i++) { + w_object(RARRAY_AREF(obj, i), arg, limit); + if (len != RARRAY_LEN(obj)) { + rb_raise(rb_eRuntimeError, "array modified during dump"); + } + } + } + break; + + case T_HASH: + w_uclass(obj, rb_cHash, arg); + if (rb_hash_compare_by_id_p(obj)) { + w_byte(TYPE_UCLASS, arg); + w_symbol(rb_sym_intern_ascii_cstr("Hash"), arg); + } + if (NIL_P(RHASH_IFNONE(obj))) { + w_byte(TYPE_HASH, arg); + } else if (FL_TEST(obj, RHASH_PROC_DEFAULT)) { - rb_raise(rb_eTypeError, "can't dump hash with default proc"); - } - else { - w_byte(TYPE_HASH_DEF, arg); - } + rb_raise(rb_eTypeError, "can't dump hash with default proc"); + } + else { + w_byte(TYPE_HASH_DEF, arg); + } w_long(rb_hash_size_num(obj), arg); - rb_hash_foreach(obj, hash_each, (st_data_t)&c_arg); - if (!NIL_P(RHASH_IFNONE(obj))) { - w_object(RHASH_IFNONE(obj), arg, limit); - } - break; - - case T_STRUCT: - w_class(TYPE_STRUCT, obj, arg, TRUE); - { - long len = RSTRUCT_LEN(obj); - VALUE mem; - long i; - - w_long(len, arg); - mem = rb_struct_members(obj); - for (i=0; i<len; i++) { - w_symbol(RARRAY_AREF(mem, i), arg); - w_object(RSTRUCT_GET(obj, i), arg, limit); - } - } - break; - - case T_OBJECT: - w_class(TYPE_OBJECT, obj, arg, TRUE); - w_objivar(obj, &c_arg); - break; - - case T_DATA: - { - VALUE v; - - if (!rb_obj_respond_to(obj, s_dump_data, TRUE)) { - rb_raise(rb_eTypeError, - "no _dump_data is defined for class %"PRIsVALUE, - rb_obj_class(obj)); - } - v = dump_funcall(arg, obj, s_dump_data, 0, 0); - w_class(TYPE_DATA, obj, arg, TRUE); - w_object(v, arg, limit); - } - break; - - default: - rb_raise(rb_eTypeError, "can't dump %"PRIsVALUE, - rb_obj_class(obj)); - break; - } - RB_GC_GUARD(obj); + rb_hash_foreach(obj, hash_each, (st_data_t)&c_arg); + if (!NIL_P(RHASH_IFNONE(obj))) { + w_object(RHASH_IFNONE(obj), arg, limit); + } + break; + + case T_STRUCT: + w_class(TYPE_STRUCT, obj, arg, TRUE); + { + long len = RSTRUCT_LEN(obj); + VALUE mem; + long i; + + w_long(len, arg); + mem = rb_struct_members(obj); + for (i=0; i<len; i++) { + w_symbol(RARRAY_AREF(mem, i), arg); + w_object(RSTRUCT_GET(obj, i), arg, limit); + } + } + break; + + case T_OBJECT: + w_class(TYPE_OBJECT, obj, arg, TRUE); + w_objivar(obj, &c_arg); + break; + + case T_DATA: + { + VALUE v; + + if (!rb_obj_respond_to(obj, s_dump_data, TRUE)) { + rb_raise(rb_eTypeError, + "no _dump_data is defined for class %"PRIsVALUE, + rb_obj_class(obj)); + } + v = dump_funcall(arg, obj, s_dump_data, 0, 0); + w_class(TYPE_DATA, obj, arg, TRUE); + w_object(v, arg, limit); + } + break; + + default: + rb_raise(rb_eTypeError, "can't dump %"PRIsVALUE, + rb_obj_class(obj)); + break; + } + RB_GC_GUARD(obj); } if (hasiv) { - w_ivar(hasiv, ivobj, encname, &c_arg); + w_ivar(hasiv, ivobj, encname, &c_arg); } } @@ -1045,12 +1045,12 @@ clear_dump_arg(struct dump_arg *arg) st_free_table(arg->data); arg->data = 0; if (arg->compat_tbl) { - st_free_table(arg->compat_tbl); - arg->compat_tbl = 0; + st_free_table(arg->compat_tbl); + arg->compat_tbl = 0; } if (arg->encodings) { - st_free_table(arg->encodings); - arg->encodings = 0; + st_free_table(arg->encodings); + arg->encodings = 0; } } @@ -1104,14 +1104,14 @@ marshal_dump(int argc, VALUE *argv, VALUE _) port = Qnil; rb_scan_args(argc, argv, "12", &obj, &a1, &a2); if (argc == 3) { - if (!NIL_P(a2)) limit = NUM2INT(a2); - if (NIL_P(a1)) io_needed(); - port = a1; + if (!NIL_P(a2)) limit = NUM2INT(a2); + if (NIL_P(a1)) io_needed(); + port = a1; } else if (argc == 2) { - if (FIXNUM_P(a1)) limit = FIX2INT(a1); - else if (NIL_P(a1)) io_needed(); - else port = a1; + if (FIXNUM_P(a1)) limit = FIX2INT(a1); + else if (NIL_P(a1)) io_needed(); + else port = a1; } return rb_marshal_dump_limited(obj, port, limit); } @@ -1130,14 +1130,14 @@ rb_marshal_dump_limited(VALUE obj, VALUE port, int limit) arg->encodings = 0; arg->str = rb_str_buf_new(0); if (!NIL_P(port)) { - if (!rb_respond_to(port, s_write)) { - io_needed(); - } - arg->dest = port; - dump_check_funcall(arg, port, s_binmode, 0, 0); + if (!rb_respond_to(port, s_write)) { + io_needed(); + } + arg->dest = port; + dump_check_funcall(arg, port, s_binmode, 0, 0); } else { - port = arg->str; + port = arg->str; } w_byte(MARSHAL_MAJOR, arg); @@ -1145,8 +1145,8 @@ rb_marshal_dump_limited(VALUE obj, VALUE port, int limit) w_object(obj, arg, limit); if (arg->dest) { - rb_io_write(arg->dest, arg->str); - rb_str_resize(arg->str, 0); + rb_io_write(arg->dest, arg->str); + rb_str_resize(arg->str, 0); } clear_dump_arg(arg); RB_GC_GUARD(wrapper); @@ -1173,7 +1173,7 @@ check_load_arg(VALUE ret, struct load_arg *arg, const char *name) { if (!arg->symbols) { rb_raise(rb_eRuntimeError, "Marshal.load reentered at %s", - name); + name); } return ret; } @@ -1237,15 +1237,15 @@ static unsigned char r_byte1_buffered(struct load_arg *arg) { if (arg->buflen == 0) { - long readable = arg->readable < BUFSIZ ? arg->readable : BUFSIZ; - VALUE str, n = LONG2NUM(readable); + long readable = arg->readable < BUFSIZ ? arg->readable : BUFSIZ; + VALUE str, n = LONG2NUM(readable); - str = load_funcall(arg, arg->src, s_read, 1, &n); - if (NIL_P(str)) too_short(); - StringValue(str); - memcpy(arg->buf, RSTRING_PTR(str), RSTRING_LEN(str)); - arg->offset = 0; - arg->buflen = RSTRING_LEN(str); + str = load_funcall(arg, arg->src, s_read, 1, &n); + if (NIL_P(str)) too_short(); + StringValue(str); + memcpy(arg->buf, RSTRING_PTR(str), RSTRING_LEN(str)); + arg->offset = 0; + arg->buflen = RSTRING_LEN(str); } arg->buflen--; return arg->buf[arg->offset++]; @@ -1257,22 +1257,22 @@ r_byte(struct load_arg *arg) int c; if (RB_TYPE_P(arg->src, T_STRING)) { - if (RSTRING_LEN(arg->src) > arg->offset) { - c = (unsigned char)RSTRING_PTR(arg->src)[arg->offset++]; - } - else { - too_short(); - } + if (RSTRING_LEN(arg->src) > arg->offset) { + c = (unsigned char)RSTRING_PTR(arg->src)[arg->offset++]; + } + else { + too_short(); + } } else { - if (arg->readable >0 || arg->buflen > 0) { - c = r_byte1_buffered(arg); - } - else { - VALUE v = load_funcall(arg, arg->src, s_getbyte, 0, 0); - if (NIL_P(v)) rb_eof_error(); - c = (unsigned char)NUM2CHR(v); - } + if (arg->readable >0 || arg->buflen > 0) { + c = r_byte1_buffered(arg); + } + else { + VALUE v = load_funcall(arg, arg->src, s_getbyte, 0, 0); + if (NIL_P(v)) rb_eof_error(); + c = (unsigned char)NUM2CHR(v); + } } return c; } @@ -1283,7 +1283,7 @@ static void long_toobig(int size) { rb_raise(rb_eTypeError, "long too big for this architecture (size " - STRINGIZE(SIZEOF_LONG)", given %d)", size); + STRINGIZE(SIZEOF_LONG)", given %d)", size); } static long @@ -1295,26 +1295,26 @@ r_long(struct load_arg *arg) if (c == 0) return 0; if (c > 0) { - if (4 < c && c < 128) { - return c - 5; - } - if (c > (int)sizeof(long)) long_toobig(c); - x = 0; - for (i=0;i<c;i++) { - x |= (long)r_byte(arg) << (8*i); - } + if (4 < c && c < 128) { + return c - 5; + } + if (c > (int)sizeof(long)) long_toobig(c); + x = 0; + for (i=0;i<c;i++) { + x |= (long)r_byte(arg) << (8*i); + } } else { - if (-129 < c && c < -4) { - return c + 5; - } - c = -c; - if (c > (int)sizeof(long)) long_toobig(c); - x = -1; - for (i=0;i<c;i++) { - x &= ~((long)0xff << (8*i)); - x |= (long)r_byte(arg) << (8*i); - } + if (-129 < c && c < -4) { + return c + 5; + } + c = -c; + if (c > (int)sizeof(long)) long_toobig(c); + x = -1; + for (i=0;i<c;i++) { + x &= ~((long)0xff << (8*i)); + x |= (long)r_byte(arg) << (8*i); + } } return x; } @@ -1351,39 +1351,39 @@ r_bytes1_buffered(long len, struct load_arg *arg) VALUE str; if (len <= arg->buflen) { - str = rb_str_new(arg->buf+arg->offset, len); - arg->offset += len; - arg->buflen -= len; + str = rb_str_new(arg->buf+arg->offset, len); + arg->offset += len; + arg->buflen -= len; } else { - long buflen = arg->buflen; - long readable = arg->readable + 1; - long tmp_len, read_len, need_len = len - buflen; - VALUE tmp, n; + long buflen = arg->buflen; + long readable = arg->readable + 1; + long tmp_len, read_len, need_len = len - buflen; + VALUE tmp, n; - readable = readable < BUFSIZ ? readable : BUFSIZ; - read_len = need_len > readable ? need_len : readable; - n = LONG2NUM(read_len); - tmp = load_funcall(arg, arg->src, s_read, 1, &n); - if (NIL_P(tmp)) too_short(); - StringValue(tmp); + readable = readable < BUFSIZ ? readable : BUFSIZ; + read_len = need_len > readable ? need_len : readable; + n = LONG2NUM(read_len); + tmp = load_funcall(arg, arg->src, s_read, 1, &n); + if (NIL_P(tmp)) too_short(); + StringValue(tmp); - tmp_len = RSTRING_LEN(tmp); + tmp_len = RSTRING_LEN(tmp); - if (tmp_len < need_len) too_short(); + if (tmp_len < need_len) too_short(); - str = rb_str_new(arg->buf+arg->offset, buflen); - rb_str_cat(str, RSTRING_PTR(tmp), need_len); + str = rb_str_new(arg->buf+arg->offset, buflen); + rb_str_cat(str, RSTRING_PTR(tmp), need_len); - if (tmp_len > need_len) { - buflen = tmp_len - need_len; - memcpy(arg->buf, RSTRING_PTR(tmp)+need_len, buflen); - arg->buflen = buflen; - } - else { - arg->buflen = 0; - } - arg->offset = 0; + if (tmp_len > need_len) { + buflen = tmp_len - need_len; + memcpy(arg->buf, RSTRING_PTR(tmp)+need_len, buflen); + arg->buflen = buflen; + } + else { + arg->buflen = 0; + } + arg->offset = 0; } return str; @@ -1398,21 +1398,21 @@ r_bytes0(long len, struct load_arg *arg) if (len == 0) return rb_str_new(0, 0); if (RB_TYPE_P(arg->src, T_STRING)) { - if (RSTRING_LEN(arg->src) - arg->offset >= len) { - str = rb_str_new(RSTRING_PTR(arg->src)+arg->offset, len); - arg->offset += len; - } - else { - too_short(); - } + if (RSTRING_LEN(arg->src) - arg->offset >= len) { + str = rb_str_new(RSTRING_PTR(arg->src)+arg->offset, len); + arg->offset += len; + } + else { + too_short(); + } } else { - if (arg->readable > 0 || arg->buflen > 0) { - str = r_bytes1_buffered(len, arg); - } - else { - str = r_bytes1(len, arg); - } + if (arg->readable > 0 || arg->buflen > 0) { + str = r_bytes1_buffered(len, arg); + } + else { + str = r_bytes1(len, arg); + } } return str; } @@ -1434,13 +1434,13 @@ sym2encidx(VALUE sym, VALUE val) RSTRING_GETMEM(sym, p, l); if (l <= 0) return -1; if (name_equal(name_encoding, sizeof(name_encoding), p, l)) { - int idx = rb_enc_find_index(StringValueCStr(val)); - return idx; + int idx = rb_enc_find_index(StringValueCStr(val)); + return idx; } if (name_equal(name_s_encoding_short, rb_strlen_lit(name_s_encoding_short), p, l)) { - if (val == Qfalse) return rb_usascii_encindex(); - else if (val == Qtrue) return rb_utf8_encindex(); - /* bogus ignore */ + if (val == Qfalse) return rb_usascii_encindex(); + else if (val == Qtrue) return rb_utf8_encindex(); + /* bogus ignore */ } return -1; } @@ -1468,7 +1468,7 @@ r_symlink(struct load_arg *arg) long num = r_long(arg); if (!st_lookup(arg->symbols, num, &sym)) { - rb_raise(rb_eArgError, "bad symbol"); + rb_raise(rb_eArgError, "bad symbol"); } return (VALUE)sym; } @@ -1484,11 +1484,11 @@ r_symreal(struct load_arg *arg, int ivar) if (rb_enc_str_asciionly_p(s)) rb_enc_associate_index(s, ENCINDEX_US_ASCII); st_insert(arg->symbols, (st_data_t)n, (st_data_t)s); if (ivar) { - long num = r_long(arg); - while (num-- > 0) { - sym = r_symbol(arg); - idx = sym2encidx(sym, r_object(arg)); - } + long num = r_long(arg); + while (num-- > 0) { + sym = r_symbol(arg); + idx = sym2encidx(sym, r_object(arg)); + } } if (idx > 0) { rb_enc_associate_index(s, idx); @@ -1509,17 +1509,17 @@ r_symbol(struct load_arg *arg) again: switch ((type = r_byte(arg))) { default: - rb_raise(rb_eArgError, "dump format error for symbol(0x%x)", type); + rb_raise(rb_eArgError, "dump format error for symbol(0x%x)", type); case TYPE_IVAR: - ivar = 1; - goto again; + ivar = 1; + goto again; case TYPE_SYMBOL: - return r_symreal(arg, ivar); + return r_symreal(arg, ivar); case TYPE_SYMLINK: - if (ivar) { - rb_raise(rb_eArgError, "dump format error (symlink with encoding)"); - } - return r_symlink(arg); + if (ivar) { + rb_raise(rb_eArgError, "dump format error (symlink with encoding)"); + } + return r_symlink(arg); } } @@ -1569,7 +1569,7 @@ static VALUE r_post_proc(VALUE v, struct load_arg *arg) { if (arg->proc) { - v = load_funcall(arg, arg->proc, s_call, 1, &v); + v = load_funcall(arg, arg->proc, s_call, 1, &v); } return v; } @@ -1579,21 +1579,21 @@ r_leave(VALUE v, struct load_arg *arg, bool partial) { v = r_fixup_compat(v, arg); if (!partial) { - st_data_t data; - st_data_t key = (st_data_t)v; - st_delete(arg->partial_objects, &key, &data); - if (arg->freeze) { - if (RB_TYPE_P(v, T_MODULE) || RB_TYPE_P(v, T_CLASS)) { - // noop - } - else if (RB_TYPE_P(v, T_STRING)) { - v = rb_str_to_interned_str(v); - } - else { - OBJ_FREEZE(v); - } - } - v = r_post_proc(v, arg); + st_data_t data; + st_data_t key = (st_data_t)v; + st_delete(arg->partial_objects, &key, &data); + if (arg->freeze) { + if (RB_TYPE_P(v, T_MODULE) || RB_TYPE_P(v, T_CLASS)) { + // noop + } + else if (RB_TYPE_P(v, T_STRING)) { + v = rb_str_to_interned_str(v); + } + else { + OBJ_FREEZE(v); + } + } + v = r_post_proc(v, arg); } return v; } @@ -1605,7 +1605,7 @@ copy_ivar_i(st_data_t key, st_data_t val, st_data_t arg) ID vid = (ID)key; if (!rb_ivar_defined(obj, vid)) - rb_ivar_set(obj, vid, value); + rb_ivar_set(obj, vid, value); return ST_CONTINUE; } @@ -1623,20 +1623,20 @@ r_ivar(VALUE obj, int *has_encoding, struct load_arg *arg) len = r_long(arg); if (len > 0) { - do { - VALUE sym = r_symbol(arg); - VALUE val = r_object(arg); - int idx = sym2encidx(sym, val); - if (idx >= 0) { + do { + VALUE sym = r_symbol(arg); + VALUE val = r_object(arg); + int idx = sym2encidx(sym, val); + if (idx >= 0) { if (rb_enc_capable(obj)) { rb_enc_associate_index(obj, idx); } else { rb_raise(rb_eArgError, "%"PRIsVALUE" is not enc_capable", obj); } - if (has_encoding) *has_encoding = TRUE; - } - else if (symname_equal_lit(sym, name_s_ruby2_keywords_flag)) { + if (has_encoding) *has_encoding = TRUE; + } + else if (symname_equal_lit(sym, name_s_ruby2_keywords_flag)) { if (RB_TYPE_P(obj, T_HASH)) { rb_hash_ruby2_keywords(obj); } @@ -1646,8 +1646,8 @@ r_ivar(VALUE obj, int *has_encoding, struct load_arg *arg) } else { rb_ivar_set(obj, rb_intern_str(sym), val); - } - } while (--len > 0); + } + } while (--len > 0); } } @@ -1657,7 +1657,7 @@ path2class(VALUE path) VALUE v = rb_path_to_class(path); if (!RB_TYPE_P(v, T_CLASS)) { - rb_raise(rb_eArgError, "%"PRIsVALUE" does not refer to class", path); + rb_raise(rb_eArgError, "%"PRIsVALUE" does not refer to class", path); } return v; } @@ -1668,7 +1668,7 @@ static VALUE must_be_module(VALUE v, VALUE path) { if (!RB_TYPE_P(v, T_MODULE)) { - rb_raise(rb_eArgError, "%"PRIsVALUE" does not refer to module", path); + rb_raise(rb_eArgError, "%"PRIsVALUE" does not refer to module", path); } return v; } @@ -1684,7 +1684,7 @@ obj_alloc_by_klass(VALUE klass, struct load_arg *arg, VALUE *oldclass) marshal_compat_t *compat = (marshal_compat_t*)data; VALUE real_obj = rb_obj_alloc(klass); VALUE obj = rb_obj_alloc(compat->oldclass); - if (oldclass) *oldclass = compat->oldclass; + if (oldclass) *oldclass = compat->oldclass; if (!arg->compat_tbl) { arg->compat_tbl = rb_init_identtable(); @@ -1707,17 +1707,17 @@ append_extmod(VALUE obj, VALUE extmod) { long i = RARRAY_LEN(extmod); while (i > 0) { - VALUE m = RARRAY_AREF(extmod, --i); - rb_extend_object(obj, m); + VALUE m = RARRAY_AREF(extmod, --i); + rb_extend_object(obj, m); } return obj; } #define prohibit_ivar(type, str) do { \ - if (!ivp || !*ivp) break; \ - rb_raise(rb_eTypeError, \ - "can't override instance variable of "type" `%"PRIsVALUE"'", \ - (str)); \ + if (!ivp || !*ivp) break; \ + rb_raise(rb_eTypeError, \ + "can't override instance variable of "type" `%"PRIsVALUE"'", \ + (str)); \ } while (0) static VALUE r_object_for(struct load_arg *arg, bool partial, int *ivp, VALUE extmod, int type); @@ -1739,437 +1739,437 @@ r_object_for(struct load_arg *arg, bool partial, int *ivp, VALUE extmod, int typ switch (type) { case TYPE_LINK: - id = r_long(arg); - if (!st_lookup(arg->data, (st_data_t)id, &link)) { - rb_raise(rb_eArgError, "dump format error (unlinked)"); - } - v = (VALUE)link; - if (!st_lookup(arg->partial_objects, (st_data_t)v, &link)) { - v = r_post_proc(v, arg); - } - break; + id = r_long(arg); + if (!st_lookup(arg->data, (st_data_t)id, &link)) { + rb_raise(rb_eArgError, "dump format error (unlinked)"); + } + v = (VALUE)link; + if (!st_lookup(arg->partial_objects, (st_data_t)v, &link)) { + v = r_post_proc(v, arg); + } + break; case TYPE_IVAR: { - int ivar = TRUE; + int ivar = TRUE; - v = r_object0(arg, true, &ivar, extmod); - if (ivar) r_ivar(v, NULL, arg); - v = r_leave(v, arg, partial); - } - break; + v = r_object0(arg, true, &ivar, extmod); + if (ivar) r_ivar(v, NULL, arg); + v = r_leave(v, arg, partial); + } + break; case TYPE_EXTENDED: - { - VALUE path = r_unique(arg); - VALUE m = rb_path_to_class(path); - if (NIL_P(extmod)) extmod = rb_ary_tmp_new(0); - - if (RB_TYPE_P(m, T_CLASS)) { /* prepended */ - VALUE c; - - v = r_object0(arg, true, 0, Qnil); - c = CLASS_OF(v); - if (c != m || FL_TEST(c, FL_SINGLETON)) { - rb_raise(rb_eArgError, - "prepended class %"PRIsVALUE" differs from class %"PRIsVALUE, - path, rb_class_name(c)); - } - c = rb_singleton_class(v); - while (RARRAY_LEN(extmod) > 0) { - m = rb_ary_pop(extmod); - rb_prepend_module(c, m); - } - } - else { - must_be_module(m, path); - rb_ary_push(extmod, m); - - v = r_object0(arg, true, 0, extmod); - while (RARRAY_LEN(extmod) > 0) { - m = rb_ary_pop(extmod); - rb_extend_object(v, m); - } - } - } - break; + { + VALUE path = r_unique(arg); + VALUE m = rb_path_to_class(path); + if (NIL_P(extmod)) extmod = rb_ary_tmp_new(0); + + if (RB_TYPE_P(m, T_CLASS)) { /* prepended */ + VALUE c; + + v = r_object0(arg, true, 0, Qnil); + c = CLASS_OF(v); + if (c != m || FL_TEST(c, FL_SINGLETON)) { + rb_raise(rb_eArgError, + "prepended class %"PRIsVALUE" differs from class %"PRIsVALUE, + path, rb_class_name(c)); + } + c = rb_singleton_class(v); + while (RARRAY_LEN(extmod) > 0) { + m = rb_ary_pop(extmod); + rb_prepend_module(c, m); + } + } + else { + must_be_module(m, path); + rb_ary_push(extmod, m); + + v = r_object0(arg, true, 0, extmod); + while (RARRAY_LEN(extmod) > 0) { + m = rb_ary_pop(extmod); + rb_extend_object(v, m); + } + } + } + break; case TYPE_UCLASS: - { - VALUE c = path2class(r_unique(arg)); - - if (FL_TEST(c, FL_SINGLETON)) { - rb_raise(rb_eTypeError, "singleton can't be loaded"); - } - type = r_byte(arg); - if ((c == rb_cHash) && - /* Hack for compare_by_identify */ - (type == TYPE_HASH || type == TYPE_HASH_DEF)) { - hash_new_with_size = rb_ident_hash_new_with_size; - goto type_hash; - } - v = r_object_for(arg, partial, 0, extmod, type); - if (rb_special_const_p(v) || RB_TYPE_P(v, T_OBJECT) || RB_TYPE_P(v, T_CLASS)) { + { + VALUE c = path2class(r_unique(arg)); + + if (FL_TEST(c, FL_SINGLETON)) { + rb_raise(rb_eTypeError, "singleton can't be loaded"); + } + type = r_byte(arg); + if ((c == rb_cHash) && + /* Hack for compare_by_identify */ + (type == TYPE_HASH || type == TYPE_HASH_DEF)) { + hash_new_with_size = rb_ident_hash_new_with_size; + goto type_hash; + } + v = r_object_for(arg, partial, 0, extmod, type); + if (rb_special_const_p(v) || RB_TYPE_P(v, T_OBJECT) || RB_TYPE_P(v, T_CLASS)) { goto format_error; - } - if (RB_TYPE_P(v, T_MODULE) || !RTEST(rb_class_inherited_p(c, RBASIC(v)->klass))) { - VALUE tmp = rb_obj_alloc(c); + } + if (RB_TYPE_P(v, T_MODULE) || !RTEST(rb_class_inherited_p(c, RBASIC(v)->klass))) { + VALUE tmp = rb_obj_alloc(c); - if (TYPE(v) != TYPE(tmp)) goto format_error; - } - RBASIC_SET_CLASS(v, c); - } - break; + if (TYPE(v) != TYPE(tmp)) goto format_error; + } + RBASIC_SET_CLASS(v, c); + } + break; format_error: rb_raise(rb_eArgError, "dump format error (user class)"); case TYPE_NIL: - v = Qnil; - v = r_leave(v, arg, false); - break; + v = Qnil; + v = r_leave(v, arg, false); + break; case TYPE_TRUE: - v = Qtrue; - v = r_leave(v, arg, false); - break; + v = Qtrue; + v = r_leave(v, arg, false); + break; case TYPE_FALSE: - v = Qfalse; - v = r_leave(v, arg, false); - break; + v = Qfalse; + v = r_leave(v, arg, false); + break; case TYPE_FIXNUM: - { - long i = r_long(arg); - v = LONG2FIX(i); - } - v = r_leave(v, arg, false); - break; + { + long i = r_long(arg); + v = LONG2FIX(i); + } + v = r_leave(v, arg, false); + break; case TYPE_FLOAT: - { - double d; - VALUE str = r_bytes(arg); - const char *ptr = RSTRING_PTR(str); - - if (strcmp(ptr, "nan") == 0) { - d = nan(""); - } - else if (strcmp(ptr, "inf") == 0) { - d = HUGE_VAL; - } - else if (strcmp(ptr, "-inf") == 0) { - d = -HUGE_VAL; - } - else { - char *e; - d = strtod(ptr, &e); - d = load_mantissa(d, e, RSTRING_LEN(str) - (e - ptr)); - } - v = DBL2NUM(d); - v = r_entry(v, arg); + { + double d; + VALUE str = r_bytes(arg); + const char *ptr = RSTRING_PTR(str); + + if (strcmp(ptr, "nan") == 0) { + d = nan(""); + } + else if (strcmp(ptr, "inf") == 0) { + d = HUGE_VAL; + } + else if (strcmp(ptr, "-inf") == 0) { + d = -HUGE_VAL; + } + else { + char *e; + d = strtod(ptr, &e); + d = load_mantissa(d, e, RSTRING_LEN(str) - (e - ptr)); + } + v = DBL2NUM(d); + v = r_entry(v, arg); v = r_leave(v, arg, false); - } - break; + } + break; case TYPE_BIGNUM: - { - long len; - VALUE data; + { + long len; + VALUE data; int sign; - sign = r_byte(arg); - len = r_long(arg); - data = r_bytes0(len * 2, arg); + sign = r_byte(arg); + len = r_long(arg); + data = r_bytes0(len * 2, arg); v = rb_integer_unpack(RSTRING_PTR(data), len, 2, 0, INTEGER_PACK_LITTLE_ENDIAN | (sign == '-' ? INTEGER_PACK_NEGATIVE : 0)); - rb_str_resize(data, 0L); - v = r_entry(v, arg); + rb_str_resize(data, 0L); + v = r_entry(v, arg); v = r_leave(v, arg, false); - } - break; + } + break; case TYPE_STRING: - v = r_entry(r_string(arg), arg); - v = r_leave(v, arg, partial); - break; + v = r_entry(r_string(arg), arg); + v = r_leave(v, arg, partial); + break; case TYPE_REGEXP: - { - VALUE str = r_bytes(arg); - int options = r_byte(arg); - int has_encoding = FALSE; - st_index_t idx = r_prepare(arg); - - if (ivp) { - r_ivar(str, &has_encoding, arg); - *ivp = FALSE; - } - if (!has_encoding) { - /* 1.8 compatibility; remove escapes undefined in 1.8 */ - char *ptr = RSTRING_PTR(str), *dst = ptr, *src = ptr; - long len = RSTRING_LEN(str); - long bs = 0; - for (; len-- > 0; *dst++ = *src++) { - switch (*src) { - case '\\': bs++; break; - case 'g': case 'h': case 'i': case 'j': case 'k': case 'l': - case 'm': case 'o': case 'p': case 'q': case 'u': case 'y': - case 'E': case 'F': case 'H': case 'I': case 'J': case 'K': - case 'L': case 'N': case 'O': case 'P': case 'Q': case 'R': - case 'S': case 'T': case 'U': case 'V': case 'X': case 'Y': - if (bs & 1) --dst; + { + VALUE str = r_bytes(arg); + int options = r_byte(arg); + int has_encoding = FALSE; + st_index_t idx = r_prepare(arg); + + if (ivp) { + r_ivar(str, &has_encoding, arg); + *ivp = FALSE; + } + if (!has_encoding) { + /* 1.8 compatibility; remove escapes undefined in 1.8 */ + char *ptr = RSTRING_PTR(str), *dst = ptr, *src = ptr; + long len = RSTRING_LEN(str); + long bs = 0; + for (; len-- > 0; *dst++ = *src++) { + switch (*src) { + case '\\': bs++; break; + case 'g': case 'h': case 'i': case 'j': case 'k': case 'l': + case 'm': case 'o': case 'p': case 'q': case 'u': case 'y': + case 'E': case 'F': case 'H': case 'I': case 'J': case 'K': + case 'L': case 'N': case 'O': case 'P': case 'Q': case 'R': + case 'S': case 'T': case 'U': case 'V': case 'X': case 'Y': + if (bs & 1) --dst; /* fall through */ - default: bs = 0; break; - } - } - rb_str_set_len(str, dst - ptr); - } - v = r_entry0(rb_reg_new_str(str, options), idx, arg); - v = r_leave(v, arg, partial); - } - break; + default: bs = 0; break; + } + } + rb_str_set_len(str, dst - ptr); + } + v = r_entry0(rb_reg_new_str(str, options), idx, arg); + v = r_leave(v, arg, partial); + } + break; case TYPE_ARRAY: - { - long len = r_long(arg); - - v = rb_ary_new2(len); - v = r_entry(v, arg); - arg->readable += len - 1; - while (len--) { - rb_ary_push(v, r_object(arg)); - arg->readable--; - } + { + long len = r_long(arg); + + v = rb_ary_new2(len); + v = r_entry(v, arg); + arg->readable += len - 1; + while (len--) { + rb_ary_push(v, r_object(arg)); + arg->readable--; + } v = r_leave(v, arg, partial); - arg->readable++; - } - break; + arg->readable++; + } + break; case TYPE_HASH: case TYPE_HASH_DEF: type_hash: - { - long len = r_long(arg); - - v = hash_new_with_size(len); - v = r_entry(v, arg); - arg->readable += (len - 1) * 2; - while (len--) { - VALUE key = r_object(arg); - VALUE value = r_object(arg); - rb_hash_aset(v, key, value); - arg->readable -= 2; - } - arg->readable += 2; - if (type == TYPE_HASH_DEF) { - RHASH_SET_IFNONE(v, r_object(arg)); - } + { + long len = r_long(arg); + + v = hash_new_with_size(len); + v = r_entry(v, arg); + arg->readable += (len - 1) * 2; + while (len--) { + VALUE key = r_object(arg); + VALUE value = r_object(arg); + rb_hash_aset(v, key, value); + arg->readable -= 2; + } + arg->readable += 2; + if (type == TYPE_HASH_DEF) { + RHASH_SET_IFNONE(v, r_object(arg)); + } v = r_leave(v, arg, partial); - } - break; + } + break; case TYPE_STRUCT: - { - VALUE mem, values; - long i; - VALUE slot; - st_index_t idx = r_prepare(arg); - VALUE klass = path2class(r_unique(arg)); - long len = r_long(arg); + { + VALUE mem, values; + long i; + VALUE slot; + st_index_t idx = r_prepare(arg); + VALUE klass = path2class(r_unique(arg)); + long len = r_long(arg); v = rb_obj_alloc(klass); - if (!RB_TYPE_P(v, T_STRUCT)) { - rb_raise(rb_eTypeError, "class %"PRIsVALUE" not a struct", rb_class_name(klass)); - } - mem = rb_struct_s_members(klass); + if (!RB_TYPE_P(v, T_STRUCT)) { + rb_raise(rb_eTypeError, "class %"PRIsVALUE" not a struct", rb_class_name(klass)); + } + mem = rb_struct_s_members(klass); if (RARRAY_LEN(mem) != len) { rb_raise(rb_eTypeError, "struct %"PRIsVALUE" not compatible (struct size differs)", rb_class_name(klass)); } - arg->readable += (len - 1) * 2; - v = r_entry0(v, idx, arg); - values = rb_ary_new2(len); - { - VALUE keywords = Qfalse; - if (RTEST(rb_struct_s_keyword_init(klass))) { - keywords = rb_hash_new(); - rb_ary_push(values, keywords); - } - - for (i=0; i<len; i++) { - VALUE n = rb_sym2str(RARRAY_AREF(mem, i)); - slot = r_symbol(arg); - - if (!rb_str_equal(n, slot)) { - rb_raise(rb_eTypeError, "struct %"PRIsVALUE" not compatible (:%"PRIsVALUE" for :%"PRIsVALUE")", - rb_class_name(klass), - slot, n); - } - if (keywords) { - rb_hash_aset(keywords, RARRAY_AREF(mem, i), r_object(arg)); - } - else { - rb_ary_push(values, r_object(arg)); - } - arg->readable -= 2; - } - } + arg->readable += (len - 1) * 2; + v = r_entry0(v, idx, arg); + values = rb_ary_new2(len); + { + VALUE keywords = Qfalse; + if (RTEST(rb_struct_s_keyword_init(klass))) { + keywords = rb_hash_new(); + rb_ary_push(values, keywords); + } + + for (i=0; i<len; i++) { + VALUE n = rb_sym2str(RARRAY_AREF(mem, i)); + slot = r_symbol(arg); + + if (!rb_str_equal(n, slot)) { + rb_raise(rb_eTypeError, "struct %"PRIsVALUE" not compatible (:%"PRIsVALUE" for :%"PRIsVALUE")", + rb_class_name(klass), + slot, n); + } + if (keywords) { + rb_hash_aset(keywords, RARRAY_AREF(mem, i), r_object(arg)); + } + else { + rb_ary_push(values, r_object(arg)); + } + arg->readable -= 2; + } + } rb_struct_initialize(v, values); v = r_leave(v, arg, partial); - arg->readable += 2; - } - break; + arg->readable += 2; + } + break; case TYPE_USERDEF: { - VALUE name = r_unique(arg); - VALUE klass = path2class(name); - VALUE data; - st_data_t d; - - if (!rb_obj_respond_to(klass, s_load, TRUE)) { - rb_raise(rb_eTypeError, "class %"PRIsVALUE" needs to have method `_load'", - name); - } - data = r_string(arg); - if (ivp) { - r_ivar(data, NULL, arg); - *ivp = FALSE; - } - v = load_funcall(arg, klass, s_load, 1, &data); - v = r_entry(v, arg); - if (st_lookup(compat_allocator_tbl, (st_data_t)rb_get_alloc_func(klass), &d)) { - marshal_compat_t *compat = (marshal_compat_t*)d; - v = compat->loader(klass, v); - } - if (!partial) v = r_post_proc(v, arg); - } + VALUE name = r_unique(arg); + VALUE klass = path2class(name); + VALUE data; + st_data_t d; + + if (!rb_obj_respond_to(klass, s_load, TRUE)) { + rb_raise(rb_eTypeError, "class %"PRIsVALUE" needs to have method `_load'", + name); + } + data = r_string(arg); + if (ivp) { + r_ivar(data, NULL, arg); + *ivp = FALSE; + } + v = load_funcall(arg, klass, s_load, 1, &data); + v = r_entry(v, arg); + if (st_lookup(compat_allocator_tbl, (st_data_t)rb_get_alloc_func(klass), &d)) { + marshal_compat_t *compat = (marshal_compat_t*)d; + v = compat->loader(klass, v); + } + if (!partial) v = r_post_proc(v, arg); + } break; case TYPE_USRMARSHAL: { - VALUE name = r_unique(arg); - VALUE klass = path2class(name); - VALUE oldclass = 0; - VALUE data; + VALUE name = r_unique(arg); + VALUE klass = path2class(name); + VALUE oldclass = 0; + VALUE data; - v = obj_alloc_by_klass(klass, arg, &oldclass); + v = obj_alloc_by_klass(klass, arg, &oldclass); + if (!NIL_P(extmod)) { + /* for the case marshal_load is overridden */ + append_extmod(v, extmod); + } + if (!rb_obj_respond_to(v, s_mload, TRUE)) { + rb_raise(rb_eTypeError, "instance of %"PRIsVALUE" needs to have method `marshal_load'", + name); + } + v = r_entry(v, arg); + data = r_object(arg); + load_funcall(arg, v, s_mload, 1, &data); + v = r_fixup_compat(v, arg); + v = r_copy_ivar(v, data); + v = r_post_proc(v, arg); if (!NIL_P(extmod)) { - /* for the case marshal_load is overridden */ - append_extmod(v, extmod); + if (oldclass) append_extmod(v, extmod); + rb_ary_clear(extmod); } - if (!rb_obj_respond_to(v, s_mload, TRUE)) { - rb_raise(rb_eTypeError, "instance of %"PRIsVALUE" needs to have method `marshal_load'", - name); - } - v = r_entry(v, arg); - data = r_object(arg); - load_funcall(arg, v, s_mload, 1, &data); - v = r_fixup_compat(v, arg); - v = r_copy_ivar(v, data); - v = r_post_proc(v, arg); - if (!NIL_P(extmod)) { - if (oldclass) append_extmod(v, extmod); - rb_ary_clear(extmod); - } - } + } break; case TYPE_OBJECT: - { - st_index_t idx = r_prepare(arg); + { + st_index_t idx = r_prepare(arg); v = obj_alloc_by_path(r_unique(arg), arg); - if (!RB_TYPE_P(v, T_OBJECT)) { - rb_raise(rb_eArgError, "dump format error"); - } - v = r_entry0(v, idx, arg); - r_ivar(v, NULL, arg); - v = r_leave(v, arg, partial); - } - break; + if (!RB_TYPE_P(v, T_OBJECT)) { + rb_raise(rb_eArgError, "dump format error"); + } + v = r_entry0(v, idx, arg); + r_ivar(v, NULL, arg); + v = r_leave(v, arg, partial); + } + break; case TYPE_DATA: - { - VALUE name = r_unique(arg); - VALUE klass = path2class(name); - VALUE oldclass = 0; - VALUE r; - - v = obj_alloc_by_klass(klass, arg, &oldclass); - if (!RB_TYPE_P(v, T_DATA)) { - rb_raise(rb_eArgError, "dump format error"); - } - v = r_entry(v, arg); - if (!rb_obj_respond_to(v, s_load_data, TRUE)) { - rb_raise(rb_eTypeError, - "class %"PRIsVALUE" needs to have instance method `_load_data'", - name); - } - r = r_object0(arg, partial, 0, extmod); - load_funcall(arg, v, s_load_data, 1, &r); - v = r_leave(v, arg, partial); - } - break; + { + VALUE name = r_unique(arg); + VALUE klass = path2class(name); + VALUE oldclass = 0; + VALUE r; + + v = obj_alloc_by_klass(klass, arg, &oldclass); + if (!RB_TYPE_P(v, T_DATA)) { + rb_raise(rb_eArgError, "dump format error"); + } + v = r_entry(v, arg); + if (!rb_obj_respond_to(v, s_load_data, TRUE)) { + rb_raise(rb_eTypeError, + "class %"PRIsVALUE" needs to have instance method `_load_data'", + name); + } + r = r_object0(arg, partial, 0, extmod); + load_funcall(arg, v, s_load_data, 1, &r); + v = r_leave(v, arg, partial); + } + break; case TYPE_MODULE_OLD: { - VALUE str = r_bytes(arg); + VALUE str = r_bytes(arg); - v = rb_path_to_class(str); - prohibit_ivar("class/module", str); - v = r_entry(v, arg); + v = rb_path_to_class(str); + prohibit_ivar("class/module", str); + v = r_entry(v, arg); v = r_leave(v, arg, partial); - } - break; + } + break; case TYPE_CLASS: { - VALUE str = r_bytes(arg); + VALUE str = r_bytes(arg); - v = path2class(str); - prohibit_ivar("class", str); - v = r_entry(v, arg); + v = path2class(str); + prohibit_ivar("class", str); + v = r_entry(v, arg); v = r_leave(v, arg, partial); - } - break; + } + break; case TYPE_MODULE: { - VALUE str = r_bytes(arg); + VALUE str = r_bytes(arg); - v = path2module(str); - prohibit_ivar("module", str); - v = r_entry(v, arg); + v = path2module(str); + prohibit_ivar("module", str); + v = r_entry(v, arg); v = r_leave(v, arg, partial); - } - break; + } + break; case TYPE_SYMBOL: - if (ivp) { - v = r_symreal(arg, *ivp); - *ivp = FALSE; - } - else { - v = r_symreal(arg, 0); - } - v = rb_str_intern(v); - v = r_leave(v, arg, partial); - break; + if (ivp) { + v = r_symreal(arg, *ivp); + *ivp = FALSE; + } + else { + v = r_symreal(arg, 0); + } + v = rb_str_intern(v); + v = r_leave(v, arg, partial); + break; case TYPE_SYMLINK: - v = rb_str_intern(r_symlink(arg)); - break; + v = rb_str_intern(r_symlink(arg)); + break; default: - rb_raise(rb_eArgError, "dump format error(0x%x)", type); - break; + rb_raise(rb_eArgError, "dump format error(0x%x)", type); + break; } if (v == Qundef) { - rb_raise(rb_eArgError, "dump format error (bad link)"); + rb_raise(rb_eArgError, "dump format error (bad link)"); } return v; @@ -2185,8 +2185,8 @@ static void clear_load_arg(struct load_arg *arg) { if (arg->buf) { - xfree(arg->buf); - arg->buf = 0; + xfree(arg->buf); + arg->buf = 0; } arg->buflen = 0; arg->offset = 0; @@ -2199,8 +2199,8 @@ clear_load_arg(struct load_arg *arg) st_free_table(arg->partial_objects); arg->partial_objects = 0; if (arg->compat_tbl) { - st_free_table(arg->compat_tbl); - arg->compat_tbl = 0; + st_free_table(arg->compat_tbl); + arg->compat_tbl = 0; } } @@ -2214,13 +2214,13 @@ rb_marshal_load_with_proc(VALUE port, VALUE proc, bool freeze) v = rb_check_string_type(port); if (!NIL_P(v)) { - port = v; + port = v; } else if (rb_respond_to(port, s_getbyte) && rb_respond_to(port, s_read)) { - rb_check_funcall(port, s_binmode, 0, 0); + rb_check_funcall(port, s_binmode, 0, 0); } else { - io_needed(); + io_needed(); } wrapper = TypedData_Make_Struct(0, struct load_arg, &load_arg_data, arg); arg->src = port; @@ -2234,22 +2234,22 @@ rb_marshal_load_with_proc(VALUE port, VALUE proc, bool freeze) arg->freeze = freeze; if (NIL_P(v)) - arg->buf = xmalloc(BUFSIZ); + arg->buf = xmalloc(BUFSIZ); else - arg->buf = 0; + arg->buf = 0; major = r_byte(arg); minor = r_byte(arg); if (major != MARSHAL_MAJOR || minor > MARSHAL_MINOR) { - clear_load_arg(arg); - rb_raise(rb_eTypeError, "incompatible marshal file format (can't be read)\n\ + clear_load_arg(arg); + rb_raise(rb_eTypeError, "incompatible marshal file format (can't be read)\n\ \tformat version %d.%d required; %d.%d given", - MARSHAL_MAJOR, MARSHAL_MINOR, major, minor); + MARSHAL_MAJOR, MARSHAL_MINOR, major, minor); } if (RTEST(ruby_verbose) && minor != MARSHAL_MINOR) { - rb_warn("incompatible marshal file format (can be read)\n\ + rb_warn("incompatible marshal file format (can be read)\n\ \tformat version %d.%d required; %d.%d given", - MARSHAL_MAJOR, MARSHAL_MINOR, major, minor); + MARSHAL_MAJOR, MARSHAL_MINOR, major, minor); } if (!NIL_P(proc)) arg->proc = proc; @@ -2414,7 +2414,7 @@ compat_allocator_table(void) #undef RUBY_UNTYPED_DATA_WARNING #define RUBY_UNTYPED_DATA_WARNING 0 compat_allocator_tbl_wrapper = - Data_Wrap_Struct(0, mark_marshal_compat_t, 0, compat_allocator_tbl); + Data_Wrap_Struct(0, mark_marshal_compat_t, 0, compat_allocator_tbl); rb_gc_register_mark_object(compat_allocator_tbl_wrapper); return compat_allocator_tbl; } |