| IO-Socket-Socks documentation | Contained in the IO-Socket-Socks distribution. |
IO::Socket::Socks - Provides a way to create socks client or server both 4 and 5 version.
use IO::Socket::Socks;
my $socks = new IO::Socket::Socks(ProxyAddr=>"proxy host",
ProxyPort=>"proxy port",
ConnectAddr=>"remote host",
ConnectPort=>"remote port",
);
print $socks "foo\n";
$socks->close();
use IO::Socket::Socks;
my $socks_server = new IO::Socket::Socks(ProxyAddr=>"localhost",
ProxyPort=>"8000",
Listen=>1,
UserAuth=>\&auth,
RequireAuth=>1
);
my $select = new IO::Select($socks_server);
while(1)
{
if ($select->can_read())
{
my $client = $socks_server->accept();
if (!defined($client))
{
print "ERROR: $SOCKS_ERROR\n";
next;
}
my $command = $client->command();
if ($command->[0] == 1) # CONNECT
{
# Handle the CONNECT
$client->command_reply(0, addr, port);
}
...
#read from the client and send to the CONNECT address
...
$client->close();
}
}
sub auth
{
my $user = shift;
my $pass = shift;
return 1 if (($user eq "foo") && ($pass eq "bar"));
return 0;
}
IO::Socket::Socks connects to a SOCKS proxy, tells it to open a connection to a remote host/port when the object is created. The object you receive can be used directly as a socket for sending and receiving data from the remote host. In addition to create socks client this module could be used to create socks server. See examples below.
For complete examples of socks 4/5 client and server see `examples' subdirectory in the distribution.
Creates a new IO::Socket::Socks client object. new_from_socket() is the same as new(), but allows to create object from an existing socket. Both takes the following config hash:
SocksVersion => 4 or 5. Default is 5
Timeout => connect/accept timeout
Blocking => Since IO::Socket::Socks version 0.5 you can perform non-blocking connect/bind by
passing false value for this option. Default is true - blocking. See ready()
below for more details.
SocksResolve => resolve host name to ip by proxy server or
not (will resolve by client). This
overrides value of $SOCKS4_RESOLVE or $SOCKS5_RESOLVE
variable. Boolean.
SocksDebug => This will cause all of the SOCKS traffic to
be presented on the command line in a form
similar to the tables in the RFCs. This overrides value
of $SOCKS_DEBUG variable. Boolean.
ProxyAddr => Hostname of the proxy
ProxyPort => Port of the proxy
ConnectAddr => Hostname of the remote machine
ConnectPort => Port of the remote machine
BindAddr => Hostname of the remote machine which will
connect to the proxy server after bind request
BindPort => Port of the remote machine which will
connect to the proxy server after bind request
UdpAddr => Associate UDP socket on the server with this client
hostname
UdpPort => Associate UDP socket on the server with this client
port
AuthType => What kind of authentication to support:
none - no authentication (default)
userpass - Username/Password. For socks5
proxy only.
RequireAuth => Do not send ANON as a valid auth mechanism.
For socks5 proxy only
Username => For socks5 if AuthType is set to userpass, then
you must provide a username. For socks4 proxy with
this option you can specify userid.
Password => If AuthType is set to userpass, then you must
provide a password. For socks5 proxy only.
The following options should be specified:
ProxyAddr and ProxyPort ConnectAddr and ConnectPort or BindAddr and BindPort or UdpAddr and UdpPort
Other options are facultative.
Returns true when socket becomes ready to transfer data (socks handshake done), false otherwise. This is useful for non-blocking connect/bind. When this method returns false value you can determine what socks handshake need for with $SOCKS_ERROR variable. It may need for read, then $SOCKS_ERROR will be SOCKS_WANT_READ or need for write, then it will be SOCKS_WANT_WRITE.
Example:
use IO::Socket::Socks;
use IO::Select;
my $sock = IO::Socket::Socks->new(
ProxyAddr => 'localhost', ProxyPort => 1080, ConnectAddr => 'mail.com', ConnectPort => 80, Blocking => 0
) or die $SOCKS_ERROR;
my $sel = IO::Select->new($sock);
until ($sock->ready) {
if ($SOCKS_ERROR == SOCKS_WANT_READ) {
$sel->can_read();
}
elsif ($SOCKS_ERROR == SOCKS_WANT_WRITE) {
$sel->can_write();
}
else {
die $SOCKS_ERROR;
}
}
# you may want to return socket to blocking state by $sock->blocking(1)
$sock->syswrite("I am ready");
Accept an incoming connection after bind request. On failed returns undef. On success returns socket. No new socket created, returned socket is same on which this method was called. Because accept(2) is not invoked on the client side, socks server calls accept(2) and proxify all traffic via socket opened by client bind request. You can call accept only once on IO::Socket::Socks client socket.
Allows to execute socks command on already opened socket. Thus you can create socks chain. For example see EXAMPLES section.
%cfg is like hash in the constructor. Only options listed below makes sence:
ConnectAddr ConnectPort BindAddr BindPort UdpAddr UdpPort SocksVersion SocksDebug SocksResolve AuthType RequireAuth Username Password AuthMethods
Values of the other options (Timeout for example) inherited from the constructor. Options like ProxyAddr and ProxyPort are not included.
Return (host, port) of the remote host after connect/accept or socks server (host, port) after bind/udpassoc.
Creates a new IO::Socket::Socks server object. new_from_socket() is the same as new(), but allows to create object from an existing socket. Both takes the following config hash:
SocksVersion => 4 for socks v4, 5 for socks v5. Default is 5
Timeout => Timeout value for various operations
SocksResolve => For socks v5: return destination address to the client
in form of 4 bytes if true, otherwise in form of host
length and host name.
For socks v4: allow use socks4a protocol extension if
true and not otherwise.
This overrides value of $SOCKS4_RESOLVE or $SOCKS5_RESOLVE.
SocksDebug => This will cause all of the SOCKS traffic to
be presented on the command line in a form
similar to the tables in the RFCs. This overrides value
of $SOCKS_DEBUG variable. Boolean.
ProxyAddr => Local host bind address
ProxyPort => Local host bind port
UserAuth => Reference to a function that returns 1 if client
allowed to use socks server, 0 otherwise. For
socks5 proxy it takes login and password as
arguments. For socks4 argument is userid.
RequireAuth => Not allow anonymous access for socks5 proxy.
Listen => Same as IO::Socket::INET listen option. Should be
specified as number > 0.
The following options should be specified:
Listen ProxyAddr ProxyPort
Other options are facultative.
Accept an incoming connection and return a new IO::Socket::Socks object that represents that connection. You must call command() on this to find out what the incoming connection wants you to do, and then call command_reply() to send back the reply.
After you call accept() the client has sent the command they want you to process. This function should be called on the socket returned by accept(). It returns a reference to an array with the following format:
[ COMMAND, HOST, PORT ]
After you call command() the client needs to be told what the result is. The REPLY CODE is as follows (integer value):
For socks v4 90: request granted 91: request rejected or failed 92: request rejected becasue SOCKS server cannot connect to identd on the client 93: request rejected because the client program and identd report different user-ids For socks v5 0: Success 1: General Failure 2: Connection Not Allowed 3: Network Unreachable 4: Host Unreachable 5: Connection Refused 6: TTL Expired 7: Command Not Supported 8: Address Not Supported
You can also use module constans. See below. HOST and PORT are the resulting host and port that you use for the command.
This scalar behaves like $! in that if undef is returned, this variable should contain a string reason for the error. Imported by default.
If this variable has true value resolving of host names will be done by proxy server, otherwise resolving will be done locally. Resolving host by socks proxy version 4 is extension to the protocol also known as socks4a. So, only socks4a proxy supports resolving of hostnames. Default value of this variable is false. This variable is not importable. See also `SocksResolve' parameter in the constructor.
If this variable has true value resolving of host names will be done by proxy server, otherwise resolving will be done locally. Note: some bugous socks5 servers doesn't support resolving of host names. Default value is true. This variable is not importable. See also `SocksResolve' parameter in the constructor.
Default value is $ENV{SOCKS_DEBUG}. If this variable has true value and no SocksDebug option in the constructor specified, then SocksDebug will has true value. This variable is not importable.
The following constants could be imported manually or using `:constants' tag:
SOCKS5_VER SOCKS4_VER ADDR_IPV4 ADDR_DOMAINNAME ADDR_IPV6 CMD_CONNECT CMD_BIND CMD_UDPASSOC AUTHMECH_ANON AUTHMECH_USERPASS AUTHMECH_INVALID AUTHREPLY_SUCCESS AUTHREPLY_FAILURE REPLY_SUCCESS REPLY_GENERAL_FAILURE REPLY_CONN_NOT_ALLOWED REPLY_NETWORK_UNREACHABLE REPLY_HOST_UNREACHABLE REPLY_CONN_REFUSED REPLY_TTL_EXPIRED REPLY_CMD_NOT_SUPPORTED REPLY_ADDR_NOT_SUPPORTED REQUEST_GRANTED REQUEST_FAILED REQUEST_REJECTED_IDENTD REQUEST_REJECTED_USERID SOCKS_WANT_READ SOCKS_WANT_WRITE
SOCKS_WANT_READ and SOCKS_WANT_WRITE are imported by default.
The following options are not implemented and not planned:
However patches are welcome.
Original author is Ryan Eatmon
Now maintained by Oleg G <oleg@cpan.org>
This module is free software, you can redistribute it and/or modify it under the terms of LGPL.
| IO-Socket-Socks documentation | Contained in the IO-Socket-Socks distribution. |
############################################################################## # # This library is free software; you can redistribute it and/or # modify it under the terms of the GNU Library General Public # License as published by the Free Software Foundation; either # version 2 of the License, or (at your option) any later version. # # This library is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU # Library General Public License for more details. # # You should have received a copy of the GNU Library General Public # License along with this library; if not, write to the # Free Software Foundation, Inc., 59 Temple Place - Suite 330, # Boston, MA 02111-1307, USA. # # Copyright (C) 2003 Ryan Eatmon # Copyright (C) 2010-2011 Oleg G # ############################################################################## package IO::Socket::Socks; use strict; use IO::Socket; use IO::Select; use Errno qw(EWOULDBLOCK EAGAIN); use Carp; use vars qw( @ISA @EXPORT @EXPORT_OK %EXPORT_TAGS $VERSION $SOCKS_ERROR $SOCKS5_RESOLVE $SOCKS4_RESOLVE $SOCKS_DEBUG %CODES ); require Exporter; use constant { SOCKS_WANT_READ => 20, SOCKS_WANT_WRITE => 21, }; @ISA = qw(Exporter IO::Socket::INET); @EXPORT = qw( $SOCKS_ERROR SOCKS_WANT_READ SOCKS_WANT_WRITE ); @EXPORT_OK = qw( SOCKS5_VER SOCKS4_VER ADDR_IPV4 ADDR_DOMAINNAME ADDR_IPV6 CMD_CONNECT CMD_BIND CMD_UDPASSOC AUTHMECH_ANON AUTHMECH_USERPASS AUTHMECH_INVALID AUTHREPLY_SUCCESS AUTHREPLY_FAILURE REPLY_SUCCESS REPLY_GENERAL_FAILURE REPLY_CONN_NOT_ALLOWED REPLY_NETWORK_UNREACHABLE REPLY_HOST_UNREACHABLE REPLY_CONN_REFUSED REPLY_TTL_EXPIRED REPLY_CMD_NOT_SUPPORTED REPLY_ADDR_NOT_SUPPORTED REQUEST_GRANTED REQUEST_FAILED REQUEST_REJECTED_IDENTD REQUEST_REJECTED_USERID ); %EXPORT_TAGS = (constants => ['SOCKS_WANT_READ', 'SOCKS_WANT_WRITE', @EXPORT_OK]); $VERSION = '0.5'; $SOCKS5_RESOLVE = 1; $SOCKS4_RESOLVE = 0; $SOCKS_DEBUG = $ENV{SOCKS_DEBUG}; use constant { SOCKS5_VER => 5, SOCKS4_VER => 4, ADDR_IPV4 => 1, ADDR_DOMAINNAME => 3, ADDR_IPV6 => 4, CMD_CONNECT => 1, CMD_BIND => 2, CMD_UDPASSOC => 3, AUTHMECH_ANON => 0, #AUTHMECH_GSSAPI => 1, AUTHMECH_USERPASS => 2, AUTHMECH_INVALID => 255, AUTHREPLY_SUCCESS => 0, AUTHREPLY_FAILURE => 1, }; $CODES{AUTHMECH}->[AUTHMECH_INVALID] = "No valid auth mechanisms"; $CODES{AUTHREPLY}->[AUTHREPLY_FAILURE] = "Failed to authenticate"; # socks5 use constant { REPLY_SUCCESS => 0, REPLY_GENERAL_FAILURE => 1, REPLY_CONN_NOT_ALLOWED => 2, REPLY_NETWORK_UNREACHABLE => 3, REPLY_HOST_UNREACHABLE => 4, REPLY_CONN_REFUSED => 5, REPLY_TTL_EXPIRED => 6, REPLY_CMD_NOT_SUPPORTED => 7, REPLY_ADDR_NOT_SUPPORTED => 8, }; $CODES{REPLY}->{&REPLY_SUCCESS} = "Success"; $CODES{REPLY}->{&REPLY_GENERAL_FAILURE} = "General failure"; $CODES{REPLY}->{&REPLY_CONN_NOT_ALLOWED} = "Not allowed"; $CODES{REPLY}->{&REPLY_NETWORK_UNREACHABLE} = "Network unreachable"; $CODES{REPLY}->{&REPLY_HOST_UNREACHABLE} = "Host unreachable"; $CODES{REPLY}->{&REPLY_CONN_REFUSED} = "Connection refused"; $CODES{REPLY}->{&REPLY_TTL_EXPIRED} = "TTL expired"; $CODES{REPLY}->{&REPLY_CMD_NOT_SUPPORTED} = "Command not supported"; $CODES{REPLY}->{&REPLY_ADDR_NOT_SUPPORTED} = "Address not supported"; # socks4 use constant { REQUEST_GRANTED => 90, REQUEST_FAILED => 91, REQUEST_REJECTED_IDENTD => 92, REQUEST_REJECTED_USERID => 93, }; $CODES{REPLY}->{&REQUEST_GRANTED} = "request granted"; $CODES{REPLY}->{&REQUEST_FAILED} = "request rejected or failed"; $CODES{REPLY}->{&REQUEST_REJECTED_IDENTD} = "request rejected becasue SOCKS server cannot connect to identd on the client"; $CODES{REPLY}->{&REQUEST_REJECTED_USERID} = "request rejected because the client program and identd report different user-ids"; # queue use constant { Q_SUB => 0, Q_ARGS => 1, Q_BUF => 2, Q_READS => 3, Q_SENDS => 4, Q_DEBUGS => 5, }; #------------------------------------------------------------------------------ # sub new is handled by IO::Socket::INET #------------------------------------------------------------------------------ sub new_from_socket { my ($class, $sock, %arg) = @_; bless $sock, $class; $sock->autoflush(1); ${*$sock}{'io_socket_timeout'} = delete $arg{Timeout}; return scalar(%arg) ? $sock->configure(\%arg) : $sock; } ############################################################################### # # configure - read in the config hash and populate the object. # ############################################################################### sub configure { my $self = shift; my $args = shift; $self->_configure($args) or return; ${*$self}->{SOCKS}->{ProxyAddr} = (exists($args->{ProxyAddr}) ? delete($args->{ProxyAddr}) : undef ); ${*$self}->{SOCKS}->{ProxyPort} = (exists($args->{ProxyPort}) ? delete($args->{ProxyPort}) : undef ); ${*$self}->{SOCKS}->{COMMAND} = []; if (exists($args->{Listen})) { $args->{LocalAddr} = ${*$self}->{SOCKS}->{ProxyAddr}; $args->{LocalPort} = ${*$self}->{SOCKS}->{ProxyPort}; $args->{Reuse} = 1; ${*$self}->{SOCKS}->{Listen} = 1; } elsif(${*$self}->{SOCKS}->{ProxyAddr} && ${*$self}->{SOCKS}->{ProxyPort}) { $args->{PeerAddr} = ${*$self}->{SOCKS}->{ProxyAddr}; $args->{PeerPort} = ${*$self}->{SOCKS}->{ProxyPort}; } unless(defined ${*$self}->{SOCKS}->{TCP}) { $args->{Proto} = "tcp"; $args->{Type} = SOCK_STREAM; } elsif(! defined $args->{Proto}) { $args->{Proto} = "udp"; $args->{Type} = SOCK_DGRAM; } $self->SUPER::configure($args); } ############################################################################### # # _configure - reusable configure operations # ############################################################################### sub _configure { my $self = shift; my $args = shift; ${*$self}->{SOCKS}->{Version} = (exists($args->{SocksVersion}) ? ($args->{SocksVersion} == 4 || $args->{SocksVersion} == 5 ? delete($args->{SocksVersion}) : croak("Unsupported socks version specified. Should be 4 or 5") ) : 5 ); ${*$self}->{SOCKS}->{AuthType} = (exists($args->{AuthType}) ? delete($args->{AuthType}) : "none" ); ${*$self}->{SOCKS}->{RequireAuth} = (exists($args->{RequireAuth}) ? delete($args->{RequireAuth}) : 0 ); ${*$self}->{SOCKS}->{UserAuth} = (exists($args->{UserAuth}) ? delete($args->{UserAuth}) : undef ); ${*$self}->{SOCKS}->{Username} = (exists($args->{Username}) ? delete($args->{Username}) : ((${*$self}->{SOCKS}->{AuthType} eq "none") ? undef : croak("If you set AuthType to userpass, then you must provide a username.") ) ); ${*$self}->{SOCKS}->{Password} = (exists($args->{Password}) ? delete($args->{Password}) : ((${*$self}->{SOCKS}->{AuthType} eq "none") ? undef : croak("If you set AuthType to userpass, then you must provide a password.") ) ); ${*$self}->{SOCKS}->{Debug} = (exists($args->{SocksDebug}) ? delete($args->{SocksDebug}) : $SOCKS_DEBUG ); ${*$self}->{SOCKS}->{Resolve} = (exists($args->{SocksResolve}) ? delete($args->{SocksResolve}) : undef ); ${*$self}->{SOCKS}->{AuthMethods} = [0,0,0]; ${*$self}->{SOCKS}->{AuthMethods}->[AUTHMECH_ANON] = 1 unless ${*$self}->{SOCKS}->{RequireAuth}; #${*$self}->{SOCKS}->{AuthMethods}->[AUTHMECH_GSSAPI] = 1 # if (${*$self}->{SOCKS}->{AuthType} eq "gssapi"); ${*$self}->{SOCKS}->{AuthMethods}->[AUTHMECH_USERPASS] = 1 if ((!exists($args->{Listen}) && (${*$self}->{SOCKS}->{AuthType} eq "userpass")) || (exists($args->{Listen}) && defined(${*$self}->{SOCKS}->{UserAuth}))); if(exists($args->{BindAddr}) && exists($args->{BindPort})) { ${*$self}->{SOCKS}->{CmdAddr} = delete($args->{BindAddr}); ${*$self}->{SOCKS}->{CmdPort} = delete($args->{BindPort}); ${*$self}->{SOCKS}->{Bind} = 1; } elsif(exists($args->{UdpAddr}) && exists($args->{UdpPort})) { if(${*$self}->{SOCKS}->{Version} == 4) { croak("Socks v4 doesn't support UDP association"); } ${*$self}->{SOCKS}->{CmdAddr} = delete($args->{UdpAddr}); ${*$self}->{SOCKS}->{CmdPort} = delete($args->{UdpPort}); $args->{LocalAddr} = ${*$self}->{SOCKS}->{CmdAddr}; $args->{LocalPort} = ${*$self}->{SOCKS}->{CmdPort}; ${*$self}->{SOCKS}->{TCP} = __PACKAGE__->new( # TCP backend for UDP socket Timeout => $args->{Timeout}, Proto => 'tcp' ) or return; } elsif(exists($args->{ConnectAddr}) && exists($args->{ConnectPort})) { ${*$self}->{SOCKS}->{CmdAddr} = delete($args->{ConnectAddr}); ${*$self}->{SOCKS}->{CmdPort} = delete($args->{ConnectPort}); } return 1; } ############################################################################### #+----------------------------------------------------------------------------- #| Connect Functions #+----------------------------------------------------------------------------- ############################################################################### ############################################################################### # # connect - On a configure, connect is called to open the connection. When # we do this we have to talk to the SOCKS proxy, log in, and # connect to the remote host. # ############################################################################### sub connect { my $self = shift; croak("Undefined IO::Socket::Socks object passed to connect.") unless defined($self); #-------------------------------------------------------------------------- # Establish a connection #-------------------------------------------------------------------------- my $sock = defined( ${*$self}->{SOCKS}->{TCP} ) ? ${*$self}->{SOCKS}->{TCP}->SUPER::connect(@_) : $self->SUPER::connect(@_); if (!$sock) { $SOCKS_ERROR = "Connection to proxy failed."; return; } $self->_connect(); } ############################################################################### # # _connect - reusable connect operations # ############################################################################### sub _connect { my $self = shift; ${*$self}->{SOCKS}->{ready} = 0; if(${*$self}->{SOCKS}->{Version} == 4) { ${*$self}->{SOCKS}->{queue} = [ # [sub, [@args], buf, [@reads], sends_cnt] [\&_socks4_connect_command, [$self, ${*$self}->{SOCKS}->{Bind} ? CMD_BIND : CMD_CONNECT], undef, [], 0], [\&_socks4_connect_reply, [$self], undef, [], 0] ]; } else { ${*$self}->{SOCKS}->{queue} = [ [\&_socks5_connect, [$self], undef, [], 0], [\&_socks5_connect_if_auth, [$self], undef, [], 0], [\&_socks5_connect_command, [ $self, ${*$self}->{SOCKS}->{Bind} ? CMD_BIND : ${*$self}->{SOCKS}->{TCP} ? CMD_UDPASSOC : CMD_CONNECT ], undef, [], 0 ], [\&_socks5_connect_reply, [$self], undef, [], 0] ]; } defined( $self->_run_queue() ) or return; return $self; } ############################################################################### # # _run_queue - run tasks from queue, return undef on error, -1 if one of the task # returned not completed because of the possible blocking on network operation # ############################################################################### sub _run_queue { my $self = shift; my $retval; while(my $elt = ${*$self}->{SOCKS}->{queue}[0]) { $retval = $elt->[Q_SUB]->(@{$elt->[Q_ARGS]}); unless (defined $retval) { ${*$self}->{SOCKS}->{queue} = []; ${*$self}->{SOCKS}->{queue_results} = {}; last; } last if ($retval == -1); ${*$self}->{SOCKS}->{queue_results}{$elt->[Q_SUB]} = $retval; shift @{${*$self}->{SOCKS}->{queue}}; } if(defined($retval) && !@{${*$self}->{SOCKS}->{queue}}) { ${*$self}->{SOCKS}->{queue_results} = {}; ${*$self}->{SOCKS}->{ready} = 1; } return $retval; } ############################################################################### # # ready - check is non-blocking socket ready to transfer user data # ############################################################################### sub ready { my $self = shift; $self->_run_queue(); return ${*$self}->{SOCKS}->{ready}; } ############################################################################### # # _socks5_connect - Send the opening handsake, and process the reply. # ############################################################################### sub _socks5_connect { my $self = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; my ($reads, $sends, $debugs) = (0, 0, 0); my $sock = defined( ${*$self}->{SOCKS}->{TCP} ) ? ${*$self}->{SOCKS}->{TCP} : $self; #-------------------------------------------------------------------------- # Send the auth mechanisms #-------------------------------------------------------------------------- # +----+----------+----------+ # |VER | NMETHODS | METHODS | # +----+----------+----------+ # | 1 | 1 | 1 to 255 | # +----+----------+----------+ my $nmethods = 0; my $methods; foreach my $method (0..$#{${*$self}->{SOCKS}->{AuthMethods}}) { if (${*$self}->{SOCKS}->{AuthMethods}->[$method] == 1) { $methods .= pack('C', $method); $nmethods++; } } my $reply; $reply = $sock->_socks_send(pack('CCa*', SOCKS5_VER, $nmethods, $methods), ++$sends) or return _fail($reply); if($debug && !$self->_debugged(++$debugs)) { $debug->add( ver => SOCKS5_VER, nmethods => $nmethods, methods => join('', unpack("C$nmethods", $methods)) ); $debug->show('Send: '); } #-------------------------------------------------------------------------- # Read the reply #-------------------------------------------------------------------------- # +----+--------+ # |VER | METHOD | # +----+--------+ # | 1 | 1 | # +----+--------+ $reply = $sock->_socks_read(2, ++$reads) or return _fail($reply); my ($version, $auth_method) = unpack('CC', $reply); if($debug && !$self->_debugged(++$debugs)) { $debug->add( ver => $version, method => $auth_method ); $debug->show('Recv: '); } if ($auth_method == AUTHMECH_INVALID) { $SOCKS_ERROR = $CODES{AUTHMECH}->[$auth_method]; return; } return $auth_method; } sub _socks5_connect_if_auth { my $self = shift; if(${*$self}->{SOCKS}->{queue_results}{\&_socks5_connect} != AUTHMECH_ANON) { unshift @{${*$self}->{SOCKS}->{queue}}, [\&_socks5_connect_auth, [$self], undef, [], 0]; (${*$self}->{SOCKS}->{queue}[0], ${*$self}->{SOCKS}->{queue}[1]) = (${*$self}->{SOCKS}->{queue}[1], ${*$self}->{SOCKS}->{queue}[0]); } 1; } ############################################################################### # # _socks5_connect_auth - Send and receive a SOCKS5 auth handshake (rfc1929) # ############################################################################### sub _socks5_connect_auth { my $self = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; my ($reads, $sends, $debugs) = (0, 0, 0); my $sock = defined( ${*$self}->{SOCKS}->{TCP} ) ? ${*$self}->{SOCKS}->{TCP} : $self; #-------------------------------------------------------------------------- # Send the auth #-------------------------------------------------------------------------- # +----+------+----------+------+----------+ # |VER | ULEN | UNAME | PLEN | PASSWD | # +----+------+----------+------+----------+ # | 1 | 1 | 1 to 255 | 1 | 1 to 255 | # +----+------+----------+------+----------+ my $uname = ${*$self}->{SOCKS}->{Username}; my $passwd = ${*$self}->{SOCKS}->{Password}; my $ulen = length($uname); my $plen = length($passwd); my $reply; $reply = $sock->_socks_send(pack("CCa${ulen}Ca*", 1, $ulen, $uname, $plen, $passwd), ++$sends) or return _fail($reply); if($debug && !$self->_debugged(++$debugs)) { $debug->add( ver => 1, ulen => $ulen, uname => $uname, plen => $plen, passwd => $passwd ); $debug->show('Send: '); } #-------------------------------------------------------------------------- # Read the reply #-------------------------------------------------------------------------- # +----+--------+ # |VER | STATUS | # +----+--------+ # | 1 | 1 | # +----+--------+ $reply = $sock->_socks_read(2, ++$reads) or return _fail($reply); my ($ver, $status) = unpack('CC', $reply); if($debug && !$self->_debugged(++$debugs)) { $debug->add( ver => $ver, status => $status ); $debug->show('Recv: '); } if ($status != AUTHREPLY_SUCCESS) { $SOCKS_ERROR = "Authentication failed with SOCKS5 proxy."; return; } return 1; } ############################################################################### # # _socks_connect_command - Process a SOCKS5 command request # ############################################################################### sub _socks5_connect_command { my $self = shift; my $command = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; my ($reads, $sends, $debugs) = (0, 0, 0); my $resolve = defined(${*$self}->{SOCKS}->{Resolve}) ? ${*$self}->{SOCKS}->{Resolve} : $SOCKS5_RESOLVE; my $sock = defined( ${*$self}->{SOCKS}->{TCP} ) ? ${*$self}->{SOCKS}->{TCP} : $self; #-------------------------------------------------------------------------- # Send the command #-------------------------------------------------------------------------- # +----+-----+-------+------+----------+----------+ # |VER | CMD | RSV | ATYP | DST.ADDR | DST.PORT | # +----+-----+-------+------+----------+----------+ # | 1 | 1 | X'00' | 1 | Variable | 2 | # +----+-----+-------+------+----------+----------+ my $atyp = $resolve ? ADDR_DOMAINNAME : ADDR_IPV4; my $dstaddr = $resolve ? ${*$self}->{SOCKS}->{CmdAddr} : inet_aton(${*$self}->{SOCKS}->{CmdAddr}); my $hlen = length($dstaddr) if $resolve; my $dstport = pack('n', ${*$self}->{SOCKS}->{CmdPort}); my $reply; $reply = $sock->_socks_send(pack('C4', SOCKS5_VER, $command, 0, $atyp) . (defined($hlen) ? pack('C', $hlen) : '') . $dstaddr . $dstport, ++$sends) or return _fail($reply); if($debug && !$self->_debugged(++$debugs)) { $debug->add( ver => SOCKS5_VER, cmd => $command, rsv => 0, atyp => $atyp ); $debug->add(hlen => $hlen) if defined $hlen; $debug->add( dstaddr => $resolve ? $dstaddr : (length($dstaddr) == 4 ? inet_ntoa($dstaddr) : undef), dstport => ${*$self}->{SOCKS}->{CmdPort} ); $debug->show('Send: '); } return 1; } sub _socks5_connect_reply { my $self = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; my ($reads, $sends, $debugs) = (0, 0, 0); my $sock = defined( ${*$self}->{SOCKS}->{TCP} ) ? ${*$self}->{SOCKS}->{TCP} : $self; #-------------------------------------------------------------------------- # Read the reply #-------------------------------------------------------------------------- # +----+-----+-------+------+----------+----------+ # |VER | REP | RSV | ATYP | BND.ADDR | BND.PORT | # +----+-----+-------+------+----------+----------+ # | 1 | 1 | X'00' | 1 | Variable | 2 | # +----+-----+-------+------+----------+----------+ my $reply; $reply = $sock->_socks_read(4, ++$reads) or return _fail($reply); my ($ver, $rep, $rsv, $atyp) = unpack('C4', $reply); if($debug) { $debug->add( ver => $ver, rep => $rep, rsv => $rsv, atyp => $atyp ); } my ($bndaddr, $bndport); if ($atyp == ADDR_DOMAINNAME) { $reply = $sock->_socks_read(1, ++$reads) or return _fail($reply); my $hlen = unpack('C', $reply); $bndaddr = $sock->_socks_read($hlen, ++$reads) or return _fail($bndaddr); if($debug) { $debug->add( hlen => $hlen, bndaddr => $bndaddr ); } } elsif ($atyp == ADDR_IPV4) { $reply = $sock->_socks_read(4, ++$reads) or return _fail($reply); $bndaddr = length($reply) == 4 ? inet_ntoa($reply) : undef; if($debug) { $debug->add(bndaddr => $bndaddr); } } else { $SOCKS_ERROR = "Unsupported address type returned by socks server: $atyp"; return; } $reply = $sock->_socks_read(2, ++$reads) or return _fail($reply); $bndport = unpack('n', $reply); ${*$self}->{SOCKS}->{DstAddr} = $bndaddr; ${*$self}->{SOCKS}->{DstPort} = $bndport; if($debug && !$self->_debugged(++$debugs)) { $debug->add(bndport => $bndport); $debug->show('Recv: '); } if($rep != REPLY_SUCCESS) { $SOCKS_ERROR = $CODES{REPLY}->{$rep}; return; } return 1; } ############################################################################### # # _socks4_connect_command - Send the opening handsake, and process the reply. # ############################################################################### sub _socks4_connect_command { # http://ss5.sourceforge.net/socks4.protocol.txt # http://ss5.sourceforge.net/socks4A.protocol.txt my $self = shift; my $command = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; my ($reads, $sends, $debugs) = (0, 0, 0); my $resolve = defined(${*$self}->{SOCKS}->{Resolve}) ? ${*$self}->{SOCKS}->{Resolve} : $SOCKS4_RESOLVE; #-------------------------------------------------------------------------- # Send the command #-------------------------------------------------------------------------- # +-----+-----+----------+---------------+----------+------+ # | VER | CMD | DST.PORT | DST.ADDR | USERID | NULL | # +-----+-----+----------+---------------+----------+------+ # | 1 | 1 | 2 | 4 | variable | 1 | # +-----+-----+----------+---------------+----------+------+ my $dstaddr = $resolve ? inet_aton('0.0.0.1') : inet_aton(${*$self}->{SOCKS}->{CmdAddr}); my $dstport = pack('n', ${*$self}->{SOCKS}->{CmdPort}); my $userid = ${*$self}->{SOCKS}->{Username}; my $dsthost; if($resolve) { # socks4a $dsthost = ${*$self}->{SOCKS}->{CmdAddr} . pack('C', 0); } my $reply; $reply = $self->_socks_send(pack('CC', SOCKS4_VER, $command) . $dstport . $dstaddr . $userid . pack('C', 0) . $dsthost, ++$sends) or return _fail($reply); if($debug && !$self->_debugged(++$debugs)) { $debug->add( ver => SOCKS4_VER, cmd => $command, dstport => ${*$self}->{SOCKS}->{CmdPort}, dstaddr => length($dstaddr) == 4 ? inet_ntoa($dstaddr) : undef, userid => $userid, null => 0 ); if($dsthost) { $debug->add( dsthost => ${*$self}->{SOCKS}->{CmdAddr}, null => 0 ); } $debug->show('Send: '); } return 1; } sub _socks4_connect_reply { my $self = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; my ($reads, $sends, $debugs) = (0, 0, 0); #-------------------------------------------------------------------------- # Read the reply #-------------------------------------------------------------------------- # +-----+-----+----------+---------------+ # | VER | REP | BND.PORT | BND.ADDR | # +-----+-----+----------+---------------+ # | 1 | 1 | 2 | 4 | # +-----+-----+----------+---------------+ my $reply; $reply = $self->_socks_read(8, ++$reads) or return _fail($reply); my ($ver, $rep, $bndport) = unpack('CCn', $reply); substr($reply, 0, 4) = ''; my $bndaddr = length($reply) == 4 ? inet_ntoa($reply) : undef; ${*$self}->{SOCKS}->{DstAddr} = $bndaddr; ${*$self}->{SOCKS}->{DstPort} = $bndport; if($debug && !$self->_debugged(++$debugs)) { $debug->add( ver => $ver, rep => $rep, bndport => $bndport, bndaddr => $bndaddr ); $debug->show('Recv: '); } if($rep != REQUEST_GRANTED) { $SOCKS_ERROR = $CODES{REPLY}->{$rep}; return; } return 1; } ############################################################################### #+----------------------------------------------------------------------------- #| Accept Functions #+----------------------------------------------------------------------------- ############################################################################### ############################################################################### # # accept - When we are accepting new connections, we need to do the SOCKS # handshaking before we return a usable socket. # ############################################################################### sub accept { my $self = shift; croak("Undefined IO::Socket::Socks object passed to accept.") unless defined($self); if(${*$self}->{SOCKS}->{Listen}) { my $client = $self->SUPER::accept(@_); if (!$client) { $SOCKS_ERROR = "Proxy accept new client failed."; return; } # inherit some socket parameters ${*$client}->{SOCKS}->{Debug} = ${*$self}->{SOCKS}->{Debug}; ${*$client}->{SOCKS}->{Version} = ${*$self}->{SOCKS}->{Version}; if(${*$self}->{SOCKS}->{Version} == 4) { return unless $self->_socks4_accept_command($client); } else { my $authmech = $self->_socks5_accept($client); return unless defined($authmech); if ($authmech == AUTHMECH_USERPASS) { return unless $self->_socks5_accept_auth($client); } return unless $self->_socks5_accept_command($client); } return $client; } else { ${*$self}->{SOCKS}->{ready} = 0; if({*$self}->{SOCKS}->{Version} == 4) { push @{${*$self}->{SOCKS}->{queue}}, [\&_socks4_connect_reply, [$self], undef, [], 0]; } else { push @{${*$self}->{SOCKS}->{queue}}, [\&_socks5_connect_reply, [$self], undef, [], 0]; } defined( $self->_run_queue() ) or return; return $self; } } ############################################################################### # # _socks5_accept - Wait for an opening handsake, and reply. # ############################################################################### sub _socks5_accept { my $self = shift; my $client = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; #-------------------------------------------------------------------------- # Read the auth mechanisms #-------------------------------------------------------------------------- # +----+----------+----------+ # |VER | NMETHODS | METHODS | # +----+----------+----------+ # | 1 | 1 | 1 to 255 | # +----+----------+----------+ my $request = $client->_socks_read(2) or return _fail(); my ($ver, $nmethods) = unpack('CC', $request); $request = $client->_socks_read($nmethods) or return _fail(); my @methods = unpack('C'x$nmethods, $request); if($debug) { $debug->add( ver => $ver, nmethods => $nmethods, methods => join('', @methods) ); $debug->show('Recv: '); } if($ver != SOCKS5_VER) { $SOCKS_ERROR = "Socks version should be 5, $ver recieved"; return; } if ($nmethods == 0) { $SOCKS_ERROR = "No auth methods sent."; return; } my $authmech; foreach my $method (@methods) { if (${*$self}->{SOCKS}->{AuthMethods}->[$method] == 1) { $authmech = $method; last; } } if (!defined($authmech)) { $authmech = AUTHMECH_INVALID; } #-------------------------------------------------------------------------- # Send the reply #-------------------------------------------------------------------------- # +----+--------+ # |VER | METHOD | # +----+--------+ # | 1 | 1 | # +----+--------+ $client->_socks_send(pack('CC', SOCKS5_VER, $authmech)) or return _fail(); if($debug) { $debug->add( ver => SOCKS5_VER, method => $authmech ); $debug->show('Send: '); } if ($authmech == AUTHMECH_INVALID) { $SOCKS_ERROR = "No available auth methods."; return; } return $authmech; } ############################################################################### # # _socks5_accept_auth - Send and receive a SOCKS5 auth handshake (rfc1929) # ############################################################################### sub _socks5_accept_auth { my $self = shift; my $client = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; #-------------------------------------------------------------------------- # Read the auth #-------------------------------------------------------------------------- # +----+------+----------+------+----------+ # |VER | ULEN | UNAME | PLEN | PASSWD | # +----+------+----------+------+----------+ # | 1 | 1 | 1 to 255 | 1 | 1 to 255 | # +----+------+----------+------+----------+ my $request = $client->_socks_read(2) or return _fail(); my ($ver, $ulen) = unpack('CC', $request); $request = $client->_socks_read($ulen+1) or return _fail(); my $uname = substr($request, 0, $ulen); my $plen = unpack('C', substr($request, $ulen)); my $passwd = $client->_socks_read($plen) or return _fail(); if($debug) { $debug->add( ver => $ver, ulen => $ulen, uname => $uname, plen => $plen, passwd => $passwd ); $debug->show('Recv: '); } my $status = 1; if (defined(${*$self}->{SOCKS}->{UserAuth})) { $status = &{${*$self}->{SOCKS}->{UserAuth}}($uname, $passwd); } #-------------------------------------------------------------------------- # Send the reply #-------------------------------------------------------------------------- # +----+--------+ # |VER | STATUS | # +----+--------+ # | 1 | 1 | # +----+--------+ $status = $status ? AUTHREPLY_SUCCESS : AUTHREPLY_FAILURE; $client->_socks_send(pack('CC', 1, $status)) or return _fail(); if($debug) { $debug->add( ver => 1, status => $status ); $debug->show('Send: '); } if ($status != AUTHREPLY_SUCCESS) { $SOCKS_ERROR = "Authentication failed with SOCKS5 proxy."; return; } return 1; } ############################################################################### # # _socks5_acccept_command - Process a SOCKS5 command request. Since this is # a library and not a server, we cannot process the # command. Let the parent program handle that. # ############################################################################### sub _socks5_accept_command { my $self = shift; my $client = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; @{${*$client}->{SOCKS}->{COMMAND}} = (); #-------------------------------------------------------------------------- # Read the command #-------------------------------------------------------------------------- # +----+-----+-------+------+----------+----------+ # |VER | CMD | RSV | ATYP | DST.ADDR | DST.PORT | # +----+-----+-------+------+----------+----------+ # | 1 | 1 | X'00' | 1 | Variable | 2 | # +----+-----+-------+------+----------+----------+ my $request = $client->_socks_read(4) or return _fail(); my ($ver, $cmd, $rsv, $atyp) = unpack('CCCC', $request); if($debug) { $debug->add( ver => $ver, cmd => $cmd, rsv => $rsv, atyp => $atyp ); } my $dstaddr; if ($atyp == ADDR_DOMAINNAME) { defined( $request = $client->_socks_read() ) or return _fail(); my $hlen = unpack('C', $request); $dstaddr = $client->_socks_read($hlen) or return _fail(); if($debug) { $debug->add(hlen => $hlen); } } elsif ($atyp == ADDR_IPV4) { $request = $client->_socks_read(4) or return _fail(); $dstaddr = length($request) == 4 ? inet_ntoa($request) : undef; } else { # unknown address type - how many bytes to read? $client->_socks5_accept_command_reply(REPLY_ADDR_NOT_SUPPORTED, '0.0.0.0', 0); $SOCKS_ERROR = $CODES{REPLY}->{REPLY_ADDR_NOT_SUPPORTED}; return; } $request = $client->_socks_read(2) or return _fail(); my $dstport = unpack('n', $request); if($debug) { $debug->add( dstaddr => $dstaddr, dstport => $dstport ); $debug->show('Recv: '); } @{${*$client}->{SOCKS}->{COMMAND}} = ($cmd, $dstaddr, $dstport); return 1; } ############################################################################### # # _socks5_acccept_command_reply - Answer a SOCKS5 command request. Since this # is a library and not a server, we cannot # process the command. Let the parent program # handle that. # ############################################################################### sub _socks5_accept_command_reply { my $self = shift; my $reply = shift; my $host = shift; my $port = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; my $resolve = defined(${*$self}->{SOCKS}->{Resolve}) ? ${*$self}->{SOCKS}->{Resolve} : $SOCKS5_RESOLVE; if (!defined($reply) || !defined($host) || !defined($port)) { croak("You must provide a reply, host, and port on the command reply."); } #-------------------------------------------------------------------------- # Send the reply #-------------------------------------------------------------------------- # +----+-----+-------+------+----------+----------+ # |VER | REP | RSV | ATYP | BND.ADDR | BND.PORT | # +----+-----+-------+------+----------+----------+ # | 1 | 1 | X'00' | 1 | Variable | 2 | # +----+-----+-------+------+----------+----------+ my $atyp = $resolve ? ADDR_IPV4 : ADDR_DOMAINNAME; my $bndaddr = $resolve ? inet_aton($host) : $host; my $hlen = length($bndaddr) unless $resolve; $self->_socks_send(pack('CCCC', SOCKS5_VER, $reply, 0, $atyp) . ($resolve ? '' : pack('C', $hlen)) . $bndaddr . pack('n', $port)) or return _fail(); if($debug) { $debug->add( ver => SOCKS5_VER, rep => $reply, rsv => 0, atyp => $atyp ); $debug->add(hlen => $hlen) unless $resolve; $debug->add( bndaddr => $resolve ? (length($bndaddr) == 4 ? inet_ntoa($bndaddr) : undef) : $bndaddr, bndport => $port ); $debug->show('Send: '); } } ############################################################################### # # _socks4_accept_command - Wait for an opening handsake and process a SOCKS4 # command request. Since this is a library and not # a server, we cannot process the command. Let the # parent program handle that. # ############################################################################### sub _socks4_accept_command { my $self = shift; my $client = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; my $resolve = defined(${*$self}->{SOCKS}->{Resolve}) ? ${*$self}->{SOCKS}->{Resolve} : $SOCKS4_RESOLVE; @{${*$client}->{SOCKS}->{COMMAND}} = (); #-------------------------------------------------------------------------- # Read the auth mechanisms #-------------------------------------------------------------------------- # +-----+-----+----------+---------------+----------+------+ # | VER | CMD | DST.PORT | DST.ADDR | USERID | NULL | # +-----+-----+----------+---------------+----------+------+ # | 1 | 1 | 2 | 4 | variable | 1 | # +-----+-----+----------+---------------+----------+------+ my $request = $client->_socks_read(8) or return _fail(); my ($ver, $cmd, $dstport) = unpack('CCn', $request); substr($request, 0, 4) = ''; my $dstaddr = length($request) == 4 ? inet_ntoa($request) : undef; my $userid = ''; my $c; while(1) { defined( $c = $client->_socks_read() ) or return _fail(); if($c ne "\0") { $userid .= $c; } else { last; } } if($debug) { $debug->add( ver => $ver, cmd => $cmd, dstport => $dstport, dstaddr => $dstaddr, userid => $userid, null => 0 ); } if($resolve && $dstaddr =~ /^0\.0\.0\.[1-9]/) { # socks4a my $dsthost = ''; while(1) { defined( $c = $client->_socks_read() ) or return _fail(); if($c ne "\0") { $dsthost .= $c; } else { last; } } if($debug) { $debug->add( dsthost => $dsthost, null => 0 ); } $dstaddr = join('.', unpack('C4', (gethostbyname($dsthost))[4])); } if($debug) { $debug->show('Recv: '); } if(defined(${*$self}->{SOCKS}->{UserAuth})) { unless( &{${*$self}->{SOCKS}->{UserAuth}}($userid) ) { $client->_socks4_accept_command_reply(REQUEST_REJECTED_USERID, '0.0.0.0', 0); $SOCKS_ERROR = 'Authentication failed with SOCKS4 proxy.'; return; } } if($ver != SOCKS4_VER) { $SOCKS_ERROR = "Socks version should be 4, $ver recieved"; return; } @{${*$client}->{SOCKS}->{COMMAND}} = ($cmd, $dstaddr, $dstport); return 1; } ############################################################################### # # _socks4_acccept_command_reply - Answer a SOCKS4 command request. Since this # is a library and not a server, we cannot # process the command. Let the parent program # handle that. # ############################################################################### sub _socks4_accept_command_reply { my $self = shift; my $reply = shift; my $host = shift; my $port = shift; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; if (!defined($reply) || !defined($host) || !defined($port)) { croak("You must provide a reply, host, and port on the command reply."); } #-------------------------------------------------------------------------- # Send the reply #-------------------------------------------------------------------------- # +-----+-----+----------+---------------+ # | VER | REP | BND.PORT | BND.ADDR | # +-----+-----+----------+---------------+ # | 1 | 1 | 2 | 4 | # +-----+-----+----------+---------------+ my $bndaddr = inet_aton($host); $self->_socks_send(pack('CCna*', 0, $reply, $port, $bndaddr)) or return _fail(); if($debug) { $debug->add( ver => 0, rep => $reply, bndport => $port, bndaddr => length($bndaddr) == 4 ? inet_ntoa($bndaddr) : undef ); $debug->show('Send: '); } } ############################################################################### # # command - return the command the user request along with the host and # port to operate on. # ############################################################################### sub command { my $self = shift; unless(exists ${*$self}->{SOCKS}->{AuthMethods}) { return ${*$self}->{SOCKS}->{COMMAND}; } else { my @keys = qw(Version AuthType RequireAuth UserAuth Username Password Debug Resolve AuthMethods CmdAddr CmdPort Bind TCP); my %tmp; $tmp{$_} = ${*$self}->{SOCKS}->{$_} for @keys; my %args = @_; $self->_configure(\%args); if( $self->_connect() ) { return 1; } ${*$self}->{SOCKS}->{$_} = $tmp{$_} for @keys; return 0; } } ############################################################################### # # command_reply - public reply wrapper to the client. # ############################################################################### sub command_reply { my $self = shift; if(${*$self}->{SOCKS}->{Version} == 4) { $self->_socks4_accept_command_reply(@_); } else { $self->_socks5_accept_command_reply(@_); } } ############################################################################### # # dst - access to the address and port selected by socks server when connect/bind/udpassoc # ############################################################################### sub dst { my $self = shift; return (${*$self}->{SOCKS}->{DstAddr}, ${*$self}->{SOCKS}->{DstPort}); } ############################################################################### # # send - send UDP datagram # ############################################################################### sub send { my $self = shift; unless(defined ${*$self}->{SOCKS}->{TCP}) { return $self->SUPER::send(@_); } my ($msg, $flags, $peer) = @_; my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; my $resolve = defined(${*$self}->{SOCKS}->{Resolve}) ? ${*$self}->{SOCKS}->{Resolve} : $SOCKS5_RESOLVE; croak "send: Cannot determine peer address" unless defined $peer; my ($dstport, $dstaddr) = sockaddr_in($peer); my ($sndaddr, $sndport) = $self->dst; if($sndaddr eq '0.0.0.0') { $sndaddr = ${*$self}->{SOCKS}->{ProxyAddr}; } $sndaddr = inet_aton($sndaddr); $peer = sockaddr_in($sndport, $sndaddr); my ($atyp, $hlen); if($resolve) { $atyp = ADDR_DOMAINNAME; $dstaddr = inet_ntoa($dstaddr); $hlen = length($dstaddr); } else { $atyp = ADDR_IPV4; } my $msglen = length($msg) if $debug; # we need to add socks header to the message # +----+------+------+----------+----------+----------+ # |RSV | FRAG | ATYP | DST.ADDR | DST.PORT | DATA | # +----+------+------+----------+----------+----------+ # | 2 | 1 | 1 | Variable | 2 | Variable | # +----+------+------+----------+----------+----------+ $msg = pack('C4', 0, 0, 0, $atyp) . ($resolve ? pack('C', $hlen) : '') . $dstaddr . pack('n', $dstport) . $msg; if($debug) { $debug->add( rsv => '00', frag => '0', atyp => $atyp ); $debug->add(hlen => $hlen) if $resolve; $debug->add( dstaddr => $resolve ? $dstaddr : (length($dstaddr) == 4 ? inet_ntoa($dstaddr) : undef), dstport => $dstport, data => "...($msglen)" ); $debug->show('Send: '); } $self->SUPER::send($msg, $flags, $peer); } ############################################################################### # # recv - receive UDP datagram # ############################################################################### sub recv { my $self = shift; unless(defined ${*$self}->{SOCKS}->{TCP}) { return $self->SUPER::recv(@_); } my $debug = IO::Socket::Socks::Debug->new() if ${*$self}->{SOCKS}->{Debug}; defined(my $peer = $self->SUPER::recv($_[0], $_[1]+262, $_[2]) ) or return; # we need to remove socks header from the message # +----+------+------+----------+----------+----------+ # |RSV | FRAG | ATYP | DST.ADDR | DST.PORT | DATA | # +----+------+------+----------+----------+----------+ # | 2 | 1 | 1 | Variable | 2 | Variable | # +----+------+------+----------+----------+----------+ my $rsv = join('', unpack('C2', $_[0])); substr($_[0], 0, 2) = ''; my ($frag, $atyp) = unpack('C2', $_[0]); substr($_[0], 0, 2) = ''; if($debug) { $debug->add( rsv => $rsv, frag => $frag, atyp => $atyp ); } my $dstaddr; if($atyp == ADDR_DOMAINNAME) { my $hlen = unpack('C', $_[0]); $dstaddr = substr($_[0], 1, $hlen); substr($_[0], 0, $hlen+1) = ''; if($debug) { $debug->add( hlen => $hlen ); } } elsif($atyp == ADDR_IPV4) { $dstaddr = substr($_[0], 0, 4); $dstaddr = length($dstaddr) == 4 ? inet_ntoa($dstaddr) : undef; substr($_[0], 0, 4) = ''; } else { $SOCKS_ERROR = "Unsupported address type returned by socks server: $atyp"; return; } my $dstport = unpack('n', $_[0]); substr($_[0], 0, 2) = ''; if($debug) { $debug->add( dstaddr => $dstaddr, dstport => $dstport, data => "...(" . length($_[0]) . ")" ); $debug->show('Recv: '); } return $peer; } ############################################################################### #+----------------------------------------------------------------------------- #| Helper Functions #+----------------------------------------------------------------------------- ############################################################################### sub _socks_send { my $self = shift; my $data = shift; my $numb = shift; $SOCKS_ERROR = undef; my $rc; my $writed = 0; my $blocking = ${*$self}{io_socket_timeout} ? $self->blocking(0) : $self->blocking; unless ($blocking || ${*$self}{io_socket_timeout}) { if(${*$self}->{SOCKS}->{queue}[0][Q_SENDS] >= $numb) { # already sent return 1; } if(defined ${*$self}->{SOCKS}->{queue}[0][Q_BUF]) { # some chunk already sent substr($data, 0, ${*$self}->{SOCKS}->{queue}[0][Q_BUF]) = ''; } while(length $data) { $rc = $self->syswrite($data); if(defined $rc) { if($rc > 0) { ${*$self}->{SOCKS}->{queue}[0][Q_BUF] += $rc; substr($data, 0, $rc) = ''; } else { last; } } elsif($! == EWOULDBLOCK || $! == EAGAIN) { $SOCKS_ERROR = SOCKS_WANT_WRITE; return undef; } else { $SOCKS_ERROR = $!; last; } } $writed = int(${*$self}->{SOCKS}->{queue}[0][Q_BUF]); ${*$self}->{SOCKS}->{queue}[0][Q_BUF] = undef; ${*$self}->{SOCKS}->{queue}[0][Q_SENDS]++; return $writed; } my $selector = IO::Select->new($self); my $start = time(); while(!${*$self}{io_socket_timeout} || time() - $start < ${*$self}{io_socket_timeout}) { unless($selector->can_write(1)) { # socket couldn't accept data for now, check if timeout expired and try again next; } $rc = $self->syswrite($data); if($rc > 0) { # reduce our message $writed += $rc; substr($data, 0, $rc) = ''; if(length($data) == 0) { # all data successfully writed last; } } else { # some error in the socket; will return false $SOCKS_ERROR = $! unless defined $rc; last; } } $self->blocking(1) if $blocking; return $writed; } sub _socks_read { my $self = shift; my $length = shift || 1; my $numb = shift; $SOCKS_ERROR = undef; my $data = ''; my ($buf, $rc); my $blocking = $self->blocking; # non-blocking read unless ($blocking || ${*$self}{io_socket_timeout}) { # no timeout should be specified for non-blocking connect if(defined ${*$self}->{SOCKS}->{queue}[0][Q_READS][$numb]) { # already readed return ${*$self}->{SOCKS}->{queue}[0][Q_READS][$numb]; } if(defined ${*$self}->{SOCKS}->{queue}[0][Q_BUF]) { # some chunk already readed $data = ${*$self}->{SOCKS}->{queue}[0][Q_BUF]; $length -= length $data; } while($length > 0) { $rc = $self->sysread($buf, $length); if(defined $rc) { if($rc > 0) { $length -= $rc; $data .= $buf; } else { last } } elsif($! == EWOULDBLOCK || $! == EAGAIN) { # no data to read if (length $data) { # save already readed data in the queue buffer ${*$self}->{SOCKS}->{queue}[0][Q_BUF] = $data; } $SOCKS_ERROR = SOCKS_WANT_READ; return undef; } else { $SOCKS_ERROR = $!; last; } } ${*$self}->{SOCKS}->{queue}[0][Q_BUF] = undef; ${*$self}->{SOCKS}->{queue}[0][Q_READS][$numb] = $data; return $data; } # blocking read my $selector = IO::Select->new($self); my $start = time(); while($length > 0 && (!${*$self}{io_socket_timeout} || time() - $start < ${*$self}{io_socket_timeout})) { unless($selector->can_read(1)) { # no data in socket for now, check if timeout expired and try again next; } $rc = $self->sysread($buf, $length); if($rc > 0) { # reduce limit and modify buffer $length -= $rc; $data .= $buf; } else { # EOF or error in the socket $SOCKS_ERROR = $! unless defined $rc; last; } } return $data; } sub _debugged { my ($self, $debugs) = @_; if(${*$self}->{SOCKS}->{queue}[0][Q_DEBUGS] >= $debugs) { return 1; } ${*$self}->{SOCKS}->{queue}[0][Q_DEBUGS] = $debugs; return 0; } sub _fail { if(!@_ || defined($_[0])) { $SOCKS_ERROR = 'Timeout' unless defined $SOCKS_ERROR; return; } return -1; } ############################################################################### #+----------------------------------------------------------------------------- #| Helper Package to display pretty debug messages #+----------------------------------------------------------------------------- ############################################################################### package IO::Socket::Socks::Debug; sub new { my ($class) = @_; my $self = []; bless $self, $class; } sub add { my $self = shift; push @{$self}, @_; } sub show { my ($self, $tag) = @_; $self->_separator($tag); $self->_row(0, $tag); $self->_separator($tag); $self->_row(1, $tag); $self->_separator($tag); print STDERR "\n"; @{$self} = (); } sub _separator { my $self = shift; my $tag = shift; my ($row1_len, $row2_len, $len); print STDERR $tag, '+'; for(my $i=0; $i<@$self; $i+=2) { $row1_len = length($self->[$i]); $row2_len = length($self->[$i+1]); $len = ($row1_len > $row2_len ? $row1_len : $row2_len)+2; print STDERR '-' x $len, '+'; } print STDERR "\n"; } sub _row { my $self = shift; my $row = shift; my $tag = shift; my ($row1_len, $row2_len, $len); print STDERR $tag, '|'; for(my $i=0; $i<@$self; $i+=2) { $row1_len = length($self->[$i]); $row2_len = length($self->[$i+1]); $len = ($row1_len > $row2_len ? $row1_len : $row2_len); printf STDERR ' %-'.$len.'s |', $self->[$i+$row]; } print STDERR "\n"; } 1; __END__