qcollator_posix.cpp

Go to the documentation of this file.

// Copyright (C) 2021 The Qt Company Ltd.
// Copyright (C) 2013 Aleix Pol Gonzalez <aleixpol@kde.org>
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
// Qt-Security score:critical reason:data-parser
 
#include "qcollator_p.h"
#include "qstringlist.h"
#include "qstring.h"
#include "qvarlengtharray.h"
 
#include <cstring>
#include <cwchar>
 
QT_BEGIN_NAMESPACE
 
void QCollatorPrivate::init()
{
    if (!isC()) {
        if (locale != QLocale::system().collation()) {
            qWarning("Only the C and system collation locales are supported "
                     "with the POSIX collation implementation");
        }
        if (options.testFlag(Opt::CaseInsensitive))
            qWarning("Case insensitive sorting unsupported in the posix collation implementation");
    }
    if (options.testFlag(Opt::NumericSort))
        qWarning("Numeric mode unsupported in the posix collation implementation");
    if (options.testFlag(Opt::IgnorePunctuation))
        qWarning("Ignoring punctuation unsupported in the posix collation implementation");
    if (options.testFlag(Opt::DiacriticInsensitive))
        qWarning("Ignoring diacritic marks unsupported in the POSIX collation implementation.");
 
    dirty = false;
}
 
void QCollatorPrivate::cleanup()
{
}
 
static void stringToWCharArray(QVarLengthArray<wchar_t> &ret, QStringView string)
{
    ret.resize(string.length());
    qsizetype len = string.toWCharArray(ret.data());
    ret.resize(len+1);
    ret[len] = 0;
}
 
int QCollator::compare(QStringView s1, QStringView s2) const
{
    if (!s1.size())
        return s2.size() ? -1 : 0;
    if (!s2.size())
        return +1;
 
    if (!d)
        d = new QCollatorPrivate(QLocale().collation());
 
    if (d->isC())
        return s1.compare(s2, caseSensitivity());
 
    d->ensureInitialized();
 
    QVarLengthArray<wchar_t> array1, array2;
    stringToWCharArray(array1, s1);
    stringToWCharArray(array2, s2);
    return std::wcscoll(array1.constData(), array2.constData());
}
 
QCollatorSortKey QCollator::sortKey(const QString &string) const
{
    if (!d)
        d = new QCollatorPrivate(QLocale().collation());
 
    d->ensureInitialized();
 
    QVarLengthArray<wchar_t> original;
    stringToWCharArray(original, string);
    QList<wchar_t> result(original.size());
    if (d->isC()) {
        std::copy(original.cbegin(), original.cend(), result.begin());
    } else {
        auto availableSizeIncludingNullTerminator = result.size();
        size_t neededSizeExcludingNullTerminator = std::wcsxfrm(
                result.data(), original.constData(), availableSizeIncludingNullTerminator);
        if (neededSizeExcludingNullTerminator > size_t(availableSizeIncludingNullTerminator - 1)) {
            result.resize(neededSizeExcludingNullTerminator + 1);
            availableSizeIncludingNullTerminator = result.size();
            neededSizeExcludingNullTerminator = std::wcsxfrm(result.data(), original.constData(),
                                                             availableSizeIncludingNullTerminator);
            Q_ASSERT(neededSizeExcludingNullTerminator
                     == size_t(availableSizeIncludingNullTerminator - 1));
        }
        result.resize(neededSizeExcludingNullTerminator + 1);
        result[neededSizeExcludingNullTerminator] = 0;
    }
    return QCollatorPrivate::sortKeyFromData(std::move(result));
}
 
int QCollatorSortKey::compare(const QCollatorSortKey &otherKey) const noexcept
{
    return std::wcscmp(d->m_key.constData(), otherKey.d->m_key.constData());
}
 
QT_END_NAMESPACE