Blame

Date:
Message:
Make use of opensmtpd's admd feature and libopensmtpd's osmtpd_register_conf
Actions:
History | Blob | Raw File
  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 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 e9e8d207 2020-07-25 martijn void admd_dataline(struct osmtpd_ctx *, const char *);

 43 e9e8d207 2020-07-25 martijn void admd_commit(struct osmtpd_ctx *);

 44 e9e8d207 2020-07-25 martijn void admd_err(struct admd_message *, const char *);

 45 e9e8d207 2020-07-25 martijn void admd_cache(struct admd_message *, const char *);

 46 e9e8d207 2020-07-25 martijn const char *admd_authservid(struct admd_message *);

 47 e9e8d207 2020-07-25 martijn void admd_freecache(struct admd_message *);

 48 e9e8d207 2020-07-25 martijn 

 49 b7482b5c 2020-10-15 martijn char *authservid;

 50 e9e8d207 2020-07-25 martijn int reject = 0;

 51 e9e8d207 2020-07-25 martijn int verbose = 0;

 52 e9e8d207 2020-07-25 martijn 

 53 e9e8d207 2020-07-25 martijn int

 54 e9e8d207 2020-07-25 martijn main(int argc, char *argv[])

 55 e9e8d207 2020-07-25 martijn {

 56 e9e8d207 2020-07-25 martijn 	int ch;

 57 e9e8d207 2020-07-25 martijn 

 58 4f29629b 2020-08-30 martijn 	if (pledge("stdio", NULL) == -1)

 59 4f29629b 2020-08-30 martijn 		osmtpd_err(1, "pledge");

 60 4f29629b 2020-08-30 martijn 

 61 4f29629b 2020-08-30 martijn 	while ((ch = getopt(argc, argv, "rv")) != -1) {

 62 e9e8d207 2020-07-25 martijn 		switch (ch) {

 63 e9e8d207 2020-07-25 martijn 		case 'r':

 64 e9e8d207 2020-07-25 martijn 			reject = 1;

 65 e9e8d207 2020-07-25 martijn 			break;

 66 e9e8d207 2020-07-25 martijn 		case 'v':

 67 e9e8d207 2020-07-25 martijn 			verbose++;

 68 e9e8d207 2020-07-25 martijn 			break;

 69 e9e8d207 2020-07-25 martijn 		default:

 70 e9e8d207 2020-07-25 martijn 			usage();

 71 e9e8d207 2020-07-25 martijn 		}

 72 e9e8d207 2020-07-25 martijn 	}

 73 4f29629b 2020-08-30 martijn 	argc -= optind;

 74 4f29629b 2020-08-30 martijn 	argv += optind;

 75 4f29629b 2020-08-30 martijn 	if (argc > 1)

 76 4f29629b 2020-08-30 martijn 		osmtpd_errx(1, "invalid authservid count");

 77 b7482b5c 2020-10-15 martijn 	if (argc == 1)

 78 b7482b5c 2020-10-15 martijn 		authservid = argv[0];

 79 e9e8d207 2020-07-25 martijn 

 80 e9e8d207 2020-07-25 martijn 	osmtpd_local_message(admd_message_new, admd_message_free);

 81 e9e8d207 2020-07-25 martijn 	osmtpd_register_filter_dataline(admd_dataline);

 82 e9e8d207 2020-07-25 martijn 	osmtpd_register_filter_commit(admd_commit);

 83 b7482b5c 2020-10-15 martijn 	osmtpd_register_conf(admd_conf);

 84 e9e8d207 2020-07-25 martijn 	osmtpd_run();

 85 e9e8d207 2020-07-25 martijn 

 86 e9e8d207 2020-07-25 martijn 	return 0;

 87 e9e8d207 2020-07-25 martijn }

 88 e9e8d207 2020-07-25 martijn 

 89 b7482b5c 2020-10-15 martijn void

 90 b7482b5c 2020-10-15 martijn admd_conf(const char *key, const char *value)

 91 b7482b5c 2020-10-15 martijn {

 92 b7482b5c 2020-10-15 martijn 	if (key == NULL) {

 93 b7482b5c 2020-10-15 martijn 		if (authservid == NULL)

 94 b7482b5c 2020-10-15 martijn 			osmtpd_errx(1, "Didn't receieve admd config option");

 95 b7482b5c 2020-10-15 martijn 		return;

 96 b7482b5c 2020-10-15 martijn 	}

 97 b7482b5c 2020-10-15 martijn 	if (strcmp(key, "admd") == 0 && authservid == NULL) {

 98 b7482b5c 2020-10-15 martijn 		if ((authservid = strdup(value)) == NULL)

 99 b7482b5c 2020-10-15 martijn 			osmtpd_err(1, "malloc");

100 b7482b5c 2020-10-15 martijn 	}

101 b7482b5c 2020-10-15 martijn }

102 b7482b5c 2020-10-15 martijn 

103 e9e8d207 2020-07-25 martijn void *

104 e9e8d207 2020-07-25 martijn admd_message_new(struct osmtpd_ctx *ctx)

105 e9e8d207 2020-07-25 martijn {

106 e9e8d207 2020-07-25 martijn 	struct admd_message *msg;

107 e9e8d207 2020-07-25 martijn 

108 e9e8d207 2020-07-25 martijn 	if ((msg = malloc(sizeof(*msg))) == NULL)

109 e9e8d207 2020-07-25 martijn 		osmtpd_err(1, "malloc");

110 e9e8d207 2020-07-25 martijn 

111 e9e8d207 2020-07-25 martijn 	msg->foundmatch = 0;

112 e9e8d207 2020-07-25 martijn 	msg->err = 0;

113 e9e8d207 2020-07-25 martijn 	msg->inheader = 0;

114 e9e8d207 2020-07-25 martijn 	msg->parsing_headers = 1;

115 e9e8d207 2020-07-25 martijn 	msg->cache = NULL;

116 e9e8d207 2020-07-25 martijn 	msg->cachelen = 0;

117 4f29629b 2020-08-30 martijn 	msg->headerlen = 0;

118 e9e8d207 2020-07-25 martijn 

119 e9e8d207 2020-07-25 martijn 	return msg;

120 e9e8d207 2020-07-25 martijn }

121 e9e8d207 2020-07-25 martijn 

122 e9e8d207 2020-07-25 martijn void

123 e9e8d207 2020-07-25 martijn admd_message_free(struct osmtpd_ctx *ctx, void *data)

124 e9e8d207 2020-07-25 martijn {

125 e9e8d207 2020-07-25 martijn 	struct admd_message *msg = data;

126 e9e8d207 2020-07-25 martijn 

127 e9e8d207 2020-07-25 martijn 	admd_freecache(msg);

128 e9e8d207 2020-07-25 martijn 	free(msg);

129 e9e8d207 2020-07-25 martijn }

130 e9e8d207 2020-07-25 martijn 

131 e9e8d207 2020-07-25 martijn void

132 e9e8d207 2020-07-25 martijn admd_dataline(struct osmtpd_ctx *ctx, const char *orig)

133 e9e8d207 2020-07-25 martijn {

134 e9e8d207 2020-07-25 martijn 	struct admd_message *msg = ctx->local_message;

135 e9e8d207 2020-07-25 martijn 	const char *line = orig;

136 e9e8d207 2020-07-25 martijn 	const char *msgauthid;

137 e9e8d207 2020-07-25 martijn 	size_t i;

138 e9e8d207 2020-07-25 martijn 

139 e9e8d207 2020-07-25 martijn 	if (msg->err) {

140 e9e8d207 2020-07-25 martijn 		if (line[0] == '.' && line[1] =='\0')

141 e9e8d207 2020-07-25 martijn 			osmtpd_filter_dataline(ctx, ".");

142 e9e8d207 2020-07-25 martijn 		return;

143 e9e8d207 2020-07-25 martijn 	}

144 e9e8d207 2020-07-25 martijn 		

145 e9e8d207 2020-07-25 martijn 	if (line[0] == '\0')

146 e9e8d207 2020-07-25 martijn 		msg->parsing_headers = 0;

147 e9e8d207 2020-07-25 martijn 	if (line[0] == '.')

148 e9e8d207 2020-07-25 martijn 		line++;

149 e9e8d207 2020-07-25 martijn 	if (msg->parsing_headers) {

150 e9e8d207 2020-07-25 martijn 		if (line[0] != ' ' && line[0] != '\t') {

151 e9e8d207 2020-07-25 martijn 			if (msg->inheader) {

152 e9e8d207 2020-07-25 martijn 				msgauthid = admd_authservid(msg);

153 4f29629b 2020-08-30 martijn 				if (msgauthid == NULL && errno != EINVAL)

154 4f29629b 2020-08-30 martijn 					return;

155 4f29629b 2020-08-30 martijn 				if (msgauthid != NULL &&

156 4f29629b 2020-08-30 martijn 				    strcmp(msgauthid, authservid) == 0)

157 e9e8d207 2020-07-25 martijn 					msg->foundmatch = 1;

158 e9e8d207 2020-07-25 martijn 				else {

159 e9e8d207 2020-07-25 martijn 					for (i = 0; i < msg->cachelen; i++)

160 e9e8d207 2020-07-25 martijn 						osmtpd_filter_dataline(ctx,

161 e9e8d207 2020-07-25 martijn 						    "%s", msg->cache[i]);

162 e9e8d207 2020-07-25 martijn 				}

163 e9e8d207 2020-07-25 martijn 				admd_freecache(msg);

164 e9e8d207 2020-07-25 martijn 			}

165 e9e8d207 2020-07-25 martijn 			msg->inheader = 0;

166 e9e8d207 2020-07-25 martijn 		}

167 4f29629b 2020-08-30 martijn 		if (strncasecmp(line, "Authentication-Results", 22) == 0) {

168 4f29629b 2020-08-30 martijn 			line += 22;

169 4f29629b 2020-08-30 martijn 			while (line[0] == ' ' || line[0] == '\t')

170 4f29629b 2020-08-30 martijn 				line++;

171 4f29629b 2020-08-30 martijn 			if (line++[0] == ':') {

172 4f29629b 2020-08-30 martijn 				msg->inheader = 1;

173 4f29629b 2020-08-30 martijn 				admd_cache(msg, orig);

174 4f29629b 2020-08-30 martijn 				return;

175 4f29629b 2020-08-30 martijn 			}

176 4f29629b 2020-08-30 martijn 		} else if (msg->inheader &&

177 4f29629b 2020-08-30 martijn 		    (line[0] == ' ' || line[0] == '\t')) {

178 e9e8d207 2020-07-25 martijn 			admd_cache(msg, orig);

179 e9e8d207 2020-07-25 martijn 			return;

180 e9e8d207 2020-07-25 martijn 		}

181 e9e8d207 2020-07-25 martijn 	}

182 e9e8d207 2020-07-25 martijn 

183 e9e8d207 2020-07-25 martijn 	osmtpd_filter_dataline(ctx, "%s", orig);

184 e9e8d207 2020-07-25 martijn 	return;

185 e9e8d207 2020-07-25 martijn }

186 e9e8d207 2020-07-25 martijn 

187 e9e8d207 2020-07-25 martijn void

188 e9e8d207 2020-07-25 martijn admd_commit(struct osmtpd_ctx *ctx)

189 e9e8d207 2020-07-25 martijn {

190 e9e8d207 2020-07-25 martijn 	struct admd_message *msg = ctx->local_message;

191 e9e8d207 2020-07-25 martijn 

192 e9e8d207 2020-07-25 martijn 	if (msg->err) {

193 e9e8d207 2020-07-25 martijn 		osmtpd_filter_disconnect(ctx, "Internal server error");

194 e9e8d207 2020-07-25 martijn 		return;

195 e9e8d207 2020-07-25 martijn 	}

196 e9e8d207 2020-07-25 martijn 	if (reject && msg->foundmatch) {

197 e9e8d207 2020-07-25 martijn 		osmtpd_filter_disconnect(ctx, "Message contains "

198 e9e8d207 2020-07-25 martijn 		    "Authentication-Results header for authserv-id '%s'",

199 e9e8d207 2020-07-25 martijn 		    authservid);

200 e9e8d207 2020-07-25 martijn 		fprintf(stderr, "%016"PRIx64" Message contains "

201 e9e8d207 2020-07-25 martijn 		    "Authentication-Results header for authserv-id '%s': "

202 e9e8d207 2020-07-25 martijn 		    "rejected\n", ctx->reqid, authservid);

203 e9e8d207 2020-07-25 martijn 		return;

204 e9e8d207 2020-07-25 martijn 	}

205 e9e8d207 2020-07-25 martijn 

206 e9e8d207 2020-07-25 martijn 	osmtpd_filter_proceed(ctx);

207 e9e8d207 2020-07-25 martijn 	if (msg->foundmatch) {

208 e9e8d207 2020-07-25 martijn 		fprintf(stderr, "%016"PRIx64" Message contains "

209 e9e8d207 2020-07-25 martijn 		    "Authentication-Results header for authserv-id '%s': "

210 e9e8d207 2020-07-25 martijn 		    "filtered\n", ctx->reqid, authservid);

211 e9e8d207 2020-07-25 martijn 	} else if (verbose)

212 e9e8d207 2020-07-25 martijn 		fprintf(stderr, "%016"PRIx64" Message contains no "

213 e9e8d207 2020-07-25 martijn 		   "Authentication-Results header for authserv-id '%s'\n",

214 e9e8d207 2020-07-25 martijn 		    ctx->reqid, authservid);

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_err(struct admd_message *message, const char *msg)

219 e9e8d207 2020-07-25 martijn {

220 e9e8d207 2020-07-25 martijn 	message->err = 1;

221 e9e8d207 2020-07-25 martijn 	fprintf(stderr, "%s: %s\n", msg, strerror(errno));

222 e9e8d207 2020-07-25 martijn }

223 e9e8d207 2020-07-25 martijn 

224 e9e8d207 2020-07-25 martijn void

225 e9e8d207 2020-07-25 martijn admd_cache(struct admd_message *msg, const char *line)

226 e9e8d207 2020-07-25 martijn {

227 e9e8d207 2020-07-25 martijn 	char **tcache;

228 e9e8d207 2020-07-25 martijn 

229 e9e8d207 2020-07-25 martijn 	if ((tcache = reallocarray(msg->cache, msg->cachelen + 1,

230 e9e8d207 2020-07-25 martijn 	    sizeof(*(msg->cache)))) == NULL) {

231 e9e8d207 2020-07-25 martijn 		admd_freecache(msg);

232 e9e8d207 2020-07-25 martijn 		admd_err(msg, "malloc");

233 e9e8d207 2020-07-25 martijn 	}

234 e9e8d207 2020-07-25 martijn 	msg->cache = tcache;

235 e9e8d207 2020-07-25 martijn 	msg->cache[msg->cachelen] = strdup(line);

236 e9e8d207 2020-07-25 martijn 	if (msg->cache[msg->cachelen] == NULL) {

237 e9e8d207 2020-07-25 martijn 		admd_freecache(msg);

238 e9e8d207 2020-07-25 martijn 		admd_err(msg, "strdup");

239 e9e8d207 2020-07-25 martijn 	}

240 e9e8d207 2020-07-25 martijn 	msg->cachelen++;

241 4f29629b 2020-08-30 martijn 	msg->headerlen += strlen(line[0] == '.' ? line + 1 : line);

242 e9e8d207 2020-07-25 martijn 	return;

243 e9e8d207 2020-07-25 martijn }

244 e9e8d207 2020-07-25 martijn 

245 e9e8d207 2020-07-25 martijn const char *

246 e9e8d207 2020-07-25 martijn admd_authservid(struct admd_message *msg)

247 e9e8d207 2020-07-25 martijn {

248 4f29629b 2020-08-30 martijn 	char *header0, *header, *line, *end;

249 4f29629b 2020-08-30 martijn 	size_t headerlen;

250 e9e8d207 2020-07-25 martijn 	size_t i = 0;

251 e9e8d207 2020-07-25 martijn 	

252 4f29629b 2020-08-30 martijn 	headerlen = msg->headerlen + (msg->cachelen * 2) + 1;

253 4f29629b 2020-08-30 martijn 	header0 = header = malloc(headerlen);

254 4f29629b 2020-08-30 martijn 	if (header == NULL) {

255 4f29629b 2020-08-30 martijn 		admd_err(msg, "malloc");

256 4f29629b 2020-08-30 martijn 		return NULL;

257 4f29629b 2020-08-30 martijn 	}

258 4f29629b 2020-08-30 martijn 	header[0] = '\0';

259 4f29629b 2020-08-30 martijn 	for (i = 0; i < msg->cachelen; i++) {

260 4f29629b 2020-08-30 martijn 		line = msg->cache[i];

261 4f29629b 2020-08-30 martijn 		if (line[0] == '.')

262 4f29629b 2020-08-30 martijn 			line++;

263 4f29629b 2020-08-30 martijn 		if (strlcat(header, line, headerlen) >= headerlen ||

264 4f29629b 2020-08-30 martijn 		    strlcat(header, "\r\n", headerlen) >= headerlen) {

265 4f29629b 2020-08-30 martijn 			osmtpd_errx(1, "miscalculated header\n");

266 4f29629b 2020-08-30 martijn 			exit(1);

267 4f29629b 2020-08-30 martijn 		}

268 4f29629b 2020-08-30 martijn 	}

269 e9e8d207 2020-07-25 martijn 

270 e9e8d207 2020-07-25 martijn 	/* Skip key */

271 4f29629b 2020-08-30 martijn 	header += 22;

272 e9e8d207 2020-07-25 martijn 	while (header[0] == ' ' || header[0] == '\t')

273 e9e8d207 2020-07-25 martijn 		header++;

274 4f29629b 2020-08-30 martijn 	/* : */

275 4f29629b 2020-08-30 martijn 	header++;

276 4f29629b 2020-08-30 martijn 

277 4f29629b 2020-08-30 martijn 	header = osmtpd_mheader_skip_cfws(header, 1);

278 4f29629b 2020-08-30 martijn 

279 4f29629b 2020-08-30 martijn 	if ((end = osmtpd_mheader_skip_value(header, 0)) == NULL) {

280 4f29629b 2020-08-30 martijn 		errno = EINVAL;

281 4f29629b 2020-08-30 martijn 		free(header0);

282 4f29629b 2020-08-30 martijn 		return NULL;

283 e9e8d207 2020-07-25 martijn 	}

284 4f29629b 2020-08-30 martijn 	memmove(header0, header, end - header);

285 4f29629b 2020-08-30 martijn 	header0[end - header] = '\0';

286 4f29629b 2020-08-30 martijn 

287 4f29629b 2020-08-30 martijn 	return header0;

288 e9e8d207 2020-07-25 martijn }

289 e9e8d207 2020-07-25 martijn 

290 e9e8d207 2020-07-25 martijn void

291 e9e8d207 2020-07-25 martijn admd_freecache(struct admd_message *msg)

292 e9e8d207 2020-07-25 martijn {

293 e9e8d207 2020-07-25 martijn 	while (msg->cachelen > 0) {

294 e9e8d207 2020-07-25 martijn 		msg->cachelen--;

295 e9e8d207 2020-07-25 martijn 		free(msg->cache[msg->cachelen]);

296 e9e8d207 2020-07-25 martijn 	}

297 e9e8d207 2020-07-25 martijn 	free(msg->cache);

298 e9e8d207 2020-07-25 martijn 	msg->cache = NULL;

299 e9e8d207 2020-07-25 martijn 	msg->cachelen = 0;

300 4f29629b 2020-08-30 martijn 	msg->headerlen = 0;

301 e9e8d207 2020-07-25 martijn }

302 e9e8d207 2020-07-25 martijn 

303 e9e8d207 2020-07-25 martijn __dead void

304 e9e8d207 2020-07-25 martijn usage(void)

305 e9e8d207 2020-07-25 martijn {

306 e9e8d207 2020-07-25 martijn 	fprintf(stderr, "usage: filter-admdscrub [-rv] [-a authserv-id]\n");

307 e9e8d207 2020-07-25 martijn 	exit(1);

308 e9e8d207 2020-07-25 martijn }