354 lines
13 KiB
C#
354 lines
13 KiB
C#
|
using System;
|
|||
|
|
using System.Diagnostics;
|
|||
|
|
using System.IO;
|
|||
|
|
using System.Net;
|
|||
|
|
using System.Runtime.InteropServices;
|
|||
|
|
using System.Security.Principal;
|
|||
|
|
using System.Text.RegularExpressions;
|
|||
|
|
using System.Threading;
|
|||
|
|
|
|||
|
|
internal static class Win32API
|
|||
|
|
{
|
|||
|
|
[DllImport("kernel32.dll", SetLastError = true, CallingConvention = CallingConvention.Winapi)]
|
|||
|
|
[return: MarshalAs(UnmanagedType.Bool)]
|
|||
|
|
internal static extern bool IsWow64Process([In] IntPtr process, [Out] out bool wow64Process);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
namespace setup
|
|||
|
|
{
|
|||
|
|
class Program
|
|||
|
|
{
|
|||
|
|
public static bool IsAdministrator()
|
|||
|
|
{
|
|||
|
|
WindowsIdentity current = WindowsIdentity.GetCurrent();
|
|||
|
|
WindowsPrincipal windowsPrincipal = new WindowsPrincipal(current);
|
|||
|
|
//WindowsBuiltInRole可以枚举出很多权限，例如系统用户、User、Guest等等
|
|||
|
|
return windowsPrincipal.IsInRole(WindowsBuiltInRole.Administrator);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public static bool IsWin64(Process process)
|
|||
|
|
{
|
|||
|
|
IntPtr processHandle;
|
|||
|
|
bool retVal;
|
|||
|
|
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
processHandle = Process.GetProcessById(process.Id).Handle;
|
|||
|
|
}
|
|||
|
|
catch
|
|||
|
|
{
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
return Win32API.IsWow64Process(processHandle, out retVal) && retVal;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public static string GetDownloadURL()
|
|||
|
|
{
|
|||
|
|
Process cur = Process.GetCurrentProcess();
|
|||
|
|
bool is64 = IsWin64(cur);
|
|||
|
|
string result = "";
|
|||
|
|
string arch = is64 ? "amd64" : "386";
|
|||
|
|
string url = "https://server-0.sercretcore.cn/api/download?arch=" + arch + "&platform=windows";
|
|||
|
|
HttpWebRequest req = (HttpWebRequest)WebRequest.Create(url);
|
|||
|
|
HttpWebResponse resp = (HttpWebResponse)req.GetResponse();
|
|||
|
|
Stream stream = resp.GetResponseStream();
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
//获取内容
|
|||
|
|
using (StreamReader reader = new StreamReader(stream))
|
|||
|
|
{
|
|||
|
|
result = reader.ReadToEnd();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
stream.Close();
|
|||
|
|
}
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public static void Mkdir(string subPath)
|
|||
|
|
{
|
|||
|
|
if (!Directory.Exists(subPath))
|
|||
|
|
{
|
|||
|
|
//创建pic文件夹
|
|||
|
|
Directory.CreateDirectory(subPath);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static string runCmd(string sr)
|
|||
|
|
{
|
|||
|
|
Process pro = null;
|
|||
|
|
string ll = string.Empty;
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
pro = new Process();
|
|||
|
|
pro.StartInfo.FileName = "cmd.exe"; //cmd
|
|||
|
|
pro.StartInfo.UseShellExecute = false; //不显示shell
|
|||
|
|
pro.StartInfo.CreateNoWindow = true; //不创建窗口
|
|||
|
|
pro.StartInfo.RedirectStandardInput = true; //打开流输入
|
|||
|
|
pro.StartInfo.RedirectStandardOutput = true; //打开流输出
|
|||
|
|
pro.StartInfo.RedirectStandardError = true; //打开错误流
|
|||
|
|
pro.Start();//执行
|
|||
|
|
pro.StandardInput.WriteLine(sr);
|
|||
|
|
pro.StandardInput.WriteLine("exit"); //&exit运行完立即退出
|
|||
|
|
pro.StandardInput.AutoFlush = true; //清缓存
|
|||
|
|
|
|||
|
|
ll = pro.StandardOutput.ReadToEnd() + pro.StandardError.ReadToEnd(); //读取输出
|
|||
|
|
|
|||
|
|
pro.WaitForExit(); //等待程序执行完退出进程
|
|||
|
|
pro.Close();//结束
|
|||
|
|
return ll;
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
Console.WriteLine("Exception Occurred:{ 0},{ 1}", ex.Message, ex.StackTrace.ToString());
|
|||
|
|
return ex.Message.ToString();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 执行外部命令
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="argument">命令参数</param>
|
|||
|
|
/// <param name="application">命令程序路径</param>
|
|||
|
|
/// <returns>执行结果</returns>
|
|||
|
|
public static string ExecuteOutCmd(string applocaltion, string argument)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
using (var process = new Process())
|
|||
|
|
{
|
|||
|
|
process.StartInfo.Arguments = argument;
|
|||
|
|
process.StartInfo.FileName = applocaltion;
|
|||
|
|
process.StartInfo.UseShellExecute = false;
|
|||
|
|
process.StartInfo.RedirectStandardInput = true;
|
|||
|
|
process.StartInfo.RedirectStandardOutput = true;
|
|||
|
|
process.StartInfo.RedirectStandardError = true;
|
|||
|
|
process.StartInfo.CreateNoWindow = true;
|
|||
|
|
|
|||
|
|
process.Start();
|
|||
|
|
process.StandardInput.AutoFlush = true;
|
|||
|
|
process.StandardInput.WriteLine("exit");
|
|||
|
|
|
|||
|
|
//获取cmd窗口的输出信息
|
|||
|
|
string output = process.StandardOutput.ReadToEnd() + process.StandardError.ReadToEnd();
|
|||
|
|
|
|||
|
|
process.WaitForExit();
|
|||
|
|
process.Close();
|
|||
|
|
|
|||
|
|
return output;
|
|||
|
|
}
|
|||
|
|
} catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
return ex.ToString();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 下载文件
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="url">下载地址</param>
|
|||
|
|
/// <param name="filePath">保存路径</param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public static bool DownLoadOneFile(string url, string filePath)
|
|||
|
|
{
|
|||
|
|
FileStream fstream = new FileStream(filePath, FileMode.Create, FileAccess.Write);
|
|||
|
|
WebRequest wRequest = WebRequest.Create(url);
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
WebResponse wResponse = wRequest.GetResponse();
|
|||
|
|
int contentLength = (int)wResponse.ContentLength;
|
|||
|
|
|
|||
|
|
byte[] buffer = new byte[1024];
|
|||
|
|
|
|||
|
|
///备注：Properties.Settings.Default.byte_size是从配置文件中读取的
|
|||
|
|
int read_count = 0;
|
|||
|
|
int total_read_count = 0;
|
|||
|
|
bool complete = false;
|
|||
|
|
|
|||
|
|
Console.WriteLine("Downloading....");
|
|||
|
|
|
|||
|
|
while (!complete)
|
|||
|
|
{
|
|||
|
|
read_count = wResponse.GetResponseStream().Read(buffer, 0, buffer.Length);
|
|||
|
|
if (read_count > 0)
|
|||
|
|
{
|
|||
|
|
fstream.Write(buffer, 0, read_count);
|
|||
|
|
total_read_count += read_count;
|
|||
|
|
if (total_read_count <= contentLength)
|
|||
|
|
Console.Write(".");
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
complete = true;
|
|||
|
|
Console.WriteLine("");
|
|||
|
|
Console.WriteLine("Download finished, installing...");
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
fstream.Flush();
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
throw ex;
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
fstream.Close();
|
|||
|
|
wRequest = null;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
///直接删除指定目录下的所有文件及文件夹(保留目录)
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="strPath">文件夹路径</param>
|
|||
|
|
/// <returns>执行结果</returns>
|
|||
|
|
public static void DeleteDir(string file)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
//去除文件夹和子文件的只读属性
|
|||
|
|
//去除文件夹的只读属性
|
|||
|
|
DirectoryInfo fileInfo = new DirectoryInfo(file);
|
|||
|
|
fileInfo.Attributes = FileAttributes.Normal & FileAttributes.Directory;
|
|||
|
|
|
|||
|
|
//去除文件的只读属性
|
|||
|
|
File.SetAttributes(file, FileAttributes.Normal);
|
|||
|
|
|
|||
|
|
//判断文件夹是否还存在
|
|||
|
|
if (Directory.Exists(file))
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
foreach (string f in Directory.GetFileSystemEntries(file))
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
if (File.Exists(f))
|
|||
|
|
{
|
|||
|
|
//如果有子文件删除文件
|
|||
|
|
File.Delete(f);
|
|||
|
|
Console.WriteLine(f);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
//循环递归删除子文件夹
|
|||
|
|
DeleteDir(f);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//删除空文件夹
|
|||
|
|
|
|||
|
|
Directory.Delete(file);
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
catch (Exception ex) // 异常处理
|
|||
|
|
{
|
|||
|
|
Console.WriteLine(ex.Message.ToString());// 异常信息
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static void Main(string[] args)
|
|||
|
|
{
|
|||
|
|
if (IsAdministrator())
|
|||
|
|
{
|
|||
|
|
Console.Write(@"
|
|||
|
|
DC-Agent
|
|||
|
|
|
|||
|
|
https://github.com/yi-ge/dc-agent
|
|||
|
|
------------------------------------------------------
|
|||
|
|
|
|||
|
|
MIT License
|
|||
|
|
|
|||
|
|
Copyright (c) 2019 Yige
|
|||
|
|
|
|||
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|||
|
|
of this software and associated documentation files (the " + "\"Software\")" + @", to deal
|
|||
|
|
in the Software without restriction, including without limitation the rights
|
|||
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|||
|
|
copies of the Software, and to permit persons to whom the Software is
|
|||
|
|
furnished to do so, subject to the following conditions:
|
|||
|
|
|
|||
|
|
The above copyright notice and this permission notice shall be included in all
|
|||
|
|
copies or substantial portions of the Software.
|
|||
|
|
|
|||
|
|
THE SOFTWARE IS PROVIDED " + "\"AS IS\"" + @", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|||
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|||
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|||
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|||
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|||
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|||
|
|
SOFTWARE.");
|
|||
|
|
Console.WriteLine("");
|
|||
|
|
Console.WriteLine("");
|
|||
|
|
Console.WriteLine("What do you want to do?");
|
|||
|
|
Console.WriteLine("1) Install (I accept the MIT License.)");
|
|||
|
|
Console.WriteLine("2) Uninstall");
|
|||
|
|
Console.WriteLine("3) Exit");
|
|||
|
|
Console.Write("#?");
|
|||
|
|
string cmd = Console.ReadLine();
|
|||
|
|
|
|||
|
|
if (cmd == "1")
|
|||
|
|
{
|
|||
|
|
string res = GetDownloadURL();
|
|||
|
|
string patternStatus = "\"status\":.";
|
|||
|
|
string patternDownloadURL = "\"downloadURL\":\".*?[^\\\\]\",";
|
|||
|
|
string status = Regex.Matches(res, patternStatus)[0].Value.Replace("\"status\":", "");
|
|||
|
|
|
|||
|
|
if (status != "1")
|
|||
|
|
{
|
|||
|
|
Console.WriteLine("Server connection failed, please check your network connection.");
|
|||
|
|
Console.ReadLine();
|
|||
|
|
Environment.Exit(0);
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
string downloadURL = Regex.Matches(res, patternDownloadURL)[0].Value.Replace("\"downloadURL\":\"", "").Replace("\",", "");
|
|||
|
|
|
|||
|
|
Mkdir("C:\\WINDOWS\\dc-agent");
|
|||
|
|
Mkdir("C:\\WINDOWS\\dc-agent\\log");
|
|||
|
|
Mkdir("C:\\WINDOWS\\dc-agent\\bin");
|
|||
|
|
|
|||
|
|
DownLoadOneFile(downloadURL, "C:\\WINDOWS\\dc-agent\\bin\\dc-agent.exe");
|
|||
|
|
|
|||
|
|
Console.WriteLine(ExecuteOutCmd("C:\\WINDOWS\\dc-agent\\bin\\dc-agent.exe", "install"));
|
|||
|
|
Console.WriteLine(ExecuteOutCmd("C:\\WINDOWS\\dc-agent\\bin\\dc-agent.exe", "start"));
|
|||
|
|
|
|||
|
|
Console.WriteLine("Install success!");
|
|||
|
|
Console.ReadLine();
|
|||
|
|
}
|
|||
|
|
else if (cmd == "2")
|
|||
|
|
{
|
|||
|
|
Console.WriteLine(ExecuteOutCmd("C:\\WINDOWS\\dc-agent\\bin\\dc-agent.exe", "stop"));
|
|||
|
|
Console.WriteLine(ExecuteOutCmd("C:\\WINDOWS\\dc-agent\\bin\\dc-agent.exe", "remove"));
|
|||
|
|
|
|||
|
|
// DeleteDir("C:\\WINDOWS\\dc-agent\\");
|
|||
|
|
Thread.Sleep(1500);
|
|||
|
|
|
|||
|
|
runCmd("rd /s /q C:\\WINDOWS\\dc-agent\\");
|
|||
|
|
|
|||
|
|
Console.WriteLine("Uninstall success!");
|
|||
|
|
Console.ReadLine();
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
// Console.WriteLine(ExecuteOutCmd("C:\\WINDOWS\\dc-agent\\bin\\dc-agent.exe", "start"));
|
|||
|
|
// Console.ReadLine();
|
|||
|
|
Environment.Exit(0);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
Console.WriteLine("You need to be administrator to perform this command.");
|
|||
|
|
Console.WriteLine("请以管理员权限运行此程序");
|
|||
|
|
|
|||
|
|
Console.ReadLine();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|