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 hang 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  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
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    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
54  # this should break out with an exception once the channel is closed
55  loop do
56    begin
57      state = ch.connectivity_state(true)
58      ch.watch_connectivity_state(state, Time.now + 360)
59    rescue RuntimeError => e
60      STDERR.puts "(expected) error occurred: #{e.inspect}"
61      break
62    end
63  end
64
65  srv.stop
66  thd.join
67end
68
69main
70