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
|
# $OpenBSD: Remote.pm,v 1.7 2016/05/03 19:13:04 bluhm Exp $
# Copyright (c) 2010-2014 Alexander Bluhm <bluhm@openbsd.org>
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
use strict;
use warnings;
package Remote;
use parent 'Proc';
use Carp;
use Cwd;
use File::Basename;
my %PIPES;
sub close_pipes {
my @pipes = @_ ? @_ : keys %PIPES
or return;
foreach (@pipes) {
# file descriptor cannot be a hash key, so use hash value
my $fh = $PIPES{$_};
# also print new line as close is delayed by forked processes
print $fh "close\n";
close($fh);
}
sleep 1; # give other end a chance to finish process
delete @PIPES{@pipes};
}
END {
close_pipes();
}
sub new {
my $class = shift;
my %args = @_;
$args{logfile} ||= "remote.log";
$args{up} ||= "listen sock: ";
$args{down} ||= $args{dryrun} ? "relayd.conf" : "parent terminating";
$args{func} = sub { Carp::confess "$class func may not be called" };
$args{remotessh}
or croak "$class remote ssh host not given";
$args{forward}
or croak "$class forward not given";
my $self = Proc::new($class, %args);
$self->{listenaddr}
or croak "$class listen addr not given";
$self->{connectaddr}
or croak "$class connect addr not given";
$self->{connectport}
or croak "$class connect port not given";
return $self;
}
sub run {
my $self = Proc::run(shift, @_);
$PIPES{$self->{pipe}} = $self->{pipe};
return $self;
}
sub up {
my $self = Proc::up(shift, @_);
my $lsock = $self->loggrep(qr/^listen sock: /)
or croak ref($self), " no 'listen sock: ' in $self->{logfile}";
my($addr, $port) = $lsock =~ /: (\S+) (\S+)$/
or croak ref($self), " no listen addr and port in $self->{logfile}";
$self->{listenaddr} = $addr;
$self->{listenport} = $port;
return $self;
}
sub child {
my $self = shift;
my @opts = $ENV{SSH_OPTIONS} ? split(' ', $ENV{SSH_OPTIONS}) : ();
my @sudo = $ENV{SUDO} ? "SUDO=$ENV{SUDO}" : ();
my @ktrace = $ENV{KTRACE} ? "KTRACE=$ENV{KTRACE}" : ();
my @relayd = $ENV{RELAYD} ? "RELAYD=$ENV{RELAYD}" : ();
my $dir = dirname($0);
$dir = getcwd() if ! $dir || $dir eq ".";
my @cmd = ("ssh", @opts, $self->{remotessh},
@sudo, @ktrace, @relayd, "perl",
"-I", $dir, "$dir/".basename($0), $self->{forward},
$self->{listenaddr}, $self->{connectaddr}, $self->{connectport},
($self->{testfile} ? "$dir/".basename($self->{testfile}) : ()));
print STDERR "execute: @cmd\n";
exec @cmd;
die ref($self), " exec '@cmd' failed: $!";
}
sub close_child {
my $self = shift;
close_pipes(delete $self->{pipe});
return $self;
}
1;
|
