8d53f039ed7ae4610c3b50e67bddf4e28bbfa602
[WebKit-https.git] / Source / WebCore / xml / XPathExpression.cpp
1 /*
2  * Copyright (C) 2005 Frerich Raabe <raabe@kde.org>
3  * Copyright (C) 2006, 2009 Apple Inc. All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  * 
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26
27 #include "config.h"
28 #include "XPathExpression.h"
29
30 #include "Document.h"
31 #include "ExceptionCode.h"
32 #include "XPathException.h"
33 #include "XPathNSResolver.h"
34 #include "XPathParser.h"
35 #include "XPathResult.h"
36 #include "XPathUtil.h"
37 #include <wtf/text/WTFString.h>
38
39 namespace WebCore {
40
41 using namespace XPath;
42     
43 inline XPathExpression::XPathExpression(std::unique_ptr<XPath::Expression> expression)
44     : m_topExpression(WTFMove(expression))
45 {
46 }
47
48 RefPtr<XPathExpression> XPathExpression::createExpression(const String& expression, RefPtr<XPathNSResolver>&& resolver, ExceptionCode& ec)
49 {
50     auto parsedExpression = Parser::parseStatement(expression, WTFMove(resolver), ec);
51     if (!parsedExpression)
52         return nullptr;
53
54     return adoptRef(*new XPathExpression(WTFMove(parsedExpression)));
55 }
56
57 XPathExpression::~XPathExpression()
58 {
59 }
60
61 RefPtr<XPathResult> XPathExpression::evaluate(Node* contextNode, unsigned short type, XPathResult*, ExceptionCode& ec)
62 {
63     if (!isValidContextNode(contextNode)) {
64         ec = NOT_SUPPORTED_ERR;
65         return nullptr;
66     }
67
68     EvaluationContext& evaluationContext = Expression::evaluationContext();
69     evaluationContext.node = contextNode;
70     evaluationContext.size = 1;
71     evaluationContext.position = 1;
72     evaluationContext.hadTypeConversionError = false;
73     RefPtr<XPathResult> result = XPathResult::create(&contextNode->document(), m_topExpression->evaluate());
74     evaluationContext.node = nullptr; // Do not hold a reference to the context node, as this may prevent the whole document from being destroyed in time.
75
76     if (evaluationContext.hadTypeConversionError) {
77         // It is not specified what to do if type conversion fails while evaluating an expression, and INVALID_EXPRESSION_ERR is not exactly right
78         // when the failure happens in an otherwise valid expression because of a variable. But XPathEvaluator does not support variables, so it's close enough.
79         ec = XPathException::INVALID_EXPRESSION_ERR;
80         return nullptr;
81     }
82
83     if (type != XPathResult::ANY_TYPE) {
84         ec = 0;
85         result->convertTo(type, ec);
86         if (ec)
87             return nullptr;
88     }
89
90     return result;
91 }
92
93 }