Backup / Restore cemail.vol

MOBILEVOLUME_API BOOL BackupMail() {
CEGUID ceguidCemail;
CEGUID ceguidBackUp;

CREATE_INVALIDGUID(&ceguidCemail);
CREATE_INVALIDGUID(&ceguidBackUp);


if(!CeMountDBVol(&ceguidCemail, L"cemail.vol",  OPEN_EXISTING)) {
DWORD dwErro = GetLastError();
return FALSE;
}


if(!CeMountDBVol( &ceguidBackUp, L"//cemail_backup.vol",  CREATE_ALWAYS)) {
return FALSE;
}

HANDLE hDatabaseEnumHandle = INVALID_HANDLE_VALUE;

if((hDatabaseEnumHandle = CeFindFirstDatabaseEx(&ceguidCemail, 0)) ==
INVALID_HANDLE_VALUE) {
return FALSE;
}

CEOID ceoidFind = 0;

while(ceoidFind = CeFindNextDatabaseEx(hDatabaseEnumHandle, NULL)) {
HANDLE hDBOpened = INVALID_HANDLE_VALUE;

hDBOpened = CeOpenDatabaseEx(&ceguidCemail,&ceoidFind, NULL, NULL,
CEDB_AUTOINCREMENT, NULL);
if(hDBOpened == INVALID_HANDLE_VALUE) {
DWORD error = GetLastError();
return FALSE;
}

BY_HANDLE_DB_INFORMATION hDBOpenedInfo;
hDBOpenedInfo.wVersion = BY_HANDLE_DB_INFORMATION_VERSION;
if(!CeGetDBInformationByHandle(hDBOpened, &hDBOpenedInfo)) {
DWORD dwError = GetLastError();
return FALSE;
}

CEOIDINFOEX cdDb;
CEOID ceoidFindRecorde = NULL;
CeOidGetInfoEx2(&ceguidCemail,ceoidFindRecorde, &cdDb);
CEOID ceoidDBaseCopy = 0;

if(!(ceoidDBaseCopy = CeCreateDatabaseEx2(&ceguidBackUp,
&hDBOpenedInfo.infDatabase))) {
DWORD dwErro = GetLastError();
return FALSE;
}

HANDLE hDBaseCopyHandle = INVALID_HANDLE_VALUE;
hDBaseCopyHandle =
CeOpenDatabaseEx(&ceguidBackUp,&ceoidDBaseCopy,hDBOpenedInfo.infDatabase.szDbaseName,0,CEDB_AUTOINCREMENT,NULL);
if(hDBaseCopyHandle == INVALID_HANDLE_VALUE) {
return FALSE;
}

hDBOpenedInfo.wVersion = BY_HANDLE_DB_INFORMATION_VERSION;
if(!CeGetDBInformationByHandle(hDBaseCopyHandle, &hDBOpenedInfo)) {
DWORD dwError = GetLastError();
return FALSE;
}

DWORD dwCpt = 0;
CEOID ceoidFindRecord = 0;
WORD dwPropId = 0;
CEPROPVAL* Buffer = NULL;
DWORD dwSizeOfBuffer = 0;
HANDLE hHeap = NULL;
hHeap = GetProcessHeap();
if(hHeap == NULL) return FALSE;
while(ceoidFindRecord =
CeReadRecordPropsEx(hDBOpened,CEDB_ALLOWREALLOC,&dwPropId,NULL,(LPBYTE*)&Buffer,&dwSizeOfBuffer,hHeap)) dwCpt++;
if(GetLastError() != ERROR_NO_MORE_ITEMS) return false;
free(Buffer);
Buffer = NULL;
CeSeekDatabaseEx(hDBOpened, CEDB_SEEK_BEGINNING, 0, 0, NULL);
ceoidFindRecord = 0;
DWORD dwNumberOfRecordFound = 0;
while(dwNumberOfRecordFound++ < dwCpt) {
ceoidFindRecord = CeReadRecordPropsEx(hDBOpened,
CEDB_ALLOWREALLOC,&dwPropId,NULL,(LPBYTE*)&Buffer, &dwSizeOfBuffer, hHeap);
if(!ceoidFindRecord) {
DWORD dwError = GetLastError();
return FALSE;
}
CEOID ceoidRecordAdd;
ceoidRecordAdd = CeWriteRecordProps(hDBaseCopyHandle,0,dwPropId,Buffer);
if(!ceoidRecordAdd) {
DWORD dwError = GetLastError();
return FALSE;
}
free(Buffer);
Buffer = NULL;
}
CloseHandle(hDBaseCopyHandle);
hDBaseCopyHandle = NULL;
}
if(!CeFlushDBVol(&ceguidCemail)) return FALSE;
if(!CeFlushDBVol(&ceguidBackUp)) return FALSE;
if(!CeUnmountDBVol(&ceguidCemail)) return FALSE;
if(!CeUnmountDBVol(&ceguidBackUp)) return FALSE;
return TRUE;
}

MOBILEVOLUME_API BOOL RestoreMail(LPWSTR fileLocation) {

CEGUID ceguidCemail;
CEGUID ceguidBackUp;

CREATE_INVALIDGUID(&ceguidCemail);
CREATE_INVALIDGUID(&ceguidBackUp);

if(!CeMountDBVol(&ceguidCemail, L"cemail.vol",  OPEN_EXISTING)) {
DWORD dwErro = GetLastError();
return FALSE;
}
if(!CeMountDBVol( &ceguidBackUp, fileLocation,  OPEN_EXISTING)) {
return FALSE;
}

HANDLE hDatabaseEnumHandle = INVALID_HANDLE_VALUE;

if((hDatabaseEnumHandle = CeFindFirstDatabaseEx(&ceguidBackUp, 0)) ==
INVALID_HANDLE_VALUE) {
return FALSE;
}
CEOID ceoidFind = 0;


/// Clear Old Values
HANDLE hRemoveHandle = INVALID_HANDLE_VALUE;
CEOID oldid = 0;
if((hRemoveHandle = CeFindFirstDatabaseEx(&ceguidCemail, 0)) ==
INVALID_HANDLE_VALUE) {
return FALSE;
}

while(oldid = CeFindNextDatabaseEx(hRemoveHandle, NULL)) {
HANDLE oldHandle = INVALID_HANDLE_VALUE;

oldHandle = CeOpenDatabaseEx(&ceguidCemail,&oldid, NULL, NULL,
CEDB_AUTOINCREMENT, NULL);
if(oldHandle == INVALID_HANDLE_VALUE) return FALSE;

BY_HANDLE_DB_INFORMATION hDBOldOpenedInfo;
hDBOldOpenedInfo.wVersion = BY_HANDLE_DB_INFORMATION_VERSION;
if(!CeGetDBInformationByHandle(oldHandle, &hDBOldOpenedInfo)) return FALSE;

DWORD dwOldCpt = 0;
CEOID ceoOldidFindRecord = 0;
WORD dwOldPropId = 0;
CEPROPVAL* oldBuffer = NULL;
DWORD dwOldSizeOfBuffer = 0;
HANDLE hOldHeap = NULL;
hOldHeap = GetProcessHeap();
if(hOldHeap == NULL) return FALSE;
while(ceoOldidFindRecord =
CeReadRecordPropsEx(oldHandle,CEDB_ALLOWREALLOC,&dwOldPropId,NULL,(LPBYTE*)&oldBuffer,&dwOldSizeOfBuffer,hOldHeap)) dwOldCpt++;
if(GetLastError() != ERROR_NO_MORE_ITEMS) return false;
free(oldBuffer);
oldBuffer = NULL;
CeSeekDatabaseEx(oldHandle, CEDB_SEEK_BEGINNING, 0, 0, NULL);
ceoOldidFindRecord = 0;
DWORD dwOldNumberOfRecordFound = 0;
while(dwOldNumberOfRecordFound++ < dwOldCpt) {
ceoOldidFindRecord = CeReadRecordPropsEx(oldHandle,
CEDB_ALLOWREALLOC,&dwOldPropId,NULL,(LPBYTE*)&oldBuffer, &dwOldSizeOfBuffer,
hOldHeap);
CeDeleteRecord(oldHandle,ceoOldidFindRecord);
}
CloseHandle(oldHandle);
oldHandle = NULL;
}
CloseHandle(hRemoveHandle);
hRemoveHandle = NULL;

while(ceoidFind = CeFindNextDatabaseEx(hDatabaseEnumHandle, NULL)) {

HANDLE hDBOpened = INVALID_HANDLE_VALUE;

hDBOpened = CeOpenDatabaseEx(&ceguidBackUp,&ceoidFind, NULL, NULL,
CEDB_AUTOINCREMENT, NULL);
if(hDBOpened == INVALID_HANDLE_VALUE) {
DWORD error = GetLastError();
return FALSE;
}
BY_HANDLE_DB_INFORMATION hDBOpenedInfo;
hDBOpenedInfo.wVersion = BY_HANDLE_DB_INFORMATION_VERSION;
if(!CeGetDBInformationByHandle(hDBOpened, &hDBOpenedInfo)) {
return FALSE;
}

CEOID ceoidDBaseOriginal = 0;
HANDLE hDBaseOriginalHandle = INVALID_HANDLE_VALUE;
hDBaseOriginalHandle = CeOpenDatabaseEx2(&ceguidCemail,
&ceoidDBaseOriginal, hDBOpenedInfo.infDatabase.szDbaseName, NULL ,
CEDB_AUTOINCREMENT, NULL);
if(hDBaseOriginalHandle == INVALID_HANDLE_VALUE) {
return FALSE;
}

WORD dwPropId = 0;
CEPROPVAL* Buffer = NULL;
DWORD dwSizeOfBuffer = 0;
HANDLE hHeap = NULL;
hHeap = GetProcessHeap();
if(hHeap == NULL) return FALSE;
BOOL bDBIsEmpty = FALSE;
CEOID ceoidFindRecord = 0;
while(!bDBIsEmpty) {

ceoidFindRecord = CeReadRecordPropsEx(hDBOpened, CEDB_ALLOWREALLOC,
&dwPropId, NULL, (LPBYTE*)&Buffer, &dwSizeOfBuffer, hHeap);
if(!ceoidFindRecord) {
if(GetLastError() != ERROR_NO_MORE_ITEMS) return FALSE;
else bDBIsEmpty = TRUE;
} else {
CEPROPVAL* temp;
temp = (CEPROPVAL*) malloc(dwSizeOfBuffer);
memcpy(temp, Buffer, dwSizeOfBuffer);
CEOID ceoidRecordAdd = 0;

ceoidRecordAdd =
CeWriteRecordProps(hDBaseOriginalHandle,temp[2].val.ulVal, temp[1].val.uiVal,
(CEPROPVAL *)(temp[0].val.blob.lpb));
if(!ceoidRecordAdd) return FALSE;
free(temp);
temp = NULL;
free(Buffer);
Buffer = NULL;
}
}

}
if(!CeFlushDBVol(&ceguidCemail)) return FALSE;
if(!CeFlushDBVol(&ceguidBackUp)) return FALSE;
if(!CeUnmountDBVol(&ceguidCemail)) return FALSE;
if(!CeUnmountDBVol(&ceguidBackUp)) return FALSE;
return TRUE;
}