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
|
#ifndef __FILEREADER_H_
#define __FILEREADER_H_
#include "config.h"
#include "acbuf.h"
#include "fileio.h"
#include "filelocks.h"
namespace acng
{
class IDecompressor;
/*!
* Helper class used to read files.
*
* Could use boost::iostream templates for most of that, but Boost became such a monster nowadays.
* And for my work, my class behaves smarter.
*/
class filereader {
public:
filereader();
~filereader();
//! Opens any supported file.
/* @param sFilename File to open. Writes the base
* filename w/o suffix or path prefix back into it
* @param bCriticalOpen Terminate program on failure
*/
bool OpenFile(const mstring & sFilename, bool bNoMagic=false, unsigned nFakeTrailingNewlines=0);
//////! Filename with all prepended path and compressed suffix stripped
//////void GetBaseFileName(mstring & sOut);
//! Returns lines when beginning with non-space, otherwise empty string.
//! @return False on errors.
bool GetOneLine(mstring & sOut, bool bForceUncompress=false);
unsigned GetCurrentLine() const { return m_nCurLine;} ;
bool CheckGoodState(bool bTerminateOnErrors, cmstring *reportFilePath=nullptr) const;
bool GetChecksum(int csType, uint8_t out[], off_t &scannedSize, FILE *pDumpFile=nullptr);
static bool GetChecksum(const mstring & sFileName, int csType, uint8_t out[],
bool bTryUnpack, off_t &scannedSize, FILE *pDumpFile=nullptr);
inline const char *GetBuffer() const { return m_szFileBuf; };
inline size_t GetSize() const { return m_nBufSize; };
void Close();
const mstring& getSErrorString() const
{
return m_sErrorString;
}
private:
bool m_bError, m_bEof;
// XXX: not totally happy, this could be a simple const char* for most usecases
mstring m_sErrorString;
char *m_szFileBuf;
size_t m_nBufSize, m_nBufPos;
acbuf m_UncompBuf; // uncompressed window
// visible position reporting
unsigned m_nCurLine;
int m_fd;
int m_nEofLines;
std::unique_ptr<IDecompressor> m_Dec;
// not to be copied
filereader& operator=(const filereader&);
filereader(const filereader&);
std::unique_ptr<TFileShrinkGuard> m_mmapLock;
};
extern uint_fast16_t hexmap[];
inline bool CsEqual(const char *sz, uint8_t b[], unsigned short binLen)
{
CUCHAR *a=(CUCHAR*)sz;
if(!a)
return false;
for(int i=0; i<binLen;i++)
{
if(!*a)
return false;
uint_fast16_t r=hexmap[a[i*2]] * 16 + hexmap[a[i*2+1]];
if(r != b[i]) return false;
}
return true;
};
bool Bz2compressFile(const char *pathIn, const char*pathOut);
}
#endif // __FILEREADER_H
|
