Blame


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 ee0aac2f 2022-05-25 op mark_nonblock_cloexec(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 ee0aac2f 2022-05-25 op if (fcntl(fd, F_SETFD, FD_CLOEXEC) == -1)
39 ee0aac2f 2022-05-25 op return 0;
40 c553191e 2022-01-11 op return 1;
41 c553191e 2022-01-11 op }
42 c553191e 2022-01-11 op
43 c553191e 2022-01-11 op int
44 c553191e 2022-01-11 op has_suffix(const char *str, const char *sufx)
45 c553191e 2022-01-11 op {
46 c553191e 2022-01-11 op size_t l, s;
47 c553191e 2022-01-11 op
48 c553191e 2022-01-11 op l = strlen(str);
49 c553191e 2022-01-11 op s = strlen(sufx);
50 c553191e 2022-01-11 op
51 c553191e 2022-01-11 op if (l < s)
52 c553191e 2022-01-11 op return 0;
53 c553191e 2022-01-11 op
54 c553191e 2022-01-11 op return !strcmp(str + (l - s), sufx);
55 c553191e 2022-01-11 op }
56 c553191e 2022-01-11 op
57 c553191e 2022-01-11 op int
58 c553191e 2022-01-11 op unicode_isspace(uint32_t cp)
59 c553191e 2022-01-11 op {
60 c553191e 2022-01-11 op if (cp < INT8_MAX)
61 c553191e 2022-01-11 op return isspace(cp);
62 c553191e 2022-01-11 op return 0;
63 c553191e 2022-01-11 op }
64 c553191e 2022-01-11 op
65 c553191e 2022-01-11 op int
66 c553191e 2022-01-11 op unicode_isgraph(uint32_t cp)
67 c553191e 2022-01-11 op {
68 c553191e 2022-01-11 op if (cp < INT8_MAX)
69 c553191e 2022-01-11 op return isgraph(cp);
70 c553191e 2022-01-11 op return 1;
71 c553191e 2022-01-11 op }
72 c553191e 2022-01-11 op
73 0f559c55 2022-02-11 op void
74 c553191e 2022-01-11 op imsg_event_add(struct imsgev *iev)
75 c553191e 2022-01-11 op {
76 c553191e 2022-01-11 op iev->events = EV_READ;
77 c553191e 2022-01-11 op if (iev->ibuf.w.queued)
78 c553191e 2022-01-11 op iev->events |= EV_WRITE;
79 c553191e 2022-01-11 op
80 c553191e 2022-01-11 op event_del(&iev->ev);
81 c553191e 2022-01-11 op event_set(&iev->ev, iev->ibuf.fd, iev->events, iev->handler, iev);
82 c553191e 2022-01-11 op event_add(&iev->ev, NULL);
83 c553191e 2022-01-11 op }
84 c553191e 2022-01-11 op
85 c553191e 2022-01-11 op int
86 c553191e 2022-01-11 op dispatch_imsg(struct imsgev *iev, short event, imsg_handlerfn **handlers,
87 c553191e 2022-01-11 op size_t size)
88 c553191e 2022-01-11 op {
89 c553191e 2022-01-11 op struct imsgbuf *ibuf;
90 c553191e 2022-01-11 op struct imsg imsg;
91 c553191e 2022-01-11 op size_t datalen, i;
92 c553191e 2022-01-11 op ssize_t n;
93 c553191e 2022-01-11 op
94 c553191e 2022-01-11 op ibuf = &iev->ibuf;
95 c553191e 2022-01-11 op
96 c553191e 2022-01-11 op if (event & EV_READ) {
97 c553191e 2022-01-11 op if ((n = imsg_read(ibuf)) == -1 && errno != EAGAIN)
98 c553191e 2022-01-11 op err(1, "imsg_read error");
99 c553191e 2022-01-11 op if (n == 0)
100 c553191e 2022-01-11 op return -1;
101 c553191e 2022-01-11 op }
102 c553191e 2022-01-11 op if (event & EV_WRITE) {
103 c553191e 2022-01-11 op if ((n = msgbuf_write(&ibuf->w)) == -1 && errno != EAGAIN)
104 c553191e 2022-01-11 op err(1, "msgbuf_write");
105 c553191e 2022-01-11 op if (n == 0)
106 c553191e 2022-01-11 op return -1;
107 c553191e 2022-01-11 op }
108 c553191e 2022-01-11 op
109 c553191e 2022-01-11 op for (;;) {
110 c553191e 2022-01-11 op if ((n = imsg_get(ibuf, &imsg)) == -1)
111 c553191e 2022-01-11 op _exit(1);
112 c553191e 2022-01-11 op if (n == 0)
113 c553191e 2022-01-11 op break;
114 7018345e 2022-02-11 op datalen = IMSG_DATA_SIZE(imsg);
115 c553191e 2022-01-11 op i = imsg.hdr.type;
116 c553191e 2022-01-11 op if (i > (size / sizeof(imsg_handlerfn*)) || handlers[i] == NULL)
117 c553191e 2022-01-11 op abort();
118 c553191e 2022-01-11 op handlers[i](&imsg, datalen);
119 c553191e 2022-01-11 op imsg_free(&imsg);
120 c553191e 2022-01-11 op }
121 c553191e 2022-01-11 op
122 c553191e 2022-01-11 op imsg_event_add(iev);
123 c553191e 2022-01-11 op return 0;
124 c553191e 2022-01-11 op }
125 c553191e 2022-01-11 op
126 c553191e 2022-01-11 op int
127 c553191e 2022-01-11 op imsg_compose_event(struct imsgev *iev, uint16_t type, uint32_t peerid,
128 c553191e 2022-01-11 op pid_t pid, int fd, const void *data, uint16_t datalen)
129 c553191e 2022-01-11 op {
130 c553191e 2022-01-11 op int ret;
131 c553191e 2022-01-11 op
132 c553191e 2022-01-11 op if ((ret = imsg_compose(&iev->ibuf, type, peerid, pid, fd, data,
133 c553191e 2022-01-11 op datalen) != -1))
134 c553191e 2022-01-11 op imsg_event_add(iev);
135 c553191e 2022-01-11 op
136 c553191e 2022-01-11 op return ret;
137 c553191e 2022-01-11 op }
138 c553191e 2022-01-11 op
139 c553191e 2022-01-11 op void *
140 c553191e 2022-01-11 op hash_alloc(size_t len, void *d)
141 c553191e 2022-01-11 op {
142 c553191e 2022-01-11 op if ((d = malloc(len)) == NULL)
143 c553191e 2022-01-11 op abort();
144 c553191e 2022-01-11 op return d;
145 c553191e 2022-01-11 op }
146 c553191e 2022-01-11 op
147 c553191e 2022-01-11 op void *
148 c553191e 2022-01-11 op hash_calloc(size_t nmemb, size_t size, void *d)
149 c553191e 2022-01-11 op {
150 c553191e 2022-01-11 op if ((d = calloc(nmemb, size)) == NULL)
151 c553191e 2022-01-11 op abort();
152 c553191e 2022-01-11 op return d;
153 c553191e 2022-01-11 op }
154 c553191e 2022-01-11 op
155 c553191e 2022-01-11 op void
156 c553191e 2022-01-11 op hash_free(void *ptr, void *d)
157 c553191e 2022-01-11 op {
158 c553191e 2022-01-11 op free(ptr);
159 c553191e 2022-01-11 op }