Don't set status code if response is closed

5 files changed, 18 insertions, 37 deletions

diff --git a/Emby.Server.Implementations/HttpServer/HttpListenerHost.cs b/Emby.Server.Implementations/HttpServer/HttpListenerHost.cs
index d1b1b5b24..eab755cef 100644
--- a/Emby.Server.Implementations/HttpServer/HttpListenerHost.cs
+++ b/Emby.Server.Implementations/HttpServer/HttpListenerHost.cs
@@ -225,7 +225,7 @@ namespace Emby.Server.Implementations.HttpServer
 
                 var httpRes = httpReq.Response;
 
-                if (httpRes.IsClosed)
+                if (httpRes.OriginalResponse.HasStarted)
                 {
                     return;
                 }
@@ -595,8 +595,6 @@ namespace Emby.Server.Implementations.HttpServer
             }
             finally
             {
-                // TODO response closes automatically after the handler is done, but some functions rely on knowing if it's closed or not
-                httpRes.IsClosed = true;
                 stopWatch.Stop();
                 var elapsed = stopWatch.Elapsed;
                 if (elapsed.TotalMilliseconds > 500)
diff --git a/Emby.Server.Implementations/Services/ServiceExec.cs b/Emby.Server.Implementations/Services/ServiceExec.cs
index 79f5c59e6..38952628d 100644
--- a/Emby.Server.Implementations/Services/ServiceExec.cs
+++ b/Emby.Server.Implementations/Services/ServiceExec.cs
@@ -78,7 +78,7 @@ namespace Emby.Server.Implementations.Services
                     foreach (var requestFilter in actionContext.RequestFilters)
                     {
                         requestFilter.RequestFilter(request, request.Response, requestDto);
-                        if (request.Response.IsClosed)
+                        if (request.Response.OriginalResponse.HasStarted)
                         {
                             Task.FromResult<object>(null);
                         }
diff --git a/Emby.Server.Implementations/SocketSharp/WebSocketSharpListener.cs b/Emby.Server.Implementations/SocketSharp/WebSocketSharpListener.cs
index 2df826957..dd313b336 100644
--- a/Emby.Server.Implementations/SocketSharp/WebSocketSharpListener.cs
+++ b/Emby.Server.Implementations/SocketSharp/WebSocketSharpListener.cs
@@ -90,7 +90,10 @@ namespace Emby.Server.Implementations.SocketSharp
             catch (Exception ex)
             {
                 _logger.LogError(ex, "AcceptWebSocketAsync error");
-                ctx.Response.StatusCode = 500;
+                if (!ctx.Response.HasStarted)
+                {
+                    ctx.Response.StatusCode = 500;
+                }
             }
         }
 
diff --git a/Emby.Server.Implementations/SocketSharp/WebSocketSharpRequest.cs b/Emby.Server.Implementations/SocketSharp/WebSocketSharpRequest.cs
index bc002dc4c..2d3ec3c8e 100644
--- a/Emby.Server.Implementations/SocketSharp/WebSocketSharpRequest.cs
+++ b/Emby.Server.Implementations/SocketSharp/WebSocketSharpRequest.cs
@@ -20,22 +20,19 @@ namespace Emby.Server.Implementations.SocketSharp
     public partial class WebSocketSharpRequest : IHttpRequest
     {
         private readonly HttpRequest request;
-        private readonly IResponse response;
 
         public WebSocketSharpRequest(HttpRequest httpContext, HttpResponse response, string operationName, ILogger logger)
         {
             this.OperationName = operationName;
             this.request = httpContext;
-            this.response = new WebSocketSharpResponse(logger, response, this);
+            this.Response = new WebSocketSharpResponse(logger, response);
 
             // HandlerFactoryPath = GetHandlerPathIfAny(UrlPrefixes[0]);
         }
 
         public HttpRequest HttpRequest => request;
 
-        public IResponse Response => response;
-
-        public IResponse HttpResponse => response;
+        public IResponse Response { get; }
 
         public string OperationName { get; set; }
 
diff --git a/Emby.Server.Implementations/SocketSharp/WebSocketSharpResponse.cs b/Emby.Server.Implementations/SocketSharp/WebSocketSharpResponse.cs
index a7e3e6c70..0f67eaa62 100644
--- a/Emby.Server.Implementations/SocketSharp/WebSocketSharpResponse.cs
+++ b/Emby.Server.Implementations/SocketSharp/WebSocketSharpResponse.cs
@@ -16,38 +16,28 @@ namespace Emby.Server.Implementations.SocketSharp
     {
         private readonly ILogger _logger;
 
-        private readonly HttpResponse _response;
-
-        public WebSocketSharpResponse(ILogger logger, HttpResponse response, IRequest request)
+        public WebSocketSharpResponse(ILogger logger, HttpResponse response)
         {
             _logger = logger;
-            this._response = response;
-            Items = new Dictionary<string, object>();
-            Request = request;
+            OriginalResponse = response;
         }
 
-        public IRequest Request { get; private set; }
-
-        public Dictionary<string, object> Items { get; private set; }
-
-        public object OriginalResponse => _response;
+        public HttpResponse OriginalResponse { get; }
 
         public int StatusCode
         {
-            get => this._response.StatusCode;
-            set => this._response.StatusCode = value;
+            get => OriginalResponse.StatusCode;
+            set => OriginalResponse.StatusCode = value;
         }
 
         public string StatusDescription { get; set; }
 
         public string ContentType
         {
-            get => _response.ContentType;
-            set => _response.ContentType = value;
+            get => OriginalResponse.ContentType;
+            set => OriginalResponse.ContentType = value;
         }
 
-        public IHeaderDictionary Headers => _response.Headers;
-
         public void AddHeader(string name, string value)
         {
             if (string.Equals(name, "Content-Type", StringComparison.OrdinalIgnoreCase))
@@ -56,22 +46,15 @@ namespace Emby.Server.Implementations.SocketSharp
                 return;
             }
 
-            _response.Headers.Add(name, value);
-        }
-
-        public string GetHeader(string name)
-        {
-            return _response.Headers[name];
+            OriginalResponse.Headers.Add(name, value);
         }
 
         public void Redirect(string url)
         {
-            _response.Redirect(url);
+            OriginalResponse.Redirect(url);
         }
 
-        public Stream OutputStream => _response.Body;
-
-        public bool IsClosed { get; set; }
+        public Stream OutputStream => OriginalResponse.Body;
 
         public bool SendChunked { get; set; }