dlls/msxml3/nodelist.c

   1 /*
   2  *    Node list 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 #define COBJMACROS
  22
  23 #include "config.h"
  24
  25 #include <stdarg.h>
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winuser.h"
  29 #include "ole2.h"
  30 #include "msxml2.h"
  31
  32 #include "msxml_private.h"
  33
  34 #include "wine/debug.h"
  35
  36 WINE_DEFAULT_DEBUG_CHANNEL(msxml);
  37
  38 #ifdef HAVE_LIBXML2
  39
  40 #ifdef HAVE_LIBXSLT
  41
  42 #ifdef HAVE_LIBXSLT_PATTERN_H
  43 #include <libxslt/pattern.h>
  44 #endif
  45 #ifdef HAVE_LIBXSLT_TRANSFORM_H
  46 #include <libxslt/transform.h>
  47 #endif
  48
  49 struct xslt_info {
  50     xsltTransformContextPtr ctxt;
  51     xsltCompMatchPtr pattern;
  52     xsltStylesheetPtr sheet;
  53 };
  54
  55 static void xslt_info_init( struct xslt_info *info )
  56 {
  57     info->ctxt = NULL;
  58     info->pattern = NULL;
  59     info->sheet = NULL;
  60 }
  61
  62 static int create_xslt_parser( struct xslt_info *info, xmlNodePtr node, const xmlChar *str )
  63 {
  64     if(!node) return 1;
  65
  66     info->sheet = xsltNewStylesheet();
  67     if (!info->sheet)
  68         return 0;
  69
  70     info->ctxt = xsltNewTransformContext( info->sheet, node->doc );
  71     if (!info->ctxt)
  72         return 0;
  73
  74     info->pattern = xsltCompilePattern( str, node->doc,
  75                                         node, info->sheet, info->ctxt );
  76     if (!info->pattern)
  77         return 0;
  78     return 1;
  79 }
  80
  81 static void free_xslt_info( struct xslt_info *info )
  82 {
  83     if (info->pattern)
  84         xsltFreeCompMatchList( info->pattern );
  85     if (info->sheet)
  86         xsltFreeStylesheet( info->sheet );
  87     if (info->ctxt)
  88         xsltFreeTransformContext( info->ctxt );
  89 }
  90
  91
  92 static xmlNodePtr get_next_node( struct xslt_info *info, xmlNodePtr node, xmlNodePtr *top_level_node );
  93
  94 static HRESULT xslt_next_match( struct xslt_info *info, xmlNodePtr *node, xmlNodePtr *top_level_node )
  95 {
  96     if (!info->ctxt)
  97         return S_FALSE;
  98
  99     /* make sure that the current element matches the pattern */
 100     while ( *node )
 101     {
 102         int r;
 103
 104         r = xsltTestCompMatchList( info->ctxt, *node, info->pattern );
 105         if ( 1 == r )
 106         {
 107             TRACE("Matched %p (%s)\n", *node, (*node)->name );
 108             return S_OK;
 109         }
 110         if (r != 0)
 111         {
 112             ERR("Pattern match failed\n");
 113             return E_FAIL;
 114         }
 115         *node = get_next_node(info, *node, top_level_node);
 116     }
 117     return S_OK;
 118 }
 119
 120 #else
 121
 122 struct xslt_info {
 123     /* empty */
 124 };
 125
 126 static void xslt_info_init( struct xslt_info *info )
 127 {
 128 }
 129
 130 void free_xslt_info( struct xslt_info *info )
 131 {
 132 }
 133
 134 static int create_xslt_parser( struct xslt_info *info, xmlNodePtr node, const xmlChar *str )
 135 {
 136     MESSAGE("libxslt was missing at compile time\n");
 137     return 0;
 138 }
 139
 140 static HRESULT xslt_next_match( struct xslt_info *info, xmlNodePtr *node, xmlNodePtr *top_level_node )
 141 {
 142     return S_FALSE;
 143 }
 144
 145 #endif
 146
 147 static xmlNodePtr get_next_node( struct xslt_info *info, xmlNodePtr node, xmlNodePtr *top_level_node )
 148 {
 149     if(!top_level_node) return node->next;
 150
 151     if(node->children) return node->children;
 152     if(node->next)
 153     {
 154         if(node == *top_level_node)
 155             *top_level_node = node->next;
 156         return node->next;
 157     }
 158
 159     if(node != *top_level_node && node->parent)
 160     {
 161         if(node->parent == *top_level_node)
 162             *top_level_node = node->parent->next;
 163         return node->parent->next;
 164     }
 165     return NULL;
 166 }
 167
 168 typedef struct _xmlnodelist
 169 {
 170     const struct IXMLDOMNodeListVtbl *lpVtbl;
 171     LONG ref;
 172     xmlNodePtr node;
 173     xmlNodePtr current;
 174     xmlNodePtr top_level_node;
 175     BOOL enum_children;
 176     struct xslt_info xinfo;
 177 } xmlnodelist;
 178
 179 static inline xmlnodelist *impl_from_IXMLDOMNodeList( IXMLDOMNodeList *iface )
 180 {
 181     return (xmlnodelist *)((char*)iface - FIELD_OFFSET(xmlnodelist, lpVtbl));
 182 }
 183
 184 static HRESULT WINAPI xmlnodelist_QueryInterface(
 185     IXMLDOMNodeList *iface,
 186     REFIID riid,
 187     void** ppvObject )
 188 {
 189     TRACE("%p %s %p\n", iface, debugstr_guid(riid), ppvObject);
 190
 191     if ( IsEqualGUID( riid, &IID_IUnknown ) ||
 192          IsEqualGUID( riid, &IID_IDispatch ) ||
 193          IsEqualGUID( riid, &IID_IXMLDOMNodeList ) )
 194     {
 195         *ppvObject = iface;
 196     }
 197     else
 198     {
 199         FIXME("interface %s not implemented\n", debugstr_guid(riid));
 200         *ppvObject = NULL;
 201         return E_NOINTERFACE;
 202     }
 203
 204     IXMLDOMNodeList_AddRef( iface );
 205
 206     return S_OK;
 207 }
 208
 209 static ULONG WINAPI xmlnodelist_AddRef(
 210     IXMLDOMNodeList *iface )
 211 {
 212     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 213     return InterlockedIncrement( &This->ref );
 214 }
 215
 216 static ULONG WINAPI xmlnodelist_Release(
 217     IXMLDOMNodeList *iface )
 218 {
 219     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 220     ULONG ref;
 221
 222     ref = InterlockedDecrement( &This->ref );
 223     if ( ref == 0 )
 224     {
 225         free_xslt_info( &This->xinfo );
 226         if(This->node) xmldoc_release( This->node->doc );
 227         HeapFree( GetProcessHeap(), 0, This );
 228     }
 229
 230     return ref;
 231 }
 232
 233 static HRESULT WINAPI xmlnodelist_GetTypeInfoCount(
 234     IXMLDOMNodeList *iface,
 235     UINT* pctinfo )
 236 {
 237     FIXME("\n");
 238     return E_NOTIMPL;
 239 }
 240
 241 static HRESULT WINAPI xmlnodelist_GetTypeInfo(
 242     IXMLDOMNodeList *iface,
 243     UINT iTInfo,
 244     LCID lcid,
 245     ITypeInfo** ppTInfo )
 246 {
 247     FIXME("\n");
 248     return E_NOTIMPL;
 249 }
 250
 251 static HRESULT WINAPI xmlnodelist_GetIDsOfNames(
 252     IXMLDOMNodeList *iface,
 253     REFIID riid,
 254     LPOLESTR* rgszNames,
 255     UINT cNames,
 256     LCID lcid,
 257     DISPID* rgDispId )
 258 {
 259     FIXME("\n");
 260     return E_NOTIMPL;
 261 }
 262
 263 static HRESULT WINAPI xmlnodelist_Invoke(
 264     IXMLDOMNodeList *iface,
 265     DISPID dispIdMember,
 266     REFIID riid,
 267     LCID lcid,
 268     WORD wFlags,
 269     DISPPARAMS* pDispParams,
 270     VARIANT* pVarResult,
 271     EXCEPINFO* pExcepInfo,
 272     UINT* puArgErr )
 273 {
 274     FIXME("\n");
 275     return E_NOTIMPL;
 276 }
 277
 278 static HRESULT WINAPI xmlnodelist_get_item(
 279         IXMLDOMNodeList* iface,
 280         long index,
 281         IXMLDOMNode** listItem)
 282 {
 283     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 284     xmlNodePtr curr, tmp;
 285     xmlNodePtr *top_level_node = NULL;
 286     long nodeIndex = 0;
 287     HRESULT r;
 288
 289     TRACE("%p %ld\n", This, index);
 290
 291     *listItem = NULL;
 292
 293     if (index < 0)
 294         return S_FALSE;
 295
 296     curr = This->node;
 297
 298     if(This->enum_children)
 299     {
 300         tmp = curr;
 301         top_level_node = &tmp;
 302     }
 303
 304     while(curr)
 305     {
 306         r = xslt_next_match( &This->xinfo, &curr, top_level_node);
 307         if(FAILED(r) || !curr) return S_FALSE;
 308         if(nodeIndex++ == index) break;
 309         curr = get_next_node(&This->xinfo, curr, top_level_node);
 310     }
 311     if(!curr) return S_FALSE;
 312
 313     *listItem = create_node( curr );
 314
 315     return S_OK;
 316 }
 317
 318 static HRESULT WINAPI xmlnodelist_get_length(
 319         IXMLDOMNodeList* iface,
 320         long* listLength)
 321 {
 322
 323     xmlNodePtr curr, tmp;
 324     xmlNodePtr *top_level_node = NULL;
 325     long nodeCount = 0;
 326     HRESULT r;
 327
 328     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 329
 330     TRACE("%p\n", This);
 331
 332     if (This->node == NULL) {
 333         *listLength = 0;
 334         return S_OK;
 335     }
 336
 337     curr = This->node;
 338
 339     if(This->enum_children)
 340     {
 341         tmp = curr;
 342         top_level_node = &tmp;
 343     }
 344
 345     while (curr)
 346     {
 347         r = xslt_next_match( &This->xinfo, &curr, top_level_node );
 348         if(FAILED(r) || !curr) break;
 349         nodeCount++;
 350         curr = get_next_node(&This->xinfo, curr, top_level_node);
 351     }
 352
 353     *listLength = nodeCount;
 354     return S_OK;
 355 }
 356
 357 static HRESULT WINAPI xmlnodelist_nextNode(
 358         IXMLDOMNodeList* iface,
 359         IXMLDOMNode** nextItem)
 360 {
 361     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 362     HRESULT r;
 363     xmlNodePtr *top_level_node = NULL;
 364
 365     TRACE("%p %p\n", This, nextItem );
 366
 367     *nextItem = NULL;
 368
 369     if(This->enum_children)
 370         top_level_node = &This->top_level_node;
 371
 372     r = xslt_next_match( &This->xinfo, &This->current, top_level_node );
 373     if (FAILED(r) )
 374         return r;
 375
 376     if (!This->current)
 377         return S_FALSE;
 378
 379     *nextItem = create_node( This->current );
 380     This->current = get_next_node(&This->xinfo, This->current, top_level_node);
 381     return S_OK;
 382 }
 383
 384 static HRESULT WINAPI xmlnodelist_reset(
 385         IXMLDOMNodeList* iface)
 386 {
 387     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 388
 389     TRACE("%p\n", This);
 390     This->current = This->node;
 391     return S_OK;
 392 }
 393
 394 static HRESULT WINAPI xmlnodelist__newEnum(
 395         IXMLDOMNodeList* iface,
 396         IUnknown** ppUnk)
 397 {
 398     FIXME("\n");
 399     return E_NOTIMPL;
 400 }
 401
 402
 403 static const struct IXMLDOMNodeListVtbl xmlnodelist_vtbl =
 404 {
 405     xmlnodelist_QueryInterface,
 406     xmlnodelist_AddRef,
 407     xmlnodelist_Release,
 408     xmlnodelist_GetTypeInfoCount,
 409     xmlnodelist_GetTypeInfo,
 410     xmlnodelist_GetIDsOfNames,
 411     xmlnodelist_Invoke,
 412     xmlnodelist_get_item,
 413     xmlnodelist_get_length,
 414     xmlnodelist_nextNode,
 415     xmlnodelist_reset,
 416     xmlnodelist__newEnum,
 417 };
 418
 419 static xmlnodelist *new_nodelist( xmlNodePtr node )
 420 {
 421     xmlnodelist *nodelist;
 422
 423     nodelist = HeapAlloc( GetProcessHeap(), 0, sizeof *nodelist );
 424     if ( !nodelist )
 425         return NULL;
 426
 427     nodelist->lpVtbl = &xmlnodelist_vtbl;
 428     nodelist->ref = 1;
 429     nodelist->node = node;
 430     nodelist->current = node;
 431     nodelist->top_level_node = node;
 432     nodelist->enum_children = FALSE;
 433     xslt_info_init( &nodelist->xinfo );
 434
 435     if(node) xmldoc_add_ref( node->doc );
 436
 437     return nodelist;
 438 }
 439
 440 IXMLDOMNodeList* create_nodelist( xmlNodePtr node )
 441 {
 442     xmlnodelist *nodelist = new_nodelist( node );
 443     return (IXMLDOMNodeList*) &nodelist->lpVtbl;
 444 }
 445
 446 IXMLDOMNodeList* create_filtered_nodelist( xmlNodePtr node, const xmlChar *str, BOOL enum_children )
 447 {
 448     xmlnodelist *This = new_nodelist( node );
 449     if (create_xslt_parser( &This->xinfo, node, str ))
 450     {
 451         This->enum_children = enum_children;
 452         return (IXMLDOMNodeList*) &This->lpVtbl;
 453     }
 454
 455     IXMLDOMNodeList_Release( (IXMLDOMNodeList*) &This->lpVtbl );
 456     return NULL;
 457 }
 458
 459 #endif