server/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
  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,
 105                                      RTMGRP_LINK |
 106                                      RTMGRP_IPV4_IFADDR |
 107                                      RTMGRP_IPV6_IFADDR |
 108                                      (1 << (RTNLGRP_ND_USEROPT - 1)),
 109          NetlinkListener::NETLINK_FORMAT_BINARY)) == NULL) {
 110         return -1;
 111     }
 112
 113     if ((mQuotaHandler = setupSocket(&mQuotaSock, NETLINK_NFLOG,
 114         NFLOG_QUOTA_GROUP, NetlinkListener::NETLINK_FORMAT_BINARY)) == NULL) {
 115         ALOGE("Unable to open quota2 logging socket");
 116         // TODO: return -1 once the emulator gets a new kernel.
 117     }
 118
 119     return 0;
 120 }
 121
 122 int NetlinkManager::stop() {
 123     int status = 0;
 124
 125     if (mUeventHandler->stop()) {
 126         ALOGE("Unable to stop uevent NetlinkHandler: %s", strerror(errno));
 127         status = -1;
 128     }
 129
 130     delete mUeventHandler;
 131     mUeventHandler = NULL;
 132
 133     close(mUeventSock);
 134     mUeventSock = -1;
 135
 136     if (mRouteHandler->stop()) {
 137         ALOGE("Unable to stop route NetlinkHandler: %s", strerror(errno));
 138         status = -1;
 139     }
 140
 141     delete mRouteHandler;
 142     mRouteHandler = NULL;
 143
 144     close(mRouteSock);
 145     mRouteSock = -1;
 146
 147     if (mQuotaHandler) {
 148         if (mQuotaHandler->stop()) {
 149             ALOGE("Unable to stop quota NetlinkHandler: %s", strerror(errno));
 150             status = -1;
 151         }
 152
 153         delete mQuotaHandler;
 154         mQuotaHandler = NULL;
 155
 156         close(mQuotaSock);
 157         mQuotaSock = -1;
 158     }
 159
 160     return status;
 161 }