Use RootedTraceableBox for unions.

1 files changed, 6 insertions, 0 deletions

diff --git a/components/script/dom/bindings/codegen/CodegenRust.py b/components/script/dom/bindings/codegen/CodegenRust.py
index e4448ff834c..4a9e7a8cc32 100644
--- a/components/script/dom/bindings/codegen/CodegenRust.py
+++ b/components/script/dom/bindings/codegen/CodegenRust.py
@@ -722,6 +722,9 @@ def getJSToNativeConversionInfo(type, descriptorProvider, failureCode=None,
         if type.nullable():
             declType = CGWrapper(declType, pre="Option<", post=" >")
 
+        if isMember != "Dictionary" and type_needs_tracing(type):
+            declType = CGTemplatedType("RootedTraceableBox", declType)
+
         templateBody = ("match FromJSValConvertible::from_jsval(cx, ${val}, ()) {\n"
                         "    Ok(ConversionResult::Success(value)) => value,\n"
                         "    Ok(ConversionResult::Failure(error)) => {\n"
@@ -6190,6 +6193,9 @@ def type_needs_tracing(t):
         if t.isSequence():
             return type_needs_tracing(t.inner)
 
+        if t.isUnion():
+            return any(type_needs_tracing(member) for member in t.flatMemberTypes)
+
         return False
 
     if t.isDictionary():