Changeset 36720 in vbox

Timestamp:

Apr 18, 2011 5:17:07 PM (14 years ago)

Author:

vboxsync

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

71237

Message:

Frontends/VBoxManage: support taking screenshots

Location:

trunk/src/VBox/Frontends/VBoxManage

Files:

• VBoxManageControlVM.cpp (modified) (3 diffs)
• VBoxManageHelp.cpp (modified) (1 diff)

trunk/src/VBox/Frontends/VBoxManage/VBoxManageControlVM.cpp

@@ -5 +5 @@
 
 /*
- * Copyright (C) 2006-2010 Oracle Corporation
+ * Copyright (C) 2006-2011 Oracle Corporation
  *
  * This file is part of VirtualBox Open Source Edition (OSE), as
@@ -36 +36 @@
 #include <iprt/string.h>
 #include <iprt/uuid.h>
+#include <iprt/file.h>
 #include <VBox/log.h>
 
@@ -1025 +1026 @@
             }
         }
+        else if (!strcmp(a->argv[1], "screenshotpng"))
+        {
+            if (a->argc <= 2 || a->argc > 4)
+            {
+                errorSyntax(USAGE_CONTROLVM, "Incorrect number of parameters");
+                rc = E_FAIL;
+                break;
+            }
+            int vrc;
+            uint32_t displayIdx = 0;
+            if (a->argc == 4)
+            {
+                vrc = RTStrToUInt32Ex(a->argv[3], NULL, 0, &displayIdx);
+                if (vrc != VINF_SUCCESS)
+                {
+                    errorArgument("Error parsing display number '%s'", a->argv[3]);
+                    rc = E_FAIL;
+                    break;
+                }
+            }
+            ComPtr<IDisplay> pDisplay;
+            CHECK_ERROR_BREAK(console, COMGETTER(Display)(pDisplay.asOutParam()));
+            ULONG width, height, bpp;
+            CHECK_ERROR_BREAK(pDisplay, GetScreenResolution(displayIdx, &width, &height, &bpp));
+            com::SafeArray<BYTE> saScreenshot;
+            CHECK_ERROR_BREAK(pDisplay, TakeScreenShotPNGToArray(displayIdx, width, height, ComSafeArrayAsOutParam(saScreenshot)));
+            RTFILE pngFile = NIL_RTFILE;
+            vrc = RTFileOpen(&pngFile, a->argv[2], RTFILE_O_OPEN_CREATE | RTFILE_O_WRITE | RTFILE_O_TRUNCATE);
+            if (RT_FAILURE(vrc))
+            {
+                RTMsgError("Failed to create file '%s'. rc=%Rrc", a->argv[2], vrc);
+                rc = E_FAIL;
+                break;
+            }
+            vrc = RTFileWrite(pngFile, saScreenshot.raw(), saScreenshot.size(), NULL);
+            if (RT_FAILURE(vrc))
+            {
+                RTMsgError("Failed to write screenshot to file '%s'. rc=%Rrc", a->argv[2], vrc);
+                rc = E_FAIL;
+            }
+            RTFileClose(pngFile);
+        }
         else
         {

trunk/src/VBox/Frontends/VBoxManage/VBoxManageHelp.cpp

@@ -399 +399 @@
         RTStrmPrintf(pStrm,
                      "                            setvideomodehint <xres> <yres> <bpp> [display] |\n"
+                     "                            screenshotpng <file> [display] |\n"
                      "                            setcredentials <username> <password> <domain>\n"
                      "                                           [--allowlocallogon <yes|no>] |\n"