Mercurial > hg > nginx-tests

comparison grpc_pass.t @ 1547:2d747d967fc0

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests: grpc_pass with variables.
author Sergey Kandaurov <pluknet@nginx.com>
date Mon, 20 Jan 2020 16:30:56 +0300
parents
children 5ac6efbe5552
comparison
equal deleted inserted replaced
1546:0902a4539c99 1547:2d747d967fc0
1 #!/usr/bin/perl
2
3 # (C) Sergey Kandaurov
4 # (C) Nginx, Inc.
5
6 # Tests for the grpc_pass directive with variables.
7
8 ###############################################################################
9
10 use warnings;
11 use strict;
12
13 use Test::More;
14
15 BEGIN { use FindBin; chdir($FindBin::Bin); }
16
17 use lib 'lib';
18 use Test::Nginx;
19
20 ###############################################################################
21
22 select STDERR; $| = 1;
23 select STDOUT; $| = 1;
24
25 my $t = Test::Nginx->new()->has(qw/http http_ssl http_v2 grpc rewrite/)
26 ->has_daemon('openssl')->write_file_expand('nginx.conf', <<'EOF');
27
28 %%TEST_GLOBALS%%
29
30 daemon off;
31
32 events {
33 }
34
35 http {
36 %%TEST_GLOBALS_HTTP%%
37
38 upstream u {
39 server 127.0.0.1:8081;
40 }
41
42 resolver 127.0.0.1:%%PORT_8982_UDP%%;
43
44 server {
45 listen 127.0.0.1:8080;
46 server_name localhost;
47
48 location / {
49 grpc_pass $host:%%PORT_8081%%;
50 }
51
52 location /grpc {
53 grpc_pass grpc://$host:%%PORT_8081%%;
54 }
55
56 location /grpcs {
57 grpc_pass grpcs://$host:%%PORT_8082%%;
58 }
59
60 location /arg {
61 grpc_pass $arg_b;
62 }
63 }
64
65 server {
66 listen 127.0.0.1:8081 http2;
67 listen 127.0.0.1:8082 http2 ssl;
68 server_name localhost;
69
70 ssl_certificate_key localhost.key;
71 ssl_certificate localhost.crt;
72
73 location / {
74 return 200 $http_host;
75 }
76 }
77 }
78
79 EOF
80
81 $t->write_file('openssl.conf', <<EOF);
82 [ req ]
83 default_bits = 2048
84 encrypt_key = no
85 distinguished_name = req_distinguished_name
86 [ req_distinguished_name ]
87 EOF
88
89 my $d = $t->testdir();
90
91 foreach my $name ('localhost') {
92 system('openssl req -x509 -new '
93 . "-config $d/openssl.conf -subj /CN=$name/ "
94 . "-out $d/$name.crt -keyout $d/$name.key "
95 . ">>$d/openssl.out 2>&1") == 0
96 or die "Can't create certificate for $name: $!\n";
97 }
98
99 $t->run_daemon(\&dns_daemon, port(8982), $t);
100 $t->try_run('no grpc_pass variables')->plan(5);
101
102 $t->waitforfile($t->testdir . '/' . port(8982));
103
104 ###############################################################################
105
106 like(http_get('/basic'), qr/200 OK/, 'no scheme');
107 like(http_get('/grpc'), qr/200 OK/, 'grpc scheme');
108
109 SKIP: {
110 $t->{_configure_args} =~ /OpenSSL ([\d\.]+)/;
111 skip 'OpenSSL too old', 1 unless defined $1 and $1 ge '1.0.2';
112
113 like(http_get('/grpcs'), qr/200 OK/, 'grpcs scheme');
114
115 }
116
117 like(http_get('/arg?b=grpc://127.0.0.1:' . port(8081)), qr/200 OK/, 'addrs');
118 like(http_get('/arg?b=grpc://u'), qr/200 OK/, 'no_port');
119
120 ###############################################################################
121
122 sub reply_handler {
123 my ($recv_data) = @_;
124
125 my (@name, @rdata);
126
127 use constant NOERROR => 0;
128 use constant A => 1;
129 use constant IN => 1;
130
131 # default values
132
133 my ($hdr, $rcode, $ttl) = (0x8180, NOERROR, 3600);
134
135 # decode name
136
137 my ($len, $offset) = (undef, 12);
138 while (1) {
139 $len = unpack("\@$offset C", $recv_data);
140 last if $len == 0;
141 $offset++;
142 push @name, unpack("\@$offset A$len", $recv_data);
143 $offset += $len;
144 }
145
146 $offset -= 1;
147 my ($id, $type, $class) = unpack("n x$offset n2", $recv_data);
148
149 my $name = join('.', @name);
150 if ($name eq 'localhost' && $type == A) {
151 push @rdata, rd_addr($ttl, '127.0.0.1');
152 }
153
154 $len = @name;
155 pack("n6 (C/a*)$len x n2", $id, $hdr | $rcode, 1, scalar @rdata,
156 0, 0, @name, $type, $class) . join('', @rdata);
157 }
158
159 sub rd_addr {
160 my ($ttl, $addr) = @_;
161
162 my $code = 'split(/\./, $addr)';
163
164 return pack 'n3N', 0xc00c, A, IN, $ttl if $addr eq '';
165
166 pack 'n3N nC4', 0xc00c, A, IN, $ttl, eval "scalar $code", eval($code);
167 }
168
169 sub dns_daemon {
170 my ($port, $t) = @_;
171
172 my ($data, $recv_data);
173 my $socket = IO::Socket::INET->new(
174 LocalAddr => '127.0.0.1',
175 LocalPort => $port,
176 Proto => 'udp',
177 )
178 or die "Can't create listening socket: $!\n";
179
180 # signal we are ready
181
182 open my $fh, '>', $t->testdir() . '/' . $port;
183 close $fh;
184
185 while (1) {
186 $socket->recv($recv_data, 65536);
187 $data = reply_handler($recv_data);
188 $socket->send($data);
189 }
190 }
191
192 ###############################################################################