72 lines
2.7 KiB
C#
72 lines
2.7 KiB
C#
using DTLib.Extensions;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Net.Sockets;
|
|
using System.Threading;
|
|
|
|
namespace DTLib.Network
|
|
{
|
|
//
|
|
// отправка/получение пакетов
|
|
//
|
|
public static class Package
|
|
{
|
|
// принимает пакет
|
|
public static byte[] GetPackage(this Socket socket)
|
|
{
|
|
int packageSize = 0;
|
|
byte[] data = new byte[2];
|
|
// цикл выполняется пока не пройдёт 2000 мс
|
|
for (ushort s = 0; s < 400; s += 1)
|
|
{
|
|
if (packageSize == 0 && socket.Available >= 2)
|
|
{
|
|
socket.Receive(data, data.Length, 0);
|
|
packageSize = data.ToInt();
|
|
}
|
|
if (packageSize != 0 && socket.Available >= packageSize)
|
|
{
|
|
data = new byte[packageSize];
|
|
socket.Receive(data, data.Length, 0);
|
|
return data;
|
|
}
|
|
else
|
|
Thread.Sleep(5);
|
|
}
|
|
throw new Exception($"GetPackage() error: timeout. socket.Available={socket.Available}\n");
|
|
}
|
|
|
|
// отправляет пакет
|
|
public static void SendPackage(this Socket socket, byte[] data)
|
|
{
|
|
if (data.Length > 65536)
|
|
throw new Exception($"SendPackage() error: package is too big ({data.Length} bytes)");
|
|
if (data.Length == 0)
|
|
throw new Exception($"SendPackage() error: package has zero size");
|
|
var list = new List<byte>();
|
|
byte[] packageSize = data.Length.ToBytes();
|
|
if (packageSize.Length == 1)
|
|
list.Add(0);
|
|
list.AddRange(packageSize);
|
|
list.AddRange(data);
|
|
socket.Send(list.ToArray());
|
|
}
|
|
public static void SendPackage(this Socket socket, string data) => SendPackage(socket, data.ToBytes());
|
|
|
|
// получает пакет и выбрасывает исключение, если пакет не соответствует образцу
|
|
public static void GetAnswer(this Socket socket, string answer)
|
|
{
|
|
string rec = socket.GetPackage().BytesToString();
|
|
if (rec != answer)
|
|
throw new Exception($"GetAnswer() error: invalid answer: <{rec}>");
|
|
}
|
|
|
|
public static byte[] RequestPackage(this Socket socket, byte[] request)
|
|
{
|
|
socket.SendPackage(request);
|
|
return socket.GetPackage();
|
|
}
|
|
public static byte[] RequestPackage(this Socket socket, string request) => socket.RequestPackage(request.ToBytes());
|
|
}
|
|
}
|