winrm 2.0.3 → 2.1.0

data/lib/winrm/wsmv/send_data.rb

@@ -1,68 +1,68 @@
-# -*- encoding: utf-8 -*-
-#
-# Copyright 2016 Matt Wrock <matt@mattwrock.com>
-#
-# 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_relative 'base'
-
-module WinRM
-  module WSMV
-    # Subclasses base, sends additional data to WinRM
-    class SendData < Base
-      attr_accessor :shell_id, :command_id, :fragment
-
-      def initialize(session_opts, shell_id, command_id, fragment)
-        @session_opts = session_opts
-        @shell_id = shell_id
-        @command_id = command_id
-        @fragment = fragment
-      end
-
-      protected
-
-      def create_header(header)
-        header << Gyoku.xml(command_headers)
-      end
-
-      def create_body(body)
-        body.tag!("#{NS_WIN_SHELL}:Send") do |cl|
-          cl << Gyoku.xml(command_body)
-        end
-      end
-
-      private
-
-      def command_body
-        {
-          "#{NS_WIN_SHELL}:Stream" => encode_bytes(fragment.bytes),
-          :attributes! => {
-            "#{NS_WIN_SHELL}:Stream" => {
-              'Name' => 'stdin',
-              'CommandId' => command_id
-            }
-          }
-        }
-      end
-
-      def command_headers
-        merge_headers(
-          shared_headers(@session_opts),
-          resource_uri_shell(RESOURCE_URI_POWERSHELL),
-          action_send,
-          selector_shell_id(shell_id)
-        )
-      end
-    end
-  end
-end
+# -*- encoding: utf-8 -*-
+#
+# Copyright 2016 Matt Wrock <matt@mattwrock.com>
+#
+# 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_relative 'base'
+
+module WinRM
+  module WSMV
+    # Subclasses base, sends additional data to WinRM
+    class SendData < Base
+      attr_accessor :shell_id, :command_id, :fragment
+
+      def initialize(session_opts, shell_id, command_id, fragment)
+        @session_opts = session_opts
+        @shell_id = shell_id
+        @command_id = command_id
+        @fragment = fragment
+      end
+
+      protected
+
+      def create_header(header)
+        header << Gyoku.xml(command_headers)
+      end
+
+      def create_body(body)
+        body.tag!("#{NS_WIN_SHELL}:Send") do |cl|
+          cl << Gyoku.xml(command_body)
+        end
+      end
+
+      private
+
+      def command_body
+        {
+          "#{NS_WIN_SHELL}:Stream" => encode_bytes(fragment.bytes),
+          :attributes! => {
+            "#{NS_WIN_SHELL}:Stream" => {
+              'Name' => 'stdin',
+              'CommandId' => command_id
+            }
+          }
+        }
+      end
+
+      def command_headers
+        merge_headers(
+          shared_headers(@session_opts),
+          resource_uri_shell(RESOURCE_URI_POWERSHELL),
+          action_send,
+          selector_shell_id(shell_id)
+        )
+      end
+    end
+  end
+end

data/lib/winrm/wsmv/soap.rb

@@ -1,51 +1,51 @@
-# -*- encoding: utf-8 -*-
-#
-# Copyright 2010 Dan Wanek <dan.wanek@gmail.com>
-# Copyright 2016 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.
-
-module WinRM
-  module WSMV
-    # WSMV SOAP namespaces mixin
-    module SOAP
-      NS_SOAP_ENV    = 's'.freeze   # http://www.w3.org/2003/05/soap-envelope
-      NS_ADDRESSING  = 'a'.freeze   # http://schemas.xmlsoap.org/ws/2004/08/addressing
-      NS_CIMBINDING  = 'b'.freeze   # http://schemas.dmtf.org/wbem/wsman/1/cimbinding.xsd
-      NS_ENUM        = 'n'.freeze   # http://schemas.xmlsoap.org/ws/2004/09/enumeration
-      NS_TRANSFER    = 'x'.freeze   # http://schemas.xmlsoap.org/ws/2004/09/transfer
-      NS_WSMAN_DMTF  = 'w'.freeze   # http://schemas.dmtf.org/wbem/wsman/1/wsman.xsd
-      NS_WSMAN_MSFT  = 'p'.freeze   # http://schemas.microsoft.com/wbem/wsman/1/wsman.xsd
-      NS_SCHEMA_INST = 'xsi'.freeze # http://www.w3.org/2001/XMLSchema-instance
-      NS_WIN_SHELL   = 'rsp'.freeze # http://schemas.microsoft.com/wbem/wsman/1/windows/shell
-      NS_WSMAN_FAULT = 'f'.freeze   # http://schemas.microsoft.com/wbem/wsman/1/wsmanfault
-      NS_WSMAN_CONF  = 'cfg'.freeze # http://schemas.microsoft.com/wbem/wsman/1/config
-
-      def namespaces
-        @namespaces ||= {
-          'xmlns:xsd' => 'http://www.w3.org/2001/XMLSchema',
-          'xmlns:xsi' => 'http://www.w3.org/2001/XMLSchema-instance',
-          'xmlns:env' => 'http://www.w3.org/2003/05/soap-envelope',
-          "xmlns:#{NS_ADDRESSING}" => 'http://schemas.xmlsoap.org/ws/2004/08/addressing',
-          "xmlns:#{NS_CIMBINDING}" => 'http://schemas.dmtf.org/wbem/wsman/1/cimbinding.xsd',
-          "xmlns:#{NS_ENUM}" => 'http://schemas.xmlsoap.org/ws/2004/09/enumeration',
-          "xmlns:#{NS_TRANSFER}" => 'http://schemas.xmlsoap.org/ws/2004/09/transfer',
-          "xmlns:#{NS_WSMAN_DMTF}" => 'http://schemas.dmtf.org/wbem/wsman/1/wsman.xsd',
-          "xmlns:#{NS_WSMAN_MSFT}" => 'http://schemas.microsoft.com/wbem/wsman/1/wsman.xsd',
-          "xmlns:#{NS_WIN_SHELL}" => 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell',
-          "xmlns:#{NS_WSMAN_CONF}" => 'http://schemas.microsoft.com/wbem/wsman/1/config'
-        }
-      end
-    end
-  end
-end
+# -*- encoding: utf-8 -*-
+#
+# Copyright 2010 Dan Wanek <dan.wanek@gmail.com>
+# Copyright 2016 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.
+
+module WinRM
+  module WSMV
+    # WSMV SOAP namespaces mixin
+    module SOAP
+      NS_SOAP_ENV    = 's'.freeze   # http://www.w3.org/2003/05/soap-envelope
+      NS_ADDRESSING  = 'a'.freeze   # http://schemas.xmlsoap.org/ws/2004/08/addressing
+      NS_CIMBINDING  = 'b'.freeze   # http://schemas.dmtf.org/wbem/wsman/1/cimbinding.xsd
+      NS_ENUM        = 'n'.freeze   # http://schemas.xmlsoap.org/ws/2004/09/enumeration
+      NS_TRANSFER    = 'x'.freeze   # http://schemas.xmlsoap.org/ws/2004/09/transfer
+      NS_WSMAN_DMTF  = 'w'.freeze   # http://schemas.dmtf.org/wbem/wsman/1/wsman.xsd
+      NS_WSMAN_MSFT  = 'p'.freeze   # http://schemas.microsoft.com/wbem/wsman/1/wsman.xsd
+      NS_SCHEMA_INST = 'xsi'.freeze # http://www.w3.org/2001/XMLSchema-instance
+      NS_WIN_SHELL   = 'rsp'.freeze # http://schemas.microsoft.com/wbem/wsman/1/windows/shell
+      NS_WSMAN_FAULT = 'f'.freeze   # http://schemas.microsoft.com/wbem/wsman/1/wsmanfault
+      NS_WSMAN_CONF  = 'cfg'.freeze # http://schemas.microsoft.com/wbem/wsman/1/config
+
+      def namespaces
+        @namespaces ||= {
+          'xmlns:xsd' => 'http://www.w3.org/2001/XMLSchema',
+          'xmlns:xsi' => 'http://www.w3.org/2001/XMLSchema-instance',
+          'xmlns:env' => 'http://www.w3.org/2003/05/soap-envelope',
+          "xmlns:#{NS_ADDRESSING}" => 'http://schemas.xmlsoap.org/ws/2004/08/addressing',
+          "xmlns:#{NS_CIMBINDING}" => 'http://schemas.dmtf.org/wbem/wsman/1/cimbinding.xsd',
+          "xmlns:#{NS_ENUM}" => 'http://schemas.xmlsoap.org/ws/2004/09/enumeration',
+          "xmlns:#{NS_TRANSFER}" => 'http://schemas.xmlsoap.org/ws/2004/09/transfer',
+          "xmlns:#{NS_WSMAN_DMTF}" => 'http://schemas.dmtf.org/wbem/wsman/1/wsman.xsd',
+          "xmlns:#{NS_WSMAN_MSFT}" => 'http://schemas.microsoft.com/wbem/wsman/1/wsman.xsd',
+          "xmlns:#{NS_WIN_SHELL}" => 'http://schemas.microsoft.com/wbem/wsman/1/windows/shell',
+          "xmlns:#{NS_WSMAN_CONF}" => 'http://schemas.microsoft.com/wbem/wsman/1/config'
+        }
+      end
+    end
+  end
+end

data/lib/winrm/wsmv/wql_query.rb

@@ -1,79 +1,79 @@
-# -*- encoding: utf-8 -*-
-#
-# Copyright 2016 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 'nori'
-require_relative 'base'
-
-module WinRM
-  module WSMV
-    # WSMV message to query Windows via WQL
-    class WqlQuery < Base
-      def initialize(session_opts, wql)
-        @session_opts = session_opts
-        @wql = wql
-      end
-
-      def process_response(response)
-        parser = Nori.new(
-          parser: :rexml,
-          advanced_typecasting: false,
-          convert_tags_to: ->(tag) { tag.snakecase.to_sym },
-          strip_namespaces: true
-        )
-        hresp = parser.parse(response.to_s)[:envelope][:body]
-
-        # Normalize items so the type always has an array even if it's just a single item.
-        items = {}
-        if hresp[:enumerate_response][:items]
-          hresp[:enumerate_response][:items].each_pair do |k, v|
-            items[k] = v.is_a?(Array) ? v : [v]
-          end
-        end
-        items
-      end
-
-      protected
-
-      def create_header(header)
-        header << Gyoku.xml(wql_header)
-      end
-
-      def create_body(body)
-        body.tag!("#{NS_ENUM}:Enumerate") { |en| en << Gyoku.xml(wql_body) }
-      end
-
-      private
-
-      def wql_header
-        merge_headers(shared_headers(@session_opts), resource_uri_wmi, action_enumerate)
-      end
-
-      def wql_body
-        {
-          "#{NS_WSMAN_DMTF}:OptimizeEnumeration" => nil,
-          "#{NS_WSMAN_DMTF}:MaxElements" => '32000',
-          "#{NS_WSMAN_DMTF}:Filter" => @wql,
-          "#{NS_WSMAN_MSFT}:SessionId" => "uuid:#{@session_opts[:session_id]}",
-          :attributes! => {
-            "#{NS_WSMAN_DMTF}:Filter" => {
-              'Dialect' => 'http://schemas.microsoft.com/wbem/wsman/1/WQL'
-            }
-          }
-        }
-      end
-    end
-  end
-end
+# -*- encoding: utf-8 -*-
+#
+# Copyright 2016 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 'nori'
+require_relative 'base'
+
+module WinRM
+  module WSMV
+    # WSMV message to query Windows via WQL
+    class WqlQuery < Base
+      def initialize(session_opts, wql)
+        @session_opts = session_opts
+        @wql = wql
+      end
+
+      def process_response(response)
+        parser = Nori.new(
+          parser: :rexml,
+          advanced_typecasting: false,
+          convert_tags_to: ->(tag) { tag.snakecase.to_sym },
+          strip_namespaces: true
+        )
+        hresp = parser.parse(response.to_s)[:envelope][:body]
+
+        # Normalize items so the type always has an array even if it's just a single item.
+        items = {}
+        if hresp[:enumerate_response][:items]
+          hresp[:enumerate_response][:items].each_pair do |k, v|
+            items[k] = v.is_a?(Array) ? v : [v]
+          end
+        end
+        items
+      end
+
+      protected
+
+      def create_header(header)
+        header << Gyoku.xml(wql_header)
+      end
+
+      def create_body(body)
+        body.tag!("#{NS_ENUM}:Enumerate") { |en| en << Gyoku.xml(wql_body) }
+      end
+
+      private
+
+      def wql_header
+        merge_headers(shared_headers(@session_opts), resource_uri_wmi, action_enumerate)
+      end
+
+      def wql_body
+        {
+          "#{NS_WSMAN_DMTF}:OptimizeEnumeration" => nil,
+          "#{NS_WSMAN_DMTF}:MaxElements" => '32000',
+          "#{NS_WSMAN_DMTF}:Filter" => @wql,
+          "#{NS_WSMAN_MSFT}:SessionId" => "uuid:#{@session_opts[:session_id]}",
+          :attributes! => {
+            "#{NS_WSMAN_DMTF}:Filter" => {
+              'Dialect' => 'http://schemas.microsoft.com/wbem/wsman/1/WQL'
+            }
+          }
+        }
+      end
+    end
+  end
+end

data/lib/winrm/wsmv/write_stdin.rb

@@ -1,88 +1,88 @@
-# -*- encoding: utf-8 -*-
-#
-# Copyright 2016 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 'base64'
-require_relative 'base'
-
-module WinRM
-  module WSMV
-    # WSMV message to send stdin to a remote shell
-    class WriteStdin < Base
-      def initialize(session_opts, stdin_opts)
-        validate_opts(session_opts, stdin_opts)
-        init_ops(session_opts, stdin_opts)
-      end
-
-      protected
-
-      def create_header(header)
-        header << Gyoku.xml(stdin_headers)
-      end
-
-      def create_body(body)
-        body << Gyoku.xml(stdin_body)
-      end
-
-      private
-
-      def init_ops(session_opts, stdin_opts)
-        @session_opts = session_opts
-        @shell_id = stdin_opts[:shell_id]
-        @command_id = stdin_opts[:command_id]
-        @stdin = stdin_opts[:stdin]
-        @shell_uri = stdin_opts[:shell_uri] || RESOURCE_URI_CMD
-      end
-
-      def validate_opts(session_opts, stdin_opts)
-        raise 'session_opts is required' unless session_opts
-        raise 'stdin_opts[:shell_id] is required' unless stdin_opts[:shell_id]
-        raise 'stdin_opts[:command_id] is required' unless stdin_opts[:command_id]
-        raise 'stdin_opts[:stdin] is required' unless stdin_opts[:stdin]
-      end
-
-      def stdin_headers
-        merge_headers(shared_headers(@session_opts),
-                      resource_uri_shell(@shell_uri),
-                      action_send,
-                      selector_shell_id(@shell_id))
-      end
-
-      def action_send
-        {
-          "#{NS_ADDRESSING}:Action" =>
-          'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Send',
-          :attributes! => {
-            "#{NS_ADDRESSING}:Action" => {
-              'mustUnderstand' => true
-            }
-          }
-        }
-      end
-
-      def stdin_body
-        {
-          "#{NS_WIN_SHELL}:Send" => {
-            "#{NS_WIN_SHELL}:Stream" => {
-              '@Name' => 'stdin',
-              '@CommandId' => @command_id,
-              :content! => Base64.encode64(@stdin)
-            }
-          }
-        }
-      end
-    end
-  end
-end
+# -*- encoding: utf-8 -*-
+#
+# Copyright 2016 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 'base64'
+require_relative 'base'
+
+module WinRM
+  module WSMV
+    # WSMV message to send stdin to a remote shell
+    class WriteStdin < Base
+      def initialize(session_opts, stdin_opts)
+        validate_opts(session_opts, stdin_opts)
+        init_ops(session_opts, stdin_opts)
+      end
+
+      protected
+
+      def create_header(header)
+        header << Gyoku.xml(stdin_headers)
+      end
+
+      def create_body(body)
+        body << Gyoku.xml(stdin_body)
+      end
+
+      private
+
+      def init_ops(session_opts, stdin_opts)
+        @session_opts = session_opts
+        @shell_id = stdin_opts[:shell_id]
+        @command_id = stdin_opts[:command_id]
+        @stdin = stdin_opts[:stdin]
+        @shell_uri = stdin_opts[:shell_uri] || RESOURCE_URI_CMD
+      end
+
+      def validate_opts(session_opts, stdin_opts)
+        raise 'session_opts is required' unless session_opts
+        raise 'stdin_opts[:shell_id] is required' unless stdin_opts[:shell_id]
+        raise 'stdin_opts[:command_id] is required' unless stdin_opts[:command_id]
+        raise 'stdin_opts[:stdin] is required' unless stdin_opts[:stdin]
+      end
+
+      def stdin_headers
+        merge_headers(shared_headers(@session_opts),
+                      resource_uri_shell(@shell_uri),
+                      action_send,
+                      selector_shell_id(@shell_id))
+      end
+
+      def action_send
+        {
+          "#{NS_ADDRESSING}:Action" =>
+          'http://schemas.microsoft.com/wbem/wsman/1/windows/shell/Send',
+          :attributes! => {
+            "#{NS_ADDRESSING}:Action" => {
+              'mustUnderstand' => true
+            }
+          }
+        }
+      end
+
+      def stdin_body
+        {
+          "#{NS_WIN_SHELL}:Send" => {
+            "#{NS_WIN_SHELL}:Stream" => {
+              '@Name' => 'stdin',
+              '@CommandId' => @command_id,
+              :content! => Base64.encode64(@stdin)
+            }
+          }
+        }
+      end
+    end
+  end
+end