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