Changeset 23904 in vbox for trunk/src/VBox/Main/ConsoleImplTeleporter.cpp

Timestamp:

Oct 20, 2009 3:01:56 PM (15 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

53705

Message:

VMM,Main: Fixed bugs related to teleporting a paused VM.

File:

• trunk/src/VBox/Main/ConsoleImplTeleporter.cpp (modified) (11 diffs)

trunk/src/VBox/Main/ConsoleImplTeleporter.cpp

@@ -92 +92 @@
     Utf8Str             mstrHostname;
     uint32_t            muPort;
-
-    TeleporterStateSrc(Console *pConsole, PVM pVM, Progress *pProgress)
+    MachineState_T      menmOldMachineState;
+
+    TeleporterStateSrc(Console *pConsole, PVM pVM, Progress *pProgress, MachineState_T enmOldMachineState)
         : TeleporterState(pConsole, pVM, pProgress, true /*fIsSource*/)
         , muPort(UINT32_MAX)
+        , menmOldMachineState(enmOldMachineState)
     {
     }
@@ -110 +112 @@
     PRTTCPSERVER        mhServer;
     PRTTIMERLR          mphTimerLR;
+    bool                mfStartPaused;
     int                 mRc;
 
     TeleporterStateTrg(Console *pConsole, PVM pVM, Progress *pProgress,
-                       IMachine *pMachine, PRTTIMERLR phTimerLR)
+                       IMachine *pMachine, PRTTIMERLR phTimerLR, bool fStartPaused)
         : TeleporterState(pConsole, pVM, pProgress, false /*fIsSource*/)
         , mpMachine(pMachine)
         , mhServer(NULL)
         , mphTimerLR(phTimerLR)
+        , mfStartPaused(false)
         , mRc(VINF_SUCCESS)
     {
@@ -617 +621 @@
     void *pvUser = static_cast<void *>(static_cast<TeleporterState *>(pState));
     vrc = VMR3Teleport(pState->mpVM, &g_teleporterTcpOps, pvUser, teleporterProgressCallback, pvUser);
-    if (vrc)
+    if (RT_FAILURE(vrc))
         return setError(E_FAIL, tr("VMR3Teleport -> %Rrc"), vrc);
 
@@ -623 +627 @@
     if (FAILED(hrc))
         return hrc;
+
+    /*
+     * If we're paused, mention this to the target side.
+     *
+     * Note: This means you have to resume the target manually if you pause it
+     *       during the teleportation.
+     */
+    if (    vrc == VINF_SSM_LIVE_SUSPENDED
+        ||  pState->menmOldMachineState == MachineState_Paused)
+    {
+        hrc = teleporterSrcSubmitCommand(pState, "pause");
+        if (FAILED(hrc))
+            return hrc;
+    }
 
     /*
@@ -686 +704 @@
                 case VMSTATE_RESETTING:
                 case VMSTATE_RESETTING_LS:
-                    pState->mptrConsole->setMachineState(MachineState_Running);
+                    if (pState->menmOldMachineState == MachineState_Running)
+                        pState->mptrConsole->setMachineState(MachineState_Running);
+                    else
+                        pState->mptrConsole->setMachineState(MachineState_Paused);
                     break;
                 case VMSTATE_GURU_MEDITATION:
@@ -777 +798 @@
     CheckComRCReturnRC(hrc);
 
-    TeleporterStateSrc *pState = new TeleporterStateSrc(this, mpVM, ptrProgress);
+    TeleporterStateSrc *pState = new TeleporterStateSrc(this, mpVM, ptrProgress, mMachineState);
     pState->mstrPassword = aPassword;
     pState->mstrHostname = aHostname;
@@ -888 +909 @@
              * Do the job, when it returns we're done.
              */
-            TeleporterStateTrg State(this, pVM, pProgress, pMachine, &hTimerLR);
+            TeleporterStateTrg State(this, pVM, pProgress, pMachine, &hTimerLR, fStartPaused);
             State.mstrPassword      = strPassword;
             State.mhServer          = hServer;
@@ -904 +925 @@
                     if (RT_SUCCESS(vrc))
                     {
-                        if (fStartPaused)
+                        if (State.mfStartPaused)
                             setMachineState(MachineState_Paused);
                         else
@@ -1085 +1106 @@
 
             vrc = teleporterTcpWriteACK(pState);
-            if (RT_FAILURE(vrc))
-                break;
         }
         /** @todo implement config verification and hardware compatability checks. Or
@@ -1096 +1115 @@
             LogRel(("Teleporter: Received cancel command.\n"));
             vrc = VERR_SSM_CANCELLED;
+        }
+        else if (!strcmp(szCmd, "pause"))
+        {
+            pState->mfStartPaused = true;
+            vrc = teleporterTcpWriteACK(pState);
         }
         else if (!strcmp(szCmd, "done"))
@@ -1118 +1142 @@
             break;
         }
+
+        if (RT_FAILURE(vrc))
+            break;
     }