[xen master] tools/xenstore: read internal state when doing live upgrade

[patchbot@xxxxxxx]

commit c0dc6a3e7c41f86dfcf22262a68ccceaa022d64d
Author:     Juergen Gross <jgross@xxxxxxxx>
AuthorDate: Wed Jan 13 14:00:20 2021 +0100
Commit:     Juergen Gross <jgross@xxxxxxxx>
CommitDate: Thu Jan 21 17:31:43 2021 +0100

    tools/xenstore: read internal state when doing live upgrade
    
    When started due to a live upgrade read the internal state and apply
    it to the data base and internal structures.
    
    Add the main control functions for that.
    
    For now only handle the daemon case.
    
    Signed-off-by: Juergen Gross <jgross@xxxxxxxx>
    Reviewed-by: Paul Durrant <paul@xxxxxxx>
    Acked-by: Julien Grall <jgrall@xxxxxxxxxx>
    Acked-by: Wei Liu <wl@xxxxxxx>
---
 tools/xenstore/xenstored_control.c | 102 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 101 insertions(+), 1 deletion(-)

diff --git a/tools/xenstore/xenstored_control.c 
b/tools/xenstore/xenstored_control.c
index 5151c1448e..3bced9e39e 100644
--- a/tools/xenstore/xenstored_control.c
+++ b/tools/xenstore/xenstored_control.c
@@ -67,6 +67,14 @@ struct live_update {
 
 static struct live_update *lu_status;
 
+struct lu_dump_state {
+       void *buf;
+       unsigned int size;
+#ifndef __MINIOS__
+       int fd;
+#endif
+};
+
 static int lu_destroy(void *data)
 {
 #ifdef __MINIOS__
@@ -318,6 +326,14 @@ static void lu_dump_close(FILE *fp)
 
        fclose(fp);
 }
+
+static void lu_get_dump_state(struct lu_dump_state *state)
+{
+}
+
+static void lu_close_dump_state(struct lu_dump_state *state)
+{
+}
 #else
 static const char *lu_binary(const void *ctx, struct connection *conn,
                             const char *filename)
@@ -373,6 +389,50 @@ static void lu_dump_close(FILE *fp)
 {
        fclose(fp);
 }
+
+static void lu_get_dump_state(struct lu_dump_state *state)
+{
+       char *filename;
+       struct stat statbuf;
+
+       state->size = 0;
+
+       filename = talloc_asprintf(NULL, "%s/state_dump", xs_daemon_rootdir());
+       if (!filename)
+               barf("Allocation failure");
+
+       state->fd = open(filename, O_RDONLY);
+       talloc_free(filename);
+       if (state->fd < 0)
+               return;
+       if (fstat(state->fd, &statbuf) != 0)
+               goto out_close;
+       state->size = statbuf.st_size;
+
+       state->buf = mmap(NULL, state->size, PROT_READ, MAP_PRIVATE,
+                         state->fd, 0);
+       if (state->buf == MAP_FAILED) {
+               state->size = 0;
+               goto out_close;
+       }
+
+       return;
+
+ out_close:
+       close(state->fd);
+}
+
+static void lu_close_dump_state(struct lu_dump_state *state)
+{
+       char *filename;
+
+       munmap(state->buf, state->size);
+       close(state->fd);
+
+       filename = talloc_asprintf(NULL, "%s/state_dump", xs_daemon_rootdir());
+       unlink(filename);
+       talloc_free(filename);
+}
 #endif
 
 static bool lu_check_lu_allowed(void)
@@ -455,7 +515,47 @@ static const char *lu_dump_state(const void *ctx, struct 
connection *conn)
 
 void lu_read_state(void)
 {
-       xprintf("live-update: read state\n");
+       struct lu_dump_state state;
+       struct xs_state_record_header *head;
+       void *ctx = talloc_new(NULL); /* Work context for subfunctions. */
+       struct xs_state_preamble *pre;
+
+       syslog(LOG_INFO, "live-update: read state\n");
+       lu_get_dump_state(&state);
+       if (state.size == 0)
+               barf_perror("No state found after live-update");
+
+       pre = state.buf;
+       if (memcmp(pre->ident, XS_STATE_IDENT, sizeof(pre->ident)) ||
+           pre->version != htobe32(XS_STATE_VERSION) ||
+           pre->flags != XS_STATE_FLAGS)
+               barf("Unknown record identifier");
+       for (head = state.buf + sizeof(*pre);
+            head->type != XS_STATE_TYPE_END &&
+               (void *)head - state.buf < state.size;
+            head = (void *)head + sizeof(*head) + head->length) {
+               switch (head->type) {
+               case XS_STATE_TYPE_GLOBAL:
+                       break;
+               case XS_STATE_TYPE_CONN:
+                       break;
+               case XS_STATE_TYPE_WATCH:
+                       break;
+               case XS_STATE_TYPE_TA:
+                       xprintf("live-update: ignore transaction record\n");
+                       break;
+               case XS_STATE_TYPE_NODE:
+                       break;
+               default:
+                       xprintf("live-update: unknown state record %08x\n",
+                               head->type);
+                       break;
+               }
+       }
+
+       lu_close_dump_state(&state);
+
+       talloc_free(ctx);
 }
 
 static const char *lu_activate_binary(const void *ctx)
--
generated by git-patchbot for /home/xen/git/xen.git#master