vagrant-unbundled 2.2.6.1 → 2.2.6.2
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/Gemfile.lock +11 -9
- data/vagrant.gemspec +1 -1
- data/vendor/bundle/ruby/2.6.0/bundler/gems/vagrant-spec-abfc34474d12/vagrant-spec.gemspec +1 -1
- data/vendor/bundle/ruby/2.6.0/gems/erubi-1.9.0/CHANGELOG +79 -0
- data/vendor/bundle/ruby/2.6.0/gems/erubi-1.9.0/MIT-LICENSE +21 -0
- data/vendor/bundle/ruby/2.6.0/gems/erubi-1.9.0/README.rdoc +109 -0
- data/vendor/bundle/ruby/2.6.0/gems/erubi-1.9.0/Rakefile +78 -0
- data/vendor/bundle/ruby/2.6.0/gems/erubi-1.9.0/lib/erubi/capture_end.rb +52 -0
- data/vendor/bundle/ruby/2.6.0/gems/erubi-1.9.0/lib/erubi.rb +211 -0
- data/vendor/bundle/ruby/2.6.0/gems/erubi-1.9.0/test/test.rb +780 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/README.md +354 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/Rakefile +18 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/TODO +15 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/central_directory.rb +208 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/compressor.rb +9 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/constants.rb +63 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/crypto/encryption.rb +11 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/crypto/null_encryption.rb +43 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/crypto/traditional_encryption.rb +99 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/decompressor.rb +13 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/deflater.rb +34 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/dos_time.rb +48 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/entry.rb +700 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/entry_set.rb +86 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/errors.rb +18 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/extra_field/generic.rb +43 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/extra_field/ntfs.rb +90 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/extra_field/old_unix.rb +44 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/extra_field/universal_time.rb +47 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/extra_field/unix.rb +37 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/extra_field/zip64.rb +68 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/extra_field/zip64_placeholder.rb +15 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/extra_field.rb +101 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/file.rb +443 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/filesystem.rb +627 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/inflater.rb +66 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/input_stream.rb +173 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/ioextras/abstract_input_stream.rb +111 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/ioextras/abstract_output_stream.rb +43 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/ioextras.rb +36 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/null_compressor.rb +15 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/null_decompressor.rb +27 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/null_input_stream.rb +10 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/output_stream.rb +189 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/pass_thru_compressor.rb +23 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/pass_thru_decompressor.rb +40 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/streamable_directory.rb +15 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/streamable_stream.rb +56 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip/version.rb +3 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/lib/zip.rb +71 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/samples/example.rb +81 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/samples/example_filesystem.rb +31 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/samples/example_recursive.rb +54 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/samples/gtk_ruby_zip.rb +84 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/samples/qtzip.rb +92 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/samples/write_simple.rb +12 -0
- data/vendor/bundle/ruby/2.6.0/gems/rubyzip-2.0.0/samples/zipfind.rb +66 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/LICENSE +202 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/README.md +276 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/bin/rwinrm +90 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/connection.rb +84 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/connection_opts.rb +90 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/exceptions.rb +88 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/http/response_handler.rb +127 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/http/transport.rb +462 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/http/transport_factory.rb +64 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/output.rb +58 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/create_pipeline.xml.erb +167 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/fragment.rb +68 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/init_runspace_pool.xml.erb +224 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message.rb +128 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_data/base.rb +47 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_data/error_record.rb +66 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_data/pipeline_host_call.rb +30 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_data/pipeline_output.rb +48 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_data/pipeline_state.rb +38 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_data/runspacepool_host_call.rb +30 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_data/runspacepool_state.rb +37 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_data/session_capability.rb +34 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_data.rb +40 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_defragmenter.rb +62 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_factory.rb +86 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/message_fragmenter.rb +58 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/powershell_output_decoder.rb +142 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/receive_response_reader.rb +95 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/session_capability.xml.erb +7 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/psrp/uuid.rb +39 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/shells/base.rb +187 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/shells/cmd.rb +63 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/shells/power_shell.rb +206 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/shells/retryable.rb +44 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/shells/shell_factory.rb +56 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/version.rb +5 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/base.rb +57 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/cleanup_command.rb +60 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/close_shell.rb +49 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/command.rb +100 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/command_output.rb +75 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/command_output_decoder.rb +54 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/configuration.rb +44 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/create_pipeline.rb +64 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/create_shell.rb +115 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/header.rb +213 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/init_runspace_pool.rb +96 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/iso8601_duration.rb +58 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/keep_alive.rb +66 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/receive_response_reader.rb +128 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/send_data.rb +66 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/soap.rb +49 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/wql_pull.rb +54 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/wql_query.rb +98 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm/wsmv/write_stdin.rb +86 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-2.3.3/lib/winrm.rb +37 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-elevated-1.1.2/LICENSE +202 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-elevated-1.1.2/README.md +99 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-elevated-1.1.2/lib/winrm/shells/elevated.rb +108 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-elevated-1.1.2/lib/winrm-elevated/scripts/elevated_shell.ps1 +116 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-elevated-1.1.2/lib/winrm-elevated.rb +17 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/LICENSE +202 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/README.md +82 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/bin/rwinrmcp +87 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/core/file_transporter.rb +569 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/core/tmp_zip.rb +178 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/exceptions.rb +29 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/file_manager.rb +158 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/scripts/check_files.ps1.erb +49 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/scripts/checksum.ps1.erb +13 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/scripts/create_dir.ps1.erb +6 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/scripts/delete.ps1.erb +6 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/scripts/download.ps1.erb +17 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/scripts/exists.ps1.erb +10 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/scripts/extract_files.ps1.erb +52 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs/scripts/scripts.rb +47 -0
- data/vendor/bundle/ruby/2.6.0/gems/winrm-fs-1.3.4/lib/winrm-fs.rb +29 -0
- data/vendor/bundle/ruby/2.6.0/specifications/erubi-1.9.0.gemspec +38 -0
- data/vendor/bundle/ruby/2.6.0/specifications/rubyzip-2.0.0.gemspec +45 -0
- data/vendor/bundle/ruby/2.6.0/specifications/winrm-2.3.3.gemspec +73 -0
- data/vendor/bundle/ruby/2.6.0/specifications/winrm-elevated-1.1.2.gemspec +50 -0
- data/vendor/bundle/ruby/2.6.0/specifications/winrm-fs-1.3.4.gemspec +58 -0
- data/version.txt +1 -1
- metadata +142 -6
@@ -0,0 +1,95 @@
|
|
1
|
+
# Copyright 2016 Matt Wrock <matt@mattwrock.com>
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
|
15
|
+
require 'nori'
|
16
|
+
require_relative 'powershell_output_decoder'
|
17
|
+
require_relative 'message_defragmenter'
|
18
|
+
|
19
|
+
module WinRM
|
20
|
+
module PSRP
|
21
|
+
# Class for reading powershell responses in Receive_Response messages
|
22
|
+
class ReceiveResponseReader < WSMV::ReceiveResponseReader
|
23
|
+
# Creates a new ReceiveResponseReader
|
24
|
+
# @param transport [HttpTransport] The WinRM SOAP transport
|
25
|
+
# @param logger [Logger] The logger to log diagnostic messages to
|
26
|
+
def initialize(transport, logger)
|
27
|
+
super
|
28
|
+
@output_decoder = PowershellOutputDecoder.new
|
29
|
+
end
|
30
|
+
|
31
|
+
# Reads PSRP messages sent in one or more receive response messages
|
32
|
+
# @param wsmv_message [WinRM::WSMV::Base] A wsmv message to send to endpoint
|
33
|
+
# @param wait_for_done_state whether to poll for a CommandState of Done
|
34
|
+
# @yield [Message] PSRP Message in response
|
35
|
+
# @yieldreturn [Array<Message>] All messages in response
|
36
|
+
def read_message(wsmv_message, wait_for_done_state = false)
|
37
|
+
messages = []
|
38
|
+
defragmenter = MessageDefragmenter.new
|
39
|
+
read_response(wsmv_message, wait_for_done_state) do |stream|
|
40
|
+
message = defragmenter.defragment(stream[:text])
|
41
|
+
next unless message
|
42
|
+
|
43
|
+
if block_given?
|
44
|
+
yield message
|
45
|
+
else
|
46
|
+
messages.push(message)
|
47
|
+
end
|
48
|
+
end
|
49
|
+
messages unless block_given?
|
50
|
+
end
|
51
|
+
|
52
|
+
# Reads streams and returns decoded output
|
53
|
+
# @param wsmv_message [WinRM::WSMV::Base] A wsmv message to send to endpoint
|
54
|
+
# @yieldparam [string] standard out response text
|
55
|
+
# @yieldparam [string] standard error response text
|
56
|
+
# @yieldreturn [WinRM::Output] The command output
|
57
|
+
def read_output(wsmv_message)
|
58
|
+
with_output do |output|
|
59
|
+
read_message(wsmv_message, true) do |message|
|
60
|
+
exit_code = find_exit_code(message)
|
61
|
+
output.exitcode = exit_code if exit_code
|
62
|
+
decoded_text = @output_decoder.decode(message)
|
63
|
+
next unless decoded_text
|
64
|
+
|
65
|
+
out = { stream_type(message) => decoded_text }
|
66
|
+
output << out
|
67
|
+
yield [out[:stdout], out[:stderr]] if block_given?
|
68
|
+
end
|
69
|
+
end
|
70
|
+
end
|
71
|
+
|
72
|
+
private
|
73
|
+
|
74
|
+
def stream_type(message)
|
75
|
+
type = :stdout
|
76
|
+
case message.type
|
77
|
+
when WinRM::PSRP::Message::MESSAGE_TYPES[:error_record]
|
78
|
+
type = :stderr
|
79
|
+
when WinRM::PSRP::Message::MESSAGE_TYPES[:pipeline_host_call]
|
80
|
+
type = :stderr if message.data.include?('WriteError')
|
81
|
+
when WinRM::PSRP::Message::MESSAGE_TYPES[:pipeline_state]
|
82
|
+
type = :stderr if message.parsed_data.pipeline_state == WinRM::PSRP::MessageData::PipelineState::FAILED
|
83
|
+
end
|
84
|
+
type
|
85
|
+
end
|
86
|
+
|
87
|
+
def find_exit_code(message)
|
88
|
+
parsed = message.parsed_data
|
89
|
+
return nil unless parsed.is_a?(MessageData::PipelineHostCall)
|
90
|
+
|
91
|
+
parsed.method_parameters[:i32].to_i if parsed.method_identifier == 'SetShouldExit'
|
92
|
+
end
|
93
|
+
end
|
94
|
+
end
|
95
|
+
end
|
@@ -0,0 +1,39 @@
|
|
1
|
+
# Copyright 2016 Shawn Neal <sneal@sneal.net>
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
|
15
|
+
module WinRM
|
16
|
+
module PSRP
|
17
|
+
# UUID helper methods
|
18
|
+
module UUID
|
19
|
+
# Format a UUID into a GUID compatible byte array for Windows
|
20
|
+
#
|
21
|
+
# https://msdn.microsoft.com/en-us/library/windows/desktop/aa373931(v=vs.85).aspx
|
22
|
+
# typedef struct _GUID {
|
23
|
+
# DWORD Data1;
|
24
|
+
# WORD Data2;
|
25
|
+
# WORD Data3;
|
26
|
+
# BYTE Data4[8];
|
27
|
+
# } GUID;
|
28
|
+
#
|
29
|
+
# @param uuid [String] Canonical hex format with hypens.
|
30
|
+
# @return [Array<byte>] UUID in a Windows GUID compatible byte array layout.
|
31
|
+
def uuid_to_windows_guid_bytes(uuid)
|
32
|
+
return [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0] unless uuid
|
33
|
+
|
34
|
+
b = uuid.scan(/[0-9a-fA-F]{2}/).map { |x| x.to_i(16) }
|
35
|
+
b[0..3].reverse + b[4..5].reverse + b[6..7].reverse + b[8..15]
|
36
|
+
end
|
37
|
+
end
|
38
|
+
end
|
39
|
+
end
|
@@ -0,0 +1,187 @@
|
|
1
|
+
# Copyright 2016 Shawn Neal <sneal@sneal.net>
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
|
15
|
+
require_relative 'retryable'
|
16
|
+
require_relative '../http/transport'
|
17
|
+
require_relative '../wsmv/cleanup_command'
|
18
|
+
require_relative '../wsmv/close_shell'
|
19
|
+
require_relative '../wsmv/command'
|
20
|
+
require_relative '../wsmv/command_output'
|
21
|
+
require_relative '../wsmv/receive_response_reader'
|
22
|
+
require_relative '../wsmv/create_shell'
|
23
|
+
require_relative '../wsmv/soap'
|
24
|
+
|
25
|
+
module WinRM
|
26
|
+
module Shells
|
27
|
+
# Base class for remote shell
|
28
|
+
class Base
|
29
|
+
TOO_MANY_COMMANDS = '2150859174'.freeze
|
30
|
+
ERROR_OPERATION_ABORTED = '995'.freeze
|
31
|
+
SHELL_NOT_FOUND = '2150858843'.freeze
|
32
|
+
|
33
|
+
FAULTS_FOR_RESET = [
|
34
|
+
'2150858843', # Shell has been closed
|
35
|
+
'2147943418', # Error reading registry key
|
36
|
+
TOO_MANY_COMMANDS, # Maximum commands per user exceeded
|
37
|
+
].freeze
|
38
|
+
|
39
|
+
include Retryable
|
40
|
+
|
41
|
+
# Create a new Cmd shell
|
42
|
+
# @param connection_opts [ConnectionOpts] The WinRM connection options
|
43
|
+
# @param transport [HttpTransport] The WinRM SOAP transport
|
44
|
+
# @param logger [Logger] The logger to log diagnostic messages to
|
45
|
+
# @param shell_opts [Hash] Options targeted for the created shell
|
46
|
+
def initialize(connection_opts, transport, logger, shell_opts = {})
|
47
|
+
@connection_opts = connection_opts
|
48
|
+
@transport = transport
|
49
|
+
@logger = logger
|
50
|
+
@shell_opts = shell_opts
|
51
|
+
end
|
52
|
+
|
53
|
+
# @return [String] shell id of the currently opn shell or nil if shell is closed
|
54
|
+
attr_reader :shell_id
|
55
|
+
|
56
|
+
# @return [String] uri that SOAP calls use to identify shell type
|
57
|
+
attr_reader :shell_uri
|
58
|
+
|
59
|
+
# @return [ConnectionOpts] connection options of the shell
|
60
|
+
attr_reader :connection_opts
|
61
|
+
|
62
|
+
# @return [WinRM::HTTP::HttpTransport] transport used to talk with endpoint
|
63
|
+
attr_reader :transport
|
64
|
+
|
65
|
+
# @return [Logger] logger used for diagnostic messages
|
66
|
+
attr_reader :logger
|
67
|
+
|
68
|
+
# @return [Hash] Options targeted for the created shell
|
69
|
+
attr_reader :shell_opts
|
70
|
+
|
71
|
+
# Runs the specified command with optional arguments
|
72
|
+
# @param command [String] The command or executable to run
|
73
|
+
# @param arguments [Array] The optional command arguments
|
74
|
+
# @param block [&block] The optional callback for any realtime output
|
75
|
+
# @yieldparam [string] standard out response text
|
76
|
+
# @yieldparam [string] standard error response text
|
77
|
+
# @yieldreturn [WinRM::Output] The command output
|
78
|
+
def run(command, arguments = [], &block)
|
79
|
+
with_command_shell(command, arguments) do |shell, cmd|
|
80
|
+
response_reader.read_output(command_output_message(shell, cmd), &block)
|
81
|
+
end
|
82
|
+
end
|
83
|
+
|
84
|
+
# Closes the shell if one is open
|
85
|
+
def close
|
86
|
+
return unless shell_id
|
87
|
+
|
88
|
+
begin
|
89
|
+
self.class.close_shell(connection_opts, transport, shell_id)
|
90
|
+
rescue WinRMWSManFault => e
|
91
|
+
raise unless [ERROR_OPERATION_ABORTED, SHELL_NOT_FOUND].include?(e.fault_code)
|
92
|
+
end
|
93
|
+
remove_finalizer
|
94
|
+
@shell_id = nil
|
95
|
+
end
|
96
|
+
|
97
|
+
protected
|
98
|
+
|
99
|
+
def send_command(_command, _arguments)
|
100
|
+
raise NotImplementedError
|
101
|
+
end
|
102
|
+
|
103
|
+
def response_reader
|
104
|
+
raise NotImplementedError
|
105
|
+
end
|
106
|
+
|
107
|
+
def open_shell
|
108
|
+
raise NotImplementedError
|
109
|
+
end
|
110
|
+
|
111
|
+
def out_streams
|
112
|
+
raise NotImplementedError
|
113
|
+
end
|
114
|
+
|
115
|
+
def command_output_message(shell_id, command_id)
|
116
|
+
cmd_out_opts = {
|
117
|
+
shell_id: shell_id,
|
118
|
+
command_id: command_id,
|
119
|
+
shell_uri: shell_uri,
|
120
|
+
out_streams: out_streams
|
121
|
+
}
|
122
|
+
WinRM::WSMV::CommandOutput.new(connection_opts, cmd_out_opts)
|
123
|
+
end
|
124
|
+
|
125
|
+
def with_command_shell(command, arguments = [])
|
126
|
+
tries ||= 2
|
127
|
+
|
128
|
+
open unless shell_id
|
129
|
+
command_id = send_command(command, arguments)
|
130
|
+
logger.debug("[WinRM] creating command_id: #{command_id} on shell_id #{shell_id}")
|
131
|
+
yield shell_id, command_id
|
132
|
+
rescue WinRMWSManFault => e
|
133
|
+
raise unless FAULTS_FOR_RESET.include?(e.fault_code) && (tries -= 1) > 0
|
134
|
+
|
135
|
+
reset_on_error(e)
|
136
|
+
retry
|
137
|
+
ensure
|
138
|
+
cleanup_command(command_id) if command_id
|
139
|
+
end
|
140
|
+
|
141
|
+
private
|
142
|
+
|
143
|
+
def reset_on_error(error)
|
144
|
+
close if error.fault_code == TOO_MANY_COMMANDS
|
145
|
+
logger.debug('[WinRM] opening new shell since the current one was deleted')
|
146
|
+
@shell_id = nil
|
147
|
+
end
|
148
|
+
|
149
|
+
def cleanup_command(command_id)
|
150
|
+
logger.debug("[WinRM] cleaning up command_id: #{command_id} on shell_id #{shell_id}")
|
151
|
+
cleanup_msg = WinRM::WSMV::CleanupCommand.new(
|
152
|
+
connection_opts,
|
153
|
+
shell_uri: shell_uri,
|
154
|
+
shell_id: shell_id,
|
155
|
+
command_id: command_id
|
156
|
+
)
|
157
|
+
transport.send_request(cleanup_msg.build)
|
158
|
+
rescue WinRMWSManFault => e
|
159
|
+
raise unless [ERROR_OPERATION_ABORTED, SHELL_NOT_FOUND].include?(e.fault_code)
|
160
|
+
rescue WinRMHTTPTransportError => t
|
161
|
+
# dont let the cleanup raise so we dont lose any errors from the command
|
162
|
+
logger.info("[WinRM] #{t.status_code} returned in cleanup with error: #{t.message}")
|
163
|
+
end
|
164
|
+
|
165
|
+
def open
|
166
|
+
close
|
167
|
+
retryable(connection_opts[:retry_limit], connection_opts[:retry_delay]) do
|
168
|
+
logger.debug("[WinRM] opening remote shell on #{connection_opts[:endpoint]}")
|
169
|
+
@shell_id = open_shell
|
170
|
+
end
|
171
|
+
logger.debug("[WinRM] remote shell created with shell_id: #{shell_id}")
|
172
|
+
add_finalizer
|
173
|
+
end
|
174
|
+
|
175
|
+
def add_finalizer
|
176
|
+
ObjectSpace.define_finalizer(
|
177
|
+
self,
|
178
|
+
self.class.finalize(connection_opts, transport, shell_id)
|
179
|
+
)
|
180
|
+
end
|
181
|
+
|
182
|
+
def remove_finalizer
|
183
|
+
ObjectSpace.undefine_finalizer(self)
|
184
|
+
end
|
185
|
+
end
|
186
|
+
end
|
187
|
+
end
|
@@ -0,0 +1,63 @@
|
|
1
|
+
# Copyright 2016 Shawn Neal <sneal@sneal.net>
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
|
15
|
+
require_relative 'base'
|
16
|
+
|
17
|
+
module WinRM
|
18
|
+
module Shells
|
19
|
+
# Proxy to a remote cmd.exe shell
|
20
|
+
class Cmd < Base
|
21
|
+
include WinRM::WSMV::SOAP
|
22
|
+
class << self
|
23
|
+
def finalize(connection_opts, transport, shell_id)
|
24
|
+
proc { Cmd.close_shell(connection_opts, transport, shell_id) }
|
25
|
+
end
|
26
|
+
|
27
|
+
def close_shell(connection_opts, transport, shell_id)
|
28
|
+
msg = WinRM::WSMV::CloseShell.new(connection_opts, shell_id: shell_id)
|
29
|
+
transport.send_request(msg.build)
|
30
|
+
end
|
31
|
+
end
|
32
|
+
|
33
|
+
protected
|
34
|
+
|
35
|
+
def send_command(command, arguments)
|
36
|
+
cmd_msg = WinRM::WSMV::Command.new(
|
37
|
+
connection_opts,
|
38
|
+
shell_id: shell_id,
|
39
|
+
command: command,
|
40
|
+
arguments: arguments
|
41
|
+
)
|
42
|
+
resp_doc = transport.send_request(cmd_msg.build)
|
43
|
+
command_id = REXML::XPath.first(resp_doc, "//*[local-name() = 'CommandId']").text
|
44
|
+
logger.debug("[WinRM] Command created for #{command} with id: #{command_id}")
|
45
|
+
command_id
|
46
|
+
end
|
47
|
+
|
48
|
+
def response_reader
|
49
|
+
@response_reader ||= WinRM::WSMV::ReceiveResponseReader.new(transport, logger)
|
50
|
+
end
|
51
|
+
|
52
|
+
def open_shell
|
53
|
+
msg = WinRM::WSMV::CreateShell.new(connection_opts, shell_opts)
|
54
|
+
resp_doc = transport.send_request(msg.build)
|
55
|
+
REXML::XPath.first(resp_doc, "//*[@Name='ShellId']").text
|
56
|
+
end
|
57
|
+
|
58
|
+
def out_streams
|
59
|
+
%w[stdout stderr]
|
60
|
+
end
|
61
|
+
end
|
62
|
+
end
|
63
|
+
end
|
@@ -0,0 +1,206 @@
|
|
1
|
+
# Copyright 2016 Shawn Neal <sneal@sneal.net>
|
2
|
+
#
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
# you may not use this file except in compliance with the License.
|
5
|
+
# You may obtain a copy of the License at
|
6
|
+
#
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
#
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
# See the License for the specific language governing permissions and
|
13
|
+
# limitations under the License.
|
14
|
+
|
15
|
+
require 'securerandom'
|
16
|
+
require_relative 'base'
|
17
|
+
require_relative '../psrp/message_fragmenter'
|
18
|
+
require_relative '../psrp/receive_response_reader'
|
19
|
+
require_relative '../wsmv/configuration'
|
20
|
+
require_relative '../wsmv/create_pipeline'
|
21
|
+
require_relative '../wsmv/send_data'
|
22
|
+
require_relative '../wsmv/init_runspace_pool'
|
23
|
+
require_relative '../wsmv/keep_alive'
|
24
|
+
|
25
|
+
module WinRM
|
26
|
+
module Shells
|
27
|
+
# Proxy to a remote PowerShell instance
|
28
|
+
class Powershell < Base
|
29
|
+
include WinRM::WSMV::SOAP
|
30
|
+
|
31
|
+
class << self
|
32
|
+
def finalize(connection_opts, transport, shell_id)
|
33
|
+
proc { Powershell.close_shell(connection_opts, transport, shell_id) }
|
34
|
+
end
|
35
|
+
|
36
|
+
def close_shell(connection_opts, transport, shell_id)
|
37
|
+
msg = WinRM::WSMV::CloseShell.new(
|
38
|
+
connection_opts,
|
39
|
+
shell_id: shell_id,
|
40
|
+
shell_uri: WinRM::WSMV::Header::RESOURCE_URI_POWERSHELL
|
41
|
+
)
|
42
|
+
transport.send_request(msg.build)
|
43
|
+
end
|
44
|
+
end
|
45
|
+
|
46
|
+
# Create a new powershell shell
|
47
|
+
# @param connection_opts [ConnectionOpts] The WinRM connection options
|
48
|
+
# @param transport [HttpTransport] The WinRM SOAP transport
|
49
|
+
# @param logger [Logger] The logger to log diagnostic messages to
|
50
|
+
def initialize(connection_opts, transport, logger)
|
51
|
+
super
|
52
|
+
@shell_uri = WinRM::WSMV::Header::RESOURCE_URI_POWERSHELL
|
53
|
+
end
|
54
|
+
|
55
|
+
# Runs the specified command
|
56
|
+
# @param command [String] The powershell script to run
|
57
|
+
# @param block [&block] The optional callback for any realtime output
|
58
|
+
# @yield [Message] PSRP Message in response
|
59
|
+
# @yieldreturn [Array<Message>] All messages in response
|
60
|
+
def send_pipeline_command(command, &block)
|
61
|
+
with_command_shell(command) do |shell, cmd|
|
62
|
+
response_reader.read_message(command_output_message(shell, cmd), true, &block)
|
63
|
+
end
|
64
|
+
end
|
65
|
+
|
66
|
+
# calculate the maimum fragment size so that they will be as large as possible yet
|
67
|
+
# no greater than the max_envelope_size_kb on the end point. To calculate this
|
68
|
+
# threshold, we:
|
69
|
+
# - determine the maximum number of bytes accepted on the endpoint
|
70
|
+
# - subtract the non-fragment characters in the SOAP envelope
|
71
|
+
# - determine the number of bytes that could be base64 encded to the above length
|
72
|
+
# - subtract the fragment header bytes (ids, length, etc)
|
73
|
+
def max_fragment_blob_size
|
74
|
+
@max_fragment_blob_size ||= begin
|
75
|
+
fragment_header_length = 21
|
76
|
+
|
77
|
+
begin
|
78
|
+
max_fragment_bytes = (max_envelope_size_kb * 1024) - empty_pipeline_envelope.length
|
79
|
+
base64_deflated(max_fragment_bytes) - fragment_header_length
|
80
|
+
rescue WinRMWSManFault => e
|
81
|
+
# A non administrator user will encounter an access denied
|
82
|
+
# error attempting to query winrm configuration.
|
83
|
+
# we will assin a small default and adjust to a protocol
|
84
|
+
# appropriate max length when that info is available
|
85
|
+
raise unless e.fault_code == '5'
|
86
|
+
|
87
|
+
WinRM::PSRP::MessageFragmenter::DEFAULT_BLOB_LENGTH
|
88
|
+
rescue WinRMSoapFault
|
89
|
+
WinRM::PSRP::MessageFragmenter::DEFAULT_BLOB_LENGTH
|
90
|
+
end
|
91
|
+
end
|
92
|
+
end
|
93
|
+
|
94
|
+
protected
|
95
|
+
|
96
|
+
def response_reader
|
97
|
+
@response_reader ||= WinRM::PSRP::ReceiveResponseReader.new(transport, logger)
|
98
|
+
end
|
99
|
+
|
100
|
+
def send_command(command, _arguments)
|
101
|
+
command_id = SecureRandom.uuid.to_s.upcase
|
102
|
+
command += "\r\nif (!$?) { if($LASTEXITCODE) { exit $LASTEXITCODE } else { exit 1 } }"
|
103
|
+
message = PSRP::MessageFactory.create_pipeline_message(@runspace_id, command_id, command)
|
104
|
+
fragmenter.fragment(message) do |fragment|
|
105
|
+
command_args = [connection_opts, shell_id, command_id, fragment]
|
106
|
+
if fragment.start_fragment
|
107
|
+
resp_doc = transport.send_request(WinRM::WSMV::CreatePipeline.new(*command_args).build)
|
108
|
+
command_id = REXML::XPath.first(resp_doc, "//*[local-name() = 'CommandId']").text
|
109
|
+
else
|
110
|
+
transport.send_request(WinRM::WSMV::SendData.new(*command_args).build)
|
111
|
+
end
|
112
|
+
end
|
113
|
+
|
114
|
+
logger.debug("[WinRM] Command created for #{command} with id: #{command_id}")
|
115
|
+
command_id
|
116
|
+
end
|
117
|
+
|
118
|
+
def open_shell
|
119
|
+
@runspace_id = SecureRandom.uuid.to_s.upcase
|
120
|
+
runspace_msg = WinRM::WSMV::InitRunspacePool.new(
|
121
|
+
connection_opts,
|
122
|
+
@runspace_id,
|
123
|
+
open_shell_payload(@runspace_id)
|
124
|
+
)
|
125
|
+
resp_doc = transport.send_request(runspace_msg.build)
|
126
|
+
shell_id = REXML::XPath.first(resp_doc, "//*[@Name='ShellId']").text
|
127
|
+
wait_for_running(shell_id)
|
128
|
+
shell_id
|
129
|
+
end
|
130
|
+
|
131
|
+
def out_streams
|
132
|
+
%w[stdout]
|
133
|
+
end
|
134
|
+
|
135
|
+
private
|
136
|
+
|
137
|
+
def base64_deflated(inflated_length)
|
138
|
+
inflated_length / 4 * 3
|
139
|
+
end
|
140
|
+
|
141
|
+
def empty_pipeline_envelope
|
142
|
+
WinRM::WSMV::CreatePipeline.new(
|
143
|
+
connection_opts,
|
144
|
+
'00000000-0000-0000-0000-000000000000',
|
145
|
+
'00000000-0000-0000-0000-000000000000'
|
146
|
+
).build
|
147
|
+
end
|
148
|
+
|
149
|
+
def max_envelope_size_kb
|
150
|
+
@max_envelope_size_kb ||= begin
|
151
|
+
config_msg = WinRM::WSMV::Configuration.new(connection_opts)
|
152
|
+
msg = config_msg.build
|
153
|
+
resp_doc = transport.send_request(msg)
|
154
|
+
REXML::XPath.first(resp_doc, "//*[local-name() = 'MaxEnvelopeSizekb']").text.to_i
|
155
|
+
ensure
|
156
|
+
logger.debug("[WinRM] Endpoint doesn't support config request for MaxEnvelopsizekb")
|
157
|
+
end
|
158
|
+
# rubocop:enable Layout/RescueEnsureAlignment
|
159
|
+
end
|
160
|
+
|
161
|
+
def open_shell_payload(shell_id)
|
162
|
+
[
|
163
|
+
WinRM::PSRP::MessageFactory.session_capability_message(shell_id),
|
164
|
+
WinRM::PSRP::MessageFactory.init_runspace_pool_message(shell_id)
|
165
|
+
].map do |message|
|
166
|
+
fragmenter.fragment(message).bytes
|
167
|
+
end.flatten
|
168
|
+
end
|
169
|
+
|
170
|
+
def wait_for_running(shell_id)
|
171
|
+
state = WinRM::PSRP::MessageData::RunspacepoolState::OPENING
|
172
|
+
keepalive_msg = WinRM::WSMV::KeepAlive.new(connection_opts, shell_id)
|
173
|
+
|
174
|
+
# 2 is "openned". if we start issuing commands while in "openning" the runspace
|
175
|
+
# seems to hang
|
176
|
+
until state == WinRM::PSRP::MessageData::RunspacepoolState::OPENED
|
177
|
+
response_reader.read_message(keepalive_msg) do |message|
|
178
|
+
logger.debug("[WinRM] polling for pipeline state. message: #{message.inspect}")
|
179
|
+
parsed = message.parsed_data
|
180
|
+
case parsed
|
181
|
+
when WinRM::PSRP::MessageData::RunspacepoolState
|
182
|
+
state = parsed.runspace_state
|
183
|
+
when WinRM::PSRP::MessageData::SessionCapability
|
184
|
+
# if the user lacks admin privileges, we cannot query the MaxEnvelopeSizeKB
|
185
|
+
# on the server and will assign to a "best effort" default based on protocol version
|
186
|
+
if fragmenter.max_blob_length == WinRM::PSRP::MessageFragmenter::DEFAULT_BLOB_LENGTH
|
187
|
+
fragmenter.max_blob_length = default_protocol_envelope_size(parsed.protocol_version)
|
188
|
+
end
|
189
|
+
end
|
190
|
+
end
|
191
|
+
end
|
192
|
+
end
|
193
|
+
|
194
|
+
# Powershell v2.0 has a protocol version of 2.1
|
195
|
+
# which defaults to a 150 MaxEnvelopeSizeKB
|
196
|
+
# later versions default to 500
|
197
|
+
def default_protocol_envelope_size(protocol_version)
|
198
|
+
protocol_version > '2.1' ? 512000 : 153600
|
199
|
+
end
|
200
|
+
|
201
|
+
def fragmenter
|
202
|
+
@fragmenter ||= WinRM::PSRP::MessageFragmenter.new(max_fragment_blob_size)
|
203
|
+
end
|
204
|
+
end
|
205
|
+
end
|
206
|
+
end
|
@@ -0,0 +1,44 @@
|
|
1
|
+
# Copyright 2016 Shawn Neal <sneal@sneal.net>
|
2
|
+
# Copyright 2015 Matt Wrock <matt@mattwrock.com>
|
3
|
+
#
|
4
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
5
|
+
# you may not use this file except in compliance with the License.
|
6
|
+
# You may obtain a copy of the License at
|
7
|
+
#
|
8
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
9
|
+
#
|
10
|
+
# Unless required by applicable law or agreed to in writing, software
|
11
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
12
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
13
|
+
# See the License for the specific language governing permissions and
|
14
|
+
# limitations under the License.
|
15
|
+
|
16
|
+
require_relative '../exceptions'
|
17
|
+
|
18
|
+
module WinRM
|
19
|
+
module Shells
|
20
|
+
# Shell mixin for retrying an operation
|
21
|
+
module Retryable
|
22
|
+
RETRYABLE_EXCEPTIONS = lambda do
|
23
|
+
[
|
24
|
+
Errno::EACCES, Errno::EADDRINUSE, Errno::ECONNREFUSED, Errno::ETIMEDOUT,
|
25
|
+
Errno::ECONNRESET, Errno::ENETUNREACH, Errno::EHOSTUNREACH, ::WinRM::WinRMWSManFault,
|
26
|
+
::WinRM::WinRMHTTPTransportError, ::WinRM::WinRMAuthorizationError,
|
27
|
+
HTTPClient::KeepAliveDisconnected, HTTPClient::ConnectTimeoutError
|
28
|
+
].freeze
|
29
|
+
end
|
30
|
+
|
31
|
+
# Retries the operation a specified number of times with a delay between
|
32
|
+
# @param retries [Integer] The number of times to retry
|
33
|
+
# @param delay [Integer] The number of seconds to wait between retry attempts
|
34
|
+
def retryable(retries, delay)
|
35
|
+
yield
|
36
|
+
rescue *RETRYABLE_EXCEPTIONS.call
|
37
|
+
raise unless (retries -= 1) > 0
|
38
|
+
|
39
|
+
sleep(delay)
|
40
|
+
retry
|
41
|
+
end
|
42
|
+
end
|
43
|
+
end
|
44
|
+
end
|