dlls/msxml3/nodemap.c

   1 /*
   2  *    Node map implementation
   3  *
   4  * Copyright 2005 Mike McCormack
   5  *
   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.
  10  *
  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.
  15  *
  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
  19  */
  20
  21 #include "config.h"
  22
  23 #define COBJMACROS
  24
  25 #include <stdarg.h>
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winuser.h"
  29 #include "winnls.h"
  30 #include "ole2.h"
  31 #include "msxml2.h"
  32
  33 #include "msxml_private.h"
  34
  35 #include "wine/debug.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(msxml);
  38
  39 #ifdef HAVE_LIBXML2
  40
  41 typedef struct _xmlnodemap
  42 {
  43     const struct IXMLDOMNamedNodeMapVtbl *lpVtbl;
  44     const struct ISupportErrorInfoVtbl *lpSEIVtbl;
  45     LONG ref;
  46     IXMLDOMNode *node;
  47     LONG iterator;
  48 } xmlnodemap;
  49
  50 static inline xmlnodemap *impl_from_IXMLDOMNamedNodeMap( IXMLDOMNamedNodeMap *iface )
  51 {
  52     return (xmlnodemap *)((char*)iface - FIELD_OFFSET(xmlnodemap, lpVtbl));
  53 }
  54
  55 static inline xmlnodemap *impl_from_ISupportErrorInfo( ISupportErrorInfo *iface )
  56 {
  57     return (xmlnodemap *)((char*)iface - FIELD_OFFSET(xmlnodemap, lpSEIVtbl));
  58 }
  59
  60 static HRESULT WINAPI xmlnodemap_QueryInterface(
  61     IXMLDOMNamedNodeMap *iface,
  62     REFIID riid, void** ppvObject )
  63 {
  64     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
  65     TRACE("%p %s %p\n", iface, debugstr_guid(riid), ppvObject);
  66
  67     if( IsEqualGUID( riid, &IID_IUnknown ) ||
  68         IsEqualGUID( riid, &IID_IDispatch ) ||
  69         IsEqualGUID( riid, &IID_IXMLDOMNamedNodeMap ) )
  70     {
  71         *ppvObject = iface;
  72     }
  73     else if( IsEqualGUID( riid, &IID_ISupportErrorInfo ))
  74     {
  75         *ppvObject = &This->lpSEIVtbl;
  76     }
  77     else
  78     {
  79         FIXME("interface %s not implemented\n", debugstr_guid(riid));
  80         return E_NOINTERFACE;
  81     }
  82
  83     IXMLDOMElement_AddRef( iface );
  84
  85     return S_OK;
  86 }
  87
  88 static ULONG WINAPI xmlnodemap_AddRef(
  89     IXMLDOMNamedNodeMap *iface )
  90 {
  91     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
  92     return InterlockedIncrement( &This->ref );
  93 }
  94
  95 static ULONG WINAPI xmlnodemap_Release(
  96     IXMLDOMNamedNodeMap *iface )
  97 {
  98     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
  99     ULONG ref;
 100
 101     ref = InterlockedDecrement( &This->ref );
 102     if ( ref == 0 )
 103     {
 104         IXMLDOMNode_Release( This->node );
 105         heap_free( This );
 106     }
 107
 108     return ref;
 109 }
 110
 111 static HRESULT WINAPI xmlnodemap_GetTypeInfoCount(
 112     IXMLDOMNamedNodeMap *iface,
 113     UINT* pctinfo )
 114 {
 115     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 116
 117     TRACE("(%p)->(%p)\n", This, pctinfo);
 118
 119     *pctinfo = 1;
 120
 121     return S_OK;
 122 }
 123
 124 static HRESULT WINAPI xmlnodemap_GetTypeInfo(
 125     IXMLDOMNamedNodeMap *iface,
 126     UINT iTInfo, LCID lcid,
 127     ITypeInfo** ppTInfo )
 128 {
 129     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 130     HRESULT hr;
 131
 132     TRACE("(%p)->(%u %u %p)\n", This, iTInfo, lcid, ppTInfo);
 133
 134     hr = get_typeinfo(IXMLDOMNamedNodeMap_tid, ppTInfo);
 135
 136     return hr;
 137 }
 138
 139 static HRESULT WINAPI xmlnodemap_GetIDsOfNames(
 140     IXMLDOMNamedNodeMap *iface,
 141     REFIID riid, LPOLESTR* rgszNames,
 142     UINT cNames, LCID lcid, DISPID* rgDispId )
 143 {
 144     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 145     ITypeInfo *typeinfo;
 146     HRESULT hr;
 147
 148     TRACE("(%p)->(%s %p %u %u %p)\n", This, debugstr_guid(riid), rgszNames, cNames,
 149           lcid, rgDispId);
 150
 151     if(!rgszNames || cNames == 0 || !rgDispId)
 152         return E_INVALIDARG;
 153
 154     hr = get_typeinfo(IXMLDOMNamedNodeMap_tid, &typeinfo);
 155     if(SUCCEEDED(hr))
 156     {
 157         hr = ITypeInfo_GetIDsOfNames(typeinfo, rgszNames, cNames, rgDispId);
 158         ITypeInfo_Release(typeinfo);
 159     }
 160
 161     return hr;
 162 }
 163
 164 static HRESULT WINAPI xmlnodemap_Invoke(
 165     IXMLDOMNamedNodeMap *iface,
 166     DISPID dispIdMember, REFIID riid, LCID lcid,
 167     WORD wFlags, DISPPARAMS* pDispParams, VARIANT* pVarResult,
 168     EXCEPINFO* pExcepInfo, UINT* puArgErr )
 169 {
 170     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 171     ITypeInfo *typeinfo;
 172     HRESULT hr;
 173
 174     TRACE("(%p)->(%d %s %d %d %p %p %p %p)\n", This, dispIdMember, debugstr_guid(riid),
 175           lcid, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
 176
 177     hr = get_typeinfo(IXMLDOMNamedNodeMap_tid, &typeinfo);
 178     if(SUCCEEDED(hr))
 179     {
 180         hr = ITypeInfo_Invoke(typeinfo, &(This->lpVtbl), dispIdMember, wFlags, pDispParams,
 181                 pVarResult, pExcepInfo, puArgErr);
 182         ITypeInfo_Release(typeinfo);
 183     }
 184
 185     return hr;
 186 }
 187
 188 xmlChar *xmlChar_from_wchar( LPWSTR str )
 189 {
 190     DWORD len;
 191     xmlChar *xmlstr;
 192
 193     len = WideCharToMultiByte( CP_UTF8, 0, str, -1, NULL, 0, NULL, NULL );
 194     xmlstr = heap_alloc( len );
 195     if ( xmlstr )
 196         WideCharToMultiByte( CP_UTF8, 0, str, -1, (LPSTR) xmlstr, len, NULL, NULL );
 197     return xmlstr;
 198 }
 199
 200 static HRESULT WINAPI xmlnodemap_getNamedItem(
 201     IXMLDOMNamedNodeMap *iface,
 202     BSTR name,
 203     IXMLDOMNode** namedItem)
 204 {
 205     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 206     xmlChar *element_name;
 207     xmlAttrPtr attr;
 208     xmlNodePtr node;
 209
 210     TRACE("%p %s %p\n", This, debugstr_w(name), namedItem );
 211
 212     if ( !namedItem )
 213         return E_INVALIDARG;
 214
 215     node = xmlNodePtr_from_domnode( This->node, 0 );
 216     if ( !node )
 217         return E_FAIL;
 218
 219     element_name = xmlChar_from_wchar( name );
 220     attr = xmlHasNsProp( node, element_name, NULL );
 221     heap_free( element_name );
 222
 223     if ( !attr )
 224     {
 225         *namedItem = NULL;
 226         return S_FALSE;
 227     }
 228
 229     *namedItem = create_node( (xmlNodePtr) attr );
 230
 231     return S_OK;
 232 }
 233
 234 static HRESULT WINAPI xmlnodemap_setNamedItem(
 235     IXMLDOMNamedNodeMap *iface,
 236     IXMLDOMNode* newItem,
 237     IXMLDOMNode** namedItem)
 238 {
 239     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 240     xmlnode *ThisNew = NULL;
 241     xmlNodePtr nodeNew;
 242     IXMLDOMNode *pAttr = NULL;
 243     xmlNodePtr node;
 244
 245     TRACE("%p %p %p\n", This, newItem, namedItem );
 246
 247     if(!newItem)
 248         return E_INVALIDARG;
 249
 250     if(namedItem) *namedItem = NULL;
 251
 252     node = xmlNodePtr_from_domnode( This->node, 0 );
 253     if ( !node )
 254         return E_FAIL;
 255
 256     /* Must be an Attribute */
 257     IUnknown_QueryInterface(newItem, &IID_IXMLDOMNode, (LPVOID*)&pAttr);
 258     if(pAttr)
 259     {
 260         ThisNew = impl_from_IXMLDOMNode( pAttr );
 261
 262         if(ThisNew->node->type != XML_ATTRIBUTE_NODE)
 263         {
 264             IUnknown_Release(pAttr);
 265             return E_FAIL;
 266         }
 267
 268         if(!ThisNew->node->parent)
 269             if(xmldoc_remove_orphan(ThisNew->node->doc, ThisNew->node) != S_OK)
 270                 WARN("%p is not an orphan of %p\n", ThisNew->node, ThisNew->node->doc);
 271
 272         nodeNew = xmlAddChild(node, ThisNew->node);
 273
 274         if(namedItem)
 275             *namedItem = create_node( nodeNew );
 276
 277         IUnknown_Release(pAttr);
 278
 279         return S_OK;
 280     }
 281
 282     return E_INVALIDARG;
 283 }
 284
 285 static HRESULT WINAPI xmlnodemap_removeNamedItem(
 286     IXMLDOMNamedNodeMap *iface,
 287     BSTR name,
 288     IXMLDOMNode** namedItem)
 289 {
 290     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 291     xmlChar *element_name;
 292     xmlAttrPtr attr;
 293     xmlNodePtr node;
 294
 295     TRACE("%p %s %p\n", This, debugstr_w(name), namedItem );
 296
 297     if ( !name)
 298         return E_INVALIDARG;
 299
 300     node = xmlNodePtr_from_domnode( This->node, 0 );
 301     if ( !node )
 302         return E_FAIL;
 303
 304     element_name = xmlChar_from_wchar( name );
 305     attr = xmlHasNsProp( node, element_name, NULL );
 306     heap_free( element_name );
 307
 308     if ( !attr )
 309     {
 310         if( namedItem )
 311             *namedItem = NULL;
 312         return S_FALSE;
 313     }
 314
 315     if ( namedItem )
 316     {
 317         xmlUnlinkNode( (xmlNodePtr) attr );
 318         xmldoc_add_orphan( attr->doc, (xmlNodePtr) attr );
 319         *namedItem = create_node( (xmlNodePtr) attr );
 320     }
 321     else
 322     {
 323         if( xmlRemoveProp( attr ) == -1 )
 324             ERR("xmlRemoveProp failed\n");
 325     }
 326
 327     return S_OK;
 328 }
 329
 330 static HRESULT WINAPI xmlnodemap_get_item(
 331     IXMLDOMNamedNodeMap *iface,
 332     LONG index,
 333     IXMLDOMNode** listItem)
 334 {
 335     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 336     xmlNodePtr node;
 337     xmlAttrPtr curr;
 338     LONG attrIndex;
 339
 340     TRACE("%p %d\n", This, index);
 341
 342     *listItem = NULL;
 343
 344     if (index < 0)
 345         return S_FALSE;
 346
 347     node = xmlNodePtr_from_domnode( This->node, 0 );
 348     curr = node->properties;
 349
 350     for (attrIndex = 0; attrIndex < index; attrIndex++) {
 351         if (curr->next == NULL)
 352             return S_FALSE;
 353         else
 354             curr = curr->next;
 355     }
 356
 357     *listItem = create_node( (xmlNodePtr) curr );
 358
 359     return S_OK;
 360 }
 361
 362 static HRESULT WINAPI xmlnodemap_get_length(
 363     IXMLDOMNamedNodeMap *iface,
 364     LONG *listLength)
 365 {
 366     xmlNodePtr node;
 367     xmlAttrPtr first;
 368     xmlAttrPtr curr;
 369     LONG attrCount;
 370
 371     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 372
 373     TRACE("%p\n", This);
 374
 375     if( !listLength )
 376         return E_INVALIDARG;
 377
 378     node = xmlNodePtr_from_domnode( This->node, 0 );
 379     if ( !node )
 380         return E_FAIL;
 381
 382     first = node->properties;
 383     if (first == NULL) {
 384         *listLength = 0;
 385         return S_OK;
 386     }
 387
 388     curr = first;
 389     attrCount = 1;
 390     while (curr->next != NULL) {
 391         attrCount++;
 392         curr = curr->next;
 393     }
 394     *listLength = attrCount;
 395
 396     return S_OK;
 397 }
 398
 399 static HRESULT WINAPI xmlnodemap_getQualifiedItem(
 400     IXMLDOMNamedNodeMap *iface,
 401     BSTR baseName,
 402     BSTR namespaceURI,
 403     IXMLDOMNode** qualifiedItem)
 404 {
 405     FIXME("\n");
 406     return E_NOTIMPL;
 407 }
 408
 409 static HRESULT WINAPI xmlnodemap_removeQualifiedItem(
 410     IXMLDOMNamedNodeMap *iface,
 411     BSTR baseName,
 412     BSTR namespaceURI,
 413     IXMLDOMNode** qualifiedItem)
 414 {
 415     FIXME("\n");
 416     return E_NOTIMPL;
 417 }
 418
 419 static HRESULT WINAPI xmlnodemap_nextNode(
 420     IXMLDOMNamedNodeMap *iface,
 421     IXMLDOMNode** nextItem)
 422 {
 423     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 424     xmlNodePtr node;
 425     xmlAttrPtr curr;
 426     LONG attrIndex;
 427
 428     TRACE("%p %d\n", This, This->iterator);
 429
 430     *nextItem = NULL;
 431
 432     node = xmlNodePtr_from_domnode( This->node, 0 );
 433     curr = node->properties;
 434
 435     for (attrIndex = 0; attrIndex < This->iterator; attrIndex++) {
 436         if (curr->next == NULL)
 437             return S_FALSE;
 438         else
 439             curr = curr->next;
 440     }
 441
 442     This->iterator++;
 443
 444     *nextItem = create_node( (xmlNodePtr) curr );
 445
 446     return S_OK;
 447 }
 448
 449 static HRESULT WINAPI xmlnodemap_reset(
 450     IXMLDOMNamedNodeMap *iface )
 451 {
 452     xmlnodemap *This = impl_from_IXMLDOMNamedNodeMap( iface );
 453
 454     TRACE("%p %d\n", This, This->iterator);
 455
 456     This->iterator = 0;
 457
 458     return S_OK;
 459 }
 460
 461 static HRESULT WINAPI xmlnodemap__newEnum(
 462     IXMLDOMNamedNodeMap *iface,
 463     IUnknown** ppUnk)
 464 {
 465     FIXME("\n");
 466     return E_NOTIMPL;
 467 }
 468
 469 static const struct IXMLDOMNamedNodeMapVtbl xmlnodemap_vtbl =
 470 {
 471     xmlnodemap_QueryInterface,
 472     xmlnodemap_AddRef,
 473     xmlnodemap_Release,
 474     xmlnodemap_GetTypeInfoCount,
 475     xmlnodemap_GetTypeInfo,
 476     xmlnodemap_GetIDsOfNames,
 477     xmlnodemap_Invoke,
 478     xmlnodemap_getNamedItem,
 479     xmlnodemap_setNamedItem,
 480     xmlnodemap_removeNamedItem,
 481     xmlnodemap_get_item,
 482     xmlnodemap_get_length,
 483     xmlnodemap_getQualifiedItem,
 484     xmlnodemap_removeQualifiedItem,
 485     xmlnodemap_nextNode,
 486     xmlnodemap_reset,
 487     xmlnodemap__newEnum,
 488 };
 489
 490 static HRESULT WINAPI support_error_QueryInterface(
 491     ISupportErrorInfo *iface,
 492     REFIID riid, void** ppvObject )
 493 {
 494     xmlnodemap *This = impl_from_ISupportErrorInfo( iface );
 495     TRACE("%p %s %p\n", iface, debugstr_guid(riid), ppvObject);
 496
 497     return IXMLDOMNamedNodeMap_QueryInterface((IXMLDOMNamedNodeMap*)&This->lpVtbl, riid, ppvObject);
 498 }
 499
 500 static ULONG WINAPI support_error_AddRef(
 501     ISupportErrorInfo *iface )
 502 {
 503     xmlnodemap *This = impl_from_ISupportErrorInfo( iface );
 504     return IXMLDOMNamedNodeMap_AddRef((IXMLDOMNamedNodeMap*)&This->lpVtbl);
 505 }
 506
 507 static ULONG WINAPI support_error_Release(
 508     ISupportErrorInfo *iface )
 509 {
 510     xmlnodemap *This = impl_from_ISupportErrorInfo( iface );
 511     return IXMLDOMNamedNodeMap_Release((IXMLDOMNamedNodeMap*)&This->lpVtbl);
 512 }
 513
 514 static HRESULT WINAPI support_error_InterfaceSupportsErrorInfo(
 515     ISupportErrorInfo *iface,
 516     REFIID riid )
 517 {
 518     FIXME("(%p)->(%s)\n", iface, debugstr_guid(riid));
 519     return S_FALSE;
 520 }
 521
 522 static const struct ISupportErrorInfoVtbl support_error_vtbl =
 523 {
 524     support_error_QueryInterface,
 525     support_error_AddRef,
 526     support_error_Release,
 527     support_error_InterfaceSupportsErrorInfo
 528 };
 529
 530 IXMLDOMNamedNodeMap *create_nodemap( IXMLDOMNode *node )
 531 {
 532     xmlnodemap *nodemap;
 533
 534     nodemap = heap_alloc( sizeof *nodemap );
 535     if ( !nodemap )
 536         return NULL;
 537
 538     nodemap->lpVtbl = &xmlnodemap_vtbl;
 539     nodemap->lpSEIVtbl = &support_error_vtbl;
 540     nodemap->node = node;
 541     nodemap->ref = 1;
 542     nodemap->iterator = 0;
 543
 544     IXMLDOMNode_AddRef( node );
 545     /* Since we AddRef a node here, we don't need to call xmldoc_add_ref() */
 546
 547     return (IXMLDOMNamedNodeMap*) &nodemap->lpVtbl;
 548 }
 549
 550 #endif