| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "net/base/dns_util.h" | |
| 6 | |
| 7 #include <cstring> | |
| 8 | |
| 9 namespace net { | |
| 10 | |
| 11 // Based on DJB's public domain code. | |
| 12 bool DNSDomainFromDot(const base::StringPiece& dotted, std::string* out) { | |
| 13 const char* buf = dotted.data(); | |
| 14 unsigned n = dotted.size(); | |
| 15 char label[63]; | |
| 16 size_t labellen = 0; /* <= sizeof label */ | |
| 17 char name[255]; | |
| 18 size_t namelen = 0; /* <= sizeof name */ | |
| 19 char ch; | |
| 20 | |
| 21 for (;;) { | |
| 22 if (!n) | |
| 23 break; | |
| 24 ch = *buf++; | |
| 25 --n; | |
| 26 if (ch == '.') { | |
| 27 if (labellen) { | |
| 28 if (namelen + labellen + 1 > sizeof name) | |
| 29 return false; | |
| 30 name[namelen++] = static_cast<char>(labellen); | |
| 31 memcpy(name + namelen, label, labellen); | |
| 32 namelen += labellen; | |
| 33 labellen = 0; | |
| 34 } | |
| 35 continue; | |
| 36 } | |
| 37 if (labellen >= sizeof label) | |
| 38 return false; | |
| 39 label[labellen++] = ch; | |
| 40 } | |
| 41 | |
| 42 if (labellen) { | |
| 43 if (namelen + labellen + 1 > sizeof name) | |
| 44 return false; | |
| 45 name[namelen++] = static_cast<char>(labellen); | |
| 46 memcpy(name + namelen, label, labellen); | |
| 47 namelen += labellen; | |
| 48 labellen = 0; | |
| 49 } | |
| 50 | |
| 51 if (namelen + 1 > sizeof name) | |
| 52 return false; | |
| 53 if (namelen == 0) // Empty names e.g. "", "." are not valid. | |
| 54 return false; | |
| 55 name[namelen++] = 0; // This is the root label (of length 0). | |
| 56 | |
| 57 *out = std::string(name, namelen); | |
| 58 return true; | |
| 59 } | |
| 60 | |
| 61 std::string DNSDomainToString(const base::StringPiece& domain) { | |
| 62 std::string ret; | |
| 63 | |
| 64 for (unsigned i = 0; i < domain.size() && domain[i]; i += domain[i] + 1) { | |
| 65 #if CHAR_MIN < 0 | |
| 66 if (domain[i] < 0) | |
| 67 return std::string(); | |
| 68 #endif | |
| 69 if (domain[i] > 63) | |
| 70 return std::string(); | |
| 71 | |
| 72 if (i) | |
| 73 ret += "."; | |
| 74 | |
| 75 if (static_cast<unsigned>(domain[i]) + i + 1 > domain.size()) | |
| 76 return std::string(); | |
| 77 | |
| 78 domain.substr(i + 1, domain[i]).AppendToString(&ret); | |
| 79 } | |
| 80 return ret; | |
| 81 } | |
| 82 | |
| 83 std::string TrimEndingDot(const base::StringPiece& host) { | |
| 84 base::StringPiece host_trimmed = host; | |
| 85 size_t len = host_trimmed.length(); | |
| 86 if (len > 1 && host_trimmed[len - 1] == '.') { | |
| 87 host_trimmed.remove_suffix(1); | |
| 88 } | |
| 89 return host_trimmed.as_string(); | |
| 90 } | |
| 91 | |
| 92 } // namespace net | |
| OLD | NEW |