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
|
#ifndef __ARCHIVE_BUFFERED_H
#define __ARCHIVE_BUFFERED_H
#include <map>
#include <boost/thread/mutex.hpp>
#include "ArchiveBase.h"
class FileBuffer
{
public:
FileBuffer();
virtual ~FileBuffer();
int size;
int pos;
char* data;
};
class ABOpenFile_t : public FileBuffer
{
public:
virtual ~ABOpenFile_t();
};
// Provides a helper implementation for archive types that can only uncompress one file to
// memory at a time
class CArchiveBuffered : public CArchiveBase
{
public:
CArchiveBuffered(const std::string& name);
virtual ~CArchiveBuffered(void);
virtual int OpenFile(const std::string& fileName);
virtual int ReadFile(int handle, void* buffer, int numBytes);
virtual void CloseFile(int handle);
virtual void Seek(int handle, int pos);
virtual int Peek(int handle);
virtual bool Eof(int handle);
virtual int FileSize(int handle);
protected:
boost::mutex archiveLock; // neither 7zip nor zlib are threadsafe
int curFileHandle;
std::map<int, FileBuffer*> fileHandles;
FileBuffer* GetEntireFile(const std::string& fileName)
{
boost::mutex::scoped_lock lock(archiveLock);
return GetEntireFileImpl(fileName);
};
virtual FileBuffer* GetEntireFileImpl(const std::string& fileName) = 0;
FileBuffer* GetOpenFile(int handle);
};
#endif
|
