aspnetcore/test/Microsoft.AspNetCore.Server.../RequestTests.cs

353 lines
16 KiB
C#

// 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.IO;
using System.Net;
using System.Net.Http;
using System.Net.Sockets;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Authentication;
using Microsoft.AspNetCore.Hosting.Server;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Http.Features;
using Microsoft.AspNetCore.HttpSys.Internal;
using Microsoft.AspNetCore.Testing.xunit;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using Xunit;
namespace Microsoft.AspNetCore.Server.HttpSys
{
public class RequestTests
{
[ConditionalFact]
public async Task Request_SimpleGet_ExpectedFieldsSet()
{
string root;
using (Utilities.CreateHttpServerReturnRoot("/basepath", out root, httpContext =>
{
try
{
var requestInfo = httpContext.Features.Get<IHttpRequestFeature>();
// Request Keys
Assert.Equal("GET", requestInfo.Method);
Assert.Equal(Stream.Null, requestInfo.Body);
Assert.NotNull(requestInfo.Headers);
Assert.Equal("http", requestInfo.Scheme);
Assert.Equal("/basepath", requestInfo.PathBase);
Assert.Equal("/SomePath", requestInfo.Path);
Assert.Equal("?SomeQuery", requestInfo.QueryString);
Assert.Equal("/basepath/SomePath?SomeQuery", requestInfo.RawTarget);
Assert.Equal("HTTP/1.1", requestInfo.Protocol);
var connectionInfo = httpContext.Features.Get<IHttpConnectionFeature>();
Assert.Equal("::1", connectionInfo.RemoteIpAddress.ToString());
Assert.NotEqual(0, connectionInfo.RemotePort);
Assert.Equal("::1", connectionInfo.LocalIpAddress.ToString());
Assert.NotEqual(0, connectionInfo.LocalPort);
Assert.NotNull(connectionInfo.ConnectionId);
// Trace identifier
var requestIdentifierFeature = httpContext.Features.Get<IHttpRequestIdentifierFeature>();
Assert.NotNull(requestIdentifierFeature);
Assert.NotNull(requestIdentifierFeature.TraceIdentifier);
// Note: Response keys are validated in the ResponseTests
}
catch (Exception ex)
{
byte[] body = Encoding.ASCII.GetBytes(ex.ToString());
httpContext.Response.Body.Write(body, 0, body.Length);
}
return Task.FromResult(0);
}))
{
string response = await SendRequestAsync(root + "/basepath/SomePath?SomeQuery");
Assert.Equal(string.Empty, response);
}
}
[ConditionalFact]
public async Task Request_FieldsCanBeSet_Set()
{
string root;
using (Utilities.CreateHttpServerReturnRoot("/basepath", out root, httpContext =>
{
try
{
var requestInfo = httpContext.Features.Get<IHttpRequestFeature>();
// Request Keys
requestInfo.Method = "TEST";
Assert.Equal("TEST", requestInfo.Method);
requestInfo.Body = new MemoryStream();
Assert.IsType<MemoryStream>(requestInfo.Body);
var customHeaders = new HeaderCollection();
requestInfo.Headers = customHeaders;
Assert.Same(customHeaders, requestInfo.Headers);
requestInfo.Scheme = "abcd";
Assert.Equal("abcd", requestInfo.Scheme);
requestInfo.PathBase = "/customized/Base";
Assert.Equal("/customized/Base", requestInfo.PathBase);
requestInfo.Path = "/customized/Path";
Assert.Equal("/customized/Path", requestInfo.Path);
requestInfo.QueryString = "?customizedQuery";
Assert.Equal("?customizedQuery", requestInfo.QueryString);
requestInfo.RawTarget = "/customized/raw?Target";
Assert.Equal("/customized/raw?Target", requestInfo.RawTarget);
requestInfo.Protocol = "Custom/2.0";
Assert.Equal("Custom/2.0", requestInfo.Protocol);
var connectionInfo = httpContext.Features.Get<IHttpConnectionFeature>();
connectionInfo.RemoteIpAddress = IPAddress.Broadcast;
Assert.Equal(IPAddress.Broadcast, connectionInfo.RemoteIpAddress);
connectionInfo.RemotePort = 12345;
Assert.Equal(12345, connectionInfo.RemotePort);
connectionInfo.LocalIpAddress = IPAddress.Any;
Assert.Equal(IPAddress.Any, connectionInfo.LocalIpAddress);
connectionInfo.LocalPort = 54321;
Assert.Equal(54321, connectionInfo.LocalPort);
connectionInfo.ConnectionId = "CustomId";
Assert.Equal("CustomId", connectionInfo.ConnectionId);
// Trace identifier
var requestIdentifierFeature = httpContext.Features.Get<IHttpRequestIdentifierFeature>();
Assert.NotNull(requestIdentifierFeature);
requestIdentifierFeature.TraceIdentifier = "customTrace";
Assert.Equal("customTrace", requestIdentifierFeature.TraceIdentifier);
// Note: Response keys are validated in the ResponseTests
}
catch (Exception ex)
{
byte[] body = Encoding.ASCII.GetBytes(ex.ToString());
httpContext.Response.Body.Write(body, 0, body.Length);
}
return Task.FromResult(0);
}))
{
string response = await SendRequestAsync(root + "/basepath/SomePath?SomeQuery");
Assert.Equal(string.Empty, response);
}
}
[ConditionalFact]
public async Task Request_FieldsCanBeSetToNull_Set()
{
string root;
using (Utilities.CreateHttpServerReturnRoot("/basepath", out root, httpContext =>
{
try
{
var requestInfo = httpContext.Features.Get<IHttpRequestFeature>();
// Request Keys
requestInfo.Method = null;
Assert.Null(requestInfo.Method);
requestInfo.Body = null;
Assert.Null(requestInfo.Body);
requestInfo.Headers = null;
Assert.Null(requestInfo.Headers);
requestInfo.Scheme = null;
Assert.Null(requestInfo.Scheme);
requestInfo.PathBase = null;
Assert.Null(requestInfo.PathBase);
requestInfo.Path = null;
Assert.Null(requestInfo.Path);
requestInfo.QueryString = null;
Assert.Null(requestInfo.QueryString);
requestInfo.RawTarget = null;
Assert.Null(requestInfo.RawTarget);
requestInfo.Protocol = null;
Assert.Null(requestInfo.Protocol);
var connectionInfo = httpContext.Features.Get<IHttpConnectionFeature>();
connectionInfo.RemoteIpAddress = null;
Assert.Null(connectionInfo.RemoteIpAddress);
connectionInfo.RemotePort = -1;
Assert.Equal(-1, connectionInfo.RemotePort);
connectionInfo.LocalIpAddress = null;
Assert.Null(connectionInfo.LocalIpAddress);
connectionInfo.LocalPort = -1;
Assert.Equal(-1, connectionInfo.LocalPort);
connectionInfo.ConnectionId = null;
Assert.Null(connectionInfo.ConnectionId);
// Trace identifier
var requestIdentifierFeature = httpContext.Features.Get<IHttpRequestIdentifierFeature>();
Assert.NotNull(requestIdentifierFeature);
requestIdentifierFeature.TraceIdentifier = null;
Assert.Null(requestIdentifierFeature.TraceIdentifier);
// Note: Response keys are validated in the ResponseTests
}
catch (Exception ex)
{
byte[] body = Encoding.ASCII.GetBytes(ex.ToString());
httpContext.Response.Body.Write(body, 0, body.Length);
}
return Task.FromResult(0);
}))
{
string response = await SendRequestAsync(root + "/basepath/SomePath?SomeQuery");
Assert.Equal(string.Empty, response);
}
}
[ConditionalTheory]
[InlineData("/", "/", "", "/")]
[InlineData("/basepath/", "/basepath", "/basepath", "")]
[InlineData("/basepath/", "/basepath/", "/basepath", "/")]
[InlineData("/basepath/", "/basepath/subpath", "/basepath", "/subpath")]
[InlineData("/base path/", "/base%20path/sub%20path", "/base path", "/sub path")]
[InlineData("/base葉path/", "/base%E8%91%89path/sub%E8%91%89path", "/base葉path", "/sub葉path")]
[InlineData("/basepath/", "/basepath/sub%2Fpath", "/basepath", "/sub%2Fpath")]
public async Task Request_PathSplitting(string pathBase, string requestPath, string expectedPathBase, string expectedPath)
{
string root;
using (Utilities.CreateHttpServerReturnRoot(pathBase, out root, httpContext =>
{
try
{
var requestInfo = httpContext.Features.Get<IHttpRequestFeature>();
var connectionInfo = httpContext.Features.Get<IHttpConnectionFeature>();
var requestIdentifierFeature = httpContext.Features.Get<IHttpRequestIdentifierFeature>();
// Request Keys
Assert.Equal("http", requestInfo.Scheme);
Assert.Equal(expectedPath, requestInfo.Path);
Assert.Equal(expectedPathBase, requestInfo.PathBase);
Assert.Equal(string.Empty, requestInfo.QueryString);
Assert.Equal(requestPath, requestInfo.RawTarget);
// Trace identifier
Assert.NotNull(requestIdentifierFeature);
Assert.NotNull(requestIdentifierFeature.TraceIdentifier);
}
catch (Exception ex)
{
byte[] body = Encoding.ASCII.GetBytes(ex.ToString());
httpContext.Response.Body.Write(body, 0, body.Length);
}
return Task.FromResult(0);
}))
{
string response = await SendRequestAsync(root + requestPath);
Assert.Equal(string.Empty, response);
}
}
[ConditionalFact]
public async Task Request_DoubleEscapingAllowed()
{
string root;
using (var server = Utilities.CreateHttpServerReturnRoot("/", out root, httpContext =>
{
var requestInfo = httpContext.Features.Get<IHttpRequestFeature>();
Assert.Equal("/%2F", requestInfo.Path);
Assert.Equal("/%252F", requestInfo.RawTarget);
return Task.FromResult(0);
}))
{
var response = await SendSocketRequestAsync(root, "/%252F");
var responseStatusCode = response.Substring(9); // Skip "HTTP/1.1 "
Assert.Equal("200", responseStatusCode);
}
}
[ConditionalTheory]
// The test server defines these prefixes: "/", "/11", "/2/3", "/2", "/11/2"
[InlineData("/", "", "/")]
[InlineData("/random", "", "/random")]
[InlineData("/11", "/11", "")]
[InlineData("/11/", "/11", "/")]
[InlineData("/11/random", "/11", "/random")]
[InlineData("/2", "/2", "")]
[InlineData("/2/", "/2", "/")]
[InlineData("/2/random", "/2", "/random")]
[InlineData("/2/3", "/2/3", "")]
[InlineData("/2/3/", "/2/3", "/")]
[InlineData("/2/3/random", "/2/3", "/random")]
public async Task Request_MultiplePrefixes(string requestPath, string expectedPathBase, string expectedPath)
{
string root;
using (CreateServer(out root, httpContext =>
{
var requestInfo = httpContext.Features.Get<IHttpRequestFeature>();
var requestIdentifierFeature = httpContext.Features.Get<IHttpRequestIdentifierFeature>();
try
{
Assert.Equal(expectedPath, requestInfo.Path);
Assert.Equal(expectedPathBase, requestInfo.PathBase);
Assert.Equal(requestPath, requestInfo.RawTarget);
// Trace identifier
Assert.NotNull(requestIdentifierFeature);
Assert.NotNull(requestIdentifierFeature.TraceIdentifier);
}
catch (Exception ex)
{
byte[] body = Encoding.ASCII.GetBytes(ex.ToString());
httpContext.Response.Body.Write(body, 0, body.Length);
}
return Task.FromResult(0);
}))
{
string response = await SendRequestAsync(root + requestPath);
Assert.Equal(string.Empty, response);
}
}
private IServer CreateServer(out string root, RequestDelegate app)
{
// TODO: We're just doing this to get a dynamic port. This can be removed later when we add support for hot-adding prefixes.
var dynamicServer = Utilities.CreateHttpServerReturnRoot("/", out root, app);
dynamicServer.Dispose();
var rootUri = new Uri(root);
var server = Utilities.CreatePump();
foreach (string path in new[] { "/", "/11", "/2/3", "/2", "/11/2" })
{
server.Listener.Options.UrlPrefixes.Add(UrlPrefix.Create(rootUri.Scheme, rootUri.Host, rootUri.Port, path));
}
server.StartAsync(new DummyApplication(app), CancellationToken.None).Wait();
return server;
}
private async Task<string> SendRequestAsync(string uri)
{
using (HttpClient client = new HttpClient())
{
return await client.GetStringAsync(uri);
}
}
private async Task<string> SendSocketRequestAsync(string address, string path)
{
var uri = new Uri(address);
StringBuilder builder = new StringBuilder();
builder.AppendLine("GET " + path + " HTTP/1.1");
builder.AppendLine("Connection: close");
builder.Append("HOST: ");
builder.AppendLine(uri.Authority);
builder.AppendLine();
byte[] request = Encoding.ASCII.GetBytes(builder.ToString());
using (var socket = new Socket(SocketType.Stream, ProtocolType.Tcp))
{
socket.Connect(uri.Host, uri.Port);
socket.Send(request);
var response = new byte[12];
await Task.Run(() => socket.Receive(response));
return Encoding.ASCII.GetString(response);
}
}
}
}