150 lines
3.5 KiB
C++
150 lines
3.5 KiB
C++
//****************************************************************************
|
|
//
|
|
// Module: ULS.DLL
|
|
// File: request.cpp
|
|
// Content: CReqMgr and CRequest classes implementation
|
|
//
|
|
// Copyright (c) Microsoft Corporation 1996-1997
|
|
//
|
|
//****************************************************************************
|
|
|
|
#include "ulsp.h"
|
|
#include "request.h"
|
|
|
|
|
|
//****************************************************************************
|
|
// CReqMgr::CReqMgr(void)
|
|
//
|
|
// Purpose: Constructor for the CReqMgr class
|
|
//
|
|
// Parameters: None
|
|
//****************************************************************************
|
|
|
|
CReqMgr::CReqMgr(void)
|
|
{
|
|
uNextReqID = REQUEST_ID_INIT;
|
|
return;
|
|
}
|
|
|
|
//****************************************************************************
|
|
// CReqMgr::~CReqMgr(void)
|
|
//
|
|
// Purpose: Destructor for the CReqMgr class
|
|
//
|
|
// Parameters: None
|
|
//****************************************************************************
|
|
|
|
CReqMgr::~CReqMgr(void)
|
|
{
|
|
COM_REQ_INFO *pRequest;
|
|
HANDLE hEnum;
|
|
|
|
// Free all the pending request
|
|
//
|
|
ReqList.Enumerate(&hEnum);
|
|
while (ReqList.Next(&hEnum, (LPVOID *)&pRequest) == NOERROR)
|
|
{
|
|
::MemFree (pRequest);
|
|
}
|
|
ReqList.Flush();
|
|
return;
|
|
}
|
|
|
|
//****************************************************************************
|
|
// HRESULT
|
|
// CReqMgr::NewRequest (COM_REQ_INFO *pri)
|
|
//
|
|
// Purpose: Add a new pending request
|
|
//
|
|
// Parameters: None
|
|
//****************************************************************************
|
|
|
|
HRESULT
|
|
CReqMgr::NewRequest (COM_REQ_INFO *pri)
|
|
{
|
|
COM_REQ_INFO *pRequest;
|
|
HRESULT hr;
|
|
|
|
// Allocate a new request node
|
|
//
|
|
pri->uReqID = uNextReqID;
|
|
pRequest = (COM_REQ_INFO *) ::MemAlloc (sizeof (COM_REQ_INFO));
|
|
if (pRequest == NULL)
|
|
{
|
|
return ILS_E_MEMORY;
|
|
};
|
|
*pRequest = *pri;
|
|
|
|
// Append the new request to the list
|
|
//
|
|
hr = ReqList.Append((PVOID)pRequest);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
delete pRequest;
|
|
}
|
|
else
|
|
{
|
|
if (++uNextReqID == ILS_INVALID_REQ_ID)
|
|
uNextReqID = REQUEST_ID_INIT;
|
|
};
|
|
return hr;
|
|
}
|
|
|
|
//****************************************************************************
|
|
// HRESULT
|
|
// CReqMgr::FindRequest (COM_REQ_INFO *pri, BOOL fRemove)
|
|
//
|
|
// Purpose: Find the request.
|
|
//
|
|
// Parameters: None
|
|
//****************************************************************************
|
|
|
|
HRESULT
|
|
CReqMgr::FindRequest (COM_REQ_INFO *pri, BOOL fRemove)
|
|
{
|
|
COM_REQ_INFO *pRequest;
|
|
ULONG uMatchingID;
|
|
HANDLE hEnum;
|
|
HRESULT hr;
|
|
|
|
// Get the ID we want to match
|
|
//
|
|
uMatchingID = (pri->uMsgID != 0 ? pri->uMsgID : pri->uReqID);
|
|
|
|
// Look for the request matching the message ID
|
|
//
|
|
ReqList.Enumerate(&hEnum);
|
|
while (ReqList.Next(&hEnum, (PVOID *)&pRequest) == NOERROR)
|
|
{
|
|
if (uMatchingID == (pri->uMsgID != 0 ? pRequest->uMsgID :
|
|
pRequest->uReqID))
|
|
{
|
|
break;
|
|
};
|
|
};
|
|
|
|
if (pRequest != NULL)
|
|
{
|
|
// Return the request associates
|
|
//
|
|
*pri = *pRequest;
|
|
|
|
// Remove the request
|
|
//
|
|
if (fRemove)
|
|
{
|
|
ReqList.Remove((PVOID)pRequest);
|
|
::MemFree (pRequest);
|
|
};
|
|
hr = NOERROR;
|
|
}
|
|
else
|
|
{
|
|
hr = ILS_E_FAIL;
|
|
};
|
|
|
|
return hr;
|
|
}
|
|
|