//////////////////////////////////////////////////////////////////////
// OpenTibia - an opensource roleplaying game
//////////////////////////////////////////////////////////////////////
//
//////////////////////////////////////////////////////////////////////
// This program 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.
//
// This program 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 this program; if not, write to the Free Software Foundation,
// Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//////////////////////////////////////////////////////////////////////
#ifndef __OTSERV_PROTOCOL_H__
#define __OTSERV_PROTOCOL_H__
#include "otsystem.h"
#include "rsa.h"
class NetworkMessage;
class OutputMessage;
class Connection;
class RSA;
typedef boost::shared_ptr<OutputMessage> OutputMessage_ptr;
class Protocol : boost::noncopyable
{
public:
Protocol(Connection* connection)
{
m_connection = connection;
m_refCount = 0;
m_rawMessages = m_encryptionEnabled = false;
m_checksumEnabled = true;
for(int8_t i = 0; i < 4; ++i)
m_key[i] = 0;
}
virtual ~Protocol() {}
virtual void onConnect() {}
virtual void onRecvFirstMessage(NetworkMessage& msg) = 0;
void onRecvMessage(NetworkMessage& msg);
void onSendMessage(OutputMessage_ptr msg);
virtual void parsePacket(NetworkMessage& msg) {}
uint32_t getIP() const;
Connection* getConnection() {return m_connection;}
const Connection* getConnection() const {return m_connection;}
void setConnection(Connection* connection) {m_connection = connection;}
int32_t addRef() {return ++m_refCount;}
int32_t unRef() {return --m_refCount;}
protected:
//Use this function for autosend messages only
OutputMessage_ptr getOutputBuffer();
void setRawMessages(bool value) {m_rawMessages = value;}
void enableChecksum() {m_checksumEnabled = true;}
void disableChecksum() {m_checksumEnabled = false;}
void enableXTEAEncryption() {m_encryptionEnabled = true;}
void disableXTEAEncryption() {m_encryptionEnabled = false;}
void setXTEAKey(const uint32_t* key) {memcpy(&m_key, key, sizeof(uint32_t) * 4);}
void XTEA_encrypt(OutputMessage& msg);
bool XTEA_decrypt(NetworkMessage& msg);
bool RSA_decrypt(RSA* rsa, NetworkMessage& msg);
virtual void releaseProtocol();
virtual void deleteProtocolTask();
friend class Connection;
private:
OutputMessage_ptr m_outputBuffer;
Connection* m_connection;
uint32_t m_refCount;
bool m_rawMessages, m_encryptionEnabled, m_checksumEnabled;
uint32_t m_key[4];
};
#endif
