460 lines
15 KiB
C#
460 lines
15 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.Sockets;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNet.Server.Kestrel;
|
|
using Microsoft.AspNet.Server.Kestrel.Http;
|
|
using Microsoft.Dnx.Runtime;
|
|
using Microsoft.Dnx.Runtime.Infrastructure;
|
|
using Xunit;
|
|
|
|
namespace Microsoft.AspNet.Server.KestrelTests
|
|
{
|
|
/// <summary>
|
|
/// Summary description for EngineTests
|
|
/// </summary>
|
|
public class EngineTests
|
|
{
|
|
private async Task App(Frame frame)
|
|
{
|
|
for (; ;)
|
|
{
|
|
var buffer = new byte[8192];
|
|
var count = await frame.RequestBody.ReadAsync(buffer, 0, buffer.Length);
|
|
if (count == 0)
|
|
{
|
|
break;
|
|
}
|
|
await frame.ResponseBody.WriteAsync(buffer, 0, count);
|
|
}
|
|
}
|
|
|
|
ILibraryManager LibraryManager
|
|
{
|
|
get
|
|
{
|
|
try
|
|
{
|
|
var locator = CallContextServiceLocator.Locator;
|
|
if (locator == null)
|
|
{
|
|
return null;
|
|
}
|
|
var services = locator.ServiceProvider;
|
|
if (services == null)
|
|
{
|
|
return null;
|
|
}
|
|
return (ILibraryManager)services.GetService(typeof(ILibraryManager));
|
|
}
|
|
catch (NullReferenceException)
|
|
{ return null; }
|
|
}
|
|
}
|
|
|
|
private async Task AppChunked(Frame frame)
|
|
{
|
|
var data = new MemoryStream();
|
|
for (; ;)
|
|
{
|
|
var buffer = new byte[8192];
|
|
var count = await frame.RequestBody.ReadAsync(buffer, 0, buffer.Length);
|
|
if (count == 0)
|
|
{
|
|
break;
|
|
}
|
|
data.Write(buffer, 0, count);
|
|
}
|
|
var bytes = data.ToArray();
|
|
frame.ResponseHeaders["Content-Length"] = new[] { bytes.Length.ToString() };
|
|
await frame.ResponseBody.WriteAsync(bytes, 0, bytes.Length);
|
|
}
|
|
|
|
[Fact]
|
|
public void EngineCanStartAndStop()
|
|
{
|
|
var engine = new KestrelEngine(LibraryManager, new ShutdownNotImplemented());
|
|
engine.Start(1);
|
|
engine.Dispose();
|
|
}
|
|
|
|
[Fact]
|
|
public void ListenerCanCreateAndDispose()
|
|
{
|
|
var engine = new KestrelEngine(LibraryManager, new ShutdownNotImplemented());
|
|
engine.Start(1);
|
|
var started = engine.CreateServer("http", "localhost", 54321, App);
|
|
started.Dispose();
|
|
engine.Dispose();
|
|
}
|
|
|
|
|
|
[Fact]
|
|
public void ConnectionCanReadAndWrite()
|
|
{
|
|
var engine = new KestrelEngine(LibraryManager, new ShutdownNotImplemented());
|
|
engine.Start(1);
|
|
var started = engine.CreateServer("http", "localhost", 54321, App);
|
|
|
|
Console.WriteLine("Started");
|
|
var socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
|
|
socket.Connect(new IPEndPoint(IPAddress.Loopback, 54321));
|
|
socket.Send(Encoding.ASCII.GetBytes("POST / HTTP/1.0\r\n\r\nHello World"));
|
|
socket.Shutdown(SocketShutdown.Send);
|
|
var buffer = new byte[8192];
|
|
for (; ;)
|
|
{
|
|
var length = socket.Receive(buffer);
|
|
if (length == 0) { break; }
|
|
var text = Encoding.ASCII.GetString(buffer, 0, length);
|
|
}
|
|
started.Dispose();
|
|
engine.Dispose();
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Http10()
|
|
{
|
|
using (var server = new TestServer(App))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.SendEnd(
|
|
"POST / HTTP/1.0",
|
|
"",
|
|
"Hello World");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.0 200 OK",
|
|
"",
|
|
"Hello World");
|
|
}
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Http11()
|
|
{
|
|
using (var server = new TestServer(AppChunked))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.SendEnd(
|
|
"GET / HTTP/1.1",
|
|
"",
|
|
"GET / HTTP/1.1",
|
|
"Connection: close",
|
|
"",
|
|
"Goodbye");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.1 200 OK",
|
|
"Content-Length: 0",
|
|
"",
|
|
"HTTP/1.1 200 OK",
|
|
"Content-Length: 7",
|
|
"Connection: close",
|
|
"",
|
|
"Goodbye");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
[Fact]
|
|
public async Task Http10ContentLength()
|
|
{
|
|
using (var server = new TestServer(App))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.Send(
|
|
"POST / HTTP/1.0",
|
|
"Content-Length: 11",
|
|
"",
|
|
"Hello World");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.0 200 OK",
|
|
"",
|
|
"Hello World");
|
|
}
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Http10TransferEncoding()
|
|
{
|
|
using (var server = new TestServer(App))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.Send(
|
|
"POST / HTTP/1.0",
|
|
"Transfer-Encoding: chunked",
|
|
"",
|
|
"5", "Hello", "6", " World", "0\r\n");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.0 200 OK",
|
|
"",
|
|
"Hello World");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
[Fact]
|
|
public async Task Http10KeepAlive()
|
|
{
|
|
using (var server = new TestServer(AppChunked))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.SendEnd(
|
|
"GET / HTTP/1.0",
|
|
"Connection: keep-alive",
|
|
"",
|
|
"POST / HTTP/1.0",
|
|
"",
|
|
"Goodbye");
|
|
await connection.Receive(
|
|
"HTTP/1.0 200 OK",
|
|
"Content-Length: 0",
|
|
"Connection: keep-alive",
|
|
"\r\n");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.0 200 OK",
|
|
"Content-Length: 7",
|
|
"",
|
|
"Goodbye");
|
|
}
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Http10KeepAliveContentLength()
|
|
{
|
|
using (var server = new TestServer(AppChunked))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.SendEnd(
|
|
"POST / HTTP/1.0",
|
|
"Connection: keep-alive",
|
|
"Content-Length: 11",
|
|
"",
|
|
"Hello WorldPOST / HTTP/1.0",
|
|
"",
|
|
"Goodbye");
|
|
await connection.Receive(
|
|
"HTTP/1.0 200 OK",
|
|
"Content-Length: 11",
|
|
"Connection: keep-alive",
|
|
"",
|
|
"Hello World");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.0 200 OK",
|
|
"Content-Length: 7",
|
|
"",
|
|
"Goodbye");
|
|
}
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Http10KeepAliveTransferEncoding()
|
|
{
|
|
using (var server = new TestServer(AppChunked))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.SendEnd(
|
|
"POST / HTTP/1.0",
|
|
"Transfer-Encoding: chunked",
|
|
"Connection: keep-alive",
|
|
"",
|
|
"5", "Hello", "6", " World", "0",
|
|
"POST / HTTP/1.0",
|
|
"",
|
|
"Goodbye");
|
|
await connection.Receive(
|
|
"HTTP/1.0 200 OK",
|
|
"Content-Length: 11",
|
|
"Connection: keep-alive",
|
|
"",
|
|
"Hello World");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.0 200 OK",
|
|
"Content-Length: 7",
|
|
"",
|
|
"Goodbye");
|
|
}
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task Expect100ContinueForBody()
|
|
{
|
|
using (var server = new TestServer(AppChunked))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.Send(
|
|
"POST / HTTP/1.1",
|
|
"Expect: 100-continue",
|
|
"Content-Length: 11",
|
|
"Connection: close",
|
|
"\r\n");
|
|
await connection.Receive("HTTP/1.1 100 Continue", "\r\n");
|
|
await connection.SendEnd("Hello World");
|
|
await connection.Receive(
|
|
"HTTP/1.1 200 OK",
|
|
"Content-Length: 11",
|
|
"Connection: close",
|
|
"",
|
|
"Hello World");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
[Fact]
|
|
public async Task DisconnectingClient()
|
|
{
|
|
using (var server = new TestServer(App))
|
|
{
|
|
var socket = new Socket(SocketType.Stream, ProtocolType.IP);
|
|
socket.Connect(IPAddress.Loopback, 54321);
|
|
await Task.Delay(200);
|
|
socket.Disconnect(false);
|
|
socket.Dispose();
|
|
|
|
await Task.Delay(200);
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.SendEnd(
|
|
"GET / HTTP/1.0",
|
|
"\r\n");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.0 200 OK",
|
|
"\r\n");
|
|
}
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ThrowingResultsIn500Response()
|
|
{
|
|
bool onStartingCalled = false;
|
|
|
|
using (var server = new TestServer(frame =>
|
|
{
|
|
frame.OnStarting(_ =>
|
|
{
|
|
onStartingCalled = true;
|
|
return Task.FromResult<object>(null);
|
|
}, null);
|
|
|
|
// Anything added to the ResponseHeaders dictionary is ignored
|
|
frame.ResponseHeaders["Content-Length"] = new[] { "11" };
|
|
throw new Exception();
|
|
}))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.SendEnd(
|
|
"GET / HTTP/1.1",
|
|
"",
|
|
"GET / HTTP/1.1",
|
|
"Connection: close",
|
|
"",
|
|
"");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.1 500 Internal Server Error",
|
|
"Content-Length: 0",
|
|
"",
|
|
"HTTP/1.1 500 Internal Server Error",
|
|
"Content-Length: 0",
|
|
"Connection: close",
|
|
"",
|
|
"");
|
|
|
|
Assert.False(onStartingCalled);
|
|
}
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ThrowingAfterWritingKillsConnection()
|
|
{
|
|
bool onStartingCalled = false;
|
|
|
|
using (var server = new TestServer(async frame =>
|
|
{
|
|
frame.OnStarting(_ =>
|
|
{
|
|
onStartingCalled = true;
|
|
return Task.FromResult<object>(null);
|
|
}, null);
|
|
|
|
frame.ResponseHeaders["Content-Length"] = new[] { "11" };
|
|
await frame.ResponseBody.WriteAsync(Encoding.ASCII.GetBytes("Hello World"), 0, 11);
|
|
throw new Exception();
|
|
}))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.Send(
|
|
"GET / HTTP/1.1",
|
|
"",
|
|
"");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.1 200 OK",
|
|
"Content-Length: 11",
|
|
"",
|
|
"Hello World");
|
|
|
|
Assert.True(onStartingCalled);
|
|
}
|
|
}
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ThrowingAfterPartialWriteKillsConnection()
|
|
{
|
|
bool onStartingCalled = false;
|
|
|
|
using (var server = new TestServer(async frame =>
|
|
{
|
|
frame.OnStarting(_ =>
|
|
{
|
|
onStartingCalled = true;
|
|
return Task.FromResult<object>(null);
|
|
}, null);
|
|
|
|
frame.ResponseHeaders["Content-Length"] = new[] { "11" };
|
|
await frame.ResponseBody.WriteAsync(Encoding.ASCII.GetBytes("Hello"), 0, 5);
|
|
throw new Exception();
|
|
}))
|
|
{
|
|
using (var connection = new TestConnection())
|
|
{
|
|
await connection.Send(
|
|
"GET / HTTP/1.1",
|
|
"",
|
|
"");
|
|
await connection.ReceiveEnd(
|
|
"HTTP/1.1 200 OK",
|
|
"Content-Length: 11",
|
|
"",
|
|
"Hello");
|
|
|
|
Assert.True(onStartingCalled);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|