bluez/android/main.c

/*
 *
 *  BlueZ - Bluetooth protocol stack for Linux
 *
 *  Copyright (C) 2013  Intel Corporation. All rights reserved.
 *
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdbool.h>
#include <signal.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>

#include <sys/signalfd.h>
#include <sys/socket.h>
#include <sys/un.h>

#include <glib.h>

#include "log.h"
#include "src/sdpd.h"

#include "lib/bluetooth.h"

#include "bluetooth.h"
#include "socket.h"
#include "hidhost.h"
#include "hal-msg.h"
#include "ipc.h"
#include "a2dp.h"
#include "pan.h"

/* TODO: Consider to remove PLATFORM_SDKVERSION check if requirement
*  for minimal Android platform version increases. */
#if defined(ANDROID) && PLATFORM_SDK_VERSION >= 18
#include <sys/capability.h>
#endif

#define STARTUP_GRACE_SECONDS 5
#define SHUTDOWN_GRACE_SECONDS 10

static guint bluetooth_start_timeout = 0;

static bdaddr_t adapter_bdaddr;

static GMainLoop *event_loop;

static GIOChannel *hal_cmd_io = NULL;
static GIOChannel *hal_notif_io = NULL;

static bool services[HAL_SERVICE_ID_MAX + 1] = { false };

static void service_register(void *buf, uint16_t len)
{
	struct hal_cmd_register_module *m = buf;
	int sk = g_io_channel_unix_get_fd(hal_notif_io);

	if (m->service_id > HAL_SERVICE_ID_MAX || services[m->service_id])
		goto failed;

	switch (m->service_id) {
	case HAL_SERVICE_ID_BLUETOOTH:
		if (!bt_bluetooth_register(sk))
			goto failed;

		break;
	case HAL_SERVICE_ID_SOCK:
		if (!bt_socket_register(sk, &adapter_bdaddr))
			goto failed;

		break;
	case HAL_SERVICE_ID_HIDHOST:
		if (!bt_hid_register(sk, &adapter_bdaddr))
			goto failed;

		break;
	case HAL_SERVICE_ID_A2DP:
		if (!bt_a2dp_register(sk, &adapter_bdaddr))
			goto failed;

		break;
	case HAL_SERVICE_ID_PAN:
		if (!bt_pan_register(sk, &adapter_bdaddr))
			goto failed;

		break;
	default:
		DBG("service %u not supported", m->service_id);
		goto failed;
	}

	services[m->service_id] = true;

	ipc_send(g_io_channel_unix_get_fd(hal_cmd_io), HAL_SERVICE_ID_CORE,
					HAL_OP_REGISTER_MODULE, 0, NULL, -1);

	info("Service ID=%u registered", m->service_id);
	return;
failed:
	ipc_send_rsp(g_io_channel_unix_get_fd(hal_cmd_io),
				HAL_SERVICE_ID_CORE, HAL_OP_REGISTER_MODULE,
				HAL_STATUS_FAILED);
}

static void service_unregister(void *buf, uint16_t len)
{
	struct hal_cmd_unregister_module *m = buf;

	if (m->service_id > HAL_SERVICE_ID_MAX || !services[m->service_id])
		goto failed;

	switch (m->service_id) {
	case HAL_SERVICE_ID_BLUETOOTH:
		bt_bluetooth_unregister();
		break;
	case HAL_SERVICE_ID_SOCK:
		bt_socket_unregister();
		break;
	case HAL_SERVICE_ID_HIDHOST:
		bt_hid_unregister();
		break;
	case HAL_SERVICE_ID_A2DP:
		bt_a2dp_unregister();
		break;
	case HAL_SERVICE_ID_PAN:
		bt_pan_unregister();
		break;
	default:
		/* This would indicate bug in HAL, as unregister should not be
		 * called in init failed */
		DBG("service %u not supported", m->service_id);
		goto failed;
	}

	services[m->service_id] = false;

	ipc_send(g_io_channel_unix_get_fd(hal_cmd_io), HAL_SERVICE_ID_CORE,
					HAL_OP_UNREGISTER_MODULE, 0, NULL, -1);

	info("Service ID=%u unregistered", m->service_id);
	return;
failed:
	ipc_send_rsp(g_io_channel_unix_get_fd(hal_cmd_io),
				HAL_SERVICE_ID_CORE, HAL_OP_UNREGISTER_MODULE,
				HAL_STATUS_FAILED);
}

static void handle_service_core(uint8_t opcode, void *buf, uint16_t len)
{
	switch (opcode) {
	case HAL_OP_REGISTER_MODULE:
		service_register(buf, len);
		break;
	case HAL_OP_UNREGISTER_MODULE:
		service_unregister(buf, len);
		break;
	default:
		ipc_send_rsp(g_io_channel_unix_get_fd(hal_cmd_io),
						HAL_SERVICE_ID_CORE, opcode,
						HAL_STATUS_FAILED);
		break;
	}
}

static void bluetooth_stopped(void)
{
	g_main_loop_quit(event_loop);
}

static gboolean quit_eventloop(gpointer user_data)
{
	g_main_loop_quit(event_loop);
	return FALSE;
}

static void stop_bluetooth(void)
{
	static bool __stop = false;

	if (__stop)
		return;

	__stop = true;

	if (!bt_bluetooth_stop(bluetooth_stopped)) {
		g_main_loop_quit(event_loop);
		return;
	}

	g_timeout_add_seconds(SHUTDOWN_GRACE_SECONDS, quit_eventloop, NULL);
}

static gboolean cmd_watch_cb(GIOChannel *io, GIOCondition cond,
							gpointer user_data)
{
	char buf[BLUEZ_HAL_MTU];
	struct hal_hdr *msg = (void *) buf;
	ssize_t ret;
	int fd;

	if (cond & (G_IO_NVAL | G_IO_ERR | G_IO_HUP)) {
		info("HAL command socket closed, terminating");
		goto fail;
	}

	fd = g_io_channel_unix_get_fd(io);

	ret = read(fd, buf, sizeof(buf));
	if (ret < 0) {
		error("HAL command read failed, terminating (%s)",
							strerror(errno));
		goto fail;
	}

	if (ret < (ssize_t) sizeof(*msg)) {
		error("HAL command too small, terminating (%zd)", ret);
		goto fail;
	}

	if (ret != (ssize_t) (sizeof(*msg) + msg->len)) {
		error("Malformed HAL command (%zd bytes), terminating", ret);
		goto fail;
	}

	DBG("service_id %u opcode %u len %u", msg->service_id, msg->opcode,
								msg->len);

	if (msg->service_id > HAL_SERVICE_ID_MAX ||
						!services[msg->service_id]) {
		error("HAL command for unregistered service %u, terminating",
							msg->service_id);
		goto fail;
	}

	switch (msg->service_id) {
	case HAL_SERVICE_ID_CORE:
		handle_service_core(msg->opcode, msg->payload, msg->len);
		break;
	case HAL_SERVICE_ID_BLUETOOTH:
		bt_bluetooth_handle_cmd(fd, msg->opcode, msg->payload,
								msg->len);
		break;
	case HAL_SERVICE_ID_HIDHOST:
		bt_hid_handle_cmd(fd, msg->opcode, msg->payload, msg->len);
		break;
	case HAL_SERVICE_ID_SOCK:
		bt_sock_handle_cmd(fd, msg->opcode, msg->payload, msg->len);
		break;
	case HAL_SERVICE_ID_A2DP:
		bt_a2dp_handle_cmd(fd, msg->opcode, msg->payload, msg->len);
		break;
	case HAL_SERVICE_ID_PAN:
		bt_pan_handle_cmd(fd, msg->opcode, msg->payload, msg->len);
		break;
	default:
		ipc_send_rsp(fd, msg->service_id, msg->opcode,
							HAL_STATUS_FAILED);
		break;
	}

	return TRUE;

fail:
	stop_bluetooth();
	return FALSE;
}

static gboolean notif_watch_cb(GIOChannel *io, GIOCondition cond,
							gpointer user_data)
{
	info("HAL notification socket closed, terminating");
	stop_bluetooth();

	return FALSE;
}

static GIOChannel *connect_hal(GIOFunc connect_cb)
{
	struct sockaddr_un addr;
	GIOCondition cond;
	GIOChannel *io;
	int sk;

	sk = socket(PF_LOCAL, SOCK_SEQPACKET, 0);
	if (sk < 0) {
		error("Failed to create socket: %d (%s)", errno,
							strerror(errno));
		return NULL;
	}

	io = g_io_channel_unix_new(sk);

	g_io_channel_set_close_on_unref(io, TRUE);
	g_io_channel_set_flags(io, G_IO_FLAG_NONBLOCK, NULL);

	memset(&addr, 0, sizeof(addr));
	addr.sun_family = AF_UNIX;

	memcpy(addr.sun_path, BLUEZ_HAL_SK_PATH, sizeof(BLUEZ_HAL_SK_PATH));

	if (connect(sk, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
		error("Failed to connect HAL socket: %d (%s)", errno,
							strerror(errno));
		g_io_channel_unref(io);
		return NULL;
	}

	cond = G_IO_OUT | G_IO_ERR | G_IO_HUP | G_IO_NVAL;

	g_io_add_watch(io, cond, connect_cb, NULL);

	return io;
}

static gboolean notif_connect_cb(GIOChannel *io, GIOCondition cond,
							gpointer user_data)
{
	DBG("");

	if (cond & (G_IO_NVAL | G_IO_ERR | G_IO_HUP)) {
		stop_bluetooth();
		return FALSE;
	}

	cond = G_IO_ERR | G_IO_HUP | G_IO_NVAL;

	g_io_add_watch(io, cond, notif_watch_cb, NULL);

	cond = G_IO_IN | G_IO_ERR | G_IO_HUP | G_IO_NVAL;

	g_io_add_watch(hal_cmd_io, cond, cmd_watch_cb, NULL);

	info("Successfully connected to HAL");

	return FALSE;
}

static gboolean cmd_connect_cb(GIOChannel *io, GIOCondition cond,
							gpointer user_data)
{
	DBG("");

	if (cond & (G_IO_NVAL | G_IO_ERR | G_IO_HUP)) {
		stop_bluetooth();
		return FALSE;
	}

	hal_notif_io = connect_hal(notif_connect_cb);
	if (!hal_notif_io) {
		error("Cannot connect to HAL, terminating");
		stop_bluetooth();
	}

	return FALSE;
}

static void adapter_ready(int err, const bdaddr_t *addr)
{
	if (err < 0) {
		error("Adapter initialization failed: %s", strerror(-err));
		exit(EXIT_FAILURE);
	}

	bacpy(&adapter_bdaddr, addr);

	if (bluetooth_start_timeout > 0) {
		g_source_remove(bluetooth_start_timeout);
		bluetooth_start_timeout = 0;
	}

	info("Adapter initialized");

	hal_cmd_io = connect_hal(cmd_connect_cb);
	if (!hal_cmd_io) {
		error("Cannot connect to HAL, terminating");
		stop_bluetooth();
	}
}

static gboolean signal_handler(GIOChannel *channel, GIOCondition cond,
							gpointer user_data)
{
	static bool __terminated = false;
	struct signalfd_siginfo si;
	ssize_t result;
	int fd;

	if (cond & (G_IO_NVAL | G_IO_ERR | G_IO_HUP))
		return FALSE;

	fd = g_io_channel_unix_get_fd(channel);

	result = read(fd, &si, sizeof(si));
	if (result != sizeof(si))
		return FALSE;

	switch (si.ssi_signo) {
	case SIGINT:
	case SIGTERM:
		if (!__terminated) {
			info("Terminating");
			stop_bluetooth();
		}

		__terminated = true;
		break;
	}

	return TRUE;
}

static guint setup_signalfd(void)
{
	GIOChannel *channel;
	guint source;
	sigset_t mask;
	int fd;

	sigemptyset(&mask);
	sigaddset(&mask, SIGINT);
	sigaddset(&mask, SIGTERM);

	if (sigprocmask(SIG_BLOCK, &mask, NULL) < 0) {
		perror("Failed to set signal mask");
		return 0;
	}

	fd = signalfd(-1, &mask, 0);
	if (fd < 0) {
		perror("Failed to create signal descriptor");
		return 0;
	}

	channel = g_io_channel_unix_new(fd);

	g_io_channel_set_close_on_unref(channel, TRUE);
	g_io_channel_set_encoding(channel, NULL, NULL);
	g_io_channel_set_buffered(channel, FALSE);

	source = g_io_add_watch(channel,
				G_IO_IN | G_IO_HUP | G_IO_ERR | G_IO_NVAL,
				signal_handler, NULL);

	g_io_channel_unref(channel);

	return source;
}

static gboolean option_version = FALSE;
static gint option_index = -1;

static GOptionEntry options[] = {
	{ "version", 'v', 0, G_OPTION_ARG_NONE, &option_version,
				"Show version information and exit", NULL },
	{ "index", 'i', 0, G_OPTION_ARG_INT, &option_index,
				"Use specified controller", "INDEX"},
	{ NULL }
};

static void cleanup_hal_connection(void)
{
	if (hal_cmd_io) {
		g_io_channel_shutdown(hal_cmd_io, TRUE, NULL);
		g_io_channel_unref(hal_cmd_io);
		hal_cmd_io = NULL;
	}

	if (hal_notif_io) {
		g_io_channel_shutdown(hal_notif_io, TRUE, NULL);
		g_io_channel_unref(hal_notif_io);
		hal_notif_io = NULL;
	}
}

static bool set_capabilities(void)
{
#if defined(ANDROID)
	struct __user_cap_header_struct header;
	struct __user_cap_data_struct cap;

	header.version = _LINUX_CAPABILITY_VERSION;
	header.pid = 0;

	cap.effective = cap.permitted =
		CAP_TO_MASK(CAP_NET_ADMIN) |
		CAP_TO_MASK(CAP_NET_BIND_SERVICE);
	cap.inheritable = 0;

	/* TODO: Move to cap_set_proc once bionic support it */
	if (capset(&header, &cap) < 0) {
		error("%s: capset(): %s", __func__, strerror(errno));
		return false;
	}

	/* TODO: Move to cap_get_proc once bionic support it */
	if (capget(&header, &cap) < 0) {
		error("%s: capget(): %s", __func__, strerror(errno));
		return false;
	}

	DBG("Caps: eff: 0x%x, perm: 0x%x, inh: 0x%x", cap.effective,
					cap.permitted, cap.inheritable);

#endif
	return true;
}

int main(int argc, char *argv[])
{
	GOptionContext *context;
	GError *err = NULL;
	guint signal;

	/* Core Service (ID=0) should always be considered registered */
	services[0] = true;

	context = g_option_context_new(NULL);
	g_option_context_add_main_entries(context, options, NULL);

	if (g_option_context_parse(context, &argc, &argv, &err) == FALSE) {
		if (err != NULL) {
			g_printerr("%s\n", err->message);
			g_error_free(err);
		} else
			g_printerr("An unknown error occurred\n");

		exit(EXIT_FAILURE);
	}

	g_option_context_free(context);

	if (option_version == TRUE) {
		printf("%s\n", VERSION);
		exit(EXIT_SUCCESS);
	}

	event_loop = g_main_loop_new(NULL, FALSE);
	signal = setup_signalfd();
	if (!signal)
		return EXIT_FAILURE;

	__btd_log_init("*", 0);

	if (!set_capabilities())
		return EXIT_FAILURE;

	bluetooth_start_timeout = g_timeout_add_seconds(STARTUP_GRACE_SECONDS,
							quit_eventloop, NULL);
	if (bluetooth_start_timeout == 0) {
		error("Failed to init startup timeout");
		return EXIT_FAILURE;
	}

	if (!bt_bluetooth_start(option_index, adapter_ready))
		return EXIT_FAILURE;

	/* Use params: mtu = 0, flags = 0 */
	start_sdp_server(0, 0);

	DBG("Entering main loop");

	g_main_loop_run(event_loop);

	g_source_remove(signal);

	cleanup_hal_connection();
	stop_sdp_server();
	bt_bluetooth_cleanup();
	g_main_loop_unref(event_loop);

	info("Exit");

	__btd_log_cleanup();

	return EXIT_SUCCESS;
}