1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#--
# =============================================================================
# Copyright (c) 2007, Chris Andrews (chris@nodnol.org),
# Jamis Buck (jgb3@email.byu.edu)
# All rights reserved.
#
# This source file is distributed as part of the Net::SSH Secure Shell Client
# library for Ruby. This file (and the library as a whole) may be used only as
# allowed by either the BSD license, or the Ruby license (or, by association
# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
# distribution for the texts of these licenses.
# -----------------------------------------------------------------------------
# net-ssh website : http://net-ssh.rubyforge.org
# project website: http://rubyforge.org/projects/net-ssh
# =============================================================================
#++
module Net
module SSH
module Service
module AgentForward
class Driver
def initialize( connection, buffers, log, agent )
@connection = connection
@buffers = buffers
@log = log
@agent = agent
@data = ''
@connection.add_channel_open_handler(
"auth-agent@openssh.com", &method(:do_open_channel) )
end
def request
@connection.channel_request( 'auth-agent-req@openssh.com' )
end
def do_open_channel( connection, channel, data )
channel.on_data(&method(:do_data))
end
# handle CHANNEL_DATA packets received on the agent-forward
# channel - pass complete received packets to the agent.
def do_data( channel, data )
@data = @data + data
reply = call_agent
if reply
channel.send_data reply
@data = ''
end
end
# Called if we have any data to forward to the
# agent. Examines the accumulated data to see if we have a
# complete packet, based on the length field (the first four
# bytes as a network long).
def call_agent
# if we have enough data to check the length of this packet
if @data.length >= 4
packet_length = @data[0..3].unpack('N').first
# send the complete packet to the agent and read the
# response
if @data.length == (4 + packet_length)
@agent.send_raw_packet @data
buffer = @agent.read_raw_packet
end
end
buffer
end
end
end # AgentForward
end # Service
end # SSH
end # Net
|
