361 lines
9.3 KiB
C++
361 lines
9.3 KiB
C++
/*++
|
|
|
|
Copyright (c) 2008 Microsoft Corporation
|
|
|
|
Title:
|
|
|
|
Express Writer sample application
|
|
|
|
Abstract:
|
|
|
|
Sample Express Writer application main code file
|
|
|
|
--*/
|
|
|
|
|
|
#include "stdafx.h"
|
|
|
|
|
|
//
|
|
// This is sample Express Writer GUID
|
|
//
|
|
// Do not reuse it: your GUID should be unique and consistent
|
|
// between writer releases
|
|
//
|
|
// In order to create your own GUID run GUID creation tool
|
|
// that comes with Visual Studio
|
|
//
|
|
// Please remember that your writer should be exposed with
|
|
// the same GUID on every system it is installed on
|
|
//
|
|
// If you want to version your Writer, please use the version
|
|
// parameter during writer creation and/or registration
|
|
//
|
|
GUID EXPRESS_WRITER_SAMPLE_GUID =
|
|
{
|
|
0x12345678,
|
|
0x1234,
|
|
0x1234,
|
|
{0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88}
|
|
};
|
|
|
|
|
|
//
|
|
// This is the name of the file where Express Writer metadata
|
|
// will be saved by one of the sample scenarios
|
|
//
|
|
// Please do not modify its content by hand
|
|
//
|
|
const PCWSTR g_wszFileName = L"ExpressWriter.xml";
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CreateAndRegister
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
bool CreateAndRegister()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
bool bResult = false;
|
|
|
|
CComPtr<IVssExpressWriter> spExpressWriter;
|
|
CComPtr<IVssCreateExpressWriterMetadata> spMetadata;
|
|
|
|
wprintf(L"INFO: CreateAndRegister called\n");
|
|
|
|
//
|
|
// Create Express Writer
|
|
//
|
|
wprintf(L"INFO: Creating Express Writer...\n");
|
|
CHECK_HR(
|
|
CreateVssExpressWriter(&spExpressWriter),
|
|
L"CreateVssExpressWriter failed");
|
|
|
|
//
|
|
// Create metadata
|
|
//
|
|
wprintf(L"INFO: Creating metadata...\n");
|
|
CHECK_HR(
|
|
spExpressWriter->CreateMetadata(
|
|
EXPRESS_WRITER_SAMPLE_GUID, // Express Writer GUID
|
|
L"Sample Express Writer", // Express Writer friendly name
|
|
VSS_UT_BOOTABLESYSTEMSTATE, // VSS_USAGE_TYPE
|
|
1, // Major version
|
|
0, // Minor version
|
|
0, // Reserved (must be 0)
|
|
&spMetadata),
|
|
L"CreateVssExpressWriter failed");
|
|
|
|
CHECK_CONDITION(
|
|
ConstructWriterDefinition(spMetadata),
|
|
L"ConstructWriterDefinition failed");
|
|
|
|
bResult = true;
|
|
|
|
//
|
|
// Register writer in the system
|
|
//
|
|
CHECK_HR(spExpressWriter->Register(), L"Register failed");
|
|
|
|
bResult = true;
|
|
|
|
_exit:
|
|
return bResult;
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CreateAndSaveToFile
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
bool CreateAndSaveToFile()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
bool bResult = false;
|
|
|
|
CComPtr<IVssExpressWriter> spExpressWriter;
|
|
CComPtr<IVssCreateExpressWriterMetadata> spMetadata;
|
|
|
|
wprintf(L"INFO: CreateAndSaveToFile called\n");
|
|
|
|
//
|
|
// Create Express Writer
|
|
//
|
|
wprintf(L"INFO: Creating Express Writer...\n");
|
|
CHECK_HR(
|
|
CreateVssExpressWriter(&spExpressWriter),
|
|
L"CreateVssExpressWriter failed");
|
|
|
|
//
|
|
// Create metadata
|
|
//
|
|
wprintf(L"INFO: Creating metadata...\n");
|
|
CHECK_HR(
|
|
spExpressWriter->CreateMetadata(
|
|
EXPRESS_WRITER_SAMPLE_GUID, // Express Writer GUID
|
|
L"Sample Express Writer", // Express Writer friendly name
|
|
VSS_UT_BOOTABLESYSTEMSTATE, // VSS_USAGE_TYPE
|
|
1, // Major version
|
|
0, // Minor version
|
|
0, // Reserved (must be 0)
|
|
&spMetadata),
|
|
L"CreateVssExpressWriter failed");
|
|
|
|
CHECK_CONDITION(
|
|
ConstructWriterDefinition(spMetadata),
|
|
L"ConstructWriterDefinition failed");
|
|
|
|
bResult = true;
|
|
|
|
//
|
|
// Save results
|
|
//
|
|
CHECK_CONDITION(SaveToFile(spMetadata, g_wszFileName), L"SaveToFile failed");
|
|
|
|
bResult = true;
|
|
|
|
_exit:
|
|
return bResult;
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// LoadFromFileAndRegister
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
bool LoadFromFileAndRegister()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
bool bResult = false;
|
|
PWSTR wszData = NULL;
|
|
|
|
CComPtr<IVssExpressWriter> spExpressWriter;
|
|
|
|
wprintf(L"INFO: LoadFromFileAndRegister called\n");
|
|
|
|
//
|
|
// Create Express Writer
|
|
//
|
|
wprintf(L"INFO: Creating Express Writer...\n");
|
|
CHECK_HR(
|
|
CreateVssExpressWriter(&spExpressWriter),
|
|
L"CreateVssExpressWriter failed");
|
|
|
|
//
|
|
// Load metadata
|
|
//
|
|
wprintf(L"INFO: Loading metadata...\n");
|
|
|
|
CHECK_CONDITION(
|
|
LoadMetadata(spExpressWriter, g_wszFileName, &wszData),
|
|
L"LoadMetadata failed");
|
|
|
|
//
|
|
// Save results
|
|
//
|
|
CHECK_HR(spExpressWriter->Register(), L"Register failed");
|
|
|
|
bResult = true;
|
|
|
|
_exit:
|
|
if (wszData)
|
|
{
|
|
free(wszData);
|
|
wszData = NULL;
|
|
}
|
|
return bResult;
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// UnregisterWriter
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
bool UnregisterWriter()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
bool bResult = false;
|
|
|
|
CComPtr<IVssExpressWriter> spExpressWriter;
|
|
CComPtr<IVssCreateExpressWriterMetadata> spMetadata;
|
|
|
|
wprintf(L"INFO: Unregister called\n");
|
|
|
|
//
|
|
// Create Express Writer
|
|
//
|
|
wprintf(L"INFO: Creating Express Writer...\n");
|
|
CHECK_HR(
|
|
CreateVssExpressWriter(&spExpressWriter),
|
|
L"CreateVssExpressWriter failed");
|
|
|
|
//
|
|
// Create metadata
|
|
//
|
|
wprintf(L"INFO: Unregistering Express Writer...\n");
|
|
CHECK_HR(
|
|
spExpressWriter->Unregister(
|
|
EXPRESS_WRITER_SAMPLE_GUID), // Express Writer GUID
|
|
L"Unregister failed");
|
|
|
|
bResult = true;
|
|
|
|
_exit:
|
|
return bResult;
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// RunSamples
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
bool RunSamples()
|
|
{
|
|
HRESULT hr = S_OK;
|
|
bool bResult = false;
|
|
|
|
wprintf(L"INFO: RunSamples called\n");
|
|
|
|
//
|
|
// Create new Express Writer and register it in the system
|
|
//
|
|
// This scenario allows you to perform full Express Writer
|
|
// creation and registration on the target machine
|
|
//
|
|
// This code can be used by your application to configure its
|
|
// writer during installation or on the first execution
|
|
// assuming that your storage varies between client machines
|
|
//
|
|
|
|
bResult = CreateAndRegister(); // Execute on target machine to
|
|
// create and register Writer
|
|
|
|
CHECK_CONDITION(bResult, L"CreateAndRegister failed");
|
|
|
|
bResult = UnregisterWriter(); // Cleanup for the next sample
|
|
|
|
CHECK_CONDITION(bResult, L"UnregisterWriter failed");
|
|
|
|
//
|
|
// Create new Express Writer and save it to a file
|
|
// Then load it from file and register in the system
|
|
//
|
|
// This scenario allows you to perform full Express Writer
|
|
// creation offline (during development) and later perform
|
|
// registration on the target machine by reading the file
|
|
//
|
|
// This code can be used by your application to configure its
|
|
// writer during installation or on the first execution
|
|
// assuming that your storage location is the same on each
|
|
// machine where your application is being deployed
|
|
//
|
|
|
|
bResult = CreateAndSaveToFile(); // Execute during development to
|
|
// create Express Writer metadata file
|
|
|
|
CHECK_CONDITION(bResult, L"CreateAndSaveToFile failed");
|
|
|
|
bResult = LoadFromFileAndRegister(); // Execute on the target machine to
|
|
// load Express Writer metadata from
|
|
// file and register Express Writer in
|
|
// the system
|
|
|
|
CHECK_CONDITION(bResult, L"LoadFromFileAndRegister failed");
|
|
|
|
bResult = UnregisterWriter(); // Cleanup after last sample
|
|
|
|
CHECK_CONDITION(bResult, L"UnregisterWriter failed");
|
|
|
|
bResult = true;
|
|
|
|
_exit:
|
|
return bResult;
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// wmain
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
int __cdecl wmain(
|
|
__in int argc,
|
|
__in_ecount(argc) WCHAR *argv[]
|
|
)
|
|
{
|
|
bool bCoInitializeCalled = false;
|
|
int iResult = -1;
|
|
int iTestIndex = -1;
|
|
HRESULT hr = S_OK;
|
|
|
|
UNREFERENCED_PARAMETER(argc);
|
|
UNREFERENCED_PARAMETER(argv);
|
|
|
|
hr = ::CoInitialize(NULL);
|
|
CHECK_HR(hr, L"CoInitialize failed");
|
|
bCoInitializeCalled = true;
|
|
|
|
//
|
|
// Run sample Express Writer code
|
|
//
|
|
CHECK_CONDITION(RunSamples(), L"RunSamples failed");
|
|
|
|
iResult = 0;
|
|
|
|
_exit:
|
|
if (bCoInitializeCalled)
|
|
{
|
|
::CoUninitialize();
|
|
}
|
|
|
|
wprintf(L"INFO: Application returning %d\n", iResult);
|
|
return iResult;
|
|
}
|
|
|
|
|