-
Notifications
You must be signed in to change notification settings - Fork 14
/
SocketStream.cpp
142 lines (119 loc) · 3.58 KB
/
SocketStream.cpp
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
#include "SocketStream.h"
#include <string.h>
#define MAXDATA (1024 * 1024)
CSocketStream::CSocketStream()
{
m_pInBuffer = new unsigned char[MAXDATA];
m_pOutBuffer = new unsigned char[MAXDATA];
m_nInBufferIndex = m_nInBufferSize = 0;
m_nOutBufferIndex = m_nOutBufferSize = 0;
}
CSocketStream::~CSocketStream()
{
delete[] m_pInBuffer;
delete[] m_pOutBuffer;
}
unsigned char *CSocketStream::GetInput(void)
{
return m_pInBuffer;
}
void CSocketStream::SetInputSize(unsigned int nSize)
{
m_nInBufferIndex = 0; //seek to the beginning of the input buffer
m_nInBufferSize = nSize;
}
unsigned char *CSocketStream::GetOutput(void)
{
return m_pOutBuffer; //output buffer
}
unsigned int CSocketStream::GetOutputSize(void)
{
return m_nOutBufferSize; //number of bytes of data in the output buffer
}
unsigned int CSocketStream::GetBufferSize(void)
{
return MAXDATA; //size of input/output buffer
}
unsigned int CSocketStream::Read(void *pData, unsigned int nSize)
{
if (nSize > m_nInBufferSize - m_nInBufferIndex) //over the number of bytes of data in the input buffer
nSize = m_nInBufferSize - m_nInBufferIndex;
memcpy(pData, m_pInBuffer + m_nInBufferIndex, nSize);
m_nInBufferIndex += nSize;
return nSize;
}
unsigned int CSocketStream::Read(unsigned long *pnValue)
{
unsigned int i, n;
unsigned char *p, pBuffer[sizeof(unsigned long)];
p = (unsigned char *)pnValue;
n = Read(pBuffer, sizeof(unsigned long));
for (i = 0; i < n; i++) //reverse byte order
p[sizeof(unsigned long) - 1 - i] = pBuffer[i];
return n;
}
unsigned int CSocketStream::Read8(unsigned __int64 *pnValue)
{
unsigned int i, n;
unsigned char *p, pBuffer[sizeof(unsigned __int64)];
p = (unsigned char *)pnValue;
n = Read(pBuffer, sizeof(unsigned __int64));
for (i = 0; i < n; i++) //reverse byte order
p[sizeof(unsigned __int64) - 1 - i] = pBuffer[i];
return n;
}
unsigned int CSocketStream::Skip(unsigned int nSize)
{
if (nSize > m_nInBufferSize - m_nInBufferIndex) //over the number of bytes of data in the input buffer
nSize = m_nInBufferSize - m_nInBufferIndex;
m_nInBufferIndex += nSize;
return nSize;
}
unsigned int CSocketStream::GetSize(void)
{
return m_nInBufferSize - m_nInBufferIndex; //number of bytes of rest data in the input buffer
}
void CSocketStream::Write(void *pData, unsigned int nSize)
{
if (m_nOutBufferIndex + nSize > MAXDATA) //over the size of output buffer
nSize = MAXDATA - m_nOutBufferIndex;
memcpy(m_pOutBuffer + m_nOutBufferIndex, pData, nSize);
m_nOutBufferIndex += nSize;
if (m_nOutBufferIndex > m_nOutBufferSize)
m_nOutBufferSize = m_nOutBufferIndex;
}
void CSocketStream::Write(unsigned long nValue)
{
int i;
unsigned char *p, pBuffer[sizeof(unsigned long)];
p = (unsigned char *)&nValue;
for (i = sizeof(unsigned long) - 1; i >= 0; i--) //reverse byte order
pBuffer[i] = p[sizeof(unsigned long) - 1 - i];
Write(pBuffer, sizeof(unsigned long));
}
void CSocketStream::Write8(unsigned __int64 nValue)
{
int i;
unsigned char *p, pBuffer[sizeof(unsigned __int64)];
p = (unsigned char *)&nValue;
for (i = sizeof(unsigned __int64) - 1; i >= 0; i--) //reverse byte order
pBuffer[i] = p[sizeof(unsigned __int64) - 1 - i];
Write(pBuffer, sizeof(unsigned __int64));
}
void CSocketStream::Seek(int nOffset, int nFrom)
{
if (nFrom == SEEK_SET)
m_nOutBufferIndex = nOffset;
else if (nFrom == SEEK_CUR)
m_nOutBufferIndex += nOffset;
else if (nFrom == SEEK_END)
m_nOutBufferIndex = m_nOutBufferSize + nOffset;
}
int CSocketStream::GetPosition(void)
{
return m_nOutBufferIndex;
}
void CSocketStream::Reset(void)
{
m_nOutBufferIndex = m_nOutBufferSize = 0; //clear output buffer
}