Bug 813867 - 'Report memory for web workers that use ctypes'. r=jlebar+jorendorff.

--HG--
extra : rebase_source : 8ecb268167e840f26dd53a774c7e978b15ae3017
This commit is contained in:
Ben Turner 2012-12-30 10:21:52 -08:00
Родитель ec25adc6cd
Коммит 3475845a40
9 изменённых файлов: 371 добавлений и 262 удалений

Просмотреть файл

@ -13,14 +13,13 @@
#include "WorkerPrivate.h"
#define CTYPES_STR "ctypes"
using namespace mozilla::dom;
USING_WORKERS_NAMESPACE
namespace {
#ifdef BUILD_CTYPES
char*
UnicodeToNative(JSContext* aCx, const jschar* aSource, size_t aSourceLen)
{
@ -42,60 +41,7 @@ UnicodeToNative(JSContext* aCx, const jschar* aSource, size_t aSourceLen)
return result;
}
JSCTypesCallbacks gCTypesCallbacks = {
UnicodeToNative
};
JSBool
CTypesLazyGetter(JSContext* aCx, JSHandleObject aObj, JSHandleId aId, JSMutableHandleValue aVp)
{
NS_ASSERTION(JS_GetGlobalObject(aCx) == aObj, "Not a global object!");
NS_ASSERTION(JSID_IS_STRING(aId), "Bad id!");
NS_ASSERTION(JS_FlatStringEqualsAscii(JSID_TO_FLAT_STRING(aId), CTYPES_STR),
"Bad id!");
WorkerPrivate* worker = GetWorkerPrivateFromContext(aCx);
NS_ASSERTION(worker->IsChromeWorker(), "This should always be true!");
if (!worker->DisableMemoryReporter()) {
return false;
}
jsval ctypes;
if (!JS_DeletePropertyById(aCx, aObj, aId) ||
!JS_InitCTypesClass(aCx, aObj) ||
!JS_GetPropertyById(aCx, aObj, aId, &ctypes)) {
return false;
}
JS_SetCTypesCallbacks(JSVAL_TO_OBJECT(ctypes), &gCTypesCallbacks);
return JS_GetPropertyById(aCx, aObj, aId, aVp.address());
}
#endif
inline bool
DefineCTypesLazyGetter(JSContext* aCx, JSObject* aGlobal)
{
#ifdef BUILD_CTYPES
{
JSString* ctypesStr = JS_InternString(aCx, CTYPES_STR);
if (!ctypesStr) {
return false;
}
jsid ctypesId = INTERNED_STRING_TO_JSID(aCx, ctypesStr);
// We use a lazy getter here to let us unregister the blocking memory
// reporter since ctypes can easily block the worker thread and we can
// deadlock. Remove once bug 673323 is fixed.
if (!JS_DefinePropertyById(aCx, aGlobal, ctypesId, JSVAL_VOID,
CTypesLazyGetter, NULL, 0)) {
return false;
}
}
#endif
return true;
}
#endif // BUILD_CTYPES
} // anonymous namespace
@ -105,7 +51,23 @@ bool
DefineChromeWorkerFunctions(JSContext* aCx, JSObject* aGlobal)
{
// Currently ctypes is the only special property given to ChromeWorkers.
return DefineCTypesLazyGetter(aCx, aGlobal);
#ifdef BUILD_CTYPES
{
jsval ctypes;
if (!JS_InitCTypesClass(aCx, aGlobal) ||
!JS_GetProperty(aCx, aGlobal, "ctypes", &ctypes)) {
return false;
}
static JSCTypesCallbacks callbacks = {
UnicodeToNative
};
JS_SetCTypesCallbacks(JSVAL_TO_OBJECT(ctypes), &callbacks);
}
#endif // BUILD_CTYPES
return true;
}
END_WORKERS_NAMESPACE

Просмотреть файл

@ -22,6 +22,7 @@
#include "nsLayoutStatics.h"
#include "jsdbgapi.h"
#include "jsfriendapi.h"
#include "mozilla/dom/EventTargetBinding.h"
#include "mozilla/Preferences.h"
#include "nsContentUtils.h"
@ -372,6 +373,27 @@ ContentSecurityPolicyAllows(JSContext* aCx)
return false;
}
void
CTypesActivityCallback(JSContext* aCx,
js::CTypesActivityType aType)
{
WorkerPrivate* worker = GetWorkerPrivateFromContext(aCx);
worker->AssertIsOnWorkerThread();
switch (aType) {
case js::CTYPES_CALL_BEGIN:
worker->BeginCTypesCall();
break;
case js::CTYPES_CALL_END:
worker->EndCTypesCall();
break;
default:
MOZ_NOT_REACHED("Unknown type flag!");
}
}
JSContext*
CreateJSContextForWorker(WorkerPrivate* aWorkerPrivate)
{
@ -418,6 +440,8 @@ CreateJSContextForWorker(WorkerPrivate* aWorkerPrivate)
JS_SetOperationCallback(workerCx, OperationCallback);
js::SetCTypesActivityCallback(runtime, CTypesActivityCallback);
NS_ASSERTION((aWorkerPrivate->GetJSContextOptions() &
kRequiredJSContextOptions) == kRequiredJSContextOptions,
"Somehow we lost our required options!");

Просмотреть файл

@ -1398,134 +1398,59 @@ public:
}
};
class CollectRuntimeStatsRunnable : public WorkerControlRunnable
class WorkerJSRuntimeStats : public JS::RuntimeStats
{
typedef mozilla::Mutex Mutex;
typedef mozilla::CondVar CondVar;
Mutex mMutex;
CondVar mCondVar;
volatile bool mDone;
bool mIsQuick;
void* mData;
bool* mSucceeded;
const nsACString& mRtPath;
public:
CollectRuntimeStatsRunnable(WorkerPrivate* aWorkerPrivate, bool aIsQuick,
void* aData, bool* aSucceeded)
: WorkerControlRunnable(aWorkerPrivate, WorkerThread, UnchangedBusyCount),
mMutex("CollectRuntimeStatsRunnable::mMutex"),
mCondVar(mMutex, "CollectRuntimeStatsRunnable::mCondVar"), mDone(false),
mIsQuick(aIsQuick), mData(aData), mSucceeded(aSucceeded)
WorkerJSRuntimeStats(const nsACString& aRtPath)
: JS::RuntimeStats(JsWorkerMallocSizeOf), mRtPath(aRtPath)
{ }
bool
PreDispatch(JSContext* aCx, WorkerPrivate* aWorkerPrivate)
~WorkerJSRuntimeStats()
{
AssertIsOnMainThread();
return true;
}
void
PostDispatch(JSContext* aCx, WorkerPrivate* aWorkerPrivate,
bool aDispatchResult)
{
AssertIsOnMainThread();
}
bool
DispatchInternal()
{
AssertIsOnMainThread();
if (!WorkerControlRunnable::DispatchInternal()) {
NS_WARNING("Failed to dispatch runnable!");
return false;
}
{
MutexAutoLock lock(mMutex);
while (!mDone) {
mCondVar.Wait();
}
}
return true;
}
bool
WorkerRun(JSContext* aCx, WorkerPrivate* aWorkerPrivate)
{
JSRuntime *rt = JS_GetRuntime(aCx);
if (mIsQuick) {
*static_cast<int64_t*>(mData) = JS::GetExplicitNonHeapForRuntime(rt, JsWorkerMallocSizeOf);
*mSucceeded = true;
} else {
*mSucceeded = JS::CollectRuntimeStats(rt, static_cast<JS::RuntimeStats*>(mData), nullptr);
}
{
MutexAutoLock lock(mMutex);
mDone = true;
mCondVar.Notify();
}
return true;
}
};
} /* anonymous namespace */
struct WorkerJSRuntimeStats : public JS::RuntimeStats
{
WorkerJSRuntimeStats(nsACString &aRtPath)
: JS::RuntimeStats(JsWorkerMallocSizeOf), mRtPath(aRtPath) { }
~WorkerJSRuntimeStats() {
for (size_t i = 0; i != compartmentStatsVector.length(); i++) {
free(compartmentStatsVector[i].extra1);
// no need to free |extra2|, because it's a static string
// No need to free |extra2| because it's a static string.
}
}
virtual void initExtraCompartmentStats(JSCompartment *c,
JS::CompartmentStats *cstats) MOZ_OVERRIDE
virtual void
initExtraCompartmentStats(JSCompartment* aCompartment,
JS::CompartmentStats* aCompartmentStats)
MOZ_OVERRIDE
{
MOZ_ASSERT(!cstats->extra1);
MOZ_ASSERT(!cstats->extra2);
MOZ_ASSERT(!aCompartmentStats->extra1);
MOZ_ASSERT(!aCompartmentStats->extra2);
// ReportJSRuntimeExplicitTreeStats expects that cstats->{extra1,extra2}
// are char pointers.
// ReportJSRuntimeExplicitTreeStats expects that
// aCompartmentStats->{extra1,extra2} are char pointers.
// This is the |cJSPathPrefix|. Each worker has exactly two compartments:
// one for atoms, and one for everything else.
nsCString cJSPathPrefix(mRtPath);
cJSPathPrefix += js::IsAtomsCompartment(c)
nsAutoCString cJSPathPrefix(mRtPath);
cJSPathPrefix += js::IsAtomsCompartment(aCompartment)
? NS_LITERAL_CSTRING("compartment(web-worker-atoms)/")
: NS_LITERAL_CSTRING("compartment(web-worker)/");
cstats->extra1 = strdup(cJSPathPrefix.get());
aCompartmentStats->extra1 = strdup(cJSPathPrefix.get());
// This is the |cDOMPathPrefix|, which should never be used when reporting
// with workers (hence the "?!").
cstats->extra2 = (void *)"explicit/workers/?!/";
// This should never be used when reporting with workers (hence the "?!").
static const char bogusMemoryReporterPath[] = "explicit/workers/?!/";
aCompartmentStats->extra2 = const_cast<char*>(bogusMemoryReporterPath);
}
private:
nsCString mRtPath;
};
BEGIN_WORKERS_NAMESPACE
class WorkerMemoryReporter MOZ_FINAL : public nsIMemoryMultiReporter
class MemoryReporter MOZ_FINAL : public nsIMemoryMultiReporter
{
friend class WorkerPrivate;
WorkerPrivate* mWorkerPrivate;
nsCString mAddressString;
nsCString mRtPath;
public:
NS_DECL_ISUPPORTS
WorkerMemoryReporter(WorkerPrivate* aWorkerPrivate)
MemoryReporter(WorkerPrivate* aWorkerPrivate)
: mWorkerPrivate(aWorkerPrivate)
{
aWorkerPrivate->AssertIsOnWorkerThread();
@ -1536,51 +1461,22 @@ public:
NS_ConvertUTF16toUTF8 escapedURL(aWorkerPrivate->ScriptURL());
escapedURL.ReplaceChar('/', '\\');
{
// 64bit address plus '0x' plus null terminator.
char address[21];
uint32_t addressSize =
JS_snprintf(address, sizeof(address), "%p", aWorkerPrivate);
if (addressSize != uint32_t(-1)) {
mAddressString.Assign(address, addressSize);
}
else {
NS_WARNING("JS_snprintf failed!");
mAddressString.AssignLiteral("<unknown address>");
}
}
nsAutoCString addressString;
addressString.AppendPrintf("0x%p", static_cast<void*>(aWorkerPrivate));
mRtPath = NS_LITERAL_CSTRING("explicit/workers/workers(") +
escapedDomain + NS_LITERAL_CSTRING(")/worker(") +
escapedURL + NS_LITERAL_CSTRING(", ") + mAddressString +
escapedURL + NS_LITERAL_CSTRING(", ") + addressString +
NS_LITERAL_CSTRING(")/");
}
nsresult
CollectForRuntime(bool aIsQuick, void* aData)
~MemoryReporter()
{
AssertIsOnMainThread();
if (!mWorkerPrivate) {
#ifdef DEBUG
nsAutoCString message("Unable to report memory for ");
message += NS_LITERAL_CSTRING("Worker (") + mAddressString +
NS_LITERAL_CSTRING(")! It is either using ctypes or is in "
"the process of being destroyed");
NS_WARNING(message.get());
#endif
*static_cast<int64_t*>(aData) = 0;
return NS_OK;
}
if (!mWorkerPrivate->BlockAndCollectRuntimeStats(aIsQuick, aData)) {
return NS_ERROR_FAILURE;
}
return NS_OK;
mWorkerPrivate->NoteDeadMemoryReporter();
}
NS_IMETHOD GetName(nsACString &aName)
NS_IMETHOD
GetName(nsACString& aName)
{
aName.AssignLiteral("workers");
return NS_OK;
@ -1593,9 +1489,9 @@ public:
AssertIsOnMainThread();
WorkerJSRuntimeStats rtStats(mRtPath);
nsresult rv = CollectForRuntime(/* isQuick = */false, &rtStats);
if (NS_FAILED(rv)) {
return rv;
if (!mWorkerPrivate->BlockAndCollectRuntimeStats(/* aIsQuick = */ false,
&rtStats)) {
return NS_ERROR_FAILURE;
}
// Always report, even if we're disabled, so that we at least get an entry
@ -1605,33 +1501,22 @@ public:
}
NS_IMETHOD
GetExplicitNonHeap(int64_t *aAmount)
GetExplicitNonHeap(int64_t* aAmount)
{
AssertIsOnMainThread();
return CollectForRuntime(/* isQuick = */true, aAmount);
}
void Disable()
{
#ifdef DEBUG
// Setting mWorkerPrivate to nullptr is safe only because we've locked the
// worker's mutex on the worker's thread, in the caller. So we check that.
//
// Also, we may have already disabled the reporter (and thus set
// mWorkerPrivate to nullptr) due to the use of CTypes (see
// ChromeWorkerScope.cpp). That's why the NULL check is necessary.
if (mWorkerPrivate) {
mWorkerPrivate->mMutex.AssertCurrentThreadOwns();
if (!mWorkerPrivate->BlockAndCollectRuntimeStats(/* aIsQuick = */ true,
aAmount)) {
return NS_ERROR_FAILURE;
}
#endif
mWorkerPrivate = nullptr;
return NS_OK;
}
};
END_WORKERS_NAMESPACE
} /* anonymous namespace */
NS_IMPL_THREADSAFE_ISUPPORTS1(WorkerMemoryReporter, nsIMemoryMultiReporter)
NS_IMPL_THREADSAFE_ISUPPORTS1(MemoryReporter, nsIMemoryMultiReporter)
#ifdef DEBUG
void
@ -1909,6 +1794,7 @@ WorkerPrivateParent<Derived>::WorkerPrivateParent(
bool aEvalAllowed)
: EventTarget(aParent ? aCx : NULL), mMutex("WorkerPrivateParent Mutex"),
mCondVar(mMutex, "WorkerPrivateParent CondVar"),
mMemoryReportCondVar(mMutex, "WorkerPrivateParent Memory Report CondVar"),
mJSObject(aObject), mParent(aParent), mParentJSContext(aParentJSContext),
mScriptURL(aScriptURL), mDomain(aDomain), mBusyCount(0),
mParentStatus(Pending), mJSContextOptions(0), mJSRuntimeHeapSize(0),
@ -2480,7 +2366,8 @@ WorkerPrivate::WorkerPrivate(JSContext* aCx, JSObject* aObject,
mJSContext(nullptr), mErrorHandlerRecursionCount(0), mNextTimeoutId(1),
mStatus(Pending), mSuspended(false), mTimerRunning(false),
mRunningExpiredTimeouts(false), mCloseHandlerStarted(false),
mCloseHandlerFinished(false), mMemoryReporterRunning(false),
mCloseHandlerFinished(false), mMemoryReporterAlive(false),
mMemoryReporterRunning(false), mBlockedForMemoryReporter(false),
mXHRParamsAllowed(aXHRParamsAllowed)
{
MOZ_COUNT_CTOR(mozilla::dom::workers::WorkerPrivate);
@ -2737,12 +2624,7 @@ WorkerPrivate::DoRunLoop(JSContext* aCx)
idleGCEvent = runnable;
}
mMemoryReporter = new WorkerMemoryReporter(this);
if (NS_FAILED(NS_RegisterMemoryMultiReporter(mMemoryReporter))) {
NS_WARNING("Failed to register memory reporter!");
mMemoryReporter = nullptr;
}
EnableMemoryReporter();
for (;;) {
Status currentStatus;
@ -2753,7 +2635,7 @@ WorkerPrivate::DoRunLoop(JSContext* aCx)
MutexAutoLock lock(mMutex);
while (!mControlQueue.Pop(event) && !mQueue.Pop(event)) {
mCondVar.Wait();
WaitForWorkerEvents();
}
bool eventIsNotIdleGCEvent;
@ -2854,17 +2736,8 @@ WorkerPrivate::DoRunLoop(JSContext* aCx)
NS_WARNING("Failed to cancel the GC timer!");
}
// Call this before unregistering the reporter as we may be racing with
// the main thread.
DisableMemoryReporter();
if (mMemoryReporter) {
if (NS_FAILED(NS_UnregisterMemoryMultiReporter(mMemoryReporter))) {
NS_WARNING("Failed to unregister memory reporter!");
}
mMemoryReporter = nullptr;
}
StopAcceptingEvents();
return;
}
@ -2898,7 +2771,7 @@ WorkerPrivate::OperationCallback(JSContext* aCx)
break;
}
mCondVar.Wait(PR_MillisecondsToInterval(RemainingRunTimeMS()));
WaitForWorkerEvents(PR_MillisecondsToInterval(RemainingRunTimeMS()));
}
}
@ -2954,49 +2827,181 @@ WorkerPrivate::BlockAndCollectRuntimeStats(bool aIsQuick, void* aData)
AssertIsOnMainThread();
NS_ASSERTION(aData, "Null data!");
JSRuntime* rt;
{
MutexAutoLock lock(mMutex);
// If the worker is being torn down then we just bail out now.
if (!mMemoryReporter) {
if (aIsQuick) {
*static_cast<int64_t*>(aData) = 0;
}
return true;
}
// This signals the worker that it should block itself as soon as possible.
mMemoryReporterRunning = true;
NS_ASSERTION(mJSContext, "This must never be null!");
rt = JS_GetRuntime(mJSContext);
// If the worker is not already blocked (e.g. waiting for a worker event or
// currently in a ctypes call) then we need to trigger the operation
// callback to trap the worker.
if (!mBlockedForMemoryReporter) {
JS_TriggerOperationCallback(rt);
// Wait until the worker actually blocks.
while (!mBlockedForMemoryReporter) {
mMemoryReportCondVar.Wait();
}
}
}
bool succeeded;
nsRefPtr<CollectRuntimeStatsRunnable> runnable =
new CollectRuntimeStatsRunnable(this, aIsQuick, aData, &succeeded);
if (!runnable->Dispatch(nullptr)) {
NS_WARNING("Failed to dispatch runnable!");
succeeded = false;
// Now do the actual report.
if (aIsQuick) {
*static_cast<int64_t*>(aData) =
JS::GetExplicitNonHeapForRuntime(rt, JsWorkerMallocSizeOf);
succeeded = true;
} else {
succeeded =
JS::CollectRuntimeStats(rt, static_cast<JS::RuntimeStats*>(aData),
nullptr);
}
{
MutexAutoLock lock(mMutex);
mMemoryReporterRunning = false;
}
MutexAutoLock lock(mMutex);
NS_ASSERTION(mMemoryReporterRunning, "This isn't possible!");
NS_ASSERTION(mBlockedForMemoryReporter, "Somehow we got unblocked!");
// Tell the worker that it can now continue its execution.
mMemoryReporterRunning = false;
// The worker may be waiting so we must notify.
mMemoryReportCondVar.Notify();
return succeeded;
}
bool
void
WorkerPrivate::NoteDeadMemoryReporter()
{
// This may happen on the worker thread or the main thread.
MutexAutoLock lock(mMutex);
NS_ASSERTION(mMemoryReporterAlive, "Must be alive!");
// Tell the worker that the memory reporter has died.
mMemoryReporterAlive = false;
// The worker may be waiting so we must notify.
mMemoryReportCondVar.Notify();
}
void
WorkerPrivate::EnableMemoryReporter()
{
AssertIsOnWorkerThread();
NS_ASSERTION(!mMemoryReporterAlive, "This isn't possible!");
// No need to lock here since the main thread can't race until we've
// successfully registered the reporter.
mMemoryReporterAlive = true;
mMemoryReporter = new MemoryReporter(this);
if (NS_FAILED(NS_RegisterMemoryMultiReporter(mMemoryReporter))) {
NS_WARNING("Failed to register memory reporter!");
// No need to lock here since a failed registration means our memory
// reporter can't start running. Just clean up.
mMemoryReporter = nullptr;
// That should have killed the memory reporter.
NS_ASSERTION(!mMemoryReporterAlive, "Must be dead!");
return;
}
}
void
WorkerPrivate::DisableMemoryReporter()
{
AssertIsOnWorkerThread();
bool result = true;
// First swap out our memory reporter so that the main thread stops trying to
// signal us when it's reporting memory.
nsCOMPtr<nsIMemoryMultiReporter> memoryReporter;
{
MutexAutoLock lock(mMutex);
while (mMemoryReporterRunning) {
MutexAutoUnlock unlock(mMutex);
result = ProcessAllControlRunnables() && result;
}
if (mMemoryReporter) {
mMemoryReporter->Disable();
}
mMemoryReporter.swap(memoryReporter);
}
return result;
// The main thread could immediately try to report memory again here but it
// will not actually do anything since we've cleared mMemoryReporter.
// If we never registered the memory reporter (e.g. some kind of error in
// NS_RegisterMemoryMultiReporter) then there's nothing else we need to do
// here.
if (!memoryReporter) {
return;
}
NS_ASSERTION(mMemoryReporterAlive, "Huh?!");
// Unregister.
if (NS_FAILED(NS_UnregisterMemoryMultiReporter(memoryReporter))) {
// If this fails then the memory reporter will probably never die and we'll
// hang below waiting for it. If this ever happens we need to fix
// NS_UnregisterMemoryMultiReporter.
MOZ_NOT_REACHED("Failed to unregister memory reporter! Worker is going to "
"hang!");
}
// Now drop our reference to the memory reporter. The main thread could be
// holding another reference so it may not die immediately.
memoryReporter = nullptr;
MutexAutoLock lock(mMutex);
// Wait once again to make sure that the main thread is done with the memory
// reporter.
while (mMemoryReporterAlive) {
mMemoryReportCondVar.Wait();
}
}
void
WorkerPrivate::WaitForWorkerEvents(PRIntervalTime aInterval)
{
AssertIsOnWorkerThread();
mMutex.AssertCurrentThreadOwns();
NS_ASSERTION(!mBlockedForMemoryReporter,
"Can't be blocked in more than one place at the same time!");
// Let the main thread know that the worker is blocked and that memory
// reporting may proceed.
mBlockedForMemoryReporter = true;
// The main thread may be waiting so we must notify.
mMemoryReportCondVar.Notify();
// Now wait for an actual worker event.
mCondVar.Wait(aInterval);
// We've gotten some kind of signal but we can't continue until the memory
// reporter has finished. Wait again.
while (mMemoryReporterRunning) {
mMemoryReportCondVar.Wait();
}
NS_ASSERTION(mBlockedForMemoryReporter, "Somehow we got unblocked!");
// No need to notify here as the main thread isn't watching for this state.
mBlockedForMemoryReporter = false;
}
bool
@ -3010,6 +3015,32 @@ WorkerPrivate::ProcessAllControlRunnables()
WorkerRunnable* event;
{
MutexAutoLock lock(mMutex);
// Block here if the memory reporter is trying to run.
if (mMemoryReporterRunning) {
NS_ASSERTION(!mBlockedForMemoryReporter,
"Can't be blocked in more than one place at the same "
"time!");
// Let the main thread know that we've received the block request and
// that memory reporting may proceed.
mBlockedForMemoryReporter = true;
// The main thread is almost certainly waiting so we must notify here.
mMemoryReportCondVar.Notify();
// Wait for the memory report to finish.
while (mMemoryReporterRunning) {
mMemoryReportCondVar.Wait();
}
NS_ASSERTION(mBlockedForMemoryReporter, "Somehow we got unblocked!");
// No need to notify here as the main thread isn't watching for this
// state.
mBlockedForMemoryReporter = false;
}
if (!mControlQueue.Pop(event)) {
break;
}
@ -3021,6 +3052,7 @@ WorkerPrivate::ProcessAllControlRunnables()
NS_RELEASE(event);
}
return result;
}
@ -3371,7 +3403,7 @@ WorkerPrivate::RunSyncLoop(JSContext* aCx, uint32_t aSyncLoopKey)
MutexAutoLock lock(mMutex);
while (!mControlQueue.Pop(event) && !syncQueue->mQueue.Pop(event)) {
mCondVar.Wait();
WaitForWorkerEvents();
}
}
@ -4119,6 +4151,46 @@ WorkerPrivate::GetContentSecurityPolicy(JSContext* aCx,
return true;
}
void
WorkerPrivate::BeginCTypesCall()
{
AssertIsOnWorkerThread();
MutexAutoLock lock(mMutex);
NS_ASSERTION(!mBlockedForMemoryReporter,
"Can't be blocked in more than one place at the same time!");
// Let the main thread know that the worker is effectively blocked while in
// this ctypes call. It isn't technically true (obviously the call could do
// non-blocking things), but we're assuming that ctypes can't call back into
// JSAPI here and therefore any work the ctypes call does will not alter the
// data structures of this JS runtime.
mBlockedForMemoryReporter = true;
// The main thread may be waiting on us so it must be notified.
mMemoryReportCondVar.Notify();
}
void
WorkerPrivate::EndCTypesCall()
{
AssertIsOnWorkerThread();
MutexAutoLock lock(mMutex);
NS_ASSERTION(mBlockedForMemoryReporter, "Somehow we got unblocked!");
// Don't continue until the memory reporter has finished.
while (mMemoryReporterRunning) {
mMemoryReportCondVar.Wait();
}
// No need to notify the main thread here as it shouldn't be waiting to see
// this state.
mBlockedForMemoryReporter = false;
}
BEGIN_WORKERS_NAMESPACE
// Force instantiation.

Просмотреть файл

@ -32,6 +32,7 @@
class JSAutoStructuredCloneBuffer;
class nsIDocument;
class nsIMemoryMultiReporter;
class nsIPrincipal;
class nsIScriptContext;
class nsIURI;
@ -41,7 +42,6 @@ class nsIXPCScriptNotify;
BEGIN_WORKERS_NAMESPACE
class WorkerMemoryReporter;
class WorkerPrivate;
class WorkerRunnable : public nsIRunnable
@ -167,6 +167,7 @@ public:
protected:
mozilla::Mutex mMutex;
mozilla::CondVar mCondVar;
mozilla::CondVar mMemoryReportCondVar;
private:
JSObject* mJSObject;
@ -518,7 +519,6 @@ public:
class WorkerPrivate : public WorkerPrivateParent<WorkerPrivate>
{
friend class WorkerPrivateParent<WorkerPrivate>;
friend class WorkerMemoryReporter;
typedef WorkerPrivateParent<WorkerPrivate> ParentType;
struct TimeoutInfo;
@ -558,7 +558,7 @@ class WorkerPrivate : public WorkerPrivateParent<WorkerPrivate>
nsTArray<nsAutoPtr<TimeoutInfo> > mTimeouts;
nsCOMPtr<nsITimer> mTimer;
nsRefPtr<WorkerMemoryReporter> mMemoryReporter;
nsCOMPtr<nsIMemoryMultiReporter> mMemoryReporter;
mozilla::TimeStamp mKillTime;
uint32_t mErrorHandlerRecursionCount;
@ -569,7 +569,9 @@ class WorkerPrivate : public WorkerPrivateParent<WorkerPrivate>
bool mRunningExpiredTimeouts;
bool mCloseHandlerStarted;
bool mCloseHandlerFinished;
bool mMemoryReporterAlive;
bool mMemoryReporterRunning;
bool mBlockedForMemoryReporter;
bool mXHRParamsAllowed;
#ifdef DEBUG
@ -710,10 +712,10 @@ public:
ScheduleDeletion(bool aWasPending);
bool
BlockAndCollectRuntimeStats(bool isQuick, void* aData);
BlockAndCollectRuntimeStats(bool aIsQuick, void* aData);
bool
DisableMemoryReporter();
void
NoteDeadMemoryReporter();
bool
XHRParamsAllowed() const
@ -761,6 +763,14 @@ public:
WorkerCrossThreadDispatcher*
GetCrossThreadDispatcher();
// This may block!
void
BeginCTypesCall();
// This may block!
void
EndCTypesCall();
private:
WorkerPrivate(JSContext* aCx, JSObject* aObject, WorkerPrivate* aParent,
JSContext* aParentJSContext, const nsAString& aScriptURL,
@ -830,6 +840,15 @@ private:
bool
ProcessAllControlRunnables();
void
EnableMemoryReporter();
void
DisableMemoryReporter();
void
WaitForWorkerEvents(PRIntervalTime interval = PR_INTERVAL_NO_TIMEOUT);
static bool
CheckXHRParamsAllowed(nsPIDOMWindow* aWindow);
};

Просмотреть файл

@ -5789,6 +5789,11 @@ FunctionType::Call(JSContext* cx,
return false;
}
// Let the runtime callback know that we are about to call into C.
js::CTypesActivityCallback activityCallback = cx->runtime->ctypesActivityCallback;
if (activityCallback)
activityCallback(cx, js::CTYPES_CALL_BEGIN);
uintptr_t fn = *reinterpret_cast<uintptr_t*>(CData::GetData(obj));
#if defined(XP_WIN)
@ -5816,6 +5821,9 @@ FunctionType::Call(JSContext* cx,
errno = savedErrno;
if (activityCallback)
activityCallback(cx, js::CTYPES_CALL_END);
// Store the error value for later consultation with |ctypes.getStatus|
JSObject *objCTypes = CType::GetGlobalCTypes(cx, typeObj);
if (!objCTypes)

Просмотреть файл

@ -888,6 +888,7 @@ JSRuntime::JSRuntime(JSUseHelperThreads useHelperThreads)
ionPcScriptCache(NULL),
threadPool(this),
ionReturnOverride_(MagicValue(JS_ARG_POISON)),
ctypesActivityCallback(NULL),
useHelperThreads_(useHelperThreads),
requestedHelperThreadCount(-1)
{

Просмотреть файл

@ -1042,6 +1042,8 @@ struct JSRuntime : js::RuntimeFriendFields
js::ThreadPool threadPool;
js::CTypesActivityCallback ctypesActivityCallback;
private:
// In certain cases, we want to optimize certain opcodes to typed instructions,
// to avoid carrying an extra register to feed into an unbox. Unfortunately,

Просмотреть файл

@ -984,3 +984,9 @@ js::GetListBaseExpandoSlot()
{
return gListBaseExpandoSlot;
}
JS_FRIEND_API(void)
js::SetCTypesActivityCallback(JSRuntime *rt, CTypesActivityCallback cb)
{
rt->ctypesActivityCallback = cb;
}

Просмотреть файл

@ -1561,6 +1561,21 @@ IsReadOnlyDateMethod(JS::IsAcceptableThis test, JS::NativeImpl method);
extern JS_FRIEND_API(bool)
IsTypedArrayThisCheck(JS::IsAcceptableThis test);
enum CTypesActivityType {
CTYPES_CALL_BEGIN,
CTYPES_CALL_END
};
typedef void
(* CTypesActivityCallback)(JSContext *cx, CTypesActivityType type);
/*
* Sets a callback that is run whenever js-ctypes is about to be used when
* calling into C.
*/
JS_FRIEND_API(void)
SetCTypesActivityCallback(JSRuntime *rt, CTypesActivityCallback cb);
} /* namespace js */
#endif /* jsfriendapi_h___ */
#endif /* jsfriendapi_h___ */