/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/*
* Copyright (C) 2006 Imendio AB
* Copyright (C) 2006 Nokia Corporation. All rights reserved.
* Copyright (C) 2007 Collabora Ltd.
*
* 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 <arpa/nameser.h>
#include <resolv.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");
_lm_socket_failed_with_error (connect_data, err);
socket->watch_connect = NULL;
return FALSE;
}
}
if (_lm_connection_async_connect_waiting (socket->connection)) {
gint res;
fd = g_io_channel_unix_get_fd (source);
res = _lm_sock_connect (fd, addr->ai_addr, (int)addr->ai_addrlen);
if (res < 0) {
err = _lm_sock_get_last_error ();
if (_lm_sock_is_blocking_success (err)) {
_lm_connection_set_async_connect_waiting (socket->connection, FALSE);
g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET,
"Connection success (1).\n");
_lm_socket_succeeded (connect_data);
}
if (_lm_connection_async_connect_waiting (socket->connection) &&
!_lm_sock_is_blocking_error (err)) {
g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET,
"Connection failed.\n");
_lm_sock_close (connect_data->fd);
_lm_socket_failed_with_error (connect_data, err);
socket->watch_connect = NULL;
return FALSE;
}
}
} else {
/* for blocking sockets, G_IO_OUT means we are connected */
g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET,
"Connection success (2).\n");
_lm_socket_succeeded (connect_data);
}
return TRUE;
}
static gboolean
socket_do_connect (LmConnectData *connect_data)
{
LmSocket *socket;
LmSocketT fd;
int res, err;
int port;
char name[NI_MAXHOST];
char portname[NI_MAXSERV];
struct addrinfo *addr;
socket = connect_data->socket;
addr = connect_data->current_addr;
if (socket->proxy) {
port = htons (lm_proxy_get_port (socket->proxy));
} else {
port = htons (socket->port);
}
((struct sockaddr_in *) addr->ai_addr)->sin_port = port;
res = getnameinfo (addr->ai_addr,
(socklen_t)addr->ai_addrlen,
name, sizeof (name),
portname, sizeof (portname),
NI_NUMERICHOST | NI_NUMERICSERV);
if (res < 0) {
return _lm_socket_failed (connect_data);
}
g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET,
"Trying %s port %s...\n", name, portname);
fd = _lm_sock_makesocket (addr->ai_family,
addr->ai_socktype,
addr->ai_protocol);
if (!_LM_SOCK_VALID (fd)) {
g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET,
"Failed making socket, error:%d...\n",
_lm_sock_get_last_error ());
return _lm_socket_failed (connect_data);
}
/* Even though it says _unix_new(), it is supported by glib on
* win32 because glib does some cool stuff to find out if it
* can treat it as a FD or a windows SOCKET.
*/
connect_data->fd = fd;
connect_data->io_channel = g_io_channel_unix_new (fd);
g_io_channel_set_encoding (connect_data->io_channel, NULL, NULL);
g_io_channel_set_buffered (connect_data->io_channel, FALSE);
_lm_sock_set_blocking (connect_data->fd, socket->blocking);
if (socket->proxy) {
socket->watch_connect =
lm_misc_add_io_watch (socket->context,
connect_data->io_channel,
G_IO_OUT|G_IO_ERR,
(GIOFunc) _lm_proxy_connect_cb,
connect_data);
} else {
socket->watch_connect =
lm_misc_add_io_watch (socket->context,
connect_data->io_channel,
G_IO_OUT|G_IO_ERR,
(GIOFunc) socket_connect_cb,
connect_data);
}
_lm_connection_set_async_connect_waiting (socket->connection, !socket->blocking);
res = _lm_sock_connect (connect_data->fd,
addr->ai_addr, (int)addr->ai_addrlen);
if (res < 0) {
err = _lm_sock_get_last_error ();
if (!_lm_sock_is_blocking_error (err)) {
_lm_sock_close (connect_data->fd);
return _lm_socket_failed_with_error (connect_data, err);
}
}
return TRUE;
}
gboolean
lm_socket_output_is_buffered (LmSocket *socket,
const gchar *buffer,
gint len)
{
if (socket->out_buf) {
lm_verbose ("Appending %d bytes to output buffer\n", len);
g_string_append_len (socket->out_buf, buffer, len);
return TRUE;
}
return FALSE;
}
void
lm_socket_setup_output_buffer (LmSocket *socket,
const gchar *buffer,
gint len)
{
lm_verbose ("OUTPUT BUFFER ENABLED\n");
socket->out_buf = g_string_new_len (buffer, len);
socket->watch_out =
lm_misc_add_io_watch (socket->context,
socket->io_channel,
G_IO_OUT,
(GIOFunc) socket_buffered_write_cb,
socket);
}
static gboolean
socket_buffered_write_cb (GIOChannel *source,
GIOCondition condition,
LmSocket *socket)
{
gint b_written;
GString *out_buf;
/* FIXME: Do the writing */
out_buf = socket->out_buf;
if (!out_buf) {
/* Should not be possible */
return FALSE;
}
b_written = lm_socket_do_write (socket, out_buf->str, out_buf->len);
if (b_written < 0) {
_lm_connection_error_event (socket,
G_IO_HUP, socket->connection);
return FALSE;
}
g_string_erase (out_buf, 0, (gsize) b_written);
if (out_buf->len == 0) {
lm_verbose ("Output buffer is empty, going back to normal output\n");
if (socket->watch_out) {
g_source_destroy (socket->watch_out);
socket->watch_out = NULL;
}
g_string_free (out_buf, TRUE);
socket->out_buf = NULL;
return FALSE;
}
return TRUE;
}
static gboolean
_parse_srv_response (unsigned char *srv, int srv_len, gchar **out_server, guint *out_port)
{
int qdcount;
int ancount;
int len;
const unsigned char *pos = srv + sizeof(HEADER);
unsigned char *end = srv + srv_len;
HEADER *head = (HEADER *)srv;
char name[256];
char pref_name[256];
guint pref_port = 0;
guint pref_prio = 9999;
pref_name[0] = 0;
qdcount = ntohs (head->qdcount);
ancount = ntohs (head->ancount);
/* Ignore the questions */
while (qdcount-- > 0 && (len = dn_expand (srv, end, pos, name, 255)) >= 0) {
g_assert (len >= 0);
pos += len + QFIXEDSZ;
}
/* Parse the answers */
while (ancount-- > 0 && (len = dn_expand (srv, end, pos, name, 255)) >= 0) {
/* Ignore the initial string */
uint16_t pref, weight, port;
g_assert (len >= 0);
pos += len;
/* Ignore type, ttl, class and dlen */
pos += 10;
GETSHORT (pref, pos);
GETSHORT (weight, pos);
GETSHORT (port, pos);
len = dn_expand (srv, end, pos, name, 255);
if (pref < pref_prio) {
pref_prio = pref;
strcpy (pref_name, name);
pref_port = port;
}
pos += len;
}
if (pref_name[0]) {
*out_server = g_strdup (pref_name);
*out_port = pref_port;
return TRUE;
}
return FALSE;
}
LmSocket *
lm_socket_create (GMainContext *context,
IncomingDataFunc func,
gpointer user_data,
LmConnection *connection,
gboolean blocking,
const gchar *server,
guint port,
LmSSL *ssl,
LmProxy *proxy,
GError **error)
{
LmSocket *socket;
struct addrinfo req;
struct addrinfo *ans;
const char *remote_addr;
LmConnectData *data;
int err;
char *srv;
#define SRV_LEN 8192
unsigned char srv_ans[SRV_LEN];
g_return_val_if_fail (server != NULL, NULL);
g_return_val_if_fail ((port >= MIN_PORT && port <= MAX_PORT), NULL);
g_return_val_if_fail (func != NULL, NULL);
socket = g_new0 (LmSocket, 1);
memset (&req, 0, sizeof(req));
req.ai_family = AF_UNSPEC;
req.ai_socktype = SOCK_STREAM;
req.ai_protocol = IPPROTO_TCP;
socket->ref_count = 1;
socket->connection = connection;
socket->server = g_strdup (server);
socket->port = port;
socket->cancel_open = FALSE;
socket->ssl = NULL;
socket->proxy = NULL;
socket->blocking = blocking;
socket->func = func;
socket->user_data = user_data;
res_init ();
srv = g_strdup_printf ("_xmpp-client._tcp.%s", socket->server);
err = res_query (srv, C_IN, T_SRV, srv_ans, SRV_LEN);
if (err > 0) {
_parse_srv_response (srv_ans, err, &(socket->server), &(socket->port));
}
g_free (srv);
if (context) {
socket->context = g_main_context_ref (context);
}
if (proxy) {
socket->proxy = lm_proxy_ref (proxy);
remote_addr = lm_proxy_get_server (socket->proxy);
} else {
remote_addr = socket->server;
}
g_log (LM_LOG_DOMAIN, LM_LOG_LEVEL_NET,
"Going to connect to %s%s:%u\n", (proxy) ? "proxy " : "",
remote_addr, socket->port);
err = getaddrinfo (remote_addr, NULL, &req, &ans);
if (err != 0) {
const char *str;
str = _lm_sock_addrinfo_get_error_str (err);
g_set_error (error,
LM_ERROR,
LM_ERROR_CONNECTION_FAILED,
str);
return NULL;
}
if (ssl) {
socket->ssl = lm_ssl_ref (ssl);
_lm_ssl_initialize (socket->ssl);
}
/* Prepare and do the nonblocking connection */
data = g_new (LmConnectData, 1);
data->socket = socket;
data->connection = connection;
data->resolved_addrs = ans;
data->current_addr = ans;
data->io_channel = NULL;
data->fd = -1;
socket->connect_data = data;
if (!socket_do_connect (data)) {
g_set_error (error,
LM_ERROR,
LM_ERROR_CONNECTION_FAILED,
"unable to connect");
return NULL;
}
return socket;
}
void
lm_socket_flush (LmSocket *socket)
{
g_return_if_fail (socket != NULL);
g_return_if_fail (socket->io_channel != NULL);
g_io_channel_flush (socket->io_channel, NULL);
}
void
lm_socket_close (LmSocket *socket)
{
LmConnectData *data;
g_return_if_fail (socket != NULL);
if (socket->watch_connect) {
g_source_destroy (socket->watch_connect);
socket->watch_connect = NULL;
}
data = socket->connect_data;
if (data) {
freeaddrinfo (data->resolved_addrs);
socket->connect_data = NULL;
g_free (data);
}
if (socket->io_channel) {
if (socket->watch_in) {
g_source_destroy (socket->watch_in);
socket->watch_in = NULL;
}
if (socket->watch_err) {
g_source_destroy (socket->watch_err);
socket->watch_err = NULL;
}
if (socket->watch_hup) {
g_source_destroy (socket->watch_hup);
socket->watch_hup = NULL;
}
if (socket->watch_out) {
g_source_destroy (socket->watch_out);
socket->watch_out = NULL;
}
g_io_channel_unref (socket->io_channel);
socket->io_channel = NULL;
socket->fd = -1;
}
}
LmSocket *
lm_socket_ref (LmSocket *socket)
{
g_return_val_if_fail (socket != NULL, NULL);
socket->ref_count++;
return socket;
}
void
lm_socket_unref (LmSocket *socket)
{
g_return_if_fail (socket != NULL);
socket->ref_count--;
if (socket->ref_count <= 0) {
socket_free (socket);
}
}