archive writer: Add function to add a file from memory

Signed-off-by: Michael Tremer <michael.tremer@ipfire.org>

diff --git a/src/pakfire/archive_writer.c b/src/pakfire/archive_writer.c
index 182977674bdb31364ed9eb0a30fbbdeb58ecfaac..1ac2e3ae2c0187116cc51d747e6a87b874317c13 100644 (file)
--- a/src/pakfire/archive_writer.c
+++ b/src/pakfire/archive_writer.c
@@ -514,8 +514,82 @@ int pakfire_archive_writer_write_files(
        if (r < 0)
                goto ERROR;
 
+       // Flush all buffers to disk
+       fflush(self->f);
+
 ERROR:
        pakfire_progress_finish(self->progress);
 
        return r;
 }
+
+/*
+ * Creates a new file and writes it to the archive.
+ */
+int pakfire_archive_writer_create_file(struct pakfire_archive_writer* self,
+               const char* filename, mode_t mode, const char* payload, const size_t length) {
+       struct archive_entry* entry = NULL;
+       int r;
+
+       // Create a new file entry
+       entry = archive_entry_new();
+       if (!entry) {
+               r = -errno;
+               goto ERROR;
+       }
+
+       // Set filename
+       archive_entry_set_pathname(entry, filename);
+
+       // This is a regular file
+    archive_entry_set_filetype(entry, AE_IFREG);
+    archive_entry_set_perm(entry, mode);
+
+       // Set length
+       archive_entry_set_size(entry, length);
+
+       // Set ownership
+       archive_entry_set_uname(entry, "root");
+       archive_entry_set_uid(entry, 0);
+       archive_entry_set_gname(entry, "root");
+       archive_entry_set_gid(entry, 0);
+
+       // Set times
+       archive_entry_set_birthtime(entry, self->time_created, 0);
+       archive_entry_set_ctime(entry, self->time_created, 0);
+       archive_entry_set_mtime(entry, self->time_created, 0);
+       archive_entry_set_atime(entry, self->time_created, 0);
+
+       // Write the header
+       r = archive_write_header(self->archive, entry);
+       if (r) {
+               ERROR(self->ctx, "Error writing header: %s\n",
+                       archive_error_string(self->archive));
+               r = -EINVAL;
+               goto ERROR;
+       }
+
+       // Write content
+       r = archive_write_data(self->archive, payload, length);
+       if (r < 0) {
+               ERROR(self->ctx, "Error writing data: %s\n",
+                       archive_error_string(self->archive));
+               r = -EINVAL;
+               goto ERROR;
+       }
+
+       // Write trailer
+       r = archive_write_finish_entry(self->archive);
+       if (r) {
+               ERROR(self->ctx, "Failed to write the trailer: %s\n",
+                       archive_error_string(self->archive));
+               r = -EINVAL;
+               goto ERROR;
+       }
+
+ERROR:
+       if (entry)
+               archive_entry_free(entry);
+
+       return r;
+}

diff --git a/src/pakfire/archive_writer.h b/src/pakfire/archive_writer.h
index 6a6aaad1e200e32f568845ff31d754242ff27b5f..92289684785968a69a1fbdaa4e13179b29985dea 100644 (file)
--- a/src/pakfire/archive_writer.h
+++ b/src/pakfire/archive_writer.h
@@ -44,4 +44,7 @@ int pakfire_archive_writer_set_title(struct pakfire_archive_writer* self,
 int pakfire_archive_writer_write_files(
        struct pakfire_archive_writer* self, struct pakfire_filelist* files);
 
+int pakfire_archive_writer_create_file(struct pakfire_archive_writer* self,
+       const char* filename, mode_t mode, const char* payload, const size_t length);
+
 #endif /* PAKFIRE_ARCHIVE_WRITER_H */