/// <summary>
/// 应用程序的主入口点。
/// </summary>
static int Main(string[] argv)
{
Environment.CurrentDirectory = Path.GetDirectoryName(Utils.ExecutablePath);
Utils.VerifySignature(Utils.LibraryPath, Utils.ExecutablePath, Path.GetFullPath(TunnelManager.FrpcExecutable));
if (Path.GetFileName(Utils.ExecutablePath) != Consts.ServiceExecutable)
{
if (Environment.UserInteractive)
{
MessageBox.Show("Do NOT rename SakuraFrpService.exe\n请不要重命名 SakuraFrpService.exe", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
}
return(1);
}
if (argv.Length != 0)
{
switch (argv[0])
{
case "--install":
try
{
var dir = new DirectoryInfo(Path.GetDirectoryName(Utils.ExecutablePath));
var acl = dir.GetAccessControl(AccessControlSections.Access);
acl.SetAccessRule(new FileSystemAccessRule(new SecurityIdentifier(WellKnownSidType.LocalServiceSid, null), FileSystemRights.FullControl, InheritanceFlags.ObjectInherit | InheritanceFlags.ContainerInherit, PropagationFlags.None, AccessControlType.Allow));
dir.SetAccessControl(acl);
ManagedInstallerClass.InstallHelper(new string[] { Utils.ExecutablePath });
var result = Utils.SetServicePermission();
if (result != null)
{
MessageBox.Show("无法设置服务操作权限, 启动器某些功能可能无法正常运行\n请记录下面的错误并反馈给开发者:\n" + result, "错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(3);
}
}
catch (Exception e) when(e.InnerException is Win32Exception w32 && w32.NativeErrorCode == 1073) // ERROR_SERVICE_EXISTS
{
MessageBox.Show("服务已存在", "操作失败", MessageBoxButtons.OK, MessageBoxIcon.Information);
}
catch (Exception e)
{
MessageBox.Show(e.ToString(), "操作失败", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(2);
}
return(0);
case "--uninstall":
try
{
ManagedInstallerClass.InstallHelper(new string[] { "/u", Utils.ExecutablePath });
}
catch (Exception e) when(e.InnerException is Win32Exception w32 && w32.NativeErrorCode == 1060) // ERROR_SERVICE_DOES_NOT_EXIST
{
MessageBox.Show("服务未安装", "操作失败", MessageBoxButtons.OK, MessageBoxIcon.Information);
}
catch (Exception e)
{
MessageBox.Show(e.ToString(), "操作失败", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(2);
}
return(0);
case "--daemon":
var args = new string[argv.Length - 1];
Array.Copy(argv, 1, args, 0, args.Length);
var main = new MainService(true);
main.DaemonRun(args);
return(main.ExitCode);
}
}
if (Environment.UserInteractive)
{
MessageBox.Show("You can't start the service directly.\nTo run as daemon, pass --daemon as the first parameter.\nKeep in mind that action above should be done by SakuraFrpLauncher automatically, not by user.", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(-1);
}
ServiceBase[] ServicesToRun;
ServicesToRun = new ServiceBase[]
{
new MainService(false)
};
ServiceBase.Run(ServicesToRun);
return(0);
}
/// <summary>
/// 应用程序的主入口点。
/// </summary>
static int Main(string[] argv)
{
Environment.CurrentDirectory = Path.GetDirectoryName(Utils.ExecutablePath);
Utils.VerifySignature(Utils.LibraryPath, Utils.ExecutablePath, Path.GetFullPath(TunnelManager.FrpcExecutable));
Utils.ValidateSettings();
if (Path.GetFileName(Utils.ExecutablePath) != Consts.ServiceExecutable)
{
if (Environment.UserInteractive)
{
MessageBox.Show("Do NOT rename SakuraFrpService.exe\n请不要重命名 SakuraFrpService.exe", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
}
return(1);
}
if (argv.Length != 0)
{
switch (argv[0])
{
case "--install":
try
{
var result = InstallService();
if (result != null)
{
MessageBox.Show("无法设置服务操作权限, 启动器可能无法正常运行\n请截图此错误并联系管理员:\n" + result, "错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(3);
}
}
catch (Exception e) when(e.InnerException is Win32Exception w32 && w32.NativeErrorCode == 1073) // ERROR_SERVICE_EXISTS
{
// Ensure the service was installed correctly
using (var searcher = new ManagementObjectSearcher("SELECT * FROM Win32_Service WHERE Name = '" + Consts.ServiceName + "'"))
using (var collection = searcher.Get())
{
var service = collection.OfType <ManagementObject>().FirstOrDefault();
if (service != null)
{
var oldPath = Path.GetFullPath((service.GetPropertyValue("PathName") as string).Trim('"'));
var newPath = Path.GetFullPath(Consts.ServiceExecutable);
if (oldPath != newPath)
{
// Delete bad service and reinstall
try
{
UninstallService();
InstallService();
}
catch (Exception e1)
{
MessageBox.Show(e1.ToString(), "操作失败", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(2);
}
}
}
}
}
catch (Exception e)
{
MessageBox.Show(e.ToString(), "操作失败", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(2);
}
return(0);
case "--uninstall":
try
{
UninstallService();
}
catch (Exception e) when(e.InnerException is Win32Exception w32 && w32.NativeErrorCode == 1060) // ERROR_SERVICE_DOES_NOT_EXIST
{
// Can be safely ignored
}
catch (Exception e)
{
MessageBox.Show(e.ToString(), "操作失败", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(2);
}
return(0);
case "--daemon":
var args = new string[argv.Length - 1];
Array.Copy(argv, 1, args, 0, args.Length);
var main = new MainService(true);
main.DaemonRun(args);
return(main.ExitCode);
case "--update":
if (argv.Length < 2)
{
MessageBox.Show("参数错误", "错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(1);
}
File.WriteAllBytes(Consts.UpdaterExecutable, Properties.Resources.Updater);
Process.Start(Consts.UpdaterExecutable, '"' + argv[1] + '"' + (argv.Length >= 3 ? ' ' + argv[2] : ""));
Environment.Exit(0);
return(0);
}
}
if (Environment.UserInteractive)
{
MessageBox.Show("You can't start the service directly.\nTo run as daemon, pass --daemon as the first parameter.\nKeep in mind that action above should be done by SakuraFrpLauncher automatically, not by user.", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
return(-1);
}
AppMutex = new Mutex(true, "SakuraFrpService_" + Utils.InstallationHash, out bool created);
if (!created)
{
if (Environment.UserInteractive)
{
MessageBox.Show("请勿重复开启守护进程", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
}
return(-1);
}
ServiceBase[] ServicesToRun;
ServicesToRun = new ServiceBase[]
{
new MainService(false)
};
ServiceBase.Run(ServicesToRun);
return(0);
}
