1 /*
2 * Copyright (C) 2008 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #define LOG_TAG "NetlinkEvent"
18
19 #include <arpa/inet.h>
20 #include <limits.h>
21 #include <linux/genetlink.h>
22 #include <linux/if.h>
23 #include <linux/if_addr.h>
24 #include <linux/if_link.h>
25 #include <linux/netfilter/nfnetlink.h>
26 #include <linux/netfilter/nfnetlink_log.h>
27 #include <linux/netfilter_ipv4/ipt_ULOG.h>
28 #include <linux/netlink.h>
29 #include <linux/rtnetlink.h>
30 #include <net/if.h>
31 #include <netinet/icmp6.h>
32 #include <netinet/in.h>
33 #include <stdlib.h>
34 #include <string.h>
35 #include <sys/socket.h>
36 #include <sys/types.h>
37
38 /* From kernel's net/netfilter/xt_quota2.c */
39 const int LOCAL_QLOG_NL_EVENT = 112;
40 const int LOCAL_NFLOG_PACKET = NFNL_SUBSYS_ULOG << 8 | NFULNL_MSG_PACKET;
41
42 #include <log/log.h>
43 #include <sysutils/NetlinkEvent.h>
44
NetlinkEvent()45 NetlinkEvent::NetlinkEvent() {
46 mAction = Action::kUnknown;
47 memset(mParams, 0, sizeof(mParams));
48 mPath = NULL;
49 mSubsystem = NULL;
50 }
51
~NetlinkEvent()52 NetlinkEvent::~NetlinkEvent() {
53 int i;
54 if (mPath)
55 free(mPath);
56 if (mSubsystem)
57 free(mSubsystem);
58 for (i = 0; i < NL_PARAMS_MAX; i++) {
59 if (!mParams[i])
60 break;
61 free(mParams[i]);
62 }
63 }
64
dump()65 void NetlinkEvent::dump() {
66 int i;
67
68 for (i = 0; i < NL_PARAMS_MAX; i++) {
69 if (!mParams[i])
70 break;
71 SLOGD("NL param '%s'\n", mParams[i]);
72 }
73 }
74
75 /*
76 * Returns the message name for a message in the NETLINK_ROUTE family, or NULL
77 * if parsing that message is not supported.
78 */
rtMessageName(int type)79 static const char *rtMessageName(int type) {
80 #define NL_EVENT_RTM_NAME(rtm) case rtm: return #rtm;
81 switch (type) {
82 NL_EVENT_RTM_NAME(RTM_NEWLINK);
83 NL_EVENT_RTM_NAME(RTM_DELLINK);
84 NL_EVENT_RTM_NAME(RTM_NEWADDR);
85 NL_EVENT_RTM_NAME(RTM_DELADDR);
86 NL_EVENT_RTM_NAME(RTM_NEWROUTE);
87 NL_EVENT_RTM_NAME(RTM_DELROUTE);
88 NL_EVENT_RTM_NAME(RTM_NEWNDUSEROPT);
89 NL_EVENT_RTM_NAME(LOCAL_QLOG_NL_EVENT);
90 NL_EVENT_RTM_NAME(LOCAL_NFLOG_PACKET);
91 default:
92 return NULL;
93 }
94 #undef NL_EVENT_RTM_NAME
95 }
96
97 /*
98 * Checks that a binary NETLINK_ROUTE message is long enough for a payload of
99 * size bytes.
100 */
checkRtNetlinkLength(const struct nlmsghdr * nh,size_t size)101 static bool checkRtNetlinkLength(const struct nlmsghdr *nh, size_t size) {
102 if (nh->nlmsg_len < NLMSG_LENGTH(size)) {
103 SLOGE("Got a short %s message\n", rtMessageName(nh->nlmsg_type));
104 return false;
105 }
106 return true;
107 }
108
109 /*
110 * Utility function to log errors.
111 */
maybeLogDuplicateAttribute(bool isDup,const char * attributeName,const char * messageName)112 static bool maybeLogDuplicateAttribute(bool isDup,
113 const char *attributeName,
114 const char *messageName) {
115 if (isDup) {
116 SLOGE("Multiple %s attributes in %s, ignoring\n", attributeName, messageName);
117 return true;
118 }
119 return false;
120 }
121
122 /*
123 * Parse a RTM_NEWLINK message.
124 */
parseIfInfoMessage(const struct nlmsghdr * nh)125 bool NetlinkEvent::parseIfInfoMessage(const struct nlmsghdr *nh) {
126 struct ifinfomsg *ifi = (struct ifinfomsg *) NLMSG_DATA(nh);
127 if (!checkRtNetlinkLength(nh, sizeof(*ifi)))
128 return false;
129
130 if ((ifi->ifi_flags & IFF_LOOPBACK) != 0) {
131 return false;
132 }
133
134 int len = IFLA_PAYLOAD(nh);
135 struct rtattr *rta;
136 for (rta = IFLA_RTA(ifi); RTA_OK(rta, len); rta = RTA_NEXT(rta, len)) {
137 switch(rta->rta_type) {
138 case IFLA_IFNAME:
139 asprintf(&mParams[0], "INTERFACE=%s", (char *) RTA_DATA(rta));
140 // We can get the interface change information from sysfs update
141 // already. But in case we missed those message when devices start.
142 // We do a update again when received a kLinkUp event. To make
143 // the message consistent, use IFINDEX here as well since sysfs
144 // uses IFINDEX.
145 asprintf(&mParams[1], "IFINDEX=%d", ifi->ifi_index);
146 mAction = (ifi->ifi_flags & IFF_LOWER_UP) ? Action::kLinkUp :
147 Action::kLinkDown;
148 mSubsystem = strdup("net");
149 return true;
150 }
151 }
152
153 return false;
154 }
155
156 /*
157 * Parse a RTM_NEWADDR or RTM_DELADDR message.
158 */
parseIfAddrMessage(const struct nlmsghdr * nh)159 bool NetlinkEvent::parseIfAddrMessage(const struct nlmsghdr *nh) {
160 struct ifaddrmsg *ifaddr = (struct ifaddrmsg *) NLMSG_DATA(nh);
161 struct ifa_cacheinfo *cacheinfo = NULL;
162 char addrstr[INET6_ADDRSTRLEN] = "";
163 char ifname[IFNAMSIZ] = "";
164
165 if (!checkRtNetlinkLength(nh, sizeof(*ifaddr)))
166 return false;
167
168 // Sanity check.
169 int type = nh->nlmsg_type;
170 if (type != RTM_NEWADDR && type != RTM_DELADDR) {
171 SLOGE("parseIfAddrMessage on incorrect message type 0x%x\n", type);
172 return false;
173 }
174
175 // For log messages.
176 const char *msgtype = rtMessageName(type);
177
178 struct rtattr *rta;
179 int len = IFA_PAYLOAD(nh);
180 for (rta = IFA_RTA(ifaddr); RTA_OK(rta, len); rta = RTA_NEXT(rta, len)) {
181 if (rta->rta_type == IFA_ADDRESS) {
182 // Only look at the first address, because we only support notifying
183 // one change at a time.
184 if (maybeLogDuplicateAttribute(*addrstr != '\0', "IFA_ADDRESS", msgtype))
185 continue;
186
187 // Convert the IP address to a string.
188 if (ifaddr->ifa_family == AF_INET) {
189 struct in_addr *addr4 = (struct in_addr *) RTA_DATA(rta);
190 if (RTA_PAYLOAD(rta) < sizeof(*addr4)) {
191 SLOGE("Short IPv4 address (%zu bytes) in %s",
192 RTA_PAYLOAD(rta), msgtype);
193 continue;
194 }
195 inet_ntop(AF_INET, addr4, addrstr, sizeof(addrstr));
196 } else if (ifaddr->ifa_family == AF_INET6) {
197 struct in6_addr *addr6 = (struct in6_addr *) RTA_DATA(rta);
198 if (RTA_PAYLOAD(rta) < sizeof(*addr6)) {
199 SLOGE("Short IPv6 address (%zu bytes) in %s",
200 RTA_PAYLOAD(rta), msgtype);
201 continue;
202 }
203 inet_ntop(AF_INET6, addr6, addrstr, sizeof(addrstr));
204 } else {
205 SLOGE("Unknown address family %d\n", ifaddr->ifa_family);
206 continue;
207 }
208
209 // Find the interface name.
210 if (!if_indextoname(ifaddr->ifa_index, ifname)) {
211 SLOGD("Unknown ifindex %d in %s", ifaddr->ifa_index, msgtype);
212 }
213
214 } else if (rta->rta_type == IFA_CACHEINFO) {
215 // Address lifetime information.
216 if (maybeLogDuplicateAttribute(cacheinfo, "IFA_CACHEINFO", msgtype))
217 continue;
218
219 if (RTA_PAYLOAD(rta) < sizeof(*cacheinfo)) {
220 SLOGE("Short IFA_CACHEINFO (%zu vs. %zu bytes) in %s",
221 RTA_PAYLOAD(rta), sizeof(cacheinfo), msgtype);
222 continue;
223 }
224
225 cacheinfo = (struct ifa_cacheinfo *) RTA_DATA(rta);
226 }
227 }
228
229 if (addrstr[0] == '\0') {
230 SLOGE("No IFA_ADDRESS in %s\n", msgtype);
231 return false;
232 }
233
234 // Fill in netlink event information.
235 mAction = (type == RTM_NEWADDR) ? Action::kAddressUpdated :
236 Action::kAddressRemoved;
237 mSubsystem = strdup("net");
238 asprintf(&mParams[0], "ADDRESS=%s/%d", addrstr, ifaddr->ifa_prefixlen);
239 asprintf(&mParams[1], "INTERFACE=%s", ifname);
240 asprintf(&mParams[2], "FLAGS=%u", ifaddr->ifa_flags);
241 asprintf(&mParams[3], "SCOPE=%u", ifaddr->ifa_scope);
242 asprintf(&mParams[4], "IFINDEX=%u", ifaddr->ifa_index);
243
244 if (cacheinfo) {
245 asprintf(&mParams[5], "PREFERRED=%u", cacheinfo->ifa_prefered);
246 asprintf(&mParams[6], "VALID=%u", cacheinfo->ifa_valid);
247 asprintf(&mParams[7], "CSTAMP=%u", cacheinfo->cstamp);
248 asprintf(&mParams[8], "TSTAMP=%u", cacheinfo->tstamp);
249 }
250
251 return true;
252 }
253
254 /*
255 * Parse a QLOG_NL_EVENT message.
256 */
parseUlogPacketMessage(const struct nlmsghdr * nh)257 bool NetlinkEvent::parseUlogPacketMessage(const struct nlmsghdr *nh) {
258 const char *devname;
259 ulog_packet_msg_t *pm = (ulog_packet_msg_t *) NLMSG_DATA(nh);
260 if (!checkRtNetlinkLength(nh, sizeof(*pm)))
261 return false;
262
263 devname = pm->indev_name[0] ? pm->indev_name : pm->outdev_name;
264 asprintf(&mParams[0], "ALERT_NAME=%s", pm->prefix);
265 asprintf(&mParams[1], "INTERFACE=%s", devname);
266 mSubsystem = strdup("qlog");
267 mAction = Action::kChange;
268 return true;
269 }
270
nlAttrLen(const nlattr * nla)271 static size_t nlAttrLen(const nlattr* nla) {
272 return nla->nla_len - NLA_HDRLEN;
273 }
274
nlAttrData(const nlattr * nla)275 static const uint8_t* nlAttrData(const nlattr* nla) {
276 return reinterpret_cast<const uint8_t*>(nla) + NLA_HDRLEN;
277 }
278
nlAttrU32(const nlattr * nla)279 static uint32_t nlAttrU32(const nlattr* nla) {
280 return *reinterpret_cast<const uint32_t*>(nlAttrData(nla));
281 }
282
283 /*
284 * Parse a LOCAL_NFLOG_PACKET message.
285 */
parseNfPacketMessage(struct nlmsghdr * nh)286 bool NetlinkEvent::parseNfPacketMessage(struct nlmsghdr *nh) {
287 int uid = -1;
288 int len = 0;
289 char* raw = NULL;
290
291 struct nlattr* uid_attr = findNlAttr(nh, sizeof(struct genlmsghdr), NFULA_UID);
292 if (uid_attr) {
293 uid = ntohl(nlAttrU32(uid_attr));
294 }
295
296 struct nlattr* payload = findNlAttr(nh, sizeof(struct genlmsghdr), NFULA_PAYLOAD);
297 if (payload) {
298 /* First 256 bytes is plenty */
299 len = nlAttrLen(payload);
300 if (len > 256) len = 256;
301 raw = (char*)nlAttrData(payload);
302 }
303
304 char* hex = (char*) calloc(1, 5 + (len * 2));
305 strcpy(hex, "HEX=");
306 for (int i = 0; i < len; i++) {
307 hex[4 + (i * 2)] = "0123456789abcdef"[(raw[i] >> 4) & 0xf];
308 hex[5 + (i * 2)] = "0123456789abcdef"[raw[i] & 0xf];
309 }
310
311 asprintf(&mParams[0], "UID=%d", uid);
312 mParams[1] = hex;
313 mSubsystem = strdup("strict");
314 mAction = Action::kChange;
315 return true;
316 }
317
318 /*
319 * Parse a RTM_NEWROUTE or RTM_DELROUTE message.
320 */
parseRtMessage(const struct nlmsghdr * nh)321 bool NetlinkEvent::parseRtMessage(const struct nlmsghdr *nh) {
322 uint8_t type = nh->nlmsg_type;
323 const char *msgname = rtMessageName(type);
324
325 // Sanity check.
326 if (type != RTM_NEWROUTE && type != RTM_DELROUTE) {
327 SLOGE("%s: incorrect message type %d (%s)\n", __func__, type, msgname);
328 return false;
329 }
330
331 struct rtmsg *rtm = (struct rtmsg *) NLMSG_DATA(nh);
332 if (!checkRtNetlinkLength(nh, sizeof(*rtm)))
333 return false;
334
335 if (// Ignore static routes we've set up ourselves.
336 (rtm->rtm_protocol != RTPROT_KERNEL &&
337 rtm->rtm_protocol != RTPROT_RA) ||
338 // We're only interested in global unicast routes.
339 (rtm->rtm_scope != RT_SCOPE_UNIVERSE) ||
340 (rtm->rtm_type != RTN_UNICAST) ||
341 // We don't support source routing.
342 (rtm->rtm_src_len != 0) ||
343 // Cloned routes aren't real routes.
344 (rtm->rtm_flags & RTM_F_CLONED)) {
345 return false;
346 }
347
348 int family = rtm->rtm_family;
349 int prefixLength = rtm->rtm_dst_len;
350
351 // Currently we only support: destination, (one) next hop, ifindex.
352 char dst[INET6_ADDRSTRLEN] = "";
353 char gw[INET6_ADDRSTRLEN] = "";
354 char dev[IFNAMSIZ] = "";
355
356 size_t len = RTM_PAYLOAD(nh);
357 struct rtattr *rta;
358 for (rta = RTM_RTA(rtm); RTA_OK(rta, len); rta = RTA_NEXT(rta, len)) {
359 switch (rta->rta_type) {
360 case RTA_DST:
361 if (maybeLogDuplicateAttribute(*dst, "RTA_DST", msgname))
362 continue;
363 if (!inet_ntop(family, RTA_DATA(rta), dst, sizeof(dst)))
364 return false;
365 continue;
366 case RTA_GATEWAY:
367 if (maybeLogDuplicateAttribute(*gw, "RTA_GATEWAY", msgname))
368 continue;
369 if (!inet_ntop(family, RTA_DATA(rta), gw, sizeof(gw)))
370 return false;
371 continue;
372 case RTA_OIF:
373 if (maybeLogDuplicateAttribute(*dev, "RTA_OIF", msgname))
374 continue;
375 if (!if_indextoname(* (int *) RTA_DATA(rta), dev))
376 return false;
377 default:
378 continue;
379 }
380 }
381
382 // If there's no RTA_DST attribute, then:
383 // - If the prefix length is zero, it's the default route.
384 // - If the prefix length is nonzero, there's something we don't understand.
385 // Ignore the event.
386 if (!*dst && !prefixLength) {
387 if (family == AF_INET) {
388 strncpy(dst, "0.0.0.0", sizeof(dst));
389 } else if (family == AF_INET6) {
390 strncpy(dst, "::", sizeof(dst));
391 }
392 }
393
394 // A useful route must have a destination and at least either a gateway or
395 // an interface.
396 if (!*dst || (!*gw && !*dev))
397 return false;
398
399 // Fill in netlink event information.
400 mAction = (type == RTM_NEWROUTE) ? Action::kRouteUpdated :
401 Action::kRouteRemoved;
402 mSubsystem = strdup("net");
403 asprintf(&mParams[0], "ROUTE=%s/%d", dst, prefixLength);
404 asprintf(&mParams[1], "GATEWAY=%s", (*gw) ? gw : "");
405 asprintf(&mParams[2], "INTERFACE=%s", (*dev) ? dev : "");
406
407 return true;
408 }
409
410 /*
411 * Parse a RTM_NEWNDUSEROPT message.
412 */
parseNdUserOptMessage(const struct nlmsghdr * nh)413 bool NetlinkEvent::parseNdUserOptMessage(const struct nlmsghdr *nh) {
414 struct nduseroptmsg *msg = (struct nduseroptmsg *) NLMSG_DATA(nh);
415 if (!checkRtNetlinkLength(nh, sizeof(*msg)))
416 return false;
417
418 // Check the length is valid.
419 int len = NLMSG_PAYLOAD(nh, sizeof(*msg));
420 if (msg->nduseropt_opts_len > len) {
421 SLOGE("RTM_NEWNDUSEROPT invalid length %d > %d\n",
422 msg->nduseropt_opts_len, len);
423 return false;
424 }
425 len = msg->nduseropt_opts_len;
426
427 // Check address family and packet type.
428 if (msg->nduseropt_family != AF_INET6) {
429 SLOGE("RTM_NEWNDUSEROPT message for unknown family %d\n",
430 msg->nduseropt_family);
431 return false;
432 }
433
434 if (msg->nduseropt_icmp_type != ND_ROUTER_ADVERT ||
435 msg->nduseropt_icmp_code != 0) {
436 SLOGE("RTM_NEWNDUSEROPT message for unknown ICMPv6 type/code %d/%d\n",
437 msg->nduseropt_icmp_type, msg->nduseropt_icmp_code);
438 return false;
439 }
440
441 // Find the interface name.
442 char ifname[IFNAMSIZ];
443 if (!if_indextoname(msg->nduseropt_ifindex, ifname)) {
444 SLOGE("RTM_NEWNDUSEROPT on unknown ifindex %d\n",
445 msg->nduseropt_ifindex);
446 return false;
447 }
448
449 // The kernel sends a separate netlink message for each ND option in the RA.
450 // So only parse the first ND option in the message.
451 struct nd_opt_hdr *opthdr = (struct nd_opt_hdr *) (msg + 1);
452
453 // The length is in multiples of 8 octets.
454 uint16_t optlen = opthdr->nd_opt_len;
455 if (optlen * 8 > len) {
456 SLOGE("Invalid option length %d > %d for ND option %d\n",
457 optlen * 8, len, opthdr->nd_opt_type);
458 return false;
459 }
460
461 if (opthdr->nd_opt_type == ND_OPT_RDNSS) {
462 // DNS Servers (RFC 6106).
463 // Each address takes up 2*8 octets, and the header takes up 8 octets.
464 // So for a valid option with one or more addresses, optlen must be
465 // odd and greater than 1.
466 if ((optlen < 3) || !(optlen & 0x1)) {
467 SLOGE("Invalid optlen %d for RDNSS option\n", optlen);
468 return false;
469 }
470 const int numaddrs = (optlen - 1) / 2;
471
472 // Find the lifetime.
473 struct nd_opt_rdnss *rndss_opt = (struct nd_opt_rdnss *) opthdr;
474 const uint32_t lifetime = ntohl(rndss_opt->nd_opt_rdnss_lifetime);
475
476 // Construct "SERVERS=<comma-separated string of DNS addresses>".
477 static const char kServerTag[] = "SERVERS=";
478 static const size_t kTagLength = strlen(kServerTag);
479 // Reserve sufficient space for an IPv6 link-local address: all but the
480 // last address are followed by ','; the last is followed by '\0'.
481 static const size_t kMaxSingleAddressLength =
482 INET6_ADDRSTRLEN + strlen("%") + IFNAMSIZ + strlen(",");
483 const size_t bufsize = kTagLength + numaddrs * kMaxSingleAddressLength;
484 char *buf = (char *) malloc(bufsize);
485 if (!buf) {
486 SLOGE("RDNSS option: out of memory\n");
487 return false;
488 }
489 strcpy(buf, kServerTag);
490 size_t pos = kTagLength;
491
492 struct in6_addr *addrs = (struct in6_addr *) (rndss_opt + 1);
493 for (int i = 0; i < numaddrs; i++) {
494 if (i > 0) {
495 buf[pos++] = ',';
496 }
497 inet_ntop(AF_INET6, addrs + i, buf + pos, bufsize - pos);
498 pos += strlen(buf + pos);
499 if (IN6_IS_ADDR_LINKLOCAL(addrs + i)) {
500 buf[pos++] = '%';
501 pos += strlcpy(buf + pos, ifname, bufsize - pos);
502 }
503 }
504 buf[pos] = '\0';
505
506 mAction = Action::kRdnss;
507 mSubsystem = strdup("net");
508 asprintf(&mParams[0], "INTERFACE=%s", ifname);
509 asprintf(&mParams[1], "LIFETIME=%u", lifetime);
510 mParams[2] = buf;
511 } else if (opthdr->nd_opt_type == ND_OPT_DNSSL) {
512 // TODO: support DNSSL.
513 } else {
514 SLOGD("Unknown ND option type %d\n", opthdr->nd_opt_type);
515 return false;
516 }
517
518 return true;
519 }
520
521 /*
522 * Parse a binary message from a NETLINK_ROUTE netlink socket.
523 *
524 * Note that this function can only parse one message, because the message's
525 * content has to be stored in the class's member variables (mAction,
526 * mSubsystem, etc.). Invalid or unrecognized messages are skipped, but if
527 * there are multiple valid messages in the buffer, only the first one will be
528 * returned.
529 *
530 * TODO: consider only ever looking at the first message.
531 */
parseBinaryNetlinkMessage(char * buffer,int size)532 bool NetlinkEvent::parseBinaryNetlinkMessage(char *buffer, int size) {
533 struct nlmsghdr *nh;
534
535 for (nh = (struct nlmsghdr *) buffer;
536 NLMSG_OK(nh, (unsigned) size) && (nh->nlmsg_type != NLMSG_DONE);
537 nh = NLMSG_NEXT(nh, size)) {
538
539 if (!rtMessageName(nh->nlmsg_type)) {
540 SLOGD("Unexpected netlink message type %d\n", nh->nlmsg_type);
541 continue;
542 }
543
544 if (nh->nlmsg_type == RTM_NEWLINK) {
545 if (parseIfInfoMessage(nh))
546 return true;
547
548 } else if (nh->nlmsg_type == LOCAL_QLOG_NL_EVENT) {
549 if (parseUlogPacketMessage(nh))
550 return true;
551
552 } else if (nh->nlmsg_type == RTM_NEWADDR ||
553 nh->nlmsg_type == RTM_DELADDR) {
554 if (parseIfAddrMessage(nh))
555 return true;
556
557 } else if (nh->nlmsg_type == RTM_NEWROUTE ||
558 nh->nlmsg_type == RTM_DELROUTE) {
559 if (parseRtMessage(nh))
560 return true;
561
562 } else if (nh->nlmsg_type == RTM_NEWNDUSEROPT) {
563 if (parseNdUserOptMessage(nh))
564 return true;
565
566 } else if (nh->nlmsg_type == LOCAL_NFLOG_PACKET) {
567 if (parseNfPacketMessage(nh))
568 return true;
569
570 }
571 }
572
573 return false;
574 }
575
576 /* If the string between 'str' and 'end' begins with 'prefixlen' characters
577 * from the 'prefix' array, then return 'str + prefixlen', otherwise return
578 * NULL.
579 */
580 static const char*
has_prefix(const char * str,const char * end,const char * prefix,size_t prefixlen)581 has_prefix(const char* str, const char* end, const char* prefix, size_t prefixlen)
582 {
583 if ((end - str) >= (ptrdiff_t)prefixlen &&
584 (prefixlen == 0 || !memcmp(str, prefix, prefixlen))) {
585 return str + prefixlen;
586 } else {
587 return NULL;
588 }
589 }
590
591 /* Same as strlen(x) for constant string literals ONLY */
592 #define CONST_STRLEN(x) (sizeof(x)-1)
593
594 /* Convenience macro to call has_prefix with a constant string literal */
595 #define HAS_CONST_PREFIX(str,end,prefix) has_prefix((str),(end),prefix,CONST_STRLEN(prefix))
596
597
598 /*
599 * Parse an ASCII-formatted message from a NETLINK_KOBJECT_UEVENT
600 * netlink socket.
601 */
parseAsciiNetlinkMessage(char * buffer,int size)602 bool NetlinkEvent::parseAsciiNetlinkMessage(char *buffer, int size) {
603 const char *s = buffer;
604 const char *end;
605 int param_idx = 0;
606 int first = 1;
607
608 if (size == 0)
609 return false;
610
611 /* Ensure the buffer is zero-terminated, the code below depends on this */
612 buffer[size-1] = '\0';
613
614 end = s + size;
615 while (s < end) {
616 if (first) {
617 const char *p;
618 /* buffer is 0-terminated, no need to check p < end */
619 for (p = s; *p != '@'; p++) {
620 if (!*p) { /* no '@', should not happen */
621 return false;
622 }
623 }
624 mPath = strdup(p+1);
625 first = 0;
626 } else {
627 const char* a;
628 if ((a = HAS_CONST_PREFIX(s, end, "ACTION=")) != NULL) {
629 if (!strcmp(a, "add"))
630 mAction = Action::kAdd;
631 else if (!strcmp(a, "remove"))
632 mAction = Action::kRemove;
633 else if (!strcmp(a, "change"))
634 mAction = Action::kChange;
635 } else if ((a = HAS_CONST_PREFIX(s, end, "SEQNUM=")) != NULL) {
636 mSeq = atoi(a);
637 } else if ((a = HAS_CONST_PREFIX(s, end, "SUBSYSTEM=")) != NULL) {
638 mSubsystem = strdup(a);
639 } else if (param_idx < NL_PARAMS_MAX) {
640 mParams[param_idx++] = strdup(s);
641 }
642 }
643 s += strlen(s) + 1;
644 }
645 return true;
646 }
647
decode(char * buffer,int size,int format)648 bool NetlinkEvent::decode(char *buffer, int size, int format) {
649 if (format == NetlinkListener::NETLINK_FORMAT_BINARY
650 || format == NetlinkListener::NETLINK_FORMAT_BINARY_UNICAST) {
651 return parseBinaryNetlinkMessage(buffer, size);
652 } else {
653 return parseAsciiNetlinkMessage(buffer, size);
654 }
655 }
656
findParam(const char * paramName)657 const char *NetlinkEvent::findParam(const char *paramName) {
658 size_t len = strlen(paramName);
659 for (int i = 0; i < NL_PARAMS_MAX && mParams[i] != NULL; ++i) {
660 const char *ptr = mParams[i] + len;
661 if (!strncmp(mParams[i], paramName, len) && *ptr == '=')
662 return ++ptr;
663 }
664
665 SLOGE("NetlinkEvent::FindParam(): Parameter '%s' not found", paramName);
666 return NULL;
667 }
668
findNlAttr(const nlmsghdr * nh,size_t hdrlen,uint16_t attr)669 nlattr* NetlinkEvent::findNlAttr(const nlmsghdr* nh, size_t hdrlen, uint16_t attr) {
670 if (nh == nullptr || NLMSG_HDRLEN + NLMSG_ALIGN(hdrlen) > SSIZE_MAX) {
671 return nullptr;
672 }
673
674 // Skip header, padding, and family header.
675 const ssize_t NLA_START = NLMSG_HDRLEN + NLMSG_ALIGN(hdrlen);
676 ssize_t left = nh->nlmsg_len - NLA_START;
677 uint8_t* hdr = ((uint8_t*)nh) + NLA_START;
678
679 while (left >= NLA_HDRLEN) {
680 nlattr* nla = (nlattr*)hdr;
681 if (nla->nla_type == attr) {
682 return nla;
683 }
684
685 hdr += NLA_ALIGN(nla->nla_len);
686 left -= NLA_ALIGN(nla->nla_len);
687 }
688
689 return nullptr;
690 }
691