blob: 06038bb5ba95bbf88f59e14bb8b319c49e3f89a3 [file] [log] [blame]
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001/*
2 *
Craig Tiller6169d5f2016-03-31 07:46:18 -07003 * Copyright 2015, Google Inc.
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08004 * 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
Mark D. Roth2137cd82016-09-14 09:04:00 -070034#include "src/core/ext/client_channel/client_channel.h"
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080035
Mark D. Roth4c0fe492016-08-31 13:51:55 -070036#include <stdbool.h>
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080037#include <stdio.h>
Craig Tillereb3b12e2015-06-26 14:42:49 -070038#include <string.h>
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080039
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080040#include <grpc/support/alloc.h>
41#include <grpc/support/log.h>
Mark D. Rothb2d24882016-10-27 15:44:07 -070042#include <grpc/support/string_util.h>
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080043#include <grpc/support/sync.h>
44#include <grpc/support/useful.h>
45
Mark D. Rothd58a9852017-01-18 08:28:57 -080046#include "src/core/ext/client_channel/http_connect_handshaker.h"
Mark D. Roth15195742016-10-07 09:02:28 -070047#include "src/core/ext/client_channel/lb_policy_registry.h"
Mark D. Rothdc9bee72017-02-07 12:29:14 -080048#include "src/core/ext/client_channel/proxy_mapper_registry.h"
Mark D. Roth21d4b2d2016-11-18 09:53:41 -080049#include "src/core/ext/client_channel/resolver_registry.h"
Mark D. Roth2137cd82016-09-14 09:04:00 -070050#include "src/core/ext/client_channel/subchannel.h"
Craig Tiller9533d042016-03-25 17:11:06 -070051#include "src/core/lib/channel/channel_args.h"
52#include "src/core/lib/channel/connected_channel.h"
Mark D. Roth72f6da82016-09-02 13:42:38 -070053#include "src/core/lib/channel/deadline_filter.h"
Craig Tiller9533d042016-03-25 17:11:06 -070054#include "src/core/lib/iomgr/iomgr.h"
Mark D. Roth4c0fe492016-08-31 13:51:55 -070055#include "src/core/lib/iomgr/polling_entity.h"
Craig Tiller9533d042016-03-25 17:11:06 -070056#include "src/core/lib/profiling/timers.h"
Craig Tiller7c70b6c2017-01-23 07:48:42 -080057#include "src/core/lib/slice/slice_internal.h"
Craig Tiller9533d042016-03-25 17:11:06 -070058#include "src/core/lib/support/string.h"
59#include "src/core/lib/surface/channel.h"
60#include "src/core/lib/transport/connectivity_state.h"
Mark D. Roth9fe284e2016-09-12 11:22:27 -070061#include "src/core/lib/transport/metadata.h"
62#include "src/core/lib/transport/metadata_batch.h"
Mark D. Rothea846a02016-11-03 11:32:54 -070063#include "src/core/lib/transport/service_config.h"
Mark D. Roth9fe284e2016-09-12 11:22:27 -070064#include "src/core/lib/transport/static_metadata.h"
Craig Tiller8910ac62015-10-08 16:49:15 -070065
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080066/* Client channel implementation */
67
Mark D. Roth26b7be42016-10-24 10:08:07 -070068/*************************************************************************
69 * METHOD-CONFIG TABLE
70 */
71
Mark D. Roth9d480942016-10-19 14:18:05 -070072typedef enum {
73 WAIT_FOR_READY_UNSET,
74 WAIT_FOR_READY_FALSE,
75 WAIT_FOR_READY_TRUE
76} wait_for_ready_value;
77
78typedef struct method_parameters {
79 gpr_timespec timeout;
80 wait_for_ready_value wait_for_ready;
81} method_parameters;
82
83static void *method_parameters_copy(void *value) {
84 void *new_value = gpr_malloc(sizeof(method_parameters));
85 memcpy(new_value, value, sizeof(method_parameters));
86 return new_value;
87}
88
Craig Tillerb28c7e82016-11-18 10:29:04 -080089static void method_parameters_free(grpc_exec_ctx *exec_ctx, void *p) {
Craig Tiller87a7e1f2016-11-09 09:42:19 -080090 gpr_free(p);
91}
92
Craig Tiller7c70b6c2017-01-23 07:48:42 -080093static const grpc_slice_hash_table_vtable method_parameters_vtable = {
Craig Tillerb28c7e82016-11-18 10:29:04 -080094 method_parameters_free, method_parameters_copy};
Mark D. Roth9d480942016-10-19 14:18:05 -070095
Mark D. Rothe30baeb2016-11-03 08:16:19 -070096static void *method_parameters_create_from_json(const grpc_json *json) {
Mark D. Rothc968e602016-11-02 14:07:36 -070097 wait_for_ready_value wait_for_ready = WAIT_FOR_READY_UNSET;
Mark D. Roth47f10842016-11-03 08:45:27 -070098 gpr_timespec timeout = {0, 0, GPR_TIMESPAN};
99 for (grpc_json *field = json->child; field != NULL; field = field->next) {
Mark D. Rothc968e602016-11-02 14:07:36 -0700100 if (field->key == NULL) continue;
Mark D. Roth84c8a022016-11-10 09:39:34 -0800101 if (strcmp(field->key, "waitForReady") == 0) {
Mark D. Rothc968e602016-11-02 14:07:36 -0700102 if (wait_for_ready != WAIT_FOR_READY_UNSET) return NULL; // Duplicate.
103 if (field->type != GRPC_JSON_TRUE && field->type != GRPC_JSON_FALSE) {
104 return NULL;
105 }
Mark D. Roth47f10842016-11-03 08:45:27 -0700106 wait_for_ready = field->type == GRPC_JSON_TRUE ? WAIT_FOR_READY_TRUE
107 : WAIT_FOR_READY_FALSE;
Mark D. Rothc968e602016-11-02 14:07:36 -0700108 } else if (strcmp(field->key, "timeout") == 0) {
109 if (timeout.tv_sec > 0 || timeout.tv_nsec > 0) return NULL; // Duplicate.
Mark D. Roth84c8a022016-11-10 09:39:34 -0800110 if (field->type != GRPC_JSON_STRING) return NULL;
111 size_t len = strlen(field->value);
112 if (field->value[len - 1] != 's') return NULL;
Mark D. Rothc19049c2016-11-10 09:43:06 -0800113 char *buf = gpr_strdup(field->value);
Mark D. Roth84c8a022016-11-10 09:39:34 -0800114 buf[len - 1] = '\0'; // Remove trailing 's'.
Mark D. Rothc19049c2016-11-10 09:43:06 -0800115 char *decimal_point = strchr(buf, '.');
Mark D. Roth84c8a022016-11-10 09:39:34 -0800116 if (decimal_point != NULL) {
117 *decimal_point = '\0';
118 timeout.tv_nsec = gpr_parse_nonnegative_int(decimal_point + 1);
119 if (timeout.tv_nsec == -1) {
120 gpr_free(buf);
Mark D. Rothc968e602016-11-02 14:07:36 -0700121 return NULL;
122 }
Mark D. Roth84c8a022016-11-10 09:39:34 -0800123 // There should always be exactly 3, 6, or 9 fractional digits.
124 int multiplier = 1;
125 switch (strlen(decimal_point + 1)) {
126 case 9:
127 break;
128 case 6:
129 multiplier *= 1000;
130 break;
131 case 3:
132 multiplier *= 1000000;
133 break;
134 default: // Unsupported number of digits.
135 gpr_free(buf);
136 return NULL;
137 }
138 timeout.tv_nsec *= multiplier;
Mark D. Rothc968e602016-11-02 14:07:36 -0700139 }
Mark D. Roth84c8a022016-11-10 09:39:34 -0800140 timeout.tv_sec = gpr_parse_nonnegative_int(buf);
141 if (timeout.tv_sec == -1) return NULL;
142 gpr_free(buf);
Mark D. Rothc968e602016-11-02 14:07:36 -0700143 }
144 }
Mark D. Roth9d480942016-10-19 14:18:05 -0700145 method_parameters *value = gpr_malloc(sizeof(method_parameters));
Mark D. Rothc968e602016-11-02 14:07:36 -0700146 value->timeout = timeout;
147 value->wait_for_ready = wait_for_ready;
Mark D. Roth9d480942016-10-19 14:18:05 -0700148 return value;
149}
150
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700151/*************************************************************************
152 * CHANNEL-WIDE FUNCTIONS
153 */
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -0800154
Craig Tiller800dacb2015-10-06 09:10:26 -0700155typedef struct client_channel_channel_data {
Craig Tillerf5f17122015-06-25 08:47:26 -0700156 /** resolver for this channel */
157 grpc_resolver *resolver;
Craig Tiller20a3c352015-08-05 08:39:50 -0700158 /** have we started resolving this channel */
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700159 bool started_resolving;
Mark D. Roth0e48a9a2016-09-08 14:14:39 -0700160 /** client channel factory */
161 grpc_client_channel_factory *client_channel_factory;
Craig Tillerf5f17122015-06-25 08:47:26 -0700162
Mark D. Roth046cf762016-09-26 11:13:51 -0700163 /** mutex protecting all variables below in this data structure */
Mark D. Rothff4df062016-08-22 15:02:49 -0700164 gpr_mu mu;
Mark D. Roth046cf762016-09-26 11:13:51 -0700165 /** currently active load balancer */
Mark D. Roth78afd772016-11-04 12:49:49 -0700166 char *lb_policy_name;
Craig Tillerf5f17122015-06-25 08:47:26 -0700167 grpc_lb_policy *lb_policy;
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800168 /** service config in JSON form */
169 char *service_config_json;
Mark D. Roth9d480942016-10-19 14:18:05 -0700170 /** maps method names to method_parameters structs */
Craig Tiller7c70b6c2017-01-23 07:48:42 -0800171 grpc_slice_hash_table *method_params_table;
Mark D. Roth046cf762016-09-26 11:13:51 -0700172 /** incoming resolver result - set by resolver.next() */
Mark D. Rothaf842452016-10-21 15:05:15 -0700173 grpc_channel_args *resolver_result;
Craig Tiller3f475422015-06-25 10:43:05 -0700174 /** a list of closures that are all waiting for config to come in */
Craig Tillerd9ccbbf2015-09-22 09:30:00 -0700175 grpc_closure_list waiting_for_config_closures;
Craig Tiller3f475422015-06-25 10:43:05 -0700176 /** resolver callback */
Mark D. Rothff4df062016-08-22 15:02:49 -0700177 grpc_closure on_resolver_result_changed;
Craig Tiller3f475422015-06-25 10:43:05 -0700178 /** connectivity state being tracked */
Craig Tillerca3e9d32015-06-27 18:37:27 -0700179 grpc_connectivity_state_tracker state_tracker;
Craig Tiller48cb07c2015-07-15 16:16:15 -0700180 /** when an lb_policy arrives, should we try to exit idle */
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700181 bool exit_idle_when_lb_policy_arrives;
Craig Tiller906e3bc2015-11-24 07:31:31 -0800182 /** owning stack */
183 grpc_channel_stack *owning_stack;
Craig Tiller69b093b2016-02-25 19:04:07 -0800184 /** interested parties (owned) */
185 grpc_pollset_set *interested_parties;
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -0800186} channel_data;
187
Craig Tillerd6c98df2015-08-18 09:33:44 -0700188/** We create one watcher for each new lb_policy that is returned from a
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700189 resolver, to watch for state changes from the lb_policy. When a state
190 change is seen, we update the channel, and create a new watcher. */
Craig Tillera82950e2015-09-22 12:33:20 -0700191typedef struct {
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700192 channel_data *chand;
Craig Tiller33825112015-09-18 07:44:19 -0700193 grpc_closure on_changed;
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700194 grpc_connectivity_state state;
195 grpc_lb_policy *lb_policy;
196} lb_policy_connectivity_watcher;
197
Craig Tillera82950e2015-09-22 12:33:20 -0700198static void watch_lb_policy(grpc_exec_ctx *exec_ctx, channel_data *chand,
199 grpc_lb_policy *lb_policy,
200 grpc_connectivity_state current_state);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700201
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800202static void set_channel_connectivity_state_locked(grpc_exec_ctx *exec_ctx,
203 channel_data *chand,
204 grpc_connectivity_state state,
Craig Tiller804ff712016-05-05 16:25:40 -0700205 grpc_error *error,
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800206 const char *reason) {
207 if ((state == GRPC_CHANNEL_TRANSIENT_FAILURE ||
Craig Tiller48ed92e2016-06-02 11:07:12 -0700208 state == GRPC_CHANNEL_SHUTDOWN) &&
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800209 chand->lb_policy != NULL) {
Mark D. Roth59c9f902016-09-28 13:33:21 -0700210 /* cancel picks with wait_for_ready=false */
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800211 grpc_lb_policy_cancel_picks(
212 exec_ctx, chand->lb_policy,
Mark D. Roth59c9f902016-09-28 13:33:21 -0700213 /* mask= */ GRPC_INITIAL_METADATA_WAIT_FOR_READY,
Mark D. Roth58f52b72016-09-09 13:55:18 -0700214 /* check= */ 0, GRPC_ERROR_REF(error));
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800215 }
Craig Tiller9ccf5f12016-05-07 21:41:01 -0700216 grpc_connectivity_state_set(exec_ctx, &chand->state_tracker, state, error,
217 reason);
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800218}
219
Craig Tiller804ff712016-05-05 16:25:40 -0700220static void on_lb_policy_state_changed_locked(grpc_exec_ctx *exec_ctx,
221 lb_policy_connectivity_watcher *w,
222 grpc_error *error) {
Craig Tillercb2609f2015-11-24 17:19:19 -0800223 grpc_connectivity_state publish_state = w->state;
Craig Tiller5795da72015-09-17 15:27:13 -0700224 /* check if the notification is for a stale policy */
Craig Tillera82950e2015-09-22 12:33:20 -0700225 if (w->lb_policy != w->chand->lb_policy) return;
Craig Tiller5795da72015-09-17 15:27:13 -0700226
Craig Tiller48ed92e2016-06-02 11:07:12 -0700227 if (publish_state == GRPC_CHANNEL_SHUTDOWN && w->chand->resolver != NULL) {
Craig Tillercb2609f2015-11-24 17:19:19 -0800228 publish_state = GRPC_CHANNEL_TRANSIENT_FAILURE;
229 grpc_resolver_channel_saw_error(exec_ctx, w->chand->resolver);
Craig Tillerf62c4d52015-12-04 07:43:07 -0800230 GRPC_LB_POLICY_UNREF(exec_ctx, w->chand->lb_policy, "channel");
231 w->chand->lb_policy = NULL;
Craig Tillercb2609f2015-11-24 17:19:19 -0800232 }
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800233 set_channel_connectivity_state_locked(exec_ctx, w->chand, publish_state,
Craig Tillerfc353d62016-05-10 12:58:03 -0700234 GRPC_ERROR_REF(error), "lb_changed");
Craig Tiller48ed92e2016-06-02 11:07:12 -0700235 if (w->state != GRPC_CHANNEL_SHUTDOWN) {
Craig Tillera82950e2015-09-22 12:33:20 -0700236 watch_lb_policy(exec_ctx, w->chand, w->lb_policy, w->state);
237 }
Craig Tiller5795da72015-09-17 15:27:13 -0700238}
239
Craig Tillera82950e2015-09-22 12:33:20 -0700240static void on_lb_policy_state_changed(grpc_exec_ctx *exec_ctx, void *arg,
Craig Tiller804ff712016-05-05 16:25:40 -0700241 grpc_error *error) {
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700242 lb_policy_connectivity_watcher *w = arg;
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700243
Mark D. Rothff4df062016-08-22 15:02:49 -0700244 gpr_mu_lock(&w->chand->mu);
Craig Tiller804ff712016-05-05 16:25:40 -0700245 on_lb_policy_state_changed_locked(exec_ctx, w, error);
Mark D. Rothff4df062016-08-22 15:02:49 -0700246 gpr_mu_unlock(&w->chand->mu);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700247
Craig Tiller906e3bc2015-11-24 07:31:31 -0800248 GRPC_CHANNEL_STACK_UNREF(exec_ctx, w->chand->owning_stack, "watch_lb_policy");
Craig Tillera82950e2015-09-22 12:33:20 -0700249 gpr_free(w);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700250}
251
Craig Tillera82950e2015-09-22 12:33:20 -0700252static void watch_lb_policy(grpc_exec_ctx *exec_ctx, channel_data *chand,
253 grpc_lb_policy *lb_policy,
254 grpc_connectivity_state current_state) {
255 lb_policy_connectivity_watcher *w = gpr_malloc(sizeof(*w));
Craig Tiller906e3bc2015-11-24 07:31:31 -0800256 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "watch_lb_policy");
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700257
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700258 w->chand = chand;
Craig Tiller91031da2016-12-28 15:44:25 -0800259 grpc_closure_init(&w->on_changed, on_lb_policy_state_changed, w,
260 grpc_schedule_on_exec_ctx);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700261 w->state = current_state;
262 w->lb_policy = lb_policy;
Craig Tillera82950e2015-09-22 12:33:20 -0700263 grpc_lb_policy_notify_on_state_change(exec_ctx, lb_policy, &w->state,
264 &w->on_changed);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700265}
266
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700267static void on_resolver_result_changed(grpc_exec_ctx *exec_ctx, void *arg,
268 grpc_error *error) {
Craig Tiller3f475422015-06-25 10:43:05 -0700269 channel_data *chand = arg;
Mark D. Rothb2d24882016-10-27 15:44:07 -0700270 char *lb_policy_name = NULL;
Craig Tiller3f475422015-06-25 10:43:05 -0700271 grpc_lb_policy *lb_policy = NULL;
272 grpc_lb_policy *old_lb_policy;
Craig Tiller7c70b6c2017-01-23 07:48:42 -0800273 grpc_slice_hash_table *method_params_table = NULL;
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700274 grpc_connectivity_state state = GRPC_CHANNEL_TRANSIENT_FAILURE;
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700275 bool exit_idle = false;
Craig Tiller804ff712016-05-05 16:25:40 -0700276 grpc_error *state_error = GRPC_ERROR_CREATE("No load balancing policy");
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800277 char *service_config_json = NULL;
Craig Tiller3f475422015-06-25 10:43:05 -0700278
Mark D. Roth046cf762016-09-26 11:13:51 -0700279 if (chand->resolver_result != NULL) {
Mark D. Roth5bd7be02016-10-21 14:19:50 -0700280 // Find LB policy name.
Mark D. Rothaf842452016-10-21 15:05:15 -0700281 const grpc_arg *channel_arg =
Mark D. Roth41124992016-11-03 11:22:20 -0700282 grpc_channel_args_find(chand->resolver_result, GRPC_ARG_LB_POLICY_NAME);
Mark D. Rothaf842452016-10-21 15:05:15 -0700283 if (channel_arg != NULL) {
284 GPR_ASSERT(channel_arg->type == GRPC_ARG_STRING);
285 lb_policy_name = channel_arg->value.string;
Mark D. Roth5bd7be02016-10-21 14:19:50 -0700286 }
Mark D. Roth88405f72016-10-03 08:24:52 -0700287 // Special case: If all of the addresses are balancer addresses,
288 // assume that we should use the grpclb policy, regardless of what the
289 // resolver actually specified.
Mark D. Rothaf842452016-10-21 15:05:15 -0700290 channel_arg =
Mark D. Roth41124992016-11-03 11:22:20 -0700291 grpc_channel_args_find(chand->resolver_result, GRPC_ARG_LB_ADDRESSES);
Mark D. Rothaf842452016-10-21 15:05:15 -0700292 if (channel_arg != NULL) {
293 GPR_ASSERT(channel_arg->type == GRPC_ARG_POINTER);
Mark D. Roth557c9902016-10-24 11:12:05 -0700294 grpc_lb_addresses *addresses = channel_arg->value.pointer.p;
Mark D. Rothaf842452016-10-21 15:05:15 -0700295 bool found_backend_address = false;
296 for (size_t i = 0; i < addresses->num_addresses; ++i) {
297 if (!addresses->addresses[i].is_balancer) {
298 found_backend_address = true;
299 break;
300 }
Mark D. Roth88405f72016-10-03 08:24:52 -0700301 }
Mark D. Rothaf842452016-10-21 15:05:15 -0700302 if (!found_backend_address) {
303 if (lb_policy_name != NULL && strcmp(lb_policy_name, "grpclb") != 0) {
304 gpr_log(GPR_INFO,
305 "resolver requested LB policy %s but provided only balancer "
306 "addresses, no backend addresses -- forcing use of grpclb LB "
307 "policy",
Mark D. Roth5f40e5d2016-10-24 13:09:05 -0700308 lb_policy_name);
Mark D. Rothaf842452016-10-21 15:05:15 -0700309 }
310 lb_policy_name = "grpclb";
Mark D. Roth88405f72016-10-03 08:24:52 -0700311 }
Mark D. Roth88405f72016-10-03 08:24:52 -0700312 }
313 // Use pick_first if nothing was specified and we didn't select grpclb
314 // above.
315 if (lb_policy_name == NULL) lb_policy_name = "pick_first";
Mark D. Roth41124992016-11-03 11:22:20 -0700316 // Instantiate LB policy.
317 grpc_lb_policy_args lb_policy_args;
318 lb_policy_args.args = chand->resolver_result;
319 lb_policy_args.client_channel_factory = chand->client_channel_factory;
Mark D. Roth88405f72016-10-03 08:24:52 -0700320 lb_policy =
321 grpc_lb_policy_create(exec_ctx, lb_policy_name, &lb_policy_args);
Craig Tillera82950e2015-09-22 12:33:20 -0700322 if (lb_policy != NULL) {
Craig Tillera82950e2015-09-22 12:33:20 -0700323 GRPC_LB_POLICY_REF(lb_policy, "config_change");
Craig Tillerf707d622016-05-06 14:26:12 -0700324 GRPC_ERROR_UNREF(state_error);
Craig Tiller804ff712016-05-05 16:25:40 -0700325 state =
326 grpc_lb_policy_check_connectivity(exec_ctx, lb_policy, &state_error);
Craig Tiller45724b32015-09-22 10:42:19 -0700327 }
Mark D. Roth41124992016-11-03 11:22:20 -0700328 // Find service config.
Mark D. Rothaf842452016-10-21 15:05:15 -0700329 channel_arg =
Mark D. Roth41124992016-11-03 11:22:20 -0700330 grpc_channel_args_find(chand->resolver_result, GRPC_ARG_SERVICE_CONFIG);
Mark D. Roth046cf762016-09-26 11:13:51 -0700331 if (channel_arg != NULL) {
Mark D. Roth9ec28af2016-11-03 12:32:39 -0700332 GPR_ASSERT(channel_arg->type == GRPC_ARG_STRING);
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800333 service_config_json = gpr_strdup(channel_arg->value.string);
Mark D. Roth70a1abd2016-11-04 09:26:37 -0700334 grpc_service_config *service_config =
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800335 grpc_service_config_create(service_config_json);
Mark D. Rothbdc58b22016-11-04 09:25:57 -0700336 if (service_config != NULL) {
337 method_params_table = grpc_service_config_create_method_config_table(
Craig Tillerb28c7e82016-11-18 10:29:04 -0800338 exec_ctx, service_config, method_parameters_create_from_json,
Mark D. Rothbdc58b22016-11-04 09:25:57 -0700339 &method_parameters_vtable);
340 grpc_service_config_destroy(service_config);
341 }
Mark D. Roth9fe284e2016-09-12 11:22:27 -0700342 }
Mark D. Rothf79ce7d2016-11-04 08:43:36 -0700343 // Before we clean up, save a copy of lb_policy_name, since it might
344 // be pointing to data inside chand->resolver_result.
345 // The copy will be saved in chand->lb_policy_name below.
346 lb_policy_name = gpr_strdup(lb_policy_name);
Craig Tiller87a7e1f2016-11-09 09:42:19 -0800347 grpc_channel_args_destroy(exec_ctx, chand->resolver_result);
Mark D. Roth046cf762016-09-26 11:13:51 -0700348 chand->resolver_result = NULL;
Craig Tillera82950e2015-09-22 12:33:20 -0700349 }
350
Craig Tiller86c99582015-11-25 15:22:26 -0800351 if (lb_policy != NULL) {
Craig Tiller69b093b2016-02-25 19:04:07 -0800352 grpc_pollset_set_add_pollset_set(exec_ctx, lb_policy->interested_parties,
353 chand->interested_parties);
Craig Tiller86c99582015-11-25 15:22:26 -0800354 }
355
Mark D. Rothff4df062016-08-22 15:02:49 -0700356 gpr_mu_lock(&chand->mu);
Mark D. Rothb2d24882016-10-27 15:44:07 -0700357 if (lb_policy_name != NULL) {
358 gpr_free(chand->lb_policy_name);
359 chand->lb_policy_name = lb_policy_name;
360 }
Craig Tiller3f475422015-06-25 10:43:05 -0700361 old_lb_policy = chand->lb_policy;
362 chand->lb_policy = lb_policy;
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800363 if (service_config_json != NULL) {
364 gpr_free(chand->service_config_json);
365 chand->service_config_json = service_config_json;
366 }
Mark D. Roth9d480942016-10-19 14:18:05 -0700367 if (chand->method_params_table != NULL) {
Craig Tiller7c70b6c2017-01-23 07:48:42 -0800368 grpc_slice_hash_table_unref(exec_ctx, chand->method_params_table);
Mark D. Roth046cf762016-09-26 11:13:51 -0700369 }
Mark D. Roth9d480942016-10-19 14:18:05 -0700370 chand->method_params_table = method_params_table;
Craig Tiller0ede5452016-04-23 12:21:45 -0700371 if (lb_policy != NULL) {
Craig Tiller91031da2016-12-28 15:44:25 -0800372 grpc_closure_list_sched(exec_ctx, &chand->waiting_for_config_closures);
Craig Tiller0ede5452016-04-23 12:21:45 -0700373 } else if (chand->resolver == NULL /* disconnected */) {
Craig Tiller804ff712016-05-05 16:25:40 -0700374 grpc_closure_list_fail_all(
375 &chand->waiting_for_config_closures,
376 GRPC_ERROR_CREATE_REFERENCING("Channel disconnected", &error, 1));
Craig Tiller91031da2016-12-28 15:44:25 -0800377 grpc_closure_list_sched(exec_ctx, &chand->waiting_for_config_closures);
Craig Tillera82950e2015-09-22 12:33:20 -0700378 }
379 if (lb_policy != NULL && chand->exit_idle_when_lb_policy_arrives) {
380 GRPC_LB_POLICY_REF(lb_policy, "exit_idle");
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700381 exit_idle = true;
382 chand->exit_idle_when_lb_policy_arrives = false;
Craig Tillera82950e2015-09-22 12:33:20 -0700383 }
Craig Tiller98465032015-06-29 14:36:42 -0700384
Craig Tiller804ff712016-05-05 16:25:40 -0700385 if (error == GRPC_ERROR_NONE && chand->resolver) {
Craig Tiller9ccf5f12016-05-07 21:41:01 -0700386 set_channel_connectivity_state_locked(
387 exec_ctx, chand, state, GRPC_ERROR_REF(state_error), "new_lb+resolver");
Craig Tillera82950e2015-09-22 12:33:20 -0700388 if (lb_policy != NULL) {
389 watch_lb_policy(exec_ctx, chand, lb_policy, state);
Craig Tiller45724b32015-09-22 10:42:19 -0700390 }
Craig Tiller906e3bc2015-11-24 07:31:31 -0800391 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "resolver");
Mark D. Roth046cf762016-09-26 11:13:51 -0700392 grpc_resolver_next(exec_ctx, chand->resolver, &chand->resolver_result,
Mark D. Rothff4df062016-08-22 15:02:49 -0700393 &chand->on_resolver_result_changed);
394 gpr_mu_unlock(&chand->mu);
Craig Tillera82950e2015-09-22 12:33:20 -0700395 } else {
Craig Tiller76a5c0e2016-03-09 09:05:30 -0800396 if (chand->resolver != NULL) {
397 grpc_resolver_shutdown(exec_ctx, chand->resolver);
398 GRPC_RESOLVER_UNREF(exec_ctx, chand->resolver, "channel");
399 chand->resolver = NULL;
400 }
Craig Tiller804ff712016-05-05 16:25:40 -0700401 grpc_error *refs[] = {error, state_error};
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800402 set_channel_connectivity_state_locked(
Craig Tillerd925c932016-06-06 08:38:50 -0700403 exec_ctx, chand, GRPC_CHANNEL_SHUTDOWN,
Craig Tiller804ff712016-05-05 16:25:40 -0700404 GRPC_ERROR_CREATE_REFERENCING("Got config after disconnection", refs,
405 GPR_ARRAY_SIZE(refs)),
406 "resolver_gone");
Mark D. Rothff4df062016-08-22 15:02:49 -0700407 gpr_mu_unlock(&chand->mu);
Craig Tillera82950e2015-09-22 12:33:20 -0700408 }
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700409
Craig Tillera82950e2015-09-22 12:33:20 -0700410 if (exit_idle) {
411 grpc_lb_policy_exit_idle(exec_ctx, lb_policy);
412 GRPC_LB_POLICY_UNREF(exec_ctx, lb_policy, "exit_idle");
413 }
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700414
Craig Tillera82950e2015-09-22 12:33:20 -0700415 if (old_lb_policy != NULL) {
Craig Tiller69b093b2016-02-25 19:04:07 -0800416 grpc_pollset_set_del_pollset_set(
417 exec_ctx, old_lb_policy->interested_parties, chand->interested_parties);
Craig Tillera82950e2015-09-22 12:33:20 -0700418 GRPC_LB_POLICY_UNREF(exec_ctx, old_lb_policy, "channel");
419 }
Craig Tiller000cd8f2015-09-18 07:20:29 -0700420
Craig Tillera82950e2015-09-22 12:33:20 -0700421 if (lb_policy != NULL) {
422 GRPC_LB_POLICY_UNREF(exec_ctx, lb_policy, "config_change");
423 }
Craig Tiller45724b32015-09-22 10:42:19 -0700424
Craig Tiller906e3bc2015-11-24 07:31:31 -0800425 GRPC_CHANNEL_STACK_UNREF(exec_ctx, chand->owning_stack, "resolver");
Craig Tiller9ccf5f12016-05-07 21:41:01 -0700426 GRPC_ERROR_UNREF(state_error);
Craig Tiller3f475422015-06-25 10:43:05 -0700427}
428
Craig Tillera82950e2015-09-22 12:33:20 -0700429static void cc_start_transport_op(grpc_exec_ctx *exec_ctx,
430 grpc_channel_element *elem,
431 grpc_transport_op *op) {
Craig Tillerca3e9d32015-06-27 18:37:27 -0700432 channel_data *chand = elem->channel_data;
Craig Tiller000cd8f2015-09-18 07:20:29 -0700433
Craig Tiller91031da2016-12-28 15:44:25 -0800434 grpc_closure_sched(exec_ctx, op->on_consumed, GRPC_ERROR_NONE);
Craig Tillerca3e9d32015-06-27 18:37:27 -0700435
Craig Tillerd7f12e32016-03-03 10:08:31 -0800436 GPR_ASSERT(op->set_accept_stream == false);
Craig Tiller28bf8912015-12-07 16:07:04 -0800437 if (op->bind_pollset != NULL) {
Craig Tiller69b093b2016-02-25 19:04:07 -0800438 grpc_pollset_set_add_pollset(exec_ctx, chand->interested_parties,
Craig Tillere2c62372015-12-07 16:11:03 -0800439 op->bind_pollset);
Craig Tiller28bf8912015-12-07 16:07:04 -0800440 }
Craig Tillerca3e9d32015-06-27 18:37:27 -0700441
Mark D. Rothff4df062016-08-22 15:02:49 -0700442 gpr_mu_lock(&chand->mu);
Craig Tillera82950e2015-09-22 12:33:20 -0700443 if (op->on_connectivity_state_change != NULL) {
444 grpc_connectivity_state_notify_on_state_change(
445 exec_ctx, &chand->state_tracker, op->connectivity_state,
446 op->on_connectivity_state_change);
447 op->on_connectivity_state_change = NULL;
448 op->connectivity_state = NULL;
449 }
450
Craig Tiller26dab312015-12-07 14:43:47 -0800451 if (op->send_ping != NULL) {
Craig Tiller87b71e22015-12-07 15:14:14 -0800452 if (chand->lb_policy == NULL) {
Craig Tiller91031da2016-12-28 15:44:25 -0800453 grpc_closure_sched(exec_ctx, op->send_ping,
454 GRPC_ERROR_CREATE("Ping with no load balancing"));
Craig Tiller26dab312015-12-07 14:43:47 -0800455 } else {
Craig Tiller28bf8912015-12-07 16:07:04 -0800456 grpc_lb_policy_ping_one(exec_ctx, chand->lb_policy, op->send_ping);
Craig Tiller26dab312015-12-07 14:43:47 -0800457 op->bind_pollset = NULL;
458 }
459 op->send_ping = NULL;
460 }
461
Craig Tiller1c51edc2016-05-07 16:18:43 -0700462 if (op->disconnect_with_error != GRPC_ERROR_NONE) {
463 if (chand->resolver != NULL) {
464 set_channel_connectivity_state_locked(
Craig Tillerd925c932016-06-06 08:38:50 -0700465 exec_ctx, chand, GRPC_CHANNEL_SHUTDOWN,
Craig Tiller1c51edc2016-05-07 16:18:43 -0700466 GRPC_ERROR_REF(op->disconnect_with_error), "disconnect");
467 grpc_resolver_shutdown(exec_ctx, chand->resolver);
468 GRPC_RESOLVER_UNREF(exec_ctx, chand->resolver, "channel");
469 chand->resolver = NULL;
470 if (!chand->started_resolving) {
471 grpc_closure_list_fail_all(&chand->waiting_for_config_closures,
472 GRPC_ERROR_REF(op->disconnect_with_error));
Craig Tiller91031da2016-12-28 15:44:25 -0800473 grpc_closure_list_sched(exec_ctx, &chand->waiting_for_config_closures);
Craig Tiller1c51edc2016-05-07 16:18:43 -0700474 }
475 if (chand->lb_policy != NULL) {
476 grpc_pollset_set_del_pollset_set(exec_ctx,
477 chand->lb_policy->interested_parties,
478 chand->interested_parties);
479 GRPC_LB_POLICY_UNREF(exec_ctx, chand->lb_policy, "channel");
480 chand->lb_policy = NULL;
481 }
Craig Tillerb12d22a2016-04-23 12:50:21 -0700482 }
Craig Tiller1c51edc2016-05-07 16:18:43 -0700483 GRPC_ERROR_UNREF(op->disconnect_with_error);
Craig Tillera82950e2015-09-22 12:33:20 -0700484 }
Mark D. Rothff4df062016-08-22 15:02:49 -0700485 gpr_mu_unlock(&chand->mu);
Craig Tillerca3e9d32015-06-27 18:37:27 -0700486}
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -0800487
Mark D. Rothb2d24882016-10-27 15:44:07 -0700488static void cc_get_channel_info(grpc_exec_ctx *exec_ctx,
489 grpc_channel_element *elem,
Mark D. Rothf79ce7d2016-11-04 08:43:36 -0700490 const grpc_channel_info *info) {
Mark D. Rothb2d24882016-10-27 15:44:07 -0700491 channel_data *chand = elem->channel_data;
492 gpr_mu_lock(&chand->mu);
493 if (info->lb_policy_name != NULL) {
494 *info->lb_policy_name = chand->lb_policy_name == NULL
Mark D. Roth78afd772016-11-04 12:49:49 -0700495 ? NULL
496 : gpr_strdup(chand->lb_policy_name);
Mark D. Rothb2d24882016-10-27 15:44:07 -0700497 }
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800498 if (info->service_config_json != NULL) {
499 *info->service_config_json = chand->service_config_json == NULL
500 ? NULL
501 : gpr_strdup(chand->service_config_json);
502 }
Mark D. Rothb2d24882016-10-27 15:44:07 -0700503 gpr_mu_unlock(&chand->mu);
504}
505
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700506/* Constructor for channel_data */
Mark D. Rothc1087882016-11-18 10:54:45 -0800507static grpc_error *cc_init_channel_elem(grpc_exec_ctx *exec_ctx,
Mark D. Roth5e2566e2016-11-18 10:53:13 -0800508 grpc_channel_element *elem,
509 grpc_channel_element_args *args) {
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700510 channel_data *chand = elem->channel_data;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700511 memset(chand, 0, sizeof(*chand));
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700512 GPR_ASSERT(args->is_last);
513 GPR_ASSERT(elem->filter == &grpc_client_channel_filter);
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800514 // Initialize data members.
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700515 gpr_mu_init(&chand->mu);
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800516 chand->owning_stack = args->channel_stack;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700517 grpc_closure_init(&chand->on_resolver_result_changed,
Craig Tiller91031da2016-12-28 15:44:25 -0800518 on_resolver_result_changed, chand,
519 grpc_schedule_on_exec_ctx);
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800520 chand->interested_parties = grpc_pollset_set_create();
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700521 grpc_connectivity_state_init(&chand->state_tracker, GRPC_CHANNEL_IDLE,
522 "client_channel");
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800523 // Record client channel factory.
524 const grpc_arg *arg = grpc_channel_args_find(args->channel_args,
525 GRPC_ARG_CLIENT_CHANNEL_FACTORY);
526 GPR_ASSERT(arg != NULL);
527 GPR_ASSERT(arg->type == GRPC_ARG_POINTER);
528 grpc_client_channel_factory_ref(arg->value.pointer.p);
529 chand->client_channel_factory = arg->value.pointer.p;
Mark D. Rothdc9bee72017-02-07 12:29:14 -0800530 // Get server name to resolve, using proxy mapper if needed.
Mark D. Roth86e90592016-11-18 09:56:40 -0800531 arg = grpc_channel_args_find(args->channel_args, GRPC_ARG_SERVER_URI);
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800532 GPR_ASSERT(arg != NULL);
533 GPR_ASSERT(arg->type == GRPC_ARG_STRING);
Mark D. Rothdc9bee72017-02-07 12:29:14 -0800534 char *proxy_name = NULL;
535 grpc_channel_args *new_args = NULL;
536 grpc_proxy_mappers_map_name(exec_ctx, arg->value.string, args->channel_args,
537 &proxy_name, &new_args);
538 // Instantiate resolver.
Mark D. Roth45ccec52017-01-18 14:04:01 -0800539 chand->resolver = grpc_resolver_create(
Mark D. Rothdc9bee72017-02-07 12:29:14 -0800540 exec_ctx, proxy_name != NULL ? proxy_name : arg->value.string,
541 new_args != NULL ? new_args : args->channel_args,
542 chand->interested_parties);
543 if (proxy_name != NULL) gpr_free(proxy_name);
544 if (new_args != NULL) grpc_channel_args_destroy(exec_ctx, new_args);
Mark D. Roth5e2566e2016-11-18 10:53:13 -0800545 if (chand->resolver == NULL) {
546 return GRPC_ERROR_CREATE("resolver creation failed");
547 }
548 return GRPC_ERROR_NONE;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700549}
550
551/* Destructor for channel_data */
552static void cc_destroy_channel_elem(grpc_exec_ctx *exec_ctx,
553 grpc_channel_element *elem) {
554 channel_data *chand = elem->channel_data;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700555 if (chand->resolver != NULL) {
556 grpc_resolver_shutdown(exec_ctx, chand->resolver);
557 GRPC_RESOLVER_UNREF(exec_ctx, chand->resolver, "channel");
558 }
Mark D. Roth0e48a9a2016-09-08 14:14:39 -0700559 if (chand->client_channel_factory != NULL) {
560 grpc_client_channel_factory_unref(exec_ctx, chand->client_channel_factory);
561 }
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700562 if (chand->lb_policy != NULL) {
563 grpc_pollset_set_del_pollset_set(exec_ctx,
564 chand->lb_policy->interested_parties,
565 chand->interested_parties);
566 GRPC_LB_POLICY_UNREF(exec_ctx, chand->lb_policy, "channel");
567 }
Mark D. Rothb2d24882016-10-27 15:44:07 -0700568 gpr_free(chand->lb_policy_name);
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800569 gpr_free(chand->service_config_json);
Mark D. Roth9d480942016-10-19 14:18:05 -0700570 if (chand->method_params_table != NULL) {
Craig Tiller7c70b6c2017-01-23 07:48:42 -0800571 grpc_slice_hash_table_unref(exec_ctx, chand->method_params_table);
Mark D. Roth9fe284e2016-09-12 11:22:27 -0700572 }
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700573 grpc_connectivity_state_destroy(exec_ctx, &chand->state_tracker);
574 grpc_pollset_set_destroy(chand->interested_parties);
575 gpr_mu_destroy(&chand->mu);
576}
577
578/*************************************************************************
579 * PER-CALL FUNCTIONS
580 */
581
582#define GET_CALL(call_data) \
583 ((grpc_subchannel_call *)(gpr_atm_acq_load(&(call_data)->subchannel_call)))
584
585#define CANCELLED_CALL ((grpc_subchannel_call *)1)
586
587typedef enum {
588 GRPC_SUBCHANNEL_CALL_HOLDER_NOT_CREATING,
589 GRPC_SUBCHANNEL_CALL_HOLDER_PICKING_SUBCHANNEL
590} subchannel_creation_phase;
591
592/** Call data. Holds a pointer to grpc_subchannel_call and the
593 associated machinery to create such a pointer.
594 Handles queueing of stream ops until a call object is ready, waiting
595 for initial metadata before trying to create a call object,
596 and handling cancellation gracefully. */
597typedef struct client_channel_call_data {
Mark D. Roth72f6da82016-09-02 13:42:38 -0700598 // State for handling deadlines.
599 // The code in deadline_filter.c requires this to be the first field.
Mark D. Roth72f6da82016-09-02 13:42:38 -0700600 // TODO(roth): This is slightly sub-optimal in that grpc_deadline_state
601 // and this struct both independently store a pointer to the call
602 // stack and each has its own mutex. If/when we have time, find a way
Mark D. Roth6ad99172016-09-09 07:52:48 -0700603 // to avoid this without breaking the grpc_deadline_state abstraction.
Mark D. Roth72f6da82016-09-02 13:42:38 -0700604 grpc_deadline_state deadline_state;
Mark D. Rothf28763c2016-09-14 15:18:40 -0700605
Craig Tiller7c70b6c2017-01-23 07:48:42 -0800606 grpc_slice path; // Request path.
Mark D. Rothe40dd292016-10-05 14:58:37 -0700607 gpr_timespec call_start_time;
608 gpr_timespec deadline;
Mark D. Roth9d480942016-10-19 14:18:05 -0700609 wait_for_ready_value wait_for_ready_from_service_config;
Mark D. Rothe40dd292016-10-05 14:58:37 -0700610 grpc_closure read_service_config;
Mark D. Rothaa850a72016-09-26 13:38:02 -0700611
Mark D. Rothf28763c2016-09-14 15:18:40 -0700612 grpc_error *cancel_error;
Mark D. Roth72f6da82016-09-02 13:42:38 -0700613
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700614 /** either 0 for no call, 1 for cancelled, or a pointer to a
615 grpc_subchannel_call */
616 gpr_atm subchannel_call;
617
618 gpr_mu mu;
619
620 subchannel_creation_phase creation_phase;
621 grpc_connected_subchannel *connected_subchannel;
622 grpc_polling_entity *pollent;
623
Craig Tiller57726ca2016-09-12 11:59:45 -0700624 grpc_transport_stream_op **waiting_ops;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700625 size_t waiting_ops_count;
626 size_t waiting_ops_capacity;
627
628 grpc_closure next_step;
629
630 grpc_call_stack *owning_call;
David Garcia Quintasd1a47f12016-09-02 12:46:44 +0200631
632 grpc_linked_mdelem lb_token_mdelem;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700633} call_data;
634
Craig Tiller8b1d59c2016-12-27 15:15:30 -0800635grpc_subchannel_call *grpc_client_channel_get_subchannel_call(
636 grpc_call_element *call_elem) {
637 grpc_subchannel_call *scc = GET_CALL((call_data *)call_elem->call_data);
638 return scc == CANCELLED_CALL ? NULL : scc;
639}
640
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700641static void add_waiting_locked(call_data *calld, grpc_transport_stream_op *op) {
642 GPR_TIMER_BEGIN("add_waiting_locked", 0);
643 if (calld->waiting_ops_count == calld->waiting_ops_capacity) {
644 calld->waiting_ops_capacity = GPR_MAX(3, 2 * calld->waiting_ops_capacity);
645 calld->waiting_ops =
646 gpr_realloc(calld->waiting_ops,
647 calld->waiting_ops_capacity * sizeof(*calld->waiting_ops));
648 }
Craig Tiller57726ca2016-09-12 11:59:45 -0700649 calld->waiting_ops[calld->waiting_ops_count++] = op;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700650 GPR_TIMER_END("add_waiting_locked", 0);
651}
652
653static void fail_locked(grpc_exec_ctx *exec_ctx, call_data *calld,
654 grpc_error *error) {
655 size_t i;
656 for (i = 0; i < calld->waiting_ops_count; i++) {
657 grpc_transport_stream_op_finish_with_failure(
Craig Tiller57726ca2016-09-12 11:59:45 -0700658 exec_ctx, calld->waiting_ops[i], GRPC_ERROR_REF(error));
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700659 }
660 calld->waiting_ops_count = 0;
661 GRPC_ERROR_UNREF(error);
662}
663
664typedef struct {
Craig Tiller57726ca2016-09-12 11:59:45 -0700665 grpc_transport_stream_op **ops;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700666 size_t nops;
667 grpc_subchannel_call *call;
668} retry_ops_args;
669
670static void retry_ops(grpc_exec_ctx *exec_ctx, void *args, grpc_error *error) {
671 retry_ops_args *a = args;
672 size_t i;
673 for (i = 0; i < a->nops; i++) {
Craig Tiller57726ca2016-09-12 11:59:45 -0700674 grpc_subchannel_call_process_op(exec_ctx, a->call, a->ops[i]);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700675 }
676 GRPC_SUBCHANNEL_CALL_UNREF(exec_ctx, a->call, "retry_ops");
677 gpr_free(a->ops);
678 gpr_free(a);
679}
680
681static void retry_waiting_locked(grpc_exec_ctx *exec_ctx, call_data *calld) {
Craig Tiller57726ca2016-09-12 11:59:45 -0700682 if (calld->waiting_ops_count == 0) {
683 return;
684 }
685
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700686 retry_ops_args *a = gpr_malloc(sizeof(*a));
687 a->ops = calld->waiting_ops;
688 a->nops = calld->waiting_ops_count;
689 a->call = GET_CALL(calld);
690 if (a->call == CANCELLED_CALL) {
691 gpr_free(a);
692 fail_locked(exec_ctx, calld, GRPC_ERROR_CANCELLED);
693 return;
694 }
695 calld->waiting_ops = NULL;
696 calld->waiting_ops_count = 0;
697 calld->waiting_ops_capacity = 0;
698 GRPC_SUBCHANNEL_CALL_REF(a->call, "retry_ops");
Craig Tiller91031da2016-12-28 15:44:25 -0800699 grpc_closure_sched(
700 exec_ctx, grpc_closure_create(retry_ops, a, grpc_schedule_on_exec_ctx),
701 GRPC_ERROR_NONE);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700702}
703
704static void subchannel_ready(grpc_exec_ctx *exec_ctx, void *arg,
705 grpc_error *error) {
Yuchen Zeng19656b12016-09-01 18:00:45 -0700706 grpc_call_element *elem = arg;
707 call_data *calld = elem->call_data;
708 channel_data *chand = elem->channel_data;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700709 gpr_mu_lock(&calld->mu);
710 GPR_ASSERT(calld->creation_phase ==
711 GRPC_SUBCHANNEL_CALL_HOLDER_PICKING_SUBCHANNEL);
Yuchen Zeng19656b12016-09-01 18:00:45 -0700712 grpc_polling_entity_del_from_pollset_set(exec_ctx, calld->pollent,
713 chand->interested_parties);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700714 calld->creation_phase = GRPC_SUBCHANNEL_CALL_HOLDER_NOT_CREATING;
715 if (calld->connected_subchannel == NULL) {
716 gpr_atm_no_barrier_store(&calld->subchannel_call, 1);
717 fail_locked(exec_ctx, calld, GRPC_ERROR_CREATE_REFERENCING(
718 "Failed to create subchannel", &error, 1));
Mark D. Roth72f6da82016-09-02 13:42:38 -0700719 } else if (GET_CALL(calld) == CANCELLED_CALL) {
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700720 /* already cancelled before subchannel became ready */
David Garcia Quintas68a9e382016-12-13 10:50:40 -0800721 grpc_error *cancellation_error = GRPC_ERROR_CREATE_REFERENCING(
722 "Cancelled before creating subchannel", &error, 1);
723 /* if due to deadline, attach the deadline exceeded status to the error */
724 if (gpr_time_cmp(calld->deadline, gpr_now(GPR_CLOCK_MONOTONIC)) < 0) {
725 cancellation_error =
726 grpc_error_set_int(cancellation_error, GRPC_ERROR_INT_GRPC_STATUS,
727 GRPC_STATUS_DEADLINE_EXCEEDED);
728 }
729 fail_locked(exec_ctx, calld, cancellation_error);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700730 } else {
Mark D. Roth9fe284e2016-09-12 11:22:27 -0700731 /* Create call on subchannel. */
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700732 grpc_subchannel_call *subchannel_call = NULL;
733 grpc_error *new_error = grpc_connected_subchannel_create_call(
Mark D. Rothaa850a72016-09-26 13:38:02 -0700734 exec_ctx, calld->connected_subchannel, calld->pollent, calld->path,
Mark D. Roth3d883412016-11-07 13:42:54 -0800735 calld->call_start_time, calld->deadline, &subchannel_call);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700736 if (new_error != GRPC_ERROR_NONE) {
737 new_error = grpc_error_add_child(new_error, error);
738 subchannel_call = CANCELLED_CALL;
739 fail_locked(exec_ctx, calld, new_error);
740 }
741 gpr_atm_rel_store(&calld->subchannel_call,
742 (gpr_atm)(uintptr_t)subchannel_call);
743 retry_waiting_locked(exec_ctx, calld);
744 }
745 gpr_mu_unlock(&calld->mu);
746 GRPC_CALL_STACK_UNREF(exec_ctx, calld->owning_call, "pick_subchannel");
747}
748
749static char *cc_get_peer(grpc_exec_ctx *exec_ctx, grpc_call_element *elem) {
750 call_data *calld = elem->call_data;
751 grpc_subchannel_call *subchannel_call = GET_CALL(calld);
752 if (subchannel_call == NULL || subchannel_call == CANCELLED_CALL) {
753 return NULL;
754 } else {
755 return grpc_subchannel_call_get_peer(exec_ctx, subchannel_call);
756 }
757}
758
Craig Tiller577c9b22015-11-02 14:11:15 -0800759typedef struct {
760 grpc_metadata_batch *initial_metadata;
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800761 uint32_t initial_metadata_flags;
Craig Tillerb5585d42015-11-17 07:18:31 -0800762 grpc_connected_subchannel **connected_subchannel;
Craig Tiller577c9b22015-11-02 14:11:15 -0800763 grpc_closure *on_ready;
764 grpc_call_element *elem;
765 grpc_closure closure;
766} continue_picking_args;
767
Yuchen Zeng144ce652016-09-01 18:19:34 -0700768/** Return true if subchannel is available immediately (in which case on_ready
769 should not be called), or false otherwise (in which case on_ready should be
770 called when the subchannel is available). */
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700771static bool pick_subchannel(grpc_exec_ctx *exec_ctx, grpc_call_element *elem,
772 grpc_metadata_batch *initial_metadata,
773 uint32_t initial_metadata_flags,
774 grpc_connected_subchannel **connected_subchannel,
Mark D. Roth72f6da82016-09-02 13:42:38 -0700775 grpc_closure *on_ready, grpc_error *error);
Craig Tiller577c9b22015-11-02 14:11:15 -0800776
Craig Tiller804ff712016-05-05 16:25:40 -0700777static void continue_picking(grpc_exec_ctx *exec_ctx, void *arg,
778 grpc_error *error) {
Craig Tiller577c9b22015-11-02 14:11:15 -0800779 continue_picking_args *cpa = arg;
Craig Tiller0ede5452016-04-23 12:21:45 -0700780 if (cpa->connected_subchannel == NULL) {
Craig Tiller577c9b22015-11-02 14:11:15 -0800781 /* cancelled, do nothing */
Craig Tiller804ff712016-05-05 16:25:40 -0700782 } else if (error != GRPC_ERROR_NONE) {
Craig Tiller91031da2016-12-28 15:44:25 -0800783 grpc_closure_sched(exec_ctx, cpa->on_ready, GRPC_ERROR_REF(error));
Mark D. Roth9dab7d52016-10-07 07:48:03 -0700784 } else {
785 call_data *calld = cpa->elem->call_data;
786 gpr_mu_lock(&calld->mu);
787 if (pick_subchannel(exec_ctx, cpa->elem, cpa->initial_metadata,
Mark D. Rothfd2ddd22016-10-07 10:11:10 -0700788 cpa->initial_metadata_flags, cpa->connected_subchannel,
789 cpa->on_ready, GRPC_ERROR_NONE)) {
Craig Tiller91031da2016-12-28 15:44:25 -0800790 grpc_closure_sched(exec_ctx, cpa->on_ready, GRPC_ERROR_NONE);
Mark D. Roth9dab7d52016-10-07 07:48:03 -0700791 }
792 gpr_mu_unlock(&calld->mu);
Craig Tiller577c9b22015-11-02 14:11:15 -0800793 }
794 gpr_free(cpa);
795}
796
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700797static bool pick_subchannel(grpc_exec_ctx *exec_ctx, grpc_call_element *elem,
798 grpc_metadata_batch *initial_metadata,
799 uint32_t initial_metadata_flags,
800 grpc_connected_subchannel **connected_subchannel,
Mark D. Roth72f6da82016-09-02 13:42:38 -0700801 grpc_closure *on_ready, grpc_error *error) {
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700802 GPR_TIMER_BEGIN("pick_subchannel", 0);
Craig Tillerbfc9adc2016-06-27 13:16:22 -0700803
Craig Tiller577c9b22015-11-02 14:11:15 -0800804 channel_data *chand = elem->channel_data;
805 call_data *calld = elem->call_data;
806 continue_picking_args *cpa;
807 grpc_closure *closure;
808
Craig Tillerb5585d42015-11-17 07:18:31 -0800809 GPR_ASSERT(connected_subchannel);
Craig Tiller577c9b22015-11-02 14:11:15 -0800810
Mark D. Rothff4df062016-08-22 15:02:49 -0700811 gpr_mu_lock(&chand->mu);
Craig Tiller577c9b22015-11-02 14:11:15 -0800812 if (initial_metadata == NULL) {
813 if (chand->lb_policy != NULL) {
Craig Tillerab33b482015-11-21 08:11:04 -0800814 grpc_lb_policy_cancel_pick(exec_ctx, chand->lb_policy,
Mark D. Roth5f844002016-09-08 08:20:53 -0700815 connected_subchannel, GRPC_ERROR_REF(error));
Craig Tiller577c9b22015-11-02 14:11:15 -0800816 }
817 for (closure = chand->waiting_for_config_closures.head; closure != NULL;
Craig Tiller804ff712016-05-05 16:25:40 -0700818 closure = closure->next_data.next) {
Craig Tiller577c9b22015-11-02 14:11:15 -0800819 cpa = closure->cb_arg;
Craig Tillerb5585d42015-11-17 07:18:31 -0800820 if (cpa->connected_subchannel == connected_subchannel) {
821 cpa->connected_subchannel = NULL;
Craig Tiller91031da2016-12-28 15:44:25 -0800822 grpc_closure_sched(
Mark D. Roth932b10c2016-09-09 08:44:30 -0700823 exec_ctx, cpa->on_ready,
Craig Tiller91031da2016-12-28 15:44:25 -0800824 GRPC_ERROR_CREATE_REFERENCING("Pick cancelled", &error, 1));
Craig Tiller577c9b22015-11-02 14:11:15 -0800825 }
826 }
Mark D. Rothff4df062016-08-22 15:02:49 -0700827 gpr_mu_unlock(&chand->mu);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700828 GPR_TIMER_END("pick_subchannel", 0);
Mark D. Roth697a1f62016-09-07 13:35:07 -0700829 GRPC_ERROR_UNREF(error);
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700830 return true;
Craig Tiller577c9b22015-11-02 14:11:15 -0800831 }
Mark D. Roth697a1f62016-09-07 13:35:07 -0700832 GPR_ASSERT(error == GRPC_ERROR_NONE);
Craig Tiller577c9b22015-11-02 14:11:15 -0800833 if (chand->lb_policy != NULL) {
Craig Tiller86c0f8a2015-12-01 20:05:40 -0800834 grpc_lb_policy *lb_policy = chand->lb_policy;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700835 GRPC_LB_POLICY_REF(lb_policy, "pick_subchannel");
Mark D. Rothff4df062016-08-22 15:02:49 -0700836 gpr_mu_unlock(&chand->mu);
Mark D. Rothe40dd292016-10-05 14:58:37 -0700837 // If the application explicitly set wait_for_ready, use that.
838 // Otherwise, if the service config specified a value for this
839 // method, use that.
Mark D. Rothc1c38582016-10-11 11:03:27 -0700840 const bool wait_for_ready_set_from_api =
841 initial_metadata_flags &
842 GRPC_INITIAL_METADATA_WAIT_FOR_READY_EXPLICITLY_SET;
843 const bool wait_for_ready_set_from_service_config =
844 calld->wait_for_ready_from_service_config != WAIT_FOR_READY_UNSET;
845 if (!wait_for_ready_set_from_api &&
846 wait_for_ready_set_from_service_config) {
Mark D. Rothe40dd292016-10-05 14:58:37 -0700847 if (calld->wait_for_ready_from_service_config == WAIT_FOR_READY_TRUE) {
848 initial_metadata_flags |= GRPC_INITIAL_METADATA_WAIT_FOR_READY;
849 } else {
850 initial_metadata_flags &= ~GRPC_INITIAL_METADATA_WAIT_FOR_READY;
851 }
852 }
David Garcia Quintas92eb6b92016-09-30 14:07:39 -0700853 const grpc_lb_policy_pick_args inputs = {
Yuchen Zengac8bc422016-10-05 14:00:02 -0700854 initial_metadata, initial_metadata_flags, &calld->lb_token_mdelem,
855 gpr_inf_future(GPR_CLOCK_MONOTONIC)};
Mark D. Roth55f25b62016-10-12 14:55:20 -0700856 const bool result = grpc_lb_policy_pick(
857 exec_ctx, lb_policy, &inputs, connected_subchannel, NULL, on_ready);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700858 GRPC_LB_POLICY_UNREF(exec_ctx, lb_policy, "pick_subchannel");
859 GPR_TIMER_END("pick_subchannel", 0);
Mark D. Roth9dab7d52016-10-07 07:48:03 -0700860 return result;
Craig Tiller577c9b22015-11-02 14:11:15 -0800861 }
862 if (chand->resolver != NULL && !chand->started_resolving) {
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700863 chand->started_resolving = true;
Craig Tiller906e3bc2015-11-24 07:31:31 -0800864 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "resolver");
Mark D. Roth046cf762016-09-26 11:13:51 -0700865 grpc_resolver_next(exec_ctx, chand->resolver, &chand->resolver_result,
Mark D. Rothff4df062016-08-22 15:02:49 -0700866 &chand->on_resolver_result_changed);
Craig Tiller577c9b22015-11-02 14:11:15 -0800867 }
Craig Tiller0eab6972016-04-23 12:59:57 -0700868 if (chand->resolver != NULL) {
869 cpa = gpr_malloc(sizeof(*cpa));
870 cpa->initial_metadata = initial_metadata;
871 cpa->initial_metadata_flags = initial_metadata_flags;
872 cpa->connected_subchannel = connected_subchannel;
873 cpa->on_ready = on_ready;
874 cpa->elem = elem;
Craig Tiller91031da2016-12-28 15:44:25 -0800875 grpc_closure_init(&cpa->closure, continue_picking, cpa,
876 grpc_schedule_on_exec_ctx);
Craig Tiller804ff712016-05-05 16:25:40 -0700877 grpc_closure_list_append(&chand->waiting_for_config_closures, &cpa->closure,
878 GRPC_ERROR_NONE);
Craig Tiller0eab6972016-04-23 12:59:57 -0700879 } else {
Craig Tiller91031da2016-12-28 15:44:25 -0800880 grpc_closure_sched(exec_ctx, on_ready, GRPC_ERROR_CREATE("Disconnected"));
Craig Tiller0eab6972016-04-23 12:59:57 -0700881 }
Mark D. Rothff4df062016-08-22 15:02:49 -0700882 gpr_mu_unlock(&chand->mu);
Craig Tillerbfc9adc2016-06-27 13:16:22 -0700883
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700884 GPR_TIMER_END("pick_subchannel", 0);
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700885 return false;
Craig Tiller577c9b22015-11-02 14:11:15 -0800886}
887
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700888// The logic here is fairly complicated, due to (a) the fact that we
889// need to handle the case where we receive the send op before the
890// initial metadata op, and (b) the need for efficiency, especially in
891// the streaming case.
892// TODO(ctiller): Explain this more thoroughly.
893static void cc_start_transport_stream_op(grpc_exec_ctx *exec_ctx,
894 grpc_call_element *elem,
895 grpc_transport_stream_op *op) {
896 call_data *calld = elem->call_data;
Yuchen Zeng19656b12016-09-01 18:00:45 -0700897 channel_data *chand = elem->channel_data;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700898 GRPC_CALL_LOG_OP(GPR_INFO, elem, op);
Mark D. Roth72f6da82016-09-02 13:42:38 -0700899 grpc_deadline_state_client_start_transport_stream_op(exec_ctx, elem, op);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700900 /* try to (atomically) get the call */
901 grpc_subchannel_call *call = GET_CALL(calld);
902 GPR_TIMER_BEGIN("cc_start_transport_stream_op", 0);
903 if (call == CANCELLED_CALL) {
Mark D. Rothf28763c2016-09-14 15:18:40 -0700904 grpc_transport_stream_op_finish_with_failure(
905 exec_ctx, op, GRPC_ERROR_REF(calld->cancel_error));
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700906 GPR_TIMER_END("cc_start_transport_stream_op", 0);
907 return;
908 }
909 if (call != NULL) {
910 grpc_subchannel_call_process_op(exec_ctx, call, op);
911 GPR_TIMER_END("cc_start_transport_stream_op", 0);
912 return;
913 }
914 /* we failed; lock and figure out what to do */
915 gpr_mu_lock(&calld->mu);
916retry:
917 /* need to recheck that another thread hasn't set the call */
918 call = GET_CALL(calld);
919 if (call == CANCELLED_CALL) {
920 gpr_mu_unlock(&calld->mu);
Mark D. Rothf28763c2016-09-14 15:18:40 -0700921 grpc_transport_stream_op_finish_with_failure(
922 exec_ctx, op, GRPC_ERROR_REF(calld->cancel_error));
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700923 GPR_TIMER_END("cc_start_transport_stream_op", 0);
924 return;
925 }
926 if (call != NULL) {
927 gpr_mu_unlock(&calld->mu);
928 grpc_subchannel_call_process_op(exec_ctx, call, op);
929 GPR_TIMER_END("cc_start_transport_stream_op", 0);
930 return;
931 }
932 /* if this is a cancellation, then we can raise our cancelled flag */
933 if (op->cancel_error != GRPC_ERROR_NONE) {
934 if (!gpr_atm_rel_cas(&calld->subchannel_call, 0,
935 (gpr_atm)(uintptr_t)CANCELLED_CALL)) {
936 goto retry;
937 } else {
Mark D. Rothf28763c2016-09-14 15:18:40 -0700938 // Stash a copy of cancel_error in our call data, so that we can use
939 // it for subsequent operations. This ensures that if the call is
940 // cancelled before any ops are passed down (e.g., if the deadline
941 // is in the past when the call starts), we can return the right
942 // error to the caller when the first op does get passed down.
943 calld->cancel_error = GRPC_ERROR_REF(op->cancel_error);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700944 switch (calld->creation_phase) {
945 case GRPC_SUBCHANNEL_CALL_HOLDER_NOT_CREATING:
946 fail_locked(exec_ctx, calld, GRPC_ERROR_REF(op->cancel_error));
947 break;
948 case GRPC_SUBCHANNEL_CALL_HOLDER_PICKING_SUBCHANNEL:
Mark D. Rothd4c0f552016-09-01 09:25:32 -0700949 pick_subchannel(exec_ctx, elem, NULL, 0, &calld->connected_subchannel,
Mark D. Roth72f6da82016-09-02 13:42:38 -0700950 NULL, GRPC_ERROR_REF(op->cancel_error));
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700951 break;
952 }
953 gpr_mu_unlock(&calld->mu);
Mark D. Roth72f6da82016-09-02 13:42:38 -0700954 grpc_transport_stream_op_finish_with_failure(
955 exec_ctx, op, GRPC_ERROR_REF(op->cancel_error));
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700956 GPR_TIMER_END("cc_start_transport_stream_op", 0);
957 return;
958 }
959 }
960 /* if we don't have a subchannel, try to get one */
961 if (calld->creation_phase == GRPC_SUBCHANNEL_CALL_HOLDER_NOT_CREATING &&
962 calld->connected_subchannel == NULL &&
963 op->send_initial_metadata != NULL) {
964 calld->creation_phase = GRPC_SUBCHANNEL_CALL_HOLDER_PICKING_SUBCHANNEL;
Craig Tiller91031da2016-12-28 15:44:25 -0800965 grpc_closure_init(&calld->next_step, subchannel_ready, elem,
966 grpc_schedule_on_exec_ctx);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700967 GRPC_CALL_STACK_REF(calld->owning_call, "pick_subchannel");
Yuchen Zeng144ce652016-09-01 18:19:34 -0700968 /* If a subchannel is not available immediately, the polling entity from
969 call_data should be provided to channel_data's interested_parties, so
970 that IO of the lb_policy and resolver could be done under it. */
Mark D. Rothd4c0f552016-09-01 09:25:32 -0700971 if (pick_subchannel(exec_ctx, elem, op->send_initial_metadata,
Mark D. Rothe40dd292016-10-05 14:58:37 -0700972 op->send_initial_metadata_flags,
973 &calld->connected_subchannel, &calld->next_step,
974 GRPC_ERROR_NONE)) {
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700975 calld->creation_phase = GRPC_SUBCHANNEL_CALL_HOLDER_NOT_CREATING;
976 GRPC_CALL_STACK_UNREF(exec_ctx, calld->owning_call, "pick_subchannel");
Yuchen Zeng19656b12016-09-01 18:00:45 -0700977 } else {
Yuchen Zeng19656b12016-09-01 18:00:45 -0700978 grpc_polling_entity_add_to_pollset_set(exec_ctx, calld->pollent,
979 chand->interested_parties);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700980 }
981 }
982 /* if we've got a subchannel, then let's ask it to create a call */
983 if (calld->creation_phase == GRPC_SUBCHANNEL_CALL_HOLDER_NOT_CREATING &&
984 calld->connected_subchannel != NULL) {
985 grpc_subchannel_call *subchannel_call = NULL;
986 grpc_error *error = grpc_connected_subchannel_create_call(
Mark D. Rothaa850a72016-09-26 13:38:02 -0700987 exec_ctx, calld->connected_subchannel, calld->pollent, calld->path,
Mark D. Roth3d883412016-11-07 13:42:54 -0800988 calld->call_start_time, calld->deadline, &subchannel_call);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700989 if (error != GRPC_ERROR_NONE) {
990 subchannel_call = CANCELLED_CALL;
991 fail_locked(exec_ctx, calld, GRPC_ERROR_REF(error));
992 grpc_transport_stream_op_finish_with_failure(exec_ctx, op, error);
993 }
994 gpr_atm_rel_store(&calld->subchannel_call,
995 (gpr_atm)(uintptr_t)subchannel_call);
996 retry_waiting_locked(exec_ctx, calld);
997 goto retry;
998 }
999 /* nothing to be done but wait */
1000 add_waiting_locked(calld, op);
1001 gpr_mu_unlock(&calld->mu);
1002 GPR_TIMER_END("cc_start_transport_stream_op", 0);
1003}
1004
Mark D. Rothe40dd292016-10-05 14:58:37 -07001005// Gets data from the service config. Invoked when the resolver returns
1006// its initial result.
1007static void read_service_config(grpc_exec_ctx *exec_ctx, void *arg,
1008 grpc_error *error) {
1009 grpc_call_element *elem = arg;
1010 channel_data *chand = elem->channel_data;
1011 call_data *calld = elem->call_data;
1012 // If this is an error, there's no point in looking at the service config.
Mark D. Roth196387a2016-10-12 14:53:36 -07001013 if (error == GRPC_ERROR_NONE) {
1014 // Get the method config table from channel data.
1015 gpr_mu_lock(&chand->mu);
Craig Tiller7c70b6c2017-01-23 07:48:42 -08001016 grpc_slice_hash_table *method_params_table = NULL;
Mark D. Roth9d480942016-10-19 14:18:05 -07001017 if (chand->method_params_table != NULL) {
1018 method_params_table =
Craig Tiller7c70b6c2017-01-23 07:48:42 -08001019 grpc_slice_hash_table_ref(chand->method_params_table);
Mark D. Rothe40dd292016-10-05 14:58:37 -07001020 }
Mark D. Roth196387a2016-10-12 14:53:36 -07001021 gpr_mu_unlock(&chand->mu);
1022 // If the method config table was present, use it.
Mark D. Roth9d480942016-10-19 14:18:05 -07001023 if (method_params_table != NULL) {
Craig Tiller87a7e1f2016-11-09 09:42:19 -08001024 const method_parameters *method_params = grpc_method_config_table_get(
1025 exec_ctx, method_params_table, calld->path);
Mark D. Roth9d480942016-10-19 14:18:05 -07001026 if (method_params != NULL) {
1027 const bool have_method_timeout =
1028 gpr_time_cmp(method_params->timeout, gpr_time_0(GPR_TIMESPAN)) != 0;
1029 if (have_method_timeout ||
1030 method_params->wait_for_ready != WAIT_FOR_READY_UNSET) {
Mark D. Roth196387a2016-10-12 14:53:36 -07001031 gpr_mu_lock(&calld->mu);
Mark D. Roth9d480942016-10-19 14:18:05 -07001032 if (have_method_timeout) {
1033 const gpr_timespec per_method_deadline =
1034 gpr_time_add(calld->call_start_time, method_params->timeout);
Mark D. Roth196387a2016-10-12 14:53:36 -07001035 if (gpr_time_cmp(per_method_deadline, calld->deadline) < 0) {
1036 calld->deadline = per_method_deadline;
1037 // Reset deadline timer.
1038 grpc_deadline_state_reset(exec_ctx, elem, calld->deadline);
1039 }
1040 }
Mark D. Roth9d480942016-10-19 14:18:05 -07001041 if (method_params->wait_for_ready != WAIT_FOR_READY_UNSET) {
Mark D. Roth196387a2016-10-12 14:53:36 -07001042 calld->wait_for_ready_from_service_config =
Mark D. Roth9d480942016-10-19 14:18:05 -07001043 method_params->wait_for_ready;
Mark D. Roth196387a2016-10-12 14:53:36 -07001044 }
1045 gpr_mu_unlock(&calld->mu);
1046 }
1047 }
Craig Tiller7c70b6c2017-01-23 07:48:42 -08001048 grpc_slice_hash_table_unref(exec_ctx, method_params_table);
Mark D. Roth196387a2016-10-12 14:53:36 -07001049 }
Mark D. Rothe40dd292016-10-05 14:58:37 -07001050 }
Mark D. Roth31292f22016-10-12 13:14:07 -07001051 GRPC_CALL_STACK_UNREF(exec_ctx, calld->owning_call, "read_service_config");
Mark D. Rothe40dd292016-10-05 14:58:37 -07001052}
1053
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001054/* Constructor for call_data */
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001055static grpc_error *cc_init_call_elem(grpc_exec_ctx *exec_ctx,
1056 grpc_call_element *elem,
1057 grpc_call_element_args *args) {
Mark D. Rothaa850a72016-09-26 13:38:02 -07001058 channel_data *chand = elem->channel_data;
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001059 call_data *calld = elem->call_data;
Mark D. Rothe40dd292016-10-05 14:58:37 -07001060 // Initialize data members.
1061 grpc_deadline_state_init(exec_ctx, elem, args->call_stack);
Craig Tiller7c70b6c2017-01-23 07:48:42 -08001062 calld->path = grpc_slice_ref_internal(args->path);
Mark D. Rothff08f332016-10-14 13:01:01 -07001063 calld->call_start_time = args->start_time;
Mark D. Rothe40dd292016-10-05 14:58:37 -07001064 calld->deadline = gpr_convert_clock_type(args->deadline, GPR_CLOCK_MONOTONIC);
1065 calld->wait_for_ready_from_service_config = WAIT_FOR_READY_UNSET;
Mark D. Rothf28763c2016-09-14 15:18:40 -07001066 calld->cancel_error = GRPC_ERROR_NONE;
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001067 gpr_atm_rel_store(&calld->subchannel_call, 0);
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001068 gpr_mu_init(&calld->mu);
1069 calld->connected_subchannel = NULL;
1070 calld->waiting_ops = NULL;
1071 calld->waiting_ops_count = 0;
1072 calld->waiting_ops_capacity = 0;
1073 calld->creation_phase = GRPC_SUBCHANNEL_CALL_HOLDER_NOT_CREATING;
1074 calld->owning_call = args->call_stack;
1075 calld->pollent = NULL;
Mark D. Rothe40dd292016-10-05 14:58:37 -07001076 // If the resolver has already returned results, then we can access
1077 // the service config parameters immediately. Otherwise, we need to
1078 // defer that work until the resolver returns an initial result.
1079 // TODO(roth): This code is almost but not quite identical to the code
1080 // in read_service_config() above. It would be nice to find a way to
1081 // combine them, to avoid having to maintain it twice.
1082 gpr_mu_lock(&chand->mu);
1083 if (chand->lb_policy != NULL) {
1084 // We already have a resolver result, so check for service config.
Mark D. Roth9d480942016-10-19 14:18:05 -07001085 if (chand->method_params_table != NULL) {
Craig Tiller7c70b6c2017-01-23 07:48:42 -08001086 grpc_slice_hash_table *method_params_table =
1087 grpc_slice_hash_table_ref(chand->method_params_table);
Mark D. Rothe40dd292016-10-05 14:58:37 -07001088 gpr_mu_unlock(&chand->mu);
Craig Tiller87a7e1f2016-11-09 09:42:19 -08001089 method_parameters *method_params = grpc_method_config_table_get(
1090 exec_ctx, method_params_table, args->path);
Mark D. Roth9d480942016-10-19 14:18:05 -07001091 if (method_params != NULL) {
1092 if (gpr_time_cmp(method_params->timeout,
1093 gpr_time_0(GPR_CLOCK_MONOTONIC)) != 0) {
Mark D. Rothe40dd292016-10-05 14:58:37 -07001094 gpr_timespec per_method_deadline =
Mark D. Roth9d480942016-10-19 14:18:05 -07001095 gpr_time_add(calld->call_start_time, method_params->timeout);
Mark D. Rothe40dd292016-10-05 14:58:37 -07001096 calld->deadline = gpr_time_min(calld->deadline, per_method_deadline);
1097 }
Mark D. Roth9d480942016-10-19 14:18:05 -07001098 if (method_params->wait_for_ready != WAIT_FOR_READY_UNSET) {
Mark D. Rothe40dd292016-10-05 14:58:37 -07001099 calld->wait_for_ready_from_service_config =
Mark D. Roth9d480942016-10-19 14:18:05 -07001100 method_params->wait_for_ready;
Mark D. Rothe40dd292016-10-05 14:58:37 -07001101 }
1102 }
Craig Tiller7c70b6c2017-01-23 07:48:42 -08001103 grpc_slice_hash_table_unref(exec_ctx, method_params_table);
Mark D. Rothe40dd292016-10-05 14:58:37 -07001104 } else {
1105 gpr_mu_unlock(&chand->mu);
1106 }
1107 } else {
1108 // We don't yet have a resolver result, so register a callback to
1109 // get the service config data once the resolver returns.
Mark D. Roth31292f22016-10-12 13:14:07 -07001110 // Take a reference to the call stack to be owned by the callback.
1111 GRPC_CALL_STACK_REF(calld->owning_call, "read_service_config");
Craig Tiller91031da2016-12-28 15:44:25 -08001112 grpc_closure_init(&calld->read_service_config, read_service_config, elem,
1113 grpc_schedule_on_exec_ctx);
Mark D. Rothe40dd292016-10-05 14:58:37 -07001114 grpc_closure_list_append(&chand->waiting_for_config_closures,
1115 &calld->read_service_config, GRPC_ERROR_NONE);
1116 gpr_mu_unlock(&chand->mu);
1117 }
1118 // Start the deadline timer with the current deadline value. If we
1119 // do not yet have service config data, then the timer may be reset
1120 // later.
1121 grpc_deadline_state_start(exec_ctx, elem, calld->deadline);
Mark D. Roth0badbe82016-06-23 10:15:12 -07001122 return GRPC_ERROR_NONE;
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001123}
1124
1125/* Destructor for call_data */
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001126static void cc_destroy_call_elem(grpc_exec_ctx *exec_ctx,
1127 grpc_call_element *elem,
1128 const grpc_call_final_info *final_info,
1129 void *and_free_memory) {
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001130 call_data *calld = elem->call_data;
Mark D. Rothf28763c2016-09-14 15:18:40 -07001131 grpc_deadline_state_destroy(exec_ctx, elem);
Craig Tiller7c70b6c2017-01-23 07:48:42 -08001132 grpc_slice_unref_internal(exec_ctx, calld->path);
Mark D. Rothf28763c2016-09-14 15:18:40 -07001133 GRPC_ERROR_UNREF(calld->cancel_error);
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001134 grpc_subchannel_call *call = GET_CALL(calld);
1135 if (call != NULL && call != CANCELLED_CALL) {
1136 GRPC_SUBCHANNEL_CALL_UNREF(exec_ctx, call, "client_channel_destroy_call");
1137 }
1138 GPR_ASSERT(calld->creation_phase == GRPC_SUBCHANNEL_CALL_HOLDER_NOT_CREATING);
1139 gpr_mu_destroy(&calld->mu);
1140 GPR_ASSERT(calld->waiting_ops_count == 0);
Craig Tiller693d3942016-10-27 16:51:25 -07001141 if (calld->connected_subchannel != NULL) {
1142 GRPC_CONNECTED_SUBCHANNEL_UNREF(exec_ctx, calld->connected_subchannel,
1143 "picked");
1144 }
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001145 gpr_free(calld->waiting_ops);
Craig Tiller2c8063c2016-03-22 22:12:15 -07001146 gpr_free(and_free_memory);
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001147}
1148
David Garcia Quintasf72eb972016-05-03 18:28:09 -07001149static void cc_set_pollset_or_pollset_set(grpc_exec_ctx *exec_ctx,
1150 grpc_call_element *elem,
David Garcia Quintas2a50dfe2016-05-31 15:09:12 -07001151 grpc_polling_entity *pollent) {
Craig Tiller577c9b22015-11-02 14:11:15 -08001152 call_data *calld = elem->call_data;
David Garcia Quintas2a50dfe2016-05-31 15:09:12 -07001153 calld->pollent = pollent;
Craig Tiller577c9b22015-11-02 14:11:15 -08001154}
1155
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001156/*************************************************************************
1157 * EXPORTED SYMBOLS
1158 */
1159
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001160const grpc_channel_filter grpc_client_channel_filter = {
Craig Tillerf40df232016-03-25 13:38:14 -07001161 cc_start_transport_stream_op,
1162 cc_start_transport_op,
1163 sizeof(call_data),
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001164 cc_init_call_elem,
David Garcia Quintas4afce7e2016-04-18 16:25:17 -07001165 cc_set_pollset_or_pollset_set,
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001166 cc_destroy_call_elem,
Craig Tillerf40df232016-03-25 13:38:14 -07001167 sizeof(channel_data),
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001168 cc_init_channel_elem,
1169 cc_destroy_channel_elem,
Craig Tillerf40df232016-03-25 13:38:14 -07001170 cc_get_peer,
Mark D. Rothb2d24882016-10-27 15:44:07 -07001171 cc_get_channel_info,
Craig Tillerf40df232016-03-25 13:38:14 -07001172 "client-channel",
Craig Tiller87d5b192015-04-16 14:37:57 -07001173};
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001174
Craig Tillera82950e2015-09-22 12:33:20 -07001175grpc_connectivity_state grpc_client_channel_check_connectivity_state(
1176 grpc_exec_ctx *exec_ctx, grpc_channel_element *elem, int try_to_connect) {
Craig Tiller48cb07c2015-07-15 16:16:15 -07001177 channel_data *chand = elem->channel_data;
1178 grpc_connectivity_state out;
Mark D. Rothff4df062016-08-22 15:02:49 -07001179 gpr_mu_lock(&chand->mu);
Craig Tiller804ff712016-05-05 16:25:40 -07001180 out = grpc_connectivity_state_check(&chand->state_tracker, NULL);
Craig Tillera82950e2015-09-22 12:33:20 -07001181 if (out == GRPC_CHANNEL_IDLE && try_to_connect) {
1182 if (chand->lb_policy != NULL) {
1183 grpc_lb_policy_exit_idle(exec_ctx, chand->lb_policy);
1184 } else {
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001185 chand->exit_idle_when_lb_policy_arrives = true;
Craig Tillera82950e2015-09-22 12:33:20 -07001186 if (!chand->started_resolving && chand->resolver != NULL) {
Craig Tiller906e3bc2015-11-24 07:31:31 -08001187 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "resolver");
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001188 chand->started_resolving = true;
Mark D. Roth046cf762016-09-26 11:13:51 -07001189 grpc_resolver_next(exec_ctx, chand->resolver, &chand->resolver_result,
Mark D. Rothff4df062016-08-22 15:02:49 -07001190 &chand->on_resolver_result_changed);
Craig Tillera82950e2015-09-22 12:33:20 -07001191 }
Craig Tiller48cb07c2015-07-15 16:16:15 -07001192 }
Craig Tillera82950e2015-09-22 12:33:20 -07001193 }
Mark D. Rothff4df062016-08-22 15:02:49 -07001194 gpr_mu_unlock(&chand->mu);
Craig Tiller48cb07c2015-07-15 16:16:15 -07001195 return out;
1196}
1197
Craig Tiller86c99582015-11-25 15:22:26 -08001198typedef struct {
1199 channel_data *chand;
1200 grpc_pollset *pollset;
1201 grpc_closure *on_complete;
1202 grpc_closure my_closure;
1203} external_connectivity_watcher;
1204
Craig Tiller1d881fb2015-12-01 07:39:04 -08001205static void on_external_watch_complete(grpc_exec_ctx *exec_ctx, void *arg,
Craig Tiller804ff712016-05-05 16:25:40 -07001206 grpc_error *error) {
Craig Tiller86c99582015-11-25 15:22:26 -08001207 external_connectivity_watcher *w = arg;
1208 grpc_closure *follow_up = w->on_complete;
Craig Tiller69b093b2016-02-25 19:04:07 -08001209 grpc_pollset_set_del_pollset(exec_ctx, w->chand->interested_parties,
Craig Tiller1d881fb2015-12-01 07:39:04 -08001210 w->pollset);
1211 GRPC_CHANNEL_STACK_UNREF(exec_ctx, w->chand->owning_stack,
1212 "external_connectivity_watcher");
Craig Tiller86c99582015-11-25 15:22:26 -08001213 gpr_free(w);
Craig Tiller804ff712016-05-05 16:25:40 -07001214 follow_up->cb(exec_ctx, follow_up->cb_arg, error);
Craig Tiller86c99582015-11-25 15:22:26 -08001215}
1216
Craig Tillera82950e2015-09-22 12:33:20 -07001217void grpc_client_channel_watch_connectivity_state(
Craig Tiller906e3bc2015-11-24 07:31:31 -08001218 grpc_exec_ctx *exec_ctx, grpc_channel_element *elem, grpc_pollset *pollset,
Craig Tillera82950e2015-09-22 12:33:20 -07001219 grpc_connectivity_state *state, grpc_closure *on_complete) {
Craig Tiller48cb07c2015-07-15 16:16:15 -07001220 channel_data *chand = elem->channel_data;
Craig Tiller86c99582015-11-25 15:22:26 -08001221 external_connectivity_watcher *w = gpr_malloc(sizeof(*w));
1222 w->chand = chand;
1223 w->pollset = pollset;
1224 w->on_complete = on_complete;
Craig Tiller69b093b2016-02-25 19:04:07 -08001225 grpc_pollset_set_add_pollset(exec_ctx, chand->interested_parties, pollset);
Craig Tiller91031da2016-12-28 15:44:25 -08001226 grpc_closure_init(&w->my_closure, on_external_watch_complete, w,
1227 grpc_schedule_on_exec_ctx);
Craig Tiller1d881fb2015-12-01 07:39:04 -08001228 GRPC_CHANNEL_STACK_REF(w->chand->owning_stack,
1229 "external_connectivity_watcher");
Mark D. Rothff4df062016-08-22 15:02:49 -07001230 gpr_mu_lock(&chand->mu);
Craig Tillera82950e2015-09-22 12:33:20 -07001231 grpc_connectivity_state_notify_on_state_change(
Craig Tiller86c99582015-11-25 15:22:26 -08001232 exec_ctx, &chand->state_tracker, state, &w->my_closure);
Mark D. Rothff4df062016-08-22 15:02:49 -07001233 gpr_mu_unlock(&chand->mu);
Craig Tiller48cb07c2015-07-15 16:16:15 -07001234}