hirura/hrr_rb_ssh

module HrrRbSsh
  class Connection
    class Channel
      class ChannelType
        class ForwardedTcpip < ChannelType
          include Loggable

          NAME = 'forwarded-tcpip'

          def initialize connection, channel, message, socket, logger: nil
            self.logger = logger
            @connection = connection
            @channel = channel
            @socket = socket
          end

          def start
            @sender_thread = sender_thread
            @receiver_thread = receiver_thread
          end

          def close
            begin
              if @sender_thread_finished && @receiver_thread_finished
                log_info { "closing forwarded-tcpip" }
                @socket.close
                log_info { "closing channel IOs" }
                @channel.io.each{ |io| io.close rescue nil }
                log_info { "channel IOs closed" }
                @channel.close from=:channel_type_instance
                log_info { "forwarded-tcpip closed" }
              end
            rescue => e
              log_error { [e.backtrace[0], ": ", e.message, " (", e.class.to_s, ")\n\t", e.backtrace[1..-1].join("\n\t")].join }
            end
          end

          def sender_thread
            Thread.new(@socket){ |s|
              begin
                loop do
                  begin
                    @channel.io[1].write s.readpartial(10240)
                  rescue EOFError
                    log_info { "socket is EOF" }
                    @channel.io[1].close rescue nil
                    break
                  rescue IOError
                    log_info { "socket is closed" }
                    @channel.io[1].close rescue nil
                    break
                  rescue => e
                    log_error { [e.backtrace[0], ": ", e.message, " (", e.class.to_s, ")\n\t", e.backtrace[1..-1].join("\n\t")].join }
                    @channel.io[1].close rescue nil
                    break
                  end
                end
                log_info { "finishing sender thread" }
                @sender_thread_finished = true
                close
              ensure
                log_info { "sender thread finished" }
              end
            }
          end

          def receiver_thread
            Thread.new(@socket){ |s|
              begin
                loop do
                  begin
                    s.write @channel.io[0].readpartial(10240)
                  rescue EOFError
                    log_info { "io is EOF" }
                    s.close_write
                    break
                  rescue IOError
                    log_info { "socket is closed" }
                    break
                  rescue => e
                    log_error { [e.backtrace[0], ": ", e.message, " (", e.class.to_s, ")\n\t", e.backtrace[1..-1].join("\n\t")].join }
                    s.close_write
                    break
                  end
                end
                log_info { "finishing receiver thread" }
                @receiver_thread_finished = true
                close
              ensure
                log_info { "receiver thread finished" }
              end
            }
          end
        end
      end
    end
  end
end