/**
 * The Forgotten Server - a free and open-source MMORPG server emulator
 * Copyright (C) 2016  Mark Samman <mark.samman@gmail.com>
 *
 * 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.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#ifndef FS_PROTOCOL_H_D71405071ACF4137A4B1203899DE80E1
#define FS_PROTOCOL_H_D71405071ACF4137A4B1203899DE80E1

#include "connection.h"

class Protocol : public std::enable_shared_from_this<Protocol>
{
public:
	explicit Protocol(Connection_ptr connection) : connection(connection), key(), encryptionEnabled(false), rawMessages(false) {}
	virtual ~Protocol() = default;

	// non-copyable
	Protocol(const Protocol&) = delete;
	Protocol& operator=(const Protocol&) = delete;

	virtual void parsePacket(NetworkMessage&) {}

	virtual void onSendMessage(const OutputMessage_ptr& msg) const;
	void onRecvMessage(NetworkMessage& msg);
	virtual void onRecvFirstMessage(NetworkMessage& msg) = 0;
	virtual void onConnect() {}

	bool isConnectionExpired() const {
		return connection.expired();
	}

	Connection_ptr getConnection() const {
		return connection.lock();
	}

	uint32_t getIP() const;

	//Use this function for autosend messages only
	OutputMessage_ptr getOutputBuffer(int32_t size);

	OutputMessage_ptr& getCurrentBuffer() {
		return outputBuffer;
	}

	void send(OutputMessage_ptr msg) const {
		if (auto connection = getConnection()) {
			connection->send(msg);
		}
	}

protected:
	void disconnect() const {
		if (auto connection = getConnection()) {
			connection->close();
		}
	}
	void enableXTEAEncryption() {
		encryptionEnabled = true;
	}
	void setXTEAKey(const uint32_t* key) {
		memcpy(this->key, key, sizeof(*key) * 4);
	}

	void XTEA_encrypt(OutputMessage& msg) const;
	bool XTEA_decrypt(NetworkMessage& msg) const;
	static bool RSA_decrypt(NetworkMessage& msg);

	void setRawMessages(bool value) {
		rawMessages = value;
	}

	virtual void release() {}
	friend class Connection;

	OutputMessage_ptr outputBuffer;
private:
	const ConnectionWeak_ptr connection;
	uint32_t key[4];
	bool encryptionEnabled;
	bool rawMessages;
};

#endif