164 lines
6.7 KiB
PowerShell
164 lines
6.7 KiB
PowerShell
#!powershell
|
|
|
|
# Copyright: (c) 2019, Ansible Project
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
#AnsibleRequires -CSharpUtil Ansible.Basic
|
|
|
|
$spec = @{
|
|
options = @{
|
|
name = @{ type = "str" }
|
|
remove_multiple = @{ type = "bool"; default = $false }
|
|
state = @{ type = "str"; default = "present"; choices = @("absent", "present") }
|
|
username = @{ type = "sid"; }
|
|
}
|
|
required_if = @(
|
|
@("state", "present", @("username")),
|
|
@("state", "absent", @("name", "username"), $true)
|
|
)
|
|
supports_check_mode = $true
|
|
}
|
|
|
|
$module = [Ansible.Basic.AnsibleModule]::Create($args, $spec)
|
|
$module.Result.path = $null
|
|
|
|
$name = $module.Params.name
|
|
$remove_multiple = $module.Params.remove_multiple
|
|
$state = $module.Params.state
|
|
$username = $module.Params.username
|
|
|
|
Add-CSharpType -AnsibleModule $module -References @'
|
|
using System;
|
|
using System.Runtime.InteropServices;
|
|
using System.Text;
|
|
|
|
namespace Ansible.WinUserProfile
|
|
{
|
|
public class NativeMethods
|
|
{
|
|
[DllImport("Userenv.dll", CharSet = CharSet.Unicode)]
|
|
public static extern int CreateProfile(
|
|
[MarshalAs(UnmanagedType.LPWStr)] string pszUserSid,
|
|
[MarshalAs(UnmanagedType.LPWStr)] string pszUserName,
|
|
[Out, MarshalAs(UnmanagedType.LPWStr)] StringBuilder pszProfilePath,
|
|
UInt32 cchProfilePath);
|
|
|
|
[DllImport("Userenv.dll", SetLastError = true, CharSet = CharSet.Unicode)]
|
|
public static extern bool DeleteProfileW(
|
|
[MarshalAs(UnmanagedType.LPWStr)] string lpSidString,
|
|
IntPtr lpProfile,
|
|
IntPtr lpComputerName);
|
|
|
|
[DllImport("Userenv.dll", SetLastError = true, CharSet = CharSet.Unicode)]
|
|
public static extern bool GetProfilesDirectoryW(
|
|
[Out, MarshalAs(UnmanagedType.LPWStr)] StringBuilder lpProfileDir,
|
|
ref UInt32 lpcchSize);
|
|
}
|
|
}
|
|
'@
|
|
|
|
Function Get-LastWin32ExceptionMessage {
|
|
param([int]$ErrorCode)
|
|
$exp = New-Object -TypeName System.ComponentModel.Win32Exception -ArgumentList $ErrorCode
|
|
$exp_msg = "{0} (Win32 ErrorCode {1} - 0x{1:X8})" -f $exp.Message, $ErrorCode
|
|
return $exp_msg
|
|
}
|
|
|
|
Function Get-ExpectedProfilePath {
|
|
param([String]$BaseName)
|
|
|
|
# Environment.GetFolderPath does not have an enumeration to get the base profile dir, use PInvoke instead
|
|
# and combine with the base name to return back to the user - best efforts
|
|
$profile_path_length = 0
|
|
[Ansible.WinUserProfile.NativeMethods]::GetProfilesDirectoryW($null,
|
|
[ref]$profile_path_length) > $null
|
|
|
|
$raw_profile_path = New-Object -TypeName System.Text.StringBuilder -ArgumentList $profile_path_length
|
|
$res = [Ansible.WinUserProfile.NativeMethods]::GetProfilesDirectoryW($raw_profile_path,
|
|
[ref]$profile_path_length)
|
|
|
|
if ($res -eq $false) {
|
|
$msg = Get-LastWin32ExceptionMessage -Error ([System.Runtime.InteropServices.Marshal]::GetLastWin32Error())
|
|
$module.FailJson("Failed to determine profile path with the base name '$BaseName': $msg")
|
|
}
|
|
$profile_path = Join-Path -Path $raw_profile_path.ToString() -ChildPath $BaseName
|
|
|
|
return $profile_path
|
|
}
|
|
|
|
$profiles = Get-ChildItem -Path "HKLM:\SOFTWARE\Microsoft\Windows NT\CurrentVersion\ProfileList"
|
|
|
|
if ($state -eq "absent") {
|
|
if ($null -ne $username) {
|
|
$user_profiles = $profiles | Where-Object { $_.PSChildName -eq $username.Value }
|
|
} else {
|
|
# If the username was not provided, or we are removing a profile for a deleted user, we need to try and find
|
|
# the correct SID to delete. We just verify that the path matches based on the name passed in
|
|
$expected_profile_path = Get-ExpectedProfilePath -BaseName $name
|
|
|
|
$user_profiles = $profiles | Where-Object {
|
|
$profile_path = (Get-ItemProperty -Path $_.PSPath -Name ProfileImagePath).ProfileImagePath
|
|
$profile_path -eq $expected_profile_path
|
|
}
|
|
|
|
if ($user_profiles.Length -gt 1 -and -not $remove_multiple) {
|
|
$module.FailJson("Found multiple profiles matching the path '$expected_profile_path', set 'remove_multiple=True' to remove all the profiles for this match")
|
|
}
|
|
}
|
|
|
|
foreach ($user_profile in $user_profiles) {
|
|
$profile_path = (Get-ItemProperty -Path $user_profile.PSPath -Name ProfileImagePath).ProfileImagePath
|
|
if (-not $module.CheckMode) {
|
|
$res = [Ansible.WinUserProfile.NativeMethods]::DeleteProfileW($user_profile.PSChildName, [IntPtr]::Zero,
|
|
[IntPtr]::Zero)
|
|
if ($res -eq $false) {
|
|
$msg = Get-LastWin32ExceptionMessage -Error ([System.Runtime.InteropServices.Marshal]::GetLastWin32Error())
|
|
$module.FailJson("Failed to delete the profile for $($user_profile.PSChildName): $msg")
|
|
}
|
|
}
|
|
|
|
# While we may have multiple profiles when the name option was used, it will always be the same path due to
|
|
# how we match name to a profile so setting it mutliple time sis fine
|
|
$module.Result.path = $profile_path
|
|
$module.Result.changed = $true
|
|
}
|
|
} elseif ($state -eq "present") {
|
|
# Now we know the SID, see if the profile already exists
|
|
$user_profile = $profiles | Where-Object { $_.PSChildName -eq $username.Value }
|
|
if ($null -eq $user_profile) {
|
|
# In case a SID was set as the username we still need to make sure the SID is mapped to a valid local account
|
|
try {
|
|
$account_name = $username.Translate([System.Security.Principal.NTAccount])
|
|
} catch [System.Security.Principal.IdentityNotMappedException] {
|
|
$module.FailJson("Fail to map the account '$($username.Value)' to a valid user")
|
|
}
|
|
|
|
# If the basename was not provided, determine it from the actual username
|
|
if ($null -eq $name) {
|
|
$name = $account_name.Value.Split('\', 2)[-1]
|
|
}
|
|
|
|
if ($module.CheckMode) {
|
|
$profile_path = Get-ExpectedProfilePath -BaseName $name
|
|
} else {
|
|
$raw_profile_path = New-Object -TypeName System.Text.StringBuilder -ArgumentList 260
|
|
$res = [Ansible.WinUserProfile.NativeMethods]::CreateProfile($username.Value, $name, $raw_profile_path,
|
|
$raw_profile_path.Capacity)
|
|
|
|
if ($res -ne 0) {
|
|
$exp = [System.Runtime.InteropServices.Marshal]::GetExceptionForHR($res)
|
|
$module.FailJson("Failed to create profile for user '$username': $($exp.Message)")
|
|
}
|
|
$profile_path = $raw_profile_path.ToString()
|
|
}
|
|
|
|
$module.Result.changed = $true
|
|
$module.Result.path = $profile_path
|
|
} else {
|
|
$module.Result.path = (Get-ItemProperty -Path $user_profile.PSPath -Name ProfileImagePath).ProfileImagePath
|
|
}
|
|
}
|
|
|
|
$module.ExitJson()
|
|
|