#include "config.h"
#include "runtime_root.h"
#include "JSGlobalObject.h"
#include "object.h"
#include "runtime.h"
#include "runtime_object.h"
#include <wtf/HashCountedSet.h>
#include <wtf/HashSet.h>
namespace KJS { namespace Bindings {
typedef HashSet<RootObject*> RootObjectSet;
static RootObjectSet* rootObjectSet()
{
static RootObjectSet staticRootObjectSet;
return &staticRootObjectSet;
}
RootObject* findProtectingRootObject(JSObject* jsObject)
{
RootObjectSet::const_iterator end = rootObjectSet()->end();
for (RootObjectSet::const_iterator it = rootObjectSet()->begin(); it != end; ++it) {
if ((*it)->gcIsProtected(jsObject))
return *it;
}
return 0;
}
RootObject* findRootObject(JSGlobalObject* globalObject)
{
RootObjectSet::const_iterator end = rootObjectSet()->end();
for (RootObjectSet::const_iterator it = rootObjectSet()->begin(); it != end; ++it) {
if ((*it)->globalObject() == globalObject)
return *it;
}
return 0;
}
#if PLATFORM(MAC)
static CFRunLoopSourceRef completionSource;
static void completedJavaScriptAccess (void *i)
{
assert (CFRunLoopGetCurrent() != RootObject::runLoop());
JSObjectCallContext *callContext = (JSObjectCallContext *)i;
CFRunLoopRef runLoop = (CFRunLoopRef)callContext->originatingLoop;
assert (CFRunLoopGetCurrent() == runLoop);
CFRunLoopStop(runLoop);
}
static pthread_once_t javaScriptAccessLockOnce = PTHREAD_ONCE_INIT;
static pthread_mutex_t javaScriptAccessLock;
static int javaScriptAccessLockCount = 0;
static void initializeJavaScriptAccessLock()
{
pthread_mutexattr_t attr;
pthread_mutexattr_init(&attr);
pthread_mutexattr_settype (&attr, PTHREAD_MUTEX_RECURSIVE);
pthread_mutex_init(&javaScriptAccessLock, &attr);
}
static inline void lockJavaScriptAccess()
{
pthread_once(&javaScriptAccessLockOnce, initializeJavaScriptAccessLock);
pthread_mutex_lock(&javaScriptAccessLock);
javaScriptAccessLockCount++;
}
static inline void unlockJavaScriptAccess()
{
javaScriptAccessLockCount--;
pthread_mutex_unlock(&javaScriptAccessLock);
}
void RootObject::dispatchToJavaScriptThread(JSObjectCallContext *context)
{
lockJavaScriptAccess();
CFRunLoopRef currentRunLoop = CFRunLoopGetCurrent();
assert (currentRunLoop != RootObject::runLoop());
context->originatingLoop = currentRunLoop;
CFRunLoopSourceContext sourceContext = {0, context, NULL, NULL, NULL, NULL, NULL, NULL, NULL, completedJavaScriptAccess};
completionSource = CFRunLoopSourceCreate(NULL, 0, &sourceContext);
CFRunLoopAddSource(currentRunLoop, completionSource, kCFRunLoopDefaultMode);
CFRunLoopSourceSignal(RootObject::performJavaScriptSource());
if (CFRunLoopIsWaiting(RootObject::runLoop())) {
CFRunLoopWakeUp(RootObject::runLoop());
}
CFRunLoopRun ();
CFRunLoopRemoveSource(currentRunLoop, completionSource, kCFRunLoopDefaultMode);
CFRelease (completionSource);
unlockJavaScriptAccess();
}
static void performJavaScriptAccess(void*)
{
assert (CFRunLoopGetCurrent() == RootObject::runLoop());
CFRunLoopSourceContext sourceContext;
CFRunLoopSourceGetContext (completionSource, &sourceContext);
JSObjectCallContext *callContext = (JSObjectCallContext *)sourceContext.info;
CFRunLoopRef originatingLoop = callContext->originatingLoop;
JavaJSObject::invoke (callContext);
CFRunLoopSourceSignal (completionSource);
if (CFRunLoopIsWaiting(originatingLoop)) {
CFRunLoopWakeUp(originatingLoop);
}
}
CreateRootObjectFunction RootObject::_createRootObject = 0;
CFRunLoopRef RootObject::_runLoop = 0;
CFRunLoopSourceRef RootObject::_performJavaScriptSource = 0;
void RootObject::setCreateRootObject(CreateRootObjectFunction createRootObject) {
ASSERT(!_createRootObject);
_createRootObject = createRootObject;
_runLoop = (CFRunLoopRef)CFRetain (CFRunLoopGetCurrent ());
CFRunLoopSourceContext sourceContext = {0, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, performJavaScriptAccess};
RootObject::_performJavaScriptSource = CFRunLoopSourceCreate(NULL, 0, &sourceContext);
CFRunLoopAddSource(RootObject::_runLoop, RootObject::_performJavaScriptSource, kCFRunLoopDefaultMode);
}
#endif
PassRefPtr<RootObject> RootObject::create(const void* nativeHandle, JSGlobalObject* globalObject)
{
return new RootObject(nativeHandle, globalObject);
}
RootObject::RootObject(const void* nativeHandle, JSGlobalObject* globalObject)
: m_isValid(true)
, m_nativeHandle(nativeHandle)
, m_globalObject(globalObject)
{
ASSERT(globalObject);
rootObjectSet()->add(this);
}
RootObject::~RootObject()
{
if (m_isValid)
invalidate();
}
void RootObject::invalidate()
{
if (!m_isValid)
return;
{
HashSet<RuntimeObjectImp*>::iterator end = m_runtimeObjects.end();
for (HashSet<RuntimeObjectImp*>::iterator it = m_runtimeObjects.begin(); it != end; ++it)
(*it)->invalidate();
m_runtimeObjects.clear();
}
m_isValid = false;
m_nativeHandle = 0;
m_globalObject = 0;
ProtectCountSet::iterator end = m_protectCountSet.end();
for (ProtectCountSet::iterator it = m_protectCountSet.begin(); it != end; ++it) {
JSLock lock;
KJS::gcUnprotect(it->first);
}
m_protectCountSet.clear();
rootObjectSet()->remove(this);
}
void RootObject::gcProtect(JSObject* jsObject)
{
ASSERT(m_isValid);
if (!m_protectCountSet.contains(jsObject)) {
JSLock lock;
KJS::gcProtect(jsObject);
}
m_protectCountSet.add(jsObject);
}
void RootObject::gcUnprotect(JSObject* jsObject)
{
ASSERT(m_isValid);
if (!jsObject)
return;
if (m_protectCountSet.count(jsObject) == 1) {
JSLock lock;
KJS::gcUnprotect(jsObject);
}
m_protectCountSet.remove(jsObject);
}
bool RootObject::gcIsProtected(JSObject* jsObject)
{
ASSERT(m_isValid);
return m_protectCountSet.contains(jsObject);
}
const void* RootObject::nativeHandle() const
{
ASSERT(m_isValid);
return m_nativeHandle;
}
JSGlobalObject* RootObject::globalObject() const
{
ASSERT(m_isValid);
return m_globalObject;
}
void RootObject::addRuntimeObject(RuntimeObjectImp* object)
{
ASSERT(m_isValid);
ASSERT(!m_runtimeObjects.contains(object));
m_runtimeObjects.add(object);
}
void RootObject::removeRuntimeObject(RuntimeObjectImp* object)
{
ASSERT(m_isValid);
ASSERT(m_runtimeObjects.contains(object));
m_runtimeObjects.remove(object);
}
} }