|
|
|
@ -38,63 +38,103 @@ |
|
|
|
#include "src/core/lib/iomgr/tcp_client.h" |
|
|
|
#include "src/core/lib/iomgr/tcp_client.h" |
|
|
|
#include "src/core/lib/slice/slice_internal.h" |
|
|
|
#include "src/core/lib/slice/slice_internal.h" |
|
|
|
|
|
|
|
|
|
|
|
typedef struct { |
|
|
|
namespace grpc_core { |
|
|
|
grpc_connector base; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
gpr_mu mu; |
|
|
|
Chttp2Connector::Chttp2Connector() { |
|
|
|
gpr_refcount refs; |
|
|
|
GRPC_CLOSURE_INIT(&connected_, Connected, this, grpc_schedule_on_exec_ctx); |
|
|
|
|
|
|
|
} |
|
|
|
bool shutdown; |
|
|
|
|
|
|
|
bool connecting; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
grpc_closure* notify; |
|
|
|
|
|
|
|
grpc_connect_in_args args; |
|
|
|
|
|
|
|
grpc_connect_out_args* result; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
grpc_endpoint* endpoint; // Non-NULL until handshaking starts.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
grpc_closure connected; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
grpc_core::RefCountedPtr<grpc_core::HandshakeManager> handshake_mgr; |
|
|
|
|
|
|
|
} chttp2_connector; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static void chttp2_connector_ref(grpc_connector* con) { |
|
|
|
Chttp2Connector::~Chttp2Connector() { |
|
|
|
chttp2_connector* c = reinterpret_cast<chttp2_connector*>(con); |
|
|
|
if (endpoint_ != nullptr) grpc_endpoint_destroy(endpoint_); |
|
|
|
gpr_ref(&c->refs); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
static void chttp2_connector_unref(grpc_connector* con) { |
|
|
|
void Chttp2Connector::Connect(const Args& args, Result* result, |
|
|
|
chttp2_connector* c = reinterpret_cast<chttp2_connector*>(con); |
|
|
|
grpc_closure* notify) { |
|
|
|
if (gpr_unref(&c->refs)) { |
|
|
|
grpc_resolved_address addr; |
|
|
|
gpr_mu_destroy(&c->mu); |
|
|
|
Subchannel::GetAddressFromSubchannelAddressArg(args.channel_args, &addr); |
|
|
|
// If handshaking is not yet in progress, destroy the endpoint.
|
|
|
|
grpc_endpoint** ep; |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
MutexLock lock(&mu_); |
|
|
|
|
|
|
|
GPR_ASSERT(notify_ == nullptr); |
|
|
|
|
|
|
|
args_ = args; |
|
|
|
|
|
|
|
result_ = result; |
|
|
|
|
|
|
|
notify_ = notify; |
|
|
|
|
|
|
|
GPR_ASSERT(!connecting_); |
|
|
|
|
|
|
|
connecting_ = true; |
|
|
|
|
|
|
|
GPR_ASSERT(endpoint_ == nullptr); |
|
|
|
|
|
|
|
ep = &endpoint_; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
// In some implementations, the closure can be flushed before
|
|
|
|
|
|
|
|
// grpc_tcp_client_connect() returns, and since the closure requires access
|
|
|
|
|
|
|
|
// to mu_, this can result in a deadlock (see
|
|
|
|
|
|
|
|
// https://github.com/grpc/grpc/issues/16427 for details).
|
|
|
|
|
|
|
|
// grpc_tcp_client_connect() will fill endpoint_ with proper contents, and we
|
|
|
|
|
|
|
|
// make sure that we still exist at that point by taking a ref.
|
|
|
|
|
|
|
|
Ref().release(); // Ref held by callback.
|
|
|
|
|
|
|
|
grpc_tcp_client_connect(&connected_, ep, args.interested_parties, |
|
|
|
|
|
|
|
args.channel_args, &addr, args.deadline); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void Chttp2Connector::Shutdown(grpc_error* error) { |
|
|
|
|
|
|
|
MutexLock lock(&mu_); |
|
|
|
|
|
|
|
shutdown_ = true; |
|
|
|
|
|
|
|
if (handshake_mgr_ != nullptr) { |
|
|
|
|
|
|
|
handshake_mgr_->Shutdown(GRPC_ERROR_REF(error)); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
// If handshaking is not yet in progress, shutdown the endpoint.
|
|
|
|
// Otherwise, the handshaker will do this for us.
|
|
|
|
// Otherwise, the handshaker will do this for us.
|
|
|
|
if (c->endpoint != nullptr) grpc_endpoint_destroy(c->endpoint); |
|
|
|
if (!connecting_ && endpoint_ != nullptr) { |
|
|
|
gpr_free(c); |
|
|
|
grpc_endpoint_shutdown(endpoint_, GRPC_ERROR_REF(error)); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
GRPC_ERROR_UNREF(error); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
static void chttp2_connector_shutdown(grpc_connector* con, grpc_error* why) { |
|
|
|
void Chttp2Connector::Connected(void* arg, grpc_error* error) { |
|
|
|
chttp2_connector* c = reinterpret_cast<chttp2_connector*>(con); |
|
|
|
Chttp2Connector* self = static_cast<Chttp2Connector*>(arg); |
|
|
|
gpr_mu_lock(&c->mu); |
|
|
|
bool unref = false; |
|
|
|
c->shutdown = true; |
|
|
|
{ |
|
|
|
if (c->handshake_mgr != nullptr) { |
|
|
|
MutexLock lock(&self->mu_); |
|
|
|
c->handshake_mgr->Shutdown(GRPC_ERROR_REF(why)); |
|
|
|
GPR_ASSERT(self->connecting_); |
|
|
|
|
|
|
|
self->connecting_ = false; |
|
|
|
|
|
|
|
if (error != GRPC_ERROR_NONE || self->shutdown_) { |
|
|
|
|
|
|
|
if (error == GRPC_ERROR_NONE) { |
|
|
|
|
|
|
|
error = GRPC_ERROR_CREATE_FROM_STATIC_STRING("connector shutdown"); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
error = GRPC_ERROR_REF(error); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
if (self->endpoint_ != nullptr) { |
|
|
|
|
|
|
|
grpc_endpoint_shutdown(self->endpoint_, GRPC_ERROR_REF(error)); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
self->result_->Reset(); |
|
|
|
|
|
|
|
grpc_closure* notify = self->notify_; |
|
|
|
|
|
|
|
self->notify_ = nullptr; |
|
|
|
|
|
|
|
ExecCtx::Run(DEBUG_LOCATION, notify, error); |
|
|
|
|
|
|
|
unref = true; |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
GPR_ASSERT(self->endpoint_ != nullptr); |
|
|
|
|
|
|
|
self->StartHandshakeLocked(); |
|
|
|
} |
|
|
|
} |
|
|
|
// If handshaking is not yet in progress, shutdown the endpoint.
|
|
|
|
|
|
|
|
// Otherwise, the handshaker will do this for us.
|
|
|
|
|
|
|
|
if (!c->connecting && c->endpoint != nullptr) { |
|
|
|
|
|
|
|
grpc_endpoint_shutdown(c->endpoint, GRPC_ERROR_REF(why)); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
gpr_mu_unlock(&c->mu); |
|
|
|
if (unref) self->Unref(); |
|
|
|
GRPC_ERROR_UNREF(why); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
static void on_handshake_done(void* arg, grpc_error* error) { |
|
|
|
void Chttp2Connector::StartHandshakeLocked() { |
|
|
|
auto* args = static_cast<grpc_core::HandshakerArgs*>(arg); |
|
|
|
handshake_mgr_ = MakeRefCounted<HandshakeManager>(); |
|
|
|
chttp2_connector* c = static_cast<chttp2_connector*>(args->user_data); |
|
|
|
HandshakerRegistry::AddHandshakers(HANDSHAKER_CLIENT, args_.channel_args, |
|
|
|
gpr_mu_lock(&c->mu); |
|
|
|
args_.interested_parties, |
|
|
|
if (error != GRPC_ERROR_NONE || c->shutdown) { |
|
|
|
handshake_mgr_.get()); |
|
|
|
|
|
|
|
grpc_endpoint_add_to_pollset_set(endpoint_, args_.interested_parties); |
|
|
|
|
|
|
|
handshake_mgr_->DoHandshake(endpoint_, args_.channel_args, args_.deadline, |
|
|
|
|
|
|
|
nullptr /* acceptor */, OnHandshakeDone, this); |
|
|
|
|
|
|
|
endpoint_ = nullptr; // Endpoint handed off to handshake manager.
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
void Chttp2Connector::OnHandshakeDone(void* arg, grpc_error* error) { |
|
|
|
|
|
|
|
auto* args = static_cast<HandshakerArgs*>(arg); |
|
|
|
|
|
|
|
Chttp2Connector* self = static_cast<Chttp2Connector*>(args->user_data); |
|
|
|
|
|
|
|
{ |
|
|
|
|
|
|
|
MutexLock lock(&self->mu_); |
|
|
|
|
|
|
|
if (error != GRPC_ERROR_NONE || self->shutdown_) { |
|
|
|
if (error == GRPC_ERROR_NONE) { |
|
|
|
if (error == GRPC_ERROR_NONE) { |
|
|
|
error = GRPC_ERROR_CREATE_FROM_STATIC_STRING("connector shutdown"); |
|
|
|
error = GRPC_ERROR_CREATE_FROM_STATIC_STRING("connector shutdown"); |
|
|
|
// We were shut down after handshaking completed successfully, so
|
|
|
|
// We were shut down after handshaking completed successfully, so
|
|
|
|
@ -111,15 +151,15 @@ static void on_handshake_done(void* arg, grpc_error* error) { |
|
|
|
} else { |
|
|
|
} else { |
|
|
|
error = GRPC_ERROR_REF(error); |
|
|
|
error = GRPC_ERROR_REF(error); |
|
|
|
} |
|
|
|
} |
|
|
|
c->result->reset(); |
|
|
|
self->result_->Reset(); |
|
|
|
} else { |
|
|
|
} else { |
|
|
|
grpc_endpoint_delete_from_pollset_set(args->endpoint, |
|
|
|
grpc_endpoint_delete_from_pollset_set(args->endpoint, |
|
|
|
c->args.interested_parties); |
|
|
|
self->args_.interested_parties); |
|
|
|
c->result->transport = |
|
|
|
self->result_->transport = |
|
|
|
grpc_create_chttp2_transport(args->args, args->endpoint, true); |
|
|
|
grpc_create_chttp2_transport(args->args, args->endpoint, true); |
|
|
|
c->result->socket = |
|
|
|
self->result_->socket_node = |
|
|
|
grpc_chttp2_transport_get_socket_node(c->result->transport); |
|
|
|
grpc_chttp2_transport_get_socket_node(self->result_->transport); |
|
|
|
GPR_ASSERT(c->result->transport); |
|
|
|
GPR_ASSERT(self->result_->transport != nullptr); |
|
|
|
// TODO(roth): We ideally want to wait until we receive HTTP/2
|
|
|
|
// TODO(roth): We ideally want to wait until we receive HTTP/2
|
|
|
|
// settings from the server before we consider the connection
|
|
|
|
// settings from the server before we consider the connection
|
|
|
|
// established. If that doesn't happen before the connection
|
|
|
|
// established. If that doesn't happen before the connection
|
|
|
|
@ -144,96 +184,16 @@ static void on_handshake_done(void* arg, grpc_error* error) { |
|
|
|
// so until after transparent retries is implemented. Otherwise, any
|
|
|
|
// so until after transparent retries is implemented. Otherwise, any
|
|
|
|
// RPC that we attempt to send on the connection before the timeout
|
|
|
|
// RPC that we attempt to send on the connection before the timeout
|
|
|
|
// would fail instead of being retried on a subsequent attempt.
|
|
|
|
// would fail instead of being retried on a subsequent attempt.
|
|
|
|
grpc_chttp2_transport_start_reading(c->result->transport, args->read_buffer, |
|
|
|
grpc_chttp2_transport_start_reading(self->result_->transport, |
|
|
|
nullptr); |
|
|
|
args->read_buffer, nullptr); |
|
|
|
c->result->channel_args = args->args; |
|
|
|
self->result_->channel_args = args->args; |
|
|
|
} |
|
|
|
|
|
|
|
grpc_closure* notify = c->notify; |
|
|
|
|
|
|
|
c->notify = nullptr; |
|
|
|
|
|
|
|
grpc_core::ExecCtx::Run(DEBUG_LOCATION, notify, error); |
|
|
|
|
|
|
|
c->handshake_mgr.reset(); |
|
|
|
|
|
|
|
gpr_mu_unlock(&c->mu); |
|
|
|
|
|
|
|
chttp2_connector_unref(reinterpret_cast<grpc_connector*>(c)); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static void start_handshake_locked(chttp2_connector* c) { |
|
|
|
|
|
|
|
c->handshake_mgr = grpc_core::MakeRefCounted<grpc_core::HandshakeManager>(); |
|
|
|
|
|
|
|
grpc_core::HandshakerRegistry::AddHandshakers( |
|
|
|
|
|
|
|
grpc_core::HANDSHAKER_CLIENT, c->args.channel_args, |
|
|
|
|
|
|
|
c->args.interested_parties, c->handshake_mgr.get()); |
|
|
|
|
|
|
|
grpc_endpoint_add_to_pollset_set(c->endpoint, c->args.interested_parties); |
|
|
|
|
|
|
|
c->handshake_mgr->DoHandshake(c->endpoint, c->args.channel_args, |
|
|
|
|
|
|
|
c->args.deadline, nullptr /* acceptor */, |
|
|
|
|
|
|
|
on_handshake_done, c); |
|
|
|
|
|
|
|
c->endpoint = nullptr; // Endpoint handed off to handshake manager.
|
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static void connected(void* arg, grpc_error* error) { |
|
|
|
|
|
|
|
chttp2_connector* c = static_cast<chttp2_connector*>(arg); |
|
|
|
|
|
|
|
gpr_mu_lock(&c->mu); |
|
|
|
|
|
|
|
GPR_ASSERT(c->connecting); |
|
|
|
|
|
|
|
c->connecting = false; |
|
|
|
|
|
|
|
if (error != GRPC_ERROR_NONE || c->shutdown) { |
|
|
|
|
|
|
|
if (error == GRPC_ERROR_NONE) { |
|
|
|
|
|
|
|
error = GRPC_ERROR_CREATE_FROM_STATIC_STRING("connector shutdown"); |
|
|
|
|
|
|
|
} else { |
|
|
|
|
|
|
|
error = GRPC_ERROR_REF(error); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
c->result->reset(); |
|
|
|
|
|
|
|
grpc_closure* notify = c->notify; |
|
|
|
|
|
|
|
c->notify = nullptr; |
|
|
|
|
|
|
|
grpc_core::ExecCtx::Run(DEBUG_LOCATION, notify, error); |
|
|
|
|
|
|
|
if (c->endpoint != nullptr) { |
|
|
|
|
|
|
|
grpc_endpoint_shutdown(c->endpoint, GRPC_ERROR_REF(error)); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
gpr_mu_unlock(&c->mu); |
|
|
|
grpc_closure* notify = self->notify_; |
|
|
|
chttp2_connector_unref(static_cast<grpc_connector*>(arg)); |
|
|
|
self->notify_ = nullptr; |
|
|
|
} else { |
|
|
|
ExecCtx::Run(DEBUG_LOCATION, notify, error); |
|
|
|
GPR_ASSERT(c->endpoint != nullptr); |
|
|
|
self->handshake_mgr_.reset(); |
|
|
|
start_handshake_locked(c); |
|
|
|
|
|
|
|
gpr_mu_unlock(&c->mu); |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
self->Unref(); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
static void chttp2_connector_connect(grpc_connector* con, |
|
|
|
} // namespace grpc_core
|
|
|
|
const grpc_connect_in_args* args, |
|
|
|
|
|
|
|
grpc_connect_out_args* result, |
|
|
|
|
|
|
|
grpc_closure* notify) { |
|
|
|
|
|
|
|
chttp2_connector* c = reinterpret_cast<chttp2_connector*>(con); |
|
|
|
|
|
|
|
grpc_resolved_address addr; |
|
|
|
|
|
|
|
grpc_core::Subchannel::GetAddressFromSubchannelAddressArg(args->channel_args, |
|
|
|
|
|
|
|
&addr); |
|
|
|
|
|
|
|
gpr_mu_lock(&c->mu); |
|
|
|
|
|
|
|
GPR_ASSERT(c->notify == nullptr); |
|
|
|
|
|
|
|
c->notify = notify; |
|
|
|
|
|
|
|
c->args = *args; |
|
|
|
|
|
|
|
c->result = result; |
|
|
|
|
|
|
|
GPR_ASSERT(c->endpoint == nullptr); |
|
|
|
|
|
|
|
chttp2_connector_ref(con); // Ref taken for callback.
|
|
|
|
|
|
|
|
GRPC_CLOSURE_INIT(&c->connected, connected, c, grpc_schedule_on_exec_ctx); |
|
|
|
|
|
|
|
GPR_ASSERT(!c->connecting); |
|
|
|
|
|
|
|
c->connecting = true; |
|
|
|
|
|
|
|
grpc_closure* closure = &c->connected; |
|
|
|
|
|
|
|
grpc_endpoint** ep = &c->endpoint; |
|
|
|
|
|
|
|
gpr_mu_unlock(&c->mu); |
|
|
|
|
|
|
|
// In some implementations, the closure can be flushed before
|
|
|
|
|
|
|
|
// grpc_tcp_client_connect and since the closure requires access to c->mu,
|
|
|
|
|
|
|
|
// this can result in a deadlock. Refer
|
|
|
|
|
|
|
|
// https://github.com/grpc/grpc/issues/16427
|
|
|
|
|
|
|
|
// grpc_tcp_client_connect would fill c->endpoint with proper contents and we
|
|
|
|
|
|
|
|
// make sure that we would still exist at that point by taking a ref.
|
|
|
|
|
|
|
|
grpc_tcp_client_connect(closure, ep, args->interested_parties, |
|
|
|
|
|
|
|
args->channel_args, &addr, args->deadline); |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static const grpc_connector_vtable chttp2_connector_vtable = { |
|
|
|
|
|
|
|
chttp2_connector_ref, chttp2_connector_unref, chttp2_connector_shutdown, |
|
|
|
|
|
|
|
chttp2_connector_connect}; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
grpc_connector* grpc_chttp2_connector_create() { |
|
|
|
|
|
|
|
chttp2_connector* c = static_cast<chttp2_connector*>(gpr_zalloc(sizeof(*c))); |
|
|
|
|
|
|
|
c->base.vtable = &chttp2_connector_vtable; |
|
|
|
|
|
|
|
gpr_mu_init(&c->mu); |
|
|
|
|
|
|
|
gpr_ref_init(&c->refs, 1); |
|
|
|
|
|
|
|
return &c->base; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
Mark D. Roth
5 years ago
committed by
GitHub