blob: 2885f7224da72099e3060b3899d44b6a7d487dd9 (
plain)
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
|
#ifndef __CSOURCE_H
#define __CSOURCE_H
class CInfo
{
public:
virtual unsigned long size() = 0;
};
class CSource : public CInfo
{
public:
virtual int get() = 0;
};
class CSink : public CInfo
{
public:
virtual void put(unsigned char c) = 0;
};
class CMemSource : public CSource
{
unsigned long m_total_size;
unsigned long m_current;
unsigned char* m_buffer;
public:
CMemSource(unsigned char* _buffer, unsigned long _size)
:
m_total_size(_size),
m_current(0),
m_buffer(_buffer)
{}
unsigned long size() { return m_current; }
int get()
{
return ((m_current < m_total_size) ? m_buffer[m_current++] : -1);
}
};
class CMemSink : public CSink
{
unsigned long m_total_size;
unsigned long m_current;
unsigned char* m_buffer;
public:
CMemSink(unsigned char* _buffer, unsigned long _size)
:
m_total_size(_size),
m_current(0),
m_buffer(_buffer)
{}
unsigned long size() { return m_current; }
void put(unsigned char c)
{
if (m_current < m_total_size)
{
m_buffer[m_current++] = c;
}
}
};
#endif
|