Blame


1 b0a6bcf7 2022-09-13 op /*
2 b0a6bcf7 2022-09-13 op * Copyright (c) 2022 Omar Polo <op@omarpolo.com>
3 b0a6bcf7 2022-09-13 op * Copyright (c) 2006 - 2015 Reyk Floeter <reyk@openbsd.org>
4 b0a6bcf7 2022-09-13 op *
5 b0a6bcf7 2022-09-13 op * Permission to use, copy, modify, and distribute this software for any
6 b0a6bcf7 2022-09-13 op * purpose with or without fee is hereby granted, provided that the above
7 b0a6bcf7 2022-09-13 op * copyright notice and this permission notice appear in all copies.
8 b0a6bcf7 2022-09-13 op *
9 b0a6bcf7 2022-09-13 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 b0a6bcf7 2022-09-13 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 b0a6bcf7 2022-09-13 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 b0a6bcf7 2022-09-13 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 b0a6bcf7 2022-09-13 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 b0a6bcf7 2022-09-13 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 b0a6bcf7 2022-09-13 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 b0a6bcf7 2022-09-13 op */
17 b0a6bcf7 2022-09-13 op
18 b0a6bcf7 2022-09-13 op #include <sys/types.h>
19 b0a6bcf7 2022-09-13 op #include <sys/queue.h>
20 b0a6bcf7 2022-09-13 op #include <sys/tree.h>
21 b0a6bcf7 2022-09-13 op #include <sys/socket.h>
22 b0a6bcf7 2022-09-13 op #include <sys/uio.h>
23 b0a6bcf7 2022-09-13 op
24 b0a6bcf7 2022-09-13 op #include <netinet/in.h>
25 b0a6bcf7 2022-09-13 op #include <arpa/inet.h>
26 b0a6bcf7 2022-09-13 op #include <netdb.h>
27 b0a6bcf7 2022-09-13 op
28 b0a6bcf7 2022-09-13 op #include <asr.h>
29 b0a6bcf7 2022-09-13 op #include <ctype.h>
30 b0a6bcf7 2022-09-13 op #include <errno.h>
31 b0a6bcf7 2022-09-13 op #include <event.h>
32 33f4bf26 2022-09-14 op #include <fnmatch.h>
33 b0a6bcf7 2022-09-13 op #include <limits.h>
34 b0a6bcf7 2022-09-13 op #include <stdlib.h>
35 b0a6bcf7 2022-09-13 op #include <stdint.h>
36 b0a6bcf7 2022-09-13 op #include <stdio.h>
37 b0a6bcf7 2022-09-13 op #include <string.h>
38 b0a6bcf7 2022-09-13 op #include <imsg.h>
39 b0a6bcf7 2022-09-13 op #include <tls.h>
40 b0a6bcf7 2022-09-13 op #include <unistd.h>
41 b0a6bcf7 2022-09-13 op
42 b0a6bcf7 2022-09-13 op #include "log.h"
43 b0a6bcf7 2022-09-13 op #include "proc.h"
44 bff58270 2022-09-19 op #include "tmpl.h"
45 b0a6bcf7 2022-09-13 op
46 b0a6bcf7 2022-09-13 op #include "galileo.h"
47 b0a6bcf7 2022-09-13 op
48 b0a6bcf7 2022-09-13 op #define MINIMUM(a, b) ((a) < (b) ? (a) : (b))
49 b0a6bcf7 2022-09-13 op
50 b0a6bcf7 2022-09-13 op /* provided by OpenBSD' base libevent but not in any header? */
51 b0a6bcf7 2022-09-13 op extern void bufferevent_read_pressure_cb(struct evbuffer *, size_t,
52 b0a6bcf7 2022-09-13 op size_t, void *);
53 b0a6bcf7 2022-09-13 op
54 b0a6bcf7 2022-09-13 op void proxy_init(struct privsep *, struct privsep_proc *, void *);
55 b0a6bcf7 2022-09-13 op int proxy_launch(struct galileo *);
56 b0a6bcf7 2022-09-13 op void proxy_inflight_dec(const char *);
57 b0a6bcf7 2022-09-13 op int proxy_dispatch_parent(int, struct privsep_proc *, struct imsg *);
58 c9dc4bc6 2022-09-13 op void proxy_translate_gemtext(struct client *);
59 b0a6bcf7 2022-09-13 op void proxy_resolved(struct asr_result *, void *);
60 b0a6bcf7 2022-09-13 op void proxy_connect(int, short, void *);
61 b0a6bcf7 2022-09-13 op void proxy_read(struct bufferevent *, void *);
62 b0a6bcf7 2022-09-13 op void proxy_write(struct bufferevent *, void *);
63 b0a6bcf7 2022-09-13 op void proxy_error(struct bufferevent *, short, void *);
64 b0a6bcf7 2022-09-13 op int proxy_bufferevent_add(struct event *, int);
65 b0a6bcf7 2022-09-13 op void proxy_tls_writecb(int, short, void *);
66 b0a6bcf7 2022-09-13 op void proxy_tls_readcb(int, short, void *);
67 b0a6bcf7 2022-09-13 op
68 b0a6bcf7 2022-09-13 op static struct privsep_proc procs[] = {
69 b0a6bcf7 2022-09-13 op { "parent", PROC_PARENT, proxy_dispatch_parent },
70 b0a6bcf7 2022-09-13 op };
71 b0a6bcf7 2022-09-13 op
72 b0a6bcf7 2022-09-13 op volatile int proxy_clients;
73 b0a6bcf7 2022-09-13 op volatile int proxy_inflight;
74 b0a6bcf7 2022-09-13 op uint32_t proxy_fcg_id;
75 b0a6bcf7 2022-09-13 op
76 b0a6bcf7 2022-09-13 op void
77 b0a6bcf7 2022-09-13 op proxy(struct privsep *ps, struct privsep_proc *p)
78 b0a6bcf7 2022-09-13 op {
79 b0a6bcf7 2022-09-13 op proc_run(ps, p, procs, nitems(procs), proxy_init, NULL);
80 b0a6bcf7 2022-09-13 op }
81 b0a6bcf7 2022-09-13 op
82 b0a6bcf7 2022-09-13 op void
83 b0a6bcf7 2022-09-13 op proxy_init(struct privsep *ps, struct privsep_proc *p, void *arg)
84 b0a6bcf7 2022-09-13 op {
85 b0a6bcf7 2022-09-13 op if (config_init(ps->ps_env) == -1)
86 b0a6bcf7 2022-09-13 op fatal("failed to initialize configuration");
87 b0a6bcf7 2022-09-13 op
88 b0a6bcf7 2022-09-13 op /* We use a custom shutdown callback */
89 b0a6bcf7 2022-09-13 op /* p->p_shutdown = proxy_shutdown */
90 b0a6bcf7 2022-09-13 op
91 b0a6bcf7 2022-09-13 op if (pledge("stdio recvfd unix inet dns", NULL) == -1)
92 b0a6bcf7 2022-09-13 op fatal("pledge");
93 b0a6bcf7 2022-09-13 op }
94 b0a6bcf7 2022-09-13 op
95 b0a6bcf7 2022-09-13 op int
96 b0a6bcf7 2022-09-13 op proxy_launch(struct galileo *env)
97 b0a6bcf7 2022-09-13 op {
98 b0a6bcf7 2022-09-13 op event_add(&env->sc_evsock, NULL);
99 b0a6bcf7 2022-09-13 op return (0);
100 b0a6bcf7 2022-09-13 op }
101 b0a6bcf7 2022-09-13 op
102 b0a6bcf7 2022-09-13 op void
103 b0a6bcf7 2022-09-13 op proxy_purge(struct server *srv)
104 b0a6bcf7 2022-09-13 op {
105 b0a6bcf7 2022-09-13 op }
106 b0a6bcf7 2022-09-13 op
107 b0a6bcf7 2022-09-13 op void
108 b0a6bcf7 2022-09-13 op proxy_inflight_dec(const char *why)
109 b0a6bcf7 2022-09-13 op {
110 b0a6bcf7 2022-09-13 op proxy_inflight--;
111 b0a6bcf7 2022-09-13 op log_debug("%s: inflight decremented, now %d, %s",
112 b0a6bcf7 2022-09-13 op __func__, proxy_inflight, why);
113 b0a6bcf7 2022-09-13 op }
114 b0a6bcf7 2022-09-13 op
115 b0a6bcf7 2022-09-13 op int
116 b0a6bcf7 2022-09-13 op proxy_dispatch_parent(int fd, struct privsep_proc *p, struct imsg *imsg)
117 b0a6bcf7 2022-09-13 op {
118 b0a6bcf7 2022-09-13 op struct privsep *ps = p->p_ps;
119 b0a6bcf7 2022-09-13 op struct galileo *env = ps->ps_env;
120 b0a6bcf7 2022-09-13 op
121 b0a6bcf7 2022-09-13 op switch (imsg->hdr.type) {
122 b0a6bcf7 2022-09-13 op case IMSG_CFG_SRV:
123 b0a6bcf7 2022-09-13 op if (config_getserver(env, imsg) == -1)
124 b0a6bcf7 2022-09-13 op fatal("config_getproxy");
125 b0a6bcf7 2022-09-13 op break;
126 b0a6bcf7 2022-09-13 op case IMSG_CFG_SOCK:
127 b0a6bcf7 2022-09-13 op /* XXX: improve */
128 b0a6bcf7 2022-09-13 op
129 b0a6bcf7 2022-09-13 op if (env->sc_sock_fd != -1) {
130 b0a6bcf7 2022-09-13 op event_del(&env->sc_evsock);
131 b0a6bcf7 2022-09-13 op close(env->sc_sock_fd);
132 b0a6bcf7 2022-09-13 op }
133 b0a6bcf7 2022-09-13 op
134 b0a6bcf7 2022-09-13 op env->sc_sock_fd = config_getsock(env, imsg);
135 b0a6bcf7 2022-09-13 op if (env->sc_sock_fd == -1)
136 b0a6bcf7 2022-09-13 op fatal("config_getsock");
137 b0a6bcf7 2022-09-13 op
138 b0a6bcf7 2022-09-13 op event_set(&env->sc_evsock, env->sc_sock_fd,
139 ea8874f1 2022-09-13 op EV_READ | EV_PERSIST, fcgi_accept, env);
140 b0a6bcf7 2022-09-13 op event_add(&env->sc_evsock, NULL);
141 ea8874f1 2022-09-13 op evtimer_set(&env->sc_evpause, fcgi_accept, env);
142 b0a6bcf7 2022-09-13 op break;
143 b0a6bcf7 2022-09-13 op case IMSG_CFG_DONE:
144 b0a6bcf7 2022-09-13 op log_debug("config done!");
145 b0a6bcf7 2022-09-13 op break;
146 b0a6bcf7 2022-09-13 op case IMSG_CTL_START:
147 b0a6bcf7 2022-09-13 op proxy_launch(env);
148 b0a6bcf7 2022-09-13 op break;
149 b0a6bcf7 2022-09-13 op default:
150 b0a6bcf7 2022-09-13 op log_warnx("unknown message %d", imsg->hdr.type);
151 b0a6bcf7 2022-09-13 op return (-1);
152 c9dc4bc6 2022-09-13 op }
153 c9dc4bc6 2022-09-13 op
154 c9dc4bc6 2022-09-13 op return (0);
155 c9dc4bc6 2022-09-13 op }
156 c9dc4bc6 2022-09-13 op
157 c9dc4bc6 2022-09-13 op static int
158 c9dc4bc6 2022-09-13 op gemtext_translate_line(struct client *clt, char *line)
159 c9dc4bc6 2022-09-13 op {
160 c9dc4bc6 2022-09-13 op /* preformatted line / closing */
161 b9de0a74 2022-09-14 op if (clt->clt_translate & TR_PRE) {
162 c9dc4bc6 2022-09-13 op if (!strncmp(line, "```", 3)) {
163 b9de0a74 2022-09-14 op clt->clt_translate &= ~TR_PRE;
164 c9dc4bc6 2022-09-13 op return (clt_puts(clt, "</pre>"));
165 c9dc4bc6 2022-09-13 op }
166 c9dc4bc6 2022-09-13 op
167 df2db214 2022-09-19 op if (tp_htmlescape(clt->clt_tp, line) == -1)
168 c9dc4bc6 2022-09-13 op return (-1);
169 c9dc4bc6 2022-09-13 op return (clt_putc(clt, '\n'));
170 b9de0a74 2022-09-14 op }
171 b9de0a74 2022-09-14 op
172 b9de0a74 2022-09-14 op /* bullet */
173 b9de0a74 2022-09-14 op if (!strncmp(line, "* ", 2)) {
174 b9de0a74 2022-09-14 op if (clt->clt_translate & TR_NAV) {
175 b9de0a74 2022-09-14 op if (clt_puts(clt, "</ul></nav>") == -1)
176 b9de0a74 2022-09-14 op return (-1);
177 b9de0a74 2022-09-14 op clt->clt_translate &= ~TR_NAV;
178 b9de0a74 2022-09-14 op }
179 b9de0a74 2022-09-14 op
180 b9de0a74 2022-09-14 op if (!(clt->clt_translate & TR_LIST)) {
181 b9de0a74 2022-09-14 op if (clt_puts(clt, "<ul>") == -1)
182 b9de0a74 2022-09-14 op return (-1);
183 b9de0a74 2022-09-14 op clt->clt_translate |= TR_LIST;
184 b9de0a74 2022-09-14 op }
185 b9de0a74 2022-09-14 op
186 b9de0a74 2022-09-14 op if (clt_puts(clt, "<li>") == -1 ||
187 df2db214 2022-09-19 op tp_htmlescape(clt->clt_tp, line + 2) == -1 ||
188 b9de0a74 2022-09-14 op clt_puts(clt, "</li>") == -1)
189 b9de0a74 2022-09-14 op return (-1);
190 b9de0a74 2022-09-14 op return (0);
191 b9de0a74 2022-09-14 op }
192 b9de0a74 2022-09-14 op
193 b9de0a74 2022-09-14 op if (clt->clt_translate & TR_LIST) {
194 b9de0a74 2022-09-14 op if (clt_puts(clt, "</ul>") == -1)
195 b9de0a74 2022-09-14 op return (-1);
196 b9de0a74 2022-09-14 op clt->clt_translate &= ~TR_LIST;
197 b9de0a74 2022-09-14 op }
198 b9de0a74 2022-09-14 op
199 b9de0a74 2022-09-14 op /* link -- TODO: relativify from SCRIPT_NAME */
200 b9de0a74 2022-09-14 op if (!strncmp(line, "=>", 2)) {
201 b9de0a74 2022-09-14 op char *label;
202 b9de0a74 2022-09-14 op
203 b9de0a74 2022-09-14 op line += 2;
204 b9de0a74 2022-09-14 op line += strspn(line, " \t");
205 b9de0a74 2022-09-14 op
206 b9de0a74 2022-09-14 op label = line + strcspn(line, " \t");
207 b9de0a74 2022-09-14 op if (*label == '\0')
208 b9de0a74 2022-09-14 op label = line;
209 b9de0a74 2022-09-14 op else
210 b9de0a74 2022-09-14 op *label++ = '\0';
211 b9de0a74 2022-09-14 op
212 33f4bf26 2022-09-14 op if (fnmatch("*.jpg", line, 0) == 0 ||
213 33f4bf26 2022-09-14 op fnmatch("*.jpeg", line, 0) == 0 ||
214 33f4bf26 2022-09-14 op fnmatch("*.gif", line, 0) == 0 ||
215 33f4bf26 2022-09-14 op fnmatch("*.png", line, 0) == 0 ||
216 33f4bf26 2022-09-14 op fnmatch("*.svg", line, 0) == 0 ||
217 33f4bf26 2022-09-14 op fnmatch("*.webp", line, 0) == 0) {
218 33f4bf26 2022-09-14 op if (clt->clt_translate & TR_NAV) {
219 33f4bf26 2022-09-14 op if (clt_puts(clt, "</ul></nav>") == -1)
220 33f4bf26 2022-09-14 op return (-1);
221 33f4bf26 2022-09-14 op clt->clt_translate &= ~TR_NAV;
222 33f4bf26 2022-09-14 op }
223 33f4bf26 2022-09-14 op
224 bff58270 2022-09-19 op if (tp_figure(clt->clt_tp, line, label) == -1)
225 33f4bf26 2022-09-14 op return (-1);
226 33f4bf26 2022-09-14 op
227 33f4bf26 2022-09-14 op return (0);
228 33f4bf26 2022-09-14 op }
229 33f4bf26 2022-09-14 op
230 33f4bf26 2022-09-14 op if (!(clt->clt_translate & TR_NAV)) {
231 33f4bf26 2022-09-14 op if (clt_puts(clt, "<nav><ul>") == -1)
232 33f4bf26 2022-09-14 op return (-1);
233 33f4bf26 2022-09-14 op clt->clt_translate |= TR_NAV;
234 33f4bf26 2022-09-14 op }
235 33f4bf26 2022-09-14 op
236 acc266fe 2022-09-19 op if (clt_puts(clt, "<li><a href='") == -1)
237 acc266fe 2022-09-19 op return (-1);
238 acc266fe 2022-09-19 op
239 acc266fe 2022-09-19 op /* XXX: do proper parsing */
240 acc266fe 2022-09-19 op if (*line == '/' || strstr(line, "//") == NULL) {
241 acc266fe 2022-09-19 op if (tp_urlescape(clt->clt_tp,
242 acc266fe 2022-09-19 op clt->clt_script_name) == -1)
243 acc266fe 2022-09-19 op return (-1);
244 acc266fe 2022-09-19 op
245 acc266fe 2022-09-19 op /* skip the first / */
246 acc266fe 2022-09-19 op line++;
247 acc266fe 2022-09-19 op }
248 acc266fe 2022-09-19 op
249 acc266fe 2022-09-19 op if (tp_urlescape(clt->clt_tp, line) == -1 ||
250 b9de0a74 2022-09-14 op clt_puts(clt, "'>") == -1 ||
251 df2db214 2022-09-19 op tp_htmlescape(clt->clt_tp, label) == -1 ||
252 b9de0a74 2022-09-14 op clt_puts(clt, "</a></li>") == -1)
253 b9de0a74 2022-09-14 op return (-1);
254 b9de0a74 2022-09-14 op
255 b9de0a74 2022-09-14 op return (0);
256 c9dc4bc6 2022-09-13 op }
257 c9dc4bc6 2022-09-13 op
258 b9de0a74 2022-09-14 op if (clt->clt_translate & TR_NAV) {
259 b9de0a74 2022-09-14 op if (clt_puts(clt, "</ul></nav>") == -1)
260 b9de0a74 2022-09-14 op return (-1);
261 b9de0a74 2022-09-14 op clt->clt_translate &= ~TR_NAV;
262 b9de0a74 2022-09-14 op }
263 b9de0a74 2022-09-14 op
264 c9dc4bc6 2022-09-13 op /* pre opening */
265 c9dc4bc6 2022-09-13 op if (!strncmp(line, "```", 3)) {
266 b9de0a74 2022-09-14 op clt->clt_translate |= TR_PRE;
267 c9dc4bc6 2022-09-13 op return (clt_puts(clt, "<pre>"));
268 c9dc4bc6 2022-09-13 op }
269 c9dc4bc6 2022-09-13 op
270 c9dc4bc6 2022-09-13 op /* citation block */
271 c9dc4bc6 2022-09-13 op if (*line == '>') {
272 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<blockquote>") == -1 ||
273 df2db214 2022-09-19 op tp_htmlescape(clt->clt_tp, line + 1) == -1 ||
274 c9dc4bc6 2022-09-13 op clt_puts(clt, "</blockquote>") == -1)
275 c9dc4bc6 2022-09-13 op return (-1);
276 c9dc4bc6 2022-09-13 op return (0);
277 c9dc4bc6 2022-09-13 op }
278 c9dc4bc6 2022-09-13 op
279 c9dc4bc6 2022-09-13 op /* headings */
280 c9dc4bc6 2022-09-13 op if (!strncmp(line, "###", 3)) {
281 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<h3>") == -1 ||
282 df2db214 2022-09-19 op tp_htmlescape(clt->clt_tp, line + 3) == -1 ||
283 c9dc4bc6 2022-09-13 op clt_puts(clt, "</h3>") == -1)
284 c9dc4bc6 2022-09-13 op return (-1);
285 c9dc4bc6 2022-09-13 op return (0);
286 c9dc4bc6 2022-09-13 op }
287 c9dc4bc6 2022-09-13 op if (!strncmp(line, "##", 2)) {
288 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<h2>") == -1 ||
289 df2db214 2022-09-19 op tp_htmlescape(clt->clt_tp, line + 2) == -1 ||
290 c9dc4bc6 2022-09-13 op clt_puts(clt, "</h2>") == -1)
291 c9dc4bc6 2022-09-13 op return (-1);
292 c9dc4bc6 2022-09-13 op return (0);
293 b0a6bcf7 2022-09-13 op }
294 c9dc4bc6 2022-09-13 op if (!strncmp(line, "#", 1)) {
295 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<h1>") == -1 ||
296 df2db214 2022-09-19 op tp_htmlescape(clt->clt_tp, line + 1) == -1 ||
297 c9dc4bc6 2022-09-13 op clt_puts(clt, "</h1>") == -1)
298 c9dc4bc6 2022-09-13 op return (-1);
299 c9dc4bc6 2022-09-13 op return (0);
300 c9dc4bc6 2022-09-13 op }
301 df5279ca 2022-09-14 op
302 df5279ca 2022-09-14 op /* Not following strictly the gemini specification... */
303 df5279ca 2022-09-14 op if (*line == '\0')
304 df5279ca 2022-09-14 op return (0);
305 b0a6bcf7 2022-09-13 op
306 c9dc4bc6 2022-09-13 op /* paragraph */
307 c9dc4bc6 2022-09-13 op if (clt_puts(clt, "<p>") == -1 ||
308 df2db214 2022-09-19 op tp_htmlescape(clt->clt_tp, line) == -1 ||
309 c9dc4bc6 2022-09-13 op clt_puts(clt, "</p>") == -1)
310 c9dc4bc6 2022-09-13 op return (-1);
311 b9de0a74 2022-09-14 op
312 b0a6bcf7 2022-09-13 op return (0);
313 b0a6bcf7 2022-09-13 op }
314 b0a6bcf7 2022-09-13 op
315 c9dc4bc6 2022-09-13 op void
316 c9dc4bc6 2022-09-13 op proxy_translate_gemtext(struct client *clt)
317 c9dc4bc6 2022-09-13 op {
318 c9dc4bc6 2022-09-13 op struct bufferevent *bev = clt->clt_bev;
319 c9dc4bc6 2022-09-13 op struct evbuffer *src = EVBUFFER_INPUT(bev);
320 c9dc4bc6 2022-09-13 op char *line;
321 c9dc4bc6 2022-09-13 op size_t len;
322 c9dc4bc6 2022-09-13 op int r;
323 c9dc4bc6 2022-09-13 op
324 c9dc4bc6 2022-09-13 op for (;;) {
325 b9de0a74 2022-09-14 op line = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF);
326 c9dc4bc6 2022-09-13 op if (line == NULL)
327 c9dc4bc6 2022-09-13 op return;
328 c9dc4bc6 2022-09-13 op
329 c9dc4bc6 2022-09-13 op r = gemtext_translate_line(clt, line);
330 c9dc4bc6 2022-09-13 op free(line);
331 c9dc4bc6 2022-09-13 op if (r == -1)
332 c9dc4bc6 2022-09-13 op return;
333 c9dc4bc6 2022-09-13 op }
334 c9dc4bc6 2022-09-13 op }
335 c9dc4bc6 2022-09-13 op
336 b0a6bcf7 2022-09-13 op static struct proxy_config *
337 b0a6bcf7 2022-09-13 op proxy_server_match(struct galileo *env, struct client *clt)
338 b0a6bcf7 2022-09-13 op {
339 b0a6bcf7 2022-09-13 op struct server *srv;
340 b0a6bcf7 2022-09-13 op
341 b0a6bcf7 2022-09-13 op if (clt->clt_server_name == NULL)
342 b0a6bcf7 2022-09-13 op return NULL;
343 b0a6bcf7 2022-09-13 op
344 b0a6bcf7 2022-09-13 op TAILQ_FOREACH(srv, &env->sc_servers, srv_entry) {
345 b0a6bcf7 2022-09-13 op if (!strcmp(clt->clt_server_name, srv->srv_conf.host))
346 b0a6bcf7 2022-09-13 op return &srv->srv_conf;
347 b0a6bcf7 2022-09-13 op }
348 b0a6bcf7 2022-09-13 op
349 b0a6bcf7 2022-09-13 op return NULL;
350 b0a6bcf7 2022-09-13 op }
351 b0a6bcf7 2022-09-13 op
352 b0a6bcf7 2022-09-13 op void
353 b0a6bcf7 2022-09-13 op proxy_start_request(struct galileo *env, struct client *clt)
354 b0a6bcf7 2022-09-13 op {
355 b0a6bcf7 2022-09-13 op struct addrinfo hints;
356 b0a6bcf7 2022-09-13 op struct asr_query *query;
357 b0a6bcf7 2022-09-13 op char port[32];
358 b0a6bcf7 2022-09-13 op
359 b0a6bcf7 2022-09-13 op if ((clt->clt_pc = proxy_server_match(env, clt)) == NULL) {
360 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Status: 501\r\n\r\n") == -1)
361 b0a6bcf7 2022-09-13 op return;
362 b0a6bcf7 2022-09-13 op fcgi_end_request(clt, 1);
363 b0a6bcf7 2022-09-13 op return;
364 b0a6bcf7 2022-09-13 op }
365 b0a6bcf7 2022-09-13 op
366 b0a6bcf7 2022-09-13 op (void)snprintf(port, sizeof(port), "%d", clt->clt_pc->proxy_port);
367 b0a6bcf7 2022-09-13 op
368 b0a6bcf7 2022-09-13 op memset(&hints, 0, sizeof(hints));
369 b0a6bcf7 2022-09-13 op hints.ai_family = AF_UNSPEC;
370 b0a6bcf7 2022-09-13 op hints.ai_socktype = SOCK_STREAM;
371 b0a6bcf7 2022-09-13 op
372 b0a6bcf7 2022-09-13 op query = getaddrinfo_async(clt->clt_pc->proxy_addr, port, &hints, NULL);
373 b0a6bcf7 2022-09-13 op if (query == NULL) {
374 b0a6bcf7 2022-09-13 op log_warn("getaddrinfo_async");
375 b0a6bcf7 2022-09-13 op fcgi_abort_request(clt);
376 b0a6bcf7 2022-09-13 op return;
377 b0a6bcf7 2022-09-13 op }
378 b0a6bcf7 2022-09-13 op
379 b0a6bcf7 2022-09-13 op clt->clt_evasr = event_asr_run(query, proxy_resolved, clt);
380 b0a6bcf7 2022-09-13 op if (clt->clt_evasr == NULL) {
381 b0a6bcf7 2022-09-13 op log_warn("event_asr_run");
382 b0a6bcf7 2022-09-13 op asr_abort(query);
383 b0a6bcf7 2022-09-13 op fcgi_abort_request(clt);
384 b0a6bcf7 2022-09-13 op return;
385 b0a6bcf7 2022-09-13 op }
386 b0a6bcf7 2022-09-13 op }
387 b0a6bcf7 2022-09-13 op
388 b0a6bcf7 2022-09-13 op void
389 b0a6bcf7 2022-09-13 op proxy_resolved(struct asr_result *res, void *d)
390 b0a6bcf7 2022-09-13 op {
391 b0a6bcf7 2022-09-13 op struct client *clt = d;
392 b0a6bcf7 2022-09-13 op struct proxy_config *pc = clt->clt_pc;
393 b0a6bcf7 2022-09-13 op
394 b0a6bcf7 2022-09-13 op clt->clt_evasr = NULL;
395 b0a6bcf7 2022-09-13 op
396 b0a6bcf7 2022-09-13 op if (res->ar_gai_errno != 0) {
397 b0a6bcf7 2022-09-13 op log_warnx("failed to resolve %s:%d: %s",
398 b0a6bcf7 2022-09-13 op pc->proxy_addr, pc->proxy_port,
399 b0a6bcf7 2022-09-13 op gai_strerror(res->ar_gai_errno));
400 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Status: 501\r\n") == -1)
401 b0a6bcf7 2022-09-13 op return;
402 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
403 b0a6bcf7 2022-09-13 op return;
404 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Proxy error; connection failed") == -1)
405 b0a6bcf7 2022-09-13 op return;
406 b0a6bcf7 2022-09-13 op fcgi_end_request(clt, 1);
407 b0a6bcf7 2022-09-13 op return;
408 b0a6bcf7 2022-09-13 op }
409 b0a6bcf7 2022-09-13 op
410 b0a6bcf7 2022-09-13 op clt->clt_addrinfo = res->ar_addrinfo;
411 b0a6bcf7 2022-09-13 op clt->clt_p = clt->clt_addrinfo;
412 b0a6bcf7 2022-09-13 op proxy_connect(-1, 0, clt);
413 b0a6bcf7 2022-09-13 op }
414 b0a6bcf7 2022-09-13 op
415 b0a6bcf7 2022-09-13 op void
416 b0a6bcf7 2022-09-13 op proxy_connect(int fd, short ev, void *d)
417 b0a6bcf7 2022-09-13 op {
418 b0a6bcf7 2022-09-13 op struct client *clt = d;
419 b0a6bcf7 2022-09-13 op struct evbuffer *out;
420 b0a6bcf7 2022-09-13 op struct addrinfo *p;
421 b0a6bcf7 2022-09-13 op struct tls_config *conf;
422 b0a6bcf7 2022-09-13 op struct timeval conntv = {5, 0};
423 b0a6bcf7 2022-09-13 op int err = 0;
424 b0a6bcf7 2022-09-13 op socklen_t len = sizeof(err);
425 b0a6bcf7 2022-09-13 op
426 b0a6bcf7 2022-09-13 op again:
427 b0a6bcf7 2022-09-13 op if (clt->clt_p == NULL)
428 b0a6bcf7 2022-09-13 op goto err;
429 b0a6bcf7 2022-09-13 op
430 b0a6bcf7 2022-09-13 op if (clt->clt_fd != -1) {
431 b0a6bcf7 2022-09-13 op if (getsockopt(clt->clt_fd, SOL_SOCKET, SO_ERROR, &err, &len)
432 b0a6bcf7 2022-09-13 op == -1)
433 b0a6bcf7 2022-09-13 op goto err;
434 b0a6bcf7 2022-09-13 op if (err != 0) {
435 b0a6bcf7 2022-09-13 op errno = err;
436 b0a6bcf7 2022-09-13 op goto err;
437 b0a6bcf7 2022-09-13 op }
438 b0a6bcf7 2022-09-13 op goto done;
439 b0a6bcf7 2022-09-13 op }
440 b0a6bcf7 2022-09-13 op
441 b0a6bcf7 2022-09-13 op p = clt->clt_p;
442 b0a6bcf7 2022-09-13 op clt->clt_fd = socket(p->ai_family, p->ai_socktype | SOCK_NONBLOCK,
443 b0a6bcf7 2022-09-13 op p->ai_protocol);
444 b0a6bcf7 2022-09-13 op if (clt->clt_fd == -1) {
445 b0a6bcf7 2022-09-13 op clt->clt_p = clt->clt_p->ai_next;
446 b0a6bcf7 2022-09-13 op goto again;
447 b0a6bcf7 2022-09-13 op }
448 b0a6bcf7 2022-09-13 op
449 b0a6bcf7 2022-09-13 op if (connect(clt->clt_fd, p->ai_addr, p->ai_addrlen) == 0)
450 b0a6bcf7 2022-09-13 op goto done;
451 b0a6bcf7 2022-09-13 op
452 b0a6bcf7 2022-09-13 op clt->clt_evconn_live = 1;
453 b0a6bcf7 2022-09-13 op event_set(&clt->clt_evconn, clt->clt_fd, EV_WRITE, proxy_connect, clt);
454 b0a6bcf7 2022-09-13 op event_add(&clt->clt_evconn, &conntv);
455 b0a6bcf7 2022-09-13 op return;
456 b0a6bcf7 2022-09-13 op
457 b0a6bcf7 2022-09-13 op done:
458 b0a6bcf7 2022-09-13 op clt->clt_evconn_live = 0;
459 b0a6bcf7 2022-09-13 op freeaddrinfo(clt->clt_addrinfo);
460 b0a6bcf7 2022-09-13 op clt->clt_addrinfo = clt->clt_p = NULL;
461 b0a6bcf7 2022-09-13 op
462 b0a6bcf7 2022-09-13 op /* initialize TLS for Gemini */
463 b0a6bcf7 2022-09-13 op if ((conf = tls_config_new()) == NULL) {
464 b0a6bcf7 2022-09-13 op log_warn("tls_config_new failed");
465 b0a6bcf7 2022-09-13 op goto err;
466 b0a6bcf7 2022-09-13 op }
467 b0a6bcf7 2022-09-13 op
468 b0a6bcf7 2022-09-13 op tls_config_insecure_noverifycert(conf);
469 b0a6bcf7 2022-09-13 op
470 b0a6bcf7 2022-09-13 op if ((clt->clt_ctx = tls_client()) == NULL) {
471 b0a6bcf7 2022-09-13 op log_warnx("tls_client failed");
472 b0a6bcf7 2022-09-13 op tls_config_free(conf);
473 b0a6bcf7 2022-09-13 op goto err;
474 b0a6bcf7 2022-09-13 op }
475 b0a6bcf7 2022-09-13 op
476 b0a6bcf7 2022-09-13 op if (tls_configure(clt->clt_ctx, conf) == -1) {
477 b0a6bcf7 2022-09-13 op log_warnx("tls_configure failed");
478 b0a6bcf7 2022-09-13 op tls_config_free(conf);
479 b0a6bcf7 2022-09-13 op goto err;
480 b0a6bcf7 2022-09-13 op }
481 b0a6bcf7 2022-09-13 op
482 b0a6bcf7 2022-09-13 op tls_config_free(conf);
483 b0a6bcf7 2022-09-13 op
484 b0a6bcf7 2022-09-13 op if (tls_connect_socket(clt->clt_ctx, clt->clt_fd,
485 b0a6bcf7 2022-09-13 op clt->clt_pc->proxy_name) == -1) {
486 b0a6bcf7 2022-09-13 op log_warnx("tls_connect_socket failed");
487 b0a6bcf7 2022-09-13 op goto err;
488 b0a6bcf7 2022-09-13 op }
489 b0a6bcf7 2022-09-13 op
490 b0a6bcf7 2022-09-13 op clt->clt_bev = bufferevent_new(clt->clt_fd, proxy_read, proxy_write,
491 b0a6bcf7 2022-09-13 op proxy_error, clt);
492 b0a6bcf7 2022-09-13 op if (clt->clt_bev == NULL) {
493 b0a6bcf7 2022-09-13 op log_warn("bufferevent_new");
494 b0a6bcf7 2022-09-13 op goto err;
495 b0a6bcf7 2022-09-13 op }
496 b0a6bcf7 2022-09-13 op out = EVBUFFER_OUTPUT(clt->clt_bev);
497 b0a6bcf7 2022-09-13 op
498 b0a6bcf7 2022-09-13 op event_set(&clt->clt_bev->ev_read, clt->clt_fd, EV_READ,
499 b0a6bcf7 2022-09-13 op proxy_tls_readcb, clt->clt_bev);
500 b0a6bcf7 2022-09-13 op event_set(&clt->clt_bev->ev_write, clt->clt_fd, EV_WRITE,
501 b0a6bcf7 2022-09-13 op proxy_tls_writecb, clt->clt_bev);
502 b0a6bcf7 2022-09-13 op
503 b0a6bcf7 2022-09-13 op /* bufferevent_settimeout(); */
504 b0a6bcf7 2022-09-13 op bufferevent_enable(clt->clt_bev, EV_READ|EV_WRITE);
505 b0a6bcf7 2022-09-13 op
506 b0a6bcf7 2022-09-13 op /* TODO: compute the URL */
507 ee80262d 2022-09-13 op if (evbuffer_add_printf(out, "gemini://%s/%s\r\n",
508 ee80262d 2022-09-13 op clt->clt_pc->proxy_name, clt->clt_path_info) == -1) {
509 b0a6bcf7 2022-09-13 op log_warn("bufferevent_printf failed");
510 b0a6bcf7 2022-09-13 op goto err;
511 b0a6bcf7 2022-09-13 op }
512 b0a6bcf7 2022-09-13 op
513 b0a6bcf7 2022-09-13 op return;
514 b0a6bcf7 2022-09-13 op
515 b0a6bcf7 2022-09-13 op err:
516 b0a6bcf7 2022-09-13 op log_warn("failed to connect to %s:%d",
517 b0a6bcf7 2022-09-13 op clt->clt_pc->proxy_addr, clt->clt_pc->proxy_port);
518 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Status: 501\r\n") == -1)
519 b0a6bcf7 2022-09-13 op return;
520 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
521 b0a6bcf7 2022-09-13 op return;
522 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Proxy error; connection failed") == -1)
523 b0a6bcf7 2022-09-13 op return;
524 b0a6bcf7 2022-09-13 op fcgi_end_request(clt, 1);
525 6edd9a00 2022-09-14 op }
526 6edd9a00 2022-09-14 op
527 6edd9a00 2022-09-14 op static inline int
528 6edd9a00 2022-09-14 op parse_mime(struct client *clt, char *mime, char *lang, size_t len)
529 6edd9a00 2022-09-14 op {
530 6edd9a00 2022-09-14 op char *t, *semi;
531 6edd9a00 2022-09-14 op
532 6edd9a00 2022-09-14 op if (strncmp(mime, "text/gemini", 11) != 0)
533 6edd9a00 2022-09-14 op return (0);
534 6edd9a00 2022-09-14 op
535 6edd9a00 2022-09-14 op clt->clt_translate = TR_ENABLED;
536 6edd9a00 2022-09-14 op
537 6edd9a00 2022-09-14 op if ((mime = strchr(mime, ';')) == NULL)
538 6edd9a00 2022-09-14 op return (0);
539 6edd9a00 2022-09-14 op
540 6edd9a00 2022-09-14 op *mime++ = '\0';
541 6edd9a00 2022-09-14 op while ((t = strsep(&mime, ";")) != NULL) {
542 6edd9a00 2022-09-14 op if (!strncmp(t, "charset=", 8)) {
543 6edd9a00 2022-09-14 op t += 8;
544 6edd9a00 2022-09-14 op if (!strncasecmp(t, "utf8", 4) ||
545 8ab2d60e 2022-09-14 op !strncasecmp(t, "utf-8", 5) ||
546 6edd9a00 2022-09-14 op !strncasecmp(t, "ascii", 5)) {
547 6edd9a00 2022-09-14 op log_debug("unknown charset %s", t);
548 6edd9a00 2022-09-14 op return (-1);
549 6edd9a00 2022-09-14 op }
550 6edd9a00 2022-09-14 op continue;
551 6edd9a00 2022-09-14 op }
552 6edd9a00 2022-09-14 op
553 6edd9a00 2022-09-14 op if (!strncmp(t, "lang=", 5)) {
554 6edd9a00 2022-09-14 op t += 5;
555 6edd9a00 2022-09-14 op if ((semi = strchr(t, ';')) != NULL)
556 6edd9a00 2022-09-14 op *semi = '\0';
557 6edd9a00 2022-09-14 op
558 6edd9a00 2022-09-14 op if (strlcpy(lang, t, len) >= len) {
559 6edd9a00 2022-09-14 op log_debug("lang too long: %s", t);
560 6edd9a00 2022-09-14 op *lang = '\0';
561 6edd9a00 2022-09-14 op }
562 6edd9a00 2022-09-14 op
563 6edd9a00 2022-09-14 op if (semi)
564 6edd9a00 2022-09-14 op *semi = ';';
565 6edd9a00 2022-09-14 op continue;
566 6edd9a00 2022-09-14 op }
567 bff58270 2022-09-19 op }
568 bff58270 2022-09-19 op
569 bff58270 2022-09-19 op return (0);
570 bff58270 2022-09-19 op }
571 bff58270 2022-09-19 op
572 bff58270 2022-09-19 op static int
573 bff58270 2022-09-19 op proxy_tp_puts(struct template *tp, const char *c)
574 bff58270 2022-09-19 op {
575 bff58270 2022-09-19 op struct client *clt = tp->tp_arg;
576 bff58270 2022-09-19 op
577 bff58270 2022-09-19 op if (clt_puts(clt, c) == -1) {
578 bff58270 2022-09-19 op tp->tp_ret = -1;
579 bff58270 2022-09-19 op return (-1);
580 6edd9a00 2022-09-14 op }
581 bff58270 2022-09-19 op return (0);
582 bff58270 2022-09-19 op }
583 6edd9a00 2022-09-14 op
584 bff58270 2022-09-19 op static int
585 bff58270 2022-09-19 op proxy_tp_putc(struct template *tp, int c)
586 bff58270 2022-09-19 op {
587 bff58270 2022-09-19 op struct client *clt = tp->tp_arg;
588 bff58270 2022-09-19 op
589 bff58270 2022-09-19 op if (clt_putc(clt, c) == -1) {
590 bff58270 2022-09-19 op tp->tp_ret = -1;
591 bff58270 2022-09-19 op return (-1);
592 bff58270 2022-09-19 op }
593 6edd9a00 2022-09-14 op return (0);
594 b0a6bcf7 2022-09-13 op }
595 b0a6bcf7 2022-09-13 op
596 b0a6bcf7 2022-09-13 op void
597 b0a6bcf7 2022-09-13 op proxy_read(struct bufferevent *bev, void *d)
598 b0a6bcf7 2022-09-13 op {
599 b0a6bcf7 2022-09-13 op struct client *clt = d;
600 fc1cb9ea 2022-09-14 op struct proxy_config *pc = clt->clt_pc;
601 b0a6bcf7 2022-09-13 op struct evbuffer *src = EVBUFFER_INPUT(bev);
602 c9dc4bc6 2022-09-13 op const char *ctype;
603 6edd9a00 2022-09-14 op char lang[16];
604 6edd9a00 2022-09-14 op char *hdr, *mime;
605 b0a6bcf7 2022-09-13 op size_t len;
606 b0a6bcf7 2022-09-13 op
607 b0a6bcf7 2022-09-13 op if (clt->clt_headersdone) {
608 c9dc4bc6 2022-09-13 op if (clt->clt_translate)
609 c9dc4bc6 2022-09-13 op proxy_translate_gemtext(clt);
610 c9dc4bc6 2022-09-13 op else
611 c9dc4bc6 2022-09-13 op clt_write_bufferevent(clt, bev);
612 b0a6bcf7 2022-09-13 op return;
613 b0a6bcf7 2022-09-13 op }
614 b0a6bcf7 2022-09-13 op
615 b0a6bcf7 2022-09-13 op hdr = evbuffer_readln(src, &len, EVBUFFER_EOL_CRLF_STRICT);
616 b0a6bcf7 2022-09-13 op if (hdr == NULL) {
617 b0a6bcf7 2022-09-13 op if (EVBUFFER_LENGTH(src) >= 1026)
618 b0a6bcf7 2022-09-13 op proxy_error(bev, EV_READ, clt);
619 b0a6bcf7 2022-09-13 op return;
620 b0a6bcf7 2022-09-13 op }
621 b0a6bcf7 2022-09-13 op
622 b0a6bcf7 2022-09-13 op if (len < 4 ||
623 b0a6bcf7 2022-09-13 op !isdigit((unsigned char)hdr[0]) ||
624 b0a6bcf7 2022-09-13 op !isdigit((unsigned char)hdr[1]) ||
625 b0a6bcf7 2022-09-13 op hdr[2] != ' ') {
626 b0a6bcf7 2022-09-13 op log_warnx("invalid ");
627 b0a6bcf7 2022-09-13 op proxy_error(bev, EV_READ, clt);
628 1337dc2b 2022-09-14 op goto err;
629 b0a6bcf7 2022-09-13 op }
630 b0a6bcf7 2022-09-13 op
631 6e8f426e 2022-09-14 op switch (hdr[0]) {
632 6e8f426e 2022-09-14 op case '2':
633 6e8f426e 2022-09-14 op /* handled below */
634 6e8f426e 2022-09-14 op break;
635 6e8f426e 2022-09-14 op default:
636 6e8f426e 2022-09-14 op if (clt_puts(clt, "Status: 501\r\n") == -1)
637 1337dc2b 2022-09-14 op goto err;
638 6e8f426e 2022-09-14 op if (clt_puts(clt,
639 8ab2d60e 2022-09-14 op "Content-Type: text/plain;charset=utf-8\r\n") == -1)
640 1337dc2b 2022-09-14 op goto err;
641 6e8f426e 2022-09-14 op if (clt_puts(clt, "\r\n") == -1)
642 1337dc2b 2022-09-14 op goto err;
643 6e8f426e 2022-09-14 op if (clt_printf(clt, "Request failed with code %c%c\n\n",
644 6e8f426e 2022-09-14 op hdr[0], hdr[1]) == -1)
645 1337dc2b 2022-09-14 op goto err;
646 6e8f426e 2022-09-14 op if (clt_printf(clt, "The server says: %s\n", &hdr[3]) == -1)
647 1337dc2b 2022-09-14 op goto err;
648 6e8f426e 2022-09-14 op fcgi_end_request(clt, 1);
649 1337dc2b 2022-09-14 op goto err;
650 b0a6bcf7 2022-09-13 op }
651 b0a6bcf7 2022-09-13 op
652 6edd9a00 2022-09-14 op mime = hdr + 2 + strspn(hdr + 2, " \t");
653 6edd9a00 2022-09-14 op if (parse_mime(clt, mime, lang, sizeof(lang)) == -1) {
654 6edd9a00 2022-09-14 op if (clt_puts(clt, "Status: 501\r\n") == -1)
655 1337dc2b 2022-09-14 op goto err;
656 6edd9a00 2022-09-14 op if (clt_puts(clt,
657 8ab2d60e 2022-09-14 op "Content-Type: text/plain;charset=utf-8\r\n") == -1)
658 1337dc2b 2022-09-14 op goto err;
659 6edd9a00 2022-09-14 op if (clt_puts(clt, "\r\n") == -1)
660 1337dc2b 2022-09-14 op goto err;
661 6edd9a00 2022-09-14 op if (clt_printf(clt, "Failed to parse the Gemini response\n")
662 6edd9a00 2022-09-14 op == -1)
663 6edd9a00 2022-09-14 op fcgi_end_request(clt, 1);
664 1337dc2b 2022-09-14 op goto err;
665 6edd9a00 2022-09-14 op }
666 6edd9a00 2022-09-14 op
667 bff58270 2022-09-19 op if (clt->clt_translate) {
668 bff58270 2022-09-19 op clt->clt_tp = template(clt, proxy_tp_puts, proxy_tp_putc);
669 bff58270 2022-09-19 op
670 8ab2d60e 2022-09-14 op ctype = "text/html;charset=utf-8";
671 bff58270 2022-09-19 op } else
672 6edd9a00 2022-09-14 op ctype = mime;
673 6edd9a00 2022-09-14 op
674 6edd9a00 2022-09-14 op if (clt_printf(clt, "Content-Type: %s\r\n\r\n", ctype) == -1)
675 1337dc2b 2022-09-14 op goto err;
676 b0a6bcf7 2022-09-13 op
677 b0a6bcf7 2022-09-13 op clt->clt_headersdone = 1;
678 c9dc4bc6 2022-09-13 op
679 bff58270 2022-09-19 op if (clt->clt_translate &&
680 bff58270 2022-09-19 op tp_head(clt->clt_tp, lang, pc->stylesheet) == -1)
681 bff58270 2022-09-19 op goto err;
682 c9dc4bc6 2022-09-13 op
683 1337dc2b 2022-09-14 op /*
684 1337dc2b 2022-09-14 op * Trigger the read again so we proceed with the response
685 1337dc2b 2022-09-14 op * body, if any.
686 1337dc2b 2022-09-14 op */
687 1337dc2b 2022-09-14 op free(hdr);
688 1337dc2b 2022-09-14 op proxy_read(bev, d);
689 1337dc2b 2022-09-14 op return;
690 1337dc2b 2022-09-14 op
691 1337dc2b 2022-09-14 op err:
692 1337dc2b 2022-09-14 op free(hdr);
693 b0a6bcf7 2022-09-13 op }
694 b0a6bcf7 2022-09-13 op
695 b0a6bcf7 2022-09-13 op void
696 b0a6bcf7 2022-09-13 op proxy_write(struct bufferevent *bev, void *d)
697 b0a6bcf7 2022-09-13 op {
698 b0a6bcf7 2022-09-13 op return;
699 b0a6bcf7 2022-09-13 op }
700 b0a6bcf7 2022-09-13 op
701 b0a6bcf7 2022-09-13 op void
702 b0a6bcf7 2022-09-13 op proxy_error(struct bufferevent *bev, short err, void *d)
703 b0a6bcf7 2022-09-13 op {
704 b0a6bcf7 2022-09-13 op struct client *clt = d;
705 b0a6bcf7 2022-09-13 op int status = !(err & EVBUFFER_EOF);
706 b0a6bcf7 2022-09-13 op
707 b0a6bcf7 2022-09-13 op log_debug("proxy error, shutting down the connection (err: %x)",
708 b0a6bcf7 2022-09-13 op err);
709 b0a6bcf7 2022-09-13 op
710 b0a6bcf7 2022-09-13 op if (!clt->clt_headersdone) {
711 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Status: 501\r\n") == -1)
712 b0a6bcf7 2022-09-13 op return;
713 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Content-Type: text/plain\r\n") == -1)
714 b0a6bcf7 2022-09-13 op return;
715 b0a6bcf7 2022-09-13 op if (clt_printf(clt, "Proxy error\n") == -1)
716 b0a6bcf7 2022-09-13 op return;
717 c9dc4bc6 2022-09-13 op } else if (status == 0) {
718 c42fce9b 2022-09-14 op if (clt->clt_translate & TR_PRE) {
719 c42fce9b 2022-09-14 op if (clt_puts(clt, "</pre>"))
720 c42fce9b 2022-09-14 op return;
721 c42fce9b 2022-09-14 op clt->clt_translate &= ~TR_PRE;
722 c42fce9b 2022-09-14 op }
723 c42fce9b 2022-09-14 op
724 c42fce9b 2022-09-14 op if (clt->clt_translate & TR_LIST) {
725 c42fce9b 2022-09-14 op if (clt_puts(clt, "</ul>") == -1)
726 c42fce9b 2022-09-14 op return;
727 c42fce9b 2022-09-14 op clt->clt_translate &= ~TR_LIST;
728 c42fce9b 2022-09-14 op }
729 c42fce9b 2022-09-14 op
730 c42fce9b 2022-09-14 op if (clt->clt_translate & TR_NAV) {
731 c42fce9b 2022-09-14 op if (clt_puts(clt, "</ul></nav>") == -1)
732 c42fce9b 2022-09-14 op return;
733 c42fce9b 2022-09-14 op clt->clt_translate &= ~TR_NAV;
734 c42fce9b 2022-09-14 op }
735 c42fce9b 2022-09-14 op
736 54acddc9 2022-09-19 op if (clt->clt_translate &&
737 54acddc9 2022-09-19 op tp_foot(clt->clt_tp) == -1)
738 8fa3006a 2022-09-14 op return;
739 b0a6bcf7 2022-09-13 op }
740 b0a6bcf7 2022-09-13 op
741 b0a6bcf7 2022-09-13 op fcgi_end_request(clt, status);
742 b0a6bcf7 2022-09-13 op }
743 b0a6bcf7 2022-09-13 op
744 b0a6bcf7 2022-09-13 op void
745 b0a6bcf7 2022-09-13 op proxy_tls_readcb(int fd, short event, void *arg)
746 b0a6bcf7 2022-09-13 op {
747 b0a6bcf7 2022-09-13 op struct bufferevent *bufev = arg;
748 b0a6bcf7 2022-09-13 op struct client *clt = bufev->cbarg;
749 b0a6bcf7 2022-09-13 op char rbuf[IBUF_READ_SIZE];
750 b0a6bcf7 2022-09-13 op int what = EVBUFFER_READ;
751 b0a6bcf7 2022-09-13 op int howmuch = IBUF_READ_SIZE;
752 b0a6bcf7 2022-09-13 op ssize_t ret;
753 b0a6bcf7 2022-09-13 op size_t len;
754 b0a6bcf7 2022-09-13 op
755 b0a6bcf7 2022-09-13 op if (event == EV_TIMEOUT) {
756 b0a6bcf7 2022-09-13 op what |= EVBUFFER_TIMEOUT;
757 b0a6bcf7 2022-09-13 op goto err;
758 b0a6bcf7 2022-09-13 op }
759 b0a6bcf7 2022-09-13 op
760 b0a6bcf7 2022-09-13 op if (bufev->wm_read.high != 0)
761 b0a6bcf7 2022-09-13 op howmuch = MINIMUM(sizeof(rbuf), bufev->wm_read.high);
762 b0a6bcf7 2022-09-13 op
763 b0a6bcf7 2022-09-13 op ret = tls_read(clt->clt_ctx, rbuf, howmuch);
764 b0a6bcf7 2022-09-13 op if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
765 b0a6bcf7 2022-09-13 op goto retry;
766 b0a6bcf7 2022-09-13 op } else if (ret == -1) {
767 b0a6bcf7 2022-09-13 op what |= EVBUFFER_ERROR;
768 b0a6bcf7 2022-09-13 op goto err;
769 b0a6bcf7 2022-09-13 op }
770 b0a6bcf7 2022-09-13 op len = ret;
771 b0a6bcf7 2022-09-13 op
772 b0a6bcf7 2022-09-13 op if (len == 0) {
773 b0a6bcf7 2022-09-13 op what |= EVBUFFER_EOF;
774 b0a6bcf7 2022-09-13 op goto err;
775 b0a6bcf7 2022-09-13 op }
776 b0a6bcf7 2022-09-13 op
777 b0a6bcf7 2022-09-13 op if (evbuffer_add(bufev->input, rbuf, len) == -1) {
778 b0a6bcf7 2022-09-13 op what |= EVBUFFER_ERROR;
779 b0a6bcf7 2022-09-13 op goto err;
780 b0a6bcf7 2022-09-13 op }
781 b0a6bcf7 2022-09-13 op
782 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
783 b0a6bcf7 2022-09-13 op
784 b0a6bcf7 2022-09-13 op len = EVBUFFER_LENGTH(bufev->input);
785 b0a6bcf7 2022-09-13 op if (bufev->wm_read.low != 0 && len < bufev->wm_read.low)
786 b0a6bcf7 2022-09-13 op return;
787 b0a6bcf7 2022-09-13 op if (bufev->wm_read.high != 0 && len > bufev->wm_read.high) {
788 b0a6bcf7 2022-09-13 op struct evbuffer *buf = bufev->input;
789 b0a6bcf7 2022-09-13 op event_del(&bufev->ev_read);
790 b0a6bcf7 2022-09-13 op evbuffer_setcb(buf, bufferevent_read_pressure_cb, bufev);
791 b0a6bcf7 2022-09-13 op return;
792 b0a6bcf7 2022-09-13 op }
793 b0a6bcf7 2022-09-13 op
794 b0a6bcf7 2022-09-13 op if (bufev->readcb != NULL)
795 b0a6bcf7 2022-09-13 op (*bufev->readcb)(bufev, bufev->cbarg);
796 b0a6bcf7 2022-09-13 op return;
797 b0a6bcf7 2022-09-13 op
798 b0a6bcf7 2022-09-13 op retry:
799 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(&bufev->ev_read, bufev->timeout_read);
800 b0a6bcf7 2022-09-13 op return;
801 b0a6bcf7 2022-09-13 op
802 b0a6bcf7 2022-09-13 op err:
803 b0a6bcf7 2022-09-13 op (*bufev->errorcb)(bufev, what, bufev->cbarg);
804 b0a6bcf7 2022-09-13 op }
805 b0a6bcf7 2022-09-13 op
806 b0a6bcf7 2022-09-13 op int
807 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(struct event *ev, int timeout)
808 b0a6bcf7 2022-09-13 op {
809 b0a6bcf7 2022-09-13 op struct timeval tv, *ptv = NULL;
810 b0a6bcf7 2022-09-13 op
811 b0a6bcf7 2022-09-13 op if (timeout) {
812 b0a6bcf7 2022-09-13 op timerclear(&tv);
813 b0a6bcf7 2022-09-13 op tv.tv_sec = timeout;
814 b0a6bcf7 2022-09-13 op ptv = &tv;
815 b0a6bcf7 2022-09-13 op }
816 b0a6bcf7 2022-09-13 op
817 b0a6bcf7 2022-09-13 op return (event_add(ev, ptv));
818 b0a6bcf7 2022-09-13 op }
819 b0a6bcf7 2022-09-13 op
820 b0a6bcf7 2022-09-13 op void
821 b0a6bcf7 2022-09-13 op proxy_tls_writecb(int fd, short event, void *arg)
822 b0a6bcf7 2022-09-13 op {
823 b0a6bcf7 2022-09-13 op struct bufferevent *bufev = arg;
824 b0a6bcf7 2022-09-13 op struct client *clt = bufev->cbarg;
825 b0a6bcf7 2022-09-13 op ssize_t ret;
826 b0a6bcf7 2022-09-13 op short what = EVBUFFER_WRITE;
827 b0a6bcf7 2022-09-13 op size_t len;
828 b0a6bcf7 2022-09-13 op
829 b0a6bcf7 2022-09-13 op if (event == EV_TIMEOUT) {
830 b0a6bcf7 2022-09-13 op what |= EVBUFFER_TIMEOUT;
831 b0a6bcf7 2022-09-13 op goto err;
832 b0a6bcf7 2022-09-13 op }
833 b0a6bcf7 2022-09-13 op
834 b0a6bcf7 2022-09-13 op if (EVBUFFER_LENGTH(bufev->output)) {
835 b0a6bcf7 2022-09-13 op ret = tls_write(clt->clt_ctx,
836 b0a6bcf7 2022-09-13 op EVBUFFER_DATA(bufev->output),
837 b0a6bcf7 2022-09-13 op EVBUFFER_LENGTH(bufev->output));
838 b0a6bcf7 2022-09-13 op if (ret == TLS_WANT_POLLIN || ret == TLS_WANT_POLLOUT) {
839 b0a6bcf7 2022-09-13 op goto retry;
840 b0a6bcf7 2022-09-13 op } else if (ret == -1) {
841 b0a6bcf7 2022-09-13 op what |= EVBUFFER_ERROR;
842 b0a6bcf7 2022-09-13 op goto err;
843 b0a6bcf7 2022-09-13 op }
844 b0a6bcf7 2022-09-13 op len = ret;
845 b0a6bcf7 2022-09-13 op evbuffer_drain(bufev->output, len);
846 b0a6bcf7 2022-09-13 op }
847 b0a6bcf7 2022-09-13 op
848 b0a6bcf7 2022-09-13 op if (EVBUFFER_LENGTH(bufev->output) != 0)
849 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
850 b0a6bcf7 2022-09-13 op
851 b0a6bcf7 2022-09-13 op if (bufev->writecb != NULL &&
852 b0a6bcf7 2022-09-13 op EVBUFFER_LENGTH(bufev->output) <= bufev->wm_write.low)
853 b0a6bcf7 2022-09-13 op (*bufev->writecb)(bufev, bufev->cbarg);
854 b0a6bcf7 2022-09-13 op return;
855 b0a6bcf7 2022-09-13 op
856 b0a6bcf7 2022-09-13 op retry:
857 b0a6bcf7 2022-09-13 op proxy_bufferevent_add(&bufev->ev_write, bufev->timeout_write);
858 b0a6bcf7 2022-09-13 op return;
859 b0a6bcf7 2022-09-13 op
860 b0a6bcf7 2022-09-13 op err:
861 b0a6bcf7 2022-09-13 op (*bufev->errorcb)(bufev, what, bufev->cbarg);
862 b0a6bcf7 2022-09-13 op }
863 b0a6bcf7 2022-09-13 op
864 b0a6bcf7 2022-09-13 op void
865 b0a6bcf7 2022-09-13 op proxy_client_free(struct client *clt)
866 b0a6bcf7 2022-09-13 op {
867 b0a6bcf7 2022-09-13 op if (clt->clt_evasr)
868 b0a6bcf7 2022-09-13 op event_asr_abort(clt->clt_evasr);
869 b0a6bcf7 2022-09-13 op
870 b0a6bcf7 2022-09-13 op if (clt->clt_addrinfo)
871 b0a6bcf7 2022-09-13 op freeaddrinfo(clt->clt_addrinfo);
872 b0a6bcf7 2022-09-13 op
873 b0a6bcf7 2022-09-13 op if (clt->clt_evconn_live)
874 b0a6bcf7 2022-09-13 op event_del(&clt->clt_evconn);
875 b0a6bcf7 2022-09-13 op
876 b0a6bcf7 2022-09-13 op if (clt->clt_fd != -1)
877 b0a6bcf7 2022-09-13 op close(clt->clt_fd);
878 b0a6bcf7 2022-09-13 op
879 b0a6bcf7 2022-09-13 op if (clt->clt_ctx)
880 b0a6bcf7 2022-09-13 op tls_free(clt->clt_ctx);
881 b0a6bcf7 2022-09-13 op
882 b0a6bcf7 2022-09-13 op if (clt->clt_bev)
883 b0a6bcf7 2022-09-13 op bufferevent_free(clt->clt_bev);
884 b0a6bcf7 2022-09-13 op
885 bff58270 2022-09-19 op free(clt->clt_tp);
886 b0a6bcf7 2022-09-13 op free(clt->clt_server_name);
887 b0a6bcf7 2022-09-13 op free(clt->clt_script_name);
888 b0a6bcf7 2022-09-13 op free(clt->clt_path_info);
889 b0a6bcf7 2022-09-13 op free(clt);
890 b0a6bcf7 2022-09-13 op }