file: Add l_file_set_contents

3 files changed, 58 insertions, 0 deletions

diff --git a/ell/ell.sym b/ell/ell.sym
index 17acd406..6ef885dd 100644
--- a/ell/ell.sym
+++ b/ell/ell.sym
@@ -312,6 +312,7 @@ global:
 	l_dir_watch_destroy;
 	/* file */
 	l_file_get_contents;
+	l_file_set_contents;
 	/* genl */
 	l_genl_new;
 	l_genl_ref;
diff --git a/ell/file.c b/ell/file.c
index fb510d17..7d82a2c6 100644
--- a/ell/file.c
+++ b/ell/file.c
@@ -15,9 +15,12 @@
 #include <fcntl.h>
 #include <unistd.h>
 #include <errno.h>
+#include <stdlib.h>
+#include <stdio.h>
 
 #include "file.h"
 #include "private.h"
+#include "useful.h"
 
 /**
  * l_file_get_contents:
@@ -72,3 +75,56 @@ error:
 	close(fd);
 	return NULL;
 }
+
+/**
+ * l_file_set_contents:
+ * @filename: Destination filename
+ * @contents: Pointer to the contents
+ * @len: Length in bytes of the contents buffer
+ *
+ * Given a content buffer, write it to a file named @filename.  This function
+ * ensures that the contents are consistent (i.e. due to a crash right after
+ * opening or during write() by writing the contents to a temporary which is then
+ * renamed to @filename.
+ *
+ * Returns: 0 if successful, a negative errno otherwise
+ **/
+LIB_EXPORT int l_file_set_contents(const char *filename,
+					const void *contents, size_t len)
+{
+	_auto_(l_free) char *tmp_path = NULL;
+	ssize_t r;
+	int fd;
+
+	if (!filename || !contents)
+		return -EINVAL;
+
+	tmp_path = l_strdup_printf("%s.XXXXXX.tmp", filename);
+
+	fd = L_TFR(mkostemps(tmp_path, 4, O_CLOEXEC));
+	if (fd == -1)
+		return -errno;
+
+	r = L_TFR(write(fd, contents, len));
+	L_TFR(close(fd));
+
+	if (r != (ssize_t) len) {
+		r = -EIO;
+		goto error_write;
+	}
+
+	/*
+	 * Now that the file contents are written, rename to the real
+	 * file name; this way we are uniquely sure that the whole
+	 * thing is there.
+	 * conserve @r's value from 'write'
+	 */
+	if (rename(tmp_path, filename) == -1)
+		r = -errno;
+
+error_write:
+	if (r < 0)
+		unlink(tmp_path);
+
+	return r < 0 ? r : 0;
+}
diff --git a/ell/file.h b/ell/file.h
index a93d8e9c..f1ed8790 100644
--- a/ell/file.h
+++ b/ell/file.h
@@ -13,6 +13,7 @@ extern "C" {
 #endif
 
 void *l_file_get_contents(const char *filename, size_t *out_len);
+int l_file_set_contents(const char *filename, const void *data, size_t len);
 
 #ifdef __cplusplus
 }