2 * fakepop - A fake POP server used by the nmh test suite
4 * This code is Copyright (c) 2012, by the authors of nmh. See the
5 * COPYRIGHT file in the root directory of the nmh distribution for
6 * complete copyright information.
15 #include <sys/socket.h>
16 #include <sys/types.h>
17 #include <sys/select.h>
23 #define PIDFILE "/tmp/fakepop.pid"
27 #define CHECKUSER() if (!user) { \
28 putpop(s, "-ERR Aren't you forgetting " \
29 "something? Like the USER command?"); \
32 #define CHECKUSERPASS() CHECKUSER() \
34 putpop(s, "-ERR Um, hello? Forget to " \
39 static void killpidfile(void);
40 static void handleterm(int);
41 static void putpop(int, char *);
42 static void putpopbulk(int, char *);
43 static int getpop(int, char *, ssize_t);
44 static char *readmessage(FILE *);
47 main(int argc, char *argv[])
49 struct addrinfo hints, *res;
56 int octets = 0, rc, l, s, on, user = 0, pass = 0, deleted = 0;
59 fprintf(stderr, "Usage: %s mail-file port username "
60 "password\n", argv[0]);
64 if (!(f = fopen(argv[1], "r"))) {
65 fprintf(stderr, "Unable to open message file \"%s\": %s\n",
66 argv[1], strerror(errno));
71 * POP wants the size of the maildrop in bytes, but with \r\n line
72 * endings. Calculate that.
75 while (fgets(line, sizeof(line), f)) {
76 octets += strlen(line);
77 if (strrchr(line, '\n'))
84 * If there is a pid file around, kill the previously running
88 if (stat(PIDFILE, &st) == 0) {
91 if (!(pid = fopen(PIDFILE, "r"))) {
92 fprintf(stderr, "Cannot open " PIDFILE
93 " (%s), continuing ...\n", strerror(errno));
95 rc = fscanf(pid, "%ld", &oldpid);
99 fprintf(stderr, "Unable to parse pid in "
100 PIDFILE ", continuing ...\n");
102 kill((pid_t) oldpid, SIGTERM);
109 memset(&hints, 0, sizeof(hints));
111 hints.ai_family = PF_INET;
112 hints.ai_socktype = SOCK_STREAM;
113 hints.ai_protocol = IPPROTO_TCP;
114 hints.ai_flags = AI_PASSIVE;
116 rc = getaddrinfo("127.0.0.1", argv[2], &hints, &res);
119 fprintf(stderr, "Unable to resolve localhost/%s: %s\n",
120 argv[2], gai_strerror(rc));
124 l = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
127 fprintf(stderr, "Unable to create listening socket: %s\n",
134 if (setsockopt(l, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1) {
135 fprintf(stderr, "Unable to set SO_REUSEADDR: %s\n",
140 if (bind(l, res->ai_addr, res->ai_addrlen) == -1) {
141 fprintf(stderr, "Unable to bind socket: %s\n", strerror(errno));
145 if (listen(l, 1) == -1) {
146 fprintf(stderr, "Unable to listen on socket: %s\n",
152 * Fork off a copy of ourselves, print out our child pid, then
156 switch (child = fork()) {
158 fprintf(stderr, "Unable to fork child: %s\n", strerror(errno));
163 * Close stdin and stdout so $() in the shell will get an
164 * EOF. For now leave stderr open.
170 printf("%ld\n", (long) child);
175 * Now that our socket and files are set up, wait 30 seconds for
176 * a connection. If there isn't one, then exit.
179 if (!(pid = fopen(PIDFILE, "w"))) {
180 fprintf(stderr, "Cannot open " PIDFILE ": %s\n",
185 fprintf(pid, "%ld\n", (long) getpid());
188 signal(SIGTERM, handleterm);
197 rc = select(l + 1, &readfd, NULL, NULL, &tv);
200 fprintf(stderr, "select() failed: %s\n", strerror(errno));
205 * If we get a timeout, just silently exit
213 * We got a connection; accept it. Right after that close our
214 * listening socket so we won't get any more connections on it.
217 if ((s = accept(l, NULL, NULL)) == -1) {
218 fprintf(stderr, "Unable to accept connection: %s\n",
226 * Pretend to be a POP server
229 putpop(s, "+OK Not really a POP server, but we play one on TV");
232 char linebuf[LINESIZE];
234 rc = getpop(s, linebuf, sizeof(linebuf));
237 break; /* Error or EOF */
239 if (strcasecmp(linebuf, "CAPA") == 0) {
240 putpopbulk(s, "+OK We have no capabilities, really\r\n"
241 "FAKE-CAPABILITY\r\n.\r\n");
242 } else if (strncasecmp(linebuf, "USER ", 5) == 0) {
243 if (strcmp(linebuf + 5, argv[3]) == 0) {
244 putpop(s, "+OK Niiiice!");
247 putpop(s, "-ERR Don't play me, bro!");
249 } else if (strncasecmp(linebuf, "PASS ", 5) == 0) {
251 if (strcmp(linebuf + 5, argv[4]) == 0) {
252 putpop(s, "+OK Aren't you a sight "
256 putpop(s, "-ERR C'mon!");
258 } else if (strcasecmp(linebuf, "STAT") == 0) {
261 strncpy(linebuf, "+OK 0 0", sizeof(linebuf));
263 snprintf(linebuf, sizeof(linebuf),
267 } else if (strcasecmp(linebuf, "RETR 1") == 0) {
270 putpop(s, "-ERR Sorry, don't have it anymore");
272 char *buf = readmessage(f);
273 putpop(s, "+OK Here you go ...");
277 } else if (strncasecmp(linebuf, "RETR ", 5) == 0) {
279 putpop(s, "-ERR Sorry man, out of range!");
280 } else if (strcasecmp(linebuf, "DELE 1") == 0) {
283 putpop(s, "-ERR Um, didn't you tell me "
284 "to delete it already?");
286 putpop(s, "+OK Alright man, I got rid of it");
289 } else if (strncasecmp(linebuf, "DELE ", 5) == 0) {
291 putpop(s, "-ERR Sorry man, out of range!");
292 } else if (strcasecmp(linebuf, "QUIT") == 0) {
293 putpop(s, "+OK See ya, wouldn't want to be ya!");
297 putpop(s, "-ERR Um, what?");
305 * Send one line to the POP client
309 putpop(int socket, char *data)
313 iov[0].iov_base = data;
314 iov[0].iov_len = strlen(data);
315 iov[1].iov_base = "\r\n";
318 writev(socket, iov, 2);
322 * Put one big buffer to the POP server. Should have already had the line
323 * endings set up and dot-stuffed if necessary.
327 putpopbulk(int socket, char *data)
329 ssize_t datalen = strlen(data);
331 write(socket, data, datalen);
335 * Get one line from the POP server. We don't do any buffering here.
339 getpop(int socket, char *data, ssize_t len)
345 cc = read(socket, data + offset, len - offset);
348 fprintf(stderr, "Read failed: %s\n", strerror(errno));
359 fprintf(stderr, "Input buffer overflow "
360 "(%d bytes)\n", (int) len);
364 if (data[offset - 1] == '\n' && data[offset - 2] == '\r') {
365 data[offset - 2] = '\0';
371 #define HAVEROOM(buf, size, used, new) do { \
372 if (used + new > size - 1) { \
373 buf = realloc(buf, size += BUFALLOC); \
378 * Read a file and return it as one malloc()'d buffer. Convert \n to \r\n
379 * and dot-stuff if necessary.
383 readmessage(FILE *file)
385 char *buffer = malloc(BUFALLOC);
386 ssize_t bufsize = BUFALLOC, used = 0;
387 char linebuf[LINESIZE];
392 while (fgets(linebuf, sizeof(linebuf), file)) {
393 if (strcmp(linebuf, ".\n") == 0) {
394 HAVEROOM(buffer, bufsize, used, 4);
395 strcat(buffer, "..\r\n");
398 if (i && linebuf[i - 1] == '\n') {
399 HAVEROOM(buffer, bufsize, used, i + 1);
400 linebuf[i - 1] = '\0';
401 strcat(buffer, linebuf);
402 strcat(buffer, "\r\n");
404 HAVEROOM(buffer, bufsize, used, i);
405 strcat(buffer, linebuf);
411 * Put a terminating dot at the end
414 HAVEROOM(buffer, bufsize, used, 3);
416 strcat(buffer, ".\r\n");
426 handleterm(int signal)
436 * Get rid of our pid file