Files
Pyarmor-Static-Unpack-1shot/data.h

53 lines
1.0 KiB
C
Raw Normal View History

2009-07-24 08:35:21 +00:00
#ifndef _PYC_FILE_H
#define _PYC_FILE_H
#include <cstdio>
class PycData {
public:
PycData() { }
virtual ~PycData() { }
virtual bool isOpen() const = 0;
virtual bool atEof() const = 0;
virtual int getByte() = 0;
virtual int getBuffer(int bytes, void* buffer) = 0;
int get16();
int get32();
};
class PycFile : public PycData {
public:
PycFile(const char* filename);
~PycFile() { if (m_stream) fclose(m_stream); }
bool isOpen() const { return (m_stream != 0); }
bool atEof() const;
int getByte();
int getBuffer(int bytes, void* buffer);
private:
FILE* m_stream;
};
class PycBuffer : public PycData {
public:
PycBuffer(const void* buffer, int size)
: m_buffer((const unsigned char*)buffer), m_size(size), m_pos(0) { }
~PycBuffer() { }
bool isOpen() const { return (m_buffer != 0); }
bool atEof() const { return (m_pos == m_size); }
int getByte();
int getBuffer(int bytes, void* buffer);
private:
const unsigned char* m_buffer;
int m_size, m_pos;
};
#endif