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 e9e8d207 2020-07-25 martijn #include "opensmtpd.h"
 26 4f29629b 2020-08-30 martijn #include "mheader.h"
 27 e9e8d207 2020-07-25 martijn 
 28 e9e8d207 2020-07-25 martijn struct admd_message {
 29 e9e8d207 2020-07-25 martijn 	int foundmatch;
 30 e9e8d207 2020-07-25 martijn 	int err;
 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 e9e8d207 2020-07-25 martijn void *admd_message_new(struct osmtpd_ctx *);
 40 e9e8d207 2020-07-25 martijn void admd_message_free(struct osmtpd_ctx *, void *);
 41 e9e8d207 2020-07-25 martijn void admd_dataline(struct osmtpd_ctx *, const char *);
 42 e9e8d207 2020-07-25 martijn void admd_commit(struct osmtpd_ctx *);
 43 e9e8d207 2020-07-25 martijn void admd_err(struct admd_message *, const char *);
 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 4f29629b 2020-08-30 martijn char authservid[256];
 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 4f29629b 2020-08-30 martijn 	if (argc == 1) {
 77 4f29629b 2020-08-30 martijn 		if (strlcpy(authservid, argv[0], sizeof(authservid)) >=
 78 4f29629b 2020-08-30 martijn 		    sizeof(authservid))
 79 4f29629b 2020-08-30 martijn 			osmtpd_errx(1, "authserv-id is too long");
 80 4f29629b 2020-08-30 martijn 	} else {
 81 e9e8d207 2020-07-25 martijn 		if (gethostname(authservid, sizeof(authservid)) == -1)
 82 e9e8d207 2020-07-25 martijn 			osmtpd_err(1, "gethostname");
 83 e9e8d207 2020-07-25 martijn 	}
 84 e9e8d207 2020-07-25 martijn 	if (strchr(authservid, '\r') != NULL ||
 85 e9e8d207 2020-07-25 martijn 	    strchr(authservid, '\n') != NULL)
 86 e9e8d207 2020-07-25 martijn 		osmtpd_errx(1, "ubsupported character in authserv-id");
 87 e9e8d207 2020-07-25 martijn 
 88 e9e8d207 2020-07-25 martijn 	osmtpd_local_message(admd_message_new, admd_message_free);
 89 e9e8d207 2020-07-25 martijn 	osmtpd_register_filter_dataline(admd_dataline);
 90 e9e8d207 2020-07-25 martijn 	osmtpd_register_filter_commit(admd_commit);
 91 e9e8d207 2020-07-25 martijn 	osmtpd_run();
 92 e9e8d207 2020-07-25 martijn 
 93 e9e8d207 2020-07-25 martijn 	return 0;
 94 e9e8d207 2020-07-25 martijn }
 95 e9e8d207 2020-07-25 martijn 
 96 e9e8d207 2020-07-25 martijn void *
 97 e9e8d207 2020-07-25 martijn admd_message_new(struct osmtpd_ctx *ctx)
 98 e9e8d207 2020-07-25 martijn {
 99 e9e8d207 2020-07-25 martijn 	struct admd_message *msg;
100 e9e8d207 2020-07-25 martijn 
101 e9e8d207 2020-07-25 martijn 	if ((msg = malloc(sizeof(*msg))) == NULL)
102 e9e8d207 2020-07-25 martijn 		osmtpd_err(1, "malloc");
103 e9e8d207 2020-07-25 martijn 
104 e9e8d207 2020-07-25 martijn 	msg->foundmatch = 0;
105 e9e8d207 2020-07-25 martijn 	msg->err = 0;
106 e9e8d207 2020-07-25 martijn 	msg->inheader = 0;
107 e9e8d207 2020-07-25 martijn 	msg->parsing_headers = 1;
108 e9e8d207 2020-07-25 martijn 	msg->cache = NULL;
109 e9e8d207 2020-07-25 martijn 	msg->cachelen = 0;
110 4f29629b 2020-08-30 martijn 	msg->headerlen = 0;
111 e9e8d207 2020-07-25 martijn 
112 e9e8d207 2020-07-25 martijn 	return msg;
113 e9e8d207 2020-07-25 martijn }
114 e9e8d207 2020-07-25 martijn 
115 e9e8d207 2020-07-25 martijn void
116 e9e8d207 2020-07-25 martijn admd_message_free(struct osmtpd_ctx *ctx, void *data)
117 e9e8d207 2020-07-25 martijn {
118 e9e8d207 2020-07-25 martijn 	struct admd_message *msg = data;
119 e9e8d207 2020-07-25 martijn 
120 e9e8d207 2020-07-25 martijn 	admd_freecache(msg);
121 e9e8d207 2020-07-25 martijn 	free(msg);
122 e9e8d207 2020-07-25 martijn }
123 e9e8d207 2020-07-25 martijn 
124 e9e8d207 2020-07-25 martijn void
125 e9e8d207 2020-07-25 martijn admd_dataline(struct osmtpd_ctx *ctx, const char *orig)
126 e9e8d207 2020-07-25 martijn {
127 e9e8d207 2020-07-25 martijn 	struct admd_message *msg = ctx->local_message;
128 e9e8d207 2020-07-25 martijn 	const char *line = orig;
129 e9e8d207 2020-07-25 martijn 	const char *msgauthid;
130 e9e8d207 2020-07-25 martijn 	size_t i;
131 e9e8d207 2020-07-25 martijn 
132 e9e8d207 2020-07-25 martijn 	if (msg->err) {
133 e9e8d207 2020-07-25 martijn 		if (line[0] == '.' && line[1] =='\0')
134 e9e8d207 2020-07-25 martijn 			osmtpd_filter_dataline(ctx, ".");
135 e9e8d207 2020-07-25 martijn 		return;
136 e9e8d207 2020-07-25 martijn 	}
137 e9e8d207 2020-07-25 martijn 		
138 e9e8d207 2020-07-25 martijn 	if (line[0] == '\0')
139 e9e8d207 2020-07-25 martijn 		msg->parsing_headers = 0;
140 e9e8d207 2020-07-25 martijn 	if (line[0] == '.')
141 e9e8d207 2020-07-25 martijn 		line++;
142 e9e8d207 2020-07-25 martijn 	if (msg->parsing_headers) {
143 e9e8d207 2020-07-25 martijn 		if (line[0] != ' ' && line[0] != '\t') {
144 e9e8d207 2020-07-25 martijn 			if (msg->inheader) {
145 e9e8d207 2020-07-25 martijn 				msgauthid = admd_authservid(msg);
146 4f29629b 2020-08-30 martijn 				if (msgauthid == NULL && errno != EINVAL)
147 4f29629b 2020-08-30 martijn 					return;
148 4f29629b 2020-08-30 martijn 				if (msgauthid != NULL &&
149 4f29629b 2020-08-30 martijn 				    strcmp(msgauthid, authservid) == 0)
150 e9e8d207 2020-07-25 martijn 					msg->foundmatch = 1;
151 e9e8d207 2020-07-25 martijn 				else {
152 e9e8d207 2020-07-25 martijn 					for (i = 0; i < msg->cachelen; i++)
153 e9e8d207 2020-07-25 martijn 						osmtpd_filter_dataline(ctx,
154 e9e8d207 2020-07-25 martijn 						    "%s", msg->cache[i]);
155 e9e8d207 2020-07-25 martijn 				}
156 e9e8d207 2020-07-25 martijn 				admd_freecache(msg);
157 e9e8d207 2020-07-25 martijn 			}
158 e9e8d207 2020-07-25 martijn 			msg->inheader = 0;
159 e9e8d207 2020-07-25 martijn 		}
160 4f29629b 2020-08-30 martijn 		if (strncasecmp(line, "Authentication-Results", 22) == 0) {
161 4f29629b 2020-08-30 martijn 			line += 22;
162 4f29629b 2020-08-30 martijn 			while (line[0] == ' ' || line[0] == '\t')
163 4f29629b 2020-08-30 martijn 				line++;
164 4f29629b 2020-08-30 martijn 			if (line++[0] == ':') {
165 4f29629b 2020-08-30 martijn 				msg->inheader = 1;
166 4f29629b 2020-08-30 martijn 				admd_cache(msg, orig);
167 4f29629b 2020-08-30 martijn 				return;
168 4f29629b 2020-08-30 martijn 			}
169 4f29629b 2020-08-30 martijn 		} else if (msg->inheader &&
170 4f29629b 2020-08-30 martijn 		    (line[0] == ' ' || line[0] == '\t')) {
171 e9e8d207 2020-07-25 martijn 			admd_cache(msg, orig);
172 e9e8d207 2020-07-25 martijn 			return;
173 e9e8d207 2020-07-25 martijn 		}
174 e9e8d207 2020-07-25 martijn 	}
175 e9e8d207 2020-07-25 martijn 
176 e9e8d207 2020-07-25 martijn 	osmtpd_filter_dataline(ctx, "%s", orig);
177 e9e8d207 2020-07-25 martijn 	return;
178 e9e8d207 2020-07-25 martijn }
179 e9e8d207 2020-07-25 martijn 
180 e9e8d207 2020-07-25 martijn void
181 e9e8d207 2020-07-25 martijn admd_commit(struct osmtpd_ctx *ctx)
182 e9e8d207 2020-07-25 martijn {
183 e9e8d207 2020-07-25 martijn 	struct admd_message *msg = ctx->local_message;
184 e9e8d207 2020-07-25 martijn 
185 e9e8d207 2020-07-25 martijn 	if (msg->err) {
186 e9e8d207 2020-07-25 martijn 		osmtpd_filter_disconnect(ctx, "Internal server error");
187 e9e8d207 2020-07-25 martijn 		return;
188 e9e8d207 2020-07-25 martijn 	}
189 e9e8d207 2020-07-25 martijn 	if (reject && msg->foundmatch) {
190 e9e8d207 2020-07-25 martijn 		osmtpd_filter_disconnect(ctx, "Message contains "
191 e9e8d207 2020-07-25 martijn 		    "Authentication-Results header for authserv-id '%s'",
192 e9e8d207 2020-07-25 martijn 		    authservid);
193 e9e8d207 2020-07-25 martijn 		fprintf(stderr, "%016"PRIx64" Message contains "
194 e9e8d207 2020-07-25 martijn 		    "Authentication-Results header for authserv-id '%s': "
195 e9e8d207 2020-07-25 martijn 		    "rejected\n", ctx->reqid, authservid);
196 e9e8d207 2020-07-25 martijn 		return;
197 e9e8d207 2020-07-25 martijn 	}
198 e9e8d207 2020-07-25 martijn 
199 e9e8d207 2020-07-25 martijn 	osmtpd_filter_proceed(ctx);
200 e9e8d207 2020-07-25 martijn 	if (msg->foundmatch) {
201 e9e8d207 2020-07-25 martijn 		fprintf(stderr, "%016"PRIx64" Message contains "
202 e9e8d207 2020-07-25 martijn 		    "Authentication-Results header for authserv-id '%s': "
203 e9e8d207 2020-07-25 martijn 		    "filtered\n", ctx->reqid, authservid);
204 e9e8d207 2020-07-25 martijn 	} else if (verbose)
205 e9e8d207 2020-07-25 martijn 		fprintf(stderr, "%016"PRIx64" Message contains no "
206 e9e8d207 2020-07-25 martijn 		   "Authentication-Results header for authserv-id '%s'\n",
207 e9e8d207 2020-07-25 martijn 		    ctx->reqid, authservid);
208 e9e8d207 2020-07-25 martijn }
209 e9e8d207 2020-07-25 martijn 
210 e9e8d207 2020-07-25 martijn void
211 e9e8d207 2020-07-25 martijn admd_err(struct admd_message *message, const char *msg)
212 e9e8d207 2020-07-25 martijn {
213 e9e8d207 2020-07-25 martijn 	message->err = 1;
214 e9e8d207 2020-07-25 martijn 	fprintf(stderr, "%s: %s\n", msg, strerror(errno));
215 e9e8d207 2020-07-25 martijn }
216 e9e8d207 2020-07-25 martijn 
217 e9e8d207 2020-07-25 martijn void
218 e9e8d207 2020-07-25 martijn admd_cache(struct admd_message *msg, const char *line)
219 e9e8d207 2020-07-25 martijn {
220 e9e8d207 2020-07-25 martijn 	char **tcache;
221 e9e8d207 2020-07-25 martijn 
222 e9e8d207 2020-07-25 martijn 	if ((tcache = reallocarray(msg->cache, msg->cachelen + 1,
223 e9e8d207 2020-07-25 martijn 	    sizeof(*(msg->cache)))) == NULL) {
224 e9e8d207 2020-07-25 martijn 		admd_freecache(msg);
225 e9e8d207 2020-07-25 martijn 		admd_err(msg, "malloc");
226 e9e8d207 2020-07-25 martijn 	}
227 e9e8d207 2020-07-25 martijn 	msg->cache = tcache;
228 e9e8d207 2020-07-25 martijn 	msg->cache[msg->cachelen] = strdup(line);
229 e9e8d207 2020-07-25 martijn 	if (msg->cache[msg->cachelen] == NULL) {
230 e9e8d207 2020-07-25 martijn 		admd_freecache(msg);
231 e9e8d207 2020-07-25 martijn 		admd_err(msg, "strdup");
232 e9e8d207 2020-07-25 martijn 	}
233 e9e8d207 2020-07-25 martijn 	msg->cachelen++;
234 4f29629b 2020-08-30 martijn 	msg->headerlen += strlen(line[0] == '.' ? line + 1 : line);
235 e9e8d207 2020-07-25 martijn 	return;
236 e9e8d207 2020-07-25 martijn }
237 e9e8d207 2020-07-25 martijn 
238 e9e8d207 2020-07-25 martijn const char *
239 e9e8d207 2020-07-25 martijn admd_authservid(struct admd_message *msg)
240 e9e8d207 2020-07-25 martijn {
241 4f29629b 2020-08-30 martijn 	char *header0, *header, *line, *end;
242 4f29629b 2020-08-30 martijn 	size_t headerlen;
243 e9e8d207 2020-07-25 martijn 	size_t i = 0;
244 e9e8d207 2020-07-25 martijn 	
245 4f29629b 2020-08-30 martijn 	headerlen = msg->headerlen + (msg->cachelen * 2) + 1;
246 4f29629b 2020-08-30 martijn 	header0 = header = malloc(headerlen);
247 4f29629b 2020-08-30 martijn 	if (header == NULL) {
248 4f29629b 2020-08-30 martijn 		admd_err(msg, "malloc");
249 4f29629b 2020-08-30 martijn 		return NULL;
250 4f29629b 2020-08-30 martijn 	}
251 4f29629b 2020-08-30 martijn 	header[0] = '\0';
252 4f29629b 2020-08-30 martijn 	for (i = 0; i < msg->cachelen; i++) {
253 4f29629b 2020-08-30 martijn 		line = msg->cache[i];
254 4f29629b 2020-08-30 martijn 		if (line[0] == '.')
255 4f29629b 2020-08-30 martijn 			line++;
256 4f29629b 2020-08-30 martijn 		if (strlcat(header, line, headerlen) >= headerlen ||
257 4f29629b 2020-08-30 martijn 		    strlcat(header, "\r\n", headerlen) >= headerlen) {
258 4f29629b 2020-08-30 martijn 			osmtpd_errx(1, "miscalculated header\n");
259 4f29629b 2020-08-30 martijn 			exit(1);
260 4f29629b 2020-08-30 martijn 		}
261 4f29629b 2020-08-30 martijn 	}
262 e9e8d207 2020-07-25 martijn 
263 e9e8d207 2020-07-25 martijn 	/* Skip key */
264 4f29629b 2020-08-30 martijn 	header += 22;
265 e9e8d207 2020-07-25 martijn 	while (header[0] == ' ' || header[0] == '\t')
266 e9e8d207 2020-07-25 martijn 		header++;
267 4f29629b 2020-08-30 martijn 	/* : */
268 4f29629b 2020-08-30 martijn 	header++;
269 4f29629b 2020-08-30 martijn 
270 4f29629b 2020-08-30 martijn 	header = osmtpd_mheader_skip_cfws(header, 1);
271 4f29629b 2020-08-30 martijn 
272 4f29629b 2020-08-30 martijn 	if ((end = osmtpd_mheader_skip_value(header, 0)) == NULL) {
273 4f29629b 2020-08-30 martijn 		errno = EINVAL;
274 4f29629b 2020-08-30 martijn 		free(header0);
275 4f29629b 2020-08-30 martijn 		return NULL;
276 e9e8d207 2020-07-25 martijn 	}
277 4f29629b 2020-08-30 martijn 	memmove(header0, header, end - header);
278 4f29629b 2020-08-30 martijn 	header0[end - header] = '\0';
279 4f29629b 2020-08-30 martijn 
280 4f29629b 2020-08-30 martijn 	return header0;
281 e9e8d207 2020-07-25 martijn }
282 e9e8d207 2020-07-25 martijn 
283 e9e8d207 2020-07-25 martijn void
284 e9e8d207 2020-07-25 martijn admd_freecache(struct admd_message *msg)
285 e9e8d207 2020-07-25 martijn {
286 e9e8d207 2020-07-25 martijn 	while (msg->cachelen > 0) {
287 e9e8d207 2020-07-25 martijn 		msg->cachelen--;
288 e9e8d207 2020-07-25 martijn 		free(msg->cache[msg->cachelen]);
289 e9e8d207 2020-07-25 martijn 	}
290 e9e8d207 2020-07-25 martijn 	free(msg->cache);
291 e9e8d207 2020-07-25 martijn 	msg->cache = NULL;
292 e9e8d207 2020-07-25 martijn 	msg->cachelen = 0;
293 4f29629b 2020-08-30 martijn 	msg->headerlen = 0;
294 e9e8d207 2020-07-25 martijn }
295 e9e8d207 2020-07-25 martijn 
296 e9e8d207 2020-07-25 martijn __dead void
297 e9e8d207 2020-07-25 martijn usage(void)
298 e9e8d207 2020-07-25 martijn {
299 e9e8d207 2020-07-25 martijn 	fprintf(stderr, "usage: filter-admdscrub [-rv] [-a authserv-id]\n");
300 e9e8d207 2020-07-25 martijn 	exit(1);
301 e9e8d207 2020-07-25 martijn }