Merge pull request #103 from shelbyKiraM/master

Create code for creating a Task Scheduler task to run at boot with elevated privileges

Author: AlbertMN

AssistantComputerControl/MainProgram.cs

@@ -399,6 +399,66 @@ static void SystemEvents_SessionSwitch(object sender, SessionSwitchEventArgs e)
             }
         }
 
+        private static bool UpdateUserTaskInScheduler(string action)
+        {
+            try
+            {
+                ProcessStartInfo startInfo = new ProcessStartInfo();
+                startInfo.FileName = "cmd.exe";
+                startInfo.Arguments = "/C schtasks /query /TN \"AssistantComputerControl startup\""; //Check if task exists
+                startInfo.RedirectStandardOutput = true;
+                startInfo.UseShellExecute = false;
+                startInfo.CreateNoWindow = true;
+                startInfo.WindowStyle = ProcessWindowStyle.Hidden;
+                if (System.Environment.OSVersion.Version.Major < 6)
+                {
+                    startInfo.Verb = "runas";
+                }
+                using (Process process = Process.Start(startInfo))
+                {
+                    // Read in all the text from the process with the StreamReader.
+                    using (StreamReader reader = process.StandardOutput)
+                    {
+                        string stdout = reader.ReadToEnd();
+                        if (stdout.Contains("<<TaskName>>")) //If task exists
+                        {
+                            startInfo.RedirectStandardOutput = false;
+                            startInfo.UseShellExecute = true;
+                            switch (action)
+                            {
+                                case "Enable":
+                                    startInfo.Arguments = "/C schtasks /Change /TN \"AssistantComputerControl startup\"  /Enable";
+                                    break;
+
+                                case "Disable":
+                                    startInfo.Arguments = "/C schtasks /Change /TN \"AssistantComputerControl startup\" /Disable";
+                                    break;
+
+                                case "Run":
+                                    startInfo.Arguments = "/C schtasks /RUN /TN \"AssistantComputerControl startup\"";
+                                    break;
+                            }
+                            Process.Start(startInfo).WaitForExit();
+                        }
+                        else
+                        {
+                            return false;
+                        }
+                        stdout = null;
+                        reader.Close();
+                        reader.Dispose();
+                    }
+                }
+                startInfo = null;
+                return true;
+            }
+            catch (Exception ex)
+            {
+                MessageBox.Show(ex.Message);
+                return false;
+            }
+        }
+
         public static void TaskSchedulerSetup () {
             //Create "Task Scheduler" service; cleanup ACC on startup, log on, workstation unlock
             try {
@@ -423,6 +483,30 @@ public static void TaskSchedulerSetup () {
             } catch {
                 DoDebug("Failed to create / update Task Scheduler service");
             }
+            //Create "Task Scheduler" service; run ACC on startup & log on, added by Shelby Marvell
+            try
+            {
+                using (TaskService ts = new TaskService())
+                {
+                    var ps1File = Path.Combine(MainProgram.currentLocation, "ExtraCleanupper.ps1");
+
+                    TaskDefinition td = ts.NewTask();
+                    td.Principal.LogonType = TaskLogonType.S4U;
+                    td.Principal.RunLevel = TaskRunLevel.Highest;
+                    td.RegistrationInfo.Author = "Albert MN. | AssistantComputerControl";
+                    td.RegistrationInfo.Description = "AssistantComputerControl startup - Runs ACC on reboot/login";
+                    td.Triggers.Add(new BootTrigger());
+                    td.Triggers.Add(new LogonTrigger());
+                    td.Actions.Add(new ExecAction(Application.ExecutablePath, null, null));
+
+                    // Register the task in the root folder
+                    ts.RootFolder.RegisterTaskDefinition(@"AssistantComputerControl startup", td);
+                }
+            }
+            catch
+            {
+                DoDebug("Failed to create / update Task Scheduler startup service");
+            }
         }
 
         private static void CurrentDomain_UnhandledException(object sender, UnhandledExceptionEventArgs args) {
@@ -627,14 +711,20 @@ public static void Exit() {
 
         public static void SetStartup(bool status, bool setThroughSoftware = false) {
             try {
-                RegistryKey rk = Registry.CurrentUser.OpenSubKey("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run", true);
-
+                bool res = false;
                 if (status) {
-                    rk.SetValue(appName, Application.ExecutablePath);
-                    DoDebug("ACC now starts with Windows");
+                    res = UpdateUserTaskInScheduler("Disable");
                 } else {
-                    rk.DeleteValue(appName, false);
-                    DoDebug("ACC no longer starts with Windows");
+                    res = UpdateUserTaskInScheduler("Enable");
+                }
+                while (!res) {
+                    // Some error occurred. Try recreating the task.
+                    TaskSchedulerSetup();
+                    if (status) {
+                        res = UpdateUserTaskInScheduler("Disable");
+                    } else {
+                        res = UpdateUserTaskInScheduler("Enable");
+                    }
                 }
             } catch {
                 DoDebug("Failed to start ACC with Windows");