regstream.c File Reference

#include <stdarg.h>
#include <string.h>
#include "winerror.h"
#include "windef.h"
#include "winbase.h"
#include "objbase.h"
#include "winreg.h"
#include "shlwapi.h"
#include "wine/debug.h"

Include dependency graph for regstream.c:

Go to the source code of this file.

Classes
struct	ISHRegStream

Macros
#define	COBJMACROS

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (shell)
static ISHRegStream *	impl_from_IStream (IStream *iface)
static HRESULT WINAPI	IStream_fnQueryInterface (IStream *iface, REFIID riid, LPVOID *ppvObj)
static ULONG WINAPI	IStream_fnAddRef (IStream *iface)
static ULONG WINAPI	IStream_fnRelease (IStream *iface)
static HRESULT WINAPI	IStream_fnRead (IStream *iface, void *pv, ULONG cb, ULONG *pcbRead)
static HRESULT WINAPI	IStream_fnWrite (IStream *iface, const void *pv, ULONG cb, ULONG *pcbWritten)
static HRESULT WINAPI	IStream_fnSeek (IStream *iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
static HRESULT WINAPI	IStream_fnSetSize (IStream *iface, ULARGE_INTEGER libNewSize)
static HRESULT WINAPI	IStream_fnCopyTo (IStream *iface, IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
static HRESULT WINAPI	IStream_fnCommit (IStream *iface, DWORD grfCommitFlags)
static HRESULT WINAPI	IStream_fnRevert (IStream *iface)
static HRESULT WINAPI	IStream_fnLockUnlockRegion (IStream *iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
static HRESULT WINAPI	IStream_fnStat (IStream *iface, STATSTG *pstatstg, DWORD grfStatFlag)
static HRESULT WINAPI	IStream_fnClone (IStream *iface, IStream **ppstm)
static ISHRegStream *	IStream_Create (HKEY hKey, LPBYTE pbBuffer, DWORD dwLength)
HRESULT WINAPI	SHCreateStreamWrapper (LPBYTE lpbData, DWORD dwDataLen, DWORD dwReserved, IStream **lppStream)

Variables
static const IStreamVtbl	rstvt

Macro Definition Documentation

COBJMACROS

#define COBJMACROS

Definition at line 25 of file regstream.c.

Function Documentation

impl_from_IStream()

static ISHRegStream * impl_from_IStream ( IStream *  iface )

inlinestatic

Definition at line 54 of file regstream.c.

{
    return CONTAINING_RECORD(iface, ISHRegStream, IStream_iface);
}

Referenced by IStream_fnAddRef(), IStream_fnClone(), IStream_fnCommit(), IStream_fnCopyTo(), IStream_fnLockUnlockRegion(), IStream_fnQueryInterface(), IStream_fnRead(), IStream_fnRelease(), IStream_fnRevert(), IStream_fnSeek(), IStream_fnSetSize(), IStream_fnStat(), and IStream_fnWrite().

IStream_Create()

static ISHRegStream * IStream_Create ( HKEY  hKey, LPBYTE  pbBuffer, DWORD  dwLength  )

static

Definition at line 377 of file regstream.c.

{
 ISHRegStream* regStream;
 
 regStream = malloc(sizeof(ISHRegStream));
 
 if (regStream)
 {
   regStream->IStream_iface.lpVtbl = &rstvt;
   regStream->ref = 1;
   regStream->hKey = hKey;
   regStream->pbBuffer = pbBuffer;
   regStream->dwLength = dwLength;
   regStream->dwPos = 0;
   regStream->dwMode = STGM_READWRITE;
   regStream->u.keyNameA = NULL;
   regStream->bUnicode = FALSE;
 }
 TRACE ("Returning %p\n", regStream);
 return regStream;
}

Referenced by SHCreateStreamWrapper().

IStream_fnAddRef()

static ULONG WINAPI IStream_fnAddRef ( IStream *  iface )

static

Definition at line 86 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
    ULONG refCount = InterlockedIncrement(&This->ref);
    
    TRACE("(%p)->(ref before=%lu)\n",This, refCount - 1);
 
    return refCount;
}

IStream_fnClone()

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

static

Definition at line 343 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
 
    TRACE("(%p)\n",This);
    *ppstm = NULL;
 
    /* clone not supported by this stream */
    return E_NOTIMPL;
}

IStream_fnCommit()

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

static

Definition at line 277 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
 
    TRACE("(%p)\n",This);
 
    /* commit not supported by this stream */
    return E_NOTIMPL;
}

IStream_fnCopyTo()

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

static

Definition at line 260 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
 
    TRACE("(%p)\n",This);
    if (pcbRead)
      pcbRead->QuadPart = 0;
    if (pcbWritten)
      pcbWritten->QuadPart = 0;
 
    /* TODO implement */
    return E_NOTIMPL;
}

IStream_fnLockUnlockRegion()

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

static

Definition at line 303 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
 
    TRACE("(%p)\n",This);
 
    /* lock/unlock not supported by this stream */
    return E_NOTIMPL;
}

IStream_fnQueryInterface()

static HRESULT WINAPI IStream_fnQueryInterface ( IStream *  iface, REFIID  riid, LPVOID *  ppvObj  )

static

Definition at line 62 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
 
    TRACE("(%p)->(\n\tIID:\t%s,%p)\n",This,debugstr_guid(riid),ppvObj);
 
    *ppvObj = NULL;
 
       if(IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IStream))
         *ppvObj = &This->IStream_iface;
 
    if(*ppvObj)
    {
      IStream_AddRef((IStream*)*ppvObj);
      TRACE("-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
      return S_OK;
    }
    TRACE("-- Interface: E_NOINTERFACE\n");
    return E_NOINTERFACE;
}

IStream_fnRead()

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

static

Definition at line 148 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
    DWORD dwBytesToRead;
 
    TRACE("(%p)->(%p,0x%08lx,%p)\n",This, pv, cb, pcbRead);
 
    if (This->dwPos >= This->dwLength)
      dwBytesToRead = 0;
        else
      dwBytesToRead = This->dwLength - This->dwPos;
 
    dwBytesToRead = (cb > dwBytesToRead) ? dwBytesToRead : cb;
    if (dwBytesToRead != 0) /* not at end of buffer and we want to read something */
    {
      memmove(pv, This->pbBuffer + This->dwPos, dwBytesToRead);
      This->dwPos += dwBytesToRead; /* adjust pointer */
    }
 
    if (pcbRead)
      *pcbRead = dwBytesToRead;
 
    return S_OK;
}

IStream_fnRelease()

static ULONG WINAPI IStream_fnRelease ( IStream *  iface )

static

Definition at line 99 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
    ULONG refCount = InterlockedDecrement(&This->ref);
 
    TRACE("(%p)->(ref before=%lu)\n",This, refCount + 1);
 
    if (!refCount)
    {
      TRACE(" destroying SHReg IStream (%p)\n",This);
 
      if (This->hKey)
      {
        /* write back data in REG_BINARY */
        if (This->dwMode == STGM_READWRITE || This->dwMode == STGM_WRITE)
        {
          if (This->dwLength)
          {
            if (This->bUnicode)
              RegSetValueExW(This->hKey, This->u.keyNameW, 0, REG_BINARY,
                             (const BYTE *) This->pbBuffer, This->dwLength);
            else
              RegSetValueExA(This->hKey, This->u.keyNameA, 0, REG_BINARY,
                            (const BYTE *) This->pbBuffer, This->dwLength);
          }
          else
          {
            if (This->bUnicode)
              RegDeleteValueW(This->hKey, This->u.keyNameW);
            else
              RegDeleteValueA(This->hKey, This->u.keyNameA);
          }
        }
 
        RegCloseKey(This->hKey);
      }
 
      HeapFree(GetProcessHeap(),0,This->u.keyNameA);
      HeapFree(GetProcessHeap(),0,This->pbBuffer);
      HeapFree(GetProcessHeap(),0,This);
      return 0;
    }
 
    return refCount;
}

IStream_fnRevert()

static HRESULT WINAPI IStream_fnRevert ( IStream *  iface )

static

Definition at line 290 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
 
    TRACE("(%p)\n",This);
 
    /* revert not supported by this stream */
    return E_NOTIMPL;
}

IStream_fnSeek()

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

static

Definition at line 207 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
    LARGE_INTEGER tmp;
    TRACE("(%p, %s, %ld %p)\n", This,
              wine_dbgstr_longlong(dlibMove.QuadPart), dwOrigin, plibNewPosition);
 
    if (dwOrigin == STREAM_SEEK_SET)
      tmp = dlibMove;
        else if (dwOrigin == STREAM_SEEK_CUR)
      tmp.QuadPart = This->dwPos + dlibMove.QuadPart;
    else if (dwOrigin == STREAM_SEEK_END)
      tmp.QuadPart = This->dwLength + dlibMove.QuadPart;
        else
      return STG_E_INVALIDPARAMETER;
 
    if (tmp.QuadPart < 0)
      return STG_E_INVALIDFUNCTION;
 
    /* we cut off the high part here */
    This->dwPos = tmp.u.LowPart;
 
    if (plibNewPosition)
      plibNewPosition->QuadPart = This->dwPos;
    return S_OK;
}

IStream_fnSetSize()

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

static

Definition at line 237 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
    DWORD newLen;
    LPBYTE newBuf;
 
    TRACE("(%p, %s)\n", This, wine_dbgstr_longlong(libNewSize.QuadPart));
 
    /* we cut off the high part here */
    newLen = libNewSize.u.LowPart;
    newBuf = _recalloc(This->pbBuffer, 1, newLen);
    if (!newBuf)
      return STG_E_INSUFFICIENTMEMORY;
 
    This->pbBuffer = newBuf;
    This->dwLength = newLen;
 
    return S_OK;
}

IStream_fnStat()

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

static

Definition at line 316 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
 
    TRACE("(%p, %p, %ld)\n",This,pstatstg,grfStatFlag);
 
    pstatstg->pwcsName = NULL;
    pstatstg->type = STGTY_STREAM;
    pstatstg->cbSize.QuadPart = This->dwLength;
    pstatstg->mtime.dwHighDateTime = 0;
    pstatstg->mtime.dwLowDateTime = 0;
    pstatstg->ctime.dwHighDateTime = 0;
    pstatstg->ctime.dwLowDateTime = 0;
    pstatstg->atime.dwHighDateTime = 0;
    pstatstg->atime.dwLowDateTime = 0;
    pstatstg->grfMode = This->dwMode;
    pstatstg->grfLocksSupported = 0;
    pstatstg->clsid = CLSID_NULL;
    pstatstg->grfStateBits = 0;
    pstatstg->reserved = 0;
 
    return S_OK;
}

IStream_fnWrite()

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

static

Definition at line 176 of file regstream.c.

{
    ISHRegStream *This = impl_from_IStream(iface);
    DWORD newLen = This->dwPos + cb;
 
    TRACE("(%p, %p, %ld, %p)\n",This, pv, cb, pcbWritten);
 
    if (newLen < This->dwPos) /* overflow */
      return STG_E_INSUFFICIENTMEMORY;
 
    if (newLen > This->dwLength)
    {
      BYTE *newBuf = _recalloc(This->pbBuffer, 1, newLen);
      if (!newBuf)
        return STG_E_INSUFFICIENTMEMORY;
 
      This->dwLength = newLen;
      This->pbBuffer = newBuf;
    }
    memmove(This->pbBuffer + This->dwPos, pv, cb);
    This->dwPos += cb; /* adjust pointer */
 
    if (pcbWritten)
      *pcbWritten = cb;
 
    return S_OK;
}

SHCreateStreamWrapper()

HRESULT WINAPI SHCreateStreamWrapper	(	LPBYTE	lpbData,
		DWORD	dwDataLen,
		DWORD	dwReserved,
		IStream **	lppStream
	)

Definition at line 418 of file regstream.c.

{
  ISHRegStream *strm;
 
  if (lppStream)
    *lppStream = NULL;
 
  if(dwReserved || !lppStream)
    return E_INVALIDARG;
 
  strm = IStream_Create(NULL, lpbData, dwDataLen);
 
  if(!strm)
    return E_OUTOFMEMORY;
 
  IStream_QueryInterface(&strm->IStream_iface, &IID_IStream, (void**)lppStream);
  IStream_Release(&strm->IStream_iface);
  return S_OK;
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

shell

)

Variable Documentation

rstvt

const IStreamVtbl rstvt

static

Initial value:

=
{
    IStream_fnQueryInterface,
    IStream_fnAddRef,
    IStream_fnRelease,
    IStream_fnRead,
    IStream_fnWrite,
    IStream_fnSeek,
    IStream_fnSetSize,
    IStream_fnCopyTo,
    IStream_fnCommit,
    IStream_fnRevert,
    IStream_fnLockUnlockRegion,
    IStream_fnLockUnlockRegion,
    IStream_fnStat,
    IStream_fnClone
}

Definition at line 354 of file regstream.c.

Referenced by IStream_Create().