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

Comparing AnyEvent-HTTP/HTTP.pm (file contents):
Revision 1.89 by root, Mon Jan 3 00:23:25 2011 UTC vs.
Revision 1.105 by root, Tue May 10 12:33:51 2011 UTC

15This module is an L<AnyEvent> user, you need to make sure that you use and 15This module is an L<AnyEvent> user, you need to make sure that you use and
16run a supported event loop. 16run a supported event loop.
17 17
18This module implements a simple, stateless and non-blocking HTTP 18This module implements a simple, stateless and non-blocking HTTP
19client. It supports GET, POST and other request methods, cookies and more, 19client. It supports GET, POST and other request methods, cookies and more,
20all on a very low level. It can follow redirects supports proxies and 20all on a very low level. It can follow redirects, supports proxies, and
21automatically limits the number of connections to the values specified in 21automatically limits the number of connections to the values specified in
22the RFC. 22the RFC.
23 23
24It should generally be a "good client" that is enough for most HTTP 24It should generally be a "good client" that is enough for most HTTP
25tasks. Simple tasks should be simple, but complex tasks should still be 25tasks. Simple tasks should be simple, but complex tasks should still be
46use AnyEvent::Util (); 46use AnyEvent::Util ();
47use AnyEvent::Handle (); 47use AnyEvent::Handle ();
48 48
49use base Exporter::; 49use base Exporter::;
50 50
51our $VERSION = '1.5'; 51our $VERSION = '2.11';
52 52
53our @EXPORT = qw(http_get http_post http_head http_request); 53our @EXPORT = qw(http_get http_post http_head http_request);
54 54
55our $USERAGENT = "Mozilla/5.0 (compatible; U; AnyEvent-HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)"; 55our $USERAGENT = "Mozilla/5.0 (compatible; U; AnyEvent-HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)";
56our $MAX_RECURSE = 10; 56our $MAX_RECURSE = 10;
57our $MAX_PERSISTENT = 8;
58our $PERSISTENT_TIMEOUT = 2; 57our $PERSISTENT_TIMEOUT = 3;
59our $TIMEOUT = 300; 58our $TIMEOUT = 300;
60 59our $MAX_PER_HOST = 4; # changing this is evil
61# changing these is evil
62our $MAX_PERSISTENT_PER_HOST = 2;
63our $MAX_PER_HOST = 4;
64 60
65our $PROXY; 61our $PROXY;
66our $ACTIVE = 0; 62our $ACTIVE = 0;
67 63
68my %KA_COUNT; # number of open keep-alive connections per host 64my %KA_CACHE; # indexed by uhost currently, points to [$handle...] array
69my %CO_SLOT; # number of open connections, and wait queue, per host 65my %CO_SLOT; # number of open connections, and wait queue, per host
70 66
71=item http_get $url, key => value..., $cb->($data, $headers) 67=item http_get $url, key => value..., $cb->($data, $headers)
72 68
73Executes an HTTP-GET request. See the http_request function for details on 69Executes an HTTP-GET request. See the http_request function for details on
169C<Host:>, C<Content-Length:>, C<Connection:> and C<Cookie:> headers and 165C<Host:>, C<Content-Length:>, C<Connection:> and C<Cookie:> headers and
170will provide defaults at least for C<TE:>, C<Referer:> and C<User-Agent:> 166will provide defaults at least for C<TE:>, C<Referer:> and C<User-Agent:>
171(this can be suppressed by using C<undef> for these headers in which case 167(this can be suppressed by using C<undef> for these headers in which case
172they won't be sent at all). 168they won't be sent at all).
173 169
170You really should provide your own C<User-Agent:> header value that is
171appropriate for your program - I wouldn't be surprised if the default
172AnyEvent string gets blocked by webservers sooner or later.
173
174Also, make sure that your headers names and values do not contain any
175embedded newlines.
176
174=item timeout => $seconds 177=item timeout => $seconds
175 178
176The time-out to use for various stages - each connect attempt will reset 179The time-out to use for various stages - each connect attempt will reset
177the timeout, as will read or write activity, i.e. this is not an overall 180the timeout, as will read or write activity, i.e. this is not an overall
178timeout. 181timeout.
179 182
180Default timeout is 5 minutes. 183Default timeout is 5 minutes.
181 184
182=item proxy => [$host, $port[, $scheme]] or undef 185=item proxy => [$host, $port[, $scheme]] or undef
183 186
184Use the given http proxy for all requests. If not specified, then the 187Use the given http proxy for all requests, or no proxy if C<undef> is
185default proxy (as specified by C<$ENV{http_proxy}>) is used. 188used.
186 189
187C<$scheme> must be either missing, C<http> for HTTP or C<https> for 190C<$scheme> must be either missing or must be C<http> for HTTP.
188HTTPS. 191
192If not specified, then the default proxy is used (see
193C<AnyEvent::HTTP::set_proxy>).
189 194
190=item body => $string 195=item body => $string
191 196
192The request body, usually empty. Will be sent as-is (future versions of 197The request body, usually empty. Will be sent as-is (future versions of
193this module might offer more options). 198this module might offer more options).
223verification, highest compatibility) and high-security (CA and common-name 228verification, highest compatibility) and high-security (CA and common-name
224verification) TLS context. 229verification) TLS context.
225 230
226The default for this option is C<low>, which could be interpreted as "give 231The default for this option is C<low>, which could be interpreted as "give
227me the page, no matter what". 232me the page, no matter what".
233
234See also the C<sessionid> parameter.
235
236=item session => $string
237
238The module might reuse connections to the same host internally. Sometimes
239(e.g. when using TLS), you do not want to reuse connections from other
240sessions. This can be achieved by setting this parameter to some unique
241ID (such as the address of an object storing your state data, or the TLS
242context) - only connections using the same unique ID will be reused.
228 243
229=item on_prepare => $callback->($fh) 244=item on_prepare => $callback->($fh)
230 245
231In rare cases you need to "tune" the socket before it is used to 246In rare cases you need to "tune" the socket before it is used to
232connect (for exmaple, to bind it on a given IP address). This parameter 247connect (for exmaple, to bind it on a given IP address). This parameter
302called. Instead of the C<$body> argument containing the body data, the 317called. Instead of the C<$body> argument containing the body data, the
303callback will receive the L<AnyEvent::Handle> object associated with the 318callback will receive the L<AnyEvent::Handle> object associated with the
304connection. In error cases, C<undef> will be passed. When there is no body 319connection. In error cases, C<undef> will be passed. When there is no body
305(e.g. status C<304>), the empty string will be passed. 320(e.g. status C<304>), the empty string will be passed.
306 321
307The handle object might or might not be in TLS mode, might be connected to 322The handle object might or might not be in TLS mode, might be connected
308a proxy, be a persistent connection etc., and configured in unspecified 323to a proxy, be a persistent connection, use chunked transfer encoding
309ways. The user is responsible for this handle (it will not be used by this 324etc., and configured in unspecified ways. The user is responsible for this
310module anymore). 325handle (it will not be used by this module anymore).
311 326
312This is useful with some push-type services, where, after the initial 327This is useful with some push-type services, where, after the initial
313headers, an interactive protocol is used (typical example would be the 328headers, an interactive protocol is used (typical example would be the
314push-style twitter API which starts a JSON/XML stream). 329push-style twitter API which starts a JSON/XML stream).
315 330
316If you think you need this, first have a look at C<on_body>, to see if 331If you think you need this, first have a look at C<on_body>, to see if
317that doesn't solve your problem in a better way. 332that doesn't solve your problem in a better way.
333
334=item persistent => $boolean
335
336Try to create/reuse a persistent connection. When this flag is set
337(default: true for idempotent requests, false for all others), then
338C<http_request> tries to re-use an existing (previously-created)
339persistent connection to the host and, failing that, tries to create a new
340one.
341
342Requests failing in certain ways will be automatically retried once, which
343is dangerous for non-idempotent requests, which is why it defaults to off
344for them. The reason for this is because the bozos who designed HTTP/1.1
345made it impossible to distinguish between a fatal error and a normal
346connection timeout, so you never know whether there was a problem with
347your request or not.
348
349When reusing an existent connection, many parameters (such as TLS context)
350will be ignored. See the C<session> parameter for a workaround.
351
352=item keepalive => $boolean
353
354Only used when C<persistent> is also true. This parameter decides whether
355C<http_request> tries to handshake a HTTP/1.0-style keep-alive connection
356(as opposed to only a HTTP/1.1 persistent connection).
357
358The default is true, except when using a proxy, in which case it defaults
359to false, as HTTP/1.0 proxies cannot support this in a meaningful way.
360
361=item handle_params => { key => value ... }
362
363The key-value pairs in this hash will be passed to any L<AnyEvent::Handle>
364constructor that is called - not all requests will create a handle, and
365sometimes more than one is created, so this parameter is only good for
366setting hints.
367
368Example: set the maximum read size to 4096, to potentially conserve memory
369at the cost of speed.
370
371 handle_params => {
372 max_read_size => 4096,
373 },
318 374
319=back 375=back
320 376
321Example: do a simple HTTP GET request for http://www.nethype.de/ and print 377Example: do a simple HTTP GET request for http://www.nethype.de/ and print
322the response body. 378the response body.
329Example: do a HTTP HEAD request on https://www.google.com/, use a 385Example: do a HTTP HEAD request on https://www.google.com/, use a
330timeout of 30 seconds. 386timeout of 30 seconds.
331 387
332 http_request 388 http_request
333 GET => "https://www.google.com", 389 GET => "https://www.google.com",
390 headers => { "user-agent" => "MySearchClient 1.0" },
334 timeout => 30, 391 timeout => 30,
335 sub { 392 sub {
336 my ($body, $hdr) = @_; 393 my ($body, $hdr) = @_;
337 use Data::Dumper; 394 use Data::Dumper;
338 print Dumper $hdr; 395 print Dumper $hdr;
348 }; 405 };
349 406
350 undef $request; 407 undef $request;
351 408
352=cut 409=cut
410
411#############################################################################
412# wait queue/slots
353 413
354sub _slot_schedule; 414sub _slot_schedule;
355sub _slot_schedule($) { 415sub _slot_schedule($) {
356 my $host = shift; 416 my $host = shift;
357 417
380 440
381 _slot_schedule $_[0]; 441 _slot_schedule $_[0];
382} 442}
383 443
384############################################################################# 444#############################################################################
445# cookie handling
385 446
386# expire cookies 447# expire cookies
387sub cookie_jar_expire($;$) { 448sub cookie_jar_expire($;$) {
388 my ($jar, $session_end) = @_; 449 my ($jar, $session_end) = @_;
389 450
413 } 474 }
414} 475}
415 476
416# extract cookies from jar 477# extract cookies from jar
417sub cookie_jar_extract($$$$) { 478sub cookie_jar_extract($$$$) {
418 my ($jar, $uscheme, $uhost, $upath) = @_; 479 my ($jar, $scheme, $host, $path) = @_;
419 480
420 %$jar = () if $jar->{version} != 1; 481 %$jar = () if $jar->{version} != 1;
421 482
422 my @cookies; 483 my @cookies;
423 484
424 while (my ($chost, $paths) = each %$jar) { 485 while (my ($chost, $paths) = each %$jar) {
425 next unless ref $paths; 486 next unless ref $paths;
426 487
427 if ($chost =~ /^\./) { 488 if ($chost =~ /^\./) {
428 next unless $chost eq substr $uhost, -length $chost; 489 next unless $chost eq substr $host, -length $chost;
429 } elsif ($chost =~ /\./) { 490 } elsif ($chost =~ /\./) {
430 next unless $chost eq $uhost; 491 next unless $chost eq $host;
431 } else { 492 } else {
432 next; 493 next;
433 } 494 }
434 495
435 while (my ($cpath, $cookies) = each %$paths) { 496 while (my ($cpath, $cookies) = each %$paths) {
436 next unless $cpath eq substr $upath, 0, length $cpath; 497 next unless $cpath eq substr $path, 0, length $cpath;
437 498
438 while (my ($cookie, $kv) = each %$cookies) { 499 while (my ($cookie, $kv) = each %$cookies) {
439 next if $uscheme ne "https" && exists $kv->{secure}; 500 next if $scheme ne "https" && exists $kv->{secure};
440 501
441 if (exists $kv->{_expires} and AE::now > $kv->{_expires}) { 502 if (exists $kv->{_expires} and AE::now > $kv->{_expires}) {
442 delete $cookies->{$cookie}; 503 delete $cookies->{$cookie};
443 next; 504 next;
444 } 505 }
458 \@cookies 519 \@cookies
459} 520}
460 521
461# parse set_cookie header into jar 522# parse set_cookie header into jar
462sub cookie_jar_set_cookie($$$$) { 523sub cookie_jar_set_cookie($$$$) {
463 my ($jar, $set_cookie, $uhost, $date) = @_; 524 my ($jar, $set_cookie, $host, $date) = @_;
464 525
465 my $anow = int AE::now; 526 my $anow = int AE::now;
466 my $snow; # server-now 527 my $snow; # server-now
467 528
468 for ($set_cookie) { 529 for ($set_cookie) {
474 while ( 535 while (
475 m{ 536 m{
476 \G\s* 537 \G\s*
477 (?: 538 (?:
478 expires \s*=\s* ([A-Z][a-z][a-z]+,\ [^,;]+) 539 expires \s*=\s* ([A-Z][a-z][a-z]+,\ [^,;]+)
479 | ([^=;,[:space:]]+) (?: \s*=\s* (?: "((?:[^\\"]+|\\.)*)" | ([^=;,[:space:]]*) ) )? 540 | ([^=;,[:space:]]+) (?: \s*=\s* (?: "((?:[^\\"]+|\\.)*)" | ([^;,[:space:]]*) ) )?
480 ) 541 )
481 }gcxsi 542 }gcxsi
482 ) { 543 ) {
483 my $name = $2; 544 my $name = $2;
484 my $value = $4; 545 my $value = $4;
491 # quoted 552 # quoted
492 $value = $3; 553 $value = $3;
493 $value =~ s/\\(.)/$1/gs; 554 $value =~ s/\\(.)/$1/gs;
494 } 555 }
495 556
496 push @kv, lc $name, $value; 557 push @kv, @kv ? lc $name : $name, $value;
497 558
498 last unless /\G\s*;/gc; 559 last unless /\G\s*;/gc;
499 } 560 }
500 561
501 last unless @kv; 562 last unless @kv;
524 585
525 # this is not rfc-like and not netscape-like. go figure. 586 # this is not rfc-like and not netscape-like. go figure.
526 my $ndots = $cdom =~ y/.//; 587 my $ndots = $cdom =~ y/.//;
527 next if $ndots < ($cdom =~ /\.[^.][^.]\.[^.][^.]$/ ? 3 : 2); 588 next if $ndots < ($cdom =~ /\.[^.][^.]\.[^.][^.]$/ ? 3 : 2);
528 } else { 589 } else {
529 $cdom = $uhost; 590 $cdom = $host;
530 } 591 }
531 592
532 # store it 593 # store it
533 $jar->{version} = 1; 594 $jar->{version} = 1;
534 $jar->{lc $cdom}{$cpath}{$name} = \%kv; 595 $jar->{lc $cdom}{$cpath}{$name} = \%kv;
535 596
536 redo if /\G\s*,/gc; 597 redo if /\G\s*,/gc;
537 } 598 }
538} 599}
539 600
601#############################################################################
602# keepalive/persistent connection cache
603
604# fetch a connection from the keepalive cache
605sub ka_fetch($) {
606 my $ka_key = shift;
607
608 my $hdl = pop @{ $KA_CACHE{$ka_key} }; # currently we reuse the MOST RECENTLY USED connection
609 delete $KA_CACHE{$ka_key}
610 unless @{ $KA_CACHE{$ka_key} };
611
612 $hdl
613}
614
615sub ka_store($$) {
616 my ($ka_key, $hdl) = @_;
617
618 my $kaa = $KA_CACHE{$ka_key} ||= [];
619
620 my $destroy = sub {
621 my @ka = grep $_ != $hdl, @{ $KA_CACHE{$ka_key} };
622
623 $hdl->destroy;
624
625 @ka
626 ? $KA_CACHE{$ka_key} = \@ka
627 : delete $KA_CACHE{$ka_key};
628 };
629
630 # on error etc., destroy
631 $hdl->on_error ($destroy);
632 $hdl->on_eof ($destroy);
633 $hdl->on_read ($destroy);
634 $hdl->timeout ($PERSISTENT_TIMEOUT);
635
636 push @$kaa, $hdl;
637 shift @$kaa while @$kaa > $MAX_PER_HOST;
638}
639
640#############################################################################
641# utilities
642
540# continue to parse $_ for headers and place them into the arg 643# continue to parse $_ for headers and place them into the arg
541sub parse_hdr() { 644sub _parse_hdr() {
542 my %hdr; 645 my %hdr;
543 646
544 # things seen, not parsed: 647 # things seen, not parsed:
545 # p3pP="NON CUR OTPi OUR NOR UNI" 648 # p3pP="NON CUR OTPi OUR NOR UNI"
546 649
560 for values %hdr; 663 for values %hdr;
561 664
562 \%hdr 665 \%hdr
563} 666}
564 667
668#############################################################################
669# http_get
670
565our $qr_nlnl = qr{(?<![^\012])\015?\012}; 671our $qr_nlnl = qr{(?<![^\012])\015?\012};
566 672
567our $TLS_CTX_LOW = { cache => 1, sslv2 => 1 }; 673our $TLS_CTX_LOW = { cache => 1, sslv2 => 1 };
568our $TLS_CTX_HIGH = { cache => 1, verify => 1, verify_peername => "https" }; 674our $TLS_CTX_HIGH = { cache => 1, verify => 1, verify_peername => "https" };
675
676# maybe it should just become a normal object :/
677
678sub _destroy_state(\%) {
679 my ($state) = @_;
680
681 $state->{handle}->destroy if $state->{handle};
682 %$state = ();
683}
684
685sub _error(\%$$) {
686 my ($state, $cb, $hdr) = @_;
687
688 &_destroy_state ($state);
689
690 $cb->(undef, $hdr);
691 ()
692}
569 693
570sub http_request($$@) { 694sub http_request($$@) {
571 my $cb = pop; 695 my $cb = pop;
572 my ($method, $url, %arg) = @_; 696 my ($method, $url, %arg) = @_;
573 697
591 my $recurse = exists $arg{recurse} ? delete $arg{recurse} : $MAX_RECURSE; 715 my $recurse = exists $arg{recurse} ? delete $arg{recurse} : $MAX_RECURSE;
592 716
593 return $cb->(undef, { @pseudo, Status => 599, Reason => "Too many redirections" }) 717 return $cb->(undef, { @pseudo, Status => 599, Reason => "Too many redirections" })
594 if $recurse < 0; 718 if $recurse < 0;
595 719
596 my $proxy = $arg{proxy} || $PROXY; 720 my $proxy = exists $arg{proxy} ? $arg{proxy} : $PROXY;
597 my $timeout = $arg{timeout} || $TIMEOUT; 721 my $timeout = $arg{timeout} || $TIMEOUT;
598 722
599 my ($uscheme, $uauthority, $upath, $query, $fragment) = 723 my ($uscheme, $uauthority, $upath, $query, undef) = # ignore fragment
600 $url =~ m|(?:([^:/?#]+):)?(?://([^/?#]*))?([^?#]*)(?:(\?[^#]*))?(?:#(.*))?|; 724 $url =~ m|^([^:]+):(?://([^/?#]*))?([^?#]*)(?:(\?[^#]*))?(?:#(.*))?$|;
601 725
602 $uscheme = lc $uscheme; 726 $uscheme = lc $uscheme;
603 727
604 my $uport = $uscheme eq "http" ? 80 728 my $uport = $uscheme eq "http" ? 80
605 : $uscheme eq "https" ? 443 729 : $uscheme eq "https" ? 443
652 if length $arg{body} || $method ne "GET"; 776 if length $arg{body} || $method ne "GET";
653 777
654 my $idempotent = $method =~ /^(?:GET|HEAD|PUT|DELETE|OPTIONS|TRACE)$/; 778 my $idempotent = $method =~ /^(?:GET|HEAD|PUT|DELETE|OPTIONS|TRACE)$/;
655 779
656 # default value for keepalive is true iff the request is for an idempotent method 780 # default value for keepalive is true iff the request is for an idempotent method
657 my $keepalive = exists $arg{keepalive} 781 my $persistent = exists $arg{persistent} ? !!$arg{persistent} : $idempotent;
658 ? $arg{keepalive}*1 782 my $keepalive = exists $arg{keepalive} ? !!$arg{keepalive} : !$proxy;
659 : $idempotent ? $PERSISTENT_TIMEOUT : 0; 783 my $was_persistent; # true if this is actually a recycled connection
660 784
785 # the key to use in the keepalive cache
786 my $ka_key = "$uscheme\x00$uhost\x00$uport\x00$arg{sessionid}";
787
661 $hdr{connection} = ($keepalive ? "" : "close ") . "Te"; #1.1 788 $hdr{connection} = ($persistent ? $keepalive ? "keep-alive " : "" : "close ") . "Te"; #1.1
662 $hdr{te} = "trailers" unless exists $hdr{te}; #1.1 789 $hdr{te} = "trailers" unless exists $hdr{te}; #1.1
663 790
664 my %state = (connect_guard => 1); 791 my %state = (connect_guard => 1);
665 792
666 my $ae_error = 595; # connecting 793 my $ae_error = 595; # connecting
667 794
668 # handle actual, non-tunneled, request 795 # handle actual, non-tunneled, request
669 my $handle_actual_request = sub { 796 my $handle_actual_request = sub {
670 $ae_error = 596; # request phase 797 $ae_error = 596; # request phase
671 798
799 my $hdl = $state{handle};
800
672 $state{handle}->starttls ("connect") if $uscheme eq "https" && !exists $state{handle}{tls}; 801 $hdl->starttls ("connect") if $uscheme eq "https" && !exists $hdl->{tls};
673 802
674 # send request 803 # send request
675 $state{handle}->push_write ( 804 $hdl->push_write (
676 "$method $rpath HTTP/1.1\015\012" 805 "$method $rpath HTTP/1.1\015\012"
677 . (join "", map "\u$_: $hdr{$_}\015\012", grep defined $hdr{$_}, keys %hdr) 806 . (join "", map "\u$_: $hdr{$_}\015\012", grep defined $hdr{$_}, keys %hdr)
678 . "\015\012" 807 . "\015\012"
679 . (delete $arg{body}) 808 . (delete $arg{body})
680 ); 809 );
685 # reduce memory usage, save a kitten, also re-use it for the response headers. 814 # reduce memory usage, save a kitten, also re-use it for the response headers.
686 %hdr = (); 815 %hdr = ();
687 816
688 # status line and headers 817 # status line and headers
689 $state{read_response} = sub { 818 $state{read_response} = sub {
819 return unless %state;
820
690 for ("$_[1]") { 821 for ("$_[1]") {
691 y/\015//d; # weed out any \015, as they show up in the weirdest of places. 822 y/\015//d; # weed out any \015, as they show up in the weirdest of places.
692 823
693 /^HTTP\/0*([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\012]*) )? \012/gxci 824 /^HTTP\/0*([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\012]*) )? \012/gxci
694 or return (%state = (), $cb->(undef, { @pseudo, Status => 599, Reason => "Invalid server response" })); 825 or return _error %state, $cb, { @pseudo, Status => 599, Reason => "Invalid server response" };
695 826
696 # 100 Continue handling 827 # 100 Continue handling
697 # should not happen as we don't send expect: 100-continue, 828 # should not happen as we don't send expect: 100-continue,
698 # but we handle it just in case. 829 # but we handle it just in case.
699 # since we send the request body regardless, if we get an error 830 # since we send the request body regardless, if we get an error
705 HTTPVersion => $1, 836 HTTPVersion => $1,
706 Status => $2, 837 Status => $2,
707 Reason => $3, 838 Reason => $3,
708 ; 839 ;
709 840
710 my $hdr = parse_hdr 841 my $hdr = _parse_hdr
711 or return (%state = (), $cb->(undef, { @pseudo, Status => 599, Reason => "Garbled response headers" })); 842 or return _error %state, $cb, { @pseudo, Status => 599, Reason => "Garbled response headers" };
712 843
713 %hdr = (%$hdr, @pseudo); 844 %hdr = (%$hdr, @pseudo);
714 } 845 }
715 846
716 # redirect handling 847 # redirect handling
746 } elsif ($status == 307) { 877 } elsif ($status == 307) {
747 $redirect = 1; 878 $redirect = 1;
748 } 879 }
749 } 880 }
750 881
751 my $finish = sub { # ($data, $err_status, $err_reason[, $keepalive]) 882 my $finish = sub { # ($data, $err_status, $err_reason[, $persistent])
752 my $may_keep_alive = $_[3]; 883 if ($state{handle}) {
753 884 # handle keepalive
885 if (
886 $persistent
887 && $_[3]
888 && ($hdr{HTTPVersion} < 1.1
889 ? $hdr{connection} =~ /\bkeep-?alive\b/i
890 : $hdr{connection} !~ /\bclose\b/i)
891 ) {
892 ka_store $ka_key, delete $state{handle};
893 } else {
894 # no keepalive, destroy the handle
754 $state{handle}->destroy if $state{handle}; 895 $state{handle}->destroy;
896 }
897 }
898
755 %state = (); 899 %state = ();
756 900
757 if (defined $_[1]) { 901 if (defined $_[1]) {
758 $hdr{OrigStatus} = $hdr{Status}; $hdr{Status} = $_[1]; 902 $hdr{OrigStatus} = $hdr{Status}; $hdr{Status} = $_[1];
759 $hdr{OrigReason} = $hdr{Reason}; $hdr{Reason} = $_[2]; 903 $hdr{OrigReason} = $hdr{Reason}; $hdr{Reason} = $_[2];
766 910
767 if ($redirect && exists $hdr{location}) { 911 if ($redirect && exists $hdr{location}) {
768 # we ignore any errors, as it is very common to receive 912 # we ignore any errors, as it is very common to receive
769 # Content-Length != 0 but no actual body 913 # Content-Length != 0 but no actual body
770 # we also access %hdr, as $_[1] might be an erro 914 # we also access %hdr, as $_[1] might be an erro
915 $state{recurse} =
771 http_request ( 916 http_request (
772 $method => $hdr{location}, 917 $method => $hdr{location},
773 %arg, 918 %arg,
774 recurse => $recurse - 1, 919 recurse => $recurse - 1,
775 Redirect => [$_[0], \%hdr], 920 Redirect => [$_[0], \%hdr],
921 sub {
922 %state = ();
923 &$cb
924 },
776 $cb); 925 );
777 } else { 926 } else {
778 $cb->($_[0], \%hdr); 927 $cb->($_[0], \%hdr);
779 } 928 }
780 }; 929 };
781 930
782 $ae_error = 597; # body phase 931 $ae_error = 597; # body phase
783 932
933 my $chunked = $hdr{"transfer-encoding"} =~ /\bchunked\b/i; # not quite correct...
934
784 my $len = $hdr{"content-length"}; 935 my $len = $chunked ? undef : $hdr{"content-length"};
785 936
786 # body handling, many different code paths 937 # body handling, many different code paths
787 # - no body expected 938 # - no body expected
788 # - want_body_handle 939 # - want_body_handle
789 # - te chunked 940 # - te chunked
804 $_[0]->on_error (undef); 955 $_[0]->on_error (undef);
805 $_[0]->on_read (undef); 956 $_[0]->on_read (undef);
806 957
807 $finish->(delete $state{handle}); 958 $finish->(delete $state{handle});
808 959
809 } elsif ($hdr{"transfer-encoding"} =~ /\bchunked\b/i) { 960 } elsif ($chunked) {
810 my $cl = 0; 961 my $cl = 0;
811 my $body = undef; 962 my $body = "";
812 my $on_body = $arg{on_body} || sub { $body .= shift; 1 }; 963 my $on_body = $arg{on_body} || sub { $body .= shift; 1 };
813 964
814 $state{read_chunk} = sub { 965 $state{read_chunk} = sub {
815 $_[1] =~ /^([0-9a-fA-F]+)/ 966 $_[1] =~ /^([0-9a-fA-F]+)/
816 or $finish->(undef, $ae_error => "Garbled chunked transfer encoding"); 967 or $finish->(undef, $ae_error => "Garbled chunked transfer encoding");
836 $_[0]->push_read (line => $qr_nlnl, sub { 987 $_[0]->push_read (line => $qr_nlnl, sub {
837 if (length $_[1]) { 988 if (length $_[1]) {
838 for ("$_[1]") { 989 for ("$_[1]") {
839 y/\015//d; # weed out any \015, as they show up in the weirdest of places. 990 y/\015//d; # weed out any \015, as they show up in the weirdest of places.
840 991
841 my $hdr = parse_hdr 992 my $hdr = _parse_hdr
842 or return $finish->(undef, $ae_error => "Garbled response trailers"); 993 or return $finish->(undef, $ae_error => "Garbled response trailers");
843 994
844 %hdr = (%hdr, %$hdr); 995 %hdr = (%hdr, %$hdr);
845 } 996 }
846 } 997 }
889 $_[0]->on_read (sub { }); 1040 $_[0]->on_read (sub { });
890 } 1041 }
891 } 1042 }
892 }; 1043 };
893 1044
1045 # if keepalive is enabled, then the server closing the connection
1046 # before a response can happen legally - we retry on idempotent methods.
1047 if ($was_persistent && $idempotent) {
1048 my $old_eof = $hdl->{on_eof};
1049 $hdl->{on_eof} = sub {
1050 _destroy_state %state;
1051
1052 %state = ();
1053 $state{recurse} =
1054 http_request (
1055 $method => $url,
1056 %arg,
1057 keepalive => 0,
1058 sub {
1059 %state = ();
1060 &$cb
1061 }
1062 );
1063 };
1064 $hdl->on_read (sub {
1065 return unless %state;
1066
1067 # as soon as we receive something, a connection close
1068 # once more becomes a hard error
1069 $hdl->{on_eof} = $old_eof;
1070 $hdl->push_read (line => $qr_nlnl, $state{read_response});
1071 });
1072 } else {
894 $state{handle}->push_read (line => $qr_nlnl, $state{read_response}); 1073 $hdl->push_read (line => $qr_nlnl, $state{read_response});
1074 }
895 }; 1075 };
896 1076
1077 my $prepare_handle = sub {
1078 my ($hdl) = $state{handle};
1079
1080 $hdl->on_error (sub {
1081 _error %state, $cb, { @pseudo, Status => $ae_error, Reason => $_[2] };
1082 });
1083 $hdl->on_eof (sub {
1084 _error %state, $cb, { @pseudo, Status => $ae_error, Reason => "Unexpected end-of-file" };
1085 });
1086 $hdl->timeout_reset;
1087 $hdl->timeout ($timeout);
1088 };
1089
1090 # connected to proxy (or origin server)
897 my $connect_cb = sub { 1091 my $connect_cb = sub {
898 $state{fh} = shift 1092 my $fh = shift
899 or do {
900 my $err = "$!";
901 %state = ();
902 return $cb->(undef, { @pseudo, Status => $ae_error, Reason => $err }); 1093 or return _error %state, $cb, { @pseudo, Status => $ae_error, Reason => "$!" };
903 };
904 1094
905 return unless delete $state{connect_guard}; 1095 return unless delete $state{connect_guard};
906 1096
907 # get handle 1097 # get handle
908 $state{handle} = new AnyEvent::Handle 1098 $state{handle} = new AnyEvent::Handle
1099 %{ $arg{handle_params} },
909 fh => $state{fh}, 1100 fh => $fh,
910 peername => $rhost, 1101 peername => $uhost,
911 tls_ctx => $arg{tls_ctx}, 1102 tls_ctx => $arg{tls_ctx},
912 # these need to be reconfigured on keepalive handles
913 timeout => $timeout,
914 on_error => sub {
915 %state = ();
916 $cb->(undef, { @pseudo, Status => $ae_error, Reason => $_[2] });
917 },
918 on_eof => sub {
919 %state = ();
920 $cb->(undef, { @pseudo, Status => $ae_error, Reason => "Unexpected end-of-file" });
921 },
922 ; 1103 ;
923 1104
924 # limit the number of persistent connections 1105 $prepare_handle->();
925 # keepalive not yet supported
926# if ($KA_COUNT{$_[1]} < $MAX_PERSISTENT_PER_HOST) {
927# ++$KA_COUNT{$_[1]};
928# $state{handle}{ka_count_guard} = AnyEvent::Util::guard {
929# --$KA_COUNT{$_[1]}
930# };
931# $hdr{connection} = "keep-alive";
932# }
933 1106
934 $state{handle}->starttls ("connect") if $rscheme eq "https"; 1107 #$state{handle}->starttls ("connect") if $rscheme eq "https";
935 1108
936 # now handle proxy-CONNECT method 1109 # now handle proxy-CONNECT method
937 if ($proxy && $uscheme eq "https") { 1110 if ($proxy && $uscheme eq "https") {
938 # oh dear, we have to wrap it into a connect request 1111 # oh dear, we have to wrap it into a connect request
939 1112
940 # maybe re-use $uauthority with patched port? 1113 # maybe re-use $uauthority with patched port?
941 $state{handle}->push_write ("CONNECT $uhost:$uport HTTP/1.0\015\012\015\012"); 1114 $state{handle}->push_write ("CONNECT $uhost:$uport HTTP/1.0\015\012\015\012");
942 $state{handle}->push_read (line => $qr_nlnl, sub { 1115 $state{handle}->push_read (line => $qr_nlnl, sub {
943 $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix 1116 $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix
944 or return (%state = (), $cb->(undef, { @pseudo, Status => 599, Reason => "Invalid proxy connect response ($_[1])" })); 1117 or return _error %state, $cb, { @pseudo, Status => 599, Reason => "Invalid proxy connect response ($_[1])" };
945 1118
946 if ($2 == 200) { 1119 if ($2 == 200) {
947 $rpath = $upath; 1120 $rpath = $upath;
948 $handle_actual_request->(); 1121 $handle_actual_request->();
949 } else { 1122 } else {
950 %state = ();
951 $cb->(undef, { @pseudo, Status => $2, Reason => $3 }); 1123 _error %state, $cb, { @pseudo, Status => $2, Reason => $3 };
952 } 1124 }
953 }); 1125 });
954 } else { 1126 } else {
955 $handle_actual_request->(); 1127 $handle_actual_request->();
956 } 1128 }
959 _get_slot $uhost, sub { 1131 _get_slot $uhost, sub {
960 $state{slot_guard} = shift; 1132 $state{slot_guard} = shift;
961 1133
962 return unless $state{connect_guard}; 1134 return unless $state{connect_guard};
963 1135
1136 # try to use an existing keepalive connection, but only if we, ourselves, plan
1137 # on a keepalive request (in theory, this should be a separate config option).
1138 if ($persistent && $KA_CACHE{$ka_key}) {
1139 $was_persistent = 1;
1140
1141 $state{handle} = ka_fetch $ka_key;
1142 $state{handle}->destroyed
1143 and die "got a destructed habndle. pah\n";#d#
1144 $prepare_handle->();
1145 $state{handle}->destroyed
1146 and die "got a destructed habndle. pa2\n";#d#
1147 $handle_actual_request->();
1148 $state{handle}->destroyed
1149 and die "got a destructed habndle. pa3\n";#d#
1150
1151 } else {
964 my $tcp_connect = $arg{tcp_connect} 1152 my $tcp_connect = $arg{tcp_connect}
965 || do { require AnyEvent::Socket; \&AnyEvent::Socket::tcp_connect }; 1153 || do { require AnyEvent::Socket; \&AnyEvent::Socket::tcp_connect };
966 1154
967 $state{connect_guard} = $tcp_connect->($rhost, $rport, $connect_cb, $arg{on_prepare} || sub { $timeout }); 1155 $state{connect_guard} = $tcp_connect->($rhost, $rport, $connect_cb, $arg{on_prepare} || sub { $timeout });
1156 }
968 }; 1157 };
969 1158
970 defined wantarray && AnyEvent::Util::guard { %state = () } 1159 defined wantarray && AnyEvent::Util::guard { _destroy_state %state }
971} 1160}
972 1161
973sub http_get($@) { 1162sub http_get($@) {
974 unshift @_, "GET"; 1163 unshift @_, "GET";
975 &http_request 1164 &http_request
993AnyEvent::HTTP uses the AnyEvent::Socket::tcp_connect function for 1182AnyEvent::HTTP uses the AnyEvent::Socket::tcp_connect function for
994the actual connection, which in turn uses AnyEvent::DNS to resolve 1183the actual connection, which in turn uses AnyEvent::DNS to resolve
995hostnames. The latter is a simple stub resolver and does no caching 1184hostnames. The latter is a simple stub resolver and does no caching
996on its own. If you want DNS caching, you currently have to provide 1185on its own. If you want DNS caching, you currently have to provide
997your own default resolver (by storing a suitable resolver object in 1186your own default resolver (by storing a suitable resolver object in
998C<$AnyEvent::DNS::RESOLVER>). 1187C<$AnyEvent::DNS::RESOLVER>) or your own C<tcp_connect> callback.
999 1188
1000=head2 GLOBAL FUNCTIONS AND VARIABLES 1189=head2 GLOBAL FUNCTIONS AND VARIABLES
1001 1190
1002=over 4 1191=over 4
1003 1192
1004=item AnyEvent::HTTP::set_proxy "proxy-url" 1193=item AnyEvent::HTTP::set_proxy "proxy-url"
1005 1194
1006Sets the default proxy server to use. The proxy-url must begin with a 1195Sets the default proxy server to use. The proxy-url must begin with a
1007string of the form C<http://host:port> (optionally C<https:...>), croaks 1196string of the form C<http://host:port>, croaks otherwise.
1008otherwise.
1009 1197
1010To clear an already-set proxy, use C<undef>. 1198To clear an already-set proxy, use C<undef>.
1199
1200When AnyEvent::HTTP is laoded for the first time it will query the
1201default proxy from the operating system, currently by looking at
1202C<$ENV{http_proxy>}.
1011 1203
1012=item AnyEvent::HTTP::cookie_jar_expire $jar[, $session_end] 1204=item AnyEvent::HTTP::cookie_jar_expire $jar[, $session_end]
1013 1205
1014Remove all cookies from the cookie jar that have been expired. If 1206Remove all cookies from the cookie jar that have been expired. If
1015C<$session_end> is given and true, then additionally remove all session 1207C<$session_end> is given and true, then additionally remove all session
1061 1253
1062=item $AnyEvent::HTTP::MAX_RECURSE 1254=item $AnyEvent::HTTP::MAX_RECURSE
1063 1255
1064The default value for the C<recurse> request parameter (default: C<10>). 1256The default value for the C<recurse> request parameter (default: C<10>).
1065 1257
1258=item $AnyEvent::HTTP::TIMEOUT
1259
1260The default timeout for conenction operations (default: C<300>).
1261
1066=item $AnyEvent::HTTP::USERAGENT 1262=item $AnyEvent::HTTP::USERAGENT
1067 1263
1068The default value for the C<User-Agent> header (the default is 1264The default value for the C<User-Agent> header (the default is
1069C<Mozilla/5.0 (compatible; U; AnyEvent-HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)>). 1265C<Mozilla/5.0 (compatible; U; AnyEvent-HTTP/$VERSION; +http://software.schmorp.de/pkg/AnyEvent)>).
1070 1266
1071=item $AnyEvent::HTTP::MAX_PER_HOST 1267=item $AnyEvent::HTTP::MAX_PER_HOST
1072 1268
1073The maximum number of concurrent connections to the same host (identified 1269The maximum number of concurrent connections to the same host (identified
1074by the hostname). If the limit is exceeded, then the additional requests 1270by the hostname). If the limit is exceeded, then the additional requests
1075are queued until previous connections are closed. 1271are queued until previous connections are closed. Both persistent and
1272non-persistent connections are counted in this limit.
1076 1273
1077The default value for this is C<4>, and it is highly advisable to not 1274The default value for this is C<4>, and it is highly advisable to not
1078increase it. 1275increase it much.
1276
1277For comparison: the RFC's recommend 4 non-persistent or 2 persistent
1278connections, older browsers used 2, newers (such as firefox 3) typically
1279use 6, and Opera uses 8 because like, they have the fastest browser and
1280give a shit for everybody else on the planet.
1281
1282=item $AnyEvent::HTTP::PERSISTENT_TIMEOUT
1283
1284The time after which idle persistent conenctions get closed by
1285AnyEvent::HTTP (default: C<3>).
1079 1286
1080=item $AnyEvent::HTTP::ACTIVE 1287=item $AnyEvent::HTTP::ACTIVE
1081 1288
1082The number of active connections. This is not the number of currently 1289The number of active connections. This is not the number of currently
1083running requests, but the number of currently open and non-idle TCP 1290running requests, but the number of currently open and non-idle TCP
1084connections. This number of can be useful for load-leveling. 1291connections. This number can be useful for load-leveling.
1085 1292
1086=back 1293=back
1087 1294
1088=cut 1295=cut
1089 1296
1132 undef 1339 undef
1133} 1340}
1134 1341
1135sub set_proxy($) { 1342sub set_proxy($) {
1136 if (length $_[0]) { 1343 if (length $_[0]) {
1137 $_[0] =~ m%^(https?):// ([^:/]+) (?: : (\d*) )?%ix 1344 $_[0] =~ m%^(http):// ([^:/]+) (?: : (\d*) )?%ix
1138 or Carp::croak "$_[0]: invalid proxy URL"; 1345 or Carp::croak "$_[0]: invalid proxy URL";
1139 $PROXY = [$2, $3 || 3128, $1] 1346 $PROXY = [$2, $3 || 3128, $1]
1140 } else { 1347 } else {
1141 undef $PROXY; 1348 undef $PROXY;
1142 } 1349 }
1145# initialise proxy from environment 1352# initialise proxy from environment
1146eval { 1353eval {
1147 set_proxy $ENV{http_proxy}; 1354 set_proxy $ENV{http_proxy};
1148}; 1355};
1149 1356
1357=head2 SHOWCASE
1358
1359This section contaisn some more elaborate "real-world" examples or code
1360snippets.
1361
1362=head2 HTTP/1.1 FILE DOWNLOAD
1363
1364Downloading files with HTTP can be quite tricky, especially when something
1365goes wrong and you want to resume.
1366
1367Here is a function that initiates and resumes a download. It uses the
1368last modified time to check for file content changes, and works with many
1369HTTP/1.0 servers as well, and usually falls back to a complete re-download
1370on older servers.
1371
1372It calls the completion callback with either C<undef>, which means a
1373nonretryable error occured, C<0> when the download was partial and should
1374be retried, and C<1> if it was successful.
1375
1376 use AnyEvent::HTTP;
1377
1378 sub download($$$) {
1379 my ($url, $file, $cb) = @_;
1380
1381 open my $fh, "+<", $file
1382 or die "$file: $!";
1383
1384 my %hdr;
1385 my $ofs = 0;
1386
1387 warn stat $fh;
1388 warn -s _;
1389 if (stat $fh and -s _) {
1390 $ofs = -s _;
1391 warn "-s is ", $ofs;#d#
1392 $hdr{"if-unmodified-since"} = AnyEvent::HTTP::format_date +(stat _)[9];
1393 $hdr{"range"} = "bytes=$ofs-";
1394 }
1395
1396 http_get $url,
1397 headers => \%hdr,
1398 on_header => sub {
1399 my ($hdr) = @_;
1400
1401 if ($hdr->{Status} == 200 && $ofs) {
1402 # resume failed
1403 truncate $fh, $ofs = 0;
1404 }
1405
1406 sysseek $fh, $ofs, 0;
1407
1408 1
1409 },
1410 on_body => sub {
1411 my ($data, $hdr) = @_;
1412
1413 if ($hdr->{Status} =~ /^2/) {
1414 length $data == syswrite $fh, $data
1415 or return; # abort on write errors
1416 }
1417
1418 1
1419 },
1420 sub {
1421 my (undef, $hdr) = @_;
1422
1423 my $status = $hdr->{Status};
1424
1425 if (my $time = AnyEvent::HTTP::parse_date $hdr->{"last-modified"}) {
1426 utime $fh, $time, $time;
1427 }
1428
1429 if ($status == 200 || $status == 206 || $status == 416) {
1430 # download ok || resume ok || file already fully downloaded
1431 $cb->(1, $hdr);
1432
1433 } elsif ($status == 412) {
1434 # file has changed while resuming, delete and retry
1435 unlink $file;
1436 $cb->(0, $hdr);
1437
1438 } elsif ($status == 500 or $status == 503 or $status =~ /^59/) {
1439 # retry later
1440 $cb->(0, $hdr);
1441
1442 } else {
1443 $cb->(undef, $hdr);
1444 }
1445 }
1446 ;
1447 }
1448
1449 download "http://server/somelargefile", "/tmp/somelargefile", sub {
1450 if ($_[0]) {
1451 print "OK!\n";
1452 } elsif (defined $_[0]) {
1453 print "please retry later\n";
1454 } else {
1455 print "ERROR\n";
1456 }
1457 };
1458
1150=head2 SOCKS PROXIES 1459=head3 SOCKS PROXIES
1151 1460
1152Socks proxies are not directly supported by AnyEvent::HTTP. You can 1461Socks proxies are not directly supported by AnyEvent::HTTP. You can
1153compile your perl to support socks, or use an external program such as 1462compile your perl to support socks, or use an external program such as
1154F<socksify> (dante) or F<tsocks> to make your program use a socks proxy 1463F<socksify> (dante) or F<tsocks> to make your program use a socks proxy
1155transparently. 1464transparently.

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines