dlls/shell32/memorystream.c

   1 /*
   2  *      this class implements a pure IStream object
   3  *      and can be used for many purposes
   4  *
   5  *      the main reason for implementing this was
   6  *      a cleaner implementation of IShellLink which
   7  *      needs to be able to load lnk's from a IStream
   8  *      interface so it was obvious to capsule the file
   9  *      access in a IStream to.
  10  *
  11  * Copyright 1999 Juergen Schmied
  12  * Copyright 2003 Mike McCormack for CodeWeavers
  13  *
  14  * This library is free software; you can redistribute it and/or
  15  * modify it under the terms of the GNU Lesser General Public
  16  * License as published by the Free Software Foundation; either
  17  * version 2.1 of the License, or (at your option) any later version.
  18  *
  19  * This library is distributed in the hope that it will be useful,
  20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  22  * Lesser General Public License for more details.
  23  *
  24  * You should have received a copy of the GNU Lesser General Public
  25  * License along with this library; if not, write to the Free Software
  26  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  27  */
  28
  29 #include <stdarg.h>
  30 #include <string.h>
  31
  32 #include "windef.h"
  33 #include "winbase.h"
  34 #include "winerror.h"
  35 #include "winuser.h"
  36 #include "wingdi.h"
  37 #include "shlobj.h"
  38 #include "wine/debug.h"
  39 #include "shell32_main.h"
  40
  41 WINE_DEFAULT_DEBUG_CHANNEL(shell);
  42
  43 static HRESULT WINAPI IStream_fnQueryInterface(IStream *iface, REFIID riid, LPVOID *ppvObj);
  44 static ULONG WINAPI IStream_fnAddRef(IStream *iface);
  45 static ULONG WINAPI IStream_fnRelease(IStream *iface);
  46 static HRESULT WINAPI IStream_fnRead (IStream * iface, void* pv, ULONG cb, ULONG* pcbRead);
  47 static HRESULT WINAPI IStream_fnWrite (IStream * iface, const void* pv, ULONG cb, ULONG* pcbWritten);
  48 static HRESULT WINAPI IStream_fnSeek (IStream * iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition);
  49 static HRESULT WINAPI IStream_fnSetSize (IStream * iface, ULARGE_INTEGER libNewSize);
  50 static HRESULT WINAPI IStream_fnCopyTo (IStream * iface, IStream* pstm, ULARGE_INTEGER cb, ULARGE_INTEGER* pcbRead, ULARGE_INTEGER* pcbWritten);
  51 static HRESULT WINAPI IStream_fnCommit (IStream * iface, DWORD grfCommitFlags);
  52 static HRESULT WINAPI IStream_fnRevert (IStream * iface);
  53 static HRESULT WINAPI IStream_fnLockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
  54 static HRESULT WINAPI IStream_fnUnlockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
  55 static HRESULT WINAPI IStream_fnStat (IStream * iface, STATSTG*   pstatstg, DWORD grfStatFlag);
  56 static HRESULT WINAPI IStream_fnClone (IStream * iface, IStream** ppstm);
  57
  58 static ICOM_VTABLE(IStream) stvt =
  59 {
  60         ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
  61         IStream_fnQueryInterface,
  62         IStream_fnAddRef,
  63         IStream_fnRelease,
  64         IStream_fnRead,
  65         IStream_fnWrite,
  66         IStream_fnSeek,
  67         IStream_fnSetSize,
  68         IStream_fnCopyTo,
  69         IStream_fnCommit,
  70         IStream_fnRevert,
  71         IStream_fnLockRegion,
  72         IStream_fnUnlockRegion,
  73         IStream_fnStat,
  74         IStream_fnClone
  75
  76 };
  77
  78 typedef struct
  79 {       ICOM_VTABLE(IStream)    *lpvtst;
  80         DWORD           ref;
  81         HANDLE          handle;
  82 } ISHFileStream;
  83
  84 /**************************************************************************
  85  *   CreateStreamOnFile()
  86  *
  87  *   similar to CreateStreamOnHGlobal
  88  */
  89 HRESULT CreateStreamOnFile (LPCWSTR pszFilename, DWORD grfMode, IStream ** ppstm)
  90 {
  91         ISHFileStream*  fstr;
  92         HANDLE          handle;
  93         DWORD           access = GENERIC_READ, creat;
  94
  95         if( grfMode & STGM_TRANSACTED )
  96                 return E_INVALIDARG;
  97
  98         if( grfMode & STGM_WRITE )
  99                 access |= GENERIC_WRITE;
 100         if( grfMode & STGM_READWRITE )
 101                 access = GENERIC_WRITE | GENERIC_READ;
 102
 103         if( grfMode & STGM_CREATE )
 104                 creat = CREATE_ALWAYS;
 105         else
 106                 creat = OPEN_EXISTING;
 107
 108         TRACE("Opening %s\n", debugstr_w(pszFilename) );
 109
 110        handle = CreateFileW( pszFilename, access, FILE_SHARE_READ, NULL, creat, 0, NULL );
 111         if( handle == INVALID_HANDLE_VALUE )
 112                 return E_FAIL;
 113
 114         fstr = (ISHFileStream*)HeapAlloc(GetProcessHeap(),
 115                 HEAP_ZERO_MEMORY,sizeof(ISHFileStream));
 116         if( !fstr )
 117                 return E_FAIL;
 118         fstr->lpvtst=&stvt;
 119         fstr->ref = 1;
 120         fstr->handle = handle;
 121
 122         (*ppstm) = (IStream*)fstr;
 123
 124         return S_OK;
 125 }
 126
 127 /**************************************************************************
 128 *  IStream_fnQueryInterface
 129 */
 130 static HRESULT WINAPI IStream_fnQueryInterface(IStream *iface, REFIID riid, LPVOID *ppvObj)
 131 {
 132         ICOM_THIS(ISHFileStream, iface);
 133
 134         TRACE("(%p)->(\n\tIID:\t%s,%p)\n",This,debugstr_guid(riid),ppvObj);
 135
 136         *ppvObj = NULL;
 137
 138         if(IsEqualIID(riid, &IID_IUnknown) ||
 139            IsEqualIID(riid, &IID_IStream))
 140         {
 141           *ppvObj = This;
 142         }
 143
 144         if(*ppvObj)
 145         {
 146           IStream_AddRef((IStream*)*ppvObj);
 147           TRACE("-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
 148           return S_OK;
 149         }
 150         TRACE("-- Interface: E_NOINTERFACE\n");
 151         return E_NOINTERFACE;
 152 }
 153
 154 /**************************************************************************
 155 *  IStream_fnAddRef
 156 */
 157 static ULONG WINAPI IStream_fnAddRef(IStream *iface)
 158 {
 159         ICOM_THIS(ISHFileStream, iface);
 160
 161         TRACE("(%p)->(count=%lu)\n",This, This->ref);
 162
 163         return ++(This->ref);
 164 }
 165
 166 /**************************************************************************
 167 *  IStream_fnRelease
 168 */
 169 static ULONG WINAPI IStream_fnRelease(IStream *iface)
 170 {
 171         ICOM_THIS(ISHFileStream, iface);
 172
 173         TRACE("(%p)->()\n",This);
 174
 175         if (!--(This->ref))
 176         {
 177                 TRACE(" destroying SHFileStream (%p)\n",This);
 178                 CloseHandle(This->handle);
 179                 HeapFree(GetProcessHeap(),0,This);
 180         }
 181         return This->ref;
 182 }
 183
 184 static HRESULT WINAPI IStream_fnRead (IStream * iface, void* pv, ULONG cb, ULONG* pcbRead)
 185 {
 186         ICOM_THIS(ISHFileStream, iface);
 187
 188         TRACE("(%p)->(%p,0x%08lx,%p)\n",This, pv, cb, pcbRead);
 189
 190         if ( !pv )
 191                 return STG_E_INVALIDPOINTER;
 192
 193         if ( ! ReadFile( This->handle, pv, cb, pcbRead, NULL ) )
 194                 return E_FAIL;
 195
 196         return S_OK;
 197 }
 198
 199 static HRESULT WINAPI IStream_fnWrite (IStream * iface, const void* pv, ULONG cb, ULONG* pcbWritten)
 200 {
 201        DWORD dummy_count;
 202         ICOM_THIS(ISHFileStream, iface);
 203
 204         TRACE("(%p)\n",This);
 205
 206         if( !pv )
 207                 return STG_E_INVALIDPOINTER;
 208
 209        /* WriteFile() doesn't allow to specify NULL as write count pointer */
 210        if (!pcbWritten)
 211                pcbWritten = &dummy_count;
 212
 213         if( ! WriteFile( This->handle, pv, cb, pcbWritten, NULL ) )
 214                 return E_FAIL;
 215
 216         return S_OK;
 217 }
 218
 219 static HRESULT WINAPI IStream_fnSeek (IStream * iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition)
 220 {
 221         DWORD pos, newposlo, newposhi;
 222
 223         ICOM_THIS(ISHFileStream, iface);
 224
 225         TRACE("(%p)\n",This);
 226
 227         pos = dlibMove.QuadPart; /* FIXME: truncates */
 228         newposhi = 0;
 229         newposlo = SetFilePointer( This->handle, pos, &newposhi, dwOrigin );
 230         if( newposlo == INVALID_SET_FILE_POINTER )
 231                 return E_FAIL;
 232
 233         plibNewPosition->QuadPart = newposlo | ( (LONGLONG)newposhi<<32);
 234
 235         return S_OK;
 236 }
 237
 238 static HRESULT WINAPI IStream_fnSetSize (IStream * iface, ULARGE_INTEGER libNewSize)
 239 {
 240         ICOM_THIS(ISHFileStream, iface);
 241
 242         TRACE("(%p)\n",This);
 243
 244         if( ! SetFilePointer( This->handle, libNewSize.QuadPart, NULL, FILE_BEGIN ) )
 245                 return E_FAIL;
 246
 247         if( ! SetEndOfFile( This->handle ) )
 248                 return E_FAIL;
 249
 250         return S_OK;
 251 }
 252 static HRESULT WINAPI IStream_fnCopyTo (IStream * iface, IStream* pstm, ULARGE_INTEGER cb, ULARGE_INTEGER* pcbRead, ULARGE_INTEGER* pcbWritten)
 253 {
 254         ICOM_THIS(ISHFileStream, iface);
 255
 256         TRACE("(%p)\n",This);
 257
 258         return E_NOTIMPL;
 259 }
 260 static HRESULT WINAPI IStream_fnCommit (IStream * iface, DWORD grfCommitFlags)
 261 {
 262         ICOM_THIS(ISHFileStream, iface);
 263
 264         TRACE("(%p)\n",This);
 265
 266         return E_NOTIMPL;
 267 }
 268 static HRESULT WINAPI IStream_fnRevert (IStream * iface)
 269 {
 270         ICOM_THIS(ISHFileStream, iface);
 271
 272         TRACE("(%p)\n",This);
 273
 274         return E_NOTIMPL;
 275 }
 276 static HRESULT WINAPI IStream_fnLockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
 277 {
 278         ICOM_THIS(ISHFileStream, iface);
 279
 280         TRACE("(%p)\n",This);
 281
 282         return E_NOTIMPL;
 283 }
 284 static HRESULT WINAPI IStream_fnUnlockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
 285 {
 286         ICOM_THIS(ISHFileStream, iface);
 287
 288         TRACE("(%p)\n",This);
 289
 290         return E_NOTIMPL;
 291 }
 292 static HRESULT WINAPI IStream_fnStat (IStream * iface, STATSTG*   pstatstg, DWORD grfStatFlag)
 293 {
 294         ICOM_THIS(ISHFileStream, iface);
 295
 296         TRACE("(%p)\n",This);
 297
 298         return E_NOTIMPL;
 299 }
 300 static HRESULT WINAPI IStream_fnClone (IStream * iface, IStream** ppstm)
 301 {
 302         ICOM_THIS(ISHFileStream, iface);
 303
 304         TRACE("(%p)\n",This);
 305
 306         return E_NOTIMPL;
 307 }