ExpandVirtualHardDisk method of the Msvm_ImageManagementService class

uint32 ExpandVirtualHardDisk(
  [in]   string Path,
  [in]   uint64 MaxInternalSize,
  [out]  CIM_ConcreteJob REF Job
);

using System;
using System.Management;
namespace HyperVSamples
{
    public class ExpandVirtualHardDiskClass
    {
        const UInt64 Size1G = 0x40000000;

        public static void ExpandVirtualHardDisk(string path, UInt64 maxInternalSize)
        {
            ManagementScope scope = new ManagementScope(@"root\virtualization", null);
            ManagementObject imageService = Utility.GetServiceObject(scope, "Msvm_ImageManagementService");

            ManagementBaseObject inParams = imageService.GetMethodParameters("ExpandVirtualHardDisk");
            inParams["Path"] = path;
            inParams["MaxInternalSize"] = maxInternalSize * Size1G;
            ManagementBaseObject outParams = imageService.InvokeMethod("ExpandVirtualHardDisk", inParams, null);
            if ((UInt32)outParams["ReturnValue"] == ReturnCode.Started)
            {
                if (Utility.JobCompleted(outParams, scope))
                {
                    Console.WriteLine("{0} was expanded successfully.", inParams["Path"]);
                }
                else
                {
                    Console.WriteLine("Unable to expand {0}", inParams["Path"]);
                }
            }
            
            inParams.Dispose();
            outParams.Dispose();
            imageService.Dispose();
        }

        static void Main(string[] args)
        {
            if (args != null && args.Length != 2)
            {
                Console.WriteLine("Usage: ExpandVHD path, maxInternalSize");
                Console.WriteLine("maxInternalSize: unit in 1 gigabyte");
                return;
            }
            string path = args[0];
            UInt64 maxInternalSize = UInt64.Parse(args[1]);
            ExpandVirtualHardDisk(path, maxInternalSize);
        }
    }
}

option explicit

dim objWMIService
dim managementService
dim fileSystem

const Size1G = &H40000000

const JobStarting = 3
const JobRunning = 4
const JobCompleted = 7
const wmiStarted = 4096


Main()

'-----------------------------------------------------------------
' Main
'-----------------------------------------------------------------
Sub Main()

    dim path, maxInternalSize, objArgs, computer
    
    set fileSystem = Wscript.CreateObject("Scripting.FileSystemObject")

    computer = "."
    set objWMIService = GetObject("winmgmts:\\" & computer & "\root\virtualization")
    set managementService = objWMIService.ExecQuery("Select * from Msvm_ImageManagementService").ItemIndex(0)

    set objArgs = WScript.Arguments
    if WScript.Arguments.Count = 2 then
       path = objArgs.Unnamed.Item(0)
       maxInternalSize = objArgs.Unnamed.Item(1)
    else
       WScript.Echo "usage: cscript ExpandVHD.vbs Path MaxInternalSize"
       WScript.Echo "MaxInternalSize: unit in Gigabytes"
       WScript.Quit(1)
    end if

    if ExpandVirtualHardDisk(path, maxInternalSize) then
        WriteLog "Done"
        WScript.Quit(0)
    else
        WriteLog "ExpandVHD failed"
        WScript.Quit(1)
    end if


End Sub


'-----------------------------------------------------------------
' Execute ExpandVirtualHardDisk
'-----------------------------------------------------------------
Function ExpandVirtualHardDisk(path, maxInternalSize)
    WriteLog Format2("Sub ExpandVirtualHardDisk({0}, {1})",  Path, maxInternalSize)

    dim objInParam, objOutParams
    
    ExpandVirtualHardDisk = false

    set objInParam = managementService.Methods_("ExpandVirtualHardDisk").InParameters.SpawnInstance_()
    objInParam.Path = Path
    objInParam.MaxInternalSize = maxInternalSize * Size1G

    set objOutParams = managementService.ExecMethod_("ExpandVirtualHardDisk", objInParam)

    if (WMIMethodStarted(objOutParams)) then
        if (WMIJobCompleted(objOutParams)) then
            WriteLog Format2("{0} was expected to {1} gigabytes.", Path, MaxInternalSize)
            ExpandVirtualHardDisk = true
        end if
    end if

End Function


'-----------------------------------------------------------------
' Handle wmi return values
'-----------------------------------------------------------------
Function WMIMethodStarted(outParam)
    WriteLog "Function WMIMethodStarted"
    
    dim wmiStatus
    
    WMIMethodStarted = false

    if Not IsNull(outParam) then
        wmiStatus = outParam.ReturnValue

        if  wmiStatus = wmiStarted then
            WMIMethodStarted = true
        else
            WriteLog Format3("Expand VHD {0} to {1} failed. The operation was not able to create ConcreteJob object. Error {2}", Path, MaxInternalSize, wmiStatus)
        end if
   end if

End Function



'-----------------------------------------------------------------
' Handle wmi Job object
'-----------------------------------------------------------------
Function WMIJobCompleted(outParam)
    WriteLog "Function WMIJobCompleted"
    
    dim WMIJob, jobState

    set WMIJob = objWMIService.Get(outParam.Job)

    WMIJobCompleted = true

    jobState = WMIJob.JobState

    while jobState = JobRunning or jobState = JobStarting
        WriteLog Format1("In progress... {0}% completed.",WMIJob.PercentComplete)
        WScript.Sleep(1000)
        set WMIJob = objWMIService.Get(outParam.Job)
        jobState = WMIJob.JobState
    wend

    if (jobState <> JobCompleted) then
        WriteLog Format1("ErrorCode:{0}", WMIJob.ErrorCode)
        WriteLog Format1("ErrorDescription:{0}", WMIJob.ErrorDescription)
        WMIJobCompleted = false
    end if

End Function

'-----------------------------------------------------------------
' Create the console log files.
'-----------------------------------------------------------------
Sub WriteLog(line)
    dim fileStream
    set fileStream = fileSystem.OpenTextFile(".\ExpandVHD.log", 8, true)
    WScript.Echo line
    fileStream.WriteLine line
    fileStream.Close

End Sub

'------------------------------------------------------------------------------
' The string formatting functions to avoid string concatenation.
'------------------------------------------------------------------------------
Function Format2(myString, arg0, arg1)
    Format2 = Format1(myString, arg0)
    Format2 = Replace(Format2, "{1}", arg1)
End Function

'------------------------------------------------------------------------------
' The string formatting functions to avoid string concatenation.
'------------------------------------------------------------------------------
Function Format1(myString, arg0)
    Format1 = Replace(myString, "{0}", arg0)
End Function