|
 |
28bab8 |
From af5ef029fe96a8703a1de7a60063c473d17d37a4 Mon Sep 17 00:00:00 2001
|
|
|
28bab8 |
From: Martin Kutlak <mkutlak@redhat.com>
|
|
|
28bab8 |
Date: Tue, 21 May 2019 10:45:55 +0200
|
|
|
28bab8 |
Subject: [PATCH] dd: introduce dd_get_env_variable
|
|
|
28bab8 |
|
|
|
28bab8 |
I was thinking about adding 'dd_open_element' function returning file
|
|
|
28bab8 |
descriptor but I thrown this idea because we should try to hide
|
|
|
28bab8 |
implementation details and do not let users to parse elements on their
|
|
|
28bab8 |
own.
|
|
|
28bab8 |
|
|
|
28bab8 |
Cherry-picked from https://github.com/abrt/libreport/commit/a1d7a0397b8251be58783dd5e8665099bf8ce868
|
|
|
28bab8 |
---
|
|
|
28bab8 |
src/include/dump_dir.h | 11 +++++++++++
|
|
|
28bab8 |
src/lib/dump_dir.c | 11 +++++++++++
|
|
|
28bab8 |
tests/dump_dir.at | 31 +++++++++++++++++++++++++++++++
|
|
|
28bab8 |
3 files changed, 53 insertions(+)
|
|
|
28bab8 |
|
|
|
28bab8 |
diff --git a/src/include/dump_dir.h b/src/include/dump_dir.h
|
|
|
28bab8 |
index b617c6c..a7ed431 100644
|
|
|
28bab8 |
--- a/src/include/dump_dir.h
|
|
|
28bab8 |
+++ b/src/include/dump_dir.h
|
|
|
28bab8 |
@@ -118,6 +118,17 @@ int dd_get_next_file(struct dump_dir *dd, char **short_name, char **full_name);
|
|
|
28bab8 |
|
|
|
28bab8 |
char* dd_load_text_ext(const struct dump_dir *dd, const char *name, unsigned flags);
|
|
|
28bab8 |
char* dd_load_text(const struct dump_dir *dd, const char *name);
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+/* Returns value of environment variable with given name.
|
|
|
28bab8 |
+ *
|
|
|
28bab8 |
+ * @param dd Dump directory
|
|
|
28bab8 |
+ * @param name Variables's name
|
|
|
28bab8 |
+ * @param value Return value.
|
|
|
28bab8 |
+ * @return 0 no success, or negative value if an error occurred (-ENOENT if the
|
|
|
28bab8 |
+ * given dd does not support environment variables).
|
|
|
28bab8 |
+ */
|
|
|
28bab8 |
+int dd_get_env_variable(struct dump_dir *dd, const char *name, char **value);
|
|
|
28bab8 |
+
|
|
|
28bab8 |
void dd_save_text(struct dump_dir *dd, const char *name, const char *data);
|
|
|
28bab8 |
void dd_save_binary(struct dump_dir *dd, const char *name, const char *data, unsigned size);
|
|
|
28bab8 |
int dd_copy_file(struct dump_dir *dd, const char *name, const char *source_path);
|
|
|
28bab8 |
diff --git a/src/lib/dump_dir.c b/src/lib/dump_dir.c
|
|
|
28bab8 |
index 5e32c08..0da96cd 100644
|
|
|
28bab8 |
--- a/src/lib/dump_dir.c
|
|
|
28bab8 |
+++ b/src/lib/dump_dir.c
|
|
|
28bab8 |
@@ -1242,6 +1242,17 @@ char* dd_load_text(const struct dump_dir *dd, const char *name)
|
|
|
28bab8 |
return dd_load_text_ext(dd, name, /*flags:*/ 0);
|
|
|
28bab8 |
}
|
|
|
28bab8 |
|
|
|
28bab8 |
+int dd_get_env_variable(struct dump_dir *dd, const char *name, char **value)
|
|
|
28bab8 |
+{
|
|
|
28bab8 |
+ const int fd = openat(dd->dd_fd, FILENAME_ENVIRON, O_RDONLY | O_NOFOLLOW);
|
|
|
28bab8 |
+ if (fd < 0)
|
|
|
28bab8 |
+ return -errno;
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+ const int r = get_env_variable_ext(fd, '\n', name, value);
|
|
|
28bab8 |
+ close(fd);
|
|
|
28bab8 |
+ return r;
|
|
|
28bab8 |
+}
|
|
|
28bab8 |
+
|
|
|
28bab8 |
void dd_save_text(struct dump_dir *dd, const char *name, const char *data)
|
|
|
28bab8 |
{
|
|
|
28bab8 |
if (!dd->locked)
|
|
|
28bab8 |
diff --git a/tests/dump_dir.at b/tests/dump_dir.at
|
|
|
28bab8 |
index 98d564c..12a0c5c 100644
|
|
|
28bab8 |
--- a/tests/dump_dir.at
|
|
|
28bab8 |
+++ b/tests/dump_dir.at
|
|
|
28bab8 |
@@ -615,3 +615,34 @@ int main(int argc, char **argv)
|
|
|
28bab8 |
return 0;
|
|
|
28bab8 |
}
|
|
|
28bab8 |
]])
|
|
|
28bab8 |
+## ------------------- ##
|
|
|
28bab8 |
+## dd_get_env_variable ##
|
|
|
28bab8 |
+## ------------------- ##
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+AT_TESTFUN([dd_get_env_variable],
|
|
|
28bab8 |
+[[
|
|
|
28bab8 |
+#include "testsuite.h"
|
|
|
28bab8 |
+#include "testsuite_tools.h"
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+TS_MAIN
|
|
|
28bab8 |
+{
|
|
|
28bab8 |
+ struct dump_dir *dd = testsuite_dump_dir_create(-1, -1, 0);
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+ dd_create_basic_files(dd, geteuid(), NULL);
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+ char *value = NULL;
|
|
|
28bab8 |
+ TS_ASSERT_SIGNED_EQ(dd_get_env_variable(dd, "FROG", &value), -ENOENT);
|
|
|
28bab8 |
+ TS_ASSERT_PTR_IS_NULL_MESSAGE(value, "Untouched return value on -ENOENT");
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+ dd_save_text(dd, FILENAME_ENVIRON, "FROG=ribbit\nROOSTER=cockle-doodle-doo");
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+ TS_ASSERT_FUNCTION(dd_get_env_variable(dd, "HORSE", &value));
|
|
|
28bab8 |
+ TS_ASSERT_PTR_IS_NULL_MESSAGE(value, "Untouched return value on missing variable");
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+ TS_ASSERT_FUNCTION(dd_get_env_variable(dd, "ROOSTER", &value));
|
|
|
28bab8 |
+ TS_ASSERT_STRING_EQ(value, "cockle-doodle-doo", "Variable ROOSTER loaded from the dump dir");
|
|
|
28bab8 |
+
|
|
|
28bab8 |
+ testsuite_dump_dir_delete(dd);
|
|
|
28bab8 |
+}
|
|
|
28bab8 |
+TS_RETURN_MAIN
|
|
|
28bab8 |
+]])
|
|
|
28bab8 |
--
|
|
|
28bab8 |
2.21.0
|
|
|
28bab8 |
|