versaplexd/versaplexd-svc.cs

   1 /*
   2  * Versaplex:
   3  *   Copyright (C)2007-2008 Versabanq Innovations Inc. and contributors.
   4  *       See the included file named LICENSE for license information.
   5  */
   6 using System;
   7 using System.Collections;
   8 using System.Collections.Generic;
   9 using System.ComponentModel;
  10 using System.Configuration.Install;
  11 using System.IO;
  12 using System.ServiceProcess;
  13 using System.Threading;
  14 using Wv;
  15 using Wv.NDesk.Options;
  16 using Wv.Extensions;
  17
  18
  19 [RunInstaller(true)]
  20 public class VersaServiceInstaller : Installer
  21 {
  22     public static bool auth = false;
  23     public static string servname = "Versaplex";
  24     ServiceProcessInstaller spi;
  25     ServiceInstaller si;
  26
  27     public VersaServiceInstaller()
  28     {
  29         spi = new ServiceProcessInstaller();
  30         si = new ServiceInstaller();
  31
  32         // if auth, leave these alone and the ServiceInstaller should prompt
  33         // for authorization.
  34         if (!auth)
  35         {
  36             spi.Account = ServiceAccount.LocalSystem;
  37             spi.Password = null;
  38             spi.Username = null;
  39         }
  40
  41         si.ServiceName = servname;
  42         si.Description = "Provides a secure connection to SQL Server";
  43         si.StartType = ServiceStartMode.Automatic;
  44
  45         Installers.AddRange(new Installer[] { si, spi });
  46     }
  47 }
  48
  49
  50 public class VersaService : ServiceBase
  51 {
  52     WvLog log = new WvLog("versaplexd-svc", WvLog.L.Info);
  53     Thread t = null;
  54
  55     public VersaService()
  56     {
  57         WvLog.recv = new WvLogFile(wv.PathJoin(VersaMain.mydir,
  58                                                "versaplex.log"),
  59                                    "a");
  60         log.print("Initializing.\n");
  61     }
  62
  63     void go()
  64     {
  65         try
  66         {
  67             Versaplexd.Go(VersaMain.config_path,
  68                           "tcp:host=127.0.0.1,port=5561",
  69                           new string[] { "tcp:0.0.0.0:5561" });
  70         }
  71         catch (Exception e)
  72         {
  73             log.print(WvLog.L.Error, "Exception: {0}\n", e.Message);
  74         }
  75     }
  76
  77     protected override void OnStart(string[] args)
  78     {
  79         log.print("Starting.\n");
  80         wv.assert(t == null);
  81         t = new Thread(go);
  82         t.Start();
  83     }
  84
  85     protected override void OnStop()
  86     {
  87         log.print("Stopping...\n");
  88         wv.assert(t != null);
  89         Versaplexd.want_to_die = true;
  90
  91         t.Join();
  92         log.print("Stopped.\n");
  93     }
  94
  95     protected override void OnContinue()
  96     {
  97         log.print("Continuing.");
  98     }
  99 }
 100
 101
 102 public static class VersaMain
 103 {
 104     static void ShowHelp()
 105     {
 106         wv.printerr("Usage: versaplexd-svc <-i | -u>\n" +
 107                     "        -i: install as a Windows service\n" +
 108                     "        -u: uninstall Windows service\n" +
 109                     "        -n: service name (default=Versaplex)\n" +
 110                     "        -A: prompt for account/password info\n");
 111         Environment.Exit(1);
 112     }
 113
 114     static void Uninstall()
 115     {
 116         using (var inst
 117                = new AssemblyInstaller(typeof(VersaService).Assembly,
 118                                        new string[] { "/logfile" }))
 119         {
 120             inst.UseNewContext = true;
 121             inst.Uninstall(new Hashtable());
 122         }
 123     }
 124
 125     static void Install()
 126     {
 127         // We might already be installed, so uninstall first.  But we might
 128         // *not* be installed, which is an error, so ignore it.
 129         try { Uninstall(); } catch { }
 130
 131         using (var inst
 132                = new AssemblyInstaller(typeof(VersaService).Assembly,
 133                                        new string[] { "/logfile" }))
 134         {
 135             inst.UseNewContext = true;
 136
 137             IDictionary state = new Hashtable();
 138             try
 139             {
 140                 inst.Install(state);
 141                 inst.Commit(state);
 142             }
 143             catch
 144             {
 145                 try { inst.Rollback(state); } catch { }
 146                 throw;
 147             }
 148         }
 149
 150         new ServiceController(VersaServiceInstaller.servname).Start();
 151     }
 152
 153     internal static string mydir
 154         = Path.GetDirectoryName(typeof(VersaMain).Assembly.Location);
 155     internal static string config_path
 156         = wv.PathJoin(mydir, "versaplexd.ini");
 157
 158     public static int Main(string[] args)
 159     {
 160         try
 161         {
 162             bool install = false, uninstall = false;
 163             new OptionSet()
 164                 .Add("i|install",
 165                      delegate(string v) { install = true; })
 166                 .Add("u|uninstall",
 167                      delegate(string v) { uninstall = true; })
 168                 .Add("n|name=",
 169                      delegate(string v) { VersaServiceInstaller.servname = v; })
 170                 .Add("A|authorize",
 171                      delegate(string v) { VersaServiceInstaller.auth = true; })
 172                 .Add("?|h|help",
 173                      delegate(string v) { ShowHelp(); })
 174                 .Parse(args);
 175
 176             if (install)
 177             {
 178
 179                 if (!File.Exists(config_path))
 180                     throw new Exception(wv.fmt("Config file missing: {0}",
 181                                                config_path));
 182                 Install();
 183             }
 184             else if (uninstall)
 185                 Uninstall();
 186             else
 187                 ServiceBase.Run(new ServiceBase[] { new VersaService() });
 188             return 0;
 189         }
 190         catch (Exception e)
 191         {
 192             wv.printerr("versaplex: {0}\n", e.Message);
 193             return 99;
 194         }
 195     }
 196 }