linux-user/host/sparc: Populate host_signal.h

[mirror_qemu.git] / migration / channel.c

diff --git a/migration/channel.c b/migration/channel.c
index 8a783baa0becdbda75991ade8938f6fdb2282a34..c4fc000a1a0a7986402dba577c55045bc86b72d3 100644 (file)
--- a/migration/channel.c
+++ b/migration/channel.c
@@ -18,6 +18,9 @@
 #include "trace.h"
 #include "qapi/error.h"
 #include "io/channel-tls.h"
+#include "io/channel-socket.h"
+#include "qemu/yank.h"
+#include "yank_functions.h"
 
 /**
  * @migration_channel_process_incoming - Create new incoming migration channel
@@ -41,6 +44,7 @@ void migration_channel_process_incoming(QIOChannel *ioc)
                              TYPE_QIO_CHANNEL_TLS)) {
         migration_tls_channel_process_incoming(s, ioc, &local_err);
     } else {
+        migration_ioc_register_yank(ioc);
         migration_ioc_process_incoming(ioc, &local_err);
     }
 
@@ -84,6 +88,8 @@ void migration_channel_connect(MigrationState *s,
         } else {
             QEMUFile *f = qemu_fopen_channel_output(ioc);
 
+            migration_ioc_register_yank(ioc);
+
             qemu_mutex_lock(&s->qemu_file_lock);
             s->to_dst_file = f;
             qemu_mutex_unlock(&s->qemu_file_lock);