public void TestPlugin()
{
if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
{
// prime Linux log files
IPBanPlugin.IPBanLoginFailed("SSH", "User1", "78.88.88.88");
foreach (IPBanLogFileScanner toParse in service.LogFilesToParse)
{
toParse.PingFiles();
}
}
service.RunCycle().Sync();
for (int i = 0; i < 5; i++)
{
IPBanPlugin.IPBanLoginFailed("SSH", "User1", "88.88.88.88");
service.RunCycle().Sync();
// attempt to read failed logins, if they do not match, sleep a bit and try again
for (int j = 0; j < 10 && (!service.DB.TryGetIPAddress("88.88.88.88", out IPBanDB.IPAddressEntry e) || e.FailedLoginCount != i + 1); j++)
{
System.Threading.Thread.Sleep(100);
foreach (IPBanLogFileScanner toParse in service.LogFilesToParse)
{
toParse.PingFiles();
}
service.RunCycle().Sync();
}
IPBanService.UtcNow += TimeSpan.FromMinutes(5.0);
}
service.RunCycle().Sync();
Assert.IsTrue(service.Firewall.IsIPAddressBlocked("88.88.88.88", out _));
if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
{
string toDelete = $"/var/log/ipbancustom_{IPBanPlugin.ProcessName}.log";
IPBanExtensionMethods.FileDeleteWithRetry(toDelete);
}
// by default, Windows plugin goes to event viewer, we want to also make sure custom log files work on Windows
if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
{
// prime log file to parse
string file = @"C:/IPBanCustomLogs/ipbancustom_test.log";
Directory.CreateDirectory(Path.GetDirectoryName(file));
IPBanExtensionMethods.FileWriteAllTextWithRetry(file, "awerfoajwerp jaeowr paojwer " + Environment.NewLine);
service.RunCycle().Sync();
System.Threading.Thread.Sleep(100);
foreach (IPBanLogFileScanner toParse in service.LogFilesToParse)
{
toParse.PingFiles();
}
string data = "ipban failed login, ip address: 99.99.99.99, source: SSH, user: User2" + Environment.NewLine;
for (int i = 0; i < 5; i++)
{
File.AppendAllText(file, data);
IPBanService.UtcNow += TimeSpan.FromMinutes(5.0);
foreach (IPBanLogFileScanner toParse in service.LogFilesToParse)
{
toParse.PingFiles();
}
// attempt to read failed logins, if they do not match, sleep a bit and try again
for (int j = 0; j < 10 && (!service.DB.TryGetIPAddress("99.99.99.99", out IPBanDB.IPAddressEntry e) || e.FailedLoginCount != i + 1); j++)
{
System.Threading.Thread.Sleep(100);
service.RunCycle().Sync();
}
service.RunCycle().Sync();
}
try
{
Assert.IsTrue(service.Firewall.IsIPAddressBlocked("99.99.99.99", out _));
}
finally
{
IPBanExtensionMethods.FileDeleteWithRetry(file);
Directory.Delete(Path.GetDirectoryName(file));
using (EventLog appLog = new EventLog("Application", System.Environment.MachineName))
{
appLog.Clear();
}
}
}
}
public void SimpleLogParseTest()
{
string fullPath = Path.Combine(tempPath, "test1.txt");
using (IPBanLogFileScanner scanner = new IPBanIPAddressLogFileScanner(this, TestDnsLookup.Instance,
source: "SSH",
pathAndMask: pathAndMask,
recursive: false,
regexFailure: "__prefix__(?<ipaddress>.+)__suffix(__(?<username>.*?)__end)?",
regexSuccess: "success_prefix__(?<ipaddress>.+)__suffix(__(?<username>.*?)__end)?",
pingIntervalMilliseconds: 0))
{
StreamWriter writer = new StreamWriter(CreateFile(fullPath), Encoding.UTF8)
{
AutoFlush = true
};
// scan once before writing any data, otherwise scanner starts at aned of file and will miss
// the first data written
scanner.PingFiles();
// start off with one ip, do not write the last newline, we will do that later
writer.Write("asdasdasdasdasdasd ");
writer.Write("__prefix__1.1.1.1__suffix message repeated 3 times");
scanner.PingFiles();
Assert.AreEqual(0, failedIPAddresses.Count, "Should not have found ip address yet");
// now write a newline, this should make it pickup the line
writer.WriteLine(" aaa ");
writer.WriteLine();
scanner.PingFiles();
Assert.AreEqual(1, failedIPAddresses.Count, "Did not find all expected ip addresses");
Assert.AreEqual("1.1.1.1", failedIPAddresses[0].IPAddress, "First ip address is wrong");
Assert.AreEqual("SSH", failedIPAddresses[0].Source, "First ip source is wrong");
Assert.AreEqual(3, failedIPAddresses[0].Count, "Repeat count is wrong");
Assert.IsNull(failedIPAddresses[0].UserName, "First user name should be null");
scanner.PingFiles();
Assert.AreEqual(1, failedIPAddresses.Count, "Should not have found more ip address yet");
writer.WriteLine("aowefjapweojfopaejfpaoe4231 343240-302843 -204 8-23084 -0");
writer.WriteLine("__prefix__2.2.2.2__suffix__THISUSER__end");
writer.WriteLine("success_prefix__4.4.4.4__suffix__THISUSER__end");
scanner.PingFiles();
Assert.AreEqual(2, failedIPAddresses.Count, "Did not find all expected ip addresses");
Assert.AreEqual("2.2.2.2", failedIPAddresses[1].IPAddress, "Second ip address is wrong");
Assert.AreEqual("SSH", failedIPAddresses[1].Source, "First ip source is wrong");
Assert.AreEqual("THISUSER", failedIPAddresses[1].UserName, "Second user name is wrong");
Assert.AreEqual(1, failedIPAddresses[1].Count, "Repeat count is wrong");
Assert.AreEqual(1, successIPAddresses.Count);
Assert.IsTrue(successIPAddresses[0].Type == IPAddressEventType.SuccessfulLogin);
Assert.AreEqual("4.4.4.4", successIPAddresses[0].IPAddress);
Assert.AreEqual("SSH", successIPAddresses[0].Source);
Assert.AreEqual("THISUSER", successIPAddresses[0].UserName);
writer.Close();
IPBanExtensionMethods.FileDeleteWithRetry(fullPath);
writer = new StreamWriter(CreateFile(fullPath), Encoding.UTF8)
{
AutoFlush = true
};
writer.WriteLine("__prefix__3.3.3.3__suffix message repeated 4 times");
scanner.PingFiles();
Assert.AreEqual(3, failedIPAddresses.Count, "Did not find all expected ip addresses");
Assert.AreEqual("3.3.3.3", failedIPAddresses[2].IPAddress, "Second ip address is wrong");
Assert.AreEqual("SSH", failedIPAddresses[2].Source, "First ip source is wrong");
Assert.AreEqual(4, failedIPAddresses[2].Count, "Repeat count is wrong");
writer.Close();
}
}