1 files changed, 137 insertions, 0 deletions
diff --git a/tests/Jellyfin.Server.Implementations.Tests/SessionManager/SessionManagerTests.cs b/tests/Jellyfin.Server.Implementations.Tests/SessionManager/SessionManagerTests.cs
index a5a67046d1..f803c69af2 100644
--- a/tests/Jellyfin.Server.Implementations.Tests/SessionManager/SessionManagerTests.cs
+++ b/tests/Jellyfin.Server.Implementations.Tests/SessionManager/SessionManagerTests.cs
@@ -1,6 +1,9 @@
 using System;
+using System.Threading;
 using System.Threading.Tasks;
+using Jellyfin.Data;
 using Jellyfin.Database.Implementations.Entities;
+using Jellyfin.Database.Implementations.Enums;
 using MediaBrowser.Controller;
 using MediaBrowser.Controller.Configuration;
 using MediaBrowser.Controller.Devices;
@@ -8,7 +11,9 @@ using MediaBrowser.Controller.Drawing;
 using MediaBrowser.Controller.Dto;
 using MediaBrowser.Controller.Events;
 using MediaBrowser.Controller.Library;
+using MediaBrowser.Controller.Net;
 using MediaBrowser.Controller.Session;
+using MediaBrowser.Model.Session;
 using Microsoft.Extensions.Hosting;
 using Microsoft.Extensions.Logging.Abstractions;
 using Moq;
@@ -108,4 +113,136 @@ public class SessionManagerTests
 
         return data;
     }
+
+    [Fact]
+    public async Task SendMessageCommand_Should_ThrowSecurityException_WhenControllingAnotherUsersSession()
+    {
+        var victim = new User("victim", "default", "default");
+        var attacker = new User("attacker", "default", "default");
+        await using var sessionManager = CreateSessionManager(victim, attacker);
+
+        var victimSession = await LogSessionActivity(sessionManager, victim);
+        var attackerSession = await LogSessionActivity(sessionManager, attacker);
+
+        await Assert.ThrowsAsync<SecurityException>(() => sessionManager.SendMessageCommand(
+            attackerSession.Id,
+            victimSession.Id,
+            new MessageCommand { Header = "Custom Message", Text = "test exploit!" },
+            CancellationToken.None));
+    }
+
+    [Fact]
+    public async Task SendMessageCommand_Should_Succeed_WhenAllowedToControlOtherUsers()
+    {
+        var victim = new User("victim", "default", "default");
+        var attacker = new User("controller", "default", "default");
+        attacker.SetPermission(PermissionKind.EnableRemoteControlOfOtherUsers, true);
+        await using var sessionManager = CreateSessionManager(victim, attacker);
+
+        var victimSession = await LogSessionActivity(sessionManager, victim);
+        var controllingSession = await LogSessionActivity(sessionManager, attacker);
+
+        await sessionManager.SendMessageCommand(
+            controllingSession.Id,
+            victimSession.Id,
+            new MessageCommand { Header = "Custom Message", Text = "hello" },
+            CancellationToken.None);
+    }
+
+    [Fact]
+    public async Task LogSessionActivity_Should_NotReuseAnotherUsersSession()
+    {
+        var victim = new User("victim", "default", "default");
+        var attacker = new User("attacker", "default", "default");
+        await using var sessionManager = CreateSessionManager(victim, attacker);
+
+        // Client name and device id are attacker controlled, so they must not identify a session on their own.
+        var victimSession = await LogSessionActivity(sessionManager, victim);
+        var attackerSession = await LogSessionActivity(sessionManager, attacker);
+
+        Assert.NotEqual(victimSession.Id, attackerSession.Id);
+        Assert.Equal(victim.Id, victimSession.UserId);
+    }
+
+    [Fact]
+    public async Task AddAdditionalUser_Should_ThrowSecurityException_WhenAttachingAnotherUser()
+    {
+        var attacker = new User("attacker", "default", "default");
+        var victim = new User("victim", "default", "default");
+        await using var sessionManager = CreateSessionManager(victim, attacker);
+
+        var attackerSession = await LogSessionActivity(sessionManager, attacker);
+
+        Assert.Throws<SecurityException>(() => sessionManager.AddAdditionalUser(attackerSession.Id, attackerSession.Id, victim.Id));
+    }
+
+    [Fact]
+    public async Task AddAdditionalUser_Should_Succeed_WhenCallerIsAdministrator()
+    {
+        var admin = new User("admin", "default", "default");
+        admin.SetPermission(PermissionKind.IsAdministrator, true);
+        var guest = new User("guest", "default", "default");
+        await using var sessionManager = CreateSessionManager(admin, guest);
+
+        var adminSession = await LogSessionActivity(sessionManager, admin);
+
+        sessionManager.AddAdditionalUser(adminSession.Id, adminSession.Id, guest.Id);
+
+        Assert.Contains(adminSession.AdditionalUsers, i => i.UserId.Equals(guest.Id));
+    }
+
+    [Fact]
+    public async Task RemoveAdditionalUser_Should_ThrowSecurityException_WhenModifyingAnotherUsersSession()
+    {
+        var victim = new User("victim", "default", "default");
+        var attacker = new User("attacker", "default", "default");
+        await using var sessionManager = CreateSessionManager(victim, attacker);
+
+        var victimSession = await LogSessionActivity(sessionManager, victim);
+        var attackerSession = await LogSessionActivity(sessionManager, attacker);
+
+        Assert.Throws<SecurityException>(() => sessionManager.RemoveAdditionalUser(attackerSession.Id, victimSession.Id, attacker.Id));
+    }
+
+    [Fact]
+    public async Task ReportCapabilities_Should_ThrowSecurityException_WhenReportingForAnotherUsersSession()
+    {
+        var victim = new User("victim", "default", "default");
+        var attacker = new User("attacker", "default", "default");
+        await using var sessionManager = CreateSessionManager(victim, attacker);
+
+        var victimSession = await LogSessionActivity(sessionManager, victim);
+        var attackerSession = await LogSessionActivity(sessionManager, attacker);
+
+        Assert.Throws<SecurityException>(() => sessionManager.ReportCapabilities(attackerSession.Id, victimSession.Id, new ClientCapabilities()));
+    }
+
+    private static Emby.Server.Implementations.Session.SessionManager CreateSessionManager(params User[] users)
+    {
+        var userManager = new Mock<IUserManager>();
+        foreach (var user in users)
+        {
+            userManager.Setup(i => i.GetUserById(user.Id)).Returns(user);
+        }
+
+        return new Emby.Server.Implementations.Session.SessionManager(
+            NullLogger<Emby.Server.Implementations.Session.SessionManager>.Instance,
+            Mock.Of<IEventManager>(),
+            Mock.Of<IUserDataManager>(),
+            Mock.Of<IServerConfigurationManager>(),
+            Mock.Of<ILibraryManager>(),
+            userManager.Object,
+            Mock.Of<IMusicManager>(),
+            Mock.Of<IDtoService>(),
+            Mock.Of<IImageProcessor>(),
+            Mock.Of<IServerApplicationHost>(),
+            Mock.Of<IDeviceManager>(),
+            Mock.Of<IMediaSourceManager>(),
+            Mock.Of<IHostApplicationLifetime>());
+    }
+
+    // All sessions are logged with the same client and device id on purpose, those values are taken
+    // from the request headers and are not bound to the access token of the calling user.
+    private static Task<SessionInfo> LogSessionActivity(ISessionManager sessionManager, User user)
+        => sessionManager.LogSessionActivity("Jellyfin Web", "1.0.0", "victim-tv-01", "device_name", "127.0.0.1", user);
 }