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  *
  4 b0a6bcf7 2022-09-13 op  * Permission to use, copy, modify, and distribute this software for any
  5 b0a6bcf7 2022-09-13 op  * purpose with or without fee is hereby granted, provided that the above
  6 b0a6bcf7 2022-09-13 op  * copyright notice and this permission notice appear in all copies.
  7 b0a6bcf7 2022-09-13 op  *
  8 b0a6bcf7 2022-09-13 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 b0a6bcf7 2022-09-13 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 b0a6bcf7 2022-09-13 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 b0a6bcf7 2022-09-13 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 b0a6bcf7 2022-09-13 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 b0a6bcf7 2022-09-13 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 b0a6bcf7 2022-09-13 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 b0a6bcf7 2022-09-13 op  */
 16 b0a6bcf7 2022-09-13 op 
 17 b0a6bcf7 2022-09-13 op #include <sys/queue.h>
 18 b0a6bcf7 2022-09-13 op #include <sys/tree.h>
 19 b0a6bcf7 2022-09-13 op #include <sys/socket.h>
 20 b0a6bcf7 2022-09-13 op 
 21 b0a6bcf7 2022-09-13 op #include <netinet/in.h>
 22 b0a6bcf7 2022-09-13 op #include <arpa/inet.h>
 23 b0a6bcf7 2022-09-13 op 
 24 b0a6bcf7 2022-09-13 op #include <errno.h>
 25 b0a6bcf7 2022-09-13 op #include <event.h>
 26 b0a6bcf7 2022-09-13 op #include <limits.h>
 27 b0a6bcf7 2022-09-13 op #include <stdarg.h>
 28 b0a6bcf7 2022-09-13 op #include <stdio.h>
 29 b0a6bcf7 2022-09-13 op #include <stdlib.h>
 30 b0a6bcf7 2022-09-13 op #include <stdint.h>
 31 b0a6bcf7 2022-09-13 op #include <string.h>
 32 b0a6bcf7 2022-09-13 op #include <unistd.h>
 33 b0a6bcf7 2022-09-13 op 
 34 b0a6bcf7 2022-09-13 op #include "log.h"
 35 232eee45 2022-09-20 op #include "tmpl.h"
 36 b0a6bcf7 2022-09-13 op 
 37 b0a6bcf7 2022-09-13 op #include "galileo.h"
 38 b0a6bcf7 2022-09-13 op 
 39 b0a6bcf7 2022-09-13 op #define MIN(a, b)	((a) < (b) ? (a) : (b))
 40 b0a6bcf7 2022-09-13 op 
 41 b0a6bcf7 2022-09-13 op struct fcgi_header {
 42 b0a6bcf7 2022-09-13 op 	unsigned char version;
 43 b0a6bcf7 2022-09-13 op 	unsigned char type;
 44 b0a6bcf7 2022-09-13 op 	unsigned char req_id1;
 45 b0a6bcf7 2022-09-13 op 	unsigned char req_id0;
 46 b0a6bcf7 2022-09-13 op 	unsigned char content_len1;
 47 b0a6bcf7 2022-09-13 op 	unsigned char content_len0;
 48 b0a6bcf7 2022-09-13 op 	unsigned char padding;
 49 b0a6bcf7 2022-09-13 op 	unsigned char reserved;
 50 b0a6bcf7 2022-09-13 op } __attribute__((packed));
 51 b0a6bcf7 2022-09-13 op 
 52 b0a6bcf7 2022-09-13 op /*
 53 b0a6bcf7 2022-09-13 op  * number of bytes in a FCGI_HEADER.  Future version of the protocol
 54 b0a6bcf7 2022-09-13 op  * will not reduce this number.
 55 b0a6bcf7 2022-09-13 op  */
 56 b0a6bcf7 2022-09-13 op #define FCGI_HEADER_LEN	8
 57 b0a6bcf7 2022-09-13 op 
 58 b0a6bcf7 2022-09-13 op /*
 59 b0a6bcf7 2022-09-13 op  * values for the version component
 60 b0a6bcf7 2022-09-13 op  */
 61 b0a6bcf7 2022-09-13 op #define FCGI_VERSION_1	1
 62 b0a6bcf7 2022-09-13 op 
 63 b0a6bcf7 2022-09-13 op /*
 64 b0a6bcf7 2022-09-13 op  * values for the type component
 65 b0a6bcf7 2022-09-13 op  */
 66 b0a6bcf7 2022-09-13 op #define FCGI_BEGIN_REQUEST	 1
 67 b0a6bcf7 2022-09-13 op #define FCGI_ABORT_REQUEST	 2
 68 b0a6bcf7 2022-09-13 op #define FCGI_END_REQUEST	 3
 69 b0a6bcf7 2022-09-13 op #define FCGI_PARAMS		 4
 70 b0a6bcf7 2022-09-13 op #define FCGI_STDIN		 5
 71 b0a6bcf7 2022-09-13 op #define FCGI_STDOUT		 6
 72 b0a6bcf7 2022-09-13 op #define FCGI_STDERR		 7
 73 b0a6bcf7 2022-09-13 op #define FCGI_DATA		 8
 74 b0a6bcf7 2022-09-13 op #define FCGI_GET_VALUES		 9
 75 b0a6bcf7 2022-09-13 op #define FCGI_GET_VALUES_RESULT	10
 76 b0a6bcf7 2022-09-13 op #define FCGI_UNKNOWN_TYPE	11
 77 b0a6bcf7 2022-09-13 op #define FCGI_MAXTYPE		(FCGI_UNKNOWN_TYPE)
 78 b0a6bcf7 2022-09-13 op 
 79 b0a6bcf7 2022-09-13 op struct fcgi_begin_req {
 80 b0a6bcf7 2022-09-13 op 	unsigned char role1;
 81 b0a6bcf7 2022-09-13 op 	unsigned char role0;
 82 b0a6bcf7 2022-09-13 op 	unsigned char flags;
 83 b0a6bcf7 2022-09-13 op 	unsigned char reserved[5];
 84 b0a6bcf7 2022-09-13 op };
 85 b0a6bcf7 2022-09-13 op 
 86 b0a6bcf7 2022-09-13 op struct fcgi_begin_req_record {
 87 b0a6bcf7 2022-09-13 op 	struct fcgi_header	header;
 88 b0a6bcf7 2022-09-13 op 	struct fcgi_begin_req	body;
 89 b0a6bcf7 2022-09-13 op };
 90 b0a6bcf7 2022-09-13 op 
 91 b0a6bcf7 2022-09-13 op /*
 92 b0a6bcf7 2022-09-13 op  * mask for flags;
 93 b0a6bcf7 2022-09-13 op  */
 94 b0a6bcf7 2022-09-13 op #define FCGI_KEEP_CONN		1
 95 b0a6bcf7 2022-09-13 op 
 96 b0a6bcf7 2022-09-13 op /*
 97 b0a6bcf7 2022-09-13 op  * values for the role
 98 b0a6bcf7 2022-09-13 op  */
 99 b0a6bcf7 2022-09-13 op #define FCGI_RESPONDER	1
100 b0a6bcf7 2022-09-13 op #define FCGI_AUTHORIZER	2
101 b0a6bcf7 2022-09-13 op #define FCGI_FILTER	3
102 b0a6bcf7 2022-09-13 op 
103 b0a6bcf7 2022-09-13 op struct fcgi_end_req_body {
104 b0a6bcf7 2022-09-13 op 	unsigned char app_status3;
105 b0a6bcf7 2022-09-13 op 	unsigned char app_status2;
106 b0a6bcf7 2022-09-13 op 	unsigned char app_status1;
107 b0a6bcf7 2022-09-13 op 	unsigned char app_status0;
108 b0a6bcf7 2022-09-13 op 	unsigned char proto_status;
109 b0a6bcf7 2022-09-13 op 	unsigned char reserved[3];
110 b0a6bcf7 2022-09-13 op };
111 b0a6bcf7 2022-09-13 op 
112 b0a6bcf7 2022-09-13 op /*
113 b0a6bcf7 2022-09-13 op  * values for proto_status
114 b0a6bcf7 2022-09-13 op  */
115 b0a6bcf7 2022-09-13 op #define FCGI_REQUEST_COMPLETE	0
116 b0a6bcf7 2022-09-13 op #define FCGI_CANT_MPX_CONN	1
117 b0a6bcf7 2022-09-13 op #define FCGI_OVERLOADED		2
118 b0a6bcf7 2022-09-13 op #define FCGI_UNKNOWN_ROLE	3
119 b0a6bcf7 2022-09-13 op 
120 b0a6bcf7 2022-09-13 op /*
121 b0a6bcf7 2022-09-13 op  * Variable names for FCGI_GET_VALUES / FCGI_GET_VALUES_RESULT
122 b0a6bcf7 2022-09-13 op  * records.
123 b0a6bcf7 2022-09-13 op  */
124 b0a6bcf7 2022-09-13 op #define FCGI_MAX_CONNS	"FCGI_MAX_CONNS"
125 b0a6bcf7 2022-09-13 op #define FCGI_MAX_REQS	"FCGI_MAX_REQS"
126 b0a6bcf7 2022-09-13 op #define FCGI_MPXS_CONNS	"FCGI_MPXS_CONNS"
127 b0a6bcf7 2022-09-13 op 
128 b0a6bcf7 2022-09-13 op #define CAT(f0, f1)	((f0) + ((f1) << 8))
129 b0a6bcf7 2022-09-13 op 
130 b0a6bcf7 2022-09-13 op enum {
131 b0a6bcf7 2022-09-13 op 	FCGI_RECORD_HEADER,
132 b0a6bcf7 2022-09-13 op 	FCGI_RECORD_BODY,
133 b0a6bcf7 2022-09-13 op };
134 b0a6bcf7 2022-09-13 op 
135 b0a6bcf7 2022-09-13 op volatile int fcgi_inflight;
136 b0a6bcf7 2022-09-13 op 
137 b0a6bcf7 2022-09-13 op static int
138 b0a6bcf7 2022-09-13 op fcgi_send_end_req(struct fcgi *fcgi, int id, int as, int ps)
139 b0a6bcf7 2022-09-13 op {
140 b0a6bcf7 2022-09-13 op 	struct bufferevent	*bev = fcgi->fcg_bev;
141 b0a6bcf7 2022-09-13 op 	struct fcgi_header	 hdr;
142 b0a6bcf7 2022-09-13 op 	struct fcgi_end_req_body end;
143 b0a6bcf7 2022-09-13 op 
144 b0a6bcf7 2022-09-13 op 	memset(&hdr, 0, sizeof(hdr));
145 b0a6bcf7 2022-09-13 op 	memset(&end, 0, sizeof(end));
146 b0a6bcf7 2022-09-13 op 
147 b0a6bcf7 2022-09-13 op 	hdr.version = FCGI_VERSION_1;
148 b0a6bcf7 2022-09-13 op 	hdr.type = FCGI_END_REQUEST;
149 b0a6bcf7 2022-09-13 op 	hdr.req_id0 = (id & 0xFF);
150 b0a6bcf7 2022-09-13 op 	hdr.req_id1 = (id >> 8);
151 b0a6bcf7 2022-09-13 op 	hdr.content_len0 = sizeof(end);
152 b0a6bcf7 2022-09-13 op 
153 b0a6bcf7 2022-09-13 op 	end.app_status0 = (unsigned char)as;
154 b0a6bcf7 2022-09-13 op 	end.proto_status = (unsigned char)ps;
155 b0a6bcf7 2022-09-13 op 
156 b0a6bcf7 2022-09-13 op 	if (bufferevent_write(bev, &hdr, sizeof(hdr)) == -1)
157 b0a6bcf7 2022-09-13 op 		return (-1);
158 b0a6bcf7 2022-09-13 op 	if (bufferevent_write(bev, &end, sizeof(end)) == -1)
159 b0a6bcf7 2022-09-13 op 		return (-1);
160 b0a6bcf7 2022-09-13 op 	return (0);
161 b0a6bcf7 2022-09-13 op }
162 b0a6bcf7 2022-09-13 op 
163 b0a6bcf7 2022-09-13 op static int
164 b0a6bcf7 2022-09-13 op end_request(struct client *clt, int status, int proto_status)
165 b0a6bcf7 2022-09-13 op {
166 b0a6bcf7 2022-09-13 op 	struct fcgi		*fcgi = clt->clt_fcgi;
167 b0a6bcf7 2022-09-13 op 	int			 r;
168 b0a6bcf7 2022-09-13 op 
169 b0a6bcf7 2022-09-13 op 	if (clt_flush(clt) == -1)
170 b0a6bcf7 2022-09-13 op 		return (-1);
171 b0a6bcf7 2022-09-13 op 
172 b0a6bcf7 2022-09-13 op 	r = fcgi_send_end_req(fcgi, clt->clt_id, status,
173 b0a6bcf7 2022-09-13 op 	    proto_status);
174 b0a6bcf7 2022-09-13 op 	if (r == -1) {
175 b0a6bcf7 2022-09-13 op 		fcgi_error(fcgi->fcg_bev, EV_WRITE, fcgi);
176 b0a6bcf7 2022-09-13 op 		return (-1);
177 b0a6bcf7 2022-09-13 op 	}
178 b0a6bcf7 2022-09-13 op 
179 b0a6bcf7 2022-09-13 op 	SPLAY_REMOVE(client_tree, &fcgi->fcg_clients, clt);
180 b0a6bcf7 2022-09-13 op 	proxy_client_free(clt);
181 247917be 2022-09-14 op 
182 247917be 2022-09-14 op 	if (!fcgi->fcg_keep_conn)
183 247917be 2022-09-14 op 		fcgi->fcg_done = 1;
184 247917be 2022-09-14 op 
185 b0a6bcf7 2022-09-13 op 	return (0);
186 b0a6bcf7 2022-09-13 op }
187 b0a6bcf7 2022-09-13 op 
188 b0a6bcf7 2022-09-13 op int
189 b0a6bcf7 2022-09-13 op fcgi_end_request(struct client *clt, int status)
190 b0a6bcf7 2022-09-13 op {
191 b0a6bcf7 2022-09-13 op 	return (end_request(clt, status, FCGI_REQUEST_COMPLETE));
192 b0a6bcf7 2022-09-13 op }
193 b0a6bcf7 2022-09-13 op 
194 b0a6bcf7 2022-09-13 op int
195 b0a6bcf7 2022-09-13 op fcgi_abort_request(struct client *clt)
196 b0a6bcf7 2022-09-13 op {
197 b0a6bcf7 2022-09-13 op 	return (end_request(clt, 1, FCGI_OVERLOADED));
198 b0a6bcf7 2022-09-13 op }
199 b0a6bcf7 2022-09-13 op 
200 b0a6bcf7 2022-09-13 op static void
201 b0a6bcf7 2022-09-13 op fcgi_inflight_dec(const char *why)
202 b0a6bcf7 2022-09-13 op {
203 b0a6bcf7 2022-09-13 op 	fcgi_inflight--;
204 b0a6bcf7 2022-09-13 op 	log_debug("%s: fcgi inflight decremented, now %d, %s",
205 b0a6bcf7 2022-09-13 op 	    __func__, fcgi_inflight, why);
206 b0a6bcf7 2022-09-13 op }
207 b0a6bcf7 2022-09-13 op 
208 b0a6bcf7 2022-09-13 op void
209 ea8874f1 2022-09-13 op fcgi_accept(int fd, short event, void *arg)
210 b0a6bcf7 2022-09-13 op {
211 ea8874f1 2022-09-13 op 	struct galileo		*env = arg;
212 b0a6bcf7 2022-09-13 op 	struct fcgi		*fcgi = NULL;
213 b0a6bcf7 2022-09-13 op 	socklen_t		 slen;
214 b0a6bcf7 2022-09-13 op 	struct sockaddr_storage	 ss;
215 b0a6bcf7 2022-09-13 op 	int			 s = -1;
216 b0a6bcf7 2022-09-13 op 
217 ea8874f1 2022-09-13 op 	event_add(&env->sc_evpause, NULL);
218 ea8874f1 2022-09-13 op 	if ((event & EV_TIMEOUT))
219 ea8874f1 2022-09-13 op 		return;
220 ea8874f1 2022-09-13 op 
221 b0a6bcf7 2022-09-13 op 	slen = sizeof(ss);
222 b0a6bcf7 2022-09-13 op 	if ((s = accept_reserve(env->sc_sock_fd, (struct sockaddr *)&ss,
223 b0a6bcf7 2022-09-13 op 	    &slen, FD_RESERVE, &fcgi_inflight)) == -1) {
224 b0a6bcf7 2022-09-13 op 		/*
225 b0a6bcf7 2022-09-13 op 		 * Pause accept if we are out of file descriptors, or
226 b0a6bcf7 2022-09-13 op 		 * libevent will haunt us here too.
227 b0a6bcf7 2022-09-13 op 		 */
228 b0a6bcf7 2022-09-13 op 		if (errno == ENFILE || errno == EMFILE) {
229 b0a6bcf7 2022-09-13 op 			struct timeval evtpause = { 1, 0 };
230 b0a6bcf7 2022-09-13 op 
231 b0a6bcf7 2022-09-13 op 			event_del(&env->sc_evsock);
232 b0a6bcf7 2022-09-13 op 			evtimer_add(&env->sc_evpause, &evtpause);
233 b0a6bcf7 2022-09-13 op 			log_debug("%s: deferring connections", __func__);
234 b0a6bcf7 2022-09-13 op 		}
235 b0a6bcf7 2022-09-13 op 		return;
236 b0a6bcf7 2022-09-13 op 	}
237 b0a6bcf7 2022-09-13 op 
238 b0a6bcf7 2022-09-13 op 	if ((fcgi = calloc(1, sizeof(*fcgi))) == NULL)
239 b0a6bcf7 2022-09-13 op 		goto err;
240 b0a6bcf7 2022-09-13 op 
241 b0a6bcf7 2022-09-13 op 	fcgi->fcg_id = ++proxy_fcg_id;
242 b0a6bcf7 2022-09-13 op 	fcgi->fcg_s = s;
243 b0a6bcf7 2022-09-13 op 	fcgi->fcg_env = env;
244 b0a6bcf7 2022-09-13 op 	fcgi->fcg_want = FCGI_RECORD_HEADER;
245 b0a6bcf7 2022-09-13 op 	fcgi->fcg_toread = sizeof(struct fcgi_header);
246 b0a6bcf7 2022-09-13 op 	SPLAY_INIT(&fcgi->fcg_clients);
247 b0a6bcf7 2022-09-13 op 
248 b0a6bcf7 2022-09-13 op 	/* assume it's enabled until we get a FCGI_BEGIN_REQUEST */
249 b0a6bcf7 2022-09-13 op 	fcgi->fcg_keep_conn = 1;
250 b0a6bcf7 2022-09-13 op 
251 b0a6bcf7 2022-09-13 op 	fcgi->fcg_bev = bufferevent_new(fcgi->fcg_s, fcgi_read, fcgi_write,
252 b0a6bcf7 2022-09-13 op 	    fcgi_error, fcgi);
253 b0a6bcf7 2022-09-13 op 	if (fcgi->fcg_bev == NULL)
254 b0a6bcf7 2022-09-13 op 		goto err;
255 b0a6bcf7 2022-09-13 op 
256 578575af 2023-03-23 op 	SPLAY_INSERT(fcgi_tree, &env->sc_fcgi_socks, fcgi);
257 b0a6bcf7 2022-09-13 op 	bufferevent_enable(fcgi->fcg_bev, EV_READ | EV_WRITE);
258 b0a6bcf7 2022-09-13 op 	return;
259 b0a6bcf7 2022-09-13 op 
260 b0a6bcf7 2022-09-13 op err:
261 b0a6bcf7 2022-09-13 op 	if (s != -1) {
262 b0a6bcf7 2022-09-13 op 		close(s);
263 b0a6bcf7 2022-09-13 op 		free(fcgi);
264 b0a6bcf7 2022-09-13 op 		fcgi_inflight_dec(__func__);
265 b0a6bcf7 2022-09-13 op 	}
266 b0a6bcf7 2022-09-13 op }
267 b0a6bcf7 2022-09-13 op 
268 b0a6bcf7 2022-09-13 op static int
269 b0a6bcf7 2022-09-13 op parse_len(struct fcgi *fcgi, struct evbuffer *src)
270 b0a6bcf7 2022-09-13 op {
271 b0a6bcf7 2022-09-13 op 	unsigned char		 c, x[3];
272 b0a6bcf7 2022-09-13 op 
273 b0a6bcf7 2022-09-13 op 	fcgi->fcg_toread--;
274 b0a6bcf7 2022-09-13 op 	evbuffer_remove(src, &c, 1);
275 b0a6bcf7 2022-09-13 op 	if (c >> 7 == 0)
276 b0a6bcf7 2022-09-13 op 		return (c);
277 b0a6bcf7 2022-09-13 op 
278 b0a6bcf7 2022-09-13 op 	if (fcgi->fcg_toread < 3)
279 b0a6bcf7 2022-09-13 op 		return (-1);
280 b0a6bcf7 2022-09-13 op 
281 b0a6bcf7 2022-09-13 op 	fcgi->fcg_toread -= 3;
282 b0a6bcf7 2022-09-13 op 	evbuffer_remove(src, x, sizeof(x));
283 b0a6bcf7 2022-09-13 op 	return (((c & 0x7F) << 24) | (x[0] << 16) | (x[1] << 8) | x[2]);
284 b0a6bcf7 2022-09-13 op }
285 b0a6bcf7 2022-09-13 op 
286 b0a6bcf7 2022-09-13 op static int
287 b0a6bcf7 2022-09-13 op fcgi_parse_params(struct fcgi *fcgi, struct evbuffer *src, struct client *clt)
288 b0a6bcf7 2022-09-13 op {
289 b0a6bcf7 2022-09-13 op 	char			 pname[32];
290 b0a6bcf7 2022-09-13 op 	char			 server[HOST_NAME_MAX + 1];
291 b0a6bcf7 2022-09-13 op 	char			 path[PATH_MAX];
292 28239bac 2022-09-20 op 	char			 query[GEMINI_MAXLEN];
293 8857131d 2022-09-20 op 	char			 method[8];
294 b0a6bcf7 2022-09-13 op 	int			 nlen, vlen;
295 b0a6bcf7 2022-09-13 op 
296 b0a6bcf7 2022-09-13 op 	while (fcgi->fcg_toread > 0) {
297 b0a6bcf7 2022-09-13 op 		if ((nlen = parse_len(fcgi, src)) < 0 ||
298 b0a6bcf7 2022-09-13 op 		    (vlen = parse_len(fcgi, src)) < 0)
299 b0a6bcf7 2022-09-13 op 			return (-1);
300 b0a6bcf7 2022-09-13 op 
301 b0a6bcf7 2022-09-13 op 		if (fcgi->fcg_toread < nlen + vlen)
302 b0a6bcf7 2022-09-13 op 			return (-1);
303 b0a6bcf7 2022-09-13 op 
304 b0a6bcf7 2022-09-13 op 		if ((size_t)nlen > sizeof(pname) - 1) {
305 b0a6bcf7 2022-09-13 op 			/* ignore this parameter */
306 b0a6bcf7 2022-09-13 op 			fcgi->fcg_toread -= nlen - vlen;
307 b0a6bcf7 2022-09-13 op 			evbuffer_drain(src, nlen + vlen);
308 b0a6bcf7 2022-09-13 op 			continue;
309 b0a6bcf7 2022-09-13 op 		}
310 b0a6bcf7 2022-09-13 op 
311 b0a6bcf7 2022-09-13 op 		fcgi->fcg_toread -= nlen;
312 b0a6bcf7 2022-09-13 op 		evbuffer_remove(src, &pname, nlen);
313 b0a6bcf7 2022-09-13 op 		pname[nlen] = '\0';
314 b0a6bcf7 2022-09-13 op 
315 b0a6bcf7 2022-09-13 op 		if (!strcmp(pname, "SERVER_NAME") &&
316 b0a6bcf7 2022-09-13 op 		    (size_t)vlen < sizeof(server)) {
317 b0a6bcf7 2022-09-13 op 			fcgi->fcg_toread -= vlen;
318 b0a6bcf7 2022-09-13 op 			evbuffer_remove(src, &server, vlen);
319 b0a6bcf7 2022-09-13 op 			server[vlen] = '\0';
320 b0a6bcf7 2022-09-13 op 
321 c5d5d231 2022-09-14 op 			free(clt->clt_server_name);
322 b0a6bcf7 2022-09-13 op 			if ((clt->clt_server_name = strdup(server)) == NULL)
323 b0a6bcf7 2022-09-13 op 				return (-1);
324 58022b94 2022-09-30 op 			DPRINTF("clt %d: server_name: %s", clt->clt_id,
325 b0a6bcf7 2022-09-13 op 			    clt->clt_server_name);
326 b0a6bcf7 2022-09-13 op 			continue;
327 b0a6bcf7 2022-09-13 op 		}
328 b0a6bcf7 2022-09-13 op 
329 b0a6bcf7 2022-09-13 op 		if (!strcmp(pname, "SCRIPT_NAME") &&
330 b0a6bcf7 2022-09-13 op 		    (size_t)vlen < sizeof(path)) {
331 b0a6bcf7 2022-09-13 op 			fcgi->fcg_toread -= vlen;
332 b0a6bcf7 2022-09-13 op 			evbuffer_remove(src, &path, vlen);
333 b0a6bcf7 2022-09-13 op 			path[vlen] = '\0';
334 b0a6bcf7 2022-09-13 op 
335 c5d5d231 2022-09-14 op 			free(clt->clt_script_name);
336 a9c69469 2022-09-19 op 			clt->clt_script_name = NULL;
337 a9c69469 2022-09-19 op 
338 c2691a5d 2022-09-21 op 			if (vlen == 0 || path[vlen - 1] != '/')
339 a9c69469 2022-09-19 op 				asprintf(&clt->clt_script_name, "%s/", path);
340 a9c69469 2022-09-19 op 			else
341 a9c69469 2022-09-19 op 				clt->clt_script_name = strdup(path);
342 a9c69469 2022-09-19 op 
343 a9c69469 2022-09-19 op 			if (clt->clt_script_name == NULL)
344 b0a6bcf7 2022-09-13 op 				return (-1);
345 a9c69469 2022-09-19 op 
346 58022b94 2022-09-30 op 			DPRINTF("clt %d: script_name: %s", clt->clt_id,
347 b0a6bcf7 2022-09-13 op 			    clt->clt_script_name);
348 b0a6bcf7 2022-09-13 op 			continue;
349 b0a6bcf7 2022-09-13 op 		}
350 b0a6bcf7 2022-09-13 op 
351 b0a6bcf7 2022-09-13 op 		if (!strcmp(pname, "PATH_INFO") &&
352 b0a6bcf7 2022-09-13 op 		    (size_t)vlen < sizeof(path)) {
353 b0a6bcf7 2022-09-13 op 			fcgi->fcg_toread -= vlen;
354 b0a6bcf7 2022-09-13 op 			evbuffer_remove(src, &path, vlen);
355 b0a6bcf7 2022-09-13 op 			path[vlen] = '\0';
356 2944b578 2022-09-14 op 
357 c5d5d231 2022-09-14 op 			free(clt->clt_path_info);
358 1296d9c0 2022-09-14 op 			clt->clt_path_info = NULL;
359 1296d9c0 2022-09-14 op 
360 1296d9c0 2022-09-14 op 			if (*path != '/')
361 1296d9c0 2022-09-14 op 				asprintf(&clt->clt_path_info, "/%s", path);
362 1296d9c0 2022-09-14 op 			else
363 1296d9c0 2022-09-14 op 				clt->clt_path_info = strdup(path);
364 1296d9c0 2022-09-14 op 
365 1296d9c0 2022-09-14 op 			if (clt->clt_path_info == NULL)
366 b0a6bcf7 2022-09-13 op 				return (-1);
367 1296d9c0 2022-09-14 op 
368 58022b94 2022-09-30 op 			DPRINTF("clt %d: path_info: %s", clt->clt_id,
369 b0a6bcf7 2022-09-13 op 			    clt->clt_path_info);
370 28239bac 2022-09-20 op 			continue;
371 28239bac 2022-09-20 op 		}
372 28239bac 2022-09-20 op 
373 28239bac 2022-09-20 op 		if (!strcmp(pname, "QUERY_STRING") &&
374 28239bac 2022-09-20 op 		    (size_t)vlen < sizeof(query) &&
375 28239bac 2022-09-20 op 		    vlen > 0) {
376 28239bac 2022-09-20 op 			fcgi->fcg_toread -= vlen;
377 28239bac 2022-09-20 op 			evbuffer_remove(src, &query, vlen);
378 8857131d 2022-09-20 op 			query[vlen] = '\0';
379 28239bac 2022-09-20 op 
380 28239bac 2022-09-20 op 			free(clt->clt_query);
381 28239bac 2022-09-20 op 			if ((clt->clt_query = strdup(query)) == NULL)
382 28239bac 2022-09-20 op 				return (-1);
383 28239bac 2022-09-20 op 
384 58022b94 2022-09-30 op 			DPRINTF("clt %d: query: %s", clt->clt_id,
385 28239bac 2022-09-20 op 			    clt->clt_query);
386 8857131d 2022-09-20 op 			continue;
387 8857131d 2022-09-20 op 		}
388 8857131d 2022-09-20 op 
389 8857131d 2022-09-20 op 		if (!strcmp(pname, "REQUEST_METHOD") &&
390 8857131d 2022-09-20 op 		    (size_t)vlen < sizeof(method)) {
391 8857131d 2022-09-20 op 			fcgi->fcg_toread -= vlen;
392 8857131d 2022-09-20 op 			evbuffer_remove(src, &method, vlen);
393 8857131d 2022-09-20 op 			method[vlen] = '\0';
394 8857131d 2022-09-20 op 
395 8857131d 2022-09-20 op 			if (!strcasecmp(method, "GET"))
396 8857131d 2022-09-20 op 				clt->clt_method = METHOD_GET;
397 8857131d 2022-09-20 op 			if (!strcasecmp(method, "POST"))
398 8857131d 2022-09-20 op 				clt->clt_method = METHOD_POST;
399 8857131d 2022-09-20 op 
400 b0a6bcf7 2022-09-13 op 			continue;
401 b0a6bcf7 2022-09-13 op 		}
402 b0a6bcf7 2022-09-13 op 
403 b0a6bcf7 2022-09-13 op 		fcgi->fcg_toread -= vlen;
404 b0a6bcf7 2022-09-13 op 		evbuffer_drain(src, vlen);
405 b0a6bcf7 2022-09-13 op 	}
406 b0a6bcf7 2022-09-13 op 
407 b0a6bcf7 2022-09-13 op 	return (0);
408 b0a6bcf7 2022-09-13 op }
409 8857131d 2022-09-20 op 
410 8857131d 2022-09-20 op static int
411 8857131d 2022-09-20 op fcgi_parse_form(struct fcgi *fcgi, struct client *clt, struct evbuffer *src)
412 8857131d 2022-09-20 op {
413 8857131d 2022-09-20 op 	char			*a, *s;
414 8857131d 2022-09-20 op 	char			 tmp[2048];
415 8857131d 2022-09-20 op 	size_t			 len;
416 8857131d 2022-09-20 op 
417 8857131d 2022-09-20 op 	if (fcgi->fcg_toread == 0) {
418 8857131d 2022-09-20 op 		clt->clt_bodydone = 1;
419 8857131d 2022-09-20 op 		return (proxy_start_request(fcgi->fcg_env, clt));
420 8857131d 2022-09-20 op 	}
421 b0a6bcf7 2022-09-13 op 
422 8857131d 2022-09-20 op 	len = sizeof(tmp) - 1;
423 8857131d 2022-09-20 op 	if (len > (size_t)fcgi->fcg_toread)
424 8857131d 2022-09-20 op 		len = fcgi->fcg_toread;
425 8857131d 2022-09-20 op 
426 8857131d 2022-09-20 op 	fcgi->fcg_toread -= len;
427 8857131d 2022-09-20 op 	evbuffer_remove(src, &tmp, len);
428 8857131d 2022-09-20 op 	tmp[len] = '\0';
429 8857131d 2022-09-20 op 
430 8857131d 2022-09-20 op 	if (clt->clt_bodydone)
431 8857131d 2022-09-20 op 		return (0);
432 8857131d 2022-09-20 op 
433 8857131d 2022-09-20 op 	if (clt->clt_bodylen > GEMINI_MAXLEN) {
434 8857131d 2022-09-20 op 		free(clt->clt_body);
435 8857131d 2022-09-20 op 		clt->clt_body = NULL;
436 8857131d 2022-09-20 op 		return (0);
437 8857131d 2022-09-20 op 	}
438 8857131d 2022-09-20 op 
439 8857131d 2022-09-20 op 	if (clt->clt_body != NULL) {
440 8857131d 2022-09-20 op 		if ((a = strchr(tmp, '&')) != NULL ||
441 8857131d 2022-09-20 op 		    (a = strchr(tmp, '\r')) != NULL) {
442 8857131d 2022-09-20 op 			*a = '\0';
443 8857131d 2022-09-20 op 			clt->clt_bodydone = 1;
444 8857131d 2022-09-20 op 		}
445 8857131d 2022-09-20 op 
446 8857131d 2022-09-20 op 		a = clt->clt_body;
447 8857131d 2022-09-20 op 		if (asprintf(&clt->clt_body, "%s%s", a, tmp) == -1) {
448 8857131d 2022-09-20 op 			free(a);
449 8857131d 2022-09-20 op 			clt->clt_body = NULL;
450 8857131d 2022-09-20 op 			return (0);
451 8857131d 2022-09-20 op 		}
452 adeb27af 2023-03-23 op 		free(a);
453 8857131d 2022-09-20 op 		clt->clt_bodylen += strlen(tmp);
454 8857131d 2022-09-20 op 		return (0);
455 8857131d 2022-09-20 op 	}
456 8857131d 2022-09-20 op 
457 8857131d 2022-09-20 op 	if (clt->clt_bodylen != 0)
458 8857131d 2022-09-20 op 		return (0); /* EOM situation */
459 8857131d 2022-09-20 op 
460 8857131d 2022-09-20 op 	if ((s = strstr(tmp, "q=")) == NULL)
461 8857131d 2022-09-20 op 		return (0);
462 8857131d 2022-09-20 op 	s += 2;
463 8857131d 2022-09-20 op 
464 8857131d 2022-09-20 op 	if ((a = strchr(s, '&')) != NULL ||
465 8857131d 2022-09-20 op 	    (a = strchr(s, '\r')) != NULL) {
466 8857131d 2022-09-20 op 		*a = '\0';
467 8857131d 2022-09-20 op 		clt->clt_bodydone = 1;
468 8857131d 2022-09-20 op 	}
469 8857131d 2022-09-20 op 
470 8857131d 2022-09-20 op 	clt->clt_bodylen = strlen(s);
471 8857131d 2022-09-20 op 	if ((clt->clt_body = strdup(s)) == NULL)
472 8857131d 2022-09-20 op 		return (0);
473 8857131d 2022-09-20 op 
474 8857131d 2022-09-20 op 	return (0);
475 8857131d 2022-09-20 op }
476 8857131d 2022-09-20 op 
477 b0a6bcf7 2022-09-13 op void
478 b0a6bcf7 2022-09-13 op fcgi_read(struct bufferevent *bev, void *d)
479 b0a6bcf7 2022-09-13 op {
480 b0a6bcf7 2022-09-13 op 	struct fcgi		*fcgi = d;
481 b0a6bcf7 2022-09-13 op 	struct galileo		*env = fcgi->fcg_env;
482 b0a6bcf7 2022-09-13 op 	struct evbuffer		*src = EVBUFFER_INPUT(bev);
483 b0a6bcf7 2022-09-13 op 	struct fcgi_header	 hdr;
484 b0a6bcf7 2022-09-13 op 	struct fcgi_begin_req	 breq;
485 b0a6bcf7 2022-09-13 op 	struct client		*clt, q;
486 b0a6bcf7 2022-09-13 op 	int			 role;
487 b0a6bcf7 2022-09-13 op 
488 b0a6bcf7 2022-09-13 op 	memset(&q, 0, sizeof(q));
489 b0a6bcf7 2022-09-13 op 
490 b0a6bcf7 2022-09-13 op 	for (;;) {
491 b0a6bcf7 2022-09-13 op 		if (EVBUFFER_LENGTH(src) < (size_t)fcgi->fcg_toread)
492 b0a6bcf7 2022-09-13 op 			return;
493 b0a6bcf7 2022-09-13 op 
494 b0a6bcf7 2022-09-13 op 		if (fcgi->fcg_want == FCGI_RECORD_HEADER) {
495 b0a6bcf7 2022-09-13 op 			fcgi->fcg_want = FCGI_RECORD_BODY;
496 b0a6bcf7 2022-09-13 op 			bufferevent_read(bev, &hdr, sizeof(hdr));
497 b0a6bcf7 2022-09-13 op 
498 58022b94 2022-09-30 op 			DPRINTF("header: v=%d t=%d id=%d len=%d p=%d",
499 b0a6bcf7 2022-09-13 op 			    hdr.version, hdr.type,
500 b0a6bcf7 2022-09-13 op 			    CAT(hdr.req_id0, hdr.req_id1),
501 b0a6bcf7 2022-09-13 op 			    CAT(hdr.content_len0, hdr.content_len1),
502 b0a6bcf7 2022-09-13 op 			    hdr.padding);
503 b0a6bcf7 2022-09-13 op 
504 b0a6bcf7 2022-09-13 op 			if (hdr.version != FCGI_VERSION_1) {
505 b0a6bcf7 2022-09-13 op 				log_warnx("unknown fastcgi version: %d",
506 b0a6bcf7 2022-09-13 op 				    hdr.version);
507 b0a6bcf7 2022-09-13 op 				fcgi_error(bev, EV_READ, d);
508 b0a6bcf7 2022-09-13 op 				return;
509 b0a6bcf7 2022-09-13 op 			}
510 b0a6bcf7 2022-09-13 op 
511 b0a6bcf7 2022-09-13 op 			fcgi->fcg_toread = CAT(hdr.content_len0,
512 b0a6bcf7 2022-09-13 op 			    hdr.content_len1);
513 b0a6bcf7 2022-09-13 op 			if (fcgi->fcg_toread < 0) {
514 b0a6bcf7 2022-09-13 op 				log_warnx("invalid record length: %d",
515 b0a6bcf7 2022-09-13 op 				    fcgi->fcg_toread);
516 b0a6bcf7 2022-09-13 op 				fcgi_error(bev, EV_READ, d);
517 b0a6bcf7 2022-09-13 op 				return;
518 b0a6bcf7 2022-09-13 op 			}
519 b0a6bcf7 2022-09-13 op 
520 b0a6bcf7 2022-09-13 op 			fcgi->fcg_padding = hdr.padding;
521 b0a6bcf7 2022-09-13 op 			if (fcgi->fcg_padding < 0) {
522 b0a6bcf7 2022-09-13 op 				log_warnx("invalid padding: %d",
523 b0a6bcf7 2022-09-13 op 				    fcgi->fcg_padding);
524 b0a6bcf7 2022-09-13 op 				fcgi_error(bev, EV_READ, d);
525 b0a6bcf7 2022-09-13 op 				return;
526 b0a6bcf7 2022-09-13 op 			}
527 b0a6bcf7 2022-09-13 op 
528 b0a6bcf7 2022-09-13 op 			fcgi->fcg_type = hdr.type;
529 b0a6bcf7 2022-09-13 op 			fcgi->fcg_rec_id = CAT(hdr.req_id0, hdr.req_id1);
530 b0a6bcf7 2022-09-13 op 			continue;
531 b0a6bcf7 2022-09-13 op 		}
532 b0a6bcf7 2022-09-13 op 
533 b0a6bcf7 2022-09-13 op 		q.clt_id = fcgi->fcg_rec_id;
534 b0a6bcf7 2022-09-13 op 		clt = SPLAY_FIND(client_tree, &fcgi->fcg_clients, &q);
535 b0a6bcf7 2022-09-13 op 
536 b0a6bcf7 2022-09-13 op 		switch (fcgi->fcg_type) {
537 b0a6bcf7 2022-09-13 op 		case FCGI_BEGIN_REQUEST:
538 b0a6bcf7 2022-09-13 op 			if (sizeof(breq) != fcgi->fcg_toread) {
539 b0a6bcf7 2022-09-13 op 				log_warnx("unexpected size for "
540 b0a6bcf7 2022-09-13 op 				    "FCGI_BEGIN_REQUEST");
541 b0a6bcf7 2022-09-13 op 				fcgi_error(bev, EV_READ, d);
542 b0a6bcf7 2022-09-13 op 				return;
543 b0a6bcf7 2022-09-13 op 			}
544 b0a6bcf7 2022-09-13 op 
545 b0a6bcf7 2022-09-13 op 			evbuffer_remove(src, &breq, sizeof(breq));
546 b0a6bcf7 2022-09-13 op 
547 b0a6bcf7 2022-09-13 op 			role = CAT(breq.role0, breq.role1);
548 b0a6bcf7 2022-09-13 op 			if (role != FCGI_RESPONDER) {
549 b0a6bcf7 2022-09-13 op 				log_warnx("unknown fastcgi role: %d",
550 b0a6bcf7 2022-09-13 op 				    role);
551 b0a6bcf7 2022-09-13 op 				if (fcgi_send_end_req(fcgi, fcgi->fcg_rec_id,
552 b0a6bcf7 2022-09-13 op 				    1, FCGI_UNKNOWN_ROLE) == -1) {
553 b0a6bcf7 2022-09-13 op 					fcgi_error(bev, EV_READ, d);
554 b0a6bcf7 2022-09-13 op 					return;
555 b0a6bcf7 2022-09-13 op 				}
556 b0a6bcf7 2022-09-13 op 				break;
557 b0a6bcf7 2022-09-13 op 			}
558 b0a6bcf7 2022-09-13 op 
559 b0a6bcf7 2022-09-13 op 			if (!fcgi->fcg_keep_conn) {
560 b0a6bcf7 2022-09-13 op 				log_warnx("trying to reuse the fastcgi "
561 b0a6bcf7 2022-09-13 op 				    "socket without marking it as so.");
562 b0a6bcf7 2022-09-13 op 				fcgi_error(bev, EV_READ, d);
563 b0a6bcf7 2022-09-13 op 				return;
564 b0a6bcf7 2022-09-13 op 			}
565 b0a6bcf7 2022-09-13 op 			fcgi->fcg_keep_conn = breq.flags & FCGI_KEEP_CONN;
566 b0a6bcf7 2022-09-13 op 
567 b0a6bcf7 2022-09-13 op 			if (clt != NULL) {
568 b0a6bcf7 2022-09-13 op 				log_warnx("ignoring attemp to re-use an "
569 b0a6bcf7 2022-09-13 op 				    "active request id (%d)",
570 b0a6bcf7 2022-09-13 op 				    fcgi->fcg_rec_id);
571 b0a6bcf7 2022-09-13 op 				break;
572 b0a6bcf7 2022-09-13 op 			}
573 b0a6bcf7 2022-09-13 op 
574 b0a6bcf7 2022-09-13 op 			if ((clt = calloc(1, sizeof(*clt))) == NULL) {
575 b0a6bcf7 2022-09-13 op 				log_warnx("calloc");
576 232eee45 2022-09-20 op 				break;
577 232eee45 2022-09-20 op 			}
578 232eee45 2022-09-20 op 
579 232eee45 2022-09-20 op 			clt->clt_tp = template(clt, clt_tp_puts, clt_tp_putc);
580 232eee45 2022-09-20 op 			if (clt->clt_tp == NULL) {
581 232eee45 2022-09-20 op 				free(clt);
582 232eee45 2022-09-20 op 				log_warn("template");
583 b0a6bcf7 2022-09-13 op 				break;
584 b0a6bcf7 2022-09-13 op 			}
585 b0a6bcf7 2022-09-13 op 
586 b0a6bcf7 2022-09-13 op 			clt->clt_id = fcgi->fcg_rec_id;
587 b0a6bcf7 2022-09-13 op 			clt->clt_fd = -1;
588 b0a6bcf7 2022-09-13 op 			clt->clt_fcgi = fcgi;
589 b0a6bcf7 2022-09-13 op 			SPLAY_INSERT(client_tree, &fcgi->fcg_clients, clt);
590 b0a6bcf7 2022-09-13 op 			break;
591 b0a6bcf7 2022-09-13 op 		case FCGI_PARAMS:
592 b0a6bcf7 2022-09-13 op 			if (clt == NULL) {
593 b0a6bcf7 2022-09-13 op 				log_warnx("got FCGI_PARAMS for inactive id "
594 b0a6bcf7 2022-09-13 op 				    "(%d)", fcgi->fcg_rec_id);
595 b0a6bcf7 2022-09-13 op 				evbuffer_drain(src, fcgi->fcg_toread);
596 b0a6bcf7 2022-09-13 op 				break;
597 b0a6bcf7 2022-09-13 op 			}
598 b0a6bcf7 2022-09-13 op 			if (fcgi->fcg_toread == 0) {
599 b0a6bcf7 2022-09-13 op 				evbuffer_drain(src, fcgi->fcg_toread);
600 8857131d 2022-09-20 op 				if (clt->clt_method == METHOD_GET) {
601 8857131d 2022-09-20 op 					if (proxy_start_request(env, clt)
602 8857131d 2022-09-20 op 					    == -1)
603 8857131d 2022-09-20 op 						return;
604 8857131d 2022-09-20 op 				}
605 b0a6bcf7 2022-09-13 op 				break;
606 b0a6bcf7 2022-09-13 op 			}
607 b0a6bcf7 2022-09-13 op 			if (fcgi_parse_params(fcgi, src, clt) == -1) {
608 b0a6bcf7 2022-09-13 op 				log_warnx("fcgi_parse_params failed");
609 b0a6bcf7 2022-09-13 op 				fcgi_error(bev, EV_READ, d);
610 b0a6bcf7 2022-09-13 op 				return;
611 b0a6bcf7 2022-09-13 op 			}
612 b0a6bcf7 2022-09-13 op 			break;
613 b0a6bcf7 2022-09-13 op 		case FCGI_STDIN:
614 d36b8fee 2023-03-23 op 			if (clt == NULL || clt->clt_method != METHOD_POST) {
615 8857131d 2022-09-20 op 				evbuffer_drain(src, fcgi->fcg_toread);
616 8857131d 2022-09-20 op 				break;
617 8857131d 2022-09-20 op 			}
618 8857131d 2022-09-20 op 			if (fcgi_parse_form(fcgi, clt, src) == -1)
619 8857131d 2022-09-20 op 				return;
620 b0a6bcf7 2022-09-13 op 			break;
621 b0a6bcf7 2022-09-13 op 		case FCGI_ABORT_REQUEST:
622 b0a6bcf7 2022-09-13 op 			if (clt == NULL) {
623 b0a6bcf7 2022-09-13 op 				log_warnx("got FCGI_ABORT_REQUEST for inactive"
624 b0a6bcf7 2022-09-13 op 				    " id (%d)", fcgi->fcg_rec_id);
625 b0a6bcf7 2022-09-13 op 				evbuffer_drain(src, fcgi->fcg_toread);
626 b0a6bcf7 2022-09-13 op 				break;
627 b0a6bcf7 2022-09-13 op 			}
628 b0a6bcf7 2022-09-13 op 			if (fcgi_end_request(clt, 1) == -1) {
629 b0a6bcf7 2022-09-13 op 				/* calls fcgi_error on failure */
630 b0a6bcf7 2022-09-13 op 				return;
631 b0a6bcf7 2022-09-13 op 			}
632 b0a6bcf7 2022-09-13 op 			break;
633 b0a6bcf7 2022-09-13 op 		default:
634 b0a6bcf7 2022-09-13 op 			log_warnx("unknown fastcgi record type %d",
635 b0a6bcf7 2022-09-13 op 			    fcgi->fcg_type);
636 b0a6bcf7 2022-09-13 op 			evbuffer_drain(src, fcgi->fcg_toread);
637 b0a6bcf7 2022-09-13 op 			break;
638 b0a6bcf7 2022-09-13 op 		}
639 b0a6bcf7 2022-09-13 op 
640 b0a6bcf7 2022-09-13 op 		/* Prepare for the next record. */
641 b0a6bcf7 2022-09-13 op 		evbuffer_drain(src, fcgi->fcg_padding);
642 b0a6bcf7 2022-09-13 op 		fcgi->fcg_want = FCGI_RECORD_HEADER;
643 b0a6bcf7 2022-09-13 op 		fcgi->fcg_toread = sizeof(struct fcgi_header);
644 b0a6bcf7 2022-09-13 op 	}
645 b0a6bcf7 2022-09-13 op }
646 b0a6bcf7 2022-09-13 op 
647 b0a6bcf7 2022-09-13 op void
648 b0a6bcf7 2022-09-13 op fcgi_write(struct bufferevent *bev, void *d)
649 b0a6bcf7 2022-09-13 op {
650 b0a6bcf7 2022-09-13 op 	struct fcgi		*fcgi = d;
651 247917be 2022-09-14 op 	struct evbuffer		*out = EVBUFFER_OUTPUT(bev);
652 b0a6bcf7 2022-09-13 op 
653 247917be 2022-09-14 op 	if (fcgi->fcg_done && EVBUFFER_LENGTH(out) == 0)
654 247917be 2022-09-14 op 		fcgi_error(bev, EVBUFFER_EOF, fcgi);
655 b0a6bcf7 2022-09-13 op }
656 b0a6bcf7 2022-09-13 op 
657 b0a6bcf7 2022-09-13 op void
658 b0a6bcf7 2022-09-13 op fcgi_error(struct bufferevent *bev, short event, void *d)
659 b0a6bcf7 2022-09-13 op {
660 b0a6bcf7 2022-09-13 op 	struct fcgi		*fcgi = d;
661 b0a6bcf7 2022-09-13 op 	struct galileo		*env = fcgi->fcg_env;
662 b0a6bcf7 2022-09-13 op 	struct client		*clt;
663 b0a6bcf7 2022-09-13 op 
664 b0a6bcf7 2022-09-13 op 	log_debug("fcgi failure, shutting down connection (ev: %x)",
665 b0a6bcf7 2022-09-13 op 	    event);
666 b0a6bcf7 2022-09-13 op 	fcgi_inflight_dec(__func__);
667 b0a6bcf7 2022-09-13 op 
668 b0a6bcf7 2022-09-13 op 	while ((clt = SPLAY_MIN(client_tree, &fcgi->fcg_clients)) != NULL) {
669 b0a6bcf7 2022-09-13 op 		SPLAY_REMOVE(client_tree, &fcgi->fcg_clients, clt);
670 b0a6bcf7 2022-09-13 op 		proxy_client_free(clt);
671 b0a6bcf7 2022-09-13 op 	}
672 b0a6bcf7 2022-09-13 op 
673 b0a6bcf7 2022-09-13 op 	SPLAY_REMOVE(fcgi_tree, &env->sc_fcgi_socks, fcgi);
674 40f762b2 2022-09-23 op 	fcgi_free(fcgi);
675 b0a6bcf7 2022-09-13 op 
676 b0a6bcf7 2022-09-13 op 	return;
677 40f762b2 2022-09-23 op }
678 40f762b2 2022-09-23 op 
679 40f762b2 2022-09-23 op void
680 40f762b2 2022-09-23 op fcgi_free(struct fcgi *fcgi)
681 40f762b2 2022-09-23 op {
682 40f762b2 2022-09-23 op 	close(fcgi->fcg_s);
683 40f762b2 2022-09-23 op 	bufferevent_free(fcgi->fcg_bev);
684 40f762b2 2022-09-23 op 	free(fcgi);
685 b0a6bcf7 2022-09-13 op }
686 b0a6bcf7 2022-09-13 op 
687 b0a6bcf7 2022-09-13 op int
688 b0a6bcf7 2022-09-13 op clt_flush(struct client *clt)
689 b0a6bcf7 2022-09-13 op {
690 b0a6bcf7 2022-09-13 op 	struct fcgi		*fcgi = clt->clt_fcgi;
691 b0a6bcf7 2022-09-13 op 	struct bufferevent	*bev = fcgi->fcg_bev;
692 b0a6bcf7 2022-09-13 op 	struct fcgi_header	 hdr;
693 b0a6bcf7 2022-09-13 op 
694 b0a6bcf7 2022-09-13 op 	if (clt->clt_buflen == 0)
695 b0a6bcf7 2022-09-13 op 		return (0);
696 b0a6bcf7 2022-09-13 op 
697 b0a6bcf7 2022-09-13 op 	memset(&hdr, 0, sizeof(hdr));
698 b0a6bcf7 2022-09-13 op 	hdr.version = FCGI_VERSION_1;
699 b0a6bcf7 2022-09-13 op 	hdr.type = FCGI_STDOUT;
700 b0a6bcf7 2022-09-13 op 	hdr.req_id0 = (clt->clt_id & 0xFF);
701 b0a6bcf7 2022-09-13 op 	hdr.req_id1 = (clt->clt_id >> 8);
702 b0a6bcf7 2022-09-13 op 	hdr.content_len0 = (clt->clt_buflen & 0xFF);
703 b0a6bcf7 2022-09-13 op 	hdr.content_len1 = (clt->clt_buflen >> 8);
704 b0a6bcf7 2022-09-13 op 
705 b0a6bcf7 2022-09-13 op 	if (bufferevent_write(bev, &hdr, sizeof(hdr)) == -1 ||
706 b0a6bcf7 2022-09-13 op 	    bufferevent_write(bev, clt->clt_buf, clt->clt_buflen) == -1) {
707 b0a6bcf7 2022-09-13 op 		fcgi_error(bev, EV_WRITE, fcgi);
708 b0a6bcf7 2022-09-13 op 		return (-1);
709 b0a6bcf7 2022-09-13 op 	}
710 b0a6bcf7 2022-09-13 op 
711 b0a6bcf7 2022-09-13 op 	clt->clt_buflen = 0;
712 b0a6bcf7 2022-09-13 op 
713 b0a6bcf7 2022-09-13 op 	return (0);
714 b0a6bcf7 2022-09-13 op }
715 b0a6bcf7 2022-09-13 op 
716 b0a6bcf7 2022-09-13 op int
717 b0a6bcf7 2022-09-13 op clt_write(struct client *clt, const uint8_t *buf, size_t len)
718 b0a6bcf7 2022-09-13 op {
719 b0a6bcf7 2022-09-13 op 	size_t			 left, copy;
720 b0a6bcf7 2022-09-13 op 
721 b0a6bcf7 2022-09-13 op 	while (len > 0) {
722 b0a6bcf7 2022-09-13 op 		left = sizeof(clt->clt_buf) - clt->clt_buflen;
723 b0a6bcf7 2022-09-13 op 		if (left == 0) {
724 b0a6bcf7 2022-09-13 op 			if (clt_flush(clt) == -1)
725 b0a6bcf7 2022-09-13 op 				return (-1);
726 b0a6bcf7 2022-09-13 op 			left = sizeof(clt->clt_buf);
727 b0a6bcf7 2022-09-13 op 		}
728 b0a6bcf7 2022-09-13 op 
729 b0a6bcf7 2022-09-13 op 		copy = MIN(left, len);
730 b0a6bcf7 2022-09-13 op 
731 b0a6bcf7 2022-09-13 op 		memcpy(&clt->clt_buf[clt->clt_buflen], buf, copy);
732 b0a6bcf7 2022-09-13 op 		clt->clt_buflen += copy;
733 b0a6bcf7 2022-09-13 op 		buf += copy;
734 b0a6bcf7 2022-09-13 op 		len -= copy;
735 b0a6bcf7 2022-09-13 op 	}
736 b0a6bcf7 2022-09-13 op 
737 b0a6bcf7 2022-09-13 op 	return (0);
738 c9dc4bc6 2022-09-13 op }
739 c9dc4bc6 2022-09-13 op 
740 c9dc4bc6 2022-09-13 op int
741 c9dc4bc6 2022-09-13 op clt_putc(struct client *clt, char ch)
742 c9dc4bc6 2022-09-13 op {
743 c9dc4bc6 2022-09-13 op 	return (clt_write(clt, &ch, 1));
744 c9dc4bc6 2022-09-13 op }
745 c9dc4bc6 2022-09-13 op 
746 c9dc4bc6 2022-09-13 op int
747 c9dc4bc6 2022-09-13 op clt_puts(struct client *clt, const char *str)
748 c9dc4bc6 2022-09-13 op {
749 c9dc4bc6 2022-09-13 op 	return (clt_write(clt, str, strlen(str)));
750 b0a6bcf7 2022-09-13 op }
751 b0a6bcf7 2022-09-13 op 
752 b0a6bcf7 2022-09-13 op int
753 b0a6bcf7 2022-09-13 op clt_write_bufferevent(struct client *clt, struct bufferevent *bev)
754 b0a6bcf7 2022-09-13 op {
755 b0a6bcf7 2022-09-13 op 	struct evbuffer		*src = EVBUFFER_INPUT(bev);
756 b0a6bcf7 2022-09-13 op 	size_t			 len, left, copy;
757 b0a6bcf7 2022-09-13 op 
758 b0a6bcf7 2022-09-13 op 	len = EVBUFFER_LENGTH(src);
759 b0a6bcf7 2022-09-13 op 	while (len > 0) {
760 b0a6bcf7 2022-09-13 op 		left = sizeof(clt->clt_buf) - clt->clt_buflen;
761 b0a6bcf7 2022-09-13 op 		if (left == 0) {
762 b0a6bcf7 2022-09-13 op 			if (clt_flush(clt) == -1)
763 b0a6bcf7 2022-09-13 op 				return (-1);
764 b0a6bcf7 2022-09-13 op 			left = sizeof(clt->clt_buf);
765 b0a6bcf7 2022-09-13 op 		}
766 b0a6bcf7 2022-09-13 op 
767 b0a6bcf7 2022-09-13 op 		copy = bufferevent_read(bev, &clt->clt_buf[clt->clt_buflen],
768 b0a6bcf7 2022-09-13 op 		    MIN(left, len));
769 b0a6bcf7 2022-09-13 op 		clt->clt_buflen += copy;
770 b0a6bcf7 2022-09-13 op 
771 b0a6bcf7 2022-09-13 op 		len = EVBUFFER_LENGTH(src);
772 b0a6bcf7 2022-09-13 op 	}
773 b0a6bcf7 2022-09-13 op 
774 b0a6bcf7 2022-09-13 op 	return (0);
775 b0a6bcf7 2022-09-13 op }
776 b0a6bcf7 2022-09-13 op 
777 b0a6bcf7 2022-09-13 op int
778 b0a6bcf7 2022-09-13 op clt_printf(struct client *clt, const char *fmt, ...)
779 b0a6bcf7 2022-09-13 op {
780 b0a6bcf7 2022-09-13 op 	struct fcgi		*fcgi = clt->clt_fcgi;
781 b0a6bcf7 2022-09-13 op 	struct bufferevent	*bev = fcgi->fcg_bev;
782 b0a6bcf7 2022-09-13 op 	char			*str;
783 b0a6bcf7 2022-09-13 op 	va_list			 ap;
784 b0a6bcf7 2022-09-13 op 	int			 r;
785 b0a6bcf7 2022-09-13 op 
786 b0a6bcf7 2022-09-13 op 	va_start(ap, fmt);
787 b0a6bcf7 2022-09-13 op 	r = vasprintf(&str, fmt, ap);
788 b0a6bcf7 2022-09-13 op 	va_end(ap);
789 b0a6bcf7 2022-09-13 op 	if (r == -1) {
790 b0a6bcf7 2022-09-13 op 		fcgi_error(bev, EV_WRITE, fcgi);
791 b0a6bcf7 2022-09-13 op 		return (-1);
792 b0a6bcf7 2022-09-13 op 	}
793 b0a6bcf7 2022-09-13 op 
794 b0a6bcf7 2022-09-13 op 	r = clt_write(clt, str, r);
795 b0a6bcf7 2022-09-13 op 	free(str);
796 b0a6bcf7 2022-09-13 op 	return (r);
797 b0a6bcf7 2022-09-13 op }
798 b0a6bcf7 2022-09-13 op 
799 b0a6bcf7 2022-09-13 op int
800 232eee45 2022-09-20 op clt_tp_puts(struct template *tp, const char *str)
801 232eee45 2022-09-20 op {
802 232eee45 2022-09-20 op 	struct client		*clt = tp->tp_arg;
803 232eee45 2022-09-20 op 
804 95448b62 2022-09-29 op 	if (clt_puts(clt, str) == -1)
805 232eee45 2022-09-20 op 		return (-1);
806 232eee45 2022-09-20 op 
807 232eee45 2022-09-20 op 	return (0);
808 232eee45 2022-09-20 op }
809 232eee45 2022-09-20 op 
810 232eee45 2022-09-20 op int
811 232eee45 2022-09-20 op clt_tp_putc(struct template *tp, int c)
812 232eee45 2022-09-20 op {
813 232eee45 2022-09-20 op 	struct client		*clt = tp->tp_arg;
814 232eee45 2022-09-20 op 
815 95448b62 2022-09-29 op 	if (clt_putc(clt, c) == -1)
816 232eee45 2022-09-20 op 		return (-1);
817 232eee45 2022-09-20 op 
818 232eee45 2022-09-20 op 	return (0);
819 232eee45 2022-09-20 op }
820 232eee45 2022-09-20 op 
821 232eee45 2022-09-20 op int
822 b0a6bcf7 2022-09-13 op fcgi_cmp(struct fcgi *a, struct fcgi *b)
823 b0a6bcf7 2022-09-13 op {
824 b0a6bcf7 2022-09-13 op 	return ((int)a->fcg_id - b->fcg_id);
825 b0a6bcf7 2022-09-13 op }
826 b0a6bcf7 2022-09-13 op 
827 b0a6bcf7 2022-09-13 op int
828 b0a6bcf7 2022-09-13 op fcgi_client_cmp(struct client *a, struct client *b)
829 b0a6bcf7 2022-09-13 op {
830 b0a6bcf7 2022-09-13 op 	return ((int)a->clt_id - b->clt_id);
831 b0a6bcf7 2022-09-13 op }
832 b0a6bcf7 2022-09-13 op 
833 b0a6bcf7 2022-09-13 op SPLAY_GENERATE(fcgi_tree, fcgi, fcg_nodes, fcgi_cmp);
834 b0a6bcf7 2022-09-13 op SPLAY_GENERATE(client_tree, client, clt_nodes, fcgi_client_cmp);