2011-01-27 Oliver Hunt <oliver@apple.com>
[WebKit-https.git] / Source / WebCore / bindings / js / JSNodeFilterCondition.cpp
1 /*
2  *  Copyright (C) 2001 Peter Kelly (pmk@post.com)
3  *  Copyright (C) 2007, 2008, 2009 Apple Inc. All rights reserved.
4  *
5  *  This library is free software; you can redistribute it and/or
6  *  modify it under the terms of the GNU Lesser General Public
7  *  License as published by the Free Software Foundation; either
8  *  version 2 of the License, or (at your option) any later version.
9  *
10  *  This library is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  *  Lesser General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Lesser General Public
16  *  License along with this library; if not, write to the Free Software
17  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
18  */
19
20 #include "config.h"
21 #include "JSNodeFilterCondition.h"
22
23 #include "JSNode.h"
24 #include "JSNodeFilter.h"
25 #include "NodeFilter.h"
26 #include <runtime/Error.h>
27 #include <runtime/JSLock.h>
28
29 namespace WebCore {
30
31 using namespace JSC;
32
33 ASSERT_CLASS_FITS_IN_CELL(JSNodeFilterCondition);
34
35 JSNodeFilterCondition::JSNodeFilterCondition(JSValue filter)
36     : m_filter(filter)
37 {
38 }
39
40 void JSNodeFilterCondition::markAggregate(MarkStack& markStack)
41 {
42     markStack.append(&m_filter);
43 }
44
45 short JSNodeFilterCondition::acceptNode(JSC::ExecState* exec, Node* filterNode) const
46 {
47     JSLock lock(SilenceAssertionsOnly);
48
49     if (!m_filter->isObject())
50         return NodeFilter::FILTER_ACCEPT;
51
52    // The exec argument here should only be null if this was called from a
53    // non-JavaScript language, and this is a JavaScript filter, and the document
54    // in question is not associated with the frame. In that case, we're going to
55    // behave incorrectly, and just reject nodes instead of calling the filter function.
56    // To fix that we'd need to come up with a way to find a suitable JavaScript
57    // execution context for the filter function to run in.
58     if (!exec)
59         return NodeFilter::FILTER_REJECT;
60
61     JSValue function = m_filter.get();
62     CallData callData;
63     CallType callType = getCallData(function, callData);
64     if (callType == CallTypeNone) {
65         function = m_filter->get(exec, Identifier(exec, "acceptNode"));
66         callType = getCallData(function, callData);
67         if (callType == CallTypeNone) {
68             throwError(exec, createTypeError(exec, "NodeFilter object does not have an acceptNode function"));
69             return NodeFilter::FILTER_REJECT;
70         }
71     }
72
73     MarkedArgumentBuffer args;
74     // FIXME: The node should have the prototype chain that came from its document, not
75     // whatever prototype chain might be on the window this filter came from. Bug 27662
76     args.append(toJS(exec, deprecatedGlobalObjectForPrototype(exec), filterNode));
77     if (exec->hadException())
78         return NodeFilter::FILTER_REJECT;
79
80     JSValue result = JSC::call(exec, function, callType, callData, m_filter.get(), args);
81     if (exec->hadException())
82         return NodeFilter::FILTER_REJECT;
83
84     int intResult = result.toInt32(exec);
85     if (exec->hadException())
86         return NodeFilter::FILTER_REJECT;
87
88     return intResult;
89 }
90
91 } // namespace WebCore