aboutsummaryrefslogtreecommitdiff
path: root/nxcomp/src/ClientReadBuffer.cpp
diff options
context:
space:
mode:
authorMihai Moldovan <ionic@ionic.de>2017-07-26 10:38:46 +0200
committerMihai Moldovan <ionic@ionic.de>2017-07-26 10:38:46 +0200
commit7897834ce578dc394649ed09acb9f95fc04915ff (patch)
treea0bcbfa2c4ae1111ad3d49390c7abd0ae3a9a2ed /nxcomp/src/ClientReadBuffer.cpp
parent9193d11eeeea933e293acd5e0f03fa4e9887186b (diff)
parent4afc641fdd8c03bd708f50e0c3691b5de7ea1dba (diff)
downloadnx-libs-7897834ce578dc394649ed09acb9f95fc04915ff.tar.gz
nx-libs-7897834ce578dc394649ed09acb9f95fc04915ff.tar.bz2
nx-libs-7897834ce578dc394649ed09acb9f95fc04915ff.zip
Merge branch 'sunweaver-pr/nxcomp-autoreconf' into 3.6.x
Attributes GH PR #480: https://github.com/ArcticaProject/nx-libs/pull/480
Diffstat (limited to 'nxcomp/src/ClientReadBuffer.cpp')
-rw-r--r--nxcomp/src/ClientReadBuffer.cpp178
1 files changed, 178 insertions, 0 deletions
diff --git a/nxcomp/src/ClientReadBuffer.cpp b/nxcomp/src/ClientReadBuffer.cpp
new file mode 100644
index 000000000..c8f4f69a4
--- /dev/null
+++ b/nxcomp/src/ClientReadBuffer.cpp
@@ -0,0 +1,178 @@
+/**************************************************************************/
+/* */
+/* Copyright (c) 2001, 2011 NoMachine (http://www.nomachine.com) */
+/* Copyright (c) 2008-2014 Oleksandr Shneyder <o.shneyder@phoca-gmbh.de> */
+/* Copyright (c) 2014-2016 Ulrich Sibiller <uli42@gmx.de> */
+/* Copyright (c) 2014-2016 Mihai Moldovan <ionic@ionic.de> */
+/* Copyright (c) 2011-2016 Mike Gabriel <mike.gabriel@das-netzwerkteam.de>*/
+/* Copyright (c) 2015-2016 Qindel Group (http://www.qindel.com) */
+/* */
+/* NXCOMP, NX protocol compression and NX extensions to this software */
+/* are copyright of the aforementioned persons and companies. */
+/* */
+/* Redistribution and use of the present software is allowed according */
+/* to terms specified in the file LICENSE.nxcomp which comes in the */
+/* source distribution. */
+/* */
+/* All rights reserved. */
+/* */
+/* NOTE: This software has received contributions from various other */
+/* contributors, only the core maintainers and supporters are listed as */
+/* copyright holders. Please contact us, if you feel you should be listed */
+/* as copyright holder, as well. */
+/* */
+/**************************************************************************/
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include "ClientReadBuffer.h"
+
+#include "ClientChannel.h"
+
+#define PANIC
+#define WARNING
+#undef TEST
+#undef DEBUG
+
+unsigned int ClientReadBuffer::suggestedLength(unsigned int pendingLength)
+{
+ //
+ // Even if the pending data is not
+ // enough to make a complete message,
+ // resize the buffer to accommodate
+ // it all.
+ //
+
+ unsigned int readLength = pendingLength;
+
+ if (pendingLength < remaining_)
+ {
+ readLength = remaining_;
+ }
+
+ return readLength;
+}
+
+int ClientReadBuffer::locateMessage(const unsigned char *start,
+ const unsigned char *end,
+ unsigned int &controlLength,
+ unsigned int &dataLength,
+ unsigned int &trailerLength)
+{
+ unsigned int size = end - start;
+
+ #ifdef TEST
+ *logofs << "ClientReadBuffer: Locating message for FD#"
+ << transport_ -> fd() << " with " << size
+ << " bytes.\n" << logofs_flush;
+ #endif
+
+ if (firstMessage_)
+ {
+ if (size < 12)
+ {
+ remaining_ = 12 - size;
+
+ #ifdef TEST
+ *logofs << "ClientReadBuffer: No message was located "
+ << "with remaining " << remaining_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ return 0;
+ }
+
+ if (*start == 0x42)
+ {
+ bigEndian_ = 1;
+ }
+ else
+ {
+ bigEndian_ = 0;
+ }
+
+ channel_ -> setBigEndian(bigEndian_);
+
+ dataLength = 12 + RoundUp4(GetUINT(start + 6, bigEndian_)) +
+ RoundUp4(GetUINT(start + 8, bigEndian_));
+
+ //
+ // Send the data immediately if this is unlikely
+ // to be a X connection attempt.
+ //
+
+ if (dataLength > 4096)
+ {
+ #ifdef WARNING
+ *logofs << "ClientReadBuffer: WARNING! Flushing suspicious X "
+ << "connection with first request of " << dataLength
+ << " bytes.\n" << logofs_flush;
+ #endif
+
+ dataLength = size;
+ }
+ }
+ else
+ {
+ if (size < 4)
+ {
+ remaining_ = 4 - size;
+
+ #ifdef TEST
+ *logofs << "ClientReadBuffer: No message was located "
+ << "with remaining " << remaining_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ return 0;
+ }
+
+ dataLength = (GetUINT(start + 2, bigEndian_) << 2);
+
+ if (dataLength < 4)
+ {
+ #ifdef TEST
+ *logofs << "ClientReadBuffer: WARNING! Assuming length 4 "
+ << "for suspicious message of length " << dataLength
+ << ".\n" << logofs_flush;
+ #endif
+
+ dataLength = 4;
+ }
+ }
+
+ #ifdef TEST
+ *logofs << "ClientReadBuffer: Length of the next message is "
+ << dataLength << ".\n" << logofs_flush;
+ #endif
+
+ if (size < dataLength)
+ {
+ remaining_ = dataLength - size;
+
+ #ifdef TEST
+ *logofs << "ClientReadBuffer: No message was located "
+ << "with remaining " << remaining_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ return 0;
+ }
+
+ firstMessage_ = 0;
+
+ controlLength = 0;
+ trailerLength = 0;
+
+ remaining_ = 0;
+
+ #ifdef TEST
+ *logofs << "ClientReadBuffer: Located message with "
+ << "remaining " << remaining_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ return 1;
+}