diff options
| author | Jacek Masiulaniec <jacekm@cvs.openbsd.org> | 2009-09-23 09:40:40 +0000 |
|---|---|---|
| committer | Jacek Masiulaniec <jacekm@cvs.openbsd.org> | 2009-09-23 09:40:40 +0000 |
| commit | 1db1103630e78eecd99203936478f1122c080290 (patch) | |
| tree | 590399c6c60489b59a7c3da7d7d197f8d95450d6 /usr.sbin | |
| parent | 912d1c2733ea9814250621b8ef0192d17275b7f4 (diff) | |
Plug memleak.
Diffstat (limited to 'usr.sbin')
| -rw-r--r-- | usr.sbin/smtpd/mda.c | 43 | ||||
| -rw-r--r-- | usr.sbin/smtpd/smtpd.h | 6 |
2 files changed, 27 insertions, 22 deletions
diff --git a/usr.sbin/smtpd/mda.c b/usr.sbin/smtpd/mda.c index 3033979f7a9..39c719c176f 100644 --- a/usr.sbin/smtpd/mda.c +++ b/usr.sbin/smtpd/mda.c @@ -1,4 +1,4 @@ -/* $OpenBSD: mda.c,v 1.29 2009/09/04 16:28:42 jacekm Exp $ */ +/* $OpenBSD: mda.c,v 1.30 2009/09/23 09:40:39 jacekm Exp $ */ /* * Copyright (c) 2008 Gilles Chehade <gilles@openbsd.org> @@ -105,7 +105,11 @@ mda_dispatch_parent(int sig, short event, void *p) fatalx("mda: mboxfd pass failure"); /* got user's mbox fd */ - b->mboxfd = imsg.fd; + if ((b->mboxfp = fdopen(imsg.fd, "w")) == NULL) { + log_warn("mda: fdopen"); + mda_done(env, b); + break; + } /* * From now on, delivery session must be deinited in @@ -133,7 +137,11 @@ mda_dispatch_parent(int sig, short event, void *p) break; } - b->datafd = imsg.fd; + if ((b->datafp = fdopen(imsg.fd, "r")) == NULL) { + log_warn("mda: fdopen"); + mda_done(env, b); + break; + } /* got message content, copy it to mbox */ if (! mda_store(b)) { @@ -141,13 +149,13 @@ mda_dispatch_parent(int sig, short event, void *p) mda_done(env, b); break; } - close(b->datafd); - b->datafd = -1; + fclose(b->datafp); + b->datafp = NULL; /* closing mboxfd will trigger EOF in forked mda */ - fsync(b->mboxfd); - close(b->mboxfd); - b->mboxfd = -1; + fsync(fileno(b->mboxfp)); + fclose(b->mboxfp); + b->mboxfp = NULL; /* ... unless it is maildir, in which case we need to * "trigger EOF" differently */ @@ -277,8 +285,8 @@ mda_dispatch_runner(int sig, short event, void *p) fatal(NULL); *b = *req; b->env = env; - b->mboxfd = -1; - b->datafd = -1; + b->mboxfp = NULL; + b->datafp = NULL; SPLAY_INSERT(batchtree, &env->batch_queue, b); break; } @@ -414,15 +422,10 @@ mda(struct smtpd *env) int mda_store(struct batch *b) { - FILE *dst; - FILE *src; + FILE *src = b->datafp; + FILE *dst = b->mboxfp; int ch; - if ((dst = fdopen(b->mboxfd, "w")) == NULL) - return 0; - if ((src = fdopen(b->datafd, "r")) == NULL) - return 0; - /* add Return-Path to preserve envelope sender */ /* XXX: remove user provided Return-Path, if any */ if (b->message.sender.user[0] && @@ -507,8 +510,10 @@ mda_done(struct smtpd *env, struct batch *b) /* deallocate resources */ SPLAY_REMOVE(batchtree, &env->batch_queue, b); - close(b->mboxfd); - close(b->datafd); + if (b->mboxfp) + fclose(b->mboxfp); + if (b->datafp) + fclose(b->datafp); free(b); } } diff --git a/usr.sbin/smtpd/smtpd.h b/usr.sbin/smtpd/smtpd.h index ea7cc42b9a6..f86c14ec146 100644 --- a/usr.sbin/smtpd/smtpd.h +++ b/usr.sbin/smtpd/smtpd.h @@ -1,4 +1,4 @@ -/* $OpenBSD: smtpd.h,v 1.143 2009/09/15 16:50:06 jacekm Exp $ */ +/* $OpenBSD: smtpd.h,v 1.144 2009/09/23 09:40:39 jacekm Exp $ */ /* * Copyright (c) 2008 Gilles Chehade <gilles@openbsd.org> @@ -451,8 +451,8 @@ struct batch { struct message message; TAILQ_HEAD(, message) messages; - int mboxfd; - int datafd; + FILE *mboxfp; + FILE *datafp; int cleanup_parent; }; |