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