1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
| | # Copyright (C) 2016-2020 all contributors <meta@public-inbox.org>
# License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
# corner case tests for the generic PSGI server
# Usage: plackup [OPTIONS] /path/to/this/file
use strict;
use warnings;
use Plack::Builder;
require Digest::SHA;
my $app = sub {
my ($env) = @_;
my $path = $env->{PATH_INFO};
my $in = $env->{'psgi.input'};
my $actual = -s $in;
my $code = 500;
my $h = [ 'Content-Type' => 'text/plain' ];
my $body = [];
if ($path eq '/sha1') {
my $sha1 = Digest::SHA->new('SHA-1');
my $buf;
while (1) {
my $r = $in->read($buf, 4096);
die "read err: $!" unless defined $r;
last if $r == 0;
$sha1->add($buf);
}
$code = 200;
push @$body, $sha1->hexdigest;
} elsif (my $fifo = $env->{HTTP_X_CHECK_FIFO}) {
if ($path eq '/slow-header') {
return sub {
open my $f, '<', $fifo or
die "open $fifo: $!\n";
local $/ = "\n";
my @r = <$f>;
$_[0]->([200, $h, \@r ]);
};
} elsif ($path eq '/slow-body') {
return sub {
my $fh = $_[0]->([200, $h]);
open my $f, '<', $fifo or
die "open $fifo: $!\n";
local $/ = "\n";
while (defined(my $l = <$f>)) {
$fh->write($l);
}
$fh->close;
};
}
} elsif ($path eq '/host-port') {
$code = 200;
push @$body, "$env->{REMOTE_ADDR}:$env->{REMOTE_PORT}";
} elsif ($path eq '/callback') {
return sub {
my ($res) = @_;
my $buf = "hello world\n";
push @$h, 'Content-Length', length($buf);
my $fh = $res->([200, $h]);
$fh->write($buf);
$fh->close;
}
} elsif ($path eq '/empty') {
$code = 200;
} elsif ($path eq '/getline-die') {
$code = 200;
$body = Plack::Util::inline_object(
getline => sub { die 'GETLINE FAIL' },
close => sub { die 'CLOSE FAIL' },
);
} elsif ($path eq '/close-die') {
$code = 200;
$body = Plack::Util::inline_object(
getline => sub { undef },
close => sub { die 'CLOSE FAIL' },
);
} elsif ($path eq '/async-big') {
require PublicInbox::Qspawn;
open my $null, '>', '/dev/null' or die;
my $rdr = { 2 => fileno($null) };
my $cmd = [qw(dd if=/dev/zero count=30 bs=1024k)];
my $qsp = PublicInbox::Qspawn->new($cmd, undef, $rdr);
return $qsp->psgi_return($env, undef, sub {
my ($r, $bref) = @_;
# make $rd_hdr retry sysread + $parse_hdr in Qspawn:
return until length($$bref) > 8000;
close $null;
[ 200, [ qw(Content-Type application/octet-stream) ]];
});
} elsif ($path eq '/psgi-return-gzip') {
require PublicInbox::Qspawn;
require PublicInbox::GzipFilter;
my $cmd = [qw(echo hello world)];
my $qsp = PublicInbox::Qspawn->new($cmd);
$env->{'qspawn.filter'} = PublicInbox::GzipFilter->new;
return $qsp->psgi_return($env, undef, sub {
[ 200, [ qw(Content-Type application/octet-stream)]]
});
} elsif ($path eq '/psgi-return-enoent') {
require PublicInbox::Qspawn;
my $cmd = [ 'this-better-not-exist-in-PATH'.rand ];
my $qsp = PublicInbox::Qspawn->new($cmd);
return $qsp->psgi_return($env, undef, sub {
[ 200, [ qw(Content-Type application/octet-stream)]]
});
} elsif ($path eq '/pid') {
$code = 200;
push @$body, "$$\n";
}
[ $code, $h, $body ]
};
builder {
enable 'ContentLength';
enable 'Head';
$app;
}
|