public async Task WriteResponseAsync(MiddlemanResponse response)
{
await WriteResponseAsync(response, CancellationToken.None)
.ConfigureAwait(false);
}
public async Task WriteResponseAsync(MiddlemanResponse response, CancellationToken ct)
{
var ms = new MemoryStream();
var sw = new StreamWriter(ms, HeaderEncoding) { NewLine = "\r\n" };
sw.WriteLine("HTTP/{0} {1} {2}", RequestVersion, response.StatusCode, response.StatusDescription);
for (var i = 0; i < response.Headers.Count; i++)
{
var key = response.Headers.GetKey(i);
var val = response.Headers.Get(i);
if (!key.Equals("X-AspNet-Version", StringComparison.InvariantCultureIgnoreCase) &&
!key.Equals("X-SourceFiles", StringComparison.InvariantCultureIgnoreCase) &&
!key.Equals("X-Powered-By", StringComparison.InvariantCultureIgnoreCase))
{
sw.WriteLine("{0}: {1}", key, val);
}
}
sw.WriteLine();
sw.Flush();
var writeStream = GetWriteStream();
ms.Position = 0;
var responseHeaderBytes = ms.ToArray();
var responseHeaders = HeaderEncoding.GetString(responseHeaderBytes.Where(x => x != 0).ToArray());
await writeStream.WriteAsync(responseHeaderBytes, 0, responseHeaderBytes.Length, ct).ConfigureAwait(false);
Log.Debug("{0}: Wrote headers ({1}b)", RemoteEndPoint, ms.Length);
var responseBody = "";
if (response.ResponseBody != null && response.ResponseBody.CanRead)
{
var buffer = new byte[8192];
int read;
long written = 0;
while (
(read = await response.ResponseBody.ReadAsync(buffer, 0, buffer.Length, ct).ConfigureAwait(false)) >
0)
{
responseBody += HeaderEncoding.GetString(buffer.Where(x => x != 0).ToArray(), 0, read);
written += read;
Log.Debug("{0}: Read {1:N0} bytes from response body", RemoteEndPoint, read);
try
{
await writeStream.WriteAsync(buffer, 0, read, ct).ConfigureAwait(false);
Log.Debug("{0}: Wrote {1:N0} bytes to client", RemoteEndPoint, read);
}
catch (Exception ex)
{
Log.ErrorException("Error writing to response stream", ex);
}
if (written >= response.ContentLength)
{
break;
}
}
Log.Debug("{0}: Wrote response body ({1:N0} bytes) to client", RemoteEndPoint, written);
}
Log.Info("RESPONSE TO CLIENT: " + Environment.NewLine + (responseHeaders.Trim() + Environment.NewLine + Environment.NewLine + responseBody.Trim()).Trim() + Environment.NewLine);
await writeStream.FlushAsync(ct).ConfigureAwait(false);
}