StringRecursionChecker.h 2.42 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
/*
 *  Copyright (C) 2011 Apple Inc. All rights reserved.
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2 of the License, or (at your option) any later version.
 *
 *  This library is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 */

#ifndef StringRecursionChecker_h
#define StringRecursionChecker_h

#include "Interpreter.h"
24
#include <wtf/StackStats.h>
25
#include <wtf/WTFThreadData.h>
26 27 28 29 30 31 32 33 34 35

namespace JSC {

class StringRecursionChecker {
    WTF_MAKE_NONCOPYABLE(StringRecursionChecker);

public:
    StringRecursionChecker(ExecState*, JSObject* thisObject);
    ~StringRecursionChecker();

36
    JSValue earlyReturnValue() const; // 0 if everything is OK, value to return for failure cases
37 38

private:
39 40 41
    JSValue throwStackOverflowError();
    JSValue emptyString();
    JSValue performCheck();
42 43 44

    ExecState* m_exec;
    JSObject* m_thisObject;
45
    JSValue m_earlyReturnValue;
46 47

    StackStats::CheckPoint stackCheckpoint;
48 49
};

50
inline JSValue StringRecursionChecker::performCheck()
51
{
52
    VM& vm = m_exec->vm();
53
    if (!vm.isSafeToRecurse())
54
        return throwStackOverflowError();
55
    bool alreadyVisited = !vm.stringRecursionCheckVisitedObjects.add(m_thisObject).isNewEntry;
56 57
    if (alreadyVisited)
        return emptyString(); // Return empty string to avoid infinite recursion.
58
    return JSValue(); // Indicate success.
59 60 61 62 63 64 65 66 67
}

inline StringRecursionChecker::StringRecursionChecker(ExecState* exec, JSObject* thisObject)
    : m_exec(exec)
    , m_thisObject(thisObject)
    , m_earlyReturnValue(performCheck())
{
}

68
inline JSValue StringRecursionChecker::earlyReturnValue() const
69 70 71 72 73 74 75 76
{
    return m_earlyReturnValue;
}

inline StringRecursionChecker::~StringRecursionChecker()
{
    if (m_earlyReturnValue)
        return;
77 78
    ASSERT(m_exec->vm().stringRecursionCheckVisitedObjects.contains(m_thisObject));
    m_exec->vm().stringRecursionCheckVisitedObjects.remove(m_thisObject);
79 80 81 82 83
}

}

#endif