From: Luca Boccassi <bluca@debian.org>
Date: Mon, 18 Sep 2023 20:15:25 +0000 (+0100)
Subject: serialize: add serialize_item_base64mem()
X-Git-Tag: v255-rc1~250^2~12
X-Git-Url: http://git-history.diyao.me/?a=commitdiff_plain;h=6dae20ec412d5ca3de4a1def0abb5077ee7e2c43;p=systemd%2F.git

serialize: add serialize_item_base64mem()
---

diff --git a/src/shared/serialize.c b/src/shared/serialize.c
index dc282e398e..6133aeb575 100644
--- a/src/shared/serialize.c
+++ b/src/shared/serialize.c
@@ -196,6 +196,29 @@ int serialize_item_hexmem(FILE *f, const char *key, const void *p, size_t l) {
         return 1;
 }
 
+int serialize_item_base64mem(FILE *f, const char *key, const void *p, size_t l) {
+        _cleanup_free_ char *encoded = NULL;
+        ssize_t len;
+        int r;
+
+        assert(f);
+        assert(key);
+        assert(p || l == 0);
+
+        if (l == 0)
+                return 0;
+
+        len = base64mem(p, l, &encoded);
+        if (len <= 0)
+                return log_oom_debug();
+
+        r = serialize_item(f, key, encoded);
+        if (r < 0)
+                return r;
+
+        return 1;
+}
+
 int deserialize_read_line(FILE *f, char **ret) {
         _cleanup_free_ char *line = NULL;
         int r;
diff --git a/src/shared/serialize.h b/src/shared/serialize.h
index 17bc968c04..5eb9be2584 100644
--- a/src/shared/serialize.h
+++ b/src/shared/serialize.h
@@ -13,6 +13,7 @@ int serialize_item(FILE *f, const char *key, const char *value);
 int serialize_item_escaped(FILE *f, const char *key, const char *value);
 int serialize_item_format(FILE *f, const char *key, const char *value, ...) _printf_(3,4);
 int serialize_item_hexmem(FILE *f, const char *key, const void *p, size_t l);
+int serialize_item_base64mem(FILE *f, const char *key, const void *p, size_t l);
 int serialize_fd(FILE *f, FDSet *fds, const char *key, int fd);
 int serialize_usec(FILE *f, const char *key, usec_t usec);
 int serialize_dual_timestamp(FILE *f, const char *key, const dual_timestamp *t);
diff --git a/src/test/test-serialize.c b/src/test/test-serialize.c
index 501f5c6f2e..9e7047f4f0 100644
--- a/src/test/test-serialize.c
+++ b/src/test/test-serialize.c
@@ -208,6 +208,23 @@ TEST(serialize_item_hexmem) {
 
 }
 
+TEST(serialize_item_base64mem) {
+        _cleanup_(unlink_tempfilep) char fn[] = "/tmp/test-serialize.XXXXXX";
+        _cleanup_fclose_ FILE *f = NULL;
+
+        assert_se(fmkostemp_safe(fn, "r+", &f) == 0);
+        log_info("/* %s (%s) */", __func__, fn);
+
+        assert_se(serialize_item_base64mem(f, "a", NULL, 0) == 0);
+        assert_se(serialize_item_base64mem(f, "a", (uint8_t []){0xff, 0xff, 0xff}, sizeof(uint8_t) * 3) == 1);
+
+        rewind(f);
+
+        _cleanup_free_ char *line = NULL;
+        assert_se(read_line(f, LONG_LINE_MAX, &line) > 0);
+        assert_se(streq(line, "a=////"));
+}
+
 static int intro(void) {
         memset(long_string, 'x', sizeof(long_string)-1);
         char_array_0(long_string);