ccbd810934
Variants like dictionaries and arrays can have cyclic references, which
caused `encode_variant` to run an infinite recursion.
Instead of keeping a stack and looking for cyclic references which would
make serialization slower, this commit adds a `MAX_RECURSION_DEPTH`
constant to Variant, and have `encode_variant` keep track of the current
recursion depth, bailing when it's too high since this likely means a
cyclic reference has been encountered.
(cherry picked from commit 324636473a)
1341 lines
29 KiB
C++
1341 lines
29 KiB
C++
/*************************************************************************/
|
|
/* marshalls.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2021 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2021 Godot Engine contributors (cf. AUTHORS.md). */
|
|
/* */
|
|
/* Permission is hereby granted, free of charge, to any person obtaining */
|
|
/* a copy of this software and associated documentation files (the */
|
|
/* "Software"), to deal in the Software without restriction, including */
|
|
/* without limitation the rights to use, copy, modify, merge, publish, */
|
|
/* distribute, sublicense, and/or sell copies of the Software, and to */
|
|
/* permit persons to whom the Software is furnished to do so, subject to */
|
|
/* the following conditions: */
|
|
/* */
|
|
/* The above copyright notice and this permission notice shall be */
|
|
/* included in all copies or substantial portions of the Software. */
|
|
/* */
|
|
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
|
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
|
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
|
|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
|
/*************************************************************************/
|
|
|
|
#include "marshalls.h"
|
|
|
|
#include "core/os/keyboard.h"
|
|
#include "core/print_string.h"
|
|
#include "core/reference.h"
|
|
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
|
|
void EncodedObjectAsID::_bind_methods() {
|
|
ClassDB::bind_method(D_METHOD("set_object_id", "id"), &EncodedObjectAsID::set_object_id);
|
|
ClassDB::bind_method(D_METHOD("get_object_id"), &EncodedObjectAsID::get_object_id);
|
|
|
|
ADD_PROPERTY(PropertyInfo(Variant::INT, "object_id"), "set_object_id", "get_object_id");
|
|
}
|
|
|
|
void EncodedObjectAsID::set_object_id(ObjectID p_id) {
|
|
id = p_id;
|
|
}
|
|
|
|
ObjectID EncodedObjectAsID::get_object_id() const {
|
|
return id;
|
|
}
|
|
|
|
EncodedObjectAsID::EncodedObjectAsID() :
|
|
id(0) {
|
|
}
|
|
|
|
#define _S(a) ((int32_t)a)
|
|
#define ERR_FAIL_ADD_OF(a, b, err) ERR_FAIL_COND_V(_S(b) < 0 || _S(a) < 0 || _S(a) > INT_MAX - _S(b), err)
|
|
#define ERR_FAIL_MUL_OF(a, b, err) ERR_FAIL_COND_V(_S(a) < 0 || _S(b) <= 0 || _S(a) > INT_MAX / _S(b), err)
|
|
|
|
#define ENCODE_MASK 0xFF
|
|
#define ENCODE_FLAG_64 1 << 16
|
|
#define ENCODE_FLAG_OBJECT_AS_ID 1 << 16
|
|
|
|
static Error _decode_string(const uint8_t *&buf, int &len, int *r_len, String &r_string) {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
|
|
int32_t strlen = decode_uint32(buf);
|
|
int32_t pad = 0;
|
|
|
|
// Handle padding
|
|
if (strlen % 4) {
|
|
pad = 4 - strlen % 4;
|
|
}
|
|
|
|
buf += 4;
|
|
len -= 4;
|
|
|
|
// Ensure buffer is big enough
|
|
ERR_FAIL_ADD_OF(strlen, pad, ERR_FILE_EOF);
|
|
ERR_FAIL_COND_V(strlen < 0 || strlen + pad > len, ERR_FILE_EOF);
|
|
|
|
String str;
|
|
ERR_FAIL_COND_V(str.parse_utf8((const char *)buf, strlen), ERR_INVALID_DATA);
|
|
r_string = str;
|
|
|
|
// Add padding
|
|
strlen += pad;
|
|
|
|
// Update buffer pos, left data count, and return size
|
|
buf += strlen;
|
|
len -= strlen;
|
|
if (r_len) {
|
|
(*r_len) += 4 + strlen;
|
|
}
|
|
|
|
return OK;
|
|
}
|
|
|
|
Error decode_variant(Variant &r_variant, const uint8_t *p_buffer, int p_len, int *r_len, bool p_allow_objects) {
|
|
const uint8_t *buf = p_buffer;
|
|
int len = p_len;
|
|
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
|
|
uint32_t type = decode_uint32(buf);
|
|
|
|
ERR_FAIL_COND_V((type & ENCODE_MASK) >= Variant::VARIANT_MAX, ERR_INVALID_DATA);
|
|
|
|
buf += 4;
|
|
len -= 4;
|
|
if (r_len) {
|
|
*r_len = 4;
|
|
}
|
|
|
|
switch (type & ENCODE_MASK) {
|
|
case Variant::NIL: {
|
|
r_variant = Variant();
|
|
} break;
|
|
case Variant::BOOL: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
bool val = decode_uint32(buf);
|
|
r_variant = val;
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
} break;
|
|
case Variant::INT: {
|
|
if (type & ENCODE_FLAG_64) {
|
|
ERR_FAIL_COND_V(len < 8, ERR_INVALID_DATA);
|
|
int64_t val = decode_uint64(buf);
|
|
r_variant = val;
|
|
if (r_len) {
|
|
(*r_len) += 8;
|
|
}
|
|
|
|
} else {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t val = decode_uint32(buf);
|
|
r_variant = val;
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
}
|
|
|
|
} break;
|
|
case Variant::REAL: {
|
|
if (type & ENCODE_FLAG_64) {
|
|
ERR_FAIL_COND_V(len < 8, ERR_INVALID_DATA);
|
|
double val = decode_double(buf);
|
|
r_variant = val;
|
|
if (r_len) {
|
|
(*r_len) += 8;
|
|
}
|
|
} else {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
float val = decode_float(buf);
|
|
r_variant = val;
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
}
|
|
|
|
} break;
|
|
case Variant::STRING: {
|
|
String str;
|
|
Error err = _decode_string(buf, len, r_len, str);
|
|
if (err) {
|
|
return err;
|
|
}
|
|
r_variant = str;
|
|
|
|
} break;
|
|
|
|
// math types
|
|
case Variant::VECTOR2: {
|
|
ERR_FAIL_COND_V(len < 4 * 2, ERR_INVALID_DATA);
|
|
Vector2 val;
|
|
val.x = decode_float(&buf[0]);
|
|
val.y = decode_float(&buf[4]);
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 2;
|
|
}
|
|
|
|
} break; // 5
|
|
case Variant::RECT2: {
|
|
ERR_FAIL_COND_V(len < 4 * 4, ERR_INVALID_DATA);
|
|
Rect2 val;
|
|
val.position.x = decode_float(&buf[0]);
|
|
val.position.y = decode_float(&buf[4]);
|
|
val.size.x = decode_float(&buf[8]);
|
|
val.size.y = decode_float(&buf[12]);
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 4;
|
|
}
|
|
|
|
} break;
|
|
case Variant::VECTOR3: {
|
|
ERR_FAIL_COND_V(len < 4 * 3, ERR_INVALID_DATA);
|
|
Vector3 val;
|
|
val.x = decode_float(&buf[0]);
|
|
val.y = decode_float(&buf[4]);
|
|
val.z = decode_float(&buf[8]);
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 3;
|
|
}
|
|
|
|
} break;
|
|
case Variant::TRANSFORM2D: {
|
|
ERR_FAIL_COND_V(len < 4 * 6, ERR_INVALID_DATA);
|
|
Transform2D val;
|
|
for (int i = 0; i < 3; i++) {
|
|
for (int j = 0; j < 2; j++) {
|
|
val.elements[i][j] = decode_float(&buf[(i * 2 + j) * 4]);
|
|
}
|
|
}
|
|
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 6;
|
|
}
|
|
|
|
} break;
|
|
case Variant::PLANE: {
|
|
ERR_FAIL_COND_V(len < 4 * 4, ERR_INVALID_DATA);
|
|
Plane val;
|
|
val.normal.x = decode_float(&buf[0]);
|
|
val.normal.y = decode_float(&buf[4]);
|
|
val.normal.z = decode_float(&buf[8]);
|
|
val.d = decode_float(&buf[12]);
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 4;
|
|
}
|
|
|
|
} break;
|
|
case Variant::QUAT: {
|
|
ERR_FAIL_COND_V(len < 4 * 4, ERR_INVALID_DATA);
|
|
Quat val;
|
|
val.x = decode_float(&buf[0]);
|
|
val.y = decode_float(&buf[4]);
|
|
val.z = decode_float(&buf[8]);
|
|
val.w = decode_float(&buf[12]);
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 4;
|
|
}
|
|
|
|
} break;
|
|
case Variant::AABB: {
|
|
ERR_FAIL_COND_V(len < 4 * 6, ERR_INVALID_DATA);
|
|
AABB val;
|
|
val.position.x = decode_float(&buf[0]);
|
|
val.position.y = decode_float(&buf[4]);
|
|
val.position.z = decode_float(&buf[8]);
|
|
val.size.x = decode_float(&buf[12]);
|
|
val.size.y = decode_float(&buf[16]);
|
|
val.size.z = decode_float(&buf[20]);
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 6;
|
|
}
|
|
|
|
} break;
|
|
case Variant::BASIS: {
|
|
ERR_FAIL_COND_V(len < 4 * 9, ERR_INVALID_DATA);
|
|
Basis val;
|
|
for (int i = 0; i < 3; i++) {
|
|
for (int j = 0; j < 3; j++) {
|
|
val.elements[i][j] = decode_float(&buf[(i * 3 + j) * 4]);
|
|
}
|
|
}
|
|
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 9;
|
|
}
|
|
|
|
} break;
|
|
case Variant::TRANSFORM: {
|
|
ERR_FAIL_COND_V(len < 4 * 12, ERR_INVALID_DATA);
|
|
Transform val;
|
|
for (int i = 0; i < 3; i++) {
|
|
for (int j = 0; j < 3; j++) {
|
|
val.basis.elements[i][j] = decode_float(&buf[(i * 3 + j) * 4]);
|
|
}
|
|
}
|
|
val.origin[0] = decode_float(&buf[36]);
|
|
val.origin[1] = decode_float(&buf[40]);
|
|
val.origin[2] = decode_float(&buf[44]);
|
|
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 12;
|
|
}
|
|
|
|
} break;
|
|
|
|
// misc types
|
|
case Variant::COLOR: {
|
|
ERR_FAIL_COND_V(len < 4 * 4, ERR_INVALID_DATA);
|
|
Color val;
|
|
val.r = decode_float(&buf[0]);
|
|
val.g = decode_float(&buf[4]);
|
|
val.b = decode_float(&buf[8]);
|
|
val.a = decode_float(&buf[12]);
|
|
r_variant = val;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 * 4;
|
|
}
|
|
|
|
} break;
|
|
case Variant::NODE_PATH: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t strlen = decode_uint32(buf);
|
|
|
|
if (strlen & 0x80000000) {
|
|
//new format
|
|
ERR_FAIL_COND_V(len < 12, ERR_INVALID_DATA);
|
|
Vector<StringName> names;
|
|
Vector<StringName> subnames;
|
|
|
|
uint32_t namecount = strlen &= 0x7FFFFFFF;
|
|
uint32_t subnamecount = decode_uint32(buf + 4);
|
|
uint32_t flags = decode_uint32(buf + 8);
|
|
|
|
len -= 12;
|
|
buf += 12;
|
|
|
|
if (flags & 2) { // Obsolete format with property separate from subpath
|
|
subnamecount++;
|
|
}
|
|
|
|
uint32_t total = namecount + subnamecount;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 12;
|
|
}
|
|
|
|
for (uint32_t i = 0; i < total; i++) {
|
|
String str;
|
|
Error err = _decode_string(buf, len, r_len, str);
|
|
if (err) {
|
|
return err;
|
|
}
|
|
|
|
if (i < namecount) {
|
|
names.push_back(str);
|
|
} else {
|
|
subnames.push_back(str);
|
|
}
|
|
}
|
|
|
|
r_variant = NodePath(names, subnames, flags & 1);
|
|
|
|
} else {
|
|
//old format, just a string
|
|
|
|
ERR_FAIL_V(ERR_INVALID_DATA);
|
|
}
|
|
|
|
} break;
|
|
case Variant::_RID: {
|
|
r_variant = RID();
|
|
} break;
|
|
case Variant::OBJECT: {
|
|
if (type & ENCODE_FLAG_OBJECT_AS_ID) {
|
|
//this _is_ allowed
|
|
ERR_FAIL_COND_V(len < 8, ERR_INVALID_DATA);
|
|
ObjectID val = decode_uint64(buf);
|
|
if (r_len) {
|
|
(*r_len) += 8;
|
|
}
|
|
|
|
if (val == 0) {
|
|
r_variant = (Object *)nullptr;
|
|
} else {
|
|
Ref<EncodedObjectAsID> obj_as_id;
|
|
obj_as_id.instance();
|
|
obj_as_id->set_object_id(val);
|
|
|
|
r_variant = obj_as_id;
|
|
}
|
|
|
|
} else {
|
|
ERR_FAIL_COND_V(!p_allow_objects, ERR_UNAUTHORIZED);
|
|
|
|
String str;
|
|
Error err = _decode_string(buf, len, r_len, str);
|
|
if (err) {
|
|
return err;
|
|
}
|
|
|
|
if (str == String()) {
|
|
r_variant = (Object *)nullptr;
|
|
} else {
|
|
Object *obj = ClassDB::instance(str);
|
|
|
|
ERR_FAIL_COND_V(!obj, ERR_UNAVAILABLE);
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
|
|
int32_t count = decode_uint32(buf);
|
|
buf += 4;
|
|
len -= 4;
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
|
|
for (int i = 0; i < count; i++) {
|
|
str = String();
|
|
err = _decode_string(buf, len, r_len, str);
|
|
if (err) {
|
|
return err;
|
|
}
|
|
|
|
Variant value;
|
|
int used;
|
|
err = decode_variant(value, buf, len, &used, p_allow_objects);
|
|
if (err) {
|
|
return err;
|
|
}
|
|
|
|
buf += used;
|
|
len -= used;
|
|
if (r_len) {
|
|
(*r_len) += used;
|
|
}
|
|
|
|
obj->set(str, value);
|
|
}
|
|
|
|
if (Object::cast_to<Reference>(obj)) {
|
|
REF ref = REF(Object::cast_to<Reference>(obj));
|
|
r_variant = ref;
|
|
} else {
|
|
r_variant = obj;
|
|
}
|
|
}
|
|
}
|
|
|
|
} break;
|
|
case Variant::DICTIONARY: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t count = decode_uint32(buf);
|
|
// bool shared = count&0x80000000;
|
|
count &= 0x7FFFFFFF;
|
|
|
|
buf += 4;
|
|
len -= 4;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
|
|
Dictionary d;
|
|
|
|
for (int i = 0; i < count; i++) {
|
|
Variant key, value;
|
|
|
|
int used;
|
|
Error err = decode_variant(key, buf, len, &used, p_allow_objects);
|
|
ERR_FAIL_COND_V_MSG(err != OK, err, "Error when trying to decode Variant.");
|
|
|
|
buf += used;
|
|
len -= used;
|
|
if (r_len) {
|
|
(*r_len) += used;
|
|
}
|
|
|
|
err = decode_variant(value, buf, len, &used, p_allow_objects);
|
|
ERR_FAIL_COND_V_MSG(err != OK, err, "Error when trying to decode Variant.");
|
|
|
|
buf += used;
|
|
len -= used;
|
|
if (r_len) {
|
|
(*r_len) += used;
|
|
}
|
|
|
|
d[key] = value;
|
|
}
|
|
|
|
r_variant = d;
|
|
|
|
} break;
|
|
case Variant::ARRAY: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t count = decode_uint32(buf);
|
|
// bool shared = count&0x80000000;
|
|
count &= 0x7FFFFFFF;
|
|
|
|
buf += 4;
|
|
len -= 4;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
|
|
Array varr;
|
|
|
|
for (int i = 0; i < count; i++) {
|
|
int used = 0;
|
|
Variant v;
|
|
Error err = decode_variant(v, buf, len, &used, p_allow_objects);
|
|
ERR_FAIL_COND_V_MSG(err != OK, err, "Error when trying to decode Variant.");
|
|
buf += used;
|
|
len -= used;
|
|
varr.push_back(v);
|
|
if (r_len) {
|
|
(*r_len) += used;
|
|
}
|
|
}
|
|
|
|
r_variant = varr;
|
|
|
|
} break;
|
|
|
|
// arrays
|
|
case Variant::POOL_BYTE_ARRAY: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t count = decode_uint32(buf);
|
|
buf += 4;
|
|
len -= 4;
|
|
ERR_FAIL_COND_V(count < 0 || count > len, ERR_INVALID_DATA);
|
|
|
|
PoolVector<uint8_t> data;
|
|
|
|
if (count) {
|
|
data.resize(count);
|
|
PoolVector<uint8_t>::Write w = data.write();
|
|
for (int32_t i = 0; i < count; i++) {
|
|
w[i] = buf[i];
|
|
}
|
|
}
|
|
|
|
r_variant = data;
|
|
|
|
if (r_len) {
|
|
if (count % 4) {
|
|
(*r_len) += 4 - count % 4;
|
|
}
|
|
(*r_len) += 4 + count;
|
|
}
|
|
|
|
} break;
|
|
case Variant::POOL_INT_ARRAY: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t count = decode_uint32(buf);
|
|
buf += 4;
|
|
len -= 4;
|
|
ERR_FAIL_MUL_OF(count, 4, ERR_INVALID_DATA);
|
|
ERR_FAIL_COND_V(count < 0 || count * 4 > len, ERR_INVALID_DATA);
|
|
|
|
PoolVector<int> data;
|
|
|
|
if (count) {
|
|
//const int*rbuf=(const int*)buf;
|
|
data.resize(count);
|
|
PoolVector<int>::Write w = data.write();
|
|
for (int32_t i = 0; i < count; i++) {
|
|
w[i] = decode_uint32(&buf[i * 4]);
|
|
}
|
|
}
|
|
r_variant = Variant(data);
|
|
if (r_len) {
|
|
(*r_len) += 4 + count * sizeof(int);
|
|
}
|
|
|
|
} break;
|
|
case Variant::POOL_REAL_ARRAY: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t count = decode_uint32(buf);
|
|
buf += 4;
|
|
len -= 4;
|
|
ERR_FAIL_MUL_OF(count, 4, ERR_INVALID_DATA);
|
|
ERR_FAIL_COND_V(count < 0 || count * 4 > len, ERR_INVALID_DATA);
|
|
|
|
PoolVector<float> data;
|
|
|
|
if (count) {
|
|
//const float*rbuf=(const float*)buf;
|
|
data.resize(count);
|
|
PoolVector<float>::Write w = data.write();
|
|
for (int32_t i = 0; i < count; i++) {
|
|
w[i] = decode_float(&buf[i * 4]);
|
|
}
|
|
}
|
|
r_variant = data;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4 + count * sizeof(float);
|
|
}
|
|
|
|
} break;
|
|
case Variant::POOL_STRING_ARRAY: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t count = decode_uint32(buf);
|
|
|
|
PoolVector<String> strings;
|
|
buf += 4;
|
|
len -= 4;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
//printf("string count: %i\n",count);
|
|
|
|
for (int32_t i = 0; i < count; i++) {
|
|
String str;
|
|
Error err = _decode_string(buf, len, r_len, str);
|
|
if (err) {
|
|
return err;
|
|
}
|
|
|
|
strings.push_back(str);
|
|
}
|
|
|
|
r_variant = strings;
|
|
|
|
} break;
|
|
case Variant::POOL_VECTOR2_ARRAY: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t count = decode_uint32(buf);
|
|
buf += 4;
|
|
len -= 4;
|
|
|
|
ERR_FAIL_MUL_OF(count, 4 * 2, ERR_INVALID_DATA);
|
|
ERR_FAIL_COND_V(count < 0 || count * 4 * 2 > len, ERR_INVALID_DATA);
|
|
PoolVector<Vector2> varray;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
|
|
if (count) {
|
|
varray.resize(count);
|
|
PoolVector<Vector2>::Write w = varray.write();
|
|
|
|
for (int32_t i = 0; i < count; i++) {
|
|
w[i].x = decode_float(buf + i * 4 * 2 + 4 * 0);
|
|
w[i].y = decode_float(buf + i * 4 * 2 + 4 * 1);
|
|
}
|
|
|
|
int adv = 4 * 2 * count;
|
|
|
|
if (r_len) {
|
|
(*r_len) += adv;
|
|
}
|
|
}
|
|
|
|
r_variant = varray;
|
|
|
|
} break;
|
|
case Variant::POOL_VECTOR3_ARRAY: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t count = decode_uint32(buf);
|
|
buf += 4;
|
|
len -= 4;
|
|
|
|
ERR_FAIL_MUL_OF(count, 4 * 3, ERR_INVALID_DATA);
|
|
ERR_FAIL_COND_V(count < 0 || count * 4 * 3 > len, ERR_INVALID_DATA);
|
|
|
|
PoolVector<Vector3> varray;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
|
|
if (count) {
|
|
varray.resize(count);
|
|
PoolVector<Vector3>::Write w = varray.write();
|
|
|
|
for (int32_t i = 0; i < count; i++) {
|
|
w[i].x = decode_float(buf + i * 4 * 3 + 4 * 0);
|
|
w[i].y = decode_float(buf + i * 4 * 3 + 4 * 1);
|
|
w[i].z = decode_float(buf + i * 4 * 3 + 4 * 2);
|
|
}
|
|
|
|
int adv = 4 * 3 * count;
|
|
|
|
if (r_len) {
|
|
(*r_len) += adv;
|
|
}
|
|
}
|
|
|
|
r_variant = varray;
|
|
|
|
} break;
|
|
case Variant::POOL_COLOR_ARRAY: {
|
|
ERR_FAIL_COND_V(len < 4, ERR_INVALID_DATA);
|
|
int32_t count = decode_uint32(buf);
|
|
buf += 4;
|
|
len -= 4;
|
|
|
|
ERR_FAIL_MUL_OF(count, 4 * 4, ERR_INVALID_DATA);
|
|
ERR_FAIL_COND_V(count < 0 || count * 4 * 4 > len, ERR_INVALID_DATA);
|
|
|
|
PoolVector<Color> carray;
|
|
|
|
if (r_len) {
|
|
(*r_len) += 4;
|
|
}
|
|
|
|
if (count) {
|
|
carray.resize(count);
|
|
PoolVector<Color>::Write w = carray.write();
|
|
|
|
for (int32_t i = 0; i < count; i++) {
|
|
w[i].r = decode_float(buf + i * 4 * 4 + 4 * 0);
|
|
w[i].g = decode_float(buf + i * 4 * 4 + 4 * 1);
|
|
w[i].b = decode_float(buf + i * 4 * 4 + 4 * 2);
|
|
w[i].a = decode_float(buf + i * 4 * 4 + 4 * 3);
|
|
}
|
|
|
|
int adv = 4 * 4 * count;
|
|
|
|
if (r_len) {
|
|
(*r_len) += adv;
|
|
}
|
|
}
|
|
|
|
r_variant = carray;
|
|
|
|
} break;
|
|
default: {
|
|
ERR_FAIL_V(ERR_BUG);
|
|
}
|
|
}
|
|
|
|
return OK;
|
|
}
|
|
|
|
static void _encode_string(const String &p_string, uint8_t *&buf, int &r_len) {
|
|
CharString utf8 = p_string.utf8();
|
|
|
|
if (buf) {
|
|
encode_uint32(utf8.length(), buf);
|
|
buf += 4;
|
|
memcpy(buf, utf8.get_data(), utf8.length());
|
|
buf += utf8.length();
|
|
}
|
|
|
|
r_len += 4 + utf8.length();
|
|
while (r_len % 4) {
|
|
r_len++; //pad
|
|
if (buf) {
|
|
*(buf++) = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
Error encode_variant(const Variant &p_variant, uint8_t *r_buffer, int &r_len, bool p_full_objects, int p_depth) {
|
|
ERR_FAIL_COND_V_MSG(p_depth > Variant::MAX_RECURSION_DEPTH, ERR_OUT_OF_MEMORY, "Potential inifite recursion detected. Bailing.");
|
|
uint8_t *buf = r_buffer;
|
|
|
|
r_len = 0;
|
|
|
|
uint32_t flags = 0;
|
|
|
|
switch (p_variant.get_type()) {
|
|
case Variant::INT: {
|
|
int64_t val = p_variant;
|
|
if (val > (int64_t)INT_MAX || val < (int64_t)INT_MIN) {
|
|
flags |= ENCODE_FLAG_64;
|
|
}
|
|
} break;
|
|
case Variant::REAL: {
|
|
double d = p_variant;
|
|
float f = d;
|
|
if (double(f) != d) {
|
|
flags |= ENCODE_FLAG_64; //always encode real as double
|
|
}
|
|
} break;
|
|
case Variant::OBJECT: {
|
|
#ifdef DEBUG_ENABLED
|
|
// Test for potential wrong values sent by the debugger when it breaks.
|
|
Object *obj = p_variant;
|
|
if (!obj || !ObjectDB::instance_validate(obj)) {
|
|
// Object is invalid, send a NULL instead.
|
|
if (buf) {
|
|
encode_uint32(Variant::NIL, buf);
|
|
}
|
|
r_len += 4;
|
|
return OK;
|
|
}
|
|
#endif // DEBUG_ENABLED
|
|
if (!p_full_objects) {
|
|
flags |= ENCODE_FLAG_OBJECT_AS_ID;
|
|
}
|
|
} break;
|
|
default: {
|
|
} // nothing to do at this stage
|
|
}
|
|
|
|
if (buf) {
|
|
encode_uint32(p_variant.get_type() | flags, buf);
|
|
buf += 4;
|
|
}
|
|
r_len += 4;
|
|
|
|
switch (p_variant.get_type()) {
|
|
case Variant::NIL: {
|
|
//nothing to do
|
|
} break;
|
|
case Variant::BOOL: {
|
|
if (buf) {
|
|
encode_uint32(p_variant.operator bool(), buf);
|
|
}
|
|
|
|
r_len += 4;
|
|
|
|
} break;
|
|
case Variant::INT: {
|
|
if (flags & ENCODE_FLAG_64) {
|
|
//64 bits
|
|
if (buf) {
|
|
encode_uint64(p_variant.operator int64_t(), buf);
|
|
}
|
|
|
|
r_len += 8;
|
|
} else {
|
|
if (buf) {
|
|
encode_uint32(p_variant.operator int32_t(), buf);
|
|
}
|
|
|
|
r_len += 4;
|
|
}
|
|
} break;
|
|
case Variant::REAL: {
|
|
if (flags & ENCODE_FLAG_64) {
|
|
if (buf) {
|
|
encode_double(p_variant.operator double(), buf);
|
|
}
|
|
|
|
r_len += 8;
|
|
|
|
} else {
|
|
if (buf) {
|
|
encode_float(p_variant.operator float(), buf);
|
|
}
|
|
|
|
r_len += 4;
|
|
}
|
|
|
|
} break;
|
|
case Variant::NODE_PATH: {
|
|
NodePath np = p_variant;
|
|
if (buf) {
|
|
encode_uint32(uint32_t(np.get_name_count()) | 0x80000000, buf); //for compatibility with the old format
|
|
encode_uint32(np.get_subname_count(), buf + 4);
|
|
uint32_t np_flags = 0;
|
|
if (np.is_absolute()) {
|
|
np_flags |= 1;
|
|
}
|
|
|
|
encode_uint32(np_flags, buf + 8);
|
|
|
|
buf += 12;
|
|
}
|
|
|
|
r_len += 12;
|
|
|
|
int total = np.get_name_count() + np.get_subname_count();
|
|
|
|
for (int i = 0; i < total; i++) {
|
|
String str;
|
|
|
|
if (i < np.get_name_count()) {
|
|
str = np.get_name(i);
|
|
} else {
|
|
str = np.get_subname(i - np.get_name_count());
|
|
}
|
|
|
|
CharString utf8 = str.utf8();
|
|
|
|
int pad = 0;
|
|
|
|
if (utf8.length() % 4) {
|
|
pad = 4 - utf8.length() % 4;
|
|
}
|
|
|
|
if (buf) {
|
|
encode_uint32(utf8.length(), buf);
|
|
buf += 4;
|
|
memcpy(buf, utf8.get_data(), utf8.length());
|
|
buf += pad + utf8.length();
|
|
}
|
|
|
|
r_len += 4 + utf8.length() + pad;
|
|
}
|
|
|
|
} break;
|
|
case Variant::STRING: {
|
|
_encode_string(p_variant, buf, r_len);
|
|
|
|
} break;
|
|
|
|
// math types
|
|
case Variant::VECTOR2: {
|
|
if (buf) {
|
|
Vector2 v2 = p_variant;
|
|
encode_float(v2.x, &buf[0]);
|
|
encode_float(v2.y, &buf[4]);
|
|
}
|
|
|
|
r_len += 2 * 4;
|
|
|
|
} break; // 5
|
|
case Variant::RECT2: {
|
|
if (buf) {
|
|
Rect2 r2 = p_variant;
|
|
encode_float(r2.position.x, &buf[0]);
|
|
encode_float(r2.position.y, &buf[4]);
|
|
encode_float(r2.size.x, &buf[8]);
|
|
encode_float(r2.size.y, &buf[12]);
|
|
}
|
|
r_len += 4 * 4;
|
|
|
|
} break;
|
|
case Variant::VECTOR3: {
|
|
if (buf) {
|
|
Vector3 v3 = p_variant;
|
|
encode_float(v3.x, &buf[0]);
|
|
encode_float(v3.y, &buf[4]);
|
|
encode_float(v3.z, &buf[8]);
|
|
}
|
|
|
|
r_len += 3 * 4;
|
|
|
|
} break;
|
|
case Variant::TRANSFORM2D: {
|
|
if (buf) {
|
|
Transform2D val = p_variant;
|
|
for (int i = 0; i < 3; i++) {
|
|
for (int j = 0; j < 2; j++) {
|
|
memcpy(&buf[(i * 2 + j) * 4], &val.elements[i][j], sizeof(float));
|
|
}
|
|
}
|
|
}
|
|
|
|
r_len += 6 * 4;
|
|
|
|
} break;
|
|
case Variant::PLANE: {
|
|
if (buf) {
|
|
Plane p = p_variant;
|
|
encode_float(p.normal.x, &buf[0]);
|
|
encode_float(p.normal.y, &buf[4]);
|
|
encode_float(p.normal.z, &buf[8]);
|
|
encode_float(p.d, &buf[12]);
|
|
}
|
|
|
|
r_len += 4 * 4;
|
|
|
|
} break;
|
|
case Variant::QUAT: {
|
|
if (buf) {
|
|
Quat q = p_variant;
|
|
encode_float(q.x, &buf[0]);
|
|
encode_float(q.y, &buf[4]);
|
|
encode_float(q.z, &buf[8]);
|
|
encode_float(q.w, &buf[12]);
|
|
}
|
|
|
|
r_len += 4 * 4;
|
|
|
|
} break;
|
|
case Variant::AABB: {
|
|
if (buf) {
|
|
AABB aabb = p_variant;
|
|
encode_float(aabb.position.x, &buf[0]);
|
|
encode_float(aabb.position.y, &buf[4]);
|
|
encode_float(aabb.position.z, &buf[8]);
|
|
encode_float(aabb.size.x, &buf[12]);
|
|
encode_float(aabb.size.y, &buf[16]);
|
|
encode_float(aabb.size.z, &buf[20]);
|
|
}
|
|
|
|
r_len += 6 * 4;
|
|
|
|
} break;
|
|
case Variant::BASIS: {
|
|
if (buf) {
|
|
Basis val = p_variant;
|
|
for (int i = 0; i < 3; i++) {
|
|
for (int j = 0; j < 3; j++) {
|
|
memcpy(&buf[(i * 3 + j) * 4], &val.elements[i][j], sizeof(float));
|
|
}
|
|
}
|
|
}
|
|
|
|
r_len += 9 * 4;
|
|
|
|
} break;
|
|
case Variant::TRANSFORM: {
|
|
if (buf) {
|
|
Transform val = p_variant;
|
|
for (int i = 0; i < 3; i++) {
|
|
for (int j = 0; j < 3; j++) {
|
|
memcpy(&buf[(i * 3 + j) * 4], &val.basis.elements[i][j], sizeof(float));
|
|
}
|
|
}
|
|
|
|
encode_float(val.origin.x, &buf[36]);
|
|
encode_float(val.origin.y, &buf[40]);
|
|
encode_float(val.origin.z, &buf[44]);
|
|
}
|
|
|
|
r_len += 12 * 4;
|
|
|
|
} break;
|
|
|
|
// misc types
|
|
case Variant::COLOR: {
|
|
if (buf) {
|
|
Color c = p_variant;
|
|
encode_float(c.r, &buf[0]);
|
|
encode_float(c.g, &buf[4]);
|
|
encode_float(c.b, &buf[8]);
|
|
encode_float(c.a, &buf[12]);
|
|
}
|
|
|
|
r_len += 4 * 4;
|
|
|
|
} break;
|
|
case Variant::_RID: {
|
|
} break;
|
|
case Variant::OBJECT: {
|
|
if (p_full_objects) {
|
|
Object *obj = p_variant;
|
|
if (!obj) {
|
|
if (buf) {
|
|
encode_uint32(0, buf);
|
|
}
|
|
r_len += 4;
|
|
|
|
} else {
|
|
_encode_string(obj->get_class(), buf, r_len);
|
|
|
|
List<PropertyInfo> props;
|
|
obj->get_property_list(&props);
|
|
|
|
int pc = 0;
|
|
for (List<PropertyInfo>::Element *E = props.front(); E; E = E->next()) {
|
|
if (!(E->get().usage & PROPERTY_USAGE_STORAGE)) {
|
|
continue;
|
|
}
|
|
pc++;
|
|
}
|
|
|
|
if (buf) {
|
|
encode_uint32(pc, buf);
|
|
buf += 4;
|
|
}
|
|
|
|
r_len += 4;
|
|
|
|
for (List<PropertyInfo>::Element *E = props.front(); E; E = E->next()) {
|
|
if (!(E->get().usage & PROPERTY_USAGE_STORAGE)) {
|
|
continue;
|
|
}
|
|
|
|
_encode_string(E->get().name, buf, r_len);
|
|
|
|
int len;
|
|
Error err = encode_variant(obj->get(E->get().name), buf, len, p_full_objects, p_depth + 1);
|
|
ERR_FAIL_COND_V(err, err);
|
|
ERR_FAIL_COND_V(len % 4, ERR_BUG);
|
|
r_len += len;
|
|
if (buf) {
|
|
buf += len;
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
if (buf) {
|
|
Object *obj = p_variant;
|
|
ObjectID id = 0;
|
|
if (obj && ObjectDB::instance_validate(obj)) {
|
|
id = obj->get_instance_id();
|
|
}
|
|
|
|
encode_uint64(id, buf);
|
|
}
|
|
|
|
r_len += 8;
|
|
}
|
|
|
|
} break;
|
|
case Variant::DICTIONARY: {
|
|
Dictionary d = p_variant;
|
|
|
|
if (buf) {
|
|
encode_uint32(uint32_t(d.size()), buf);
|
|
buf += 4;
|
|
}
|
|
r_len += 4;
|
|
|
|
List<Variant> keys;
|
|
d.get_key_list(&keys);
|
|
|
|
for (List<Variant>::Element *E = keys.front(); E; E = E->next()) {
|
|
/*
|
|
CharString utf8 = E->->utf8();
|
|
|
|
if (buf) {
|
|
encode_uint32(utf8.length()+1,buf);
|
|
buf+=4;
|
|
memcpy(buf,utf8.get_data(),utf8.length()+1);
|
|
}
|
|
|
|
r_len+=4+utf8.length()+1;
|
|
while (r_len%4)
|
|
r_len++; //pad
|
|
*/
|
|
Variant *v = d.getptr(E->get());
|
|
int len;
|
|
Error err = encode_variant(v ? E->get() : Variant("[Deleted Object]"), buf, len, p_full_objects, p_depth + 1);
|
|
ERR_FAIL_COND_V(err, err);
|
|
ERR_FAIL_COND_V(len % 4, ERR_BUG);
|
|
r_len += len;
|
|
if (buf) {
|
|
buf += len;
|
|
}
|
|
err = encode_variant(v ? *v : Variant(), buf, len, p_full_objects, p_depth + 1);
|
|
ERR_FAIL_COND_V(err, err);
|
|
ERR_FAIL_COND_V(len % 4, ERR_BUG);
|
|
r_len += len;
|
|
if (buf) {
|
|
buf += len;
|
|
}
|
|
}
|
|
|
|
} break;
|
|
case Variant::ARRAY: {
|
|
Array v = p_variant;
|
|
|
|
if (buf) {
|
|
encode_uint32(uint32_t(v.size()), buf);
|
|
buf += 4;
|
|
}
|
|
|
|
r_len += 4;
|
|
|
|
for (int i = 0; i < v.size(); i++) {
|
|
int len;
|
|
Error err = encode_variant(v.get(i), buf, len, p_full_objects, p_depth + 1);
|
|
ERR_FAIL_COND_V(err, err);
|
|
ERR_FAIL_COND_V(len % 4, ERR_BUG);
|
|
r_len += len;
|
|
if (buf) {
|
|
buf += len;
|
|
}
|
|
}
|
|
|
|
} break;
|
|
// arrays
|
|
case Variant::POOL_BYTE_ARRAY: {
|
|
PoolVector<uint8_t> data = p_variant;
|
|
int datalen = data.size();
|
|
int datasize = sizeof(uint8_t);
|
|
|
|
if (buf) {
|
|
encode_uint32(datalen, buf);
|
|
buf += 4;
|
|
PoolVector<uint8_t>::Read r = data.read();
|
|
memcpy(buf, &r[0], datalen * datasize);
|
|
buf += datalen * datasize;
|
|
}
|
|
|
|
r_len += 4 + datalen * datasize;
|
|
while (r_len % 4) {
|
|
r_len++;
|
|
if (buf) {
|
|
*(buf++) = 0;
|
|
}
|
|
}
|
|
|
|
} break;
|
|
case Variant::POOL_INT_ARRAY: {
|
|
PoolVector<int> data = p_variant;
|
|
int datalen = data.size();
|
|
int datasize = sizeof(int32_t);
|
|
|
|
if (buf) {
|
|
encode_uint32(datalen, buf);
|
|
buf += 4;
|
|
PoolVector<int>::Read r = data.read();
|
|
for (int i = 0; i < datalen; i++) {
|
|
encode_uint32(r[i], &buf[i * datasize]);
|
|
}
|
|
}
|
|
|
|
r_len += 4 + datalen * datasize;
|
|
|
|
} break;
|
|
case Variant::POOL_REAL_ARRAY: {
|
|
PoolVector<real_t> data = p_variant;
|
|
int datalen = data.size();
|
|
int datasize = sizeof(real_t);
|
|
|
|
if (buf) {
|
|
encode_uint32(datalen, buf);
|
|
buf += 4;
|
|
PoolVector<real_t>::Read r = data.read();
|
|
for (int i = 0; i < datalen; i++) {
|
|
encode_float(r[i], &buf[i * datasize]);
|
|
}
|
|
}
|
|
|
|
r_len += 4 + datalen * datasize;
|
|
|
|
} break;
|
|
case Variant::POOL_STRING_ARRAY: {
|
|
PoolVector<String> data = p_variant;
|
|
int len = data.size();
|
|
|
|
if (buf) {
|
|
encode_uint32(len, buf);
|
|
buf += 4;
|
|
}
|
|
|
|
r_len += 4;
|
|
|
|
for (int i = 0; i < len; i++) {
|
|
CharString utf8 = data.get(i).utf8();
|
|
|
|
if (buf) {
|
|
encode_uint32(utf8.length() + 1, buf);
|
|
buf += 4;
|
|
memcpy(buf, utf8.get_data(), utf8.length() + 1);
|
|
buf += utf8.length() + 1;
|
|
}
|
|
|
|
r_len += 4 + utf8.length() + 1;
|
|
while (r_len % 4) {
|
|
r_len++; //pad
|
|
if (buf) {
|
|
*(buf++) = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
} break;
|
|
case Variant::POOL_VECTOR2_ARRAY: {
|
|
PoolVector<Vector2> data = p_variant;
|
|
int len = data.size();
|
|
|
|
if (buf) {
|
|
encode_uint32(len, buf);
|
|
buf += 4;
|
|
}
|
|
|
|
r_len += 4;
|
|
|
|
if (buf) {
|
|
for (int i = 0; i < len; i++) {
|
|
Vector2 v = data.get(i);
|
|
|
|
encode_float(v.x, &buf[0]);
|
|
encode_float(v.y, &buf[4]);
|
|
buf += 4 * 2;
|
|
}
|
|
}
|
|
|
|
r_len += 4 * 2 * len;
|
|
|
|
} break;
|
|
case Variant::POOL_VECTOR3_ARRAY: {
|
|
PoolVector<Vector3> data = p_variant;
|
|
int len = data.size();
|
|
|
|
if (buf) {
|
|
encode_uint32(len, buf);
|
|
buf += 4;
|
|
}
|
|
|
|
r_len += 4;
|
|
|
|
if (buf) {
|
|
for (int i = 0; i < len; i++) {
|
|
Vector3 v = data.get(i);
|
|
|
|
encode_float(v.x, &buf[0]);
|
|
encode_float(v.y, &buf[4]);
|
|
encode_float(v.z, &buf[8]);
|
|
buf += 4 * 3;
|
|
}
|
|
}
|
|
|
|
r_len += 4 * 3 * len;
|
|
|
|
} break;
|
|
case Variant::POOL_COLOR_ARRAY: {
|
|
PoolVector<Color> data = p_variant;
|
|
int len = data.size();
|
|
|
|
if (buf) {
|
|
encode_uint32(len, buf);
|
|
buf += 4;
|
|
}
|
|
|
|
r_len += 4;
|
|
|
|
if (buf) {
|
|
for (int i = 0; i < len; i++) {
|
|
Color c = data.get(i);
|
|
|
|
encode_float(c.r, &buf[0]);
|
|
encode_float(c.g, &buf[4]);
|
|
encode_float(c.b, &buf[8]);
|
|
encode_float(c.a, &buf[12]);
|
|
buf += 4 * 4;
|
|
}
|
|
}
|
|
|
|
r_len += 4 * 4 * len;
|
|
|
|
} break;
|
|
default: {
|
|
ERR_FAIL_V(ERR_BUG);
|
|
}
|
|
}
|
|
|
|
return OK;
|
|
}
|