230 lines
9 KiB
C#
230 lines
9 KiB
C#
using System; //Copyright (c) NorbiPeti 2015 - See LICENSE file
|
|
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<SettingType, string> Settings = new Dictionary<SettingType, string>(); //2015.05.21.
|
|
/// <summary>
|
|
/// Csak betöltéskor (2015.05.24.)
|
|
/// </summary>
|
|
public static Dictionary<string, string> LoggedInSettings = new Dictionary<string, string>();
|
|
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)
|
|
{
|
|
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 += (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));
|
|
}
|
|
}
|
|
|
|
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[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<string, string>();
|
|
}
|
|
|
|
public static bool IsLoaded { get; private set; }
|
|
}
|
|
public enum SettingType
|
|
{
|
|
Email,
|
|
WindowState,
|
|
Lang,
|
|
Port,
|
|
LastUsedEmail,
|
|
FileLen,
|
|
ChatWindow,
|
|
Theme
|
|
}
|
|
}
|