blob: bb39dc6e82a0363797fa4b0a156145cf7268887b (
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
63
64
65
66
67
68
|
#ifndef __lib_base_rawfile_h
#define __lib_base_rawfile_h
#include <string>
#include <lib/base/object.h>
class iDataSource: public iObject
{
public:
virtual off_t lseek(off_t offset, int whence)=0;
virtual ssize_t read(void *buf, size_t count)=0; /* NOTE: you must be able to handle short reads! */
virtual off_t length()=0;
virtual off_t position()=0;
virtual int valid()=0;
virtual eSingleLock &getLock()=0;
virtual bool is_shared()=0;
};
class iDataSourcePositionRestorer
{
ePtr<iDataSource> &m_source;
off_t m_position;
public:
iDataSourcePositionRestorer(ePtr<iDataSource> &source)
:m_source(source)
{
if (m_source->is_shared())
m_position = m_source->position();
}
~iDataSourcePositionRestorer()
{
if (m_source->is_shared())
m_source->lseek(m_position, SEEK_SET);
}
};
class eRawFile: public iDataSource
{
DECLARE_REF(eRawFile);
eSingleLock m_lock;
public:
eRawFile();
~eRawFile();
int open(const char *filename, int cached = 0);
void setfd(int fd);
off_t lseek(off_t offset, int whence);
int close();
ssize_t read(void *buf, size_t count); /* NOTE: you must be able to handle short reads! */
off_t length();
off_t position();
int valid();
eSingleLock &getLock();
bool is_shared() { return ref.count > 1; }
private:
int m_fd; /* for uncached */
FILE *m_file; /* for cached */
int m_cached;
std::string m_basename;
off_t m_splitsize, m_totallength, m_current_offset, m_base_offset, m_last_offset;
int m_nrfiles;
void scan();
int m_current_file;
int switchOffset(off_t off);
FILE *openFileCached(int nr);
int openFileUncached(int nr);
};
#endif
|
