1#!/usr/bin/env ruby 2 3# Copyright 2015 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 17require_relative './end2end_common' 18 19# Calls '#close' on a Channel when "shutdown" called. This tries to 20# trigger a freeze or crash bug by closing a channel actively being watched 21class ChannelClosingClientController < ClientControl::ClientController::Service 22 def initialize(ch) 23 @ch = ch 24 end 25 26 def shutdown(_, _) 27 @ch.close 28 ClientControl::Void.new 29 end 30end 31 32def main 33 parent_controller_port = '' 34 server_port = '' 35 OptionParser.new do |opts| 36 opts.on('--parent_controller_port=P', String) do |p| 37 parent_controller_port = p 38 end 39 opts.on('--server_port=P', String) do |p| 40 server_port = p 41 end 42 end.parse! 43 44 ch = GRPC::Core::Channel.new("localhost:#{server_port}", {}, 45 :this_channel_is_insecure) 46 47 srv = new_rpc_server_for_testing 48 thd = Thread.new do 49 port = srv.add_http2_port('localhost:0', :this_port_is_insecure) 50 report_controller_port_to_parent(parent_controller_port, port) 51 srv.handle(ChannelClosingClientController.new(ch)) 52 srv.run 53 end 54 55 # this should break out with an exception once the channel is closed 56 loop do 57 begin 58 state = ch.connectivity_state(true) 59 ch.watch_connectivity_state(state, Time.now + 360) 60 rescue RuntimeError => e 61 STDERR.puts "(expected) error occurred: #{e.inspect}" 62 break 63 end 64 end 65 66 srv.stop 67 thd.join 68end 69 70main 71