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
|
#
# (c) Jan Gehring <jan.gehring@gmail.com>
#
# vim: set ts=2 sw=2 tw=0:
# vim: set expandtab:
package Rex::Interface::File::Sudo;
use strict;
use warnings;
our $VERSION = '1.4.1'; # VERSION
use Fcntl;
use File::Basename;
require Rex::Commands;
use Rex::Interface::Fs;
use Rex::Interface::File::Base;
use Rex::Helper::Path;
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 ) = @_;
if ( my $ssh = Rex::is_ssh() ) {
if ( ref $ssh eq "Net::OpenSSH" ) {
$self->{fh} = Rex::Interface::File->create("OpenSSH");
}
else {
$self->{fh} = Rex::Interface::File->create("SSH");
}
}
else {
$self->{fh} = Rex::Interface::File->create("Local");
}
# always use current logged in user for sudo fs operations
Rex::get_current_connection_object()->push_sudo_options( {} );
$self->{mode} = $mode;
$self->{file} = $file;
$self->{rndfile} = get_tmp_file;
if ( $self->_fs->is_file($file) ) {
# resolving symlinks
while ( my $link = $self->_fs->readlink($file) ) {
if ( $link !~ m/^\// ) {
$file = dirname($file) . "/" . $link;
}
else {
$file = $link;
}
$link = $self->_fs->readlink($link);
}
$self->{file_stat} = { $self->_fs->stat( $self->{file} ) };
$self->_fs->cp( $file, $self->{rndfile} );
$self->_fs->chmod( 600, $self->{rndfile} );
$self->_fs->chown( Rex::Commands::connection->get_auth_user,
$self->{rndfile} );
}
$self->{fh}->open( $mode, $self->{rndfile} );
Rex::get_current_connection_object()->pop_sudo_options();
return $self->{fh};
}
sub read {
my ( $self, $len ) = @_;
return $self->{fh}->read($len);
}
sub write {
my ( $self, $buf ) = @_;
$self->{fh}->write($buf);
}
sub seek {
my ( $self, $pos ) = @_;
$self->{fh}->seek($pos);
}
sub close {
my ($self) = @_;
return unless $self->{fh};
# always use current logged in user for sudo fs operations
Rex::get_current_connection_object()->push_sudo_options( {} );
$self->{fh}->close;
if ( exists $self->{mode}
&& ( $self->{mode} eq ">" || $self->{mode} eq ">>" ) )
{
my $exec = Rex::Interface::Exec->create;
if ( $self->{file_stat} ) {
my %stat = $self->_fs->stat( $self->{file} );
$self->_fs->chmod( $stat{mode}, $self->{rndfile} );
$self->_fs->chown( $stat{uid}, $self->{rndfile} );
$self->_fs->chgrp( $stat{gid}, $self->{rndfile} );
}
$self->_fs->rename( $self->{rndfile}, $self->{file} );
#$exec->exec("cat " . $self->{rndfile} . " >'" . $self->{file} . "'");
}
$self->{fh} = undef;
$self->_fs->unlink( $self->{rndfile} );
Rex::get_current_connection_object()->pop_sudo_options();
$self = undef;
}
sub _fs {
my ($self) = @_;
return Rex::Interface::Fs->create("Sudo");
}
1;
|
