David Garcia Quintas
7 years ago
parent
e7cd8355ea
commit
956fe5e089
8 changed files with 412 additions and 1 deletions
@ -0,0 +1,284 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015 gRPC authors. |
||||
* |
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
* you may not use this file except in compliance with the License. |
||||
* You may obtain a copy of the License at |
||||
* |
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
* |
||||
* Unless required by applicable law or agreed to in writing, software |
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
* See the License for the specific language governing permissions and |
||||
* limitations under the License. |
||||
* |
||||
*/ |
||||
|
||||
#include <cstring> |
||||
|
||||
#include <grpc/support/log.h> |
||||
|
||||
#include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.h" |
||||
#include "src/core/ext/filters/client_channel/resolver_registry.h" |
||||
#include "src/core/lib/channel/channel_args.h" |
||||
#include "src/core/lib/iomgr/combiner.h" |
||||
#include "src/core/lib/iomgr/sockaddr_utils.h" |
||||
#include "test/core/util/test_config.h" |
||||
|
||||
static grpc_combiner* g_combiner; |
||||
|
||||
static void (*g_default_grpc_resolve_address)( |
||||
const char* name, const char* default_port, |
||||
grpc_pollset_set* interested_parties, grpc_closure* on_done, |
||||
grpc_resolved_addresses** addrs); |
||||
|
||||
grpc_ares_request* (*g_default_dns_lookup_ares)( |
||||
const char* dns_server, const char* name, const char* default_port, |
||||
grpc_pollset_set* interested_parties, grpc_closure* on_done, |
||||
grpc_lb_addresses** addrs, bool check_grpclb, char** service_config_json); |
||||
|
||||
// Counter incremented by test_resolve_address_impl indicating the number of
|
||||
// times a system-level resolution has happened.
|
||||
static int g_resolution_count; |
||||
|
||||
typedef struct args_struct { |
||||
gpr_event ev; |
||||
grpc_resolved_addresses* addrs; |
||||
gpr_atm done_atm; |
||||
gpr_mu* mu; |
||||
grpc_pollset* pollset; |
||||
grpc_pollset_set* pollset_set; |
||||
} args_struct; |
||||
|
||||
args_struct g_iomgr_args; |
||||
|
||||
// Wrapper around g_default_grpc_resolve_address in order to count the number of
|
||||
// times we incur in a system-level name resolution.
|
||||
static void test_resolve_address_impl(const char* name, |
||||
const char* default_port, |
||||
grpc_pollset_set* interested_parties, |
||||
grpc_closure* on_done, |
||||
grpc_resolved_addresses** addrs) { |
||||
g_default_grpc_resolve_address(name, default_port, g_iomgr_args.pollset_set, |
||||
on_done, addrs); |
||||
++g_resolution_count; |
||||
} |
||||
|
||||
grpc_ares_request* test_dns_lookup_ares( |
||||
const char* dns_server, const char* name, const char* default_port, |
||||
grpc_pollset_set* interested_parties, grpc_closure* on_done, |
||||
grpc_lb_addresses** addrs, bool check_grpclb, char** service_config_json) { |
||||
grpc_ares_request* result = g_default_dns_lookup_ares( |
||||
dns_server, name, default_port, g_iomgr_args.pollset_set, on_done, addrs, |
||||
check_grpclb, service_config_json); |
||||
++g_resolution_count; |
||||
return result; |
||||
} |
||||
|
||||
static gpr_timespec test_deadline(void) { |
||||
return grpc_timeout_seconds_to_deadline(100); |
||||
} |
||||
|
||||
static void do_nothing(void* arg, grpc_error* error) {} |
||||
|
||||
void args_init(args_struct* args) { |
||||
gpr_event_init(&args->ev); |
||||
args->pollset = static_cast<grpc_pollset*>(gpr_zalloc(grpc_pollset_size())); |
||||
grpc_pollset_init(args->pollset, &args->mu); |
||||
args->pollset_set = grpc_pollset_set_create(); |
||||
grpc_pollset_set_add_pollset(args->pollset_set, args->pollset); |
||||
args->addrs = nullptr; |
||||
gpr_atm_rel_store(&args->done_atm, 0); |
||||
} |
||||
|
||||
void args_finish(args_struct* args) { |
||||
GPR_ASSERT(gpr_event_wait(&args->ev, test_deadline())); |
||||
grpc_resolved_addresses_destroy(args->addrs); |
||||
grpc_pollset_set_del_pollset(args->pollset_set, args->pollset); |
||||
grpc_pollset_set_destroy(args->pollset_set); |
||||
grpc_closure do_nothing_cb; |
||||
GRPC_CLOSURE_INIT(&do_nothing_cb, do_nothing, nullptr, |
||||
grpc_schedule_on_exec_ctx); |
||||
gpr_mu_lock(args->mu); |
||||
grpc_pollset_shutdown(args->pollset, &do_nothing_cb); |
||||
gpr_mu_unlock(args->mu); |
||||
// exec_ctx needs to be flushed before calling grpc_pollset_destroy()
|
||||
grpc_core::ExecCtx::Get()->Flush(); |
||||
grpc_pollset_destroy(args->pollset); |
||||
gpr_free(args->pollset); |
||||
} |
||||
|
||||
static grpc_millis n_sec_deadline(int seconds) { |
||||
return grpc_timespec_to_millis_round_up( |
||||
grpc_timeout_seconds_to_deadline(seconds)); |
||||
} |
||||
|
||||
static void poll_pollset_until_request_done(args_struct* args) { |
||||
grpc_core::ExecCtx exec_ctx; |
||||
grpc_millis deadline = n_sec_deadline(10); |
||||
while (true) { |
||||
bool done = gpr_atm_acq_load(&args->done_atm) != 0; |
||||
if (done) { |
||||
break; |
||||
} |
||||
grpc_millis time_left = deadline - grpc_core::ExecCtx::Get()->Now(); |
||||
gpr_log(GPR_DEBUG, "done=%d, time_left=%" PRIdPTR, done, time_left); |
||||
GPR_ASSERT(time_left >= 0); |
||||
grpc_pollset_worker* worker = nullptr; |
||||
gpr_mu_lock(args->mu); |
||||
GRPC_LOG_IF_ERROR("pollset_work", grpc_pollset_work(args->pollset, &worker, |
||||
n_sec_deadline(1))); |
||||
gpr_mu_unlock(args->mu); |
||||
grpc_core::ExecCtx::Get()->Flush(); |
||||
} |
||||
gpr_event_set(&args->ev, (void*)1); |
||||
} |
||||
|
||||
typedef struct on_resolution_cb_arg { |
||||
grpc_resolver* resolver; |
||||
grpc_channel_args* result; |
||||
grpc_millis delay_before_second_resolution; |
||||
} on_resolution_cb_arg; |
||||
|
||||
// Counter for the number of times a resolution notification callback has been
|
||||
// invoked.
|
||||
static int g_on_resolution_invocations_count; |
||||
|
||||
// Set to true by the last callback in the resolution chain.
|
||||
bool g_all_callbacks_invoked; |
||||
|
||||
void on_third_resolution(void* arg, grpc_error* error) { |
||||
on_resolution_cb_arg* cb_arg = static_cast<on_resolution_cb_arg*>(arg); |
||||
++g_on_resolution_invocations_count; |
||||
grpc_channel_args_destroy(cb_arg->result); |
||||
|
||||
gpr_log(GPR_INFO, |
||||
"3rd: g_on_resolution_invocations_count: %d, g_resolution_count: %d", |
||||
g_on_resolution_invocations_count, g_resolution_count); |
||||
// In this case we expect to have incurred in another system-level resolution
|
||||
// because on_second_resolution slept for longer than the min resolution
|
||||
// period.
|
||||
GPR_ASSERT(g_on_resolution_invocations_count == 3); |
||||
GPR_ASSERT(g_resolution_count == 2); |
||||
|
||||
grpc_resolver_shutdown_locked(cb_arg->resolver); |
||||
GRPC_RESOLVER_UNREF(cb_arg->resolver, "on_third_resolution"); |
||||
gpr_free(cb_arg); |
||||
g_all_callbacks_invoked = true; |
||||
} |
||||
|
||||
void on_second_resolution(void* arg, grpc_error* error) { |
||||
on_resolution_cb_arg* cb_arg = static_cast<on_resolution_cb_arg*>(arg); |
||||
++g_on_resolution_invocations_count; |
||||
grpc_channel_args_destroy(cb_arg->result); |
||||
|
||||
gpr_log(GPR_INFO, |
||||
"2nd: g_on_resolution_invocations_count: %d, g_resolution_count: %d", |
||||
g_on_resolution_invocations_count, g_resolution_count); |
||||
// The resolution request for which this function is the callback happened
|
||||
// before the min resolution period. Therefore, no new system-level
|
||||
// resolutions happened, as indicated by g_resolution_count.
|
||||
GPR_ASSERT(g_on_resolution_invocations_count == 2); |
||||
GPR_ASSERT(g_resolution_count == 1); |
||||
|
||||
grpc_core::ExecCtx::Get()->TestOnlySetNow( |
||||
cb_arg->delay_before_second_resolution * 2); |
||||
grpc_resolver_next_locked( |
||||
cb_arg->resolver, &cb_arg->result, |
||||
GRPC_CLOSURE_CREATE(on_third_resolution, arg, grpc_schedule_on_exec_ctx)); |
||||
grpc_resolver_channel_saw_error_locked(cb_arg->resolver); |
||||
} |
||||
|
||||
void on_first_resolution(void* arg, grpc_error* error) { |
||||
on_resolution_cb_arg* cb_arg = static_cast<on_resolution_cb_arg*>(arg); |
||||
++g_on_resolution_invocations_count; |
||||
grpc_channel_args_destroy(cb_arg->result); |
||||
grpc_resolver_next_locked(cb_arg->resolver, &cb_arg->result, |
||||
GRPC_CLOSURE_CREATE(on_second_resolution, arg, |
||||
grpc_schedule_on_exec_ctx)); |
||||
grpc_resolver_channel_saw_error_locked(cb_arg->resolver); |
||||
gpr_log(GPR_INFO, |
||||
"1st: g_on_resolution_invocations_count: %d, g_resolution_count: %d", |
||||
g_on_resolution_invocations_count, g_resolution_count); |
||||
// Theres one initial system-level resolution and one invocation of a
|
||||
// notification callback (the current function).
|
||||
GPR_ASSERT(g_on_resolution_invocations_count == 1); |
||||
GPR_ASSERT(g_resolution_count == 1); |
||||
} |
||||
|
||||
static void test_cooldown(const char* uri_str, bool use_cares) { |
||||
grpc_core::ExecCtx exec_ctx; |
||||
grpc_resolver_factory* factory = grpc_resolver_factory_lookup("dns"); |
||||
grpc_uri* uri = grpc_uri_parse(uri_str, 0); |
||||
grpc_resolver_args args; |
||||
|
||||
if (use_cares) args_init(&g_iomgr_args); |
||||
|
||||
grpc_resolver* resolver; |
||||
gpr_log(GPR_DEBUG, "test: '%s' should be valid for '%s'", uri_str, |
||||
factory->vtable->scheme); |
||||
GPR_ASSERT(uri); |
||||
memset(&args, 0, sizeof(args)); |
||||
args.uri = uri; |
||||
args.combiner = g_combiner; |
||||
g_on_resolution_invocations_count = 0; |
||||
g_resolution_count = 0; |
||||
constexpr int kMinResolutionPeriodMs = 1000; |
||||
|
||||
grpc_arg cooldown_arg; |
||||
cooldown_arg.key = |
||||
const_cast<char*>(GRPC_ARG_DNS_MIN_TIME_BETWEEN_RESOLUTIONS_MS); |
||||
cooldown_arg.type = GRPC_ARG_INTEGER; |
||||
cooldown_arg.value.integer = kMinResolutionPeriodMs; |
||||
auto* cooldown_channel_args = |
||||
grpc_channel_args_copy_and_add(nullptr, &cooldown_arg, 1); |
||||
args.args = cooldown_channel_args; |
||||
|
||||
resolver = grpc_resolver_factory_create_resolver(factory, &args); |
||||
grpc_channel_args_destroy(cooldown_channel_args); |
||||
GPR_ASSERT(resolver != nullptr); |
||||
|
||||
on_resolution_cb_arg* arg = (on_resolution_cb_arg*)gpr_zalloc(sizeof(*arg)); |
||||
arg->resolver = resolver; |
||||
arg->delay_before_second_resolution = kMinResolutionPeriodMs; |
||||
// First resolution, would incur in system-level resolution.
|
||||
grpc_resolver_next_locked( |
||||
resolver, &arg->result, |
||||
GRPC_CLOSURE_CREATE(on_first_resolution, arg, grpc_schedule_on_exec_ctx)); |
||||
grpc_uri_destroy(uri); |
||||
|
||||
if (use_cares) { |
||||
grpc_core::ExecCtx::Get()->Flush(); |
||||
poll_pollset_until_request_done(&g_iomgr_args); |
||||
args_finish(&g_iomgr_args); |
||||
} |
||||
|
||||
grpc_resolver_factory_unref(factory); |
||||
} |
||||
|
||||
int main(int argc, char** argv) { |
||||
grpc_test_init(argc, argv); |
||||
grpc_init(); |
||||
|
||||
g_combiner = grpc_combiner_create(); |
||||
|
||||
const bool use_cares = (grpc_resolve_address == grpc_resolve_address_ares); |
||||
|
||||
g_default_grpc_resolve_address = grpc_resolve_address; |
||||
g_default_dns_lookup_ares = grpc_dns_lookup_ares; |
||||
grpc_dns_lookup_ares = test_dns_lookup_ares; |
||||
grpc_resolve_address = test_resolve_address_impl; |
||||
|
||||
test_cooldown("dns:127.0.0.1", use_cares); |
||||
|
||||
{ |
||||
grpc_core::ExecCtx exec_ctx; |
||||
GRPC_COMBINER_UNREF(g_combiner, "test"); |
||||
} |
||||
grpc_shutdown(); |
||||
GPR_ASSERT(g_all_callbacks_invoked); |
||||
|
||||
return 0; |
||||
} |
||||
Loading…
Reference in new issue