vagrant-windows-hyperv 1.0.2 → 1.0.3

data/lib/vagrant-windows-hyperv/monkey_patch/action/provision.rb

@@ -11,7 +11,11 @@ module Vagrant
       class Provision
 
         alias_method :original_run_provisioner, :run_provisioner
-        # Override this method from core vagrant, here we branch out the provision for windows
+        # Override this method from core vagrant, here we branch out the
+        # provision for windows
+        # FIXME:
+        # This should be removed when the core vagrant supports communication
+        # class for windows guests
         def run_provisioner(env)
           if env[:machine].config.vm.guest == :windows
             case env[:provisioner].class.to_s

data/lib/vagrant-windows-hyperv/monkey_patch/config.rb

@@ -0,0 +1,51 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the MIT License.
+#--------------------------------------------------------------------------
+
+require "vagrant"
+
+module VagrantPlugins
+  module HyperV
+    class Config < Vagrant.plugin("2", :config)
+      # The timeout to wait for an IP address when booting the machine,
+      # in seconds.
+      #
+      # @return [Integer]
+      attr_accessor :ip_address_timeout
+      attr_reader :customizations
+
+      def initialize
+        @ip_address_timeout = UNSET_VALUE
+        @customizations   = []
+      end
+
+      def customize(*command)
+        @customizations ||= []
+        event   = command.first.is_a?(String) ? command.shift : "pre-boot"
+        command = command[0]
+        options = command[1]
+        @customizations << [event, command, options]
+      end
+
+      def finalize!
+        if @ip_address_timeout == UNSET_VALUE
+          @ip_address_timeout = 120
+        end
+      end
+
+      def validate(machine)
+        errors = _detected_errors
+
+        valid_events = ["pre-boot"]
+        @customizations.each do |event, _|
+          if !valid_events.include?(event)
+            errors << "Invalid custom event #{event} use pre-boot"
+          end
+        end
+
+        { "Hyper-V" => errors }
+      end
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/monkey_patch/driver.rb

@@ -0,0 +1,52 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+require "#{Vagrant::source_root}/plugins/providers/hyperv/driver"
+
+module VagrantPlugins
+  module HyperV
+    class Driver
+
+      # Override the existing import method to Override the import_vm.ps1 script
+      def import(options)
+        load_path = Pathname.new(File.expand_path("../scripts", __FILE__))
+        script_path = load_path.join('import_vm.ps1').to_s
+        execute(script_path, options)
+      end
+
+      # New methods for network customization
+      def create_network_switch(options)
+        load_path = Pathname.new(File.expand_path("../scripts", __FILE__))
+        script_path = load_path.join('create_switch.ps1')
+        execute(script_path, options)
+      end
+
+      def list_net_adapters
+        load_path = Pathname.new(File.expand_path("../scripts", __FILE__))
+        script_path = load_path.join('get_adapters.ps1')
+        execute(script_path, {})
+      end
+
+      def find_vm_switch_name
+        load_path = Pathname.new(File.expand_path("../scripts", __FILE__))
+        script_path = load_path.join('find_vm_switch_name.ps1')
+        execute(script_path, {vm_id: vm_id})
+      end
+
+      def add_swith_to_vm(options)
+        load_path = Pathname.new(File.expand_path("../scripts", __FILE__))
+        script_path = load_path.join('add_switch_to_vm.ps1')
+        execute(script_path, options)
+      end
+
+      def switch_exist(options)
+        load_path = Pathname.new(File.expand_path("../scripts", __FILE__))
+        script_path = load_path.join('switch_exist.ps1')
+        execute(script_path, options)
+      end
+
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/monkey_patch/machine.rb

@@ -6,14 +6,14 @@
 module Vagrant
   class Machine
 
-    alias_method :original_communicate, :communicate
+    ssh_communicate = instance_method(:communicate)
 
-    def communicate
+    define_method(:communicate) do
       unless @communicator
         if @config.vm.guest == :windows
           @communicator = VagrantPlugins::VagrantHyperV::Communicator::PowerShell.new(self)
         else
-         @communicator = original_communicate
+         @communicator = ssh_communicate.bind(self).()
         end
       end
       @communicator

data/lib/vagrant-windows-hyperv/monkey_patch/plugins/synced_folders/smb/synced_folders.rb

@@ -11,6 +11,9 @@ module VagrantPlugins
 
       alias_method :original_enable, :enable
 
+      # FIXME:
+      # This is a workaround to enable the guest VM to access the SMB share
+      # set in the host.
       def enable(machine, folders, nfsopts)
         response = machine.provider.driver.get_host_ip
         host_ip = response["host_ip"]

data/lib/vagrant-windows-hyperv/monkey_patch/scripts/add_switch_to_vm.ps1

@@ -0,0 +1,38 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the MIT License.
+#--------------------------------------------------------------------------
+
+param (
+  [Parameter(Mandatory=$true)]
+  [string]$type,
+  [Parameter(Mandatory=$true)]
+  [string]$name,
+  [Parameter(Mandatory=$true)]
+  [string]$vm_id,
+  [Parameter(Mandatory=$false)]
+  [string]$adapter
+ )
+
+# Include the following modules
+$Dir = Split-Path $script:MyInvocation.MyCommand.Path
+. ([System.IO.Path]::Combine($Dir, "utils\write_messages.ps1"))
+
+try {
+
+   $vm = Get-VM -Id $vm_id -ErrorAction "stop"
+    Get-VMSwitch "$name" | Where-Object { $_.SwitchType -eq "$type" } `
+    | Connect-VMNetworkAdapter -VMName $vm.Name
+
+   $resultHash = @{
+     message = "OK"
+   }
+   Write-Output-Message $(ConvertTo-JSON $resultHash)
+ }
+ catch {
+  $errortHash = @{
+    type = "PowerShellError"
+    error = "$_"
+  }
+  Write-Error-Message $(ConvertTo-JSON $errortHash)
+ }

data/lib/vagrant-windows-hyperv/monkey_patch/scripts/create_switch.ps1

@@ -0,0 +1,75 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the MIT License.
+#--------------------------------------------------------------------------
+
+param (
+    [Parameter(Mandatory=$true)]
+    [string]$type,
+    [Parameter(Mandatory=$true)]
+    [string]$name,
+    [Parameter(Mandatory=$true)]
+    [string]$vm_id,
+    [Parameter(Mandatory=$true)]
+    [string]$adapter
+ )
+
+# Include the following modules
+$Dir = Split-Path $script:MyInvocation.MyCommand.Path
+. ([System.IO.Path]::Combine($Dir, "utils\write_messages.ps1"))
+
+try {
+
+  # Find the current IP address of the host. This will be used later to test the
+  # network connectivity upon creating a new switch to a network adapter
+  $ip = (Get-WmiObject -class win32_NetworkAdapterConfiguration -Filter 'ipenabled = "true"').ipaddress[0]
+
+  $max_attempts = 5
+  $operation_pass = $false
+  do {
+    try {
+      if ($type -eq "external") {
+        New-VMSwitch -Name "$name" -NetAdapterName $adapter -ErrorAction "stop"
+      }
+      $operation_pass = $true
+    } catch {
+      $max_attempts = $max_attempts - 1
+      sleep 5
+    }
+  }
+  while (!$operation_pass -and $max_attempts -gt 0)
+
+   # On creating a new switch / a new network adapter, there are chances that
+   # the network may get disconnected for a while.
+
+   # Keep checking for network availability before exiting this script
+   $max_attempts = 10
+   $network_available = $false
+   do {
+     try {
+      $ping_response = Test-Connection "$ip" -ErrorAction "stop"
+      $network_available = $true
+     } catch {
+        $max_attempts = $max_attempts - 1
+        sleep 5
+     }
+   }
+   while (!$network_available -and $max_attempts -gt 0)
+   if (-not $network_available) {
+    $resptHash = @{
+      message = "Network down"
+    }
+   } else {
+    $resptHash = @{
+      message = "Success"
+    }
+   }
+   Write-Output-Message $(ConvertTo-JSON $resptHash)
+   return
+} catch {
+    $errortHash = @{
+      type = "PowerShellError"
+      error = "$_"
+    }
+    Write-Error-Message $(ConvertTo-JSON $errortHash)
+}

data/lib/vagrant-windows-hyperv/monkey_patch/scripts/find_vm_switch_name.ps1

@@ -0,0 +1,29 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the MIT License.
+#--------------------------------------------------------------------------
+
+param (
+    [Parameter(Mandatory=$true)]
+    [string]$vm_id
+ )
+
+# Include the following modules
+$Dir = Split-Path $script:MyInvocation.MyCommand.Path
+. ([System.IO.Path]::Combine($Dir, "utils\write_messages.ps1"))
+
+
+try {
+  $vm = Get-VM -Id $vm_id -ErrorAction "stop"
+  $network_adapter = Get-VMNetworkAdapter -vm $vm
+  $resultHash = @{}
+  $resultHash["switch_name"] = $network_adapter.SwitchName
+  $resultHash["network_adapter"] = $network_adapter.Name
+  Write-Output-Message $(ConvertTo-JSON $resultHash)
+} catch [Microsoft.HyperV.PowerShell.VirtualizationOperationFailedException] {
+  $errortHash = @{
+    type = "PowerShellError"
+    error = "$_"
+  }
+  Write-Error-Message $(ConvertTo-JSON $errortHash)
+}

data/lib/vagrant-windows-hyperv/monkey_patch/scripts/get_adapters.ps1

@@ -0,0 +1,17 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the MIT License.
+#--------------------------------------------------------------------------
+# Include the following modules
+$Dir = Split-Path $script:MyInvocation.MyCommand.Path
+. ([System.IO.Path]::Combine($Dir, "utils\write_messages.ps1"))
+
+$adapters = @()
+
+  Get-NetAdapter `
+    | Select-Object Name,InterfaceDescription,Status `
+    | Where-Object {$_.Status-eq "up"}  `
+    | ForEach-Object  -Process {
+      $adapters += $_
+    }
+Write-Output-Message $(ConvertTo-JSON $adapters)

data/lib/vagrant-windows-hyperv/monkey_patch/scripts/import_vm.ps1

@@ -0,0 +1,136 @@
+Param(
+    [Parameter(Mandatory=$true)]
+    [string]$vm_xml_config,
+    [Parameter(Mandatory=$true)]
+    [string]$vhdx_path
+)
+
+# Include the following modules
+$Dir = Split-Path $script:MyInvocation.MyCommand.Path
+. ([System.IO.Path]::Combine($Dir, "utils\write_messages.ps1"))
+
+[xml]$vmconfig = Get-Content -Path  $vm_xml_config
+
+$vm_name = $vmconfig.configuration.properties.name.'#text'
+$processors = $vmconfig.configuration.settings.processors.count.'#text'
+
+function GetUniqueName($name) {
+    Get-VM | ForEach-Object -Process {
+        if ($name -eq $_.Name) {
+            $name =  $name + "_1"
+        }
+    }
+    return $name
+}
+
+do {
+    $name = $vm_name
+    $vm_name = GetUniqueName $name
+} while ($vm_name -ne $name)
+
+$memory = (Select-Xml -xml $vmconfig -XPath "//memory").node.Bank
+if ($memory.dynamic_memory_enabled."#text" -eq "True") {
+    $dynamicmemory = $True
+}
+else {
+    $dynamicmemory = $False
+}
+
+# Memory values need to be in bytes
+$MemoryMaximumBytes = ($memory.limit."#text" -as [int]) * 1MB
+$MemoryStartupBytes = ($memory.size."#text" -as [int]) * 1MB
+$MemoryMinimumBytes = ($memory.reservation."#text" -as [int]) * 1MB
+
+# Determine boot device
+Switch ((Select-Xml -xml $vmconfig -XPath "//boot").node.device0."#text") {
+    "Floppy"    { $bootdevice = "floppy" }
+    "HardDrive" { $bootdevice = "IDE" }
+    "Optical"   { $bootdevice = "CD" }
+    "Network"   { $bootdevice = "LegacyNetworkAdapter" }
+    "Default"   { $bootdevice = "IDE" }
+} #switch
+
+# Define a hash map of parameter values for New-VM
+
+$vm_params = @{
+    Name = $vm_name
+    NoVHD = $True
+    MemoryStartupBytes = $MemoryStartupBytes
+    BootDevice = $bootdevice
+    ErrorAction = "Stop"
+}
+
+# Create the VM using the values in the hash map
+
+$vm = New-VM @vm_params
+
+$notes = (Select-Xml -xml $vmconfig -XPath "//notes").node.'#text'
+
+# Set-VM parameters to configure new VM with old values
+
+$more_vm_params = @{
+    ProcessorCount = $processors
+    MemoryStartupBytes = $MemoryStartupBytes
+}
+
+If ($dynamicmemory) {
+    $more_vm_params.Add("DynamicMemory",$True)
+    $more_vm_params.Add("MemoryMinimumBytes",$MemoryMinimumBytes)
+    $more_vm_params.Add("MemoryMaximumBytes", $MemoryMaximumBytes)
+} else {
+    $more_vm_params.Add("StaticMemory",$True)
+}
+
+if ($notes) {
+    $more_vm_params.Add("Notes",$notes)
+}
+
+# Set the values on the VM
+$vm | Set-VM @more_vm_params -Passthru
+
+# Add drives to the virtual machine
+$controllers = Select-Xml -xml $vmconfig -xpath "//*[starts-with(name(.),'controller')]"
+
+# A regular expression pattern to pull the number from controllers
+[regex]$rx="\d"
+
+foreach ($controller in $controllers) {
+    $node = $controller.Node
+
+    # Check for SCSI
+    if ($node.ParentNode.ChannelInstanceGuid) {
+        $ControllerType = "SCSI"
+    } else {
+        $ControllerType = "IDE"
+    }
+
+    $drives = $node.ChildNodes | where {$_.pathname."#text"}
+    foreach ($drive in $drives) {
+        #if drive type is ISO then set DVD Drive accordingly
+        $driveType = $drive.type."#text"
+
+        $addDriveParam = @{
+            ControllerNumber = $rx.Match($controller.node.name).value
+            Path = $vhdx_path
+        }
+
+        if ($drive.pool_id."#text") {
+            $ResourcePoolName = $drive.pool_id."#text"
+            $addDriveParam.Add("ResourcePoolname",$ResourcePoolName)
+        }
+
+        if ($drivetype -eq 'VHD') {
+            $addDriveParam.add("ControllerType",$ControllerType)
+            $vm | Add-VMHardDiskDrive @AddDriveparam
+        }
+    }
+}
+
+$vm_id = (Get-VM $vm_name).id.guid
+$resultHash = @{
+    name = $vm_name
+    id = $vm_id
+}
+
+$result = ConvertTo-Json $resultHash
+Write-Output-Message $result