using System; using System.Collections.Generic; using System.Globalization; using System.IO; using System.Linq; using System.Security.Cryptography; using System.Text; using System.Threading.Tasks; namespace MSGer.tk { static class Storage { //2014.08.07. public static string FileName; //public static Dictionary Settings = new Dictionary(); public static Dictionary Settings = new Dictionary(); //2015.05.21. /// /// Csak betöltéskor (2015.05.24.) /// public static Dictionary LoggedInSettings = new Dictionary(); public static readonly string PasswordHash = "PWPassword"; public static string SaltKey; //Bejelentkezéskor kapja meg public static readonly string VIKey = "SGf546HEfg56h45K"; public static void Load(bool loggedin) { if (!loggedin) { SaltKey = "nologinnologinnologinnologin"; FileName = "program.db"; } if (!File.Exists(FileName)) { if (!loggedin) { /*Settings.Add("email", ""); Settings.Add("windowstate", "3"); Settings.Add("lang", CultureInfo.InstalledUICulture.TwoLetterISOLanguageName); Settings.Add("port", "4510"); //Use this to connect to different users <-- És fogalmam sincs, miért angolul írtam... Settings.Add("lastusedemail", "0"); Settings.Add("filelen", "-1"); //(long) Maximum fájlméret, ameddig bemásolhatja a memóriába Settings.Add("chatwindow", "0");*/ //string[] settings = typeof(SettingType).GetEnumNames(); var settings = typeof(SettingType).GetEnumValues(); foreach (SettingType setting in settings) Settings.Add(setting, ""); //TO!DO: Az alapértelmezett értékeket állítsa be IsLoaded = true; //2015.06.04. } } else { Parse(Decrypt(Read(loggedin)), loggedin); if (loggedin) UserInfo.Load(); if (!loggedin) IsLoaded = true; //2015.06.14. } } public static void Save(bool loggedin) { if (!IsLoaded && !loggedin) //!loggedin: 2015.06.14. return; //2015.06.04. if (!loggedin) SaltKey = "nologinnologinnologinnologin"; Write(Encrypt(GetString(loggedin)), loggedin); } public static string GetString(bool loggedin) { string s = ""; if (loggedin) { foreach (var entry in LoggedInSettings) { s += entry.Key; s += "="; s += entry.Value; s += "\n"; } } else { foreach (var entry in Settings) { //s += entry.Key; s += (int)entry.Key; //2015.05.21. s += "="; s += entry.Value; s += "\n"; } } return s; } private static void Parse(string filecontent, bool loggedin) { string[] splitCache = filecontent.Split(new string[] { "\n", "\r\n" }, StringSplitOptions.RemoveEmptyEntries); if (loggedin) { LoggedInSettings = splitCache.ToDictionary( entry => entry.Substring(0, entry.IndexOf("=")), entry => entry.Substring(entry.IndexOf("=") + 1)); } else { Settings = splitCache.ToDictionary( entry => (SettingType)int.Parse(entry.Substring(0, entry.IndexOf("="))), //string --> SettingType: 2015.05.21. entry => entry.Substring(entry.IndexOf("=") + 1)); } /*if (loggedin) LoggedInSettings = tmp; else Settings = tmp;*/ } public static void Parse(string filecontent) //Publikus metódus { string[] splitCache = filecontent.Split(new string[] { "\n", "\r\n" }, StringSplitOptions.RemoveEmptyEntries); var tmp = splitCache.ToDictionary( //entry => "userinfo_" + entry.Substring(0, entry.IndexOf("=")), - 2015.05.10. entry => entry.Substring(0, entry.IndexOf("=")), entry => entry.Substring(entry.IndexOf("=") + 1)); LoggedInSettings = LoggedInSettings.Concat(tmp) .ToLookup(pair => pair.Key, pair => pair.Value) .ToDictionary(group => group.Key, group => group.Last()); } public static byte[] Encrypt(byte[] content, string code) { byte[] keyBytes = new Rfc2898DeriveBytes(PasswordHash, Encoding.ASCII.GetBytes(code)).GetBytes(256 / 8); var symmetricKey = new RijndaelManaged() { Mode = CipherMode.CBC, Padding = PaddingMode.None }; var encryptor = symmetricKey.CreateEncryptor(keyBytes, Encoding.ASCII.GetBytes(VIKey)); byte[] plainTextBytes; int targetsize = content.Length + 16 - content.Length % 16; //Hozzáadja a hosszához a hossz 16-tal való osztásának maradékát - Tehát 16-tal osztható lesz plainTextBytes = new byte[targetsize]; Array.Copy(content, plainTextBytes, content.Length); byte[] cipherTextBytes; using (var memoryStream = new MemoryStream()) { using (var cryptoStream = new CryptoStream(memoryStream, encryptor, CryptoStreamMode.Write)) { cryptoStream.Write(plainTextBytes, 0, plainTextBytes.Length); cryptoStream.FlushFinalBlock(); cipherTextBytes = memoryStream.ToArray(); cryptoStream.Close(); } memoryStream.Close(); } byte[] final = new byte[cipherTextBytes.Length + 4]; Array.Copy(BitConverter.GetBytes(cipherTextBytes.Length), final, 4); Array.Copy(cipherTextBytes, 0, final, 4, cipherTextBytes.Length); return final; } private static byte[] Encrypt(byte[] filecontent) { //2014.09.01. return Encrypt(filecontent, SaltKey); } private static byte[] Encrypt(string filecontent) { return Encrypt(Encoding.UTF8.GetBytes(filecontent)); } public static byte[] Decrypt(byte[] b, bool tr, string code) { int len = BitConverter.ToInt32(b, 0); byte[] cipherTextBytes = new byte[b.Length - 4]; //b.Length: 2015.04.03. Array.Copy(b, 4, cipherTextBytes, 0, b.Length - 4); //Itt még az eredeti, feltöltött hosszal számol byte[] keyBytes = new Rfc2898DeriveBytes(PasswordHash, Encoding.ASCII.GetBytes(code)).GetBytes(256 / 8); var symmetricKey = new RijndaelManaged() { Mode = CipherMode.CBC, Padding = PaddingMode.None }; var decryptor = symmetricKey.CreateDecryptor(keyBytes, Encoding.ASCII.GetBytes(VIKey)); var memoryStream = new MemoryStream(cipherTextBytes); var cryptoStream = new CryptoStream(memoryStream, decryptor, CryptoStreamMode.Read); //byte[] plainTextBytes = new byte[cipherTextBytes.Length]; byte[] plainTextBytes = new byte[len]; //2015.04.03. int decryptedByteCount = cryptoStream.Read(plainTextBytes, 0, plainTextBytes.Length); memoryStream.Close(); cryptoStream.Close(); byte[] ret = new byte[len]; Array.Copy(plainTextBytes, ret, len); return ret; } private static byte[] Decrypt(byte[] b, bool tr) { //2014.09.01. return Decrypt(b, tr, SaltKey); } private static string Decrypt(byte[] b) { byte[] tmp = Decrypt(b, true, SaltKey); return Encoding.UTF8.GetString(tmp, 0, tmp.Length).TrimEnd("\0".ToCharArray()); } private static byte[] Read(bool loggedin) { FileStream fs; if (loggedin) fs = new FileStream(FileName, FileMode.Open); else fs = new FileStream("program.db", FileMode.Open); byte[] b = new byte[4]; fs.Read(b, 0, b.Length); var file_len = BitConverter.ToInt32(b, 0); b = new byte[file_len]; fs.Read(b, 0, b.Length); fs.Close(); return b; } private static void Write(byte[] b, bool loggedin) { var len = BitConverter.GetBytes(b.Length); FileStream fs; if (loggedin) fs = new FileStream(FileName, FileMode.Create); else fs = new FileStream("program.db", FileMode.Create); fs.Write(len, 0, len.Length); fs.Write(b, 0, b.Length); fs.Close(); } public static void Dispose() { LoggedInSettings = new Dictionary(); } public static bool IsLoaded { get; private set; } } public enum SettingType { Email, WindowState, Lang, Port, LastUsedEmail, FileLen, ChatWindow, Theme } }