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 7e9b7812 2021-02-07 op #include <errno.h>
20 7e9b7812 2021-02-07 op #include <netdb.h>
21 7e9b7812 2021-02-07 op #include <stdarg.h>
22 7e9b7812 2021-02-07 op #include <stdio.h>
23 7e9b7812 2021-02-07 op #include <string.h>
24 7e9b7812 2021-02-07 op #include <syslog.h>
25 7e9b7812 2021-02-07 op
26 7e9b7812 2021-02-07 op void
27 7e9b7812 2021-02-07 op fatal(const char *fmt, ...)
28 7e9b7812 2021-02-07 op {
29 7e9b7812 2021-02-07 op va_list ap;
30 7e9b7812 2021-02-07 op
31 7e9b7812 2021-02-07 op va_start(ap, fmt);
32 7e9b7812 2021-02-07 op
33 7e9b7812 2021-02-07 op if (conf.foreground) {
34 7e9b7812 2021-02-07 op vfprintf(stderr, fmt, ap);
35 7e9b7812 2021-02-07 op fprintf(stderr, "\n");
36 7e9b7812 2021-02-07 op } else
37 7e9b7812 2021-02-07 op vsyslog(LOG_DAEMON | LOG_CRIT, fmt, ap);
38 7e9b7812 2021-02-07 op
39 7e9b7812 2021-02-07 op va_end(ap);
40 7e9b7812 2021-02-07 op exit(1);
41 7e9b7812 2021-02-07 op }
42 7e9b7812 2021-02-07 op
43 7e9b7812 2021-02-07 op static inline int
44 7e9b7812 2021-02-07 op should_log(int priority)
45 7e9b7812 2021-02-07 op {
46 7e9b7812 2021-02-07 op switch (priority) {
47 7e9b7812 2021-02-07 op case LOG_ERR:
48 7e9b7812 2021-02-07 op return 1;
49 7e9b7812 2021-02-07 op case LOG_WARNING:
50 7e9b7812 2021-02-07 op return 1;
51 7e9b7812 2021-02-07 op case LOG_NOTICE:
52 7e9b7812 2021-02-07 op return conf.verbose >= 1;
53 7e9b7812 2021-02-07 op case LOG_INFO:
54 7e9b7812 2021-02-07 op return conf.verbose >= 2;
55 7e9b7812 2021-02-07 op case LOG_DEBUG:
56 7e9b7812 2021-02-07 op return conf.verbose >= 3;
57 7e9b7812 2021-02-07 op default:
58 7e9b7812 2021-02-07 op return 0;
59 7e9b7812 2021-02-07 op }
60 7e9b7812 2021-02-07 op }
61 7e9b7812 2021-02-07 op
62 7e9b7812 2021-02-07 op static void
63 7e9b7812 2021-02-07 op do_log(int priority, struct client *c,
64 7e9b7812 2021-02-07 op const char *fmt, va_list ap)
65 7e9b7812 2021-02-07 op {
66 7e9b7812 2021-02-07 op char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV];
67 7e9b7812 2021-02-07 op char *fmted, *s;
68 7e9b7812 2021-02-07 op size_t len;
69 7e9b7812 2021-02-07 op int ec;
70 7e9b7812 2021-02-07 op
71 7e9b7812 2021-02-07 op if (!should_log(priority))
72 7e9b7812 2021-02-07 op return;
73 7e9b7812 2021-02-07 op
74 7e9b7812 2021-02-07 op if (c == NULL) {
75 7e9b7812 2021-02-07 op strncpy(hbuf, "<internal>", sizeof(hbuf));
76 7e9b7812 2021-02-07 op sbuf[0] = '\0';
77 7e9b7812 2021-02-07 op } else {
78 7e9b7812 2021-02-07 op len = sizeof(c->addr);
79 7e9b7812 2021-02-07 op ec = getnameinfo((struct sockaddr*)&c->addr, len,
80 7e9b7812 2021-02-07 op hbuf, sizeof(hbuf),
81 7e9b7812 2021-02-07 op sbuf, sizeof(sbuf),
82 7e9b7812 2021-02-07 op NI_NUMERICHOST | NI_NUMERICSERV);
83 7e9b7812 2021-02-07 op if (ec != 0)
84 7e9b7812 2021-02-07 op fatal("getnameinfo: %s", gai_strerror(ec));
85 7e9b7812 2021-02-07 op }
86 7e9b7812 2021-02-07 op
87 7e9b7812 2021-02-07 op if (vasprintf(&fmted, fmt, ap) == -1)
88 7e9b7812 2021-02-07 op fatal("vasprintf: %s", strerror(errno));
89 7e9b7812 2021-02-07 op
90 7e9b7812 2021-02-07 op if (conf.foreground)
91 7e9b7812 2021-02-07 op fprintf(stderr, "%s:%s %s\n", hbuf, sbuf, fmted);
92 7e9b7812 2021-02-07 op else {
93 7e9b7812 2021-02-07 op if (asprintf(&s, "%s:%s %s", hbuf, sbuf, fmted) == -1)
94 7e9b7812 2021-02-07 op fatal("asprintf: %s", strerror(errno));
95 7e9b7812 2021-02-07 op syslog(priority | LOG_DAEMON, "%s", s);
96 7e9b7812 2021-02-07 op free(s);
97 7e9b7812 2021-02-07 op }
98 7e9b7812 2021-02-07 op
99 7e9b7812 2021-02-07 op free(fmted);
100 7e9b7812 2021-02-07 op }
101 7e9b7812 2021-02-07 op
102 7e9b7812 2021-02-07 op void
103 7e9b7812 2021-02-07 op log_err(struct client *c, const char *fmt, ...)
104 7e9b7812 2021-02-07 op {
105 7e9b7812 2021-02-07 op va_list ap;
106 7e9b7812 2021-02-07 op
107 7e9b7812 2021-02-07 op va_start(ap, fmt);
108 7e9b7812 2021-02-07 op do_log(LOG_ERR, c, fmt, ap);
109 7e9b7812 2021-02-07 op va_end(ap);
110 7e9b7812 2021-02-07 op }
111 7e9b7812 2021-02-07 op
112 7e9b7812 2021-02-07 op void
113 7e9b7812 2021-02-07 op log_warn(struct client *c, const char *fmt, ...)
114 7e9b7812 2021-02-07 op {
115 7e9b7812 2021-02-07 op va_list ap;
116 7e9b7812 2021-02-07 op
117 7e9b7812 2021-02-07 op va_start(ap, fmt);
118 7e9b7812 2021-02-07 op do_log(LOG_WARNING, c, fmt, ap);
119 7e9b7812 2021-02-07 op va_end(ap);
120 7e9b7812 2021-02-07 op }
121 7e9b7812 2021-02-07 op
122 7e9b7812 2021-02-07 op void
123 7e9b7812 2021-02-07 op log_notice(struct client *c, const char *fmt, ...)
124 7e9b7812 2021-02-07 op {
125 7e9b7812 2021-02-07 op va_list ap;
126 7e9b7812 2021-02-07 op
127 7e9b7812 2021-02-07 op va_start(ap, fmt);
128 7e9b7812 2021-02-07 op do_log(LOG_NOTICE, c, fmt, ap);
129 7e9b7812 2021-02-07 op va_end(ap);
130 7e9b7812 2021-02-07 op }
131 7e9b7812 2021-02-07 op
132 7e9b7812 2021-02-07 op void
133 7e9b7812 2021-02-07 op log_info(struct client *c, const char *fmt, ...)
134 7e9b7812 2021-02-07 op {
135 7e9b7812 2021-02-07 op va_list ap;
136 7e9b7812 2021-02-07 op
137 7e9b7812 2021-02-07 op va_start(ap, fmt);
138 7e9b7812 2021-02-07 op do_log(LOG_INFO, c, fmt, ap);
139 7e9b7812 2021-02-07 op va_end(ap);
140 7e9b7812 2021-02-07 op }
141 7e9b7812 2021-02-07 op
142 7e9b7812 2021-02-07 op void
143 7e9b7812 2021-02-07 op log_debug(struct client *c, const char *fmt, ...)
144 7e9b7812 2021-02-07 op {
145 7e9b7812 2021-02-07 op va_list ap;
146 7e9b7812 2021-02-07 op
147 7e9b7812 2021-02-07 op va_start(ap, fmt);
148 7e9b7812 2021-02-07 op do_log(LOG_DEBUG, c, fmt, ap);
149 7e9b7812 2021-02-07 op va_end(ap);
150 7e9b7812 2021-02-07 op }
151 7e9b7812 2021-02-07 op
152 7e9b7812 2021-02-07 op /* strchr, but with a bound */
153 7e9b7812 2021-02-07 op static char *
154 7e9b7812 2021-02-07 op gmid_strnchr(char *s, int c, size_t len)
155 7e9b7812 2021-02-07 op {
156 7e9b7812 2021-02-07 op size_t i;
157 7e9b7812 2021-02-07 op
158 7e9b7812 2021-02-07 op for (i = 0; i < len; ++i)
159 7e9b7812 2021-02-07 op if (s[i] == c)
160 7e9b7812 2021-02-07 op return &s[i];
161 7e9b7812 2021-02-07 op return NULL;
162 7e9b7812 2021-02-07 op }
163 7e9b7812 2021-02-07 op
164 7e9b7812 2021-02-07 op void
165 7e9b7812 2021-02-07 op log_request(struct client *c, char *meta, size_t l)
166 7e9b7812 2021-02-07 op {
167 7e9b7812 2021-02-07 op char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV], b[GEMINI_URL_LEN];
168 7e9b7812 2021-02-07 op char *t;
169 7e9b7812 2021-02-07 op size_t len;
170 7e9b7812 2021-02-07 op int ec;
171 7e9b7812 2021-02-07 op
172 7e9b7812 2021-02-07 op len = sizeof(c->addr);
173 7e9b7812 2021-02-07 op ec = getnameinfo((struct sockaddr*)&c->addr, len,
174 7e9b7812 2021-02-07 op hbuf, sizeof(hbuf),
175 7e9b7812 2021-02-07 op sbuf, sizeof(sbuf),
176 7e9b7812 2021-02-07 op NI_NUMERICHOST | NI_NUMERICSERV);
177 7e9b7812 2021-02-07 op if (ec != 0)
178 7e9b7812 2021-02-07 op fatal("getnameinfo: %s", gai_strerror(ec));
179 7e9b7812 2021-02-07 op
180 7e9b7812 2021-02-07 op if (c->iri.schema != NULL) {
181 7e9b7812 2021-02-07 op /* serialize the IRI */
182 7e9b7812 2021-02-07 op strlcpy(b, c->iri.schema, sizeof(b));
183 7e9b7812 2021-02-07 op strlcat(b, "://", sizeof(b));
184 7e9b7812 2021-02-07 op
185 7e9b7812 2021-02-07 op /* log the decoded host name, but if it was invalid
186 7e9b7812 2021-02-07 op * use the raw one. */
187 7e9b7812 2021-02-07 op if (*c->domain != '\0')
188 7e9b7812 2021-02-07 op strlcat(b, c->domain, sizeof(b));
189 7e9b7812 2021-02-07 op else
190 7e9b7812 2021-02-07 op strlcat(b, c->iri.host, sizeof(b));
191 7e9b7812 2021-02-07 op
192 7e9b7812 2021-02-07 op strlcat(b, "/", sizeof(b));
193 7e9b7812 2021-02-07 op strlcat(b, c->iri.path, sizeof(b)); /* TODO: sanitize UTF8 */
194 7e9b7812 2021-02-07 op if (*c->iri.query != '\0') { /* TODO: sanitize UTF8 */
195 7e9b7812 2021-02-07 op strlcat(b, "?", sizeof(b));
196 7e9b7812 2021-02-07 op strlcat(b, c->iri.query, sizeof(b));
197 7e9b7812 2021-02-07 op }
198 7e9b7812 2021-02-07 op } else {
199 7e9b7812 2021-02-07 op strlcpy(b, c->req, sizeof(b));
200 7e9b7812 2021-02-07 op }
201 7e9b7812 2021-02-07 op
202 7e9b7812 2021-02-07 op if ((t = gmid_strnchr(meta, '\r', l)) == NULL)
203 7e9b7812 2021-02-07 op t = meta + len;
204 7e9b7812 2021-02-07 op
205 7e9b7812 2021-02-07 op if (conf.foreground)
206 7e9b7812 2021-02-07 op fprintf(stderr, "%s:%s GET %s %.*s\n", hbuf, sbuf, b,
207 7e9b7812 2021-02-07 op (int)(t - meta), meta);
208 7e9b7812 2021-02-07 op else
209 7e9b7812 2021-02-07 op syslog(LOG_INFO | LOG_DAEMON, "%s:%s GET %s %.*s",
210 7e9b7812 2021-02-07 op hbuf, sbuf, b, (int)(t - meta), meta);
211 7e9b7812 2021-02-07 op }