[Modify] Polish it

This commit is contained in:
sta 2017-12-28 19:42:07 +09:00
parent 05e9629652
commit 3651d12408

View File

@ -462,21 +462,24 @@ namespace WebSocketSharp.Net
#region Internal Methods
internal void AddHeader (string header)
internal void AddHeader (string headerLine)
{
var colon = header.IndexOf (':');
if (colon == -1) {
_context.ErrorMessage = "Invalid header";
var colon = headerLine.IndexOf (':');
if (colon < 1) {
_context.ErrorMessage = "Invalid header line";
return;
}
var name = header.Substring (0, colon).Trim ();
var val = header.Substring (colon + 1).Trim ();
var name = headerLine.Substring (0, colon).Trim ();
var val = colon < headerLine.Length - 1
? headerLine.Substring (colon + 1).Trim ()
: String.Empty;
_headers.InternalSet (name, val, false);
var lower = name.ToLower (CultureInfo.InvariantCulture);
if (lower == "accept") {
_acceptTypes = new List<string> (val.SplitHeaderValue (',')).ToArray ();
_acceptTypes = val.SplitHeaderValue (',').ToList ().ToArray ();
return;
}
@ -509,8 +512,17 @@ namespace WebSocketSharp.Net
return;
}
if (lower == "referer")
_referer = val.ToUri ();
if (lower == "referer") {
var referer = val.ToUri ();
if (referer != null) {
_referer = referer;
}
else {
_context.ErrorMessage = "Invalid Referer header";
}
return;
}
}
internal void FinishInitialization ()