NNTPGrab

/* 

    Copyright (C) 2005-2011  Erik van Pienbroek

    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 Place, Suite 330, Boston, MA  02111-1307  USA

*/

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include 

#include 

#include "nntpconnection_utils.h"

#include "nntpconnection_backend_native.h"

#if GLIB_CHECK_VERSION(2,22,0)

#include "nntpconnection_backend_gio.h"

#endif

/* Fake placeholders */

typedef void* NGPlugin;

void

ng_plugin_emit_log_msg(NGPlugin *plugin_data, NGLogLevel log_level, const char *format, ...)

{

    va_list args;

    char msg[1024];

    va_start(args, format);

    memset(msg, 0, sizeof(msg));

    vsnprintf(msg, sizeof(msg) - 1, format, args);

    va_end(args);

    g_test_message("%s", msg);

}

void

nntpgrab_core_emit_traffic_monitor_update(gboolean direct, int bytes_received[10], time_t stamp)

{

    g_test_message("Signal was emitted: traffic_monitor_update (bytes_received[9] = %i, bytes_received[8] = %i, stamp = %i", bytes_received[9], bytes_received[8], (int) stamp);

}

void

nntpgrab_core_emit_connecting(gboolean direct, const char *servername, int conn_id)

{

    g_test_message("Signal was emitted: connecting (servername = %s, conn_id = %i", servername, conn_id);

}

void

nntpgrab_core_emit_connected(gboolean direct, const char *servername, int conn_id, const char *welcome_msg)

{

    g_test_message("Signal was emitted: connected (servername = %s, conn_id = %i, welcome_msg = %s", servername, conn_id, welcome_msg);

}

void

nntpgrab_core_emit_disconnect(gboolean direct, const char *servername, int conn_id, NNTPDisconnectType disconnect_type, const char *errmsg)

{

    g_test_message("Signal was emitted: disconnect (servername = %s, conn_id = %i, disconnect_type = %i, errmsg = %s", servername, conn_id, disconnect_type, errmsg);

}

/* Are the functions below really necessary */

void

file_unref(NNTPFile *file)

{

}

void

collection_unref(NNTPCollection *collection)

{

}

void

download_queue_update_part_status(const char *servername, int connection_id, NNTPCollection *collection, NNTPFile *file, NNTPPart *part, int server_id, gboolean success, gboolean reset_to_zero, gboolean dont_retry_anymore)

{

}

NGSchedularState

download_thread_get_state (void)

{

    return SCHEDULAR_STATE_RUNNING;

}

void

nntpconnection_process_welcome_msg(NntpConnection *conn)

{

}

void

throttle_pause(struct timeval start_time, off_t xferlen, int max_bandwidth)

{

}

/* Test functions */

static void

test_backend_internal(NntpConnection *conn)

{

    char *errmsg = NULL;

    gboolean retval = TRUE;

    char line[256];

    int length;

    gboolean more_data_ready;

    /* Try to connect to the server */

    conn->poll_fd.fd = -1;

    g_test_message("Connecting to usenet server");

    retval = nntpconnection_connect_to_server(conn, &errmsg);

    if (retval != NNTP_CONNECTION_ERROR_NONE) {

        g_assert_cmpint(conn->poll_fd.fd, ==, -1);

        g_assert(errmsg != NULL);

        g_test_message("Errmsg = %s\n", errmsg);

        g_free(errmsg);

        g_assert_cmpint(retval, ==, NNTP_CONNECTION_ERROR_NONE);

        g_assert_not_reached();

    }

    g_assert_cmpint(conn->poll_fd.fd, >, -1);

    /* Wait until there's data */

    g_test_message("Waiting for data to arrive");

    g_assert(nntpconnection_ensure_data_is_ready(conn, G_USEC_PER_SEC * 5));

    /* Read the welcome msg */

    g_test_message("Reading welcome message");

    memset(&line, 0, sizeof(line));

    g_assert(nntpconnection_read_msg(conn, TRUE, sizeof(line) - 1, &line, &length, &more_data_ready));

    /* The welcome msg must start with '200' */

    g_test_message("Verifying welcome message");

    g_assert_cmpint(nntpconnection_utils_get_status_code(line), ==, 200);

    g_assert_cmpint(strlen(line), ==, length);

    /* Send the MODE READER command */

    g_test_message("Sending the command MODE READER to server");

    memset(&line, 0, sizeof(line));

    snprintf(line, sizeof(line) - 1, "MODE READER\r\n");

    g_assert(nntpconnection_send_msg(conn, line, strlen(line)));

    /* Wait again for data to arrive */

    g_test_message("Waiting for data to arrive");

    g_assert(nntpconnection_ensure_data_is_ready(conn, G_USEC_PER_SEC * 5));

    /* Read the response */

    g_test_message("Reading response");

    memset(&line, 0, sizeof(line));

    g_assert(nntpconnection_read_msg(conn, TRUE, sizeof(line) - 1, &line, &length, &more_data_ready));

    /* The response must start with '200' */

    g_test_message("Verifying MODE READER response message");

    g_assert_cmpint(nntpconnection_utils_get_status_code(line), ==, 200);

    g_assert_cmpint(strlen(line), ==, length);

    /* Disconnect from the server */

    g_test_message("Disconnecting from server");

    nntpconnection_disconnect_from_server(conn, DISCONNECT_NORMAL, NULL, __FILE__, __LINE__);

    g_assert_cmpint(conn->poll_fd.fd, ==, -1);

}

static void

test_backend_normal(NntpConnection *conn)

{

    /* Configure a fake server */

    strncpy(conn->server_info.servername, "GigaNews", sizeof(conn->server_info.servername) - 1);

    strncpy(conn->server_info.hostname, "news.giganews.com", sizeof(conn->server_info.hostname) - 1);

    conn->server_info.port = 119;

    conn->server_info.use_ssl = FALSE;

    conn->server_info.enabled = TRUE;

    test_backend_internal(conn);

}

static void

test_backend_ssl(NntpConnection *conn)

{

    /* Configure a fake server */

    strncpy(conn->server_info.servername, "GigaNews", sizeof(conn->server_info.servername) - 1);

    strncpy(conn->server_info.hostname, "news.giganews.com", sizeof(conn->server_info.hostname) - 1);

    conn->server_info.port = 563;

    conn->server_info.use_ssl = TRUE;

    conn->server_info.enabled = TRUE;

    test_backend_internal(conn);

}

static

void test_backend_native(void)

{

    NntpConnection *conn = nntpconnection_backend_native_new();

    test_backend_normal(conn);

    g_object_unref(conn);

}

static

void test_backend_native_ssl(void)

{

    NntpConnection *conn = nntpconnection_backend_native_new();

    g_test_message("Testing for the SSL support flag");

#ifdef HAVE_SSL

    g_assert_cmpint(nntpconnection_get_has_ssl_support(conn), ==, TRUE);

#else

    g_assert_cmpint(nntpconnection_get_has_ssl_support(conn), ==, FALSE);

#endif

    test_backend_ssl(conn);

    g_object_unref(conn);

}

#if GLIB_CHECK_VERSION(2,22,0)

static

void test_backend_gio(void)

{

    NntpConnection *conn = nntpconnection_backend_gio_new();

    test_backend_normal(conn);

    g_object_unref(conn);

}

#endif

#ifdef HAVE_GLIB_NETWORKING

static

void test_backend_gio_ssl(void)

{

    NntpConnection *conn = nntpconnection_backend_gio_new();

    g_test_message("Testing for the SSL support flag");

    g_assert_cmpint(nntpconnection_get_has_ssl_support(conn), ==, TRUE);

    test_backend_ssl(conn);

    g_object_unref(conn);

}

#endif

int main(int argc, char **argv)

{

    g_type_init();

    g_test_init(&argc, &argv, NULL);

    g_test_bug_base("https://www.nntpgrab.nl/issues/");

    g_test_add_func("/nntpconnection/backend/native", test_backend_native);

    g_test_add_func("/nntpconnection/backend/native/ssl", test_backend_native_ssl);

#if GLIB_CHECK_VERSION(2,22,0)

    g_test_add_func("/nntpconnection/backend/gio", test_backend_gio);

#endif

#ifdef HAVE_GLIB_NETWORKING

    g_test_add_func("/nntpconnection/backend/gio/ssl", test_backend_gio_ssl);

#endif

    return g_test_run();

}