[PATCH 03/18] insert: open Maildir tmp file

Peter Wang novalazy at gmail.com
Wed Jul 25 06:42:32 PDT 2012


Open a unique file in the Maildir tmp directory.
The new message is not yet copied into the file.
---
 notmuch-insert.c |   80 +++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 79 insertions(+), 1 deletions(-)

diff --git a/notmuch-insert.c b/notmuch-insert.c
index 21424cf..f01a6f2 100644
--- a/notmuch-insert.c
+++ b/notmuch-insert.c
@@ -20,12 +20,86 @@
 
 #include "notmuch-client.h"
 
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+
+static notmuch_bool_t
+safe_gethostname (char *hostname, size_t hostname_size)
+{
+    if (gethostname (hostname, hostname_size) == -1) {
+	strncpy (hostname, "unknown", hostname_size);
+    }
+    hostname[hostname_size - 1] = '\0';
+
+    return (strchr (hostname, '/') == NULL);
+}
+
+static int
+maildir_open_tmp (void *ctx, const char *dir, char **tmppath, char **newpath)
+{
+    pid_t pid;
+    char hostname[256];
+    struct timeval tv;
+    char *filename;
+    int fd = -1;
+
+    /* This is the file name format used by Dovecot. */
+    pid = getpid ();
+    if (! safe_gethostname (hostname, sizeof (hostname))) {
+	fprintf (stderr, "Error: invalid host name.\n");
+	return -1;
+    }
+    gettimeofday (&tv, NULL);
+    filename = talloc_asprintf (ctx, "%ld.M%ldP%d.%s",
+				tv.tv_sec, tv.tv_usec, pid, hostname);
+
+    *tmppath = talloc_asprintf (ctx, "%s/tmp/%s", dir, filename);
+
+    do {
+	fd = open (*tmppath, O_WRONLY | O_CREAT | O_TRUNC | O_EXCL, 0666);
+    } while (fd == -1 && errno == EEXIST);
+
+    if (fd != -1) {
+	*newpath = talloc_asprintf (ctx, "%s/new/%s", dir, filename);
+    }
+    else {
+	fprintf (stderr, "Error: opening %s: %s\n",
+		 *tmppath, strerror (errno));
+	talloc_free (*tmppath);
+    }
+
+    talloc_free (filename);
+
+    return fd;
+}
+
+static notmuch_bool_t
+insert_message (void *ctx, notmuch_database_t *notmuch, int fdin,
+		const char *dir)
+{
+    char *tmppath;
+    char *newpath;
+    int fdout;
+
+    fdout = maildir_open_tmp (ctx, dir, &tmppath, &newpath);
+    if (fdout < 0) {
+	return FALSE;
+    }
+
+    close (fdout);
+    unlink (tmppath);
+    return FALSE;
+}
+
 int
 notmuch_insert_command (void *ctx, int argc, char *argv[])
 {
     notmuch_config_t *config;
     notmuch_database_t *notmuch;
     const char *db_path;
+    char *maildir;
+    notmuch_bool_t ret;
 
     config = notmuch_config_open (ctx, NULL, NULL);
     if (config == NULL)
@@ -33,11 +107,15 @@ notmuch_insert_command (void *ctx, int argc, char *argv[])
 
     db_path = notmuch_config_get_database_path (config);
 
+    maildir = talloc_asprintf (ctx, "%s", db_path);
+
     if (notmuch_database_open (notmuch_config_get_database_path (config),
 			       NOTMUCH_DATABASE_MODE_READ_WRITE, &notmuch))
 	return 1;
 
+    ret = insert_message (ctx, notmuch, STDIN_FILENO, maildir);
+
     notmuch_database_destroy (notmuch);
 
-    return 1;
+    return (ret) ? 0 : 1;
 }
-- 
1.7.4.4



More information about the notmuch mailing list