blob: 3244d425fa90f503168d4e4a1850236ee028a194 [file] [log] [blame]
/*
*
* BlueZ - Bluetooth protocol stack for Linux
*
* Copyright (C) 2012 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 <stdio.h>
#include <errno.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdbool.h>
#include <signal.h>
#include <sys/signalfd.h>
#include <readline/readline.h>
#include <readline/history.h>
#include <glib.h>
#include <gdbus.h>
#include "monitor/uuid.h"
#include "agent.h"
#include "display.h"
/* String display constants */
#define COLORED_NEW COLOR_GREEN "NEW" COLOR_OFF
#define COLORED_CHG COLOR_YELLOW "CHG" COLOR_OFF
#define COLORED_DEL COLOR_RED "DEL" COLOR_OFF
#define PROMPT_ON COLOR_BLUE "[bluetooth]" COLOR_OFF "# "
#define PROMPT_OFF "[bluetooth]# "
static GMainLoop *main_loop;
static DBusConnection *dbus_conn;
static GDBusProxy *agent_manager;
static char *auto_register_agent = NULL;
static GDBusProxy *default_ctrl;
static GList *ctrl_list;
static GList *dev_list;
static const char * const agent_arguments[] = {
"on",
"off",
"DisplayOnly",
"DisplayYesNo",
"KeyboardDisplay",
"KeyboardOnly",
"NoInputNoOutput",
NULL
};
static void proxy_leak(gpointer data)
{
printf("Leaking proxy %p\n", data);
}
static void connect_handler(DBusConnection *connection, void *user_data)
{
rl_set_prompt(PROMPT_ON);
printf("\r");
rl_on_new_line();
rl_redisplay();
}
static void disconnect_handler(DBusConnection *connection, void *user_data)
{
rl_set_prompt(PROMPT_OFF);
printf("\r");
rl_on_new_line();
rl_redisplay();
g_list_free(ctrl_list);
ctrl_list = NULL;
default_ctrl = NULL;
g_list_free(dev_list);
dev_list = NULL;
}
static void print_adapter(GDBusProxy *proxy, const char *description)
{
DBusMessageIter iter;
const char *address, *name;
if (g_dbus_proxy_get_property(proxy, "Address", &iter) == FALSE)
return;
dbus_message_iter_get_basic(&iter, &address);
if (g_dbus_proxy_get_property(proxy, "Alias", &iter) == TRUE)
dbus_message_iter_get_basic(&iter, &name);
else
name = "<unknown>";
rl_printf("%s%s%sController %s %s %s\n",
description ? "[" : "",
description ? : "",
description ? "] " : "",
address, name,
default_ctrl == proxy ? "[default]" : "");
}
static void print_device(GDBusProxy *proxy, const char *description)
{
DBusMessageIter iter;
const char *address, *name;
if (g_dbus_proxy_get_property(proxy, "Address", &iter) == FALSE)
return;
dbus_message_iter_get_basic(&iter, &address);
if (g_dbus_proxy_get_property(proxy, "Alias", &iter) == TRUE)
dbus_message_iter_get_basic(&iter, &name);
else
name = "<unknown>";
rl_printf("%s%s%sDevice %s %s\n",
description ? "[" : "",
description ? : "",
description ? "] " : "",
address, name);
}
static void print_iter(const char *label, const char *name,
DBusMessageIter *iter)
{
dbus_bool_t valbool;
dbus_uint32_t valu32;
dbus_uint16_t valu16;
dbus_int16_t vals16;
const char *valstr;
DBusMessageIter subiter;
int type;
if (iter == NULL) {
rl_printf("%s%s is nil\n", label, name);
return;
}
switch (dbus_message_iter_get_arg_type(iter)) {
case DBUS_TYPE_INVALID:
rl_printf("%s%s is inavlid\n", label, name);
break;
case DBUS_TYPE_STRING:
case DBUS_TYPE_OBJECT_PATH:
dbus_message_iter_get_basic(iter, &valstr);
rl_printf("%s%s: %s\n", label, name, valstr);
break;
case DBUS_TYPE_BOOLEAN:
dbus_message_iter_get_basic(iter, &valbool);
rl_printf("%s%s: %s\n", label, name,
valbool == TRUE ? "yes" : "no");
break;
case DBUS_TYPE_UINT32:
dbus_message_iter_get_basic(iter, &valu32);
rl_printf("%s%s: 0x%06x\n", label, name, valu32);
break;
case DBUS_TYPE_UINT16:
dbus_message_iter_get_basic(iter, &valu16);
rl_printf("%s%s: 0x%04x\n", label, name, valu16);
break;
case DBUS_TYPE_INT16:
dbus_message_iter_get_basic(iter, &vals16);
rl_printf("%s%s: %d\n", label, name, vals16);
break;
case DBUS_TYPE_ARRAY:
dbus_message_iter_recurse(iter, &subiter);
rl_printf("%s%s:\n", label, name);
do {
type = dbus_message_iter_get_arg_type(&subiter);
if (type == DBUS_TYPE_INVALID)
break;
if (type == DBUS_TYPE_STRING) {
dbus_message_iter_get_basic(&subiter, &valstr);
rl_printf("\t%s\n", valstr);
}
dbus_message_iter_next(&subiter);
} while(true);
break;
default:
rl_printf("%s%s has unsupported type\n", label, name);
break;
}
}
static void print_property(GDBusProxy *proxy, const char *name)
{
DBusMessageIter iter;
if (g_dbus_proxy_get_property(proxy, name, &iter) == FALSE)
return;
print_iter("\t", name, &iter);
}
static void print_uuids(GDBusProxy *proxy)
{
DBusMessageIter iter, value;
if (g_dbus_proxy_get_property(proxy, "UUIDs", &iter) == FALSE)
return;
dbus_message_iter_recurse(&iter, &value);
while (dbus_message_iter_get_arg_type(&value) == DBUS_TYPE_STRING) {
const char *uuid, *text;
dbus_message_iter_get_basic(&value, &uuid);
text = uuidstr_to_str(uuid);
if (text) {
char str[26];
unsigned int n;
str[sizeof(str) - 1] = '\0';
n = snprintf(str, sizeof(str), "%s", text);
if (n > sizeof(str) - 1) {
str[sizeof(str) - 2] = '.';
str[sizeof(str) - 3] = '.';
if (str[sizeof(str) - 4] == ' ')
str[sizeof(str) - 4] = '.';
n = sizeof(str) - 1;
}
rl_printf("\tUUID: %s%*c(%s)\n",
str, 26 - n, ' ', uuid);
} else
rl_printf("\tUUID: %*c(%s)\n", 26, ' ', uuid);
dbus_message_iter_next(&value);
}
}
static gboolean device_is_child(GDBusProxy *device, GDBusProxy *master)
{
DBusMessageIter iter;
const char *adapter, *path;
if (!master)
return FALSE;
if (g_dbus_proxy_get_property(device, "Adapter", &iter) == FALSE)
return FALSE;
dbus_message_iter_get_basic(&iter, &adapter);
path = g_dbus_proxy_get_path(master);
if (!strcmp(path, adapter))
return TRUE;
return FALSE;
}
static void proxy_added(GDBusProxy *proxy, void *user_data)
{
const char *interface;
interface = g_dbus_proxy_get_interface(proxy);
if (!strcmp(interface, "org.bluez.Device1")) {
if (device_is_child(proxy, default_ctrl) == TRUE) {
dev_list = g_list_append(dev_list, proxy);
print_device(proxy, COLORED_NEW);
}
} else if (!strcmp(interface, "org.bluez.Adapter1")) {
ctrl_list = g_list_append(ctrl_list, proxy);
if (!default_ctrl)
default_ctrl = proxy;
print_adapter(proxy, COLORED_NEW);
} else if (!strcmp(interface, "org.bluez.AgentManager1")) {
if (!agent_manager) {
agent_manager = proxy;
if (auto_register_agent)
agent_register(dbus_conn, agent_manager,
auto_register_agent);
}
}
}
static void proxy_removed(GDBusProxy *proxy, void *user_data)
{
const char *interface;
interface = g_dbus_proxy_get_interface(proxy);
if (!strcmp(interface, "org.bluez.Device1")) {
if (device_is_child(proxy, default_ctrl) == TRUE) {
dev_list = g_list_remove(dev_list, proxy);
print_device(proxy, COLORED_DEL);
}
} else if (!strcmp(interface, "org.bluez.Adapter1")) {
ctrl_list = g_list_remove(ctrl_list, proxy);
print_adapter(proxy, COLORED_DEL);
if (default_ctrl == proxy) {
default_ctrl = NULL;
g_list_free(dev_list);
dev_list = NULL;
}
} else if (!strcmp(interface, "org.bluez.AgentManager1")) {
if (agent_manager == proxy) {
agent_manager = NULL;
if (auto_register_agent)
agent_unregister(dbus_conn, NULL);
}
}
}
static void property_changed(GDBusProxy *proxy, const char *name,
DBusMessageIter *iter, void *user_data)
{
const char *interface;
interface = g_dbus_proxy_get_interface(proxy);
if (!strcmp(interface, "org.bluez.Device1")) {
if (device_is_child(proxy, default_ctrl) == TRUE) {
DBusMessageIter addr_iter;
char *str;
if (g_dbus_proxy_get_property(proxy, "Address",
&addr_iter) == TRUE) {
const char *address;
dbus_message_iter_get_basic(&addr_iter,
&address);
str = g_strdup_printf("[" COLORED_CHG
"] Device %s ", address);
} else
str = g_strdup("");
print_iter(str, name, iter);
g_free(str);
}
} else if (!strcmp(interface, "org.bluez.Adapter1")) {
DBusMessageIter addr_iter;
char *str;
if (g_dbus_proxy_get_property(proxy, "Address",
&addr_iter) == TRUE) {
const char *address;
dbus_message_iter_get_basic(&addr_iter, &address);
str = g_strdup_printf("[" COLORED_CHG
"] Controller %s ", address);
} else
str = g_strdup("");
print_iter(str, name, iter);
g_free(str);
}
}
static void message_handler(DBusConnection *connection,
DBusMessage *message, void *user_data)
{
rl_printf("[SIGNAL] %s.%s\n", dbus_message_get_interface(message),
dbus_message_get_member(message));
}
static GDBusProxy *find_proxy_by_address(GList *source, const char *address)
{
GList *list;
for (list = g_list_first(source); list; list = g_list_next(list)) {
GDBusProxy *proxy = list->data;
DBusMessageIter iter;
const char *str;
if (g_dbus_proxy_get_property(proxy, "Address", &iter) == FALSE)
continue;
dbus_message_iter_get_basic(&iter, &str);
if (!strcmp(str, address))
return proxy;
}
return NULL;
}
static gboolean check_default_ctrl(void)
{
if (!default_ctrl) {
rl_printf("No default controller available\n");
return FALSE;
}
return TRUE;
}
static gboolean parse_argument_on_off(const char *arg, dbus_bool_t *value)
{
if (!arg || !strlen(arg)) {
rl_printf("Missing on/off argument\n");
return FALSE;
}
if (!strcmp(arg, "on") || !strcmp(arg, "yes")) {
*value = TRUE;
return TRUE;
}
if (!strcmp(arg, "off") || !strcmp(arg, "no")) {
*value = FALSE;
return TRUE;
}
rl_printf("Invalid argument %s\n", arg);
return FALSE;
}
static gboolean parse_argument_agent(const char *arg, dbus_bool_t *value,
const char **capability)
{
const char * const *opt;
if (arg == NULL || strlen(arg) == 0) {
rl_printf("Missing on/off/capability argument\n");
return FALSE;
}
if (strcmp(arg, "on") == 0 || strcmp(arg, "yes") == 0) {
*value = TRUE;
*capability = "";
return TRUE;
}
if (strcmp(arg, "off") == 0 || strcmp(arg, "no") == 0) {
*value = FALSE;
return TRUE;
}
for (opt = agent_arguments; *opt; opt++) {
if (strcmp(arg, *opt) == 0) {
*value = TRUE;
*capability = *opt;
return TRUE;
}
}
rl_printf("Invalid argument %s\n", arg);
return FALSE;
}
static void cmd_list(const char *arg)
{
GList *list;
for (list = g_list_first(ctrl_list); list; list = g_list_next(list)) {
GDBusProxy *proxy = list->data;
print_adapter(proxy, NULL);
}
}
static void cmd_show(const char *arg)
{
GDBusProxy *proxy;
DBusMessageIter iter;
const char *address;
if (!arg || !strlen(arg)) {
if (check_default_ctrl() == FALSE)
return;
proxy = default_ctrl;
} else {
proxy = find_proxy_by_address(ctrl_list, arg);
if (!proxy) {
rl_printf("Controller %s not available\n", arg);
return;
}
}
if (g_dbus_proxy_get_property(proxy, "Address", &iter) == FALSE)
return;
dbus_message_iter_get_basic(&iter, &address);
rl_printf("Controller %s\n", address);
print_property(proxy, "Name");
print_property(proxy, "Alias");
print_property(proxy, "Class");
print_property(proxy, "Powered");
print_property(proxy, "Discoverable");
print_property(proxy, "Pairable");
print_uuids(proxy);
print_property(proxy, "Modalias");
print_property(proxy, "Discovering");
}
static void cmd_select(const char *arg)
{
GDBusProxy *proxy;
if (!arg || !strlen(arg)) {
rl_printf("Missing controller address argument\n");
return;
}
proxy = find_proxy_by_address(ctrl_list, arg);
if (!proxy) {
rl_printf("Controller %s not available\n", arg);
return;
}
if (default_ctrl == proxy)
return;
default_ctrl = proxy;
print_adapter(proxy, NULL);
g_list_free(dev_list);
dev_list = NULL;
}
static void cmd_devices(const char *arg)
{
GList *list;
for (list = g_list_first(dev_list); list; list = g_list_next(list)) {
GDBusProxy *proxy = list->data;
print_device(proxy, NULL);
}
}
static void cmd_paired_devices(const char *arg)
{
GList *list;
for (list = g_list_first(dev_list); list; list = g_list_next(list)) {
GDBusProxy *proxy = list->data;
DBusMessageIter iter;
dbus_bool_t paired;
if (g_dbus_proxy_get_property(proxy, "Paired", &iter) == FALSE)
continue;
dbus_message_iter_get_basic(&iter, &paired);
if (!paired)
continue;
print_device(proxy, NULL);
}
}
static void generic_callback(const DBusError *error, void *user_data)
{
char *str = user_data;
if (dbus_error_is_set(error))
rl_printf("Failed to set %s: %s\n", str, error->name);
else
rl_printf("Changing %s succeeded\n", str);
}
static void cmd_system_alias(const char *arg)
{
char *name;
if (!arg || !strlen(arg)) {
rl_printf("Missing name argument\n");
return;
}
if (check_default_ctrl() == FALSE)
return;
name = g_strdup(arg);
if (g_dbus_proxy_set_property_basic(default_ctrl, "Alias",
DBUS_TYPE_STRING, &name,
generic_callback, name, g_free) == TRUE)
return;
g_free(name);
}
static void cmd_reset_alias(const char *arg)
{
char *name;
if (check_default_ctrl() == FALSE)
return;
name = g_strdup("");
if (g_dbus_proxy_set_property_basic(default_ctrl, "Alias",
DBUS_TYPE_STRING, &name,
generic_callback, name, g_free) == TRUE)
return;
g_free(name);
}
static void cmd_power(const char *arg)
{
dbus_bool_t powered;
char *str;
if (parse_argument_on_off(arg, &powered) == FALSE)
return;
if (check_default_ctrl() == FALSE)
return;
str = g_strdup_printf("power %s", powered == TRUE ? "on" : "off");
if (g_dbus_proxy_set_property_basic(default_ctrl, "Powered",
DBUS_TYPE_BOOLEAN, &powered,
generic_callback, str, g_free) == TRUE)
return;
g_free(str);
}
static void cmd_pairable(const char *arg)
{
dbus_bool_t pairable;
char *str;
if (parse_argument_on_off(arg, &pairable) == FALSE)
return;
if (check_default_ctrl() == FALSE)
return;
str = g_strdup_printf("pairable %s", pairable == TRUE ? "on" : "off");
if (g_dbus_proxy_set_property_basic(default_ctrl, "Pairable",
DBUS_TYPE_BOOLEAN, &pairable,
generic_callback, str, g_free) == TRUE)
return;
g_free(str);
}
static void cmd_discoverable(const char *arg)
{
dbus_bool_t discoverable;
char *str;
if (parse_argument_on_off(arg, &discoverable) == FALSE)
return;
if (check_default_ctrl() == FALSE)
return;
str = g_strdup_printf("discoverable %s",
discoverable == TRUE ? "on" : "off");
if (g_dbus_proxy_set_property_basic(default_ctrl, "Discoverable",
DBUS_TYPE_BOOLEAN, &discoverable,
generic_callback, str, g_free) == TRUE)
return;
g_free(str);
}
static void cmd_agent(const char *arg)
{
dbus_bool_t enable;
const char *capability;
if (parse_argument_agent(arg, &enable, &capability) == FALSE)
return;
if (enable == TRUE) {
g_free(auto_register_agent);
auto_register_agent = g_strdup(capability);
if (agent_manager)
agent_register(dbus_conn, agent_manager,
auto_register_agent);
else
rl_printf("Agent registration enabled\n");
} else {
g_free(auto_register_agent);
auto_register_agent = NULL;
if (agent_manager)
agent_unregister(dbus_conn, agent_manager);
else
rl_printf("Agent registration disabled\n");
}
}
static void cmd_default_agent(const char *arg)
{
agent_default(dbus_conn, agent_manager);
}
static void start_discovery_reply(DBusMessage *message, void *user_data)
{
dbus_bool_t enable = GPOINTER_TO_UINT(user_data);
DBusError error;
dbus_error_init(&error);
if (dbus_set_error_from_message(&error, message) == TRUE) {
rl_printf("Failed to %s discovery: %s\n",
enable == TRUE ? "start" : "stop", error.name);
dbus_error_free(&error);
return;
}
rl_printf("Discovery %s\n", enable == TRUE ? "started" : "stopped");
}
static void cmd_scan(const char *arg)
{
dbus_bool_t enable;
const char *method;
if (parse_argument_on_off(arg, &enable) == FALSE)
return;
if (check_default_ctrl() == FALSE)
return;
if (enable == TRUE)
method = "StartDiscovery";
else
method = "StopDiscovery";
if (g_dbus_proxy_method_call(default_ctrl, method,
NULL, start_discovery_reply,
GUINT_TO_POINTER(enable), NULL) == FALSE) {
rl_printf("Failed to %s discovery\n",
enable == TRUE ? "start" : "stop");
return;
}
}
static void cmd_info(const char *arg)
{
GDBusProxy *proxy;
DBusMessageIter iter;
const char *address;
if (!arg || !strlen(arg)) {
rl_printf("Missing device address argument\n");
return;
}
proxy = find_proxy_by_address(dev_list, arg);
if (!proxy) {
rl_printf("Device %s not available\n", arg);
return;
}
if (g_dbus_proxy_get_property(proxy, "Address", &iter) == FALSE)
return;
dbus_message_iter_get_basic(&iter, &address);
rl_printf("Device %s\n", address);
print_property(proxy, "Name");
print_property(proxy, "Alias");
print_property(proxy, "Class");
print_property(proxy, "Appearance");
print_property(proxy, "Icon");
print_property(proxy, "Paired");
print_property(proxy, "Trusted");
print_property(proxy, "Blocked");
print_property(proxy, "Connected");
print_property(proxy, "LegacyPairing");
print_uuids(proxy);
print_property(proxy, "Modalias");
}
static void pair_reply(DBusMessage *message, void *user_data)
{
DBusError error;
dbus_error_init(&error);
if (dbus_set_error_from_message(&error, message) == TRUE) {
rl_printf("Failed to pair: %s\n", error.name);
dbus_error_free(&error);
return;
}
rl_printf("Pairing successful\n");
}
static void cmd_pair(const char *arg)
{
GDBusProxy *proxy;
if (!arg || !strlen(arg)) {
rl_printf("Missing device address argument\n");
return;
}
proxy = find_proxy_by_address(dev_list, arg);
if (!proxy) {
rl_printf("Device %s not available\n", arg);
return;
}
if (g_dbus_proxy_method_call(proxy, "Pair", NULL, pair_reply,
NULL, NULL) == FALSE) {
rl_printf("Failed to pair\n");
return;
}
rl_printf("Attempting to pair with %s\n", arg);
}
static void cmd_trust(const char *arg)
{
GDBusProxy *proxy;
dbus_bool_t trusted;
char *str;
if (!arg || !strlen(arg)) {
rl_printf("Missing device address argument\n");
return;
}
proxy = find_proxy_by_address(dev_list, arg);
if (!proxy) {
rl_printf("Device %s not available\n", arg);
return;
}
trusted = TRUE;
str = g_strdup_printf("%s trust", arg);
if (g_dbus_proxy_set_property_basic(proxy, "Trusted",
DBUS_TYPE_BOOLEAN, &trusted,
generic_callback, str, g_free) == TRUE)
return;
g_free(str);
}
static void cmd_untrust(const char *arg)
{
GDBusProxy *proxy;
dbus_bool_t trusted;
char *str;
if (!arg || !strlen(arg)) {
rl_printf("Missing device address argument\n");
return;
}
proxy = find_proxy_by_address(dev_list, arg);
if (!proxy) {
rl_printf("Device %s not available\n", arg);
return;
}
trusted = FALSE;
str = g_strdup_printf("%s untrust", arg);
if (g_dbus_proxy_set_property_basic(proxy, "Trusted",
DBUS_TYPE_BOOLEAN, &trusted,
generic_callback, str, g_free) == TRUE)
return;
g_free(str);
}
static void cmd_block(const char *arg)
{
GDBusProxy *proxy;
dbus_bool_t blocked;
char *str;
if (!arg || !strlen(arg)) {
rl_printf("Missing device address argument\n");
return;
}
proxy = find_proxy_by_address(dev_list, arg);
if (!proxy) {
rl_printf("Device %s not available\n", arg);
return;
}
blocked = TRUE;
str = g_strdup_printf("%s block", arg);
if (g_dbus_proxy_set_property_basic(proxy, "Blocked",
DBUS_TYPE_BOOLEAN, &blocked,
generic_callback, str, g_free) == TRUE)
return;
g_free(str);
}
static void cmd_unblock(const char *arg)
{
GDBusProxy *proxy;
dbus_bool_t blocked;
char *str;
if (!arg || !strlen(arg)) {
rl_printf("Missing device address argument\n");
return;
}
proxy = find_proxy_by_address(dev_list, arg);
if (!proxy) {
rl_printf("Device %s not available\n", arg);
return;
}
blocked = FALSE;
str = g_strdup_printf("%s unblock", arg);
if (g_dbus_proxy_set_property_basic(proxy, "Blocked",
DBUS_TYPE_BOOLEAN, &blocked,
generic_callback, str, g_free) == TRUE)
return;
g_free(str);
}
static void remove_device_reply(DBusMessage *message, void *user_data)
{
DBusError error;
dbus_error_init(&error);
if (dbus_set_error_from_message(&error, message) == TRUE) {
rl_printf("Failed to remove device: %s\n", error.name);
dbus_error_free(&error);
return;
}
rl_printf("Device has been removed\n");
}
static void remove_device_setup(DBusMessageIter *iter, void *user_data)
{
const char *path = user_data;
dbus_message_iter_append_basic(iter, DBUS_TYPE_OBJECT_PATH, &path);
}
static void cmd_remove(const char *arg)
{
GDBusProxy *proxy;
char *path;
if (!arg || !strlen(arg)) {
rl_printf("Missing device address argument\n");
return;
}
if (check_default_ctrl() == FALSE)
return;
proxy = find_proxy_by_address(dev_list, arg);
if (!proxy) {
rl_printf("Device %s not available\n", arg);
return;
}
path = g_strdup(g_dbus_proxy_get_path(proxy));
if (g_dbus_proxy_method_call(default_ctrl, "RemoveDevice",
remove_device_setup,
remove_device_reply,
path, g_free) == FALSE) {
rl_printf("Failed to remove device\n");
g_free(path);
return;
}
}
static void connect_reply(DBusMessage *message, void *user_data)
{
DBusError error;
dbus_error_init(&error);
if (dbus_set_error_from_message(&error, message) == TRUE) {
rl_printf("Failed to connect: %s\n", error.name);
dbus_error_free(&error);
return;
}
rl_printf("Connection successful\n");
}
static void cmd_connect(const char *arg)
{
GDBusProxy *proxy;
if (!arg || !strlen(arg)) {
rl_printf("Missing device address argument\n");
return;
}
proxy = find_proxy_by_address(dev_list, arg);
if (!proxy) {
rl_printf("Device %s not available\n", arg);
return;
}
if (g_dbus_proxy_method_call(proxy, "Connect", NULL, connect_reply,
NULL, NULL) == FALSE) {
rl_printf("Failed to connect\n");
return;
}
rl_printf("Attempting to connect to %s\n", arg);
}
static void disconn_reply(DBusMessage *message, void *user_data)
{
DBusError error;
dbus_error_init(&error);
if (dbus_set_error_from_message(&error, message) == TRUE) {
rl_printf("Failed to disconnect: %s\n", error.name);
dbus_error_free(&error);
return;
}
rl_printf("Successful disconnected\n");
}
static void cmd_disconn(const char *arg)
{
GDBusProxy *proxy;
if (!arg || !strlen(arg)) {
rl_printf("Missing device address argument\n");
return;
}
proxy = find_proxy_by_address(dev_list, arg);
if (!proxy) {
rl_printf("Device %s not available\n", arg);
return;
}
if (g_dbus_proxy_method_call(proxy, "Disconnect", NULL, disconn_reply,
NULL, NULL) == FALSE) {
rl_printf("Failed to disconnect\n");
return;
}
rl_printf("Attempting to disconnect from %s\n", arg);
}
static void cmd_version(const char *arg)
{
rl_printf("Version %s\n", VERSION);
}
static void cmd_quit(const char *arg)
{
g_main_loop_quit(main_loop);
}
static char *generic_generator(const char *text, int state,
GList *source, const char *property)
{
static int index, len;
GList *list;
if (!state) {
index = 0;
len = strlen(text);
}
for (list = g_list_nth(source, index); list;
list = g_list_next(list)) {
GDBusProxy *proxy = list->data;
DBusMessageIter iter;
const char *str;
index++;
if (g_dbus_proxy_get_property(proxy, property, &iter) == FALSE)
continue;
dbus_message_iter_get_basic(&iter, &str);
if (!strncmp(str, text, len))
return strdup(str);
}
return NULL;
}
static char *ctrl_generator(const char *text, int state)
{
return generic_generator(text, state, ctrl_list, "Address");
}
static char *dev_generator(const char *text, int state)
{
return generic_generator(text, state, dev_list, "Address");
}
static char *capability_generator(const char *text, int state)
{
static int index, len;
const char *arg;
if (!state) {
index = 0;
len = strlen(text);
}
while ((arg = agent_arguments[index])) {
index++;
if (!strncmp(arg, text, len))
return strdup(arg);
}
return NULL;
}
static const struct {
const char *cmd;
const char *arg;
void (*func) (const char *arg);
const char *desc;
char * (*gen) (const char *text, int state);
void (*disp) (char **matches, int num_matches, int max_length);
} cmd_table[] = {
{ "list", NULL, cmd_list, "List available controllers" },
{ "show", "[ctrl]", cmd_show, "Controller information",
ctrl_generator },
{ "select", "<ctrl>", cmd_select, "Select default controller",
ctrl_generator },
{ "devices", NULL, cmd_devices, "List available devices" },
{ "paired-devices", NULL, cmd_paired_devices,
"List paired devices"},
{ "system-alias", "<name>", cmd_system_alias },
{ "reset-alias", NULL, cmd_reset_alias },
{ "power", "<on/off>", cmd_power, "Set controller power" },
{ "pairable", "<on/off>", cmd_pairable,
"Set controller pairable mode" },
{ "discoverable", "<on/off>", cmd_discoverable,
"Set controller discoverable mode" },
{ "agent", "<on/off/capability>", cmd_agent,
"Enable/disable agent with given capability",
capability_generator},
{ "default-agent",NULL, cmd_default_agent,
"Set agent as the default one" },
{ "scan", "<on/off>", cmd_scan, "Scan for devices" },
{ "info", "<dev>", cmd_info, "Device information",
dev_generator },
{ "pair", "<dev>", cmd_pair, "Pair with device",
dev_generator },
{ "trust", "<dev>", cmd_trust, "Trust device",
dev_generator },
{ "untrust", "<dev>", cmd_untrust, "Untrust device",
dev_generator },
{ "block", "<dev>", cmd_block, "Block device",
dev_generator },
{ "unblock", "<dev>", cmd_unblock, "Unblock device",
dev_generator },
{ "remove", "<dev>", cmd_remove, "Remove device",
dev_generator },
{ "connect", "<dev>", cmd_connect, "Connect device",
dev_generator },
{ "disconnect", "<dev>", cmd_disconn, "Disconnect device",
dev_generator },
{ "version", NULL, cmd_version, "Display version" },
{ "quit", NULL, cmd_quit, "Quit program" },
{ "exit", NULL, cmd_quit },
{ "help" },
{ }
};
static char *cmd_generator(const char *text, int state)
{
static int index, len;
const char *cmd;
if (!state) {
index = 0;
len = strlen(text);
}
while ((cmd = cmd_table[index].cmd)) {
index++;
if (!strncmp(cmd, text, len))
return strdup(cmd);
}
return NULL;
}
static char **cmd_completion(const char *text, int start, int end)
{
char **matches = NULL;
if (agent_completion() == TRUE) {
rl_attempted_completion_over = 1;
return NULL;
}
if (start > 0) {
int i;
for (i = 0; cmd_table[i].cmd; i++) {
if (strncmp(cmd_table[i].cmd,
rl_line_buffer, start - 1))
continue;
if (!cmd_table[i].gen)
continue;
rl_completion_display_matches_hook = cmd_table[i].disp;
matches = rl_completion_matches(text, cmd_table[i].gen);
break;
}
} else {
rl_completion_display_matches_hook = NULL;
matches = rl_completion_matches(text, cmd_generator);
}
if (!matches)
rl_attempted_completion_over = 1;
return matches;
}
static void rl_handler(char *input)
{
char *cmd, *arg;
int i;
if (!input) {
rl_insert_text("quit");
rl_redisplay();
rl_crlf();
g_main_loop_quit(main_loop);
return;
}
if (!strlen(input))
goto done;
if (agent_input(dbus_conn, input) == TRUE)
goto done;
add_history(input);
cmd = strtok_r(input, " ", &arg);
if (!cmd)
goto done;
if (arg) {
int len = strlen(arg);
if (len > 0 && arg[len - 1] == ' ')
arg[len - 1] = '\0';
}
for (i = 0; cmd_table[i].cmd; i++) {
if (strcmp(cmd, cmd_table[i].cmd))
continue;
if (cmd_table[i].func) {
cmd_table[i].func(arg);
goto done;
}
}
if (strcmp(cmd, "help")) {
printf("Invalid command\n");
goto done;
}
printf("Available commands:\n");
for (i = 0; cmd_table[i].cmd; i++) {
if (cmd_table[i].desc)
printf(" %s %-*s %s\n", cmd_table[i].cmd,
(int)(25 - strlen(cmd_table[i].cmd)),
cmd_table[i].arg ? : "",
cmd_table[i].desc ? : "");
}
done:
free(input);
}
static gboolean input_handler(GIOChannel *channel, GIOCondition condition,
gpointer user_data)
{
if (condition & G_IO_IN) {
rl_callback_read_char();
return TRUE;
}
if (condition & (G_IO_HUP | G_IO_ERR | G_IO_NVAL)) {
g_main_loop_quit(main_loop);
return FALSE;
}
return TRUE;
}
static guint setup_standard_input(void)
{
GIOChannel *channel;
guint source;
channel = g_io_channel_unix_new(fileno(stdin));
source = g_io_add_watch(channel,
G_IO_IN | G_IO_HUP | G_IO_ERR | G_IO_NVAL,
input_handler, NULL);
g_io_channel_unref(channel);
return source;
}
static gboolean signal_handler(GIOChannel *channel, GIOCondition condition,
gpointer user_data)
{
static unsigned int __terminated = 0;
struct signalfd_siginfo si;
ssize_t result;
int fd;
if (condition & (G_IO_NVAL | G_IO_ERR | G_IO_HUP)) {
g_main_loop_quit(main_loop);
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:
rl_replace_line("", 0);
rl_crlf();
rl_on_new_line();
rl_redisplay();
break;
case SIGTERM:
if (__terminated == 0) {
rl_replace_line("", 0);
rl_crlf();
g_main_loop_quit(main_loop);
}
__terminated = 1;
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 gboolean parse_agent(const char *key, const char *value,
gpointer user_data, GError **error)
{
if (value)
auto_register_agent = g_strdup(value);
else
auto_register_agent = g_strdup("");
return TRUE;
}
static GOptionEntry options[] = {
{ "version", 'v', 0, G_OPTION_ARG_NONE, &option_version,
"Show version information and exit" },
{ "agent", 'a', G_OPTION_FLAG_OPTIONAL_ARG,
G_OPTION_ARG_CALLBACK, parse_agent,
"Register agent handler", "CAPABILITY" },
{ NULL },
};
int main(int argc, char *argv[])
{
GOptionContext *context;
GError *error = NULL;
GDBusClient *client;
guint signal, input;
context = g_option_context_new(NULL);
g_option_context_add_main_entries(context, options, NULL);
if (g_option_context_parse(context, &argc, &argv, &error) == FALSE) {
if (error != NULL) {
g_printerr("%s\n", error->message);
g_error_free(error);
} else
g_printerr("An unknown error occurred\n");
exit(1);
}
g_option_context_free(context);
if (option_version == TRUE) {
printf("%s\n", VERSION);
exit(0);
}
main_loop = g_main_loop_new(NULL, FALSE);
dbus_conn = g_dbus_setup_bus(DBUS_BUS_SYSTEM, NULL, NULL);
rl_attempted_completion_function = cmd_completion;
rl_erase_empty_line = 1;
rl_callback_handler_install(NULL, rl_handler);
rl_set_prompt(PROMPT_OFF);
rl_redisplay();
input = setup_standard_input();
signal = setup_signalfd();
client = g_dbus_client_new(dbus_conn, "org.bluez", "/org/bluez");
g_dbus_client_set_connect_watch(client, connect_handler, NULL);
g_dbus_client_set_disconnect_watch(client, disconnect_handler, NULL);
g_dbus_client_set_signal_watch(client, message_handler, NULL);
g_dbus_client_set_proxy_handlers(client, proxy_added, proxy_removed,
property_changed, NULL);
g_main_loop_run(main_loop);
g_dbus_client_unref(client);
g_source_remove(signal);
g_source_remove(input);
rl_message("");
rl_callback_handler_remove();
dbus_connection_unref(dbus_conn);
g_main_loop_unref(main_loop);
g_list_free_full(ctrl_list, proxy_leak);
g_list_free_full(dev_list, proxy_leak);
g_free(auto_register_agent);
return 0;
}