]> Sergey Matveev's repositories - public-inbox.git/commitdiff
httpd: make internals slightly more generic
authorEric Wong <e@80x24.org>
Mon, 1 Aug 2022 21:24:42 +0000 (21:24 +0000)
committerEric Wong <e@80x24.org>
Tue, 2 Aug 2022 19:20:10 +0000 (19:20 +0000)
This brings the HTTP server closer to the IMAP/NNTP/POP3
implementations and eliminates package-wide globals in
PublicInbox::HTTPD.  The end goal is to be able to host
completely different PSGI applications on different listen
ports.

lib/PublicInbox/Daemon.pm
lib/PublicInbox/HTTP.pm
lib/PublicInbox/HTTPD.pm

index bceae6e5699f5d13412f7cdc4e544074b58fd1d7..1af03cc47947c9cd7f5ab10fa9a71409fc39af0b 100644 (file)
@@ -81,11 +81,11 @@ sub load_mod ($) {
        my $mod = $modc.'D';
        eval "require $mod"; # IMAPD|HTTPD|NNTPD|POP3D
        die $@ if $@;
-       my %xn = map { $_ => $mod->can($_) } qw(refresh post_accept);
-       $xn{tlsd} = $mod->new if $mod->can('refresh_groups'); #!HTTPD
-       my $tlsd = $xn{tlsd};
-       $xn{refresh} //= sub { $tlsd->refresh_groups(@_) };
-       $xn{post_accept} //= sub { $modc->new($_[0], $tlsd) };
+       my %xn;
+       my $tlsd = $xn{tlsd} = $mod->new;
+       $xn{refresh} = sub { $tlsd->refresh_groups(@_) };
+       $xn{post_accept} = $tlsd->can('post_accept_cb') ?
+                       $tlsd->post_accept_cb : sub { $modc->new($_[0], $tlsd) };
        $xn{af_default} = 'httpready' if $modc eq 'PublicInbox::HTTP';
        \%xn;
 }
index 76e978a249261f2c8d4872973dfb293db720fb03..669211e38099fbacf3e4b4309deeced1568c5d00 100644 (file)
@@ -1,4 +1,4 @@
-# Copyright (C) 2016-2021 all contributors <meta@public-inbox.org>
+# Copyright (C) all contributors <meta@public-inbox.org>
 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
 #
 # Generic PSGI server for convenience.  It aims to provide
@@ -52,8 +52,8 @@ sub http_date () {
 }
 
 sub new ($$$) {
-       my ($class, $sock, $addr, $httpd) = @_;
-       my $self = bless { httpd => $httpd }, $class;
+       my ($class, $sock, $addr, $srv_env) = @_;
+       my $self = bless { srv_env => $srv_env }, $class;
        my $ev = EPOLLIN;
        my $wbuf;
        if ($sock->can('accept_SSL') && !$sock->accept_SSL) {
@@ -78,7 +78,7 @@ sub event_step { # called by PublicInbox::DS
        return read_input($self) if ref($self->{env});
 
        my $rbuf = $self->{rbuf} // (\(my $x = ''));
-       my %env = %{$self->{httpd}->{env}}; # full hash copy
+       my %env = %{$self->{srv_env}}; # full hash copy
        my $r;
        while (($r = parse_http_request($$rbuf, \%env)) < 0) {
                # We do not support Trailers in chunked requests, for
@@ -145,7 +145,7 @@ sub app_dispatch {
        # note: NOT $self->{sock}, we want our close (+ PublicInbox::DS::close),
        # to do proper cleanup:
        $env->{'psgix.io'} = $self; # for ->close or async_pass
-       my $res = Plack::Util::run_app($self->{httpd}->{app}, $env);
+       my $res = Plack::Util::run_app($env->{'pi-httpd.app'}, $env);
        eval {
                if (ref($res) eq 'CODE') {
                        $res->(sub { response_write($self, $env, $_[0]) });
index 715e453841be3fd4fe0b2a1d389d299a8a24ab15..bcdbb9f9aa2b360cc6c4640b726e48b8f46d73e3 100644 (file)
@@ -13,12 +13,17 @@ use PublicInbox::HTTPD::Async;
 
 sub pi_httpd_async { PublicInbox::HTTPD::Async->new(@_) }
 
-sub new {
-       my ($class, $sock, $app, $client) = @_;
-       my $n = getsockname($sock) or die "not a socket: $sock $!\n";
-       my ($host, $port) = PublicInbox::Daemon::host_with_port($n);
+# we have a different env for ever listener socket for
+# SERVER_NAME, SERVER_PORT and psgi.url_scheme
+# envs: listener FD => PSGI env
+sub new { bless { envs => {} }, __PACKAGE__ }
 
-       my %env = (
+# this becomes {srv_env} in PublicInbox::HTTP
+sub env_for ($$$) {
+       my ($self, $srv, $client) = @_;
+       my $n = getsockname($srv) or die "not a socket: $srv $!\n";
+       my ($host, $port) = PublicInbox::Daemon::host_with_port($n);
+       {
                SERVER_NAME => $host,
                SERVER_PORT => $port,
                SCRIPT_NAME => '',
@@ -40,26 +45,24 @@ sub new {
                # this to limit git-http-backend(1) parallelism.
                # We also check for the truthiness of this to
                # detect when to use async paths for slow blobs
-               'pi-httpd.async' => \&pi_httpd_async
-       );
-       bless { app => $app, env => \%env }, $class;
+               'pi-httpd.async' => \&pi_httpd_async,
+               'pi-httpd.app' => $self->{app},
+       }
 }
 
-my %httpds; # per-listen-FD mapping for HTTPD->{env}->{SERVER_<NAME|PORT>}
-my $default_app; # ugh...
-
-sub refresh {
+sub refresh_groups {
+       my ($self) = @_;
+       my $app;
        if (@main::ARGV) {
-               eval { $default_app = Plack::Util::load_psgi(@ARGV) };
-               if ($@) {
-                       die $@,
-"$0 runs in /, command-line paths must be absolute\n";
-               }
+               eval { $app = Plack::Util::load_psgi(@ARGV) };
+               die $@, <<EOM if $@;
+$0 runs in /, command-line paths must be absolute
+EOM
        } else {
                require PublicInbox::WWW;
                my $www = PublicInbox::WWW->new;
                $www->preload;
-               $default_app = builder {
+               $app = builder {
                        eval { enable 'ReverseProxy' };
                        $@ and warn <<EOM;
 Plack::Middleware::ReverseProxy missing,
@@ -69,14 +72,18 @@ EOM
                        sub { $www->call(@_) };
                };
        }
-       %httpds = (); # invalidate cache
+       $_->{'pi-httpd.app'} = $app for values %{$self->{envs}};
+       $self->{app} = $app;
 }
 
-sub post_accept { # Listener->{post_accept}
-       my ($client, $addr, $srv) = @_; # $_[3] - tls_wrap (unused)
-       my $httpd = $httpds{fileno($srv)} //=
-                               __PACKAGE__->new($srv, $default_app, $client);
-       PublicInbox::HTTP->new($client, $addr, $httpd),
+sub post_accept_cb { # for Listener->{post_accept}
+       my ($self) = @_;
+       sub {
+               my ($client, $addr, $srv) = @_; # $_[4] - tls_wrap (unused)
+               PublicInbox::HTTP->new($client, $addr,
+                               $self->{envs}->{fileno($srv)} //=
+                                       env_for($self, $srv, $client));
+       }
 }
 
 1;