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
|
#
# (c) Jan Gehring <jan.gehring@gmail.com>
#
package Rex::Interface::File::OpenSSH;
use v5.12.5;
use warnings;
our $VERSION = '1.14.1'; # VERSION
use Fcntl;
use Rex::Interface::Fs;
use Rex::Interface::File::Base;
BEGIN {
use Rex::Require;
Net::SFTP::Foreign::Constants->use(qw(:flags :fxp));
}
use base qw(Rex::Interface::File::Base);
sub new {
my $that = shift;
my $proto = ref($that) || $that;
my $self = $proto->SUPER::new(@_);
bless( $self, $proto );
return $self;
}
sub open {
my ( $self, $mode, $file ) = @_;
Rex::Logger::debug("Opening $file with mode: $mode");
my $sftp = Rex::get_sftp();
if ( $mode eq ">" ) {
$self->{fh} =
$sftp->open( $file, SSH2_FXF_WRITE | SSH2_FXF_CREAT | SSH2_FXF_TRUNC );
}
elsif ( $mode eq ">>" ) {
$self->{fh} = $sftp->open( $file, SSH2_FXF_WRITE | SSH2_FXF_APPEND );
my $fs = Rex::Interface::Fs->create;
my %stat = $fs->stat($file);
$self->{fh}->seek( $stat{size}, 0 );
}
elsif ( $mode eq "<" ) {
$self->{fh} = $sftp->open( $file, SSH2_FXF_READ );
}
return $self->{fh};
}
sub read {
my ( $self, $len ) = @_;
my $sftp = Rex::get_sftp();
my $buf = $sftp->read( $self->{fh}, $len );
return $buf;
}
sub write {
my ( $self, $buf ) = @_;
utf8::encode($buf)
if Rex::Config->get_write_utf8_files && utf8::is_utf8($buf);
my $sftp = Rex::get_sftp();
$sftp->write( $self->{fh}, $buf );
}
sub seek {
my ( $self, $pos ) = @_;
my $sftp = Rex::get_sftp();
$sftp->seek( $self->{fh}, $pos, 0 );
}
sub close {
my ($self) = @_;
my $sftp = Rex::get_sftp();
$sftp->close( $self->{fh} );
}
1;
|
