Bug 1968644 - Part 5: Populate triggeringFirstPartyClassificationFlags and triggeringThirdPartyClassificationFlags for CSS loads. r=emilio

Differential Revision: https://phabricator.services.mozilla.com/D252447

Author: artines1

layout/style/Loader.cpp

@@ -42,6 +42,7 @@
 #include "nsContentPolicyUtils.h"
 #include "nsIHttpChannel.h"
 #include "nsIHttpChannelInternal.h"
+#include "nsIClassifiedChannel.h"
 #include "nsIClassOfService.h"
 #include "nsIScriptError.h"
 #include "nsMimeTypes.h"
@@ -1069,6 +1070,7 @@ nsresult Loader::NewStyleSheetChannel(SheetLoadData& aLoadData,
                                       nsIChannel** aOutChannel) {
   nsCOMPtr<nsILoadGroup> loadGroup;
   nsCOMPtr<nsICookieJarSettings> cookieJarSettings;
+  net::ClassificationFlags triggeringClassificationFlags;
   if (aUseLoadGroup == UseLoadGroup::Yes && mDocument) {
     loadGroup = mDocument->GetDocumentLoadGroup();
     // load for a document with no loadgrup indicates that something is
@@ -1079,6 +1081,11 @@ nsresult Loader::NewStyleSheetChannel(SheetLoadData& aLoadData,
     }
 
     cookieJarSettings = mDocument->CookieJarSettings();
+
+    // If the script context is tracking, we use the flags from the script
+    // tracking flags. Otherwise, we fallback to use the flags from the
+    // document.
+    triggeringClassificationFlags = mDocument->GetScriptTrackingFlags();
   }
 
   nsSecurityFlags securityFlags =
@@ -1121,9 +1128,17 @@ nsresult Loader::NewStyleSheetChannel(SheetLoadData& aLoadData,
     }
   }
 
-  return NS_NewChannel(aOutChannel, aLoadData.mURI, triggeringPrincipal,
-                       securityFlags, contentPolicyType, cookieJarSettings,
-                       /* aPerformanceStorage = */ nullptr, loadGroup);
+  MOZ_TRY(NS_NewChannel(aOutChannel, aLoadData.mURI, triggeringPrincipal,
+                        securityFlags, contentPolicyType, cookieJarSettings,
+                        /* aPerformanceStorage = */ nullptr, loadGroup));
+
+  nsCOMPtr<nsILoadInfo> loadInfo = (*aOutChannel)->LoadInfo();
+  loadInfo->SetTriggeringFirstPartyClassificationFlags(
+      triggeringClassificationFlags.firstPartyFlags);
+  loadInfo->SetTriggeringThirdPartyClassificationFlags(
+      triggeringClassificationFlags.thirdPartyFlags);
+
+  return NS_OK;
 }
 
 nsresult Loader::LoadSheetSyncInternal(SheetLoadData& aLoadData,