JSDocumentCustom.cpp [plain text]
#include "config.h"
#include "JSDocument.h"
#include "Frame.h"
#include "JSDOMWindowCustom.h"
#include "JSHTMLDocument.h"
#include "JSXMLDocument.h"
#include "NodeTraversal.h"
#include "SVGDocument.h"
#include <JavaScriptCore/HeapSnapshotBuilder.h>
namespace WebCore {
using namespace JSC;
static inline JSValue createNewDocumentWrapper(ExecState& state, JSDOMGlobalObject& globalObject, Ref<Document>&& passedDocument)
{
auto& document = passedDocument.get();
JSObject* wrapper;
if (document.isHTMLDocument())
wrapper = createWrapper<HTMLDocument>(&globalObject, WTFMove(passedDocument));
else if (document.isXMLDocument())
wrapper = createWrapper<XMLDocument>(&globalObject, WTFMove(passedDocument));
else
wrapper = createWrapper<Document>(&globalObject, WTFMove(passedDocument));
reportMemoryForDocumentIfFrameless(state, document);
return wrapper;
}
JSObject* cachedDocumentWrapper(ExecState& state, JSDOMGlobalObject& globalObject, Document& document)
{
if (auto* wrapper = getCachedWrapper(globalObject.world(), document))
return wrapper;
auto* window = document.domWindow();
if (!window)
return nullptr;
auto* documentGlobalObject = toJSDOMWindow(state.vm(), toJS(&state, *window));
if (!documentGlobalObject)
return nullptr;
return getCachedWrapper(documentGlobalObject->world(), document);
}
void reportMemoryForDocumentIfFrameless(ExecState& state, Document& document)
{
if (document.frame())
return;
VM& vm = state.vm();
size_t memoryCost = 0;
for (Node* node = &document; node; node = NodeTraversal::next(*node))
memoryCost += node->approximateMemoryCost();
vm.heap.deprecatedReportExtraMemory(memoryCost);
}
JSValue toJSNewlyCreated(ExecState* state, JSDOMGlobalObject* globalObject, Ref<Document>&& document)
{
return createNewDocumentWrapper(*state, *globalObject, WTFMove(document));
}
JSValue toJS(ExecState* state, JSDOMGlobalObject* globalObject, Document& document)
{
if (auto* wrapper = cachedDocumentWrapper(*state, *globalObject, document))
return wrapper;
return toJSNewlyCreated(state, globalObject, Ref<Document>(document));
}
void JSDocument::visitAdditionalChildren(SlotVisitor& visitor)
{
visitor.addOpaqueRoot(static_cast<ScriptExecutionContext*>(&wrapped()));
}
void JSDocument::heapSnapshot(JSCell* cell, HeapSnapshotBuilder& builder)
{
Base::heapSnapshot(cell, builder);
auto* thisObject = jsCast<JSDocument*>(cell);
builder.setLabelForCell(cell, thisObject->wrapped().url().string());
}
}