Class: Snapsync::SSHPopen

Inherits:
Object
  • Object
show all
Defined in:
lib/snapsync/ssh_popen.rb

Defined Under Namespace

Classes: ExitSignal, NonZeroExitCode

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(machine, command, **options) ⇒ SSHPopen

Returns a new instance of SSHPopen.

Parameters:



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
79
80
81
82
83
84
85
86
87
88
89
 
# File 'lib/snapsync/ssh_popen.rb', line 17

def initialize(machine, command, **options)
    @read_buffer, read_buffer_in = IO.pipe
    write_buffer_out, @write_buffer = IO.pipe

    if options[:out]
        read_buffer_in = File.open(options[:out], "w")
    end

    ready = Concurrent::AtomicBoolean.new(false)
    @ssh_thr = Thread.new do
        machine.dup_ssh do |ssh|
            ready.make_true
            if Snapsync.SSH_DEBUG
                log = Logger.new(STDOUT)
                log.level = Logger::DEBUG
                ssh.logger = log
                ssh.logger.sev_threshold = Logger::Severity::DEBUG
            end
            # @type [Net::SSH::Connection::Channel]
            channel = ssh.open_channel do |channel|
                Snapsync.debug "SSHPopen channel opened: #{channel}"

                channel.on_data do |ch, data|
                    read_buffer_in.write(data)
                end
                channel.on_extended_data do |ch, type, data|
                    data = data.chomp
                    if data.length > 0
                        Snapsync.error data
                    end
                end

                channel.on_request("exit-status") do |ch2, data|
                    code = data.read_long
                    if code != 0
                        raise NonZeroExitCode, "Exited with code: #{code}"
                    else
                        Snapsync.debug "SSHPopen command finished."
                    end
                end

                channel.on_request("exit-signal") do |ch2, data|
                    exit_signal = data.read_long
                    raise ExitSignal, "Exited due to signal: #{exit_signal}"
                end

                channel.on_process do
                    begin
                        channel.send_data(write_buffer_out.read_nonblock(2 << 20))
                    rescue IO::EAGAINWaitReadable
                    end
                end

                channel.exec(Shellwords.join command)
            end

            ssh.loop(0.001) {
                if write_buffer_out.closed?
                    channel.close
                end

                channel.active?
            }

            Snapsync.debug "SSHPopen session closed"
            read_buffer_in.close
            write_buffer_out.close
        end
    end
    while ready.false?
        sleep 0.001
    end
end

Instance Attribute Details

#read_buffer ⇒ IO (readonly)

Returns:

  • (IO) 


9
10
11
 
# File 'lib/snapsync/ssh_popen.rb', line 9

def read_buffer
  @read_buffer
end

#write_buffer ⇒ IO (readonly)

Returns:

  • (IO) 


12
13
14
 
# File 'lib/snapsync/ssh_popen.rb', line 12

def write_buffer
  @write_buffer
end

Instance Method Details

#close ⇒ Object 



103
104
105
 
# File 'lib/snapsync/ssh_popen.rb', line 103

def close
    write_buffer.close
end

#read(nbytes = nil) ⇒ Object 



95
96
97
 
# File 'lib/snapsync/ssh_popen.rb', line 95

def read(nbytes = nil)
    read_buffer.read nbytes
end

#sync=(sync) ⇒ Object 



91
92
93
 
# File 'lib/snapsync/ssh_popen.rb', line 91

def sync=(sync)
    # ignore
end

#write(data) ⇒ Object 



99
100
101
 
# File 'lib/snapsync/ssh_popen.rb', line 99

def write(data)
    write_buffer.write data
end