wininet: Support Punycode in RetrieveUrlCacheEntryStream.
[wine/multimedia.git] / dlls / msctf / compartmentmgr.c
bloba5bae3f08bb78c50bd774d88d16f5c5ed34fb9a6
1 /*
2 * ITfCompartmentMgr implementation
4 * Copyright 2009 Aric Stewart, CodeWeavers
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.
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.
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include "config.h"
23 #include <stdarg.h>
25 #define COBJMACROS
27 #include "wine/debug.h"
28 #include "windef.h"
29 #include "winbase.h"
30 #include "winreg.h"
31 #include "winuser.h"
32 #include "shlwapi.h"
33 #include "winerror.h"
34 #include "objbase.h"
35 #include "oleauto.h"
36 #include "olectl.h"
38 #include "wine/unicode.h"
39 #include "wine/list.h"
41 #include "msctf.h"
42 #include "msctf_internal.h"
44 WINE_DEFAULT_DEBUG_CHANNEL(msctf);
46 typedef struct tagCompartmentValue {
47 struct list entry;
48 GUID guid;
49 TfClientId owner;
50 ITfCompartment *compartment;
51 } CompartmentValue;
53 typedef struct tagCompartmentMgr {
54 const ITfCompartmentMgrVtbl *CompartmentMgrVtbl;
55 LONG refCount;
57 IUnknown *pUnkOuter;
59 struct list values;
60 } CompartmentMgr;
62 typedef struct tagCompartmentEnumGuid {
63 const IEnumGUIDVtbl *Vtbl;
64 LONG refCount;
66 struct list *values;
67 struct list *cursor;
68 } CompartmentEnumGuid;
71 typedef struct tagCompartmentSink {
72 struct list entry;
73 union {
74 IUnknown *pIUnknown;
75 ITfCompartmentEventSink *pITfCompartmentEventSink;
76 } interfaces;
77 } CompartmentSink;
79 typedef struct tagCompartment {
80 const ITfCompartmentVtbl *Vtbl;
81 const ITfSourceVtbl *SourceVtbl;
82 LONG refCount;
84 /* Only VT_I4, VT_UNKNOWN and VT_BSTR data types are allowed */
85 VARIANT variant;
86 CompartmentValue *valueData;
87 struct list CompartmentEventSink;
88 } Compartment;
90 static HRESULT CompartmentEnumGuid_Constructor(struct list* values, IEnumGUID **ppOut);
91 static HRESULT Compartment_Constructor(CompartmentValue *value, ITfCompartment **ppOut);
93 static inline Compartment *impl_from_ITfSourceVtbl(ITfSource *iface)
95 return (Compartment *)((char *)iface - FIELD_OFFSET(Compartment,SourceVtbl));
98 HRESULT CompartmentMgr_Destructor(ITfCompartmentMgr *iface)
100 CompartmentMgr *This = (CompartmentMgr *)iface;
101 struct list *cursor, *cursor2;
103 LIST_FOR_EACH_SAFE(cursor, cursor2, &This->values)
105 CompartmentValue* value = LIST_ENTRY(cursor,CompartmentValue,entry);
106 list_remove(cursor);
107 ITfCompartment_Release(value->compartment);
108 HeapFree(GetProcessHeap(),0,value);
111 HeapFree(GetProcessHeap(),0,This);
112 return S_OK;
115 /*****************************************************
116 * ITfCompartmentMgr functions
117 *****************************************************/
118 static HRESULT WINAPI CompartmentMgr_QueryInterface(ITfCompartmentMgr *iface, REFIID iid, LPVOID *ppvOut)
120 CompartmentMgr *This = (CompartmentMgr *)iface;
121 if (This->pUnkOuter)
122 return IUnknown_QueryInterface(This->pUnkOuter, iid, *ppvOut);
123 else
125 *ppvOut = NULL;
127 if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfCompartmentMgr))
129 *ppvOut = This;
132 if (*ppvOut)
134 ITfCompartmentMgr_AddRef(iface);
135 return S_OK;
138 WARN("unsupported interface: %s\n", debugstr_guid(iid));
139 return E_NOINTERFACE;
143 static ULONG WINAPI CompartmentMgr_AddRef(ITfCompartmentMgr *iface)
145 CompartmentMgr *This = (CompartmentMgr *)iface;
146 if (This->pUnkOuter)
147 return IUnknown_AddRef(This->pUnkOuter);
148 else
149 return InterlockedIncrement(&This->refCount);
152 static ULONG WINAPI CompartmentMgr_Release(ITfCompartmentMgr *iface)
154 CompartmentMgr *This = (CompartmentMgr *)iface;
155 if (This->pUnkOuter)
156 return IUnknown_Release(This->pUnkOuter);
157 else
159 ULONG ret;
161 ret = InterlockedDecrement(&This->refCount);
162 if (ret == 0)
163 CompartmentMgr_Destructor(iface);
164 return ret;
168 static HRESULT WINAPI CompartmentMgr_GetCompartment(ITfCompartmentMgr *iface,
169 REFGUID rguid, ITfCompartment **ppcomp)
171 CompartmentMgr *This = (CompartmentMgr *)iface;
172 CompartmentValue* value;
173 struct list *cursor;
174 HRESULT hr;
176 TRACE("(%p) %s %p\n",This,debugstr_guid(rguid),ppcomp);
178 LIST_FOR_EACH(cursor, &This->values)
180 value = LIST_ENTRY(cursor,CompartmentValue,entry);
181 if (IsEqualGUID(rguid,&value->guid))
183 ITfCompartment_AddRef(value->compartment);
184 *ppcomp = value->compartment;
185 return S_OK;
189 value = HeapAlloc(GetProcessHeap(),0,sizeof(CompartmentValue));
190 value->guid = *rguid;
191 value->owner = 0;
192 hr = Compartment_Constructor(value,&value->compartment);
193 if (SUCCEEDED(hr))
195 list_add_head(&This->values,&value->entry);
196 ITfCompartment_AddRef(value->compartment);
197 *ppcomp = value->compartment;
199 else
201 HeapFree(GetProcessHeap(),0,value);
202 *ppcomp = NULL;
204 return hr;
207 static HRESULT WINAPI CompartmentMgr_ClearCompartment(ITfCompartmentMgr *iface,
208 TfClientId tid, REFGUID rguid)
210 struct list *cursor;
211 CompartmentMgr *This = (CompartmentMgr *)iface;
212 TRACE("(%p) %i %s\n",This,tid,debugstr_guid(rguid));
214 LIST_FOR_EACH(cursor, &This->values)
216 CompartmentValue* value = LIST_ENTRY(cursor,CompartmentValue,entry);
217 if (IsEqualGUID(rguid,&value->guid))
219 if (value->owner && tid != value->owner)
220 return E_UNEXPECTED;
221 list_remove(cursor);
222 ITfCompartment_Release(value->compartment);
223 HeapFree(GetProcessHeap(),0,value);
224 return S_OK;
228 return CONNECT_E_NOCONNECTION;
231 static HRESULT WINAPI CompartmentMgr_EnumCompartments(ITfCompartmentMgr *iface,
232 IEnumGUID **ppEnum)
234 CompartmentMgr *This = (CompartmentMgr *)iface;
235 TRACE("(%p) %p\n",This,ppEnum);
236 if (!ppEnum)
237 return E_INVALIDARG;
238 return CompartmentEnumGuid_Constructor(&This->values, ppEnum);
241 static const ITfCompartmentMgrVtbl CompartmentMgr_CompartmentMgrVtbl =
243 CompartmentMgr_QueryInterface,
244 CompartmentMgr_AddRef,
245 CompartmentMgr_Release,
247 CompartmentMgr_GetCompartment,
248 CompartmentMgr_ClearCompartment,
249 CompartmentMgr_EnumCompartments
252 HRESULT CompartmentMgr_Constructor(IUnknown *pUnkOuter, REFIID riid, IUnknown **ppOut)
254 CompartmentMgr *This;
256 if (!ppOut)
257 return E_POINTER;
259 if (pUnkOuter && !IsEqualIID (riid, &IID_IUnknown))
260 return CLASS_E_NOAGGREGATION;
262 This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(CompartmentMgr));
263 if (This == NULL)
264 return E_OUTOFMEMORY;
266 This->CompartmentMgrVtbl = &CompartmentMgr_CompartmentMgrVtbl;
267 This->pUnkOuter = pUnkOuter;
268 list_init(&This->values);
270 if (pUnkOuter)
272 TRACE("returning %p\n", This);
273 *ppOut = (IUnknown*)This;
274 return S_OK;
276 else
278 HRESULT hr;
279 hr = IUnknown_QueryInterface((IUnknown*)This, riid, (LPVOID*)ppOut);
280 if (FAILED(hr))
281 HeapFree(GetProcessHeap(),0,This);
282 return hr;
286 /**************************************************
287 * IEnumGUID implementation for ITfCompartmentMgr::EnumCompartments
288 **************************************************/
289 static void CompartmentEnumGuid_Destructor(CompartmentEnumGuid *This)
291 TRACE("destroying %p\n", This);
292 HeapFree(GetProcessHeap(),0,This);
295 static HRESULT WINAPI CompartmentEnumGuid_QueryInterface(IEnumGUID *iface, REFIID iid, LPVOID *ppvOut)
297 CompartmentEnumGuid *This = (CompartmentEnumGuid *)iface;
298 *ppvOut = NULL;
300 if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_IEnumGUID))
302 *ppvOut = This;
305 if (*ppvOut)
307 IEnumGUID_AddRef(iface);
308 return S_OK;
311 WARN("unsupported interface: %s\n", debugstr_guid(iid));
312 return E_NOINTERFACE;
315 static ULONG WINAPI CompartmentEnumGuid_AddRef(IEnumGUID *iface)
317 CompartmentEnumGuid *This = (CompartmentEnumGuid*)iface;
318 return InterlockedIncrement(&This->refCount);
321 static ULONG WINAPI CompartmentEnumGuid_Release(IEnumGUID *iface)
323 CompartmentEnumGuid *This = (CompartmentEnumGuid *)iface;
324 ULONG ret;
326 ret = InterlockedDecrement(&This->refCount);
327 if (ret == 0)
328 CompartmentEnumGuid_Destructor(This);
329 return ret;
332 /*****************************************************
333 * IEnumGuid functions
334 *****************************************************/
335 static HRESULT WINAPI CompartmentEnumGuid_Next( LPENUMGUID iface,
336 ULONG celt, GUID *rgelt, ULONG *pceltFetched)
338 CompartmentEnumGuid *This = (CompartmentEnumGuid *)iface;
339 ULONG fetched = 0;
341 TRACE("(%p)\n",This);
343 if (rgelt == NULL) return E_POINTER;
345 while (fetched < celt && This->cursor)
347 CompartmentValue* value = LIST_ENTRY(This->cursor,CompartmentValue,entry);
348 if (!value)
349 break;
351 This->cursor = list_next(This->values,This->cursor);
352 *rgelt = value->guid;
354 ++fetched;
355 ++rgelt;
358 if (pceltFetched) *pceltFetched = fetched;
359 return fetched == celt ? S_OK : S_FALSE;
362 static HRESULT WINAPI CompartmentEnumGuid_Skip( LPENUMGUID iface, ULONG celt)
364 CompartmentEnumGuid *This = (CompartmentEnumGuid *)iface;
365 TRACE("(%p)\n",This);
367 This->cursor = list_next(This->values,This->cursor);
368 return S_OK;
371 static HRESULT WINAPI CompartmentEnumGuid_Reset( LPENUMGUID iface)
373 CompartmentEnumGuid *This = (CompartmentEnumGuid *)iface;
374 TRACE("(%p)\n",This);
375 This->cursor = list_head(This->values);
376 return S_OK;
379 static HRESULT WINAPI CompartmentEnumGuid_Clone( LPENUMGUID iface,
380 IEnumGUID **ppenum)
382 CompartmentEnumGuid *This = (CompartmentEnumGuid *)iface;
383 HRESULT res;
385 TRACE("(%p)\n",This);
387 if (ppenum == NULL) return E_POINTER;
389 res = CompartmentEnumGuid_Constructor(This->values, ppenum);
390 if (SUCCEEDED(res))
392 CompartmentEnumGuid *new_This = (CompartmentEnumGuid *)*ppenum;
393 new_This->cursor = This->cursor;
395 return res;
398 static const IEnumGUIDVtbl IEnumGUID_Vtbl ={
399 CompartmentEnumGuid_QueryInterface,
400 CompartmentEnumGuid_AddRef,
401 CompartmentEnumGuid_Release,
403 CompartmentEnumGuid_Next,
404 CompartmentEnumGuid_Skip,
405 CompartmentEnumGuid_Reset,
406 CompartmentEnumGuid_Clone
409 static HRESULT CompartmentEnumGuid_Constructor(struct list *values, IEnumGUID **ppOut)
411 CompartmentEnumGuid *This;
413 This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(CompartmentEnumGuid));
414 if (This == NULL)
415 return E_OUTOFMEMORY;
417 This->Vtbl= &IEnumGUID_Vtbl;
418 This->refCount = 1;
420 This->values = values;
421 This->cursor = list_head(values);
423 TRACE("returning %p\n", This);
424 *ppOut = (IEnumGUID*)This;
425 return S_OK;
428 /**************************************************
429 * ITfCompartment
430 **************************************************/
431 static void free_sink(CompartmentSink *sink)
433 IUnknown_Release(sink->interfaces.pIUnknown);
434 HeapFree(GetProcessHeap(),0,sink);
437 static void Compartment_Destructor(Compartment *This)
439 struct list *cursor, *cursor2;
440 TRACE("destroying %p\n", This);
441 VariantClear(&This->variant);
442 LIST_FOR_EACH_SAFE(cursor, cursor2, &This->CompartmentEventSink)
444 CompartmentSink* sink = LIST_ENTRY(cursor,CompartmentSink,entry);
445 list_remove(cursor);
446 free_sink(sink);
448 HeapFree(GetProcessHeap(),0,This);
451 static HRESULT WINAPI Compartment_QueryInterface(ITfCompartment *iface, REFIID iid, LPVOID *ppvOut)
453 Compartment *This = (Compartment *)iface;
454 *ppvOut = NULL;
456 if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfCompartment))
458 *ppvOut = This;
460 else if (IsEqualIID(iid, &IID_ITfSource))
462 *ppvOut = &This->SourceVtbl;
465 if (*ppvOut)
467 ITfCompartment_AddRef(iface);
468 return S_OK;
471 WARN("unsupported interface: %s\n", debugstr_guid(iid));
472 return E_NOINTERFACE;
475 static ULONG WINAPI Compartment_AddRef(ITfCompartment *iface)
477 Compartment *This = (Compartment*)iface;
478 return InterlockedIncrement(&This->refCount);
481 static ULONG WINAPI Compartment_Release(ITfCompartment *iface)
483 Compartment *This = (Compartment *)iface;
484 ULONG ret;
486 ret = InterlockedDecrement(&This->refCount);
487 if (ret == 0)
488 Compartment_Destructor(This);
489 return ret;
492 static HRESULT WINAPI Compartment_SetValue(ITfCompartment *iface,
493 TfClientId tid, const VARIANT *pvarValue)
495 Compartment *This = (Compartment *)iface;
496 struct list *cursor;
498 TRACE("(%p) %i %p\n",This,tid,pvarValue);
500 if (!pvarValue)
501 return E_INVALIDARG;
503 if (!(V_VT(pvarValue) == VT_BSTR || V_VT(pvarValue) == VT_I4 ||
504 V_VT(pvarValue) == VT_UNKNOWN))
505 return E_INVALIDARG;
507 if (!This->valueData->owner)
508 This->valueData->owner = tid;
510 VariantClear(&This->variant);
512 /* Shallow copy of value and type */
513 This->variant = *pvarValue;
515 if (V_VT(pvarValue) == VT_BSTR)
516 V_BSTR(&This->variant) = SysAllocStringByteLen((char*)V_BSTR(pvarValue),
517 SysStringByteLen(V_BSTR(pvarValue)));
518 else if (V_VT(pvarValue) == VT_UNKNOWN)
519 IUnknown_AddRef(V_UNKNOWN(&This->variant));
521 LIST_FOR_EACH(cursor, &This->CompartmentEventSink)
523 CompartmentSink* sink = LIST_ENTRY(cursor,CompartmentSink,entry);
524 ITfCompartmentEventSink_OnChange(sink->interfaces.pITfCompartmentEventSink,&This->valueData->guid);
527 return S_OK;
530 static HRESULT WINAPI Compartment_GetValue(ITfCompartment *iface,
531 VARIANT *pvarValue)
533 Compartment *This = (Compartment *)iface;
534 TRACE("(%p) %p\n",This, pvarValue);
536 if (!pvarValue)
537 return E_INVALIDARG;
539 VariantInit(pvarValue);
540 if (V_VT(&This->variant) == VT_EMPTY) return S_FALSE;
541 return VariantCopy(pvarValue,&This->variant);
544 static const ITfCompartmentVtbl ITfCompartment_Vtbl ={
545 Compartment_QueryInterface,
546 Compartment_AddRef,
547 Compartment_Release,
549 Compartment_SetValue,
550 Compartment_GetValue
553 /*****************************************************
554 * ITfSource functions
555 *****************************************************/
557 static HRESULT WINAPI Source_QueryInterface(ITfSource *iface, REFIID iid, LPVOID *ppvOut)
559 Compartment *This = impl_from_ITfSourceVtbl(iface);
560 return Compartment_QueryInterface((ITfCompartment *)This, iid, *ppvOut);
563 static ULONG WINAPI Source_AddRef(ITfSource *iface)
565 Compartment *This = impl_from_ITfSourceVtbl(iface);
566 return Compartment_AddRef((ITfCompartment*)This);
569 static ULONG WINAPI Source_Release(ITfSource *iface)
571 Compartment *This = impl_from_ITfSourceVtbl(iface);
572 return Compartment_Release((ITfCompartment *)This);
575 static HRESULT WINAPI CompartmentSource_AdviseSink(ITfSource *iface,
576 REFIID riid, IUnknown *punk, DWORD *pdwCookie)
578 CompartmentSink *cs;
579 Compartment *This = impl_from_ITfSourceVtbl(iface);
581 TRACE("(%p) %s %p %p\n",This,debugstr_guid(riid),punk,pdwCookie);
583 if (!riid || !punk || !pdwCookie)
584 return E_INVALIDARG;
586 if (IsEqualIID(riid, &IID_ITfCompartmentEventSink))
588 cs = HeapAlloc(GetProcessHeap(),0,sizeof(CompartmentSink));
589 if (!cs)
590 return E_OUTOFMEMORY;
591 if (FAILED(IUnknown_QueryInterface(punk, riid, (LPVOID *)&cs->interfaces.pITfCompartmentEventSink)))
593 HeapFree(GetProcessHeap(),0,cs);
594 return CONNECT_E_CANNOTCONNECT;
596 list_add_head(&This->CompartmentEventSink,&cs->entry);
597 *pdwCookie = generate_Cookie(COOKIE_MAGIC_COMPARTMENTSINK , cs);
599 else
601 FIXME("(%p) Unhandled Sink: %s\n",This,debugstr_guid(riid));
602 return E_NOTIMPL;
605 TRACE("cookie %x\n",*pdwCookie);
607 return S_OK;
610 static HRESULT WINAPI CompartmentSource_UnadviseSink(ITfSource *iface, DWORD pdwCookie)
612 CompartmentSink *sink;
613 Compartment *This = impl_from_ITfSourceVtbl(iface);
615 TRACE("(%p) %x\n",This,pdwCookie);
617 if (get_Cookie_magic(pdwCookie)!=COOKIE_MAGIC_COMPARTMENTSINK)
618 return E_INVALIDARG;
620 sink = (CompartmentSink*)remove_Cookie(pdwCookie);
621 if (!sink)
622 return CONNECT_E_NOCONNECTION;
624 list_remove(&sink->entry);
625 free_sink(sink);
627 return S_OK;
630 static const ITfSourceVtbl Compartment_SourceVtbl =
632 Source_QueryInterface,
633 Source_AddRef,
634 Source_Release,
636 CompartmentSource_AdviseSink,
637 CompartmentSource_UnadviseSink,
640 static HRESULT Compartment_Constructor(CompartmentValue *valueData, ITfCompartment **ppOut)
642 Compartment *This;
644 This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(Compartment));
645 if (This == NULL)
646 return E_OUTOFMEMORY;
648 This->Vtbl= &ITfCompartment_Vtbl;
649 This->SourceVtbl = &Compartment_SourceVtbl;
650 This->refCount = 1;
652 This->valueData = valueData;
653 VariantInit(&This->variant);
655 list_init(&This->CompartmentEventSink);
657 TRACE("returning %p\n", This);
658 *ppOut = (ITfCompartment*)This;
659 return S_OK;