public void SetUp() { string seedXml = "<?xml version=\"1.0\" encoding=\"utf-8\" ?>" + "<service> " + " <id>SERVICE_NAME</id> " + " <name>Jenkins Slave</name> " + " <description>This service runs a slave for Jenkins continuous integration system.</description> " + " <executable>C:\\Program Files\\Java\\jre7\\bin\\java.exe</executable> " + " <arguments>-Xrs -jar \\\"%BASE%\\slave.jar\\\" -jnlpUrl ...</arguments> " + " <logmode>rotate</logmode> " + " <extensions> " + " <extension enabled=\"true\" className=\"" + testExtension + "\" id=\"mapNetworDirs\"> " + " <mapping> " + " <map enabled=\"false\" label=\"N:\" uncpath=\"\\\\UNC\"/> " + " <map enabled=\"false\" label=\"M:\" uncpath=\"\\\\UNC2\"/> " + " </mapping> " + " </extension> " + " <extension enabled=\"true\" className=\"" + testExtension + "\" id=\"mapNetworDirs2\"> " + " <mapping> " + " <map enabled=\"false\" label=\"X:\" uncpath=\"\\\\UNC\"/> " + " <map enabled=\"false\" label=\"Y:\" uncpath=\"\\\\UNC2\"/> " + " </mapping> " + " </extension> " + " </extensions> " + "</service>"; _testServiceDescriptor = ServiceDescriptor.FromXML(seedXml); }
public void IncorrectStartMode() { const string SeedXml = "<service>" + "<id>service.exe</id>" + "<name>Service</name>" + "<description>The service.</description>" + "<executable>node.exe</executable>" + "<arguments>My Arguments</arguments>" + "<startmode>rotate</startmode>" + "<logmode>rotate</logmode>" + "<serviceaccount>" + "<domain>" + Domain + "</domain>" + "<user>" + Username + "</user>" + "<password>" + Password + "</password>" + "<allowservicelogon>" + AllowServiceAccountLogonRight + "</allowservicelogon>" + "</serviceaccount>" + "<workingdirectory>" + ExpectedWorkingDirectory + "</workingdirectory>" + @"<logpath>C:\logs</logpath>" + "</service>"; _extendedServiceDescriptor = ServiceDescriptor.FromXML(SeedXml); Assert.That(_extendedServiceDescriptor.StartMode, Is.EqualTo(StartMode.Manual)); }
public WrapperService() { this.descriptor = new ServiceDescriptor(); this.ServiceName = descriptor.Id; this.CanShutdown = true; this.CanStop = true; this.CanPauseAndContinue = false; this.AutoLog = true; this.systemShuttingdown = false; }
public WrapperService(ServiceDescriptor descriptor) { _descriptor = descriptor; ServiceName = _descriptor.Id; CanShutdown = true; CanStop = true; CanPauseAndContinue = false; AutoLog = true; _systemShuttingdown = false; }
public static string CLITest(String[] args, ServiceDescriptor descriptor = null) { using (StringWriter sw = new StringWriter()) { TextWriter tmp = Console.Out; Console.SetOut(sw); WrapperService.Run(args, descriptor ?? DefaultServiceDescriptor); Console.SetOut(tmp); Console.Write(sw.ToString()); return sw.ToString(); } }
public WrapperService(ServiceDescriptor descriptor) { _descriptor = descriptor; ServiceName = _descriptor.Id; ExtensionManager = new WinSWExtensionManager(_descriptor); CanShutdown = true; CanStop = true; CanPauseAndContinue = false; AutoLog = true; _systemShuttingdown = false; // Register the event log provider eventLogProvider.service = this; }
public void SetUp() { string seedXml = "<?xml version=\"1.0\" encoding=\"utf-8\" ?>" + "<service> " + " <id>SERVICE_NAME</id> " + " <name>Jenkins Slave</name> " + " <description>This service runs a slave for Jenkins continuous integration system.</description> " + " <executable>C:\\Program Files\\Java\\jre7\\bin\\java.exe</executable> " + " <arguments>-Xrs -jar \\\"%BASE%\\slave.jar\\\" -jnlpUrl ...</arguments> " + " <logmode>rotate</logmode> " + " <extensions> " + " <extension enabled=\"true\" className=\"" + testExtension + "\" id=\"killRunawayProcess\"> " + " <pidfile>foo/bar/pid.txt</pidfile>" + " <stopTimeout>5000</stopTimeout> " + " <stopParentFirst>true</stopParentFirst>" + " </extension> " + " </extensions> " + "</service>"; _testServiceDescriptor = ServiceDescriptor.FromXML(seedXml); }
public void SetUp() { const string SeedXml = "<service>" + "<id>service.exe</id>" + "<name>Service</name>" + "<description>The service.</description>" + "<executable>node.exe</executable>" + "<arguments>My Arguments</arguments>" + "<logmode>rotate</logmode>" + "<serviceaccount>" + "<domain>" + Domain + "</domain>" + "<user>" + Username + "</user>" + "<password>" + Password + "</password>" + "</serviceaccount>" + "<workingdirectory>" + ExpectedWorkingDirectory + "</workingdirectory>" + @"<logpath>C:\logs</logpath>" + "</service>"; extendedServiceDescriptor = ServiceDescriptor.FromXML(SeedXml); }
public static CLITestResult CLIErrorTest(String[] args, ServiceDescriptor descriptor = null) { StringWriter swOut, swErr; Exception testEx = null; TextWriter tmpOut = Console.Out; TextWriter tmpErr = Console.Error; using (swOut = new StringWriter()) using (swErr = new StringWriter()) try { Console.SetOut(swOut); Console.SetError(swErr); WrapperService.Run(args, descriptor ?? DefaultServiceDescriptor); } catch (Exception ex) { testEx = ex; } finally { Console.SetOut(tmpOut); Console.SetError(tmpErr); Console.WriteLine("\n>>> Output: "); Console.Write(swOut.ToString()); Console.WriteLine("\n>>> Error: "); Console.Write(swErr.ToString()); if (testEx != null) { Console.WriteLine("\n>>> Exception: "); Console.WriteLine(testEx); } } return new CLITestResult(swOut.ToString(), swErr.ToString(), testEx); }
public static void Run(string[] _args) { if (_args.Length > 0) { var d = new ServiceDescriptor(); Win32Services svc = new WmiRoot().GetCollection<Win32Services>(); Win32Service s = svc.Select(d.Id); var args = new List<string>(Array.AsReadOnly(_args)); if (args[0] == "/redirect") { // Redirect output // One might ask why we support this when the caller // can redirect the output easily. The answer is for supporting UAC. // On UAC-enabled Windows such as Vista, SCM operation requires // elevated privileges, thus winsw.exe needs to be launched // accordingly. This in turn limits what the caller can do, // and among other things it makes it difficult for the caller // to read stdout/stderr. Thus redirection becomes handy. var f = new FileStream(args[1], FileMode.Create); var w = new StreamWriter(f); w.AutoFlush = true; Console.SetOut(w); Console.SetError(w); var handle = f.Handle; SetStdHandle(-11, handle); // set stdout SetStdHandle(-12, handle); // set stder args = args.GetRange(2, args.Count - 2); } args[0] = args[0].ToLower(); if (args[0] == "install") { svc.Create( d.Id, d.Caption, "\""+ServiceDescriptor.ExecutablePath+"\"", WMI.ServiceType.OwnProcess, ErrorControl.UserNotified, StartMode.Automatic, d.Interactive, d.ServiceDependencies); // update the description /* Somehow this doesn't work, even though it doesn't report an error Win32Service s = svc.Select(d.Id); s.Description = d.Description; s.Commit(); */ // so using a classic method to set the description. Ugly. Registry.LocalMachine.OpenSubKey("System").OpenSubKey("CurrentControlSet").OpenSubKey("Services") .OpenSubKey(d.Id, true).SetValue("Description", d.Description); } if (args[0] == "uninstall") { if (s == null) return; // there's no such service, so consider it already uninstalled try { s.Delete(); } catch (WmiException e) { if (e.ErrorCode == ReturnValue.ServiceMarkedForDeletion) return; // it's already uninstalled, so consider it a success throw e; } } if (args[0] == "start") { if (s == null) ThrowNoSuchService(); s.StartService(); } if (args[0] == "stop") { if (s == null) ThrowNoSuchService(); s.StopService(); } if (args[0] == "restart") { if (s == null) ThrowNoSuchService(); if(s.Started) s.StopService(); while (s.Started) { Thread.Sleep(1000); s = svc.Select(d.Id); } s.StartService(); } if (args[0] == "status") { if (s == null) Console.WriteLine("NonExistent"); else if (s.Started) Console.WriteLine("Started"); else Console.WriteLine("Stopped"); } if (args[0] == "test") { WrapperService wsvc = new WrapperService(); wsvc.OnStart(args.ToArray()); Thread.Sleep(1000); wsvc.OnStop(); } return; } ServiceBase.Run(new WrapperService()); }
public FileOperations(WrapperService wrapper, ServiceDescriptor descriptor) { _descriptor = descriptor; _wrapper = wrapper; }
// ReSharper disable once InconsistentNaming public static void Run(string[] _args, ServiceDescriptor descriptor = null) { if (_args.Length > 0) { var d = descriptor ?? new ServiceDescriptor(); var svc = new WmiRoot().GetCollection<Win32Services>(); var s = svc.Select(d.Id); var args = new List<string>(Array.AsReadOnly(_args)); if (args[0] == "/redirect") { // Redirect output // One might ask why we support this when the caller // can redirect the output easily. The answer is for supporting UAC. // On UAC-enabled Windows such as Vista, SCM operation requires // elevated privileges, thus winsw.exe needs to be launched // accordingly. This in turn limits what the caller can do, // and among other things it makes it difficult for the caller // to read stdout/stderr. Thus redirection becomes handy. var f = new FileStream(args[1], FileMode.Create); var w = new StreamWriter(f) {AutoFlush = true}; Console.SetOut(w); Console.SetError(w); var handle = f.Handle; Kernel32.SetStdHandle(-11, handle); // set stdout Kernel32.SetStdHandle(-12, handle); // set stder args = args.GetRange(2, args.Count - 2); } args[0] = args[0].ToLower(); if (args[0] == "install") { // Check if the service exists if (s != null) { Console.WriteLine("Service with id '" + d.Id + "' already exists"); Console.WriteLine( "To install the service, delete the existing one or change service Id in the configuration file"); throw new Exception("Installation failure: Service with id '" + d.Id + "' already exists"); } string username = null, password = null; var setallowlogonasaserviceright = false; if (args.Count > 1 && args[1] == "/p") { // we expected username/password on stdin Console.Write("Username: "******"Password: "******"Set Account rights to allow log on as a service (y/n)?: "); var keypressed = Console.ReadKey(); Console.WriteLine(); if (keypressed.Key == ConsoleKey.Y) { setallowlogonasaserviceright = true; } } else { if (d.HasServiceAccount()) { username = d.ServiceAccountUser; password = d.ServiceAccountPassword; setallowlogonasaserviceright = d.AllowServiceAcountLogonRight; } } if (setallowlogonasaserviceright) { LogonAsAService.AddLogonAsAServiceRight(username); } svc.Create( d.Id, d.Caption, "\"" + d.ExecutablePath + "\"", ServiceType.OwnProcess, ErrorControl.UserNotified, d.StartMode, d.Interactive, username, password, d.ServiceDependencies); // update the description /* Somehow this doesn't work, even though it doesn't report an error Win32Service s = svc.Select(d.Id); s.Description = d.Description; s.Commit(); */ // so using a classic method to set the description. Ugly. Registry.LocalMachine.OpenSubKey("System").OpenSubKey("CurrentControlSet").OpenSubKey("Services") .OpenSubKey(d.Id, true).SetValue("Description", d.Description); var actions = d.FailureActions; if (actions.Count > 0) { // set the failure actions using (var scm = new ServiceManager()) { using (var sc = scm.Open(d.Id)) { sc.ChangeConfig(d.ResetFailureAfter, actions); } } } return; } if (args[0] == "uninstall") { if (s == null) { Console.WriteLine("Warning! The service with id '" + d.Id + "' does not exist. Nothing to uninstall"); return; // there's no such service, so consider it already uninstalled } try { s.Delete(); } catch (WmiException e) { if (e.ErrorCode == ReturnValue.ServiceMarkedForDeletion) return; // it's already uninstalled, so consider it a success throw e; } return; } if (args[0] == "start") { if (s == null) ThrowNoSuchService(); s.StartService(); return; } if (args[0] == "stop") { if (s == null) ThrowNoSuchService(); s.StopService(); return; } if (args[0] == "restart") { if (s == null) ThrowNoSuchService(); if (s.Started) s.StopService(); while (s.Started) { Thread.Sleep(1000); s = svc.Select(d.Id); } s.StartService(); return; } if (args[0] == "restart!") { // run restart from another process group. see README.md for why this is useful. var si = new STARTUPINFO(); var pi = new PROCESS_INFORMATION(); var result = Kernel32.CreateProcess(null, d.ExecutablePath + " restart", IntPtr.Zero, IntPtr.Zero, false, 0x200 /*CREATE_NEW_PROCESS_GROUP*/, IntPtr.Zero, null, ref si, out pi); if (!result) { throw new Exception("Failed to invoke restart: " + Marshal.GetLastWin32Error()); } return; } if (args[0] == "status") { if (s == null) Console.WriteLine("NonExistent"); else if (s.Started) Console.WriteLine("Started"); else Console.WriteLine("Stopped"); return; } if (args[0] == "test") { var wsvc = new WrapperService(); wsvc.OnStart(args.ToArray()); Thread.Sleep(1000); wsvc.OnStop(); return; } if (args[0] == "help" || args[0] == "--help" || args[0] == "-h" || args[0] == "-?" || args[0] == "/?") { printHelp(); return; } if (args[0] == "version") { printVersion(); return; } Console.WriteLine("Unknown command: " + args[0]); printAvailableCommandsInfo(); throw new Exception("Unknown command: " + args[0]); } Run(new WrapperService()); }
public static void Run(string[] _args) { if (_args.Length > 0) { var d = new ServiceDescriptor(); Win32Services svc = new WmiRoot().GetCollection<Win32Services>(); Console.WriteLine("Looking for service..."); Win32Service s = svc.Select(d.Id); var args = new List<string>(Array.AsReadOnly(_args)); if (args[0] == "/redirect") { // Redirect output // One might ask why we support this when the caller // can redirect the output easily. The answer is for supporting UAC. // On UAC-enabled Windows such as Vista, SCM operation requires // elevated privileges, thus winsw.exe needs to be launched // accordingly. This in turn limits what the caller can do, // and among other things it makes it difficult for the caller // to read stdout/stderr. Thus redirection becomes handy. var f = new FileStream(args[1], FileMode.Create); var w = new StreamWriter(f); w.AutoFlush = true; Console.SetOut(w); Console.SetError(w); var safeHandle = f.SafeFileHandle; SetStdHandle(-11, safeHandle.DangerousGetHandle()); // set stdout SetStdHandle(-12, safeHandle.DangerousGetHandle()); // set stder args = args.GetRange(2, args.Count - 2); } args[0] = args[0].ToLower(); if (args[0] == "install") { svc.Create( d.Id, d.Caption, "\""+ServiceDescriptor.ExecutablePath+"\"", WMI.ServiceType.OwnProcess, ErrorControl.UserNotified, StartMode.Automatic, d.Interactive, d.ServiceDependencies); // update the description /* Somehow this doesn't work, even though it doesn't report an error Win32Service s = svc.Select(d.Id); s.Description = d.Description; s.Commit(); */ // so using a classic method to set the description. Ugly. Registry.LocalMachine.OpenSubKey("System").OpenSubKey("CurrentControlSet").OpenSubKey("Services") .OpenSubKey(d.Id, true).SetValue("Description", d.Description); Console.WriteLine("Installed"); } if (args[0] == "uninstall") { if (s == null) return; // there's no such service, so consider it already uninstalled try { s.Delete(); } catch (WmiException e) { if (e.ErrorCode == ReturnValue.ServiceMarkedForDeletion) return; // it's already uninstalled, so consider it a success throw e; } Console.WriteLine("Uninstalled"); } if (args[0] == "start") { Console.WriteLine("Requesting start"); if (s == null) ThrowNoSuchService(); s.StartService(); } if (args[0] == "stop") { Console.WriteLine("Requesting stop"); if (s == null) ThrowNoSuchService(); if (s.Started) { s.StopService(); Console.WriteLine("Stopped."); } else { Console.WriteLine("Not running."); } } if (args[0] == "restart") { Console.WriteLine("Requesting restart"); if (s == null) ThrowNoSuchService(); if(s.Started) s.StopService(); while (s.Started) { // XXX: Should give up after a while, and display error-message. Thread.Sleep(1000); s = svc.Select(d.Id); } s.StartService(); } if (args[0] == "status") { if (s == null) Console.WriteLine("NonExistent"); else if (s.Started) Console.WriteLine("Started"); else Console.WriteLine("Stopped"); } if (args[0] == "test") { WrapperService wsvc = new WrapperService(); wsvc.OnStart(args.ToArray()); Thread.Sleep(1000); wsvc.OnStop(); } if (args[0] == "version") { Version version = new Version(); Console.WriteLine("Compiled on " + version.version); } if (args[0] == "specs") { Dictionary<string, string> specs = new Dictionary<string, string>() { {"argument", d.Arguments}, {"beeponshutdown", d.BeepOnShutdown.ToString()}, {"name", d.Caption}, {"description", d.Description}, {"executable", d.Executable}, {"startargument", d.Startarguments}, {"stopexecutable", d.StopExecutable}, {"stopargument", d.Stoparguments}, {"id", d.Id}, {"interactive", d.Interactive.ToString()}, {"logpath", d.LogDirectory}, {"sleeptime", d.SleepTime.ToString()}, {"waithint", d.WaitHint.ToString()}, }; foreach (KeyValuePair<string, string> spec in specs) { Console.WriteLine(spec.Key + ": " + spec.Value); } Console.WriteLine("env:"); foreach (var k in d.EnvironmentVariables) { Console.WriteLine("\t" + k.Key + "=" + k.Value); } Console.WriteLine("downloads:"); foreach (var k in d.Downloads) { Console.WriteLine("\tFrom: " + k.From + "\tTo: " + k.To); } Console.WriteLine("depend:"); foreach (var k in d.ServiceDependencies) { Console.WriteLine("\t" + k); } Console.WriteLine("log:"); Console.WriteLine("\tmode: " + d.LogHandler.logmode()); if (d.LogHandler.GetType() == typeof(TimeBasedRollingLogAppender)) { Console.WriteLine("\tpattern: " + ((TimeBasedRollingLogAppender)d.LogHandler).Pattern); Console.WriteLine("\tperiod: " + ((TimeBasedRollingLogAppender)d.LogHandler).Period); } else if (d.LogHandler.GetType() == typeof(SizeBasedRollingLogAppender)) { Console.WriteLine("\tkeepFiles: " + ((SizeBasedRollingLogAppender)d.LogHandler).FilesToKeep); Console.WriteLine("\tsizeThreshold: " + ((SizeBasedRollingLogAppender)d.LogHandler).SizeThreshold); } } return; } ServiceBase.Run(new WrapperService()); }
private static void InitLoggers(ServiceDescriptor d, bool enableCLILogging) { // TODO: Make logging levels configurable Level logLevel = Level.Debug; Level eventLogLevel = Level.Warn; // Legacy format from winsw-1.x: (DateTime.Now.ToString("yyyy-MM-dd HH:mm:ss") + " - " + message); PatternLayout pl = new PatternLayout { ConversionPattern = "%d %-5p - %m%n" }; pl.ActivateOptions(); List<IAppender> appenders = new List<IAppender>(); // wrapper.log String wrapperLogPath = Path.Combine(d.LogDirectory, d.BaseName + ".wrapper.log"); var wrapperLog = new FileAppender { AppendToFile = true, File = wrapperLogPath, ImmediateFlush = true, Name = "Wrapper file log", Threshold = logLevel, LockingModel = new FileAppender.MinimalLock(), Layout = pl }; wrapperLog.ActivateOptions(); appenders.Add(wrapperLog); // Also display logs in CLI if required if (enableCLILogging) { var consoleAppender = new ConsoleAppender { Name = "Wrapper console log", Threshold = logLevel, Layout = pl, }; consoleAppender.ActivateOptions(); appenders.Add(consoleAppender); } // System log var systemEventLogger = new ServiceEventLogAppender { Name = "System event log", Threshold = eventLogLevel, provider = eventLogProvider }; systemEventLogger.ActivateOptions(); appenders.Add(systemEventLogger); BasicConfigurator.Configure(appenders.ToArray()); }
public static void Run(string[] _args) { if (_args.Length > 0) { var d = new ServiceDescriptor(); Win32Services svc = new WmiRoot().GetCollection <Win32Services>(); Win32Service s = svc.Select(d.Id); var args = new List <string>(Array.AsReadOnly(_args)); if (args[0] == "/redirect") { // Redirect output // One might ask why we support this when the caller // can redirect the output easily. The answer is for supporting UAC. // On UAC-enabled Windows such as Vista, SCM operation requires // elevated privileges, thus winsw.exe needs to be launched // accordingly. This in turn limits what the caller can do, // and among other things it makes it difficult for the caller // to read stdout/stderr. Thus redirection becomes handy. var f = new FileStream(args[1], FileMode.Create); var w = new StreamWriter(f); w.AutoFlush = true; Console.SetOut(w); Console.SetError(w); var handle = f.Handle; SetStdHandle(-11, handle); // set stdout SetStdHandle(-12, handle); // set stder args = args.GetRange(2, args.Count - 2); } args[0] = args[0].ToLower(); if (args[0] == "install") { string username = null, password = null; if (args.Count > 1 && args[1] == "/p") { // we expected username/password on stdin Console.Write("Username: "******"Password: "******"\"" + ServiceDescriptor.ExecutablePath + "\"", WMI.ServiceType.OwnProcess, ErrorControl.UserNotified, StartMode.Automatic, d.Interactive, username, password, d.ServiceDependencies); // update the description /* Somehow this doesn't work, even though it doesn't report an error * Win32Service s = svc.Select(d.Id); * s.Description = d.Description; * s.Commit(); */ // so using a classic method to set the description. Ugly. Registry.LocalMachine.OpenSubKey("System").OpenSubKey("CurrentControlSet").OpenSubKey("Services") .OpenSubKey(d.Id, true).SetValue("Description", d.Description); var actions = d.FailureActions; if (actions.Count > 0) {// set the failure actions using (Advapi32.ServiceManager scm = new Advapi32.ServiceManager()) { using (Advapi32.Service sc = scm.Open(d.Id)) { sc.ChangeConfig(d.ResetFailureAfter, actions); } } } } if (args[0] == "uninstall") { if (s == null) { return; // there's no such service, so consider it already uninstalled } try { s.Delete(); } catch (WmiException e) { if (e.ErrorCode == ReturnValue.ServiceMarkedForDeletion) { return; // it's already uninstalled, so consider it a success } throw e; } } if (args[0] == "start") { if (s == null) { ThrowNoSuchService(); } s.StartService(); } if (args[0] == "stop") { if (s == null) { ThrowNoSuchService(); } s.StopService(); } if (args[0] == "restart") { if (s == null) { ThrowNoSuchService(); } if (s.Started) { s.StopService(); } while (s.Started) { Thread.Sleep(1000); s = svc.Select(d.Id); } s.StartService(); } if (args[0] == "status") { if (s == null) { Console.WriteLine("NonExistent"); } else if (s.Started) { Console.WriteLine("Started"); } else { Console.WriteLine("Stopped"); } } if (args[0] == "test") { WrapperService wsvc = new WrapperService(); wsvc.OnStart(args.ToArray()); Thread.Sleep(1000); wsvc.OnStop(); } return; } ServiceBase.Run(new WrapperService()); }
// ReSharper disable once InconsistentNaming /// <summary> /// Runs the wrapper. /// </summary> /// <param name="_args">Arguments. If empty, WinSW will behave in the service mode. Otherwise - CLI mode</param> /// <param name="descriptor">Service descriptor. If null, it will be initialized within the method. /// In such case configs will be loaded from the XML Configuration File.</param> /// <exception cref="Exception">Any unhandled exception</exception> public static void Run(string[] _args, ServiceDescriptor descriptor = null) { bool isCLIMode = _args.Length > 0; // If descriptor is not specified, initialize the new one (and load configs from there) var d = descriptor ?? new ServiceDescriptor(); // Configure the wrapper-internal logging. // STDIN and STDOUT of the child process will be handled independently. InitLoggers(d, isCLIMode); if (isCLIMode) // CLI mode, in-service mode otherwise { Log.Info("Starting ServiceWrapper in the CLI mode"); // Get service info for the future use Win32Services svc = new WmiRoot().GetCollection <Win32Services>(); Win32Service s = svc.Select(d.Id); var args = new List <string>(Array.AsReadOnly(_args)); if (args[0] == "/redirect") { // Redirect output // One might ask why we support this when the caller // can redirect the output easily. The answer is for supporting UAC. // On UAC-enabled Windows such as Vista, SCM operation requires // elevated privileges, thus winsw.exe needs to be launched // accordingly. This in turn limits what the caller can do, // and among other things it makes it difficult for the caller // to read stdout/stderr. Thus redirection becomes handy. var f = new FileStream(args[1], FileMode.Create); var w = new StreamWriter(f) { AutoFlush = true }; Console.SetOut(w); Console.SetError(w); var handle = f.SafeFileHandle; Kernel32.SetStdHandle(-11, handle); // set stdout Kernel32.SetStdHandle(-12, handle); // set stder args = args.GetRange(2, args.Count - 2); } args[0] = args[0].ToLower(); if (args[0] == "install") { // Check if the service exists if (s != null) { Console.WriteLine("Service with id '" + d.Id + "' already exists"); Console.WriteLine("To install the service, delete the existing one or change service Id in the configuration file"); throw new Exception("Installation failure: Service with id '" + d.Id + "' already exists"); } string username = null, password = null; bool setallowlogonasaserviceright = false; if (args.Count > 1 && args[1] == "/p") { // we expected username/password on stdin Console.Write("Username: "******"Password: "******"Set Account rights to allow log on as a service (y/n)?: "); var keypressed = Console.ReadKey(); Console.WriteLine(); if (keypressed.Key == ConsoleKey.Y) { setallowlogonasaserviceright = true; } } else { if (d.HasServiceAccount()) { username = d.ServiceAccountUser; password = d.ServiceAccountPassword; setallowlogonasaserviceright = d.AllowServiceAcountLogonRight; } } if (setallowlogonasaserviceright) { LogonAsAService.AddLogonAsAServiceRight(username); } svc.Create( d.Id, d.Caption, "\"" + d.ExecutablePath + "\"", ServiceType.OwnProcess, ErrorControl.UserNotified, d.StartMode, d.Interactive, username, password, d.ServiceDependencies); // update the description /* Somehow this doesn't work, even though it doesn't report an error * Win32Service s = svc.Select(d.Id); * s.Description = d.Description; * s.Commit(); */ // so using a classic method to set the description. Ugly. Registry.LocalMachine.OpenSubKey("System").OpenSubKey("CurrentControlSet").OpenSubKey("Services") .OpenSubKey(d.Id, true).SetValue("Description", d.Description); var actions = d.FailureActions; if (actions.Count > 0) {// set the failure actions using (ServiceManager scm = new ServiceManager()) { using (Service sc = scm.Open(d.Id)) { sc.ChangeConfig(d.ResetFailureAfter, actions); } } } return; } if (args[0] == "uninstall") { Log.Info("Uninstalling the service with id '" + d.Id + "'"); if (s == null) { Log.Warn("The service with id '" + d.Id + "' does not exist. Nothing to uninstall"); return; // there's no such service, so consider it already uninstalled } if (s.Started) { // We could fail the opeartion here, but it would be an incompatible change. // So it is just a warning Log.Warn("The service with id '" + d.Id + "' is running. It may be impossible to uninstall it"); } try { s.Delete(); } catch (WmiException e) { if (e.ErrorCode == ReturnValue.ServiceMarkedForDeletion) { Log.Error("Failed to uninstall the service with id '" + d.Id + "'" + ". It has been marked for deletion."); // TODO: change the default behavior to Error? return; // it's already uninstalled, so consider it a success } else { Log.Fatal("Failed to uninstall the service with id '" + d.Id + "'. WMI Error code is '" + e.ErrorCode + "'"); } throw e; } return; } if (args[0] == "start") { if (s == null) { ThrowNoSuchService(); } s.StartService(); return; } if (args[0] == "stop") { if (s == null) { ThrowNoSuchService(); } s.StopService(); return; } if (args[0] == "restart") { if (s == null) { ThrowNoSuchService(); } if (s.Started) { s.StopService(); } while (s.Started) { Thread.Sleep(1000); s = svc.Select(d.Id); } s.StartService(); return; } if (args[0] == "restart!") { // run restart from another process group. see README.md for why this is useful. STARTUPINFO si = new STARTUPINFO(); PROCESS_INFORMATION pi = new PROCESS_INFORMATION(); bool result = Kernel32.CreateProcess(null, d.ExecutablePath + " restart", IntPtr.Zero, IntPtr.Zero, false, 0x200 /*CREATE_NEW_PROCESS_GROUP*/, IntPtr.Zero, null, ref si, out pi); if (!result) { throw new Exception("Failed to invoke restart: " + Marshal.GetLastWin32Error()); } return; } if (args[0] == "status") { Log.Warn("User requested the status"); if (s == null) { Console.WriteLine("NonExistent"); } else if (s.Started) { Console.WriteLine("Started"); } else { Console.WriteLine("Stopped"); } return; } if (args[0] == "test") { WrapperService wsvc = new WrapperService(); wsvc.OnStart(args.ToArray()); Thread.Sleep(1000); wsvc.OnStop(); return; } if (args[0] == "help" || args[0] == "--help" || args[0] == "-h" || args[0] == "-?" || args[0] == "/?") { printHelp(); return; } if (args[0] == "version") { printVersion(); return; } Console.WriteLine("Unknown command: " + args[0]); printAvailableCommandsInfo(); throw new Exception("Unknown command: " + args[0]); } else { Log.Info("Starting ServiceWrapper in the service mode"); } Run(new WrapperService()); }