@ -3,6 +3,7 @@
# Copyright: (c) 2015, George Frank <george@georgefrank.net> # Copyright: (c) 2015, George Frank <george@georgefrank.net>
# Copyright: (c) 2015, Adam Keech <akeech@chathamfinancial.com> # Copyright: (c) 2015, Adam Keech <akeech@chathamfinancial.com>
# Copyright: (c) 2015, Hans-Joachim Kliemeck <git@kliemeck.de> # Copyright: (c) 2015, Hans-Joachim Kliemeck <git@kliemeck.de>
# Copyright: (c) 2019, Kevin Subileau (@ksubileau)
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt) # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
# Requires -Module Ansible.ModuleUtils.Legacy # Requires -Module Ansible.ModuleUtils.Legacy
@ -11,671 +12,472 @@
$ErrorActionPreference = " Stop " $ErrorActionPreference = " Stop "
$params = Parse-Args $args $start_modes_map = @ {
" auto " = " SERVICE_AUTO_START "
$result = @ { " delayed " = " SERVICE_DELAYED_AUTO_START "
changed = $false
" manual " = " SERVICE_DEMAND_START "
" disabled " = " SERVICE_DISABLED "
} }
$params = Parse-Args -arguments $args -supports_check_mode $true
$check_mode = Get-AnsibleParam -obj $params -name " _ansible_check_mode " -type " bool " -default $false
$diff_mode = Get-AnsibleParam -obj $params -name " _ansible_diff " -type " bool " -default $false
$name = Get-AnsibleParam -obj $params -name " name " -type " str " -failifempty $true $name = Get-AnsibleParam -obj $params -name " name " -type " str " -failifempty $true
$state = Get-AnsibleParam -obj $params -name " state " -type " str " -default " present " -validateset " present " , " absent " , " started " , " stopped " , " restarted " -resultobj $result $state = Get-AnsibleParam -obj $params -name " state " -type " str " -default " present " -validateset " present " , " absent " , " started " , " stopped " , " restarted " -resultobj $result
$display_name = Get-AnsibleParam -obj $params -name 'display_name' -type 'str'
$description = Get-AnsibleParam -obj $params -name 'description' -type 'str'
$application = Get-AnsibleParam -obj $params -name " application " -type " str " $application = Get-AnsibleParam -obj $params -name " application " -type " path "
$appDirectory = Get-AnsibleParam -obj $params -name " working_directory " -aliases " app_directory " , " chdir " -type " path "
$appParameters = Get-AnsibleParam -obj $params -name " app_parameters " $appParameters = Get-AnsibleParam -obj $params -name " app_parameters "
$appParametersFree = Get-AnsibleParam -obj $params -name " app_parameters_free_form " -type " str " $appArguments = Get-AnsibleParam -obj $params -name " arguments " -aliases " app_parameters_free_form "
$startMode = Get-AnsibleParam -obj $params -name " start_mode " -type " str " -default " auto " -validateset " auto " , " delayed " , " manual " , " disabled " -resultobj $result
$stdoutFile = Get-AnsibleParam -obj $params -name " stdout_file " -type " str " $stdoutFile = Get-AnsibleParam -obj $params -name " stdout_file " -type " path "
$stderrFile = Get-AnsibleParam -obj $params -name " stderr_file " -type " str " $stderrFile = Get-AnsibleParam -obj $params -name " stderr_file " -type " path "
$dependencies = Get-AnsibleParam -obj $params -name " dependencies " -type " list "
$executable = Get-AnsibleParam -obj $params -name " executable " -type " path " -default " nssm.exe "
# Deprecated options since 2.8. Remove in 2.12
$startMode = Get-AnsibleParam -obj $params -name " start_mode " -type " str " -default " auto " -validateset $start_modes_map . Keys -resultobj $result
$dependencies = Get-AnsibleParam -obj $params -name " dependencies " -type " list "
$user = Get-AnsibleParam -obj $params -name " user " -type " str " $user = Get-AnsibleParam -obj $params -name " user " -type " str "
$password = Get-AnsibleParam -obj $params -name " password " -type " str " $password = Get-AnsibleParam -obj $params -name " password " -type " str "
if ( ( $appParameters -ne $null ) -and ( $appParametersFree -ne $null ) ) $result = @ {
{ changed = $false
Fail-Json $result " Use either app_parameters or app_parameteres_free_form, but not both "
}
if ( ( $appParameters -ne $null ) -and ( $appParameters -isnot [ string ] ) ) {
Fail-Json -obj $result -message " The app_parameters parameter must be a string representing a dictionary. "
} }
$diff_text = $null
Function Nssm-Invoke function Invoke-NssmCommand {
{
[ CmdletBinding ( ) ]
[ CmdletBinding ( ) ]
param (
param (
[ Parameter ( Mandatory = $true ]
[ Parameter ( Mandatory = $true ,ValueFromRemainingArguments = $true )]
[ string ] $cmd
[ string []] $arguments
)
)
$nssm_result = Run-Command -command " nssm $cmd "
$command = Argv-ToString -arguments ( @ ( $executable ) + $arguments )
$result = Run-Command -command $command
return $nssm_result
$result . arguments = $command
}
Function Service-Exists return $result
{
[ CmdletBinding ( ) ]
param (
[ Parameter ( Mandatory = $true ) ]
[ string ] $name
)
return [ bool ] ( Get-Service " $name " -ErrorAction SilentlyContinue )
} }
Function Nssm-Remove function Get-NssmServiceStatus {
{
[ CmdletBinding ( ) ]
[ CmdletBinding ( ) ]
param (
param (
[ Parameter ( Mandatory = $true ) ]
[ Parameter ( Mandatory = $true ) ]
[ string ] $ nam e
[ string ] $service
)
)
if ( Service-Exists -name $name )
return Invoke-NssmCommand -arguments @ ( " status " , $service )
{
if ( ( Get-Service -Name $name ) . Status -ne " Stopped " ) {
$cmd = " stop "" $name "" "
$nssm_result = Nssm-Invoke $cmd
}
$cmd = " remove "" $name "" confirm "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error removing service "" $name "" "
}
$result . changed_by = " remove_service "
$result . changed = $true
}
} }
Function Nssm-Install function Get-NssmServiceParameter {
{
[ CmdletBinding ( ) ]
[ CmdletBinding ( ) ]
param (
param (
[ Parameter ( Mandatory = $true ) ]
[ Parameter ( Mandatory = $true ) ]
[ string ] $ nam e,
[ string ] $service ,
[ Parameter ( Mandatory = $true ) ]
[ Parameter ( Mandatory = $true ) ]
[ AllowEmptyString ( ) ]
[ Alias ( " param " ) ]
[ string ] $application
[ string ] $parameter ,
[ Parameter ( Mandatory = $false ) ]
[ string ] $subparameter
)
)
if ( ! $application )
$arguments = @ ( " get " , $service , $parameter )
{
if ( $subparameter -ne " " ) {
Throw " Error installing service "" $name "" . No application was supplied. "
$arguments + = $subparameter
}
}
If ( -Not ( Test-Path -Path $application -PathType Leaf ) ) {
return Invoke-NssmCommand -arguments $arguments
Throw " $application does not exist on the host "
}
if ( ! ( Service-Exists -name $name ) )
{
$nssm_result = Nssm-Invoke " install "" $name "" "" $application "" "
if ( $nssm_result . rc -ne 0 )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error installing service "" $name "" "
}
$result . changed_by = " install_service "
$result . changed = $true
} else {
$nssm_result = Nssm-Invoke " get "" $name "" Application "
if ( $nssm_result . rc -ne 0 )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error installing service "" $name "" "
}
if ( $nssm_result . stdout . split ( " `n `r " ) [ 0 ] -ne $application )
{
$cmd = " set "" $name "" Application "" $application "" "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error installing service "" $name "" "
}
$result . application = " $application "
$result . changed_by = " reinstall_service "
$result . changed = $true
}
}
if ( $result . changed )
{
$applicationPath = ( Get-Item $application ) . DirectoryName
$cmd = " set "" $name "" AppDirectory "" $applicationPath "" "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error installing service "" $name "" "
}
}
} }
Function ParseAppParameters ( ) function Set-NssmServiceParameter {
{ [ CmdletBinding ( ) ]
[ CmdletBinding ( ) ]
param (
param (
[ Parameter ( Mandatory = $true ) ]
[ Parameter ( Mandatory = $true ) ]
[ AllowEmptyString ( ) ]
[ string ] $service ,
[ string ] $appParameters
[ Parameter ( Mandatory = $true ) ]
[ string ] $parameter ,
[ Parameter ( Mandatory = $true , ValueFromRemainingArguments = $true ) ]
[ Alias ( " value " ) ]
[ string[] ] $arguments
)
)
$escapedAppParameters = $appParameters . TrimStart ( " @ " ) . TrimStart ( " { " ) . TrimEnd ( " } " ) . Replace ( " ; " , " `n " ) . Replace ( " \ " , " \\ " )
return Invoke-NssmCommand -arguments ( @ ( " set " , $service , $parameter ) + $arguments )
return ConvertFrom-StringData -StringData $escapedAppParameters
} }
Function Nssm-Update-AppParameters function Reset-NssmServiceParameter {
{
[ CmdletBinding ( ) ]
[ CmdletBinding ( ) ]
param (
param (
[ Parameter ( Mandatory = $true ) ]
[ Parameter ( Mandatory = $true ) ]
[ string ] $name ,
[ string ] $service ,
$appParameters ,
[ Parameter ( Mandatory = $true ) ]
[ string ] $appParametersFree
[ Alias ( " param " ) ]
[ string ] $parameter
)
)
$cmd = " get "" $name "" AppParameters "
return Invoke-NssmCommand -arguments @ ( " reset " , $service , $parameter )
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error updating AppParameters for service "" $name "" "
}
$appParamKeys = @ ( )
$appParamVals = @ ( )
$singleLineParams = " "
if ( $null -ne $appParameters )
{
$appParametersHash = ParseAppParameters -appParameters $appParameters
$appParamsArray = @ ( )
$appParametersHash . GetEnumerator ( ) | foreach {
$key = $ ( $_ . Name )
$val = $ ( $_ . Value )
$appParamKeys + = $key
$appParamVals + = $val
if ( $key -ne " _ " ) {
$appParamsArray + = $key
}
$appParamsArray + = $val
}
$result . nssm_app_parameters_keys = $appParamKeys
$result . nssm_app_parameters_vals = $appParamVals
$singleLineParams = Argv-ToString -arguments $appParamsArray
}
elseif ( $null -ne $appParametersFree ) {
$result . nssm_app_parameters_free_form = $appParametersFree
$singleLineParams = $appParametersFree
}
$result . nssm_app_parameters = $appParameters
$result . nssm_single_line_app_parameters = $singleLineParams
if ( $nssm_result . stdout . split ( " `n `r " ) [ 0 ] -ne $singleLineParams )
{
# Escape argument line to preserve possible quotes and spaces
$singleLineParams = Escape-Argument -argument $singleLineParams
$cmd = " set "" $name "" AppParameters $singleLineParams "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error updating AppParameters for service "" $name "" "
}
$result . changed_by = " update_app_parameters "
$result . changed = $true
}
} }
Function Nssm-Set-Output-Files function Update-NssmServiceParameter {
{ <#
[ CmdletBinding ( ) ]
. SYNOPSIS
A generic cmdlet to idempotently set a nssm service parameter .
. PARAMETER service
[ String ] The service name
. PARAMETER parameter
[ String ] The name of the nssm parameter to set .
. PARAMETER arguments
[ String[] ] Target value ( or list of value ) or array of arguments to pass to the 'nssm set' command .
. PARAMETER compare
[ scriptblock ] An optionnal idempotency check scriptblock that must return true when
the current value is equal to the desired value . Usefull when 'nssm get' doesn ' t return
the same value as 'nssm set' takes in argument , like for the ObjectName parameter .
#>
[ CmdletBinding ( SupportsShouldProcess = $true ) ]
param (
param (
[ Parameter ( Mandatory = $true ) ]
[ Parameter ( Mandatory = $true ) ]
[ string ] $name ,
[ string ] $service ,
[ string ] $stdout ,
[ string ] $stderr
)
$cmd = " get "" $name "" AppStdout "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error retrieving existing stdout file for service "" $name "" "
}
if ( $nssm_result . stdout . split ( " `n `r " ) [ 0 ] -ne $stdout )
[ Parameter ( Mandatory = $true ) ]
{
[ string ] $parameter ,
if ( ! $stdout )
{
$cmd = " reset "" $name "" AppStdout "
} else {
$cmd = " set "" $name "" AppStdout $stdout "
}
$nssm_result = Nssm-Invoke $cmd
[ Parameter ( Mandatory = $true , ValueFromRemainingArguments = $true ) ]
[ AllowEmptyString ( ) ]
[ AllowNull ( ) ]
[ Alias ( " value " ) ]
[ string[] ] $arguments ,
if ( $nssm_result . rc -ne 0 )
[ Parameter ( ) ]
{
[ scriptblock ] $compare = { param ( $actual , $expected ) @ ( Compare-Object -ReferenceObject $actual -DifferenceObject $expected ) . Length -eq 0 }
$result . nssm_error_cmd = $cmd
)
$result . nssm_error_log = $nssm_result . stderr
Throw " Error setting stdout file for service "" $name "" "
}
$result . changed_by = " set_stdout "
if ( $null -eq $arguments ) { return }
$result . changed = $true
$arguments = @ ( $arguments | Where-Object { $_ -ne '' } )
}
$cmd = " get "" $name "" AppStderr "
$nssm_result = Get-NssmServiceParameter -service $service -parameter $parameter
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
if ( $nssm_result . rc -ne 0 ) {
{
$result . nssm_error_cmd = $nssm_result . arguments
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
$result . nssm_error_log = $nssm_result . stderr
Throw " Error retrieving existing stderr file for service "" $nam "" "
Fail-Json -obj $result -message " Error retrieving $parameter for service "" $service "" "
}
}
if ( $nssm_result . stdout . split ( " `n `r " ) [ 0 ] -ne $stderr )
$current_values = @ ( $nssm_result . stdout . split ( " `n `r " ) | Where-Object { $_ -ne '' } )
{
if ( ! $stderr )
{
$cmd = " reset "" $name "" AppStderr "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
if ( -not $compare . Invoke ( $current_values , $arguments ) ) {
{
if ( $PSCmdlet . ShouldProcess ( $service , " Update ' $parameter ' parameter " ) ) {
$result . nssm_error_cmd = $cmd
if ( $arguments . Count -gt 0 ) {
$result . nssm_error_log = $nssm_result . stderr
$nssm_result = Set-NssmServiceParameter -service $service -parameter $parameter -arguments $arguments
Throw " Error clearing stderr file setting for service "" $name "" "
}
else {
$nssm_result = Reset-NssmServiceParameter -service $service -parameter $parameter
}
}
} else {
$cmd = " set "" $name "" AppStderr $stderr "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
if ( $nssm_result . rc -ne 0 ) {
{
$result . nssm_error_cmd = $nssm_result . arguments
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
$result . nssm_error_log = $nssm_result . stderr
Throw " Error setting stderr file for service "" $nam "" "
Fail-Json -obj $result -message " Error setting $parameter for service "" $service "" "
}
}
}
}
$result . changed_by = " set_stderr "
$script:diff_text + = " - $parameter = $( $current_values -join ', ' ) `n + $parameter = $( $arguments -join ', ' ) `n "
$result . changed_by = $parameter
$result . changed = $true
$result . changed = $true
}
}
}
###
function Test-NssmServiceExists {
# Setup file rotation so we don't accidentally consume too much disk
[ CmdletBinding ( ) ]
###
param (
[ Parameter ( Mandatory = $true ) ]
#set files to overwrite
[ string ] $service
$cmd = " set "" $name "" AppStdoutCreationDisposition 2 "
)
$nssm_result = Nssm-Invoke $cmd
$cmd = " set "" $name "" AppStderrCreationDisposition 2 "
$nssm_result = Nssm-Invoke $cmd
#enable file rotation
$cmd = " set "" $name "" AppRotateFiles 1 "
$nssm_result = Nssm-Invoke $cmd
#don't rotate until the service restarts
$cmd = " set "" $name "" AppRotateOnline 0 "
$nssm_result = Nssm-Invoke $cmd
#both of the below conditions must be met before rotation will happen
#minimum age before rotating
$cmd = " set "" $name "" AppRotateSeconds 86400 "
$nssm_result = Nssm-Invoke $cmd
#minimum size before rotating
return [ bool ] ( Get-Service -Name $service -ErrorAction SilentlyContinue )
$cmd = " set "" $name "" AppRotateBytes 104858 "
$nssm_result = Nssm-Invoke $cmd
} }
Function Nssm-Update-Credentials function Invoke-NssmStart {
{
[ CmdletBinding ( ) ]
[ CmdletBinding ( ) ]
param (
param (
[ Parameter ( Mandatory = $true ) ]
[ Parameter ( Mandatory = $true ) ]
[ string ] $name ,
[ string ] $service
[ Parameter ( Mandatory = $false ) ]
[ string ] $user ,
[ Parameter ( Mandatory = $false ) ]
[ string ] $password
)
)
$cmd = " get "" $name "" ObjectName "
$nssm_result = Invoke-NssmCommand -arguments @ ( " start " , $service )
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
if ( $nssm_result . rc -ne 0 ) {
{
$result . nssm_error_cmd = $nssm_result . arguments
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
$result . nssm_error_log = $nssm_result . stderr
Throw " Error updating credentials for service "" $nam "" "
Fail-Json -obj $result -message " Error starting service "" $service "" "
}
}
}
if ( $user ) {
function Invoke-NssmStop {
if ( ! $password ) {
[ CmdletBinding ( ) ]
Throw " User without password is informed for service "" $name "" "
param (
}
[ Parameter ( Mandatory = $true ) ]
else {
[ string ] $service
$fullUser = $user
)
If ( -Not ( $user . contains ( " @ " ) ) -And ( $user . Split ( " \ " ) . count -eq 1 ) ) {
$fullUser = " .\ " + $user
}
If ( $nssm_result . stdout . split ( " `n `r " ) [ 0 ] -ne $fullUser ) {
$cmd = Argv-ToString @ ( " set " , $name , " ObjectName " , $fullUser , $password )
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
$nssm_result = Invoke-NssmCommand -arguments @ ( " stop " , $service )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error updating credentials for service "" $name "" "
}
$result . changed_by = " update_credentials "
if ( $nssm_result . rc -ne 0 ) {
$result . changed = $true
$result . nssm_error_cmd = $nssm_result . arguments
}
$result . nssm_error_log = $nssm_result . stderr
}
Fail-Json -obj $result -message " Error stopping service "" $service "" "
}
}
} }
Function Nssm-Update-Dependencies function Start-NssmService {
{ [ CmdletBinding ( SupportsShouldProcess = $true ) ]
[ CmdletBinding ( ) ]
param (
param (
[ Parameter ( Mandatory = $true ) ]
[ Parameter ( Mandatory = $true ) ]
[ string ] $name ,
[ string ] $service
[ Parameter ( Mandatory = $false ) ]
$dependencies
)
)
if ( $null -eq $dependencies ) {
$currentStatus = Get-NssmServiceStatus -service $service
# Don't make any change to dependencies if the parameter is omitted
return
}
$cmd = " get "" $name "" DependOnService "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
if ( $currentStatus . rc -ne 0 ) {
{
$result . nssm_error_cmd = $currentStatus . arguments
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $currentStatus . stderr
$result . nssm_error_log = $nssm_result . stderr
Fail-Json -obj $result -message " Error starting service "" $service "" "
Throw " Error updating dependencies for service "" $name "" "
}
}
$current_dependencies = @ ( $nssm_result . stdout . split ( " `n `r " ) | where { $_ -ne '' } )
if ( $currentStatus . stdout -notlike " *SERVICE_RUNNING* " ) {
if ( $PSCmdlet . ShouldProcess ( $service , " Start service " ) ) {
If ( @ ( Compare-Object -ReferenceObject $current_dependencies -DifferenceObject $dependencies ) . Length -ne 0 ) {
switch -wildcard ( $currentStatus . stdout ) {
$dependencies_str = Argv-ToString -arguments $dependencies
" *SERVICE_STOPPED* " { Invoke-NssmStart -service $service }
$cmd = " set "" $name "" DependOnService $dependencies_str "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
" *SERVICE_CONTINUE_PENDING* " { Invoke-NssmStop -service $service ; Invoke-NssmStart -service $service }
{
" *SERVICE_PAUSE_PENDING* " { Invoke-NssmStop -service $service ; Invoke-NssmStart -service $service }
$result . nssm_error_cmd = $cmd
" *SERVICE_PAUSED* " { Invoke-NssmStop -service $service ; Invoke-NssmStart -service $service }
$result . nssm_error_log = $nssm_result . stderr
" *SERVICE_START_PENDING* " { Invoke-NssmStop -service $service ; Invoke-NssmStart -service $service }
Throw " Error updating dependencies for service "" $name "" "
" *SERVICE_STOP_PENDING* " { Invoke-NssmStop -service $service ; Invoke-NssmStart -service $service }
}
}
}
$result . changed_by = " update-dependencies "
$result . changed_by = " start_service "
$result . changed = $true
$result . changed = $true
}
}
} }
Function Nssm-Update-StartMode function Stop-NssmService {
{ [ CmdletBinding ( SupportsShouldProcess = $true ) ]
[ CmdletBinding ( ) ]
param (
param (
[ Parameter ( Mandatory = $true ) ]
[ Parameter ( Mandatory = $true ) ]
[ string ] $name ,
[ string ] $service
[ Parameter ( Mandatory = $true ) ]
[ string ] $mode
)
)
$cmd = " get "" $name "" Start "
$currentStatus = Get-NssmServiceStatus -service $service
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
if ( $currentStatus . rc -ne 0 ) {
{
$result . nssm_error_cmd = $currentStatus . arguments
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $currentStatus . stderr
$result . nssm_error_log = $nssm_result . stderr
Fail-Json -obj $result -message " Error stopping service "" $service "" "
Throw " Error updating start mode for service "" $name "" "
}
}
$modes = @ { " auto " = " SERVICE_AUTO_START " ; " delayed " = " SERVICE_DELAYED_AUTO_START " ; " manual " = " SERVICE_DEMAND_START " ; " disabled " = " SERVICE_DISABLED " }
if ( $currentStatus . stdout -notlike " *SERVICE_STOPPED* " ) {
$mappedMode = $modes . $mode
if ( $PSCmdlet . ShouldProcess ( $service , " Stop service " ) ) {
if ( $nssm_result . stdout -notlike " * $mappedMode * " ) {
Invoke-NssmStop -service $service
$cmd = " set "" $name "" Start $mappedMode "
$nssm_result = Nssm-Invoke $cmd
if ( $nssm_result . rc -ne 0 )
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error updating start mode for service "" $name "" "
}
}
$result . changed_by = " st art_mod e"
$result . changed_by = " stop_service "
$result . changed = $true
$result . changed = $true
}
}
} }
Function Nssm-Get-Status if ( ( $null -ne $appParameters ) -and ( $null -ne $appArguments ) ) {
{ Fail-Json $result " 'app_parameters' and 'arguments' are mutually exclusive but have both been set. "
[ CmdletBinding ( ) ]
param (
[ Parameter ( Mandatory = $true ) ]
[ string ] $name
)
$cmd = " status "" $name "" "
$nssm_result = Nssm-Invoke $cmd
return $nssm_result
} }
Function Nssm-Start # Backward compatibility for old parameters style. Remove the block bellow in 2.12
{ if ( $null -ne $appParameters ) {
[ CmdletBinding ( ) ]
Add-DeprecationWarning -obj $result -message " The parameter 'app_parameters' will be removed soon, use 'arguments' instead " -version 2.12
param (
[ Parameter ( Mandatory = $true ) ]
[ string ] $name
)
$currentStatus = Nssm-Get -Status -name $name
if ( $currentStatus . rc -ne 0 )
if ( $appParameters -isnot [ string ] ) {
{
Fail-Json -obj $result -message " The app_parameters parameter must be a string representing a dictionary. "
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $currentStatus . stderr
Throw " Error starting service "" $name "" "
}
}
switch -wildcard ( $currentStatus . stdout )
# Convert dict-as-string form to list
{
$escapedAppParameters = $appParameters . TrimStart ( " @ " ) . TrimStart ( " { " ) . TrimEnd ( " } " ) . Replace ( " ; " , " `n " ) . Replace ( " \ " , " \\ " )
" *SERVICE_RUNNING* " { <# Nothing to do #> }
$appParametersHash = ConvertFrom-StringData -StringData $escapedAppParameters
" *SERVICE_STOPPED* " { Nssm-Start -Service -Command -name $name }
" *SERVICE_CONTINUE_PENDING* " { Nssm-Stop -Service -Command -name $name ; Nssm-Start -Service -Command -name $name }
$appParamsArray = @ ( )
" *SERVICE_PAUSE_PENDING* " { Nssm-Stop -Service -Command -name $name ; Nssm-Start -Service -Command -name $name }
$appParametersHash . GetEnumerator ( ) | Foreach-Object {
" *SERVICE_PAUSED* " { Nssm-Stop -Service -Command -name $name ; Nssm-Start -Service -Command -name $name }
if ( $_ . Name -ne " _ " ) {
" *SERVICE_START_PENDING* " { Nssm-Stop -Service -Command -name $name ; Nssm-Start -Service -Command -name $name }
$appParamsArray + = $_ . Name
" *SERVICE_STOP_PENDING* " { Nssm-Stop -Service -Command -name $name ; Nssm-Start -Service -Command -name $name }
}
$appParamsArray + = $_ . Value
}
}
$appArguments = @ ( $appParamsArray )
# The rest of the code should use only the new $appArguments variable
} }
Function Nssm-Start-Service-Command if ( $state -in @ ( " started " , " stopped " , " restarted " ) ) {
{ Add-DeprecationWarning -obj $result -message " The values 'started', 'stopped', and 'restarted' for 'state' will be removed soon, use the win_service module to start or stop the service instead " -version 2.12
[ CmdletBinding ( ) ]
}
param (
if ( $params . ContainsKey ( 'start_mode' ) ) {
[ Parameter ( Mandatory = $true ) ]
Add-DeprecationWarning -obj $result -message " The parameter 'start_mode' will be removed soon, use the win_service module instead " -version 2.12
[ string ] $name
}
)
if ( $null -ne $dependencies ) {
Add-DeprecationWarning -obj $result -message " The parameter 'dependencies' will be removed soon, use the win_service module instead " -version 2.12
}
if ( $null -ne $user ) {
Add-DeprecationWarning -obj $result -message " The parameter 'user' will be removed soon, use the win_service module instead " -version 2.12
}
if ( $null -ne $password ) {
Add-DeprecationWarning -obj $result -message " The parameter 'password' will be removed soon, use the win_service module instead " -version 2.12
}
$cmd = " start "" $name "" "
if ( $state -ne 'absent' ) {
if ( $null -eq $application ) {
Fail-Json -obj $result -message " The application parameter must be defined when the state is not absent. "
}
$nssm_result = Nssm-Invoke $cmd
if ( -not ( Test-Path -LiteralPath $application -PathType Leaf ) ) {
Fail-Json -obj $result -message " The application specified "" $application "" does not exist on the host. "
}
if ( $nssm_result . rc -ne 0 )
if ( $null -eq $appDirectory ) {
{
$appDirectory = ( Get-Item -LiteralPath $application ) . DirectoryName
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error starting service "" $name "" "
}
}
$result . changed_by = " start_service "
if ( $user -and -not $password ) {
$result . changed = $true
Fail-Json -obj $result -message " User without password is informed for service "" $name "" "
}
} }
Function Nssm-Stop-Service-Command
{
[ CmdletBinding ( ) ]
param (
[ Parameter ( Mandatory = $true ) ]
[ string ] $name
)
$cmd = " stop "" $name "" " $service_exists = Test-NssmServiceExists -service $name
$nssm_result = Nssm-Invoke $cmd
if ( $state -eq 'absent' ) {
if ( $service_exists ) {
if ( -not $check_mode ) {
if ( ( Get-Service -Name $name ) . Status -ne " Stopped " ) {
$nssm_result = Invoke-NssmStop -service $name
}
if ( $nssm_result . rc -ne 0 )
$nssm_result = Invoke-NssmCommand -arguments @ ( " remove " , $name , " confirm " )
{
$result . nssm_error_cmd = $cmd
if ( $nssm_result . rc -ne 0 ) {
$result . nssm_error_log = $nssm_result . stderr
$result . nssm_error_cmd = $nssm_result . arguments
Throw " Error stopping service "" $name "" "
$result . nssm_error_log = $nssm_result . stderr
Fail-Json -obj $result -message " Error removing service "" $name "" "
}
}
$diff_text + = " -[ $name ] "
$result . changed_by = " remove_service "
$result . changed = $true
}
}
} else {
$diff_text_added_prefix = ''
if ( -not $service_exists ) {
if ( -not $check_mode ) {
$nssm_result = Invoke-NssmCommand -arguments @ ( " install " , $name , $application )
if ( $nssm_result . rc -ne 0 ) {
$result . nssm_error_cmd = $nssm_result . arguments
$result . nssm_error_log = $nssm_result . stderr
Fail-Json -obj $result -message " Error installing service "" $name "" "
}
$service_exists = $true
}
$result . changed_by = " stop_service_command "
$diff_text_added_prefix = '+'
$result . changed = $true
$result . changed_by = " install_service "
} $result . changed = $true
}
Function Nssm-Stop $diff_text + = " $diff_text_added_prefix [ $name ] `n "
{
[ CmdletBinding ( ) ]
param (
[ Parameter ( Mandatory = $true ) ]
[ string ] $name
)
$currentStatus = Nssm-Get -Status -name $name
# We cannot configure a service that was created above in check mode as it won't actually exist
if ( $service_exists ) {
$common_params = @ {
service = $name
WhatIf = $check_mode
}
if ( $currentStatus . rc -ne 0 )
Update-NssmServiceParameter -parameter " Application " -value $application @common_params
{
Update-NssmServiceParameter -parameter " DisplayName " -value $display_name @common_params
$result . nssm_error_cmd = $cmd
Update-NssmServiceParameter -parameter " Description " -value $description @common_params
$result . nssm_error_log = $currentStatus . stderr
Throw " Error stopping service "" $name "" "
Update-NssmServiceParameter -parameter " AppDirectory " -value $appDirectory @common_params
}
if ( $currentStatus . stdout -notlike " *SERVICE_STOPPED* " )
{
$cmd = " stop "" $name "" "
$nssm_result = Nssm-Invoke $cmd
if ( $null -ne $appArguments ) {
$singleLineParams = " "
if ( $appArguments -is [ array ] ) {
$singleLineParams = Argv-ToString -arguments $appArguments
} else {
$singleLineParams = $appArguments . ToString ( )
}
$result . nssm_app_parameters = $appArguments
$result . nssm_single_line_app_parameters = $singleLineParams
if ( $nssm_result . rc -ne 0 )
Update-NssmServiceParameter -parameter " AppParameters " -value $singleLineParams @common_params
{
$result . nssm_error_cmd = $cmd
$result . nssm_error_log = $nssm_result . stderr
Throw " Error stopping service "" $name "" "
}
}
$result . changed_by = " stop_service "
$result . changed = $true
}
}
Function Nssm-Restart Update-NssmServiceParameter -parameter " AppStdout " -value $stdoutFile @common_params
{ Update-NssmServiceParameter -parameter " AppStderr " -value $stderrFile @common_params
[ CmdletBinding ( ) ]
param (
[ Parameter ( Mandatory = $true ) ]
[ string ] $name
)
Nssm-Stop -Service -Command -name $name
###
Nssm-Start -Service -Command -name $name
# Setup file rotation so we don't accidentally consume too much disk
} ###
Function NssmProcedure #set files to overwrite
{ Update-NssmServiceParameter -parameter " AppStdoutCreationDisposition " -value 2 @common_params
Nssm-Install -name $name -application $application
Update-NssmServiceParameter -parameter " AppStderrCreationDisposition " -value 2 @common_params
Nssm-Update -AppParameters -name $name -appParameters $appParameters -appParametersFree $appParametersFree
Nssm-Set -Output -Files -name $name -stdout $stdoutFile -stderr $stderrFile
Nssm-Update -Dependencies -name $name -dependencies $dependencies
Nssm-Update -Credentials -name $name -user $user -password $password
Nssm-Update -StartMode -name $name -mode $startMode
}
Try #enable file rotation
{ Update-NssmServiceParameter -parameter " AppRotateFiles " -value 1 @common_params
switch ( $state )
{
#don't rotate until the service restarts
" absent " {
Update-NssmServiceParameter -parameter " AppRotateOnline " -value 0 @common_params
Nssm-Remove -name $name
}
#both of the below conditions must be met before rotation will happen
" present " {
#minimum age before rotating
NssmProcedure
Update-NssmServiceParameter -parameter " AppRotateSeconds " -value 86400 @common_params
}
" started " {
#minimum size before rotating
NssmProcedure
Update-NssmServiceParameter -parameter " AppRotateBytes " -value 104858 @common_params
Nssm-Start -name $name
############## DEPRECATED block since 2.8. Remove in 2.12 ##############
Update-NssmServiceParameter -parameter " DependOnService " -arguments $dependencies @common_params
if ( $user ) {
$fullUser = $user
if ( -Not ( $user . contains ( " @ " ) ) -And ( $user . Split ( " \ " ) . count -eq 1 ) ) {
$fullUser = " .\ " + $user
}
# Use custom compare callback to test only the username (and not the password)
Update-NssmServiceParameter -parameter " ObjectName " -arguments @ ( $fullUser , $password ) -compare { param ( $actual , $expected ) $actual [ 0 ] -eq $expected [ 0 ] } @common_params
}
}
" stopped " {
$mappedMode = $start_modes_map . $startMode
NssmProcedure
Update-NssmServiceParameter -parameter " Start " -value $mappedMode @common_params
Nssm-Stop -name $name
if ( $state -in " stopped " , " restarted " ) {
Stop-NssmService @common_params
}
}
" restarted " {
NssmProcedure
if ( $state -in " started " , " restarted " ) {
Nssm-Restart -name $name
Start-NssmService @common_params
}
}
}
########################################################################
Exit-Json $result
}
} }
Catch
{ if ( $diff_mode -and $result . changed -eq $true ) {
Fail-Json $result $_ . Exception . Message
$result . diff = @ {
prepared = $diff_text
}
} }
Exit-Json $result
Kevin Subileau
6 years ago
committed by