diff options
Diffstat (limited to 'src/network/access/qhttpheaderparser.cpp')
-rw-r--r-- | src/network/access/qhttpheaderparser.cpp | 80 |
1 files changed, 32 insertions, 48 deletions
diff --git a/src/network/access/qhttpheaderparser.cpp b/src/network/access/qhttpheaderparser.cpp index 345f548c51..0b7882c18a 100644 --- a/src/network/access/qhttpheaderparser.cpp +++ b/src/network/access/qhttpheaderparser.cpp @@ -1,4 +1,4 @@ -// Copyright (C) 2021 The Qt Company Ltd. +// Copyright (C) 2022 The Qt Company Ltd. // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only #include "qhttpheaderparser_p.h" @@ -7,12 +7,6 @@ QT_BEGIN_NAMESPACE -// both constants are taken from the default settings of Apache -// see: http://httpd.apache.org/docs/2.2/mod/core.html#limitrequestfieldsize and -// http://httpd.apache.org/docs/2.2/mod/core.html#limitrequestfields -static const int MAX_HEADER_FIELD_SIZE = 8 * 1024; -static const int MAX_HEADER_FIELDS = 100; - QHttpHeaderParser::QHttpHeaderParser() : statusCode(100) // Required by tst_QHttpNetworkConnection::ignoresslerror(failure) , majorVersion(0) @@ -37,7 +31,7 @@ static bool fieldNameCheck(QByteArrayView name) || otherCharacters.contains(c); }; - return name.size() > 0 && std::all_of(name.begin(), name.end(), fieldNameChar); + return !name.empty() && std::all_of(name.begin(), name.end(), fieldNameChar); } bool QHttpHeaderParser::parseHeaders(QByteArrayView header) @@ -49,43 +43,46 @@ bool QHttpHeaderParser::parseHeaders(QByteArrayView header) return h.startsWith(' ') || h.startsWith('\t'); }; // Headers, if non-empty, start with a non-space and end with a newline: - if (hSpaceStart(header) || (header.size() && !header.endsWith('\n'))) + if (hSpaceStart(header) || (!header.empty() && !header.endsWith('\n'))) return false; while (int tail = header.endsWith("\n\r\n") ? 2 : header.endsWith("\n\n") ? 1 : 0) header.chop(tail); - QList<QPair<QByteArray, QByteArray>> result; - while (header.size()) { - const int colon = header.indexOf(':'); + if (header.size() - (header.endsWith("\r\n") ? 2 : 1) > maxTotalSize) + return false; + + QHttpHeaders result; + while (!header.empty()) { + const qsizetype colon = header.indexOf(':'); if (colon == -1) // if no colon check if empty headers - return result.size() == 0 && (header == "\n" || header == "\r\n"); - if (result.size() >= MAX_HEADER_FIELDS) + return result.isEmpty() && (header == "\n" || header == "\r\n"); + if (result.size() >= maxFieldCount) return false; QByteArrayView name = header.first(colon); if (!fieldNameCheck(name)) return false; header = header.sliced(colon + 1); QByteArray value; - int valueSpace = MAX_HEADER_FIELD_SIZE - name.size() - 1; + qsizetype valueSpace = maxFieldSize - name.size() - 1; do { - const int endLine = header.indexOf('\n'); + const qsizetype endLine = header.indexOf('\n'); Q_ASSERT(endLine != -1); auto line = header.first(endLine); // includes space valueSpace -= line.size() - (line.endsWith('\r') ? 1 : 0); if (valueSpace < 0) return false; line = line.trimmed(); - if (line.size()) { + if (!line.empty()) { if (value.size()) - value += ' ' + line.toByteArray(); + value += ' ' + line; else value = line.toByteArray(); } header = header.sliced(endLine + 1); } while (hSpaceStart(header)); - Q_ASSERT(name.size() + 1 + value.size() <= MAX_HEADER_FIELD_SIZE); - result.append(qMakePair(name.toByteArray(), value)); + Q_ASSERT(name.size() + 1 + value.size() <= maxFieldSize); + result.append(name, value); } fields = result; @@ -105,7 +102,7 @@ bool QHttpHeaderParser::parseStatus(QByteArrayView status) static const int spacePos = 8; static const char httpMagic[] = "HTTP/"; - if (status.length() < minLength + if (status.size() < minLength || !status.startsWith(httpMagic) || status.at(dotPos) != '.' || status.at(spacePos) != ' ') { @@ -118,11 +115,11 @@ bool QHttpHeaderParser::parseStatus(QByteArrayView status) minorVersion = status.at(dotPos + 1) - '0'; int i = spacePos; - int j = status.indexOf(' ', i + 1); + qsizetype j = status.indexOf(' ', i + 1); const QByteArrayView code = j > i ? status.sliced(i + 1, j - i - 1) : status.sliced(i + 1); - bool ok; + bool ok = false; statusCode = code.toInt(&ok); reasonPhrase = j > i ? QString::fromLatin1(status.sliced(j + 1)) @@ -131,62 +128,49 @@ bool QHttpHeaderParser::parseStatus(QByteArrayView status) return ok && uint(majorVersion) <= 9 && uint(minorVersion) <= 9; } -const QList<QPair<QByteArray, QByteArray> >& QHttpHeaderParser::headers() const +const QHttpHeaders& QHttpHeaderParser::headers() const { return fields; } -QByteArray QHttpHeaderParser::firstHeaderField(const QByteArray &name, +QByteArray QHttpHeaderParser::firstHeaderField(QByteArrayView name, const QByteArray &defaultValue) const { - for (auto it = fields.constBegin(); it != fields.constEnd(); ++it) { - if (name.compare(it->first, Qt::CaseInsensitive) == 0) - return it->second; - } - return defaultValue; + return fields.value(name, defaultValue).toByteArray(); } -QByteArray QHttpHeaderParser::combinedHeaderValue(const QByteArray &name, const QByteArray &defaultValue) const +QByteArray QHttpHeaderParser::combinedHeaderValue(QByteArrayView name, const QByteArray &defaultValue) const { const QList<QByteArray> allValues = headerFieldValues(name); if (allValues.isEmpty()) return defaultValue; - else - return allValues.join(", "); + return allValues.join(", "); } -QList<QByteArray> QHttpHeaderParser::headerFieldValues(const QByteArray &name) const +QList<QByteArray> QHttpHeaderParser::headerFieldValues(QByteArrayView name) const { - QList<QByteArray> result; - for (auto it = fields.constBegin(); it != fields.constEnd(); ++it) - if (name.compare(it->first, Qt::CaseInsensitive) == 0) - result += it->second; - - return result; + return fields.values(name); } -void QHttpHeaderParser::removeHeaderField(const QByteArray &name) +void QHttpHeaderParser::removeHeaderField(QByteArrayView name) { - auto firstEqualsName = [&name](const QPair<QByteArray, QByteArray> &header) { - return name.compare(header.first, Qt::CaseInsensitive) == 0; - }; - fields.removeIf(firstEqualsName); + fields.removeAll(name); } void QHttpHeaderParser::setHeaderField(const QByteArray &name, const QByteArray &data) { removeHeaderField(name); - fields.append(qMakePair(name, data)); + fields.append(name, data); } void QHttpHeaderParser::prependHeaderField(const QByteArray &name, const QByteArray &data) { - fields.prepend(qMakePair(name, data)); + fields.insert(0, name, data); } void QHttpHeaderParser::appendHeaderField(const QByteArray &name, const QByteArray &data) { - fields.append(qMakePair(name, data)); + fields.append(name, data); } void QHttpHeaderParser::clearHeaders() |