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