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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
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
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
#
# (c) Jan Gehring <jan.gehring@gmail.com>
#
package Rex::Helper::SSH2;
use v5.12.5;
use warnings;
our $VERSION = '1.14.1'; # VERSION
require Exporter;
use Data::Dumper;
require Rex::Commands;
use Time::HiRes qw(sleep);
use base qw(Exporter);
use vars qw(@EXPORT);
@EXPORT = qw(net_ssh2_exec net_ssh2_exec_output net_ssh2_shell_exec);
our $READ_STDERR = 1;
our $EXEC_AND_SLEEP = 0;
sub net_ssh2_exec {
my ( $ssh, $cmd, $base, $option ) = @_;
my $chan = $ssh->channel;
# REQUIRE_TTY can be turned off by feature no_tty
if ( !Rex::Config->get_no_tty ) {
$chan->pty("xterm"); # set to xterm, due to problems with vt100.
# if vt100 sometimes the restart of services doesn't work and need a sleep .000001 after the command...
# strange bug...
$chan->pty_size( 4000, 80 );
}
$chan->blocking(1);
$chan->exec($cmd);
my $in;
my $in_err = "";
my $rex_int_conf = Rex::Commands::get("rex_internals") || {};
my $buffer_size = 1024;
if ( exists $rex_int_conf->{read_buffer_size} ) {
$buffer_size = $rex_int_conf->{read_buffer_size};
}
my @lines;
my $last_line;
my $current_line = "";
while ( my $len = $chan->read( my $buf, $buffer_size ) ) {
$in .= $buf;
$current_line .= $buf;
if ( $buf =~ m/\n/ms ) {
@lines = split( /\n/, $current_line );
unshift @lines, $last_line if ($last_line);
$last_line = pop @lines;
for my $line (@lines) {
$line =~ s/[\r\n]//gms;
$line .= "\n";
$base->execute_line_based_operation( $line, $option ) && goto END_READ;
}
$current_line = "";
}
}
my @lines_err;
my $last_line_err = "";
while ( my $len = $chan->read( my $buf_err, $buffer_size, 1 ) ) {
$in_err .= $buf_err;
@lines_err = split( /\n/, $buf_err );
unshift @lines_err, $last_line_err if ($last_line_err);
$last_line_err = pop @lines_err;
for my $line (@lines_err) {
$line =~ s/[\r\n]//gms;
$line .= "\n";
$base->execute_line_based_operation( $line, $option ) && goto END_READ;
}
}
#select undef, undef, undef, 0.002; # wait a little before closing the channel
#sleep 1;
END_READ:
$chan->send_eof;
my $wait_c = 0;
my $wait_max = $rex_int_conf->{ssh2_channel_closewait_max} || 500;
while ( !$chan->eof ) {
Rex::Logger::debug("Waiting for eof on ssh channel.");
sleep 0.002; # wait a little for retry
$wait_c++;
if ( $wait_c >= $wait_max ) {
# channel will be force closed.
Rex::Logger::debug(
"Rex::Helper::SSH2::net_ssh2_exec: force closing channel for command: $cmd"
);
last;
}
}
$chan->wait_closed;
$? = $chan->exit_status;
# if used with $chan->pty() we have to remove \r
if ( !Rex::Config->get_no_tty ) {
$in =~ s/\r//g if $in;
$in_err =~ s/\r//g if $in_err;
}
if (wantarray) {
return ( $in, $in_err );
}
return $in;
}
sub net_ssh2_exec_output {
my ( $ssh, $cmd, $callback ) = @_;
my $chan = $ssh->channel;
$chan->blocking(1);
$chan->exec($cmd);
while (1) {
my $buf;
my $buf_err;
$chan->read( $buf, 15 );
$chan->read( $buf_err, 15 );
if ($callback) {
&$callback( $buf, $buf_err );
}
else {
print $buf;
print $buf_err;
}
last unless $buf;
}
$chan->close;
$? = $chan->exit_status;
}
1;
|
