1 /***************************************************************************************
2 * BindCtx implementation
4 * Copyright 1999 Noomen Hamza
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 ***************************************************************************************/
27 #include "wine/unicode.h"
29 #include "wine/debug.h"
31 WINE_DEFAULT_DEBUG_CHANNEL(ole);
33 /* represent the first size table and it's increment block size */
34 #define BLOCK_TAB_SIZE 10
35 #define MAX_TAB_SIZE 0xFFFFFFFF
37 /* data structure of the BindCtx table elements */
38 typedef struct BindCtxObject{
40 IUnknown* pObj; /* point on a bound object */
42 LPOLESTR pkeyObj; /* key associated to this bound object */
44 BYTE regType; /* registration type: 1 if RegisterObjectParam and 0 if RegisterObjectBound */
48 /* BindCtx data strucrture */
49 typedef struct BindCtxImpl{
51 ICOM_VFIELD(IBindCtx); /* VTable relative to the IBindCtx interface.*/
53 ULONG ref; /* reference counter for this object */
55 BindCtxObject* bindCtxTable; /* this is a table in which all bounded objects are stored*/
56 DWORD bindCtxTableLastIndex; /* first free index in the table */
57 DWORD bindCtxTableSize; /* size table */
59 BIND_OPTS2 bindOption2; /* a structure which contains the bind options*/
63 /* IBindCtx prototype functions : */
65 /* IUnknown functions*/
66 static HRESULT WINAPI BindCtxImpl_QueryInterface(IBindCtx* iface,REFIID riid,void** ppvObject);
67 static ULONG WINAPI BindCtxImpl_AddRef(IBindCtx* iface);
68 static ULONG WINAPI BindCtxImpl_Release(IBindCtx* iface);
69 /* IBindCtx functions */
70 static HRESULT WINAPI BindCtxImpl_RegisterObjectBound(IBindCtx* iface,IUnknown* punk);
71 static HRESULT WINAPI BindCtxImpl_RevokeObjectBound(IBindCtx* iface, IUnknown* punk);
72 static HRESULT WINAPI BindCtxImpl_ReleaseBoundObjects(IBindCtx* iface);
73 static HRESULT WINAPI BindCtxImpl_SetBindOptions(IBindCtx* iface,BIND_OPTS *pbindopts);
74 static HRESULT WINAPI BindCtxImpl_GetBindOptions(IBindCtx* iface,BIND_OPTS *pbindopts);
75 static HRESULT WINAPI BindCtxImpl_GetRunningObjectTable(IBindCtx* iface,IRunningObjectTable** pprot);
76 static HRESULT WINAPI BindCtxImpl_RegisterObjectParam(IBindCtx* iface,LPOLESTR pszkey, IUnknown* punk);
77 static HRESULT WINAPI BindCtxImpl_GetObjectParam(IBindCtx* iface,LPOLESTR pszkey, IUnknown** punk);
78 static HRESULT WINAPI BindCtxImpl_EnumObjectParam(IBindCtx* iface,IEnumString** ppenum);
79 static HRESULT WINAPI BindCtxImpl_RevokeObjectParam(IBindCtx* iface,LPOLESTR pszkey);
81 HRESULT WINAPI BindCtxImpl_Construct(BindCtxImpl* This);
82 HRESULT WINAPI BindCtxImpl_Destroy(BindCtxImpl* This);
83 HRESULT WINAPI BindCtxImpl_GetObjectIndex(BindCtxImpl* This,IUnknown* punk,LPOLESTR pszkey,DWORD *index);
85 /* Virtual function table for the BindCtx class. */
86 static ICOM_VTABLE(IBindCtx) VT_BindCtxImpl =
88 ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
89 BindCtxImpl_QueryInterface,
92 BindCtxImpl_RegisterObjectBound,
93 BindCtxImpl_RevokeObjectBound,
94 BindCtxImpl_ReleaseBoundObjects,
95 BindCtxImpl_SetBindOptions,
96 BindCtxImpl_GetBindOptions,
97 BindCtxImpl_GetRunningObjectTable,
98 BindCtxImpl_RegisterObjectParam,
99 BindCtxImpl_GetObjectParam,
100 BindCtxImpl_EnumObjectParam,
101 BindCtxImpl_RevokeObjectParam
104 /*******************************************************************************
105 * BindCtx_QueryInterface
106 *******************************************************************************/
107 HRESULT WINAPI BindCtxImpl_QueryInterface(IBindCtx* iface,REFIID riid,void** ppvObject)
109 ICOM_THIS(BindCtxImpl,iface);
111 TRACE("(%p,%p,%p)\n",This,riid,ppvObject);
113 /* Perform a sanity check on the parameters.*/
114 if ( (This==0) || (ppvObject==0) )
117 /* Initialize the return parameter.*/
120 /* Compare the riid with the interface IDs implemented by this object.*/
121 if (IsEqualIID(&IID_IUnknown, riid))
122 *ppvObject = (IBindCtx*)This;
124 if (IsEqualIID(&IID_IBindCtx, riid))
125 *ppvObject = (IBindCtx*)This;
127 /* Check that we obtained an interface.*/
129 return E_NOINTERFACE;
131 /* Query Interface always increases the reference count by one when it is successful */
132 BindCtxImpl_AddRef(iface);
137 /******************************************************************************
139 ******************************************************************************/
140 ULONG WINAPI BindCtxImpl_AddRef(IBindCtx* iface)
142 ICOM_THIS(BindCtxImpl,iface);
144 TRACE("(%p)\n",This);
146 return ++(This->ref);
149 /******************************************************************************
151 ******************************************************************************/
152 ULONG WINAPI BindCtxImpl_Release(IBindCtx* iface)
154 ICOM_THIS(BindCtxImpl,iface);
156 TRACE("(%p)\n",This);
162 /* release all registered objects */
163 BindCtxImpl_ReleaseBoundObjects((IBindCtx*)This);
165 BindCtxImpl_Destroy(This);
173 /******************************************************************************
174 * BindCtx_Construct (local function)
175 *******************************************************************************/
176 HRESULT WINAPI BindCtxImpl_Construct(BindCtxImpl* This)
178 TRACE("(%p)\n",This);
180 /* Initialize the virtual function table.*/
181 This->lpVtbl = &VT_BindCtxImpl;
184 /* Initialize the BIND_OPTS2 structure */
185 This->bindOption2.cbStruct = sizeof(BIND_OPTS2);
186 This->bindOption2.grfFlags = 0;
187 This->bindOption2.grfMode = STGM_READWRITE;
188 This->bindOption2.dwTickCountDeadline = 0;
190 This->bindOption2.dwTrackFlags = 0;
191 This->bindOption2.dwClassContext = CLSCTX_SERVER;
192 This->bindOption2.locale = 1033;
193 This->bindOption2.pServerInfo = 0;
195 /* Initialize the bindctx table */
196 This->bindCtxTableSize=BLOCK_TAB_SIZE;
197 This->bindCtxTableLastIndex=0;
198 This->bindCtxTable= HeapAlloc(GetProcessHeap(), 0,This->bindCtxTableSize*sizeof(BindCtxObject));
200 if (This->bindCtxTable==NULL)
201 return E_OUTOFMEMORY;
206 /******************************************************************************
207 * BindCtx_Destroy (local function)
208 *******************************************************************************/
209 HRESULT WINAPI BindCtxImpl_Destroy(BindCtxImpl* This)
211 TRACE("(%p)\n",This);
213 /* free the table space memory */
214 HeapFree(GetProcessHeap(),0,This->bindCtxTable);
216 /* free the bindctx structure */
217 HeapFree(GetProcessHeap(),0,This);
223 /******************************************************************************
224 * BindCtx_RegisterObjectBound
225 ******************************************************************************/
226 HRESULT WINAPI BindCtxImpl_RegisterObjectBound(IBindCtx* iface,IUnknown* punk)
228 ICOM_THIS(BindCtxImpl,iface);
229 DWORD lastIndex=This->bindCtxTableLastIndex;
231 TRACE("(%p,%p)\n",This,punk);
236 IUnknown_AddRef(punk);
238 /* put the object in the first free element in the table */
239 This->bindCtxTable[lastIndex].pObj = punk;
240 This->bindCtxTable[lastIndex].pkeyObj = NULL;
241 This->bindCtxTable[lastIndex].regType = 0;
242 lastIndex= ++This->bindCtxTableLastIndex;
244 if (lastIndex == This->bindCtxTableSize){ /* the table is full so it must be resized */
246 if (This->bindCtxTableSize > (MAX_TAB_SIZE-BLOCK_TAB_SIZE)){
247 FIXME("This->bindCtxTableSize: %ld is out of data limite \n",This->bindCtxTableSize);
251 This->bindCtxTableSize+=BLOCK_TAB_SIZE; /* new table size */
253 This->bindCtxTable = HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,This->bindCtxTable,
254 This->bindCtxTableSize * sizeof(BindCtxObject));
255 if (!This->bindCtxTable)
256 return E_OUTOFMEMORY;
261 /******************************************************************************
262 * BindCtx_RevokeObjectBound
263 ******************************************************************************/
264 HRESULT WINAPI BindCtxImpl_RevokeObjectBound(IBindCtx* iface, IUnknown* punk)
268 ICOM_THIS(BindCtxImpl,iface);
270 TRACE("(%p,%p)\n",This,punk);
272 /* check if the object was registred or not */
273 if (BindCtxImpl_GetObjectIndex(This,punk,NULL,&index)==S_FALSE)
274 return MK_E_NOTBOUND;
276 if(This->bindCtxTable[index].pObj)
277 IUnknown_Release(This->bindCtxTable[index].pObj);
278 if(This->bindCtxTable[index].pkeyObj)
279 HeapFree(GetProcessHeap(),0,This->bindCtxTable[index].pkeyObj);
281 /* left-shift all elements in the right side of the current revoked object */
282 for(j=index; j<This->bindCtxTableLastIndex-1; j++)
283 This->bindCtxTable[j]= This->bindCtxTable[j+1];
285 This->bindCtxTableLastIndex--;
290 /******************************************************************************
291 * BindCtx_ReleaseBoundObjects
292 ******************************************************************************/
293 HRESULT WINAPI BindCtxImpl_ReleaseBoundObjects(IBindCtx* iface)
297 ICOM_THIS(BindCtxImpl,iface);
299 TRACE("(%p)\n",This);
301 for(i=0;i<This->bindCtxTableLastIndex;i++)
303 if(This->bindCtxTable[i].pObj)
304 IUnknown_Release(This->bindCtxTable[i].pObj);
305 if(This->bindCtxTable[i].pkeyObj)
306 HeapFree(GetProcessHeap(),0,This->bindCtxTable[i].pkeyObj);
309 This->bindCtxTableLastIndex = 0;
314 /******************************************************************************
315 * BindCtx_SetBindOptions
316 ******************************************************************************/
317 HRESULT WINAPI BindCtxImpl_SetBindOptions(IBindCtx* iface,BIND_OPTS *pbindopts)
319 ICOM_THIS(BindCtxImpl,iface);
321 TRACE("(%p,%p)\n",This,pbindopts);
326 if (pbindopts->cbStruct > sizeof(BIND_OPTS2))
328 WARN("invalid size\n");
329 return E_INVALIDARG; /* FIXME : not verified */
331 memcpy(&This->bindOption2, pbindopts, pbindopts->cbStruct);
335 /******************************************************************************
336 * BindCtx_GetBindOptions
337 ******************************************************************************/
338 HRESULT WINAPI BindCtxImpl_GetBindOptions(IBindCtx* iface,BIND_OPTS *pbindopts)
340 ICOM_THIS(BindCtxImpl,iface);
342 TRACE("(%p,%p)\n",This,pbindopts);
347 if (pbindopts->cbStruct > sizeof(BIND_OPTS2))
349 WARN("invalid size\n");
350 return E_INVALIDARG; /* FIXME : not verified */
352 memcpy(pbindopts, &This->bindOption2, pbindopts->cbStruct);
356 /******************************************************************************
357 * BindCtx_GetRunningObjectTable
358 ******************************************************************************/
359 HRESULT WINAPI BindCtxImpl_GetRunningObjectTable(IBindCtx* iface,IRunningObjectTable** pprot)
363 ICOM_THIS(BindCtxImpl,iface);
365 TRACE("(%p,%p)\n",This,pprot);
370 res=GetRunningObjectTable(0, pprot);
375 /******************************************************************************
376 * BindCtx_RegisterObjectParam
377 ******************************************************************************/
378 HRESULT WINAPI BindCtxImpl_RegisterObjectParam(IBindCtx* iface,LPOLESTR pszkey, IUnknown* punk)
381 ICOM_THIS(BindCtxImpl,iface);
383 TRACE("(%p,%s,%p)\n",This,debugstr_w(pszkey),punk);
388 if (pszkey!=NULL && BindCtxImpl_GetObjectIndex(This,NULL,pszkey,&index)==S_OK)
390 TRACE("Overwriting existing key\n");
391 if(This->bindCtxTable[index].pObj!=NULL)
392 IUnknown_Release(This->bindCtxTable[index].pObj);
393 This->bindCtxTable[index].pObj=punk;
394 IUnknown_AddRef(punk);
397 This->bindCtxTable[This->bindCtxTableLastIndex].pObj = punk;
398 This->bindCtxTable[This->bindCtxTableLastIndex].regType = 1;
402 This->bindCtxTable[This->bindCtxTableLastIndex].pkeyObj=NULL;
406 This->bindCtxTable[This->bindCtxTableLastIndex].pkeyObj=
407 HeapAlloc(GetProcessHeap(),0,(sizeof(WCHAR)*(1+lstrlenW(pszkey))));
409 if (This->bindCtxTable[This->bindCtxTableLastIndex].pkeyObj==NULL)
410 return E_OUTOFMEMORY;
411 strcpyW(This->bindCtxTable[This->bindCtxTableLastIndex].pkeyObj,pszkey);
414 This->bindCtxTableLastIndex++;
416 if (This->bindCtxTableLastIndex == This->bindCtxTableSize){ /* table is full ! must be resized */
418 This->bindCtxTableSize+=BLOCK_TAB_SIZE; /* new table size */
420 if (This->bindCtxTableSize > (MAX_TAB_SIZE-BLOCK_TAB_SIZE)){
421 FIXME("This->bindCtxTableSize: %ld is out of data limite \n",This->bindCtxTableSize);
424 This->bindCtxTable = HeapReAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,This->bindCtxTable,
425 This->bindCtxTableSize * sizeof(BindCtxObject));
426 if (!This->bindCtxTable)
427 return E_OUTOFMEMORY;
429 IUnknown_AddRef(punk);
433 /******************************************************************************
434 * BindCtx_GetObjectParam
435 ******************************************************************************/
436 HRESULT WINAPI BindCtxImpl_GetObjectParam(IBindCtx* iface,LPOLESTR pszkey, IUnknown** punk)
439 ICOM_THIS(BindCtxImpl,iface);
441 TRACE("(%p,%s,%p)\n",This,debugstr_w(pszkey),punk);
448 if (BindCtxImpl_GetObjectIndex(This,NULL,pszkey,&index)==S_FALSE)
451 IUnknown_AddRef(This->bindCtxTable[index].pObj);
453 *punk = This->bindCtxTable[index].pObj;
458 /******************************************************************************
459 * BindCtx_RevokeObjectParam
460 ******************************************************************************/
461 HRESULT WINAPI BindCtxImpl_RevokeObjectParam(IBindCtx* iface,LPOLESTR ppenum)
465 ICOM_THIS(BindCtxImpl,iface);
467 TRACE("(%p,%s)\n",This,debugstr_w(ppenum));
469 if (BindCtxImpl_GetObjectIndex(This,NULL,ppenum,&index)==S_FALSE)
472 /* release the object if it's found */
473 if(This->bindCtxTable[index].pObj)
474 IUnknown_Release(This->bindCtxTable[index].pObj);
475 if(This->bindCtxTable[index].pkeyObj)
476 HeapFree(GetProcessHeap(),0,This->bindCtxTable[index].pkeyObj);
478 /* remove the object from the table with a left-shifting of all objects in the right side */
479 for(j=index; j<This->bindCtxTableLastIndex-1; j++)
480 This->bindCtxTable[j]= This->bindCtxTable[j+1];
482 This->bindCtxTableLastIndex--;
487 /******************************************************************************
488 * BindCtx_EnumObjectParam
489 ******************************************************************************/
490 HRESULT WINAPI BindCtxImpl_EnumObjectParam(IBindCtx* iface,IEnumString** pszkey)
492 FIXME("(%p,%p),stub!\n",iface,pszkey);
496 /********************************************************************************
497 * GetObjectIndex (local function)
498 ********************************************************************************/
499 HRESULT WINAPI BindCtxImpl_GetObjectIndex(BindCtxImpl* This,
508 TRACE("(%p,%p,%p,%p)\n",This,punk,pszkey,index);
511 /* search object identified by a register key */
512 for(i=0; ( (i<This->bindCtxTableLastIndex) && (!found));i++){
514 if(This->bindCtxTable[i].regType==1){
516 if ( ( (This->bindCtxTable[i].pkeyObj==NULL) && (pszkey==NULL) ) ||
517 ( (This->bindCtxTable[i].pkeyObj!=NULL) &&
519 (lstrcmpW(This->bindCtxTable[i].pkeyObj,pszkey)==0)
527 /* search object identified by a moniker*/
528 for(i=0; ( (i<This->bindCtxTableLastIndex) && (!found));i++)
529 if(This->bindCtxTable[i].pObj==punk)
537 TRACE("key not found\n");
541 /******************************************************************************
543 ******************************************************************************/
544 HRESULT WINAPI CreateBindCtx16(DWORD reserved, LPBC * ppbc)
546 FIXME("(%ld,%p),stub!\n",reserved,ppbc);
550 /******************************************************************************
551 * CreateBindCtx (OLE32.@)
552 ******************************************************************************/
553 HRESULT WINAPI CreateBindCtx(DWORD reserved, LPBC * ppbc)
555 BindCtxImpl* newBindCtx = 0;
557 IID riid=IID_IBindCtx;
559 TRACE("(%ld,%p)\n",reserved,ppbc);
561 newBindCtx = HeapAlloc(GetProcessHeap(), 0, sizeof(BindCtxImpl));
564 return E_OUTOFMEMORY;
566 hr = BindCtxImpl_Construct(newBindCtx);
570 HeapFree(GetProcessHeap(),0,newBindCtx);
574 hr = BindCtxImpl_QueryInterface((IBindCtx*)newBindCtx,&riid,(void**)ppbc);
579 HRESULT WINAPI BindMoniker(LPMONIKER pmk, DWORD grfOpt, REFIID riid, LPVOID * ppvResult)
584 TRACE("(%p, %lx, %s, %p)\n", pmk, grfOpt, debugstr_guid(riid), ppvResult);
586 res = CreateBindCtx(grfOpt, &pbc);
588 res = IMoniker_BindToObject(pmk, pbc, NULL, riid, ppvResult);