Merge branch 'zsv-ldap-2@@2' into 'feature-zsv-5.0.0-vm-support-vtpm-and-secuceboot'

<feature>[kvm]: add hostfile backup agent command

See merge request zstackio/zstack!9497

Author: gitlab

plugin/kvm/src/main/java/org/zstack/kvm/KVMAgentCommands.java

@@ -11,6 +11,7 @@
 import org.zstack.header.host.VmNicRedirectConfig;
 import org.zstack.header.log.NoLogging;
 import org.zstack.header.vm.*;
+import org.zstack.header.vm.additions.VmHostFileBackupJob;
 import org.zstack.header.vm.devices.DeviceAddress;
 import org.zstack.header.vm.devices.VirtualDeviceInfo;
 import org.zstack.kvm.tpm.TpmTO;
@@ -3015,6 +3016,21 @@ public void setHostFiles(List<VmHostFileTO> hostFiles) {
     public static class WriteVmHostFileContentResponse extends AgentResponse {
     }
 
+    public static class BackupVmHostFileCmd extends AgentCommand {
+        private List<VmHostFileBackupJob> vmHostFileBackupJobs;
+
+        public List<VmHostFileBackupJob> getVmHostFileBackupJobs() {
+            return vmHostFileBackupJobs;
+        }
+
+        public void setVmHostFileBackupJobs(List<VmHostFileBackupJob> vmHostFileBackupJobs) {
+            this.vmHostFileBackupJobs = vmHostFileBackupJobs;
+        }
+    }
+
+    public static class BackupVmHostFileResponse extends AgentResponse {
+    }
+
     public static class VmNicInfo {
         private String macAddress;
         private DeviceAddress deviceAddress;

plugin/kvm/src/main/java/org/zstack/kvm/KVMConstant.java

@@ -89,6 +89,7 @@ public interface KVMConstant {
     String FSTRIM_VM_PATH = "/vm/fstrim";
     String READ_VM_HOST_FILE_PATH = "/vm/hostfile/read";
     String WRITE_VM_HOST_FILE_PATH = "/vm/hostfile/write";
+    String BACKUP_VM_HOST_FILE_PATH = "/vm/hostfile/backup";
 
     String ISO_TO = "kvm.isoto";
     String ANSIBLE_PLAYBOOK_NAME = "kvm.py";

plugin/kvm/src/main/java/org/zstack/kvm/efi/BackupVmHostFileOnHypervisorMsg.java

@@ -0,0 +1,29 @@
+package org.zstack.kvm.efi;
+
+import org.zstack.header.host.HostMessage;
+import org.zstack.header.message.NeedReplyMessage;
+import org.zstack.header.vm.additions.VmHostFileBackupJob;
+
+import java.util.List;
+
+public class BackupVmHostFileOnHypervisorMsg extends NeedReplyMessage implements HostMessage {
+    private String hostUuid;
+    private List<VmHostFileBackupJob> vmHostFileBackupJobs;
+
+    @Override
+    public String getHostUuid() {
+        return hostUuid;
+    }
+
+    public void setHostUuid(String hostUuid) {
+        this.hostUuid = hostUuid;
+    }
+
+    public List<VmHostFileBackupJob> getVmHostFileBackupJobs() {
+        return vmHostFileBackupJobs;
+    }
+
+    public void setVmHostFileBackupJobs(List<VmHostFileBackupJob> vmHostFileBackupJobs) {
+        this.vmHostFileBackupJobs = vmHostFileBackupJobs;
+    }
+}

plugin/kvm/src/main/java/org/zstack/kvm/efi/BackupVmHostFileOnHypervisorReply.java

@@ -0,0 +1,6 @@
+package org.zstack.kvm.efi;
+
+import org.zstack.header.message.MessageReply;
+
+public class BackupVmHostFileOnHypervisorReply extends MessageReply {
+}

plugin/kvm/src/main/java/org/zstack/kvm/efi/KvmSecureBootManager.java

@@ -178,6 +178,8 @@ public void handleMessage(Message msg) {
             handle((BackupVmHostFileMsg) msg);
         } else if (msg instanceof RestoreVmHostFileMsg) {
             handle((RestoreVmHostFileMsg) msg);
+        } else if (msg instanceof BackupVmHostFileOnHypervisorMsg) {
+            handle((BackupVmHostFileOnHypervisorMsg) msg);
         } else {
             bus.dealWithUnknownMessage(msg);
         }
@@ -938,4 +940,31 @@ public void fail(ErrorCode errorCode) {
             })
             .start();
     }
+
+    private void handle(BackupVmHostFileOnHypervisorMsg msg) {
+        KvmCommandSender sender = new KvmCommandSender(msg.getHostUuid());
+
+        KVMAgentCommands.BackupVmHostFileCmd cmd = new KVMAgentCommands.BackupVmHostFileCmd();
+        cmd.setVmHostFileBackupJobs(msg.getVmHostFileBackupJobs());
+
+        BackupVmHostFileOnHypervisorReply reply = new BackupVmHostFileOnHypervisorReply();
+        sender.send(cmd, BACKUP_VM_HOST_FILE_PATH, wrapper -> {
+            KVMAgentCommands.BackupVmHostFileResponse rsp = wrapper.getResponse(KVMAgentCommands.BackupVmHostFileResponse.class);
+            return rsp.isSuccess() ? null :
+                    operr("failed to backup vm host file on hypervisor[hostUuid=%s]", msg.getHostUuid())
+                            .withOpaque("host.uuid", msg.getHostUuid())
+                            .withException(rsp.getError());
+        }, new ReturnValueCompletion<KvmResponseWrapper>(msg) {
+            @Override
+            public void success(KvmResponseWrapper wrapper) {
+                bus.reply(msg, reply);
+            }
+
+            @Override
+            public void fail(ErrorCode errorCode) {
+                reply.setError(errorCode);
+                bus.reply(msg, reply);
+            }
+        });
+    }
 }