This reverts commit a638c407bb.
pull/31623/head
Craig Tiller
2 years ago
committed by
GitHub
parent
51f296b4f8
commit
867dc6cae2
35 changed files with 21 additions and 837 deletions
@ -1,34 +0,0 @@ |
|||||||
# 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. |
|
||||||
|
|
||||||
|
|
||||||
find_package(systemd QUIET CONFIG) |
|
||||||
if(systemd_FOUND) |
|
||||||
message(STATUS "Found systemd via CMake.") |
|
||||||
return() |
|
||||||
endif() |
|
||||||
|
|
||||||
if(TARGET systemd) |
|
||||||
message(STATUS "Found systemd via pkg-config already?") |
|
||||||
return() |
|
||||||
endif() |
|
||||||
|
|
||||||
find_package(PkgConfig) |
|
||||||
pkg_check_modules(SYSTEMD libsystemd) |
|
||||||
|
|
||||||
if(SYSTEMD_FOUND) |
|
||||||
set(systemd_FOUND "${SYSTEMD_FOUND}") |
|
||||||
add_library(systemd INTERFACE IMPORTED) |
|
||||||
message(STATUS "Found systemd via pkg-config.") |
|
||||||
endif() |
|
||||||
@ -1,20 +0,0 @@ |
|||||||
# 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. |
|
||||||
|
|
||||||
find_package(systemd) |
|
||||||
if(TARGET systemd) |
|
||||||
set(_gRPC_SYSTEMD_LIBRARIES systemd ${SYSTEMD_LINK_LIBRARIES}) |
|
||||||
add_definitions(-DHAVE_LIBSYSTEMD) |
|
||||||
endif() |
|
||||||
set(_gRPC_FIND_SYSTEMD "if(NOT systemd_FOUND)\n find_package(systemd)\nendif()") |
|
||||||
@ -1,34 +0,0 @@ |
|||||||
# Copyright 2022 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. |
|
||||||
|
|
||||||
licenses(["notice"]) |
|
||||||
|
|
||||||
cc_binary( |
|
||||||
name = "client", |
|
||||||
srcs = ["client.cc"], |
|
||||||
deps = [ |
|
||||||
"//:grpc++", |
|
||||||
"//examples/protos:helloworld_cc_grpc", |
|
||||||
], |
|
||||||
) |
|
||||||
|
|
||||||
cc_binary( |
|
||||||
name = "server", |
|
||||||
srcs = ["server.cc"], |
|
||||||
deps = [ |
|
||||||
"//:grpc++", |
|
||||||
"//:grpc++_reflection", |
|
||||||
"//examples/protos:helloworld_cc_grpc", |
|
||||||
], |
|
||||||
) |
|
||||||
@ -1,100 +0,0 @@ |
|||||||
// Copyright 2022 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 <iostream> |
|
||||||
#include <memory> |
|
||||||
#include <string> |
|
||||||
|
|
||||||
#include "examples/protos/helloworld.grpc.pb.h" |
|
||||||
|
|
||||||
#include <grpcpp/grpcpp.h> |
|
||||||
|
|
||||||
using grpc::Channel; |
|
||||||
using grpc::ClientContext; |
|
||||||
using grpc::Status; |
|
||||||
using helloworld::Greeter; |
|
||||||
using helloworld::HelloReply; |
|
||||||
using helloworld::HelloRequest; |
|
||||||
|
|
||||||
class GreeterClient { |
|
||||||
public: |
|
||||||
GreeterClient(std::shared_ptr<Channel> channel) |
|
||||||
: stub_(Greeter::NewStub(channel)) {} |
|
||||||
|
|
||||||
// Assembles the client's payload, sends it and presents the response back
|
|
||||||
// from the server.
|
|
||||||
std::string SayHello(const std::string& user) { |
|
||||||
// Data we are sending to the server.
|
|
||||||
HelloRequest request; |
|
||||||
request.set_name(user); |
|
||||||
|
|
||||||
// Container for the data we expect from the server.
|
|
||||||
HelloReply reply; |
|
||||||
|
|
||||||
// Context for the client. It could be used to convey extra information to
|
|
||||||
// the server and/or tweak certain RPC behaviors.
|
|
||||||
ClientContext context; |
|
||||||
|
|
||||||
// The actual RPC.
|
|
||||||
Status status = stub_->SayHello(&context, request, &reply); |
|
||||||
|
|
||||||
// Act upon its status.
|
|
||||||
if (status.ok()) { |
|
||||||
return reply.message(); |
|
||||||
} else { |
|
||||||
std::cout << status.error_code() << ": " << status.error_message() |
|
||||||
<< std::endl; |
|
||||||
return "RPC failed"; |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
private: |
|
||||||
std::unique_ptr<Greeter::Stub> stub_; |
|
||||||
}; |
|
||||||
|
|
||||||
int main(int argc, char** argv) { |
|
||||||
// Instantiate the client. It requires a channel, out of which the actual RPCs
|
|
||||||
// are created. This channel models a connection to an endpoint specified by
|
|
||||||
// the argument "--target=" which is the only expected argument.
|
|
||||||
// We indicate that the channel isn't authenticated (use of
|
|
||||||
// InsecureChannelCredentials()).
|
|
||||||
std::string target_str; |
|
||||||
std::string arg_str("--target"); |
|
||||||
if (argc > 1) { |
|
||||||
std::string arg_val = argv[1]; |
|
||||||
size_t start_pos = arg_val.find(arg_str); |
|
||||||
if (start_pos != std::string::npos) { |
|
||||||
start_pos += arg_str.size(); |
|
||||||
if (arg_val[start_pos] == '=') { |
|
||||||
target_str = arg_val.substr(start_pos + 1); |
|
||||||
} else { |
|
||||||
std::cout << "The only correct argument syntax is --target=" |
|
||||||
<< std::endl; |
|
||||||
return 0; |
|
||||||
} |
|
||||||
} else { |
|
||||||
std::cout << "The only acceptable argument is --target=" << std::endl; |
|
||||||
return 0; |
|
||||||
} |
|
||||||
} else { |
|
||||||
target_str = "unix:/tmp/server"; |
|
||||||
} |
|
||||||
GreeterClient greeter( |
|
||||||
grpc::CreateChannel(target_str, grpc::InsecureChannelCredentials())); |
|
||||||
std::string user("world"); |
|
||||||
std::string reply = greeter.SayHello(user); |
|
||||||
std::cout << "Greeter received: " << reply << std::endl; |
|
||||||
|
|
||||||
return 0; |
|
||||||
} |
|
||||||
@ -1,68 +0,0 @@ |
|||||||
// Copyright 2022 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 <iostream> |
|
||||||
#include <memory> |
|
||||||
#include <string> |
|
||||||
|
|
||||||
#include "examples/protos/helloworld.grpc.pb.h" |
|
||||||
|
|
||||||
#include <grpcpp/ext/proto_server_reflection_plugin.h> |
|
||||||
#include <grpcpp/grpcpp.h> |
|
||||||
#include <grpcpp/health_check_service_interface.h> |
|
||||||
|
|
||||||
using grpc::Server; |
|
||||||
using grpc::ServerBuilder; |
|
||||||
using grpc::ServerContext; |
|
||||||
using grpc::Status; |
|
||||||
using helloworld::Greeter; |
|
||||||
using helloworld::HelloReply; |
|
||||||
using helloworld::HelloRequest; |
|
||||||
|
|
||||||
// Logic and data behind the server's behavior.
|
|
||||||
class GreeterServiceImpl final : public Greeter::Service { |
|
||||||
Status SayHello(ServerContext* context, const HelloRequest* request, |
|
||||||
HelloReply* reply) override { |
|
||||||
std::string prefix("Hello "); |
|
||||||
reply->set_message(prefix + request->name()); |
|
||||||
return Status::OK; |
|
||||||
} |
|
||||||
}; |
|
||||||
|
|
||||||
void RunServer() { |
|
||||||
std::string server_address("unix:/tmp/server"); |
|
||||||
GreeterServiceImpl service; |
|
||||||
|
|
||||||
grpc::EnableDefaultHealthCheckService(true); |
|
||||||
grpc::reflection::InitProtoReflectionServerBuilderPlugin(); |
|
||||||
ServerBuilder builder; |
|
||||||
// Listen on the given address without any authentication mechanism.
|
|
||||||
builder.AddListeningPort(server_address, grpc::InsecureServerCredentials()); |
|
||||||
// Register "service" as the instance through which we'll communicate with
|
|
||||||
// clients. In this case it corresponds to an *synchronous* service.
|
|
||||||
builder.RegisterService(&service); |
|
||||||
// Finally assemble the server.
|
|
||||||
std::unique_ptr<Server> server(builder.BuildAndStart()); |
|
||||||
std::cout << "Server listening on " << server_address << std::endl; |
|
||||||
|
|
||||||
// Wait for the server to shutdown. Note that some other thread must be
|
|
||||||
// responsible for shutting down the server for this call to ever return.
|
|
||||||
server->Wait(); |
|
||||||
} |
|
||||||
|
|
||||||
int main(int argc, char** argv) { |
|
||||||
RunServer(); |
|
||||||
|
|
||||||
return 0; |
|
||||||
} |
|
||||||
@ -1,73 +0,0 @@ |
|||||||
#!/usr/bin/env bash |
|
||||||
# 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. |
|
||||||
|
|
||||||
# Test structure borrowed with gratitude from |
|
||||||
# https://github.com/grpc/grpc-go/tree/master/examples/features |
|
||||||
|
|
||||||
# Run this script as root |
|
||||||
|
|
||||||
clean() { |
|
||||||
echo "Cleaning..." |
|
||||||
systemctl stop sdsockact.socket |
|
||||||
systemctl stop sdsockact.service |
|
||||||
systemctl daemon-reload |
|
||||||
rm /tmp/greeter_server |
|
||||||
rm /tmp/greeter_client |
|
||||||
rm /etc/systemd/system/sdsockact.service |
|
||||||
rm /etc/systemd/system/sdsockact.socket |
|
||||||
} |
|
||||||
|
|
||||||
fail() { |
|
||||||
clean |
|
||||||
echo "FAIL: $@" >&2 |
|
||||||
exit 1 |
|
||||||
} |
|
||||||
|
|
||||||
pass() { |
|
||||||
echo "SUCCESS: $1" |
|
||||||
} |
|
||||||
|
|
||||||
bazel build --define=use_systemd=true //examples/cpp/features/sd_sock_act:all || fail "Failed to build sd_sock_act" |
|
||||||
cp ../../../../bazel-bin/examples/cpp/features/sd_sock_act/server /tmp/greeter_server |
|
||||||
cp ../../../../bazel-bin/examples/cpp/features/sd_sock_act/client /tmp/greeter_client |
|
||||||
|
|
||||||
cat << EOF > /etc/systemd/system/sdsockact.service |
|
||||||
[Service] |
|
||||||
ExecStart=/tmp/greeter_server |
|
||||||
EOF |
|
||||||
|
|
||||||
cat << EOF > /etc/systemd/system/sdsockact.socket |
|
||||||
[Socket] |
|
||||||
ListenStream=/tmp/server |
|
||||||
ReusePort=true |
|
||||||
|
|
||||||
[Install] |
|
||||||
WantedBy=sockets.target |
|
||||||
EOF |
|
||||||
|
|
||||||
systemctl daemon-reload |
|
||||||
systemctl enable sdsockact.socket |
|
||||||
systemctl start sdsockact.socket |
|
||||||
|
|
||||||
pushd /tmp |
|
||||||
./greeter_client | grep "Hello" |
|
||||||
if [ $? -ne 0 ]; then |
|
||||||
popd |
|
||||||
fail "Response not received" |
|
||||||
fi |
|
||||||
|
|
||||||
popd |
|
||||||
pass "Response received" |
|
||||||
clean |
|
||||||
@ -1,115 +0,0 @@ |
|||||||
/*
|
|
||||||
* |
|
||||||
* 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. |
|
||||||
* |
|
||||||
*/ |
|
||||||
|
|
||||||
#include <grpc/support/port_platform.h> |
|
||||||
|
|
||||||
#include "src/core/lib/iomgr/port.h" |
|
||||||
|
|
||||||
#ifdef GRPC_POSIX_SOCKET_TCP_SERVER_UTILS_COMMON |
|
||||||
|
|
||||||
#ifdef HAVE_LIBSYSTEMD |
|
||||||
#include <systemd/sd-daemon.h> |
|
||||||
#endif |
|
||||||
|
|
||||||
#include "src/core/lib/address_utils/sockaddr_utils.h" |
|
||||||
#include "src/core/lib/iomgr/resolve_address.h" |
|
||||||
#include "src/core/lib/iomgr/sockaddr.h" |
|
||||||
#include "src/core/lib/iomgr/systemd_utils.h" |
|
||||||
|
|
||||||
#ifdef HAVE_LIBSYSTEMD |
|
||||||
bool set_matching_sd_unix_fd(grpc_tcp_server* s, |
|
||||||
const grpc_resolved_address* addr, int fd_start, |
|
||||||
int n) { |
|
||||||
absl::StatusOr<std::string> addr_name = grpc_sockaddr_to_string(addr, true); |
|
||||||
for (int i = fd_start; i < fd_start + n; i++) { |
|
||||||
if (sd_is_socket_unix(i, SOCK_STREAM, 1, addr_name.value().c_str(), 0)) { |
|
||||||
grpc_tcp_server_set_pre_allocated_fd(s, i); |
|
||||||
return true; |
|
||||||
} |
|
||||||
} |
|
||||||
return false; |
|
||||||
} |
|
||||||
|
|
||||||
bool set_matching_sd_inet_fd(grpc_tcp_server* s, |
|
||||||
const grpc_resolved_address* addr, int family, |
|
||||||
int port, int fd_start, int n) { |
|
||||||
for (int i = fd_start; i < fd_start + n; i++) { |
|
||||||
int r_inet = sd_is_socket_inet(i, family, SOCK_STREAM, 1, (uint16_t)port); |
|
||||||
int r_addr = sd_is_socket_sockaddr( |
|
||||||
i, SOCK_STREAM, |
|
||||||
reinterpret_cast<grpc_sockaddr*>(const_cast<char*>(addr->addr)), |
|
||||||
addr->len, 1); |
|
||||||
|
|
||||||
if (r_inet > 0 && r_addr > 0) { |
|
||||||
grpc_tcp_server_set_pre_allocated_fd(s, i); |
|
||||||
return true; |
|
||||||
} |
|
||||||
} |
|
||||||
return false; |
|
||||||
} |
|
||||||
|
|
||||||
void set_matching_sd_fds(grpc_tcp_server* s, const grpc_resolved_address* addr, |
|
||||||
int requested_port) { |
|
||||||
int n = sd_listen_fds(0); |
|
||||||
if (n <= 0) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
int fd_start = SD_LISTEN_FDS_START; |
|
||||||
grpc_resolved_address addr6_v4mapped; |
|
||||||
|
|
||||||
if (grpc_sockaddr_to_v4mapped(addr, &addr6_v4mapped)) { |
|
||||||
addr = &addr6_v4mapped; |
|
||||||
} |
|
||||||
|
|
||||||
int family = grpc_sockaddr_get_family(addr); |
|
||||||
int port = grpc_sockaddr_get_port(addr); |
|
||||||
|
|
||||||
if (family == AF_UNIX) { |
|
||||||
set_matching_sd_unix_fd(s, addr, fd_start, n); |
|
||||||
} else { |
|
||||||
if (grpc_sockaddr_is_wildcard(addr, &requested_port)) { |
|
||||||
grpc_resolved_address wild4; |
|
||||||
grpc_resolved_address wild6; |
|
||||||
grpc_resolved_address wildcard_addrs[2]; |
|
||||||
|
|
||||||
grpc_sockaddr_make_wildcards(requested_port, &wild4, &wild6); |
|
||||||
wildcard_addrs[0] = wild4; |
|
||||||
wildcard_addrs[1] = wild6; |
|
||||||
|
|
||||||
for (grpc_resolved_address addr_w : wildcard_addrs) { |
|
||||||
int family_w = grpc_sockaddr_get_family(&addr_w); |
|
||||||
int port_w = grpc_sockaddr_get_port(&addr_w); |
|
||||||
if (set_matching_sd_inet_fd(s, &addr_w, family_w, port_w, fd_start, |
|
||||||
n)) { |
|
||||||
return; |
|
||||||
} |
|
||||||
} |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
set_matching_sd_inet_fd(s, addr, family, port, fd_start, n); |
|
||||||
} |
|
||||||
} |
|
||||||
#else |
|
||||||
void set_matching_sd_fds(GRPC_UNUSED grpc_tcp_server* s, |
|
||||||
GRPC_UNUSED const grpc_resolved_address* addr, |
|
||||||
GRPC_UNUSED int requested_port) {} |
|
||||||
#endif /* HAVE_LIBSYSTEMD */ |
|
||||||
|
|
||||||
#endif /* GRPC_POSIX_SOCKET_TCP_SERVER_UTILS_COMMON */ |
|
||||||
@ -1,33 +0,0 @@ |
|||||||
/*
|
|
||||||
* |
|
||||||
* 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_CORE_LIB_IOMGR_SYSTEMD_UTILS_H |
|
||||||
#define GRPC_CORE_LIB_IOMGR_SYSTEMD_UTILS_H |
|
||||||
|
|
||||||
#include <grpc/support/port_platform.h> |
|
||||||
|
|
||||||
#include "src/core/lib/iomgr/tcp_server_utils_posix.h" |
|
||||||
|
|
||||||
/* Check whether systemd has pre-allocated FDs. If so, check whether any
|
|
||||||
* pre-allocated FD is valid, i.e. matches addr and its family. If there is |
|
||||||
* any valid FD, set its value to s->pre_allocated_fd |
|
||||||
*/ |
|
||||||
void set_matching_sd_fds(grpc_tcp_server* s, const grpc_resolved_address* addr, |
|
||||||
int requested_port); |
|
||||||
|
|
||||||
#endif /* GRPC_CORE_LIB_IOMGR_SYSTEMD_UTILS_H */ |
|
||||||
Loading…
Reference in new issue