NetUserSetInfo uses ffi

2 files changed, 40 insertions, 3 deletions

diff --git a/lib/chef/win32/api/net.rb b/lib/chef/win32/api/net.rb
index 5eb77ae2b0..91d665c49e 100644
--- a/lib/chef/win32/api/net.rb
+++ b/lib/chef/win32/api/net.rb
@@ -84,10 +84,16 @@ class Chef
             :usri3_home_dir_drive, :LPWSTR,
             :usri3_password_expired, :DWORD
 
-
           def set(key, val)
-            if val.is_a? String
-              val = FFI::MemoryPointer.from_string(val.to_wstring)
+            val = if val.is_a? String
+              encoded = if val.encoding == Encoding::UTF_16LE
+                val
+              else
+                val.to_wstring
+              end
+              FFI::MemoryPointer.from_string(encoded)
+            else
+              val
             end
             self[key] = val
           end
@@ -176,6 +182,16 @@ class Chef
 #  _In_ LPVOID Buffer
 #);
         safe_attach_function :NetApiBufferFree, [:LPVOID], :DWORD
+
+#NET_API_STATUS NetUserSetInfo(
+#  _In_  LPCWSTR servername,
+#  _In_  LPCWSTR username,
+#  _In_  DWORD   level,
+#  _In_  LPBYTE  buf,
+#  _Out_ LPDWORD parm_err
+#);
+        safe_attach_function :NetUserSetInfo, [:LPCWSTR, :LPCWSTR, :DWORD, :LPBYTE, :LPDWORD], :DWORD
+
       end
     end
   end
diff --git a/lib/chef/win32/user.rb b/lib/chef/win32/user.rb
index 345eb3b6c7..8b28bf8b29 100644
--- a/lib/chef/win32/user.rb
+++ b/lib/chef/win32/user.rb
@@ -152,6 +152,27 @@ END
         ui3
       end
 
+      def self.net_user_set_info_l3(server_name, user_name, info)
+        param_err = FFI::Buffer.new(:long)
+        buf = default_user_info_3
+
+        info.each do |k, v|
+          buf.set(k, v)
+        end
+
+        server_name = wstring(server_name)
+        user_name = wstring(user_name)
+
+        rc = NetUserSetInfo(server_name, user_name, 3, buf, param_err)
+        if rc != NERR_Success
+          if Chef::ReservedNames::Win32::Error.get_last_error != 0
+            Chef::ReservedNames::Win32::Error.raise!
+          else
+            net_api_error!(rc)
+          end
+        end
+      end
+
       def self.net_local_group_add_member(server_name, group_name, domain_user)
         server_name = server_name.to_wstring if server_name
         group_name = group_name.to_wstring