blob: a8a7a37be01705906544f7c8c648c3d1c345a276 [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>
32#include <grpc/support/useful.h>
33
Yuchen Zeng0bad30a2017-10-05 21:47:39 -070034#include "src/core/ext/filters/client_channel/backup_poller.h"
Craig Tiller9eb0fde2017-03-31 16:59:30 -070035#include "src/core/ext/filters/client_channel/http_connect_handshaker.h"
36#include "src/core/ext/filters/client_channel/lb_policy_registry.h"
37#include "src/core/ext/filters/client_channel/proxy_mapper_registry.h"
38#include "src/core/ext/filters/client_channel/resolver_registry.h"
39#include "src/core/ext/filters/client_channel/retry_throttle.h"
40#include "src/core/ext/filters/client_channel/subchannel.h"
Craig Tiller3be7dd02017-04-03 14:30:03 -070041#include "src/core/ext/filters/deadline/deadline_filter.h"
Craig Tiller9533d042016-03-25 17:11:06 -070042#include "src/core/lib/channel/channel_args.h"
43#include "src/core/lib/channel/connected_channel.h"
Mark D. Rothdbdf4952018-01-18 11:21:12 -080044#include "src/core/lib/gpr/string.h"
Craig Tillerbefafe62017-02-09 11:30:54 -080045#include "src/core/lib/iomgr/combiner.h"
Craig Tiller9533d042016-03-25 17:11:06 -070046#include "src/core/lib/iomgr/iomgr.h"
Mark D. Roth4c0fe492016-08-31 13:51:55 -070047#include "src/core/lib/iomgr/polling_entity.h"
Craig Tiller9533d042016-03-25 17:11:06 -070048#include "src/core/lib/profiling/timers.h"
Craig Tiller7c70b6c2017-01-23 07:48:42 -080049#include "src/core/lib/slice/slice_internal.h"
Craig Tiller9533d042016-03-25 17:11:06 -070050#include "src/core/lib/surface/channel.h"
51#include "src/core/lib/transport/connectivity_state.h"
Mark D. Roth9fe284e2016-09-12 11:22:27 -070052#include "src/core/lib/transport/metadata.h"
53#include "src/core/lib/transport/metadata_batch.h"
Mark D. Rothea846a02016-11-03 11:32:54 -070054#include "src/core/lib/transport/service_config.h"
Mark D. Roth9fe284e2016-09-12 11:22:27 -070055#include "src/core/lib/transport/static_metadata.h"
Craig Tiller8910ac62015-10-08 16:49:15 -070056
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -080057/* Client channel implementation */
58
Craig Tiller694580f2017-10-18 14:48:14 -070059grpc_core::TraceFlag grpc_client_channel_trace(false, "client_channel");
Mark D. Roth60751fe2017-07-07 12:50:33 -070060
Mark D. Roth26b7be42016-10-24 10:08:07 -070061/*************************************************************************
62 * METHOD-CONFIG TABLE
63 */
64
Mark D. Roth9d480942016-10-19 14:18:05 -070065typedef enum {
Craig Tiller7acc37e2017-02-28 10:01:37 -080066 /* zero so it can be default initialized */
67 WAIT_FOR_READY_UNSET = 0,
Mark D. Roth9d480942016-10-19 14:18:05 -070068 WAIT_FOR_READY_FALSE,
69 WAIT_FOR_READY_TRUE
70} wait_for_ready_value;
71
Mark D. Roth95b627b2017-02-24 11:02:58 -080072typedef struct {
73 gpr_refcount refs;
Craig Tiller89c14282017-07-19 15:32:27 -070074 grpc_millis timeout;
Mark D. Roth9d480942016-10-19 14:18:05 -070075 wait_for_ready_value wait_for_ready;
76} method_parameters;
77
Craig Tillerbaa14a92017-11-03 09:09:36 -070078static method_parameters* method_parameters_ref(
79 method_parameters* method_params) {
Mark D. Roth95b627b2017-02-24 11:02:58 -080080 gpr_ref(&method_params->refs);
81 return method_params;
Mark D. Roth9d480942016-10-19 14:18:05 -070082}
83
Craig Tillerbaa14a92017-11-03 09:09:36 -070084static void method_parameters_unref(method_parameters* method_params) {
Mark D. Roth95b627b2017-02-24 11:02:58 -080085 if (gpr_unref(&method_params->refs)) {
86 gpr_free(method_params);
87 }
88}
89
Mark D. Roth76d0ec42017-10-26 11:08:14 -070090// Wrappers to pass to grpc_service_config_create_method_config_table().
Craig Tillera64b2b12017-11-03 15:23:13 -070091static void* method_parameters_ref_wrapper(void* value) {
92 return method_parameters_ref((method_parameters*)value);
Mark D. Roth76d0ec42017-10-26 11:08:14 -070093}
Yash Tibrewal8cf14702017-12-06 09:47:54 -080094static void method_parameters_unref_wrapper(void* value) {
Craig Tillerbaa14a92017-11-03 09:09:36 -070095 method_parameters_unref((method_parameters*)value);
Craig Tiller87a7e1f2016-11-09 09:42:19 -080096}
97
Craig Tillerbaa14a92017-11-03 09:09:36 -070098static bool parse_wait_for_ready(grpc_json* field,
99 wait_for_ready_value* wait_for_ready) {
Mark D. Roth95b627b2017-02-24 11:02:58 -0800100 if (field->type != GRPC_JSON_TRUE && field->type != GRPC_JSON_FALSE) {
101 return false;
102 }
103 *wait_for_ready = field->type == GRPC_JSON_TRUE ? WAIT_FOR_READY_TRUE
104 : WAIT_FOR_READY_FALSE;
105 return true;
106}
107
Craig Tillerbaa14a92017-11-03 09:09:36 -0700108static bool parse_timeout(grpc_json* field, grpc_millis* timeout) {
Mark D. Roth95b627b2017-02-24 11:02:58 -0800109 if (field->type != GRPC_JSON_STRING) return false;
110 size_t len = strlen(field->value);
111 if (field->value[len - 1] != 's') return false;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700112 char* buf = gpr_strdup(field->value);
Mark D. Roth95b627b2017-02-24 11:02:58 -0800113 buf[len - 1] = '\0'; // Remove trailing 's'.
Craig Tillerbaa14a92017-11-03 09:09:36 -0700114 char* decimal_point = strchr(buf, '.');
Craig Tiller89c14282017-07-19 15:32:27 -0700115 int nanos = 0;
Craig Tiller4782d922017-11-10 09:53:21 -0800116 if (decimal_point != nullptr) {
Mark D. Roth95b627b2017-02-24 11:02:58 -0800117 *decimal_point = '\0';
Craig Tiller89c14282017-07-19 15:32:27 -0700118 nanos = gpr_parse_nonnegative_int(decimal_point + 1);
119 if (nanos == -1) {
Mark D. Roth95b627b2017-02-24 11:02:58 -0800120 gpr_free(buf);
121 return false;
122 }
Mark D. Rotha2821462017-10-26 11:31:58 -0700123 int num_digits = (int)strlen(decimal_point + 1);
124 if (num_digits > 9) { // We don't accept greater precision than nanos.
125 gpr_free(buf);
126 return false;
Mark D. Roth95b627b2017-02-24 11:02:58 -0800127 }
Mark D. Rotha2821462017-10-26 11:31:58 -0700128 for (int i = 0; i < (9 - num_digits); ++i) {
129 nanos *= 10;
130 }
Mark D. Roth95b627b2017-02-24 11:02:58 -0800131 }
Mark D. Rotha2821462017-10-26 11:31:58 -0700132 int seconds = decimal_point == buf ? 0 : gpr_parse_nonnegative_int(buf);
Mark D. Roth95b627b2017-02-24 11:02:58 -0800133 gpr_free(buf);
Craig Tiller89c14282017-07-19 15:32:27 -0700134 if (seconds == -1) return false;
135 *timeout = seconds * GPR_MS_PER_SEC + nanos / GPR_NS_PER_MS;
Mark D. Roth95b627b2017-02-24 11:02:58 -0800136 return true;
137}
138
Craig Tillerbaa14a92017-11-03 09:09:36 -0700139static void* method_parameters_create_from_json(const grpc_json* json) {
Mark D. Rothc968e602016-11-02 14:07:36 -0700140 wait_for_ready_value wait_for_ready = WAIT_FOR_READY_UNSET;
Craig Tiller89c14282017-07-19 15:32:27 -0700141 grpc_millis timeout = 0;
Craig Tiller4782d922017-11-10 09:53:21 -0800142 for (grpc_json* field = json->child; field != nullptr; field = field->next) {
143 if (field->key == nullptr) continue;
Mark D. Roth84c8a022016-11-10 09:39:34 -0800144 if (strcmp(field->key, "waitForReady") == 0) {
Craig Tiller4782d922017-11-10 09:53:21 -0800145 if (wait_for_ready != WAIT_FOR_READY_UNSET) return nullptr; // Duplicate.
146 if (!parse_wait_for_ready(field, &wait_for_ready)) return nullptr;
Mark D. Rothc968e602016-11-02 14:07:36 -0700147 } else if (strcmp(field->key, "timeout") == 0) {
Craig Tiller4782d922017-11-10 09:53:21 -0800148 if (timeout > 0) return nullptr; // Duplicate.
149 if (!parse_timeout(field, &timeout)) return nullptr;
Mark D. Rothc968e602016-11-02 14:07:36 -0700150 }
151 }
Craig Tillerbaa14a92017-11-03 09:09:36 -0700152 method_parameters* value =
153 (method_parameters*)gpr_malloc(sizeof(method_parameters));
Mark D. Roth95b627b2017-02-24 11:02:58 -0800154 gpr_ref_init(&value->refs, 1);
Mark D. Rothc968e602016-11-02 14:07:36 -0700155 value->timeout = timeout;
156 value->wait_for_ready = wait_for_ready;
Mark D. Roth9d480942016-10-19 14:18:05 -0700157 return value;
158}
159
Alexander Polcync3b1f182017-04-18 13:51:36 -0700160struct external_connectivity_watcher;
161
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700162/*************************************************************************
163 * CHANNEL-WIDE FUNCTIONS
164 */
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -0800165
Craig Tiller800dacb2015-10-06 09:10:26 -0700166typedef struct client_channel_channel_data {
Craig Tillerf5f17122015-06-25 08:47:26 -0700167 /** resolver for this channel */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700168 grpc_resolver* resolver;
Craig Tiller20a3c352015-08-05 08:39:50 -0700169 /** have we started resolving this channel */
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700170 bool started_resolving;
Craig Tiller3be7dd02017-04-03 14:30:03 -0700171 /** is deadline checking enabled? */
172 bool deadline_checking_enabled;
Mark D. Roth0e48a9a2016-09-08 14:14:39 -0700173 /** client channel factory */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700174 grpc_client_channel_factory* client_channel_factory;
Craig Tillerf5f17122015-06-25 08:47:26 -0700175
Craig Tillerbefafe62017-02-09 11:30:54 -0800176 /** combiner protecting all variables below in this data structure */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700177 grpc_combiner* combiner;
Mark D. Roth046cf762016-09-26 11:13:51 -0700178 /** currently active load balancer */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700179 grpc_lb_policy* lb_policy;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800180 /** retry throttle data */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700181 grpc_server_retry_throttle_data* retry_throttle_data;
Mark D. Roth9d480942016-10-19 14:18:05 -0700182 /** maps method names to method_parameters structs */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700183 grpc_slice_hash_table* method_params_table;
Mark D. Roth046cf762016-09-26 11:13:51 -0700184 /** incoming resolver result - set by resolver.next() */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700185 grpc_channel_args* resolver_result;
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700186 /** a list of closures that are all waiting for resolver result to come in */
187 grpc_closure_list waiting_for_resolver_result_closures;
Craig Tiller3f475422015-06-25 10:43:05 -0700188 /** resolver callback */
Mark D. Rothff4df062016-08-22 15:02:49 -0700189 grpc_closure on_resolver_result_changed;
Craig Tiller3f475422015-06-25 10:43:05 -0700190 /** connectivity state being tracked */
Craig Tillerca3e9d32015-06-27 18:37:27 -0700191 grpc_connectivity_state_tracker state_tracker;
Craig Tiller48cb07c2015-07-15 16:16:15 -0700192 /** when an lb_policy arrives, should we try to exit idle */
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700193 bool exit_idle_when_lb_policy_arrives;
Craig Tiller906e3bc2015-11-24 07:31:31 -0800194 /** owning stack */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700195 grpc_channel_stack* owning_stack;
Craig Tiller69b093b2016-02-25 19:04:07 -0800196 /** interested parties (owned) */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700197 grpc_pollset_set* interested_parties;
Craig Tiller613dafa2017-02-09 12:00:43 -0800198
Alexander Polcync3b1f182017-04-18 13:51:36 -0700199 /* external_connectivity_watcher_list head is guarded by its own mutex, since
200 * counts need to be grabbed immediately without polling on a cq */
201 gpr_mu external_connectivity_watcher_list_mu;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700202 struct external_connectivity_watcher* external_connectivity_watcher_list_head;
Alexander Polcync3b1f182017-04-18 13:51:36 -0700203
Craig Tiller613dafa2017-02-09 12:00:43 -0800204 /* the following properties are guarded by a mutex since API's require them
Craig Tiller46dd7902017-02-23 09:42:16 -0800205 to be instantaneously available */
Craig Tiller613dafa2017-02-09 12:00:43 -0800206 gpr_mu info_mu;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700207 char* info_lb_policy_name;
Craig Tiller613dafa2017-02-09 12:00:43 -0800208 /** service config in JSON form */
Craig Tillerbaa14a92017-11-03 09:09:36 -0700209 char* info_service_config_json;
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -0800210} channel_data;
211
Juanli Shen592cf342017-12-04 20:52:01 -0800212typedef struct {
213 channel_data* chand;
214 /** used as an identifier, don't dereference it because the LB policy may be
215 * non-existing when the callback is run */
216 grpc_lb_policy* lb_policy;
217 grpc_closure closure;
218} reresolution_request_args;
219
Craig Tillerd6c98df2015-08-18 09:33:44 -0700220/** We create one watcher for each new lb_policy that is returned from a
Mark D. Roth4c0fe492016-08-31 13:51:55 -0700221 resolver, to watch for state changes from the lb_policy. When a state
222 change is seen, we update the channel, and create a new watcher. */
Craig Tillera82950e2015-09-22 12:33:20 -0700223typedef struct {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700224 channel_data* chand;
Craig Tiller33825112015-09-18 07:44:19 -0700225 grpc_closure on_changed;
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700226 grpc_connectivity_state state;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700227 grpc_lb_policy* lb_policy;
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700228} lb_policy_connectivity_watcher;
229
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800230static void watch_lb_policy_locked(channel_data* chand,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700231 grpc_lb_policy* lb_policy,
Craig Tiller2400bf52017-02-09 16:25:19 -0800232 grpc_connectivity_state current_state);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700233
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800234static void set_channel_connectivity_state_locked(channel_data* chand,
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800235 grpc_connectivity_state state,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700236 grpc_error* error,
237 const char* reason) {
David Garcia Quintas37251282017-04-14 13:46:03 -0700238 /* TODO: Improve failure handling:
239 * - Make it possible for policies to return GRPC_CHANNEL_TRANSIENT_FAILURE.
240 * - Hand over pending picks from old policies during the switch that happens
241 * when resolver provides an update. */
Craig Tiller4782d922017-11-10 09:53:21 -0800242 if (chand->lb_policy != nullptr) {
David Garcia Quintas956f7002017-04-13 15:40:06 -0700243 if (state == GRPC_CHANNEL_TRANSIENT_FAILURE) {
244 /* cancel picks with wait_for_ready=false */
245 grpc_lb_policy_cancel_picks_locked(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800246 chand->lb_policy,
David Garcia Quintas956f7002017-04-13 15:40:06 -0700247 /* mask= */ GRPC_INITIAL_METADATA_WAIT_FOR_READY,
248 /* check= */ 0, GRPC_ERROR_REF(error));
249 } else if (state == GRPC_CHANNEL_SHUTDOWN) {
250 /* cancel all picks */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800251 grpc_lb_policy_cancel_picks_locked(chand->lb_policy,
David Garcia Quintas956f7002017-04-13 15:40:06 -0700252 /* mask= */ 0, /* check= */ 0,
253 GRPC_ERROR_REF(error));
254 }
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800255 }
Craig Tiller6014e8a2017-10-16 13:50:29 -0700256 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700257 gpr_log(GPR_DEBUG, "chand=%p: setting connectivity state to %s", chand,
258 grpc_connectivity_state_name(state));
259 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800260 grpc_connectivity_state_set(&chand->state_tracker, state, error, reason);
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800261}
262
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800263static void on_lb_policy_state_changed_locked(void* arg, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700264 lb_policy_connectivity_watcher* w = (lb_policy_connectivity_watcher*)arg;
Craig Tillerc5de8352017-02-09 14:08:05 -0800265 /* check if the notification is for the latest policy */
266 if (w->lb_policy == w->chand->lb_policy) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700267 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700268 gpr_log(GPR_DEBUG, "chand=%p: lb_policy=%p state changed to %s", w->chand,
269 w->lb_policy, grpc_connectivity_state_name(w->state));
270 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800271 set_channel_connectivity_state_locked(w->chand, w->state,
Craig Tillerc5de8352017-02-09 14:08:05 -0800272 GRPC_ERROR_REF(error), "lb_changed");
273 if (w->state != GRPC_CHANNEL_SHUTDOWN) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800274 watch_lb_policy_locked(w->chand, w->lb_policy, w->state);
Craig Tillerc5de8352017-02-09 14:08:05 -0800275 }
Craig Tillera82950e2015-09-22 12:33:20 -0700276 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800277 GRPC_CHANNEL_STACK_UNREF(w->chand->owning_stack, "watch_lb_policy");
Craig Tillera82950e2015-09-22 12:33:20 -0700278 gpr_free(w);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700279}
280
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800281static void watch_lb_policy_locked(channel_data* chand,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700282 grpc_lb_policy* lb_policy,
Craig Tiller2400bf52017-02-09 16:25:19 -0800283 grpc_connectivity_state current_state) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700284 lb_policy_connectivity_watcher* w =
285 (lb_policy_connectivity_watcher*)gpr_malloc(sizeof(*w));
Craig Tiller906e3bc2015-11-24 07:31:31 -0800286 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "watch_lb_policy");
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700287 w->chand = chand;
ncteisen274bbbe2017-06-08 14:57:11 -0700288 GRPC_CLOSURE_INIT(&w->on_changed, on_lb_policy_state_changed_locked, w,
Craig Tilleree4b1452017-05-12 10:56:03 -0700289 grpc_combiner_scheduler(chand->combiner));
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700290 w->state = current_state;
291 w->lb_policy = lb_policy;
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800292 grpc_lb_policy_notify_on_state_change_locked(lb_policy, &w->state,
Craig Tiller2400bf52017-02-09 16:25:19 -0800293 &w->on_changed);
Craig Tiller1ada6ad2015-07-16 16:19:14 -0700294}
295
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800296static void start_resolving_locked(channel_data* chand) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700297 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700298 gpr_log(GPR_DEBUG, "chand=%p: starting name resolution", chand);
299 }
300 GPR_ASSERT(!chand->started_resolving);
301 chand->started_resolving = true;
302 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "resolver");
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800303 grpc_resolver_next_locked(chand->resolver, &chand->resolver_result,
Mark D. Roth60751fe2017-07-07 12:50:33 -0700304 &chand->on_resolver_result_changed);
305}
306
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800307typedef struct {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700308 char* server_name;
309 grpc_server_retry_throttle_data* retry_throttle_data;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800310} service_config_parsing_state;
311
Craig Tillerbaa14a92017-11-03 09:09:36 -0700312static void parse_retry_throttle_params(const grpc_json* field, void* arg) {
313 service_config_parsing_state* parsing_state =
314 (service_config_parsing_state*)arg;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800315 if (strcmp(field->key, "retryThrottling") == 0) {
Craig Tiller4782d922017-11-10 09:53:21 -0800316 if (parsing_state->retry_throttle_data != nullptr) return; // Duplicate.
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800317 if (field->type != GRPC_JSON_OBJECT) return;
318 int max_milli_tokens = 0;
319 int milli_token_ratio = 0;
Craig Tiller4782d922017-11-10 09:53:21 -0800320 for (grpc_json* sub_field = field->child; sub_field != nullptr;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800321 sub_field = sub_field->next) {
Craig Tiller4782d922017-11-10 09:53:21 -0800322 if (sub_field->key == nullptr) return;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800323 if (strcmp(sub_field->key, "maxTokens") == 0) {
324 if (max_milli_tokens != 0) return; // Duplicate.
325 if (sub_field->type != GRPC_JSON_NUMBER) return;
326 max_milli_tokens = gpr_parse_nonnegative_int(sub_field->value);
327 if (max_milli_tokens == -1) return;
328 max_milli_tokens *= 1000;
329 } else if (strcmp(sub_field->key, "tokenRatio") == 0) {
330 if (milli_token_ratio != 0) return; // Duplicate.
331 if (sub_field->type != GRPC_JSON_NUMBER) return;
332 // We support up to 3 decimal digits.
333 size_t whole_len = strlen(sub_field->value);
334 uint32_t multiplier = 1;
335 uint32_t decimal_value = 0;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700336 const char* decimal_point = strchr(sub_field->value, '.');
Craig Tiller4782d922017-11-10 09:53:21 -0800337 if (decimal_point != nullptr) {
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800338 whole_len = (size_t)(decimal_point - sub_field->value);
339 multiplier = 1000;
340 size_t decimal_len = strlen(decimal_point + 1);
341 if (decimal_len > 3) decimal_len = 3;
342 if (!gpr_parse_bytes_to_uint32(decimal_point + 1, decimal_len,
343 &decimal_value)) {
344 return;
345 }
346 uint32_t decimal_multiplier = 1;
347 for (size_t i = 0; i < (3 - decimal_len); ++i) {
348 decimal_multiplier *= 10;
349 }
350 decimal_value *= decimal_multiplier;
351 }
352 uint32_t whole_value;
353 if (!gpr_parse_bytes_to_uint32(sub_field->value, whole_len,
354 &whole_value)) {
355 return;
356 }
357 milli_token_ratio = (int)((whole_value * multiplier) + decimal_value);
Mark D. Rothb3322562017-02-23 14:38:02 -0800358 if (milli_token_ratio <= 0) return;
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800359 }
360 }
361 parsing_state->retry_throttle_data =
362 grpc_retry_throttle_map_get_data_for_server(
363 parsing_state->server_name, max_milli_tokens, milli_token_ratio);
364 }
365}
366
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800367static void request_reresolution_locked(void* arg, grpc_error* error) {
Juanli Shen592cf342017-12-04 20:52:01 -0800368 reresolution_request_args* args = (reresolution_request_args*)arg;
369 channel_data* chand = args->chand;
370 // If this invocation is for a stale LB policy, treat it as an LB shutdown
371 // signal.
372 if (args->lb_policy != chand->lb_policy || error != GRPC_ERROR_NONE ||
373 chand->resolver == nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800374 GRPC_CHANNEL_STACK_UNREF(chand->owning_stack, "re-resolution");
Juanli Shen592cf342017-12-04 20:52:01 -0800375 gpr_free(args);
376 return;
377 }
378 if (grpc_client_channel_trace.enabled()) {
379 gpr_log(GPR_DEBUG, "chand=%p: started name re-resolving", chand);
380 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800381 grpc_resolver_channel_saw_error_locked(chand->resolver);
Juanli Shen592cf342017-12-04 20:52:01 -0800382 // Give back the closure to the LB policy.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800383 grpc_lb_policy_set_reresolve_closure_locked(chand->lb_policy, &args->closure);
Juanli Shen592cf342017-12-04 20:52:01 -0800384}
385
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800386static void on_resolver_result_changed_locked(void* arg, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700387 channel_data* chand = (channel_data*)arg;
Craig Tiller6014e8a2017-10-16 13:50:29 -0700388 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700389 gpr_log(GPR_DEBUG, "chand=%p: got resolver result: error=%s", chand,
390 grpc_error_string(error));
391 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700392 // Extract the following fields from the resolver result, if non-NULL.
Mark D. Roth15494b52017-07-12 15:26:55 -0700393 bool lb_policy_updated = false;
Craig Tiller4782d922017-11-10 09:53:21 -0800394 char* lb_policy_name_dup = nullptr;
Mark D. Roth60751fe2017-07-07 12:50:33 -0700395 bool lb_policy_name_changed = false;
Craig Tiller4782d922017-11-10 09:53:21 -0800396 grpc_lb_policy* new_lb_policy = nullptr;
397 char* service_config_json = nullptr;
398 grpc_server_retry_throttle_data* retry_throttle_data = nullptr;
399 grpc_slice_hash_table* method_params_table = nullptr;
400 if (chand->resolver_result != nullptr) {
Juanli Shen592cf342017-12-04 20:52:01 -0800401 if (chand->resolver != nullptr) {
402 // Find LB policy name.
403 const char* lb_policy_name = nullptr;
404 const grpc_arg* channel_arg = grpc_channel_args_find(
405 chand->resolver_result, GRPC_ARG_LB_POLICY_NAME);
406 if (channel_arg != nullptr) {
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800407 GPR_ASSERT(channel_arg->type == GRPC_ARG_STRING);
Juanli Shen592cf342017-12-04 20:52:01 -0800408 lb_policy_name = channel_arg->value.string;
Mark D. Rothbdc58b22016-11-04 09:25:57 -0700409 }
Juanli Shen592cf342017-12-04 20:52:01 -0800410 // Special case: If at least one balancer address is present, we use
411 // the grpclb policy, regardless of what the resolver actually specified.
412 channel_arg =
413 grpc_channel_args_find(chand->resolver_result, GRPC_ARG_LB_ADDRESSES);
414 if (channel_arg != nullptr && channel_arg->type == GRPC_ARG_POINTER) {
415 grpc_lb_addresses* addresses =
416 (grpc_lb_addresses*)channel_arg->value.pointer.p;
417 bool found_balancer_address = false;
418 for (size_t i = 0; i < addresses->num_addresses; ++i) {
419 if (addresses->addresses[i].is_balancer) {
420 found_balancer_address = true;
421 break;
422 }
423 }
424 if (found_balancer_address) {
425 if (lb_policy_name != nullptr &&
426 strcmp(lb_policy_name, "grpclb") != 0) {
427 gpr_log(GPR_INFO,
428 "resolver requested LB policy %s but provided at least one "
429 "balancer address -- forcing use of grpclb LB policy",
430 lb_policy_name);
431 }
432 lb_policy_name = "grpclb";
433 }
434 }
435 // Use pick_first if nothing was specified and we didn't select grpclb
436 // above.
437 if (lb_policy_name == nullptr) lb_policy_name = "pick_first";
438 grpc_lb_policy_args lb_policy_args;
439 lb_policy_args.args = chand->resolver_result;
440 lb_policy_args.client_channel_factory = chand->client_channel_factory;
441 lb_policy_args.combiner = chand->combiner;
442 // Check to see if we're already using the right LB policy.
443 // Note: It's safe to use chand->info_lb_policy_name here without
444 // taking a lock on chand->info_mu, because this function is the
445 // only thing that modifies its value, and it can only be invoked
446 // once at any given time.
447 lb_policy_name_changed =
448 chand->info_lb_policy_name == nullptr ||
449 gpr_stricmp(chand->info_lb_policy_name, lb_policy_name) != 0;
450 if (chand->lb_policy != nullptr && !lb_policy_name_changed) {
451 // Continue using the same LB policy. Update with new addresses.
452 lb_policy_updated = true;
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800453 grpc_lb_policy_update_locked(chand->lb_policy, &lb_policy_args);
Juanli Shen592cf342017-12-04 20:52:01 -0800454 } else {
455 // Instantiate new LB policy.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800456 new_lb_policy = grpc_lb_policy_create(lb_policy_name, &lb_policy_args);
Juanli Shen592cf342017-12-04 20:52:01 -0800457 if (new_lb_policy == nullptr) {
458 gpr_log(GPR_ERROR, "could not create LB policy \"%s\"",
459 lb_policy_name);
460 } else {
461 reresolution_request_args* args =
462 (reresolution_request_args*)gpr_zalloc(sizeof(*args));
463 args->chand = chand;
464 args->lb_policy = new_lb_policy;
465 GRPC_CLOSURE_INIT(&args->closure, request_reresolution_locked, args,
466 grpc_combiner_scheduler(chand->combiner));
467 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "re-resolution");
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800468 grpc_lb_policy_set_reresolve_closure_locked(new_lb_policy,
Juanli Shen592cf342017-12-04 20:52:01 -0800469 &args->closure);
470 }
471 }
472 // Find service config.
473 channel_arg = grpc_channel_args_find(chand->resolver_result,
474 GRPC_ARG_SERVICE_CONFIG);
475 if (channel_arg != nullptr) {
476 GPR_ASSERT(channel_arg->type == GRPC_ARG_STRING);
477 service_config_json = gpr_strdup(channel_arg->value.string);
478 grpc_service_config* service_config =
479 grpc_service_config_create(service_config_json);
480 if (service_config != nullptr) {
481 channel_arg = grpc_channel_args_find(chand->resolver_result,
482 GRPC_ARG_SERVER_URI);
483 GPR_ASSERT(channel_arg != nullptr);
484 GPR_ASSERT(channel_arg->type == GRPC_ARG_STRING);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800485 grpc_uri* uri = grpc_uri_parse(channel_arg->value.string, true);
Juanli Shen592cf342017-12-04 20:52:01 -0800486 GPR_ASSERT(uri->path[0] != '\0');
487 service_config_parsing_state parsing_state;
488 memset(&parsing_state, 0, sizeof(parsing_state));
489 parsing_state.server_name =
490 uri->path[0] == '/' ? uri->path + 1 : uri->path;
491 grpc_service_config_parse_global_params(
492 service_config, parse_retry_throttle_params, &parsing_state);
493 grpc_uri_destroy(uri);
494 retry_throttle_data = parsing_state.retry_throttle_data;
495 method_params_table = grpc_service_config_create_method_config_table(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800496 service_config, method_parameters_create_from_json,
Juanli Shen592cf342017-12-04 20:52:01 -0800497 method_parameters_ref_wrapper, method_parameters_unref_wrapper);
498 grpc_service_config_destroy(service_config);
499 }
500 }
501 // Before we clean up, save a copy of lb_policy_name, since it might
502 // be pointing to data inside chand->resolver_result.
503 // The copy will be saved in chand->lb_policy_name below.
504 lb_policy_name_dup = gpr_strdup(lb_policy_name);
Mark D. Roth9fe284e2016-09-12 11:22:27 -0700505 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800506 grpc_channel_args_destroy(chand->resolver_result);
Craig Tiller4782d922017-11-10 09:53:21 -0800507 chand->resolver_result = nullptr;
Craig Tillera82950e2015-09-22 12:33:20 -0700508 }
Craig Tiller6014e8a2017-10-16 13:50:29 -0700509 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700510 gpr_log(GPR_DEBUG,
511 "chand=%p: resolver result: lb_policy_name=\"%s\"%s, "
512 "service_config=\"%s\"",
Yash Tibrewal9eb86722017-09-17 23:43:30 -0700513 chand, lb_policy_name_dup,
514 lb_policy_name_changed ? " (changed)" : "", service_config_json);
Mark D. Roth60751fe2017-07-07 12:50:33 -0700515 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700516 // Now swap out fields in chand. Note that the new values may still
517 // be NULL if (e.g.) the resolver failed to return results or the
518 // results did not contain the necessary data.
519 //
520 // First, swap out the data used by cc_get_channel_info().
Craig Tiller613dafa2017-02-09 12:00:43 -0800521 gpr_mu_lock(&chand->info_mu);
Craig Tiller4782d922017-11-10 09:53:21 -0800522 if (lb_policy_name_dup != nullptr) {
Craig Tiller613dafa2017-02-09 12:00:43 -0800523 gpr_free(chand->info_lb_policy_name);
Yash Tibrewal9eb86722017-09-17 23:43:30 -0700524 chand->info_lb_policy_name = lb_policy_name_dup;
Mark D. Rothb2d24882016-10-27 15:44:07 -0700525 }
Craig Tiller4782d922017-11-10 09:53:21 -0800526 if (service_config_json != nullptr) {
Craig Tiller613dafa2017-02-09 12:00:43 -0800527 gpr_free(chand->info_service_config_json);
528 chand->info_service_config_json = service_config_json;
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800529 }
Craig Tiller613dafa2017-02-09 12:00:43 -0800530 gpr_mu_unlock(&chand->info_mu);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700531 // Swap out the retry throttle data.
Craig Tiller4782d922017-11-10 09:53:21 -0800532 if (chand->retry_throttle_data != nullptr) {
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800533 grpc_server_retry_throttle_data_unref(chand->retry_throttle_data);
534 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700535 chand->retry_throttle_data = retry_throttle_data;
536 // Swap out the method params table.
Craig Tiller4782d922017-11-10 09:53:21 -0800537 if (chand->method_params_table != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800538 grpc_slice_hash_table_unref(chand->method_params_table);
Mark D. Roth046cf762016-09-26 11:13:51 -0700539 }
Mark D. Roth9d480942016-10-19 14:18:05 -0700540 chand->method_params_table = method_params_table;
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700541 // If we have a new LB policy or are shutting down (in which case
Juanli Shen592cf342017-12-04 20:52:01 -0800542 // new_lb_policy will be NULL), swap out the LB policy, unreffing the old one
543 // and removing its fds from chand->interested_parties. Note that we do NOT do
544 // this if either (a) we updated the existing LB policy above or (b) we failed
545 // to create the new LB policy (in which case we want to continue using the
546 // most recent one we had).
Craig Tiller4782d922017-11-10 09:53:21 -0800547 if (new_lb_policy != nullptr || error != GRPC_ERROR_NONE ||
548 chand->resolver == nullptr) {
549 if (chand->lb_policy != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700550 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700551 gpr_log(GPR_DEBUG, "chand=%p: unreffing lb_policy=%p", chand,
552 chand->lb_policy);
553 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800554 grpc_pollset_set_del_pollset_set(chand->lb_policy->interested_parties,
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700555 chand->interested_parties);
Mark D. Rothc0febd32018-01-09 10:25:24 -0800556 grpc_lb_policy_shutdown_locked(chand->lb_policy, new_lb_policy);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800557 GRPC_LB_POLICY_UNREF(chand->lb_policy, "channel");
Craig Tiller45724b32015-09-22 10:42:19 -0700558 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700559 chand->lb_policy = new_lb_policy;
560 }
561 // Now that we've swapped out the relevant fields of chand, check for
562 // error or shutdown.
Craig Tiller4782d922017-11-10 09:53:21 -0800563 if (error != GRPC_ERROR_NONE || chand->resolver == nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700564 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700565 gpr_log(GPR_DEBUG, "chand=%p: shutting down", chand);
566 }
Craig Tiller4782d922017-11-10 09:53:21 -0800567 if (chand->resolver != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700568 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700569 gpr_log(GPR_DEBUG, "chand=%p: shutting down resolver", chand);
570 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800571 grpc_resolver_shutdown_locked(chand->resolver);
572 GRPC_RESOLVER_UNREF(chand->resolver, "channel");
Craig Tiller4782d922017-11-10 09:53:21 -0800573 chand->resolver = nullptr;
Craig Tiller76a5c0e2016-03-09 09:05:30 -0800574 }
Craig Tiller8c0d96f2016-03-11 14:27:52 -0800575 set_channel_connectivity_state_locked(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800576 chand, GRPC_CHANNEL_SHUTDOWN,
ncteisen4b36a3d2017-03-13 19:08:06 -0700577 GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING(
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700578 "Got resolver result after disconnection", &error, 1),
Craig Tiller804ff712016-05-05 16:25:40 -0700579 "resolver_gone");
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800580 GRPC_CHANNEL_STACK_UNREF(chand->owning_stack, "resolver");
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700581 grpc_closure_list_fail_all(&chand->waiting_for_resolver_result_closures,
582 GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING(
583 "Channel disconnected", &error, 1));
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800584 GRPC_CLOSURE_LIST_SCHED(&chand->waiting_for_resolver_result_closures);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700585 } else { // Not shutting down.
586 grpc_connectivity_state state = GRPC_CHANNEL_TRANSIENT_FAILURE;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700587 grpc_error* state_error =
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700588 GRPC_ERROR_CREATE_FROM_STATIC_STRING("No load balancing policy");
Craig Tiller4782d922017-11-10 09:53:21 -0800589 if (new_lb_policy != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -0700590 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700591 gpr_log(GPR_DEBUG, "chand=%p: initializing new LB policy", chand);
592 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700593 GRPC_ERROR_UNREF(state_error);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800594 state =
595 grpc_lb_policy_check_connectivity_locked(new_lb_policy, &state_error);
596 grpc_pollset_set_add_pollset_set(new_lb_policy->interested_parties,
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700597 chand->interested_parties);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800598 GRPC_CLOSURE_LIST_SCHED(&chand->waiting_for_resolver_result_closures);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700599 if (chand->exit_idle_when_lb_policy_arrives) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800600 grpc_lb_policy_exit_idle_locked(new_lb_policy);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700601 chand->exit_idle_when_lb_policy_arrives = false;
602 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800603 watch_lb_policy_locked(chand, new_lb_policy, state);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700604 }
Mark D. Roth15494b52017-07-12 15:26:55 -0700605 if (!lb_policy_updated) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800606 set_channel_connectivity_state_locked(
607 chand, state, GRPC_ERROR_REF(state_error), "new_lb+resolver");
Mark D. Roth15494b52017-07-12 15:26:55 -0700608 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800609 grpc_resolver_next_locked(chand->resolver, &chand->resolver_result,
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700610 &chand->on_resolver_result_changed);
611 GRPC_ERROR_UNREF(state_error);
Craig Tillera82950e2015-09-22 12:33:20 -0700612 }
Craig Tiller3f475422015-06-25 10:43:05 -0700613}
614
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800615static void start_transport_op_locked(void* arg, grpc_error* error_ignored) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700616 grpc_transport_op* op = (grpc_transport_op*)arg;
617 grpc_channel_element* elem =
618 (grpc_channel_element*)op->handler_private.extra_arg;
619 channel_data* chand = (channel_data*)elem->channel_data;
Craig Tiller000cd8f2015-09-18 07:20:29 -0700620
Craig Tiller4782d922017-11-10 09:53:21 -0800621 if (op->on_connectivity_state_change != nullptr) {
Craig Tillera82950e2015-09-22 12:33:20 -0700622 grpc_connectivity_state_notify_on_state_change(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800623 &chand->state_tracker, op->connectivity_state,
Craig Tillera82950e2015-09-22 12:33:20 -0700624 op->on_connectivity_state_change);
Craig Tiller4782d922017-11-10 09:53:21 -0800625 op->on_connectivity_state_change = nullptr;
626 op->connectivity_state = nullptr;
Craig Tillera82950e2015-09-22 12:33:20 -0700627 }
628
Yuchen Zengc272dd72017-12-05 12:18:34 -0800629 if (op->send_ping.on_initiate != nullptr || op->send_ping.on_ack != nullptr) {
Craig Tiller4782d922017-11-10 09:53:21 -0800630 if (chand->lb_policy == nullptr) {
ncteisen274bbbe2017-06-08 14:57:11 -0700631 GRPC_CLOSURE_SCHED(
Yash Tibrewald6c292f2017-12-07 19:38:43 -0800632 op->send_ping.on_initiate,
Yuchen Zengc272dd72017-12-05 12:18:34 -0800633 GRPC_ERROR_CREATE_FROM_STATIC_STRING("Ping with no load balancing"));
634 GRPC_CLOSURE_SCHED(
Yash Tibrewald6c292f2017-12-07 19:38:43 -0800635 op->send_ping.on_ack,
ncteisen4b36a3d2017-03-13 19:08:06 -0700636 GRPC_ERROR_CREATE_FROM_STATIC_STRING("Ping with no load balancing"));
Craig Tiller26dab312015-12-07 14:43:47 -0800637 } else {
Yash Tibrewald6c292f2017-12-07 19:38:43 -0800638 grpc_lb_policy_ping_one_locked(
639 chand->lb_policy, op->send_ping.on_initiate, op->send_ping.on_ack);
Craig Tiller4782d922017-11-10 09:53:21 -0800640 op->bind_pollset = nullptr;
Craig Tiller26dab312015-12-07 14:43:47 -0800641 }
Yuchen Zengc272dd72017-12-05 12:18:34 -0800642 op->send_ping.on_initiate = nullptr;
643 op->send_ping.on_ack = nullptr;
Craig Tiller26dab312015-12-07 14:43:47 -0800644 }
645
Craig Tiller1c51edc2016-05-07 16:18:43 -0700646 if (op->disconnect_with_error != GRPC_ERROR_NONE) {
Craig Tiller4782d922017-11-10 09:53:21 -0800647 if (chand->resolver != nullptr) {
Craig Tiller1c51edc2016-05-07 16:18:43 -0700648 set_channel_connectivity_state_locked(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800649 chand, GRPC_CHANNEL_SHUTDOWN,
Craig Tiller1c51edc2016-05-07 16:18:43 -0700650 GRPC_ERROR_REF(op->disconnect_with_error), "disconnect");
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800651 grpc_resolver_shutdown_locked(chand->resolver);
652 GRPC_RESOLVER_UNREF(chand->resolver, "channel");
Craig Tiller4782d922017-11-10 09:53:21 -0800653 chand->resolver = nullptr;
Craig Tiller1c51edc2016-05-07 16:18:43 -0700654 if (!chand->started_resolving) {
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700655 grpc_closure_list_fail_all(&chand->waiting_for_resolver_result_closures,
Craig Tiller1c51edc2016-05-07 16:18:43 -0700656 GRPC_ERROR_REF(op->disconnect_with_error));
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800657 GRPC_CLOSURE_LIST_SCHED(&chand->waiting_for_resolver_result_closures);
Craig Tiller1c51edc2016-05-07 16:18:43 -0700658 }
Craig Tiller4782d922017-11-10 09:53:21 -0800659 if (chand->lb_policy != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800660 grpc_pollset_set_del_pollset_set(chand->lb_policy->interested_parties,
Craig Tiller1c51edc2016-05-07 16:18:43 -0700661 chand->interested_parties);
Mark D. Rothc0febd32018-01-09 10:25:24 -0800662 grpc_lb_policy_shutdown_locked(chand->lb_policy, nullptr);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800663 GRPC_LB_POLICY_UNREF(chand->lb_policy, "channel");
Craig Tiller4782d922017-11-10 09:53:21 -0800664 chand->lb_policy = nullptr;
Craig Tiller1c51edc2016-05-07 16:18:43 -0700665 }
Craig Tillerb12d22a2016-04-23 12:50:21 -0700666 }
Craig Tiller1c51edc2016-05-07 16:18:43 -0700667 GRPC_ERROR_UNREF(op->disconnect_with_error);
Craig Tillera82950e2015-09-22 12:33:20 -0700668 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800669 GRPC_CHANNEL_STACK_UNREF(chand->owning_stack, "start_transport_op");
Craig Tillerd2e5cfc2017-02-09 13:02:20 -0800670
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800671 GRPC_CLOSURE_SCHED(op->on_consumed, GRPC_ERROR_NONE);
Craig Tillerbefafe62017-02-09 11:30:54 -0800672}
673
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800674static void cc_start_transport_op(grpc_channel_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700675 grpc_transport_op* op) {
676 channel_data* chand = (channel_data*)elem->channel_data;
Craig Tillerbefafe62017-02-09 11:30:54 -0800677
Craig Tillerbefafe62017-02-09 11:30:54 -0800678 GPR_ASSERT(op->set_accept_stream == false);
Craig Tiller4782d922017-11-10 09:53:21 -0800679 if (op->bind_pollset != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800680 grpc_pollset_set_add_pollset(chand->interested_parties, op->bind_pollset);
Craig Tillerbefafe62017-02-09 11:30:54 -0800681 }
682
Craig Tillerc55c1022017-03-10 10:26:42 -0800683 op->handler_private.extra_arg = elem;
Craig Tillerd2e5cfc2017-02-09 13:02:20 -0800684 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "start_transport_op");
ncteisen274bbbe2017-06-08 14:57:11 -0700685 GRPC_CLOSURE_SCHED(
ncteisen274bbbe2017-06-08 14:57:11 -0700686 GRPC_CLOSURE_INIT(&op->handler_private.closure, start_transport_op_locked,
Craig Tilleree4b1452017-05-12 10:56:03 -0700687 op, grpc_combiner_scheduler(chand->combiner)),
Craig Tillerbefafe62017-02-09 11:30:54 -0800688 GRPC_ERROR_NONE);
Craig Tillerca3e9d32015-06-27 18:37:27 -0700689}
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -0800690
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800691static void cc_get_channel_info(grpc_channel_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700692 const grpc_channel_info* info) {
693 channel_data* chand = (channel_data*)elem->channel_data;
Craig Tiller613dafa2017-02-09 12:00:43 -0800694 gpr_mu_lock(&chand->info_mu);
Craig Tiller4782d922017-11-10 09:53:21 -0800695 if (info->lb_policy_name != nullptr) {
696 *info->lb_policy_name = chand->info_lb_policy_name == nullptr
697 ? nullptr
Craig Tiller613dafa2017-02-09 12:00:43 -0800698 : gpr_strdup(chand->info_lb_policy_name);
Mark D. Rothb2d24882016-10-27 15:44:07 -0700699 }
Craig Tiller4782d922017-11-10 09:53:21 -0800700 if (info->service_config_json != nullptr) {
Craig Tiller613dafa2017-02-09 12:00:43 -0800701 *info->service_config_json =
Craig Tiller4782d922017-11-10 09:53:21 -0800702 chand->info_service_config_json == nullptr
703 ? nullptr
Craig Tiller613dafa2017-02-09 12:00:43 -0800704 : gpr_strdup(chand->info_service_config_json);
Mark D. Rothc625c7a2016-11-09 14:12:37 -0800705 }
Craig Tiller613dafa2017-02-09 12:00:43 -0800706 gpr_mu_unlock(&chand->info_mu);
Mark D. Rothb2d24882016-10-27 15:44:07 -0700707}
708
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700709/* Constructor for channel_data */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800710static grpc_error* cc_init_channel_elem(grpc_channel_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700711 grpc_channel_element_args* args) {
712 channel_data* chand = (channel_data*)elem->channel_data;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700713 GPR_ASSERT(args->is_last);
714 GPR_ASSERT(elem->filter == &grpc_client_channel_filter);
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800715 // Initialize data members.
Craig Tilleree4b1452017-05-12 10:56:03 -0700716 chand->combiner = grpc_combiner_create();
Craig Tillerd85477512017-02-09 12:02:39 -0800717 gpr_mu_init(&chand->info_mu);
Alexander Polcync3b1f182017-04-18 13:51:36 -0700718 gpr_mu_init(&chand->external_connectivity_watcher_list_mu);
719
720 gpr_mu_lock(&chand->external_connectivity_watcher_list_mu);
Craig Tiller4782d922017-11-10 09:53:21 -0800721 chand->external_connectivity_watcher_list_head = nullptr;
Alexander Polcync3b1f182017-04-18 13:51:36 -0700722 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
723
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800724 chand->owning_stack = args->channel_stack;
ncteisen274bbbe2017-06-08 14:57:11 -0700725 GRPC_CLOSURE_INIT(&chand->on_resolver_result_changed,
Craig Tillerbefafe62017-02-09 11:30:54 -0800726 on_resolver_result_changed_locked, chand,
Craig Tilleree4b1452017-05-12 10:56:03 -0700727 grpc_combiner_scheduler(chand->combiner));
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800728 chand->interested_parties = grpc_pollset_set_create();
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700729 grpc_connectivity_state_init(&chand->state_tracker, GRPC_CHANNEL_IDLE,
730 "client_channel");
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800731 grpc_client_channel_start_backup_polling(chand->interested_parties);
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800732 // Record client channel factory.
Craig Tillerbaa14a92017-11-03 09:09:36 -0700733 const grpc_arg* arg = grpc_channel_args_find(args->channel_args,
Mark D. Roth21d4b2d2016-11-18 09:53:41 -0800734 GRPC_ARG_CLIENT_CHANNEL_FACTORY);
Craig Tiller4782d922017-11-10 09:53:21 -0800735 if (arg == nullptr) {
David Garcia Quintas228a5142017-03-30 19:43:00 -0700736 return GRPC_ERROR_CREATE_FROM_STATIC_STRING(
737 "Missing client channel factory in args for client channel filter");
738 }
739 if (arg->type != GRPC_ARG_POINTER) {
740 return GRPC_ERROR_CREATE_FROM_STATIC_STRING(
741 "client channel factory arg must be a pointer");
742 }
Yash Tibrewalbc130da2017-09-12 22:44:08 -0700743 grpc_client_channel_factory_ref(
Craig Tillerbaa14a92017-11-03 09:09:36 -0700744 (grpc_client_channel_factory*)arg->value.pointer.p);
Yash Tibrewalca3c1c02017-09-07 22:47:16 -0700745 chand->client_channel_factory =
Craig Tillerbaa14a92017-11-03 09:09:36 -0700746 (grpc_client_channel_factory*)arg->value.pointer.p;
Mark D. Rothdc9bee72017-02-07 12:29:14 -0800747 // Get server name to resolve, using proxy mapper if needed.
Mark D. Roth86e90592016-11-18 09:56:40 -0800748 arg = grpc_channel_args_find(args->channel_args, GRPC_ARG_SERVER_URI);
Craig Tiller4782d922017-11-10 09:53:21 -0800749 if (arg == nullptr) {
David Garcia Quintas228a5142017-03-30 19:43:00 -0700750 return GRPC_ERROR_CREATE_FROM_STATIC_STRING(
751 "Missing server uri in args for client channel filter");
752 }
753 if (arg->type != GRPC_ARG_STRING) {
754 return GRPC_ERROR_CREATE_FROM_STATIC_STRING(
755 "server uri arg must be a string");
756 }
Craig Tiller4782d922017-11-10 09:53:21 -0800757 char* proxy_name = nullptr;
758 grpc_channel_args* new_args = nullptr;
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800759 grpc_proxy_mappers_map_name(arg->value.string, args->channel_args,
Mark D. Rothdc9bee72017-02-07 12:29:14 -0800760 &proxy_name, &new_args);
761 // Instantiate resolver.
Mark D. Roth45ccec52017-01-18 14:04:01 -0800762 chand->resolver = grpc_resolver_create(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800763 proxy_name != nullptr ? proxy_name : arg->value.string,
Craig Tiller4782d922017-11-10 09:53:21 -0800764 new_args != nullptr ? new_args : args->channel_args,
Craig Tiller972470b2017-02-09 15:05:36 -0800765 chand->interested_parties, chand->combiner);
Craig Tiller4782d922017-11-10 09:53:21 -0800766 if (proxy_name != nullptr) gpr_free(proxy_name);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800767 if (new_args != nullptr) grpc_channel_args_destroy(new_args);
Craig Tiller4782d922017-11-10 09:53:21 -0800768 if (chand->resolver == nullptr) {
ncteisen4b36a3d2017-03-13 19:08:06 -0700769 return GRPC_ERROR_CREATE_FROM_STATIC_STRING("resolver creation failed");
Mark D. Roth5e2566e2016-11-18 10:53:13 -0800770 }
Craig Tiller3be7dd02017-04-03 14:30:03 -0700771 chand->deadline_checking_enabled =
772 grpc_deadline_checking_enabled(args->channel_args);
Mark D. Roth5e2566e2016-11-18 10:53:13 -0800773 return GRPC_ERROR_NONE;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700774}
775
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800776static void shutdown_resolver_locked(void* arg, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700777 grpc_resolver* resolver = (grpc_resolver*)arg;
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800778 grpc_resolver_shutdown_locked(resolver);
779 GRPC_RESOLVER_UNREF(resolver, "channel");
Craig Tiller972470b2017-02-09 15:05:36 -0800780}
781
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700782/* Destructor for channel_data */
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800783static void cc_destroy_channel_elem(grpc_channel_element* elem) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700784 channel_data* chand = (channel_data*)elem->channel_data;
Craig Tiller4782d922017-11-10 09:53:21 -0800785 if (chand->resolver != nullptr) {
ncteisen274bbbe2017-06-08 14:57:11 -0700786 GRPC_CLOSURE_SCHED(
Yash Tibrewal0ee75742017-10-13 16:07:13 -0700787 GRPC_CLOSURE_CREATE(shutdown_resolver_locked, chand->resolver,
788 grpc_combiner_scheduler(chand->combiner)),
Craig Tiller972470b2017-02-09 15:05:36 -0800789 GRPC_ERROR_NONE);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700790 }
Craig Tiller4782d922017-11-10 09:53:21 -0800791 if (chand->client_channel_factory != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800792 grpc_client_channel_factory_unref(chand->client_channel_factory);
Mark D. Roth0e48a9a2016-09-08 14:14:39 -0700793 }
Craig Tiller4782d922017-11-10 09:53:21 -0800794 if (chand->lb_policy != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800795 grpc_pollset_set_del_pollset_set(chand->lb_policy->interested_parties,
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700796 chand->interested_parties);
Mark D. Rothc0febd32018-01-09 10:25:24 -0800797 grpc_lb_policy_shutdown_locked(chand->lb_policy, nullptr);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800798 GRPC_LB_POLICY_UNREF(chand->lb_policy, "channel");
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700799 }
Craig Tiller613dafa2017-02-09 12:00:43 -0800800 gpr_free(chand->info_lb_policy_name);
801 gpr_free(chand->info_service_config_json);
Craig Tiller4782d922017-11-10 09:53:21 -0800802 if (chand->retry_throttle_data != nullptr) {
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800803 grpc_server_retry_throttle_data_unref(chand->retry_throttle_data);
804 }
Craig Tiller4782d922017-11-10 09:53:21 -0800805 if (chand->method_params_table != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800806 grpc_slice_hash_table_unref(chand->method_params_table);
Mark D. Roth9fe284e2016-09-12 11:22:27 -0700807 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800808 grpc_client_channel_stop_backup_polling(chand->interested_parties);
809 grpc_connectivity_state_destroy(&chand->state_tracker);
810 grpc_pollset_set_destroy(chand->interested_parties);
811 GRPC_COMBINER_UNREF(chand->combiner, "client_channel");
Craig Tillerd85477512017-02-09 12:02:39 -0800812 gpr_mu_destroy(&chand->info_mu);
Alexander Polcync3b1f182017-04-18 13:51:36 -0700813 gpr_mu_destroy(&chand->external_connectivity_watcher_list_mu);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700814}
815
816/*************************************************************************
817 * PER-CALL FUNCTIONS
818 */
819
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700820// Max number of batches that can be pending on a call at any given
821// time. This includes:
822// recv_initial_metadata
823// send_initial_metadata
824// recv_message
825// send_message
826// recv_trailing_metadata
827// send_trailing_metadata
Mark D. Roth76e264b2017-08-25 09:03:33 -0700828// We also add room for a single cancel_stream batch.
829#define MAX_WAITING_BATCHES 7
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700830
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700831/** Call data. Holds a pointer to grpc_subchannel_call and the
832 associated machinery to create such a pointer.
833 Handles queueing of stream ops until a call object is ready, waiting
834 for initial metadata before trying to create a call object,
835 and handling cancellation gracefully. */
836typedef struct client_channel_call_data {
Mark D. Roth72f6da82016-09-02 13:42:38 -0700837 // State for handling deadlines.
838 // The code in deadline_filter.c requires this to be the first field.
Mark D. Roth72f6da82016-09-02 13:42:38 -0700839 // TODO(roth): This is slightly sub-optimal in that grpc_deadline_state
Mark D. Roth66f3d2b2017-09-01 09:02:17 -0700840 // and this struct both independently store pointers to the call stack
841 // and call combiner. If/when we have time, find a way to avoid this
842 // without breaking the grpc_deadline_state abstraction.
Mark D. Roth72f6da82016-09-02 13:42:38 -0700843 grpc_deadline_state deadline_state;
Mark D. Rothf28763c2016-09-14 15:18:40 -0700844
Craig Tiller7c70b6c2017-01-23 07:48:42 -0800845 grpc_slice path; // Request path.
Mark D. Rothe40dd292016-10-05 14:58:37 -0700846 gpr_timespec call_start_time;
Craig Tiller89c14282017-07-19 15:32:27 -0700847 grpc_millis deadline;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700848 gpr_arena* arena;
849 grpc_call_stack* owning_call;
850 grpc_call_combiner* call_combiner;
Mark D. Roth76e264b2017-08-25 09:03:33 -0700851
Craig Tillerbaa14a92017-11-03 09:09:36 -0700852 grpc_server_retry_throttle_data* retry_throttle_data;
853 method_parameters* method_params;
Mark D. Rothaa850a72016-09-26 13:38:02 -0700854
Craig Tillerbaa14a92017-11-03 09:09:36 -0700855 grpc_subchannel_call* subchannel_call;
856 grpc_error* error;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700857
Mark D. Rothc0febd32018-01-09 10:25:24 -0800858 grpc_lb_policy_pick_state pick;
Mark D. Roth60751fe2017-07-07 12:50:33 -0700859 grpc_closure lb_pick_closure;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -0700860 grpc_closure lb_pick_cancel_closure;
Mark D. Roth60751fe2017-07-07 12:50:33 -0700861
Craig Tillerbaa14a92017-11-03 09:09:36 -0700862 grpc_polling_entity* pollent;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700863
Craig Tillerbaa14a92017-11-03 09:09:36 -0700864 grpc_transport_stream_op_batch* waiting_for_pick_batches[MAX_WAITING_BATCHES];
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700865 size_t waiting_for_pick_batches_count;
Mark D. Roth76e264b2017-08-25 09:03:33 -0700866 grpc_closure handle_pending_batch_in_call_combiner[MAX_WAITING_BATCHES];
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700867
Craig Tillerbaa14a92017-11-03 09:09:36 -0700868 grpc_transport_stream_op_batch* initial_metadata_batch;
David Garcia Quintasd1a47f12016-09-02 12:46:44 +0200869
Mark D. Rothd6d192d2017-02-23 08:58:42 -0800870 grpc_closure on_complete;
Craig Tillerbaa14a92017-11-03 09:09:36 -0700871 grpc_closure* original_on_complete;
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700872} call_data;
873
Craig Tillerbaa14a92017-11-03 09:09:36 -0700874grpc_subchannel_call* grpc_client_channel_get_subchannel_call(
875 grpc_call_element* elem) {
876 call_data* calld = (call_data*)elem->call_data;
Mark D. Roth76e264b2017-08-25 09:03:33 -0700877 return calld->subchannel_call;
Craig Tiller8b1d59c2016-12-27 15:15:30 -0800878}
879
Mark D. Roth76e264b2017-08-25 09:03:33 -0700880// This is called via the call combiner, so access to calld is synchronized.
881static void waiting_for_pick_batches_add(
Craig Tillerbaa14a92017-11-03 09:09:36 -0700882 call_data* calld, grpc_transport_stream_op_batch* batch) {
Mark D. Roth76e264b2017-08-25 09:03:33 -0700883 if (batch->send_initial_metadata) {
Craig Tiller4782d922017-11-10 09:53:21 -0800884 GPR_ASSERT(calld->initial_metadata_batch == nullptr);
Mark D. Roth76e264b2017-08-25 09:03:33 -0700885 calld->initial_metadata_batch = batch;
886 } else {
887 GPR_ASSERT(calld->waiting_for_pick_batches_count < MAX_WAITING_BATCHES);
888 calld->waiting_for_pick_batches[calld->waiting_for_pick_batches_count++] =
889 batch;
890 }
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700891}
892
Mark D. Roth76e264b2017-08-25 09:03:33 -0700893// This is called via the call combiner, so access to calld is synchronized.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800894static void fail_pending_batch_in_call_combiner(void* arg, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700895 call_data* calld = (call_data*)arg;
Mark D. Roth76e264b2017-08-25 09:03:33 -0700896 if (calld->waiting_for_pick_batches_count > 0) {
897 --calld->waiting_for_pick_batches_count;
898 grpc_transport_stream_op_batch_finish_with_failure(
Mark D. Roth76e264b2017-08-25 09:03:33 -0700899 calld->waiting_for_pick_batches[calld->waiting_for_pick_batches_count],
900 GRPC_ERROR_REF(error), calld->call_combiner);
901 }
902}
903
904// This is called via the call combiner, so access to calld is synchronized.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800905static void waiting_for_pick_batches_fail(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -0700906 grpc_error* error) {
907 call_data* calld = (call_data*)elem->call_data;
Craig Tiller6014e8a2017-10-16 13:50:29 -0700908 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700909 gpr_log(GPR_DEBUG,
Mark D. Rothe9b10832017-10-26 13:18:25 -0700910 "chand=%p calld=%p: failing %" PRIuPTR " pending batches: %s",
Mark D. Roth60751fe2017-07-07 12:50:33 -0700911 elem->channel_data, calld, calld->waiting_for_pick_batches_count,
912 grpc_error_string(error));
913 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700914 for (size_t i = 0; i < calld->waiting_for_pick_batches_count; ++i) {
Mark D. Roth76e264b2017-08-25 09:03:33 -0700915 GRPC_CLOSURE_INIT(&calld->handle_pending_batch_in_call_combiner[i],
916 fail_pending_batch_in_call_combiner, calld,
917 grpc_schedule_on_exec_ctx);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800918 GRPC_CALL_COMBINER_START(
919 calld->call_combiner, &calld->handle_pending_batch_in_call_combiner[i],
920 GRPC_ERROR_REF(error), "waiting_for_pick_batches_fail");
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700921 }
Craig Tiller4782d922017-11-10 09:53:21 -0800922 if (calld->initial_metadata_batch != nullptr) {
Mark D. Roth76e264b2017-08-25 09:03:33 -0700923 grpc_transport_stream_op_batch_finish_with_failure(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800924 calld->initial_metadata_batch, GRPC_ERROR_REF(error),
Mark D. Roth76e264b2017-08-25 09:03:33 -0700925 calld->call_combiner);
926 } else {
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800927 GRPC_CALL_COMBINER_STOP(calld->call_combiner,
Mark D. Roth76e264b2017-08-25 09:03:33 -0700928 "waiting_for_pick_batches_fail");
929 }
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700930 GRPC_ERROR_UNREF(error);
931}
932
Mark D. Roth76e264b2017-08-25 09:03:33 -0700933// This is called via the call combiner, so access to calld is synchronized.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800934static void run_pending_batch_in_call_combiner(void* arg, grpc_error* ignored) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700935 call_data* calld = (call_data*)arg;
Mark D. Roth76e264b2017-08-25 09:03:33 -0700936 if (calld->waiting_for_pick_batches_count > 0) {
937 --calld->waiting_for_pick_batches_count;
938 grpc_subchannel_call_process_op(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800939 calld->subchannel_call,
Mark D. Roth76e264b2017-08-25 09:03:33 -0700940 calld->waiting_for_pick_batches[calld->waiting_for_pick_batches_count]);
Craig Tiller57726ca2016-09-12 11:59:45 -0700941 }
Mark D. Roth76e264b2017-08-25 09:03:33 -0700942}
943
944// This is called via the call combiner, so access to calld is synchronized.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800945static void waiting_for_pick_batches_resume(grpc_call_element* elem) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700946 channel_data* chand = (channel_data*)elem->channel_data;
947 call_data* calld = (call_data*)elem->call_data;
Craig Tiller6014e8a2017-10-16 13:50:29 -0700948 if (grpc_client_channel_trace.enabled()) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700949 gpr_log(GPR_DEBUG,
950 "chand=%p calld=%p: sending %" PRIuPTR
951 " pending batches to subchannel_call=%p",
Mark D. Roth76e264b2017-08-25 09:03:33 -0700952 chand, calld, calld->waiting_for_pick_batches_count,
953 calld->subchannel_call);
Mark D. Roth60751fe2017-07-07 12:50:33 -0700954 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700955 for (size_t i = 0; i < calld->waiting_for_pick_batches_count; ++i) {
Mark D. Roth76e264b2017-08-25 09:03:33 -0700956 GRPC_CLOSURE_INIT(&calld->handle_pending_batch_in_call_combiner[i],
957 run_pending_batch_in_call_combiner, calld,
958 grpc_schedule_on_exec_ctx);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800959 GRPC_CALL_COMBINER_START(
960 calld->call_combiner, &calld->handle_pending_batch_in_call_combiner[i],
961 GRPC_ERROR_NONE, "waiting_for_pick_batches_resume");
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700962 }
Craig Tiller4782d922017-11-10 09:53:21 -0800963 GPR_ASSERT(calld->initial_metadata_batch != nullptr);
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800964 grpc_subchannel_call_process_op(calld->subchannel_call,
Mark D. Roth76e264b2017-08-25 09:03:33 -0700965 calld->initial_metadata_batch);
Mark D. Roth2a5959f2016-09-01 08:20:27 -0700966}
967
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700968// Applies service config to the call. Must be invoked once we know
969// that the resolver has returned results to the channel.
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800970static void apply_service_config_to_call_locked(grpc_call_element* elem) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700971 channel_data* chand = (channel_data*)elem->channel_data;
972 call_data* calld = (call_data*)elem->call_data;
Craig Tiller6014e8a2017-10-16 13:50:29 -0700973 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -0700974 gpr_log(GPR_DEBUG, "chand=%p calld=%p: applying service config to call",
975 chand, calld);
976 }
Craig Tiller4782d922017-11-10 09:53:21 -0800977 if (chand->retry_throttle_data != nullptr) {
Mark D. Roth9ccbc4d2017-03-15 08:30:04 -0700978 calld->retry_throttle_data =
979 grpc_server_retry_throttle_data_ref(chand->retry_throttle_data);
980 }
Craig Tiller4782d922017-11-10 09:53:21 -0800981 if (chand->method_params_table != nullptr) {
Craig Tillerbaa14a92017-11-03 09:09:36 -0700982 calld->method_params = (method_parameters*)grpc_method_config_table_get(
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800983 chand->method_params_table, calld->path);
Craig Tiller4782d922017-11-10 09:53:21 -0800984 if (calld->method_params != nullptr) {
Craig Tiller11c17d42017-03-13 13:36:34 -0700985 method_parameters_ref(calld->method_params);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700986 // If the deadline from the service config is shorter than the one
987 // from the client API, reset the deadline timer.
988 if (chand->deadline_checking_enabled &&
Craig Tiller89c14282017-07-19 15:32:27 -0700989 calld->method_params->timeout != 0) {
990 const grpc_millis per_method_deadline =
Craig Tiller9a8c3f32017-07-21 13:14:14 -0700991 grpc_timespec_to_millis_round_up(calld->call_start_time) +
Craig Tiller89c14282017-07-19 15:32:27 -0700992 calld->method_params->timeout;
993 if (per_method_deadline < calld->deadline) {
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700994 calld->deadline = per_method_deadline;
Yash Tibrewal8cf14702017-12-06 09:47:54 -0800995 grpc_deadline_state_reset(elem, calld->deadline);
Mark D. Roth0ca0be82017-06-20 07:49:33 -0700996 }
Craig Tiller11c17d42017-03-13 13:36:34 -0700997 }
998 }
999 }
Craig Tiller11c17d42017-03-13 13:36:34 -07001000}
Craig Tillerea4a4f12017-03-13 13:36:52 -07001001
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001002static void create_subchannel_call_locked(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001003 grpc_error* error) {
1004 channel_data* chand = (channel_data*)elem->channel_data;
1005 call_data* calld = (call_data*)elem->call_data;
David Garcia Quintasbaf1ac72018-01-09 14:24:32 -08001006 const grpc_core::ConnectedSubchannel::CallArgs call_args = {
Mark D. Rothc0febd32018-01-09 10:25:24 -08001007 calld->pollent, // pollent
1008 calld->path, // path
1009 calld->call_start_time, // start_time
1010 calld->deadline, // deadline
1011 calld->arena, // arena
1012 calld->pick.subchannel_call_context, // context
1013 calld->call_combiner // call_combiner
Yash Tibrewald8b84a22017-09-25 13:38:03 -07001014 };
David Garcia Quintas70fbe622018-01-09 19:27:46 -08001015 grpc_error* new_error = calld->pick.connected_subchannel->CreateCall(
David Garcia Quintasbaf1ac72018-01-09 14:24:32 -08001016 call_args, &calld->subchannel_call);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001017 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001018 gpr_log(GPR_DEBUG, "chand=%p calld=%p: create subchannel_call=%p: error=%s",
Mark D. Roth76e264b2017-08-25 09:03:33 -07001019 chand, calld, calld->subchannel_call, grpc_error_string(new_error));
Mark D. Roth60751fe2017-07-07 12:50:33 -07001020 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001021 if (new_error != GRPC_ERROR_NONE) {
1022 new_error = grpc_error_add_child(new_error, error);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001023 waiting_for_pick_batches_fail(elem, new_error);
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001024 } else {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001025 waiting_for_pick_batches_resume(elem);
Craig Tiller11c17d42017-03-13 13:36:34 -07001026 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001027 GRPC_ERROR_UNREF(error);
Craig Tiller11c17d42017-03-13 13:36:34 -07001028}
1029
Mark D. Rothb2929602017-09-11 09:31:11 -07001030// Invoked when a pick is completed, on both success or failure.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001031static void pick_done_locked(grpc_call_element* elem, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001032 call_data* calld = (call_data*)elem->call_data;
1033 channel_data* chand = (channel_data*)elem->channel_data;
David Garcia Quintasbe1b7f92018-01-12 14:01:38 -08001034 if (calld->pick.connected_subchannel == nullptr) {
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001035 // Failed to create subchannel.
Mark D. Roth76e264b2017-08-25 09:03:33 -07001036 GRPC_ERROR_UNREF(calld->error);
1037 calld->error = error == GRPC_ERROR_NONE
1038 ? GRPC_ERROR_CREATE_FROM_STATIC_STRING(
1039 "Call dropped by load balancing policy")
1040 : GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING(
1041 "Failed to create subchannel", &error, 1);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001042 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001043 gpr_log(GPR_DEBUG,
1044 "chand=%p calld=%p: failed to create subchannel: error=%s", chand,
Mark D. Roth76e264b2017-08-25 09:03:33 -07001045 calld, grpc_error_string(calld->error));
Mark D. Roth60751fe2017-07-07 12:50:33 -07001046 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001047 waiting_for_pick_batches_fail(elem, GRPC_ERROR_REF(calld->error));
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001048 } else {
Mark D. Roth9fe284e2016-09-12 11:22:27 -07001049 /* Create call on subchannel. */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001050 create_subchannel_call_locked(elem, GRPC_ERROR_REF(error));
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001051 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001052 GRPC_ERROR_UNREF(error);
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001053}
1054
Mark D. Rothb2929602017-09-11 09:31:11 -07001055// A wrapper around pick_done_locked() that is used in cases where
1056// either (a) the pick was deferred pending a resolver result or (b) the
1057// pick was done asynchronously. Removes the call's polling entity from
1058// chand->interested_parties before invoking pick_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001059static void async_pick_done_locked(grpc_call_element* elem, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001060 channel_data* chand = (channel_data*)elem->channel_data;
1061 call_data* calld = (call_data*)elem->call_data;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001062 grpc_polling_entity_del_from_pollset_set(calld->pollent,
Mark D. Rothb2929602017-09-11 09:31:11 -07001063 chand->interested_parties);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001064 pick_done_locked(elem, error);
Mark D. Rothb2929602017-09-11 09:31:11 -07001065}
1066
1067// Note: This runs under the client_channel combiner, but will NOT be
1068// holding the call combiner.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001069static void pick_callback_cancel_locked(void* arg, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001070 grpc_call_element* elem = (grpc_call_element*)arg;
1071 channel_data* chand = (channel_data*)elem->channel_data;
1072 call_data* calld = (call_data*)elem->call_data;
Mark D. Rothc0febd32018-01-09 10:25:24 -08001073 // Note: chand->lb_policy may have changed since we started our pick,
1074 // in which case we will be cancelling the pick on a policy other than
1075 // the one we started it on. However, this will just be a no-op.
1076 if (error != GRPC_ERROR_NONE && chand->lb_policy != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001077 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001078 gpr_log(GPR_DEBUG, "chand=%p calld=%p: cancelling pick from LB policy %p",
Mark D. Rothc0febd32018-01-09 10:25:24 -08001079 chand, calld, chand->lb_policy);
Mark D. Rothb2929602017-09-11 09:31:11 -07001080 }
Mark D. Rothc0febd32018-01-09 10:25:24 -08001081 grpc_lb_policy_cancel_pick_locked(chand->lb_policy, &calld->pick,
1082 GRPC_ERROR_REF(error));
Mark D. Rothb2929602017-09-11 09:31:11 -07001083 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001084 GRPC_CALL_STACK_UNREF(calld->owning_call, "pick_callback_cancel");
Mark D. Rothb2929602017-09-11 09:31:11 -07001085}
1086
1087// Callback invoked by grpc_lb_policy_pick_locked() for async picks.
1088// Unrefs the LB policy and invokes async_pick_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001089static void pick_callback_done_locked(void* arg, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001090 grpc_call_element* elem = (grpc_call_element*)arg;
1091 channel_data* chand = (channel_data*)elem->channel_data;
1092 call_data* calld = (call_data*)elem->call_data;
Craig Tiller6014e8a2017-10-16 13:50:29 -07001093 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001094 gpr_log(GPR_DEBUG, "chand=%p calld=%p: pick completed asynchronously",
1095 chand, calld);
1096 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001097 async_pick_done_locked(elem, GRPC_ERROR_REF(error));
Mark D. Rothb2929602017-09-11 09:31:11 -07001098}
1099
1100// Takes a ref to chand->lb_policy and calls grpc_lb_policy_pick_locked().
1101// If the pick was completed synchronously, unrefs the LB policy and
1102// returns true.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001103static bool pick_callback_start_locked(grpc_call_element* elem) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001104 channel_data* chand = (channel_data*)elem->channel_data;
1105 call_data* calld = (call_data*)elem->call_data;
Craig Tiller6014e8a2017-10-16 13:50:29 -07001106 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001107 gpr_log(GPR_DEBUG, "chand=%p calld=%p: starting pick on lb_policy=%p",
1108 chand, calld, chand->lb_policy);
1109 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001110 apply_service_config_to_call_locked(elem);
Mark D. Rothb2929602017-09-11 09:31:11 -07001111 // If the application explicitly set wait_for_ready, use that.
1112 // Otherwise, if the service config specified a value for this
1113 // method, use that.
1114 uint32_t initial_metadata_flags =
1115 calld->initial_metadata_batch->payload->send_initial_metadata
1116 .send_initial_metadata_flags;
1117 const bool wait_for_ready_set_from_api =
1118 initial_metadata_flags &
1119 GRPC_INITIAL_METADATA_WAIT_FOR_READY_EXPLICITLY_SET;
1120 const bool wait_for_ready_set_from_service_config =
Craig Tiller4782d922017-11-10 09:53:21 -08001121 calld->method_params != nullptr &&
Mark D. Rothb2929602017-09-11 09:31:11 -07001122 calld->method_params->wait_for_ready != WAIT_FOR_READY_UNSET;
1123 if (!wait_for_ready_set_from_api && wait_for_ready_set_from_service_config) {
1124 if (calld->method_params->wait_for_ready == WAIT_FOR_READY_TRUE) {
1125 initial_metadata_flags |= GRPC_INITIAL_METADATA_WAIT_FOR_READY;
1126 } else {
1127 initial_metadata_flags &= ~GRPC_INITIAL_METADATA_WAIT_FOR_READY;
1128 }
1129 }
Mark D. Rothc0febd32018-01-09 10:25:24 -08001130 calld->pick.initial_metadata =
Mark D. Rothb2929602017-09-11 09:31:11 -07001131 calld->initial_metadata_batch->payload->send_initial_metadata
Mark D. Rothc0febd32018-01-09 10:25:24 -08001132 .send_initial_metadata;
1133 calld->pick.initial_metadata_flags = initial_metadata_flags;
Mark D. Rothb2929602017-09-11 09:31:11 -07001134 GRPC_CLOSURE_INIT(&calld->lb_pick_closure, pick_callback_done_locked, elem,
1135 grpc_combiner_scheduler(chand->combiner));
Mark D. Rothc0febd32018-01-09 10:25:24 -08001136 calld->pick.on_complete = &calld->lb_pick_closure;
1137 const bool pick_done =
1138 grpc_lb_policy_pick_locked(chand->lb_policy, &calld->pick);
Mark D. Rothb2929602017-09-11 09:31:11 -07001139 if (pick_done) {
1140 /* synchronous grpc_lb_policy_pick call. Unref the LB policy. */
Craig Tiller6014e8a2017-10-16 13:50:29 -07001141 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001142 gpr_log(GPR_DEBUG, "chand=%p calld=%p: pick completed synchronously",
1143 chand, calld);
1144 }
Mark D. Rothb2929602017-09-11 09:31:11 -07001145 } else {
1146 GRPC_CALL_STACK_REF(calld->owning_call, "pick_callback_cancel");
1147 grpc_call_combiner_set_notify_on_cancel(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001148 calld->call_combiner,
Mark D. Rothb2929602017-09-11 09:31:11 -07001149 GRPC_CLOSURE_INIT(&calld->lb_pick_cancel_closure,
1150 pick_callback_cancel_locked, elem,
1151 grpc_combiner_scheduler(chand->combiner)));
1152 }
1153 return pick_done;
1154}
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001155
Craig Tiller577c9b22015-11-02 14:11:15 -08001156typedef struct {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001157 grpc_call_element* elem;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001158 bool finished;
Craig Tiller577c9b22015-11-02 14:11:15 -08001159 grpc_closure closure;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001160 grpc_closure cancel_closure;
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001161} pick_after_resolver_result_args;
Craig Tiller577c9b22015-11-02 14:11:15 -08001162
Mark D. Roth76e264b2017-08-25 09:03:33 -07001163// Note: This runs under the client_channel combiner, but will NOT be
1164// holding the call combiner.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001165static void pick_after_resolver_result_cancel_locked(void* arg,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001166 grpc_error* error) {
1167 pick_after_resolver_result_args* args = (pick_after_resolver_result_args*)arg;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001168 if (args->finished) {
1169 gpr_free(args);
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001170 return;
Mark D. Roth764cf042017-09-01 09:00:06 -07001171 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001172 // If we don't yet have a resolver result, then a closure for
1173 // pick_after_resolver_result_done_locked() will have been added to
1174 // chand->waiting_for_resolver_result_closures, and it may not be invoked
1175 // until after this call has been destroyed. We mark the operation as
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001176 // finished, so that when pick_after_resolver_result_done_locked()
Mark D. Roth76e264b2017-08-25 09:03:33 -07001177 // is called, it will be a no-op. We also immediately invoke
Mark D. Rothb2929602017-09-11 09:31:11 -07001178 // async_pick_done_locked() to propagate the error back to the caller.
1179 args->finished = true;
Craig Tillerbaa14a92017-11-03 09:09:36 -07001180 grpc_call_element* elem = args->elem;
1181 channel_data* chand = (channel_data*)elem->channel_data;
1182 call_data* calld = (call_data*)elem->call_data;
Craig Tiller6014e8a2017-10-16 13:50:29 -07001183 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001184 gpr_log(GPR_DEBUG,
1185 "chand=%p calld=%p: cancelling pick waiting for resolver result",
1186 chand, calld);
Mark D. Roth76e264b2017-08-25 09:03:33 -07001187 }
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001188 // Note: Although we are not in the call combiner here, we are
1189 // basically stealing the call combiner from the pending pick, so
Mark D. Rothb2929602017-09-11 09:31:11 -07001190 // it's safe to call async_pick_done_locked() here -- we are
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001191 // essentially calling it here instead of calling it in
1192 // pick_after_resolver_result_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001193 async_pick_done_locked(elem, GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING(
1194 "Pick cancelled", &error, 1));
Mark D. Roth76e264b2017-08-25 09:03:33 -07001195}
1196
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001197static void pick_after_resolver_result_start_locked(grpc_call_element* elem);
Mark D. Roth48be9de2017-10-23 12:27:37 -07001198
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001199static void pick_after_resolver_result_done_locked(void* arg,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001200 grpc_error* error) {
1201 pick_after_resolver_result_args* args = (pick_after_resolver_result_args*)arg;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001202 if (args->finished) {
Craig Tiller577c9b22015-11-02 14:11:15 -08001203 /* cancelled, do nothing */
Craig Tiller6014e8a2017-10-16 13:50:29 -07001204 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001205 gpr_log(GPR_DEBUG, "call cancelled before resolver result");
1206 }
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001207 gpr_free(args);
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001208 return;
1209 }
1210 args->finished = true;
Craig Tillerbaa14a92017-11-03 09:09:36 -07001211 grpc_call_element* elem = args->elem;
1212 channel_data* chand = (channel_data*)elem->channel_data;
1213 call_data* calld = (call_data*)elem->call_data;
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001214 if (error != GRPC_ERROR_NONE) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001215 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001216 gpr_log(GPR_DEBUG, "chand=%p calld=%p: resolver failed to return data",
1217 chand, calld);
1218 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001219 async_pick_done_locked(elem, GRPC_ERROR_REF(error));
Craig Tiller4782d922017-11-10 09:53:21 -08001220 } else if (chand->lb_policy != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001221 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2b9a0f2017-09-01 09:06:47 -07001222 gpr_log(GPR_DEBUG, "chand=%p calld=%p: resolver returned, doing pick",
1223 chand, calld);
1224 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001225 if (pick_callback_start_locked(elem)) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001226 // Even if the LB policy returns a result synchronously, we have
1227 // already added our polling entity to chand->interested_parties
1228 // in order to wait for the resolver result, so we need to
1229 // remove it here. Therefore, we call async_pick_done_locked()
1230 // instead of pick_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001231 async_pick_done_locked(elem, GRPC_ERROR_NONE);
Mark D. Roth9dab7d52016-10-07 07:48:03 -07001232 }
Craig Tiller577c9b22015-11-02 14:11:15 -08001233 }
Mark D. Roth48be9de2017-10-23 12:27:37 -07001234 // TODO(roth): It should be impossible for chand->lb_policy to be NULL
1235 // here, so the rest of this code should never actually be executed.
1236 // However, we have reports of a crash on iOS that triggers this case,
1237 // so we are temporarily adding this to restore branches that were
1238 // removed in https://github.com/grpc/grpc/pull/12297. Need to figure
1239 // out what is actually causing this to occur and then figure out the
1240 // right way to deal with it.
Craig Tiller4782d922017-11-10 09:53:21 -08001241 else if (chand->resolver != nullptr) {
Mark D. Roth48be9de2017-10-23 12:27:37 -07001242 // No LB policy, so try again.
ncteisen72afb762017-11-10 12:23:12 -08001243 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth48be9de2017-10-23 12:27:37 -07001244 gpr_log(GPR_DEBUG,
1245 "chand=%p calld=%p: resolver returned but no LB policy, "
1246 "trying again",
1247 chand, calld);
1248 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001249 pick_after_resolver_result_start_locked(elem);
Mark D. Roth48be9de2017-10-23 12:27:37 -07001250 } else {
ncteisen72afb762017-11-10 12:23:12 -08001251 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth48be9de2017-10-23 12:27:37 -07001252 gpr_log(GPR_DEBUG, "chand=%p calld=%p: resolver disconnected", chand,
1253 calld);
1254 }
1255 async_pick_done_locked(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001256 elem, GRPC_ERROR_CREATE_FROM_STATIC_STRING("Disconnected"));
Mark D. Roth48be9de2017-10-23 12:27:37 -07001257 }
Craig Tiller577c9b22015-11-02 14:11:15 -08001258}
1259
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001260static void pick_after_resolver_result_start_locked(grpc_call_element* elem) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001261 channel_data* chand = (channel_data*)elem->channel_data;
1262 call_data* calld = (call_data*)elem->call_data;
Craig Tiller6014e8a2017-10-16 13:50:29 -07001263 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001264 gpr_log(GPR_DEBUG,
1265 "chand=%p calld=%p: deferring pick pending resolver result", chand,
1266 calld);
Mark D. Roth64a317c2017-05-02 08:27:08 -07001267 }
Craig Tillerbaa14a92017-11-03 09:09:36 -07001268 pick_after_resolver_result_args* args =
1269 (pick_after_resolver_result_args*)gpr_zalloc(sizeof(*args));
Mark D. Roth60751fe2017-07-07 12:50:33 -07001270 args->elem = elem;
1271 GRPC_CLOSURE_INIT(&args->closure, pick_after_resolver_result_done_locked,
1272 args, grpc_combiner_scheduler(chand->combiner));
1273 grpc_closure_list_append(&chand->waiting_for_resolver_result_closures,
1274 &args->closure, GRPC_ERROR_NONE);
Mark D. Roth76e264b2017-08-25 09:03:33 -07001275 grpc_call_combiner_set_notify_on_cancel(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001276 calld->call_combiner,
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001277 GRPC_CLOSURE_INIT(&args->cancel_closure,
1278 pick_after_resolver_result_cancel_locked, args,
Mark D. Roth76e264b2017-08-25 09:03:33 -07001279 grpc_combiner_scheduler(chand->combiner)));
Mark D. Roth60751fe2017-07-07 12:50:33 -07001280}
1281
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001282static void start_pick_locked(void* arg, grpc_error* ignored) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001283 grpc_call_element* elem = (grpc_call_element*)arg;
1284 call_data* calld = (call_data*)elem->call_data;
1285 channel_data* chand = (channel_data*)elem->channel_data;
David Garcia Quintasbe1b7f92018-01-12 14:01:38 -08001286 GPR_ASSERT(calld->pick.connected_subchannel == nullptr);
Craig Tiller4782d922017-11-10 09:53:21 -08001287 if (chand->lb_policy != nullptr) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001288 // We already have an LB policy, so ask it for a pick.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001289 if (pick_callback_start_locked(elem)) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001290 // Pick completed synchronously.
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001291 pick_done_locked(elem, GRPC_ERROR_NONE);
Mark D. Rothb2929602017-09-11 09:31:11 -07001292 return;
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001293 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001294 } else {
Mark D. Rothb2929602017-09-11 09:31:11 -07001295 // We do not yet have an LB policy, so wait for a resolver result.
Craig Tiller4782d922017-11-10 09:53:21 -08001296 if (chand->resolver == nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001297 pick_done_locked(elem,
Mark D. Rothb2929602017-09-11 09:31:11 -07001298 GRPC_ERROR_CREATE_FROM_STATIC_STRING("Disconnected"));
1299 return;
1300 }
1301 if (!chand->started_resolving) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001302 start_resolving_locked(chand);
Mark D. Rothb2929602017-09-11 09:31:11 -07001303 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001304 pick_after_resolver_result_start_locked(elem);
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001305 }
Mark D. Rothb2929602017-09-11 09:31:11 -07001306 // We need to wait for either a resolver result or for an async result
1307 // from the LB policy. Add the polling entity from call_data to the
1308 // channel_data's interested_parties, so that the I/O of the LB policy
1309 // and resolver can be done under it. The polling entity will be
1310 // removed in async_pick_done_locked().
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001311 grpc_polling_entity_add_to_pollset_set(calld->pollent,
Mark D. Rothb2929602017-09-11 09:31:11 -07001312 chand->interested_parties);
Craig Tillera11bfc82017-02-14 09:56:33 -08001313}
1314
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001315static void on_complete(void* arg, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001316 grpc_call_element* elem = (grpc_call_element*)arg;
1317 call_data* calld = (call_data*)elem->call_data;
Craig Tiller4782d922017-11-10 09:53:21 -08001318 if (calld->retry_throttle_data != nullptr) {
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001319 if (error == GRPC_ERROR_NONE) {
1320 grpc_server_retry_throttle_data_record_success(
Mark D. Roth9ccbc4d2017-03-15 08:30:04 -07001321 calld->retry_throttle_data);
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001322 } else {
1323 // TODO(roth): In a subsequent PR, check the return value here and
Mark D. Rothb3322562017-02-23 14:38:02 -08001324 // decide whether or not to retry. Note that we should only
1325 // record failures whose statuses match the configured retryable
1326 // or non-fatal status codes.
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001327 grpc_server_retry_throttle_data_record_failure(
Mark D. Roth9ccbc4d2017-03-15 08:30:04 -07001328 calld->retry_throttle_data);
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001329 }
1330 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001331 GRPC_CLOSURE_RUN(calld->original_on_complete, GRPC_ERROR_REF(error));
Mark D. Rothd6d192d2017-02-23 08:58:42 -08001332}
1333
Craig Tillere1b51da2017-03-31 15:44:33 -07001334static void cc_start_transport_stream_op_batch(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001335 grpc_call_element* elem, grpc_transport_stream_op_batch* batch) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001336 call_data* calld = (call_data*)elem->call_data;
1337 channel_data* chand = (channel_data*)elem->channel_data;
Craig Tiller3be7dd02017-04-03 14:30:03 -07001338 if (chand->deadline_checking_enabled) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001339 grpc_deadline_state_client_start_transport_stream_op_batch(elem, batch);
Craig Tiller3be7dd02017-04-03 14:30:03 -07001340 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001341 GPR_TIMER_BEGIN("cc_start_transport_stream_op_batch", 0);
1342 // If we've previously been cancelled, immediately fail any new batches.
1343 if (calld->error != GRPC_ERROR_NONE) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001344 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth76e264b2017-08-25 09:03:33 -07001345 gpr_log(GPR_DEBUG, "chand=%p calld=%p: failing batch with error: %s",
1346 chand, calld, grpc_error_string(calld->error));
1347 }
1348 grpc_transport_stream_op_batch_finish_with_failure(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001349 batch, GRPC_ERROR_REF(calld->error), calld->call_combiner);
Mark D. Roth76e264b2017-08-25 09:03:33 -07001350 goto done;
1351 }
1352 if (batch->cancel_stream) {
1353 // Stash a copy of cancel_error in our call data, so that we can use
1354 // it for subsequent operations. This ensures that if the call is
1355 // cancelled before any batches are passed down (e.g., if the deadline
1356 // is in the past when the call starts), we can return the right
1357 // error to the caller when the first batch does get passed down.
1358 GRPC_ERROR_UNREF(calld->error);
1359 calld->error = GRPC_ERROR_REF(batch->payload->cancel_stream.cancel_error);
Craig Tiller6014e8a2017-10-16 13:50:29 -07001360 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth76e264b2017-08-25 09:03:33 -07001361 gpr_log(GPR_DEBUG, "chand=%p calld=%p: recording cancel_error=%s", chand,
1362 calld, grpc_error_string(calld->error));
1363 }
1364 // If we have a subchannel call, send the cancellation batch down.
1365 // Otherwise, fail all pending batches.
Craig Tiller4782d922017-11-10 09:53:21 -08001366 if (calld->subchannel_call != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001367 grpc_subchannel_call_process_op(calld->subchannel_call, batch);
Mark D. Roth76e264b2017-08-25 09:03:33 -07001368 } else {
1369 waiting_for_pick_batches_add(calld, batch);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001370 waiting_for_pick_batches_fail(elem, GRPC_ERROR_REF(calld->error));
Mark D. Roth76e264b2017-08-25 09:03:33 -07001371 }
1372 goto done;
1373 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001374 // Intercept on_complete for recv_trailing_metadata so that we can
1375 // check retry throttle status.
Mark D. Roth60751fe2017-07-07 12:50:33 -07001376 if (batch->recv_trailing_metadata) {
Craig Tiller4782d922017-11-10 09:53:21 -08001377 GPR_ASSERT(batch->on_complete != nullptr);
Mark D. Roth60751fe2017-07-07 12:50:33 -07001378 calld->original_on_complete = batch->on_complete;
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001379 GRPC_CLOSURE_INIT(&calld->on_complete, on_complete, elem,
1380 grpc_schedule_on_exec_ctx);
Mark D. Roth60751fe2017-07-07 12:50:33 -07001381 batch->on_complete = &calld->on_complete;
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001382 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001383 // Check if we've already gotten a subchannel call.
1384 // Note that once we have completed the pick, we do not need to enter
1385 // the channel combiner, which is more efficient (especially for
1386 // streaming calls).
Craig Tiller4782d922017-11-10 09:53:21 -08001387 if (calld->subchannel_call != nullptr) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001388 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth60751fe2017-07-07 12:50:33 -07001389 gpr_log(GPR_DEBUG,
1390 "chand=%p calld=%p: sending batch to subchannel_call=%p", chand,
Mark D. Roth76e264b2017-08-25 09:03:33 -07001391 calld, calld->subchannel_call);
Mark D. Roth60751fe2017-07-07 12:50:33 -07001392 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001393 grpc_subchannel_call_process_op(calld->subchannel_call, batch);
Mark D. Roth60751fe2017-07-07 12:50:33 -07001394 goto done;
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001395 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001396 // We do not yet have a subchannel call.
1397 // Add the batch to the waiting-for-pick list.
1398 waiting_for_pick_batches_add(calld, batch);
1399 // For batches containing a send_initial_metadata op, enter the channel
1400 // combiner to start a pick.
1401 if (batch->send_initial_metadata) {
Craig Tiller6014e8a2017-10-16 13:50:29 -07001402 if (grpc_client_channel_trace.enabled()) {
Mark D. Rothb2929602017-09-11 09:31:11 -07001403 gpr_log(GPR_DEBUG, "chand=%p calld=%p: entering client_channel combiner",
1404 chand, calld);
Mark D. Roth76e264b2017-08-25 09:03:33 -07001405 }
1406 GRPC_CLOSURE_SCHED(
Mark D. Roth76e264b2017-08-25 09:03:33 -07001407 GRPC_CLOSURE_INIT(&batch->handler_private.closure, start_pick_locked,
1408 elem, grpc_combiner_scheduler(chand->combiner)),
1409 GRPC_ERROR_NONE);
1410 } else {
1411 // For all other batches, release the call combiner.
Craig Tiller6014e8a2017-10-16 13:50:29 -07001412 if (grpc_client_channel_trace.enabled()) {
Mark D. Roth76e264b2017-08-25 09:03:33 -07001413 gpr_log(GPR_DEBUG,
1414 "chand=%p calld=%p: saved batch, yeilding call combiner", chand,
1415 calld);
1416 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001417 GRPC_CALL_COMBINER_STOP(calld->call_combiner,
Mark D. Roth76e264b2017-08-25 09:03:33 -07001418 "batch does not include send_initial_metadata");
Mark D. Roth60751fe2017-07-07 12:50:33 -07001419 }
Mark D. Roth60751fe2017-07-07 12:50:33 -07001420done:
Craig Tillera0f3abd2017-03-31 15:42:16 -07001421 GPR_TIMER_END("cc_start_transport_stream_op_batch", 0);
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001422}
1423
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001424/* Constructor for call_data */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001425static grpc_error* cc_init_call_elem(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001426 const grpc_call_element_args* args) {
1427 call_data* calld = (call_data*)elem->call_data;
1428 channel_data* chand = (channel_data*)elem->channel_data;
Mark D. Rothe40dd292016-10-05 14:58:37 -07001429 // Initialize data members.
Craig Tiller7c70b6c2017-01-23 07:48:42 -08001430 calld->path = grpc_slice_ref_internal(args->path);
Mark D. Rothff08f332016-10-14 13:01:01 -07001431 calld->call_start_time = args->start_time;
Craig Tiller89c14282017-07-19 15:32:27 -07001432 calld->deadline = args->deadline;
Craig Tillerd426cac2017-03-13 12:30:45 -07001433 calld->arena = args->arena;
Mark D. Roth66f3d2b2017-09-01 09:02:17 -07001434 calld->owning_call = args->call_stack;
Mark D. Roth76e264b2017-08-25 09:03:33 -07001435 calld->call_combiner = args->call_combiner;
Craig Tiller3be7dd02017-04-03 14:30:03 -07001436 if (chand->deadline_checking_enabled) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001437 grpc_deadline_state_init(elem, args->call_stack, args->call_combiner,
1438 calld->deadline);
Craig Tiller3be7dd02017-04-03 14:30:03 -07001439 }
Mark D. Roth0badbe82016-06-23 10:15:12 -07001440 return GRPC_ERROR_NONE;
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001441}
1442
1443/* Destructor for call_data */
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001444static void cc_destroy_call_elem(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001445 const grpc_call_final_info* final_info,
1446 grpc_closure* then_schedule_closure) {
1447 call_data* calld = (call_data*)elem->call_data;
1448 channel_data* chand = (channel_data*)elem->channel_data;
Craig Tiller3be7dd02017-04-03 14:30:03 -07001449 if (chand->deadline_checking_enabled) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001450 grpc_deadline_state_destroy(elem);
Craig Tiller3be7dd02017-04-03 14:30:03 -07001451 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001452 grpc_slice_unref_internal(calld->path);
Craig Tiller4782d922017-11-10 09:53:21 -08001453 if (calld->method_params != nullptr) {
Mark D. Roth95b627b2017-02-24 11:02:58 -08001454 method_parameters_unref(calld->method_params);
1455 }
Mark D. Roth76e264b2017-08-25 09:03:33 -07001456 GRPC_ERROR_UNREF(calld->error);
Craig Tiller4782d922017-11-10 09:53:21 -08001457 if (calld->subchannel_call != nullptr) {
Mark D. Roth76e264b2017-08-25 09:03:33 -07001458 grpc_subchannel_call_set_cleanup_closure(calld->subchannel_call,
Craig Tillerf7c8c9f2017-05-17 15:22:05 -07001459 then_schedule_closure);
Craig Tiller4782d922017-11-10 09:53:21 -08001460 then_schedule_closure = nullptr;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001461 GRPC_SUBCHANNEL_CALL_UNREF(calld->subchannel_call,
Craig Tillerf7c8c9f2017-05-17 15:22:05 -07001462 "client_channel_destroy_call");
Mark D. Roth4c0fe492016-08-31 13:51:55 -07001463 }
Mark D. Roth0ca0be82017-06-20 07:49:33 -07001464 GPR_ASSERT(calld->waiting_for_pick_batches_count == 0);
David Garcia Quintasbe1b7f92018-01-12 14:01:38 -08001465 if (calld->pick.connected_subchannel != nullptr) {
David Garcia Quintasdfa28512018-01-11 18:31:13 -08001466 calld->pick.connected_subchannel.reset();
Craig Tiller693d3942016-10-27 16:51:25 -07001467 }
Mark D. Roth09e458c2017-05-02 08:13:26 -07001468 for (size_t i = 0; i < GRPC_CONTEXT_COUNT; ++i) {
Mark D. Rothc0febd32018-01-09 10:25:24 -08001469 if (calld->pick.subchannel_call_context[i].value != nullptr) {
1470 calld->pick.subchannel_call_context[i].destroy(
1471 calld->pick.subchannel_call_context[i].value);
Mark D. Roth09e458c2017-05-02 08:13:26 -07001472 }
1473 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001474 GRPC_CLOSURE_SCHED(then_schedule_closure, GRPC_ERROR_NONE);
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001475}
1476
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001477static void cc_set_pollset_or_pollset_set(grpc_call_element* elem,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001478 grpc_polling_entity* pollent) {
1479 call_data* calld = (call_data*)elem->call_data;
David Garcia Quintas2a50dfe2016-05-31 15:09:12 -07001480 calld->pollent = pollent;
Craig Tiller577c9b22015-11-02 14:11:15 -08001481}
1482
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001483/*************************************************************************
1484 * EXPORTED SYMBOLS
1485 */
1486
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001487const grpc_channel_filter grpc_client_channel_filter = {
Craig Tillera0f3abd2017-03-31 15:42:16 -07001488 cc_start_transport_stream_op_batch,
Craig Tillerf40df232016-03-25 13:38:14 -07001489 cc_start_transport_op,
1490 sizeof(call_data),
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001491 cc_init_call_elem,
David Garcia Quintas4afce7e2016-04-18 16:25:17 -07001492 cc_set_pollset_or_pollset_set,
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001493 cc_destroy_call_elem,
Craig Tillerf40df232016-03-25 13:38:14 -07001494 sizeof(channel_data),
Mark D. Roth2a5959f2016-09-01 08:20:27 -07001495 cc_init_channel_elem,
1496 cc_destroy_channel_elem,
Mark D. Rothb2d24882016-10-27 15:44:07 -07001497 cc_get_channel_info,
Craig Tillerf40df232016-03-25 13:38:14 -07001498 "client-channel",
Craig Tiller87d5b192015-04-16 14:37:57 -07001499};
Nicolas Nobleb7ebd3b2014-11-26 16:33:03 -08001500
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001501static void try_to_connect_locked(void* arg, grpc_error* error_ignored) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001502 channel_data* chand = (channel_data*)arg;
Craig Tiller4782d922017-11-10 09:53:21 -08001503 if (chand->lb_policy != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001504 grpc_lb_policy_exit_idle_locked(chand->lb_policy);
Craig Tiller613dafa2017-02-09 12:00:43 -08001505 } else {
1506 chand->exit_idle_when_lb_policy_arrives = true;
Craig Tiller4782d922017-11-10 09:53:21 -08001507 if (!chand->started_resolving && chand->resolver != nullptr) {
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001508 start_resolving_locked(chand);
Craig Tiller613dafa2017-02-09 12:00:43 -08001509 }
1510 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001511 GRPC_CHANNEL_STACK_UNREF(chand->owning_stack, "try_to_connect");
Craig Tiller613dafa2017-02-09 12:00:43 -08001512}
1513
Craig Tillera82950e2015-09-22 12:33:20 -07001514grpc_connectivity_state grpc_client_channel_check_connectivity_state(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001515 grpc_channel_element* elem, int try_to_connect) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001516 channel_data* chand = (channel_data*)elem->channel_data;
Craig Tillera8610c02017-02-14 10:05:11 -08001517 grpc_connectivity_state out =
1518 grpc_connectivity_state_check(&chand->state_tracker);
Craig Tillera82950e2015-09-22 12:33:20 -07001519 if (out == GRPC_CHANNEL_IDLE && try_to_connect) {
Craig Tillerd2e5cfc2017-02-09 13:02:20 -08001520 GRPC_CHANNEL_STACK_REF(chand->owning_stack, "try_to_connect");
ncteisen274bbbe2017-06-08 14:57:11 -07001521 GRPC_CLOSURE_SCHED(
Yash Tibrewal0ee75742017-10-13 16:07:13 -07001522 GRPC_CLOSURE_CREATE(try_to_connect_locked, chand,
1523 grpc_combiner_scheduler(chand->combiner)),
Craig Tiller613dafa2017-02-09 12:00:43 -08001524 GRPC_ERROR_NONE);
Craig Tillera82950e2015-09-22 12:33:20 -07001525 }
Craig Tiller48cb07c2015-07-15 16:16:15 -07001526 return out;
1527}
1528
Alexander Polcync3b1f182017-04-18 13:51:36 -07001529typedef struct external_connectivity_watcher {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001530 channel_data* chand;
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001531 grpc_polling_entity pollent;
Craig Tillerbaa14a92017-11-03 09:09:36 -07001532 grpc_closure* on_complete;
1533 grpc_closure* watcher_timer_init;
1534 grpc_connectivity_state* state;
Craig Tiller86c99582015-11-25 15:22:26 -08001535 grpc_closure my_closure;
Craig Tillerbaa14a92017-11-03 09:09:36 -07001536 struct external_connectivity_watcher* next;
Craig Tiller86c99582015-11-25 15:22:26 -08001537} external_connectivity_watcher;
1538
Craig Tillerbaa14a92017-11-03 09:09:36 -07001539static external_connectivity_watcher* lookup_external_connectivity_watcher(
1540 channel_data* chand, grpc_closure* on_complete) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001541 gpr_mu_lock(&chand->external_connectivity_watcher_list_mu);
Craig Tillerbaa14a92017-11-03 09:09:36 -07001542 external_connectivity_watcher* w =
Alexander Polcync3b1f182017-04-18 13:51:36 -07001543 chand->external_connectivity_watcher_list_head;
Craig Tiller4782d922017-11-10 09:53:21 -08001544 while (w != nullptr && w->on_complete != on_complete) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001545 w = w->next;
1546 }
1547 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
1548 return w;
1549}
1550
1551static void external_connectivity_watcher_list_append(
Craig Tillerbaa14a92017-11-03 09:09:36 -07001552 channel_data* chand, external_connectivity_watcher* w) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001553 GPR_ASSERT(!lookup_external_connectivity_watcher(chand, w->on_complete));
1554
1555 gpr_mu_lock(&w->chand->external_connectivity_watcher_list_mu);
1556 GPR_ASSERT(!w->next);
1557 w->next = chand->external_connectivity_watcher_list_head;
1558 chand->external_connectivity_watcher_list_head = w;
1559 gpr_mu_unlock(&w->chand->external_connectivity_watcher_list_mu);
1560}
1561
1562static void external_connectivity_watcher_list_remove(
Craig Tillerbaa14a92017-11-03 09:09:36 -07001563 channel_data* chand, external_connectivity_watcher* too_remove) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001564 GPR_ASSERT(
1565 lookup_external_connectivity_watcher(chand, too_remove->on_complete));
1566 gpr_mu_lock(&chand->external_connectivity_watcher_list_mu);
1567 if (too_remove == chand->external_connectivity_watcher_list_head) {
1568 chand->external_connectivity_watcher_list_head = too_remove->next;
1569 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
1570 return;
1571 }
Craig Tillerbaa14a92017-11-03 09:09:36 -07001572 external_connectivity_watcher* w =
Alexander Polcync3b1f182017-04-18 13:51:36 -07001573 chand->external_connectivity_watcher_list_head;
Craig Tiller4782d922017-11-10 09:53:21 -08001574 while (w != nullptr) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001575 if (w->next == too_remove) {
1576 w->next = w->next->next;
1577 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
1578 return;
1579 }
1580 w = w->next;
1581 }
1582 GPR_UNREACHABLE_CODE(return );
1583}
1584
1585int grpc_client_channel_num_external_connectivity_watchers(
Craig Tillerbaa14a92017-11-03 09:09:36 -07001586 grpc_channel_element* elem) {
1587 channel_data* chand = (channel_data*)elem->channel_data;
Alexander Polcync3b1f182017-04-18 13:51:36 -07001588 int count = 0;
1589
1590 gpr_mu_lock(&chand->external_connectivity_watcher_list_mu);
Craig Tillerbaa14a92017-11-03 09:09:36 -07001591 external_connectivity_watcher* w =
Alexander Polcync3b1f182017-04-18 13:51:36 -07001592 chand->external_connectivity_watcher_list_head;
Craig Tiller4782d922017-11-10 09:53:21 -08001593 while (w != nullptr) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001594 count++;
1595 w = w->next;
1596 }
1597 gpr_mu_unlock(&chand->external_connectivity_watcher_list_mu);
1598
1599 return count;
1600}
1601
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001602static void on_external_watch_complete_locked(void* arg, grpc_error* error) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001603 external_connectivity_watcher* w = (external_connectivity_watcher*)arg;
1604 grpc_closure* follow_up = w->on_complete;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001605 grpc_polling_entity_del_from_pollset_set(&w->pollent,
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001606 w->chand->interested_parties);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001607 GRPC_CHANNEL_STACK_UNREF(w->chand->owning_stack,
Craig Tiller1d881fb2015-12-01 07:39:04 -08001608 "external_connectivity_watcher");
Alexander Polcync3b1f182017-04-18 13:51:36 -07001609 external_connectivity_watcher_list_remove(w->chand, w);
Craig Tiller86c99582015-11-25 15:22:26 -08001610 gpr_free(w);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001611 GRPC_CLOSURE_RUN(follow_up, GRPC_ERROR_REF(error));
Craig Tiller613dafa2017-02-09 12:00:43 -08001612}
1613
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001614static void watch_connectivity_state_locked(void* arg,
Craig Tillerbaa14a92017-11-03 09:09:36 -07001615 grpc_error* error_ignored) {
1616 external_connectivity_watcher* w = (external_connectivity_watcher*)arg;
Craig Tiller4782d922017-11-10 09:53:21 -08001617 external_connectivity_watcher* found = nullptr;
1618 if (w->state != nullptr) {
Alexander Polcync3b1f182017-04-18 13:51:36 -07001619 external_connectivity_watcher_list_append(w->chand, w);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001620 GRPC_CLOSURE_RUN(w->watcher_timer_init, GRPC_ERROR_NONE);
Alexander Polcyn2004e392017-10-16 15:14:46 -07001621 GRPC_CLOSURE_INIT(&w->my_closure, on_external_watch_complete_locked, w,
1622 grpc_combiner_scheduler(w->chand->combiner));
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001623 grpc_connectivity_state_notify_on_state_change(&w->chand->state_tracker,
1624 w->state, &w->my_closure);
Alexander Polcync3b1f182017-04-18 13:51:36 -07001625 } else {
Craig Tiller4782d922017-11-10 09:53:21 -08001626 GPR_ASSERT(w->watcher_timer_init == nullptr);
Alexander Polcync3b1f182017-04-18 13:51:36 -07001627 found = lookup_external_connectivity_watcher(w->chand, w->on_complete);
1628 if (found) {
1629 GPR_ASSERT(found->on_complete == w->on_complete);
1630 grpc_connectivity_state_notify_on_state_change(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001631 &found->chand->state_tracker, nullptr, &found->my_closure);
Alexander Polcync3b1f182017-04-18 13:51:36 -07001632 }
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001633 grpc_polling_entity_del_from_pollset_set(&w->pollent,
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001634 w->chand->interested_parties);
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001635 GRPC_CHANNEL_STACK_UNREF(w->chand->owning_stack,
Alexander Polcync3b1f182017-04-18 13:51:36 -07001636 "external_connectivity_watcher");
1637 gpr_free(w);
1638 }
Craig Tiller86c99582015-11-25 15:22:26 -08001639}
1640
Craig Tillera82950e2015-09-22 12:33:20 -07001641void grpc_client_channel_watch_connectivity_state(
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001642 grpc_channel_element* elem, grpc_polling_entity pollent,
1643 grpc_connectivity_state* state, grpc_closure* closure,
1644 grpc_closure* watcher_timer_init) {
Craig Tillerbaa14a92017-11-03 09:09:36 -07001645 channel_data* chand = (channel_data*)elem->channel_data;
1646 external_connectivity_watcher* w =
1647 (external_connectivity_watcher*)gpr_zalloc(sizeof(*w));
Craig Tiller86c99582015-11-25 15:22:26 -08001648 w->chand = chand;
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001649 w->pollent = pollent;
Mark D. Roth92210832017-05-02 15:04:39 -07001650 w->on_complete = closure;
Craig Tiller613dafa2017-02-09 12:00:43 -08001651 w->state = state;
Alexander Polcync3b1f182017-04-18 13:51:36 -07001652 w->watcher_timer_init = watcher_timer_init;
Yash Tibrewal8cf14702017-12-06 09:47:54 -08001653 grpc_polling_entity_add_to_pollset_set(&w->pollent,
David Garcia Quintas87d5a312017-06-06 19:45:58 -07001654 chand->interested_parties);
Craig Tiller1d881fb2015-12-01 07:39:04 -08001655 GRPC_CHANNEL_STACK_REF(w->chand->owning_stack,
1656 "external_connectivity_watcher");
ncteisen274bbbe2017-06-08 14:57:11 -07001657 GRPC_CLOSURE_SCHED(
ncteisen274bbbe2017-06-08 14:57:11 -07001658 GRPC_CLOSURE_INIT(&w->my_closure, watch_connectivity_state_locked, w,
Craig Tilleree4b1452017-05-12 10:56:03 -07001659 grpc_combiner_scheduler(chand->combiner)),
Craig Tiller613dafa2017-02-09 12:00:43 -08001660 GRPC_ERROR_NONE);
Craig Tiller48cb07c2015-07-15 16:16:15 -07001661}