awesome/dbus.c

/*
 * dbus.c - awesome dbus support
 *
 * Copyright © 2008 Julien Danjou <julien@danjou.info>
 *
 * 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 Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */

#include <dbus/dbus.h>

#include "dbus.h"
#include "statusbar.h"
#include "widget.h"

extern AwesomeConf globalconf;

static DBusError err;
static DBusConnection *dbus_connection = NULL;

/** Check a dbus object path format and its number of element.
 * \param path The path.
 * \param nelem The number of element it should have.
 * \return true if the path is ok, false otherwise.
 */
static bool
a_dbus_path_check(char **path, int nelem)
{
    int i;

    for(i = 0; path[i]; i++);
    if(i != nelem)
        return false;

    return (!a_strcmp(path[0], "org")&& !a_strcmp(path[1], "awesome"));
}

/** Process widget.set method call.
 * \param req The dbus message.
 */
static void
a_dbus_process_widget_set(DBusMessage *req)
{
    char *arg, **path;
    int i, screen;
    DBusMessageIter iter;
    statusbar_t *statusbar;
    widget_t *widget;

    if(!dbus_message_get_path_decomposed(req, &path)
       || !a_dbus_path_check(path, 10)
       || a_strcmp(path[2], "screen")
       || a_strcmp(path[4], "statusbar")
       || a_strcmp(path[6], "widget")
       || a_strcmp(path[8], "property"))
    {
        warn("invalid object path 2\n");
        dbus_error_free(&err);
        return;
    }

    if(!dbus_message_iter_init(req, &iter))
    {
        warn("message has no argument: %s\n", err.message);
        dbus_error_free(&err);
        return;
    }
    else if(DBUS_TYPE_STRING != dbus_message_iter_get_arg_type(&iter))
    {
        warn("argument must be a string\n");
        dbus_error_free(&err);
        return;
    }
    else
        dbus_message_iter_get_basic(&iter, &arg);

    if((screen = atoi(path[3])) >= globalconf.screens_info->nscreen)
        return warn("bad screen number\n");

    if(!(statusbar = statusbar_getbyname(screen, path[5])))
        return warn("no such statusbar: %s\n", path[5]);

    if(!(widget = widget_getbyname(statusbar, path[7])))
        return warn("no such widget: %s in statusbar %s.\n", path[7], statusbar->name);

    widget->tell(widget, path[9], arg);

    for(i = 0; path[i]; i++)
        p_delete(&path[i]);
    p_delete(&path);
}

void
a_dbus_process_requests(int *fd)
{
    DBusMessage *msg;
    int nmsg = 0;

    if(!dbus_connection && !a_dbus_init(fd))
        return;

    while(true)
    {
        dbus_connection_read_write(dbus_connection, 0);

        if(!(msg = dbus_connection_pop_message(dbus_connection)))
            break;


        if(dbus_message_is_method_call(msg, "org.awesome.statusbar.widget", "set"))
            a_dbus_process_widget_set(msg);
        else if(dbus_message_is_signal(msg, DBUS_INTERFACE_LOCAL, "Disconnected"))
        {
            a_dbus_cleanup();
            dbus_message_unref(msg);
            return;
        }

        dbus_message_unref(msg);

        nmsg++;
    }

    if(nmsg)
        dbus_connection_flush(dbus_connection);
}

bool
a_dbus_init(int *fd)
{
    bool ret;

    dbus_error_init(&err);

    dbus_connection = dbus_bus_get(DBUS_BUS_SESSION, &err);
    if(dbus_error_is_set(&err))
    {
        warn("DBus system bus connection failed: %s\n", err.message);
        dbus_connection = NULL;
        dbus_error_free(&err);
        return false;
    }

    dbus_connection_set_exit_on_disconnect(dbus_connection, FALSE);

    ret = dbus_bus_request_name(dbus_connection, "org.awesome", 0, &err);

    if(dbus_error_is_set(&err))
    {
        warn("failed to request DBus name: %s\n", err.message);
        a_dbus_cleanup();
        return false;
    }

    if(ret != DBUS_REQUEST_NAME_REPLY_PRIMARY_OWNER)
    {
        warn("not primary DBus name owner\n");
        a_dbus_cleanup();
        return false;
    }

    if(!dbus_connection_get_unix_fd(dbus_connection, fd))
    {
        warn("cannot get DBus connection file descriptor\n");
        a_dbus_cleanup();
        return false;
    }

    return true;
}

void
a_dbus_cleanup(void)
{
    if(!dbus_connection)
        return;

    dbus_error_free(&err);

    /* This is a shared connection owned by libdbus
     * Do not close it, only unref
     */
    dbus_connection_unref(dbus_connection);
    dbus_connection = NULL;
}

// vim: filetype=c:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=80