net-ssh 2.9.2 → 4.0.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (138) hide show
  1. checksums.yaml +4 -4
  2. checksums.yaml.gz.sig +0 -0
  3. data/.gitignore +6 -0
  4. data/.rubocop.yml +5 -0
  5. data/.rubocop_todo.yml +1129 -0
  6. data/.travis.yml +41 -5
  7. data/CHANGES.txt +133 -1
  8. data/Gemfile +13 -0
  9. data/Gemfile.norbnacl +10 -0
  10. data/Gemfile.norbnacl.lock +41 -0
  11. data/ISSUE_TEMPLATE.md +30 -0
  12. data/README.rdoc +26 -81
  13. data/Rakefile +63 -45
  14. data/appveyor.yml +51 -0
  15. data/lib/net/ssh/authentication/agent.rb +174 -14
  16. data/lib/net/ssh/authentication/ed25519.rb +137 -0
  17. data/lib/net/ssh/authentication/ed25519_loader.rb +21 -0
  18. data/lib/net/ssh/authentication/key_manager.rb +36 -30
  19. data/lib/net/ssh/authentication/methods/abstract.rb +4 -0
  20. data/lib/net/ssh/authentication/methods/keyboard_interactive.rb +16 -9
  21. data/lib/net/ssh/authentication/methods/password.rb +17 -4
  22. data/lib/net/ssh/authentication/pageant.rb +166 -45
  23. data/lib/net/ssh/authentication/session.rb +3 -2
  24. data/lib/net/ssh/buffer.rb +49 -10
  25. data/lib/net/ssh/buffered_io.rb +17 -12
  26. data/lib/net/ssh/config.rb +39 -8
  27. data/lib/net/ssh/connection/channel.rb +42 -20
  28. data/lib/net/ssh/connection/event_loop.rb +114 -0
  29. data/lib/net/ssh/connection/keepalive.rb +2 -2
  30. data/lib/net/ssh/connection/session.rb +120 -34
  31. data/lib/net/ssh/errors.rb +6 -6
  32. data/lib/net/ssh/key_factory.rb +49 -43
  33. data/lib/net/ssh/known_hosts.rb +49 -3
  34. data/lib/net/ssh/prompt.rb +47 -78
  35. data/lib/net/ssh/proxy/command.rb +31 -5
  36. data/lib/net/ssh/proxy/http.rb +15 -11
  37. data/lib/net/ssh/proxy/https.rb +49 -0
  38. data/lib/net/ssh/proxy/socks4.rb +2 -1
  39. data/lib/net/ssh/proxy/socks5.rb +3 -2
  40. data/lib/net/ssh/ruby_compat.rb +2 -29
  41. data/lib/net/ssh/service/forward.rb +2 -2
  42. data/lib/net/ssh/test/channel.rb +7 -0
  43. data/lib/net/ssh/test/extensions.rb +17 -0
  44. data/lib/net/ssh/test/kex.rb +4 -4
  45. data/lib/net/ssh/test/packet.rb +18 -2
  46. data/lib/net/ssh/test/script.rb +16 -2
  47. data/lib/net/ssh/test/socket.rb +1 -1
  48. data/lib/net/ssh/test.rb +5 -5
  49. data/lib/net/ssh/transport/algorithms.rb +92 -75
  50. data/lib/net/ssh/transport/cipher_factory.rb +19 -26
  51. data/lib/net/ssh/transport/ctr.rb +7 -9
  52. data/lib/net/ssh/transport/kex/diffie_hellman_group1_sha1.rb +20 -9
  53. data/lib/net/ssh/transport/kex/diffie_hellman_group_exchange_sha1.rb +5 -3
  54. data/lib/net/ssh/transport/kex/ecdh_sha2_nistp256.rb +1 -1
  55. data/lib/net/ssh/transport/key_expander.rb +1 -0
  56. data/lib/net/ssh/transport/openssl.rb +1 -1
  57. data/lib/net/ssh/transport/packet_stream.rb +11 -3
  58. data/lib/net/ssh/transport/server_version.rb +13 -6
  59. data/lib/net/ssh/transport/session.rb +20 -10
  60. data/lib/net/ssh/transport/state.rb +1 -1
  61. data/lib/net/ssh/verifiers/secure.rb +8 -10
  62. data/lib/net/ssh/version.rb +4 -4
  63. data/lib/net/ssh.rb +62 -14
  64. data/net-ssh-public_cert.pem +19 -18
  65. data/net-ssh.gemspec +34 -194
  66. data/support/arcfour_check.rb +1 -1
  67. data/support/ssh_tunnel_bug.rb +1 -1
  68. data.tar.gz.sig +0 -0
  69. metadata +125 -109
  70. metadata.gz.sig +0 -0
  71. data/Rudyfile +0 -96
  72. data/lib/net/ssh/authentication/agent/java_pageant.rb +0 -85
  73. data/lib/net/ssh/authentication/agent/socket.rb +0 -178
  74. data/setup.rb +0 -1585
  75. data/test/README.txt +0 -47
  76. data/test/authentication/methods/common.rb +0 -28
  77. data/test/authentication/methods/test_abstract.rb +0 -51
  78. data/test/authentication/methods/test_hostbased.rb +0 -114
  79. data/test/authentication/methods/test_keyboard_interactive.rb +0 -100
  80. data/test/authentication/methods/test_none.rb +0 -41
  81. data/test/authentication/methods/test_password.rb +0 -95
  82. data/test/authentication/methods/test_publickey.rb +0 -148
  83. data/test/authentication/test_agent.rb +0 -224
  84. data/test/authentication/test_key_manager.rb +0 -227
  85. data/test/authentication/test_session.rb +0 -107
  86. data/test/common.rb +0 -108
  87. data/test/configs/auth_off +0 -5
  88. data/test/configs/auth_on +0 -4
  89. data/test/configs/empty +0 -0
  90. data/test/configs/eqsign +0 -3
  91. data/test/configs/exact_match +0 -8
  92. data/test/configs/host_plus +0 -10
  93. data/test/configs/multihost +0 -4
  94. data/test/configs/negative_match +0 -6
  95. data/test/configs/nohost +0 -19
  96. data/test/configs/numeric_host +0 -4
  97. data/test/configs/send_env +0 -2
  98. data/test/configs/substitutes +0 -8
  99. data/test/configs/wild_cards +0 -14
  100. data/test/connection/test_channel.rb +0 -467
  101. data/test/connection/test_session.rb +0 -543
  102. data/test/known_hosts/github +0 -1
  103. data/test/manual/test_forward.rb +0 -285
  104. data/test/manual/test_pageant.rb +0 -37
  105. data/test/start/test_connection.rb +0 -53
  106. data/test/start/test_options.rb +0 -43
  107. data/test/start/test_transport.rb +0 -28
  108. data/test/test_all.rb +0 -11
  109. data/test/test_buffer.rb +0 -433
  110. data/test/test_buffered_io.rb +0 -63
  111. data/test/test_config.rb +0 -221
  112. data/test/test_key_factory.rb +0 -191
  113. data/test/test_known_hosts.rb +0 -13
  114. data/test/transport/hmac/test_md5.rb +0 -41
  115. data/test/transport/hmac/test_md5_96.rb +0 -27
  116. data/test/transport/hmac/test_none.rb +0 -34
  117. data/test/transport/hmac/test_ripemd160.rb +0 -36
  118. data/test/transport/hmac/test_sha1.rb +0 -36
  119. data/test/transport/hmac/test_sha1_96.rb +0 -27
  120. data/test/transport/hmac/test_sha2_256.rb +0 -37
  121. data/test/transport/hmac/test_sha2_256_96.rb +0 -27
  122. data/test/transport/hmac/test_sha2_512.rb +0 -37
  123. data/test/transport/hmac/test_sha2_512_96.rb +0 -27
  124. data/test/transport/kex/test_diffie_hellman_group14_sha1.rb +0 -13
  125. data/test/transport/kex/test_diffie_hellman_group1_sha1.rb +0 -146
  126. data/test/transport/kex/test_diffie_hellman_group_exchange_sha1.rb +0 -92
  127. data/test/transport/kex/test_diffie_hellman_group_exchange_sha256.rb +0 -34
  128. data/test/transport/kex/test_ecdh_sha2_nistp256.rb +0 -161
  129. data/test/transport/kex/test_ecdh_sha2_nistp384.rb +0 -38
  130. data/test/transport/kex/test_ecdh_sha2_nistp521.rb +0 -38
  131. data/test/transport/test_algorithms.rb +0 -324
  132. data/test/transport/test_cipher_factory.rb +0 -443
  133. data/test/transport/test_hmac.rb +0 -34
  134. data/test/transport/test_identity_cipher.rb +0 -40
  135. data/test/transport/test_packet_stream.rb +0 -1761
  136. data/test/transport/test_server_version.rb +0 -78
  137. data/test/transport/test_session.rb +0 -331
  138. data/test/transport/test_state.rb +0 -181
@@ -1,543 +0,0 @@
1
- require 'common'
2
- require 'net/ssh/connection/session'
3
-
4
- module Connection
5
-
6
- class TestSession < Test::Unit::TestCase
7
- include Net::SSH::Connection::Constants
8
-
9
- def test_constructor_should_set_defaults
10
- assert session.channels.empty?
11
- assert session.pending_requests.empty?
12
- assert_equal({ socket => nil }, session.listeners)
13
- end
14
-
15
- def test_on_open_channel_should_register_block_with_given_channel_type
16
- flag = false
17
- session.on_open_channel("testing") { flag = true }
18
- assert_not_nil session.channel_open_handlers["testing"]
19
- session.channel_open_handlers["testing"].call
20
- assert flag, "callback should have been invoked"
21
- end
22
-
23
- def test_forward_should_create_and_cache_instance_of_forward_service
24
- assert_instance_of Net::SSH::Service::Forward, session.forward
25
- assert_equal session.forward.object_id, session.forward.object_id
26
- end
27
-
28
- def test_listen_to_without_callback_should_add_argument_as_listener
29
- io = stub("io")
30
- session.listen_to(io)
31
- assert session.listeners.key?(io)
32
- assert_nil session.listeners[io]
33
- end
34
-
35
- def test_listen_to_should_add_argument_to_listeners_list_if_block_is_given
36
- io = stub("io", :pending_write? => true)
37
- flag = false
38
- session.listen_to(io) { flag = true }
39
- assert !flag, "callback should not be invoked immediately"
40
- assert session.listeners.key?(io)
41
- session.listeners[io].call
42
- assert flag, "callback should have been invoked"
43
- end
44
-
45
- def test_stop_listening_to_should_remove_argument_from_listeners
46
- io = stub("io", :pending_write? => true)
47
-
48
- session.listen_to(io)
49
- assert session.listeners.key?(io)
50
-
51
- session.stop_listening_to(io)
52
- assert !session.listeners.key?(io)
53
- end
54
-
55
- def test_send_message_should_enqueue_message_at_transport_layer
56
- packet = P(:byte, REQUEST_SUCCESS)
57
- session.send_message(packet)
58
- assert_equal packet.to_s, socket.write_buffer
59
- end
60
-
61
- def test_open_channel_defaults_should_use_session_channel
62
- flag = false
63
- channel = session.open_channel { flag = true }
64
- assert !flag, "callback should not be invoked immediately"
65
- channel.do_open_confirmation(1,2,3)
66
- assert flag, "callback should have been invoked"
67
- assert_equal "session", channel.type
68
- assert_equal 0, channel.local_id
69
- assert_equal channel, session.channels[channel.local_id]
70
-
71
- packet = P(:byte, CHANNEL_OPEN, :string, "session", :long, channel.local_id,
72
- :long, channel.local_maximum_window_size, :long, channel.local_maximum_packet_size)
73
- assert_equal packet.to_s, socket.write_buffer
74
- end
75
-
76
- def test_open_channel_with_type_should_use_type
77
- channel = session.open_channel("direct-tcpip")
78
- assert_equal "direct-tcpip", channel.type
79
- packet = P(:byte, CHANNEL_OPEN, :string, "direct-tcpip", :long, channel.local_id,
80
- :long, channel.local_maximum_window_size, :long, channel.local_maximum_packet_size)
81
- assert_equal packet.to_s, socket.write_buffer
82
- end
83
-
84
- def test_open_channel_with_extras_should_append_extras_to_packet
85
- channel = session.open_channel("direct-tcpip", :string, "other.host", :long, 1234)
86
- packet = P(:byte, CHANNEL_OPEN, :string, "direct-tcpip", :long, channel.local_id,
87
- :long, channel.local_maximum_window_size, :long, channel.local_maximum_packet_size,
88
- :string, "other.host", :long, 1234)
89
- assert_equal packet.to_s, socket.write_buffer
90
- end
91
-
92
- def test_send_global_request_without_callback_should_not_expect_reply
93
- packet = P(:byte, GLOBAL_REQUEST, :string, "testing", :bool, false)
94
- session.send_global_request("testing")
95
- assert_equal packet.to_s, socket.write_buffer
96
- assert session.pending_requests.empty?
97
- end
98
-
99
- def test_send_global_request_with_callback_should_expect_reply
100
- packet = P(:byte, GLOBAL_REQUEST, :string, "testing", :bool, true)
101
- proc = Proc.new {}
102
- session.send_global_request("testing", &proc)
103
- assert_equal packet.to_s, socket.write_buffer
104
- assert_equal [proc], session.pending_requests
105
- end
106
-
107
- def test_send_global_request_with_extras_should_append_extras_to_packet
108
- packet = P(:byte, GLOBAL_REQUEST, :string, "testing", :bool, false, :string, "other.host", :long, 1234)
109
- session.send_global_request("testing", :string, "other.host", :long, 1234)
110
- assert_equal packet.to_s, socket.write_buffer
111
- end
112
-
113
- def test_process_should_exit_immediately_if_block_is_false
114
- session.channels[0] = stub("channel", :closing? => false)
115
- session.channels[0].expects(:process).never
116
- process_times(0)
117
- end
118
-
119
- def test_process_should_exit_after_processing_if_block_is_true_then_false
120
- session.channels[0] = stub("channel", :closing? => false)
121
- session.channels[0].expects(:process)
122
- IO.expects(:select).never
123
- process_times(2)
124
- end
125
-
126
- def test_process_should_not_process_channels_that_are_closing
127
- session.channels[0] = stub("channel", :closing? => true)
128
- session.channels[0].expects(:process).never
129
- IO.expects(:select).never
130
- process_times(2)
131
- end
132
-
133
- def test_global_request_packets_should_be_silently_handled_if_no_handler_exists_for_them
134
- transport.return(GLOBAL_REQUEST, :string, "testing", :bool, false)
135
- process_times(2)
136
- assert transport.queue.empty?
137
- assert !socket.pending_write?
138
- end
139
-
140
- def test_global_request_packets_should_be_auto_replied_to_even_if_no_handler_exists
141
- transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
142
- process_times(2)
143
- assert_equal P(:byte, REQUEST_FAILURE).to_s, socket.write_buffer
144
- end
145
-
146
- def test_global_request_handler_should_not_trigger_auto_reply_if_no_reply_is_wanted
147
- flag = false
148
- session.on_global_request("testing") { flag = true }
149
- assert !flag, "callback should not be invoked yet"
150
- transport.return(GLOBAL_REQUEST, :string, "testing", :bool, false)
151
- process_times(2)
152
- assert transport.queue.empty?
153
- assert !socket.pending_write?
154
- assert flag, "callback should have been invoked"
155
- end
156
-
157
- def test_global_request_handler_returning_true_should_trigger_success_auto_reply
158
- flag = false
159
- session.on_global_request("testing") { flag = true }
160
- transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
161
- process_times(2)
162
- assert_equal P(:byte, REQUEST_SUCCESS).to_s, socket.write_buffer
163
- assert flag
164
- end
165
-
166
- def test_global_request_handler_returning_false_should_trigger_failure_auto_reply
167
- flag = false
168
- session.on_global_request("testing") { flag = true; false }
169
- transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
170
- process_times(2)
171
- assert_equal P(:byte, REQUEST_FAILURE).to_s, socket.write_buffer
172
- assert flag
173
- end
174
-
175
- def test_global_request_handler_returning_sent_should_not_trigger_auto_reply
176
- flag = false
177
- session.on_global_request("testing") { flag = true; :sent }
178
- transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
179
- process_times(2)
180
- assert !socket.pending_write?
181
- assert flag
182
- end
183
-
184
- def test_global_request_handler_returning_other_value_should_raise_error
185
- session.on_global_request("testing") { "bug" }
186
- transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
187
- assert_raises(RuntimeError) { process_times(2) }
188
- end
189
-
190
- def test_request_success_packets_should_invoke_next_pending_request_with_true
191
- result = nil
192
- session.pending_requests << Proc.new { |*args| result = args }
193
- transport.return(REQUEST_SUCCESS)
194
- process_times(2)
195
- assert_equal [true, P(:byte, REQUEST_SUCCESS)], result
196
- assert session.pending_requests.empty?
197
- end
198
-
199
- def test_request_failure_packets_should_invoke_next_pending_request_with_false
200
- result = nil
201
- session.pending_requests << Proc.new { |*args| result = args }
202
- transport.return(REQUEST_FAILURE)
203
- process_times(2)
204
- assert_equal [false, P(:byte, REQUEST_FAILURE)], result
205
- assert session.pending_requests.empty?
206
- end
207
-
208
- def test_channel_open_packet_without_corresponding_channel_open_handler_should_result_in_channel_open_failure
209
- transport.return(CHANNEL_OPEN, :string, "auth-agent", :long, 14, :long, 0x20000, :long, 0x10000)
210
- process_times(2)
211
- assert_equal P(:byte, CHANNEL_OPEN_FAILURE, :long, 14, :long, 3, :string, "unknown channel type auth-agent", :string, "").to_s, socket.write_buffer
212
- end
213
-
214
- def test_channel_open_packet_with_corresponding_handler_should_result_in_channel_open_failure_when_handler_returns_an_error
215
- transport.return(CHANNEL_OPEN, :string, "auth-agent", :long, 14, :long, 0x20000, :long, 0x10000)
216
- session.on_open_channel "auth-agent" do |s, ch, p|
217
- raise Net::SSH::ChannelOpenFailed.new(1234, "we iz in ur channelz!")
218
- end
219
- process_times(2)
220
- assert_equal P(:byte, CHANNEL_OPEN_FAILURE, :long, 14, :long, 1234, :string, "we iz in ur channelz!", :string, "").to_s, socket.write_buffer
221
- end
222
-
223
- def test_channel_open_packet_with_corresponding_handler_should_result_in_channel_open_confirmation_when_handler_succeeds
224
- transport.return(CHANNEL_OPEN, :string, "auth-agent", :long, 14, :long, 0x20001, :long, 0x10001)
225
- result = nil
226
- session.on_open_channel("auth-agent") { |*args| result = args }
227
- process_times(2)
228
- assert_equal P(:byte, CHANNEL_OPEN_CONFIRMATION, :long, 14, :long, 0, :long, 0x20000, :long, 0x8000).to_s, socket.write_buffer
229
- assert_not_nil(ch = session.channels[0])
230
- assert_equal [session, ch, P(:byte, CHANNEL_OPEN, :string, "auth-agent", :long, 14, :long, 0x20001, :long, 0x10001)], result
231
- assert_equal 0, ch.local_id
232
- assert_equal 14, ch.remote_id
233
- assert_equal 0x20001, ch.remote_maximum_window_size
234
- assert_equal 0x10001, ch.remote_maximum_packet_size
235
- assert_equal 0x20000, ch.local_maximum_window_size
236
- assert_equal 0x8000, ch.local_maximum_packet_size
237
- assert_equal "auth-agent", ch.type
238
- end
239
-
240
- def test_channel_open_failure_should_remove_channel_and_tell_channel_that_open_failed
241
- session.channels[1] = stub("channel")
242
- session.channels[1].expects(:do_open_failed).with(1234, "some reason")
243
- transport.return(CHANNEL_OPEN_FAILURE, :long, 1, :long, 1234, :string, "some reason", :string, "lang tag")
244
- process_times(2)
245
- assert session.channels.empty?
246
- end
247
-
248
- def test_channel_open_confirmation_packet_should_be_routed_to_corresponding_channel
249
- channel_at(14).expects(:do_open_confirmation).with(1234, 0x20001, 0x10001)
250
- transport.return(CHANNEL_OPEN_CONFIRMATION, :long, 14, :long, 1234, :long, 0x20001, :long, 0x10001)
251
- process_times(2)
252
- end
253
-
254
- def test_channel_window_adjust_packet_should_be_routed_to_corresponding_channel
255
- channel_at(14).expects(:do_window_adjust).with(5000)
256
- transport.return(CHANNEL_WINDOW_ADJUST, :long, 14, :long, 5000)
257
- process_times(2)
258
- end
259
-
260
- def test_channel_request_for_nonexistant_channel_should_be_ignored
261
- transport.return(CHANNEL_REQUEST, :long, 14, :string, "testing", :bool, false)
262
- assert_nothing_raised { process_times(2) }
263
- end
264
-
265
- def test_channel_request_packet_should_be_routed_to_corresponding_channel
266
- channel_at(14).expects(:do_request).with("testing", false, Net::SSH::Buffer.new)
267
- transport.return(CHANNEL_REQUEST, :long, 14, :string, "testing", :bool, false)
268
- process_times(2)
269
- end
270
-
271
- def test_channel_data_packet_should_be_routed_to_corresponding_channel
272
- channel_at(14).expects(:do_data).with("bring it on down")
273
- transport.return(CHANNEL_DATA, :long, 14, :string, "bring it on down")
274
- process_times(2)
275
- end
276
-
277
- def test_channel_extended_data_packet_should_be_routed_to_corresponding_channel
278
- channel_at(14).expects(:do_extended_data).with(1, "bring it on down")
279
- transport.return(CHANNEL_EXTENDED_DATA, :long, 14, :long, 1, :string, "bring it on down")
280
- process_times(2)
281
- end
282
-
283
- def test_channel_eof_packet_should_be_routed_to_corresponding_channel
284
- channel_at(14).expects(:do_eof).with()
285
- transport.return(CHANNEL_EOF, :long, 14)
286
- process_times(2)
287
- end
288
-
289
- def test_channel_success_packet_should_be_routed_to_corresponding_channel
290
- channel_at(14).expects(:do_success).with()
291
- transport.return(CHANNEL_SUCCESS, :long, 14)
292
- process_times(2)
293
- end
294
-
295
- def test_channel_failure_packet_should_be_routed_to_corresponding_channel
296
- channel_at(14).expects(:do_failure).with()
297
- transport.return(CHANNEL_FAILURE, :long, 14)
298
- process_times(2)
299
- end
300
-
301
- def test_channel_close_packet_should_be_routed_to_corresponding_channel_and_channel_should_be_closed_and_removed
302
- channel_at(14).expects(:do_close).with()
303
- session.channels[14].expects(:close).with()
304
- transport.return(CHANNEL_CLOSE, :long, 14)
305
- process_times(2)
306
- assert session.channels.empty?
307
- end
308
-
309
- def test_multiple_pending_dispatches_should_be_dispatched_together
310
- channel_at(14).expects(:do_eof).with()
311
- session.channels[14].expects(:do_success).with()
312
- transport.return(CHANNEL_SUCCESS, :long, 14)
313
- transport.return(CHANNEL_EOF, :long, 14)
314
- process_times(2)
315
- end
316
-
317
- def test_writers_without_pending_writes_should_not_be_considered_for_select
318
- IO.expects(:select).with([socket],[],nil,nil).returns([[],[],[]])
319
- session.process
320
- end
321
-
322
- def test_writers_with_pending_writes_should_be_considered_for_select
323
- socket.enqueue("laksdjflasdkf")
324
- IO.expects(:select).with([socket],[socket],nil,nil).returns([[],[],[]])
325
- session.process
326
- end
327
-
328
- def test_ready_readers_should_be_filled
329
- socket.expects(:recv).returns("this is some data")
330
- IO.expects(:select).with([socket],[],nil,nil).returns([[socket],[],[]])
331
- session.process
332
- assert_equal [socket], session.listeners.keys
333
- end
334
-
335
- def test_ready_readers_that_cant_be_filled_should_be_removed
336
- socket.expects(:recv).returns("")
337
- socket.expects(:close)
338
- IO.expects(:select).with([socket],[],nil,nil).returns([[socket],[],[]])
339
- session.process
340
- assert session.listeners.empty?
341
- end
342
-
343
- def test_ready_readers_that_are_registered_with_a_block_should_call_block_instead_of_fill
344
- io = stub("io", :pending_write? => false)
345
- flag = false
346
- session.stop_listening_to(socket) # so that we only have to test the presence of a single IO object
347
- session.listen_to(io) { flag = true }
348
- IO.expects(:select).with([io],[],nil,nil).returns([[io],[],[]])
349
- session.process
350
- assert flag, "callback should have been invoked"
351
- end
352
-
353
- def test_ready_writers_should_call_send_pending
354
- socket.enqueue("laksdjflasdkf")
355
- socket.expects(:send).with("laksdjflasdkf", 0).returns(13)
356
- IO.expects(:select).with([socket],[socket],nil,nil).returns([[],[socket],[]])
357
- session.process
358
- end
359
-
360
- def test_process_should_call_rekey_as_needed
361
- transport.expects(:rekey_as_needed)
362
- IO.expects(:select).with([socket],[],nil,nil).returns([[],[],[]])
363
- session.process
364
- end
365
-
366
- def test_process_should_call_enqueue_message_if_io_select_timed_out
367
- timeout = Net::SSH::Connection::Session::DEFAULT_IO_SELECT_TIMEOUT
368
- options = { :keepalive => true }
369
- expected_packet = P(:byte, Net::SSH::Packet::GLOBAL_REQUEST, :string, "keepalive@openssh.com", :bool, true)
370
- IO.stubs(:select).with([socket],[],nil,timeout).returns(nil)
371
- transport.expects(:enqueue_message).with{ |msg| msg.content == expected_packet.content }
372
- session(options).process
373
- end
374
-
375
- def test_process_should_raise_if_keepalives_not_answered
376
- timeout = Net::SSH::Connection::Session::DEFAULT_IO_SELECT_TIMEOUT
377
- options = { :keepalive => true, :keepalive_interval => 300, :keepalive_maxcount => 3 }
378
- expected_packet = P(:byte, Net::SSH::Packet::GLOBAL_REQUEST, :string, "keepalive@openssh.com", :bool, true)
379
- [1,2,3].each do |i|
380
- Time.stubs(:now).returns(i*300)
381
- IO.stubs(:select).with([socket],[],nil,timeout).returns(nil)
382
- transport.expects(:enqueue_message).with{ |msg| msg.content == expected_packet.content }
383
- session(options).process
384
- end
385
-
386
- Time.stubs(:now).returns(4*300)
387
- IO.stubs(:select).with([socket],[],nil,timeout).returns(nil)
388
- transport.expects(:enqueue_message).with{ |msg| msg.content == expected_packet.content }
389
- assert_raises(Net::SSH::Timeout) { session(options).process }
390
- end
391
-
392
- def test_process_should_not_call_enqueue_message_unless_io_select_timed_out
393
- timeout = Net::SSH::Connection::Session::DEFAULT_IO_SELECT_TIMEOUT
394
- options = { :keepalive => true }
395
- IO.stubs(:select).with([socket],[],nil,timeout).returns([[],[],[]])
396
- transport.expects(:enqueue_message).never
397
- session(options).process
398
- end
399
-
400
- def test_process_should_not_call_enqueue_message_unless_keepalive_interval_not_go_on
401
- timeout = 10
402
- options = { :keepalive => true, :keepalive_interval => timeout }
403
- Time.stubs(:now).returns(Time.at(0), Time.at(9), Time.at(timeout))
404
- IO.stubs(:select).with([socket],[],nil,timeout).returns(nil)
405
- transport.expects(:enqueue_message).times(2)
406
- 3.times { session(options).process }
407
- end
408
-
409
- def test_process_should_call_io_select_with_nil_as_last_arg_if_keepalive_disabled
410
- IO.expects(:select).with([socket],[],nil,nil).returns([[],[],[]])
411
- session.process
412
- end
413
-
414
- def test_process_should_call_io_select_with_interval_as_last_arg_if_keepalive_interval_passed
415
- timeout = 10
416
- options = { :keepalive => true, :keepalive_interval => timeout }
417
- IO.expects(:select).with([socket],[],nil,timeout).returns([[],[],[]])
418
- session(options).process
419
- end
420
-
421
- def test_loop_should_call_process_until_process_returns_false
422
- IO.stubs(:select).with([socket],[],nil,nil).returns([[],[],[]])
423
- session.expects(:process).with(nil).times(4).returns(true,true,true,false).yields
424
- n = 0
425
- session.loop { n += 1 }
426
- assert_equal 4, n
427
- end
428
-
429
- def test_exec_should_open_channel_and_configure_default_callbacks
430
- prep_exec("ls", :stdout, "data packet", :stderr, "extended data packet")
431
-
432
- call = :first
433
- session.exec "ls" do |channel, type, data|
434
- if call == :first
435
- assert_equal :stdout, type
436
- assert_equal "data packet", data
437
- call = :second
438
- elsif call == :second
439
- assert_equal :stderr, type
440
- assert_equal "extended data packet", data
441
- call = :third
442
- else
443
- flunk "should never get here, call == #{call.inspect}"
444
- end
445
- end
446
-
447
- session.loop
448
- assert_equal :third, call
449
- end
450
-
451
- def test_exec_without_block_should_use_print_to_display_result
452
- prep_exec("ls", :stdout, "data packet", :stderr, "extended data packet")
453
- $stdout.expects(:print).with("data packet")
454
- $stderr.expects(:print).with("extended data packet")
455
-
456
- session.exec "ls"
457
- session.loop
458
- end
459
-
460
- def test_exec_bang_should_block_until_command_finishes
461
- prep_exec("ls", :stdout, "some data")
462
- called = false
463
- session.exec! "ls" do |channel, type, data|
464
- called = true
465
- assert_equal :stdout, type
466
- assert_equal "some data", data
467
- end
468
- assert called
469
- end
470
-
471
- def test_exec_bang_without_block_should_return_data_as_string
472
- prep_exec("ls", :stdout, "some data")
473
- assert_equal "some data", session.exec!("ls")
474
- end
475
-
476
- private
477
-
478
- def prep_exec(command, *data)
479
- transport.mock_enqueue = true
480
- transport.expect do |t, p|
481
- assert_equal CHANNEL_OPEN, p.type
482
- t.return(CHANNEL_OPEN_CONFIRMATION, :long, p[:remote_id], :long, 0, :long, 0x20000, :long, 0x10000)
483
- t.expect do |t2, p2|
484
- assert_equal CHANNEL_REQUEST, p2.type
485
- assert_equal "exec", p2[:request]
486
- assert_equal true, p2[:want_reply]
487
- assert_equal "ls", p2.read_string
488
-
489
- t2.return(CHANNEL_SUCCESS, :long, p[:remote_id])
490
-
491
- 0.step(data.length-1, 2) do |index|
492
- type = data[index]
493
- datum = data[index+1]
494
-
495
- if type == :stdout
496
- t2.return(CHANNEL_DATA, :long, p[:remote_id], :string, datum)
497
- else
498
- t2.return(CHANNEL_EXTENDED_DATA, :long, p[:remote_id], :long, 1, :string, datum)
499
- end
500
- end
501
-
502
- t2.return(CHANNEL_CLOSE, :long, p[:remote_id])
503
- t2.expect { |t3,p3| assert_equal CHANNEL_CLOSE, p3.type }
504
- end
505
- end
506
- end
507
-
508
- module MockSocket
509
- # so that we can easily test the contents that were enqueued, without
510
- # worrying about all the packet stream overhead
511
- def enqueue_packet(message)
512
- enqueue(message.to_s)
513
- end
514
- end
515
-
516
- def socket
517
- @socket ||= begin
518
- socket ||= Object.new
519
- socket.extend(Net::SSH::Transport::PacketStream)
520
- socket.extend(MockSocket)
521
- socket
522
- end
523
- end
524
-
525
- def channel_at(local_id)
526
- session.channels[local_id] = stub("channel", :process => true, :closing? => false)
527
- end
528
-
529
- def transport(options={})
530
- @transport ||= MockTransport.new(options.merge(:socket => socket))
531
- end
532
-
533
- def session(options={})
534
- @session ||= Net::SSH::Connection::Session.new(transport, options)
535
- end
536
-
537
- def process_times(n)
538
- i = 0
539
- session.process { (i += 1) < n }
540
- end
541
- end
542
-
543
- end
@@ -1 +0,0 @@
1
- github.com ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAq2A7hRGmdnm9tUDbO9IDSwBK6TbQa+PXYPCPy6rbTrTtw7PHkccKrpp0yVhp5HdEIcKr6pLlVDBfOLX9QUsyCOV0wzfjIJNlGEYsdlLJizHhbn2mUjvSAHQqZETYP81eFzLQNnPHt4EVVUh7VfDESU84KezmD5QlWpXLmvU31/yMf+Se8xhHTvKSCZIFImWwoG6mbUoWf9nzpIoaSjB+weqqUUmpaaasXVal72J+UX2B+2RPW3RcT0eOzQgqlJL3RKrTJvdsjE3JEAvGq3lGHSZXy28G3skua2SmVi/w4yCE6gbODqnTWlg7+wC604ydGXA8VJiS5ap43JXiUFFAaQ==