HeaderFieldTokenizer.cpp [plain text]
#include "config.h"
#include "HeaderFieldTokenizer.h"
#include "HTTPHeaderField.h"
#include <wtf/text/StringBuilder.h>
namespace WebCore {
HeaderFieldTokenizer::HeaderFieldTokenizer(const String& headerField)
: m_input(headerField)
{
skipSpaces();
}
bool HeaderFieldTokenizer::consume(UChar c)
{
ASSERT(c != ' ' && c != '\t');
if (isConsumed() || m_input[m_index] != c)
return false;
++m_index;
skipSpaces();
return true;
}
String HeaderFieldTokenizer::consumeQuotedString()
{
StringBuilder builder;
ASSERT(m_input[m_index] == '"');
++m_index;
while (!isConsumed()) {
if (m_input[m_index] == '"') {
String output = builder.toString();
++m_index;
skipSpaces();
return output;
}
if (m_input[m_index] == '\\') {
++m_index;
if (isConsumed())
return String();
}
builder.append(m_input[m_index]);
++m_index;
}
return String();
}
String HeaderFieldTokenizer::consumeToken()
{
auto start = m_index;
while (!isConsumed() && RFC7230::isTokenCharacter(m_input[m_index]))
++m_index;
if (start == m_index)
return String();
String output = m_input.substring(start, m_index - start);
skipSpaces();
return output;
}
String HeaderFieldTokenizer::consumeTokenOrQuotedString()
{
if (isConsumed())
return String();
if (m_input[m_index] == '"')
return consumeQuotedString();
return consumeToken();
}
void HeaderFieldTokenizer::skipSpaces()
{
while (!isConsumed() && RFC7230::isWhitespace(m_input[m_index]))
++m_index;
}
void HeaderFieldTokenizer::consumeBeforeAnyCharMatch(const Vector<UChar>& chars)
{
ASSERT(chars.size() > 0U && chars.size() < 3U);
while (!isConsumed()) {
for (const auto& c : chars) {
if (c == m_input[m_index])
return;
}
++m_index;
}
}
}