jellyfin/SocketHttpListener/PayloadData.cs
2017-05-24 15:12:55 -04:00

150 lines
2.9 KiB
C#

using System;
using System.Collections;
using System.Collections.Generic;
using System.Text;
namespace SocketHttpListener
{
internal class PayloadData : IEnumerable<byte>
{
#region Private Fields
private byte [] _applicationData;
private byte [] _extensionData;
private bool _masked;
#endregion
#region Public Const Fields
public const ulong MaxLength = long.MaxValue;
#endregion
#region Public Constructors
public PayloadData ()
: this (new byte [0], new byte [0], false)
{
}
public PayloadData (byte [] applicationData)
: this (new byte [0], applicationData, false)
{
}
public PayloadData (string applicationData)
: this (new byte [0], Encoding.UTF8.GetBytes (applicationData), false)
{
}
public PayloadData (byte [] applicationData, bool masked)
: this (new byte [0], applicationData, masked)
{
}
public PayloadData (byte [] extensionData, byte [] applicationData, bool masked)
{
_extensionData = extensionData;
_applicationData = applicationData;
_masked = masked;
}
#endregion
#region Internal Properties
internal bool ContainsReservedCloseStatusCode {
get {
return _applicationData.Length > 1 &&
_applicationData.SubArray (0, 2).ToUInt16 (ByteOrder.Big).IsReserved ();
}
}
#endregion
#region Public Properties
public byte [] ApplicationData {
get {
return _applicationData;
}
}
public byte [] ExtensionData {
get {
return _extensionData;
}
}
public bool IsMasked {
get {
return _masked;
}
}
public ulong Length {
get {
return (ulong) (_extensionData.Length + _applicationData.Length);
}
}
#endregion
#region Private Methods
private static void mask (byte [] src, byte [] key)
{
for (long i = 0; i < src.Length; i++)
src [i] = (byte) (src [i] ^ key [i % 4]);
}
#endregion
#region Public Methods
public IEnumerator<byte> GetEnumerator ()
{
foreach (byte b in _extensionData)
yield return b;
foreach (byte b in _applicationData)
yield return b;
}
public void Mask (byte [] maskingKey)
{
if (_extensionData.Length > 0)
mask (_extensionData, maskingKey);
if (_applicationData.Length > 0)
mask (_applicationData, maskingKey);
_masked = !_masked;
}
public byte [] ToByteArray ()
{
return _extensionData.Length > 0
? new List<byte> (this).ToArray ()
: _applicationData;
}
public override string ToString ()
{
return BitConverter.ToString (ToByteArray ());
}
#endregion
#region Explicitly Implemented Interface Members
IEnumerator IEnumerable.GetEnumerator ()
{
return GetEnumerator ();
}
#endregion
}
}