#include "config.h"
#ifdef HAVE_STDLIB_H
#include <stdlib.h>
#endif
#ifdef HAVE_STRING_H
#include <string.h>
#endif
#include "icalcap.h"
#include "icalcap_session.h"
#include "icalcap_server.h"
#include "icalcap_impl.h"
#include "icalcap_message_impl.h"
#include "icalcap_session_impl.h"
#include "icalcap_server_impl.h"
icalcomponent *icalcap_component_new_from_string(const char *data);
static int default_msg_handler(RRCAP *cap, RRFrame *frame, GError **error);
static void client_final_handler(RRCAP *cap);
/**
* Implementation of functions in icalcap
*/
void
icalcap_free_rr(icalcap *cap) {
g_return_if_fail(cap);
if (cap->username)
free(cap->username);
if (cap->authname)
free(cap->authname);
g_free(cap);
}
/**
* Implementation of functions in icalcap_session
*/
icalcap_session *
icalcap_session_new_rr(void) {
icalcap_session *sess;
GError *error = NULL;
/* Initialize roadrunner */
if (!rr_init(0, NULL, &error))
return 0;
if ((sess = g_new0(icalcap_session, 1)) == NULL) {
/* FIXME return an error */
return NULL;
}
sess->cfg = rr_cap_config_new();
/* Tell roadrunner which profiles we want to support */
sess->profreg = rr_profile_registry_new();
rr_profile_registry_add_profile(sess->profreg, RR_TYPE_CAP, NULL);
rr_profile_registry_add_profile(sess->profreg, RR_TYPE_TLS, NULL);
rr_profile_registry_add_profile(sess->profreg, RR_TYPE_SASL_DIGEST_MD5, NULL);
return sess;
}
int
icalcap_session_connect_rr(icalcap_session *sess, const char *hostname, const int port) {
GError *error = NULL;
if (sess == NULL) {
/* FIXME return the error */
return FALSE;
}
/* Create a connection object */
sess->connection = rr_tcp_connection_new(sess->profreg, hostname, port, &error);
if (sess->connection == NULL) {
/* FIXME return the error */
return FALSE;
}
return TRUE;
}
int
icalcap_session_login_rr(icalcap_session *sess, const char *username, const char *authname,
const char *password) {
/* assert cap != NULL */
GError *error = NULL;
rr_sasl_set_username(sess->connection, username);
rr_sasl_set_authname(sess->connection, authname);
rr_sasl_set_password(sess->connection, password);
/* FIXME */
if (!rr_sasl_login(sess->connection, RR_TYPE_SASL_DIGEST_MD5,
"foo.example.com", NULL, &error)) {
/* FIXME return the error */
return 0;
}
return 1;
}
icalcap *
icalcap_session_start_rr(const icalcap_session *sess, icalcap_msg_handler handler) {
/* assert sess != NULL */
icalcap *cap;
RRCAP *channel;
GError *error = NULL;
if ((cap = g_new0(icalcap, 1)) == NULL) {
/* FIXME return an error */
return NULL;
}
if (handler != NULL) {
rr_cap_config_set_msg_handler(sess->cfg, default_msg_handler, (void *)handler);
/* FIXME rr_cap_config_set_final_handler(cfg, client_final_handler); */
}
if ((channel = rr_cap_client_start(sess->connection, sess->cfg, &error)) == NULL) {
/* FIXME return the error */
goto FAILED;
}
cap->chan = channel;
channel->hl = cap;
return cap;
FAILED:
g_free(cap);
return NULL;
}
int
icalcap_stop_rr(icalcap *cap) {
/* assert cap != NULL */
GError *error = NULL;
if (!rr_cap_close(cap->chan, &error)) {
/* FIXME return the error */
return 0;
}
cap->chan = NULL;
g_free(cap);
return 1;
}
int
icalcap_session_disconnect_rr(icalcap_session *sess) {
/* assert cap != NULL */
GError *error = NULL;
if (!rr_connection_disconnect(sess->connection, &error)) {
/* FIXME return the error */
return 0;
}
sess->connection = NULL;
g_free(sess);
if (!rr_exit(&error)) {
/* FIXME return the error */
return 0;
}
return 1;
}
/**
* Implementation of functions in icalcap_server
*/
/**
* If the user properly authenticated (via SASL), initialize the channel
* credentials. Otherwise, refuse to open the channel.
*/
static int
server_init_handler(RRCAP *chan, const gchar *piggyback, GError **error) {
icalcap *cap;
RRConnection *connection;
icalcap_auth_handler func;
const gchar *username, *authname;
int rc;
g_return_val_if_fail(chan, FALSE);
g_return_val_if_fail(chan->cfg, FALSE);
g_return_val_if_fail(chan->cfg->server_init_data, FALSE);
connection = rr_channel_get_connection(RR_CHANNEL(chan));
if ((username = rr_sasl_get_username(connection)) == NULL)
return FALSE;
if ((authname = rr_sasl_get_authname(connection)) == NULL)
return FALSE;
if ((cap = g_new0(icalcap, 1)) == NULL) {
return FALSE;
}
cap->chan = chan;
chan->hl = cap;
cap->username = strdup(username);
cap->authname = strdup(authname);
func = (icalcap_auth_handler)chan->cfg->server_init_data;
return func(cap, piggyback);
}
static void
server_confirmation_handler(RRCAP *chan) {
icalcap_chanup_handler func;
g_return_if_fail(chan);
g_return_if_fail(chan->cfg);
g_return_if_fail(chan->cfg->server_confirmation_data);
func = (icalcap_chanup_handler)chan->cfg->server_confirmation_data;
func(chan->hl);
}
static gboolean
server_frame_handler(RRCAP *cap, RRFrame *frame, GError **error)
{
if (frame->type == RR_FRAME_TYPE_MSG)
/* FIXME */
return default_msg_handler(cap, frame, error);
else
return FALSE;
}
static void
server_final_handler(RRCAP *cap)
{
g_return_if_fail(cap);
g_return_if_fail(RR_CAP(cap));
if (cap->hl != NULL) {
icalcap_free(cap->hl);
cap->hl = NULL;
}
}
/*
* FIXME Do we want to pass argc and argv in?
*/
icalcap_server *
icalcap_server_new_rr(icalcap_auth_handler auth_cb, icalcap_chanup_handler chanup_cb,
icalcap_msg_handler msg_cb) {
icalcap_server *serv;
GError *error = NULL;
/* Initialize roadrunner */
if (!rr_init(0, NULL, &error))
return 0;
if ((serv = g_new0(icalcap_server, 1)) == NULL) {
/* FIXME return an error */
return NULL;
}
serv->handler = msg_cb;
/* This is somewhat hackish */
serv->cfg = rr_cap_config_new();
rr_cap_config_set_msg_handler(serv->cfg, NULL, (void *)msg_cb);
rr_cap_config_set_frame_handler(serv->cfg, server_frame_handler);
rr_cap_config_set_final_handler(serv->cfg, server_final_handler);
rr_cap_config_set_server_init_handler(serv->cfg, server_init_handler, (void *)auth_cb);
rr_cap_config_set_server_confirmation_handler(serv->cfg,
server_confirmation_handler, (void *)chanup_cb);
/* Tell roadrunner which profiles we want to support */
serv->profreg = rr_profile_registry_new();
rr_profile_registry_add_profile(serv->profreg, RR_TYPE_CAP, serv->cfg);
rr_profile_registry_add_profile(serv->profreg, RR_TYPE_TLS, NULL);
rr_profile_registry_add_profile(serv->profreg, RR_TYPE_SASL_DIGEST_MD5, NULL);
return serv;
}
int
icalcap_server_listen_rr(icalcap_server *serv, const char *hostname, const int port) {
GError *error = NULL;
g_return_val_if_fail(serv, FALSE);
/* Create a listener object */
serv->listener = rr_tcp_listener_new(serv->profreg, hostname, port, &error);
if (serv->listener == NULL) {
/* FIXME return the error */
return FALSE;
}
return TRUE;
}
int
icalcap_server_run_rr(const icalcap_server *serv) {
/* assert cap != NULL */
GError *error = NULL;
if (!rr_wait_until_done(&error)) {
/* FIXME return the error */
return FALSE;
}
return TRUE;
}
int
icalcap_server_shutdown_rr(icalcap_server *serv) {
/* assert cap != NULL */
GError *error = NULL;
if (!rr_listener_shutdown(serv->listener, &error)) {
/* FIXME return the error */
return 0;
}
serv->listener = NULL;
g_free(serv);
return 1;
}
/*
* icalcap_message.c
*/
/*
* Internal constructor
*/
static struct _icalcap_message_rr *
_icalcap_message_new_from_component_rr(const icalcap *cap, int type, icalcomponent *comp) {
struct _icalcap_message_rr *ret;
if ((ret = g_new0(struct _icalcap_message_rr, 1)) == NULL) {
/* FIXME return an error */
return NULL;
}
ret->cap = cap;
ret->type = type;
ret->comp = comp;
return ret;
}
static icalcap_message *
_icalcap_message_new_from_frame_rr(const icalcap *cap, int type, RRFrame *frame) {
struct _icalcap_message_rr *ret;
if ((ret = g_new0(struct _icalcap_message_rr, 1)) == NULL) {
/* FIXME return an error */
return NULL;
}
ret->cap = cap;
ret->type = type;
ret->frame = frame;
ret->comp = icalcap_component_new_from_string(ret->frame->payload);
return (icalcap_message *)ret;
}
icalcap_message *
icalcap_message_new_rr(const icalcap *cap, const icalcomponent *comp) {
struct _icalcap_message_rr *ret;
gchar *str;
if (comp == NULL) {
/* FIXME return an error */
return NULL;
}
ret = _icalcap_message_new_from_component_rr(cap, ICALCAP_MESSAGE_CMD, NULL);
str = g_strdup_printf("%s\n\n%s",
"Content-Type: text/calendar",
icalcomponent_as_ical_string(comp));
ret->msg = rr_message_static_new(RR_FRAME_TYPE_MSG, str, strlen(str), TRUE);
return (icalcap_message *)ret;
}
/*
* This method and its implementation are critical. It has the responsibility for
* serializing the component. The tricky part is that we have a options for encoding XROOT.
*/
icalcap_message *
icalcap_message_new_reply_rr(const icalcap_message *orig, const icalcomponent *comp) {
struct _icalcap_message_rr *in, *ret;
icalcomponent *cc;
GString *str;
if ((in = (struct _icalcap_message_rr *)orig) == NULL) {
/* FIXME return an error */
return NULL;
}
ret = _icalcap_message_new_from_component_rr(in->cap, ICALCAP_MESSAGE_REPLY, NULL);
/* FIXME */
if (icalcomponent_isa(comp) != ICAL_XROOT_COMPONENT)
return NULL;
/* FIXME don't hardcode */
str = g_string_new("Content-Type: text/calendar\n\n");
for (cc = icalcomponent_get_first_component(comp,
ICAL_VCALENDAR_COMPONENT);
cc != NULL;
cc = icalcomponent_get_next_component(comp,
ICAL_VCALENDAR_COMPONENT)) {
g_string_append(str, icalcomponent_as_ical_string(cc));
}
ret->msg = rr_message_static_new(RR_FRAME_TYPE_RPY, str->str, strlen(str->str), TRUE);
/* FIXME this should now be ok but if change the API we need to change */
ret->msg->msgno = in->frame->msgno;
return (icalcap_message *)ret;
}
void
icalcap_message_free_rr(icalcap_message *in) {
struct _icalcap_message_rr *capmsg = (struct _icalcap_message_rr *)in;
g_return_if_fail(capmsg);
if (capmsg->comp != NULL) {
icalcomponent_free(capmsg->comp);
capmsg->comp = NULL;
}
if (capmsg->msg != NULL) {
g_free(capmsg->msg);
}
g_free(capmsg);
}
int
icalcap_message_send_reply_rr(icalcap_message *in) {
struct _icalcap_message_rr *capmsg = (struct _icalcap_message_rr *)in;
GError *error = NULL;
int rc;
if ((rc = rr_channel_send_message(RR_CHANNEL(capmsg->cap->chan), capmsg->msg, &error)) == 0)
g_message("error = %s", error->message);
/* FIXME handle error */
capmsg->msg = NULL;
return rc;
}
icalcomponent *
icalcap_message_sync_send_rr(icalcap_message *in, int timeout) {
struct _icalcap_message_rr *capmsg = (struct _icalcap_message_rr *)in;
icalcomponent *comp, *ret;
gchar *str2;
GError *error = NULL;
int rc;
/* FIXME */
rc = rr_cap_cmd(capmsg->cap->chan, capmsg->msg, 3 * timeout, &str2, &error);
capmsg->msg = NULL;
if (rc == 0) {
g_message("error = %s", error->message);
/* FIXME handle error */
return 0;
}
comp = icalcap_component_new_from_string(str2);
g_free(str2);
if (ret == NULL)
return NULL;
return comp;
}
/*
*
*/
/*
* FIXME We assume we can safely create an icalcap_message derived object
* without calls to the base object
*/
int
default_msg_handler(RRCAP *cap, RRFrame *frame, GError **error) {
icalcap_msg_handler func;
icalcap_message *msg;
int ret;
g_return_val_if_fail(cap, FALSE);
g_return_val_if_fail(cap->cfg, FALSE);
g_return_val_if_fail(cap->cfg->msg_handler_data, FALSE);
func = (icalcap_msg_handler)cap->cfg->msg_handler_data;
msg = _icalcap_message_new_from_frame_rr(cap->hl, ICALCAP_MESSAGE_CMD, frame);
if (msg == NULL) {
/* FIXME */
g_message("error");
return FALSE;
}
ret = func(msg);
icalcap_message_free(msg);
return ret;
}
