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.
 
 
 
 
 
 

325 lines
14 KiB

#include "precomp.h"
#pragma hdrstop
#include <simstr.h>
#include <wiadebug.h>
#include "wiadefui.h"
#include "uicommon.h"
#include "wiauiext.h"
#include "resource.h"
extern HINSTANCE g_hInstance;
STDMETHODIMP CWiaDefaultUI::BandedDataCallback(
LONG lReason,
LONG lStatus,
LONG lPercentComplete,
LONG lOffset,
LONG lLength,
LONG lReserved,
LONG lResLength,
PBYTE pbBuffer)
{
WIA_PUSH_FUNCTION((TEXT("CWiaDefaultUI::BandedDataCallback(%08X,%08X,%08X,%08X,%08X,%08X,%08X,%08X)"),lReason,lStatus,lPercentComplete,lOffset,lLength,lReserved,lResLength,pbBuffer));
HRESULT hr = S_OK;
if (m_pSecondaryCallback)
hr = m_pSecondaryCallback->BandedDataCallback(lReason,lStatus,lPercentComplete,lOffset,lLength,lReserved,lResLength,pbBuffer);
WIA_PRINTHRESULT((hr,TEXT("m_pSecondaryCallback->BandedDataCallback returned")));
if (SUCCEEDED(hr) && hr != S_FALSE && lReason == IT_MSG_STATUS)
{
if (m_pWiaProgressDialog)
{
m_pWiaProgressDialog->SetPercentComplete(lPercentComplete);
BOOL bCancelled = FALSE;
if (SUCCEEDED(m_pWiaProgressDialog->Cancelled(&bCancelled)) && bCancelled)
{
hr = S_FALSE;
}
}
}
else if (SUCCEEDED(hr) && hr != S_FALSE && lReason == IT_MSG_DATA)
{
if (m_pWiaProgressDialog)
{
m_pWiaProgressDialog->SetPercentComplete(lPercentComplete);
BOOL bCancelled = FALSE;
if (SUCCEEDED(m_pWiaProgressDialog->Cancelled(&bCancelled)) && bCancelled)
{
hr = S_FALSE;
}
}
}
WIA_PRINTHRESULT((hr,TEXT("CWiaDefaultUI::BandedDataCallback is returning")));
return(hr);
}
STDMETHODIMP CWiaDefaultUI::TransferItemBanded( IWiaItem *pWiaItem, HWND hwndParent, DWORD dwFlags, GUID guidFormat, ULONG ulBufferSize, IWiaDataCallback *pSecondaryCallback )
{
if (!pWiaItem)
{
return E_INVALIDARG;
}
// Create status window, if requested
m_pSecondaryCallback = pSecondaryCallback;
HRESULT hr;
if (0 == (dwFlags&WIA_TRANSFERHELPER_NOPROGRESS))
{
hr = CoCreateInstance( CLSID_WiaDefaultUi, NULL, CLSCTX_INPROC_SERVER, IID_IWiaProgressDialog, (void**)&m_pWiaProgressDialog );
if (SUCCEEDED(hr))
{
m_pWiaProgressDialog->Create( hwndParent, WIA_PROGRESSDLG_NO_ANIM );
m_pWiaProgressDialog->SetTitle(CSimpleStringConvert::WideString(CSimpleString(IDS_TRANSFER_IMAGE_ACQUIRING,g_hInstance)));
m_pWiaProgressDialog->SetMessage(CSimpleStringConvert::WideString(CSimpleString(IDS_TRANSFER_IMAGE_PERCENT,g_hInstance)));
m_pWiaProgressDialog->Show();
}
}
if (m_pWiaProgressDialog || (dwFlags&WIA_TRANSFERHELPER_NOPROGRESS))
{
CComPtr<IWiaDataCallback> pWiaDataCallback;
hr = this->QueryInterface(IID_IWiaDataCallback, (void **)&pWiaDataCallback);
if (SUCCEEDED(hr))
{
CComPtr<IWiaPropertyStorage> pWiaPropertyStorage;
hr = pWiaItem->QueryInterface(IID_IWiaPropertyStorage, (void **)&pWiaPropertyStorage);
if (SUCCEEDED(hr))
{
if (guidFormat == GUID_NULL)
{
CComPtr<IWiaSupportedFormats> pWiaSupportedFormats;
hr = this->QueryInterface(IID_IWiaSupportedFormats, (void **)&pWiaSupportedFormats);
if (SUCCEEDED(hr))
{
hr = pWiaSupportedFormats->Initialize( pWiaItem, TYMED_CALLBACK );
if (SUCCEEDED(hr))
{
hr = GetDefaultClipboardFileFormat( &guidFormat );
}
}
}
if (guidFormat == GUID_NULL)
guidFormat = WiaImgFmt_MEMORYBMP;
PROPSPEC PropSpec[2] = {0};
PROPVARIANT PropVariant[2] = {0};
PropSpec[0].ulKind = PRSPEC_PROPID;
PropSpec[0].propid = WIA_IPA_FORMAT;
PropSpec[1].ulKind = PRSPEC_PROPID;
PropSpec[1].propid = WIA_IPA_TYMED;
PropVariant[0].vt = VT_CLSID;
PropVariant[0].puuid = &guidFormat;
PropVariant[1].vt = VT_I4;
PropVariant[1].lVal = TYMED_CALLBACK;
hr = pWiaPropertyStorage->WriteMultiple( ARRAYSIZE(PropVariant), PropSpec, PropVariant, WIA_IPA_FIRST );
if (SUCCEEDED(hr))
{
CComPtr<IWiaDataTransfer> pIWiaDataTransfer;
hr = pWiaItem->QueryInterface(IID_IWiaDataTransfer, (void **)&pIWiaDataTransfer);
if (SUCCEEDED(hr))
{
if (!ulBufferSize)
{
LONG lMinimumBufferSize;
if (!PropStorageHelpers::GetProperty( pWiaItem, WIA_IPA_MIN_BUFFER_SIZE, lMinimumBufferSize ))
{
lMinimumBufferSize = 0x0000FFFF; // 64K
}
ulBufferSize = static_cast<ULONG>(lMinimumBufferSize);
}
WIA_DATA_TRANSFER_INFO WiaDataTransferInfo = {0};
WiaDataTransferInfo.ulSize = sizeof(WIA_DATA_TRANSFER_INFO);
WiaDataTransferInfo.ulBufferSize = ulBufferSize;
hr = pIWiaDataTransfer->idtGetBandedData(&WiaDataTransferInfo, pWiaDataCallback);
WIA_PRINTHRESULT((hr,TEXT("pIWiaDataTransfer->idtGetBandedData returned")));
}
else
{
WIA_PRINTHRESULT((hr,TEXT("TransferItemBanded, QI on IID_IWiaDataTransfer")));
}
}
else
{
WIA_PRINTHRESULT((hr,TEXT("TransferItemBanded, pIWiaPropertyStorage->WriteMultiple failed")));
}
}
else
{
WIA_PRINTHRESULT((hr,TEXT("TransferItemBanded, QI of IID_IWiaPropertyStorage failed.")));
}
}
if (m_pWiaProgressDialog)
{
m_pWiaProgressDialog->Destroy();
m_pWiaProgressDialog = NULL;
}
}
else
{
WIA_ERROR((TEXT("TransferItemBanded, unable to create status window")));
hr = HRESULT_FROM_WIN32(GetLastError());
}
return hr;
}
STDMETHODIMP CWiaDefaultUI::TransferItemFile( IWiaItem *pWiaItem, HWND hwndParent, DWORD dwFlags, GUID guidFormat, LPCWSTR pszFilename, IWiaDataCallback *pSecondaryCallback, LONG nMediaType )
{
WIA_PUSH_FUNCTION((TEXT("CWiaDefaultUI::TransferItemFile( %08X )"), pWiaItem ));
if (!pszFilename || !lstrlenW(pszFilename) || !pWiaItem)
{
return E_INVALIDARG;
}
//
// Create status window, if requested
//
m_pSecondaryCallback = pSecondaryCallback;
HRESULT hr;
if (0 == (dwFlags&WIA_TRANSFERHELPER_NOPROGRESS))
{
hr = CoCreateInstance( CLSID_WiaDefaultUi, NULL, CLSCTX_INPROC_SERVER, IID_IWiaProgressDialog, (void**)&m_pWiaProgressDialog );
if (SUCCEEDED(hr))
{
m_pWiaProgressDialog->Create( hwndParent, WIA_PROGRESSDLG_NO_ANIM );
m_pWiaProgressDialog->SetTitle(CSimpleStringConvert::WideString(CSimpleString(IDS_TRANSFER_IMAGE_ACQUIRING,g_hInstance)));
m_pWiaProgressDialog->SetMessage(CSimpleStringConvert::WideString(CSimpleString(IDS_TRANSFER_IMAGE_PERCENT,g_hInstance)));
m_pWiaProgressDialog->Show();
}
}
if (m_pWiaProgressDialog || (dwFlags&WIA_TRANSFERHELPER_NOPROGRESS))
{
CComPtr<IWiaDataCallback> pWiaDataCallback;
hr = this->QueryInterface(IID_IWiaDataCallback, (void **)&pWiaDataCallback);
if (SUCCEEDED(hr))
{
CComPtr<IWiaPropertyStorage> pWiaPropertyStorage;
hr = pWiaItem->QueryInterface(IID_IWiaPropertyStorage, (void **)&pWiaPropertyStorage);
if (SUCCEEDED(hr))
{
if (guidFormat == GUID_NULL)
{
CComPtr<IWiaSupportedFormats> pWiaSupportedFormats;
hr = this->QueryInterface(IID_IWiaSupportedFormats, (void **)&pWiaSupportedFormats);
if (SUCCEEDED(hr))
{
hr = pWiaSupportedFormats->Initialize( pWiaItem, nMediaType );
if (SUCCEEDED(hr))
{
hr = GetDefaultClipboardFileFormat( &guidFormat );
}
}
}
if (guidFormat == GUID_NULL)
guidFormat = WiaImgFmt_BMP;
PROPSPEC PropSpec[2] = {0};
PROPVARIANT PropVariant[2] = {0};
PropSpec[0].ulKind = PRSPEC_PROPID;
PropSpec[0].propid = WIA_IPA_FORMAT;
PropSpec[1].ulKind = PRSPEC_PROPID;
PropSpec[1].propid = WIA_IPA_TYMED;
PropVariant[0].vt = VT_CLSID;
PropVariant[0].puuid = &guidFormat;
PropVariant[1].vt = VT_I4;
PropVariant[1].lVal = nMediaType;
WIA_PRINTGUID((guidFormat,TEXT("TYMED: %d, guidFormat"), nMediaType ));
hr = pWiaPropertyStorage->WriteMultiple( ARRAYSIZE(PropVariant), PropSpec, PropVariant, WIA_IPA_FIRST );
if (SUCCEEDED(hr))
{
CComPtr<IWiaDataTransfer> pIWiaDataTransfer;
hr = pWiaItem->QueryInterface(IID_IWiaDataTransfer, (void **)&pIWiaDataTransfer);
if (SUCCEEDED(hr))
{
STGMEDIUM stgMedium = {0};
hr = pIWiaDataTransfer->idtGetData(&stgMedium, pWiaDataCallback);
WIA_PRINTHRESULT((hr,TEXT("pIWiaDataTransfer->idtGetData returned")));
//
// If the transfer succeeded, or it failed with the possibility of having a file
// left over AND we want to preserve any leftover file.
//
if (SUCCEEDED(hr))
{
//
// Make sure we didn't cancel.
//
if (hr != S_FALSE)
{
//
// Get the file names
//
CSimpleString strName = CSimpleStringConvert::NaturalString(CSimpleStringWide(stgMedium.lpszFileName));
CSimpleString strTarget = CSimpleStringConvert::NaturalString(CSimpleStringWide(pszFilename));
if (strName.Length() && strTarget.Length())
{
//
// This function will first try to move, then copy+erase the src file to the tgt file
//
hr = WiaUiUtil::MoveOrCopyFile( strName.String(), strTarget.String() );
}
else
{
hr = E_OUTOFMEMORY;
WIA_ERROR((TEXT("TransferItemFile, unable to create source filename")));
}
}
else
{
WIA_ERROR((TEXT("TransferItemFile, user cancelled")));
}
}
else if ((WIA_ERROR_PAPER_JAM == hr || WIA_ERROR_PAPER_EMPTY == hr || WIA_ERROR_PAPER_PROBLEM == hr) && (dwFlags & WIA_TRANSFERHELPER_PRESERVEFAILEDFILE))
{
//
// Get the file names
//
CSimpleString strName = CSimpleStringConvert::NaturalString(CSimpleStringWide(stgMedium.lpszFileName));
CSimpleString strTarget = CSimpleStringConvert::NaturalString(CSimpleStringWide(pszFilename));
if (strName.Length() && strTarget.Length())
{
//
// Don't preserve any errors we encounter here.
//
WiaUiUtil::MoveOrCopyFile( strName.String(), strTarget.String() );
}
}
else
{
WIA_PRINTHRESULT((hr,TEXT("TransferItemFile, idtGetData failed")));
}
//
// Call this even in the error case, in case the driver/service left a file laying around.
//
ReleaseStgMedium(&stgMedium);
}
}
else
{
WIA_PRINTHRESULT((hr,TEXT("TransferItemFile, pIWiaPropertyStorage->WriteMultiple failed")));
}
}
else
{
WIA_PRINTHRESULT((hr,TEXT("TransferItemFile, QI of IWiaPropertyStorage failed.")));
}
}
if (m_pWiaProgressDialog)
{
m_pWiaProgressDialog->Destroy();
m_pWiaProgressDialog = NULL;
}
}
else
{
WIA_ERROR((TEXT("TransferItemFile, unable to create status window")));
hr = HRESULT_FROM_WIN32(GetLastError());
}
return hr;
}