david.tahir/PrimoCacheBackupDetect
1
0
Fork 0
Code Issues Pull Requests Releases 1 Wiki Activity
Files
9ae9aea93ea5b2b7422e618ce2eef3a4d2e8af09
PrimoCacheBackupDetect/PrimoCacheBackupDetect.au3
David 9ae9aea93e Update PrimoCacheBackupDetect.au3
2019-02-12 10:40:23 +01:00

209 lines
8.6 KiB
AutoIt
Raw Blame History

#NoTrayIcon
#RequireAdmin
#AutoIt3Wrapper_Res_Fileversion=0.0.2.0
#include <AutoItConstants.au3>
#include <TrayConstants.au3>
#include <MsgBoxConstants.au3>
#include "_ProcessListProperties.au3"
Opt("TrayMenuMode", 3)
Global Const $iconFile = "shell32.dll"
Global Const $idIconError = -110
Global Const $idIconRunning = -138
Global Const $idIconPaused = -245
Global $sTaskDir = ''
Global $sTaskName = 'PrimoCacheBackupDetect'
Global $sTaskFullName = $sTaskDir & '\' & $sTaskName
ReadConfig()
PrimoCacheBackupDetect()
Func ReadConfig()
Global Const $configFile = @ScriptDir & "\PrimoCacheBackupDetect.ini"
Global $configInterval = Int(IniRead($configFile, "Config", "Interval", "60000"))
Global $configProcess = IniRead($configFile, "Config", "Process", "backupService-ab.exe")
Global $configThreshold = Int(IniRead($configFile, "Config", "Threshold", "50000000"))
Global $configPauseCmd = IniRead($configFile, "Config", "PauseCmd", '"C:\Program Files\PrimoCache\rxpcc.exe" pause -s -a')
Global $configResumeCmd = IniRead($configFile, "Config", "ResumeCmd", '"C:\Program Files\PrimoCache\rxpcc.exe" resume -s -a')
EndFunc
Func PrimoCacheBackupDetect()
Local $idCheckProcess = TrayCreateItem("Check Process")
Global $idTask = TrayCreateItem("-")
RenameTrayTask()
Local $idExit = TrayCreateItem("Exit")
TraySetIcon($iconFile, $idIconRunning)
TraySetToolTip("PrimoCache running, no Backup-Process detected.")
TraySetState($TRAY_ICONSTATE_SHOW)
Local $hTimer = TimerInit() ; Begin the timer and store the handle in a variable.
Local $fDiff = 0
Local $isRunning = 0, $wasRunning = 0
Local $processList
While 1
$fDiff = TimerDiff($hTimer) ; Find the difference in time from the previous call of TimerInit
If $fDiff > $configInterval Then ; If the difference is greater than interval then check if the named process is running
TraySetState($TRAY_ICONSTATE_SHOW) ; try to show the trayicon if it has not been initialized correctly on startup, due to a premature boot via scheduled tasks
$isRunning = 0
If ProcessExists($configProcess) Then ; Check if the process is running.
$processList = _ProcessListProperties($configProcess) ; check if the process has a memory consumption higher than the configured threshold
If $processList[1][7] > $configThreshold Then
$isRunning = 1
EndIf
EndIf
If $isRunning == 1 Then
If $wasRunning == 0 Then
PauseCache()
EndIf
$wasRunning = 1
Else
If $wasRunning == 1 Then
ResumeCache()
EndIf
$wasRunning = 0
EndIf
$hTimer = TimerInit() ; Reset the timer.
EndIf
Switch TrayGetMsg()
Case $idExit ; Exit the loop.
ExitLoop
Case $idCheckProcess
CheckProcess()
Case $idTask
If IsTaskInstalled() Then
If Not UninstallTask() Then
MsgBox(BitOR($MB_SYSTEMMODAL, $MB_ICONERROR), "Error!", "Could not remove Autostart!")
EndIf
Else
If MsgBox($MB_ICONQUESTION + $MB_YESNO, 'Proceed with Installation?', 'This will install a new scheduled Task to run this Program on boot.' & @CRLF & 'The Program has to remain at the current location which is: ' & @ScriptFullPath & @CRLF & @CRLF & 'Do you want to proceed?') = $IDYES Then
If Not InstallTask() Then
MsgBox(BitOR($MB_SYSTEMMODAL, $MB_ICONERROR), "Error!", "Could not create Autostart!")
EndIf
EndIf
EndIf
RenameTrayTask()
EndSwitch
WEnd
EndFunc
Func PauseCache()
Local $iPID = Run(@ComSpec & ' /C ' & $configPauseCmd, @SystemDir, @SW_HIDE, BitOR($STDERR_CHILD, $STDOUT_CHILD))
ProcessWaitClose($iPID)
Local $sOutput = StdoutRead($iPID)
Local $sError = StderrRead($iPID)
If $sError <> "" Then
TraySetToolTip("Could not switch PrimoCache! An Error occured!")
TraySetIcon($iconFile, $idIconError)
MsgBox(BitOR($MB_SYSTEMMODAL, $MB_ICONERROR), "Could not pause the PrimoCache!", $sError)
Else
TraySetToolTip("PrimoCache paused, Process " & $configProcess & " is running.")
TraySetIcon($iconFile, $idIconPaused)
EndIf
EndFunc
Func ResumeCache()
Local $iPID = Run(@ComSpec & ' /C ' & $configResumeCmd, @SystemDir, @SW_HIDE, BitOR($STDERR_CHILD, $STDOUT_CHILD))
ProcessWaitClose($iPID)
Local $sOutput = StdoutRead($iPID)
Local $sError = StderrRead($iPID)
If $sError <> "" Then
TraySetToolTip("Could not switch PrimoCache! An Error occured!")
TraySetIcon($iconFile, $idIconError)
MsgBox(BitOR($MB_SYSTEMMODAL, $MB_ICONERROR), "Could not resume the PrimoCache!", $sError)
Else
TraySetToolTip("PrimoCache running, no Backup-Process detected.")
TraySetIcon($iconFile, $idIconRunning)
EndIf
EndFunc
Func CheckProcess()
If ProcessExists($configProcess) Then ; Check if the process is running.
$processList = _ProcessListProperties($configProcess) ; check if the process has a memory consumption higher than the configured threshold
MsgBox($MB_SYSTEMMODAL, "Process-Details", $processList[1][0] & @CRLF & "PID: " & $processList[1][1] & @CRLF & "PPID: " & $processList[1][2] & @CRLF & "OWNER: " & $processList[1][3] & @CRLF & "PRIORITY: " & $processList[1][4] & @CRLF & "PATH: " & $processList[1][5] & @CRLF &"CPU: " & $processList[1][6] & @CRLF &"MEMORY: " & $processList[1][7] & @CRLF &"CREATION: " & $processList[1][8] & @CRLF &"CLI: " & $processList[1][9])
If $processList[1][7] > $configThreshold Then
MsgBox($MB_SYSTEMMODAL, "Threshold reached", "The process " & $configProcess & " has reached the configured Threshold of " & $configThreshold & " Bytes.")
Else
MsgBox($MB_SYSTEMMODAL, "Threshold not reached", "The process " & $configProcess & " has not reached the configured Threshold of " & $configThreshold & " Bytes.")
EndIf
Else
MsgBox($MB_SYSTEMMODAL, "Process not running", "The configured process " & $configProcess & " is not running.")
EndIf
EndFunc
Func InstallTask()
Local $sXML = _
'<?xml version="1.0" encoding="UTF-16"?>' & @CRLF & _
'<Task version="1.3" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">' & @CRLF & _
' <Triggers>' & @CRLF & _
' <LogonTrigger>' & @CRLF & _
' <Enabled>true</Enabled>' & @CRLF & _
' </LogonTrigger>' & @CRLF & _
' </Triggers>' & @CRLF & _
' <Principals>' & @CRLF & _
' <Principal id="Author">' & @CRLF & _
' <GroupId>S-1-1-0</GroupId>' & @CRLF & _
' <RunLevel>HighestAvailable</RunLevel>' & @CRLF & _
' </Principal>' & @CRLF & _
' </Principals>' & @CRLF & _
' <Settings>' & @CRLF & _
' <MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>' & @CRLF & _
' <DisallowStartIfOnBatteries>false</DisallowStartIfOnBatteries>' & @CRLF & _
' <StopIfGoingOnBatteries>false</StopIfGoingOnBatteries>' & @CRLF & _
' <AllowHardTerminate>true</AllowHardTerminate>' & @CRLF & _
' <StartWhenAvailable>false</StartWhenAvailable>' & @CRLF & _
' <RunOnlyIfNetworkAvailable>false</RunOnlyIfNetworkAvailable>' & @CRLF & _
' <IdleSettings>' & @CRLF & _
' <StopOnIdleEnd>false</StopOnIdleEnd>' & @CRLF & _
' <RestartOnIdle>false</RestartOnIdle>' & @CRLF & _
' </IdleSettings>' & @CRLF & _
' <AllowStartOnDemand>true</AllowStartOnDemand>' & @CRLF & _
' <Enabled>true</Enabled>' & @CRLF & _
' <Hidden>false</Hidden>' & @CRLF & _
' <RunOnlyIfIdle>false</RunOnlyIfIdle>' & @CRLF & _
' <DisallowStartOnRemoteAppSession>false</DisallowStartOnRemoteAppSession>' & @CRLF & _
' <UseUnifiedSchedulingEngine>true</UseUnifiedSchedulingEngine>' & @CRLF & _
' <WakeToRun>false</WakeToRun>' & @CRLF & _
' <ExecutionTimeLimit>PT0S</ExecutionTimeLimit>' & @CRLF & _
' <Priority>7</Priority>' & @CRLF & _
' </Settings>' & @CRLF & _
' <Actions Context="Author">' & @CRLF & _
' <Exec>' & @CRLF & _
' <Command>###FILE###</Command>' & @CRLF & _
' </Exec>' & @CRLF & _
' </Actions>' & @CRLF & _
'</Task>'
$sXML = StringReplace($sXML, '###FILE###', @ScriptFullPath)
Local $sFileXML = @TempDir & '\' & $sTaskName & '.xml'
FileDelete($sFileXML)
FileWrite($sFileXML, $sXML)
If FileRead($sFileXML) <> $sXML Then Return False
If RunWait('schtasks.exe /Create /XML "' & $sFileXML & '" /TN ' & $sTaskFullName, '', @SW_HIDE) <> 0 Then Return False
FileDelete($sFileXML)
Return True
EndFunc
Func UninstallTask()
If RunWait('schtasks.exe /Delete /F /TN ' & $sTaskFullName, '', @SW_HIDE) <> 0 Then Return False
Return True
EndFunc
Func IsTaskInstalled()
If RunWait('schtasks.exe /Query /TN ' & $sTaskFullName, '', @SW_HIDE) <> 0 Then Return False
Return True
EndFunc
Func RenameTrayTask()
If IsTaskInstalled() Then
TrayItemSetText($idTask, "Remove Autostart")
Else
TrayItemSetText($idTask, "Set Autostart on Boot")
EndIf
EndFunc
Reference in New Issue View Git Blame Copy Permalink