bluez/client/agent.c
Vinit Mehta 60d60166e4 mgmt: Fix crash after pair command
After pair command, if the user doesn't provide any input on bluetoothctl
CLI interface after receiving the prompt(yes/no) below crash is observed:

dbus[782]: arguments to dbus_message_get_no_reply() were incorrect,
assertion "message != NULL" failed in file
/usr/src/debug/dbus/1.14.10-r0/dbus/dbus-message.c line 3250.
This is normally a bug in some application using the D-Bus library.
/usr/lib/libc.so.6(+0x27534) [0xffffa1b67534]
/usr/lib/libc.so.6(__libc_start_main+0x9c) [0xffffa1b6760c]
bluetoothctl(+0x188f0) [0xaaaac9c088f0]
Aborted (core dumped)
2024-01-03 15:59:23 -05:00

517 lines
13 KiB
C

// SPDX-License-Identifier: GPL-2.0-or-later
/*
*
* BlueZ - Bluetooth protocol stack for Linux
*
* Copyright (C) 2012 Intel Corporation. All rights reserved.
*
*
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <unistd.h>
#include <string.h>
#include <glib.h>
#include "src/shared/shell.h"
#include "gdbus/gdbus.h"
#include "agent.h"
#define AGENT_PATH "/org/bluez/agent"
#define AGENT_INTERFACE "org.bluez.Agent1"
#define AGENT_PROMPT COLOR_RED "[agent]" COLOR_OFF " "
static gboolean agent_registered = FALSE;
static const char *agent_capability = NULL;
static DBusMessage *pending_message = NULL;
static void agent_release_prompt(void)
{
if (!pending_message)
return;
bt_shell_release_prompt("");
}
dbus_bool_t agent_completion(void)
{
if (!pending_message)
return FALSE;
return TRUE;
}
static void pincode_response(const char *input, void *user_data)
{
DBusConnection *conn = user_data;
g_dbus_send_reply(conn, pending_message, DBUS_TYPE_STRING, &input,
DBUS_TYPE_INVALID);
}
static void passkey_response(const char *input, void *user_data)
{
DBusConnection *conn = user_data;
dbus_uint32_t passkey;
if (sscanf(input, "%u", &passkey) == 1)
g_dbus_send_reply(conn, pending_message, DBUS_TYPE_UINT32,
&passkey, DBUS_TYPE_INVALID);
else if (!strcmp(input, "no"))
g_dbus_send_error(conn, pending_message,
"org.bluez.Error.Rejected", NULL);
else
g_dbus_send_error(conn, pending_message,
"org.bluez.Error.Canceled", NULL);
}
static void confirm_response(const char *input, void *user_data)
{
DBusConnection *conn = user_data;
if (pending_message != NULL) {
if (!strcmp(input, "yes"))
g_dbus_send_reply(conn, pending_message,
DBUS_TYPE_INVALID);
else if (!strcmp(input, "no"))
g_dbus_send_error(conn, pending_message,
"org.bluez.Error.Rejected", NULL);
else
g_dbus_send_error(conn, pending_message,
"org.bluez.Error.Canceled", NULL);
}
}
static void agent_release(DBusConnection *conn)
{
agent_registered = FALSE;
agent_capability = NULL;
if (pending_message) {
dbus_message_unref(pending_message);
pending_message = NULL;
}
agent_release_prompt();
g_dbus_unregister_interface(conn, AGENT_PATH, AGENT_INTERFACE);
}
static DBusMessage *release_agent(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
bt_shell_printf("Agent released\n");
agent_release(conn);
return dbus_message_new_method_return(msg);
}
static DBusMessage *request_pincode(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device;
bt_shell_printf("Request PIN code\n");
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_INVALID);
bt_shell_prompt_input("agent", "Enter PIN code:", pincode_response,
conn);
pending_message = dbus_message_ref(msg);
return NULL;
}
static DBusMessage *display_pincode(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device;
const char *pincode;
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_STRING, &pincode, DBUS_TYPE_INVALID);
bt_shell_printf(AGENT_PROMPT "PIN code: %s\n", pincode);
return dbus_message_new_method_return(msg);
}
static DBusMessage *request_passkey(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device;
bt_shell_printf("Request passkey\n");
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_INVALID);
bt_shell_prompt_input("agent", "Enter passkey (number in 0-999999):",
passkey_response, conn);
pending_message = dbus_message_ref(msg);
return NULL;
}
static DBusMessage *display_passkey(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device;
dbus_uint32_t passkey;
dbus_uint16_t entered;
char passkey_full[7];
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_UINT32, &passkey, DBUS_TYPE_UINT16, &entered,
DBUS_TYPE_INVALID);
snprintf(passkey_full, sizeof(passkey_full), "%.6u", passkey);
passkey_full[6] = '\0';
if (entered > strlen(passkey_full))
entered = strlen(passkey_full);
bt_shell_printf(AGENT_PROMPT "Passkey: "
COLOR_BOLDGRAY "%.*s" COLOR_BOLDWHITE "%s\n" COLOR_OFF,
entered, passkey_full, passkey_full + entered);
return dbus_message_new_method_return(msg);
}
static DBusMessage *request_confirmation(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device;
dbus_uint32_t passkey;
char *str;
bt_shell_printf("Request confirmation\n");
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_UINT32, &passkey, DBUS_TYPE_INVALID);
str = g_strdup_printf("Confirm passkey %06u (yes/no):", passkey);
bt_shell_prompt_input("agent", str, confirm_response, conn);
g_free(str);
pending_message = dbus_message_ref(msg);
return NULL;
}
static DBusMessage *request_authorization(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device;
bt_shell_printf("Request authorization\n");
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_INVALID);
bt_shell_prompt_input("agent", "Accept pairing (yes/no):",
confirm_response, conn);
pending_message = dbus_message_ref(msg);
return NULL;
}
static DBusMessage *authorize_service(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device, *uuid;
char *str;
bt_shell_printf("Authorize service\n");
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_STRING, &uuid, DBUS_TYPE_INVALID);
str = g_strdup_printf("Authorize service %s (yes/no):", uuid);
bt_shell_prompt_input("agent", str, confirm_response, conn);
g_free(str);
pending_message = dbus_message_ref(msg);
return NULL;
}
static DBusMessage *cancel_request(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
bt_shell_printf("Request canceled\n");
agent_release_prompt();
dbus_message_unref(pending_message);
pending_message = NULL;
return dbus_message_new_method_return(msg);
}
static const GDBusMethodTable agent_methods[] = {
{ GDBUS_METHOD("Release", NULL, NULL, release_agent) },
{ GDBUS_ASYNC_METHOD("RequestPinCode",
GDBUS_ARGS({ "device", "o" }),
GDBUS_ARGS({ "pincode", "s" }), request_pincode) },
{ GDBUS_METHOD("DisplayPinCode",
GDBUS_ARGS({ "device", "o" }, { "pincode", "s" }),
NULL, display_pincode) },
{ GDBUS_ASYNC_METHOD("RequestPasskey",
GDBUS_ARGS({ "device", "o" }),
GDBUS_ARGS({ "passkey", "u" }), request_passkey) },
{ GDBUS_METHOD("DisplayPasskey",
GDBUS_ARGS({ "device", "o" }, { "passkey", "u" },
{ "entered", "q" }),
NULL, display_passkey) },
{ GDBUS_ASYNC_METHOD("RequestConfirmation",
GDBUS_ARGS({ "device", "o" }, { "passkey", "u" }),
NULL, request_confirmation) },
{ GDBUS_ASYNC_METHOD("RequestAuthorization",
GDBUS_ARGS({ "device", "o" }),
NULL, request_authorization) },
{ GDBUS_ASYNC_METHOD("AuthorizeService",
GDBUS_ARGS({ "device", "o" }, { "uuid", "s" }),
NULL, authorize_service) },
{ GDBUS_METHOD("Cancel", NULL, NULL, cancel_request) },
{ }
};
static DBusMessage *auto_confirmation(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device;
dbus_uint32_t passkey;
bt_shell_printf("Request confirmation\n");
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_UINT32, &passkey, DBUS_TYPE_INVALID);
bt_shell_printf("Confirm passkey %06u (auto)", passkey);
return dbus_message_new_method_return(msg);
}
static DBusMessage *auto_authorization(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device;
bt_shell_printf("Request authorization\n");
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_INVALID);
bt_shell_printf("Accept pairing (auto)");
return dbus_message_new_method_return(msg);
}
static DBusMessage *auto_authorize_service(DBusConnection *conn,
DBusMessage *msg, void *user_data)
{
const char *device, *uuid;
dbus_message_get_args(msg, NULL, DBUS_TYPE_OBJECT_PATH, &device,
DBUS_TYPE_STRING, &uuid, DBUS_TYPE_INVALID);
bt_shell_printf("Authorize service %s (auto)", uuid);
return dbus_message_new_method_return(msg);
}
static const GDBusMethodTable auto_methods[] = {
{ GDBUS_METHOD("Release", NULL, NULL, release_agent) },
{ GDBUS_ASYNC_METHOD("RequestPinCode",
GDBUS_ARGS({ "device", "o" }),
GDBUS_ARGS({ "pincode", "s" }), request_pincode) },
{ GDBUS_METHOD("DisplayPinCode",
GDBUS_ARGS({ "device", "o" }, { "pincode", "s" }),
NULL, display_pincode) },
{ GDBUS_ASYNC_METHOD("RequestPasskey",
GDBUS_ARGS({ "device", "o" }),
GDBUS_ARGS({ "passkey", "u" }), request_passkey) },
{ GDBUS_METHOD("DisplayPasskey",
GDBUS_ARGS({ "device", "o" }, { "passkey", "u" },
{ "entered", "q" }),
NULL, display_passkey) },
{ GDBUS_ASYNC_METHOD("RequestConfirmation",
GDBUS_ARGS({ "device", "o" }, { "passkey", "u" }),
NULL, auto_confirmation) },
{ GDBUS_ASYNC_METHOD("RequestAuthorization",
GDBUS_ARGS({ "device", "o" }),
NULL, auto_authorization) },
{ GDBUS_ASYNC_METHOD("AuthorizeService",
GDBUS_ARGS({ "device", "o" }, { "uuid", "s" }),
NULL, auto_authorize_service) },
{ GDBUS_METHOD("Cancel", NULL, NULL, cancel_request) },
{ }
};
static void register_agent_setup(DBusMessageIter *iter, void *user_data)
{
const char *path = AGENT_PATH;
const char *capability = agent_capability;
dbus_message_iter_append_basic(iter, DBUS_TYPE_OBJECT_PATH, &path);
dbus_message_iter_append_basic(iter, DBUS_TYPE_STRING, &capability);
}
static void register_agent_reply(DBusMessage *message, void *user_data)
{
DBusConnection *conn = user_data;
DBusError error;
dbus_error_init(&error);
if (dbus_set_error_from_message(&error, message) == FALSE) {
agent_registered = TRUE;
bt_shell_printf("Agent registered\n");
} else {
bt_shell_printf("Failed to register agent: %s\n", error.name);
dbus_error_free(&error);
if (g_dbus_unregister_interface(conn, AGENT_PATH,
AGENT_INTERFACE) == FALSE)
bt_shell_printf("Failed to unregister agent object\n");
}
}
void agent_register(DBusConnection *conn, GDBusProxy *manager,
const char *capability)
{
const GDBusMethodTable *methods = agent_methods;
if (agent_registered == TRUE) {
bt_shell_printf("Agent is already registered\n");
return;
}
agent_capability = capability;
if (!strcasecmp(agent_capability, "auto")) {
bt_shell_printf("Warning: setting auto response is not secure, "
"it bypass user confirmation/authorization, it "
"shall only be used for test automation.\n");
agent_capability = "";
methods = auto_methods;
}
if (g_dbus_register_interface(conn, AGENT_PATH,
AGENT_INTERFACE, methods,
NULL, NULL, NULL, NULL) == FALSE) {
bt_shell_printf("Failed to register agent object\n");
return;
}
if (g_dbus_proxy_method_call(manager, "RegisterAgent",
register_agent_setup,
register_agent_reply,
conn, NULL) == FALSE) {
bt_shell_printf("Failed to call register agent method\n");
return;
}
agent_capability = NULL;
}
static void unregister_agent_setup(DBusMessageIter *iter, void *user_data)
{
const char *path = AGENT_PATH;
dbus_message_iter_append_basic(iter, DBUS_TYPE_OBJECT_PATH, &path);
}
static void unregister_agent_reply(DBusMessage *message, void *user_data)
{
DBusConnection *conn = user_data;
DBusError error;
dbus_error_init(&error);
if (dbus_set_error_from_message(&error, message) == FALSE) {
bt_shell_printf("Agent unregistered\n");
agent_release(conn);
} else {
bt_shell_printf("Failed to unregister agent: %s\n", error.name);
dbus_error_free(&error);
}
}
void agent_unregister(DBusConnection *conn, GDBusProxy *manager)
{
if (agent_registered == FALSE) {
bt_shell_printf("No agent is registered\n");
return;
}
if (!manager) {
bt_shell_printf("Agent unregistered\n");
agent_release(conn);
return;
}
if (g_dbus_proxy_method_call(manager, "UnregisterAgent",
unregister_agent_setup,
unregister_agent_reply,
conn, NULL) == FALSE) {
bt_shell_printf("Failed to call unregister agent method\n");
return;
}
}
static void request_default_setup(DBusMessageIter *iter, void *user_data)
{
const char *path = AGENT_PATH;
dbus_message_iter_append_basic(iter, DBUS_TYPE_OBJECT_PATH, &path);
}
static void request_default_reply(DBusMessage *message, void *user_data)
{
DBusError error;
dbus_error_init(&error);
if (dbus_set_error_from_message(&error, message) == TRUE) {
bt_shell_printf("Failed to request default agent: %s\n",
error.name);
dbus_error_free(&error);
return bt_shell_noninteractive_quit(EXIT_FAILURE);
}
bt_shell_printf("Default agent request successful\n");
return bt_shell_noninteractive_quit(EXIT_SUCCESS);
}
void agent_default(DBusConnection *conn, GDBusProxy *manager)
{
if (agent_registered == FALSE) {
bt_shell_printf("No agent is registered\n");
return bt_shell_noninteractive_quit(EXIT_FAILURE);
}
if (g_dbus_proxy_method_call(manager, "RequestDefaultAgent",
request_default_setup,
request_default_reply,
NULL, NULL) == FALSE) {
bt_shell_printf("Failed to call RequestDefaultAgent method\n");
return bt_shell_noninteractive_quit(EXIT_FAILURE);
}
}