Файл WPD портативного устройства Win32 в локальный каталог. Потоковое копирование не выполняется для некоторых файлов. ⇐ C++

[Anonymous]

Я пытаюсь скопировать файл портативного устройства WPD в локальную папку C:\. Я получил этот код от Google и несколько примеров MSDN. Он работает правильно с большинством файлов. Однако для других я получаю исключение:

Код: Выделить всё

Read access violation: pDestStream was nullptr

онлайн:

Код: Выделить всё

hr = pDestStream->Write(pObjectData, cbBytesRead, &cbBytesWritten);

Обратите внимание, что этот код находится в служебной функцииstreamCopy, которую я получил из некоторых примеров, также показанных ниже.
Внешний код:

Код: Выделить всё

std::wstring destinationFilename = ...; // Where to copy to
std::wstring specialDeviceObjectId = ...; // Object ID on WPD device

CComPtr pResources;
CComPtr pObjectDataStream;
CComPtr pFinalFileStream;
DWORD cbOptimalTransferSize = 0;

HRESULT hrResources = pContent->Transfer(&pResources);
// Get the stream from the device
HRESULT hrGetStream = pResources->GetStream(
specialDeviceObjectId.c_str(),
WPD_RESOURCE_DEFAULT, // Transfer the entire data
STGM_READ,            // Open in READ mode
&cbOptimalTransferSize,
&pObjectDataStream);
// Create the local destination file stream
HRESULT hrCreateStreamOnFile = SHCreateStreamOnFile(
destinationFilename.c_str(),
STGM_CREATE | STGM_WRITE,
&pFinalFileStream);
// Copy the data
DWORD cbTotalBytesWritten = 0;
// Call utility function StreamCopy
HRESULT hrStreamCopy = StreamCopy(pFinalFileStream,
pObjectDataStream,
cbOptimalTransferSize,
&cbTotalBytesWritten);

Вспомогательная функция StreamCopy

Код: Выделить всё

// Copies data from a source stream to a destination stream using the
// specified cbTransferSize as the temporary buffer size.
HRESULT StreamCopy(
IStream* pDestStream,
IStream* pSourceStream,
DWORD       cbTransferSize,
DWORD* pcbWritten)
{

HRESULT hr = S_OK;

// Allocate a temporary buffer (of Optimal transfer size) for the read results to
// be written to.
BYTE* pObjectData = new (std::nothrow) BYTE[cbTransferSize];
if (pObjectData != NULL)
{
DWORD cbTotalBytesRead = 0;
DWORD cbTotalBytesWritten = 0;

DWORD cbBytesRead = 0;
DWORD cbBytesWritten = 0;

// Read until the number of bytes returned from the source stream is 0, or
// an error occured during transfer.
do
{
// Read object data from the source stream
hr = pSourceStream->Read(pObjectData, cbTransferSize, &cbBytesRead);
if (FAILED(hr))
{
printf("! Failed to read %d bytes from the source stream, hr = 0x%lx\n", cbTransferSize, hr);
}

// Write object data to the destination stream
if (SUCCEEDED(hr))
{
cbTotalBytesRead += cbBytesRead; // Calculating total bytes read from device for debugging purposes only

// THIS IS WHERE THE ERROR SOMETIMES OCCURS
// pDestStream->Write: pDestStream was nullptr
hr = pDestStream->Write(pObjectData, cbBytesRead, &cbBytesWritten);
if (FAILED(hr))
{
printf("! Failed to write %d bytes of object data to the destination stream, hr = 0x%lx\n", cbBytesRead, hr);
}

if (SUCCEEDED(hr))
{
cbTotalBytesWritten += cbBytesWritten; // Calculating total bytes written to the file for debugging purposes only
}
}

// Output Read/Write operation information only if we have received data and if no error has occured so far.
if (SUCCEEDED(hr) && (cbBytesRead > 0))
{
printf("Read %d bytes from the source stream...Wrote %d bytes to the destination stream...\n", cbBytesRead, cbBytesWritten);
}

} while (SUCCEEDED(hr) && (cbBytesRead > 0));

// If the caller supplied a pcbWritten parameter and we
// and we are successful, set it to cbTotalBytesWritten
// before exiting.
if ((SUCCEEDED(hr)) &&  (pcbWritten != NULL))
{
*pcbWritten = cbTotalBytesWritten;
}

// Remember to delete the temporary transfer buffer
delete[] pObjectData;
pObjectData = NULL;
}
else
{
printf("! Failed to allocate %d bytes for the temporary transfer buffer.\n", cbTransferSize);
}

return hr;
}

ПРИМЕЧАНИЕ. Обертывание этой pDestStream->Write внутри проверки if (pDestStream != nullptr) устранило проблему. Но мне понятно, почему иногда возникает эта ошибка.

Подробнее здесь: https://stackoverflow.com/questions/799 ... -some-file