diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 18:45:59 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 18:45:59 +0000 |
commit | 19fcec84d8d7d21e796c7624e521b60d28ee21ed (patch) | |
tree | 42d26aa27d1e3f7c0b8bd3fd14e7d7082f5008dc /src/jaegertracing/thrift/lib/c_glib/test/testcompactprotocol.c | |
parent | Initial commit. (diff) | |
download | ceph-6d07fdb6bb33b1af39833b850bb6cf8af79fe293.tar.xz ceph-6d07fdb6bb33b1af39833b850bb6cf8af79fe293.zip |
Adding upstream version 16.2.11+ds.upstream/16.2.11+dsupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rwxr-xr-x | src/jaegertracing/thrift/lib/c_glib/test/testcompactprotocol.c | 1293 |
1 files changed, 1293 insertions, 0 deletions
diff --git a/src/jaegertracing/thrift/lib/c_glib/test/testcompactprotocol.c b/src/jaegertracing/thrift/lib/c_glib/test/testcompactprotocol.c new file mode 100755 index 000000000..e5ad49120 --- /dev/null +++ b/src/jaegertracing/thrift/lib/c_glib/test/testcompactprotocol.c @@ -0,0 +1,1293 @@ +/* + * 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. + */ + +/* Disable string-function optimizations when glibc is used, as these produce + compiler warnings about string length when a string function is used inside + a call to g_assert () */ +#if !defined(__APPLE__) && !defined(__FreeBSD__) && \ + !defined(__OpenBSD__) && !defined(__NetBSD__) +#include <features.h> +#endif + +#ifdef __GLIBC__ +#define __NO_STRING_INLINES 1 +#endif + +#include <unistd.h> +#include <stdlib.h> +#include <stdio.h> +#include <netdb.h> +#include <string.h> +#include <sys/wait.h> + +#include <thrift/c_glib/protocol/thrift_protocol.h> +#include <thrift/c_glib/transport/thrift_socket.h> +#include <thrift/c_glib/transport/thrift_server_socket.h> +#include <thrift/c_glib/transport/thrift_framed_transport.h> + +#define TEST_BOOL TRUE +#define TEST_BYTE 123 +#define TEST_I16 12345 +#define TEST_I32 1234567890 +#define TEST_I64 123456789012345 +#define TEST_NI16 (-12345) +#define TEST_NI32 (-1234567890) +#define TEST_NI64 (-123456789012345) +#define TEST_DOUBLE 1234567890.123 +#define TEST_STRING "this is a test string 1234567890!@#$%^&*()" +#define TEST_PORT 51199 + +static int transport_read_count = 0; +static int transport_read_error = 0; +static int transport_read_error_at = -1; +gint32 +my_thrift_transport_read_all (ThriftTransport *transport, gpointer buf, + guint32 len, GError **error) +{ + if (transport_read_count != transport_read_error_at + && transport_read_error == 0) + { + transport_read_count++; + return thrift_transport_read_all (transport, buf, len, error); + } + return -1; +} + +static int transport_write_count = 0; +static int transport_write_error = 0; +static int transport_write_error_at = -1; +gboolean +my_thrift_transport_write (ThriftTransport *transport, const gpointer buf, + const guint32 len, GError **error) +{ + if (transport_write_count != transport_write_error_at + && transport_write_error == 0) + { + transport_write_count++; + return thrift_transport_write (transport, buf, len, error); + } + return FALSE; +} + +#define thrift_transport_read_all my_thrift_transport_read_all +#define thrift_transport_write my_thrift_transport_write +#include "../src/thrift/c_glib/protocol/thrift_compact_protocol.c" +#undef thrift_transport_read_all +#undef thrift_transport_write + +static void thrift_server_primitives (const int port); +static void thrift_server_complex_types (const int port); +static void thrift_server_many_frames (const int port); + +static void +test_create_and_destroy (void) +{ + GObject *object = NULL; + + /* create an object and then destroy it */ + object = g_object_new (THRIFT_TYPE_COMPACT_PROTOCOL, NULL); + g_assert (object != NULL); + g_object_unref (object); +} + +static void +test_initialize (void) +{ + ThriftSocket *tsocket = NULL; + ThriftCompactProtocol *protocol = NULL; + ThriftSocket *temp = NULL; + + /* create a ThriftTransport */ + tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost", + "port", 51188, NULL); + g_assert (tsocket != NULL); + /* create a ThriftCompactProtocol using the Transport */ + protocol = g_object_new (THRIFT_TYPE_COMPACT_PROTOCOL, "transport", + tsocket, NULL); + g_assert (protocol != NULL); + /* fetch the properties */ + g_object_get (G_OBJECT (protocol), "transport", &temp, NULL); + g_object_unref (temp); + + /* clean up memory */ + g_object_unref (protocol); + g_object_unref (tsocket); +} + +static void +test_read_and_write_primitives (void) +{ + int status; + pid_t pid; + ThriftSocket *tsocket = NULL; + ThriftTransport *transport = NULL; + ThriftCompactProtocol *tc = NULL; + ThriftProtocol *protocol = NULL; + gpointer binary = (gpointer *) TEST_STRING; + guint32 len = strlen (TEST_STRING); + int port = TEST_PORT; + + /* fork a server from the client */ + pid = fork (); + g_assert (pid >= 0); + + if (pid == 0) + { + /* child listens */ + thrift_server_primitives (port); + exit (0); + } else { + /* parent. wait a bit for the socket to be created. */ + sleep (1); + + /* create a ThriftSocket */ + tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost", + "port", port, NULL); + transport = THRIFT_TRANSPORT (tsocket); + thrift_transport_open (transport, NULL); + g_assert (thrift_transport_is_open (transport)); + + /* create a ThriftCompactTransport */ + tc = g_object_new (THRIFT_TYPE_COMPACT_PROTOCOL, "transport", + tsocket, NULL); + protocol = THRIFT_PROTOCOL (tc); + g_assert (protocol != NULL); + + /* write a bunch of primitives */ + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, NULL) > 0); + g_assert (thrift_compact_protocol_write_byte (protocol, TEST_BYTE, NULL) > 0); + g_assert (thrift_compact_protocol_write_i16 (protocol, TEST_I16, NULL) > 0); + g_assert (thrift_compact_protocol_write_i32 (protocol, TEST_I32, NULL) > 0); + g_assert (thrift_compact_protocol_write_i64 (protocol, TEST_I64, NULL) > 0); + g_assert (thrift_compact_protocol_write_i16 (protocol, TEST_NI16, NULL) > 0); + g_assert (thrift_compact_protocol_write_i32 (protocol, TEST_NI32, NULL) > 0); + g_assert (thrift_compact_protocol_write_i64 (protocol, TEST_NI64, NULL) > 0); + g_assert (thrift_compact_protocol_write_i16 (protocol, 2, NULL) > 0); + g_assert (thrift_compact_protocol_write_i32 (protocol, 2, NULL) > 0); + g_assert (thrift_compact_protocol_write_i64 (protocol, 2, NULL) > 0); + g_assert (thrift_compact_protocol_write_i16 (protocol, -2, NULL) > 0); + g_assert (thrift_compact_protocol_write_i32 (protocol, -2, NULL) > 0); + g_assert (thrift_compact_protocol_write_i64 (protocol, -2, NULL) > 0); + g_assert (thrift_compact_protocol_write_double (protocol, + TEST_DOUBLE, NULL) > 0); + g_assert (thrift_compact_protocol_write_string (protocol, + TEST_STRING, NULL) > 0); + g_assert (thrift_compact_protocol_write_string (protocol, "", NULL) > 0); + g_assert (thrift_compact_protocol_write_binary (protocol, binary, + len, NULL) > 0); + g_assert (thrift_compact_protocol_write_binary (protocol, NULL, 0, NULL) > 0); + g_assert (thrift_compact_protocol_write_binary (protocol, binary, + len, NULL) > 0); + + /* test write errors */ + transport_write_error = 1; + g_assert (thrift_compact_protocol_write_byte (protocol, TEST_BYTE, + NULL) == -1); + g_assert (thrift_compact_protocol_write_i16 (protocol, TEST_I16, NULL) == -1); + g_assert (thrift_compact_protocol_write_i32 (protocol, TEST_I32, NULL) == -1); + g_assert (thrift_compact_protocol_write_i64 (protocol, TEST_I64, NULL) == -1); + g_assert (thrift_compact_protocol_write_i16 (protocol, TEST_NI16, + NULL) == -1); + g_assert (thrift_compact_protocol_write_i32 (protocol, TEST_NI32, + NULL) == -1); + g_assert (thrift_compact_protocol_write_i64 (protocol, TEST_NI64, + NULL) == -1); + g_assert (thrift_compact_protocol_write_double (protocol, TEST_DOUBLE, + NULL) == -1); + g_assert (thrift_compact_protocol_write_binary (protocol, binary, len, + NULL) == -1); + transport_write_error = 0; + + /* test binary partial failure */ + transport_write_count = 0; + transport_write_error_at = 1; + g_assert (thrift_compact_protocol_write_binary (protocol, binary, + len, NULL) == -1); + transport_write_error_at = -1; + + /* clean up */ + thrift_transport_close (transport, NULL); + g_object_unref (tsocket); + g_object_unref (protocol); + g_assert (wait (&status) == pid); + g_assert (status == 0); + } +} + +static void +test_read_and_write_complex_types (void) +{ + int status; + pid_t pid; + ThriftSocket *tsocket = NULL; + ThriftTransport *transport = NULL; + ThriftCompactProtocol *tc = NULL; + ThriftProtocol *protocol = NULL; + int port = TEST_PORT; + + /* fork a server from the client */ + pid = fork (); + g_assert (pid >= 0); + + if (pid == 0) + { + /* child listens */ + thrift_server_complex_types (port); + exit (0); + } else { + /* parent. wait a bit for the socket to be created. */ + sleep (1); + + /* create a ThriftSocket */ + tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost", + "port", port, NULL); + transport = THRIFT_TRANSPORT (tsocket); + thrift_transport_open (transport, NULL); + g_assert (thrift_transport_is_open (transport)); + + /* create a ThriftCompactTransport */ + tc = g_object_new (THRIFT_TYPE_COMPACT_PROTOCOL, "transport", + tsocket, NULL); + protocol = THRIFT_PROTOCOL (tc); + g_assert (protocol != NULL); + + /* test structures */ + g_assert (thrift_compact_protocol_write_struct_begin (protocol, + NULL, NULL) == 0); + g_assert (thrift_compact_protocol_write_struct_end (protocol, NULL) == 0); + + /* test field state w.r.t. deltas */ + + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, 1, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 16, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 17, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 15, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 30, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 46, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 47, NULL) == 1); + + /* test fields */ + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 1, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + + /* test field state w.r.t. structs */ + + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 1, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 16, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + + g_assert (thrift_compact_protocol_write_struct_begin (protocol, + NULL, NULL) == 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 17, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + + g_assert (thrift_compact_protocol_write_struct_begin (protocol, + NULL, NULL) == 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 18, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 19, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_struct_end (protocol, NULL) == 0); + + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 18, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 25, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_struct_end (protocol, NULL) == 0); + + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 17, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + + /* test field state w.r.t. bools */ + + /* deltas */ + /* non-bool field -> bool field -> non-bool field */ + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 18, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", T_BOOL, + 19, NULL) == 0); + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, + NULL) == 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 20, NULL) == 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + /* bool -> bool field -> bool */ + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, NULL) > 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", T_BOOL, + 21, NULL) == 0); + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, + NULL) == 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, NULL) > 0); + + /* no deltas */ + /* non-bool field -> bool field -> non-bool field */ + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 1, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", T_BOOL, + 1, NULL) == 0); + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 1, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + /* bool -> bool field -> bool */ + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, NULL) > 0); + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", T_BOOL, + 1, NULL) == 0); + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, NULL) > 1); + g_assert (thrift_compact_protocol_write_field_end (protocol, NULL) == 0); + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, NULL) > 0); + + /* test write error */ + transport_write_error = 1; + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 1, NULL) == -1); + transport_write_error = 0; + + /* test 2nd write error */ + transport_write_count = 0; + transport_write_error_at = 1; + g_assert (thrift_compact_protocol_write_field_begin (protocol, "test", + T_DOUBLE, + 1, NULL) == -1); + transport_write_error_at = -1; + + /* test 2nd read failure on a field */ + thrift_compact_protocol_write_byte (protocol, T_DOUBLE, NULL); + + /* test write_field_stop */ + g_assert (thrift_compact_protocol_write_field_stop (protocol, NULL) > 0); + + /* write a map */ + g_assert (thrift_compact_protocol_write_map_begin (protocol, T_DOUBLE, + T_DOUBLE, + 1, NULL) > 0); + g_assert (thrift_compact_protocol_write_map_end (protocol, NULL) == 0); + + /* test 1st read failure on map---nothing to do on our side */ + + /* test 2nd read failure on a map */ + thrift_compact_protocol_write_byte (protocol, T_DOUBLE, NULL); + + /* test 1st write failure on a map */ + transport_write_error = 1; + g_assert (thrift_compact_protocol_write_map_begin (protocol, T_DOUBLE, + T_DOUBLE, + 1, NULL) == -1); + transport_write_error = 0; + + /* test 2nd write failure on a map */ + transport_write_count = 0; + transport_write_error_at = 1; + g_assert (thrift_compact_protocol_write_map_begin (protocol, T_DOUBLE, + T_DOUBLE, + 1, NULL) == -1); + transport_write_error_at = -1; + + /* test negative map size */ + thrift_compact_protocol_write_varint32 (tc, -10, NULL); + thrift_compact_protocol_write_byte (protocol, T_DOUBLE, NULL); + + /* test list operations */ + g_assert (thrift_compact_protocol_write_list_begin (protocol, T_DOUBLE, + 15, NULL) > 0); + g_assert (thrift_compact_protocol_write_list_end (protocol, NULL) == 0); + + /* test 1st read failure on a small list---nothing to do on our end */ + + /* test 1st read failure on a big list---nothing to do on our end */ + + /* test 2nd read failure on a big list */ + thrift_compact_protocol_write_byte (protocol, (gint8) 0xf0, NULL); + + /* test negative list size */ + thrift_compact_protocol_write_byte (protocol, (gint8) 0xf0, NULL); + thrift_compact_protocol_write_varint32 (tc, -10, NULL); + + /* test first write error on a small list */ + transport_write_error = 1; + g_assert (thrift_compact_protocol_write_list_begin (protocol, T_DOUBLE, + 14, NULL) == -1); + transport_write_error = 0; + + /* test first write error on a big list */ + transport_write_error = 1; + g_assert (thrift_compact_protocol_write_list_begin (protocol, T_DOUBLE, + 15, NULL) == -1); + transport_write_error = 0; + + /* test 2nd write error on a big list */ + transport_write_count = 0; + transport_write_error_at = 1; + g_assert (thrift_compact_protocol_write_list_begin (protocol, T_DOUBLE, + 15, NULL) == -1); + transport_write_error_at = -1; + + /* test set operation s*/ + g_assert (thrift_compact_protocol_write_set_begin (protocol, T_DOUBLE, + 1, NULL) > 0); + g_assert (thrift_compact_protocol_write_set_end (protocol, NULL) == 0); + + /* invalid protocol */ + g_assert (thrift_compact_protocol_write_byte (protocol, 0, NULL) > 0); + + /* invalid version */ + g_assert (thrift_compact_protocol_write_byte (protocol, (gint8) 0x82u, + NULL) > 0); + g_assert (thrift_compact_protocol_write_byte (protocol, 0, NULL) > 0); + + /* send a valid message */ + g_assert (thrift_compact_protocol_write_byte (protocol, (gint8) 0x82u, + NULL) > 0); + g_assert (thrift_compact_protocol_write_byte (protocol, 0x01u, NULL) > 0); + thrift_compact_protocol_write_varint32 (tc, 1, NULL); + thrift_compact_protocol_write_string (protocol, "test", NULL); + + /* broken 2nd read */ + thrift_compact_protocol_write_byte (protocol, (gint8) 0x82u, NULL); + + /* send a broken 3rd read */ + thrift_compact_protocol_write_byte (protocol, (gint8) 0x82u, NULL); + thrift_compact_protocol_write_byte (protocol, 0x01u, NULL); + + /* send a broken 4th read */ + thrift_compact_protocol_write_byte (protocol, (gint8) 0x82u, NULL); + thrift_compact_protocol_write_byte (protocol, 0x01u, NULL); + thrift_compact_protocol_write_varint32 (tc, 1, NULL); + + /* send a valid message */ + g_assert (thrift_compact_protocol_write_message_begin (protocol, "test", + T_CALL, 1, NULL) > 0); + + g_assert (thrift_compact_protocol_write_message_end (protocol, NULL) == 0); + + /* send broken writes */ + transport_write_error = 1; + g_assert (thrift_compact_protocol_write_message_begin (protocol, "test", + T_CALL, 1, NULL) == -1); + transport_write_error = 0; + + transport_write_count = 0; + transport_write_error_at = 1; + g_assert (thrift_compact_protocol_write_message_begin (protocol, "test", + T_CALL, 1, NULL) == -1); + transport_write_error_at = -1; + + transport_write_count = 0; + transport_write_error_at = 2; + g_assert (thrift_compact_protocol_write_message_begin (protocol, "test", + T_CALL, 1, NULL) == -1); + transport_write_error_at = -1; + + transport_write_count = 0; + transport_write_error_at = 3; + g_assert (thrift_compact_protocol_write_message_begin (protocol, "test", + T_CALL, 1, NULL) == -1); + transport_write_error_at = -1; + + /* clean up */ + thrift_transport_close (transport, NULL); + g_object_unref (tsocket); + g_object_unref (protocol); + g_assert (wait (&status) == pid); + g_assert (status == 0); + } +} + +static void +test_read_and_write_many_frames (void) +{ + int status; + pid_t pid; + ThriftSocket *tsocket = NULL; + ThriftTransport *transport = NULL; + ThriftFramedTransport *ft = NULL; + ThriftCompactProtocol *tc = NULL; + ThriftProtocol *protocol = NULL; + gpointer binary = (gpointer *) TEST_STRING; + const guint32 len = strlen (TEST_STRING); + int port = TEST_PORT; + + /* fork a server from the client */ + pid = fork (); + g_assert (pid >= 0); + + if (pid == 0) + { + /* child listens */ + thrift_server_many_frames (port); + exit (0); + } else { + /* parent. wait a bit for the socket to be created. */ + sleep (1); + + /* create a ThriftSocket */ + tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost", + "port", port, NULL); + g_assert (tsocket != NULL); + transport = THRIFT_TRANSPORT (tsocket); + + /* wrap in a framed transport */ + ft = g_object_new (THRIFT_TYPE_FRAMED_TRANSPORT, "transport", transport, + "w_buf_size", 1, NULL); + g_assert (ft != NULL); + transport = THRIFT_TRANSPORT (ft); + + thrift_transport_open (transport, NULL); + g_assert (thrift_transport_is_open (transport)); + + /* create a compact protocol */ + tc = g_object_new (THRIFT_TYPE_COMPACT_PROTOCOL, "transport", + transport, NULL); + protocol = THRIFT_PROTOCOL (tc); + g_assert (protocol != NULL); + + /* write a bunch of primitives */ + g_assert (thrift_compact_protocol_write_bool (protocol, TEST_BOOL, + NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_byte (protocol, TEST_BYTE, + NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i16 (protocol, TEST_I16, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i32 (protocol, TEST_I32, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i64 (protocol, TEST_I64, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i16 (protocol, TEST_NI16, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i32 (protocol, TEST_NI32, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i64 (protocol, TEST_NI64, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i16 (protocol, 2, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i32 (protocol, 2, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i64 (protocol, 2, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i16 (protocol, -2, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i32 (protocol, -2, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_i64 (protocol, -2, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_double (protocol, + TEST_DOUBLE, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_string (protocol, + TEST_STRING, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_string (protocol, "", NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_binary (protocol, binary, + len, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_binary (protocol, NULL, + 0, NULL) > 0); + thrift_transport_flush (transport, NULL); + g_assert (thrift_compact_protocol_write_binary (protocol, binary, + len, NULL) > 0); + thrift_transport_flush (transport, NULL); + + /* clean up */ + thrift_transport_write_end (transport, NULL); + thrift_transport_close (transport, NULL); + g_object_unref (ft); + g_object_unref (tsocket); + g_object_unref (tc); + g_assert (wait (&status) == pid); + g_assert (status == 0); + } +} + + +static void +thrift_server_primitives (const int port) +{ + ThriftServerTransport *transport = NULL; + ThriftTransport *client = NULL; + ThriftCompactProtocol *tc = NULL; + ThriftProtocol *protocol = NULL; + gboolean value_boolean = FALSE; + gint8 value_byte = 0, + zigzag_p16 = 0, zigzag_p32 = 0, zigzag_p64 = 0, + zigzag_n16 = 0, zigzag_n32 = 0, zigzag_n64 = 0; + gint16 value_16 = 0; + gint32 value_32 = 0; + gint64 value_64 = 0; + gint16 value_n16 = 0; + gint32 value_n32 = 0; + gint64 value_n64 = 0; + gdouble value_double = 0; + gchar *string = NULL; + gchar *empty_string = NULL; + gpointer binary = NULL; + guint32 len = 0; + void *comparator = (void *) TEST_STRING; + + 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); + + tc = g_object_new (THRIFT_TYPE_COMPACT_PROTOCOL, "transport", + client, NULL); + protocol = THRIFT_PROTOCOL (tc); + + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &value_byte, NULL) > 0); + g_assert (thrift_compact_protocol_read_i16 (protocol, &value_16, NULL) > 0); + g_assert (thrift_compact_protocol_read_i32 (protocol, &value_32, NULL) > 0); + g_assert (thrift_compact_protocol_read_i64 (protocol, &value_64, NULL) > 0); + g_assert (thrift_compact_protocol_read_i16 (protocol, &value_n16, NULL) > 0); + g_assert (thrift_compact_protocol_read_i32 (protocol, &value_n32, NULL) > 0); + g_assert (thrift_compact_protocol_read_i64 (protocol, &value_n64, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_p16, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_p32, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_p64, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_n16, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_n32, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_n64, NULL) > 0); + g_assert (thrift_compact_protocol_read_double (protocol, + &value_double, NULL) > 0); + g_assert (thrift_compact_protocol_read_string (protocol, &string, NULL) > 0); + g_assert (thrift_compact_protocol_read_string (protocol, &empty_string, + NULL) > 0); + g_assert (thrift_compact_protocol_read_binary (protocol, &binary, + &len, NULL) > 0); + + g_assert (value_boolean == TEST_BOOL); + g_assert (value_byte == TEST_BYTE); + g_assert (value_16 == TEST_I16); + g_assert (value_32 == TEST_I32); + g_assert (value_64 == TEST_I64); + g_assert (value_n16 == TEST_NI16); + g_assert (value_n32 == TEST_NI32); + g_assert (value_n64 == TEST_NI64); + g_assert (zigzag_p16 == 4); + g_assert (zigzag_p32 == 4); + g_assert (zigzag_p64 == 4); + g_assert (zigzag_n16 == 3); + g_assert (zigzag_n32 == 3); + g_assert (zigzag_n64 == 3); + g_assert (value_double == TEST_DOUBLE); + g_assert (strcmp (TEST_STRING, string) == 0); + g_assert (strcmp ("", empty_string) == 0); + g_assert (memcmp (comparator, binary, len) == 0); + + g_free (string); + g_free (empty_string); + g_free (binary); + + g_assert (thrift_compact_protocol_read_binary (protocol, &binary, + &len, NULL) > 0); + g_assert (binary == NULL); + g_assert (len == 0); + g_free (binary); + + transport_read_count = 0; + transport_read_error_at = 0; + g_assert (thrift_compact_protocol_read_binary (protocol, &binary, + &len, NULL) == -1); + transport_read_error_at = -1; + + transport_read_count = 0; + transport_read_error_at = 1; + g_assert (thrift_compact_protocol_read_binary (protocol, &binary, + &len, NULL) == -1); + transport_read_error_at = -1; + + transport_read_error = 1; + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) == -1); + g_assert (thrift_compact_protocol_read_byte (protocol, + &value_byte, NULL) == -1); + g_assert (thrift_compact_protocol_read_i16 (protocol, + &value_16, NULL) == -1); + g_assert (thrift_compact_protocol_read_i32 (protocol, &value_32, NULL) == -1); + g_assert (thrift_compact_protocol_read_i64 (protocol, &value_64, NULL) == -1); + g_assert (thrift_compact_protocol_read_i16 (protocol, + &value_n16, NULL) == -1); + g_assert (thrift_compact_protocol_read_i32 (protocol, &value_n32, NULL) == -1); + g_assert (thrift_compact_protocol_read_i64 (protocol, &value_n64, NULL) == -1); + g_assert (thrift_compact_protocol_read_double (protocol, + &value_double, NULL) == -1); + transport_read_error = 0; + + /* test partial write failure */ + thrift_protocol_read_i32 (protocol, &value_32, NULL); + + thrift_transport_read_end (client, NULL); + thrift_transport_close (client, NULL); + + g_object_unref (tc); + g_object_unref (client); + g_object_unref (tsocket); +} + +static void +thrift_server_complex_types (const int port) +{ + ThriftServerTransport *transport = NULL; + ThriftTransport *client = NULL; + ThriftCompactProtocol *tc = NULL; + ThriftProtocol *protocol = NULL; + gchar *struct_name = NULL; + gchar *field_name = NULL; + gchar *message_name = NULL; + ThriftType element_type, key_type, value_type, field_type; + ThriftMessageType message_type; + gboolean value_boolean = ! TEST_BOOL; + gint8 value = 0; + gint16 field_id = 0; + guint32 size = 0; + gint32 seqid = 0; + gint8 version_and_type = 0; + gint8 protocol_id = 0; + + 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); + + tc = g_object_new (THRIFT_TYPE_COMPACT_PROTOCOL, "transport", + client, NULL); + protocol = THRIFT_PROTOCOL (tc); + + /* test struct operations */ + + thrift_compact_protocol_read_struct_begin (protocol, &struct_name, NULL); + thrift_compact_protocol_read_struct_end (protocol, NULL); + + /* test field state w.r.t. deltas */ + + field_id = 0; + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 1); + field_id = 0; + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 16); + field_id = 0; + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 17); + field_id = 0; + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) > 1); + g_assert (field_id == 15); + field_id = 0; + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 30); + field_id = 0; + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) > 1); + g_assert (field_id == 46); + field_id = 0; + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 47); + field_id = 0; + + /* test field operations */ + + thrift_compact_protocol_read_field_begin (protocol, &field_name, &field_type, + &field_id, NULL); + thrift_compact_protocol_read_field_end (protocol, NULL); + + /* test field state w.r.t. structs */ + + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) > 1); + g_assert (field_id == 1); + field_id = 0; + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 16); + field_id = 0; + thrift_compact_protocol_read_field_end (protocol, NULL); + + g_assert (thrift_compact_protocol_read_struct_begin (protocol, + &struct_name, NULL) == 0); + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) > 1); + g_assert (field_id == 17); + field_id = 0; + thrift_compact_protocol_read_field_end (protocol, NULL); + + g_assert (thrift_compact_protocol_read_struct_begin (protocol, + &struct_name, NULL) == 0); + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) > 1); + g_assert (field_id == 18); + field_id = 0; + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 19); + field_id = 0; + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_struct_end (protocol, NULL) == 0); + + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 18); + field_id = 0; + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 25); + field_id = 0; + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_struct_end (protocol, NULL) == 0); + + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_id == 17); + field_id = 0; + thrift_compact_protocol_read_field_end (protocol, NULL); + + /* test field state w.r.t. bools */ + + /* deltas */ + /* non-bool field -> bool field -> non-bool field */ + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_type == T_BOOL); + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) == 0); + g_assert (value_boolean == TEST_BOOL); + value_boolean = ! TEST_BOOL; + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + thrift_compact_protocol_read_field_end (protocol, NULL); + /* bool -> bool field -> bool */ + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) > 0); + g_assert (value_boolean == TEST_BOOL); + value_boolean = ! TEST_BOOL; + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) == 1); + g_assert (field_type == T_BOOL); + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) == 0); + g_assert (value_boolean == TEST_BOOL); + value_boolean = ! TEST_BOOL; + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) > 0); + g_assert (value_boolean == TEST_BOOL); + value_boolean = ! TEST_BOOL; + + /* no deltas */ + /* non-bool field -> bool field -> non-bool field */ + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) > 1); + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) > 1); + g_assert (field_type == T_BOOL); + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) == 0); + g_assert (value_boolean == TEST_BOOL); + value_boolean = ! TEST_BOOL; + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) > 1); + thrift_compact_protocol_read_field_end (protocol, NULL); + /* bool -> bool field -> bool */ + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) > 0); + g_assert (value_boolean == TEST_BOOL); + value_boolean = ! TEST_BOOL; + g_assert (thrift_compact_protocol_read_field_begin (protocol, &field_name, + &field_type, + &field_id, NULL) > 1); + g_assert (field_type == T_BOOL); + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) == 0); + g_assert (value_boolean == TEST_BOOL); + value_boolean = ! TEST_BOOL; + thrift_compact_protocol_read_field_end (protocol, NULL); + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) > 0); + g_assert (value_boolean == TEST_BOOL); + value_boolean = ! TEST_BOOL; + + /* test first read error on a field */ + transport_read_error = 1; + g_assert (thrift_compact_protocol_read_field_begin (protocol, + &field_name, &field_type, + &field_id, NULL) == -1); + transport_read_error = 0; + + /* test 2nd write failure */ + thrift_compact_protocol_read_byte (protocol, &value, NULL); + + /* test 2nd read failure on a field */ + transport_read_count = 0; + transport_read_error_at = 1; + g_assert (thrift_compact_protocol_read_field_begin (protocol, + &field_name, &field_type, + &field_id, NULL) == -1); + transport_read_error_at = -1; + + /* test field stop */ + thrift_compact_protocol_read_field_begin (protocol, &field_name, &field_type, + &field_id, NULL); + + /* test map operations */ + + thrift_compact_protocol_read_map_begin (protocol, &key_type, &value_type, + &size, NULL); + thrift_compact_protocol_read_map_end (protocol, NULL); + + /* test 1st read failure on a map */ + transport_read_count = 0; + transport_read_error_at = 0; + g_assert (thrift_compact_protocol_read_map_begin (protocol, + &key_type, &value_type, + &size, NULL) == -1); + transport_read_error_at = -1; + + /* test 2nd read failure on a map */ + transport_read_count = 0; + transport_read_error_at = 1; + g_assert (thrift_compact_protocol_read_map_begin (protocol, + &key_type, &value_type, + &size, NULL) == -1); + transport_read_error_at = -1; + + /* test 1st write failure on map---nothing to do on our side */ + + /* test 2nd write failure */ + thrift_compact_protocol_read_byte (protocol, &value, NULL); + + /* test negative map size */ + g_assert (thrift_compact_protocol_read_map_begin (protocol, + &key_type, &value_type, + &size, NULL) == -1); + + /* test list operations */ + thrift_compact_protocol_read_list_begin (protocol, &element_type, &size, + NULL); + thrift_compact_protocol_read_list_end (protocol, NULL); + + /* test small list 1st read failure */ + transport_read_error = 1; + g_assert (thrift_compact_protocol_read_list_begin (protocol, &element_type, + &size, NULL) == -1); + transport_read_error = 0; + + /* test big list 1st read failure */ + transport_read_error = 1; + g_assert (thrift_compact_protocol_read_list_begin (protocol, &element_type, + &size, NULL) == -1); + transport_read_error = 0; + + /* test big list 2nd read failure */ + transport_read_count = 0; + transport_read_error_at = 1; + thrift_compact_protocol_read_list_begin (protocol, &element_type, &size, + NULL); + transport_read_error_at = -1; + + /* test negative list size failure */ + thrift_compact_protocol_read_list_begin (protocol, &element_type, &size, + NULL); + + /* test small list 1st write failure---nothing to do on our end */ + + /* test big list 1st write failure---nothing to do on our end */ + + /* test big list 2nd write failure */ + thrift_compact_protocol_read_byte (protocol, &value, NULL); + + /* test set operations */ + thrift_compact_protocol_read_set_begin (protocol, &element_type, &size, NULL); + thrift_compact_protocol_read_set_end (protocol, NULL); + + /* broken read */ + transport_read_error = 1; + g_assert (thrift_compact_protocol_read_message_begin (protocol, &message_name, + &message_type, &seqid, + NULL) == -1); + transport_read_error = 0; + + /* invalid protocol */ + g_assert (thrift_compact_protocol_read_message_begin (protocol, &message_name, + &message_type, &seqid, + NULL) == -1); + + /* invalid version */ + g_assert (thrift_compact_protocol_read_message_begin (protocol, &message_name, + &message_type, &seqid, + NULL) == -1); + + /* read a valid message */ + g_assert (thrift_compact_protocol_read_message_begin (protocol, &message_name, + &message_type, &seqid, + NULL) > 0); + g_free (message_name); + + /* broken 2nd read on a message */ + transport_read_count = 0; + transport_read_error_at = 1; + g_assert (thrift_compact_protocol_read_message_begin (protocol, &message_name, + &message_type, &seqid, + NULL) == -1); + transport_read_error_at = -1; + + /* broken 3rd read on a message */ + transport_read_count = 0; + transport_read_error_at = 2; + g_assert (thrift_compact_protocol_read_message_begin (protocol, &message_name, + &message_type, &seqid, + NULL) == -1); + transport_read_error_at = -1; + + /* broken 4th read on a message */ + transport_read_count = 0; + transport_read_error_at = 3; + g_assert (thrift_compact_protocol_read_message_begin (protocol, &message_name, + &message_type, &seqid, + NULL) == -1); + transport_read_error_at = -1; + + /* read a valid message */ + g_assert (thrift_compact_protocol_read_message_begin (protocol, &message_name, + &message_type, &seqid, + NULL) > 0); + g_free (message_name); + + g_assert (thrift_compact_protocol_read_message_end (protocol, NULL) == 0); + + /* handle 2nd write failure on a message */ + thrift_compact_protocol_read_byte (protocol, &protocol_id, NULL); + + /* handle 3rd write failure on a message */ + thrift_compact_protocol_read_byte (protocol, &protocol_id, NULL); + thrift_compact_protocol_read_byte (protocol, &version_and_type, NULL); + + /* handle 4th write failure on a message */ + thrift_compact_protocol_read_byte (protocol, &protocol_id, NULL); + thrift_compact_protocol_read_byte (protocol, &version_and_type, NULL); + thrift_compact_protocol_read_varint32 (tc, &seqid, NULL); + + g_object_unref (client); + g_object_unref (tsocket); +} + +static void +thrift_server_many_frames (const int port) +{ + ThriftServerTransport *transport = NULL; + ThriftTransport *client = NULL; + ThriftCompactProtocol *tcp = NULL; + ThriftProtocol *protocol = NULL; + ThriftServerSocket *tsocket = NULL; + gboolean value_boolean = FALSE; + gint8 value_byte = 0, + zigzag_p16 = 0, zigzag_p32 = 0, zigzag_p64 = 0, + zigzag_n16 = 0, zigzag_n32 = 0, zigzag_n64 = 0; + gint16 value_16 = 0; + gint32 value_32 = 0; + gint64 value_64 = 0; + gint16 value_n16 = 0; + gint32 value_n32 = 0; + gint64 value_n64 = 0; + gdouble value_double = 0; + gchar *string = NULL; + gchar *empty_string = NULL; + gpointer binary = NULL; + guint32 len = 0; + void *comparator = (void *) TEST_STRING; + + 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 framed transport */ + client = g_object_new (THRIFT_TYPE_FRAMED_TRANSPORT, "transport", + thrift_server_transport_accept (transport, NULL), + "r_buf_size", 1, NULL); + g_assert (client != NULL); + + tcp = g_object_new (THRIFT_TYPE_COMPACT_PROTOCOL, "transport", + client, NULL); + protocol = THRIFT_PROTOCOL (tcp); + + g_assert (thrift_compact_protocol_read_bool (protocol, + &value_boolean, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &value_byte, NULL) > 0); + g_assert (thrift_compact_protocol_read_i16 (protocol, &value_16, NULL) > 0); + g_assert (thrift_compact_protocol_read_i32 (protocol, &value_32, NULL) > 0); + g_assert (thrift_compact_protocol_read_i64 (protocol, &value_64, NULL) > 0); + g_assert (thrift_compact_protocol_read_i16 (protocol, &value_n16, NULL) > 0); + g_assert (thrift_compact_protocol_read_i32 (protocol, &value_n32, NULL) > 0); + g_assert (thrift_compact_protocol_read_i64 (protocol, &value_n64, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_p16, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_p32, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_p64, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_n16, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_n32, NULL) > 0); + g_assert (thrift_compact_protocol_read_byte (protocol, &zigzag_n64, NULL) > 0); + g_assert (thrift_compact_protocol_read_double (protocol, + &value_double, NULL) > 0); + g_assert (thrift_compact_protocol_read_string (protocol, &string, NULL) > 0); + g_assert (thrift_compact_protocol_read_string (protocol, &empty_string, + NULL) > 0); + g_assert (thrift_compact_protocol_read_binary (protocol, &binary, + &len, NULL) > 0); + + g_assert (value_boolean == TEST_BOOL); + g_assert (value_byte == TEST_BYTE); + g_assert (value_16 == TEST_I16); + g_assert (value_32 == TEST_I32); + g_assert (value_64 == TEST_I64); + g_assert (value_n16 == TEST_NI16); + g_assert (value_n32 == TEST_NI32); + g_assert (value_n64 == TEST_NI64); + g_assert (zigzag_p16 == 4); + g_assert (zigzag_p32 == 4); + g_assert (zigzag_p64 == 4); + g_assert (zigzag_n16 == 3); + g_assert (zigzag_n32 == 3); + g_assert (zigzag_n64 == 3); + g_assert (value_double == TEST_DOUBLE); + g_assert (strcmp (TEST_STRING, string) == 0); + g_assert (strcmp ("", empty_string) == 0); + g_assert (memcmp (comparator, binary, len) == 0); + + g_free (string); + g_free (empty_string); + g_free (binary); + + g_assert (thrift_compact_protocol_read_binary (protocol, &binary, + &len, NULL) > 0); + g_assert (binary == NULL); + g_assert (len == 0); + g_free (binary); + + thrift_transport_read_end (client, NULL); + thrift_transport_close (client, NULL); + + g_object_unref (tcp); + g_object_unref (client); + g_object_unref (tsocket); +} + +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 ("/testcompactprotocol/CreateAndDestroy", + test_create_and_destroy); + g_test_add_func ("/testcompactprotocol/Initialize", test_initialize); + g_test_add_func ("/testcompactprotocol/ReadAndWritePrimitives", + test_read_and_write_primitives); + g_test_add_func ("/testcompactprotocol/ReadAndWriteComplexTypes", + test_read_and_write_complex_types); + g_test_add_func ("/testcompactprotocol/ReadAndWriteManyFrames", + test_read_and_write_many_frames); + + return g_test_run (); +} |