virtualx-engine/core/io/packet_peer.cpp

296 lines
8.7 KiB
C++
Raw Normal View History

2014-02-10 02:10:30 +01:00
/*************************************************************************/
/* packet_peer.cpp */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* https://godotengine.org */
2014-02-10 02:10:30 +01:00
/*************************************************************************/
/* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur. */
/* Copyright (c) 2014-2017 Godot Engine contributors (cf. AUTHORS.md) */
2014-02-10 02:10:30 +01:00
/* */
/* 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 "packet_peer.h"
#include "io/marshalls.h"
#include "project_settings.h"
2014-02-10 02:10:30 +01:00
/* helpers / binders */
PacketPeer::PacketPeer() {
allow_object_decoding = false;
last_get_error = OK;
2014-02-10 02:10:30 +01:00
}
void PacketPeer::set_allow_object_decoding(bool p_enable) {
allow_object_decoding = p_enable;
}
bool PacketPeer::is_object_decoding_allowed() const {
return allow_object_decoding;
}
Error PacketPeer::get_packet_buffer(PoolVector<uint8_t> &r_buffer) {
2014-02-10 02:10:30 +01:00
const uint8_t *buffer;
int buffer_size;
Error err = get_packet(&buffer, buffer_size);
2014-02-10 02:10:30 +01:00
if (err)
return err;
r_buffer.resize(buffer_size);
if (buffer_size == 0)
2014-02-10 02:10:30 +01:00
return OK;
PoolVector<uint8_t>::Write w = r_buffer.write();
for (int i = 0; i < buffer_size; i++)
w[i] = buffer[i];
2014-02-10 02:10:30 +01:00
return OK;
}
Error PacketPeer::put_packet_buffer(const PoolVector<uint8_t> &p_buffer) {
2014-02-10 02:10:30 +01:00
int len = p_buffer.size();
if (len == 0)
2014-02-10 02:10:30 +01:00
return OK;
PoolVector<uint8_t>::Read r = p_buffer.read();
return put_packet(&r[0], len);
2014-02-10 02:10:30 +01:00
}
Error PacketPeer::get_var(Variant &r_variant) {
2014-02-10 02:10:30 +01:00
const uint8_t *buffer;
int buffer_size;
Error err = get_packet(&buffer, buffer_size);
2014-02-10 02:10:30 +01:00
if (err)
return err;
return decode_variant(r_variant, buffer, buffer_size, NULL, allow_object_decoding);
2014-02-10 02:10:30 +01:00
}
Error PacketPeer::put_var(const Variant &p_packet) {
2014-02-10 02:10:30 +01:00
int len;
Error err = encode_variant(p_packet, NULL, len, !allow_object_decoding); // compute len first
2014-02-10 02:10:30 +01:00
if (err)
return err;
if (len == 0)
2014-02-10 02:10:30 +01:00
return OK;
uint8_t *buf = (uint8_t *)alloca(len);
ERR_FAIL_COND_V(!buf, ERR_OUT_OF_MEMORY);
err = encode_variant(p_packet, buf, len, !allow_object_decoding);
2014-02-10 02:10:30 +01:00
ERR_FAIL_COND_V(err, err);
return put_packet(buf, len);
}
Variant PacketPeer::_bnd_get_var() {
2014-02-10 02:10:30 +01:00
Variant var;
get_var(var);
return var;
};
Error PacketPeer::_put_packet(const PoolVector<uint8_t> &p_buffer) {
return put_packet_buffer(p_buffer);
}
PoolVector<uint8_t> PacketPeer::_get_packet() {
PoolVector<uint8_t> raw;
last_get_error = get_packet_buffer(raw);
return raw;
}
Error PacketPeer::_get_packet_error() const {
return last_get_error;
}
2014-02-10 02:10:30 +01:00
void PacketPeer::_bind_methods() {
ClassDB::bind_method(D_METHOD("get_var"), &PacketPeer::_bnd_get_var);
ClassDB::bind_method(D_METHOD("put_var", "var"), &PacketPeer::put_var);
ClassDB::bind_method(D_METHOD("get_packet"), &PacketPeer::_get_packet);
ClassDB::bind_method(D_METHOD("put_packet", "buffer"), &PacketPeer::_put_packet);
ClassDB::bind_method(D_METHOD("get_packet_error"), &PacketPeer::_get_packet_error);
ClassDB::bind_method(D_METHOD("get_available_packet_count"), &PacketPeer::get_available_packet_count);
ClassDB::bind_method(D_METHOD("set_allow_object_decoding", "enable"), &PacketPeer::set_allow_object_decoding);
ClassDB::bind_method(D_METHOD("is_object_decoding_allowed"), &PacketPeer::is_object_decoding_allowed);
2014-02-10 02:10:30 +01:00
};
/***************/
void PacketPeerStream::_set_stream_peer(REF p_peer) {
ERR_FAIL_COND(p_peer.is_null());
set_stream_peer(p_peer);
}
void PacketPeerStream::_bind_methods() {
ClassDB::bind_method(D_METHOD("set_stream_peer", "peer"), &PacketPeerStream::_set_stream_peer);
ClassDB::bind_method(D_METHOD("set_input_buffer_max_size", "max_size_bytes"), &PacketPeerStream::set_input_buffer_max_size);
ClassDB::bind_method(D_METHOD("set_output_buffer_max_size", "max_size_bytes"), &PacketPeerStream::set_output_buffer_max_size);
ClassDB::bind_method(D_METHOD("get_input_buffer_max_size"), &PacketPeerStream::get_input_buffer_max_size);
ClassDB::bind_method(D_METHOD("get_output_buffer_max_size"), &PacketPeerStream::get_output_buffer_max_size);
2014-02-10 02:10:30 +01:00
}
Error PacketPeerStream::_poll_buffer() const {
ERR_FAIL_COND_V(peer.is_null(), ERR_UNCONFIGURED);
2014-02-10 02:10:30 +01:00
int read = 0;
Error err = peer->get_partial_data(&input_buffer[0], ring_buffer.space_left(), read);
2014-02-10 02:10:30 +01:00
if (err)
return err;
if (read == 0)
2014-02-10 02:10:30 +01:00
return OK;
int w = ring_buffer.write(&input_buffer[0], read);
ERR_FAIL_COND_V(w != read, ERR_BUG);
2014-02-10 02:10:30 +01:00
return OK;
}
int PacketPeerStream::get_available_packet_count() const {
_poll_buffer();
uint32_t remaining = ring_buffer.data_left();
int ofs = 0;
int count = 0;
2014-02-10 02:10:30 +01:00
while (remaining >= 4) {
2014-02-10 02:10:30 +01:00
uint8_t lbuf[4];
ring_buffer.copy(lbuf, ofs, 4);
2014-02-10 02:10:30 +01:00
uint32_t len = decode_uint32(lbuf);
remaining -= 4;
ofs += 4;
if (len > remaining)
2014-02-10 02:10:30 +01:00
break;
remaining -= len;
ofs += len;
2014-02-10 02:10:30 +01:00
count++;
}
return count;
}
Error PacketPeerStream::get_packet(const uint8_t **r_buffer, int &r_buffer_size) {
2014-02-10 02:10:30 +01:00
ERR_FAIL_COND_V(peer.is_null(), ERR_UNCONFIGURED);
2014-02-10 02:10:30 +01:00
_poll_buffer();
int remaining = ring_buffer.data_left();
ERR_FAIL_COND_V(remaining < 4, ERR_UNAVAILABLE);
2014-02-10 02:10:30 +01:00
uint8_t lbuf[4];
ring_buffer.copy(lbuf, 0, 4);
remaining -= 4;
2016-03-09 00:00:52 +01:00
uint32_t len = decode_uint32(lbuf);
ERR_FAIL_COND_V(remaining < (int)len, ERR_UNAVAILABLE);
2014-02-10 02:10:30 +01:00
ring_buffer.read(lbuf, 4); //get rid of first 4 bytes
ring_buffer.read(&input_buffer[0], len); // read packet
2014-02-10 02:10:30 +01:00
*r_buffer = &input_buffer[0];
r_buffer_size = len;
2014-02-10 02:10:30 +01:00
return OK;
}
Error PacketPeerStream::put_packet(const uint8_t *p_buffer, int p_buffer_size) {
2014-02-10 02:10:30 +01:00
ERR_FAIL_COND_V(peer.is_null(), ERR_UNCONFIGURED);
2014-02-10 02:10:30 +01:00
Error err = _poll_buffer(); //won't hurt to poll here too
if (err)
return err;
if (p_buffer_size == 0)
2014-02-10 02:10:30 +01:00
return OK;
ERR_FAIL_COND_V(p_buffer_size < 0, ERR_INVALID_PARAMETER);
ERR_FAIL_COND_V(p_buffer_size + 4 > output_buffer.size(), ERR_INVALID_PARAMETER);
2014-02-10 02:10:30 +01:00
encode_uint32(p_buffer_size, &output_buffer[0]);
uint8_t *dst = &output_buffer[4];
for (int i = 0; i < p_buffer_size; i++)
dst[i] = p_buffer[i];
2014-02-10 02:10:30 +01:00
return peer->put_data(&output_buffer[0], p_buffer_size + 4);
2014-02-10 02:10:30 +01:00
}
int PacketPeerStream::get_max_packet_size() const {
return output_buffer.size();
2014-02-10 02:10:30 +01:00
}
void PacketPeerStream::set_stream_peer(const Ref<StreamPeer> &p_peer) {
//ERR_FAIL_COND(p_peer.is_null());
2014-02-10 02:10:30 +01:00
if (p_peer.ptr() != peer.ptr()) {
ring_buffer.advance_read(ring_buffer.data_left()); // reset the ring buffer
};
peer = p_peer;
2014-02-10 02:10:30 +01:00
}
void PacketPeerStream::set_input_buffer_max_size(int p_max_size) {
//warning may lose packets
ERR_EXPLAIN("Buffer in use, resizing would cause loss of data");
ERR_FAIL_COND(ring_buffer.data_left());
ring_buffer.resize(nearest_shift(p_max_size + 4));
input_buffer.resize(next_power_of_2(p_max_size + 4));
}
int PacketPeerStream::get_input_buffer_max_size() const {
return input_buffer.size() - 4;
}
void PacketPeerStream::set_output_buffer_max_size(int p_max_size) {
output_buffer.resize(next_power_of_2(p_max_size + 4));
2014-02-10 02:10:30 +01:00
}
int PacketPeerStream::get_output_buffer_max_size() const {
return output_buffer.size() - 4;
}
2014-02-10 02:10:30 +01:00
PacketPeerStream::PacketPeerStream() {
int rbsize = GLOBAL_GET("network/limits/packet_peer_stream/max_buffer_po2");
2014-02-10 02:10:30 +01:00
ring_buffer.resize(rbsize);
input_buffer.resize(1 << rbsize);
output_buffer.resize(1 << rbsize);
2014-02-10 02:10:30 +01:00
}