312 lines
7.3 KiB
Perl
312 lines
7.3 KiB
Perl
# IO::Socket::INET.pm
|
|
#
|
|
# Copyright (c) 1997-8 Graham Barr <gbarr@pobox.com>. All rights reserved.
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the same terms as Perl itself.
|
|
|
|
package IO::Socket::INET;
|
|
|
|
use strict;
|
|
our(@ISA, $VERSION);
|
|
use IO::Socket;
|
|
use Socket;
|
|
use Carp;
|
|
use Exporter;
|
|
use Errno;
|
|
|
|
@ISA = qw(IO::Socket);
|
|
$VERSION = "1.35";
|
|
|
|
my $EINVAL = exists(&Errno::EINVAL) ? Errno::EINVAL() : 1;
|
|
|
|
IO::Socket::INET->register_domain( AF_INET );
|
|
|
|
my %socket_type = ( tcp => SOCK_STREAM,
|
|
udp => SOCK_DGRAM,
|
|
icmp => SOCK_RAW
|
|
);
|
|
my %proto_number;
|
|
$proto_number{tcp} = Socket::IPPROTO_TCP() if defined &Socket::IPPROTO_TCP;
|
|
$proto_number{udp} = Socket::IPPROTO_UDP() if defined &Socket::IPPROTO_UDP;
|
|
$proto_number{icmp} = Socket::IPPROTO_ICMP() if defined &Socket::IPPROTO_ICMP;
|
|
my %proto_name = reverse %proto_number;
|
|
|
|
sub new {
|
|
my $class = shift;
|
|
unshift(@_, "PeerAddr") if @_ == 1;
|
|
return $class->SUPER::new(@_);
|
|
}
|
|
|
|
sub _cache_proto {
|
|
my @proto = @_;
|
|
for (map lc($_), $proto[0], split(' ', $proto[1])) {
|
|
$proto_number{$_} = $proto[2];
|
|
}
|
|
$proto_name{$proto[2]} = $proto[0];
|
|
}
|
|
|
|
sub _get_proto_number {
|
|
my $name = lc(shift);
|
|
return undef unless defined $name;
|
|
return $proto_number{$name} if exists $proto_number{$name};
|
|
|
|
my @proto = eval { getprotobyname($name) };
|
|
return undef unless @proto;
|
|
_cache_proto(@proto);
|
|
|
|
return $proto[2];
|
|
}
|
|
|
|
sub _get_proto_name {
|
|
my $num = shift;
|
|
return undef unless defined $num;
|
|
return $proto_name{$num} if exists $proto_name{$num};
|
|
|
|
my @proto = eval { getprotobynumber($num) };
|
|
return undef unless @proto;
|
|
_cache_proto(@proto);
|
|
|
|
return $proto[0];
|
|
}
|
|
|
|
sub _sock_info {
|
|
my($addr,$port,$proto) = @_;
|
|
my $origport = $port;
|
|
my @serv = ();
|
|
|
|
$port = $1
|
|
if(defined $addr && $addr =~ s,:([\w\(\)/]+)$,,);
|
|
|
|
if(defined $proto && $proto =~ /\D/) {
|
|
my $num = _get_proto_number($proto);
|
|
unless (defined $num) {
|
|
$@ = "Bad protocol '$proto'";
|
|
return;
|
|
}
|
|
$proto = $num;
|
|
}
|
|
|
|
if(defined $port) {
|
|
my $defport = ($port =~ s,\((\d+)\)$,,) ? $1 : undef;
|
|
my $pnum = ($port =~ m,^(\d+)$,)[0];
|
|
|
|
@serv = getservbyname($port, _get_proto_name($proto) || "")
|
|
if ($port =~ m,\D,);
|
|
|
|
$port = $serv[2] || $defport || $pnum;
|
|
unless (defined $port) {
|
|
$@ = "Bad service '$origport'";
|
|
return;
|
|
}
|
|
|
|
$proto = _get_proto_number($serv[3]) if @serv && !$proto;
|
|
}
|
|
|
|
return ($addr || undef,
|
|
$port || undef,
|
|
$proto || undef
|
|
);
|
|
}
|
|
|
|
sub _error {
|
|
my $sock = shift;
|
|
my $err = shift;
|
|
{
|
|
local($!);
|
|
my $title = ref($sock).": ";
|
|
$@ = join("", $_[0] =~ /^$title/ ? "" : $title, @_);
|
|
$sock->close()
|
|
if(defined fileno($sock));
|
|
}
|
|
$! = $err;
|
|
return undef;
|
|
}
|
|
|
|
sub _get_addr {
|
|
my($sock,$addr_str, $multi) = @_;
|
|
my @addr;
|
|
if ($multi && $addr_str !~ /^\d+(?:\.\d+){3}$/) {
|
|
(undef, undef, undef, undef, @addr) = gethostbyname($addr_str);
|
|
} else {
|
|
my $h = inet_aton($addr_str);
|
|
push(@addr, $h) if defined $h;
|
|
}
|
|
@addr;
|
|
}
|
|
|
|
sub configure {
|
|
my($sock,$arg) = @_;
|
|
my($lport,$rport,$laddr,$raddr,$proto,$type);
|
|
|
|
$arg->{LocalAddr} = $arg->{LocalHost}
|
|
if exists $arg->{LocalHost} && !exists $arg->{LocalAddr};
|
|
|
|
($laddr,$lport,$proto) = _sock_info($arg->{LocalAddr},
|
|
$arg->{LocalPort},
|
|
$arg->{Proto})
|
|
or return _error($sock, $!, $@);
|
|
|
|
$laddr = defined $laddr ? inet_aton($laddr)
|
|
: INADDR_ANY;
|
|
|
|
return _error($sock, $EINVAL, "Bad hostname '",$arg->{LocalAddr},"'")
|
|
unless(defined $laddr);
|
|
|
|
$arg->{PeerAddr} = $arg->{PeerHost}
|
|
if exists $arg->{PeerHost} && !exists $arg->{PeerAddr};
|
|
|
|
unless(exists $arg->{Listen}) {
|
|
($raddr,$rport,$proto) = _sock_info($arg->{PeerAddr},
|
|
$arg->{PeerPort},
|
|
$proto)
|
|
or return _error($sock, $!, $@);
|
|
}
|
|
|
|
$proto ||= _get_proto_number('tcp');
|
|
|
|
$type = $arg->{Type} || $socket_type{lc _get_proto_name($proto)};
|
|
|
|
my @raddr = ();
|
|
|
|
if(defined $raddr) {
|
|
@raddr = $sock->_get_addr($raddr, $arg->{MultiHomed});
|
|
return _error($sock, $EINVAL, "Bad hostname '",$arg->{PeerAddr},"'")
|
|
unless @raddr;
|
|
}
|
|
|
|
while(1) {
|
|
|
|
$sock->socket(AF_INET, $type, $proto) or
|
|
return _error($sock, $!, "$!");
|
|
|
|
if (defined $arg->{Blocking}) {
|
|
defined $sock->blocking($arg->{Blocking})
|
|
or return _error($sock, $!, "$!");
|
|
}
|
|
|
|
if ($arg->{Reuse} || $arg->{ReuseAddr}) {
|
|
$sock->sockopt(SO_REUSEADDR,1) or
|
|
return _error($sock, $!, "$!");
|
|
}
|
|
|
|
if ($arg->{ReusePort}) {
|
|
$sock->sockopt(SO_REUSEPORT,1) or
|
|
return _error($sock, $!, "$!");
|
|
}
|
|
|
|
if ($arg->{Broadcast}) {
|
|
$sock->sockopt(SO_BROADCAST,1) or
|
|
return _error($sock, $!, "$!");
|
|
}
|
|
|
|
if($lport || ($laddr ne INADDR_ANY) || exists $arg->{Listen}) {
|
|
$sock->bind($lport || 0, $laddr) or
|
|
return _error($sock, $!, "$!");
|
|
}
|
|
|
|
if(exists $arg->{Listen}) {
|
|
$sock->listen($arg->{Listen} || 5) or
|
|
return _error($sock, $!, "$!");
|
|
last;
|
|
}
|
|
|
|
# don't try to connect unless we're given a PeerAddr
|
|
last unless exists($arg->{PeerAddr});
|
|
|
|
$raddr = shift @raddr;
|
|
|
|
return _error($sock, $EINVAL, 'Cannot determine remote port')
|
|
unless($rport || $type == SOCK_DGRAM || $type == SOCK_RAW);
|
|
|
|
last
|
|
unless($type == SOCK_STREAM || defined $raddr);
|
|
|
|
return _error($sock, $EINVAL, "Bad hostname '",$arg->{PeerAddr},"'")
|
|
unless defined $raddr;
|
|
|
|
# my $timeout = ${*$sock}{'io_socket_timeout'};
|
|
# my $before = time() if $timeout;
|
|
|
|
undef $@;
|
|
if ($sock->connect(pack_sockaddr_in($rport, $raddr))) {
|
|
# ${*$sock}{'io_socket_timeout'} = $timeout;
|
|
return $sock;
|
|
}
|
|
|
|
return _error($sock, $!, $@ || "Timeout")
|
|
unless @raddr;
|
|
|
|
# if ($timeout) {
|
|
# my $new_timeout = $timeout - (time() - $before);
|
|
# return _error($sock,
|
|
# (exists(&Errno::ETIMEDOUT) ? Errno::ETIMEDOUT() : $EINVAL),
|
|
# "Timeout") if $new_timeout <= 0;
|
|
# ${*$sock}{'io_socket_timeout'} = $new_timeout;
|
|
# }
|
|
|
|
}
|
|
|
|
$sock;
|
|
}
|
|
|
|
sub connect {
|
|
@_ == 2 || @_ == 3 or
|
|
croak 'usage: $sock->connect(NAME) or $sock->connect(PORT, ADDR)';
|
|
my $sock = shift;
|
|
return $sock->SUPER::connect(@_ == 1 ? shift : pack_sockaddr_in(@_));
|
|
}
|
|
|
|
sub bind {
|
|
@_ == 2 || @_ == 3 or
|
|
croak 'usage: $sock->bind(NAME) or $sock->bind(PORT, ADDR)';
|
|
my $sock = shift;
|
|
return $sock->SUPER::bind(@_ == 1 ? shift : pack_sockaddr_in(@_))
|
|
}
|
|
|
|
sub sockaddr {
|
|
@_ == 1 or croak 'usage: $sock->sockaddr()';
|
|
my($sock) = @_;
|
|
my $name = $sock->sockname;
|
|
$name ? (sockaddr_in($name))[1] : undef;
|
|
}
|
|
|
|
sub sockport {
|
|
@_ == 1 or croak 'usage: $sock->sockport()';
|
|
my($sock) = @_;
|
|
my $name = $sock->sockname;
|
|
$name ? (sockaddr_in($name))[0] : undef;
|
|
}
|
|
|
|
sub sockhost {
|
|
@_ == 1 or croak 'usage: $sock->sockhost()';
|
|
my($sock) = @_;
|
|
my $addr = $sock->sockaddr;
|
|
$addr ? inet_ntoa($addr) : undef;
|
|
}
|
|
|
|
sub peeraddr {
|
|
@_ == 1 or croak 'usage: $sock->peeraddr()';
|
|
my($sock) = @_;
|
|
my $name = $sock->peername;
|
|
$name ? (sockaddr_in($name))[1] : undef;
|
|
}
|
|
|
|
sub peerport {
|
|
@_ == 1 or croak 'usage: $sock->peerport()';
|
|
my($sock) = @_;
|
|
my $name = $sock->peername;
|
|
$name ? (sockaddr_in($name))[0] : undef;
|
|
}
|
|
|
|
sub peerhost {
|
|
@_ == 1 or croak 'usage: $sock->peerhost()';
|
|
my($sock) = @_;
|
|
my $addr = $sock->peeraddr;
|
|
$addr ? inet_ntoa($addr) : undef;
|
|
}
|
|
|
|
1;
|
|
|
|
__END__
|
|
|