mirror of https://github.com/grpc/grpc.git
Mark D. Roth
11 months ago
parent
599b50228a
commit
46a41400e0
25 changed files with 205 additions and 43 deletions
@ -0,0 +1,80 @@ |
||||
//
|
||||
// Copyright 2024 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 <grpc/support/port_platform.h> |
||||
|
||||
#include "src/core/lib/transport/endpoint_info_handshaker.h" |
||||
|
||||
#include <memory> |
||||
|
||||
#include "absl/status/status.h" |
||||
|
||||
#include "src/core/lib/channel/channel_args.h" |
||||
#include "src/core/lib/gprpp/debug_location.h" |
||||
#include "src/core/lib/gprpp/ref_counted_ptr.h" |
||||
#include "src/core/lib/iomgr/closure.h" |
||||
#include "src/core/lib/iomgr/endpoint.h" |
||||
#include "src/core/lib/iomgr/exec_ctx.h" |
||||
#include "src/core/lib/transport/handshaker.h" |
||||
#include "src/core/lib/transport/handshaker_factory.h" |
||||
#include "src/core/lib/transport/handshaker_registry.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
namespace { |
||||
|
||||
class EndpointInfoHandshaker : public Handshaker { |
||||
public: |
||||
const char* name() const override { return "endpoint_info"; } |
||||
|
||||
void DoHandshake(grpc_tcp_server_acceptor* /*acceptor*/, |
||||
grpc_closure* on_handshake_done, |
||||
HandshakerArgs* args) override { |
||||
args->args = args->args |
||||
.Set(GRPC_ARG_ENDPOINT_LOCAL_ADDRESS, |
||||
grpc_endpoint_get_local_address(args->endpoint)) |
||||
.Set(GRPC_ARG_ENDPOINT_PEER_ADDRESS, |
||||
grpc_endpoint_get_peer(args->endpoint)); |
||||
ExecCtx::Run(DEBUG_LOCATION, on_handshake_done, absl::OkStatus()); |
||||
} |
||||
|
||||
void Shutdown(grpc_error_handle /*why*/) override {} |
||||
}; |
||||
|
||||
class EndpointInfoHandshakerFactory : public HandshakerFactory { |
||||
public: |
||||
void AddHandshakers(const ChannelArgs& /*args*/, |
||||
grpc_pollset_set* /*interested_parties*/, |
||||
HandshakeManager* handshake_mgr) override { |
||||
handshake_mgr->Add(MakeRefCounted<EndpointInfoHandshaker>()); |
||||
} |
||||
|
||||
HandshakerPriority Priority() override { |
||||
// Needs to be after kTCPConnectHandshakers.
|
||||
return HandshakerPriority::kSecurityHandshakers; |
||||
} |
||||
}; |
||||
|
||||
} // namespace
|
||||
|
||||
void RegisterEndpointInfoHandshaker(CoreConfiguration::Builder* builder) { |
||||
builder->handshaker_registry()->RegisterHandshakerFactory( |
||||
HANDSHAKER_CLIENT, std::make_unique<EndpointInfoHandshakerFactory>()); |
||||
builder->handshaker_registry()->RegisterHandshakerFactory( |
||||
HANDSHAKER_SERVER, std::make_unique<EndpointInfoHandshakerFactory>()); |
||||
} |
||||
|
||||
} // namespace grpc_core
|
||||
@ -0,0 +1,37 @@ |
||||
//
|
||||
// Copyright 2022 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.
|
||||
//
|
||||
|
||||
#ifndef GRPC_SRC_CORE_LIB_TRANSPORT_ENDPOINT_INFO_HANDSHAKER_H |
||||
#define GRPC_SRC_CORE_LIB_TRANSPORT_ENDPOINT_INFO_HANDSHAKER_H |
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include "src/core/lib/config/core_configuration.h" |
||||
|
||||
// Set by the handshaker to indicate the local address of the endpoint.
|
||||
#define GRPC_ARG_ENDPOINT_LOCAL_ADDRESS "grpc.internal.endpoint_local_address" |
||||
|
||||
// Set by the handshaker to indicate the peer address of the endpoint.
|
||||
#define GRPC_ARG_ENDPOINT_PEER_ADDRESS "grpc.internal.endpoint_peer_address" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
// Register the endpoint info handshaker into the configuration builder.
|
||||
void RegisterEndpointInfoHandshaker(CoreConfiguration::Builder* builder); |
||||
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif // GRPC_SRC_CORE_LIB_TRANSPORT_ENDPOINT_INFO_HANDSHAKER_H
|
||||
Loading…
Reference in new issue