[Spice-devel] [PATCHv2 08/22] session: set session for migration when connecting

[Marc-André Lureau]

---
 gtk/channel-main.c  | 1 -
 gtk/spice-session.c | 3 +++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/gtk/channel-main.c b/gtk/channel-main.c
index 490d21c..bf7c204 100644
--- a/gtk/channel-main.c
+++ b/gtk/channel-main.c
@@ -2073,7 +2073,6 @@ static gboolean migrate_connect(gpointer data)
     g_return_val_if_fail(c != NULL, FALSE);
     g_return_val_if_fail(mig->session != NULL, FALSE);
 
-    mig->session->priv->for_migration = true;
     spice_session_set_migration_state(mig->session, SPICE_SESSION_MIGRATION_CONNECTING);
 
     if ((c->peer_hdr.major_version == 1) &&
diff --git a/gtk/spice-session.c b/gtk/spice-session.c
index d1e1599..8b9655d 100644
--- a/gtk/spice-session.c
+++ b/gtk/spice-session.c
@@ -2092,6 +2092,9 @@ void spice_session_set_migration_state(SpiceSession *session, SpiceSessionMigrat
 
     SpiceSessionPrivate *s = session->priv;
 
+    if (state == SPICE_SESSION_MIGRATION_CONNECTING)
+        s->for_migration = true;
+
     s->migration_state = state;
     g_coroutine_object_notify(G_OBJECT(session), "migration-state");
 }
-- 
2.1.0