AfRApay/AfRApay.FTM/Program.cs

195 lines
7.4 KiB
C#

using System.CommandLine;
using PCSC;
using PCSC.Monitoring;
using PCSC.Iso7816;
const int defaultAmount = 150;
var rootCommand = new RootCommand("Fancy Test Machine for AfRApay");
var listReadersOption = new Option<bool>("--list-readers", "List card readers and exit");
rootCommand.Add(listReadersOption);
var webAddrOption = new Option<Uri>("--web-addr", "Base URL for AfRApay.Web");
webAddrOption.SetDefaultValue(new Uri("http://127.0.0.1:5296"));
rootCommand.Add(webAddrOption);
rootCommand.SetHandler((listReaders, webAddr) => {
using var context = ContextFactory.Instance.Establish(SCardScope.System);
// Ignore Yubikeys.
var readerNames = context.GetReaders()
.Where((readerName) => !readerName.Contains("Yubico"))
.ToArray();
// We need at least one card reader or this won't work!
if (readerNames.Length == 0) {
Console.Error.WriteLine("Error: no card reader detected");
Environment.Exit(1);
}
// If --list-readers is passed, list readers and exit.
if (listReaders) {
Console.Error.WriteLine("----------- Connected Readers ----------");
foreach (var name in readerNames) {
Console.WriteLine(name);
}
Console.Error.WriteLine("----------------------------------------");
Environment.Exit(0);
}
// Setup!
using HttpClient httpClient = new();
httpClient.BaseAddress = webAddr;
var state = TerminalState.Debit;
decimal amount = defaultAmount;
Console.Error.WriteLine("----------------------------------------");
Console.Error.WriteLine("--- AfRApay FTM - Fancy Test Machine ---");
Console.Error.WriteLine("----------------------------------------");
Console.Error.WriteLine();
Console.Error.WriteLine("AfRApay.Web: {0}", httpClient.BaseAddress);
Console.Error.WriteLine();
Console.Error.WriteLine("Hotkeys (case insensitive):");
Console.Error.WriteLine(" [-] Debit (default)");
Console.Error.WriteLine(" [+] Credit");
Console.Error.WriteLine(" [=] Set amount (default: €{0:C})", amount/100M);
Console.Error.WriteLine();
Console.Error.WriteLine(" [B] Balance query");
Console.Error.WriteLine(" [L] Link card (initiate from web UI)");
Console.Error.WriteLine(" [Esc] Cancel, reset state and amount");
Console.Error.WriteLine();
Console.Error.WriteLine("----------------------------------------");
// Listen for events on all connected readers.
using var monitor = MonitorFactory.Instance.Create(SCardScope.System);
monitor.Initialized += (_, args) => Console.WriteLine("[ Reader Initialized: {0} ]", args.ReaderName);
monitor.MonitorException += (_, args) => {
Console.Error.WriteLine("! ERROR: {0}", args);
Environment.Exit(1);
};
monitor.StatusChanged += (_, args) => Console.WriteLine("~ {0} -> {1}", args.LastState, args.NewState);
monitor.CardInserted += (_, args) => {
Console.WriteLine("> TAP: {0}", Convert.ToHexString(args.Atr));
var reader = new IsoReader(context, args.ReaderName, SCardShareMode.Shared, SCardProtocol.Any);
HandleTap(reader, httpClient, state, amount);
};
monitor.CardRemoved += (_, _) => {
Console.WriteLine("< OFF");
Console.WriteLine(); // Write a blank line between card taps for readability.
};
Console.WriteLine("[ Starting... ]");
monitor.Start(readerNames);
// Handle hotkeys.
while (true) {
var key = Console.ReadKey();
var dontPrint = false;
switch (key.Key) {
case ConsoleKey.Subtract:
case ConsoleKey.OemMinus:
state = TerminalState.Debit;
break;
case ConsoleKey.Add:
case ConsoleKey.OemPlus:
state = TerminalState.Credit;
break;
case 0 when key.KeyChar == '=':
Console.Error.Write("\b => ENTER AMOUNT: ");
amount = Math.Abs(int.Parse(Console.ReadLine() ?? "150".Trim()));
break;
case ConsoleKey.L:
state = TerminalState.Link;
break;
case ConsoleKey.B:
state = TerminalState.Balance;
break;
case ConsoleKey.Escape:
state = TerminalState.Debit;
amount = defaultAmount;
break;
default:
Console.Error.Write("\b");
dontPrint = true;
break;
}
if (dontPrint) {
// Invalid input, just ignore it.
} else if (state is TerminalState.Debit or TerminalState.Credit) {
Console.Error.WriteLine("\b => {0}: €{1}", state, amount);
} else {
Console.Error.WriteLine("\b => {0}", state);
}
}
// ReSharper disable once FunctionNeverReturns
}, listReadersOption, webAddrOption);
return await rootCommand.InvokeAsync(args);
// Queries a card for data when one is tapped.
static async void HandleTap(IIsoReader reader, HttpClient httpClient, TerminalState state, decimal amount) {
// Send a PCSC pseudo-APDU to query the ISO 14443 UID.
var uidRsp = reader.Transmit(new CommandApdu(IsoCase.Case2Short, SCardProtocol.Any) {
CLA = 0xFF,
Instruction = InstructionCode.GetData,
P1 = 0x00,
P2 = 0x00,
});
if (!IsSucc(uidRsp)) {
Console.Error.WriteLine("--> Card Error: SW1={0} SW2={1}", (SW1Code)uidRsp.SW1, uidRsp.SW2);
return;
}
var uid = uidRsp.GetData();
Console.WriteLine(" UID: {0}", Convert.ToHexString(uid));
// Query the backend, which endpoint depending on terminal state.
switch (state) {
case TerminalState.Debit:
case TerminalState.Credit:
var finalAmount = Math.Abs(amount) * (state == TerminalState.Debit ? -1 : 1);
var idempotencyKey = new byte[16];
Random.Shared.NextBytes(idempotencyKey);
var ik = Convert.ToHexString(idempotencyKey);
await CallPut(httpClient, $"/api/card/{Convert.ToHexString(uid)}/transaction/{ik}?amount={finalAmount}");
break;
case TerminalState.Link:
await CallGet(httpClient, $"/api/card/{Convert.ToHexString(uid)}/link");
break;
case TerminalState.Balance:
await CallGet(httpClient, $"/api/card/{Convert.ToHexString(uid)}/balance");
break;
default:
Console.Error.WriteLine("UNKNOWN TERMINAL STATE: {0}", state);
break;
}
}
// Was the command successful?
static bool IsSucc(Response rsp) {
return rsp is { SW1: (byte)SW1Code.Normal, SW2: 0x00 };
}
static async Task<string> CallGet(HttpClient client, string path) {
Console.WriteLine(" -> GET {0}", path);
var rsp = await client.GetStringAsync(path);
Console.WriteLine(" <- {0}", rsp);
return rsp;
}
static async Task<string> CallPut(HttpClient client, string path) {
Console.WriteLine(" -> PUT {0}", path);
var rsp = await client.PutAsync(path, new StringContent("")).Result.Content.ReadAsStringAsync();
Console.WriteLine(" <- {0}", rsp);
return rsp;
}
// Terminal State.
internal enum TerminalState {
Debit,
Credit,
Link,
Balance,
};