a7b9285
From b61e15d1ead74c97c9fa709f6d2868529bae1098 Mon Sep 17 00:00:00 2001
Alon Levy 408bdb5
From: Paolo Bonzini <pbonzini@redhat.com>
Alon Levy 408bdb5
Date: Fri, 22 Feb 2013 17:36:20 +0100
a7b9285
Subject: [PATCH] migration: prepare to access s->state outside critical
a7b9285
 sections
Alon Levy 408bdb5
Alon Levy 408bdb5
Accessing s->state outside the big QEMU lock will simplify a bit the
Alon Levy 408bdb5
locking/unlocking of the iothread lock.
Alon Levy 408bdb5
Alon Levy 408bdb5
We need to keep the lock in migrate_fd_error and migrate_fd_completed,
Alon Levy 408bdb5
however, because they call migrate_fd_cleanup.
Alon Levy 408bdb5
Alon Levy 408bdb5
Reviewed-by: Orit Wasserman <owasserm@redhat.com>
Alon Levy 408bdb5
Reviewed-by: Juan Quintela <quintela@redhat.com>
Alon Levy 408bdb5
Signed-off-by: Paolo Bonzini <pbonzini@redhat.com>
Alon Levy 408bdb5
Signed-off-by: Juan Quintela <quintela@redhat.com>
a7b9285
(cherry picked from commit f4410a5d9926886c36d9fa9fdd969d0469d62724)
Alon Levy 408bdb5
---
Alon Levy 408bdb5
 migration.c | 33 +++++++++++++++++++++------------
Alon Levy 408bdb5
 1 file changed, 21 insertions(+), 12 deletions(-)
Alon Levy 408bdb5
Alon Levy 408bdb5
diff --git a/migration.c b/migration.c
Alon Levy 408bdb5
index c0a9239..e844b2c 100644
Alon Levy 408bdb5
--- a/migration.c
Alon Levy 408bdb5
+++ b/migration.c
Alon Levy 408bdb5
@@ -279,19 +279,25 @@ static void migrate_fd_cleanup(MigrationState *s)
Alon Levy 408bdb5
     notifier_list_notify(&migration_state_notifiers, s);
Alon Levy 408bdb5
 }
Alon Levy 408bdb5
 
Alon Levy 408bdb5
+static void migrate_finish_set_state(MigrationState *s, int new_state)
Alon Levy 408bdb5
+{
Alon Levy 408bdb5
+    if (__sync_val_compare_and_swap(&s->state, MIG_STATE_ACTIVE,
Alon Levy 408bdb5
+                                    new_state) == new_state) {
Alon Levy 408bdb5
+        trace_migrate_set_state(new_state);
Alon Levy 408bdb5
+    }
Alon Levy 408bdb5
+}
Alon Levy 408bdb5
+
Alon Levy 408bdb5
 void migrate_fd_error(MigrationState *s)
Alon Levy 408bdb5
 {
Alon Levy 408bdb5
     DPRINTF("setting error state\n");
Alon Levy 408bdb5
-    s->state = MIG_STATE_ERROR;
Alon Levy 408bdb5
-    trace_migrate_set_state(MIG_STATE_ERROR);
Alon Levy 408bdb5
+    migrate_finish_set_state(s, MIG_STATE_ERROR);
Alon Levy 408bdb5
     migrate_fd_cleanup(s);
Alon Levy 408bdb5
 }
Alon Levy 408bdb5
 
Alon Levy 408bdb5
 static void migrate_fd_completed(MigrationState *s)
Alon Levy 408bdb5
 {
Alon Levy 408bdb5
     DPRINTF("setting completed state\n");
Alon Levy 408bdb5
-    s->state = MIG_STATE_COMPLETED;
Alon Levy 408bdb5
-    trace_migrate_set_state(MIG_STATE_COMPLETED);
Alon Levy 408bdb5
+    migrate_finish_set_state(s, MIG_STATE_COMPLETED);
Alon Levy 408bdb5
     migrate_fd_cleanup(s);
Alon Levy 408bdb5
 }
Alon Levy 408bdb5
 
Alon Levy 408bdb5
@@ -316,13 +322,9 @@ static ssize_t migrate_fd_put_buffer(MigrationState *s, const void *data,
Alon Levy 408bdb5
 
Alon Levy 408bdb5
 static void migrate_fd_cancel(MigrationState *s)
Alon Levy 408bdb5
 {
Alon Levy 408bdb5
-    if (s->state != MIG_STATE_ACTIVE)
Alon Levy 408bdb5
-        return;
Alon Levy 408bdb5
-
Alon Levy 408bdb5
     DPRINTF("cancelling migration\n");
Alon Levy 408bdb5
 
Alon Levy 408bdb5
-    s->state = MIG_STATE_CANCELLED;
Alon Levy 408bdb5
-    trace_migrate_set_state(MIG_STATE_CANCELLED);
Alon Levy 408bdb5
+    migrate_finish_set_state(s, MIG_STATE_CANCELLED);
Alon Levy 408bdb5
     migrate_fd_cleanup(s);
Alon Levy 408bdb5
 }
Alon Levy 408bdb5
 
Alon Levy 408bdb5
@@ -657,12 +659,14 @@ static void *buffered_file_thread(void *opaque)
Alon Levy 408bdb5
     qemu_mutex_lock_iothread();
Alon Levy 408bdb5
     DPRINTF("beginning savevm\n");
Alon Levy 408bdb5
     qemu_savevm_state_begin(s->file, &s->params);
Alon Levy 408bdb5
+    qemu_mutex_unlock_iothread();
Alon Levy 408bdb5
 
Alon Levy 408bdb5
     while (s->state == MIG_STATE_ACTIVE) {
Alon Levy 408bdb5
         int64_t current_time;
Alon Levy 408bdb5
         uint64_t pending_size;
Alon Levy 408bdb5
 
Alon Levy 408bdb5
         if (s->bytes_xfer < s->xfer_limit) {
Alon Levy 408bdb5
+            qemu_mutex_lock_iothread();
Alon Levy 408bdb5
             DPRINTF("iterate\n");
Alon Levy 408bdb5
             pending_size = qemu_savevm_state_pending(s->file, max_size);
Alon Levy 408bdb5
             DPRINTF("pending size %lu max %lu\n", pending_size, max_size);
Alon Levy 408bdb5
@@ -678,8 +682,9 @@ static void *buffered_file_thread(void *opaque)
Alon Levy 408bdb5
                 qemu_savevm_state_complete(s->file);
Alon Levy 408bdb5
                 last_round = true;
Alon Levy 408bdb5
             }
Alon Levy 408bdb5
+            qemu_mutex_unlock_iothread();
Alon Levy 408bdb5
         }
Alon Levy 408bdb5
-        qemu_mutex_unlock_iothread();
Alon Levy 408bdb5
+
Alon Levy 408bdb5
         current_time = qemu_get_clock_ms(rt_clock);
Alon Levy 408bdb5
         if (current_time >= initial_time + BUFFER_DELAY) {
Alon Levy 408bdb5
             uint64_t transferred_bytes = s->bytes_xfer;
Alon Levy 408bdb5
@@ -706,14 +711,18 @@ static void *buffered_file_thread(void *opaque)
Alon Levy 408bdb5
             sleep_time += qemu_get_clock_ms(rt_clock) - current_time;
Alon Levy 408bdb5
         }
Alon Levy 408bdb5
         buffered_flush(s);
Alon Levy 408bdb5
-        qemu_mutex_lock_iothread();
Alon Levy 408bdb5
         if (qemu_file_get_error(s->file)) {
Alon Levy 408bdb5
+            qemu_mutex_lock_iothread();
Alon Levy 408bdb5
             migrate_fd_error(s);
Alon Levy 408bdb5
+            qemu_mutex_unlock_iothread();
Alon Levy 408bdb5
         } else if (last_round && s->buffer_size == 0) {
Alon Levy 408bdb5
+            qemu_mutex_lock_iothread();
Alon Levy 408bdb5
             migrate_fd_completed(s);
Alon Levy 408bdb5
+            qemu_mutex_unlock_iothread();
Alon Levy 408bdb5
         }
Alon Levy 408bdb5
     }
Alon Levy 408bdb5
 
Alon Levy 408bdb5
+    qemu_mutex_lock_iothread();
Alon Levy 408bdb5
     if (s->state == MIG_STATE_COMPLETED) {
Alon Levy 408bdb5
         int64_t end_time = qemu_get_clock_ms(rt_clock);
Alon Levy 408bdb5
         s->total_time = end_time - s->total_time;
Alon Levy 408bdb5
@@ -725,8 +734,8 @@ static void *buffered_file_thread(void *opaque)
Alon Levy 408bdb5
             vm_start();
Alon Levy 408bdb5
         }
Alon Levy 408bdb5
     }
Alon Levy 408bdb5
-
Alon Levy 408bdb5
     qemu_mutex_unlock_iothread();
Alon Levy 408bdb5
+
Alon Levy 408bdb5
     g_free(s->buffer);
Alon Levy 408bdb5
     return NULL;
Alon Levy 408bdb5
 }