Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 1 | /* |
| 2 | * |
| 3 | * Copyright 2015, Google Inc. |
| 4 | * All rights reserved. |
| 5 | * |
| 6 | * Redistribution and use in source and binary forms, with or without |
| 7 | * modification, are permitted provided that the following conditions are |
| 8 | * met: |
| 9 | * |
| 10 | * * Redistributions of source code must retain the above copyright |
| 11 | * notice, this list of conditions and the following disclaimer. |
| 12 | * * Redistributions in binary form must reproduce the above |
| 13 | * copyright notice, this list of conditions and the following disclaimer |
| 14 | * in the documentation and/or other materials provided with the |
| 15 | * distribution. |
| 16 | * * Neither the name of Google Inc. nor the names of its |
| 17 | * contributors may be used to endorse or promote products derived from |
| 18 | * this software without specific prior written permission. |
| 19 | * |
| 20 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 21 | * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 22 | * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 23 | * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 24 | * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 25 | * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 26 | * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 27 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 28 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 29 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 30 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 31 | * |
| 32 | */ |
| 33 | |
| 34 | #include "test/core/bad_client/bad_client.h" |
| 35 | |
| 36 | #include "src/core/channel/channel_stack.h" |
| 37 | #include "src/core/channel/http_server_filter.h" |
| 38 | #include "src/core/iomgr/endpoint_pair.h" |
| 39 | #include "src/core/surface/completion_queue.h" |
| 40 | #include "src/core/surface/server.h" |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 41 | #include "src/core/support/string.h" |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 42 | #include "src/core/transport/chttp2_transport.h" |
| 43 | |
Craig Tiller | e46f00a | 2015-06-30 14:27:27 -0700 | [diff] [blame] | 44 | #include <grpc/support/alloc.h> |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 45 | #include <grpc/support/sync.h> |
| 46 | #include <grpc/support/thd.h> |
| 47 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 48 | typedef struct |
| 49 | { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 50 | grpc_server *server; |
| 51 | grpc_completion_queue *cq; |
| 52 | grpc_bad_client_server_side_validator validator; |
| 53 | gpr_event done_thd; |
| 54 | gpr_event done_write; |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 55 | } thd_args; |
| 56 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 57 | static void |
| 58 | thd_func (void *arg) |
| 59 | { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 60 | thd_args *a = arg; |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 61 | a->validator (a->server, a->cq); |
| 62 | gpr_event_set (&a->done_thd, (void *) 1); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 63 | } |
| 64 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 65 | static void |
| 66 | done_write (void *arg, int success, grpc_closure_list * closure_list) |
| 67 | { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 68 | thd_args *a = arg; |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 69 | gpr_event_set (&a->done_write, (void *) 1); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 70 | } |
| 71 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 72 | static void |
| 73 | server_setup_transport (void *ts, grpc_transport * transport, grpc_mdctx * mdctx) |
| 74 | { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 75 | thd_args *a = ts; |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 76 | static grpc_channel_filter const *extra_filters[] = { |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 77 | &grpc_http_server_filter |
| 78 | }; |
Craig Tiller | d9ccbbf | 2015-09-22 09:30:00 -0700 | [diff] [blame] | 79 | grpc_closure_list closure_list = GRPC_CLOSURE_LIST_INIT; |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 80 | grpc_server_setup_transport (a->server, transport, extra_filters, GPR_ARRAY_SIZE (extra_filters), mdctx, grpc_server_get_channel_args (a->server), &closure_list); |
| 81 | grpc_closure_list_run (&closure_list); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 82 | } |
| 83 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 84 | void |
| 85 | grpc_run_bad_client_test (grpc_bad_client_server_side_validator validator, const char *client_payload, size_t client_payload_length, gpr_uint32 flags) |
| 86 | { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 87 | grpc_endpoint_pair sfd; |
| 88 | thd_args a; |
| 89 | gpr_thd_id id; |
Craig Tiller | 83b2f25 | 2015-06-05 14:50:00 -0700 | [diff] [blame] | 90 | char *hex; |
Craig Tiller | acf0f07 | 2015-06-29 08:24:16 -0700 | [diff] [blame] | 91 | grpc_transport *transport; |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 92 | grpc_mdctx *mdctx = grpc_mdctx_create (); |
| 93 | gpr_slice slice = gpr_slice_from_copied_buffer (client_payload, client_payload_length); |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame] | 94 | gpr_slice_buffer outgoing; |
Craig Tiller | 3382511 | 2015-09-18 07:44:19 -0700 | [diff] [blame] | 95 | grpc_closure done_write_closure; |
Craig Tiller | d9ccbbf | 2015-09-22 09:30:00 -0700 | [diff] [blame] | 96 | grpc_closure_list closure_list = GRPC_CLOSURE_LIST_INIT; |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 97 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 98 | hex = gpr_dump (client_payload, client_payload_length, GPR_DUMP_HEX | GPR_DUMP_ASCII); |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 99 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 100 | /* Add a debug log */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 101 | gpr_log (GPR_INFO, "TEST: %s", hex); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 102 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 103 | gpr_free (hex); |
Craig Tiller | e46f00a | 2015-06-30 14:27:27 -0700 | [diff] [blame] | 104 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 105 | /* Init grpc */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 106 | grpc_init (); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 107 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 108 | /* Create endpoints */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 109 | sfd = grpc_iomgr_create_endpoint_pair ("fixture", 65536); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 110 | |
| 111 | /* Create server, completion events */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 112 | a.server = grpc_server_create_from_filters (NULL, 0, NULL); |
| 113 | a.cq = grpc_completion_queue_create (NULL); |
| 114 | gpr_event_init (&a.done_thd); |
| 115 | gpr_event_init (&a.done_write); |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 116 | a.validator = validator; |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 117 | grpc_server_register_completion_queue (a.server, a.cq, NULL); |
| 118 | grpc_server_start (a.server); |
| 119 | transport = grpc_create_chttp2_transport (NULL, sfd.server, mdctx, 0, &closure_list); |
| 120 | server_setup_transport (&a, transport, mdctx); |
| 121 | grpc_chttp2_transport_start_reading (transport, NULL, 0, &closure_list); |
| 122 | grpc_closure_list_run (&closure_list); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 123 | |
| 124 | /* Bind everything into the same pollset */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 125 | grpc_endpoint_add_to_pollset (sfd.client, grpc_cq_pollset (a.cq), &closure_list); |
| 126 | grpc_endpoint_add_to_pollset (sfd.server, grpc_cq_pollset (a.cq), &closure_list); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 127 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 128 | /* Check a ground truth */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 129 | GPR_ASSERT (grpc_server_has_open_connections (a.server)); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 130 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 131 | /* Start validator */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 132 | gpr_thd_new (&id, thd_func, &a, NULL); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 133 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 134 | gpr_slice_buffer_init (&outgoing); |
| 135 | gpr_slice_buffer_add (&outgoing, slice); |
| 136 | grpc_closure_init (&done_write_closure, done_write, &a); |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame] | 137 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 138 | /* Write data */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 139 | grpc_endpoint_write (sfd.client, &outgoing, &done_write_closure, &closure_list); |
| 140 | grpc_closure_list_run (&closure_list); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 141 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 142 | /* Await completion */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 143 | GPR_ASSERT (gpr_event_wait (&a.done_write, GRPC_TIMEOUT_SECONDS_TO_DEADLINE (5))); |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 144 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 145 | if (flags & GRPC_BAD_CLIENT_DISCONNECT) |
| 146 | { |
| 147 | grpc_endpoint_shutdown (sfd.client, &closure_list); |
| 148 | grpc_endpoint_destroy (sfd.client, &closure_list); |
| 149 | grpc_closure_list_run (&closure_list); |
| 150 | sfd.client = NULL; |
| 151 | } |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 152 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 153 | GPR_ASSERT (gpr_event_wait (&a.done_thd, GRPC_TIMEOUT_SECONDS_TO_DEADLINE (5))); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 154 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 155 | /* Shutdown */ |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 156 | if (sfd.client) |
| 157 | { |
| 158 | grpc_endpoint_shutdown (sfd.client, &closure_list); |
| 159 | grpc_endpoint_destroy (sfd.client, &closure_list); |
| 160 | grpc_closure_list_run (&closure_list); |
| 161 | } |
| 162 | grpc_server_shutdown_and_notify (a.server, a.cq, NULL); |
| 163 | GPR_ASSERT (grpc_completion_queue_pluck (a.cq, NULL, GRPC_TIMEOUT_SECONDS_TO_DEADLINE (1), NULL).type == GRPC_OP_COMPLETE); |
| 164 | grpc_server_destroy (a.server); |
| 165 | grpc_completion_queue_destroy (a.cq); |
| 166 | gpr_slice_buffer_destroy (&outgoing); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 167 | |
Craig Tiller | 45724b3 | 2015-09-22 10:42:19 -0700 | [diff] [blame^] | 168 | grpc_closure_list_run (&closure_list); |
| 169 | grpc_shutdown (); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 170 | } |