| OLD | NEW |
| 1 // Copyright 2011, Google Inc. | 1 // Copyright 2011, Google Inc. |
| 2 // All rights reserved. | 2 // All rights reserved. |
| 3 // | 3 // |
| 4 // Redistribution and use in source and binary forms, with or without | 4 // Redistribution and use in source and binary forms, with or without |
| 5 // modification, are permitted provided that the following conditions are | 5 // modification, are permitted provided that the following conditions are |
| 6 // met: | 6 // met: |
| 7 // | 7 // |
| 8 // * Redistributions of source code must retain the above copyright | 8 // * Redistributions of source code must retain the above copyright |
| 9 // notice, this list of conditions and the following disclaimer. | 9 // notice, this list of conditions and the following disclaimer. |
| 10 // * Redistributions in binary form must reproduce the above | 10 // * Redistributions in binary form must reproduce the above |
| (...skipping 17 matching lines...) Expand all Loading... |
| 28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | 28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 29 | 29 |
| 30 // This file is intended to be included in another C++ file where the character | 30 // This file is intended to be included in another C++ file where the character |
| 31 // types are defined. This allows us to write mostly generic code, but not have | 31 // types are defined. This allows us to write mostly generic code, but not have |
| 32 // templace bloat because everything is inlined when anybody calls any of our | 32 // templace bloat because everything is inlined when anybody calls any of our |
| 33 // functions. | 33 // functions. |
| 34 | 34 |
| 35 #ifndef GOOGLEURL_SRC_URL_CANON_INTERNAL_H__ | 35 #ifndef GOOGLEURL_SRC_URL_CANON_INTERNAL_H__ |
| 36 #define GOOGLEURL_SRC_URL_CANON_INTERNAL_H__ | 36 #define GOOGLEURL_SRC_URL_CANON_INTERNAL_H__ |
| 37 | 37 |
| 38 #include <stdlib.h> | 38 #include "url/url_canon_internal.h" |
| 39 | |
| 40 #include "base/logging.h" | |
| 41 #include "googleurl/src/url_canon.h" | |
| 42 | |
| 43 namespace url_canon { | |
| 44 | |
| 45 // Character type handling ----------------------------------------------------- | |
| 46 | |
| 47 // Bits that identify different character types. These types identify different | |
| 48 // bits that are set for each 8-bit character in the kSharedCharTypeTable. | |
| 49 enum SharedCharTypes { | |
| 50 // Characters that do not require escaping in queries. Characters that do | |
| 51 // not have this flag will be escaped; see url_canon_query.cc | |
| 52 CHAR_QUERY = 1, | |
| 53 | |
| 54 // Valid in the username/password field. | |
| 55 CHAR_USERINFO = 2, | |
| 56 | |
| 57 // Valid in a IPv4 address (digits plus dot and 'x' for hex). | |
| 58 CHAR_IPV4 = 4, | |
| 59 | |
| 60 // Valid in an ASCII-representation of a hex digit (as in %-escaped). | |
| 61 CHAR_HEX = 8, | |
| 62 | |
| 63 // Valid in an ASCII-representation of a decimal digit. | |
| 64 CHAR_DEC = 16, | |
| 65 | |
| 66 // Valid in an ASCII-representation of an octal digit. | |
| 67 CHAR_OCT = 32, | |
| 68 | |
| 69 // Characters that do not require escaping in encodeURIComponent. Characters | |
| 70 // that do not have this flag will be escaped; see url_util.cc. | |
| 71 CHAR_COMPONENT = 64, | |
| 72 }; | |
| 73 | |
| 74 // This table contains the flags in SharedCharTypes for each 8-bit character. | |
| 75 // Some canonicalization functions have their own specialized lookup table. | |
| 76 // For those with simple requirements, we have collected the flags in one | |
| 77 // place so there are fewer lookup tables to load into the CPU cache. | |
| 78 // | |
| 79 // Using an unsigned char type has a small but measurable performance benefit | |
| 80 // over using a 32-bit number. | |
| 81 extern const unsigned char kSharedCharTypeTable[0x100]; | |
| 82 | |
| 83 // More readable wrappers around the character type lookup table. | |
| 84 inline bool IsCharOfType(unsigned char c, SharedCharTypes type) { | |
| 85 return !!(kSharedCharTypeTable[c] & type); | |
| 86 } | |
| 87 inline bool IsQueryChar(unsigned char c) { | |
| 88 return IsCharOfType(c, CHAR_QUERY); | |
| 89 } | |
| 90 inline bool IsIPv4Char(unsigned char c) { | |
| 91 return IsCharOfType(c, CHAR_IPV4); | |
| 92 } | |
| 93 inline bool IsHexChar(unsigned char c) { | |
| 94 return IsCharOfType(c, CHAR_HEX); | |
| 95 } | |
| 96 inline bool IsComponentChar(unsigned char c) { | |
| 97 return IsCharOfType(c, CHAR_COMPONENT); | |
| 98 } | |
| 99 | |
| 100 // Appends the given string to the output, escaping characters that do not | |
| 101 // match the given |type| in SharedCharTypes. | |
| 102 void AppendStringOfType(const char* source, int length, | |
| 103 SharedCharTypes type, | |
| 104 CanonOutput* output); | |
| 105 void AppendStringOfType(const char16* source, int length, | |
| 106 SharedCharTypes type, | |
| 107 CanonOutput* output); | |
| 108 | |
| 109 // Maps the hex numerical values 0x0 to 0xf to the corresponding ASCII digit | |
| 110 // that will be used to represent it. | |
| 111 GURL_API extern const char kHexCharLookup[0x10]; | |
| 112 | |
| 113 // This lookup table allows fast conversion between ASCII hex letters and their | |
| 114 // corresponding numerical value. The 8-bit range is divided up into 8 | |
| 115 // regions of 0x20 characters each. Each of the three character types (numbers, | |
| 116 // uppercase, lowercase) falls into different regions of this range. The table | |
| 117 // contains the amount to subtract from characters in that range to get at | |
| 118 // the corresponding numerical value. | |
| 119 // | |
| 120 // See HexDigitToValue for the lookup. | |
| 121 extern const char kCharToHexLookup[8]; | |
| 122 | |
| 123 // Assumes the input is a valid hex digit! Call IsHexChar before using this. | |
| 124 inline unsigned char HexCharToValue(unsigned char c) { | |
| 125 return c - kCharToHexLookup[c / 0x20]; | |
| 126 } | |
| 127 | |
| 128 // Indicates if the given character is a dot or dot equivalent, returning the | |
| 129 // number of characters taken by it. This will be one for a literal dot, 3 for | |
| 130 // an escaped dot. If the character is not a dot, this will return 0. | |
| 131 template<typename CHAR> | |
| 132 inline int IsDot(const CHAR* spec, int offset, int end) { | |
| 133 if (spec[offset] == '.') { | |
| 134 return 1; | |
| 135 } else if (spec[offset] == '%' && offset + 3 <= end && | |
| 136 spec[offset + 1] == '2' && | |
| 137 (spec[offset + 2] == 'e' || spec[offset + 2] == 'E')) { | |
| 138 // Found "%2e" | |
| 139 return 3; | |
| 140 } | |
| 141 return 0; | |
| 142 } | |
| 143 | |
| 144 // Returns the canonicalized version of the input character according to scheme | |
| 145 // rules. This is implemented alongside the scheme canonicalizer, and is | |
| 146 // required for relative URL resolving to test for scheme equality. | |
| 147 // | |
| 148 // Returns 0 if the input character is not a valid scheme character. | |
| 149 char CanonicalSchemeChar(char16 ch); | |
| 150 | |
| 151 // Write a single character, escaped, to the output. This always escapes: it | |
| 152 // does no checking that thee character requires escaping. | |
| 153 // Escaping makes sense only 8 bit chars, so code works in all cases of | |
| 154 // input parameters (8/16bit). | |
| 155 template<typename UINCHAR, typename OUTCHAR> | |
| 156 inline void AppendEscapedChar(UINCHAR ch, | |
| 157 CanonOutputT<OUTCHAR>* output) { | |
| 158 output->push_back('%'); | |
| 159 output->push_back(kHexCharLookup[(ch >> 4) & 0xf]); | |
| 160 output->push_back(kHexCharLookup[ch & 0xf]); | |
| 161 } | |
| 162 | |
| 163 // The character we'll substitute for undecodable or invalid characters. | |
| 164 extern const char16 kUnicodeReplacementCharacter; | |
| 165 | |
| 166 // UTF-8 functions ------------------------------------------------------------ | |
| 167 | |
| 168 // Reads one character in UTF-8 starting at |*begin| in |str| and places | |
| 169 // the decoded value into |*code_point|. If the character is valid, we will | |
| 170 // return true. If invalid, we'll return false and put the | |
| 171 // kUnicodeReplacementCharacter into |*code_point|. | |
| 172 // | |
| 173 // |*begin| will be updated to point to the last character consumed so it | |
| 174 // can be incremented in a loop and will be ready for the next character. | |
| 175 // (for a single-byte ASCII character, it will not be changed). | |
| 176 // | |
| 177 // Implementation is in url_canon_icu.cc. | |
| 178 GURL_API bool ReadUTFChar(const char* str, int* begin, int length, | |
| 179 unsigned* code_point_out); | |
| 180 | |
| 181 // Generic To-UTF-8 converter. This will call the given append method for each | |
| 182 // character that should be appended, with the given output method. Wrappers | |
| 183 // are provided below for escaped and non-escaped versions of this. | |
| 184 // | |
| 185 // The char_value must have already been checked that it's a valid Unicode | |
| 186 // character. | |
| 187 template<class Output, void Appender(unsigned char, Output*)> | |
| 188 inline void DoAppendUTF8(unsigned char_value, Output* output) { | |
| 189 if (char_value <= 0x7f) { | |
| 190 Appender(static_cast<unsigned char>(char_value), output); | |
| 191 } else if (char_value <= 0x7ff) { | |
| 192 // 110xxxxx 10xxxxxx | |
| 193 Appender(static_cast<unsigned char>(0xC0 | (char_value >> 6)), | |
| 194 output); | |
| 195 Appender(static_cast<unsigned char>(0x80 | (char_value & 0x3f)), | |
| 196 output); | |
| 197 } else if (char_value <= 0xffff) { | |
| 198 // 1110xxxx 10xxxxxx 10xxxxxx | |
| 199 Appender(static_cast<unsigned char>(0xe0 | (char_value >> 12)), | |
| 200 output); | |
| 201 Appender(static_cast<unsigned char>(0x80 | ((char_value >> 6) & 0x3f)), | |
| 202 output); | |
| 203 Appender(static_cast<unsigned char>(0x80 | (char_value & 0x3f)), | |
| 204 output); | |
| 205 } else if (char_value <= 0x10FFFF) { // Max unicode code point. | |
| 206 // 11110xxx 10xxxxxx 10xxxxxx 10xxxxxx | |
| 207 Appender(static_cast<unsigned char>(0xf0 | (char_value >> 18)), | |
| 208 output); | |
| 209 Appender(static_cast<unsigned char>(0x80 | ((char_value >> 12) & 0x3f)), | |
| 210 output); | |
| 211 Appender(static_cast<unsigned char>(0x80 | ((char_value >> 6) & 0x3f)), | |
| 212 output); | |
| 213 Appender(static_cast<unsigned char>(0x80 | (char_value & 0x3f)), | |
| 214 output); | |
| 215 } else { | |
| 216 // Invalid UTF-8 character (>20 bits). | |
| 217 NOTREACHED(); | |
| 218 } | |
| 219 } | |
| 220 | |
| 221 // Helper used by AppendUTF8Value below. We use an unsigned parameter so there | |
| 222 // are no funny sign problems with the input, but then have to convert it to | |
| 223 // a regular char for appending. | |
| 224 inline void AppendCharToOutput(unsigned char ch, CanonOutput* output) { | |
| 225 output->push_back(static_cast<char>(ch)); | |
| 226 } | |
| 227 | |
| 228 // Writes the given character to the output as UTF-8. This does NO checking | |
| 229 // of the validity of the unicode characters; the caller should ensure that | |
| 230 // the value it is appending is valid to append. | |
| 231 inline void AppendUTF8Value(unsigned char_value, CanonOutput* output) { | |
| 232 DoAppendUTF8<CanonOutput, AppendCharToOutput>(char_value, output); | |
| 233 } | |
| 234 | |
| 235 // Writes the given character to the output as UTF-8, escaping ALL | |
| 236 // characters (even when they are ASCII). This does NO checking of the | |
| 237 // validity of the unicode characters; the caller should ensure that the value | |
| 238 // it is appending is valid to append. | |
| 239 inline void AppendUTF8EscapedValue(unsigned char_value, CanonOutput* output) { | |
| 240 DoAppendUTF8<CanonOutput, AppendEscapedChar>(char_value, output); | |
| 241 } | |
| 242 | |
| 243 // UTF-16 functions ----------------------------------------------------------- | |
| 244 | |
| 245 // Reads one character in UTF-16 starting at |*begin| in |str| and places | |
| 246 // the decoded value into |*code_point|. If the character is valid, we will | |
| 247 // return true. If invalid, we'll return false and put the | |
| 248 // kUnicodeReplacementCharacter into |*code_point|. | |
| 249 // | |
| 250 // |*begin| will be updated to point to the last character consumed so it | |
| 251 // can be incremented in a loop and will be ready for the next character. | |
| 252 // (for a single-16-bit-word character, it will not be changed). | |
| 253 // | |
| 254 // Implementation is in url_canon_icu.cc. | |
| 255 GURL_API bool ReadUTFChar(const char16* str, int* begin, int length, | |
| 256 unsigned* code_point); | |
| 257 | |
| 258 // Equivalent to U16_APPEND_UNSAFE in ICU but uses our output method. | |
| 259 inline void AppendUTF16Value(unsigned code_point, | |
| 260 CanonOutputT<char16>* output) { | |
| 261 if (code_point > 0xffff) { | |
| 262 output->push_back(static_cast<char16>((code_point >> 10) + 0xd7c0)); | |
| 263 output->push_back(static_cast<char16>((code_point & 0x3ff) | 0xdc00)); | |
| 264 } else { | |
| 265 output->push_back(static_cast<char16>(code_point)); | |
| 266 } | |
| 267 } | |
| 268 | |
| 269 // Escaping functions --------------------------------------------------------- | |
| 270 | |
| 271 // Writes the given character to the output as UTF-8, escaped. Call this | |
| 272 // function only when the input is wide. Returns true on success. Failure | |
| 273 // means there was some problem with the encoding, we'll still try to | |
| 274 // update the |*begin| pointer and add a placeholder character to the | |
| 275 // output so processing can continue. | |
| 276 // | |
| 277 // We will append the character starting at ch[begin] with the buffer ch | |
| 278 // being |length|. |*begin| will be updated to point to the last character | |
| 279 // consumed (we may consume more than one for UTF-16) so that if called in | |
| 280 // a loop, incrementing the pointer will move to the next character. | |
| 281 // | |
| 282 // Every single output character will be escaped. This means that if you | |
| 283 // give it an ASCII character as input, it will be escaped. Some code uses | |
| 284 // this when it knows that a character is invalid according to its rules | |
| 285 // for validity. If you don't want escaping for ASCII characters, you will | |
| 286 // have to filter them out prior to calling this function. | |
| 287 // | |
| 288 // Assumes that ch[begin] is within range in the array, but does not assume | |
| 289 // that any following characters are. | |
| 290 inline bool AppendUTF8EscapedChar(const char16* str, int* begin, int length, | |
| 291 CanonOutput* output) { | |
| 292 // UTF-16 input. Readchar16 will handle invalid characters for us and give | |
| 293 // us the kUnicodeReplacementCharacter, so we don't have to do special | |
| 294 // checking after failure, just pass through the failure to the caller. | |
| 295 unsigned char_value; | |
| 296 bool success = ReadUTFChar(str, begin, length, &char_value); | |
| 297 AppendUTF8EscapedValue(char_value, output); | |
| 298 return success; | |
| 299 } | |
| 300 | |
| 301 // Handles UTF-8 input. See the wide version above for usage. | |
| 302 inline bool AppendUTF8EscapedChar(const char* str, int* begin, int length, | |
| 303 CanonOutput* output) { | |
| 304 // ReadUTF8Char will handle invalid characters for us and give us the | |
| 305 // kUnicodeReplacementCharacter, so we don't have to do special checking | |
| 306 // after failure, just pass through the failure to the caller. | |
| 307 unsigned ch; | |
| 308 bool success = ReadUTFChar(str, begin, length, &ch); | |
| 309 AppendUTF8EscapedValue(ch, output); | |
| 310 return success; | |
| 311 } | |
| 312 | |
| 313 // Given a '%' character at |*begin| in the string |spec|, this will decode | |
| 314 // the escaped value and put it into |*unescaped_value| on success (returns | |
| 315 // true). On failure, this will return false, and will not write into | |
| 316 // |*unescaped_value|. | |
| 317 // | |
| 318 // |*begin| will be updated to point to the last character of the escape | |
| 319 // sequence so that when called with the index of a for loop, the next time | |
| 320 // through it will point to the next character to be considered. On failure, | |
| 321 // |*begin| will be unchanged. | |
| 322 inline bool Is8BitChar(char c) { | |
| 323 return true; // this case is specialized to avoid a warning | |
| 324 } | |
| 325 inline bool Is8BitChar(char16 c) { | |
| 326 return c <= 255; | |
| 327 } | |
| 328 | |
| 329 template<typename CHAR> | |
| 330 inline bool DecodeEscaped(const CHAR* spec, int* begin, int end, | |
| 331 unsigned char* unescaped_value) { | |
| 332 if (*begin + 3 > end || | |
| 333 !Is8BitChar(spec[*begin + 1]) || !Is8BitChar(spec[*begin + 2])) { | |
| 334 // Invalid escape sequence because there's not enough room, or the | |
| 335 // digits are not ASCII. | |
| 336 return false; | |
| 337 } | |
| 338 | |
| 339 unsigned char first = static_cast<unsigned char>(spec[*begin + 1]); | |
| 340 unsigned char second = static_cast<unsigned char>(spec[*begin + 2]); | |
| 341 if (!IsHexChar(first) || !IsHexChar(second)) { | |
| 342 // Invalid hex digits, fail. | |
| 343 return false; | |
| 344 } | |
| 345 | |
| 346 // Valid escape sequence. | |
| 347 *unescaped_value = (HexCharToValue(first) << 4) + HexCharToValue(second); | |
| 348 *begin += 2; | |
| 349 return true; | |
| 350 } | |
| 351 | |
| 352 // Appends the given substring to the output, escaping "some" characters that | |
| 353 // it feels may not be safe. It assumes the input values are all contained in | |
| 354 // 8-bit although it allows any type. | |
| 355 // | |
| 356 // This is used in error cases to append invalid output so that it looks | |
| 357 // approximately correct. Non-error cases should not call this function since | |
| 358 // the escaping rules are not guaranteed! | |
| 359 void AppendInvalidNarrowString(const char* spec, int begin, int end, | |
| 360 CanonOutput* output); | |
| 361 void AppendInvalidNarrowString(const char16* spec, int begin, int end, | |
| 362 CanonOutput* output); | |
| 363 | |
| 364 // Misc canonicalization helpers ---------------------------------------------- | |
| 365 | |
| 366 // Converts between UTF-8 and UTF-16, returning true on successful conversion. | |
| 367 // The output will be appended to the given canonicalizer output (so make sure | |
| 368 // it's empty if you want to replace). | |
| 369 // | |
| 370 // On invalid input, this will still write as much output as possible, | |
| 371 // replacing the invalid characters with the "invalid character". It will | |
| 372 // return false in the failure case, and the caller should not continue as | |
| 373 // normal. | |
| 374 GURL_API bool ConvertUTF16ToUTF8(const char16* input, int input_len, | |
| 375 CanonOutput* output); | |
| 376 GURL_API bool ConvertUTF8ToUTF16(const char* input, int input_len, | |
| 377 CanonOutputT<char16>* output); | |
| 378 | |
| 379 // Converts from UTF-16 to 8-bit using the character set converter. If the | |
| 380 // converter is NULL, this will use UTF-8. | |
| 381 void ConvertUTF16ToQueryEncoding(const char16* input, | |
| 382 const url_parse::Component& query, | |
| 383 CharsetConverter* converter, | |
| 384 CanonOutput* output); | |
| 385 | |
| 386 // Applies the replacements to the given component source. The component source | |
| 387 // should be pre-initialized to the "old" base. That is, all pointers will | |
| 388 // point to the spec of the old URL, and all of the Parsed components will | |
| 389 // be indices into that string. | |
| 390 // | |
| 391 // The pointers and components in the |source| for all non-NULL strings in the | |
| 392 // |repl| (replacements) will be updated to reference those strings. | |
| 393 // Canonicalizing with the new |source| and |parsed| can then combine URL | |
| 394 // components from many different strings. | |
| 395 void SetupOverrideComponents(const char* base, | |
| 396 const Replacements<char>& repl, | |
| 397 URLComponentSource<char>* source, | |
| 398 url_parse::Parsed* parsed); | |
| 399 | |
| 400 // Like the above 8-bit version, except that it additionally converts the | |
| 401 // UTF-16 input to UTF-8 before doing the overrides. | |
| 402 // | |
| 403 // The given utf8_buffer is used to store the converted components. They will | |
| 404 // be appended one after another, with the parsed structure identifying the | |
| 405 // appropriate substrings. This buffer is a parameter because the source has | |
| 406 // no storage, so the buffer must have the same lifetime as the source | |
| 407 // parameter owned by the caller. | |
| 408 // | |
| 409 // THE CALLER MUST NOT ADD TO THE |utf8_buffer| AFTER THIS CALL. Members of | |
| 410 // |source| will point into this buffer, which could be invalidated if | |
| 411 // additional data is added and the CanonOutput resizes its buffer. | |
| 412 // | |
| 413 // Returns true on success. Fales means that the input was not valid UTF-16, | |
| 414 // although we will have still done the override with "invalid characters" in | |
| 415 // place of errors. | |
| 416 bool SetupUTF16OverrideComponents(const char* base, | |
| 417 const Replacements<char16>& repl, | |
| 418 CanonOutput* utf8_buffer, | |
| 419 URLComponentSource<char>* source, | |
| 420 url_parse::Parsed* parsed); | |
| 421 | |
| 422 // Implemented in url_canon_path.cc, these are required by the relative URL | |
| 423 // resolver as well, so we declare them here. | |
| 424 bool CanonicalizePartialPath(const char* spec, | |
| 425 const url_parse::Component& path, | |
| 426 int path_begin_in_output, | |
| 427 CanonOutput* output); | |
| 428 bool CanonicalizePartialPath(const char16* spec, | |
| 429 const url_parse::Component& path, | |
| 430 int path_begin_in_output, | |
| 431 CanonOutput* output); | |
| 432 | |
| 433 #ifndef WIN32 | |
| 434 | |
| 435 // Implementations of Windows' int-to-string conversions | |
| 436 GURL_API int _itoa_s(int value, char* buffer, size_t size_in_chars, int radix); | |
| 437 GURL_API int _itow_s(int value, char16* buffer, size_t size_in_chars, | |
| 438 int radix); | |
| 439 | |
| 440 // Secure template overloads for these functions | |
| 441 template<size_t N> | |
| 442 inline int _itoa_s(int value, char (&buffer)[N], int radix) { | |
| 443 return _itoa_s(value, buffer, N, radix); | |
| 444 } | |
| 445 | |
| 446 template<size_t N> | |
| 447 inline int _itow_s(int value, char16 (&buffer)[N], int radix) { | |
| 448 return _itow_s(value, buffer, N, radix); | |
| 449 } | |
| 450 | |
| 451 // _strtoui64 and strtoull behave the same | |
| 452 inline unsigned long long _strtoui64(const char* nptr, | |
| 453 char** endptr, int base) { | |
| 454 return strtoull(nptr, endptr, base); | |
| 455 } | |
| 456 | |
| 457 #endif // WIN32 | |
| 458 | |
| 459 } // namespace url_canon | |
| 460 | 39 |
| 461 #endif // GOOGLEURL_SRC_URL_CANON_INTERNAL_H__ | 40 #endif // GOOGLEURL_SRC_URL_CANON_INTERNAL_H__ |
| OLD | NEW |