1 files changed, 146 insertions, 0 deletions
diff --git a/intl/locale/windows/nsCollationWin.cpp b/intl/locale/windows/nsCollationWin.cpp
new file mode 100644
index 000000000..8dcdb8f07
--- /dev/null
+++ b/intl/locale/windows/nsCollationWin.cpp
@@ -0,0 +1,146 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* 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 "nsCollationWin.h"
+#include "nsIServiceManager.h"
+#include "nsIComponentManager.h"
+#include "nsILocaleService.h"
+#include "nsIPlatformCharset.h"
+#include "nsWin32Locale.h"
+#include "nsCOMPtr.h"
+#include "prmem.h"
+#include "plstr.h"
+#include <windows.h>
+
+#undef CompareString
+
+NS_IMPL_ISUPPORTS(nsCollationWin, nsICollation)
+
+
+nsCollationWin::nsCollationWin() : mCollation(nullptr)
+{
+}
+
+nsCollationWin::~nsCollationWin() 
+{
+  if (mCollation)
+    delete mCollation;
+}
+
+nsresult nsCollationWin::Initialize(nsILocale* locale) 
+{
+  NS_ASSERTION(!mCollation, "Should only be initialized once.");
+
+  nsresult res;
+
+  mCollation = new nsCollation;
+
+  // default LCID (en-US)
+  mLCID = 1033;
+
+  nsAutoString localeStr;
+
+  // get locale string, use app default if no locale specified
+  if (!locale) {
+    nsCOMPtr<nsILocaleService> localeService = 
+             do_GetService(NS_LOCALESERVICE_CONTRACTID);
+    if (localeService) {
+      nsCOMPtr<nsILocale> appLocale;
+      res = localeService->GetApplicationLocale(getter_AddRefs(appLocale));
+      if (NS_SUCCEEDED(res)) {
+        res = appLocale->GetCategory(NS_LITERAL_STRING("NSILOCALE_COLLATE"), 
+                                     localeStr);
+      }
+    }
+  }
+  else {
+    res = locale->GetCategory(NS_LITERAL_STRING("NSILOCALE_COLLATE"), 
+                              localeStr);
+  }
+
+  // Get LCID and charset name from locale, if available
+  LCID lcid;
+  res = nsWin32Locale::GetPlatformLocale(localeStr, &lcid);
+  if (NS_SUCCEEDED(res)) {
+    mLCID = lcid;
+  }
+
+  nsCOMPtr <nsIPlatformCharset> platformCharset = 
+      do_GetService(NS_PLATFORMCHARSET_CONTRACTID);
+  if (platformCharset) {
+    nsAutoCString mappedCharset;
+    res = platformCharset->GetDefaultCharsetForLocale(localeStr, mappedCharset);
+    if (NS_SUCCEEDED(res)) {
+      mCollation->SetCharset(mappedCharset.get());
+    }
+  }
+
+  return NS_OK;
+}
+
+
+NS_IMETHODIMP nsCollationWin::CompareString(int32_t strength, 
+                                            const nsAString & string1, 
+                                            const nsAString & string2, 
+                                            int32_t *result)
+{
+  int retval;
+  nsresult res;
+  DWORD dwMapFlags = 0;
+
+  if (strength == kCollationCaseInSensitive)
+    dwMapFlags |= NORM_IGNORECASE;
+
+  retval = ::CompareStringW(mLCID, 
+                            dwMapFlags,
+                            (LPCWSTR) PromiseFlatString(string1).get(), 
+                            -1,
+                            (LPCWSTR) PromiseFlatString(string2).get(), 
+                            -1);
+  if (retval) {
+    res = NS_OK;
+    *result = retval - 2;
+  } else {
+    res = NS_ERROR_FAILURE;
+  }
+
+  return res;
+}
+ 
+
+nsresult nsCollationWin::AllocateRawSortKey(int32_t strength, 
+                                            const nsAString& stringIn, uint8_t** key, uint32_t* outLen)
+{
+  int byteLen;
+  void *buffer;
+  nsresult res = NS_OK;
+  DWORD dwMapFlags = LCMAP_SORTKEY;
+
+  if (strength == kCollationCaseInSensitive)
+    dwMapFlags |= NORM_IGNORECASE;
+
+  byteLen = LCMapStringW(mLCID, dwMapFlags, 
+                         (LPCWSTR) PromiseFlatString(stringIn).get(),
+                         -1, nullptr, 0);
+  buffer = PR_Malloc(byteLen);
+  if (!buffer) {
+    res = NS_ERROR_OUT_OF_MEMORY;
+  } else {
+    *key = (uint8_t *)buffer;
+    *outLen = LCMapStringW(mLCID, dwMapFlags, 
+                           (LPCWSTR) PromiseFlatString(stringIn).get(),
+                           -1, (LPWSTR) buffer, byteLen);
+  }
+  return res;
+}
+
+nsresult nsCollationWin::CompareRawSortKey(const uint8_t* key1, uint32_t len1, 
+                                           const uint8_t* key2, uint32_t len2, 
+                                           int32_t* result)
+{
+  *result = PL_strcmp((const char *)key1, (const char *)key2);
+  return NS_OK;
+}