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