62 lines
2.0 KiB
C#
62 lines
2.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Text;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Steamworks;
|
|
|
|
namespace Torch.Utils
|
|
{
|
|
public static class MiscExtensions
|
|
{
|
|
private static readonly ThreadLocal<WeakReference<byte[]>> _streamBuffer = new ThreadLocal<WeakReference<byte[]>>(() => new WeakReference<byte[]>(null));
|
|
|
|
private static long LengthSafe(this Stream stream)
|
|
{
|
|
try
|
|
{
|
|
return stream.Length;
|
|
}
|
|
catch
|
|
{
|
|
return 512;
|
|
}
|
|
}
|
|
|
|
public static byte[] ReadToEnd(this Stream stream, int optionalDataLength = -1)
|
|
{
|
|
byte[] buffer;
|
|
if (!_streamBuffer.Value.TryGetTarget(out buffer))
|
|
buffer = new byte[stream.LengthSafe()];
|
|
var initialBufferSize = optionalDataLength > 0 ? optionalDataLength : stream.LengthSafe();
|
|
if (buffer.Length < initialBufferSize)
|
|
buffer = new byte[initialBufferSize];
|
|
if (buffer.Length < 1024)
|
|
buffer = new byte[1024];
|
|
var streamPosition = 0;
|
|
while (true)
|
|
{
|
|
if (buffer.Length == streamPosition)
|
|
Array.Resize(ref buffer, Math.Max((int)stream.LengthSafe(), buffer.Length * 2));
|
|
int count = stream.Read(buffer, streamPosition, buffer.Length - streamPosition);
|
|
if (count == 0)
|
|
break;
|
|
streamPosition += count;
|
|
}
|
|
var result = new byte[streamPosition];
|
|
Array.Copy(buffer, 0, result, 0, result.Length);
|
|
_streamBuffer.Value.SetTarget(buffer);
|
|
return result;
|
|
}
|
|
|
|
public static IPAddress GetRemoteIP(this P2PSessionState_t state)
|
|
{
|
|
// What is endianness anyway?
|
|
return new IPAddress(BitConverter.GetBytes(state.m_nRemoteIP).Reverse().ToArray());
|
|
}
|
|
}
|
|
}
|