mirror of
https://github.com/valitydev/thrift.git
synced 2024-11-07 18:58:51 +00:00
291 lines
9.2 KiB
C
Executable File
291 lines
9.2 KiB
C
Executable File
/*
|
|
* 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 <assert.h>
|
|
#include <netdb.h>
|
|
#include <signal.h>
|
|
#include <sys/wait.h>
|
|
|
|
#include <thrift/c_glib/transport/thrift_transport.h>
|
|
#include <thrift/c_glib/transport/thrift_socket.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' }
|
|
|
|
#include "../src/thrift/c_glib/transport/thrift_buffered_transport.c"
|
|
|
|
static const char TEST_ADDRESS[] = "localhost";
|
|
static const short TEST_PORT = 64444;
|
|
|
|
static void thrift_server (const int port);
|
|
|
|
/* test object creation and destruction */
|
|
static void
|
|
test_create_and_destroy(void)
|
|
{
|
|
ThriftTransport *transport = NULL;
|
|
guint r_buf_size = 0;
|
|
guint w_buf_size = 0;
|
|
|
|
GObject *object = NULL;
|
|
object = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT, NULL);
|
|
assert (object != NULL);
|
|
g_object_get (G_OBJECT (object), "transport", &transport,
|
|
"r_buf_size", &r_buf_size,
|
|
"w_buf_size", &w_buf_size, NULL);
|
|
g_object_unref (object);
|
|
}
|
|
|
|
static void
|
|
test_open_and_close(void)
|
|
{
|
|
ThriftSocket *tsocket = NULL;
|
|
ThriftTransport *transport = NULL;
|
|
GError *err = NULL;
|
|
|
|
/* create a ThriftSocket */
|
|
tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost",
|
|
"port", 51188, NULL);
|
|
|
|
/* create a BufferedTransport wrapper of the Socket */
|
|
transport = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT,
|
|
"transport", THRIFT_TRANSPORT (tsocket), NULL);
|
|
|
|
/* this shouldn't work */
|
|
assert (thrift_buffered_transport_open (transport, NULL) == FALSE);
|
|
assert (thrift_buffered_transport_is_open (transport) == TRUE);
|
|
assert (thrift_buffered_transport_close (transport, NULL) == TRUE);
|
|
g_object_unref (transport);
|
|
g_object_unref (tsocket);
|
|
|
|
/* try and underlying socket failure */
|
|
tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost.broken",
|
|
NULL);
|
|
|
|
/* create a BufferedTransport wrapper of the Socket */
|
|
transport = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT,
|
|
"transport", THRIFT_TRANSPORT (tsocket), NULL);
|
|
|
|
assert (thrift_buffered_transport_open (transport, &err) == FALSE);
|
|
g_object_unref (transport);
|
|
g_object_unref (tsocket);
|
|
g_error_free (err);
|
|
err = NULL;
|
|
}
|
|
|
|
static void
|
|
test_read_and_write(void)
|
|
{
|
|
int status;
|
|
pid_t pid;
|
|
ThriftSocket *tsocket = NULL;
|
|
ThriftTransport *transport = NULL;
|
|
int port = 51199;
|
|
guchar buf[10] = TEST_DATA; /* a buffer */
|
|
|
|
pid = fork ();
|
|
assert ( pid >= 0 );
|
|
|
|
if ( pid == 0 )
|
|
{
|
|
/* child listens */
|
|
thrift_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 = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT,
|
|
"transport", THRIFT_TRANSPORT (tsocket),
|
|
"w_buf_size", 4, NULL);
|
|
|
|
assert (thrift_buffered_transport_open (transport, NULL) == TRUE);
|
|
assert (thrift_buffered_transport_is_open (transport));
|
|
|
|
/* write 10 bytes */
|
|
thrift_buffered_transport_write (transport, buf, 10, NULL);
|
|
|
|
/* write 1 byte at a time */
|
|
thrift_buffered_transport_write (transport, buf, 1, NULL);
|
|
thrift_buffered_transport_write (transport, buf, 1, NULL);
|
|
thrift_buffered_transport_write (transport, buf, 1, NULL);
|
|
|
|
/* overflow the buffer */
|
|
thrift_buffered_transport_write (transport, buf, 2, NULL);
|
|
thrift_buffered_transport_write (transport, buf, 1, NULL);
|
|
thrift_buffered_transport_flush (transport, NULL);
|
|
|
|
/* write 1 byte and flush */
|
|
thrift_buffered_transport_write (transport, buf, 1, NULL);
|
|
thrift_buffered_transport_flush (transport, NULL);
|
|
|
|
/* write and overflow buffer with 2 system calls */
|
|
thrift_buffered_transport_write (transport, buf, 1, NULL);
|
|
thrift_buffered_transport_write (transport, buf, 3, NULL);
|
|
|
|
/* write 10 bytes */
|
|
thrift_buffered_transport_write (transport, buf, 10, NULL);
|
|
|
|
thrift_buffered_transport_write_end (transport, NULL);
|
|
thrift_buffered_transport_flush (transport, NULL);
|
|
thrift_buffered_transport_close (transport, NULL);
|
|
|
|
g_object_unref (transport);
|
|
g_object_unref (tsocket);
|
|
|
|
assert ( wait (&status) == pid );
|
|
assert ( status == 0 );
|
|
}
|
|
}
|
|
|
|
static void
|
|
thrift_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);
|
|
|
|
/* wrap the client in a BufferedTransport */
|
|
client = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT, "transport",
|
|
thrift_server_transport_accept (transport, NULL),
|
|
"r_buf_size", 5, NULL);
|
|
assert (client != NULL);
|
|
|
|
/* read 10 bytes */
|
|
bytes = thrift_buffered_transport_read (client, buf, 10, NULL);
|
|
assert (bytes == 10); /* make sure we've read 10 bytes */
|
|
assert ( memcmp (buf, match, 10) == 0 ); /* make sure what we got matches */
|
|
|
|
/* read 1 byte */
|
|
bytes = thrift_buffered_transport_read (client, buf, 1, NULL);
|
|
|
|
bytes = thrift_buffered_transport_read (client, buf, 6, NULL);
|
|
bytes = thrift_buffered_transport_read (client, buf, 2, NULL);
|
|
bytes = thrift_buffered_transport_read (client, buf, 1, NULL);
|
|
|
|
thrift_buffered_transport_read_end (client, NULL);
|
|
thrift_buffered_transport_close (client, NULL);
|
|
g_object_unref (client);
|
|
g_object_unref (tsocket);
|
|
}
|
|
|
|
static void
|
|
test_write_fail(void)
|
|
{
|
|
int status;
|
|
pid_t pid;
|
|
ThriftSocket *tsocket = NULL;
|
|
ThriftTransport *transport = NULL;
|
|
int port = 51198;
|
|
guchar buf[10] = TEST_DATA; /* a buffer */
|
|
|
|
/* SIGPIPE when send to disconnected socket */
|
|
signal(SIGPIPE, SIG_IGN);
|
|
|
|
pid = fork ();
|
|
assert ( pid >= 0 );
|
|
|
|
if ( pid == 0 )
|
|
{
|
|
/* child listens */
|
|
ThriftServerTransport *transport = NULL;
|
|
ThriftTransport *client = NULL;
|
|
|
|
ThriftServerSocket *tsocket = g_object_new (THRIFT_TYPE_SERVER_SOCKET,
|
|
"port", port, NULL);
|
|
|
|
transport = THRIFT_SERVER_TRANSPORT (tsocket);
|
|
thrift_server_transport_listen (transport, NULL);
|
|
|
|
/* wrap the client in a BufferedTransport */
|
|
client = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT, "transport",
|
|
thrift_server_transport_accept (transport, NULL),
|
|
"r_buf_size", 5, NULL);
|
|
assert (client != NULL);
|
|
|
|
/* just close socket */
|
|
thrift_buffered_transport_close (client, NULL);
|
|
g_object_unref (client);
|
|
g_object_unref (tsocket);
|
|
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 = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT,
|
|
"transport", THRIFT_TRANSPORT (tsocket),
|
|
"w_buf_size", 4, NULL);
|
|
|
|
|
|
assert (thrift_buffered_transport_open (transport, NULL) == TRUE);
|
|
assert (thrift_buffered_transport_is_open (transport));
|
|
|
|
/* recognize disconnection */
|
|
sleep(1);
|
|
assert (thrift_buffered_transport_write (transport, buf, 10, NULL) == TRUE);
|
|
assert (thrift_buffered_transport_write (transport, buf, 10, NULL) == FALSE);
|
|
|
|
/* write and overflow buffer */
|
|
assert (thrift_buffered_transport_write (transport, buf, 10, NULL) == FALSE);
|
|
|
|
/* write 1 and flush */
|
|
assert (thrift_buffered_transport_write (transport, buf, 1, NULL) == TRUE);
|
|
assert (thrift_buffered_transport_flush (transport, NULL) == FALSE);
|
|
|
|
thrift_buffered_transport_close (transport, NULL);
|
|
|
|
g_object_unref (transport);
|
|
g_object_unref (tsocket);
|
|
|
|
assert ( wait (&status) == pid );
|
|
assert ( status == 0 );
|
|
}
|
|
}
|
|
|
|
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 ("/testbufferedtransport/CreateAndDestroy", test_create_and_destroy);
|
|
g_test_add_func ("/testbufferedtransport/OpenAndClose", test_open_and_close);
|
|
g_test_add_func ("/testbufferedtransport/ReadAndWrite", test_read_and_write);
|
|
g_test_add_func ("/testbufferedtransport/WriteFail", test_write_fail);
|
|
|
|
return g_test_run ();
|
|
}
|
|
|