diff options
Diffstat (limited to 'src/shared/networkmessage.h')
| -rw-r--r-- | src/shared/networkmessage.h | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/src/shared/networkmessage.h b/src/shared/networkmessage.h index c072070..327f544 100644 --- a/src/shared/networkmessage.h +++ b/src/shared/networkmessage.h @@ -12,7 +12,9 @@ #define NM_READ_INCOMPLETE (2) #define NM_READ_FAILED (0) -#include <QtCore> +#include <QHostAddress> +#include <QByteArray> +#include <QString> class QAbstractSocket; class QUdpSocket; @@ -62,16 +64,16 @@ private: public: NetworkMessage(); virtual ~NetworkMessage(); - int readMessage(QAbstractSocket* socket); + int readMessage(QAbstractSocket *socket); int readMessage(char* data, quint32 len); - bool writeMessage(QAbstractSocket * const socket); + bool writeMessage(QAbstractSocket *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 readComplete() const { return _mode == 3; } + bool writeComplete() const { 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); } + 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 |