dlls/shell32/regstream.c

   1 /*
   2  *      SHRegOpenStream
   3  */
   4 #include <string.h>
   5
   6 #include "wine/obj_storage.h"
   7
   8 #include "debug.h"
   9 #include "heap.h"
  10 #include "winerror.h"
  11 #include "winreg.h"
  12
  13 #include "shell.h"
  14 #include "pidl.h"
  15 #include "shell32_main.h"
  16
  17 DEFAULT_DEBUG_CHANNEL(shell)
  18
  19 typedef struct
  20 {       ICOM_VTABLE(IStream)* lpvtbl;
  21         DWORD           ref;
  22         HKEY            hKey;
  23         LPSTR           pszSubKey;
  24         LPSTR           pszValue;
  25         LPBYTE          pbBuffer;
  26         DWORD           dwLength;
  27         DWORD           dwPos;
  28 } ISHRegStream;
  29
  30 static struct ICOM_VTABLE(IStream) rstvt;
  31
  32 /**************************************************************************
  33 *   IStream_Constructor()
  34 */
  35 IStream *IStream_Constructor(HKEY hKey, LPCSTR pszSubKey, LPCSTR pszValue, DWORD grfMode)
  36 {       ISHRegStream*   rstr;
  37         DWORD           dwType;
  38
  39         rstr = (ISHRegStream*)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(ISHRegStream));
  40         rstr->lpvtbl=&rstvt;
  41         rstr->ref = 1;
  42
  43         if ( ERROR_SUCCESS == RegOpenKeyExA (hKey, pszSubKey, 0, KEY_READ, &(rstr->hKey)))
  44         { if ( ERROR_SUCCESS == RegQueryValueExA(rstr->hKey, (LPSTR)pszValue,0,0,0,&(rstr->dwLength)))
  45           {
  46             /* read the binary data into the buffer */
  47             rstr->pbBuffer = HeapAlloc(GetProcessHeap(),0,rstr->dwLength);
  48             if (rstr->pbBuffer)
  49             { if ( ERROR_SUCCESS == RegQueryValueExA(rstr->hKey, (LPSTR)pszValue,0,&dwType,rstr->pbBuffer,&(rstr->dwLength)))
  50               { if (dwType == REG_BINARY )
  51                 { rstr->pszSubKey = HEAP_strdupA (GetProcessHeap(),0, pszSubKey);
  52                   rstr->pszValue = HEAP_strdupA (GetProcessHeap(),0, pszValue);
  53                   TRACE(shell,"(%p)->0x%08x,%s,%s,0x%08lx\n", rstr, hKey, pszSubKey, pszValue, grfMode);
  54                   shell32_ObjCount++;
  55                   return (IStream*)rstr;
  56                 }
  57               }
  58               HeapFree (GetProcessHeap(),0,rstr->pbBuffer);
  59             }
  60           }
  61           RegCloseKey(rstr->hKey);
  62
  63         }
  64         HeapFree (GetProcessHeap(),0,rstr);
  65         return NULL;
  66 }
  67
  68 /**************************************************************************
  69 *  IStream_fnQueryInterface
  70 */
  71 static HRESULT WINAPI IStream_fnQueryInterface(IStream *iface, REFIID riid, LPVOID *ppvObj)
  72 {
  73         ICOM_THIS(ISHRegStream, iface);
  74
  75         char    xriid[50];
  76         WINE_StringFromCLSID((LPCLSID)riid,xriid);
  77
  78         TRACE(shell,"(%p)->(\n\tIID:\t%s,%p)\n",This,xriid,ppvObj);
  79
  80         *ppvObj = NULL;
  81
  82         if(IsEqualIID(riid, &IID_IUnknown))     /*IUnknown*/
  83         { *ppvObj = This;
  84         }
  85         else if(IsEqualIID(riid, &IID_IStream)) /*IStream*/
  86         { *ppvObj = This;
  87         }
  88
  89         if(*ppvObj)
  90         {
  91           IStream_AddRef((IStream*)*ppvObj);
  92           TRACE(shell,"-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
  93           return S_OK;
  94         }
  95         TRACE(shell,"-- Interface: E_NOINTERFACE\n");
  96         return E_NOINTERFACE;
  97 }
  98
  99 /**************************************************************************
 100 *  IStream_fnAddRef
 101 */
 102 static ULONG WINAPI IStream_fnAddRef(IStream *iface)
 103 {
 104         ICOM_THIS(ISHRegStream, iface);
 105
 106         TRACE(shell,"(%p)->(count=%lu)\n",This, This->ref);
 107
 108         shell32_ObjCount++;
 109         return ++(This->ref);
 110 }
 111
 112 /**************************************************************************
 113 *  IStream_fnRelease
 114 */
 115 static ULONG WINAPI IStream_fnRelease(IStream *iface)
 116 {
 117         ICOM_THIS(ISHRegStream, iface);
 118
 119         TRACE(shell,"(%p)->()\n",This);
 120
 121         shell32_ObjCount--;
 122
 123         if (!--(This->ref))
 124         { TRACE(shell," destroying SHReg IStream (%p)\n",This);
 125
 126           if (This->pszSubKey)
 127             HeapFree(GetProcessHeap(),0,This->pszSubKey);
 128
 129           if (This->pszValue)
 130             HeapFree(GetProcessHeap(),0,This->pszValue);
 131
 132           if (This->pbBuffer)
 133             HeapFree(GetProcessHeap(),0,This->pbBuffer);
 134
 135           if (This->hKey)
 136             RegCloseKey(This->hKey);
 137
 138           HeapFree(GetProcessHeap(),0,This);
 139           return 0;
 140         }
 141         return This->ref;
 142 }
 143
 144 HRESULT WINAPI IStream_fnRead (IStream * iface, void* pv, ULONG cb, ULONG* pcbRead)
 145 {
 146         ICOM_THIS(ISHRegStream, iface);
 147
 148         DWORD dwBytesToRead, dwBytesLeft;
 149
 150         TRACE(shell,"(%p)->(%p,0x%08lx,%p)\n",This, pv, cb, pcbRead);
 151
 152         if ( !pv )
 153           return STG_E_INVALIDPOINTER;
 154
 155         dwBytesLeft = This->dwLength - This->dwPos;
 156
 157         if ( 0 >= dwBytesLeft )                                         /* end of buffer */
 158           return S_FALSE;
 159
 160         dwBytesToRead = ( cb > dwBytesLeft) ? dwBytesLeft : cb;
 161
 162         memmove ( pv, (This->pbBuffer) + (This->dwPos), dwBytesToRead);
 163
 164         This->dwPos += dwBytesToRead;                                   /* adjust pointer */
 165
 166         if (pcbRead)
 167           *pcbRead = dwBytesToRead;
 168
 169         return S_OK;
 170 }
 171 HRESULT WINAPI IStream_fnWrite (IStream * iface, const void* pv, ULONG cb, ULONG* pcbWritten)
 172 {
 173         ICOM_THIS(ISHRegStream, iface);
 174
 175         TRACE(shell,"(%p)\n",This);
 176
 177         return E_NOTIMPL;
 178 }
 179 HRESULT WINAPI IStream_fnSeek (IStream * iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition)
 180 {
 181         ICOM_THIS(ISHRegStream, iface);
 182
 183         TRACE(shell,"(%p)\n",This);
 184
 185         return E_NOTIMPL;
 186 }
 187 HRESULT WINAPI IStream_fnSetSize (IStream * iface, ULARGE_INTEGER libNewSize)
 188 {
 189         ICOM_THIS(ISHRegStream, iface);
 190
 191         TRACE(shell,"(%p)\n",This);
 192
 193         return E_NOTIMPL;
 194 }
 195 HRESULT WINAPI IStream_fnCopyTo (IStream * iface, IStream* pstm, ULARGE_INTEGER cb, ULARGE_INTEGER* pcbRead, ULARGE_INTEGER* pcbWritten)
 196 {
 197         ICOM_THIS(ISHRegStream, iface);
 198
 199         TRACE(shell,"(%p)\n",This);
 200
 201         return E_NOTIMPL;
 202 }
 203 HRESULT WINAPI IStream_fnCommit (IStream * iface, DWORD grfCommitFlags)
 204 {
 205         ICOM_THIS(ISHRegStream, iface);
 206
 207         TRACE(shell,"(%p)\n",This);
 208
 209         return E_NOTIMPL;
 210 }
 211 HRESULT WINAPI IStream_fnRevert (IStream * iface)
 212 {
 213         ICOM_THIS(ISHRegStream, iface);
 214
 215         TRACE(shell,"(%p)\n",This);
 216
 217         return E_NOTIMPL;
 218 }
 219 HRESULT WINAPI IStream_fnLockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
 220 {
 221         ICOM_THIS(ISHRegStream, iface);
 222
 223         TRACE(shell,"(%p)\n",This);
 224
 225         return E_NOTIMPL;
 226 }
 227 HRESULT WINAPI IStream_fnUnlockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
 228 {
 229         ICOM_THIS(ISHRegStream, iface);
 230
 231         TRACE(shell,"(%p)\n",This);
 232
 233         return E_NOTIMPL;
 234 }
 235 HRESULT WINAPI IStream_fnStat (IStream * iface, STATSTG*   pstatstg, DWORD grfStatFlag)
 236 {
 237         ICOM_THIS(ISHRegStream, iface);
 238
 239         TRACE(shell,"(%p)\n",This);
 240
 241         return E_NOTIMPL;
 242 }
 243 HRESULT WINAPI IStream_fnClone (IStream * iface, IStream** ppstm)
 244 {
 245         ICOM_THIS(ISHRegStream, iface);
 246
 247         TRACE(shell,"(%p)\n",This);
 248
 249         return E_NOTIMPL;
 250 }
 251
 252 static struct ICOM_VTABLE(IStream) rstvt =
 253 {
 254         IStream_fnQueryInterface,
 255         IStream_fnAddRef,
 256         IStream_fnRelease,
 257         IStream_fnRead,
 258         IStream_fnWrite,
 259         IStream_fnSeek,
 260         IStream_fnSetSize,
 261         IStream_fnCopyTo,
 262         IStream_fnCommit,
 263         IStream_fnRevert,
 264         IStream_fnLockRegion,
 265         IStream_fnUnlockRegion,
 266         IStream_fnStat,
 267         IStream_fnClone
 268
 269 };
 270
 271 /*************************************************************************
 272  * OpenRegStream                                [SHELL32.85]
 273  *
 274  * NOTES
 275  *     exported by ordinal
 276  */
 277 IStream * WINAPI OpenRegStream(HKEY hkey, LPCSTR pszSubkey, LPCSTR pszValue, DWORD grfMode)
 278 {
 279         TRACE(shell,"(0x%08x,%s,%s,0x%08lx)\n",hkey, pszSubkey, pszValue, grfMode);
 280         return IStream_Constructor(hkey, pszSubkey, pszValue, grfMode);
 281 }