ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/AnyEvent/lib/AnyEvent/Socket.pm
(Generate patch)

Comparing AnyEvent/lib/AnyEvent/Socket.pm (file contents):
Revision 1.61 by root, Thu Aug 21 23:48:35 2008 UTC vs.
Revision 1.157 by root, Wed Oct 31 15:42:06 2012 UTC

1=head1 NAME 1=head1 NAME
2 2
3AnyEvent::Socket - useful IPv4 and IPv6 stuff. 3AnyEvent::Socket - useful IPv4 and IPv6 stuff. also unix domain sockets. and stuff.
4 4
5=head1 SYNOPSIS 5=head1 SYNOPSIS
6 6
7 use AnyEvent::Socket; 7 use AnyEvent::Socket;
8 8
33 33
34=cut 34=cut
35 35
36package AnyEvent::Socket; 36package AnyEvent::Socket;
37 37
38no warnings;
39use strict;
40
41use Carp (); 38use Carp ();
42use Errno (); 39use Errno ();
43use Socket qw(AF_INET AF_UNIX SOCK_STREAM SOCK_DGRAM SOL_SOCKET SO_REUSEADDR); 40use Socket qw(AF_INET AF_UNIX SOCK_STREAM SOCK_DGRAM SOL_SOCKET SO_REUSEADDR);
44 41
45use AnyEvent (); 42use AnyEvent (); BEGIN { AnyEvent::common_sense }
46use AnyEvent::Util qw(guard fh_nonblocking AF_INET6); 43use AnyEvent::Util qw(guard fh_nonblocking AF_INET6);
47use AnyEvent::DNS (); 44use AnyEvent::DNS ();
48 45
49use base 'Exporter'; 46use base 'Exporter';
50 47
51our @EXPORT = qw( 48our @EXPORT = qw(
52 parse_hostport 49 getprotobyname
50 parse_hostport format_hostport
53 parse_ipv4 parse_ipv6 51 parse_ipv4 parse_ipv6
54 parse_ip parse_address 52 parse_ip parse_address
53 format_ipv4 format_ipv6
55 format_ip format_address 54 format_ip format_address
56 address_family 55 address_family
57 inet_aton 56 inet_aton
58 tcp_server 57 tcp_server
59 tcp_connect 58 tcp_connect
60); 59);
61 60
62our $VERSION = 4.233; 61our $VERSION = $AnyEvent::VERSION;
63 62
64=item $ipn = parse_ipv4 $dotted_quad 63=item $ipn = parse_ipv4 $dotted_quad
65 64
66Tries to parse the given dotted quad IPv4 address and return it in 65Tries to parse the given dotted quad IPv4 address and return it in
67octet form (or undef when it isn't in a parsable format). Supports all 66octet form (or undef when it isn't in a parsable format). Supports all
98forms supported by parse_ipv4). Note that scope-id's are not supported 97forms supported by parse_ipv4). Note that scope-id's are not supported
99(and will not parse). 98(and will not parse).
100 99
101This function works similarly to C<inet_pton AF_INET6, ...>. 100This function works similarly to C<inet_pton AF_INET6, ...>.
102 101
102Example:
103
104 print unpack "H*", parse_ipv6 "2002:5345::10.0.0.1";
105 # => 2002534500000000000000000a000001
106
103=cut 107=cut
104 108
105sub parse_ipv6($) { 109sub parse_ipv6($) {
106 # quick test to avoid longer processing 110 # quick test to avoid longer processing
107 my $n = $_[0] =~ y/://; 111 my $n = $_[0] =~ y/://;
137 141
138 # and done 142 # and done
139 pack "n*", map hex, @h, @t 143 pack "n*", map hex, @h, @t
140} 144}
141 145
146=item $token = parse_unix $hostname
147
148This fucntion exists mainly for symmetry to the other C<parse_protocol>
149functions - it takes a hostname and, if it is C<unix/>, it returns a
150special address token, otherwise C<undef>.
151
152The only use for this function is probably to detect whether a hostname
153matches whatever AnyEvent uses for unix domain sockets.
154
155=cut
156
142sub parse_unix($) { 157sub parse_unix($) {
143 $_[0] eq "unix/" 158 $_[0] eq "unix/"
144 ? pack "S", AF_UNIX 159 ? pack "S", AF_UNIX
145 : undef 160 : undef
146 161
147} 162}
148 163
149=item $ipn = parse_address $text 164=item $ipn = parse_address $ip
150 165
151Combines C<parse_ipv4> and C<parse_ipv6> in one function. The address 166Combines C<parse_ipv4> and C<parse_ipv6> in one function. The address
152here refers to the host address (not socket address) in network form 167here refers to the host address (not socket address) in network form
153(binary). 168(binary).
154 169
155If the C<$text> is C<unix/>, then this function returns a special token 170If the C<$text> is C<unix/>, then this function returns a special token
156recognised by the other functions in this module to mean "UNIX domain 171recognised by the other functions in this module to mean "UNIX domain
157socket". 172socket".
158 173
174If the C<$text> to parse is a mapped IPv4 in IPv6 address (:ffff::<ipv4>),
175then it will be treated as an IPv4 address. If you don't want that, you
176have to call C<parse_ipv4> and/or C<parse_ipv6> manually.
177
178Example:
179
180 print unpack "H*", parse_address "10.1.2.3";
181 # => 0a010203
182
159=item $text = AnyEvent::Socket::aton $ipn 183=item $ipn = AnyEvent::Socket::aton $ip
160 184
161Same as C<parse_address>, but not exported (think C<Socket::inet_aton> but 185Same as C<parse_address>, but not exported (think C<Socket::inet_aton> but
162I<without> name resolution). 186I<without> name resolution).
163 187
164=cut 188=cut
165 189
166sub parse_address($) { 190sub parse_address($) {
167 &parse_ipv4 || &parse_ipv6 || &parse_unix 191 for (&parse_ipv6) {
192 if ($_) {
193 s/^\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff//;
194 return $_;
195 } else {
196 return &parse_ipv4 || &parse_unix
197 }
198 }
168} 199}
169 200
170*aton = \&parse_address; 201*aton = \&parse_address;
202
203=item ($name, $aliases, $proto) = getprotobyname $name
204
205Works like the builtin function of the same name, except it tries hard to
206work even on broken platforms (well, that's windows), where getprotobyname
207is traditionally very unreliable.
208
209Example: get the protocol number for TCP (usually 6)
210
211 my $proto = getprotobyname "tcp";
212
213=cut
214
215# microsoft can't even get getprotobyname working (the etc/protocols file
216# gets lost fairly often on windows), so we have to hardcode some common
217# protocol numbers ourselves.
218our %PROTO_BYNAME;
219
220$PROTO_BYNAME{tcp} = Socket::IPPROTO_TCP () if defined &Socket::IPPROTO_TCP;
221$PROTO_BYNAME{udp} = Socket::IPPROTO_UDP () if defined &Socket::IPPROTO_UDP;
222$PROTO_BYNAME{icmp} = Socket::IPPROTO_ICMP() if defined &Socket::IPPROTO_ICMP;
223
224sub getprotobyname($) {
225 my $name = lc shift;
226
227 defined (my $proton = $PROTO_BYNAME{$name} || (getprotobyname $name)[2])
228 or return;
229
230 ($name, uc $name, $proton)
231}
171 232
172=item ($host, $service) = parse_hostport $string[, $default_service] 233=item ($host, $service) = parse_hostport $string[, $default_service]
173 234
174Splitting a string of the form C<hostname:port> is a common 235Splitting a string of the form C<hostname:port> is a common
175problem. Unfortunately, just splitting on the colon makes it hard to 236problem. Unfortunately, just splitting on the colon makes it hard to
176specify IPv6 addresses and doesn't support the less common but well 237specify IPv6 addresses and doesn't support the less common but well
177standardised C<[ip literal]> syntax. 238standardised C<[ip literal]> syntax.
178 239
179This function tries to do this job in a better way, it supports the 240This function tries to do this job in a better way, it supports (at
180following formats, where C<port> can be a numerical port number of a 241least) the following formats, where C<port> can be a numerical port
181service name, or a C<name=port> string, and the C< port> and C<:port> 242number of a service name, or a C<name=port> string, and the C< port> and
182parts are optional. Also, everywhere where an IP address is supported 243C<:port> parts are optional. Also, everywhere where an IP address is
183a hostname or unix domain socket address is also supported (see 244supported a hostname or unix domain socket address is also supported (see
184C<parse_unix>). 245C<parse_unix>), and strings starting with C</> will also be interpreted as
246unix domain sockets.
185 247
186 hostname:port e.g. "www.linux.org", "www.x.de:443", "www.x.de:https=443" 248 hostname:port e.g. "www.linux.org", "www.x.de:443", "www.x.de:https=443",
187 ipv4:port e.g. "198.182.196.56", "127.1:22" 249 ipv4:port e.g. "198.182.196.56", "127.1:22"
188 ipv6 e.g. "::1", "affe::1" 250 ipv6 e.g. "::1", "affe::1"
189 [ipv4or6]:port e.g. "[::1]", "[10.0.1]:80" 251 [ipv4or6]:port e.g. "[::1]", "[10.0.1]:80"
190 [ipv4or6] port e.g. "[127.0.0.1]", "[www.x.org] 17" 252 [ipv4or6] port e.g. "[127.0.0.1]", "[www.x.org] 17"
191 ipv4or6 port e.g. "::1 443", "10.0.0.1 smtp" 253 ipv4or6 port e.g. "::1 443", "10.0.0.1 smtp"
254 unix/:path e.g. "unix/:/path/to/socket"
255 /path e.g. "/path/to/socket"
192 256
193It also supports defaulting the service name in a simple way by using 257It also supports defaulting the service name in a simple way by using
194C<$default_service> if no service was detected. If neither a service was 258C<$default_service> if no service was detected. If neither a service was
195detected nor a default was specified, then this function returns the 259detected nor a default was specified, then this function returns the
196empty list. The same happens when a parse error weas detected, such as a 260empty list. The same happens when a parse error was detected, such as a
197hostname with a colon in it (the function is rather conservative, though). 261hostname with a colon in it (the function is rather conservative, though).
198 262
199Example: 263Example:
200 264
201 print join ",", parse_hostport "localhost:443"; 265 print join ",", parse_hostport "localhost:443";
205 # => "localhost,https" 269 # => "localhost,https"
206 270
207 print join ",", parse_hostport "[::1]"; 271 print join ",", parse_hostport "[::1]";
208 # => "," (empty list) 272 # => "," (empty list)
209 273
274 print join ",", parse_host_port "/tmp/debug.sock";
275 # => "unix/", "/tmp/debug.sock"
276
210=cut 277=cut
211 278
212sub parse_hostport($;$) { 279sub parse_hostport($;$) {
213 my ($host, $port); 280 my ($host, $port);
214 281
215 for ("$_[0]") { # work on a copy, just in case, and also reset pos 282 for ("$_[0]") { # work on a copy, just in case, and also reset pos
216 283
284 # shortcut for /path
285 return ("unix/", $_)
286 if m%^/%;
287
217 # parse host, special cases: "ipv6" or "ipv6 port" 288 # parse host, special cases: "ipv6" or "ipv6[#p ]port"
218 unless ( 289 unless (
219 ($host) = /^\s* ([0-9a-fA-F:]*:[0-9a-fA-F:]*:[0-9a-fA-F\.:]*)/xgc 290 ($host) = /^\s* ([0-9a-fA-F:]*:[0-9a-fA-F:]*:[0-9a-fA-F\.:]*)/xgc
220 and parse_ipv6 $host 291 and parse_ipv6 $host
221 ) { 292 ) {
222 /^\s*/xgc; 293 /^\s*/xgc;
229 return; 300 return;
230 } 301 }
231 } 302 }
232 303
233 # parse port 304 # parse port
234 if (/\G (?:\s+|:) ([^:[:space:]]+) \s*$/xgc) { 305 if (/\G (?:\s+|:|\#) ([^:[:space:]]+) \s*$/xgc) {
235 $port = $1; 306 $port = $1;
236 } elsif (/\G\s*$/gc && length $_[1]) { 307 } elsif (/\G\s*$/gc && length $_[1]) {
237 $port = $_[1]; 308 $port = $_[1];
238 } else { 309 } else {
239 return; 310 return;
240 } 311 }
312
241 } 313 }
242 314
243 # hostnames must not contain :'s 315 # hostnames must not contain :'s
244 return if $host =~ /:/ && !parse_ipv6 $host; 316 return if $host =~ /:/ && !parse_ipv6 $host;
245 317
246 ($host, $port) 318 ($host, $port)
319}
320
321=item $string = format_hostport $host, $port
322
323Takes a host (in textual form) and a port and formats in unambigiously in
324a way that C<parse_hostport> can parse it again. C<$port> can be C<undef>.
325
326=cut
327
328sub format_hostport($;$) {
329 my ($host, $port) = @_;
330
331 $port = ":$port" if length $port;
332 $host = "[$host]" if $host =~ /:/;
333
334 "$host$port"
247} 335}
248 336
249=item $sa_family = address_family $ipn 337=item $sa_family = address_family $ipn
250 338
251Returns the address family/protocol-family (AF_xxx/PF_xxx, in one value :) 339Returns the address family/protocol-family (AF_xxx/PF_xxx, in one value :)
259 : 16 == length $_[0] 347 : 16 == length $_[0]
260 ? AF_INET6 348 ? AF_INET6
261 : unpack "S", $_[0] 349 : unpack "S", $_[0]
262} 350}
263 351
352=item $text = format_ipv4 $ipn
353
354Expects a four octet string representing a binary IPv4 address and returns
355its textual format. Rarely used, see C<format_address> for a nicer
356interface.
357
358=item $text = format_ipv6 $ipn
359
360Expects a sixteen octet string representing a binary IPv6 address and
361returns its textual format. Rarely used, see C<format_address> for a
362nicer interface.
363
264=item $text = format_address $ipn 364=item $text = format_address $ipn
265 365
266Covnvert a host address in network format (e.g. 4 octets for IPv4 or 16 366Covnvert a host address in network format (e.g. 4 octets for IPv4 or 16
267octets for IPv6) and convert it into textual form. 367octets for IPv6) and convert it into textual form.
268 368
271This function works similarly to C<inet_ntop AF_INET || AF_INET6, ...>, 371This function works similarly to C<inet_ntop AF_INET || AF_INET6, ...>,
272except it automatically detects the address type. 372except it automatically detects the address type.
273 373
274Returns C<undef> if it cannot detect the type. 374Returns C<undef> if it cannot detect the type.
275 375
376If the C<$ipn> is a mapped IPv4 in IPv6 address (:ffff::<ipv4>), then just
377the contained IPv4 address will be returned. If you do not want that, you
378have to call C<format_ipv6> manually.
379
380Example:
381
382 print format_address "\x01\x02\x03\x05";
383 => 1.2.3.5
384
276=item $text = AnyEvent::Socket::ntoa $ipn 385=item $text = AnyEvent::Socket::ntoa $ipn
277 386
278Same as format_address, but not exported (think C<inet_ntoa>). 387Same as format_address, but not exported (think C<inet_ntoa>).
279 388
280=cut 389=cut
281 390
282sub format_address; 391sub format_ipv4($) {
283sub format_address($) {
284 my $af = address_family $_[0];
285 if ($af == AF_INET) {
286 return join ".", unpack "C4", $_[0] 392 join ".", unpack "C4", $_[0]
287 } elsif ($af == AF_INET6) { 393}
394
395sub format_ipv6($) {
396 if ($_[0] =~ /^\x00\x00\x00\x00\x00\x00\x00\x00/) {
288 if (v0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0 eq $_[0]) { 397 if (v0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0 eq $_[0]) {
289 return "::"; 398 return "::";
290 } elsif (v0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.1 eq $_[0]) { 399 } elsif (v0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.1 eq $_[0]) {
291 return "::1"; 400 return "::1";
292 } elsif (v0.0.0.0.0.0.0.0.0.0.0.0 eq substr $_[0], 0, 12) { 401 } elsif (v0.0.0.0.0.0.0.0.0.0.0.0 eq substr $_[0], 0, 12) {
293 # v4compatible 402 # v4compatible
294 return "::" . format_address substr $_[0], 12; 403 return "::" . format_ipv4 substr $_[0], 12;
295 } elsif (v0.0.0.0.0.0.0.0.0.0.255.255 eq substr $_[0], 0, 12) { 404 } elsif (v0.0.0.0.0.0.0.0.0.0.255.255 eq substr $_[0], 0, 12) {
296 # v4mapped 405 # v4mapped
297 return "::ffff:" . format_address substr $_[0], 12; 406 return "::ffff:" . format_ipv4 substr $_[0], 12;
298 } elsif (v0.0.0.0.0.0.0.0.255.255.0.0 eq substr $_[0], 0, 12) { 407 } elsif (v0.0.0.0.0.0.0.0.255.255.0.0 eq substr $_[0], 0, 12) {
299 # v4translated 408 # v4translated
300 return "::ffff:0:" . format_address substr $_[0], 12; 409 return "::ffff:0:" . format_ipv4 substr $_[0], 12;
301 } else {
302 my $ip = sprintf "%x:%x:%x:%x:%x:%x:%x:%x", unpack "n8", $_[0];
303
304 # this is rather sucky, I admit
305 $ip =~ s/^0:(?:0:)*(0$)?/::/
306 or $ip =~ s/(:0){7}$/::/ or $ip =~ s/(:0){7}/:/
307 or $ip =~ s/(:0){6}$/::/ or $ip =~ s/(:0){6}/:/
308 or $ip =~ s/(:0){5}$/::/ or $ip =~ s/(:0){5}/:/
309 or $ip =~ s/(:0){4}$/::/ or $ip =~ s/(:0){4}/:/
310 or $ip =~ s/(:0){3}$/::/ or $ip =~ s/(:0){3}/:/
311 or $ip =~ s/(:0){2}$/::/ or $ip =~ s/(:0){2}/:/
312 or $ip =~ s/(:0){1}$/::/ or $ip =~ s/(:0){1}/:/;
313 return $ip
314 } 410 }
315 } elsif ($af == AF_UNIX) { 411 }
412
413 my $ip = sprintf "%x:%x:%x:%x:%x:%x:%x:%x", unpack "n8", $_[0];
414
415 # this is admittedly rather sucky
416 $ip =~ s/(?:^|:) 0:0:0:0:0:0:0 (?:$|:)/::/x
417 or $ip =~ s/(?:^|:) 0:0:0:0:0:0 (?:$|:)/::/x
418 or $ip =~ s/(?:^|:) 0:0:0:0:0 (?:$|:)/::/x
419 or $ip =~ s/(?:^|:) 0:0:0:0 (?:$|:)/::/x
420 or $ip =~ s/(?:^|:) 0:0:0 (?:$|:)/::/x
421 or $ip =~ s/(?:^|:) 0:0 (?:$|:)/::/x
422 or $ip =~ s/(?:^|:) 0 (?:$|:)/::/x;
423
424 $ip
425}
426
427sub format_address($) {
428 if (4 == length $_[0]) {
429 return &format_ipv4;
430 } elsif (16 == length $_[0]) {
431 return $_[0] =~ /^\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff(....)$/s
432 ? format_ipv4 $1
433 : &format_ipv6;
434 } elsif (AF_UNIX == address_family $_[0]) {
316 return "unix/" 435 return "unix/"
317 } else { 436 } else {
318 return undef 437 return undef
319 } 438 }
320} 439}
322*ntoa = \&format_address; 441*ntoa = \&format_address;
323 442
324=item inet_aton $name_or_address, $cb->(@addresses) 443=item inet_aton $name_or_address, $cb->(@addresses)
325 444
326Works similarly to its Socket counterpart, except that it uses a 445Works similarly to its Socket counterpart, except that it uses a
327callback. Also, if a host has only an IPv6 address, this might be passed 446callback. Use the length to distinguish between ipv4 and ipv6 (4 octets
328to the callback instead (use the length to detect this - 4 for IPv4, 16 447for IPv4, 16 for IPv6), or use C<format_address> to convert it to a more
329for IPv6). 448readable format.
330 449
331Unlike the L<Socket> function of the same name, you can get multiple IPv4 450Note that C<resolve_sockaddr>, while initially a more complex interface,
332and IPv6 addresses as result (and maybe even other adrdess types). 451resolves host addresses, IDNs, service names and SRV records and gives you
452an ordered list of socket addresses to try and should be preferred over
453C<inet_aton>.
454
455Example.
456
457 inet_aton "www.google.com", my $cv = AE::cv;
458 say unpack "H*", $_
459 for $cv->recv;
460 # => d155e363
461 # => d155e367 etc.
462
463 inet_aton "ipv6.google.com", my $cv = AE::cv;
464 say unpack "H*", $_
465 for $cv->recv;
466 # => 20014860a00300000000000000000068
333 467
334=cut 468=cut
335 469
336sub inet_aton { 470sub inet_aton {
337 my ($name, $cb) = @_; 471 my ($name, $cb) = @_;
341 } elsif (my $ipn = &parse_ipv6) { 475 } elsif (my $ipn = &parse_ipv6) {
342 $cb->($ipn); 476 $cb->($ipn);
343 } elsif ($name eq "localhost") { # rfc2606 et al. 477 } elsif ($name eq "localhost") { # rfc2606 et al.
344 $cb->(v127.0.0.1, v0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.1); 478 $cb->(v127.0.0.1, v0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.1);
345 } else { 479 } else {
346 require AnyEvent::DNS; 480 require AnyEvent::DNS unless $AnyEvent::DNS::VERSION;
347 481
348 # simple, bad suboptimal algorithm 482 my $ipv4 = $AnyEvent::PROTOCOL{ipv4};
483 my $ipv6 = $AnyEvent::PROTOCOL{ipv6};
484
485 my @res;
486
487 my $cv = AE::cv {
488 $cb->(map @$_, reverse @res);
489 };
490
491 $cv->begin;
492
493 if ($ipv4) {
494 $cv->begin;
349 AnyEvent::DNS::a ($name, sub { 495 AnyEvent::DNS::a ($name, sub {
350 if (@_) { 496 $res[$ipv4] = [map { parse_ipv4 $_ } @_];
351 $cb->(map +(parse_ipv4 $_), @_);
352 } else {
353 $cb->(); 497 $cv->end;
354 #AnyEvent::DNS::aaaa ($name, $cb); need inet_pton
355 } 498 });
356 }); 499 };
357 }
358}
359 500
501 if ($ipv6) {
502 $cv->begin;
503 AnyEvent::DNS::aaaa ($name, sub {
504 $res[$ipv6] = [map { parse_ipv6 $_ } @_];
505 $cv->end;
506 });
507 };
508
509 $cv->end;
510 }
511}
512
513BEGIN {
514 *sockaddr_family = $Socket::VERSION >= 1.75
515 ? \&Socket::sockaddr_family
516 : # for 5.6.x, we need to do something much more horrible
517 (Socket::pack_sockaddr_in 0x5555, "\x55\x55\x55\x55"
518 | eval { Socket::pack_sockaddr_un "U" }) =~ /^\x00/
519 ? sub { unpack "xC", $_[0] }
520 : sub { unpack "S" , $_[0] };
521}
522
360# check for broken platforms with extra field in sockaddr structure 523# check for broken platforms with an extra field in sockaddr structure
361# kind of a rfc vs. bsd issue, as usual (ok, normally it's a 524# kind of a rfc vs. bsd issue, as usual (ok, normally it's a
362# unix vs. bsd issue, a iso C vs. bsd issue or simply a 525# unix vs. bsd issue, a iso C vs. bsd issue or simply a
363# correctness vs. bsd issue. 526# correctness vs. bsd issue.)
364my $pack_family = (0x55 == Socket::sockaddr_family "\x55\x55") 527my $pack_family = 0x55 == sockaddr_family ("\x55\x55")
365 ? "xC" : "S"; 528 ? "xC" : "S";
366 529
367=item $sa = AnyEvent::Socket::pack_sockaddr $service, $host 530=item $sa = AnyEvent::Socket::pack_sockaddr $service, $host
368 531
369Pack the given port/host combination into a binary sockaddr 532Pack the given port/host combination into a binary sockaddr
370structure. Handles both IPv4 and IPv6 host addresses, as well as UNIX 533structure. Handles both IPv4 and IPv6 host addresses, as well as UNIX
371domain sockets (C<$host> == C<unix/> and C<$service> == absolute 534domain sockets (C<$host> == C<unix/> and C<$service> == absolute
372pathname). 535pathname).
536
537Example:
538
539 my $bind = AnyEvent::Socket::pack_sockaddr 43, v195.234.53.120;
540 bind $socket, $bind
541 or die "bind: $!";
373 542
374=cut 543=cut
375 544
376sub pack_sockaddr($$) { 545sub pack_sockaddr($$) {
377 my $af = address_family $_[1]; 546 my $af = address_family $_[1];
404is a special token that is understood by the other functions in this 573is a special token that is understood by the other functions in this
405module (C<format_address> converts it to C<unix/>). 574module (C<format_address> converts it to C<unix/>).
406 575
407=cut 576=cut
408 577
578# perl contains a bug (imho) where it requires that the kernel always returns
579# sockaddr_un structures of maximum length (which is not, AFAICS, required
580# by any standard). try to 0-pad structures for the benefit of those platforms.
581
582my $sa_un_zero = eval { Socket::pack_sockaddr_un "" }; $sa_un_zero ^= $sa_un_zero;
583
409sub unpack_sockaddr($) { 584sub unpack_sockaddr($) {
410 my $af = Socket::sockaddr_family $_[0]; 585 my $af = sockaddr_family $_[0];
411 586
412 if ($af == AF_INET) { 587 if ($af == AF_INET) {
413 Socket::unpack_sockaddr_in $_[0] 588 Socket::unpack_sockaddr_in $_[0]
414 } elsif ($af == AF_INET6) { 589 } elsif ($af == AF_INET6) {
415 unpack "x2 n x4 a16", $_[0] 590 unpack "x2 n x4 a16", $_[0]
416 } elsif ($af == AF_UNIX) { 591 } elsif ($af == AF_UNIX) {
417 ((Socket::unpack_sockaddr_un $_[0]), pack "S", AF_UNIX) 592 ((Socket::unpack_sockaddr_un $_[0] ^ $sa_un_zero), pack "S", AF_UNIX)
418 } else { 593 } else {
419 Carp::croak "unpack_sockaddr: unsupported protocol family $af"; 594 Carp::croak "unpack_sockaddr: unsupported protocol family $af";
420 } 595 }
421} 596}
422 597
425Tries to resolve the given nodename and service name into protocol families 600Tries to resolve the given nodename and service name into protocol families
426and sockaddr structures usable to connect to this node and service in a 601and sockaddr structures usable to connect to this node and service in a
427protocol-independent way. It works remotely similar to the getaddrinfo 602protocol-independent way. It works remotely similar to the getaddrinfo
428posix function. 603posix function.
429 604
430For internet addresses, C<$node> is either an IPv4 or IPv6 address or an 605For internet addresses, C<$node> is either an IPv4 or IPv6 address, an
431internet hostname, and C<$service> is either a service name (port name 606internet hostname (DNS domain name or IDN), and C<$service> is either
432from F</etc/services>) or a numerical port number. If both C<$node> and 607a service name (port name from F</etc/services>) or a numerical port
433C<$service> are names, then SRV records will be consulted to find the real 608number. If both C<$node> and C<$service> are names, then SRV records
434service, otherwise they will be used as-is. If you know that the service 609will be consulted to find the real service, otherwise they will be
435name is not in your services database, then you can specify the service in 610used as-is. If you know that the service name is not in your services
436the format C<name=port> (e.g. C<http=80>). 611database, then you can specify the service in the format C<name=port>
612(e.g. C<http=80>).
613
614If a host cannot be found via DNS, then it will be looked up in
615F</etc/hosts> (or the file specified via C<< $ENV{PERL_ANYEVENT_HOSTS}
616>>). If they are found, the addresses there will be used. The effect is as
617if entries from F</etc/hosts> would yield C<A> and C<AAAA> records for the
618host name unless DNS already had records for them.
437 619
438For UNIX domain sockets, C<$node> must be the string C<unix/> and 620For UNIX domain sockets, C<$node> must be the string C<unix/> and
439C<$service> must be the absolute pathname of the socket. In this case, 621C<$service> must be the absolute pathname of the socket. In this case,
440C<$proto> will be ignored. 622C<$proto> will be ignored.
441 623
443C<sctp>. The default is currently C<tcp>, but in the future, this function 625C<sctp>. The default is currently C<tcp>, but in the future, this function
444might try to use other protocols such as C<sctp>, depending on the socket 626might try to use other protocols such as C<sctp>, depending on the socket
445type and any SRV records it might find. 627type and any SRV records it might find.
446 628
447C<$family> must be either C<0> (meaning any protocol is OK), C<4> (use 629C<$family> must be either C<0> (meaning any protocol is OK), C<4> (use
448only IPv4) or C<6> (use only IPv6). This setting might be influenced by 630only IPv4) or C<6> (use only IPv6). The default is influenced by
449C<$ENV{PERL_ANYEVENT_PROTOCOLS}>. 631C<$ENV{PERL_ANYEVENT_PROTOCOLS}>.
450 632
451C<$type> must be C<SOCK_STREAM>, C<SOCK_DGRAM> or C<SOCK_SEQPACKET> (or 633C<$type> must be C<SOCK_STREAM>, C<SOCK_DGRAM> or C<SOCK_SEQPACKET> (or
452C<undef> in which case it gets automatically chosen). 634C<undef> in which case it gets automatically chosen to be C<SOCK_STREAM>
635unless C<$proto> is C<udp>).
453 636
454The callback will receive zero or more array references that contain 637The callback will receive zero or more array references that contain
455C<$family, $type, $proto> for use in C<socket> and a binary 638C<$family, $type, $proto> for use in C<socket> and a binary
456C<$sockaddr> for use in C<connect> (or C<bind>). 639C<$sockaddr> for use in C<connect> (or C<bind>).
457 640
461 644
462 resolve_sockaddr "google.com", "http", 0, undef, undef, sub { ... }; 645 resolve_sockaddr "google.com", "http", 0, undef, undef, sub { ... };
463 646
464=cut 647=cut
465 648
466# microsoft can't even get getprotobyname working (the etc/protocols file 649our %HOSTS; # $HOSTS{$nodename}[$ipv6] = [@aliases...]
467# gets lost fairly often on windows), so we have to hardcode some common 650our @HOSTS_CHECKING; # callbacks to call when hosts have been loaded
468# protocol numbers ourselves. 651our $HOSTS_MTIME;
469our %PROTO_BYNAME;
470 652
471$PROTO_BYNAME{tcp} = &Socket::IPPROTO_TCP if defined &Socket::IPPROTO_TCP; 653sub _parse_hosts($) {
472$PROTO_BYNAME{udp} = &Socket::IPPROTO_UDP if defined &Socket::IPPROTO_UDP; 654 %HOSTS = ();
473$PROTO_BYNAME{icmp} = &Socket::IPPROTO_ICMP if defined &Socket::IPPROTO_ICMP; 655
656 for (split /\n/, $_[0]) {
657 s/#.*$//;
658 s/^[ \t]+//;
659 y/A-Z/a-z/;
660
661 my ($addr, @aliases) = split /[ \t]+/;
662 next unless @aliases;
663
664 if (my $ip = parse_ipv4 $addr) {
665 push @{ $HOSTS{$_}[0] }, $ip
666 for @aliases;
667 } elsif (my $ip = parse_ipv6 $addr) {
668 push @{ $HOSTS{$_}[1] }, $ip
669 for @aliases;
670 }
671 }
672}
673
674# helper function - unless dns delivered results, check and parse hosts, then clal continuation code
675sub _load_hosts_unless(&$@) {
676 my ($cont, $cv, @dns) = @_;
677
678 if (@dns) {
679 $cv->end;
680 } else {
681 my $etc_hosts = length $ENV{PERL_ANYEVENT_HOSTS} ? $ENV{PERL_ANYEVENT_HOSTS}
682 : AnyEvent::WIN32 ? "$ENV{SystemRoot}/system32/drivers/etc/hosts"
683 : "/etc/hosts";
684
685 push @HOSTS_CHECKING, sub {
686 $cont->();
687 $cv->end;
688 };
689
690 unless ($#HOSTS_CHECKING) {
691 # we are not the first, so we actually have to do the work
692 require AnyEvent::IO;
693
694 AnyEvent::IO::aio_stat ($etc_hosts, sub {
695 if ((stat _)[9] ne $HOSTS_MTIME) {
696 AE::log 8 => "(re)loading $etc_hosts.";
697 $HOSTS_MTIME = (stat _)[9];
698 # we might load a newer version of hosts,but that's a harmless race,
699 # as the next call will just load it again.
700 AnyEvent::IO::aio_load ($etc_hosts, sub {
701 _parse_hosts $_[0];
702 (shift @HOSTS_CHECKING)->() while @HOSTS_CHECKING;
703 });
704 } else {
705 (shift @HOSTS_CHECKING)->() while @HOSTS_CHECKING;
706 }
707 });
708 }
709 }
710}
474 711
475sub resolve_sockaddr($$$$$$) { 712sub resolve_sockaddr($$$$$$) {
476 my ($node, $service, $proto, $family, $type, $cb) = @_; 713 my ($node, $service, $proto, $family, $type, $cb) = @_;
477 714
478 if ($node eq "unix/") { 715 if ($node eq "unix/") {
479 return $cb->() if $family || !/^\//; # no can do 716 return $cb->() if $family || $service !~ /^\//; # no can do
480 717
481 return $cb->([AF_UNIX, $type, 0, Socket::pack_sockaddr_un $service]); 718 return $cb->([AF_UNIX, defined $type ? $type : SOCK_STREAM, 0, Socket::pack_sockaddr_un $service]);
482 } 719 }
483 720
484 unless (AF_INET6) { 721 unless (AF_INET6) {
485 $family != 6 722 $family != 6
486 or return $cb->(); 723 or return $cb->();
495 $family ||= 6 unless $AnyEvent::PROTOCOL{ipv4}; 732 $family ||= 6 unless $AnyEvent::PROTOCOL{ipv4};
496 733
497 $proto ||= "tcp"; 734 $proto ||= "tcp";
498 $type ||= $proto eq "udp" ? SOCK_DGRAM : SOCK_STREAM; 735 $type ||= $proto eq "udp" ? SOCK_DGRAM : SOCK_STREAM;
499 736
500 my $proton = $PROTO_BYNAME{lc $proto} || (getprotobyname $proto)[2] 737 my $proton = AnyEvent::Socket::getprotobyname $proto
501 or Carp::croak "$proto: protocol unknown"; 738 or Carp::croak "$proto: protocol unknown";
502 739
503 my $port; 740 my $port;
504 741
505 if ($service =~ /^(\S+)=(\d+)$/) { 742 if ($service =~ /^(\S+)=(\d+)$/) {
509 } else { 746 } else {
510 $port = (getservbyname $service, $proto)[2] 747 $port = (getservbyname $service, $proto)[2]
511 or Carp::croak "$service/$proto: service unknown"; 748 or Carp::croak "$service/$proto: service unknown";
512 } 749 }
513 750
514 my @target = [$node, $port];
515
516 # resolve a records / provide sockaddr structures 751 # resolve a records / provide sockaddr structures
517 my $resolve = sub { 752 my $resolve = sub {
753 my @target = @_;
754
518 my @res; 755 my @res;
519 my $cv = AnyEvent->condvar (cb => sub { 756 my $cv = AE::cv {
520 $cb->( 757 $cb->(
521 map $_->[2], 758 map $_->[2],
522 sort { 759 sort {
523 $AnyEvent::PROTOCOL{$b->[1]} <=> $AnyEvent::PROTOCOL{$a->[1]} 760 $AnyEvent::PROTOCOL{$b->[1]} <=> $AnyEvent::PROTOCOL{$a->[1]}
524 or $a->[0] <=> $b->[0] 761 or $a->[0] <=> $b->[0]
525 } 762 }
526 @res 763 @res
527 ) 764 )
528 }); 765 };
529 766
530 $cv->begin; 767 $cv->begin;
531 for my $idx (0 .. $#target) { 768 for my $idx (0 .. $#target) {
532 my ($node, $port) = @{ $target[$idx] }; 769 my ($node, $port) = @{ $target[$idx] };
533 770
542 if ($af == AF_INET6 && $family != 4) { 779 if ($af == AF_INET6 && $family != 4) {
543 push @res, [$idx, "ipv6", [AF_INET6, $type, $proton, 780 push @res, [$idx, "ipv6", [AF_INET6, $type, $proton,
544 pack_sockaddr $port, $noden]] 781 pack_sockaddr $port, $noden]]
545 } 782 }
546 } else { 783 } else {
547 # ipv4 784 $node =~ y/A-Z/a-z/;
785
786 my $hosts = $HOSTS{$node};
787
788 # a records
548 if ($family != 6) { 789 if ($family != 6) {
549 $cv->begin; 790 $cv->begin;
550 AnyEvent::DNS::a $node, sub { 791 AnyEvent::DNS::a $node, sub {
551 push @res, [$idx, "ipv4", [AF_INET, $type, $proton, 792 push @res, [$idx, "ipv4", [AF_INET , $type, $proton, pack_sockaddr $port, parse_ipv4 $_]]
552 pack_sockaddr $port, parse_ipv4 $_]]
553 for @_; 793 for @_;
794
795 # dns takes precedence over hosts
796 _load_hosts_unless {
797 push @res,
798 map [$idx, "ipv4", [AF_INET , $type, $proton, pack_sockaddr $port, $_]],
799 @{ $HOSTS{$node}[0] };
554 $cv->end; 800 } $cv, @_;
555 }; 801 };
556 } 802 }
557 803
558 # ipv6 804 # aaaa records
559 if ($family != 4) { 805 if ($family != 4) {
560 $cv->begin; 806 $cv->begin;
561 AnyEvent::DNS::aaaa $node, sub { 807 AnyEvent::DNS::aaaa $node, sub {
562 push @res, [$idx, "ipv6", [AF_INET6, $type, $proton, 808 push @res, [$idx, "ipv6", [AF_INET6, $type, $proton, pack_sockaddr $port, parse_ipv6 $_]]
563 pack_sockaddr $port, parse_ipv6 $_]]
564 for @_; 809 for @_;
810
811 _load_hosts_unless {
812 push @res,
813 map [$idx + 0.5, "ipv6", [AF_INET6, $type, $proton, pack_sockaddr $port, $_]],
814 @{ $HOSTS{$node}[1] }
565 $cv->end; 815 } $cv, @_;
566 }; 816 };
567 } 817 }
568 } 818 }
569 } 819 }
570 $cv->end; 820 $cv->end;
571 }; 821 };
572 822
823 $node = AnyEvent::Util::idn_to_ascii $node
824 if $node =~ /[^\x00-\x7f]/;
825
573 # try srv records, if applicable 826 # try srv records, if applicable
574 if ($node eq "localhost") { 827 if ($node eq "localhost") {
575 @target = (["127.0.0.1", $port], ["::1", $port]); 828 $resolve->(["127.0.0.1", $port], ["::1", $port]);
576 &$resolve;
577 } elsif (defined $service && !parse_address $node) { 829 } elsif (defined $service && !parse_address $node) {
578 AnyEvent::DNS::srv $service, $proto, $node, sub { 830 AnyEvent::DNS::srv $service, $proto, $node, sub {
579 my (@srv) = @_; 831 my (@srv) = @_;
580 832
581 # no srv records, continue traditionally
582 @srv 833 if (@srv) {
583 or return &$resolve;
584
585 # the only srv record has "." ("" here) => abort 834 # the only srv record has "." ("" here) => abort
586 $srv[0][2] ne "" || $#srv 835 $srv[0][2] ne "" || $#srv
587 or return $cb->(); 836 or return $cb->();
588 837
589 # use srv records then 838 # use srv records then
839 $resolve->(
590 @target = map ["$_->[3].", $_->[2]], 840 map ["$_->[3].", $_->[2]],
591 grep $_->[3] ne ".", 841 grep $_->[3] ne ".",
592 @srv; 842 @srv
593 843 );
594 &$resolve; 844 } else {
845 # no srv records, continue traditionally
846 $resolve->([$node, $port]);
847 }
595 }; 848 };
596 } else { 849 } else {
597 &$resolve; 850 # most common case
851 $resolve->([$node, $port]);
598 } 852 }
599} 853}
600 854
601=item $guard = tcp_connect $host, $service, $connect_cb[, $prepare_cb] 855=item $guard = tcp_connect $host, $service, $connect_cb[, $prepare_cb]
602 856
603This is a convenience function that creates a TCP socket and makes a 100% 857This is a convenience function that creates a TCP socket and makes a
604non-blocking connect to the given C<$host> (which can be a hostname or 858100% non-blocking connect to the given C<$host> (which can be a DNS/IDN
605a textual IP address, or the string C<unix/> for UNIX domain sockets) 859hostname or a textual IP address, or the string C<unix/> for UNIX domain
606and C<$service> (which can be a numeric port number or a service name, 860sockets) and C<$service> (which can be a numeric port number or a service
607or a C<servicename=portnumber> string, or the pathname to a UNIX domain 861name, or a C<servicename=portnumber> string, or the pathname to a UNIX
608socket). 862domain socket).
609 863
610If both C<$host> and C<$port> are names, then this function will use SRV 864If both C<$host> and C<$port> are names, then this function will use SRV
611records to locate the real target(s). 865records to locate the real target(s).
612 866
613In either case, it will create a list of target hosts (e.g. for multihomed 867In either case, it will create a list of target hosts (e.g. for multihomed
614hosts or hosts with both IPv4 and IPv6 addresses) and try to connect to 868hosts or hosts with both IPv4 and IPv6 addresses) and try to connect to
615each in turn. 869each in turn.
616 870
617If the connect is successful, then the C<$connect_cb> will be invoked with 871After the connection is established, then the C<$connect_cb> will be
618the socket file handle (in non-blocking mode) as first and the peer host 872invoked with the socket file handle (in non-blocking mode) as first, and
619(as a textual IP address) and peer port as second and third arguments, 873the peer host (as a textual IP address) and peer port as second and third
620respectively. The fourth argument is a code reference that you can call 874arguments, respectively. The fourth argument is a code reference that you
621if, for some reason, you don't like this connection, which will cause 875can call if, for some reason, you don't like this connection, which will
622C<tcp_connect> to try the next one (or call your callback without any 876cause C<tcp_connect> to try the next one (or call your callback without
623arguments if there are no more connections). In most cases, you can simply 877any arguments if there are no more connections). In most cases, you can
624ignore this argument. 878simply ignore this argument.
625 879
626 $cb->($filehandle, $host, $port, $retry) 880 $cb->($filehandle, $host, $port, $retry)
627 881
628If the connect is unsuccessful, then the C<$connect_cb> will be invoked 882If the connect is unsuccessful, then the C<$connect_cb> will be invoked
629without any arguments and C<$!> will be set appropriately (with C<ENXIO> 883without any arguments and C<$!> will be set appropriately (with C<ENXIO>
630indicating a DNS resolution failure). 884indicating a DNS resolution failure).
631 885
886The callback will I<never> be invoked before C<tcp_connect> returns, even
887if C<tcp_connect> was able to connect immediately (e.g. on unix domain
888sockets).
889
632The file handle is perfect for being plugged into L<AnyEvent::Handle>, but 890The file handle is perfect for being plugged into L<AnyEvent::Handle>, but
633can be used as a normal perl file handle as well. 891can be used as a normal perl file handle as well.
634 892
635Unless called in void context, C<tcp_connect> returns a guard object that 893Unless called in void context, C<tcp_connect> returns a guard object that
636will automatically abort connecting when it gets destroyed (it does not do 894will automatically cancel the connection attempt when it gets destroyed
895- in which case the callback will not be invoked. Destroying it does not
637anything to the socket after the connect was successful). 896do anything to the socket after the connect was successful - you cannot
897"uncall" a callback that has been invoked already.
638 898
639Sometimes you need to "prepare" the socket before connecting, for example, 899Sometimes you need to "prepare" the socket before connecting, for example,
640to C<bind> it to some port, or you want a specific connect timeout that 900to C<bind> it to some port, or you want a specific connect timeout that
641is lower than your kernel's default timeout. In this case you can specify 901is lower than your kernel's default timeout. In this case you can specify
642a second callback, C<$prepare_cb>. It will be called with the file handle 902a second callback, C<$prepare_cb>. It will be called with the file handle
674 or die "unable to connect: $!"; 934 or die "unable to connect: $!";
675 935
676 my $handle; # avoid direct assignment so on_eof has it in scope. 936 my $handle; # avoid direct assignment so on_eof has it in scope.
677 $handle = new AnyEvent::Handle 937 $handle = new AnyEvent::Handle
678 fh => $fh, 938 fh => $fh,
939 on_error => sub {
940 AE::log error => $_[2];
941 $_[0]->destroy;
942 },
679 on_eof => sub { 943 on_eof => sub {
680 undef $handle; # keep it alive till eof 944 $handle->destroy; # destroy handle
681 warn "done.\n"; 945 AE::log info => "Done.";
682 }; 946 };
683 947
684 $handle->push_write ("GET / HTTP/1.0\015\012\015\012"); 948 $handle->push_write ("GET / HTTP/1.0\015\012\015\012");
685 949
686 $handle->push_read_line ("\015\012\015\012", sub { 950 $handle->push_read (line => "\015\012\015\012", sub {
687 my ($handle, $line) = @_; 951 my ($handle, $line) = @_;
688 952
689 # print response header 953 # print response header
690 print "HEADER\n$line\n\nBODY\n"; 954 print "HEADER\n$line\n\nBODY\n";
691 955
711=cut 975=cut
712 976
713sub tcp_connect($$$;$) { 977sub tcp_connect($$$;$) {
714 my ($host, $port, $connect, $prepare) = @_; 978 my ($host, $port, $connect, $prepare) = @_;
715 979
716 # see http://cr.yp.to/docs/connect.html for some background 980 # see http://cr.yp.to/docs/connect.html for some tricky aspects
717 # also http://advogato.org/article/672.html 981 # also http://advogato.org/article/672.html
718 982
719 my %state = ( fh => undef ); 983 my %state = ( fh => undef );
720 984
721 # name/service to type/sockaddr resolution 985 # name/service to type/sockaddr resolution
722 resolve_sockaddr $host, $port, 0, 0, 0, sub { 986 resolve_sockaddr $host, $port, 0, 0, undef, sub {
723 my @target = @_; 987 my @target = @_;
724 988
725 $state{next} = sub { 989 $state{next} = sub {
726 return unless exists $state{fh}; 990 return unless exists $state{fh};
727 991
992 my $errno = $!;
728 my $target = shift @target 993 my $target = shift @target
729 or do { 994 or return AE::postpone {
995 return unless exists $state{fh};
730 %state = (); 996 %state = ();
997 $! = $errno;
731 return $connect->(); 998 $connect->();
732 }; 999 };
733 1000
734 my ($domain, $type, $proto, $sockaddr) = @$target; 1001 my ($domain, $type, $proto, $sockaddr) = @$target;
735 1002
736 # socket creation 1003 # socket creation
741 1008
742 my $timeout = $prepare && $prepare->($state{fh}); 1009 my $timeout = $prepare && $prepare->($state{fh});
743 1010
744 $timeout ||= 30 if AnyEvent::WIN32; 1011 $timeout ||= 30 if AnyEvent::WIN32;
745 1012
746 $state{to} = AnyEvent->timer (after => $timeout, cb => sub { 1013 $state{to} = AE::timer $timeout, 0, sub {
747 $! = &Errno::ETIMEDOUT; 1014 $! = Errno::ETIMEDOUT;
748 $state{next}(); 1015 $state{next}();
749 }) if $timeout; 1016 } if $timeout;
750 1017
751 # called when the connect was successful, which, 1018 # now connect
752 # in theory, could be the case immediately (but never is in practise) 1019 if (
753 my $connected = sub { 1020 (connect $state{fh}, $sockaddr)
754 delete $state{ww}; 1021 || ($! == Errno::EINPROGRESS # POSIX
755 delete $state{to}; 1022 || $! == Errno::EWOULDBLOCK
756 1023 # WSAEINPROGRESS intentionally not checked - it means something else entirely
1024 || $! == AnyEvent::Util::WSAEINVAL # not convinced, but doesn't hurt
1025 || $! == AnyEvent::Util::WSAEWOULDBLOCK)
1026 ) {
1027 $state{ww} = AE::io $state{fh}, 1, sub {
757 # we are connected, or maybe there was an error 1028 # we are connected, or maybe there was an error
758 if (my $sin = getpeername $state{fh}) { 1029 if (my $sin = getpeername $state{fh}) {
759 my ($port, $host) = unpack_sockaddr $sin; 1030 my ($port, $host) = unpack_sockaddr $sin;
760 1031
1032 delete $state{ww}; delete $state{to};
1033
761 my $guard = guard { 1034 my $guard = guard { %state = () };
762 %state = ();
763 };
764 1035
765 $connect->($state{fh}, format_address $host, $port, sub { 1036 $connect->(delete $state{fh}, format_address $host, $port, sub {
766 $guard->cancel; 1037 $guard->cancel;
1038 $state{next}();
1039 });
1040 } else {
1041 if ($! == Errno::ENOTCONN) {
1042 # dummy read to fetch real error code if !cygwin
1043 sysread $state{fh}, my $buf, 1;
1044
1045 # cygwin 1.5 continously reports "ready' but never delivers
1046 # an error with getpeername or sysread.
1047 # cygwin 1.7 only reports readyness *once*, but is otherwise
1048 # the same, which is actually more broken.
1049 # Work around both by using unportable SO_ERROR for cygwin.
1050 $! = (unpack "l", getsockopt $state{fh}, Socket::SOL_SOCKET(), Socket::SO_ERROR()) || Errno::EAGAIN
1051 if AnyEvent::CYGWIN && $! == Errno::EAGAIN;
1052 }
1053
1054 return if $! == Errno::EAGAIN; # skip spurious wake-ups
1055
1056 delete $state{ww}; delete $state{to};
1057
767 $state{next}(); 1058 $state{next}();
768 }); 1059 }
769 } else {
770 # dummy read to fetch real error code
771 sysread $state{fh}, my $buf, 1 if $! == &Errno::ENOTCONN;
772 $state{next}();
773 } 1060 };
774 };
775
776 # now connect
777 if (connect $state{fh}, $sockaddr) {
778 $connected->();
779 } elsif ($! == &Errno::EINPROGRESS # POSIX
780 || $! == &Errno::EWOULDBLOCK
781 # WSAEINPROGRESS intentionally not checked - it means something else entirely
782 || $! == AnyEvent::Util::WSAEINVAL # not convinced, but doesn't hurt
783 || $! == AnyEvent::Util::WSAEWOULDBLOCK) {
784 $state{ww} = AnyEvent->io (fh => $state{fh}, poll => 'w', cb => $connected);
785 } else { 1061 } else {
786 $state{next}(); 1062 $state{next}();
787 } 1063 }
788 }; 1064 };
789 1065
790 $! = &Errno::ENXIO; 1066 $! = Errno::ENXIO;
791 $state{next}(); 1067 $state{next}();
792 }; 1068 };
793 1069
794 defined wantarray && guard { %state = () } 1070 defined wantarray && guard { %state = () }
795} 1071}
796 1072
797=item $guard = tcp_server $host, $service, $accept_cb[, $prepare_cb] 1073=item $guard = tcp_server $host, $service, $accept_cb[, $prepare_cb]
798 1074
799Create and bind a stream socket to the given host, and port, set the 1075Create and bind a stream socket to the given host address and port, set
800SO_REUSEADDR flag (if applicable) and call C<listen>. Unlike the name 1076the SO_REUSEADDR flag (if applicable) and call C<listen>. Unlike the name
801implies, this function can also bind on UNIX domain sockets. 1077implies, this function can also bind on UNIX domain sockets.
802 1078
803For internet sockets, C<$host> must be an IPv4 or IPv6 address (or 1079For internet sockets, C<$host> must be an IPv4 or IPv6 address (or
804C<undef>, in which case it binds either to C<0> or to C<::>, depending 1080C<undef>, in which case it binds either to C<0> or to C<::>, depending
805on whether IPv4 or IPv6 is the preferred protocol, and maybe to both in 1081on whether IPv4 or IPv6 is the preferred protocol, and maybe to both in
806future versions, as applicable). 1082future versions, as applicable).
807 1083
808To bind to the IPv4 wildcard address, use C<0>, to bind to the IPv6 1084To bind to the IPv4 wildcard address, use C<0>, to bind to the IPv6
809wildcard address, use C<::>. 1085wildcard address, use C<::>.
810 1086
811The port is specified by C<$service>, which must be either a service name or 1087The port is specified by C<$service>, which must be either a service name
812a numeric port number (or C<0> or C<undef>, in which case an ephemeral 1088or a numeric port number (or C<0> or C<undef>, in which case an ephemeral
813port will be used). 1089port will be used).
814 1090
815For UNIX domain sockets, C<$host> must be C<unix/> and C<$service> must be 1091For UNIX domain sockets, C<$host> must be C<unix/> and C<$service> must be
816the absolute pathname of the socket. This function will try to C<unlink> 1092the absolute pathname of the socket. This function will try to C<unlink>
817the socket before it tries to bind to it. See SECURITY CONSIDERATIONS, 1093the socket before it tries to bind to it, and will try to unlink it after
818below. 1094it stops using it. See SECURITY CONSIDERATIONS, below.
819 1095
820For each new connection that could be C<accept>ed, call the C<< 1096For each new connection that could be C<accept>ed, call the C<<
821$accept_cb->($fh, $host, $port) >> with the file handle (in non-blocking 1097$accept_cb->($fh, $host, $port) >> with the file handle (in non-blocking
822mode) as first and the peer host and port as second and third arguments 1098mode) as first, and the peer host and port as second and third arguments
823(see C<tcp_connect> for details). 1099(see C<tcp_connect> for details).
824 1100
825Croaks on any errors it can detect before the listen. 1101Croaks on any errors it can detect before the listen.
826 1102
827If called in non-void context, then this function returns a guard object 1103If called in non-void context, then this function returns a guard object
828whose lifetime it tied to the TCP server: If the object gets destroyed, 1104whose lifetime it tied to the TCP server: If the object gets destroyed,
829the server will be stopped (but existing accepted connections will 1105the server will be stopped (but existing accepted connections will
830continue). 1106not be affected).
1107
1108Regardless, when the function returns to the caller, the socket is bound
1109and in listening state.
831 1110
832If you need more control over the listening socket, you can provide a 1111If you need more control over the listening socket, you can provide a
833C<< $prepare_cb->($fh, $host, $port) >>, which is called just before the 1112C<< $prepare_cb->($fh, $host, $port) >>, which is called just before the
834C<listen ()> call, with the listen file handle as first argument, and IP 1113C<listen ()> call, with the listen file handle as first argument, and IP
835address and port number of the local socket endpoint as second and third 1114address and port number of the local socket endpoint as second and third
851 my ($fh, $host, $port) = @_; 1130 my ($fh, $host, $port) = @_;
852 1131
853 syswrite $fh, "The internet is full, $host:$port. Go away!\015\012"; 1132 syswrite $fh, "The internet is full, $host:$port. Go away!\015\012";
854 }, sub { 1133 }, sub {
855 my ($fh, $thishost, $thisport) = @_; 1134 my ($fh, $thishost, $thisport) = @_;
856 warn "bound to $thishost, port $thisport\n"; 1135 AE::log info => "Bound to $thishost, port $thisport.";
1136 };
1137
1138Example: bind a server on a unix domain socket.
1139
1140 tcp_server "unix/", "/tmp/mydir/mysocket", sub {
1141 my ($fh) = @_;
857 }; 1142 };
858 1143
859=cut 1144=cut
860 1145
861sub tcp_server($$$;$) { 1146sub tcp_server($$$;$) {
893 } 1178 }
894 1179
895 bind $state{fh}, pack_sockaddr $service, $ipn 1180 bind $state{fh}, pack_sockaddr $service, $ipn
896 or Carp::croak "bind: $!"; 1181 or Carp::croak "bind: $!";
897 1182
1183 if ($af == AF_UNIX) {
1184 my $fh = $state{fh};
1185 my $ino = (stat $fh)[1];
1186 $state{unlink} = guard {
1187 # this is racy, but is not designed to be foolproof, just best-effort
1188 unlink $service
1189 if $ino == (stat $fh)[1];
1190 };
1191 }
1192
898 fh_nonblocking $state{fh}, 1; 1193 fh_nonblocking $state{fh}, 1;
899 1194
900 my $len; 1195 my $len;
901 1196
902 if ($prepare) { 1197 if ($prepare) {
907 $len ||= 128; 1202 $len ||= 128;
908 1203
909 listen $state{fh}, $len 1204 listen $state{fh}, $len
910 or Carp::croak "listen: $!"; 1205 or Carp::croak "listen: $!";
911 1206
912 $state{aw} = AnyEvent->io (fh => $state{fh}, poll => 'r', cb => sub { 1207 $state{aw} = AE::io $state{fh}, 0, sub {
913 # this closure keeps $state alive 1208 # this closure keeps $state alive
914 while (my $peer = accept my $fh, $state{fh}) { 1209 while ($state{fh} && (my $peer = accept my $fh, $state{fh})) {
915 fh_nonblocking $fh, 1; # POSIX requires inheritance, the outside world does not 1210 fh_nonblocking $fh, 1; # POSIX requires inheritance, the outside world does not
916 1211
917 my ($service, $host) = unpack_sockaddr $peer; 1212 my ($service, $host) = unpack_sockaddr $peer;
918 $accept->($fh, format_address $host, $service); 1213 $accept->($fh, format_address $host, $service);
919 } 1214 }
920 }); 1215 };
921 1216
922 defined wantarray 1217 defined wantarray
923 ? guard { %state = () } # clear fh and watcher, which breaks the circular dependency 1218 ? guard { %state = () } # clear fh and watcher, which breaks the circular dependency
924 : () 1219 : ()
925} 1220}
926 1221
9271; 1222=item tcp_nodelay $fh, $enable
1223
1224Enables (or disables) the C<TCP_NODELAY> socket option (also known as
1225Nagle's algorithm). Returns false on error, true otherwise.
1226
1227=cut
1228
1229sub tcp_nodelay($$) {
1230 my $onoff = int ! ! $_[1];
1231
1232 setsockopt $_[0], Socket::IPPROTO_TCP (), Socket::TCP_NODELAY (), $onoff
1233}
1234
1235=item tcp_congestion $fh, $algorithm
1236
1237Sets the tcp congestion avoidance algorithm (via the C<TCP_CONGESTION>
1238socket option). The default is OS-specific, but is usually
1239C<reno>. Typical other available choices include C<cubic>, C<lp>, C<bic>,
1240C<highspeed>, C<htcp>, C<hybla>, C<illinois>, C<scalable>, C<vegas>,
1241C<veno>, C<westwood> and C<yeah>.
1242
1243=cut
1244
1245sub tcp_congestion($$) {
1246 defined TCP_CONGESTION
1247 ? setsockopt $_[0], Socket::IPPROTO_TCP (), TCP_CONGESTION, "$_[1]"
1248 : undef
1249}
928 1250
929=back 1251=back
930 1252
931=head1 SECURITY CONSIDERATIONS 1253=head1 SECURITY CONSIDERATIONS
932 1254
938harmful in general. 1260harmful in general.
939 1261
940=head1 AUTHOR 1262=head1 AUTHOR
941 1263
942 Marc Lehmann <schmorp@schmorp.de> 1264 Marc Lehmann <schmorp@schmorp.de>
943 http://home.schmorp.de/ 1265 http://anyevent.schmorp.de
944 1266
945=cut 1267=cut
946 1268
12691
1270

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines