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.42 by root, Tue May 27 06:23:15 2008 UTC vs.
Revision 1.61 by root, Fri Jun 6 10:23:50 2008 UTC

6use AnyEvent (); 6use AnyEvent ();
7use AnyEvent::Util qw(WSAEWOULDBLOCK); 7use AnyEvent::Util qw(WSAEWOULDBLOCK);
8use Scalar::Util (); 8use Scalar::Util ();
9use Carp (); 9use Carp ();
10use Fcntl (); 10use Fcntl ();
11use Errno qw/EAGAIN EINTR/; 11use Errno qw(EAGAIN EINTR);
12 12
13=head1 NAME 13=head1 NAME
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 = '0.04'; 19our $VERSION = 4.14;
20 20
21=head1 SYNOPSIS 21=head1 SYNOPSIS
22 22
23 use AnyEvent; 23 use AnyEvent;
24 use AnyEvent::Handle; 24 use AnyEvent::Handle;
75NOTE: The filehandle will be set to non-blocking (using 75NOTE: The filehandle will be set to non-blocking (using
76AnyEvent::Util::fh_nonblocking). 76AnyEvent::Util::fh_nonblocking).
77 77
78=item on_eof => $cb->($handle) 78=item on_eof => $cb->($handle)
79 79
80Set the callback to be called on EOF. 80Set the callback to be called when an end-of-file condition is detcted,
81i.e. in the case of a socket, when the other side has closed the
82connection cleanly.
81 83
82While not mandatory, it is highly recommended to set an eof callback, 84While not mandatory, it is highly recommended to set an eof callback,
83otherwise you might end up with a closed socket while you are still 85otherwise you might end up with a closed socket while you are still
84waiting for data. 86waiting for data.
85 87
86=item on_error => $cb->($handle) 88=item on_error => $cb->($handle, $fatal)
87 89
88This is the fatal error callback, that is called when, well, a fatal error 90This is the error callback, which is called when, well, some error
89occurs, such as not being able to resolve the hostname, failure to connect 91occured, such as not being able to resolve the hostname, failure to
90or a read error. 92connect or a read error.
91 93
92The object will not be in a usable state when this callback has been 94Some errors are fatal (which is indicated by C<$fatal> being true). On
93called. 95fatal errors the handle object will be shut down and will not be
96usable. Non-fatal errors can be retried by simply returning, but it is
97recommended to simply ignore this parameter and instead abondon the handle
98object when this callback is invoked.
94 99
95On callback entrance, the value of C<$!> contains the operating system 100On callback entrance, the value of C<$!> contains the operating system
96error (or C<ENOSPC>, C<EPIPE> or C<EBADMSG>). 101error (or C<ENOSPC>, C<EPIPE>, C<ETIMEDOUT> or C<EBADMSG>).
97
98The callback should throw an exception. If it returns, then
99AnyEvent::Handle will C<croak> for you.
100 102
101While not mandatory, it is I<highly> recommended to set this callback, as 103While not mandatory, it is I<highly> recommended to set this callback, as
102you will not be notified of errors otherwise. The default simply calls 104you will not be notified of errors otherwise. The default simply calls
103die. 105C<croak>.
104 106
105=item on_read => $cb->($handle) 107=item on_read => $cb->($handle)
106 108
107This sets the default read callback, which is called when data arrives 109This sets the default read callback, which is called when data arrives
108and no read request is in the queue. 110and no read request is in the queue (unlike read queue callbacks, this
111callback will only be called when at least one octet of data is in the
112read buffer).
109 113
110To access (and remove data from) the read buffer, use the C<< ->rbuf >> 114To access (and remove data from) the read buffer, use the C<< ->rbuf >>
111method or access the C<$handle->{rbuf}> member directly. 115method or access the C<$handle->{rbuf}> member directly.
112 116
113When an EOF condition is detected then AnyEvent::Handle will first try to 117When an EOF condition is detected then AnyEvent::Handle will first try to
119 123
120This sets the callback that is called when the write buffer becomes empty 124This sets the callback that is called when the write buffer becomes empty
121(or when the callback is set and the buffer is empty already). 125(or when the callback is set and the buffer is empty already).
122 126
123To append to the write buffer, use the C<< ->push_write >> method. 127To append to the write buffer, use the C<< ->push_write >> method.
128
129=item timeout => $fractional_seconds
130
131If non-zero, then this enables an "inactivity" timeout: whenever this many
132seconds pass without a successful read or write on the underlying file
133handle, the C<on_timeout> callback will be invoked (and if that one is
134missing, an C<ETIMEDOUT> error will be raised).
135
136Note that timeout processing is also active when you currently do not have
137any outstanding read or write requests: If you plan to keep the connection
138idle then you should disable the timout temporarily or ignore the timeout
139in the C<on_timeout> callback.
140
141Zero (the default) disables this timeout.
142
143=item on_timeout => $cb->($handle)
144
145Called whenever the inactivity timeout passes. If you return from this
146callback, then the timeout will be reset as if some activity had happened,
147so this condition is not fatal in any way.
124 148
125=item rbuf_max => <bytes> 149=item rbuf_max => <bytes>
126 150
127If defined, then a fatal error will be raised (with C<$!> set to C<ENOSPC>) 151If defined, then a fatal error will be raised (with C<$!> set to C<ENOSPC>)
128when the read buffer ever (strictly) exceeds this size. This is useful to 152when the read buffer ever (strictly) exceeds this size. This is useful to
135isn't finished). 159isn't finished).
136 160
137=item read_size => <bytes> 161=item read_size => <bytes>
138 162
139The default read block size (the amount of bytes this module will try to read 163The default read block size (the amount of bytes this module will try to read
140on each [loop iteration). Default: C<4096>. 164during each (loop iteration). Default: C<8192>.
141 165
142=item low_water_mark => <bytes> 166=item low_water_mark => <bytes>
143 167
144Sets the amount of bytes (default: C<0>) that make up an "empty" write 168Sets the amount of bytes (default: C<0>) that make up an "empty" write
145buffer: If the write reaches this size or gets even samller it is 169buffer: If the write reaches this size or gets even samller it is
202 if ($self->{tls}) { 226 if ($self->{tls}) {
203 require Net::SSLeay; 227 require Net::SSLeay;
204 $self->starttls (delete $self->{tls}, delete $self->{tls_ctx}); 228 $self->starttls (delete $self->{tls}, delete $self->{tls_ctx});
205 } 229 }
206 230
207 $self->on_eof (delete $self->{on_eof} ) if $self->{on_eof}; 231 $self->{_activity} = AnyEvent->now;
208 $self->on_error (delete $self->{on_error}) if $self->{on_error}; 232 $self->_timeout;
233
209 $self->on_drain (delete $self->{on_drain}) if $self->{on_drain}; 234 $self->on_drain (delete $self->{on_drain}) if $self->{on_drain};
210 $self->on_read (delete $self->{on_read} ) if $self->{on_read};
211
212 $self->start_read;
213 235
214 $self 236 $self
215} 237}
216 238
217sub _shutdown { 239sub _shutdown {
218 my ($self) = @_; 240 my ($self) = @_;
219 241
242 delete $self->{_tw};
220 delete $self->{_rw}; 243 delete $self->{_rw};
221 delete $self->{_ww}; 244 delete $self->{_ww};
222 delete $self->{fh}; 245 delete $self->{fh};
223}
224 246
247 $self->stoptls;
248}
249
225sub error { 250sub _error {
226 my ($self) = @_; 251 my ($self, $errno, $fatal) = @_;
227 252
228 {
229 local $!;
230 $self->_shutdown; 253 $self->_shutdown
231 } 254 if $fatal;
232 255
233 $self->{on_error}($self) 256 $! = $errno;
257
234 if $self->{on_error}; 258 if ($self->{on_error}) {
235 259 $self->{on_error}($self, $fatal);
260 } else {
236 Carp::croak "AnyEvent::Handle uncaught fatal error: $!"; 261 Carp::croak "AnyEvent::Handle uncaught error: $!";
262 }
237} 263}
238 264
239=item $fh = $handle->fh 265=item $fh = $handle->fh
240 266
241This method returns the file handle of the L<AnyEvent::Handle> object. 267This method returns the file handle of the L<AnyEvent::Handle> object.
260 286
261=cut 287=cut
262 288
263sub on_eof { 289sub on_eof {
264 $_[0]{on_eof} = $_[1]; 290 $_[0]{on_eof} = $_[1];
291}
292
293=item $handle->on_timeout ($cb)
294
295Replace the current C<on_timeout> callback, or disables the callback
296(but not the timeout) if C<$cb> = C<undef>. See C<timeout> constructor
297argument.
298
299=cut
300
301sub on_timeout {
302 $_[0]{on_timeout} = $_[1];
303}
304
305#############################################################################
306
307=item $handle->timeout ($seconds)
308
309Configures (or disables) the inactivity timeout.
310
311=cut
312
313sub timeout {
314 my ($self, $timeout) = @_;
315
316 $self->{timeout} = $timeout;
317 $self->_timeout;
318}
319
320# reset the timeout watcher, as neccessary
321# also check for time-outs
322sub _timeout {
323 my ($self) = @_;
324
325 if ($self->{timeout}) {
326 my $NOW = AnyEvent->now;
327
328 # when would the timeout trigger?
329 my $after = $self->{_activity} + $self->{timeout} - $NOW;
330
331 # now or in the past already?
332 if ($after <= 0) {
333 $self->{_activity} = $NOW;
334
335 if ($self->{on_timeout}) {
336 $self->{on_timeout}($self);
337 } else {
338 $self->_error (&Errno::ETIMEDOUT);
339 }
340
341 # callback could have changed timeout value, optimise
342 return unless $self->{timeout};
343
344 # calculate new after
345 $after = $self->{timeout};
346 }
347
348 Scalar::Util::weaken $self;
349 return unless $self; # ->error could have destroyed $self
350
351 $self->{_tw} ||= AnyEvent->timer (after => $after, cb => sub {
352 delete $self->{_tw};
353 $self->_timeout;
354 });
355 } else {
356 delete $self->{_tw};
357 }
265} 358}
266 359
267############################################################################# 360#############################################################################
268 361
269=back 362=back
316 my $len = syswrite $self->{fh}, $self->{wbuf}; 409 my $len = syswrite $self->{fh}, $self->{wbuf};
317 410
318 if ($len >= 0) { 411 if ($len >= 0) {
319 substr $self->{wbuf}, 0, $len, ""; 412 substr $self->{wbuf}, 0, $len, "";
320 413
414 $self->{_activity} = AnyEvent->now;
415
321 $self->{on_drain}($self) 416 $self->{on_drain}($self)
322 if $self->{low_water_mark} >= length $self->{wbuf} 417 if $self->{low_water_mark} >= length $self->{wbuf}
323 && $self->{on_drain}; 418 && $self->{on_drain};
324 419
325 delete $self->{_ww} unless length $self->{wbuf}; 420 delete $self->{_ww} unless length $self->{wbuf};
326 } elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) { 421 } elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) {
327 $self->error; 422 $self->_error ($!, 1);
328 } 423 }
329 }; 424 };
330 425
331 # try to write data immediately 426 # try to write data immediately
332 $cb->(); 427 $cb->();
352 @_ = ($WH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_write") 447 @_ = ($WH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_write")
353 ->($self, @_); 448 ->($self, @_);
354 } 449 }
355 450
356 if ($self->{filter_w}) { 451 if ($self->{filter_w}) {
357 $self->{filter_w}->($self, \$_[0]); 452 $self->{filter_w}($self, \$_[0]);
358 } else { 453 } else {
359 $self->{wbuf} .= $_[0]; 454 $self->{wbuf} .= $_[0];
360 $self->_drain_wbuf; 455 $self->_drain_wbuf;
361 } 456 }
362} 457}
363 458
364=item $handle->push_write (type => @args) 459=item $handle->push_write (type => @args)
365 460
366=item $handle->unshift_write (type => @args)
367
368Instead of formatting your data yourself, you can also let this module do 461Instead of formatting your data yourself, you can also let this module do
369the job by specifying a type and type-specific arguments. 462the job by specifying a type and type-specific arguments.
370 463
371Predefined types are (if you have ideas for additional types, feel free to 464Predefined types are (if you have ideas for additional types, feel free to
372drop by and tell us): 465drop by and tell us):
376=item netstring => $string 469=item netstring => $string
377 470
378Formats the given value as netstring 471Formats the given value as netstring
379(http://cr.yp.to/proto/netstrings.txt, this is not a recommendation to use them). 472(http://cr.yp.to/proto/netstrings.txt, this is not a recommendation to use them).
380 473
381=back
382
383=cut 474=cut
384 475
385register_write_type netstring => sub { 476register_write_type netstring => sub {
386 my ($self, $string) = @_; 477 my ($self, $string) = @_;
387 478
388 sprintf "%d:%s,", (length $string), $string 479 sprintf "%d:%s,", (length $string), $string
480};
481
482=item packstring => $format, $data
483
484An octet string prefixed with an encoded length. The encoding C<$format>
485uses the same format as a Perl C<pack> format, but must specify a single
486integer only (only one of C<cCsSlLqQiInNvVjJw> is allowed, plus an
487optional C<!>, C<< < >> or C<< > >> modifier).
488
489=cut
490
491register_write_type packstring => sub {
492 my ($self, $format, $string) = @_;
493
494 pack "$format/a", $string
389}; 495};
390 496
391=item json => $array_or_hashref 497=item json => $array_or_hashref
392 498
393Encodes the given hash or array reference into a JSON object. Unless you 499Encodes the given hash or array reference into a JSON object. Unless you
427 533
428 $self->{json} ? $self->{json}->encode ($ref) 534 $self->{json} ? $self->{json}->encode ($ref)
429 : JSON::encode_json ($ref) 535 : JSON::encode_json ($ref)
430}; 536};
431 537
538=back
539
432=item AnyEvent::Handle::register_write_type type => $coderef->($handle, @args) 540=item AnyEvent::Handle::register_write_type type => $coderef->($handle, @args)
433 541
434This function (not method) lets you add your own types to C<push_write>. 542This function (not method) lets you add your own types to C<push_write>.
435Whenever the given C<type> is used, C<push_write> will invoke the code 543Whenever the given C<type> is used, C<push_write> will invoke the code
436reference with the handle object and the remaining arguments. 544reference with the handle object and the remaining arguments.
461enough is there) from the read buffer (C<< $handle->rbuf >>) if you want 569enough is there) from the read buffer (C<< $handle->rbuf >>) if you want
462or not. 570or not.
463 571
464In the more complex case, you want to queue multiple callbacks. In this 572In the more complex case, you want to queue multiple callbacks. In this
465case, AnyEvent::Handle will call the first queued callback each time new 573case, AnyEvent::Handle will call the first queued callback each time new
466data arrives and removes it when it has done its job (see C<push_read>, 574data arrives (also the first time it is queued) and removes it when it has
467below). 575done its job (see C<push_read>, below).
468 576
469This way you can, for example, push three line-reads, followed by reading 577This way you can, for example, push three line-reads, followed by reading
470a chunk of data, and AnyEvent::Handle will execute them in order. 578a chunk of data, and AnyEvent::Handle will execute them in order.
471 579
472Example 1: EPP protocol parser. EPP sends 4 byte length info, followed by 580Example 1: EPP protocol parser. EPP sends 4 byte length info, followed by
473the specified number of bytes which give an XML datagram. 581the specified number of bytes which give an XML datagram.
474 582
475 # in the default state, expect some header bytes 583 # in the default state, expect some header bytes
476 $handle->on_read (sub { 584 $handle->on_read (sub {
477 # some data is here, now queue the length-header-read (4 octets) 585 # some data is here, now queue the length-header-read (4 octets)
478 shift->unshift_read_chunk (4, sub { 586 shift->unshift_read (chunk => 4, sub {
479 # header arrived, decode 587 # header arrived, decode
480 my $len = unpack "N", $_[1]; 588 my $len = unpack "N", $_[1];
481 589
482 # now read the payload 590 # now read the payload
483 shift->unshift_read_chunk ($len, sub { 591 shift->unshift_read (chunk => $len, sub {
484 my $xml = $_[1]; 592 my $xml = $_[1];
485 # handle xml 593 # handle xml
486 }); 594 });
487 }); 595 });
488 }); 596 });
495 603
496 # request one 604 # request one
497 $handle->push_write ("request 1\015\012"); 605 $handle->push_write ("request 1\015\012");
498 606
499 # we expect "ERROR" or "OK" as response, so push a line read 607 # we expect "ERROR" or "OK" as response, so push a line read
500 $handle->push_read_line (sub { 608 $handle->push_read (line => sub {
501 # if we got an "OK", we have to _prepend_ another line, 609 # if we got an "OK", we have to _prepend_ another line,
502 # so it will be read before the second request reads its 64 bytes 610 # so it will be read before the second request reads its 64 bytes
503 # which are already in the queue when this callback is called 611 # which are already in the queue when this callback is called
504 # we don't do this in case we got an error 612 # we don't do this in case we got an error
505 if ($_[1] eq "OK") { 613 if ($_[1] eq "OK") {
506 $_[0]->unshift_read_line (sub { 614 $_[0]->unshift_read (line => sub {
507 my $response = $_[1]; 615 my $response = $_[1];
508 ... 616 ...
509 }); 617 });
510 } 618 }
511 }); 619 });
512 620
513 # request two 621 # request two
514 $handle->push_write ("request 2\015\012"); 622 $handle->push_write ("request 2\015\012");
515 623
516 # simply read 64 bytes, always 624 # simply read 64 bytes, always
517 $handle->push_read_chunk (64, sub { 625 $handle->push_read (chunk => 64, sub {
518 my $response = $_[1]; 626 my $response = $_[1];
519 ... 627 ...
520 }); 628 });
521 629
522=over 4 630=over 4
523 631
524=cut 632=cut
525 633
526sub _drain_rbuf { 634sub _drain_rbuf {
527 my ($self) = @_; 635 my ($self) = @_;
636
637 local $self->{_in_drain} = 1;
528 638
529 if ( 639 if (
530 defined $self->{rbuf_max} 640 defined $self->{rbuf_max}
531 && $self->{rbuf_max} < length $self->{rbuf} 641 && $self->{rbuf_max} < length $self->{rbuf}
532 ) { 642 ) {
533 $! = &Errno::ENOSPC; 643 return $self->_error (&Errno::ENOSPC, 1);
534 $self->error;
535 } 644 }
536 645
537 return if $self->{in_drain}; 646 while () {
538 local $self->{in_drain} = 1;
539
540 while (my $len = length $self->{rbuf}) {
541 no strict 'refs'; 647 no strict 'refs';
648
649 my $len = length $self->{rbuf};
650
542 if (my $cb = shift @{ $self->{_queue} }) { 651 if (my $cb = shift @{ $self->{_queue} }) {
543 unless ($cb->($self)) { 652 unless ($cb->($self)) {
544 if ($self->{_eof}) { 653 if ($self->{_eof}) {
545 # no progress can be made (not enough data and no data forthcoming) 654 # no progress can be made (not enough data and no data forthcoming)
546 $! = &Errno::EPIPE; 655 $self->_error (&Errno::EPIPE, 1), last;
547 $self->error;
548 } 656 }
549 657
550 unshift @{ $self->{_queue} }, $cb; 658 unshift @{ $self->{_queue} }, $cb;
551 return; 659 last;
552 } 660 }
553 } elsif ($self->{on_read}) { 661 } elsif ($self->{on_read}) {
662 last unless $len;
663
554 $self->{on_read}($self); 664 $self->{on_read}($self);
555 665
556 if ( 666 if (
557 $self->{_eof} # if no further data will arrive
558 && $len == length $self->{rbuf} # and no data has been consumed 667 $len == length $self->{rbuf} # if no data has been consumed
559 && !@{ $self->{_queue} } # and the queue is still empty 668 && !@{ $self->{_queue} } # and the queue is still empty
560 && $self->{on_read} # and we still want to read data 669 && $self->{on_read} # but we still have on_read
561 ) { 670 ) {
671 # no further data will arrive
562 # then no progress can be made 672 # so no progress can be made
563 $! = &Errno::EPIPE; 673 $self->_error (&Errno::EPIPE, 1), last
564 $self->error; 674 if $self->{_eof};
675
676 last; # more data might arrive
565 } 677 }
566 } else { 678 } else {
567 # read side becomes idle 679 # read side becomes idle
568 delete $self->{_rw}; 680 delete $self->{_rw};
569 return; 681 last;
570 } 682 }
571 } 683 }
572 684
573 if ($self->{_eof}) {
574 $self->_shutdown;
575 $self->{on_eof}($self) 685 $self->{on_eof}($self)
576 if $self->{on_eof}; 686 if $self->{_eof} && $self->{on_eof};
687
688 # may need to restart read watcher
689 unless ($self->{_rw}) {
690 $self->start_read
691 if $self->{on_read} || @{ $self->{_queue} };
577 } 692 }
578} 693}
579 694
580=item $handle->on_read ($cb) 695=item $handle->on_read ($cb)
581 696
587 702
588sub on_read { 703sub on_read {
589 my ($self, $cb) = @_; 704 my ($self, $cb) = @_;
590 705
591 $self->{on_read} = $cb; 706 $self->{on_read} = $cb;
707 $self->_drain_rbuf if $cb && !$self->{_in_drain};
592} 708}
593 709
594=item $handle->rbuf 710=item $handle->rbuf
595 711
596Returns the read buffer (as a modifiable lvalue). 712Returns the read buffer (as a modifiable lvalue).
645 $cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_read") 761 $cb = ($RH{$type} or Carp::croak "unsupported type passed to AnyEvent::Handle::push_read")
646 ->($self, $cb, @_); 762 ->($self, $cb, @_);
647 } 763 }
648 764
649 push @{ $self->{_queue} }, $cb; 765 push @{ $self->{_queue} }, $cb;
650 $self->_drain_rbuf; 766 $self->_drain_rbuf unless $self->{_in_drain};
651} 767}
652 768
653sub unshift_read { 769sub unshift_read {
654 my $self = shift; 770 my $self = shift;
655 my $cb = pop; 771 my $cb = pop;
661 ->($self, $cb, @_); 777 ->($self, $cb, @_);
662 } 778 }
663 779
664 780
665 unshift @{ $self->{_queue} }, $cb; 781 unshift @{ $self->{_queue} }, $cb;
666 $self->_drain_rbuf; 782 $self->_drain_rbuf unless $self->{_in_drain};
667} 783}
668 784
669=item $handle->push_read (type => @args, $cb) 785=item $handle->push_read (type => @args, $cb)
670 786
671=item $handle->unshift_read (type => @args, $cb) 787=item $handle->unshift_read (type => @args, $cb)
756 872
757sub unshift_read_line { 873sub unshift_read_line {
758 my $self = shift; 874 my $self = shift;
759 $self->unshift_read (line => @_); 875 $self->unshift_read (line => @_);
760} 876}
761
762=item netstring => $cb->($handle, $string)
763
764A netstring (http://cr.yp.to/proto/netstrings.txt, this is not an endorsement).
765
766Throws an error with C<$!> set to EBADMSG on format violations.
767
768=cut
769
770register_read_type netstring => sub {
771 my ($self, $cb) = @_;
772
773 sub {
774 unless ($_[0]{rbuf} =~ s/^(0|[1-9][0-9]*)://) {
775 if ($_[0]{rbuf} =~ /[^0-9]/) {
776 $! = &Errno::EBADMSG;
777 $self->error;
778 }
779 return;
780 }
781
782 my $len = $1;
783
784 $self->unshift_read (chunk => $len, sub {
785 my $string = $_[1];
786 $_[0]->unshift_read (chunk => 1, sub {
787 if ($_[1] eq ",") {
788 $cb->($_[0], $string);
789 } else {
790 $! = &Errno::EBADMSG;
791 $self->error;
792 }
793 });
794 });
795
796 1
797 }
798};
799 877
800=item regex => $accept[, $reject[, $skip], $cb->($handle, $data) 878=item regex => $accept[, $reject[, $skip], $cb->($handle, $data)
801 879
802Makes a regex match against the regex object C<$accept> and returns 880Makes a regex match against the regex object C<$accept> and returns
803everything up to and including the match. 881everything up to and including the match.
853 return 1; 931 return 1;
854 } 932 }
855 933
856 # reject 934 # reject
857 if ($reject && $$rbuf =~ $reject) { 935 if ($reject && $$rbuf =~ $reject) {
858 $! = &Errno::EBADMSG; 936 $self->_error (&Errno::EBADMSG);
859 $self->error;
860 } 937 }
861 938
862 # skip 939 # skip
863 if ($skip && $$rbuf =~ $skip) { 940 if ($skip && $$rbuf =~ $skip) {
864 $data .= substr $$rbuf, 0, $+[0], ""; 941 $data .= substr $$rbuf, 0, $+[0], "";
865 } 942 }
866 943
867 () 944 ()
945 }
946};
947
948=item netstring => $cb->($handle, $string)
949
950A netstring (http://cr.yp.to/proto/netstrings.txt, this is not an endorsement).
951
952Throws an error with C<$!> set to EBADMSG on format violations.
953
954=cut
955
956register_read_type netstring => sub {
957 my ($self, $cb) = @_;
958
959 sub {
960 unless ($_[0]{rbuf} =~ s/^(0|[1-9][0-9]*)://) {
961 if ($_[0]{rbuf} =~ /[^0-9]/) {
962 $self->_error (&Errno::EBADMSG);
963 }
964 return;
965 }
966
967 my $len = $1;
968
969 $self->unshift_read (chunk => $len, sub {
970 my $string = $_[1];
971 $_[0]->unshift_read (chunk => 1, sub {
972 if ($_[1] eq ",") {
973 $cb->($_[0], $string);
974 } else {
975 $self->_error (&Errno::EBADMSG);
976 }
977 });
978 });
979
980 1
981 }
982};
983
984=item packstring => $format, $cb->($handle, $string)
985
986An octet string prefixed with an encoded length. The encoding C<$format>
987uses the same format as a Perl C<pack> format, but must specify a single
988integer only (only one of C<cCsSlLqQiInNvVjJw> is allowed, plus an
989optional C<!>, C<< < >> or C<< > >> modifier).
990
991DNS over TCP uses a prefix of C<n>, EPP uses a prefix of C<N>.
992
993Example: read a block of data prefixed by its length in BER-encoded
994format (very efficient).
995
996 $handle->push_read (packstring => "w", sub {
997 my ($handle, $data) = @_;
998 });
999
1000=cut
1001
1002register_read_type packstring => sub {
1003 my ($self, $cb, $format) = @_;
1004
1005 sub {
1006 # when we can use 5.10 we can use ".", but for 5.8 we use the re-pack method
1007 defined (my $len = eval { unpack $format, $_[0]->{rbuf} })
1008 or return;
1009
1010 # remove prefix
1011 substr $_[0]->{rbuf}, 0, (length pack $format, $len), "";
1012
1013 # read rest
1014 $_[0]->unshift_read (chunk => $len, $cb);
1015
1016 1
868 } 1017 }
869}; 1018};
870 1019
871=item json => $cb->($handle, $hash_or_arrayref) 1020=item json => $cb->($handle, $hash_or_arrayref)
872 1021
937=item $handle->stop_read 1086=item $handle->stop_read
938 1087
939=item $handle->start_read 1088=item $handle->start_read
940 1089
941In rare cases you actually do not want to read anything from the 1090In rare cases you actually do not want to read anything from the
942socket. In this case you can call C<stop_read>. Neither C<on_read> no 1091socket. In this case you can call C<stop_read>. Neither C<on_read> nor
943any queued callbacks will be executed then. To start reading again, call 1092any queued callbacks will be executed then. To start reading again, call
944C<start_read>. 1093C<start_read>.
1094
1095Note that AnyEvent::Handle will automatically C<start_read> for you when
1096you change the C<on_read> callback or push/unshift a read callback, and it
1097will automatically C<stop_read> for you when neither C<on_read> is set nor
1098there are any read requests in the queue.
945 1099
946=cut 1100=cut
947 1101
948sub stop_read { 1102sub stop_read {
949 my ($self) = @_; 1103 my ($self) = @_;
960 $self->{_rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub { 1114 $self->{_rw} = AnyEvent->io (fh => $self->{fh}, poll => "r", cb => sub {
961 my $rbuf = $self->{filter_r} ? \my $buf : \$self->{rbuf}; 1115 my $rbuf = $self->{filter_r} ? \my $buf : \$self->{rbuf};
962 my $len = sysread $self->{fh}, $$rbuf, $self->{read_size} || 8192, length $$rbuf; 1116 my $len = sysread $self->{fh}, $$rbuf, $self->{read_size} || 8192, length $$rbuf;
963 1117
964 if ($len > 0) { 1118 if ($len > 0) {
1119 $self->{_activity} = AnyEvent->now;
1120
965 $self->{filter_r} 1121 $self->{filter_r}
966 ? $self->{filter_r}->($self, $rbuf) 1122 ? $self->{filter_r}($self, $rbuf)
967 : $self->_drain_rbuf; 1123 : $self->{_in_drain} || $self->_drain_rbuf;
968 1124
969 } elsif (defined $len) { 1125 } elsif (defined $len) {
970 delete $self->{_rw}; 1126 delete $self->{_rw};
971 $self->{_eof} = 1; 1127 $self->{_eof} = 1;
972 $self->_drain_rbuf; 1128 $self->_drain_rbuf unless $self->{_in_drain};
973 1129
974 } elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) { 1130 } elsif ($! != EAGAIN && $! != EINTR && $! != WSAEWOULDBLOCK) {
975 return $self->error; 1131 return $self->_error ($!, 1);
976 } 1132 }
977 }); 1133 });
978 } 1134 }
979} 1135}
980 1136
981sub _dotls { 1137sub _dotls {
982 my ($self) = @_; 1138 my ($self) = @_;
1139
1140 my $buf;
983 1141
984 if (length $self->{_tls_wbuf}) { 1142 if (length $self->{_tls_wbuf}) {
985 while ((my $len = Net::SSLeay::write ($self->{tls}, $self->{_tls_wbuf})) > 0) { 1143 while ((my $len = Net::SSLeay::write ($self->{tls}, $self->{_tls_wbuf})) > 0) {
986 substr $self->{_tls_wbuf}, 0, $len, ""; 1144 substr $self->{_tls_wbuf}, 0, $len, "";
987 } 1145 }
988 } 1146 }
989 1147
990 if (defined (my $buf = Net::SSLeay::BIO_read ($self->{_wbio}))) { 1148 if (length ($buf = Net::SSLeay::BIO_read ($self->{_wbio}))) {
991 $self->{wbuf} .= $buf; 1149 $self->{wbuf} .= $buf;
992 $self->_drain_wbuf; 1150 $self->_drain_wbuf;
993 } 1151 }
994 1152
995 while (defined (my $buf = Net::SSLeay::read ($self->{tls}))) { 1153 while (defined ($buf = Net::SSLeay::read ($self->{tls}))) {
1154 if (length $buf) {
996 $self->{rbuf} .= $buf; 1155 $self->{rbuf} .= $buf;
997 $self->_drain_rbuf; 1156 $self->_drain_rbuf unless $self->{_in_drain};
1157 } else {
1158 # let's treat SSL-eof as we treat normal EOF
1159 $self->{_eof} = 1;
1160 $self->_shutdown;
1161 return;
1162 }
998 } 1163 }
999 1164
1000 my $err = Net::SSLeay::get_error ($self->{tls}, -1); 1165 my $err = Net::SSLeay::get_error ($self->{tls}, -1);
1001 1166
1002 if ($err!= Net::SSLeay::ERROR_WANT_READ ()) { 1167 if ($err!= Net::SSLeay::ERROR_WANT_READ ()) {
1003 if ($err == Net::SSLeay::ERROR_SYSCALL ()) { 1168 if ($err == Net::SSLeay::ERROR_SYSCALL ()) {
1004 $self->error; 1169 return $self->_error ($!, 1);
1005 } elsif ($err == Net::SSLeay::ERROR_SSL ()) { 1170 } elsif ($err == Net::SSLeay::ERROR_SSL ()) {
1006 $! = &Errno::EIO; 1171 return $self->_error (&Errno::EIO, 1);
1007 $self->error;
1008 } 1172 }
1009 1173
1010 # all others are fine for our purposes 1174 # all others are fine for our purposes
1011 } 1175 }
1012} 1176}
1027call and can be used or changed to your liking. Note that the handshake 1191call and can be used or changed to your liking. Note that the handshake
1028might have already started when this function returns. 1192might have already started when this function returns.
1029 1193
1030=cut 1194=cut
1031 1195
1032# TODO: maybe document...
1033sub starttls { 1196sub starttls {
1034 my ($self, $ssl, $ctx) = @_; 1197 my ($self, $ssl, $ctx) = @_;
1035 1198
1036 $self->stoptls; 1199 $self->stoptls;
1037 1200

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines