/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */

/*

 * Copyright (C) 2003-2006 Imendio AB

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser 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

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with this program; if not, write to the

 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,

 * Boston, MA 02111-1307, USA.

 */

#include <config.h>

#include <string.h>

#include <sys/stat.h> 

#include <sys/types.h>

#include <fcntl.h>

#include <glib.h>

#include "lm-sock.h"

#include "lm-debug.h"

#include "lm-error.h"

#include "lm-internals.h"

#include "lm-ssl-internals.h"

#include "lm-parser.h"

#include "lm-sha.h"

#include "lm-connection.h"

#include "lm-utils.h"

#define IN_BUFFER_SIZE 1024

typedef struct {

	LmHandlerPriority  priority;

	LmMessageHandler  *handler;

} HandlerData;

typedef struct {

	GSource       source;

	LmConnection *connection;

} LmIncomingSource;

struct _LmConnection {

	/* Parameters */

	GMainContext *context;

	gchar        *server;

	gchar        *jid;

	gchar        *resource;

	guint         port;

	LmSSL        *ssl;

	LmProxy      *proxy;

	LmParser     *parser;

	gchar        *stream_id;

	GHashTable   *id_handlers;

	GSList       *handlers[LM_MESSAGE_TYPE_UNKNOWN];

	/* Communication */

	GIOChannel   *io_channel;

	guint         io_watch_in;

	guint         io_watch_err;

	guint         io_watch_hup;

	LmSocket      fd;

	guint         io_watch_connect;

	guint         open_id;

	LmCallback   *open_cb;

 	gboolean      async_connect_waiting;

	gboolean      blocking;

	gboolean      cancel_open;

	LmCallback   *close_cb;     /* unused */

	LmCallback   *auth_cb;

	LmCallback   *register_cb;  /* unused */

	LmCallback   *disconnect_cb;

	GQueue       *incoming_messages;

	GSource      *incoming_source;

	LmConnectionState state;

	guint         keep_alive_rate;

	guint         keep_alive_id;

	guint         io_watch_out;

	GString      *out_buf;

	LmConnectData *connect_data;

	LmDisconnectReason disconnect_reason;

	gint          ref_count;

};

typedef enum {

	AUTH_TYPE_PLAIN  = 1,

	AUTH_TYPE_DIGEST = 2,

	AUTH_TYPE_0K     = 4

} AuthType;

static void     connection_free (LmConnection *connection);

static void     connection_handle_message    (LmConnection         *connection,

					      LmMessage            *message);

static void     connection_new_message_cb    (LmParser             *parser,

					      LmMessage            *message,

					      LmConnection         *connection);

static gboolean connection_do_open           (LmConnection         *connection,

					      GError              **error);

static void     connection_do_close           (LmConnection        *connection);

static gint     connection_do_write          (LmConnection         *connection,

					      const gchar          *buf,

					      gint                  len);

static gboolean connection_in_event          (GIOChannel   *source,

					      GIOCondition  condition,

					      LmConnection *connection);

static gboolean connection_error_event       (GIOChannel   *source,

					      GIOCondition  condition,

					      LmConnection *connection);

static gboolean connection_hup_event         (GIOChannel   *source,

					      GIOCondition  condition,

					      LmConnection *connection);

static gboolean connection_send              (LmConnection         *connection,

					      const gchar          *str,

					      gint                  len,

					      GError               **error);

static LmMessage *     connection_create_auth_req_msg (const gchar *username);

static LmMessage *     connection_create_auth_msg     (LmConnection *connection,

						       const gchar  *username,

						       const gchar  *password,

						       const gchar  *resource,

						       gint          auth_type);

static LmHandlerResult connection_auth_req_reply (LmMessageHandler *handler,

						  LmConnection     *connection,

						  LmMessage        *m,

						  gpointer          user_data);

static int connection_check_auth_type   (LmMessage           *auth_req_rpl);

static LmHandlerResult connection_auth_reply (LmMessageHandler    *handler,

					      LmConnection        *connection,

					      LmMessage           *m,

					      gpointer             user_data);

static void     connection_stream_error      (LmConnection        *connection,

					      LmMessage           *m);

static void     connection_stream_received   (LmConnection        *connection, 

					      LmMessage           *m);

static gint     connection_handler_compare_func (HandlerData  *a,

						 HandlerData  *b);

static gboolean connection_incoming_prepare  (GSource         *source,

					      gint            *timeout);

static gboolean connection_incoming_check    (GSource         *source);

static gboolean connection_incoming_dispatch (GSource         *source,

					      GSourceFunc      callback,

					      gpointer           user_data);

static GSource * connection_create_source       (LmConnection *connection);

static void      connection_signal_disconnect   (LmConnection *connection,

						 LmDisconnectReason reason);

static gboolean connection_do_connect           (LmConnectData *connect_data);

static guint    connection_add_watch            (LmConnection  *connection,

						 GIOChannel    *channel,

						 GIOCondition   condition,

						 GIOFunc        func,

						 gpointer       user_data);

static gboolean connection_send_keep_alive      (LmConnection  *connection);

static void     connection_start_keep_alive     (LmConnection  *connection);

static void     connection_stop_keep_alive      (LmConnection  *connection);

static gboolean connection_buffered_write_cb    (GIOChannel    *source, 

						 GIOCondition   condition,

						 LmConnection  *connection);

static gboolean connection_output_is_buffered   (LmConnection  *connection,

						 const gchar   *buffer,

						 gint           len);

static void     connection_setup_output_buffer  (LmConnection  *connection,

						 const gchar   *buffer,

						 gint           len);

static GSourceFuncs incoming_funcs = {

	connection_incoming_prepare,

	connection_incoming_check,

	connection_incoming_dispatch,

	NULL

};

static void

connection_free (LmConnection *connection)

{

	int        i;

	LmMessage *m;

	g_free (connection->server);

	g_free (connection->jid);

	g_free (connection->stream_id);

	g_free (connection->resource);

	if (connection->parser) {

		lm_parser_free (connection->parser);

	}

	/* Unref handlers */

	for (i = 0; i < LM_MESSAGE_TYPE_UNKNOWN; ++i) {

		GSList *l;

		for (l = connection->handlers[i]; l; l = l->next) {

			HandlerData *hd = (HandlerData *) l->data;

			lm_message_handler_unref (hd->handler);

			g_free (hd);

		}

		g_slist_free (connection->handlers[i]);

	}

	g_hash_table_destroy (connection->id_handlers);

	if (connection->state >= LM_CONNECTION_STATE_OPENING) {

		connection_do_close (connection);

	}

	if (connection->open_cb) {

		_lm_utils_free_callback (connection->open_cb);

	}

	if (connection->auth_cb) {

		_lm_utils_free_callback (connection->auth_cb);

	}

	lm_connection_set_disconnect_function (connection, NULL, NULL, NULL);

	while ((m = g_queue_pop_head (connection->incoming_messages)) != NULL) {

		lm_message_unref (m);

	}

	if (connection->ssl) {

		lm_ssl_unref (connection->ssl);

	}

	if (connection->proxy) {

		lm_proxy_unref (connection->proxy);

	}

	g_queue_free (connection->incoming_messages);

        if (connection->context) {

                g_main_context_unref (connection->context);

        }

	if (connection->out_buf) {

		g_string_free (connection->out_buf, TRUE);

	}

        g_free (connection);

}

static void

connection_handle_message (LmConnection *connection, LmMessage *m)

{

	LmMessageHandler *handler;

	GSList           *l;

	const gchar      *id;

	LmHandlerResult   result = LM_HANDLER_RESULT_ALLOW_MORE_HANDLERS;

	lm_connection_ref (connection);

	if (lm_message_get_type (m) == LM_MESSAGE_TYPE_STREAM) {

		connection_stream_received (connection, m);

		goto out;

	}

	else if (lm_message_get_type (m) == LM_MESSAGE_TYPE_STREAM_ERROR) {

		connection_stream_error (connection, m);

	}

	id = lm_message_node_get_attribute (m->node, "id");

	if (id) {

		handler = g_hash_table_lookup (connection->id_handlers, id);

		if (handler) {

			result = _lm_message_handler_handle_message (handler, 

								     connection,

								     m);

			g_hash_table_remove (connection->id_handlers, id);

		}

	}

	if (result == LM_HANDLER_RESULT_REMOVE_MESSAGE) {

		goto out;

	}

	for (l = connection->handlers[lm_message_get_type (m)]; 

	     l && result == LM_HANDLER_RESULT_ALLOW_MORE_HANDLERS; 

	     l = l->next) {

		HandlerData *hd = (HandlerData *) l->data;

		result = _lm_message_handler_handle_message (hd->handler,

							     connection,

							     m);

	}

out:

	lm_connection_unref (connection);

	return;

}

static void

connection_new_message_cb (LmParser     *parser,

			   LmMessage    *m,

			   LmConnection *connection)

{

	const gchar *from;

	lm_message_ref (m);

	from = lm_message_node_get_attribute (m->node, "from");

	if (!from) {

		from = "unknown";

	}

	lm_verbose ("New message with type=\"%s\" from: %s\n",

		    _lm_message_type_to_string (lm_message_get_type (m)),

		    from);

	g_queue_push_tail (connection->incoming_messages, m);

}

void

_lm_connection_succeeded (LmConnectData *connect_data)

{

	LmConnection *connection;

	LmMessage    *m;

	gchar        *server_from_jid;

	gchar        *ch;

	gchar        *ch_end;

	connection = connect_data->connection;

	if (connection->io_watch_connect != 0) {

		GSource *source;

		source = g_main_context_find_source_by_id (connection->context,

							   connection->io_watch_connect);

		if (source) {

			g_source_destroy (source);

		}

		connection->io_watch_connect = 0;

	}

	/* Need some way to report error/success */

	if (connection->cancel_open) {

		lm_verbose ("Cancelling connection...\n");

		return;

	}

	connection->fd = connect_data->fd;

	connection->io_channel = connect_data->io_channel;

	freeaddrinfo (connect_data->resolved_addrs);

	connection->connect_data = NULL;

	g_free (connect_data);

	if (connection->ssl) {

		GError *error = NULL;

		lm_verbose ("Setting up SSL...\n");

#ifdef HAVE_GNUTLS

		/* GNU TLS requires the socket to be blocking */

		_lm_sock_set_blocking (connection->fd, TRUE);

#endif

		if (!_lm_ssl_begin (connection->ssl, connection->fd,

				    connection->server,

				    &error)) {

			lm_verbose ("Could not begin SSL\n");

			if (error) {

				g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET,

				       "%s\n", error->message);

				g_error_free (error);

			}

 			_lm_sock_shutdown (connection->fd);

 			_lm_sock_close (connection->fd);

			connection_do_close (connection);