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.92 by root, Wed Oct 1 08:52:06 2008 UTC vs.
Revision 1.96 by root, Thu Oct 2 08:10:27 2008 UTC

240write data and will install a watcher that will write this data to the 240write data and will install a watcher that will write this data to the
241socket. No errors will be reported (this mostly matches how the operating 241socket. No errors will be reported (this mostly matches how the operating
242system treats outstanding data at socket close time). 242system treats outstanding data at socket close time).
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. 245yet. This data will be lost. Calling the C<stoptls> method in time might
246help.
246 247
247=item tls => "accept" | "connect" | Net::SSLeay::SSL object 248=item tls => "accept" | "connect" | Net::SSLeay::SSL object
248 249
249When this parameter is given, it enables TLS (SSL) mode, that means 250When this parameter is given, it enables TLS (SSL) mode, that means
250AnyEvent will start a TLS handshake as soon as the conenction has been 251AnyEvent will start a TLS handshake as soon as the conenction has been
281texts. 282texts.
282 283
283Note that you are responsible to depend on the JSON module if you want to 284Note that you are responsible to depend on the JSON module if you want to
284use this functionality, as AnyEvent does not have a dependency itself. 285use this functionality, as AnyEvent does not have a dependency itself.
285 286
286=item filter_r => $cb
287
288=item filter_w => $cb
289
290These exist, but are undocumented at this time. (They are used internally
291by the TLS code).
292
293=back 287=back
294 288
295=cut 289=cut
296 290
297sub new { 291sub new {
301 295
302 $self->{fh} or Carp::croak "mandatory argument fh is missing"; 296 $self->{fh} or Carp::croak "mandatory argument fh is missing";
303 297
304 AnyEvent::Util::fh_nonblocking $self->{fh}, 1; 298 AnyEvent::Util::fh_nonblocking $self->{fh}, 1;
305 299
306 if ($self->{tls}) {
307 require Net::SSLeay;
308 $self->starttls (delete $self->{tls}, delete $self->{tls_ctx}); 300 $self->starttls (delete $self->{tls}, delete $self->{tls_ctx})
309 } 301 if $self->{tls};
310 302
311 $self->{_activity} = AnyEvent->now; 303 $self->{_activity} = AnyEvent->now;
312 $self->_timeout; 304 $self->_timeout;
313 305
314 $self->on_drain (delete $self->{on_drain}) if exists $self->{on_drain}; 306 $self->on_drain (delete $self->{on_drain}) if exists $self->{on_drain};
495 my ($self, $cb) = @_; 487 my ($self, $cb) = @_;
496 488
497 $self->{on_drain} = $cb; 489 $self->{on_drain} = $cb;
498 490
499 $cb->($self) 491 $cb->($self)
500 if $cb && $self->{low_water_mark} >= length $self->{wbuf}; 492 if $cb && $self->{low_water_mark} >= (length $self->{wbuf}) + (length $self->{_tls_wbuf});
501} 493}
502 494
503=item $handle->push_write ($data) 495=item $handle->push_write ($data)
504 496
505Queues the given scalar to be written. You can push as much data as you 497Queues the given scalar to be written. You can push as much data as you
522 substr $self->{wbuf}, 0, $len, ""; 514 substr $self->{wbuf}, 0, $len, "";
523 515
524 $self->{_activity} = AnyEvent->now; 516 $self->{_activity} = AnyEvent->now;
525 517
526 $self->{on_drain}($self) 518 $self->{on_drain}($self)
527 if $self->{low_water_mark} >= length $self->{wbuf} 519 if $self->{low_water_mark} >= (length $self->{wbuf}) + (length $self->{_tls_wbuf})
528 && $self->{on_drain}; 520 && $self->{on_drain};
529 521
530 delete $self->{_ww} unless length $self->{wbuf}; 522 delete $self->{_ww} unless length $self->{wbuf};
531 } elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) { 523 } elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) {
532 $self->_error ($!, 1); 524 $self->_error ($!, 1);
556 548
557 @_ = ($WH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_write") 549 @_ = ($WH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_write")
558 ->($self, @_); 550 ->($self, @_);
559 } 551 }
560 552
561 if ($self->{filter_w}) { 553 if ($self->{tls}) {
562 $self->{filter_w}($self, \$_[0]); 554 $self->{_tls_wbuf} .= $_[0];
555 &_dotls ($self);
563 } else { 556 } else {
564 $self->{wbuf} .= $_[0]; 557 $self->{wbuf} .= $_[0];
565 $self->_drain_wbuf; 558 $self->_drain_wbuf;
566 } 559 }
567} 560}
584=cut 577=cut
585 578
586register_write_type netstring => sub { 579register_write_type netstring => sub {
587 my ($self, $string) = @_; 580 my ($self, $string) = @_;
588 581
589 sprintf "%d:%s,", (length $string), $string 582 (length $string) . ":$string,"
590}; 583};
591 584
592=item packstring => $format, $data 585=item packstring => $format, $data
593 586
594An octet string prefixed with an encoded length. The encoding C<$format> 587An octet string prefixed with an encoded length. The encoding C<$format>
803 796
804 last; # more data might arrive 797 last; # more data might arrive
805 } 798 }
806 } else { 799 } else {
807 # read side becomes idle 800 # read side becomes idle
808 delete $self->{_rw}; 801 delete $self->{_rw} unless $self->{tls};
809 last; 802 last;
810 } 803 }
811 } 804 }
812 805
813 if ($self->{_eof}) { 806 if ($self->{_eof}) {
1108An octet string prefixed with an encoded length. The encoding C<$format> 1101An octet string prefixed with an encoded length. The encoding C<$format>
1109uses the same format as a Perl C<pack> format, but must specify a single 1102uses the same format as a Perl C<pack> format, but must specify a single
1110integer only (only one of C<cCsSlLqQiInNvVjJw> is allowed, plus an 1103integer only (only one of C<cCsSlLqQiInNvVjJw> is allowed, plus an
1111optional C<!>, C<< < >> or C<< > >> modifier). 1104optional C<!>, C<< < >> or C<< > >> modifier).
1112 1105
1113DNS over TCP uses a prefix of C<n>, EPP uses a prefix of C<N>. 1106For example, DNS over TCP uses a prefix of C<n> (2 octet network order),
1107EPP uses a prefix of C<N> (4 octtes).
1114 1108
1115Example: read a block of data prefixed by its length in BER-encoded 1109Example: read a block of data prefixed by its length in BER-encoded
1116format (very efficient). 1110format (very efficient).
1117 1111
1118 $handle->push_read (packstring => "w", sub { 1112 $handle->push_read (packstring => "w", sub {
1271Note that AnyEvent::Handle will automatically C<start_read> for you when 1265Note that AnyEvent::Handle will automatically C<start_read> for you when
1272you change the C<on_read> callback or push/unshift a read callback, and it 1266you change the C<on_read> callback or push/unshift a read callback, and it
1273will automatically C<stop_read> for you when neither C<on_read> is set nor 1267will automatically C<stop_read> for you when neither C<on_read> is set nor
1274there are any read requests in the queue. 1268there are any read requests in the queue.
1275 1269
1270These methods will have no effect when in TLS mode (as TLS doesn't support
1271half-duplex connections).
1272
1276=cut 1273=cut
1277 1274
1278sub stop_read { 1275sub stop_read {
1279 my ($self) = @_; 1276 my ($self) = @_;
1280 1277
1281 delete $self->{_rw}; 1278 delete $self->{_rw} unless $self->{tls};
1282} 1279}
1283 1280
1284sub start_read { 1281sub start_read {
1285 my ($self) = @_; 1282 my ($self) = @_;
1286 1283
1287 unless ($self->{_rw} || $self->{_eof}) { 1284 unless ($self->{_rw} || $self->{_eof}) {
1288 Scalar::Util::weaken $self; 1285 Scalar::Util::weaken $self;
1289 1286
1290 $self->{_rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub { 1287 $self->{_rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub {
1291 my $rbuf = $self->{filter_r} ? \my $buf : \$self->{rbuf}; 1288 my $rbuf = \($self->{tls} ? my $buf : $self->{rbuf});
1292 my $len = sysread $self->{fh}, $$rbuf, $self->{read_size} || 8192, length $$rbuf; 1289 my $len = sysread $self->{fh}, $$rbuf, $self->{read_size} || 8192, length $$rbuf;
1293 1290
1294 if ($len > 0) { 1291 if ($len > 0) {
1295 $self->{_activity} = AnyEvent->now; 1292 $self->{_activity} = AnyEvent->now;
1296 1293
1297 $self->{filter_r} 1294 if ($self->{tls}) {
1298 ? $self->{filter_r}($self, $rbuf) 1295 Net::SSLeay::BIO_write ($self->{_rbio}, $$rbuf);
1299 : $self->{_in_drain} || $self->_drain_rbuf; 1296 &_dotls ($self);
1297 } else {
1298 $self->_drain_rbuf unless $self->{_in_drain};
1299 }
1300 1300
1301 } elsif (defined $len) { 1301 } elsif (defined $len) {
1302 delete $self->{_rw}; 1302 delete $self->{_rw};
1303 $self->{_eof} = 1; 1303 $self->{_eof} = 1;
1304 $self->_drain_rbuf unless $self->{_in_drain}; 1304 $self->_drain_rbuf unless $self->{_in_drain};
1344 } 1344 }
1345 1345
1346 # all others are fine for our purposes 1346 # all others are fine for our purposes
1347 } 1347 }
1348 1348
1349 if (length ($buf = Net::SSLeay::BIO_read ($self->{_wbio}))) { 1349 while (length ($buf = Net::SSLeay::BIO_read ($self->{_wbio}))) {
1350 $self->{wbuf} .= $buf; 1350 $self->{wbuf} .= $buf;
1351 $self->_drain_wbuf; 1351 $self->_drain_wbuf;
1352 } 1352 }
1353} 1353}
1354 1354
1373 1373
1374=cut 1374=cut
1375 1375
1376sub starttls { 1376sub starttls {
1377 my ($self, $ssl, $ctx) = @_; 1377 my ($self, $ssl, $ctx) = @_;
1378
1379 require Net::SSLeay;
1378 1380
1379 Carp::croak "it is an error to call starttls more than once on an Anyevent::Handle object" 1381 Carp::croak "it is an error to call starttls more than once on an Anyevent::Handle object"
1380 if $self->{tls}; 1382 if $self->{tls};
1381 1383
1382 if ($ssl eq "accept") { 1384 if ($ssl eq "accept") {
1395 # and mismaintained ssleay-module doesn't even offer them). 1397 # and mismaintained ssleay-module doesn't even offer them).
1396 # http://www.mail-archive.com/openssl-dev@openssl.org/msg22420.html 1398 # http://www.mail-archive.com/openssl-dev@openssl.org/msg22420.html
1397 # 1399 #
1398 # in short: this is a mess. 1400 # in short: this is a mess.
1399 # 1401 #
1400 # note that we do not try to kepe the length constant between writes as we are required to do. 1402 # note that we do not try to keep the length constant between writes as we are required to do.
1401 # we assume that most (but not all) of this insanity only applies to non-blocking cases, 1403 # we assume that most (but not all) of this insanity only applies to non-blocking cases,
1402 # and we drive openssl fully in blocking mode here. 1404 # and we drive openssl fully in blocking mode here. Or maybe we don't - openssl seems to
1405 # have identity issues in that area.
1403 Net::SSLeay::CTX_set_mode ($self->{tls}, 1406 Net::SSLeay::CTX_set_mode ($self->{tls},
1404 (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1) 1407 (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1)
1405 | (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2)); 1408 | (eval { local $SIG{__DIE__}; Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2));
1406 1409
1407 $self->{_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ()); 1410 $self->{_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
1408 $self->{_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ()); 1411 $self->{_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
1409 1412
1410 Net::SSLeay::set_bio ($ssl, $self->{_rbio}, $self->{_wbio}); 1413 Net::SSLeay::set_bio ($ssl, $self->{_rbio}, $self->{_wbio});
1411 1414
1412 $self->{filter_w} = sub {
1413 $_[0]{_tls_wbuf} .= ${$_[1]};
1414 &_dotls;
1415 };
1416 $self->{filter_r} = sub {
1417 Net::SSLeay::BIO_write ($_[0]{_rbio}, ${$_[1]});
1418 &_dotls;
1419 };
1420
1421 &_dotls; # need to trigger the initial negotiation exchange 1415 &_dotls; # need to trigger the initial handshake
1416 $self->start_read; # make sure we actually do read
1422} 1417}
1423 1418
1424=item $handle->stoptls 1419=item $handle->stoptls
1425 1420
1426Shuts down the SSL connection - this makes a proper EOF handshake by 1421Shuts down the SSL connection - this makes a proper EOF handshake by
1432 1427
1433sub stoptls { 1428sub stoptls {
1434 my ($self) = @_; 1429 my ($self) = @_;
1435 1430
1436 if ($self->{tls}) { 1431 if ($self->{tls}) {
1437 Net::SSLeay::shutdown $self->{tls}; 1432 Net::SSLeay::shutdown ($self->{tls});
1438 1433
1439 &_dotls; 1434 &_dotls;
1440 1435
1441 # we don't give a shit. no, we do, but we can't. no... 1436 # we don't give a shit. no, we do, but we can't. no...
1442 # we, we... have to use openssl :/ 1437 # we, we... have to use openssl :/
1449 1444
1450 return unless $self->{tls}; 1445 return unless $self->{tls};
1451 1446
1452 Net::SSLeay::free (delete $self->{tls}); 1447 Net::SSLeay::free (delete $self->{tls});
1453 1448
1454 delete @$self{qw(_rbio filter_w _wbio filter_r)}; 1449 delete @$self{qw(_rbio _wbio _tls_wbuf)};
1455} 1450}
1456 1451
1457sub DESTROY { 1452sub DESTROY {
1458 my $self = shift; 1453 my $self = shift;
1459 1454
1517 } 1512 }
1518} 1513}
1519 1514
1520=back 1515=back
1521 1516
1517
1518=head1 NONFREQUENTLY ASKED QUESTIONS
1519
1520=over 4
1521
1522=item How do I read data until the other side closes the connection?
1523
1524If 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,
1526clearing the C<on_eof> callback and in the C<on_error> callback, the data
1527will be in C<$_[0]{rbuf}>:
1528
1529 $handle->on_read (sub { });
1530 $handle->on_eof (undef);
1531 $handle->on_error (sub {
1532 my $data = delete $_[0]{rbuf};
1533 undef $handle;
1534 });
1535
1536The 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
1538fact, all data has been received.
1539
1540It is usually better to use acknowledgements when transfering data,
1541to 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
1543explicit QUIT command.
1544
1545
1546=item I don't want to destroy the handle too early - how do I wait until
1547all data has been written?
1548
1549After writing your last bits of data, set the C<on_drain> callback
1550and destroy the handle in there - with the default setting of
1551C<low_water_mark> this will be called precisely when all data has been
1552written to the socket:
1553
1554 $handle->push_write (...);
1555 $handle->on_drain (sub {
1556 warn "all data submitted to the kernel\n";
1557 undef $handle;
1558 });
1559
1560=back
1561
1562
1522=head1 SUBCLASSING AnyEvent::Handle 1563=head1 SUBCLASSING AnyEvent::Handle
1523 1564
1524In many cases, you might want to subclass AnyEvent::Handle. 1565In many cases, you might want to subclass AnyEvent::Handle.
1525 1566
1526To make this easier, a given version of AnyEvent::Handle uses these 1567To make this easier, a given version of AnyEvent::Handle uses these

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines