Blame


1 7e9b7812 2021-02-07 op /*
2 7e9b7812 2021-02-07 op * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3 7e9b7812 2021-02-07 op *
4 7e9b7812 2021-02-07 op * Permission to use, copy, modify, and distribute this software for any
5 7e9b7812 2021-02-07 op * purpose with or without fee is hereby granted, provided that the above
6 7e9b7812 2021-02-07 op * copyright notice and this permission notice appear in all copies.
7 7e9b7812 2021-02-07 op *
8 7e9b7812 2021-02-07 op * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 7e9b7812 2021-02-07 op * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 7e9b7812 2021-02-07 op * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 7e9b7812 2021-02-07 op * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 7e9b7812 2021-02-07 op * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 7e9b7812 2021-02-07 op * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 7e9b7812 2021-02-07 op * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 7e9b7812 2021-02-07 op */
16 7e9b7812 2021-02-07 op
17 52418c8d 2021-02-12 op #include "gmid.h"
18 52418c8d 2021-02-12 op
19 d278a0c3 2021-02-23 op #include <sys/types.h>
20 d278a0c3 2021-02-23 op #include <sys/uio.h>
21 d278a0c3 2021-02-23 op
22 7e9b7812 2021-02-07 op #include <errno.h>
23 d278a0c3 2021-02-23 op #include <event.h>
24 d278a0c3 2021-02-23 op #include <imsg.h>
25 7e9b7812 2021-02-07 op #include <netdb.h>
26 2ef7f631 2021-04-28 op #include <poll.h>
27 7e9b7812 2021-02-07 op #include <stdarg.h>
28 7e9b7812 2021-02-07 op #include <stdio.h>
29 7e9b7812 2021-02-07 op #include <string.h>
30 7e9b7812 2021-02-07 op #include <syslog.h>
31 b3342582 2021-04-14 op #include <time.h>
32 7e9b7812 2021-02-07 op
33 bc99d868 2021-03-19 op static struct event imsgev;
34 d278a0c3 2021-02-23 op
35 e952c505 2021-06-15 op static FILE *log;
36 e952c505 2021-06-15 op
37 bc99d868 2021-03-19 op static void handle_imsg_quit(struct imsgbuf*, struct imsg*, size_t);
38 bc99d868 2021-03-19 op static void handle_imsg_log(struct imsgbuf*, struct imsg*, size_t);
39 e952c505 2021-06-15 op static void handle_imsg_log_type(struct imsgbuf*, struct imsg*, size_t);
40 bc99d868 2021-03-19 op static void handle_dispatch_imsg(int, short, void*);
41 d278a0c3 2021-02-23 op
42 bc99d868 2021-03-19 op static imsg_handlerfn *handlers[] = {
43 bc99d868 2021-03-19 op [IMSG_QUIT] = handle_imsg_quit,
44 bc99d868 2021-03-19 op [IMSG_LOG] = handle_imsg_log,
45 41395640 2021-07-19 op [IMSG_LOG_REQUEST] = handle_imsg_log,
46 e952c505 2021-06-15 op [IMSG_LOG_TYPE] = handle_imsg_log_type,
47 bc99d868 2021-03-19 op };
48 b3342582 2021-04-14 op
49 b3342582 2021-04-14 op static inline void
50 e952c505 2021-06-15 op print_date(FILE *f)
51 b3342582 2021-04-14 op {
52 b3342582 2021-04-14 op struct tm tminfo;
53 b3342582 2021-04-14 op time_t t;
54 b3342582 2021-04-14 op char buf[20];
55 b3342582 2021-04-14 op
56 b3342582 2021-04-14 op time(&t);
57 b3342582 2021-04-14 op strftime(buf, sizeof(buf), "%F %T",
58 b3342582 2021-04-14 op localtime_r(&t, &tminfo));
59 e952c505 2021-06-15 op fprintf(f, "[%s] ", buf);
60 7e9b7812 2021-02-07 op }
61 7e9b7812 2021-02-07 op
62 7e9b7812 2021-02-07 op static inline int
63 7e9b7812 2021-02-07 op should_log(int priority)
64 7e9b7812 2021-02-07 op {
65 7e9b7812 2021-02-07 op switch (priority) {
66 7e9b7812 2021-02-07 op case LOG_ERR:
67 7e9b7812 2021-02-07 op return 1;
68 7e9b7812 2021-02-07 op case LOG_WARNING:
69 7e9b7812 2021-02-07 op return 1;
70 7e9b7812 2021-02-07 op case LOG_NOTICE:
71 7e9b7812 2021-02-07 op return conf.verbose >= 1;
72 7e9b7812 2021-02-07 op case LOG_INFO:
73 7e9b7812 2021-02-07 op return conf.verbose >= 2;
74 7e9b7812 2021-02-07 op case LOG_DEBUG:
75 7e9b7812 2021-02-07 op return conf.verbose >= 3;
76 7e9b7812 2021-02-07 op default:
77 7e9b7812 2021-02-07 op return 0;
78 7e9b7812 2021-02-07 op }
79 7e9b7812 2021-02-07 op }
80 7e9b7812 2021-02-07 op
81 d278a0c3 2021-02-23 op static inline void
82 2b5e2a5f 2021-07-19 op send_log(int type, int priority, const char *msg, size_t len)
83 d278a0c3 2021-02-23 op {
84 2b5e2a5f 2021-07-19 op imsg_compose(&logibuf, type, priority, 0, -1, msg, len);
85 bc99d868 2021-03-19 op imsg_flush(&logibuf);
86 d278a0c3 2021-02-23 op }
87 d278a0c3 2021-02-23 op
88 d89a9060 2021-04-28 op void
89 d89a9060 2021-04-28 op fatal(const char *fmt, ...)
90 d89a9060 2021-04-28 op {
91 2ef7f631 2021-04-28 op struct pollfd pfd;
92 d89a9060 2021-04-28 op va_list ap;
93 d89a9060 2021-04-28 op int r;
94 d89a9060 2021-04-28 op char *fmted;
95 d89a9060 2021-04-28 op
96 d89a9060 2021-04-28 op va_start(ap, fmt);
97 d89a9060 2021-04-28 op if ((r = vasprintf(&fmted, fmt, ap)) != -1) {
98 ef352ec3 2021-08-23 op send_log(IMSG_LOG, LOG_CRIT, fmted, r+1);
99 d89a9060 2021-04-28 op free(fmted);
100 2ef7f631 2021-04-28 op
101 2ef7f631 2021-04-28 op /* wait for the logger process to shut down */
102 2ef7f631 2021-04-28 op pfd.fd = logibuf.fd;
103 2ef7f631 2021-04-28 op pfd.events = POLLIN;
104 2ef7f631 2021-04-28 op poll(&pfd, 1, 1000);
105 d89a9060 2021-04-28 op }
106 d89a9060 2021-04-28 op va_end(ap);
107 d89a9060 2021-04-28 op exit(1);
108 d89a9060 2021-04-28 op }
109 d89a9060 2021-04-28 op
110 d278a0c3 2021-02-23 op static inline void
111 d278a0c3 2021-02-23 op vlog(int priority, struct client *c,
112 7e9b7812 2021-02-07 op const char *fmt, va_list ap)
113 7e9b7812 2021-02-07 op {
114 7e9b7812 2021-02-07 op char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV];
115 7e9b7812 2021-02-07 op char *fmted, *s;
116 7e9b7812 2021-02-07 op size_t len;
117 7e9b7812 2021-02-07 op int ec;
118 7e9b7812 2021-02-07 op
119 7e9b7812 2021-02-07 op if (!should_log(priority))
120 7e9b7812 2021-02-07 op return;
121 7e9b7812 2021-02-07 op
122 d278a0c3 2021-02-23 op if (c != NULL) {
123 7e9b7812 2021-02-07 op len = sizeof(c->addr);
124 7e9b7812 2021-02-07 op ec = getnameinfo((struct sockaddr*)&c->addr, len,
125 7e9b7812 2021-02-07 op hbuf, sizeof(hbuf),
126 7e9b7812 2021-02-07 op sbuf, sizeof(sbuf),
127 7e9b7812 2021-02-07 op NI_NUMERICHOST | NI_NUMERICSERV);
128 7e9b7812 2021-02-07 op if (ec != 0)
129 d0071d83 2021-10-02 op fatal("getnameinfo: %s: %s",
130 d0071d83 2021-10-02 op gai_strerror(ec), strerror(errno));
131 7e9b7812 2021-02-07 op }
132 7e9b7812 2021-02-07 op
133 7e9b7812 2021-02-07 op if (vasprintf(&fmted, fmt, ap) == -1)
134 7e9b7812 2021-02-07 op fatal("vasprintf: %s", strerror(errno));
135 7e9b7812 2021-02-07 op
136 7700ab07 2021-07-06 op if (c == NULL)
137 d278a0c3 2021-02-23 op ec = asprintf(&s, "internal: %s", fmted);
138 d278a0c3 2021-02-23 op else
139 d278a0c3 2021-02-23 op ec = asprintf(&s, "%s:%s %s", hbuf, sbuf, fmted);
140 7e9b7812 2021-02-07 op
141 d278a0c3 2021-02-23 op if (ec < 0)
142 d278a0c3 2021-02-23 op fatal("asprintf: %s", strerror(errno));
143 d278a0c3 2021-02-23 op
144 ef352ec3 2021-08-23 op send_log(IMSG_LOG, priority, s, ec+1);
145 d278a0c3 2021-02-23 op
146 7e9b7812 2021-02-07 op free(fmted);
147 d278a0c3 2021-02-23 op free(s);
148 7e9b7812 2021-02-07 op }
149 7e9b7812 2021-02-07 op
150 7e9b7812 2021-02-07 op void
151 7e9b7812 2021-02-07 op log_err(struct client *c, const char *fmt, ...)
152 7e9b7812 2021-02-07 op {
153 7e9b7812 2021-02-07 op va_list ap;
154 7e9b7812 2021-02-07 op
155 7e9b7812 2021-02-07 op va_start(ap, fmt);
156 d278a0c3 2021-02-23 op vlog(LOG_ERR, c, fmt, ap);
157 7e9b7812 2021-02-07 op va_end(ap);
158 7e9b7812 2021-02-07 op }
159 7e9b7812 2021-02-07 op
160 7e9b7812 2021-02-07 op void
161 7e9b7812 2021-02-07 op log_warn(struct client *c, const char *fmt, ...)
162 7e9b7812 2021-02-07 op {
163 7e9b7812 2021-02-07 op va_list ap;
164 7e9b7812 2021-02-07 op
165 7e9b7812 2021-02-07 op va_start(ap, fmt);
166 d278a0c3 2021-02-23 op vlog(LOG_WARNING, c, fmt, ap);
167 7e9b7812 2021-02-07 op va_end(ap);
168 7e9b7812 2021-02-07 op }
169 7e9b7812 2021-02-07 op
170 7e9b7812 2021-02-07 op void
171 7e9b7812 2021-02-07 op log_notice(struct client *c, const char *fmt, ...)
172 7e9b7812 2021-02-07 op {
173 7e9b7812 2021-02-07 op va_list ap;
174 7e9b7812 2021-02-07 op
175 7e9b7812 2021-02-07 op va_start(ap, fmt);
176 d278a0c3 2021-02-23 op vlog(LOG_NOTICE, c, fmt, ap);
177 7e9b7812 2021-02-07 op va_end(ap);
178 7e9b7812 2021-02-07 op }
179 7e9b7812 2021-02-07 op
180 7e9b7812 2021-02-07 op void
181 7e9b7812 2021-02-07 op log_info(struct client *c, const char *fmt, ...)
182 7e9b7812 2021-02-07 op {
183 7e9b7812 2021-02-07 op va_list ap;
184 7e9b7812 2021-02-07 op
185 7e9b7812 2021-02-07 op va_start(ap, fmt);
186 d278a0c3 2021-02-23 op vlog(LOG_INFO, c, fmt, ap);
187 7e9b7812 2021-02-07 op va_end(ap);
188 7e9b7812 2021-02-07 op }
189 7e9b7812 2021-02-07 op
190 7e9b7812 2021-02-07 op void
191 7e9b7812 2021-02-07 op log_debug(struct client *c, const char *fmt, ...)
192 7e9b7812 2021-02-07 op {
193 7e9b7812 2021-02-07 op va_list ap;
194 7e9b7812 2021-02-07 op
195 7e9b7812 2021-02-07 op va_start(ap, fmt);
196 d278a0c3 2021-02-23 op vlog(LOG_DEBUG, c, fmt, ap);
197 7e9b7812 2021-02-07 op va_end(ap);
198 7e9b7812 2021-02-07 op }
199 7e9b7812 2021-02-07 op
200 7e9b7812 2021-02-07 op /* strchr, but with a bound */
201 7e9b7812 2021-02-07 op static char *
202 7e9b7812 2021-02-07 op gmid_strnchr(char *s, int c, size_t len)
203 7e9b7812 2021-02-07 op {
204 7e9b7812 2021-02-07 op size_t i;
205 7e9b7812 2021-02-07 op
206 7e9b7812 2021-02-07 op for (i = 0; i < len; ++i)
207 7e9b7812 2021-02-07 op if (s[i] == c)
208 7e9b7812 2021-02-07 op return &s[i];
209 7e9b7812 2021-02-07 op return NULL;
210 7e9b7812 2021-02-07 op }
211 7e9b7812 2021-02-07 op
212 7e9b7812 2021-02-07 op void
213 7e9b7812 2021-02-07 op log_request(struct client *c, char *meta, size_t l)
214 7e9b7812 2021-02-07 op {
215 7e9b7812 2021-02-07 op char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV], b[GEMINI_URL_LEN];
216 387b976b 2021-10-15 op char *fmted;
217 387b976b 2021-10-15 op const char *t;
218 7e9b7812 2021-02-07 op size_t len;
219 7e9b7812 2021-02-07 op int ec;
220 7e9b7812 2021-02-07 op
221 7e9b7812 2021-02-07 op len = sizeof(c->addr);
222 7e9b7812 2021-02-07 op ec = getnameinfo((struct sockaddr*)&c->addr, len,
223 7e9b7812 2021-02-07 op hbuf, sizeof(hbuf),
224 7e9b7812 2021-02-07 op sbuf, sizeof(sbuf),
225 7e9b7812 2021-02-07 op NI_NUMERICHOST | NI_NUMERICSERV);
226 7e9b7812 2021-02-07 op if (ec != 0)
227 7e9b7812 2021-02-07 op fatal("getnameinfo: %s", gai_strerror(ec));
228 7e9b7812 2021-02-07 op
229 7e9b7812 2021-02-07 op if (c->iri.schema != NULL) {
230 7e9b7812 2021-02-07 op /* serialize the IRI */
231 7e9b7812 2021-02-07 op strlcpy(b, c->iri.schema, sizeof(b));
232 7e9b7812 2021-02-07 op strlcat(b, "://", sizeof(b));
233 7e9b7812 2021-02-07 op
234 7e9b7812 2021-02-07 op /* log the decoded host name, but if it was invalid
235 7e9b7812 2021-02-07 op * use the raw one. */
236 7e9b7812 2021-02-07 op if (*c->domain != '\0')
237 7e9b7812 2021-02-07 op strlcat(b, c->domain, sizeof(b));
238 7e9b7812 2021-02-07 op else
239 7e9b7812 2021-02-07 op strlcat(b, c->iri.host, sizeof(b));
240 7e9b7812 2021-02-07 op
241 a81dd6cd 2021-10-24 op if (*c->iri.path != '/')
242 a81dd6cd 2021-10-24 op strlcat(b, "/", sizeof(b));
243 7e9b7812 2021-02-07 op strlcat(b, c->iri.path, sizeof(b)); /* TODO: sanitize UTF8 */
244 7e9b7812 2021-02-07 op if (*c->iri.query != '\0') { /* TODO: sanitize UTF8 */
245 7e9b7812 2021-02-07 op strlcat(b, "?", sizeof(b));
246 7e9b7812 2021-02-07 op strlcat(b, c->iri.query, sizeof(b));
247 7e9b7812 2021-02-07 op }
248 7e9b7812 2021-02-07 op } else {
249 387b976b 2021-10-15 op if ((t = c->req) == NULL)
250 387b976b 2021-10-15 op t = "";
251 387b976b 2021-10-15 op strlcpy(b, t, sizeof(b));
252 7e9b7812 2021-02-07 op }
253 7e9b7812 2021-02-07 op
254 7e9b7812 2021-02-07 op if ((t = gmid_strnchr(meta, '\r', l)) == NULL)
255 7e9b7812 2021-02-07 op t = meta + len;
256 7e9b7812 2021-02-07 op
257 d278a0c3 2021-02-23 op ec = asprintf(&fmted, "%s:%s GET %s %.*s", hbuf, sbuf, b,
258 d278a0c3 2021-02-23 op (int)(t-meta), meta);
259 d278a0c3 2021-02-23 op if (ec < 0)
260 d278a0c3 2021-02-23 op err(1, "asprintf");
261 ef352ec3 2021-08-23 op send_log(IMSG_LOG_REQUEST, LOG_NOTICE, fmted, ec+1);
262 d278a0c3 2021-02-23 op free(fmted);
263 d278a0c3 2021-02-23 op }
264 d278a0c3 2021-02-23 op
265 d278a0c3 2021-02-23 op
266 d278a0c3 2021-02-23 op
267 d278a0c3 2021-02-23 op static void
268 41395640 2021-07-19 op do_log(int type, int priority, const char *msg)
269 e952c505 2021-06-15 op {
270 e952c505 2021-06-15 op int quit = 0;
271 e952c505 2021-06-15 op
272 e952c505 2021-06-15 op if (priority == LOG_CRIT) {
273 e952c505 2021-06-15 op quit = 1;
274 e952c505 2021-06-15 op priority = LOG_ERR;
275 e952c505 2021-06-15 op }
276 e952c505 2021-06-15 op
277 e952c505 2021-06-15 op if (log != NULL) {
278 41395640 2021-07-19 op if (type != IMSG_LOG_REQUEST)
279 41395640 2021-07-19 op print_date(log);
280 e952c505 2021-06-15 op fprintf(log, "%s\n", msg);
281 e952c505 2021-06-15 op } else
282 e952c505 2021-06-15 op syslog(LOG_DAEMON | priority, "%s", msg);
283 e952c505 2021-06-15 op
284 e952c505 2021-06-15 op if (quit)
285 e952c505 2021-06-15 op exit(1);
286 e952c505 2021-06-15 op }
287 e952c505 2021-06-15 op
288 e952c505 2021-06-15 op static void
289 bc99d868 2021-03-19 op handle_imsg_quit(struct imsgbuf *ibuf, struct imsg *imsg, size_t datalen)
290 d278a0c3 2021-02-23 op {
291 bc99d868 2021-03-19 op event_loopbreak();
292 bc99d868 2021-03-19 op }
293 d278a0c3 2021-02-23 op
294 bc99d868 2021-03-19 op static void
295 bc99d868 2021-03-19 op handle_imsg_log(struct imsgbuf *ibuf, struct imsg *imsg, size_t datalen)
296 bc99d868 2021-03-19 op {
297 984c46a8 2021-06-16 op int priority;
298 2ef7f631 2021-04-28 op char *msg;
299 d278a0c3 2021-02-23 op
300 bc99d868 2021-03-19 op msg = imsg->data;
301 bc99d868 2021-03-19 op msg[datalen-1] = '\0';
302 2ef7f631 2021-04-28 op priority = imsg->hdr.peerid;
303 41395640 2021-07-19 op do_log(imsg->hdr.type, priority, msg);
304 e952c505 2021-06-15 op }
305 2ef7f631 2021-04-28 op
306 e952c505 2021-06-15 op static void
307 e952c505 2021-06-15 op handle_imsg_log_type(struct imsgbuf *ibuf, struct imsg *imsg, size_t datalen)
308 e952c505 2021-06-15 op {
309 8a50fc03 2021-07-07 op if (log != NULL && log != stderr) {
310 e952c505 2021-06-15 op fflush(log);
311 e952c505 2021-06-15 op fclose(log);
312 2ef7f631 2021-04-28 op }
313 8a50fc03 2021-07-07 op log = NULL;
314 2ef7f631 2021-04-28 op
315 e952c505 2021-06-15 op if (imsg->fd != -1) {
316 e952c505 2021-06-15 op if ((log = fdopen(imsg->fd, "a")) == NULL) {
317 e952c505 2021-06-15 op syslog(LOG_DAEMON | LOG_ERR, "fdopen: %s",
318 e952c505 2021-06-15 op strerror(errno));
319 e952c505 2021-06-15 op exit(1);
320 e952c505 2021-06-15 op }
321 e952c505 2021-06-15 op }
322 bc99d868 2021-03-19 op }
323 d278a0c3 2021-02-23 op
324 bc99d868 2021-03-19 op static void
325 bc99d868 2021-03-19 op handle_dispatch_imsg(int fd, short ev, void *d)
326 bc99d868 2021-03-19 op {
327 bc99d868 2021-03-19 op struct imsgbuf *ibuf = d;
328 bc99d868 2021-03-19 op dispatch_imsg(ibuf, handlers, sizeof(handlers));
329 7e9b7812 2021-02-07 op }
330 d278a0c3 2021-02-23 op
331 376a5407 2021-02-23 op int
332 d278a0c3 2021-02-23 op logger_main(int fd, struct imsgbuf *ibuf)
333 d278a0c3 2021-02-23 op {
334 8a50fc03 2021-07-07 op log = stderr;
335 db623d75 2021-09-17 op
336 db623d75 2021-09-17 op openlog(getprogname(), LOG_NDELAY, LOG_DAEMON);
337 bd2eb095 2022-07-07 op tzset();
338 8a50fc03 2021-07-07 op
339 d278a0c3 2021-02-23 op event_init();
340 d278a0c3 2021-02-23 op
341 bc99d868 2021-03-19 op event_set(&imsgev, fd, EV_READ | EV_PERSIST, &handle_dispatch_imsg, ibuf);
342 bc99d868 2021-03-19 op event_add(&imsgev, NULL);
343 d278a0c3 2021-02-23 op
344 62e001b0 2021-03-20 op sandbox_logger_process();
345 d278a0c3 2021-02-23 op
346 d278a0c3 2021-02-23 op event_dispatch();
347 d278a0c3 2021-02-23 op
348 db623d75 2021-09-17 op closelog();
349 db623d75 2021-09-17 op
350 d278a0c3 2021-02-23 op return 0;
351 d278a0c3 2021-02-23 op }