OLD | NEW |
1 // Copyright (c) 2010 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/address_list.h" | 5 #include "net/base/address_list.h" |
6 | 6 |
7 #include <stdlib.h> | 7 #include <stdlib.h> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "net/base/net_util.h" | 10 #include "net/base/net_util.h" |
11 #include "net/base/sys_addrinfo.h" | 11 #include "net/base/sys_addrinfo.h" |
12 | 12 |
13 namespace net { | 13 namespace net { |
14 | 14 |
15 namespace { | 15 namespace { |
16 | 16 |
17 char* do_strdup(const char* src) { | 17 char* do_strdup(const char* src) { |
18 #if defined(OS_WIN) | 18 #if defined(OS_WIN) |
19 return _strdup(src); | 19 return _strdup(src); |
20 #else | 20 #else |
21 return strdup(src); | 21 return strdup(src); |
22 #endif | 22 #endif |
23 } | 23 } |
24 | 24 |
25 // Make a copy of |info| (the dynamically-allocated parts are copied as well). | |
26 // If |recursive| is true, chained entries via ai_next are copied too. | |
27 // Copy returned by this function should be deleted using | |
28 // DeleteCopyOfAddrinfo(), and NOT freeaddrinfo(). | |
29 struct addrinfo* CreateCopyOfAddrinfo(const struct addrinfo* info, | |
30 bool recursive) { | |
31 DCHECK(info); | |
32 struct addrinfo* copy = new addrinfo; | |
33 | |
34 // Copy all the fields (some of these are pointers, we will fix that next). | |
35 memcpy(copy, info, sizeof(addrinfo)); | |
36 | |
37 // ai_canonname is a NULL-terminated string. | |
38 if (info->ai_canonname) { | |
39 copy->ai_canonname = do_strdup(info->ai_canonname); | |
40 } | |
41 | |
42 // ai_addr is a buffer of length ai_addrlen. | |
43 if (info->ai_addr) { | |
44 copy->ai_addr = reinterpret_cast<sockaddr *>(new char[info->ai_addrlen]); | |
45 memcpy(copy->ai_addr, info->ai_addr, info->ai_addrlen); | |
46 } | |
47 | |
48 // Recursive copy. | |
49 if (recursive && info->ai_next) | |
50 copy->ai_next = CreateCopyOfAddrinfo(info->ai_next, recursive); | |
51 else | |
52 copy->ai_next = NULL; | |
53 | |
54 return copy; | |
55 } | |
56 | |
57 // Free an addrinfo that was created by CreateCopyOfAddrinfo(). | |
58 void FreeMyAddrinfo(struct addrinfo* info) { | |
59 DCHECK(info); | |
60 if (info->ai_canonname) | |
61 free(info->ai_canonname); // Allocated by strdup. | |
62 | |
63 if (info->ai_addr) | |
64 delete [] reinterpret_cast<char*>(info->ai_addr); | |
65 | |
66 struct addrinfo* next = info->ai_next; | |
67 | |
68 delete info; | |
69 | |
70 // Recursive free. | |
71 if (next) | |
72 FreeMyAddrinfo(next); | |
73 } | |
74 | |
75 // Assign the port for all addresses in the list. | 25 // Assign the port for all addresses in the list. |
76 void SetPortRecursive(struct addrinfo* info, int port) { | 26 void SetPortRecursive(struct addrinfo* info, int port) { |
77 uint16* port_field = GetPortFieldFromAddrinfo(info); | 27 uint16* port_field = GetPortFieldFromAddrinfo(info); |
78 if (port_field) | 28 if (port_field) |
79 *port_field = htons(port); | 29 *port_field = htons(port); |
80 | 30 |
81 // Assign recursively. | 31 // Assign recursively. |
82 if (info->ai_next) | 32 if (info->ai_next) |
83 SetPortRecursive(info->ai_next, port); | 33 SetPortRecursive(info->ai_next, port); |
84 } | 34 } |
(...skipping 191 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
276 return new AddressList(new Data(ai, false /*is_system_created*/)); | 226 return new AddressList(new Data(ai, false /*is_system_created*/)); |
277 } | 227 } |
278 | 228 |
279 | 229 |
280 AddressList::Data::Data(struct addrinfo* ai, bool is_system_created) | 230 AddressList::Data::Data(struct addrinfo* ai, bool is_system_created) |
281 : head(ai), is_system_created(is_system_created) { | 231 : head(ai), is_system_created(is_system_created) { |
282 DCHECK(head); | 232 DCHECK(head); |
283 } | 233 } |
284 | 234 |
285 AddressList::Data::~Data() { | 235 AddressList::Data::~Data() { |
286 // Call either freeaddrinfo(head), or FreeMyAddrinfo(head), depending who | 236 // Call either freeaddrinfo(head), or FreeCopyOfAddrinfo(head), depending on |
287 // created the data. | 237 // who created the data. |
288 if (is_system_created) | 238 if (is_system_created) |
289 freeaddrinfo(head); | 239 freeaddrinfo(head); |
290 else | 240 else |
291 FreeMyAddrinfo(head); | 241 FreeCopyOfAddrinfo(head); |
292 } | 242 } |
293 | 243 |
294 } // namespace net | 244 } // namespace net |
OLD | NEW |