+++ /dev/null
-/*
- * Licensed to the Apache Software Foundation (ASF) under one
- * or more contributor license agreements. See the NOTICE file
- * distributed with this work for additional information
- * regarding copyright ownership. The ASF licenses this file
- * to you under the Apache License, Version 2.0 (the
- * "License"); you may not use this file except in compliance
- * with the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing,
- * software distributed under the License is distributed on an
- * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
- * KIND, either express or implied. See the License for the
- * specific language governing permissions and limitations
- * under the License.
- */
-
-#include <netdb.h>
-#include <sys/wait.h>
-
-#include <thrift/c_glib/transport/thrift_transport.h>
-#include <thrift/c_glib/transport/thrift_buffered_transport.h>
-#include <thrift/c_glib/transport/thrift_server_transport.h>
-#include <thrift/c_glib/transport/thrift_server_socket.h>
-
-#define TEST_DATA { 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j' }
-
-/* substituted functions to test failures of system and library calls */
-static int socket_error = 0;
-int
-my_socket(int domain, int type, int protocol)
-{
- if (socket_error == 0)
- {
- return socket (domain, type, protocol);
- }
- return -1;
-}
-
-static int recv_error = 0;
-ssize_t
-my_recv(int socket, void *buffer, size_t length, int flags)
-{
- if (recv_error == 0)
- {
- return recv (socket, buffer, length, flags);
- }
- return -1;
-}
-
-static int send_error = 0;
-ssize_t
-my_send(int socket, const void *buffer, size_t length, int flags)
-{
- if (send_error == 0)
- {
- return send (socket, buffer, length, flags);
- }
- return -1;
-}
-
-#define socket my_socket
-#define recv my_recv
-#define send my_send
-#include "../src/thrift/c_glib/transport/thrift_socket.c"
-#undef socket
-#undef recv
-#undef send
-
-static void thrift_socket_server (const int port);
-static void thrift_socket_server_open (const int port, int times);
-/* test object creation and destruction */
-static void
-test_create_and_destroy(void)
-{
- gchar *hostname = NULL;
- guint port = 0;
-
- GObject *object = NULL;
- object = g_object_new (THRIFT_TYPE_SOCKET, NULL);
- g_assert (object != NULL);
- g_object_get (G_OBJECT(object), "hostname", &hostname, "port", &port, NULL);
- g_free (hostname);
-
- g_object_unref (object);
-}
-
-static void
-test_open_and_close(void)
-{
- ThriftSocket *tsocket = NULL;
- ThriftTransport *transport = NULL;
- GError *err = NULL;
- int port = 51199;
- pid_t pid;
- int status;
-
- pid = fork ();
- g_assert ( pid >= 0 );
-
- if ( pid == 0 )
- {
- /* child listens */
- thrift_socket_server_open (port, 1);
- exit (0);
- } else {
- /* parent connects, wait a bit for the socket to be created */
- sleep (1);
-
- /* open a connection and close it */
- tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost",
- "port", port, NULL);
- transport = THRIFT_TRANSPORT (tsocket);
- thrift_socket_open (transport, NULL);
- g_assert (thrift_socket_is_open (transport) == TRUE);
- thrift_socket_close (transport, NULL);
- g_assert (thrift_socket_is_open (transport) == FALSE);
-
- /* test close failure */
- tsocket->sd = -1;
- thrift_socket_close (transport, NULL);
- g_object_unref (tsocket);
-
- /* try a hostname lookup failure */
- tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost.broken",
- NULL);
- transport = THRIFT_TRANSPORT (tsocket);
- g_assert (thrift_socket_open (transport, &err) == FALSE);
- g_object_unref (tsocket);
- g_error_free (err);
- err = NULL;
-
- /* try an error call to socket() */
- tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost", NULL);
- transport = THRIFT_TRANSPORT (tsocket);
- socket_error = 1;
- g_assert (thrift_socket_open (transport, &err) == FALSE);
- socket_error = 0;
- g_object_unref (tsocket);
- g_error_free (err);
- g_assert ( wait (&status) == pid );
- g_assert ( status == 0 );
- }
-}
-
-static void
-test_read_and_write(void)
-{
- ThriftSocket *tsocket = NULL;
- ThriftTransport *transport = NULL;
- pid_t pid;
- int port = 51199;
- int status;
- guchar buf[10] = TEST_DATA; /* a buffer */
-
- pid = fork ();
- g_assert ( pid >= 0 );
-
- if ( pid == 0 )
- {
- /* child listens */
- thrift_socket_server (port);
- exit (0);
- } else {
- /* parent connects, wait a bit for the socket to be created */
- sleep (1);
-
- tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost",
- "port", port, NULL);
- transport = THRIFT_TRANSPORT (tsocket);
- g_assert (thrift_socket_open (transport, NULL) == TRUE);
- g_assert (thrift_socket_is_open (transport));
- thrift_socket_write (transport, buf, 10, NULL);
-
- /* write fail */
- send_error = 1;
- thrift_socket_write (transport, buf, 1, NULL);
- send_error = 0;
-
- thrift_socket_write_end (transport, NULL);
- thrift_socket_flush (transport, NULL);
- thrift_socket_close (transport, NULL);
- g_object_unref (tsocket);
-
- g_assert ( wait (&status) == pid );
- g_assert ( status == 0 );
- }
-}
-
-/* test ThriftSocket's peek() implementation */
-static void
-test_peek(void)
-{
- gint status;
- pid_t pid;
- guint port = 51199;
- gchar data = 'A';
- ThriftTransport *client_transport;
- GError *error = NULL;
-
- client_transport = g_object_new (THRIFT_TYPE_SOCKET,
- "hostname", "localhost",
- "port", port,
- NULL);
-
- /* thrift_transport_peek returns FALSE when the socket is closed */
- g_assert (thrift_transport_is_open (client_transport) == FALSE);
- g_assert (thrift_transport_peek (client_transport, &error) == FALSE);
- g_assert (error == NULL);
-
- pid = fork ();
- g_assert (pid >= 0);
-
- if (pid == 0)
- {
- ThriftServerTransport *server_transport = NULL;
-
- g_object_unref (client_transport);
-
- /* child listens */
- server_transport = g_object_new (THRIFT_TYPE_SERVER_SOCKET,
- "port", port,
- NULL);
- g_assert (server_transport != NULL);
-
- thrift_server_transport_listen (server_transport, &error);
- g_assert (error == NULL);
-
- client_transport = g_object_new
- (THRIFT_TYPE_BUFFERED_TRANSPORT,
- "transport", thrift_server_transport_accept (server_transport, &error),
- "r_buf_size", 0,
- "w_buf_size", sizeof data,
- NULL);
- g_assert (error == NULL);
- g_assert (client_transport != NULL);
-
- /* write exactly one character to the client */
- g_assert (thrift_transport_write (client_transport,
- &data,
- sizeof data,
- &error) == TRUE);
-
- thrift_transport_flush (client_transport, &error);
- thrift_transport_write_end (client_transport, &error);
- thrift_transport_close (client_transport, &error);
-
- g_object_unref (client_transport);
- g_object_unref (server_transport);
-
- exit (0);
- }
- else {
- /* parent connects, wait a bit for the socket to be created */
- sleep (1);
-
- /* connect to the child */
- thrift_transport_open (client_transport, &error);
- g_assert (error == NULL);
- g_assert (thrift_transport_is_open (client_transport) == TRUE);
-
- /* thrift_transport_peek returns TRUE when the socket is open and there is
- data available to be read */
- g_assert (thrift_transport_peek (client_transport, &error) == TRUE);
- g_assert (error == NULL);
-
- /* read exactly one character from the server */
- g_assert_cmpint (thrift_transport_read (client_transport,
- &data,
- sizeof data,
- &error), ==, sizeof data);
-
- /* thrift_transport_peek returns FALSE when the socket is open but there is
- no (more) data available to be read */
- g_assert (thrift_transport_is_open (client_transport) == TRUE);
- g_assert (thrift_transport_peek (client_transport, &error) == FALSE);
- g_assert (error == NULL);
-
- thrift_transport_read_end (client_transport, &error);
- thrift_transport_close (client_transport, &error);
-
- g_object_unref (client_transport);
-
- g_assert (wait (&status) == pid);
- g_assert (status == 0);
- }
-}
-
-static void
-thrift_socket_server_open (const int port, int times)
-{
- ThriftServerTransport *transport = NULL;
- ThriftTransport *client = NULL;
- int i;
- ThriftServerSocket *tsocket = g_object_new (THRIFT_TYPE_SERVER_SOCKET,
- "port", port, NULL);
-
- transport = THRIFT_SERVER_TRANSPORT (tsocket);
- thrift_server_transport_listen (transport, NULL);
- for(i=0;i<times;i++){
- client = thrift_server_transport_accept (transport, NULL);
- g_assert (client != NULL);
- thrift_socket_close (client, NULL);
- g_object_unref (client);
- }
- g_object_unref (tsocket);
-}
-
-
-static void
-thrift_socket_server (const int port)
-{
- int bytes = 0;
- ThriftServerTransport *transport = NULL;
- ThriftTransport *client = NULL;
- guchar buf[10]; /* a buffer */
- guchar match[10] = TEST_DATA;
-
- ThriftServerSocket *tsocket = g_object_new (THRIFT_TYPE_SERVER_SOCKET,
- "port", port, NULL);
-
- transport = THRIFT_SERVER_TRANSPORT (tsocket);
- thrift_server_transport_listen (transport, NULL);
- client = thrift_server_transport_accept (transport, NULL);
- g_assert (client != NULL);
-
- /* read 10 bytes */
- bytes = thrift_socket_read (client, buf, 10, NULL);
- g_assert (bytes == 10); /* make sure we've read 10 bytes */
- g_assert ( memcmp(buf, match, 10) == 0 ); /* make sure what we got matches */
-
- /* failed read */
- recv_error = 1;
- thrift_socket_read (client, buf, 1, NULL);
- recv_error = 0;
-
- thrift_socket_read_end (client, NULL);
- thrift_socket_close (client, NULL);
- g_object_unref (tsocket);
- g_object_unref (client);
-}
-
-int
-main(int argc, char *argv[])
-{
-#if (!GLIB_CHECK_VERSION (2, 36, 0))
- g_type_init();
-#endif
-
- g_test_init (&argc, &argv, NULL);
-
- g_test_add_func ("/testtransportsocket/CreateAndDestroy", test_create_and_destroy);
- g_test_add_func ("/testtransportsocket/OpenAndClose", test_open_and_close);
- g_test_add_func ("/testtransportsocket/ReadAndWrite", test_read_and_write);
- g_test_add_func ("/testtransportsocket/Peek", test_peek);
-
- return g_test_run ();
-}
-