mirror of
https://github.com/jellyfin/jellyfin.git
synced 2026-05-30 12:28:27 +01:00
b555de4cea
Wrap the downloaded stream in an explicit await using block with ConfigureAwait(false), matching the pattern already used in LiveStreamHelper and similar callers. Also add ConfigureAwait(false) to the ZipFile.ExtractToDirectoryAsync call. Part of #2149
587 lines
23 KiB
C#
587 lines
23 KiB
C#
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.IO.Compression;
|
|
using System.Linq;
|
|
using System.Net.Http;
|
|
using System.Net.Http.Json;
|
|
using System.Security.Cryptography;
|
|
using System.Text.Json;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Jellyfin.Data.Events;
|
|
using Jellyfin.Extensions;
|
|
using Jellyfin.Extensions.Json;
|
|
using MediaBrowser.Common.Configuration;
|
|
using MediaBrowser.Common.Net;
|
|
using MediaBrowser.Common.Plugins;
|
|
using MediaBrowser.Common.Updates;
|
|
using MediaBrowser.Controller;
|
|
using MediaBrowser.Controller.Configuration;
|
|
using MediaBrowser.Controller.Events;
|
|
using MediaBrowser.Controller.Events.Updates;
|
|
using MediaBrowser.Model.Plugins;
|
|
using MediaBrowser.Model.Updates;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace Emby.Server.Implementations.Updates
|
|
{
|
|
/// <summary>
|
|
/// Manages all install, uninstall, and update operations for the system and individual plugins.
|
|
/// </summary>
|
|
public class InstallationManager : IInstallationManager
|
|
{
|
|
/// <summary>
|
|
/// The logger.
|
|
/// </summary>
|
|
private readonly ILogger<InstallationManager> _logger;
|
|
private readonly IApplicationPaths _appPaths;
|
|
private readonly IEventManager _eventManager;
|
|
private readonly IHttpClientFactory _httpClientFactory;
|
|
private readonly IServerConfigurationManager _config;
|
|
private readonly JsonSerializerOptions _jsonSerializerOptions;
|
|
private readonly IPluginManager _pluginManager;
|
|
|
|
/// <summary>
|
|
/// Gets the application host.
|
|
/// </summary>
|
|
/// <value>The application host.</value>
|
|
private readonly IServerApplicationHost _applicationHost;
|
|
private readonly Lock _currentInstallationsLock = new();
|
|
|
|
/// <summary>
|
|
/// The current installations.
|
|
/// </summary>
|
|
private readonly List<(InstallationInfo Info, CancellationTokenSource Token)> _currentInstallations;
|
|
|
|
/// <summary>
|
|
/// The completed installations.
|
|
/// </summary>
|
|
private readonly ConcurrentBag<InstallationInfo> _completedInstallationsInternal;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="InstallationManager"/> class.
|
|
/// </summary>
|
|
/// <param name="logger">The <see cref="ILogger{InstallationManager}"/>.</param>
|
|
/// <param name="appHost">The <see cref="IServerApplicationHost"/>.</param>
|
|
/// <param name="appPaths">The <see cref="IApplicationPaths"/>.</param>
|
|
/// <param name="eventManager">The <see cref="IEventManager"/>.</param>
|
|
/// <param name="httpClientFactory">The <see cref="IHttpClientFactory"/>.</param>
|
|
/// <param name="config">The <see cref="IServerConfigurationManager"/>.</param>
|
|
/// <param name="pluginManager">The <see cref="IPluginManager"/>.</param>
|
|
public InstallationManager(
|
|
ILogger<InstallationManager> logger,
|
|
IServerApplicationHost appHost,
|
|
IApplicationPaths appPaths,
|
|
IEventManager eventManager,
|
|
IHttpClientFactory httpClientFactory,
|
|
IServerConfigurationManager config,
|
|
IPluginManager pluginManager)
|
|
{
|
|
_currentInstallations = new List<(InstallationInfo, CancellationTokenSource)>();
|
|
_completedInstallationsInternal = new ConcurrentBag<InstallationInfo>();
|
|
|
|
_logger = logger;
|
|
_applicationHost = appHost;
|
|
_appPaths = appPaths;
|
|
_eventManager = eventManager;
|
|
_httpClientFactory = httpClientFactory;
|
|
_config = config;
|
|
_jsonSerializerOptions = JsonDefaults.Options;
|
|
_pluginManager = pluginManager;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public IEnumerable<InstallationInfo> CompletedInstallations => _completedInstallationsInternal;
|
|
|
|
/// <inheritdoc />
|
|
public async Task<PackageInfo[]> GetPackages(string manifestName, string manifest, bool filterIncompatible, CancellationToken cancellationToken = default)
|
|
{
|
|
try
|
|
{
|
|
PackageInfo[]? packages = await _httpClientFactory.CreateClient(NamedClient.Default)
|
|
.GetFromJsonAsync<PackageInfo[]>(new Uri(manifest), _jsonSerializerOptions, cancellationToken).ConfigureAwait(false);
|
|
|
|
if (packages is null)
|
|
{
|
|
return Array.Empty<PackageInfo>();
|
|
}
|
|
|
|
var minimumVersion = new Version(0, 0, 0, 1);
|
|
// Store the repository and repository url with each version, as they may be spread apart.
|
|
foreach (var entry in packages)
|
|
{
|
|
for (int a = entry.Versions.Count - 1; a >= 0; a--)
|
|
{
|
|
var ver = entry.Versions[a];
|
|
ver.RepositoryName = manifestName;
|
|
ver.RepositoryUrl = manifest;
|
|
|
|
if (!filterIncompatible)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (!Version.TryParse(ver.TargetAbi, out var targetAbi))
|
|
{
|
|
targetAbi = minimumVersion;
|
|
}
|
|
|
|
// Only show plugins that are greater than or equal to targetAbi.
|
|
if (_applicationHost.ApplicationVersion >= targetAbi)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// Not compatible with this version so remove it.
|
|
entry.Versions.Remove(ver);
|
|
}
|
|
}
|
|
|
|
return packages;
|
|
}
|
|
catch (IOException ex)
|
|
{
|
|
_logger.LogError(ex, "Cannot locate the plugin manifest {Manifest}", manifest);
|
|
return Array.Empty<PackageInfo>();
|
|
}
|
|
catch (JsonException ex)
|
|
{
|
|
_logger.LogError(ex, "Failed to deserialize the plugin manifest retrieved from {Manifest}", manifest);
|
|
return Array.Empty<PackageInfo>();
|
|
}
|
|
catch (UriFormatException ex)
|
|
{
|
|
_logger.LogError(ex, "The URL configured for the plugin repository manifest URL is not valid: {Manifest}", manifest);
|
|
return Array.Empty<PackageInfo>();
|
|
}
|
|
catch (NotSupportedException ex)
|
|
{
|
|
_logger.LogError(ex, "The URL scheme configured for the plugin repository is not supported: {Manifest}", manifest);
|
|
return Array.Empty<PackageInfo>();
|
|
}
|
|
catch (HttpRequestException ex)
|
|
{
|
|
_logger.LogError(ex, "An error occurred while accessing the plugin manifest: {Manifest}", manifest);
|
|
return Array.Empty<PackageInfo>();
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public async Task<IReadOnlyList<PackageInfo>> GetAvailablePackages(CancellationToken cancellationToken = default)
|
|
{
|
|
var result = new List<PackageInfo>();
|
|
foreach (RepositoryInfo repository in _config.Configuration.PluginRepositories)
|
|
{
|
|
if (repository.Enabled && repository.Url is not null)
|
|
{
|
|
// Where repositories have the same content, the details from the first is taken.
|
|
foreach (var package in await GetPackages(repository.Name ?? "Unnamed Repo", repository.Url, true, cancellationToken).ConfigureAwait(true))
|
|
{
|
|
var existing = FilterPackages(result, package.Name, package.Id).FirstOrDefault();
|
|
|
|
// Remove invalid versions from the valid package.
|
|
for (var i = package.Versions.Count - 1; i >= 0; i--)
|
|
{
|
|
var version = package.Versions[i];
|
|
|
|
var plugin = _pluginManager.GetPlugin(package.Id, version.VersionNumber);
|
|
if (plugin is not null)
|
|
{
|
|
await _pluginManager.PopulateManifest(package, version.VersionNumber, plugin.Path, plugin.Manifest.Status).ConfigureAwait(false);
|
|
}
|
|
|
|
// Remove versions with a target ABI greater than the current application version.
|
|
if (Version.TryParse(version.TargetAbi, out var targetAbi) && _applicationHost.ApplicationVersion < targetAbi)
|
|
{
|
|
package.Versions.RemoveAt(i);
|
|
}
|
|
}
|
|
|
|
// Don't add a package that doesn't have any compatible versions.
|
|
if (package.Versions.Count == 0)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (existing is not null)
|
|
{
|
|
// Assumption is both lists are ordered, so slot these into the correct place.
|
|
MergeSortedList(existing.Versions, package.Versions);
|
|
}
|
|
else
|
|
{
|
|
result.Add(package);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public IEnumerable<PackageInfo> FilterPackages(
|
|
IEnumerable<PackageInfo> availablePackages,
|
|
string? name = null,
|
|
Guid id = default,
|
|
Version? specificVersion = null)
|
|
{
|
|
if (!id.IsEmpty())
|
|
{
|
|
availablePackages = availablePackages.Where(x => x.Id.Equals(id));
|
|
}
|
|
else if (name is not null)
|
|
{
|
|
availablePackages = availablePackages.Where(x => x.Name.Equals(name, StringComparison.OrdinalIgnoreCase));
|
|
}
|
|
|
|
if (specificVersion is not null)
|
|
{
|
|
availablePackages = availablePackages.Where(x => x.Versions.Any(y => y.VersionNumber.Equals(specificVersion)));
|
|
}
|
|
|
|
return availablePackages;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public IEnumerable<InstallationInfo> GetCompatibleVersions(
|
|
IEnumerable<PackageInfo> availablePackages,
|
|
string? name = null,
|
|
Guid id = default,
|
|
Version? minVersion = null,
|
|
Version? specificVersion = null)
|
|
{
|
|
var package = FilterPackages(availablePackages, name, id, specificVersion).FirstOrDefault();
|
|
|
|
// Package not found in repository
|
|
if (package is null)
|
|
{
|
|
yield break;
|
|
}
|
|
|
|
var appVer = _applicationHost.ApplicationVersion;
|
|
var availableVersions = package.Versions
|
|
.Where(x => string.IsNullOrEmpty(x.TargetAbi) || Version.Parse(x.TargetAbi) <= appVer);
|
|
|
|
if (specificVersion is not null)
|
|
{
|
|
availableVersions = availableVersions.Where(x => x.VersionNumber.Equals(specificVersion));
|
|
}
|
|
else if (minVersion is not null)
|
|
{
|
|
availableVersions = availableVersions.Where(x => x.VersionNumber >= minVersion);
|
|
}
|
|
|
|
foreach (var v in availableVersions.OrderByDescending(x => x.VersionNumber))
|
|
{
|
|
yield return new InstallationInfo
|
|
{
|
|
Changelog = v.Changelog,
|
|
Id = package.Id,
|
|
Name = package.Name,
|
|
Version = v.VersionNumber,
|
|
SourceUrl = v.SourceUrl,
|
|
Checksum = v.Checksum,
|
|
PackageInfo = package
|
|
};
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public async Task<IEnumerable<InstallationInfo>> GetAvailablePluginUpdates(CancellationToken cancellationToken = default)
|
|
{
|
|
var catalog = await GetAvailablePackages(cancellationToken).ConfigureAwait(false);
|
|
return GetAvailablePluginUpdates(catalog);
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public async Task InstallPackage(InstallationInfo package, CancellationToken cancellationToken)
|
|
{
|
|
ArgumentNullException.ThrowIfNull(package);
|
|
|
|
var innerCancellationTokenSource = new CancellationTokenSource();
|
|
|
|
var tuple = (package, innerCancellationTokenSource);
|
|
|
|
// Add it to the in-progress list
|
|
lock (_currentInstallationsLock)
|
|
{
|
|
_currentInstallations.Add(tuple);
|
|
}
|
|
|
|
using var linkedTokenSource = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, innerCancellationTokenSource.Token);
|
|
var linkedToken = linkedTokenSource.Token;
|
|
|
|
await _eventManager.PublishAsync(new PluginInstallingEventArgs(package)).ConfigureAwait(false);
|
|
|
|
try
|
|
{
|
|
var isUpdate = await InstallPackageInternal(package, linkedToken).ConfigureAwait(false);
|
|
|
|
lock (_currentInstallationsLock)
|
|
{
|
|
_currentInstallations.Remove(tuple);
|
|
}
|
|
|
|
_completedInstallationsInternal.Add(package);
|
|
|
|
if (isUpdate)
|
|
{
|
|
await _eventManager.PublishAsync(new PluginUpdatedEventArgs(package)).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
await _eventManager.PublishAsync(new PluginInstalledEventArgs(package)).ConfigureAwait(false);
|
|
}
|
|
|
|
_applicationHost.NotifyPendingRestart();
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
lock (_currentInstallationsLock)
|
|
{
|
|
_currentInstallations.Remove(tuple);
|
|
}
|
|
|
|
_logger.LogInformation("Package installation cancelled: {0} {1}", package.Name, package.Version);
|
|
|
|
await _eventManager.PublishAsync(new PluginInstallationCancelledEventArgs(package)).ConfigureAwait(false);
|
|
|
|
throw;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "Package installation failed");
|
|
|
|
lock (_currentInstallationsLock)
|
|
{
|
|
_currentInstallations.Remove(tuple);
|
|
}
|
|
|
|
await _eventManager.PublishAsync(new InstallationFailedEventArgs
|
|
{
|
|
InstallationInfo = package,
|
|
Exception = ex
|
|
}).ConfigureAwait(false);
|
|
|
|
throw;
|
|
}
|
|
finally
|
|
{
|
|
// Dispose the progress object and remove the installation from the in-progress list
|
|
tuple.innerCancellationTokenSource.Dispose();
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Uninstalls a plugin.
|
|
/// </summary>
|
|
/// <param name="plugin">The <see cref="LocalPlugin"/> to uninstall.</param>
|
|
public void UninstallPlugin(LocalPlugin plugin)
|
|
{
|
|
if (plugin is null)
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (plugin.Instance?.CanUninstall == false)
|
|
{
|
|
_logger.LogWarning("Attempt to delete non removable plugin {PluginName}, ignoring request", plugin.Name);
|
|
return;
|
|
}
|
|
|
|
plugin.Instance?.OnUninstalling();
|
|
|
|
// Remove it the quick way for now
|
|
_pluginManager.RemovePlugin(plugin);
|
|
|
|
_eventManager.Publish(new PluginUninstalledEventArgs(plugin.GetPluginInfo()));
|
|
|
|
_applicationHost.NotifyPendingRestart();
|
|
}
|
|
|
|
/// <inheritdoc/>
|
|
public bool CancelInstallation(Guid id)
|
|
{
|
|
lock (_currentInstallationsLock)
|
|
{
|
|
var install = _currentInstallations.Find(x => x.Info.Id.Equals(id));
|
|
if (install == default((InstallationInfo, CancellationTokenSource)))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
install.Token.Cancel();
|
|
_currentInstallations.Remove(install);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public void Dispose()
|
|
{
|
|
Dispose(true);
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Releases unmanaged and optionally managed resources.
|
|
/// </summary>
|
|
/// <param name="dispose"><c>true</c> to release both managed and unmanaged resources or <c>false</c> to release only unmanaged resources.</param>
|
|
protected virtual void Dispose(bool dispose)
|
|
{
|
|
if (dispose)
|
|
{
|
|
lock (_currentInstallationsLock)
|
|
{
|
|
foreach (var (info, token) in _currentInstallations)
|
|
{
|
|
token.Dispose();
|
|
}
|
|
|
|
_currentInstallations.Clear();
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Merges two sorted lists.
|
|
/// </summary>
|
|
/// <param name="source">The source <see cref="IList{VersionInfo}"/> instance to merge.</param>
|
|
/// <param name="dest">The destination <see cref="IList{VersionInfo}"/> instance to merge with.</param>
|
|
private static void MergeSortedList(IList<VersionInfo> source, IList<VersionInfo> dest)
|
|
{
|
|
int sLength = source.Count - 1;
|
|
int dLength = dest.Count;
|
|
int s = 0, d = 0;
|
|
var sourceVersion = source[0].VersionNumber;
|
|
var destVersion = dest[0].VersionNumber;
|
|
|
|
while (d < dLength)
|
|
{
|
|
if (sourceVersion.CompareTo(destVersion) >= 0)
|
|
{
|
|
if (s < sLength)
|
|
{
|
|
sourceVersion = source[++s].VersionNumber;
|
|
}
|
|
else
|
|
{
|
|
// Append all of destination to the end of source.
|
|
while (d < dLength)
|
|
{
|
|
source.Add(dest[d++]);
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
source.Insert(s++, dest[d++]);
|
|
if (d >= dLength)
|
|
{
|
|
break;
|
|
}
|
|
|
|
sLength++;
|
|
destVersion = dest[d].VersionNumber;
|
|
}
|
|
}
|
|
}
|
|
|
|
private IEnumerable<InstallationInfo> GetAvailablePluginUpdates(IReadOnlyList<PackageInfo> pluginCatalog)
|
|
{
|
|
var plugins = _pluginManager.Plugins;
|
|
foreach (var plugin in plugins)
|
|
{
|
|
// Don't auto update when plugin marked not to, or when it's disabled.
|
|
if (plugin.Manifest?.AutoUpdate == false || plugin.Manifest?.Status == PluginStatus.Disabled)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
var compatibleVersions = GetCompatibleVersions(pluginCatalog, plugin.Name, plugin.Id, minVersion: plugin.Version);
|
|
var version = compatibleVersions.FirstOrDefault(y => y.Version > plugin.Version);
|
|
|
|
if (version is not null && CompletedInstallations.All(x => !x.Id.Equals(version.Id)))
|
|
{
|
|
yield return version;
|
|
}
|
|
}
|
|
}
|
|
|
|
private async Task PerformPackageInstallation(InstallationInfo package, PluginStatus status, CancellationToken cancellationToken)
|
|
{
|
|
if (!Path.GetExtension(package.SourceUrl.AsSpan()).Equals(".zip", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
_logger.LogError("Only zip packages are supported. {SourceUrl} is not a zip archive.", package.SourceUrl);
|
|
return;
|
|
}
|
|
|
|
// Always override the passed-in target (which is a file) and figure it out again
|
|
string targetDir = Path.Combine(_appPaths.PluginsPath, package.Name);
|
|
|
|
using var response = await _httpClientFactory.CreateClient(NamedClient.Default)
|
|
.GetAsync(new Uri(package.SourceUrl), cancellationToken).ConfigureAwait(false);
|
|
response.EnsureSuccessStatusCode();
|
|
Stream stream = await response.Content.ReadAsStreamAsync(cancellationToken).ConfigureAwait(false);
|
|
await using (stream.ConfigureAwait(false))
|
|
{
|
|
// CA5351: Do Not Use Broken Cryptographic Algorithms
|
|
#pragma warning disable CA5351
|
|
cancellationToken.ThrowIfCancellationRequested();
|
|
|
|
var hash = Convert.ToHexString(await MD5.HashDataAsync(stream, cancellationToken).ConfigureAwait(false));
|
|
if (!string.Equals(package.Checksum, hash, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
_logger.LogError(
|
|
"The checksums didn't match while installing {Package}, expected: {Expected}, got: {Received}",
|
|
package.Name,
|
|
package.Checksum,
|
|
hash);
|
|
throw new InvalidDataException("The checksum of the received data doesn't match.");
|
|
}
|
|
|
|
// Version folder as they cannot be overwritten in Windows.
|
|
targetDir += "_" + package.Version;
|
|
|
|
if (Directory.Exists(targetDir))
|
|
{
|
|
try
|
|
{
|
|
Directory.Delete(targetDir, true);
|
|
}
|
|
#pragma warning disable CA1031 // Do not catch general exception types
|
|
catch
|
|
#pragma warning restore CA1031 // Do not catch general exception types
|
|
{
|
|
// Ignore any exceptions.
|
|
}
|
|
}
|
|
|
|
stream.Position = 0;
|
|
await ZipFile.ExtractToDirectoryAsync(stream, targetDir, true, cancellationToken).ConfigureAwait(false);
|
|
}
|
|
|
|
// Ensure we create one or populate existing ones with missing data.
|
|
await _pluginManager.PopulateManifest(package.PackageInfo, package.Version, targetDir, status).ConfigureAwait(false);
|
|
|
|
_pluginManager.ImportPluginFrom(targetDir);
|
|
}
|
|
|
|
private async Task<bool> InstallPackageInternal(InstallationInfo package, CancellationToken cancellationToken)
|
|
{
|
|
LocalPlugin? plugin = _pluginManager.Plugins.FirstOrDefault(p => p.Id.Equals(package.Id) && p.Version.Equals(package.Version))
|
|
?? _pluginManager.Plugins.FirstOrDefault(p => p.Name.Equals(package.Name, StringComparison.OrdinalIgnoreCase) && p.Version.Equals(package.Version));
|
|
|
|
await PerformPackageInstallation(package, plugin?.Manifest.Status ?? PluginStatus.Active, cancellationToken).ConfigureAwait(false);
|
|
_logger.LogInformation("Plugin {Action}: {PluginName} {PluginVersion}", plugin is null ? "installed" : "updated", package.Name, package.Version);
|
|
|
|
return plugin is not null;
|
|
}
|
|
}
|
|
}
|