345 lines
11 KiB
C#
345 lines
11 KiB
C#
using Microsoft.AspNet.Server.Kestrel;
|
|
using Microsoft.AspNet.Server.Kestrel.Http;
|
|
using Microsoft.Framework.Runtime;
|
|
using Microsoft.Framework.Runtime.Infrastructure;
|
|
using System;
|
|
using System.IO;
|
|
using System.Net;
|
|
using System.Net.Sockets;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Xunit;
|
|
|
|
namespace Microsoft.AspNet.Server.KestrelTests
|
|
{
|
|
/// <summary>
|
|
/// Summary description for EngineTests
|
|
/// </summary>
|
|
internal 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 async Task EngineCanStartAndStop()
|
|
{
|
|
var engine = new KestrelEngine(LibraryManager);
|
|
engine.Start(1);
|
|
engine.Dispose();
|
|
}
|
|
|
|
[Fact]
|
|
public async Task ListenerCanCreateAndDispose()
|
|
{
|
|
var engine = new KestrelEngine(LibraryManager);
|
|
engine.Start(1);
|
|
var started = engine.CreateServer("http", "localhost", 54321, App);
|
|
started.Dispose();
|
|
engine.Dispose();
|
|
}
|
|
|
|
|
|
[Fact]
|
|
public async Task ConnectionCanReadAndWrite()
|
|
{
|
|
var engine = new KestrelEngine(LibraryManager);
|
|
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");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|