Do not serve response body for HEAD requests

Author: jbagga

src/Microsoft.AspNetCore.Mvc.Core/Infrastructure/FileResultExecutorBase.cs

@@ -295,6 +295,7 @@ private static PreconditionState GetMaxPreconditionState(params PreconditionStat
         {
             var response = context.HttpContext.Response;
             var httpResponseHeaders = response.GetTypedHeaders();
+            var serveBody = !HttpMethods.IsHead(context.HttpContext.Request.Method);
 
             // Range may be null for empty range header, invalid ranges, parsing errors, multiple ranges 
             // and when the file length is zero.
@@ -306,7 +307,7 @@ private static PreconditionState GetMaxPreconditionState(params PreconditionStat
 
             if (!isRangeRequest)
             {
-                return (range: null, rangeLength: 0, serveBody: true);
+                return (range: null, rangeLength: 0, serveBody);
             }
 
             // Requested range is not satisfiable
@@ -330,7 +331,7 @@ private static PreconditionState GetMaxPreconditionState(params PreconditionStat
             // Overwrite the Content-Length header for valid range requests with the range length.
             var rangeLength = SetContentLength(response, range);
 
-            return (range, rangeLength, serveBody: true);
+            return (range, rangeLength, serveBody);
         }
 
         private static long SetContentLength(HttpResponse response, RangeItemHeaderValue range)

test/Microsoft.AspNetCore.Mvc.Core.Test/FileResultTest.cs

@@ -7,13 +7,11 @@
 using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Mvc.Abstractions;
 using Microsoft.AspNetCore.Mvc.Infrastructure;
-using Microsoft.AspNetCore.Mvc.Internal;
 using Microsoft.AspNetCore.Routing;
 using Microsoft.AspNetCore.Testing.xunit;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Logging.Abstractions;
-using Microsoft.Extensions.Logging.Testing;
 using Microsoft.Net.Http.Headers;
 using Moq;
 using Xunit;

test/Microsoft.AspNetCore.Mvc.FunctionalTests/FileResultTests.cs

@@ -1,12 +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.
 
-using System;
 using System.Net;
 using System.Net.Http;
 using System.Net.Http.Headers;
 using System.Threading.Tasks;
-using Microsoft.AspNetCore.Mvc.Infrastructure;
 using Microsoft.AspNetCore.Testing.xunit;
 using Xunit;
 
@@ -189,11 +187,13 @@ public async Task FileFromDisk_ReturnsFileWithFileName()
             Assert.Equal("attachment; filename=downloadName.txt; filename*=UTF-8''downloadName.txt", contentDisposition);
         }
 
-        [Fact]
-        public async Task FileFromDisk_ReturnsFileWithFileName_RangeProcessingNotEnabled_RangeRequestedIgnored()
+        [Theory]
+        [InlineData("GET", "This is a sample text file")]
+        [InlineData("HEAD", "")]
+        public async Task FileFromDisk_ReturnsFileWithFileName_RangeProcessingNotEnabled_RangeRequestedIgnored(string httpMethod, string expectedBody)
         {
             // Arrange
-            var httpRequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://localhost/DownloadFiles/DownloadFromDiskWithFileName");
+            var httpRequestMessage = new HttpRequestMessage(new HttpMethod(httpMethod), "http://localhost/DownloadFiles/DownloadFromDiskWithFileName");
             httpRequestMessage.Headers.Range = new RangeHeaderValue(0, 6);
 
             // Act
@@ -204,7 +204,7 @@ public async Task FileFromDisk_ReturnsFileWithFileName_RangeProcessingNotEnabled
             Assert.NotNull(response.Content.Headers.ContentType);
             Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
             var body = await response.Content.ReadAsStringAsync();
-            Assert.Equal("This is a sample text file", body);
+            Assert.Equal(expectedBody, body);
         }
 
         [Fact]
@@ -246,6 +246,41 @@ public async Task FileFromDisk_ReturnsFileWithFileName_IfRangeHeaderInvalid_Rang
             Assert.Equal("This is a sample text file", body);
         }
 
+        [Theory]
+        [InlineData("", HttpStatusCode.OK)]
+        [InlineData("bytes = 0-6", HttpStatusCode.PartialContent)]
+        [InlineData("bytes = 17-25", HttpStatusCode.PartialContent)]
+        [InlineData("bytes = 0-50", HttpStatusCode.PartialContent)]
+        [InlineData("0-6", HttpStatusCode.OK)]
+        [InlineData("bytes = ", HttpStatusCode.OK)]
+        [InlineData("bytes = 1-4, 5-11", HttpStatusCode.OK)]
+        [InlineData("bytes = 35-36", HttpStatusCode.RequestedRangeNotSatisfiable)]
+        [InlineData("bytes = -0", HttpStatusCode.RequestedRangeNotSatisfiable)]
+        public async Task FileFromDisk_ReturnsFileWithFileName_DoesNotServeBody_ForHeadRequest_WithLastModifiedAndEtag(string rangeString, HttpStatusCode httpStatusCode)
+        {
+            // Arrange
+            var httpRequestMessage = new HttpRequestMessage(HttpMethod.Head, "http://localhost/DownloadFiles/DownloadFromDiskWithFileName_WithLastModifiedAndEtag");
+            httpRequestMessage.Headers.TryAddWithoutValidation("Range", rangeString);
+            httpRequestMessage.Headers.IfRange = new RangeConditionHeaderValue(new EntityTagHeaderValue("\"Etag\""));
+
+            // Act
+            var response = await Client.SendAsync(httpRequestMessage);
+
+            // Assert
+            Assert.Equal(httpStatusCode, response.StatusCode);
+
+            Assert.NotNull(response.Content.Headers.ContentType);
+            Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
+
+            var body = await response.Content.ReadAsStringAsync();
+            Assert.NotNull(body);
+            Assert.Equal(string.Empty, body);
+
+            var contentDisposition = response.Content.Headers.ContentDisposition.ToString();
+            Assert.NotNull(contentDisposition);
+            Assert.Equal("attachment; filename=downloadName.txt; filename*=UTF-8''downloadName.txt", contentDisposition);
+        }
+
         [Fact]
         public async Task FileFromStream_ReturnsFile()
         {
@@ -347,11 +382,13 @@ public async Task FileFromStream_ReturnsFileWithFileName()
             Assert.Equal("attachment; filename=downloadName.txt; filename*=UTF-8''downloadName.txt", contentDisposition);
         }
 
-        [Fact]
-        public async Task FileFromStream_ReturnsFileWithFileName_RangeProcessingNotEnabled_RangeRequestedIgnored()
+        [Theory]
+        [InlineData("GET", "This is sample text from a stream")]
+        [InlineData("HEAD", "")]
+        public async Task FileFromStream_ReturnsFileWithFileName_RangeProcessingNotEnabled_RangeRequestedIgnored(string httpMethod, string expectedBody)
         {
             // Arrange
-            var httpRequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://localhost/DownloadFiles/DownloadFromStreamWithFileName");
+            var httpRequestMessage = new HttpRequestMessage(new HttpMethod(httpMethod), "http://localhost/DownloadFiles/DownloadFromStreamWithFileName");
             httpRequestMessage.Headers.Range = new RangeHeaderValue(0, 6);
 
             // Act
@@ -362,7 +399,7 @@ public async Task FileFromStream_ReturnsFileWithFileName_RangeProcessingNotEnabl
             Assert.NotNull(response.Content.Headers.ContentType);
             Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
             var body = await response.Content.ReadAsStringAsync();
-            Assert.Equal("This is sample text from a stream", body);
+            Assert.Equal(expectedBody, body);
         }
 
         [Fact]
@@ -402,6 +439,41 @@ public async Task FileFromStream_ReturnsFileWithFileName_IfRangeHeaderInvalid_Ra
             Assert.Equal("This is sample text from a stream", body);
         }
 
+        [Theory]
+        [InlineData("", HttpStatusCode.OK)]
+        [InlineData("bytes = 0-6", HttpStatusCode.PartialContent)]
+        [InlineData("bytes = 17-25", HttpStatusCode.PartialContent)]
+        [InlineData("bytes = 0-50", HttpStatusCode.PartialContent)]
+        [InlineData("0-6", HttpStatusCode.OK)]
+        [InlineData("bytes = ", HttpStatusCode.OK)]
+        [InlineData("bytes = 1-4, 5-11", HttpStatusCode.OK)]
+        [InlineData("bytes = 35-36", HttpStatusCode.RequestedRangeNotSatisfiable)]
+        [InlineData("bytes = -0", HttpStatusCode.RequestedRangeNotSatisfiable)]
+        public async Task FileFromStream_ReturnsFileWithFileName_DoesNotServeBody_ForHeadRequest(string rangeString, HttpStatusCode httpStatusCode)
+        {
+            // Arrange
+            var httpRequestMessage = new HttpRequestMessage(HttpMethod.Head, "http://localhost/DownloadFiles/DownloadFromStreamWithFileName_WithEtag");
+            httpRequestMessage.Headers.TryAddWithoutValidation("Range", rangeString);
+            httpRequestMessage.Headers.IfRange = new RangeConditionHeaderValue(new EntityTagHeaderValue("\"Etag\""));
+
+            // Act
+            var response = await Client.SendAsync(httpRequestMessage);
+
+            // Assert
+            Assert.Equal(httpStatusCode, response.StatusCode);
+
+            Assert.NotNull(response.Content.Headers.ContentType);
+            Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
+
+            var body = await response.Content.ReadAsStringAsync();
+            Assert.NotNull(body);
+            Assert.Equal(string.Empty, body);
+
+            var contentDisposition = response.Content.Headers.ContentDisposition.ToString();
+            Assert.NotNull(contentDisposition);
+            Assert.Equal("attachment; filename=downloadName.txt; filename*=UTF-8''downloadName.txt", contentDisposition);
+        }
+
         [Fact]
         public async Task FileFromBinaryData_ReturnsFile()
         {
@@ -506,11 +578,13 @@ public async Task FileFromBinaryData_ReturnsFileWithFileName()
             Assert.Equal("attachment; filename=downloadName.txt; filename*=UTF-8''downloadName.txt", contentDisposition);
         }
 
-        [Fact]
-        public async Task FileFromBinaryData_ReturnsFileWithFileName_RangeProcessingNotEnabled_RangeRequestedIgnored()
+        [Theory]
+        [InlineData("GET", "This is a sample text from a binary array")]
+        [InlineData("HEAD", "")]
+        public async Task FileFromBinaryData_ReturnsFileWithFileName_RangeProcessingNotEnabled_RangeRequestedIgnored(string httpMethod, string expectedBody)
         {
             // Arrange
-            var httpRequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://localhost/DownloadFiles/DownloadFromBinaryDataWithFileName");
+            var httpRequestMessage = new HttpRequestMessage(new HttpMethod(httpMethod), "http://localhost/DownloadFiles/DownloadFromBinaryDataWithFileName");
             httpRequestMessage.Headers.Range = new RangeHeaderValue(0, 6);
 
             // Act
@@ -521,7 +595,7 @@ public async Task FileFromBinaryData_ReturnsFileWithFileName_RangeProcessingNotE
             Assert.NotNull(response.Content.Headers.ContentType);
             Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
             var body = await response.Content.ReadAsStringAsync();
-            Assert.Equal("This is a sample text from a binary array", body);
+            Assert.Equal(expectedBody, body);
         }
 
         [Fact]
@@ -563,6 +637,41 @@ public async Task FileFromBinaryData_ReturnsFileWithFileName_IfRangeHeaderInvali
             Assert.Equal("This is a sample text from a binary array", body);
         }
 
+        [Theory]
+        [InlineData("", HttpStatusCode.OK)]
+        [InlineData("bytes = 0-6", HttpStatusCode.PartialContent)]
+        [InlineData("bytes = 17-25", HttpStatusCode.PartialContent)]
+        [InlineData("bytes = 0-50", HttpStatusCode.PartialContent)]
+        [InlineData("0-6", HttpStatusCode.OK)]
+        [InlineData("bytes = ", HttpStatusCode.OK)]
+        [InlineData("bytes = 1-4, 5-11", HttpStatusCode.OK)]
+        [InlineData("bytes = 45-46", HttpStatusCode.RequestedRangeNotSatisfiable)]
+        [InlineData("bytes = -0", HttpStatusCode.RequestedRangeNotSatisfiable)]
+        public async Task FileFromBinaryData_ReturnsFileWithFileName_DoesNotServeBody_ForHeadRequest(string rangeString, HttpStatusCode httpStatusCode)
+        {
+            // Arrange
+            var httpRequestMessage = new HttpRequestMessage(HttpMethod.Head, "http://localhost/DownloadFiles/DownloadFromBinaryDataWithFileName_WithEtag");
+            httpRequestMessage.Headers.TryAddWithoutValidation("Range", rangeString);
+            httpRequestMessage.Headers.IfRange = new RangeConditionHeaderValue(new EntityTagHeaderValue("\"Etag\""));
+
+            // Act
+            var response = await Client.SendAsync(httpRequestMessage);
+
+            // Assert
+            Assert.Equal(httpStatusCode, response.StatusCode);
+
+            Assert.NotNull(response.Content.Headers.ContentType);
+            Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
+
+            var body = await response.Content.ReadAsStringAsync();
+            Assert.NotNull(body);
+            Assert.Equal(string.Empty, body);
+
+            var contentDisposition = response.Content.Headers.ContentDisposition.ToString();
+            Assert.NotNull(contentDisposition);
+            Assert.Equal("attachment; filename=downloadName.txt; filename*=UTF-8''downloadName.txt", contentDisposition);
+        }
+
         [Fact]
         public async Task FileFromEmbeddedResources_ReturnsFileWithFileName()
         {
@@ -612,11 +721,13 @@ public async Task FileFromEmbeddedResources_ReturnsFileWithFileName_RangeRequest
             Assert.Equal("attachment; filename=downloadName.txt; filename*=UTF-8''downloadName.txt", contentDisposition);
         }
 
-        [Fact]
-        public async Task FileFromEmbeddedResources_ReturnsFileWithFileName_RangeProcessingNotEnabled_RangeRequestedIgnored()
+        [Theory]
+        [InlineData("GET", "Sample text file as embedded resource.")]
+        [InlineData("HEAD", "")]
+        public async Task FileFromEmbeddedResources_ReturnsFileWithFileName_RangeProcessingNotEnabled_RangeRequestedIgnored(string httpMethod, string expectedBody)
         {
             // Arrange
-            var httpRequestMessage = new HttpRequestMessage(HttpMethod.Get, "http://localhost/EmbeddedFiles/DownloadFileWithFileName_RangeProcessingNotEnabled");
+            var httpRequestMessage = new HttpRequestMessage(new HttpMethod(httpMethod), "http://localhost/EmbeddedFiles/DownloadFileWithFileName_RangeProcessingNotEnabled");
             httpRequestMessage.Headers.Range = new RangeHeaderValue(0, 6);
 
             // Act
@@ -627,7 +738,7 @@ public async Task FileFromEmbeddedResources_ReturnsFileWithFileName_RangeProcess
             Assert.NotNull(response.Content.Headers.ContentType);
             Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
             var body = await response.Content.ReadAsStringAsync();
-            Assert.Equal("Sample text file as embedded resource.", body);
+            Assert.Equal(expectedBody, body);
         }
 
         [Fact]
@@ -721,5 +832,40 @@ public async Task FileFromEmbeddedResources_ReturnsFileWithFileName_RangeRequest
             Assert.NotNull(contentDisposition);
             Assert.Equal("attachment; filename=downloadName.txt; filename*=UTF-8''downloadName.txt", contentDisposition);
         }
+
+        [Theory]
+        [InlineData("", HttpStatusCode.OK)]
+        [InlineData("bytes = 0-6", HttpStatusCode.PartialContent)]
+        [InlineData("bytes = 17-25", HttpStatusCode.PartialContent)]
+        [InlineData("bytes = 0-50", HttpStatusCode.PartialContent)]
+        [InlineData("0-6", HttpStatusCode.OK)]
+        [InlineData("bytes = ", HttpStatusCode.OK)]
+        [InlineData("bytes = 1-4, 5-11", HttpStatusCode.OK)]
+        [InlineData("bytes = 45-46", HttpStatusCode.RequestedRangeNotSatisfiable)]
+        [InlineData("bytes = -0", HttpStatusCode.RequestedRangeNotSatisfiable)]
+        public async Task FileFromEmbeddedResources_ReturnsFileWithFileName_DoesNotServeBody_ForHeadRequest(string rangeString, HttpStatusCode httpStatusCode)
+        {
+            // Arrange
+            var httpRequestMessage = new HttpRequestMessage(HttpMethod.Head, "http://localhost/EmbeddedFiles/DownloadFileWithFileName");
+            httpRequestMessage.Headers.TryAddWithoutValidation("Range", rangeString);
+            httpRequestMessage.Headers.IfRange = new RangeConditionHeaderValue(new EntityTagHeaderValue("\"Etag\""));
+
+            // Act
+            var response = await Client.SendAsync(httpRequestMessage);
+
+            // Assert
+            Assert.Equal(httpStatusCode, response.StatusCode);
+
+            Assert.NotNull(response.Content.Headers.ContentType);
+            Assert.Equal("text/plain", response.Content.Headers.ContentType.ToString());
+
+            var body = await response.Content.ReadAsStringAsync();
+            Assert.NotNull(body);
+            Assert.Equal(string.Empty, body);
+
+            var contentDisposition = response.Content.Headers.ContentDisposition.ToString();
+            Assert.NotNull(contentDisposition);
+            Assert.Equal("attachment; filename=downloadName.txt; filename*=UTF-8''downloadName.txt", contentDisposition);
+        }
     }
 }