/
ScheduledTaskCommand.cs
127 lines (107 loc) · 4.4 KB
/
ScheduledTaskCommand.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
using System;
using System.Diagnostics;
using System.IO;
namespace IPFilter.Commands
{
class ScheduledTaskCommand
{
public enum _TASK_LOGON_TYPE
{
TASK_LOGON_NONE,
TASK_LOGON_PASSWORD,
TASK_LOGON_S4U,
TASK_LOGON_INTERACTIVE_TOKEN,
TASK_LOGON_GROUP,
TASK_LOGON_SERVICE_ACCOUNT,
TASK_LOGON_INTERACTIVE_TOKEN_OR_PASSWORD,
}
public enum _TASK_TRIGGER_TYPE2
{
TASK_TRIGGER_EVENT = 0,
TASK_TRIGGER_TIME = 1,
TASK_TRIGGER_DAILY = 2,
TASK_TRIGGER_WEEKLY = 3,
TASK_TRIGGER_MONTHLY = 4,
TASK_TRIGGER_MONTHLYDOW = 5,
TASK_TRIGGER_IDLE = 6,
TASK_TRIGGER_REGISTRATION = 7,
TASK_TRIGGER_BOOT = 8,
TASK_TRIGGER_LOGON = 9,
TASK_TRIGGER_SESSION_STATE_CHANGE = 11,
TASK_TRIGGER_CUSTOM_TRIGGER_01 = 12
}
public enum _TASK_CREATION
{
TASK_VALIDATE_ONLY = 1,
TASK_CREATE = 2,
TASK_UPDATE = 4,
TASK_CREATE_OR_UPDATE = 6,
TASK_DISABLE = 8,
TASK_DONT_ADD_PRINCIPAL_ACE = 16,
TASK_IGNORE_REGISTRATION_TRIGGERS = 32
}
public enum _TASK_ACTION_TYPE
{
TASK_ACTION_EXEC = 0,
TASK_ACTION_COM_HANDLER = 5,
TASK_ACTION_SEND_EMAIL = 6,
TASK_ACTION_SHOW_MESSAGE = 7,
}
public enum _TASK_RUNLEVEL
{
TASK_RUNLEVEL_LUA,
TASK_RUNLEVEL_HIGHEST,
}
const string taskPath = "IPFilter";
public static void Execute(bool enable)
{
var type = Type.GetTypeFromProgID("Schedule.Service");
dynamic service = Activator.CreateInstance(type);
service.Connect();
if(!enable)
{
Trace.TraceInformation("Deleting the automatic schedule task...");
var rootFolder = service.GetFolder("\\");
try
{
var existingTask = rootFolder.GetTask(taskPath);
existingTask.Enabled = false;
existingTask.Stop(0);
rootFolder.DeleteTask(taskPath, 0);
Trace.TraceInformation("Successfully deleted the scheduled task.");
}
catch (FileNotFoundException)
{
Trace.TraceInformation("No task found to delete.");
}
return;
}
Trace.TraceInformation("Setting up the automatic schedule...");
var task = service.NewTask(0);
task.RegistrationInfo.Description = "Updates the IP Filter for bit torrent clients";
task.Triggers.Clear();
// Schedule to run daily at 4am
var now = DateTime.Now.AddDays(-1);
var date = new DateTime(now.Year, now.Month, now.Day, 4, 0, 0);
var trigger = task.Triggers.Create(_TASK_TRIGGER_TYPE2.TASK_TRIGGER_DAILY);
trigger.DaysInterval = 1;
trigger.StartBoundary = date.ToString("s");
trigger.RandomDelay = "PT15M"; // Delay randomly by 15 minutes to stagger the amount of requests hitting list servers
// Execute silently
//var action = (IExecAction) task.Actions.Create(_TASK_ACTION_TYPE.TASK_ACTION_EXEC);
var action = task.Actions.Create(_TASK_ACTION_TYPE.TASK_ACTION_EXEC);
action.Path = Process.GetCurrentProcess().MainModule.FileName;
action.Arguments = "/silent";
task.Settings.RunOnlyIfNetworkAvailable = true;
task.Settings.StartWhenAvailable = true;
task.Settings.WakeToRun = false;
task.Settings.Enabled = true;
task.Principal.RunLevel = _TASK_RUNLEVEL.TASK_RUNLEVEL_LUA;
task.Principal.LogonType = _TASK_LOGON_TYPE.TASK_LOGON_INTERACTIVE_TOKEN;
//task.Principal.UserId = identity.Name;
var folder = service.GetFolder("\\");
var registered = folder.RegisterTaskDefinition(taskPath, task, (int)_TASK_CREATION.TASK_CREATE_OR_UPDATE, null,null,_TASK_LOGON_TYPE.TASK_LOGON_INTERACTIVE_TOKEN);
Trace.TraceInformation("Finished creating / updating scheduled task");
}
}
}