train-core 2.1.7 → 2.1.13

data/lib/train/transports/mock.rb

@@ -1,11 +1,9 @@
-# encoding: utf-8
-
-require 'train/plugins'
-require 'digest'
+require "train/plugins"
+require "digest"
 
 module Train::Transports
   class Mock < Train.plugin(1)
-    name 'mock'
+    name "mock"
 
     def initialize(conf = nil)
       @conf = conf || {}
@@ -17,39 +15,39 @@ module Train::Transports
     end
 
     def to_s
-      'Mock Transport'
+      "Mock Transport"
     end
 
     private
 
     def trace_calls
       interface_methods = {
-        'Train::Transports::Mock' =>
+        "Train::Transports::Mock" =>
           Train::Transports::Mock.instance_methods(false),
-        'Train::Transports::Mock::Connection' =>
+        "Train::Transports::Mock::Connection" =>
           Connection.instance_methods(false),
-        'Train::Transports::Mock::Connection::File' =>
+        "Train::Transports::Mock::Connection::File" =>
           Connection::FileCommon.instance_methods(false),
-        'Train::Transports::Mock::Connection::OS' =>
+        "Train::Transports::Mock::Connection::OS" =>
           Train::Platform.instance_methods(false),
       }
 
       # rubocop:disable Metrics/ParameterLists
       # rubocop:disable Lint/Eval
-      set_trace_func proc { |event, _file, _line, id, binding, classname|
-        unless classname.to_s.start_with?('Train::Transports::Mock') and
-               event == 'call' and
-               interface_methods[classname.to_s].include?(id)
+      set_trace_func(proc { |event, _file, _line, id, binding, classname|
+        unless classname.to_s.start_with?("Train::Transports::Mock") &&
+            (event == "call") &&
+            interface_methods[classname.to_s].include?(id)
           next
         end
         # kindly borrowed from the wonderful simple-tracer by matugm
         arg_names = eval(
-          'method(__method__).parameters.map { |arg| arg[1].to_s }',
+          "method(__method__).parameters.map { |arg| arg[1].to_s }",
           binding)
-        args = eval("#{arg_names}.map { |arg| eval(arg) }", binding).join(', ')
-        prefix = '-' * (classname.to_s.count(':') - 2) + '> '
+        args = eval("#{arg_names}.map { |arg| eval(arg) }", binding).join(", ")
+        prefix = "-" * (classname.to_s.count(":") - 2) + "> "
         puts("#{prefix}#{id} #{args}")
-      }
+      })
       # rubocop:enable all
     end
   end
@@ -67,14 +65,14 @@ class Train::Transports::Mock
     end
 
     def uri
-      'mock://'
+      "mock://"
     end
 
     def mock_os(value = {})
       # if a user passes a nil value, set to an empty hash so the merge still succeeds
       value ||= {}
-      value.each { |k, v| value[k] = 'unknown' if v.nil? }
-      value = { name: 'mock', family: 'mock', release: 'unknown', arch: 'unknown' }.merge(value)
+      value.each { |k, v| value[k] = "unknown" if v.nil? }
+      value = { name: "mock", family: "mock", release: "unknown", arch: "unknown" }.merge(value)
 
       platform = Train::Platforms.name(value[:name])
       platform.find_family_hierarchy
@@ -100,26 +98,26 @@ class Train::Transports::Mock
     end
 
     def mock_command(cmd, stdout = nil, stderr = nil, exit_status = 0)
-      @cache[:command][cmd] = Command.new(stdout || '', stderr || '', exit_status)
+      @cache[:command][cmd] = Command.new(stdout || "", stderr || "", exit_status)
     end
 
     def command_not_found(cmd)
       if @options[:verbose]
-        STDERR.puts('Command not mocked:')
-        STDERR.puts('    '+cmd.to_s.split("\n").join("\n    "))
-        STDERR.puts('    SHA: ' + Digest::SHA256.hexdigest(cmd.to_s))
+        $stderr.puts("Command not mocked:")
+        $stderr.puts("    " + cmd.to_s.split("\n").join("\n    "))
+        $stderr.puts("    SHA: " + Digest::SHA256.hexdigest(cmd.to_s))
       end
       # return a non-zero exit code
       mock_command(cmd, nil, nil, 1)
     end
 
     def file_not_found(path)
-      STDERR.puts('File not mocked: '+path.to_s) if @options[:verbose]
+      $stderr.puts("File not mocked: " + path.to_s) if @options[:verbose]
       File.new(self, path)
     end
 
     def to_s
-      'Mock Connection'
+      "Mock Connection"
     end
 
     private
@@ -142,23 +140,23 @@ end
 class Train::Transports::Mock::Connection
   class File < Train::File
     def self.from_json(json)
-      res = new(json['backend'],
-                json['path'],
-                json['follow_symlink'])
-      res.type = json['type']
+      res = new(json["backend"],
+                json["path"],
+                json["follow_symlink"])
+      res.type = json["type"]
       Train::File::DATA_FIELDS.each do |f|
-        m = (f.tr('?', '') + '=').to_sym
+        m = (f.tr("?", "") + "=").to_sym
         res.method(m).call(json[f])
       end
       res
     end
 
     Train::File::DATA_FIELDS.each do |m|
-      attr_accessor m.tr('?', '').to_sym
-      next unless m.include?('?')
+      attr_accessor m.tr("?", "").to_sym
+      next unless m.include?("?")
 
       define_method m.to_sym do
-        method(m.tr('?', '').to_sym).call
+        method(m.tr("?", "").to_sym).call
       end
     end
     attr_accessor :type

data/lib/train/transports/ssh.rb

@@ -18,9 +18,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-require 'net/ssh'
-require 'net/scp'
-require 'train/errors'
+require "net/ssh"
+require "net/scp"
+require "train/errors"
 
 module Train::Transports
   # Wrapped exception for any internally raised SSH-related errors.
@@ -34,10 +34,10 @@ module Train::Transports
   #
   # @author Fletcher Nichol <fnichol@nichol.ca>
   class SSH < Train.plugin(1) # rubocop:disable Metrics/ClassLength
-    name 'ssh'
+    name "ssh"
 
-    require 'train/transports/ssh_connection'
-    require 'train/transports/cisco_ios_connection'
+    require "train/transports/ssh_connection"
+    require "train/transports/cisco_ios_connection"
 
     # add options for submodules
     include_options Train::Extras::CommandWrapper
@@ -45,7 +45,7 @@ module Train::Transports
     # common target configuration
     option :host,      required: true
     option :port,      default: 22, required: true
-    option :user,      default: 'root', required: true
+    option :user,      default: "root", required: true
     option :key_files, default: nil
     option :password,  default: nil
 
@@ -60,7 +60,7 @@ module Train::Transports
     option :pty, default: false
     option :proxy_command, default: nil
     option :bastion_host, default: nil
-    option :bastion_user, default: 'root'
+    option :bastion_user, default: "root"
     option :bastion_port, default: 22
     option :non_interactive, default: false
     option :verify_host_key, default: false
@@ -89,34 +89,34 @@ module Train::Transports
       super(options)
 
       key_files = Array(options[:key_files])
-      options[:auth_methods] ||= ['none']
+      options[:auth_methods] ||= ["none"]
 
       unless key_files.empty?
-        options[:auth_methods].push('publickey')
+        options[:auth_methods].push("publickey")
         options[:keys_only] = true if options[:password].nil?
         options[:key_files] = key_files
       end
 
       unless options[:password].nil?
-        options[:auth_methods].push('password', 'keyboard-interactive')
+        options[:auth_methods].push("password", "keyboard-interactive")
       end
 
-      if options[:auth_methods] == ['none']
+      if options[:auth_methods] == ["none"]
         if ssh_known_identities.empty?
-          fail Train::ClientError,
-               'Your SSH Agent has no keys added, and you have not specified a password or a key file'
+          raise Train::ClientError,
+               "Your SSH Agent has no keys added, and you have not specified a password or a key file"
         else
-          logger.debug('[SSH] Using Agent keys as no password or key file have been specified')
-          options[:auth_methods].push('publickey')
+          logger.debug("[SSH] Using Agent keys as no password or key file have been specified")
+          options[:auth_methods].push("publickey")
         end
       end
 
       if options[:pty]
-        logger.warn('[SSH] PTY requested: stderr will be merged into stdout')
+        logger.warn("[SSH] PTY requested: stderr will be merged into stdout")
       end
 
       if [options[:proxy_command], options[:bastion_host]].all? { |type| !type.nil? }
-        fail Train::ClientError, 'Only one of proxy_command or bastion_host needs to be specified'
+        raise Train::ClientError, "Only one of proxy_command or bastion_host needs to be specified"
       end
 
       super
@@ -142,30 +142,30 @@ module Train::Transports
     # @api private
     def connection_options(opts)
       connection_options = {
-        logger:                 logger,
-        user_known_hosts_file:  '/dev/null',
-        hostname:               opts[:host],
-        port:                   opts[:port],
-        username:               opts[:user],
-        compression:            opts[:compression],
-        compression_level:      opts[:compression_level],
-        keepalive:              opts[:keepalive],
-        keepalive_interval:     opts[:keepalive_interval],
-        timeout:                opts[:connection_timeout],
-        connection_retries:     opts[:connection_retries],
+        logger: logger,
+        user_known_hosts_file: "/dev/null",
+        hostname: opts[:host],
+        port: opts[:port],
+        username: opts[:user],
+        compression: opts[:compression],
+        compression_level: opts[:compression_level],
+        keepalive: opts[:keepalive],
+        keepalive_interval: opts[:keepalive_interval],
+        timeout: opts[:connection_timeout],
+        connection_retries: opts[:connection_retries],
         connection_retry_sleep: opts[:connection_retry_sleep],
-        max_wait_until_ready:   opts[:max_wait_until_ready],
-        auth_methods:           opts[:auth_methods],
-        keys_only:              opts[:keys_only],
-        keys:                   opts[:key_files],
-        password:               opts[:password],
-        forward_agent:          opts[:forward_agent],
-        proxy_command:          opts[:proxy_command],
-        bastion_host:           opts[:bastion_host],
-        bastion_user:           opts[:bastion_user],
-        bastion_port:           opts[:bastion_port],
-        non_interactive:        opts[:non_interactive],
-        transport_options:      opts,
+        max_wait_until_ready: opts[:max_wait_until_ready],
+        auth_methods: opts[:auth_methods],
+        keys_only: opts[:keys_only],
+        keys: opts[:key_files],
+        password: opts[:password],
+        forward_agent: opts[:forward_agent],
+        proxy_command: opts[:proxy_command],
+        bastion_host: opts[:bastion_host],
+        bastion_user: opts[:bastion_user],
+        bastion_port: opts[:bastion_port],
+        non_interactive: opts[:non_interactive],
+        transport_options: opts,
       }
       # disable host key verification. The hash key and value to use
       # depends on the version of net-ssh in use.
@@ -203,20 +203,20 @@ module Train::Transports
         # 5.0+ style
         {
           # It's not a boolean anymore.
-          'true' => :always,
-          'false' => :never,
+          "true" => :always,
+          "false" => :never,
           true => :always,
           false => :never,
           # May be correct value, but strings from JSON config
-          'always' => :always,
-          'never' => :never,
+          "always" => :always,
+          "never" => :never,
           nil => :never,
         }.fetch(given, given)
       else
         # up to 4.2 style
         {
-          'true' => true,
-          'false' => false,
+          "true" => true,
+          "false" => false,
           nil => false,
         }.fetch(given, given)
       end

data/lib/train/transports/ssh_connection.rb

@@ -18,9 +18,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-require 'net/ssh'
-require 'net/scp'
-require 'timeout'
+require "net/ssh"
+require "net/scp"
+require "timeout"
 
 class Train::Transports::SSH
   # A Connection instance can be generated and re-generated, given new
@@ -63,17 +63,17 @@ class Train::Transports::SSH
     end
 
     def ssh_opts
-      level = logger.debug? ? 'VERBOSE' : 'ERROR'
-      fwd_agent = options[:forward_agent] ? 'yes' : 'no'
+      level = logger.debug? ? "VERBOSE" : "ERROR"
+      fwd_agent = options[:forward_agent] ? "yes" : "no"
 
       args  = %w{ -o UserKnownHostsFile=/dev/null }
       args += %w{ -o StrictHostKeyChecking=no }
       args += %w{ -o IdentitiesOnly=yes } if options[:keys]
       args += %w{ -o BatchMode=yes } if options[:non_interactive]
-      args += %W( -o LogLevel=#{level} )
-      args += %W( -o ForwardAgent=#{fwd_agent} ) if options.key?(:forward_agent)
+      args += %W{ -o LogLevel=#{level} }
+      args += %W{ -o ForwardAgent=#{fwd_agent} } if options.key?(:forward_agent)
       Array(options[:keys]).each do |ssh_key|
-        args += %W( -i #{ssh_key} )
+        args += %W{ -i #{ssh_key} }
       end
       args
     end
@@ -86,19 +86,19 @@ class Train::Transports::SSH
       return @proxy_command unless @proxy_command.nil?
       args = %w{ ssh }
       args += ssh_opts
-      args += %W( #{@bastion_user}@#{@bastion_host} )
-      args += %W( -p #{@bastion_port} )
+      args += %W{ #{@bastion_user}@#{@bastion_host} }
+      args += %W{ -p #{@bastion_port} }
       args += %w{ -W %h:%p }
-      args.join(' ')
+      args.join(" ")
     end
 
     # (see Base::Connection#login_command)
     def login_command
       args = ssh_opts
-      args += %W( -o ProxyCommand='#{generate_proxy_command}' ) if check_proxy
-      args += %W( -p #{@port} )
-      args += %W( #{@username}@#{@hostname} )
-      LoginCommand.new('ssh', args)
+      args += %W{ -o ProxyCommand='#{generate_proxy_command}' } if check_proxy
+      args += %W{ -p #{@port} }
+      args += %W{ #{@username}@#{@hostname} }
+      LoginCommand.new("ssh", args)
     end
 
     # (see Base::Connection#upload)
@@ -138,7 +138,7 @@ class Train::Transports::SSH
         retries: @max_wait_until_ready / delay,
         delay:   delay,
         message: "Waiting for SSH service on #{@hostname}:#{@port}, " \
-                 "retrying in #{delay} seconds",
+                 "retrying in #{delay} seconds"
       )
       run_command(PING_COMMAND.dup)
     end
@@ -172,14 +172,14 @@ class Train::Transports::SSH
     def establish_connection(opts)
       logger.debug("[SSH] opening connection to #{self}")
       if check_proxy
-        require 'net/ssh/proxy/command'
+        require "net/ssh/proxy/command"
         @options[:proxy] = Net::SSH::Proxy::Command.new(generate_proxy_command)
       end
       Net::SSH.start(@hostname, @username, @options.clone.delete_if { |_key, value| value.nil? })
     rescue *RESCUE_EXCEPTIONS_ON_ESTABLISH => e
       if (opts[:retries] -= 1) <= 0
         logger.warn("[SSH] connection failed, terminating (#{e.inspect})")
-        raise Train::Transports::SSHFailed, 'SSH session could not be established'
+        raise Train::Transports::SSHFailed, "SSH session could not be established"
       end
 
       if opts[:message]
@@ -200,7 +200,7 @@ class Train::Transports::SSH
         Train::File::Remote::Aix.new(self, path)
       elsif os.solaris?
         Train::File::Remote::Unix.new(self, path)
-      elsif os[:name] == 'qnx'
+      elsif os[:name] == "qnx"
         Train::File::Remote::Qnx.new(self, path)
       elsif os.windows?
         Train::File::Remote::Windows.new(self, path)
@@ -210,7 +210,7 @@ class Train::Transports::SSH
     end
 
     def run_command_via_connection(cmd, &data_handler)
-      cmd.dup.force_encoding('binary') if cmd.respond_to?(:force_encoding)
+      cmd.dup.force_encoding("binary") if cmd.respond_to?(:force_encoding)
       logger.debug("[SSH] #{self} (#{cmd})")
 
       reset_session if session.closed?
@@ -228,7 +228,7 @@ class Train::Transports::SSH
       # transport. This retries the command if this is the case.
       # See:
       #  https://github.com/inspec/train/pull/271
-      logger.debug('[SSH] Possible Cisco IOS race condition, retrying command')
+      logger.debug("[SSH] Possible Cisco IOS race condition, retrying command")
 
       # Only attempt retry up to 5 times to avoid infinite loop
       @ios_cmd_retries += 1
@@ -246,7 +246,7 @@ class Train::Transports::SSH
     def session(retry_options = {})
       @session ||= establish_connection({
         retries: @connection_retries.to_i,
-        delay:   @connection_retry_sleep.to_i,
+        delay: @connection_retry_sleep.to_i,
       }.merge(retry_options))
     end
 
@@ -260,7 +260,7 @@ class Train::Transports::SSH
     # @api private
     def to_s
       options_to_print = @options.clone
-      options_to_print[:password] = '<hidden>' if options_to_print.key?(:password)
+      options_to_print[:password] = "<hidden>" if options_to_print.key?(:password)
       "#{@username}@#{@hostname}<#{options_to_print.inspect}>"
     end
 
@@ -274,7 +274,7 @@ class Train::Transports::SSH
     #
     # @api private
     def execute_on_channel(cmd, &data_handler)
-      stdout = stderr = ''
+      stdout = stderr = ""
       exit_status = nil
       session.open_channel do |channel|
         # wrap commands if that is configured
@@ -282,11 +282,11 @@ class Train::Transports::SSH
 
         if @transport_options[:pty]
           channel.request_pty do |_ch, success|
-            fail Train::Transports::SSHPTYFailed, 'Requesting PTY failed' unless success
+            raise Train::Transports::SSHPTYFailed, "Requesting PTY failed" unless success
           end
         end
         channel.exec(cmd) do |_, success|
-          abort 'Couldn\'t execute command on SSH.' unless success
+          abort "Couldn't execute command on SSH." unless success
           channel.on_data do |_, data|
             yield(data) unless data_handler.nil?
             stdout += data
@@ -297,11 +297,11 @@ class Train::Transports::SSH
             stderr += data
           end
 
-          channel.on_request('exit-status') do |_, data|
+          channel.on_request("exit-status") do |_, data|
             exit_status = data.read_long
           end
 
-          channel.on_request('exit-signal') do |_, data|
+          channel.on_request("exit-signal") do |_, data|
             exit_status = data.read_long
           end
         end