Refactored PayloadData.cs
This commit is contained in:
parent
5109d88ebb
commit
64938831ab
@ -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 ()
|
||||||
|
@ -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;
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user