BOOL VSSClientGeneric::CreateSnapshots(char* szDriveLetters)
{
/* szDriveLetters contains all drive letters in uppercase */
/* if a drive can not being added, it's converted to lowercase in szDriveLetters */
/* http://msdn.microsoft.com/library/default.asp?url=/library/en-us/vss/base/ivssbackupcomponents_startsnapshotset.asp */
if (!m_pVssObject || m_bBackupIsInitialized) {
errno = ENOSYS;
return FALSE;
}
m_uidCurrentSnapshotSet = GUID_NULL;
IVssBackupComponents *pVss = (IVssBackupComponents*)m_pVssObject;
/* startSnapshotSet */
pVss->StartSnapshotSet(&m_uidCurrentSnapshotSet);
/* AddToSnapshotSet */
wchar_t szDrive[3];
szDrive[1] = ':';
szDrive[2] = 0;
wstring volume;
CComPtr<IVssAsync> pAsync1;
CComPtr<IVssAsync> pAsync2;
VSS_ID pid;
for (size_t i=0; i < strlen (szDriveLetters); i++) {
szDrive[0] = szDriveLetters[i];
volume = GetUniqueVolumeNameForPath(szDrive);
// store uniquevolumname
if (SUCCEEDED(pVss->AddToSnapshotSet((LPWSTR)volume.c_str(), GUID_NULL, &pid))) {
wcsncpy (m_wszUniqueVolumeName[szDriveLetters[i]-'A'], (LPWSTR) volume.c_str(), MAX_PATH);
} else {
szDriveLetters[i] = tolower (szDriveLetters[i]);
}
}
/* PrepareForBackup */
if (FAILED(pVss->PrepareForBackup(&pAsync1.p))) {
errno = b_errno_win32;
return FALSE;
}
// Waits for the async operation to finish and checks the result
WaitAndCheckForAsyncOperation(pAsync1.p);
/* get latest info about writer status */
if (!CheckWriterStatus()) {
errno = b_errno_win32;
return FALSE;
}
/* DoSnapShotSet */
if (FAILED(pVss->DoSnapshotSet(&pAsync2.p))) {
errno = b_errno_win32;
return FALSE;
}
// Waits for the async operation to finish and checks the result
WaitAndCheckForAsyncOperation(pAsync2.p);
/* query snapshot info */
QuerySnapshotSet(m_uidCurrentSnapshotSet);
SetVSSPathConvert(VSSPathConvert, VSSPathConvertW);
m_bBackupIsInitialized = true;
return TRUE;
}
