NetlinkManager.cpp

   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 const int NetlinkManager::IDLETIMER_GROUP = 1;
  38
  39 NetlinkManager *NetlinkManager::sInstance = NULL;
  40
  41 NetlinkManager *NetlinkManager::Instance() {
  42     if (!sInstance)
  43         sInstance = new NetlinkManager();
  44     return sInstance;
  45 }
  46
  47 NetlinkManager::NetlinkManager() {
  48     mBroadcaster = NULL;
  49 }
  50
  51 NetlinkManager::~NetlinkManager() {
  52 }
  53
  54 NetlinkHandler *NetlinkManager::setupSocket(int *sock, int netlinkFamily,
  55     int groups, int format) {
  56
  57     struct sockaddr_nl nladdr;
  58     int sz = 64 * 1024;
  59     int on = 1;
  60
  61     memset(&nladdr, 0, sizeof(nladdr));
  62     nladdr.nl_family = AF_NETLINK;
  63     nladdr.nl_pid = getpid();
  64     nladdr.nl_groups = groups;
  65
  66     if ((*sock = socket(PF_NETLINK, SOCK_DGRAM, netlinkFamily)) < 0) {
  67         ALOGE("Unable to create netlink socket: %s", strerror(errno));
  68         return NULL;
  69     }
  70
  71     if (setsockopt(*sock, SOL_SOCKET, SO_RCVBUFFORCE, &sz, sizeof(sz)) < 0) {
  72         ALOGE("Unable to set uevent socket SO_RCVBUFFORCE option: %s", strerror(errno));
  73         close(*sock);
  74         return NULL;
  75     }
  76
  77     if (setsockopt(*sock, SOL_SOCKET, SO_PASSCRED, &on, sizeof(on)) < 0) {
  78         SLOGE("Unable to set uevent socket SO_PASSCRED option: %s", strerror(errno));
  79         close(*sock);
  80         return NULL;
  81     }
  82
  83     if (bind(*sock, (struct sockaddr *) &nladdr, sizeof(nladdr)) < 0) {
  84         ALOGE("Unable to bind netlink socket: %s", strerror(errno));
  85         close(*sock);
  86         return NULL;
  87     }
  88
  89     NetlinkHandler *handler = new NetlinkHandler(this, *sock, format);
  90     if (handler->start()) {
  91         ALOGE("Unable to start NetlinkHandler: %s", strerror(errno));
  92         close(*sock);
  93         return NULL;
  94     }
  95
  96     return handler;
  97 }
  98
  99 int NetlinkManager::start() {
 100     if ((mUeventHandler = setupSocket(&mUeventSock, NETLINK_KOBJECT_UEVENT,
 101          0xffffffff, NetlinkListener::NETLINK_FORMAT_ASCII)) == NULL) {
 102         return -1;
 103     }
 104
 105     if ((mRouteHandler = setupSocket(&mRouteSock, NETLINK_ROUTE, RTMGRP_LINK,
 106          NetlinkListener::NETLINK_FORMAT_BINARY)) == NULL) {
 107         return -1;
 108     }
 109
 110     if ((mQuotaHandler = setupSocket(&mQuotaSock, NETLINK_NFLOG,
 111         NFLOG_QUOTA_GROUP, NetlinkListener::NETLINK_FORMAT_BINARY)) == NULL) {
 112         ALOGE("Unable to open quota2 logging socket");
 113         // TODO: return -1 once the emulator gets a new kernel.
 114     }
 115
 116     if ((mIfaceIdleTimerHandler = setupSocket(&mIfaceIdleTimerSock, NETLINK_IDLETIMER,
 117         IDLETIMER_GROUP, NetlinkListener::NETLINK_FORMAT_BINARY)) == NULL) {
 118         // TODO: switch back to using NETLINK_NFLOG with a custom type.
 119         ALOGE("Unable to open iface idletimer socket");
 120     }
 121     return 0;
 122 }
 123
 124 int NetlinkManager::stop() {
 125     int status = 0;
 126
 127     if (mUeventHandler->stop()) {
 128         ALOGE("Unable to stop uevent NetlinkHandler: %s", strerror(errno));
 129         status = -1;
 130     }
 131
 132     delete mUeventHandler;
 133     mUeventHandler = NULL;
 134
 135     close(mUeventSock);
 136     mUeventSock = -1;
 137
 138     if (mRouteHandler->stop()) {
 139         ALOGE("Unable to stop route NetlinkHandler: %s", strerror(errno));
 140         status = -1;
 141     }
 142
 143     delete mRouteHandler;
 144     mRouteHandler = NULL;
 145
 146     close(mRouteSock);
 147     mRouteSock = -1;
 148
 149     if (mQuotaHandler) {
 150         if (mQuotaHandler->stop()) {
 151             ALOGE("Unable to stop quota NetlinkHandler: %s", strerror(errno));
 152             status = -1;
 153         }
 154
 155         delete mQuotaHandler;
 156         mQuotaHandler = NULL;
 157
 158         close(mQuotaSock);
 159         mQuotaSock = -1;
 160     }
 161
 162     if (mIfaceIdleTimerHandler) {
 163         if (mIfaceIdleTimerHandler->stop()) {
 164             ALOGE("Unable to stop iface IDLETIMER NetlinkHandler: %s", strerror(errno));
 165             status = -1;
 166         }
 167
 168         delete mIfaceIdleTimerHandler;
 169         mIfaceIdleTimerHandler = NULL;
 170
 171         close(mIfaceIdleTimerSock);
 172         mIfaceIdleTimerSock = -1;
 173     }
 174
 175     return status;
 176 }