train-core 2.0.12 → 2.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6428f10f1fcb461fff9d6a262ed8a8c643ca0aaf21cc60a0ddeff1573f4e3c0e
-  data.tar.gz: a1d8450b00960078fb2116ec40dd3f1e61a5976a4aa6aa14e4c0a3e73596f27b
+  metadata.gz: 5f219653ce1764d525f68ee34a5a0fe84bed8c76b8fd28fd05d01b5cbef33200
+  data.tar.gz: bcce1e0cf610ee3e765b8a4a9bfad310a7baebe54578d133b3a8b242314521ec
 SHA512:
-  metadata.gz: 338cd5ee7abda4182c71c3132336b647ab5353d90934673fdf156577506167e109e557b6c139288b5dcc89476742a34eb692b95bfd99ad7159775f531f90c3ee
-  data.tar.gz: d15361a0bc4362272ea783c5198dea4ef6dbb5beb1a3878e04eb4484126f0b5daa2fdb0dbe725b5828da9cf48519f3d43730860a5a4e7de3ceaac7284b13f765
+  metadata.gz: 4f2ab55ffe10956e5133cc6863138c514a4e21d75ecfc7ecb0f6b1ac002426be11b859b2f9a2d6c2da045a4d4720829610abccc993586c6e5d9e25aff31c09f7
+  data.tar.gz: 7cf6b5888e1cf3cc4f757736fa4f623d4e525fa45bd91f893e26ae049b915ee6522fd8109cfd8752632c29ce57a7d5a84b297f878f45a30bdbe016ab573d3809

data/lib/train/extras/command_wrapper.rb

@@ -121,8 +121,45 @@ module Train::Extras
     end
   end
 
+  # Wrap windows commands.
+  class WindowsCommand < CommandWrapperBase
+    Train::Options.attach(self)
+
+    option :shell_command, default: nil
+
+    def initialize(backend, options)
+      @backend = backend
+      validate_options(options)
+
+      @shell_command = options[:shell_command] # e.g. 'powershell'
+    end
+
+    # (see CommandWrapperBase::run)
+    def run(command)
+      powershell_wrap(command)
+    end
+
+    private
+
+    # Wrap the cmd in an encoded command to allow pipes and quotes
+    def powershell_wrap(cmd)
+      shell = @shell_command || 'powershell'
+
+      # Prevent progress stream from leaking into stderr
+      script = "$ProgressPreference='SilentlyContinue';" + cmd
+
+      # Encode script so PowerShell can use it
+      script = script.encode('UTF-16LE', 'UTF-8')
+      base64_script = Base64.strict_encode64(script)
+
+      cmd = "#{shell} -NoProfile -EncodedCommand #{base64_script}"
+      cmd
+    end
+  end
+
   class CommandWrapper
     include_options LinuxCommand
+    include_options WindowsCommand
 
     def self.load(transport, options)
       if transport.os.unix?
@@ -134,6 +171,9 @@ module Train::Extras
           raise Train::UserError.new("Sudo failed: #{msg}", reason)
         end
         res
+      elsif transport.os.windows?
+        res = WindowsCommand.new(transport, options)
+        res
       end
     end
   end

data/lib/train/platforms/detect/helpers/os_windows.rb

@@ -3,7 +3,8 @@
 module Train::Platforms::Detect::Helpers
   module Windows
     def detect_windows
-      res = @backend.run_command('cmd /c ver')
+      # try to detect windows, use cmd.exe to also support Microsoft OpenSSH
+      res = @backend.run_command('cmd.exe /c ver')
       return false if res.exit_status != 0 or res.stdout.empty?
 
       # if the ver contains `Windows`, we know its a Windows system

data/lib/train/platforms/detect/specifications/os.rb

@@ -25,6 +25,11 @@ module Train::Platforms::Detect::Specifications
               is_windows = true if ruby_host_os(/mswin|mingw32|windows/)
             end
 
+            # Try to detect windows even for ssh transport
+            if !is_windows && detect_windows == true
+              is_windows = true
+            end
+
             is_windows
           }
       # windows platform

data/lib/train/transports/cisco_ios_connection.rb

@@ -46,7 +46,8 @@ class Train::Transports::SSH
         # This verifies we are not in privileged exec mode before running the
         # enable command. Otherwise, the password will be in history.
         if run_command_via_connection('show privilege').stdout.split[-1] != '15'
-          run_command_via_connection("enable\r\n#{@enable_password}")
+          # Extra newlines to get back to prompt if incorrect password is used
+          run_command_via_connection("enable\n#{@enable_password}\n\n\n")
         end
       end
 
@@ -65,7 +66,9 @@ class Train::Transports::SSH
 
       logger.debug('[SSH] waiting for prompt')
       until @buf =~ @prompt
-        raise BadEnablePassword if @buf =~ /Bad secrets/
+        if @buf =~ /Bad (secrets|password)|Access denied/
+          raise BadEnablePassword
+        end
         session.connection.process(0)
       end
 

data/lib/train/transports/ssh_connection.rb

@@ -202,6 +202,8 @@ class Train::Transports::SSH
         Train::File::Remote::Unix.new(self, path)
       elsif os[:name] == 'qnx'
         Train::File::Remote::Qnx.new(self, path)
+      elsif os.windows?
+        Train::File::Remote::Windows.new(self, path)
       else
         Train::File::Remote::Linux.new(self, path)
       end

data/lib/train/version.rb

@@ -3,5 +3,5 @@
 # Author:: Dominik Richter (<dominik.richter@gmail.com>)
 
 module Train
-  VERSION = '2.0.12'.freeze
+  VERSION = '2.1.0'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: train-core
 version: !ruby/object:Gem::Version
-  version: 2.0.12
+  version: 2.1.0
 platform: ruby
 authors:
 - Dominik Richter
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-04-23 00:00:00.000000000 Z
+date: 2019-05-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json