source: trunk/src/shlwapi/regstream.c@ 9022

Last change on this file since 9022 was 8584, checked in by sandervl, 23 years ago

resynced with latest Wine
File size: 8.1 KB
Line 
1/*
2 * SHRegOpenStream
3 *
4 * Copyright 1999 Juergen Schmied
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 */
20
21#include <string.h>
22
23#include "winerror.h"
24#include "winbase.h"
25#include "winreg.h"
26#include "shlobj.h"
27
28#include "wine/debug.h"
29
30WINE_DEFAULT_DEBUG_CHANNEL(shell);
31
32typedef struct
33{ ICOM_VFIELD(IStream);
34 DWORD ref;
35 HKEY hKey;
36 LPBYTE pbBuffer;
37 DWORD dwLength;
38 DWORD dwPos;
39} ISHRegStream;
40
41static struct ICOM_VTABLE(IStream) rstvt;
42
43/**************************************************************************
44* IStream_ConstructorA [internal]
45*/
46static IStream *IStream_ConstructorA(HKEY hKey, LPCSTR pszSubKey, LPCSTR pszValue, DWORD grfMode)
47{
48 ISHRegStream* rstr;
49 DWORD dwType;
50
51 rstr = (ISHRegStream*)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(ISHRegStream));
52
53 ICOM_VTBL(rstr)=&rstvt;
54 rstr->ref = 1;
55
56 if (!(RegOpenKeyExA (hKey, pszSubKey, 0, KEY_READ, &(rstr->hKey))))
57 {
58 if (!(RegQueryValueExA(rstr->hKey, pszValue,0,0,0,&(rstr->dwLength))))
59 {
60 /* read the binary data into the buffer */
61 if((rstr->pbBuffer = HeapAlloc(GetProcessHeap(),0,rstr->dwLength)))
62 {
63 if (!(RegQueryValueExA(rstr->hKey, pszValue,0,&dwType,rstr->pbBuffer,&(rstr->dwLength))))
64 {
65 if (dwType == REG_BINARY )
66 {
67 TRACE ("%p\n", rstr);
68 return (IStream*)rstr;
69 }
70 }
71 HeapFree (GetProcessHeap(),0,rstr->pbBuffer);
72 }
73 }
74 RegCloseKey(rstr->hKey);
75 }
76 HeapFree (GetProcessHeap(),0,rstr);
77 return NULL;
78}
79
80/**************************************************************************
81* IStream_ConstructorW [internal]
82*/
83static IStream *IStream_ConstructorW(HKEY hKey, LPCWSTR pszSubKey, LPCWSTR pszValue, DWORD grfMode)
84{
85 ISHRegStream* rstr;
86 DWORD dwType;
87
88 rstr = (ISHRegStream*)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(ISHRegStream));
89
90 ICOM_VTBL(rstr)=&rstvt;
91 rstr->ref = 1;
92
93 if (!(RegOpenKeyExW (hKey, pszSubKey, 0, KEY_READ, &(rstr->hKey))))
94 {
95 if (!(RegQueryValueExW(rstr->hKey, pszValue,0,0,0,&(rstr->dwLength))))
96 {
97 /* read the binary data into the buffer */
98 if((rstr->pbBuffer = HeapAlloc(GetProcessHeap(),0,rstr->dwLength)))
99 {
100 if (!(RegQueryValueExW(rstr->hKey, pszValue,0,&dwType,rstr->pbBuffer,&(rstr->dwLength))))
101 {
102 if (dwType == REG_BINARY )
103 {
104 TRACE ("%p\n", rstr);
105 return (IStream*)rstr;
106 }
107 }
108 HeapFree (GetProcessHeap(),0,rstr->pbBuffer);
109 }
110 }
111 RegCloseKey(rstr->hKey);
112 }
113 HeapFree (GetProcessHeap(),0,rstr);
114 return NULL;
115}
116
117/**************************************************************************
118* IStream_fnQueryInterface
119*/
120static HRESULT WINAPI IStream_fnQueryInterface(IStream *iface, REFIID riid, LPVOID *ppvObj)
121{
122 ICOM_THIS(ISHRegStream, iface);
123
124 TRACE("(%p)->(\n\tIID:\t%s,%p)\n",This,debugstr_guid(riid),ppvObj);
125
126 *ppvObj = NULL;
127
128 if(IsEqualIID(riid, &IID_IUnknown)) /*IUnknown*/
129 { *ppvObj = This;
130 }
131 else if(IsEqualIID(riid, &IID_IStream)) /*IStream*/
132 { *ppvObj = This;
133 }
134
135 if(*ppvObj)
136 {
137 IStream_AddRef((IStream*)*ppvObj);
138 TRACE("-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
139 return S_OK;
140 }
141 TRACE("-- Interface: E_NOINTERFACE\n");
142 return E_NOINTERFACE;
143}
144
145/**************************************************************************
146* IStream_fnAddRef
147*/
148static ULONG WINAPI IStream_fnAddRef(IStream *iface)
149{
150 ICOM_THIS(ISHRegStream, iface);
151
152 TRACE("(%p)->(count=%lu)\n",This, This->ref);
153
154 return ++(This->ref);
155}
156
157/**************************************************************************
158* IStream_fnRelease
159*/
160static ULONG WINAPI IStream_fnRelease(IStream *iface)
161{
162 ICOM_THIS(ISHRegStream, iface);
163
164 TRACE("(%p)->()\n",This);
165
166 if (!--(This->ref))
167 { TRACE(" destroying SHReg IStream (%p)\n",This);
168
169 if (This->pbBuffer)
170 HeapFree(GetProcessHeap(),0,This->pbBuffer);
171
172 if (This->hKey)
173 RegCloseKey(This->hKey);
174
175 HeapFree(GetProcessHeap(),0,This);
176 return 0;
177 }
178 return This->ref;
179}
180
181static HRESULT WINAPI IStream_fnRead (IStream * iface, void* pv, ULONG cb, ULONG* pcbRead)
182{
183 ICOM_THIS(ISHRegStream, iface);
184
185 DWORD dwBytesToRead, dwBytesLeft;
186
187 TRACE("(%p)->(%p,0x%08lx,%p)\n",This, pv, cb, pcbRead);
188
189 if ( !pv )
190 return STG_E_INVALIDPOINTER;
191
192 dwBytesLeft = This->dwLength - This->dwPos;
193
194 if ( 0 >= dwBytesLeft ) /* end of buffer */
195 return S_FALSE;
196
197 dwBytesToRead = ( cb > dwBytesLeft) ? dwBytesLeft : cb;
198
199 memmove ( pv, (This->pbBuffer) + (This->dwPos), dwBytesToRead);
200
201 This->dwPos += dwBytesToRead; /* adjust pointer */
202
203 if (pcbRead)
204 *pcbRead = dwBytesToRead;
205
206 return S_OK;
207}
208static HRESULT WINAPI IStream_fnWrite (IStream * iface, const void* pv, ULONG cb, ULONG* pcbWritten)
209{
210 ICOM_THIS(ISHRegStream, iface);
211
212 TRACE("(%p)\n",This);
213
214 return E_NOTIMPL;
215}
216static HRESULT WINAPI IStream_fnSeek (IStream * iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition)
217{
218 ICOM_THIS(ISHRegStream, iface);
219
220 TRACE("(%p)\n",This);
221
222 return E_NOTIMPL;
223}
224static HRESULT WINAPI IStream_fnSetSize (IStream * iface, ULARGE_INTEGER libNewSize)
225{
226 ICOM_THIS(ISHRegStream, iface);
227
228 TRACE("(%p)\n",This);
229
230 return E_NOTIMPL;
231}
232static HRESULT WINAPI IStream_fnCopyTo (IStream * iface, IStream* pstm, ULARGE_INTEGER cb, ULARGE_INTEGER* pcbRead, ULARGE_INTEGER* pcbWritten)
233{
234 ICOM_THIS(ISHRegStream, iface);
235
236 TRACE("(%p)\n",This);
237
238 return E_NOTIMPL;
239}
240static HRESULT WINAPI IStream_fnCommit (IStream * iface, DWORD grfCommitFlags)
241{
242 ICOM_THIS(ISHRegStream, iface);
243
244 TRACE("(%p)\n",This);
245
246 return E_NOTIMPL;
247}
248static HRESULT WINAPI IStream_fnRevert (IStream * iface)
249{
250 ICOM_THIS(ISHRegStream, iface);
251
252 TRACE("(%p)\n",This);
253
254 return E_NOTIMPL;
255}
256static HRESULT WINAPI IStream_fnLockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
257{
258 ICOM_THIS(ISHRegStream, iface);
259
260 TRACE("(%p)\n",This);
261
262 return E_NOTIMPL;
263}
264static HRESULT WINAPI IStream_fnUnlockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
265{
266 ICOM_THIS(ISHRegStream, iface);
267
268 TRACE("(%p)\n",This);
269
270 return E_NOTIMPL;
271}
272static HRESULT WINAPI IStream_fnStat (IStream * iface, STATSTG* pstatstg, DWORD grfStatFlag)
273{
274 ICOM_THIS(ISHRegStream, iface);
275
276 TRACE("(%p)\n",This);
277
278 return E_NOTIMPL;
279}
280static HRESULT WINAPI IStream_fnClone (IStream * iface, IStream** ppstm)
281{
282 ICOM_THIS(ISHRegStream, iface);
283
284 TRACE("(%p)\n",This);
285
286 return E_NOTIMPL;
287}
288
289static struct ICOM_VTABLE(IStream) rstvt =
290{
291 ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
292 IStream_fnQueryInterface,
293 IStream_fnAddRef,
294 IStream_fnRelease,
295 IStream_fnRead,
296 IStream_fnWrite,
297 IStream_fnSeek,
298 IStream_fnSetSize,
299 IStream_fnCopyTo,
300 IStream_fnCommit,
301 IStream_fnRevert,
302 IStream_fnLockRegion,
303 IStream_fnUnlockRegion,
304 IStream_fnStat,
305 IStream_fnClone
306
307};
308
309/*************************************************************************
310 * SHOpenRegStreamA [SHLWAPI.@]
311 * SHOpenRegStream2A [SHLWAPI.@]
312 */
313IStream * WINAPI SHOpenRegStreamA(
314 HKEY hkey,
315 LPCSTR pszSubkey,
316 LPCSTR pszValue,
317 DWORD grfMode)
318{
319 TRACE("(0x%08x,%s,%s,0x%08lx)\n",
320 hkey, pszSubkey, pszValue, grfMode);
321
322 return IStream_ConstructorA(hkey, pszSubkey, pszValue, grfMode);
323}
324
325/*************************************************************************
326 * SHOpenRegStreamW [SHLWAPI.@]
327 * SHOpenRegStream2W [SHLWAPI.@]
328 */
329IStream * WINAPI SHOpenRegStreamW(
330 HKEY hkey,
331 LPCWSTR pszSubkey,
332 LPCWSTR pszValue,
333 DWORD grfMode)
334{
335 TRACE("(0x%08x,%s,%s,0x%08lx)\n",
336 hkey, debugstr_w(pszSubkey), debugstr_w(pszValue), grfMode);
337
338 return IStream_ConstructorW(hkey, pszSubkey, pszValue, grfMode);
339}
Note: See TracBrowser for help on using the repository browser.