File Coverage

blib/lib/Vayne/SSH/Tunnel.pm

Criterion	Covered	Total	%
statement	26	72	36.1
branch	0	20	0.0
condition	0	16	0.0
subroutine	9	13	69.2
pod	0	1	0.0
total	35	122	28.6

package Vayne::SSH::Tunnel;

use strict;
use 5.008_005;

our $VERSION = '0.01';

use POSIX;
use YAML::XS;
use File::Spec;
use Sys::Hostname;
use FindBin qw($Bin);
use Fcntl qw( :flock );
use Net::EmptyPort qw(empty_port);

sub Run
{
    my(%option, $cmd, $lport) = @_;

    $lport = empty_port();
    $cmd = _fmt_ssh(%option, lport => $lport );

    die "can not find user $option{user}" unless my(undef, undef, $uid, $gid) = getpwnam $option{user};

    POSIX::setgid($gid);
    POSIX::setuid($uid);
    print $cmd, "\n";

    #fork
    setpgrp(0,0);
    $SIG{TERM} = $SIG{HUP} = sub{ print "Sig recv, killed!\n";      kill 'KILL', -$$};
    $SIG{ALRM} =             sub{ print "Alarm timeout, killed!\n"; kill 'KILL', -$$};

    my($pid, %grep, $lock_fh) = open my $fh, "$cmd|" or die "can't fork cmd $cmd";

    alarm $option{timeout} if $option{timeout};

    #read
    $|++;

    READ:  while(my $line = <$fh>)
    {
        alarm 0;
        print $line;

        #write conf
        if( $line =~ /Entering interactive session/ and $option{confdir} )
        {
            my $path = File::Spec->join($option{confdir}, $option{name}. ".". $$);

            unless (
                not defined $lock_fh
                and open $lock_fh, '>', $path
                and flock $lock_fh, LOCK_EX | LOCK_NB
            )
            {
                warn "can not open $path";
                last READ;
            }

            print $lock_fh YAML::XS::Dump { $option{title} => "127.0.0.1:$lport" };
            $lock_fh->flush;


        }

        print "channel open reach $option{max_channel}!\n" and last if $line =~ /channel (\d+):/ && $1 >= $option{max_channel};

        while(my($word, $times) = each %{ $option{grep_word} })
        {
            next unless $line =~ /$word/;
            $grep{$word} += 1;
            print "match '$word' reach $times!\n" and last READ if $grep{$word} >= $times;
        }

        alarm $option{timeout} if $option{timeout};
    }

    kill 'KILL', -$$;
}


#ssh -tt -L 127.0.0.1:7920:127.0.0.1:7920 [e-mail] ssh -v -o 'ServerAliveInterval=10' -o 'ServerAliveCountMax=3' -tt -L localhost:7920:127.0.0.1:7920 ser2.net ssh -v -o 'ServerAliveInterval=10' -o 'ServerAliveCountMax=1' -N -L localhost:7920:127.0.0.1:1080 [e-mail] 2>&1
sub _fmt_ssh
{
    my(%opt, @way, $option, $cmd, $hostname) = @_;
    @way = @{ $opt{way} };
    $option = join ' ', map{ "-o '$_'" }@{$opt{ option }};

    $hostname = hostname;
    shift @way if $way[0] =~ /\@$hostname$/;

    for(splice @way, 0, -1)
    {
        $cmd .= sprintf "ssh -tt %s -L 127.0.0.1:%s:127.0.0.1:%s %s ", $option, $opt{lport}, $opt{lport}, $_;
    }

    $cmd .= sprintf "ssh -v %s -N -L 127.0.0.1:%s:127.0.0.1:%s %s 2>&1", $option, $opt{lport}, $opt{dport}, $way[-1];
}

1;
__END__