1031 lines
18 KiB
C++
1031 lines
18 KiB
C++
/*
|
|
** marshal.cpp
|
|
**
|
|
** This file is part of mkxp.
|
|
**
|
|
** Copyright (C) 2013 Jonas Kulla <Nyocurio@gmail.com>
|
|
**
|
|
** mkxp is free software: you can redistribute it and/or modify
|
|
** it under the terms of the GNU General Public License as published by
|
|
** the Free Software Foundation, either version 2 of the License, or
|
|
** (at your option) any later version.
|
|
**
|
|
** mkxp is distributed in the hope that it will be useful,
|
|
** but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
** GNU General Public License for more details.
|
|
**
|
|
** You should have received a copy of the GNU General Public License
|
|
** along with mkxp. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "marshal.h"
|
|
|
|
#include <mruby/array.h>
|
|
#include <mruby/hash.h>
|
|
#include <mruby/khash.h>
|
|
#include <mruby/string.h>
|
|
#include <mruby/variable.h>
|
|
#include <mruby/class.h>
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <stdint.h>
|
|
#include <vector>
|
|
|
|
#include "../binding-util.h"
|
|
#include "file.h"
|
|
#include "rwmem.h"
|
|
#include "exception.h"
|
|
#include "boost-hash.h"
|
|
#include "debugwriter.h"
|
|
|
|
#include <SDL_timer.h>
|
|
|
|
|
|
#define MARSHAL_MAJOR 4
|
|
#define MARSHAL_MINOR 8
|
|
|
|
#define TYPE_NIL '0'
|
|
#define TYPE_TRUE 'T'
|
|
#define TYPE_FALSE 'F'
|
|
#define TYPE_FIXNUM 'i'
|
|
|
|
#define TYPE_OBJECT 'o'
|
|
//#define TYPE_DATA 'd'
|
|
#define TYPE_USERDEF 'u'
|
|
#define TYPE_USRMARSHAL 'U'
|
|
#define TYPE_FLOAT 'f'
|
|
//#define TYPE_BIGNUM 'l'
|
|
#define TYPE_STRING '"'
|
|
//#define TYPE_REGEXP '/'
|
|
#define TYPE_ARRAY '['
|
|
#define TYPE_HASH '{'
|
|
#define TYPE_HASH_DEF '}'
|
|
#define TYPE_MODULE_OLD 'M'
|
|
#define TYPE_IVAR 'I'
|
|
#define TYPE_LINK '@'
|
|
|
|
#define TYPE_SYMBOL ':'
|
|
#define TYPE_SYMLINK ';'
|
|
|
|
// FIXME make these dynamically allocatd, per MarshalContext
|
|
static char gpbuffer[512];
|
|
|
|
inline size_t hash_value(mrb_value key)
|
|
{
|
|
return boost::hash_value(key.value.p);
|
|
}
|
|
|
|
inline bool operator==(mrb_value v1, mrb_value v2)
|
|
{
|
|
if (mrb_type(v1) != mrb_type(v2))
|
|
return false;
|
|
|
|
switch (mrb_type(v1))
|
|
{
|
|
case MRB_TT_TRUE:
|
|
return true;
|
|
|
|
case MRB_TT_FALSE:
|
|
case MRB_TT_FIXNUM:
|
|
return (v1.value.i == v2.value.i);
|
|
case MRB_TT_SYMBOL:
|
|
return (v1.value.sym == v2.value.sym);
|
|
|
|
case MRB_TT_FLOAT:
|
|
return (mrb_float(v1) == mrb_float(v2));
|
|
|
|
default:
|
|
return (mrb_ptr(v1) == mrb_ptr(v2));
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
struct LinkBuffer
|
|
{
|
|
/* Key: val, Value: idx in vec */
|
|
BoostHash<T, int> hash;
|
|
std::vector<T> vec;
|
|
|
|
bool contains(T value)
|
|
{
|
|
return hash.contains(value);
|
|
}
|
|
|
|
bool containsIdx(size_t idx)
|
|
{
|
|
if (vec.empty())
|
|
return false;
|
|
|
|
return (idx < vec.size());
|
|
}
|
|
|
|
int add(T value)
|
|
{
|
|
int idx = vec.size();
|
|
|
|
hash.insert(value, idx);
|
|
vec.push_back(value);
|
|
|
|
return idx;
|
|
}
|
|
|
|
T lookup(size_t idx)
|
|
{
|
|
return vec[idx];
|
|
}
|
|
|
|
int lookupIdx(T value)
|
|
{
|
|
return hash[value];
|
|
}
|
|
};
|
|
|
|
struct MarshalContext
|
|
{
|
|
SDL_RWops *ops;
|
|
mrb_state *mrb;
|
|
mrb_int limit;
|
|
|
|
/* For Symlinks/Links */
|
|
LinkBuffer<mrb_sym> symbols;
|
|
LinkBuffer<mrb_value> objects;
|
|
|
|
char readByte()
|
|
{
|
|
char byte;
|
|
int result = SDL_RWread(ops, &byte, 1, 1);
|
|
|
|
if (result < 1)
|
|
throw Exception(Exception::ArgumentError, "dump format error");
|
|
|
|
return byte;
|
|
}
|
|
|
|
void readData(char *dest, int len)
|
|
{
|
|
int result = SDL_RWread(ops, dest, 1, len);
|
|
|
|
if (result < len)
|
|
throw Exception(Exception::ArgumentError, "dump format error");
|
|
}
|
|
|
|
void writeByte(char byte)
|
|
{
|
|
int result = SDL_RWwrite(ops, &byte, 1, 1);
|
|
|
|
if (result < 1) // FIXME not sure what the correct error would be here
|
|
throw Exception(Exception::IOError, "dump writing error");
|
|
}
|
|
|
|
void writeData(const char *data, int len)
|
|
{
|
|
int result = SDL_RWwrite(ops, data, 1, len);
|
|
|
|
if (result < len) // FIXME not sure what the correct error would be here
|
|
throw Exception(Exception::IOError, "dump writing error");
|
|
}
|
|
};
|
|
|
|
|
|
static int
|
|
read_fixnum(MarshalContext *ctx)
|
|
{
|
|
char head = ctx->readByte();
|
|
|
|
if (head == 0)
|
|
return 0;
|
|
else if (head > 5)
|
|
return head - 5;
|
|
else if (head < -4)
|
|
return head + 5;
|
|
|
|
int pos = (head > 0);
|
|
int len = pos ? head : head * -1;
|
|
|
|
char n1, n2, n3, n4;
|
|
|
|
if (pos)
|
|
n2 = n3 = n4 = 0;
|
|
else
|
|
n2 = n3 = n4 = 0xFF;
|
|
|
|
n1 = ctx->readByte();
|
|
if (len >= 2)
|
|
n2 = ctx->readByte();
|
|
if (len >= 3)
|
|
n3 = ctx->readByte();
|
|
if (len >= 4)
|
|
n4 = ctx->readByte();
|
|
|
|
int result = ((0xFF << 0x00) & (n1 << 0x00))
|
|
| ((0xFF << 0x08) & (n2 << 0x08))
|
|
| ((0xFF << 0x10) & (n3 << 0x10))
|
|
| ((0xFF << 0x18) & (n4 << 0x18));
|
|
|
|
// if (!pos)
|
|
// result = -((result ^ 0xFFFFFFFF) + 1);
|
|
|
|
return result;
|
|
}
|
|
|
|
static float
|
|
read_float(MarshalContext *ctx)
|
|
{
|
|
int len = read_fixnum(ctx);
|
|
|
|
ctx->readData(gpbuffer, len);
|
|
gpbuffer[len] = '\0';
|
|
|
|
return strtof(gpbuffer, 0);
|
|
}
|
|
|
|
static char *
|
|
read_string(MarshalContext *ctx)
|
|
{
|
|
int len = read_fixnum(ctx);
|
|
|
|
ctx->readData(gpbuffer, len);
|
|
gpbuffer[len] = '\0';
|
|
|
|
return gpbuffer;
|
|
}
|
|
|
|
static mrb_value
|
|
read_string_value(MarshalContext *ctx)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
int len = read_fixnum(ctx);
|
|
|
|
mrb_value str = mrb_str_new(mrb, 0, len);
|
|
char *ptr = RSTR_PTR(RSTRING(str));
|
|
|
|
ctx->readData(ptr, len);
|
|
ptr[len] = '\0';
|
|
|
|
return str;
|
|
}
|
|
|
|
static mrb_value read_value(MarshalContext *ctx);
|
|
|
|
static mrb_value
|
|
read_array(MarshalContext *ctx)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
int len = read_fixnum(ctx);
|
|
int i;
|
|
|
|
mrb_value array = mrb_ary_new_capa(mrb, len);
|
|
|
|
ctx->objects.add(array);
|
|
|
|
for (i = 0; i < len; ++i)
|
|
{
|
|
mrb_value val = read_value(ctx);
|
|
mrb_ary_set(mrb, array, i, val);
|
|
}
|
|
|
|
return array;
|
|
}
|
|
|
|
static mrb_value
|
|
read_hash(MarshalContext *ctx)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
int len = read_fixnum(ctx);
|
|
int i;
|
|
|
|
mrb_value hash = mrb_hash_new_capa(mrb, len);
|
|
|
|
ctx->objects.add(hash);
|
|
|
|
for (i = 0; i < len; ++i)
|
|
{
|
|
mrb_value key = read_value(ctx);
|
|
mrb_value val = read_value(ctx);
|
|
|
|
mrb_hash_set(mrb, hash, key, val);
|
|
}
|
|
|
|
return hash;
|
|
}
|
|
|
|
static mrb_sym
|
|
read_symbol(MarshalContext *ctx)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
|
|
mrb_sym symbol = mrb_intern_cstr(mrb, read_string(ctx));
|
|
ctx->symbols.add(symbol);
|
|
|
|
return symbol;
|
|
}
|
|
|
|
static mrb_sym
|
|
read_symlink(MarshalContext *ctx)
|
|
{
|
|
int idx = read_fixnum(ctx);
|
|
|
|
if (!ctx->symbols.containsIdx(idx))
|
|
throw Exception(Exception::ArgumentError, "bad symlink");
|
|
|
|
return ctx->symbols.lookup(idx);
|
|
}
|
|
|
|
static mrb_value
|
|
read_link(MarshalContext *ctx)
|
|
{
|
|
int idx = read_fixnum(ctx);
|
|
|
|
if (!ctx->objects.containsIdx(idx))
|
|
throw Exception(Exception::ArgumentError, "dump format error (unlinked)");
|
|
|
|
return ctx->objects.lookup(idx);
|
|
}
|
|
|
|
mrb_value
|
|
read_instance_var(MarshalContext *ctx)
|
|
{
|
|
mrb_value obj = read_value(ctx);
|
|
|
|
int iv_count = read_fixnum(ctx);
|
|
int i;
|
|
|
|
ctx->objects.add(obj);
|
|
|
|
for (i = 0; i < iv_count; ++i)
|
|
{
|
|
mrb_value iv_name = read_value(ctx);
|
|
mrb_value iv_value = read_value(ctx);
|
|
(void)iv_name;
|
|
(void)iv_value;
|
|
|
|
// Instance vars for String/Array not supported yet
|
|
// mrb_iv_set(mrb, obj, mrb_symbol(iv_name), iv_value);
|
|
}
|
|
|
|
return obj;
|
|
}
|
|
|
|
static struct RClass *
|
|
mrb_class_from_path(mrb_state *mrb, mrb_value path)
|
|
{
|
|
char *path_s;
|
|
switch (path.tt)
|
|
{
|
|
case MRB_TT_SYMBOL :
|
|
path_s = (char*) mrb_sym2name(mrb, mrb_symbol(path));
|
|
break;
|
|
case MRB_TT_STRING :
|
|
path_s = (char*) RSTRING_PTR(path);
|
|
break;
|
|
default:
|
|
throw Exception(Exception::ArgumentError, "dump format error for symbol");
|
|
}
|
|
|
|
/* If a symbol contains any colons, mrb_sym2name
|
|
* will return the string with "s around it
|
|
* (e.g. :Mod::Class => :"Mod::Class"),
|
|
* so we need to skip those. */
|
|
if (path_s[0] == '\"')
|
|
path_s++;
|
|
|
|
char *p, *pbgn;
|
|
mrb_value klass = mrb_obj_value(mrb->object_class);
|
|
|
|
p = pbgn = path_s;
|
|
|
|
while (*p && *p != '\"')
|
|
{
|
|
while (*p && *p != ':' && *p != '\"') p++;
|
|
|
|
mrb_sym sym = mrb_intern(mrb, pbgn, p-pbgn);
|
|
klass = mrb_const_get(mrb, klass, sym);
|
|
|
|
if (p[0] == ':')
|
|
{
|
|
if (p[1] != ':')
|
|
return 0;
|
|
p += 2;
|
|
pbgn = p;
|
|
}
|
|
}
|
|
|
|
return (struct RClass*) mrb_obj_ptr(klass);
|
|
}
|
|
|
|
static mrb_value
|
|
read_object(MarshalContext *ctx)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
mrb_value class_path = read_value(ctx);
|
|
|
|
struct RClass *klass =
|
|
mrb_class_from_path(mrb, class_path);
|
|
|
|
mrb_value obj = mrb_obj_value(mrb_obj_alloc(mrb, MRB_TT_OBJECT, klass));
|
|
|
|
ctx->objects.add(obj);
|
|
|
|
int iv_count = read_fixnum(ctx);
|
|
int i;
|
|
|
|
for (i = 0; i < iv_count; ++i)
|
|
{
|
|
mrb_value iv_name = read_value(ctx);
|
|
mrb_value iv_value = read_value(ctx);
|
|
|
|
mrb_obj_iv_set(mrb, mrb_obj_ptr(obj),
|
|
mrb_symbol(iv_name), iv_value);
|
|
}
|
|
|
|
return obj;
|
|
}
|
|
|
|
static mrb_value
|
|
read_userdef(MarshalContext *ctx)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
mrb_value class_path = read_value(ctx);
|
|
|
|
struct RClass *klass =
|
|
mrb_class_from_path(mrb, class_path);
|
|
|
|
/* Should check here if klass implements '_load()' */
|
|
if (!mrb_obj_respond_to(mrb, mrb_class(mrb, mrb_obj_value(klass)), mrb_intern_cstr(mrb, "_load")))
|
|
throw Exception(Exception::TypeError,
|
|
"class %s needs to have method '_load'",
|
|
RSTRING_PTR(class_path));
|
|
|
|
mrb_value data = read_string_value(ctx);
|
|
mrb_value obj = mrb_funcall(mrb, mrb_obj_value(klass), "_load", 1, data);
|
|
|
|
return obj;
|
|
}
|
|
|
|
static int maxArena = 0;
|
|
|
|
static mrb_value
|
|
read_value(MarshalContext *ctx)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
char type = ctx->readByte();
|
|
mrb_value value;
|
|
if (mrb->arena_idx > maxArena)
|
|
maxArena = mrb->arena_idx;
|
|
|
|
int arena = mrb_gc_arena_save(mrb);
|
|
|
|
switch (type)
|
|
{
|
|
case TYPE_NIL :
|
|
value = mrb_nil_value();
|
|
break;
|
|
|
|
case TYPE_TRUE :
|
|
value = mrb_true_value();
|
|
break;
|
|
|
|
case TYPE_FALSE :
|
|
value = mrb_false_value();
|
|
break;
|
|
|
|
case TYPE_FIXNUM :
|
|
value = mrb_fixnum_value(read_fixnum(ctx));
|
|
break;
|
|
|
|
case TYPE_FLOAT :
|
|
value = mrb_float_value(mrb, read_float(ctx));
|
|
ctx->objects.add(value);
|
|
break;
|
|
|
|
case TYPE_STRING :
|
|
value = read_string_value(ctx);
|
|
ctx->objects.add(value);
|
|
break;
|
|
|
|
case TYPE_ARRAY :
|
|
value = read_array(ctx);
|
|
break;
|
|
|
|
case TYPE_HASH :
|
|
value = read_hash(ctx);
|
|
break;
|
|
|
|
case TYPE_SYMBOL :
|
|
value = mrb_symbol_value(read_symbol(ctx));
|
|
break;
|
|
|
|
case TYPE_SYMLINK :
|
|
value = mrb_symbol_value(read_symlink(ctx));
|
|
break;
|
|
|
|
case TYPE_OBJECT :
|
|
value = read_object(ctx);
|
|
break;
|
|
|
|
case TYPE_IVAR :
|
|
value = read_instance_var(ctx);
|
|
break;
|
|
|
|
case TYPE_LINK :
|
|
value = read_link(ctx);
|
|
break;
|
|
|
|
case TYPE_USERDEF :
|
|
value = read_userdef(ctx);
|
|
ctx->objects.add(value);
|
|
break;
|
|
|
|
default :
|
|
throw Exception(Exception::MKXPError, "Marshal.load: unsupported value type '%c'",
|
|
(char) type);
|
|
}
|
|
|
|
mrb_gc_arena_restore(mrb, arena);
|
|
|
|
return value;
|
|
}
|
|
|
|
static void
|
|
write_fixnum(MarshalContext *ctx, int value)
|
|
{
|
|
if (value == 0)
|
|
{
|
|
ctx->writeByte(0);
|
|
return;
|
|
}
|
|
else if (value > 0 && value < 123)
|
|
{
|
|
ctx->writeByte((char) value + 5);
|
|
return;
|
|
}
|
|
else if (value < 0 && value > -124)
|
|
{
|
|
ctx->writeByte((char) value - 5);
|
|
return;
|
|
}
|
|
|
|
char len;
|
|
|
|
if (value > 0)
|
|
{
|
|
/* Positive number */
|
|
if (value <= 0x7F)
|
|
{
|
|
/* 1 byte wide */
|
|
len = 1;
|
|
}
|
|
else if (value <= 0x7FFF)
|
|
{
|
|
/* 2 bytes wide */
|
|
len = 2;
|
|
}
|
|
else if (value <= 0x7FFFFF)
|
|
{
|
|
/* 3 bytes wide */
|
|
len = 3;
|
|
}
|
|
else
|
|
{
|
|
/* 4 bytes wide */
|
|
len = 4;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* Negative number */
|
|
if (value >= (int)0x80)
|
|
{
|
|
/* 1 byte wide */
|
|
len = -1;
|
|
}
|
|
else if (value >= (int)0x8000)
|
|
{
|
|
/* 2 bytes wide */
|
|
len = -2;
|
|
}
|
|
else if (value <= (int)0x800000)
|
|
{
|
|
/* 3 bytes wide */
|
|
len = -3;
|
|
}
|
|
else
|
|
{
|
|
/* 4 bytes wide */
|
|
len = -4;
|
|
}
|
|
}
|
|
|
|
/* Write length */
|
|
ctx->writeByte(len);
|
|
|
|
/* Write bytes */
|
|
if (len >= 1 || len <= -1)
|
|
{
|
|
ctx->writeByte((value & 0x000000FF) >> 0x00);
|
|
}
|
|
if (len >= 2 || len <= -2)
|
|
{
|
|
ctx->writeByte((value & 0x0000FF00) >> 0x08);
|
|
}
|
|
if (len >= 3 || len <= -3)
|
|
{
|
|
ctx->writeByte((value & 0x00FF0000) >> 0x10);
|
|
}
|
|
if (len == 4 || len == -4)
|
|
{
|
|
ctx->writeByte((value & 0xFF000000) >> 0x18);
|
|
}
|
|
}
|
|
|
|
static void
|
|
write_float(MarshalContext *ctx, float value)
|
|
{
|
|
sprintf(gpbuffer, "%.16g", value);
|
|
|
|
int len = strlen(gpbuffer);
|
|
write_fixnum(ctx, len);
|
|
ctx->writeData(gpbuffer, len);
|
|
}
|
|
|
|
static void
|
|
write_string(MarshalContext *ctx, const char *string)
|
|
{
|
|
int len = strlen(string);
|
|
|
|
write_fixnum(ctx, len);
|
|
ctx->writeData(string, len);
|
|
}
|
|
|
|
static void
|
|
write_string_value(MarshalContext *ctx, mrb_value string)
|
|
{
|
|
int len = RSTRING_LEN(string);
|
|
const char *ptr = RSTRING_PTR(string);
|
|
|
|
write_fixnum(ctx, len);
|
|
ctx->writeData(ptr, len);
|
|
}
|
|
|
|
static void write_value(MarshalContext *, mrb_value);
|
|
|
|
static void
|
|
write_array(MarshalContext *ctx, mrb_value array)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
int len = mrb_ary_len(mrb, array);
|
|
write_fixnum(ctx, len);
|
|
|
|
int i;
|
|
for (i = 0; i < len; ++i)
|
|
{
|
|
mrb_value value = mrb_ary_entry(array, i);
|
|
write_value(ctx, value);
|
|
}
|
|
}
|
|
|
|
KHASH_DECLARE(ht, mrb_value, mrb_value, 1)
|
|
|
|
static void
|
|
write_hash(MarshalContext *ctx, mrb_value hash)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
int len = 0;
|
|
|
|
kh_ht *h = mrb_hash_tbl(mrb, hash);
|
|
if (h)
|
|
len = kh_size(h);
|
|
|
|
write_fixnum(ctx, len);
|
|
|
|
for (khiter_t k = kh_begin(h); k != kh_end(h); ++k)
|
|
{
|
|
if (!kh_exist(h, k))
|
|
continue;
|
|
|
|
write_value(ctx, kh_key(h, k));
|
|
write_value(ctx, kh_val(h, k));
|
|
}
|
|
}
|
|
|
|
static void
|
|
write_symbol(MarshalContext *ctx, mrb_value symbol)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
mrb_sym sym = mrb_symbol(symbol);
|
|
mrb_int len;
|
|
const char *p = mrb_sym2name_len(mrb, sym, &len);
|
|
|
|
write_string(ctx, p);
|
|
}
|
|
|
|
static void
|
|
write_object(MarshalContext *ctx, mrb_value object)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
struct RObject *o = mrb_obj_ptr(object);
|
|
|
|
mrb_value path = mrb_class_path(mrb, o->c);
|
|
write_value(ctx, mrb_str_intern(mrb, path));
|
|
|
|
mrb_value iv_ary = mrb_obj_instance_variables(mrb, object);
|
|
int iv_count = mrb_ary_len(mrb, iv_ary);
|
|
write_fixnum(ctx, iv_count);
|
|
|
|
int i;
|
|
for (i = 0; i < iv_count; ++i)
|
|
{
|
|
mrb_value iv_name = mrb_ary_entry(iv_ary, i);
|
|
mrb_value iv_value = mrb_obj_iv_get(mrb, o, mrb_symbol(iv_name));
|
|
|
|
write_value(ctx, iv_name);
|
|
write_value(ctx, iv_value);
|
|
}
|
|
}
|
|
|
|
static void
|
|
write_userdef(MarshalContext *ctx, mrb_value object)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
struct RObject *o = mrb_obj_ptr(object);
|
|
|
|
mrb_value path = mrb_class_path(mrb, o->c);
|
|
write_value(ctx, mrb_str_intern(mrb, path));
|
|
|
|
mrb_value dump = mrb_funcall(mrb, object, "_dump", 0);
|
|
write_string_value(ctx, dump);
|
|
}
|
|
|
|
static void
|
|
write_value(MarshalContext *ctx, mrb_value value)
|
|
{
|
|
mrb_state *mrb = ctx->mrb;
|
|
|
|
int arena = mrb_gc_arena_save(mrb);
|
|
|
|
if (ctx->objects.contains(value))
|
|
{
|
|
ctx->writeByte(TYPE_LINK);
|
|
write_fixnum(ctx, ctx->objects.lookupIdx(value));
|
|
mrb_gc_arena_restore(mrb, arena);
|
|
return;
|
|
}
|
|
|
|
switch (mrb_type(value))
|
|
{
|
|
case MRB_TT_TRUE :
|
|
ctx->writeByte(TYPE_TRUE);
|
|
break;
|
|
|
|
case MRB_TT_FALSE :
|
|
if (mrb_fixnum(value))
|
|
ctx->writeByte(TYPE_FALSE);
|
|
else
|
|
ctx->writeByte(TYPE_NIL);
|
|
|
|
break;
|
|
|
|
case MRB_TT_FIXNUM :
|
|
ctx->writeByte(TYPE_FIXNUM);
|
|
write_fixnum(ctx, mrb_fixnum(value));
|
|
|
|
break;
|
|
|
|
case MRB_TT_FLOAT :
|
|
ctx->writeByte(TYPE_FLOAT);
|
|
write_float(ctx, mrb_float(value));
|
|
ctx->objects.add(value);
|
|
|
|
break;
|
|
|
|
case MRB_TT_STRING :
|
|
ctx->objects.add(value);
|
|
ctx->writeByte(TYPE_STRING);
|
|
write_string_value(ctx, value);
|
|
|
|
break;
|
|
|
|
case MRB_TT_ARRAY :
|
|
ctx->objects.add(value);
|
|
ctx->writeByte(TYPE_ARRAY);
|
|
write_array(ctx, value);
|
|
|
|
break;
|
|
|
|
case MRB_TT_HASH :
|
|
ctx->objects.add(value);
|
|
ctx->writeByte(TYPE_HASH);
|
|
write_hash(ctx, value);
|
|
|
|
break;
|
|
|
|
case MRB_TT_SYMBOL :
|
|
if (ctx->symbols.contains(mrb_symbol(value)))
|
|
{
|
|
ctx->writeByte(TYPE_SYMLINK);
|
|
write_fixnum(ctx, ctx->symbols.lookupIdx(mrb_symbol(value)));
|
|
ctx->symbols.add(mrb_symbol(value));
|
|
}
|
|
else
|
|
{
|
|
ctx->writeByte(TYPE_SYMBOL);
|
|
write_symbol(ctx, value);
|
|
}
|
|
|
|
break;
|
|
|
|
/* Objects seem to have wrong tt */
|
|
case MRB_TT_CLASS :
|
|
case MRB_TT_OBJECT :
|
|
ctx->objects.add(value);
|
|
|
|
if (mrb_obj_respond_to(mrb, mrb_obj_ptr(value)->c,
|
|
mrb_intern_lit(mrb, "_dump")))
|
|
{
|
|
ctx->writeByte(TYPE_USERDEF);
|
|
write_userdef(ctx, value);
|
|
}
|
|
else
|
|
{
|
|
ctx->writeByte(TYPE_OBJECT);
|
|
write_object(ctx, value);
|
|
}
|
|
|
|
break;
|
|
|
|
default :
|
|
printf("Warning! Could not write value type '%c'(%d)\n", // FIXME not sure if raise or just print error
|
|
mrb_type(value), mrb_type(value));
|
|
fflush(stdout);
|
|
}
|
|
|
|
mrb_gc_arena_restore(mrb, arena);
|
|
}
|
|
|
|
static void
|
|
writeMarshalHeader(MarshalContext *ctx)
|
|
{
|
|
ctx->writeByte(MARSHAL_MAJOR);
|
|
ctx->writeByte(MARSHAL_MINOR);
|
|
}
|
|
|
|
static void
|
|
verifyMarshalHeader(MarshalContext *ctx)
|
|
{
|
|
char maj = ctx->readByte();
|
|
char min = ctx->readByte();
|
|
|
|
if (maj != MARSHAL_MAJOR || min != MARSHAL_MINOR)
|
|
throw Exception(Exception::TypeError, "incompatible marshal file format (can't be read)");
|
|
}
|
|
|
|
MRB_FUNCTION(marshalDump)
|
|
{
|
|
mrb_value val;
|
|
mrb_value port = mrb_nil_value();
|
|
mrb_int limit = 100;
|
|
|
|
mrb_get_args(mrb, "o|oi", &val, &port, &limit);
|
|
|
|
SDL_RWops *ops;
|
|
|
|
bool havePort = mrb_type(port) == MRB_TT_OBJECT;
|
|
bool dumpString = false;
|
|
mrb_value ret = port;
|
|
|
|
if (havePort)
|
|
{
|
|
FileImpl *file = getPrivateDataCheck<FileImpl>(mrb, port, FileType);
|
|
ops = file->ops;
|
|
}
|
|
else
|
|
{
|
|
ops = SDL_AllocRW();
|
|
RWMemOpen(ops);
|
|
dumpString = true;
|
|
}
|
|
|
|
try
|
|
{
|
|
MarshalContext ctx;
|
|
ctx.mrb = mrb;
|
|
ctx.limit = limit;
|
|
ctx.ops = ops;
|
|
|
|
writeMarshalHeader(&ctx);
|
|
write_value(&ctx, val);
|
|
}
|
|
catch (const Exception &e)
|
|
{
|
|
if (dumpString)
|
|
{
|
|
SDL_RWclose(ops);
|
|
SDL_FreeRW(ops);
|
|
}
|
|
|
|
raiseMrbExc(mrb, e);
|
|
}
|
|
|
|
if (dumpString)
|
|
{
|
|
int dataSize = RWMemGetData(ops, 0);
|
|
|
|
ret = mrb_str_new(mrb, 0, dataSize);
|
|
RWMemGetData(ops, RSTRING_PTR(ret));
|
|
|
|
SDL_RWclose(ops);
|
|
SDL_FreeRW(ops);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
MRB_FUNCTION(marshalLoad)
|
|
{
|
|
mrb_value port;
|
|
|
|
mrb_get_args(mrb, "o", &port);
|
|
|
|
SDL_RWops *ops;
|
|
|
|
bool freeOps = false;
|
|
|
|
if (mrb_type(port) == MRB_TT_OBJECT)
|
|
{
|
|
FileImpl *file = getPrivateDataCheck<FileImpl>(mrb, port, FileType);
|
|
ops = file->ops;
|
|
}
|
|
else if (mrb_string_p(port))
|
|
{
|
|
ops = SDL_RWFromConstMem(RSTRING_PTR(port), RSTRING_LEN(port));
|
|
freeOps = true;
|
|
}
|
|
else
|
|
{
|
|
Debug() << "FIXME: Marshal.load: generic IO port not implemented";
|
|
return mrb_nil_value();
|
|
}
|
|
|
|
mrb_value val;
|
|
|
|
try
|
|
{
|
|
MarshalContext ctx;
|
|
ctx.mrb = mrb;
|
|
ctx.ops = ops;
|
|
|
|
verifyMarshalHeader(&ctx);
|
|
val = read_value(&ctx);
|
|
}
|
|
catch (const Exception &e)
|
|
{
|
|
if (freeOps)
|
|
SDL_RWclose(ops);
|
|
}
|
|
|
|
if (freeOps)
|
|
SDL_RWclose(ops);
|
|
|
|
return val;
|
|
}
|
|
|
|
void
|
|
marshalBindingInit(mrb_state *mrb)
|
|
{
|
|
RClass *module = mrb_define_module(mrb, "Marshal");
|
|
|
|
mrb_define_module_function(mrb, module, "dump", marshalDump, MRB_ARGS_REQ(1) | MRB_ARGS_OPT(2));
|
|
mrb_define_module_function(mrb, module, "load", marshalLoad, MRB_ARGS_REQ(1));
|
|
}
|
|
|
|
/* Exceptions from these internal functions will be caught higher up */
|
|
void
|
|
marshalDumpInt(mrb_state *mrb, SDL_RWops *ops, mrb_value val)
|
|
{
|
|
MarshalContext ctx;
|
|
ctx.mrb = mrb;
|
|
ctx.ops = ops;
|
|
ctx.limit = 100;
|
|
|
|
writeMarshalHeader(&ctx);
|
|
write_value(&ctx, val);
|
|
}
|
|
|
|
mrb_value
|
|
marshalLoadInt(mrb_state *mrb, SDL_RWops *ops)
|
|
{
|
|
MarshalContext ctx;
|
|
ctx.mrb = mrb;
|
|
ctx.ops = ops;
|
|
|
|
verifyMarshalHeader(&ctx);
|
|
|
|
mrb_value val = read_value(&ctx);
|
|
|
|
return val;
|
|
}
|
|
|