blob: ea1d24ae701c7f679c613b3d47b8306ea5f1889b [file] [log] [blame]
/*
*
* Connection Manager
*
* Copyright (C) 2007-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 version 2 as
* published by the Free Software Foundation.
*
* 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 <string.h>
#include <signal.h>
#include <dbus/dbus.h>
static volatile sig_atomic_t __io_terminate = 0;
static void sig_term(int sig)
{
__io_terminate = 1;
}
static void add_dict_with_string_value(DBusMessageIter *iter,
const char *key, const char *str)
{
DBusMessageIter dict, entry, value;
dbus_message_iter_open_container(iter, DBUS_TYPE_ARRAY,
DBUS_DICT_ENTRY_BEGIN_CHAR_AS_STRING
DBUS_TYPE_STRING_AS_STRING DBUS_TYPE_VARIANT_AS_STRING
DBUS_DICT_ENTRY_END_CHAR_AS_STRING, &dict);
dbus_message_iter_open_container(&dict, DBUS_TYPE_DICT_ENTRY,
NULL, &entry);
dbus_message_iter_append_basic(&entry, DBUS_TYPE_STRING, &key);
dbus_message_iter_open_container(&entry, DBUS_TYPE_VARIANT,
DBUS_TYPE_STRING_AS_STRING, &value);
dbus_message_iter_append_basic(&value, DBUS_TYPE_STRING, &str);
dbus_message_iter_close_container(&entry, &value);
dbus_message_iter_close_container(&dict, &entry);
dbus_message_iter_close_container(iter, &dict);
}
static void add_empty_string_dict(DBusMessageIter *iter)
{
DBusMessageIter dict;
dbus_message_iter_open_container(iter, DBUS_TYPE_ARRAY,
DBUS_DICT_ENTRY_BEGIN_CHAR_AS_STRING
DBUS_TYPE_STRING_AS_STRING DBUS_TYPE_STRING_AS_STRING
DBUS_DICT_ENTRY_END_CHAR_AS_STRING, &dict);
dbus_message_iter_close_container(iter, &dict);
}
static void add_arguments(DBusConnection *conn, DBusMessageIter *iter)
{
const char *busname = dbus_bus_get_unique_name(conn);
const char *kind = "system-bus-name";
const char *action = "org.freedesktop.policykit.exec";
const char *cancel = "";
dbus_uint32_t flags = 0x00000001;
DBusMessageIter subject;
dbus_message_iter_open_container(iter, DBUS_TYPE_STRUCT,
NULL, &subject);
dbus_message_iter_append_basic(&subject, DBUS_TYPE_STRING, &kind);
add_dict_with_string_value(&subject, "name", busname);
dbus_message_iter_close_container(iter, &subject);
dbus_message_iter_append_basic(iter, DBUS_TYPE_STRING, &action);
add_empty_string_dict(iter);
dbus_message_iter_append_basic(iter, DBUS_TYPE_UINT32, &flags);
dbus_message_iter_append_basic(iter, DBUS_TYPE_STRING, &cancel);
}
static void print_arguments(DBusMessageIter *iter)
{
DBusMessageIter result;
dbus_bool_t authorized, challenge;
dbus_message_iter_recurse(iter, &result);
dbus_message_iter_get_basic(&result, &authorized);
dbus_message_iter_get_basic(&result, &challenge);
printf("Authorized %d (Challenge %d)\n", authorized, challenge);
}
#define AUTHORITY_DBUS "org.freedesktop.PolicyKit1"
#define AUTHORITY_INTF "org.freedesktop.PolicyKit1.Authority"
#define AUTHORITY_PATH "/org/freedesktop/PolicyKit1/Authority"
static int check_authorization(DBusConnection *conn)
{
DBusMessage *msg, *reply;
DBusMessageIter iter;
DBusError err;
msg = dbus_message_new_method_call(AUTHORITY_DBUS, AUTHORITY_PATH,
AUTHORITY_INTF, "CheckAuthorization");
if (!msg) {
fprintf(stderr, "Can't allocate new method call\n");
return -ENOMEM;
}
dbus_message_iter_init_append(msg, &iter);
add_arguments(conn, &iter);
dbus_error_init(&err);
reply = dbus_connection_send_with_reply_and_block(conn, msg, -1, &err);
dbus_message_unref(msg);
if (!reply) {
if (dbus_error_is_set(&err)) {
fprintf(stderr, "%s\n", err.message);
dbus_error_free(&err);
} else
fprintf(stderr, "Can't check authorization\n");
return -EIO;
}
if (dbus_message_has_signature(reply, "(bba{ss})")) {
dbus_message_iter_init(reply, &iter);
print_arguments(&iter);
}
dbus_message_unref(reply);
return 0;
}
int main(int argc, char *argv[])
{
DBusConnection *conn;
struct sigaction sa;
conn = dbus_bus_get(DBUS_BUS_SYSTEM, NULL);
if (!conn) {
fprintf(stderr, "Can't get on system bus");
return 1;
}
check_authorization(conn);
memset(&sa, 0, sizeof(sa));
sa.sa_flags = SA_NOCLDSTOP;
sa.sa_handler = sig_term;
sigaction(SIGTERM, &sa, NULL);
sigaction(SIGINT, &sa, NULL);
#if 0
while (!__io_terminate) {
if (dbus_connection_read_write_dispatch(conn, 500) == FALSE)
break;
}
#endif
dbus_connection_unref(conn);
return 0;
}