1 files changed, 128 insertions, 0 deletions
diff --git a/nxcomp/src/ReadBuffer.h b/nxcomp/src/ReadBuffer.h
new file mode 100644
index 000000000..a12bcc9b3
--- /dev/null
+++ b/nxcomp/src/ReadBuffer.h
@@ -0,0 +1,128 @@
+/**************************************************************************/
+/*                                                                        */
+/* 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.                                          */
+/*                                                                        */
+/**************************************************************************/
+
+#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 */