diff options
| author | Simon Rettberg | 2016-09-09 16:31:04 +0200 |
|---|---|---|
| committer | Simon Rettberg | 2016-09-09 16:31:04 +0200 |
| commit | a39269e11e7c2d1843a4a4bf67749ae0224e4830 (patch) | |
| tree | 9870de008a4ecd0fcf3902668cffb3f574e17ba4 /src/shared/networkmessage.h | |
| parent | Rename lock/unlock scripts to kb-lock and kb-unlock so you know it's for the ... (diff) | |
| download | pvs2-a39269e11e7c2d1843a4a4bf67749ae0224e4830.tar.gz pvs2-a39269e11e7c2d1843a4a4bf67749ae0224e4830.tar.xz pvs2-a39269e11e7c2d1843a4a4bf67749ae0224e4830.zip | |
Fix infinite loop when receiving incomplete headers
Diffstat (limited to 'src/shared/networkmessage.h')
| -rw-r--r-- | src/shared/networkmessage.h | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/shared/networkmessage.h b/src/shared/networkmessage.h index 3742890..2e8c1c5 100644 --- a/src/shared/networkmessage.h +++ b/src/shared/networkmessage.h @@ -8,6 +8,10 @@ #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; @@ -56,8 +60,8 @@ private: public: NetworkMessage(); virtual ~NetworkMessage(); - bool readMessage(QAbstractSocket* socket); - bool readMessage(char* data, quint32 len); + 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; } |