2 // System.Net.ChunkedInputStream
5 // Gonzalo Paniagua Javier (gonzalo@novell.com)
7 // Copyright (c) 2005 Novell, Inc (http://www.novell.com)
9 // Permission is hereby granted, free of charge, to any person obtaining
10 // a copy of this software and associated documentation files (the
11 // "Software"), to deal in the Software without restriction, including
12 // without limitation the rights to use, copy, modify, merge, publish,
13 // distribute, sublicense, and/or sell copies of the Software, and to
14 // permit persons to whom the Software is furnished to do so, subject to
15 // the following conditions:
17 // The above copyright notice and this permission notice shall be
18 // included in all copies or substantial portions of the Software.
20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
28 #if NET_2_0 && SECURITY_DEP
31 using System.Net.Sockets;
32 using System.Runtime.InteropServices;
33 namespace System.Net {
34 class ChunkedInputStream : RequestStream {
37 HttpListenerContext context;
40 class ReadBufferState {
41 public byte [] Buffer;
44 public int InitialCount;
45 public HttpStreamAsyncResult Ares;
46 public ReadBufferState (byte [] buffer, int offset, int count,
47 HttpStreamAsyncResult ares)
57 public ChunkedInputStream (HttpListenerContext context, Stream stream,
58 byte [] buffer, int offset, int length)
59 : base (stream, buffer, offset, length)
61 this.context = context;
62 WebHeaderCollection coll = (WebHeaderCollection) context.Request.Headers;
63 decoder = new ChunkStream (coll);
66 public ChunkStream Decoder {
67 get { return decoder; }
68 set { decoder = value; }
71 public override int Read ([In,Out] byte [] buffer, int offset, int count)
73 IAsyncResult ares = BeginRead (buffer, offset, count, null, null);
74 return EndRead (ares);
77 public override IAsyncResult BeginRead (byte [] buffer, int offset, int count,
78 AsyncCallback cback, object state)
81 throw new ObjectDisposedException (GetType ().ToString ());
84 throw new ArgumentNullException ("buffer");
86 int len = buffer.Length;
87 if (offset < 0 || offset > len)
88 throw new ArgumentOutOfRangeException ("offset exceeds the size of buffer");
90 if (count < 0 || offset > len - count)
91 throw new ArgumentOutOfRangeException ("offset+size exceeds the size of buffer");
93 HttpStreamAsyncResult ares = new HttpStreamAsyncResult ();
94 ares.Callback = cback;
100 int nread = decoder.Read (buffer, offset, count);
104 // got all we wanted, no need to bother the decoder yet
109 if (!decoder.WantMore) {
110 no_more_data = nread == 0;
115 ares.Buffer = new byte [8192];
118 ReadBufferState rb = new ReadBufferState (buffer, offset, count, ares);
119 rb.InitialCount += nread;
120 base.BeginRead (ares.Buffer, ares.Offset, ares.Count, OnRead, rb);
124 void OnRead (IAsyncResult base_ares)
126 ReadBufferState rb = (ReadBufferState) base_ares.AsyncState;
127 HttpStreamAsyncResult ares = rb.Ares;
129 int nread = base.EndRead (base_ares);
130 decoder.Write (ares.Buffer, ares.Offset, nread);
131 nread = decoder.Read (rb.Buffer, rb.Offset, rb.Count);
134 if (rb.Count == 0 || !decoder.WantMore || nread == 0) {
135 no_more_data = !decoder.WantMore && nread == 0;
136 ares.Count = rb.InitialCount - rb.Count;
141 ares.Count = Math.Min (8192, decoder.ChunkLeft + 6);
142 base.BeginRead (ares.Buffer, ares.Offset, ares.Count, OnRead, rb);
143 } catch (Exception e) {
144 context.Connection.SendError (e.Message, 400);
149 public override int EndRead (IAsyncResult ares)
152 throw new ObjectDisposedException (GetType ().ToString ());
154 HttpStreamAsyncResult my_ares = ares as HttpStreamAsyncResult;
156 throw new ArgumentException ("Invalid IAsyncResult", "ares");
158 if (!ares.IsCompleted)
159 ares.AsyncWaitHandle.WaitOne ();
161 if (my_ares.Error != null)
162 throw new HttpListenerException (400, "I/O operation aborted.");
164 return my_ares.Count;
167 public override void Close ()