Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 1 | /* |
| 2 | * |
Craig Tiller | a93a25f | 2016-01-28 13:55:49 -0800 | [diff] [blame] | 3 | * Copyright 2015-2016, Google Inc. |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 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 | |
Craig Tiller | adcb92d | 2016-03-28 10:14:05 -0700 | [diff] [blame] | 36 | #include "src/core/ext/transport/chttp2/transport/chttp2_transport.h" |
Craig Tiller | 9533d04 | 2016-03-25 17:11:06 -0700 | [diff] [blame] | 37 | #include "src/core/lib/channel/channel_stack.h" |
| 38 | #include "src/core/lib/channel/http_server_filter.h" |
| 39 | #include "src/core/lib/iomgr/endpoint_pair.h" |
| 40 | #include "src/core/lib/support/string.h" |
| 41 | #include "src/core/lib/surface/completion_queue.h" |
| 42 | #include "src/core/lib/surface/server.h" |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 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 | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 48 | typedef struct { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 49 | grpc_server *server; |
| 50 | grpc_completion_queue *cq; |
| 51 | grpc_bad_client_server_side_validator validator; |
Craig Tiller | ac3220b | 2015-12-14 12:27:03 -0800 | [diff] [blame] | 52 | void *registered_method; |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 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 | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 57 | static void thd_func(void *arg) { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 58 | thd_args *a = arg; |
Craig Tiller | ac3220b | 2015-12-14 12:27:03 -0800 | [diff] [blame] | 59 | a->validator(a->server, a->cq, a->registered_method); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 60 | gpr_event_set(&a->done_thd, (void *)1); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 61 | } |
| 62 | |
Craig Tiller | 6c39686 | 2016-01-28 13:53:40 -0800 | [diff] [blame] | 63 | static void done_write(grpc_exec_ctx *exec_ctx, void *arg, bool success) { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 64 | thd_args *a = arg; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 65 | gpr_event_set(&a->done_write, (void *)1); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 66 | } |
| 67 | |
Craig Tiller | b2b4261 | 2015-11-20 12:02:17 -0800 | [diff] [blame] | 68 | static void server_setup_transport(void *ts, grpc_transport *transport) { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 69 | thd_args *a = ts; |
Craig Tiller | f5768a6 | 2015-09-22 10:54:34 -0700 | [diff] [blame] | 70 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
Craig Tiller | 178edfa | 2016-02-17 20:54:46 -0800 | [diff] [blame] | 71 | grpc_server_setup_transport(&exec_ctx, a->server, transport, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 72 | grpc_server_get_channel_args(a->server)); |
| 73 | grpc_exec_ctx_finish(&exec_ctx); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 74 | } |
| 75 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 76 | void grpc_run_bad_client_test(grpc_bad_client_server_side_validator validator, |
| 77 | const char *client_payload, |
Craig Tiller | 7536af0 | 2015-12-22 13:49:30 -0800 | [diff] [blame] | 78 | size_t client_payload_length, uint32_t flags) { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 79 | grpc_endpoint_pair sfd; |
| 80 | thd_args a; |
| 81 | gpr_thd_id id; |
Craig Tiller | 83b2f25 | 2015-06-05 14:50:00 -0700 | [diff] [blame] | 82 | char *hex; |
Craig Tiller | acf0f07 | 2015-06-29 08:24:16 -0700 | [diff] [blame] | 83 | grpc_transport *transport; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 84 | gpr_slice slice = |
| 85 | gpr_slice_from_copied_buffer(client_payload, client_payload_length); |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame] | 86 | gpr_slice_buffer outgoing; |
Craig Tiller | 3382511 | 2015-09-18 07:44:19 -0700 | [diff] [blame] | 87 | grpc_closure done_write_closure; |
Craig Tiller | f5768a6 | 2015-09-22 10:54:34 -0700 | [diff] [blame] | 88 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 89 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 90 | hex = gpr_dump(client_payload, client_payload_length, |
| 91 | GPR_DUMP_HEX | GPR_DUMP_ASCII); |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 92 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 93 | /* Add a debug log */ |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 94 | gpr_log(GPR_INFO, "TEST: %s", hex); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 95 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 96 | gpr_free(hex); |
Craig Tiller | e46f00a | 2015-06-30 14:27:27 -0700 | [diff] [blame] | 97 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 98 | /* Init grpc */ |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 99 | grpc_init(); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 100 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 101 | /* Create endpoints */ |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 102 | sfd = grpc_iomgr_create_endpoint_pair("fixture", 65536); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 103 | |
| 104 | /* Create server, completion events */ |
Craig Tiller | 178edfa | 2016-02-17 20:54:46 -0800 | [diff] [blame] | 105 | a.server = grpc_server_create(NULL, NULL); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 106 | a.cq = grpc_completion_queue_create(NULL); |
| 107 | gpr_event_init(&a.done_thd); |
| 108 | gpr_event_init(&a.done_write); |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 109 | a.validator = validator; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 110 | grpc_server_register_completion_queue(a.server, a.cq, NULL); |
Craig Tiller | 6cb6917 | 2015-12-14 13:35:53 -0800 | [diff] [blame] | 111 | a.registered_method = |
| 112 | grpc_server_register_method(a.server, GRPC_BAD_CLIENT_REGISTERED_METHOD, |
| 113 | GRPC_BAD_CLIENT_REGISTERED_HOST); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 114 | grpc_server_start(a.server); |
Craig Tiller | b2b4261 | 2015-11-20 12:02:17 -0800 | [diff] [blame] | 115 | transport = grpc_create_chttp2_transport(&exec_ctx, NULL, sfd.server, 0); |
| 116 | server_setup_transport(&a, transport); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 117 | grpc_chttp2_transport_start_reading(&exec_ctx, transport, NULL, 0); |
| 118 | grpc_exec_ctx_finish(&exec_ctx); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 119 | |
| 120 | /* Bind everything into the same pollset */ |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 121 | grpc_endpoint_add_to_pollset(&exec_ctx, sfd.client, grpc_cq_pollset(a.cq)); |
| 122 | grpc_endpoint_add_to_pollset(&exec_ctx, sfd.server, grpc_cq_pollset(a.cq)); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 123 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 124 | /* Check a ground truth */ |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 125 | GPR_ASSERT(grpc_server_has_open_connections(a.server)); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 126 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 127 | /* Start validator */ |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 128 | gpr_thd_new(&id, thd_func, &a, NULL); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 129 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 130 | gpr_slice_buffer_init(&outgoing); |
| 131 | gpr_slice_buffer_add(&outgoing, slice); |
| 132 | grpc_closure_init(&done_write_closure, done_write, &a); |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame] | 133 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 134 | /* Write data */ |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 135 | grpc_endpoint_write(&exec_ctx, sfd.client, &outgoing, &done_write_closure); |
| 136 | grpc_exec_ctx_finish(&exec_ctx); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 137 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 138 | /* Await completion */ |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 139 | GPR_ASSERT( |
| 140 | gpr_event_wait(&a.done_write, GRPC_TIMEOUT_SECONDS_TO_DEADLINE(5))); |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 141 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 142 | if (flags & GRPC_BAD_CLIENT_DISCONNECT) { |
| 143 | grpc_endpoint_shutdown(&exec_ctx, sfd.client); |
| 144 | grpc_endpoint_destroy(&exec_ctx, sfd.client); |
| 145 | grpc_exec_ctx_finish(&exec_ctx); |
| 146 | sfd.client = NULL; |
| 147 | } |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 148 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 149 | 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] | 150 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 151 | /* Shutdown */ |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 152 | if (sfd.client) { |
| 153 | grpc_endpoint_shutdown(&exec_ctx, sfd.client); |
| 154 | grpc_endpoint_destroy(&exec_ctx, sfd.client); |
| 155 | grpc_exec_ctx_finish(&exec_ctx); |
| 156 | } |
| 157 | grpc_server_shutdown_and_notify(a.server, a.cq, NULL); |
Craig Tiller | f40df23 | 2016-03-25 13:38:14 -0700 | [diff] [blame] | 158 | GPR_ASSERT(grpc_completion_queue_pluck( |
| 159 | a.cq, NULL, GRPC_TIMEOUT_SECONDS_TO_DEADLINE(1), NULL) |
| 160 | .type == GRPC_OP_COMPLETE); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 161 | grpc_server_destroy(a.server); |
| 162 | grpc_completion_queue_destroy(a.cq); |
| 163 | gpr_slice_buffer_destroy(&outgoing); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 164 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 165 | grpc_exec_ctx_finish(&exec_ctx); |
| 166 | grpc_shutdown(); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 167 | } |