?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
d3dx8: Implement the C++ stuff of the D3DXVECTOR3 structure.
[wine/multimedia.git] / dlls / ole32 / stg_prop.c
blob76a86ca59f8ddccbf0e28dfa1d680131bb1ed991
1 /*
2 * Compound Storage (32 bit version)
3 * Storage implementation
4 *
5 * This file contains the compound file implementation
6 * of the storage interface.
7 *
8 * Copyright 1999 Francis Beaudet
9 * Copyright 1999 Sylvain St-Germain
10 * Copyright 1999 Thuy Nguyen
11 * Copyright 2005 Mike McCormack
12 * Copyright 2005 Juan Lang
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
27 *
28 * There's a decent overview of property set storage here:
29 * http://msdn.microsoft.com/archive/en-us/dnarolegen/html/msdn_propset.asp
30 * It's a little bit out of date, and more definitive references are given
31 * below, but it gives the best "big picture" that I've found.
32 *
33 * TODO:
34 * - I don't honor the maximum property set size.
35 * - Certain bogus files could result in reading past the end of a buffer.
36 * - Mac-generated files won't be read correctly, even if they're little
37 * endian, because I disregard whether the generator was a Mac. This means
38 * strings will probably be munged (as I don't understand Mac scripts.)
39 * - Not all PROPVARIANT types are supported.
40 * - User defined properties are not supported, see comment in
41 * PropertyStorage_ReadFromStream
42 */
43
44 #include <assert.h>
45 #include <stdarg.h>
46 #include <stdio.h>
47 #include <stdlib.h>
48 #include <string.h>
49
50 #define COBJMACROS
51 #define NONAMELESSUNION
52 #define NONAMELESSSTRUCT
53
54 #include "windef.h"
55 #include "winbase.h"
56 #include "winnls.h"
57 #include "winuser.h"
58 #include "wine/unicode.h"
59 #include "wine/debug.h"
60 #include "dictionary.h"
61 #include "storage32.h"
62 #include "enumx.h"
63
64 WINE_DEFAULT_DEBUG_CHANNEL(storage);
65
66 static inline StorageImpl *impl_from_IPropertySetStorage( IPropertySetStorage *iface )
67 {
68 return (StorageImpl *)((char*)iface - FIELD_OFFSET(StorageImpl, base.pssVtbl));
69 }
70
71 /* These are documented in MSDN, e.g.
72 * http://msdn.microsoft.com/library/en-us/stg/stg/property_set_header.asp
73 * http://msdn.microsoft.com/library/library/en-us/stg/stg/section.asp
74 * but they don't seem to be in any header file.
75 */
76 #define PROPSETHDR_BYTEORDER_MAGIC 0xfffe
77 #define PROPSETHDR_OSVER_KIND_WIN16 0
78 #define PROPSETHDR_OSVER_KIND_MAC 1
79 #define PROPSETHDR_OSVER_KIND_WIN32 2
80
81 #define CP_UNICODE 1200
82
83 #define MAX_VERSION_0_PROP_NAME_LENGTH 256
84
85 #define CFTAG_WINDOWS (-1L)
86 #define CFTAG_MACINTOSH (-2L)
87 #define CFTAG_FMTID (-3L)
88 #define CFTAG_NODATA 0L
89
90 /* The format version (and what it implies) is described here:
91 * http://msdn.microsoft.com/library/en-us/stg/stg/format_version.asp
92 */
93 typedef struct tagPROPERTYSETHEADER
94 {
95 WORD wByteOrder; /* always 0xfffe */
96 WORD wFormat; /* can be zero or one */
97 DWORD dwOSVer; /* OS version of originating system */
98 CLSID clsid; /* application CLSID */
99 DWORD reserved; /* always 1 */
100 } PROPERTYSETHEADER;
101
102 typedef struct tagFORMATIDOFFSET
103 {
104 FMTID fmtid;
105 DWORD dwOffset; /* from beginning of stream */
106 } FORMATIDOFFSET;
107
108 typedef struct tagPROPERTYSECTIONHEADER
109 {
110 DWORD cbSection;
111 DWORD cProperties;
112 } PROPERTYSECTIONHEADER;
113
114 typedef struct tagPROPERTYIDOFFSET
115 {
116 DWORD propid;
117 DWORD dwOffset; /* from beginning of section */
118 } PROPERTYIDOFFSET;
119
120 typedef struct tagPropertyStorage_impl PropertyStorage_impl;
121
122 /* Initializes the property storage from the stream (and undoes any uncommitted
123 * changes in the process.) Returns an error if there is an error reading or
124 * if the stream format doesn't match what's expected.
125 */
126 static HRESULT PropertyStorage_ReadFromStream(PropertyStorage_impl *);
127
128 static HRESULT PropertyStorage_WriteToStream(PropertyStorage_impl *);
129
130 /* Creates the dictionaries used by the property storage. If successful, all
131 * the dictionaries have been created. If failed, none has been. (This makes
132 * it a bit easier to deal with destroying them.)
133 */
134 static HRESULT PropertyStorage_CreateDictionaries(PropertyStorage_impl *);
135
136 static void PropertyStorage_DestroyDictionaries(PropertyStorage_impl *);
137
138 /* Copies from propvar to prop. If propvar's type is VT_LPSTR, copies the
139 * string using PropertyStorage_StringCopy.
140 */
141 static HRESULT PropertyStorage_PropVariantCopy(PROPVARIANT *prop,
142 const PROPVARIANT *propvar, LCID targetCP, LCID srcCP);
143
144 /* Copies the string src, which is encoded using code page srcCP, and returns
145 * it in *dst, in the code page specified by targetCP. The returned string is
146 * allocated using CoTaskMemAlloc.
147 * If srcCP is CP_UNICODE, src is in fact an LPCWSTR. Similarly, if targetCP
148 * is CP_UNICODE, the returned string is in fact an LPWSTR.
149 * Returns S_OK on success, something else on failure.
150 */
151 static HRESULT PropertyStorage_StringCopy(LPCSTR src, LCID srcCP, LPSTR *dst,
152 LCID targetCP);
153
154 static const IPropertyStorageVtbl IPropertyStorage_Vtbl;
155 static const IEnumSTATPROPSETSTGVtbl IEnumSTATPROPSETSTG_Vtbl;
156 static const IEnumSTATPROPSTGVtbl IEnumSTATPROPSTG_Vtbl;
157 static HRESULT create_EnumSTATPROPSETSTG(StorageImpl *, IEnumSTATPROPSETSTG**);
158 static HRESULT create_EnumSTATPROPSTG(PropertyStorage_impl *, IEnumSTATPROPSTG**);
159
160 /***********************************************************************
161 * Implementation of IPropertyStorage
162 */
163 struct tagPropertyStorage_impl
164 {
165 const IPropertyStorageVtbl *vtbl;
166 LONG ref;
167 CRITICAL_SECTION cs;
168 IStream *stm;
169 BOOL dirty;
170 FMTID fmtid;
171 CLSID clsid;
172 WORD format;
173 DWORD originatorOS;
174 DWORD grfFlags;
175 DWORD grfMode;
176 UINT codePage;
177 LCID locale;
178 PROPID highestProp;
179 struct dictionary *name_to_propid;
180 struct dictionary *propid_to_name;
181 struct dictionary *propid_to_prop;
182 };
183
184 /************************************************************************
185 * IPropertyStorage_fnQueryInterface (IPropertyStorage)
186 */
187 static HRESULT WINAPI IPropertyStorage_fnQueryInterface(
188 IPropertyStorage *iface,
189 REFIID riid,
190 void** ppvObject)
191 {
192 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
193
194 if ( (This==0) || (ppvObject==0) )
195 return E_INVALIDARG;
196
197 *ppvObject = 0;
198
199 if (IsEqualGUID(&IID_IUnknown, riid) ||
200 IsEqualGUID(&IID_IPropertyStorage, riid))
201 {
202 IPropertyStorage_AddRef(iface);
203 *ppvObject = (IPropertyStorage*)iface;
204 return S_OK;
205 }
206
207 return E_NOINTERFACE;
208 }
209
210 /************************************************************************
211 * IPropertyStorage_fnAddRef (IPropertyStorage)
212 */
213 static ULONG WINAPI IPropertyStorage_fnAddRef(
214 IPropertyStorage *iface)
215 {
216 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
217 return InterlockedIncrement(&This->ref);
218 }
219
220 /************************************************************************
221 * IPropertyStorage_fnRelease (IPropertyStorage)
222 */
223 static ULONG WINAPI IPropertyStorage_fnRelease(
224 IPropertyStorage *iface)
225 {
226 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
227 ULONG ref;
228
229 ref = InterlockedDecrement(&This->ref);
230 if (ref == 0)
231 {
232 TRACE("Destroying %p\n", This);
233 if (This->dirty)
234 IPropertyStorage_Commit(iface, STGC_DEFAULT);
235 IStream_Release(This->stm);
236 This->cs.DebugInfo->Spare[0] = 0;
237 DeleteCriticalSection(&This->cs);
238 PropertyStorage_DestroyDictionaries(This);
239 HeapFree(GetProcessHeap(), 0, This);
240 }
241 return ref;
242 }
243
244 static PROPVARIANT *PropertyStorage_FindProperty(PropertyStorage_impl *This,
245 DWORD propid)
246 {
247 PROPVARIANT *ret = NULL;
248
249 dictionary_find(This->propid_to_prop, (void *)propid, (void **)&ret);
250 TRACE("returning %p\n", ret);
251 return ret;
252 }
253
254 /* Returns NULL if name is NULL. */
255 static PROPVARIANT *PropertyStorage_FindPropertyByName(
256 PropertyStorage_impl *This, LPCWSTR name)
257 {
258 PROPVARIANT *ret = NULL;
259 PROPID propid;
260
261 if (!name)
262 return NULL;
263 if (This->codePage == CP_UNICODE)
264 {
265 if (dictionary_find(This->name_to_propid, name, (void **)&propid))
266 ret = PropertyStorage_FindProperty(This, (PROPID)propid);
267 }
268 else
269 {
270 LPSTR ansiName;
271 HRESULT hr = PropertyStorage_StringCopy((LPCSTR)name, CP_UNICODE,
272 &ansiName, This->codePage);
273
274 if (SUCCEEDED(hr))
275 {
276 if (dictionary_find(This->name_to_propid, ansiName,
277 (void **)&propid))
278 ret = PropertyStorage_FindProperty(This, (PROPID)propid);
279 CoTaskMemFree(ansiName);
280 }
281 }
282 TRACE("returning %p\n", ret);
283 return ret;
284 }
285
286 static LPWSTR PropertyStorage_FindPropertyNameById(PropertyStorage_impl *This,
287 DWORD propid)
288 {
289 LPWSTR ret = NULL;
290
291 dictionary_find(This->propid_to_name, (void *)propid, (void **)&ret);
292 TRACE("returning %p\n", ret);
293 return ret;
294 }
295
296 /************************************************************************
297 * IPropertyStorage_fnReadMultiple (IPropertyStorage)
298 */
299 static HRESULT WINAPI IPropertyStorage_fnReadMultiple(
300 IPropertyStorage* iface,
301 ULONG cpspec,
302 const PROPSPEC rgpspec[],
303 PROPVARIANT rgpropvar[])
304 {
305 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
306 HRESULT hr = S_OK;
307 ULONG i;
308
309 TRACE("(%p, %d, %p, %p)\n", iface, cpspec, rgpspec, rgpropvar);
310
311 if (!cpspec)
312 return S_FALSE;
313 if (!rgpspec || !rgpropvar)
314 return E_INVALIDARG;
315 EnterCriticalSection(&This->cs);
316 for (i = 0; i < cpspec; i++)
317 {
318 PropVariantInit(&rgpropvar[i]);
319 if (rgpspec[i].ulKind == PRSPEC_LPWSTR)
320 {
321 PROPVARIANT *prop = PropertyStorage_FindPropertyByName(This,
322 rgpspec[i].u.lpwstr);
323
324 if (prop)
325 PropertyStorage_PropVariantCopy(&rgpropvar[i], prop, GetACP(),
326 This->codePage);
327 }
328 else
329 {
330 switch (rgpspec[i].u.propid)
331 {
332 case PID_CODEPAGE:
333 rgpropvar[i].vt = VT_I2;
334 rgpropvar[i].u.iVal = This->codePage;
335 break;
336 case PID_LOCALE:
337 rgpropvar[i].vt = VT_I4;
338 rgpropvar[i].u.lVal = This->locale;
339 break;
340 default:
341 {
342 PROPVARIANT *prop = PropertyStorage_FindProperty(This,
343 rgpspec[i].u.propid);
344
345 if (prop)
346 PropertyStorage_PropVariantCopy(&rgpropvar[i], prop,
347 GetACP(), This->codePage);
348 else
349 hr = S_FALSE;
350 }
351 }
352 }
353 }
354 LeaveCriticalSection(&This->cs);
355 return hr;
356 }
357
358 static HRESULT PropertyStorage_StringCopy(LPCSTR src, LCID srcCP, LPSTR *dst,
359 LCID dstCP)
360 {
361 HRESULT hr = S_OK;
362 int len;
363
364 TRACE("%s, %p, %d, %d\n",
365 srcCP == CP_UNICODE ? debugstr_w((LPCWSTR)src) : debugstr_a(src), dst,
366 dstCP, srcCP);
367 assert(src);
368 assert(dst);
369 *dst = NULL;
370 if (dstCP == srcCP)
371 {
372 size_t len;
373
374 if (dstCP == CP_UNICODE)
375 len = (strlenW((LPCWSTR)src) + 1) * sizeof(WCHAR);
376 else
377 len = strlen(src) + 1;
378 *dst = CoTaskMemAlloc(len * sizeof(WCHAR));
379 if (!*dst)
380 hr = STG_E_INSUFFICIENTMEMORY;
381 else
382 memcpy(*dst, src, len);
383 }
384 else
385 {
386 if (dstCP == CP_UNICODE)
387 {
388 len = MultiByteToWideChar(srcCP, 0, src, -1, NULL, 0);
389 *dst = CoTaskMemAlloc(len * sizeof(WCHAR));
390 if (!*dst)
391 hr = STG_E_INSUFFICIENTMEMORY;
392 else
393 MultiByteToWideChar(srcCP, 0, src, -1, (LPWSTR)*dst, len);
394 }
395 else
396 {
397 LPCWSTR wideStr = NULL;
398 LPWSTR wideStr_tmp = NULL;
399
400 if (srcCP == CP_UNICODE)
401 wideStr = (LPCWSTR)src;
402 else
403 {
404 len = MultiByteToWideChar(srcCP, 0, src, -1, NULL, 0);
405 wideStr_tmp = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
406 if (wideStr_tmp)
407 {
408 MultiByteToWideChar(srcCP, 0, src, -1, wideStr_tmp, len);
409 wideStr = wideStr_tmp;
410 }
411 else
412 hr = STG_E_INSUFFICIENTMEMORY;
413 }
414 if (SUCCEEDED(hr))
415 {
416 len = WideCharToMultiByte(dstCP, 0, wideStr, -1, NULL, 0,
417 NULL, NULL);
418 *dst = CoTaskMemAlloc(len);
419 if (!*dst)
420 hr = STG_E_INSUFFICIENTMEMORY;
421 else
422 {
423 BOOL defCharUsed = FALSE;
424
425 if (WideCharToMultiByte(dstCP, 0, wideStr, -1, *dst, len,
426 NULL, &defCharUsed) == 0 || defCharUsed)
427 {
428 CoTaskMemFree(*dst);
429 *dst = NULL;
430 hr = HRESULT_FROM_WIN32(ERROR_NO_UNICODE_TRANSLATION);
431 }
432 }
433 }
434 HeapFree(GetProcessHeap(), 0, wideStr_tmp);
435 }
436 }
437 TRACE("returning 0x%08x (%s)\n", hr,
438 dstCP == CP_UNICODE ? debugstr_w((LPCWSTR)*dst) : debugstr_a(*dst));
439 return hr;
440 }
441
442 static HRESULT PropertyStorage_PropVariantCopy(PROPVARIANT *prop,
443 const PROPVARIANT *propvar, LCID targetCP, LCID srcCP)
444 {
445 HRESULT hr = S_OK;
446
447 assert(prop);
448 assert(propvar);
449 if (propvar->vt == VT_LPSTR)
450 {
451 hr = PropertyStorage_StringCopy(propvar->u.pszVal, srcCP,
452 &prop->u.pszVal, targetCP);
453 if (SUCCEEDED(hr))
454 prop->vt = VT_LPSTR;
455 }
456 else
457 PropVariantCopy(prop, propvar);
458 return hr;
459 }
460
461 /* Stores the property with id propid and value propvar into this property
462 * storage. lcid is ignored if propvar's type is not VT_LPSTR. If propvar's
463 * type is VT_LPSTR, converts the string using lcid as the source code page
464 * and This->codePage as the target code page before storing.
465 * As a side effect, may change This->format to 1 if the type of propvar is
466 * a version 1-only property.
467 */
468 static HRESULT PropertyStorage_StorePropWithId(PropertyStorage_impl *This,
469 PROPID propid, const PROPVARIANT *propvar, LCID lcid)
470 {
471 HRESULT hr = S_OK;
472 PROPVARIANT *prop = PropertyStorage_FindProperty(This, propid);
473
474 assert(propvar);
475 if (propvar->vt & VT_BYREF || propvar->vt & VT_ARRAY)
476 This->format = 1;
477 switch (propvar->vt)
478 {
479 case VT_DECIMAL:
480 case VT_I1:
481 case VT_INT:
482 case VT_UINT:
483 case VT_VECTOR|VT_I1:
484 This->format = 1;
485 }
486 TRACE("Setting 0x%08x to type %d\n", propid, propvar->vt);
487 if (prop)
488 {
489 PropVariantClear(prop);
490 hr = PropertyStorage_PropVariantCopy(prop, propvar, This->codePage,
491 lcid);
492 }
493 else
494 {
495 prop = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
496 sizeof(PROPVARIANT));
497 if (prop)
498 {
499 hr = PropertyStorage_PropVariantCopy(prop, propvar, This->codePage,
500 lcid);
501 if (SUCCEEDED(hr))
502 {
503 dictionary_insert(This->propid_to_prop, (void *)propid, prop);
504 if (propid > This->highestProp)
505 This->highestProp = propid;
506 }
507 else
508 HeapFree(GetProcessHeap(), 0, prop);
509 }
510 else
511 hr = STG_E_INSUFFICIENTMEMORY;
512 }
513 return hr;
514 }
515
516 /* Adds the name srcName to the name dictionaries, mapped to property ID id.
517 * srcName is encoded in code page cp, and is converted to This->codePage.
518 * If cp is CP_UNICODE, srcName is actually a unicode string.
519 * As a side effect, may change This->format to 1 if srcName is too long for
520 * a version 0 property storage.
521 * Doesn't validate id.
522 */
523 static HRESULT PropertyStorage_StoreNameWithId(PropertyStorage_impl *This,
524 LPCSTR srcName, LCID cp, PROPID id)
525 {
526 LPSTR name;
527 HRESULT hr;
528
529 assert(srcName);
530
531 hr = PropertyStorage_StringCopy((LPCSTR)srcName, cp, &name, This->codePage);
532 if (SUCCEEDED(hr))
533 {
534 if (This->codePage == CP_UNICODE)
535 {
536 if (lstrlenW((LPWSTR)name) >= MAX_VERSION_0_PROP_NAME_LENGTH)
537 This->format = 1;
538 }
539 else
540 {
541 if (strlen(name) >= MAX_VERSION_0_PROP_NAME_LENGTH)
542 This->format = 1;
543 }
544 TRACE("Adding prop name %s, propid %d\n",
545 This->codePage == CP_UNICODE ? debugstr_w((LPCWSTR)name) :
546 debugstr_a(name), id);
547 dictionary_insert(This->name_to_propid, name, (void *)id);
548 dictionary_insert(This->propid_to_name, (void *)id, name);
549 }
550 return hr;
551 }
552
553 /************************************************************************
554 * IPropertyStorage_fnWriteMultiple (IPropertyStorage)
555 */
556 static HRESULT WINAPI IPropertyStorage_fnWriteMultiple(
557 IPropertyStorage* iface,
558 ULONG cpspec,
559 const PROPSPEC rgpspec[],
560 const PROPVARIANT rgpropvar[],
561 PROPID propidNameFirst)
562 {
563 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
564 HRESULT hr = S_OK;
565 ULONG i;
566
567 TRACE("(%p, %d, %p, %p)\n", iface, cpspec, rgpspec, rgpropvar);
568
569 if (cpspec && (!rgpspec || !rgpropvar))
570 return E_INVALIDARG;
571 if (!(This->grfMode & STGM_READWRITE))
572 return STG_E_ACCESSDENIED;
573 EnterCriticalSection(&This->cs);
574 This->dirty = TRUE;
575 This->originatorOS = (DWORD)MAKELONG(LOWORD(GetVersion()),
576 PROPSETHDR_OSVER_KIND_WIN32) ;
577 for (i = 0; i < cpspec; i++)
578 {
579 if (rgpspec[i].ulKind == PRSPEC_LPWSTR)
580 {
581 PROPVARIANT *prop = PropertyStorage_FindPropertyByName(This,
582 rgpspec[i].u.lpwstr);
583
584 if (prop)
585 PropVariantCopy(prop, &rgpropvar[i]);
586 else
587 {
588 /* Note that I don't do the special cases here that I do below,
589 * because naming the special PIDs isn't supported.
590 */
591 if (propidNameFirst < PID_FIRST_USABLE ||
592 propidNameFirst >= PID_MIN_READONLY)
593 hr = STG_E_INVALIDPARAMETER;
594 else
595 {
596 PROPID nextId = max(propidNameFirst, This->highestProp + 1);
597
598 hr = PropertyStorage_StoreNameWithId(This,
599 (LPCSTR)rgpspec[i].u.lpwstr, CP_UNICODE, nextId);
600 if (SUCCEEDED(hr))
601 hr = PropertyStorage_StorePropWithId(This, nextId,
602 &rgpropvar[i], GetACP());
603 }
604 }
605 }
606 else
607 {
608 switch (rgpspec[i].u.propid)
609 {
610 case PID_DICTIONARY:
611 /* Can't set the dictionary */
612 hr = STG_E_INVALIDPARAMETER;
613 break;
614 case PID_CODEPAGE:
615 /* Can only set the code page if nothing else has been set */
616 if (dictionary_num_entries(This->propid_to_prop) == 0 &&
617 rgpropvar[i].vt == VT_I2)
618 {
619 This->codePage = rgpropvar[i].u.iVal;
620 if (This->codePage == CP_UNICODE)
621 This->grfFlags &= ~PROPSETFLAG_ANSI;
622 else
623 This->grfFlags |= PROPSETFLAG_ANSI;
624 }
625 else
626 hr = STG_E_INVALIDPARAMETER;
627 break;
628 case PID_LOCALE:
629 /* Can only set the locale if nothing else has been set */
630 if (dictionary_num_entries(This->propid_to_prop) == 0 &&
631 rgpropvar[i].vt == VT_I4)
632 This->locale = rgpropvar[i].u.lVal;
633 else
634 hr = STG_E_INVALIDPARAMETER;
635 break;
636 case PID_ILLEGAL:
637 /* silently ignore like MSDN says */
638 break;
639 default:
640 if (rgpspec[i].u.propid >= PID_MIN_READONLY)
641 hr = STG_E_INVALIDPARAMETER;
642 else
643 hr = PropertyStorage_StorePropWithId(This,
644 rgpspec[i].u.propid, &rgpropvar[i], GetACP());
645 }
646 }
647 }
648 if (This->grfFlags & PROPSETFLAG_UNBUFFERED)
649 IPropertyStorage_Commit(iface, STGC_DEFAULT);
650 LeaveCriticalSection(&This->cs);
651 return hr;
652 }
653
654 /************************************************************************
655 * IPropertyStorage_fnDeleteMultiple (IPropertyStorage)
656 */
657 static HRESULT WINAPI IPropertyStorage_fnDeleteMultiple(
658 IPropertyStorage* iface,
659 ULONG cpspec,
660 const PROPSPEC rgpspec[])
661 {
662 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
663 ULONG i;
664 HRESULT hr;
665
666 TRACE("(%p, %d, %p)\n", iface, cpspec, rgpspec);
667
668 if (cpspec && !rgpspec)
669 return E_INVALIDARG;
670 if (!(This->grfMode & STGM_READWRITE))
671 return STG_E_ACCESSDENIED;
672 hr = S_OK;
673 EnterCriticalSection(&This->cs);
674 This->dirty = TRUE;
675 for (i = 0; i < cpspec; i++)
676 {
677 if (rgpspec[i].ulKind == PRSPEC_LPWSTR)
678 {
679 PROPID propid;
680
681 if (dictionary_find(This->name_to_propid,
682 (void *)rgpspec[i].u.lpwstr, (void **)&propid))
683 dictionary_remove(This->propid_to_prop, (void *)propid);
684 }
685 else
686 {
687 if (rgpspec[i].u.propid >= PID_FIRST_USABLE &&
688 rgpspec[i].u.propid < PID_MIN_READONLY)
689 dictionary_remove(This->propid_to_prop,
690 (void *)rgpspec[i].u.propid);
691 else
692 hr = STG_E_INVALIDPARAMETER;
693 }
694 }
695 if (This->grfFlags & PROPSETFLAG_UNBUFFERED)
696 IPropertyStorage_Commit(iface, STGC_DEFAULT);
697 LeaveCriticalSection(&This->cs);
698 return hr;
699 }
700
701 /************************************************************************
702 * IPropertyStorage_fnReadPropertyNames (IPropertyStorage)
703 */
704 static HRESULT WINAPI IPropertyStorage_fnReadPropertyNames(
705 IPropertyStorage* iface,
706 ULONG cpropid,
707 const PROPID rgpropid[],
708 LPOLESTR rglpwstrName[])
709 {
710 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
711 ULONG i;
712 HRESULT hr = S_FALSE;
713
714 TRACE("(%p, %d, %p, %p)\n", iface, cpropid, rgpropid, rglpwstrName);
715
716 if (cpropid && (!rgpropid || !rglpwstrName))
717 return E_INVALIDARG;
718 EnterCriticalSection(&This->cs);
719 for (i = 0; i < cpropid && SUCCEEDED(hr); i++)
720 {
721 LPWSTR name = PropertyStorage_FindPropertyNameById(This, rgpropid[i]);
722
723 if (name)
724 {
725 size_t len = lstrlenW(name);
726
727 hr = S_OK;
728 rglpwstrName[i] = CoTaskMemAlloc((len + 1) * sizeof(WCHAR));
729 if (rglpwstrName)
730 memcpy(rglpwstrName, name, (len + 1) * sizeof(WCHAR));
731 else
732 hr = STG_E_INSUFFICIENTMEMORY;
733 }
734 else
735 rglpwstrName[i] = NULL;
736 }
737 LeaveCriticalSection(&This->cs);
738 return hr;
739 }
740
741 /************************************************************************
742 * IPropertyStorage_fnWritePropertyNames (IPropertyStorage)
743 */
744 static HRESULT WINAPI IPropertyStorage_fnWritePropertyNames(
745 IPropertyStorage* iface,
746 ULONG cpropid,
747 const PROPID rgpropid[],
748 const LPOLESTR rglpwstrName[])
749 {
750 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
751 ULONG i;
752 HRESULT hr;
753
754 TRACE("(%p, %d, %p, %p)\n", iface, cpropid, rgpropid, rglpwstrName);
755
756 if (cpropid && (!rgpropid || !rglpwstrName))
757 return E_INVALIDARG;
758 if (!(This->grfMode & STGM_READWRITE))
759 return STG_E_ACCESSDENIED;
760 hr = S_OK;
761 EnterCriticalSection(&This->cs);
762 This->dirty = TRUE;
763 for (i = 0; SUCCEEDED(hr) && i < cpropid; i++)
764 {
765 if (rgpropid[i] != PID_ILLEGAL)
766 hr = PropertyStorage_StoreNameWithId(This, (LPCSTR)rglpwstrName[i],
767 CP_UNICODE, rgpropid[i]);
768 }
769 if (This->grfFlags & PROPSETFLAG_UNBUFFERED)
770 IPropertyStorage_Commit(iface, STGC_DEFAULT);
771 LeaveCriticalSection(&This->cs);
772 return hr;
773 }
774
775 /************************************************************************
776 * IPropertyStorage_fnDeletePropertyNames (IPropertyStorage)
777 */
778 static HRESULT WINAPI IPropertyStorage_fnDeletePropertyNames(
779 IPropertyStorage* iface,
780 ULONG cpropid,
781 const PROPID rgpropid[])
782 {
783 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
784 ULONG i;
785 HRESULT hr;
786
787 TRACE("(%p, %d, %p)\n", iface, cpropid, rgpropid);
788
789 if (cpropid && !rgpropid)
790 return E_INVALIDARG;
791 if (!(This->grfMode & STGM_READWRITE))
792 return STG_E_ACCESSDENIED;
793 hr = S_OK;
794 EnterCriticalSection(&This->cs);
795 This->dirty = TRUE;
796 for (i = 0; i < cpropid; i++)
797 {
798 LPWSTR name = NULL;
799
800 if (dictionary_find(This->propid_to_name, (void *)rgpropid[i],
801 (void **)&name))
802 {
803 dictionary_remove(This->propid_to_name, (void *)rgpropid[i]);
804 dictionary_remove(This->name_to_propid, name);
805 }
806 }
807 if (This->grfFlags & PROPSETFLAG_UNBUFFERED)
808 IPropertyStorage_Commit(iface, STGC_DEFAULT);
809 LeaveCriticalSection(&This->cs);
810 return hr;
811 }
812
813 /************************************************************************
814 * IPropertyStorage_fnCommit (IPropertyStorage)
815 */
816 static HRESULT WINAPI IPropertyStorage_fnCommit(
817 IPropertyStorage* iface,
818 DWORD grfCommitFlags)
819 {
820 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
821 HRESULT hr;
822
823 TRACE("(%p, 0x%08x)\n", iface, grfCommitFlags);
824
825 if (!(This->grfMode & STGM_READWRITE))
826 return STG_E_ACCESSDENIED;
827 EnterCriticalSection(&This->cs);
828 if (This->dirty)
829 hr = PropertyStorage_WriteToStream(This);
830 else
831 hr = S_OK;
832 LeaveCriticalSection(&This->cs);
833 return hr;
834 }
835
836 /************************************************************************
837 * IPropertyStorage_fnRevert (IPropertyStorage)
838 */
839 static HRESULT WINAPI IPropertyStorage_fnRevert(
840 IPropertyStorage* iface)
841 {
842 HRESULT hr;
843 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
844
845 TRACE("%p\n", iface);
846
847 EnterCriticalSection(&This->cs);
848 if (This->dirty)
849 {
850 PropertyStorage_DestroyDictionaries(This);
851 hr = PropertyStorage_CreateDictionaries(This);
852 if (SUCCEEDED(hr))
853 hr = PropertyStorage_ReadFromStream(This);
854 }
855 else
856 hr = S_OK;
857 LeaveCriticalSection(&This->cs);
858 return hr;
859 }
860
861 /************************************************************************
862 * IPropertyStorage_fnEnum (IPropertyStorage)
863 */
864 static HRESULT WINAPI IPropertyStorage_fnEnum(
865 IPropertyStorage* iface,
866 IEnumSTATPROPSTG** ppenum)
867 {
868 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
869 return create_EnumSTATPROPSTG(This, ppenum);
870 }
871
872 /************************************************************************
873 * IPropertyStorage_fnSetTimes (IPropertyStorage)
874 */
875 static HRESULT WINAPI IPropertyStorage_fnSetTimes(
876 IPropertyStorage* iface,
877 const FILETIME* pctime,
878 const FILETIME* patime,
879 const FILETIME* pmtime)
880 {
881 FIXME("\n");
882 return E_NOTIMPL;
883 }
884
885 /************************************************************************
886 * IPropertyStorage_fnSetClass (IPropertyStorage)
887 */
888 static HRESULT WINAPI IPropertyStorage_fnSetClass(
889 IPropertyStorage* iface,
890 REFCLSID clsid)
891 {
892 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
893
894 TRACE("%p, %s\n", iface, debugstr_guid(clsid));
895
896 if (!clsid)
897 return E_INVALIDARG;
898 if (!(This->grfMode & STGM_READWRITE))
899 return STG_E_ACCESSDENIED;
900 memcpy(&This->clsid, clsid, sizeof(This->clsid));
901 This->dirty = TRUE;
902 if (This->grfFlags & PROPSETFLAG_UNBUFFERED)
903 IPropertyStorage_Commit(iface, STGC_DEFAULT);
904 return S_OK;
905 }
906
907 /************************************************************************
908 * IPropertyStorage_fnStat (IPropertyStorage)
909 */
910 static HRESULT WINAPI IPropertyStorage_fnStat(
911 IPropertyStorage* iface,
912 STATPROPSETSTG* statpsstg)
913 {
914 PropertyStorage_impl *This = (PropertyStorage_impl *)iface;
915 STATSTG stat;
916 HRESULT hr;
917
918 TRACE("%p, %p\n", iface, statpsstg);
919
920 if (!statpsstg)
921 return E_INVALIDARG;
922
923 hr = IStream_Stat(This->stm, &stat, STATFLAG_NONAME);
924 if (SUCCEEDED(hr))
925 {
926 memcpy(&statpsstg->fmtid, &This->fmtid, sizeof(statpsstg->fmtid));
927 memcpy(&statpsstg->clsid, &This->clsid, sizeof(statpsstg->clsid));
928 statpsstg->grfFlags = This->grfFlags;
929 memcpy(&statpsstg->mtime, &stat.mtime, sizeof(statpsstg->mtime));
930 memcpy(&statpsstg->ctime, &stat.ctime, sizeof(statpsstg->ctime));
931 memcpy(&statpsstg->atime, &stat.atime, sizeof(statpsstg->atime));
932 statpsstg->dwOSVersion = This->originatorOS;
933 }
934 return hr;
935 }
936
937 static int PropertyStorage_PropNameCompare(const void *a, const void *b,
938 void *extra)
939 {
940 PropertyStorage_impl *This = (PropertyStorage_impl *)extra;
941
942 if (This->codePage == CP_UNICODE)
943 {
944 TRACE("(%s, %s)\n", debugstr_w(a), debugstr_w(b));
945 if (This->grfFlags & PROPSETFLAG_CASE_SENSITIVE)
946 return lstrcmpW((LPCWSTR)a, (LPCWSTR)b);
947 else
948 return lstrcmpiW((LPCWSTR)a, (LPCWSTR)b);
949 }
950 else
951 {
952 TRACE("(%s, %s)\n", debugstr_a(a), debugstr_a(b));
953 if (This->grfFlags & PROPSETFLAG_CASE_SENSITIVE)
954 return lstrcmpA((LPCSTR)a, (LPCSTR)b);
955 else
956 return lstrcmpiA((LPCSTR)a, (LPCSTR)b);
957 }
958 }
959
960 static void PropertyStorage_PropNameDestroy(void *k, void *d, void *extra)
961 {
962 CoTaskMemFree(k);
963 }
964
965 static int PropertyStorage_PropCompare(const void *a, const void *b,
966 void *extra)
967 {
968 TRACE("(%d, %d)\n", (PROPID)a, (PROPID)b);
969 return (PROPID)a - (PROPID)b;
970 }
971
972 static void PropertyStorage_PropertyDestroy(void *k, void *d, void *extra)
973 {
974 PropVariantClear((PROPVARIANT *)d);
975 HeapFree(GetProcessHeap(), 0, d);
976 }
977
978 #ifdef WORDS_BIGENDIAN
979 /* Swaps each character in str to or from little endian; assumes the conversion
980 * is symmetric, that is, that le16toh is equivalent to htole16.
981 */
982 static void PropertyStorage_ByteSwapString(LPWSTR str, size_t len)
983 {
984 DWORD i;
985
986 /* Swap characters to host order.
987 * FIXME: alignment?
988 */
989 for (i = 0; i < len; i++)
990 str[i] = le16toh(str[i]);
991 }
992 #else
993 #define PropertyStorage_ByteSwapString(s, l)
994 #endif
995
996 /* Reads the dictionary from the memory buffer beginning at ptr. Interprets
997 * the entries according to the values of This->codePage and This->locale.
998 * FIXME: there isn't any checking whether the read property extends past the
999 * end of the buffer.
1000 */
1001 static HRESULT PropertyStorage_ReadDictionary(PropertyStorage_impl *This,
1002 BYTE *ptr)
1003 {
1004 DWORD numEntries, i;
1005 HRESULT hr = S_OK;
1006
1007 assert(This->name_to_propid);
1008 assert(This->propid_to_name);
1009
1010 StorageUtl_ReadDWord(ptr, 0, &numEntries);
1011 TRACE("Reading %d entries:\n", numEntries);
1012 ptr += sizeof(DWORD);
1013 for (i = 0; SUCCEEDED(hr) && i < numEntries; i++)
1014 {
1015 PROPID propid;
1016 DWORD cbEntry;
1017
1018 StorageUtl_ReadDWord(ptr, 0, &propid);
1019 ptr += sizeof(PROPID);
1020 StorageUtl_ReadDWord(ptr, 0, &cbEntry);
1021 ptr += sizeof(DWORD);
1022 TRACE("Reading entry with ID 0x%08x, %d bytes\n", propid, cbEntry);
1023 /* Make sure the source string is NULL-terminated */
1024 if (This->codePage != CP_UNICODE)
1025 ptr[cbEntry - 1] = '\0';
1026 else
1027 *((LPWSTR)ptr + cbEntry / sizeof(WCHAR)) = '\0';
1028 hr = PropertyStorage_StoreNameWithId(This, (char*)ptr, This->codePage, propid);
1029 if (This->codePage == CP_UNICODE)
1030 {
1031 /* Unicode entries are padded to DWORD boundaries */
1032 if (cbEntry % sizeof(DWORD))
1033 ptr += sizeof(DWORD) - (cbEntry % sizeof(DWORD));
1034 }
1035 ptr += sizeof(DWORD) + cbEntry;
1036 }
1037 return hr;
1038 }
1039
1040 /* FIXME: there isn't any checking whether the read property extends past the
1041 * end of the buffer.
1042 */
1043 static HRESULT PropertyStorage_ReadProperty(PropertyStorage_impl *This,
1044 PROPVARIANT *prop, const BYTE *data)
1045 {
1046 HRESULT hr = S_OK;
1047
1048 assert(prop);
1049 assert(data);
1050 StorageUtl_ReadDWord(data, 0, (DWORD *)&prop->vt);
1051 data += sizeof(DWORD);
1052 switch (prop->vt)
1053 {
1054 case VT_EMPTY:
1055 case VT_NULL:
1056 break;
1057 case VT_I1:
1058 prop->u.cVal = *(const char *)data;
1059 TRACE("Read char 0x%x ('%c')\n", prop->u.cVal, prop->u.cVal);
1060 break;
1061 case VT_UI1:
1062 prop->u.bVal = *(const UCHAR *)data;
1063 TRACE("Read byte 0x%x\n", prop->u.bVal);
1064 break;
1065 case VT_I2:
1066 StorageUtl_ReadWord(data, 0, (WORD*)&prop->u.iVal);
1067 TRACE("Read short %d\n", prop->u.iVal);
1068 break;
1069 case VT_UI2:
1070 StorageUtl_ReadWord(data, 0, &prop->u.uiVal);
1071 TRACE("Read ushort %d\n", prop->u.uiVal);
1072 break;
1073 case VT_INT:
1074 case VT_I4:
1075 StorageUtl_ReadDWord(data, 0, (DWORD*)&prop->u.lVal);
1076 TRACE("Read long %ld\n", prop->u.lVal);
1077 break;
1078 case VT_UINT:
1079 case VT_UI4:
1080 StorageUtl_ReadDWord(data, 0, &prop->u.ulVal);
1081 TRACE("Read ulong %d\n", prop->u.ulVal);
1082 break;
1083 case VT_LPSTR:
1084 {
1085 DWORD count;
1086
1087 StorageUtl_ReadDWord(data, 0, &count);
1088 if (This->codePage == CP_UNICODE && count / 2)
1089 {
1090 WARN("Unicode string has odd number of bytes\n");
1091 hr = STG_E_INVALIDHEADER;
1092 }
1093 else
1094 {
1095 prop->u.pszVal = CoTaskMemAlloc(count);
1096 if (prop->u.pszVal)
1097 {
1098 memcpy(prop->u.pszVal, data + sizeof(DWORD), count);
1099 /* This is stored in the code page specified in This->codePage.
1100 * Don't convert it, the caller will just store it as-is.
1101 */
1102 if (This->codePage == CP_UNICODE)
1103 {
1104 /* Make sure it's NULL-terminated */
1105 prop->u.pszVal[count / sizeof(WCHAR) - 1] = '\0';
1106 TRACE("Read string value %s\n",
1107 debugstr_w(prop->u.pwszVal));
1108 }
1109 else
1110 {
1111 /* Make sure it's NULL-terminated */
1112 prop->u.pszVal[count - 1] = '\0';
1113 TRACE("Read string value %s\n", debugstr_a(prop->u.pszVal));
1114 }
1115 }
1116 else
1117 hr = STG_E_INSUFFICIENTMEMORY;
1118 }
1119 break;
1120 }
1121 case VT_LPWSTR:
1122 {
1123 DWORD count;
1124
1125 StorageUtl_ReadDWord(data, 0, &count);
1126 prop->u.pwszVal = CoTaskMemAlloc(count * sizeof(WCHAR));
1127 if (prop->u.pwszVal)
1128 {
1129 memcpy(prop->u.pwszVal, data + sizeof(DWORD),
1130 count * sizeof(WCHAR));
1131 /* make sure string is NULL-terminated */
1132 prop->u.pwszVal[count - 1] = '\0';
1133 PropertyStorage_ByteSwapString(prop->u.pwszVal, count);
1134 TRACE("Read string value %s\n", debugstr_w(prop->u.pwszVal));
1135 }
1136 else
1137 hr = STG_E_INSUFFICIENTMEMORY;
1138 break;
1139 }
1140 case VT_FILETIME:
1141 StorageUtl_ReadULargeInteger(data, 0,
1142 (ULARGE_INTEGER *)&prop->u.filetime);
1143 break;
1144 case VT_CF:
1145 {
1146 DWORD len = 0, tag = 0;
1147
1148 StorageUtl_ReadDWord(data, 0, &len);
1149 StorageUtl_ReadDWord(data, 4, &tag);
1150 if (len > 8)
1151 {
1152 len -= 8;
1153 prop->u.pclipdata = CoTaskMemAlloc(sizeof (CLIPDATA));
1154 prop->u.pclipdata->cbSize = len;
1155 prop->u.pclipdata->ulClipFmt = tag;
1156 prop->u.pclipdata->pClipData = CoTaskMemAlloc(len);
1157 memcpy(prop->u.pclipdata->pClipData, data+8, len);
1158 }
1159 else
1160 hr = STG_E_INVALIDPARAMETER;
1161 }
1162 break;
1163 default:
1164 FIXME("unsupported type %d\n", prop->vt);
1165 hr = STG_E_INVALIDPARAMETER;
1166 }
1167 return hr;
1168 }
1169
1170 static HRESULT PropertyStorage_ReadHeaderFromStream(IStream *stm,
1171 PROPERTYSETHEADER *hdr)
1172 {
1173 BYTE buf[sizeof(PROPERTYSETHEADER)];
1174 ULONG count = 0;
1175 HRESULT hr;
1176
1177 assert(stm);
1178 assert(hdr);
1179 hr = IStream_Read(stm, buf, sizeof(buf), &count);
1180 if (SUCCEEDED(hr))
1181 {
1182 if (count != sizeof(buf))
1183 {
1184 WARN("read only %d\n", count);
1185 hr = STG_E_INVALIDHEADER;
1186 }
1187 else
1188 {
1189 StorageUtl_ReadWord(buf, offsetof(PROPERTYSETHEADER, wByteOrder),
1190 &hdr->wByteOrder);
1191 StorageUtl_ReadWord(buf, offsetof(PROPERTYSETHEADER, wFormat),
1192 &hdr->wFormat);
1193 StorageUtl_ReadDWord(buf, offsetof(PROPERTYSETHEADER, dwOSVer),
1194 &hdr->dwOSVer);
1195 StorageUtl_ReadGUID(buf, offsetof(PROPERTYSETHEADER, clsid),
1196 &hdr->clsid);
1197 StorageUtl_ReadDWord(buf, offsetof(PROPERTYSETHEADER, reserved),
1198 &hdr->reserved);
1199 }
1200 }
1201 TRACE("returning 0x%08x\n", hr);
1202 return hr;
1203 }
1204
1205 static HRESULT PropertyStorage_ReadFmtIdOffsetFromStream(IStream *stm,
1206 FORMATIDOFFSET *fmt)
1207 {
1208 BYTE buf[sizeof(FORMATIDOFFSET)];
1209 ULONG count = 0;
1210 HRESULT hr;
1211
1212 assert(stm);
1213 assert(fmt);
1214 hr = IStream_Read(stm, buf, sizeof(buf), &count);
1215 if (SUCCEEDED(hr))
1216 {
1217 if (count != sizeof(buf))
1218 {
1219 WARN("read only %d\n", count);
1220 hr = STG_E_INVALIDHEADER;
1221 }
1222 else
1223 {
1224 StorageUtl_ReadGUID(buf, offsetof(FORMATIDOFFSET, fmtid),
1225 &fmt->fmtid);
1226 StorageUtl_ReadDWord(buf, offsetof(FORMATIDOFFSET, dwOffset),
1227 &fmt->dwOffset);
1228 }
1229 }
1230 TRACE("returning 0x%08x\n", hr);
1231 return hr;
1232 }
1233
1234 static HRESULT PropertyStorage_ReadSectionHeaderFromStream(IStream *stm,
1235 PROPERTYSECTIONHEADER *hdr)
1236 {
1237 BYTE buf[sizeof(PROPERTYSECTIONHEADER)];
1238 ULONG count = 0;
1239 HRESULT hr;
1240
1241 assert(stm);
1242 assert(hdr);
1243 hr = IStream_Read(stm, buf, sizeof(buf), &count);
1244 if (SUCCEEDED(hr))
1245 {
1246 if (count != sizeof(buf))
1247 {
1248 WARN("read only %d\n", count);
1249 hr = STG_E_INVALIDHEADER;
1250 }
1251 else
1252 {
1253 StorageUtl_ReadDWord(buf, offsetof(PROPERTYSECTIONHEADER,
1254 cbSection), &hdr->cbSection);
1255 StorageUtl_ReadDWord(buf, offsetof(PROPERTYSECTIONHEADER,
1256 cProperties), &hdr->cProperties);
1257 }
1258 }
1259 TRACE("returning 0x%08x\n", hr);
1260 return hr;
1261 }
1262
1263 static HRESULT PropertyStorage_ReadFromStream(PropertyStorage_impl *This)
1264 {
1265 PROPERTYSETHEADER hdr;
1266 FORMATIDOFFSET fmtOffset;
1267 PROPERTYSECTIONHEADER sectionHdr;
1268 LARGE_INTEGER seek;
1269 ULONG i;
1270 STATSTG stat;
1271 HRESULT hr;
1272 BYTE *buf = NULL;
1273 ULONG count = 0;
1274 DWORD dictOffset = 0;
1275
1276 This->dirty = FALSE;
1277 This->highestProp = 0;
1278 hr = IStream_Stat(This->stm, &stat, STATFLAG_NONAME);
1279 if (FAILED(hr))
1280 goto end;
1281 if (stat.cbSize.u.HighPart)
1282 {
1283 WARN("stream too big\n");
1284 /* maximum size varies, but it can't be this big */
1285 hr = STG_E_INVALIDHEADER;
1286 goto end;
1287 }
1288 if (stat.cbSize.u.LowPart == 0)
1289 {
1290 /* empty stream is okay */
1291 hr = S_OK;
1292 goto end;
1293 }
1294 else if (stat.cbSize.u.LowPart < sizeof(PROPERTYSETHEADER) +
1295 sizeof(FORMATIDOFFSET))
1296 {
1297 WARN("stream too small\n");
1298 hr = STG_E_INVALIDHEADER;
1299 goto end;
1300 }
1301 seek.QuadPart = 0;
1302 hr = IStream_Seek(This->stm, seek, STREAM_SEEK_SET, NULL);
1303 if (FAILED(hr))
1304 goto end;
1305 hr = PropertyStorage_ReadHeaderFromStream(This->stm, &hdr);
1306 /* I've only seen reserved == 1, but the article says I shouldn't disallow
1307 * higher values.
1308 */
1309 if (hdr.wByteOrder != PROPSETHDR_BYTEORDER_MAGIC || hdr.reserved < 1)
1310 {
1311 WARN("bad magic in prop set header\n");
1312 hr = STG_E_INVALIDHEADER;
1313 goto end;
1314 }
1315 if (hdr.wFormat != 0 && hdr.wFormat != 1)
1316 {
1317 WARN("bad format version %d\n", hdr.wFormat);
1318 hr = STG_E_INVALIDHEADER;
1319 goto end;
1320 }
1321 This->format = hdr.wFormat;
1322 memcpy(&This->clsid, &hdr.clsid, sizeof(This->clsid));
1323 This->originatorOS = hdr.dwOSVer;
1324 if (PROPSETHDR_OSVER_KIND(hdr.dwOSVer) == PROPSETHDR_OSVER_KIND_MAC)
1325 WARN("File comes from a Mac, strings will probably be screwed up\n");
1326 hr = PropertyStorage_ReadFmtIdOffsetFromStream(This->stm, &fmtOffset);
1327 if (FAILED(hr))
1328 goto end;
1329 if (fmtOffset.dwOffset > stat.cbSize.u.LowPart)
1330 {
1331 WARN("invalid offset %d (stream length is %d)\n", fmtOffset.dwOffset,
1332 stat.cbSize.u.LowPart);
1333 hr = STG_E_INVALIDHEADER;
1334 goto end;
1335 }
1336 /* wackiness alert: if the format ID is FMTID_DocSummaryInformation, there
1337 * follow not one, but two sections. The first is the standard properties
1338 * for the document summary information, and the second is user-defined
1339 * properties. This is the only case in which multiple sections are
1340 * allowed.
1341 * Reading the second stream isn't implemented yet.
1342 */
1343 hr = PropertyStorage_ReadSectionHeaderFromStream(This->stm, &sectionHdr);
1344 if (FAILED(hr))
1345 goto end;
1346 /* The section size includes the section header, so check it */
1347 if (sectionHdr.cbSection < sizeof(PROPERTYSECTIONHEADER))
1348 {
1349 WARN("section header too small, got %d\n", sectionHdr.cbSection);
1350 hr = STG_E_INVALIDHEADER;
1351 goto end;
1352 }
1353 buf = HeapAlloc(GetProcessHeap(), 0, sectionHdr.cbSection -
1354 sizeof(PROPERTYSECTIONHEADER));
1355 if (!buf)
1356 {
1357 hr = STG_E_INSUFFICIENTMEMORY;
1358 goto end;
1359 }
1360 hr = IStream_Read(This->stm, buf, sectionHdr.cbSection -
1361 sizeof(PROPERTYSECTIONHEADER), &count);
1362 if (FAILED(hr))
1363 goto end;
1364 TRACE("Reading %d properties:\n", sectionHdr.cProperties);
1365 for (i = 0; SUCCEEDED(hr) && i < sectionHdr.cProperties; i++)
1366 {
1367 PROPERTYIDOFFSET *idOffset = (PROPERTYIDOFFSET *)(buf +
1368 i * sizeof(PROPERTYIDOFFSET));
1369
1370 if (idOffset->dwOffset < sizeof(PROPERTYSECTIONHEADER) ||
1371 idOffset->dwOffset >= sectionHdr.cbSection - sizeof(DWORD))
1372 hr = STG_E_INVALIDPOINTER;
1373 else
1374 {
1375 if (idOffset->propid >= PID_FIRST_USABLE &&
1376 idOffset->propid < PID_MIN_READONLY && idOffset->propid >
1377 This->highestProp)
1378 This->highestProp = idOffset->propid;
1379 if (idOffset->propid == PID_DICTIONARY)
1380 {
1381 /* Don't read the dictionary yet, its entries depend on the
1382 * code page. Just store the offset so we know to read it
1383 * later.
1384 */
1385 dictOffset = idOffset->dwOffset;
1386 TRACE("Dictionary offset is %d\n", dictOffset);
1387 }
1388 else
1389 {
1390 PROPVARIANT prop;
1391
1392 PropVariantInit(&prop);
1393 if (SUCCEEDED(PropertyStorage_ReadProperty(This, &prop,
1394 buf + idOffset->dwOffset - sizeof(PROPERTYSECTIONHEADER))))
1395 {
1396 TRACE("Read property with ID 0x%08x, type %d\n",
1397 idOffset->propid, prop.vt);
1398 switch(idOffset->propid)
1399 {
1400 case PID_CODEPAGE:
1401 if (prop.vt == VT_I2)
1402 This->codePage = (UINT)prop.u.iVal;
1403 break;
1404 case PID_LOCALE:
1405 if (prop.vt == VT_I4)
1406 This->locale = (LCID)prop.u.lVal;
1407 break;
1408 case PID_BEHAVIOR:
1409 if (prop.vt == VT_I4 && prop.u.lVal)
1410 This->grfFlags |= PROPSETFLAG_CASE_SENSITIVE;
1411 /* The format should already be 1, but just in case */
1412 This->format = 1;
1413 break;
1414 default:
1415 hr = PropertyStorage_StorePropWithId(This,
1416 idOffset->propid, &prop, This->codePage);
1417 }
1418 }
1419 }
1420 }
1421 }
1422 if (!This->codePage)
1423 {
1424 /* default to Unicode unless told not to, as specified here:
1425 * http://msdn.microsoft.com/library/en-us/stg/stg/names_in_istorage.asp
1426 */
1427 if (This->grfFlags & PROPSETFLAG_ANSI)
1428 This->codePage = GetACP();
1429 else
1430 This->codePage = CP_UNICODE;
1431 }
1432 if (!This->locale)
1433 This->locale = LOCALE_SYSTEM_DEFAULT;
1434 TRACE("Code page is %d, locale is %d\n", This->codePage, This->locale);
1435 if (dictOffset)
1436 hr = PropertyStorage_ReadDictionary(This,
1437 buf + dictOffset - sizeof(PROPERTYSECTIONHEADER));
1438
1439 end:
1440 HeapFree(GetProcessHeap(), 0, buf);
1441 if (FAILED(hr))
1442 {
1443 dictionary_destroy(This->name_to_propid);
1444 This->name_to_propid = NULL;
1445 dictionary_destroy(This->propid_to_name);
1446 This->propid_to_name = NULL;
1447 dictionary_destroy(This->propid_to_prop);
1448 This->propid_to_prop = NULL;
1449 }
1450 return hr;
1451 }
1452
1453 static void PropertyStorage_MakeHeader(PropertyStorage_impl *This,
1454 PROPERTYSETHEADER *hdr)
1455 {
1456 assert(hdr);
1457 StorageUtl_WriteWord((BYTE *)&hdr->wByteOrder, 0,
1458 PROPSETHDR_BYTEORDER_MAGIC);
1459 StorageUtl_WriteWord((BYTE *)&hdr->wFormat, 0, This->format);
1460 StorageUtl_WriteDWord((BYTE *)&hdr->dwOSVer, 0, This->originatorOS);
1461 StorageUtl_WriteGUID((BYTE *)&hdr->clsid, 0, &This->clsid);
1462 StorageUtl_WriteDWord((BYTE *)&hdr->reserved, 0, 1);
1463 }
1464
1465 static void PropertyStorage_MakeFmtIdOffset(PropertyStorage_impl *This,
1466 FORMATIDOFFSET *fmtOffset)
1467 {
1468 assert(fmtOffset);
1469 StorageUtl_WriteGUID((BYTE *)fmtOffset, 0, &This->fmtid);
1470 StorageUtl_WriteDWord((BYTE *)fmtOffset, offsetof(FORMATIDOFFSET, dwOffset),
1471 sizeof(PROPERTYSETHEADER) + sizeof(FORMATIDOFFSET));
1472 }
1473
1474 static void PropertyStorage_MakeSectionHdr(DWORD cbSection, DWORD numProps,
1475 PROPERTYSECTIONHEADER *hdr)
1476 {
1477 assert(hdr);
1478 StorageUtl_WriteDWord((BYTE *)hdr, 0, cbSection);
1479 StorageUtl_WriteDWord((BYTE *)hdr,
1480 offsetof(PROPERTYSECTIONHEADER, cProperties), numProps);
1481 }
1482
1483 static void PropertyStorage_MakePropertyIdOffset(DWORD propid, DWORD dwOffset,
1484 PROPERTYIDOFFSET *propIdOffset)
1485 {
1486 assert(propIdOffset);
1487 StorageUtl_WriteDWord((BYTE *)propIdOffset, 0, propid);
1488 StorageUtl_WriteDWord((BYTE *)propIdOffset,
1489 offsetof(PROPERTYIDOFFSET, dwOffset), dwOffset);
1490 }
1491
1492 struct DictionaryClosure
1493 {
1494 HRESULT hr;
1495 DWORD bytesWritten;
1496 };
1497
1498 static BOOL PropertyStorage_DictionaryWriter(const void *key,
1499 const void *value, void *extra, void *closure)
1500 {
1501 PropertyStorage_impl *This = (PropertyStorage_impl *)extra;
1502 struct DictionaryClosure *c = (struct DictionaryClosure *)closure;
1503 DWORD propid;
1504 ULONG count;
1505
1506 assert(key);
1507 assert(closure);
1508 StorageUtl_WriteDWord((LPBYTE)&propid, 0, (DWORD)value);
1509 c->hr = IStream_Write(This->stm, &propid, sizeof(propid), &count);
1510 if (FAILED(c->hr))
1511 goto end;
1512 c->bytesWritten += sizeof(DWORD);
1513 if (This->codePage == CP_UNICODE)
1514 {
1515 DWORD keyLen, pad = 0;
1516
1517 StorageUtl_WriteDWord((LPBYTE)&keyLen, 0,
1518 (lstrlenW((LPCWSTR)key) + 1) * sizeof(WCHAR));
1519 c->hr = IStream_Write(This->stm, &keyLen, sizeof(keyLen), &count);
1520 if (FAILED(c->hr))
1521 goto end;
1522 c->bytesWritten += sizeof(DWORD);
1523 /* Rather than allocate a copy, I'll swap the string to little-endian
1524 * in-place, write it, then swap it back.
1525 */
1526 PropertyStorage_ByteSwapString(key, keyLen);
1527 c->hr = IStream_Write(This->stm, key, keyLen, &count);
1528 PropertyStorage_ByteSwapString(key, keyLen);
1529 if (FAILED(c->hr))
1530 goto end;
1531 c->bytesWritten += keyLen;
1532 if (keyLen % sizeof(DWORD))
1533 {
1534 c->hr = IStream_Write(This->stm, &pad,
1535 sizeof(DWORD) - keyLen % sizeof(DWORD), &count);
1536 if (FAILED(c->hr))
1537 goto end;
1538 c->bytesWritten += sizeof(DWORD) - keyLen % sizeof(DWORD);
1539 }
1540 }
1541 else
1542 {
1543 DWORD keyLen;
1544
1545 StorageUtl_WriteDWord((LPBYTE)&keyLen, 0, strlen((LPCSTR)key) + 1);
1546 c->hr = IStream_Write(This->stm, &keyLen, sizeof(keyLen), &count);
1547 if (FAILED(c->hr))
1548 goto end;
1549 c->bytesWritten += sizeof(DWORD);
1550 c->hr = IStream_Write(This->stm, key, keyLen, &count);
1551 if (FAILED(c->hr))
1552 goto end;
1553 c->bytesWritten += keyLen;
1554 }
1555 end:
1556 return SUCCEEDED(c->hr);
1557 }
1558
1559 #define SECTIONHEADER_OFFSET sizeof(PROPERTYSETHEADER) + sizeof(FORMATIDOFFSET)
1560
1561 /* Writes the dictionary to the stream. Assumes without checking that the
1562 * dictionary isn't empty.
1563 */
1564 static HRESULT PropertyStorage_WriteDictionaryToStream(
1565 PropertyStorage_impl *This, DWORD *sectionOffset)
1566 {
1567 HRESULT hr;
1568 LARGE_INTEGER seek;
1569 PROPERTYIDOFFSET propIdOffset;
1570 ULONG count;
1571 DWORD dwTemp;
1572 struct DictionaryClosure closure;
1573
1574 assert(sectionOffset);
1575
1576 /* The dictionary's always the first property written, so seek to its
1577 * spot.
1578 */
1579 seek.QuadPart = SECTIONHEADER_OFFSET + sizeof(PROPERTYSECTIONHEADER);
1580 hr = IStream_Seek(This->stm, seek, STREAM_SEEK_SET, NULL);
1581 if (FAILED(hr))
1582 goto end;
1583 PropertyStorage_MakePropertyIdOffset(PID_DICTIONARY, *sectionOffset,
1584 &propIdOffset);
1585 hr = IStream_Write(This->stm, &propIdOffset, sizeof(propIdOffset), &count);
1586 if (FAILED(hr))
1587 goto end;
1588
1589 seek.QuadPart = SECTIONHEADER_OFFSET + *sectionOffset;
1590 hr = IStream_Seek(This->stm, seek, STREAM_SEEK_SET, NULL);
1591 if (FAILED(hr))
1592 goto end;
1593 StorageUtl_WriteDWord((LPBYTE)&dwTemp, 0,
1594 dictionary_num_entries(This->name_to_propid));
1595 hr = IStream_Write(This->stm, &dwTemp, sizeof(dwTemp), &count);
1596 if (FAILED(hr))
1597 goto end;
1598 *sectionOffset += sizeof(dwTemp);
1599
1600 closure.hr = S_OK;
1601 closure.bytesWritten = 0;
1602 dictionary_enumerate(This->name_to_propid, PropertyStorage_DictionaryWriter,
1603 &closure);
1604 hr = closure.hr;
1605 if (FAILED(hr))
1606 goto end;
1607 *sectionOffset += closure.bytesWritten;
1608 if (closure.bytesWritten % sizeof(DWORD))
1609 {
1610 DWORD padding = sizeof(DWORD) - closure.bytesWritten % sizeof(DWORD);
1611 TRACE("adding %d bytes of padding\n", padding);
1612 *sectionOffset += padding;
1613 }
1614
1615 end:
1616 return hr;
1617 }
1618
1619 static HRESULT PropertyStorage_WritePropertyToStream(PropertyStorage_impl *This,
1620 DWORD propNum, DWORD propid, const PROPVARIANT *var, DWORD *sectionOffset)
1621 {
1622 HRESULT hr;
1623 LARGE_INTEGER seek;
1624 PROPERTYIDOFFSET propIdOffset;
1625 ULONG count;
1626 DWORD dwType, bytesWritten;
1627
1628 assert(var);
1629 assert(sectionOffset);
1630
1631 TRACE("%p, %d, 0x%08x, (%d), (%d)\n", This, propNum, propid, var->vt,
1632 *sectionOffset);
1633
1634 seek.QuadPart = SECTIONHEADER_OFFSET + sizeof(PROPERTYSECTIONHEADER) +
1635 propNum * sizeof(PROPERTYIDOFFSET);
1636 hr = IStream_Seek(This->stm, seek, STREAM_SEEK_SET, NULL);
1637 if (FAILED(hr))
1638 goto end;
1639 PropertyStorage_MakePropertyIdOffset(propid, *sectionOffset, &propIdOffset);
1640 hr = IStream_Write(This->stm, &propIdOffset, sizeof(propIdOffset), &count);
1641 if (FAILED(hr))
1642 goto end;
1643
1644 seek.QuadPart = SECTIONHEADER_OFFSET + *sectionOffset;
1645 hr = IStream_Seek(This->stm, seek, STREAM_SEEK_SET, NULL);
1646 if (FAILED(hr))
1647 goto end;
1648 StorageUtl_WriteDWord((LPBYTE)&dwType, 0, var->vt);
1649 hr = IStream_Write(This->stm, &dwType, sizeof(dwType), &count);
1650 if (FAILED(hr))
1651 goto end;
1652 *sectionOffset += sizeof(dwType);
1653
1654 switch (var->vt)
1655 {
1656 case VT_EMPTY:
1657 case VT_NULL:
1658 bytesWritten = 0;
1659 break;
1660 case VT_I1:
1661 case VT_UI1:
1662 hr = IStream_Write(This->stm, &var->u.cVal, sizeof(var->u.cVal),
1663 &count);
1664 bytesWritten = count;
1665 break;
1666 case VT_I2:
1667 case VT_UI2:
1668 {
1669 WORD wTemp;
1670
1671 StorageUtl_WriteWord((LPBYTE)&wTemp, 0, var->u.iVal);
1672 hr = IStream_Write(This->stm, &wTemp, sizeof(wTemp), &count);
1673 bytesWritten = count;
1674 break;
1675 }
1676 case VT_I4:
1677 case VT_UI4:
1678 {
1679 DWORD dwTemp;
1680
1681 StorageUtl_WriteDWord((LPBYTE)&dwTemp, 0, var->u.lVal);
1682 hr = IStream_Write(This->stm, &dwTemp, sizeof(dwTemp), &count);
1683 bytesWritten = count;
1684 break;
1685 }
1686 case VT_LPSTR:
1687 {
1688 DWORD len, dwTemp;
1689
1690 if (This->codePage == CP_UNICODE)
1691 len = (lstrlenW(var->u.pwszVal) + 1) * sizeof(WCHAR);
1692 else
1693 len = lstrlenA(var->u.pszVal) + 1;
1694 StorageUtl_WriteDWord((LPBYTE)&dwTemp, 0, len);
1695 hr = IStream_Write(This->stm, &dwTemp, sizeof(dwTemp), &count);
1696 if (FAILED(hr))
1697 goto end;
1698 hr = IStream_Write(This->stm, var->u.pszVal, len, &count);
1699 bytesWritten = count + sizeof(DWORD);
1700 break;
1701 }
1702 case VT_LPWSTR:
1703 {
1704 DWORD len = lstrlenW(var->u.pwszVal) + 1, dwTemp;
1705
1706 StorageUtl_WriteDWord((LPBYTE)&dwTemp, 0, len);
1707 hr = IStream_Write(This->stm, &dwTemp, sizeof(dwTemp), &count);
1708 if (FAILED(hr))
1709 goto end;
1710 hr = IStream_Write(This->stm, var->u.pwszVal, len * sizeof(WCHAR),
1711 &count);
1712 bytesWritten = count + sizeof(DWORD);
1713 break;
1714 }
1715 case VT_FILETIME:
1716 {
1717 FILETIME temp;
1718
1719 StorageUtl_WriteULargeInteger((BYTE *)&temp, 0,
1720 (const ULARGE_INTEGER *)&var->u.filetime);
1721 hr = IStream_Write(This->stm, &temp, sizeof(FILETIME), &count);
1722 bytesWritten = count;
1723 break;
1724 }
1725 case VT_CF:
1726 {
1727 DWORD cf_hdr[2], len;
1728
1729 len = var->u.pclipdata->cbSize;
1730 StorageUtl_WriteDWord((LPBYTE)&cf_hdr[0], 0, len + 8);
1731 StorageUtl_WriteDWord((LPBYTE)&cf_hdr[1], 0, var->u.pclipdata->ulClipFmt);
1732 hr = IStream_Write(This->stm, &cf_hdr, sizeof(cf_hdr), &count);
1733 if (FAILED(hr))
1734 goto end;
1735 hr = IStream_Write(This->stm, &var->u.pclipdata->pClipData, len, &count);
1736 if (FAILED(hr))
1737 goto end;
1738 bytesWritten = count + sizeof cf_hdr;
1739 break;
1740 }
1741 default:
1742 FIXME("unsupported type: %d\n", var->vt);
1743 return STG_E_INVALIDPARAMETER;
1744 }
1745
1746 if (SUCCEEDED(hr))
1747 {
1748 *sectionOffset += bytesWritten;
1749 if (bytesWritten % sizeof(DWORD))
1750 {
1751 DWORD padding = sizeof(DWORD) - bytesWritten % sizeof(DWORD);
1752 TRACE("adding %d bytes of padding\n", padding);
1753 *sectionOffset += padding;
1754 }
1755 }
1756
1757 end:
1758 return hr;
1759 }
1760
1761 struct PropertyClosure
1762 {
1763 HRESULT hr;
1764 DWORD propNum;
1765 DWORD *sectionOffset;
1766 };
1767
1768 static BOOL PropertyStorage_PropertiesWriter(const void *key, const void *value,
1769 void *extra, void *closure)
1770 {
1771 PropertyStorage_impl *This = (PropertyStorage_impl *)extra;
1772 struct PropertyClosure *c = (struct PropertyClosure *)closure;
1773
1774 assert(key);
1775 assert(value);
1776 assert(extra);
1777 assert(closure);
1778 c->hr = PropertyStorage_WritePropertyToStream(This, c->propNum++,
1779 (DWORD)key, value, c->sectionOffset);
1780 return SUCCEEDED(c->hr);
1781 }
1782
1783 static HRESULT PropertyStorage_WritePropertiesToStream(
1784 PropertyStorage_impl *This, DWORD startingPropNum, DWORD *sectionOffset)
1785 {
1786 struct PropertyClosure closure;
1787
1788 assert(sectionOffset);
1789 closure.hr = S_OK;
1790 closure.propNum = startingPropNum;
1791 closure.sectionOffset = sectionOffset;
1792 dictionary_enumerate(This->propid_to_prop, PropertyStorage_PropertiesWriter,
1793 &closure);
1794 return closure.hr;
1795 }
1796
1797 static HRESULT PropertyStorage_WriteHeadersToStream(PropertyStorage_impl *This)
1798 {
1799 HRESULT hr;
1800 ULONG count = 0;
1801 LARGE_INTEGER seek = { {0} };
1802 PROPERTYSETHEADER hdr;
1803 FORMATIDOFFSET fmtOffset;
1804
1805 hr = IStream_Seek(This->stm, seek, STREAM_SEEK_SET, NULL);
1806 if (FAILED(hr))
1807 goto end;
1808 PropertyStorage_MakeHeader(This, &hdr);
1809 hr = IStream_Write(This->stm, &hdr, sizeof(hdr), &count);
1810 if (FAILED(hr))
1811 goto end;
1812 if (count != sizeof(hdr))
1813 {
1814 hr = STG_E_WRITEFAULT;
1815 goto end;
1816 }
1817
1818 PropertyStorage_MakeFmtIdOffset(This, &fmtOffset);
1819 hr = IStream_Write(This->stm, &fmtOffset, sizeof(fmtOffset), &count);
1820 if (FAILED(hr))
1821 goto end;
1822 if (count != sizeof(fmtOffset))
1823 {
1824 hr = STG_E_WRITEFAULT;
1825 goto end;
1826 }
1827 hr = S_OK;
1828
1829 end:
1830 return hr;
1831 }
1832
1833 static HRESULT PropertyStorage_WriteToStream(PropertyStorage_impl *This)
1834 {
1835 PROPERTYSECTIONHEADER sectionHdr;
1836 HRESULT hr;
1837 ULONG count;
1838 LARGE_INTEGER seek;
1839 DWORD numProps, prop, sectionOffset, dwTemp;
1840 PROPVARIANT var;
1841
1842 PropertyStorage_WriteHeadersToStream(This);
1843
1844 /* Count properties. Always at least one property, the code page */
1845 numProps = 1;
1846 if (dictionary_num_entries(This->name_to_propid))
1847 numProps++;
1848 if (This->locale != LOCALE_SYSTEM_DEFAULT)
1849 numProps++;
1850 if (This->grfFlags & PROPSETFLAG_CASE_SENSITIVE)
1851 numProps++;
1852 numProps += dictionary_num_entries(This->propid_to_prop);
1853
1854 /* Write section header with 0 bytes right now, I'll adjust it after
1855 * writing properties.
1856 */
1857 PropertyStorage_MakeSectionHdr(0, numProps, &sectionHdr);
1858 seek.QuadPart = SECTIONHEADER_OFFSET;
1859 hr = IStream_Seek(This->stm, seek, STREAM_SEEK_SET, NULL);
1860 if (FAILED(hr))
1861 goto end;
1862 hr = IStream_Write(This->stm, &sectionHdr, sizeof(sectionHdr), &count);
1863 if (FAILED(hr))
1864 goto end;
1865
1866 prop = 0;
1867 sectionOffset = sizeof(PROPERTYSECTIONHEADER) +
1868 numProps * sizeof(PROPERTYIDOFFSET);
1869
1870 if (dictionary_num_entries(This->name_to_propid))
1871 {
1872 prop++;
1873 hr = PropertyStorage_WriteDictionaryToStream(This, &sectionOffset);
1874 if (FAILED(hr))
1875 goto end;
1876 }
1877
1878 PropVariantInit(&var);
1879
1880 var.vt = VT_I2;
1881 var.u.iVal = This->codePage;
1882 hr = PropertyStorage_WritePropertyToStream(This, prop++, PID_CODEPAGE,
1883 &var, &sectionOffset);
1884 if (FAILED(hr))
1885 goto end;
1886
1887 if (This->locale != LOCALE_SYSTEM_DEFAULT)
1888 {
1889 var.vt = VT_I4;
1890 var.u.lVal = This->locale;
1891 hr = PropertyStorage_WritePropertyToStream(This, prop++, PID_LOCALE,
1892 &var, &sectionOffset);
1893 if (FAILED(hr))
1894 goto end;
1895 }
1896
1897 if (This->grfFlags & PROPSETFLAG_CASE_SENSITIVE)
1898 {
1899 var.vt = VT_I4;
1900 var.u.lVal = 1;
1901 hr = PropertyStorage_WritePropertyToStream(This, prop++, PID_BEHAVIOR,
1902 &var, &sectionOffset);
1903 if (FAILED(hr))
1904 goto end;
1905 }
1906
1907 hr = PropertyStorage_WritePropertiesToStream(This, prop, &sectionOffset);
1908 if (FAILED(hr))
1909 goto end;
1910
1911 /* Now write the byte count of the section */
1912 seek.QuadPart = SECTIONHEADER_OFFSET;
1913 hr = IStream_Seek(This->stm, seek, STREAM_SEEK_SET, NULL);
1914 if (FAILED(hr))
1915 goto end;
1916 StorageUtl_WriteDWord((LPBYTE)&dwTemp, 0, sectionOffset);
1917 hr = IStream_Write(This->stm, &dwTemp, sizeof(dwTemp), &count);
1918
1919 end:
1920 return hr;
1921 }
1922
1923 /***********************************************************************
1924 * PropertyStorage_Construct
1925 */
1926 static void PropertyStorage_DestroyDictionaries(PropertyStorage_impl *This)
1927 {
1928 dictionary_destroy(This->name_to_propid);
1929 This->name_to_propid = NULL;
1930 dictionary_destroy(This->propid_to_name);
1931 This->propid_to_name = NULL;
1932 dictionary_destroy(This->propid_to_prop);
1933 This->propid_to_prop = NULL;
1934 }
1935
1936 static HRESULT PropertyStorage_CreateDictionaries(PropertyStorage_impl *This)
1937 {
1938 HRESULT hr = S_OK;
1939
1940 This->name_to_propid = dictionary_create(
1941 PropertyStorage_PropNameCompare, PropertyStorage_PropNameDestroy,
1942 This);
1943 if (!This->name_to_propid)
1944 {
1945 hr = STG_E_INSUFFICIENTMEMORY;
1946 goto end;
1947 }
1948 This->propid_to_name = dictionary_create(PropertyStorage_PropCompare,
1949 NULL, This);
1950 if (!This->propid_to_name)
1951 {
1952 hr = STG_E_INSUFFICIENTMEMORY;
1953 goto end;
1954 }
1955 This->propid_to_prop = dictionary_create(PropertyStorage_PropCompare,
1956 PropertyStorage_PropertyDestroy, This);
1957 if (!This->propid_to_prop)
1958 {
1959 hr = STG_E_INSUFFICIENTMEMORY;
1960 goto end;
1961 }
1962 end:
1963 if (FAILED(hr))
1964 PropertyStorage_DestroyDictionaries(This);
1965 return hr;
1966 }
1967
1968 static HRESULT PropertyStorage_BaseConstruct(IStream *stm,
1969 REFFMTID rfmtid, DWORD grfMode, PropertyStorage_impl **pps)
1970 {
1971 HRESULT hr = S_OK;
1972
1973 assert(pps);
1974 assert(rfmtid);
1975 *pps = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof **pps);
1976 if (!*pps)
1977 return E_OUTOFMEMORY;
1978
1979 (*pps)->vtbl = &IPropertyStorage_Vtbl;
1980 (*pps)->ref = 1;
1981 InitializeCriticalSection(&(*pps)->cs);
1982 (*pps)->cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": PropertyStorage_impl.cs");
1983 (*pps)->stm = stm;
1984 memcpy(&(*pps)->fmtid, rfmtid, sizeof((*pps)->fmtid));
1985 (*pps)->grfMode = grfMode;
1986
1987 hr = PropertyStorage_CreateDictionaries(*pps);
1988 if (FAILED(hr))
1989 {
1990 IStream_Release(stm);
1991 (*pps)->cs.DebugInfo->Spare[0] = 0;
1992 DeleteCriticalSection(&(*pps)->cs);
1993 HeapFree(GetProcessHeap(), 0, *pps);
1994 *pps = NULL;
1995 }
1996
1997 return hr;
1998 }
1999
2000 static HRESULT PropertyStorage_ConstructFromStream(IStream *stm,
2001 REFFMTID rfmtid, DWORD grfMode, IPropertyStorage** pps)
2002 {
2003 PropertyStorage_impl *ps;
2004 HRESULT hr;
2005
2006 assert(pps);
2007 hr = PropertyStorage_BaseConstruct(stm, rfmtid, grfMode, &ps);
2008 if (SUCCEEDED(hr))
2009 {
2010 hr = PropertyStorage_ReadFromStream(ps);
2011 if (SUCCEEDED(hr))
2012 {
2013 *pps = (IPropertyStorage *)ps;
2014 TRACE("PropertyStorage %p constructed\n", ps);
2015 hr = S_OK;
2016 }
2017 else
2018 {
2019 PropertyStorage_DestroyDictionaries(ps);
2020 HeapFree(GetProcessHeap(), 0, ps);
2021 }
2022 }
2023 return hr;
2024 }
2025
2026 static HRESULT PropertyStorage_ConstructEmpty(IStream *stm,
2027 REFFMTID rfmtid, DWORD grfFlags, DWORD grfMode, IPropertyStorage** pps)
2028 {
2029 PropertyStorage_impl *ps;
2030 HRESULT hr;
2031
2032 assert(pps);
2033 hr = PropertyStorage_BaseConstruct(stm, rfmtid, grfMode, &ps);
2034 if (SUCCEEDED(hr))
2035 {
2036 ps->format = 0;
2037 ps->grfFlags = grfFlags;
2038 if (ps->grfFlags & PROPSETFLAG_CASE_SENSITIVE)
2039 ps->format = 1;
2040 /* default to Unicode unless told not to, as specified here:
2041 * http://msdn.microsoft.com/library/en-us/stg/stg/names_in_istorage.asp
2042 */
2043 if (ps->grfFlags & PROPSETFLAG_ANSI)
2044 ps->codePage = GetACP();
2045 else
2046 ps->codePage = CP_UNICODE;
2047 ps->locale = LOCALE_SYSTEM_DEFAULT;
2048 TRACE("Code page is %d, locale is %d\n", ps->codePage, ps->locale);
2049 *pps = (IPropertyStorage *)ps;
2050 TRACE("PropertyStorage %p constructed\n", ps);
2051 hr = S_OK;
2052 }
2053 return hr;
2054 }
2055
2056
2057 /***********************************************************************
2058 * Implementation of IPropertySetStorage
2059 */
2060
2061 /************************************************************************
2062 * IPropertySetStorage_fnQueryInterface (IUnknown)
2063 *
2064 * This method forwards to the common QueryInterface implementation
2065 */
2066 static HRESULT WINAPI IPropertySetStorage_fnQueryInterface(
2067 IPropertySetStorage *ppstg,
2068 REFIID riid,
2069 void** ppvObject)
2070 {
2071 StorageImpl *This = impl_from_IPropertySetStorage(ppstg);
2072 return IStorage_QueryInterface( (IStorage*)This, riid, ppvObject );
2073 }
2074
2075 /************************************************************************
2076 * IPropertySetStorage_fnAddRef (IUnknown)
2077 *
2078 * This method forwards to the common AddRef implementation
2079 */
2080 static ULONG WINAPI IPropertySetStorage_fnAddRef(
2081 IPropertySetStorage *ppstg)
2082 {
2083 StorageImpl *This = impl_from_IPropertySetStorage(ppstg);
2084 return IStorage_AddRef( (IStorage*)This );
2085 }
2086
2087 /************************************************************************
2088 * IPropertySetStorage_fnRelease (IUnknown)
2089 *
2090 * This method forwards to the common Release implementation
2091 */
2092 static ULONG WINAPI IPropertySetStorage_fnRelease(
2093 IPropertySetStorage *ppstg)
2094 {
2095 StorageImpl *This = impl_from_IPropertySetStorage(ppstg);
2096 return IStorage_Release( (IStorage*)This );
2097 }
2098
2099 /************************************************************************
2100 * IPropertySetStorage_fnCreate (IPropertySetStorage)
2101 */
2102 static HRESULT WINAPI IPropertySetStorage_fnCreate(
2103 IPropertySetStorage *ppstg,
2104 REFFMTID rfmtid,
2105 const CLSID* pclsid,
2106 DWORD grfFlags,
2107 DWORD grfMode,
2108 IPropertyStorage** ppprstg)
2109 {
2110 StorageImpl *This = impl_from_IPropertySetStorage(ppstg);
2111 WCHAR name[CCH_MAX_PROPSTG_NAME];
2112 IStream *stm = NULL;
2113 HRESULT r;
2114
2115 TRACE("%p %s %08x %08x %p\n", This, debugstr_guid(rfmtid), grfFlags,
2116 grfMode, ppprstg);
2117
2118 /* be picky */
2119 if (grfMode != (STGM_CREATE|STGM_READWRITE|STGM_SHARE_EXCLUSIVE))
2120 {
2121 r = STG_E_INVALIDFLAG;
2122 goto end;
2123 }
2124
2125 if (!rfmtid)
2126 {
2127 r = E_INVALIDARG;
2128 goto end;
2129 }
2130
2131 /* FIXME: if (grfFlags & PROPSETFLAG_NONSIMPLE), we need to create a
2132 * storage, not a stream. For now, disallow it.
2133 */
2134 if (grfFlags & PROPSETFLAG_NONSIMPLE)
2135 {
2136 FIXME("PROPSETFLAG_NONSIMPLE not supported\n");
2137 r = STG_E_INVALIDFLAG;
2138 goto end;
2139 }
2140
2141 r = FmtIdToPropStgName(rfmtid, name);
2142 if (FAILED(r))
2143 goto end;
2144
2145 r = IStorage_CreateStream( (IStorage*)This, name, grfMode, 0, 0, &stm );
2146 if (FAILED(r))
2147 goto end;
2148
2149 r = PropertyStorage_ConstructEmpty(stm, rfmtid, grfFlags, grfMode, ppprstg);
2150
2151 end:
2152 TRACE("returning 0x%08x\n", r);
2153 return r;
2154 }
2155
2156 /************************************************************************
2157 * IPropertySetStorage_fnOpen (IPropertySetStorage)
2158 */
2159 static HRESULT WINAPI IPropertySetStorage_fnOpen(
2160 IPropertySetStorage *ppstg,
2161 REFFMTID rfmtid,
2162 DWORD grfMode,
2163 IPropertyStorage** ppprstg)
2164 {
2165 StorageImpl *This = impl_from_IPropertySetStorage(ppstg);
2166 IStream *stm = NULL;
2167 WCHAR name[CCH_MAX_PROPSTG_NAME];
2168 HRESULT r;
2169
2170 TRACE("%p %s %08x %p\n", This, debugstr_guid(rfmtid), grfMode, ppprstg);
2171
2172 /* be picky */
2173 if (grfMode != (STGM_READWRITE|STGM_SHARE_EXCLUSIVE) &&
2174 grfMode != (STGM_READ|STGM_SHARE_EXCLUSIVE))
2175 {
2176 r = STG_E_INVALIDFLAG;
2177 goto end;
2178 }
2179
2180 if (!rfmtid)
2181 {
2182 r = E_INVALIDARG;
2183 goto end;
2184 }
2185
2186 r = FmtIdToPropStgName(rfmtid, name);
2187 if (FAILED(r))
2188 goto end;
2189
2190 r = IStorage_OpenStream((IStorage*) This, name, 0, grfMode, 0, &stm );
2191 if (FAILED(r))
2192 goto end;
2193
2194 r = PropertyStorage_ConstructFromStream(stm, rfmtid, grfMode, ppprstg);
2195
2196 end:
2197 TRACE("returning 0x%08x\n", r);
2198 return r;
2199 }
2200
2201 /************************************************************************
2202 * IPropertySetStorage_fnDelete (IPropertySetStorage)
2203 */
2204 static HRESULT WINAPI IPropertySetStorage_fnDelete(
2205 IPropertySetStorage *ppstg,
2206 REFFMTID rfmtid)
2207 {
2208 StorageImpl *This = impl_from_IPropertySetStorage(ppstg);
2209 IStorage *stg = NULL;
2210 WCHAR name[CCH_MAX_PROPSTG_NAME];
2211 HRESULT r;
2212
2213 TRACE("%p %s\n", This, debugstr_guid(rfmtid));
2214
2215 if (!rfmtid)
2216 return E_INVALIDARG;
2217
2218 r = FmtIdToPropStgName(rfmtid, name);
2219 if (FAILED(r))
2220 return r;
2221
2222 stg = (IStorage*) This;
2223 return IStorage_DestroyElement(stg, name);
2224 }
2225
2226 /************************************************************************
2227 * IPropertySetStorage_fnEnum (IPropertySetStorage)
2228 */
2229 static HRESULT WINAPI IPropertySetStorage_fnEnum(
2230 IPropertySetStorage *ppstg,
2231 IEnumSTATPROPSETSTG** ppenum)
2232 {
2233 StorageImpl *This = impl_from_IPropertySetStorage(ppstg);
2234 return create_EnumSTATPROPSETSTG(This, ppenum);
2235 }
2236
2237 /************************************************************************
2238 * Implement IEnumSTATPROPSETSTG using enumx
2239 */
2240 static HRESULT WINAPI IEnumSTATPROPSETSTG_fnQueryInterface(
2241 IEnumSTATPROPSETSTG *iface,
2242 REFIID riid,
2243 void** ppvObject)
2244 {
2245 return enumx_QueryInterface((enumx_impl*)iface, riid, ppvObject);
2246 }
2247
2248 static ULONG WINAPI IEnumSTATPROPSETSTG_fnAddRef(
2249 IEnumSTATPROPSETSTG *iface)
2250 {
2251 return enumx_AddRef((enumx_impl*)iface);
2252 }
2253
2254 static ULONG WINAPI IEnumSTATPROPSETSTG_fnRelease(
2255 IEnumSTATPROPSETSTG *iface)
2256 {
2257 return enumx_Release((enumx_impl*)iface);
2258 }
2259
2260 static HRESULT WINAPI IEnumSTATPROPSETSTG_fnNext(
2261 IEnumSTATPROPSETSTG *iface,
2262 ULONG celt,
2263 STATPROPSETSTG *rgelt,
2264 ULONG *pceltFetched)
2265 {
2266 return enumx_Next((enumx_impl*)iface, celt, rgelt, pceltFetched);
2267 }
2268
2269 static HRESULT WINAPI IEnumSTATPROPSETSTG_fnSkip(
2270 IEnumSTATPROPSETSTG *iface,
2271 ULONG celt)
2272 {
2273 return enumx_Skip((enumx_impl*)iface, celt);
2274 }
2275
2276 static HRESULT WINAPI IEnumSTATPROPSETSTG_fnReset(
2277 IEnumSTATPROPSETSTG *iface)
2278 {
2279 return enumx_Reset((enumx_impl*)iface);
2280 }
2281
2282 static HRESULT WINAPI IEnumSTATPROPSETSTG_fnClone(
2283 IEnumSTATPROPSETSTG *iface,
2284 IEnumSTATPROPSETSTG **ppenum)
2285 {
2286 return enumx_Clone((enumx_impl*)iface, (enumx_impl**)ppenum);
2287 }
2288
2289 static HRESULT create_EnumSTATPROPSETSTG(
2290 StorageImpl *This,
2291 IEnumSTATPROPSETSTG** ppenum)
2292 {
2293 IStorage *stg = (IStorage*) &This->base.lpVtbl;
2294 IEnumSTATSTG *penum = NULL;
2295 STATSTG stat;
2296 ULONG count;
2297 HRESULT r;
2298 STATPROPSETSTG statpss;
2299 enumx_impl *enumx;
2300
2301 TRACE("%p %p\n", This, ppenum);
2302
2303 enumx = enumx_allocate(&IID_IEnumSTATPROPSETSTG,
2304 &IEnumSTATPROPSETSTG_Vtbl,
2305 sizeof (STATPROPSETSTG));
2306
2307 /* add all the property set elements into a list */
2308 r = IStorage_EnumElements(stg, 0, NULL, 0, &penum);
2309 if (FAILED(r))
2310 return E_OUTOFMEMORY;
2311
2312 while (1)
2313 {
2314 count = 0;
2315 r = IEnumSTATSTG_Next(penum, 1, &stat, &count);
2316 if (FAILED(r))
2317 break;
2318 if (!count)
2319 break;
2320 if (!stat.pwcsName)
2321 continue;
2322 if (stat.pwcsName[0] == 5 && stat.type == STGTY_STREAM)
2323 {
2324 PropStgNameToFmtId(stat.pwcsName, &statpss.fmtid);
2325 TRACE("adding %s (%s)\n", debugstr_w(stat.pwcsName),
2326 debugstr_guid(&statpss.fmtid));
2327 statpss.mtime = stat.mtime;
2328 statpss.atime = stat.atime;
2329 statpss.ctime = stat.ctime;
2330 statpss.grfFlags = stat.grfMode;
2331 memcpy(&statpss.clsid, &stat.clsid, sizeof stat.clsid);
2332 enumx_add_element(enumx, &statpss);
2333 }
2334 CoTaskMemFree(stat.pwcsName);
2335 }
2336 IEnumSTATSTG_Release(penum);
2337
2338 *ppenum = (IEnumSTATPROPSETSTG*) enumx;
2339
2340 return S_OK;
2341 }
2342
2343 /************************************************************************
2344 * Implement IEnumSTATPROPSTG using enumx
2345 */
2346 static HRESULT WINAPI IEnumSTATPROPSTG_fnQueryInterface(
2347 IEnumSTATPROPSTG *iface,
2348 REFIID riid,
2349 void** ppvObject)
2350 {
2351 return enumx_QueryInterface((enumx_impl*)iface, riid, ppvObject);
2352 }
2353
2354 static ULONG WINAPI IEnumSTATPROPSTG_fnAddRef(
2355 IEnumSTATPROPSTG *iface)
2356 {
2357 return enumx_AddRef((enumx_impl*)iface);
2358 }
2359
2360 static ULONG WINAPI IEnumSTATPROPSTG_fnRelease(
2361 IEnumSTATPROPSTG *iface)
2362 {
2363 return enumx_Release((enumx_impl*)iface);
2364 }
2365
2366 static HRESULT WINAPI IEnumSTATPROPSTG_fnNext(
2367 IEnumSTATPROPSTG *iface,
2368 ULONG celt,
2369 STATPROPSTG *rgelt,
2370 ULONG *pceltFetched)
2371 {
2372 return enumx_Next((enumx_impl*)iface, celt, rgelt, pceltFetched);
2373 }
2374
2375 static HRESULT WINAPI IEnumSTATPROPSTG_fnSkip(
2376 IEnumSTATPROPSTG *iface,
2377 ULONG celt)
2378 {
2379 return enumx_Skip((enumx_impl*)iface, celt);
2380 }
2381
2382 static HRESULT WINAPI IEnumSTATPROPSTG_fnReset(
2383 IEnumSTATPROPSTG *iface)
2384 {
2385 return enumx_Reset((enumx_impl*)iface);
2386 }
2387
2388 static HRESULT WINAPI IEnumSTATPROPSTG_fnClone(
2389 IEnumSTATPROPSTG *iface,
2390 IEnumSTATPROPSTG **ppenum)
2391 {
2392 return enumx_Clone((enumx_impl*)iface, (enumx_impl**)ppenum);
2393 }
2394
2395 static BOOL prop_enum_stat(const void *k, const void *v, void *extra, void *arg)
2396 {
2397 enumx_impl *enumx = arg;
2398 PROPID propid = (PROPID) k;
2399 const PROPVARIANT *prop = v;
2400 STATPROPSTG stat;
2401
2402 stat.lpwstrName = NULL;
2403 stat.propid = propid;
2404 stat.vt = prop->vt;
2405
2406 enumx_add_element(enumx, &stat);
2407
2408 return TRUE;
2409 }
2410
2411 static HRESULT create_EnumSTATPROPSTG(
2412 PropertyStorage_impl *This,
2413 IEnumSTATPROPSTG** ppenum)
2414 {
2415 enumx_impl *enumx;
2416
2417 TRACE("%p %p\n", This, ppenum);
2418
2419 enumx = enumx_allocate(&IID_IEnumSTATPROPSTG,
2420 &IEnumSTATPROPSTG_Vtbl,
2421 sizeof (STATPROPSTG));
2422
2423 dictionary_enumerate(This->propid_to_prop, prop_enum_stat, enumx);
2424
2425 *ppenum = (IEnumSTATPROPSTG*) enumx;
2426
2427 return S_OK;
2428 }
2429
2430 /***********************************************************************
2431 * vtables
2432 */
2433 const IPropertySetStorageVtbl IPropertySetStorage_Vtbl =
2434 {
2435 IPropertySetStorage_fnQueryInterface,
2436 IPropertySetStorage_fnAddRef,
2437 IPropertySetStorage_fnRelease,
2438 IPropertySetStorage_fnCreate,
2439 IPropertySetStorage_fnOpen,
2440 IPropertySetStorage_fnDelete,
2441 IPropertySetStorage_fnEnum
2442 };
2443
2444 static const IPropertyStorageVtbl IPropertyStorage_Vtbl =
2445 {
2446 IPropertyStorage_fnQueryInterface,
2447 IPropertyStorage_fnAddRef,
2448 IPropertyStorage_fnRelease,
2449 IPropertyStorage_fnReadMultiple,
2450 IPropertyStorage_fnWriteMultiple,
2451 IPropertyStorage_fnDeleteMultiple,
2452 IPropertyStorage_fnReadPropertyNames,
2453 IPropertyStorage_fnWritePropertyNames,
2454 IPropertyStorage_fnDeletePropertyNames,
2455 IPropertyStorage_fnCommit,
2456 IPropertyStorage_fnRevert,
2457 IPropertyStorage_fnEnum,
2458 IPropertyStorage_fnSetTimes,
2459 IPropertyStorage_fnSetClass,
2460 IPropertyStorage_fnStat,
2461 };
2462
2463 static const IEnumSTATPROPSETSTGVtbl IEnumSTATPROPSETSTG_Vtbl =
2464 {
2465 IEnumSTATPROPSETSTG_fnQueryInterface,
2466 IEnumSTATPROPSETSTG_fnAddRef,
2467 IEnumSTATPROPSETSTG_fnRelease,
2468 IEnumSTATPROPSETSTG_fnNext,
2469 IEnumSTATPROPSETSTG_fnSkip,
2470 IEnumSTATPROPSETSTG_fnReset,
2471 IEnumSTATPROPSETSTG_fnClone,
2472 };
2473
2474 static const IEnumSTATPROPSTGVtbl IEnumSTATPROPSTG_Vtbl =
2475 {
2476 IEnumSTATPROPSTG_fnQueryInterface,
2477 IEnumSTATPROPSTG_fnAddRef,
2478 IEnumSTATPROPSTG_fnRelease,
2479 IEnumSTATPROPSTG_fnNext,
2480 IEnumSTATPROPSTG_fnSkip,
2481 IEnumSTATPROPSTG_fnReset,
2482 IEnumSTATPROPSTG_fnClone,
2483 };
2484
2485 /***********************************************************************
2486 * Format ID <-> name conversion
2487 */
2488 static const WCHAR szSummaryInfo[] = { 5,'S','u','m','m','a','r','y',
2489 'I','n','f','o','r','m','a','t','i','o','n',0 };
2490 static const WCHAR szDocSummaryInfo[] = { 5,'D','o','c','u','m','e','n','t',
2491 'S','u','m','m','a','r','y','I','n','f','o','r','m','a','t','i','o','n',0 };
2492
2493 #define BITS_PER_BYTE 8
2494 #define CHARMASK 0x1f
2495 #define BITS_IN_CHARMASK 5
2496 #define NUM_ALPHA_CHARS 26
2497
2498 /***********************************************************************
2499 * FmtIdToPropStgName [ole32.@]
2500 * Returns the storage name of the format ID rfmtid.
2501 * PARAMS
2502 * rfmtid [I] Format ID for which to return a storage name
2503 * str [O] Storage name associated with rfmtid.
2504 *
2505 * RETURNS
2506 * E_INVALIDARG if rfmtid or str i NULL, S_OK otherwise.
2507 *
2508 * NOTES
2509 * str must be at least CCH_MAX_PROPSTG_NAME characters in length.
2510 * Based on the algorithm described here:
2511 * http://msdn.microsoft.com/library/en-us/stg/stg/names_in_istorage.asp
2512 */
2513 HRESULT WINAPI FmtIdToPropStgName(const FMTID *rfmtid, LPOLESTR str)
2514 {
2515 static const char fmtMap[] = "abcdefghijklmnopqrstuvwxyz012345";
2516
2517 TRACE("%s, %p\n", debugstr_guid(rfmtid), str);
2518
2519 if (!rfmtid) return E_INVALIDARG;
2520 if (!str) return E_INVALIDARG;
2521
2522 if (IsEqualGUID(&FMTID_SummaryInformation, rfmtid))
2523 lstrcpyW(str, szSummaryInfo);
2524 else if (IsEqualGUID(&FMTID_DocSummaryInformation, rfmtid))
2525 lstrcpyW(str, szDocSummaryInfo);
2526 else if (IsEqualGUID(&FMTID_UserDefinedProperties, rfmtid))
2527 lstrcpyW(str, szDocSummaryInfo);
2528 else
2529 {
2530 const BYTE *fmtptr;
2531 WCHAR *pstr = str;
2532 ULONG bitsRemaining = BITS_PER_BYTE;
2533
2534 *pstr++ = 5;
2535 for (fmtptr = (const BYTE *)rfmtid; fmtptr < (const BYTE *)rfmtid + sizeof(FMTID); )
2536 {
2537 ULONG i = *fmtptr >> (BITS_PER_BYTE - bitsRemaining);
2538
2539 if (bitsRemaining >= BITS_IN_CHARMASK)
2540 {
2541 *pstr = (WCHAR)(fmtMap[i & CHARMASK]);
2542 if (bitsRemaining == BITS_PER_BYTE && *pstr >= 'a' &&
2543 *pstr <= 'z')
2544 *pstr += 'A' - 'a';
2545 pstr++;
2546 bitsRemaining -= BITS_IN_CHARMASK;
2547 if (bitsRemaining == 0)
2548 {
2549 fmtptr++;
2550 bitsRemaining = BITS_PER_BYTE;
2551 }
2552 }
2553 else
2554 {
2555 if (++fmtptr < (const BYTE *)rfmtid + sizeof(FMTID))
2556 i |= *fmtptr << bitsRemaining;
2557 *pstr++ = (WCHAR)(fmtMap[i & CHARMASK]);
2558 bitsRemaining += BITS_PER_BYTE - BITS_IN_CHARMASK;
2559 }
2560 }
2561 *pstr = 0;
2562 }
2563 TRACE("returning %s\n", debugstr_w(str));
2564 return S_OK;
2565 }
2566
2567 /***********************************************************************
2568 * PropStgNameToFmtId [ole32.@]
2569 * Returns the format ID corresponding to the given name.
2570 * PARAMS
2571 * str [I] Storage name to convert to a format ID.
2572 * rfmtid [O] Format ID corresponding to str.
2573 *
2574 * RETURNS
2575 * E_INVALIDARG if rfmtid or str is NULL or if str can't be converted to
2576 * a format ID, S_OK otherwise.
2577 *
2578 * NOTES
2579 * Based on the algorithm described here:
2580 * http://msdn.microsoft.com/library/en-us/stg/stg/names_in_istorage.asp
2581 */
2582 HRESULT WINAPI PropStgNameToFmtId(const LPOLESTR str, FMTID *rfmtid)
2583 {
2584 HRESULT hr = STG_E_INVALIDNAME;
2585
2586 TRACE("%s, %p\n", debugstr_w(str), rfmtid);
2587
2588 if (!rfmtid) return E_INVALIDARG;
2589 if (!str) return STG_E_INVALIDNAME;
2590
2591 if (!lstrcmpiW(str, szDocSummaryInfo))
2592 {
2593 memcpy(rfmtid, &FMTID_DocSummaryInformation, sizeof(*rfmtid));
2594 hr = S_OK;
2595 }
2596 else if (!lstrcmpiW(str, szSummaryInfo))
2597 {
2598 memcpy(rfmtid, &FMTID_SummaryInformation, sizeof(*rfmtid));
2599 hr = S_OK;
2600 }
2601 else
2602 {
2603 ULONG bits;
2604 BYTE *fmtptr = (BYTE *)rfmtid - 1;
2605 const WCHAR *pstr = str;
2606
2607 memset(rfmtid, 0, sizeof(*rfmtid));
2608 for (bits = 0; bits < sizeof(FMTID) * BITS_PER_BYTE;
2609 bits += BITS_IN_CHARMASK)
2610 {
2611 ULONG bitsUsed = bits % BITS_PER_BYTE, bitsStored;
2612 WCHAR wc;
2613
2614 if (bitsUsed == 0)
2615 fmtptr++;
2616 wc = *++pstr - 'A';
2617 if (wc > NUM_ALPHA_CHARS)
2618 {
2619 wc += 'A' - 'a';
2620 if (wc > NUM_ALPHA_CHARS)
2621 {
2622 wc += 'a' - '0' + NUM_ALPHA_CHARS;
2623 if (wc > CHARMASK)
2624 {
2625 WARN("invalid character (%d)\n", *pstr);
2626 goto end;
2627 }
2628 }
2629 }
2630 *fmtptr |= wc << bitsUsed;
2631 bitsStored = min(BITS_PER_BYTE - bitsUsed, BITS_IN_CHARMASK);
2632 if (bitsStored < BITS_IN_CHARMASK)
2633 {
2634 wc >>= BITS_PER_BYTE - bitsUsed;
2635 if (bits + bitsStored == sizeof(FMTID) * BITS_PER_BYTE)
2636 {
2637 if (wc != 0)
2638 {
2639 WARN("extra bits\n");
2640 goto end;
2641 }
2642 break;
2643 }
2644 fmtptr++;
2645 *fmtptr |= (BYTE)wc;
2646 }
2647 }
2648 hr = S_OK;
2649 }
2650 end:
2651 return hr;
2652 }
Maarten's multimedia branch