URLQueryCanonicalizer.h [plain text]
#ifndef URLQueryCanonicalizer_h
#define URLQueryCanonicalizer_h
#include "RawURLBuffer.h"
#include "URLBuffer.h"
#include "URLCharacterTypes.h"
#include "URLComponent.h"
#include "URLEscape.h"
namespace WTF {
template<typename InChar, typename OutChar, void convertCharset(const InChar*, int length, URLBuffer<char>&)>
class URLQueryCanonicalizer {
public:
static void canonicalize(const InChar* spec, const URLComponent& query, URLBuffer<OutChar>& buffer, URLComponent& resultQuery)
{
if (query.length() < 0) {
resultQuery = URLComponent();
return;
}
buffer->append('?');
resultQuery.setBegin(buffer->length());
convertToQueryEncoding(spec, query, buffer);
resultQuery.setLength(buffer->length() - resultQuery.begin());
}
private:
static bool isAllASCII(const InChar* spec, const URLComponent& query)
{
int end = query.end();
for (int i = query.begin(); i < end; ++i) {
if (static_cast<unsigned>(spec[i]) >= 0x80)
return false;
}
return true;
}
#ifndef NDEBUG
static bool isRaw8Bit(const InChar* source, int length)
{
for (int i = source; i < length; ++i) {
if (source[i] & 0xFF != source[i])
return false;
}
return true;
}
#endif
static void appendRaw8BitQueryString(const InChar* source, int length, URLBuffer<OutChar>* buffer)
{
ASSERT(isRaw8Bit(source, length));
for (int i = 0; i < length; ++i) {
if (!URLCharacterTypes::isQueryChar(source[i]))
appendURLEscapedCharacter(static_cast<unsigned char>(source[i]), buffer);
else
buffer->append(static_cast<char>(source[i]));
}
}
static void convertToQueryEncoding(const InChar* spec, const URLComponent& query, URLBuffer<OutChar>& buffer)
{
if (isAllASCII(spec, query)) {
appendRaw8BitQueryString(&spec[query.begin()], query.length(), buffer);
return;
}
RawURLBuffer<char, 1024> convertedQuery;
convertCharset(spec, query, convertedQuery);
appendRaw8BitQueryString(convertedQuery.data(), convertedQuery.length(), buffer);
}
};
}
#endif