/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/* camel-remote-store.c : class for an remote store */
/*
* Authors: Peter Williams <peterw@helixcode.com>
* based on camel-imap-provider.c
*
* Copyright 2000 Helix Code, Inc. (www.helixcode.com)
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA.
*
*/
#include <config.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <gal/util/e-util.h>
#include "camel-remote-store.h"
#include "camel-folder.h"
#include "camel-exception.h"
#include "camel-session.h"
#include "camel-stream.h"
#include "camel-stream-buffer.h"
#include "camel-stream-fs.h"
#include "camel-url.h"
#include "string-utils.h"
#define d(x) x
#if d(!)0
extern gboolean camel_verbose_debug;
#endif
#define CSRVC(obj) (CAMEL_SERVICE_CLASS (CAMEL_OBJECT_GET_CLASS (obj)))
#define CSTRC(obj) (CAMEL_STORE_CLASS (CAMEL_OBJECT_GET_CLASS (obj)))
#define CRSC(obj) (CAMEL_REMOTE_STORE_CLASS (CAMEL_OBJECT_GET_CLASS (obj)))
static CamelStoreClass *store_class = NULL;
static gboolean remote_connect (CamelService *service, CamelException *ex);
static gboolean remote_disconnect (CamelService *service, gboolean clean, CamelException *ex);
static GList *remote_query_auth_types_generic (CamelService *service, CamelException *ex);
static GList *remote_query_auth_types_connected (CamelService *service, CamelException *ex);
static void remote_free_auth_types (CamelService *service, GList *authtypes);
static char *remote_get_name (CamelService *service, gboolean brief);
static char *remote_get_folder_name (CamelStore *store,
const char *folder_name,
CamelException *ex);
static gint remote_send_string (CamelRemoteStore *store, CamelException *ex,
char *fmt, va_list ap);
static gint remote_send_stream (CamelRemoteStore *store, CamelStream *stream,
CamelException *ex);
static gint remote_recv_line (CamelRemoteStore *store, char **dest,
CamelException *ex);
static void
camel_remote_store_class_init (CamelRemoteStoreClass *camel_remote_store_class)
{
/* virtual method overload */
CamelServiceClass *camel_service_class =
CAMEL_SERVICE_CLASS (camel_remote_store_class);
CamelStoreClass *camel_store_class =
CAMEL_STORE_CLASS (camel_remote_store_class);
store_class = CAMEL_STORE_CLASS (camel_type_get_global_classfuncs (camel_store_get_type ()));
/* virtual method overload */
camel_service_class->connect = remote_connect;
camel_service_class->disconnect = remote_disconnect;
camel_service_class->query_auth_types_generic = remote_query_auth_types_generic;
camel_service_class->query_auth_types_connected = remote_query_auth_types_connected;
camel_service_class->free_auth_types = remote_free_auth_types;
camel_service_class->get_name = remote_get_name;
camel_store_class->get_folder_name = remote_get_folder_name;
camel_remote_store_class->send_string = remote_send_string;
camel_remote_store_class->send_stream = remote_send_stream;
camel_remote_store_class->recv_line = remote_recv_line;
camel_remote_store_class->keepalive = NULL;
}
static void
camel_remote_store_init (CamelObject *object)
{
CamelStore *store = CAMEL_STORE (object);
CamelRemoteStore *remote_store = CAMEL_REMOTE_STORE (object);
store->folders = g_hash_table_new (g_str_hash, g_str_equal);
remote_store->istream = NULL;
remote_store->ostream = NULL;
remote_store->timeout_id = 0;
}
/*
*static void
*camel_remote_store_finalize (CamelObject *object)
*{
* CamelRemoteStore *remote_store = CAMEL_REMOTE_STORE (object);
*
* g_free (remote_store->nice_name);
*}
*/
CamelType
camel_remote_store_get_type (void)
{
static CamelType camel_remote_store_type = CAMEL_INVALID_TYPE;
if (camel_remote_store_type == CAMEL_INVALID_TYPE) {
camel_remote_store_type =
camel_type_register (CAMEL_STORE_TYPE, "CamelRemoteStore",
sizeof (CamelRemoteStore),
sizeof (CamelRemoteStoreClass),
(CamelObjectClassInitFunc) camel_remote_store_class_init,
NULL,
(CamelObjectInitFunc) camel_remote_store_init,
(CamelObjectFinalizeFunc) NULL);
}
return camel_remote_store_type;
}
/* Auth stuff */
/*
static CamelServiceAuthType password_authtype = {
N_("SSH Tunneling"),
N_("This option will connect to the server using a "
"SSH tunnel."),
"",
TRUE
};
*/
static GList *
remote_query_auth_types_connected (CamelService *service, CamelException *ex)
{
return NULL;
}
static GList *
remote_query_auth_types_generic (CamelService *service, CamelException *ex)
{
return NULL;
}
static void
remote_free_auth_types (CamelService *service, GList *authtypes)
{
g_list_free (authtypes);
}
static char *
remote_get_name (CamelService *service, gboolean brief)
{
if (brief)
return g_strdup_printf (_("%s server %s"),
service->provider->name,
service->url->host);
else {
return g_strdup_printf (_("%s service for %s on %s"),
service->provider->name,
service->url->user,
service->url->host);
}
}
static gboolean
timeout_cb (gpointer data)
{
CRSC (data)->keepalive (CAMEL_REMOTE_STORE (data));
return TRUE;
}
static gboolean
remote_connect (CamelService *service, CamelException *ex)
{
CamelRemoteStore *store = CAMEL_REMOTE_STORE (service);
struct hostent *h;
struct sockaddr_in sin;
gint fd;
gint port;
h = camel_service_gethost (service, ex);
if (!h)
return FALSE;
/* connect to the server */
sin.sin_family = h->h_addrtype;
if (service->url->port)
port = service->url->port;
else
port = store->default_port;
sin.sin_port = htons (port);
memcpy (&sin.sin_addr, h->h_addr, sizeof (sin.sin_addr));
fd = socket (h->h_addrtype, SOCK_STREAM, 0);
if (fd == -1 || connect (fd, (struct sockaddr *)&sin, sizeof (sin)) == -1) {
camel_exception_setv (ex, CAMEL_EXCEPTION_SERVICE_UNAVAILABLE,
_("Could not connect to %s (port %d): %s"),
service->url->host ? service->url->host : _("(unknown host)"),
port, g_strerror (errno));
if (fd > -1)
close (fd);
return FALSE;
}
/* parent class connect initialization */
if (CAMEL_SERVICE_CLASS (store_class)->connect (service, ex) == FALSE)
return FALSE;
store->ostream = camel_stream_fs_new_with_fd (fd);
store->istream = camel_stream_buffer_new (store->ostream, CAMEL_STREAM_BUFFER_READ);
/* Okay, good enough for us */
CAMEL_SERVICE (store)->connected = TRUE;
/* Add a timeout so that we can hopefully prevent getting disconnected */
/* (Only if the implementation supports it) */
if (CRSC (store)->keepalive) {
CamelSession *session = camel_service_get_session (CAMEL_SERVICE (store));
store->timeout_id = camel_session_register_timeout (session, 10 * 60 * 1000,
timeout_cb,
store);
}
return TRUE;
}
static gboolean
remote_disconnect (CamelService *service, gboolean clean, CamelException *ex)
{
CamelRemoteStore *store = CAMEL_REMOTE_STORE (service);
if (store->timeout_id) {
camel_session_remove_timeout (camel_service_get_session (CAMEL_SERVICE (store)),
store->timeout_id);
store->timeout_id = 0;
}
if (!CAMEL_SERVICE_CLASS (store_class)->disconnect (service, clean, ex))
return FALSE;
if (store->istream) {
camel_object_unref (CAMEL_OBJECT (store->istream));
store->istream = NULL;
}
if (store->ostream) {
camel_object_unref (CAMEL_OBJECT (store->ostream));
store->ostream = NULL;
}
return TRUE;
}
static gchar *
remote_get_folder_name (CamelStore *store, const char *folder_name, CamelException *ex)
{
return g_strdup (folder_name);
}
static gint
remote_send_string (CamelRemoteStore *store, CamelException *ex, char *fmt, va_list ap)
{
gchar *cmdbuf;
/* Check for connectedness. Failed (or cancelled) operations will
* close the connection. */
if (store->ostream == NULL) {
d(g_message ("remote: (send) disconnected, reconnecting."));
if (!camel_service_connect (CAMEL_SERVICE (store), ex))
return -1;
}
/* create the command */
cmdbuf = g_strdup_vprintf (fmt, ap);
#if d(!)0
if (camel_verbose_debug) {
if (strncmp (cmdbuf, "PASS ", 5) == 0)
fprintf (stderr, "sending : PASS xxxx\n");
else if (strstr (cmdbuf, "LOGIN \""))
fprintf (stderr, "sending : ---- LOGIN \"xxxx\" \"xxxx\"\n");
else
fprintf (stderr, "sending : %s", cmdbuf);
}
#endif
if (camel_stream_printf (store->ostream, "%s", cmdbuf) == -1) {
g_free (cmdbuf);
camel_exception_set (ex, CAMEL_EXCEPTION_SERVICE_UNAVAILABLE,
g_strerror (errno));
camel_service_disconnect (CAMEL_SERVICE (store), FALSE, NULL);
return -1;
}
g_free (cmdbuf);
return 0;
}
/**
* camel_remote_store_send_string: Writes a string to the server
* @store: a CamelRemoteStore
* @ex: a CamelException
* @fmt: the printf-style format to use for creating the string to send
* @...: the arguments to the printf string @fmt
* Return value: 0 on success, nonzero on error
*
* Formats the string and sends it to the server.
**/
gint
camel_remote_store_send_string (CamelRemoteStore *store, CamelException *ex,
char *fmt, ...)
{
va_list ap;
gint ret;
g_return_val_if_fail (CAMEL_IS_REMOTE_STORE (store), -1);
g_return_val_if_fail (fmt, -1);
va_start (ap, fmt);
ret = CRSC (store)->send_string (store, ex, fmt, ap);
va_end (ap);
return ret;
}
static gint
remote_send_stream (CamelRemoteStore *store, CamelStream *stream, CamelException *ex)
{
/* Check for connectedness. Failed (or cancelled) operations will
* close the connection. */
if (store->ostream == NULL) {
d(g_message ("remote: (sendstream) disconnected, reconnecting."));
if (!camel_service_connect (CAMEL_SERVICE (store), ex))
return -1;
}
d(fprintf (stderr, "(sending stream)\n"));
if (camel_stream_write_to_stream (stream, store->ostream) < 0) {
camel_exception_set (ex, CAMEL_EXCEPTION_SERVICE_UNAVAILABLE,
g_strerror (errno));
camel_service_disconnect (CAMEL_SERVICE (store), FALSE, NULL);
return -1;
}
return 0;
}
/**
* camel_remote_store_send_stream: Writes a CamelStream to the server
* @store: a CamelRemoteStore
* @stream: the stream to write
* @ex: a CamelException
* Return value: 0 on success, nonzero on error
*
* Sends the stream to the server.
**/
gint
camel_remote_store_send_stream (CamelRemoteStore *store, CamelStream *stream, CamelException *ex)
{
g_return_val_if_fail (CAMEL_IS_REMOTE_STORE (store), -1);
g_return_val_if_fail (CAMEL_IS_STREAM (stream), -1);
return CRSC (store)->send_stream (store, stream, ex);
}
static int
remote_recv_line (CamelRemoteStore *store, char **dest, CamelException *ex)
{
CamelStreamBuffer *stream = CAMEL_STREAM_BUFFER (store->istream);
GByteArray *bytes;
gchar buf[1024], *ret;
gint nread;
*dest = NULL;
/* Check for connectedness. Failed (or cancelled) operations will
* close the connection. We can't expect a read to have any
* meaning if we reconnect, so always set an exception.
*/
if (store->istream == NULL) {
g_message ("remote: (recv) disconnected, reconnecting.");
camel_service_connect (CAMEL_SERVICE (store), ex);
if (!camel_exception_is_set (ex))
camel_exception_set (ex, CAMEL_EXCEPTION_SERVICE_NOT_CONNECTED,
g_strerror (errno));
return -1;
}
bytes = g_byte_array_new ();
do {
nread = camel_stream_buffer_gets (stream, buf, sizeof (buf));
if (nread > 0)
g_byte_array_append (bytes, buf, nread);
} while (nread == sizeof (buf) - 1);
g_byte_array_append (bytes, "", 1);
ret = bytes->data;
nread = bytes->len - 1;
g_byte_array_free (bytes, FALSE);
if (nread <= 0) {
g_free (ret);
ret = NULL;
camel_exception_set (ex, CAMEL_EXCEPTION_SERVICE_UNAVAILABLE,
nread ? g_strerror (errno) :
_("Server disconnected."));
camel_service_disconnect (CAMEL_SERVICE (store), FALSE, NULL);
return -1;
}
/* strip off the CRLF sequence */
while (nread > 0 && ret[nread] != '\r')
ret[nread--] = '\0';
ret[nread] = '\0';
*dest = ret;
#if d(!)0
if (camel_verbose_debug)
fprintf (stderr, "received: %s\n", *dest);
#endif
return nread;
}
/**
* camel_remote_store_recv_line: Reads a line from the server
* @store: a CamelRemoteStore
* @dest: a pointer that will be set to the location of a buffer
* holding the server's response
* @ex: a CamelException
* Return value: 0 on success, -1 on error
*
* Reads a line from the server (terminated by \n or \r\n).
**/
gint
camel_remote_store_recv_line (CamelRemoteStore *store, char **dest,
CamelException *ex)
{
g_return_val_if_fail (CAMEL_IS_REMOTE_STORE (store), -1);
g_return_val_if_fail (dest, -1);
return CRSC (store)->recv_line (store, dest, ex);
}
static void
refresh_folder_info (gpointer key, gpointer value, gpointer data)
{
CamelFolder *folder = CAMEL_FOLDER (value);
camel_folder_refresh_info (folder, (CamelException *) data);
}
/**
* camel_remote_store_refresh_folders: Refresh the folders that I
* contain
* @store: a CamelRemoteStore
* @ex: a CamelException
*
* Refreshes the folders listed in the folders hashtable.
**/
void
camel_remote_store_refresh_folders (CamelRemoteStore *store, CamelException *ex)
{
g_hash_table_foreach (CAMEL_STORE (store)->folders, refresh_folder_info, ex);
}
