Merge pull request #6722 from Icinga/feature/notification-result

[icinga2] / agent / windows-setup-agent / Program.cs

diff --git a/agent/windows-setup-agent/Program.cs b/agent/windows-setup-agent/Program.cs
index a36af0ef928a723a40e82c34c1542bf1711fe252..b22b042eb70e077df81744489982638ec4a3e44b 100644 (file)
--- a/agent/windows-setup-agent/Program.cs
+++ b/agent/windows-setup-agent/Program.cs
@@ -2,25 +2,78 @@
 using System.IO;
 using System.Windows.Forms;
 using Microsoft.Win32;
+using System.Runtime.InteropServices;
+using System.Text;
 
 namespace Icinga
 {
-       static class Program
-       {
+    internal static class NativeMethods
+    {
+        [DllImport("msi.dll", CharSet = CharSet.Unicode)]
+        internal static extern int MsiEnumProducts(int iProductIndex, StringBuilder lpProductBuf);
+
+        [DllImport("msi.dll", CharSet = CharSet.Unicode)]
+        internal static extern Int32 MsiGetProductInfo(string product, string property, [Out] StringBuilder valueBuf, ref Int32 len);
+    }
 
+    static class Program
+       {
                public static string Icinga2InstallDir
                {
                        get
                        {
-                               RegistryKey rk = Registry.LocalMachine.OpenSubKey("SOFTWARE\\Icinga Development Team\\ICINGA2");
+                               StringBuilder szProduct;
+
+                               for (int index = 0; ; index++) {
+                                       szProduct = new StringBuilder(39);
+                                       if (NativeMethods.MsiEnumProducts(index, szProduct) != 0)
+                                               break;
+
+                                       int cbName = 128;
+                                       StringBuilder szName = new StringBuilder(cbName);
+
+                                       if (NativeMethods.MsiGetProductInfo(szProduct.ToString(), "ProductName", szName, ref cbName) != 0)
+                                               continue;
+
+                                       if (szName.ToString() != "Icinga 2")
+                                               continue;
+
+                                       int cbLocation = 1024;
+                                       StringBuilder szLocation = new StringBuilder(cbLocation);
+                                       if (NativeMethods.MsiGetProductInfo(szProduct.ToString(), "InstallLocation", szLocation, ref cbLocation) == 0)
+                                               return szLocation.ToString();
+                               }
+
+                               return "";
+                       }
+               }
+
+               public static string Icinga2DataDir
+               {
+                       get
+                       {
+                               return Environment.GetFolderPath(Environment.SpecialFolder.CommonApplicationData) + "\\icinga2";
+                       }
+               }
 
-                               if (rk == null)
-                                       return "";
+               public static string Icinga2User
+               {
+                       get
+                       {
+                               if (!File.Exists(Icinga2DataDir + "\\etc\\icinga2\\user"))
+                                       return "NT AUTHORITY\\NetworkService";
+                               System.IO.StreamReader file = new System.IO.StreamReader(Icinga2DataDir + "\\etc\\icinga2\\user");
+                               string line = file.ReadLine();
+                               file.Close();
 
-                               return (string)rk.GetValue("");
+                               if (line != null)
+                                       return line;
+                               else
+                                       return "NT AUTHORITY\\NetworkService";
                        }
                }
 
+
                public static void FatalError(Form owner, string message)
                {
                        MessageBox.Show(owner, message, "Icinga 2 Setup Wizard", MessageBoxButtons.OK, MessageBoxIcon.Error);
@@ -38,12 +91,14 @@ namespace Icinga
 
                        string installDir = Program.Icinga2InstallDir;
 
-                       if (installDir == "")
+                       if (installDir == "") {
                                FatalError(null, "Icinga 2 does not seem to be installed properly.");
+                               return;
+                       }
 
                        Form form;
 
-                       if (File.Exists(installDir + "\\etc\\icinga2\\features-enabled\\api.conf"))
+                       if (File.Exists(Program.Icinga2DataDir + "\\etc\\icinga2\\features-enabled\\api.conf"))
                                form = new ServiceStatus();
                        else
                                form = new SetupWizard();