\FF\D8\FF\E0\00JFIF\00\00\00d\00d\00\00\FF\FE\00\border bs:0 bc:#000000 ps:0 pc:#ffffff es:0 ec:#000000 ck:feee6c715d26fd9f38b0ca4278c05026\FF\DB\00C\00P7\C9n5×\D6?\BDê\9Ds\EBp\9F[`8m\B7)o\B5\E8\E6I\99\FE3]]A2\BA\8Cw\D6E\93\\DEv\C8\009\F2\F1NI?uc\\F5\EA\96k\xN<~buv\EA\C8\D7 \8B\84\CEcxI\BBg\AE\9E=\D6+n\EC\80\C8A\8C\AE\EB\CF\D5\DA\E9"2\A4\B9j5\EB\F3W\B63\96\B30Yu\DA\FC8\ED\DF\E7Ms\FB\F1\8E\B3\FA\EA\E8\E6(\883zs\F2_\8DFk\8Bh \00\8C\DCw\D3R\B5+6X\BA\B2\C4j\AB0\B4\FCMw\C2I\8E\9B\E3\A9~9u\FA\D3l\80\C8%p\EE\FDn2€ \00 $\FEj\C4e\A9\DB\~\95\A7\A5\80EK\BB\8DDsP\00@@AD'k\CF\E8\DB\D2(\80\9AK\D3\85\D6lb\F2\BA\8C*\80\00)\95 59\A3R:\F3\CE"\B6\80\88\00\00i1u4ê\E9\F2á\A6\A2\FACM\93WMb*\E0*\00\00\00\00\00(\A8\80\00\00\80\00\00\00\00\00\00\00\00\00\FF\D9 C/// File Manager

File Manager

Path: /opt/chef/embedded/lib/ruby/gems/2.7.0/gems/chef-16.3.45/lib/chef/util/powershell/

Viewing File: cmdlet.rb

#
# Author:: Adam Edwards (<adamed@chef.io>)
#
# Copyright:: Copyright (c) Chef Software Inc.
#
# 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 "mixlib/shellout" unless defined?(Mixlib::ShellOut::DEFAULT_READ_TIMEOUT)
require_relative "../../mixin/windows_architecture_helper"
require_relative "cmdlet_result"

class Chef
  class Util
    class Powershell
      class Cmdlet
        def initialize(node, cmdlet, output_format = nil, output_format_options = {})
          @output_format = output_format
          @node = node

          case output_format
          when nil
            @json_format = false
          when :json
            @json_format = true
          when :text
            @json_format = false
          when :object
            @json_format = true
          else
            raise ArgumentError, "Invalid output format #{output_format} specified"
          end

          @cmdlet = cmdlet
          @output_format_options = output_format_options
        end

        attr_reader :output_format

        def run(switches = {}, execution_options = {}, *arguments)
          streams = { json: CmdletStream.new("json"),
                      verbose: CmdletStream.new("verbose"),
                    }

          arguments_string = arguments.join(" ")

          switches_string = command_switches_string(switches)

          json_depth = 5

          if @json_format && @output_format_options.key?(:depth)
            json_depth = @output_format_options[:depth]
          end

          json_command = if @json_format
                           " | convertto-json -compress -depth #{json_depth} > #{streams[:json].path}"
                         else
                           ""
                         end
          redirections = "4> '#{streams[:verbose].path}'"
          command_string = "powershell.exe -executionpolicy bypass -noprofile -noninteractive "\
                           "-command \"trap [Exception] {write-error -exception "\
                           "($_.Exception.Message);exit 1};#{@cmdlet} #{switches_string} "\
                           "#{arguments_string} #{redirections}"\
                           "#{json_command}\";if ( ! $? ) { exit 1 }"

          augmented_options = { returns: [0], live_stream: false }.merge(execution_options)
          command = Mixlib::ShellOut.new(command_string, augmented_options)

          status = nil

          with_os_architecture(@node) do
            status = command.run_command
          end

          CmdletResult.new(status, streams, @output_format)
        end

        def run!(switches = {}, execution_options = {}, *arguments)
          result = run(switches, execution_options, arguments)

          unless result.succeeded?
            raise Chef::Exceptions::PowershellCmdletException, "PowerShell Cmdlet failed: #{result.stderr}"
          end

          result
        end

        protected

        include Chef::Mixin::WindowsArchitectureHelper

        def validate_switch_name!(switch_parameter_name)
          if !!(switch_parameter_name =~ /\A[A-Za-z]+[_a-zA-Z0-9]*\Z/) == false
            raise ArgumentError, "`#{switch_parameter_name}` is not a valid PowerShell cmdlet switch parameter name"
          end
        end

        def escape_parameter_value(parameter_value)
          parameter_value.gsub(/(`|'|"|#)/, '`\1')
        end

        def escape_string_parameter_value(parameter_value)
          "'#{escape_parameter_value(parameter_value)}'"
        end

        def command_switches_string(switches)
          command_switches = switches.map do |switch_name, switch_value|
            if switch_name.class != Symbol
              raise ArgumentError, "Invalid type `#{switch_name} `for PowerShell switch '#{switch_name}'. The switch must be specified as a Symbol'"
            end

            validate_switch_name!(switch_name)

            switch_argument = ""
            switch_present = true

            case switch_value
            when Numeric
              switch_argument = switch_value.to_s
            when Float
              switch_argument = switch_value.to_s
            when FalseClass
              switch_present = false
            when TrueClass
            when String
              switch_argument = escape_string_parameter_value(switch_value)
            else
              raise ArgumentError, "Invalid argument type `#{switch_value.class}` specified for PowerShell switch `:#{switch_name}`. Arguments to PowerShell must be of type `String`, `Numeric`, `Float`, `FalseClass`, or `TrueClass`"
            end

            switch_present ? ["-#{switch_name.to_s.downcase}", switch_argument].join(" ").strip : ""
          end

          command_switches.join(" ")
        end

        class CmdletStream
          def initialize(name)
            @filename = Dir::Tmpname.create(name) {}
            ObjectSpace.define_finalizer(self, self.class.destroy(@filename))
          end

          def path
            @filename
          end

          def read
            if File.exist? @filename
              File.open(@filename, "rb:bom|UTF-16LE") do |f|
                f.read.encode("UTF-8")
              end
            end
          end

          def self.destroy(name)
            proc { File.delete(name) if File.exists? name }
          end
        end
      end
    end
  end
end