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

85 lines
3.0 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.Threading.Tasks;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Testing;
using Xunit;
namespace Microsoft.AspNetCore.Server.Kestrel.FunctionalTests
{
public class MaxRequestLineSizeTests
{
[Theory]
[InlineData("GET / HTTP/1.1\r\n", 16)]
[InlineData("GET / HTTP/1.1\r\n", 17)]
[InlineData("GET / HTTP/1.1\r\n", 137)]
[InlineData("POST /abc/de HTTP/1.1\r\n", 23)]
[InlineData("POST /abc/de HTTP/1.1\r\n", 24)]
[InlineData("POST /abc/de HTTP/1.1\r\n", 287)]
[InlineData("PUT /abc/de?f=ghi HTTP/1.1\r\n", 28)]
[InlineData("PUT /abc/de?f=ghi HTTP/1.1\r\n", 29)]
[InlineData("PUT /abc/de?f=ghi HTTP/1.1\r\n", 589)]
[InlineData("DELETE /a%20b%20c/d%20e?f=ghi HTTP/1.1\r\n", 40)]
[InlineData("DELETE /a%20b%20c/d%20e?f=ghi HTTP/1.1\r\n", 41)]
[InlineData("DELETE /a%20b%20c/d%20e?f=ghi HTTP/1.1\r\n", 1027)]
public async Task ServerAcceptsRequestLineWithinLimit(string requestLine, int limit)
{
var maxRequestLineSize = limit;
using (var host = BuildWebHost(options =>
{
options.Limits.MaxRequestLineSize = maxRequestLineSize;
}))
{
host.Start();
using (var connection = new TestConnection(host.GetPort()))
{
await connection.SendEnd($"{requestLine}\r\n");
await connection.Receive($"HTTP/1.1 200 OK\r\n");
}
}
}
[Theory]
[InlineData("GET / HTTP/1.1\r\n")]
[InlineData("POST /abc/de HTTP/1.1\r\n")]
[InlineData("PUT /abc/de?f=ghi HTTP/1.1\r\n")]
[InlineData("DELETE /a%20b%20c/d%20e?f=ghi HTTP/1.1\r\n")]
public async Task ServerRejectsRequestLineExceedingLimit(string requestLine)
{
using (var host = BuildWebHost(options =>
{
options.Limits.MaxRequestLineSize = requestLine.Length - 1; // stop short of the '\n'
}))
{
host.Start();
using (var connection = new TestConnection(host.GetPort()))
{
await connection.SendAllTryEnd($"{requestLine}\r\n");
await connection.Receive($"HTTP/1.1 400 Bad Request\r\n");
}
}
}
private IWebHost BuildWebHost(Action<KestrelServerOptions> options)
{
var host = new WebHostBuilder()
.UseKestrel(options)
.UseUrls("http://127.0.0.1:0/")
.Configure(app => app.Run(async context =>
{
await context.Response.WriteAsync("hello, world");
}))
.Build();
return host;
}
}
}