winrm 2.2.3 → 2.3.0

data/lib/winrm/wsmv/send_data.rb

@@ -1,5 +1,3 @@
-# -*- encoding: utf-8 -*-
-#
 # Copyright 2016 Matt Wrock <matt@mattwrock.com>
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

data/lib/winrm/wsmv/soap.rb

@@ -1,5 +1,3 @@
-# -*- encoding: utf-8 -*-
-#
 # Copyright 2010 Dan Wanek <dan.wanek@gmail.com>
 # Copyright 2016 Shawn Neal <sneal@sneal.net>
 #

data/lib/winrm/wsmv/wql_pull.rb

@@ -1,56 +1,54 @@
-# -*- encoding: utf-8 -*-
-
-require 'nori'
-require_relative 'base'
-
-module WinRM
-  module WSMV
-    # WSMV message to 'pull' rest of enumeration results from Windows via WQL
-    class WqlPull < Base
-      def initialize(session_opts, namespace, enumeration_context)
-        @session_opts = session_opts
-        @namespace = namespace
-        @enumeration_context = enumeration_context
-      end
-
-      def process_response(response)
-        parser = Nori.new(
-          parser: :rexml,
-          advanced_typecasting: false,
-          convert_tags_to: ->(tag) { tag.snakecase.to_sym },
-          strip_namespaces: true
-        )
-        parser.parse(response.to_s)[:envelope][:body]
-      end
-
-      protected
-
-      def create_header(header)
-        header << Gyoku.xml(wql_header)
-      end
-
-      def create_body(body)
-        body.tag!("#{NS_ENUM}:Pull") { |en| en << Gyoku.xml(wql_body) }
-      end
-
-      private
-
-      def wql_header
-        merge_headers(
-          shared_headers(@session_opts),
-          resource_uri_wmi(@namespace),
-          action_enumerate_pull
-        )
-      end
-
-      def wql_body
-        {
-          "#{NS_ENUM}:EnumerationContext" => @enumeration_context,
-          "#{NS_WSMAN_DMTF}:OptimizeEnumeration" => nil,
-          "#{NS_ENUM}:MaxElements" => '32000',
-          "#{NS_WSMAN_MSFT}:SessionId" => "uuid:#{@session_opts[:session_id]}"
-        }
-      end
-    end
-  end
-end
+require 'nori'
+require_relative 'base'
+
+module WinRM
+  module WSMV
+    # WSMV message to 'pull' rest of enumeration results from Windows via WQL
+    class WqlPull < Base
+      def initialize(session_opts, namespace, enumeration_context)
+        @session_opts = session_opts
+        @namespace = namespace
+        @enumeration_context = enumeration_context
+      end
+
+      def process_response(response)
+        parser = Nori.new(
+          parser: :rexml,
+          advanced_typecasting: false,
+          convert_tags_to: ->(tag) { tag.snakecase.to_sym },
+          strip_namespaces: true
+        )
+        parser.parse(response.to_s)[:envelope][:body]
+      end
+
+      protected
+
+      def create_header(header)
+        header << Gyoku.xml(wql_header)
+      end
+
+      def create_body(body)
+        body.tag!("#{NS_ENUM}:Pull") { |en| en << Gyoku.xml(wql_body) }
+      end
+
+      private
+
+      def wql_header
+        merge_headers(
+          shared_headers(@session_opts),
+          resource_uri_wmi(@namespace),
+          action_enumerate_pull
+        )
+      end
+
+      def wql_body
+        {
+          "#{NS_ENUM}:EnumerationContext" => @enumeration_context,
+          "#{NS_WSMAN_DMTF}:OptimizeEnumeration" => nil,
+          "#{NS_ENUM}:MaxElements" => '32000',
+          "#{NS_WSMAN_MSFT}:SessionId" => "uuid:#{@session_opts[:session_id]}"
+        }
+      end
+    end
+  end
+end

data/lib/winrm/wsmv/wql_query.rb

@@ -1,99 +1,98 @@
-# -*- 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(transport, session_opts, wql, namespace = nil)
-        @session_opts = session_opts
-        @wql = wql
-        @namespace = namespace
-        @transport = transport
-      end
-
-      def process_response(response, &block)
-        parser = Nori.new(
-          parser: :rexml,
-          advanced_typecasting: false,
-          convert_tags_to: ->(tag) { tag.snakecase.to_sym },
-          strip_namespaces: true
-        )
-        @items = Hash.new { |h, k| h[k] = [] }
-
-        hresp = parser.parse(response.to_s)[:envelope][:body][:enumerate_response]
-        process_items hresp[:items], &block
-
-        # Perform WS-Enum PULL's until we have all the elements
-        enumeration_context = hresp[:enumeration_context]
-        until enumeration_context.nil?
-          query = WqlPull.new(@session_opts, @namespace, enumeration_context)
-          hresp = query.process_response(@transport.send_request(query.build))[:pull_response]
-          process_items hresp[:items], &block
-          enumeration_context = hresp[:enumeration_context]
-        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 process_items(items, &block)
-        return if items.nil?
-        items.each_pair do |k, v|
-          # Normalize items so the type always has an array even if it's just a single item.
-          v_ary = v.is_a?(Array) ? v : [v]
-          if block
-            v_ary.each { |val| yield k, val }
-          else
-            @items[k] += v_ary
-          end
-        end
-      end
-
-      def wql_header
-        merge_headers(shared_headers(@session_opts), resource_uri_wmi(@namespace), 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
+# 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(transport, session_opts, wql, namespace = nil)
+        @session_opts = session_opts
+        @wql = wql
+        @namespace = namespace
+        @transport = transport
+      end
+
+      def process_response(response, &block)
+        parser = Nori.new(
+          parser: :rexml,
+          advanced_typecasting: false,
+          convert_tags_to: ->(tag) { tag.snakecase.to_sym },
+          strip_namespaces: true
+        )
+        @items = Hash.new { |h, k| h[k] = [] }
+
+        hresp = parser.parse(response.to_s)[:envelope][:body][:enumerate_response]
+        process_items hresp[:items], &block
+
+        # Perform WS-Enum PULL's until we have all the elements
+        enumeration_context = hresp[:enumeration_context]
+        until enumeration_context.nil?
+          query = WqlPull.new(@session_opts, @namespace, enumeration_context)
+          hresp = query.process_response(@transport.send_request(query.build))[:pull_response]
+          process_items hresp[:items], &block
+          enumeration_context = hresp[:enumeration_context]
+        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 process_items(items, &block)
+        return if items.nil?
+
+        items.each_pair do |k, v|
+          # Normalize items so the type always has an array even if it's just a single item.
+          v_ary = v.is_a?(Array) ? v : [v]
+          if block
+            v_ary.each { |val| yield k, val }
+          else
+            @items[k] += v_ary
+          end
+        end
+      end
+
+      def wql_header
+        merge_headers(shared_headers(@session_opts), resource_uri_wmi(@namespace), 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,5 +1,3 @@
-# -*- encoding: utf-8 -*-
-#
 # Copyright 2016 Shawn Neal <sneal@sneal.net>
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

data/tests/integration/auth_timeout_spec.rb

@@ -1,4 +1,3 @@
-# encoding: UTF-8
 require_relative 'spec_helper'
 
 # This test may only be meaningful with kerberos auth

data/tests/integration/cmd_spec.rb

@@ -1,4 +1,3 @@
-# encoding: UTF-8
 require_relative 'spec_helper'
 
 describe 'winrm client cmd' do
@@ -21,7 +20,7 @@ describe 'winrm client cmd' do
   end
 
   describe 'codepage' do
-    let(:options) { Hash.new }
+    let(:options) { {} }
     let(:shell) { winrm_connection.shell(:cmd, options) }
 
     after { shell.close }
@@ -98,7 +97,7 @@ describe 'winrm client cmd' do
   end
 
   describe 'ipconfig with /all argument' do
-    subject(:output) { @cmd_shell.run('ipconfig', %w(/all)) }
+    subject(:output) { @cmd_shell.run('ipconfig', %w[/all]) }
     it { should have_exit_code 0 }
     it { should have_stdout_match(/Windows IP Configuration/) }
     it { should have_no_stderr }

data/tests/integration/issue_59_spec.rb

@@ -1,4 +1,3 @@
-# encoding: UTF-8
 require_relative 'spec_helper'
 
 describe 'issue 59' do

data/tests/integration/powershell_spec.rb

@@ -1,4 +1,3 @@
-# encoding: UTF-8
 require_relative 'spec_helper'
 
 describe 'winrm client powershell' do
@@ -75,10 +74,10 @@ describe 'winrm client powershell' do
 
   describe 'capturing output from Write-Host and Write-Error' do
     subject(:output) do
-      script = <<-eos
+      script = <<-EOS
       Write-Host 'Hello'
       $host.ui.WriteErrorLine(', world!')
-      eos
+      EOS
 
       @captured_stdout = ''
       @captured_stderr = ''
@@ -105,10 +104,10 @@ describe 'winrm client powershell' do
 
   describe 'capturing output from pipeline followed by Host' do
     subject(:output) do
-      script = <<-eos
+      script = <<-EOS
       Write-Output 'output'
       $host.UI.Writeline('host')
-      eos
+      EOS
 
       @captured_stdout = ''
       @captured_stderr = ''

data/tests/integration/spec_helper.rb

@@ -1,4 +1,3 @@
-# encoding: UTF-8
 require 'rubygems'
 require 'bundler/setup'
 require 'winrm'
@@ -12,8 +11,8 @@ module ConnectionHelper
   end
 
   def connection_opts
-    @config ||= begin
-      cfg = symbolize_keys(YAML.load(File.read(winrm_config_path)))
+    @connection_opts ||= begin
+      cfg = symbolize_keys(YAML.safe_load(File.read(winrm_config_path)))
       merge_environment(cfg)
     end
   end
@@ -22,9 +21,7 @@ module ConnectionHelper
     merge_config_option_from_environment(config, 'user')
     merge_config_option_from_environment(config, 'password')
     merge_config_option_from_environment(config, 'no_ssl_peer_verification')
-    if ENV['use_ssl_peer_fingerprint']
-      config[:ssl_peer_fingerprint] = ENV['winrm_cert']
-    end
+    config[:ssl_peer_fingerprint] = ENV['winrm_cert'] if ENV['use_ssl_peer_fingerprint']
     config[:endpoint] = ENV['winrm_endpoint'] if ENV['winrm_endpoint']
     config
   end

data/tests/integration/transport_spec.rb

@@ -1,4 +1,3 @@
-# encoding: UTF-8
 require_relative 'spec_helper'
 
 describe 'WinRM connection' do

data/tests/integration/wql_spec.rb

@@ -1,34 +1,33 @@
-# encoding: UTF-8
-require_relative 'spec_helper'
-
-describe 'winrm client wql' do
-  before(:all) do
-    @winrm = winrm_connection
-  end
-
-  it 'should query Win32_OperatingSystem' do
-    output = @winrm.run_wql('select * from Win32_OperatingSystem')
-    expect(output).to_not be_empty
-    output_caption = output[:win32_operating_system][0][:caption]
-    expect(output_caption).to include('Microsoft')
-    expect(output_caption).to include('Windows')
-  end
-
-  it 'should query Win32_Process' do
-    output = @winrm.run_wql('select * from Win32_Process')
-    expect(output).to_not be_empty
-    process_count = output[:win32_process].count
-    expect(process_count).to be > 1
-    expect(output[:win32_process]).to all(include(:command_line))
-  end
-
-  it 'should query Win32_Process with block' do
-    count = 0
-    @winrm.run_wql('select * from Win32_Process') do |type, item|
-      expect(type).to eq(:win32_process)
-      expect(item).to include(:command_line)
-      count += 1
-    end
-    expect(count).to be > 1
-  end
-end
+require_relative 'spec_helper'
+
+describe 'winrm client wql' do
+  before(:all) do
+    @winrm = winrm_connection
+  end
+
+  it 'should query Win32_OperatingSystem' do
+    output = @winrm.run_wql('select * from Win32_OperatingSystem')
+    expect(output).to_not be_empty
+    output_caption = output[:win32_operating_system][0][:caption]
+    expect(output_caption).to include('Microsoft')
+    expect(output_caption).to include('Windows')
+  end
+
+  it 'should query Win32_Process' do
+    output = @winrm.run_wql('select * from Win32_Process')
+    expect(output).to_not be_empty
+    process_count = output[:win32_process].count
+    expect(process_count).to be > 1
+    expect(output[:win32_process]).to all(include(:command_line))
+  end
+
+  it 'should query Win32_Process with block' do
+    count = 0
+    @winrm.run_wql('select * from Win32_Process') do |type, item|
+      expect(type).to eq(:win32_process)
+      expect(item).to include(:command_line)
+      count += 1
+    end
+    expect(count).to be > 1
+  end
+end