Home | History | Annotate | Download | only in netd 
      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 #include <stdio.h>
     18 #include <errno.h>
     19 
     20 #include <sys/socket.h>
     21 #include <sys/select.h>
     22 #include <sys/time.h>
     23 #include <sys/types.h>
     24 #include <sys/un.h>
     25 
     26 #include <linux/netlink.h>
     27 #include <linux/rtnetlink.h>
     28 
     29 #define LOG_TAG "Netd"
     30 
     31 #include <cutils/log.h>
     32 
     33 #include "NetlinkManager.h"
     34 #include "NetlinkHandler.h"
     35 
     36 const int NetlinkManager::NFLOG_QUOTA_GROUP = 1;
     37 
     38 NetlinkManager *NetlinkManager::sInstance = NULL;
     39 
     40 NetlinkManager *NetlinkManager::Instance() {
     41     if (!sInstance)
     42         sInstance = new NetlinkManager();
     43     return sInstance;
     44 }
     45 
     46 NetlinkManager::NetlinkManager() {
     47     mBroadcaster = NULL;
     48 }
     49 
     50 NetlinkManager::~NetlinkManager() {
     51 }
     52 
     53 NetlinkHandler *NetlinkManager::setupSocket(int *sock, int netlinkFamily,
     54     int groups, int format) {
     55 
     56     struct sockaddr_nl nladdr;
     57     int sz = 64 * 1024;
     58     int on = 1;
     59 
     60     memset(&nladdr, 0, sizeof(nladdr));
     61     nladdr.nl_family = AF_NETLINK;
     62     nladdr.nl_pid = getpid();
     63     nladdr.nl_groups = groups;
     64 
     65     if ((*sock = socket(PF_NETLINK, SOCK_DGRAM, netlinkFamily)) < 0) {
     66         ALOGE("Unable to create netlink socket: %s", strerror(errno));
     67         return NULL;
     68     }
     69 
     70     if (setsockopt(*sock, SOL_SOCKET, SO_RCVBUFFORCE, &sz, sizeof(sz)) < 0) {
     71         ALOGE("Unable to set uevent socket SO_RCVBUFFORCE option: %s", strerror(errno));
     72         close(*sock);
     73         return NULL;
     74     }
     75 
     76     if (setsockopt(*sock, SOL_SOCKET, SO_PASSCRED, &on, sizeof(on)) < 0) {
     77         SLOGE("Unable to set uevent socket SO_PASSCRED option: %s", strerror(errno));
     78         close(*sock);
     79         return NULL;
     80     }
     81 
     82     if (bind(*sock, (struct sockaddr *) &nladdr, sizeof(nladdr)) < 0) {
     83         ALOGE("Unable to bind netlink socket: %s", strerror(errno));
     84         close(*sock);
     85         return NULL;
     86     }
     87 
     88     NetlinkHandler *handler = new NetlinkHandler(this, *sock, format);
     89     if (handler->start()) {
     90         ALOGE("Unable to start NetlinkHandler: %s", strerror(errno));
     91         close(*sock);
     92         return NULL;
     93     }
     94 
     95     return handler;
     96 }
     97 
     98 int NetlinkManager::start() {
     99     if ((mUeventHandler = setupSocket(&mUeventSock, NETLINK_KOBJECT_UEVENT,
    100          0xffffffff, NetlinkListener::NETLINK_FORMAT_ASCII)) == NULL) {
    101         return -1;
    102     }
    103 
    104     if ((mRouteHandler = setupSocket(&mRouteSock, NETLINK_ROUTE, RTMGRP_LINK,
    105          NetlinkListener::NETLINK_FORMAT_BINARY)) == NULL) {
    106         return -1;
    107     }
    108 
    109     if ((mQuotaHandler = setupSocket(&mQuotaSock, NETLINK_NFLOG,
    110         NFLOG_QUOTA_GROUP, NetlinkListener::NETLINK_FORMAT_BINARY)) == NULL) {
    111         ALOGE("Unable to open quota2 logging socket");
    112         // TODO: return -1 once the emulator gets a new kernel.
    113     }
    114 
    115     return 0;
    116 }
    117 
    118 int NetlinkManager::stop() {
    119     int status = 0;
    120 
    121     if (mUeventHandler->stop()) {
    122         ALOGE("Unable to stop uevent NetlinkHandler: %s", strerror(errno));
    123         status = -1;
    124     }
    125 
    126     delete mUeventHandler;
    127     mUeventHandler = NULL;
    128 
    129     close(mUeventSock);
    130     mUeventSock = -1;
    131 
    132     if (mRouteHandler->stop()) {
    133         ALOGE("Unable to stop route NetlinkHandler: %s", strerror(errno));
    134         status = -1;
    135     }
    136 
    137     delete mRouteHandler;
    138     mRouteHandler = NULL;
    139 
    140     close(mRouteSock);
    141     mRouteSock = -1;
    142 
    143     if (mQuotaHandler) {
    144         if (mQuotaHandler->stop()) {
    145             ALOGE("Unable to stop quota NetlinkHandler: %s", strerror(errno));
    146             status = -1;
    147         }
    148 
    149         delete mQuotaHandler;
    150         mQuotaHandler = NULL;
    151 
    152         close(mQuotaSock);
    153         mQuotaSock = -1;
    154     }
    155 
    156     return status;
    157 }
    158