mirror of
https://github.com/jellyfin/jellyfin.git
synced 2025-06-03 13:44:22 -04:00
Add check for ProviderIds to prevent '=' from appearing in keys, also support '=' in the values. (#12274)
This commit is contained in:
parent
c666f9d050
commit
3262f8dc2a
@ -1046,9 +1046,10 @@ namespace Emby.Server.Implementations.Data
|
|||||||
foreach (var part in value.SpanSplit('|'))
|
foreach (var part in value.SpanSplit('|'))
|
||||||
{
|
{
|
||||||
var providerDelimiterIndex = part.IndexOf('=');
|
var providerDelimiterIndex = part.IndexOf('=');
|
||||||
if (providerDelimiterIndex != -1 && providerDelimiterIndex == part.LastIndexOf('='))
|
// Don't let empty values through
|
||||||
|
if (providerDelimiterIndex != -1 && part.Length != providerDelimiterIndex + 1)
|
||||||
{
|
{
|
||||||
item.SetProviderId(part.Slice(0, providerDelimiterIndex).ToString(), part.Slice(providerDelimiterIndex + 1).ToString());
|
item.SetProviderId(part[..providerDelimiterIndex].ToString(), part[(providerDelimiterIndex + 1)..].ToString());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -111,31 +111,32 @@ namespace MediaBrowser.Model.Entities
|
|||||||
/// Sets a provider id.
|
/// Sets a provider id.
|
||||||
/// </summary>
|
/// </summary>
|
||||||
/// <param name="instance">The instance.</param>
|
/// <param name="instance">The instance.</param>
|
||||||
/// <param name="name">The name.</param>
|
/// <param name="name">The name, this should not contain a '=' character.</param>
|
||||||
/// <param name="value">The value.</param>
|
/// <param name="value">The value.</param>
|
||||||
public static void SetProviderId(this IHasProviderIds instance, string name, string? value)
|
/// <remarks>Due to how deserialization from the database works the name can not contain '='.</remarks>
|
||||||
|
public static void SetProviderId(this IHasProviderIds instance, string name, string value)
|
||||||
{
|
{
|
||||||
ArgumentNullException.ThrowIfNull(instance);
|
ArgumentNullException.ThrowIfNull(instance);
|
||||||
|
ArgumentException.ThrowIfNullOrEmpty(name);
|
||||||
|
ArgumentException.ThrowIfNullOrEmpty(value);
|
||||||
|
|
||||||
// If it's null remove the key from the dictionary
|
// When name contains a '=' it can't be deserialized from the database
|
||||||
if (string.IsNullOrEmpty(value))
|
if (name.Contains('=', StringComparison.Ordinal))
|
||||||
{
|
{
|
||||||
instance.ProviderIds?.Remove(name);
|
throw new ArgumentException("Provider id name cannot contain '='", nameof(name));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ensure it exists
|
||||||
|
instance.ProviderIds ??= new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
|
||||||
|
|
||||||
|
// Match on internal MetadataProvider enum string values before adding arbitrary providers
|
||||||
|
if (_metadataProviderEnumDictionary.TryGetValue(name, out var enumValue))
|
||||||
|
{
|
||||||
|
instance.ProviderIds[enumValue] = value;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
// Ensure it exists
|
instance.ProviderIds[name] = value;
|
||||||
instance.ProviderIds ??= new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
|
|
||||||
|
|
||||||
// Match on internal MetadataProvider enum string values before adding arbitrary providers
|
|
||||||
if (_metadataProviderEnumDictionary.TryGetValue(name, out var enumValue))
|
|
||||||
{
|
|
||||||
instance.ProviderIds[enumValue] = value;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
instance.ProviderIds[name] = value;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -149,5 +150,30 @@ namespace MediaBrowser.Model.Entities
|
|||||||
{
|
{
|
||||||
instance.SetProviderId(provider.ToString(), value);
|
instance.SetProviderId(provider.ToString(), value);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Removes a provider id.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="instance">The instance.</param>
|
||||||
|
/// <param name="name">The name.</param>
|
||||||
|
public static void RemoveProviderId(this IHasProviderIds instance, string name)
|
||||||
|
{
|
||||||
|
ArgumentNullException.ThrowIfNull(instance);
|
||||||
|
ArgumentException.ThrowIfNullOrEmpty(name);
|
||||||
|
|
||||||
|
instance.ProviderIds?.Remove(name);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Removes a provider id.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="instance">The instance.</param>
|
||||||
|
/// <param name="provider">The provider.</param>
|
||||||
|
public static void RemoveProviderId(this IHasProviderIds instance, MetadataProvider provider)
|
||||||
|
{
|
||||||
|
ArgumentNullException.ThrowIfNull(instance);
|
||||||
|
|
||||||
|
instance.ProviderIds?.Remove(provider.ToString());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -141,7 +141,7 @@ namespace Jellyfin.Model.Tests.Entities
|
|||||||
public void SetProviderId_Null_Remove()
|
public void SetProviderId_Null_Remove()
|
||||||
{
|
{
|
||||||
var provider = new ProviderIdsExtensionsTestsObject();
|
var provider = new ProviderIdsExtensionsTestsObject();
|
||||||
provider.SetProviderId(MetadataProvider.Imdb, null!);
|
Assert.Throws<ArgumentNullException>(() => provider.SetProviderId(MetadataProvider.Imdb, null!));
|
||||||
Assert.Empty(provider.ProviderIds);
|
Assert.Empty(provider.ProviderIds);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -150,8 +150,8 @@ namespace Jellyfin.Model.Tests.Entities
|
|||||||
{
|
{
|
||||||
var provider = new ProviderIdsExtensionsTestsObject();
|
var provider = new ProviderIdsExtensionsTestsObject();
|
||||||
provider.ProviderIds[MetadataProvider.Imdb.ToString()] = ExampleImdbId;
|
provider.ProviderIds[MetadataProvider.Imdb.ToString()] = ExampleImdbId;
|
||||||
provider.SetProviderId(MetadataProvider.Imdb, string.Empty);
|
Assert.Throws<ArgumentException>(() => provider.SetProviderId(MetadataProvider.Imdb, string.Empty));
|
||||||
Assert.Empty(provider.ProviderIds);
|
Assert.Single(provider.ProviderIds);
|
||||||
}
|
}
|
||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
@ -182,10 +182,20 @@ namespace Jellyfin.Model.Tests.Entities
|
|||||||
ProviderIds = null!
|
ProviderIds = null!
|
||||||
};
|
};
|
||||||
|
|
||||||
nullProvider.SetProviderId(MetadataProvider.Imdb, string.Empty);
|
Assert.Throws<ArgumentException>(() => nullProvider.SetProviderId(MetadataProvider.Imdb, string.Empty));
|
||||||
Assert.Null(nullProvider.ProviderIds);
|
Assert.Null(nullProvider.ProviderIds);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public void RemoveProviderId_Null_Remove()
|
||||||
|
{
|
||||||
|
var provider = new ProviderIdsExtensionsTestsObject();
|
||||||
|
|
||||||
|
provider.ProviderIds[MetadataProvider.Imdb.ToString()] = ExampleImdbId;
|
||||||
|
provider.RemoveProviderId(MetadataProvider.Imdb);
|
||||||
|
Assert.Empty(provider.ProviderIds);
|
||||||
|
}
|
||||||
|
|
||||||
private sealed class ProviderIdsExtensionsTestsObject : IHasProviderIds
|
private sealed class ProviderIdsExtensionsTestsObject : IHasProviderIds
|
||||||
{
|
{
|
||||||
public static readonly ProviderIdsExtensionsTestsObject Empty = new ProviderIdsExtensionsTestsObject();
|
public static readonly ProviderIdsExtensionsTestsObject Empty = new ProviderIdsExtensionsTestsObject();
|
||||||
|
Loading…
x
Reference in New Issue
Block a user