mirror of https://github.com/grpc/grpc.git
[EventEngine] Implement Windows Native Resolver (#35458)
Closes #35458
COPYBARA_INTEGRATE_REVIEW=https://github.com/grpc/grpc/pull/35458 from drfloob:windows-native-ee-resolver 364b710342
PiperOrigin-RevId: 596981099
pull/35486/head
AJ Heller
11 months ago
committed by
Copybara-Service
parent
d8ae1d28c5
commit
745a61222d
18 changed files with 218 additions and 4 deletions
@ -0,0 +1,114 @@ |
|||||||
|
// Copyright 2024 The 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> |
||||||
|
|
||||||
|
#ifdef GPR_WINDOWS |
||||||
|
#include <inttypes.h> |
||||||
|
#include <string.h> |
||||||
|
#include <sys/types.h> |
||||||
|
|
||||||
|
#include <string> |
||||||
|
|
||||||
|
#include "absl/strings/str_format.h" |
||||||
|
|
||||||
|
#include <grpc/event_engine/event_engine.h> |
||||||
|
|
||||||
|
#include "src/core/lib/event_engine/windows/native_windows_dns_resolver.h" |
||||||
|
#include "src/core/lib/gprpp/host_port.h" |
||||||
|
#include "src/core/lib/gprpp/status_helper.h" |
||||||
|
#include "src/core/lib/iomgr/error.h" |
||||||
|
|
||||||
|
namespace grpc_event_engine { |
||||||
|
namespace experimental { |
||||||
|
|
||||||
|
namespace { |
||||||
|
absl::StatusOr<std::vector<EventEngine::ResolvedAddress>> |
||||||
|
LookupHostnameBlocking(absl::string_view name, absl::string_view default_port) { |
||||||
|
std::vector<EventEngine::ResolvedAddress> addresses; |
||||||
|
// parse name, splitting it into host and port parts
|
||||||
|
std::string host; |
||||||
|
std::string port; |
||||||
|
grpc_core::SplitHostPort(name, &host, &port); |
||||||
|
if (host.empty()) { |
||||||
|
return absl::InvalidArgumentError(absl::StrCat("Unparseable name: ", name)); |
||||||
|
} |
||||||
|
if (port.empty()) { |
||||||
|
if (default_port.empty()) { |
||||||
|
return absl::InvalidArgumentError( |
||||||
|
absl::StrFormat("No port in name %s or default_port argument", name)); |
||||||
|
} |
||||||
|
port = std::string(default_port); |
||||||
|
} |
||||||
|
// Call getaddrinfo
|
||||||
|
struct addrinfo hints; |
||||||
|
memset(&hints, 0, sizeof(hints)); |
||||||
|
hints.ai_family = AF_UNSPEC; // ipv4 or ipv6
|
||||||
|
hints.ai_socktype = SOCK_STREAM; // stream socket
|
||||||
|
hints.ai_flags = AI_PASSIVE; // for wildcard IP address
|
||||||
|
struct addrinfo* result = nullptr; |
||||||
|
int getaddrinfo_error = |
||||||
|
getaddrinfo(host.c_str(), port.c_str(), &hints, &result); |
||||||
|
if (getaddrinfo_error != 0) { |
||||||
|
return absl::UnknownError( |
||||||
|
absl::StrFormat("Address lookup failed for %s os_error: %s", name, |
||||||
|
grpc_core::StatusToString( |
||||||
|
GRPC_WSA_ERROR(WSAGetLastError(), "getaddrinfo")) |
||||||
|
.c_str())); |
||||||
|
} |
||||||
|
// Success path: collect and return all addresses
|
||||||
|
for (auto* resp = result; resp != nullptr; resp = resp->ai_next) { |
||||||
|
addresses.emplace_back(resp->ai_addr, resp->ai_addrlen); |
||||||
|
} |
||||||
|
if (result) freeaddrinfo(result); |
||||||
|
return addresses; |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace
|
||||||
|
NativeWindowsDNSResolver::NativeWindowsDNSResolver( |
||||||
|
std::shared_ptr<EventEngine> event_engine) |
||||||
|
: event_engine_(std::move(event_engine)) {} |
||||||
|
|
||||||
|
void NativeWindowsDNSResolver::LookupHostname( |
||||||
|
EventEngine::DNSResolver::LookupHostnameCallback on_resolved, |
||||||
|
absl::string_view name, absl::string_view default_port) { |
||||||
|
event_engine_->Run( |
||||||
|
[name, default_port, on_resolved = std::move(on_resolved)]() mutable { |
||||||
|
on_resolved(LookupHostnameBlocking(name, default_port)); |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
void NativeWindowsDNSResolver::LookupSRV( |
||||||
|
EventEngine::DNSResolver::LookupSRVCallback on_resolved, |
||||||
|
absl::string_view /* name */) { |
||||||
|
// Not supported
|
||||||
|
event_engine_->Run([on_resolved = std::move(on_resolved)]() mutable { |
||||||
|
on_resolved(absl::UnimplementedError( |
||||||
|
"The Native resolver does not support looking up SRV records")); |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
void NativeWindowsDNSResolver::LookupTXT( |
||||||
|
EventEngine::DNSResolver::LookupTXTCallback on_resolved, |
||||||
|
absl::string_view /* name */) { |
||||||
|
// Not supported
|
||||||
|
event_engine_->Run([on_resolved = std::move(on_resolved)]() mutable { |
||||||
|
on_resolved(absl::UnimplementedError( |
||||||
|
"The Native resolver does not support looking up TXT records")); |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace experimental
|
||||||
|
} // namespace grpc_event_engine
|
||||||
|
|
||||||
|
#endif // GPR_WINDOWS
|
||||||
@ -0,0 +1,51 @@ |
|||||||
|
// Copyright 2024 The 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_EVENT_ENGINE_WINDOWS_NATIVE_WINDOWS_DNS_RESOLVER_H |
||||||
|
#define GRPC_SRC_CORE_LIB_EVENT_ENGINE_WINDOWS_NATIVE_WINDOWS_DNS_RESOLVER_H |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#ifdef GPR_WINDOWS |
||||||
|
|
||||||
|
#include <grpc/event_engine/event_engine.h> |
||||||
|
|
||||||
|
namespace grpc_event_engine { |
||||||
|
namespace experimental { |
||||||
|
|
||||||
|
// An asynchronous DNS resolver which uses the native platform's getaddrinfo
|
||||||
|
// API. Only supports A/AAAA records.
|
||||||
|
class NativeWindowsDNSResolver : public EventEngine::DNSResolver { |
||||||
|
public: |
||||||
|
explicit NativeWindowsDNSResolver(std::shared_ptr<EventEngine> event_engine); |
||||||
|
|
||||||
|
void LookupHostname( |
||||||
|
EventEngine::DNSResolver::LookupHostnameCallback on_resolved, |
||||||
|
absl::string_view name, absl::string_view default_port) override; |
||||||
|
|
||||||
|
void LookupSRV(EventEngine::DNSResolver::LookupSRVCallback on_resolved, |
||||||
|
absl::string_view name) override; |
||||||
|
|
||||||
|
void LookupTXT(EventEngine::DNSResolver::LookupTXTCallback on_resolved, |
||||||
|
absl::string_view name) override; |
||||||
|
|
||||||
|
private: |
||||||
|
std::shared_ptr<EventEngine> event_engine_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace experimental
|
||||||
|
} // namespace grpc_event_engine
|
||||||
|
|
||||||
|
#endif |
||||||
|
|
||||||
|
#endif // GRPC_SRC_CORE_LIB_EVENT_ENGINE_WINDOWS_NATIVE_WINDOWS_DNS_RESOLVER_H
|
||||||
Loading…
Reference in new issue