46 lines
1.4 KiB
C#
46 lines
1.4 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.Buffers;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.Connections;
|
|
using Microsoft.AspNetCore.Http.Connections;
|
|
|
|
namespace FunctionalTests
|
|
{
|
|
public class EchoConnectionHandler : ConnectionHandler
|
|
{
|
|
public async override Task OnConnectedAsync(ConnectionContext connection)
|
|
{
|
|
var context = connection.GetHttpContext();
|
|
// The 'withCredentials' tests wont send a cookie for cross-site requests
|
|
if (!context.WebSockets.IsWebSocketRequest && !context.Request.Cookies.ContainsKey("testCookie"))
|
|
{
|
|
return;
|
|
}
|
|
|
|
while (true)
|
|
{
|
|
var result = await connection.Transport.Input.ReadAsync();
|
|
var buffer = result.Buffer;
|
|
|
|
try
|
|
{
|
|
if (!buffer.IsEmpty)
|
|
{
|
|
await connection.Transport.Output.WriteAsync(buffer.ToArray());
|
|
}
|
|
else if (result.IsCompleted)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
connection.Transport.Input.AdvanceTo(result.Buffer.End);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|