COLO: Add 'x-colo-lost-heartbeat' command to trigger failover

[mirror_qemu.git] / migration / colo-comm.c

diff --git a/migration/colo-comm.c b/migration/colo-comm.c
index bf44f764406aa8aebdd7c4535cd766be66f9457e..20b60ec3847981c39a0f2e4bebcdb7e4e6096aad 100644 (file)
--- a/migration/colo-comm.c
+++ b/migration/colo-comm.c
@@ -21,6 +21,17 @@ typedef struct {
 
 static COLOInfo colo_info;
 
+COLOMode get_colo_mode(void)
+{
+    if (migration_in_colo_state()) {
+        return COLO_MODE_PRIMARY;
+    } else if (migration_incoming_in_colo_state()) {
+        return COLO_MODE_SECONDARY;
+    } else {
+        return COLO_MODE_UNKNOWN;
+    }
+}
+
 static void colo_info_pre_save(void *opaque)
 {
     COLOInfo *s = opaque;