You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
210 lines
5.4 KiB
210 lines
5.4 KiB
/* The MIT License
|
|
*
|
|
* Copyright (c) 2010 OTClient, https://github.com/edubart/otclient
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
|
|
#ifndef NETWORKMESSAGE_H
|
|
#define NETWORKMESSAGE_H
|
|
|
|
#include <prerequisites.h>
|
|
|
|
class InputMessage
|
|
{
|
|
public:
|
|
enum {
|
|
INPUTMESSAGE_MAXSIZE = 16834,
|
|
HEADER_LENGTH = 2
|
|
};
|
|
|
|
InputMessage() : m_messageSize(0), m_readPos(HEADER_LENGTH) { }
|
|
~InputMessage() { }
|
|
|
|
inline void reset() {
|
|
m_messageSize = 0;
|
|
m_readPos = HEADER_LENGTH;
|
|
}
|
|
|
|
uint16_t decodeHeader() {
|
|
return (int32_t)(m_buffer[0] | m_buffer[1] << 8);
|
|
}
|
|
|
|
uint8_t getByte() {
|
|
return m_buffer[m_readPos++];
|
|
}
|
|
|
|
uint16_t getU16() {
|
|
uint16_t v = *(uint16_t*)(m_buffer + m_readPos);
|
|
m_readPos += 2;
|
|
return v;
|
|
}
|
|
|
|
uint32_t getU32() {
|
|
uint32_t v = *(uint32_t*)(m_buffer + m_readPos);
|
|
m_readPos += 4;
|
|
return v;
|
|
}
|
|
|
|
uint64_t getU64() {
|
|
uint64_t v = *(uint64_t*)(m_buffer + m_readPos);
|
|
m_readPos += 8;
|
|
return v;
|
|
}
|
|
|
|
std::string getString() {
|
|
uint16_t stringlen = getU16();
|
|
if(stringlen >= (INPUTMESSAGE_MAXSIZE - m_readPos))
|
|
return std::string();
|
|
|
|
char* v = (char*)(m_buffer + m_readPos);
|
|
m_readPos += stringlen;
|
|
return std::string(v, stringlen);
|
|
}
|
|
|
|
void skipBytes(int count) { m_readPos += count; }
|
|
|
|
int32_t getMessageLength() const {return m_messageSize; }
|
|
void setMessageLength(int32_t newSize) { m_messageSize = newSize; }
|
|
|
|
int32_t getReadPos() const { return m_readPos; }
|
|
const char *getBuffer() const { return (char*)&m_buffer[0]; }
|
|
|
|
private:
|
|
uint16_t m_messageSize;
|
|
uint16_t m_readPos;
|
|
uint8_t m_buffer[INPUTMESSAGE_MAXSIZE];
|
|
};
|
|
|
|
class OutputMessage
|
|
{
|
|
public:
|
|
enum {
|
|
OUTPUTMESSAGE_MAXSIZE = 1460
|
|
};
|
|
|
|
OutputMessage() : m_outputBufferStart(4), m_messageSize(0), m_writePos(4) { }
|
|
~OutputMessage() { }
|
|
|
|
void reset() {
|
|
m_messageSize = 0;
|
|
m_writePos = 4;
|
|
m_outputBufferStart = 4;
|
|
}
|
|
|
|
void addByte(uint8_t value)
|
|
{
|
|
if(!canAdd(1))
|
|
return;
|
|
m_buffer[m_writePos++] = value;
|
|
m_messageSize++;
|
|
}
|
|
|
|
void addU16(uint16_t value)
|
|
{
|
|
if(!canAdd(2))
|
|
return;
|
|
*(uint16_t*)(m_buffer + m_writePos) = value;
|
|
m_writePos += 2;
|
|
m_messageSize += 2;
|
|
}
|
|
|
|
void addU32(uint32_t value)
|
|
{
|
|
if(!canAdd(4))
|
|
return;
|
|
*(uint32_t*)(m_buffer + m_writePos) = value;
|
|
m_writePos += 4;
|
|
m_messageSize += 4;
|
|
}
|
|
|
|
void addU64(uint64_t value)
|
|
{
|
|
if(!canAdd(8))
|
|
return;
|
|
*(uint64_t*)(m_buffer + m_writePos) = value;
|
|
m_writePos += 8;
|
|
m_messageSize += 8;
|
|
}
|
|
|
|
void addBytes(const char* bytes, uint32_t size)
|
|
{
|
|
if(!canAdd(size) || size > 8192)
|
|
return;
|
|
|
|
memcpy(m_buffer + m_writePos, bytes, size);
|
|
m_writePos += size;
|
|
m_messageSize += size;
|
|
}
|
|
|
|
void addPaddingBytes(uint32_t n) {
|
|
if(!canAdd(n))
|
|
return;
|
|
|
|
memset((void*)&m_buffer[m_writePos], 0x33, n);
|
|
m_messageSize = m_messageSize + n;
|
|
}
|
|
|
|
void addString(const char* value)
|
|
{
|
|
uint32_t stringlen = (uint32_t)strlen(value);
|
|
if(!canAdd(stringlen + 2) || stringlen > 8192)
|
|
return;
|
|
|
|
addU16(stringlen);
|
|
strcpy((char*)(m_buffer + m_writePos), value);
|
|
m_writePos += stringlen;
|
|
m_messageSize += stringlen;
|
|
}
|
|
|
|
void addString(const std::string &value) {
|
|
addString(value.c_str());
|
|
}
|
|
|
|
void writeMessageLength() {
|
|
*(uint16_t*)(m_buffer + 2) = m_messageSize;
|
|
m_messageSize += 2;
|
|
m_outputBufferStart = 2;
|
|
}
|
|
|
|
void writeCryptoHeader() {
|
|
*(uint16_t*)(m_buffer) = m_messageSize;
|
|
m_messageSize += 2;
|
|
m_outputBufferStart = 0;
|
|
}
|
|
|
|
int32_t getMessageLength() const { return m_messageSize; }
|
|
void setMessageLength(int32_t newSize) { m_messageSize = newSize; }
|
|
|
|
const char *getBuffer() const { return (char*)&m_buffer[0]; }
|
|
const char *getOutputBuffer() const { return (char*)&m_buffer[m_outputBufferStart]; }
|
|
|
|
private:
|
|
inline bool canAdd(int size) {
|
|
return (size + m_writePos < OUTPUTMESSAGE_MAXSIZE);
|
|
}
|
|
|
|
uint16_t m_outputBufferStart;
|
|
uint16_t m_messageSize;
|
|
uint16_t m_writePos;
|
|
uint8_t m_buffer[OUTPUTMESSAGE_MAXSIZE];
|
|
};
|
|
|
|
#endif //NETWORKMESSAGE_H
|