Leaked source code of windows server 2003
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.
 
 
 
 
 
 

759 lines
18 KiB

/*****************************************************************************\
* 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;
}