Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(444)

Side by Side Diff: net/base/network_change_notifier_netlink_linux.cc

Issue 10689015: [net] Adds AddressTrackerLinux which keeps track of interface addresses using rtnetlink. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Responded to vandebo's review. Created 8 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
(Empty)
1 // Copyright (c) 2010 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/network_change_notifier_netlink_linux.h"
6
7 #include <fcntl.h>
8 // socket.h is needed to define types for the linux kernel header netlink.h
9 // so it needs to come before netlink.h.
10 #include <sys/socket.h>
11 #include <linux/netlink.h>
12 #include <linux/rtnetlink.h>
13 #include <string.h>
14 #include <unistd.h>
15
16 #include "base/logging.h"
17
18 namespace {
19
20 // Return true on success, false on failure.
21 // Too small a function to bother putting in a library?
22 bool SetNonBlocking(int fd) {
23 int flags = fcntl(fd, F_GETFL, 0);
24 if (-1 == flags)
25 return false;
26 return fcntl(fd, F_SETFL, flags | O_NONBLOCK) == 0 ? true : false;
27 }
28
29 bool IsIPv6Update(const struct nlmsghdr* netlink_message_header) {
30 const struct ifaddrmsg* address_message =
31 reinterpret_cast<struct ifaddrmsg*>(NLMSG_DATA(netlink_message_header));
32 return address_message->ifa_family == AF_INET6;
33 }
34
35 bool IsDuplicateIPv6AddressUpdate(
36 const struct nlmsghdr* netlink_message_header) {
37 const struct ifaddrmsg* address_message =
38 reinterpret_cast<struct ifaddrmsg*>(NLMSG_DATA(netlink_message_header));
39 int address_message_length = IFA_PAYLOAD(netlink_message_header);
40 const struct rtattr* route_attribute =
41 reinterpret_cast<struct rtattr*>(IFA_RTA(address_message));
42 DCHECK_EQ(address_message->ifa_family, AF_INET6);
43
44 // Look for a cacheinfo attribute, and ignore new address broadcasts
45 // where the updated time stamp is newer than the created time stamp.
46 while (RTA_OK(route_attribute, address_message_length)) {
47 if (route_attribute->rta_type == IFA_CACHEINFO) {
48 struct ifa_cacheinfo* cache_info =
49 reinterpret_cast<struct ifa_cacheinfo*>(RTA_DATA(route_attribute));
50 if (cache_info->cstamp != cache_info->tstamp)
51 return true;
52 }
53 route_attribute = RTA_NEXT(route_attribute, address_message_length);
54 }
55 return false;
56 }
57
58 } // namespace
59
60 int InitializeNetlinkSocket() {
61 int sock = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
62 if (sock < 0) {
63 PLOG(ERROR) << "Error creating netlink socket";
64 return -1;
65 }
66
67 if (!SetNonBlocking(sock)) {
68 PLOG(ERROR) << "Failed to set netlink socket to non-blocking mode.";
69 if (close(sock) != 0)
70 PLOG(ERROR) << "Failed to close socket";
71 return -1;
72 }
73
74 struct sockaddr_nl local_addr;
75 memset(&local_addr, 0, sizeof(local_addr));
76 local_addr.nl_family = AF_NETLINK;
77 local_addr.nl_pid = getpid();
78 local_addr.nl_groups = RTMGRP_IPV4_IFADDR | RTMGRP_IPV6_IFADDR |
79 RTMGRP_NOTIFY;
80 int ret = bind(sock, reinterpret_cast<struct sockaddr*>(&local_addr),
81 sizeof(local_addr));
82 if (ret < 0) {
83 PLOG(ERROR) << "Error binding netlink socket";
84 if (close(sock) != 0)
85 PLOG(ERROR) << "Failed to close socket";
86 return -1;
87 }
88
89 return sock;
90 }
91
92 bool HandleNetlinkMessage(char* buf, size_t len) {
93 const struct nlmsghdr* netlink_message_header =
94 reinterpret_cast<struct nlmsghdr*>(buf);
95 DCHECK(netlink_message_header);
96 for (; NLMSG_OK(netlink_message_header, len);
97 netlink_message_header = NLMSG_NEXT(netlink_message_header, len)) {
98 int netlink_message_type = netlink_message_header->nlmsg_type;
99 switch (netlink_message_type) {
100 case NLMSG_DONE:
101 NOTREACHED()
102 << "This is a monitoring netlink socket. It should never be done.";
103 return false;
104 case NLMSG_ERROR:
105 LOG(ERROR) << "Unexpected netlink error.";
106 return false;
107 // During IP address changes, we will see all these messages. Only fire
108 // the notification when we get a new address or remove an address. We
109 // may still end up notifying observers more than strictly necessary, but
110 // if the primary interface goes down and back up, then this is necessary.
111 case RTM_NEWADDR:
112 if (IsIPv6Update(netlink_message_header) &&
113 IsDuplicateIPv6AddressUpdate(netlink_message_header))
114 return false;
115 return true;
116 case RTM_DELADDR:
117 return true;
118 case RTM_NEWLINK:
119 case RTM_DELLINK:
120 return false;
121 default:
122 LOG(DFATAL) << "Received unexpected netlink message type: "
123 << netlink_message_type;
124 return false;
125 }
126 }
127
128 return false;
129 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698