- Add async packet processing

FraudCapturer/DomainHelper.cs

@@ -22,7 +22,7 @@ internal class DomainHelper
         return domains.Distinct().ToArray();
     }
 
-    public static DomainInfo? GetDomainReputation(string domain)
+    public static async Task<DomainInfo?> GetDomainReputation(string domain)
     {
         try
         {
@@ -41,10 +41,19 @@ internal class DomainHelper
             };
 
             HttpContent httpContent = new StringContent(JsonSerializer.Serialize(reqestBody), Encoding.UTF8, "application/json");
+            HttpResponseMessage responseMessage;
 
-            HttpResponseMessage responseMessage = httpClient.PostAsync("https://anti-fish.bitflow.dev/check", httpContent).GetAwaiter().GetResult(); ;
+            try
+            {
+                responseMessage = await httpClient.PostAsync("https://anti-fish.bitflow.dev/check", httpContent);
+            }
+            catch (HttpRequestException ex)
+            {
+                Console.WriteLine($"error: {ex.Message}");
+                return null;
+            }
 
-            string resultString = responseMessage.Content.ReadAsStringAsync().GetAwaiter().GetResult(); ;
+            string resultString = await responseMessage.Content.ReadAsStringAsync();
             AntiFishResultBody? resultBody = JsonSerializer.Deserialize<AntiFishResultBody>(resultString);
 
             AntiFishResult? result = resultBody?.Matches?.FirstOrDefault(m => m.Domain == domain);

FraudCapturer/FirewallHelper.cs

@@ -11,6 +11,8 @@ internal class FirewallHelper
             throw new ArgumentNullException(nameof(ipAddress));
         }
 
+        lock (Program.IpStorePath)
+        {
             AddRuleIfDoesnotExist(ipAddress);
 
             File.AppendAllText(Program.IpStorePath, $"{Environment.NewLine}{ipAddress}");
@@ -27,6 +29,7 @@ internal class FirewallHelper
             process.StartInfo = startInfo;
             _ = process.Start();
         }
+    }
 
     public static void UnblockIp(IPAddress? ipAddress)
     {
@@ -35,6 +38,8 @@ internal class FirewallHelper
             throw new ArgumentNullException(nameof(ipAddress));
         }
 
+        lock (Program.IpStorePath)
+        {
             List<string> iPs = File.ReadAllLines(Program.IpStorePath).ToList();
             iPs.Remove(ipAddress.ToString());
 
@@ -50,8 +55,11 @@ internal class FirewallHelper
             process.StartInfo = startInfo;
             _ = process.Start();
         }
+    }
 
     public static void AddRuleIfDoesnotExist(IPAddress ipAddress)
+    {
+        lock (Program.IpStorePath)
         {
             System.Diagnostics.Process process = new System.Diagnostics.Process();
             System.Diagnostics.ProcessStartInfo startInfo = new System.Diagnostics.ProcessStartInfo
@@ -64,4 +72,5 @@ internal class FirewallHelper
             _ = process.Start();
             process.WaitForExit();
         }
+    }
 }

FraudCapturer/IpHelper.cs

@@ -7,11 +7,20 @@ internal class IpHelper
 {
     public static string? ProxycheckApiKey { get; set; }
 
-    public static IpInfo? GetIpReputation(IPAddress ipAddress)
+    public static async Task<IpInfo?> GetIpReputation(IPAddress ipAddress)
     {
         HttpClient httpClient = new HttpClient();
+        string rawResponseData;
 
-        string rawResponseData = httpClient.GetStringAsync($"http://proxycheck.io/v2/{ipAddress}?key={ProxycheckApiKey}&risk=2&vpn=1&asn=1&tag={Program.AppName}({Environment.MachineName})").GetAwaiter().GetResult();
+        try
+        {
+            rawResponseData = await httpClient.GetStringAsync($"http://proxycheck.io/v2/{ipAddress}?key={ProxycheckApiKey}&risk=2&vpn=1&asn=1&tag={Program.AppName}({Environment.MachineName})");
+        }
+        catch (HttpRequestException ex)
+        {
+            Console.WriteLine($"error: {ex.Message}");
+            return null;
+        }
 
         JsonDocument responseData = JsonDocument.Parse(rawResponseData);
 

FraudCapturer/Program.cs

@@ -3,6 +3,7 @@ using PacketDotNet;
 
 using SharpPcap;
 
+using System.Collections.Concurrent;
 using System.Net;
 
 namespace FraudCapturer;
@@ -19,8 +20,8 @@ public class Program
     private static DateTime lastCacheClear;
     private static string lastDomain = string.Empty;
 
-    private static readonly List<string> capturedIpsCache = new();
+    private static readonly ConcurrentBag<string> capturedIpsCache = new();
-    private static readonly Dictionary<string, DomainInfo> capturedDomainsCache = new();
+    private static readonly ConcurrentDictionary<string, DomainInfo?> capturedDomainsCache = new();
 
     /// <summary>
     /// The main entry point for the application.
@@ -101,6 +102,11 @@ public class Program
     private static void Device_OnPacketArrival(object sender, PacketCapture e)
     {
         RawCapture rawPacket = e.GetPacket();
+        ProcessRawPacket(rawPacket);
+    }
+
+    private static async void ProcessRawPacket(RawCapture rawPacket)
+    {
         Packet packet = Packet.ParsePacket(rawPacket.LinkLayerType, rawPacket.Data);
         if (packet is EthernetPacket)
         {
@@ -137,27 +143,27 @@ public class Program
                 }
 
                 //Check if a DNS packet contains a "dangerous" domain.
-                CheckDns(packet, remoteIpAddress, direction);
+                await CheckDns(packet, remoteIpAddress, direction);
 
                 if (capturedIpsCache.Contains(remoteIpAddress.ToString()))
                 {
                     return;
                 }
 
-                TimeSpan timeRemainingUntilCacheReset = new TimeSpan(0, 10, 0) - (DateTime.Now - lastCacheClear);
-                Console.WriteLine($"Next cache reset in {timeRemainingUntilCacheReset.Minutes} minute(s) and {timeRemainingUntilCacheReset.Seconds} second(s)");
-
                 capturedIpsCache.Add(remoteIpAddress.ToString());
 
                 //Check if ip address is "dangerous" or blocked
-                CheckIpAddress(remoteIpAddress, direction);
+                await CheckIpAddress(remoteIpAddress, direction);
+
+                TimeSpan timeRemainingUntilCacheReset = new TimeSpan(0, 10, 0) - (DateTime.Now - lastCacheClear);
+                Console.WriteLine($"Next cache reset in {timeRemainingUntilCacheReset.Minutes} minute(s) and {timeRemainingUntilCacheReset.Seconds} second(s)");
             }
         }
     }
 
-    private static void CheckIpAddress(IPAddress remoteIpAddress, string direction)
+    private static async Task CheckIpAddress(IPAddress remoteIpAddress, string direction)
     {
-        IpInfo? ipInfo = IpHelper.GetIpReputation(remoteIpAddress);
+        IpInfo? ipInfo = await IpHelper.GetIpReputation(remoteIpAddress);
 
         if (IpHelper.IsInternalIpAddress(remoteIpAddress.ToString()))
         {
@@ -200,7 +206,7 @@ public class Program
         Console.ResetColor();
     }
 
-    private static void CheckDns(Packet packet, IPAddress remoteIpAddress, string direction)
+    private static async Task CheckDns(Packet packet, IPAddress remoteIpAddress, string direction)
     {
         TransportPacket transportPacket = packet.Extract<TcpPacket>();
         transportPacket ??= packet.Extract<UdpPacket>();
@@ -219,7 +225,8 @@ public class Program
                 }
                 else
                 {
-                    domainInfo = DomainHelper.GetDomainReputation(domain);
+                    domainInfo = await DomainHelper.GetDomainReputation(domain);
+                    _ = capturedDomainsCache.TryAdd(domain, domainInfo);
                 }
 
                 if (domainInfo is null)