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 | |
| 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; |
| 52 | gpr_event done_thd; |
| 53 | gpr_event done_write; |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 54 | } thd_args; |
| 55 | |
| 56 | static void thd_func(void *arg) { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 57 | thd_args *a = arg; |
| 58 | a->validator(a->server, a->cq); |
| 59 | gpr_event_set(&a->done_thd, (void *)1); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 60 | } |
| 61 | |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame^] | 62 | static void done_write(void *arg, int success) { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 63 | thd_args *a = arg; |
| 64 | gpr_event_set(&a->done_write, (void *)1); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 65 | } |
| 66 | |
Craig Tiller | 079a11b | 2015-06-30 10:07:15 -0700 | [diff] [blame] | 67 | static void server_setup_transport(void *ts, grpc_transport *transport, |
| 68 | grpc_mdctx *mdctx) { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 69 | thd_args *a = ts; |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 70 | static grpc_channel_filter const *extra_filters[] = { |
| 71 | &grpc_http_server_filter}; |
Craig Tiller | acf0f07 | 2015-06-29 08:24:16 -0700 | [diff] [blame] | 72 | grpc_server_setup_transport(a->server, transport, extra_filters, |
Craig Tiller | 079a11b | 2015-06-30 10:07:15 -0700 | [diff] [blame] | 73 | GPR_ARRAY_SIZE(extra_filters), mdctx, |
| 74 | grpc_server_get_channel_args(a->server)); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 75 | } |
| 76 | |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 77 | void grpc_run_bad_client_test(grpc_bad_client_server_side_validator validator, |
| 78 | const char *client_payload, |
Craig Tiller | 83b2f25 | 2015-06-05 14:50:00 -0700 | [diff] [blame] | 79 | size_t client_payload_length, gpr_uint32 flags) { |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 80 | grpc_endpoint_pair sfd; |
| 81 | thd_args a; |
| 82 | gpr_thd_id id; |
Craig Tiller | 83b2f25 | 2015-06-05 14:50:00 -0700 | [diff] [blame] | 83 | char *hex; |
Craig Tiller | acf0f07 | 2015-06-29 08:24:16 -0700 | [diff] [blame] | 84 | grpc_transport *transport; |
| 85 | grpc_mdctx *mdctx = grpc_mdctx_create(); |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 86 | gpr_slice slice = |
| 87 | gpr_slice_from_copied_buffer(client_payload, client_payload_length); |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame^] | 88 | gpr_slice_buffer outgoing; |
| 89 | grpc_iomgr_closure done_write_closure; |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 90 | |
Julien Boeuf | da13cd2 | 2015-06-29 19:25:32 +0200 | [diff] [blame] | 91 | hex = gpr_dump(client_payload, client_payload_length, |
| 92 | GPR_DUMP_HEX | GPR_DUMP_ASCII); |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 93 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 94 | /* Add a debug log */ |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 95 | gpr_log(GPR_INFO, "TEST: %s", hex); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 96 | |
Craig Tiller | e46f00a | 2015-06-30 14:27:27 -0700 | [diff] [blame] | 97 | gpr_free(hex); |
| 98 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 99 | /* Init grpc */ |
| 100 | grpc_init(); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 101 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 102 | /* Create endpoints */ |
Craig Tiller | fa275a9 | 2015-06-01 13:55:54 -0700 | [diff] [blame] | 103 | sfd = grpc_iomgr_create_endpoint_pair("fixture", 65536); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 104 | |
| 105 | /* Create server, completion events */ |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 106 | a.server = grpc_server_create_from_filters(NULL, 0, NULL); |
Nicolas "Pixel" Noble | 9a123df | 2015-07-29 23:45:08 +0200 | [diff] [blame] | 107 | a.cq = grpc_completion_queue_create(NULL); |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 108 | gpr_event_init(&a.done_thd); |
| 109 | gpr_event_init(&a.done_write); |
| 110 | a.validator = validator; |
Nicolas "Pixel" Noble | 9a123df | 2015-07-29 23:45:08 +0200 | [diff] [blame] | 111 | grpc_server_register_completion_queue(a.server, a.cq, NULL); |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 112 | grpc_server_start(a.server); |
Craig Tiller | 1dd7026 | 2015-07-06 11:45:30 -0700 | [diff] [blame] | 113 | transport = grpc_create_chttp2_transport(NULL, sfd.server, mdctx, 0); |
Craig Tiller | acf0f07 | 2015-06-29 08:24:16 -0700 | [diff] [blame] | 114 | server_setup_transport(&a, transport, mdctx); |
Craig Tiller | 1dd7026 | 2015-07-06 11:45:30 -0700 | [diff] [blame] | 115 | grpc_chttp2_transport_start_reading(transport, NULL, 0); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 116 | |
| 117 | /* Bind everything into the same pollset */ |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 118 | grpc_endpoint_add_to_pollset(sfd.client, grpc_cq_pollset(a.cq)); |
| 119 | grpc_endpoint_add_to_pollset(sfd.server, grpc_cq_pollset(a.cq)); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 120 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 121 | /* Check a ground truth */ |
| 122 | GPR_ASSERT(grpc_server_has_open_connections(a.server)); |
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 | /* Start validator */ |
| 125 | gpr_thd_new(&id, thd_func, &a, NULL); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 126 | |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame^] | 127 | gpr_slice_buffer_init(&outgoing); |
| 128 | gpr_slice_buffer_add(&outgoing, slice); |
| 129 | grpc_iomgr_closure_init(&done_write_closure, done_write, &a); |
| 130 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 131 | /* Write data */ |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame^] | 132 | switch (grpc_endpoint_write(sfd.client, &outgoing, &done_write_closure)) { |
| 133 | case GRPC_ENDPOINT_DONE: |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 134 | done_write(&a, 1); |
| 135 | break; |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame^] | 136 | case GRPC_ENDPOINT_PENDING: |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 137 | break; |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame^] | 138 | case GRPC_ENDPOINT_ERROR: |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 139 | done_write(&a, 0); |
| 140 | break; |
| 141 | } |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 142 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 143 | /* Await completion */ |
| 144 | GPR_ASSERT( |
| 145 | gpr_event_wait(&a.done_write, GRPC_TIMEOUT_SECONDS_TO_DEADLINE(5))); |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 146 | |
| 147 | if (flags & GRPC_BAD_CLIENT_DISCONNECT) { |
| 148 | grpc_endpoint_destroy(sfd.client); |
| 149 | sfd.client = NULL; |
| 150 | } |
| 151 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 152 | 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] | 153 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 154 | /* Shutdown */ |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 155 | if (sfd.client) { |
| 156 | grpc_endpoint_destroy(sfd.client); |
| 157 | } |
Craig Tiller | e511fdb | 2015-05-27 13:00:38 -0700 | [diff] [blame] | 158 | grpc_server_shutdown_and_notify(a.server, a.cq, NULL); |
Craig Tiller | d6c98df | 2015-08-18 09:33:44 -0700 | [diff] [blame] | 159 | GPR_ASSERT(grpc_completion_queue_pluck( |
| 160 | a.cq, NULL, GRPC_TIMEOUT_SECONDS_TO_DEADLINE(1), NULL) |
Craig Tiller | e511fdb | 2015-05-27 13:00:38 -0700 | [diff] [blame] | 161 | .type == GRPC_OP_COMPLETE); |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 162 | grpc_server_destroy(a.server); |
| 163 | grpc_completion_queue_destroy(a.cq); |
Craig Tiller | b029859 | 2015-08-27 07:38:01 -0700 | [diff] [blame^] | 164 | gpr_slice_buffer_destroy(&outgoing); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 165 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 166 | grpc_shutdown(); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 167 | } |