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 | |
| 62 | static void done_write(void *arg, grpc_endpoint_cb_status status) { |
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 | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 67 | static grpc_transport_setup_result server_setup_transport( |
| 68 | void *ts, grpc_transport *transport, grpc_mdctx *mdctx) { |
| 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}; |
| 72 | return grpc_server_setup_transport(a->server, transport, extra_filters, |
Julien Boeuf | ff69cd1 | 2015-05-29 21:55:51 -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 | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 84 | gpr_slice slice = |
| 85 | gpr_slice_from_copied_buffer(client_payload, client_payload_length); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 86 | |
Craig Tiller | 83b2f25 | 2015-06-05 14:50:00 -0700 | [diff] [blame] | 87 | hex = |
| 88 | gpr_hexdump(client_payload, client_payload_length, GPR_HEXDUMP_PLAINTEXT); |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 89 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 90 | /* Add a debug log */ |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 91 | gpr_log(GPR_INFO, "TEST: %s", hex); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 92 | |
Craig Tiller | e46f00a | 2015-06-30 14:27:27 -0700 | [diff] [blame^] | 93 | gpr_free(hex); |
| 94 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 95 | /* Init grpc */ |
| 96 | grpc_init(); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 97 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 98 | /* Create endpoints */ |
Craig Tiller | fa275a9 | 2015-06-01 13:55:54 -0700 | [diff] [blame] | 99 | sfd = grpc_iomgr_create_endpoint_pair("fixture", 65536); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 100 | |
| 101 | /* Create server, completion events */ |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 102 | a.server = grpc_server_create_from_filters(NULL, 0, NULL); |
| 103 | a.cq = grpc_completion_queue_create(); |
| 104 | gpr_event_init(&a.done_thd); |
| 105 | gpr_event_init(&a.done_write); |
| 106 | a.validator = validator; |
| 107 | grpc_server_register_completion_queue(a.server, a.cq); |
| 108 | grpc_server_start(a.server); |
| 109 | grpc_create_chttp2_transport(server_setup_transport, &a, NULL, sfd.server, |
| 110 | NULL, 0, grpc_mdctx_create(), 0); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 111 | |
| 112 | /* Bind everything into the same pollset */ |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 113 | grpc_endpoint_add_to_pollset(sfd.client, grpc_cq_pollset(a.cq)); |
| 114 | grpc_endpoint_add_to_pollset(sfd.server, grpc_cq_pollset(a.cq)); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 115 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 116 | /* Check a ground truth */ |
| 117 | GPR_ASSERT(grpc_server_has_open_connections(a.server)); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 118 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 119 | /* Start validator */ |
| 120 | gpr_thd_new(&id, thd_func, &a, NULL); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 121 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 122 | /* Write data */ |
| 123 | switch (grpc_endpoint_write(sfd.client, &slice, 1, done_write, &a)) { |
| 124 | case GRPC_ENDPOINT_WRITE_DONE: |
| 125 | done_write(&a, 1); |
| 126 | break; |
| 127 | case GRPC_ENDPOINT_WRITE_PENDING: |
| 128 | break; |
| 129 | case GRPC_ENDPOINT_WRITE_ERROR: |
| 130 | done_write(&a, 0); |
| 131 | break; |
| 132 | } |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 133 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 134 | /* Await completion */ |
| 135 | GPR_ASSERT( |
| 136 | gpr_event_wait(&a.done_write, GRPC_TIMEOUT_SECONDS_TO_DEADLINE(5))); |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 137 | |
| 138 | if (flags & GRPC_BAD_CLIENT_DISCONNECT) { |
| 139 | grpc_endpoint_destroy(sfd.client); |
| 140 | sfd.client = NULL; |
| 141 | } |
| 142 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 143 | 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] | 144 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 145 | /* Shutdown */ |
Craig Tiller | 3c26d09 | 2015-06-05 14:48:30 -0700 | [diff] [blame] | 146 | if (sfd.client) { |
| 147 | grpc_endpoint_destroy(sfd.client); |
| 148 | } |
Craig Tiller | e511fdb | 2015-05-27 13:00:38 -0700 | [diff] [blame] | 149 | grpc_server_shutdown_and_notify(a.server, a.cq, NULL); |
| 150 | GPR_ASSERT(grpc_completion_queue_pluck(a.cq, NULL, |
| 151 | GRPC_TIMEOUT_SECONDS_TO_DEADLINE(1)) |
| 152 | .type == GRPC_OP_COMPLETE); |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 153 | grpc_server_destroy(a.server); |
| 154 | grpc_completion_queue_destroy(a.cq); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 155 | |
Craig Tiller | 4ebeb40 | 2015-05-26 13:31:52 +0000 | [diff] [blame] | 156 | grpc_shutdown(); |
Craig Tiller | ba3c3cd | 2015-05-26 06:28:10 -0700 | [diff] [blame] | 157 | } |