Class: Http2::Connection

Inherits:
Object
  • Object
show all
Defined in:
lib/http2/connection.rb

Instance Method Summary collapse

Constructor Details

#initialize(http2) ⇒ Connection

Returns a new instance of Connection.



2
3
4
5
6
7
8
 
# File 'lib/http2/connection.rb', line 2

def initialize(http2)
  @http2 = http2
  @debug = http2.debug
  @args = http2.args
  @nl = http2.nl
  reconnect
end

Instance Method Details

#apply_ssl ⇒ Object 



148
149
150
151
152
153
154
155
156
157
158
159
 
# File 'lib/http2/connection.rb', line 148

def apply_ssl
  puts "Http2: Initializing SSL." if @debug
  require "openssl" unless ::Kernel.const_defined?(:OpenSSL)

  ssl_context = OpenSSL::SSL::SSLContext.new
  ssl_context.verify_mode = OpenSSL::SSL::VERIFY_PEER unless @args[:ssl_skip_verify]
  @sock_ssl = OpenSSL::SSL::SSLSocket.new(@sock_plain, ssl_context)
  @sock_ssl.sync_close = true
  @sock_ssl.connect

  @sock = @sock_ssl
end

#close ⇒ Object

Closes the current connection if any.



102
103
104
105
106
 
# File 'lib/http2/connection.rb', line 102

def close
  @sock.close if @sock && !@sock.closed?
  @sock_ssl.close if @sock_ssl && !@sock_ssl.closed?
  @sock_plain.close if @sock_plain && !@sock_plain.closed?
end

#closed? ⇒ Boolean

Returns:



29
30
31
 
# File 'lib/http2/connection.rb', line 29

def closed?
  @sock.closed?
end

#connect_proxy ⇒ Object 



143
144
145
146
 
# File 'lib/http2/connection.rb', line 143

def connect_proxy
  puts "Http2: Opening socket connection to '#{@args[:host]}:#{@args[:port]}' through proxy '#{@args[:proxy][:host]}:#{@args[:proxy][:port]}'." if @debug
  @sock_plain = TCPSocket.new(@args[:proxy][:host], @args[:proxy][:port].to_i)
end

#connect_proxy_connect ⇒ Object 



108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/http2/connection.rb', line 108

def connect_proxy_connect
  puts "Http2: Initializing proxy connect to '#{@args[:host]}:#{@args[:port]}' through proxy '#{@args[:proxy][:host]}:#{@args[:proxy][:port]}'." if @debug
  @sock_plain = TCPSocket.new(@args[:proxy][:host], @args[:proxy][:port])

  connect = "CONNECT #{@args[:host]}:#{@args[:port]} HTTP/1.1#{@nl}"
  puts "Http2: Sending connect: #{connect}" if @debug
  @sock_plain.write(connect)

  headers = {
    "Host" => "#{@args[:host]}:#{@args[:port]}"
  }

  if @args[:proxy][:user] && @args[:proxy][:passwd]
    headers["Proxy-Authorization"] = "Basic #{["#{@args[:proxy][:user]}:#{@args[:proxy][:passwd]}"].pack("m").chomp}"
  end

  headers.each do |key, value|
    header = "#{key}: #{value}"
    puts "Http2: Sending header to proxy: #{header}" if @debug
    @sock_plain.write("#{header}#{@nl}")
  end

  @sock_plain.write(@nl)

  res = @sock_plain.gets.to_s
  raise "Couldn't connect through proxy: #{res}" unless res.match(/^http\/1\.(0|1)\s+200/i)
  @sock_plain.gets

  @proxy_connect = true
end

#destroy ⇒ Object 



10
11
12
13
14
15
16
17
18
19
 
# File 'lib/http2/connection.rb', line 10

def destroy
  @sock.close if @sock && !@sock.closed?
  @sock = nil

  @sock_plain.close if @sock_plain && !@sock_plain.closed?
  @sock_plain = nil

  @sock_ssl.close if @sock_ssl && !@sock_ssl.closed?
  @sock_ssl = nil
end

#gets ⇒ Object 



21
22
23
 
# File 'lib/http2/connection.rb', line 21

def gets
  @sock.gets
end

#proxy_connect? ⇒ Boolean

Returns:



139
140
141
 
# File 'lib/http2/connection.rb', line 139

def proxy_connect?
  @proxy_connect
end

#read(length) ⇒ Object 



25
26
27
 
# File 'lib/http2/connection.rb', line 25

def read(length)
  @sock.read(length)
end

#reconnect ⇒ Object

Reconnects to the host.



62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
 
# File 'lib/http2/connection.rb', line 62

def reconnect
  puts "Http2: Reconnect." if @debug

  # Open connection.
  if @args[:proxy]
    if @args[:proxy][:connect]
      connect_proxy_connect
    else
      connect_proxy
    end
  else
    puts "Http2: Opening socket connection to '#{@http2.host}:#{@http2.port}'." if @debug
    @sock_plain = TCPSocket.new(@http2.host, @http2.port)
  end

  if @args[:ssl]
    apply_ssl
  else
    @sock = @sock_plain
  end
end

#sock_puts(str) ⇒ Object 



40
41
42
 
# File 'lib/http2/connection.rb', line 40

def sock_puts(str)
  sock_write("#{str}#{@nl}")
end

#sock_write(str) ⇒ Object 



33
34
35
36
37
38
 
# File 'lib/http2/connection.rb', line 33

def sock_write(str)
  str = str.to_s
  return if str.empty?
  count = @sock.write(str)
  raise "Couldnt write to socket: '#{count}', '#{str}'." if count <= 0
end

#socket_working? ⇒ Boolean

Returns boolean based on the if the object is connected and the socket is working.

Examples

puts “Socket is working.” if http.socket_working?

Returns:



87
88
89
90
91
92
93
94
95
96
97
98
99
 
# File 'lib/http2/connection.rb', line 87

def socket_working?
  return false if !@sock || @sock.closed?

  if @keepalive_timeout && @request_last
    between = Time.now.to_i - @request_last.to_i
    if between >= @keepalive_timeout
      puts "Http2: We are over the keepalive-wait - returning false for socket_working?." if @debug
      return false
    end
  end

  true
end

#write(str) ⇒ Object

Tries to write a string to the socket. If it fails it reconnects and tries again.



45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
 
# File 'lib/http2/connection.rb', line 45

def write(str)
  reconnect unless socket_working?

  puts "Http2: Writing: #{str}" if @debug

  begin
    raise Errno::EPIPE, "The socket is closed." if !@sock || @sock.closed?
    sock_write(str)
  rescue Errno::EPIPE # this can also be thrown by puts.
    reconnect
    sock_write(str)
  end

  @request_last = Time.now
end