Refactored PayloadData.cs

This commit is contained in:
sta 2014-09-07 22:18:45 +09:00
parent 5109d88ebb
commit 64938831ab
2 changed files with 42 additions and 38 deletions

View File

@ -37,59 +37,63 @@ namespace WebSocketSharp
{ {
#region Private Fields #region Private Fields
private byte [] _applicationData; private byte[] _appData;
private byte [] _extensionData; private byte[] _extData;
private bool _masked; private ulong _length;
private bool _masked;
#endregion #endregion
#region Public Const Fields #region Public Fields
public const ulong MaxLength = long.MaxValue; public const ulong MaxLength = Int64.MaxValue;
#endregion #endregion
#region Public Constructors #region Public Constructors
public PayloadData () public PayloadData ()
: this (new byte [0], new byte [0], false) : this (new byte[0], new byte[0], false)
{ {
} }
public PayloadData (byte [] applicationData) public PayloadData (byte[] applicationData)
: this (new byte [0], applicationData, false) : this (new byte[0], applicationData, false)
{ {
} }
public PayloadData (string applicationData) public PayloadData (string applicationData)
: this (new byte [0], Encoding.UTF8.GetBytes (applicationData), false) : this (new byte[0], Encoding.UTF8.GetBytes (applicationData), false)
{ {
} }
public PayloadData (byte [] applicationData, bool masked) public PayloadData (byte[] applicationData, bool masked)
: this (new byte [0], applicationData, masked) : this (new byte[0], applicationData, masked)
{ {
} }
public PayloadData (byte [] extensionData, byte [] applicationData, bool masked) public PayloadData (byte[] extensionData, byte[] applicationData, bool masked)
{ {
if ((ulong) extensionData.LongLength + (ulong) applicationData.LongLength > MaxLength) var extLen = (ulong) extensionData.LongLength;
var appLen = (ulong) applicationData.LongLength;
if (appLen > MaxLength - extLen)
throw new ArgumentOutOfRangeException ( throw new ArgumentOutOfRangeException (
"The length of 'extensionData' plus 'applicationData' is greater than MaxLength."); "The length of payload data is greater than the allowable length.");
_extensionData = extensionData; _extData = extensionData;
_applicationData = applicationData; _appData = applicationData;
_masked = masked; _masked = masked;
_length = extLen + appLen;
} }
#endregion #endregion
#region Internal Properties #region Internal Properties
internal bool ContainsReservedCloseStatusCode { internal bool IncludesReservedCloseStatusCode {
get { get {
return _applicationData.Length > 1 && return _appData.Length > 1 &&
_applicationData.SubArray (0, 2).ToUInt16 (ByteOrder.Big).IsReserved (); _appData.SubArray (0, 2).ToUInt16 (ByteOrder.Big).IsReserved ();
} }
} }
@ -97,15 +101,15 @@ namespace WebSocketSharp
#region Public Properties #region Public Properties
public byte [] ApplicationData { public byte[] ApplicationData {
get { get {
return _applicationData; return _appData;
} }
} }
public byte [] ExtensionData { public byte[] ExtensionData {
get { get {
return _extensionData; return _extData;
} }
} }
@ -117,7 +121,7 @@ namespace WebSocketSharp
public ulong Length { public ulong Length {
get { get {
return (ulong) (_extensionData.LongLength + _applicationData.LongLength); return _length;
} }
} }
@ -125,10 +129,10 @@ namespace WebSocketSharp
#region Private Methods #region Private Methods
private static void mask (byte [] src, byte [] key) private static void mask (byte[] source, byte[] key)
{ {
for (long i = 0; i < src.LongLength; i++) for (long i = 0; i < source.LongLength; i++)
src [i] = (byte) (src [i] ^ key [i % 4]); source[i] = (byte) (source[i] ^ key[i % 4]);
} }
#endregion #endregion
@ -137,29 +141,29 @@ namespace WebSocketSharp
public IEnumerator<byte> GetEnumerator () public IEnumerator<byte> GetEnumerator ()
{ {
foreach (byte b in _extensionData) foreach (byte b in _extData)
yield return b; yield return b;
foreach (byte b in _applicationData) foreach (byte b in _appData)
yield return b; yield return b;
} }
public void Mask (byte [] maskingKey) public void Mask (byte[] maskingKey)
{ {
if (_extensionData.LongLength > 0) if (_extData.LongLength > 0)
mask (_extensionData, maskingKey); mask (_extData, maskingKey);
if (_applicationData.LongLength > 0) if (_appData.LongLength > 0)
mask (_applicationData, maskingKey); mask (_appData, maskingKey);
_masked = !_masked; _masked = !_masked;
} }
public byte [] ToByteArray () public byte[] ToByteArray ()
{ {
return _extensionData.LongLength > 0 return _extData.LongLength > 0
? new List<byte> (this).ToArray () ? new List<byte> (this).ToArray ()
: _applicationData; : _appData;
} }
public override string ToString () public override string ToString ()

View File

@ -913,7 +913,7 @@ namespace WebSocketSharp
private bool processCloseFrame (WebSocketFrame frame) private bool processCloseFrame (WebSocketFrame frame)
{ {
var payload = frame.PayloadData; var payload = frame.PayloadData;
close (payload, !payload.ContainsReservedCloseStatusCode, false); close (payload, !payload.IncludesReservedCloseStatusCode, false);
return false; return false;
} }