mcs/class/System.Net.Http/System.Net.Http/HttpContent.cs

   1 //
   2 // HttpContent.cs
   3 //
   4 // Authors:
   5 //      Marek Safar  <marek.safar@gmail.com>
   6 //
   7 // Copyright (C) 2011 Xamarin Inc (http://www.xamarin.com)
   8 //
   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:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  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.
  27 //
  28
  29 using System.Net.Http.Headers;
  30 using System.IO;
  31 using System.Threading.Tasks;
  32 using System.Text;
  33
  34 namespace System.Net.Http
  35 {
  36         public abstract class HttpContent : IDisposable
  37         {
  38                 sealed class FixedMemoryStream : MemoryStream
  39                 {
  40                         readonly long maxSize;
  41
  42                         public FixedMemoryStream (long maxSize)
  43                                 : base ()
  44                         {
  45                                 this.maxSize = maxSize;
  46                         }
  47
  48                         void CheckOverflow (int count)
  49                         {
  50                                 if (Length + count > maxSize)
  51                                         throw new HttpRequestException (string.Format ("Cannot write more bytes to the buffer than the configured maximum buffer size: {0}", maxSize));
  52                         }
  53
  54                         public override void WriteByte (byte value)
  55                         {
  56                                 CheckOverflow (1);
  57                                 base.WriteByte (value);
  58                         }
  59
  60                         public override void Write (byte[] buffer, int offset, int count)
  61                         {
  62                                 CheckOverflow (count);
  63                                 base.Write (buffer, offset, count);
  64                         }
  65                 }
  66
  67                 FixedMemoryStream buffer;
  68                 Stream stream;
  69                 bool disposed;
  70                 HttpContentHeaders headers;
  71
  72                 public HttpContentHeaders Headers {
  73                         get {
  74                                 return headers ?? (headers = new HttpContentHeaders (this));
  75                         }
  76                 }
  77
  78                 public Task CopyToAsync (Stream stream)
  79                 {
  80                         return CopyToAsync (stream, null);
  81                 }
  82
  83                 public Task CopyToAsync (Stream stream, TransportContext context)
  84                 {
  85                         if (stream == null)
  86                                 throw new ArgumentNullException ("stream");
  87
  88                         return SerializeToStreamAsync (stream, context);
  89                 }
  90
  91                 protected async virtual Task<Stream> CreateContentReadStreamAsync ()
  92                 {
  93                         await LoadIntoBufferAsync ().ConfigureAwait (false);
  94                         return buffer;
  95                 }
  96
  97                 static FixedMemoryStream CreateFixedMemoryStream (long maxBufferSize)
  98                 {
  99                         return new FixedMemoryStream (maxBufferSize);
 100                 }
 101
 102                 public void Dispose ()
 103                 {
 104                         Dispose (true);
 105                 }
 106
 107                 protected virtual void Dispose (bool disposing)
 108                 {
 109                         if (disposing && !disposed) {
 110                                 disposed = true;
 111
 112                                 if (buffer != null)
 113                                         buffer.Dispose ();
 114                         }
 115                 }
 116
 117                 public Task LoadIntoBufferAsync ()
 118                 {
 119                         return LoadIntoBufferAsync (65536);
 120                 }
 121
 122                 public async Task LoadIntoBufferAsync (long maxBufferSize)
 123                 {
 124                         if (disposed)
 125                                 throw new ObjectDisposedException (GetType ().ToString ());
 126
 127                         if (buffer != null)
 128                                 return;
 129
 130                         buffer = CreateFixedMemoryStream (maxBufferSize);
 131                         await SerializeToStreamAsync (buffer, null).ConfigureAwait (false);
 132                         buffer.Seek (0, SeekOrigin.Begin);
 133                 }
 134
 135                 public async Task<Stream> ReadAsStreamAsync ()
 136                 {
 137                         if (disposed)
 138                                 throw new ObjectDisposedException (GetType ().ToString ());
 139
 140                         if (stream == null)
 141                                 stream = await CreateContentReadStreamAsync ().ConfigureAwait (false);
 142
 143                         return stream;
 144                 }
 145
 146                 public async Task<byte[]> ReadAsByteArrayAsync ()
 147                 {
 148                         await LoadIntoBufferAsync ().ConfigureAwait (false);
 149                         return buffer.ToArray ();
 150                 }
 151
 152                 public async Task<string> ReadAsStringAsync ()
 153                 {
 154                         await LoadIntoBufferAsync ().ConfigureAwait (false);
 155                         if (buffer.Length == 0)
 156                                 return string.Empty;
 157
 158                         Encoding encoding;
 159                         if (headers != null && headers.ContentType != null && headers.ContentType.CharSet != null) {
 160                                 encoding = Encoding.GetEncoding (headers.ContentType.CharSet);
 161                         } else {
 162                                 encoding = Encoding.UTF8;
 163                         }
 164
 165                         return encoding.GetString (buffer.GetBuffer (), 0, (int) buffer.Length);
 166                 }
 167
 168                 protected internal abstract Task SerializeToStreamAsync (Stream stream, TransportContext context);
 169                 protected internal abstract bool TryComputeLength (out long length);
 170         }
 171 }