2 // System.Net.ResponseStream
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.
32 using System.Net.Sockets;
34 using System.Runtime.InteropServices;
35 namespace System.Net {
36 // FIXME: Does this buffer the response until Close?
37 // Update: we send a single packet for the first non-chunked Write
38 // What happens when we set content-length to X and write X-1 bytes then close?
39 // what if we don't set content-length at all?
40 class ResponseStream : Stream
42 HttpListenerResponse response;
48 internal ResponseStream (Stream stream, HttpListenerResponse response, bool ignore_errors)
50 this.response = response;
51 this.ignore_errors = ignore_errors;
55 public override bool CanRead {
59 public override bool CanSeek {
63 public override bool CanWrite {
67 public override long Length {
68 get { throw new NotSupportedException (); }
71 public override long Position {
72 get { throw new NotSupportedException (); }
73 set { throw new NotSupportedException (); }
77 public override void Close ()
79 if (disposed == false) {
82 MemoryStream ms = GetHeaders (true);
83 bool chunked = response.SendChunked;
84 if (stream.CanWrite) {
87 long start = ms.Position;
88 if (chunked && !trailer_sent) {
89 bytes = GetChunkSizeBytes (0, true);
90 ms.Position = ms.Length;
91 ms.Write (bytes, 0, bytes.Length);
93 InternalWrite (ms.GetBuffer (), (int) start, (int) (ms.Length - start));
95 } else if (chunked && !trailer_sent) {
96 bytes = GetChunkSizeBytes (0, true);
97 InternalWrite (bytes, 0, bytes.Length);
100 } catch (IOException ex) {
101 // Ignore error due to connection reset by peer
108 MemoryStream GetHeaders (bool closing)
110 // SendHeaders works on shared headers
111 lock (response.headers_lock) {
112 if (response.HeadersSent)
114 MemoryStream ms = new MemoryStream ();
115 response.SendHeaders (closing, ms);
120 public override void Flush ()
124 static byte [] crlf = new byte [] { 13, 10 };
125 static byte [] GetChunkSizeBytes (int size, bool final)
127 string str = String.Format ("{0:x}\r\n{1}", size, final ? "\r\n" : "");
128 return Encoding.ASCII.GetBytes (str);
131 internal void InternalWrite (byte [] buffer, int offset, int count)
135 stream.Write (buffer, offset, count);
138 stream.Write (buffer, offset, count);
142 public override void Write (byte [] buffer, int offset, int count)
145 throw new ObjectDisposedException (GetType ().ToString ());
147 byte [] bytes = null;
148 MemoryStream ms = GetHeaders (false);
149 bool chunked = response.SendChunked;
151 long start = ms.Position; // After the possible preamble for the encoding
152 ms.Position = ms.Length;
154 bytes = GetChunkSizeBytes (count, false);
155 ms.Write (bytes, 0, bytes.Length);
158 int new_count = Math.Min (count, 16384 - (int) ms.Position + (int) start);
159 ms.Write (buffer, offset, new_count);
162 InternalWrite (ms.GetBuffer (), (int) start, (int) (ms.Length - start));
164 ms.Capacity = 0; // 'dispose' the buffer in ms.
165 } else if (chunked) {
166 bytes = GetChunkSizeBytes (count, false);
167 InternalWrite (bytes, 0, bytes.Length);
171 InternalWrite (buffer, offset, count);
173 InternalWrite (crlf, 0, 2);
176 public override IAsyncResult BeginWrite (byte [] buffer, int offset, int count,
177 AsyncCallback cback, object state)
180 throw new ObjectDisposedException (GetType ().ToString ());
182 byte [] bytes = null;
183 MemoryStream ms = GetHeaders (false);
184 bool chunked = response.SendChunked;
186 long start = ms.Position;
187 ms.Position = ms.Length;
189 bytes = GetChunkSizeBytes (count, false);
190 ms.Write (bytes, 0, bytes.Length);
192 ms.Write (buffer, offset, count);
193 buffer = ms.GetBuffer ();
194 offset = (int) start;
195 count = (int) (ms.Position - start);
196 } else if (chunked) {
197 bytes = GetChunkSizeBytes (count, false);
198 InternalWrite (bytes, 0, bytes.Length);
201 return stream.BeginWrite (buffer, offset, count, cback, state);
204 public override void EndWrite (IAsyncResult ares)
207 throw new ObjectDisposedException (GetType ().ToString ());
211 stream.EndWrite (ares);
212 if (response.SendChunked)
213 stream.Write (crlf, 0, 2);
216 stream.EndWrite (ares);
217 if (response.SendChunked)
218 stream.Write (crlf, 0, 2);
222 public override int Read ([In,Out] byte[] buffer, int offset, int count)
224 throw new NotSupportedException ();
227 public override IAsyncResult BeginRead (byte [] buffer, int offset, int count,
228 AsyncCallback cback, object state)
230 throw new NotSupportedException ();
233 public override int EndRead (IAsyncResult ares)
235 throw new NotSupportedException ();
238 public override long Seek (long offset, SeekOrigin origin)
240 throw new NotSupportedException ();
243 public override void SetLength (long value)
245 throw new NotSupportedException ();