From 1a9f4cad0871bcde9420f633c527b933ac288057 Mon Sep 17 00:00:00 2001
From: Connor Brewster <connor.brewster@eagles.oc.edu>
Date: Thu, 20 Jul 2017 16:35:46 -0600
Subject: [PATCH] Fix compartment mismatch issue

---
 components/script/dom/bindings/codegen/CodegenRust.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/components/script/dom/bindings/codegen/CodegenRust.py b/components/script/dom/bindings/codegen/CodegenRust.py
index 482b713e10c..7f97868a378 100644
--- a/components/script/dom/bindings/codegen/CodegenRust.py
+++ b/components/script/dom/bindings/codegen/CodegenRust.py
@@ -5377,7 +5377,12 @@ let result = match result {
     },
 };
 
-JS_SetPrototype(cx, result.reflector().get_jsobject(), prototype.handle());
+rooted!(in(cx) let mut element = result.reflector().get_jsobject().get());
+if !JS_WrapObject(cx, element.handle_mut()) {
+    return false;
+}
+
+JS_SetPrototype(cx, element.handle(), prototype.handle());
 
 (result).to_jsval(cx, args.rval());
 return true;