busybox/mailutils/popmaildir.c

/* vi: set sw=4 ts=4: */
/*
 * popmaildir: a simple yet powerful POP3 client
 * Delivers contents of remote mailboxes to local Maildir
 *
 * Inspired by original utility by Nikola Vladov
 *
 * Copyright (C) 2008 by Vladimir Dronnikov <dronnikov@gmail.com>
 *
 * Licensed under GPLv2, see file LICENSE in this tarball for details.
 */
#include "libbb.h"
#include "mail.h"

static void pop3_checkr(const char *fmt, const char *param, char **ret)
{
	const char *msg = command(fmt, param);
	char *answer = xmalloc_fgetline(stdin);
	if (answer && '+' == *answer) {
		if (timeout)
			alarm(0);
		if (ret)
			*ret = answer+4; // skip "+OK "
		else if (ENABLE_FEATURE_CLEAN_UP)
			free(answer);
		return;
	}
	bb_error_msg_and_die("%s failed: %s", msg, answer);
}

static void pop3_check(const char *fmt, const char *param)
{
	pop3_checkr(fmt, param, NULL);
}

int popmaildir_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int popmaildir_main(int argc UNUSED_PARAM, char **argv)
{
	char *buf;
	unsigned nmsg;
	char *hostname;
	pid_t pid;
	const char *retr;
#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
	const char *delivery;
#endif
	unsigned opt_nlines = 0;

	enum {
		OPT_b = 1 << 0,		// -b binary mode. Ignored
		OPT_d = 1 << 1,		// -d,-dd,-ddd debug. Ignored
		OPT_m = 1 << 2,		// -m show used memory. Ignored
		OPT_V = 1 << 3,		// -V version. Ignored
		OPT_c = 1 << 4,		// -c use tcpclient. Ignored
		OPT_a = 1 << 5,		// -a use APOP protocol
		OPT_s = 1 << 6,		// -s skip authorization
		OPT_T = 1 << 7,		// -T get messages with TOP instead with RETR
		OPT_k = 1 << 8,		// -k keep retrieved messages on the server
		OPT_t = 1 << 9,		// -t90 set timeout to 90 sec
		OPT_R = 1 << 10,	// -R20000 remove old messages on the server >= 20000 bytes (requires -k). Ignored
		OPT_Z = 1 << 11,	// -Z11-23 remove messages from 11 to 23 (dangerous). Ignored
		OPT_L = 1 << 12,	// -L50000 not retrieve new messages >= 50000 bytes. Ignored
		OPT_H = 1 << 13,	// -H30 type first 30 lines of a message; (-L12000 -H30). Ignored
		OPT_M = 1 << 14,	// -M\"program arg1 arg2 ...\"; deliver by program. Treated like -F
		OPT_F = 1 << 15,	// -F\"program arg1 arg2 ...\"; filter by program. Treated like -M
	};

	// init global variables
	INIT_G();

	// parse options
	opt_complementary = "-1:dd:t+:R+:L+:H+";
	opts = getopt32(argv,
		"bdmVcasTkt:" "R:Z:L:H:" USE_FEATURE_POPMAILDIR_DELIVERY("M:F:"),
		&timeout, NULL, NULL, NULL, &opt_nlines
		USE_FEATURE_POPMAILDIR_DELIVERY(, &delivery, &delivery) // we treat -M and -F the same
	);
	//argc -= optind;
	argv += optind;

	// get auth info
	if (!(opts & OPT_s))
		get_cred_or_die(STDIN_FILENO);

	// goto maildir
	xchdir(*argv++);

	// launch connect helper, if any
	if (*argv)
		launch_helper((const char **)argv);

	// get server greeting
	pop3_checkr(NULL, NULL, &buf);

	// authenticate (if no -s given)
	if (!(opts & OPT_s)) {
		// server supports APOP and we want it? -> use it
		if ('<' == *buf && (opts & OPT_a)) {
			md5_ctx_t md5;
			// yes! compose <stamp><password>
			char *s = strchr(buf, '>');
			if (s)
				strcpy(s+1, G.pass);
			s = buf;
			// get md5 sum of <stamp><password>
			md5_begin(&md5);
			md5_hash(s, strlen(s), &md5);
			md5_end(s, &md5);
			// NOTE: md5 struct contains enough space
			// so we reuse md5 space instead of xzalloc(16*2+1)
#define md5_hex ((uint8_t *)&md5)
//			uint8_t *md5_hex = (uint8_t *)&md5;
			*bin2hex((char *)md5_hex, s, 16) = '\0';
			// APOP
			s = xasprintf("%s %s", G.user, md5_hex);
#undef md5_hex
			pop3_check("APOP %s", s);
			if (ENABLE_FEATURE_CLEAN_UP) {
				free(s);
				free(buf-4); // buf is "+OK " away from malloc'ed string
			}
		// server ignores APOP -> use simple text authentication
		} else {
			// USER
			pop3_check("USER %s", G.user);
			// PASS
			pop3_check("PASS %s", G.pass);
		}
	}

	// get mailbox statistics
	pop3_checkr("STAT", NULL, &buf);

	// prepare message filename suffix
	hostname = safe_gethostname();
	pid = getpid();

	// get messages counter
	// NOTE: we don't use xatou(buf) since buf is "nmsg nbytes"
	// we only need nmsg and atoi is just exactly what we need
	// if atoi fails to convert buf into number it returns 0
	// in this case the following loop simply will not be executed
	nmsg = atoi(buf);
	if (ENABLE_FEATURE_CLEAN_UP)
		free(buf-4); // buf is "+OK " away from malloc'ed string

	// loop through messages
	retr = (opts & OPT_T) ? xasprintf("TOP %%u %u", opt_nlines) : "RETR %u";
	for (; nmsg; nmsg--) {

		char *filename;
		char *target;
		char *answer;
		FILE *fp;
#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
		int rc;
#endif
		// generate unique filename
		filename  = xasprintf("tmp/%llu.%u.%s",
			monotonic_us(), (unsigned)pid, hostname);

		// retrieve message in ./tmp/ unless filter is specified
		pop3_check(retr, (const char *)(ptrdiff_t)nmsg);

#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
		// delivery helper ordered? -> setup pipe
		if (opts & (OPT_F|OPT_M)) {
			// helper will have $FILENAME set to filename
			xsetenv("FILENAME", filename);
			fp = popen(delivery, "w");
			unsetenv("FILENAME");
			if (!fp) {
				bb_perror_msg("delivery helper");
				break;
			}
		} else
#endif
		// create and open file filename
		fp = xfopen_for_write(filename);

		// copy stdin to fp (either filename or delivery helper)
		while ((answer = xmalloc_fgets_str(stdin, "\r\n")) != NULL) {
			char *s = answer;
			if ('.' == answer[0]) {
				if ('.' == answer[1])
					s++;
				else if ('\r' == answer[1] && '\n' == answer[2] && '\0' == answer[3])
					break;
			}
			//*strchrnul(s, '\r') = '\n';
			fputs(s, fp);
			free(answer);
		}

#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
		// analyse delivery status
		if (opts & (OPT_F|OPT_M)) {
			rc = pclose(fp);
			if (99 == rc) // 99 means bail out
				break;
//			if (rc) // !0 means skip to the next message
				goto skip;
//			// 0 means continue
		} else {
			// close filename
			fclose(fp);
		}
#endif

		// delete message from server
		if (!(opts & OPT_k))
			pop3_check("DELE %u", (const char*)(ptrdiff_t)nmsg);

		// atomically move message to ./new/
		target = xstrdup(filename);
		strncpy(target, "new", 3);
		// ... or just stop receiving on failure
		if (rename_or_warn(filename, target))
			break;
		free(target);

#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
 skip:
#endif
		free(filename);
	}

	// Bye
	pop3_check("QUIT", NULL);

	if (ENABLE_FEATURE_CLEAN_UP) {
		free(G.user);
		free(G.pass);
	}

	return EXIT_SUCCESS;
}
1	/* vi: set sw=4 ts=4: */
2	/*
3	* popmaildir: a simple yet powerful POP3 client
4	* Delivers contents of remote mailboxes to local Maildir
5	*
6	* Inspired by original utility by Nikola Vladov
7	*
8	* Copyright (C) 2008 by Vladimir Dronnikov <dronnikov@gmail.com>
9	*
10	* Licensed under GPLv2, see file LICENSE in this tarball for details.
11	*/
12	#include "libbb.h"
13	#include "mail.h"
14
15	static void pop3_checkr(const char fmt, const char param, char **ret)
16	{
17	const char *msg = command(fmt, param);
18	char *answer = xmalloc_fgetline(stdin);
19	if (answer && '+' == *answer) {
20	if (timeout)
21	alarm(0);
22	if (ret)
23	*ret = answer+4; // skip "+OK "
24	else if (ENABLE_FEATURE_CLEAN_UP)
25	free(answer);
26	return;
27	}
28	bb_error_msg_and_die("%s failed: %s", msg, answer);
29	}
30
31	static void pop3_check(const char fmt, const char param)
32	{
33	pop3_checkr(fmt, param, NULL);
34	}
35
36	int popmaildir_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
37	int popmaildir_main(int argc UNUSED_PARAM, char **argv)
38	{
39	char *buf;
40	unsigned nmsg;
41	char *hostname;
42	pid_t pid;
43	const char *retr;
44	#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
45	const char *delivery;
46	#endif
47	unsigned opt_nlines = 0;
48
49	enum {
50	OPT_b = 1 << 0, // -b binary mode. Ignored
51	OPT_d = 1 << 1, // -d,-dd,-ddd debug. Ignored
52	OPT_m = 1 << 2, // -m show used memory. Ignored
53	OPT_V = 1 << 3, // -V version. Ignored
54	OPT_c = 1 << 4, // -c use tcpclient. Ignored
55	OPT_a = 1 << 5, // -a use APOP protocol
56	OPT_s = 1 << 6, // -s skip authorization
57	OPT_T = 1 << 7, // -T get messages with TOP instead with RETR
58	OPT_k = 1 << 8, // -k keep retrieved messages on the server
59	OPT_t = 1 << 9, // -t90 set timeout to 90 sec
60	OPT_R = 1 << 10, // -R20000 remove old messages on the server >= 20000 bytes (requires -k). Ignored
61	OPT_Z = 1 << 11, // -Z11-23 remove messages from 11 to 23 (dangerous). Ignored
62	OPT_L = 1 << 12, // -L50000 not retrieve new messages >= 50000 bytes. Ignored
63	OPT_H = 1 << 13, // -H30 type first 30 lines of a message; (-L12000 -H30). Ignored
64	OPT_M = 1 << 14, // -M\"program arg1 arg2 ...\"; deliver by program. Treated like -F
65	OPT_F = 1 << 15, // -F\"program arg1 arg2 ...\"; filter by program. Treated like -M
66	};
67
68	// init global variables
69	INIT_G();
70
71	// parse options
72	opt_complementary = "-1:dd:t+:R+:L+:H+";
73	opts = getopt32(argv,
74	"bdmVcasTkt:" "R:Z:L:H:" USE_FEATURE_POPMAILDIR_DELIVERY("M:F:"),
75	&timeout, NULL, NULL, NULL, &opt_nlines
76	USE_FEATURE_POPMAILDIR_DELIVERY(, &delivery, &delivery) // we treat -M and -F the same
77	);
78	//argc -= optind;
79	argv += optind;
80
81	// get auth info
82	if (!(opts & OPT_s))
83	get_cred_or_die(STDIN_FILENO);
84
85	// goto maildir
86	xchdir(*argv++);
87
88	// launch connect helper, if any
89	if (*argv)
90	launch_helper((const char **)argv);
91
92	// get server greeting
93	pop3_checkr(NULL, NULL, &buf);
94
95	// authenticate (if no -s given)
96	if (!(opts & OPT_s)) {
97	// server supports APOP and we want it? -> use it
98	if ('<' == *buf && (opts & OPT_a)) {
99	md5_ctx_t md5;
100	// yes! compose <stamp><password>
101	char *s = strchr(buf, '>');
102	if (s)
103	strcpy(s+1, G.pass);
104	s = buf;
105	// get md5 sum of <stamp><password>
106	md5_begin(&md5);
107	md5_hash(s, strlen(s), &md5);
108	md5_end(s, &md5);
109	// NOTE: md5 struct contains enough space
110	// so we reuse md5 space instead of xzalloc(16*2+1)
111	#define md5_hex ((uint8_t *)&md5)
112	// uint8_t md5_hex = (uint8_t )&md5;
113	bin2hex((char )md5_hex, s, 16) = '\0';
114	// APOP
115	s = xasprintf("%s %s", G.user, md5_hex);
116	#undef md5_hex
117	pop3_check("APOP %s", s);
118	if (ENABLE_FEATURE_CLEAN_UP) {
119	free(s);
120	free(buf-4); // buf is "+OK " away from malloc'ed string
121	}
122	// server ignores APOP -> use simple text authentication
123	} else {
124	// USER
125	pop3_check("USER %s", G.user);
126	// PASS
127	pop3_check("PASS %s", G.pass);
128	}
129	}
130
131	// get mailbox statistics
132	pop3_checkr("STAT", NULL, &buf);
133
134	// prepare message filename suffix
135	hostname = safe_gethostname();
136	pid = getpid();
137
138	// get messages counter
139	// NOTE: we don't use xatou(buf) since buf is "nmsg nbytes"
140	// we only need nmsg and atoi is just exactly what we need
141	// if atoi fails to convert buf into number it returns 0
142	// in this case the following loop simply will not be executed
143	nmsg = atoi(buf);
144	if (ENABLE_FEATURE_CLEAN_UP)
145	free(buf-4); // buf is "+OK " away from malloc'ed string
146
147	// loop through messages
148	retr = (opts & OPT_T) ? xasprintf("TOP %%u %u", opt_nlines) : "RETR %u";
149	for (; nmsg; nmsg--) {
150
151	char *filename;
152	char *target;
153	char *answer;
154	FILE *fp;
155	#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
156	int rc;
157	#endif
158	// generate unique filename
159	filename = xasprintf("tmp/%llu.%u.%s",
160	monotonic_us(), (unsigned)pid, hostname);
161
162	// retrieve message in ./tmp/ unless filter is specified
163	pop3_check(retr, (const char *)(ptrdiff_t)nmsg);
164
165	#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
166	// delivery helper ordered? -> setup pipe
167	if (opts & (OPT_F\|OPT_M)) {
168	// helper will have $FILENAME set to filename
169	xsetenv("FILENAME", filename);
170	fp = popen(delivery, "w");
171	unsetenv("FILENAME");
172	if (!fp) {
173	bb_perror_msg("delivery helper");
174	break;
175	}
176	} else
177	#endif
178	// create and open file filename
179	fp = xfopen_for_write(filename);
180
181	// copy stdin to fp (either filename or delivery helper)
182	while ((answer = xmalloc_fgets_str(stdin, "\r\n")) != NULL) {
183	char *s = answer;
184	if ('.' == answer[0]) {
185	if ('.' == answer[1])
186	s++;
187	else if ('\r' == answer[1] && '\n' == answer[2] && '\0' == answer[3])
188	break;
189	}
190	//*strchrnul(s, '\r') = '\n';
191	fputs(s, fp);
192	free(answer);
193	}
194
195	#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
196	// analyse delivery status
197	if (opts & (OPT_F\|OPT_M)) {
198	rc = pclose(fp);
199	if (99 == rc) // 99 means bail out
200	break;
201	// if (rc) // !0 means skip to the next message
202	goto skip;
203	// // 0 means continue
204	} else {
205	// close filename
206	fclose(fp);
207	}
208	#endif
209
210	// delete message from server
211	if (!(opts & OPT_k))
212	pop3_check("DELE %u", (const char*)(ptrdiff_t)nmsg);
213
214	// atomically move message to ./new/
215	target = xstrdup(filename);
216	strncpy(target, "new", 3);
217	// ... or just stop receiving on failure
218	if (rename_or_warn(filename, target))
219	break;
220	free(target);
221
222	#if ENABLE_FEATURE_POPMAILDIR_DELIVERY
223	skip:
224	#endif
225	free(filename);
226	}
227
228	// Bye
229	pop3_check("QUIT", NULL);
230
231	if (ENABLE_FEATURE_CLEAN_UP) {
232	free(G.user);
233	free(G.pass);
234	}
235
236	return EXIT_SUCCESS;
237	}