[ServiceManager] Add Failure Actions

Author: CodingWonders

Elements/ServiceManagement/WindowsService.vb

@@ -26,6 +26,44 @@
         Critical = 3
     End Enum
 
+    Class ServiceFailureActions
+        Public Property FirstFailure As ServiceFailureAction
+        Public Property FirstDelayInMillis As Long
+        Public Property SecondFailure As ServiceFailureAction
+        Public Property SecondDelayInMillis As Long
+        Public Property SubsequentFailure As ServiceFailureAction
+        Public Property SubsequentDelaysInMillis As Long
+        Public Property ResetDelayInSeconds As Integer
+
+        Public Sub New()
+            FirstFailure = ServiceFailureAction.NoAction
+            FirstDelayInMillis = 0
+            SecondFailure = ServiceFailureAction.NoAction
+            SecondDelayInMillis = 0
+            SubsequentFailure = ServiceFailureAction.NoAction
+            SubsequentDelaysInMillis = 0
+            ResetDelayInSeconds = 0
+        End Sub
+
+        Public Sub New(firstFail As ServiceFailureAction, firstDelay As Long, secondFail As ServiceFailureAction, secondDelay As Long, subsequentFail As ServiceFailureAction, subsequentDelays As Long, resetDelays As Long)
+            FirstFailure = firstFail
+            FirstDelayInMillis = If(firstDelay >= 0, firstDelay, 0)
+            SecondFailure = secondFail
+            SecondDelayInMillis = If(secondDelay >= 0, secondDelay, 0)
+            SubsequentFailure = subsequentFail
+            SubsequentDelaysInMillis = If(subsequentDelays >= 0, subsequentDelays, 0)
+            ResetDelayInSeconds = If(resetDelays >= 0, resetDelays, 0)
+        End Sub
+    End Class
+
+    Enum ServiceFailureAction As Integer
+        Unknown = -1
+        NoAction = 0
+        RestartService = 1
+        RestartComputer = 2
+        RunProgram = 3
+    End Enum
+
     Public Property Name As String
     Public Property DisplayName As String
     Public Property Description As String
@@ -37,8 +75,9 @@
     Public Property ErrorControl As ServiceErrorControl
     Public Property RequiredPrivileges As New List(Of NTSecurityPrivilegeConstant)
     Public Property Dependencies As String()
+    Public Property FailureActions As ServiceFailureActions
 
-    Public Sub New(name As String, displayName As String, description As String, objectName As String, imagePath As String, startType As ServiceStartType, delayedStart As Boolean, type As ServiceType, errorControl As ServiceErrorControl, ntPrivileges As List(Of NTSecurityPrivilegeConstant), deps As String())
+    Public Sub New(name As String, displayName As String, description As String, objectName As String, imagePath As String, startType As ServiceStartType, delayedStart As Boolean, type As ServiceType, errorControl As ServiceErrorControl, ntPrivileges As List(Of NTSecurityPrivilegeConstant), deps As String(), failureActions As ServiceFailureActions)
         Me.Name = name
         Me.DisplayName = displayName
         Me.Description = description
@@ -50,6 +89,7 @@
         Me.ErrorControl = errorControl
         Me.RequiredPrivileges = ntPrivileges
         Me.Dependencies = deps
+        Me.FailureActions = failureActions
     End Sub
 
     Public Function StartTypeToString() As String
@@ -101,4 +141,18 @@
         End Select
     End Function
 
+    Public Function FailureActionToString(FailureAction As ServiceFailureAction) As String
+        Select Case FailureAction
+            Case ServiceFailureAction.NoAction
+                Return "Take no action"
+            Case ServiceFailureAction.RestartService
+                Return "Restart Service"
+            Case ServiceFailureAction.RestartComputer
+                Return "Restart Computer"
+            Case ServiceFailureAction.RunProgram
+                Return "Run a program"
+        End Select
+        Return ""
+    End Function
+
 End Class

Elements/ServiceManagement/WindowsServiceHelper.vb

@@ -254,7 +254,8 @@ Module WindowsServiceHelper
                         serviceType As WindowsService.ServiceType = WindowsService.ServiceType.Unknown,
                         serviceErrorControl As WindowsService.ServiceErrorControl = WindowsService.ServiceErrorControl.Unknown,
                         serviceRequiredPrivilegesString() As String = New String() {},
-                        serviceDependencies() As String = New String() {}
+                        serviceDependencies() As String = New String() {},
+                        serviceFailActionByteArr() As Byte = New Byte() {}
                     Using ServiceInfoRk As RegistryKey = Registry.LocalMachine.OpenSubKey(String.Format("zSYS\ControlSet{0}\Services\{1}", DefaultControlSet.ToString().PadLeft(3, "0"), ServiceName), False)
                         ' We explicitly tell that we want to grab the raw data without env var expansion because REG_EXPAND_SZ values
                         ' are still string values, but with unexpanded environment variables. If the variable exists in the target system,
@@ -286,6 +287,7 @@ Module WindowsServiceHelper
                         serviceRequiredPrivilegesString = ServiceInfoRk.GetValue("RequiredPrivileges", New String() {})
                         ' Same goes for dependencies
                         serviceDependencies = ServiceInfoRk.GetValue("DependOnService", New String() {})
+                        serviceFailActionByteArr = ServiceInfoRk.GetValue("FailureActions", New Byte() {})
 
                         Dim serviceRequiredPrivilegeList As New List(Of NTSecurityPrivilegeConstant)
 
@@ -311,7 +313,8 @@ Module WindowsServiceHelper
                                                            serviceType,
                                                            serviceErrorControl,
                                                            serviceRequiredPrivilegeList,
-                                                           serviceDependencies))
+                                                           serviceDependencies,
+                                                           ParseFailureActionByteArray(serviceFailActionByteArr)))
                     End Using
                 Next
             Catch ex As Exception
@@ -325,4 +328,51 @@ Module WindowsServiceHelper
         Return serviceList
     End Function
 
+    Private Function ParseFailureActionByteArray(FailureActions As Byte()) As WindowsService.ServiceFailureActions
+        Dim scFailure As WindowsService.ServiceFailureActions = New WindowsService.ServiceFailureActions()
+        Dim firstFail As WindowsService.ServiceFailureAction = WindowsService.ServiceFailureAction.Unknown,
+            secondFail As WindowsService.ServiceFailureAction = WindowsService.ServiceFailureAction.Unknown,
+            subsequentFails As WindowsService.ServiceFailureAction = WindowsService.ServiceFailureAction.Unknown
+        Dim firstDelay As Long = 0,
+            secondDelay As Long = 0,
+            subsequentDelay As Long = 0
+        Dim resetDelay As Integer = 0
+
+        If FailureActions.Count >= 1 Then
+            Try
+                resetDelay = GetDelay(FailureActions.Skip(0).Take(4).ToArray())
+
+                ' We have to get the number of byte elements twice because undefined failure measures
+                ' cause our byte array to be smaller than expected, therefore causing indexes out of bounds.
+                firstFail = FailureActions(20)
+                firstDelay = GetDelay(FailureActions.Skip(24).Take(4).ToArray())
+                If FailureActions.Count > 28 Then
+                    ' We have defined second failure measures
+                    secondFail = FailureActions(28)
+                    secondDelay = GetDelay(FailureActions.Skip(32).Take(4).ToArray())
+                End If
+                If FailureActions.Count > 36 Then
+                    ' We have defined subsequent failure measures
+                    subsequentFails = FailureActions(36)
+                    subsequentDelay = GetDelay(FailureActions.Skip(40).Take(4).ToArray())
+                End If
+            Catch ex As Exception
+
+            End Try
+
+            scFailure = New WindowsService.ServiceFailureActions(firstFail, firstDelay, secondFail, secondDelay, subsequentFails, subsequentDelay, resetDelay)
+        End If
+        Return scFailure
+    End Function
+
+    Private Function GetDelay(ByteArray As Byte()) As Long
+        Dim binary As String = ""
+
+        For x = ByteArray.Length - 1 To 0 Step -1
+            binary &= Convert.ToString(ByteArray(x), 2).PadLeft(8, "0"c)
+        Next
+
+        Return Convert.ToInt32(binary, 2)
+    End Function
+
 End Module