aboutsummaryrefslogtreecommitdiff
path: root/SocketHttpListener/Net/ChunkStream.cs
diff options
context:
space:
mode:
Diffstat (limited to 'SocketHttpListener/Net/ChunkStream.cs')
-rw-r--r--SocketHttpListener/Net/ChunkStream.cs385
1 files changed, 0 insertions, 385 deletions
diff --git a/SocketHttpListener/Net/ChunkStream.cs b/SocketHttpListener/Net/ChunkStream.cs
deleted file mode 100644
index 3836947d4..000000000
--- a/SocketHttpListener/Net/ChunkStream.cs
+++ /dev/null
@@ -1,385 +0,0 @@
-using System;
-using System.Collections.Generic;
-using System.Globalization;
-using System.IO;
-using System.Net;
-using System.Text;
-
-namespace SocketHttpListener.Net
-{
- // Licensed to the .NET Foundation under one or more agreements.
- // See the LICENSE file in the project root for more information.
- //
- // System.Net.ResponseStream
- //
- // Author:
- // Gonzalo Paniagua Javier (gonzalo@novell.com)
- //
- // Copyright (c) 2005 Novell, Inc. (http://www.novell.com)
- //
- // Permission is hereby granted, free of charge, to any person obtaining
- // a copy of this software and associated documentation files (the
- // "Software"), to deal in the Software without restriction, including
- // without limitation the rights to use, copy, modify, merge, publish,
- // distribute, sublicense, and/or sell copies of the Software, and to
- // permit persons to whom the Software is furnished to do so, subject to
- // the following conditions:
- //
- // The above copyright notice and this permission notice shall be
- // included in all copies or substantial portions of the Software.
- //
- // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
- // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
- // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
- // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
- // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
- // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
- // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
- //
-
- internal sealed class ChunkStream
- {
- private enum State
- {
- None,
- PartialSize,
- Body,
- BodyFinished,
- Trailer
- }
-
- private class Chunk
- {
- public byte[] Bytes;
- public int Offset;
-
- public Chunk(byte[] chunk)
- {
- Bytes = chunk;
- }
-
- public int Read(byte[] buffer, int offset, int size)
- {
- int nread = (size > Bytes.Length - Offset) ? Bytes.Length - Offset : size;
- Buffer.BlockCopy(Bytes, Offset, buffer, offset, nread);
- Offset += nread;
- return nread;
- }
- }
-
- internal WebHeaderCollection _headers;
- private int _chunkSize;
- private int _chunkRead;
- private int _totalWritten;
- private State _state;
- private StringBuilder _saved;
- private bool _sawCR;
- private bool _gotit;
- private int _trailerState;
- private List<Chunk> _chunks;
-
- public ChunkStream(WebHeaderCollection headers)
- {
- _headers = headers;
- _saved = new StringBuilder();
- _chunks = new List<Chunk>();
- _chunkSize = -1;
- _totalWritten = 0;
- }
-
- public void ResetBuffer()
- {
- _chunkSize = -1;
- _chunkRead = 0;
- _totalWritten = 0;
- _chunks.Clear();
- }
-
- public int Read(byte[] buffer, int offset, int size)
- {
- return ReadFromChunks(buffer, offset, size);
- }
-
- private int ReadFromChunks(byte[] buffer, int offset, int size)
- {
- int count = _chunks.Count;
- int nread = 0;
-
- var chunksForRemoving = new List<Chunk>(count);
- for (int i = 0; i < count; i++)
- {
- var chunk = _chunks[i];
-
- if (chunk.Offset == chunk.Bytes.Length)
- {
- chunksForRemoving.Add(chunk);
- continue;
- }
-
- nread += chunk.Read(buffer, offset + nread, size - nread);
- if (nread == size)
- break;
- }
-
- foreach (var chunk in chunksForRemoving)
- _chunks.Remove(chunk);
-
- return nread;
- }
-
- public void Write(byte[] buffer, int offset, int size)
- {
- // Note, the logic here only works when offset is 0 here.
- // Otherwise, it would treat "size" as the end offset instead of an actual byte count from offset.
-
- if (offset < size)
- InternalWrite(buffer, ref offset, size);
- }
-
- private void InternalWrite(byte[] buffer, ref int offset, int size)
- {
- if (_state == State.None || _state == State.PartialSize)
- {
- _state = GetChunkSize(buffer, ref offset, size);
- if (_state == State.PartialSize)
- return;
-
- _saved.Length = 0;
- _sawCR = false;
- _gotit = false;
- }
-
- if (_state == State.Body && offset < size)
- {
- _state = ReadBody(buffer, ref offset, size);
- if (_state == State.Body)
- return;
- }
-
- if (_state == State.BodyFinished && offset < size)
- {
- _state = ReadCRLF(buffer, ref offset, size);
- if (_state == State.BodyFinished)
- return;
-
- _sawCR = false;
- }
-
- if (_state == State.Trailer && offset < size)
- {
- _state = ReadTrailer(buffer, ref offset, size);
- if (_state == State.Trailer)
- return;
-
- _saved.Length = 0;
- _sawCR = false;
- _gotit = false;
- }
-
- if (offset < size)
- InternalWrite(buffer, ref offset, size);
- }
-
- public bool WantMore => (_chunkRead != _chunkSize || _chunkSize != 0 || _state != State.None);
-
- public bool DataAvailable
- {
- get
- {
- int count = _chunks.Count;
- for (int i = 0; i < count; i++)
- {
- var ch = _chunks[i];
- if (ch == null || ch.Bytes == null)
- continue;
- if (ch.Bytes.Length > 0 && ch.Offset < ch.Bytes.Length)
- return (_state != State.Body);
- }
- return false;
- }
- }
-
- public int TotalDataSize => _totalWritten;
-
- public int ChunkLeft => _chunkSize - _chunkRead;
-
- private State ReadBody(byte[] buffer, ref int offset, int size)
- {
- if (_chunkSize == 0)
- return State.BodyFinished;
-
- int diff = size - offset;
- if (diff + _chunkRead > _chunkSize)
- diff = _chunkSize - _chunkRead;
-
- byte[] chunk = new byte[diff];
- Buffer.BlockCopy(buffer, offset, chunk, 0, diff);
- _chunks.Add(new Chunk(chunk));
- offset += diff;
- _chunkRead += diff;
- _totalWritten += diff;
-
- return (_chunkRead == _chunkSize) ? State.BodyFinished : State.Body;
- }
-
- private State GetChunkSize(byte[] buffer, ref int offset, int size)
- {
- _chunkRead = 0;
- _chunkSize = 0;
- char c = '\0';
- while (offset < size)
- {
- c = (char)buffer[offset++];
- if (c == '\r')
- {
- if (_sawCR)
- ThrowProtocolViolation("2 CR found");
-
- _sawCR = true;
- continue;
- }
-
- if (_sawCR && c == '\n')
- break;
-
- if (c == ' ')
- _gotit = true;
-
- if (!_gotit)
- _saved.Append(c);
-
- if (_saved.Length > 20)
- ThrowProtocolViolation("chunk size too long.");
- }
-
- if (!_sawCR || c != '\n')
- {
- if (offset < size)
- ThrowProtocolViolation("Missing \\n");
-
- try
- {
- if (_saved.Length > 0)
- {
- _chunkSize = int.Parse(RemoveChunkExtension(_saved.ToString()), NumberStyles.HexNumber);
- }
- }
- catch (Exception)
- {
- ThrowProtocolViolation("Cannot parse chunk size.");
- }
-
- return State.PartialSize;
- }
-
- _chunkRead = 0;
- try
- {
- _chunkSize = int.Parse(RemoveChunkExtension(_saved.ToString()), NumberStyles.HexNumber);
- }
- catch (Exception)
- {
- ThrowProtocolViolation("Cannot parse chunk size.");
- }
-
- if (_chunkSize == 0)
- {
- _trailerState = 2;
- return State.Trailer;
- }
-
- return State.Body;
- }
-
- private static string RemoveChunkExtension(string input)
- {
- int idx = input.IndexOf(';');
- if (idx == -1)
- return input;
- return input.Substring(0, idx);
- }
-
- private State ReadCRLF(byte[] buffer, ref int offset, int size)
- {
- if (!_sawCR)
- {
- if ((char)buffer[offset++] != '\r')
- ThrowProtocolViolation("Expecting \\r");
-
- _sawCR = true;
- if (offset == size)
- return State.BodyFinished;
- }
-
- if (_sawCR && (char)buffer[offset++] != '\n')
- ThrowProtocolViolation("Expecting \\n");
-
- return State.None;
- }
-
- private State ReadTrailer(byte[] buffer, ref int offset, int size)
- {
- char c = '\0';
-
- // short path
- if (_trailerState == 2 && (char)buffer[offset] == '\r' && _saved.Length == 0)
- {
- offset++;
- if (offset < size && (char)buffer[offset] == '\n')
- {
- offset++;
- return State.None;
- }
- offset--;
- }
-
- int st = _trailerState;
- string stString = "\r\n\r";
- while (offset < size && st < 4)
- {
- c = (char)buffer[offset++];
- if ((st == 0 || st == 2) && c == '\r')
- {
- st++;
- continue;
- }
-
- if ((st == 1 || st == 3) && c == '\n')
- {
- st++;
- continue;
- }
-
- if (st > 0)
- {
- _saved.Append(stString.Substring(0, _saved.Length == 0 ? st - 2 : st));
- st = 0;
- if (_saved.Length > 4196)
- ThrowProtocolViolation("Error reading trailer (too long).");
- }
- }
-
- if (st < 4)
- {
- _trailerState = st;
- if (offset < size)
- ThrowProtocolViolation("Error reading trailer.");
-
- return State.Trailer;
- }
-
- var reader = new StringReader(_saved.ToString());
- string line;
- while ((line = reader.ReadLine()) != null && line != "")
- _headers.Add(line);
-
- return State.None;
- }
-
- private static void ThrowProtocolViolation(string message)
- {
- var we = new WebException(message, null, WebExceptionStatus.ServerProtocolViolation, null);
- throw we;
- }
- }
-}