| /* |
| * 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 (); |
| } |