Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

84 lines
2.6 KiB

// Copyright 1997 Microsoft Corporation. All Rights Reserved.
#if _MSC_VER > 1000
#pragma once
#endif
#ifndef _FSCLIENT_H_
#define _FSCLIENT_H_
//#include "msitstg.h"
#include "fs.h"
#ifdef ReleaseObjPtr
#undef ReleaseObjPtr
#endif
#define ReleaseObjPtr(pObj) \
{ \
if( pObj ) \
{ \
pObj->Release(); \
pObj= NULL; \
} \
}
class CFSClient
{
public:
CFSClient();
CFSClient(CFileSystem* pFS, PCSTR pszSubFile);
~CFSClient();
BOOL Initialize(PCSTR pszCompiledFile);
void Initialize(CFileSystem* pFS) { m_pFS = pFS; m_fNoDeleteFS = TRUE; }
HRESULT OpenStream(PCSTR pszFile, DWORD dwAccess = STGM_READ);
BOOL isStreamOpen() { return m_pSubFS != NULL; }
// ********** Internal stream functions **********
HRESULT Read(void* pDst, const ULONG cbRead, ULONG* pcbRead);
ULONG Read(PBYTE pbDest, ULONG cbBytes); // stream read
HRESULT doRead(void* pbDst, ULONG cbBytes) {
if (Read((PBYTE) pbDst, cbBytes) == cbBytes)
return S_OK;
else
return STG_E_READFAULT;
}
int tell(void) const { return m_lFileBuf + (int)(m_pCurBuf - m_pbuf); };
HRESULT seek(int pos, SEEK_TYPE seek = SK_SET);
// ********** End Internal stream functions **********
void CloseStream(void);
ULONG GetElementStat() { return GetElementStat(1, &m_Stat); }
ULONG GetElementStat(IEnumSTATSTG* pEnum, STATSTG* pStat) { return GetElementStat(1, pStat, pEnum); }
ULONG GetElementStat(ULONG nNumber, STATSTG* stat, IEnumSTATSTG* pEnum = NULL);
LPWSTR GetStatName() const { return m_Stat.pwcsName; }
DWORD GetStatType() const { return m_Stat.type; }
void WriteStorageContents(PCSTR pszRootFolder, OLECHAR* wszFSName);
// void WaitForReadAhead(void);
void ReadBuf(void);
ULONG SeekSub(int cb, int iOrigin) {
ASSERT(m_pSubFS);
return m_pSubFS->SeekSub(cb, iOrigin); }
STATSTG m_Stat;
IEnumSTATSTG* m_pEnum;
CFileSystem* m_pFS;
CSubFileSystem* m_pSubFS;
PBYTE m_pCurBuf; // current position in the buffer
PBYTE m_pEndBuf; // last position in buffer
BOOL m_fEndOfFile;
int m_lFilePos; // position in the file
int m_lFileBuf; // file position at first of buffer
PBYTE m_pbuf; // address of allocated buffer
int m_cbBuf; // buffer size
int m_cThrdRead; // result from read-ahead thread
HANDLE m_hthrd;
DWORD m_idThrd;
BOOL m_fDualCPU;
BOOL m_fNoDeleteFS;
};
#endif // _FSCLIENT_H_