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>
19f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
20f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/socket.h>
21f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/select.h>
22f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/time.h>
23f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/types.h>
24f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/un.h>
25f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
26f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <linux/netlink.h>
27f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
28f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#define LOG_TAG "Vold"
29f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
30f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <cutils/log.h>
31f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
32f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include "NetlinkManager.h"
33f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include "NetlinkHandler.h"
34f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
35f1b736bc5605e92e917ab27f5abf3ba839be2270San MehatNetlinkManager *NetlinkManager::sInstance = NULL;
36f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
37f1b736bc5605e92e917ab27f5abf3ba839be2270San MehatNetlinkManager *NetlinkManager::Instance() {
38f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (!sInstance)
39f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat        sInstance = new NetlinkManager();
40f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    return sInstance;
41f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
42f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
43f1b736bc5605e92e917ab27f5abf3ba839be2270San MehatNetlinkManager::NetlinkManager() {
44f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    mBroadcaster = NULL;
45f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
46f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
47f1b736bc5605e92e917ab27f5abf3ba839be2270San MehatNetlinkManager::~NetlinkManager() {
48f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
49f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
50f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehatint NetlinkManager::start() {
51f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    struct sockaddr_nl nladdr;
52f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    int sz = 64 * 1024;
53c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich    int on = 1;
54f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
55f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    memset(&nladdr, 0, sizeof(nladdr));
56f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    nladdr.nl_family = AF_NETLINK;
57f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    nladdr.nl_pid = getpid();
58f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    nladdr.nl_groups = 0xffffffff;
59f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
60f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if ((mSock = socket(PF_NETLINK,
61f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat                        SOCK_DGRAM,NETLINK_KOBJECT_UEVENT)) < 0) {
6297ac40e4e6f3ed0bd5d6878d7d8d4a54fcaecb76San Mehat        SLOGE("Unable to create uevent socket: %s", strerror(errno));
63f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat        return -1;
64f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
65f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
66f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (setsockopt(mSock, SOL_SOCKET, SO_RCVBUFFORCE, &sz, sizeof(sz)) < 0) {
67c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich        SLOGE("Unable to set uevent socket SO_RECBUFFORCE option: %s", strerror(errno));
68c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich        return -1;
69c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich    }
70c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich
71c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich    if (setsockopt(mSock, SOL_SOCKET, SO_PASSCRED, &on, sizeof(on)) < 0) {
72c51920c82463b240e2be0430849837d6fdc5352eNick Kralevich        SLOGE("Unable to set uevent socket SO_PASSCRED option: %s", strerror(errno));
73f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat        return -1;
74f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
75f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
76f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (bind(mSock, (struct sockaddr *) &nladdr, sizeof(nladdr)) < 0) {
7797ac40e4e6f3ed0bd5d6878d7d8d4a54fcaecb76San Mehat        SLOGE("Unable to bind uevent socket: %s", strerror(errno));
78f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat        return -1;
79f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
80f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
81f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    mHandler = new NetlinkHandler(mSock);
82f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (mHandler->start()) {
8397ac40e4e6f3ed0bd5d6878d7d8d4a54fcaecb76San Mehat        SLOGE("Unable to start NetlinkHandler: %s", strerror(errno));
84f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat        return -1;
85f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
86f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    return 0;
87f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
88f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
89f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehatint NetlinkManager::stop() {
90f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    if (mHandler->stop()) {
9197ac40e4e6f3ed0bd5d6878d7d8d4a54fcaecb76San Mehat        SLOGE("Unable to stop NetlinkHandler: %s", strerror(errno));
92f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat        return -1;
93f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    }
94f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    delete mHandler;
95f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    mHandler = NULL;
96f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
97f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    close(mSock);
98f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    mSock = -1;
99f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
100f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat    return 0;
101f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
102