ryantimwilson / rpms / systemd

Forked from rpms/systemd 2 months ago
Clone
Zbigniew Jędrzejewski-Szmek e4a83a
From 91a3ba906422127bb12095d1c7d0c7f0cb385588 Mon Sep 17 00:00:00 2001
Zbigniew Jędrzejewski-Szmek e4a83a
From: =?UTF-8?q?Zbigniew=20J=C4=99drzejewski-Szmek?= <zbyszek@in.waw.pl>
Zbigniew Jędrzejewski-Szmek e4a83a
Date: Mon, 2 Mar 2015 10:34:51 -0500
Zbigniew Jędrzejewski-Szmek e4a83a
Subject: [PATCH] journal-remote: fix saving of binary fields
Zbigniew Jędrzejewski-Szmek e4a83a
Zbigniew Jędrzejewski-Szmek e4a83a
Binary fields were not processed properly, and resulting journal files
Zbigniew Jędrzejewski-Szmek e4a83a
were non-conforming, resulting in an error ("Invalid field.") when reading.
Zbigniew Jędrzejewski-Szmek e4a83a
Zbigniew Jędrzejewski-Szmek e4a83a
https://bugs.freedesktop.org/show_bug.cgi?id=89391
Zbigniew Jędrzejewski-Szmek e4a83a
(cherry picked from commit 09d801a82a46df518dd752e40bf13ac404daa2ce)
Zbigniew Jędrzejewski-Szmek e4a83a
---
Zbigniew Jędrzejewski-Szmek e4a83a
 src/journal-remote/journal-remote-parse.c | 31 ++++++++++++++++++-------------
Zbigniew Jędrzejewski-Szmek e4a83a
 src/journal-remote/journal-remote-parse.h |  4 +++-
Zbigniew Jędrzejewski-Szmek e4a83a
 2 files changed, 21 insertions(+), 14 deletions(-)
Zbigniew Jędrzejewski-Szmek e4a83a
Zbigniew Jędrzejewski-Szmek e4a83a
diff --git a/src/journal-remote/journal-remote-parse.c b/src/journal-remote/journal-remote-parse.c
Zbigniew Jędrzejewski-Szmek e4a83a
index d9dea8deb0..afded7e380 100644
Zbigniew Jędrzejewski-Szmek e4a83a
--- a/src/journal-remote/journal-remote-parse.c
Zbigniew Jędrzejewski-Szmek e4a83a
+++ b/src/journal-remote/journal-remote-parse.c
Zbigniew Jędrzejewski-Szmek e4a83a
@@ -344,22 +344,25 @@ int process_data(RemoteSource *source) {
Zbigniew Jędrzejewski-Szmek e4a83a
                    LLLLLLLL0011223344...\n
Zbigniew Jędrzejewski-Szmek e4a83a
                 */
Zbigniew Jędrzejewski-Szmek e4a83a
                 sep = memchr(line, '=', n);
Zbigniew Jędrzejewski-Szmek e4a83a
-                if (sep)
Zbigniew Jędrzejewski-Szmek e4a83a
+                if (sep) {
Zbigniew Jędrzejewski-Szmek e4a83a
                         /* chomp newline */
Zbigniew Jędrzejewski-Szmek e4a83a
                         n--;
Zbigniew Jędrzejewski-Szmek e4a83a
-                else
Zbigniew Jędrzejewski-Szmek e4a83a
+
Zbigniew Jędrzejewski-Szmek e4a83a
+                        r = iovw_put(&source->iovw, line, n);
Zbigniew Jędrzejewski-Szmek e4a83a
+                        if (r < 0)
Zbigniew Jędrzejewski-Szmek e4a83a
+                                return r;
Zbigniew Jędrzejewski-Szmek e4a83a
+                } else {
Zbigniew Jędrzejewski-Szmek e4a83a
                         /* replace \n with = */
Zbigniew Jędrzejewski-Szmek e4a83a
                         line[n-1] = '=';
Zbigniew Jędrzejewski-Szmek e4a83a
-                log_trace("Received: %.*s", (int) n, line);
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
-                r = iovw_put(&source->iovw, line, n);
Zbigniew Jędrzejewski-Szmek e4a83a
-                if (r < 0) {
Zbigniew Jędrzejewski-Szmek e4a83a
-                        log_error("Failed to put line in iovect");
Zbigniew Jędrzejewski-Szmek e4a83a
-                        return r;
Zbigniew Jędrzejewski-Szmek e4a83a
+                        source->field_len = n;
Zbigniew Jędrzejewski-Szmek e4a83a
+                        source->state = STATE_DATA_START;
Zbigniew Jędrzejewski-Szmek e4a83a
+
Zbigniew Jędrzejewski-Szmek e4a83a
+                        /* we cannot put the field in iovec until we have all data */
Zbigniew Jędrzejewski-Szmek e4a83a
                 }
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
-                if (!sep)
Zbigniew Jędrzejewski-Szmek e4a83a
-                        source->state = STATE_DATA_START;
Zbigniew Jędrzejewski-Szmek e4a83a
+                log_trace("Received: %.*s (%s)", (int) n, line, sep ? "text" : "binary");
Zbigniew Jędrzejewski-Szmek e4a83a
+
Zbigniew Jędrzejewski-Szmek e4a83a
                 return 0; /* continue */
Zbigniew Jędrzejewski-Szmek e4a83a
         }
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
@@ -382,6 +385,7 @@ int process_data(RemoteSource *source) {
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
         case STATE_DATA: {
Zbigniew Jędrzejewski-Szmek e4a83a
                 void *data;
Zbigniew Jędrzejewski-Szmek e4a83a
+                char *field;
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
                 assert(source->data_size > 0);
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
@@ -396,11 +400,12 @@ int process_data(RemoteSource *source) {
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
                 assert(data);
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
-                r = iovw_put(&source->iovw, data, source->data_size);
Zbigniew Jędrzejewski-Szmek e4a83a
-                if (r < 0) {
Zbigniew Jędrzejewski-Szmek e4a83a
-                        log_error("failed to put binary buffer in iovect");
Zbigniew Jędrzejewski-Szmek e4a83a
+                field = (char*) data - sizeof(uint64_t) - source->field_len;
Zbigniew Jędrzejewski-Szmek e4a83a
+                memmove(field + sizeof(uint64_t), field, source->field_len);
Zbigniew Jędrzejewski-Szmek e4a83a
+
Zbigniew Jędrzejewski-Szmek e4a83a
+                r = iovw_put(&source->iovw, field + sizeof(uint64_t), source->field_len + source->data_size);
Zbigniew Jędrzejewski-Szmek e4a83a
+                if (r < 0)
Zbigniew Jędrzejewski-Szmek e4a83a
                         return r;
Zbigniew Jędrzejewski-Szmek e4a83a
-                }
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
                 source->state = STATE_DATA_FINISH;
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
diff --git a/src/journal-remote/journal-remote-parse.h b/src/journal-remote/journal-remote-parse.h
Zbigniew Jędrzejewski-Szmek e4a83a
index 8499f4eb82..22db550913 100644
Zbigniew Jędrzejewski-Szmek e4a83a
--- a/src/journal-remote/journal-remote-parse.h
Zbigniew Jędrzejewski-Szmek e4a83a
+++ b/src/journal-remote/journal-remote-parse.h
Zbigniew Jędrzejewski-Szmek e4a83a
@@ -42,7 +42,9 @@ typedef struct RemoteSource {
Zbigniew Jędrzejewski-Szmek e4a83a
         size_t offset;     /* offset to the beginning of live data in the buffer */
Zbigniew Jędrzejewski-Szmek e4a83a
         size_t scanned;    /* number of bytes since the beginning of data without a newline */
Zbigniew Jędrzejewski-Szmek e4a83a
         size_t filled;     /* total number of bytes in the buffer */
Zbigniew Jędrzejewski-Szmek e4a83a
-        size_t data_size;  /* size of the binary data chunk being processed */
Zbigniew Jędrzejewski-Szmek e4a83a
+
Zbigniew Jędrzejewski-Szmek e4a83a
+        size_t field_len;  /* used for binary fields: the field name length */
Zbigniew Jędrzejewski-Szmek e4a83a
+        size_t data_size;  /* and the size of the binary data chunk being processed */
Zbigniew Jędrzejewski-Szmek e4a83a
 
Zbigniew Jędrzejewski-Szmek e4a83a
         struct iovec_wrapper iovw;
Zbigniew Jędrzejewski-Szmek e4a83a