From 1ff3980c98e3c228291b35266fd100edd5a65965 Mon Sep 17 00:00:00 2001 From: Carson McManus Date: Sat, 27 Aug 2016 18:42:43 -0400 Subject: [PATCH] added namespace SteamGuard --- Manifest.cs | 1032 ++++++++++++++++++++++----------------------- Program.cs | 1153 ++++++++++++++++++++++++++------------------------- 2 files changed, 1095 insertions(+), 1090 deletions(-) diff --git a/Manifest.cs b/Manifest.cs index f8cac74..45e0035 100644 --- a/Manifest.cs +++ b/Manifest.cs @@ -8,535 +8,537 @@ using System.Text; using System.Threading.Tasks; using System.Security.Cryptography; -public class Manifest +namespace SteamGuard { - private const int PBKDF2_ITERATIONS = 50000; //Set to 50k to make program not unbearably slow. May increase in future. - private const int SALT_LENGTH = 8; - private const int KEY_SIZE_BYTES = 32; - private const int IV_LENGTH = 16; - - [JsonProperty("encrypted")] - public bool Encrypted { get; set; } - - [JsonProperty("first_run")] - public bool FirstRun { get; set; } = true; - - [JsonProperty("entries")] - public List Entries { get; set; } - - [JsonProperty("periodic_checking")] - public bool PeriodicChecking { get; set; } = false; - - [JsonProperty("periodic_checking_interval")] - public int PeriodicCheckingInterval { get; set; } = 5; - - [JsonProperty("periodic_checking_checkall")] - public bool CheckAllAccounts { get; set; } = false; - - [JsonProperty("auto_confirm_market_transactions")] - public bool AutoConfirmMarketTransactions { get; set; } = false; - - [JsonProperty("auto_confirm_trades")] - public bool AutoConfirmTrades { get; set; } = false; - - private static Manifest _manifest { get; set; } - - public static Manifest GetManifest(bool forceLoad = false) - { - // Check if already staticly loaded - if (_manifest != null && !forceLoad) - { - return _manifest; - } - - // Find config dir and manifest file - string maFile = Path.Combine(Program.SteamGuardPath, "manifest.json"); - - // If there's no config dir, create it - if (!Directory.Exists(Program.SteamGuardPath)) - { - _manifest = _generateNewManifest(); - return _manifest; - } - - // If there's no manifest, create it - if (!File.Exists(maFile)) - { - if (Program.Verbose) Console.WriteLine("warn: No manifest file found at {0}", maFile); - bool? createNewManifest = Program.SteamGuardPath == - Program.defaultSteamGuardPath.Replace("~", Environment.GetEnvironmentVariable("HOME")) ? true : (bool?) null; - while (createNewManifest == null) - { - Console.Write($"Generate new manifest.json in {Program.SteamGuardPath}? [Y/n]"); - var answer = Console.ReadLine(); - if (answer != null) - createNewManifest = !answer.StartsWith("n") && !answer.StartsWith("N"); - } - if ((bool) createNewManifest) - { - _manifest = _generateNewManifest(true); - return _manifest; - } - return null; - } - - try - { - string manifestContents = File.ReadAllText(maFile); - _manifest = JsonConvert.DeserializeObject(manifestContents); - - if (_manifest.Encrypted && _manifest.Entries.Count == 0) - { - _manifest.Encrypted = false; - _manifest.Save(); - } - - _manifest.RecomputeExistingEntries(); - - return _manifest; - } - catch (Exception ex) - { - Console.WriteLine("error: Could not open manifest file: {0}", ex.ToString()); - return null; - } - } - - private static Manifest _generateNewManifest(bool scanDir = false) - { - if (Program.Verbose) Console.WriteLine("Generating new manifest..."); - - // No directory means no manifest file anyways. - Manifest newManifest = new Manifest(); - newManifest.Encrypted = false; - newManifest.PeriodicCheckingInterval = 5; - newManifest.PeriodicChecking = false; - newManifest.AutoConfirmMarketTransactions = false; - newManifest.AutoConfirmTrades = false; - newManifest.Entries = new List(); - newManifest.FirstRun = true; - - // Take a pre-manifest version and generate a manifest for it. - if (scanDir) - { - if (Directory.Exists(Program.SteamGuardPath)) - { - DirectoryInfo dir = new DirectoryInfo(Program.SteamGuardPath); - var files = dir.GetFiles(); - - foreach (var file in files) - { - if (file.Extension != ".maFile") continue; - - string contents = File.ReadAllText(file.FullName); - try - { - SteamGuardAccount account = JsonConvert.DeserializeObject(contents); - ManifestEntry newEntry = new ManifestEntry() - { - Filename = file.Name, - SteamID = account.Session.SteamID - }; - newManifest.Entries.Add(newEntry); - } - catch (Exception ex) - { - if (Program.Verbose) Console.WriteLine("warn: {0}", ex.Message); - } - } - - if (newManifest.Entries.Count > 0) - { - newManifest.Save(); - newManifest.PromptSetupPassKey(true); - } - } - } - - if (newManifest.Save()) - { - return newManifest; - } - - return null; - } - - public class IncorrectPassKeyException : Exception { } - public class ManifestNotEncryptedException : Exception { } - - // TODO: move PromptForPassKey to Program.cs - // TODO: make PromptForPassKey more secure - public string PromptForPassKey() - { - if (!this.Encrypted) - { - throw new ManifestNotEncryptedException(); - } - - bool passKeyValid = false; - string passKey = ""; - while (!passKeyValid) - { - Console.WriteLine("Please enter encryption password: "); - passKey = Console.ReadLine(); - if (passKey == "") - continue; - passKeyValid = this.VerifyPasskey(passKey); - if (!passKeyValid) - { - Console.WriteLine("Incorrect."); - } - } - return passKey; - } - - // TODO: move PromptSetupPassKey to Program.cs - public string PromptSetupPassKey(bool inAccountSetupProcess = false) - { - if (inAccountSetupProcess) - { - Console.Write("Would you like to use encryption? [Y/n] "); - string doEncryptAnswer = Console.ReadLine(); - if (doEncryptAnswer == "n" || doEncryptAnswer == "N") - { - Console.WriteLine("WARNING: You chose to not encrypt your files. Doing so imposes a security risk for yourself. If an attacker were to gain access to your computer, they could completely lock you out of your account and steal all your items."); - Console.WriteLine("You may add encryption later using the --encrypt argument."); - return null; - } - } - - string newPassKey = ""; - string confirmPassKey = ""; - do - { - Console.Write("Enter" + (inAccountSetupProcess ? " " : " new ") + "passkey: "); - newPassKey = Console.ReadLine(); - Console.Write("Confirm" + (inAccountSetupProcess ? " " : " new ") + "passkey: "); - confirmPassKey = Console.ReadLine(); - - if (newPassKey != confirmPassKey) - { - Console.WriteLine("Passkeys do not match."); - } - } while (newPassKey != confirmPassKey || newPassKey == ""); - - return newPassKey; - } - - public SteamAuth.SteamGuardAccount[] GetAllAccounts(string passKey = null, int limit = -1) - { - if (passKey == null && this.Encrypted) return new SteamGuardAccount[0]; - - List accounts = new List(); - foreach (var entry in this.Entries) - { - var account = GetAccount(entry, passKey); - if (account == null) continue; - accounts.Add(account); - - if (limit != -1 && limit >= accounts.Count) - break; - } - - return accounts.ToArray(); - } - - public SteamGuardAccount GetAccount(ManifestEntry entry, string passKey = null) - { - string fileText = ""; - Stream stream = null; - RijndaelManaged aes256; - - if (this.Encrypted) - { - MemoryStream ms = new MemoryStream(Convert.FromBase64String(File.ReadAllText(Path.Combine(Program.SteamGuardPath, entry.Filename)))); - byte[] key = GetEncryptionKey(passKey, entry.Salt); - - aes256 = new RijndaelManaged - { - IV = Convert.FromBase64String(entry.IV), - Key = key, - Padding = PaddingMode.PKCS7, - Mode = CipherMode.CBC - }; - - ICryptoTransform decryptor = aes256.CreateDecryptor(aes256.Key, aes256.IV); - stream = new CryptoStream(ms, decryptor, CryptoStreamMode.Read); - } - else - { - FileStream fileStream = File.OpenRead(Path.Combine(Program.SteamGuardPath, entry.Filename)); - stream = fileStream; - } - - if (Program.Verbose) Console.WriteLine("Decrypting..."); - using (StreamReader reader = new StreamReader(stream)) - { - fileText = reader.ReadToEnd(); - } - stream.Close(); - - return JsonConvert.DeserializeObject(fileText); - } - - public bool VerifyPasskey(string passkey) - { - if (!this.Encrypted || this.Entries.Count == 0) return true; - - var accounts = this.GetAllAccounts(passkey, 1); - return accounts != null && accounts.Length == 1; - } - - public bool RemoveAccount(SteamGuardAccount account, bool deleteMaFile = true) - { - ManifestEntry entry = (from e in this.Entries where e.SteamID == account.Session.SteamID select e).FirstOrDefault(); - if (entry == null) return true; // If something never existed, did you do what they asked? - - string filename = Path.Combine(Program.SteamGuardPath, entry.Filename); - this.Entries.Remove(entry); - - if (this.Entries.Count == 0) - { - this.Encrypted = false; - } - - if (this.Save() && deleteMaFile) - { - try - { - File.Delete(filename); - return true; - } - catch (Exception) - { - return false; - } - } - - return false; - } - - public bool SaveAccount(SteamGuardAccount account, bool encrypt, string passKey = null, string salt = null, string iV = null) - { - if (encrypt && (String.IsNullOrEmpty(passKey) || String.IsNullOrEmpty(salt) || String.IsNullOrEmpty(iV))) return false; - - string jsonAccount = JsonConvert.SerializeObject(account); - - string filename = account.Session.SteamID.ToString() + ".maFile"; - if (Program.Verbose) Console.WriteLine($"Saving account {account.AccountName} to {filename}..."); - - ManifestEntry newEntry = new ManifestEntry() - { - SteamID = account.Session.SteamID, - IV = iV, - Salt = salt, - Filename = filename - }; - - bool foundExistingEntry = false; - for (int i = 0; i < this.Entries.Count; i++) - { - if (this.Entries[i].SteamID == account.Session.SteamID) - { - this.Entries[i] = newEntry; - foundExistingEntry = true; - break; - } - } - - if (!foundExistingEntry) - { - this.Entries.Add(newEntry); - } - - bool wasEncrypted = this.Encrypted; - this.Encrypted = encrypt; - - if (!this.Save()) - { - this.Encrypted = wasEncrypted; - return false; - } - - try - { - Stream stream = null; - MemoryStream ms = null; - RijndaelManaged aes256; - - if (encrypt) - { - ms = new MemoryStream(); - byte[] key = GetEncryptionKey(passKey, newEntry.Salt); - - aes256 = new RijndaelManaged - { - IV = Convert.FromBase64String(newEntry.IV), - Key = key, - Padding = PaddingMode.PKCS7, - Mode = CipherMode.CBC - }; - - ICryptoTransform encryptor = aes256.CreateEncryptor(aes256.Key, aes256.IV); - stream = new CryptoStream(ms, encryptor, CryptoStreamMode.Write); - } - else - { - // An unencrypted maFile is shorter than the encrypted version, - // so when an unencrypted maFile gets written this way, the file does not get wiped - // leaving encrypted text after the final } bracket. Deleting and recreating the file fixes this. - File.Delete(Path.Combine(Program.SteamGuardPath, newEntry.Filename)); - stream = File.OpenWrite(Path.Combine(Program.SteamGuardPath, newEntry.Filename)); // open or create - } - - using (StreamWriter writer = new StreamWriter(stream)) - { - writer.Write(jsonAccount); - } - - if (encrypt) - { - File.WriteAllText(Path.Combine(Program.SteamGuardPath, newEntry.Filename), Convert.ToBase64String(ms.ToArray())); - } - - stream.Close(); - return true; - } - catch (Exception ex) - { - if (Program.Verbose) Console.WriteLine("error: {0}", ex.ToString()); - return false; - } - } - - public bool Save() - { - string filename = Path.Combine(Program.SteamGuardPath, "manifest.json"); - if (!Directory.Exists(Program.SteamGuardPath)) - { - try - { - if (Program.Verbose) Console.WriteLine("Creating {0}", Program.SteamGuardPath); - Directory.CreateDirectory(Program.SteamGuardPath); - } - catch (Exception ex) - { + public class Manifest + { + private const int PBKDF2_ITERATIONS = 50000; //Set to 50k to make program not unbearably slow. May increase in future. + private const int SALT_LENGTH = 8; + private const int KEY_SIZE_BYTES = 32; + private const int IV_LENGTH = 16; + + [JsonProperty("encrypted")] + public bool Encrypted { get; set; } + + [JsonProperty("first_run")] + public bool FirstRun { get; set; } = true; + + [JsonProperty("entries")] + public List Entries { get; set; } + + [JsonProperty("periodic_checking")] + public bool PeriodicChecking { get; set; } = false; + + [JsonProperty("periodic_checking_interval")] + public int PeriodicCheckingInterval { get; set; } = 5; + + [JsonProperty("periodic_checking_checkall")] + public bool CheckAllAccounts { get; set; } = false; + + [JsonProperty("auto_confirm_market_transactions")] + public bool AutoConfirmMarketTransactions { get; set; } = false; + + [JsonProperty("auto_confirm_trades")] + public bool AutoConfirmTrades { get; set; } = false; + + private static Manifest _manifest { get; set; } + + public static Manifest GetManifest(bool forceLoad = false) + { + // Check if already staticly loaded + if (_manifest != null && !forceLoad) + { + return _manifest; + } + + // Find config dir and manifest file + string maFile = Path.Combine(Program.SteamGuardPath, "manifest.json"); + + // If there's no config dir, create it + if (!Directory.Exists(Program.SteamGuardPath)) + { + _manifest = _generateNewManifest(); + return _manifest; + } + + // If there's no manifest, create it + if (!File.Exists(maFile)) + { + if (Program.Verbose) Console.WriteLine("warn: No manifest file found at {0}", maFile); + bool? createNewManifest = Program.SteamGuardPath == + Program.defaultSteamGuardPath.Replace("~", Environment.GetEnvironmentVariable("HOME")) ? true : (bool?) null; + while (createNewManifest == null) + { + Console.Write($"Generate new manifest.json in {Program.SteamGuardPath}? [Y/n]"); + var answer = Console.ReadLine(); + if (answer != null) + createNewManifest = !answer.StartsWith("n") && !answer.StartsWith("N"); + } + if ((bool) createNewManifest) + { + _manifest = _generateNewManifest(true); + return _manifest; + } + return null; + } + + try + { + string manifestContents = File.ReadAllText(maFile); + _manifest = JsonConvert.DeserializeObject(manifestContents); + + if (_manifest.Encrypted && _manifest.Entries.Count == 0) + { + _manifest.Encrypted = false; + _manifest.Save(); + } + + _manifest.RecomputeExistingEntries(); + + return _manifest; + } + catch (Exception ex) + { + Console.WriteLine("error: Could not open manifest file: {0}", ex.ToString()); + return null; + } + } + + private static Manifest _generateNewManifest(bool scanDir = false) + { + if (Program.Verbose) Console.WriteLine("Generating new manifest..."); + + // No directory means no manifest file anyways. + Manifest newManifest = new Manifest(); + newManifest.Encrypted = false; + newManifest.PeriodicCheckingInterval = 5; + newManifest.PeriodicChecking = false; + newManifest.AutoConfirmMarketTransactions = false; + newManifest.AutoConfirmTrades = false; + newManifest.Entries = new List(); + newManifest.FirstRun = true; + + // Take a pre-manifest version and generate a manifest for it. + if (scanDir) + { + if (Directory.Exists(Program.SteamGuardPath)) + { + DirectoryInfo dir = new DirectoryInfo(Program.SteamGuardPath); + var files = dir.GetFiles(); + + foreach (var file in files) + { + if (file.Extension != ".maFile") continue; + + string contents = File.ReadAllText(file.FullName); + try + { + SteamGuardAccount account = JsonConvert.DeserializeObject(contents); + ManifestEntry newEntry = new ManifestEntry() + { + Filename = file.Name, + SteamID = account.Session.SteamID + }; + newManifest.Entries.Add(newEntry); + } + catch (Exception ex) + { + if (Program.Verbose) Console.WriteLine("warn: {0}", ex.Message); + } + } + + if (newManifest.Entries.Count > 0) + { + newManifest.Save(); + newManifest.PromptSetupPassKey(true); + } + } + } + + if (newManifest.Save()) + { + return newManifest; + } + + return null; + } + + public class IncorrectPassKeyException : Exception { } + public class ManifestNotEncryptedException : Exception { } + + // TODO: move PromptForPassKey to Program.cs + // TODO: make PromptForPassKey more secure + public string PromptForPassKey() + { + if (!this.Encrypted) + { + throw new ManifestNotEncryptedException(); + } + + bool passKeyValid = false; + string passKey = ""; + while (!passKeyValid) + { + Console.WriteLine("Please enter encryption password: "); + passKey = Console.ReadLine(); + if (passKey == "") + continue; + passKeyValid = this.VerifyPasskey(passKey); + if (!passKeyValid) + { + Console.WriteLine("Incorrect."); + } + } + return passKey; + } + + // TODO: move PromptSetupPassKey to Program.cs + public string PromptSetupPassKey(bool inAccountSetupProcess = false) + { + if (inAccountSetupProcess) + { + Console.Write("Would you like to use encryption? [Y/n] "); + string doEncryptAnswer = Console.ReadLine(); + if (doEncryptAnswer == "n" || doEncryptAnswer == "N") + { + Console.WriteLine("WARNING: You chose to not encrypt your files. Doing so imposes a security risk for yourself. If an attacker were to gain access to your computer, they could completely lock you out of your account and steal all your items."); + Console.WriteLine("You may add encryption later using the --encrypt argument."); + return null; + } + } + + string newPassKey = ""; + string confirmPassKey = ""; + do + { + Console.Write("Enter" + (inAccountSetupProcess ? " " : " new ") + "passkey: "); + newPassKey = Console.ReadLine(); + Console.Write("Confirm" + (inAccountSetupProcess ? " " : " new ") + "passkey: "); + confirmPassKey = Console.ReadLine(); + + if (newPassKey != confirmPassKey) + { + Console.WriteLine("Passkeys do not match."); + } + } while (newPassKey != confirmPassKey || newPassKey == ""); + + return newPassKey; + } + + public SteamAuth.SteamGuardAccount[] GetAllAccounts(string passKey = null, int limit = -1) + { + if (passKey == null && this.Encrypted) return new SteamGuardAccount[0]; + + List accounts = new List(); + foreach (var entry in this.Entries) + { + var account = GetAccount(entry, passKey); + if (account == null) continue; + accounts.Add(account); + + if (limit != -1 && limit >= accounts.Count) + break; + } + + return accounts.ToArray(); + } + + public SteamGuardAccount GetAccount(ManifestEntry entry, string passKey = null) + { + string fileText = ""; + Stream stream = null; + RijndaelManaged aes256; + + if (this.Encrypted) + { + MemoryStream ms = new MemoryStream(Convert.FromBase64String(File.ReadAllText(Path.Combine(Program.SteamGuardPath, entry.Filename)))); + byte[] key = GetEncryptionKey(passKey, entry.Salt); + + aes256 = new RijndaelManaged + { + IV = Convert.FromBase64String(entry.IV), + Key = key, + Padding = PaddingMode.PKCS7, + Mode = CipherMode.CBC + }; + + ICryptoTransform decryptor = aes256.CreateDecryptor(aes256.Key, aes256.IV); + stream = new CryptoStream(ms, decryptor, CryptoStreamMode.Read); + } + else + { + FileStream fileStream = File.OpenRead(Path.Combine(Program.SteamGuardPath, entry.Filename)); + stream = fileStream; + } + + if (Program.Verbose) Console.WriteLine("Decrypting..."); + using (StreamReader reader = new StreamReader(stream)) + { + fileText = reader.ReadToEnd(); + } + stream.Close(); + + return JsonConvert.DeserializeObject(fileText); + } + + public bool VerifyPasskey(string passkey) + { + if (!this.Encrypted || this.Entries.Count == 0) return true; + + var accounts = this.GetAllAccounts(passkey, 1); + return accounts != null && accounts.Length == 1; + } + + public bool RemoveAccount(SteamGuardAccount account, bool deleteMaFile = true) + { + ManifestEntry entry = (from e in this.Entries where e.SteamID == account.Session.SteamID select e).FirstOrDefault(); + if (entry == null) return true; // If something never existed, did you do what they asked? + + string filename = Path.Combine(Program.SteamGuardPath, entry.Filename); + this.Entries.Remove(entry); + + if (this.Entries.Count == 0) + { + this.Encrypted = false; + } + + if (this.Save() && deleteMaFile) + { + try + { + File.Delete(filename); + return true; + } + catch (Exception) + { + return false; + } + } + + return false; + } + + public bool SaveAccount(SteamGuardAccount account, bool encrypt, string passKey = null, string salt = null, string iV = null) + { + if (encrypt && (String.IsNullOrEmpty(passKey) || String.IsNullOrEmpty(salt) || String.IsNullOrEmpty(iV))) return false; + + string jsonAccount = JsonConvert.SerializeObject(account); + + string filename = account.Session.SteamID.ToString() + ".maFile"; + if (Program.Verbose) Console.WriteLine($"Saving account {account.AccountName} to {filename}..."); + + ManifestEntry newEntry = new ManifestEntry() + { + SteamID = account.Session.SteamID, + IV = iV, + Salt = salt, + Filename = filename + }; + + bool foundExistingEntry = false; + for (int i = 0; i < this.Entries.Count; i++) + { + if (this.Entries[i].SteamID == account.Session.SteamID) + { + this.Entries[i] = newEntry; + foundExistingEntry = true; + break; + } + } + + if (!foundExistingEntry) + { + this.Entries.Add(newEntry); + } + + bool wasEncrypted = this.Encrypted; + this.Encrypted = encrypt; + + if (!this.Save()) + { + this.Encrypted = wasEncrypted; + return false; + } + + try + { + Stream stream = null; + MemoryStream ms = null; + RijndaelManaged aes256; + + if (encrypt) + { + ms = new MemoryStream(); + byte[] key = GetEncryptionKey(passKey, newEntry.Salt); + + aes256 = new RijndaelManaged + { + IV = Convert.FromBase64String(newEntry.IV), + Key = key, + Padding = PaddingMode.PKCS7, + Mode = CipherMode.CBC + }; + + ICryptoTransform encryptor = aes256.CreateEncryptor(aes256.Key, aes256.IV); + stream = new CryptoStream(ms, encryptor, CryptoStreamMode.Write); + } + else + { + // An unencrypted maFile is shorter than the encrypted version, + // so when an unencrypted maFile gets written this way, the file does not get wiped + // leaving encrypted text after the final } bracket. Deleting and recreating the file fixes this. + File.Delete(Path.Combine(Program.SteamGuardPath, newEntry.Filename)); + stream = File.OpenWrite(Path.Combine(Program.SteamGuardPath, newEntry.Filename)); // open or create + } + + using (StreamWriter writer = new StreamWriter(stream)) + { + writer.Write(jsonAccount); + } + + if (encrypt) + { + File.WriteAllText(Path.Combine(Program.SteamGuardPath, newEntry.Filename), Convert.ToBase64String(ms.ToArray())); + } + + stream.Close(); + return true; + } + catch (Exception ex) + { + if (Program.Verbose) Console.WriteLine("error: {0}", ex.ToString()); + return false; + } + } + + public bool Save() + { + string filename = Path.Combine(Program.SteamGuardPath, "manifest.json"); + if (!Directory.Exists(Program.SteamGuardPath)) + { + try + { + if (Program.Verbose) Console.WriteLine("Creating {0}", Program.SteamGuardPath); + Directory.CreateDirectory(Program.SteamGuardPath); + } + catch (Exception ex) + { + if (Program.Verbose) Console.WriteLine($"error: {ex.Message}"); + return false; + } + } + + try + { + string contents = JsonConvert.SerializeObject(this); + File.WriteAllText(filename, contents); + return true; + } + catch (Exception ex) + { if (Program.Verbose) Console.WriteLine($"error: {ex.Message}"); - return false; - } - } + return false; + } + } - try - { - string contents = JsonConvert.SerializeObject(this); - File.WriteAllText(filename, contents); - return true; - } - catch (Exception ex) - { - if (Program.Verbose) Console.WriteLine($"error: {ex.Message}"); - return false; - } - } + private void RecomputeExistingEntries() + { + List newEntries = new List(); - private void RecomputeExistingEntries() - { - List newEntries = new List(); + foreach (var entry in this.Entries) + { + string filename = Path.Combine(Program.SteamGuardPath, entry.Filename); - foreach (var entry in this.Entries) - { - string filename = Path.Combine(Program.SteamGuardPath, entry.Filename); + if (File.Exists(filename)) + { + newEntries.Add(entry); + } + } - if (File.Exists(filename)) - { - newEntries.Add(entry); - } - } + this.Entries = newEntries; - this.Entries = newEntries; + if (this.Entries.Count == 0) + { + this.Encrypted = false; + } + } - if (this.Entries.Count == 0) - { - this.Encrypted = false; - } - } + public void MoveEntry(int from, int to) + { + if (from < 0 || to < 0 || from > Entries.Count || to > Entries.Count - 1) return; + ManifestEntry sel = Entries[from]; + Entries.RemoveAt(from); + Entries.Insert(to, sel); + Save(); + } - public void MoveEntry(int from, int to) - { - if (from < 0 || to < 0 || from > Entries.Count || to > Entries.Count - 1) return; - ManifestEntry sel = Entries[from]; - Entries.RemoveAt(from); - Entries.Insert(to, sel); - Save(); - } + public class ManifestEntry + { + [JsonProperty("encryption_iv")] + public string IV { get; set; } - public class ManifestEntry - { - [JsonProperty("encryption_iv")] - public string IV { get; set; } + [JsonProperty("encryption_salt")] + public string Salt { get; set; } - [JsonProperty("encryption_salt")] - public string Salt { get; set; } + [JsonProperty("filename")] + public string Filename { get; set; } - [JsonProperty("filename")] - public string Filename { get; set; } + [JsonProperty("steamid")] + public ulong SteamID { get; set; } + } - [JsonProperty("steamid")] - public ulong SteamID { get; set; } - } + /* + Crypto Functions + */ - /* - Crypto Functions - */ + /// + /// Returns an 8-byte cryptographically random salt in base64 encoding + /// + /// + public static string GetRandomSalt() + { + byte[] salt = new byte[SALT_LENGTH]; + using (RNGCryptoServiceProvider rng = new RNGCryptoServiceProvider()) + { + rng.GetBytes(salt); + } + return Convert.ToBase64String(salt); + } - /// - /// Returns an 8-byte cryptographically random salt in base64 encoding - /// - /// - public static string GetRandomSalt() - { - byte[] salt = new byte[SALT_LENGTH]; - using (RNGCryptoServiceProvider rng = new RNGCryptoServiceProvider()) - { - rng.GetBytes(salt); - } - return Convert.ToBase64String(salt); - } + /// + /// Returns a 16-byte cryptographically random initialization vector (IV) in base64 encoding + /// + /// + public static string GetInitializationVector() + { + byte[] IV = new byte[IV_LENGTH]; + using (RNGCryptoServiceProvider rng = new RNGCryptoServiceProvider()) + { + rng.GetBytes(IV); + } + return Convert.ToBase64String(IV); + } - /// - /// Returns a 16-byte cryptographically random initialization vector (IV) in base64 encoding - /// - /// - public static string GetInitializationVector() - { - byte[] IV = new byte[IV_LENGTH]; - using (RNGCryptoServiceProvider rng = new RNGCryptoServiceProvider()) - { - rng.GetBytes(IV); - } - return Convert.ToBase64String(IV); - } - - - /// - /// Generates an encryption key derived using a password, a random salt, and specified number of rounds of PBKDF2 - /// - /// - /// - /// - private static byte[] GetEncryptionKey(string password, string salt) - { - if (string.IsNullOrEmpty(password)) - { - throw new ArgumentException("Password is empty"); - } - if (string.IsNullOrEmpty(salt)) - { - throw new ArgumentException("Salt is empty"); - } - using (Rfc2898DeriveBytes pbkdf2 = new Rfc2898DeriveBytes(password, Convert.FromBase64String(salt), PBKDF2_ITERATIONS)) - { - return pbkdf2.GetBytes(KEY_SIZE_BYTES); - } - } + /// + /// Generates an encryption key derived using a password, a random salt, and specified number of rounds of PBKDF2 + /// + /// + /// + /// + private static byte[] GetEncryptionKey(string password, string salt) + { + if (string.IsNullOrEmpty(password)) + { + throw new ArgumentException("Password is empty"); + } + if (string.IsNullOrEmpty(salt)) + { + throw new ArgumentException("Salt is empty"); + } + using (Rfc2898DeriveBytes pbkdf2 = new Rfc2898DeriveBytes(password, Convert.FromBase64String(salt), PBKDF2_ITERATIONS)) + { + return pbkdf2.GetBytes(KEY_SIZE_BYTES); + } + } + } } diff --git a/Program.cs b/Program.cs index 828a214..a0f7fdd 100644 --- a/Program.cs +++ b/Program.cs @@ -9,611 +9,625 @@ using System.Runtime.InteropServices; using System.Text; using System.Threading.Tasks; -public static class Program +namespace SteamGuard { - public const string defaultSteamGuardPath = "~/maFiles"; - - public static string SteamGuardPath { get; set; } = defaultSteamGuardPath; - public static Manifest Manifest { get; set; } - public static SteamGuardAccount[] SteamGuardAccounts { get; set; } - public static bool Verbose { get; set; } = false; - - /// - /// The main entry point for the application - /// - [STAThread] - public static void Main(string[] args) - { - string action = ""; - string user = ""; - - // Parse cli arguments - for (int i = 0; i < args.Length; i++) - { - if (args[i].StartsWith("-")) - { - if (args[i] == "-v" || args[i] == "--verbose") - { - Verbose = true; - } - else if (args[i] == "-m" || args[i] == "--mafiles-path") - { - i++; - if (i < args.Length) - SteamGuardPath = args[i]; - else - { - Console.WriteLine($"Expected path after {args[i-1]}"); - return; - } - } - else if (args[i] == "--help" || args[i] == "-h") - { - ShowHelp(); - return; - } - } - else // Parse as action or username - { - if (string.IsNullOrEmpty(action)) - { - if (args[i] == "add" || args[i] == "setup") - { - action = "setup"; - } - else if (args[i] == "trade") - { - action = "trade"; - } - else if (args[i] == "encrypt") - { - action = "encrypt"; - } - else if (args[i] == "decrypt") - { - action = "decrypt"; - } - else if (args[i] == "remove") - { - action = "remove"; - } - else if (args[i] == "2fa" || args[i] == "code" || args[i] == "generate-code") - { - action = "generate-code"; - } - else if (string.IsNullOrEmpty(user)) - user = args[i]; - } - else if (string.IsNullOrEmpty(user)) - user = args[i]; - } - } - - if (string.IsNullOrEmpty(action)) - action = "generate-code"; - - // Do some configuring - SteamGuardPath = SteamGuardPath.Replace("~", Environment.GetEnvironmentVariable("HOME")); - if (!Directory.Exists(SteamGuardPath)) - { - if (SteamGuardPath == defaultSteamGuardPath.Replace("~", Environment.GetEnvironmentVariable("HOME"))) - { - if (Verbose) Console.WriteLine("warn: {0} does not exist, creating...", SteamGuardPath); - Directory.CreateDirectory(SteamGuardPath); - } - else - { - Console.WriteLine("error: {0} does not exist.", SteamGuardPath); - return; - } - } - - if (Verbose) Console.WriteLine($"Action: {action}"); - if (Verbose) Console.WriteLine($"User: {user}"); - if (Verbose) Console.WriteLine($"maFiles path: {SteamGuardPath}"); - - // Perform desired action - switch (action) - { - case "generate-code": - GenerateCode(user); - break; - case "encrypt": // Can also be used to change passkey - Console.WriteLine(Encrypt()); - break; - case "decrypt": - Console.WriteLine(Decrypt()); - break; - case "setup": - Setup(user); - break; - case "trade": - Trade(user); - break; - case "accept-all": - AcceptAllTrades(user); - break; - default: - Console.WriteLine("error: Unknown action: {0}", action); - return; - } - } - - static void ShowHelp() + public static class Program { - var descPadding = 26; - var descWidth = Console.BufferWidth - descPadding; - if (descWidth < 20) - descWidth = 20; - else if (descWidth > 56) - descWidth = 56; + public const string defaultSteamGuardPath = "~/maFiles"; - var flags = new Dictionary - { - { "-h, --help", "Display this help message." }, - { "-v, --verbose", "Display some extra information when the program is running." }, - { "-m, --mafiles", "Specify which folder your maFiles are in. Ex: ~/maFiles" }, - }; - var actions = new Dictionary - { - { "generate-code", "Generate a Steam Guard code for the specified user (if any) and exit. (default)" }, - { "encrypt", "Encrypt your maFiles or change your encryption passkey." }, - { "decrypt", "Remove encryption from your maFiles." }, - { "code", "Same as generate-code" }, - { "2fa", "Same as generate-code" }, - { "add", "Set up Steam Guard for 2 factor authentication." }, - { "setup", "Same as add" } - }; + public static string SteamGuardPath { get; set; } = defaultSteamGuardPath; + public static Manifest Manifest { get; set; } + public static SteamGuardAccount[] SteamGuardAccounts { get; set; } + public static bool Verbose { get; set; } = false; - Console.WriteLine($"steamguard-cli - v{Assembly.GetExecutingAssembly().GetName().Version}"); - Console.WriteLine("usage: steamguard (action) (steam username) -v -h"); - Console.WriteLine(); - foreach (var flag in flags) + /// + /// The main entry point for the application + /// + [STAThread] + public static void Main(string[] args) { - // word wrap the descriptions, if needed - var desc = flag.Value; - if (desc.Length > descWidth) + string action = ""; + string user = ""; + + // Parse cli arguments + for (int i = 0; i < args.Length; i++) { - var sb = new StringBuilder(); - for (int i = 0; i < desc.Length; i += descWidth) + if (args[i].StartsWith("-")) { - if (i > 0) - sb.Append("".PadLeft((flag.Key.StartsWith("--") ? 5 : 2) + descPadding)); - sb.AppendLine(desc.Substring(i, i + descWidth > desc.Length ? desc.Length - i : descWidth).Trim()); - } - desc = sb.ToString().TrimEnd('\n'); - } - Console.WriteLine($"{(flag.Key.StartsWith("--") ? " " : " " )}{flag.Key.PadRight(descPadding)}{desc}"); - } - Console.WriteLine(); - Console.WriteLine("Actions:"); - foreach (var action in actions) - { - // word wrap the descriptions, if needed - var desc = action.Value; - if (desc.Length > descWidth) - { - var sb = new StringBuilder(); - for (int i = 0; i < desc.Length; i += descWidth) - { - if (i > 0) - sb.Append("".PadLeft(descPadding + 2)); - sb.AppendLine(desc.Substring(i, i + descWidth > desc.Length ? desc.Length - i : descWidth).Trim()); - } - desc = sb.ToString().TrimEnd('\n'); - } - Console.WriteLine($" {action.Key.PadRight(descPadding)}{desc}"); - } - } - - static void GenerateCode(string user = "") - { - if (Verbose) Console.WriteLine("Aligning time..."); - TimeAligner.AlignTime(); - if (Verbose) Console.WriteLine("Opening manifest..."); - Manifest = Manifest.GetManifest(true); - if (Verbose) Console.WriteLine("Reading accounts from manifest..."); - if (Manifest.Encrypted) - { - string passkey = Manifest.PromptForPassKey(); - SteamGuardAccounts = Manifest.GetAllAccounts(passkey); - } - else - { - SteamGuardAccounts = Manifest.GetAllAccounts(); - } - if (SteamGuardAccounts.Length == 0) - { - Console.WriteLine("error: No accounts read."); - return; - } - if (Verbose) Console.WriteLine("Selecting account..."); - string code = ""; - for (int i = 0; i < SteamGuardAccounts.Length; i++) - { - SteamGuardAccount account = SteamGuardAccounts[i]; - if (user != "") - { - if (account.AccountName.ToLower() == user.ToLower()) - { - if (Verbose) Console.WriteLine("Generating Code..."); - code = account.GenerateSteamGuardCode(); - break; - } - } - else - { - if (Verbose) Console.WriteLine("Generating Code for {0}...", account.AccountName); - code = account.GenerateSteamGuardCode(); - break; - } - } - if (code != "") - Console.WriteLine(code); - else - Console.WriteLine("error: No Steam accounts found in {0}", SteamGuardAccounts); - } - - static bool Encrypt() - { - if (Verbose) Console.WriteLine("Opening manifest..."); - Manifest = Manifest.GetManifest(true); - if (Verbose) Console.WriteLine("Reading accounts from manifest..."); - if (Manifest.Encrypted) - { - string passkey = Manifest.PromptForPassKey(); - SteamGuardAccounts = Manifest.GetAllAccounts(passkey); - } - else - { - SteamGuardAccounts = Manifest.GetAllAccounts(); - } - - string newPassKey = Manifest.PromptSetupPassKey(); - - for (int i = 0; i < SteamGuardAccounts.Length; i++) - { - var account = SteamGuardAccounts[i]; - var salt = Manifest.GetRandomSalt(); - var iv = Manifest.GetInitializationVector(); - bool success = Manifest.SaveAccount(account, true, newPassKey, salt, iv); - if (Verbose) Console.WriteLine("Encrypted {0}: {1}", account.AccountName, success); - if (!success) return false; - } - return true; - } - - static bool Decrypt() - { - if (Verbose) Console.WriteLine("Opening manifest..."); - Manifest = Manifest.GetManifest(true); - if (Verbose) Console.WriteLine("Reading accounts from manifest..."); - if (Manifest.Encrypted) - { - string passkey = Manifest.PromptForPassKey(); - SteamGuardAccounts = Manifest.GetAllAccounts(passkey); - } - else - { - if (Verbose) Console.WriteLine("Decryption not required."); - return true; - } - - for (int i = 0; i < SteamGuardAccounts.Length; i++) - { - var account = SteamGuardAccounts[i]; - bool success = Manifest.SaveAccount(account, false); - if (Verbose) Console.WriteLine("Decrypted {0}: {1}", account.AccountName, success); - if (!success) return false; - } - return true; - } - - static void Setup(string username = "") - { - if (Verbose) Console.WriteLine("Opening manifest..."); - Manifest = Manifest.GetManifest(true); - - if (string.IsNullOrWhiteSpace(username)) - { - Console.Write("Username: "); - username = Console.ReadLine(); - } - Console.Write("Password: "); - var password = Console.ReadLine(); - - UserLogin login = new UserLogin(username, password); - Console.Write($"Logging in {username}... "); - LoginResult loginResult = login.DoLogin(); - Console.WriteLine(loginResult); - if (!login.LoggedIn) return; - - AuthenticatorLinker linker = new AuthenticatorLinker(login.Session); - AuthenticatorLinker.LinkResult linkResult = AuthenticatorLinker.LinkResult.GeneralFailure; - - do - { - linkResult = linker.AddAuthenticator(); - Console.WriteLine($"Link result: {linkResult}"); - switch (linkResult) - { - case AuthenticatorLinker.LinkResult.MustProvidePhoneNumber: - var phonenumber = ""; - do + if (args[i] == "-v" || args[i] == "--verbose") { - Console.WriteLine("Enter your mobile phone number in the following format: +{cC} phoneNumber. EG, +1 123-456-7890"); - phonenumber = Console.ReadLine(); - phonenumber = FilterPhoneNumber(phonenumber); - linker.PhoneNumber = phonenumber; - } while (!PhoneNumberOkay(phonenumber)); - break; - case AuthenticatorLinker.LinkResult.MustRemovePhoneNumber: - linker.PhoneNumber = null; - break; - case AuthenticatorLinker.LinkResult.AwaitingFinalization: - break; - case AuthenticatorLinker.LinkResult.GeneralFailure: - Console.WriteLine("error: Unable to add your phone number. Steam returned GeneralFailure"); - return; - case AuthenticatorLinker.LinkResult.AuthenticatorPresent: - Console.WriteLine("error: Can't link authenticator, remove the previous authenticator."); - return; - default: - Console.WriteLine($"error: Unexpected linker result: {linkResult}"); - return; + Verbose = true; + } + else if (args[i] == "-m" || args[i] == "--mafiles-path") + { + i++; + if (i < args.Length) + SteamGuardPath = args[i]; + else + { + Console.WriteLine($"Expected path after {args[i-1]}"); + return; + } + } + else if (args[i] == "--help" || args[i] == "-h") + { + ShowHelp(); + return; + } + } + else // Parse as action or username + { + if (string.IsNullOrEmpty(action)) + { + if (args[i] == "add" || args[i] == "setup") + { + action = "setup"; + } + else if (args[i] == "trade") + { + action = "trade"; + } + else if (args[i] == "encrypt") + { + action = "encrypt"; + } + else if (args[i] == "decrypt") + { + action = "decrypt"; + } + else if (args[i] == "remove") + { + action = "remove"; + } + else if (args[i] == "2fa" || args[i] == "code" || args[i] == "generate-code") + { + action = "generate-code"; + } + else if (string.IsNullOrEmpty(user)) + user = args[i]; + } + else if (string.IsNullOrEmpty(user)) + user = args[i]; + } } - } while (linkResult != AuthenticatorLinker.LinkResult.AwaitingFinalization); - string passKey = null; - if (Manifest.Entries.Count == 0) - { - Console.WriteLine("Looks like we are setting up your first account."); - passKey = Manifest.PromptSetupPassKey(); - } - else if (Manifest.Entries.Count > 0 && Manifest.Encrypted) - { - passKey = Manifest.PromptForPassKey(); - } + if (string.IsNullOrEmpty(action)) + action = "generate-code"; - //Save the file immediately; losing this would be bad. - if (!Manifest.SaveAccount(linker.LinkedAccount, passKey != null, passKey)) - { - Manifest.RemoveAccount(linker.LinkedAccount); - Console.WriteLine("Unable to save mobile authenticator file. The mobile authenticator has not been linked."); - return; - } - - Console.WriteLine( - $"The Mobile Authenticator has not yet been linked. Before finalizing the authenticator, please write down your revocation code: {linker.LinkedAccount.RevocationCode}"); - - AuthenticatorLinker.FinalizeResult finalizeResponse = AuthenticatorLinker.FinalizeResult.GeneralFailure; - do - { - Console.Write("Please input the SMS message sent to your phone number: "); - string smsCode = Console.ReadLine(); - - finalizeResponse = linker.FinalizeAddAuthenticator(smsCode); - if (Verbose) Console.WriteLine(finalizeResponse); - - switch (finalizeResponse) + // Do some configuring + SteamGuardPath = SteamGuardPath.Replace("~", Environment.GetEnvironmentVariable("HOME")); + if (!Directory.Exists(SteamGuardPath)) { - case AuthenticatorLinker.FinalizeResult.BadSMSCode: - continue; - - case AuthenticatorLinker.FinalizeResult.UnableToGenerateCorrectCodes: - Console.WriteLine( - "Unable to generate the proper codes to finalize this authenticator. The authenticator should not have been linked."); - Console.WriteLine( - $"In the off-chance it was, please write down your revocation code, as this is the last chance to see it: {linker.LinkedAccount.RevocationCode}"); - Manifest.RemoveAccount(linker.LinkedAccount); + if (SteamGuardPath == defaultSteamGuardPath.Replace("~", Environment.GetEnvironmentVariable("HOME"))) + { + if (Verbose) Console.WriteLine("warn: {0} does not exist, creating...", SteamGuardPath); + Directory.CreateDirectory(SteamGuardPath); + } + else + { + Console.WriteLine("error: {0} does not exist.", SteamGuardPath); return; + } + } - case AuthenticatorLinker.FinalizeResult.GeneralFailure: - Console.WriteLine("Unable to finalize this authenticator. The authenticator should not have been linked."); - Console.WriteLine( - $"In the off-chance it was, please write down your revocation code, as this is the last chance to see it: {linker.LinkedAccount.RevocationCode}"); - Manifest.RemoveAccount(linker.LinkedAccount); + if (Verbose) Console.WriteLine($"Action: {action}"); + if (Verbose) Console.WriteLine($"User: {user}"); + if (Verbose) Console.WriteLine($"maFiles path: {SteamGuardPath}"); + + // Perform desired action + switch (action) + { + case "generate-code": + GenerateCode(user); + break; + case "encrypt": // Can also be used to change passkey + Console.WriteLine(Encrypt()); + break; + case "decrypt": + Console.WriteLine(Decrypt()); + break; + case "setup": + Setup(user); + break; + case "trade": + Trade(user); + break; + case "accept-all": + AcceptAllTrades(user); + break; + default: + Console.WriteLine("error: Unknown action: {0}", action); return; } - } while (finalizeResponse != AuthenticatorLinker.FinalizeResult.Success); - - //Linked, finally. Re-save with FullyEnrolled property. - Manifest.SaveAccount(linker.LinkedAccount, passKey != null, passKey); - Console.WriteLine( - $"Mobile authenticator successfully linked. Please actually write down your revocation code: {linker.LinkedAccount.RevocationCode}"); - } - - public static string FilterPhoneNumber(string phoneNumber) - => phoneNumber.Replace("-", "").Replace("(", "").Replace(")", ""); - - public static bool PhoneNumberOkay(string phoneNumber) - { - if (phoneNumber == null || phoneNumber.Length == 0) return false; - if (phoneNumber[0] != '+') return false; - return true; - } - - static void Trade(string user = "") - { - if (Verbose) Console.WriteLine("Opening manifest..."); - Manifest = Manifest.GetManifest(true); - if (Verbose) Console.WriteLine("Reading accounts from manifest..."); - if (Manifest.Encrypted) - { - var passkey = Manifest.PromptForPassKey(); - SteamGuardAccounts = Manifest.GetAllAccounts(passkey); - } - else - { - SteamGuardAccounts = Manifest.GetAllAccounts(); - } - if (SteamGuardAccounts.Length == 0) - { - Console.WriteLine("error: No accounts read."); - return; } - foreach (var account in SteamGuardAccounts) + static void ShowHelp() { - if (user != "") - if (!string.Equals(account.AccountName, user, StringComparison.CurrentCultureIgnoreCase)) - break; + var descPadding = 26; + var descWidth = Console.BufferWidth - descPadding; + if (descWidth < 20) + descWidth = 20; + else if (descWidth > 56) + descWidth = 56; - processConfirmations(account); - } - } + var flags = new Dictionary + { + { "-h, --help", "Display this help message." }, + { "-v, --verbose", "Display some extra information when the program is running." }, + { "-m, --mafiles", "Specify which folder your maFiles are in. Ex: ~/maFiles" }, + }; + var actions = new Dictionary + { + { "generate-code", "Generate a Steam Guard code for the specified user (if any) and exit. (default)" }, + { "encrypt", "Encrypt your maFiles or change your encryption passkey." }, + { "decrypt", "Remove encryption from your maFiles." }, + { "code", "Same as generate-code" }, + { "2fa", "Same as generate-code" }, + { "add", "Set up Steam Guard for 2 factor authentication." }, + { "setup", "Same as add" } + }; - enum TradeAction - { - Accept = 1, - Deny = 0, - Ignore = -1 - } - - static void processConfirmations(SteamGuardAccount account) - { - if (Verbose) Console.WriteLine("Refeshing Session..."); - account.RefreshSession(); - Console.WriteLine("Retrieving trade confirmations..."); - var trades = account.FetchConfirmations(); - var tradeActions = new TradeAction[trades.Length]; - for (var i = 0; i < tradeActions.Length; i++) - { - tradeActions[i] = TradeAction.Ignore; - } - if (trades.Length == 0) - { - Console.WriteLine($"No trade confirmations for {account.AccountName}."); - return; - } - var selected = 0; - var colorAccept = ConsoleColor.Green; - var colorDeny = ConsoleColor.Red; - var colorIgnore = ConsoleColor.Gray; - var colorSelected = ConsoleColor.Yellow; - var confirm = false; - - do - { - Console.Clear(); - if (selected >= trades.Length) - selected = trades.Length - 1; - else if (selected < 0) - selected = 0; - Console.ResetColor(); - Console.WriteLine($"Trade confirmations for {account.AccountName}..."); - Console.WriteLine("No action will be made without your confirmation."); - Console.WriteLine("[a]ccept [d]eny [i]gnore [enter] Confirm [q]uit"); // accept = 1, deny = 0, ignore = -1 + Console.WriteLine($"steamguard-cli - v{Assembly.GetExecutingAssembly().GetName().Version}"); + Console.WriteLine("usage: steamguard (action) (steam username) -v -h"); Console.WriteLine(); + foreach (var flag in flags) + { + // word wrap the descriptions, if needed + var desc = flag.Value; + if (desc.Length > descWidth) + { + var sb = new StringBuilder(); + for (int i = 0; i < desc.Length; i += descWidth) + { + if (i > 0) + sb.Append("".PadLeft((flag.Key.StartsWith("--") ? 5 : 2) + descPadding)); + sb.AppendLine(desc.Substring(i, i + descWidth > desc.Length ? desc.Length - i : descWidth).Trim()); + } + desc = sb.ToString().TrimEnd('\n'); + } + Console.WriteLine($"{(flag.Key.StartsWith("--") ? " " : " " )}{flag.Key.PadRight(descPadding)}{desc}"); + } + Console.WriteLine(); + Console.WriteLine("Actions:"); + foreach (var action in actions) + { + // word wrap the descriptions, if needed + var desc = action.Value; + if (desc.Length > descWidth) + { + var sb = new StringBuilder(); + for (int i = 0; i < desc.Length; i += descWidth) + { + if (i > 0) + sb.Append("".PadLeft(descPadding + 2)); + sb.AppendLine(desc.Substring(i, i + descWidth > desc.Length ? desc.Length - i : descWidth).Trim()); + } + desc = sb.ToString().TrimEnd('\n'); + } + Console.WriteLine($" {action.Key.PadRight(descPadding)}{desc}"); + } + } + static void GenerateCode(string user = "") + { + if (Verbose) Console.WriteLine("Aligning time..."); + TimeAligner.AlignTime(); + if (Verbose) Console.WriteLine("Opening manifest..."); + Manifest = Manifest.GetManifest(true); + if (Verbose) Console.WriteLine("Reading accounts from manifest..."); + if (Manifest.Encrypted) + { + string passkey = Manifest.PromptForPassKey(); + SteamGuardAccounts = Manifest.GetAllAccounts(passkey); + } + else + { + SteamGuardAccounts = Manifest.GetAllAccounts(); + } + if (SteamGuardAccounts.Length == 0) + { + Console.WriteLine("error: No accounts read."); + return; + } + if (Verbose) Console.WriteLine("Selecting account..."); + string code = ""; + for (int i = 0; i < SteamGuardAccounts.Length; i++) + { + SteamGuardAccount account = SteamGuardAccounts[i]; + if (user != "") + { + if (account.AccountName.ToLower() == user.ToLower()) + { + if (Verbose) Console.WriteLine("Generating Code..."); + code = account.GenerateSteamGuardCode(); + break; + } + } + else + { + if (Verbose) Console.WriteLine("Generating Code for {0}...", account.AccountName); + code = account.GenerateSteamGuardCode(); + break; + } + } + if (code != "") + Console.WriteLine(code); + else + Console.WriteLine("error: No Steam accounts found in {0}", SteamGuardAccounts); + } + + static bool Encrypt() + { + if (Verbose) Console.WriteLine("Opening manifest..."); + Manifest = Manifest.GetManifest(true); + if (Verbose) Console.WriteLine("Reading accounts from manifest..."); + if (Manifest.Encrypted) + { + string passkey = Manifest.PromptForPassKey(); + SteamGuardAccounts = Manifest.GetAllAccounts(passkey); + } + else + { + SteamGuardAccounts = Manifest.GetAllAccounts(); + } + + string newPassKey = Manifest.PromptSetupPassKey(); + + for (int i = 0; i < SteamGuardAccounts.Length; i++) + { + var account = SteamGuardAccounts[i]; + var salt = Manifest.GetRandomSalt(); + var iv = Manifest.GetInitializationVector(); + bool success = Manifest.SaveAccount(account, true, newPassKey, salt, iv); + if (Verbose) Console.WriteLine("Encrypted {0}: {1}", account.AccountName, success); + if (!success) return false; + } + return true; + } + + static bool Decrypt() + { + if (Verbose) Console.WriteLine("Opening manifest..."); + Manifest = Manifest.GetManifest(true); + if (Verbose) Console.WriteLine("Reading accounts from manifest..."); + if (Manifest.Encrypted) + { + string passkey = Manifest.PromptForPassKey(); + SteamGuardAccounts = Manifest.GetAllAccounts(passkey); + } + else + { + if (Verbose) Console.WriteLine("Decryption not required."); + return true; + } + + for (int i = 0; i < SteamGuardAccounts.Length; i++) + { + var account = SteamGuardAccounts[i]; + bool success = Manifest.SaveAccount(account, false); + if (Verbose) Console.WriteLine("Decrypted {0}: {1}", account.AccountName, success); + if (!success) return false; + } + return true; + } + + static void Setup(string username = "") + { + if (Verbose) Console.WriteLine("Opening manifest..."); + Manifest = Manifest.GetManifest(true); + + if (string.IsNullOrWhiteSpace(username)) + { + Console.Write("Username: "); + username = Console.ReadLine(); + } + Console.Write("Password: "); + var password = Console.ReadLine(); + + UserLogin login = new UserLogin(username, password); + Console.Write($"Logging in {username}... "); + LoginResult loginResult = login.DoLogin(); + Console.WriteLine(loginResult); + if (!login.LoggedIn) return; + + AuthenticatorLinker linker = new AuthenticatorLinker(login.Session); + AuthenticatorLinker.LinkResult linkResult = AuthenticatorLinker.LinkResult.GeneralFailure; + + do + { + linkResult = linker.AddAuthenticator(); + Console.WriteLine($"Link result: {linkResult}"); + switch (linkResult) + { + case AuthenticatorLinker.LinkResult.MustProvidePhoneNumber: + var phonenumber = ""; + do + { + Console.WriteLine("Enter your mobile phone number in the following format: +{cC} phoneNumber. EG, +1 123-456-7890"); + phonenumber = Console.ReadLine(); + phonenumber = FilterPhoneNumber(phonenumber); + linker.PhoneNumber = phonenumber; + } while (!PhoneNumberOkay(phonenumber)); + break; + case AuthenticatorLinker.LinkResult.MustRemovePhoneNumber: + linker.PhoneNumber = null; + break; + case AuthenticatorLinker.LinkResult.AwaitingFinalization: + break; + case AuthenticatorLinker.LinkResult.GeneralFailure: + Console.WriteLine("error: Unable to add your phone number. Steam returned GeneralFailure"); + return; + case AuthenticatorLinker.LinkResult.AuthenticatorPresent: + Console.WriteLine("error: Can't link authenticator, remove the previous authenticator."); + return; + default: + Console.WriteLine($"error: Unexpected linker result: {linkResult}"); + return; + } + } while (linkResult != AuthenticatorLinker.LinkResult.AwaitingFinalization); + + string passKey = null; + if (Manifest.Entries.Count == 0) + { + Console.WriteLine("Looks like we are setting up your first account."); + passKey = Manifest.PromptSetupPassKey(); + } + else if (Manifest.Entries.Count > 0 && Manifest.Encrypted) + { + passKey = Manifest.PromptForPassKey(); + } + + //Save the file immediately; losing this would be bad. + if (!Manifest.SaveAccount(linker.LinkedAccount, passKey != null, passKey)) + { + Manifest.RemoveAccount(linker.LinkedAccount); + Console.WriteLine("Unable to save mobile authenticator file. The mobile authenticator has not been linked."); + return; + } + + Console.WriteLine( + $"The Mobile Authenticator has not yet been linked. Before finalizing the authenticator, please write down your revocation code: {linker.LinkedAccount.RevocationCode}"); + + AuthenticatorLinker.FinalizeResult finalizeResponse = AuthenticatorLinker.FinalizeResult.GeneralFailure; + do + { + Console.Write("Please input the SMS message sent to your phone number: "); + string smsCode = Console.ReadLine(); + + finalizeResponse = linker.FinalizeAddAuthenticator(smsCode); + if (Verbose) Console.WriteLine(finalizeResponse); + + switch (finalizeResponse) + { + case AuthenticatorLinker.FinalizeResult.BadSMSCode: + continue; + + case AuthenticatorLinker.FinalizeResult.UnableToGenerateCorrectCodes: + Console.WriteLine( + "Unable to generate the proper codes to finalize this authenticator. The authenticator should not have been linked."); + Console.WriteLine( + $"In the off-chance it was, please write down your revocation code, as this is the last chance to see it: {linker.LinkedAccount.RevocationCode}"); + Manifest.RemoveAccount(linker.LinkedAccount); + return; + + case AuthenticatorLinker.FinalizeResult.GeneralFailure: + Console.WriteLine("Unable to finalize this authenticator. The authenticator should not have been linked."); + Console.WriteLine( + $"In the off-chance it was, please write down your revocation code, as this is the last chance to see it: {linker.LinkedAccount.RevocationCode}"); + Manifest.RemoveAccount(linker.LinkedAccount); + return; + } + } while (finalizeResponse != AuthenticatorLinker.FinalizeResult.Success); + + //Linked, finally. Re-save with FullyEnrolled property. + Manifest.SaveAccount(linker.LinkedAccount, passKey != null, passKey); + Console.WriteLine( + $"Mobile authenticator successfully linked. Please actually write down your revocation code: {linker.LinkedAccount.RevocationCode}"); + } + + public static string FilterPhoneNumber(string phoneNumber) + => phoneNumber.Replace("-", "").Replace("(", "").Replace(")", ""); + + public static bool PhoneNumberOkay(string phoneNumber) + { + if (phoneNumber == null || phoneNumber.Length == 0) return false; + if (phoneNumber[0] != '+') return false; + return true; + } + + static void Trade(string user = "") + { + if (Verbose) Console.WriteLine("Opening manifest..."); + Manifest = Manifest.GetManifest(true); + if (Verbose) Console.WriteLine("Reading accounts from manifest..."); + if (Manifest.Encrypted) + { + var passkey = Manifest.PromptForPassKey(); + SteamGuardAccounts = Manifest.GetAllAccounts(passkey); + } + else + { + SteamGuardAccounts = Manifest.GetAllAccounts(); + } + if (SteamGuardAccounts.Length == 0) + { + Console.WriteLine("error: No accounts read."); + return; + } + + foreach (var account in SteamGuardAccounts) + { + if (user != "") + if (!string.Equals(account.AccountName, user, StringComparison.CurrentCultureIgnoreCase)) + break; + + processConfirmations(account); + } + } + + enum TradeAction + { + Accept = 1, + Deny = 0, + Ignore = -1 + } + + static void processConfirmations(SteamGuardAccount account) + { + if (Verbose) Console.WriteLine("Refeshing Session..."); + account.RefreshSession(); + Console.WriteLine("Retrieving trade confirmations..."); + var trades = account.FetchConfirmations(); + var tradeActions = new TradeAction[trades.Length]; + for (var i = 0; i < tradeActions.Length; i++) + { + tradeActions[i] = TradeAction.Ignore; + } + if (trades.Length == 0) + { + Console.WriteLine($"No trade confirmations for {account.AccountName}."); + return; + } + var selected = 0; + var colorAccept = ConsoleColor.Green; + var colorDeny = ConsoleColor.Red; + var colorIgnore = ConsoleColor.Gray; + var colorSelected = ConsoleColor.Yellow; + var confirm = false; + + do + { + Console.Clear(); + if (selected >= trades.Length) + selected = trades.Length - 1; + else if (selected < 0) + selected = 0; + Console.ResetColor(); + Console.WriteLine($"Trade confirmations for {account.AccountName}..."); + Console.WriteLine("No action will be made without your confirmation."); + Console.WriteLine("[a]ccept [d]eny [i]gnore [enter] Confirm [q]uit"); // accept = 1, deny = 0, ignore = -1 + Console.WriteLine(); + + for (var t = 0; t < trades.Length; t++) + { + ConsoleColor itemColor; + switch (tradeActions[t]) + { + case TradeAction.Accept: + itemColor = colorAccept; + break; + case TradeAction.Deny: + itemColor = colorDeny; + break; + case TradeAction.Ignore: + itemColor = colorIgnore; + break; + default: + throw new ArgumentOutOfRangeException(); + } + + Console.ForegroundColor = t == selected ? colorSelected : itemColor; + + Console.WriteLine($" [{t}] [{tradeActions[t]}] {trades[t].Description}"); + } + var key = Console.ReadKey(); + switch (key.Key) + { + case ConsoleKey.UpArrow: + case ConsoleKey.W: + selected--; + break; + case ConsoleKey.DownArrow: + case ConsoleKey.S: + selected++; + break; + case ConsoleKey.A: + tradeActions[selected] = TradeAction.Accept; + break; + case ConsoleKey.D: + tradeActions[selected] = TradeAction.Deny; + break; + case ConsoleKey.I: + tradeActions[selected] = TradeAction.Ignore; + break; + case ConsoleKey.Enter: + confirm = true; + break; + case ConsoleKey.Escape: + case ConsoleKey.Q: + Console.ResetColor(); + Console.WriteLine("Quitting..."); + return; + default: + break; + } + } while (!confirm); + Console.ResetColor(); + Console.WriteLine(); + Console.WriteLine("Processing..."); for (var t = 0; t < trades.Length; t++) { - ConsoleColor itemColor; + bool success = false; switch (tradeActions[t]) { case TradeAction.Accept: - itemColor = colorAccept; + if (Verbose) Console.Write($"Accepting {trades[t].Description}..."); + success = account.AcceptConfirmation(trades[t]); break; case TradeAction.Deny: - itemColor = colorDeny; + if (Verbose) Console.Write($"Denying {trades[t].Description}..."); + success = account.AcceptConfirmation(trades[t]); break; case TradeAction.Ignore: - itemColor = colorIgnore; + if (Verbose) Console.Write($"Ignoring {trades[t].Description}..."); + success = true; break; default: throw new ArgumentOutOfRangeException(); } - - Console.ForegroundColor = t == selected ? colorSelected : itemColor; - - Console.WriteLine($" [{t}] [{tradeActions[t]}] {trades[t].Description}"); + if (Verbose) Console.WriteLine(success); } - var key = Console.ReadKey(); - switch (key.Key) - { - case ConsoleKey.UpArrow: - case ConsoleKey.W: - selected--; - break; - case ConsoleKey.DownArrow: - case ConsoleKey.S: - selected++; - break; - case ConsoleKey.A: - tradeActions[selected] = TradeAction.Accept; - break; - case ConsoleKey.D: - tradeActions[selected] = TradeAction.Deny; - break; - case ConsoleKey.I: - tradeActions[selected] = TradeAction.Ignore; - break; - case ConsoleKey.Enter: - confirm = true; - break; - case ConsoleKey.Escape: - case ConsoleKey.Q: - Console.ResetColor(); - Console.WriteLine("Quitting..."); - return; - default: - break; - } - } while (!confirm); - Console.ResetColor(); - Console.WriteLine(); - Console.WriteLine("Processing..."); - for (var t = 0; t < trades.Length; t++) - { - bool success = false; - switch (tradeActions[t]) - { - case TradeAction.Accept: - if (Verbose) Console.Write($"Accepting {trades[t].Description}..."); - success = account.AcceptConfirmation(trades[t]); - break; - case TradeAction.Deny: - if (Verbose) Console.Write($"Denying {trades[t].Description}..."); - success = account.AcceptConfirmation(trades[t]); - break; - case TradeAction.Ignore: - if (Verbose) Console.Write($"Ignoring {trades[t].Description}..."); - success = true; - break; - default: - throw new ArgumentOutOfRangeException(); - } - if (Verbose) Console.WriteLine(success); - } - Console.WriteLine("Done."); - } - - static void AcceptAllTrades(string user = "") - { - if (Verbose) Console.WriteLine("Opening manifest..."); - Manifest = Manifest.GetManifest(true); - if (Verbose) Console.WriteLine("Reading accounts from manifest..."); - if (Manifest.Encrypted) - { - string passkey = Manifest.PromptForPassKey(); - SteamGuardAccounts = Manifest.GetAllAccounts(passkey); - } - else - { - SteamGuardAccounts = Manifest.GetAllAccounts(); - } - if (SteamGuardAccounts.Length == 0) - { - Console.WriteLine("error: No accounts read."); - return; + Console.WriteLine("Done."); } - for (int i = 0; i < SteamGuardAccounts.Length; i++) + static void AcceptAllTrades(string user = "") { - SteamGuardAccount account = SteamGuardAccounts[i]; - if (user != "") + if (Verbose) Console.WriteLine("Opening manifest..."); + Manifest = Manifest.GetManifest(true); + if (Verbose) Console.WriteLine("Reading accounts from manifest..."); + if (Manifest.Encrypted) { - if (account.AccountName.ToLower() == user.ToLower()) + string passkey = Manifest.PromptForPassKey(); + SteamGuardAccounts = Manifest.GetAllAccounts(passkey); + } + else + { + SteamGuardAccounts = Manifest.GetAllAccounts(); + } + if (SteamGuardAccounts.Length == 0) + { + Console.WriteLine("error: No accounts read."); + return; + } + + for (int i = 0; i < SteamGuardAccounts.Length; i++) + { + SteamGuardAccount account = SteamGuardAccounts[i]; + if (user != "") + { + if (account.AccountName.ToLower() == user.ToLower()) + { + Console.WriteLine($"Accepting Confirmations on {account.AccountName}"); + if (Verbose) Console.WriteLine("Refeshing Session..."); + account.RefreshSession(); + if (Verbose) Console.WriteLine("Fetching Confirmations..."); + Confirmation[] confirmations = account.FetchConfirmations(); + if (Verbose) Console.WriteLine("Accepting Confirmations..."); + account.AcceptMultipleConfirmations(confirmations); + break; + } + } + else { Console.WriteLine($"Accepting Confirmations on {account.AccountName}"); if (Verbose) Console.WriteLine("Refeshing Session..."); @@ -622,19 +636,8 @@ public static class Program Confirmation[] confirmations = account.FetchConfirmations(); if (Verbose) Console.WriteLine("Accepting Confirmations..."); account.AcceptMultipleConfirmations(confirmations); - break; } } - else - { - Console.WriteLine($"Accepting Confirmations on {account.AccountName}"); - if (Verbose) Console.WriteLine("Refeshing Session..."); - account.RefreshSession(); - if (Verbose) Console.WriteLine("Fetching Confirmations..."); - Confirmation[] confirmations = account.FetchConfirmations(); - if (Verbose) Console.WriteLine("Accepting Confirmations..."); - account.AcceptMultipleConfirmations(confirmations); - } } } }