88 lines
3.1 KiB
C#
88 lines
3.1 KiB
C#
using System.Globalization;
|
|
using System.Text;
|
|
using Jellyfin.Networking.Configuration;
|
|
using Jellyfin.Networking.Manager;
|
|
using Jellyfin.Server.Extensions;
|
|
using MediaBrowser.Common.Configuration;
|
|
using Microsoft.AspNetCore.Builder;
|
|
using Microsoft.Extensions.Logging.Abstractions;
|
|
using Moq;
|
|
using Xunit;
|
|
|
|
namespace Jellyfin.Server.Tests
|
|
{
|
|
public class ParseNetworkTests
|
|
{
|
|
/// <summary>
|
|
/// Order of the result has always got to be hosts, then networks.
|
|
/// </summary>
|
|
/// <param name="ip4">IP4 enabled.</param>
|
|
/// <param name="ip6">IP6 enabled.</param>
|
|
/// <param name="hostList">List to parse.</param>
|
|
/// <param name="match">What it should match.</param>
|
|
[Theory]
|
|
// [InlineData(true, true, "192.168.0.0/16,www.yahoo.co.uk", "::ffff:212.82.100.150,::ffff:192.168.0.0/16")] <- fails on Max. www.yahoo.co.uk resolves to a different ip address.
|
|
// [InlineData(true, false, "192.168.0.0/16,www.yahoo.co.uk", "212.82.100.150,192.168.0.0/16")]
|
|
[InlineData(true, true, "192.168.t,127.0.0.1,1234.1232.12.1234", "::ffff:127.0.0.1")]
|
|
[InlineData(true, false, "192.168.x,127.0.0.1,1234.1232.12.1234", "127.0.0.1")]
|
|
[InlineData(true, true, "::1", "::1/128")]
|
|
public void TestNetworks(bool ip4, bool ip6, string hostList, string match)
|
|
{
|
|
using var nm = CreateNetworkManager();
|
|
|
|
var settings = new NetworkConfiguration
|
|
{
|
|
EnableIPV4 = ip4,
|
|
EnableIPV6 = ip6
|
|
};
|
|
|
|
var result = match + ",";
|
|
ForwardedHeadersOptions options = new ForwardedHeadersOptions();
|
|
|
|
// Need this here as ::1 and 127.0.0.1 are in them by default.
|
|
options.KnownProxies.Clear();
|
|
options.KnownNetworks.Clear();
|
|
|
|
ApiServiceCollectionExtensions.AddProxyAddresses(settings, hostList.Split(','), options);
|
|
|
|
var sb = new StringBuilder();
|
|
foreach (var item in options.KnownProxies)
|
|
{
|
|
sb.Append(item)
|
|
.Append(',');
|
|
}
|
|
|
|
foreach (var item in options.KnownNetworks)
|
|
{
|
|
sb.Append(item.Prefix)
|
|
.Append('/')
|
|
.Append(item.PrefixLength.ToString(CultureInfo.InvariantCulture))
|
|
.Append(',');
|
|
}
|
|
|
|
Assert.Equal(sb.ToString(), result);
|
|
}
|
|
|
|
private static IConfigurationManager GetMockConfig(NetworkConfiguration conf)
|
|
{
|
|
var configManager = new Mock<IConfigurationManager>
|
|
{
|
|
CallBase = true
|
|
};
|
|
configManager.Setup(x => x.GetConfiguration(It.IsAny<string>())).Returns(conf);
|
|
return configManager.Object;
|
|
}
|
|
|
|
private static NetworkManager CreateNetworkManager()
|
|
{
|
|
var conf = new NetworkConfiguration()
|
|
{
|
|
EnableIPV6 = true,
|
|
EnableIPV4 = true,
|
|
};
|
|
|
|
return new NetworkManager(GetMockConfig(conf), new NullLogger<NetworkManager>());
|
|
}
|
|
}
|
|
}
|