Skip to content

Commit

Permalink
[export] [endpoint] Remove grpc_endpoint_shutdown().
Browse files Browse the repository at this point in the history
This gives grpc_endpoint the same destruction-is-shutdown semantic as
EventEngine::Endpoint, which will make the migration easier.

----
DO NOT SUBMIT. This PR is for testing purposes only. [cl/620101818](http://cl/620101818)

PiperOrigin-RevId: 620101818
  • Loading branch information
markdroth authored and copybara-github committed Jun 3, 2024
1 parent b2e1405 commit 6cf0b4a
Show file tree
Hide file tree
Showing 44 changed files with 323 additions and 345 deletions.
1 change: 1 addition & 0 deletions BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -4792,6 +4792,7 @@ grpc_cc_library(
"chttp2_context_list_entry",
"chttp2_legacy_frame",
"chttp2_varint",
"config_vars",
"debug_location",
"exec_ctx",
"gpr",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -289,7 +289,6 @@ void ChaoticGoodConnector::OnHandshakeDone(void* arg, grpc_error_handle error) {
// We were shut down after handshaking completed successfully, so
// destroy the endpoint here.
if (args->endpoint != nullptr) {
grpc_endpoint_shutdown(args->endpoint, error);
grpc_endpoint_destroy(args->endpoint);
}
}
Expand Down
28 changes: 5 additions & 23 deletions src/core/ext/transport/chttp2/client/chttp2_connector.cc
Original file line number Diff line number Diff line change
Expand Up @@ -96,12 +96,6 @@ void NullThenSchedClosure(const DebugLocation& location, grpc_closure** closure,
}
} // namespace

Chttp2Connector::~Chttp2Connector() {
if (endpoint_ != nullptr) {
grpc_endpoint_destroy(endpoint_);
}
}

void Chttp2Connector::Connect(const Args& args, Result* result,
grpc_closure* notify) {
{
Expand All @@ -110,7 +104,6 @@ void Chttp2Connector::Connect(const Args& args, Result* result,
args_ = args;
result_ = result;
notify_ = notify;
CHECK_EQ(endpoint_, nullptr);
event_engine_ = args_.channel_args.GetObject<EventEngine>();
}
absl::StatusOr<std::string> address = grpc_sockaddr_to_uri(args.address);
Expand Down Expand Up @@ -153,11 +146,6 @@ void Chttp2Connector::OnHandshakeDone(void* arg, grpc_error_handle error) {
// We were shut down after handshaking completed successfully, so
// destroy the endpoint here.
if (args->endpoint != nullptr) {
// TODO(ctiller): It is currently necessary to shutdown endpoints
// before destroying them, even if we know that there are no
// pending read/write callbacks. This should be fixed, at which
// point this can be removed.
grpc_endpoint_shutdown(args->endpoint, error);
grpc_endpoint_destroy(args->endpoint);
grpc_slice_buffer_destroy(args->read_buffer);
gpr_free(args->read_buffer);
Expand All @@ -172,13 +160,12 @@ void Chttp2Connector::OnHandshakeDone(void* arg, grpc_error_handle error) {
self->result_->socket_node =
grpc_chttp2_transport_get_socket_node(self->result_->transport);
self->result_->channel_args = args->args;
self->endpoint_ = args->endpoint;
self->Ref().release(); // Ref held by OnReceiveSettings()
GRPC_CLOSURE_INIT(&self->on_receive_settings_, OnReceiveSettings, self,
grpc_schedule_on_exec_ctx);
grpc_chttp2_transport_start_reading(self->result_->transport,
args->read_buffer,
&self->on_receive_settings_, nullptr);
grpc_chttp2_transport_start_reading(
self->result_->transport, args->read_buffer,
&self->on_receive_settings_, self->args_.interested_parties, nullptr);
self->timer_handle_ = self->event_engine_->RunAfter(
self->args_.deadline - Timestamp::Now(),
[self = self->RefAsSubclass<Chttp2Connector>()] {
Expand All @@ -203,8 +190,6 @@ void Chttp2Connector::OnReceiveSettings(void* arg, grpc_error_handle error) {
{
MutexLock lock(&self->mu_);
if (!self->notify_error_.has_value()) {
grpc_endpoint_delete_from_pollset_set(self->endpoint_,
self->args_.interested_parties);
if (!error.ok()) {
// Transport got an error while waiting on SETTINGS frame.
self->result_->Reset();
Expand Down Expand Up @@ -233,7 +218,6 @@ void Chttp2Connector::OnTimeout() {
if (!notify_error_.has_value()) {
// The transport did not receive the settings frame in time. Destroy the
// transport.
grpc_endpoint_delete_from_pollset_set(endpoint_, args_.interested_parties);
result_->Reset();
MaybeNotify(GRPC_ERROR_CREATE(
"connection attempt timed out before receiving SETTINGS frame"));
Expand All @@ -248,9 +232,6 @@ void Chttp2Connector::MaybeNotify(grpc_error_handle error) {
if (notify_error_.has_value()) {
NullThenSchedClosure(DEBUG_LOCATION, &notify_, notify_error_.value());
// Clear state for a new Connect().
// Clear out the endpoint_, since it is the responsibility of
// the transport to shut it down.
endpoint_ = nullptr;
notify_error_.reset();
} else {
notify_error_ = error;
Expand Down Expand Up @@ -409,7 +390,8 @@ grpc_channel* grpc_channel_create_from_fd(const char* target, int fd,
auto channel = grpc_core::ChannelCreate(
target, final_args, GRPC_CLIENT_DIRECT_CHANNEL, transport);
if (channel.ok()) {
grpc_chttp2_transport_start_reading(transport, nullptr, nullptr, nullptr);
grpc_chttp2_transport_start_reading(transport, nullptr, nullptr, nullptr,
nullptr);
grpc_core::ExecCtx::Get()->Flush();
return channel->release()->c_ptr();
} else {
Expand Down
5 changes: 0 additions & 5 deletions src/core/ext/transport/chttp2/client/chttp2_connector.h
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,6 @@ namespace grpc_core {

class Chttp2Connector : public SubchannelConnector {
public:
~Chttp2Connector() override;

void Connect(const Args& args, Result* result, grpc_closure* notify) override;
void Shutdown(grpc_error_handle error) override;

Expand All @@ -63,9 +61,6 @@ class Chttp2Connector : public SubchannelConnector {
Result* result_ = nullptr;
grpc_closure* notify_ = nullptr;
bool shutdown_ = false;
// Holds the endpoint when first created before being handed off to
// the handshake manager, and then again after handshake is done.
grpc_endpoint* endpoint_ = nullptr;
grpc_closure on_receive_settings_;
absl::optional<grpc_event_engine::experimental::EventEngine::TaskHandle>
timer_handle_ ABSL_GUARDED_BY(mu_);
Expand Down
35 changes: 12 additions & 23 deletions src/core/ext/transport/chttp2/server/chttp2_server.cc
Original file line number Diff line number Diff line change
Expand Up @@ -460,11 +460,6 @@ void Chttp2ServerListener::ActiveConnection::HandshakingState::OnHandshakeDone(
if (error.ok() && args->endpoint != nullptr) {
// We were shut down or stopped serving after handshaking completed
// successfully, so destroy the endpoint here.
// TODO(ctiller): It is currently necessary to shutdown endpoints
// before destroying them, even if we know that there are no
// pending read/write callbacks. This should be fixed, at which
// point this can be removed.
grpc_endpoint_shutdown(args->endpoint, absl::OkStatus());
grpc_endpoint_destroy(args->endpoint);
grpc_slice_buffer_destroy(args->read_buffer);
gpr_free(args->read_buffer);
Expand Down Expand Up @@ -517,7 +512,7 @@ void Chttp2ServerListener::ActiveConnection::HandshakingState::OnHandshakeDone(
}
grpc_chttp2_transport_start_reading(
transport.get(), args->read_buffer, &self->on_receive_settings_,
on_close);
nullptr, on_close);
self->timer_handle_ = self->connection_->event_engine_->RunAfter(
self->deadline_ - Timestamp::Now(),
[self = self->Ref()]() mutable {
Expand Down Expand Up @@ -645,7 +640,6 @@ void Chttp2ServerListener::ActiveConnection::Start(
// owning Chttp2ServerListener and all associated ActiveConnections have
// been orphaned. The generated endpoints need to be shutdown here to
// ensure the tcp connections are closed appropriately.
grpc_endpoint_shutdown(endpoint, absl::OkStatus());
grpc_endpoint_destroy(endpoint);
return;
}
Expand Down Expand Up @@ -856,36 +850,30 @@ void Chttp2ServerListener::OnAccept(void* arg, grpc_endpoint* tcp,
MutexLock lock(&self->mu_);
connection_manager = self->connection_manager_;
}
auto endpoint_cleanup = [&](grpc_error_handle error) {
grpc_endpoint_shutdown(tcp, error);
auto endpoint_cleanup = [&]() {
grpc_endpoint_destroy(tcp);
gpr_free(acceptor);
};
if (!self->connection_quota_->AllowIncomingConnection(
self->memory_quota_, grpc_endpoint_get_peer(tcp))) {
grpc_error_handle error = GRPC_ERROR_CREATE(
"Rejected incoming connection because configured connection quota "
"limits have been exceeded.");
endpoint_cleanup(error);
endpoint_cleanup();
return;
}
if (self->config_fetcher_ != nullptr) {
if (connection_manager == nullptr) {
grpc_error_handle error = GRPC_ERROR_CREATE(
"No ConnectionManager configured. Closing connection.");
endpoint_cleanup(error);
endpoint_cleanup();
return;
}
absl::StatusOr<ChannelArgs> args_result =
connection_manager->UpdateChannelArgsForConnection(args, tcp);
if (!args_result.ok()) {
endpoint_cleanup(GRPC_ERROR_CREATE(args_result.status().ToString()));
endpoint_cleanup();
return;
}
grpc_error_handle error;
args = self->args_modifier_(*args_result, &error);
if (!error.ok()) {
endpoint_cleanup(error);
endpoint_cleanup();
return;
}
}
Expand Down Expand Up @@ -915,7 +903,7 @@ void Chttp2ServerListener::OnAccept(void* arg, grpc_endpoint* tcp,
}
}
if (connection != nullptr) {
endpoint_cleanup(absl::OkStatus());
endpoint_cleanup();
} else {
connection_ref->Start(std::move(listener_ref), tcp, args);
}
Expand Down Expand Up @@ -1177,16 +1165,17 @@ void grpc_server_add_channel_from_fd(grpc_server* server, int fd,
grpc_fd_create(fd, name.c_str(), true),
grpc_event_engine::experimental::ChannelArgsEndpointConfig(server_args),
name);
for (grpc_pollset* pollset : core_server->pollsets()) {
grpc_endpoint_add_to_pollset(server_endpoint, pollset);
}
grpc_core::Transport* transport = grpc_create_chttp2_transport(
server_args, server_endpoint, false // is_client
);
grpc_error_handle error =
core_server->SetupTransport(transport, nullptr, server_args, nullptr);
if (error.ok()) {
for (grpc_pollset* pollset : core_server->pollsets()) {
grpc_endpoint_add_to_pollset(server_endpoint, pollset);
}
grpc_chttp2_transport_start_reading(transport, nullptr, nullptr, nullptr);
grpc_chttp2_transport_start_reading(transport, nullptr, nullptr, nullptr,
nullptr);
} else {
LOG(ERROR) << "Failed to create channel: "
<< grpc_core::StatusToString(error);
Expand Down
48 changes: 39 additions & 9 deletions src/core/ext/transport/chttp2/transport/chttp2_transport.cc
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,7 @@
#include "src/core/ext/transport/chttp2/transport/varint.h"
#include "src/core/ext/transport/chttp2/transport/write_size_policy.h"
#include "src/core/lib/channel/channel_args.h"
#include "src/core/lib/config/config_vars.h"
#include "src/core/lib/experiments/experiments.h"
#include "src/core/lib/gprpp/bitset.h"
#include "src/core/lib/gprpp/crash.h"
Expand All @@ -83,6 +84,7 @@
#include "src/core/lib/gprpp/time.h"
#include "src/core/lib/iomgr/combiner.h"
#include "src/core/lib/iomgr/error.h"
#include "src/core/lib/iomgr/ev_posix.h"
#include "src/core/lib/iomgr/exec_ctx.h"
#include "src/core/lib/iomgr/iomgr_fwd.h"
#include "src/core/lib/iomgr/port.h"
Expand Down Expand Up @@ -110,10 +112,6 @@
#include "src/core/util/string.h"
#include "src/core/util/useful.h"

#ifdef GRPC_POSIX_SOCKET_TCP
#include "src/core/lib/iomgr/ev_posix.h"
#endif

#define DEFAULT_CONNECTION_WINDOW_TARGET (1024 * 1024)
#define MAX_WINDOW 0x7fffffffu
#define MAX_WRITE_BUFFER_SIZE (64 * 1024 * 1024)
Expand Down Expand Up @@ -380,7 +378,7 @@ grpc_chttp2_transport::~grpc_chttp2_transport() {
channelz_socket.reset();
}

grpc_endpoint_destroy(ep);
if (ep != nullptr) grpc_endpoint_destroy(ep);

grpc_slice_buffer_destroy(&qbuf);

Expand Down Expand Up @@ -749,9 +747,21 @@ static void close_transport_locked(grpc_chttp2_transport* t,
GRPC_CHTTP2_STREAM_UNREF(s, "chttp2_writing:close");
}
CHECK(t->write_state == GRPC_CHTTP2_WRITE_STATE_IDLE);
grpc_endpoint_shutdown(t->ep, error);
if (t->interested_parties_until_recv_settings != nullptr) {
grpc_endpoint_delete_from_pollset_set(
t->ep, t->interested_parties_until_recv_settings);
t->interested_parties_until_recv_settings = nullptr;
}
grpc_core::MutexLock lock(&t->ep_destroy_mu);
grpc_endpoint_destroy(t->ep);
t->ep = nullptr;
}
if (t->notify_on_receive_settings != nullptr) {
if (t->interested_parties_until_recv_settings != nullptr) {
grpc_endpoint_delete_from_pollset_set(
t->ep, t->interested_parties_until_recv_settings);
t->interested_parties_until_recv_settings = nullptr;
}
grpc_core::ExecCtx::Run(DEBUG_LOCATION, t->notify_on_receive_settings,
error);
t->notify_on_receive_settings = nullptr;
Expand Down Expand Up @@ -2997,12 +3007,22 @@ static void connectivity_state_set(grpc_chttp2_transport* t,

void grpc_chttp2_transport::SetPollset(grpc_stream* /*gs*/,
grpc_pollset* pollset) {
grpc_endpoint_add_to_pollset(ep, pollset);
// We don't want the overhead of acquiring the mutex unless we're
// using the "poll" polling engine, which is the only one that
// actually uses pollsets.
if (strcmp(grpc_get_poll_strategy_name(), "poll") != 0) return;
grpc_core::MutexLock lock(&ep_destroy_mu);
if (ep != nullptr) grpc_endpoint_add_to_pollset(ep, pollset);
}

void grpc_chttp2_transport::SetPollsetSet(grpc_stream* /*gs*/,
grpc_pollset_set* pollset_set) {
grpc_endpoint_add_to_pollset_set(ep, pollset_set);
// We don't want the overhead of acquiring the mutex unless we're
// using the "poll" polling engine, which is the only one that
// actually uses pollsets.
if (strcmp(grpc_get_poll_strategy_name(), "poll") != 0) return;
grpc_core::MutexLock lock(&ep_destroy_mu);
if (ep != nullptr) grpc_endpoint_add_to_pollset_set(ep, pollset_set);
}

//
Expand Down Expand Up @@ -3188,7 +3208,9 @@ grpc_core::Transport* grpc_create_chttp2_transport(

void grpc_chttp2_transport_start_reading(
grpc_core::Transport* transport, grpc_slice_buffer* read_buffer,
grpc_closure* notify_on_receive_settings, grpc_closure* notify_on_close) {
grpc_closure* notify_on_receive_settings,
grpc_pollset_set* interested_parties_until_recv_settings,
grpc_closure* notify_on_close) {
auto t = reinterpret_cast<grpc_chttp2_transport*>(transport)->Ref();
if (read_buffer != nullptr) {
grpc_slice_buffer_move_into(read_buffer, &t->read_buffer);
Expand All @@ -3197,9 +3219,15 @@ void grpc_chttp2_transport_start_reading(
auto* tp = t.get();
tp->combiner->Run(
grpc_core::NewClosure([t = std::move(t), notify_on_receive_settings,
interested_parties_until_recv_settings,
notify_on_close](grpc_error_handle) mutable {
if (!t->closed_with_error.ok()) {
if (notify_on_receive_settings != nullptr) {
if (t->ep != nullptr &&
interested_parties_until_recv_settings != nullptr) {
grpc_endpoint_delete_from_pollset_set(
t->ep, interested_parties_until_recv_settings);
}
grpc_core::ExecCtx::Run(DEBUG_LOCATION, notify_on_receive_settings,
t->closed_with_error);
}
Expand All @@ -3209,6 +3237,8 @@ void grpc_chttp2_transport_start_reading(
}
return;
}
t->interested_parties_until_recv_settings =
interested_parties_until_recv_settings;
t->notify_on_receive_settings = notify_on_receive_settings;
t->notify_on_close = notify_on_close;
read_action_locked(std::move(t), absl::OkStatus());
Expand Down
7 changes: 6 additions & 1 deletion src/core/ext/transport/chttp2/transport/chttp2_transport.h
Original file line number Diff line number Diff line change
Expand Up @@ -59,9 +59,14 @@ grpc_chttp2_transport_get_socket_node(grpc_core::Transport* transport);
/// leftover bytes previously read from the endpoint (e.g., by handshakers).
/// If non-null, \a notify_on_receive_settings will be scheduled when
/// HTTP/2 settings are received from the peer.
/// If non-null, the endpoint will be removed from
/// interested_parties_until_recv_settings before
/// notify_on_receive_settings is invoked.
void grpc_chttp2_transport_start_reading(
grpc_core::Transport* transport, grpc_slice_buffer* read_buffer,
grpc_closure* notify_on_receive_settings, grpc_closure* notify_on_close);
grpc_closure* notify_on_receive_settings,
grpc_pollset_set* interested_parties_until_recv_settings,
grpc_closure* notify_on_close);

namespace grpc_core {
typedef void (*TestOnlyGlobalHttp2TransportInitCallback)();
Expand Down
5 changes: 5 additions & 0 deletions src/core/ext/transport/chttp2/transport/frame_settings.cc
Original file line number Diff line number Diff line change
Expand Up @@ -109,6 +109,11 @@ grpc_error_handle grpc_chttp2_settings_parser_parse(void* p,
grpc_chttp2_initiate_write(t,
GRPC_CHTTP2_INITIATE_WRITE_SETTINGS_ACK);
if (t->notify_on_receive_settings != nullptr) {
if (t->interested_parties_until_recv_settings != nullptr) {
grpc_endpoint_delete_from_pollset_set(
t->ep, t->interested_parties_until_recv_settings);
t->interested_parties_until_recv_settings = nullptr;
}
grpc_core::ExecCtx::Run(DEBUG_LOCATION,
t->notify_on_receive_settings,
absl::OkStatus());
Expand Down
10 changes: 10 additions & 0 deletions src/core/ext/transport/chttp2/transport/internal.h
Original file line number Diff line number Diff line change
Expand Up @@ -258,6 +258,8 @@ struct grpc_chttp2_transport final : public grpc_core::FilterStackTransport,
void PerformOp(grpc_transport_op* op) override;

grpc_endpoint* ep;
grpc_core::Mutex ep_destroy_mu; // Guards endpoint destruction only.

grpc_core::Slice peer_string;

grpc_core::MemoryOwner memory_owner;
Expand All @@ -268,6 +270,14 @@ struct grpc_chttp2_transport final : public grpc_core::FilterStackTransport,
grpc_core::Combiner* combiner;
absl::BitGen bitgen;

// On the client side, when the transport is first created, the
// endpoint will already have been added to this pollset_set, and it
// needs to stay there until the notify_on_receive_settings callback
// is invoked. After that, the polling will be coordinated via the
// bind_pollset_set transport op, sent by the subchannel when it
// starts a connectivity watch.
grpc_pollset_set* interested_parties_until_recv_settings = nullptr;

grpc_closure* notify_on_receive_settings = nullptr;
grpc_closure* notify_on_close = nullptr;

Expand Down
Loading

0 comments on commit 6cf0b4a

Please sign in to comment.