blob: ee2f21a064aedc82e87cd78d7a05d2c15b5be69b [file] [log] [blame]
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001/*
2 *
Jan Tattermusch7897ae92017-06-07 22:57:36 +02003 * Copyright 2015 gRPC authors.
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08004 *
Jan Tattermusch7897ae92017-06-07 22:57:36 +02005 * 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 Nobleb7ebd3b2014-11-26 16:33:03 -08008 *
Jan Tattermusch7897ae92017-06-07 22:57:36 +02009 * http://www.apache.org/licenses/LICENSE-2.0
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080010 *
Jan Tattermusch7897ae92017-06-07 22:57:36 +020011 * 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 Nobleb7ebd3b2014-11-26 16:33:03 -080016 *
17 */
18
Yash Tibrewal37fdb732017-09-25 16:45:02 -070019#include <grpc/support/port_platform.h>
20
Craig Tiller9eb0fde2017-03-31 16:59:30 -070021#include "src/core/ext/filters/client_channel/client_channel.h"
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080022
Yash Tibrewalfcd26bc2017-09-25 15:08:28 -070023#include <inttypes.h>
Mark D. Roth4c0fe492016-08-31 13:51:55 -070024#include <stdbool.h>
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080025#include <stdio.h>
Craig Tillereb3b12e2015-06-26 14:42:49 -070026#include <string.h>
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080027
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080028#include <grpc/support/alloc.h>
29#include <grpc/support/log.h>
Mark D. Rothb2d24882016-10-27 15:44:07 -070030#include <grpc/support/string_util.h>
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080031#include <grpc/support/sync.h>
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080032
Yuchen Zeng0bad30a2017-10-05 21:47:39 -070033#include "src/core/ext/filters/client_channel/backup_poller.h"
Craig Tiller9eb0fde2017-03-31 16:59:30 -070034#include "src/core/ext/filters/client_channel/http_connect_handshaker.h"
35#include "src/core/ext/filters/client_channel/lb_policy_registry.h"
36#include "src/core/ext/filters/client_channel/proxy_mapper_registry.h"
37#include "src/core/ext/filters/client_channel/resolver_registry.h"
38#include "src/core/ext/filters/client_channel/retry_throttle.h"
39#include "src/core/ext/filters/client_channel/subchannel.h"
Craig Tiller3be7dd02017-04-03 14:30:03 -070040#include "src/core/ext/filters/deadline/deadline_filter.h"
Craig Tiller9533d042016-03-25 17:11:06 -070041#include "src/core/lib/channel/channel_args.h"
42#include "src/core/lib/channel/connected_channel.h"
Mark D. Rothdbdf4952018-01-18 11:21:12 -080043#include "src/core/lib/gpr/string.h"
Craig Tillerbefafe62017-02-09 11:30:54 -080044#include "src/core/lib/iomgr/combiner.h"
Craig Tiller9533d042016-03-25 17:11:06 -070045#include "src/core/lib/iomgr/iomgr.h"
Mark D. Roth4c0fe492016-08-31 13:51:55 -070046#include "src/core/lib/iomgr/polling_entity.h"
Craig Tiller9533d042016-03-25 17:11:06 -070047#include "src/core/lib/profiling/timers.h"
Craig Tiller7c70b6c2017-01-23 07:48:42 -080048#include "src/core/lib/slice/slice_internal.h"
Craig Tiller9533d042016-03-25 17:11:06 -070049#include "src/core/lib/surface/channel.h"
50#include "src/core/lib/transport/connectivity_state.h"
Mark D. Roth9fe284e2016-09-12 11:22:27 -070051#include "src/core/lib/transport/metadata.h"
52#include "src/core/lib/transport/metadata_batch.h"
Mark D. Rothea846a02016-11-03 11:32:54 -070053#include "src/core/lib/transport/service_config.h"
Mark D. Roth9fe284e2016-09-12 11:22:27 -070054#include "src/core/lib/transport/static_metadata.h"
Craig Tiller8910ac62015-10-08 16:49:15 -070055
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080056/* Client channel implementation */
57
Craig Tiller694580f2017-10-18 14:48:14 -070058grpc_core::TraceFlag grpc_client_channel_trace(false, "client_channel");
Mark D. Roth60751fe2017-07-07 12:50:33 -070059
Mark D. Roth26b7be42016-10-24 10:08:07 -070060/*************************************************************************
61 * METHOD-CONFIG TABLE
62 */
63
Mark D. Roth9d480942016-10-19 14:18:05 -070064typedef enum {
Craig Tiller7acc37e2017-02-28 10:01:37 -080065 /* zero so it can be default initialized */
66 WAIT_FOR_READY_UNSET = 0,
Mark D. Roth9d480942016-10-19 14:18:05 -070067 WAIT_FOR_READY_FALSE,
68 WAIT_FOR_READY_TRUE
69} wait_for_ready_value;
70
Mark D. Roth95b627b2017-02-24 11:02:58 -080071typedef struct {
72 gpr_refcount refs;
Craig Tiller89c14282017-07-19 15:32:27 -070073 grpc_millis timeout;
Mark D. Roth9d480942016-10-19 14:18:05 -070074 wait_for_ready_value wait_for_ready;
75} method_parameters;
76
Craig Tillerbaa14a92017-11-03 09:09:36 -070077static method_parameters* method_parameters_ref(
78 method_parameters* method_params) {
Mark D. Roth95b627b2017-02-24 11:02:58 -080079 gpr_ref(&method_params->refs);
80 return method_params;
Mark D. Roth9d480942016-10-19 14:18:05 -070081}
82
Craig Tillerbaa14a92017-11-03 09:09:36 -070083static void method_parameters_unref(method_parameters* method_params) {
Mark D. Roth95b627b2017-02-24 11:02:58 -080084 if (gpr_unref(&method_params->refs)) {
85 gpr_free(method_params);
86 }
87}
88
Mark D. Roth76d0ec42017-10-26 11:08:14 -070089// Wrappers to pass to grpc_service_config_create_method_config_table().
Craig Tillera64b2b12017-11-03 15:23:13 -070090static void* method_parameters_ref_wrapper(void* value) {
Noah Eisenbe82e642018-02-09 09:16:55 -080091 return method_parameters_ref(static_cast<method_parameters*>(value));
Mark D. Roth76d0ec42017-10-26 11:08:14 -070092}
Yash Tibrewal8cf14702017-12-06 09:47:54 -080093static void method_parameters_unref_wrapper(void* value) {
Noah Eisenbe82e642018-02-09 09:16:55 -080094 method_parameters_unref(static_cast<method_parameters*>(value));
Craig Tiller87a7e1f2016-11-09 09:42:19 -080095}
96
Craig Tillerbaa14a92017-11-03 09:09:36 -070097static bool parse_wait_for_ready(grpc_json* field,
98 wait_for_ready_value* wait_for_ready) {
Mark D. Roth95b627b2017-02-24 11:02:58 -080099 if (field->type != GRPC_JSON_TRUE && field->type != GRPC_JSON_FALSE) {
100 return false;
101 }
102 *wait_for_ready = field->type == GRPC_JSON_TRUE ? WAIT_FOR_READY_TRUE
103 : WAIT_FOR_READY_FALSE;
104 return true;
105}
106
Craig Tillerbaa14a92017-11-03 09:09:36 -0700107static bool parse_timeout(grpc_json* field, grpc_millis* timeout) {
Mark D. Roth95b627b2017-02-24 11:02:58 -0800108 if (field->type != GRPC_JSON_STRING) return false;
109 size_t len = strlen(field->value);
110 if (field->value[len - 1] != 's') return false;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700111 char* buf = gpr_strdup(field->value);
Mark D. Roth95b627b2017-02-24 11:02:58 -0800112 buf[len - 1] = '\0'; // Remove trailing 's'.
Craig Tillerbaa14a92017-11-03 09:09:36 -0700113 char* decimal_point = strchr(buf, '.');
Craig Tiller89c14282017-07-19 15:32:27 -0700114 int nanos = 0;
Craig Tiller4782d922017-11-10 09:53:21 -0800115 if (decimal_point != nullptr) {
Mark D. Roth95b627b2017-02-24 11:02:58 -0800116 *decimal_point = '\0';
Craig Tiller89c14282017-07-19 15:32:27 -0700117 nanos = gpr_parse_nonnegative_int(decimal_point + 1);
118 if (nanos == -1) {
Mark D. Roth95b627b2017-02-24 11:02:58 -0800119 gpr_free(buf);
120 return false;
121 }
Noah Eisenbe82e642018-02-09 09:16:55 -0800122 int num_digits = static_cast<int>(strlen(decimal_point + 1));
Mark D. Rotha2821462017-10-26 11:31:58 -0700123 if (num_digits > 9) { // We don't accept greater precision than nanos.
124 gpr_free(buf);
125 return false;
Mark D. Roth95b627b2017-02-24 11:02:58 -0800126 }
Mark D. Rotha2821462017-10-26 11:31:58 -0700127 for (int i = 0; i < (9 - num_digits); ++i) {
128 nanos *= 10;
129 }
Mark D. Roth95b627b2017-02-24 11:02:58 -0800130 }
Mark D. Rotha2821462017-10-26 11:31:58 -0700131 int seconds = decimal_point == buf ? 0 : gpr_parse_nonnegative_int(buf);
Mark D. Roth95b627b2017-02-24 11:02:58 -0800132 gpr_free(buf);
Craig Tiller89c14282017-07-19 15:32:27 -0700133 if (seconds == -1) return false;
134 *timeout = seconds * GPR_MS_PER_SEC + nanos / GPR_NS_PER_MS;
Mark D. Roth95b627b2017-02-24 11:02:58 -0800135 return true;
136}
137
Craig Tillerbaa14a92017-11-03 09:09:36 -0700138static void* method_parameters_create_from_json(const grpc_json* json) {
Mark D. Rothc968e602016-11-02 14:07:36 -0700139 wait_for_ready_value wait_for_ready = WAIT_FOR_READY_UNSET;
Craig Tiller89c14282017-07-19 15:32:27 -0700140 grpc_millis timeout = 0;
Craig Tiller4782d922017-11-10 09:53:21 -0800141 for (grpc_json* field = json->child; field != nullptr; field = field->next) {
142 if (field->key == nullptr) continue;
Mark D. Roth84c8a022016-11-10 09:39:34 -0800143 if (strcmp(field->key, "waitForReady") == 0) {
Craig Tiller4782d922017-11-10 09:53:21 -0800144 if (wait_for_ready != WAIT_FOR_READY_UNSET) return nullptr; // Duplicate.
145 if (!parse_wait_for_ready(field, &wait_for_ready)) return nullptr;
Mark D. Rothc968e602016-11-02 14:07:36 -0700146 } else if (strcmp(field->key, "timeout") == 0) {
Craig Tiller4782d922017-11-10 09:53:21 -0800147 if (timeout > 0) return nullptr; // Duplicate.
148 if (!parse_timeout(field, &timeout)) return nullptr;
Mark D. Rothc968e602016-11-02 14:07:36 -0700149 }
150 }
Craig Tillerbaa14a92017-11-03 09:09:36 -0700151 method_parameters* value =
Noah Eisenbe82e642018-02-09 09:16:55 -0800152 static_cast<method_parameters*>(gpr_malloc(sizeof(method_parameters)));
Mark D. Roth95b627b2017-02-24 11:02:58 -0800153 gpr_ref_init(&value->refs, 1);
Mark D. Rothc968e602016-11-02 14:07:36 -0700154 value->timeout = timeout;
155 value->wait_for_ready = wait_for_ready;
Mark D. Roth9d480942016-10-19 14:18:05 -0700156 return value;
157}
158
Alexander Polcync3b1f182017-04-18 13:51:36 -0700159struct external_connectivity_watcher;
160
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700161/*************************************************************************
162 * CHANNEL-WIDE FUNCTIONS
163 */
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -0800164
Craig Tiller800dacb2015-10-06 09:10:26 -0700165typedef struct client_channel_channel_data {
Craig Tillerf5f17122015-06-25 08:47:26 -0700166 /** resolver for this channel */
Mark D. Roth209f6442018-02-08 10:26:46 -0800167 grpc_core::OrphanablePtr<grpc_core::Resolver> resolver;
Craig Tiller20a3c352015-08-05 08:39:50 -0700168 /** have we started resolving this channel */
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700169 bool started_resolving;
Craig Tiller3be7dd02017-04-03 14:30:03 -0700170 /** is deadline checking enabled? */
171 bool deadline_checking_enabled;
Mark D. Roth0e48a9a2016-09-08 14:14:39 -0700172 /** client channel factory */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700173 grpc_client_channel_factory* client_channel_factory;
Craig Tillerf5f17122015-06-25 08:47:26 -0700174
Craig Tillerbefafe62017-02-09 11:30:54 -0800175 /** combiner protecting all variables below in this data structure */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700176 grpc_combiner* combiner;
Mark D. Roth046cf762016-09-26 11:13:51 -0700177 /** currently active load balancer */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700178 grpc_lb_policy* lb_policy;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800179 /** retry throttle data */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700180 grpc_server_retry_throttle_data* retry_throttle_data;
Mark D. Roth9d480942016-10-19 14:18:05 -0700181 /** maps method names to method_parameters structs */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700182 grpc_slice_hash_table* method_params_table;
Mark D. Roth046cf762016-09-26 11:13:51 -0700183 /** incoming resolver result - set by resolver.next() */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700184 grpc_channel_args* resolver_result;
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700185 /** a list of closures that are all waiting for resolver result to come in */
186 grpc_closure_list waiting_for_resolver_result_closures;
Craig Tiller3f475422015-06-25 10:43:05 -0700187 /** resolver callback */
Mark D. Rothff4df062016-08-22 15:02:49 -0700188 grpc_closure on_resolver_result_changed;
Craig Tiller3f475422015-06-25 10:43:05 -0700189 /** connectivity state being tracked */
Craig Tillerca3e9d32015-06-27 18:37:27 -0700190 grpc_connectivity_state_tracker state_tracker;
Craig Tiller48cb07c2015-07-15 16:16:15 -0700191 /** when an lb_policy arrives, should we try to exit idle */
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700192 bool exit_idle_when_lb_policy_arrives;
Craig Tiller906e3bc2015-11-24 07:31:31 -0800193 /** owning stack */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700194 grpc_channel_stack* owning_stack;
Craig Tiller69b093b2016-02-25 19:04:07 -0800195 /** interested parties (owned) */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700196 grpc_pollset_set* interested_parties;
Craig Tiller613dafa2017-02-09 12:00:43 -0800197
Alexander Polcync3b1f182017-04-18 13:51:36 -0700198 /* external_connectivity_watcher_list head is guarded by its own mutex, since
199 * counts need to be grabbed immediately without polling on a cq */
200 gpr_mu external_connectivity_watcher_list_mu;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700201 struct external_connectivity_watcher* external_connectivity_watcher_list_head;
Alexander Polcync3b1f182017-04-18 13:51:36 -0700202
Craig Tiller613dafa2017-02-09 12:00:43 -0800203 /* the following properties are guarded by a mutex since API's require them
Craig Tiller46dd7902017-02-23 09:42:16 -0800204 to be instantaneously available */
Craig Tiller613dafa2017-02-09 12:00:43 -0800205 gpr_mu info_mu;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700206 char* info_lb_policy_name;
Craig Tiller613dafa2017-02-09 12:00:43 -0800207 /** service config in JSON form */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700208 char* info_service_config_json;
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -0800209} channel_data;
210
Juanli Shen592cf342017-12-04 20:52:01 -0800211typedef struct {
212 channel_data* chand;
213 /** used as an identifier, don't dereference it because the LB policy may be
214 * non-existing when the callback is run */
215 grpc_lb_policy* lb_policy;
216 grpc_closure closure;
217} reresolution_request_args;
218
Craig Tillerd6c98df2015-08-18 09:33:44 -0700219/** We create one watcher for each new lb_policy that is returned from a
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700220 resolver, to watch for state changes from the lb_policy. When a state
221 change is seen, we update the channel, and create a new watcher. */
Craig Tillera82950e2015-09-22 12:33:20 -0700222typedef struct {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700223 channel_data* chand;
Craig Tiller33825112015-09-18 07:44:19 -0700224 grpc_closure on_changed;
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700225 grpc_connectivity_state state;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700226 grpc_lb_policy* lb_policy;
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700227} lb_policy_connectivity_watcher;
228
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800229static void watch_lb_policy_locked(channel_data* chand,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700230 grpc_lb_policy* lb_policy,
Craig Tiller2400bf52017-02-09 16:25:19 -0800231 grpc_connectivity_state current_state);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700232
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800233static void set_channel_connectivity_state_locked(channel_data* chand,
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800234 grpc_connectivity_state state,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700235 grpc_error* error,
236 const char* reason) {
David Garcia Quintas37251282017-04-14 13:46:03 -0700237 /* TODO: Improve failure handling:
238 * - Make it possible for policies to return GRPC_CHANNEL_TRANSIENT_FAILURE.
239 * - Hand over pending picks from old policies during the switch that happens
240 * when resolver provides an update. */
Craig Tiller4782d922017-11-10 09:53:21 -0800241 if (chand->lb_policy != nullptr) {
David Garcia Quintas956f7002017-04-13 15:40:06 -0700242 if (state == GRPC_CHANNEL_TRANSIENT_FAILURE) {
243 /* cancel picks with wait_for_ready=false */
244 grpc_lb_policy_cancel_picks_locked(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800245 chand->lb_policy,
David Garcia Quintas956f7002017-04-13 15:40:06 -0700246 /* mask= */ GRPC_INITIAL_METADATA_WAIT_FOR_READY,
247 /* check= */ 0, GRPC_ERROR_REF(error));
248 } else if (state == GRPC_CHANNEL_SHUTDOWN) {
249 /* cancel all picks */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800250 grpc_lb_policy_cancel_picks_locked(chand->lb_policy,
David Garcia Quintas956f7002017-04-13 15:40:06 -0700251 /* mask= */ 0, /* check= */ 0,
252 GRPC_ERROR_REF(error));
253 }
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800254 }
Craig Tiller6014e8a2017-10-16 13:50:29 -0700255 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700256 gpr_log(GPR_DEBUG, "chand=%p: setting connectivity state to %s", chand,
257 grpc_connectivity_state_name(state));
258 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800259 grpc_connectivity_state_set(&chand->state_tracker, state, error, reason);
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800260}
261
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800262static void on_lb_policy_state_changed_locked(void* arg, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800263 lb_policy_connectivity_watcher* w = static_cast<lb_policy_connectivity_watcher*>(arg);
Craig Tillerc5de8352017-02-09 14:08:05 -0800264 /* check if the notification is for the latest policy */
265 if (w->lb_policy == w->chand->lb_policy) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700266 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700267 gpr_log(GPR_DEBUG, "chand=%p: lb_policy=%p state changed to %s", w->chand,
268 w->lb_policy, grpc_connectivity_state_name(w->state));
269 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800270 set_channel_connectivity_state_locked(w->chand, w->state,
Craig Tillerc5de8352017-02-09 14:08:05 -0800271 GRPC_ERROR_REF(error), "lb_changed");
272 if (w->state != GRPC_CHANNEL_SHUTDOWN) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800273 watch_lb_policy_locked(w->chand, w->lb_policy, w->state);
Craig Tillerc5de8352017-02-09 14:08:05 -0800274 }
Craig Tillera82950e2015-09-22 12:33:20 -0700275 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800276 GRPC_CHANNEL_STACK_UNREF(w->chand->owning_stack, "watch_lb_policy");
Craig Tillera82950e2015-09-22 12:33:20 -0700277 gpr_free(w);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700278}
279
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800280static void watch_lb_policy_locked(channel_data* chand,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700281 grpc_lb_policy* lb_policy,
Craig Tiller2400bf52017-02-09 16:25:19 -0800282 grpc_connectivity_state current_state) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700283 lb_policy_connectivity_watcher* w =
Noah Eisenbe82e642018-02-09 09:16:55 -0800284 static_cast<lb_policy_connectivity_watcher*>(gpr_malloc(sizeof(*w)));
Craig Tiller906e3bc2015-11-24 07:31:31 -0800285 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "watch_lb_policy");
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700286 w->chand = chand;
ncteisen274bbbe2017-06-08 14:57:11 -0700287 GRPC_CLOSURE_INIT(&w->on_changed, on_lb_policy_state_changed_locked, w,
Craig Tilleree4b1452017-05-12 10:56:03 -0700288 grpc_combiner_scheduler(chand->combiner));
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700289 w->state = current_state;
290 w->lb_policy = lb_policy;
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800291 grpc_lb_policy_notify_on_state_change_locked(lb_policy, &w->state,
Craig Tiller2400bf52017-02-09 16:25:19 -0800292 &w->on_changed);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700293}
294
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800295static void start_resolving_locked(channel_data* chand) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700296 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700297 gpr_log(GPR_DEBUG, "chand=%p: starting name resolution", chand);
298 }
299 GPR_ASSERT(!chand->started_resolving);
300 chand->started_resolving = true;
301 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "resolver");
Mark D. Roth209f6442018-02-08 10:26:46 -0800302 chand->resolver->NextLocked(&chand->resolver_result,
303 &chand->on_resolver_result_changed);
Mark D. Roth60751fe2017-07-07 12:50:33 -0700304}
305
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800306typedef struct {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700307 char* server_name;
308 grpc_server_retry_throttle_data* retry_throttle_data;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800309} service_config_parsing_state;
310
Craig Tillerbaa14a92017-11-03 09:09:36 -0700311static void parse_retry_throttle_params(const grpc_json* field, void* arg) {
312 service_config_parsing_state* parsing_state =
Noah Eisenbe82e642018-02-09 09:16:55 -0800313 static_cast<service_config_parsing_state*>(arg);
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800314 if (strcmp(field->key, "retryThrottling") == 0) {
Craig Tiller4782d922017-11-10 09:53:21 -0800315 if (parsing_state->retry_throttle_data != nullptr) return; // Duplicate.
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800316 if (field->type != GRPC_JSON_OBJECT) return;
317 int max_milli_tokens = 0;
318 int milli_token_ratio = 0;
Craig Tiller4782d922017-11-10 09:53:21 -0800319 for (grpc_json* sub_field = field->child; sub_field != nullptr;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800320 sub_field = sub_field->next) {
Craig Tiller4782d922017-11-10 09:53:21 -0800321 if (sub_field->key == nullptr) return;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800322 if (strcmp(sub_field->key, "maxTokens") == 0) {
323 if (max_milli_tokens != 0) return; // Duplicate.
324 if (sub_field->type != GRPC_JSON_NUMBER) return;
325 max_milli_tokens = gpr_parse_nonnegative_int(sub_field->value);
326 if (max_milli_tokens == -1) return;
327 max_milli_tokens *= 1000;
328 } else if (strcmp(sub_field->key, "tokenRatio") == 0) {
329 if (milli_token_ratio != 0) return; // Duplicate.
330 if (sub_field->type != GRPC_JSON_NUMBER) return;
331 // We support up to 3 decimal digits.
332 size_t whole_len = strlen(sub_field->value);
333 uint32_t multiplier = 1;
334 uint32_t decimal_value = 0;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700335 const char* decimal_point = strchr(sub_field->value, '.');
Craig Tiller4782d922017-11-10 09:53:21 -0800336 if (decimal_point != nullptr) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800337 whole_len = static_cast<size_t>(decimal_point - sub_field->value);
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800338 multiplier = 1000;
339 size_t decimal_len = strlen(decimal_point + 1);
340 if (decimal_len > 3) decimal_len = 3;
341 if (!gpr_parse_bytes_to_uint32(decimal_point + 1, decimal_len,
342 &decimal_value)) {
343 return;
344 }
345 uint32_t decimal_multiplier = 1;
346 for (size_t i = 0; i < (3 - decimal_len); ++i) {
347 decimal_multiplier *= 10;
348 }
349 decimal_value *= decimal_multiplier;
350 }
351 uint32_t whole_value;
352 if (!gpr_parse_bytes_to_uint32(sub_field->value, whole_len,
353 &whole_value)) {
354 return;
355 }
Noah Eisenbe82e642018-02-09 09:16:55 -0800356 milli_token_ratio = static_cast<int>((whole_value * multiplier) + decimal_value);
Mark D. Rothb3322562017-02-23 14:38:02 -0800357 if (milli_token_ratio <= 0) return;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800358 }
359 }
360 parsing_state->retry_throttle_data =
361 grpc_retry_throttle_map_get_data_for_server(
362 parsing_state->server_name, max_milli_tokens, milli_token_ratio);
363 }
364}
365
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800366static void request_reresolution_locked(void* arg, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800367 reresolution_request_args* args = static_cast<reresolution_request_args*>(arg);
Juanli Shen592cf342017-12-04 20:52:01 -0800368 channel_data* chand = args->chand;
369 // If this invocation is for a stale LB policy, treat it as an LB shutdown
370 // signal.
371 if (args->lb_policy != chand->lb_policy || error != GRPC_ERROR_NONE ||
372 chand->resolver == nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800373 GRPC_CHANNEL_STACK_UNREF(chand->owning_stack, "re-resolution");
Juanli Shen592cf342017-12-04 20:52:01 -0800374 gpr_free(args);
375 return;
376 }
377 if (grpc_client_channel_trace.enabled()) {
378 gpr_log(GPR_DEBUG, "chand=%p: started name re-resolving", chand);
379 }
Mark D. Roth209f6442018-02-08 10:26:46 -0800380 chand->resolver->RequestReresolutionLocked();
Juanli Shen592cf342017-12-04 20:52:01 -0800381 // Give back the closure to the LB policy.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800382 grpc_lb_policy_set_reresolve_closure_locked(chand->lb_policy, &args->closure);
Juanli Shen592cf342017-12-04 20:52:01 -0800383}
384
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800385static void on_resolver_result_changed_locked(void* arg, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800386 channel_data* chand = static_cast<channel_data*>(arg);
Craig Tiller6014e8a2017-10-16 13:50:29 -0700387 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700388 gpr_log(GPR_DEBUG, "chand=%p: got resolver result: error=%s", chand,
389 grpc_error_string(error));
390 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700391 // Extract the following fields from the resolver result, if non-NULL.
Mark D. Roth15494b52017-07-12 15:26:55 -0700392 bool lb_policy_updated = false;
Craig Tiller4782d922017-11-10 09:53:21 -0800393 char* lb_policy_name_dup = nullptr;
Mark D. Roth60751fe2017-07-07 12:50:33 -0700394 bool lb_policy_name_changed = false;
Craig Tiller4782d922017-11-10 09:53:21 -0800395 grpc_lb_policy* new_lb_policy = nullptr;
396 char* service_config_json = nullptr;
397 grpc_server_retry_throttle_data* retry_throttle_data = nullptr;
398 grpc_slice_hash_table* method_params_table = nullptr;
399 if (chand->resolver_result != nullptr) {
Juanli Shen592cf342017-12-04 20:52:01 -0800400 if (chand->resolver != nullptr) {
401 // Find LB policy name.
402 const char* lb_policy_name = nullptr;
403 const grpc_arg* channel_arg = grpc_channel_args_find(
404 chand->resolver_result, GRPC_ARG_LB_POLICY_NAME);
405 if (channel_arg != nullptr) {
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800406 GPR_ASSERT(channel_arg->type == GRPC_ARG_STRING);
Juanli Shen592cf342017-12-04 20:52:01 -0800407 lb_policy_name = channel_arg->value.string;
Mark D. Rothbdc58b22016-11-04 09:25:57 -0700408 }
Juanli Shen592cf342017-12-04 20:52:01 -0800409 // Special case: If at least one balancer address is present, we use
410 // the grpclb policy, regardless of what the resolver actually specified.
411 channel_arg =
412 grpc_channel_args_find(chand->resolver_result, GRPC_ARG_LB_ADDRESSES);
413 if (channel_arg != nullptr && channel_arg->type == GRPC_ARG_POINTER) {
414 grpc_lb_addresses* addresses =
Noah Eisenbe82e642018-02-09 09:16:55 -0800415 static_cast<grpc_lb_addresses*>(channel_arg->value.pointer.p);
Juanli Shen592cf342017-12-04 20:52:01 -0800416 bool found_balancer_address = false;
417 for (size_t i = 0; i < addresses->num_addresses; ++i) {
418 if (addresses->addresses[i].is_balancer) {
419 found_balancer_address = true;
420 break;
421 }
422 }
423 if (found_balancer_address) {
424 if (lb_policy_name != nullptr &&
425 strcmp(lb_policy_name, "grpclb") != 0) {
426 gpr_log(GPR_INFO,
427 "resolver requested LB policy %s but provided at least one "
428 "balancer address -- forcing use of grpclb LB policy",
429 lb_policy_name);
430 }
431 lb_policy_name = "grpclb";
432 }
433 }
434 // Use pick_first if nothing was specified and we didn't select grpclb
435 // above.
436 if (lb_policy_name == nullptr) lb_policy_name = "pick_first";
437 grpc_lb_policy_args lb_policy_args;
438 lb_policy_args.args = chand->resolver_result;
439 lb_policy_args.client_channel_factory = chand->client_channel_factory;
440 lb_policy_args.combiner = chand->combiner;
441 // Check to see if we're already using the right LB policy.
442 // Note: It's safe to use chand->info_lb_policy_name here without
443 // taking a lock on chand->info_mu, because this function is the
444 // only thing that modifies its value, and it can only be invoked
445 // once at any given time.
446 lb_policy_name_changed =
447 chand->info_lb_policy_name == nullptr ||
448 gpr_stricmp(chand->info_lb_policy_name, lb_policy_name) != 0;
449 if (chand->lb_policy != nullptr && !lb_policy_name_changed) {
450 // Continue using the same LB policy. Update with new addresses.
451 lb_policy_updated = true;
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800452 grpc_lb_policy_update_locked(chand->lb_policy, &lb_policy_args);
Juanli Shen592cf342017-12-04 20:52:01 -0800453 } else {
454 // Instantiate new LB policy.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800455 new_lb_policy = grpc_lb_policy_create(lb_policy_name, &lb_policy_args);
Juanli Shen592cf342017-12-04 20:52:01 -0800456 if (new_lb_policy == nullptr) {
457 gpr_log(GPR_ERROR, "could not create LB policy \"%s\"",
458 lb_policy_name);
459 } else {
460 reresolution_request_args* args =
Noah Eisenbe82e642018-02-09 09:16:55 -0800461 static_cast<reresolution_request_args*>(gpr_zalloc(sizeof(*args)));
Juanli Shen592cf342017-12-04 20:52:01 -0800462 args->chand = chand;
463 args->lb_policy = new_lb_policy;
464 GRPC_CLOSURE_INIT(&args->closure, request_reresolution_locked, args,
465 grpc_combiner_scheduler(chand->combiner));
466 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "re-resolution");
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800467 grpc_lb_policy_set_reresolve_closure_locked(new_lb_policy,
Juanli Shen592cf342017-12-04 20:52:01 -0800468 &args->closure);
469 }
470 }
471 // Find service config.
472 channel_arg = grpc_channel_args_find(chand->resolver_result,
473 GRPC_ARG_SERVICE_CONFIG);
474 if (channel_arg != nullptr) {
475 GPR_ASSERT(channel_arg->type == GRPC_ARG_STRING);
476 service_config_json = gpr_strdup(channel_arg->value.string);
477 grpc_service_config* service_config =
478 grpc_service_config_create(service_config_json);
479 if (service_config != nullptr) {
480 channel_arg = grpc_channel_args_find(chand->resolver_result,
481 GRPC_ARG_SERVER_URI);
482 GPR_ASSERT(channel_arg != nullptr);
483 GPR_ASSERT(channel_arg->type == GRPC_ARG_STRING);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800484 grpc_uri* uri = grpc_uri_parse(channel_arg->value.string, true);
Juanli Shen592cf342017-12-04 20:52:01 -0800485 GPR_ASSERT(uri->path[0] != '\0');
486 service_config_parsing_state parsing_state;
487 memset(&parsing_state, 0, sizeof(parsing_state));
488 parsing_state.server_name =
489 uri->path[0] == '/' ? uri->path + 1 : uri->path;
490 grpc_service_config_parse_global_params(
491 service_config, parse_retry_throttle_params, &parsing_state);
492 grpc_uri_destroy(uri);
493 retry_throttle_data = parsing_state.retry_throttle_data;
494 method_params_table = grpc_service_config_create_method_config_table(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800495 service_config, method_parameters_create_from_json,
Juanli Shen592cf342017-12-04 20:52:01 -0800496 method_parameters_ref_wrapper, method_parameters_unref_wrapper);
497 grpc_service_config_destroy(service_config);
498 }
499 }
500 // Before we clean up, save a copy of lb_policy_name, since it might
501 // be pointing to data inside chand->resolver_result.
502 // The copy will be saved in chand->lb_policy_name below.
503 lb_policy_name_dup = gpr_strdup(lb_policy_name);
Mark D. Roth9fe284e2016-09-12 11:22:27 -0700504 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800505 grpc_channel_args_destroy(chand->resolver_result);
Craig Tiller4782d922017-11-10 09:53:21 -0800506 chand->resolver_result = nullptr;
Craig Tillera82950e2015-09-22 12:33:20 -0700507 }
Craig Tiller6014e8a2017-10-16 13:50:29 -0700508 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700509 gpr_log(GPR_DEBUG,
510 "chand=%p: resolver result: lb_policy_name=\"%s\"%s, "
511 "service_config=\"%s\"",
Yash Tibrewal9eb86722017-09-17 23:43:30 -0700512 chand, lb_policy_name_dup,
513 lb_policy_name_changed ? " (changed)" : "", service_config_json);
Mark D. Roth60751fe2017-07-07 12:50:33 -0700514 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700515 // Now swap out fields in chand. Note that the new values may still
516 // be NULL if (e.g.) the resolver failed to return results or the
517 // results did not contain the necessary data.
518 //
519 // First, swap out the data used by cc_get_channel_info().
Craig Tiller613dafa2017-02-09 12:00:43 -0800520 gpr_mu_lock(&chand->info_mu);
Craig Tiller4782d922017-11-10 09:53:21 -0800521 if (lb_policy_name_dup != nullptr) {
Craig Tiller613dafa2017-02-09 12:00:43 -0800522 gpr_free(chand->info_lb_policy_name);
Yash Tibrewal9eb86722017-09-17 23:43:30 -0700523 chand->info_lb_policy_name = lb_policy_name_dup;
Mark D. Rothb2d24882016-10-27 15:44:07 -0700524 }
Craig Tiller4782d922017-11-10 09:53:21 -0800525 if (service_config_json != nullptr) {
Craig Tiller613dafa2017-02-09 12:00:43 -0800526 gpr_free(chand->info_service_config_json);
527 chand->info_service_config_json = service_config_json;
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800528 }
Craig Tiller613dafa2017-02-09 12:00:43 -0800529 gpr_mu_unlock(&chand->info_mu);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700530 // Swap out the retry throttle data.
Craig Tiller4782d922017-11-10 09:53:21 -0800531 if (chand->retry_throttle_data != nullptr) {
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800532 grpc_server_retry_throttle_data_unref(chand->retry_throttle_data);
533 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700534 chand->retry_throttle_data = retry_throttle_data;
535 // Swap out the method params table.
Craig Tiller4782d922017-11-10 09:53:21 -0800536 if (chand->method_params_table != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800537 grpc_slice_hash_table_unref(chand->method_params_table);
Mark D. Roth046cf762016-09-26 11:13:51 -0700538 }
Mark D. Roth9d480942016-10-19 14:18:05 -0700539 chand->method_params_table = method_params_table;
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700540 // If we have a new LB policy or are shutting down (in which case
Juanli Shen592cf342017-12-04 20:52:01 -0800541 // new_lb_policy will be NULL), swap out the LB policy, unreffing the old one
542 // and removing its fds from chand->interested_parties. Note that we do NOT do
543 // this if either (a) we updated the existing LB policy above or (b) we failed
544 // to create the new LB policy (in which case we want to continue using the
545 // most recent one we had).
Craig Tiller4782d922017-11-10 09:53:21 -0800546 if (new_lb_policy != nullptr || error != GRPC_ERROR_NONE ||
547 chand->resolver == nullptr) {
548 if (chand->lb_policy != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700549 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700550 gpr_log(GPR_DEBUG, "chand=%p: unreffing lb_policy=%p", chand,
551 chand->lb_policy);
552 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800553 grpc_pollset_set_del_pollset_set(chand->lb_policy->interested_parties,
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700554 chand->interested_parties);
Mark D. Rothc0febd32018-01-09 10:25:24 -0800555 grpc_lb_policy_shutdown_locked(chand->lb_policy, new_lb_policy);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800556 GRPC_LB_POLICY_UNREF(chand->lb_policy, "channel");
Craig Tiller45724b32015-09-22 10:42:19 -0700557 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700558 chand->lb_policy = new_lb_policy;
559 }
560 // Now that we've swapped out the relevant fields of chand, check for
561 // error or shutdown.
Craig Tiller4782d922017-11-10 09:53:21 -0800562 if (error != GRPC_ERROR_NONE || chand->resolver == nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700563 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700564 gpr_log(GPR_DEBUG, "chand=%p: shutting down", chand);
565 }
Craig Tiller4782d922017-11-10 09:53:21 -0800566 if (chand->resolver != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700567 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700568 gpr_log(GPR_DEBUG, "chand=%p: shutting down resolver", chand);
569 }
Mark D. Roth209f6442018-02-08 10:26:46 -0800570 chand->resolver.reset();
Craig Tiller76a5c0e2016-03-09 09:05:30 -0800571 }
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800572 set_channel_connectivity_state_locked(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800573 chand, GRPC_CHANNEL_SHUTDOWN,
ncteisen4b36a3d2017-03-13 19:08:06 -0700574 GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING(
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700575 "Got resolver result after disconnection", &error, 1),
Craig Tiller804ff712016-05-05 16:25:40 -0700576 "resolver_gone");
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800577 GRPC_CHANNEL_STACK_UNREF(chand->owning_stack, "resolver");
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700578 grpc_closure_list_fail_all(&chand->waiting_for_resolver_result_closures,
579 GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING(
580 "Channel disconnected", &error, 1));
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800581 GRPC_CLOSURE_LIST_SCHED(&chand->waiting_for_resolver_result_closures);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700582 } else { // Not shutting down.
583 grpc_connectivity_state state = GRPC_CHANNEL_TRANSIENT_FAILURE;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700584 grpc_error* state_error =
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700585 GRPC_ERROR_CREATE_FROM_STATIC_STRING("No load balancing policy");
Craig Tiller4782d922017-11-10 09:53:21 -0800586 if (new_lb_policy != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700587 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700588 gpr_log(GPR_DEBUG, "chand=%p: initializing new LB policy", chand);
589 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700590 GRPC_ERROR_UNREF(state_error);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800591 state =
592 grpc_lb_policy_check_connectivity_locked(new_lb_policy, &state_error);
593 grpc_pollset_set_add_pollset_set(new_lb_policy->interested_parties,
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700594 chand->interested_parties);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800595 GRPC_CLOSURE_LIST_SCHED(&chand->waiting_for_resolver_result_closures);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700596 if (chand->exit_idle_when_lb_policy_arrives) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800597 grpc_lb_policy_exit_idle_locked(new_lb_policy);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700598 chand->exit_idle_when_lb_policy_arrives = false;
599 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800600 watch_lb_policy_locked(chand, new_lb_policy, state);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700601 }
Mark D. Roth15494b52017-07-12 15:26:55 -0700602 if (!lb_policy_updated) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800603 set_channel_connectivity_state_locked(
604 chand, state, GRPC_ERROR_REF(state_error), "new_lb+resolver");
Mark D. Roth15494b52017-07-12 15:26:55 -0700605 }
Mark D. Roth209f6442018-02-08 10:26:46 -0800606 chand->resolver->NextLocked(&chand->resolver_result,
607 &chand->on_resolver_result_changed);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700608 GRPC_ERROR_UNREF(state_error);
Craig Tillera82950e2015-09-22 12:33:20 -0700609 }
Craig Tiller3f475422015-06-25 10:43:05 -0700610}
611
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800612static void start_transport_op_locked(void* arg, grpc_error* error_ignored) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800613 grpc_transport_op* op = static_cast<grpc_transport_op*>(arg);
Craig Tillerbaa14a92017-11-03 09:09:36 -0700614 grpc_channel_element* elem =
Noah Eisenbe82e642018-02-09 09:16:55 -0800615 static_cast<grpc_channel_element*>(op->handler_private.extra_arg);
616 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Craig Tiller000cd8f2015-09-18 07:20:29 -0700617
Craig Tiller4782d922017-11-10 09:53:21 -0800618 if (op->on_connectivity_state_change != nullptr) {
Craig Tillera82950e2015-09-22 12:33:20 -0700619 grpc_connectivity_state_notify_on_state_change(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800620 &chand->state_tracker, op->connectivity_state,
Craig Tillera82950e2015-09-22 12:33:20 -0700621 op->on_connectivity_state_change);
Craig Tiller4782d922017-11-10 09:53:21 -0800622 op->on_connectivity_state_change = nullptr;
623 op->connectivity_state = nullptr;
Craig Tillera82950e2015-09-22 12:33:20 -0700624 }
625
Yuchen Zengc272dd72017-12-05 12:18:34 -0800626 if (op->send_ping.on_initiate != nullptr || op->send_ping.on_ack != nullptr) {
Craig Tiller4782d922017-11-10 09:53:21 -0800627 if (chand->lb_policy == nullptr) {
ncteisen274bbbe2017-06-08 14:57:11 -0700628 GRPC_CLOSURE_SCHED(
Yash Tibrewald6c292f2017-12-07 19:38:43 -0800629 op->send_ping.on_initiate,
Yuchen Zengc272dd72017-12-05 12:18:34 -0800630 GRPC_ERROR_CREATE_FROM_STATIC_STRING("Ping with no load balancing"));
631 GRPC_CLOSURE_SCHED(
Yash Tibrewald6c292f2017-12-07 19:38:43 -0800632 op->send_ping.on_ack,
ncteisen4b36a3d2017-03-13 19:08:06 -0700633 GRPC_ERROR_CREATE_FROM_STATIC_STRING("Ping with no load balancing"));
Craig Tiller26dab312015-12-07 14:43:47 -0800634 } else {
Yash Tibrewald6c292f2017-12-07 19:38:43 -0800635 grpc_lb_policy_ping_one_locked(
636 chand->lb_policy, op->send_ping.on_initiate, op->send_ping.on_ack);
Craig Tiller4782d922017-11-10 09:53:21 -0800637 op->bind_pollset = nullptr;
Craig Tiller26dab312015-12-07 14:43:47 -0800638 }
Yuchen Zengc272dd72017-12-05 12:18:34 -0800639 op->send_ping.on_initiate = nullptr;
640 op->send_ping.on_ack = nullptr;
Craig Tiller26dab312015-12-07 14:43:47 -0800641 }
642
Craig Tiller1c51edc2016-05-07 16:18:43 -0700643 if (op->disconnect_with_error != GRPC_ERROR_NONE) {
Craig Tiller4782d922017-11-10 09:53:21 -0800644 if (chand->resolver != nullptr) {
Craig Tiller1c51edc2016-05-07 16:18:43 -0700645 set_channel_connectivity_state_locked(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800646 chand, GRPC_CHANNEL_SHUTDOWN,
Craig Tiller1c51edc2016-05-07 16:18:43 -0700647 GRPC_ERROR_REF(op->disconnect_with_error), "disconnect");
Mark D. Roth209f6442018-02-08 10:26:46 -0800648 chand->resolver.reset();
Craig Tiller1c51edc2016-05-07 16:18:43 -0700649 if (!chand->started_resolving) {
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700650 grpc_closure_list_fail_all(&chand->waiting_for_resolver_result_closures,
Craig Tiller1c51edc2016-05-07 16:18:43 -0700651 GRPC_ERROR_REF(op->disconnect_with_error));
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800652 GRPC_CLOSURE_LIST_SCHED(&chand->waiting_for_resolver_result_closures);
Craig Tiller1c51edc2016-05-07 16:18:43 -0700653 }
Craig Tiller4782d922017-11-10 09:53:21 -0800654 if (chand->lb_policy != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800655 grpc_pollset_set_del_pollset_set(chand->lb_policy->interested_parties,
Craig Tiller1c51edc2016-05-07 16:18:43 -0700656 chand->interested_parties);
Mark D. Rothc0febd32018-01-09 10:25:24 -0800657 grpc_lb_policy_shutdown_locked(chand->lb_policy, nullptr);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800658 GRPC_LB_POLICY_UNREF(chand->lb_policy, "channel");
Craig Tiller4782d922017-11-10 09:53:21 -0800659 chand->lb_policy = nullptr;
Craig Tiller1c51edc2016-05-07 16:18:43 -0700660 }
Craig Tillerb12d22a2016-04-23 12:50:21 -0700661 }
Craig Tiller1c51edc2016-05-07 16:18:43 -0700662 GRPC_ERROR_UNREF(op->disconnect_with_error);
Craig Tillera82950e2015-09-22 12:33:20 -0700663 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800664 GRPC_CHANNEL_STACK_UNREF(chand->owning_stack, "start_transport_op");
Craig Tillerd2e5cfc2017-02-09 13:02:20 -0800665
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800666 GRPC_CLOSURE_SCHED(op->on_consumed, GRPC_ERROR_NONE);
Craig Tillerbefafe62017-02-09 11:30:54 -0800667}
668
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800669static void cc_start_transport_op(grpc_channel_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700670 grpc_transport_op* op) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800671 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Craig Tillerbefafe62017-02-09 11:30:54 -0800672
Craig Tillerbefafe62017-02-09 11:30:54 -0800673 GPR_ASSERT(op->set_accept_stream == false);
Craig Tiller4782d922017-11-10 09:53:21 -0800674 if (op->bind_pollset != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800675 grpc_pollset_set_add_pollset(chand->interested_parties, op->bind_pollset);
Craig Tillerbefafe62017-02-09 11:30:54 -0800676 }
677
Craig Tillerc55c1022017-03-10 10:26:42 -0800678 op->handler_private.extra_arg = elem;
Craig Tillerd2e5cfc2017-02-09 13:02:20 -0800679 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "start_transport_op");
ncteisen274bbbe2017-06-08 14:57:11 -0700680 GRPC_CLOSURE_SCHED(
ncteisen274bbbe2017-06-08 14:57:11 -0700681 GRPC_CLOSURE_INIT(&op->handler_private.closure, start_transport_op_locked,
Craig Tilleree4b1452017-05-12 10:56:03 -0700682 op, grpc_combiner_scheduler(chand->combiner)),
Craig Tillerbefafe62017-02-09 11:30:54 -0800683 GRPC_ERROR_NONE);
Craig Tillerca3e9d32015-06-27 18:37:27 -0700684}
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -0800685
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800686static void cc_get_channel_info(grpc_channel_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700687 const grpc_channel_info* info) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800688 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Craig Tiller613dafa2017-02-09 12:00:43 -0800689 gpr_mu_lock(&chand->info_mu);
Craig Tiller4782d922017-11-10 09:53:21 -0800690 if (info->lb_policy_name != nullptr) {
691 *info->lb_policy_name = chand->info_lb_policy_name == nullptr
692 ? nullptr
Craig Tiller613dafa2017-02-09 12:00:43 -0800693 : gpr_strdup(chand->info_lb_policy_name);
Mark D. Rothb2d24882016-10-27 15:44:07 -0700694 }
Craig Tiller4782d922017-11-10 09:53:21 -0800695 if (info->service_config_json != nullptr) {
Craig Tiller613dafa2017-02-09 12:00:43 -0800696 *info->service_config_json =
Craig Tiller4782d922017-11-10 09:53:21 -0800697 chand->info_service_config_json == nullptr
698 ? nullptr
Craig Tiller613dafa2017-02-09 12:00:43 -0800699 : gpr_strdup(chand->info_service_config_json);
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800700 }
Craig Tiller613dafa2017-02-09 12:00:43 -0800701 gpr_mu_unlock(&chand->info_mu);
Mark D. Rothb2d24882016-10-27 15:44:07 -0700702}
703
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700704/* Constructor for channel_data */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800705static grpc_error* cc_init_channel_elem(grpc_channel_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700706 grpc_channel_element_args* args) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800707 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700708 GPR_ASSERT(args->is_last);
709 GPR_ASSERT(elem->filter == &grpc_client_channel_filter);
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800710 // Initialize data members.
Craig Tilleree4b1452017-05-12 10:56:03 -0700711 chand->combiner = grpc_combiner_create();
Craig Tillerd85477512017-02-09 12:02:39 -0800712 gpr_mu_init(&chand->info_mu);
Alexander Polcync3b1f182017-04-18 13:51:36 -0700713 gpr_mu_init(&chand->external_connectivity_watcher_list_mu);
714
715 gpr_mu_lock(&chand->external_connectivity_watcher_list_mu);
Craig Tiller4782d922017-11-10 09:53:21 -0800716 chand->external_connectivity_watcher_list_head = nullptr;
Alexander Polcync3b1f182017-04-18 13:51:36 -0700717 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
718
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800719 chand->owning_stack = args->channel_stack;
ncteisen274bbbe2017-06-08 14:57:11 -0700720 GRPC_CLOSURE_INIT(&chand->on_resolver_result_changed,
Craig Tillerbefafe62017-02-09 11:30:54 -0800721 on_resolver_result_changed_locked, chand,
Craig Tilleree4b1452017-05-12 10:56:03 -0700722 grpc_combiner_scheduler(chand->combiner));
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800723 chand->interested_parties = grpc_pollset_set_create();
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700724 grpc_connectivity_state_init(&chand->state_tracker, GRPC_CHANNEL_IDLE,
725 "client_channel");
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800726 grpc_client_channel_start_backup_polling(chand->interested_parties);
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800727 // Record client channel factory.
Craig Tillerbaa14a92017-11-03 09:09:36 -0700728 const grpc_arg* arg = grpc_channel_args_find(args->channel_args,
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800729 GRPC_ARG_CLIENT_CHANNEL_FACTORY);
Craig Tiller4782d922017-11-10 09:53:21 -0800730 if (arg == nullptr) {
David Garcia Quintas228a5142017-03-30 19:43:00 -0700731 return GRPC_ERROR_CREATE_FROM_STATIC_STRING(
732 "Missing client channel factory in args for client channel filter");
733 }
734 if (arg->type != GRPC_ARG_POINTER) {
735 return GRPC_ERROR_CREATE_FROM_STATIC_STRING(
736 "client channel factory arg must be a pointer");
737 }
Yash Tibrewalbc130da2017-09-12 22:44:08 -0700738 grpc_client_channel_factory_ref(
Noah Eisenbe82e642018-02-09 09:16:55 -0800739 static_cast<grpc_client_channel_factory*>(arg->value.pointer.p));
Yash Tibrewalca3c1c02017-09-07 22:47:16 -0700740 chand->client_channel_factory =
Noah Eisenbe82e642018-02-09 09:16:55 -0800741 static_cast<grpc_client_channel_factory*>(arg->value.pointer.p);
Mark D. Rothdc9bee72017-02-07 12:29:14 -0800742 // Get server name to resolve, using proxy mapper if needed.
Mark D. Roth86e90592016-11-18 09:56:40 -0800743 arg = grpc_channel_args_find(args->channel_args, GRPC_ARG_SERVER_URI);
Craig Tiller4782d922017-11-10 09:53:21 -0800744 if (arg == nullptr) {
David Garcia Quintas228a5142017-03-30 19:43:00 -0700745 return GRPC_ERROR_CREATE_FROM_STATIC_STRING(
746 "Missing server uri in args for client channel filter");
747 }
748 if (arg->type != GRPC_ARG_STRING) {
749 return GRPC_ERROR_CREATE_FROM_STATIC_STRING(
750 "server uri arg must be a string");
751 }
Craig Tiller4782d922017-11-10 09:53:21 -0800752 char* proxy_name = nullptr;
753 grpc_channel_args* new_args = nullptr;
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800754 grpc_proxy_mappers_map_name(arg->value.string, args->channel_args,
Mark D. Rothdc9bee72017-02-07 12:29:14 -0800755 &proxy_name, &new_args);
756 // Instantiate resolver.
Mark D. Roth209f6442018-02-08 10:26:46 -0800757 chand->resolver = grpc_core::ResolverRegistry::CreateResolver(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800758 proxy_name != nullptr ? proxy_name : arg->value.string,
Craig Tiller4782d922017-11-10 09:53:21 -0800759 new_args != nullptr ? new_args : args->channel_args,
Craig Tiller972470b2017-02-09 15:05:36 -0800760 chand->interested_parties, chand->combiner);
Craig Tiller4782d922017-11-10 09:53:21 -0800761 if (proxy_name != nullptr) gpr_free(proxy_name);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800762 if (new_args != nullptr) grpc_channel_args_destroy(new_args);
Craig Tiller4782d922017-11-10 09:53:21 -0800763 if (chand->resolver == nullptr) {
ncteisen4b36a3d2017-03-13 19:08:06 -0700764 return GRPC_ERROR_CREATE_FROM_STATIC_STRING("resolver creation failed");
Mark D. Roth5e2566e2016-11-18 10:53:13 -0800765 }
Craig Tiller3be7dd02017-04-03 14:30:03 -0700766 chand->deadline_checking_enabled =
767 grpc_deadline_checking_enabled(args->channel_args);
Mark D. Roth5e2566e2016-11-18 10:53:13 -0800768 return GRPC_ERROR_NONE;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700769}
770
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800771static void shutdown_resolver_locked(void* arg, grpc_error* error) {
Mark D. Roth209f6442018-02-08 10:26:46 -0800772 grpc_core::Resolver* resolver = static_cast<grpc_core::Resolver*>(arg);
773 resolver->Orphan();
Craig Tiller972470b2017-02-09 15:05:36 -0800774}
775
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700776/* Destructor for channel_data */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800777static void cc_destroy_channel_elem(grpc_channel_element* elem) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800778 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Craig Tiller4782d922017-11-10 09:53:21 -0800779 if (chand->resolver != nullptr) {
ncteisen274bbbe2017-06-08 14:57:11 -0700780 GRPC_CLOSURE_SCHED(
Mark D. Roth209f6442018-02-08 10:26:46 -0800781 GRPC_CLOSURE_CREATE(shutdown_resolver_locked, chand->resolver.release(),
Yash Tibrewal0ee75742017-10-13 16:07:13 -0700782 grpc_combiner_scheduler(chand->combiner)),
Craig Tiller972470b2017-02-09 15:05:36 -0800783 GRPC_ERROR_NONE);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700784 }
Craig Tiller4782d922017-11-10 09:53:21 -0800785 if (chand->client_channel_factory != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800786 grpc_client_channel_factory_unref(chand->client_channel_factory);
Mark D. Roth0e48a9a2016-09-08 14:14:39 -0700787 }
Craig Tiller4782d922017-11-10 09:53:21 -0800788 if (chand->lb_policy != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800789 grpc_pollset_set_del_pollset_set(chand->lb_policy->interested_parties,
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700790 chand->interested_parties);
Mark D. Rothc0febd32018-01-09 10:25:24 -0800791 grpc_lb_policy_shutdown_locked(chand->lb_policy, nullptr);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800792 GRPC_LB_POLICY_UNREF(chand->lb_policy, "channel");
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700793 }
Craig Tiller613dafa2017-02-09 12:00:43 -0800794 gpr_free(chand->info_lb_policy_name);
795 gpr_free(chand->info_service_config_json);
Craig Tiller4782d922017-11-10 09:53:21 -0800796 if (chand->retry_throttle_data != nullptr) {
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800797 grpc_server_retry_throttle_data_unref(chand->retry_throttle_data);
798 }
Craig Tiller4782d922017-11-10 09:53:21 -0800799 if (chand->method_params_table != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800800 grpc_slice_hash_table_unref(chand->method_params_table);
Mark D. Roth9fe284e2016-09-12 11:22:27 -0700801 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800802 grpc_client_channel_stop_backup_polling(chand->interested_parties);
803 grpc_connectivity_state_destroy(&chand->state_tracker);
804 grpc_pollset_set_destroy(chand->interested_parties);
805 GRPC_COMBINER_UNREF(chand->combiner, "client_channel");
Craig Tillerd85477512017-02-09 12:02:39 -0800806 gpr_mu_destroy(&chand->info_mu);
Alexander Polcync3b1f182017-04-18 13:51:36 -0700807 gpr_mu_destroy(&chand->external_connectivity_watcher_list_mu);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700808}
809
810/*************************************************************************
811 * PER-CALL FUNCTIONS
812 */
813
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700814// Max number of batches that can be pending on a call at any given
815// time. This includes:
816// recv_initial_metadata
817// send_initial_metadata
818// recv_message
819// send_message
820// recv_trailing_metadata
821// send_trailing_metadata
Mark D. Roth76e264b2017-08-25 09:03:33 -0700822// We also add room for a single cancel_stream batch.
823#define MAX_WAITING_BATCHES 7
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700824
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700825/** Call data. Holds a pointer to grpc_subchannel_call and the
826 associated machinery to create such a pointer.
827 Handles queueing of stream ops until a call object is ready, waiting
828 for initial metadata before trying to create a call object,
829 and handling cancellation gracefully. */
830typedef struct client_channel_call_data {
Mark D. Roth72f6da82016-09-02 13:42:38 -0700831 // State for handling deadlines.
832 // The code in deadline_filter.c requires this to be the first field.
Mark D. Roth72f6da82016-09-02 13:42:38 -0700833 // TODO(roth): This is slightly sub-optimal in that grpc_deadline_state
Mark D. Roth66f3d2b2017-09-01 09:02:17 -0700834 // and this struct both independently store pointers to the call stack
835 // and call combiner. If/when we have time, find a way to avoid this
836 // without breaking the grpc_deadline_state abstraction.
Mark D. Roth72f6da82016-09-02 13:42:38 -0700837 grpc_deadline_state deadline_state;
Mark D. Rothf28763c2016-09-14 15:18:40 -0700838
Craig Tiller7c70b6c2017-01-23 07:48:42 -0800839 grpc_slice path; // Request path.
Mark D. Rothe40dd292016-10-05 14:58:37 -0700840 gpr_timespec call_start_time;
Craig Tiller89c14282017-07-19 15:32:27 -0700841 grpc_millis deadline;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700842 gpr_arena* arena;
843 grpc_call_stack* owning_call;
844 grpc_call_combiner* call_combiner;
Mark D. Roth76e264b2017-08-25 09:03:33 -0700845
Craig Tillerbaa14a92017-11-03 09:09:36 -0700846 grpc_server_retry_throttle_data* retry_throttle_data;
847 method_parameters* method_params;
Mark D. Rothaa850a72016-09-26 13:38:02 -0700848
Craig Tillerbaa14a92017-11-03 09:09:36 -0700849 grpc_subchannel_call* subchannel_call;
850 grpc_error* error;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700851
Mark D. Rothc0febd32018-01-09 10:25:24 -0800852 grpc_lb_policy_pick_state pick;
Mark D. Roth60751fe2017-07-07 12:50:33 -0700853 grpc_closure lb_pick_closure;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -0700854 grpc_closure lb_pick_cancel_closure;
Mark D. Roth60751fe2017-07-07 12:50:33 -0700855
Craig Tillerbaa14a92017-11-03 09:09:36 -0700856 grpc_polling_entity* pollent;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700857
Craig Tillerbaa14a92017-11-03 09:09:36 -0700858 grpc_transport_stream_op_batch* waiting_for_pick_batches[MAX_WAITING_BATCHES];
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700859 size_t waiting_for_pick_batches_count;
Mark D. Roth76e264b2017-08-25 09:03:33 -0700860 grpc_closure handle_pending_batch_in_call_combiner[MAX_WAITING_BATCHES];
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700861
Craig Tillerbaa14a92017-11-03 09:09:36 -0700862 grpc_transport_stream_op_batch* initial_metadata_batch;
David Garcia Quintasd1a47f12016-09-02 12:46:44 +0200863
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800864 grpc_closure on_complete;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700865 grpc_closure* original_on_complete;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700866} call_data;
867
Craig Tillerbaa14a92017-11-03 09:09:36 -0700868grpc_subchannel_call* grpc_client_channel_get_subchannel_call(
869 grpc_call_element* elem) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800870 call_data* calld = static_cast<call_data*>(elem->call_data);
Mark D. Roth76e264b2017-08-25 09:03:33 -0700871 return calld->subchannel_call;
Craig Tiller8b1d59c2016-12-27 15:15:30 -0800872}
873
Mark D. Roth76e264b2017-08-25 09:03:33 -0700874// This is called via the call combiner, so access to calld is synchronized.
875static void waiting_for_pick_batches_add(
Craig Tillerbaa14a92017-11-03 09:09:36 -0700876 call_data* calld, grpc_transport_stream_op_batch* batch) {
Mark D. Roth76e264b2017-08-25 09:03:33 -0700877 if (batch->send_initial_metadata) {
Craig Tiller4782d922017-11-10 09:53:21 -0800878 GPR_ASSERT(calld->initial_metadata_batch == nullptr);
Mark D. Roth76e264b2017-08-25 09:03:33 -0700879 calld->initial_metadata_batch = batch;
880 } else {
881 GPR_ASSERT(calld->waiting_for_pick_batches_count < MAX_WAITING_BATCHES);
882 calld->waiting_for_pick_batches[calld->waiting_for_pick_batches_count++] =
883 batch;
884 }
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700885}
886
Mark D. Roth76e264b2017-08-25 09:03:33 -0700887// This is called via the call combiner, so access to calld is synchronized.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800888static void fail_pending_batch_in_call_combiner(void* arg, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800889 call_data* calld = static_cast<call_data*>(arg);
Mark D. Roth76e264b2017-08-25 09:03:33 -0700890 if (calld->waiting_for_pick_batches_count > 0) {
891 --calld->waiting_for_pick_batches_count;
892 grpc_transport_stream_op_batch_finish_with_failure(
Mark D. Roth76e264b2017-08-25 09:03:33 -0700893 calld->waiting_for_pick_batches[calld->waiting_for_pick_batches_count],
894 GRPC_ERROR_REF(error), calld->call_combiner);
895 }
896}
897
898// This is called via the call combiner, so access to calld is synchronized.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800899static void waiting_for_pick_batches_fail(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700900 grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800901 call_data* calld = static_cast<call_data*>(elem->call_data);
Craig Tiller6014e8a2017-10-16 13:50:29 -0700902 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700903 gpr_log(GPR_DEBUG,
Mark D. Rothe9b10832017-10-26 13:18:25 -0700904 "chand=%p calld=%p: failing %" PRIuPTR " pending batches: %s",
Mark D. Roth60751fe2017-07-07 12:50:33 -0700905 elem->channel_data, calld, calld->waiting_for_pick_batches_count,
906 grpc_error_string(error));
907 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700908 for (size_t i = 0; i < calld->waiting_for_pick_batches_count; ++i) {
Mark D. Roth76e264b2017-08-25 09:03:33 -0700909 GRPC_CLOSURE_INIT(&calld->handle_pending_batch_in_call_combiner[i],
910 fail_pending_batch_in_call_combiner, calld,
911 grpc_schedule_on_exec_ctx);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800912 GRPC_CALL_COMBINER_START(
913 calld->call_combiner, &calld->handle_pending_batch_in_call_combiner[i],
914 GRPC_ERROR_REF(error), "waiting_for_pick_batches_fail");
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700915 }
Craig Tiller4782d922017-11-10 09:53:21 -0800916 if (calld->initial_metadata_batch != nullptr) {
Mark D. Roth76e264b2017-08-25 09:03:33 -0700917 grpc_transport_stream_op_batch_finish_with_failure(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800918 calld->initial_metadata_batch, GRPC_ERROR_REF(error),
Mark D. Roth76e264b2017-08-25 09:03:33 -0700919 calld->call_combiner);
920 } else {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800921 GRPC_CALL_COMBINER_STOP(calld->call_combiner,
Mark D. Roth76e264b2017-08-25 09:03:33 -0700922 "waiting_for_pick_batches_fail");
923 }
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700924 GRPC_ERROR_UNREF(error);
925}
926
Mark D. Roth76e264b2017-08-25 09:03:33 -0700927// This is called via the call combiner, so access to calld is synchronized.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800928static void run_pending_batch_in_call_combiner(void* arg, grpc_error* ignored) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800929 call_data* calld = static_cast<call_data*>(arg);
Mark D. Roth76e264b2017-08-25 09:03:33 -0700930 if (calld->waiting_for_pick_batches_count > 0) {
931 --calld->waiting_for_pick_batches_count;
932 grpc_subchannel_call_process_op(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800933 calld->subchannel_call,
Mark D. Roth76e264b2017-08-25 09:03:33 -0700934 calld->waiting_for_pick_batches[calld->waiting_for_pick_batches_count]);
Craig Tiller57726ca2016-09-12 11:59:45 -0700935 }
Mark D. Roth76e264b2017-08-25 09:03:33 -0700936}
937
938// This is called via the call combiner, so access to calld is synchronized.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800939static void waiting_for_pick_batches_resume(grpc_call_element* elem) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800940 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
941 call_data* calld = static_cast<call_data*>(elem->call_data);
Craig Tiller6014e8a2017-10-16 13:50:29 -0700942 if (grpc_client_channel_trace.enabled()) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700943 gpr_log(GPR_DEBUG,
944 "chand=%p calld=%p: sending %" PRIuPTR
945 " pending batches to subchannel_call=%p",
Mark D. Roth76e264b2017-08-25 09:03:33 -0700946 chand, calld, calld->waiting_for_pick_batches_count,
947 calld->subchannel_call);
Mark D. Roth60751fe2017-07-07 12:50:33 -0700948 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700949 for (size_t i = 0; i < calld->waiting_for_pick_batches_count; ++i) {
Mark D. Roth76e264b2017-08-25 09:03:33 -0700950 GRPC_CLOSURE_INIT(&calld->handle_pending_batch_in_call_combiner[i],
951 run_pending_batch_in_call_combiner, calld,
952 grpc_schedule_on_exec_ctx);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800953 GRPC_CALL_COMBINER_START(
954 calld->call_combiner, &calld->handle_pending_batch_in_call_combiner[i],
955 GRPC_ERROR_NONE, "waiting_for_pick_batches_resume");
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700956 }
Craig Tiller4782d922017-11-10 09:53:21 -0800957 GPR_ASSERT(calld->initial_metadata_batch != nullptr);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800958 grpc_subchannel_call_process_op(calld->subchannel_call,
Mark D. Roth76e264b2017-08-25 09:03:33 -0700959 calld->initial_metadata_batch);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700960}
961
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700962// Applies service config to the call. Must be invoked once we know
963// that the resolver has returned results to the channel.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800964static void apply_service_config_to_call_locked(grpc_call_element* elem) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800965 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
966 call_data* calld = static_cast<call_data*>(elem->call_data);
Craig Tiller6014e8a2017-10-16 13:50:29 -0700967 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700968 gpr_log(GPR_DEBUG, "chand=%p calld=%p: applying service config to call",
969 chand, calld);
970 }
Craig Tiller4782d922017-11-10 09:53:21 -0800971 if (chand->retry_throttle_data != nullptr) {
Mark D. Roth9ccbc4d2017-03-15 08:30:04 -0700972 calld->retry_throttle_data =
973 grpc_server_retry_throttle_data_ref(chand->retry_throttle_data);
974 }
Craig Tiller4782d922017-11-10 09:53:21 -0800975 if (chand->method_params_table != nullptr) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800976 calld->method_params = static_cast<method_parameters*>(grpc_method_config_table_get(
977 chand->method_params_table, calld->path));
Craig Tiller4782d922017-11-10 09:53:21 -0800978 if (calld->method_params != nullptr) {
Craig Tiller11c17d42017-03-13 13:36:34 -0700979 method_parameters_ref(calld->method_params);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700980 // If the deadline from the service config is shorter than the one
981 // from the client API, reset the deadline timer.
982 if (chand->deadline_checking_enabled &&
Craig Tiller89c14282017-07-19 15:32:27 -0700983 calld->method_params->timeout != 0) {
984 const grpc_millis per_method_deadline =
Craig Tiller9a8c3f32017-07-21 13:14:14 -0700985 grpc_timespec_to_millis_round_up(calld->call_start_time) +
Craig Tiller89c14282017-07-19 15:32:27 -0700986 calld->method_params->timeout;
987 if (per_method_deadline < calld->deadline) {
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700988 calld->deadline = per_method_deadline;
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800989 grpc_deadline_state_reset(elem, calld->deadline);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700990 }
Craig Tiller11c17d42017-03-13 13:36:34 -0700991 }
992 }
993 }
Craig Tiller11c17d42017-03-13 13:36:34 -0700994}
Craig Tillerea4a4f12017-03-13 13:36:52 -0700995
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800996static void create_subchannel_call_locked(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700997 grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -0800998 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
999 call_data* calld = static_cast<call_data*>(elem->call_data);
David Garcia Quintasbaf1ac72018-01-09 14:24:32 -08001000 const grpc_core::ConnectedSubchannel::CallArgs call_args = {
Mark D. Rothc0febd32018-01-09 10:25:24 -08001001 calld->pollent, // pollent
1002 calld->path, // path
1003 calld->call_start_time, // start_time
1004 calld->deadline, // deadline
1005 calld->arena, // arena
1006 calld->pick.subchannel_call_context, // context
1007 calld->call_combiner // call_combiner
Yash Tibrewald8b84a22017-09-25 13:38:03 -07001008 };
David Garcia Quintas70fbe622018-01-09 19:27:46 -08001009 grpc_error* new_error = calld->pick.connected_subchannel->CreateCall(
David Garcia Quintasbaf1ac72018-01-09 14:24:32 -08001010 call_args, &calld->subchannel_call);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001011 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001012 gpr_log(GPR_DEBUG, "chand=%p calld=%p: create subchannel_call=%p: error=%s",
Mark D. Roth76e264b2017-08-25 09:03:33 -07001013 chand, calld, calld->subchannel_call, grpc_error_string(new_error));
Mark D. Roth60751fe2017-07-07 12:50:33 -07001014 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001015 if (new_error != GRPC_ERROR_NONE) {
1016 new_error = grpc_error_add_child(new_error, error);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001017 waiting_for_pick_batches_fail(elem, new_error);
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001018 } else {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001019 waiting_for_pick_batches_resume(elem);
Craig Tiller11c17d42017-03-13 13:36:34 -07001020 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001021 GRPC_ERROR_UNREF(error);
Craig Tiller11c17d42017-03-13 13:36:34 -07001022}
1023
Mark D. Rothb2929602017-09-11 09:31:11 -07001024// Invoked when a pick is completed, on both success or failure.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001025static void pick_done_locked(grpc_call_element* elem, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001026 call_data* calld = static_cast<call_data*>(elem->call_data);
1027 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
David Garcia Quintasbe1b7f92018-01-12 14:01:38 -08001028 if (calld->pick.connected_subchannel == nullptr) {
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001029 // Failed to create subchannel.
Mark D. Roth76e264b2017-08-25 09:03:33 -07001030 GRPC_ERROR_UNREF(calld->error);
1031 calld->error = error == GRPC_ERROR_NONE
1032 ? GRPC_ERROR_CREATE_FROM_STATIC_STRING(
1033 "Call dropped by load balancing policy")
1034 : GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING(
1035 "Failed to create subchannel", &error, 1);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001036 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001037 gpr_log(GPR_DEBUG,
1038 "chand=%p calld=%p: failed to create subchannel: error=%s", chand,
Mark D. Roth76e264b2017-08-25 09:03:33 -07001039 calld, grpc_error_string(calld->error));
Mark D. Roth60751fe2017-07-07 12:50:33 -07001040 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001041 waiting_for_pick_batches_fail(elem, GRPC_ERROR_REF(calld->error));
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001042 } else {
Mark D. Roth9fe284e2016-09-12 11:22:27 -07001043 /* Create call on subchannel. */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001044 create_subchannel_call_locked(elem, GRPC_ERROR_REF(error));
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001045 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001046 GRPC_ERROR_UNREF(error);
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001047}
1048
Mark D. Rothb2929602017-09-11 09:31:11 -07001049// A wrapper around pick_done_locked() that is used in cases where
1050// either (a) the pick was deferred pending a resolver result or (b) the
1051// pick was done asynchronously. Removes the call's polling entity from
1052// chand->interested_parties before invoking pick_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001053static void async_pick_done_locked(grpc_call_element* elem, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001054 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
1055 call_data* calld = static_cast<call_data*>(elem->call_data);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001056 grpc_polling_entity_del_from_pollset_set(calld->pollent,
Mark D. Rothb2929602017-09-11 09:31:11 -07001057 chand->interested_parties);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001058 pick_done_locked(elem, error);
Mark D. Rothb2929602017-09-11 09:31:11 -07001059}
1060
1061// Note: This runs under the client_channel combiner, but will NOT be
1062// holding the call combiner.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001063static void pick_callback_cancel_locked(void* arg, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001064 grpc_call_element* elem = static_cast<grpc_call_element*>(arg);
1065 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
1066 call_data* calld = static_cast<call_data*>(elem->call_data);
Mark D. Rothc0febd32018-01-09 10:25:24 -08001067 // Note: chand->lb_policy may have changed since we started our pick,
1068 // in which case we will be cancelling the pick on a policy other than
1069 // the one we started it on. However, this will just be a no-op.
1070 if (error != GRPC_ERROR_NONE && chand->lb_policy != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001071 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001072 gpr_log(GPR_DEBUG, "chand=%p calld=%p: cancelling pick from LB policy %p",
Mark D. Rothc0febd32018-01-09 10:25:24 -08001073 chand, calld, chand->lb_policy);
Mark D. Rothb2929602017-09-11 09:31:11 -07001074 }
Mark D. Rothc0febd32018-01-09 10:25:24 -08001075 grpc_lb_policy_cancel_pick_locked(chand->lb_policy, &calld->pick,
1076 GRPC_ERROR_REF(error));
Mark D. Rothb2929602017-09-11 09:31:11 -07001077 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001078 GRPC_CALL_STACK_UNREF(calld->owning_call, "pick_callback_cancel");
Mark D. Rothb2929602017-09-11 09:31:11 -07001079}
1080
1081// Callback invoked by grpc_lb_policy_pick_locked() for async picks.
1082// Unrefs the LB policy and invokes async_pick_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001083static void pick_callback_done_locked(void* arg, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001084 grpc_call_element* elem = static_cast<grpc_call_element*>(arg);
1085 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
1086 call_data* calld = static_cast<call_data*>(elem->call_data);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001087 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001088 gpr_log(GPR_DEBUG, "chand=%p calld=%p: pick completed asynchronously",
1089 chand, calld);
1090 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001091 async_pick_done_locked(elem, GRPC_ERROR_REF(error));
Ken Paysonf069dd42018-02-05 09:15:05 -08001092 GRPC_CALL_STACK_UNREF(calld->owning_call, "pick_callback");
Mark D. Rothb2929602017-09-11 09:31:11 -07001093}
1094
1095// Takes a ref to chand->lb_policy and calls grpc_lb_policy_pick_locked().
1096// If the pick was completed synchronously, unrefs the LB policy and
1097// returns true.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001098static bool pick_callback_start_locked(grpc_call_element* elem) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001099 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
1100 call_data* calld = static_cast<call_data*>(elem->call_data);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001101 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001102 gpr_log(GPR_DEBUG, "chand=%p calld=%p: starting pick on lb_policy=%p",
1103 chand, calld, chand->lb_policy);
1104 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001105 apply_service_config_to_call_locked(elem);
Mark D. Rothb2929602017-09-11 09:31:11 -07001106 // If the application explicitly set wait_for_ready, use that.
1107 // Otherwise, if the service config specified a value for this
1108 // method, use that.
1109 uint32_t initial_metadata_flags =
1110 calld->initial_metadata_batch->payload->send_initial_metadata
1111 .send_initial_metadata_flags;
1112 const bool wait_for_ready_set_from_api =
1113 initial_metadata_flags &
1114 GRPC_INITIAL_METADATA_WAIT_FOR_READY_EXPLICITLY_SET;
1115 const bool wait_for_ready_set_from_service_config =
Craig Tiller4782d922017-11-10 09:53:21 -08001116 calld->method_params != nullptr &&
Mark D. Rothb2929602017-09-11 09:31:11 -07001117 calld->method_params->wait_for_ready != WAIT_FOR_READY_UNSET;
1118 if (!wait_for_ready_set_from_api && wait_for_ready_set_from_service_config) {
1119 if (calld->method_params->wait_for_ready == WAIT_FOR_READY_TRUE) {
1120 initial_metadata_flags |= GRPC_INITIAL_METADATA_WAIT_FOR_READY;
1121 } else {
1122 initial_metadata_flags &= ~GRPC_INITIAL_METADATA_WAIT_FOR_READY;
1123 }
1124 }
Mark D. Rothc0febd32018-01-09 10:25:24 -08001125 calld->pick.initial_metadata =
Mark D. Rothb2929602017-09-11 09:31:11 -07001126 calld->initial_metadata_batch->payload->send_initial_metadata
Mark D. Rothc0febd32018-01-09 10:25:24 -08001127 .send_initial_metadata;
1128 calld->pick.initial_metadata_flags = initial_metadata_flags;
Mark D. Rothb2929602017-09-11 09:31:11 -07001129 GRPC_CLOSURE_INIT(&calld->lb_pick_closure, pick_callback_done_locked, elem,
1130 grpc_combiner_scheduler(chand->combiner));
Mark D. Rothc0febd32018-01-09 10:25:24 -08001131 calld->pick.on_complete = &calld->lb_pick_closure;
Ken Paysonf069dd42018-02-05 09:15:05 -08001132 GRPC_CALL_STACK_REF(calld->owning_call, "pick_callback");
Mark D. Rothc0febd32018-01-09 10:25:24 -08001133 const bool pick_done =
1134 grpc_lb_policy_pick_locked(chand->lb_policy, &calld->pick);
Mark D. Rothb2929602017-09-11 09:31:11 -07001135 if (pick_done) {
1136 /* synchronous grpc_lb_policy_pick call. Unref the LB policy. */
Craig Tiller6014e8a2017-10-16 13:50:29 -07001137 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001138 gpr_log(GPR_DEBUG, "chand=%p calld=%p: pick completed synchronously",
1139 chand, calld);
1140 }
Ken Paysonf069dd42018-02-05 09:15:05 -08001141 GRPC_CALL_STACK_UNREF(calld->owning_call, "pick_callback");
Mark D. Rothb2929602017-09-11 09:31:11 -07001142 } else {
1143 GRPC_CALL_STACK_REF(calld->owning_call, "pick_callback_cancel");
1144 grpc_call_combiner_set_notify_on_cancel(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001145 calld->call_combiner,
Mark D. Rothb2929602017-09-11 09:31:11 -07001146 GRPC_CLOSURE_INIT(&calld->lb_pick_cancel_closure,
1147 pick_callback_cancel_locked, elem,
1148 grpc_combiner_scheduler(chand->combiner)));
1149 }
1150 return pick_done;
1151}
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001152
Craig Tiller577c9b22015-11-02 14:11:15 -08001153typedef struct {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001154 grpc_call_element* elem;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001155 bool finished;
Craig Tiller577c9b22015-11-02 14:11:15 -08001156 grpc_closure closure;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001157 grpc_closure cancel_closure;
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001158} pick_after_resolver_result_args;
Craig Tiller577c9b22015-11-02 14:11:15 -08001159
Mark D. Roth76e264b2017-08-25 09:03:33 -07001160// Note: This runs under the client_channel combiner, but will NOT be
1161// holding the call combiner.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001162static void pick_after_resolver_result_cancel_locked(void* arg,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001163 grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001164 pick_after_resolver_result_args* args = static_cast<pick_after_resolver_result_args*>(arg);
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001165 if (args->finished) {
1166 gpr_free(args);
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001167 return;
Mark D. Roth764cf042017-09-01 09:00:06 -07001168 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001169 // If we don't yet have a resolver result, then a closure for
1170 // pick_after_resolver_result_done_locked() will have been added to
1171 // chand->waiting_for_resolver_result_closures, and it may not be invoked
1172 // until after this call has been destroyed. We mark the operation as
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001173 // finished, so that when pick_after_resolver_result_done_locked()
Mark D. Roth76e264b2017-08-25 09:03:33 -07001174 // is called, it will be a no-op. We also immediately invoke
Mark D. Rothb2929602017-09-11 09:31:11 -07001175 // async_pick_done_locked() to propagate the error back to the caller.
1176 args->finished = true;
Craig Tillerbaa14a92017-11-03 09:09:36 -07001177 grpc_call_element* elem = args->elem;
Noah Eisenbe82e642018-02-09 09:16:55 -08001178 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
1179 call_data* calld = static_cast<call_data*>(elem->call_data);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001180 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001181 gpr_log(GPR_DEBUG,
1182 "chand=%p calld=%p: cancelling pick waiting for resolver result",
1183 chand, calld);
Mark D. Roth76e264b2017-08-25 09:03:33 -07001184 }
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001185 // Note: Although we are not in the call combiner here, we are
1186 // basically stealing the call combiner from the pending pick, so
Mark D. Rothb2929602017-09-11 09:31:11 -07001187 // it's safe to call async_pick_done_locked() here -- we are
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001188 // essentially calling it here instead of calling it in
1189 // pick_after_resolver_result_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001190 async_pick_done_locked(elem, GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING(
1191 "Pick cancelled", &error, 1));
Mark D. Roth76e264b2017-08-25 09:03:33 -07001192}
1193
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001194static void pick_after_resolver_result_start_locked(grpc_call_element* elem);
Mark D. Roth48be9de2017-10-23 12:27:37 -07001195
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001196static void pick_after_resolver_result_done_locked(void* arg,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001197 grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001198 pick_after_resolver_result_args* args = static_cast<pick_after_resolver_result_args*>(arg);
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001199 if (args->finished) {
Craig Tiller577c9b22015-11-02 14:11:15 -08001200 /* cancelled, do nothing */
Craig Tiller6014e8a2017-10-16 13:50:29 -07001201 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001202 gpr_log(GPR_DEBUG, "call cancelled before resolver result");
1203 }
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001204 gpr_free(args);
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001205 return;
1206 }
1207 args->finished = true;
Craig Tillerbaa14a92017-11-03 09:09:36 -07001208 grpc_call_element* elem = args->elem;
Noah Eisenbe82e642018-02-09 09:16:55 -08001209 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
1210 call_data* calld = static_cast<call_data*>(elem->call_data);
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001211 if (error != GRPC_ERROR_NONE) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001212 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001213 gpr_log(GPR_DEBUG, "chand=%p calld=%p: resolver failed to return data",
1214 chand, calld);
1215 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001216 async_pick_done_locked(elem, GRPC_ERROR_REF(error));
Craig Tiller4782d922017-11-10 09:53:21 -08001217 } else if (chand->lb_policy != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001218 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001219 gpr_log(GPR_DEBUG, "chand=%p calld=%p: resolver returned, doing pick",
1220 chand, calld);
1221 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001222 if (pick_callback_start_locked(elem)) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001223 // Even if the LB policy returns a result synchronously, we have
1224 // already added our polling entity to chand->interested_parties
1225 // in order to wait for the resolver result, so we need to
1226 // remove it here. Therefore, we call async_pick_done_locked()
1227 // instead of pick_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001228 async_pick_done_locked(elem, GRPC_ERROR_NONE);
Mark D. Roth9dab7d52016-10-07 07:48:03 -07001229 }
Craig Tiller577c9b22015-11-02 14:11:15 -08001230 }
Mark D. Roth48be9de2017-10-23 12:27:37 -07001231 // TODO(roth): It should be impossible for chand->lb_policy to be NULL
1232 // here, so the rest of this code should never actually be executed.
1233 // However, we have reports of a crash on iOS that triggers this case,
1234 // so we are temporarily adding this to restore branches that were
1235 // removed in https://github.com/grpc/grpc/pull/12297. Need to figure
1236 // out what is actually causing this to occur and then figure out the
1237 // right way to deal with it.
Craig Tiller4782d922017-11-10 09:53:21 -08001238 else if (chand->resolver != nullptr) {
Mark D. Roth48be9de2017-10-23 12:27:37 -07001239 // No LB policy, so try again.
ncteisen72afb762017-11-10 12:23:12 -08001240 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth48be9de2017-10-23 12:27:37 -07001241 gpr_log(GPR_DEBUG,
1242 "chand=%p calld=%p: resolver returned but no LB policy, "
1243 "trying again",
1244 chand, calld);
1245 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001246 pick_after_resolver_result_start_locked(elem);
Mark D. Roth48be9de2017-10-23 12:27:37 -07001247 } else {
ncteisen72afb762017-11-10 12:23:12 -08001248 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth48be9de2017-10-23 12:27:37 -07001249 gpr_log(GPR_DEBUG, "chand=%p calld=%p: resolver disconnected", chand,
1250 calld);
1251 }
1252 async_pick_done_locked(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001253 elem, GRPC_ERROR_CREATE_FROM_STATIC_STRING("Disconnected"));
Mark D. Roth48be9de2017-10-23 12:27:37 -07001254 }
Craig Tiller577c9b22015-11-02 14:11:15 -08001255}
1256
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001257static void pick_after_resolver_result_start_locked(grpc_call_element* elem) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001258 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
1259 call_data* calld = static_cast<call_data*>(elem->call_data);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001260 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001261 gpr_log(GPR_DEBUG,
1262 "chand=%p calld=%p: deferring pick pending resolver result", chand,
1263 calld);
Mark D. Roth64a317c2017-05-02 08:27:08 -07001264 }
Craig Tillerbaa14a92017-11-03 09:09:36 -07001265 pick_after_resolver_result_args* args =
Noah Eisenbe82e642018-02-09 09:16:55 -08001266 static_cast<pick_after_resolver_result_args*>(gpr_zalloc(sizeof(*args)));
Mark D. Roth60751fe2017-07-07 12:50:33 -07001267 args->elem = elem;
1268 GRPC_CLOSURE_INIT(&args->closure, pick_after_resolver_result_done_locked,
1269 args, grpc_combiner_scheduler(chand->combiner));
1270 grpc_closure_list_append(&chand->waiting_for_resolver_result_closures,
1271 &args->closure, GRPC_ERROR_NONE);
Mark D. Roth76e264b2017-08-25 09:03:33 -07001272 grpc_call_combiner_set_notify_on_cancel(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001273 calld->call_combiner,
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001274 GRPC_CLOSURE_INIT(&args->cancel_closure,
1275 pick_after_resolver_result_cancel_locked, args,
Mark D. Roth76e264b2017-08-25 09:03:33 -07001276 grpc_combiner_scheduler(chand->combiner)));
Mark D. Roth60751fe2017-07-07 12:50:33 -07001277}
1278
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001279static void start_pick_locked(void* arg, grpc_error* ignored) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001280 grpc_call_element* elem = static_cast<grpc_call_element*>(arg);
1281 call_data* calld = static_cast<call_data*>(elem->call_data);
1282 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
David Garcia Quintasbe1b7f92018-01-12 14:01:38 -08001283 GPR_ASSERT(calld->pick.connected_subchannel == nullptr);
Craig Tiller4782d922017-11-10 09:53:21 -08001284 if (chand->lb_policy != nullptr) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001285 // We already have an LB policy, so ask it for a pick.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001286 if (pick_callback_start_locked(elem)) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001287 // Pick completed synchronously.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001288 pick_done_locked(elem, GRPC_ERROR_NONE);
Mark D. Rothb2929602017-09-11 09:31:11 -07001289 return;
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001290 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001291 } else {
Mark D. Rothb2929602017-09-11 09:31:11 -07001292 // We do not yet have an LB policy, so wait for a resolver result.
Craig Tiller4782d922017-11-10 09:53:21 -08001293 if (chand->resolver == nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001294 pick_done_locked(elem,
Mark D. Rothb2929602017-09-11 09:31:11 -07001295 GRPC_ERROR_CREATE_FROM_STATIC_STRING("Disconnected"));
1296 return;
1297 }
1298 if (!chand->started_resolving) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001299 start_resolving_locked(chand);
Mark D. Rothb2929602017-09-11 09:31:11 -07001300 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001301 pick_after_resolver_result_start_locked(elem);
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001302 }
Mark D. Rothb2929602017-09-11 09:31:11 -07001303 // We need to wait for either a resolver result or for an async result
1304 // from the LB policy. Add the polling entity from call_data to the
1305 // channel_data's interested_parties, so that the I/O of the LB policy
1306 // and resolver can be done under it. The polling entity will be
1307 // removed in async_pick_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001308 grpc_polling_entity_add_to_pollset_set(calld->pollent,
Mark D. Rothb2929602017-09-11 09:31:11 -07001309 chand->interested_parties);
Craig Tillera11bfc82017-02-14 09:56:33 -08001310}
1311
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001312static void on_complete(void* arg, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001313 grpc_call_element* elem = static_cast<grpc_call_element*>(arg);
1314 call_data* calld = static_cast<call_data*>(elem->call_data);
Craig Tiller4782d922017-11-10 09:53:21 -08001315 if (calld->retry_throttle_data != nullptr) {
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001316 if (error == GRPC_ERROR_NONE) {
1317 grpc_server_retry_throttle_data_record_success(
Mark D. Roth9ccbc4d2017-03-15 08:30:04 -07001318 calld->retry_throttle_data);
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001319 } else {
1320 // TODO(roth): In a subsequent PR, check the return value here and
Mark D. Rothb3322562017-02-23 14:38:02 -08001321 // decide whether or not to retry. Note that we should only
1322 // record failures whose statuses match the configured retryable
1323 // or non-fatal status codes.
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001324 grpc_server_retry_throttle_data_record_failure(
Mark D. Roth9ccbc4d2017-03-15 08:30:04 -07001325 calld->retry_throttle_data);
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001326 }
1327 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001328 GRPC_CLOSURE_RUN(calld->original_on_complete, GRPC_ERROR_REF(error));
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001329}
1330
Craig Tillere1b51da2017-03-31 15:44:33 -07001331static void cc_start_transport_stream_op_batch(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001332 grpc_call_element* elem, grpc_transport_stream_op_batch* batch) {
yang-gce1cfea2018-01-31 15:59:50 -08001333 GPR_TIMER_SCOPE("cc_start_transport_stream_op_batch", 0);
Noah Eisenbe82e642018-02-09 09:16:55 -08001334 call_data* calld = static_cast<call_data*>(elem->call_data);
1335 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Craig Tiller3be7dd02017-04-03 14:30:03 -07001336 if (chand->deadline_checking_enabled) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001337 grpc_deadline_state_client_start_transport_stream_op_batch(elem, batch);
Craig Tiller3be7dd02017-04-03 14:30:03 -07001338 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001339 // If we've previously been cancelled, immediately fail any new batches.
1340 if (calld->error != GRPC_ERROR_NONE) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001341 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth76e264b2017-08-25 09:03:33 -07001342 gpr_log(GPR_DEBUG, "chand=%p calld=%p: failing batch with error: %s",
1343 chand, calld, grpc_error_string(calld->error));
1344 }
1345 grpc_transport_stream_op_batch_finish_with_failure(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001346 batch, GRPC_ERROR_REF(calld->error), calld->call_combiner);
yang-gce1cfea2018-01-31 15:59:50 -08001347 return;
Mark D. Roth76e264b2017-08-25 09:03:33 -07001348 }
1349 if (batch->cancel_stream) {
1350 // Stash a copy of cancel_error in our call data, so that we can use
1351 // it for subsequent operations. This ensures that if the call is
1352 // cancelled before any batches are passed down (e.g., if the deadline
1353 // is in the past when the call starts), we can return the right
1354 // error to the caller when the first batch does get passed down.
1355 GRPC_ERROR_UNREF(calld->error);
1356 calld->error = GRPC_ERROR_REF(batch->payload->cancel_stream.cancel_error);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001357 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth76e264b2017-08-25 09:03:33 -07001358 gpr_log(GPR_DEBUG, "chand=%p calld=%p: recording cancel_error=%s", chand,
1359 calld, grpc_error_string(calld->error));
1360 }
1361 // If we have a subchannel call, send the cancellation batch down.
1362 // Otherwise, fail all pending batches.
Craig Tiller4782d922017-11-10 09:53:21 -08001363 if (calld->subchannel_call != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001364 grpc_subchannel_call_process_op(calld->subchannel_call, batch);
Mark D. Roth76e264b2017-08-25 09:03:33 -07001365 } else {
1366 waiting_for_pick_batches_add(calld, batch);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001367 waiting_for_pick_batches_fail(elem, GRPC_ERROR_REF(calld->error));
Mark D. Roth76e264b2017-08-25 09:03:33 -07001368 }
yang-gce1cfea2018-01-31 15:59:50 -08001369 return;
Mark D. Roth76e264b2017-08-25 09:03:33 -07001370 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001371 // Intercept on_complete for recv_trailing_metadata so that we can
1372 // check retry throttle status.
Mark D. Roth60751fe2017-07-07 12:50:33 -07001373 if (batch->recv_trailing_metadata) {
Craig Tiller4782d922017-11-10 09:53:21 -08001374 GPR_ASSERT(batch->on_complete != nullptr);
Mark D. Roth60751fe2017-07-07 12:50:33 -07001375 calld->original_on_complete = batch->on_complete;
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001376 GRPC_CLOSURE_INIT(&calld->on_complete, on_complete, elem,
1377 grpc_schedule_on_exec_ctx);
Mark D. Roth60751fe2017-07-07 12:50:33 -07001378 batch->on_complete = &calld->on_complete;
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001379 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001380 // Check if we've already gotten a subchannel call.
1381 // Note that once we have completed the pick, we do not need to enter
1382 // the channel combiner, which is more efficient (especially for
1383 // streaming calls).
Craig Tiller4782d922017-11-10 09:53:21 -08001384 if (calld->subchannel_call != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001385 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001386 gpr_log(GPR_DEBUG,
1387 "chand=%p calld=%p: sending batch to subchannel_call=%p", chand,
Mark D. Roth76e264b2017-08-25 09:03:33 -07001388 calld, calld->subchannel_call);
Mark D. Roth60751fe2017-07-07 12:50:33 -07001389 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001390 grpc_subchannel_call_process_op(calld->subchannel_call, batch);
yang-gce1cfea2018-01-31 15:59:50 -08001391 return;
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001392 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001393 // We do not yet have a subchannel call.
1394 // Add the batch to the waiting-for-pick list.
1395 waiting_for_pick_batches_add(calld, batch);
1396 // For batches containing a send_initial_metadata op, enter the channel
1397 // combiner to start a pick.
1398 if (batch->send_initial_metadata) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001399 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001400 gpr_log(GPR_DEBUG, "chand=%p calld=%p: entering client_channel combiner",
1401 chand, calld);
Mark D. Roth76e264b2017-08-25 09:03:33 -07001402 }
1403 GRPC_CLOSURE_SCHED(
Mark D. Roth76e264b2017-08-25 09:03:33 -07001404 GRPC_CLOSURE_INIT(&batch->handler_private.closure, start_pick_locked,
1405 elem, grpc_combiner_scheduler(chand->combiner)),
1406 GRPC_ERROR_NONE);
1407 } else {
1408 // For all other batches, release the call combiner.
Craig Tiller6014e8a2017-10-16 13:50:29 -07001409 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth76e264b2017-08-25 09:03:33 -07001410 gpr_log(GPR_DEBUG,
1411 "chand=%p calld=%p: saved batch, yeilding call combiner", chand,
1412 calld);
1413 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001414 GRPC_CALL_COMBINER_STOP(calld->call_combiner,
Mark D. Roth76e264b2017-08-25 09:03:33 -07001415 "batch does not include send_initial_metadata");
Mark D. Roth60751fe2017-07-07 12:50:33 -07001416 }
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001417}
1418
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001419/* Constructor for call_data */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001420static grpc_error* cc_init_call_elem(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001421 const grpc_call_element_args* args) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001422 call_data* calld = static_cast<call_data*>(elem->call_data);
1423 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Mark D. Rothe40dd292016-10-05 14:58:37 -07001424 // Initialize data members.
Craig Tiller7c70b6c2017-01-23 07:48:42 -08001425 calld->path = grpc_slice_ref_internal(args->path);
Mark D. Rothff08f332016-10-14 13:01:01 -07001426 calld->call_start_time = args->start_time;
Craig Tiller89c14282017-07-19 15:32:27 -07001427 calld->deadline = args->deadline;
Craig Tillerd426cac2017-03-13 12:30:45 -07001428 calld->arena = args->arena;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001429 calld->owning_call = args->call_stack;
Mark D. Roth76e264b2017-08-25 09:03:33 -07001430 calld->call_combiner = args->call_combiner;
Craig Tiller3be7dd02017-04-03 14:30:03 -07001431 if (chand->deadline_checking_enabled) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001432 grpc_deadline_state_init(elem, args->call_stack, args->call_combiner,
1433 calld->deadline);
Craig Tiller3be7dd02017-04-03 14:30:03 -07001434 }
Mark D. Roth0badbe82016-06-23 10:15:12 -07001435 return GRPC_ERROR_NONE;
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001436}
1437
1438/* Destructor for call_data */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001439static void cc_destroy_call_elem(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001440 const grpc_call_final_info* final_info,
1441 grpc_closure* then_schedule_closure) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001442 call_data* calld = static_cast<call_data*>(elem->call_data);
1443 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Craig Tiller3be7dd02017-04-03 14:30:03 -07001444 if (chand->deadline_checking_enabled) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001445 grpc_deadline_state_destroy(elem);
Craig Tiller3be7dd02017-04-03 14:30:03 -07001446 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001447 grpc_slice_unref_internal(calld->path);
Craig Tiller4782d922017-11-10 09:53:21 -08001448 if (calld->method_params != nullptr) {
Mark D. Roth95b627b2017-02-24 11:02:58 -08001449 method_parameters_unref(calld->method_params);
1450 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001451 GRPC_ERROR_UNREF(calld->error);
Craig Tiller4782d922017-11-10 09:53:21 -08001452 if (calld->subchannel_call != nullptr) {
Mark D. Roth76e264b2017-08-25 09:03:33 -07001453 grpc_subchannel_call_set_cleanup_closure(calld->subchannel_call,
Craig Tillerf7c8c9f2017-05-17 15:22:05 -07001454 then_schedule_closure);
Craig Tiller4782d922017-11-10 09:53:21 -08001455 then_schedule_closure = nullptr;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001456 GRPC_SUBCHANNEL_CALL_UNREF(calld->subchannel_call,
Craig Tillerf7c8c9f2017-05-17 15:22:05 -07001457 "client_channel_destroy_call");
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001458 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001459 GPR_ASSERT(calld->waiting_for_pick_batches_count == 0);
David Garcia Quintasbe1b7f92018-01-12 14:01:38 -08001460 if (calld->pick.connected_subchannel != nullptr) {
David Garcia Quintasdfa28512018-01-11 18:31:13 -08001461 calld->pick.connected_subchannel.reset();
Craig Tiller693d3942016-10-27 16:51:25 -07001462 }
Mark D. Roth09e458c2017-05-02 08:13:26 -07001463 for (size_t i = 0; i < GRPC_CONTEXT_COUNT; ++i) {
Mark D. Rothc0febd32018-01-09 10:25:24 -08001464 if (calld->pick.subchannel_call_context[i].value != nullptr) {
1465 calld->pick.subchannel_call_context[i].destroy(
1466 calld->pick.subchannel_call_context[i].value);
Mark D. Roth09e458c2017-05-02 08:13:26 -07001467 }
1468 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001469 GRPC_CLOSURE_SCHED(then_schedule_closure, GRPC_ERROR_NONE);
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001470}
1471
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001472static void cc_set_pollset_or_pollset_set(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001473 grpc_polling_entity* pollent) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001474 call_data* calld = static_cast<call_data*>(elem->call_data);
David Garcia Quintas2a50dfe2016-05-31 15:09:12 -07001475 calld->pollent = pollent;
Craig Tiller577c9b22015-11-02 14:11:15 -08001476}
1477
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001478/*************************************************************************
1479 * EXPORTED SYMBOLS
1480 */
1481
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001482const grpc_channel_filter grpc_client_channel_filter = {
Craig Tillera0f3abd2017-03-31 15:42:16 -07001483 cc_start_transport_stream_op_batch,
Craig Tillerf40df232016-03-25 13:38:14 -07001484 cc_start_transport_op,
1485 sizeof(call_data),
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001486 cc_init_call_elem,
David Garcia Quintas4afce7e2016-04-18 16:25:17 -07001487 cc_set_pollset_or_pollset_set,
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001488 cc_destroy_call_elem,
Craig Tillerf40df232016-03-25 13:38:14 -07001489 sizeof(channel_data),
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001490 cc_init_channel_elem,
1491 cc_destroy_channel_elem,
Mark D. Rothb2d24882016-10-27 15:44:07 -07001492 cc_get_channel_info,
Craig Tillerf40df232016-03-25 13:38:14 -07001493 "client-channel",
Craig Tiller87d5b192015-04-16 14:37:57 -07001494};
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001495
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001496static void try_to_connect_locked(void* arg, grpc_error* error_ignored) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001497 channel_data* chand = static_cast<channel_data*>(arg);
Craig Tiller4782d922017-11-10 09:53:21 -08001498 if (chand->lb_policy != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001499 grpc_lb_policy_exit_idle_locked(chand->lb_policy);
Craig Tiller613dafa2017-02-09 12:00:43 -08001500 } else {
1501 chand->exit_idle_when_lb_policy_arrives = true;
Craig Tiller4782d922017-11-10 09:53:21 -08001502 if (!chand->started_resolving && chand->resolver != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001503 start_resolving_locked(chand);
Craig Tiller613dafa2017-02-09 12:00:43 -08001504 }
1505 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001506 GRPC_CHANNEL_STACK_UNREF(chand->owning_stack, "try_to_connect");
Craig Tiller613dafa2017-02-09 12:00:43 -08001507}
1508
Craig Tillera82950e2015-09-22 12:33:20 -07001509grpc_connectivity_state grpc_client_channel_check_connectivity_state(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001510 grpc_channel_element* elem, int try_to_connect) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001511 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Craig Tillera8610c02017-02-14 10:05:11 -08001512 grpc_connectivity_state out =
1513 grpc_connectivity_state_check(&chand->state_tracker);
Craig Tillera82950e2015-09-22 12:33:20 -07001514 if (out == GRPC_CHANNEL_IDLE && try_to_connect) {
Craig Tillerd2e5cfc2017-02-09 13:02:20 -08001515 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "try_to_connect");
ncteisen274bbbe2017-06-08 14:57:11 -07001516 GRPC_CLOSURE_SCHED(
Yash Tibrewal0ee75742017-10-13 16:07:13 -07001517 GRPC_CLOSURE_CREATE(try_to_connect_locked, chand,
1518 grpc_combiner_scheduler(chand->combiner)),
Craig Tiller613dafa2017-02-09 12:00:43 -08001519 GRPC_ERROR_NONE);
Craig Tillera82950e2015-09-22 12:33:20 -07001520 }
Craig Tiller48cb07c2015-07-15 16:16:15 -07001521 return out;
1522}
1523
Alexander Polcync3b1f182017-04-18 13:51:36 -07001524typedef struct external_connectivity_watcher {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001525 channel_data* chand;
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001526 grpc_polling_entity pollent;
Craig Tillerbaa14a92017-11-03 09:09:36 -07001527 grpc_closure* on_complete;
1528 grpc_closure* watcher_timer_init;
1529 grpc_connectivity_state* state;
Craig Tiller86c99582015-11-25 15:22:26 -08001530 grpc_closure my_closure;
Craig Tillerbaa14a92017-11-03 09:09:36 -07001531 struct external_connectivity_watcher* next;
Craig Tiller86c99582015-11-25 15:22:26 -08001532} external_connectivity_watcher;
1533
Craig Tillerbaa14a92017-11-03 09:09:36 -07001534static external_connectivity_watcher* lookup_external_connectivity_watcher(
1535 channel_data* chand, grpc_closure* on_complete) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001536 gpr_mu_lock(&chand->external_connectivity_watcher_list_mu);
Craig Tillerbaa14a92017-11-03 09:09:36 -07001537 external_connectivity_watcher* w =
Alexander Polcync3b1f182017-04-18 13:51:36 -07001538 chand->external_connectivity_watcher_list_head;
Craig Tiller4782d922017-11-10 09:53:21 -08001539 while (w != nullptr && w->on_complete != on_complete) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001540 w = w->next;
1541 }
1542 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
1543 return w;
1544}
1545
1546static void external_connectivity_watcher_list_append(
Craig Tillerbaa14a92017-11-03 09:09:36 -07001547 channel_data* chand, external_connectivity_watcher* w) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001548 GPR_ASSERT(!lookup_external_connectivity_watcher(chand, w->on_complete));
1549
1550 gpr_mu_lock(&w->chand->external_connectivity_watcher_list_mu);
1551 GPR_ASSERT(!w->next);
1552 w->next = chand->external_connectivity_watcher_list_head;
1553 chand->external_connectivity_watcher_list_head = w;
1554 gpr_mu_unlock(&w->chand->external_connectivity_watcher_list_mu);
1555}
1556
1557static void external_connectivity_watcher_list_remove(
Craig Tillerbaa14a92017-11-03 09:09:36 -07001558 channel_data* chand, external_connectivity_watcher* too_remove) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001559 GPR_ASSERT(
1560 lookup_external_connectivity_watcher(chand, too_remove->on_complete));
1561 gpr_mu_lock(&chand->external_connectivity_watcher_list_mu);
1562 if (too_remove == chand->external_connectivity_watcher_list_head) {
1563 chand->external_connectivity_watcher_list_head = too_remove->next;
1564 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
1565 return;
1566 }
Craig Tillerbaa14a92017-11-03 09:09:36 -07001567 external_connectivity_watcher* w =
Alexander Polcync3b1f182017-04-18 13:51:36 -07001568 chand->external_connectivity_watcher_list_head;
Craig Tiller4782d922017-11-10 09:53:21 -08001569 while (w != nullptr) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001570 if (w->next == too_remove) {
1571 w->next = w->next->next;
1572 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
1573 return;
1574 }
1575 w = w->next;
1576 }
1577 GPR_UNREACHABLE_CODE(return );
1578}
1579
1580int grpc_client_channel_num_external_connectivity_watchers(
Craig Tillerbaa14a92017-11-03 09:09:36 -07001581 grpc_channel_element* elem) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001582 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Alexander Polcync3b1f182017-04-18 13:51:36 -07001583 int count = 0;
1584
1585 gpr_mu_lock(&chand->external_connectivity_watcher_list_mu);
Craig Tillerbaa14a92017-11-03 09:09:36 -07001586 external_connectivity_watcher* w =
Alexander Polcync3b1f182017-04-18 13:51:36 -07001587 chand->external_connectivity_watcher_list_head;
Craig Tiller4782d922017-11-10 09:53:21 -08001588 while (w != nullptr) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001589 count++;
1590 w = w->next;
1591 }
1592 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
1593
1594 return count;
1595}
1596
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001597static void on_external_watch_complete_locked(void* arg, grpc_error* error) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001598 external_connectivity_watcher* w = static_cast<external_connectivity_watcher*>(arg);
Craig Tillerbaa14a92017-11-03 09:09:36 -07001599 grpc_closure* follow_up = w->on_complete;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001600 grpc_polling_entity_del_from_pollset_set(&w->pollent,
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001601 w->chand->interested_parties);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001602 GRPC_CHANNEL_STACK_UNREF(w->chand->owning_stack,
Craig Tiller1d881fb2015-12-01 07:39:04 -08001603 "external_connectivity_watcher");
Alexander Polcync3b1f182017-04-18 13:51:36 -07001604 external_connectivity_watcher_list_remove(w->chand, w);
Craig Tiller86c99582015-11-25 15:22:26 -08001605 gpr_free(w);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001606 GRPC_CLOSURE_RUN(follow_up, GRPC_ERROR_REF(error));
Craig Tiller613dafa2017-02-09 12:00:43 -08001607}
1608
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001609static void watch_connectivity_state_locked(void* arg,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001610 grpc_error* error_ignored) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001611 external_connectivity_watcher* w = static_cast<external_connectivity_watcher*>(arg);
Craig Tiller4782d922017-11-10 09:53:21 -08001612 external_connectivity_watcher* found = nullptr;
1613 if (w->state != nullptr) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001614 external_connectivity_watcher_list_append(w->chand, w);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001615 GRPC_CLOSURE_RUN(w->watcher_timer_init, GRPC_ERROR_NONE);
Alexander Polcyn2004e392017-10-16 15:14:46 -07001616 GRPC_CLOSURE_INIT(&w->my_closure, on_external_watch_complete_locked, w,
1617 grpc_combiner_scheduler(w->chand->combiner));
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001618 grpc_connectivity_state_notify_on_state_change(&w->chand->state_tracker,
1619 w->state, &w->my_closure);
Alexander Polcync3b1f182017-04-18 13:51:36 -07001620 } else {
Craig Tiller4782d922017-11-10 09:53:21 -08001621 GPR_ASSERT(w->watcher_timer_init == nullptr);
Alexander Polcync3b1f182017-04-18 13:51:36 -07001622 found = lookup_external_connectivity_watcher(w->chand, w->on_complete);
1623 if (found) {
1624 GPR_ASSERT(found->on_complete == w->on_complete);
1625 grpc_connectivity_state_notify_on_state_change(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001626 &found->chand->state_tracker, nullptr, &found->my_closure);
Alexander Polcync3b1f182017-04-18 13:51:36 -07001627 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001628 grpc_polling_entity_del_from_pollset_set(&w->pollent,
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001629 w->chand->interested_parties);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001630 GRPC_CHANNEL_STACK_UNREF(w->chand->owning_stack,
Alexander Polcync3b1f182017-04-18 13:51:36 -07001631 "external_connectivity_watcher");
1632 gpr_free(w);
1633 }
Craig Tiller86c99582015-11-25 15:22:26 -08001634}
1635
Craig Tillera82950e2015-09-22 12:33:20 -07001636void grpc_client_channel_watch_connectivity_state(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001637 grpc_channel_element* elem, grpc_polling_entity pollent,
1638 grpc_connectivity_state* state, grpc_closure* closure,
1639 grpc_closure* watcher_timer_init) {
Noah Eisenbe82e642018-02-09 09:16:55 -08001640 channel_data* chand = static_cast<channel_data*>(elem->channel_data);
Craig Tillerbaa14a92017-11-03 09:09:36 -07001641 external_connectivity_watcher* w =
Noah Eisenbe82e642018-02-09 09:16:55 -08001642 static_cast<external_connectivity_watcher*>(gpr_zalloc(sizeof(*w)));
Craig Tiller86c99582015-11-25 15:22:26 -08001643 w->chand = chand;
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001644 w->pollent = pollent;
Mark D. Roth92210832017-05-02 15:04:39 -07001645 w->on_complete = closure;
Craig Tiller613dafa2017-02-09 12:00:43 -08001646 w->state = state;
Alexander Polcync3b1f182017-04-18 13:51:36 -07001647 w->watcher_timer_init = watcher_timer_init;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001648 grpc_polling_entity_add_to_pollset_set(&w->pollent,
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001649 chand->interested_parties);
Craig Tiller1d881fb2015-12-01 07:39:04 -08001650 GRPC_CHANNEL_STACK_REF(w->chand->owning_stack,
1651 "external_connectivity_watcher");
ncteisen274bbbe2017-06-08 14:57:11 -07001652 GRPC_CLOSURE_SCHED(
ncteisen274bbbe2017-06-08 14:57:11 -07001653 GRPC_CLOSURE_INIT(&w->my_closure, watch_connectivity_state_locked, w,
Craig Tilleree4b1452017-05-12 10:56:03 -07001654 grpc_combiner_scheduler(chand->combiner)),
Craig Tiller613dafa2017-02-09 12:00:43 -08001655 GRPC_ERROR_NONE);
Craig Tiller48cb07c2015-07-15 16:16:15 -07001656}