1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
/*
* NetworkMessage.h
*
* Created on: 18.01.2013
* Author: sr
*/
#ifndef NETWORKMESSAGE_H_
#define NETWORKMESSAGE_H_
#include <QtCore>
class QAbstractSocket;
class QUdpSocket;
// define qstrings for message ids. this prevents implicit instanciation of the same qstrings over and over again
#define MSGTYPE(name) static const QString _ ## 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(ERROR);
class NetworkMessage
{
private:
char *_buffer;
quint32 _bufferSize, _bufferPos, _lastBufferSize;
QHash<QString, 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();
bool readMessage(QAbstractSocket* socket);
bool 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; }
const bool readComplete() const { return _mode == 3; }
const bool writeComplete() const { return _mode == 4; }
const bool hasField(const QString& key) const { return _fields.contains(key); }
const QString getFieldString(const QString& key) const { return QString::fromUtf8(_fields.value(key)); }
const QByteArray getFieldBytes(const QString& key) const { return _fields.value(key); }
void setField(const QString& key, const QByteArray& value) { if (_mode == 1 || _mode == 2) qFatal("setField called in bad state."); _fields.insert(key, value); _mode = 0; }
void setField(const QString& key, const QString& value) { setField(key, value.toUtf8()); }
// Convenience
void buildErrorMessage(const QString& error);
void buildErrorMessage(const char* error);
};
#endif /* NETWORKMESSAGE_H_ */
|
