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.16 by root, Fri May 23 05:16:57 2008 UTC vs.
Revision 1.21 by root, Sat May 24 15:03:42 2008 UTC

… …
83waiting for data. 83waiting for data.
84 84
85=item on_error => $cb->($self) 85=item on_error => $cb->($self)
86 86
87This is the fatal error callback, that is called when, well, a fatal error 87This is the fatal error callback, that is called when, well, a fatal error
88ocurs, such as not being able to resolve the hostname, failure to connect 88occurs, such as not being able to resolve the hostname, failure to connect
89or a read error. 89or a read error.
90 90
91The object will not be in a usable state when this callback has been 91The object will not be in a usable state when this callback has been
92called. 92called.
93 93
… …
102 102
103This sets the default read callback, which is called when data arrives 103This sets the default read callback, which is called when data arrives
104and no read request is in the queue. 104and no read request is in the queue.
105 105
106To access (and remove data from) the read buffer, use the C<< ->rbuf >> 106To access (and remove data from) the read buffer, use the C<< ->rbuf >>
107method or acces sthe C<$self->{rbuf}> member directly. 107method or access the C<$self->{rbuf}> member directly.
108 108
109When an EOF condition is detected then AnyEvent::Handle will first try to 109When an EOF condition is detected then AnyEvent::Handle will first try to
110feed all the remaining data to the queued callbacks and C<on_read> before 110feed all the remaining data to the queued callbacks and C<on_read> before
111calling the C<on_eof> callback. If no progress can be made, then a fatal 111calling the C<on_eof> callback. If no progress can be made, then a fatal
112error will be raised (with C<$!> set to C<EPIPE>). 112error will be raised (with C<$!> set to C<EPIPE>).
… …
139 139
140Sets the amount of bytes (default: C<0>) that make up an "empty" write 140Sets the amount of bytes (default: C<0>) that make up an "empty" write
141buffer: If the write reaches this size or gets even samller it is 141buffer: If the write reaches this size or gets even samller it is
142considered empty. 142considered empty.
143 143
144=item tls => "accept" | "connect" | Net::SSLeay::SSL object
145
146When this parameter is given, it enables TLS (SSL) mode, that means it
147will start making tls handshake and will transparently encrypt/decrypt
148data.
149
150For the TLS server side, use C<accept>, and for the TLS client side of a
151connection, use C<connect> mode.
152
153You can also provide your own TLS connection object, but you have
154to make sure that you call either C<Net::SSLeay::set_connect_state>
155or C<Net::SSLeay::set_accept_state> on it before you pass it to
156AnyEvent::Handle.
157
158=item tls_ctx => $ssl_ctx
159
160Use the given Net::SSLeay::CTX object to create the new TLS connection
161(unless a connection object was specified directly). If this parameter is
162missing, then AnyEvent::Handle will use C<AnyEvent::Handle::TLS_CTX>.
163
144=back 164=back
145 165
146=cut 166=cut
147 167
148sub new { 168sub new {
… …
151 my $self = bless { @_ }, $class; 171 my $self = bless { @_ }, $class;
152 172
153 $self->{fh} or Carp::croak "mandatory argument fh is missing"; 173 $self->{fh} or Carp::croak "mandatory argument fh is missing";
154 174
155 AnyEvent::Util::fh_nonblocking $self->{fh}, 1; 175 AnyEvent::Util::fh_nonblocking $self->{fh}, 1;
176
177 if ($self->{tls}) {
178 require Net::SSLeay;
179 $self->starttls (delete $self->{tls}, delete $self->{tls_ctx});
180 }
156 181
157 $self->on_eof (delete $self->{on_eof} ) if $self->{on_eof}; 182 $self->on_eof (delete $self->{on_eof} ) if $self->{on_eof};
158 $self->on_error (delete $self->{on_error}) if $self->{on_error}; 183 $self->on_error (delete $self->{on_error}) if $self->{on_error};
159 $self->on_drain (delete $self->{on_drain}) if $self->{on_drain}; 184 $self->on_drain (delete $self->{on_drain}) if $self->{on_drain};
160 $self->on_read (delete $self->{on_read} ) if $self->{on_read}; 185 $self->on_read (delete $self->{on_read} ) if $self->{on_read};
… …
225for reading. 250for reading.
226 251
227The write queue is very simple: you can add data to its end, and 252The write queue is very simple: you can add data to its end, and
228AnyEvent::Handle will automatically try to get rid of it for you. 253AnyEvent::Handle will automatically try to get rid of it for you.
229 254
230When data could be writtena nd the write buffer is shorter then the low 255When data could be written and the write buffer is shorter then the low
231water mark, the C<on_drain> callback will be invoked. 256water mark, the C<on_drain> callback will be invoked.
232 257
233=over 4 258=over 4
234 259
235=item $handle->on_drain ($cb) 260=item $handle->on_drain ($cb)
… …
254want (only limited by the available memory), as C<AnyEvent::Handle> 279want (only limited by the available memory), as C<AnyEvent::Handle>
255buffers it independently of the kernel. 280buffers it independently of the kernel.
256 281
257=cut 282=cut
258 283
259sub push_write { 284sub _drain_wbuf {
260 my ($self, $data) = @_; 285 my ($self) = @_;
261
262 $self->{wbuf} .= $data;
263 286
264 unless ($self->{ww}) { 287 unless ($self->{ww}) {
265 Scalar::Util::weaken $self; 288 Scalar::Util::weaken $self;
266 my $cb = sub { 289 my $cb = sub {
267 my $len = syswrite $self->{fh}, $self->{wbuf}; 290 my $len = syswrite $self->{fh}, $self->{wbuf};
268 291
269 if ($len > 0) { 292 if ($len > 0) {
270 substr $self->{wbuf}, 0, $len, ""; 293 substr $self->{wbuf}, 0, $len, "";
271
272 294
273 $self->{on_drain}($self) 295 $self->{on_drain}($self)
274 if $self->{low_water_mark} >= length $self->{wbuf} 296 if $self->{low_water_mark} >= length $self->{wbuf}
275 && $self->{on_drain}; 297 && $self->{on_drain};
276 298
… …
282 304
283 $self->{ww} = AnyEvent->io (fh => $self->{fh}, poll => "w", cb => $cb); 305 $self->{ww} = AnyEvent->io (fh => $self->{fh}, poll => "w", cb => $cb);
284 306
285 $cb->($self); 307 $cb->($self);
286 }; 308 };
309}
310
311sub push_write {
312 my $self = shift;
313
314 if ($self->{filter_w}) {
315 $self->{filter_w}->($self, \$_[0]);
316 } else {
317 $self->{wbuf} .= $_[0];
318 $self->_drain_wbuf;
319 }
287} 320}
288 321
289############################################################################# 322#############################################################################
290 323
291=back 324=back
… …
366 399
367=cut 400=cut
368 401
369sub _drain_rbuf { 402sub _drain_rbuf {
370 my ($self) = @_; 403 my ($self) = @_;
404
405 if (
406 defined $self->{rbuf_max}
407 && $self->{rbuf_max} < length $self->{rbuf}
408 ) {
409 $! = &Errno::ENOSPC; return $self->error;
410 }
371 411
372 return if $self->{in_drain}; 412 return if $self->{in_drain};
373 local $self->{in_drain} = 1; 413 local $self->{in_drain} = 1;
374 414
375 while (my $len = length $self->{rbuf}) { 415 while (my $len = length $self->{rbuf}) {
… …
448Append the given callback to the end of the queue (C<push_read>) or 488Append the given callback to the end of the queue (C<push_read>) or
449prepend it (C<unshift_read>). 489prepend it (C<unshift_read>).
450 490
451The callback is called each time some additional read data arrives. 491The callback is called each time some additional read data arrives.
452 492
453It must check wether enough data is in the read buffer already. 493It must check whether enough data is in the read buffer already.
454 494
455If not enough data is available, it must return the empty list or a false 495If not enough data is available, it must return the empty list or a false
456value, in which case it will be called repeatedly until enough data is 496value, in which case it will be called repeatedly until enough data is
457available (or an error condition is detected). 497available (or an error condition is detected).
458 498
… …
560 600
561=item $handle->stop_read 601=item $handle->stop_read
562 602
563=item $handle->start_read 603=item $handle->start_read
564 604
565In rare cases you actually do not want to read anything form the 605In rare cases you actually do not want to read anything from the
566socket. In this case you can call C<stop_read>. Neither C<on_read> no 606socket. In this case you can call C<stop_read>. Neither C<on_read> no
567any queued callbacks will be executed then. To start readign again, call 607any queued callbacks will be executed then. To start readign again, call
568C<start_read>. 608C<start_read>.
569 609
570=cut 610=cut
… …
580 620
581 unless ($self->{rw} || $self->{eof}) { 621 unless ($self->{rw} || $self->{eof}) {
582 Scalar::Util::weaken $self; 622 Scalar::Util::weaken $self;
583 623
584 $self->{rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub { 624 $self->{rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub {
625 my $rbuf = $self->{filter_r} ? \my $buf : \$self->{rbuf};
585 my $len = sysread $self->{fh}, $self->{rbuf}, $self->{read_size} || 8192, length $self->{rbuf}; 626 my $len = sysread $self->{fh}, $$rbuf, $self->{read_size} || 8192, length $$rbuf;
586 627
587 if ($len > 0) { 628 if ($len > 0) {
588 if (defined $self->{rbuf_max}) { 629 $self->{filter_r}
589 if ($self->{rbuf_max} < length $self->{rbuf}) { 630 ? $self->{filter_r}->($self, $rbuf)
590 $! = &Errno::ENOSPC; return $self->error; 631 : $self->_drain_rbuf;
591 }
592 }
593 632
594 } elsif (defined $len) { 633 } elsif (defined $len) {
634 delete $self->{rw};
595 $self->{eof} = 1; 635 $self->{eof} = 1;
596 delete $self->{rw}; 636 $self->_drain_rbuf;
597 637
598 } elsif ($! != EAGAIN && $! != EINTR) { 638 } elsif ($! != EAGAIN && $! != EINTR) {
599 return $self->error; 639 return $self->error;
600 } 640 }
601
602 $self->_drain_rbuf;
603 }); 641 });
604 } 642 }
605} 643}
606 644
645sub _dotls {
646 my ($self) = @_;
647
648 if (length $self->{tls_wbuf}) {
649 my $len = Net::SSLeay::write ($self->{tls}, $self->{tls_wbuf});
650 substr $self->{tls_wbuf}, 0, $len, "" if $len > 0;
651 }
652
653 if (defined (my $buf = Net::SSLeay::BIO_read ($self->{tls_wbio}))) {
654 $self->{wbuf} .= $buf;
655 $self->_drain_wbuf;
656 }
657
658 if (defined (my $buf = Net::SSLeay::read ($self->{tls}))) {
659 $self->{rbuf} .= $buf;
660 $self->_drain_rbuf;
661 } elsif (
662 (my $err = Net::SSLeay::get_error ($self->{tls}, -1))
663 != Net::SSLeay::ERROR_WANT_READ ()
664 ) {
665 if ($err == Net::SSLeay::ERROR_SYSCALL ()) {
666 $self->error;
667 } elsif ($err == Net::SSLeay::ERROR_SSL ()) {
668 $! = &Errno::EIO;
669 $self->error;
670 }
671
672 # all others are fine for our purposes
673 }
674}
675
676# TODO: maybe document...
677sub starttls {
678 my ($self, $ssl, $ctx) = @_;
679
680 if ($ssl eq "accept") {
681 $ssl = Net::SSLeay::new ($ctx || TLS_CTX ());
682 Net::SSLeay::set_accept_state ($ssl);
683 } elsif ($ssl eq "connect") {
684 $ssl = Net::SSLeay::new ($ctx || TLS_CTX ());
685 Net::SSLeay::set_connect_state ($ssl);
686 }
687
688 $self->{tls} = $ssl;
689
690 # basically, this is deep magic (because SSL_read should have the same issues)
691 # but the openssl maintainers basically said: "trust us, it just works".
692 # (unfortunately, we have to hardcode constants because the abysmally misdesigned
693 # and mismaintained ssleay-module doesn't even offer them).
694 Net::SSLeay::CTX_set_mode ($self->{tls},
695 (eval { Net::SSLeay::MODE_ENABLE_PARTIAL_WRITE () } || 1)
696 | (eval { Net::SSLeay::MODE_ACCEPT_MOVING_WRITE_BUFFER () } || 2));
697
698 $self->{tls_rbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
699 $self->{tls_wbio} = Net::SSLeay::BIO_new (Net::SSLeay::BIO_s_mem ());
700
701 Net::SSLeay::set_bio ($ssl, $self->{tls_rbio}, $self->{tls_wbio});
702
703 $self->{filter_w} = sub {
704 $_[0]{tls_wbuf} .= ${$_[1]};
705 &_dotls;
706 };
707 $self->{filter_r} = sub {
708 Net::SSLeay::BIO_write ($_[0]{tls_rbio}, ${$_[1]});
709 &_dotls;
710 };
711}
712
713sub DESTROY {
714 my $self = shift;
715
716 Net::SSLeay::free (delete $self->{tls}) if $self->{tls};
717}
718
719=item AnyEvent::Handle::TLS_CTX
720
721This function creates and returns the Net::SSLeay::CTX object used by
722default for TLS mode.
723
724The context is created like this:
725
726 Net::SSLeay::load_error_strings;
727 Net::SSLeay::SSLeay_add_ssl_algorithms;
728 Net::SSLeay::randomize;
729
730 my $CTX = Net::SSLeay::CTX_new;
731
732 Net::SSLeay::CTX_set_options $CTX, Net::SSLeay::OP_ALL
733
734=cut
735
736our $TLS_CTX;
737
738sub TLS_CTX() {
739 $TLS_CTX || do {
740 require Net::SSLeay;
741
742 Net::SSLeay::load_error_strings ();
743 Net::SSLeay::SSLeay_add_ssl_algorithms ();
744 Net::SSLeay::randomize ();
745
746 $TLS_CTX = Net::SSLeay::CTX_new ();
747
748 Net::SSLeay::CTX_set_options ($TLS_CTX, Net::SSLeay::OP_ALL ());
749
750 $TLS_CTX
751 }
752}
753
607=back 754=back
608 755
609=head1 AUTHOR 756=head1 AUTHOR
610 757
611Robin Redeker C<< <elmex at ta-sa.org> >>, Marc Lehmann <schmorp@schmorp.de>. 758Robin Redeker C<< <elmex at ta-sa.org> >>, Marc Lehmann <schmorp@schmorp.de>.

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines