DevWiki/websocket-sharp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactored ChunkedInputStream.cs

Browse Source
This commit is contained in:
sta 2014-05-16 15:48:56 +09:00
parent be85033f8a
commit 696cfd686d
3 changed files with 260 additions and 165 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

340
websocket-sharp/Net/ChunkedInputStream.cs
View File

@ -1,204 +1,214 @@
// #region License
// ChunkedInputStream.cs /*
// Copied from System.Net.ChunkedInputStream.cs * ChunkedInputStream.cs
// *
// Authors: * This code is derived from System.Net.ChunkedInputStream.cs of Mono
// Gonzalo Paniagua Javier (gonzalo@novell.com) * (http://www.mono-project.com).
// *
// Copyright (c) 2005 Novell, Inc (http://www.novell.com) * The MIT License
// *
// Permission is hereby granted, free of charge, to any person obtaining * Copyright (c) 2005 Novell, Inc. (http://www.novell.com)
// a copy of this software and associated documentation files (the * Copyright (c) 2012-2014 sta.blockhead
// "Software"), to deal in the Software without restriction, including *
// without limitation the rights to use, copy, modify, merge, publish, * Permission is hereby granted, free of charge, to any person obtaining a copy
// distribute, sublicense, and/or sell copies of the Software, and to * of this software and associated documentation files (the "Software"), to deal
// permit persons to whom the Software is furnished to do so, subject to * in the Software without restriction, including without limitation the rights
// the following conditions: * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// * copies of the Software, and to permit persons to whom the Software is
// The above copyright notice and this permission notice shall be * furnished to do so, subject to the following conditions:
// included in all copies or substantial portions of the Software. *
// * The above copyright notice and this permission notice shall be included in
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, * all copies or substantial portions of the Software.
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF *
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. * 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.
*/
#endregion
#region Authors
/*
* Authors:
* - Gonzalo Paniagua Javier <gonzalo@novell.com>
*/
#endregion
using System; using System;
using System.IO; using System.IO;
using System.Runtime.InteropServices;
namespace WebSocketSharp.Net { namespace WebSocketSharp.Net
{
internal class ChunkedInputStream : RequestStream
{
#region Private Const Fields
class ChunkedInputStream : RequestStream { private const int _bufferSize = 8192;
class ReadBufferState { #endregion
public HttpStreamAsyncResult Ares; #region Private Fields
public byte [] Buffer;
public int Count;
public int InitialCount;
public int Offset;
public ReadBufferState ( private HttpListenerContext _context;
byte [] buffer, int offset, int count, HttpStreamAsyncResult ares) private ChunkStream _decoder;
{ private bool _disposed;
Buffer = buffer; private bool _noMoreData;
Offset = offset;
Count = count;
InitialCount = count;
Ares = ares;
}
}
#region Fields #endregion
HttpListenerContext context; #region Public Constructors
ChunkStream decoder;
bool disposed;
bool no_more_data;
#endregion public ChunkedInputStream (
HttpListenerContext context, Stream stream, byte [] buffer, int offset, int length)
: base (stream, buffer, offset, length)
{
_context = context;
_decoder = new ChunkStream ((WebHeaderCollection) context.Request.Headers);
}
#region Constructor #endregion
public ChunkedInputStream ( #region Public Properties
HttpListenerContext context, Stream stream, byte [] buffer, int offset, int length)
: base (stream, buffer, offset, length)
{
this.context = context;
WebHeaderCollection coll = (WebHeaderCollection) context.Request.Headers;
decoder = new ChunkStream (coll);
}
#endregion public ChunkStream Decoder {
get {
return _decoder;
}
#region Property set {
_decoder = value;
}
}
public ChunkStream Decoder { #endregion
get { return decoder; }
set { decoder = value; }
}
#endregion #region Private Methods
#region Private Method private void onRead (IAsyncResult asyncResult)
{
var readState = (ReadBufferState) asyncResult.AsyncState;
var ares = readState.AsyncResult;
try {
var nread = base.EndRead (asyncResult);
_decoder.Write (ares.Buffer, ares.Offset, nread);
nread = _decoder.Read (readState.Buffer, readState.Offset, readState.Count);
readState.Offset += nread;
readState.Count -= nread;
if (readState.Count == 0 || !_decoder.WantMore || nread == 0) {
_noMoreData = !_decoder.WantMore && nread == 0;
ares.Count = readState.InitialCount - readState.Count;
ares.Complete ();
void OnRead (IAsyncResult base_ares) return;
{ }
var rb = (ReadBufferState) base_ares.AsyncState;
var ares = rb.Ares;
try {
int nread = base.EndRead (base_ares);
decoder.Write (ares.Buffer, ares.Offset, nread);
nread = decoder.Read (rb.Buffer, rb.Offset, rb.Count);
rb.Offset += nread;
rb.Count -= nread;
if (rb.Count == 0 || !decoder.WantMore || nread == 0) {
no_more_data = !decoder.WantMore && nread == 0;
ares.Count = rb.InitialCount - rb.Count;
ares.Complete ();
return;
}
ares.Offset = 0; ares.Offset = 0;
ares.Count = Math.Min (8192, decoder.ChunkLeft + 6); ares.Count = Math.Min (_bufferSize, _decoder.ChunkLeft + 6);
base.BeginRead (ares.Buffer, ares.Offset, ares.Count, OnRead, rb); base.BeginRead (ares.Buffer, ares.Offset, ares.Count, onRead, readState);
} catch (Exception e) { }
context.Connection.SendError (e.Message, 400); catch (Exception ex) {
ares.Complete (e); _context.Connection.SendError (ex.Message, 400);
} ares.Complete (ex);
} }
}
#endregion #endregion
#region Public Methods #region Public Methods
public override IAsyncResult BeginRead ( public override IAsyncResult BeginRead (
byte [] buffer, int offset, int count, AsyncCallback cback, object state) byte [] buffer, int offset, int count, AsyncCallback callback, object state)
{ {
if (disposed) if (_disposed)
throw new ObjectDisposedException (GetType ().ToString ()); throw new ObjectDisposedException (GetType ().ToString ());
if (buffer == null) if (buffer == null)
throw new ArgumentNullException ("buffer"); throw new ArgumentNullException ("buffer");
int len = buffer.Length; var len = buffer.Length;
if (offset < 0 || offset > len) if (offset < 0 || offset > len)
throw new ArgumentOutOfRangeException ("'offset' exceeds the size of buffer."); throw new ArgumentOutOfRangeException ("'offset' exceeds the size of buffer.");
if (count < 0 || offset > len - count) if (count < 0 || offset > len - count)
throw new ArgumentOutOfRangeException ("'offset' + 'count' exceeds the size of buffer."); throw new ArgumentOutOfRangeException ("'offset' + 'count' exceeds the size of buffer.");
var ares = new HttpStreamAsyncResult (); var ares = new HttpStreamAsyncResult ();
ares.Callback = cback; ares.Callback = callback;
ares.State = state; ares.State = state;
if (no_more_data) { if (_noMoreData) {
ares.Complete (); ares.Complete ();
return ares; return ares;
} }
int nread = decoder.Read (buffer, offset, count); var nread = _decoder.Read (buffer, offset, count);
offset += nread; offset += nread;
count -= nread; count -= nread;
if (count == 0) { if (count == 0) {
// got all we wanted, no need to bother the decoder yet // Got all we wanted, no need to bother the decoder yet.
ares.Count = nread; ares.Count = nread;
ares.Complete (); ares.Complete ();
return ares;
}
if (!decoder.WantMore) { return ares;
no_more_data = nread == 0; }
ares.Count = nread;
ares.Complete ();
return ares;
}
ares.Buffer = new byte [8192]; if (!_decoder.WantMore) {
ares.Offset = 0; _noMoreData = nread == 0;
ares.Count = 8192; ares.Count = nread;
var rb = new ReadBufferState (buffer, offset, count, ares); ares.Complete ();
rb.InitialCount += nread;
base.BeginRead (ares.Buffer, ares.Offset, ares.Count, OnRead, rb);
return ares;
}
public override void Close () return ares;
{ }
if (!disposed) {
disposed = true;
base.Close ();
}
}
public override int EndRead (IAsyncResult ares) ares.Buffer = new byte [_bufferSize];
{ ares.Offset = 0;
if (disposed) ares.Count = _bufferSize;
throw new ObjectDisposedException (GetType ().ToString ());
if (ares == null) var readState = new ReadBufferState (buffer, offset, count, ares);
throw new ArgumentException ("Invalid IAsyncResult.", "ares"); readState.InitialCount += nread;
base.BeginRead (ares.Buffer, ares.Offset, ares.Count, onRead, readState);
if (!ares.IsCompleted) return ares;
ares.AsyncWaitHandle.WaitOne (); }
var ares_ = ares as HttpStreamAsyncResult; public override void Close ()
if (ares_.Error != null) {
throw new HttpListenerException (400, "I/O operation aborted."); if (_disposed)
return;
return ares_.Count; _disposed = true;
} base.Close ();
}
public override int Read ([In,Out] byte [] buffer, int offset, int count) public override int EndRead (IAsyncResult asyncResult)
{ {
var ares = BeginRead (buffer, offset, count, null, null); if (_disposed)
return EndRead (ares); throw new ObjectDisposedException (GetType ().ToString ());
}
#endregion if (asyncResult == null)
} throw new ArgumentNullException ("asyncResult");
var ares = asyncResult as HttpStreamAsyncResult;
if (ares == null)
throw new ArgumentException ("Wrong IAsyncResult.", "asyncResult");
if (!ares.IsCompleted)
ares.AsyncWaitHandle.WaitOne ();
if (ares.Error != null)
throw new HttpListenerException (400, "I/O operation aborted.");
return ares.Count;
}
public override int Read (byte [] buffer, int offset, int count)
{
var ares = BeginRead (buffer, offset, count, null, null);
return EndRead (ares);
}
#endregion
}
} }

84
websocket-sharp/Net/ReadBufferState.cs Normal file
View File

@ -0,0 +1,84 @@
#region License
/*
* ReadBufferState.cs
*
* This code is derived from System.Net.ChunkedInputStream.cs of Mono
* (http://www.mono-project.com).
*
* The MIT License
*
* Copyright (c) 2005 Novell, Inc. (http://www.novell.com)
* Copyright (c) 2014 sta.blockhead
*
* 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.
*/
#endregion
#region Authors
/*
* Authors:
* - Gonzalo Paniagua Javier <gonzalo@novell.com>
*/
#endregion
using System;
namespace WebSocketSharp.Net
{
internal class ReadBufferState
{
#region Public Constructors
public ReadBufferState (
byte [] buffer, int offset, int count, HttpStreamAsyncResult asyncResult)
{
Buffer = buffer;
Offset = offset;
Count = count;
InitialCount = count;
AsyncResult = asyncResult;
}
#endregion
#region Public Properties
public HttpStreamAsyncResult AsyncResult {
get; set;
}
public byte [] Buffer {
get; set;
}
public int Count {
get; set;
}
public int InitialCount {
get; set;
}
public int Offset {
get; set;
}
#endregion
}
}

1
websocket-sharp/websocket-sharp.csproj
View File

@ -131,6 +131,7 @@
<Compile Include="Net\InputState.cs" /> <Compile Include="Net\InputState.cs" />
<Compile Include="Net\LineState.cs" /> <Compile Include="Net\LineState.cs" />
<Compile Include="WebSocketStream.cs" /> <Compile Include="WebSocketStream.cs" />
<Compile Include="Net\ReadBufferState.cs" />
</ItemGroup> </ItemGroup>
<Import Project="$(MSBuildBinPath)\Microsoft.CSharp.targets" /> <Import Project="$(MSBuildBinPath)\Microsoft.CSharp.targets" />
<ItemGroup> <ItemGroup>