source: protocols/rpc/rpc.c @ 10d089d

#include <sys/socket.h>
#include <sys/un.h>
#include <sys/types.h>
#include <dirent.h>

#include "bitlbee.h"
#include "bee.h"
#include "nogaim.h"
#include "parson.h"

#define JSON_O_FOREACH(o, k, v) \
    const char *k; const JSON_Value *v; int __i; \
    for (__i = 0; json_object_get_tuple(o, __i, &k, &v); __i++)

static int next_rpc_id = 1;

struct rpc_plugin {
        /* Socket address of the RPC server. */
        struct sockaddr *addr;
        socklen_t addrlen;
        /* Full copy of the "settings" section of the init message. This info
         * can only be applied later on, when an account is created (but well
         * before logging in). */
        JSON_Value *settings;
        /* Supported away states returned by the away_states() function. Since
         * RPC servers can't do return values, just get this info at init time.
         * This means the list of possible away states is static from init time
         * which hopefully won't be a problem. If NULL, the away_states function
         * will not be set on this protocol. */
        GList *away_states;
        /* Account flags. See account.h. Plugin lib should provide constants. */
        int account_flags;
};

struct rpc_connection {
        int fd;
        char *buf;
        int buflen;
        GHashTable *groupchats;
};

struct rpc_groupchat {
        int id;
        struct groupchat *gc;
};

static JSON_Value *jsonrpc_error(int code, const char *msg) {
        JSON_Value *ret = json_value_init_object();
        json_object_set_null(json_object(ret), "result");
        if (TRUE) {
                /* Format from http://jsonrpc.org/historical/json-rpc-1-1-alt.html.
                 * Not sure whether to use it. */
                JSON_Value *error = json_value_init_object();
                json_object_set_integer(json_object(error), "code", code);
                json_object_set_string(json_object(error), "message", msg);
                json_object_set_value(json_object(ret), "error", error);
        } else {
                json_object_set_string(json_object(ret), "error", msg);
        }
        
        return ret;
}

// Might have liked to have this one in the library for optional values/etc.
static void json_array_append_string_or_null(JSON_Array *array, const char *string) {
        if (string)
                json_array_append_string(array, string);
        else
                json_array_append_null(array);
}

static void json_object_set_string_or_null(JSON_Object *object, const char *key, const char *string) {
        if (string)
                json_object_set_string(object, key, string);
        else
                json_object_set_null(object, key);
}

static JSON_Value *rpc_out_new(const char *method, JSON_Array **params_) {
        JSON_Value *rpc = json_value_init_object();
        json_object_set_string(json_object(rpc), "method", method);
        json_object_set_integer(json_object(rpc), "id", next_rpc_id++);

        JSON_Value *params = json_value_init_array();
        json_object_set_value(json_object(rpc), "params", params);

        if (params_)
                *params_ = json_array(params);

        return rpc;
}

#define RPC_OUT_INIT(method) \
        JSON_Array *params; \
        JSON_Value *rpc = rpc_out_new(method, &params);

/** Sends an RPC object. Takes ownership (i.e. frees it when done). */
static gboolean rpc_send(struct im_connection *ic, JSON_Value *rpc) {
        struct rpc_connection *rd = ic->proto_data;
        char *buf = json_serialize_to_string(rpc);
        int len = strlen(buf);
        int st;

        buf = g_realloc(buf, len + 3);
        strcpy(buf + len, "\r\n");
        len += 2;

        st = write(rd->fd, buf, len);
        g_free(buf);
        json_value_free(rpc);

        if (st != len) {
                if (!(ic->flags & OPT_LOGGING_OUT))
                        imcb_log(ic, "Write error");
                imc_logout(ic, TRUE);
                return FALSE;
        }

        return TRUE;
}

static JSON_Value *rpc_ser_settings(set_t **set);

static JSON_Value *rpc_ser_account(account_t *acc) {
        JSON_Value *v = json_value_init_object();
        JSON_Object *o = json_object(v);
        json_object_set_string(o, "user", acc->user);
        json_object_set_string(o, "pass", acc->pass);
        if (acc->server)
                json_object_set_string(o, "server", acc->server);
        json_object_set_value(o, "settings", rpc_ser_settings(&acc->set));
        return v;
}

static JSON_Value *rpc_ser_bee_user(bee_user_t *bu) {
        JSON_Value *v = json_value_init_object();
        JSON_Object *o = json_object(v);
        json_object_set_string_or_null(o, "handle", bu->handle);
        json_object_set_string_or_null(o, "fullname", bu->fullname);
        json_object_set_string_or_null(o, "nick", bu->nick);
        json_object_set_string_or_null(o, "group", bu->group ? bu->group->name : NULL);
        json_object_set_integer(o, "flags", bu->flags);
        json_object_set_string_or_null(o, "status", bu->status);
        json_object_set_string_or_null(o, "status_msg", bu->status_msg);
        json_object_set_integer(o, "login_time", bu->login_time);
        json_object_set_integer(o, "idle_time", bu->idle_time);
        return v;
}

static char *rpc_set_evaluator(set_t *set, char *value);

static void rpc_init(account_t *acc) {
        struct rpc_plugin *pd = acc->prpl->data;

        JSON_O_FOREACH(json_object(pd->settings), name, value) {
                JSON_Object *o = json_object(value);
                char *defs = NULL;
                JSON_Value *defv = json_object_get_value(o, "default");
                if (json_type(defv) == JSONString)
                        defs = g_strdup(json_string(defv));
                else if(json_type(defv) == JSONInteger)
                        defs = g_strdup_printf("%lld", (long long) json_integer(defv));
                else if(json_type(defv) == JSONBoolean)
                        defs = g_strdup(json_boolean(defv) ? "true" : "false");
                set_t *set = set_add(&acc->set, name, defs, NULL, acc);
                g_free(defs);
                set->flags |= json_object_get_integer(o, "flags");
                set->eval = rpc_set_evaluator;
                set->eval_data = o;
                /* eval_list turns out to be a memory leak so don't implement it
                 * for now.
                 * Allowing a plugin to define its own evaluator is not really
                 * possible without having BitlBee block on it responding which
                 * I don't want to do.
                 * Should a module want to override a user's setting, it can
                 * use set_setstr(). */
        }

        acc->flags |= pd->account_flags;
}

set_eval rpc_type_set_eval(const set_t *set) {
        JSON_Object *o = set->eval_data;
        const char *type = json_object_get_string(o, "type");

        set_eval type_eval = NULL;
        if (type == NULL) {
                /* Try to do something sane for settings that aren't ours. */
                if (set->eval == set_eval_int || set->eval == set_eval_bool) {
                        type_eval = set->eval;
                }
        } else if (g_str_has_prefix(type, "int")) {
                type_eval = set_eval_int;
        } else if (g_str_has_prefix(type, "bool")) {
                type_eval = set_eval_bool;
        }

        return type_eval;
}

static JSON_Value *set_make_json_value(set_eval type, const char *value) {
        JSON_Value *ret;

        if (value == NULL) {
                ret = json_value_init_null();
        } else if (type == set_eval_int) {
                long long num = 0;
                /* Evaluator already did validation so ignore retval. */
                sscanf(value, "%lld", &num);
                ret = json_value_init_integer(num);
        } else if (type == set_eval_bool) {
                ret = json_value_init_boolean(bool2int(value));
        } else {
                ret = json_value_init_string(value);
        }
        return ret;
}

static char *rpc_set_evaluator(set_t *set, char *value) {
        set_eval type_eval = rpc_type_set_eval(set);
        if (type_eval) {
                char *new = type_eval(set, value);
                if (new == SET_INVALID) {
                        return SET_INVALID;
                }
        }

        account_t *acc = set->data;
        if (acc->ic) {
                /* But do send RPCs to the plugin for each changed setting so
                 * it always has up-to-date values. */
                RPC_OUT_INIT("set_set");
                json_array_append_string(params, set->key);
                json_array_append_value(params, set_make_json_value(rpc_type_set_eval(set), value));
                rpc_send(acc->ic, rpc);
        }

        return value;
}

static JSON_Value *rpc_ser_settings(set_t **set) {
        const set_t *s;
        JSON_Value *ret = json_value_init_object();
       
        for (s = *set; s; s = s->next) {
                JSON_Value *v = set_make_json_value(rpc_type_set_eval(s), set_value(s));
                json_object_set_value(json_object(ret), s->key, v);
        }

        return ret;
}

static gboolean rpc_login_cb(gpointer data, gint fd, b_input_condition cond);
static gboolean rpc_in_event(gpointer data, gint fd, b_input_condition cond);
static JSON_Value *rpc_init_isup();

static void rpc_login(account_t *acc) {
        struct im_connection *ic = imcb_new(acc);
        struct rpc_connection *rd = ic->proto_data = g_new0(struct rpc_connection, 1);
        struct rpc_plugin *pd = acc->prpl->data;
        imcb_log(ic, "Logging in via RPC server");
        rd->fd = socket(pd->addr->sa_family, SOCK_STREAM, 0);
        sock_make_nonblocking(rd->fd);
        if (connect(rd->fd, pd->addr, pd->addrlen) == -1) {
                closesocket(rd->fd);
                imcb_error(ic, "RPC server unreachable");
                imc_logout(ic, TRUE);
                return;
        }
        ic->inpa = b_input_add(rd->fd, B_EV_IO_WRITE, rpc_login_cb, ic);
        rd->groupchats = g_hash_table_new(g_int_hash, g_int_equal);
}

static gboolean rpc_login_cb(gpointer data, gint fd, b_input_condition cond) {
        struct im_connection *ic = data;
        struct rpc_connection *rd = ic->proto_data;

        /* Need to repeat this since each IM connection means an actual new
         * RPC session. */
        JSON_Value *init = rpc_init_isup();
        if (!rpc_send(ic, init))
                return FALSE;

        RPC_OUT_INIT("login");
        json_array_append_value(params, rpc_ser_account(ic->acc));
        if (!rpc_send(ic, rpc))
                return FALSE;

        ic->inpa = b_input_add(rd->fd, B_EV_IO_READ, rpc_in_event, ic);

        return FALSE;
}

static void rpc_keepalive(struct im_connection *ic) {
        RPC_OUT_INIT("keepalive");
        rpc_send(ic, rpc);
}

static void rpc_groupchat_free(struct groupchat *gc);

static void rpc_logout(struct im_connection *ic) {
        RPC_OUT_INIT("logout");
        if (!rpc_send(ic, rpc))
                return;

        struct rpc_connection *rd = ic->proto_data;

        GHashTableIter iter;
        struct rpc_groupchat *rc;
        g_hash_table_iter_init(&iter, rd->groupchats);
        while (g_hash_table_iter_next(&iter, NULL, (gpointer*)&rc)) {
                rpc_groupchat_free(rc->gc);
        }

        b_event_remove(ic->inpa);
        closesocket(rd->fd);
        g_free(rd->buf);
        g_hash_table_destroy(rd->groupchats);
        g_free(rd);
}

static int rpc_buddy_msg(struct im_connection *ic, char *to, char *message, int flags) {
        RPC_OUT_INIT("buddy_msg");
        json_array_append_string(params, to);
        json_array_append_string(params, message);
        json_array_append_integer(params, flags);
        return rpc_send(ic, rpc);
}

static void rpc_set_away(struct im_connection *ic, char *state, char *message) {
        RPC_OUT_INIT("set_away");
        json_array_append_string_or_null(params, state);
        json_array_append_string_or_null(params, message);
        rpc_send(ic, rpc);
}

static int rpc_send_typing(struct im_connection *ic, char *who, int flags) {
        RPC_OUT_INIT("send_typing");
        json_array_append_string(params, who);
        json_array_append_integer(params, flags);
        return rpc_send(ic, rpc);
}

static void rpc_add_buddy(struct im_connection *ic, char *name, char *group) {
        RPC_OUT_INIT("add_buddy");
        json_array_append_string(params, name);
        json_array_append_string_or_null(params, group);
        rpc_send(ic, rpc);
}

static void rpc_remove_buddy(struct im_connection *ic, char *name, char *group) {
        RPC_OUT_INIT("remove_buddy");
        json_array_append_string(params, name);
        json_array_append_string_or_null(params, group);
        rpc_send(ic, rpc);
}

static void rpc_add_permit(struct im_connection *ic, char *who) {
        RPC_OUT_INIT("add_permit");
        json_array_append_string(params, who);
        rpc_send(ic, rpc);
}

static void rpc_add_deny(struct im_connection *ic, char *who) {
        RPC_OUT_INIT("add_deny");
        json_array_append_string(params, who);
        rpc_send(ic, rpc);
}

static void rpc_rem_permit(struct im_connection *ic, char *who) {
        RPC_OUT_INIT("rem_permit");
        json_array_append_string(params, who);
        rpc_send(ic, rpc);
}

static void rpc_rem_deny(struct im_connection *ic, char *who) {
        RPC_OUT_INIT("rem_deny");
        json_array_append_string(params, who);
        rpc_send(ic, rpc);
}

static void rpc_get_info(struct im_connection *ic, char *who) {
        RPC_OUT_INIT("get_info");
        json_array_append_string(params, who);
        rpc_send(ic, rpc);
}

static void rpc_chat_invite(struct groupchat *gc, char *who, char *message) {
        RPC_OUT_INIT("chat_invite");
        struct rpc_groupchat *rc = gc->data;
        json_array_append_integer(params, rc->id);
        json_array_append_string(params, who);
        json_array_append_string_or_null(params, message);
        rpc_send(gc->ic, rpc);
}

static void rpc_chat_kick(struct groupchat *gc, char *who, const char *message) {
        RPC_OUT_INIT("chat_kick");
        struct rpc_groupchat *rc = gc->data;
        json_array_append_integer(params, rc->id);
        json_array_append_string(params, who);
        json_array_append_string_or_null(params, message);
        rpc_send(gc->ic, rpc);
}

static void rpc_chat_leave(struct groupchat *gc) {
        RPC_OUT_INIT("chat_leave");
        struct rpc_groupchat *rc = gc->data;
        json_array_append_integer(params, rc->id);
        rpc_send(gc->ic, rpc);

        struct rpc_connection *rd = gc->ic->proto_data;
        g_hash_table_remove(rd->groupchats, &rc->id);
        rpc_groupchat_free(gc);
}

static void rpc_chat_msg(struct groupchat *gc, char *msg, int flags) {
        RPC_OUT_INIT("chat_msg");       
        struct rpc_groupchat *rc = gc->data;
        json_array_append_integer(params, rc->id);
        json_array_append_string(params, msg);
        json_array_append_integer(params, flags);
        rpc_send(gc->ic, rpc);
}

static struct rpc_groupchat *rpc_groupchat_new(struct im_connection *ic, const char *handle) {
        struct rpc_connection *rd = ic->proto_data;
        struct groupchat *gc = imcb_chat_new(ic, handle);
        struct rpc_groupchat *rc = gc->data = g_new0(struct rpc_groupchat, 1);
        rc->id = next_rpc_id++;
        rc->gc = gc;
        g_hash_table_insert(rd->groupchats, &rc->id, rc);
        return rc;  // TODO: RETVAL HERE AND BELOW
}

static struct rpc_groupchat *rpc_groupchat_by_id(struct im_connection *ic, int id) {
        struct rpc_connection *rd = ic->proto_data;
        struct rpc_groupchat *rc = g_hash_table_lookup(rd->groupchats, &id);

        return rc;
}

/* When calling this, also make sure the element gets removes from
   rpc_connection->groupchat. This function doesn't do that because it gets
   called from within an iterator. */
static void rpc_groupchat_free(struct groupchat *gc) {
        g_free(gc->data);
        imcb_chat_free(gc);
}

/* Boilerplate for all incoming RPCs (where groupchat is identified using
 * numeric ID). */
#define SET_GROUPCHAT(rc) \
        do { \
                rc = rpc_groupchat_by_id(ic, json_array_get_integer(params, 0)); \
                if (rc == NULL) \
                        return jsonrpc_error(ENOENT, "No groupchat with that id."); \
        } while (0)

static struct groupchat *rpc_chat_with(struct im_connection *ic, char *who) {
        RPC_OUT_INIT("chat_with");
        struct rpc_groupchat *rc = rpc_groupchat_new(ic, who);
        json_array_append_integer(params, rc->id);
        json_array_append_string(params, who);
        rpc_send(ic, rpc);

        return rc->gc; 
}

static struct groupchat *rpc_chat_join(struct im_connection *ic, const char *room, const char *nick,
                                       const char *password, set_t **sets) {
        RPC_OUT_INIT("chat_join");
        struct rpc_groupchat *rc = rpc_groupchat_new(ic, room);
        json_array_append_integer(params, rc->id);
        json_array_append_string(params, room);
        json_array_append_string_or_null(params, nick);
        json_array_append_string_or_null(params, password);
        json_array_append_value(params, rpc_ser_settings(sets));
        rpc_send(ic, rpc);

        return rc->gc;
}

static void rpc_chat_topic(struct groupchat *gc, char *topic) {
        RPC_OUT_INIT("chat_topic");
        struct rpc_groupchat *rc = gc->data;
        json_array_append_integer(params, rc->id);
        json_array_append_string(params, topic);
        rpc_send(gc->ic, rpc);
}

static GList *rpc_away_states(struct im_connection *ic) {
        struct rpc_plugin *pd = ic->acc->prpl->data;
        return pd->away_states;
}

static JSON_Value *rpc_cmd_in(struct im_connection *ic, const char *cmd, JSON_Array *params);

static gboolean rpc_in(struct im_connection *ic, JSON_Object *rpc) {
        const char *cmd = json_object_get_string(rpc, "method");
        JSON_Value *id = json_object_get_value(rpc, "id");
        JSON_Value *error = json_object_get_value(rpc, "error");
        JSON_Array *params = json_object_get_array(rpc, "params");

        /* Removed checks for result/error/etc. as it's all too free-form and
         * at least for now this code is not going to care about retvals as
         * they come in late anyway. */
        if (!id) {
                imcb_log(ic, "Received invalid JSON-RPC object.");
                imc_logout(ic, TRUE);
                return FALSE;
        }

        if (cmd) {
                JSON_Value *resp = rpc_cmd_in(ic, cmd, params);
                if (!resp) {
                        resp = json_value_init_object();
                        json_object_set_boolean(json_object(resp), "result", TRUE);
                }
                json_object_set_value(json_object(resp), "id", json_value_deep_copy(id));
                return rpc_send(ic, resp);
        } else if (error && json_type(error) != JSONNull) {
                char *error_str = json_serialize_to_string(error);
                /* Maybe sanitise/truncate? Though really that should be done at
                 * a different layer. */
                imcb_error(ic, "RPC Error: %s", error_str);
                g_free(error_str);
        }

        return TRUE;
}

static gboolean rpc_in_event(gpointer data, gint fd, b_input_condition cond) {
        struct im_connection *ic = data;
        struct rpc_connection *rd = ic->proto_data;
        char buf[2048];
        int st;

        while ((st = read(rd->fd, buf, sizeof(buf))) > 0) {
                rd->buf = g_realloc(rd->buf, rd->buflen + st + 1);
                memcpy(rd->buf + rd->buflen, buf, st);
                rd->buflen += st;
        }

        if (st == 0 || (st == -1 && !(sockerr_again() || errno == EAGAIN))) {
                imcb_log(ic, "Lost RPC connection");
                imc_logout(ic, TRUE);
                return FALSE;
        }
        rd->buf[rd->buflen] = '\0';

        JSON_Value *parsed;
        const char *end;
        while ((parsed = json_parse_first(rd->buf, &end))) {
                st = rpc_in(ic, json_object(parsed));
                json_value_free(parsed);

                if (!st)
                        return FALSE;

                if (end == rd->buf + rd->buflen) {
                        g_free(rd->buf);
                        rd->buf = NULL;
                } else {
                        int newlen = rd->buf + rd->buflen - end;
                        char new[newlen];
                        memcpy(new, end, newlen);
                        rd->buf = g_realloc(rd->buf, newlen + 1);
                        memcpy(rd->buf, new, newlen);
                        rd->buflen = newlen;
                        rd->buf[rd->buflen] = '\0';
                }
        }

        return TRUE;
}

static JSON_Value *rpc_imcb_log(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct im_connection*, const char*, ...) = func_;
        func(ic, "%s", json_array_get_string(params, 0));
        return NULL;
}

static JSON_Value *rpc_imcb_connected(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct im_connection*) = func_;
        func(ic);
        return NULL;
}

static JSON_Value *rpc_imc_logout(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct im_connection*, gboolean) = func_;
        func(ic, json_array_get_boolean(params, 0));
        return NULL;
}

static JSON_Value *rpc_imcb_add_buddy(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct im_connection*, const char*, const char*) = func_;
        func(ic, json_array_get_string(params, 0), json_array_get_string(params, 1));
        return NULL;
}

static JSON_Value *rpc_imcb_buddy_status(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct im_connection*, const char*, int, const char*, const char*) = func_;
        func(ic, json_array_get_string(params, 0), json_array_get_integer(params, 1),
                 json_array_get_string(params, 2), json_array_get_string(params, 3));
        return NULL;
}

static JSON_Value *rpc_imcb_buddy_times(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct im_connection*, const char*, int, int) = func_;
        func(ic, json_array_get_string(params, 0), json_array_get_integer(params, 1),
                 json_array_get_integer(params, 2));
        return NULL;
}

static JSON_Value *rpc_imcb_buddy_msg(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct im_connection*, const char*, const char*, int, int) = func_;
        func(ic, json_array_get_string(params, 0), json_array_get_string(params, 1),
                 json_array_get_integer(params, 2), json_array_get_integer(params, 3));
        return NULL;
}

static JSON_Value *rpc_imcb_buddy_typing(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct im_connection*, const char*, int) = func_;
        func(ic, (char*) json_array_get_string(params, 0), json_array_get_integer(params, 1));
        return NULL;
}

static JSON_Value *rpc_imcb_get_local_contacts(struct im_connection *ic, void *func_, JSON_Array *params) {
        JSON_Value *resp = json_value_init_object();
        JSON_Value *arr = json_value_init_array();
        GSList *contacts = imcb_get_local_contacts(ic);
        GSList *c;
        for (c = contacts; c; c = c->next) {
                json_array_append_string(json_array(arr), (const char*) c->data);
        }
        g_slist_free(contacts);
        json_object_set_value(json_object(resp), "result", arr);
        return resp;
}

static JSON_Value *rpc_imcb_chat_new(struct im_connection *ic, void *func_, JSON_Array *params) {
        struct rpc_groupchat *rc = rpc_groupchat_new(ic, json_array_get_string(params, 0));
        JSON_Value *resp = json_value_init_object();
        json_object_set_integer(json_object(resp), "result", rc->id);
        return resp;
}

static JSON_Value *rpc_imcb_chat_name_hint(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct groupchat*, const char*) = func_;
        struct rpc_groupchat *rc;
        SET_GROUPCHAT(rc);
        func(rc->gc, json_array_get_string(params, 1));
        return NULL;
}

static JSON_Value *rpc_imcb_chat_msg(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct groupchat*, const char*, const char*, guint32, time_t) = func_;
        struct rpc_groupchat *rc;
        SET_GROUPCHAT(rc);
        func(rc->gc, json_array_get_string(params, 1), json_array_get_string(params, 2),
             json_array_get_integer(params, 3), json_array_get_integer(params, 4));
        return NULL;
}

static JSON_Value *rpc_imcb_chat_log(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct groupchat*, const char*, ...) = func_;
        struct rpc_groupchat *rc;
        SET_GROUPCHAT(rc);
        func(rc->gc, "%s", json_array_get_string(params, 1));
        return NULL;
}

static JSON_Value *rpc_imcb_chat_topic(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct groupchat*, const char*, const char*, time_t) = func_;
        struct rpc_groupchat *rc;
        SET_GROUPCHAT(rc);
        func(rc->gc, json_array_get_string(params, 1), json_array_get_string(params, 2),
             json_array_get_integer(params, 3));
        return NULL;
}

static JSON_Value *rpc_imcb_chat_remove_buddy(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct groupchat*, const char*, const char*) = func_;
        struct rpc_groupchat *rc;
        SET_GROUPCHAT(rc);
        func(rc->gc, json_array_get_string(params, 1), json_array_get_string(params, 2));
        return NULL;
}

static JSON_Value *rpc_imcb_chat_invite(struct im_connection *ic, void *func_, JSON_Array *params) {
        void (*func)(struct groupchat*, const char*, const char*, const char*) = func_;
        struct rpc_groupchat *rc;
        SET_GROUPCHAT(rc);
        func(rc->gc, json_array_get_string(params, 1), json_array_get_string(params, 2),
             json_array_get_string(params, 3));
        return NULL;
}

static JSON_Value *rpc_set_getstr(struct im_connection *ic, void *func_, JSON_Array *params) {
        char *rets = set_getstr(&ic->acc->set, json_array_get_string(params, 0));
        JSON_Value *ret = json_value_init_object();
        if (rets)
                json_object_set_string(json_object(ret), "result", rets);
        else
                json_object_set_null(json_object(ret), "result");
        return ret;
}

static JSON_Value *rpc_set_setstr(struct im_connection *ic, void *func_, JSON_Array *params) {
        /* Sadly use of const is very poor in BitlBee. :-( */
        char *newval = g_strdup(json_array_get_string(params, 1));
        set_setstr(&ic->acc->set, json_array_get_string(params, 0), newval);
        g_free(newval);
        return rpc_set_getstr(ic, func_, params);
}

static JSON_Value *rpc_set_reset(struct im_connection *ic, void *func_, JSON_Array *params) {
        set_reset(&ic->acc->set, json_array_get_string(params, 0));
        return rpc_set_getstr(ic, func_, params);
}

static JSON_Value *rpc_bee_user_by_handle(struct im_connection *ic, void *func_, JSON_Array *params) {
        bee_user_t *bu = bee_user_by_handle(ic->bee, ic, json_array_get_string(params, 0));
        JSON_Value *ret = json_value_init_object();
        if (bu)
                json_object_set_value(json_object(ret), "result", rpc_ser_bee_user(bu));
        else
                json_object_set_value(json_object(ret), "error", jsonrpc_error(ENOENT, "Contact not found"));
        return ret;
}

struct rpc_in_method {
        char *name;
        void *func;
        JSON_Value* (* wfunc) (struct im_connection *ic, void *cmd, JSON_Array *params);
        char args[8];
};

static const struct rpc_in_method methods[] = {
        /* All these RPCs are equivalent of BitlBee C functions but with the
         * struct im_connection* removed as this is in the object context. */
        { "imcb_log", imcb_log, rpc_imcb_log, "s" },
        { "imcb_error", imcb_error, rpc_imcb_log, "s" },
        { "imcb_connected", imcb_connected, rpc_imcb_connected, "" },
        { "imc_logout", imc_logout, rpc_imc_logout, "b" },
        { "imcb_add_buddy", imcb_add_buddy, rpc_imcb_add_buddy, "ss" },
        { "imcb_remove_buddy", imcb_remove_buddy, rpc_imcb_add_buddy, "ss" },
        { "imcb_rename_buddy", imcb_rename_buddy, rpc_imcb_add_buddy, "ss" },
        { "imcb_buddy_nick_hint", imcb_buddy_nick_hint, rpc_imcb_add_buddy, "ss" },
        { "imcb_buddy_status", imcb_buddy_status, rpc_imcb_buddy_status, "siss" },
        { "imcb_buddy_status_msg", imcb_buddy_status_msg, rpc_imcb_add_buddy, "ss" },
        { "imcb_buddy_times", imcb_buddy_times, rpc_imcb_buddy_times, "sii" },
        { "imcb_buddy_msg", imcb_buddy_msg, rpc_imcb_buddy_msg, "ssii" },
        { "imcb_buddy_typing", imcb_buddy_typing, rpc_imcb_buddy_typing, "si" },
        { "imcb_get_local_contacts", NULL, rpc_imcb_get_local_contacts, "" },
        { "imcb_chat_new", NULL, rpc_imcb_chat_new, "s" },
        
        /* RPCs below are equivalent, but with the struct groupchat* replaced
         * with the numeric id of the chat. */
        { "imcb_chat_name_hint", imcb_chat_name_hint, rpc_imcb_chat_name_hint, "is" },
        { "imcb_chat_msg", imcb_chat_msg, rpc_imcb_chat_msg, "issii" },
        { "imcb_chat_log", imcb_chat_log, rpc_imcb_chat_log, "is" },
        { "imcb_chat_topic", imcb_chat_topic, rpc_imcb_chat_topic, "issi" },
        { "imcb_chat_add_buddy", imcb_chat_add_buddy, rpc_imcb_chat_name_hint, "is" },
        { "imcb_chat_remove_buddy", imcb_chat_remove_buddy, rpc_imcb_chat_remove_buddy, "iss" },
        { "imcb_chat_invite", imcb_chat_invite, rpc_imcb_chat_invite, "isss" },

        /* These are not imcb* functions but should still be exported. */
        /* Setting functions. Starting with just providing access to account
         * settings. See later whether access to chat/chan settings is necessary.
         * All of these will return the (new) value of given setting. */
        { "set_getstr", NULL, rpc_set_getstr, "s" },
        { "set_setstr", NULL, rpc_set_setstr, "ss" },
        { "set_reset", NULL, rpc_set_reset, "s" },
        
        { "bee_user_by_handle", NULL, rpc_bee_user_by_handle, "s" },

        { NULL },
};

static JSON_Value *rpc_cmd_in(struct im_connection *ic, const char *cmd, JSON_Array *params) {
        int i;

        for (i = 0; methods[i].name; i++) {
                if (strcmp(cmd, methods[i].name) == 0) {
                        if (json_array_get_count(params) != strlen(methods[i].args)) {
                                imcb_error(ic, "Invalid argument count to method %s: %d, wanted %zd", cmd, (int) json_array_get_count(params), strlen(methods[i].args));
                                return jsonrpc_error(E2BIG, "Invalid integer of arguments");
                        }
                        int j;
                        for (j = 0; methods[i].args[j]; j++) {
                                JSON_Value_Type type = json_value_get_type(json_array_get_value(params, j));
                                gboolean ok = FALSE;
                                switch (methods[i].args[j]) {
                                case 's':
                                        ok = type == JSONString || type == JSONNull;
                                        break;
                                case 'i':
                                        ok = type == JSONInteger;
                                        break;
                                case 'o':
                                        ok = type == JSONObject;
                                        break;
                                case 'a':
                                        ok = type == JSONArray;
                                        break;
                                case 'b':
                                        ok = type == JSONBoolean;
                                        break;
                                }
                                if (!ok) {
                                        // This error sucks, but just get your types right!
                                        imcb_error(ic, "Invalid argument type, %s parameter %d: %d not %c", cmd, j, type, methods[i].args[j]);
                                        return jsonrpc_error(EINVAL, "Invalid argument type");
                                }
                        }
                        return methods[i].wfunc(ic, methods[i].func, params);
                }
        }
        return jsonrpc_error(ENOSYS, "Function not implemented");
}

#define RPC_ADD_FUNC(func) \
        ret->func = rpc_ ## func
#define RPC_ADD_OPT_FUNC(func) \
        if (g_hash_table_lookup(methods, #func)) \
                RPC_ADD_FUNC(func)

static JSON_Value *rpc_init_isup() {
        int i;

        RPC_OUT_INIT("init");
        JSON_Value *d = json_value_init_object();
        json_object_set_string(json_object(d), "version_str", BITLBEE_VERSION);
        json_object_set_integer(json_object(d), "version", BITLBEE_VERSION_CODE);
        
        JSON_Value *ml = json_value_init_array();
        for (i = 0; methods[i].name; i++) {
                json_array_append_string(json_array(ml), methods[i].name);
        }
        json_object_set_value(json_object(d), "method_list", ml);
        json_array_append_value(params, d);

        return rpc;
}

gboolean rpc_initmodule_sock(struct sockaddr *address, socklen_t addrlen) {
        int st, fd, i;

        fd = socket(address->sa_family, SOCK_STREAM, 0);
        if (fd == -1 || connect(fd, address, addrlen) == -1) {
                log_message(LOGLVL_WARNING, "Failed to connect to RPC server: %s", strerror(errno));
                if (fd != -1)
                        closesocket(fd);
                return FALSE;
        }

        JSON_Value *rpc = rpc_init_isup();
        char *s = json_serialize_to_string(rpc);
        json_value_free(rpc);

        int len = strlen(s);
        s = g_realloc(s, len + 3);
        strcpy(s + len, "\r\n");
        len += 2;

        if ((st = write(fd, s, len)) != len) {
                log_message(LOGLVL_WARNING, "Error while writing to RPC server: %s", strerror(errno));
                return FALSE;
        }
        g_free(s);

        char *resp = NULL;
        int buflen = 4096, resplen = 0;
        JSON_Value *parsed;
        do {
                fd_set rfds;
                struct timeval to;

                FD_ZERO(&rfds);
                FD_SET(fd, &rfds);
                to.tv_sec = 1;
                to.tv_usec = 0;
                st = select(fd + 1, &rfds, NULL, NULL, &to);

                if (st == 0) {
                        log_message(LOGLVL_WARNING, "Error while reading from RPC server: %s", strerror(errno));
                        closesocket(fd);
                        return FALSE;
                }
                
                if (resplen >= buflen)
                        buflen *= 2;
                resp = g_realloc(resp, buflen + 1);
                st = read(fd, resp + resplen, buflen - resplen);
                if (st == -1) {
                        if (sockerr_again())
                                continue;
                        log_message(LOGLVL_WARNING, "Error while reading from RPC server: %s", strerror(errno));
                        closesocket(fd);
                        return FALSE;
                }
                resplen += st;
                resp[resplen] = '\0';
        }
        while (!(parsed = json_parse_string(resp)));
        closesocket(fd);

        JSON_Object *isup = json_object_get_object(json_object(parsed), "result");
        if (isup == NULL) {
                log_message(LOGLVL_WARNING, "Error while parsing RPC server response");
                return FALSE;
        }

        struct prpl *ret = g_new0(struct prpl, 1);
        struct rpc_plugin *proto_data = g_new0(struct rpc_plugin, 1);
        proto_data->addr = g_memdup(address, addrlen);
        proto_data->addrlen = addrlen;
        ret->name = g_strdup(json_object_get_string(isup, "name"));
        ret->data = proto_data;

        proto_data->account_flags = json_object_get_integer(isup, "account_flags");

        /* Keep a full copy of the settings list, we can only use it when we
         * have an account to work on. */
        JSON_Value *settings = json_object_get_value(isup, "settings");
        if (json_type(settings) == JSONObject)
                proto_data->settings = json_value_deep_copy(settings);

        JSON_Array *aways_a = json_object_get_array(isup, "away_state_list");
        for (i = 0; i < json_array_get_count(aways_a); ++i) {
                JSON_Value *state = json_array_get_value(aways_a, i);
                if (json_type(state) == JSONString)
                        proto_data->away_states =
                                g_list_append(proto_data->away_states,
                                              g_strdup(json_string(state)));
        }

        JSON_Array *methods_a = json_object_get_array(isup, "method_list");
        GHashTable *methods = g_hash_table_new(g_str_hash, g_str_equal);
        for (i = 0; i < json_array_get_count(methods_a); i++) {
                gpointer func = (void*) json_array_get_string(methods_a, i);
                g_hash_table_replace(methods, func, func);
        }

        ret->init = rpc_init;
        RPC_ADD_FUNC(login);
        RPC_ADD_OPT_FUNC(keepalive);
        RPC_ADD_FUNC(logout);
        RPC_ADD_FUNC(buddy_msg);
        RPC_ADD_OPT_FUNC(set_away);
        RPC_ADD_OPT_FUNC(send_typing);
        RPC_ADD_OPT_FUNC(add_buddy);
        RPC_ADD_OPT_FUNC(remove_buddy);
        RPC_ADD_OPT_FUNC(add_permit);
        RPC_ADD_OPT_FUNC(add_deny);
        RPC_ADD_OPT_FUNC(rem_permit);
        RPC_ADD_OPT_FUNC(rem_deny);
        RPC_ADD_OPT_FUNC(get_info);
        RPC_ADD_OPT_FUNC(chat_invite);
        RPC_ADD_OPT_FUNC(chat_kick);
        RPC_ADD_OPT_FUNC(chat_leave);
        RPC_ADD_OPT_FUNC(chat_msg);
        RPC_ADD_OPT_FUNC(chat_with);
        RPC_ADD_OPT_FUNC(chat_join);
        RPC_ADD_OPT_FUNC(chat_topic);
        if (proto_data->away_states)
                ret->away_states = rpc_away_states;
        
        g_hash_table_destroy(methods);

        // TODO: Property for a few standard nickcmp implementations.
        ret->handle_cmp = g_ascii_strcasecmp;
        
        register_protocol(ret);

        return TRUE;
}

#define PDIR "/tmp/rpcplugins"

/* YA RLY :-/ */
#ifndef UNIX_PATH_MAX
struct sockaddr_un sizecheck;
#define UNIX_PATH_MAX sizeof(sizecheck.sun_path)
#endif

void rpc_initmodule() {
        DIR *pdir = opendir(PDIR);
        struct dirent *de;

        if (!pdir)
                return;

        while ((de = readdir(pdir))) {
                if (de->d_type != DT_SOCK && de->d_type != DT_UNKNOWN)
                        continue;

                char *fn = g_build_filename(PDIR, de->d_name, NULL);
                struct sockaddr_un su;

                strncpy(su.sun_path, fn, UNIX_PATH_MAX);

#if 0
                struct stat fdata;
                if (stat(fn, &fdata) == -1) {
                        log_message(LOGLVL_WARNING, "Could not stat %s: %s", fn, strerror(errno));
                        g_free(fn);
                        continue;
                }
                /* For now just skip anything that is not a Unix domain socket. */
                if (!S_ISSOCK(fdata.st_mode))
                        continue;
#endif

                su.sun_path[UNIX_PATH_MAX-1] = '\0';
                su.sun_family = AF_UNIX;
                gboolean st = rpc_initmodule_sock((struct sockaddr*) &su, sizeof(su));
                g_free(fn);
                if (!st)
                        log_message(LOGLVL_WARNING, "Failed to register protocol %s", fn);
                /* Idea: Also support textfiles containing a host:port tuple to
                 * connect to. Not that remote RPC'ing would be a great idea,
                 * but maybe some jsonrpc libs don't support Unix domain sockets. */
        }
        closedir(pdir);
}