Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 1 | #!/usr/bin/env ruby |
| 2 | |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 3 | # Copyright 2015 gRPC authors. |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 4 | # |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 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 |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 8 | # |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 9 | # http://www.apache.org/licenses/LICENSE-2.0 |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 10 | # |
Jan Tattermusch | 7897ae9 | 2017-06-07 22:57:36 +0200 | [diff] [blame] | 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. |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 16 | |
| 17 | require_relative './end2end_common' |
| 18 | |
Alexander Polcyn | 4e60675 | 2017-03-19 23:32:54 -0700 | [diff] [blame] | 19 | # Calls '#close' on a Channel when "shutdown" called. This tries to |
| 20 | # trigger a hang or crash bug by closing a channel actively being watched |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 21 | class ChannelClosingClientController < ClientControl::ClientController::Service |
| 22 | def initialize(ch) |
| 23 | @ch = ch |
| 24 | end |
Alexander Polcyn | 4e60675 | 2017-03-19 23:32:54 -0700 | [diff] [blame] | 25 | |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 26 | def shutdown(_, _) |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 27 | @ch.close |
Alexander Polcyn | 563ec53 | 2017-03-15 15:54:49 -0700 | [diff] [blame] | 28 | ClientControl::Void.new |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 29 | end |
| 30 | end |
| 31 | |
| 32 | def main |
| 33 | client_control_port = '' |
| 34 | server_port = '' |
| 35 | OptionParser.new do |opts| |
| 36 | opts.on('--client_control_port=P', String) do |p| |
| 37 | client_control_port = p |
| 38 | end |
| 39 | opts.on('--server_port=P', String) do |p| |
| 40 | server_port = p |
| 41 | end |
| 42 | end.parse! |
| 43 | |
Alexander Polcyn | 4e60675 | 2017-03-19 23:32:54 -0700 | [diff] [blame] | 44 | ch = GRPC::Core::Channel.new("localhost:#{server_port}", {}, |
| 45 | :this_channel_is_insecure) |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 46 | |
| 47 | srv = GRPC::RpcServer.new |
| 48 | thd = Thread.new do |
| 49 | srv.add_http2_port("0.0.0.0:#{client_control_port}", :this_port_is_insecure) |
| 50 | srv.handle(ChannelClosingClientController.new(ch)) |
| 51 | srv.run |
| 52 | end |
| 53 | |
Alexander Polcyn | 563ec53 | 2017-03-15 15:54:49 -0700 | [diff] [blame] | 54 | # this should break out with an exception once the channel is closed |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 55 | loop do |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 56 | begin |
Alexander Polcyn | 563ec53 | 2017-03-15 15:54:49 -0700 | [diff] [blame] | 57 | state = ch.connectivity_state(true) |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 58 | ch.watch_connectivity_state(state, Time.now + 360) |
Alexander Polcyn | 563ec53 | 2017-03-15 15:54:49 -0700 | [diff] [blame] | 59 | rescue RuntimeError => e |
| 60 | STDERR.puts "(expected) error occurred: #{e.inspect}" |
Alexander Polcyn | 70bc492 | 2017-03-15 14:00:46 -0700 | [diff] [blame] | 61 | break |
| 62 | end |
| 63 | end |
| 64 | |
| 65 | srv.stop |
| 66 | thd.join |
| 67 | end |
| 68 | |
| 69 | main |