X-Git-Url: https://granicus.if.org/sourcecode?a=blobdiff_plain;f=agent%2Fwindows-setup-agent%2FProgram.cs;h=b22b042eb70e077df81744489982638ec4a3e44b;hb=6de4cef3ae7517d0d728612c4734c9e53883a9fe;hp=ee30b986f6ae67beb62f52c13b2646d21ae08b51;hpb=954debf2358a63dec582cba9da3259238e8a8905;p=icinga2

diff --git a/agent/windows-setup-agent/Program.cs b/agent/windows-setup-agent/Program.cs
index ee30b986f..b22b042eb 100644
--- a/agent/windows-setup-agent/Program.cs
+++ b/agent/windows-setup-agent/Program.cs
@@ -1,10 +1,85 @@
 using System;
+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
+			{
+				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";
+			}
+		}
+
+		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();
+
+				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);
+			Application.Exit();
+		}
+
 		/// <summary>
 		/// The main entry point for the application.
 		/// </summary>
@@ -13,7 +88,22 @@ namespace Icinga
 		{
 			Application.EnableVisualStyles();
 			Application.SetCompatibleTextRenderingDefault(false);
-			Application.Run(new AgentWizard());
+
+			string installDir = Program.Icinga2InstallDir;
+
+			if (installDir == "") {
+				FatalError(null, "Icinga 2 does not seem to be installed properly.");
+				return;
+			}
+
+			Form form;
+
+			if (File.Exists(Program.Icinga2DataDir + "\\etc\\icinga2\\features-enabled\\api.conf"))
+				form = new ServiceStatus();
+			else
+				form = new SetupWizard();
+
+			Application.Run(form);
 		}
 	}
 }