1 files changed, 161 insertions, 0 deletions
diff --git a/js/xpconnect/loader/mozJSComponentLoader.h b/js/xpconnect/loader/mozJSComponentLoader.h
new file mode 100644
index 000000000..6ebe4a371
--- /dev/null
+++ b/js/xpconnect/loader/mozJSComponentLoader.h
@@ -0,0 +1,161 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/* vim: set ts=8 sts=4 et sw=4 tw=99: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef mozJSComponentLoader_h
+#define mozJSComponentLoader_h
+
+#include "mozilla/dom/ScriptSettings.h"
+#include "mozilla/MemoryReporting.h"
+#include "mozilla/ModuleLoader.h"
+#include "nsAutoPtr.h"
+#include "nsISupports.h"
+#include "nsIObserver.h"
+#include "nsIURI.h"
+#include "xpcIJSModuleLoader.h"
+#include "nsClassHashtable.h"
+#include "nsDataHashtable.h"
+#include "jsapi.h"
+
+#include "xpcIJSGetFactory.h"
+
+class nsIFile;
+class nsIPrincipal;
+class nsIXPConnectJSObjectHolder;
+class ComponentLoaderInfo;
+
+/* 6bd13476-1dd2-11b2-bbef-f0ccb5fa64b6 (thanks, mozbot) */
+
+#define MOZJSCOMPONENTLOADER_CID                                              \
+  {0x6bd13476, 0x1dd2, 0x11b2,                                                \
+    { 0xbb, 0xef, 0xf0, 0xcc, 0xb5, 0xfa, 0x64, 0xb6 }}
+#define MOZJSCOMPONENTLOADER_CONTRACTID "@mozilla.org/moz/jsloader;1"
+
+class mozJSComponentLoader : public mozilla::ModuleLoader,
+                             public xpcIJSModuleLoader,
+                             public nsIObserver
+{
+ public:
+    NS_DECL_ISUPPORTS
+    NS_DECL_XPCIJSMODULELOADER
+    NS_DECL_NSIOBSERVER
+
+    mozJSComponentLoader();
+
+    // ModuleLoader
+    const mozilla::Module* LoadModule(mozilla::FileLocation& aFile) override;
+
+    nsresult FindTargetObject(JSContext* aCx,
+                              JS::MutableHandleObject aTargetObject);
+
+    static mozJSComponentLoader* Get() { return sSelf; }
+
+    size_t SizeOfIncludingThis(mozilla::MallocSizeOf aMallocSizeOf);
+
+ protected:
+    virtual ~mozJSComponentLoader();
+
+    static mozJSComponentLoader* sSelf;
+
+    nsresult ReallyInit();
+    void UnloadModules();
+
+    JSObject* PrepareObjectForLocation(JSContext* aCx,
+                                       nsIFile* aComponentFile,
+                                       nsIURI* aComponent,
+                                       bool aReuseLoaderGlobal,
+                                       bool* aRealFile);
+
+    nsresult ObjectForLocation(ComponentLoaderInfo& aInfo,
+                               nsIFile* aComponentFile,
+                               JS::MutableHandleObject aObject,
+                               JS::MutableHandleScript aTableScript,
+                               char** location,
+                               bool aCatchException,
+                               JS::MutableHandleValue aException);
+
+    nsresult ImportInto(const nsACString& aLocation,
+                        JS::HandleObject targetObj,
+                        JSContext* callercx,
+                        JS::MutableHandleObject vp);
+
+    nsCOMPtr<nsIComponentManager> mCompMgr;
+    nsCOMPtr<nsIPrincipal> mSystemPrincipal;
+    nsCOMPtr<nsIXPConnectJSObjectHolder> mLoaderGlobal;
+
+    class ModuleEntry : public mozilla::Module
+    {
+    public:
+        explicit ModuleEntry(JS::RootingContext* aRootingCx)
+          : mozilla::Module(), obj(aRootingCx), thisObjectKey(aRootingCx)
+        {
+            mVersion = mozilla::Module::kVersion;
+            mCIDs = nullptr;
+            mContractIDs = nullptr;
+            mCategoryEntries = nullptr;
+            getFactoryProc = GetFactory;
+            loadProc = nullptr;
+            unloadProc = nullptr;
+
+            location = nullptr;
+        }
+
+        ~ModuleEntry() {
+            Clear();
+        }
+
+        void Clear() {
+            getfactoryobj = nullptr;
+
+            if (obj) {
+                mozilla::AutoJSContext cx;
+                JSAutoCompartment ac(cx, obj);
+
+                if (JS_HasExtensibleLexicalEnvironment(obj)) {
+                    JS_SetAllNonReservedSlotsToUndefined(cx, JS_ExtensibleLexicalEnvironment(obj));
+                }
+                JS_SetAllNonReservedSlotsToUndefined(cx, obj);
+                obj = nullptr;
+                thisObjectKey = nullptr;
+            }
+
+            if (location)
+                free(location);
+
+            obj = nullptr;
+            thisObjectKey = nullptr;
+            location = nullptr;
+        }
+
+        size_t SizeOfIncludingThis(mozilla::MallocSizeOf aMallocSizeOf) const;
+
+        static already_AddRefed<nsIFactory> GetFactory(const mozilla::Module& module,
+                                                       const mozilla::Module::CIDEntry& entry);
+
+        nsCOMPtr<xpcIJSGetFactory> getfactoryobj;
+        JS::PersistentRootedObject obj;
+        JS::PersistentRootedScript thisObjectKey;
+        char*               location;
+    };
+
+    friend class ModuleEntry;
+
+    static size_t DataEntrySizeOfExcludingThis(const nsACString& aKey, ModuleEntry* const& aData,
+                                               mozilla::MallocSizeOf aMallocSizeOf, void* arg);
+    static size_t ClassEntrySizeOfExcludingThis(const nsACString& aKey,
+                                                const nsAutoPtr<ModuleEntry>& aData,
+                                                mozilla::MallocSizeOf aMallocSizeOf, void* arg);
+
+    // Modules are intentionally leaked, but still cleared.
+    nsDataHashtable<nsCStringHashKey, ModuleEntry*> mModules;
+
+    nsClassHashtable<nsCStringHashKey, ModuleEntry> mImports;
+    nsDataHashtable<nsCStringHashKey, ModuleEntry*> mInProgressImports;
+
+    bool mInitialized;
+    bool mReuseLoaderGlobal;
+};
+
+#endif