feat: ✨ Check if the service already exists

frack113 · Oct 22, 2024 · f610ea2 · f610ea2
1 parent b9b19f9
commit f610ea2
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/src/actions/drivers/byovd.rs b/src/actions/drivers/byovd.rs
@@ -7,9 +7,13 @@ use clap::Parser;
 use std::{error::Error, path::PathBuf};
 use windows::{
     core::{Owned, HSTRING, PCWSTR},
-    Win32::System::Services::{
-        CreateServiceW, OpenSCManagerW, StartServiceW, SC_HANDLE, SC_MANAGER_ALL_ACCESS,
-        SC_MANAGER_CREATE_SERVICE, SERVICE_AUTO_START, SERVICE_ERROR_IGNORE, SERVICE_KERNEL_DRIVER,
+    Win32::{
+        Foundation::GENERIC_READ,
+        System::Services::{
+            CreateServiceW, OpenSCManagerW, OpenServiceW, StartServiceW, SC_HANDLE,
+            SC_MANAGER_ALL_ACCESS, SC_MANAGER_CREATE_SERVICE, SERVICE_AUTO_START,
+            SERVICE_ERROR_IGNORE, SERVICE_KERNEL_DRIVER,
+        },
     },
 };
 
@@ -40,6 +44,16 @@ impl Runnable for Byovd {
                 SC_MANAGER_CREATE_SERVICE,
             )?);
 
+            if OpenServiceW(
+                *service_manager,
+                &HSTRING::from(self.service_name.as_str()),
+                GENERIC_READ.0,
+            )
+            .is_ok()
+            {
+                return Ok(());
+            }
+
             let service: Owned<SC_HANDLE> = Owned::new(CreateServiceW(
                 *service_manager,
                 &HSTRING::from(self.service_name.as_str()),