1 12eb29d8 2024-01-07 op /*	$OpenBSD: imsg-buffer.c,v 1.18 2023/12/12 15:47:41 claudio Exp $	*/
  2 6b191ed5 2021-02-23 op 
  3 6b191ed5 2021-02-23 op /*
  4 12eb29d8 2024-01-07 op  * Copyright (c) 2023 Claudio Jeker <claudio@openbsd.org>
  5 6b191ed5 2021-02-23 op  * Copyright (c) 2003, 2004 Henning Brauer <henning@openbsd.org>
  6 6b191ed5 2021-02-23 op  *
  7 6b191ed5 2021-02-23 op  * Permission to use, copy, modify, and distribute this software for any
  8 6b191ed5 2021-02-23 op  * purpose with or without fee is hereby granted, provided that the above
  9 6b191ed5 2021-02-23 op  * copyright notice and this permission notice appear in all copies.
 10 6b191ed5 2021-02-23 op  *
 11 6b191ed5 2021-02-23 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 12 6b191ed5 2021-02-23 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 13 6b191ed5 2021-02-23 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 14 6b191ed5 2021-02-23 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 15 6b191ed5 2021-02-23 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 16 6b191ed5 2021-02-23 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 17 6b191ed5 2021-02-23 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 18 6b191ed5 2021-02-23 op  */
 19 6b191ed5 2021-02-23 op 
 20 6b191ed5 2021-02-23 op #include "../config.h"
 21 6b191ed5 2021-02-23 op 
 22 6b191ed5 2021-02-23 op #include <sys/types.h>
 23 6b191ed5 2021-02-23 op #include <sys/socket.h>
 24 6b191ed5 2021-02-23 op #include <sys/uio.h>
 25 6b191ed5 2021-02-23 op 
 26 6b191ed5 2021-02-23 op #include <limits.h>
 27 6b191ed5 2021-02-23 op #include <errno.h>
 28 12eb29d8 2024-01-07 op #include <stdint.h>
 29 6b191ed5 2021-02-23 op #include <stdlib.h>
 30 6b191ed5 2021-02-23 op #include <string.h>
 31 6b191ed5 2021-02-23 op #include <unistd.h>
 32 6b191ed5 2021-02-23 op 
 33 6b191ed5 2021-02-23 op #include "imsg.h"
 34 6b191ed5 2021-02-23 op 
 35 6b191ed5 2021-02-23 op static int	ibuf_realloc(struct ibuf *, size_t);
 36 6b191ed5 2021-02-23 op static void	ibuf_enqueue(struct msgbuf *, struct ibuf *);
 37 6b191ed5 2021-02-23 op static void	ibuf_dequeue(struct msgbuf *, struct ibuf *);
 38 c9c44c65 2023-07-02 op static void	msgbuf_drain(struct msgbuf *, size_t);
 39 6b191ed5 2021-02-23 op 
 40 6b191ed5 2021-02-23 op struct ibuf *
 41 6b191ed5 2021-02-23 op ibuf_open(size_t len)
 42 6b191ed5 2021-02-23 op {
 43 6b191ed5 2021-02-23 op 	struct ibuf	*buf;
 44 6b191ed5 2021-02-23 op 
 45 c9c44c65 2023-07-02 op 	if (len == 0) {
 46 c9c44c65 2023-07-02 op 		errno = EINVAL;
 47 c9c44c65 2023-07-02 op 		return (NULL);
 48 c9c44c65 2023-07-02 op 	}
 49 6b191ed5 2021-02-23 op 	if ((buf = calloc(1, sizeof(struct ibuf))) == NULL)
 50 6b191ed5 2021-02-23 op 		return (NULL);
 51 c9c44c65 2023-07-02 op 	if ((buf->buf = calloc(len, 1)) == NULL) {
 52 6b191ed5 2021-02-23 op 		free(buf);
 53 6b191ed5 2021-02-23 op 		return (NULL);
 54 6b191ed5 2021-02-23 op 	}
 55 6b191ed5 2021-02-23 op 	buf->size = buf->max = len;
 56 6b191ed5 2021-02-23 op 	buf->fd = -1;
 57 6b191ed5 2021-02-23 op 
 58 6b191ed5 2021-02-23 op 	return (buf);
 59 6b191ed5 2021-02-23 op }
 60 6b191ed5 2021-02-23 op 
 61 6b191ed5 2021-02-23 op struct ibuf *
 62 6b191ed5 2021-02-23 op ibuf_dynamic(size_t len, size_t max)
 63 6b191ed5 2021-02-23 op {
 64 6b191ed5 2021-02-23 op 	struct ibuf	*buf;
 65 6b191ed5 2021-02-23 op 
 66 12eb29d8 2024-01-07 op 	if (max == 0 || max < len) {
 67 c9c44c65 2023-07-02 op 		errno = EINVAL;
 68 6b191ed5 2021-02-23 op 		return (NULL);
 69 c9c44c65 2023-07-02 op 	}
 70 6b191ed5 2021-02-23 op 
 71 c9c44c65 2023-07-02 op 	if ((buf = calloc(1, sizeof(struct ibuf))) == NULL)
 72 6b191ed5 2021-02-23 op 		return (NULL);
 73 c9c44c65 2023-07-02 op 	if (len > 0) {
 74 c9c44c65 2023-07-02 op 		if ((buf->buf = calloc(len, 1)) == NULL) {
 75 c9c44c65 2023-07-02 op 			free(buf);
 76 c9c44c65 2023-07-02 op 			return (NULL);
 77 c9c44c65 2023-07-02 op 		}
 78 c9c44c65 2023-07-02 op 	}
 79 c9c44c65 2023-07-02 op 	buf->size = len;
 80 c9c44c65 2023-07-02 op 	buf->max = max;
 81 c9c44c65 2023-07-02 op 	buf->fd = -1;
 82 6b191ed5 2021-02-23 op 
 83 6b191ed5 2021-02-23 op 	return (buf);
 84 6b191ed5 2021-02-23 op }
 85 6b191ed5 2021-02-23 op 
 86 6b191ed5 2021-02-23 op static int
 87 6b191ed5 2021-02-23 op ibuf_realloc(struct ibuf *buf, size_t len)
 88 6b191ed5 2021-02-23 op {
 89 6b191ed5 2021-02-23 op 	unsigned char	*b;
 90 6b191ed5 2021-02-23 op 
 91 6b191ed5 2021-02-23 op 	/* on static buffers max is eq size and so the following fails */
 92 c9c44c65 2023-07-02 op 	if (len > SIZE_MAX - buf->wpos || buf->wpos + len > buf->max) {
 93 6b191ed5 2021-02-23 op 		errno = ERANGE;
 94 6b191ed5 2021-02-23 op 		return (-1);
 95 6b191ed5 2021-02-23 op 	}
 96 6b191ed5 2021-02-23 op 
 97 6b191ed5 2021-02-23 op 	b = recallocarray(buf->buf, buf->size, buf->wpos + len, 1);
 98 6b191ed5 2021-02-23 op 	if (b == NULL)
 99 6b191ed5 2021-02-23 op 		return (-1);
100 6b191ed5 2021-02-23 op 	buf->buf = b;
101 6b191ed5 2021-02-23 op 	buf->size = buf->wpos + len;
102 6b191ed5 2021-02-23 op 
103 6b191ed5 2021-02-23 op 	return (0);
104 6b191ed5 2021-02-23 op }
105 6b191ed5 2021-02-23 op 
106 6b191ed5 2021-02-23 op void *
107 6b191ed5 2021-02-23 op ibuf_reserve(struct ibuf *buf, size_t len)
108 6b191ed5 2021-02-23 op {
109 6b191ed5 2021-02-23 op 	void	*b;
110 6b191ed5 2021-02-23 op 
111 12eb29d8 2024-01-07 op 	if (len > SIZE_MAX - buf->wpos || buf->max == 0) {
112 c9c44c65 2023-07-02 op 		errno = ERANGE;
113 c9c44c65 2023-07-02 op 		return (NULL);
114 c9c44c65 2023-07-02 op 	}
115 c9c44c65 2023-07-02 op 
116 6b191ed5 2021-02-23 op 	if (buf->wpos + len > buf->size)
117 6b191ed5 2021-02-23 op 		if (ibuf_realloc(buf, len) == -1)
118 6b191ed5 2021-02-23 op 			return (NULL);
119 6b191ed5 2021-02-23 op 
120 6b191ed5 2021-02-23 op 	b = buf->buf + buf->wpos;
121 6b191ed5 2021-02-23 op 	buf->wpos += len;
122 6b191ed5 2021-02-23 op 	return (b);
123 6b191ed5 2021-02-23 op }
124 6b191ed5 2021-02-23 op 
125 c9c44c65 2023-07-02 op int
126 c9c44c65 2023-07-02 op ibuf_add(struct ibuf *buf, const void *data, size_t len)
127 c9c44c65 2023-07-02 op {
128 c9c44c65 2023-07-02 op 	void *b;
129 c9c44c65 2023-07-02 op 
130 c9c44c65 2023-07-02 op 	if ((b = ibuf_reserve(buf, len)) == NULL)
131 c9c44c65 2023-07-02 op 		return (-1);
132 c9c44c65 2023-07-02 op 
133 c9c44c65 2023-07-02 op 	memcpy(b, data, len);
134 c9c44c65 2023-07-02 op 	return (0);
135 c9c44c65 2023-07-02 op }
136 c9c44c65 2023-07-02 op 
137 c9c44c65 2023-07-02 op int
138 12eb29d8 2024-01-07 op ibuf_add_ibuf(struct ibuf *buf, const struct ibuf *from)
139 12eb29d8 2024-01-07 op {
140 12eb29d8 2024-01-07 op 	return ibuf_add(buf, ibuf_data(from), ibuf_size(from));
141 12eb29d8 2024-01-07 op }
142 12eb29d8 2024-01-07 op 
143 12eb29d8 2024-01-07 op /* remove after tree is converted */
144 12eb29d8 2024-01-07 op int
145 c9c44c65 2023-07-02 op ibuf_add_buf(struct ibuf *buf, const struct ibuf *from)
146 c9c44c65 2023-07-02 op {
147 12eb29d8 2024-01-07 op 	return ibuf_add_ibuf(buf, from);
148 c9c44c65 2023-07-02 op }
149 c9c44c65 2023-07-02 op 
150 c9c44c65 2023-07-02 op int
151 c9c44c65 2023-07-02 op ibuf_add_n8(struct ibuf *buf, uint64_t value)
152 c9c44c65 2023-07-02 op {
153 c9c44c65 2023-07-02 op 	uint8_t v;
154 c9c44c65 2023-07-02 op 
155 c9c44c65 2023-07-02 op 	if (value > UINT8_MAX) {
156 c9c44c65 2023-07-02 op 		errno = EINVAL;
157 c9c44c65 2023-07-02 op 		return (-1);
158 c9c44c65 2023-07-02 op 	}
159 c9c44c65 2023-07-02 op 	v = value;
160 c9c44c65 2023-07-02 op 	return ibuf_add(buf, &v, sizeof(v));
161 c9c44c65 2023-07-02 op }
162 c9c44c65 2023-07-02 op 
163 c9c44c65 2023-07-02 op int
164 c9c44c65 2023-07-02 op ibuf_add_n16(struct ibuf *buf, uint64_t value)
165 c9c44c65 2023-07-02 op {
166 c9c44c65 2023-07-02 op 	uint16_t v;
167 c9c44c65 2023-07-02 op 
168 c9c44c65 2023-07-02 op 	if (value > UINT16_MAX) {
169 c9c44c65 2023-07-02 op 		errno = EINVAL;
170 c9c44c65 2023-07-02 op 		return (-1);
171 c9c44c65 2023-07-02 op 	}
172 c9c44c65 2023-07-02 op 	v = htobe16(value);
173 c9c44c65 2023-07-02 op 	return ibuf_add(buf, &v, sizeof(v));
174 c9c44c65 2023-07-02 op }
175 c9c44c65 2023-07-02 op 
176 c9c44c65 2023-07-02 op int
177 c9c44c65 2023-07-02 op ibuf_add_n32(struct ibuf *buf, uint64_t value)
178 c9c44c65 2023-07-02 op {
179 c9c44c65 2023-07-02 op 	uint32_t v;
180 c9c44c65 2023-07-02 op 
181 c9c44c65 2023-07-02 op 	if (value > UINT32_MAX) {
182 c9c44c65 2023-07-02 op 		errno = EINVAL;
183 c9c44c65 2023-07-02 op 		return (-1);
184 c9c44c65 2023-07-02 op 	}
185 c9c44c65 2023-07-02 op 	v = htobe32(value);
186 c9c44c65 2023-07-02 op 	return ibuf_add(buf, &v, sizeof(v));
187 c9c44c65 2023-07-02 op }
188 c9c44c65 2023-07-02 op 
189 c9c44c65 2023-07-02 op int
190 c9c44c65 2023-07-02 op ibuf_add_n64(struct ibuf *buf, uint64_t value)
191 c9c44c65 2023-07-02 op {
192 c9c44c65 2023-07-02 op 	value = htobe64(value);
193 12eb29d8 2024-01-07 op 	return ibuf_add(buf, &value, sizeof(value));
194 12eb29d8 2024-01-07 op }
195 12eb29d8 2024-01-07 op 
196 12eb29d8 2024-01-07 op int
197 12eb29d8 2024-01-07 op ibuf_add_h16(struct ibuf *buf, uint64_t value)
198 12eb29d8 2024-01-07 op {
199 12eb29d8 2024-01-07 op 	uint16_t v;
200 12eb29d8 2024-01-07 op 
201 12eb29d8 2024-01-07 op 	if (value > UINT16_MAX) {
202 12eb29d8 2024-01-07 op 		errno = EINVAL;
203 12eb29d8 2024-01-07 op 		return (-1);
204 12eb29d8 2024-01-07 op 	}
205 12eb29d8 2024-01-07 op 	v = value;
206 12eb29d8 2024-01-07 op 	return ibuf_add(buf, &v, sizeof(v));
207 12eb29d8 2024-01-07 op }
208 12eb29d8 2024-01-07 op 
209 12eb29d8 2024-01-07 op int
210 12eb29d8 2024-01-07 op ibuf_add_h32(struct ibuf *buf, uint64_t value)
211 12eb29d8 2024-01-07 op {
212 12eb29d8 2024-01-07 op 	uint32_t v;
213 12eb29d8 2024-01-07 op 
214 12eb29d8 2024-01-07 op 	if (value > UINT32_MAX) {
215 12eb29d8 2024-01-07 op 		errno = EINVAL;
216 12eb29d8 2024-01-07 op 		return (-1);
217 12eb29d8 2024-01-07 op 	}
218 12eb29d8 2024-01-07 op 	v = value;
219 12eb29d8 2024-01-07 op 	return ibuf_add(buf, &v, sizeof(v));
220 12eb29d8 2024-01-07 op }
221 12eb29d8 2024-01-07 op 
222 12eb29d8 2024-01-07 op int
223 12eb29d8 2024-01-07 op ibuf_add_h64(struct ibuf *buf, uint64_t value)
224 12eb29d8 2024-01-07 op {
225 c9c44c65 2023-07-02 op 	return ibuf_add(buf, &value, sizeof(value));
226 c9c44c65 2023-07-02 op }
227 c9c44c65 2023-07-02 op 
228 c9c44c65 2023-07-02 op int
229 c9c44c65 2023-07-02 op ibuf_add_zero(struct ibuf *buf, size_t len)
230 c9c44c65 2023-07-02 op {
231 c9c44c65 2023-07-02 op 	void *b;
232 c9c44c65 2023-07-02 op 
233 c9c44c65 2023-07-02 op 	if ((b = ibuf_reserve(buf, len)) == NULL)
234 c9c44c65 2023-07-02 op 		return (-1);
235 12eb29d8 2024-01-07 op 	memset(b, 0, len);
236 c9c44c65 2023-07-02 op 	return (0);
237 c9c44c65 2023-07-02 op }
238 c9c44c65 2023-07-02 op 
239 6b191ed5 2021-02-23 op void *
240 6b191ed5 2021-02-23 op ibuf_seek(struct ibuf *buf, size_t pos, size_t len)
241 6b191ed5 2021-02-23 op {
242 12eb29d8 2024-01-07 op 	/* only allow seeking between rpos and wpos */
243 12eb29d8 2024-01-07 op 	if (ibuf_size(buf) < pos || SIZE_MAX - pos < len ||
244 12eb29d8 2024-01-07 op 	    ibuf_size(buf) < pos + len) {
245 c9c44c65 2023-07-02 op 		errno = ERANGE;
246 6b191ed5 2021-02-23 op 		return (NULL);
247 c9c44c65 2023-07-02 op 	}
248 6b191ed5 2021-02-23 op 
249 12eb29d8 2024-01-07 op 	return (buf->buf + buf->rpos + pos);
250 6b191ed5 2021-02-23 op }
251 6b191ed5 2021-02-23 op 
252 c9c44c65 2023-07-02 op int
253 c9c44c65 2023-07-02 op ibuf_set(struct ibuf *buf, size_t pos, const void *data, size_t len)
254 c9c44c65 2023-07-02 op {
255 c9c44c65 2023-07-02 op 	void *b;
256 c9c44c65 2023-07-02 op 
257 c9c44c65 2023-07-02 op 	if ((b = ibuf_seek(buf, pos, len)) == NULL)
258 c9c44c65 2023-07-02 op 		return (-1);
259 c9c44c65 2023-07-02 op 
260 c9c44c65 2023-07-02 op 	memcpy(b, data, len);
261 c9c44c65 2023-07-02 op 	return (0);
262 c9c44c65 2023-07-02 op }
263 c9c44c65 2023-07-02 op 
264 c9c44c65 2023-07-02 op int
265 c9c44c65 2023-07-02 op ibuf_set_n8(struct ibuf *buf, size_t pos, uint64_t value)
266 c9c44c65 2023-07-02 op {
267 c9c44c65 2023-07-02 op 	uint8_t v;
268 c9c44c65 2023-07-02 op 
269 c9c44c65 2023-07-02 op 	if (value > UINT8_MAX) {
270 c9c44c65 2023-07-02 op 		errno = EINVAL;
271 c9c44c65 2023-07-02 op 		return (-1);
272 c9c44c65 2023-07-02 op 	}
273 c9c44c65 2023-07-02 op 	v = value;
274 c9c44c65 2023-07-02 op 	return (ibuf_set(buf, pos, &v, sizeof(v)));
275 c9c44c65 2023-07-02 op }
276 c9c44c65 2023-07-02 op 
277 c9c44c65 2023-07-02 op int
278 c9c44c65 2023-07-02 op ibuf_set_n16(struct ibuf *buf, size_t pos, uint64_t value)
279 c9c44c65 2023-07-02 op {
280 c9c44c65 2023-07-02 op 	uint16_t v;
281 c9c44c65 2023-07-02 op 
282 c9c44c65 2023-07-02 op 	if (value > UINT16_MAX) {
283 c9c44c65 2023-07-02 op 		errno = EINVAL;
284 c9c44c65 2023-07-02 op 		return (-1);
285 c9c44c65 2023-07-02 op 	}
286 c9c44c65 2023-07-02 op 	v = htobe16(value);
287 c9c44c65 2023-07-02 op 	return (ibuf_set(buf, pos, &v, sizeof(v)));
288 c9c44c65 2023-07-02 op }
289 c9c44c65 2023-07-02 op 
290 c9c44c65 2023-07-02 op int
291 c9c44c65 2023-07-02 op ibuf_set_n32(struct ibuf *buf, size_t pos, uint64_t value)
292 c9c44c65 2023-07-02 op {
293 c9c44c65 2023-07-02 op 	uint32_t v;
294 c9c44c65 2023-07-02 op 
295 c9c44c65 2023-07-02 op 	if (value > UINT32_MAX) {
296 c9c44c65 2023-07-02 op 		errno = EINVAL;
297 c9c44c65 2023-07-02 op 		return (-1);
298 c9c44c65 2023-07-02 op 	}
299 c9c44c65 2023-07-02 op 	v = htobe32(value);
300 c9c44c65 2023-07-02 op 	return (ibuf_set(buf, pos, &v, sizeof(v)));
301 c9c44c65 2023-07-02 op }
302 c9c44c65 2023-07-02 op 
303 c9c44c65 2023-07-02 op int
304 c9c44c65 2023-07-02 op ibuf_set_n64(struct ibuf *buf, size_t pos, uint64_t value)
305 c9c44c65 2023-07-02 op {
306 c9c44c65 2023-07-02 op 	value = htobe64(value);
307 c9c44c65 2023-07-02 op 	return (ibuf_set(buf, pos, &value, sizeof(value)));
308 c9c44c65 2023-07-02 op }
309 c9c44c65 2023-07-02 op 
310 12eb29d8 2024-01-07 op int
311 12eb29d8 2024-01-07 op ibuf_set_h16(struct ibuf *buf, size_t pos, uint64_t value)
312 c9c44c65 2023-07-02 op {
313 12eb29d8 2024-01-07 op 	uint16_t v;
314 c9c44c65 2023-07-02 op 
315 12eb29d8 2024-01-07 op 	if (value > UINT16_MAX) {
316 12eb29d8 2024-01-07 op 		errno = EINVAL;
317 12eb29d8 2024-01-07 op 		return (-1);
318 12eb29d8 2024-01-07 op 	}
319 12eb29d8 2024-01-07 op 	v = value;
320 12eb29d8 2024-01-07 op 	return (ibuf_set(buf, pos, &v, sizeof(v)));
321 6b191ed5 2021-02-23 op }
322 6b191ed5 2021-02-23 op 
323 12eb29d8 2024-01-07 op int
324 12eb29d8 2024-01-07 op ibuf_set_h32(struct ibuf *buf, size_t pos, uint64_t value)
325 12eb29d8 2024-01-07 op {
326 12eb29d8 2024-01-07 op 	uint32_t v;
327 12eb29d8 2024-01-07 op 
328 12eb29d8 2024-01-07 op 	if (value > UINT32_MAX) {
329 12eb29d8 2024-01-07 op 		errno = EINVAL;
330 12eb29d8 2024-01-07 op 		return (-1);
331 12eb29d8 2024-01-07 op 	}
332 12eb29d8 2024-01-07 op 	v = value;
333 12eb29d8 2024-01-07 op 	return (ibuf_set(buf, pos, &v, sizeof(v)));
334 12eb29d8 2024-01-07 op }
335 12eb29d8 2024-01-07 op 
336 12eb29d8 2024-01-07 op int
337 12eb29d8 2024-01-07 op ibuf_set_h64(struct ibuf *buf, size_t pos, uint64_t value)
338 12eb29d8 2024-01-07 op {
339 12eb29d8 2024-01-07 op 	return (ibuf_set(buf, pos, &value, sizeof(value)));
340 12eb29d8 2024-01-07 op }
341 12eb29d8 2024-01-07 op 
342 12eb29d8 2024-01-07 op void *
343 12eb29d8 2024-01-07 op ibuf_data(const struct ibuf *buf)
344 12eb29d8 2024-01-07 op {
345 12eb29d8 2024-01-07 op 	return (buf->buf + buf->rpos);
346 12eb29d8 2024-01-07 op }
347 12eb29d8 2024-01-07 op 
348 6b191ed5 2021-02-23 op size_t
349 12eb29d8 2024-01-07 op ibuf_size(const struct ibuf *buf)
350 6b191ed5 2021-02-23 op {
351 12eb29d8 2024-01-07 op 	return (buf->wpos - buf->rpos);
352 12eb29d8 2024-01-07 op }
353 12eb29d8 2024-01-07 op 
354 12eb29d8 2024-01-07 op size_t
355 12eb29d8 2024-01-07 op ibuf_left(const struct ibuf *buf)
356 12eb29d8 2024-01-07 op {
357 12eb29d8 2024-01-07 op 	if (buf->max == 0)
358 12eb29d8 2024-01-07 op 		return (0);
359 6b191ed5 2021-02-23 op 	return (buf->max - buf->wpos);
360 6b191ed5 2021-02-23 op }
361 6b191ed5 2021-02-23 op 
362 12eb29d8 2024-01-07 op int
363 12eb29d8 2024-01-07 op ibuf_truncate(struct ibuf *buf, size_t len)
364 12eb29d8 2024-01-07 op {
365 12eb29d8 2024-01-07 op 	if (ibuf_size(buf) >= len) {
366 12eb29d8 2024-01-07 op 		buf->wpos = buf->rpos + len;
367 12eb29d8 2024-01-07 op 		return (0);
368 12eb29d8 2024-01-07 op 	}
369 12eb29d8 2024-01-07 op 	if (buf->max == 0) {
370 12eb29d8 2024-01-07 op 		/* only allow to truncate down */
371 12eb29d8 2024-01-07 op 		errno = ERANGE;
372 12eb29d8 2024-01-07 op 		return (-1);
373 12eb29d8 2024-01-07 op 	}
374 12eb29d8 2024-01-07 op 	return ibuf_add_zero(buf, len - ibuf_size(buf));
375 12eb29d8 2024-01-07 op }
376 12eb29d8 2024-01-07 op 
377 6b191ed5 2021-02-23 op void
378 12eb29d8 2024-01-07 op ibuf_rewind(struct ibuf *buf)
379 12eb29d8 2024-01-07 op {
380 12eb29d8 2024-01-07 op 	buf->rpos = 0;
381 12eb29d8 2024-01-07 op }
382 12eb29d8 2024-01-07 op 
383 12eb29d8 2024-01-07 op void
384 6b191ed5 2021-02-23 op ibuf_close(struct msgbuf *msgbuf, struct ibuf *buf)
385 6b191ed5 2021-02-23 op {
386 6b191ed5 2021-02-23 op 	ibuf_enqueue(msgbuf, buf);
387 6b191ed5 2021-02-23 op }
388 6b191ed5 2021-02-23 op 
389 c9c44c65 2023-07-02 op void
390 12eb29d8 2024-01-07 op ibuf_from_buffer(struct ibuf *buf, void *data, size_t len)
391 12eb29d8 2024-01-07 op {
392 12eb29d8 2024-01-07 op 	memset(buf, 0, sizeof(*buf));
393 12eb29d8 2024-01-07 op 	buf->buf = data;
394 12eb29d8 2024-01-07 op 	buf->size = buf->wpos = len;
395 12eb29d8 2024-01-07 op 	buf->fd = -1;
396 12eb29d8 2024-01-07 op }
397 12eb29d8 2024-01-07 op 
398 12eb29d8 2024-01-07 op void
399 12eb29d8 2024-01-07 op ibuf_from_ibuf(struct ibuf *buf, const struct ibuf *from)
400 12eb29d8 2024-01-07 op {
401 12eb29d8 2024-01-07 op 	ibuf_from_buffer(buf, ibuf_data(from), ibuf_size(from));
402 12eb29d8 2024-01-07 op }
403 12eb29d8 2024-01-07 op 
404 12eb29d8 2024-01-07 op int
405 12eb29d8 2024-01-07 op ibuf_get(struct ibuf *buf, void *data, size_t len)
406 12eb29d8 2024-01-07 op {
407 12eb29d8 2024-01-07 op 	if (ibuf_size(buf) < len) {
408 12eb29d8 2024-01-07 op 		errno = EBADMSG;
409 12eb29d8 2024-01-07 op 		return (-1);
410 12eb29d8 2024-01-07 op 	}
411 12eb29d8 2024-01-07 op 
412 12eb29d8 2024-01-07 op 	memcpy(data, ibuf_data(buf), len);
413 12eb29d8 2024-01-07 op 	buf->rpos += len;
414 12eb29d8 2024-01-07 op 	return (0);
415 12eb29d8 2024-01-07 op }
416 12eb29d8 2024-01-07 op 
417 12eb29d8 2024-01-07 op int
418 12eb29d8 2024-01-07 op ibuf_get_ibuf(struct ibuf *buf, size_t len, struct ibuf *new)
419 12eb29d8 2024-01-07 op {
420 12eb29d8 2024-01-07 op 	if (ibuf_size(buf) < len) {
421 12eb29d8 2024-01-07 op 		errno = EBADMSG;
422 12eb29d8 2024-01-07 op 		return (-1);
423 12eb29d8 2024-01-07 op 	}
424 12eb29d8 2024-01-07 op 
425 12eb29d8 2024-01-07 op 	ibuf_from_buffer(new, ibuf_data(buf), len);
426 12eb29d8 2024-01-07 op 	buf->rpos += len;
427 12eb29d8 2024-01-07 op 	return (0);
428 12eb29d8 2024-01-07 op }
429 12eb29d8 2024-01-07 op 
430 12eb29d8 2024-01-07 op int
431 12eb29d8 2024-01-07 op ibuf_get_n8(struct ibuf *buf, uint8_t *value)
432 12eb29d8 2024-01-07 op {
433 12eb29d8 2024-01-07 op 	return ibuf_get(buf, value, sizeof(*value));
434 12eb29d8 2024-01-07 op }
435 12eb29d8 2024-01-07 op 
436 12eb29d8 2024-01-07 op int
437 12eb29d8 2024-01-07 op ibuf_get_n16(struct ibuf *buf, uint16_t *value)
438 12eb29d8 2024-01-07 op {
439 12eb29d8 2024-01-07 op 	int rv;
440 12eb29d8 2024-01-07 op 
441 12eb29d8 2024-01-07 op 	rv = ibuf_get(buf, value, sizeof(*value));
442 12eb29d8 2024-01-07 op 	*value = be16toh(*value);
443 12eb29d8 2024-01-07 op 	return (rv);
444 12eb29d8 2024-01-07 op }
445 12eb29d8 2024-01-07 op 
446 12eb29d8 2024-01-07 op int
447 12eb29d8 2024-01-07 op ibuf_get_n32(struct ibuf *buf, uint32_t *value)
448 12eb29d8 2024-01-07 op {
449 12eb29d8 2024-01-07 op 	int rv;
450 12eb29d8 2024-01-07 op 
451 12eb29d8 2024-01-07 op 	rv = ibuf_get(buf, value, sizeof(*value));
452 12eb29d8 2024-01-07 op 	*value = be32toh(*value);
453 12eb29d8 2024-01-07 op 	return (rv);
454 12eb29d8 2024-01-07 op }
455 12eb29d8 2024-01-07 op 
456 12eb29d8 2024-01-07 op int
457 12eb29d8 2024-01-07 op ibuf_get_n64(struct ibuf *buf, uint64_t *value)
458 12eb29d8 2024-01-07 op {
459 12eb29d8 2024-01-07 op 	int rv;
460 12eb29d8 2024-01-07 op 
461 12eb29d8 2024-01-07 op 	rv = ibuf_get(buf, value, sizeof(*value));
462 12eb29d8 2024-01-07 op 	*value = be64toh(*value);
463 12eb29d8 2024-01-07 op 	return (rv);
464 12eb29d8 2024-01-07 op }
465 12eb29d8 2024-01-07 op 
466 12eb29d8 2024-01-07 op int
467 12eb29d8 2024-01-07 op ibuf_get_h16(struct ibuf *buf, uint16_t *value)
468 12eb29d8 2024-01-07 op {
469 12eb29d8 2024-01-07 op 	return ibuf_get(buf, value, sizeof(*value));
470 12eb29d8 2024-01-07 op }
471 12eb29d8 2024-01-07 op 
472 12eb29d8 2024-01-07 op int
473 12eb29d8 2024-01-07 op ibuf_get_h32(struct ibuf *buf, uint32_t *value)
474 12eb29d8 2024-01-07 op {
475 12eb29d8 2024-01-07 op 	return ibuf_get(buf, value, sizeof(*value));
476 12eb29d8 2024-01-07 op }
477 12eb29d8 2024-01-07 op 
478 12eb29d8 2024-01-07 op int
479 12eb29d8 2024-01-07 op ibuf_get_h64(struct ibuf *buf, uint64_t *value)
480 12eb29d8 2024-01-07 op {
481 12eb29d8 2024-01-07 op 	return ibuf_get(buf, value, sizeof(*value));
482 12eb29d8 2024-01-07 op }
483 12eb29d8 2024-01-07 op 
484 12eb29d8 2024-01-07 op int
485 12eb29d8 2024-01-07 op ibuf_skip(struct ibuf *buf, size_t len)
486 12eb29d8 2024-01-07 op {
487 12eb29d8 2024-01-07 op 	if (ibuf_size(buf) < len) {
488 12eb29d8 2024-01-07 op 		errno = EBADMSG;
489 12eb29d8 2024-01-07 op 		return (-1);
490 12eb29d8 2024-01-07 op 	}
491 12eb29d8 2024-01-07 op 
492 12eb29d8 2024-01-07 op 	buf->rpos += len;
493 12eb29d8 2024-01-07 op 	return (0);
494 12eb29d8 2024-01-07 op }
495 12eb29d8 2024-01-07 op 
496 12eb29d8 2024-01-07 op void
497 c9c44c65 2023-07-02 op ibuf_free(struct ibuf *buf)
498 c9c44c65 2023-07-02 op {
499 c9c44c65 2023-07-02 op 	if (buf == NULL)
500 c9c44c65 2023-07-02 op 		return;
501 12eb29d8 2024-01-07 op 	if (buf->max == 0)	/* if buf lives on the stack */
502 12eb29d8 2024-01-07 op 		abort();	/* abort before causing more harm */
503 c9c44c65 2023-07-02 op 	if (buf->fd != -1)
504 c9c44c65 2023-07-02 op 		close(buf->fd);
505 c9c44c65 2023-07-02 op 	freezero(buf->buf, buf->size);
506 c9c44c65 2023-07-02 op 	free(buf);
507 c9c44c65 2023-07-02 op }
508 c9c44c65 2023-07-02 op 
509 6b191ed5 2021-02-23 op int
510 c9c44c65 2023-07-02 op ibuf_fd_avail(struct ibuf *buf)
511 c9c44c65 2023-07-02 op {
512 c9c44c65 2023-07-02 op 	return (buf->fd != -1);
513 c9c44c65 2023-07-02 op }
514 c9c44c65 2023-07-02 op 
515 c9c44c65 2023-07-02 op int
516 c9c44c65 2023-07-02 op ibuf_fd_get(struct ibuf *buf)
517 c9c44c65 2023-07-02 op {
518 c9c44c65 2023-07-02 op 	int fd;
519 c9c44c65 2023-07-02 op 
520 c9c44c65 2023-07-02 op 	fd = buf->fd;
521 c9c44c65 2023-07-02 op 	buf->fd = -1;
522 c9c44c65 2023-07-02 op 	return (fd);
523 c9c44c65 2023-07-02 op }
524 c9c44c65 2023-07-02 op 
525 c9c44c65 2023-07-02 op void
526 c9c44c65 2023-07-02 op ibuf_fd_set(struct ibuf *buf, int fd)
527 c9c44c65 2023-07-02 op {
528 12eb29d8 2024-01-07 op 	if (buf->max == 0)	/* if buf lives on the stack */
529 12eb29d8 2024-01-07 op 		abort();	/* abort before causing more harm */
530 c9c44c65 2023-07-02 op 	if (buf->fd != -1)
531 c9c44c65 2023-07-02 op 		close(buf->fd);
532 c9c44c65 2023-07-02 op 	buf->fd = fd;
533 c9c44c65 2023-07-02 op }
534 c9c44c65 2023-07-02 op 
535 c9c44c65 2023-07-02 op int
536 6b191ed5 2021-02-23 op ibuf_write(struct msgbuf *msgbuf)
537 6b191ed5 2021-02-23 op {
538 6b191ed5 2021-02-23 op 	struct iovec	 iov[IOV_MAX];
539 6b191ed5 2021-02-23 op 	struct ibuf	*buf;
540 6b191ed5 2021-02-23 op 	unsigned int	 i = 0;
541 6b191ed5 2021-02-23 op 	ssize_t	n;
542 6b191ed5 2021-02-23 op 
543 6b191ed5 2021-02-23 op 	memset(&iov, 0, sizeof(iov));
544 6b191ed5 2021-02-23 op 	TAILQ_FOREACH(buf, &msgbuf->bufs, entry) {
545 6b191ed5 2021-02-23 op 		if (i >= IOV_MAX)
546 6b191ed5 2021-02-23 op 			break;
547 12eb29d8 2024-01-07 op 		iov[i].iov_base = ibuf_data(buf);
548 12eb29d8 2024-01-07 op 		iov[i].iov_len = ibuf_size(buf);
549 6b191ed5 2021-02-23 op 		i++;
550 6b191ed5 2021-02-23 op 	}
551 6b191ed5 2021-02-23 op 
552 6b191ed5 2021-02-23 op again:
553 6b191ed5 2021-02-23 op 	if ((n = writev(msgbuf->fd, iov, i)) == -1) {
554 6b191ed5 2021-02-23 op 		if (errno == EINTR)
555 6b191ed5 2021-02-23 op 			goto again;
556 6b191ed5 2021-02-23 op 		if (errno == ENOBUFS)
557 6b191ed5 2021-02-23 op 			errno = EAGAIN;
558 6b191ed5 2021-02-23 op 		return (-1);
559 6b191ed5 2021-02-23 op 	}
560 6b191ed5 2021-02-23 op 
561 6b191ed5 2021-02-23 op 	if (n == 0) {			/* connection closed */
562 6b191ed5 2021-02-23 op 		errno = 0;
563 6b191ed5 2021-02-23 op 		return (0);
564 6b191ed5 2021-02-23 op 	}
565 6b191ed5 2021-02-23 op 
566 6b191ed5 2021-02-23 op 	msgbuf_drain(msgbuf, n);
567 6b191ed5 2021-02-23 op 
568 6b191ed5 2021-02-23 op 	return (1);
569 6b191ed5 2021-02-23 op }
570 6b191ed5 2021-02-23 op 
571 6b191ed5 2021-02-23 op void
572 6b191ed5 2021-02-23 op msgbuf_init(struct msgbuf *msgbuf)
573 6b191ed5 2021-02-23 op {
574 6b191ed5 2021-02-23 op 	msgbuf->queued = 0;
575 6b191ed5 2021-02-23 op 	msgbuf->fd = -1;
576 6b191ed5 2021-02-23 op 	TAILQ_INIT(&msgbuf->bufs);
577 6b191ed5 2021-02-23 op }
578 6b191ed5 2021-02-23 op 
579 c9c44c65 2023-07-02 op static void
580 6b191ed5 2021-02-23 op msgbuf_drain(struct msgbuf *msgbuf, size_t n)
581 6b191ed5 2021-02-23 op {
582 6b191ed5 2021-02-23 op 	struct ibuf	*buf, *next;
583 6b191ed5 2021-02-23 op 
584 6b191ed5 2021-02-23 op 	for (buf = TAILQ_FIRST(&msgbuf->bufs); buf != NULL && n > 0;
585 6b191ed5 2021-02-23 op 	    buf = next) {
586 6b191ed5 2021-02-23 op 		next = TAILQ_NEXT(buf, entry);
587 12eb29d8 2024-01-07 op 		if (n >= ibuf_size(buf)) {
588 12eb29d8 2024-01-07 op 			n -= ibuf_size(buf);
589 6b191ed5 2021-02-23 op 			ibuf_dequeue(msgbuf, buf);
590 6b191ed5 2021-02-23 op 		} else {
591 6b191ed5 2021-02-23 op 			buf->rpos += n;
592 6b191ed5 2021-02-23 op 			n = 0;
593 6b191ed5 2021-02-23 op 		}
594 6b191ed5 2021-02-23 op 	}
595 6b191ed5 2021-02-23 op }
596 6b191ed5 2021-02-23 op 
597 6b191ed5 2021-02-23 op void
598 6b191ed5 2021-02-23 op msgbuf_clear(struct msgbuf *msgbuf)
599 6b191ed5 2021-02-23 op {
600 6b191ed5 2021-02-23 op 	struct ibuf	*buf;
601 6b191ed5 2021-02-23 op 
602 6b191ed5 2021-02-23 op 	while ((buf = TAILQ_FIRST(&msgbuf->bufs)) != NULL)
603 6b191ed5 2021-02-23 op 		ibuf_dequeue(msgbuf, buf);
604 6b191ed5 2021-02-23 op }
605 6b191ed5 2021-02-23 op 
606 6b191ed5 2021-02-23 op int
607 6b191ed5 2021-02-23 op msgbuf_write(struct msgbuf *msgbuf)
608 6b191ed5 2021-02-23 op {
609 6b191ed5 2021-02-23 op 	struct iovec	 iov[IOV_MAX];
610 894e9984 2022-03-19 op 	struct ibuf	*buf, *buf0 = NULL;
611 6b191ed5 2021-02-23 op 	unsigned int	 i = 0;
612 6b191ed5 2021-02-23 op 	ssize_t		 n;
613 6b191ed5 2021-02-23 op 	struct msghdr	 msg;
614 6b191ed5 2021-02-23 op 	struct cmsghdr	*cmsg;
615 6b191ed5 2021-02-23 op 	union {
616 6b191ed5 2021-02-23 op 		struct cmsghdr	hdr;
617 6b191ed5 2021-02-23 op 		char		buf[CMSG_SPACE(sizeof(int))];
618 6b191ed5 2021-02-23 op 	} cmsgbuf;
619 6b191ed5 2021-02-23 op 
620 6b191ed5 2021-02-23 op 	memset(&iov, 0, sizeof(iov));
621 6b191ed5 2021-02-23 op 	memset(&msg, 0, sizeof(msg));
622 6b191ed5 2021-02-23 op 	memset(&cmsgbuf, 0, sizeof(cmsgbuf));
623 6b191ed5 2021-02-23 op 	TAILQ_FOREACH(buf, &msgbuf->bufs, entry) {
624 6b191ed5 2021-02-23 op 		if (i >= IOV_MAX)
625 6b191ed5 2021-02-23 op 			break;
626 894e9984 2022-03-19 op 		if (i > 0 && buf->fd != -1)
627 894e9984 2022-03-19 op 			break;
628 12eb29d8 2024-01-07 op 		iov[i].iov_base = ibuf_data(buf);
629 12eb29d8 2024-01-07 op 		iov[i].iov_len = ibuf_size(buf);
630 6b191ed5 2021-02-23 op 		i++;
631 6b191ed5 2021-02-23 op 		if (buf->fd != -1)
632 894e9984 2022-03-19 op 			buf0 = buf;
633 6b191ed5 2021-02-23 op 	}
634 6b191ed5 2021-02-23 op 
635 6b191ed5 2021-02-23 op 	msg.msg_iov = iov;
636 6b191ed5 2021-02-23 op 	msg.msg_iovlen = i;
637 6b191ed5 2021-02-23 op 
638 894e9984 2022-03-19 op 	if (buf0 != NULL) {
639 6b191ed5 2021-02-23 op 		msg.msg_control = (caddr_t)&cmsgbuf.buf;
640 6b191ed5 2021-02-23 op 		msg.msg_controllen = sizeof(cmsgbuf.buf);
641 6b191ed5 2021-02-23 op 		cmsg = CMSG_FIRSTHDR(&msg);
642 6b191ed5 2021-02-23 op 		cmsg->cmsg_len = CMSG_LEN(sizeof(int));
643 6b191ed5 2021-02-23 op 		cmsg->cmsg_level = SOL_SOCKET;
644 6b191ed5 2021-02-23 op 		cmsg->cmsg_type = SCM_RIGHTS;
645 894e9984 2022-03-19 op 		*(int *)CMSG_DATA(cmsg) = buf0->fd;
646 6b191ed5 2021-02-23 op 	}
647 6b191ed5 2021-02-23 op 
648 6b191ed5 2021-02-23 op again:
649 6b191ed5 2021-02-23 op 	if ((n = sendmsg(msgbuf->fd, &msg, 0)) == -1) {
650 6b191ed5 2021-02-23 op 		if (errno == EINTR)
651 6b191ed5 2021-02-23 op 			goto again;
652 6b191ed5 2021-02-23 op 		if (errno == ENOBUFS)
653 6b191ed5 2021-02-23 op 			errno = EAGAIN;
654 6b191ed5 2021-02-23 op 		return (-1);
655 6b191ed5 2021-02-23 op 	}
656 6b191ed5 2021-02-23 op 
657 6b191ed5 2021-02-23 op 	if (n == 0) {			/* connection closed */
658 6b191ed5 2021-02-23 op 		errno = 0;
659 6b191ed5 2021-02-23 op 		return (0);
660 6b191ed5 2021-02-23 op 	}
661 6b191ed5 2021-02-23 op 
662 6b191ed5 2021-02-23 op 	/*
663 6b191ed5 2021-02-23 op 	 * assumption: fd got sent if sendmsg sent anything
664 6b191ed5 2021-02-23 op 	 * this works because fds are passed one at a time
665 6b191ed5 2021-02-23 op 	 */
666 894e9984 2022-03-19 op 	if (buf0 != NULL) {
667 894e9984 2022-03-19 op 		close(buf0->fd);
668 894e9984 2022-03-19 op 		buf0->fd = -1;
669 6b191ed5 2021-02-23 op 	}
670 6b191ed5 2021-02-23 op 
671 6b191ed5 2021-02-23 op 	msgbuf_drain(msgbuf, n);
672 6b191ed5 2021-02-23 op 
673 6b191ed5 2021-02-23 op 	return (1);
674 6b191ed5 2021-02-23 op }
675 6b191ed5 2021-02-23 op 
676 12eb29d8 2024-01-07 op uint32_t
677 12eb29d8 2024-01-07 op msgbuf_queuelen(struct msgbuf *msgbuf)
678 12eb29d8 2024-01-07 op {
679 12eb29d8 2024-01-07 op 	return (msgbuf->queued);
680 12eb29d8 2024-01-07 op }
681 12eb29d8 2024-01-07 op 
682 6b191ed5 2021-02-23 op static void
683 6b191ed5 2021-02-23 op ibuf_enqueue(struct msgbuf *msgbuf, struct ibuf *buf)
684 6b191ed5 2021-02-23 op {
685 12eb29d8 2024-01-07 op 	if (buf->max == 0)	/* if buf lives on the stack */
686 12eb29d8 2024-01-07 op 		abort();	/* abort before causing more harm */
687 6b191ed5 2021-02-23 op 	TAILQ_INSERT_TAIL(&msgbuf->bufs, buf, entry);
688 6b191ed5 2021-02-23 op 	msgbuf->queued++;
689 6b191ed5 2021-02-23 op }
690 6b191ed5 2021-02-23 op 
691 6b191ed5 2021-02-23 op static void
692 6b191ed5 2021-02-23 op ibuf_dequeue(struct msgbuf *msgbuf, struct ibuf *buf)
693 6b191ed5 2021-02-23 op {
694 6b191ed5 2021-02-23 op 	TAILQ_REMOVE(&msgbuf->bufs, buf, entry);
695 6b191ed5 2021-02-23 op 	msgbuf->queued--;
696 6b191ed5 2021-02-23 op 	ibuf_free(buf);
697 6b191ed5 2021-02-23 op }