source: trunk/src/shell32/memorystream.c@ 6688

Last change on this file since 6688 was 6650, checked in by bird, 24 years ago

Added $Id:$ keyword.
File size: 7.5 KB
Line 
1/* $Id: memorystream.c,v 1.3 2001-09-05 13:46:56 bird Exp $ */
2/*
3 * this class implements a pure IStream object
4 * and can be used for many purposes
5 *
6 * the main reason for implementing this was
7 * a cleaner implementation of IShellLink which
8 * needs to be able to load lnk's from a IStream
9 * interface so it was obvious to capsule the file
10 * access in a IStream to.
11 */
12
13#include <string.h>
14
15#include "winbase.h"
16#include "winerror.h"
17#include "shlobj.h"
18#include "debugtools.h"
19#include "heap.h"
20#include "shell32_main.h"
21
22DEFAULT_DEBUG_CHANNEL(shell);
23
24static HRESULT WINAPI IStream_fnQueryInterface(IStream *iface, REFIID riid, LPVOID *ppvObj);
25static ULONG WINAPI IStream_fnAddRef(IStream *iface);
26static ULONG WINAPI IStream_fnRelease(IStream *iface);
27static HRESULT WINAPI IStream_fnRead (IStream * iface, void* pv, ULONG cb, ULONG* pcbRead);
28static HRESULT WINAPI IStream_fnWrite (IStream * iface, const void* pv, ULONG cb, ULONG* pcbWritten);
29static HRESULT WINAPI IStream_fnSeek (IStream * iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition);
30static HRESULT WINAPI IStream_fnSetSize (IStream * iface, ULARGE_INTEGER libNewSize);
31static HRESULT WINAPI IStream_fnCopyTo (IStream * iface, IStream* pstm, ULARGE_INTEGER cb, ULARGE_INTEGER* pcbRead, ULARGE_INTEGER* pcbWritten);
32static HRESULT WINAPI IStream_fnCommit (IStream * iface, DWORD grfCommitFlags);
33static HRESULT WINAPI IStream_fnRevert (IStream * iface);
34static HRESULT WINAPI IStream_fnLockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
35static HRESULT WINAPI IStream_fnUnlockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
36static HRESULT WINAPI IStream_fnStat (IStream * iface, STATSTG* pstatstg, DWORD grfStatFlag);
37static HRESULT WINAPI IStream_fnClone (IStream * iface, IStream** ppstm);
38
39static ICOM_VTABLE(IStream) stvt =
40{
41 ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
42 IStream_fnQueryInterface,
43 IStream_fnAddRef,
44 IStream_fnRelease,
45 IStream_fnRead,
46 IStream_fnWrite,
47 IStream_fnSeek,
48 IStream_fnSetSize,
49 IStream_fnCopyTo,
50 IStream_fnCommit,
51 IStream_fnRevert,
52 IStream_fnLockRegion,
53 IStream_fnUnlockRegion,
54 IStream_fnStat,
55 IStream_fnClone
56
57};
58
59typedef struct
60{ ICOM_VTABLE(IStream) *lpvtst;
61 DWORD ref;
62 LPBYTE pImage;
63 HANDLE hMapping;
64 DWORD dwLength;
65 DWORD dwPos;
66} ISHFileStream;
67
68/**************************************************************************
69 * CreateStreamOnFile()
70 *
71 * similar to CreateStreamOnHGlobal
72 */
73HRESULT CreateStreamOnFile (LPCSTR pszFilename, IStream ** ppstm)
74{
75 ISHFileStream* fstr;
76 OFSTRUCT ofs;
77 HFILE hFile = OpenFile( pszFilename, &ofs, OF_READ );
78 HRESULT ret = E_FAIL;
79
80 fstr = (ISHFileStream*)HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(ISHFileStream));
81 fstr->lpvtst=&stvt;
82 fstr->ref = 1;
83 fstr->dwLength = GetFileSize (hFile, NULL);
84
85 shell32_ObjCount++;
86
87 if (!(fstr->hMapping = CreateFileMappingA(hFile,NULL,PAGE_READONLY|SEC_COMMIT,0,0,NULL)))
88 {
89 WARN("failed to create filemap.\n");
90 goto end_2;
91 }
92
93 if (!(fstr->pImage = MapViewOfFile(fstr->hMapping,FILE_MAP_READ,0,0,0)))
94 {
95 WARN("failed to mmap filemap.\n");
96 goto end_3;
97 }
98
99 ret = S_OK;
100 goto end_1;
101
102end_3: CloseHandle(fstr->hMapping);
103end_2: HeapFree(GetProcessHeap(), 0, fstr);
104 fstr = NULL;
105
106end_1: _lclose(hFile);
107 (*ppstm) = (IStream*)fstr;
108 return ret;
109}
110
111/**************************************************************************
112* IStream_fnQueryInterface
113*/
114static HRESULT WINAPI IStream_fnQueryInterface(IStream *iface, REFIID riid, LPVOID *ppvObj)
115{
116 ICOM_THIS(ISHFileStream, iface);
117
118 TRACE("(%p)->(\n\tIID:\t%s,%p)\n",This,debugstr_guid(riid),ppvObj);
119
120 *ppvObj = NULL;
121
122 if(IsEqualIID(riid, &IID_IUnknown) ||
123 IsEqualIID(riid, &IID_IStream))
124 {
125 *ppvObj = This;
126 }
127
128 if(*ppvObj)
129 {
130 IStream_AddRef((IStream*)*ppvObj);
131 TRACE("-- Interface: (%p)->(%p)\n",ppvObj,*ppvObj);
132 return S_OK;
133 }
134 TRACE("-- Interface: E_NOINTERFACE\n");
135 return E_NOINTERFACE;
136}
137
138/**************************************************************************
139* IStream_fnAddRef
140*/
141static ULONG WINAPI IStream_fnAddRef(IStream *iface)
142{
143 ICOM_THIS(ISHFileStream, iface);
144
145 TRACE("(%p)->(count=%lu)\n",This, This->ref);
146
147 shell32_ObjCount++;
148 return ++(This->ref);
149}
150
151/**************************************************************************
152* IStream_fnRelease
153*/
154static ULONG WINAPI IStream_fnRelease(IStream *iface)
155{
156 ICOM_THIS(ISHFileStream, iface);
157
158 TRACE("(%p)->()\n",This);
159
160 shell32_ObjCount--;
161
162 if (!--(This->ref))
163 { TRACE(" destroying SHFileStream (%p)\n",This);
164
165 UnmapViewOfFile(This->pImage);
166 CloseHandle(This->hMapping);
167
168 HeapFree(GetProcessHeap(),0,This);
169 return 0;
170 }
171 return This->ref;
172}
173
174static HRESULT WINAPI IStream_fnRead (IStream * iface, void* pv, ULONG cb, ULONG* pcbRead)
175{
176 ICOM_THIS(ISHFileStream, iface);
177
178 DWORD dwBytesToRead, dwBytesLeft;
179
180 TRACE("(%p)->(%p,0x%08lx,%p)\n",This, pv, cb, pcbRead);
181
182 if ( !pv )
183 return STG_E_INVALIDPOINTER;
184
185 dwBytesLeft = This->dwLength - This->dwPos;
186
187 if ( 0 >= dwBytesLeft ) /* end of buffer */
188 return S_FALSE;
189
190 dwBytesToRead = ( cb > dwBytesLeft) ? dwBytesLeft : cb;
191
192 memmove ( pv, (This->pImage) + (This->dwPos), dwBytesToRead);
193
194 This->dwPos += dwBytesToRead; /* adjust pointer */
195
196 if (pcbRead)
197 *pcbRead = dwBytesToRead;
198
199 return S_OK;
200}
201static HRESULT WINAPI IStream_fnWrite (IStream * iface, const void* pv, ULONG cb, ULONG* pcbWritten)
202{
203 ICOM_THIS(ISHFileStream, iface);
204
205 TRACE("(%p)\n",This);
206
207 return E_NOTIMPL;
208}
209static HRESULT WINAPI IStream_fnSeek (IStream * iface, LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER* plibNewPosition)
210{
211 ICOM_THIS(ISHFileStream, iface);
212
213 TRACE("(%p)\n",This);
214
215 return E_NOTIMPL;
216}
217static HRESULT WINAPI IStream_fnSetSize (IStream * iface, ULARGE_INTEGER libNewSize)
218{
219 ICOM_THIS(ISHFileStream, iface);
220
221 TRACE("(%p)\n",This);
222
223 return E_NOTIMPL;
224}
225static HRESULT WINAPI IStream_fnCopyTo (IStream * iface, IStream* pstm, ULARGE_INTEGER cb, ULARGE_INTEGER* pcbRead, ULARGE_INTEGER* pcbWritten)
226{
227 ICOM_THIS(ISHFileStream, iface);
228
229 TRACE("(%p)\n",This);
230
231 return E_NOTIMPL;
232}
233static HRESULT WINAPI IStream_fnCommit (IStream * iface, DWORD grfCommitFlags)
234{
235 ICOM_THIS(ISHFileStream, iface);
236
237 TRACE("(%p)\n",This);
238
239 return E_NOTIMPL;
240}
241static HRESULT WINAPI IStream_fnRevert (IStream * iface)
242{
243 ICOM_THIS(ISHFileStream, iface);
244
245 TRACE("(%p)\n",This);
246
247 return E_NOTIMPL;
248}
249static HRESULT WINAPI IStream_fnLockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
250{
251 ICOM_THIS(ISHFileStream, iface);
252
253 TRACE("(%p)\n",This);
254
255 return E_NOTIMPL;
256}
257static HRESULT WINAPI IStream_fnUnlockRegion (IStream * iface, ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
258{
259 ICOM_THIS(ISHFileStream, iface);
260
261 TRACE("(%p)\n",This);
262
263 return E_NOTIMPL;
264}
265static HRESULT WINAPI IStream_fnStat (IStream * iface, STATSTG* pstatstg, DWORD grfStatFlag)
266{
267 ICOM_THIS(ISHFileStream, iface);
268
269 TRACE("(%p)\n",This);
270
271 return E_NOTIMPL;
272}
273static HRESULT WINAPI IStream_fnClone (IStream * iface, IStream** ppstm)
274{
275 ICOM_THIS(ISHFileStream, iface);
276
277 TRACE("(%p)\n",This);
278
279 return E_NOTIMPL;
280}
Note: See TracBrowser for help on using the repository browser.