winrm-fs 0.3.1 → 0.3.2

data/changelog.md

@@ -1,31 +1,34 @@
-# WinRM-fs Gem Changelog
-
-# 0.3.1
-- Widen logging version constraints to include 2.0 (matching WinRM core gem)
-
-# 0.3.0
-- Jetisons `CommandExecutor` now living in the core WinRM gem and swaps in implementation currently used in the winrm-transport gem. These changes should have little visible effect on current consumers of the `FileManager` class with these exceptions:
-  - BREAKING CHANGE: When uploading a directory and the destination directory exists on the endpoint, the source base directory will be created below the destination directory on the endpoint and the source directory contents will be unzipped to that location. Prior to this release, the contents of the source directory would be unzipped to an existing destination directory without creating the source base directory. This new behavior is more consistent with SCP and other well known shell copy commands.
-  - `Upload` may now receive an array of source files and directories rather than just a single file or directory path.
-
-# 0.2.4
-- Fix issue 21, downloading files is extremely slow.
-- Add zip file creation debug logging.
-
-# 0.2.3
-- Fix yielding progress data, issue #23
-
-# 0.2.2
-- Fix powershell streams leaking to standard error breaking Windows 10, issue #18
-
-# 0.2.1
-- Fixed issue 16 creating zip file on Windows
-
-# 0.2.0
-- Redesigned temp zip file creation system
-- Fixed lots of small edge case issues especially with directory uploads
-- Simplified file manager upload method API to take only a single source file or directory
-- Expanded acceptable username and hostnames for rwinrmcp
-
-# 0.1.0
-- Initial alpha quality release
+# WinRM-fs Gem Changelog
+
+# 0.3.2
+- Fix re-extraction of cached directories from temp folder when there is more than one "clean" directory deleted from destination
+
+# 0.3.1
+- Widen logging version constraints to include 2.0 (matching WinRM core gem)
+
+# 0.3.0
+- Jetisons `CommandExecutor` now living in the core WinRM gem and swaps in implementation currently used in the winrm-transport gem. These changes should have little visible effect on current consumers of the `FileManager` class with these exceptions:
+  - BREAKING CHANGE: When uploading a directory and the destination directory exists on the endpoint, the source base directory will be created below the destination directory on the endpoint and the source directory contents will be unzipped to that location. Prior to this release, the contents of the source directory would be unzipped to an existing destination directory without creating the source base directory. This new behavior is more consistent with SCP and other well known shell copy commands.
+  - `Upload` may now receive an array of source files and directories rather than just a single file or directory path.
+
+# 0.2.4
+- Fix issue 21, downloading files is extremely slow.
+- Add zip file creation debug logging.
+
+# 0.2.3
+- Fix yielding progress data, issue #23
+
+# 0.2.2
+- Fix powershell streams leaking to standard error breaking Windows 10, issue #18
+
+# 0.2.1
+- Fixed issue 16 creating zip file on Windows
+
+# 0.2.0
+- Redesigned temp zip file creation system
+- Fixed lots of small edge case issues especially with directory uploads
+- Simplified file manager upload method API to take only a single source file or directory
+- Expanded acceptable username and hostnames for rwinrmcp
+
+# 0.1.0
+- Initial alpha quality release

data/lib/winrm-fs.rb

@@ -1,28 +1,28 @@
-# encoding: UTF-8
-#
-# Copyright 2015 Shawn Neal <sneal@sneal.net>
-#
-# 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 'winrm'
-require 'logger'
-require 'pathname'
-require_relative 'winrm-fs/exceptions'
-require_relative 'winrm-fs/file_manager'
-
-module WinRM
-  # WinRM File System
-  module FS
-    # Top level module code
-  end
-end
+# encoding: UTF-8
+#
+# Copyright 2015 Shawn Neal <sneal@sneal.net>
+#
+# 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 'winrm'
+require 'logger'
+require 'pathname'
+require_relative 'winrm-fs/exceptions'
+require_relative 'winrm-fs/file_manager'
+
+module WinRM
+  # WinRM File System
+  module FS
+    # Top level module code
+  end
+end

data/lib/winrm-fs/core/file_transporter.rb

@@ -1,506 +1,507 @@
-# -*- encoding: utf-8 -*-
-#
-# Author:: Fletcher (<fnichol@nichol.ca>)
-#
-# Copyright (C) 2015, Fletcher Nichol
-#
-# 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 'benchmark'
-require 'csv'
-require 'digest'
-require 'securerandom'
-require 'stringio'
-
-require 'winrm-fs/core/tmp_zip'
-
-module WinRM
-  module FS
-    module Core
-      # Wrapped exception for any internally raised WinRM-related errors.
-      #
-      # @author Fletcher Nichol <fnichol@nichol.ca>
-      class FileTransporterFailed < ::WinRM::WinRMError; end
-      # rubocop:disable MethodLength, AbcSize, ClassLength
-
-      # Object which can upload one or more files or directories to a remote
-      # host over WinRM using PowerShell scripts and CMD commands. Note that
-      # this form of file transfer is *not* ideal and extremely costly on both
-      # the local and remote sides. Great pains are made to minimize round
-      # trips to  the remote host and to minimize the number of PowerShell
-      # sessions being invoked which can be 2 orders of magnitude more
-      # expensive than vanilla CMD commands.
-      #
-      # This object is supported by either a `CommandExecutor` instance as it
-      # depends on the `#run_cmd` and `#run_powershell_script` API contracts.
-      #
-      # An optional logger can be supplied, assuming it can respond to the
-      # `#debug` and `#debug?` messages.
-      #
-      # @author Fletcher Nichol <fnichol@nichol.ca>
-      # @author Matt Wrock <matt@mattwrock.com>
-      class FileTransporter
-        # Creates a FileTransporter given a CommandExecutor object.
-        #
-        # @param executor [CommandExecutor] a winrm CommandExecutor object
-        def initialize(executor, opts = {})
-          @executor  = executor
-          @logger = executor.service.logger
-          @id_generator = opts.fetch(:id_generator) { -> { SecureRandom.uuid } }
-        end
-
-        # Uploads a collection of files and/or directories to the remote host.
-        #
-        # **TODO Notes:**
-        # * options could specify zip mode, zip options, etc.
-        # * maybe option to set tmpfile base dir to override $env:PATH?
-        # * progress yields block like net-scp progress
-        # * final API: def upload(locals, remote, _options = {}, &_progress)
-        #
-        # @param locals [Array<String>,String] one or more local file or
-        #   directory paths
-        # @param remote [String] the base destination path on the remote host
-        # @return [Hash] report hash, keyed by the local MD5 digest
-        def upload(locals, remote)
-          files = nil
-          report = nil
-
-          elapsed1 = Benchmark.measure do
-            files = make_files_hash(Array(locals), remote)
-            report = check_files(files)
-            merge_with_report!(files, report)
-          end
-          total_size = total_base64_transfer_size(files)
-
-          elapsed2 = Benchmark.measure do
-            report = stream_upload_files(files) do |local_path, xfered|
-              yield xfered, total_size, local_path, remote if block_given?
-            end
-            merge_with_report!(files, report)
-          end
-
-          elapsed3 = Benchmark.measure do
-            report = decode_files(files)
-            merge_with_report!(files, report)
-            cleanup(files)
-          end
-
-          logger.debug(
-            "Uploaded #{files.keys.size} items " \
-            "dirty_check: #{duration(elapsed1.real)} " \
-            "stream_files: #{duration(elapsed2.real)} " \
-            "decode: #{duration(elapsed3.real)} " \
-          )
-
-          [total_size, files]
-        end
-
-        private
-
-        # @return [Integer] the maximum number of bytes that can be supplied on
-        #   a Windows CMD prompt without exceeded the maximum command line
-        #   length
-        # @api private
-        MAX_ENCODED_WRITE = 8000
-
-        # @return [String] the Array pack template for Base64 encoding a stream
-        #   of data
-        # @api private
-        BASE64_PACK = 'm0'.freeze
-
-        # @return [String] the directory where temporary upload artifacts are
-        #   persisted
-        # @api private
-        TEMP_UPLOAD_DIRECTORY = '$env:TEMP\\winrm-upload'.freeze
-
-        # @return [#debug,#debug?] the logger
-        # @api private
-        attr_reader :logger
-
-        # @return [Winrm::CommandExecutor] a WinRM CommandExecutor
-        # @api private
-        attr_reader :executor
-
-        # Adds an entry to a files Hash (keyed by local MD5 digest) for a
-        # directory. When a directory is added, a temporary Zip file is created
-        # containing the contents of the directory and any file-related data
-        # such as MD5 digest, size, etc. will be referring to the Zip file.
-        #
-        # @param hash [Hash] hash to be mutated
-        # @param dir [String] directory path to be Zipped and added
-        # @param remote [String] path to destination on remote host
-        # @api private
-        def add_directory_hash!(hash, dir, remote)
-          logger.debug "creating hash for directory #{remote}"
-          zip_io = TmpZip.new(dir, logger)
-          zip_md5 = md5sum(zip_io.path)
-
-          hash[zip_md5] = {
-            'src'     => dir,
-            'src_zip' => zip_io.path.to_s,
-            'zip_io'  => zip_io,
-            'tmpzip'  => "#{TEMP_UPLOAD_DIRECTORY}\\tmpzip-#{zip_md5}.zip",
-            'dst'     => "#{remote}\\#{File.basename(dir)}",
-            'size'    => File.size(zip_io.path)
-          }
-        end
-
-        # Adds an entry to a files Hash (keyed by local MD5 digest) for a file.
-        #
-        # @param hash [Hash] hash to be mutated
-        # @param local [String] file path
-        # @param remote [String] path to destination on remote host
-        # @api private
-        def add_file_hash!(hash, local, remote)
-          logger.debug "creating hash for file #{remote}"
-
-          # If the src has a file extension and the destination does not
-          # we can assume the caller specified the dest as a directory
-          if File.extname(local) != '' && File.extname(remote) == ''
-            remote = File.join(remote, File.basename(local))
-          end
-
-          hash[md5sum(local)] = {
-            'src'   => local,
-            'dst'   => remote,
-            'size'  => File.size(local)
-          }
-        end
-
-        # Runs the check_files PowerShell script against a collection of
-        # destination path/MD5 checksum pairs. The PowerShell script returns
-        # its results as a CSV-formatted report which is converted into a Ruby
-        # Hash.
-        #
-        # @param files [Hash] files hash, keyed by the local MD5 digest
-        # @return [Hash] a report hash, keyed by the local MD5 digest
-        # @api private
-        def check_files(files)
-          logger.debug 'Running check_files.ps1'
-          hash_file = create_remote_hash_file(check_files_ps_hash(files))
-          script = WinRM::FS::Scripts.render('check_files', hash_file: hash_file)
-          parse_response(executor.run_powershell_script(script))
-        end
-
-        # Constructs a collection of destination path/MD5 checksum pairs as a
-        # String representation of the contents of a PowerShell Hash Table.
-        #
-        # @param files [Hash] files hash, keyed by the local MD5 digest
-        # @return [String] the inner contents of a PowerShell Hash Table
-        # @api private
-        def check_files_ps_hash(files)
-          ps_hash(Hash[
-            files.map { |md5, data| [data.fetch('tmpzip', data['dst']), md5] }
-          ])
-        end
-
-        # Performs any final cleanup on the report Hash and removes any
-        # temporary files/resources used in the upload task.
-        #
-        # @param files [Hash] a files hash
-        # @api private
-        def cleanup(files)
-          files.select { |_, data| data.key?('zip_io') }.each do |md5, data|
-            data.fetch('zip_io').unlink
-            files.fetch(md5).delete('zip_io')
-            logger.debug "Cleaned up src_zip #{data['src_zip']}"
-          end
-        end
-
-        # Creates a remote Base64-encoded temporary file containing a
-        # PowerShell hash table.
-        #
-        # @param hash [String] a String representation of a PowerShell hash
-        #   table
-        # @return [String] the remote path to the temporary file
-        # @api private
-        def create_remote_hash_file(hash)
-          hash_file = "$env:TEMP\\hash-#{@id_generator.call}.txt"
-          hash.lines.each { |line| logger.debug line.chomp }
-          StringIO.open(hash) { |io| stream_upload(io, hash_file) }
-          hash_file
-        end
-
-        # Runs the decode_files PowerShell script against a collection of
-        # temporary file/destination path pairs. The PowerShell script returns
-        # its results as a CSV-formatted report which is converted into a Ruby
-        # Hash. The script will not be invoked if there are no "dirty" files
-        # present in the incoming files Hash.
-        #
-        # @param files [Hash] files hash, keyed by the local MD5 digest
-        # @return [Hash] a report hash, keyed by the local MD5 digest
-        # @api private
-        def decode_files(files)
-          decoded_files = decode_files_ps_hash(files)
-
-          if decoded_files == ps_hash({})
-            logger.debug 'No remote files to decode, skipping'
-            {}
-          else
-            logger.debug 'Running decode_files.ps1'
-            hash_file = create_remote_hash_file(decoded_files)
-            script = WinRM::FS::Scripts.render('decode_files', hash_file: hash_file)
-
-            parse_response(executor.run_powershell_script(script))
-          end
-        end
-
-        # Constructs a collection of temporary file/destination path pairs for
-        # all "dirty" files as a String representation of the contents of a
-        # PowerShell Hash Table. A "dirty" file is one which has the
-        # `"chk_dirty"` option set to `"True"` in the incoming files Hash.
-        #
-        # @param files [Hash] files hash, keyed by the local MD5 digest
-        # @return [String] the inner contents of a PowerShell Hash Table
-        # @api private
-        def decode_files_ps_hash(files)
-          file_data = files.select do |_, data|
-            data['chk_dirty'] == 'True' || data.key?('tmpzip')
-          end
-
-          result = file_data.map do |_, data|
-            val = { 'dst' => data['dst'] }
-            val['tmpzip'] = data['tmpzip'] if data['tmpzip']
-
-            [data['tmpfile'], val]
-          end
-
-          ps_hash(Hash[result])
-        end
-
-        # Returns a formatted string representing a duration in seconds.
-        #
-        # @param total [Integer] the total number of seconds
-        # @return [String] a formatted string of the form (XmYY.00s)
-        def duration(total)
-          total = 0 if total.nil?
-          minutes = (total / 60).to_i
-          seconds = (total - (minutes * 60))
-          format('(%dm%.2fs)', minutes, seconds)
-        end
-
-        # Contructs a Hash of files or directories, keyed by the local MD5
-        # digest. Each file entry has a source and destination set, at a
-        # minimum.
-        #
-        # @param locals [Array<String>] a collection of local files or
-        #   directories
-        # @param remote [String] the base destination path on the remote host
-        # @return [Hash] files hash, keyed by the local MD5 digest
-        # @api private
-        def make_files_hash(locals, remote)
-          hash = {}
-          locals.each do |local|
-            expanded = File.expand_path(local)
-            expanded += local[-1] if local.end_with?('/', '\\')
-
-            if File.file?(expanded)
-              add_file_hash!(hash, expanded, remote)
-            elsif File.directory?(expanded)
-              add_directory_hash!(hash, expanded, remote)
-            else
-              fail Errno::ENOENT, "No such file or directory #{expanded}"
-            end
-          end
-          hash
-        end
-
-        # @return [String] the MD5 digest of a local file
-        # @api private
-        def md5sum(local)
-          Digest::MD5.file(local).hexdigest
-        end
-
-        # Destructively merges a report Hash into an existing files Hash.
-        # **Note:** this method mutates the files Hash.
-        #
-        # @param files [Hash] files hash, keyed by the local MD5 digest
-        # @param report [Hash] report hash, keyed by the local MD5 digest
-        # @api private
-        def merge_with_report!(files, report)
-          files.merge!(report) { |_, oldval, newval| oldval.merge(newval) }
-        end
-
-        # @param depth [Integer] number of padding characters (default: `0`)
-        # @return [String] a whitespace padded string of the given length
-        # @api private
-        def pad(depth = 0)
-          ' ' * depth
-        end
-
-        # Parses CLIXML String into regular String (without any XML syntax).
-        # Inspired by https://github.com/WinRb/WinRM/issues/106.
-        #
-        # @param  clixml [String] clixml text
-        # @return [String] parsed clixml into String
-        def clixml_to_s(clixml)
-          doc = REXML::Document.new(clixml)
-          text = doc.get_elements('//S').map(&:text).join
-          text.gsub(/_x(\h\h\h\h)_/) do
-            code = Regexp.last_match[1]
-            code.hex.chr
-          end
-        end
-
-        # Parses response of a PowerShell script or CMD command which contains
-        # a CSV-formatted document in the standard output stream.
-        #
-        # @param output [WinRM::Output] output object with stdout, stderr, and
-        #   exit code
-        # @return [Hash] report hash, keyed by the local MD5 digest
-        # @api private
-        def parse_response(output)
-          exitcode = output[:exitcode]
-          stderr = output.stderr
-          if stderr.include?('The command line is too long')
-            # The powershell script which should result in `output` parameter
-            # is too long, remove some newlines, comments, etc from it.
-            fail StandardError, 'The command line is too long' \
-              ' (powershell script is too long)'
-          end
-          pretty_stderr = clixml_to_s(stderr)
-
-          if exitcode != 0
-            fail FileTransporterFailed, "[#{self.class}] Upload failed " \
-              "(exitcode: #{exitcode})\n#{pretty_stderr}"
-          elsif stderr != '\r\n' && stderr != ''
-            fail FileTransporterFailed, "[#{self.class}] Upload failed " \
-              "(exitcode: 0), but stderr present\n#{pretty_stderr}"
-          end
-
-          array = CSV.parse(output.stdout, headers: true).map(&:to_hash)
-          array.each { |h| h.each { |key, value| h[key] = nil if value == '' } }
-          Hash[array.map { |entry| [entry.fetch('src_md5'), entry] }]
-        end
-
-        # Converts a Ruby hash into a PowerShell hash table, represented in a
-        # String.
-        #
-        # @param obj [Object] source Hash or object when used in recursive
-        #   calls
-        # @param depth [Integer] padding depth, used in recursive calls
-        #   (default: `0`)
-        # @return [String] a PowerShell hash table
-        # @api private
-        def ps_hash(obj, depth = 0)
-          if obj.is_a?(Hash)
-            obj.map do |k, v|
-              %(#{pad(depth + 2)}#{ps_hash(k)} = #{ps_hash(v, depth + 2)})
-            end.join(";\n").insert(0, "@{\n").insert(-1, "\n#{pad(depth)}}")
-          else
-            %("#{obj}")
-          end
-        end
-
-        # Uploads an IO stream to a Base64-encoded destination file.
-        #
-        # **Implementation Note:** Some of the code in this method may appear
-        # slightly too dense and while adding additional variables would help,
-        # the code is written very precisely to avoid unwanted allocations
-        # which will bloat the Ruby VM's object space (and memory footprint).
-        # The goal here is to stream potentially large files to a remote host
-        # while not loading the entire file into memory first, then Base64
-        # encoding it--duplicating the file in memory again.
-        #
-        # @param input_io [#read] a readable stream or object to be uploaded
-        # @param dest [String] path to the destination file on the remote host
-        # @return [Integer,Integer] the number of resulting upload chunks and
-        #   the number of bytes transferred to the remote host
-        # @api private
-        def stream_upload(input_io, dest)
-          dest_cmd = dest.sub('$env:TEMP', '%TEMP%')
-          read_size = (MAX_ENCODED_WRITE.to_i / 4) * 3
-          chunk, bytes = 1, 0
-          buffer = ''
-          executor.run_cmd(%(echo|set /p=>"#{dest_cmd}")) # truncate empty file
-          while input_io.read(read_size, buffer)
-            bytes += (buffer.bytesize / 3 * 4)
-            executor.run_cmd([buffer].pack(BASE64_PACK)
-              .insert(0, 'echo ').concat(%( >> "#{dest_cmd}")))
-            logger.debug "Wrote chunk #{chunk} for #{dest}" if chunk % 25 == 0
-            chunk += 1
-            yield bytes if block_given?
-          end
-          buffer = nil # rubocop:disable Lint/UselessAssignment
-
-          [chunk - 1, bytes]
-        end
-
-        # Uploads a local file to a Base64-encoded temporary file.
-        #
-        # @param src [String] path to a local file
-        # @param tmpfile [String] path to the temporary file on the remote
-        #   host
-        # @return [Integer,Integer] the number of resulting upload chunks and
-        #   the number of bytes transferred to the remote host
-        # @api private
-        def stream_upload_file(src, tmpfile, &block)
-          logger.debug "Uploading #{src} to encoded tmpfile #{tmpfile}"
-          chunks, bytes = 0, 0
-          elapsed = Benchmark.measure do
-            File.open(src, 'rb') do |io|
-              chunks, bytes = stream_upload(io, tmpfile, &block)
-            end
-          end
-          logger.debug(
-            "Finished uploading #{src} to encoded tmpfile #{tmpfile} " \
-            "(#{bytes.to_f / 1000} KB over #{chunks} chunks) " \
-            "in #{duration(elapsed.real)}"
-          )
-
-          [chunks, bytes]
-        end
-
-        # Uploads a collection of "dirty" files to the remote host as
-        # Base64-encoded temporary files. A "dirty" file is one which has the
-        # `"chk_dirty"` option set to `"True"` in the incoming files Hash.
-        #
-        # @param files [Hash] files hash, keyed by the local MD5 digest
-        # @return [Hash] a report hash, keyed by the local MD5 digest
-        # @api private
-        def stream_upload_files(files)
-          response = {}
-          files.each do |md5, data|
-            src = data.fetch('src_zip', data['src'])
-            if data['chk_dirty'] == 'True'
-              tmpfile = "$env:TEMP\\b64-#{md5}.txt"
-              response[md5] = { 'tmpfile' => tmpfile }
-              chunks, bytes = stream_upload_file(src, tmpfile) do |xfered|
-                yield data['src'], xfered
-              end
-              response[md5]['chunks'] = chunks
-              response[md5]['xfered'] = bytes
-            else
-              logger.debug "File #{data['dst']} is up to date, skipping"
-            end
-          end
-          response
-        end
-
-        # Total by byte count to be transferred.
-        # Calculates count based on the sum of base64 encoded content size
-        # of all files base 64 that are dirty.
-        #
-        # @param files [Hash] files hash, keyed by the local MD5 digest
-        # @return [Fixnum] total byte size
-        # @api private
-        def total_base64_transfer_size(files)
-          size = 0
-          files.values.each { |file| size += file['size'] if file['chk_dirty'] == 'True' }
-          size / 3 * 4
-        end
-      end
-      # rubocop:enable MethodLength, AbcSize, ClassLength
-    end
-  end
-end
+# -*- encoding: utf-8 -*-
+#
+# Author:: Fletcher (<fnichol@nichol.ca>)
+#
+# Copyright (C) 2015, Fletcher Nichol
+#
+# 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 'benchmark'
+require 'csv'
+require 'digest'
+require 'securerandom'
+require 'stringio'
+
+require 'winrm-fs/core/tmp_zip'
+
+module WinRM
+  module FS
+    module Core
+      # Wrapped exception for any internally raised WinRM-related errors.
+      #
+      # @author Fletcher Nichol <fnichol@nichol.ca>
+      class FileTransporterFailed < ::WinRM::WinRMError; end
+      # rubocop:disable MethodLength, AbcSize, ClassLength
+
+      # Object which can upload one or more files or directories to a remote
+      # host over WinRM using PowerShell scripts and CMD commands. Note that
+      # this form of file transfer is *not* ideal and extremely costly on both
+      # the local and remote sides. Great pains are made to minimize round
+      # trips to  the remote host and to minimize the number of PowerShell
+      # sessions being invoked which can be 2 orders of magnitude more
+      # expensive than vanilla CMD commands.
+      #
+      # This object is supported by either a `CommandExecutor` instance as it
+      # depends on the `#run_cmd` and `#run_powershell_script` API contracts.
+      #
+      # An optional logger can be supplied, assuming it can respond to the
+      # `#debug` and `#debug?` messages.
+      #
+      # @author Fletcher Nichol <fnichol@nichol.ca>
+      # @author Matt Wrock <matt@mattwrock.com>
+      class FileTransporter
+        # Creates a FileTransporter given a CommandExecutor object.
+        #
+        # @param executor [CommandExecutor] a winrm CommandExecutor object
+        def initialize(executor, opts = {})
+          @executor  = executor
+          @logger = executor.service.logger
+          @id_generator = opts.fetch(:id_generator) { -> { SecureRandom.uuid } }
+        end
+
+        # Uploads a collection of files and/or directories to the remote host.
+        #
+        # **TODO Notes:**
+        # * options could specify zip mode, zip options, etc.
+        # * maybe option to set tmpfile base dir to override $env:PATH?
+        # * progress yields block like net-scp progress
+        # * final API: def upload(locals, remote, _options = {}, &_progress)
+        #
+        # @param locals [Array<String>,String] one or more local file or
+        #   directory paths
+        # @param remote [String] the base destination path on the remote host
+        # @return [Hash] report hash, keyed by the local MD5 digest
+        def upload(locals, remote)
+          files = nil
+          report = nil
+
+          elapsed1 = Benchmark.measure do
+            files = make_files_hash(Array(locals), remote)
+            report = check_files(files)
+            merge_with_report!(files, report)
+          end
+          total_size = total_base64_transfer_size(files)
+
+          elapsed2 = Benchmark.measure do
+            report = stream_upload_files(files) do |local_path, xfered|
+              yield xfered, total_size, local_path, remote if block_given?
+            end
+            merge_with_report!(files, report)
+          end
+
+          elapsed3 = Benchmark.measure do
+            report = decode_files(files)
+            merge_with_report!(files, report)
+            cleanup(files)
+          end
+
+          logger.debug(
+            "Uploaded #{files.keys.size} items " \
+            "dirty_check: #{duration(elapsed1.real)} " \
+            "stream_files: #{duration(elapsed2.real)} " \
+            "decode: #{duration(elapsed3.real)} " \
+          )
+
+          [total_size, files]
+        end
+
+        private
+
+        # @return [Integer] the maximum number of bytes that can be supplied on
+        #   a Windows CMD prompt without exceeded the maximum command line
+        #   length
+        # @api private
+        MAX_ENCODED_WRITE = 8000
+
+        # @return [String] the Array pack template for Base64 encoding a stream
+        #   of data
+        # @api private
+        BASE64_PACK = 'm0'.freeze
+
+        # @return [String] the directory where temporary upload artifacts are
+        #   persisted
+        # @api private
+        TEMP_UPLOAD_DIRECTORY = '$env:TEMP\\winrm-upload'.freeze
+
+        # @return [#debug,#debug?] the logger
+        # @api private
+        attr_reader :logger
+
+        # @return [Winrm::CommandExecutor] a WinRM CommandExecutor
+        # @api private
+        attr_reader :executor
+
+        # Adds an entry to a files Hash (keyed by local MD5 digest) for a
+        # directory. When a directory is added, a temporary Zip file is created
+        # containing the contents of the directory and any file-related data
+        # such as MD5 digest, size, etc. will be referring to the Zip file.
+        #
+        # @param hash [Hash] hash to be mutated
+        # @param dir [String] directory path to be Zipped and added
+        # @param remote [String] path to destination on remote host
+        # @api private
+        def add_directory_hash!(hash, dir, remote)
+          logger.debug "creating hash for directory #{remote}"
+          zip_io = TmpZip.new(dir, logger)
+          zip_md5 = md5sum(zip_io.path)
+
+          hash[zip_md5] = {
+            'src'     => dir,
+            'src_zip' => zip_io.path.to_s,
+            'zip_io'  => zip_io,
+            'tmpzip'  => "#{TEMP_UPLOAD_DIRECTORY}\\tmpzip-#{zip_md5}.zip",
+            'dst'     => "#{remote}\\#{File.basename(dir)}",
+            'size'    => File.size(zip_io.path)
+          }
+        end
+
+        # Adds an entry to a files Hash (keyed by local MD5 digest) for a file.
+        #
+        # @param hash [Hash] hash to be mutated
+        # @param local [String] file path
+        # @param remote [String] path to destination on remote host
+        # @api private
+        def add_file_hash!(hash, local, remote)
+          logger.debug "creating hash for file #{remote}"
+
+          # If the src has a file extension and the destination does not
+          # we can assume the caller specified the dest as a directory
+          if File.extname(local) != '' && File.extname(remote) == ''
+            remote = File.join(remote, File.basename(local))
+          end
+
+          hash[md5sum(local)] = {
+            'src'   => local,
+            'dst'   => remote,
+            'size'  => File.size(local)
+          }
+        end
+
+        # Runs the check_files PowerShell script against a collection of
+        # destination path/MD5 checksum pairs. The PowerShell script returns
+        # its results as a CSV-formatted report which is converted into a Ruby
+        # Hash.
+        #
+        # @param files [Hash] files hash, keyed by the local MD5 digest
+        # @return [Hash] a report hash, keyed by the local MD5 digest
+        # @api private
+        def check_files(files)
+          logger.debug 'Running check_files.ps1'
+          hash_file = create_remote_hash_file(check_files_ps_hash(files))
+          script = WinRM::FS::Scripts.render('check_files', hash_file: hash_file)
+          parse_response(executor.run_powershell_script(script))
+        end
+
+        # Constructs a collection of destination path/MD5 checksum pairs as a
+        # String representation of the contents of a PowerShell Hash Table.
+        #
+        # @param files [Hash] files hash, keyed by the local MD5 digest
+        # @return [String] the inner contents of a PowerShell Hash Table
+        # @api private
+        def check_files_ps_hash(files)
+          ps_hash(Hash[
+            files.map { |md5, data| [data.fetch('tmpzip', data['dst']), md5] }
+          ])
+        end
+
+        # Performs any final cleanup on the report Hash and removes any
+        # temporary files/resources used in the upload task.
+        #
+        # @param files [Hash] a files hash
+        # @api private
+        def cleanup(files)
+          files.select { |_, data| data.key?('zip_io') }.each do |md5, data|
+            data.fetch('zip_io').unlink
+            files.fetch(md5).delete('zip_io')
+            logger.debug "Cleaned up src_zip #{data['src_zip']}"
+          end
+        end
+
+        # Creates a remote Base64-encoded temporary file containing a
+        # PowerShell hash table.
+        #
+        # @param hash [String] a String representation of a PowerShell hash
+        #   table
+        # @return [String] the remote path to the temporary file
+        # @api private
+        def create_remote_hash_file(hash)
+          hash_file = "$env:TEMP\\hash-#{@id_generator.call}.txt"
+          hash.lines.each { |line| logger.debug line.chomp }
+          StringIO.open(hash) { |io| stream_upload(io, hash_file) }
+          hash_file
+        end
+
+        # Runs the decode_files PowerShell script against a collection of
+        # temporary file/destination path pairs. The PowerShell script returns
+        # its results as a CSV-formatted report which is converted into a Ruby
+        # Hash. The script will not be invoked if there are no "dirty" files
+        # present in the incoming files Hash.
+        #
+        # @param files [Hash] files hash, keyed by the local MD5 digest
+        # @return [Hash] a report hash, keyed by the local MD5 digest
+        # @api private
+        def decode_files(files)
+          decoded_files = decode_files_ps_hash(files)
+
+          if decoded_files == ps_hash({})
+            logger.debug 'No remote files to decode, skipping'
+            {}
+          else
+            logger.debug 'Running decode_files.ps1'
+            hash_file = create_remote_hash_file(decoded_files)
+            script = WinRM::FS::Scripts.render('decode_files', hash_file: hash_file)
+
+            parse_response(executor.run_powershell_script(script))
+          end
+        end
+
+        # Constructs a collection of temporary file/destination path pairs for
+        # all "dirty" files as a String representation of the contents of a
+        # PowerShell Hash Table. A "dirty" file is one which has the
+        # `"chk_dirty"` option set to `"True"` in the incoming files Hash.
+        #
+        # @param files [Hash] files hash, keyed by the local MD5 digest
+        # @return [String] the inner contents of a PowerShell Hash Table
+        # @api private
+        def decode_files_ps_hash(files)
+          file_data = files.select do |_, data|
+            data['chk_dirty'] == 'True' || data.key?('tmpzip')
+          end
+
+          i = 0
+          result = file_data.map do |_, data|
+            val = { 'dst' => data['dst'] }
+            val['tmpzip'] = data['tmpzip'] if data['tmpzip']
+
+            [data['tmpfile'] || "clean#{i += 1}", val]
+          end
+
+          ps_hash(Hash[result])
+        end
+
+        # Returns a formatted string representing a duration in seconds.
+        #
+        # @param total [Integer] the total number of seconds
+        # @return [String] a formatted string of the form (XmYY.00s)
+        def duration(total)
+          total = 0 if total.nil?
+          minutes = (total / 60).to_i
+          seconds = (total - (minutes * 60))
+          format('(%dm%.2fs)', minutes, seconds)
+        end
+
+        # Contructs a Hash of files or directories, keyed by the local MD5
+        # digest. Each file entry has a source and destination set, at a
+        # minimum.
+        #
+        # @param locals [Array<String>] a collection of local files or
+        #   directories
+        # @param remote [String] the base destination path on the remote host
+        # @return [Hash] files hash, keyed by the local MD5 digest
+        # @api private
+        def make_files_hash(locals, remote)
+          hash = {}
+          locals.each do |local|
+            expanded = File.expand_path(local)
+            expanded += local[-1] if local.end_with?('/', '\\')
+
+            if File.file?(expanded)
+              add_file_hash!(hash, expanded, remote)
+            elsif File.directory?(expanded)
+              add_directory_hash!(hash, expanded, remote)
+            else
+              fail Errno::ENOENT, "No such file or directory #{expanded}"
+            end
+          end
+          hash
+        end
+
+        # @return [String] the MD5 digest of a local file
+        # @api private
+        def md5sum(local)
+          Digest::MD5.file(local).hexdigest
+        end
+
+        # Destructively merges a report Hash into an existing files Hash.
+        # **Note:** this method mutates the files Hash.
+        #
+        # @param files [Hash] files hash, keyed by the local MD5 digest
+        # @param report [Hash] report hash, keyed by the local MD5 digest
+        # @api private
+        def merge_with_report!(files, report)
+          files.merge!(report) { |_, oldval, newval| oldval.merge(newval) }
+        end
+
+        # @param depth [Integer] number of padding characters (default: `0`)
+        # @return [String] a whitespace padded string of the given length
+        # @api private
+        def pad(depth = 0)
+          ' ' * depth
+        end
+
+        # Parses CLIXML String into regular String (without any XML syntax).
+        # Inspired by https://github.com/WinRb/WinRM/issues/106.
+        #
+        # @param  clixml [String] clixml text
+        # @return [String] parsed clixml into String
+        def clixml_to_s(clixml)
+          doc = REXML::Document.new(clixml)
+          text = doc.get_elements('//S').map(&:text).join
+          text.gsub(/_x(\h\h\h\h)_/) do
+            code = Regexp.last_match[1]
+            code.hex.chr
+          end
+        end
+
+        # Parses response of a PowerShell script or CMD command which contains
+        # a CSV-formatted document in the standard output stream.
+        #
+        # @param output [WinRM::Output] output object with stdout, stderr, and
+        #   exit code
+        # @return [Hash] report hash, keyed by the local MD5 digest
+        # @api private
+        def parse_response(output)
+          exitcode = output[:exitcode]
+          stderr = output.stderr
+          if stderr.include?('The command line is too long')
+            # The powershell script which should result in `output` parameter
+            # is too long, remove some newlines, comments, etc from it.
+            fail StandardError, 'The command line is too long' \
+              ' (powershell script is too long)'
+          end
+          pretty_stderr = clixml_to_s(stderr)
+
+          if exitcode != 0
+            fail FileTransporterFailed, "[#{self.class}] Upload failed " \
+              "(exitcode: #{exitcode})\n#{pretty_stderr}"
+          elsif stderr != '\r\n' && stderr != ''
+            fail FileTransporterFailed, "[#{self.class}] Upload failed " \
+              "(exitcode: 0), but stderr present\n#{pretty_stderr}"
+          end
+
+          array = CSV.parse(output.stdout, headers: true).map(&:to_hash)
+          array.each { |h| h.each { |key, value| h[key] = nil if value == '' } }
+          Hash[array.map { |entry| [entry.fetch('src_md5'), entry] }]
+        end
+
+        # Converts a Ruby hash into a PowerShell hash table, represented in a
+        # String.
+        #
+        # @param obj [Object] source Hash or object when used in recursive
+        #   calls
+        # @param depth [Integer] padding depth, used in recursive calls
+        #   (default: `0`)
+        # @return [String] a PowerShell hash table
+        # @api private
+        def ps_hash(obj, depth = 0)
+          if obj.is_a?(Hash)
+            obj.map do |k, v|
+              %(#{pad(depth + 2)}#{ps_hash(k)} = #{ps_hash(v, depth + 2)})
+            end.join(";\n").insert(0, "@{\n").insert(-1, "\n#{pad(depth)}}")
+          else
+            %("#{obj}")
+          end
+        end
+
+        # Uploads an IO stream to a Base64-encoded destination file.
+        #
+        # **Implementation Note:** Some of the code in this method may appear
+        # slightly too dense and while adding additional variables would help,
+        # the code is written very precisely to avoid unwanted allocations
+        # which will bloat the Ruby VM's object space (and memory footprint).
+        # The goal here is to stream potentially large files to a remote host
+        # while not loading the entire file into memory first, then Base64
+        # encoding it--duplicating the file in memory again.
+        #
+        # @param input_io [#read] a readable stream or object to be uploaded
+        # @param dest [String] path to the destination file on the remote host
+        # @return [Integer,Integer] the number of resulting upload chunks and
+        #   the number of bytes transferred to the remote host
+        # @api private
+        def stream_upload(input_io, dest)
+          dest_cmd = dest.sub('$env:TEMP', '%TEMP%')
+          read_size = (MAX_ENCODED_WRITE.to_i / 4) * 3
+          chunk, bytes = 1, 0
+          buffer = ''
+          executor.run_cmd(%(echo|set /p=>"#{dest_cmd}")) # truncate empty file
+          while input_io.read(read_size, buffer)
+            bytes += (buffer.bytesize / 3 * 4)
+            executor.run_cmd([buffer].pack(BASE64_PACK)
+              .insert(0, 'echo ').concat(%( >> "#{dest_cmd}")))
+            logger.debug "Wrote chunk #{chunk} for #{dest}" if chunk % 25 == 0
+            chunk += 1
+            yield bytes if block_given?
+          end
+          buffer = nil # rubocop:disable Lint/UselessAssignment
+
+          [chunk - 1, bytes]
+        end
+
+        # Uploads a local file to a Base64-encoded temporary file.
+        #
+        # @param src [String] path to a local file
+        # @param tmpfile [String] path to the temporary file on the remote
+        #   host
+        # @return [Integer,Integer] the number of resulting upload chunks and
+        #   the number of bytes transferred to the remote host
+        # @api private
+        def stream_upload_file(src, tmpfile, &block)
+          logger.debug "Uploading #{src} to encoded tmpfile #{tmpfile}"
+          chunks, bytes = 0, 0
+          elapsed = Benchmark.measure do
+            File.open(src, 'rb') do |io|
+              chunks, bytes = stream_upload(io, tmpfile, &block)
+            end
+          end
+          logger.debug(
+            "Finished uploading #{src} to encoded tmpfile #{tmpfile} " \
+            "(#{bytes.to_f / 1000} KB over #{chunks} chunks) " \
+            "in #{duration(elapsed.real)}"
+          )
+
+          [chunks, bytes]
+        end
+
+        # Uploads a collection of "dirty" files to the remote host as
+        # Base64-encoded temporary files. A "dirty" file is one which has the
+        # `"chk_dirty"` option set to `"True"` in the incoming files Hash.
+        #
+        # @param files [Hash] files hash, keyed by the local MD5 digest
+        # @return [Hash] a report hash, keyed by the local MD5 digest
+        # @api private
+        def stream_upload_files(files)
+          response = {}
+          files.each do |md5, data|
+            src = data.fetch('src_zip', data['src'])
+            if data['chk_dirty'] == 'True'
+              tmpfile = "$env:TEMP\\b64-#{md5}.txt"
+              response[md5] = { 'tmpfile' => tmpfile }
+              chunks, bytes = stream_upload_file(src, tmpfile) do |xfered|
+                yield data['src'], xfered
+              end
+              response[md5]['chunks'] = chunks
+              response[md5]['xfered'] = bytes
+            else
+              logger.debug "File #{data['dst']} is up to date, skipping"
+            end
+          end
+          response
+        end
+
+        # Total by byte count to be transferred.
+        # Calculates count based on the sum of base64 encoded content size
+        # of all files base 64 that are dirty.
+        #
+        # @param files [Hash] files hash, keyed by the local MD5 digest
+        # @return [Fixnum] total byte size
+        # @api private
+        def total_base64_transfer_size(files)
+          size = 0
+          files.values.each { |file| size += file['size'] if file['chk_dirty'] == 'True' }
+          size / 3 * 4
+        end
+      end
+      # rubocop:enable MethodLength, AbcSize, ClassLength
+    end
+  end
+end