path: root/src/shared/networkmessage.h

/*
 * NetworkMessage.h
 *
 *  Created on: 18.01.2013
 *      Author: sr
 */

#ifndef NETWORKMESSAGE_H_
#define NETWORKMESSAGE_H_

#define NM_READ_OK (1)
#define NM_READ_INCOMPLETE (2)
#define NM_READ_FAILED (0)

#include <QHostAddress>
#include <QByteArray>
#include <QString>

class QAbstractSocket;
class QUdpSocket;

// define QByteArrays for message ids. this prevents implicit instantiation of the same QByteArray over and over again
#define MSGTYPE(name) static const QByteArray _ ## name ( #name )
MSGTYPE(ID);
MSGTYPE(IMG);
MSGTYPE(LOGIN);
MSGTYPE(THUMB);
MSGTYPE(X);
MSGTYPE(Y);
MSGTYPE(VNCSERVER);
MSGTYPE(VNCCLIENT);
MSGTYPE(LOCK);
MSGTYPE(HASH);
MSGTYPE(SALT1);
MSGTYPE(SALT2);
MSGTYPE(IPLIST);
MSGTYPE(PORT);
MSGTYPE(CERT);
MSGTYPE(CHALLENGE);
MSGTYPE(ENABLE);
MSGTYPE(ERROR);
MSGTYPE(TUTOR);
MSGTYPE(EXAMMODE);
MSGTYPE(ATTENTION);


static const QByteArray __TRUE("1");
static const QByteArray __FALSE("0");
static const inline QByteArray& _BOOL(bool val) { if (val) return __TRUE; return __FALSE; }

class NetworkMessage
{
private:
	char *_buffer;
	quint32 _bufferSize, _bufferPos, _lastBufferSize;
	QHash<QByteArray, QByteArray> _fields;
	int _mode; // 0 = none, 1 = reading, 2 = writing, 3 = read complete, 4 = write complete

	void allocBuffer();
	bool parseHeader(char *header);
	bool parseMessage(char *buffer);
	void serializeMessage();

public:
	NetworkMessage();
	virtual ~NetworkMessage();
	int readMessage(QAbstractSocket *socket);
	int readMessage(char* data, quint32 len);
	bool writeMessage(QAbstractSocket *socket);
	bool writeMessage(QUdpSocket* socket, const QHostAddress& address, quint16 port);
	void reset() { _fields.clear(); _bufferSize = 0; _mode = 0; }
	bool readComplete() const { return _mode == 3; }
	bool writeComplete() const { return _mode == 4; }
	bool hasField(QByteArray& key) { return _fields.contains(key); }
	QString getFieldString(const QByteArray& key) const { return QString::fromUtf8(_fields.value(key)); }
	QByteArray getFieldBytes(const QByteArray& key) const { return _fields.value(key); }
	void setField(const QByteArray& key, const QByteArray& value) { if (_mode == 1 || _mode == 2) qFatal("setField called in bad state."); _fields.insert(key, value); _mode = 0; }
	void setField(const QByteArray& key, const QString& value) { setField(key, value.toUtf8()); }
	// Convenience
	void buildErrorMessage(const QString& error);
};

#endif /* NETWORKMESSAGE_H_ */