Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 1 | /* |
| 2 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame^] | 3 | * Copyright 2015 gRPC authors. |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 4 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame^] | 5 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | * you may not use this file except in compliance with the License. |
| 7 | * You may obtain a copy of the License at |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 8 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame^] | 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 10 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame^] | 11 | * Unless required by applicable law or agreed to in writing, software |
| 12 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | * See the License for the specific language governing permissions and |
| 15 | * limitations under the License. |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 16 | * |
| 17 | */ |
| 18 | |
Craig Tiller | 9533d04 | 2016-03-25 17:11:06 -0700 | [diff] [blame] | 19 | #include "src/core/lib/surface/channel.h" |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 20 | |
| 21 | #include <stdlib.h> |
| 22 | #include <string.h> |
| 23 | |
David Garcia Quintas | 73dcbda | 2016-04-23 00:17:05 -0700 | [diff] [blame] | 24 | #include <grpc/compression.h> |
Craig Tiller | 1b22b9d | 2015-07-20 13:42:22 -0700 | [diff] [blame] | 25 | #include <grpc/support/alloc.h> |
| 26 | #include <grpc/support/log.h> |
| 27 | #include <grpc/support/string_util.h> |
| 28 | |
David Garcia Quintas | 73dcbda | 2016-04-23 00:17:05 -0700 | [diff] [blame] | 29 | #include "src/core/lib/channel/channel_args.h" |
Craig Tiller | 9533d04 | 2016-03-25 17:11:06 -0700 | [diff] [blame] | 30 | #include "src/core/lib/iomgr/iomgr.h" |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 31 | #include "src/core/lib/slice/slice_internal.h" |
Craig Tiller | 9533d04 | 2016-03-25 17:11:06 -0700 | [diff] [blame] | 32 | #include "src/core/lib/support/string.h" |
| 33 | #include "src/core/lib/surface/api_trace.h" |
| 34 | #include "src/core/lib/surface/call.h" |
| 35 | #include "src/core/lib/surface/channel_init.h" |
Craig Tiller | 9533d04 | 2016-03-25 17:11:06 -0700 | [diff] [blame] | 36 | #include "src/core/lib/transport/static_metadata.h" |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 37 | |
Craig Tiller | 5d5bd22 | 2015-06-08 17:01:49 -0700 | [diff] [blame] | 38 | /** Cache grpc-status: X mdelems for X = 0..NUM_CACHED_STATUS_ELEMS. |
| 39 | * Avoids needing to take a metadata context lock for sending status |
| 40 | * if the status code is <= NUM_CACHED_STATUS_ELEMS. |
| 41 | * Sized to allow the most commonly used codes to fit in |
| 42 | * (OK, Cancelled, Unknown). */ |
Craig Tiller | 3fc8e82 | 2015-06-08 16:31:28 -0700 | [diff] [blame] | 43 | #define NUM_CACHED_STATUS_ELEMS 3 |
| 44 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 45 | typedef struct registered_call { |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 46 | grpc_mdelem path; |
| 47 | grpc_mdelem authority; |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 48 | struct registered_call *next; |
| 49 | } registered_call; |
| 50 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 51 | struct grpc_channel { |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 52 | int is_client; |
David Garcia Quintas | 73dcbda | 2016-04-23 00:17:05 -0700 | [diff] [blame] | 53 | grpc_compression_options compression_options; |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 54 | grpc_mdelem default_authority; |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 55 | |
Craig Tiller | a6bec8f | 2017-03-14 08:26:04 -0700 | [diff] [blame] | 56 | gpr_atm call_size_estimate; |
| 57 | |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 58 | gpr_mu registered_call_mu; |
| 59 | registered_call *registered_calls; |
David Garcia Quintas | b0dd253 | 2016-05-17 19:14:51 -0700 | [diff] [blame] | 60 | |
Craig Tiller | 1b22b9d | 2015-07-20 13:42:22 -0700 | [diff] [blame] | 61 | char *target; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 62 | }; |
| 63 | |
Craig Tiller | 6902ad2 | 2015-04-16 08:01:49 -0700 | [diff] [blame] | 64 | #define CHANNEL_STACK_FROM_CHANNEL(c) ((grpc_channel_stack *)((c) + 1)) |
Craig Tiller | b20111c | 2015-04-10 23:27:11 +0000 | [diff] [blame] | 65 | #define CHANNEL_FROM_CHANNEL_STACK(channel_stack) \ |
| 66 | (((grpc_channel *)(channel_stack)) - 1) |
Craig Tiller | da66937 | 2015-02-05 10:10:15 -0800 | [diff] [blame] | 67 | #define CHANNEL_FROM_TOP_ELEM(top_elem) \ |
| 68 | CHANNEL_FROM_CHANNEL_STACK(grpc_channel_stack_from_top_element(top_elem)) |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 69 | |
Craig Tiller | f51457b | 2016-05-03 17:06:32 -0700 | [diff] [blame] | 70 | static void destroy_channel(grpc_exec_ctx *exec_ctx, void *arg, |
| 71 | grpc_error *error); |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 72 | |
Craig Tiller | b7f35a6 | 2017-03-16 13:29:05 -0700 | [diff] [blame] | 73 | grpc_channel *grpc_channel_create_with_builder( |
| 74 | grpc_exec_ctx *exec_ctx, grpc_channel_stack_builder *builder, |
| 75 | grpc_channel_stack_type channel_stack_type) { |
| 76 | char *target = gpr_strdup(grpc_channel_stack_builder_get_target(builder)); |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 77 | grpc_channel_args *args = grpc_channel_args_copy( |
Mark D. Roth | 5e2566e | 2016-11-18 10:53:13 -0800 | [diff] [blame] | 78 | grpc_channel_stack_builder_get_channel_arguments(builder)); |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 79 | grpc_channel *channel; |
Mark D. Roth | c108788 | 2016-11-18 10:54:45 -0800 | [diff] [blame] | 80 | grpc_error *error = grpc_channel_stack_builder_finish( |
Mark D. Roth | 5e2566e | 2016-11-18 10:53:13 -0800 | [diff] [blame] | 81 | exec_ctx, builder, sizeof(grpc_channel), 1, destroy_channel, NULL, |
Mark D. Roth | c108788 | 2016-11-18 10:54:45 -0800 | [diff] [blame] | 82 | (void **)&channel); |
Mark D. Roth | 5e2566e | 2016-11-18 10:53:13 -0800 | [diff] [blame] | 83 | if (error != GRPC_ERROR_NONE) { |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 84 | gpr_log(GPR_ERROR, "channel stack builder failed: %s", |
| 85 | grpc_error_string(error)); |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 86 | GRPC_ERROR_UNREF(error); |
Craig Tiller | b7f35a6 | 2017-03-16 13:29:05 -0700 | [diff] [blame] | 87 | gpr_free(target); |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 88 | goto done; |
Craig Tiller | 839bebe | 2016-04-06 08:07:11 -0700 | [diff] [blame] | 89 | } |
Craig Tiller | 178edfa | 2016-02-17 20:54:46 -0800 | [diff] [blame] | 90 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 91 | memset(channel, 0, sizeof(*channel)); |
Craig Tiller | b7f35a6 | 2017-03-16 13:29:05 -0700 | [diff] [blame] | 92 | channel->target = target; |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 93 | channel->is_client = grpc_channel_stack_type_is_client(channel_stack_type); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 94 | gpr_mu_init(&channel->registered_call_mu); |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 95 | channel->registered_calls = NULL; |
Craig Tiller | 629b0ed | 2015-04-22 11:14:26 -0700 | [diff] [blame] | 96 | |
Craig Tiller | a6bec8f | 2017-03-14 08:26:04 -0700 | [diff] [blame] | 97 | gpr_atm_no_barrier_store( |
| 98 | &channel->call_size_estimate, |
Craig Tiller | 2ccd502 | 2017-03-16 09:57:46 -0700 | [diff] [blame] | 99 | (gpr_atm)CHANNEL_STACK_FROM_CHANNEL(channel)->call_stack_size); |
Craig Tiller | a6bec8f | 2017-03-14 08:26:04 -0700 | [diff] [blame] | 100 | |
David Garcia Quintas | 73dcbda | 2016-04-23 00:17:05 -0700 | [diff] [blame] | 101 | grpc_compression_options_init(&channel->compression_options); |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 102 | for (size_t i = 0; i < args->num_args; i++) { |
| 103 | if (0 == strcmp(args->args[i].key, GRPC_ARG_DEFAULT_AUTHORITY)) { |
| 104 | if (args->args[i].type != GRPC_ARG_STRING) { |
| 105 | gpr_log(GPR_ERROR, "%s ignored: it must be a string", |
| 106 | GRPC_ARG_DEFAULT_AUTHORITY); |
| 107 | } else { |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 108 | if (!GRPC_MDISNULL(channel->default_authority)) { |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 109 | /* setting this takes precedence over anything else */ |
Craig Tiller | 4cc1c35 | 2016-12-27 08:48:01 -0800 | [diff] [blame] | 110 | GRPC_MDELEM_UNREF(exec_ctx, channel->default_authority); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 111 | } |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 112 | channel->default_authority = grpc_mdelem_from_slices( |
| 113 | exec_ctx, GRPC_MDSTR_AUTHORITY, |
Craig Tiller | 5f4264f | 2017-02-08 14:53:04 -0800 | [diff] [blame] | 114 | grpc_slice_intern( |
| 115 | grpc_slice_from_static_string(args->args[i].value.string))); |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 116 | } |
| 117 | } else if (0 == |
| 118 | strcmp(args->args[i].key, GRPC_SSL_TARGET_NAME_OVERRIDE_ARG)) { |
| 119 | if (args->args[i].type != GRPC_ARG_STRING) { |
| 120 | gpr_log(GPR_ERROR, "%s ignored: it must be a string", |
| 121 | GRPC_SSL_TARGET_NAME_OVERRIDE_ARG); |
| 122 | } else { |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 123 | if (!GRPC_MDISNULL(channel->default_authority)) { |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 124 | /* other ways of setting this (notably ssl) take precedence */ |
| 125 | gpr_log(GPR_ERROR, |
| 126 | "%s ignored: default host already set some other way", |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 127 | GRPC_SSL_TARGET_NAME_OVERRIDE_ARG); |
| 128 | } else { |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 129 | channel->default_authority = grpc_mdelem_from_slices( |
| 130 | exec_ctx, GRPC_MDSTR_AUTHORITY, |
Craig Tiller | 5f4264f | 2017-02-08 14:53:04 -0800 | [diff] [blame] | 131 | grpc_slice_intern( |
| 132 | grpc_slice_from_static_string(args->args[i].value.string))); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 133 | } |
| 134 | } |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 135 | } else if (0 == strcmp(args->args[i].key, |
| 136 | GRPC_COMPRESSION_CHANNEL_DEFAULT_LEVEL)) { |
| 137 | channel->compression_options.default_level.is_set = true; |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 138 | channel->compression_options.default_level.level = |
David Garcia Quintas | 228a514 | 2017-03-30 19:43:00 -0700 | [diff] [blame] | 139 | (grpc_compression_level)grpc_channel_arg_get_integer( |
| 140 | &args->args[i], |
| 141 | (grpc_integer_options){GRPC_COMPRESS_LEVEL_NONE, |
| 142 | GRPC_COMPRESS_LEVEL_NONE, |
| 143 | GRPC_COMPRESS_LEVEL_COUNT - 1}); |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 144 | } else if (0 == strcmp(args->args[i].key, |
| 145 | GRPC_COMPRESSION_CHANNEL_DEFAULT_ALGORITHM)) { |
| 146 | channel->compression_options.default_algorithm.is_set = true; |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 147 | channel->compression_options.default_algorithm.algorithm = |
David Garcia Quintas | 228a514 | 2017-03-30 19:43:00 -0700 | [diff] [blame] | 148 | (grpc_compression_algorithm)grpc_channel_arg_get_integer( |
| 149 | &args->args[i], |
| 150 | (grpc_integer_options){GRPC_COMPRESS_NONE, GRPC_COMPRESS_NONE, |
| 151 | GRPC_COMPRESS_ALGORITHMS_COUNT - 1}); |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 152 | } else if (0 == |
| 153 | strcmp(args->args[i].key, |
| 154 | GRPC_COMPRESSION_CHANNEL_ENABLED_ALGORITHMS_BITSET)) { |
| 155 | channel->compression_options.enabled_algorithms_bitset = |
| 156 | (uint32_t)args->args[i].value.integer | |
| 157 | 0x1; /* always support no compression */ |
Craig Tiller | 629b0ed | 2015-04-22 11:14:26 -0700 | [diff] [blame] | 158 | } |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 159 | } |
Craig Tiller | 629b0ed | 2015-04-22 11:14:26 -0700 | [diff] [blame] | 160 | |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 161 | done: |
Craig Tiller | 4cc1c35 | 2016-12-27 08:48:01 -0800 | [diff] [blame] | 162 | grpc_channel_args_destroy(exec_ctx, args); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 163 | return channel; |
| 164 | } |
| 165 | |
Craig Tiller | b7f35a6 | 2017-03-16 13:29:05 -0700 | [diff] [blame] | 166 | grpc_channel *grpc_channel_create(grpc_exec_ctx *exec_ctx, const char *target, |
| 167 | const grpc_channel_args *input_args, |
| 168 | grpc_channel_stack_type channel_stack_type, |
| 169 | grpc_transport *optional_transport) { |
| 170 | grpc_channel_stack_builder *builder = grpc_channel_stack_builder_create(); |
| 171 | grpc_channel_stack_builder_set_channel_arguments(exec_ctx, builder, |
| 172 | input_args); |
| 173 | grpc_channel_stack_builder_set_target(builder, target); |
| 174 | grpc_channel_stack_builder_set_transport(builder, optional_transport); |
| 175 | if (!grpc_channel_init_create_stack(exec_ctx, builder, channel_stack_type)) { |
| 176 | grpc_channel_stack_builder_destroy(exec_ctx, builder); |
| 177 | return NULL; |
| 178 | } |
| 179 | return grpc_channel_create_with_builder(exec_ctx, builder, |
| 180 | channel_stack_type); |
| 181 | } |
| 182 | |
Craig Tiller | a6bec8f | 2017-03-14 08:26:04 -0700 | [diff] [blame] | 183 | size_t grpc_channel_get_call_size_estimate(grpc_channel *channel) { |
| 184 | #define ROUND_UP_SIZE 256 |
Craig Tiller | 878a7c2 | 2017-03-23 15:14:48 -0700 | [diff] [blame] | 185 | /* We round up our current estimate to the NEXT value of ROUND_UP_SIZE. |
| 186 | This ensures: |
| 187 | 1. a consistent size allocation when our estimate is drifting slowly |
| 188 | (which is common) - which tends to help most allocators reuse memory |
| 189 | 2. a small amount of allowed growth over the estimate without hitting |
| 190 | the arena size doubling case, reducing overall memory usage */ |
Craig Tiller | b64d652 | 2017-03-23 15:17:18 -0700 | [diff] [blame] | 191 | return ((size_t)gpr_atm_no_barrier_load(&channel->call_size_estimate) + |
| 192 | 2 * ROUND_UP_SIZE) & |
| 193 | ~(size_t)(ROUND_UP_SIZE - 1); |
Craig Tiller | a6bec8f | 2017-03-14 08:26:04 -0700 | [diff] [blame] | 194 | } |
| 195 | |
| 196 | void grpc_channel_update_call_size_estimate(grpc_channel *channel, |
| 197 | size_t size) { |
| 198 | size_t cur = (size_t)gpr_atm_no_barrier_load(&channel->call_size_estimate); |
| 199 | if (cur < size) { |
| 200 | /* size grew: update estimate */ |
| 201 | gpr_atm_no_barrier_cas(&channel->call_size_estimate, (gpr_atm)cur, |
| 202 | (gpr_atm)size); |
| 203 | /* if we lose: never mind, something else will likely update soon enough */ |
| 204 | } else if (cur == size) { |
| 205 | /* no change: holding pattern */ |
| 206 | } else if (cur > 0) { |
| 207 | /* size shrank: decrease estimate */ |
| 208 | gpr_atm_no_barrier_cas( |
| 209 | &channel->call_size_estimate, (gpr_atm)cur, |
| 210 | (gpr_atm)(GPR_MIN(cur - 1, (255 * cur + size) / 256))); |
| 211 | /* if we lose: never mind, something else will likely update soon enough */ |
| 212 | } |
| 213 | } |
| 214 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 215 | char *grpc_channel_get_target(grpc_channel *channel) { |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 216 | GRPC_API_TRACE("grpc_channel_get_target(channel=%p)", 1, (channel)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 217 | return gpr_strdup(channel->target); |
Craig Tiller | 1b22b9d | 2015-07-20 13:42:22 -0700 | [diff] [blame] | 218 | } |
| 219 | |
Mark D. Roth | b2d2488 | 2016-10-27 15:44:07 -0700 | [diff] [blame] | 220 | void grpc_channel_get_info(grpc_channel *channel, |
Mark D. Roth | f79ce7d | 2016-11-04 08:43:36 -0700 | [diff] [blame] | 221 | const grpc_channel_info *channel_info) { |
Mark D. Roth | b2d2488 | 2016-10-27 15:44:07 -0700 | [diff] [blame] | 222 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
| 223 | grpc_channel_element *elem = |
| 224 | grpc_channel_stack_element(CHANNEL_STACK_FROM_CHANNEL(channel), 0); |
| 225 | elem->filter->get_channel_info(&exec_ctx, elem, channel_info); |
| 226 | grpc_exec_ctx_finish(&exec_ctx); |
| 227 | } |
| 228 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 229 | static grpc_call *grpc_channel_create_call_internal( |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 230 | grpc_exec_ctx *exec_ctx, grpc_channel *channel, grpc_call *parent_call, |
| 231 | uint32_t propagation_mask, grpc_completion_queue *cq, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 232 | grpc_pollset_set *pollset_set_alternative, grpc_mdelem path_mdelem, |
| 233 | grpc_mdelem authority_mdelem, gpr_timespec deadline) { |
| 234 | grpc_mdelem send_metadata[2]; |
Craig Tiller | 32ca48c | 2015-09-10 11:47:15 -0700 | [diff] [blame] | 235 | size_t num_metadata = 0; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 236 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 237 | GPR_ASSERT(channel->is_client); |
David Garcia Quintas | 879b3b9 | 2016-04-25 11:23:38 -0700 | [diff] [blame] | 238 | GPR_ASSERT(!(cq != NULL && pollset_set_alternative != NULL)); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 239 | |
Craig Tiller | c4b56b6 | 2015-07-23 17:44:11 -0700 | [diff] [blame] | 240 | send_metadata[num_metadata++] = path_mdelem; |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 241 | if (!GRPC_MDISNULL(authority_mdelem)) { |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 242 | send_metadata[num_metadata++] = authority_mdelem; |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 243 | } else if (!GRPC_MDISNULL(channel->default_authority)) { |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 244 | send_metadata[num_metadata++] = GRPC_MDELEM_REF(channel->default_authority); |
| 245 | } |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 246 | |
Craig Tiller | 8e21465 | 2016-08-19 09:54:31 -0700 | [diff] [blame] | 247 | grpc_call_create_args args; |
| 248 | memset(&args, 0, sizeof(args)); |
| 249 | args.channel = channel; |
| 250 | args.parent_call = parent_call; |
| 251 | args.propagation_mask = propagation_mask; |
| 252 | args.cq = cq; |
| 253 | args.pollset_set_alternative = pollset_set_alternative; |
| 254 | args.server_transport_data = NULL; |
| 255 | args.add_initial_metadata = send_metadata; |
| 256 | args.add_initial_metadata_count = num_metadata; |
| 257 | args.send_deadline = deadline; |
| 258 | |
| 259 | grpc_call *call; |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 260 | GRPC_LOG_IF_ERROR("call_create", grpc_call_create(exec_ctx, &args, &call)); |
Craig Tiller | 8e21465 | 2016-08-19 09:54:31 -0700 | [diff] [blame] | 261 | return call; |
Craig Tiller | 6902ad2 | 2015-04-16 08:01:49 -0700 | [diff] [blame] | 262 | } |
klempner | c463f74 | 2014-12-19 13:03:35 -0800 | [diff] [blame] | 263 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 264 | grpc_call *grpc_channel_create_call(grpc_channel *channel, |
| 265 | grpc_call *parent_call, |
Craig Tiller | 7536af0 | 2015-12-22 13:49:30 -0800 | [diff] [blame] | 266 | uint32_t propagation_mask, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 267 | grpc_completion_queue *cq, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 268 | grpc_slice method, const grpc_slice *host, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 269 | gpr_timespec deadline, void *reserved) { |
| 270 | GPR_ASSERT(!reserved); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 271 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
| 272 | grpc_call *call = grpc_channel_create_call_internal( |
| 273 | &exec_ctx, channel, parent_call, propagation_mask, cq, NULL, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 274 | grpc_mdelem_from_slices(&exec_ctx, GRPC_MDSTR_PATH, |
| 275 | grpc_slice_ref_internal(method)), |
| 276 | host != NULL ? grpc_mdelem_from_slices(&exec_ctx, GRPC_MDSTR_AUTHORITY, |
| 277 | grpc_slice_ref_internal(*host)) |
| 278 | : GRPC_MDNULL, |
David Garcia Quintas | 4afce7e | 2016-04-18 16:25:17 -0700 | [diff] [blame] | 279 | deadline); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 280 | grpc_exec_ctx_finish(&exec_ctx); |
| 281 | return call; |
David Garcia Quintas | 4afce7e | 2016-04-18 16:25:17 -0700 | [diff] [blame] | 282 | } |
| 283 | |
| 284 | grpc_call *grpc_channel_create_pollset_set_call( |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 285 | grpc_exec_ctx *exec_ctx, grpc_channel *channel, grpc_call *parent_call, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 286 | uint32_t propagation_mask, grpc_pollset_set *pollset_set, grpc_slice method, |
| 287 | const grpc_slice *host, gpr_timespec deadline, void *reserved) { |
David Garcia Quintas | 4afce7e | 2016-04-18 16:25:17 -0700 | [diff] [blame] | 288 | GPR_ASSERT(!reserved); |
| 289 | return grpc_channel_create_call_internal( |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 290 | exec_ctx, channel, parent_call, propagation_mask, NULL, pollset_set, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 291 | grpc_mdelem_from_slices(exec_ctx, GRPC_MDSTR_PATH, |
| 292 | grpc_slice_ref_internal(method)), |
| 293 | host != NULL ? grpc_mdelem_from_slices(exec_ctx, GRPC_MDSTR_AUTHORITY, |
| 294 | grpc_slice_ref_internal(*host)) |
| 295 | : GRPC_MDNULL, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 296 | deadline); |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 297 | } |
| 298 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 299 | void *grpc_channel_register_call(grpc_channel *channel, const char *method, |
| 300 | const char *host, void *reserved) { |
| 301 | registered_call *rc = gpr_malloc(sizeof(registered_call)); |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 302 | GRPC_API_TRACE( |
| 303 | "grpc_channel_register_call(channel=%p, method=%s, host=%s, reserved=%p)", |
| 304 | 4, (channel, method, host, reserved)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 305 | GPR_ASSERT(!reserved); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 306 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 307 | |
| 308 | rc->path = grpc_mdelem_from_slices( |
| 309 | &exec_ctx, GRPC_MDSTR_PATH, |
| 310 | grpc_slice_intern(grpc_slice_from_static_string(method))); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 311 | rc->authority = |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 312 | host ? grpc_mdelem_from_slices( |
| 313 | &exec_ctx, GRPC_MDSTR_AUTHORITY, |
| 314 | grpc_slice_intern(grpc_slice_from_static_string(host))) |
| 315 | : GRPC_MDNULL; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 316 | gpr_mu_lock(&channel->registered_call_mu); |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 317 | rc->next = channel->registered_calls; |
| 318 | channel->registered_calls = rc; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 319 | gpr_mu_unlock(&channel->registered_call_mu); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 320 | grpc_exec_ctx_finish(&exec_ctx); |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 321 | return rc; |
| 322 | } |
| 323 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 324 | grpc_call *grpc_channel_create_registered_call( |
Craig Tiller | 7536af0 | 2015-12-22 13:49:30 -0800 | [diff] [blame] | 325 | grpc_channel *channel, grpc_call *parent_call, uint32_t propagation_mask, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 326 | grpc_completion_queue *completion_queue, void *registered_call_handle, |
| 327 | gpr_timespec deadline, void *reserved) { |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 328 | registered_call *rc = registered_call_handle; |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 329 | GRPC_API_TRACE( |
| 330 | "grpc_channel_create_registered_call(" |
Craig Tiller | 4de3e4f | 2015-10-05 08:55:50 -0700 | [diff] [blame] | 331 | "channel=%p, parent_call=%p, propagation_mask=%x, completion_queue=%p, " |
| 332 | "registered_call_handle=%p, " |
Ken Payson | 62a6c92 | 2016-06-24 11:53:54 -0700 | [diff] [blame] | 333 | "deadline=gpr_timespec { tv_sec: %" PRId64 |
| 334 | ", tv_nsec: %d, clock_type: %d }, " |
Craig Tiller | 4de3e4f | 2015-10-05 08:55:50 -0700 | [diff] [blame] | 335 | "reserved=%p)", |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 336 | 9, (channel, parent_call, (unsigned)propagation_mask, completion_queue, |
Ken Payson | 62a6c92 | 2016-06-24 11:53:54 -0700 | [diff] [blame] | 337 | registered_call_handle, deadline.tv_sec, deadline.tv_nsec, |
| 338 | (int)deadline.clock_type, reserved)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 339 | GPR_ASSERT(!reserved); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 340 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
| 341 | grpc_call *call = grpc_channel_create_call_internal( |
| 342 | &exec_ctx, channel, parent_call, propagation_mask, completion_queue, NULL, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 343 | GRPC_MDELEM_REF(rc->path), GRPC_MDELEM_REF(rc->authority), deadline); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 344 | grpc_exec_ctx_finish(&exec_ctx); |
| 345 | return call; |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 346 | } |
| 347 | |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 348 | #ifdef GRPC_STREAM_REFCOUNT_DEBUG |
| 349 | #define REF_REASON reason |
| 350 | #define REF_ARG , const char *reason |
Craig Tiller | 9ec2a52 | 2015-05-29 22:46:54 -0700 | [diff] [blame] | 351 | #else |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 352 | #define REF_REASON "" |
| 353 | #define REF_ARG |
Craig Tiller | 9ec2a52 | 2015-05-29 22:46:54 -0700 | [diff] [blame] | 354 | #endif |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 355 | void grpc_channel_internal_ref(grpc_channel *c REF_ARG) { |
| 356 | GRPC_CHANNEL_STACK_REF(CHANNEL_STACK_FROM_CHANNEL(c), REF_REASON); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 357 | } |
| 358 | |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 359 | void grpc_channel_internal_unref(grpc_exec_ctx *exec_ctx, |
| 360 | grpc_channel *c REF_ARG) { |
| 361 | GRPC_CHANNEL_STACK_UNREF(exec_ctx, CHANNEL_STACK_FROM_CHANNEL(c), REF_REASON); |
| 362 | } |
| 363 | |
| 364 | static void destroy_channel(grpc_exec_ctx *exec_ctx, void *arg, |
Craig Tiller | f51457b | 2016-05-03 17:06:32 -0700 | [diff] [blame] | 365 | grpc_error *error) { |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 366 | grpc_channel *channel = arg; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 367 | grpc_channel_stack_destroy(exec_ctx, CHANNEL_STACK_FROM_CHANNEL(channel)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 368 | while (channel->registered_calls) { |
| 369 | registered_call *rc = channel->registered_calls; |
| 370 | channel->registered_calls = rc->next; |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 371 | GRPC_MDELEM_UNREF(exec_ctx, rc->path); |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 372 | GRPC_MDELEM_UNREF(exec_ctx, rc->authority); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 373 | gpr_free(rc); |
| 374 | } |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 375 | GRPC_MDELEM_UNREF(exec_ctx, channel->default_authority); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 376 | gpr_mu_destroy(&channel->registered_call_mu); |
| 377 | gpr_free(channel->target); |
| 378 | gpr_free(channel); |
Craig Tiller | 7bd5ab1 | 2015-02-17 22:29:04 -0800 | [diff] [blame] | 379 | } |
| 380 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 381 | void grpc_channel_destroy(grpc_channel *channel) { |
Craig Tiller | e0221ff | 2016-07-11 15:56:08 -0700 | [diff] [blame] | 382 | grpc_transport_op *op = grpc_make_transport_op(NULL); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 383 | grpc_channel_element *elem; |
Craig Tiller | f5768a6 | 2015-09-22 10:54:34 -0700 | [diff] [blame] | 384 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 385 | GRPC_API_TRACE("grpc_channel_destroy(channel=%p)", 1, (channel)); |
ncteisen | 4b36a3d | 2017-03-13 19:08:06 -0700 | [diff] [blame] | 386 | op->disconnect_with_error = |
| 387 | GRPC_ERROR_CREATE_FROM_STATIC_STRING("Channel Destroyed"); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 388 | elem = grpc_channel_stack_element(CHANNEL_STACK_FROM_CHANNEL(channel), 0); |
Craig Tiller | e0221ff | 2016-07-11 15:56:08 -0700 | [diff] [blame] | 389 | elem->filter->start_transport_op(&exec_ctx, elem, op); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 390 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 391 | GRPC_CHANNEL_INTERNAL_UNREF(&exec_ctx, channel, "channel"); |
Craig Tiller | dfff1b8 | 2015-09-21 14:39:57 -0700 | [diff] [blame] | 392 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 393 | grpc_exec_ctx_finish(&exec_ctx); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 394 | } |
| 395 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 396 | grpc_channel_stack *grpc_channel_get_channel_stack(grpc_channel *channel) { |
| 397 | return CHANNEL_STACK_FROM_CHANNEL(channel); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 398 | } |
| 399 | |
David Garcia Quintas | ac09447 | 2016-05-18 20:25:57 -0700 | [diff] [blame] | 400 | grpc_compression_options grpc_channel_compression_options( |
David Garcia Quintas | 73dcbda | 2016-04-23 00:17:05 -0700 | [diff] [blame] | 401 | const grpc_channel *channel) { |
| 402 | return channel->compression_options; |
| 403 | } |
| 404 | |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 405 | grpc_mdelem grpc_channel_get_reffed_status_elem(grpc_exec_ctx *exec_ctx, |
| 406 | grpc_channel *channel, int i) { |
Craig Tiller | ebdef9d | 2015-11-19 17:09:49 -0800 | [diff] [blame] | 407 | char tmp[GPR_LTOA_MIN_BUFSIZE]; |
| 408 | switch (i) { |
| 409 | case 0: |
| 410 | return GRPC_MDELEM_GRPC_STATUS_0; |
| 411 | case 1: |
| 412 | return GRPC_MDELEM_GRPC_STATUS_1; |
| 413 | case 2: |
| 414 | return GRPC_MDELEM_GRPC_STATUS_2; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 415 | } |
Craig Tiller | ebdef9d | 2015-11-19 17:09:49 -0800 | [diff] [blame] | 416 | gpr_ltoa(i, tmp); |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 417 | return grpc_mdelem_from_slices(exec_ctx, GRPC_MDSTR_GRPC_STATUS, |
| 418 | grpc_slice_from_copied_string(tmp)); |
Craig Tiller | 190d360 | 2015-02-18 09:23:38 -0800 | [diff] [blame] | 419 | } |