private static void PlainText(ref WritableBuffer outputBuffer)
{
var writer = new WritableBufferWriter(outputBuffer);
// HTTP 1.1 OK
writer.Write(_http11OK);
// Server headers
writer.Write(_headerServer);
// Date header
writer.Write(_dateHeaderValueManager.GetDateHeaderValues().Bytes);
writer.Write(_crlf);
// Content-Type header
writer.Write(_headerContentTypeText);
writer.Write(_crlf);
// Content-Length header
writer.Write(_headerContentLength);
PipelineExtensions.WriteNumeric(ref writer, (ulong)_plainTextBody.Length);
writer.Write(_crlf);
// End of headers
writer.Write(_crlf);
// Body
writer.Write(_plainTextBody);
}
private void Json(ref WritableBuffer outputBuffer)
{
var writer = new WritableBufferWriter(outputBuffer);
// HTTP 1.1 OK
writer.Write(_bytesHttpVersion11);
writer.Write(ReasonPhrases.ToStatusBytes(200, reasonPhrase: null));
// Headers
var values = _dateHeaderValueManager.GetDateHeaderValues();
ResponseHeaders.SetRawDate(values.String, values.Bytes);
ResponseHeaders.SetRawServer("Custom", _headerServer);
ResponseHeaders.HeaderContentType = "application/json";
var jsonPayload = JsonSerializer.SerializeUnsafe(new { message = "Hello, World!" });
ResponseHeaders.ContentLength = jsonPayload.Count;
// Write headers
ResponseHeaders.CopyTo(ref writer);
writer.Write(_bytesEndHeaders);
// Body
writer.Write(jsonPayload.Array, jsonPayload.Offset, jsonPayload.Count);
}
public void ThrowsForInvalidParameters(int arrayLength, int offset, int length)
{
_buffer = _pipe.Writer.Alloc(1);
var initialLength = _buffer.Buffer.Length;
var writer = new WritableBufferWriter(_buffer);
var array = new byte[arrayLength];
for (int i = 0; i < array.Length; i++)
{
array[i] = (byte)(i + 1);
}
writer.Write(array, 0, 0);
writer.Write(array, array.Length, 0);
try
{
writer.Write(array, offset, length);
Assert.True(false);
}
catch (Exception ex)
{
Assert.True(ex is ArgumentOutOfRangeException);
}
writer.Write(array, 0, array.Length);
Assert.Equal(array, Read());
}
private static void Json(ref WritableBuffer outputBuffer)
{
var writer = new WritableBufferWriter(outputBuffer);
// HTTP 1.1 OK
writer.Write(_http11OK);
// Server headers
writer.Write(_headerServer);
// Date header
writer.Write(_dateHeaderValueManager.GetDateHeaderValues().Bytes);
writer.Write(_crlf);
// Content-Type header
writer.Write(_headerContentTypeJson);
writer.Write(_crlf);
var jsonPayload = JsonSerializer.SerializeUnsafe(new { message = "Hello, World!" });
// Content-Length header
writer.Write(_headerContentLength);
PipelineExtensions.WriteNumeric(ref writer, (ulong)jsonPayload.Count);
writer.Write(_crlf);
// End of headers
writer.Write(_crlf);
// Body
writer.Write(jsonPayload.Array, jsonPayload.Offset, jsonPayload.Count);
}
public void CanWriteEmpty()
{
_buffer = _pipe.Writer.Alloc();
var writer = new WritableBufferWriter(_buffer);
var array = new byte[] { };
writer.Write(array);
writer.Write(array, 0, array.Length);
Assert.Equal(array, Read());
}
public void CanWriteMultipleTimes()
{
_buffer = _pipe.Writer.Alloc();
var writer = new WritableBufferWriter(_buffer);
writer.Write(new byte[] { 1 });
writer.Write(new byte[] { 2 });
writer.Write(new byte[] { 3 });
Assert.Equal(new byte[] { 1, 2, 3 }, Read());
}
public static int WriteBeginChunkBytes(ref WritableBufferWriter start, int dataCount)
{
var chunkSegment = BeginChunkBytes(dataCount);
start.Write(chunkSegment.Array, chunkSegment.Offset, chunkSegment.Count);
return(chunkSegment.Count);
}
public Task WriteAsync(
ArraySegment <byte> buffer,
CancellationToken cancellationToken)
{
var writableBuffer = default(WritableBuffer);
lock (_contextLock)
{
if (_completed)
{
throw new ObjectDisposedException("Response is already completed");
}
writableBuffer = _pipe.Writer.Alloc(1);
// TODO obsolete
#pragma warning disable CS0618 // Type or member is obsolete
var writer = new WritableBufferWriter(writableBuffer);
#pragma warning restore CS0618 // Type or member is obsolete
if (buffer.Count > 0)
{
writer.Write(buffer.Array, buffer.Offset, buffer.Count);
}
writableBuffer.Commit();
}
return(FlushAsync(writableBuffer, cancellationToken));
}
private Task WriteAsync(
ArraySegment <byte> buffer,
CancellationToken cancellationToken)
{
var writableBuffer = default(WritableBuffer);
long bytesWritten = 0;
lock (_contextLock)
{
if (_completed)
{
return(Task.CompletedTask);
}
writableBuffer = _pipeWriter.Alloc(1);
var writer = new WritableBufferWriter(writableBuffer);
if (buffer.Count > 0)
{
writer.Write(buffer.Array, buffer.Offset, buffer.Count);
bytesWritten += buffer.Count;
}
writableBuffer.Commit();
}
return(FlushAsync(writableBuffer, bytesWritten, cancellationToken));
}
public void CanWriteIntoHeadlessBuffer()
{
_buffer = _pipe.Writer.Alloc();
var writer = new WritableBufferWriter(_buffer);
writer.Write(new byte[] { 1, 2, 3 });
Assert.Equal(new byte[] { 1, 2, 3 }, Read());
}
private void Default(ref WritableBuffer outputBuffer)
{
var writer = new WritableBufferWriter(outputBuffer);
// HTTP 1.1 OK
writer.Write(_bytesHttpVersion11);
writer.Write(ReasonPhrases.ToStatusBytes(200, reasonPhrase: null));
// Headers
var values = _dateHeaderValueManager.GetDateHeaderValues();
ResponseHeaders.SetRawDate(values.String, values.Bytes);
ResponseHeaders.SetRawServer("Custom", _headerServer);
ResponseHeaders.ContentLength = 0;
// Write headers
ResponseHeaders.CopyTo(ref writer);
writer.Write(_bytesEndHeaders);
}
public void CanWriteWithOffsetAndLenght(int alloc, int offset, int length)
{
_buffer = _pipe.Writer.Alloc(alloc);
var writer = new WritableBufferWriter(_buffer);
var array = new byte[] { 1, 2, 3 };
writer.Write(array, offset, length);
Assert.Equal(array.Skip(offset).Take(length).ToArray(), Read());
}
public void WriteResponseHeaders(int statusCode, string reasonPhrase, HttpResponseHeaders responseHeaders)
{
lock (_contextLock)
{
if (_completed)
{
return;
}
var buffer = _pipeWriter.Alloc(1);
var writer = new WritableBufferWriter(buffer);
writer.Write(_bytesHttpVersion11);
var statusBytes = ReasonPhrases.ToStatusBytes(statusCode, reasonPhrase);
writer.Write(statusBytes);
responseHeaders.CopyTo(ref writer);
writer.Write(_bytesEndHeaders);
buffer.Commit();
}
}
public void CopyTo(ref WritableBufferWriter output)
{
CopyToFast(ref output);
if (MaybeUnknown != null)
{
foreach (var kv in MaybeUnknown)
{
foreach (var value in kv.Value)
{
if (value != null)
{
output.Write(_CrLf);
PipelineExtensions.WriteAsciiNoValidation(ref output, kv.Key);
output.Write(_colonSpace);
PipelineExtensions.WriteAsciiNoValidation(ref output, value);
}
}
}
}
}
public void CanWriteOverTheBlockLength()
{
_buffer = _pipe.Writer.Alloc(1);
var writer = new WritableBufferWriter(_buffer);
var source = Enumerable.Range(0, _buffer.Buffer.Length).Select(i => (byte)i);
var expectedBytes = source.Concat(source).Concat(source).ToArray();
writer.Write(expectedBytes);
Assert.Equal(expectedBytes, Read());
}
public void SlicesSpanAndAdvancesAfterWrite()
{
_buffer = _pipe.Writer.Alloc(1);
var initialLength = _buffer.Buffer.Length;
var writer = new WritableBufferWriter(_buffer);
writer.Write(new byte[] { 1, 2, 3 });
Assert.Equal(initialLength - 3, writer.Span.Length);
Assert.Equal(_buffer.Buffer.Length, writer.Span.Length);
Assert.Equal(new byte[] { 1, 2, 3 }, Read());
}
private static void Default(ref WritableBuffer outputBuffer)
{
var writer = new WritableBufferWriter(outputBuffer);
// HTTP 1.1 OK
writer.Write(_http11OK);
// Server headers
writer.Write(_headerServer);
writer.Write(_crlf);
// Date header
writer.Write(_headerDate);
writer.Write(_dateHeaderValueManager.GetDateHeaderValues().Bytes);
writer.Write(_crlf);
// Content-Length 0
writer.Write(_headerContentLengthZero);
writer.Write(_crlf);
// End of headers
writer.Write(_crlf);
}
private static void WriteNumericMultiWrite(ref WritableBufferWriter buffer, ulong number)
{
const byte AsciiDigitStart = (byte)'0';
var value = number;
var position = _maxULongByteLength;
var byteBuffer = NumericBytesScratch;
do
{
// Consider using Math.DivRem() if available
var quotient = value / 10;
byteBuffer[--position] = (byte)(AsciiDigitStart + (value - quotient * 10)); // 0x30 = '0'
value = quotient;
}while (value != 0);
var length = _maxULongByteLength - position;
buffer.Write(byteBuffer, position, length);
}
public void WritesNumericAcrossSpanBoundaries(int gapSize)
{
var writerBuffer = _pipe.Writer.Alloc(100);
var writer = new WritableBufferWriter(writerBuffer);
// almost fill up the first block
var spacer = new byte[writer.Span.Length - gapSize];
writer.Write(spacer);
var bufferLength = writer.Span.Length;
PipelineExtensions.WriteNumeric(ref writer, ulong.MaxValue);
Assert.NotEqual(bufferLength, writer.Span.Length);
writerBuffer.FlushAsync().GetAwaiter().GetResult();
var reader = _pipe.Reader.ReadAsync().GetAwaiter().GetResult();
var numAsString = ulong.MaxValue.ToString();
var written = reader.Buffer.Slice(spacer.Length, numAsString.Length);
Assert.False(written.IsSingleSpan, "The buffer should cross spans");
AssertExtensions.Equal(Encoding.ASCII.GetBytes(numAsString), written.ToArray());
}
private Task WriteAsync(
ArraySegment <byte> buffer,
CancellationToken cancellationToken,
bool chunk = false)
{
var writableBuffer = default(WritableBuffer);
lock (_contextLock)
{
if (_completed)
{
return(Task.CompletedTask);
}
writableBuffer = _pipe.Writer.Alloc(1);
var writer = new WritableBufferWriter(writableBuffer);
if (buffer.Count > 0)
{
if (chunk)
{
ChunkWriter.WriteBeginChunkBytes(ref writer, buffer.Count);
}
writer.Write(buffer.Array, buffer.Offset, buffer.Count);
if (chunk)
{
ChunkWriter.WriteEndChunkBytes(ref writer);
}
}
writableBuffer.Commit();
}
return(FlushAsync(writableBuffer, cancellationToken));
}
public void WritesAsciiAcrossBlockBoundaries(int stringLength, int gapSize)
{
var testString = new string(' ', stringLength);
var writerBuffer = _pipe.Writer.Alloc(100);
var writer = new WritableBufferWriter(writerBuffer);
// almost fill up the first block
var spacer = new byte[writer.Span.Length - gapSize];
writer.Write(spacer);
Assert.Equal(gapSize, writer.Span.Length);
var bufferLength = writer.Span.Length;
PipelineExtensions.WriteAsciiNoValidation(ref writer, testString);
Assert.NotEqual(bufferLength, writer.Span.Length);
writerBuffer.FlushAsync().GetAwaiter().GetResult();
var reader = _pipe.Reader.ReadAsync().GetAwaiter().GetResult();
var written = reader.Buffer.Slice(spacer.Length, stringLength);
Assert.False(written.IsSingleSpan, "The buffer should cross spans");
AssertExtensions.Equal(Encoding.ASCII.GetBytes(testString), written.ToArray());
}
public static void WriteEndChunkBytes(ref WritableBufferWriter start)
{
start.Write(_endChunkBytes.Array, _endChunkBytes.Offset, _endChunkBytes.Count);
}
