|
|
/*++
Copyright (c) 1999 Microsoft Corporation
Module Name:
opaque.c
Abstract:
Implements a basic secure server transport module
Author:
Jim Schmidt (jimschm) 08-Mar-2000
Revision History:
<alias> <date> <comments>
--*/
//
// Includes
//
#include "pch.h"
#include "logmsg.h"
#include <compress.h>
#define DBG_OPAQUE "OpaqueUnc"
//
// Strings
//
#define S_TRANSPORT_DIR TEXT("USMT2.UNC")
#define S_TRANSPORT_DAT_FILE TEXT("TRANSDB.DAT")
#define S_TRANSPORT_IMG_FILE TEXT("IMG%05X.DAT")
#define S_TRANSPORT_STATUS_FILE TEXT("status")
#define S_FILEOBJECT_NAME TEXT("File")
#define S_REGOBJECT_NAME TEXT("Registry")
#define S_DATABASEOBJECT_NAME TEXT("Database")
#define S_DETAILS_PREFIX TEXT("details-")
//
// Constants
//
#define TRFLAG_FILE 0x01
#define TRFLAG_MEMORY 0x02
#define COPY_BUFFER_SIZE 32768
#define OPAQUETR_OLDSIG1 0x55534D31 //USM1
#define OPAQUETR_OLDSIG2 0x55534D32 //USM2
#define OPAQUETR_OLDSIG3 0x55534D33 //USM3
#define OPAQUETR_CONVSIG 0x55534D33 //USM3
#define OPAQUETR_SIG 0x55534D34 //USM4
#define TRSTATUS_DIRTY 0x00000001
#define TRSTATUS_READY 0x00000002
#define TRSTATUS_LOCKED 0x00000003
//
// Macros
//
// None
//
// Types
//
typedef struct { TCHAR TempFile [MAX_PATH]; PCVOID AllocPtr; PCVOID DetailsPtr; HANDLE FileHandle; HANDLE MapHandle; } ALLOCSTATE, *PALLOCSTATE;
//
// Globals
//
MIG_TRANSPORTSTORAGEID g_ReliableStorageId; PCTSTR g_TransportPath = NULL; PCTSTR g_TransportTempPath = NULL; PCTSTR g_TransportStatus = NULL; HANDLE g_TransportStatusHandle = NULL; BOOL g_OtCompressData = FALSE; BOOL g_OtOldFormat = FALSE; UINT g_Platform; MIG_PROGRESSSLICEID g_PersistentSlice; MIG_PROGRESSSLICEID g_DatabaseSlice; UINT g_CompressedTicks; UINT g_CompressedTicked; MIG_PROGRESSSLICEID g_CompressedSlice; UINT g_UncompressTicks; UINT g_UncompressTicked; MIG_PROGRESSSLICEID g_UncompressSlice; LONGLONG g_TotalFiles; LONGLONG g_FilesRead;
//
// Macro expansion list
//
// None
//
// Private function prototypes
//
BOOL pOtSaveAllState ( IN BOOL Compressed );
//
// Macro expansion definition
//
// None
//
// Code
//
BOOL pSetOpaqueTransportStatus ( IN HANDLE TrJournalHandle, IN BOOL Compressed, IN DWORD Status ) { DWORD signature = OPAQUETR_SIG; BOOL result = FALSE;
if (BfSetFilePointer (TrJournalHandle, 0)) { result = TRUE; result = result && BfWriteFile (TrJournalHandle, (PBYTE)(&signature), sizeof (DWORD)); result = result && BfWriteFile (TrJournalHandle, (PBYTE)(&Compressed), sizeof (BOOL)); result = result && BfWriteFile (TrJournalHandle, (PBYTE)(&Status), sizeof (DWORD)); result = result && FlushFileBuffers (TrJournalHandle); } return result; }
DWORD pGetOpaqueTransportStatus ( IN PCTSTR TrJournal, OUT PBOOL Compressed, OPTIONAL OUT PBOOL OldStorage, OPTIONAL IN BOOL OpenReadOnly ) { HANDLE trJrnHandle; BOOL compressed = FALSE; DWORD signature = 0; DWORD fileSize = 0; DWORD error; DWORD result = 0;
if (OldStorage) { *OldStorage = FALSE; }
if (TrJournal && TrJournal [0]) { if (OpenReadOnly) { trJrnHandle = BfOpenReadFile (TrJournal); } else { trJrnHandle = BfOpenFile (TrJournal); } if (trJrnHandle) { if (BfSetFilePointer (trJrnHandle, 0)) { fileSize = GetFileSize (trJrnHandle, NULL); if (fileSize == (sizeof (BOOL) + sizeof (DWORD))) { if (OldStorage) { *OldStorage = TRUE; } if (BfReadFile (trJrnHandle, (PBYTE)(&compressed), sizeof (BOOL))) { BfReadFile (trJrnHandle, (PBYTE)(&result), sizeof (DWORD)); } } if (fileSize == (sizeof (BOOL) + sizeof (DWORD) + sizeof (DWORD))) { if (BfReadFile (trJrnHandle, (PBYTE)(&signature), sizeof (DWORD))) { if (signature == OPAQUETR_SIG) { if (BfReadFile (trJrnHandle, (PBYTE)(&compressed), sizeof (BOOL))) { BfReadFile (trJrnHandle, (PBYTE)(&result), sizeof (DWORD)); } } if ((signature == OPAQUETR_OLDSIG1) || (signature == OPAQUETR_OLDSIG2) || (signature == OPAQUETR_OLDSIG3) ) { if (OldStorage) { *OldStorage = TRUE; } if (BfReadFile (trJrnHandle, (PBYTE)(&compressed), sizeof (BOOL))) { BfReadFile (trJrnHandle, (PBYTE)(&result), sizeof (DWORD)); } } } } } CloseHandle (trJrnHandle); } else { error = GetLastError (); if ((error == ERROR_ACCESS_DENIED) || (error == ERROR_SHARING_VIOLATION) ) { result = TRSTATUS_LOCKED; } } } if (Compressed) { *Compressed = compressed; } return result; }
PCTSTR pGetOpaqueImageFile ( IN PCTSTR BasePath, IN UINT ImageIdx ) { TCHAR imageFileName [13]; PCTSTR imageFile = NULL; HANDLE imageFileHandle = NULL;
wsprintf (imageFileName, S_TRANSPORT_IMG_FILE, ImageIdx); return JoinPaths (BasePath, imageFileName); }
PCTSTR pGetRealTransportPath ( VOID ) { return g_OtCompressData?g_TransportTempPath:g_TransportPath; }
BOOL WINAPI OpaqueTransportInitialize ( IN PMIG_LOGCALLBACK LogCallback ) { //
// Initialize globals
//
LogReInit (NULL, NULL, NULL, (PLOGCALLBACK) LogCallback); g_ReliableStorageId = IsmRegisterTransport (S_RELIABLE_STORAGE_TRANSPORT);
return TRUE; }
VOID WINAPI OpaqueTransportEstimateProgressBar ( MIG_PLATFORMTYPEID PlatformTypeId ) { UINT ticks; PMIG_OBJECTCOUNT objectCount;
if (PlatformTypeId == PLATFORM_SOURCE) {
//
// If saving, we know the number of ticks based on the count of the
// persistent attribute.
//
objectCount = IsmGetObjectsStatistics (PLATFORM_SOURCE);
if (objectCount) { ticks = objectCount->PersistentObjects; } else { ticks = 0; }
g_PersistentSlice = IsmRegisterProgressSlice (ticks, max (1, ticks / 5));
if (g_OtCompressData) { g_DatabaseSlice = IsmRegisterProgressSlice (3, 1); } else { g_DatabaseSlice = IsmRegisterProgressSlice (1, 1); }
} else { if (g_OtCompressData) { g_UncompressTicked = 0; g_UncompressTicks = 1000; g_UncompressSlice = IsmRegisterProgressSlice (g_UncompressTicks, 180); } else { //
// If restoring, we have almost no work to account for, since
// we download from the secure server file-by-file.
//
DEBUGMSG ((DBG_VERBOSE, "Assuming transport download has no progress impact")); } } }
BOOL WINAPI OpaqueTransportQueryCapabilities ( IN MIG_TRANSPORTSTORAGEID TransportStorageId, OUT PMIG_TRANSPORTTYPE TransportType, OUT PMIG_TRANSPORTCAPABILITIES Capabilities, OUT PCTSTR *FriendlyDescription ) { if (TransportStorageId != g_ReliableStorageId) { return FALSE; }
*TransportType = TRANSPORTTYPE_FULL; *Capabilities = CAPABILITY_COMPRESSED; *FriendlyDescription = TEXT("Local computer or another computer on the Network"); return TRUE; }
VOID pOtCleanUpTempDir ( VOID ) { if (g_TransportTempPath) { FiRemoveAllFilesInTree (g_TransportTempPath); } }
PCTSTR pOtCreateTemporaryDir ( VOID ) { TCHAR tempFile[MAX_PATH];
if (!IsmGetTempDirectory (tempFile, ARRAYSIZE(tempFile))) { return NULL; } return DuplicatePathString (tempFile, 0); }
BOOL WINAPI OpaqueTransportSetStorage ( IN MIG_PLATFORMTYPEID Platform, IN MIG_TRANSPORTSTORAGEID TransportStorageId, IN MIG_TRANSPORTCAPABILITIES RequiredCapabilities, IN PCTSTR StoragePath, OUT PBOOL Valid, OUT PBOOL ImageExists ) { PCTSTR lastDirPtr = NULL; PCTSTR transportPath = NULL; PCTSTR transportStatus = NULL; HANDLE transportStatusHandle = INVALID_HANDLE_VALUE; MIG_OBJECTSTRINGHANDLE encodedPath; DWORD status; BOOL result = FALSE;
if (Valid) { *Valid = FALSE; } if (ImageExists) { *ImageExists = FALSE; }
if (TransportStorageId == g_ReliableStorageId) {
if ((!RequiredCapabilities) || (RequiredCapabilities == CAPABILITY_COMPRESSED)) {
if (RequiredCapabilities == CAPABILITY_COMPRESSED) { g_OtCompressData = TRUE; } else { g_OtCompressData = FALSE; }
// If this is the destination, let's check if the user included USMT2.UNC in
// the path by mistake. If he did (the StoragePath ends in USMT2.UNC and
// StoragePath has our transport status) we will accept it.
if (Platform == PLATFORM_DESTINATION) { lastDirPtr = _tcsrchr (StoragePath, TEXT('\\')); if (lastDirPtr) { lastDirPtr = _tcsinc (lastDirPtr); if (lastDirPtr) { if (StringIMatch (lastDirPtr, S_TRANSPORT_DIR)) { transportPath = DuplicatePathString (StoragePath, 0); transportStatus = JoinPaths (transportPath, S_TRANSPORT_STATUS_FILE); if ((!DoesFileExist (transportPath)) || (pGetOpaqueTransportStatus (transportStatus, NULL, NULL, TRUE) != TRSTATUS_READY) ) { FreePathString (transportPath); transportPath = NULL; FreePathString (transportStatus); transportStatus = NULL; } } } } }
if (!transportPath) { transportPath = JoinPaths (StoragePath, S_TRANSPORT_DIR); } if (!transportStatus) { transportStatus = JoinPaths (transportPath, S_TRANSPORT_STATUS_FILE); }
if (!DoesFileExist (transportPath)) {
// we require UNC path or a full path (like c:\...)
if (transportPath[0] == '\\' && transportPath[1] == '\\') { // this is a UNC path
*Valid = TRUE; } else if (transportPath[1] == ':') { // this is a normal full path
*Valid = TRUE; } else { *Valid = FALSE; }
if ((*Valid) && (Platform == PLATFORM_SOURCE)) { // we need to actually try to create our transport directory and the status file
// After that we will remove them but this is just the way for us to validate the
// store
*Valid = FALSE; if (BfCreateDirectory (transportPath)) { transportStatusHandle = BfCreateFile (transportStatus); if (transportStatusHandle) { *Valid = TRUE; CloseHandle (transportStatusHandle); } PushError (); FiRemoveAllFilesInTree (transportPath); PopError (); } }
*ImageExists = FALSE;
} else {
// we try to open this transport status file with Read/Write. If we
// fail this is either because somebody has it opened or we don't
// have access rights. In both cases we should get back TRSTATUS_LOCKED
status = pGetOpaqueTransportStatus (transportStatus, NULL, NULL, TRUE);
switch (status) { case TRSTATUS_LOCKED: *ImageExists = TRUE; *Valid = FALSE; break; case TRSTATUS_READY: *ImageExists = TRUE; *Valid = TRUE; break; case TRSTATUS_DIRTY: *ImageExists = FALSE; *Valid = TRUE; break; default: *ImageExists = FALSE; *Valid = TRUE; } }
result = TRUE; } }
if (result && *Valid) {
if (g_TransportPath) { FreePathString (g_TransportPath); g_TransportPath = NULL; }
if (g_TransportStatus) { FreePathString (g_TransportStatus); g_TransportStatus = NULL; }
if (transportPath) { g_TransportPath = transportPath; } else { g_TransportPath = JoinPaths (StoragePath, S_TRANSPORT_DIR); } if (transportStatus) { g_TransportStatus = transportStatus; } else { g_TransportStatus = JoinPaths (g_TransportPath, S_TRANSPORT_STATUS_FILE); }
encodedPath = IsmCreateSimpleObjectPattern (g_TransportPath, FALSE, NULL, FALSE); if (encodedPath) { IsmRegisterStaticExclusion (MIG_FILE_TYPE, encodedPath); IsmDestroyObjectHandle (encodedPath); } }
return result; }
BOOL WINAPI OpaqueTransportSaveState ( VOID ) { MIG_APPINFO appInfo; ERRUSER_EXTRADATA extraData; DWORD status; BOOL result = FALSE; BOOL allowDelete = FALSE;
ZeroMemory (&extraData, sizeof (ERRUSER_EXTRADATA)); extraData.Error = ERRUSER_ERROR_UNKNOWN;
__try {
ZeroMemory (&appInfo, sizeof (MIG_APPINFO)); appInfo.Phase = MIG_TRANSPORT_PHASE; appInfo.SubPhase = SUBPHASE_PREPARING; IsmSendMessageToApp (ISMMESSAGE_APP_INFO, (ULONG_PTR)(&appInfo));
if (!g_TransportPath) { DEBUGMSG ((DBG_ERROR, "Transport Path is not selected")); extraData.Error = ERRUSER_ERROR_NOTRANSPORTPATH; __leave; }
if (DoesFileExist (g_TransportPath)) {
status = pGetOpaqueTransportStatus (g_TransportStatus, NULL, NULL, FALSE);
switch (status) { case TRSTATUS_LOCKED: SetLastError (ERROR_ACCESS_DENIED); LOG ((LOG_ERROR, (PCSTR) MSG_TRANSPORT_DIR_BUSY, g_TransportPath)); extraData.Error = ERRUSER_ERROR_TRANSPORTPATHBUSY; __leave; case TRSTATUS_DIRTY: result = FiRemoveAllFilesInTree (g_TransportPath); if (!result) { LOG ((LOG_ERROR, (PCSTR) MSG_CANT_EMPTY_DIR, g_TransportPath)); extraData.Error = ERRUSER_ERROR_CANTEMPTYDIR; __leave; } break; case TRSTATUS_READY: default: if (IsmSendMessageToApp (TRANSPORTMESSAGE_IMAGE_EXISTS, 0)) { if (!FiRemoveAllFilesInTree (g_TransportPath)) { LOG ((LOG_ERROR, (PCSTR) MSG_CANT_EMPTY_DIR, g_TransportPath)); extraData.Error = ERRUSER_ERROR_CANTEMPTYDIR; __leave; } } else { LOG ((LOG_ERROR, (PCSTR) MSG_NOT_EMPTY, g_TransportPath)); SetLastError (ERROR_ALREADY_EXISTS); extraData.Error = ERRUSER_ERROR_ALREADYEXISTS; __leave; } break; } }
allowDelete = TRUE;
if (!BfCreateDirectory (g_TransportPath)) { PushError (); LOG ((LOG_ERROR, (PCSTR) MSG_CANT_CREATE_DIR, g_TransportPath)); PopError (); extraData.Error = ERRUSER_ERROR_CANTCREATEDIR; __leave; }
g_TransportStatusHandle = BfCreateFile (g_TransportStatus); if (!g_TransportStatusHandle) { PushError (); LOG ((LOG_ERROR, (PCSTR) MSG_CANT_CREATE_STATUS_FILE, g_TransportStatus)); PopError (); extraData.Error = ERRUSER_ERROR_CANTCREATESTATUS; __leave; }
result = pSetOpaqueTransportStatus (g_TransportStatusHandle, g_OtCompressData, TRSTATUS_DIRTY); if (!result) { PushError (); LOG ((LOG_ERROR, (PCSTR) MSG_CANT_CREATE_STATUS_FILE, g_TransportStatus)); PopError (); extraData.Error = ERRUSER_ERROR_CANTCREATESTATUS; __leave; }
g_Platform = PLATFORM_SOURCE; result = pOtSaveAllState (g_OtCompressData);
if (result) { result = result && pSetOpaqueTransportStatus (g_TransportStatusHandle, g_OtCompressData, TRSTATUS_READY); }
result = result && FlushFileBuffers (g_TransportStatusHandle); result = result && CloseHandle (g_TransportStatusHandle); if (result) { g_TransportStatusHandle = NULL; } } __finally { PushError (); if (g_TransportStatusHandle) { // if we got here there was some error. Let's just close the handle
CloseHandle (g_TransportStatusHandle); } if (allowDelete && (!result)) { FiRemoveAllFilesInTree (g_TransportPath); } PopError (); }
PushError ();
if (!result) { extraData.ErrorArea = ERRUSER_AREA_SAVE; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); }
PopError ();
return result; }
PCTSTR pOpaqueGetNewFileName ( IN PCTSTR FileName ) { PCTSTR newFileName = NULL; PTSTR tempPtr1 = NULL; PCTSTR endStr1 = NULL; PCTSTR tempPtr2 = NULL; PCTSTR endStr2 = NULL; INT i;
// let's modify the file to extract. The file name will
// be split in 2 after the first 5 characters
newFileName = DuplicatePathString (FileName, 1); if (!newFileName) { return NULL; } tempPtr1 = (PTSTR) GetFileNameFromPath (newFileName); if (!tempPtr1) { FreePathString (newFileName); return NULL; } endStr1 = GetEndOfString (newFileName); if (!endStr1) { FreePathString (newFileName); return NULL; } tempPtr2 = GetFileNameFromPath (FileName); if (!tempPtr2) { FreePathString (newFileName); return NULL; } endStr2 = GetEndOfString (FileName); if (!endStr2) { FreePathString (newFileName); return NULL; } for (i = 0; i < 5; i ++) { tempPtr1 = _tcsinc (tempPtr1); tempPtr2 = _tcsinc (tempPtr2); }
if ((tempPtr1 < endStr1) && (tempPtr2 < endStr2) ) { StringCopy (tempPtr1, TEXT("\\")); tempPtr1 = _tcsinc (tempPtr1); StringCopy (tempPtr1, tempPtr2); } else { FreePathString (newFileName); newFileName = NULL; } return newFileName; }
BOOL pOpaqueCallback ( IN PCTSTR FileToExtract, IN LONGLONG FileSize, OUT PBOOL ExtractFile, IN OUT PCTSTR *NewFileName ) { LONGLONG numerator; LONGLONG divisor; LONGLONG tick; UINT delta;
if (NewFileName) { *NewFileName = pOpaqueGetNewFileName (FileToExtract); }
g_FilesRead ++; // now update the progress bar
numerator = (LONGLONG) g_FilesRead * (LONGLONG) g_UncompressTicks; divisor = (LONGLONG) g_TotalFiles; if (divisor) { tick = numerator / divisor; } else { tick = 0; } delta = (UINT) tick - g_UncompressTicked; if (delta) { IsmTickProgressBar (g_UncompressSlice, delta); g_UncompressTicked += delta; }
if (ExtractFile) { *ExtractFile = TRUE; }
return (!IsmCheckCancel()); }
BOOL pOtReadAllImages ( VOID ) { COMPRESS_HANDLE compressedHandle; BOOL result = FALSE;
if (CompressOpenHandle (g_TransportPath, S_TRANSPORT_IMG_FILE, 1, &compressedHandle)) { g_TotalFiles = compressedHandle.FilesStored; if (CompressExtractAllFiles (g_TransportTempPath, &compressedHandle, pOpaqueCallback)) { result = TRUE; } CompressCleanupHandle (&compressedHandle); }
return result; }
BOOL WINAPI OpaqueTransportBeginApply ( VOID ) { ERRUSER_EXTRADATA extraData; DWORD status = 0; PCTSTR memDbFile; INT_PTR appReply; PCTSTR newTransDbFile = NULL; BOOL result = FALSE;
g_Platform = PLATFORM_DESTINATION;
ZeroMemory (&extraData, sizeof (ERRUSER_EXTRADATA)); extraData.Error = ERRUSER_ERROR_UNKNOWN;
__try { if (!g_TransportPath) { DEBUGMSG ((DBG_ERROR, "Transport Path is not selected")); extraData.Error = ERRUSER_ERROR_NOTRANSPORTPATH; __leave; }
while (status != TRSTATUS_READY) {
status = pGetOpaqueTransportStatus (g_TransportStatus, &g_OtCompressData, &g_OtOldFormat, TRUE);
switch (status) { case TRSTATUS_LOCKED: if (!IsmSendMessageToApp (TRANSPORTMESSAGE_IMAGE_LOCKED, 0)) { SetLastError (ERROR_ACCESS_DENIED); LOG ((LOG_ERROR, (PCSTR) MSG_TRANSPORT_DIR_BUSY, g_TransportPath)); extraData.Error = ERRUSER_ERROR_TRANSPORTPATHBUSY; __leave; } break; case TRSTATUS_DIRTY: SetLastError (ERROR_INVALID_DATA); LOG ((LOG_ERROR, (PCSTR) MSG_INVALID_IMAGE, g_TransportPath)); extraData.Error = ERRUSER_ERROR_TRANSPORTINVALIDIMAGE; __leave; case TRSTATUS_READY: break; default: SetLastError (ERROR_INVALID_DATA); LOG ((LOG_ERROR, (PCSTR) MSG_INVALID_IMAGE, g_TransportPath)); extraData.Error = ERRUSER_ERROR_TRANSPORTINVALIDIMAGE; __leave; } }
if (g_OtOldFormat) { IsmSendMessageToApp (TRANSPORTMESSAGE_OLD_STORAGE, 0); SetLastError (ERROR_INVALID_DATA); LOG ((LOG_ERROR, (PCSTR) MSG_INVALID_IMAGE, g_TransportPath)); extraData.Error = ERRUSER_ERROR_TRANSPORTINVALIDIMAGE; __leave; }
g_TransportStatusHandle = BfOpenReadFile (g_TransportStatus); if (!g_TransportStatusHandle) { LOG ((LOG_ERROR, (PCSTR) MSG_CANT_OPEN_STATUS_FILE, g_TransportStatus)); extraData.Error = ERRUSER_ERROR_CANTOPENSTATUS; __leave; }
if (g_OtCompressData) { g_TransportTempPath = pOtCreateTemporaryDir ();
if (!g_TransportTempPath) { CloseHandle (g_TransportStatusHandle); g_TransportStatusHandle = NULL; extraData.Error = ERRUSER_ERROR_CANTCREATETEMPDIR; __leave; }
if (!pOtReadAllImages ()) { CloseHandle (g_TransportStatusHandle); g_TransportStatusHandle = NULL; extraData.Error = ERRUSER_ERROR_CANTUNPACKIMAGE; __leave; } CloseHandle (g_TransportStatusHandle); g_TransportStatusHandle = NULL; }
newTransDbFile = pOpaqueGetNewFileName (S_TRANSPORT_DAT_FILE);
memDbFile = JoinPaths (pGetRealTransportPath (), newTransDbFile?newTransDbFile:S_TRANSPORT_DAT_FILE);
if (newTransDbFile) { FreePathString (newTransDbFile); newTransDbFile = NULL; }
result = MemDbLoad (memDbFile); if (!result) { extraData.Error = ERRUSER_ERROR_CANTREADIMAGE; } FreePathString (memDbFile); } __finally { }
if (!result) { extraData.ErrorArea = ERRUSER_AREA_LOAD; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); }
return result; }
VOID WINAPI OpaqueTransportEndApply ( VOID ) { MYASSERT (g_Platform == PLATFORM_DESTINATION);
if (g_OtCompressData) { pOtCleanUpTempDir (); } else { CloseHandle (g_TransportStatusHandle); g_TransportStatusHandle = NULL; } }
VOID WINAPI OpaqueTransportTerminate ( VOID ) { pOtCleanUpTempDir();
if (g_TransportTempPath) { FreePathString (g_TransportTempPath); g_TransportTempPath = NULL; } if (g_TransportStatus) { FreePathString (g_TransportStatus); g_TransportStatus = NULL; } if (g_TransportPath) { FreePathString (g_TransportPath); g_TransportPath = NULL; } }
static VOID pGetTempFileName ( OUT PTSTR Buffer ) { static fileIndex = 0;
fileIndex ++; wsprintf (Buffer, TEXT("%08X.DAT"), fileIndex); }
PCTSTR pOpaqueAllocStorageFileName ( IN PCTSTR FileName OPTIONAL ) { TCHAR buffer[32];
if (FileName) { StringCopy (buffer, FileName); } else { pGetTempFileName (buffer); }
return JoinPaths (g_TransportPath, buffer); }
VOID pFreeStorageFileName ( IN PCTSTR FileName ) { FreePathString (FileName); }
BOOL pOpaqueSaveDetails ( IN PCTSTR DecoratedObject, IN PMIG_DETAILS Details ) { PCTSTR key; BOOL b = FALSE;
if ((!Details) || (!Details->DetailsSize)) { return TRUE; }
key = JoinText (S_DETAILS_PREFIX, DecoratedObject);
if (key) {
b = (MemDbSetUnorderedBlob (key, 0, Details->DetailsData, Details->DetailsSize) != 0);
FreeText (key); }
return b; }
BOOL pOtAddFileToImage ( IN PCTSTR FileName, IN PCTSTR StoredName, IN OUT PCOMPRESS_HANDLE CompressedHandle ) { return CompressAddFileToHandle (FileName, StoredName, CompressedHandle); }
BOOL pOtSaveContentInFile ( IN MIG_OBJECTTYPEID ObjectTypeId, IN PCTSTR EncodedFileName, IN PCTSTR DecoratedObject, IN PMIG_CONTENT Content, IN OUT PCOMPRESS_HANDLE CompressedHandle OPTIONAL ) { BOOL result = FALSE; PCTSTR destPath = NULL; DWORD attributes = INVALID_ATTRIBUTES;
MYASSERT (Content->ContentInFile); if (!Content->ContentInFile) { return FALSE; }
//
// Use the CopyFile API to move the file from local to storage.
//
__try { if (Content && (Content->Details.DetailsSize == sizeof (WIN32_FIND_DATAW)) && Content->Details.DetailsData) { attributes = ((PWIN32_FIND_DATAW)Content->Details.DetailsData)->dwFileAttributes; } if ((attributes != INVALID_ATTRIBUTES) && (attributes & FILE_ATTRIBUTE_DIRECTORY)) {
// this must be a directory, let's just write the key
if (!MemDbSetValue (DecoratedObject, TRFLAG_FILE)) { __leave; }
} else {
//
// Get a temp file, assemble the src path, copy the file
//
destPath = pOpaqueAllocStorageFileName (NULL); if (!destPath) { __leave; }
if (CompressedHandle) { if (!pOtAddFileToImage (Content->FileContent.ContentPath, GetFileNameFromPath (destPath), CompressedHandle)) { __leave; } } else { if (!CopyFile (Content->FileContent.ContentPath, destPath, FALSE)) { __leave; } }
//
// Keep track of where the file went
//
if (!MemDbSetValue (DecoratedObject, TRFLAG_FILE)) { __leave; }
if (!MemDbAddSingleLinkage (DecoratedObject, GetFileNameFromPath (destPath), 0)) { __leave; } }
//
// Save details
//
result = pOpaqueSaveDetails (DecoratedObject, &(Content->Details));
} __finally { pFreeStorageFileName (destPath); INVALID_POINTER (destPath); }
return result; }
BOOL pOtSaveContentInMemory ( IN MIG_OBJECTTYPEID ObjectTypeId, IN PCTSTR EncodedObjectName, IN PCTSTR DecoratedObject, IN PMIG_CONTENT Content ) { BOOL result = FALSE;
MYASSERT (!Content->ContentInFile); if (Content->ContentInFile) { return FALSE; }
MemDbSetValue (DecoratedObject, TRFLAG_MEMORY);
if (Content->MemoryContent.ContentBytes && Content->MemoryContent.ContentSize) {
MemDbSetUnorderedBlob ( DecoratedObject, 0, Content->MemoryContent.ContentBytes, Content->MemoryContent.ContentSize ); }
result = pOpaqueSaveDetails (DecoratedObject, &(Content->Details));
return result; }
PCTSTR pOpaqueBuildDecoratedObject ( IN MIG_OBJECTTYPEID ObjectTypeId, IN ENCODEDSTRHANDLE ObjectName ) { PCTSTR typeStr;
typeStr = IsmGetObjectTypeName (ObjectTypeId); if (!typeStr) { return NULL; }
return JoinPaths (typeStr, ObjectName); }
VOID pOtDestroyDecoratedObject ( IN PCTSTR String ) { FreePathString (String); }
BOOL pOtSaveAllState ( IN BOOL Compressed ) { MIG_APPINFO appInfo; MIG_CONTENT value; PMIG_CONTENT convValue; ULONGLONG size; MIG_OBJECTWITHATTRIBUTE_ENUM objEnum; PCTSTR ourDbFile = NULL; PCTSTR decoratedObject = NULL; ULONGLONG bytesSaved = 0; UINT lastTick = GetTickCount(); TCHAR text[64]; UINT fraction; COMPRESS_HANDLE compressedHandle; INT_PTR appReply; BOOL okSave = FALSE; TRANSCOPY_ERROR transCopyError; ERRUSER_EXTRADATA extraData; BOOL result = FALSE;
__try {
ZeroMemory (&compressedHandle, sizeof (COMPRESS_HANDLE));
if (Compressed) { g_TransportTempPath = pOtCreateTemporaryDir ();
if (!g_TransportTempPath) { extraData.Error = ERRUSER_ERROR_CANTCREATETEMPDIR; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = 0; extraData.ObjectName = NULL; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); __leave; }
if (!CompressCreateHandle (g_TransportPath, S_TRANSPORT_IMG_FILE, 1, 0, &compressedHandle)) { extraData.Error = ERRUSER_ERROR_CANTCREATECABFILE; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = 0; extraData.ObjectName = NULL; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); __leave; } }
//
// Enumerate all objects with "save" attribute
//
if (IsmEnumFirstPersistentObject (&objEnum)) { do { //
// For each object to be saved, do the appropriate
// data copy action
//
okSave = FALSE; while (!okSave) {
if (!IsmAcquireObjectEx ( objEnum.ObjectTypeId, objEnum.ObjectName, &value, CONTENTTYPE_ANY, 0 )) {
transCopyError.ObjectType = IsmGetObjectTypeName (objEnum.ObjectTypeId); transCopyError.ObjectName = IsmGetNativeObjectName (objEnum.ObjectTypeId, objEnum.ObjectName); transCopyError.Error = GetLastError ();
if (IsmIsNonCriticalObject (objEnum.ObjectTypeId, objEnum.ObjectName)) { appReply = APPRESPONSE_IGNORE; } else { appReply = IsmSendMessageToApp (TRANSPORTMESSAGE_SRC_COPY_ERROR, (ULONG_PTR)&transCopyError); if ((appReply == APPRESPONSE_NONE) || (appReply == APPRESPONSE_FAIL) ) { LOG ((LOG_ERROR, (PCSTR) MSG_CANT_COPYSOURCE, transCopyError.ObjectName)); IsmReleaseMemory (transCopyError.ObjectName); extraData.Error = ERRUSER_ERROR_CANTSAVEOBJECT; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = objEnum.ObjectTypeId; extraData.ObjectName = objEnum.ObjectName; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); IsmAbortPersistentObjectEnum (&objEnum); __leave; } } if (appReply == APPRESPONSE_IGNORE) { LOG ((LOG_WARNING, (PCSTR) MSG_IGNORE_COPYSOURCE, transCopyError.ObjectName)); IsmReleaseMemory (transCopyError.ObjectName); extraData.Error = ERRUSER_ERROR_CANTSAVEOBJECT; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = objEnum.ObjectTypeId; extraData.ObjectName = objEnum.ObjectName; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYWARNING, (ULONG_PTR)(&extraData)); break; } IsmReleaseMemory (transCopyError.ObjectName); continue; } okSave = TRUE; }
if (okSave) {
#ifdef UNICODE
convValue = &value; #else
// now let's convert this object content to UNICODE
convValue = IsmConvertObjectContentToUnicode (objEnum.ObjectTypeId, objEnum.ObjectName, &value); if (!convValue) { convValue = &value; } #endif
decoratedObject = pOpaqueBuildDecoratedObject (objEnum.ObjectTypeId, objEnum.ObjectName);
ZeroMemory (&appInfo, sizeof (MIG_APPINFO)); appInfo.Phase = MIG_TRANSPORT_PHASE; appInfo.SubPhase = SUBPHASE_COMPRESSING; appInfo.ObjectTypeId = (objEnum.ObjectTypeId & (~PLATFORM_MASK)); appInfo.ObjectName = objEnum.ObjectName; IsmSendMessageToApp (ISMMESSAGE_APP_INFO, (ULONG_PTR)(&appInfo));
if (convValue->ContentInFile) { okSave = FALSE; while (!okSave) { if (!pOtSaveContentInFile (objEnum.ObjectTypeId, objEnum.ObjectName, decoratedObject, convValue, &compressedHandle)) { if (GetLastError () == ERROR_DISK_FULL) { // we just failed because we don't have enough space on the destination
// path. Let's tell that to the user
extraData.Error = ERRUSER_ERROR_CANTCREATECABFILE; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = 0; extraData.ObjectName = NULL; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); __leave; }
transCopyError.ObjectType = IsmGetObjectTypeName (objEnum.ObjectTypeId); transCopyError.ObjectName = IsmGetNativeObjectName (objEnum.ObjectTypeId, objEnum.ObjectName); transCopyError.Error = GetLastError ();
if (IsmIsNonCriticalObject (objEnum.ObjectTypeId, objEnum.ObjectName)) { appReply = APPRESPONSE_IGNORE; } else { appReply = IsmSendMessageToApp (TRANSPORTMESSAGE_SRC_COPY_ERROR, (ULONG_PTR)&transCopyError); if ((appReply == APPRESPONSE_NONE) || (appReply == APPRESPONSE_FAIL) ) { LOG ((LOG_ERROR, (PCSTR) MSG_CANT_COPYSOURCE, transCopyError.ObjectName)); IsmReleaseMemory (transCopyError.ObjectName); extraData.Error = ERRUSER_ERROR_CANTSAVEOBJECT; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = objEnum.ObjectTypeId; extraData.ObjectName = objEnum.ObjectName; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); IsmAbortPersistentObjectEnum (&objEnum); __leave; } } if (appReply == APPRESPONSE_IGNORE) { LOG ((LOG_WARNING, (PCSTR) MSG_IGNORE_COPYSOURCE, transCopyError.ObjectName)); IsmReleaseMemory (transCopyError.ObjectName); extraData.Error = ERRUSER_ERROR_CANTSAVEOBJECT; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = objEnum.ObjectTypeId; extraData.ObjectName = objEnum.ObjectName; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYWARNING, (ULONG_PTR)(&extraData)); break; } IsmReleaseMemory (transCopyError.ObjectName); continue; } okSave = TRUE; } size = convValue->FileContent.ContentSize; } else { okSave = FALSE; while (!okSave) { if (!pOtSaveContentInMemory (objEnum.ObjectTypeId, objEnum.ObjectName, decoratedObject, convValue)) { if (GetLastError () == ERROR_DISK_FULL) { // we just failed because we don't have enough space on the destination
// path. Let's tell that to the user
extraData.Error = ERRUSER_ERROR_CANTCREATECABFILE; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = 0; extraData.ObjectName = NULL; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); __leave; }
transCopyError.ObjectType = IsmGetObjectTypeName (objEnum.ObjectTypeId); transCopyError.ObjectName = IsmGetNativeObjectName (objEnum.ObjectTypeId, objEnum.ObjectName); transCopyError.Error = GetLastError ();
if (IsmIsNonCriticalObject (objEnum.ObjectTypeId, objEnum.ObjectName)) { appReply = APPRESPONSE_IGNORE; } else { appReply = IsmSendMessageToApp (TRANSPORTMESSAGE_SRC_COPY_ERROR, (ULONG_PTR)&transCopyError); if ((appReply == APPRESPONSE_NONE) || (appReply == APPRESPONSE_FAIL) ) { LOG ((LOG_ERROR, (PCSTR) MSG_CANT_COPYSOURCE, transCopyError.ObjectName)); IsmReleaseMemory (transCopyError.ObjectName); extraData.Error = ERRUSER_ERROR_CANTSAVEOBJECT; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = objEnum.ObjectTypeId; extraData.ObjectName = objEnum.ObjectName; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); IsmAbortPersistentObjectEnum (&objEnum); __leave; } } if (appReply == APPRESPONSE_IGNORE) { LOG ((LOG_WARNING, (PCSTR) MSG_IGNORE_COPYSOURCE, transCopyError.ObjectName)); IsmReleaseMemory (transCopyError.ObjectName); extraData.Error = ERRUSER_ERROR_CANTSAVEOBJECT; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = objEnum.ObjectTypeId; extraData.ObjectName = objEnum.ObjectName; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYWARNING, (ULONG_PTR)(&extraData)); break; } IsmReleaseMemory (transCopyError.ObjectName); continue; } okSave = TRUE; } size = convValue->MemoryContent.ContentSize; }
#ifndef UNICODE
if (convValue != (&value)) { IsmFreeConvertedObjectContent (objEnum.ObjectTypeId, convValue); } #endif
IsmReleaseObject (&value);
pOtDestroyDecoratedObject (decoratedObject); decoratedObject = NULL; }
IsmTickProgressBar (g_PersistentSlice, 1);
if (IsmCheckCancel()) { __leave; }
//
// Send bytes saved to app every 3 seconds
//
bytesSaved += size;
if (GetTickCount() - lastTick > 3000) {
if (bytesSaved < 1048576) { wsprintf (text, TEXT("Saved: %u K"), (UINT) (bytesSaved / 1024)); } else if (bytesSaved < 8388608) { fraction = (UINT) (bytesSaved / 10485); wsprintf (text, TEXT("Saved: %u.%02u M"), fraction / 100, fraction % 100); } else if (bytesSaved < 1073741824) { wsprintf (text, TEXT("Saved: %u M"), (UINT) (bytesSaved / 1048576)); } else { fraction = (UINT) (bytesSaved / 10737418); wsprintf (text, TEXT("Saved: %u.%02u G"), fraction / 100, fraction % 100); }
IsmSendMessageToApp (TRANSPORTMESSAGE_SIZE_SAVED, (ULONG_PTR) text);
lastTick = GetTickCount(); }
} while (IsmEnumNextPersistentObject (&objEnum)); }
if (IsmCheckCancel()) { __leave; }
ZeroMemory (&appInfo, sizeof (MIG_APPINFO)); appInfo.Phase = MIG_TRANSPORT_PHASE; appInfo.SubPhase = SUBPHASE_FINISHING; IsmSendMessageToApp (ISMMESSAGE_APP_INFO, (ULONG_PTR)(&appInfo));
if (Compressed) { ourDbFile = JoinPaths (g_TransportTempPath, S_TRANSPORT_DAT_FILE); } else { ourDbFile = pOpaqueAllocStorageFileName (S_TRANSPORT_DAT_FILE); if (!ourDbFile) { extraData.Error = ERRUSER_ERROR_CANTSAVEINTERNALDATA; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = 0; extraData.ObjectName = NULL; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); __leave; } }
if (!MemDbSave (ourDbFile)) { extraData.Error = ERRUSER_ERROR_CANTSAVEINTERNALDATA; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = 0; extraData.ObjectName = NULL; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); __leave; }
IsmTickProgressBar (g_DatabaseSlice, 1);
if (Compressed) { if (IsmCheckCancel()) { __leave; }
if (!pOtAddFileToImage (ourDbFile, S_TRANSPORT_DAT_FILE, &compressedHandle)) { extraData.Error = ERRUSER_ERROR_CANTSAVEINTERNALDATA; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = 0; extraData.ObjectName = NULL; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); __leave; }
IsmTickProgressBar (g_DatabaseSlice, 1);
if (IsmCheckCancel()) { __leave; }
if (!CompressFlushAndCloseHandle (&compressedHandle)) { extraData.Error = ERRUSER_ERROR_CANTCREATECABFILE; extraData.ErrorArea = ERRUSER_AREA_SAVE; extraData.ObjectTypeId = 0; extraData.ObjectName = NULL; IsmSendMessageToApp (MODULEMESSAGE_DISPLAYERROR, (ULONG_PTR)(&extraData)); __leave; }
IsmTickProgressBar (g_DatabaseSlice, 1);
if (IsmCheckCancel()) { __leave; } }
IsmSendMessageToApp (TRANSPORTMESSAGE_SIZE_SAVED, 0); result = TRUE;
} __finally { PushError (); CompressCleanupHandle (&compressedHandle);
pFreeStorageFileName (ourDbFile); INVALID_POINTER (ourDbFile);
pOtDestroyDecoratedObject (decoratedObject); INVALID_POINTER (decoratedObject);
if (g_OtCompressData && g_TransportTempPath) { pOtCleanUpTempDir (); } PopError (); }
return result; }
BOOL WINAPI OpaqueTransportAcquireObject ( IN MIG_OBJECTTYPEID ObjectTypeId, IN MIG_OBJECTSTRINGHANDLE ObjectName, OUT PMIG_CONTENT ObjectContent, CALLER_INITIALIZED IN MIG_CONTENTTYPE ContentType, IN UINT MemoryContentLimit ) { UINT value; PCBYTE memValue; UINT memValueSize; PCTSTR fileValue = NULL; PCTSTR newFileValue = NULL; BOOL valueInFile; KEYHANDLE keyHandle; PALLOCSTATE allocState; PCTSTR detailsKey = NULL; PBYTE details; UINT detailsSize; PCTSTR sourceFile; PCTSTR decoratedObject = NULL; HANDLE fileHandle; BOOL result = FALSE;
if (!ObjectContent) { return FALSE; }
MYASSERT (g_Platform == PLATFORM_DESTINATION); MYASSERT ((ObjectTypeId & PLATFORM_MASK) == PLATFORM_SOURCE);
decoratedObject = pOpaqueBuildDecoratedObject (ObjectTypeId, ObjectName);
allocState = (PALLOCSTATE) MemAllocZeroed (sizeof (ALLOCSTATE));
if (MemDbGetValue (decoratedObject, &value)) { if (value == TRFLAG_FILE) { valueInFile = TRUE; keyHandle = MemDbGetSingleLinkage (decoratedObject, 0, 0); if (keyHandle) { fileValue = MemDbGetKeyFromHandle (keyHandle, 0); newFileValue = pOpaqueGetNewFileName (fileValue); result = fileValue != NULL; } else { fileValue = NULL; result = TRUE; } } else if (value == TRFLAG_MEMORY) { valueInFile = FALSE; memValueSize = 0; memValue = MemDbGetUnorderedBlob (decoratedObject, 0, &memValueSize); result = TRUE; } else { LOG ((LOG_ERROR, (PCSTR) MSG_UNSUPPORTED_DATA, value)); SetLastError (ERROR_RESOURCE_NAME_NOT_FOUND); } if (result) { result = FALSE;
if (valueInFile) { if ((ContentType == CONTENTTYPE_ANY) || (ContentType == CONTENTTYPE_FILE) || (ContentType == CONTENTTYPE_DETAILS_ONLY) ) { // this is stored as a file and it's wanted as a file
ObjectContent->ObjectTypeId = ObjectTypeId; ObjectContent->ContentInFile = TRUE; if (fileValue) { ObjectContent->FileContent.ContentPath = JoinPaths (pGetRealTransportPath (), newFileValue?newFileValue:fileValue); ObjectContent->FileContent.ContentSize = BfGetFileSize (ObjectContent->FileContent.ContentPath); } else { ObjectContent->FileContent.ContentSize = 0; ObjectContent->FileContent.ContentPath = NULL; } result = TRUE; } else { // this is stored as a file and it's wanted as memory
ObjectContent->ObjectTypeId = ObjectTypeId; ObjectContent->ContentInFile = FALSE; if (fileValue) { sourceFile = JoinPaths (pGetRealTransportPath (), newFileValue?newFileValue:fileValue); ObjectContent->MemoryContent.ContentSize = (UINT) BfGetFileSize (sourceFile); ObjectContent->MemoryContent.ContentBytes = MapFileIntoMemory ( sourceFile, &allocState->FileHandle, &allocState->MapHandle ); FreePathString (sourceFile); result = (ObjectContent->MemoryContent.ContentBytes != NULL); } else { ObjectContent->MemoryContent.ContentSize = 0; ObjectContent->MemoryContent.ContentBytes = NULL; result = TRUE; } } if (newFileValue) { FreePathString (newFileValue); newFileValue = NULL; } if (fileValue) { MemDbReleaseMemory (fileValue); fileValue = NULL; } } else { if ((ContentType == CONTENTTYPE_ANY) || (ContentType == CONTENTTYPE_MEMORY) || (ContentType == CONTENTTYPE_DETAILS_ONLY) ) { // this is stored as memory and it's wanted as memory
ObjectContent->ObjectTypeId = ObjectTypeId; ObjectContent->ContentInFile = FALSE; ObjectContent->MemoryContent.ContentSize = memValueSize; ObjectContent->MemoryContent.ContentBytes = memValue; result = TRUE; } else { // this is stored as memory and it's wanted as a file
if (memValue) { if (IsmGetTempFile (allocState->TempFile, ARRAYSIZE(allocState->TempFile))) { fileHandle = BfCreateFile (allocState->TempFile); if (fileHandle) { if (BfWriteFile (fileHandle, memValue, memValueSize)) { ObjectContent->ObjectTypeId = ObjectTypeId; ObjectContent->ContentInFile = TRUE; ObjectContent->FileContent.ContentSize = memValueSize; ObjectContent->FileContent.ContentPath = DuplicatePathString (allocState->TempFile, 0); result = TRUE; } CloseHandle (fileHandle); } } MemDbReleaseMemory (memValue); } else { ObjectContent->ObjectTypeId = ObjectTypeId; ObjectContent->ContentInFile = TRUE; ObjectContent->FileContent.ContentSize = 0; ObjectContent->FileContent.ContentPath = NULL; } } } } } else { SetLastError (ERROR_RESOURCE_NAME_NOT_FOUND); }
if (result) { //
// Fill the details
//
detailsKey = JoinText (S_DETAILS_PREFIX, decoratedObject);
details = MemDbGetUnorderedBlob (detailsKey, 0, &detailsSize);
if (!details) { detailsSize = 0; }
allocState->DetailsPtr = details;
ObjectContent->Details.DetailsSize = detailsSize; ObjectContent->Details.DetailsData = details;
FreeText (detailsKey);
ObjectContent->TransHandle = allocState; }
if (!result) { FreeAlloc (allocState); }
FreePathString (decoratedObject);
return result; }
BOOL WINAPI OpaqueTransportReleaseObject ( IN OUT PMIG_CONTENT ObjectContent ) { PALLOCSTATE allocState;
MYASSERT (g_Platform == PLATFORM_DESTINATION);
allocState = (PALLOCSTATE) ObjectContent->TransHandle;
if (ObjectContent->ContentInFile) { FreePathString (ObjectContent->FileContent.ContentPath); if (allocState && allocState->TempFile[0]) { DeleteFile (allocState->TempFile); } } else { if (allocState && allocState->FileHandle && allocState->MapHandle) { UnmapFile ( ObjectContent->MemoryContent.ContentBytes, allocState->MapHandle, allocState->FileHandle ); } else { MemDbReleaseMemory (ObjectContent->MemoryContent.ContentBytes); } }
if (allocState && allocState->DetailsPtr) { MemDbReleaseMemory (allocState->DetailsPtr); }
FreeAlloc (allocState);
return TRUE; }
|