-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathProgram.cs
139 lines (124 loc) · 4.86 KB
/
Program.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
128
129
130
131
132
133
134
135
136
137
138
139
using System;
using System.IO;
using Microsoft.Win32;
using System.Diagnostics;
namespace CodeRedDropper
{
internal class Program
{
static void Write(string str)
{
Console.WriteLine("[" + DateTime.Now.ToString() + "] " + str);
}
static bool IsValidProcess(Process process)
{
if (process != null
&& (process.Id > 8) // A process with an id of 8 or lower is a system process, we shouldn't be trying to access those.
&& (process.MainWindowHandle != IntPtr.Zero))
{
return true;
}
return false;
}
static List<Process> GetFilteredProcesses(string filter)
{
List<Process> returnList = new List<Process>();
Process[] processList = Process.GetProcessesByName(filter);
foreach (Process process in processList)
{
if (IsValidProcess(process))
{
if (process.ProcessName.Contains(filter) || process.MainWindowTitle.Contains(filter))
{
returnList.Add(process);
}
}
}
return returnList;
}
// Launcher should already be closed by this point, so this is just for the sake of sanity checking.
static bool CloseLauncher()
{
List<Process> launchers = GetFilteredProcesses("CodeRedLauncher");
foreach (Process launcher in launchers)
{
if (IsValidProcess(launcher))
{
Write("Found launcher running, attempting to close \"" + launcher.Id.ToString() + "\"...");
try
{
launcher.Kill();
}
catch (Exception ex)
{
Write("Failed to close process: " + ex.ToString());
return false;
}
}
}
return true;
}
static void Main(string[] args)
{
string tempFolder = (Path.GetTempPath() + "\\CodeRedLauncher");
if (Directory.Exists(tempFolder))
{
string newLauncher = (tempFolder + "\\CodeRedLauncher.exe");
string launcherPath = (tempFolder + "\\LauncherPath.txt");
if (File.Exists(newLauncher))
{
if (File.Exists(launcherPath))
{
string oldLauncher = File.ReadAllText(launcherPath);
if (File.Exists(oldLauncher))
{
if (CloseLauncher())
{
Write("Deleting old launcher \"" + oldLauncher + "\"...");
File.Delete(oldLauncher);
Write("Replacing with new launcher...");
File.Move(newLauncher, oldLauncher, true);
if (File.Exists(oldLauncher))
{
Write("Successfully installed the new launcher, attempting to open...");
Process.Start(new ProcessStartInfo(oldLauncher) { UseShellExecute = false });
Environment.Exit(0);
}
else
{
Write("Failed to overwrite old launcher with new one!");
Console.ReadKey();
}
}
else
{
Write("Failed to close the launcher, cannot overwrite with the new one!");
Console.ReadKey();
}
}
else
{
Write("Failed to locate old launcher to replace, cannot install new one!");
Console.ReadKey();
}
}
else
{
Write("Failed to locate old launchers install path, cannot install new one!");
Console.ReadKey();
}
}
else
{
Write("Failed to find newly downloaded launcher, cannot install!");
Console.ReadKey();
}
}
else
{
Write("Failed to find temporary folder, cannot install launcher!");
Console.ReadKey();
}
}
}
}