diff options
author | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
---|---|---|
committer | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
commit | 5f8de423f190bbb79a62f804151bc24824fa32d8 (patch) | |
tree | 10027f336435511475e392454359edea8e25895d /storage/SQLCollations.cpp | |
parent | 49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff) | |
download | UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip |
Add m-esr52 at 52.6.0
Diffstat (limited to 'storage/SQLCollations.cpp')
-rw-r--r-- | storage/SQLCollations.cpp | 242 |
1 files changed, 242 insertions, 0 deletions
diff --git a/storage/SQLCollations.cpp b/storage/SQLCollations.cpp new file mode 100644 index 000000000..392bcd804 --- /dev/null +++ b/storage/SQLCollations.cpp @@ -0,0 +1,242 @@ +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- + * vim: sw=2 ts=2 et lcs=trail\:.,tab\:>~ : + * 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 "mozilla/ArrayUtils.h" + +#include "SQLCollations.h" + +namespace mozilla { +namespace storage { + +//////////////////////////////////////////////////////////////////////////////// +//// Local Helper Functions + +namespace { + +/** + * Helper function for the UTF-8 locale collations. + * + * @param aService + * The Service that owns the nsICollation used by this collation. + * @param aLen1 + * The number of bytes in aStr1. + * @param aStr1 + * The string to be compared against aStr2 as provided by SQLite. It + * must be a non-null-terminated char* buffer. + * @param aLen2 + * The number of bytes in aStr2. + * @param aStr2 + * The string to be compared against aStr1 as provided by SQLite. It + * must be a non-null-terminated char* buffer. + * @param aComparisonStrength + * The sorting strength, one of the nsICollation constants. + * @return aStr1 - aStr2. That is, if aStr1 < aStr2, returns a negative number. + * If aStr1 > aStr2, returns a positive number. If aStr1 == aStr2, + * returns 0. + */ +int +localeCollationHelper8(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2, + int32_t aComparisonStrength) +{ + NS_ConvertUTF8toUTF16 str1(static_cast<const char *>(aStr1), aLen1); + NS_ConvertUTF8toUTF16 str2(static_cast<const char *>(aStr2), aLen2); + Service *serv = static_cast<Service *>(aService); + return serv->localeCompareStrings(str1, str2, aComparisonStrength); +} + +/** + * Helper function for the UTF-16 locale collations. + * + * @param aService + * The Service that owns the nsICollation used by this collation. + * @param aLen1 + * The number of bytes (not characters) in aStr1. + * @param aStr1 + * The string to be compared against aStr2 as provided by SQLite. It + * must be a non-null-terminated char16_t* buffer. + * @param aLen2 + * The number of bytes (not characters) in aStr2. + * @param aStr2 + * The string to be compared against aStr1 as provided by SQLite. It + * must be a non-null-terminated char16_t* buffer. + * @param aComparisonStrength + * The sorting strength, one of the nsICollation constants. + * @return aStr1 - aStr2. That is, if aStr1 < aStr2, returns a negative number. + * If aStr1 > aStr2, returns a positive number. If aStr1 == aStr2, + * returns 0. + */ +int +localeCollationHelper16(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2, + int32_t aComparisonStrength) +{ + const char16_t *buf1 = static_cast<const char16_t *>(aStr1); + const char16_t *buf2 = static_cast<const char16_t *>(aStr2); + + // The second argument to the nsDependentSubstring constructor is exclusive: + // It points to the char16_t immediately following the last one in the target + // substring. Since aLen1 and aLen2 are in bytes, divide by sizeof(char16_t) + // so that the pointer arithmetic is correct. + nsDependentSubstring str1(buf1, buf1 + (aLen1 / sizeof(char16_t))); + nsDependentSubstring str2(buf2, buf2 + (aLen2 / sizeof(char16_t))); + Service *serv = static_cast<Service *>(aService); + return serv->localeCompareStrings(str1, str2, aComparisonStrength); +} + +// This struct is used only by registerCollations below, but ISO C++98 forbids +// instantiating a template dependent on a locally-defined type. Boo-urns! +struct Collations { + const char *zName; + int enc; + int(*xCompare)(void*, int, const void*, int, const void*); +}; + +} // namespace + +//////////////////////////////////////////////////////////////////////////////// +//// Exposed Functions + +int +registerCollations(sqlite3 *aDB, + Service *aService) +{ + Collations collations[] = { + {"locale", + SQLITE_UTF8, + localeCollation8}, + {"locale_case_sensitive", + SQLITE_UTF8, + localeCollationCaseSensitive8}, + {"locale_accent_sensitive", + SQLITE_UTF8, + localeCollationAccentSensitive8}, + {"locale_case_accent_sensitive", + SQLITE_UTF8, + localeCollationCaseAccentSensitive8}, + {"locale", + SQLITE_UTF16, + localeCollation16}, + {"locale_case_sensitive", + SQLITE_UTF16, + localeCollationCaseSensitive16}, + {"locale_accent_sensitive", + SQLITE_UTF16, + localeCollationAccentSensitive16}, + {"locale_case_accent_sensitive", + SQLITE_UTF16, + localeCollationCaseAccentSensitive16}, + }; + + int rv = SQLITE_OK; + for (size_t i = 0; SQLITE_OK == rv && i < ArrayLength(collations); ++i) { + struct Collations *p = &collations[i]; + rv = ::sqlite3_create_collation(aDB, p->zName, p->enc, aService, + p->xCompare); + } + + return rv; +} + +//////////////////////////////////////////////////////////////////////////////// +//// SQL Collations + +int +localeCollation8(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2) +{ + return localeCollationHelper8(aService, aLen1, aStr1, aLen2, aStr2, + nsICollation::kCollationCaseInSensitive); +} + +int +localeCollationCaseSensitive8(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2) +{ + return localeCollationHelper8(aService, aLen1, aStr1, aLen2, aStr2, + nsICollation::kCollationAccentInsenstive); +} + +int +localeCollationAccentSensitive8(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2) +{ + return localeCollationHelper8(aService, aLen1, aStr1, aLen2, aStr2, + nsICollation::kCollationCaseInsensitiveAscii); +} + +int +localeCollationCaseAccentSensitive8(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2) +{ + return localeCollationHelper8(aService, aLen1, aStr1, aLen2, aStr2, + nsICollation::kCollationCaseSensitive); +} + +int +localeCollation16(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2) +{ + return localeCollationHelper16(aService, aLen1, aStr1, aLen2, aStr2, + nsICollation::kCollationCaseInSensitive); +} + +int +localeCollationCaseSensitive16(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2) +{ + return localeCollationHelper16(aService, aLen1, aStr1, aLen2, aStr2, + nsICollation::kCollationAccentInsenstive); +} + +int +localeCollationAccentSensitive16(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2) +{ + return localeCollationHelper16(aService, aLen1, aStr1, aLen2, aStr2, + nsICollation::kCollationCaseInsensitiveAscii); +} + +int +localeCollationCaseAccentSensitive16(void *aService, + int aLen1, + const void *aStr1, + int aLen2, + const void *aStr2) +{ + return localeCollationHelper16(aService, aLen1, aStr1, aLen2, aStr2, + nsICollation::kCollationCaseSensitive); +} + +} // namespace storage +} // namespace mozilla |