/*
* add-passkey.c:
* registers as an agent for the bluez bluetooth linux stack, the code is
* shamelessly stolen from bluez source found at http://bluez.sf.net
*
* BlueZ - Bluetooth protocol stack for Linux
*
* Copyright (C) 2005-2006 Marcel Holtmann <marcel@holtmann.org>
*
*
* 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 <signal.h>
#include <getopt.h>
#include <string.h>
#include <dbus/dbus.h>
#define INTERFACE "org.bluez.Security"
// FIXME assumption
#define PASSKEYMAXLENGTH 255
static char *passkey = NULL;
static char *address = NULL;
static volatile sig_atomic_t __io_canceled = 0;
static volatile sig_atomic_t __io_terminated = 0;
static void sig_term(int sig)
{
__io_canceled = 1;
}
static DBusHandlerResult agent_filter(DBusConnection *conn,
DBusMessage *msg, void *data)
{
const char *name, *old, *new;
if (!dbus_message_is_signal(msg, DBUS_INTERFACE_DBUS, "NameOwnerChanged"))
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
if (!dbus_message_get_args(msg, NULL,
DBUS_TYPE_STRING, &name, DBUS_TYPE_STRING, &old,
DBUS_TYPE_STRING, &new, DBUS_TYPE_INVALID)) {
fprintf(stderr, "Invalid arguments for NameOwnerChanged signal");
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}
if (!strcmp(name, "org.bluez") && *new == '\0') {
__io_terminated = 1;
}
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}
static DBusHandlerResult request_message(DBusConnection *conn,
DBusMessage *msg, void *data)
{
DBusMessage *reply;
const char *path, *address;
if (!passkey)
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
if (!dbus_message_get_args(msg, NULL, DBUS_TYPE_STRING, &path,
DBUS_TYPE_STRING, &address, DBUS_TYPE_INVALID)) {
fprintf(stderr, "Invalid arguments for passkey Request method");
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}
reply = dbus_message_new_method_return(msg);
if (!reply) {
fprintf(stderr, "Can't create reply message\n");
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}
dbus_message_append_args(reply, DBUS_TYPE_STRING, &passkey,
DBUS_TYPE_INVALID);
dbus_connection_send(conn, reply, NULL);
dbus_connection_flush(conn);
dbus_message_unref(reply);
return DBUS_HANDLER_RESULT_HANDLED;
}
static DBusHandlerResult release_message(DBusConnection *conn,
DBusMessage *msg, void *data)
{
DBusMessage *reply;
if (!dbus_message_get_args(msg, NULL, DBUS_TYPE_INVALID)) {
fprintf(stderr, "Invalid arguments for passkey Release method");
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}
reply = dbus_message_new_method_return(msg);
if (!reply) {
fprintf(stderr, "Can't create reply message\n");
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}
dbus_message_append_args(reply, DBUS_TYPE_INVALID);
dbus_connection_send(conn, reply, NULL);
dbus_connection_flush(conn);
dbus_message_unref(reply);
if (!__io_canceled)
fprintf(stderr, "Passkey service has been released\n");
__io_terminated = 1;
return DBUS_HANDLER_RESULT_HANDLED;
}
static DBusHandlerResult agent_message(DBusConnection *conn,
DBusMessage *msg, void *data)
{
if (dbus_message_is_method_call(msg, "org.bluez.PasskeyAgent", "Request"))
return request_message(conn, msg, data);
if (dbus_message_is_method_call(msg, "org.bluez.PasskeyAgent", "Release"))
return release_message(conn, msg, data);
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}
static const DBusObjectPathVTable agent_table = {
.message_function = agent_message,
};
static int register_agent(DBusConnection *conn, const char *agent_path,
const char *remote_address, int use_default)
{
DBusMessage *msg, *reply;
DBusError err;
const char *path, *method, *address = remote_address;
if (!dbus_connection_register_object_path(conn, agent_path,
&agent_table, NULL)) {
fprintf(stderr, "Can't register path object path for agent\n");
return -1;
}
if (use_default) {
path = "/org/bluez";
method = "RegisterDefaultPasskeyAgent";
} else {
path = "/org/bluez/hci0";
method = "RegisterPasskeyAgent";
}
msg = dbus_message_new_method_call("org.bluez", path, INTERFACE, method);
if (!msg) {
fprintf(stderr, "Can't allocate new method call\n");
return -1;
}
if (use_default)
dbus_message_append_args(msg, DBUS_TYPE_STRING, &agent_path,
DBUS_TYPE_INVALID);
else
dbus_message_append_args(msg, DBUS_TYPE_STRING, &agent_path,
DBUS_TYPE_STRING, &address, DBUS_TYPE_INVALID);
dbus_error_init(&err);
reply = dbus_connection_send_with_reply_and_block(conn, msg, -1, &err);
dbus_message_unref(msg);
if (!reply) {
fprintf(stderr, "Can't register passkey agent\n");
if (dbus_error_is_set(&err)) {
fprintf(stderr, "%s\n", err.message);
dbus_error_free(&err);
}
return -1;
}
dbus_message_unref(reply);
dbus_connection_flush(conn);
return 0;
}
static int unregister_agent(DBusConnection *conn, const char *agent_path,
const char *remote_address, int use_default)
{
DBusMessage *msg, *reply;
DBusError err;
const char *path, *method, *address = remote_address;
if (use_default) {
path = "/org/bluez";
method = "UnregisterDefaultPasskeyAgent";
} else {
path = "/org/bluez/hci0";
method = "UnregisterPasskeyAgent";
}
msg = dbus_message_new_method_call("org.bluez", path, INTERFACE, method);
if (!msg) {
fprintf(stderr, "Can't allocate new method call\n");
dbus_connection_close(conn);
exit(1);
}
if (use_default)
dbus_message_append_args(msg, DBUS_TYPE_STRING, &agent_path,
DBUS_TYPE_INVALID);
else
dbus_message_append_args(msg, DBUS_TYPE_STRING, &agent_path,
DBUS_TYPE_STRING, &address, DBUS_TYPE_INVALID);
dbus_error_init(&err);
reply = dbus_connection_send_with_reply_and_block(conn, msg, -1, &err);
dbus_message_unref(msg);
if (!reply) {
fprintf(stderr, "Can't unregister passkey agent\n");
if (dbus_error_is_set(&err)) {
fprintf(stderr, "%s\n", err.message);
dbus_error_free(&err);
}
return -1;
}
dbus_message_unref(reply);
dbus_connection_flush(conn);
dbus_connection_unregister_object_path(conn, agent_path);
return 0;
}
static void usage(void)
{
// printf("Bluetooth passkey agent ver %s\n\n", VERSION);
printf("Usage:\n"
"\tadd-passkey [--passkey-fd n] [--default] [--path agent-path] [address]\n"
"\n"
"add-passkey will read from passkey-fd (default: stdin) adding a default passkey\n"
"if --default is given, or for a specific address if supplied on commandline.\n\n"
"--default or address are mandatory.\n"
"this program is based on bluez passkey-agent.c from http://bluez.sf.net\n"
"\n");
}
static struct option main_options[] = {
{ "passkey-fd", 1, 0, 'f'},
{ "default", 0, 0, 'd' },
{ "path", 1, 0, 'p' },
{ "help", 0, 0, 'h' },
{ 0, 0, 0, 0 }
};
int main(int argc, char *argv[])
{
struct sigaction sa;
DBusConnection *conn;
char match_string[128], default_path[128], *agent_path = NULL;
int opt, use_default = 0, passkey_fd = 0;
char *tmppasskey;
snprintf(default_path, sizeof(default_path),
"/org/bluez/passkey_agent_%d", getpid());
while ((opt = getopt_long(argc, argv, "+fdp:h", main_options, NULL)) != EOF) {
switch(opt) {
case 'f':
passkey_fd = atoi(optarg);
case 'd':
use_default = 1;
break;
case 'p':
if (optarg[0] != '/') {
fprintf(stderr, "Invalid path\n");
exit(1);
}
agent_path = strdup(optarg);
break;
case 'h':
usage();
exit(0);
default:
exit(1);
}
}
argc -= optind;
argv += optind;
optind = 0;
if (argc < 1 && !use_default) {
usage();
exit(1);
}
passkey = malloc(PASSKEYMAXLENGTH * sizeof(char));
if (!read(passkey_fd, passkey, PASSKEYMAXLENGTH)) {
fprintf(stderr, "Unable to read passkey from fd %d\n", passkey_fd);
exit(1);
}
if (tmppasskey = index(passkey, '\n')) {
*tmppasskey = '\0';
}
address = (argc > 0) ? strdup(argv[0]) : NULL;
if (!use_default && !address) {
usage();
exit(1);
}
if (!agent_path)
agent_path = strdup(default_path);
conn = dbus_bus_get(DBUS_BUS_SYSTEM, NULL);
if (!conn) {
fprintf(stderr, "Can't get on system bus");
exit(1);
}
if (register_agent(conn, agent_path, address, use_default) < 0) {
dbus_connection_close(conn);
exit(1);
}
if (!dbus_connection_add_filter(conn, agent_filter, NULL, NULL))
fprintf(stderr, "Can't add signal filter");
snprintf(match_string, sizeof(match_string),
"interface=%s,member=NameOwnerChanged,arg0=%s",
DBUS_INTERFACE_DBUS, "org.bluez");
dbus_bus_add_match(conn, match_string, NULL);
memset(&sa, 0, sizeof(sa));
sa.sa_flags = SA_NOCLDSTOP;
sa.sa_handler = sig_term;
sigaction(SIGTERM, &sa, NULL);
sigaction(SIGINT, &sa, NULL);
while (!__io_canceled && !__io_terminated) {
if (dbus_connection_read_write_dispatch(conn, 100) != TRUE)
break;
}
if (!__io_terminated)
unregister_agent(conn, agent_path, address, use_default);
if (passkey)
free(passkey);
dbus_connection_close(conn);
return 0;
}
