1 c553191e 2022-01-11 op /*
  2 c553191e 2022-01-11 op  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
  3 c553191e 2022-01-11 op  *
  4 c553191e 2022-01-11 op  * Permission to use, copy, modify, and distribute this software for any
  5 c553191e 2022-01-11 op  * purpose with or without fee is hereby granted, provided that the above
  6 c553191e 2022-01-11 op  * copyright notice and this permission notice appear in all copies.
  7 c553191e 2022-01-11 op  *
  8 c553191e 2022-01-11 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 c553191e 2022-01-11 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 c553191e 2022-01-11 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 c553191e 2022-01-11 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 c553191e 2022-01-11 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 c553191e 2022-01-11 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 c553191e 2022-01-11 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 c553191e 2022-01-11 op  */
 16 c553191e 2022-01-11 op 
 17 c553191e 2022-01-11 op #include "compat.h"
 18 c553191e 2022-01-11 op 
 19 c553191e 2022-01-11 op #include <ctype.h>
 20 c553191e 2022-01-11 op #include <errno.h>
 21 c553191e 2022-01-11 op #include <fcntl.h>
 22 c553191e 2022-01-11 op #include <stdlib.h>
 23 c553191e 2022-01-11 op #include <string.h>
 24 c553191e 2022-01-11 op #include <unistd.h>
 25 c553191e 2022-01-11 op 
 26 c553191e 2022-01-11 op #include "telescope.h"
 27 c553191e 2022-01-11 op #include "utils.h"
 28 c553191e 2022-01-11 op 
 29 c553191e 2022-01-11 op int
 30 c553191e 2022-01-11 op mark_nonblock(int fd)
 31 c553191e 2022-01-11 op {
 32 c553191e 2022-01-11 op 	int flags;
 33 c553191e 2022-01-11 op 
 34 c553191e 2022-01-11 op 	if ((flags = fcntl(fd, F_GETFL)) == -1)
 35 c553191e 2022-01-11 op 		return 0;
 36 c553191e 2022-01-11 op 	if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1)
 37 c553191e 2022-01-11 op 		return 0;
 38 c553191e 2022-01-11 op 	return 1;
 39 c553191e 2022-01-11 op }
 40 c553191e 2022-01-11 op 
 41 c553191e 2022-01-11 op int
 42 c553191e 2022-01-11 op has_prefix(const char *str, const char *prfx)
 43 c553191e 2022-01-11 op {
 44 c553191e 2022-01-11 op 	size_t i;
 45 c553191e 2022-01-11 op 
 46 c553191e 2022-01-11 op 	for (i = 0; str[i] != '\0' && prfx[i] != '\0'; ++i)
 47 c553191e 2022-01-11 op 		if (str[i] != prfx[i])
 48 c553191e 2022-01-11 op 			return 0;
 49 c553191e 2022-01-11 op 	return prfx[i] == '\0';
 50 c553191e 2022-01-11 op }
 51 c553191e 2022-01-11 op 
 52 c553191e 2022-01-11 op int
 53 c553191e 2022-01-11 op has_suffix(const char *str, const char *sufx)
 54 c553191e 2022-01-11 op {
 55 c553191e 2022-01-11 op 	size_t l, s;
 56 c553191e 2022-01-11 op 
 57 c553191e 2022-01-11 op 	l = strlen(str);
 58 c553191e 2022-01-11 op 	s = strlen(sufx);
 59 c553191e 2022-01-11 op 
 60 c553191e 2022-01-11 op 	if (l < s)
 61 c553191e 2022-01-11 op 		return 0;
 62 c553191e 2022-01-11 op 
 63 c553191e 2022-01-11 op 	return !strcmp(str + (l - s), sufx);
 64 c553191e 2022-01-11 op }
 65 c553191e 2022-01-11 op 
 66 c553191e 2022-01-11 op int
 67 c553191e 2022-01-11 op unicode_isspace(uint32_t cp)
 68 c553191e 2022-01-11 op {
 69 c553191e 2022-01-11 op 	if (cp < INT8_MAX)
 70 c553191e 2022-01-11 op 		return isspace(cp);
 71 c553191e 2022-01-11 op 	return 0;
 72 c553191e 2022-01-11 op }
 73 c553191e 2022-01-11 op 
 74 c553191e 2022-01-11 op int
 75 c553191e 2022-01-11 op unicode_isgraph(uint32_t cp)
 76 c553191e 2022-01-11 op {
 77 c553191e 2022-01-11 op 	if (cp < INT8_MAX)
 78 c553191e 2022-01-11 op 		return isgraph(cp);
 79 c553191e 2022-01-11 op 	return 1;
 80 c553191e 2022-01-11 op }
 81 c553191e 2022-01-11 op 
 82 0f559c55 2022-02-11 op void
 83 c553191e 2022-01-11 op imsg_event_add(struct imsgev *iev)
 84 c553191e 2022-01-11 op {
 85 c553191e 2022-01-11 op 	iev->events = EV_READ;
 86 c553191e 2022-01-11 op 	if (iev->ibuf.w.queued)
 87 c553191e 2022-01-11 op 		iev->events |= EV_WRITE;
 88 c553191e 2022-01-11 op 
 89 c553191e 2022-01-11 op 	event_del(&iev->ev);
 90 c553191e 2022-01-11 op 	event_set(&iev->ev, iev->ibuf.fd, iev->events, iev->handler, iev);
 91 c553191e 2022-01-11 op 	event_add(&iev->ev, NULL);
 92 c553191e 2022-01-11 op }
 93 c553191e 2022-01-11 op 
 94 c553191e 2022-01-11 op int
 95 c553191e 2022-01-11 op dispatch_imsg(struct imsgev *iev, short event, imsg_handlerfn **handlers,
 96 c553191e 2022-01-11 op     size_t size)
 97 c553191e 2022-01-11 op {
 98 c553191e 2022-01-11 op 	struct imsgbuf	*ibuf;
 99 c553191e 2022-01-11 op 	struct imsg	 imsg;
100 c553191e 2022-01-11 op 	size_t		 datalen, i;
101 c553191e 2022-01-11 op 	ssize_t		 n;
102 c553191e 2022-01-11 op 
103 c553191e 2022-01-11 op 	ibuf = &iev->ibuf;
104 c553191e 2022-01-11 op 
105 c553191e 2022-01-11 op 	if (event & EV_READ) {
106 c553191e 2022-01-11 op 		if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
107 c553191e 2022-01-11 op 			err(1, "imsg_read error");
108 c553191e 2022-01-11 op 		if (n == 0)
109 c553191e 2022-01-11 op 			return -1;
110 c553191e 2022-01-11 op 	}
111 c553191e 2022-01-11 op 	if (event & EV_WRITE) {
112 c553191e 2022-01-11 op 		if ((n = msgbuf_write(&ibuf->w)) == -1 && errno != EAGAIN)
113 c553191e 2022-01-11 op 			err(1, "msgbuf_write");
114 c553191e 2022-01-11 op 		if (n == 0)
115 c553191e 2022-01-11 op 			return -1;
116 c553191e 2022-01-11 op 	}
117 c553191e 2022-01-11 op 
118 c553191e 2022-01-11 op 	for (;;) {
119 c553191e 2022-01-11 op 		if ((n = imsg_get(ibuf, &imsg)) == -1)
120 c553191e 2022-01-11 op 			_exit(1);
121 c553191e 2022-01-11 op 		if (n == 0)
122 c553191e 2022-01-11 op 			break;
123 7018345e 2022-02-11 op 		datalen = IMSG_DATA_SIZE(imsg);
124 c553191e 2022-01-11 op 		i = imsg.hdr.type;
125 c553191e 2022-01-11 op 		if (i > (size / sizeof(imsg_handlerfn*)) || handlers[i] == NULL)
126 c553191e 2022-01-11 op 			abort();
127 c553191e 2022-01-11 op 		handlers[i](&imsg, datalen);
128 c553191e 2022-01-11 op 		imsg_free(&imsg);
129 c553191e 2022-01-11 op 	}
130 c553191e 2022-01-11 op 
131 c553191e 2022-01-11 op 	imsg_event_add(iev);
132 c553191e 2022-01-11 op 	return 0;
133 c553191e 2022-01-11 op }
134 c553191e 2022-01-11 op 
135 c553191e 2022-01-11 op int
136 c553191e 2022-01-11 op imsg_compose_event(struct imsgev *iev, uint16_t type, uint32_t peerid,
137 c553191e 2022-01-11 op     pid_t pid, int fd, const void *data, uint16_t datalen)
138 c553191e 2022-01-11 op {
139 c553191e 2022-01-11 op 	int	ret;
140 c553191e 2022-01-11 op 
141 c553191e 2022-01-11 op 	if ((ret = imsg_compose(&iev->ibuf, type, peerid, pid, fd, data,
142 c553191e 2022-01-11 op 	    datalen) != -1))
143 c553191e 2022-01-11 op 		imsg_event_add(iev);
144 c553191e 2022-01-11 op 
145 c553191e 2022-01-11 op 	return ret;
146 c553191e 2022-01-11 op }
147 c553191e 2022-01-11 op 
148 c553191e 2022-01-11 op void *
149 c553191e 2022-01-11 op hash_alloc(size_t len, void *d)
150 c553191e 2022-01-11 op {
151 c553191e 2022-01-11 op 	if ((d = malloc(len)) == NULL)
152 c553191e 2022-01-11 op 		abort();
153 c553191e 2022-01-11 op 	return d;
154 c553191e 2022-01-11 op }
155 c553191e 2022-01-11 op 
156 c553191e 2022-01-11 op void *
157 c553191e 2022-01-11 op hash_calloc(size_t nmemb, size_t size, void *d)
158 c553191e 2022-01-11 op {
159 c553191e 2022-01-11 op 	if ((d = calloc(nmemb, size)) == NULL)
160 c553191e 2022-01-11 op 		abort();
161 c553191e 2022-01-11 op 	return d;
162 c553191e 2022-01-11 op }
163 c553191e 2022-01-11 op 
164 c553191e 2022-01-11 op void
165 c553191e 2022-01-11 op hash_free(void *ptr, void *d)
166 c553191e 2022-01-11 op {
167 c553191e 2022-01-11 op 	free(ptr);
168 c553191e 2022-01-11 op }