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.52)
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
);