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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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 xlst_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     info->sheet = xsltNewStylesheet();
  65     if (!info->sheet)
  66         return 0;
  67
  68     info->ctxt = xsltNewTransformContext( info->sheet, node->doc );
  69     if (!info->ctxt)
  70         return 0;
  71
  72     info->pattern = xsltCompilePattern( str, node->doc,
  73                                         node, info->sheet, info->ctxt );
  74     if (!info->pattern)
  75         return 0;
  76     return 1;
  77 }
  78
  79 void free_xslt_info( struct xslt_info *info )
  80 {
  81     if (info->pattern)
  82         xsltFreeCompMatchList( info->pattern );
  83     if (info->sheet)
  84         xsltFreeStylesheet( info->sheet );
  85     if (info->ctxt)
  86         xsltFreeTransformContext( info->ctxt );
  87 }
  88
  89 static HRESULT xslt_next_match( struct xslt_info *info, xmlNodePtr *node )
  90 {
  91     if (!info->ctxt)
  92         return S_FALSE;
  93
  94     /* make sure that the current element matches the pattern */
  95     while ( *node )
  96     {
  97         int r;
  98
  99         r = xsltTestCompMatchList( info->ctxt, *node, info->pattern );
 100         if ( 1 == r )
 101         {
 102             TRACE("Matched %p (%s)\n", *node, (*node)->name );
 103             return S_OK;
 104         }
 105         if (r != 0)
 106         {
 107             ERR("Pattern match failed\n");
 108             return E_FAIL;
 109         }
 110         *node = (*node)->next;
 111     }
 112     return S_OK;
 113 }
 114
 115 #else
 116
 117 struct xslt_info {
 118     /* empty */
 119 };
 120
 121 static void xlst_info_init( struct xslt_info *info )
 122 {
 123 }
 124
 125 void free_xslt_info( struct xslt_info *info )
 126 {
 127 }
 128
 129 static int create_xslt_parser( struct xslt_info *info, xmlNodePtr node, const xmlChar *str )
 130 {
 131     MESSAGE("libxslt was missing at compile time\n");
 132     return 0;
 133 }
 134
 135 static HRESULT xslt_next_match( struct xslt_info *info, xmlNodePtr *node )
 136 {
 137     return S_FALSE;
 138 }
 139
 140 #endif
 141
 142 typedef struct _xmlnodelist
 143 {
 144     const struct IXMLDOMNodeListVtbl *lpVtbl;
 145     LONG ref;
 146     xmlNodePtr node;
 147     xmlNodePtr current;
 148     struct xslt_info xinfo;
 149 } xmlnodelist;
 150
 151 static inline xmlnodelist *impl_from_IXMLDOMNodeList( IXMLDOMNodeList *iface )
 152 {
 153     return (xmlnodelist *)((char*)iface - FIELD_OFFSET(xmlnodelist, lpVtbl));
 154 }
 155
 156 static HRESULT WINAPI xmlnodelist_QueryInterface(
 157     IXMLDOMNodeList *iface,
 158     REFIID riid,
 159     void** ppvObject )
 160 {
 161     TRACE("%p %s %p\n", iface, debugstr_guid(riid), ppvObject);
 162
 163     if ( IsEqualGUID( riid, &IID_IUnknown ) ||
 164          IsEqualGUID( riid, &IID_IDispatch ) ||
 165          IsEqualGUID( riid, &IID_IXMLDOMNodeList ) )
 166     {
 167         *ppvObject = iface;
 168     }
 169     else
 170         return E_NOINTERFACE;
 171
 172     IXMLDOMNodeList_AddRef( iface );
 173
 174     return S_OK;
 175 }
 176
 177 static ULONG WINAPI xmlnodelist_AddRef(
 178     IXMLDOMNodeList *iface )
 179 {
 180     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 181     return InterlockedIncrement( &This->ref );
 182 }
 183
 184 static ULONG WINAPI xmlnodelist_Release(
 185     IXMLDOMNodeList *iface )
 186 {
 187     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 188     ULONG ref;
 189
 190     ref = InterlockedDecrement( &This->ref );
 191     if ( ref == 0 )
 192     {
 193         free_xslt_info( &This->xinfo );
 194         xmldoc_release( This->node->doc );
 195         HeapFree( GetProcessHeap(), 0, This );
 196     }
 197
 198     return ref;
 199 }
 200
 201 static HRESULT WINAPI xmlnodelist_GetTypeInfoCount(
 202     IXMLDOMNodeList *iface,
 203     UINT* pctinfo )
 204 {
 205     FIXME("\n");
 206     return E_NOTIMPL;
 207 }
 208
 209 static HRESULT WINAPI xmlnodelist_GetTypeInfo(
 210     IXMLDOMNodeList *iface,
 211     UINT iTInfo,
 212     LCID lcid,
 213     ITypeInfo** ppTInfo )
 214 {
 215     FIXME("\n");
 216     return E_NOTIMPL;
 217 }
 218
 219 static HRESULT WINAPI xmlnodelist_GetIDsOfNames(
 220     IXMLDOMNodeList *iface,
 221     REFIID riid,
 222     LPOLESTR* rgszNames,
 223     UINT cNames,
 224     LCID lcid,
 225     DISPID* rgDispId )
 226 {
 227     FIXME("\n");
 228     return E_NOTIMPL;
 229 }
 230
 231 static HRESULT WINAPI xmlnodelist_Invoke(
 232     IXMLDOMNodeList *iface,
 233     DISPID dispIdMember,
 234     REFIID riid,
 235     LCID lcid,
 236     WORD wFlags,
 237     DISPPARAMS* pDispParams,
 238     VARIANT* pVarResult,
 239     EXCEPINFO* pExcepInfo,
 240     UINT* puArgErr )
 241 {
 242     FIXME("\n");
 243     return E_NOTIMPL;
 244 }
 245
 246 static HRESULT WINAPI xmlnodelist_get_item(
 247         IXMLDOMNodeList* iface,
 248         long index,
 249         IXMLDOMNode** listItem)
 250 {
 251     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 252     xmlNodePtr curr;
 253     long nodeIndex = 0;
 254     HRESULT r;
 255
 256     TRACE("%p %ld\n", This, index);
 257
 258     *listItem = NULL;
 259
 260     if (index < 0)
 261         return S_FALSE;
 262
 263     curr = This->node;
 264
 265     while(curr)
 266     {
 267         r = xslt_next_match( &This->xinfo, &curr );
 268         if(FAILED(r) || !curr) return S_FALSE;
 269         if(nodeIndex++ == index) break;
 270         curr = curr->next;
 271     }
 272     if(!curr) return S_FALSE;
 273
 274     *listItem = create_node( curr );
 275
 276     return S_OK;
 277 }
 278
 279 static HRESULT WINAPI xmlnodelist_get_length(
 280         IXMLDOMNodeList* iface,
 281         long* listLength)
 282 {
 283
 284     xmlNodePtr curr;
 285     long nodeCount = 0;
 286     HRESULT r;
 287
 288     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 289
 290     TRACE("%p\n", This);
 291
 292     if (This->node == NULL) {
 293         *listLength = 0;
 294         return S_OK;
 295     }
 296
 297     for(curr = This->node; curr; curr = curr->next)
 298     {
 299         r = xslt_next_match( &This->xinfo, &curr );
 300         if(FAILED(r) || !curr) break;
 301         nodeCount++;
 302     }
 303
 304     *listLength = nodeCount;
 305     return S_OK;
 306 }
 307
 308 static HRESULT WINAPI xmlnodelist_nextNode(
 309         IXMLDOMNodeList* iface,
 310         IXMLDOMNode** nextItem)
 311 {
 312     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 313     HRESULT r;
 314
 315     TRACE("%p %p\n", This, nextItem );
 316
 317     r = xslt_next_match( &This->xinfo, &This->current );
 318     if (FAILED(r) )
 319         return r;
 320
 321     if (!This->current)
 322         return S_FALSE;
 323
 324     *nextItem = create_node( This->current );
 325     This->current = This->current->next;
 326     return S_OK;
 327 }
 328
 329 static HRESULT WINAPI xmlnodelist_reset(
 330         IXMLDOMNodeList* iface)
 331 {
 332     xmlnodelist *This = impl_from_IXMLDOMNodeList( iface );
 333
 334     TRACE("%p\n", This);
 335     This->current = This->node;
 336     return S_OK;
 337 }
 338
 339 static HRESULT WINAPI xmlnodelist__newEnum(
 340         IXMLDOMNodeList* iface,
 341         IUnknown** ppUnk)
 342 {
 343     FIXME("\n");
 344     return E_NOTIMPL;
 345 }
 346
 347
 348 static const struct IXMLDOMNodeListVtbl xmlnodelist_vtbl =
 349 {
 350     xmlnodelist_QueryInterface,
 351     xmlnodelist_AddRef,
 352     xmlnodelist_Release,
 353     xmlnodelist_GetTypeInfoCount,
 354     xmlnodelist_GetTypeInfo,
 355     xmlnodelist_GetIDsOfNames,
 356     xmlnodelist_Invoke,
 357     xmlnodelist_get_item,
 358     xmlnodelist_get_length,
 359     xmlnodelist_nextNode,
 360     xmlnodelist_reset,
 361     xmlnodelist__newEnum,
 362 };
 363
 364 static xmlnodelist *new_nodelist( xmlNodePtr node )
 365 {
 366     xmlnodelist *nodelist;
 367
 368     nodelist = HeapAlloc( GetProcessHeap(), 0, sizeof *nodelist );
 369     if ( !nodelist )
 370         return NULL;
 371
 372     nodelist->lpVtbl = &xmlnodelist_vtbl;
 373     nodelist->ref = 1;
 374     nodelist->node = node;
 375     nodelist->current = node;
 376     xlst_info_init( &nodelist->xinfo );
 377
 378     xmldoc_add_ref( node->doc );
 379
 380     return nodelist;
 381 }
 382
 383 IXMLDOMNodeList* create_nodelist( xmlNodePtr node )
 384 {
 385     xmlnodelist *nodelist = new_nodelist( node );
 386     if (!node)
 387         return NULL;
 388     return (IXMLDOMNodeList*) &nodelist->lpVtbl;
 389 }
 390
 391 IXMLDOMNodeList* create_filtered_nodelist( xmlNodePtr node, const xmlChar *str )
 392 {
 393     xmlnodelist *This = new_nodelist( node );
 394
 395     if (create_xslt_parser( &This->xinfo, node, str ))
 396         return (IXMLDOMNodeList*) &This->lpVtbl;
 397
 398     IXMLDOMNodeList_Release( (IXMLDOMNodeList*) &This->lpVtbl );
 399     return NULL;
 400 }
 401
 402 #endif