dmc

commit 109860e2aef53382f2c6c65747f03b95706f3d6c
parent 216bc6709eef0522d05c312913521b7e45c7afe6
Author: nibble <unknown>
Date:   Wed,  4 Nov 2009 18:47:05 +0100

* dmc-mbox is now working
Diffstat:
mbox.c
 | 
172
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++--------------------

1 file changed, 130 insertions(+), 42 deletions(-)
diff --git a/mbox.c b/mbox.c
@@ -1,67 +1,155 @@
-/* dmc :: Copyleft 2009 -- pancake (at) nopcode (dot) org */
+/* dmc :: Copyleft 2009 -- nibble (at) develsec (dot) org */
 
 #include <stdio.h>
 #include <string.h>
 #include <stdlib.h>
+#include <unistd.h>
 
 FILE *fd;
 static char word[1024];
 
 static void mbox_ls () {
-  // TODO
-  // printf("%d %d\n", i, size);
+	char b[1024], from[1024], subject[1024], date[1024], *ptr;
+	int m = 0, headers = 1;
+
+	b[1023] = '\0';
+	fseek (fd, 0, SEEK_SET);
+	while(fgets (b, 1023, fd)) {
+		if (b[0]=='\n') {
+			if (headers) {
+				printf ("%i  %s  %s  %s\n", m++, from, date, subject);
+				headers = 0;
+			} else {
+				fgets (b, 1023, fd);
+				if (!memcmp (b, "From ", 5))
+					headers = 1;
+			}
+		}
+		if (headers) {
+			if (!memcmp (b, "From: ", 6)) {
+				strncpy (from, b+6, 1023);
+				from[1023] = '\0';
+				if ((ptr = strchr (from, '\n')))
+					ptr[0] = '\0';
+			} else
+			if (!memcmp (b, "Subject: ", 9)) {
+				strncpy (subject, b+9, 1023);
+				subject[1023] = '\0';
+				if ((ptr = strchr (subject, '\n')))
+					ptr[0] = '\0';
+			} else
+			if (!memcmp (b, "Date: ", 6)) {
+				strncpy (date, b+6, 1023);
+				date[1023] = '\0';
+				if ((ptr = strchr (date, '\n')))
+					ptr[0] = '\0';
+			}
+		}
+	}
 }
 
 static void mbox_cat (int idx, int body) {
-  // TODO
+	char b[1024];
+	int m = 0, headers = 1;
+
+	b[1023] = '\0';
+	fseek (fd, 0, SEEK_SET);
+	while(fgets (b, 1023, fd)) {
+		if (b[0]=='\n') {
+			if (headers)
+				headers = 0;
+			else {
+				fgets (b, 1023, fd);
+				if (!memcmp (b, "From ", 5)) {
+					headers = 1;
+					m++;
+				}
+			}
+		}
+		if (m == idx && (headers || body))
+			fputs (b, stdout);
+	}
 }
 
 static void mbox_rm (int idx) {
-  // TODO
+	char b[1024], *buf;
+	int m = 0, i = 0, headers = 1, size;
+
+	fseek(fd, 0, SEEK_END);
+	size = ftell(fd);
+	if (!(buf = malloc (size)))
+		return;
+	b[1023] = '\0';
+	fseek (fd, 0, SEEK_SET);
+	while(fgets (b, 1023, fd)) {
+		if (b[0]=='\n') {
+			if (headers)
+				headers = 0;
+			else {
+				fgets (b, 1023, fd);
+				if (!memcmp (b, "From ", 5)) {
+					headers = 1;
+					if (++m == idx) {
+						strcpy(buf + i, "\n");
+						i += 1;
+					}
+				}
+			}
+		}
+		if (m != idx) {
+			strcpy(buf + i, b);
+			i += strlen(b);
+		}
+	}
+	fseek (fd, 0, SEEK_SET);
+	fwrite (buf, 1, i, fd);
+	fflush (fd);
+	ftruncate (fileno(fd), i);
+	free(buf);
 }
 
 static char *getword () {
-  fscanf (stdin, "%255s", word);
-  if (feof (stdin))
-    *word = '\0';
-  return word;
+	fscanf (stdin, "%255s", word);
+	if (feof (stdin))
+		*word = '\0';
+	return word;
 }
 
 static int doword (char *word) {
-  int ret = 1;
-  if (*word == '\0') {
-    /* Do nothing */
-  } else
-  if (!strcmp (word, "ls")) {
-    mbox_ls ();
-  } else
-  if (!strcmp (word, "cat")) {
-    mbox_cat (atoi (getword ()), 1);
-  } else
-  if (!strcmp (word, "rm")) {
-    mbox_rm (atoi (getword ()));
-  } else
-  if (!strcmp (word, "head")) {
-    mbox_cat (atoi (getword ()), 0);
-  } else
-  if (!strcmp (word, "login")) {
-    getword (); // ignore login
-    getword (); // ignore password
-  } else
-  if (!strcmp (word, "exit"))
-    ret = 0;
-  return ret;
+	int ret = 1;
+	if (*word == '\0') {
+		/* Do nothing */
+	} else
+	if (!strcmp (word, "ls")) {
+		mbox_ls ();
+	} else
+	if (!strcmp (word, "cat")) {
+		mbox_cat (atoi (getword ()), 1);
+	} else
+	if (!strcmp (word, "rm")) {
+		mbox_rm (atoi (getword ()));
+	} else
+	if (!strcmp (word, "head")) {
+		mbox_cat (atoi (getword ()), 0);
+	} else
+	if (!strcmp (word, "login")) {
+		getword (); // ignore login
+		getword (); // ignore password
+	} else
+	if (!strcmp (word, "exit"))
+		ret = 0;
+	return ret;
 }
 
 int main (int argc, char **argv) {
-  int ret = 1;
-  if (argc>1) {
-    fd = fopen (argv[1], "r");
-    if (fd != NULL) {
-      ret = 0;
-      while (doword (getword ()));
-      fclose (fd);
-    } else fprintf (stderr, "Cannot open %s\n", argv[1]);
-  } else fprintf (stderr, "Usage: dmc-mbox [mbox-file] 2> body > fifo < input\n");
-  return ret;
+	int ret = 1;
+	if (argc>1) {
+		fd = fopen (argv[1], "r+");
+		if (fd != NULL) {
+			ret = 0;
+			while (doword (getword ()));
+			fclose (fd);
+		} else fprintf (stderr, "Cannot open %s\n", argv[1]);
+	} else fprintf (stderr, "Usage: dmc-mbox [mbox-file] 2> body > fifo < input\n");
+	return ret;
 }