1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/**************************************************************************/
/* */
/* Copyright (c) 2001, 2007 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 NoMachine S.r.l. */
/* */
/* 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 */
|
