NetlinkEvent.cpp revision e6f80149a201e02ddd1e251e0690ad100b688cd6 
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#include <stdlib.h>
17#include <string.h>
18
19#define LOG_TAG "NetlinkEvent"
20#include <cutils/log.h>
21
22#include <sysutils/NetlinkEvent.h>
23
24#include <sys/types.h>
25#include <sys/socket.h>
26#include <linux/if.h>
27#include <linux/netfilter/nfnetlink.h>
28#include <linux/netfilter_ipv4/ipt_ULOG.h>
29/* From kernel's net/netfilter/xt_quota2.c */
30const int QLOG_NL_EVENT  = 112;
31
32#include <linux/netlink.h>
33#include <linux/rtnetlink.h>
34
35const int NetlinkEvent::NlActionUnknown = 0;
36const int NetlinkEvent::NlActionAdd = 1;
37const int NetlinkEvent::NlActionRemove = 2;
38const int NetlinkEvent::NlActionChange = 3;
39const int NetlinkEvent::NlActionLinkUp = 4;
40const int NetlinkEvent::NlActionLinkDown = 5;
41
42NetlinkEvent::NetlinkEvent() {
43    mAction = NlActionUnknown;
44    memset(mParams, 0, sizeof(mParams));
45    mPath = NULL;
46    mSubsystem = NULL;
47}
48
49NetlinkEvent::~NetlinkEvent() {
50    int i;
51    if (mPath)
52        free(mPath);
53    if (mSubsystem)
54        free(mSubsystem);
55    for (i = 0; i < NL_PARAMS_MAX; i++) {
56        if (!mParams[i])
57            break;
58        free(mParams[i]);
59    }
60}
61
62void NetlinkEvent::dump() {
63    int i;
64
65    for (i = 0; i < NL_PARAMS_MAX; i++) {
66        if (!mParams[i])
67            break;
68        SLOGD("NL param '%s'\n", mParams[i]);
69    }
70}
71
72/*
73 * Parse an binary message from a NETLINK_ROUTE netlink socket.
74 */
75bool NetlinkEvent::parseBinaryNetlinkMessage(char *buffer, int size) {
76    size_t sz = size;
77    const struct nlmsghdr *nh = (struct nlmsghdr *) buffer;
78
79    while (NLMSG_OK(nh, sz) && (nh->nlmsg_type != NLMSG_DONE)) {
80
81        if (nh->nlmsg_type == RTM_NEWLINK) {
82            int len = nh->nlmsg_len - sizeof(*nh);
83            struct ifinfomsg *ifi;
84
85            if (sizeof(*ifi) > (size_t) len) {
86                SLOGE("Got a short RTM_NEWLINK message\n");
87                continue;
88            }
89
90            ifi = (ifinfomsg *)NLMSG_DATA(nh);
91            if ((ifi->ifi_flags & IFF_LOOPBACK) != 0) {
92                continue;
93            }
94
95            struct rtattr *rta = (struct rtattr *)
96              ((char *) ifi + NLMSG_ALIGN(sizeof(*ifi)));
97            len = NLMSG_PAYLOAD(nh, sizeof(*ifi));
98
99            while(RTA_OK(rta, len)) {
100                switch(rta->rta_type) {
101                case IFLA_IFNAME:
102                    char buffer[16 + IFNAMSIZ];
103                    snprintf(buffer, sizeof(buffer), "INTERFACE=%s",
104                             (char *) RTA_DATA(rta));
105                    mParams[0] = strdup(buffer);
106                    mAction = (ifi->ifi_flags & IFF_LOWER_UP) ?
107                      NlActionLinkUp : NlActionLinkDown;
108                    mSubsystem = strdup("net");
109                    break;
110                }
111
112                rta = RTA_NEXT(rta, len);
113            }
114
115        } else if (nh->nlmsg_type == QLOG_NL_EVENT) {
116            char *devname;
117            ulog_packet_msg_t *pm;
118            size_t len = nh->nlmsg_len - sizeof(*nh);
119            if (sizeof(*pm) > len) {
120                SLOGE("Got a short QLOG message\n");
121                continue;
122            }
123            pm = (ulog_packet_msg_t *)NLMSG_DATA(nh);
124            devname = pm->indev_name[0] ? pm->indev_name : pm->outdev_name;
125            SLOGD("QLOG prefix=%s dev=%s\n", pm->prefix, devname);
126            asprintf(&mParams[0], "ALERT_NAME=%s", pm->prefix);
127            asprintf(&mParams[1], "INTERFACE=%s", devname);
128            mSubsystem = strdup("qlog");
129            mAction = NlActionChange;
130
131        } else {
132                SLOGD("Unexpected netlink message. type=0x%x\n", nh->nlmsg_type);
133        }
134        nh = NLMSG_NEXT(nh, size);
135    }
136
137    return true;
138}
139
140/* If the string between 'str' and 'end' begins with 'prefixlen' characters
141 * from the 'prefix' array, then return 'str + prefixlen', otherwise return
142 * NULL.
143 */
144static const char*
145has_prefix(const char* str, const char* end, const char* prefix, size_t prefixlen)
146{
147    if ((end-str) >= (ptrdiff_t)prefixlen && !memcmp(str, prefix, prefixlen))
148        return str + prefixlen;
149    else
150        return NULL;
151}
152
153/* Same as strlen(x) for constant string literals ONLY */
154#define CONST_STRLEN(x)  (sizeof(x)-1)
155
156/* Convenience macro to call has_prefix with a constant string literal  */
157#define HAS_CONST_PREFIX(str,end,prefix)  has_prefix((str),(end),prefix,CONST_STRLEN(prefix))
158
159
160/*
161 * Parse an ASCII-formatted message from a NETLINK_KOBJECT_UEVENT
162 * netlink socket.
163 */
164bool NetlinkEvent::parseAsciiNetlinkMessage(char *buffer, int size) {
165    const char *s = buffer;
166    const char *end;
167    int param_idx = 0;
168    int i;
169    int first = 1;
170
171    if (size == 0)
172        return false;
173
174    /* Ensure the buffer is zero-terminated, the code below depends on this */
175    buffer[size-1] = '\0';
176
177    end = s + size;
178    while (s < end) {
179        if (first) {
180            const char *p;
181            /* buffer is 0-terminated, no need to check p < end */
182            for (p = s; *p != '@'; p++) {
183                if (!*p) { /* no '@', should not happen */
184                    return false;
185                }
186            }
187            mPath = strdup(p+1);
188            first = 0;
189        } else {
190            const char* a;
191            if ((a = HAS_CONST_PREFIX(s, end, "ACTION=")) != NULL) {
192                if (!strcmp(a, "add"))
193                    mAction = NlActionAdd;
194                else if (!strcmp(a, "remove"))
195                    mAction = NlActionRemove;
196                else if (!strcmp(a, "change"))
197                    mAction = NlActionChange;
198            } else if ((a = HAS_CONST_PREFIX(s, end, "SEQNUM=")) != NULL) {
199                mSeq = atoi(a);
200            } else if ((a = HAS_CONST_PREFIX(s, end, "SUBSYSTEM=")) != NULL) {
201                mSubsystem = strdup(a);
202            } else if (param_idx < NL_PARAMS_MAX) {
203                mParams[param_idx++] = strdup(s);
204            }
205        }
206        s += strlen(s) + 1;
207    }
208    return true;
209}
210
211bool NetlinkEvent::decode(char *buffer, int size, int format) {
212    if (format == NetlinkListener::NETLINK_FORMAT_BINARY) {
213        return parseBinaryNetlinkMessage(buffer, size);
214    } else {
215        return parseAsciiNetlinkMessage(buffer, size);
216    }
217}
218
219const char *NetlinkEvent::findParam(const char *paramName) {
220    size_t len = strlen(paramName);
221    for (int i = 0; i < NL_PARAMS_MAX && mParams[i] != NULL; ++i) {
222        const char *ptr = mParams[i] + len;
223        if (!strncmp(mParams[i], paramName, len) && *ptr == '=')
224            return ++ptr;
225    }
226
227    SLOGE("NetlinkEvent::FindParam(): Parameter '%s' not found", paramName);
228    return NULL;
229}
230