/// <summary>
/// Create a test IPBanService
/// </summary>
/// <param name="directory">Root directory</param>
/// <param name="configFileName">Config file name</param>
/// <param name="configFileModifier">Change config file (param are file text, returns new file text)</param>
/// <returns>Service</returns>
public static T CreateAndStartIPBanTestService <T>(string directory = null, string configFileName = null,
Func <string, string> configFileModifier = null) where T : IPBanService
{
DefaultHttpRequestMaker.DisableLiveRequests = true;
// cleanup any db, set or tbl files
foreach (string file in Directory.GetFiles(AppDomain.CurrentDomain.BaseDirectory, "*.set")
.Union(Directory.GetFiles(AppDomain.CurrentDomain.BaseDirectory, "*.tbl"))
.Union(Directory.GetFiles(AppDomain.CurrentDomain.BaseDirectory, "*.set6"))
.Union(Directory.GetFiles(AppDomain.CurrentDomain.BaseDirectory, "*.tbl6"))
.Union(Directory.GetFiles(AppDomain.CurrentDomain.BaseDirectory, "*.sqlite*"))
.Union(Directory.GetFiles(AppDomain.CurrentDomain.BaseDirectory, "*journal*")))
{
for (int i = 0; i < 10; i++)
{
try
{
File.Delete(file);
break;
}
catch
{
Thread.Sleep(1000);
}
}
}
if (string.IsNullOrWhiteSpace(directory))
{
directory = Path.GetDirectoryName(IPBanAssembly.Location);
}
if (string.IsNullOrWhiteSpace(configFileName))
{
configFileName = IPBanService.ConfigFileName;
}
string configFilePath = Path.Combine(directory, configFileName);
string configFileText = File.ReadAllText(configFilePath);
configFilePath += ".tmp";
if (configFileModifier != null)
{
configFileText = configFileModifier(configFileText);
}
File.WriteAllText(configFilePath, configFileText);
T service = IPBanService.CreateService <T>() as T;
service.ExternalIPAddressLookup = LocalMachineExternalIPAddressLookupTest.Instance;
service.ConfigFilePath = configFilePath;
service.MultiThreaded = false;
service.ManualCycle = true;
service.BannedIPAddressHandler = service;
service.Version = "1.1.1.1";
service.StartAsync().Sync();
service.DB.Truncate(true);
service.Firewall.Truncate();
return(service);
}
private static void CreateService()
{
if (service != null)
{
service.Dispose();
}
service = IPBanService.CreateService <IPBanService>();
service.Start();
}
public static void MacMain(string[] args)
{
IPBanService service = IPBanService.CreateService <IPBanService>();
service.Start();
IPBanLog.Warn("IPBan Mac Service Running, Press Ctrl-C to quit.");
ManualResetEvent wait = new ManualResetEvent(false);
wait.WaitOne();
}
public static Task <int> MainService <T>(string[] args, out T service) where T : IPBanService
{
T _service = IPBanService.CreateService <T>();
service = _service;
return(MainService(args, (_args) =>
{
_service.Start();
}, () =>
{
_service.Stop();
}, (_timeout) =>
{
return _service.Wait(_timeout);
}));
}
public static Task <int> MainService <T>(string[] args, out T service) where T : IPBanService
{
T _service = IPBanService.CreateService <T>();
service = _service;
return(MainService(args, (_args) =>
{
// kick off start in background thread, make sure service starts up in a timely manner
return Task.Run(() => _service.Start());
}, () =>
{
_service.Stop();
}, (_timeout) =>
{
return _service.Wait(_timeout);
}));
}
public static async Task MainService <T>(string[] args, Action started = null) where T : IPBanService
{
T _service = IPBanService.CreateService <T>();
await MainService(args, async (_args) =>
{
// kick off start in background thread, make sure service starts up in a timely manner
await _service.StartAsync();
started?.Invoke();
// wait for service to end
await _service.WaitAsync(Timeout.Infinite);
}, () =>
{
// stop the service, will cause any WaitAsync to exit
_service.Stop();
});
}
/// <summary>
/// IPBan main method
/// </summary>
/// <param name="args">Args</param>
/// <returns>Task</returns>
public static async Task Main(string[] args)
{
if (args.Length != 0 && (args[0].Equals("info", StringComparison.OrdinalIgnoreCase) ||
args[0].Equals("-info", StringComparison.OrdinalIgnoreCase)))
{
Logger.Warn("System info: {0}", OSUtility.OSString());
return;
}
IPBanService service = null;
await IPBanServiceRunner.MainService(args, (CancellationToken cancelToken) =>
{
service = IPBanService.CreateService <IPBanService>();
return(service.RunAsync(cancelToken));
}, (CancellationToken cancelToken) =>
{
service?.Dispose();
return(Task.CompletedTask);
});
}
/// <summary>
/// IPBan main method
/// </summary>
/// <param name="args">Args</param>
/// <returns>Task</returns>
public static async Task Main(string[] args)
{
if (ProcessCommandLine(args))
{
return;
}
IPBanService service = null;
await IPBanServiceRunner.MainService(args, (CancellationToken cancelToken) =>
{
service = IPBanService.CreateService <IPBanService>();
Logger.Warn("IPBan is free software created and refined over many years.");
Logger.Warn("Please consider upgrading to the pro version for more advanced functions, shared ban lists and much more.");
Logger.Warn("Learn more at https://ipban.com");
return(service.RunAsync(cancelToken));
}, (CancellationToken cancelToken) =>
{
service?.Dispose();
return(Task.CompletedTask);
});
}
/// <summary>
/// IPBan main method
/// </summary>
/// <param name="args">Args</param>
/// <returns>Task</returns>
public static async Task Main(string[] args)
{
if (args.Length != 0 && (args[0].Equals("info", StringComparison.OrdinalIgnoreCase) ||
args[0].Equals("-info", StringComparison.OrdinalIgnoreCase)))
{
Logger.Warn("System info: {0}", OSUtility.OSString());
return;
}
IPBanService service = null;
await IPBanServiceRunner.MainService(args, (CancellationToken cancelToken) =>
{
service = IPBanService.CreateService <IPBanService>();
Logger.Warn("IPBan is free software created and refined over many years.");
Logger.Warn("Please consider upgrading to the pro version for more advanced functions, shared ban lists and much more.");
Logger.Warn("Learn more at https://ipban.com");
return(service.RunAsync(cancelToken));
}, (CancellationToken cancelToken) =>
{
service?.Dispose();
return(Task.CompletedTask);
});
}