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;

unsigned int options;           /* Program options. */
int verbosity;                  /* Program verbosity level (-2..2). */
unsigned int flags = 0;         /* Program flags. */
unsigned int interval = 0;      /* Poll at the specified interval. */
conn_t connpri, connaux;        /* Primary and auxiliary IMAP connection. */

char logfile[PATH_MAX];         /* Log file. */
char *home = NULL;              /* User's home directory. */
char charset[CHARSET_LEN];      /* Charset for IMAP SEARCH requests. */

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


void usage(void);
void version(void);


/*
 * IMAPFilter: an IMAP mail filtering utility.
 */
int
main(int argc, char *argv[])
{
        int c, r;
        char *conffile;         /* Configuration file. */
        account_t *ca;          /* Current account. */
        mbox_t *cm;             /* Current mailbox. */

        setlocale(LC_ALL, "");

        home = getenv("HOME");
        options = (OPTION_NAMESPACE | OPTION_PEEK);
        verbosity = 0;
        *charset = 0;
        *logfile = 0;
        conffile = NULL;
        connpri.sock = connaux.sock = -1;

        while ((c = getopt(argc, argv, "c:d:Dhkl:"
#ifdef ENCRYPTED_PASSWORDS
                    "p"
#endif
                    "qvV")) != -1) {
                switch (c) {
                case 'c':
                        conffile = optarg;
                        break;
                case 'd':
                        options |= OPTION_DAEMON_MODE;
                        errno = 0;
                        interval = strtoul(optarg, NULL, 10);
                        if (errno)
                                interval = 0;
                        break;
                case 'D':
                        options |= OPTION_DEBUG;
                        break;
                case 'h':
                        usage();
                        exit(ERROR_UNDEFINED);
                        break;
                case 'k':
                        kill_imapfilter();
                        break;
                case 'l':
                        strncat(logfile, optarg, PATH_MAX - 1);
                        break;
#ifdef ENCRYPTED_PASSWORDS
                case 'p':
                        options |= OPTION_PASSWORD_EDITOR;
                        break;
#endif
                case 'q':
                        if (verbosity > -2)
                                verbosity--;
                        break;
                case 'v':
                        if (verbosity < 2)
                                verbosity++;
                        break;
                case 'V':
                        version();
                        exit(ERROR_UNDEFINED);
                        break;
                default:
                        usage();
                        exit(ERROR_UNDEFINED);
                        break;
                }
        }

        debug_start();

        create_homedir();

        lockfile_check();
        lockfile_create();

        if (!(options & OPTION_DEBUG))
                corefile_disable();

        tty_store();
        catch_signals();

        read_config(conffile);

#ifdef ENCRYPTED_PASSWORDS
        read_passwords();

        if ((options & OPTION_PASSWORD_EDITOR)) {
                password_editor();

                secmem_clear();
                lockfile_remove();

                exit(0);
        }
#endif

        log_start();

        init_buffer(&ibuf);
        init_buffer(&obuf);

        if (options & OPTION_DAEMON_MODE)
                verbosity = -2;

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

                        if (setjmp(acctloop))
                                continue;

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

                        r = greeting_response(&connpri);

                        if (options & OPTION_DEBUG)
                                test(&connpri);

                        if (check_capabilities(&connpri))
                                continue;

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

                        log_info(LOG_ACCOUNT, ca->key);

                        if (r != RESPONSE_PREAUTH) {
                                if (ca->passwdattr == PASSWORD_NONE) {
                                        printf("Enter password for %s@%s: ",
                                            ca->username, ca->server);
                                        get_password(ca->password, PASSWORD_LEN);
                                        ca->passwdattr = PASSWORD_PLAIN;
                                }
#ifdef CRAM_MD5
                                if (connpri.caps & CAPABILITY_AUTH_CRAM_MD5)
                                        r = auth_cram_md5(&connpri,
                                            ca->username, ca->password);
                                else
#endif
                                        r = login(&connpri, ca->username,
                                            ca->password);

                                if (r == RESPONSE_NO) {
                                        error("username %s or password rejected "
                                            "at %s\n", ca->username, 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. */
                if (options & OPTION_DAEMON_MODE &&
                    !(flags & FLAG_DAEMON_MODE)) {

                        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_MODE;
                }
                if (options & OPTION_DAEMON_MODE &&
                    flags & FLAG_SIGUSR1_RECEIVED) {
                        reread_config(conffile);
                        continue;
                }
                if (interval)
                        sleep(interval);
        } while (options & OPTION_DAEMON_MODE && interval);

        log_stop();
        secmem_clear();

        lockfile_remove();

        debug_stop();

        exit(0);
}


/*
 * Print a very brief usage message.
 */
void
usage(void)
{
        fprintf(stderr,
            "usage: imapfilter [-bDhk"
#ifdef ENCRYPTED_PASSWORDS
            "p"
#endif
            "qvV] [-c configfile] [-d interval] [-l logfile]\n");
}


/*
 * 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
        );
}
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.10	unsigned int options; /* Program options. */
30	lefcha	1.62	int verbosity; /* Program verbosity level (-2..2). */
31	lefcha	1.29	unsigned int flags = 0; /* Program flags. */
32			unsigned int interval = 0; /* Poll at the specified interval. */
33	lefcha	1.51	conn_t connpri, connaux; /* Primary and auxiliary IMAP connection. */
34	lefcha	1.41
35	lefcha	1.5	char logfile[PATH_MAX]; /* Log file. */
36	lefcha	1.25	char home = NULL; / User's home directory. */
37	lefcha	1.40	char charset[CHARSET_LEN]; /* Charset for IMAP SEARCH requests. */
38	lefcha	1.42
39	lefcha	1.46	jmp_buf acctloop; /* Non-local exit in case of network error. */
40	lefcha	1.1
41	lefcha	1.29
42	lefcha	1.52	void usage(void);
43			void version(void);
44
45
46	lefcha	1.2	/*
47	lefcha	1.46	* IMAPFilter: an IMAP mail filtering utility.
48	lefcha	1.2	*/
49	lefcha	1.42	int
50			main(int argc, char *argv[])
51	lefcha	1.1	{
52	lefcha	1.58	int c, r;
53	lefcha	1.55	char conffile; / Configuration file. */
54	lefcha	1.42	account_t ca; / Current account. */
55			mbox_t cm; / Current mailbox. */
56	lefcha	1.33
57	lefcha	1.45	setlocale(LC_ALL, "");
58	lefcha	1.40
59	lefcha	1.42	home = getenv("HOME");
60	lefcha	1.62	options = (OPTION_NAMESPACE \| OPTION_PEEK);
61			verbosity = 0;
62	lefcha	1.42	*charset = 0;
63			*logfile = 0;
64	lefcha	1.55	conffile = NULL;
65			connpri.sock = connaux.sock = -1;
66	lefcha	1.33
67	lefcha	1.60	while ((c = getopt(argc, argv, "c:d:Dhkl:"
68	lefcha	1.29	#ifdef ENCRYPTED_PASSWORDS
69	lefcha	1.42	"p"
70	lefcha	1.25	#endif
71	lefcha	1.62	"qvV")) != -1) {
72	lefcha	1.42	switch (c) {
73			case 'c':
74	lefcha	1.55	conffile = optarg;
75	lefcha	1.42	break;
76			case 'd':
77			options \|= OPTION_DAEMON_MODE;
78			errno = 0;
79			interval = strtoul(optarg, NULL, 10);
80			if (errno)
81			interval = 0;
82			break;
83	lefcha	1.60	case 'D':
84			options \|= OPTION_DEBUG;
85			break;
86	lefcha	1.42	case 'h':
87			usage();
88			exit(ERROR_UNDEFINED);
89			break;
90			case 'k':
91			kill_imapfilter();
92			break;
93			case 'l':
94			strncat(logfile, optarg, PATH_MAX - 1);
95			break;
96	lefcha	1.25	#ifdef ENCRYPTED_PASSWORDS
97	lefcha	1.42	case 'p':
98			options \|= OPTION_PASSWORD_EDITOR;
99			break;
100			#endif
101			case 'q':
102	lefcha	1.62	if (verbosity > -2)
103			verbosity--;
104	lefcha	1.42	break;
105			case 'v':
106	lefcha	1.62	if (verbosity < 2)
107			verbosity++;
108	lefcha	1.42	break;
109			case 'V':
110			version();
111			exit(ERROR_UNDEFINED);
112			break;
113			default:
114			usage();
115			exit(ERROR_UNDEFINED);
116			break;
117			}
118	lefcha	1.1	}
119	lefcha	1.32
120	lefcha	1.60	debug_start();
121
122	lefcha	1.42	create_homedir();
123	lefcha	1.33
124	lefcha	1.42	lockfile_check();
125			lockfile_create();
126	lefcha	1.28
127	lefcha	1.60	if (!(options & OPTION_DEBUG))
128			corefile_disable();
129	lefcha	1.33
130	lefcha	1.42	tty_store();
131			catch_signals();
132	lefcha	1.24
133	lefcha	1.55	read_config(conffile);
134	lefcha	1.33
135	lefcha	1.25	#ifdef ENCRYPTED_PASSWORDS
136	lefcha	1.42	read_passwords();
137	lefcha	1.33
138	lefcha	1.42	if ((options & OPTION_PASSWORD_EDITOR)) {
139			password_editor();
140	lefcha	1.33
141	lefcha	1.42	secmem_clear();
142			lockfile_remove();
143	lefcha	1.33
144	lefcha	1.42	exit(0);
145			}
146	lefcha	1.25	#endif
147	lefcha	1.33
148	lefcha	1.60	log_start();
149	lefcha	1.32
150	lefcha	1.50	init_buffer(&ibuf);
151			init_buffer(&obuf);
152	lefcha	1.32
153	lefcha	1.62	if (options & OPTION_DAEMON_MODE)
154			verbosity = -2;
155
156	lefcha	1.42	do {
157			for (ca = accounts; ca != NULL; ca = ca->next) {
158	lefcha	1.7
159	lefcha	1.42	if (setjmp(acctloop))
160			continue;
161	lefcha	1.35
162	lefcha	1.51	if (init_connection(&connpri, ca->server, ca->port,
163	lefcha	1.42	ca->ssl))
164			continue;
165	lefcha	1.21
166	lefcha	1.51	r = greeting_response(&connpri);
167	lefcha	1.18
168	lefcha	1.60	if (options & OPTION_DEBUG)
169			test(&connpri);
170	lefcha	1.45
171	lefcha	1.51	if (check_capabilities(&connpri))
172	lefcha	1.42	continue;
173	lefcha	1.1
174	lefcha	1.49	#ifdef SSL_TLS
175			if (ca->ssl == SSL_DISABLED &&
176	lefcha	1.51	connpri.caps & CAPABILITY_STARTTLS)
177	lefcha	1.52	if (negotiate_tls(&connpri) == RESPONSE_OK)
178	lefcha	1.51	check_capabilities(&connpri);
179	lefcha	1.49	#endif
180
181	lefcha	1.42	log_info(LOG_ACCOUNT, ca->key);
182	lefcha	1.1
183	lefcha	1.42	if (r != RESPONSE_PREAUTH) {
184			if (ca->passwdattr == PASSWORD_NONE) {
185			printf("Enter password for %s@%s: ",
186			ca->username, ca->server);
187			get_password(ca->password, PASSWORD_LEN);
188			ca->passwdattr = PASSWORD_PLAIN;
189			}
190	lefcha	1.48	#ifdef CRAM_MD5
191	lefcha	1.51	if (connpri.caps & CAPABILITY_AUTH_CRAM_MD5)
192	lefcha	1.52	r = auth_cram_md5(&connpri,
193	lefcha	1.48	ca->username, ca->password);
194			else
195			#endif
196	lefcha	1.51	r = login(&connpri, ca->username,
197	lefcha	1.48	ca->password);
198
199			if (r == RESPONSE_NO) {
200	lefcha	1.42	error("username %s or password rejected "
201			"at %s\n", ca->username, ca->server);
202			continue;
203			}
204			}
205	lefcha	1.51	check_namespace(&connpri);
206	lefcha	1.42
207			for (cm = ca->mboxes; cm != NULL; cm = cm->next)
208			if (*cm->filters == NULL)
209	lefcha	1.51	mailbox_status(&connpri, cm->name);
210			else if (!select_mailbox(&connpri, cm->name)) {
211	lefcha	1.43	apply_filters(cm->name, cm->filters);
212	lefcha	1.51	close_mailbox(&connpri);
213	lefcha	1.42	}
214	lefcha	1.51	logout(&connpri);
215	lefcha	1.33
216	lefcha	1.51	close_connection(&connpri);
217	lefcha	1.33	}
218	lefcha	1.11
219	lefcha	1.42	/* Fork if in daemon mode. */
220	lefcha	1.58	if (options & OPTION_DAEMON_MODE &&
221			!(flags & FLAG_DAEMON_MODE)) {
222
223			switch (fork()) {
224			case -1:
225			fatal(ERROR_FORK, "forking; %s\n",
226			strerror(errno));
227			break;
228			case 0:
229			break;
230			default:
231	lefcha	1.60	log_stop();
232	lefcha	1.58	secmem_clear();
233	lefcha	1.60	debug_stop();
234	lefcha	1.58	exit(0);
235			break;
236			}
237
238			if (setsid() == -1)
239			fatal(ERROR_FORK, "creating session; %s\n",
240			strerror(errno));
241
242			switch (fork()) {
243	lefcha	1.42	case -1:
244			fatal(ERROR_FORK, "forking; %s\n",
245			strerror(errno));
246			break;
247			case 0:
248			break;
249			default:
250	lefcha	1.60	log_stop();
251	lefcha	1.42	secmem_clear();
252	lefcha	1.60	debug_stop();
253	lefcha	1.42	exit(0);
254			break;
255			}
256	lefcha	1.58
257			close(STDIN_FILENO);
258			close(STDOUT_FILENO);
259			close(STDERR_FILENO);
260			if (open("/dev/null", O_RDWR) != -1) {
261			dup(STDIN_FILENO);
262			dup(STDIN_FILENO);
263			}
264			lockfile_create();
265			corefile_disable();
266
267			flags \|= FLAG_DAEMON_MODE;
268	lefcha	1.46	}
269			if (options & OPTION_DAEMON_MODE &&
270	lefcha	1.54	flags & FLAG_SIGUSR1_RECEIVED) {
271	lefcha	1.55	reread_config(conffile);
272	lefcha	1.46	continue;
273	lefcha	1.42	}
274			if (interval)
275			sleep(interval);
276			} while (options & OPTION_DAEMON_MODE && interval);
277	lefcha	1.17
278	lefcha	1.60	log_stop();
279	lefcha	1.42	secmem_clear();
280	lefcha	1.33
281	lefcha	1.42	lockfile_remove();
282	lefcha	1.33
283	lefcha	1.60	debug_stop();
284
285	lefcha	1.55	exit(0);
286	lefcha	1.1	}
287
288
289			/*
290	lefcha	1.11	* Print a very brief usage message.
291	lefcha	1.2	*/
292	lefcha	1.42	void
293			usage(void)
294	lefcha	1.2	{
295	lefcha	1.42	fprintf(stderr,
296	lefcha	1.60	"usage: imapfilter [-bDhk"
297	lefcha	1.26	#ifdef ENCRYPTED_PASSWORDS
298			"p"
299	lefcha	1.33	#endif
300	lefcha	1.62	"qvV] [-c configfile] [-d interval] [-l logfile]\n");
301	lefcha	1.38	}
302
303
304			/*
305			* Print program's version, and if it is built in, OpenSSL's version number.
306			*/
307	lefcha	1.42	void
308			version(void)
309	lefcha	1.38	{
310	lefcha	1.42	fprintf(stderr, "IMAPFilter %s"
311	lefcha	1.53	#if defined SSL_TLS \|\| defined ENCRYPTED_PASSWORDS \|\| defined CRAM_MD5
312	lefcha	1.38	", OpenSSL 0x%8.8lx"
313			#endif
314			"\n", IMAPFILTER_VERSION
315	lefcha	1.53	#if defined SSL_TLS \|\| defined ENCRYPTED_PASSWORDS \|\| defined CRAM_MD5
316	lefcha	1.38	,SSLeay()
317			#endif
318	lefcha	1.42	);
319	lefcha	1.1	}