1 e9e8d207 2020-07-25 martijn /*
  2 e9e8d207 2020-07-25 martijn  * Copyright (c) 2019 Martijn van Duren <martijn@openbsd.org>
  3 e9e8d207 2020-07-25 martijn  *
  4 e9e8d207 2020-07-25 martijn  * Permission to use, copy, modify, and distribute this software for any
  5 e9e8d207 2020-07-25 martijn  * purpose with or without fee is hereby granted, provided that the above
  6 e9e8d207 2020-07-25 martijn  * copyright notice and this permission notice appear in all copies.
  7 e9e8d207 2020-07-25 martijn  *
  8 e9e8d207 2020-07-25 martijn  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 e9e8d207 2020-07-25 martijn  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 e9e8d207 2020-07-25 martijn  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 e9e8d207 2020-07-25 martijn  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 e9e8d207 2020-07-25 martijn  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 e9e8d207 2020-07-25 martijn  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 e9e8d207 2020-07-25 martijn  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 e9e8d207 2020-07-25 martijn  */
 16 e9e8d207 2020-07-25 martijn 
 17 e9e8d207 2020-07-25 martijn #include <errno.h>
 18 e9e8d207 2020-07-25 martijn #include <inttypes.h>
 19 e9e8d207 2020-07-25 martijn #include <stdio.h>
 20 e9e8d207 2020-07-25 martijn #include <stdlib.h>
 21 e9e8d207 2020-07-25 martijn #include <string.h>
 22 e9e8d207 2020-07-25 martijn #include <time.h>
 23 e9e8d207 2020-07-25 martijn #include <unistd.h>
 24 e9e8d207 2020-07-25 martijn 
 25 cb6053ee 2021-05-11 martijn #include "openbsd-compat.h"
 26 e9e8d207 2020-07-25 martijn #include "opensmtpd.h"
 27 4f29629b 2020-08-30 martijn #include "mheader.h"
 28 e9e8d207 2020-07-25 martijn 
 29 e9e8d207 2020-07-25 martijn struct admd_message {
 30 e9e8d207 2020-07-25 martijn 	int foundmatch;
 31 e9e8d207 2020-07-25 martijn 	int inheader;
 32 e9e8d207 2020-07-25 martijn 	int parsing_headers;
 33 e9e8d207 2020-07-25 martijn 	char **cache;
 34 e9e8d207 2020-07-25 martijn 	size_t cachelen;
 35 4f29629b 2020-08-30 martijn 	size_t headerlen;
 36 e9e8d207 2020-07-25 martijn };
 37 e9e8d207 2020-07-25 martijn 
 38 e9e8d207 2020-07-25 martijn void usage(void);
 39 b7482b5c 2020-10-15 martijn void admd_conf(const char *, const char *);
 40 e9e8d207 2020-07-25 martijn void *admd_message_new(struct osmtpd_ctx *);
 41 e9e8d207 2020-07-25 martijn void admd_message_free(struct osmtpd_ctx *, void *);
 42 79ef769f 2025-05-19 martijn int admd_dataline(struct osmtpd_ctx *, const char *);
 43 79ef769f 2025-05-19 martijn int admd_commit(struct osmtpd_ctx *);
 44 e9e8d207 2020-07-25 martijn void admd_cache(struct admd_message *, const char *);
 45 e9e8d207 2020-07-25 martijn const char *admd_authservid(struct admd_message *);
 46 e9e8d207 2020-07-25 martijn void admd_freecache(struct admd_message *);
 47 e9e8d207 2020-07-25 martijn 
 48 b7482b5c 2020-10-15 martijn char *authservid;
 49 e9e8d207 2020-07-25 martijn int reject = 0;
 50 e9e8d207 2020-07-25 martijn int verbose = 0;
 51 e9e8d207 2020-07-25 martijn 
 52 e9e8d207 2020-07-25 martijn int
 53 e9e8d207 2020-07-25 martijn main(int argc, char *argv[])
 54 e9e8d207 2020-07-25 martijn {
 55 e9e8d207 2020-07-25 martijn 	int ch;
 56 e9e8d207 2020-07-25 martijn 
 57 4f29629b 2020-08-30 martijn 	if (pledge("stdio", NULL) == -1)
 58 4f29629b 2020-08-30 martijn 		osmtpd_err(1, "pledge");
 59 4f29629b 2020-08-30 martijn 
 60 4f29629b 2020-08-30 martijn 	while ((ch = getopt(argc, argv, "rv")) != -1) {
 61 e9e8d207 2020-07-25 martijn 		switch (ch) {
 62 e9e8d207 2020-07-25 martijn 		case 'r':
 63 e9e8d207 2020-07-25 martijn 			reject = 1;
 64 e9e8d207 2020-07-25 martijn 			break;
 65 e9e8d207 2020-07-25 martijn 		case 'v':
 66 e9e8d207 2020-07-25 martijn 			verbose++;
 67 e9e8d207 2020-07-25 martijn 			break;
 68 e9e8d207 2020-07-25 martijn 		default:
 69 e9e8d207 2020-07-25 martijn 			usage();
 70 e9e8d207 2020-07-25 martijn 		}
 71 e9e8d207 2020-07-25 martijn 	}
 72 4f29629b 2020-08-30 martijn 	argc -= optind;
 73 4f29629b 2020-08-30 martijn 	argv += optind;
 74 4f29629b 2020-08-30 martijn 	if (argc > 1)
 75 4f29629b 2020-08-30 martijn 		osmtpd_errx(1, "invalid authservid count");
 76 b7482b5c 2020-10-15 martijn 	if (argc == 1)
 77 b7482b5c 2020-10-15 martijn 		authservid = argv[0];
 78 e9e8d207 2020-07-25 martijn 
 79 e9e8d207 2020-07-25 martijn 	osmtpd_local_message(admd_message_new, admd_message_free);
 80 e9e8d207 2020-07-25 martijn 	osmtpd_register_filter_dataline(admd_dataline);
 81 e9e8d207 2020-07-25 martijn 	osmtpd_register_filter_commit(admd_commit);
 82 b7482b5c 2020-10-15 martijn 	osmtpd_register_conf(admd_conf);
 83 e9e8d207 2020-07-25 martijn 	osmtpd_run();
 84 e9e8d207 2020-07-25 martijn 
 85 e9e8d207 2020-07-25 martijn 	return 0;
 86 e9e8d207 2020-07-25 martijn }
 87 e9e8d207 2020-07-25 martijn 
 88 b7482b5c 2020-10-15 martijn void
 89 b7482b5c 2020-10-15 martijn admd_conf(const char *key, const char *value)
 90 b7482b5c 2020-10-15 martijn {
 91 b7482b5c 2020-10-15 martijn 	if (key == NULL) {
 92 b7482b5c 2020-10-15 martijn 		if (authservid == NULL)
 93 08490891 2020-12-14 martijn 			osmtpd_errx(1, "Didn't receive admd config option");
 94 b7482b5c 2020-10-15 martijn 		return;
 95 b7482b5c 2020-10-15 martijn 	}
 96 b7482b5c 2020-10-15 martijn 	if (strcmp(key, "admd") == 0 && authservid == NULL) {
 97 b7482b5c 2020-10-15 martijn 		if ((authservid = strdup(value)) == NULL)
 98 b7482b5c 2020-10-15 martijn 			osmtpd_err(1, "malloc");
 99 b7482b5c 2020-10-15 martijn 	}
100 b7482b5c 2020-10-15 martijn }
101 b7482b5c 2020-10-15 martijn 
102 e9e8d207 2020-07-25 martijn void *
103 e9e8d207 2020-07-25 martijn admd_message_new(struct osmtpd_ctx *ctx)
104 e9e8d207 2020-07-25 martijn {
105 e9e8d207 2020-07-25 martijn 	struct admd_message *msg;
106 e9e8d207 2020-07-25 martijn 
107 e9e8d207 2020-07-25 martijn 	if ((msg = malloc(sizeof(*msg))) == NULL)
108 e9e8d207 2020-07-25 martijn 		osmtpd_err(1, "malloc");
109 e9e8d207 2020-07-25 martijn 
110 e9e8d207 2020-07-25 martijn 	msg->foundmatch = 0;
111 e9e8d207 2020-07-25 martijn 	msg->inheader = 0;
112 e9e8d207 2020-07-25 martijn 	msg->parsing_headers = 1;
113 e9e8d207 2020-07-25 martijn 	msg->cache = NULL;
114 e9e8d207 2020-07-25 martijn 	msg->cachelen = 0;
115 4f29629b 2020-08-30 martijn 	msg->headerlen = 0;
116 e9e8d207 2020-07-25 martijn 
117 e9e8d207 2020-07-25 martijn 	return msg;
118 e9e8d207 2020-07-25 martijn }
119 e9e8d207 2020-07-25 martijn 
120 e9e8d207 2020-07-25 martijn void
121 e9e8d207 2020-07-25 martijn admd_message_free(struct osmtpd_ctx *ctx, void *data)
122 e9e8d207 2020-07-25 martijn {
123 e9e8d207 2020-07-25 martijn 	struct admd_message *msg = data;
124 e9e8d207 2020-07-25 martijn 
125 e9e8d207 2020-07-25 martijn 	admd_freecache(msg);
126 e9e8d207 2020-07-25 martijn 	free(msg);
127 e9e8d207 2020-07-25 martijn }
128 e9e8d207 2020-07-25 martijn 
129 79ef769f 2025-05-19 martijn int
130 e9e8d207 2020-07-25 martijn admd_dataline(struct osmtpd_ctx *ctx, const char *orig)
131 e9e8d207 2020-07-25 martijn {
132 e9e8d207 2020-07-25 martijn 	struct admd_message *msg = ctx->local_message;
133 e9e8d207 2020-07-25 martijn 	const char *line = orig;
134 e9e8d207 2020-07-25 martijn 	const char *msgauthid;
135 e9e8d207 2020-07-25 martijn 	size_t i;
136 e9e8d207 2020-07-25 martijn 
137 e9e8d207 2020-07-25 martijn 	if (line[0] == '\0')
138 e9e8d207 2020-07-25 martijn 		msg->parsing_headers = 0;
139 e9e8d207 2020-07-25 martijn 	if (line[0] == '.')
140 e9e8d207 2020-07-25 martijn 		line++;
141 e9e8d207 2020-07-25 martijn 	if (msg->parsing_headers) {
142 e9e8d207 2020-07-25 martijn 		if (line[0] != ' ' && line[0] != '\t') {
143 e9e8d207 2020-07-25 martijn 			if (msg->inheader) {
144 e9e8d207 2020-07-25 martijn 				msgauthid = admd_authservid(msg);
145 4f29629b 2020-08-30 martijn 				if (msgauthid == NULL && errno != EINVAL)
146 79ef769f 2025-05-19 martijn 					return 0;
147 4f29629b 2020-08-30 martijn 				if (msgauthid != NULL &&
148 4f29629b 2020-08-30 martijn 				    strcmp(msgauthid, authservid) == 0)
149 e9e8d207 2020-07-25 martijn 					msg->foundmatch = 1;
150 e9e8d207 2020-07-25 martijn 				else {
151 e9e8d207 2020-07-25 martijn 					for (i = 0; i < msg->cachelen; i++)
152 e9e8d207 2020-07-25 martijn 						osmtpd_filter_dataline(ctx,
153 e9e8d207 2020-07-25 martijn 						    "%s", msg->cache[i]);
154 e9e8d207 2020-07-25 martijn 				}
155 e9e8d207 2020-07-25 martijn 				admd_freecache(msg);
156 e9e8d207 2020-07-25 martijn 			}
157 e9e8d207 2020-07-25 martijn 			msg->inheader = 0;
158 e9e8d207 2020-07-25 martijn 		}
159 4f29629b 2020-08-30 martijn 		if (strncasecmp(line, "Authentication-Results", 22) == 0) {
160 4f29629b 2020-08-30 martijn 			line += 22;
161 4f29629b 2020-08-30 martijn 			while (line[0] == ' ' || line[0] == '\t')
162 4f29629b 2020-08-30 martijn 				line++;
163 4f29629b 2020-08-30 martijn 			if (line++[0] == ':') {
164 4f29629b 2020-08-30 martijn 				msg->inheader = 1;
165 4f29629b 2020-08-30 martijn 				admd_cache(msg, orig);
166 79ef769f 2025-05-19 martijn 				return 0;
167 4f29629b 2020-08-30 martijn 			}
168 4f29629b 2020-08-30 martijn 		} else if (msg->inheader &&
169 4f29629b 2020-08-30 martijn 		    (line[0] == ' ' || line[0] == '\t')) {
170 e9e8d207 2020-07-25 martijn 			admd_cache(msg, orig);
171 79ef769f 2025-05-19 martijn 			return 0;
172 e9e8d207 2020-07-25 martijn 		}
173 e9e8d207 2020-07-25 martijn 	}
174 e9e8d207 2020-07-25 martijn 
175 e9e8d207 2020-07-25 martijn 	osmtpd_filter_dataline(ctx, "%s", orig);
176 79ef769f 2025-05-19 martijn 	return 0;
177 e9e8d207 2020-07-25 martijn }
178 e9e8d207 2020-07-25 martijn 
179 79ef769f 2025-05-19 martijn int
180 e9e8d207 2020-07-25 martijn admd_commit(struct osmtpd_ctx *ctx)
181 e9e8d207 2020-07-25 martijn {
182 e9e8d207 2020-07-25 martijn 	struct admd_message *msg = ctx->local_message;
183 e9e8d207 2020-07-25 martijn 
184 e9e8d207 2020-07-25 martijn 	if (reject && msg->foundmatch) {
185 e9e8d207 2020-07-25 martijn 		osmtpd_filter_disconnect(ctx, "Message contains "
186 e9e8d207 2020-07-25 martijn 		    "Authentication-Results header for authserv-id '%s'",
187 e9e8d207 2020-07-25 martijn 		    authservid);
188 e9e8d207 2020-07-25 martijn 		fprintf(stderr, "%016"PRIx64" Message contains "
189 e9e8d207 2020-07-25 martijn 		    "Authentication-Results header for authserv-id '%s': "
190 e9e8d207 2020-07-25 martijn 		    "rejected\n", ctx->reqid, authservid);
191 79ef769f 2025-05-19 martijn 		return 0;
192 e9e8d207 2020-07-25 martijn 	}
193 e9e8d207 2020-07-25 martijn 
194 e9e8d207 2020-07-25 martijn 	osmtpd_filter_proceed(ctx);
195 e9e8d207 2020-07-25 martijn 	if (msg->foundmatch) {
196 e9e8d207 2020-07-25 martijn 		fprintf(stderr, "%016"PRIx64" Message contains "
197 e9e8d207 2020-07-25 martijn 		    "Authentication-Results header for authserv-id '%s': "
198 e9e8d207 2020-07-25 martijn 		    "filtered\n", ctx->reqid, authservid);
199 e9e8d207 2020-07-25 martijn 	} else if (verbose)
200 e9e8d207 2020-07-25 martijn 		fprintf(stderr, "%016"PRIx64" Message contains no "
201 e9e8d207 2020-07-25 martijn 		   "Authentication-Results header for authserv-id '%s'\n",
202 e9e8d207 2020-07-25 martijn 		    ctx->reqid, authservid);
203 e9e8d207 2020-07-25 martijn 
204 79ef769f 2025-05-19 martijn 	return 0;
205 e9e8d207 2020-07-25 martijn }
206 e9e8d207 2020-07-25 martijn 
207 e9e8d207 2020-07-25 martijn void
208 e9e8d207 2020-07-25 martijn admd_cache(struct admd_message *msg, const char *line)
209 e9e8d207 2020-07-25 martijn {
210 e9e8d207 2020-07-25 martijn 	char **tcache;
211 e9e8d207 2020-07-25 martijn 
212 e9e8d207 2020-07-25 martijn 	if ((tcache = reallocarray(msg->cache, msg->cachelen + 1,
213 e9e8d207 2020-07-25 martijn 	    sizeof(*(msg->cache)))) == NULL) {
214 e9e8d207 2020-07-25 martijn 		admd_freecache(msg);
215 79ef769f 2025-05-19 martijn 		osmtpd_err(1, "malloc");
216 e9e8d207 2020-07-25 martijn 	}
217 e9e8d207 2020-07-25 martijn 	msg->cache = tcache;
218 e9e8d207 2020-07-25 martijn 	msg->cache[msg->cachelen] = strdup(line);
219 e9e8d207 2020-07-25 martijn 	if (msg->cache[msg->cachelen] == NULL) {
220 e9e8d207 2020-07-25 martijn 		admd_freecache(msg);
221 79ef769f 2025-05-19 martijn 		osmtpd_err(1, "strdup");
222 e9e8d207 2020-07-25 martijn 	}
223 e9e8d207 2020-07-25 martijn 	msg->cachelen++;
224 4f29629b 2020-08-30 martijn 	msg->headerlen += strlen(line[0] == '.' ? line + 1 : line);
225 e9e8d207 2020-07-25 martijn }
226 e9e8d207 2020-07-25 martijn 
227 e9e8d207 2020-07-25 martijn const char *
228 e9e8d207 2020-07-25 martijn admd_authservid(struct admd_message *msg)
229 e9e8d207 2020-07-25 martijn {
230 4f29629b 2020-08-30 martijn 	char *header0, *header, *line, *end;
231 4f29629b 2020-08-30 martijn 	size_t headerlen;
232 e9e8d207 2020-07-25 martijn 	size_t i = 0;
233 e9e8d207 2020-07-25 martijn 	
234 4f29629b 2020-08-30 martijn 	headerlen = msg->headerlen + (msg->cachelen * 2) + 1;
235 4f29629b 2020-08-30 martijn 	header0 = header = malloc(headerlen);
236 4f29629b 2020-08-30 martijn 	if (header == NULL) {
237 79ef769f 2025-05-19 martijn 		osmtpd_err(1, "malloc");
238 4f29629b 2020-08-30 martijn 		return NULL;
239 4f29629b 2020-08-30 martijn 	}
240 4f29629b 2020-08-30 martijn 	header[0] = '\0';
241 4f29629b 2020-08-30 martijn 	for (i = 0; i < msg->cachelen; i++) {
242 4f29629b 2020-08-30 martijn 		line = msg->cache[i];
243 4f29629b 2020-08-30 martijn 		if (line[0] == '.')
244 4f29629b 2020-08-30 martijn 			line++;
245 4f29629b 2020-08-30 martijn 		if (strlcat(header, line, headerlen) >= headerlen ||
246 4f29629b 2020-08-30 martijn 		    strlcat(header, "\r\n", headerlen) >= headerlen) {
247 4f29629b 2020-08-30 martijn 			osmtpd_errx(1, "miscalculated header\n");
248 4f29629b 2020-08-30 martijn 			exit(1);
249 4f29629b 2020-08-30 martijn 		}
250 4f29629b 2020-08-30 martijn 	}
251 e9e8d207 2020-07-25 martijn 
252 e9e8d207 2020-07-25 martijn 	/* Skip key */
253 4f29629b 2020-08-30 martijn 	header += 22;
254 e9e8d207 2020-07-25 martijn 	while (header[0] == ' ' || header[0] == '\t')
255 e9e8d207 2020-07-25 martijn 		header++;
256 4f29629b 2020-08-30 martijn 	/* : */
257 4f29629b 2020-08-30 martijn 	header++;
258 4f29629b 2020-08-30 martijn 
259 4f29629b 2020-08-30 martijn 	header = osmtpd_mheader_skip_cfws(header, 1);
260 4f29629b 2020-08-30 martijn 
261 4f29629b 2020-08-30 martijn 	if ((end = osmtpd_mheader_skip_value(header, 0)) == NULL) {
262 4f29629b 2020-08-30 martijn 		errno = EINVAL;
263 4f29629b 2020-08-30 martijn 		free(header0);
264 4f29629b 2020-08-30 martijn 		return NULL;
265 e9e8d207 2020-07-25 martijn 	}
266 4f29629b 2020-08-30 martijn 	memmove(header0, header, end - header);
267 4f29629b 2020-08-30 martijn 	header0[end - header] = '\0';
268 4f29629b 2020-08-30 martijn 
269 4f29629b 2020-08-30 martijn 	return header0;
270 e9e8d207 2020-07-25 martijn }
271 e9e8d207 2020-07-25 martijn 
272 e9e8d207 2020-07-25 martijn void
273 e9e8d207 2020-07-25 martijn admd_freecache(struct admd_message *msg)
274 e9e8d207 2020-07-25 martijn {
275 e9e8d207 2020-07-25 martijn 	while (msg->cachelen > 0) {
276 e9e8d207 2020-07-25 martijn 		msg->cachelen--;
277 e9e8d207 2020-07-25 martijn 		free(msg->cache[msg->cachelen]);
278 e9e8d207 2020-07-25 martijn 	}
279 e9e8d207 2020-07-25 martijn 	free(msg->cache);
280 e9e8d207 2020-07-25 martijn 	msg->cache = NULL;
281 e9e8d207 2020-07-25 martijn 	msg->cachelen = 0;
282 4f29629b 2020-08-30 martijn 	msg->headerlen = 0;
283 e9e8d207 2020-07-25 martijn }
284 e9e8d207 2020-07-25 martijn 
285 e9e8d207 2020-07-25 martijn __dead void
286 e9e8d207 2020-07-25 martijn usage(void)
287 e9e8d207 2020-07-25 martijn {
288 e9e8d207 2020-07-25 martijn 	fprintf(stderr, "usage: filter-admdscrub [-rv] [-a authserv-id]\n");
289 e9e8d207 2020-07-25 martijn 	exit(1);
290 e9e8d207 2020-07-25 martijn }