760 lines
18 KiB
C++
760 lines
18 KiB
C++
|
/*****************************************************************************\
|
||
|
|
* MODULE: splapi.cpp
|
||
|
|
*
|
||
|
|
* PURPOSE: Implementation of COM interface for BidiSpooler
|
||
|
|
*
|
||
|
|
* Copyright (C) 2000 Microsoft Corporation
|
||
|
|
*
|
||
|
|
* History:
|
||
|
|
*
|
||
|
|
* 03/09/00 Weihai Chen (weihaic) Created
|
||
|
|
*
|
||
|
|
\*****************************************************************************/
|
||
|
|
|
||
|
|
#include "precomp.h"
|
||
|
|
#include "priv.h"
|
||
|
|
|
||
|
|
TBidiSpl::TBidiSpl():
|
||
|
|
m_bValid (FALSE),
|
||
|
|
m_cRef (1),
|
||
|
|
m_CritSec (),
|
||
|
|
m_hPrinter (NULL),
|
||
|
|
m_pfnSendRecvBidiData (NULL),
|
||
|
|
m_pfnRouterFreeBidiResponseContainer (NULL)
|
||
|
|
|
||
|
|
|
||
|
|
{
|
||
|
|
HMODULE hModule = NULL;
|
||
|
|
|
||
|
|
InterlockedIncrement(&g_cComponents) ;
|
||
|
|
|
||
|
|
if (m_CritSec.bValid ()) {
|
||
|
|
|
||
|
|
hModule = GetModuleHandle (_T ("winspool.drv"));
|
||
|
|
|
||
|
|
if (hModule) {
|
||
|
|
|
||
|
|
m_pfnSendRecvBidiData = (PFN_SENDRECVBIDIDATA) GetProcAddress (
|
||
|
|
hModule, MAKEINTRESOURCEA (222));
|
||
|
|
|
||
|
|
m_pfnRouterFreeBidiResponseContainer = (PFN_ROUTERFREEBIDIRESPONSECONTAINER) GetProcAddress (
|
||
|
|
hModule, MAKEINTRESOURCEA (223));
|
||
|
|
|
||
|
|
if (m_pfnSendRecvBidiData && m_pfnRouterFreeBidiResponseContainer) {
|
||
|
|
m_bValid = TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
DBGMSG(DBG_TRACE,("TBidiSpl Created\n"));
|
||
|
|
}
|
||
|
|
|
||
|
|
TBidiSpl::~TBidiSpl()
|
||
|
|
{
|
||
|
|
UnbindDevice ();
|
||
|
|
InterlockedDecrement(&g_cComponents) ;
|
||
|
|
|
||
|
|
DBGMSG(DBG_TRACE,("TBidiSpl Dstroy self\n"));
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
TBidiSpl::QueryInterface (
|
||
|
|
REFIID iid,
|
||
|
|
void** ppv)
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
DBGMSG(DBG_TRACE,("Enter TBidiSpl QI\n"));
|
||
|
|
|
||
|
|
if (iid == IID_IUnknown) {
|
||
|
|
*ppv = static_cast<IBidiSpl*>(this) ;
|
||
|
|
}
|
||
|
|
else if (iid == IID_IBidiSpl) {
|
||
|
|
|
||
|
|
*ppv = static_cast<IBidiSpl*>(this) ;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
*ppv = NULL ;
|
||
|
|
hr = E_NOINTERFACE ;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (*ppv) {
|
||
|
|
reinterpret_cast<IUnknown*>(*ppv)->AddRef() ;
|
||
|
|
}
|
||
|
|
|
||
|
|
DBGMSG(DBG_TRACE,("Leave TBidiSpl QI hr=%x\n", hr));
|
||
|
|
return hr ;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP_ (ULONG)
|
||
|
|
TBidiSpl::AddRef ()
|
||
|
|
{
|
||
|
|
DBGMSG(DBG_TRACE,("Enter TBidiSpl::AddRef ref= %d\n", m_cRef));
|
||
|
|
return InterlockedIncrement(&m_cRef) ;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP_ (ULONG)
|
||
|
|
TBidiSpl::Release ()
|
||
|
|
{
|
||
|
|
DBGMSG(DBG_TRACE,("Enter TBidiSpl::Release ref= %d\n", m_cRef));
|
||
|
|
if (InterlockedDecrement(&m_cRef) == 0)
|
||
|
|
{
|
||
|
|
delete this ;
|
||
|
|
return 0 ;
|
||
|
|
}
|
||
|
|
return m_cRef ;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
TBidiSpl::BindDevice (
|
||
|
|
IN CONST LPCWSTR pszDeviceName,
|
||
|
|
IN CONST DWORD dwAccess)
|
||
|
|
{
|
||
|
|
HRESULT hr (E_FAIL);
|
||
|
|
|
||
|
|
HANDLE hPrinter = NULL;
|
||
|
|
HANDLE hOldPrinter = NULL;
|
||
|
|
PRINTER_DEFAULTS PrinterDefault = {NULL, NULL, 0};
|
||
|
|
|
||
|
|
BOOL bRet;
|
||
|
|
|
||
|
|
if (m_bValid) {
|
||
|
|
|
||
|
|
if (pszDeviceName) {
|
||
|
|
|
||
|
|
if (dwAccess == BIDI_ACCESS_ADMINISTRATOR ) {
|
||
|
|
PrinterDefault.DesiredAccess = PRINTER_ALL_ACCESS;
|
||
|
|
}
|
||
|
|
else if (dwAccess == BIDI_ACCESS_USER) {
|
||
|
|
PrinterDefault.DesiredAccess = PRINTER_ACCESS_USE;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
hr = E_INVALIDARG;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (hr != E_INVALIDARG) {
|
||
|
|
|
||
|
|
bRet = OpenPrinter ((LPWSTR)pszDeviceName, &hPrinter, &PrinterDefault);
|
||
|
|
|
||
|
|
if (bRet) {
|
||
|
|
|
||
|
|
TAutoCriticalSection CritSec (m_CritSec);
|
||
|
|
|
||
|
|
bRet = CritSec.bValid ();
|
||
|
|
if (bRet) {
|
||
|
|
if (m_hPrinter != NULL) {
|
||
|
|
// Opened before
|
||
|
|
|
||
|
|
// Do not cache the handle, since the calling thread may
|
||
|
|
// impersonate different user credentials
|
||
|
|
//
|
||
|
|
|
||
|
|
hOldPrinter = m_hPrinter;
|
||
|
|
}
|
||
|
|
|
||
|
|
m_hPrinter = hPrinter;
|
||
|
|
hPrinter = NULL;
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (hOldPrinter) {
|
||
|
|
ClosePrinter (hOldPrinter);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (hPrinter) {
|
||
|
|
ClosePrinter (hPrinter);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (FAILED (hr)) {
|
||
|
|
hr = LastError2HRESULT ();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = E_INVALIDARG;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = E_HANDLE;
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
TBidiSpl::UnbindDevice ()
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
BOOL bRet;
|
||
|
|
HANDLE hPrinter = NULL;
|
||
|
|
|
||
|
|
if (m_bValid) {
|
||
|
|
|
||
|
|
{
|
||
|
|
TAutoCriticalSection CritSec (m_CritSec);
|
||
|
|
|
||
|
|
bRet = CritSec.bValid ();
|
||
|
|
|
||
|
|
if (bRet) {
|
||
|
|
|
||
|
|
if (hPrinter = m_hPrinter) {
|
||
|
|
m_hPrinter = NULL;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
// Nothing to unbind
|
||
|
|
bRet = FALSE;
|
||
|
|
SetLastError (ERROR_INVALID_HANDLE_STATE);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// Leave Critical Section
|
||
|
|
|
||
|
|
if (hPrinter) {
|
||
|
|
bRet = ClosePrinter (hPrinter);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (bRet) {
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
hr = LastError2HRESULT();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = E_HANDLE;
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
TBidiSpl::SendRecv (
|
||
|
|
IN CONST LPCWSTR pszAction,
|
||
|
|
IN IBidiRequest * pRequest)
|
||
|
|
{
|
||
|
|
IBidiRequestContainer * pIReqContainer = NULL ;
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
hr = ValidateContext();
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
if (!pRequest) {
|
||
|
|
hr = E_POINTER;
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = ::CoCreateInstance(CLSID_BidiRequestContainer,
|
||
|
|
NULL,
|
||
|
|
CLSCTX_INPROC_SERVER,
|
||
|
|
IID_IBidiRequestContainer,
|
||
|
|
(void**)&pIReqContainer) ;
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
hr = pIReqContainer->AddRequest (pRequest);
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
hr = MultiSendRecv (pszAction, pIReqContainer);
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
Failed:
|
||
|
|
|
||
|
|
if (pIReqContainer) {
|
||
|
|
pIReqContainer->Release ();
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
TBidiSpl::MultiSendRecv (
|
||
|
|
IN CONST LPCWSTR pszAction,
|
||
|
|
IN IBidiRequestContainer * pRequestContainer)
|
||
|
|
{
|
||
|
|
|
||
|
|
DWORD dwTotal;
|
||
|
|
DWORD dwRet;
|
||
|
|
IBidiRequestContainer * pIReqContainer = NULL;
|
||
|
|
PBIDI_RESPONSE_CONTAINER pResponse = NULL;
|
||
|
|
TRequestContainer * pReqContainer = NULL;
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
|
||
|
|
hr = ValidateContext();
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
if (!pRequestContainer) {
|
||
|
|
hr = E_POINTER;
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
hr = pRequestContainer->QueryInterface (
|
||
|
|
IID_IBidiRequestContainer,
|
||
|
|
(void **) & pIReqContainer);
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
hr = pIReqContainer->GetRequestCount (&dwTotal);
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
if (dwTotal == 0 && lstrcmpi (BIDI_ACTION_ENUM_SCHEMA, pszAction)) {
|
||
|
|
// There is no request in the container
|
||
|
|
//
|
||
|
|
hr = E_INVALIDARG;
|
||
|
|
DBGMSG (DBG_INFO, ("No request in an action"));
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = ComposeRequestData (pRequestContainer, &pReqContainer);
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
dwRet = (*m_pfnSendRecvBidiData) (m_hPrinter,
|
||
|
|
pszAction,
|
||
|
|
pReqContainer->GetContainerPointer (),
|
||
|
|
&pResponse);
|
||
|
|
|
||
|
|
if (!dwRet && pResponse) {
|
||
|
|
hr = ComposeReponseData (
|
||
|
|
pIReqContainer,
|
||
|
|
(PBIDI_RESPONSE_CONTAINER) pResponse);
|
||
|
|
|
||
|
|
(*m_pfnRouterFreeBidiResponseContainer) (pResponse);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = WinError2HRESULT (dwRet);
|
||
|
|
|
||
|
|
Failed:
|
||
|
|
|
||
|
|
if (pReqContainer)
|
||
|
|
delete pReqContainer;
|
||
|
|
|
||
|
|
if (pIReqContainer)
|
||
|
|
pIReqContainer ->Release ();
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
TBidiSpl::ValidateContext ()
|
||
|
|
{
|
||
|
|
BOOL bRet;
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
if (m_bValid) {
|
||
|
|
TAutoCriticalSection CritSec (m_CritSec);
|
||
|
|
|
||
|
|
bRet = CritSec.bValid ();
|
||
|
|
if (bRet) {
|
||
|
|
if (m_hPrinter == NULL) {
|
||
|
|
// The application has not called BindDevice yet.
|
||
|
|
hr = E_HANDLE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = LastError2HRESULT ();
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hr = E_HANDLE;
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
TBidiSpl::ComposeRequestData (
|
||
|
|
IN IBidiRequestContainer *pIReqContainer,
|
||
|
|
IN TRequestContainer **ppReqContainer)
|
||
|
|
{
|
||
|
|
|
||
|
|
|
||
|
|
DWORD dwFetched;
|
||
|
|
DWORD dwSize;
|
||
|
|
DWORD dwTotal;
|
||
|
|
DWORD dwType;
|
||
|
|
DWORD i;
|
||
|
|
LPWSTR pszSchema;
|
||
|
|
PBYTE pData = NULL;
|
||
|
|
|
||
|
|
IBidiRequestSpl * pISpl = NULL;
|
||
|
|
IEnumUnknown * pEnumIunk = NULL;
|
||
|
|
IUnknown * pIunk = NULL;
|
||
|
|
TRequestContainer * pReqContainer = NULL;
|
||
|
|
|
||
|
|
BOOL bRet;
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
*ppReqContainer = NULL;
|
||
|
|
|
||
|
|
hr = pIReqContainer->GetEnumObject (&pEnumIunk) ;
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
hr = pIReqContainer->GetRequestCount (&dwTotal) ;
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
pReqContainer = new TRequestContainer (dwTotal);
|
||
|
|
*ppReqContainer = pReqContainer;
|
||
|
|
|
||
|
|
if (! (pReqContainer && pReqContainer->bValid ())) {
|
||
|
|
hr = LastError2HRESULT();
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (i = 0; i < dwTotal; i++){
|
||
|
|
|
||
|
|
hr = pEnumIunk->Next (1, &pIunk, &dwFetched);
|
||
|
|
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
if (dwFetched != 1) {
|
||
|
|
hr = E_INVALIDARG;
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = pIunk->QueryInterface (IID_IBidiRequestSpl, (void **) & pISpl);
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
pIunk->Release ();
|
||
|
|
pIunk = NULL;
|
||
|
|
|
||
|
|
// Create the request
|
||
|
|
hr = pISpl->GetSchema (&pszSchema);
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
hr = pISpl->GetInputData (&dwType, &pData, &dwSize);
|
||
|
|
if (FAILED (hr)) goto Failed;
|
||
|
|
|
||
|
|
bRet = pReqContainer->AddRequest (i, pszSchema, (BIDI_TYPE) dwType, pData, dwSize);
|
||
|
|
|
||
|
|
if (!bRet) {
|
||
|
|
hr = LastError2HRESULT();
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
pISpl->Release ();
|
||
|
|
pISpl = NULL;
|
||
|
|
|
||
|
|
}
|
||
|
|
hr = S_OK;
|
||
|
|
|
||
|
|
Failed:
|
||
|
|
|
||
|
|
if (pISpl)
|
||
|
|
pISpl->Release ();
|
||
|
|
|
||
|
|
if (pEnumIunk)
|
||
|
|
pEnumIunk->Release ();
|
||
|
|
|
||
|
|
if (pIunk)
|
||
|
|
pIunk->Release();
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
TBidiSpl::SetData (
|
||
|
|
IN IBidiRequestSpl *pISpl,
|
||
|
|
IN PBIDI_RESPONSE_DATA pResponseData)
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
hr = pISpl->SetResult (pResponseData->dwResult);
|
||
|
|
|
||
|
|
if (!SUCCEEDED (hr)) {
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
PBIDI_DATA pData = & (pResponseData->data);
|
||
|
|
|
||
|
|
switch (pData->dwBidiType) {
|
||
|
|
case BIDI_NULL:
|
||
|
|
break;
|
||
|
|
case BIDI_INT:
|
||
|
|
hr = pISpl->AppendOutputData (pResponseData->pSchema,
|
||
|
|
pData->dwBidiType,
|
||
|
|
(PBYTE) & pData->u.iData,
|
||
|
|
sizeof (ULONG));
|
||
|
|
break;
|
||
|
|
case BIDI_BOOL:
|
||
|
|
hr = pISpl->AppendOutputData (pResponseData->pSchema,
|
||
|
|
pData->dwBidiType,
|
||
|
|
(PBYTE) & pData->u.bData,
|
||
|
|
sizeof (BOOL));
|
||
|
|
break;
|
||
|
|
case BIDI_FLOAT:
|
||
|
|
hr = pISpl->AppendOutputData (pResponseData->pSchema,
|
||
|
|
pData->dwBidiType,
|
||
|
|
(PBYTE) & pData->u.fData,
|
||
|
|
sizeof (FLOAT));
|
||
|
|
break;
|
||
|
|
case BIDI_TEXT:
|
||
|
|
case BIDI_ENUM:
|
||
|
|
case BIDI_STRING:
|
||
|
|
|
||
|
|
if (pData->u.sData)
|
||
|
|
{
|
||
|
|
hr = pISpl->AppendOutputData (pResponseData->pSchema,
|
||
|
|
pData->dwBidiType,
|
||
|
|
(PBYTE) pData->u.sData,
|
||
|
|
sizeof (WCHAR) * (lstrlen (pData->u.sData) + 1));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hr = pISpl->AppendOutputData (pResponseData->pSchema,
|
||
|
|
pData->dwBidiType,
|
||
|
|
NULL,
|
||
|
|
0);
|
||
|
|
}
|
||
|
|
|
||
|
|
break;
|
||
|
|
case BIDI_BLOB:
|
||
|
|
hr = pISpl->AppendOutputData (pResponseData->pSchema,
|
||
|
|
pData->dwBidiType,
|
||
|
|
pData->u.biData.pData,
|
||
|
|
pData->u.biData.cbBuf);
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
hr = E_INVALIDARG;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
TBidiSpl::ComposeReponseData (
|
||
|
|
IN IBidiRequestContainer *pIReqContainer,
|
||
|
|
IN PBIDI_RESPONSE_CONTAINER pResponse)
|
||
|
|
{
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
IEnumUnknown* pEnumIunk = NULL;
|
||
|
|
|
||
|
|
hr = pIReqContainer->GetEnumObject (&pEnumIunk) ;
|
||
|
|
|
||
|
|
if (SUCCEEDED (hr)) {
|
||
|
|
|
||
|
|
hr = pEnumIunk->Reset ();
|
||
|
|
|
||
|
|
if (SUCCEEDED (hr)) {
|
||
|
|
|
||
|
|
DWORD dwReqIndex = 0;
|
||
|
|
BOOL bGotInterface = FALSE;;
|
||
|
|
|
||
|
|
|
||
|
|
for (DWORD i = 0; i < pResponse->Count; i++) {
|
||
|
|
|
||
|
|
PBIDI_RESPONSE_DATA pResponseData = & (pResponse->aData[i]);
|
||
|
|
|
||
|
|
// Locate the request
|
||
|
|
if (dwReqIndex <= pResponse->aData[i].dwReqNumber) {
|
||
|
|
hr = pEnumIunk->Skip (pResponse->aData[i].dwReqNumber - dwReqIndex);
|
||
|
|
dwReqIndex = pResponse->aData[i].dwReqNumber;
|
||
|
|
|
||
|
|
if (FAILED (hr)) {
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
bGotInterface = FALSE;
|
||
|
|
|
||
|
|
}
|
||
|
|
else if (dwReqIndex > pResponse->aData[i].dwReqNumber) {
|
||
|
|
hr = pEnumIunk->Reset ();
|
||
|
|
|
||
|
|
if (FAILED (hr)) {
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = pEnumIunk->Skip (pResponse->aData[i].dwReqNumber);
|
||
|
|
dwReqIndex = pResponse->aData[i].dwReqNumber;
|
||
|
|
|
||
|
|
if (FAILED (hr)) {
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
bGotInterface = FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
IUnknown *pIunk = NULL;
|
||
|
|
|
||
|
|
if (!bGotInterface) {
|
||
|
|
DWORD dwFetched;
|
||
|
|
|
||
|
|
hr = pEnumIunk->Next (1, &pIunk, &dwFetched);
|
||
|
|
dwReqIndex++;
|
||
|
|
|
||
|
|
if (FAILED (hr)) {
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwFetched != 1) {
|
||
|
|
hr = E_INVALIDARG;
|
||
|
|
goto Failed;
|
||
|
|
}
|
||
|
|
|
||
|
|
bGotInterface = TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
IBidiRequestSpl *pISpl;
|
||
|
|
|
||
|
|
hr = pIunk->QueryInterface (IID_IBidiRequestSpl,
|
||
|
|
(void **) & pISpl);
|
||
|
|
pIunk->Release();
|
||
|
|
|
||
|
|
if (SUCCEEDED (hr)) {
|
||
|
|
hr = SetData (pISpl, pResponseData);
|
||
|
|
pISpl->Release ();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
Failed:
|
||
|
|
|
||
|
|
if (pEnumIunk)
|
||
|
|
pEnumIunk->Release ();
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
TBidiSpl::TRequestContainer::TRequestContainer (
|
||
|
|
DWORD dwCount):
|
||
|
|
m_bValid (FALSE)
|
||
|
|
{
|
||
|
|
DWORD dwSize = sizeof (BIDI_REQUEST_CONTAINER) + (dwCount - 1) * sizeof (BIDI_REQUEST_DATA);
|
||
|
|
m_pContainer = (PBIDI_REQUEST_CONTAINER) new BYTE [dwSize];
|
||
|
|
|
||
|
|
if (m_pContainer) {
|
||
|
|
|
||
|
|
ZeroMemory (m_pContainer, dwSize);
|
||
|
|
|
||
|
|
m_pContainer->Version = 1;
|
||
|
|
m_pContainer->Count = dwCount;
|
||
|
|
m_bValid = TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
TBidiSpl::TRequestContainer::~TRequestContainer ()
|
||
|
|
{
|
||
|
|
if (m_pContainer) {
|
||
|
|
PBIDI_DATA pBidiData;
|
||
|
|
|
||
|
|
|
||
|
|
for (DWORD i = 0; i < m_pContainer->Count; i++) {
|
||
|
|
|
||
|
|
pBidiData = & (m_pContainer->aData[i].data);
|
||
|
|
|
||
|
|
switch (pBidiData->dwBidiType) {
|
||
|
|
case BIDI_STRING:
|
||
|
|
case BIDI_TEXT:
|
||
|
|
case BIDI_ENUM:
|
||
|
|
if (pBidiData->u.sData)
|
||
|
|
CoTaskMemFree (pBidiData->u.sData);
|
||
|
|
break;
|
||
|
|
case BIDI_BLOB:
|
||
|
|
if (pBidiData->u.biData.pData) {
|
||
|
|
CoTaskMemFree (pBidiData->u.biData.pData);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
delete [] m_pContainer;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
TBidiSpl::TRequestContainer::AddRequest (
|
||
|
|
IN CONST DWORD dwIndex,
|
||
|
|
IN CONST LPCWSTR pszSchema,
|
||
|
|
IN CONST BIDI_TYPE dwDataType,
|
||
|
|
IN PBYTE pData,
|
||
|
|
IN CONST DWORD dwSize)
|
||
|
|
{
|
||
|
|
BOOL bRet (FALSE);
|
||
|
|
BOOL bFreeData;
|
||
|
|
|
||
|
|
if (m_pContainer) {
|
||
|
|
PBIDI_DATA pBidiData;
|
||
|
|
bFreeData = TRUE;
|
||
|
|
|
||
|
|
m_pContainer->aData[dwIndex].dwReqNumber = dwIndex;
|
||
|
|
m_pContainer->aData[dwIndex].pSchema = (LPWSTR) pszSchema;
|
||
|
|
m_pContainer->aData[dwIndex].data.dwBidiType = dwDataType;
|
||
|
|
pBidiData = & (m_pContainer->aData[dwIndex].data);
|
||
|
|
|
||
|
|
switch (pBidiData->dwBidiType) {
|
||
|
|
case BIDI_NULL:
|
||
|
|
bRet = TRUE;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case BIDI_BOOL:
|
||
|
|
bRet = dwSize == BIDI_BOOL_SIZE;
|
||
|
|
|
||
|
|
if(bRet) {
|
||
|
|
pBidiData->u.iData = *(PBOOL)pData;
|
||
|
|
}
|
||
|
|
|
||
|
|
break;
|
||
|
|
|
||
|
|
case BIDI_INT:
|
||
|
|
bRet = dwSize == BIDI_INT_SIZE;
|
||
|
|
|
||
|
|
if (bRet) {
|
||
|
|
pBidiData->u.iData = *(PINT)pData;
|
||
|
|
bRet = TRUE;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case BIDI_FLOAT:
|
||
|
|
|
||
|
|
bRet = dwSize == BIDI_FLOAT_SIZE;
|
||
|
|
|
||
|
|
if (bRet) {
|
||
|
|
pBidiData->u.iData = *(PINT)pData;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case BIDI_STRING:
|
||
|
|
case BIDI_TEXT:
|
||
|
|
case BIDI_ENUM:
|
||
|
|
pBidiData->u.sData = (LPWSTR) pData;
|
||
|
|
bFreeData = FALSE;
|
||
|
|
bRet = TRUE;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case BIDI_BLOB:
|
||
|
|
pBidiData->u.biData.pData = pData;
|
||
|
|
pBidiData->u.biData.cbBuf = dwSize;
|
||
|
|
bFreeData = FALSE;
|
||
|
|
bRet = TRUE;
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
bRet = FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pData && bFreeData) {
|
||
|
|
CoTaskMemFree (pData);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!bRet) {
|
||
|
|
SetLastError (ERROR_INVALID_PARAMETER);
|
||
|
|
}
|
||
|
|
|
||
|
|
return bRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
|