| /* |
| * |
| * Copyright 2015 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 _POSIX_SOURCE |
| #define _POSIX_SOURCE |
| #endif |
| |
| #include <assert.h> |
| #include <signal.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <sys/types.h> |
| #include <sys/wait.h> |
| #include <unistd.h> |
| |
| #include <gflags/gflags.h> |
| #include <grpc/support/alloc.h> |
| #include <grpc/support/host_port.h> |
| #include <grpc/support/log.h> |
| #include <grpc/support/string_util.h> |
| #include "test/core/util/port.h" |
| #include "test/cpp/util/test_config.h" |
| |
| #include "src/core/lib/gpr/string.h" |
| #include "src/core/lib/iomgr/socket_utils_posix.h" |
| |
| DEFINE_string(extra_server_flags, "", "Extra flags to pass to server."); |
| |
| int test_client(const char* root, const char* host, int port) { |
| int status; |
| pid_t cli; |
| cli = fork(); |
| if (cli == 0) { |
| char* binary_path; |
| char* port_arg; |
| gpr_asprintf(&binary_path, "%s/interop_client", root); |
| gpr_asprintf(&port_arg, "--server_port=%d", port); |
| |
| execl(binary_path, binary_path, port_arg, NULL); |
| |
| gpr_free(binary_path); |
| gpr_free(port_arg); |
| return 1; |
| } |
| /* wait for client */ |
| gpr_log(GPR_INFO, "Waiting for client: %s", host); |
| if (waitpid(cli, &status, 0) == -1) return 2; |
| if (!WIFEXITED(status)) return 4; |
| if (WEXITSTATUS(status)) return WEXITSTATUS(status); |
| return 0; |
| } |
| |
| int main(int argc, char** argv) { |
| grpc::testing::InitTest(&argc, &argv, true); |
| char* me = argv[0]; |
| char* lslash = strrchr(me, '/'); |
| char root[1024]; |
| int port = grpc_pick_unused_port_or_die(); |
| int status; |
| pid_t svr; |
| int ret; |
| int do_ipv6 = 1; |
| /* seed rng with pid, so we don't end up with the same random numbers as a |
| concurrently running test binary */ |
| srand(getpid()); |
| if (!grpc_ipv6_loopback_available()) { |
| gpr_log(GPR_INFO, "Can't bind to ::1. Skipping IPv6 tests."); |
| do_ipv6 = 0; |
| } |
| /* figure out where we are */ |
| if (lslash) { |
| memcpy(root, me, lslash - me); |
| root[lslash - me] = 0; |
| } else { |
| strcpy(root, "."); |
| } |
| /* start the server */ |
| svr = fork(); |
| if (svr == 0) { |
| const size_t num_args = 3 + !FLAGS_extra_server_flags.empty(); |
| char** args = (char**)gpr_malloc(sizeof(char*) * num_args); |
| memset(args, 0, sizeof(char*) * num_args); |
| gpr_asprintf(&args[0], "%s/interop_server", root); |
| gpr_asprintf(&args[1], "--port=%d", port); |
| if (!FLAGS_extra_server_flags.empty()) { |
| args[2] = gpr_strdup(FLAGS_extra_server_flags.c_str()); |
| } |
| execv(args[0], args); |
| for (size_t i = 0; i < num_args - 1; ++i) { |
| gpr_free(args[i]); |
| } |
| gpr_free(args); |
| return 1; |
| } |
| /* wait a little */ |
| sleep(10); |
| /* start the clients */ |
| ret = test_client(root, "127.0.0.1", port); |
| if (ret != 0) return ret; |
| ret = test_client(root, "::ffff:127.0.0.1", port); |
| if (ret != 0) return ret; |
| ret = test_client(root, "localhost", port); |
| if (ret != 0) return ret; |
| if (do_ipv6) { |
| ret = test_client(root, "::1", port); |
| if (ret != 0) return ret; |
| } |
| /* wait for server */ |
| gpr_log(GPR_INFO, "Waiting for server"); |
| kill(svr, SIGINT); |
| if (waitpid(svr, &status, 0) == -1) return 2; |
| if (!WIFEXITED(status)) return 4; |
| if (WEXITSTATUS(status)) return WEXITSTATUS(status); |
| return 0; |
| } |