183 lines
4.5 KiB
C++
183 lines
4.5 KiB
C++
// TestStg.cpp -- A simple text program for the IITStorage interface wrapper.
|
|
|
|
#include <Windows.h>
|
|
#include <malloc.h>
|
|
#include <stdio.h>
|
|
#include <wchar.h>
|
|
#include <objbase.h>
|
|
#include <urlmon.h>
|
|
#include "MSITSTG.h"
|
|
|
|
typedef IUnknown *PIUnknown;
|
|
|
|
#define ReleaseObjPtr(pObj) \
|
|
{ \
|
|
if (pObj) \
|
|
{ \
|
|
IUnknown *pUnk= pObj; \
|
|
pObj= NULL; \
|
|
pUnk->Release(); \
|
|
} \
|
|
}
|
|
|
|
#define StorageName L".\\test.df"
|
|
|
|
|
|
const char aItemTypes[] = "?\\:#@!";
|
|
|
|
// ITControlData defines the current structure of the control data
|
|
// used by ITS at the moment. Don't count on it being correct forever.
|
|
// Instead you should use IITStorage::EditControlData to set up control
|
|
// data. [Real soon now...]
|
|
|
|
typedef struct _ITControlData
|
|
{
|
|
UINT cdwFollowing;
|
|
|
|
DWORD cdwITFS_Control;
|
|
DWORD dwMagicITS;
|
|
DWORD dwVersionITS;
|
|
DWORD cbDirectoryBlock;
|
|
DWORD cMinCacheEntries;
|
|
DWORD fFlags;
|
|
|
|
UINT cdwLZXData;
|
|
|
|
DWORD dwLZXMagic;
|
|
DWORD dwVersionLZX;
|
|
DWORD cbResetBlock;
|
|
DWORD cbWindowSize;
|
|
DWORD cbSecondPartition;
|
|
DWORD cbExeOpt;
|
|
|
|
} ITControlData, *PITControlData;
|
|
|
|
#define Buff_Cnt (200)
|
|
|
|
void main(int cArgs, char **ppchArgs)
|
|
{
|
|
#define SHUTDOWN { DebugBreak(); goto shutdown; }
|
|
#define BAILOUT { DebugBreak(); goto bail_out; }
|
|
|
|
IITStorage *pITStorage = NULL;
|
|
IStorage *pStorage = NULL;
|
|
|
|
IStorage *rgpstg[100];
|
|
IStream *pstm = NULL;
|
|
|
|
|
|
WCHAR StmName[1024];
|
|
WCHAR StgName[1024];
|
|
|
|
int i,j;
|
|
ULONG ulData[Buff_Cnt];
|
|
ULONG iData;
|
|
ULONG cbWritten= 0;
|
|
ULONG ulDataRead[Buff_Cnt];
|
|
ULONG cbRead = 0;
|
|
WCHAR suffix[10];
|
|
ITControlData itcd;
|
|
PITControlData pitcd = &itcd;
|
|
|
|
itcd.cdwFollowing = 13;
|
|
itcd.cdwITFS_Control = 5;
|
|
itcd.dwMagicITS = 'I' | ('T' << 8) | ('S' << 16) | ('C' << 24);
|
|
itcd.dwVersionITS = 1;
|
|
itcd.cbDirectoryBlock = 512;
|
|
itcd.cMinCacheEntries = 20;
|
|
itcd.fFlags = 0x00000000; // Make compression the default.
|
|
itcd.cdwLZXData = 6;
|
|
itcd.dwLZXMagic = 'L' | ('Z' << 8 ) | ('X' << 16) | ('C' << 24);
|
|
itcd.dwVersionLZX = 2;
|
|
itcd.cbResetBlock = 2;//32*1024; // 0x40000;
|
|
itcd.cbWindowSize = 2;//28*1024; //0x20000;
|
|
itcd.cbSecondPartition = 1;//64*1024; //0x10000;
|
|
itcd.cbExeOpt = FALSE;
|
|
|
|
if (!SUCCEEDED(CoInitialize(NULL)))
|
|
return;
|
|
|
|
IClassFactory *pICFITStorage = NULL;
|
|
|
|
HRESULT hr= CoGetClassObject(CLSID_ITStorage, CLSCTX_INPROC_SERVER, NULL,
|
|
IID_IClassFactory, (VOID **)&pICFITStorage
|
|
);
|
|
|
|
if (!SUCCEEDED(hr))
|
|
SHUTDOWN
|
|
|
|
hr= pICFITStorage->CreateInstance(NULL, IID_ITStorage,
|
|
(VOID **)&pITStorage
|
|
);
|
|
|
|
ReleaseObjPtr(pICFITStorage);
|
|
|
|
if (!SUCCEEDED(hr))
|
|
SHUTDOWN
|
|
|
|
hr = pITStorage->EditControlData((PITS_Control_Data *)&pitcd, 0);
|
|
hr = pITStorage->SetControlData(PITS_Control_Data(&itcd));
|
|
|
|
hr= pITStorage->StgCreateDocfile(StorageName,
|
|
STGM_READWRITE | STGM_SHARE_EXCLUSIVE
|
|
| STGM_DIRECT
|
|
| STGM_CREATE,
|
|
0, &pStorage
|
|
);
|
|
|
|
if (!SUCCEEDED(hr))
|
|
BAILOUT
|
|
|
|
for (iData= Buff_Cnt; iData--; )
|
|
ulData[iData] = iData;
|
|
|
|
for (i = 0; SUCCEEDED(hr) && (i < 100); i++)
|
|
{
|
|
_itow(i, suffix, 10);
|
|
wcscpy(StgName, L"stg");
|
|
wcscat(StgName, suffix);
|
|
|
|
hr= pStorage->CreateStorage(StgName, STGM_READWRITE, 0, 0, &rgpstg[i]);
|
|
|
|
for (j = 0; SUCCEEDED(hr) && (j < 100); j++)
|
|
{
|
|
_itow(j, suffix, 10);
|
|
wcscpy(StmName, L"stm");
|
|
wcscat(StmName, suffix);
|
|
|
|
hr= rgpstg[i]->CreateStream(StmName, STGM_READWRITE, 0, 0, &pstm);
|
|
hr= pstm->Write(ulData, Buff_Cnt * sizeof(ULONG), &cbWritten);
|
|
pstm->Release();
|
|
}
|
|
}
|
|
|
|
|
|
for (i = 0; SUCCEEDED(hr) && (i < 100); i++)
|
|
{
|
|
for (j = 0; SUCCEEDED(hr) && (j < 100); j++)
|
|
{
|
|
_itow(j, suffix, 10);
|
|
wcscpy(StmName, L"stm");
|
|
wcscat(StmName, suffix);
|
|
|
|
hr= rgpstg[i]->OpenStream(StmName, 0, STGM_WRITE, 0, &pstm);
|
|
hr= pstm->Write(ulData, Buff_Cnt * sizeof(ULONG), &cbWritten);
|
|
pstm->Release();
|
|
}
|
|
ReleaseObjPtr(rgpstg[i]);
|
|
}
|
|
|
|
printf("Everything worked!\n");
|
|
|
|
|
|
bail_out:
|
|
ReleaseObjPtr(pStorage);
|
|
ReleaseObjPtr(pITStorage);
|
|
|
|
ReleaseObjPtr(pICFITStorage);
|
|
|
|
shutdown:
|
|
|
|
CoUninitialize();
|
|
}
|