… | |
… | |
57 | our $MAX_PERSISTENT = 8; |
57 | our $MAX_PERSISTENT = 8; |
58 | our $PERSISTENT_TIMEOUT = 2; |
58 | our $PERSISTENT_TIMEOUT = 2; |
59 | our $TIMEOUT = 300; |
59 | our $TIMEOUT = 300; |
60 | |
60 | |
61 | # changing these is evil |
61 | # changing these is evil |
62 | our $MAX_PERSISTENT_PER_HOST = 0; |
62 | our $MAX_PERSISTENT_PER_HOST = 2; |
63 | our $MAX_PER_HOST = 4; |
63 | our $MAX_PER_HOST = 4; |
64 | |
64 | |
65 | our $PROXY; |
65 | our $PROXY; |
66 | our $ACTIVE = 0; |
66 | our $ACTIVE = 0; |
67 | |
67 | |
… | |
… | |
414 | } |
414 | } |
415 | |
415 | |
416 | # extract cookies from jar |
416 | # extract cookies from jar |
417 | sub cookie_jar_extract($$$$) { |
417 | sub cookie_jar_extract($$$$) { |
418 | my ($jar, $uscheme, $uhost, $upath) = @_; |
418 | my ($jar, $uscheme, $uhost, $upath) = @_; |
419 | |
|
|
420 | $uhost = lc $uhost; |
|
|
421 | |
419 | |
422 | %$jar = () if $jar->{version} != 1; |
420 | %$jar = () if $jar->{version} != 1; |
423 | |
421 | |
424 | my @cookies; |
422 | my @cookies; |
425 | |
423 | |
… | |
… | |
608 | : return $cb->(undef, { @pseudo, Status => 599, Reason => "Only http and https URL schemes supported" }); |
606 | : return $cb->(undef, { @pseudo, Status => 599, Reason => "Only http and https URL schemes supported" }); |
609 | |
607 | |
610 | $uauthority =~ /^(?: .*\@ )? ([^\@:]+) (?: : (\d+) )?$/x |
608 | $uauthority =~ /^(?: .*\@ )? ([^\@:]+) (?: : (\d+) )?$/x |
611 | or return $cb->(undef, { @pseudo, Status => 599, Reason => "Unparsable URL" }); |
609 | or return $cb->(undef, { @pseudo, Status => 599, Reason => "Unparsable URL" }); |
612 | |
610 | |
613 | my $uhost = $1; |
611 | my $uhost = lc $1; |
614 | $uport = $2 if defined $2; |
612 | $uport = $2 if defined $2; |
615 | |
613 | |
616 | $hdr{host} = defined $2 ? "$uhost:$2" : "$uhost" |
614 | $hdr{host} = defined $2 ? "$uhost:$2" : "$uhost" |
617 | unless exists $hdr{host}; |
615 | unless exists $hdr{host}; |
618 | |
616 | |
… | |
… | |
637 | $rscheme = "http" unless defined $rscheme; |
635 | $rscheme = "http" unless defined $rscheme; |
638 | |
636 | |
639 | # don't support https requests over https-proxy transport, |
637 | # don't support https requests over https-proxy transport, |
640 | # can't be done with tls as spec'ed, unless you double-encrypt. |
638 | # can't be done with tls as spec'ed, unless you double-encrypt. |
641 | $rscheme = "http" if $uscheme eq "https" && $rscheme eq "https"; |
639 | $rscheme = "http" if $uscheme eq "https" && $rscheme eq "https"; |
|
|
640 | |
|
|
641 | $rhost = lc $rhost; |
|
|
642 | $rscheme = lc $rscheme; |
642 | } else { |
643 | } else { |
643 | ($rhost, $rport, $rscheme, $rpath) = ($uhost, $uport, $uscheme, $upath); |
644 | ($rhost, $rport, $rscheme, $rpath) = ($uhost, $uport, $uscheme, $upath); |
644 | } |
645 | } |
645 | |
646 | |
646 | # leave out fragment and query string, just a heuristic |
647 | # leave out fragment and query string, just a heuristic |
… | |
… | |
648 | $hdr{"user-agent"} = $USERAGENT unless exists $hdr{"user-agent"}; |
649 | $hdr{"user-agent"} = $USERAGENT unless exists $hdr{"user-agent"}; |
649 | |
650 | |
650 | $hdr{"content-length"} = length $arg{body} |
651 | $hdr{"content-length"} = length $arg{body} |
651 | if length $arg{body} || $method ne "GET"; |
652 | if length $arg{body} || $method ne "GET"; |
652 | |
653 | |
653 | $hdr{connection} = "close TE"; #1.1 |
654 | my $idempotent = $method =~ /^(?:GET|HEAD|PUT|DELETE|OPTIONS|TRACE)$/; |
|
|
655 | |
|
|
656 | # default value for keepalive is true iff the request is for an idempotent method |
|
|
657 | my $keepalive = exists $arg{keepalive} |
|
|
658 | ? $arg{keepalive}*1 |
|
|
659 | : $idempotent ? $PERSISTENT_TIMEOUT : 0; |
|
|
660 | |
|
|
661 | $hdr{connection} = ($keepalive ? "" : "close ") . "Te"; #1.1 |
654 | $hdr{te} = "trailers" unless exists $hdr{te}; #1.1 |
662 | $hdr{te} = "trailers" unless exists $hdr{te}; #1.1 |
655 | |
663 | |
656 | my %state = (connect_guard => 1); |
664 | my %state = (connect_guard => 1); |
657 | |
665 | |
658 | _get_slot $uhost, sub { |
|
|
659 | $state{slot_guard} = shift; |
|
|
660 | |
|
|
661 | return unless $state{connect_guard}; |
|
|
662 | |
|
|
663 | my $ae_error = 595; # connecting |
666 | my $ae_error = 595; # connecting |
664 | |
667 | |
665 | # handle actual, non-tunneled, request |
668 | # handle actual, non-tunneled, request |
666 | my $handle_actual_request = sub { |
669 | my $handle_actual_request = sub { |
667 | $ae_error = 596; # request phase |
670 | $ae_error = 596; # request phase |
668 | |
671 | |
669 | $state{handle}->starttls ("connect") if $uscheme eq "https" && !exists $state{handle}{tls}; |
672 | $state{handle}->starttls ("connect") if $uscheme eq "https" && !exists $state{handle}{tls}; |
670 | |
673 | |
671 | # send request |
674 | # send request |
672 | $state{handle}->push_write ( |
675 | $state{handle}->push_write ( |
673 | "$method $rpath HTTP/1.1\015\012" |
676 | "$method $rpath HTTP/1.1\015\012" |
674 | . (join "", map "\u$_: $hdr{$_}\015\012", grep defined $hdr{$_}, keys %hdr) |
677 | . (join "", map "\u$_: $hdr{$_}\015\012", grep defined $hdr{$_}, keys %hdr) |
675 | . "\015\012" |
678 | . "\015\012" |
676 | . (delete $arg{body}) |
679 | . (delete $arg{body}) |
677 | ); |
680 | ); |
678 | |
681 | |
679 | # return if error occured during push_write() |
682 | # return if error occured during push_write() |
680 | return unless %state; |
683 | return unless %state; |
681 | |
684 | |
682 | %hdr = (); # reduce memory usage, save a kitten, also make it possible to re-use |
685 | # reduce memory usage, save a kitten, also re-use it for the response headers. |
|
|
686 | %hdr = (); |
683 | |
687 | |
684 | # status line and headers |
688 | # status line and headers |
685 | $state{read_response} = sub { |
689 | $state{read_response} = sub { |
686 | for ("$_[1]") { |
690 | for ("$_[1]") { |
687 | y/\015//d; # weed out any \015, as they show up in the weirdest of places. |
691 | y/\015//d; # weed out any \015, as they show up in the weirdest of places. |
688 | |
692 | |
689 | /^HTTP\/0*([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\012]*) )? \012/gxci |
693 | /^HTTP\/0*([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\012]*) )? \012/gxci |
690 | or return (%state = (), $cb->(undef, { @pseudo, Status => 599, Reason => "Invalid server response" })); |
694 | or return (%state = (), $cb->(undef, { @pseudo, Status => 599, Reason => "Invalid server response" })); |
691 | |
695 | |
692 | # 100 Continue handling |
696 | # 100 Continue handling |
693 | # should not happen as we don't send expect: 100-continue, |
697 | # should not happen as we don't send expect: 100-continue, |
694 | # but we handle it just in case. |
698 | # but we handle it just in case. |
695 | # since we send the request body regardless, if we get an error |
699 | # since we send the request body regardless, if we get an error |
696 | # we are out of-sync, which we currently do NOT handle correctly. |
700 | # we are out of-sync, which we currently do NOT handle correctly. |
697 | return $state{handle}->push_read (line => $qr_nlnl, $state{read_response}) |
701 | return $state{handle}->push_read (line => $qr_nlnl, $state{read_response}) |
698 | if $2 eq 100; |
702 | if $2 eq 100; |
699 | |
703 | |
700 | push @pseudo, |
704 | push @pseudo, |
701 | HTTPVersion => $1, |
705 | HTTPVersion => $1, |
702 | Status => $2, |
706 | Status => $2, |
703 | Reason => $3, |
707 | Reason => $3, |
704 | ; |
708 | ; |
705 | |
709 | |
706 | my $hdr = parse_hdr |
710 | my $hdr = parse_hdr |
707 | or return (%state = (), $cb->(undef, { @pseudo, Status => 599, Reason => "Garbled response headers" })); |
711 | or return (%state = (), $cb->(undef, { @pseudo, Status => 599, Reason => "Garbled response headers" })); |
708 | |
712 | |
709 | %hdr = (%$hdr, @pseudo); |
713 | %hdr = (%$hdr, @pseudo); |
|
|
714 | } |
|
|
715 | |
|
|
716 | # redirect handling |
|
|
717 | # microsoft and other shitheads don't give a shit for following standards, |
|
|
718 | # try to support some common forms of broken Location headers. |
|
|
719 | if ($hdr{location} !~ /^(?: $ | [^:\/?\#]+ : )/x) { |
|
|
720 | $hdr{location} =~ s/^\.\/+//; |
|
|
721 | |
|
|
722 | my $url = "$rscheme://$uhost:$uport"; |
|
|
723 | |
|
|
724 | unless ($hdr{location} =~ s/^\///) { |
|
|
725 | $url .= $upath; |
|
|
726 | $url =~ s/\/[^\/]*$//; |
710 | } |
727 | } |
711 | |
728 | |
712 | # redirect handling |
|
|
713 | # microsoft and other shitheads don't give a shit for following standards, |
|
|
714 | # try to support some common forms of broken Location headers. |
|
|
715 | if ($hdr{location} !~ /^(?: $ | [^:\/?\#]+ : )/x) { |
|
|
716 | $hdr{location} =~ s/^\.\/+//; |
|
|
717 | |
|
|
718 | my $url = "$rscheme://$uhost:$uport"; |
|
|
719 | |
|
|
720 | unless ($hdr{location} =~ s/^\///) { |
|
|
721 | $url .= $upath; |
|
|
722 | $url =~ s/\/[^\/]*$//; |
|
|
723 | } |
|
|
724 | |
|
|
725 | $hdr{location} = "$url/$hdr{location}"; |
729 | $hdr{location} = "$url/$hdr{location}"; |
|
|
730 | } |
|
|
731 | |
|
|
732 | my $redirect; |
|
|
733 | |
|
|
734 | if ($recurse) { |
|
|
735 | my $status = $hdr{Status}; |
|
|
736 | |
|
|
737 | # industry standard is to redirect POST as GET for |
|
|
738 | # 301, 302 and 303, in contrast to HTTP/1.0 and 1.1. |
|
|
739 | # also, the UA should ask the user for 301 and 307 and POST, |
|
|
740 | # industry standard seems to be to simply follow. |
|
|
741 | # we go with the industry standard. |
|
|
742 | if ($status == 301 or $status == 302 or $status == 303) { |
|
|
743 | # HTTP/1.1 is unclear on how to mutate the method |
|
|
744 | $method = "GET" unless $method eq "HEAD"; |
|
|
745 | $redirect = 1; |
|
|
746 | } elsif ($status == 307) { |
|
|
747 | $redirect = 1; |
726 | } |
748 | } |
727 | |
|
|
728 | my $redirect; |
|
|
729 | |
|
|
730 | if ($recurse) { |
|
|
731 | my $status = $hdr{Status}; |
|
|
732 | |
|
|
733 | # industry standard is to redirect POST as GET for |
|
|
734 | # 301, 302 and 303, in contrast to HTTP/1.0 and 1.1. |
|
|
735 | # also, the UA should ask the user for 301 and 307 and POST, |
|
|
736 | # industry standard seems to be to simply follow. |
|
|
737 | # we go with the industry standard. |
|
|
738 | if ($status == 301 or $status == 302 or $status == 303) { |
|
|
739 | # HTTP/1.1 is unclear on how to mutate the method |
|
|
740 | $method = "GET" unless $method eq "HEAD"; |
|
|
741 | $redirect = 1; |
|
|
742 | } elsif ($status == 307) { |
|
|
743 | $redirect = 1; |
|
|
744 | } |
749 | } |
|
|
750 | |
|
|
751 | my $finish = sub { # ($data, $err_status, $err_reason[, $keepalive]) |
|
|
752 | my $may_keep_alive = $_[3]; |
|
|
753 | |
|
|
754 | $state{handle}->destroy if $state{handle}; |
|
|
755 | %state = (); |
|
|
756 | |
|
|
757 | if (defined $_[1]) { |
|
|
758 | $hdr{OrigStatus} = $hdr{Status}; $hdr{Status} = $_[1]; |
|
|
759 | $hdr{OrigReason} = $hdr{Reason}; $hdr{Reason} = $_[2]; |
745 | } |
760 | } |
746 | |
761 | |
747 | my $finish = sub { # ($data, $err_status, $err_reason[, $keepalive]) |
|
|
748 | my $may_keep_alive = $_[3]; |
|
|
749 | |
|
|
750 | $state{handle}->destroy if $state{handle}; |
|
|
751 | %state = (); |
|
|
752 | |
|
|
753 | if (defined $_[1]) { |
|
|
754 | $hdr{OrigStatus} = $hdr{Status}; $hdr{Status} = $_[1]; |
|
|
755 | $hdr{OrigReason} = $hdr{Reason}; $hdr{Reason} = $_[2]; |
|
|
756 | } |
|
|
757 | |
|
|
758 | # set-cookie processing |
762 | # set-cookie processing |
759 | if ($arg{cookie_jar}) { |
763 | if ($arg{cookie_jar}) { |
760 | cookie_jar_set_cookie $arg{cookie_jar}, $hdr{"set-cookie"}, $uhost, $hdr{date}; |
764 | cookie_jar_set_cookie $arg{cookie_jar}, $hdr{"set-cookie"}, $uhost, $hdr{date}; |
761 | } |
765 | } |
762 | |
766 | |
763 | if ($redirect && exists $hdr{location}) { |
767 | if ($redirect && exists $hdr{location}) { |
764 | # we ignore any errors, as it is very common to receive |
768 | # we ignore any errors, as it is very common to receive |
765 | # Content-Length != 0 but no actual body |
769 | # Content-Length != 0 but no actual body |
766 | # we also access %hdr, as $_[1] might be an erro |
770 | # we also access %hdr, as $_[1] might be an erro |
767 | http_request ( |
771 | http_request ( |
768 | $method => $hdr{location}, |
772 | $method => $hdr{location}, |
769 | %arg, |
773 | %arg, |
770 | recurse => $recurse - 1, |
774 | recurse => $recurse - 1, |
771 | Redirect => [$_[0], \%hdr], |
775 | Redirect => [$_[0], \%hdr], |
772 | $cb); |
776 | $cb); |
|
|
777 | } else { |
|
|
778 | $cb->($_[0], \%hdr); |
|
|
779 | } |
|
|
780 | }; |
|
|
781 | |
|
|
782 | $ae_error = 597; # body phase |
|
|
783 | |
|
|
784 | my $len = $hdr{"content-length"}; |
|
|
785 | |
|
|
786 | # body handling, many different code paths |
|
|
787 | # - no body expected |
|
|
788 | # - want_body_handle |
|
|
789 | # - te chunked |
|
|
790 | # - 2x length known (with or without on_body) |
|
|
791 | # - 2x length not known (with or without on_body) |
|
|
792 | if (!$redirect && $arg{on_header} && !$arg{on_header}(\%hdr)) { |
|
|
793 | $finish->(undef, 598 => "Request cancelled by on_header"); |
|
|
794 | } elsif ( |
|
|
795 | $hdr{Status} =~ /^(?:1..|204|205|304)$/ |
|
|
796 | or $method eq "HEAD" |
|
|
797 | or (defined $len && $len == 0) # == 0, not !, because "0 " is true |
|
|
798 | ) { |
|
|
799 | # no body |
|
|
800 | $finish->("", undef, undef, 1); |
|
|
801 | |
|
|
802 | } elsif (!$redirect && $arg{want_body_handle}) { |
|
|
803 | $_[0]->on_eof (undef); |
|
|
804 | $_[0]->on_error (undef); |
|
|
805 | $_[0]->on_read (undef); |
|
|
806 | |
|
|
807 | $finish->(delete $state{handle}); |
|
|
808 | |
|
|
809 | } elsif ($hdr{"transfer-encoding"} =~ /\bchunked\b/i) { |
|
|
810 | my $cl = 0; |
|
|
811 | my $body = undef; |
|
|
812 | my $on_body = $arg{on_body} || sub { $body .= shift; 1 }; |
|
|
813 | |
|
|
814 | $state{read_chunk} = sub { |
|
|
815 | $_[1] =~ /^([0-9a-fA-F]+)/ |
|
|
816 | or $finish->(undef, $ae_error => "Garbled chunked transfer encoding"); |
|
|
817 | |
|
|
818 | my $len = hex $1; |
|
|
819 | |
|
|
820 | if ($len) { |
|
|
821 | $cl += $len; |
|
|
822 | |
|
|
823 | $_[0]->push_read (chunk => $len, sub { |
|
|
824 | $on_body->($_[1], \%hdr) |
|
|
825 | or return $finish->(undef, 598 => "Request cancelled by on_body"); |
|
|
826 | |
|
|
827 | $_[0]->push_read (line => sub { |
|
|
828 | length $_[1] |
|
|
829 | and return $finish->(undef, $ae_error => "Garbled chunked transfer encoding"); |
|
|
830 | $_[0]->push_read (line => $state{read_chunk}); |
|
|
831 | }); |
|
|
832 | }); |
773 | } else { |
833 | } else { |
774 | $cb->($_[0], \%hdr); |
834 | $hdr{"content-length"} ||= $cl; |
|
|
835 | |
|
|
836 | $_[0]->push_read (line => $qr_nlnl, sub { |
|
|
837 | if (length $_[1]) { |
|
|
838 | for ("$_[1]") { |
|
|
839 | y/\015//d; # weed out any \015, as they show up in the weirdest of places. |
|
|
840 | |
|
|
841 | my $hdr = parse_hdr |
|
|
842 | or return $finish->(undef, $ae_error => "Garbled response trailers"); |
|
|
843 | |
|
|
844 | %hdr = (%hdr, %$hdr); |
|
|
845 | } |
|
|
846 | } |
|
|
847 | |
|
|
848 | $finish->($body, undef, undef, 1); |
|
|
849 | }); |
775 | } |
850 | } |
776 | }; |
851 | }; |
777 | |
852 | |
778 | $ae_error = 597; # body phase |
|
|
779 | |
|
|
780 | my $len = $hdr{"content-length"}; |
|
|
781 | |
|
|
782 | # body handling, many different code paths |
|
|
783 | # - no body expected |
|
|
784 | # - want_body_handle |
|
|
785 | # - te chunked |
|
|
786 | # - 2x length known (with or without on_body) |
|
|
787 | # - 2x length not known (with or without on_body) |
|
|
788 | if (!$redirect && $arg{on_header} && !$arg{on_header}(\%hdr)) { |
|
|
789 | $finish->(undef, 598 => "Request cancelled by on_header"); |
|
|
790 | } elsif ( |
|
|
791 | $hdr{Status} =~ /^(?:1..|204|205|304)$/ |
|
|
792 | or $method eq "HEAD" |
|
|
793 | or (defined $len && $len == 0) # == 0, not !, because "0 " is true |
|
|
794 | ) { |
|
|
795 | # no body |
|
|
796 | $finish->("", undef, undef, 1); |
|
|
797 | |
|
|
798 | } elsif (!$redirect && $arg{want_body_handle}) { |
|
|
799 | $_[0]->on_eof (undef); |
|
|
800 | $_[0]->on_error (undef); |
|
|
801 | $_[0]->on_read (undef); |
|
|
802 | |
|
|
803 | $finish->(delete $state{handle}); |
|
|
804 | |
|
|
805 | } elsif ($hdr{"transfer-encoding"} =~ /\bchunked\b/i) { |
|
|
806 | my $cl = 0; |
|
|
807 | my $body = undef; |
|
|
808 | my $on_body = $arg{on_body} || sub { $body .= shift; 1 }; |
|
|
809 | |
|
|
810 | $state{read_chunk} = sub { |
|
|
811 | $_[1] =~ /^([0-9a-fA-F]+)/ |
|
|
812 | or $finish->(undef, $ae_error => "Garbled chunked transfer encoding"); |
|
|
813 | |
|
|
814 | my $len = hex $1; |
|
|
815 | |
|
|
816 | if ($len) { |
|
|
817 | $cl += $len; |
|
|
818 | |
|
|
819 | $_[0]->push_read (chunk => $len, sub { |
|
|
820 | $on_body->($_[1], \%hdr) |
|
|
821 | or return $finish->(undef, 598 => "Request cancelled by on_body"); |
|
|
822 | |
|
|
823 | $_[0]->push_read (line => sub { |
|
|
824 | length $_[1] |
|
|
825 | and return $finish->(undef, $ae_error => "Garbled chunked transfer encoding"); |
|
|
826 | $_[0]->push_read (line => $state{read_chunk}); |
|
|
827 | }); |
|
|
828 | }); |
|
|
829 | } else { |
|
|
830 | $hdr{"content-length"} ||= $cl; |
|
|
831 | |
|
|
832 | $_[0]->push_read (line => $qr_nlnl, sub { |
|
|
833 | if (length $_[1]) { |
|
|
834 | for ("$_[1]") { |
|
|
835 | y/\015//d; # weed out any \015, as they show up in the weirdest of places. |
|
|
836 | |
|
|
837 | my $hdr = parse_hdr |
|
|
838 | or return $finish->(undef, $ae_error => "Garbled response trailers"); |
|
|
839 | |
|
|
840 | %hdr = (%hdr, %$hdr); |
|
|
841 | } |
|
|
842 | } |
|
|
843 | |
|
|
844 | $finish->($body, undef, undef, 1); |
|
|
845 | }); |
|
|
846 | } |
|
|
847 | }; |
|
|
848 | |
|
|
849 | $_[0]->push_read (line => $state{read_chunk}); |
853 | $_[0]->push_read (line => $state{read_chunk}); |
850 | |
854 | |
851 | } elsif ($arg{on_body}) { |
855 | } elsif ($arg{on_body}) { |
852 | if (defined $len) { |
856 | if (defined $len) { |
853 | $_[0]->on_read (sub { |
857 | $_[0]->on_read (sub { |
854 | $len -= length $_[0]{rbuf}; |
858 | $len -= length $_[0]{rbuf}; |
855 | |
859 | |
856 | $arg{on_body}(delete $_[0]{rbuf}, \%hdr) |
860 | $arg{on_body}(delete $_[0]{rbuf}, \%hdr) |
857 | or return $finish->(undef, 598 => "Request cancelled by on_body"); |
861 | or return $finish->(undef, 598 => "Request cancelled by on_body"); |
858 | |
862 | |
859 | $len > 0 |
863 | $len > 0 |
860 | or $finish->("", undef, undef, 1); |
864 | or $finish->("", undef, undef, 1); |
861 | }); |
|
|
862 | } else { |
|
|
863 | $_[0]->on_eof (sub { |
|
|
864 | $finish->(""); |
|
|
865 | }); |
|
|
866 | $_[0]->on_read (sub { |
|
|
867 | $arg{on_body}(delete $_[0]{rbuf}, \%hdr) |
|
|
868 | or $finish->(undef, 598 => "Request cancelled by on_body"); |
|
|
869 | }); |
|
|
870 | } |
865 | }); |
871 | } else { |
866 | } else { |
872 | $_[0]->on_eof (undef); |
867 | $_[0]->on_eof (sub { |
873 | |
868 | $finish->(""); |
874 | if (defined $len) { |
869 | }); |
875 | $_[0]->on_read (sub { |
870 | $_[0]->on_read (sub { |
876 | $finish->((substr delete $_[0]{rbuf}, 0, $len, ""), undef, undef, 1) |
871 | $arg{on_body}(delete $_[0]{rbuf}, \%hdr) |
877 | if $len <= length $_[0]{rbuf}; |
872 | or $finish->(undef, 598 => "Request cancelled by on_body"); |
878 | }); |
|
|
879 | } else { |
|
|
880 | $_[0]->on_error (sub { |
|
|
881 | ($! == Errno::EPIPE || !$!) |
|
|
882 | ? $finish->(delete $_[0]{rbuf}) |
|
|
883 | : $finish->(undef, $ae_error => $_[2]); |
|
|
884 | }); |
|
|
885 | $_[0]->on_read (sub { }); |
|
|
886 | } |
873 | }); |
887 | } |
874 | } |
|
|
875 | } else { |
|
|
876 | $_[0]->on_eof (undef); |
|
|
877 | |
|
|
878 | if (defined $len) { |
|
|
879 | $_[0]->on_read (sub { |
|
|
880 | $finish->((substr delete $_[0]{rbuf}, 0, $len, ""), undef, undef, 1) |
|
|
881 | if $len <= length $_[0]{rbuf}; |
|
|
882 | }); |
|
|
883 | } else { |
|
|
884 | $_[0]->on_error (sub { |
|
|
885 | ($! == Errno::EPIPE || !$!) |
|
|
886 | ? $finish->(delete $_[0]{rbuf}) |
|
|
887 | : $finish->(undef, $ae_error => $_[2]); |
|
|
888 | }); |
|
|
889 | $_[0]->on_read (sub { }); |
|
|
890 | } |
|
|
891 | } |
|
|
892 | }; |
|
|
893 | |
|
|
894 | $state{handle}->push_read (line => $qr_nlnl, $state{read_response}); |
|
|
895 | }; |
|
|
896 | |
|
|
897 | my $connect_cb = sub { |
|
|
898 | $state{fh} = shift |
|
|
899 | or do { |
|
|
900 | my $err = "$!"; |
|
|
901 | %state = (); |
|
|
902 | return $cb->(undef, { @pseudo, Status => $ae_error, Reason => $err }); |
888 | }; |
903 | }; |
889 | |
904 | |
890 | $state{handle}->push_read (line => $qr_nlnl, $state{read_response}); |
|
|
891 | }; |
|
|
892 | |
|
|
893 | my $connect_cb = sub { |
|
|
894 | $state{fh} = shift |
|
|
895 | or do { |
|
|
896 | my $err = "$!"; |
|
|
897 | %state = (); |
|
|
898 | return $cb->(undef, { @pseudo, Status => $ae_error, Reason => $err }); |
|
|
899 | }; |
|
|
900 | |
|
|
901 | return unless delete $state{connect_guard}; |
905 | return unless delete $state{connect_guard}; |
902 | |
906 | |
903 | # get handle |
907 | # get handle |
904 | $state{handle} = new AnyEvent::Handle |
908 | $state{handle} = new AnyEvent::Handle |
905 | fh => $state{fh}, |
909 | fh => $state{fh}, |
906 | peername => $rhost, |
910 | peername => $rhost, |
907 | tls_ctx => $arg{tls_ctx}, |
911 | tls_ctx => $arg{tls_ctx}, |
908 | # these need to be reconfigured on keepalive handles |
912 | # these need to be reconfigured on keepalive handles |
909 | timeout => $timeout, |
913 | timeout => $timeout, |
910 | on_error => sub { |
914 | on_error => sub { |
911 | %state = (); |
915 | %state = (); |
912 | $cb->(undef, { @pseudo, Status => $ae_error, Reason => $_[2] }); |
916 | $cb->(undef, { @pseudo, Status => $ae_error, Reason => $_[2] }); |
913 | }, |
917 | }, |
914 | on_eof => sub { |
918 | on_eof => sub { |
915 | %state = (); |
919 | %state = (); |
916 | $cb->(undef, { @pseudo, Status => $ae_error, Reason => "Unexpected end-of-file" }); |
920 | $cb->(undef, { @pseudo, Status => $ae_error, Reason => "Unexpected end-of-file" }); |
917 | }, |
921 | }, |
918 | ; |
922 | ; |
919 | |
923 | |
920 | # limit the number of persistent connections |
924 | # limit the number of persistent connections |
921 | # keepalive not yet supported |
925 | # keepalive not yet supported |
922 | # if ($KA_COUNT{$_[1]} < $MAX_PERSISTENT_PER_HOST) { |
926 | # if ($KA_COUNT{$_[1]} < $MAX_PERSISTENT_PER_HOST) { |
923 | # ++$KA_COUNT{$_[1]}; |
927 | # ++$KA_COUNT{$_[1]}; |
924 | # $state{handle}{ka_count_guard} = AnyEvent::Util::guard { |
928 | # $state{handle}{ka_count_guard} = AnyEvent::Util::guard { |
925 | # --$KA_COUNT{$_[1]} |
929 | # --$KA_COUNT{$_[1]} |
926 | # }; |
930 | # }; |
927 | # $hdr{connection} = "keep-alive"; |
931 | # $hdr{connection} = "keep-alive"; |
928 | # } |
932 | # } |
929 | |
933 | |
930 | $state{handle}->starttls ("connect") if $rscheme eq "https"; |
934 | $state{handle}->starttls ("connect") if $rscheme eq "https"; |
931 | |
935 | |
932 | # now handle proxy-CONNECT method |
936 | # now handle proxy-CONNECT method |
933 | if ($proxy && $uscheme eq "https") { |
937 | if ($proxy && $uscheme eq "https") { |
934 | # oh dear, we have to wrap it into a connect request |
938 | # oh dear, we have to wrap it into a connect request |
935 | |
939 | |
936 | # maybe re-use $uauthority with patched port? |
940 | # maybe re-use $uauthority with patched port? |
937 | $state{handle}->push_write ("CONNECT $uhost:$uport HTTP/1.0\015\012Host: $uhost\015\012\015\012"); |
941 | $state{handle}->push_write ("CONNECT $uhost:$uport HTTP/1.0\015\012\015\012"); |
938 | $state{handle}->push_read (line => $qr_nlnl, sub { |
942 | $state{handle}->push_read (line => $qr_nlnl, sub { |
939 | $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix |
943 | $_[1] =~ /^HTTP\/([0-9\.]+) \s+ ([0-9]{3}) (?: \s+ ([^\015\012]*) )?/ix |
940 | or return (%state = (), $cb->(undef, { @pseudo, Status => 599, Reason => "Invalid proxy connect response ($_[1])" })); |
944 | or return (%state = (), $cb->(undef, { @pseudo, Status => 599, Reason => "Invalid proxy connect response ($_[1])" })); |
941 | |
945 | |
942 | if ($2 == 200) { |
946 | if ($2 == 200) { |
943 | $rpath = $upath; |
947 | $rpath = $upath; |
944 | $handle_actual_request->(); |
948 | $handle_actual_request->(); |
945 | } else { |
949 | } else { |
946 | %state = (); |
950 | %state = (); |
947 | $cb->(undef, { @pseudo, Status => $2, Reason => $3 }); |
951 | $cb->(undef, { @pseudo, Status => $2, Reason => $3 }); |
948 | } |
|
|
949 | }); |
952 | } |
|
|
953 | }); |
950 | } else { |
954 | } else { |
951 | $handle_actual_request->(); |
955 | $handle_actual_request->(); |
952 | } |
|
|
953 | }; |
956 | } |
|
|
957 | }; |
|
|
958 | |
|
|
959 | _get_slot $uhost, sub { |
|
|
960 | $state{slot_guard} = shift; |
|
|
961 | |
|
|
962 | return unless $state{connect_guard}; |
954 | |
963 | |
955 | my $tcp_connect = $arg{tcp_connect} |
964 | my $tcp_connect = $arg{tcp_connect} |
956 | || do { require AnyEvent::Socket; \&AnyEvent::Socket::tcp_connect }; |
965 | || do { require AnyEvent::Socket; \&AnyEvent::Socket::tcp_connect }; |
957 | |
966 | |
958 | $state{connect_guard} = $tcp_connect->($rhost, $rport, $connect_cb, $arg{on_prepare} || sub { $timeout }); |
967 | $state{connect_guard} = $tcp_connect->($rhost, $rport, $connect_cb, $arg{on_prepare} || sub { $timeout }); |