334 lines
14 KiB
C#
334 lines
14 KiB
C#
using AcDream.Launcher.Core.Updates;
|
|
|
|
namespace AcDream.Launcher.Core.Tests.Updates;
|
|
|
|
public sealed class LauncherSelfUpdateManagerTests : IDisposable
|
|
{
|
|
private readonly string _root = Path.Combine(
|
|
Path.GetTempPath(),
|
|
"acdream-self-update-tests",
|
|
"target & $(literal)-" + Guid.NewGuid().ToString("N"));
|
|
|
|
public void Dispose()
|
|
{
|
|
if (Directory.Exists(_root))
|
|
{
|
|
Directory.Delete(_root, recursive: true);
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task StartupWithoutPlanReclaimsOnlyExactOwnedResidue()
|
|
{
|
|
using var harness = new Harness(_root);
|
|
string orphan = harness.Manager.GetTransactionDirectory(Guid.NewGuid().ToString("N"));
|
|
Directory.CreateDirectory(orphan);
|
|
await File.WriteAllTextAsync(Path.Combine(orphan, "partial"), "partial");
|
|
Directory.CreateDirectory(harness.Manager.RootDirectory);
|
|
string temporary = Path.Combine(
|
|
harness.Manager.RootDirectory,
|
|
$".pending.json.{Guid.NewGuid():N}.tmp");
|
|
string unrelated = Path.Combine(harness.Manager.RootDirectory, "pending.user.tmp");
|
|
await File.WriteAllTextAsync(temporary, "partial");
|
|
await File.WriteAllTextAsync(unrelated, "preserve");
|
|
|
|
Assert.Null(await harness.Manager.LoadPendingAsync());
|
|
using UpdateSessionBarrier.ExclusiveLease lease =
|
|
harness.Manager.Barrier.AcquireExclusive();
|
|
Assert.True(harness.Manager.CleanupOwnedResidueUnderLease(
|
|
pending: null,
|
|
harness.Target,
|
|
lease));
|
|
|
|
Assert.False(Directory.Exists(orphan));
|
|
Assert.False(File.Exists(temporary));
|
|
Assert.True(File.Exists(unrelated));
|
|
}
|
|
|
|
[Fact]
|
|
public async Task VerifiedStageIsDurableAndDoesNotTouchRunningTarget()
|
|
{
|
|
using var harness = new Harness(_root);
|
|
|
|
SelfUpdateStageResult result = await harness.StageAsync();
|
|
SelfUpdatePlan plan = Assert.IsType<SelfUpdatePlan>(
|
|
await harness.Manager.LoadPendingAsync());
|
|
|
|
Assert.Equal("2.0.0", result.Version.Value);
|
|
Assert.Equal(SelfUpdatePlanState.Staged, plan.State);
|
|
Assert.Null(plan.Apply);
|
|
Assert.Equal("old-launcher", await File.ReadAllTextAsync(harness.LauncherPath));
|
|
Assert.Equal("new-launcher", await File.ReadAllTextAsync(
|
|
Path.Combine(harness.Manager.GetPayloadDirectory(plan.TransactionId), harness.LauncherName)));
|
|
string pendingJson = await File.ReadAllTextAsync(result.PendingPlanPath);
|
|
Assert.Contains("\"state\": \"staged\"", pendingJson, StringComparison.Ordinal);
|
|
Assert.DoesNotContain("\"state\": \"Staged\"", pendingJson, StringComparison.Ordinal);
|
|
Assert.DoesNotContain("cmd", pendingJson,
|
|
StringComparison.OrdinalIgnoreCase);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ApplyConfirmAndCompletionUseMoveJournalAndRemoveTransaction()
|
|
{
|
|
using var harness = new Harness(_root);
|
|
_ = await harness.StageAsync();
|
|
|
|
SelfUpdatePlan applied = await harness.Manager.ApplyPendingAsync(harness.Target);
|
|
|
|
Assert.Equal(SelfUpdatePlanState.AwaitingConfirmation, applied.State);
|
|
Assert.NotNull(applied.Apply);
|
|
Assert.Equal("new-launcher", await File.ReadAllTextAsync(harness.LauncherPath));
|
|
Assert.Equal("support-new-launcher", await File.ReadAllTextAsync(harness.SupportPath));
|
|
await harness.Manager.ConfirmAsync(
|
|
applied.TransactionId,
|
|
harness.Target,
|
|
harness.LauncherPath);
|
|
Assert.True(harness.Manager.IsConfirmed(applied.TransactionId));
|
|
|
|
await harness.Manager.CompleteConfirmedAsync(applied.TransactionId, harness.Target);
|
|
|
|
Assert.False(File.Exists(harness.Manager.PendingPlanPath));
|
|
Assert.False(Directory.Exists(
|
|
harness.Manager.GetTransactionDirectory(applied.TransactionId)));
|
|
Assert.Equal("new-launcher", await File.ReadAllTextAsync(harness.LauncherPath));
|
|
}
|
|
|
|
[Fact]
|
|
public async Task AwaitingConfirmationRollbackRestoresEveryOldFileAndCanRetry()
|
|
{
|
|
using var harness = new Harness(_root);
|
|
_ = await harness.StageAsync();
|
|
SelfUpdatePlan applied = await harness.Manager.ApplyPendingAsync(harness.Target);
|
|
|
|
SelfUpdatePlan rolledBack = await harness.Manager
|
|
.RollbackAwaitingConfirmationAsync(harness.Target);
|
|
|
|
Assert.Equal(SelfUpdatePlanState.Staged, rolledBack.State);
|
|
Assert.Null(rolledBack.Apply);
|
|
Assert.Equal("old-launcher", await File.ReadAllTextAsync(harness.LauncherPath));
|
|
Assert.Equal("old-support", await File.ReadAllTextAsync(harness.SupportPath));
|
|
SelfUpdatePlan retried = await harness.Manager.ApplyPendingAsync(harness.Target);
|
|
Assert.Equal(SelfUpdatePlanState.AwaitingConfirmation, retried.State);
|
|
Assert.Equal("new-launcher", await File.ReadAllTextAsync(harness.LauncherPath));
|
|
}
|
|
|
|
[Fact]
|
|
public async Task PriorOwnershipRemovesObsoleteFilesAndRollbackRestoresThem()
|
|
{
|
|
using var harness = new Harness(_root);
|
|
byte[] firstArchive = UpdateTestData.CreateZip(
|
|
[
|
|
(harness.LauncherName, "launcher-v2"u8.ToArray(), 0x81ED),
|
|
("support.dat", "support-v2"u8.ToArray(), 0x81A4),
|
|
("obsolete.dll", "obsolete-v2"u8.ToArray(), 0x81A4),
|
|
]);
|
|
_ = await harness.StageAsync("2.0.0", firstArchive);
|
|
SelfUpdatePlan first = await harness.Manager.ApplyPendingAsync(harness.Target);
|
|
await harness.Manager.ConfirmAsync(
|
|
first.TransactionId,
|
|
harness.Target,
|
|
harness.LauncherPath);
|
|
await harness.Manager.CompleteConfirmedAsync(first.TransactionId, harness.Target);
|
|
string obsoletePath = Path.Combine(harness.Target, "obsolete.dll");
|
|
Assert.Equal("obsolete-v2", await File.ReadAllTextAsync(obsoletePath));
|
|
|
|
byte[] secondArchive = UpdateTestData.CreateZip(
|
|
[
|
|
(harness.LauncherName, "launcher-v3"u8.ToArray(), 0x81ED),
|
|
("support.dat", "support-v3"u8.ToArray(), 0x81A4),
|
|
]);
|
|
_ = await harness.StageAsync("3.0.0", secondArchive);
|
|
SelfUpdatePlan applied = await harness.Manager.ApplyPendingAsync(harness.Target);
|
|
|
|
Assert.False(File.Exists(obsoletePath));
|
|
Assert.Contains(
|
|
applied.Apply!,
|
|
entry => entry.Path == "obsolete.dll"
|
|
&& entry.Operation == SelfUpdateApplyOperation.Remove
|
|
&& entry.HadOriginal);
|
|
|
|
SelfUpdatePlan rolledBack = await harness.Manager
|
|
.RollbackAwaitingConfirmationAsync(harness.Target);
|
|
|
|
Assert.Equal(SelfUpdatePlanState.Staged, rolledBack.State);
|
|
Assert.Equal("launcher-v2", await File.ReadAllTextAsync(harness.LauncherPath));
|
|
Assert.Equal("support-v2", await File.ReadAllTextAsync(harness.SupportPath));
|
|
Assert.Equal("obsolete-v2", await File.ReadAllTextAsync(obsoletePath));
|
|
|
|
SelfUpdatePlan retried = await harness.Manager.ApplyPendingAsync(harness.Target);
|
|
await harness.Manager.ConfirmAsync(
|
|
retried.TransactionId,
|
|
harness.Target,
|
|
harness.LauncherPath);
|
|
await harness.Manager.CompleteConfirmedAsync(retried.TransactionId, harness.Target);
|
|
|
|
Assert.False(File.Exists(obsoletePath));
|
|
string ownership = await File.ReadAllTextAsync(Path.Combine(
|
|
harness.Target,
|
|
LauncherSelfUpdateManager.InstallRecordFileName));
|
|
Assert.DoesNotContain("obsolete.dll", ownership, StringComparison.Ordinal);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ApplyFailpointAfterCanonicalAtomicReplaceRollsBackToStagedState()
|
|
{
|
|
using var harness = new Harness(_root);
|
|
_ = await harness.StageAsync();
|
|
LauncherSelfUpdateManager faulting = harness.CreateManagerWithObserver(observation =>
|
|
{
|
|
if (observation.Boundary == SelfUpdateApplyBoundary.AfterTargetMutation
|
|
&& string.Equals(
|
|
observation.Path,
|
|
harness.LauncherName,
|
|
StringComparison.Ordinal))
|
|
{
|
|
Assert.True(File.Exists(harness.LauncherPath));
|
|
throw new InvalidOperationException("failpoint");
|
|
}
|
|
});
|
|
|
|
InvalidOperationException failure = await Assert.ThrowsAsync<InvalidOperationException>(
|
|
() => faulting.ApplyPendingAsync(harness.Target));
|
|
SelfUpdatePlan recovered = Assert.IsType<SelfUpdatePlan>(
|
|
await harness.Manager.LoadPendingAsync());
|
|
|
|
Assert.Equal("failpoint", failure.Message);
|
|
Assert.Equal(SelfUpdatePlanState.Staged, recovered.State);
|
|
Assert.Equal("old-launcher", await File.ReadAllTextAsync(harness.LauncherPath));
|
|
Assert.Equal("old-support", await File.ReadAllTextAsync(harness.SupportPath));
|
|
Assert.Equal("new-launcher", await File.ReadAllTextAsync(Path.Combine(
|
|
harness.Manager.GetPayloadDirectory(recovered.TransactionId),
|
|
harness.LauncherName)));
|
|
}
|
|
|
|
[Fact]
|
|
public async Task CorruptPayloadWrongTargetAndUnknownPlanFieldFailClosed()
|
|
{
|
|
using var harness = new Harness(_root);
|
|
_ = await harness.StageAsync();
|
|
SelfUpdatePlan staged = Assert.IsType<SelfUpdatePlan>(
|
|
await harness.Manager.LoadPendingAsync());
|
|
await File.WriteAllTextAsync(
|
|
Path.Combine(
|
|
harness.Manager.GetPayloadDirectory(staged.TransactionId),
|
|
harness.LauncherName),
|
|
"bad-payload!");
|
|
|
|
await Assert.ThrowsAsync<LauncherUpdateException>(() =>
|
|
harness.Manager.ApplyPendingAsync(harness.Target));
|
|
Assert.Equal("old-launcher", await File.ReadAllTextAsync(harness.LauncherPath));
|
|
string wrongTarget = Path.Combine(_root, "other-target");
|
|
Directory.CreateDirectory(wrongTarget);
|
|
await Assert.ThrowsAsync<LauncherUpdateException>(() =>
|
|
harness.Manager.ApplyPendingAsync(wrongTarget));
|
|
|
|
string json = await File.ReadAllTextAsync(harness.Manager.PendingPlanPath);
|
|
await File.WriteAllTextAsync(
|
|
harness.Manager.PendingPlanPath,
|
|
json.TrimEnd().TrimEnd('}') + ",\"unknown\":true}");
|
|
await Assert.ThrowsAsync<LauncherUpdateException>(() =>
|
|
harness.Manager.LoadPendingAsync());
|
|
}
|
|
|
|
[Fact]
|
|
public async Task BootstrapConfirmationAndOrdinaryStartupDoNotUseShellParsing()
|
|
{
|
|
using var harness = new Harness(_root);
|
|
SelfUpdateStartupResult ordinary = await LauncherSelfUpdateBootstrap.HandleAsync(
|
|
["--literal", "argument with spaces & metacharacters"],
|
|
harness.Manager,
|
|
harness.Target,
|
|
harness.LauncherPath);
|
|
Assert.False(ordinary.ShouldExit);
|
|
Assert.Equal(["--literal", "argument with spaces & metacharacters"],
|
|
ordinary.RemainingArguments);
|
|
|
|
_ = await harness.StageAsync();
|
|
SelfUpdatePlan applied = await harness.Manager.ApplyPendingAsync(harness.Target);
|
|
SelfUpdateStartupResult confirmation = await LauncherSelfUpdateBootstrap.HandleAsync(
|
|
[LauncherSelfUpdateBootstrap.ConfirmArgument, applied.TransactionId],
|
|
harness.Manager,
|
|
harness.Target,
|
|
harness.LauncherPath);
|
|
|
|
Assert.False(confirmation.ShouldExit);
|
|
Assert.Empty(confirmation.RemainingArguments);
|
|
Assert.False(File.Exists(harness.Manager.PendingPlanPath));
|
|
Assert.False(harness.Manager.IsConfirmed(applied.TransactionId));
|
|
}
|
|
|
|
private sealed class Harness : IDisposable
|
|
{
|
|
private readonly LocalHttpFixture _server = new();
|
|
private readonly HttpClient _http = new();
|
|
private readonly byte[] _archive;
|
|
private readonly ReleaseArtifact _artifact;
|
|
private readonly string _root;
|
|
|
|
public Harness(string root)
|
|
{
|
|
_root = root;
|
|
Target = Path.Combine(root, "published launcher");
|
|
Directory.CreateDirectory(Target);
|
|
Rid = LauncherRuntimeIdentity.DetectRid();
|
|
LauncherName = "acdream-launcher"
|
|
+ (Rid.StartsWith("win-", StringComparison.Ordinal) ? ".exe" : string.Empty);
|
|
LauncherPath = Path.Combine(Target, LauncherName);
|
|
SupportPath = Path.Combine(Target, "support.dat");
|
|
File.WriteAllText(LauncherPath, "old-launcher");
|
|
File.WriteAllText(SupportPath, "old-support");
|
|
_archive = UpdateTestData.LauncherZip(Rid, "new-launcher");
|
|
_server.Add("launcher.zip", _archive);
|
|
_artifact = new ReleaseArtifact(
|
|
_server.UriFor("launcher.zip"),
|
|
UpdateTestData.Sha256(_archive),
|
|
_archive.LongLength);
|
|
Manager = new LauncherSelfUpdateManager(UpdateTestData.Paths(root), _http);
|
|
}
|
|
|
|
public string Target { get; }
|
|
|
|
public string Rid { get; }
|
|
|
|
public string LauncherName { get; }
|
|
|
|
public string LauncherPath { get; }
|
|
|
|
public string SupportPath { get; }
|
|
|
|
public LauncherSelfUpdateManager Manager { get; }
|
|
|
|
public Task<SelfUpdateStageResult> StageAsync() => Manager.StageAsync(
|
|
LauncherVersion.Parse("2.0.0"),
|
|
Rid,
|
|
_artifact,
|
|
Target,
|
|
progress: null,
|
|
CancellationToken.None);
|
|
|
|
public Task<SelfUpdateStageResult> StageAsync(string version, byte[] archive)
|
|
{
|
|
_server.Add("launcher.zip", archive);
|
|
return Manager.StageAsync(
|
|
LauncherVersion.Parse(version),
|
|
Rid,
|
|
new ReleaseArtifact(
|
|
_server.UriFor("launcher.zip"),
|
|
UpdateTestData.Sha256(archive),
|
|
archive.LongLength),
|
|
Target,
|
|
progress: null,
|
|
CancellationToken.None);
|
|
}
|
|
|
|
public LauncherSelfUpdateManager CreateManagerWithObserver(
|
|
Action<SelfUpdateApplyObservation> observer) =>
|
|
new(UpdateTestData.Paths(_root), _http, null, observer);
|
|
|
|
public void Dispose()
|
|
{
|
|
_http.Dispose();
|
|
_server.Dispose();
|
|
}
|
|
}
|
|
}
|