Initial commit

1 files changed, 632 insertions, 0 deletions

diff --git a/src/dutil/rssutil.cpp b/src/dutil/rssutil.cpp
new file mode 100644
index 00000000..db49d954
--- /dev/null
+++ b/src/dutil/rssutil.cpp
@@ -0,0 +1,632 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+#include "precomp.h"
+
+static HRESULT ParseRssDocument(
+    __in IXMLDOMDocument *pixd,
+    __out RSS_CHANNEL **ppChannel
+    );
+static HRESULT ParseRssChannel(
+    __in IXMLDOMNode *pixnChannel,
+    __out RSS_CHANNEL **ppChannel
+    );
+static HRESULT ParseRssItem(
+    __in IXMLDOMNode *pixnItem,
+    __in DWORD cItem,
+    __in_xcount(pChannel->cItems) RSS_CHANNEL *pChannel
+    );
+static HRESULT ParseRssUnknownElement(
+    __in IXMLDOMNode *pNode,
+    __inout RSS_UNKNOWN_ELEMENT** ppUnknownElement
+    );
+static HRESULT ParseRssUnknownAttribute(
+    __in IXMLDOMNode *pNode,
+    __inout RSS_UNKNOWN_ATTRIBUTE** ppUnknownAttribute
+    );
+static void FreeRssUnknownElementList(
+    __in_opt RSS_UNKNOWN_ELEMENT* pUnknownElement
+    );
+static void FreeRssUnknownAttributeList(
+    __in_opt RSS_UNKNOWN_ATTRIBUTE* pUnknownAttribute
+    );
+
+
+/********************************************************************
+ RssInitialize - Initialize RSS utilities.
+
+*********************************************************************/
+extern "C" HRESULT DAPI RssInitialize()
+{
+    return XmlInitialize();
+}
+
+
+/********************************************************************
+ RssUninitialize - Uninitialize RSS utilities.
+
+*********************************************************************/
+extern "C" void DAPI RssUninitialize()
+{
+    XmlUninitialize();
+}
+
+
+/********************************************************************
+ RssParseFromString - parses out an RSS channel from a string.
+
+*********************************************************************/
+extern "C" HRESULT DAPI RssParseFromString(
+    __in_z LPCWSTR wzRssString,
+    __out RSS_CHANNEL **ppChannel
+    )
+{
+    Assert(wzRssString);
+    Assert(ppChannel);
+
+    HRESULT hr = S_OK;
+    RSS_CHANNEL *pNewChannel = NULL;
+    IXMLDOMDocument *pixdRss = NULL;
+
+    hr = XmlLoadDocument(wzRssString, &pixdRss);
+    ExitOnFailure(hr, "Failed to load RSS string as XML document.");
+
+    hr = ParseRssDocument(pixdRss, &pNewChannel);
+    ExitOnFailure(hr, "Failed to parse RSS document.");
+
+    *ppChannel = pNewChannel;
+    pNewChannel = NULL;
+
+LExit:
+    ReleaseObject(pixdRss);
+
+    ReleaseRssChannel(pNewChannel);
+
+    return hr;
+}
+
+
+/********************************************************************
+ RssParseFromFile - parses out an RSS channel from a file path.
+
+*********************************************************************/
+extern "C" HRESULT DAPI RssParseFromFile(
+    __in_z LPCWSTR wzRssFile,
+    __out RSS_CHANNEL **ppChannel
+    )
+{
+    Assert(wzRssFile);
+    Assert(ppChannel);
+
+    HRESULT hr = S_OK;
+    RSS_CHANNEL *pNewChannel = NULL;
+    IXMLDOMDocument *pixdRss = NULL;
+
+    hr = XmlLoadDocumentFromFile(wzRssFile, &pixdRss);
+    ExitOnFailure(hr, "Failed to load RSS string as XML document.");
+
+    hr = ParseRssDocument(pixdRss, &pNewChannel);
+    ExitOnFailure(hr, "Failed to parse RSS document.");
+
+    *ppChannel = pNewChannel;
+    pNewChannel = NULL;
+
+LExit:
+    ReleaseObject(pixdRss);
+
+    ReleaseRssChannel(pNewChannel);
+
+    return hr;
+}
+
+
+/********************************************************************
+ RssFreeChannel - parses out an RSS channel from a string.
+
+*********************************************************************/
+extern "C" void DAPI RssFreeChannel(
+    __in_xcount(pChannel->cItems) RSS_CHANNEL *pChannel
+    )
+{
+    if (pChannel)
+    {
+        for (DWORD i = 0; i < pChannel->cItems; ++i)
+        {
+            ReleaseStr(pChannel->rgItems[i].wzTitle);
+            ReleaseStr(pChannel->rgItems[i].wzLink);
+            ReleaseStr(pChannel->rgItems[i].wzDescription);
+            ReleaseStr(pChannel->rgItems[i].wzGuid);
+            ReleaseStr(pChannel->rgItems[i].wzEnclosureUrl);
+            ReleaseStr(pChannel->rgItems[i].wzEnclosureType);
+
+            FreeRssUnknownElementList(pChannel->rgItems[i].pUnknownElements);
+        }
+
+        ReleaseStr(pChannel->wzTitle);
+        ReleaseStr(pChannel->wzLink);
+        ReleaseStr(pChannel->wzDescription);
+        FreeRssUnknownElementList(pChannel->pUnknownElements);
+
+        MemFree(pChannel);
+    }
+}
+
+
+/********************************************************************
+ ParseRssDocument - parses out an RSS channel from a loaded XML DOM document.
+
+*********************************************************************/
+static HRESULT ParseRssDocument(
+    __in IXMLDOMDocument *pixd,
+    __out RSS_CHANNEL **ppChannel
+    )
+{
+    Assert(pixd);
+    Assert(ppChannel);
+
+    HRESULT hr = S_OK;
+    IXMLDOMElement *pRssElement = NULL;
+    IXMLDOMNodeList *pChannelNodes = NULL;
+    IXMLDOMNode *pNode = NULL;
+    BSTR bstrNodeName = NULL;
+
+    RSS_CHANNEL *pNewChannel = NULL;
+
+    //
+    // Get the document element and start processing channels.
+    //
+    hr = pixd ->get_documentElement(&pRssElement);
+    ExitOnFailure(hr, "failed get_documentElement in ParseRssDocument");
+
+    hr = pRssElement->get_childNodes(&pChannelNodes);
+    ExitOnFailure(hr, "Failed to get child nodes of Rss Document element.");
+
+    while (S_OK == (hr = XmlNextElement(pChannelNodes, &pNode, &bstrNodeName)))
+    {
+        if (0 == lstrcmpW(bstrNodeName, L"channel"))
+        {
+            hr = ParseRssChannel(pNode, &pNewChannel);
+            ExitOnFailure(hr, "Failed to parse RSS channel.");
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"link"))
+        {
+        }
+
+        ReleaseNullBSTR(bstrNodeName);
+        ReleaseNullObject(pNode);
+    }
+
+    if (S_FALSE == hr)
+    {
+        hr = S_OK;
+    }
+
+    *ppChannel = pNewChannel;
+    pNewChannel = NULL;
+
+LExit:
+    ReleaseBSTR(bstrNodeName);
+    ReleaseObject(pNode);
+    ReleaseObject(pChannelNodes);
+    ReleaseObject(pRssElement);
+
+    ReleaseRssChannel(pNewChannel);
+
+    return hr;
+}
+
+
+/********************************************************************
+ ParseRssChannel - parses out an RSS channel from a loaded XML DOM element.
+
+*********************************************************************/
+static HRESULT ParseRssChannel(
+    __in IXMLDOMNode *pixnChannel,
+    __out RSS_CHANNEL **ppChannel
+    )
+{
+    Assert(pixnChannel);
+    Assert(ppChannel);
+
+    HRESULT hr = S_OK;
+    IXMLDOMNodeList *pNodeList = NULL;
+
+    RSS_CHANNEL *pNewChannel = NULL;
+    long cItems = 0;
+
+    IXMLDOMNode *pNode = NULL;
+    BSTR bstrNodeName = NULL;
+    BSTR bstrNodeValue = NULL;
+
+    //
+    // First, calculate how many RSS items we're going to have and allocate
+    // the RSS_CHANNEL structure
+    //
+    hr = XmlSelectNodes(pixnChannel, L"item", &pNodeList);
+    ExitOnFailure(hr, "Failed to select all RSS items in an RSS channel.");
+
+    hr = pNodeList->get_length(&cItems);
+    ExitOnFailure(hr, "Failed to count the number of RSS items in RSS channel.");
+
+    pNewChannel = static_cast<RSS_CHANNEL*>(MemAlloc(sizeof(RSS_CHANNEL) + sizeof(RSS_ITEM) * cItems, TRUE));
+    ExitOnNull(pNewChannel, hr, E_OUTOFMEMORY, "Failed to allocate RSS channel structure.");
+
+    pNewChannel->cItems = cItems;
+
+    //
+    // Process the elements under a channel now.
+    //
+    hr = pixnChannel->get_childNodes(&pNodeList);
+    ExitOnFailure(hr, "Failed to get child nodes of RSS channel element.");
+
+    cItems = 0; // reset the counter and use this to walk through the channel items
+    while (S_OK == (hr = XmlNextElement(pNodeList, &pNode, &bstrNodeName)))
+    {
+        if (0 == lstrcmpW(bstrNodeName, L"title"))
+        {
+            hr = XmlGetText(pNode, &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS channel title.");
+
+            hr = StrAllocString(&pNewChannel->wzTitle, bstrNodeValue, 0);
+            ExitOnFailure(hr, "Failed to allocate RSS channel title.");
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"link"))
+        {
+            hr = XmlGetText(pNode, &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS channel link.");
+
+            hr = StrAllocString(&pNewChannel->wzLink, bstrNodeValue, 0);
+            ExitOnFailure(hr, "Failed to allocate RSS channel link.");
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"description"))
+        {
+            hr = XmlGetText(pNode, &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS channel description.");
+
+            hr = StrAllocString(&pNewChannel->wzDescription, bstrNodeValue, 0);
+            ExitOnFailure(hr, "Failed to allocate RSS channel description.");
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"ttl"))
+        {
+            hr = XmlGetText(pNode, &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS channel description.");
+
+            pNewChannel->dwTimeToLive = (DWORD)wcstoul(bstrNodeValue, NULL, 10);
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"item"))
+        {
+            hr = ParseRssItem(pNode, cItems, pNewChannel);
+            ExitOnFailure(hr, "Failed to parse RSS item.");
+
+            ++cItems;
+        }
+        else
+        {
+            hr = ParseRssUnknownElement(pNode, &pNewChannel->pUnknownElements);
+            ExitOnFailure(hr, "Failed to parse unknown RSS channel element: %ls", bstrNodeName);
+        }
+
+        ReleaseNullBSTR(bstrNodeValue);
+        ReleaseNullBSTR(bstrNodeName);
+        ReleaseNullObject(pNode);
+    }
+
+    *ppChannel = pNewChannel;
+    pNewChannel = NULL;
+
+LExit:
+    ReleaseBSTR(bstrNodeName);
+    ReleaseObject(pNode);
+    ReleaseObject(pNodeList);
+
+    ReleaseRssChannel(pNewChannel);
+
+    return hr;
+}
+
+
+/********************************************************************
+ ParseRssItem - parses out an RSS item from a loaded XML DOM node.
+
+*********************************************************************/
+static HRESULT ParseRssItem(
+    __in IXMLDOMNode *pixnItem,
+    __in DWORD cItem,
+    __in_xcount(pChannel->cItems) RSS_CHANNEL *pChannel
+    )
+{
+    HRESULT hr = S_OK;
+
+    RSS_ITEM *pItem = NULL;
+    IXMLDOMNodeList *pNodeList = NULL;
+
+    IXMLDOMNode *pNode = NULL;
+    BSTR bstrNodeName = NULL;
+    BSTR bstrNodeValue = NULL;
+
+    //
+    // First make sure we're dealing with a valid item.
+    //
+    if (pChannel->cItems <= cItem)
+    {
+        hr = E_UNEXPECTED;
+        ExitOnFailure(hr, "Unexpected number of items parsed.");
+    }
+
+    pItem = pChannel->rgItems + cItem;
+
+    //
+    // Process the elements under an item now.
+    //
+    hr = pixnItem->get_childNodes(&pNodeList);
+    ExitOnFailure(hr, "Failed to get child nodes of RSS item element.");
+    while (S_OK == (hr = XmlNextElement(pNodeList, &pNode, &bstrNodeName)))
+    {
+        if (0 == lstrcmpW(bstrNodeName, L"title"))
+        {
+            hr = XmlGetText(pNode, &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS channel title.");
+
+            hr = StrAllocString(&pItem->wzTitle, bstrNodeValue, 0);
+            ExitOnFailure(hr, "Failed to allocate RSS item title.");
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"link"))
+        {
+            hr = XmlGetText(pNode, &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS channel link.");
+
+            hr = StrAllocString(&pItem->wzLink, bstrNodeValue, 0);
+            ExitOnFailure(hr, "Failed to allocate RSS item link.");
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"description"))
+        {
+            hr = XmlGetText(pNode, &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS item description.");
+
+            hr = StrAllocString(&pItem->wzDescription, bstrNodeValue, 0);
+            ExitOnFailure(hr, "Failed to allocate RSS item description.");
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"guid"))
+        {
+            hr = XmlGetText(pNode, &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS item guid.");
+
+            hr = StrAllocString(&pItem->wzGuid, bstrNodeValue, 0);
+            ExitOnFailure(hr, "Failed to allocate RSS item guid.");
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"pubDate"))
+        {
+            hr = XmlGetText(pNode, &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS item guid.");
+
+            hr = TimeFromString(bstrNodeValue, &pItem->ftPublished);
+            ExitOnFailure(hr, "Failed to convert RSS item time.");
+        }
+        else if (0 == lstrcmpW(bstrNodeName, L"enclosure"))
+        {
+            hr = XmlGetAttribute(pNode, L"url", &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS item enclosure url.");
+
+            hr = StrAllocString(&pItem->wzEnclosureUrl, bstrNodeValue, 0);
+            ExitOnFailure(hr, "Failed to allocate RSS item enclosure url.");
+            ReleaseNullBSTR(bstrNodeValue);
+
+            hr = XmlGetAttributeNumber(pNode, L"length", &pItem->dwEnclosureSize);
+            ExitOnFailure(hr, "Failed to get RSS item enclosure length.");
+
+            hr = XmlGetAttribute(pNode, L"type", &bstrNodeValue);
+            ExitOnFailure(hr, "Failed to get RSS item enclosure type.");
+
+            hr = StrAllocString(&pItem->wzEnclosureType, bstrNodeValue, 0);
+            ExitOnFailure(hr, "Failed to allocate RSS item enclosure type.");
+        }
+        else
+        {
+            hr = ParseRssUnknownElement(pNode, &pItem->pUnknownElements);
+            ExitOnFailure(hr, "Failed to parse unknown RSS item element: %ls", bstrNodeName);
+        }
+
+        ReleaseNullBSTR(bstrNodeValue);
+        ReleaseNullBSTR(bstrNodeName);
+        ReleaseNullObject(pNode);
+    }
+
+LExit:
+    ReleaseBSTR(bstrNodeValue);
+    ReleaseBSTR(bstrNodeName);
+    ReleaseObject(pNode);
+    ReleaseObject(pNodeList);
+
+    return hr;
+}
+
+
+/********************************************************************
+ ParseRssUnknownElement - parses out an unknown item from the RSS feed from a loaded XML DOM node.
+
+*********************************************************************/
+static HRESULT ParseRssUnknownElement(
+    __in IXMLDOMNode *pNode,
+    __inout RSS_UNKNOWN_ELEMENT** ppUnknownElement
+    )
+{
+    Assert(ppUnknownElement);
+
+    HRESULT hr = S_OK;
+    BSTR bstrNodeNamespace = NULL;
+    BSTR bstrNodeName = NULL;
+    BSTR bstrNodeValue = NULL;
+    IXMLDOMNamedNodeMap* pixnnmAttributes = NULL;
+    IXMLDOMNode* pixnAttribute = NULL;
+    RSS_UNKNOWN_ELEMENT* pNewUnknownElement;
+
+    pNewUnknownElement = static_cast<RSS_UNKNOWN_ELEMENT*>(MemAlloc(sizeof(RSS_UNKNOWN_ELEMENT), TRUE));
+    ExitOnNull(pNewUnknownElement, hr, E_OUTOFMEMORY, "Failed to allocate unknown element.");
+
+    hr = pNode->get_namespaceURI(&bstrNodeNamespace);
+    if (S_OK == hr)
+    {
+        hr = StrAllocString(&pNewUnknownElement->wzNamespace, bstrNodeNamespace, 0);
+        ExitOnFailure(hr, "Failed to allocate RSS unknown element namespace.");
+    }
+    else if (S_FALSE == hr)
+    {
+        hr = S_OK;
+    }
+    ExitOnFailure(hr, "Failed to get unknown element namespace.");
+
+    hr = pNode->get_baseName(&bstrNodeName);
+    ExitOnFailure(hr, "Failed to get unknown element name.");
+
+    hr = StrAllocString(&pNewUnknownElement->wzElement, bstrNodeName, 0);
+    ExitOnFailure(hr, "Failed to allocate RSS unknown element name.");
+
+    hr = XmlGetText(pNode, &bstrNodeValue);
+    ExitOnFailure(hr, "Failed to get unknown element value.");
+
+    hr = StrAllocString(&pNewUnknownElement->wzValue, bstrNodeValue, 0);
+    ExitOnFailure(hr, "Failed to allocate RSS unknown element value.");
+
+    hr = pNode->get_attributes(&pixnnmAttributes);
+    ExitOnFailure(hr, "Failed get attributes on RSS unknown element.");
+
+    while (S_OK == (hr = pixnnmAttributes->nextNode(&pixnAttribute)))
+    {
+        hr = ParseRssUnknownAttribute(pixnAttribute, &pNewUnknownElement->pAttributes);
+        ExitOnFailure(hr, "Failed to parse attribute on RSS unknown element.");
+
+        ReleaseNullObject(pixnAttribute);
+    }
+
+    if (S_FALSE == hr)
+    {
+        hr = S_OK;
+    }
+    ExitOnFailure(hr, "Failed to enumerate all attributes on RSS unknown element.");
+
+    RSS_UNKNOWN_ELEMENT** ppTail = ppUnknownElement;
+    while (*ppTail)
+    {
+        ppTail = &(*ppTail)->pNext;
+    }
+
+    *ppTail = pNewUnknownElement;
+    pNewUnknownElement = NULL;
+
+LExit:
+    FreeRssUnknownElementList(pNewUnknownElement);
+
+    ReleaseBSTR(bstrNodeNamespace);
+    ReleaseBSTR(bstrNodeName);
+    ReleaseBSTR(bstrNodeValue);
+    ReleaseObject(pixnnmAttributes);
+    ReleaseObject(pixnAttribute);
+
+    return hr;
+}
+
+
+/********************************************************************
+ ParseRssUnknownAttribute - parses out attribute from an unknown element
+
+*********************************************************************/
+static HRESULT ParseRssUnknownAttribute(
+    __in IXMLDOMNode *pNode,
+    __inout RSS_UNKNOWN_ATTRIBUTE** ppUnknownAttribute
+    )
+{
+    Assert(ppUnknownAttribute);
+
+    HRESULT hr = S_OK;
+    BSTR bstrNodeNamespace = NULL;
+    BSTR bstrNodeName = NULL;
+    BSTR bstrNodeValue = NULL;
+    RSS_UNKNOWN_ATTRIBUTE* pNewUnknownAttribute;
+
+    pNewUnknownAttribute = static_cast<RSS_UNKNOWN_ATTRIBUTE*>(MemAlloc(sizeof(RSS_UNKNOWN_ATTRIBUTE), TRUE));
+    ExitOnNull(pNewUnknownAttribute, hr, E_OUTOFMEMORY, "Failed to allocate unknown attribute.");
+
+    hr = pNode->get_namespaceURI(&bstrNodeNamespace);
+    if (S_OK == hr)
+    {
+        hr = StrAllocString(&pNewUnknownAttribute->wzNamespace, bstrNodeNamespace, 0);
+        ExitOnFailure(hr, "Failed to allocate RSS unknown attribute namespace.");
+    }
+    else if (S_FALSE == hr)
+    {
+        hr = S_OK;
+    }
+    ExitOnFailure(hr, "Failed to get unknown attribute namespace.");
+
+    hr = pNode->get_baseName(&bstrNodeName);
+    ExitOnFailure(hr, "Failed to get unknown attribute name.");
+
+    hr = StrAllocString(&pNewUnknownAttribute->wzAttribute, bstrNodeName, 0);
+    ExitOnFailure(hr, "Failed to allocate RSS unknown attribute name.");
+
+    hr = XmlGetText(pNode, &bstrNodeValue);
+    ExitOnFailure(hr, "Failed to get unknown attribute value.");
+
+    hr = StrAllocString(&pNewUnknownAttribute->wzValue, bstrNodeValue, 0);
+    ExitOnFailure(hr, "Failed to allocate RSS unknown attribute value.");
+
+    RSS_UNKNOWN_ATTRIBUTE** ppTail = ppUnknownAttribute;
+    while (*ppTail)
+    {
+        ppTail = &(*ppTail)->pNext;
+    }
+
+    *ppTail = pNewUnknownAttribute;
+    pNewUnknownAttribute = NULL;
+
+LExit:
+    FreeRssUnknownAttributeList(pNewUnknownAttribute);
+
+    ReleaseBSTR(bstrNodeNamespace);
+    ReleaseBSTR(bstrNodeName);
+    ReleaseBSTR(bstrNodeValue);
+
+    return hr;
+}
+
+
+/********************************************************************
+ FreeRssUnknownElement - releases all of the memory used by a list of unknown elements
+
+*********************************************************************/
+static void FreeRssUnknownElementList(
+    __in_opt RSS_UNKNOWN_ELEMENT* pUnknownElement
+    )
+{
+    while (pUnknownElement)
+    {
+        RSS_UNKNOWN_ELEMENT* pFree = pUnknownElement;
+        pUnknownElement = pUnknownElement->pNext;
+
+        FreeRssUnknownAttributeList(pFree->pAttributes);
+        ReleaseStr(pFree->wzNamespace);
+        ReleaseStr(pFree->wzElement);
+        ReleaseStr(pFree->wzValue);
+        MemFree(pFree);
+    }
+}
+
+
+/********************************************************************
+ FreeRssUnknownAttribute - releases all of the memory used by a list of unknown attributes
+
+*********************************************************************/
+static void FreeRssUnknownAttributeList(
+    __in_opt RSS_UNKNOWN_ATTRIBUTE* pUnknownAttribute
+    )
+{
+    while (pUnknownAttribute)
+    {
+        RSS_UNKNOWN_ATTRIBUTE* pFree = pUnknownAttribute;
+        pUnknownAttribute = pUnknownAttribute->pNext;
+
+        ReleaseStr(pFree->wzNamespace);
+        ReleaseStr(pFree->wzAttribute);
+        ReleaseStr(pFree->wzValue);
+        MemFree(pFree);
+    }
+}