stg_stream.c File Reference

#include <stdlib.h>
#include <stdarg.h>
#include <stdio.h>
#include <string.h>
#include "windef.h"
#include "winbase.h"
#include "winerror.h"
#include "winternl.h"
#include "wine/debug.h"
#include "storage32.h"

Include dependency graph for stg_stream.c:

Go to the source code of this file.

Macros
#define	COBJMACROS
#define	NONAMELESSUNION

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (storage)
static HRESULT WINAPI	StgStreamImpl_QueryInterface (IStream *iface, REFIID riid, void **ppvObject)
static ULONG WINAPI	StgStreamImpl_AddRef (IStream *iface)
static ULONG WINAPI	StgStreamImpl_Release (IStream *iface)
static HRESULT WINAPI	StgStreamImpl_Read (IStream *iface, void *pv, ULONG cb, ULONG *pcbRead)
static HRESULT WINAPI	StgStreamImpl_Write (IStream *iface, const void *pv, ULONG cb, ULONG *pcbWritten)
static HRESULT WINAPI	StgStreamImpl_Seek (IStream *iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
static HRESULT WINAPI	StgStreamImpl_SetSize (IStream *iface, ULARGE_INTEGER libNewSize)
static HRESULT WINAPI	StgStreamImpl_CopyTo (IStream *iface, IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
static HRESULT WINAPI	StgStreamImpl_Commit (IStream *iface, DWORD grfCommitFlags)
static HRESULT WINAPI	StgStreamImpl_Revert (IStream *iface)
static HRESULT WINAPI	StgStreamImpl_LockRegion (IStream *iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
static HRESULT WINAPI	StgStreamImpl_UnlockRegion (IStream *iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
static HRESULT WINAPI	StgStreamImpl_Stat (IStream *iface, STATSTG *pstatstg, DWORD grfStatFlag)
static HRESULT WINAPI	StgStreamImpl_Clone (IStream *iface, IStream **ppstm)
StgStreamImpl *	StgStreamImpl_Construct (StorageBaseImpl *parentStorage, DWORD grfMode, DirRef dirEntry)

Variables
static const IStreamVtbl	StgStreamVtbl

Macro Definition Documentation

COBJMACROS

#define COBJMACROS

Definition at line 31 of file stg_stream.c.

NONAMELESSUNION

#define NONAMELESSUNION

Definition at line 32 of file stg_stream.c.

Function Documentation

StgStreamImpl_AddRef()

static ULONG WINAPI StgStreamImpl_AddRef ( IStream *  iface )

static

Definition at line 78 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
  return InterlockedIncrement(&This->ref);
}

StgStreamImpl_Clone()

static HRESULT WINAPI StgStreamImpl_Clone ( IStream *  iface, IStream **  ppstm  )

static

Definition at line 601 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
  StgStreamImpl* new_stream;
  LARGE_INTEGER seek_pos;
 
  TRACE("%p %p\n", This, ppstm);
 
  /*
   * Sanity check
   */
 
  if (!This->parentStorage)
    return STG_E_REVERTED;
 
  if ( ppstm == 0 )
    return STG_E_INVALIDPOINTER;
 
  new_stream = StgStreamImpl_Construct (This->parentStorage, This->grfMode, This->dirEntry);
 
  if (!new_stream)
    return STG_E_INSUFFICIENTMEMORY; /* Currently the only reason for new_stream=0 */
 
  *ppstm = &new_stream->IStream_iface;
  IStream_AddRef(*ppstm);
 
  seek_pos.QuadPart = This->currentPosition.QuadPart;
 
  return IStream_Seek(*ppstm, seek_pos, STREAM_SEEK_SET, NULL);
}

StgStreamImpl_Commit()

static HRESULT WINAPI StgStreamImpl_Commit ( IStream *  iface, DWORD  grfCommitFlags  )

static

Definition at line 467 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
 
  if (!This->parentStorage)
  {
    WARN("storage reverted\n");
    return STG_E_REVERTED;
  }
 
  return StorageBaseImpl_Flush(This->parentStorage);
}

StgStreamImpl_Construct()

StgStreamImpl * StgStreamImpl_Construct	(	StorageBaseImpl *	parentStorage,
		DWORD	grfMode,
		DirRef	dirEntry
	)

Definition at line 666 of file stg_stream.c.

{
  StgStreamImpl* newStream;
 
  newStream = HeapAlloc(GetProcessHeap(), 0, sizeof(StgStreamImpl));
 
  if (newStream)
  {
    /*
     * Set-up the virtual function table and reference count.
     */
    newStream->IStream_iface.lpVtbl = &StgStreamVtbl;
    newStream->ref       = 0;
 
    newStream->parentStorage = parentStorage;
 
    /*
     * We want to nail-down the reference to the storage in case the
     * stream out-lives the storage in the client application.
     *
     * -- IStorage_AddRef(&newStream->parentStorage->IStorage_iface);
     *
     * No, don't do this. Some apps call IStorage_Release without
     * calling IStream_Release first. If we grab a reference the
     * file is not closed, and the app fails when it tries to
     * reopen the file (Easy-PC, for example)
     */
 
    newStream->grfMode = grfMode;
    newStream->dirEntry = dirEntry;
 
    /*
     * Start the stream at the beginning.
     */
    newStream->currentPosition.u.HighPart = 0;
    newStream->currentPosition.u.LowPart = 0;
 
    /* add us to the storage's list of active streams */
    StorageBaseImpl_AddStream(parentStorage, newStream);
  }
 
  return newStream;
}

Referenced by StgStreamImpl_Clone(), StorageBaseImpl_CopyChildEntryTo(), StorageBaseImpl_CreateStream(), and StorageBaseImpl_OpenStream().

StgStreamImpl_CopyTo()

static HRESULT WINAPI StgStreamImpl_CopyTo ( IStream *  iface, IStream *  pstm, ULARGE_INTEGER  cb, ULARGE_INTEGER *  pcbRead, ULARGE_INTEGER *  pcbWritten  )

static

Definition at line 389 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
  HRESULT        hr = S_OK;
  BYTE           tmpBuffer[128];
  ULONG          bytesRead, bytesWritten, copySize;
  ULARGE_INTEGER totalBytesRead;
  ULARGE_INTEGER totalBytesWritten;
 
  TRACE("(%p, %p, %d, %p, %p)\n",
    iface, pstm, cb.u.LowPart, pcbRead, pcbWritten);
 
  /*
   * Sanity check
   */
 
  if (!This->parentStorage)
  {
    WARN("storage reverted\n");
    return STG_E_REVERTED;
  }
 
  if ( pstm == 0 )
    return STG_E_INVALIDPOINTER;
 
  totalBytesRead.QuadPart = 0;
  totalBytesWritten.QuadPart = 0;
 
  while ( cb.QuadPart > 0 )
  {
    if ( cb.QuadPart >= sizeof(tmpBuffer) )
      copySize = sizeof(tmpBuffer);
    else
      copySize = cb.u.LowPart;
 
    IStream_Read(iface, tmpBuffer, copySize, &bytesRead);
 
    totalBytesRead.QuadPart += bytesRead;
 
    IStream_Write(pstm, tmpBuffer, bytesRead, &bytesWritten);
 
    totalBytesWritten.QuadPart += bytesWritten;
 
    /*
     * Check that read & write operations were successful
     */
    if (bytesRead != bytesWritten)
    {
      hr = STG_E_MEDIUMFULL;
      WARN("medium full\n");
      break;
    }
 
    if (bytesRead!=copySize)
      cb.QuadPart = 0;
    else
      cb.QuadPart -= bytesRead;
  }
 
  if (pcbRead) pcbRead->QuadPart = totalBytesRead.QuadPart;
  if (pcbWritten) pcbWritten->QuadPart = totalBytesWritten.QuadPart;
 
  return hr;
}

StgStreamImpl_LockRegion()

static HRESULT WINAPI StgStreamImpl_LockRegion ( IStream *  iface, ULARGE_INTEGER  libOffset, ULARGE_INTEGER  cb, DWORD  dwLockType  )

static

Definition at line 496 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
 
  if (!This->parentStorage)
  {
    WARN("storage reverted\n");
    return STG_E_REVERTED;
  }
 
  FIXME("not implemented!\n");
  return E_NOTIMPL;
}

StgStreamImpl_QueryInterface()

static HRESULT WINAPI StgStreamImpl_QueryInterface ( IStream *  iface, REFIID  riid, void **  ppvObject  )

static

Definition at line 48 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
 
  if (ppvObject==0)
    return E_INVALIDARG;
 
  *ppvObject = 0;
 
  if (IsEqualIID(&IID_IUnknown, riid) ||
      IsEqualIID(&IID_ISequentialStream, riid) ||
      IsEqualIID(&IID_IStream, riid))
  {
    *ppvObject = &This->IStream_iface;
  }
  else
    return E_NOINTERFACE;
 
  IStream_AddRef(iface);
 
  return S_OK;
}

StgStreamImpl_Read()

static HRESULT WINAPI StgStreamImpl_Read ( IStream *  iface, void *  pv, ULONG  cb, ULONG *  pcbRead  )

static

Definition at line 128 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
 
  ULONG bytesReadBuffer;
  HRESULT res;
 
  TRACE("(%p, %p, %d, %p)\n",
    iface, pv, cb, pcbRead);
 
  if (!This->parentStorage)
  {
    WARN("storage reverted\n");
    return STG_E_REVERTED;
  }
 
  /*
   * If the caller is not interested in the number of bytes read,
   * we use another buffer to avoid "if" statements in the code.
   */
  if (pcbRead==0)
    pcbRead = &bytesReadBuffer;
 
  res = StorageBaseImpl_StreamReadAt(This->parentStorage,
                                     This->dirEntry,
                                     This->currentPosition,
                                     cb,
                                     pv,
                                     pcbRead);
 
  if (SUCCEEDED(res))
  {
    /*
     * Advance the pointer for the number of positions read.
     */
    This->currentPosition.QuadPart += *pcbRead;
  }
 
  TRACE("<-- %08x\n", res);
  return res;
}

StgStreamImpl_Release()

static ULONG WINAPI StgStreamImpl_Release ( IStream *  iface )

static

Definition at line 89 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
  ULONG ref = InterlockedDecrement(&This->ref);
 
  if (!ref)
  {
    TRACE("(%p)\n", This);
 
    /*
     * Release the reference we are holding on the parent storage.
     * IStorage_Release(&This->parentStorage->IStorage_iface);
     *
     * No, don't do this. Some apps call IStorage_Release without
     * calling IStream_Release first. If we grab a reference the
     * file is not closed, and the app fails when it tries to
     * reopen the file (Easy-PC, for example). Just inform the
     * storage that we have closed the stream
     */
 
    if (This->parentStorage)
      StorageBaseImpl_RemoveStream(This->parentStorage, This);
    This->parentStorage = 0;
    HeapFree(GetProcessHeap(), 0, This);
  }
 
  return ref;
}

StgStreamImpl_Revert()

static HRESULT WINAPI StgStreamImpl_Revert ( IStream *  iface )

static

Definition at line 490 of file stg_stream.c.

{
  return S_OK;
}

StgStreamImpl_Seek()

static HRESULT WINAPI StgStreamImpl_Seek ( IStream *  iface, LARGE_INTEGER  dlibMove, DWORD  dwOrigin, ULARGE_INTEGER *  plibNewPosition  )

static

Definition at line 265 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
 
  ULARGE_INTEGER newPosition;
  DirEntry currentEntry;
  HRESULT hr;
 
  TRACE("(%p, %d, %d, %p)\n",
    iface, dlibMove.u.LowPart, dwOrigin, plibNewPosition);
 
  /*
   * fail if the stream has no parent (as does windows)
   */
 
  if (!This->parentStorage)
  {
    WARN("storage reverted\n");
    return STG_E_REVERTED;
  }
 
  /*
   * The caller is allowed to pass in NULL as the new position return value.
   * If it happens, we assign it to a dynamic variable to avoid special cases
   * in the code below.
   */
  if (plibNewPosition == 0)
  {
    plibNewPosition = &newPosition;
  }
 
  /*
   * The file pointer is moved depending on the given "function"
   * parameter.
   */
  switch (dwOrigin)
  {
    case STREAM_SEEK_SET:
      plibNewPosition->u.HighPart = 0;
      plibNewPosition->u.LowPart  = 0;
      break;
    case STREAM_SEEK_CUR:
      *plibNewPosition = This->currentPosition;
      break;
    case STREAM_SEEK_END:
      hr = StorageBaseImpl_ReadDirEntry(This->parentStorage, This->dirEntry, &currentEntry);
      if (FAILED(hr)) return hr;
      *plibNewPosition = currentEntry.size;
      break;
    default:
      WARN("invalid dwOrigin %d\n", dwOrigin);
      return STG_E_INVALIDFUNCTION;
  }
 
  plibNewPosition->QuadPart += dlibMove.QuadPart;
 
  /*
   * tell the caller what we calculated
   */
  This->currentPosition = *plibNewPosition;
 
  return S_OK;
}

StgStreamImpl_SetSize()

static HRESULT WINAPI StgStreamImpl_SetSize ( IStream *  iface, ULARGE_INTEGER  libNewSize  )

static

Definition at line 340 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
 
  HRESULT      hr;
 
  TRACE("(%p, %d)\n", iface, libNewSize.u.LowPart);
 
  if(!This->parentStorage)
  {
    WARN("storage reverted\n");
    return STG_E_REVERTED;
  }
 
  /*
   * As documented.
   */
  if (libNewSize.u.HighPart != 0)
  {
    WARN("invalid value for libNewSize.u.HighPart %d\n", libNewSize.u.HighPart);
    return STG_E_INVALIDFUNCTION;
  }
 
  /*
   * Do we have permission?
   */
  if (!(This->grfMode & (STGM_WRITE | STGM_READWRITE)))
  {
    WARN("access denied\n");
    return STG_E_ACCESSDENIED;
  }
 
  hr = StorageBaseImpl_StreamSetSize(This->parentStorage, This->dirEntry, libNewSize);
 
  if (SUCCEEDED(hr))
    hr = StorageBaseImpl_Flush(This->parentStorage);
 
  return hr;
}

StgStreamImpl_Stat()

static HRESULT WINAPI StgStreamImpl_Stat ( IStream *  iface, STATSTG *  pstatstg, DWORD  grfStatFlag  )

static

Definition at line 540 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
 
  DirEntry     currentEntry;
  HRESULT      hr;
 
  TRACE("%p %p %d\n", This, pstatstg, grfStatFlag);
 
  /*
   * if stream has no parent, return STG_E_REVERTED
   */
 
  if (!This->parentStorage)
  {
    WARN("storage reverted\n");
    return STG_E_REVERTED;
  }
 
  /*
   * Read the information from the directory entry.
   */
  hr = StorageBaseImpl_ReadDirEntry(This->parentStorage,
                         This->dirEntry,
                         &currentEntry);
 
  if (SUCCEEDED(hr))
  {
    StorageUtl_CopyDirEntryToSTATSTG(This->parentStorage,
                     pstatstg,
                     &currentEntry,
                     grfStatFlag);
 
    pstatstg->grfMode = This->grfMode;
 
    /* In simple create mode cbSize is the current pos */
    if((This->parentStorage->openFlags & STGM_SIMPLE) && This->parentStorage->create)
      pstatstg->cbSize = This->currentPosition;
 
    return S_OK;
  }
 
  WARN("failed to read entry\n");
  return hr;
}

StgStreamImpl_UnlockRegion()

static HRESULT WINAPI StgStreamImpl_UnlockRegion ( IStream *  iface, ULARGE_INTEGER  libOffset, ULARGE_INTEGER  cb, DWORD  dwLockType  )

static

Definition at line 514 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
 
  if (!This->parentStorage)
  {
    WARN("storage reverted\n");
    return STG_E_REVERTED;
  }
 
  FIXME("not implemented!\n");
  return E_NOTIMPL;
}

StgStreamImpl_Write()

static HRESULT WINAPI StgStreamImpl_Write ( IStream *  iface, const void *  pv, ULONG  cb, ULONG *  pcbWritten  )

static

Definition at line 184 of file stg_stream.c.

{
  StgStreamImpl* This = impl_from_IStream(iface);
 
  ULONG bytesWritten = 0;
  HRESULT res;
 
  TRACE("(%p, %p, %d, %p)\n",
    iface, pv, cb, pcbWritten);
 
  /*
   * Do we have permission to write to this stream?
   */
  switch(STGM_ACCESS_MODE(This->grfMode))
  {
  case STGM_WRITE:
  case STGM_READWRITE:
      break;
  default:
      WARN("access denied by flags: 0x%x\n", STGM_ACCESS_MODE(This->grfMode));
      return STG_E_ACCESSDENIED;
  }
 
  if (!pv)
    return STG_E_INVALIDPOINTER;
 
  if (!This->parentStorage)
  {
    WARN("storage reverted\n");
    return STG_E_REVERTED;
  }
 
  /*
   * If the caller is not interested in the number of bytes written,
   * we use another buffer to avoid "if" statements in the code.
   */
  if (pcbWritten == 0)
    pcbWritten = &bytesWritten;
 
  /*
   * Initialize the out parameter
   */
  *pcbWritten = 0;
 
  if (cb == 0)
  {
    TRACE("<-- S_OK, written 0\n");
    return S_OK;
  }
 
  res = StorageBaseImpl_StreamWriteAt(This->parentStorage,
                                      This->dirEntry,
                                      This->currentPosition,
                                      cb,
                                      pv,
                                      pcbWritten);
 
  /*
   * Advance the position pointer for the number of positions written.
   */
  This->currentPosition.QuadPart += *pcbWritten;
 
  if (SUCCEEDED(res))
    res = StorageBaseImpl_Flush(This->parentStorage);
 
  TRACE("<-- %08x, written %u\n", res, *pcbWritten);
  return res;
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

storage

)

Variable Documentation

StgStreamVtbl

const IStreamVtbl StgStreamVtbl

static

Initial value:

=
{
    StgStreamImpl_QueryInterface,
    StgStreamImpl_AddRef,
    StgStreamImpl_Release,
    StgStreamImpl_Read,
    StgStreamImpl_Write,
    StgStreamImpl_Seek,
    StgStreamImpl_SetSize,
    StgStreamImpl_CopyTo,
    StgStreamImpl_Commit,
    StgStreamImpl_Revert,
    StgStreamImpl_LockRegion,
    StgStreamImpl_UnlockRegion,
    StgStreamImpl_Stat,
    StgStreamImpl_Clone
}

Definition at line 637 of file stg_stream.c.

Referenced by StgStreamImpl_Construct().