|
moel@82
|
1 |
/*
|
|
moel@82
|
2 |
|
|
moel@82
|
3 |
Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
moel@82
|
4 |
|
|
moel@82
|
5 |
The contents of this file are subject to the Mozilla Public License Version
|
|
moel@82
|
6 |
1.1 (the "License"); you may not use this file except in compliance with
|
|
moel@82
|
7 |
the License. You may obtain a copy of the License at
|
|
moel@82
|
8 |
|
|
moel@82
|
9 |
http://www.mozilla.org/MPL/
|
|
moel@82
|
10 |
|
|
moel@82
|
11 |
Software distributed under the License is distributed on an "AS IS" basis,
|
|
moel@82
|
12 |
WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
moel@82
|
13 |
for the specific language governing rights and limitations under the License.
|
|
moel@82
|
14 |
|
|
moel@82
|
15 |
The Original Code is the Open Hardware Monitor code.
|
|
moel@82
|
16 |
|
|
moel@82
|
17 |
The Initial Developer of the Original Code is
|
|
moel@82
|
18 |
Michael Möller <m.moeller@gmx.ch>.
|
|
moel@82
|
19 |
Portions created by the Initial Developer are Copyright (C) 2009-2010
|
|
moel@82
|
20 |
the Initial Developer. All Rights Reserved.
|
|
moel@82
|
21 |
|
|
moel@82
|
22 |
Contributor(s):
|
|
moel@82
|
23 |
|
|
moel@82
|
24 |
Alternatively, the contents of this file may be used under the terms of
|
|
moel@82
|
25 |
either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
moel@82
|
26 |
the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
moel@82
|
27 |
in which case the provisions of the GPL or the LGPL are applicable instead
|
|
moel@82
|
28 |
of those above. If you wish to allow use of your version of this file only
|
|
moel@82
|
29 |
under the terms of either the GPL or the LGPL, and not to allow others to
|
|
moel@82
|
30 |
use your version of this file under the terms of the MPL, indicate your
|
|
moel@82
|
31 |
decision by deleting the provisions above and replace them with the notice
|
|
moel@82
|
32 |
and other provisions required by the GPL or the LGPL. If you do not delete
|
|
moel@82
|
33 |
the provisions above, a recipient may use your version of this file under
|
|
moel@82
|
34 |
the terms of any one of the MPL, the GPL or the LGPL.
|
|
moel@82
|
35 |
|
|
moel@82
|
36 |
*/
|
|
moel@82
|
37 |
|
|
moel@82
|
38 |
using System;
|
|
moel@82
|
39 |
using System.Collections.Generic;
|
|
moel@82
|
40 |
using System.IO;
|
|
moel@139
|
41 |
using System.Runtime.InteropServices;
|
|
moel@123
|
42 |
using System.Security.Principal;
|
|
moel@82
|
43 |
using System.Windows.Forms;
|
|
moel@82
|
44 |
using Microsoft.Win32;
|
|
moel@82
|
45 |
using OpenHardwareMonitor.TaskScheduler;
|
|
moel@82
|
46 |
|
|
moel@82
|
47 |
namespace OpenHardwareMonitor.GUI {
|
|
moel@82
|
48 |
public class StartupManager {
|
|
moel@82
|
49 |
|
|
moel@82
|
50 |
private TaskSchedulerClass scheduler;
|
|
moel@82
|
51 |
private bool startup;
|
|
moel@82
|
52 |
|
|
moel@82
|
53 |
private const string REGISTRY_RUN =
|
|
moel@82
|
54 |
@"Software\Microsoft\Windows\CurrentVersion\Run";
|
|
moel@82
|
55 |
|
|
moel@123
|
56 |
private bool IsAdministrator() {
|
|
moel@123
|
57 |
try {
|
|
moel@123
|
58 |
WindowsIdentity identity = WindowsIdentity.GetCurrent();
|
|
moel@123
|
59 |
WindowsPrincipal principal = new WindowsPrincipal(identity);
|
|
moel@123
|
60 |
return principal.IsInRole(WindowsBuiltInRole.Administrator);
|
|
moel@123
|
61 |
} catch {
|
|
moel@123
|
62 |
return false;
|
|
moel@123
|
63 |
}
|
|
moel@123
|
64 |
}
|
|
moel@123
|
65 |
|
|
moel@82
|
66 |
public StartupManager() {
|
|
moel@123
|
67 |
if (IsAdministrator()) {
|
|
moel@123
|
68 |
try {
|
|
moel@123
|
69 |
scheduler = new TaskSchedulerClass();
|
|
moel@123
|
70 |
scheduler.Connect(null, null, null, null);
|
|
moel@123
|
71 |
} catch {
|
|
moel@123
|
72 |
scheduler = null;
|
|
moel@123
|
73 |
}
|
|
moel@123
|
74 |
|
|
moel@123
|
75 |
if (scheduler != null) {
|
|
moel@123
|
76 |
try {
|
|
moel@139
|
77 |
// check if the taskscheduler is running
|
|
moel@139
|
78 |
IRunningTaskCollection collection = scheduler.GetRunningTasks(0);
|
|
moel@139
|
79 |
|
|
moel@123
|
80 |
ITaskFolder folder = scheduler.GetFolder("\\Open Hardware Monitor");
|
|
moel@123
|
81 |
IRegisteredTask task = folder.GetTask("Startup");
|
|
moel@129
|
82 |
startup = (task != null) &&
|
|
moel@129
|
83 |
(task.Definition.Triggers.Count > 0) &&
|
|
moel@129
|
84 |
(task.Definition.Triggers[1].Type ==
|
|
moel@129
|
85 |
TASK_TRIGGER_TYPE2.TASK_TRIGGER_LOGON) &&
|
|
moel@129
|
86 |
(task.Definition.Actions.Count > 0) &&
|
|
moel@129
|
87 |
(task.Definition.Actions[1].Type ==
|
|
moel@129
|
88 |
TASK_ACTION_TYPE.TASK_ACTION_EXEC) &&
|
|
moel@129
|
89 |
(task.Definition.Actions[1] as IExecAction != null) &&
|
|
moel@129
|
90 |
((task.Definition.Actions[1] as IExecAction).Path ==
|
|
moel@129
|
91 |
Application.ExecutablePath);
|
|
moel@129
|
92 |
|
|
moel@123
|
93 |
} catch (IOException) {
|
|
moel@123
|
94 |
startup = false;
|
|
moel@123
|
95 |
} catch (UnauthorizedAccessException) {
|
|
moel@123
|
96 |
scheduler = null;
|
|
moel@139
|
97 |
} catch (COMException) {
|
|
moel@139
|
98 |
scheduler = null;
|
|
moel@123
|
99 |
}
|
|
moel@123
|
100 |
}
|
|
moel@123
|
101 |
} else {
|
|
moel@82
|
102 |
scheduler = null;
|
|
moel@82
|
103 |
}
|
|
moel@120
|
104 |
|
|
moel@120
|
105 |
if (scheduler == null) {
|
|
moel@82
|
106 |
RegistryKey key = Registry.CurrentUser.OpenSubKey(REGISTRY_RUN);
|
|
moel@82
|
107 |
startup = false;
|
|
moel@82
|
108 |
if (key != null) {
|
|
moel@82
|
109 |
string value = (string)key.GetValue("OpenHardwareMonitor");
|
|
moel@82
|
110 |
if (value != null)
|
|
moel@82
|
111 |
startup = value == Application.ExecutablePath;
|
|
moel@82
|
112 |
}
|
|
moel@82
|
113 |
}
|
|
moel@82
|
114 |
}
|
|
moel@82
|
115 |
|
|
moel@82
|
116 |
private void CreateSchedulerTask() {
|
|
moel@82
|
117 |
ITaskDefinition definition = scheduler.NewTask(0);
|
|
moel@82
|
118 |
definition.RegistrationInfo.Description =
|
|
moel@82
|
119 |
"This task starts the Open Hardware Monitor on Windows startup.";
|
|
moel@82
|
120 |
definition.Principal.RunLevel =
|
|
moel@82
|
121 |
TASK_RUNLEVEL.TASK_RUNLEVEL_HIGHEST;
|
|
moel@82
|
122 |
definition.Settings.DisallowStartIfOnBatteries = false;
|
|
moel@82
|
123 |
definition.Settings.StopIfGoingOnBatteries = false;
|
|
moel@82
|
124 |
definition.Settings.ExecutionTimeLimit = "PT0S";
|
|
moel@82
|
125 |
|
|
moel@82
|
126 |
ILogonTrigger trigger = (ILogonTrigger)definition.Triggers.Create(
|
|
moel@82
|
127 |
TASK_TRIGGER_TYPE2.TASK_TRIGGER_LOGON);
|
|
moel@82
|
128 |
|
|
moel@82
|
129 |
IExecAction action = (IExecAction)definition.Actions.Create(
|
|
moel@82
|
130 |
TASK_ACTION_TYPE.TASK_ACTION_EXEC);
|
|
moel@82
|
131 |
action.Path = Application.ExecutablePath;
|
|
moel@82
|
132 |
action.WorkingDirectory =
|
|
moel@82
|
133 |
Path.GetDirectoryName(Application.ExecutablePath);
|
|
moel@82
|
134 |
|
|
moel@82
|
135 |
ITaskFolder root = scheduler.GetFolder("\\");
|
|
moel@82
|
136 |
ITaskFolder folder;
|
|
moel@82
|
137 |
try {
|
|
moel@82
|
138 |
folder = root.GetFolder("Open Hardware Monitor");
|
|
moel@104
|
139 |
} catch (IOException) {
|
|
moel@82
|
140 |
folder = root.CreateFolder("Open Hardware Monitor", "");
|
|
moel@82
|
141 |
}
|
|
moel@82
|
142 |
folder.RegisterTaskDefinition("Startup", definition,
|
|
moel@82
|
143 |
(int)TASK_CREATION.TASK_CREATE_OR_UPDATE, null, null,
|
|
moel@82
|
144 |
TASK_LOGON_TYPE.TASK_LOGON_INTERACTIVE_TOKEN, "");
|
|
moel@82
|
145 |
}
|
|
moel@82
|
146 |
|
|
moel@82
|
147 |
private void DeleteSchedulerTask() {
|
|
moel@82
|
148 |
ITaskFolder root = scheduler.GetFolder("\\");
|
|
moel@82
|
149 |
try {
|
|
moel@82
|
150 |
ITaskFolder folder = root.GetFolder("Open Hardware Monitor");
|
|
moel@82
|
151 |
folder.DeleteTask("Startup", 0);
|
|
moel@104
|
152 |
} catch (IOException) { }
|
|
moel@82
|
153 |
try {
|
|
moel@82
|
154 |
root.DeleteFolder("Open Hardware Monitor", 0);
|
|
moel@104
|
155 |
} catch (IOException) { }
|
|
moel@82
|
156 |
}
|
|
moel@82
|
157 |
|
|
moel@82
|
158 |
private void CreateRegistryRun() {
|
|
moel@82
|
159 |
RegistryKey key = Registry.CurrentUser.CreateSubKey(REGISTRY_RUN);
|
|
moel@82
|
160 |
key.SetValue("OpenHardwareMonitor", Application.ExecutablePath);
|
|
moel@82
|
161 |
}
|
|
moel@82
|
162 |
|
|
moel@82
|
163 |
private void DeleteRegistryRun() {
|
|
moel@82
|
164 |
RegistryKey key = Registry.CurrentUser.CreateSubKey(REGISTRY_RUN);
|
|
moel@82
|
165 |
key.DeleteValue("OpenHardwareMonitor");
|
|
moel@82
|
166 |
}
|
|
moel@82
|
167 |
|
|
moel@82
|
168 |
public bool Startup {
|
|
moel@82
|
169 |
get {
|
|
moel@82
|
170 |
return startup;
|
|
moel@82
|
171 |
}
|
|
moel@82
|
172 |
set {
|
|
moel@82
|
173 |
if (startup != value) {
|
|
moel@82
|
174 |
startup = value;
|
|
moel@82
|
175 |
if (scheduler != null) {
|
|
moel@82
|
176 |
if (startup)
|
|
moel@82
|
177 |
CreateSchedulerTask();
|
|
moel@82
|
178 |
else
|
|
moel@82
|
179 |
DeleteSchedulerTask();
|
|
moel@82
|
180 |
} else {
|
|
moel@82
|
181 |
if (startup)
|
|
moel@82
|
182 |
CreateRegistryRun();
|
|
moel@82
|
183 |
else
|
|
moel@82
|
184 |
DeleteRegistryRun();
|
|
moel@82
|
185 |
}
|
|
moel@82
|
186 |
}
|
|
moel@82
|
187 |
}
|
|
moel@82
|
188 |
}
|
|
moel@82
|
189 |
}
|
|
moel@82
|
190 |
|
|
moel@82
|
191 |
|
|
moel@82
|
192 |
|
|
moel@82
|
193 |
|
|
moel@82
|
194 |
|
|
moel@82
|
195 |
}
|