mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
862 lines
21 KiB
862 lines
21 KiB
//+-------------------------------------------------------------------
|
|
//
|
|
// File: smarshal.cxx
|
|
//
|
|
// Synopsis: Source code for Interface Marshaling stress test main
|
|
// driver functions. Source for individual tests is in
|
|
// testvar.cxx
|
|
//
|
|
// History: 21-Aug-95 Rickhi Created
|
|
//
|
|
//--------------------------------------------------------------------
|
|
#include <smarshal.hxx>
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Globals:
|
|
//
|
|
//--------------------------------------------------------------------
|
|
BOOL gfVerbose = FALSE; // print execution messages
|
|
BOOL gfDebug = FALSE; // print debug messages
|
|
|
|
DWORD giThreadModel = OPF_INITAPARTMENT; // threading model to use
|
|
int giTestVar = 0; // test variation to run
|
|
int giHighestTestVar = 2; // highest test var supported
|
|
|
|
int gicReps = 5; // number of repetitions of each test
|
|
int gicThreads = 1; // number of threads to use on each test
|
|
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Private Function ProtoTypes:
|
|
//
|
|
//--------------------------------------------------------------------
|
|
HRESULT DoIfOperation(DWORD dwFlags, INTERFACEPARAMS *pIFD);
|
|
void DisplayHelp(void);
|
|
BOOL GetSwitch(CHAR *pCmdParam, CHAR *pszSwitch);
|
|
BOOL GetSwitchInt(CHAR *pCmdParam, CHAR *pszSwitch, int *pInt);
|
|
BOOL ParseCmdLine(int argc, char **argv);
|
|
int _cdecl main(int argc, char **argv);
|
|
|
|
BOOL FreeWaitForEvent(HANDLE hEvent);
|
|
BOOL AptWaitForEvent(HANDLE hEvent);
|
|
|
|
STDAPI_(LPSTREAM) CreateMemStm(DWORD cb, LPHANDLE phdl);
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Misc:
|
|
//
|
|
//--------------------------------------------------------------------
|
|
|
|
#ifndef _CAIRO_
|
|
// COM initialization flags; passed to CoInitialize.
|
|
typedef enum tagCOINIT
|
|
{
|
|
COINIT_MULTITHREADED = 0, // OLE calls objects on any thread.
|
|
COINIT_SINGLETHREADED = 1, // OLE calls objects on single thread.
|
|
COINIT_APARTMENTTHREADED = 2 // Apartment model
|
|
} COINIT;
|
|
|
|
WINOLEAPI CoInitializeEx(LPVOID pvReserved, DWORD);
|
|
#endif
|
|
|
|
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: main
|
|
//
|
|
// Synopsis: Entry point to EXE. Parse the command line, then run
|
|
// whatever test variations were selected.
|
|
//
|
|
// Notes: The test variation code is in testvars.cxx.
|
|
// The rest of this file is helper routines.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
int _cdecl main(int argc, char **argv)
|
|
{
|
|
// parse the command line
|
|
BOOL fRes = ParseCmdLine(argc, argv);
|
|
|
|
if (fRes)
|
|
{
|
|
// run the selected test variations
|
|
switch (giTestVar)
|
|
{
|
|
case 1: fRes = TestVar1();
|
|
break;
|
|
case 2: fRes = TestVar2();
|
|
break;
|
|
default: break;
|
|
}
|
|
}
|
|
|
|
// check the results
|
|
CHKTESTRESULT(fRes, "Marshal Stress Tests");
|
|
return 0;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: ParseCmdLine
|
|
//
|
|
// Synopsis: parses the execution parameters
|
|
//
|
|
//--------------------------------------------------------------------
|
|
BOOL ParseCmdLine(int argc, char **argv)
|
|
{
|
|
BOOL fDontRun = (argc == 1) ? TRUE : FALSE;
|
|
|
|
// the first parameter is the EXE name, skip it.
|
|
argc--;
|
|
argv++;
|
|
|
|
for (int i=0; i<argc; i++, argv++)
|
|
{
|
|
if (GetSwitch(*argv, "v"))
|
|
{
|
|
// run verbose
|
|
gfVerbose = TRUE;
|
|
}
|
|
else if (GetSwitch(*argv, "d"))
|
|
{
|
|
// run debug mode
|
|
gfVerbose = TRUE;
|
|
gfDebug = TRUE;
|
|
}
|
|
else if (GetSwitch(*argv, "h") || GetSwitch(*argv, "?"))
|
|
{
|
|
// help wanted
|
|
fDontRun = TRUE;
|
|
}
|
|
else if (GetSwitchInt(*argv, "var:", &giTestVar))
|
|
{
|
|
// selected test variation, ensure the variation is valid
|
|
if (giTestVar > giHighestTestVar)
|
|
{
|
|
ERROUT("Unknown Test Variation:%x\n", giTestVar);
|
|
fDontRun = TRUE;
|
|
}
|
|
}
|
|
else if (GetSwitchInt(*argv, "reps:", &gicReps))
|
|
{
|
|
; // selected repetition count
|
|
}
|
|
else if (GetSwitchInt(*argv, "threads:", &gicThreads))
|
|
{
|
|
; // selected thread count
|
|
}
|
|
else if (GetSwitch(*argv, "model:apt"))
|
|
{
|
|
// run apartment model
|
|
giThreadModel = OPF_INITAPARTMENT;
|
|
}
|
|
else if (GetSwitch(*argv, "model:free"))
|
|
{
|
|
// run freethreaded model
|
|
giThreadModel = OPF_INITFREE;
|
|
}
|
|
else
|
|
{
|
|
// unknown switch, show help
|
|
ERROUT("Unknown command line switch:<%s>\n", *argv);
|
|
fDontRun = TRUE;
|
|
}
|
|
}
|
|
|
|
if (fDontRun)
|
|
{
|
|
// help is wanted
|
|
DisplayHelp();
|
|
return FALSE;
|
|
}
|
|
|
|
// success, run the test
|
|
return TRUE;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: DisplayHelp
|
|
//
|
|
// Synopsis: displays a command line help message
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void DisplayHelp(void)
|
|
{
|
|
printf("\nCommand Line Switches for Marshal Stress Test:\n\n");
|
|
|
|
printf("-d - debug mode\n");
|
|
printf("-v - verbose mode\n");
|
|
printf("-h | -? - display this help message\n\n");
|
|
|
|
printf("-model:[apt|free] - threading model to use for test\n");
|
|
printf("-var:# - test variation to execute 1-%x\n", giHighestTestVar);
|
|
printf("-threads:# - number of threads per variation\n");
|
|
printf("-reps:# - number of repetitions of the test\n");
|
|
|
|
printf("\n");
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: GetSwitch
|
|
//
|
|
// Synopsis: returns TRUE if the specified cmd line switch is set.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
BOOL GetSwitch(CHAR *pCmdParam, CHAR *pszSwitch)
|
|
{
|
|
if (*pCmdParam == '-' || *pCmdParam == '/')
|
|
{
|
|
pCmdParam++;
|
|
return (!stricmp(pCmdParam, pszSwitch));
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: GetSwitchInt
|
|
//
|
|
// Synopsis: returns TRUE if the specified cmd line switch is set,
|
|
// and sets the value of that switch.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
BOOL GetSwitchInt(CHAR *pCmdParam, CHAR *pszSwitch, int *pInt)
|
|
{
|
|
if (*pCmdParam == '-' || *pCmdParam == '/')
|
|
{
|
|
pCmdParam++;
|
|
int len = strlen(pszSwitch);
|
|
if (!strnicmp(pCmdParam, pszSwitch, len))
|
|
{
|
|
pCmdParam += len;
|
|
*pInt = atoi(pCmdParam);
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: CHKRESULT
|
|
//
|
|
// Synopsis: prints a failure message if the result code indicates
|
|
// failure., and success message only in verbose mode.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void CHKRESULT(HRESULT hr, CHAR *pszOperation)
|
|
{
|
|
if (FAILED(hr))
|
|
{
|
|
printf("FAILED hr:%x Op:%s\n", hr, pszOperation);
|
|
}
|
|
else if (gfVerbose)
|
|
{
|
|
printf("PASSED hr:%x Op:%s\n", hr, pszOperation);
|
|
}
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: CHKOP
|
|
//
|
|
// Synopsis: prints a failure message if the result code indicates
|
|
// failure, and success message only in debug mode.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void CHKOP(HRESULT hr, CHAR *pszOperation)
|
|
{
|
|
if (FAILED(hr))
|
|
{
|
|
printf("FAILED hr:%x Op:%s\n", hr, pszOperation);
|
|
}
|
|
else if (gfDebug)
|
|
{
|
|
printf("PASSED hr:%x Op:%s\n", hr, pszOperation);
|
|
}
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: CHKTESTRESULT
|
|
//
|
|
// Synopsis: prints a pass or fail message
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void CHKTESTRESULT(BOOL fRes, CHAR *pszMsg)
|
|
{
|
|
if (fRes)
|
|
printf("%s PASSED\n", pszMsg);
|
|
else
|
|
printf("%s FAILED\n", pszMsg);
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: GetEvent / ReleaseEvent
|
|
//
|
|
// Synopsis: allocates or releases an event
|
|
//
|
|
// CODEWORK: cache these for frequent use
|
|
//
|
|
//--------------------------------------------------------------------
|
|
HANDLE GetEvent()
|
|
{
|
|
HANDLE hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
|
|
if (hEvent == NULL)
|
|
{
|
|
ERROUT("GetEvent FAILED\n");
|
|
}
|
|
DBGOUT("CreateEvent hEvent:%x\n", hEvent);
|
|
return hEvent;
|
|
}
|
|
|
|
void ReleaseEvent(HANDLE hEvent)
|
|
{
|
|
if (hEvent)
|
|
{
|
|
DBGOUT("ReleaseEvent hEvent:%x\n", hEvent);
|
|
CloseHandle(hEvent);
|
|
}
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: WaitForEvent
|
|
//
|
|
// Synopsis: waits on the given event (if there is one) for a
|
|
// certain amount of time, returns FALSE if timedout.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
BOOL WaitForEvent(HANDLE hEvent)
|
|
{
|
|
if (hEvent)
|
|
{
|
|
// CODEWORK: base off the Threading Model
|
|
|
|
return (TRUE) ? AptWaitForEvent(hEvent) :
|
|
FreeWaitForEvent(hEvent);
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: FreeWaitForEvent
|
|
//
|
|
// Synopsis: FreeThreaded version of WaitForEvent. Waits on the
|
|
// given event.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
BOOL FreeWaitForEvent(HANDLE hEvent)
|
|
{
|
|
DBGOUT("FreeWaitForEvent hEvent:%x\n", hEvent);
|
|
if (WaitForSingleObject(hEvent, 30000) == WAIT_TIMEOUT)
|
|
{
|
|
ERROUT("WaitForSingleObject TimedOut");
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: AptWaitForEvent
|
|
//
|
|
// Synopsis: Apartment model version of WaitForEvent. Waits on the
|
|
// given event. Dispatches all incoming windows messages
|
|
// while waiting.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
BOOL AptWaitForEvent(HANDLE hEvent)
|
|
{
|
|
DBGOUT("AptWaitForEvent hEvent:%x\n", hEvent);
|
|
|
|
while (1)
|
|
{
|
|
HANDLE arEvent[] = {hEvent};
|
|
DWORD dwWakeReason = MsgWaitForMultipleObjects(1, arEvent, FALSE,
|
|
1000, QS_ALLINPUT);
|
|
|
|
if (dwWakeReason == WAIT_OBJECT_0)
|
|
{
|
|
// event was signalled. exit.
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
// check for and dispatch any messages that have arrived
|
|
MSG msg;
|
|
while (PeekMessage(&msg, 0, WM_NULL, WM_NULL, PM_REMOVE))
|
|
{
|
|
DispatchMessage(&msg);
|
|
}
|
|
}
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: SignalEvent
|
|
//
|
|
// Synopsis: signals an event (if there is one)
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void SignalEvent(HANDLE hEvent)
|
|
{
|
|
if (hEvent)
|
|
{
|
|
DBGOUT("SignalEvent hEvent:%x\n", hEvent);
|
|
SetEvent(hEvent);
|
|
}
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: GetStream / ReleaseStream
|
|
//
|
|
// Synopsis: allocates or releases a Stream
|
|
//
|
|
// CODEWORK: cache these for frequent use
|
|
// CODEWORK: use CStreamOnFile for cross process/machine marshaling.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
IStream * GetStream(void)
|
|
{
|
|
IStream *pStm = CreateMemStm(600, NULL);
|
|
if (pStm == NULL)
|
|
{
|
|
ERROUT("ERROR: GetStream FAILED\n");
|
|
}
|
|
DBGOUT("GetStream pStm:%x\n", pStm);
|
|
return pStm;
|
|
}
|
|
|
|
void ReleaseStream(IStream *pStm)
|
|
{
|
|
if (pStm)
|
|
{
|
|
DBGOUT("ReleaseStream pStm:%x\n", pStm);
|
|
pStm->Release();
|
|
}
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: ResetStream
|
|
//
|
|
// Synopsis: resets a steam back to the start
|
|
//
|
|
//--------------------------------------------------------------------
|
|
HRESULT ResetStream(IStream *pStm)
|
|
{
|
|
DBGOUT("ResetStream pStm:%x\n", pStm);
|
|
|
|
LARGE_INTEGER libMove;
|
|
libMove.LowPart = 0;
|
|
libMove.HighPart = 0;
|
|
|
|
HRESULT hr = pStm->Seek(libMove, STREAM_SEEK_SET, 0);
|
|
if (FAILED(hr))
|
|
{
|
|
ERROUT("ERROR: ResetStream FAILED hr:%x\n",hr);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: GetInterface / ReleaseInterface
|
|
//
|
|
// Synopsis: allocates or releases an object interface
|
|
//
|
|
//--------------------------------------------------------------------
|
|
IUnknown *GetInterface(void)
|
|
{
|
|
IUnknown *punk = (IUnknown *) new CTestUnk();
|
|
if (punk == NULL)
|
|
{
|
|
ERROUT("ERROR: GetInterface FAILED\n");
|
|
}
|
|
DBGOUT("GetInterface punk:%x\n", punk);
|
|
return punk;
|
|
}
|
|
|
|
void ReleaseInterface(IUnknown *punk)
|
|
{
|
|
if (punk)
|
|
{
|
|
DBGOUT("ReleaseInterface punk:%x\n", punk);
|
|
punk->Release();
|
|
}
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: GenericExecute
|
|
//
|
|
// Synopsis: run all the parameter blocks on different threads
|
|
// simultaneously.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
BOOL GenericExecute(ULONG cEPs, EXECPARAMS *pEP[])
|
|
{
|
|
BOOL fRes = TRUE;
|
|
DBGOUT("Start GenericExecute cEPs:%x\n", cEPs);
|
|
|
|
HANDLE hThread[50];
|
|
|
|
// launch a thread to run each command block
|
|
for (ULONG i=0; i<cEPs; i++)
|
|
{
|
|
// launch a thread to execute the parameter block
|
|
DWORD dwThreadId;
|
|
hThread[i] = CreateThread(NULL, 0,
|
|
WorkerThread,
|
|
pEP[i],
|
|
0,
|
|
&dwThreadId);
|
|
}
|
|
|
|
// signal all the threads to start their work
|
|
for (i=0; i<cEPs; i++)
|
|
{
|
|
SignalEvent(pEP[i]->hEventThreadStart);
|
|
}
|
|
|
|
// wait for all the threads to complete their work
|
|
for (i=0; i<cEPs; i++)
|
|
{
|
|
if (pEP[i]->hEventThreadDone)
|
|
{
|
|
WaitForSingleObject(pEP[i]->hEventThreadDone, 60000);
|
|
}
|
|
}
|
|
|
|
// signal all the threads to exit
|
|
for (i=0; i<cEPs; i++)
|
|
{
|
|
HANDLE hEventThreadExit = pEP[i]->hEventThreadExit;
|
|
pEP[i]->hEventThreadExit = NULL;// set to NULL so only the thread will
|
|
// release it, genericcleanup wont.
|
|
SignalEvent(hEventThreadExit);
|
|
}
|
|
|
|
// wait for all the threads to exit
|
|
for (i=0; i<cEPs; i++)
|
|
{
|
|
WaitForSingleObject(hThread[i], 5000);
|
|
CloseHandle(hThread[i]);
|
|
}
|
|
|
|
DBGOUT("Done GenericExecute fRes:%x\n", fRes);
|
|
return fRes;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: GenericCleanup
|
|
//
|
|
// Synopsis: clean all the parameter blocks
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void GenericCleanup(ULONG cEPs, EXECPARAMS *pEP[])
|
|
{
|
|
DBGOUT("GenericCleanup\n");
|
|
|
|
// delete the execution parameter blocks
|
|
for (ULONG i=0; i<cEPs; i++)
|
|
{
|
|
ReleaseExecParam(pEP[i]);
|
|
// CODEWORK: get results from the parameter block?
|
|
}
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: WorkerThread
|
|
//
|
|
// Synopsis: entry point for thread that executes a series of
|
|
// interface commands
|
|
//
|
|
//--------------------------------------------------------------------
|
|
DWORD _stdcall WorkerThread(void *params)
|
|
{
|
|
DBGOUT("StartWorkerThread TID:%x pEP:%x\n", GetCurrentThreadId(), params);
|
|
|
|
EXECPARAMS *pEP = (EXECPARAMS *)params;
|
|
if (pEP == NULL)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
HRESULT hr = S_OK;
|
|
|
|
// Initialize OLE for this thread.
|
|
if (pEP->dwFlags & OPF_INITAPARTMENT)
|
|
{
|
|
hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
|
|
}
|
|
else
|
|
{
|
|
hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
ERROUT("ERROR: CoInitializeEx FAILED\n");
|
|
return hr;
|
|
}
|
|
|
|
// wait for signal to start the test
|
|
if (!WaitForEvent(pEP->hEventThreadStart))
|
|
{
|
|
return E_OUTOFMEMORY; // BUGBUG
|
|
}
|
|
|
|
// loop for the number of reps requested
|
|
for (ULONG iRep = 0; iRep < pEP->cReps; iRep++)
|
|
{
|
|
// wait for the start signal
|
|
if (!WaitForEvent(pEP->hEventRepStart))
|
|
{
|
|
return E_OUTOFMEMORY; // BUGBUG
|
|
}
|
|
|
|
MSGOUT(" TID:%x Rep:%x of %x\n",
|
|
GetCurrentThreadId(), iRep, pEP->cReps);
|
|
|
|
// loop for the number of INTERFACEPARAMSs, performing
|
|
// the requested operation(s) on each interface.
|
|
|
|
for (ULONG iIP=0; iIP < pEP->cIPs; iIP++)
|
|
{
|
|
hr = DoIfOperation(pEP->dwFlags, &(pEP->aIP[iIP]));
|
|
}
|
|
|
|
// signal the completion event
|
|
SignalEvent(pEP->hEventRepDone);
|
|
}
|
|
|
|
// signal the thread completion event. Cant touch pEP after this
|
|
// point in time since the main thread may delete it. We extract
|
|
// the ThreadExit event and NULL it in the parameter block so that
|
|
// the main thread wont release it. We release it after the event
|
|
// has been signaled.
|
|
|
|
HANDLE hEventThreadExit = pEP->hEventThreadExit;
|
|
SignalEvent(pEP->hEventThreadDone);
|
|
|
|
// wait on the thread exit event. This allows other threads to
|
|
// complete their work (eg unmarshaling/Releasing interfaces on
|
|
// object in this thread.
|
|
|
|
WaitForEvent(hEventThreadExit);
|
|
ReleaseEvent(hEventThreadExit);
|
|
|
|
// uninitialize OLE for this thread
|
|
CoUninitialize();
|
|
|
|
DBGOUT("ExitWorkerThread TID:%x hr:%x\n", GetCurrentThreadId(), hr);
|
|
return hr;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: DoIfOperation
|
|
//
|
|
// Synopsis: executes one interface operation
|
|
//
|
|
//--------------------------------------------------------------------
|
|
HRESULT DoIfOperation(DWORD dwFlags, INTERFACEPARAMS *pIP)
|
|
{
|
|
// find the interface pointers and IID in the params
|
|
IUnknown *punk = pIP->punk;
|
|
IStream *pStm = pIP->pStm;
|
|
REFIID riid = pIP->iid;
|
|
HRESULT hr = S_OK;
|
|
|
|
DBGOUT("DoIfOperation Oper:%x pUnk:%x pStm:%x\n", dwFlags, punk, pStm);
|
|
|
|
// wait for the start signal
|
|
if (!WaitForEvent(pIP->hEventStart))
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
// do the requested operation(s) on the interface
|
|
if (dwFlags & OPF_MARSHAL)
|
|
{
|
|
// marshal the interface into the stream
|
|
ResetStream(pStm);
|
|
hr = CoMarshalInterface(pStm, riid, punk, 0, NULL, MSHLFLAGS_NORMAL);
|
|
CHKOP(hr, "CoMarshalInterface");
|
|
}
|
|
|
|
if (dwFlags & OPF_DISCONNECT)
|
|
{
|
|
hr = CoDisconnectObject(punk, 0);
|
|
CHKOP(hr, "CoDisconnectObject");
|
|
}
|
|
|
|
if (dwFlags & OPF_RELEASEMARSHALDATA)
|
|
{
|
|
// call RMD on the stream
|
|
ResetStream(pStm);
|
|
hr = CoReleaseMarshalData(pStm);
|
|
CHKOP(hr, "CoReleaseMarshalData");
|
|
}
|
|
|
|
if (dwFlags & OPF_UNMARSHAL)
|
|
{
|
|
// unmarshal the interface from the stream
|
|
ResetStream(pStm);
|
|
hr = CoUnmarshalInterface(pStm, riid, (void **)&punk);
|
|
CHKOP(hr, "CoUnmarshalInterface");
|
|
}
|
|
|
|
if (dwFlags & OPF_RELEASE)
|
|
{
|
|
// release the interface pointer (if there is one).
|
|
if (punk != NULL)
|
|
{
|
|
ULONG cRefs = punk->Release();
|
|
}
|
|
}
|
|
|
|
SignalEvent(pIP->hEventDone);
|
|
|
|
DBGOUT("DoIfOperation Oper:%x hr:%x\n", dwFlags, hr);
|
|
return hr;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: CreateExecParam
|
|
//
|
|
// Synopsis: allocates an exec parameter packet for the given # of
|
|
// INTERFACEPARAMSs.
|
|
//
|
|
//--------------------------------------------------------------------
|
|
EXECPARAMS *CreateExecParam(ULONG cIP)
|
|
{
|
|
// allocate memory
|
|
ULONG ulSize = sizeof(EXECPARAMS) + (cIP * sizeof(INTERFACEPARAMS));
|
|
|
|
EXECPARAMS *pEP = (EXECPARAMS *) new BYTE[ulSize];
|
|
if (pEP == NULL)
|
|
{
|
|
DBGOUT("CreateExecParams OOM\n");
|
|
return NULL;
|
|
}
|
|
|
|
// zero fill the packet
|
|
memset((BYTE*)pEP, 0, ulSize);
|
|
pEP->cIPs = cIP;
|
|
|
|
DBGOUT("CreateExecParam pEP:%x\n", pEP);
|
|
return pEP;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: FillExecParam
|
|
//
|
|
// Synopsis: fills an exec parameter packet
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void FillExecParam(EXECPARAMS *pEP, DWORD dwFlags, ULONG cReps,
|
|
HANDLE hEventRepStart, HANDLE hEventRepDone,
|
|
HANDLE hEventThreadStart, HANDLE hEventThreadDone)
|
|
{
|
|
DBGOUT("FillExecParam pEP:%x\n", pEP);
|
|
|
|
pEP->dwFlags = dwFlags;
|
|
pEP->hEventThreadStart = hEventThreadStart;
|
|
pEP->hEventThreadDone = hEventThreadDone;
|
|
pEP->hEventThreadExit = GetEvent();
|
|
|
|
pEP->cReps = cReps;
|
|
pEP->hEventRepStart = hEventRepStart;
|
|
pEP->hEventRepDone = hEventRepDone;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: ReleaseExecParam
|
|
//
|
|
// Synopsis: releases an exec parameter packet
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void ReleaseExecParam(EXECPARAMS *pEP)
|
|
{
|
|
DBGOUT("ReleaseExecParam pEP:%x\n", pEP);
|
|
|
|
if (!pEP)
|
|
return;
|
|
|
|
// release the events.
|
|
ReleaseEvent(pEP->hEventThreadStart);
|
|
ReleaseEvent(pEP->hEventThreadDone);
|
|
ReleaseEvent(pEP->hEventThreadExit);
|
|
ReleaseEvent(pEP->hEventRepStart);
|
|
ReleaseEvent(pEP->hEventRepDone);
|
|
|
|
// release the interface parameter blocks
|
|
for (ULONG i=0; i<pEP->cIPs; i++)
|
|
{
|
|
ReleaseInterfaceParam(&(pEP->aIP[i]));
|
|
}
|
|
|
|
// free the memory
|
|
delete pEP;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: FillInterfaceParam
|
|
//
|
|
// Synopsis: fills default info into the interface parms
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void FillInterfaceParam(INTERFACEPARAMS *pIP, REFIID riid, IUnknown *punk,
|
|
IStream *pStm, HANDLE hEventStart, HANDLE hEventDone)
|
|
{
|
|
DBGOUT("FillInterfaceParam pIP:%x\n", pIP);
|
|
|
|
pIP->iid = riid;
|
|
pIP->punk = punk;
|
|
pIP->pStm = pStm;
|
|
pIP->hEventStart = hEventStart;
|
|
pIP->hEventDone = hEventDone;
|
|
}
|
|
|
|
//+-------------------------------------------------------------------
|
|
//
|
|
// Function: ReleaseInterfaceParam
|
|
//
|
|
// Synopsis: releases an interface parameter packet
|
|
//
|
|
//--------------------------------------------------------------------
|
|
void ReleaseInterfaceParam(INTERFACEPARAMS *pIP)
|
|
{
|
|
DBGOUT("ReleaseInterfaceParam pIP:%x\n", pIP);
|
|
|
|
if (!pIP)
|
|
return;
|
|
|
|
// release the interfaces
|
|
ReleaseInterface(pIP->punk);
|
|
ReleaseInterface(pIP->pStm);
|
|
|
|
// release the events
|
|
ReleaseEvent(pIP->hEventStart);
|
|
ReleaseEvent(pIP->hEventDone);
|
|
}
|