Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 1 | /* |
| 2 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | * Copyright 2016 gRPC authors. |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 4 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 5 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | * you may not use this file except in compliance with the License. |
| 7 | * You may obtain a copy of the License at |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 8 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | * http://www.apache.org/licenses/LICENSE-2.0 |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 10 | * |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 11 | * Unless required by applicable law or agreed to in writing, software |
| 12 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | * See the License for the specific language governing permissions and |
| 15 | * limitations under the License. |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 16 | * |
| 17 | */ |
| 18 | |
| 19 | #include <grpc/support/port_platform.h> |
Yuchen Zeng | d6b4985 | 2017-03-20 01:44:27 -0700 | [diff] [blame] | 20 | #if GRPC_ARES == 1 && !defined(GRPC_UV) |
Yuchen Zeng | d79f92a | 2016-08-08 01:18:32 -0700 | [diff] [blame] | 21 | |
Craig Tiller | 9eb0fde | 2017-03-31 16:59:30 -0700 | [diff] [blame] | 22 | #include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.h" |
Yuchen Zeng | 299dd8d | 2016-08-16 21:40:13 -0700 | [diff] [blame] | 23 | #include "src/core/lib/iomgr/sockaddr.h" |
Yuchen Zeng | 6eb83cc | 2016-08-23 00:44:18 -0700 | [diff] [blame] | 24 | #include "src/core/lib/iomgr/socket_utils_posix.h" |
Yuchen Zeng | 85750b0 | 2016-08-08 14:16:34 -0700 | [diff] [blame] | 25 | |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 26 | #include <string.h> |
| 27 | #include <sys/types.h> |
| 28 | |
| 29 | #include <ares.h> |
| 30 | #include <grpc/support/alloc.h> |
| 31 | #include <grpc/support/host_port.h> |
| 32 | #include <grpc/support/log.h> |
| 33 | #include <grpc/support/string_util.h> |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 34 | #include <grpc/support/time.h> |
| 35 | #include <grpc/support/useful.h> |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 36 | |
| 37 | #include "src/core/ext/filters/client_channel/parse_address.h" |
Craig Tiller | 9eb0fde | 2017-03-31 16:59:30 -0700 | [diff] [blame] | 38 | #include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver.h" |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 39 | #include "src/core/lib/iomgr/error.h" |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 40 | #include "src/core/lib/iomgr/executor.h" |
| 41 | #include "src/core/lib/iomgr/iomgr_internal.h" |
Yuchen Zeng | f630ff2 | 2017-07-25 13:28:45 -0700 | [diff] [blame] | 42 | #include "src/core/lib/iomgr/nameser.h" |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 43 | #include "src/core/lib/iomgr/sockaddr_utils.h" |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 44 | #include "src/core/lib/support/string.h" |
| 45 | |
Yuchen Zeng | 459480b | 2016-08-09 11:40:26 -0700 | [diff] [blame] | 46 | static gpr_once g_basic_init = GPR_ONCE_INIT; |
| 47 | static gpr_mu g_init_mu; |
| 48 | |
Yuchen Zeng | 3b4bed2 | 2017-06-04 01:41:15 -0700 | [diff] [blame] | 49 | struct grpc_ares_request { |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 50 | /** indicates the DNS server to use, if specified */ |
| 51 | struct ares_addr_port_node dns_server_addr; |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 52 | /** following members are set in grpc_resolve_address_ares_impl */ |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 53 | /** closure to call when the request completes */ |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 54 | grpc_closure *on_done; |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 55 | /** the pointer to receive the resolved addresses */ |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 56 | grpc_lb_addresses **lb_addrs_out; |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 57 | /** the pointer to receive the service config in JSON */ |
| 58 | char **service_config_json_out; |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 59 | /** the evernt driver used by this request */ |
Yuchen Zeng | 2648a95 | 2016-11-02 18:52:25 -0700 | [diff] [blame] | 60 | grpc_ares_ev_driver *ev_driver; |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 61 | /** number of ongoing queries */ |
Yuchen Zeng | 2648a95 | 2016-11-02 18:52:25 -0700 | [diff] [blame] | 62 | gpr_refcount pending_queries; |
| 63 | |
| 64 | /** mutex guarding the rest of the state */ |
| 65 | gpr_mu mu; |
Yuchen Zeng | 912327e | 2016-10-25 18:27:17 -0700 | [diff] [blame] | 66 | /** is there at least one successful query, set in on_done_cb */ |
| 67 | bool success; |
| 68 | /** the errors explaining the request failure, set in on_done_cb */ |
Yuchen Zeng | 1d0c1c4 | 2016-08-18 15:36:24 -0700 | [diff] [blame] | 69 | grpc_error *error; |
Yuchen Zeng | 3b4bed2 | 2017-06-04 01:41:15 -0700 | [diff] [blame] | 70 | }; |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 71 | |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 72 | typedef struct grpc_ares_hostbyname_request { |
| 73 | /** following members are set in create_hostbyname_request */ |
| 74 | /** the top-level request instance */ |
| 75 | grpc_ares_request *parent_request; |
| 76 | /** host to resolve, parsed from the name to resolve */ |
| 77 | char *host; |
| 78 | /** port to fill in sockaddr_in, parsed from the name to resolve */ |
| 79 | uint16_t port; |
| 80 | /** is it a grpclb address */ |
| 81 | bool is_balancer; |
| 82 | } grpc_ares_hostbyname_request; |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 83 | |
Yuchen Zeng | b48000c | 2016-08-09 22:44:23 -0700 | [diff] [blame] | 84 | static void do_basic_init(void) { gpr_mu_init(&g_init_mu); } |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 85 | |
Yuchen Zeng | a0dbd9f | 2016-08-16 23:08:09 -0700 | [diff] [blame] | 86 | static uint16_t strhtons(const char *port) { |
| 87 | if (strcmp(port, "http") == 0) { |
| 88 | return htons(80); |
| 89 | } else if (strcmp(port, "https") == 0) { |
| 90 | return htons(443); |
| 91 | } |
| 92 | return htons((unsigned short)atoi(port)); |
| 93 | } |
| 94 | |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 95 | static void grpc_ares_request_ref(grpc_ares_request *r) { |
| 96 | gpr_ref(&r->pending_queries); |
| 97 | } |
| 98 | |
Yuchen Zeng | 117a300 | 2016-11-17 19:43:36 -0800 | [diff] [blame] | 99 | static void grpc_ares_request_unref(grpc_exec_ctx *exec_ctx, |
| 100 | grpc_ares_request *r) { |
Yuchen Zeng | e5230b7 | 2017-03-17 15:10:49 -0700 | [diff] [blame] | 101 | /* If there are no pending queries, invoke on_done callback and destroy the |
| 102 | request */ |
Yuchen Zeng | 117a300 | 2016-11-17 19:43:36 -0800 | [diff] [blame] | 103 | if (gpr_unref(&r->pending_queries)) { |
Yuchen Zeng | 1561862 | 2017-03-19 22:31:14 -0700 | [diff] [blame] | 104 | /* TODO(zyc): Sort results with RFC6724 before invoking on_done. */ |
Yuchen Zeng | 117a300 | 2016-11-17 19:43:36 -0800 | [diff] [blame] | 105 | if (exec_ctx == NULL) { |
Yuchen Zeng | e5230b7 | 2017-03-17 15:10:49 -0700 | [diff] [blame] | 106 | /* A new exec_ctx is created here, as the c-ares interface does not |
| 107 | provide one in ares_host_callback. It's safe to schedule on_done with |
| 108 | the newly created exec_ctx, since the caller has been warned not to |
| 109 | acquire locks in on_done. ares_dns_resolver is using combiner to |
| 110 | protect resources needed by on_done. */ |
Yuchen Zeng | 117a300 | 2016-11-17 19:43:36 -0800 | [diff] [blame] | 111 | grpc_exec_ctx new_exec_ctx = GRPC_EXEC_CTX_INIT; |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 112 | GRPC_CLOSURE_SCHED(&new_exec_ctx, r->on_done, r->error); |
Yuchen Zeng | 117a300 | 2016-11-17 19:43:36 -0800 | [diff] [blame] | 113 | grpc_exec_ctx_finish(&new_exec_ctx); |
| 114 | } else { |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 115 | GRPC_CLOSURE_SCHED(exec_ctx, r->on_done, r->error); |
Yuchen Zeng | 117a300 | 2016-11-17 19:43:36 -0800 | [diff] [blame] | 116 | } |
| 117 | gpr_mu_destroy(&r->mu); |
| 118 | grpc_ares_ev_driver_destroy(r->ev_driver); |
Yuchen Zeng | 117a300 | 2016-11-17 19:43:36 -0800 | [diff] [blame] | 119 | gpr_free(r); |
| 120 | } |
| 121 | } |
| 122 | |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 123 | static grpc_ares_hostbyname_request *create_hostbyname_request( |
| 124 | grpc_ares_request *parent_request, char *host, uint16_t port, |
| 125 | bool is_balancer) { |
Yash Tibrewal | bc130da | 2017-09-12 22:44:08 -0700 | [diff] [blame] | 126 | grpc_ares_hostbyname_request *hr = (grpc_ares_hostbyname_request *)gpr_zalloc( |
| 127 | sizeof(grpc_ares_hostbyname_request)); |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 128 | hr->parent_request = parent_request; |
| 129 | hr->host = gpr_strdup(host); |
| 130 | hr->port = port; |
| 131 | hr->is_balancer = is_balancer; |
| 132 | grpc_ares_request_ref(parent_request); |
| 133 | return hr; |
| 134 | } |
| 135 | |
| 136 | static void destroy_hostbyname_request(grpc_exec_ctx *exec_ctx, |
| 137 | grpc_ares_hostbyname_request *hr) { |
| 138 | grpc_ares_request_unref(exec_ctx, hr->parent_request); |
| 139 | gpr_free(hr->host); |
| 140 | gpr_free(hr); |
| 141 | } |
| 142 | |
| 143 | static void on_hostbyname_done_cb(void *arg, int status, int timeouts, |
| 144 | struct hostent *hostent) { |
| 145 | grpc_ares_hostbyname_request *hr = (grpc_ares_hostbyname_request *)arg; |
| 146 | grpc_ares_request *r = hr->parent_request; |
Yuchen Zeng | 2648a95 | 2016-11-02 18:52:25 -0700 | [diff] [blame] | 147 | gpr_mu_lock(&r->mu); |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 148 | if (status == ARES_SUCCESS) { |
Yuchen Zeng | 1d0c1c4 | 2016-08-18 15:36:24 -0700 | [diff] [blame] | 149 | GRPC_ERROR_UNREF(r->error); |
| 150 | r->error = GRPC_ERROR_NONE; |
Yuchen Zeng | 912327e | 2016-10-25 18:27:17 -0700 | [diff] [blame] | 151 | r->success = true; |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 152 | grpc_lb_addresses **lb_addresses = r->lb_addrs_out; |
| 153 | if (*lb_addresses == NULL) { |
| 154 | *lb_addresses = grpc_lb_addresses_create(0, NULL); |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 155 | } |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 156 | size_t prev_naddr = (*lb_addresses)->num_addresses; |
Yuchen Zeng | 912327e | 2016-10-25 18:27:17 -0700 | [diff] [blame] | 157 | size_t i; |
Yuchen Zeng | 9248d35 | 2016-08-16 16:33:10 -0700 | [diff] [blame] | 158 | for (i = 0; hostent->h_addr_list[i] != NULL; i++) { |
| 159 | } |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 160 | (*lb_addresses)->num_addresses += i; |
Yash Tibrewal | ca3c1c0 | 2017-09-07 22:47:16 -0700 | [diff] [blame] | 161 | (*lb_addresses)->addresses = (grpc_lb_address *)gpr_realloc( |
| 162 | (*lb_addresses)->addresses, |
| 163 | sizeof(grpc_lb_address) * (*lb_addresses)->num_addresses); |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 164 | for (i = prev_naddr; i < (*lb_addresses)->num_addresses; i++) { |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 165 | switch (hostent->h_addrtype) { |
| 166 | case AF_INET6: { |
| 167 | size_t addr_len = sizeof(struct sockaddr_in6); |
| 168 | struct sockaddr_in6 addr; |
Yuchen Zeng | 3b4bed2 | 2017-06-04 01:41:15 -0700 | [diff] [blame] | 169 | memset(&addr, 0, addr_len); |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 170 | memcpy(&addr.sin6_addr, hostent->h_addr_list[i - prev_naddr], |
| 171 | sizeof(struct in6_addr)); |
| 172 | addr.sin6_family = (sa_family_t)hostent->h_addrtype; |
| 173 | addr.sin6_port = hr->port; |
| 174 | grpc_lb_addresses_set_address( |
| 175 | *lb_addresses, i, &addr, addr_len, |
| 176 | hr->is_balancer /* is_balancer */, |
Alexander Polcyn | 3a24364 | 2017-09-12 12:06:27 -0700 | [diff] [blame] | 177 | hr->is_balancer ? hr->host : NULL /* balancer_name */, |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 178 | NULL /* user_data */); |
| 179 | char output[INET6_ADDRSTRLEN]; |
| 180 | ares_inet_ntop(AF_INET6, &addr.sin6_addr, output, INET6_ADDRSTRLEN); |
| 181 | gpr_log(GPR_DEBUG, |
| 182 | "c-ares resolver gets a AF_INET6 result: \n" |
| 183 | " addr: %s\n port: %d\n sin6_scope_id: %d\n", |
| 184 | output, ntohs(hr->port), addr.sin6_scope_id); |
| 185 | break; |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 186 | } |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 187 | case AF_INET: { |
| 188 | size_t addr_len = sizeof(struct sockaddr_in); |
| 189 | struct sockaddr_in addr; |
Yuchen Zeng | 3b4bed2 | 2017-06-04 01:41:15 -0700 | [diff] [blame] | 190 | memset(&addr, 0, addr_len); |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 191 | memcpy(&addr.sin_addr, hostent->h_addr_list[i - prev_naddr], |
| 192 | sizeof(struct in_addr)); |
| 193 | addr.sin_family = (sa_family_t)hostent->h_addrtype; |
| 194 | addr.sin_port = hr->port; |
| 195 | grpc_lb_addresses_set_address( |
| 196 | *lb_addresses, i, &addr, addr_len, |
| 197 | hr->is_balancer /* is_balancer */, |
Alexander Polcyn | 3a24364 | 2017-09-12 12:06:27 -0700 | [diff] [blame] | 198 | hr->is_balancer ? hr->host : NULL /* balancer_name */, |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 199 | NULL /* user_data */); |
| 200 | char output[INET_ADDRSTRLEN]; |
| 201 | ares_inet_ntop(AF_INET, &addr.sin_addr, output, INET_ADDRSTRLEN); |
| 202 | gpr_log(GPR_DEBUG, |
| 203 | "c-ares resolver gets a AF_INET result: \n" |
| 204 | " addr: %s\n port: %d\n", |
| 205 | output, ntohs(hr->port)); |
| 206 | break; |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 207 | } |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 208 | } |
| 209 | } |
Yuchen Zeng | 15a71c8 | 2016-08-17 11:50:06 -0700 | [diff] [blame] | 210 | } else if (!r->success) { |
Yuchen Zeng | 912327e | 2016-10-25 18:27:17 -0700 | [diff] [blame] | 211 | char *error_msg; |
| 212 | gpr_asprintf(&error_msg, "C-ares status is not ARES_SUCCESS: %s", |
| 213 | ares_strerror(status)); |
Yuchen Zeng | 0e8bc45 | 2017-03-23 14:44:14 -0700 | [diff] [blame] | 214 | grpc_error *error = GRPC_ERROR_CREATE_FROM_COPIED_STRING(error_msg); |
Yuchen Zeng | 912327e | 2016-10-25 18:27:17 -0700 | [diff] [blame] | 215 | gpr_free(error_msg); |
Yuchen Zeng | 1d0c1c4 | 2016-08-18 15:36:24 -0700 | [diff] [blame] | 216 | if (r->error == GRPC_ERROR_NONE) { |
Yuchen Zeng | 912327e | 2016-10-25 18:27:17 -0700 | [diff] [blame] | 217 | r->error = error; |
| 218 | } else { |
| 219 | r->error = grpc_error_add_child(error, r->error); |
Yuchen Zeng | 15a71c8 | 2016-08-17 11:50:06 -0700 | [diff] [blame] | 220 | } |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 221 | } |
Yuchen Zeng | 2648a95 | 2016-11-02 18:52:25 -0700 | [diff] [blame] | 222 | gpr_mu_unlock(&r->mu); |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 223 | destroy_hostbyname_request(NULL, hr); |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 224 | } |
| 225 | |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 226 | static void on_srv_query_done_cb(void *arg, int status, int timeouts, |
| 227 | unsigned char *abuf, int alen) { |
| 228 | grpc_ares_request *r = (grpc_ares_request *)arg; |
| 229 | grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
| 230 | gpr_log(GPR_DEBUG, "on_query_srv_done_cb"); |
| 231 | if (status == ARES_SUCCESS) { |
| 232 | gpr_log(GPR_DEBUG, "on_query_srv_done_cb ARES_SUCCESS"); |
| 233 | struct ares_srv_reply *reply; |
| 234 | const int parse_status = ares_parse_srv_reply(abuf, alen, &reply); |
| 235 | if (parse_status == ARES_SUCCESS) { |
| 236 | ares_channel *channel = grpc_ares_ev_driver_get_channel(r->ev_driver); |
| 237 | for (struct ares_srv_reply *srv_it = reply; srv_it != NULL; |
| 238 | srv_it = srv_it->next) { |
| 239 | if (grpc_ipv6_loopback_available()) { |
| 240 | grpc_ares_hostbyname_request *hr = create_hostbyname_request( |
Alex Polcyn | ddf6c49 | 2017-06-27 22:06:35 +0000 | [diff] [blame] | 241 | r, srv_it->host, htons(srv_it->port), true /* is_balancer */); |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 242 | ares_gethostbyname(*channel, hr->host, AF_INET6, |
| 243 | on_hostbyname_done_cb, hr); |
| 244 | } |
| 245 | grpc_ares_hostbyname_request *hr = create_hostbyname_request( |
Alex Polcyn | ddf6c49 | 2017-06-27 22:06:35 +0000 | [diff] [blame] | 246 | r, srv_it->host, htons(srv_it->port), true /* is_balancer */); |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 247 | ares_gethostbyname(*channel, hr->host, AF_INET, on_hostbyname_done_cb, |
| 248 | hr); |
| 249 | grpc_ares_ev_driver_start(&exec_ctx, r->ev_driver); |
| 250 | } |
| 251 | } |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 252 | if (reply != NULL) { |
| 253 | ares_free_data(reply); |
| 254 | } |
| 255 | } else if (!r->success) { |
| 256 | char *error_msg; |
| 257 | gpr_asprintf(&error_msg, "C-ares status is not ARES_SUCCESS: %s", |
| 258 | ares_strerror(status)); |
| 259 | grpc_error *error = GRPC_ERROR_CREATE_FROM_COPIED_STRING(error_msg); |
| 260 | gpr_free(error_msg); |
| 261 | if (r->error == GRPC_ERROR_NONE) { |
| 262 | r->error = error; |
| 263 | } else { |
| 264 | r->error = grpc_error_add_child(error, r->error); |
| 265 | } |
| 266 | } |
| 267 | grpc_ares_request_unref(&exec_ctx, r); |
| 268 | grpc_exec_ctx_finish(&exec_ctx); |
| 269 | } |
| 270 | |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 271 | static const char g_service_config_attribute_prefix[] = "grpc_config="; |
| 272 | |
| 273 | static void on_txt_done_cb(void *arg, int status, int timeouts, |
| 274 | unsigned char *buf, int len) { |
| 275 | gpr_log(GPR_DEBUG, "on_txt_done_cb"); |
| 276 | char *error_msg; |
| 277 | grpc_ares_request *r = (grpc_ares_request *)arg; |
Yash Tibrewal | 533d118 | 2017-09-18 10:48:22 -0700 | [diff] [blame^] | 278 | const size_t prefix_len = sizeof(g_service_config_attribute_prefix) - 1; |
| 279 | struct ares_txt_ext *result; |
| 280 | struct ares_txt_ext *reply = NULL; |
| 281 | grpc_error *error = GRPC_ERROR_NONE; |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 282 | gpr_mu_lock(&r->mu); |
| 283 | if (status != ARES_SUCCESS) goto fail; |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 284 | status = ares_parse_txt_reply_ext(buf, len, &reply); |
| 285 | if (status != ARES_SUCCESS) goto fail; |
| 286 | // Find service config in TXT record. |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 287 | for (result = reply; result != NULL; result = result->next) { |
| 288 | if (result->record_start && |
| 289 | memcmp(result->txt, g_service_config_attribute_prefix, prefix_len) == |
| 290 | 0) { |
| 291 | break; |
| 292 | } |
| 293 | } |
| 294 | // Found a service config record. |
| 295 | if (result != NULL) { |
| 296 | size_t service_config_len = result->length - prefix_len; |
Yash Tibrewal | ca3c1c0 | 2017-09-07 22:47:16 -0700 | [diff] [blame] | 297 | *r->service_config_json_out = (char *)gpr_malloc(service_config_len + 1); |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 298 | memcpy(*r->service_config_json_out, result->txt + prefix_len, |
| 299 | service_config_len); |
| 300 | for (result = result->next; result != NULL && !result->record_start; |
| 301 | result = result->next) { |
Yash Tibrewal | ca3c1c0 | 2017-09-07 22:47:16 -0700 | [diff] [blame] | 302 | *r->service_config_json_out = (char *)gpr_realloc( |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 303 | *r->service_config_json_out, service_config_len + result->length + 1); |
| 304 | memcpy(*r->service_config_json_out + service_config_len, result->txt, |
| 305 | result->length); |
| 306 | service_config_len += result->length; |
| 307 | } |
| 308 | (*r->service_config_json_out)[service_config_len] = '\0'; |
| 309 | gpr_log(GPR_INFO, "found service config: %s", *r->service_config_json_out); |
| 310 | } |
| 311 | // Clean up. |
| 312 | ares_free_data(reply); |
| 313 | goto done; |
| 314 | fail: |
| 315 | gpr_asprintf(&error_msg, "C-ares TXT lookup status is not ARES_SUCCESS: %s", |
| 316 | ares_strerror(status)); |
Yash Tibrewal | 533d118 | 2017-09-18 10:48:22 -0700 | [diff] [blame^] | 317 | error = GRPC_ERROR_CREATE_FROM_COPIED_STRING(error_msg); |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 318 | gpr_free(error_msg); |
| 319 | if (r->error == GRPC_ERROR_NONE) { |
| 320 | r->error = error; |
| 321 | } else { |
| 322 | r->error = grpc_error_add_child(error, r->error); |
| 323 | } |
| 324 | done: |
| 325 | gpr_mu_unlock(&r->mu); |
| 326 | grpc_ares_request_unref(NULL, r); |
| 327 | } |
| 328 | |
Yuchen Zeng | 4ebace7 | 2017-06-05 17:24:06 -0700 | [diff] [blame] | 329 | static grpc_ares_request *grpc_dns_lookup_ares_impl( |
Yuchen Zeng | 3b4bed2 | 2017-06-04 01:41:15 -0700 | [diff] [blame] | 330 | grpc_exec_ctx *exec_ctx, const char *dns_server, const char *name, |
| 331 | const char *default_port, grpc_pollset_set *interested_parties, |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 332 | grpc_closure *on_done, grpc_lb_addresses **addrs, bool check_grpclb, |
| 333 | char **service_config_json) { |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 334 | grpc_error *error = GRPC_ERROR_NONE; |
Yash Tibrewal | 533d118 | 2017-09-18 10:48:22 -0700 | [diff] [blame^] | 335 | grpc_ares_hostbyname_request *hr = NULL; |
| 336 | grpc_ares_request *r = NULL; |
| 337 | ares_channel *channel = NULL; |
Yuchen Zeng | e5230b7 | 2017-03-17 15:10:49 -0700 | [diff] [blame] | 338 | /* TODO(zyc): Enable tracing after #9603 is checked in */ |
| 339 | /* if (grpc_dns_trace) { |
| 340 | gpr_log(GPR_DEBUG, "resolve_address (blocking): name=%s, default_port=%s", |
| 341 | name, default_port); |
| 342 | } */ |
| 343 | |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 344 | /* parse name, splitting it into host and port parts */ |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 345 | char *host; |
| 346 | char *port; |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 347 | gpr_split_host_port(name, &host, &port); |
| 348 | if (host == NULL) { |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 349 | error = grpc_error_set_str( |
Yuchen Zeng | 0e8bc45 | 2017-03-23 14:44:14 -0700 | [diff] [blame] | 350 | GRPC_ERROR_CREATE_FROM_STATIC_STRING("unparseable host:port"), |
| 351 | GRPC_ERROR_STR_TARGET_ADDRESS, grpc_slice_from_copied_string(name)); |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 352 | goto error_cleanup; |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 353 | } else if (port == NULL) { |
| 354 | if (default_port == NULL) { |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 355 | error = grpc_error_set_str( |
Yuchen Zeng | 0e8bc45 | 2017-03-23 14:44:14 -0700 | [diff] [blame] | 356 | GRPC_ERROR_CREATE_FROM_STATIC_STRING("no port in name"), |
| 357 | GRPC_ERROR_STR_TARGET_ADDRESS, grpc_slice_from_copied_string(name)); |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 358 | goto error_cleanup; |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 359 | } |
| 360 | port = gpr_strdup(default_port); |
Yuchen Zeng | db53cc0 | 2016-08-10 05:04:53 -0700 | [diff] [blame] | 361 | } |
| 362 | |
Yuchen Zeng | 117a300 | 2016-11-17 19:43:36 -0800 | [diff] [blame] | 363 | grpc_ares_ev_driver *ev_driver; |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 364 | error = grpc_ares_ev_driver_create(&ev_driver, interested_parties); |
| 365 | if (error != GRPC_ERROR_NONE) goto error_cleanup; |
Yuchen Zeng | 117a300 | 2016-11-17 19:43:36 -0800 | [diff] [blame] | 366 | |
Yash Tibrewal | 533d118 | 2017-09-18 10:48:22 -0700 | [diff] [blame^] | 367 | r = (grpc_ares_request *)gpr_zalloc(sizeof(grpc_ares_request)); |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 368 | gpr_mu_init(&r->mu); |
| 369 | r->ev_driver = ev_driver; |
| 370 | r->on_done = on_done; |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 371 | r->lb_addrs_out = addrs; |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 372 | r->service_config_json_out = service_config_json; |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 373 | r->success = false; |
| 374 | r->error = GRPC_ERROR_NONE; |
Yash Tibrewal | 533d118 | 2017-09-18 10:48:22 -0700 | [diff] [blame^] | 375 | channel = grpc_ares_ev_driver_get_channel(r->ev_driver); |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 376 | |
| 377 | // If dns_server is specified, use it. |
| 378 | if (dns_server != NULL) { |
| 379 | gpr_log(GPR_INFO, "Using DNS server %s", dns_server); |
| 380 | grpc_resolved_address addr; |
| 381 | if (grpc_parse_ipv4_hostport(dns_server, &addr, false /* log_errors */)) { |
| 382 | r->dns_server_addr.family = AF_INET; |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 383 | struct sockaddr_in *in = (struct sockaddr_in *)addr.addr; |
| 384 | memcpy(&r->dns_server_addr.addr.addr4, &in->sin_addr, |
| 385 | sizeof(struct in_addr)); |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 386 | r->dns_server_addr.tcp_port = grpc_sockaddr_get_port(&addr); |
| 387 | r->dns_server_addr.udp_port = grpc_sockaddr_get_port(&addr); |
| 388 | } else if (grpc_parse_ipv6_hostport(dns_server, &addr, |
| 389 | false /* log_errors */)) { |
| 390 | r->dns_server_addr.family = AF_INET6; |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 391 | struct sockaddr_in6 *in6 = (struct sockaddr_in6 *)addr.addr; |
| 392 | memcpy(&r->dns_server_addr.addr.addr6, &in6->sin6_addr, |
| 393 | sizeof(struct in6_addr)); |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 394 | r->dns_server_addr.tcp_port = grpc_sockaddr_get_port(&addr); |
| 395 | r->dns_server_addr.udp_port = grpc_sockaddr_get_port(&addr); |
| 396 | } else { |
| 397 | error = grpc_error_set_str( |
| 398 | GRPC_ERROR_CREATE_FROM_STATIC_STRING("cannot parse authority"), |
| 399 | GRPC_ERROR_STR_TARGET_ADDRESS, grpc_slice_from_copied_string(name)); |
Yuchen Zeng | 42e1f69 | 2017-05-21 15:34:37 -0700 | [diff] [blame] | 400 | gpr_free(r); |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 401 | goto error_cleanup; |
| 402 | } |
| 403 | int status = ares_set_servers_ports(*channel, &r->dns_server_addr); |
| 404 | if (status != ARES_SUCCESS) { |
| 405 | char *error_msg; |
| 406 | gpr_asprintf(&error_msg, "C-ares status is not ARES_SUCCESS: %s", |
| 407 | ares_strerror(status)); |
| 408 | error = GRPC_ERROR_CREATE_FROM_COPIED_STRING(error_msg); |
| 409 | gpr_free(error_msg); |
Yuchen Zeng | 42e1f69 | 2017-05-21 15:34:37 -0700 | [diff] [blame] | 410 | gpr_free(r); |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 411 | goto error_cleanup; |
| 412 | } |
| 413 | } |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 414 | gpr_ref_init(&r->pending_queries, 1); |
Yuchen Zeng | dc6b569 | 2017-02-13 22:44:01 -0800 | [diff] [blame] | 415 | if (grpc_ipv6_loopback_available()) { |
Yash Tibrewal | 533d118 | 2017-09-18 10:48:22 -0700 | [diff] [blame^] | 416 | hr = create_hostbyname_request(r, host, strhtons(port), |
| 417 | false /* is_balancer */); |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 418 | ares_gethostbyname(*channel, hr->host, AF_INET6, on_hostbyname_done_cb, hr); |
Yuchen Zeng | dc6b569 | 2017-02-13 22:44:01 -0800 | [diff] [blame] | 419 | } |
Yash Tibrewal | 533d118 | 2017-09-18 10:48:22 -0700 | [diff] [blame^] | 420 | hr = create_hostbyname_request(r, host, strhtons(port), |
| 421 | false /* is_balancer */); |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 422 | ares_gethostbyname(*channel, hr->host, AF_INET, on_hostbyname_done_cb, hr); |
Yuchen Zeng | 94c5776 | 2017-05-19 16:30:08 -0700 | [diff] [blame] | 423 | if (check_grpclb) { |
Yuchen Zeng | f3bdbb7 | 2017-05-18 18:22:23 -0700 | [diff] [blame] | 424 | /* Query the SRV record */ |
| 425 | grpc_ares_request_ref(r); |
| 426 | char *service_name; |
| 427 | gpr_asprintf(&service_name, "_grpclb._tcp.%s", host); |
| 428 | ares_query(*channel, service_name, ns_c_in, ns_t_srv, on_srv_query_done_cb, |
| 429 | r); |
| 430 | gpr_free(service_name); |
| 431 | } |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 432 | if (service_config_json != NULL) { |
| 433 | grpc_ares_request_ref(r); |
| 434 | ares_search(*channel, hr->host, ns_c_in, ns_t_txt, on_txt_done_cb, r); |
| 435 | } |
Yuchen Zeng | e5230b7 | 2017-03-17 15:10:49 -0700 | [diff] [blame] | 436 | /* TODO(zyc): Handle CNAME records here. */ |
Yuchen Zeng | dc6b569 | 2017-02-13 22:44:01 -0800 | [diff] [blame] | 437 | grpc_ares_ev_driver_start(exec_ctx, r->ev_driver); |
| 438 | grpc_ares_request_unref(exec_ctx, r); |
Yuchen Zeng | 42e1f69 | 2017-05-21 15:34:37 -0700 | [diff] [blame] | 439 | gpr_free(host); |
| 440 | gpr_free(port); |
Yuchen Zeng | 3b4bed2 | 2017-06-04 01:41:15 -0700 | [diff] [blame] | 441 | return r; |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 442 | |
Yuchen Zeng | 3ae2663 | 2016-11-09 15:53:50 -0800 | [diff] [blame] | 443 | error_cleanup: |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 444 | GRPC_CLOSURE_SCHED(exec_ctx, on_done, error); |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 445 | gpr_free(host); |
| 446 | gpr_free(port); |
Yuchen Zeng | 3b4bed2 | 2017-06-04 01:41:15 -0700 | [diff] [blame] | 447 | return NULL; |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 448 | } |
| 449 | |
Yuchen Zeng | 3b4bed2 | 2017-06-04 01:41:15 -0700 | [diff] [blame] | 450 | grpc_ares_request *(*grpc_dns_lookup_ares)( |
| 451 | grpc_exec_ctx *exec_ctx, const char *dns_server, const char *name, |
| 452 | const char *default_port, grpc_pollset_set *interested_parties, |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 453 | grpc_closure *on_done, grpc_lb_addresses **addrs, bool check_grpclb, |
| 454 | char **service_config_json) = grpc_dns_lookup_ares_impl; |
Mark D. Roth | 96fc54c | 2017-05-18 14:33:07 -0700 | [diff] [blame] | 455 | |
Yuchen Zeng | 3b4bed2 | 2017-06-04 01:41:15 -0700 | [diff] [blame] | 456 | void grpc_cancel_ares_request(grpc_exec_ctx *exec_ctx, grpc_ares_request *r) { |
| 457 | if (grpc_dns_lookup_ares == grpc_dns_lookup_ares_impl) { |
| 458 | grpc_ares_ev_driver_shutdown(exec_ctx, r->ev_driver); |
| 459 | } |
| 460 | } |
Yuchen Zeng | 8917aec | 2016-08-09 18:41:31 -0700 | [diff] [blame] | 461 | |
Yuchen Zeng | 85750b0 | 2016-08-08 14:16:34 -0700 | [diff] [blame] | 462 | grpc_error *grpc_ares_init(void) { |
Yuchen Zeng | 459480b | 2016-08-09 11:40:26 -0700 | [diff] [blame] | 463 | gpr_once_init(&g_basic_init, do_basic_init); |
| 464 | gpr_mu_lock(&g_init_mu); |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 465 | int status = ares_library_init(ARES_LIB_INIT_ALL); |
Yuchen Zeng | 459480b | 2016-08-09 11:40:26 -0700 | [diff] [blame] | 466 | gpr_mu_unlock(&g_init_mu); |
| 467 | |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 468 | if (status != ARES_SUCCESS) { |
Yuchen Zeng | 912327e | 2016-10-25 18:27:17 -0700 | [diff] [blame] | 469 | char *error_msg; |
| 470 | gpr_asprintf(&error_msg, "ares_library_init failed: %s", |
| 471 | ares_strerror(status)); |
Yuchen Zeng | 0e8bc45 | 2017-03-23 14:44:14 -0700 | [diff] [blame] | 472 | grpc_error *error = GRPC_ERROR_CREATE_FROM_COPIED_STRING(error_msg); |
Yuchen Zeng | 912327e | 2016-10-25 18:27:17 -0700 | [diff] [blame] | 473 | gpr_free(error_msg); |
| 474 | return error; |
Yuchen Zeng | d4bbfc7 | 2016-08-05 10:33:16 -0700 | [diff] [blame] | 475 | } |
Yuchen Zeng | 85750b0 | 2016-08-08 14:16:34 -0700 | [diff] [blame] | 476 | return GRPC_ERROR_NONE; |
| 477 | } |
| 478 | |
Yuchen Zeng | 459480b | 2016-08-09 11:40:26 -0700 | [diff] [blame] | 479 | void grpc_ares_cleanup(void) { |
| 480 | gpr_mu_lock(&g_init_mu); |
| 481 | ares_library_cleanup(); |
| 482 | gpr_mu_unlock(&g_init_mu); |
| 483 | } |
Yuchen Zeng | 299dd8d | 2016-08-16 21:40:13 -0700 | [diff] [blame] | 484 | |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 485 | /* |
| 486 | * grpc_resolve_address_ares related structs and functions |
| 487 | */ |
| 488 | |
| 489 | typedef struct grpc_resolve_address_ares_request { |
| 490 | /** the pointer to receive the resolved addresses */ |
| 491 | grpc_resolved_addresses **addrs_out; |
| 492 | /** currently resolving lb addresses */ |
| 493 | grpc_lb_addresses *lb_addrs; |
| 494 | /** closure to call when the resolve_address_ares request completes */ |
| 495 | grpc_closure *on_resolve_address_done; |
| 496 | /** a closure wrapping on_dns_lookup_done_cb, which should be invoked when the |
| 497 | grpc_dns_lookup_ares operation is done. */ |
| 498 | grpc_closure on_dns_lookup_done; |
| 499 | } grpc_resolve_address_ares_request; |
| 500 | |
| 501 | static void on_dns_lookup_done_cb(grpc_exec_ctx *exec_ctx, void *arg, |
| 502 | grpc_error *error) { |
| 503 | grpc_resolve_address_ares_request *r = |
| 504 | (grpc_resolve_address_ares_request *)arg; |
| 505 | grpc_resolved_addresses **resolved_addresses = r->addrs_out; |
| 506 | if (r->lb_addrs == NULL || r->lb_addrs->num_addresses == 0) { |
| 507 | *resolved_addresses = NULL; |
| 508 | } else { |
Yash Tibrewal | ca3c1c0 | 2017-09-07 22:47:16 -0700 | [diff] [blame] | 509 | *resolved_addresses = |
| 510 | (grpc_resolved_addresses *)gpr_zalloc(sizeof(grpc_resolved_addresses)); |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 511 | (*resolved_addresses)->naddrs = r->lb_addrs->num_addresses; |
Yash Tibrewal | ca3c1c0 | 2017-09-07 22:47:16 -0700 | [diff] [blame] | 512 | (*resolved_addresses)->addrs = (grpc_resolved_address *)gpr_zalloc( |
| 513 | sizeof(grpc_resolved_address) * (*resolved_addresses)->naddrs); |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 514 | for (size_t i = 0; i < (*resolved_addresses)->naddrs; i++) { |
| 515 | GPR_ASSERT(!r->lb_addrs->addresses[i].is_balancer); |
| 516 | memcpy(&(*resolved_addresses)->addrs[i], |
| 517 | &r->lb_addrs->addresses[i].address, sizeof(grpc_resolved_address)); |
| 518 | } |
| 519 | } |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 520 | GRPC_CLOSURE_SCHED(exec_ctx, r->on_resolve_address_done, |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 521 | GRPC_ERROR_REF(error)); |
Yuchen Zeng | 4c15fac | 2017-05-24 16:40:20 -0700 | [diff] [blame] | 522 | grpc_lb_addresses_destroy(exec_ctx, r->lb_addrs); |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 523 | gpr_free(r); |
| 524 | } |
| 525 | |
Yuchen Zeng | 4ebace7 | 2017-06-05 17:24:06 -0700 | [diff] [blame] | 526 | static void grpc_resolve_address_ares_impl(grpc_exec_ctx *exec_ctx, |
| 527 | const char *name, |
| 528 | const char *default_port, |
| 529 | grpc_pollset_set *interested_parties, |
| 530 | grpc_closure *on_done, |
| 531 | grpc_resolved_addresses **addrs) { |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 532 | grpc_resolve_address_ares_request *r = |
Yash Tibrewal | bc130da | 2017-09-12 22:44:08 -0700 | [diff] [blame] | 533 | (grpc_resolve_address_ares_request *)gpr_zalloc( |
| 534 | sizeof(grpc_resolve_address_ares_request)); |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 535 | r->addrs_out = addrs; |
| 536 | r->on_resolve_address_done = on_done; |
ncteisen | 274bbbe | 2017-06-08 14:57:11 -0700 | [diff] [blame] | 537 | GRPC_CLOSURE_INIT(&r->on_dns_lookup_done, on_dns_lookup_done_cb, r, |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 538 | grpc_schedule_on_exec_ctx); |
| 539 | grpc_dns_lookup_ares(exec_ctx, NULL /* dns_server */, name, default_port, |
| 540 | interested_parties, &r->on_dns_lookup_done, &r->lb_addrs, |
Mark D. Roth | f9bf428 | 2017-08-03 14:47:23 -0700 | [diff] [blame] | 541 | false /* check_grpclb */, |
| 542 | NULL /* service_config_json */); |
Yuchen Zeng | 3385b4f | 2017-05-24 11:40:20 -0700 | [diff] [blame] | 543 | } |
| 544 | |
| 545 | void (*grpc_resolve_address_ares)( |
| 546 | grpc_exec_ctx *exec_ctx, const char *name, const char *default_port, |
| 547 | grpc_pollset_set *interested_parties, grpc_closure *on_done, |
| 548 | grpc_resolved_addresses **addrs) = grpc_resolve_address_ares_impl; |
| 549 | |
Yuchen Zeng | d6b4985 | 2017-03-20 01:44:27 -0700 | [diff] [blame] | 550 | #endif /* GRPC_ARES == 1 && !defined(GRPC_UV) */ |