492 lines
14 KiB
C#
492 lines
14 KiB
C#
using Newtonsoft.Json;
|
|
using SteamAuth;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
|
|
public static class Program
|
|
{
|
|
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;
|
|
|
|
/// <summary>
|
|
/// The main entry point for the application
|
|
/// </summary>
|
|
[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("maFiles path: {0}", SteamGuardPath);
|
|
|
|
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":
|
|
throw new NotSupportedException();
|
|
break;
|
|
case "trade":
|
|
Trade(user);
|
|
break;
|
|
case "accept-all":
|
|
AcceptAllTrades(user);
|
|
break;
|
|
default:
|
|
Console.WriteLine("error: Unknown action: {0}", action);
|
|
return;
|
|
}
|
|
}
|
|
|
|
static void ShowHelp()
|
|
{
|
|
var descPadding = 26;
|
|
var descWidth = Console.BufferWidth - descPadding;
|
|
if (descWidth < 20)
|
|
descWidth = 20;
|
|
else if (descWidth > 56)
|
|
descWidth = 56;
|
|
|
|
var flags = new Dictionary<string, string>
|
|
{
|
|
{ "-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<string, string>
|
|
{
|
|
{ "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" },
|
|
};
|
|
|
|
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 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 colorBackground = Console.BackgroundColor;
|
|
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"); // 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();
|
|
}
|
|
|
|
if (t == selected)
|
|
{
|
|
Console.BackgroundColor = itemColor;
|
|
Console.ForegroundColor = colorBackground;
|
|
}
|
|
else
|
|
{
|
|
Console.ForegroundColor = itemColor;
|
|
Console.BackgroundColor = colorBackground;
|
|
}
|
|
|
|
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;
|
|
default:
|
|
break;
|
|
}
|
|
} while (!confirm);
|
|
|
|
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
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...");
|
|
account.RefreshSession();
|
|
if (Verbose) Console.WriteLine("Fetching Confirmations...");
|
|
Confirmation[] confirmations = account.FetchConfirmations();
|
|
if (Verbose) Console.WriteLine("Accepting Confirmations...");
|
|
account.AcceptMultipleConfirmations(confirmations);
|
|
}
|
|
}
|
|
}
|
|
}
|