Bug 1968644 - Part 8: Populate the triggeringFirstPartyClassificationFlags and triggeringThirdPartyClassificationFlags for WebSocket requests. r=valentin,necko-reviewers

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

Author: artines1

netwerk/protocol/websocket/BaseWebSocketChannel.cpp

@@ -6,7 +6,10 @@
 
 #include "WebSocketLog.h"
 #include "BaseWebSocketChannel.h"
+#include "mozilla/dom/Document.h"
 #include "MainThreadUtils.h"
+#include "nsContentUtils.h"
+#include "nsIClassifiedChannel.h"
 #include "nsILoadGroup.h"
 #include "nsINode.h"
 #include "nsIInterfaceRequestor.h"
@@ -226,6 +229,17 @@ BaseWebSocketChannel::InitLoadInfoNative(
   if (aCookieJarSettings) {
     mLoadInfo->SetCookieJarSettings(aCookieJarSettings);
   }
+
+  if (aLoadingNode) {
+    RefPtr<dom::Document> doc = aLoadingNode->OwnerDoc();
+    if (doc) {
+      ClassificationFlags flags = doc->GetScriptTrackingFlags();
+      mLoadInfo->SetTriggeringFirstPartyClassificationFlags(
+          flags.firstPartyFlags);
+      mLoadInfo->SetTriggeringThirdPartyClassificationFlags(
+          flags.thirdPartyFlags);
+    }
+  }
   return NS_OK;
 }