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

/*

 * Copyright (C) 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 "lm-debug.h"

#include "lm-internals.h"

#include "lm-misc.h"

#include "lm-ssl.h"

#include "lm-ssl-internals.h"

#include "lm-proxy.h"

#include "lm-socket.h"

#include "lm-sock.h"

#include "lm-error.h"

#define IN_BUFFER_SIZE 1024

#define MIN_PORT 1

#define MAX_PORT 65536

struct _LmSocket {

	LmConnection *connection;

	GMainContext *context;

	gchar        *server;

	guint         port;

	gboolean      blocking;

	LmSSL        *ssl;

	LmProxy      *proxy;

	GIOChannel   *io_channel;

	GSource      *watch_in;

	GSource      *watch_err;

	GSource      *watch_hup;

	LmSocketT      fd;

	GSource      *watch_connect;

	gboolean      cancel_open;

	GSource      *watch_out;

	GString      *out_buf;

	LmConnectData *connect_data;

	IncomingDataFunc func;

	gpointer         user_data;

	guint          ref_count;

}; 

static void         socket_free                 (LmSocket      *socket);

static gboolean     socket_do_connect        (LmConnectData *connect_data);

static gboolean     socket_connect_cb       (GIOChannel    *source, 

						 GIOCondition   condition,

						 LmConnectData *connect_data);

static gboolean socket_in_event          (GIOChannel   *source,

					      GIOCondition  condition,

					      LmSocket     *socket);

static gboolean socket_hup_event         (GIOChannel   *source,

					      GIOCondition  condition,

					      LmSocket     *socket);

static gboolean

socket_buffered_write_cb (GIOChannel   *source, 

			      GIOCondition  condition,

			      LmSocket     *socket);

static void

socket_free (LmSocket *socket)

{

	g_free (socket->server);

	if (socket->ssl) {

		lm_ssl_unref (socket->ssl);

	}

	if (socket->proxy) {

		lm_proxy_unref (socket->proxy);

	}

	if (socket->out_buf) {

		g_string_free (socket->out_buf, TRUE);

	}

	g_free (socket);

}

gint

lm_socket_do_write (LmSocket     *socket,

		     const gchar  *buf,

		     gint          len)

{

	gint b_written;

	if (socket->ssl) {

		b_written = _lm_ssl_send (socket->ssl, buf, len);

	} else {

		GIOStatus io_status = G_IO_STATUS_AGAIN;

		gsize     bytes_written;

		while (io_status == G_IO_STATUS_AGAIN) {

			io_status = g_io_channel_write_chars (socket->io_channel, 

							      buf, len, 

							      &bytes_written,

							      NULL);

		}

		b_written = bytes_written;

		if (io_status != G_IO_STATUS_NORMAL) {

			b_written = -1;

		}

	}

	return b_written;

}

static gboolean

socket_read_incoming (LmSocket     *socket,

		      gchar        *buf,

		      gsize         buf_size,

		      gsize        *bytes_read,

		      gboolean     *hangup)

{

	GIOStatus status;

	*hangup = FALSE;

	if (socket->ssl) {

		status = _lm_ssl_read (socket->ssl, 

				       buf, buf_size - 1, bytes_read);

	} else {

		status = g_io_channel_read_chars (socket->io_channel,

						  buf, buf_size - 1,

						  bytes_read,

						  NULL);

	}

	if (status != G_IO_STATUS_NORMAL || *bytes_read < 0) {

		gint reason;

		switch (status) {

		case G_IO_STATUS_EOF:

			reason = LM_DISCONNECT_REASON_HUP;

			break;

		case G_IO_STATUS_AGAIN:

			/* No data readable but we didn't hangup */

			return FALSE;

			break;

		case G_IO_STATUS_ERROR:

			reason = LM_DISCONNECT_REASON_ERROR;

			break;

		default:

			reason = LM_DISCONNECT_REASON_UNKNOWN;

		}

		_lm_connection_do_close (socket->connection);

		_lm_connection_signal_disconnect (socket->connection, reason);

		/* Notify connection_in_event that we hangup the connection */

		*hangup = TRUE;

		return FALSE;

	}

	buf[*bytes_read] = '\0';

	/* There is more data to be read */

	return TRUE;

}

static gboolean

socket_in_event (GIOChannel   *source,

		     GIOCondition  condition,

		     LmSocket     *socket)

{

	gchar     buf[IN_BUFFER_SIZE];

	gsize     bytes_read;

	gboolean  hangup;

	if (!socket->io_channel) {

		return FALSE;

	}

	while (socket_read_incoming (socket, buf, IN_BUFFER_SIZE, 

				     &bytes_read, &hangup)) {

		g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET, "\nRECV [%d]:\n", 

		       (int)bytes_read);

		g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET, 

		       "-----------------------------------\n");

		g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET, "'%s'\n", buf);

		g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET, 

		       "-----------------------------------\n");

		lm_verbose ("Read: %d chars\n", (int)bytes_read);

		(socket->func) (socket, buf, socket->user_data);

	}

	if (hangup) {

		return FALSE;

	}

	return TRUE;

}

static gboolean

socket_hup_event (GIOChannel   *source,

		      GIOCondition  condition,

		      LmSocket     *socket)

{

	lm_verbose ("HUP event: %d->'%s'\n", 

		    condition, lm_misc_io_condition_to_str (condition));

	if (!socket->io_channel) {

		return FALSE;

	}

	_lm_connection_do_close (socket->connection);

	_lm_connection_signal_disconnect (socket->connection, 

					  LM_DISCONNECT_REASON_HUP);

	return TRUE;

}

void

_lm_socket_succeeded (LmConnectData *connect_data)

{

	LmSocket     *socket;

	socket = connect_data->socket;

	if (socket->watch_connect) {

		g_source_destroy (socket->watch_connect);

		socket->watch_connect = NULL;

	}

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

	if (socket->cancel_open) {

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

		_lm_connection_socket_result (socket->connection, FALSE);

		return;

	}

	socket->fd = connect_data->fd;

	socket->io_channel = connect_data->io_channel;

	freeaddrinfo (connect_data->resolved_addrs);

	socket->connect_data = NULL;

	g_free (connect_data);

	if (socket->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 (socket->fd, TRUE);

#endif

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

				    socket->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 (socket->fd);

 			_lm_sock_close (socket->fd);

			_lm_connection_do_close (socket->connection);

			_lm_connection_socket_result (socket->connection, FALSE);

			return;

		}

#ifdef HAVE_GNUTLS

		_lm_sock_set_blocking (socket->fd, FALSE); 

#endif

	}

	socket->watch_in = 

		lm_misc_add_io_watch (socket->context,

				      socket->io_channel,

				      G_IO_IN,

				      (GIOFunc) socket_in_event,

				      socket);

	/* FIXME: if we add these, we don't get ANY

	 * response from the server, this is to do with the way that

	 * windows handles watches, see bug #331214.

	 */

#ifndef G_OS_WIN32

	socket->watch_err = 

		lm_misc_add_io_watch (socket->context,

				      socket->io_channel,

				      G_IO_ERR,

				      (GIOFunc) _lm_connection_error_event,

				      socket);

	socket->watch_hup =

		lm_misc_add_io_watch (socket->context,

				      socket->io_channel,

				      G_IO_HUP,

				      (GIOFunc) socket_hup_event,

				      socket);

#endif

	_lm_connection_socket_result (socket->connection, TRUE);

}

gboolean 

_lm_socket_failed_with_error (LmConnectData *connect_data, int error) 

{

	LmSocket *socket;

	g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET,

	       "Connection failed: %s (error %d)\n",

	       _lm_sock_get_error_str (error), error);

	socket = connect_data->socket;

	connect_data->current_addr = connect_data->current_addr->ai_next;

	if (socket->watch_connect) {

		g_source_destroy (socket->watch_connect);

		socket->watch_connect = NULL;

	}

	if (connect_data->io_channel != NULL) {

		g_io_channel_unref (connect_data->io_channel);

		/* FIXME: need to check for last unref and close the socket */

	}

	if (connect_data->current_addr == NULL) {

		_lm_connection_socket_result (socket->connection, FALSE);

		 /* if the user callback called connection_close(), this is already freed */

		if (socket->connect_data != NULL) {

			freeaddrinfo (connect_data->resolved_addrs);

			socket->connect_data = NULL;

			g_free (connect_data);

		}

	} else {

		/* try to connect to the next host */

		return socket_do_connect (connect_data);

	}

	return FALSE;

}

gboolean 

_lm_socket_failed (LmConnectData *connect_data)

{

	return _lm_socket_failed_with_error (connect_data,

                                       _lm_sock_get_last_error());

}

static gboolean 

socket_connect_cb (GIOChannel   *source, 

		       GIOCondition  condition,

		       LmConnectData *connect_data) 

{

	LmSocket        *socket;

	struct addrinfo *addr;

	int              err;

	socklen_t        len;

	LmSocketT         fd; 

	socket = connect_data->socket;

	addr = connect_data->current_addr;

	fd = g_io_channel_unix_get_fd (source);

	if (condition == G_IO_ERR) {

		len = sizeof (err);

		_lm_sock_get_error (fd, &err, &len);

		if (!_lm_sock_is_blocking_error (err)) {

			g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET,

			       "Connection failed.\n");