update powershell class to support optional timeout parameter

Signed-off-by: rishichawda <rishichawda@users.noreply.github.com>

1 files changed, 5 insertions, 5 deletions

diff --git a/lib/chef/powershell.rb b/lib/chef/powershell.rb
index b49d3c58e4..0797846b4d 100644
--- a/lib/chef/powershell.rb
+++ b/lib/chef/powershell.rb
@@ -34,14 +34,14 @@ class Chef
     #
     # @param script [String] script to run
     # @return [Object] output
-    def initialize(script)
+    def initialize(script, timeout = -1)
       # This Powershell DLL source lives here: https://github.com/chef/chef-powershell-shim
       # Every merge into that repo triggers a Habitat build and promotion. Running
       # the rake :update_chef_exec_dll task in this (chef/chef) repo will pull down
       # the built packages and copy the binaries to distro/ruby_bin_folder. Bundle install
       # ensures that the correct architecture binaries are installed into the path.
       @dll ||= "Chef.PowerShell.Wrapper.dll"
-      exec(script)
+      exec(script, timeout)
     end
 
     #
@@ -66,10 +66,10 @@ class Chef
 
     protected
 
-    def exec(script)
+    def exec(script, timeout)
       FFI.ffi_lib @dll
-      FFI.attach_function :execute_powershell, :ExecuteScript, [:string], :pointer
-      execution = FFI.execute_powershell(script).read_utf16string
+      FFI.attach_function :execute_powershell, :ExecuteScript, [:string, :int], :pointer
+      execution = FFI.execute_powershell(script, timeout).read_utf16string
       hashed_outcome = Chef::JSONCompat.parse(execution)
       @result = Chef::JSONCompat.parse(hashed_outcome["result"])
       @errors = hashed_outcome["errors"]