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, |
Muxi Yan | d37411e | 2017-08-17 18:44:12 -0700 | [diff] [blame] | 145 | GRPC_STREAM_COMPRESSION_CHANNEL_DEFAULT_LEVEL)) { |
| 146 | channel->compression_options.default_stream_compression_level.is_set = |
| 147 | true; |
| 148 | channel->compression_options.default_stream_compression_level.level = |
Muxi Yan | da4f087 | 2017-08-18 17:05:27 -0700 | [diff] [blame^] | 149 | (grpc_stream_compression_level)grpc_channel_arg_get_integer( |
Muxi Yan | d37411e | 2017-08-17 18:44:12 -0700 | [diff] [blame] | 150 | &args->args[i], |
Muxi Yan | da4f087 | 2017-08-18 17:05:27 -0700 | [diff] [blame^] | 151 | (grpc_integer_options){GRPC_STREAM_COMPRESS_LEVEL_NONE, |
| 152 | GRPC_STREAM_COMPRESS_LEVEL_NONE, |
| 153 | GRPC_STREAM_COMPRESS_LEVEL_COUNT - 1}); |
Muxi Yan | d37411e | 2017-08-17 18:44:12 -0700 | [diff] [blame] | 154 | } else if (0 == strcmp(args->args[i].key, |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 155 | GRPC_COMPRESSION_CHANNEL_DEFAULT_ALGORITHM)) { |
| 156 | channel->compression_options.default_algorithm.is_set = true; |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 157 | channel->compression_options.default_algorithm.algorithm = |
David Garcia Quintas | 228a514 | 2017-03-30 19:43:00 -0700 | [diff] [blame] | 158 | (grpc_compression_algorithm)grpc_channel_arg_get_integer( |
| 159 | &args->args[i], |
| 160 | (grpc_integer_options){GRPC_COMPRESS_NONE, GRPC_COMPRESS_NONE, |
| 161 | GRPC_COMPRESS_ALGORITHMS_COUNT - 1}); |
Muxi Yan | 68a0fd5 | 2017-07-21 09:26:04 -0700 | [diff] [blame] | 162 | } else if (0 == strcmp(args->args[i].key, |
| 163 | GRPC_STREAM_COMPRESSION_CHANNEL_DEFAULT_ALGORITHM)) { |
| 164 | channel->compression_options.default_stream_compression_algorithm.is_set = |
| 165 | true; |
| 166 | channel->compression_options.default_stream_compression_algorithm |
| 167 | .algorithm = |
| 168 | (grpc_stream_compression_algorithm)grpc_channel_arg_get_integer( |
| 169 | &args->args[i], |
| 170 | (grpc_integer_options){ |
| 171 | GRPC_STREAM_COMPRESS_NONE, GRPC_STREAM_COMPRESS_NONE, |
| 172 | GRPC_STREAM_COMPRESS_ALGORITHMS_COUNT - 1}); |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 173 | } else if (0 == |
| 174 | strcmp(args->args[i].key, |
| 175 | GRPC_COMPRESSION_CHANNEL_ENABLED_ALGORITHMS_BITSET)) { |
| 176 | channel->compression_options.enabled_algorithms_bitset = |
| 177 | (uint32_t)args->args[i].value.integer | |
| 178 | 0x1; /* always support no compression */ |
Muxi Yan | 68a0fd5 | 2017-07-21 09:26:04 -0700 | [diff] [blame] | 179 | } else if (0 == |
| 180 | strcmp( |
| 181 | args->args[i].key, |
| 182 | GRPC_STREAM_COMPRESSION_CHANNEL_ENABLED_ALGORITHMS_BITSET)) { |
| 183 | channel->compression_options |
| 184 | .enabled_stream_compression_algorithms_bitset = |
| 185 | (uint32_t)args->args[i].value.integer | |
| 186 | 0x1; /* always support no compression */ |
Craig Tiller | 629b0ed | 2015-04-22 11:14:26 -0700 | [diff] [blame] | 187 | } |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 188 | } |
Craig Tiller | 629b0ed | 2015-04-22 11:14:26 -0700 | [diff] [blame] | 189 | |
Mark D. Roth | e62605f | 2016-11-29 16:31:36 +0000 | [diff] [blame] | 190 | done: |
Craig Tiller | 4cc1c35 | 2016-12-27 08:48:01 -0800 | [diff] [blame] | 191 | grpc_channel_args_destroy(exec_ctx, args); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 192 | return channel; |
| 193 | } |
| 194 | |
Craig Tiller | b7f35a6 | 2017-03-16 13:29:05 -0700 | [diff] [blame] | 195 | grpc_channel *grpc_channel_create(grpc_exec_ctx *exec_ctx, const char *target, |
| 196 | const grpc_channel_args *input_args, |
| 197 | grpc_channel_stack_type channel_stack_type, |
| 198 | grpc_transport *optional_transport) { |
| 199 | grpc_channel_stack_builder *builder = grpc_channel_stack_builder_create(); |
| 200 | grpc_channel_stack_builder_set_channel_arguments(exec_ctx, builder, |
| 201 | input_args); |
| 202 | grpc_channel_stack_builder_set_target(builder, target); |
| 203 | grpc_channel_stack_builder_set_transport(builder, optional_transport); |
| 204 | if (!grpc_channel_init_create_stack(exec_ctx, builder, channel_stack_type)) { |
| 205 | grpc_channel_stack_builder_destroy(exec_ctx, builder); |
| 206 | return NULL; |
| 207 | } |
| 208 | return grpc_channel_create_with_builder(exec_ctx, builder, |
| 209 | channel_stack_type); |
| 210 | } |
| 211 | |
Craig Tiller | a6bec8f | 2017-03-14 08:26:04 -0700 | [diff] [blame] | 212 | size_t grpc_channel_get_call_size_estimate(grpc_channel *channel) { |
| 213 | #define ROUND_UP_SIZE 256 |
Craig Tiller | 878a7c2 | 2017-03-23 15:14:48 -0700 | [diff] [blame] | 214 | /* We round up our current estimate to the NEXT value of ROUND_UP_SIZE. |
| 215 | This ensures: |
| 216 | 1. a consistent size allocation when our estimate is drifting slowly |
| 217 | (which is common) - which tends to help most allocators reuse memory |
| 218 | 2. a small amount of allowed growth over the estimate without hitting |
| 219 | the arena size doubling case, reducing overall memory usage */ |
Craig Tiller | b64d652 | 2017-03-23 15:17:18 -0700 | [diff] [blame] | 220 | return ((size_t)gpr_atm_no_barrier_load(&channel->call_size_estimate) + |
| 221 | 2 * ROUND_UP_SIZE) & |
| 222 | ~(size_t)(ROUND_UP_SIZE - 1); |
Craig Tiller | a6bec8f | 2017-03-14 08:26:04 -0700 | [diff] [blame] | 223 | } |
| 224 | |
| 225 | void grpc_channel_update_call_size_estimate(grpc_channel *channel, |
| 226 | size_t size) { |
| 227 | size_t cur = (size_t)gpr_atm_no_barrier_load(&channel->call_size_estimate); |
| 228 | if (cur < size) { |
| 229 | /* size grew: update estimate */ |
| 230 | gpr_atm_no_barrier_cas(&channel->call_size_estimate, (gpr_atm)cur, |
| 231 | (gpr_atm)size); |
| 232 | /* if we lose: never mind, something else will likely update soon enough */ |
| 233 | } else if (cur == size) { |
| 234 | /* no change: holding pattern */ |
| 235 | } else if (cur > 0) { |
| 236 | /* size shrank: decrease estimate */ |
| 237 | gpr_atm_no_barrier_cas( |
| 238 | &channel->call_size_estimate, (gpr_atm)cur, |
| 239 | (gpr_atm)(GPR_MIN(cur - 1, (255 * cur + size) / 256))); |
| 240 | /* if we lose: never mind, something else will likely update soon enough */ |
| 241 | } |
| 242 | } |
| 243 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 244 | char *grpc_channel_get_target(grpc_channel *channel) { |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 245 | GRPC_API_TRACE("grpc_channel_get_target(channel=%p)", 1, (channel)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 246 | return gpr_strdup(channel->target); |
Craig Tiller | 1b22b9d | 2015-07-20 13:42:22 -0700 | [diff] [blame] | 247 | } |
| 248 | |
Mark D. Roth | b2d2488 | 2016-10-27 15:44:07 -0700 | [diff] [blame] | 249 | void grpc_channel_get_info(grpc_channel *channel, |
Mark D. Roth | f79ce7d | 2016-11-04 08:43:36 -0700 | [diff] [blame] | 250 | const grpc_channel_info *channel_info) { |
Mark D. Roth | b2d2488 | 2016-10-27 15:44:07 -0700 | [diff] [blame] | 251 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
| 252 | grpc_channel_element *elem = |
| 253 | grpc_channel_stack_element(CHANNEL_STACK_FROM_CHANNEL(channel), 0); |
| 254 | elem->filter->get_channel_info(&exec_ctx, elem, channel_info); |
| 255 | grpc_exec_ctx_finish(&exec_ctx); |
| 256 | } |
| 257 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 258 | static grpc_call *grpc_channel_create_call_internal( |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 259 | grpc_exec_ctx *exec_ctx, grpc_channel *channel, grpc_call *parent_call, |
| 260 | uint32_t propagation_mask, grpc_completion_queue *cq, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 261 | grpc_pollset_set *pollset_set_alternative, grpc_mdelem path_mdelem, |
| 262 | grpc_mdelem authority_mdelem, gpr_timespec deadline) { |
| 263 | grpc_mdelem send_metadata[2]; |
Craig Tiller | 32ca48c | 2015-09-10 11:47:15 -0700 | [diff] [blame] | 264 | size_t num_metadata = 0; |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 265 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 266 | GPR_ASSERT(channel->is_client); |
David Garcia Quintas | 879b3b9 | 2016-04-25 11:23:38 -0700 | [diff] [blame] | 267 | GPR_ASSERT(!(cq != NULL && pollset_set_alternative != NULL)); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 268 | |
Craig Tiller | c4b56b6 | 2015-07-23 17:44:11 -0700 | [diff] [blame] | 269 | send_metadata[num_metadata++] = path_mdelem; |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 270 | if (!GRPC_MDISNULL(authority_mdelem)) { |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 271 | send_metadata[num_metadata++] = authority_mdelem; |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 272 | } else if (!GRPC_MDISNULL(channel->default_authority)) { |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 273 | send_metadata[num_metadata++] = GRPC_MDELEM_REF(channel->default_authority); |
| 274 | } |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 275 | |
Craig Tiller | 8e21465 | 2016-08-19 09:54:31 -0700 | [diff] [blame] | 276 | grpc_call_create_args args; |
| 277 | memset(&args, 0, sizeof(args)); |
| 278 | args.channel = channel; |
| 279 | args.parent_call = parent_call; |
| 280 | args.propagation_mask = propagation_mask; |
| 281 | args.cq = cq; |
| 282 | args.pollset_set_alternative = pollset_set_alternative; |
| 283 | args.server_transport_data = NULL; |
| 284 | args.add_initial_metadata = send_metadata; |
| 285 | args.add_initial_metadata_count = num_metadata; |
| 286 | args.send_deadline = deadline; |
| 287 | |
| 288 | grpc_call *call; |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 289 | 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] | 290 | return call; |
Craig Tiller | 6902ad2 | 2015-04-16 08:01:49 -0700 | [diff] [blame] | 291 | } |
klempner | c463f74 | 2014-12-19 13:03:35 -0800 | [diff] [blame] | 292 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 293 | grpc_call *grpc_channel_create_call(grpc_channel *channel, |
| 294 | grpc_call *parent_call, |
Craig Tiller | 7536af0 | 2015-12-22 13:49:30 -0800 | [diff] [blame] | 295 | uint32_t propagation_mask, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 296 | grpc_completion_queue *cq, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 297 | grpc_slice method, const grpc_slice *host, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 298 | gpr_timespec deadline, void *reserved) { |
| 299 | GPR_ASSERT(!reserved); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 300 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
| 301 | grpc_call *call = grpc_channel_create_call_internal( |
| 302 | &exec_ctx, channel, parent_call, propagation_mask, cq, NULL, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 303 | grpc_mdelem_from_slices(&exec_ctx, GRPC_MDSTR_PATH, |
| 304 | grpc_slice_ref_internal(method)), |
| 305 | host != NULL ? grpc_mdelem_from_slices(&exec_ctx, GRPC_MDSTR_AUTHORITY, |
| 306 | grpc_slice_ref_internal(*host)) |
| 307 | : GRPC_MDNULL, |
David Garcia Quintas | 4afce7e | 2016-04-18 16:25:17 -0700 | [diff] [blame] | 308 | deadline); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 309 | grpc_exec_ctx_finish(&exec_ctx); |
| 310 | return call; |
David Garcia Quintas | 4afce7e | 2016-04-18 16:25:17 -0700 | [diff] [blame] | 311 | } |
| 312 | |
| 313 | grpc_call *grpc_channel_create_pollset_set_call( |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 314 | grpc_exec_ctx *exec_ctx, grpc_channel *channel, grpc_call *parent_call, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 315 | uint32_t propagation_mask, grpc_pollset_set *pollset_set, grpc_slice method, |
| 316 | const grpc_slice *host, gpr_timespec deadline, void *reserved) { |
David Garcia Quintas | 4afce7e | 2016-04-18 16:25:17 -0700 | [diff] [blame] | 317 | GPR_ASSERT(!reserved); |
| 318 | return grpc_channel_create_call_internal( |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 319 | exec_ctx, channel, parent_call, propagation_mask, NULL, pollset_set, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 320 | grpc_mdelem_from_slices(exec_ctx, GRPC_MDSTR_PATH, |
| 321 | grpc_slice_ref_internal(method)), |
| 322 | host != NULL ? grpc_mdelem_from_slices(exec_ctx, GRPC_MDSTR_AUTHORITY, |
| 323 | grpc_slice_ref_internal(*host)) |
| 324 | : GRPC_MDNULL, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 325 | deadline); |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 326 | } |
| 327 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 328 | void *grpc_channel_register_call(grpc_channel *channel, const char *method, |
| 329 | const char *host, void *reserved) { |
| 330 | registered_call *rc = gpr_malloc(sizeof(registered_call)); |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 331 | GRPC_API_TRACE( |
| 332 | "grpc_channel_register_call(channel=%p, method=%s, host=%s, reserved=%p)", |
| 333 | 4, (channel, method, host, reserved)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 334 | GPR_ASSERT(!reserved); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 335 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 336 | |
| 337 | rc->path = grpc_mdelem_from_slices( |
| 338 | &exec_ctx, GRPC_MDSTR_PATH, |
| 339 | grpc_slice_intern(grpc_slice_from_static_string(method))); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 340 | rc->authority = |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 341 | host ? grpc_mdelem_from_slices( |
| 342 | &exec_ctx, GRPC_MDSTR_AUTHORITY, |
| 343 | grpc_slice_intern(grpc_slice_from_static_string(host))) |
| 344 | : GRPC_MDNULL; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 345 | gpr_mu_lock(&channel->registered_call_mu); |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 346 | rc->next = channel->registered_calls; |
| 347 | channel->registered_calls = rc; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 348 | gpr_mu_unlock(&channel->registered_call_mu); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 349 | grpc_exec_ctx_finish(&exec_ctx); |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 350 | return rc; |
| 351 | } |
| 352 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 353 | grpc_call *grpc_channel_create_registered_call( |
Craig Tiller | 7536af0 | 2015-12-22 13:49:30 -0800 | [diff] [blame] | 354 | grpc_channel *channel, grpc_call *parent_call, uint32_t propagation_mask, |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 355 | grpc_completion_queue *completion_queue, void *registered_call_handle, |
| 356 | gpr_timespec deadline, void *reserved) { |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 357 | registered_call *rc = registered_call_handle; |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 358 | GRPC_API_TRACE( |
| 359 | "grpc_channel_create_registered_call(" |
Craig Tiller | 4de3e4f | 2015-10-05 08:55:50 -0700 | [diff] [blame] | 360 | "channel=%p, parent_call=%p, propagation_mask=%x, completion_queue=%p, " |
| 361 | "registered_call_handle=%p, " |
Ken Payson | 62a6c92 | 2016-06-24 11:53:54 -0700 | [diff] [blame] | 362 | "deadline=gpr_timespec { tv_sec: %" PRId64 |
| 363 | ", tv_nsec: %d, clock_type: %d }, " |
Craig Tiller | 4de3e4f | 2015-10-05 08:55:50 -0700 | [diff] [blame] | 364 | "reserved=%p)", |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 365 | 9, (channel, parent_call, (unsigned)propagation_mask, completion_queue, |
Ken Payson | 62a6c92 | 2016-06-24 11:53:54 -0700 | [diff] [blame] | 366 | registered_call_handle, deadline.tv_sec, deadline.tv_nsec, |
| 367 | (int)deadline.clock_type, reserved)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 368 | GPR_ASSERT(!reserved); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 369 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
| 370 | grpc_call *call = grpc_channel_create_call_internal( |
| 371 | &exec_ctx, channel, parent_call, propagation_mask, completion_queue, NULL, |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 372 | GRPC_MDELEM_REF(rc->path), GRPC_MDELEM_REF(rc->authority), deadline); |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 373 | grpc_exec_ctx_finish(&exec_ctx); |
| 374 | return call; |
Craig Tiller | 0845337 | 2015-04-10 16:05:38 -0700 | [diff] [blame] | 375 | } |
| 376 | |
ncteisen | 9c43fc0 | 2017-06-08 16:06:23 -0700 | [diff] [blame] | 377 | #ifndef NDEBUG |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 378 | #define REF_REASON reason |
| 379 | #define REF_ARG , const char *reason |
Craig Tiller | 9ec2a52 | 2015-05-29 22:46:54 -0700 | [diff] [blame] | 380 | #else |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 381 | #define REF_REASON "" |
| 382 | #define REF_ARG |
Craig Tiller | 9ec2a52 | 2015-05-29 22:46:54 -0700 | [diff] [blame] | 383 | #endif |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 384 | void grpc_channel_internal_ref(grpc_channel *c REF_ARG) { |
| 385 | GRPC_CHANNEL_STACK_REF(CHANNEL_STACK_FROM_CHANNEL(c), REF_REASON); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 386 | } |
| 387 | |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 388 | void grpc_channel_internal_unref(grpc_exec_ctx *exec_ctx, |
| 389 | grpc_channel *c REF_ARG) { |
| 390 | GRPC_CHANNEL_STACK_UNREF(exec_ctx, CHANNEL_STACK_FROM_CHANNEL(c), REF_REASON); |
| 391 | } |
| 392 | |
| 393 | static void destroy_channel(grpc_exec_ctx *exec_ctx, void *arg, |
Craig Tiller | f51457b | 2016-05-03 17:06:32 -0700 | [diff] [blame] | 394 | grpc_error *error) { |
Craig Tiller | 7b43561 | 2015-11-24 08:15:05 -0800 | [diff] [blame] | 395 | grpc_channel *channel = arg; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 396 | grpc_channel_stack_destroy(exec_ctx, CHANNEL_STACK_FROM_CHANNEL(channel)); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 397 | while (channel->registered_calls) { |
| 398 | registered_call *rc = channel->registered_calls; |
| 399 | channel->registered_calls = rc->next; |
Craig Tiller | a59c16c | 2016-10-31 07:25:01 -0700 | [diff] [blame] | 400 | GRPC_MDELEM_UNREF(exec_ctx, rc->path); |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 401 | GRPC_MDELEM_UNREF(exec_ctx, rc->authority); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 402 | gpr_free(rc); |
| 403 | } |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 404 | GRPC_MDELEM_UNREF(exec_ctx, channel->default_authority); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 405 | gpr_mu_destroy(&channel->registered_call_mu); |
| 406 | gpr_free(channel->target); |
| 407 | gpr_free(channel); |
Craig Tiller | 7bd5ab1 | 2015-02-17 22:29:04 -0800 | [diff] [blame] | 408 | } |
| 409 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 410 | void grpc_channel_destroy(grpc_channel *channel) { |
Craig Tiller | e0221ff | 2016-07-11 15:56:08 -0700 | [diff] [blame] | 411 | grpc_transport_op *op = grpc_make_transport_op(NULL); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 412 | grpc_channel_element *elem; |
Craig Tiller | f5768a6 | 2015-09-22 10:54:34 -0700 | [diff] [blame] | 413 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
Masood Malekghassemi | 76c3d74 | 2015-08-19 18:22:53 -0700 | [diff] [blame] | 414 | GRPC_API_TRACE("grpc_channel_destroy(channel=%p)", 1, (channel)); |
ncteisen | 4b36a3d | 2017-03-13 19:08:06 -0700 | [diff] [blame] | 415 | op->disconnect_with_error = |
| 416 | GRPC_ERROR_CREATE_FROM_STATIC_STRING("Channel Destroyed"); |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 417 | elem = grpc_channel_stack_element(CHANNEL_STACK_FROM_CHANNEL(channel), 0); |
Craig Tiller | e0221ff | 2016-07-11 15:56:08 -0700 | [diff] [blame] | 418 | elem->filter->start_transport_op(&exec_ctx, elem, op); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 419 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 420 | GRPC_CHANNEL_INTERNAL_UNREF(&exec_ctx, channel, "channel"); |
Craig Tiller | dfff1b8 | 2015-09-21 14:39:57 -0700 | [diff] [blame] | 421 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 422 | grpc_exec_ctx_finish(&exec_ctx); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 423 | } |
| 424 | |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 425 | grpc_channel_stack *grpc_channel_get_channel_stack(grpc_channel *channel) { |
| 426 | return CHANNEL_STACK_FROM_CHANNEL(channel); |
Nicolas Noble | b7ebd3b | 2014-11-26 16:33:03 -0800 | [diff] [blame] | 427 | } |
| 428 | |
David Garcia Quintas | ac09447 | 2016-05-18 20:25:57 -0700 | [diff] [blame] | 429 | grpc_compression_options grpc_channel_compression_options( |
David Garcia Quintas | 73dcbda | 2016-04-23 00:17:05 -0700 | [diff] [blame] | 430 | const grpc_channel *channel) { |
| 431 | return channel->compression_options; |
| 432 | } |
| 433 | |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 434 | grpc_mdelem grpc_channel_get_reffed_status_elem(grpc_exec_ctx *exec_ctx, |
| 435 | grpc_channel *channel, int i) { |
Craig Tiller | ebdef9d | 2015-11-19 17:09:49 -0800 | [diff] [blame] | 436 | char tmp[GPR_LTOA_MIN_BUFSIZE]; |
| 437 | switch (i) { |
| 438 | case 0: |
| 439 | return GRPC_MDELEM_GRPC_STATUS_0; |
| 440 | case 1: |
| 441 | return GRPC_MDELEM_GRPC_STATUS_1; |
| 442 | case 2: |
| 443 | return GRPC_MDELEM_GRPC_STATUS_2; |
Craig Tiller | a82950e | 2015-09-22 12:33:20 -0700 | [diff] [blame] | 444 | } |
Craig Tiller | ebdef9d | 2015-11-19 17:09:49 -0800 | [diff] [blame] | 445 | gpr_ltoa(i, tmp); |
Craig Tiller | 7c70b6c | 2017-01-23 07:48:42 -0800 | [diff] [blame] | 446 | return grpc_mdelem_from_slices(exec_ctx, GRPC_MDSTR_GRPC_STATUS, |
| 447 | grpc_slice_from_copied_string(tmp)); |
Craig Tiller | 190d360 | 2015-02-18 09:23:38 -0800 | [diff] [blame] | 448 | } |