572 lines
18 KiB
C#
572 lines
18 KiB
C#
using System;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Net;
|
|
using System.Text;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using MediaBrowser.Model.IO;
|
|
using MediaBrowser.Model.Logging;
|
|
using MediaBrowser.Model.Text;
|
|
using SocketHttpListener.Primitives;
|
|
|
|
namespace SocketHttpListener.Net
|
|
{
|
|
public sealed class HttpListenerResponse : IDisposable
|
|
{
|
|
bool disposed;
|
|
Encoding content_encoding;
|
|
long content_length;
|
|
bool cl_set;
|
|
string content_type;
|
|
CookieCollection cookies;
|
|
WebHeaderCollection headers = new WebHeaderCollection();
|
|
bool keep_alive = true;
|
|
Stream output_stream;
|
|
Version version = HttpVersion.Version11;
|
|
string location;
|
|
int status_code = 200;
|
|
string status_description = "OK";
|
|
bool chunked;
|
|
HttpListenerContext context;
|
|
|
|
internal bool HeadersSent;
|
|
internal object headers_lock = new object();
|
|
|
|
private readonly ILogger _logger;
|
|
private readonly ITextEncoding _textEncoding;
|
|
private readonly IFileSystem _fileSystem;
|
|
|
|
internal HttpListenerResponse(HttpListenerContext context, ILogger logger, ITextEncoding textEncoding, IFileSystem fileSystem)
|
|
{
|
|
this.context = context;
|
|
_logger = logger;
|
|
_textEncoding = textEncoding;
|
|
_fileSystem = fileSystem;
|
|
}
|
|
|
|
internal bool CloseConnection
|
|
{
|
|
get
|
|
{
|
|
return headers["Connection"] == "close";
|
|
}
|
|
}
|
|
|
|
public bool ForceCloseChunked
|
|
{
|
|
get { return false; }
|
|
}
|
|
|
|
public Encoding ContentEncoding
|
|
{
|
|
get
|
|
{
|
|
if (content_encoding == null)
|
|
content_encoding = _textEncoding.GetDefaultEncoding();
|
|
return content_encoding;
|
|
}
|
|
set
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException(GetType().ToString());
|
|
|
|
content_encoding = value;
|
|
}
|
|
}
|
|
|
|
public long ContentLength64
|
|
{
|
|
get { return content_length; }
|
|
set
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException(GetType().ToString());
|
|
|
|
if (HeadersSent)
|
|
throw new InvalidOperationException("Cannot be changed after headers are sent.");
|
|
|
|
if (value < 0)
|
|
throw new ArgumentOutOfRangeException("Must be >= 0", "value");
|
|
|
|
cl_set = true;
|
|
content_length = value;
|
|
}
|
|
}
|
|
|
|
public string ContentType
|
|
{
|
|
get { return content_type; }
|
|
set
|
|
{
|
|
// TODO: is null ok?
|
|
if (disposed)
|
|
throw new ObjectDisposedException(GetType().ToString());
|
|
|
|
content_type = value;
|
|
}
|
|
}
|
|
|
|
// RFC 2109, 2965 + the netscape specification at http://wp.netscape.com/newsref/std/cookie_spec.html
|
|
public CookieCollection Cookies
|
|
{
|
|
get
|
|
{
|
|
if (cookies == null)
|
|
cookies = new CookieCollection();
|
|
return cookies;
|
|
}
|
|
set { cookies = value; } // null allowed?
|
|
}
|
|
|
|
public WebHeaderCollection Headers
|
|
{
|
|
get { return headers; }
|
|
set
|
|
{
|
|
/**
|
|
* "If you attempt to set a Content-Length, Keep-Alive, Transfer-Encoding, or
|
|
* WWW-Authenticate header using the Headers property, an exception will be
|
|
* thrown. Use the KeepAlive or ContentLength64 properties to set these headers.
|
|
* You cannot set the Transfer-Encoding or WWW-Authenticate headers manually."
|
|
*/
|
|
// TODO: check if this is marked readonly after headers are sent.
|
|
headers = value;
|
|
}
|
|
}
|
|
|
|
public bool KeepAlive
|
|
{
|
|
get { return keep_alive; }
|
|
set
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException(GetType().ToString());
|
|
|
|
keep_alive = value;
|
|
}
|
|
}
|
|
|
|
public Stream OutputStream
|
|
{
|
|
get
|
|
{
|
|
if (output_stream == null)
|
|
output_stream = context.Connection.GetResponseStream();
|
|
return output_stream;
|
|
}
|
|
}
|
|
|
|
public Version ProtocolVersion
|
|
{
|
|
get { return version; }
|
|
set
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException(GetType().ToString());
|
|
|
|
if (value == null)
|
|
throw new ArgumentNullException("value");
|
|
|
|
if (value.Major != 1 || (value.Minor != 0 && value.Minor != 1))
|
|
throw new ArgumentException("Must be 1.0 or 1.1", "value");
|
|
|
|
if (disposed)
|
|
throw new ObjectDisposedException(GetType().ToString());
|
|
|
|
version = value;
|
|
}
|
|
}
|
|
|
|
public string RedirectLocation
|
|
{
|
|
get { return location; }
|
|
set
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException(GetType().ToString());
|
|
|
|
location = value;
|
|
}
|
|
}
|
|
|
|
public bool SendChunked
|
|
{
|
|
get { return chunked; }
|
|
set
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException(GetType().ToString());
|
|
|
|
chunked = value;
|
|
}
|
|
}
|
|
|
|
public int StatusCode
|
|
{
|
|
get { return status_code; }
|
|
set
|
|
{
|
|
if (disposed)
|
|
throw new ObjectDisposedException(GetType().ToString());
|
|
|
|
if (value < 100 || value > 999)
|
|
throw new ProtocolViolationException("StatusCode must be between 100 and 999.");
|
|
status_code = value;
|
|
status_description = GetStatusDescription(value);
|
|
}
|
|
}
|
|
|
|
internal static string GetStatusDescription(int code)
|
|
{
|
|
switch (code)
|
|
{
|
|
case 100: return "Continue";
|
|
case 101: return "Switching Protocols";
|
|
case 102: return "Processing";
|
|
case 200: return "OK";
|
|
case 201: return "Created";
|
|
case 202: return "Accepted";
|
|
case 203: return "Non-Authoritative Information";
|
|
case 204: return "No Content";
|
|
case 205: return "Reset Content";
|
|
case 206: return "Partial Content";
|
|
case 207: return "Multi-Status";
|
|
case 300: return "Multiple Choices";
|
|
case 301: return "Moved Permanently";
|
|
case 302: return "Found";
|
|
case 303: return "See Other";
|
|
case 304: return "Not Modified";
|
|
case 305: return "Use Proxy";
|
|
case 307: return "Temporary Redirect";
|
|
case 400: return "Bad Request";
|
|
case 401: return "Unauthorized";
|
|
case 402: return "Payment Required";
|
|
case 403: return "Forbidden";
|
|
case 404: return "Not Found";
|
|
case 405: return "Method Not Allowed";
|
|
case 406: return "Not Acceptable";
|
|
case 407: return "Proxy Authentication Required";
|
|
case 408: return "Request Timeout";
|
|
case 409: return "Conflict";
|
|
case 410: return "Gone";
|
|
case 411: return "Length Required";
|
|
case 412: return "Precondition Failed";
|
|
case 413: return "Request Entity Too Large";
|
|
case 414: return "Request-Uri Too Long";
|
|
case 415: return "Unsupported Media Type";
|
|
case 416: return "Requested Range Not Satisfiable";
|
|
case 417: return "Expectation Failed";
|
|
case 422: return "Unprocessable Entity";
|
|
case 423: return "Locked";
|
|
case 424: return "Failed Dependency";
|
|
case 500: return "Internal Server Error";
|
|
case 501: return "Not Implemented";
|
|
case 502: return "Bad Gateway";
|
|
case 503: return "Service Unavailable";
|
|
case 504: return "Gateway Timeout";
|
|
case 505: return "Http Version Not Supported";
|
|
case 507: return "Insufficient Storage";
|
|
}
|
|
return "";
|
|
}
|
|
|
|
public string StatusDescription
|
|
{
|
|
get { return status_description; }
|
|
set
|
|
{
|
|
status_description = value;
|
|
}
|
|
}
|
|
|
|
void IDisposable.Dispose()
|
|
{
|
|
Close(true); //TODO: Abort or Close?
|
|
}
|
|
|
|
public void Abort()
|
|
{
|
|
if (disposed)
|
|
return;
|
|
|
|
Close(true);
|
|
}
|
|
|
|
public void AddHeader(string name, string value)
|
|
{
|
|
if (name == null)
|
|
throw new ArgumentNullException("name");
|
|
|
|
if (name == "")
|
|
throw new ArgumentException("'name' cannot be empty", "name");
|
|
|
|
//TODO: check for forbidden headers and invalid characters
|
|
if (value.Length > 65535)
|
|
throw new ArgumentOutOfRangeException("value");
|
|
|
|
headers.Set(name, value);
|
|
}
|
|
|
|
public void AppendCookie(Cookie cookie)
|
|
{
|
|
if (cookie == null)
|
|
throw new ArgumentNullException("cookie");
|
|
|
|
Cookies.Add(cookie);
|
|
}
|
|
|
|
public void AppendHeader(string name, string value)
|
|
{
|
|
if (name == null)
|
|
throw new ArgumentNullException("name");
|
|
|
|
if (name == "")
|
|
throw new ArgumentException("'name' cannot be empty", "name");
|
|
|
|
if (value.Length > 65535)
|
|
throw new ArgumentOutOfRangeException("value");
|
|
|
|
headers.Add(name, value);
|
|
}
|
|
|
|
private void Close(bool force)
|
|
{
|
|
if (force)
|
|
{
|
|
_logger.Debug("HttpListenerResponse force closing HttpConnection");
|
|
}
|
|
disposed = true;
|
|
context.Connection.Close(force);
|
|
}
|
|
|
|
public void Close(byte[] responseEntity, bool willBlock)
|
|
{
|
|
//CheckDisposed();
|
|
|
|
if (responseEntity == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(responseEntity));
|
|
}
|
|
|
|
//if (_boundaryType != BoundaryType.Chunked)
|
|
{
|
|
ContentLength64 = responseEntity.Length;
|
|
}
|
|
|
|
if (willBlock)
|
|
{
|
|
try
|
|
{
|
|
OutputStream.Write(responseEntity, 0, responseEntity.Length);
|
|
}
|
|
finally
|
|
{
|
|
Close(false);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
OutputStream.BeginWrite(responseEntity, 0, responseEntity.Length, iar =>
|
|
{
|
|
var thisRef = (HttpListenerResponse)iar.AsyncState;
|
|
try
|
|
{
|
|
thisRef.OutputStream.EndWrite(iar);
|
|
}
|
|
finally
|
|
{
|
|
thisRef.Close(false);
|
|
}
|
|
}, this);
|
|
}
|
|
}
|
|
|
|
public void Close()
|
|
{
|
|
if (disposed)
|
|
return;
|
|
|
|
Close(false);
|
|
}
|
|
|
|
public void Redirect(string url)
|
|
{
|
|
StatusCode = 302; // Found
|
|
location = url;
|
|
}
|
|
|
|
bool FindCookie(Cookie cookie)
|
|
{
|
|
string name = cookie.Name;
|
|
string domain = cookie.Domain;
|
|
string path = cookie.Path;
|
|
foreach (Cookie c in cookies)
|
|
{
|
|
if (name != c.Name)
|
|
continue;
|
|
if (domain != c.Domain)
|
|
continue;
|
|
if (path == c.Path)
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public void DetermineIfChunked()
|
|
{
|
|
if (chunked)
|
|
{
|
|
return;
|
|
}
|
|
|
|
Version v = context.Request.ProtocolVersion;
|
|
if (!cl_set && !chunked && v >= HttpVersion.Version11)
|
|
chunked = true;
|
|
if (!chunked && string.Equals(headers["Transfer-Encoding"], "chunked"))
|
|
{
|
|
chunked = true;
|
|
}
|
|
}
|
|
|
|
internal void SendHeaders(bool closing, MemoryStream ms)
|
|
{
|
|
Encoding encoding = content_encoding;
|
|
if (encoding == null)
|
|
encoding = _textEncoding.GetDefaultEncoding();
|
|
|
|
if (content_type != null)
|
|
{
|
|
if (content_encoding != null && content_type.IndexOf("charset=", StringComparison.OrdinalIgnoreCase) == -1)
|
|
{
|
|
string enc_name = content_encoding.WebName;
|
|
headers.SetInternal("Content-Type", content_type + "; charset=" + enc_name);
|
|
}
|
|
else
|
|
{
|
|
headers.SetInternal("Content-Type", content_type);
|
|
}
|
|
}
|
|
|
|
if (headers["Server"] == null)
|
|
headers.SetInternal("Server", "Mono-HTTPAPI/1.0");
|
|
|
|
CultureInfo inv = CultureInfo.InvariantCulture;
|
|
if (headers["Date"] == null)
|
|
headers.SetInternal("Date", DateTime.UtcNow.ToString("r", inv));
|
|
|
|
if (!chunked)
|
|
{
|
|
if (!cl_set && closing)
|
|
{
|
|
cl_set = true;
|
|
content_length = 0;
|
|
}
|
|
|
|
if (cl_set)
|
|
headers.SetInternal("Content-Length", content_length.ToString(inv));
|
|
}
|
|
|
|
Version v = context.Request.ProtocolVersion;
|
|
if (!cl_set && !chunked && v >= HttpVersion.Version11)
|
|
chunked = true;
|
|
|
|
/* Apache forces closing the connection for these status codes:
|
|
* HttpStatusCode.BadRequest 400
|
|
* HttpStatusCode.RequestTimeout 408
|
|
* HttpStatusCode.LengthRequired 411
|
|
* HttpStatusCode.RequestEntityTooLarge 413
|
|
* HttpStatusCode.RequestUriTooLong 414
|
|
* HttpStatusCode.InternalServerError 500
|
|
* HttpStatusCode.ServiceUnavailable 503
|
|
*/
|
|
bool conn_close = status_code == 400 || status_code == 408 || status_code == 411 ||
|
|
status_code == 413 || status_code == 414 ||
|
|
status_code == 500 ||
|
|
status_code == 503;
|
|
|
|
if (conn_close == false)
|
|
conn_close = !context.Request.KeepAlive;
|
|
|
|
// They sent both KeepAlive: true and Connection: close!?
|
|
if (!keep_alive || conn_close)
|
|
{
|
|
headers.SetInternal("Connection", "close");
|
|
conn_close = true;
|
|
}
|
|
|
|
if (chunked)
|
|
headers.SetInternal("Transfer-Encoding", "chunked");
|
|
|
|
//int reuses = context.Connection.Reuses;
|
|
//if (reuses >= 100)
|
|
//{
|
|
// _logger.Debug("HttpListenerResponse - keep alive has exceeded 100 uses and will be closed.");
|
|
|
|
// force_close_chunked = true;
|
|
// if (!conn_close)
|
|
// {
|
|
// headers.SetInternal("Connection", "close");
|
|
// conn_close = true;
|
|
// }
|
|
//}
|
|
|
|
if (!conn_close)
|
|
{
|
|
if (context.Request.ProtocolVersion <= HttpVersion.Version10)
|
|
headers.SetInternal("Connection", "keep-alive");
|
|
}
|
|
|
|
if (location != null)
|
|
headers.SetInternal("Location", location);
|
|
|
|
if (cookies != null)
|
|
{
|
|
foreach (Cookie cookie in cookies)
|
|
headers.SetInternal("Set-Cookie", cookie.ToString());
|
|
}
|
|
|
|
headers.SetInternal("Status", status_code.ToString(CultureInfo.InvariantCulture));
|
|
|
|
using (StreamWriter writer = new StreamWriter(ms, encoding, 256, true))
|
|
{
|
|
writer.Write("HTTP/{0} {1} {2}\r\n", version, status_code, status_description);
|
|
string headers_str = headers.ToStringMultiValue();
|
|
writer.Write(headers_str);
|
|
writer.Flush();
|
|
}
|
|
|
|
int preamble = encoding.GetPreamble().Length;
|
|
if (output_stream == null)
|
|
output_stream = context.Connection.GetResponseStream();
|
|
|
|
/* Assumes that the ms was at position 0 */
|
|
ms.Position = preamble;
|
|
HeadersSent = true;
|
|
}
|
|
|
|
public void SetCookie(Cookie cookie)
|
|
{
|
|
if (cookie == null)
|
|
throw new ArgumentNullException("cookie");
|
|
|
|
if (cookies != null)
|
|
{
|
|
if (FindCookie(cookie))
|
|
throw new ArgumentException("The cookie already exists.");
|
|
}
|
|
else
|
|
{
|
|
cookies = new CookieCollection();
|
|
}
|
|
|
|
cookies.Add(cookie);
|
|
}
|
|
|
|
public Task TransmitFile(string path, long offset, long count, FileShareMode fileShareMode, CancellationToken cancellationToken)
|
|
{
|
|
return ((HttpResponseStream)OutputStream).TransmitFile(path, offset, count, fileShareMode, cancellationToken);
|
|
}
|
|
}
|
|
} |