knife-cloud 1.0.0.rc.0

data/lib/chef/knife/cloud/chefbootstrap/bootstrap_options.rb

@@ -0,0 +1,191 @@
+# Author:: Kaustubh Deorukhkar (<kaustubh@clogeny.com>)
+# Author:: Prabhu Das (<prabhu.das@clogeny.com>)
+#
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+require 'chef/knife/winrm_base'
+require 'chef/knife/core/bootstrap_context'
+require 'net/ssh/multi'
+
+class Chef
+  class Knife
+    class Cloud
+      # Ideally chef/knife/bootstap should expose this as module.
+      module BootstrapOptions
+
+        def self.included(includer)
+          includer.class_eval do
+
+            deps do
+              require 'chef/json_compat'
+              require 'tempfile'
+              require 'highline'
+              require 'net/ssh'
+              require 'chef/knife/ssh'
+              Chef::Knife::Ssh.load_deps
+            end
+
+            include Chef::Knife::WinrmBase
+
+            option :ssh_user,
+              :short => "-x USERNAME",
+              :long => "--ssh-user USERNAME",
+              :description => "The ssh username",
+              :default => "root"
+
+            option :ssh_password,
+              :short => "-P PASSWORD",
+              :long => "--ssh-password PASSWORD",
+              :description => "The ssh password"
+
+            option :ssh_port,
+              :short => "-p PORT",
+              :long => "--ssh-port PORT",
+              :description => "The ssh port",
+              :proc => Proc.new { |key| Chef::Config[:knife][:ssh_port] = key },
+              :default => "22"
+
+            option :ssh_gateway,
+              :long => "--ssh-gateway GATEWAY",
+              :description => "The ssh gateway server. Any proxies configured in your ssh config are automatically used by default.",
+              :proc => Proc.new { |key| Chef::Config[:knife][:ssh_gateway] = key }
+
+            option :ssh_gateway_identity,
+              :long => "--ssh-gateway-identity IDENTITY_FILE",
+              :description => "The private key for ssh gateway server",
+              :proc => Proc.new { |key| Chef::Config[:knife][:ssh_gateway_identity] = key }
+
+            option :forward_agent,
+              :long => "--forward-agent",
+              :description => "Enable SSH agent forwarding",
+              :boolean => true
+
+            option :identity_file,
+              :short => "-i IDENTITY_FILE",
+              :long => "--identity-file IDENTITY_FILE",
+              :description => "The SSH identity file used for authentication"
+
+            option :chef_node_name,
+              :short => "-N NAME",
+              :long => "--node-name NAME",
+              :description => "The Chef node name for your new node"
+
+            option :prerelease,
+              :long => "--prerelease",
+              :description => "Install the pre-release chef gems"
+
+            option :bootstrap_version,
+              :long => "--bootstrap-version VERSION",
+              :description => "The version of Chef to install",
+              :proc => lambda { |v| Chef::Config[:knife][:bootstrap_version] = v }
+
+            option :bootstrap_proxy,
+              :long => "--bootstrap-proxy PROXY_URL",
+              :description => "The proxy server for the node being bootstrapped",
+              :proc => Proc.new { |p| Chef::Config[:knife][:bootstrap_proxy] = p }
+
+            option :bootstrap_no_proxy,
+              :long => "--bootstrap-no-proxy [NO_PROXY_URL|NO_PROXY_IP]",
+              :description => "Do not proxy locations for the node being bootstrapped; this option is used internally by Opscode",
+              :proc => Proc.new { |np| Chef::Config[:knife][:bootstrap_no_proxy] = np }
+
+            option :distro,
+              :short => "-d DISTRO",
+              :long => "--distro DISTRO",
+              :description => "Bootstrap a distro using a template"
+
+            option :use_sudo,
+              :long => "--sudo",
+              :description => "Execute the bootstrap via sudo",
+              :boolean => true
+
+            option :use_sudo_password,
+              :long => "--use-sudo-password",
+              :description => "Execute the bootstrap via sudo with password",
+              :boolean => false
+
+            option :template_file,
+              :long => "--template-file TEMPLATE",
+              :description => "Full path to location of template to use",
+              :proc => Proc.new { |t| Chef::Config[:knife][:template_file] = t },
+              :default => false
+
+            option :run_list,
+              :short => "-r RUN_LIST",
+              :long => "--run-list RUN_LIST",
+              :description => "Comma separated list of roles/recipes to apply",
+              :proc => lambda { |o| o.split(/[\s,]+/) },
+              :default => []
+
+            option :first_boot_attributes,
+              :short => "-j JSON_ATTRIBS",
+              :long => "--json-attributes",
+              :description => "A JSON string to be added to the first run of chef-client",
+              :proc => lambda { |o| JSON.parse(o) },
+              :default => {}
+
+            option :host_key_verify,
+              :long => "--[no-]host-key-verify",
+              :description => "Verify host key, enabled by default.",
+              :boolean => true,
+              :default => true
+
+            option :hint,
+              :long => "--hint HINT_NAME[=HINT_FILE]",
+              :description => "Specify Ohai Hint to be set on the bootstrap target.  Use multiple --hint options to specify multiple hints.",
+              :proc => Proc.new { |h|
+                Chef::Config[:knife][:hints] ||= Hash.new
+                name, path = h.split("=")
+                Chef::Config[:knife][:hints][name] = path ? JSON.parse(::File.read(path)) : Hash.new  }
+
+            option :secret,
+              :short => "-s SECRET",
+              :long  => "--secret ",
+              :description => "The secret key to use to encrypt data bag item values"
+
+            option :secret_file,
+              :long => "--secret-file SECRET_FILE",
+              :description => "A file containing the secret key to use to encrypt data bag item values"
+
+            option :bootstrap_url,
+              :long        => "--bootstrap-url URL",
+              :description => "URL to a custom installation script",
+              :proc        => Proc.new { |u| Chef::Config[:knife][:bootstrap_url] = u }
+
+            option :bootstrap_install_command,
+              :long        => "--bootstrap-install-command COMMANDS",
+              :description => "Custom command to install chef-client",
+              :proc        => Proc.new { |ic| Chef::Config[:knife][:bootstrap_install_command] = ic }
+
+            option :bootstrap_wget_options,
+              :long        => "--bootstrap-wget-options OPTIONS",
+              :description => "Add options to wget when installing chef-client",
+              :proc        => Proc.new { |wo| Chef::Config[:knife][:bootstrap_wget_options] = wo }
+
+            option :bootstrap_curl_options,
+              :long        => "--bootstrap-curl-options OPTIONS",
+              :description => "Add options to curl when install chef-client",
+              :proc        => Proc.new { |co| Chef::Config[:knife][:bootstrap_curl_options] = co }
+
+            option :auth_timeout,
+              :long => "--auth-timeout MINUTES",
+              :description => "The maximum time in minutes to wait to for authentication over the transport to the node to succeed. The default value is 25 minutes.",
+              :default => 25
+          end
+        end
+      end # module ends
+    end
+  end
+end

data/lib/chef/knife/cloud/chefbootstrap/bootstrap_protocol.rb

@@ -0,0 +1,69 @@
+# Author:: Kaustubh Deorukhkar (<kaustubh@clogeny.com>)
+# Author:: Prabhu Das (<prabhu.das@clogeny.com>)
+#
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/core/ui'
+require 'chef/knife/cloud/helpers'
+
+class Chef
+  class Knife
+    class Cloud
+      class BootstrapProtocol
+        include Chef::Knife::Cloud::Helpers
+
+        attr_accessor :bootstrap, :ui, :initial_sleep_delay
+        attr_reader :config
+
+        def initialize(config)
+          @config = config
+          @ui ||= Chef::Knife::UI.new(STDOUT, STDERR, STDIN, {})
+        end
+
+        def wait_for_server_ready
+          raise Chef::Exceptions::Override, "You must override wait_for_server_ready in #{self.to_s}"
+        end
+
+        def send_bootstrap_command
+          wait_for_server_ready
+          init_bootstrap_options
+          @bootstrap.run
+        end
+
+        def init_bootstrap_options
+          # set the command bootstrap options.
+          bootstrap.name_args = locate_config_value(:bootstrap_ip_address)
+          bootstrap.config[:chef_node_name] = locate_config_value(:chef_node_name)
+          bootstrap.config[:run_list] = locate_config_value(:run_list)
+          bootstrap.config[:prerelease] = locate_config_value(:prerelease)
+          bootstrap.config[:bootstrap_version] = locate_config_value(:bootstrap_version)
+          bootstrap.config[:distro] = locate_config_value(:distro)
+          bootstrap.config[:bootstrap_proxy] = locate_config_value(:bootstrap_proxy)
+          bootstrap.config[:environment] = locate_config_value(:environment)
+          # see chef/knife/bootstrap.rb #warn_chef_config_secret_key.
+          bootstrap.config[:encrypted_data_bag_secret] = locate_config_value(:encrypted_data_bag_secret)
+          bootstrap.config[:encrypted_data_bag_secret_file] = locate_config_value(:encrypted_data_bag_secret_file)
+          bootstrap.config[:first_boot_attributes] = locate_config_value(:first_boot_attributes)
+          bootstrap.config[:secret] = locate_config_value(:secret)
+          bootstrap.config[:secret_file] = locate_config_value(:secret_file)
+          bootstrap.config[:template_file] =  locate_config_value(:template_file)
+        end
+
+      end
+    end
+  end
+end

data/lib/chef/knife/cloud/chefbootstrap/bootstrapper.rb

@@ -0,0 +1,78 @@
+# Author:: Kaustubh Deorukhkar (<kaustubh@clogeny.com>)
+# Author:: Prabhu Das (<prabhu.das@clogeny.com>)
+#
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/core/ui'
+require 'chef/knife/cloud/chefbootstrap/ssh_bootstrap_protocol'
+require 'chef/knife/cloud/chefbootstrap/winrm_bootstrap_protocol'
+require 'chef/knife/cloud/chefbootstrap/windows_distribution'
+require 'chef/knife/cloud/chefbootstrap/unix_distribution'
+require 'chef/knife/cloud/exceptions'
+
+class Chef
+  class Knife
+    class Cloud
+      class Bootstrapper
+
+        def initialize(config)
+          @config = config
+          @ui ||= Chef::Knife::UI.new(STDOUT, STDERR, STDIN, {})
+        end
+
+        def bootstrap
+          # uses BootstrapDistribution and BootstrapProtocol to perform bootstrap
+          @protocol = create_bootstrap_protocol
+          @distribution = create_bootstrap_distribution
+          begin
+            @protocol.send_bootstrap_command
+          rescue Net::SSH::AuthenticationFailed => e
+            error_message = "Authentication Failed during bootstrapping. #{e.message}."
+            raise CloudExceptions::BootstrapError, error_message
+          end
+        end
+
+        def create_bootstrap_protocol
+          if @config[:bootstrap_protocol].nil? or @config[:bootstrap_protocol] == 'ssh'
+            SshBootstrapProtocol.new(@config)
+          elsif @config[:bootstrap_protocol] == 'winrm'
+            WinrmBootstrapProtocol.new(@config)
+          else
+            # raise an exception, invalid bootstrap protocol.
+            error_message = "Invalid bootstrap protocol."
+            ui.fatal(error_message)
+            raise CloudExceptions::BootstrapError, error_message
+          end
+        end
+
+        def create_bootstrap_distribution
+          if @config[:image_os_type] == 'windows'
+            Chef::Knife::Cloud::WindowsDistribution.new(@config)
+          elsif @config[:image_os_type] == 'linux'
+            Chef::Knife::Cloud::UnixDistribution.new(@config)
+          else
+            # raise an exception, invalid bootstrap distribution.
+            error_message = "Invalid bootstrap distribution. image_os_type should be either windows or linux."
+            ui.fatal(error_message)
+            raise CloudExceptions::BootstrapError, error_message
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/chef/knife/cloud/chefbootstrap/ssh_bootstrap_protocol.rb

@@ -0,0 +1,179 @@
+# Author:: Kaustubh Deorukhkar (<kaustubh@clogeny.com>)
+# Author:: Prabhu Das (<prabhu.das@clogeny.com>)
+#
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/cloud/chefbootstrap/bootstrap_protocol'
+require 'chef/knife/core/windows_bootstrap_context'
+require 'chef/knife/bootstrap'
+
+
+class Chef
+  class Knife
+    class Cloud
+      class SshBootstrapProtocol < BootstrapProtocol
+
+        def initialize(config)
+          @bootstrap = (config[:image_os_type] == 'linux') ? Chef::Knife::Bootstrap.new : Chef::Knife::BootstrapWindowsSsh.new
+          super
+        end
+
+        def init_bootstrap_options
+          bootstrap.config[:ssh_user] = @config[:ssh_user]
+          bootstrap.config[:ssh_password] = @config[:ssh_password]
+          bootstrap.config[:ssh_port] = locate_config_value(:ssh_port)
+          bootstrap.config[:identity_file] = @config[:identity_file]
+          bootstrap.config[:host_key_verify] = @config[:host_key_verify]
+          bootstrap.config[:use_sudo] = true unless @config[:ssh_user] == 'root'
+          bootstrap.config[:template_file] =  @config[:template_file]
+          bootstrap.config[:ssh_gateway] = locate_config_value(:ssh_gateway)
+          bootstrap.config[:forward_agent] = locate_config_value(:forward_agent)
+          bootstrap.config[:use_sudo_password] = locate_config_value(:use_sudo_password)
+          super
+        end
+
+        def wait_for_server_ready
+          print "\n#{ui.color("Waiting for sshd to host (#{@config[:bootstrap_ip_address]})", :magenta)}"
+
+          ssh_gateway = get_ssh_gateway_for(@config[:bootstrap_ip_address])
+
+          # The ssh_gateway & subnet_id are currently supported only in EC2.
+          if ssh_gateway
+            print(".") until tunnel_test_ssh(ssh_gateway, @config[:bootstrap_ip_address]) {
+              @initial_sleep_delay = !!locate_config_value(:subnet_id) ? 40 : 10
+              sleep @initial_sleep_delay
+              puts("done")
+            }
+          else
+            print(".") until tcp_test_ssh(@config[:bootstrap_ip_address], locate_config_value(:ssh_port)) {
+              @initial_sleep_delay = !!locate_config_value(:subnet_id) ? 40 : 10
+              sleep @initial_sleep_delay
+              puts("done")
+            }
+          end
+        end
+
+        def get_ssh_gateway_for(hostname)
+          if locate_config_value(:ssh_gateway)
+            # The ssh_gateway specified in the knife config (if any) takes
+            # precedence over anything in the SSH configuration
+            Chef::Log.debug("Using ssh gateway #{locate_config_value(:ssh_gateway)} from knife config")
+            locate_config_value(:ssh_gateway)
+          else
+            # Next, check if the SSH configuration has a ProxyCommand
+            # directive for this host. If there is one, parse out the
+            # host from the proxy command
+            ssh_proxy = Net::SSH::Config.for(hostname)[:proxy]
+            if ssh_proxy.respond_to?(:command_line_template)
+              # ssh gateway_hostname nc %h %p
+              proxy_pattern = /ssh\s+(\S+)\s+nc/
+              matchdata = proxy_pattern.match(ssh_proxy.command_line_template)
+              if matchdata.nil?
+                Chef::Log.debug("Unable to determine ssh gateway for '#{hostname}' from ssh config template: #{ssh_proxy.command_line_template}")
+                nil
+              else
+                # Return hostname extracted from command line template
+                Chef::Log.debug("Using ssh gateway #{matchdata[1]} from ssh config")
+                matchdata[1]
+              end
+            else
+              # Return nil if we cannot find an ssh_gateway
+              Chef::Log.debug("No ssh gateway found, making a direct connection")
+              nil
+            end
+          end
+        end
+
+        def tcp_test_ssh(hostname, ssh_port)
+          begin
+            tcp_socket = TCPSocket.new(hostname, ssh_port)
+            readable = IO.select([tcp_socket], nil, nil, 5)
+            if readable
+              ssh_banner = tcp_socket.gets
+              if ssh_banner.nil? or ssh_banner.empty?
+                false
+              else
+                Chef::Log.debug("ssh accepting connections on #{hostname}, banner is #{tcp_socket.gets}")
+                yield
+                true
+              end
+            else
+              false
+            end
+          rescue Errno::EPERM, Errno::ETIMEDOUT
+            Chef::Log.debug("ssh timed out: #{hostname}")
+            false
+          rescue SocketError, Errno::ECONNREFUSED, Errno::EHOSTUNREACH, Errno::ENETUNREACH, IOError
+            Chef::Log.debug("ssh failed to connect: #{hostname}")
+            sleep 2
+            false
+           # This happens on some mobile phone networks
+          rescue Errno::ECONNRESET
+             Chef::Log.debug("ssh reset its connection: #{hostname}")
+            sleep 2
+            false
+          ensure
+            tcp_socket && tcp_socket.close
+          end
+        end
+
+        def tunnel_test_ssh(ssh_gateway, hostname, &block)
+          begin
+            status = false
+            gateway = configure_ssh_gateway(ssh_gateway)
+            gateway.open(hostname, locate_config_value(:ssh_port)) do |local_tunnel_port|
+              status = tcp_test_ssh('localhost', local_tunnel_port, &block)
+            end
+            status
+          rescue SocketError, Errno::ECONNREFUSED, Errno::EHOSTUNREACH, Errno::ENETUNREACH, IOError
+            sleep 2
+            false
+          rescue Errno::EPERM, Errno::ETIMEDOUT
+            false
+          end
+        end
+
+        def configure_ssh_gateway(ssh_gateway)
+          gw_host, gw_user = ssh_gateway.split('@').reverse
+          gw_host, gw_port = gw_host.split(':')
+          gateway_options = { :port => gw_port || 22 }
+
+          # Load the SSH config for the SSH gateway host.
+          # Set the gateway user if it was not part of the
+          # SSH gateway string, and use any configured
+          # SSH keys.
+          ssh_gateway_config = Net::SSH::Config.for(gw_host)
+          gw_user ||= ssh_gateway_config[:user]
+
+          # Always use the gateway keys from the SSH Config
+          gateway_keys = ssh_gateway_config[:keys]        
+
+          # Use the keys specificed on the command line if available (overrides SSH Config)
+          if locate_config_value(:ssh_gateway_identity)
+            gateway_keys = Array(locate_config_value(:ssh_gateway_identity))
+          end
+
+          unless gateway_keys.nil?
+            gateway_options[:keys] = gateway_keys
+          end
+
+          Net::SSH::Gateway.new(gw_host, gw_user, gateway_options)
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/cloud/chefbootstrap/unix_distribution.rb

@@ -0,0 +1,31 @@
+#
+# Author:: Prabhu Das (<prabhu.das@clogeny.com>)
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+require 'chef/knife/cloud/chefbootstrap/bootstrap_distribution'
+
+class Chef
+  class Knife
+    class Cloud
+      class UnixDistribution < BootstrapDistribution
+
+        def initialize(config)
+          super
+          config[:distro] =  config[:distro] || "chef-full"
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/cloud/chefbootstrap/windows_distribution.rb

@@ -0,0 +1,32 @@
+#
+# Author:: Prabhu Das (<prabhu.das@clogeny.com>)
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+require 'chef/knife/cloud/chefbootstrap/bootstrap_distribution'
+require 'chef/knife/bootstrap_windows_base'
+
+class Chef
+  class Knife
+    class Cloud
+      class WindowsDistribution < BootstrapDistribution
+        def initialize(config)
+          super
+          config[:distro] ||= "windows-chef-client-msi"
+        end
+      end
+    end
+  end
+end

data/lib/chef/knife/cloud/chefbootstrap/winrm_bootstrap_protocol.rb

@@ -0,0 +1,85 @@
+# Author:: Kaustubh Deorukhkar (<kaustubh@clogeny.com>)
+# Author:: Prabhu Das (<prabhu.das@clogeny.com>)
+#
+# Copyright:: Copyright (c) 2013 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'chef/knife/cloud/chefbootstrap/bootstrap_protocol'
+
+class Chef
+  class Knife
+    class Cloud
+      class WinrmBootstrapProtocol < BootstrapProtocol
+
+        def initialize(config)
+          load_winrm_deps
+          @bootstrap = Chef::Knife::BootstrapWindowsWinrm.new
+          super
+        end
+
+        def load_winrm_deps
+          require 'winrm'
+          require 'em-winrm'
+          require 'chef/knife/bootstrap_windows_winrm'
+          require 'chef/knife/core/windows_bootstrap_context'
+          require 'chef/knife/winrm'
+        end
+
+        def init_bootstrap_options
+          bootstrap.config[:winrm_user] = @config[:winrm_user] || 'Administrator'
+          bootstrap.config[:winrm_password] = @config[:winrm_password]
+          bootstrap.config[:winrm_transport] = @config[:winrm_transport]
+          bootstrap.config[:winrm_port] = @config[:winrm_port]
+          bootstrap.config[:auth_timeout] =  @config[:auth_timeout]
+          super
+        end
+
+        def wait_for_server_ready
+          print "\n#{ui.color("Waiting for winrm to host (#{@config[:bootstrap_ip_address]})", :magenta)}"
+          print(".") until tcp_test_winrm(@config[:bootstrap_ip_address], @config[:winrm_port]){
+            sleep @initial_sleep_delay ||= 10
+            puts("done")
+          }
+        end
+
+        def tcp_test_winrm(hostname, port)
+          tcp_socket = TCPSocket.new(hostname, port)
+          return true
+        rescue SocketError
+          sleep 2
+          false
+        rescue Errno::ETIMEDOUT
+          false
+        rescue Errno::EPERM
+          false
+        rescue Errno::ECONNREFUSED
+          sleep 2
+          false
+        rescue Errno::EHOSTUNREACH
+          sleep 2
+          false
+        rescue Errno::ENETUNREACH
+          sleep 2
+          false
+        ensure
+          tcp_socket && tcp_socket.close          
+        end
+
+      end
+    end
+  end
+end
+