aboutsummaryrefslogtreecommitdiff
path: root/nxcomp/ReadBuffer.h
diff options
context:
space:
mode:
Diffstat (limited to 'nxcomp/ReadBuffer.h')
-rw-r--r--nxcomp/ReadBuffer.h120
1 files changed, 120 insertions, 0 deletions
diff --git a/nxcomp/ReadBuffer.h b/nxcomp/ReadBuffer.h
new file mode 100644
index 000000000..20130120a
--- /dev/null
+++ b/nxcomp/ReadBuffer.h
@@ -0,0 +1,120 @@
+/**************************************************************************/
+/* */
+/* Copyright (c) 2001, 2010 NoMachine, http://www.nomachine.com/. */
+/* */
+/* NXCOMP, NX protocol compression and NX extensions to this software */
+/* are copyright of NoMachine. Redistribution and use of the present */
+/* software is allowed according to terms specified in the file LICENSE */
+/* which comes in the source distribution. */
+/* */
+/* Check http://www.nomachine.com/licensing.html for applicability. */
+/* */
+/* NX and NoMachine are trademarks of Medialogic S.p.A. */
+/* */
+/* All rights reserved. */
+/* */
+/**************************************************************************/
+
+#ifndef ReadBuffer_H
+#define ReadBuffer_H
+
+#include "Misc.h"
+#include "Timestamp.h"
+#include "Transport.h"
+
+#define READ_BUFFER_DEFAULT_SIZE 8192
+
+#define PANIC
+#define WARNING
+#undef TEST
+#undef DEBUG
+
+class ReadBuffer
+{
+ public:
+
+ ReadBuffer(Transport *transport);
+
+ virtual ~ReadBuffer();
+
+ int readMessage();
+
+ void readMessage(const unsigned char *message, unsigned int length);
+
+ const unsigned char *getMessage(unsigned int &dataLength)
+ {
+ unsigned int controlLength;
+
+ return getMessage(controlLength, dataLength);
+ }
+
+ const unsigned char *getMessage(unsigned int &controlLength, unsigned int &dataLength);
+
+ unsigned int getLength() const
+ {
+ return length_;
+ }
+
+ unsigned int getRemaining() const
+ {
+ return remaining_;
+ }
+
+ int setSize(int initialReadSize, int initialbufferSize);
+
+ void fullReset();
+
+ //
+ // Check if there is a complete
+ // message in the buffer.
+ //
+
+ int checkMessage()
+ {
+ if (length_ == 0)
+ {
+ return 0;
+ }
+ else
+ {
+ unsigned int controlLength;
+ unsigned int dataLength;
+ unsigned int trailerLength;
+
+ return (locateMessage(buffer_ + start_, buffer_ + start_ + length_,
+ controlLength, dataLength, trailerLength));
+ }
+ }
+
+ protected:
+
+ virtual unsigned int suggestedLength(unsigned int pendingLength) = 0;
+
+ virtual int locateMessage(const unsigned char *start,
+ const unsigned char *end,
+ unsigned int &controlLength,
+ unsigned int &dataLength,
+ unsigned int &trailerLength) = 0;
+
+ unsigned char *allocateBuffer(unsigned int newSize);
+
+ void appendBuffer(const unsigned char *message, unsigned int length);
+
+ void convertBuffer();
+
+ Transport *transport_;
+
+ unsigned char *buffer_;
+
+ unsigned int length_;
+ unsigned int size_;
+ unsigned int start_;
+ unsigned int remaining_;
+
+ int owner_;
+
+ unsigned int initialReadSize_;
+ unsigned int maximumBufferSize_;
+};
+
+#endif /* ReadBuffer_H */