fix issue with incorrect user detection when Invoking for User

src/Microsoft.AspNetCore.SignalR.Core/DefaultHubLifetimeManager.cs

@@ -14,7 +14,7 @@ public class DefaultHubLifetimeManager<THub> : HubLifetimeManager<THub>
     {
         private long _nextInvocationId = 0;
         private readonly HubConnectionList _connections = new HubConnectionList();
-
+        
         public override Task AddGroupAsync(string connectionId, string groupName)
         {
             if (connectionId == null)
@@ -138,17 +138,14 @@ public override Task InvokeGroupAsync(string groupName, string methodName, objec
 
         public override Task InvokeUserAsync(string userId, string methodName, object[] args)
         {
-            return InvokeAllWhere(methodName, args, connection =>
-            {
-                return string.Equals(connection.User.Identity.Name, userId, StringComparison.Ordinal);
-            });
+            return InvokeAllWhere(methodName, args, connection => 
+                string.Equals(connection.UserIdentifier, userId, StringComparison.Ordinal));
         }
 
         public override Task OnConnectedAsync(HubConnectionContext connection)
         {
             // Set the hub groups feature
             connection.Features.Set<IHubGroupsFeature>(new HubGroupsFeature());
-
             _connections.Add(connection);
             return Task.CompletedTask;
         }
@@ -166,7 +163,7 @@ private async Task WriteAsync(HubConnectionContext connection, HubMessage hubMes
                 if (connection.Output.TryWrite(hubMessage))
                 {
                     break;
-                }
+                } 
             }
         }
 
@@ -194,4 +191,4 @@ private class HubGroupsFeature : IHubGroupsFeature
             public HashSet<string> Groups { get; } = new HashSet<string>(StringComparer.OrdinalIgnoreCase);
         }
     }
-}
+}

src/Microsoft.AspNetCore.SignalR.Core/DefaultUserIdProvider.cs

@@ -0,0 +1,15 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+using System.Security.Claims;
+
+namespace Microsoft.AspNetCore.SignalR.Core
+{
+    public class DefaultUserIdProvider : IUserIdProvider
+    {
+        public string GetUserId(HubConnectionContext connection)
+        {
+            return connection.User.FindFirst(ClaimTypes.NameIdentifier)?.Value;
+        }
+    }
+}

src/Microsoft.AspNetCore.SignalR.Core/HubConnectionContext.cs

@@ -9,6 +9,7 @@
 using System.Threading.Tasks;
 using System.Threading.Tasks.Channels;
 using Microsoft.AspNetCore.Http.Features;
+using Microsoft.AspNetCore.SignalR.Core;
 using Microsoft.AspNetCore.SignalR.Features;
 using Microsoft.AspNetCore.SignalR.Internal;
 using Microsoft.AspNetCore.SignalR.Internal.Protocol;
@@ -25,12 +26,15 @@ public class HubConnectionContext
         private readonly ConnectionContext _connectionContext;
         private readonly CancellationTokenSource _connectionAbortedTokenSource = new CancellationTokenSource();
         private readonly TaskCompletionSource<object> _abortCompletedTcs = new TaskCompletionSource<object>();
+        private readonly IUserIdProvider _userIdProvider;
+        private string _userIdCache = null;
 
-        public HubConnectionContext(WritableChannel<HubMessage> output, ConnectionContext connectionContext)
+        public HubConnectionContext(WritableChannel<HubMessage> output, ConnectionContext connectionContext, IUserIdProvider userIdProvider)
         {
             _output = output;
             _connectionContext = connectionContext;
             ConnectionAbortedToken = _connectionAbortedTokenSource.Token;
+            _userIdProvider = userIdProvider;
         }
 
         private IHubFeature HubFeature => Features.Get<IHubFeature>();
@@ -67,6 +71,18 @@ public virtual void Abort()
             Task.Factory.StartNew(_abortedCallback, this);
         }
 
+        public string UserIdentifier
+        {
+            get
+            {
+                if (String.IsNullOrEmpty(_userIdCache))
+                {
+                    _userIdCache = _userIdProvider.GetUserId(this);
+                }
+                return _userIdCache;
+            }
+        }
+
         internal void Abort(Exception exception)
         {
             AbortException = ExceptionDispatchInfo.Capture(exception);

src/Microsoft.AspNetCore.SignalR.Core/HubEndPoint.cs

@@ -11,6 +11,7 @@
 using System.Threading.Tasks;
 using System.Threading.Tasks.Channels;
 using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.SignalR.Core;
 using Microsoft.AspNetCore.SignalR.Core.Internal;
 using Microsoft.AspNetCore.SignalR.Features;
 using Microsoft.AspNetCore.SignalR.Internal;
@@ -39,20 +40,23 @@ public class HubEndPoint<THub> : IInvocationBinder where THub : Hub
         private readonly IServiceScopeFactory _serviceScopeFactory;
         private readonly IHubProtocolResolver _protocolResolver;
         private readonly IOptions<HubOptions> _hubOptions;
+        private readonly IUserIdProvider _userIdProvider;
 
         public HubEndPoint(HubLifetimeManager<THub> lifetimeManager,
                            IHubProtocolResolver protocolResolver,
                            IHubContext<THub> hubContext,
                            IOptions<HubOptions> hubOptions,
                            ILogger<HubEndPoint<THub>> logger,
-                           IServiceScopeFactory serviceScopeFactory)
+                           IServiceScopeFactory serviceScopeFactory, 
+                           IUserIdProvider userIdProvider)
         {
             _protocolResolver = protocolResolver;
             _lifetimeManager = lifetimeManager;
             _hubContext = hubContext;
             _hubOptions = hubOptions;
             _logger = logger;
             _serviceScopeFactory = serviceScopeFactory;
+            _userIdProvider = userIdProvider;
 
             DiscoverHubMethods();
         }
@@ -65,7 +69,7 @@ public async Task OnConnectedAsync(ConnectionContext connection)
             // all the relevant state for a SignalR Hub connection.
             connection.Features.Set<IHubFeature>(new HubFeature());
 
-            var connectionContext = new HubConnectionContext(output, connection);
+            var connectionContext = new HubConnectionContext(output, connection, _userIdProvider);
 
             if (!await ProcessNegotiate(connectionContext))
             {

src/Microsoft.AspNetCore.SignalR.Core/IUserIdProvider.cs

@@ -0,0 +1,10 @@
+// Copyright (c) .NET Foundation. All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+
+namespace Microsoft.AspNetCore.SignalR.Core
+{
+    public interface IUserIdProvider
+    {
+        string GetUserId(HubConnectionContext connection);
+    }
+}

src/Microsoft.AspNetCore.SignalR.Core/SignalRDependencyInjectionExtensions.cs

@@ -2,6 +2,7 @@
 // Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
 
 using Microsoft.AspNetCore.SignalR;
+using Microsoft.AspNetCore.SignalR.Core;
 using Microsoft.AspNetCore.SignalR.Internal;
 
 namespace Microsoft.Extensions.DependencyInjection
@@ -15,6 +16,7 @@ public static ISignalRBuilder AddSignalRCore(this IServiceCollection services)
             services.AddSingleton(typeof(IHubContext<>), typeof(HubContext<>));
             services.AddSingleton(typeof(IHubContext<,>), typeof(HubContext<,>));
             services.AddSingleton(typeof(HubEndPoint<>), typeof(HubEndPoint<>));
+            services.AddSingleton(typeof(IUserIdProvider), typeof(DefaultUserIdProvider));
             services.AddScoped(typeof(IHubActivator<>), typeof(DefaultHubActivator<>));
 
             services.AddAuthorization();

src/Microsoft.AspNetCore.SignalR.Redis/RedisHubLifetimeManager.cs

@@ -245,9 +245,9 @@ public override Task OnConnectedAsync(HubConnectionContext connection)
                 previousConnectionTask = WriteAsync(connection, message);
             });
 
-            if (connection.User.Identity.IsAuthenticated)
+            if (!String.IsNullOrEmpty(connection.UserIdentifier))
             {
-                var userChannel = _channelNamePrefix + ".user." + connection.User.Identity.Name;
+                var userChannel = _channelNamePrefix + ".user." + connection.UserIdentifier;
                 redisSubscriptions.Add(userChannel);
 
                 var previousUserTask = Task.CompletedTask;

test/Microsoft.AspNetCore.SignalR.Tests/DefaultHubLifetimeManagerTests.cs

@@ -1,5 +1,6 @@
 using System.Threading.Tasks;
 using System.Threading.Tasks.Channels;
+using Microsoft.AspNetCore.SignalR.Core;
 using Microsoft.AspNetCore.SignalR.Internal.Protocol;
 using Xunit;
 
@@ -17,8 +18,8 @@ public async Task InvokeAllAsyncWritesToAllConnectionsOutput()
                 var output2 = Channel.CreateUnbounded<HubMessage>();
 
                 var manager = new DefaultHubLifetimeManager<MyHub>();
-                var connection1 = new HubConnectionContext(output1, client1.Connection);
-                var connection2 = new HubConnectionContext(output2, client2.Connection);
+                var connection1 = new HubConnectionContext(output1, client1.Connection, new DefaultUserIdProvider());
+                var connection2 = new HubConnectionContext(output2, client2.Connection, new DefaultUserIdProvider());
 
                 await manager.OnConnectedAsync(connection1);
                 await manager.OnConnectedAsync(connection2);
@@ -51,8 +52,8 @@ public async Task InvokeAllAsyncDoesNotWriteToDisconnectedConnectionsOutput()
                 var output2 = Channel.CreateUnbounded<HubMessage>();
 
                 var manager = new DefaultHubLifetimeManager<MyHub>();
-                var connection1 = new HubConnectionContext(output1, client1.Connection);
-                var connection2 = new HubConnectionContext(output2, client2.Connection);
+                var connection1 = new HubConnectionContext(output1, client1.Connection, new DefaultUserIdProvider());
+                var connection2 = new HubConnectionContext(output2, client2.Connection, new DefaultUserIdProvider());
 
                 await manager.OnConnectedAsync(connection1);
                 await manager.OnConnectedAsync(connection2);
@@ -82,8 +83,8 @@ public async Task InvokeGroupAsyncWritesToAllConnectionsInGroupOutput()
                 var output2 = Channel.CreateUnbounded<HubMessage>();
 
                 var manager = new DefaultHubLifetimeManager<MyHub>();
-                var connection1 = new HubConnectionContext(output1, client1.Connection);
-                var connection2 = new HubConnectionContext(output2, client2.Connection);
+                var connection1 = new HubConnectionContext(output1, client1.Connection, new DefaultUserIdProvider());
+                var connection2 = new HubConnectionContext(output2, client2.Connection, new DefaultUserIdProvider());
 
                 await manager.OnConnectedAsync(connection1);
                 await manager.OnConnectedAsync(connection2);
@@ -110,7 +111,7 @@ public async Task InvokeConnectionAsyncWritesToConnectionOutput()
             {
                 var output = Channel.CreateUnbounded<HubMessage>();
                 var manager = new DefaultHubLifetimeManager<MyHub>();
-                var connection = new HubConnectionContext(output, client.Connection);
+                var connection = new HubConnectionContext(output, client.Connection, new DefaultUserIdProvider());
 
                 await manager.OnConnectedAsync(connection);
 

test/Microsoft.AspNetCore.SignalR.Tests/HubEndpointTests.cs

@@ -850,15 +850,15 @@ public async Task HubsCanSendToUser(Type hubType)
 
             dynamic endPoint = serviceProvider.GetService(GetEndPointType(hubType));
 
-            using (var firstClient = new TestClient())
-            using (var secondClient = new TestClient())
+            using (var firstClient = new TestClient(addClaimId: true))
+            using (var secondClient = new TestClient(addClaimId: true))
             {
                 Task firstEndPointTask = endPoint.OnConnectedAsync(firstClient.Connection);
                 Task secondEndPointTask = endPoint.OnConnectedAsync(secondClient.Connection);
 
                 await Task.WhenAll(firstClient.Connected, secondClient.Connected).OrTimeout();
 
-                await firstClient.SendInvocationAsync("ClientSendMethod", secondClient.Connection.User.Identity.Name, "test").OrTimeout();
+                await firstClient.SendInvocationAsync("ClientSendMethod", secondClient.Connection.User.FindFirst(ClaimTypes.NameIdentifier)?.Value, "test").OrTimeout();
 
                 // check that 'secondConnection' has received the group send
                 var hubMessage = await secondClient.ReadAsync().OrTimeout();

test/Microsoft.AspNetCore.SignalR.Tests/TestClient.cs

@@ -28,7 +28,7 @@ public class TestClient : IDisposable, IInvocationBinder
         public Channel<byte[]> Application { get; }
         public Task Connected => ((TaskCompletionSource<bool>)Connection.Metadata["ConnectedTask"]).Task;
 
-        public TestClient(bool synchronousCallbacks = false, IHubProtocol protocol = null)
+        public TestClient(bool synchronousCallbacks = false, IHubProtocol protocol = null, bool addClaimId = false)
         {
             var options = new ChannelOptimizations { AllowSynchronousContinuations = synchronousCallbacks };
             var transportToApplication = Channel.CreateUnbounded<byte[]>(options);
@@ -38,7 +38,15 @@ public TestClient(bool synchronousCallbacks = false, IHubProtocol protocol = nul
             _transport = ChannelConnection.Create<byte[]>(input: transportToApplication, output: applicationToTransport);
 
             Connection = new DefaultConnectionContext(Guid.NewGuid().ToString(), _transport, Application);
-            Connection.User = new ClaimsPrincipal(new ClaimsIdentity(new[] { new Claim(ClaimTypes.Name, Interlocked.Increment(ref _id).ToString()) }));
+
+            var claimValue = Interlocked.Increment(ref _id).ToString();
+            var claims = new List<Claim>{ new Claim(ClaimTypes.Name, claimValue) };
+            if (addClaimId)
+            {
+                claims.Add(new Claim(ClaimTypes.NameIdentifier, claimValue));
+            }
+
+            Connection.User = new ClaimsPrincipal(new ClaimsIdentity(claims));
             Connection.Metadata["ConnectedTask"] = new TaskCompletionSource<bool>();
 
             protocol = protocol ?? new JsonHubProtocol();
@@ -182,4 +190,4 @@ Type IInvocationBinder.GetReturnType(string invocationId)
             return typeof(object);
         }
     }
-}
+}