1#!/usr/bin/env ruby 2# 3# Copyright 2016 gRPC authors. 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16 17ENV['GRPC_ENABLE_FORK_SUPPORT'] = "1" 18fail "forking only supported on linux" unless RUBY_PLATFORM =~ /linux/ 19 20this_dir = File.expand_path(File.dirname(__FILE__)) 21protos_lib_dir = File.join(this_dir, 'lib') 22grpc_lib_dir = File.join(File.dirname(this_dir), 'lib') 23$LOAD_PATH.unshift(grpc_lib_dir) unless $LOAD_PATH.include?(grpc_lib_dir) 24$LOAD_PATH.unshift(protos_lib_dir) unless $LOAD_PATH.include?(protos_lib_dir) 25$LOAD_PATH.unshift(this_dir) unless $LOAD_PATH.include?(this_dir) 26 27require 'sanity_check_dlopen' 28require 'grpc' 29require 'end2end_common' 30 31def do_rpc(stub) 32 stub.echo(Echo::EchoRequest.new(request: 'hello'), deadline: Time.now + 1) 33rescue GRPC::Unavailable => e 34 STDERR.puts "RPC terminated with expected error: #{e}" 35rescue GRPC::DeadlineExceeded => e 36 STDERR.puts "RPC terminated with expected error: #{e}" 37end 38 39def expect_error_for(action) 40 STDERR.puts "#{action}: begin (pid=#{Process.pid})" 41 begin 42 yield 43 rescue RuntimeError => e 44 STDERR.puts "got (expected) error: #{e}" 45 STDERR.puts "#{action}: done (pid=#{Process.pid})" 46 return 47 end 48 fail "expected an exception due to: #{action}" 49end 50 51def main 52 # TODO(apolcyn): point this to a guaranteed-non-listening port 53 stub = Echo::EchoServer::Stub.new("localhost:443", :this_channel_is_insecure) 54 do_rpc(stub) 55 STDERR.puts "GRPC::pre_fork begin" 56 t = Thread.new do 57 expect_error_for("running prefork in a different thread than gRPC was initialized on") do 58 GRPC.prefork 59 end 60 end 61 t.join 62 expect_error_for("calling postfork_parent before prefork") { GRPC.postfork_parent } 63 expect_error_for("calling postfork_child before prefork") { GRPC.postfork_child } 64 with_logging("parent: GRPC.prefork") { GRPC.prefork } 65 expect_error_for("calling prefork twice") { GRPC.prefork } 66 expect_error_for("calling prefork twice") { GRPC.prefork } 67 expect_error_for("using gRPC after prefork") { do_rpc(stub) } 68 pid = fork do 69 expect_error_for("using gRPC before postfork_child") { do_rpc(stub) } 70 expect_error_for("calling postfork_parent from child") { GRPC.postfork_parent } 71 with_logging("child: GRPC.postfork_child") { GRPC.postfork_child } 72 expect_error_for("calling postfork_child twice") { GRPC.postfork_child } 73 with_logging("child: first post-fork RPC") { do_rpc(stub) } 74 with_logging("child: second post-fork RPC") { do_rpc(stub) } 75 STDERR.puts "child: done" 76 end 77 expect_error_for("using gRPC before postfork_parent") { do_rpc(stub) } 78 expect_error_for("calling postfork_child from parent") { GRPC.postfork_child } 79 t = Thread.new do 80 expect_error_for("running postfork_parent in a different thread than gRPC was initialized on") do 81 GRPC.postfork_parent 82 end 83 end 84 t.join 85 with_logging("parent: GRPC.postfork_parent") { GRPC.postfork_parent } 86 expect_error_for("calling postfork_parent twice") { GRPC.postfork_parent } 87 with_logging("parent: first post-fork RPC") { do_rpc(stub) } 88 with_logging("parent: second post-fork RPC") { do_rpc(stub) } 89 Process.wait pid 90 STDERR.puts "parent: done" 91end 92 93main 94