ec1f5dc317
Add Argument*Exceptions now use proper nameof operators. Added exception messages to quite a few Argument*Exceptions. Fixed rethorwing to be proper syntax. Added a ton of null checkes. (This is only a start, there are about 500 places that need proper null handling) Added some TODOs to log certain exceptions. Fix sln again. Fixed all AssemblyInfo's and added proper copyright (where I could find them) We live in *current year*. Fixed the use of braces. Fixed a ton of properties, and made a fair amount of functions static that should be and can be static. Made more Methods that should be static static. You can now use static to find bad functions! Removed unused variable. And added one more proper XML comment.
368 lines
12 KiB
C#
368 lines
12 KiB
C#
using System;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Specialized;
|
|
using System.Net;
|
|
using System.Runtime.InteropServices;
|
|
using System.Runtime.Serialization;
|
|
using System.Text;
|
|
using MediaBrowser.Model.Services;
|
|
using MediaBrowser.Model.Extensions;
|
|
|
|
namespace SocketHttpListener.Net
|
|
{
|
|
[ComVisible(true)]
|
|
public class WebHeaderCollection : QueryParamCollection
|
|
{
|
|
[Flags]
|
|
internal enum HeaderInfo
|
|
{
|
|
Request = 1,
|
|
Response = 1 << 1,
|
|
MultiValue = 1 << 10
|
|
}
|
|
|
|
static readonly bool[] allowed_chars = {
|
|
false, false, false, false, false, false, false, false, false, false, false, false, false, false,
|
|
false, false, false, false, false, false, false, false, false, false, false, false, false, false,
|
|
false, false, false, false, false, true, false, true, true, true, true, false, false, false, true,
|
|
true, false, true, true, false, true, true, true, true, true, true, true, true, true, true, false,
|
|
false, false, false, false, false, false, true, true, true, true, true, true, true, true, true,
|
|
true, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true,
|
|
false, false, false, true, true, true, true, true, true, true, true, true, true, true, true, true,
|
|
true, true, true, true, true, true, true, true, true, true, true, true, true, true, true, true,
|
|
false, true, false
|
|
};
|
|
|
|
static readonly Dictionary<string, HeaderInfo> headers;
|
|
|
|
static WebHeaderCollection()
|
|
{
|
|
headers = new Dictionary<string, HeaderInfo>(StringComparer.OrdinalIgnoreCase) {
|
|
{ "Allow", HeaderInfo.MultiValue },
|
|
{ "Accept", HeaderInfo.Request | HeaderInfo.MultiValue },
|
|
{ "Accept-Charset", HeaderInfo.MultiValue },
|
|
{ "Accept-Encoding", HeaderInfo.MultiValue },
|
|
{ "Accept-Language", HeaderInfo.MultiValue },
|
|
{ "Accept-Ranges", HeaderInfo.MultiValue },
|
|
{ "Age", HeaderInfo.Response },
|
|
{ "Authorization", HeaderInfo.MultiValue },
|
|
{ "Cache-Control", HeaderInfo.MultiValue },
|
|
{ "Cookie", HeaderInfo.MultiValue },
|
|
{ "Connection", HeaderInfo.Request | HeaderInfo.MultiValue },
|
|
{ "Content-Encoding", HeaderInfo.MultiValue },
|
|
{ "Content-Length", HeaderInfo.Request | HeaderInfo.Response },
|
|
{ "Content-Type", HeaderInfo.Request },
|
|
{ "Content-Language", HeaderInfo.MultiValue },
|
|
{ "Date", HeaderInfo.Request },
|
|
{ "Expect", HeaderInfo.Request | HeaderInfo.MultiValue},
|
|
{ "Host", HeaderInfo.Request },
|
|
{ "If-Match", HeaderInfo.MultiValue },
|
|
{ "If-Modified-Since", HeaderInfo.Request },
|
|
{ "If-None-Match", HeaderInfo.MultiValue },
|
|
{ "Keep-Alive", HeaderInfo.Response },
|
|
{ "Pragma", HeaderInfo.MultiValue },
|
|
{ "Proxy-Authenticate", HeaderInfo.MultiValue },
|
|
{ "Proxy-Authorization", HeaderInfo.MultiValue },
|
|
{ "Proxy-Connection", HeaderInfo.Request | HeaderInfo.MultiValue },
|
|
{ "Range", HeaderInfo.Request | HeaderInfo.MultiValue },
|
|
{ "Referer", HeaderInfo.Request },
|
|
{ "Set-Cookie", HeaderInfo.MultiValue },
|
|
{ "Set-Cookie2", HeaderInfo.MultiValue },
|
|
{ "Server", HeaderInfo.Response },
|
|
{ "TE", HeaderInfo.MultiValue },
|
|
{ "Trailer", HeaderInfo.MultiValue },
|
|
{ "Transfer-Encoding", HeaderInfo.Request | HeaderInfo.Response | HeaderInfo.MultiValue },
|
|
{ "Translate", HeaderInfo.Request | HeaderInfo.Response },
|
|
{ "Upgrade", HeaderInfo.MultiValue },
|
|
{ "User-Agent", HeaderInfo.Request },
|
|
{ "Vary", HeaderInfo.MultiValue },
|
|
{ "Via", HeaderInfo.MultiValue },
|
|
{ "Warning", HeaderInfo.MultiValue },
|
|
{ "WWW-Authenticate", HeaderInfo.Response | HeaderInfo. MultiValue },
|
|
{ "SecWebSocketAccept", HeaderInfo.Response },
|
|
{ "SecWebSocketExtensions", HeaderInfo.Request | HeaderInfo.Response | HeaderInfo. MultiValue },
|
|
{ "SecWebSocketKey", HeaderInfo.Request },
|
|
{ "Sec-WebSocket-Protocol", HeaderInfo.Request | HeaderInfo.Response | HeaderInfo. MultiValue },
|
|
{ "SecWebSocketVersion", HeaderInfo.Response | HeaderInfo. MultiValue }
|
|
};
|
|
}
|
|
|
|
// Methods
|
|
|
|
public void Add(string header)
|
|
{
|
|
if (header == null)
|
|
throw new ArgumentNullException(nameof(header));
|
|
int pos = header.IndexOf(':');
|
|
if (pos == -1)
|
|
throw new ArgumentException("no colon found", nameof(header));
|
|
|
|
this.Add(header.Substring(0, pos), header.Substring(pos + 1));
|
|
}
|
|
|
|
public override void Add(string name, string value)
|
|
{
|
|
if (name == null)
|
|
throw new ArgumentNullException(nameof(name));
|
|
|
|
this.AddWithoutValidate(name, value);
|
|
}
|
|
|
|
protected void AddWithoutValidate(string headerName, string headerValue)
|
|
{
|
|
if (!IsHeaderName(headerName))
|
|
throw new ArgumentException("invalid header name: " + headerName, nameof(headerName));
|
|
if (headerValue == null)
|
|
headerValue = string.Empty;
|
|
else
|
|
headerValue = headerValue.Trim();
|
|
if (!IsHeaderValue(headerValue))
|
|
throw new ArgumentException("invalid header value: " + headerValue, nameof(headerValue));
|
|
|
|
AddValue(headerName, headerValue);
|
|
}
|
|
|
|
internal void AddValue(string headerName, string headerValue)
|
|
{
|
|
base.Add(headerName, headerValue);
|
|
}
|
|
|
|
internal List<string> GetValues_internal(string header, bool split)
|
|
{
|
|
if (header == null)
|
|
throw new ArgumentNullException(nameof(header));
|
|
|
|
var values = base.GetValues(header);
|
|
if (values == null || values.Count == 0)
|
|
return null;
|
|
|
|
if (split && IsMultiValue(header))
|
|
{
|
|
List<string> separated = null;
|
|
foreach (var value in values)
|
|
{
|
|
if (value.IndexOf(',') < 0)
|
|
{
|
|
if (separated != null)
|
|
separated.Add(value);
|
|
|
|
continue;
|
|
}
|
|
|
|
if (separated == null)
|
|
{
|
|
separated = new List<string>(values.Count + 1);
|
|
foreach (var v in values)
|
|
{
|
|
if (v == value)
|
|
break;
|
|
|
|
separated.Add(v);
|
|
}
|
|
}
|
|
|
|
var slices = value.Split(',');
|
|
var slices_length = slices.Length;
|
|
if (value[value.Length - 1] == ',')
|
|
--slices_length;
|
|
|
|
for (int i = 0; i < slices_length; ++i)
|
|
{
|
|
separated.Add(slices[i].Trim());
|
|
}
|
|
}
|
|
|
|
if (separated != null)
|
|
return separated;
|
|
}
|
|
|
|
return values;
|
|
}
|
|
|
|
public override List<string> GetValues(string header)
|
|
{
|
|
return GetValues_internal(header, true);
|
|
}
|
|
|
|
public override string[] GetValues(int index)
|
|
{
|
|
string[] values = base.GetValues(index);
|
|
|
|
if (values == null || values.Length == 0)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
return values;
|
|
}
|
|
|
|
public static bool IsRestricted(string headerName)
|
|
{
|
|
return IsRestricted(headerName, false);
|
|
}
|
|
|
|
public static bool IsRestricted(string headerName, bool response)
|
|
{
|
|
if (headerName == null)
|
|
throw new ArgumentNullException(nameof(headerName));
|
|
|
|
if (headerName.Length == 0)
|
|
throw new ArgumentException("empty string", nameof(headerName));
|
|
|
|
if (!IsHeaderName(headerName))
|
|
throw new ArgumentException("Invalid character in header");
|
|
|
|
HeaderInfo info;
|
|
if (!headers.TryGetValue(headerName, out info))
|
|
return false;
|
|
|
|
var flag = response ? HeaderInfo.Response : HeaderInfo.Request;
|
|
return (info & flag) != 0;
|
|
}
|
|
|
|
public override void Set(string name, string value)
|
|
{
|
|
if (name == null)
|
|
throw new ArgumentNullException(nameof(name));
|
|
if (!IsHeaderName(name))
|
|
throw new ArgumentException("invalid header name");
|
|
if (value == null)
|
|
value = string.Empty;
|
|
else
|
|
value = value.Trim();
|
|
if (!IsHeaderValue(value))
|
|
throw new ArgumentException("invalid header value");
|
|
|
|
base.Set(name, value);
|
|
}
|
|
|
|
internal string ToStringMultiValue()
|
|
{
|
|
StringBuilder sb = new StringBuilder();
|
|
|
|
int count = base.Count;
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
string key = GetKey(i);
|
|
if (IsMultiValue(key))
|
|
{
|
|
foreach (string v in GetValues(i))
|
|
{
|
|
sb.Append(key)
|
|
.Append(": ")
|
|
.Append(v)
|
|
.Append("\r\n");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
sb.Append(key)
|
|
.Append(": ")
|
|
.Append(Get(i))
|
|
.Append("\r\n");
|
|
}
|
|
}
|
|
return sb.Append("\r\n").ToString();
|
|
}
|
|
|
|
public override string ToString()
|
|
{
|
|
StringBuilder sb = new StringBuilder();
|
|
|
|
int count = base.Count;
|
|
for (int i = 0; i < count; i++)
|
|
sb.Append(GetKey(i))
|
|
.Append(": ")
|
|
.Append(Get(i))
|
|
.Append("\r\n");
|
|
|
|
return sb.Append("\r\n").ToString();
|
|
}
|
|
|
|
|
|
// Internal Methods
|
|
|
|
// With this we don't check for invalid characters in header. See bug #55994.
|
|
internal void SetInternal(string header)
|
|
{
|
|
int pos = header.IndexOf(':');
|
|
if (pos == -1)
|
|
throw new ArgumentException("no colon found", nameof(header));
|
|
|
|
SetInternal(header.Substring(0, pos), header.Substring(pos + 1));
|
|
}
|
|
|
|
internal void SetInternal(string name, string value)
|
|
{
|
|
if (value == null)
|
|
value = string.Empty;
|
|
else
|
|
value = value.Trim();
|
|
if (!IsHeaderValue(value))
|
|
throw new ArgumentException("invalid header value");
|
|
|
|
if (IsMultiValue(name))
|
|
{
|
|
base.Add(name, value);
|
|
}
|
|
else
|
|
{
|
|
base.Remove(name);
|
|
base.Set(name, value);
|
|
}
|
|
}
|
|
|
|
internal static bool IsMultiValue(string headerName)
|
|
{
|
|
if (headerName == null)
|
|
return false;
|
|
|
|
HeaderInfo info;
|
|
return headers.TryGetValue(headerName, out info) && (info & HeaderInfo.MultiValue) != 0;
|
|
}
|
|
|
|
internal static bool IsHeaderValue(string value)
|
|
{
|
|
// TEXT any 8 bit value except CTL's (0-31 and 127)
|
|
// but including \r\n space and \t
|
|
// after a newline at least one space or \t must follow
|
|
// certain header fields allow comments ()
|
|
|
|
int len = value.Length;
|
|
for (int i = 0; i < len; i++)
|
|
{
|
|
char c = value[i];
|
|
if (c == 127)
|
|
return false;
|
|
if (c < 0x20 && (c != '\r' && c != '\n' && c != '\t'))
|
|
return false;
|
|
if (c == '\n' && ++i < len)
|
|
{
|
|
c = value[i];
|
|
if (c != ' ' && c != '\t')
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
internal static bool IsHeaderName(string name)
|
|
{
|
|
if (name == null || name.Length == 0)
|
|
return false;
|
|
|
|
int len = name.Length;
|
|
for (int i = 0; i < len; i++)
|
|
{
|
|
char c = name[i];
|
|
if (c > 126 || !allowed_chars[c])
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
}
|