~frank-cizmich/percona-toolkit/percona-toolkit

« back to all changes in this revision

Viewing changes to bin/pt-heartbeat

  • Committer: Daniel Nichter
  • Date: 2014-02-20 03:00:02 UTC
  • Revision ID: daniel@percona.com-20140220030002-gsj22qr101mb2fbp
Remove version check bin type.  Update all tools.

Show diffs side-by-side

added added

removed removed

Lines of Context:
22
22
      TableParser
23
23
      Retry
24
24
      Transformers
25
 
      HTTPMicro
 
25
      HTTP::Micro
26
26
      VersionCheck
27
27
   ));
28
28
}
3522
3522
# ###########################################################################
3523
3523
 
3524
3524
# ###########################################################################
3525
 
# HTTPMicro package
 
3525
# HTTP::Micro package
3526
3526
# This package is a copy without comments from the original.  The original
3527
3527
# with comments and its test file can be found in the Bazaar repository at,
3528
 
#   lib/HTTPMicro.pm
3529
 
#   t/lib/HTTPMicro.t
 
3528
#   lib/HTTP/Micro.pm
 
3529
#   t/lib/HTTP/Micro.t
3530
3530
# See https://launchpad.net/percona-toolkit for more information.
3531
3531
# ###########################################################################
3532
3532
{
3533
 
 
3534
 
package HTTPMicro;
3535
 
BEGIN {
3536
 
  $HTTPMicro::VERSION = '0.001';
3537
 
}
 
3533
package HTTP::Micro;
 
3534
 
 
3535
our $VERSION = '0.01';
 
3536
 
3538
3537
use strict;
3539
 
use warnings;
3540
 
 
 
3538
use warnings FATAL => 'all';
 
3539
use English qw(-no_match_vars);
3541
3540
use Carp ();
3542
3541
 
3543
 
 
3544
3542
my @attributes;
3545
3543
BEGIN {
3546
3544
    @attributes = qw(agent timeout);
3611
3609
        headers   => {},
3612
3610
    };
3613
3611
 
3614
 
    my $handle  = HTTPMicro::Handle->new(timeout => $self->{timeout});
 
3612
    my $handle  = HTTP::Micro::Handle->new(timeout => $self->{timeout});
3615
3613
 
3616
3614
    $handle->connect($scheme, $host, $port);
3617
3615
 
3676
3674
    return ($scheme, $host, $port, $path_query);
3677
3675
}
3678
3676
 
3679
 
package
3680
 
    HTTPMicro::Handle; # hide from PAUSE/indexers
3681
 
use strict;
3682
 
use warnings;
3683
 
 
3684
 
use Carp       qw[croak];
3685
 
use Errno      qw[EINTR EPIPE];
3686
 
use IO::Socket qw[SOCK_STREAM];
3687
 
 
3688
 
sub BUFSIZE () { 32768 }
3689
 
 
3690
 
my $Printable = sub {
3691
 
    local $_ = shift;
3692
 
    s/\r/\\r/g;
3693
 
    s/\n/\\n/g;
3694
 
    s/\t/\\t/g;
3695
 
    s/([^\x20-\x7E])/sprintf('\\x%.2X', ord($1))/ge;
3696
 
    $_;
3697
 
};
3698
 
 
3699
 
sub new {
3700
 
    my ($class, %args) = @_;
3701
 
    return bless {
3702
 
        rbuf             => '',
3703
 
        timeout          => 60,
3704
 
        max_line_size    => 16384,
3705
 
        %args
3706
 
    }, $class;
3707
 
}
3708
 
 
3709
 
my $ssl_verify_args = {
3710
 
    check_cn => "when_only",
3711
 
    wildcards_in_alt => "anywhere",
3712
 
    wildcards_in_cn => "anywhere"
3713
 
};
3714
 
 
3715
 
sub connect {
3716
 
    @_ == 4 || croak(q/Usage: $handle->connect(scheme, host, port)/);
3717
 
    my ($self, $scheme, $host, $port) = @_;
3718
 
 
3719
 
    if ( $scheme eq 'https' ) {
3720
 
        eval "require IO::Socket::SSL"
3721
 
            unless exists $INC{'IO/Socket/SSL.pm'};
3722
 
        croak(qq/IO::Socket::SSL must be installed for https support\n/)
3723
 
            unless $INC{'IO/Socket/SSL.pm'};
3724
 
    }
3725
 
    elsif ( $scheme ne 'http' ) {
3726
 
      croak(qq/Unsupported URL scheme '$scheme'\n/);
3727
 
    }
3728
 
 
3729
 
    $self->{fh} = 'IO::Socket::INET'->new(
3730
 
        PeerHost  => $host,
3731
 
        PeerPort  => $port,
3732
 
        Proto     => 'tcp',
3733
 
        Type      => SOCK_STREAM,
3734
 
        Timeout   => $self->{timeout}
3735
 
    ) or croak(qq/Could not connect to '$host:$port': $@/);
3736
 
 
3737
 
    binmode($self->{fh})
3738
 
      or croak(qq/Could not binmode() socket: '$!'/);
3739
 
 
3740
 
    if ( $scheme eq 'https') {
3741
 
        IO::Socket::SSL->start_SSL($self->{fh});
3742
 
        ref($self->{fh}) eq 'IO::Socket::SSL'
3743
 
            or die(qq/SSL connection failed for $host\n/);
3744
 
        if ( $self->{fh}->can("verify_hostname") ) {
3745
 
            $self->{fh}->verify_hostname( $host, $ssl_verify_args );
3746
 
        }
3747
 
        else {
3748
 
         my $fh = $self->{fh};
3749
 
         _verify_hostname_of_cert($host, _peer_certificate($fh), $ssl_verify_args)
3750
 
               or die(qq/SSL certificate not valid for $host\n/);
3751
 
         }
3752
 
    }
3753
 
      
3754
 
    $self->{host} = $host;
3755
 
    $self->{port} = $port;
3756
 
 
3757
 
    return $self;
3758
 
}
3759
 
 
3760
 
sub close {
3761
 
    @_ == 1 || croak(q/Usage: $handle->close()/);
3762
 
    my ($self) = @_;
3763
 
    CORE::close($self->{fh})
3764
 
      or croak(qq/Could not close socket: '$!'/);
3765
 
}
3766
 
 
3767
 
sub write {
3768
 
    @_ == 2 || croak(q/Usage: $handle->write(buf)/);
3769
 
    my ($self, $buf) = @_;
3770
 
 
3771
 
    my $len = length $buf;
3772
 
    my $off = 0;
3773
 
 
3774
 
    local $SIG{PIPE} = 'IGNORE';
3775
 
 
3776
 
    while () {
3777
 
        $self->can_write
3778
 
          or croak(q/Timed out while waiting for socket to become ready for writing/);
3779
 
        my $r = syswrite($self->{fh}, $buf, $len, $off);
3780
 
        if (defined $r) {
3781
 
            $len -= $r;
3782
 
            $off += $r;
3783
 
            last unless $len > 0;
3784
 
        }
3785
 
        elsif ($! == EPIPE) {
3786
 
            croak(qq/Socket closed by remote server: $!/);
3787
 
        }
3788
 
        elsif ($! != EINTR) {
3789
 
            croak(qq/Could not write to socket: '$!'/);
3790
 
        }
3791
 
    }
3792
 
    return $off;
3793
 
}
3794
 
 
3795
 
sub read {
3796
 
    @_ == 2 || @_ == 3 || croak(q/Usage: $handle->read(len)/);
3797
 
    my ($self, $len) = @_;
3798
 
 
3799
 
    my $buf  = '';
3800
 
    my $got = length $self->{rbuf};
3801
 
 
3802
 
    if ($got) {
3803
 
        my $take = ($got < $len) ? $got : $len;
3804
 
        $buf  = substr($self->{rbuf}, 0, $take, '');
3805
 
        $len -= $take;
3806
 
    }
3807
 
 
3808
 
    while ($len > 0) {
3809
 
        $self->can_read
3810
 
          or croak(q/Timed out while waiting for socket to become ready for reading/);
3811
 
        my $r = sysread($self->{fh}, $buf, $len, length $buf);
3812
 
        if (defined $r) {
3813
 
            last unless $r;
3814
 
            $len -= $r;
3815
 
        }
3816
 
        elsif ($! != EINTR) {
3817
 
            croak(qq/Could not read from socket: '$!'/);
3818
 
        }
3819
 
    }
3820
 
    if ($len) {
3821
 
        croak(q/Unexpected end of stream/);
3822
 
    }
3823
 
    return $buf;
3824
 
}
3825
 
 
3826
 
sub readline {
3827
 
    @_ == 1 || croak(q/Usage: $handle->readline()/);
3828
 
    my ($self) = @_;
3829
 
 
3830
 
    while () {
3831
 
        if ($self->{rbuf} =~ s/\A ([^\x0D\x0A]* \x0D?\x0A)//x) {
3832
 
            return $1;
3833
 
        }
3834
 
        $self->can_read
3835
 
          or croak(q/Timed out while waiting for socket to become ready for reading/);
3836
 
        my $r = sysread($self->{fh}, $self->{rbuf}, BUFSIZE, length $self->{rbuf});
3837
 
        if (defined $r) {
3838
 
            last unless $r;
3839
 
        }
3840
 
        elsif ($! != EINTR) {
3841
 
            croak(qq/Could not read from socket: '$!'/);
3842
 
        }
3843
 
    }
3844
 
    croak(q/Unexpected end of stream while looking for line/);
3845
 
}
3846
 
 
3847
 
sub read_header_lines {
3848
 
    @_ == 1 || @_ == 2 || croak(q/Usage: $handle->read_header_lines([headers])/);
3849
 
    my ($self, $headers) = @_;
3850
 
    $headers ||= {};
3851
 
    my $lines   = 0;
3852
 
    my $val;
3853
 
 
3854
 
    while () {
3855
 
         my $line = $self->readline;
3856
 
 
3857
 
         if ($line =~ /\A ([^\x00-\x1F\x7F:]+) : [\x09\x20]* ([^\x0D\x0A]*)/x) {
3858
 
             my ($field_name) = lc $1;
3859
 
             $val = \($headers->{$field_name} = $2);
3860
 
         }
3861
 
         elsif ($line =~ /\A [\x09\x20]+ ([^\x0D\x0A]*)/x) {
3862
 
             $val
3863
 
               or croak(q/Unexpected header continuation line/);
3864
 
             next unless length $1;
3865
 
             $$val .= ' ' if length $$val;
3866
 
             $$val .= $1;
3867
 
         }
3868
 
         elsif ($line =~ /\A \x0D?\x0A \z/x) {
3869
 
            last;
3870
 
         }
3871
 
         else {
3872
 
            croak(q/Malformed header line: / . $Printable->($line));
3873
 
         }
3874
 
    }
3875
 
    return $headers;
3876
 
}
3877
 
 
3878
 
sub write_header_lines {
3879
 
    (@_ == 2 && ref $_[1] eq 'HASH') || croak(q/Usage: $handle->write_header_lines(headers)/);
3880
 
    my($self, $headers) = @_;
3881
 
 
3882
 
    my $buf = '';
3883
 
    while (my ($k, $v) = each %$headers) {
3884
 
        my $field_name = lc $k;
3885
 
         $field_name =~ /\A [\x21\x23-\x27\x2A\x2B\x2D\x2E\x30-\x39\x41-\x5A\x5E-\x7A\x7C\x7E]+ \z/x
3886
 
            or croak(q/Invalid HTTP header field name: / . $Printable->($field_name));
3887
 
         $field_name =~ s/\b(\w)/\u$1/g;
3888
 
         $buf .= "$field_name: $v\x0D\x0A";
3889
 
    }
3890
 
    $buf .= "\x0D\x0A";
3891
 
    return $self->write($buf);
3892
 
}
3893
 
 
3894
 
sub read_content_body {
3895
 
    @_ == 3 || @_ == 4 || croak(q/Usage: $handle->read_content_body(callback, response, [read_length])/);
3896
 
    my ($self, $cb, $response, $len) = @_;
3897
 
    $len ||= $response->{headers}{'content-length'};
3898
 
 
3899
 
    croak("No content-length in the returned response, and this "
3900
 
        . "UA doesn't implement chunking") unless defined $len;
3901
 
 
3902
 
    while ($len > 0) {
3903
 
        my $read = ($len > BUFSIZE) ? BUFSIZE : $len;
3904
 
        $cb->($self->read($read), $response);
3905
 
        $len -= $read;
3906
 
    }
3907
 
 
3908
 
    return;
3909
 
}
3910
 
 
3911
 
sub write_content_body {
3912
 
    @_ == 2 || croak(q/Usage: $handle->write_content_body(request)/);
3913
 
    my ($self, $request) = @_;
3914
 
    my ($len, $content_length) = (0, $request->{headers}{'content-length'});
3915
 
 
3916
 
    $len += $self->write($request->{content});
3917
 
 
3918
 
    $len == $content_length
3919
 
      or croak(qq/Content-Length missmatch (got: $len expected: $content_length)/);
3920
 
 
3921
 
    return $len;
3922
 
}
3923
 
 
3924
 
sub read_response_header {
3925
 
    @_ == 1 || croak(q/Usage: $handle->read_response_header()/);
3926
 
    my ($self) = @_;
3927
 
 
3928
 
    my $line = $self->readline;
3929
 
 
3930
 
    $line =~ /\A (HTTP\/(0*\d+\.0*\d+)) [\x09\x20]+ ([0-9]{3}) [\x09\x20]+ ([^\x0D\x0A]*) \x0D?\x0A/x
3931
 
      or croak(q/Malformed Status-Line: / . $Printable->($line));
3932
 
 
3933
 
    my ($protocol, $version, $status, $reason) = ($1, $2, $3, $4);
3934
 
 
3935
 
    return {
3936
 
        status   => $status,
3937
 
        reason   => $reason,
3938
 
        headers  => $self->read_header_lines,
3939
 
        protocol => $protocol,
3940
 
    };
3941
 
}
3942
 
 
3943
 
sub write_request_header {
3944
 
    @_ == 4 || croak(q/Usage: $handle->write_request_header(method, request_uri, headers)/);
3945
 
    my ($self, $method, $request_uri, $headers) = @_;
3946
 
 
3947
 
    return $self->write("$method $request_uri HTTP/1.1\x0D\x0A")
3948
 
         + $self->write_header_lines($headers);
3949
 
}
3950
 
 
3951
 
sub _do_timeout {
3952
 
    my ($self, $type, $timeout) = @_;
3953
 
    $timeout = $self->{timeout}
3954
 
        unless defined $timeout && $timeout >= 0;
3955
 
 
3956
 
    my $fd = fileno $self->{fh};
3957
 
    defined $fd && $fd >= 0
3958
 
      or croak(q/select(2): 'Bad file descriptor'/);
3959
 
 
3960
 
    my $initial = time;
3961
 
    my $pending = $timeout;
3962
 
    my $nfound;
3963
 
 
3964
 
    vec(my $fdset = '', $fd, 1) = 1;
3965
 
 
3966
 
    while () {
3967
 
        $nfound = ($type eq 'read')
3968
 
            ? select($fdset, undef, undef, $pending)
3969
 
            : select(undef, $fdset, undef, $pending) ;
3970
 
        if ($nfound == -1) {
3971
 
            $! == EINTR
3972
 
              or croak(qq/select(2): '$!'/);
3973
 
            redo if !$timeout || ($pending = $timeout - (time - $initial)) > 0;
3974
 
            $nfound = 0;
3975
 
        }
3976
 
        last;
3977
 
    }
3978
 
    $! = 0;
3979
 
    return $nfound;
3980
 
}
3981
 
 
3982
 
sub can_read {
3983
 
    @_ == 1 || @_ == 2 || croak(q/Usage: $handle->can_read([timeout])/);
3984
 
    my $self = shift;
3985
 
    return $self->_do_timeout('read', @_)
3986
 
}
3987
 
 
3988
 
sub can_write {
3989
 
    @_ == 1 || @_ == 2 || croak(q/Usage: $handle->can_write([timeout])/);
3990
 
    my $self = shift;
3991
 
    return $self->_do_timeout('write', @_)
3992
 
}
 
3677
} # HTTP::Micro
 
3678
 
 
3679
{
 
3680
   package HTTP::Micro::Handle;
 
3681
 
 
3682
   use strict;
 
3683
   use warnings FATAL => 'all';
 
3684
   use English qw(-no_match_vars);
 
3685
 
 
3686
   use Carp       qw(croak);
 
3687
   use Errno      qw(EINTR EPIPE);
 
3688
   use IO::Socket qw(SOCK_STREAM);
 
3689
 
 
3690
   sub BUFSIZE () { 32768 }
 
3691
 
 
3692
   my $Printable = sub {
 
3693
       local $_ = shift;
 
3694
       s/\r/\\r/g;
 
3695
       s/\n/\\n/g;
 
3696
       s/\t/\\t/g;
 
3697
       s/([^\x20-\x7E])/sprintf('\\x%.2X', ord($1))/ge;
 
3698
       $_;
 
3699
   };
 
3700
 
 
3701
   sub new {
 
3702
       my ($class, %args) = @_;
 
3703
       return bless {
 
3704
           rbuf          => '',
 
3705
           timeout       => 60,
 
3706
           max_line_size => 16384,
 
3707
           %args
 
3708
       }, $class;
 
3709
   }
 
3710
 
 
3711
   my $ssl_verify_args = {
 
3712
       check_cn         => "when_only",
 
3713
       wildcards_in_alt => "anywhere",
 
3714
       wildcards_in_cn  => "anywhere"
 
3715
   };
 
3716
 
 
3717
   sub connect {
 
3718
       @_ == 4 || croak(q/Usage: $handle->connect(scheme, host, port)/);
 
3719
       my ($self, $scheme, $host, $port) = @_;
 
3720
 
 
3721
       if ( $scheme eq 'https' ) {
 
3722
           eval "require IO::Socket::SSL"
 
3723
               unless exists $INC{'IO/Socket/SSL.pm'};
 
3724
           croak(qq/IO::Socket::SSL must be installed for https support\n/)
 
3725
               unless $INC{'IO/Socket/SSL.pm'};
 
3726
       }
 
3727
       elsif ( $scheme ne 'http' ) {
 
3728
         croak(qq/Unsupported URL scheme '$scheme'\n/);
 
3729
       }
 
3730
 
 
3731
       $self->{fh} = IO::Socket::INET->new(
 
3732
           PeerHost  => $host,
 
3733
           PeerPort  => $port,
 
3734
           Proto     => 'tcp',
 
3735
           Type      => SOCK_STREAM,
 
3736
           Timeout   => $self->{timeout}
 
3737
       ) or croak(qq/Could not connect to '$host:$port': $@/);
 
3738
 
 
3739
       binmode($self->{fh})
 
3740
         or croak(qq/Could not binmode() socket: '$!'/);
 
3741
 
 
3742
       if ( $scheme eq 'https') {
 
3743
           IO::Socket::SSL->start_SSL($self->{fh});
 
3744
           ref($self->{fh}) eq 'IO::Socket::SSL'
 
3745
               or die(qq/SSL connection failed for $host\n/);
 
3746
           if ( $self->{fh}->can("verify_hostname") ) {
 
3747
               $self->{fh}->verify_hostname( $host, $ssl_verify_args );
 
3748
           }
 
3749
           else {
 
3750
            my $fh = $self->{fh};
 
3751
            _verify_hostname_of_cert($host, _peer_certificate($fh), $ssl_verify_args)
 
3752
                  or die(qq/SSL certificate not valid for $host\n/);
 
3753
            }
 
3754
       }
 
3755
         
 
3756
       $self->{host} = $host;
 
3757
       $self->{port} = $port;
 
3758
 
 
3759
       return $self;
 
3760
   }
 
3761
 
 
3762
   sub close {
 
3763
       @_ == 1 || croak(q/Usage: $handle->close()/);
 
3764
       my ($self) = @_;
 
3765
       CORE::close($self->{fh})
 
3766
         or croak(qq/Could not close socket: '$!'/);
 
3767
   }
 
3768
 
 
3769
   sub write {
 
3770
       @_ == 2 || croak(q/Usage: $handle->write(buf)/);
 
3771
       my ($self, $buf) = @_;
 
3772
 
 
3773
       my $len = length $buf;
 
3774
       my $off = 0;
 
3775
 
 
3776
       local $SIG{PIPE} = 'IGNORE';
 
3777
 
 
3778
       while () {
 
3779
           $self->can_write
 
3780
             or croak(q/Timed out while waiting for socket to become ready for writing/);
 
3781
           my $r = syswrite($self->{fh}, $buf, $len, $off);
 
3782
           if (defined $r) {
 
3783
               $len -= $r;
 
3784
               $off += $r;
 
3785
               last unless $len > 0;
 
3786
           }
 
3787
           elsif ($! == EPIPE) {
 
3788
               croak(qq/Socket closed by remote server: $!/);
 
3789
           }
 
3790
           elsif ($! != EINTR) {
 
3791
               croak(qq/Could not write to socket: '$!'/);
 
3792
           }
 
3793
       }
 
3794
       return $off;
 
3795
   }
 
3796
 
 
3797
   sub read {
 
3798
       @_ == 2 || @_ == 3 || croak(q/Usage: $handle->read(len)/);
 
3799
       my ($self, $len) = @_;
 
3800
 
 
3801
       my $buf  = '';
 
3802
       my $got = length $self->{rbuf};
 
3803
 
 
3804
       if ($got) {
 
3805
           my $take = ($got < $len) ? $got : $len;
 
3806
           $buf  = substr($self->{rbuf}, 0, $take, '');
 
3807
           $len -= $take;
 
3808
       }
 
3809
 
 
3810
       while ($len > 0) {
 
3811
           $self->can_read
 
3812
             or croak(q/Timed out while waiting for socket to become ready for reading/);
 
3813
           my $r = sysread($self->{fh}, $buf, $len, length $buf);
 
3814
           if (defined $r) {
 
3815
               last unless $r;
 
3816
               $len -= $r;
 
3817
           }
 
3818
           elsif ($! != EINTR) {
 
3819
               croak(qq/Could not read from socket: '$!'/);
 
3820
           }
 
3821
       }
 
3822
       if ($len) {
 
3823
           croak(q/Unexpected end of stream/);
 
3824
       }
 
3825
       return $buf;
 
3826
   }
 
3827
 
 
3828
   sub readline {
 
3829
       @_ == 1 || croak(q/Usage: $handle->readline()/);
 
3830
       my ($self) = @_;
 
3831
 
 
3832
       while () {
 
3833
           if ($self->{rbuf} =~ s/\A ([^\x0D\x0A]* \x0D?\x0A)//x) {
 
3834
               return $1;
 
3835
           }
 
3836
           $self->can_read
 
3837
             or croak(q/Timed out while waiting for socket to become ready for reading/);
 
3838
           my $r = sysread($self->{fh}, $self->{rbuf}, BUFSIZE, length $self->{rbuf});
 
3839
           if (defined $r) {
 
3840
               last unless $r;
 
3841
           }
 
3842
           elsif ($! != EINTR) {
 
3843
               croak(qq/Could not read from socket: '$!'/);
 
3844
           }
 
3845
       }
 
3846
       croak(q/Unexpected end of stream while looking for line/);
 
3847
   }
 
3848
 
 
3849
   sub read_header_lines {
 
3850
       @_ == 1 || @_ == 2 || croak(q/Usage: $handle->read_header_lines([headers])/);
 
3851
       my ($self, $headers) = @_;
 
3852
       $headers ||= {};
 
3853
       my $lines   = 0;
 
3854
       my $val;
 
3855
 
 
3856
       while () {
 
3857
            my $line = $self->readline;
 
3858
 
 
3859
            if ($line =~ /\A ([^\x00-\x1F\x7F:]+) : [\x09\x20]* ([^\x0D\x0A]*)/x) {
 
3860
                my ($field_name) = lc $1;
 
3861
                $val = \($headers->{$field_name} = $2);
 
3862
            }
 
3863
            elsif ($line =~ /\A [\x09\x20]+ ([^\x0D\x0A]*)/x) {
 
3864
                $val
 
3865
                  or croak(q/Unexpected header continuation line/);
 
3866
                next unless length $1;
 
3867
                $$val .= ' ' if length $$val;
 
3868
                $$val .= $1;
 
3869
            }
 
3870
            elsif ($line =~ /\A \x0D?\x0A \z/x) {
 
3871
               last;
 
3872
            }
 
3873
            else {
 
3874
               croak(q/Malformed header line: / . $Printable->($line));
 
3875
            }
 
3876
       }
 
3877
       return $headers;
 
3878
   }
 
3879
 
 
3880
   sub write_header_lines {
 
3881
       (@_ == 2 && ref $_[1] eq 'HASH') || croak(q/Usage: $handle->write_header_lines(headers)/);
 
3882
       my($self, $headers) = @_;
 
3883
 
 
3884
       my $buf = '';
 
3885
       while (my ($k, $v) = each %$headers) {
 
3886
           my $field_name = lc $k;
 
3887
            $field_name =~ /\A [\x21\x23-\x27\x2A\x2B\x2D\x2E\x30-\x39\x41-\x5A\x5E-\x7A\x7C\x7E]+ \z/x
 
3888
               or croak(q/Invalid HTTP header field name: / . $Printable->($field_name));
 
3889
            $field_name =~ s/\b(\w)/\u$1/g;
 
3890
            $buf .= "$field_name: $v\x0D\x0A";
 
3891
       }
 
3892
       $buf .= "\x0D\x0A";
 
3893
       return $self->write($buf);
 
3894
   }
 
3895
 
 
3896
   sub read_content_body {
 
3897
       @_ == 3 || @_ == 4 || croak(q/Usage: $handle->read_content_body(callback, response, [read_length])/);
 
3898
       my ($self, $cb, $response, $len) = @_;
 
3899
       $len ||= $response->{headers}{'content-length'};
 
3900
 
 
3901
       croak("No content-length in the returned response, and this "
 
3902
           . "UA doesn't implement chunking") unless defined $len;
 
3903
 
 
3904
       while ($len > 0) {
 
3905
           my $read = ($len > BUFSIZE) ? BUFSIZE : $len;
 
3906
           $cb->($self->read($read), $response);
 
3907
           $len -= $read;
 
3908
       }
 
3909
 
 
3910
       return;
 
3911
   }
 
3912
 
 
3913
   sub write_content_body {
 
3914
       @_ == 2 || croak(q/Usage: $handle->write_content_body(request)/);
 
3915
       my ($self, $request) = @_;
 
3916
       my ($len, $content_length) = (0, $request->{headers}{'content-length'});
 
3917
 
 
3918
       $len += $self->write($request->{content});
 
3919
 
 
3920
       $len == $content_length
 
3921
         or croak(qq/Content-Length missmatch (got: $len expected: $content_length)/);
 
3922
 
 
3923
       return $len;
 
3924
   }
 
3925
 
 
3926
   sub read_response_header {
 
3927
       @_ == 1 || croak(q/Usage: $handle->read_response_header()/);
 
3928
       my ($self) = @_;
 
3929
 
 
3930
       my $line = $self->readline;
 
3931
 
 
3932
       $line =~ /\A (HTTP\/(0*\d+\.0*\d+)) [\x09\x20]+ ([0-9]{3}) [\x09\x20]+ ([^\x0D\x0A]*) \x0D?\x0A/x
 
3933
         or croak(q/Malformed Status-Line: / . $Printable->($line));
 
3934
 
 
3935
       my ($protocol, $version, $status, $reason) = ($1, $2, $3, $4);
 
3936
 
 
3937
       return {
 
3938
           status   => $status,
 
3939
           reason   => $reason,
 
3940
           headers  => $self->read_header_lines,
 
3941
           protocol => $protocol,
 
3942
       };
 
3943
   }
 
3944
 
 
3945
   sub write_request_header {
 
3946
       @_ == 4 || croak(q/Usage: $handle->write_request_header(method, request_uri, headers)/);
 
3947
       my ($self, $method, $request_uri, $headers) = @_;
 
3948
 
 
3949
       return $self->write("$method $request_uri HTTP/1.1\x0D\x0A")
 
3950
            + $self->write_header_lines($headers);
 
3951
   }
 
3952
 
 
3953
   sub _do_timeout {
 
3954
       my ($self, $type, $timeout) = @_;
 
3955
       $timeout = $self->{timeout}
 
3956
           unless defined $timeout && $timeout >= 0;
 
3957
 
 
3958
       my $fd = fileno $self->{fh};
 
3959
       defined $fd && $fd >= 0
 
3960
         or croak(q/select(2): 'Bad file descriptor'/);
 
3961
 
 
3962
       my $initial = time;
 
3963
       my $pending = $timeout;
 
3964
       my $nfound;
 
3965
 
 
3966
       vec(my $fdset = '', $fd, 1) = 1;
 
3967
 
 
3968
       while () {
 
3969
           $nfound = ($type eq 'read')
 
3970
               ? select($fdset, undef, undef, $pending)
 
3971
               : select(undef, $fdset, undef, $pending) ;
 
3972
           if ($nfound == -1) {
 
3973
               $! == EINTR
 
3974
                 or croak(qq/select(2): '$!'/);
 
3975
               redo if !$timeout || ($pending = $timeout - (time - $initial)) > 0;
 
3976
               $nfound = 0;
 
3977
           }
 
3978
           last;
 
3979
       }
 
3980
       $! = 0;
 
3981
       return $nfound;
 
3982
   }
 
3983
 
 
3984
   sub can_read {
 
3985
       @_ == 1 || @_ == 2 || croak(q/Usage: $handle->can_read([timeout])/);
 
3986
       my $self = shift;
 
3987
       return $self->_do_timeout('read', @_)
 
3988
   }
 
3989
 
 
3990
   sub can_write {
 
3991
       @_ == 1 || @_ == 2 || croak(q/Usage: $handle->can_write([timeout])/);
 
3992
       my $self = shift;
 
3993
       return $self->_do_timeout('write', @_)
 
3994
   }
 
3995
}  # HTTP::Micro::Handle
3993
3996
 
3994
3997
my $prog = <<'EOP';
3995
3998
BEGIN {
4010
4013
   }
4011
4014
}
4012
4015
{
 
4016
   use Carp qw(croak);
4013
4017
   my %dispatcher = (
4014
4018
      issuer =>  sub { Net::SSLeay::X509_NAME_oneline( Net::SSLeay::X509_get_issuer_name( shift )) },
4015
4019
      subject => sub { Net::SSLeay::X509_NAME_oneline( Net::SSLeay::X509_get_subject_name( shift )) },
4165
4169
}
4166
4170
 
4167
4171
1;
4168
 
}
4169
4172
# ###########################################################################
4170
 
# End HTTPMicro package
 
4173
# End HTTP::Micro package
4171
4174
# ###########################################################################
4172
4175
 
4173
4176
# ###########################################################################
4201
4204
 
4202
4205
eval {
4203
4206
   require Percona::Toolkit;
4204
 
   require HTTPMicro;
 
4207
   require HTTP::Micro;
4205
4208
};
4206
4209
 
4207
4210
{
4432
4435
   my $url       = $args{url};
4433
4436
   my $instances = $args{instances};
4434
4437
 
4435
 
   my $ua = $args{ua} || HTTPMicro->new( timeout => 3 );
 
4438
   my $ua = $args{ua} || HTTP::Micro->new( timeout => 3 );
4436
4439
 
4437
4440
   my $response = $ua->request('GET', $url);
4438
4441
   PTDEBUG && _d('Server response:', Dumper($response));
4546
4549
   perl_version        => \&get_perl_version,
4547
4550
   perl_module_version => \&get_perl_module_version,
4548
4551
   mysql_variable      => \&get_mysql_variable,
4549
 
   bin_version         => \&get_bin_version,
4550
4552
);
4551
4553
 
4552
4554
sub valid_item {
4729
4731
   return \%version_for;
4730
4732
}
4731
4733
 
4732
 
sub get_bin_version {
4733
 
   my (%args) = @_;
4734
 
   my $item = $args{item};
4735
 
   my $cmd  = $item->{item};
4736
 
   return unless $cmd;
4737
 
 
4738
 
   my $sanitized_command = File::Basename::basename($cmd);
4739
 
   PTDEBUG && _d('cmd:', $cmd, 'sanitized:', $sanitized_command);
4740
 
   return if $sanitized_command !~ /\A[a-zA-Z0-9_-]+\z/;
4741
 
 
4742
 
   my $output = `$sanitized_command --version 2>&1`;
4743
 
   PTDEBUG && _d('output:', $output);
4744
 
 
4745
 
   my ($version) = $output =~ /v?([0-9]+\.[0-9]+(?:\.[\w-]+)?)/;
4746
 
 
4747
 
   PTDEBUG && _d('Version for', $sanitized_command, '=', $version);
4748
 
   return $version;
4749
 
}
4750
 
 
4751
4734
sub _d {
4752
4735
   my ($package, undef, $line) = caller 0;
4753
4736
   @_ = map { (my $temp = $_) =~ s/\n/\n# /g; $temp; }