networkmessage.h

//////////////////////////////////////////////////////////////////////
// OTAdmin - OpenTibia
//////////////////////////////////////////////////////////////////////
//
//////////////////////////////////////////////////////////////////////
// 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 __NETWORK_MESSAGE_H__
#define __NETWORK_MESSAGE_H__

#include "definitions.h"
#include "rsa.h"

#define RETRY_TIME 10000

enum SocketCode{
	SOCKET_CODE_OK,
	SOCKET_CODE_TIMEOUT,
	SOCKET_CODE_ERROR,
};

class NetworkMessage
{
public:
	// constructor/destructor
	NetworkMessage();
	~NetworkMessage();

	// resets the internal buffer to an empty message
	void Reset();

	// socket functions
	SocketCode ReadFromSocket(SOCKET socket, int timeout = RETRY_TIME);
	SocketCode WriteToSocket(SOCKET socket, int timeout = RETRY_TIME);

	// simply read functions for incoming message
	uint8_t  InspectByte();
	uint8_t  GetByte();
	uint16_t GetU16();
	uint32_t   GetU32();
	std::string    GetString();
	std::string	 GetRaw();

	void setEncryptionState(bool state);
	void setEncryptionKey(const uint32_t* key);

	// skips count unknown/unused bytes in an incoming message
	void SkipBytes(int count);

	// simply write functions for outgoing message
	void AddByte(uint8_t  value);
	void AddU16 (uint16_t value);
	void AddU32 (uint32_t value);

	void AddString(const std::string &value);
	void AddString(const char* value);

  	int getMessageLength(){
		return m_MsgSize;
	}
	
	bool RSA_encrypt();

	void setRSAInstance(RSA* rsa);

protected:
	inline bool canAdd(int size){
    	return (size + m_ReadPos < NETWORKMESSAGE_MAXSIZE - 16);
  	};
  	
  	void XTEA_encrypt();
  	void XTEA_decrypt();
  	
  	int m_MsgSize;
  	int m_ReadPos;

	uint8_t m_MsgBuf[NETWORKMESSAGE_MAXSIZE];
	
	static bool m_encryptionEnabled;
	static bool m_keyset;
	static uint32_t m_key[4];

	static RSA* m_RSA;
};


#endif // #ifndef __NETWORK_MESSAGE_H__