summaryrefslogtreecommitdiff
path: root/lib/chef/win32/api/command_line_helper.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/chef/win32/api/command_line_helper.rb')
-rw-r--r--lib/chef/win32/api/command_line_helper.rb89
1 files changed, 89 insertions, 0 deletions
diff --git a/lib/chef/win32/api/command_line_helper.rb b/lib/chef/win32/api/command_line_helper.rb
new file mode 100644
index 0000000000..6ddc74ae0f
--- /dev/null
+++ b/lib/chef/win32/api/command_line_helper.rb
@@ -0,0 +1,89 @@
+#
+# Author:: Kapil Chouhan <kapil.chouhan@msystechnologies.com>
+# Copyright:: Copyright 2013-2020, Chef Software, Inc.
+# License:: Apache License, Version 2.0
+#
+# 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 "../api"
+
+class Chef
+ module ReservedNames::Win32
+ module API
+ module CommandLineHelper
+ # extend Chef::ReservedNames::Win32
+ extend Chef::ReservedNames::Win32::API
+
+ ###############################################
+ # Win32 API Bindings
+ ###############################################
+
+ ffi_lib "Shell32"
+
+=begin
+LPWSTR * CommandLineToArgvW(
+ LPCWSTR lpCmdLine,
+ int *pNumArgs
+);
+=end
+
+ safe_attach_function :command_line_to_argv_w, :CommandLineToArgvW, %i{pointer pointer}, :pointer
+
+ ffi_lib "Kernel32"
+
+=begin
+LPSTR GetCommandLineA();
+=end
+
+ safe_attach_function :get_command_line, :GetCommandLineA, [], :pointer
+
+=begin
+HLOCAL LocalFree(
+ _Frees_ptr_opt_ HLOCAL hMem
+);
+=end
+
+ safe_attach_function :local_free, :LocalFree, [:pointer], :pointer
+
+ ###############################################
+ # Helpers
+ ###############################################
+
+ # It takes the supplied string and splits it into an array.
+ def command_line_to_argv_w_helper(args)
+ arguments_list = []
+ argv = args.to_wstring
+ result = get_command_line
+ argc = FFI::MemoryPointer.new(:int)
+
+ # Parses a Unicode command line string
+ # It is return an array of pointers to the command line arguments.
+ # Along with a count of such arguments
+ result = command_line_to_argv_w(argv, argc)
+ str_ptr = result.read_pointer
+ offset = 0
+ number_of_agrs = argc.read_int
+ number_of_agrs.times do
+ new_str_pointer = str_ptr.+(offset)
+ argument = new_str_pointer.read_wstring
+ arguments_list << argument
+ offset = offset + argument.length * 2 + 2
+ end
+ local_free(result)
+ arguments_list
+ end
+ end
+ end
+ end
+end