1f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat/*
2f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * Copyright (C) 2008 The Android Open Source Project
3f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat *
4f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * Licensed under the Apache License, Version 2.0 (the "License");
5f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * you may not use this file except in compliance with the License.
6f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * You may obtain a copy of the License at
7f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat *
8f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat *      http://www.apache.org/licenses/LICENSE-2.0
9f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat *
10f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * Unless required by applicable law or agreed to in writing, software
11f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * distributed under the License is distributed on an "AS IS" BASIS,
12f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * See the License for the specific language governing permissions and
14f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * limitations under the License.
15f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat */
16f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
17f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <stdio.h>
18f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <errno.h>
1938a73c6a2014153163235368cfd95ca28139daa8Elliott Hughes#include <string.h>
20f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
21f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/socket.h>
22f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/select.h>
23f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/time.h>
24f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/types.h>
25f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/un.h>
26f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
27f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <linux/netlink.h>
28f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
29f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#define LOG_TAG "Vold"
30f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
31f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <cutils/log.h>
32f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
33f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include "NetlinkManager.h"
34f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include "NetlinkHandler.h"
35f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
36f1b736bc5605e92e917ab27f5abf3ba839be2270San MehatNetlinkManager *NetlinkManager::sInstance = NULL;
37f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
38f1b736bc5605e92e917ab27f5abf3ba839be2270San MehatNetlinkManager *NetlinkManager::Instance() {
39f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (!sInstance)
40f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat        sInstance = new NetlinkManager();
41f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    return sInstance;
42f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
43f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
44f1b736bc5605e92e917ab27f5abf3ba839be2270San MehatNetlinkManager::NetlinkManager() {
45f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    mBroadcaster = NULL;
46f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
47f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
48f1b736bc5605e92e917ab27f5abf3ba839be2270San MehatNetlinkManager::~NetlinkManager() {
49f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
50f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
51f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehatint NetlinkManager::start() {
52f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    struct sockaddr_nl nladdr;
53f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    int sz = 64 * 1024;
54c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich    int on = 1;
55f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
56f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    memset(&nladdr, 0, sizeof(nladdr));
57f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    nladdr.nl_family = AF_NETLINK;
58f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    nladdr.nl_pid = getpid();
59f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    nladdr.nl_groups = 0xffffffff;
60f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
6136801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    if ((mSock = socket(PF_NETLINK, SOCK_DGRAM | SOCK_CLOEXEC,
6236801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            NETLINK_KOBJECT_UEVENT)) < 0) {
6397ac40e4e6f3ed0bd5d6878d7d8d4a54fcaecb76San Mehat        SLOGE("Unable to create uevent socket: %s", strerror(errno));
64f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat        return -1;
65f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
66f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
67f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (setsockopt(mSock, SOL_SOCKET, SO_RCVBUFFORCE, &sz, sizeof(sz)) < 0) {
683b20c39c9023eaaad5f7eeb1130104187fa0e057Nick Kralevich        SLOGE("Unable to set uevent socket SO_RCVBUFFORCE option: %s", strerror(errno));
6915bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju        goto out;
70c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich    }
71c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich
72c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich    if (setsockopt(mSock, SOL_SOCKET, SO_PASSCRED, &on, sizeof(on)) < 0) {
73c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich        SLOGE("Unable to set uevent socket SO_PASSCRED option: %s", strerror(errno));
7415bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju        goto out;
75f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
76f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
77f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (bind(mSock, (struct sockaddr *) &nladdr, sizeof(nladdr)) < 0) {
7897ac40e4e6f3ed0bd5d6878d7d8d4a54fcaecb76San Mehat        SLOGE("Unable to bind uevent socket: %s", strerror(errno));
7915bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju        goto out;
80f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
81f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
82f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    mHandler = new NetlinkHandler(mSock);
83f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (mHandler->start()) {
8497ac40e4e6f3ed0bd5d6878d7d8d4a54fcaecb76San Mehat        SLOGE("Unable to start NetlinkHandler: %s", strerror(errno));
8515bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju        goto out;
86f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
8715bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju
88f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    return 0;
8915bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju
9015bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Juout:
9115bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju    close(mSock);
9215bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju    return -1;
93f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
94f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
95f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehatint NetlinkManager::stop() {
9615bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju    int status = 0;
9715bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju
98f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (mHandler->stop()) {
9997ac40e4e6f3ed0bd5d6878d7d8d4a54fcaecb76San Mehat        SLOGE("Unable to stop NetlinkHandler: %s", strerror(errno));
10015bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju        status = -1;
101f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
102f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    delete mHandler;
103f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    mHandler = NULL;
104f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
105f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    close(mSock);
106f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    mSock = -1;
107f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
10815bd71f82fbb48c4e99eb54dd7d465df121c3f2cKaspter Ju    return status;
109f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
110