#include "Stream.hpp"
const int MAX_VARINT_LENGTH = 5;
bool StreamInput::ReadBool() {
unsigned char value;
ReadData(&value, 1);
return value != 0;
}
signed char StreamInput::ReadByte() {
signed char value;
ReadData((unsigned char *) &value, 1);
endswap(value);
return value;
}
unsigned char StreamInput::ReadUByte() {
unsigned char value;
ReadData(&value, 1);
endswap(value);
return value;
}
short StreamInput::ReadShort() {
unsigned short value;
ReadData((unsigned char *) &value, 2);
endswap(value);
return value;
}
unsigned short StreamInput::ReadUShort() {
unsigned char buff[2];
ReadData(buff, 2);
unsigned short val = *(reinterpret_cast<unsigned short *>(buff));
endswap(val);
return val;
}
int StreamInput::ReadInt() {
int value;
ReadData((unsigned char *) &value, 4);
endswap(value);
return value;
}
long long StreamInput::ReadLong() {
long long value;
ReadData((unsigned char *) &value, 8);
endswap(value);
return value;
}
float StreamInput::ReadFloat() {
float value;
ReadData((unsigned char *) &value, 4);
endswap(value);
return value;
}
double StreamInput::ReadDouble() {
double value;
ReadData((unsigned char *) &value, 8);
endswap(value);
return value;
}
std::string StreamInput::ReadString() {
int strLength = ReadVarInt();
unsigned char *buff = new unsigned char[strLength + 1];
ReadData(buff, strLength);
buff[strLength] = 0;
std::string str((char *) buff);
delete buff;
return str;
}
std::string StreamInput::ReadChat() {
std::string str, jsonStr = ReadString();
nlohmann::json json;
try {
json = nlohmann::json::parse(jsonStr);
} catch (std::exception &e) {
LOG(WARNING) << "Chat json parsing failed: " << e.what();
LOG(WARNING) << "Corrupted json: " << jsonStr;
return "";
}
if (json.find("translate") != json.end())
if (json["translate"].get<std::string>() == "multiplayer.disconnect.kicked")
return "kicked by operator";
for (auto &it:json["extra"]) {
str += it["text"].get<std::string>();
}
return str;
}
int StreamInput::ReadVarInt() {
unsigned char data[MAX_VARINT_LENGTH] = {0};
size_t dataLen = 0;
do {
ReadData(&data[dataLen], 1);
} while ((data[dataLen++] & 0x80) != 0);
int readed = 0;
int result = 0;
char read;
do {
read = data[readed];
int value = (read & 0b01111111);
result |= (value << (7 * readed));
readed++;
} while ((read & 0b10000000) != 0);
return result;
}
long long StreamInput::ReadVarLong() {
return 0;
}
std::vector<unsigned char> StreamInput::ReadEntityMetadata() {
return std::vector<unsigned char>();
}
std::vector<unsigned char> StreamInput::ReadSlot() {
return std::vector<unsigned char>();
}
std::vector<unsigned char> StreamInput::ReadNbtTag() {
return std::vector<unsigned char>();
}
Vector StreamInput::ReadPosition() {
unsigned long long t = ReadLong();
int x = t >> 38;
int y = (t >> 26) & 0xFFF;
int z = t << 38 >> 38;
if (x >= pow(2, 25)) {
x -= pow(2, 26);
}
if (y >= pow(2, 11)) {
y -= pow(2, 12);
}
if (z >= pow(2, 25)) {
z -= pow(2, 26);
}
return Vector(x, y, z);
}
unsigned char StreamInput::ReadAngle() {
return ReadUByte();
}
std::vector<unsigned char> StreamInput::ReadUuid() {
unsigned char buff[16];
ReadData(buff, 16);
endswap(buff, 16);
return std::vector<unsigned char>(buff, buff + 16);
}
std::vector<unsigned char> StreamInput::ReadByteArray(size_t arrLength) {
unsigned char *buffer = new unsigned char[arrLength];
ReadData(buffer, arrLength);
std::vector<unsigned char> ret(buffer, buffer + arrLength);
delete buffer;
return ret;
}
void StreamOutput::WriteBool(bool value) {
unsigned char val = value ? 1 : 0;
endswap(val);
WriteData(&val, 1);
}
void StreamOutput::WriteByte(signed char value) {
}
void StreamOutput::WriteUByte(unsigned char value) {
endswap(value);
WriteData(&value,1);
}
void StreamOutput::WriteShort(short value) {
}
void StreamOutput::WriteUShort(unsigned short value) {
endswap(value);
WriteData((unsigned char *) &value, 2);
}
void StreamOutput::WriteInt(int value) {
endswap(value);
WriteData((unsigned char *) &value, 4);
}
void StreamOutput::WriteLong(long long value) {
}
void StreamOutput::WriteFloat(float value) {
endswap(value);
WriteData((unsigned char *) &value, 4);
}
void StreamOutput::WriteDouble(double value) {
endswap(value);
WriteData((unsigned char *) &value, 8);
}
void StreamOutput::WriteString(std::string value) {
WriteVarInt(value.size());
WriteData((unsigned char *) value.data(), value.size());
}
void StreamOutput::WriteChat(std::string value) {
}
void StreamOutput::WriteVarInt(int value) {
unsigned char buff[5];
size_t len = 0;
do {
unsigned char temp = (unsigned char) (value & 0b01111111);
value >>= 7;
if (value != 0) {
temp |= 0b10000000;
}
buff[len] = temp;
len++;
} while (value != 0);
WriteData(buff, len);
}
void StreamOutput::WriteVarLong(long long value) {
}
void StreamOutput::WriteEntityMetadata(std::vector<unsigned char> value) {
}
void StreamOutput::WriteSlot(std::vector<unsigned char> value) {
}
void StreamOutput::WriteNbtTag(std::vector<unsigned char> value) {
}
void StreamOutput::WritePosition(Vector value) {
}
void StreamOutput::WriteAngle(unsigned char value) {
}
void StreamOutput::WriteUuid(std::vector<unsigned char> value) {
}
void StreamOutput::WriteByteArray(std::vector<unsigned char> value) {
}
void StreamBuffer::ReadData(unsigned char *buffPtr, size_t buffLen) {
size_t bufferLengthLeft = buffer + bufferLength - bufferPtr;
if (bufferLengthLeft < buffLen)
throw std::runtime_error("Required data is more, than in buffer available");
std::memcpy(buffPtr, bufferPtr, buffLen);
bufferPtr += buffLen;
}
void StreamBuffer::WriteData(unsigned char *buffPtr, size_t buffLen) {
size_t bufferLengthLeft = buffer + bufferLength - bufferPtr;
if (bufferLengthLeft < buffLen)
throw std::runtime_error("Required data is more, than in buffer available");
std::memcpy(bufferPtr, buffPtr, buffLen);
bufferPtr += buffLen;
}
StreamBuffer::StreamBuffer(unsigned char *data, size_t dataLen) {
buffer = new unsigned char[dataLen];
bufferPtr = buffer;
bufferLength = dataLen;
std::memcpy(buffer, data, dataLen);
}
StreamBuffer::StreamBuffer(size_t bufferLen) {
buffer = new unsigned char[bufferLen];
bufferPtr = buffer;
bufferLength = bufferLen;
for (unsigned char *p = buffer; p != buffer + bufferLength; ++p)
*p = 0;
}
StreamBuffer::~StreamBuffer() {
delete buffer;
}
std::vector<unsigned char> StreamBuffer::GetBuffer() {
return std::vector<unsigned char>(buffer, buffer + bufferLength);
}
void StreamCounter::WriteData(unsigned char *buffPtr, size_t buffLen) {
size += buffLen;
}
StreamCounter::StreamCounter(size_t initialSize) : size(initialSize) {
}
StreamCounter::~StreamCounter() {
}
size_t StreamCounter::GetCountedSize() {
return size;
}
void StreamSocket::ReadData(unsigned char *buffPtr, size_t buffLen) {
socket->Read(buffPtr, buffLen);
}
void StreamSocket::WriteData(unsigned char *buffPtr, size_t buffLen) {
socket->Write(buffPtr, buffLen);
}
StreamSocket::StreamSocket(Socket *socketPtr) : socket(socketPtr) {
}
