Mabinogi Environment Classes

MabinogiEnv.cs

/*!
 * Mabinogi Environment Class
 * Copyright (C) 2012 by Logue <http://logue.be/>
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * as published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

using System;
using System.Collections;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Net;
using System.Text;
using System.Windows.Forms;
using Microsoft.Win32;

namespace MabiLauncher // <-- Change here yourself
{
	class MabiEnvironment
	{
		// Local Enviroment
		public string MabinogiDir;
		public uint LocalVersion;
		// Server Enviroment
		public bool isDownloadable;
		public uint Version;
		public string Arg;
		public string LoginIP;
		public Uri PatchServer;
		public uint Fullver;
		public string LangPack;
		// Define
		public uint Code = 1622;
		public uint LoginPort = 11000;
		public string CrackShieldBinName = "Solaris.exe";

		/// <summary>
		/// Get Mabinogi Environment
		/// </summary>
		public MabiEnvironment() {
			this.MabinogiDir = GetMabiDir();
			this.LocalVersion = GetMabiVer();
			this.isDownloadable = false;
		}
		/// <summary>
		/// Get Mabinogi Environment
		/// </summary>
		/// <param name="url">Url to Patch.txt</param>
		public MabiEnvironment(string url)
		{
			this.MabinogiDir = GetMabiDir();
			this.LocalVersion = GetMabiVer();

			Dictionary<string, string> p = PatchText(url);
			this.isDownloadable = (p["patch_accept"] == "0") ? false : true;
			this.Version = uint.Parse(p["main_version"]);
			this.Arg = p["arg"];
			this.LoginIP = p["login"];
			this.LangPack = p.ContainsKey("lang") ? p["lang"] : "";	// language.pack

			// Maybe Korean server only.
			this.Fullver = p.ContainsKey("main_fullversion") ? uint.Parse(p["main_fullversion"]) : 0;

			// Notice:
			// * US server seems not read main_ftp value.
			// * In some countries uses HTTP for download patch, 
			//   then the beginning of the address http://, to pass to the URI object.
			// * Whether FTP is to be determined by the port.
			this.PatchServer = new Uri("ftp://" +
				(p["main_ftp"] == "mabipatch.nexon.net/game/" ? "mabipatch.nexon.net" : p["main_ftp"]));
		}
		/// <summary>
		/// Get Mabinogi installed directory from Registory.
		/// </summary>
		/// <returns>Fullpath of Mabinogi directory</returns>
		private String GetMabiDir()
		{
			// Get Mabinogi Directory from Registory
			RegistryKey regkey = Registry.CurrentUser.OpenSubKey(@"Software\Nexon\Mabinogi", false);
			if (regkey == null)
			{
				regkey = Registry.CurrentUser.OpenSubKey(@"Software\Nexon\Mabinogi_test", false);
				if (regkey == null)
				{
					Registry.CurrentUser.OpenSubKey(@"Software\Nexon\Mabinogi_hangame", false);
					if (regkey == null) return "C:\\Nexon\\Mabinogi";
				}
			}
			string reg = (string)regkey.GetValue("");	// Returns Mabinogi Directory
			regkey.Close();
			return reg;
		}
		/// <summary>
		/// Read Mabinogi Version from version.dat
		/// </summary>
		/// <returns>Local Version</returns>
		private uint GetMabiVer()
		{
			String MabiDir = this.GetMabiDir();
			// Get Client Version from version.dat
			//string version_dat = MabiDir + "\\version.dat";
			string version_dat = "version.dat";
			if (File.Exists(version_dat))
			{
				byte[] data = File.ReadAllBytes(version_dat);
				return BitConverter.ToUInt32(data, 0);
			}
			else if (File.Exists(MabiDir + "\\" + version_dat))
			{
				// binary is not exists same directory,
				// load registory.
				byte[] data = File.ReadAllBytes(MabiDir + "\\" + version_dat);
				return BitConverter.ToUInt32(data, 0);
			}
			else
			{
				return 0;
			}
		}
		/// <summary>
		/// Fetch and parse patch.txt
		/// </summary>
		/// <param name="url">URL to patch text.</param>
		/// <returns>Key-value data of patch.txt.</returns>
		private Dictionary<string, string> PatchText(string url)
		{
			string line = "";
			ArrayList al = new ArrayList();
			HttpWebRequest webreq = (HttpWebRequest)WebRequest.Create(url);
			HttpWebResponse webres = (HttpWebResponse)webreq.GetResponse();
			Stream st = webres.GetResponseStream();

			// Fetch patch.txt
			Encoding enc = Encoding.GetEncoding("UTF-8");	// assume UTF-8
			StreamReader sr = new StreamReader(st, enc);

			Dictionary<string, string> data = new Dictionary<string, string>();
			while ((line = sr.ReadLine()) != null)
			{
				if (line.Trim().Length == 0 || line[0].Equals('#'))
				{
					continue;
				}
				string[] result = line.Split(new char[] { '=' }, 2);
				if (result.Length == 2)
				{
					data.Add(result[0], result[1]);
				}
			}
			webres.Close();
			sr.Close();
			return data;
		}
		/// <summary>
		/// Launch Mabinogi client. If Crackshild is detected, launch crackshield.
		/// Notice : The parent program MUST be put on the same directory as client.exe. 
		/// </summary>
		/// <param name="form">The window of a parent program.</param>
		/// <returns>When client is launched returns true, otherwise returns false.</returns>
		public bool Launch(String[] args, Form form, bool isLocal = false)
		{
			ProcessStartInfo psi = new ProcessStartInfo();

			String cArgs;

			if (isLocal)
			{
				cArgs = "code:" + this.Code + " ver:" + this.LocalVersion +
						" logip:127.0.0.1 logport:127.0.0.1 " + this.Arg + " " + String.Join(" ", args);
			}
			else
			{
				cArgs = "code:" + this.Code + " ver:" + this.LocalVersion +
						" logip:" + this.LoginIP + " logport:" + this.LoginPort + " " + this.Arg + " " + String.Join(" ", args);
			}

			if (File.Exists("client.exe")) {
				if (File.Exists(this.CrackShieldBinName) &&
					File.Exists("dinput8.dll") &&
					Process.GetProcessesByName(CrackShieldBinName).Length == 0)
				{
					Console.WriteLine("Detect CrackSheild. Launch CrackShield first...");
					RunElevated(CrackShieldBinName, "", form, false);
				}
				
				Console.WriteLine("Command Line: client.exe " + cArgs);
				Console.WriteLine("Launch Mabinogi client.");

				// Multiple launch client is not checked. :)
				return RunElevated("client.exe", cArgs, form, false);

			}
			else
			{
				String MabiDir = this.GetMabiDir();

				// Whwn detect CrackShield, launch CrackShield first.
				// If CrackShield process detected, ignore launch code.
				if (File.Exists(MabiDir + "\\" + CrackShieldBinName) &&
					File.Exists(MabiDir + "\\dinput8.dll") &&
					Process.GetProcessesByName(CrackShieldBinName).Length == 0)
				{
					Console.WriteLine("Detect CrackSheild. Launch CrackShield first...");
					RunElevated(MabiDir + "\\"+ CrackShieldBinName, "", form, false);
				}
				Console.WriteLine("Command Line:" + MabinogiDir + "\\client.exe " + cArgs);
				Console.WriteLine("Launch Mabinogi client.");

				// Multiple launch client is not checked. :)
				if (File.Exists(MabiDir + "\\" + CrackShieldBinName))
				{
					return RunElevated(MabiDir + "\\client.exe", cArgs, form, false);
				}
				else
				{
					return false;
				}
			}
		}
		/// <summary>
		/// Launch other program as Administrator.
		/// </summary>
		/// <param name="fileName">Fullpath of program you would launch</param>
		/// <param name="arguments">Argument of program.</param>
		/// <param name="parentForm">The window of a parent program.</param>
		/// <param name="waitExit">Wait for program exit.</param>
		/// <returns>When launch succeed returns true, cancelled by UAC returns false.</returns>
		private static bool RunElevated(string fileName, string arguments, Form parentForm, bool waitExit)
		{
			if (!File.Exists(fileName))
			{
				throw new FileNotFoundException();
			}

			ProcessStartInfo psi = new ProcessStartInfo();
			psi.UseShellExecute = true;
			psi.FileName = fileName;
			//動詞に「runas」をつける
			psi.Verb = "runas";
			psi.Arguments = arguments;

			if (parentForm != null)
			{
				psi.ErrorDialog = true;
				psi.ErrorDialogParentHandle = parentForm.Handle;
			}

			try
			{
				Process p = Process.Start(psi);
				if (waitExit)
				{
					p.WaitForExit();
				}
			}
			catch (System.ComponentModel.Win32Exception ex)
			{
                Console.WriteLine(ex);
                return false;
			}

			return true;
		}
	}
}