1 files changed, 112 insertions, 0 deletions
diff --git a/netwerk/protocol/http/HttpWinUtils.cpp b/netwerk/protocol/http/HttpWinUtils.cpp
new file mode 100644
index 0000000000..f60ce77932
--- /dev/null
+++ b/netwerk/protocol/http/HttpWinUtils.cpp
@@ -0,0 +1,112 @@
+/* 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/. */
+
+#include "HttpWinUtils.h"
+#include "nsIURI.h"
+#include "nsHttpChannel.h"
+#include "mozilla/ClearOnShutdown.h"
+#include <proofofpossessioncookieinfo.h>
+
+namespace mozilla {
+namespace net {
+
+static StaticRefPtr<IProofOfPossessionCookieInfoManager> sPopCookieManager;
+static bool sPopCookieManagerAvailable = true;
+
+void AddWindowsSSO(nsHttpChannel* channel) {
+  if (!sPopCookieManagerAvailable) {
+    return;
+  }
+  HRESULT hr;
+  if (!sPopCookieManager) {
+    GUID CLSID_ProofOfPossessionCookieInfoManager;
+    GUID IID_IProofOfPossessionCookieInfoManager;
+
+    CLSIDFromString(L"{A9927F85-A304-4390-8B23-A75F1C668600}",
+                    &CLSID_ProofOfPossessionCookieInfoManager);
+    IIDFromString(L"{CDAECE56-4EDF-43DF-B113-88E4556FA1BB}",
+                  &IID_IProofOfPossessionCookieInfoManager);
+
+    hr = CoCreateInstance(CLSID_ProofOfPossessionCookieInfoManager, NULL,
+                          CLSCTX_INPROC_SERVER,
+                          IID_IProofOfPossessionCookieInfoManager,
+                          reinterpret_cast<void**>(&sPopCookieManager));
+    if (FAILED(hr)) {
+      sPopCookieManagerAvailable = false;
+      return;
+    }
+
+    RunOnShutdown([&] {
+      if (sPopCookieManager) {
+        sPopCookieManager = nullptr;
+      }
+    });
+  }
+
+  DWORD cookieCount = 0;
+  ProofOfPossessionCookieInfo* cookieInfo = nullptr;
+
+  nsCOMPtr<nsIURI> uri;
+  channel->GetURI(getter_AddRefs(uri));
+
+  nsAutoCString urispec;
+  uri->GetSpec(urispec);
+
+  hr = sPopCookieManager->GetCookieInfoForUri(
+      NS_ConvertUTF8toUTF16(urispec).get(), &cookieCount, &cookieInfo);
+  if (FAILED(hr)) {
+    return;
+  }
+
+  nsAutoCString host;
+  uri->GetHost(host);
+  bool addCookies = false;
+  if (StringEndsWith(host, ".live.com"_ns)) {
+    addCookies = true;
+  }
+
+  nsAutoString allCookies;
+
+  for (DWORD i = 0; i < cookieCount; i++) {
+    nsAutoString cookieData;
+    cookieData.Assign(cookieInfo[i].data);
+    // Strip old Set-Cookie info for WinInet
+    int32_t semicolon = cookieData.FindChar(';');
+    if (semicolon >= 0) {
+      cookieData.SetLength(semicolon);
+    }
+    if (StringBeginsWith(nsDependentString(cookieInfo[i].name), u"x-ms-"_ns)) {
+      channel->SetRequestHeader(NS_ConvertUTF16toUTF8(cookieInfo[i].name),
+                                NS_ConvertUTF16toUTF8(cookieData),
+                                true /* merge */);
+    } else if (addCookies) {
+      if (!allCookies.IsEmpty()) {
+        allCookies.AppendLiteral("; ");
+      }
+      allCookies.Append(cookieInfo[i].name);
+      allCookies.AppendLiteral("=");
+      allCookies.Append(cookieData);
+    }
+  }
+
+  // Merging cookie headers doesn't work correctly as it separates the new
+  // cookies using commas instead of semicolons, so we have to replace
+  // the entire header.
+  if (!allCookies.IsEmpty()) {
+    nsAutoCString cookieHeader;
+    channel->GetRequestHeader(nsDependentCString(nsHttp::Cookie), cookieHeader);
+    if (!cookieHeader.IsEmpty()) {
+      cookieHeader.AppendLiteral("; ");
+    }
+    cookieHeader.Append(NS_ConvertUTF16toUTF8(allCookies));
+    channel->SetRequestHeader(nsDependentCString(nsHttp::Cookie), cookieHeader,
+                              false);
+  }
+  if (cookieInfo) {
+    FreeProofOfPossessionCookieInfoArray(cookieInfo, cookieCount);
+  }
+}
+
+}  // namespace net
+}  // namespace mozilla