masqmail: src/masqmail.c comparison

comparison src/masqmail.c @ 251:2babd21e7c75

moved run_queue to a new function plus various small cleanups, mostly cosmetic

author	markus schnalke <meillo@marmaro.de>
date	Thu, 04 Nov 2010 13:00:19 -0300
parents	a41c013c8458
children	82d168dd52fd

comparison

equal deleted inserted replaced

-:a41c013c8458
+:2babd21e7c75
 		ok = queue_delete(*id);
 	}
 	return ok;
 }
+static int
+run_queue(gboolean do_runq, gboolean do_runq_online, char* route_name)
+{
+	int ret;
+	/* queue runs */
+	set_identity(conf.orig_uid, "queue run");
+	if (do_runq) {
+		ret = queue_run();
+	}
+	if (do_runq_online) {
+		if (route_name) {
+			conf.online_detect = g_strdup("argument");
+			set_online_name(route_name);
+		}
+		ret = queue_run_online();
+	}
+	return ret;
+}
 int
 main(int argc, char *argv[])
 {
-	/* cmd line flags */
+	gchar *progname;
-	gchar *conf_file = CONF_FILE;
 	char* opt;
 	gint arg;
+	mta_mode mta_mode = MODE_ACCEPT;
 	gboolean do_listen = FALSE;
 	gboolean do_runq = FALSE;
 	gboolean do_runq_online = FALSE;
 	gboolean do_queue = FALSE;
-	gboolean do_verbose = FALSE;
-	gint debug_level = -1;
-	mta_mode mta_mode = MODE_ACCEPT;
 	gint queue_interval = 0;
+	gchar *M_cmd = NULL;
 	gboolean opt_t = FALSE;
 	gboolean opt_i = FALSE;
 	gboolean exit_failure = FALSE;
-	gchar *M_cmd = NULL;
 	gint exit_code = EXIT_SUCCESS;
+	gchar *conf_file = CONF_FILE;
 	gchar *route_name = NULL;
-	gchar *progname;
 	gchar *f_address = NULL;
+	address *return_path = NULL;  /* may be changed by -f option */
 	gchar *full_sender_name = NULL;
-	address *return_path = NULL;  /* may be changed by -f option */
+	gboolean do_verbose = FALSE;
+	gint debug_level = -1;
 	progname = get_progname(argv[0]);
 	if (strcmp(progname, "mailq") == 0) {
 		mta_mode = MODE_LIST;
 		mta_mode = MODE_SMTP;
 	}
 	/* parse cmd line */
 	for (arg=1; arg<argc && argv[arg][0]=='-'; arg++) {
-		opt = argv[arg] + 1;
+		opt = argv[arg] + 1;  /* points to the char after the dash */
 		if (strcmp(opt, "-") == 0) {
 			/* everything after `--' are address arguments */
 			arg++;
 			break;
 		} else if (strncmp(opt, "B", 1) == 0) {
 			/* we ignore this and throw the argument away */
 			get_optarg(argv, &arg, opt+1);
 		} else if (strncmp(opt, "C", 1) == 0) {
-			if (!(conf_file = get_optarg(argv, &arg, opt+1))) {
+			conf_file = get_optarg(argv, &arg, opt+1);
+			if (!conf_file) {
 				fprintf(stderr, "-C requires a filename as argument.\n");
 				exit(EXIT_FAILURE);
 			}
 		} else if (strncmp(opt, "d", 1) == 0) {
 			gchar *optarg;
 			int dummy;
 			do_runq = TRUE;
 			mta_mode = MODE_RUNQUEUE;
-			if ((optarg = get_optarg(argv, &arg, opt+1))) {
+			optarg = get_optarg(argv, &arg, opt+1);
+			if (optarg) {
+				/* not just one single queue run but regular runs */
 				mta_mode = MODE_DAEMON;
 				queue_interval = time_interval(optarg, &dummy);
 			}
 		} else if (strcmp(opt, "t") == 0) {
 	/* close all possibly open file descriptors, except std{in,out,err} */
 	{
 		int i, max_fd = sysconf(_SC_OPEN_MAX);
-		if (max_fd <= 0)
+		if (max_fd <= 0) {
 			max_fd = 64;
-		for (i = 3; i < max_fd; i++)
+		}
+		for (i=3; i<max_fd; i++) {
 			close(i);
+		}
 	}
 	init_conf();
 	/* if we are not privileged, and the config file was changed we
 	   privileges.
 	   So it is possible for a user to run his own daemon without
 	   breaking security.
 	 */
-	if (strcmp(conf_file, CONF_FILE) != 0) {
+	if ((strcmp(conf_file, CONF_FILE) != 0) && (conf.orig_uid != 0)) {
-		if (conf.orig_uid != 0) {
+		conf.run_as_user = TRUE;
-			conf.run_as_user = TRUE;
+		seteuid(conf.orig_uid);
-			seteuid(conf.orig_uid);
+		setegid(conf.orig_gid);
-			setegid(conf.orig_gid);
+		setuid(conf.orig_uid);
-			setuid(conf.orig_uid);
+		setgid(conf.orig_gid);
-			setgid(conf.orig_gid);
-		}
 	}
 	conf.log_dir = LOG_DIR;
 	logopen();
 	if (!read_conf(conf_file)) {
 		logwrite(LOG_ALERT, "SHUTTING DOWN due to problems reading config\n");
 		exit(5);
 	}
 	logclose();
-	if (do_queue)
+	if (do_queue) {
 		conf.do_queue = TRUE;
-	if (do_verbose)
+	}
+	if (do_verbose) {
 		conf.do_verbose = TRUE;
-	if (debug_level >= 0)  /* if >= 0, it was given by argument */
+	}
+	if (debug_level >= 0) {  /* if >= 0, it was given by argument */
 		conf.debug_level = debug_level;
+	}
 	/* It appears that changing to / ensures that we are never in
 	   a directory which we cannot access. This situation could be
 	   possible after changing identity.
 	   Maybe we should only change to / if we not run as user, to
 	switch (mta_mode) {
 	case MODE_DAEMON:
 		mode_daemon(do_listen, queue_interval, argv);
 		break;
 	case MODE_RUNQUEUE:
-		{
+		exit(run_queue(do_runq, do_runq_online, route_name) ? 0 : 1);
-			/* queue runs */
-			set_identity(conf.orig_uid, "queue run");
-			if (do_runq)
-				exit_code = queue_run() ? EXIT_SUCCESS : EXIT_FAILURE;
-			if (do_runq_online) {
-				if (route_name != NULL) {
-					conf.online_detect = g_strdup("argument");
-					set_online_name(route_name);
-				}
-				exit_code =
-					queue_run_online() ? EXIT_SUCCESS : EXIT_FAILURE;
-			}
-		}
 		break;
 	case MODE_SMTP:
 		mode_smtp();
 		break;
 			exit(exit_failure ? EXIT_FAILURE : EXIT_SUCCESS);
 		}
 		break;
 	case MODE_NONE:
 		break;
 	default:
 		fprintf(stderr, "unknown mode: %d\n", mta_mode);
 		break;
 	}

Mercurial > masqmail

comparison src/masqmail.c @ 251:2babd21e7c75