imapfilter/imapfilter/imapfilter.c

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <limits.h>
#include <fcntl.h>
#include <setjmp.h>
#include <locale.h>

#include "config.h"
#include "imapfilter.h"
#include "version.h"
#include "account.h"
#include "filter.h"
#include "buffer.h"

#if defined SSL_TLS || defined ENCRYPTED_PASSWORDS || defined CRAM_MD5
#include <openssl/opensslv.h>
#include <openssl/crypto.h>
#endif


extern account_t *accounts;
extern filter_t *filters;
extern buffer_t ibuf, obuf;

options_t opts;                 /* Program options. */
unsigned int flags = 0;         /* Program flags. */

connection_t connpri, connaux;  /* Primary and auxiliary IMAP connection. */

char *home = NULL;              /* User's home directory. */

jmp_buf acctloop;               /* Non-local exit in case of network error. */


void imapfilter(void);
void daemonize(void);
void usage(void);
void version(void);


/*
 * IMAPFilter: an IMAP mail filtering utility.
 */
int
main(int argc, char *argv[])
{
        int c;
        char *cf;

        setlocale(LC_ALL, "");

        opts.debug = 0;
        opts.verbosity = 0;
        opts.timeout = -1;
        opts.daemon = 0;
        opts.headers = 0;
        opts.errors = 0;
        opts.namespace = 1;
        opts.expunge = 0;
        opts.subscribe = 0;
        opts.peek = 1;
        opts.passwd_editor = 0;
        opts.charset[0] = '\0';
        opts.logfile[0] = '\0';

        home = getenv("HOME");
        cf = NULL;
        connpri.sock = connaux.sock = -1;

        while ((c = getopt(argc, argv, "DVc:d:kl:"
#ifdef ENCRYPTED_PASSWORDS
            "p"
#endif
            "qv")) != -1) {
                switch (c) {
                case 'D':
                        if (opts.debug < 2)
                                opts.debug++;
                        break;
                case 'V':
                        version();
                        /* NOTREACHED */
                case 'c':
                        cf = optarg;
                        break;
                case 'd':
                        errno = 0;
                        opts.daemon = strtoul(optarg, NULL, 10);
                        if (errno)
                                opts.daemon = 0;
                        break;
                case 'k':
                        kill_imapfilter();
                        break;
                case 'l':
                        strncat(opts.logfile, optarg, PATH_MAX - 1);
                        break;
#ifdef ENCRYPTED_PASSWORDS
                case 'p':
                        opts.passwd_editor = 1;
                        break;
#endif
                case 'q':
                        if (opts.verbosity > -2)
                                opts.verbosity--;
                        break;
                case 'v':
                        if (opts.verbosity < 2)
                                opts.verbosity++;
                        break;
                case '?':
                default:
                        usage();
                        /* NOTREACHED */
                }
        }

        debug_start();

        create_homedir();

        lockfile_check();
        lockfile_create();

        if (!opts.debug)
                corefile_disable();

        tty_store();

        catch_signals();

        read_config(cf);

#ifdef ENCRYPTED_PASSWORDS
        read_passwords();

        if (opts.passwd_editor) {
                password_editor();

                secmem_clear();
                lockfile_remove();

                exit(0);
        }
#endif

        log_start();

        buffer_init(&ibuf);
        buffer_init(&obuf);

        if (opts.daemon)
                opts.verbosity = -2;

        do {
                imapfilter();

                if (opts.daemon && !(flags & FLAG_DAEMON))
                        daemonize();
                if (opts.daemon && flags & FLAG_SIGUSR1) {
                        reread_config(cf);
                        continue;
                }
                if (opts.daemon > 0)
                        sleep(opts.daemon);
        } while (opts.daemon);

        log_stop();

        secmem_clear();

        lockfile_remove();

        debug_stop();

        exit(0);
}


/*
 * Go through all accounts and apply filters to mailboxes as defined.
 */
void
imapfilter(void)
{
        int r;
        account_t *ca;
        mbox_t *cm;

        for (ca = accounts; ca != NULL; ca = ca->next) {

                if (setjmp(acctloop))
                        continue;

                if (init_connection(&connpri, ca->server, ca->port, ca->ssl))
                        continue;

                r = response_greeting(&connpri);

                if (opts.debug)
                        test(&connpri);

                if (check_capabilities(&connpri))
                        continue;

#ifdef SSL_TLS
                if (ca->ssl == SSL_DISABLED && connpri.caps & CAPS_STARTTLS)
                        if (negotiate_tls(&connpri) == RESPONSE_OK)
                                check_capabilities(&connpri);
#endif

                log_info(LOG_ACCOUNT, ca->key);

                if (r != RESPONSE_PREAUTH) {
                        if (ca->pass_attr == PASS_ATTR_NONE) {
                                printf("Enter password for %s@%s: ",
                                    ca->user, ca->server);
                                get_password(ca->pass, PASS_LEN);
                                ca->pass_attr = PASS_ATTR_PLAIN;
                        }
#ifdef CRAM_MD5
                        if (connpri.caps & CAPS_CRAMMD5)
                                r = auth_cram_md5(&connpri, ca->user, ca->pass);
                        else
#endif
                                r = login(&connpri, ca->user, ca->pass);

                        if (r == RESPONSE_NO) {
                                error("username %s or password rejected "
                                    "at %s\n", ca->user, ca->server);
                                continue;
                        }
                }
                check_namespace(&connpri);

                for (cm = ca->mboxes; cm != NULL; cm = cm->next)
                        if (*cm->filters == NULL)
                                mailbox_status(&connpri, cm->name);
                        else if (!select_mailbox(&connpri, cm->name)) {
                                apply_filters(cm->name, cm->filters);
                                close_mailbox(&connpri);
                        }
                logout(&connpri);

                close_connection(&connpri);
        }
}


/*
 * Fork if in daemon mode.
 */
void
daemonize(void)
{

        switch (fork()) {
                case -1:
                fatal(ERROR_FORK, "forking; %s\n", strerror(errno));
                break;
        case 0:
                break;
        default:
                log_stop();
                secmem_clear();
                debug_stop();
                exit(0);
                break;
        }

        if (setsid() == -1)
                fatal(ERROR_FORK, "creating session; %s\n", strerror(errno));

        switch (fork()) {
        case -1:
                fatal(ERROR_FORK, "forking; %s\n", strerror(errno));
                break;
        case 0:
                break;
        default:
                log_stop();
                secmem_clear();
                debug_stop();
                exit(0);
                break;
        }

        close(STDIN_FILENO);
        close(STDOUT_FILENO);
        close(STDERR_FILENO);
        if (open("/dev/null", O_RDWR) != -1) {
                dup(STDIN_FILENO);
                dup(STDIN_FILENO);
        }
        lockfile_create();
        corefile_disable();

        flags |= FLAG_DAEMON;
}


/*
 * Print a very brief usage message.
 */
void
usage(void)
{

        fprintf(stderr,
            "usage: imapfilter [-DVbk"
#ifdef ENCRYPTED_PASSWORDS
            "p"
#endif
            "qv] [-c configfile] [-d interval] [-l logfile]\n");

        exit(ERROR_UNDEFINED);
}


/*
 * Print program's version, and if it is built in, OpenSSL's version number.
 */
void
version(void)
{

        fprintf(stderr, "IMAPFilter %s"
#if defined SSL_TLS || defined ENCRYPTED_PASSWORDS || defined CRAM_MD5
            ", OpenSSL 0x%8.8lx"
#endif
            "\n", IMAPFILTER_VERSION
#if defined SSL_TLS || defined ENCRYPTED_PASSWORDS || defined CRAM_MD5
            ,SSLeay()
#endif
        );

        exit(ERROR_UNDEFINED);
}
1	lefcha	1.1	#include <stdio.h>
2	lefcha	1.53	#include <stdlib.h>
3	lefcha	1.55	#include <sys/types.h>
4			#include <unistd.h>
5	lefcha	1.52	#include <string.h>
6			#include <errno.h>
7	lefcha	1.7	#include <limits.h>
8	lefcha	1.58	#include <fcntl.h>
9	lefcha	1.35	#include <setjmp.h>
10	lefcha	1.40	#include <locale.h>
11	lefcha	1.1
12	lefcha	1.39	#include "config.h"
13			#include "imapfilter.h"
14	lefcha	1.52	#include "version.h"
15			#include "account.h"
16			#include "filter.h"
17			#include "buffer.h"
18	lefcha	1.39
19	lefcha	1.53	#if defined SSL_TLS \|\| defined ENCRYPTED_PASSWORDS \|\| defined CRAM_MD5
20	lefcha	1.55	#include <openssl/opensslv.h>
21	lefcha	1.38	#include <openssl/crypto.h>
22			#endif
23	lefcha	1.1
24
25	lefcha	1.7	extern account_t *accounts;
26	lefcha	1.11	extern filter_t *filters;
27	lefcha	1.50	extern buffer_t ibuf, obuf;
28	lefcha	1.7
29	lefcha	1.67	options_t opts; /* Program options. */
30	lefcha	1.29	unsigned int flags = 0; /* Program flags. */
31	lefcha	1.64
32	lefcha	1.65	connection_t connpri, connaux; /* Primary and auxiliary IMAP connection. */
33	lefcha	1.41
34	lefcha	1.25	char home = NULL; / User's home directory. */
35	lefcha	1.42
36	lefcha	1.46	jmp_buf acctloop; /* Non-local exit in case of network error. */
37	lefcha	1.1
38	lefcha	1.29
39	lefcha	1.66	void imapfilter(void);
40			void daemonize(void);
41	lefcha	1.52	void usage(void);
42			void version(void);
43
44
45	lefcha	1.2	/*
46	lefcha	1.46	* IMAPFilter: an IMAP mail filtering utility.
47	lefcha	1.2	*/
48	lefcha	1.42	int
49			main(int argc, char *argv[])
50	lefcha	1.1	{
51	lefcha	1.66	int c;
52			char *cf;
53	lefcha	1.33
54	lefcha	1.45	setlocale(LC_ALL, "");
55	lefcha	1.40
56	lefcha	1.64	opts.debug = 0;
57			opts.verbosity = 0;
58			opts.timeout = -1;
59			opts.daemon = 0;
60			opts.headers = 0;
61			opts.errors = 0;
62			opts.namespace = 1;
63			opts.expunge = 0;
64			opts.subscribe = 0;
65			opts.peek = 1;
66			opts.passwd_editor = 0;
67			opts.charset[0] = '\0';
68			opts.logfile[0] = '\0';
69
70	lefcha	1.42	home = getenv("HOME");
71	lefcha	1.64	cf = NULL;
72	lefcha	1.55	connpri.sock = connaux.sock = -1;
73	lefcha	1.33
74	lefcha	1.63	while ((c = getopt(argc, argv, "DVc:d:kl:"
75	lefcha	1.29	#ifdef ENCRYPTED_PASSWORDS
76	lefcha	1.68	"p"
77	lefcha	1.25	#endif
78	lefcha	1.68	"qv")) != -1) {
79	lefcha	1.42	switch (c) {
80	lefcha	1.63	case 'D':
81	lefcha	1.64	if (opts.debug < 2)
82			opts.debug++;
83	lefcha	1.63	break;
84			case 'V':
85			version();
86			/* NOTREACHED */
87	lefcha	1.42	case 'c':
88	lefcha	1.64	cf = optarg;
89	lefcha	1.42	break;
90			case 'd':
91			errno = 0;
92	lefcha	1.64	opts.daemon = strtoul(optarg, NULL, 10);
93	lefcha	1.42	if (errno)
94	lefcha	1.64	opts.daemon = 0;
95	lefcha	1.42	break;
96			case 'k':
97			kill_imapfilter();
98			break;
99			case 'l':
100	lefcha	1.64	strncat(opts.logfile, optarg, PATH_MAX - 1);
101	lefcha	1.42	break;
102	lefcha	1.25	#ifdef ENCRYPTED_PASSWORDS
103	lefcha	1.42	case 'p':
104	lefcha	1.64	opts.passwd_editor = 1;
105	lefcha	1.42	break;
106			#endif
107			case 'q':
108	lefcha	1.64	if (opts.verbosity > -2)
109			opts.verbosity--;
110	lefcha	1.42	break;
111			case 'v':
112	lefcha	1.64	if (opts.verbosity < 2)
113			opts.verbosity++;
114	lefcha	1.42	break;
115	lefcha	1.67	case '?':
116	lefcha	1.42	default:
117			usage();
118	lefcha	1.63	/* NOTREACHED */
119	lefcha	1.42	}
120	lefcha	1.1	}
121	lefcha	1.32
122	lefcha	1.60	debug_start();
123
124	lefcha	1.42	create_homedir();
125	lefcha	1.33
126	lefcha	1.42	lockfile_check();
127			lockfile_create();
128	lefcha	1.28
129	lefcha	1.64	if (!opts.debug)
130	lefcha	1.60	corefile_disable();
131	lefcha	1.33
132	lefcha	1.42	tty_store();
133	lefcha	1.67
134	lefcha	1.42	catch_signals();
135	lefcha	1.24
136	lefcha	1.64	read_config(cf);
137	lefcha	1.33
138	lefcha	1.25	#ifdef ENCRYPTED_PASSWORDS
139	lefcha	1.42	read_passwords();
140	lefcha	1.33
141	lefcha	1.64	if (opts.passwd_editor) {
142	lefcha	1.42	password_editor();
143	lefcha	1.33
144	lefcha	1.42	secmem_clear();
145			lockfile_remove();
146	lefcha	1.33
147	lefcha	1.42	exit(0);
148			}
149	lefcha	1.25	#endif
150	lefcha	1.33
151	lefcha	1.60	log_start();
152	lefcha	1.32
153	lefcha	1.65	buffer_init(&ibuf);
154			buffer_init(&obuf);
155	lefcha	1.32
156	lefcha	1.64	if (opts.daemon)
157			opts.verbosity = -2;
158	lefcha	1.62
159	lefcha	1.42	do {
160	lefcha	1.66	imapfilter();
161	lefcha	1.7
162	lefcha	1.66	if (opts.daemon && !(flags & FLAG_DAEMON))
163			daemonize();
164			if (opts.daemon && flags & FLAG_SIGUSR1) {
165			reread_config(cf);
166			continue;
167			}
168			if (opts.daemon > 0)
169			sleep(opts.daemon);
170			} while (opts.daemon);
171
172			log_stop();
173	lefcha	1.67
174	lefcha	1.66	secmem_clear();
175
176			lockfile_remove();
177
178			debug_stop();
179
180			exit(0);
181			}
182
183
184			/*
185			* Go through all accounts and apply filters to mailboxes as defined.
186			*/
187			void
188			imapfilter(void)
189			{
190			int r;
191			account_t *ca;
192			mbox_t *cm;
193
194			for (ca = accounts; ca != NULL; ca = ca->next) {
195
196			if (setjmp(acctloop))
197			continue;
198	lefcha	1.35
199	lefcha	1.68	if (init_connection(&connpri, ca->server, ca->port, ca->ssl))
200	lefcha	1.66	continue;
201	lefcha	1.21
202	lefcha	1.66	r = response_greeting(&connpri);
203	lefcha	1.18
204	lefcha	1.66	if (opts.debug)
205			test(&connpri);
206	lefcha	1.45
207	lefcha	1.66	if (check_capabilities(&connpri))
208			continue;
209	lefcha	1.1
210	lefcha	1.49	#ifdef SSL_TLS
211	lefcha	1.68	if (ca->ssl == SSL_DISABLED && connpri.caps & CAPS_STARTTLS)
212	lefcha	1.66	if (negotiate_tls(&connpri) == RESPONSE_OK)
213			check_capabilities(&connpri);
214	lefcha	1.49	#endif
215
216	lefcha	1.66	log_info(LOG_ACCOUNT, ca->key);
217	lefcha	1.1
218	lefcha	1.66	if (r != RESPONSE_PREAUTH) {
219			if (ca->pass_attr == PASS_ATTR_NONE) {
220			printf("Enter password for %s@%s: ",
221			ca->user, ca->server);
222			get_password(ca->pass, PASS_LEN);
223			ca->pass_attr = PASS_ATTR_PLAIN;
224			}
225	lefcha	1.48	#ifdef CRAM_MD5
226	lefcha	1.66	if (connpri.caps & CAPS_CRAMMD5)
227	lefcha	1.68	r = auth_cram_md5(&connpri, ca->user, ca->pass);
228	lefcha	1.66	else
229	lefcha	1.48	#endif
230	lefcha	1.68	r = login(&connpri, ca->user, ca->pass);
231	lefcha	1.48
232	lefcha	1.66	if (r == RESPONSE_NO) {
233			error("username %s or password rejected "
234			"at %s\n", ca->user, ca->server);
235			continue;
236	lefcha	1.42	}
237	lefcha	1.33	}
238	lefcha	1.66	check_namespace(&connpri);
239	lefcha	1.11
240	lefcha	1.66	for (cm = ca->mboxes; cm != NULL; cm = cm->next)
241			if (*cm->filters == NULL)
242			mailbox_status(&connpri, cm->name);
243			else if (!select_mailbox(&connpri, cm->name)) {
244			apply_filters(cm->name, cm->filters);
245			close_mailbox(&connpri);
246	lefcha	1.58	}
247	lefcha	1.66	logout(&connpri);
248	lefcha	1.58
249	lefcha	1.66	close_connection(&connpri);
250			}
251			}
252	lefcha	1.58
253
254	lefcha	1.66	/*
255			* Fork if in daemon mode.
256			*/
257			void
258			daemonize(void)
259			{
260	lefcha	1.17
261	lefcha	1.66	switch (fork()) {
262			case -1:
263	lefcha	1.68	fatal(ERROR_FORK, "forking; %s\n", strerror(errno));
264	lefcha	1.66	break;
265			case 0:
266			break;
267			default:
268			log_stop();
269			secmem_clear();
270			debug_stop();
271			exit(0);
272			break;
273			}
274	lefcha	1.33
275	lefcha	1.66	if (setsid() == -1)
276	lefcha	1.68	fatal(ERROR_FORK, "creating session; %s\n", strerror(errno));
277	lefcha	1.66
278			switch (fork()) {
279			case -1:
280	lefcha	1.68	fatal(ERROR_FORK, "forking; %s\n", strerror(errno));
281	lefcha	1.66	break;
282			case 0:
283			break;
284			default:
285			log_stop();
286			secmem_clear();
287			debug_stop();
288			exit(0);
289			break;
290			}
291	lefcha	1.33
292	lefcha	1.66	close(STDIN_FILENO);
293			close(STDOUT_FILENO);
294			close(STDERR_FILENO);
295			if (open("/dev/null", O_RDWR) != -1) {
296			dup(STDIN_FILENO);
297			dup(STDIN_FILENO);
298			}
299			lockfile_create();
300			corefile_disable();
301	lefcha	1.60
302	lefcha	1.66	flags \|= FLAG_DAEMON;
303	lefcha	1.1	}
304
305
306			/*
307	lefcha	1.11	* Print a very brief usage message.
308	lefcha	1.2	*/
309	lefcha	1.42	void
310			usage(void)
311	lefcha	1.2	{
312	lefcha	1.66
313	lefcha	1.42	fprintf(stderr,
314	lefcha	1.63	"usage: imapfilter [-DVbk"
315	lefcha	1.26	#ifdef ENCRYPTED_PASSWORDS
316			"p"
317	lefcha	1.33	#endif
318	lefcha	1.63	"qv] [-c configfile] [-d interval] [-l logfile]\n");
319
320			exit(ERROR_UNDEFINED);
321	lefcha	1.38	}
322
323
324			/*
325			* Print program's version, and if it is built in, OpenSSL's version number.
326			*/
327	lefcha	1.42	void
328			version(void)
329	lefcha	1.38	{
330	lefcha	1.66
331	lefcha	1.42	fprintf(stderr, "IMAPFilter %s"
332	lefcha	1.53	#if defined SSL_TLS \|\| defined ENCRYPTED_PASSWORDS \|\| defined CRAM_MD5
333	lefcha	1.38	", OpenSSL 0x%8.8lx"
334			#endif
335			"\n", IMAPFILTER_VERSION
336	lefcha	1.53	#if defined SSL_TLS \|\| defined ENCRYPTED_PASSWORDS \|\| defined CRAM_MD5
337	lefcha	1.38	,SSLeay()
338			#endif
339	lefcha	1.42	);
340	lefcha	1.63
341			exit(ERROR_UNDEFINED);
342	lefcha	1.1	}