117 lines
4.0 KiB
Diff
117 lines
4.0 KiB
Diff
From 1e4f0fdb8d684007fa0c9747181a8ca70da5eaae Mon Sep 17 00:00:00 2001
|
|
From: Antonio Alvarez Feijoo <antonio.feijoo@suse.com>
|
|
Date: Mon, 10 Apr 2023 15:18:33 +0200
|
|
Subject: [PATCH] generator: add generator_open_unit_file_full to allow
|
|
creating temporary units
|
|
|
|
This function is like `generator_open_unit_file`, but if `ret_temp_path` is
|
|
passed, a temporary unit is created instead.
|
|
|
|
(cherry picked from commit 8a84e0d7960b7970c16a4efd4c5b0b26c810d22d)
|
|
|
|
Related: RHEL-27512
|
|
---
|
|
src/shared/generator.c | 46 ++++++++++++++++++++++++++++--------------
|
|
src/shared/generator.h | 10 ++++-----
|
|
2 files changed, 36 insertions(+), 20 deletions(-)
|
|
|
|
diff --git a/src/shared/generator.c b/src/shared/generator.c
|
|
index 284e5fc580..29de8ada6b 100644
|
|
--- a/src/shared/generator.c
|
|
+++ b/src/shared/generator.c
|
|
@@ -20,42 +20,58 @@
|
|
#include "specifier.h"
|
|
#include "string-util.h"
|
|
#include "time-util.h"
|
|
+#include "tmpfile-util.h"
|
|
#include "unit-name.h"
|
|
#include "util.h"
|
|
|
|
-int generator_open_unit_file(
|
|
+int generator_open_unit_file_full(
|
|
const char *dir,
|
|
const char *source,
|
|
const char *fn,
|
|
- FILE **ret) {
|
|
+ FILE **ret_file,
|
|
+ char **ret_temp_path) {
|
|
|
|
_cleanup_free_ char *p = NULL;
|
|
FILE *f;
|
|
int r;
|
|
|
|
assert(dir);
|
|
- assert(fn);
|
|
- assert(ret);
|
|
+ assert(ret_file);
|
|
|
|
- p = path_join(dir, fn);
|
|
- if (!p)
|
|
- return log_oom();
|
|
+ /* If <ret_temp_path> is specified, it creates a temporary unit file and also returns its
|
|
+ * temporary path. */
|
|
|
|
- r = fopen_unlocked(p, "wxe", &f);
|
|
- if (r < 0) {
|
|
- if (source && r == -EEXIST)
|
|
- return log_error_errno(r,
|
|
- "Failed to create unit file '%s', as it already exists. Duplicate entry in '%s'?",
|
|
- p, source);
|
|
+ if (ret_temp_path) {
|
|
+ r = fopen_temporary(dir, &f, &p);
|
|
+ if (r < 0)
|
|
+ return log_error_errno(r, "Failed to create temporary unit file in '%s': %m", dir);
|
|
+
|
|
+ (void) fchmod(fileno(f), 0644);
|
|
|
|
- return log_error_errno(r, "Failed to create unit file '%s': %m", p);
|
|
+ *ret_temp_path = TAKE_PTR(p);
|
|
+ } else {
|
|
+ assert(fn);
|
|
+
|
|
+ p = path_join(dir, fn);
|
|
+ if (!p)
|
|
+ return log_oom();
|
|
+
|
|
+ r = fopen_unlocked(p, "wxe", &f);
|
|
+ if (r < 0) {
|
|
+ if (source && r == -EEXIST)
|
|
+ return log_error_errno(r,
|
|
+ "Failed to create unit file '%s', as it already exists. Duplicate entry in '%s'?",
|
|
+ p, source);
|
|
+
|
|
+ return log_error_errno(r, "Failed to create unit file '%s': %m", p);
|
|
+ }
|
|
}
|
|
|
|
fprintf(f,
|
|
"# Automatically generated by %s\n\n",
|
|
program_invocation_short_name);
|
|
|
|
- *ret = f;
|
|
+ *ret_file = f;
|
|
return 0;
|
|
}
|
|
|
|
diff --git a/src/shared/generator.h b/src/shared/generator.h
|
|
index 111900fd45..d97d6edc67 100644
|
|
--- a/src/shared/generator.h
|
|
+++ b/src/shared/generator.h
|
|
@@ -6,11 +6,11 @@
|
|
#include "macro.h"
|
|
#include "main-func.h"
|
|
|
|
-int generator_open_unit_file(
|
|
- const char *dest,
|
|
- const char *source,
|
|
- const char *name,
|
|
- FILE **file);
|
|
+int generator_open_unit_file_full(const char *dest, const char *source, const char *name, FILE **ret_file, char **ret_temp_path);
|
|
+
|
|
+static inline int generator_open_unit_file(const char *dest, const char *source, const char *name, FILE **ret_file) {
|
|
+ return generator_open_unit_file_full(dest, source, name, ret_file, NULL);
|
|
+}
|
|
|
|
int generator_add_symlink_full(const char *dir, const char *dst, const char *dep_type, const char *src, const char *instance);
|
|
|