/* * * 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 #include #include #include #include #include #include #include #include #include #include #include #include #include "test/core/util/port.h" #include "test/cpp/util/test_config.h" #include "src/core/lib/iomgr/socket_utils_posix.h" #include "src/core/lib/support/string.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; }