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 + 300) 33end 34 35def create_channel_creds 36 test_root = File.join(File.dirname(__FILE__), '..', 'spec', 'testdata') 37 files = ['ca.pem', 'client.key', 'client.pem'] 38 creds = files.map { |f| File.open(File.join(test_root, f)).read } 39 GRPC::Core::ChannelCredentials.new(creds[0], creds[1], creds[2]) 40end 41 42def client_cert 43 test_root = File.join(File.dirname(__FILE__), '..', 'spec', 'testdata') 44 cert = File.open(File.join(test_root, 'client.pem')).read 45 fail unless cert.is_a?(String) 46 cert 47end 48 49def run_client(stub) 50 do_rpc(stub) 51 with_logging("parent: GRPC.prefork") { GRPC.prefork } 52 pid = fork do 53 with_logging("child1: GRPC.postfork_child") { GRPC.postfork_child } 54 with_logging("child1: first post-fork RPC") { do_rpc(stub) } 55 with_logging("child1: GRPC.prefork") { GRPC.prefork } 56 pid2 = fork do 57 with_logging("child2: GRPC.postfork_child") { GRPC.postfork_child } 58 with_logging("child2: first post-fork RPC") { do_rpc(stub) } 59 with_logging("child2: second post-fork RPC") { do_rpc(stub) } 60 STDERR.puts "child2: done" 61 end 62 with_logging("child1: GRPC.postfork_parent") { GRPC.postfork_parent } 63 with_logging("child1: second post-fork RPC") { do_rpc(stub) } 64 Process.wait(pid2) 65 STDERR.puts "child1: done" 66 end 67 with_logging("parent: GRPC.postfork_parent") { GRPC.postfork_parent } 68 with_logging("parent: first post-fork RPC") { do_rpc(stub) } 69 with_logging("parent: second post-fork RPC") { do_rpc(stub) } 70 Process.wait pid 71 STDERR.puts "parent: done" 72end 73 74# This is designed to test fork support around three key things: 75# - call credentials (relies on a special background thread) 76# - channel credentials (ownership semantics are interesting for re-creating channels) 77# - channel args (ownership semantics are interesting for re-creating channels) 78def main 79 this_dir = File.expand_path(File.dirname(__FILE__)) 80 echo_server_path = File.join(this_dir, 'echo_server.rb') 81 to_child_r, _to_child_w = IO.pipe 82 to_parent_r, to_parent_w = IO.pipe 83 Process.spawn(RbConfig.ruby, echo_server_path, "--secure", in: to_child_r, out: to_parent_w, err: "server_log") 84 to_child_r.close 85 to_parent_w.close 86 child_port = to_parent_r.gets.strip 87 STDERR.puts "server running on port: #{child_port}" 88 channel_creds = create_channel_creds.compose( 89 GRPC::Core::CallCredentials.new(proc do |args| 90 { 'authorization' => 'test' }.merge(args) 91 end)) 92 stub = Echo::EchoServer::Stub.new( 93 "localhost:#{child_port}", channel_creds, 94 channel_args: { GRPC::Core::Channel::SSL_TARGET => 'foo.test.google.fr' }) 95 2.times do 96 run_client(stub) 97 end 98end 99 100main 101