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.137 by root, Thu Aug 25 02:10:59 2011 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
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 the
180following formats, where C<port> can be a numerical port number of a 241following formats, where C<port> can be a numerical port number of a
181service name, or a C<name=port> string, and the C< port> and C<:port> 242service name, or a C<name=port> string, and the C< port> and C<:port>
182parts are optional. Also, everywhere where an IP address is supported 243parts are optional. Also, everywhere where an IP address is supported
183a hostname or unix domain socket address is also supported (see 244a 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
283
284 # shortcut for /path
285 return ("unix/", $_)
286 if m%^/%;
216 287
217 # parse host, special cases: "ipv6" or "ipv6 port" 288 # parse host, special cases: "ipv6" or "ipv6 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
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) = @_;
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;
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;
467# gets lost fairly often on windows), so we have to hardcode some common 650our $HOSTS;
468# protocol numbers ourselves.
469our %PROTO_BYNAME;
470 651
471$PROTO_BYNAME{tcp} = &Socket::IPPROTO_TCP if defined &Socket::IPPROTO_TCP; 652if (
472$PROTO_BYNAME{udp} = &Socket::IPPROTO_UDP if defined &Socket::IPPROTO_UDP; 653 open my $fh, "<",
473$PROTO_BYNAME{icmp} = &Socket::IPPROTO_ICMP if defined &Socket::IPPROTO_ICMP; 654 length $ENV{PERL_ANYEVENT_HOSTS} ? $ENV{PERL_ANYEVENT_HOSTS}
655 : AnyEvent::WIN32 ? "$ENV{SystemRoot}/system32/drivers/etc/hosts"
656 : "/etc/hosts"
657) {
658 local $/;
659 binmode $fh;
660 $HOSTS = <$fh>;
661} else {
662 $HOSTS = "";
663}
664
665sub _parse_hosts() {
666 #%HOSTS = ();
667
668 for (split /\n/, $HOSTS) {
669 s/#.*$//;
670 s/^[ \t]+//;
671 y/A-Z/a-z/;
672
673 my ($addr, @aliases) = split /[ \t]+/;
674 next unless @aliases;
675
676 if (my $ip = parse_ipv4 $addr) {
677 push @{ $HOSTS{$_}[0] }, $ip
678 for @aliases;
679 } elsif (my $ip = parse_ipv6 $addr) {
680 push @{ $HOSTS{$_}[1] }, $ip
681 for @aliases;
682 }
683 }
684
685 undef $HOSTS;
686}
474 687
475sub resolve_sockaddr($$$$$$) { 688sub resolve_sockaddr($$$$$$) {
476 my ($node, $service, $proto, $family, $type, $cb) = @_; 689 my ($node, $service, $proto, $family, $type, $cb) = @_;
477 690
478 if ($node eq "unix/") { 691 if ($node eq "unix/") {
479 return $cb->() if $family || !/^\//; # no can do 692 return $cb->() if $family || $service !~ /^\//; # no can do
480 693
481 return $cb->([AF_UNIX, $type, 0, Socket::pack_sockaddr_un $service]); 694 return $cb->([AF_UNIX, defined $type ? $type : SOCK_STREAM, 0, Socket::pack_sockaddr_un $service]);
482 } 695 }
483 696
484 unless (AF_INET6) { 697 unless (AF_INET6) {
485 $family != 6 698 $family != 6
486 or return $cb->(); 699 or return $cb->();
495 $family ||= 6 unless $AnyEvent::PROTOCOL{ipv4}; 708 $family ||= 6 unless $AnyEvent::PROTOCOL{ipv4};
496 709
497 $proto ||= "tcp"; 710 $proto ||= "tcp";
498 $type ||= $proto eq "udp" ? SOCK_DGRAM : SOCK_STREAM; 711 $type ||= $proto eq "udp" ? SOCK_DGRAM : SOCK_STREAM;
499 712
500 my $proton = $PROTO_BYNAME{lc $proto} || (getprotobyname $proto)[2] 713 my $proton = AnyEvent::Socket::getprotobyname $proto
501 or Carp::croak "$proto: protocol unknown"; 714 or Carp::croak "$proto: protocol unknown";
502 715
503 my $port; 716 my $port;
504 717
505 if ($service =~ /^(\S+)=(\d+)$/) { 718 if ($service =~ /^(\S+)=(\d+)$/) {
509 } else { 722 } else {
510 $port = (getservbyname $service, $proto)[2] 723 $port = (getservbyname $service, $proto)[2]
511 or Carp::croak "$service/$proto: service unknown"; 724 or Carp::croak "$service/$proto: service unknown";
512 } 725 }
513 726
514 my @target = [$node, $port];
515
516 # resolve a records / provide sockaddr structures 727 # resolve a records / provide sockaddr structures
517 my $resolve = sub { 728 my $resolve = sub {
729 my @target = @_;
730
518 my @res; 731 my @res;
519 my $cv = AnyEvent->condvar (cb => sub { 732 my $cv = AE::cv {
520 $cb->( 733 $cb->(
521 map $_->[2], 734 map $_->[2],
522 sort { 735 sort {
523 $AnyEvent::PROTOCOL{$b->[1]} <=> $AnyEvent::PROTOCOL{$a->[1]} 736 $AnyEvent::PROTOCOL{$b->[1]} <=> $AnyEvent::PROTOCOL{$a->[1]}
524 or $a->[0] <=> $b->[0] 737 or $a->[0] <=> $b->[0]
525 } 738 }
526 @res 739 @res
527 ) 740 )
528 }); 741 };
529 742
530 $cv->begin; 743 $cv->begin;
531 for my $idx (0 .. $#target) { 744 for my $idx (0 .. $#target) {
532 my ($node, $port) = @{ $target[$idx] }; 745 my ($node, $port) = @{ $target[$idx] };
533 746
542 if ($af == AF_INET6 && $family != 4) { 755 if ($af == AF_INET6 && $family != 4) {
543 push @res, [$idx, "ipv6", [AF_INET6, $type, $proton, 756 push @res, [$idx, "ipv6", [AF_INET6, $type, $proton,
544 pack_sockaddr $port, $noden]] 757 pack_sockaddr $port, $noden]]
545 } 758 }
546 } else { 759 } else {
547 # ipv4 760 $node =~ y/A-Z/a-z/;
761
762 my $hosts = $HOSTS{$node};
763
764 # a records
548 if ($family != 6) { 765 if ($family != 6) {
549 $cv->begin; 766 $cv->begin;
550 AnyEvent::DNS::a $node, sub { 767 AnyEvent::DNS::a $node, sub {
551 push @res, [$idx, "ipv4", [AF_INET, $type, $proton, 768 push @res, [$idx, "ipv4", [AF_INET , $type, $proton, pack_sockaddr $port, parse_ipv4 $_]]
552 pack_sockaddr $port, parse_ipv4 $_]]
553 for @_; 769 for @_;
770
771 # dns takes precedence over hosts
772 push @res,
773 map [$idx, "ipv4", [AF_INET , $type, $proton, pack_sockaddr $port, $_]],
774 @{ $hosts->[0] }
775 unless @_;
776
554 $cv->end; 777 $cv->end;
555 }; 778 };
556 } 779 }
557 780
558 # ipv6 781 # aaaa records
559 if ($family != 4) { 782 if ($family != 4) {
560 $cv->begin; 783 $cv->begin;
561 AnyEvent::DNS::aaaa $node, sub { 784 AnyEvent::DNS::aaaa $node, sub {
562 push @res, [$idx, "ipv6", [AF_INET6, $type, $proton, 785 push @res, [$idx, "ipv6", [AF_INET6, $type, $proton, pack_sockaddr $port, parse_ipv6 $_]]
563 pack_sockaddr $port, parse_ipv6 $_]]
564 for @_; 786 for @_;
787
788 push @res,
789 map [$idx + 0.5, "ipv6", [AF_INET6, $type, $proton, pack_sockaddr $port, $_]],
790 @{ $hosts->[1] }
791 unless @_;
792
565 $cv->end; 793 $cv->end;
566 }; 794 };
567 } 795 }
568 } 796 }
569 } 797 }
570 $cv->end; 798 $cv->end;
571 }; 799 };
572 800
801 $node = AnyEvent::Util::idn_to_ascii $node
802 if $node =~ /[^\x00-\x7f]/;
803
804 # parse hosts
805 if (defined $HOSTS) {
806 _parse_hosts;
807 undef &_parse_hosts;
808 }
809
573 # try srv records, if applicable 810 # try srv records, if applicable
574 if ($node eq "localhost") { 811 if ($node eq "localhost") {
575 @target = (["127.0.0.1", $port], ["::1", $port]); 812 $resolve->(["127.0.0.1", $port], ["::1", $port]);
576 &$resolve;
577 } elsif (defined $service && !parse_address $node) { 813 } elsif (defined $service && !parse_address $node) {
578 AnyEvent::DNS::srv $service, $proto, $node, sub { 814 AnyEvent::DNS::srv $service, $proto, $node, sub {
579 my (@srv) = @_; 815 my (@srv) = @_;
580 816
581 # no srv records, continue traditionally
582 @srv 817 if (@srv) {
583 or return &$resolve;
584
585 # the only srv record has "." ("" here) => abort 818 # the only srv record has "." ("" here) => abort
586 $srv[0][2] ne "" || $#srv 819 $srv[0][2] ne "" || $#srv
587 or return $cb->(); 820 or return $cb->();
588 821
589 # use srv records then 822 # use srv records then
823 $resolve->(
590 @target = map ["$_->[3].", $_->[2]], 824 map ["$_->[3].", $_->[2]],
591 grep $_->[3] ne ".", 825 grep $_->[3] ne ".",
592 @srv; 826 @srv
593 827 );
594 &$resolve; 828 } else {
829 # no srv records, continue traditionally
830 $resolve->([$node, $port]);
831 }
595 }; 832 };
596 } else { 833 } else {
597 &$resolve; 834 # most common case
835 $resolve->([$node, $port]);
598 } 836 }
599} 837}
600 838
601=item $guard = tcp_connect $host, $service, $connect_cb[, $prepare_cb] 839=item $guard = tcp_connect $host, $service, $connect_cb[, $prepare_cb]
602 840
603This is a convenience function that creates a TCP socket and makes a 100% 841This 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 842100% 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) 843hostname 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, 844sockets) 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 845name, or a C<servicename=portnumber> string, or the pathname to a UNIX
608socket). 846domain socket).
609 847
610If both C<$host> and C<$port> are names, then this function will use SRV 848If both C<$host> and C<$port> are names, then this function will use SRV
611records to locate the real target(s). 849records to locate the real target(s).
612 850
613In either case, it will create a list of target hosts (e.g. for multihomed 851In 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 852hosts or hosts with both IPv4 and IPv6 addresses) and try to connect to
615each in turn. 853each in turn.
616 854
617If the connect is successful, then the C<$connect_cb> will be invoked with 855After 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 856invoked 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, 857the 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 858arguments, respectively. The fourth argument is a code reference that you
621if, for some reason, you don't like this connection, which will cause 859can 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 860cause 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 861any arguments if there are no more connections). In most cases, you can
624ignore this argument. 862simply ignore this argument.
625 863
626 $cb->($filehandle, $host, $port, $retry) 864 $cb->($filehandle, $host, $port, $retry)
627 865
628If the connect is unsuccessful, then the C<$connect_cb> will be invoked 866If the connect is unsuccessful, then the C<$connect_cb> will be invoked
629without any arguments and C<$!> will be set appropriately (with C<ENXIO> 867without any arguments and C<$!> will be set appropriately (with C<ENXIO>
630indicating a DNS resolution failure). 868indicating a DNS resolution failure).
631 869
870The callback will I<never> be invoked before C<tcp_connect> returns, even
871if C<tcp_connect> was able to connect immediately (e.g. on unix domain
872sockets).
873
632The file handle is perfect for being plugged into L<AnyEvent::Handle>, but 874The file handle is perfect for being plugged into L<AnyEvent::Handle>, but
633can be used as a normal perl file handle as well. 875can be used as a normal perl file handle as well.
634 876
635Unless called in void context, C<tcp_connect> returns a guard object that 877Unless called in void context, C<tcp_connect> returns a guard object that
636will automatically abort connecting when it gets destroyed (it does not do 878will automatically cancel the connection attempt when it gets destroyed
879- in which case the callback will not be invoked. Destroying it does not
637anything to the socket after the connect was successful). 880do anything to the socket after the connect was successful - you cannot
881"uncall" a callback that has been invoked already.
638 882
639Sometimes you need to "prepare" the socket before connecting, for example, 883Sometimes 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 884to 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 885is 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 886a second callback, C<$prepare_cb>. It will be called with the file handle
674 or die "unable to connect: $!"; 918 or die "unable to connect: $!";
675 919
676 my $handle; # avoid direct assignment so on_eof has it in scope. 920 my $handle; # avoid direct assignment so on_eof has it in scope.
677 $handle = new AnyEvent::Handle 921 $handle = new AnyEvent::Handle
678 fh => $fh, 922 fh => $fh,
923 on_error => sub {
924 warn "error $_[2]\n";
925 $_[0]->destroy;
926 },
679 on_eof => sub { 927 on_eof => sub {
680 undef $handle; # keep it alive till eof 928 $handle->destroy; # destroy handle
681 warn "done.\n"; 929 warn "done.\n";
682 }; 930 };
683 931
684 $handle->push_write ("GET / HTTP/1.0\015\012\015\012"); 932 $handle->push_write ("GET / HTTP/1.0\015\012\015\012");
685 933
686 $handle->push_read_line ("\015\012\015\012", sub { 934 $handle->push_read (line => "\015\012\015\012", sub {
687 my ($handle, $line) = @_; 935 my ($handle, $line) = @_;
688 936
689 # print response header 937 # print response header
690 print "HEADER\n$line\n\nBODY\n"; 938 print "HEADER\n$line\n\nBODY\n";
691 939
711=cut 959=cut
712 960
713sub tcp_connect($$$;$) { 961sub tcp_connect($$$;$) {
714 my ($host, $port, $connect, $prepare) = @_; 962 my ($host, $port, $connect, $prepare) = @_;
715 963
716 # see http://cr.yp.to/docs/connect.html for some background 964 # see http://cr.yp.to/docs/connect.html for some tricky aspects
717 # also http://advogato.org/article/672.html 965 # also http://advogato.org/article/672.html
718 966
719 my %state = ( fh => undef ); 967 my %state = ( fh => undef );
720 968
721 # name/service to type/sockaddr resolution 969 # name/service to type/sockaddr resolution
722 resolve_sockaddr $host, $port, 0, 0, 0, sub { 970 resolve_sockaddr $host, $port, 0, 0, undef, sub {
723 my @target = @_; 971 my @target = @_;
724 972
725 $state{next} = sub { 973 $state{next} = sub {
726 return unless exists $state{fh}; 974 return unless exists $state{fh};
727 975
728 my $target = shift @target 976 my $target = shift @target
729 or do { 977 or return AE::postpone {
978 return unless exists $state{fh};
730 %state = (); 979 %state = ();
731 return $connect->(); 980 $connect->();
732 }; 981 };
733 982
734 my ($domain, $type, $proto, $sockaddr) = @$target; 983 my ($domain, $type, $proto, $sockaddr) = @$target;
735 984
736 # socket creation 985 # socket creation
741 990
742 my $timeout = $prepare && $prepare->($state{fh}); 991 my $timeout = $prepare && $prepare->($state{fh});
743 992
744 $timeout ||= 30 if AnyEvent::WIN32; 993 $timeout ||= 30 if AnyEvent::WIN32;
745 994
746 $state{to} = AnyEvent->timer (after => $timeout, cb => sub { 995 $state{to} = AE::timer $timeout, 0, sub {
747 $! = &Errno::ETIMEDOUT; 996 $! = Errno::ETIMEDOUT;
748 $state{next}(); 997 $state{next}();
749 }) if $timeout; 998 } if $timeout;
750 999
751 # called when the connect was successful, which, 1000 # now connect
752 # in theory, could be the case immediately (but never is in practise) 1001 if (
753 my $connected = sub { 1002 (connect $state{fh}, $sockaddr)
754 delete $state{ww}; 1003 || ($! == Errno::EINPROGRESS # POSIX
755 delete $state{to}; 1004 || $! == Errno::EWOULDBLOCK
756 1005 # WSAEINPROGRESS intentionally not checked - it means something else entirely
1006 || $! == AnyEvent::Util::WSAEINVAL # not convinced, but doesn't hurt
1007 || $! == AnyEvent::Util::WSAEWOULDBLOCK)
1008 ) {
1009 $state{ww} = AE::io $state{fh}, 1, sub {
757 # we are connected, or maybe there was an error 1010 # we are connected, or maybe there was an error
758 if (my $sin = getpeername $state{fh}) { 1011 if (my $sin = getpeername $state{fh}) {
759 my ($port, $host) = unpack_sockaddr $sin; 1012 my ($port, $host) = unpack_sockaddr $sin;
760 1013
1014 delete $state{ww}; delete $state{to};
1015
761 my $guard = guard { 1016 my $guard = guard { %state = () };
762 %state = ();
763 };
764 1017
765 $connect->($state{fh}, format_address $host, $port, sub { 1018 $connect->(delete $state{fh}, format_address $host, $port, sub {
766 $guard->cancel; 1019 $guard->cancel;
1020 $state{next}();
1021 });
1022 } else {
1023 if ($! == Errno::ENOTCONN) {
1024 # dummy read to fetch real error code if !cygwin
1025 sysread $state{fh}, my $buf, 1;
1026
1027 # cygwin 1.5 continously reports "ready' but never delivers
1028 # an error with getpeername or sysread.
1029 # cygwin 1.7 only reports readyness *once*, but is otherwise
1030 # the same, which is actually more broken.
1031 # Work around both by using unportable SO_ERROR for cygwin.
1032 $! = (unpack "l", getsockopt $state{fh}, Socket::SOL_SOCKET(), Socket::SO_ERROR()) || Errno::EAGAIN
1033 if AnyEvent::CYGWIN && $! == Errno::EAGAIN;
1034 }
1035
1036 return if $! == Errno::EAGAIN; # skip spurious wake-ups
1037
1038 delete $state{ww}; delete $state{to};
1039
767 $state{next}(); 1040 $state{next}();
768 }); 1041 }
769 } else {
770 # dummy read to fetch real error code
771 sysread $state{fh}, my $buf, 1 if $! == &Errno::ENOTCONN;
772 $state{next}();
773 } 1042 };
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 { 1043 } else {
786 $state{next}(); 1044 $state{next}();
787 } 1045 }
788 }; 1046 };
789 1047
790 $! = &Errno::ENXIO; 1048 $! = Errno::ENXIO;
791 $state{next}(); 1049 $state{next}();
792 }; 1050 };
793 1051
794 defined wantarray && guard { %state = () } 1052 defined wantarray && guard { %state = () }
795} 1053}
812a numeric port number (or C<0> or C<undef>, in which case an ephemeral 1070a numeric port number (or C<0> or C<undef>, in which case an ephemeral
813port will be used). 1071port will be used).
814 1072
815For UNIX domain sockets, C<$host> must be C<unix/> and C<$service> must be 1073For 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> 1074the absolute pathname of the socket. This function will try to C<unlink>
817the socket before it tries to bind to it. See SECURITY CONSIDERATIONS, 1075the socket before it tries to bind to it, and will try to unlink it after
818below. 1076it stops using it. See SECURITY CONSIDERATIONS, below.
819 1077
820For each new connection that could be C<accept>ed, call the C<< 1078For 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 1079$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 1080mode) as first, and the peer host and port as second and third arguments
823(see C<tcp_connect> for details). 1081(see C<tcp_connect> for details).
824 1082
825Croaks on any errors it can detect before the listen. 1083Croaks on any errors it can detect before the listen.
826 1084
827If called in non-void context, then this function returns a guard object 1085If 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, 1086whose lifetime it tied to the TCP server: If the object gets destroyed,
829the server will be stopped (but existing accepted connections will 1087the server will be stopped (but existing accepted connections will
830continue). 1088not be affected).
831 1089
832If you need more control over the listening socket, you can provide a 1090If you need more control over the listening socket, you can provide a
833C<< $prepare_cb->($fh, $host, $port) >>, which is called just before the 1091C<< $prepare_cb->($fh, $host, $port) >>, which is called just before the
834C<listen ()> call, with the listen file handle as first argument, and IP 1092C<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 1093address and port number of the local socket endpoint as second and third
854 }, sub { 1112 }, sub {
855 my ($fh, $thishost, $thisport) = @_; 1113 my ($fh, $thishost, $thisport) = @_;
856 warn "bound to $thishost, port $thisport\n"; 1114 warn "bound to $thishost, port $thisport\n";
857 }; 1115 };
858 1116
1117Example: bind a server on a unix domain socket.
1118
1119 tcp_server "unix/", "/tmp/mydir/mysocket", sub {
1120 my ($fh) = @_;
1121 };
1122
859=cut 1123=cut
860 1124
861sub tcp_server($$$;$) { 1125sub tcp_server($$$;$) {
862 my ($host, $service, $accept, $prepare) = @_; 1126 my ($host, $service, $accept, $prepare) = @_;
863 1127
893 } 1157 }
894 1158
895 bind $state{fh}, pack_sockaddr $service, $ipn 1159 bind $state{fh}, pack_sockaddr $service, $ipn
896 or Carp::croak "bind: $!"; 1160 or Carp::croak "bind: $!";
897 1161
1162 if ($af == AF_UNIX) {
1163 my $fh = $state{fh};
1164 my $ino = (stat $fh)[1];
1165 $state{unlink} = guard {
1166 # this is racy, but is not designed to be foolproof, just best-effort
1167 unlink $service
1168 if $ino == (stat $fh)[1];
1169 };
1170 }
1171
898 fh_nonblocking $state{fh}, 1; 1172 fh_nonblocking $state{fh}, 1;
899 1173
900 my $len; 1174 my $len;
901 1175
902 if ($prepare) { 1176 if ($prepare) {
907 $len ||= 128; 1181 $len ||= 128;
908 1182
909 listen $state{fh}, $len 1183 listen $state{fh}, $len
910 or Carp::croak "listen: $!"; 1184 or Carp::croak "listen: $!";
911 1185
912 $state{aw} = AnyEvent->io (fh => $state{fh}, poll => 'r', cb => sub { 1186 $state{aw} = AE::io $state{fh}, 0, sub {
913 # this closure keeps $state alive 1187 # this closure keeps $state alive
914 while (my $peer = accept my $fh, $state{fh}) { 1188 while ($state{fh} && (my $peer = accept my $fh, $state{fh})) {
915 fh_nonblocking $fh, 1; # POSIX requires inheritance, the outside world does not 1189 fh_nonblocking $fh, 1; # POSIX requires inheritance, the outside world does not
916 1190
917 my ($service, $host) = unpack_sockaddr $peer; 1191 my ($service, $host) = unpack_sockaddr $peer;
918 $accept->($fh, format_address $host, $service); 1192 $accept->($fh, format_address $host, $service);
919 } 1193 }
920 }); 1194 };
921 1195
922 defined wantarray 1196 defined wantarray
923 ? guard { %state = () } # clear fh and watcher, which breaks the circular dependency 1197 ? guard { %state = () } # clear fh and watcher, which breaks the circular dependency
924 : () 1198 : ()
1199}
1200
1201=item tcp_nodelay $fh, $enable
1202
1203Enables (or disables) the C<TCP_NODELAY> socket option (also known as
1204Nagle's algorithm). Returns false on error, true otherwise.
1205
1206=cut
1207
1208sub tcp_nodelay($$) {
1209 my $onoff = int ! ! $_[1];
1210
1211 setsockopt $_[0], Socket::IPPROTO_TCP (), Socket::TCP_NODELAY (), $onoff
1212}
1213
1214=item tcp_congestion $fh, $algorithm
1215
1216Sets the tcp congestion avoidance algorithm (via the C<TCP_CONGESTION>
1217socket option). The default is OS-specific, but is usually
1218C<reno>. Typical other available choices include C<cubic>, C<lp>, C<bic>,
1219C<highspeed>, C<htcp>, C<hybla>, C<illinois>, C<scalable>, C<vegas>,
1220C<veno>, C<westwood> and C<yeah>.
1221
1222=cut
1223
1224sub tcp_congestion($$) {
1225 defined TCP_CONGESTION
1226 ? setsockopt $_[0], Socket::IPPROTO_TCP (), TCP_CONGESTION, "$_[1]"
1227 : undef
925} 1228}
926 1229
9271; 12301;
928 1231
929=back 1232=back

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines