Chromium Code Reviews| Index: src/uri.cc |
| diff --git a/src/uri.cc b/src/uri.cc |
| index c459be5e53f4ad915d669a0122cec77b1d1889e2..3924bcc3f0520cbca6b32e2d26c5c835fbe8d8c7 100644 |
| --- a/src/uri.cc |
| +++ b/src/uri.cc |
| @@ -60,36 +60,29 @@ void AddHexEncodedToBuffer(uint8_t octet, List<uint8_t>* buffer) { |
| } |
| void EncodeSingle(uc16 c, List<uint8_t>* buffer) { |
| - uint8_t x = (c >> 12) & 0xF; |
| - uint8_t y = (c >> 6) & 63; |
| - uint8_t z = c & 63; |
| - if (c <= 0x007F) { |
| - AddHexEncodedToBuffer(c, buffer); |
| - } else if (c <= 0x07FF) { |
| - AddHexEncodedToBuffer(y + 192, buffer); |
| - AddHexEncodedToBuffer(z + 128, buffer); |
| - } else { |
| - AddHexEncodedToBuffer(x + 224, buffer); |
| - AddHexEncodedToBuffer(y + 128, buffer); |
| - AddHexEncodedToBuffer(z + 128, buffer); |
| + char s[4]; |
| + int number_of_bytes; |
| + number_of_bytes = |
| + unibrow::Utf8::Encode(s, c, unibrow::Utf16::kNoPreviousCharacter, false); |
| + for (int k = 0; k < number_of_bytes; k++) { |
| + AddHexEncodedToBuffer(s[k], buffer); |
| } |
| } |
| void EncodePair(uc16 cc1, uc16 cc2, List<uint8_t>* buffer) { |
| - uint8_t u = ((cc1 >> 6) & 0xF) + 1; |
| - uint8_t w = (cc1 >> 2) & 0xF; |
| - uint8_t x = cc1 & 3; |
| - uint8_t y = (cc2 >> 6) & 0xF; |
| - uint8_t z = cc2 & 63; |
| - AddHexEncodedToBuffer((u >> 2) + 240, buffer); |
| - AddHexEncodedToBuffer((((u & 3) << 4) | w) + 128, buffer); |
| - AddHexEncodedToBuffer(((x << 4) | y) + 128, buffer); |
| - AddHexEncodedToBuffer(z + 128, buffer); |
| + char s[4]; |
| + int number_of_bytes = |
| + unibrow::Utf8::Encode(s, unibrow::Utf16::CombineSurrogatePair(cc1, cc2), |
| + unibrow::Utf16::kNoPreviousCharacter, false); |
| + for (int k = 0; k < number_of_bytes; k++) { |
| + AddHexEncodedToBuffer(s[k], buffer); |
| + } |
| } |
| } // anonymous namespace |
| -Object* Uri::Encode(Isolate* isolate, Handle<String> uri, bool is_uri) { |
| +MaybeHandle<Object> Uri::Encode(Isolate* isolate, Handle<String> uri, |
| + bool is_uri) { |
| uri = String::Flatten(uri); |
| int uri_length = uri->length(); |
| List<uint8_t> buffer(uri_length); |
| @@ -120,15 +113,216 @@ Object* Uri::Encode(Isolate* isolate, Handle<String> uri, bool is_uri) { |
| } |
| AllowHeapAllocation allocate_error_and_return; |
| - THROW_NEW_ERROR_RETURN_FAILURE(isolate, NewURIError()); |
| + THROW_NEW_ERROR(isolate, NewURIError(), Object); |
| } |
| } |
| Handle<String> result; |
| - ASSIGN_RETURN_FAILURE_ON_EXCEPTION( |
| + ASSIGN_RETURN_ON_EXCEPTION( |
| isolate, result, |
| - isolate->factory()->NewStringFromOneByte(buffer.ToConstVector())); |
| - return *result; |
| + isolate->factory()->NewStringFromOneByte(buffer.ToConstVector()), Object); |
| + return result; |
| +} |
| + |
| +namespace { // anonymous namespace for DecodeURI helper functions |
| + |
| +bool IsReservedPredicate(uc16 c) { |
| + switch (c) { |
| + case '#': |
| + case '$': |
| + case '&': |
| + case '+': |
| + case ',': |
| + case '/': |
| + case ':': |
| + case ';': |
| + case '=': |
| + case '?': |
| + case '@': |
| + return true; |
| + default: |
| + return false; |
| + } |
| +} |
| + |
| +bool IsReplacementCharacter(const uint8_t* octets, int length) { |
| + // The replacement character is at codepoint U+FFFD in the Unicode Specials |
| + // table. Its UTF-8 encoding is 0xEF 0xBF 0xBD. |
| + if (length != 3 || octets[0] != 0xef || octets[1] != 0xbf || |
| + octets[2] != 0xbd) { |
| + return false; |
| + } |
| + return true; |
| +} |
| + |
| +bool DecodeOctets(const uint8_t* octets, int length, |
| + List<uc16>* two_byte_buffer) { |
| + size_t cursor = 0; |
| + uc32 value = unibrow::Utf8::ValueOf(octets, length, &cursor); |
| + if (value == unibrow::Utf8::kBadChar && |
| + !IsReplacementCharacter(octets, length)) { |
| + return false; |
| + } |
| + |
| + if (value <= unibrow::Utf16::kMaxNonSurrogateCharCode) { |
| + two_byte_buffer->Add(value); |
| + } else { |
| + two_byte_buffer->Add(unibrow::Utf16::LeadSurrogate(value)); |
| + two_byte_buffer->Add(unibrow::Utf16::TrailSurrogate(value)); |
| + } |
| + return true; |
| +} |
| + |
| +bool TwoDigitHex(uc16* decoded, int index, String::FlatContent* uri_content) { |
| + char high = HexValue(uri_content->Get(index + 1)); |
| + char low = HexValue(uri_content->Get(index + 2)); |
| + if (high < 0 || low < 0) { |
| + return false; |
| + } |
| + *decoded = (high << 4) | low; |
| + return true; |
| +} |
| + |
| +template <typename T> |
| +bool AddToBuffer(uc16 decoded, String::FlatContent* uri_content, int index, |
| + bool is_uri, List<T>* buffer) { |
| + if (is_uri && IsReservedPredicate(decoded)) { |
| + buffer->Add('%'); |
| + uc16 first = uri_content->Get(index + 1); |
| + uc16 second = uri_content->Get(index + 2); |
| + if (first > std::numeric_limits<T>::max() || |
|
Yang
2016/05/23 11:24:59
Do you expect this to happen in some cases? Imo th
Franzi
2016/05/24 15:07:38
You're right. Can never happen. Done.
|
| + second > std::numeric_limits<T>::max()) { |
| + return false; |
| + } |
| + buffer->Add(first); |
| + buffer->Add(second); |
| + } else { |
| + buffer->Add(decoded); |
| + } |
| + return true; |
| +} |
| + |
| +bool IntoTwoByte(int index, bool is_uri, int uri_length, |
| + String::FlatContent* uri_content, |
| + List<uc16>* two_byte_buffer) { |
| + for (int k = index; k < uri_length; k++) { |
| + uc16 code = uri_content->Get(k); |
| + if (code == '%') { |
| + uc16 decoded; |
| + if (k + 2 >= uri_length || !TwoDigitHex(&decoded, k, uri_content)) { |
| + return false; |
| + } |
| + k += 2; |
| + if (decoded > unibrow::Utf8::kMaxOneByteChar) { |
| + int n = 1; |
| + while ((decoded << n) & 0x80) { |
|
Yang
2016/05/23 11:24:59
Actually, you could merge this while loop with the
Franzi
2016/05/24 15:07:38
Merged. Also renamed n to number_of_continuation_b
|
| + n++; |
| + } |
| + if (n == 1 || n > 4 || k + 3 * (n - 1) >= uri_length) { |
| + return false; |
| + } |
| + uint8_t octets[4]; |
| + int octet_length = 0; |
| + octets[octet_length++] = decoded; |
| + |
| + for (int i = 1; i < n; i++) { |
| + uc16 decodedTrail; |
| + |
| + if (uri_content->Get(++k) != '%' || k + 2 >= uri_length || |
| + !TwoDigitHex(&decodedTrail, k, uri_content)) { |
| + return false; |
| + } |
| + k += 2; |
| + octets[octet_length++] = decodedTrail; |
| + } |
| + |
| + if (!DecodeOctets(octets, octet_length, two_byte_buffer)) { |
| + return false; |
| + } |
| + } else { |
| + if (!AddToBuffer(decoded, uri_content, k - 2, is_uri, |
| + two_byte_buffer)) { |
| + return false; |
| + } |
| + } |
| + } else { |
| + two_byte_buffer->Add(code); |
| + } |
| + } |
| + return true; |
| +} |
| + |
| +bool IntoOneAndTwoByte(Handle<String> uri, bool is_uri, |
| + List<uint8_t>* one_byte_buffer, |
| + List<uc16>* two_byte_buffer) { |
| + DisallowHeapAllocation no_gc; |
| + String::FlatContent uri_content = uri->GetFlatContent(); |
| + |
| + int uri_length = uri->length(); |
| + for (int k = 0; k < uri_length; k++) { |
| + uc16 code = uri_content.Get(k); |
| + if (code == '%') { |
| + uc16 decoded; |
| + if (k + 2 >= uri_length || !TwoDigitHex(&decoded, k, &uri_content)) { |
| + return false; |
| + } |
| + |
| + if (decoded > unibrow::Utf8::kMaxOneByteChar) { |
| + return IntoTwoByte(k, is_uri, uri_length, &uri_content, |
| + two_byte_buffer); |
| + } |
| + |
| + if (!AddToBuffer(decoded, &uri_content, k, is_uri, one_byte_buffer)) { |
| + return false; |
| + } |
| + k += 2; |
| + } else { |
| + if (code > unibrow::Utf8::kMaxOneByteChar) { |
| + return IntoTwoByte(k, is_uri, uri_length, &uri_content, |
| + two_byte_buffer); |
| + } |
| + one_byte_buffer->Add(code); |
| + } |
| + } |
| + return true; |
| +} |
| + |
| +} // anonymous namespace |
| + |
| +MaybeHandle<Object> Uri::Decode(Isolate* isolate, Handle<String> uri, |
| + bool is_uri) { |
| + uri = String::Flatten(uri); |
| + List<uint8_t> one_byte_buffer; |
| + List<uc16> two_byte_buffer; |
| + |
| + if (!IntoOneAndTwoByte(uri, is_uri, &one_byte_buffer, &two_byte_buffer)) { |
| + THROW_NEW_ERROR(isolate, NewURIError(), Object); |
| + } |
| + |
| + if (two_byte_buffer.is_empty()) { |
| + Handle<SeqOneByteString> result; |
| + |
| + ASSIGN_RETURN_ON_EXCEPTION( |
| + isolate, result, |
| + isolate->factory()->NewRawOneByteString(one_byte_buffer.length()), |
|
Yang
2016/05/23 11:24:59
You can use NewStringFromOneByte here.
Franzi
2016/05/24 15:07:38
Done.
|
| + Object); |
| + CopyChars(result->GetChars(), one_byte_buffer.ToConstVector().start(), |
| + one_byte_buffer.length()); |
| + return result; |
| + } |
| + |
| + Handle<SeqTwoByteString> result; |
| + ASSIGN_RETURN_ON_EXCEPTION( |
| + isolate, result, isolate->factory()->NewRawTwoByteString( |
| + one_byte_buffer.length() + two_byte_buffer.length()), |
| + Object); |
| + |
| + CopyChars(result->GetChars(), one_byte_buffer.ToConstVector().start(), |
| + one_byte_buffer.length()); |
| + CopyChars(result->GetChars() + one_byte_buffer.length(), |
| + two_byte_buffer.ToConstVector().start(), two_byte_buffer.length()); |
| + |
| + return result; |
| } |
| } // namespace internal |