blob: c07207025353abdfb70286e383e58f53fbe37a82 (
plain) (
tree)
|
|
/*
* 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 <QtCore>
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 * const socket);
bool writeMessage(QUdpSocket* socket, const QHostAddress& address, quint16 port);
void reset() { _fields.clear(); _bufferSize = 0; _mode = 0; }
bool readComplete() { return _mode == 3; }
bool writeComplete() { return _mode == 4; }
bool hasField(QByteArray& key) { return _fields.contains(key); }
const QString getFieldString(const QByteArray& key) const { return QString::fromUtf8(_fields.value(key)); }
const 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_ */
|