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

Comparing AnyEvent/lib/AnyEvent/Handle.pm (file contents):
Revision 1.96 by root, Thu Oct 2 08:10:27 2008 UTC vs.
Revision 1.142 by root, Mon Jul 6 20:24:47 2009 UTC

14 14
15AnyEvent::Handle - non-blocking I/O on file handles via AnyEvent 15AnyEvent::Handle - non-blocking I/O on file handles via AnyEvent
16 16
17=cut 17=cut
18 18
19our $VERSION = 4.3; 19our $VERSION = 4.452;
20 20
21=head1 SYNOPSIS 21=head1 SYNOPSIS
22 22
23 use AnyEvent; 23 use AnyEvent;
24 use AnyEvent::Handle; 24 use AnyEvent::Handle;
27 27
28 my $handle = 28 my $handle =
29 AnyEvent::Handle->new ( 29 AnyEvent::Handle->new (
30 fh => \*STDIN, 30 fh => \*STDIN,
31 on_eof => sub { 31 on_eof => sub {
32 $cv->broadcast; 32 $cv->send;
33 }, 33 },
34 ); 34 );
35 35
36 # send some request line 36 # send some request line
37 $handle->push_write ("getinfo\015\012"); 37 $handle->push_write ("getinfo\015\012");
59treatment of characters applies to this module as well. 59treatment of characters applies to this module as well.
60 60
61All callbacks will be invoked with the handle object as their first 61All callbacks will be invoked with the handle object as their first
62argument. 62argument.
63 63
64=head2 SIGPIPE is not handled by this module
65
66SIGPIPE is not handled by this module, so one of the practical
67requirements of using it is to ignore SIGPIPE (C<$SIG{PIPE} =
68'IGNORE'>). At least, this is highly recommend in a networked program: If
69you use AnyEvent::Handle in a filter program (like sort), exiting on
70SIGPIPE is probably the right thing to do.
71
72=head1 METHODS 64=head1 METHODS
73 65
74=over 4 66=over 4
75 67
76=item B<new (%args)> 68=item $handle = B<new> AnyEvent::TLS fh => $filehandle, key => value...
77 69
78The constructor supports these arguments (all as key => value pairs). 70The constructor supports these arguments (all as C<< key => value >> pairs).
79 71
80=over 4 72=over 4
81 73
82=item fh => $filehandle [MANDATORY] 74=item fh => $filehandle [MANDATORY]
83 75
92Set the callback to be called when an end-of-file condition is detected, 84Set the callback to be called when an end-of-file condition is detected,
93i.e. in the case of a socket, when the other side has closed the 85i.e. in the case of a socket, when the other side has closed the
94connection cleanly. 86connection cleanly.
95 87
96For sockets, this just means that the other side has stopped sending data, 88For sockets, this just means that the other side has stopped sending data,
97you can still try to write data, and, in fact, one can return from the eof 89you can still try to write data, and, in fact, one can return from the EOF
98callback and continue writing data, as only the read part has been shut 90callback and continue writing data, as only the read part has been shut
99down. 91down.
100 92
101While not mandatory, it is I<highly> recommended to set an eof callback, 93While not mandatory, it is I<highly> recommended to set an EOF callback,
102otherwise you might end up with a closed socket while you are still 94otherwise you might end up with a closed socket while you are still
103waiting for data. 95waiting for data.
104 96
105If an EOF condition has been detected but no C<on_eof> callback has been 97If an EOF condition has been detected but no C<on_eof> callback has been
106set, then a fatal error will be raised with C<$!> set to <0>. 98set, then a fatal error will be raised with C<$!> set to <0>.
107 99
108=item on_error => $cb->($handle, $fatal) 100=item on_error => $cb->($handle, $fatal, $message)
109 101
110This is the error callback, which is called when, well, some error 102This is the error callback, which is called when, well, some error
111occured, such as not being able to resolve the hostname, failure to 103occured, such as not being able to resolve the hostname, failure to
112connect or a read error. 104connect or a read error.
113 105
115fatal errors the handle object will be shut down and will not be usable 107fatal errors the handle object will be shut down and will not be usable
116(but you are free to look at the current C<< ->rbuf >>). Examples of fatal 108(but you are free to look at the current C<< ->rbuf >>). Examples of fatal
117errors are an EOF condition with active (but unsatisifable) read watchers 109errors are an EOF condition with active (but unsatisifable) read watchers
118(C<EPIPE>) or I/O errors. 110(C<EPIPE>) or I/O errors.
119 111
112AnyEvent::Handle tries to find an appropriate error code for you to check
113against, but in some cases (TLS errors), this does not work well. It is
114recommended to always output the C<$message> argument in human-readable
115error messages (it's usually the same as C<"$!">).
116
120Non-fatal errors can be retried by simply returning, but it is recommended 117Non-fatal errors can be retried by simply returning, but it is recommended
121to simply ignore this parameter and instead abondon the handle object 118to simply ignore this parameter and instead abondon the handle object
122when this callback is invoked. Examples of non-fatal errors are timeouts 119when this callback is invoked. Examples of non-fatal errors are timeouts
123C<ETIMEDOUT>) or badly-formatted data (C<EBADMSG>). 120C<ETIMEDOUT>) or badly-formatted data (C<EBADMSG>).
124 121
125On callback entrance, the value of C<$!> contains the operating system 122On callback entrance, the value of C<$!> contains the operating system
126error (or C<ENOSPC>, C<EPIPE>, C<ETIMEDOUT> or C<EBADMSG>). 123error code (or C<ENOSPC>, C<EPIPE>, C<ETIMEDOUT>, C<EBADMSG> or
124C<EPROTO>).
127 125
128While not mandatory, it is I<highly> recommended to set this callback, as 126While not mandatory, it is I<highly> recommended to set this callback, as
129you will not be notified of errors otherwise. The default simply calls 127you will not be notified of errors otherwise. The default simply calls
130C<croak>. 128C<croak>.
131 129
135and no read request is in the queue (unlike read queue callbacks, this 133and no read request is in the queue (unlike read queue callbacks, this
136callback will only be called when at least one octet of data is in the 134callback will only be called when at least one octet of data is in the
137read buffer). 135read buffer).
138 136
139To access (and remove data from) the read buffer, use the C<< ->rbuf >> 137To access (and remove data from) the read buffer, use the C<< ->rbuf >>
140method or access the C<$handle->{rbuf}> member directly. 138method or access the C<< $handle->{rbuf} >> member directly. Note that you
139must not enlarge or modify the read buffer, you can only remove data at
140the beginning from it.
141 141
142When an EOF condition is detected then AnyEvent::Handle will first try to 142When an EOF condition is detected then AnyEvent::Handle will first try to
143feed all the remaining data to the queued callbacks and C<on_read> before 143feed all the remaining data to the queued callbacks and C<on_read> before
144calling the C<on_eof> callback. If no progress can be made, then a fatal 144calling the C<on_eof> callback. If no progress can be made, then a fatal
145error will be raised (with C<$!> set to C<EPIPE>). 145error will be raised (with C<$!> set to C<EPIPE>).
243 243
244This will not work for partial TLS data that could not be encoded 244This will not work for partial TLS data that could not be encoded
245yet. This data will be lost. Calling the C<stoptls> method in time might 245yet. This data will be lost. Calling the C<stoptls> method in time might
246help. 246help.
247 247
248=item peername => $string
249
250A string used to identify the remote site - usually the DNS hostname
251(I<not> IDN!) used to create the connection, rarely the IP address.
252
253Apart from being useful in error messages, this string is also used in TLS
254peername verification (see C<verify_peername> in L<AnyEvent::TLS>).
255
248=item tls => "accept" | "connect" | Net::SSLeay::SSL object 256=item tls => "accept" | "connect" | Net::SSLeay::SSL object
249 257
250When this parameter is given, it enables TLS (SSL) mode, that means 258When this parameter is given, it enables TLS (SSL) mode, that means
251AnyEvent will start a TLS handshake as soon as the conenction has been 259AnyEvent will start a TLS handshake as soon as the conenction has been
252established and will transparently encrypt/decrypt data afterwards. 260established and will transparently encrypt/decrypt data afterwards.
261
262All TLS protocol errors will be signalled as C<EPROTO>, with an
263appropriate error message.
253 264
254TLS mode requires Net::SSLeay to be installed (it will be loaded 265TLS mode requires Net::SSLeay to be installed (it will be loaded
255automatically when you try to create a TLS handle): this module doesn't 266automatically when you try to create a TLS handle): this module doesn't
256have a dependency on that module, so if your module requires it, you have 267have a dependency on that module, so if your module requires it, you have
257to add the dependency yourself. 268to add the dependency yourself.
261mode. 272mode.
262 273
263You can also provide your own TLS connection object, but you have 274You can also provide your own TLS connection object, but you have
264to make sure that you call either C<Net::SSLeay::set_connect_state> 275to make sure that you call either C<Net::SSLeay::set_connect_state>
265or C<Net::SSLeay::set_accept_state> on it before you pass it to 276or C<Net::SSLeay::set_accept_state> on it before you pass it to
266AnyEvent::Handle. 277AnyEvent::Handle. Also, this module will take ownership of this connection
278object.
279
280At some future point, AnyEvent::Handle might switch to another TLS
281implementation, then the option to use your own session object will go
282away.
283
284B<IMPORTANT:> since Net::SSLeay "objects" are really only integers,
285passing in the wrong integer will lead to certain crash. This most often
286happens when one uses a stylish C<< tls => 1 >> and is surprised about the
287segmentation fault.
267 288
268See the C<< ->starttls >> method for when need to start TLS negotiation later. 289See the C<< ->starttls >> method for when need to start TLS negotiation later.
269 290
270=item tls_ctx => $ssl_ctx 291=item tls_ctx => $anyevent_tls
271 292
272Use the given C<Net::SSLeay::CTX> object to create the new TLS connection 293Use the given C<AnyEvent::TLS> object to create the new TLS connection
273(unless a connection object was specified directly). If this parameter is 294(unless a connection object was specified directly). If this parameter is
274missing, then AnyEvent::Handle will use C<AnyEvent::Handle::TLS_CTX>. 295missing, then AnyEvent::Handle will use C<AnyEvent::Handle::TLS_CTX>.
296
297Instead of an object, you can also specify a hash reference with C<< key
298=> value >> pairs. Those will be passed to L<AnyEvent::TLS> to create a
299new TLS context object.
300
301=item on_starttls => $cb->($handle, $success)
302
303This callback will be invoked when the TLS/SSL handshake has finished. If
304C<$success> is true, then the TLS handshake succeeded, otherwise it failed
305(C<on_stoptls> will not be called in this case).
306
307The session in C<< $handle->{tls} >> can still be examined in this
308callback, even when the handshake was not successful.
309
310=item on_stoptls => $cb->($handle)
311
312When a SSLv3/TLS shutdown/close notify/EOF is detected and this callback is
313set, then it will be invoked after freeing the TLS session. If it is not,
314then a TLS shutdown condition will be treated like a normal EOF condition
315on the handle.
316
317The session in C<< $handle->{tls} >> can still be examined in this
318callback.
319
320This callback will only be called on TLS shutdowns, not when the
321underlying handle signals EOF.
275 322
276=item json => JSON or JSON::XS object 323=item json => JSON or JSON::XS object
277 324
278This is the json coder object used by the C<json> read and write types. 325This is the json coder object used by the C<json> read and write types.
279 326
288 335
289=cut 336=cut
290 337
291sub new { 338sub new {
292 my $class = shift; 339 my $class = shift;
293
294 my $self = bless { @_ }, $class; 340 my $self = bless { @_ }, $class;
295 341
296 $self->{fh} or Carp::croak "mandatory argument fh is missing"; 342 $self->{fh} or Carp::croak "mandatory argument fh is missing";
297 343
298 AnyEvent::Util::fh_nonblocking $self->{fh}, 1; 344 AnyEvent::Util::fh_nonblocking $self->{fh}, 1;
345
346 $self->{_activity} = AnyEvent->now;
347 $self->_timeout;
348
349 $self->no_delay (delete $self->{no_delay}) if exists $self->{no_delay};
299 350
300 $self->starttls (delete $self->{tls}, delete $self->{tls_ctx}) 351 $self->starttls (delete $self->{tls}, delete $self->{tls_ctx})
301 if $self->{tls}; 352 if $self->{tls};
302 353
303 $self->{_activity} = AnyEvent->now;
304 $self->_timeout;
305
306 $self->on_drain (delete $self->{on_drain}) if exists $self->{on_drain}; 354 $self->on_drain (delete $self->{on_drain}) if exists $self->{on_drain};
307 $self->no_delay (delete $self->{no_delay}) if exists $self->{no_delay};
308 355
309 $self->start_read 356 $self->start_read
310 if $self->{on_read}; 357 if $self->{on_read};
311 358
312 $self 359 $self->{fh} && $self
313} 360}
314 361
315sub _shutdown { 362sub _shutdown {
316 my ($self) = @_; 363 my ($self) = @_;
317 364
318 delete $self->{_tw}; 365 delete @$self{qw(_tw _rw _ww fh wbuf on_read _queue)};
319 delete $self->{_rw}; 366 $self->{_eof} = 1; # tell starttls et. al to stop trying
320 delete $self->{_ww};
321 delete $self->{fh};
322 367
323 &_freetls; 368 &_freetls;
324
325 delete $self->{on_read};
326 delete $self->{_queue};
327} 369}
328 370
329sub _error { 371sub _error {
330 my ($self, $errno, $fatal) = @_; 372 my ($self, $errno, $fatal, $message) = @_;
331 373
332 $self->_shutdown 374 $self->_shutdown
333 if $fatal; 375 if $fatal;
334 376
335 $! = $errno; 377 $! = $errno;
378 $message ||= "$!";
336 379
337 if ($self->{on_error}) { 380 if ($self->{on_error}) {
338 $self->{on_error}($self, $fatal); 381 $self->{on_error}($self, $fatal, $message);
339 } else { 382 } elsif ($self->{fh}) {
340 Carp::croak "AnyEvent::Handle uncaught error: $!"; 383 Carp::croak "AnyEvent::Handle uncaught error: $message";
341 } 384 }
342} 385}
343 386
344=item $fh = $handle->fh 387=item $fh = $handle->fh
345 388
382} 425}
383 426
384=item $handle->autocork ($boolean) 427=item $handle->autocork ($boolean)
385 428
386Enables or disables the current autocork behaviour (see C<autocork> 429Enables or disables the current autocork behaviour (see C<autocork>
387constructor argument). 430constructor argument). Changes will only take effect on the next write.
388 431
389=cut 432=cut
433
434sub autocork {
435 $_[0]{autocork} = $_[1];
436}
390 437
391=item $handle->no_delay ($boolean) 438=item $handle->no_delay ($boolean)
392 439
393Enables or disables the C<no_delay> setting (see constructor argument of 440Enables or disables the C<no_delay> setting (see constructor argument of
394the same name for details). 441the same name for details).
400 447
401 eval { 448 eval {
402 local $SIG{__DIE__}; 449 local $SIG{__DIE__};
403 setsockopt $_[0]{fh}, &Socket::IPPROTO_TCP, &Socket::TCP_NODELAY, int $_[1]; 450 setsockopt $_[0]{fh}, &Socket::IPPROTO_TCP, &Socket::TCP_NODELAY, int $_[1];
404 }; 451 };
452}
453
454=item $handle->on_starttls ($cb)
455
456Replace the current C<on_starttls> callback (see the C<on_starttls> constructor argument).
457
458=cut
459
460sub on_starttls {
461 $_[0]{on_starttls} = $_[1];
462}
463
464=item $handle->on_stoptls ($cb)
465
466Replace the current C<on_stoptls> callback (see the C<on_stoptls> constructor argument).
467
468=cut
469
470sub on_starttls {
471 $_[0]{on_stoptls} = $_[1];
405} 472}
406 473
407############################################################################# 474#############################################################################
408 475
409=item $handle->timeout ($seconds) 476=item $handle->timeout ($seconds)
550 ->($self, @_); 617 ->($self, @_);
551 } 618 }
552 619
553 if ($self->{tls}) { 620 if ($self->{tls}) {
554 $self->{_tls_wbuf} .= $_[0]; 621 $self->{_tls_wbuf} .= $_[0];
622
555 &_dotls ($self); 623 &_dotls ($self);
556 } else { 624 } else {
557 $self->{wbuf} .= $_[0]; 625 $self->{wbuf} .= $_[0];
558 $self->_drain_wbuf; 626 $self->_drain_wbuf;
559 } 627 }
652 720
653 pack "w/a*", Storable::nfreeze ($ref) 721 pack "w/a*", Storable::nfreeze ($ref)
654}; 722};
655 723
656=back 724=back
725
726=item $handle->push_shutdown
727
728Sometimes you know you want to close the socket after writing your data
729before it was actually written. One way to do that is to replace your
730C<on_drain> handler by a callback that shuts down the socket (and set
731C<low_water_mark> to C<0>). This method is a shorthand for just that, and
732replaces the C<on_drain> callback with:
733
734 sub { shutdown $_[0]{fh}, 1 } # for push_shutdown
735
736This simply shuts down the write side and signals an EOF condition to the
737the peer.
738
739You can rely on the normal read queue and C<on_eof> handling
740afterwards. This is the cleanest way to close a connection.
741
742=cut
743
744sub push_shutdown {
745 my ($self) = @_;
746
747 delete $self->{low_water_mark};
748 $self->on_drain (sub { shutdown $_[0]{fh}, 1 });
749}
657 750
658=item AnyEvent::Handle::register_write_type type => $coderef->($handle, @args) 751=item AnyEvent::Handle::register_write_type type => $coderef->($handle, @args)
659 752
660This function (not method) lets you add your own types to C<push_write>. 753This function (not method) lets you add your own types to C<push_write>.
661Whenever the given C<type> is used, C<push_write> will invoke the code 754Whenever the given C<type> is used, C<push_write> will invoke the code
765 ) { 858 ) {
766 $self->_error (&Errno::ENOSPC, 1), return; 859 $self->_error (&Errno::ENOSPC, 1), return;
767 } 860 }
768 861
769 while () { 862 while () {
863 # we need to use a separate tls read buffer, as we must not receive data while
864 # we are draining the buffer, and this can only happen with TLS.
865 $self->{rbuf} .= delete $self->{_tls_rbuf} if exists $self->{_tls_rbuf};
866
770 my $len = length $self->{rbuf}; 867 my $len = length $self->{rbuf};
771 868
772 if (my $cb = shift @{ $self->{_queue} }) { 869 if (my $cb = shift @{ $self->{_queue} }) {
773 unless ($cb->($self)) { 870 unless ($cb->($self)) {
774 if ($self->{_eof}) { 871 if ($self->{_eof}) {
805 902
806 if ($self->{_eof}) { 903 if ($self->{_eof}) {
807 if ($self->{on_eof}) { 904 if ($self->{on_eof}) {
808 $self->{on_eof}($self) 905 $self->{on_eof}($self)
809 } else { 906 } else {
810 $self->_error (0, 1); 907 $self->_error (0, 1, "Unexpected end-of-file");
811 } 908 }
812 } 909 }
813 910
814 # may need to restart read watcher 911 # may need to restart read watcher
815 unless ($self->{_rw}) { 912 unless ($self->{_rw}) {
835 932
836=item $handle->rbuf 933=item $handle->rbuf
837 934
838Returns the read buffer (as a modifiable lvalue). 935Returns the read buffer (as a modifiable lvalue).
839 936
840You can access the read buffer directly as the C<< ->{rbuf} >> member, if 937You can access the read buffer directly as the C<< ->{rbuf} >>
841you want. 938member, if you want. However, the only operation allowed on the
939read buffer (apart from looking at it) is removing data from its
940beginning. Otherwise modifying or appending to it is not allowed and will
941lead to hard-to-track-down bugs.
842 942
843NOTE: The read buffer should only be used or modified if the C<on_read>, 943NOTE: The read buffer should only be used or modified if the C<on_read>,
844C<push_read> or C<unshift_read> methods are used. The other read methods 944C<push_read> or C<unshift_read> methods are used. The other read methods
845automatically manage the read buffer. 945automatically manage the read buffer.
846 946
1142 } 1242 }
1143}; 1243};
1144 1244
1145=item json => $cb->($handle, $hash_or_arrayref) 1245=item json => $cb->($handle, $hash_or_arrayref)
1146 1246
1147Reads a JSON object or array, decodes it and passes it to the callback. 1247Reads a JSON object or array, decodes it and passes it to the
1248callback. When a parse error occurs, an C<EBADMSG> error will be raised.
1148 1249
1149If a C<json> object was passed to the constructor, then that will be used 1250If a C<json> object was passed to the constructor, then that will be used
1150for the final decode, otherwise it will create a JSON coder expecting UTF-8. 1251for the final decode, otherwise it will create a JSON coder expecting UTF-8.
1151 1252
1152This read type uses the incremental parser available with JSON version 1253This read type uses the incremental parser available with JSON version
1161=cut 1262=cut
1162 1263
1163register_read_type json => sub { 1264register_read_type json => sub {
1164 my ($self, $cb) = @_; 1265 my ($self, $cb) = @_;
1165 1266
1166 require JSON; 1267 my $json = $self->{json} ||=
1268 eval { require JSON::XS; JSON::XS->new->utf8 }
1269 || do { require JSON; JSON->new->utf8 };
1167 1270
1168 my $data; 1271 my $data;
1169 my $rbuf = \$self->{rbuf}; 1272 my $rbuf = \$self->{rbuf};
1170 1273
1171 my $json = $self->{json} ||= JSON->new->utf8;
1172
1173 sub { 1274 sub {
1174 my $ref = $json->incr_parse ($self->{rbuf}); 1275 my $ref = eval { $json->incr_parse ($self->{rbuf}) };
1175 1276
1176 if ($ref) { 1277 if ($ref) {
1177 $self->{rbuf} = $json->incr_text; 1278 $self->{rbuf} = $json->incr_text;
1178 $json->incr_text = ""; 1279 $json->incr_text = "";
1179 $cb->($self, $ref); 1280 $cb->($self, $ref);
1180 1281
1181 1 1282 1
1283 } elsif ($@) {
1284 # error case
1285 $json->incr_skip;
1286
1287 $self->{rbuf} = $json->incr_text;
1288 $json->incr_text = "";
1289
1290 $self->_error (&Errno::EBADMSG);
1291
1292 ()
1182 } else { 1293 } else {
1183 $self->{rbuf} = ""; 1294 $self->{rbuf} = "";
1295
1184 () 1296 ()
1185 } 1297 }
1186 } 1298 }
1187}; 1299};
1188 1300
1291 if ($len > 0) { 1403 if ($len > 0) {
1292 $self->{_activity} = AnyEvent->now; 1404 $self->{_activity} = AnyEvent->now;
1293 1405
1294 if ($self->{tls}) { 1406 if ($self->{tls}) {
1295 Net::SSLeay::BIO_write ($self->{_rbio}, $$rbuf); 1407 Net::SSLeay::BIO_write ($self->{_rbio}, $$rbuf);
1408
1296 &_dotls ($self); 1409 &_dotls ($self);
1297 } else { 1410 } else {
1298 $self->_drain_rbuf unless $self->{_in_drain}; 1411 $self->_drain_rbuf unless $self->{_in_drain};
1299 } 1412 }
1300 1413
1308 } 1421 }
1309 }); 1422 });
1310 } 1423 }
1311} 1424}
1312 1425
1426our $ERROR_SYSCALL;
1427our $ERROR_WANT_READ;
1428
1429sub _tls_error {
1430 my ($self, $err) = @_;
1431
1432 return $self->_error ($!, 1)
1433 if $err == Net::SSLeay::ERROR_SYSCALL ();
1434
1435 my $err =Net::SSLeay::ERR_error_string (Net::SSLeay::ERR_get_error ());
1436
1437 # reduce error string to look less scary
1438 $err =~ s/^error:[0-9a-fA-F]{8}:[^:]+:([^:]+):/\L$1: /;
1439
1440 $self->_error (&Errno::EPROTO, 1, $err);
1441}
1442
1443# poll the write BIO and send the data if applicable
1444# also decode read data if possible
1445# this is basiclaly our TLS state machine
1446# more efficient implementations are possible with openssl,
1447# but not with the buggy and incomplete Net::SSLeay.
1313sub _dotls { 1448sub _dotls {
1314 my ($self) = @_; 1449 my ($self) = @_;
1315 1450
1316 my $buf; 1451 my $tmp;
1317 1452
1318 if (length $self->{_tls_wbuf}) { 1453 if (length $self->{_tls_wbuf}) {
1319 while ((my $len = Net::SSLeay::write ($self->{tls}, $self->{_tls_wbuf})) > 0) { 1454 while (($tmp = Net::SSLeay::write ($self->{tls}, $self->{_tls_wbuf})) > 0) {
1320 substr $self->{_tls_wbuf}, 0, $len, ""; 1455 substr $self->{_tls_wbuf}, 0, $tmp, "";
1321 } 1456 }
1322 }
1323 1457
1458 $tmp = Net::SSLeay::get_error ($self->{tls}, $tmp);
1459 return $self->_tls_error ($tmp)
1460 if $tmp != $ERROR_WANT_READ
1461 && ($tmp != $ERROR_SYSCALL || $!);
1462 }
1463
1324 while (defined ($buf = Net::SSLeay::read ($self->{tls}))) { 1464 while (defined ($tmp = Net::SSLeay::read ($self->{tls}))) {
1325 unless (length $buf) { 1465 unless (length $tmp) {
1326 # let's treat SSL-eof as we treat normal EOF
1327 delete $self->{_rw};
1328 $self->{_eof} = 1;
1329 &_freetls; 1466 &_freetls;
1467 if ($self->{on_stoptls}) {
1468 $self->{on_stoptls}($self);
1469 return;
1470 } else {
1471 # let's treat SSL-eof as we treat normal EOF
1472 delete $self->{_rw};
1473 $self->{_eof} = 1;
1474 }
1330 } 1475 }
1331 1476
1332 $self->{rbuf} .= $buf; 1477 $self->{_tls_rbuf} .= $tmp;
1333 $self->_drain_rbuf unless $self->{_in_drain}; 1478 $self->_drain_rbuf unless $self->{_in_drain};
1334 $self->{tls} or return; # tls session might have gone away in callback 1479 $self->{tls} or return; # tls session might have gone away in callback
1335 } 1480 }
1336 1481
1337 my $err = Net::SSLeay::get_error ($self->{tls}, -1); 1482 $tmp = Net::SSLeay::get_error ($self->{tls}, -1);
1338
1339 if ($err!= Net::SSLeay::ERROR_WANT_READ ()) {
1340 if ($err == Net::SSLeay::ERROR_SYSCALL ()) {
1341 return $self->_error ($!, 1); 1483 return $self->_tls_error ($tmp)
1342 } elsif ($err == Net::SSLeay::ERROR_SSL ()) { 1484 if $tmp != $ERROR_WANT_READ
1343 return $self->_error (&Errno::EIO, 1); 1485 && ($tmp != $ERROR_SYSCALL || $!);
1344 }
1345 1486
1346 # all others are fine for our purposes
1347 }
1348
1349 while (length ($buf = Net::SSLeay::BIO_read ($self->{_wbio}))) { 1487 while (length ($tmp = Net::SSLeay::BIO_read ($self->{_wbio}))) {
1350 $self->{wbuf} .= $buf; 1488 $self->{wbuf} .= $tmp;
1351 $self->_drain_wbuf; 1489 $self->_drain_wbuf;
1352 } 1490 }
1491
1492 $self->{_on_starttls}
1493 and Net::SSLeay::state ($self->{tls}) == Net::SSLeay::ST_OK ()
1494 and (delete $self->{_on_starttls})->($self, 1);
1353} 1495}
1354 1496
1355=item $handle->starttls ($tls[, $tls_ctx]) 1497=item $handle->starttls ($tls[, $tls_ctx])
1356 1498
1357Instead of starting TLS negotiation immediately when the AnyEvent::Handle 1499Instead of starting TLS negotiation immediately when the AnyEvent::Handle
1359C<starttls>. 1501C<starttls>.
1360 1502
1361The first argument is the same as the C<tls> constructor argument (either 1503The first argument is the same as the C<tls> constructor argument (either
1362C<"connect">, C<"accept"> or an existing Net::SSLeay object). 1504C<"connect">, C<"accept"> or an existing Net::SSLeay object).
1363 1505
1364The second argument is the optional C<Net::SSLeay::CTX> object that is 1506The second argument is the optional C<AnyEvent::TLS> object that is used
1365used when AnyEvent::Handle has to create its own TLS connection object. 1507when AnyEvent::Handle has to create its own TLS connection object, or
1508a hash reference with C<< key => value >> pairs that will be used to
1509construct a new context.
1366 1510
1367The TLS connection object will end up in C<< $handle->{tls} >> after this 1511The TLS connection object will end up in C<< $handle->{tls} >>, the TLS
1368call and can be used or changed to your liking. Note that the handshake 1512context in C<< $handle->{tls_ctx} >> after this call and can be used or
1369might have already started when this function returns. 1513changed to your liking. Note that the handshake might have already started
1514when this function returns.
1370 1515
1371If it an error to start a TLS handshake more than once per 1516If it an error to start a TLS handshake more than once per
1372AnyEvent::Handle object (this is due to bugs in OpenSSL). 1517AnyEvent::Handle object (this is due to bugs in OpenSSL).
1373 1518
1374=cut 1519=cut
1375 1520
1521our %TLS_CACHE; #TODO not yet documented, should we?
1522
1376sub starttls { 1523sub starttls {
1377 my ($self, $ssl, $ctx) = @_; 1524 my ($self, $ssl, $ctx) = @_;
1378 1525
1379 require Net::SSLeay; 1526 require Net::SSLeay;
1380 1527
1381 Carp::croak "it is an error to call starttls more than once on an Anyevent::Handle object" 1528 Carp::croak "it is an error to call starttls more than once on an AnyEvent::Handle object"
1382 if $self->{tls}; 1529 if $self->{tls};
1530
1531 $ERROR_SYSCALL = Net::SSLeay::ERROR_SYSCALL ();
1532 $ERROR_WANT_READ = Net::SSLeay::ERROR_WANT_READ ();
1533
1534 $ctx ||= $self->{tls_ctx};
1535
1536 if ("HASH" eq ref $ctx) {
1537 require AnyEvent::TLS;
1538
1539 local $Carp::CarpLevel = 1; # skip ourselves when creating a new context
1540
1541 if ($ctx->{cache}) {
1542 my $key = $ctx+0;
1543 $ctx = $TLS_CACHE{$key} ||= new AnyEvent::TLS %$ctx;
1544 } else {
1545 $ctx = new AnyEvent::TLS %$ctx;
1546 }
1547 }
1383 1548
1384 if ($ssl eq "accept") { 1549 $self->{tls_ctx} = $ctx || TLS_CTX ();
1385 $ssl = Net::SSLeay::new ($ctx || TLS_CTX ()); 1550 $self->{tls} = $ssl = $self->{tls_ctx}->_get_session ($ssl, $self, $self->{peername});
1386 Net::SSLeay::set_accept_state ($ssl);
1387 } elsif ($ssl eq "connect") {
1388 $ssl = Net::SSLeay::new ($ctx || TLS_CTX ());
1389 Net::SSLeay::set_connect_state ($ssl);
1390 }
1391
1392 $self->{tls} = $ssl;
1393 1551
1394 # basically, this is deep magic (because SSL_read should have the same issues) 1552 # basically, this is deep magic (because SSL_read should have the same issues)
1395 # but the openssl maintainers basically said: "trust us, it just works". 1553 # but the openssl maintainers basically said: "trust us, it just works".
1396 # (unfortunately, we have to hardcode constants because the abysmally misdesigned 1554 # (unfortunately, we have to hardcode constants because the abysmally misdesigned
1397 # and mismaintained ssleay-module doesn't even offer them). 1555 # and mismaintained ssleay-module doesn't even offer them).
1401 # 1559 #
1402 # note that we do not try to keep the length constant between writes as we are required to do. 1560 # note that we do not try to keep the length constant between writes as we are required to do.
1403 # we assume that most (but not all) of this insanity only applies to non-blocking cases, 1561 # we assume that most (but not all) of this insanity only applies to non-blocking cases,
1404 # and we drive openssl fully in blocking mode here. Or maybe we don't - openssl seems to 1562 # and we drive openssl fully in blocking mode here. Or maybe we don't - openssl seems to
1405 # have identity issues in that area. 1563 # have identity issues in that area.
1406 Net::SSLeay::CTX_set_mode ($self->{tls}, 1564# Net::SSLeay::CTX_set_mode ($ssl,
1407 (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1) 1565# (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1)
1408 | (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2)); 1566# | (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2));
1567 Net::SSLeay::CTX_set_mode ($ssl, 1|2);
1409 1568
1410 $self->{_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ()); 1569 $self->{_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
1411 $self->{_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ()); 1570 $self->{_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
1412 1571
1413 Net::SSLeay::set_bio ($ssl, $self->{_rbio}, $self->{_wbio}); 1572 Net::SSLeay::set_bio ($ssl, $self->{_rbio}, $self->{_wbio});
1573
1574 $self->{_on_starttls} = sub { $_[0]{on_starttls}(@_) }
1575 if exists $self->{on_starttls};
1414 1576
1415 &_dotls; # need to trigger the initial handshake 1577 &_dotls; # need to trigger the initial handshake
1416 $self->start_read; # make sure we actually do read 1578 $self->start_read; # make sure we actually do read
1417} 1579}
1418 1580
1431 if ($self->{tls}) { 1593 if ($self->{tls}) {
1432 Net::SSLeay::shutdown ($self->{tls}); 1594 Net::SSLeay::shutdown ($self->{tls});
1433 1595
1434 &_dotls; 1596 &_dotls;
1435 1597
1436 # we don't give a shit. no, we do, but we can't. no... 1598# # we don't give a shit. no, we do, but we can't. no...#d#
1437 # we, we... have to use openssl :/ 1599# # we, we... have to use openssl :/#d#
1438 &_freetls; 1600# &_freetls;#d#
1439 } 1601 }
1440} 1602}
1441 1603
1442sub _freetls { 1604sub _freetls {
1443 my ($self) = @_; 1605 my ($self) = @_;
1444 1606
1445 return unless $self->{tls}; 1607 return unless $self->{tls};
1446 1608
1447 Net::SSLeay::free (delete $self->{tls}); 1609 $self->{_on_starttls}
1610 and (delete $self->{_on_starttls})->($self, undef);
1611
1612 $self->{tls_ctx}->_put_session (delete $self->{tls});
1448 1613
1449 delete @$self{qw(_rbio _wbio _tls_wbuf)}; 1614 delete @$self{qw(_rbio _wbio _tls_wbuf)};
1450} 1615}
1451 1616
1452sub DESTROY { 1617sub DESTROY {
1453 my $self = shift; 1618 my ($self) = @_;
1454 1619
1455 &_freetls; 1620 &_freetls;
1456 1621
1457 my $linger = exists $self->{linger} ? $self->{linger} : 3600; 1622 my $linger = exists $self->{linger} ? $self->{linger} : 3600;
1458 1623
1475 @linger = (); 1640 @linger = ();
1476 }); 1641 });
1477 } 1642 }
1478} 1643}
1479 1644
1645=item $handle->destroy
1646
1647Shuts down the handle object as much as possible - this call ensures that
1648no further callbacks will be invoked and as many resources as possible
1649will be freed. You must not call any methods on the object afterwards.
1650
1651Normally, you can just "forget" any references to an AnyEvent::Handle
1652object and it will simply shut down. This works in fatal error and EOF
1653callbacks, as well as code outside. It does I<NOT> work in a read or write
1654callback, so when you want to destroy the AnyEvent::Handle object from
1655within such an callback. You I<MUST> call C<< ->destroy >> explicitly in
1656that case.
1657
1658The handle might still linger in the background and write out remaining
1659data, as specified by the C<linger> option, however.
1660
1661=cut
1662
1663sub destroy {
1664 my ($self) = @_;
1665
1666 $self->DESTROY;
1667 %$self = ();
1668}
1669
1480=item AnyEvent::Handle::TLS_CTX 1670=item AnyEvent::Handle::TLS_CTX
1481 1671
1482This function creates and returns the Net::SSLeay::CTX object used by 1672This function creates and returns the AnyEvent::TLS object used by default
1483default for TLS mode. 1673for TLS mode.
1484 1674
1485The context is created like this: 1675The context is created by calling L<AnyEvent::TLS> without any arguments.
1486
1487 Net::SSLeay::load_error_strings;
1488 Net::SSLeay::SSLeay_add_ssl_algorithms;
1489 Net::SSLeay::randomize;
1490
1491 my $CTX = Net::SSLeay::CTX_new;
1492
1493 Net::SSLeay::CTX_set_options $CTX, Net::SSLeay::OP_ALL
1494 1676
1495=cut 1677=cut
1496 1678
1497our $TLS_CTX; 1679our $TLS_CTX;
1498 1680
1499sub TLS_CTX() { 1681sub TLS_CTX() {
1500 $TLS_CTX || do { 1682 $TLS_CTX ||= do {
1501 require Net::SSLeay; 1683 require AnyEvent::TLS;
1502 1684
1503 Net::SSLeay::load_error_strings (); 1685 new AnyEvent::TLS
1504 Net::SSLeay::SSLeay_add_ssl_algorithms ();
1505 Net::SSLeay::randomize ();
1506
1507 $TLS_CTX = Net::SSLeay::CTX_new ();
1508
1509 Net::SSLeay::CTX_set_options ($TLS_CTX, Net::SSLeay::OP_ALL ());
1510
1511 $TLS_CTX
1512 } 1686 }
1513} 1687}
1514 1688
1515=back 1689=back
1516 1690
1517 1691
1518=head1 NONFREQUENTLY ASKED QUESTIONS 1692=head1 NONFREQUENTLY ASKED QUESTIONS
1519 1693
1520=over 4 1694=over 4
1695
1696=item I C<undef> the AnyEvent::Handle reference inside my callback and
1697still get further invocations!
1698
1699That's because AnyEvent::Handle keeps a reference to itself when handling
1700read or write callbacks.
1701
1702It is only safe to "forget" the reference inside EOF or error callbacks,
1703from within all other callbacks, you need to explicitly call the C<<
1704->destroy >> method.
1705
1706=item I get different callback invocations in TLS mode/Why can't I pause
1707reading?
1708
1709Unlike, say, TCP, TLS connections do not consist of two independent
1710communication channels, one for each direction. Or put differently. The
1711read and write directions are not independent of each other: you cannot
1712write data unless you are also prepared to read, and vice versa.
1713
1714This can mean than, in TLS mode, you might get C<on_error> or C<on_eof>
1715callback invocations when you are not expecting any read data - the reason
1716is that AnyEvent::Handle always reads in TLS mode.
1717
1718During the connection, you have to make sure that you always have a
1719non-empty read-queue, or an C<on_read> watcher. At the end of the
1720connection (or when you no longer want to use it) you can call the
1721C<destroy> method.
1521 1722
1522=item How do I read data until the other side closes the connection? 1723=item How do I read data until the other side closes the connection?
1523 1724
1524If you just want to read your data into a perl scalar, the easiest way 1725If you just want to read your data into a perl scalar, the easiest way
1525to achieve this is by setting an C<on_read> callback that does nothing, 1726to achieve this is by setting an C<on_read> callback that does nothing,
1535 1736
1536The reason to use C<on_error> is that TCP connections, due to latencies 1737The reason to use C<on_error> is that TCP connections, due to latencies
1537and packets loss, might get closed quite violently with an error, when in 1738and packets loss, might get closed quite violently with an error, when in
1538fact, all data has been received. 1739fact, all data has been received.
1539 1740
1540It is usually better to use acknowledgements when transfering data, 1741It is usually better to use acknowledgements when transferring data,
1541to make sure the other side hasn't just died and you got the data 1742to make sure the other side hasn't just died and you got the data
1542intact. This is also one reason why so many internet protocols have an 1743intact. This is also one reason why so many internet protocols have an
1543explicit QUIT command. 1744explicit QUIT command.
1544
1545 1745
1546=item I don't want to destroy the handle too early - how do I wait until 1746=item I don't want to destroy the handle too early - how do I wait until
1547all data has been written? 1747all data has been written?
1548 1748
1549After writing your last bits of data, set the C<on_drain> callback 1749After writing your last bits of data, set the C<on_drain> callback

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines