vagrant-windows-hyperv 1.0.1

data/lib/vagrant-windows-hyperv/command/rdp/command.rb

@@ -0,0 +1,22 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+module VagrantPlugins
+  module VagrantHyperV
+    class Command < Vagrant.plugin("2", :command)
+      def self.synopsis
+        "opens a RDP session for a vagrant machine"
+      end
+
+      def execute
+        with_target_vms do |vm|
+          vm.action(:rdp)
+        end
+        # Success, exit status 0
+        0
+      end
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/communication/powershell.rb

@@ -0,0 +1,37 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+module VagrantPlugins
+  module VagrantHyperV
+    module Communicator
+      class PowerShell < Vagrant.plugin("2", :communicator)
+        def initialize(machine)
+          @machine = machine
+        end
+
+        def wait_for_ready(timeout)
+          ready?
+        end
+
+        def ready?
+          # Return True when the guest has enabled WinRM
+          # In this case we can try any remote PowerShell commands to see if
+          # further vagrant can be carried out using this communication
+          if !@winrm_status
+            status = false
+            response = @machine.provider.driver.check_winrm
+            @winrm_status = response["message"] == "Running"
+            raise Errors::WinRMNotReady if !@winrm_status
+          end
+          @winrm_status
+        end
+
+        def test(command, opts=nil)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,89 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+require "json"
+require "#{Vagrant::source_root}/plugins/providers/hyperv/driver"
+
+module VagrantPlugins
+  module VagrantHyperV
+    class Driver < VagrantPlugins::HyperV::Driver
+
+      def initialize(machine)
+        @vm_id = machine.id
+        @machine = machine
+      end
+
+      def ssh_info
+        @ssh_info ||= @machine.ssh_info
+      end
+
+      def remote_credentials
+        @remote_credentials ||= {  guest_ip: ssh_info[:host],
+           username:  ssh_info[:username],
+           password: "vagrant" }
+      end
+
+      def run_remote_ps(command, &block)
+        options = remote_credentials.merge(command: command)
+        script_path = local_script_path('run_in_remote.ps1')
+        # FIXME:  Vagrant's core driver method which executes PowerShell,
+        # need to take an &block so that this piece of code can be avoided.
+        ps_options = []
+        options.each do |key, value|
+          ps_options << "-#{key}"
+          ps_options << "'#{value}'"
+        end
+        ps_options << "-ErrorAction" << "Stop"
+        opts = { notify: [:stdout, :stderr, :stdin] }
+        Vagrant::Util::PowerShell.execute(script_path, *ps_options, **opts, &block)
+      end
+
+      def export_vm_to(path)
+        options = {
+           vm_id: vm_id,
+           path: windows_path(path)
+         }
+        script_path = local_script_path('export_vm.ps1')
+        execute(script_path, options)
+      end
+
+      def upload(from, to)
+        options = {
+          vm_id: vm_id,
+          host_path: windows_path(from),
+          guest_path: windows_path(to)
+        }.merge(remote_credentials)
+        script_path = local_script_path('upload_file.ps1')
+        execute(script_path, options)
+      end
+
+      def check_winrm
+        script_path = local_script_path('check_winrm.ps1')
+        execute(script_path, remote_credentials)
+      end
+
+      def get_host_ip
+        script_path = local_script_path('host_info.ps1')
+        execute(script_path, {})
+      end
+
+      protected
+
+      def local_script_path(path)
+        lib_path = Pathname.new(File.expand_path("../scripts", __FILE__))
+        lib_path.join(path).to_s
+      end
+
+      def windows_path(path)
+        if path
+          path = path.gsub("/","\\")
+          path = "c:#{path}" if path =~ /^\\/
+        end
+        path
+      end
+
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/errors.rb

@@ -0,0 +1,31 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+module VagrantPlugins
+  module VagrantHyperV
+    module Errors
+      class VagrantHyperVError < Vagrant::Errors::VagrantError
+        error_namespace("vagrant_win_hyperv.errors")
+      end
+
+      class SSHNotAvailable < VagrantHyperVError
+        error_key(:ssh_not_available)
+      end
+
+      class RDPNotAvailable < VagrantHyperVError
+        error_key(:rdp_not_available)
+      end
+
+      class InvalidSMBCredentials < VagrantHyperVError
+        error_key(:no_smb_credentials)
+      end
+
+      class WinRMNotReady < VagrantHyperVError
+        error_key(:win_rm_not_available)
+      end
+
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/guest/cap/halt.rb

@@ -0,0 +1,19 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+module VagrantPlugins
+  module VagrantHyperV
+    module Guest
+      module Cap
+        class Halt
+          def self.halt(machine)
+            machine.provider.driver.run_remote_ps("shutdown -a")
+            machine.provider.driver.run_remote_ps("shutdown /s /t 1")
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/guest/windows.rb

@@ -0,0 +1,25 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require "vagrant"
+
+module VagrantPlugins
+  module VagrantHyperV
+    module Guest
+      class Windows < Vagrant.plugin("2", :guest)
+        attr_reader :machin
+
+        def initialize(machine = nil)
+          super(machine) unless machine == nil
+          @machine = machine
+        end
+
+        def halt
+          VagrantHyperV::VagrantPlugins::Guest::Cap::Halt.halt(@machine)
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,32 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+require "#{Vagrant::source_root}/lib/vagrant/action/builtin/provision"
+
+module Vagrant
+  module Action
+    module Builtin
+      class Provision
+
+        alias_method :original_run_provisioner, :run_provisioner
+        # Override this method from core vagrant, here we branch out the provision for windows
+        def run_provisioner(env)
+          if env[:machine].config.vm.guest == :windows
+            case env[:provisioner].class.to_s
+            when "VagrantPlugins::Shell::Provisioner"
+              VagrantPlugins::VagrantHyperV::Provisioner::Shell.new(env).provision_for_windows
+            when "VagrantPlugins::Puppet::Provisioner::Puppet"
+              VagrantPlugins::VagrantHyperV::Provisioner::Puppet.new(env).provision_for_windows
+            when "VagrantPlugins::Chef::Provisioner::ChefSolo"
+              VagrantPlugins::VagrantHyperV::Provisioner::ChefSolo.new(env).provision_for_windows
+            end
+          else
+            original_run_provisioner(env)
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,22 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+module Vagrant
+  class Machine
+
+    alias_method :original_communicate, :communicate
+
+    def communicate
+      unless @communicator
+        if @config.vm.guest == :windows
+          @communicator = VagrantPlugins::VagrantHyperV::Communicator::PowerShell.new(self)
+        else
+         @communicator = original_communicate
+        end
+      end
+      @communicator
+    end
+  end
+end

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

@@ -0,0 +1,55 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+require "#{Vagrant::source_root}/plugins/synced_folders/smb/synced_folder"
+
+module VagrantPlugins
+  module SyncedFolderSMB
+    class SyncedFolder < Vagrant.plugin("2", :synced_folder)
+
+      alias_method :original_enable, :enable
+
+      def enable(machine, folders, nfsopts)
+        response = machine.provider.driver.get_host_ip
+        host_ip = response["host_ip"]
+        if machine.config.vm.guest == :windows
+          folders.each do |id, data|
+            machine.ui.output "#{data[:hostpath]} ==>"
+            machine.ui.output "\\\\#{host_ip}\\#{data[:smb_id]}"
+          end
+        guest_startup_script(machine, folders, host_ip)
+        else
+          original_enable(machine, folders, nfsopts)
+        end
+      end
+
+      def guest_startup_script(machine, folders, host_ip)
+        # Upload a startup script to VM,
+        # This script will authenticate the Network share with the host, and
+        # the guest can access the share path from a RDP session
+        file = Tempfile.new(['vagrant-smb-auth', '.ps1'])
+        begin
+          folders.each do |id, data|
+            smb_map_command = "New-SmbMapping"
+            smb_map_command += " -RemotePath \\\\#{host_ip}\\#{data[:smb_id]}"
+            smb_map_command += " -UserName #{@creds[:username]}"
+            smb_map_command += " -Password #{@creds[:password]}"
+            file.puts(smb_map_command)
+          end
+          file.fsync
+          file.close
+        ensure
+          file.close
+        end
+        machine.provider.driver.upload(file.path.to_s, "c:\\tmp\\vagrant-smb-auth.ps1")
+        # Invoke Remote Schedule task command in VM
+        command = 'schtasks /create /sc ONLOGON /tn vagrant-smb-auth /tr \"powershell c:\tmp\vagrant-smb-auth.ps1\"'
+        machine.provider.driver.run_remote_ps(command)
+      end
+
+    end
+  end
+end
+

data/lib/vagrant-windows-hyperv/monkey_patch/util/powershell.rb

@@ -0,0 +1,37 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+require "#{Vagrant::source_root}/lib/vagrant/util/which"
+require "#{Vagrant::source_root}/lib/vagrant/util/subprocess"
+
+module Vagrant
+  module Util
+    # Executes PowerShell scripts.
+    #
+    # This is primarily a convenience wrapper around Subprocess that
+    # properly sets powershell flags for you.
+    class PowerShell
+      # Monkey patch to fix a bug with Vagrant 1.5.1.
+      # https://github.com/mitchellh/vagrant/issues/3192.
+      # This has been fixed in 1.5.2. by
+      # https://github.com/jyggen/vagrant/commit/d7320399e2497aae9b9c3fa83d94b7210d21bfb5
+      def self.execute(path, *args, **opts, &block)
+        command = [
+          "powershell",
+          "-NoProfile",
+          "-ExecutionPolicy", "Bypass",
+          "&('#{path}')",
+          args
+        ].flatten
+
+        # Append on the options hash since Subprocess doesn't use
+        # Ruby 2.0 style options yet.
+        command << opts
+
+        Subprocess.execute(*command, &block)
+      end
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/plugin.rb

@@ -0,0 +1,85 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+# This is a sanity check to make sure no one is attempting to install
+# this into an early Vagrant version.
+if Vagrant::VERSION < "1.5.0"
+  raise "The Vagrant Hyper-V plugin is only compatible with Vagrant 1.5+"
+end
+
+module VagrantPlugins
+  module VagrantHyperV
+    class Plugin < Vagrant.plugin("2")
+      name "VagrantHyperV"
+      description <<-DESC
+      This plugin installs a provider that allows Vagrant to manage
+      windows guest machines in Hyper-V.
+      DESC
+
+      command "rdp" do
+        require_relative "command/rdp/command"
+        Command
+      end
+
+      provider(:hyperv) do
+        # Setup logging and i18n
+        # setup_logging
+        setup_i18n_hyperv
+
+        # Return the provider
+        require_relative "provider"
+        Provider
+      end
+
+      guest(:windows) do
+        require_relative "guest/windows"
+        Guest::Windows
+      end
+
+      guest_capability(:windows, :halt) do
+        require_relative "guest/cap/halt"
+        Guest::Cap::Halt
+      end
+
+
+
+      # This initializes the internationalization strings.
+      def self.setup_i18n_hyperv
+        I18n.load_path << File.expand_path("locales/en.yml", VagrantHyperV.source_root)
+        I18n.load_path << File.expand_path(
+          "templates/locales/providers_hyperv.yml", Vagrant.source_root)
+        I18n.reload!
+      end
+
+      # This sets up our log level to be whatever VAGRANT_LOG is.
+      def self.setup_logging
+        require "log4r"
+        level = nil
+        begin
+          level = Log4r.const_get(ENV["VAGRANT_LOG"].upcase)
+        rescue NameError
+          # This means that the logging constant wasn't found,
+          # which is fine. We just keep `level` as `nil`. But
+          # we tell the user.
+          level = nil
+        end
+
+        # Some constants, such as "true" resolve to booleans, so the
+        # above error checking doesn't catch it. This will check to make
+        # sure that the log level is an integer, as Log4r requires.
+        level = nil if !level.is_a?(Integer)
+
+        # Set the logging level on all "vagrant" namespaced
+        # logs as long as we have a valid level.
+        if level
+          logger = Log4r::Logger.new("vagrant_hyperv")
+          logger.outputters = Log4r::Outputter.stderr
+          logger.level = level
+          logger = nil
+        end
+      end
+    end
+  end
+end