long candy support
This commit is contained in:
meelstorm
118
Plugins/Pralinen/PralinenPLC/NetworkResultInformer.cs
Normal file
118
Plugins/Pralinen/PralinenPLC/NetworkResultInformer.cs
Normal file
@@ -0,0 +1,118 @@
|
||||
using System.Collections;
|
||||
using PralinenPLC.TCP;
|
||||
|
||||
namespace PralinenPLC;
|
||||
|
||||
public class NetworkResultInformer : TcpServer, IDisposable
|
||||
{
|
||||
|
||||
private bool _systemReady = false;
|
||||
private bool _dataReady = false;
|
||||
private bool _dataOk = false;
|
||||
private bool _firstConnection = false;
|
||||
private static bool _isStarted = false;
|
||||
public NetworkResultInformer() : base("0.0.0.0", 42010)
|
||||
{
|
||||
Console.WriteLine("------");
|
||||
Console.WriteLine("NetworkResultInformer");
|
||||
Console.WriteLine("------");
|
||||
if (!_isStarted)
|
||||
{
|
||||
Start();
|
||||
_isStarted = true;
|
||||
}
|
||||
_systemReady = true;
|
||||
_dataReady = false;
|
||||
_dataOk = false;
|
||||
|
||||
}
|
||||
List<TcpSession> _sessions = new List<TcpSession>();
|
||||
protected override void OnConnected(TcpSession session)
|
||||
{
|
||||
Console.WriteLine("Client connected");
|
||||
Console.WriteLine("------");
|
||||
|
||||
_sessions.Add(session);
|
||||
_firstConnection = true;
|
||||
base.OnConnected(session);
|
||||
SendStatus();
|
||||
}
|
||||
|
||||
protected override void OnDisconnected(TcpSession session)
|
||||
{
|
||||
Console.WriteLine("Client disconnected");
|
||||
Console.WriteLine("------");
|
||||
_sessions.Remove(session);
|
||||
base.OnDisconnected(session);
|
||||
}
|
||||
byte ConvertToByte(BitArray bits)
|
||||
{
|
||||
if (bits.Count != 8)
|
||||
{
|
||||
throw new ArgumentException("bits");
|
||||
}
|
||||
byte[] bytes = new byte[1];
|
||||
bits.CopyTo(bytes, 0);
|
||||
return bytes[0];
|
||||
}
|
||||
public void SendStatus()
|
||||
{
|
||||
try
|
||||
{
|
||||
Console.WriteLine($"Sessions:{_sessions.Count}");
|
||||
foreach (TcpSession session in _sessions)
|
||||
{
|
||||
|
||||
byte[] data = new byte[10];
|
||||
|
||||
BitArray arr = new BitArray(8);
|
||||
|
||||
arr.Set(0, _systemReady);
|
||||
arr.Set(1, _dataReady);
|
||||
arr.Set(2, _dataOk);
|
||||
arr.Set(7, _firstConnection);
|
||||
|
||||
Console.WriteLine($"Sending status: system:{_systemReady} data:{_dataReady} ok:{_dataOk} init:{_firstConnection}");
|
||||
|
||||
|
||||
|
||||
data[0] = ConvertToByte(arr);
|
||||
|
||||
session.Send(data);
|
||||
|
||||
|
||||
}
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
Console.WriteLine(e);
|
||||
throw;
|
||||
}
|
||||
_firstConnection = false;
|
||||
}
|
||||
|
||||
public void Reset()
|
||||
{
|
||||
_dataReady = true;
|
||||
_dataOk = true;
|
||||
SendStatus();
|
||||
}
|
||||
|
||||
public int Delay { get; } = 0;
|
||||
|
||||
public void SignalBad()
|
||||
{
|
||||
_dataReady = true;
|
||||
_dataOk = false;
|
||||
SendStatus();
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
_systemReady = false;
|
||||
_dataReady = false;
|
||||
_dataOk = false;
|
||||
SendStatus();
|
||||
|
||||
}
|
||||
}
|
||||
20
Plugins/Pralinen/PralinenPLC/Plugin.cs
Normal file
20
Plugins/Pralinen/PralinenPLC/Plugin.cs
Normal file
@@ -0,0 +1,20 @@
|
||||
using Ninject;
|
||||
using VisionBuilder.UI.Common;
|
||||
using VisionBuilder.UI.Common.Plugins;
|
||||
|
||||
namespace PralinenPLC
|
||||
{
|
||||
public class Plugin: IPlugin
|
||||
{
|
||||
public void RegisterGlobalModules(IKernel kernel)
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
public void RegisterCameraModules(IKernel kernel, string cameraName)
|
||||
{
|
||||
// ONLY ONE CAMERA !!!
|
||||
kernel.RegisterModule<PralinenModule>();
|
||||
}
|
||||
}
|
||||
}
|
||||
33
Plugins/Pralinen/PralinenPLC/PralinenModule.cs
Normal file
33
Plugins/Pralinen/PralinenPLC/PralinenModule.cs
Normal file
@@ -0,0 +1,33 @@
|
||||
using VisionBuilder.UI.Common;
|
||||
using VisionBuilder.UI.Common.RecipeProcessing;
|
||||
|
||||
namespace PralinenPLC;
|
||||
|
||||
public class PralinenModule:IVisionBuilderModule
|
||||
{
|
||||
private readonly IRecognitionControl _recognitionControl;
|
||||
private NetworkResultInformer _networkResultInformer;
|
||||
|
||||
public PralinenModule(IRecognitionControl recognitionControl)
|
||||
{
|
||||
_recognitionControl = recognitionControl;
|
||||
}
|
||||
|
||||
public void InitializeModule()
|
||||
{
|
||||
_networkResultInformer = new NetworkResultInformer();
|
||||
_recognitionControl.ImageProcessed += _recognitionControl_ImageProcessed;
|
||||
}
|
||||
|
||||
private void _recognitionControl_ImageProcessed(VisionBuilder.UI.Common.Commands.ImageProcessedEvent obj)
|
||||
{
|
||||
if (obj.HasError)
|
||||
{
|
||||
_networkResultInformer.SignalBad();
|
||||
}
|
||||
else
|
||||
{
|
||||
_networkResultInformer.Reset();
|
||||
}
|
||||
}
|
||||
}
|
||||
21
Plugins/Pralinen/PralinenPLC/PralinenPLC.csproj
Normal file
21
Plugins/Pralinen/PralinenPLC/PralinenPLC.csproj
Normal file
@@ -0,0 +1,21 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
|
||||
<PropertyGroup>
|
||||
<TargetFramework>net8.0</TargetFramework>
|
||||
<ImplicitUsings>enable</ImplicitUsings>
|
||||
<Nullable>enable</Nullable>
|
||||
<EnableDynamicLoading>true</EnableDynamicLoading>
|
||||
<OutDir>D:\Inspectron\Hawkeye\code\VisionBuilder5\VisionBuilder.UI\VisionBuilder.UI.Windows.Test\bin\Debug\Data\Plugins\PralinenPLC</OutDir>
|
||||
</PropertyGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\..\..\framework\Inspectron.Settings\Inspectron.Settings.csproj">
|
||||
<Private>False</Private>
|
||||
<ExcludeAssets>runtime</ExcludeAssets>
|
||||
</ProjectReference>
|
||||
<ProjectReference Include="..\..\..\VisionBuilder.UI.Common\VisionBuilder.UI.Common.csproj">
|
||||
<Private>false</Private>
|
||||
<ExcludeAssets>runtime</ExcludeAssets>
|
||||
</ProjectReference>
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
182
Plugins/Pralinen/PralinenPLC/TCP/Buffer.cs
Normal file
182
Plugins/Pralinen/PralinenPLC/TCP/Buffer.cs
Normal file
@@ -0,0 +1,182 @@
|
||||
using System.Diagnostics;
|
||||
using System.Text;
|
||||
|
||||
namespace PralinenPLC.TCP
|
||||
{
|
||||
/// <summary>
|
||||
/// Dynamic byte buffer
|
||||
/// </summary>
|
||||
public class Buffer
|
||||
{
|
||||
private byte[] _data;
|
||||
private long _size;
|
||||
private long _offset;
|
||||
|
||||
/// <summary>
|
||||
/// Is the buffer empty?
|
||||
/// </summary>
|
||||
public bool IsEmpty => (_data == null) || (_size == 0);
|
||||
/// <summary>
|
||||
/// Bytes memory buffer
|
||||
/// </summary>
|
||||
public byte[] Data => _data;
|
||||
/// <summary>
|
||||
/// Bytes memory buffer capacity
|
||||
/// </summary>
|
||||
public long Capacity => _data.Length;
|
||||
/// <summary>
|
||||
/// Bytes memory buffer size
|
||||
/// </summary>
|
||||
public long Size => _size;
|
||||
/// <summary>
|
||||
/// Bytes memory buffer offset
|
||||
/// </summary>
|
||||
public long Offset => _offset;
|
||||
|
||||
/// <summary>
|
||||
/// Buffer indexer operator
|
||||
/// </summary>
|
||||
public byte this[int index] => _data[index];
|
||||
|
||||
/// <summary>
|
||||
/// Initialize a new expandable buffer with zero capacity
|
||||
/// </summary>
|
||||
public Buffer() { _data = new byte[0]; _size = 0; _offset = 0; }
|
||||
/// <summary>
|
||||
/// Initialize a new expandable buffer with the given capacity
|
||||
/// </summary>
|
||||
public Buffer(long capacity) { _data = new byte[capacity]; _size = 0; _offset = 0; }
|
||||
/// <summary>
|
||||
/// Initialize a new expandable buffer with the given data
|
||||
/// </summary>
|
||||
public Buffer(byte[] data) { _data = data; _size = data.Length; _offset = 0; }
|
||||
|
||||
#region Memory buffer methods
|
||||
|
||||
/// <summary>
|
||||
/// Get string from the current buffer
|
||||
/// </summary>
|
||||
public override string ToString()
|
||||
{
|
||||
return ExtractString(0, _size);
|
||||
}
|
||||
|
||||
// Clear the current buffer and its offset
|
||||
public void Clear()
|
||||
{
|
||||
_size = 0;
|
||||
_offset = 0;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Extract the string from buffer of the given offset and size
|
||||
/// </summary>
|
||||
public string ExtractString(long offset, long size)
|
||||
{
|
||||
Debug.Assert(((offset + size) <= Size), "Invalid offset & size!");
|
||||
if ((offset + size) > Size)
|
||||
throw new ArgumentException("Invalid offset & size!", nameof(offset));
|
||||
|
||||
return Encoding.UTF8.GetString(_data, (int)offset, (int)size);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Remove the buffer of the given offset and size
|
||||
/// </summary>
|
||||
public void Remove(long offset, long size)
|
||||
{
|
||||
Debug.Assert(((offset + size) <= Size), "Invalid offset & size!");
|
||||
if ((offset + size) > Size)
|
||||
throw new ArgumentException("Invalid offset & size!", nameof(offset));
|
||||
|
||||
Array.Copy(_data, offset + size, _data, offset, _size - size - offset);
|
||||
_size -= size;
|
||||
if (_offset >= (offset + size))
|
||||
_offset -= size;
|
||||
else if (_offset >= offset)
|
||||
{
|
||||
_offset -= _offset - offset;
|
||||
if (_offset > Size)
|
||||
_offset = Size;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Reserve the buffer of the given capacity
|
||||
/// </summary>
|
||||
public void Reserve(long capacity)
|
||||
{
|
||||
Debug.Assert((capacity >= 0), "Invalid reserve capacity!");
|
||||
if (capacity < 0)
|
||||
throw new ArgumentException("Invalid reserve capacity!", nameof(capacity));
|
||||
|
||||
if (capacity > Capacity)
|
||||
{
|
||||
byte[] data = new byte[Math.Max(capacity, 2 * Capacity)];
|
||||
Array.Copy(_data, 0, data, 0, _size);
|
||||
_data = data;
|
||||
}
|
||||
}
|
||||
|
||||
// Resize the current buffer
|
||||
public void Resize(long size)
|
||||
{
|
||||
Reserve(size);
|
||||
_size = size;
|
||||
if (_offset > _size)
|
||||
_offset = _size;
|
||||
}
|
||||
|
||||
// Shift the current buffer offset
|
||||
public void Shift(long offset) { _offset += offset; }
|
||||
// Unshift the current buffer offset
|
||||
public void Unshift(long offset) { _offset -= offset; }
|
||||
|
||||
#endregion
|
||||
|
||||
#region Buffer I/O methods
|
||||
|
||||
/// <summary>
|
||||
/// Append the given buffer
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to append</param>
|
||||
/// <returns>Count of append bytes</returns>
|
||||
public long Append(byte[] buffer)
|
||||
{
|
||||
Reserve(_size + buffer.Length);
|
||||
Array.Copy(buffer, 0, _data, _size, buffer.Length);
|
||||
_size += buffer.Length;
|
||||
return buffer.Length;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Append the given buffer fragment
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to append</param>
|
||||
/// <param name="offset">Buffer offset</param>
|
||||
/// <param name="size">Buffer size</param>
|
||||
/// <returns>Count of append bytes</returns>
|
||||
public long Append(byte[] buffer, long offset, long size)
|
||||
{
|
||||
Reserve(_size + size);
|
||||
Array.Copy(buffer, offset, _data, _size, size);
|
||||
_size += size;
|
||||
return size;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Append the given text in UTF-8 encoding
|
||||
/// </summary>
|
||||
/// <param name="text">Text to append</param>
|
||||
/// <returns>Count of append bytes</returns>
|
||||
public long Append(string text)
|
||||
{
|
||||
Reserve(_size + Encoding.UTF8.GetMaxByteCount(text.Length));
|
||||
long result = Encoding.UTF8.GetBytes(text, 0, text.Length, _data, (int)_size);
|
||||
_size += result;
|
||||
return result;
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
525
Plugins/Pralinen/PralinenPLC/TCP/TcpServer.cs
Normal file
525
Plugins/Pralinen/PralinenPLC/TCP/TcpServer.cs
Normal file
@@ -0,0 +1,525 @@
|
||||
using System.Collections.Concurrent;
|
||||
using System.Diagnostics;
|
||||
using System.Net;
|
||||
using System.Net.Sockets;
|
||||
using System.Text;
|
||||
|
||||
namespace PralinenPLC.TCP
|
||||
{
|
||||
/// <summary>
|
||||
/// TCP server is used to connect, disconnect and manage TCP sessions
|
||||
/// </summary>
|
||||
/// <remarks>Thread-safe</remarks>
|
||||
public class TcpServer : IDisposable
|
||||
{
|
||||
/// <summary>
|
||||
/// Initialize TCP server with a given IP address and port number
|
||||
/// </summary>
|
||||
/// <param name="address">IP address</param>
|
||||
/// <param name="port">Port number</param>
|
||||
public TcpServer(IPAddress address, int port) : this(new IPEndPoint(address, port)) {}
|
||||
/// <summary>
|
||||
/// Initialize TCP server with a given IP address and port number
|
||||
/// </summary>
|
||||
/// <param name="address">IP address</param>
|
||||
/// <param name="port">Port number</param>
|
||||
public TcpServer(string address, int port) : this(new IPEndPoint(IPAddress.Parse(address), port)) {}
|
||||
/// <summary>
|
||||
/// Initialize TCP server with a given IP endpoint
|
||||
/// </summary>
|
||||
/// <param name="endpoint">IP endpoint</param>
|
||||
public TcpServer(IPEndPoint endpoint)
|
||||
{
|
||||
Id = Guid.NewGuid();
|
||||
Endpoint = endpoint;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Server Id
|
||||
/// </summary>
|
||||
public Guid Id { get; }
|
||||
|
||||
/// <summary>
|
||||
/// IP endpoint
|
||||
/// </summary>
|
||||
public IPEndPoint Endpoint { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// Number of sessions connected to the server
|
||||
/// </summary>
|
||||
public long ConnectedSessions { get { return Sessions.Count; } }
|
||||
/// <summary>
|
||||
/// Number of bytes pending sent by the server
|
||||
/// </summary>
|
||||
public long BytesPending { get { return _bytesPending; } }
|
||||
/// <summary>
|
||||
/// Number of bytes sent by the server
|
||||
/// </summary>
|
||||
public long BytesSent { get { return _bytesSent; } }
|
||||
/// <summary>
|
||||
/// Number of bytes received by the server
|
||||
/// </summary>
|
||||
public long BytesReceived { get { return _bytesReceived; } }
|
||||
|
||||
/// <summary>
|
||||
/// Option: acceptor backlog size
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// This option will set the listening socket's backlog size
|
||||
/// </remarks>
|
||||
public int OptionAcceptorBacklog { get; set; } = 1024;
|
||||
/// <summary>
|
||||
/// Option: dual mode socket
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// Specifies whether the Socket is a dual-mode socket used for both IPv4 and IPv6.
|
||||
/// Will work only if socket is bound on IPv6 address.
|
||||
/// </remarks>
|
||||
public bool OptionDualMode { get; set; }
|
||||
/// <summary>
|
||||
/// Option: keep alive
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// This option will setup SO_KEEPALIVE if the OS support this feature
|
||||
/// </remarks>
|
||||
public bool OptionKeepAlive { get; set; }
|
||||
/// <summary>
|
||||
/// Option: no delay
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// This option will enable/disable Nagle's algorithm for TCP protocol
|
||||
/// </remarks>
|
||||
public bool OptionNoDelay { get; set; }
|
||||
/// <summary>
|
||||
/// Option: reuse address
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// This option will enable/disable SO_REUSEADDR if the OS support this feature
|
||||
/// </remarks>
|
||||
public bool OptionReuseAddress { get; set; }
|
||||
/// <summary>
|
||||
/// Option: enables a socket to be bound for exclusive access
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// This option will enable/disable SO_EXCLUSIVEADDRUSE if the OS support this feature
|
||||
/// </remarks>
|
||||
public bool OptionExclusiveAddressUse { get; set; }
|
||||
/// <summary>
|
||||
/// Option: receive buffer size
|
||||
/// </summary>
|
||||
public int OptionReceiveBufferSize { get; set; } = 8192;
|
||||
/// <summary>
|
||||
/// Option: send buffer size
|
||||
/// </summary>
|
||||
public int OptionSendBufferSize { get; set; } = 8192;
|
||||
|
||||
#region Start/Stop server
|
||||
|
||||
// Server acceptor
|
||||
private Socket _acceptorSocket;
|
||||
private SocketAsyncEventArgs _acceptorEventArg;
|
||||
|
||||
// Server statistic
|
||||
internal long _bytesPending;
|
||||
internal long _bytesSent;
|
||||
internal long _bytesReceived;
|
||||
|
||||
/// <summary>
|
||||
/// Is the server started?
|
||||
/// </summary>
|
||||
public bool IsStarted { get; private set; }
|
||||
/// <summary>
|
||||
/// Is the server accepting new clients?
|
||||
/// </summary>
|
||||
public bool IsAccepting { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// Create a new socket object
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// Method may be override if you need to prepare some specific socket object in your implementation.
|
||||
/// </remarks>
|
||||
/// <returns>Socket object</returns>
|
||||
protected virtual Socket CreateSocket()
|
||||
{
|
||||
return new Socket(Endpoint.AddressFamily, SocketType.Stream, ProtocolType.Tcp);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Start the server
|
||||
/// </summary>
|
||||
/// <returns>'true' if the server was successfully started, 'false' if the server failed to start</returns>
|
||||
public virtual bool Start()
|
||||
{
|
||||
Debug.Assert(!IsStarted, "TCP server is already started!");
|
||||
if (IsStarted)
|
||||
return false;
|
||||
|
||||
// Setup acceptor event arg
|
||||
_acceptorEventArg = new SocketAsyncEventArgs();
|
||||
_acceptorEventArg.Completed += OnAsyncCompleted;
|
||||
|
||||
// Create a new acceptor socket
|
||||
_acceptorSocket = CreateSocket();
|
||||
|
||||
// Update the acceptor socket disposed flag
|
||||
IsSocketDisposed = false;
|
||||
|
||||
// Apply the option: reuse address
|
||||
_acceptorSocket.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, OptionReuseAddress);
|
||||
// Apply the option: exclusive address use
|
||||
_acceptorSocket.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ExclusiveAddressUse, OptionExclusiveAddressUse);
|
||||
// Apply the option: dual mode (this option must be applied before listening)
|
||||
if (_acceptorSocket.AddressFamily == AddressFamily.InterNetworkV6)
|
||||
_acceptorSocket.DualMode = OptionDualMode;
|
||||
|
||||
// Bind the acceptor socket to the IP endpoint
|
||||
_acceptorSocket.Bind(Endpoint);
|
||||
// Refresh the endpoint property based on the actual endpoint created
|
||||
Endpoint = (IPEndPoint)_acceptorSocket.LocalEndPoint;
|
||||
// Start listen to the acceptor socket with the given accepting backlog size
|
||||
_acceptorSocket.Listen(OptionAcceptorBacklog);
|
||||
|
||||
// Reset statistic
|
||||
_bytesPending = 0;
|
||||
_bytesSent = 0;
|
||||
_bytesReceived = 0;
|
||||
|
||||
// Update the started flag
|
||||
IsStarted = true;
|
||||
|
||||
// Call the server started handler
|
||||
OnStarted();
|
||||
|
||||
// Perform the first server accept
|
||||
IsAccepting = true;
|
||||
StartAccept(_acceptorEventArg);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Stop the server
|
||||
/// </summary>
|
||||
/// <returns>'true' if the server was successfully stopped, 'false' if the server is already stopped</returns>
|
||||
public virtual bool Stop()
|
||||
{
|
||||
Debug.Assert(IsStarted, "TCP server is not started!");
|
||||
if (!IsStarted)
|
||||
return false;
|
||||
|
||||
// Stop accepting new clients
|
||||
IsAccepting = false;
|
||||
|
||||
// Reset acceptor event arg
|
||||
_acceptorEventArg.Completed -= OnAsyncCompleted;
|
||||
|
||||
// Close the acceptor socket
|
||||
_acceptorSocket.Close();
|
||||
|
||||
// Dispose the acceptor socket
|
||||
_acceptorSocket.Dispose();
|
||||
|
||||
// Dispose event arguments
|
||||
_acceptorEventArg.Dispose();
|
||||
|
||||
// Update the acceptor socket disposed flag
|
||||
IsSocketDisposed = true;
|
||||
|
||||
// Disconnect all sessions
|
||||
DisconnectAll();
|
||||
|
||||
// Update the started flag
|
||||
IsStarted = false;
|
||||
|
||||
// Call the server stopped handler
|
||||
OnStopped();
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Restart the server
|
||||
/// </summary>
|
||||
/// <returns>'true' if the server was successfully restarted, 'false' if the server failed to restart</returns>
|
||||
public virtual bool Restart()
|
||||
{
|
||||
if (!Stop())
|
||||
return false;
|
||||
|
||||
while (IsStarted)
|
||||
Thread.Yield();
|
||||
|
||||
return Start();
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Accepting clients
|
||||
|
||||
/// <summary>
|
||||
/// Start accept a new client connection
|
||||
/// </summary>
|
||||
private void StartAccept(SocketAsyncEventArgs e)
|
||||
{
|
||||
// Socket must be cleared since the context object is being reused
|
||||
e.AcceptSocket = null;
|
||||
|
||||
// Async accept a new client connection
|
||||
if (!_acceptorSocket.AcceptAsync(e))
|
||||
ProcessAccept(e);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Process accepted client connection
|
||||
/// </summary>
|
||||
private void ProcessAccept(SocketAsyncEventArgs e)
|
||||
{
|
||||
if (e.SocketError == SocketError.Success)
|
||||
{
|
||||
// Create a new session to register
|
||||
var session = CreateSession();
|
||||
|
||||
// Register the session
|
||||
RegisterSession(session);
|
||||
|
||||
// Connect new session
|
||||
session.Connect(e.AcceptSocket);
|
||||
}
|
||||
else
|
||||
SendError(e.SocketError);
|
||||
|
||||
// Accept the next client connection
|
||||
if (IsAccepting)
|
||||
StartAccept(e);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// This method is the callback method associated with Socket.AcceptAsync()
|
||||
/// operations and is invoked when an accept operation is complete
|
||||
/// </summary>
|
||||
private void OnAsyncCompleted(object sender, SocketAsyncEventArgs e)
|
||||
{
|
||||
ProcessAccept(e);
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Session factory
|
||||
|
||||
/// <summary>
|
||||
/// Create TCP session factory method
|
||||
/// </summary>
|
||||
/// <returns>TCP session</returns>
|
||||
protected virtual TcpSession CreateSession() { return new TcpSession(this); }
|
||||
|
||||
#endregion
|
||||
|
||||
#region Session management
|
||||
|
||||
// Server sessions
|
||||
protected readonly ConcurrentDictionary<Guid, TcpSession> Sessions = new ConcurrentDictionary<Guid, TcpSession>();
|
||||
|
||||
/// <summary>
|
||||
/// Disconnect all connected sessions
|
||||
/// </summary>
|
||||
/// <returns>'true' if all sessions were successfully disconnected, 'false' if the server is not started</returns>
|
||||
public virtual bool DisconnectAll()
|
||||
{
|
||||
if (!IsStarted)
|
||||
return false;
|
||||
|
||||
// Disconnect all sessions
|
||||
foreach (var session in Sessions.Values)
|
||||
session.Disconnect();
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Find a session with a given Id
|
||||
/// </summary>
|
||||
/// <param name="id">Session Id</param>
|
||||
/// <returns>Session with a given Id or null if the session it not connected</returns>
|
||||
public TcpSession FindSession(Guid id)
|
||||
{
|
||||
// Try to find the required session
|
||||
return Sessions.TryGetValue(id, out TcpSession result) ? result : null;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Register a new session
|
||||
/// </summary>
|
||||
/// <param name="session">Session to register</param>
|
||||
internal void RegisterSession(TcpSession session)
|
||||
{
|
||||
// Register a new session
|
||||
Sessions.TryAdd(session.Id, session);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Unregister session by Id
|
||||
/// </summary>
|
||||
/// <param name="id">Session Id</param>
|
||||
internal void UnregisterSession(Guid id)
|
||||
{
|
||||
// Unregister session by Id
|
||||
Sessions.TryRemove(id, out TcpSession temp);
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Multicasting
|
||||
|
||||
/// <summary>
|
||||
/// Multicast data to all connected sessions
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to multicast</param>
|
||||
/// <returns>'true' if the data was successfully multicasted, 'false' if the data was not multicasted</returns>
|
||||
public virtual bool Multicast(byte[] buffer) { return Multicast(buffer, 0, buffer.Length); }
|
||||
|
||||
/// <summary>
|
||||
/// Multicast data to all connected clients
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to multicast</param>
|
||||
/// <param name="offset">Buffer offset</param>
|
||||
/// <param name="size">Buffer size</param>
|
||||
/// <returns>'true' if the data was successfully multicasted, 'false' if the data was not multicasted</returns>
|
||||
public virtual bool Multicast(byte[] buffer, long offset, long size)
|
||||
{
|
||||
if (!IsStarted)
|
||||
return false;
|
||||
|
||||
if (size == 0)
|
||||
return true;
|
||||
|
||||
// Multicast data to all sessions
|
||||
foreach (var session in Sessions.Values)
|
||||
session.SendAsync(buffer, offset, size);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Multicast text to all connected clients
|
||||
/// </summary>
|
||||
/// <param name="text">Text string to multicast</param>
|
||||
/// <returns>'true' if the text was successfully multicasted, 'false' if the text was not multicasted</returns>
|
||||
public virtual bool Multicast(string text) { return Multicast(Encoding.UTF8.GetBytes(text)); }
|
||||
|
||||
#endregion
|
||||
|
||||
#region Server handlers
|
||||
|
||||
/// <summary>
|
||||
/// Handle server started notification
|
||||
/// </summary>
|
||||
protected virtual void OnStarted() {}
|
||||
/// <summary>
|
||||
/// Handle server stopped notification
|
||||
/// </summary>
|
||||
protected virtual void OnStopped() {}
|
||||
|
||||
/// <summary>
|
||||
/// Handle session connected notification
|
||||
/// </summary>
|
||||
/// <param name="session">Connected session</param>
|
||||
protected virtual void OnConnected(TcpSession session) {}
|
||||
/// <summary>
|
||||
/// Handle session disconnected notification
|
||||
/// </summary>
|
||||
/// <param name="session">Disconnected session</param>
|
||||
protected virtual void OnDisconnected(TcpSession session) {}
|
||||
|
||||
/// <summary>
|
||||
/// Handle error notification
|
||||
/// </summary>
|
||||
/// <param name="error">Socket error code</param>
|
||||
protected virtual void OnError(SocketError error) {}
|
||||
|
||||
internal void OnConnectedInternal(TcpSession session) { OnConnected(session); }
|
||||
internal void OnDisconnectedInternal(TcpSession session) { OnDisconnected(session); }
|
||||
|
||||
#endregion
|
||||
|
||||
#region Error handling
|
||||
|
||||
/// <summary>
|
||||
/// Send error notification
|
||||
/// </summary>
|
||||
/// <param name="error">Socket error code</param>
|
||||
private void SendError(SocketError error)
|
||||
{
|
||||
// Skip disconnect errors
|
||||
if ((error == SocketError.ConnectionAborted) ||
|
||||
(error == SocketError.ConnectionRefused) ||
|
||||
(error == SocketError.ConnectionReset) ||
|
||||
(error == SocketError.OperationAborted) ||
|
||||
(error == SocketError.Shutdown))
|
||||
return;
|
||||
|
||||
OnError(error);
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region IDisposable implementation
|
||||
|
||||
/// <summary>
|
||||
/// Disposed flag
|
||||
/// </summary>
|
||||
public bool IsDisposed { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// Acceptor socket disposed flag
|
||||
/// </summary>
|
||||
public bool IsSocketDisposed { get; private set; } = true;
|
||||
|
||||
// Implement IDisposable.
|
||||
public void Dispose()
|
||||
{
|
||||
Dispose(true);
|
||||
GC.SuppressFinalize(this);
|
||||
}
|
||||
|
||||
protected virtual void Dispose(bool disposingManagedResources)
|
||||
{
|
||||
// The idea here is that Dispose(Boolean) knows whether it is
|
||||
// being called to do explicit cleanup (the Boolean is true)
|
||||
// versus being called due to a garbage collection (the Boolean
|
||||
// is false). This distinction is useful because, when being
|
||||
// disposed explicitly, the Dispose(Boolean) method can safely
|
||||
// execute code using reference type fields that refer to other
|
||||
// objects knowing for sure that these other objects have not been
|
||||
// finalized or disposed of yet. When the Boolean is false,
|
||||
// the Dispose(Boolean) method should not execute code that
|
||||
// refer to reference type fields because those objects may
|
||||
// have already been finalized."
|
||||
|
||||
if (!IsDisposed)
|
||||
{
|
||||
if (disposingManagedResources)
|
||||
{
|
||||
// Dispose managed resources here...
|
||||
Stop();
|
||||
}
|
||||
|
||||
// Dispose unmanaged resources here...
|
||||
|
||||
// Set large fields to null here...
|
||||
|
||||
// Mark as disposed.
|
||||
IsDisposed = true;
|
||||
}
|
||||
}
|
||||
|
||||
// Use C# destructor syntax for finalization code.
|
||||
~TcpServer()
|
||||
{
|
||||
// Simply call Dispose(false).
|
||||
Dispose(false);
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
722
Plugins/Pralinen/PralinenPLC/TCP/TcpSession.cs
Normal file
722
Plugins/Pralinen/PralinenPLC/TCP/TcpSession.cs
Normal file
@@ -0,0 +1,722 @@
|
||||
using System.Net.Sockets;
|
||||
using System.Text;
|
||||
|
||||
namespace PralinenPLC.TCP
|
||||
{
|
||||
/// <summary>
|
||||
/// TCP session is used to read and write data from the connected TCP client
|
||||
/// </summary>
|
||||
/// <remarks>Thread-safe</remarks>
|
||||
public class TcpSession : IDisposable
|
||||
{
|
||||
/// <summary>
|
||||
/// Initialize the session with a given server
|
||||
/// </summary>
|
||||
/// <param name="server">TCP server</param>
|
||||
public TcpSession(TcpServer server)
|
||||
{
|
||||
Id = Guid.NewGuid();
|
||||
Server = server;
|
||||
OptionReceiveBufferSize = server.OptionReceiveBufferSize;
|
||||
OptionSendBufferSize = server.OptionSendBufferSize;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Session Id
|
||||
/// </summary>
|
||||
public Guid Id { get; }
|
||||
|
||||
/// <summary>
|
||||
/// Server
|
||||
/// </summary>
|
||||
public TcpServer Server { get; }
|
||||
/// <summary>
|
||||
/// Socket
|
||||
/// </summary>
|
||||
public Socket Socket { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// Number of bytes pending sent by the session
|
||||
/// </summary>
|
||||
public long BytesPending { get; private set; }
|
||||
/// <summary>
|
||||
/// Number of bytes sending by the session
|
||||
/// </summary>
|
||||
public long BytesSending { get; private set; }
|
||||
/// <summary>
|
||||
/// Number of bytes sent by the session
|
||||
/// </summary>
|
||||
public long BytesSent { get; private set; }
|
||||
/// <summary>
|
||||
/// Number of bytes received by the session
|
||||
/// </summary>
|
||||
public long BytesReceived { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// Option: receive buffer size
|
||||
/// </summary>
|
||||
public int OptionReceiveBufferSize { get; set; } = 8192;
|
||||
/// <summary>
|
||||
/// Option: send buffer size
|
||||
/// </summary>
|
||||
public int OptionSendBufferSize { get; set; } = 8192;
|
||||
|
||||
#region Connect/Disconnect session
|
||||
|
||||
/// <summary>
|
||||
/// Is the session connected?
|
||||
/// </summary>
|
||||
public bool IsConnected { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// Connect the session
|
||||
/// </summary>
|
||||
/// <param name="socket">Session socket</param>
|
||||
internal void Connect(Socket socket)
|
||||
{
|
||||
Socket = socket;
|
||||
|
||||
// Update the session socket disposed flag
|
||||
IsSocketDisposed = false;
|
||||
|
||||
// Setup buffers
|
||||
_receiveBuffer = new Buffer();
|
||||
_sendBufferMain = new Buffer();
|
||||
_sendBufferFlush = new Buffer();
|
||||
|
||||
// Setup event args
|
||||
_receiveEventArg = new SocketAsyncEventArgs();
|
||||
_receiveEventArg.Completed += OnAsyncCompleted;
|
||||
_sendEventArg = new SocketAsyncEventArgs();
|
||||
_sendEventArg.Completed += OnAsyncCompleted;
|
||||
|
||||
// Apply the option: keep alive
|
||||
if (Server.OptionKeepAlive)
|
||||
Socket.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.KeepAlive, true);
|
||||
// Apply the option: no delay
|
||||
if (Server.OptionNoDelay)
|
||||
Socket.SetSocketOption(SocketOptionLevel.Tcp, SocketOptionName.NoDelay, true);
|
||||
|
||||
// Prepare receive & send buffers
|
||||
_receiveBuffer.Reserve(OptionReceiveBufferSize);
|
||||
_sendBufferMain.Reserve(OptionSendBufferSize);
|
||||
_sendBufferFlush.Reserve(OptionSendBufferSize);
|
||||
|
||||
// Reset statistic
|
||||
BytesPending = 0;
|
||||
BytesSending = 0;
|
||||
BytesSent = 0;
|
||||
BytesReceived = 0;
|
||||
|
||||
// Update the connected flag
|
||||
IsConnected = true;
|
||||
|
||||
// Try to receive something from the client
|
||||
TryReceive();
|
||||
|
||||
// Call the session connected handler
|
||||
OnConnected();
|
||||
|
||||
// Call the session connected handler in the server
|
||||
Server.OnConnectedInternal(this);
|
||||
|
||||
// Call the empty send buffer handler
|
||||
if (_sendBufferMain.IsEmpty)
|
||||
OnEmpty();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Disconnect the session
|
||||
/// </summary>
|
||||
/// <returns>'true' if the section was successfully disconnected, 'false' if the section is already disconnected</returns>
|
||||
public virtual bool Disconnect()
|
||||
{
|
||||
if (!IsConnected)
|
||||
return false;
|
||||
|
||||
// Reset event args
|
||||
_receiveEventArg.Completed -= OnAsyncCompleted;
|
||||
_sendEventArg.Completed -= OnAsyncCompleted;
|
||||
|
||||
try
|
||||
{
|
||||
try
|
||||
{
|
||||
// Shutdown the socket associated with the client
|
||||
Socket.Shutdown(SocketShutdown.Both);
|
||||
}
|
||||
catch (SocketException) {}
|
||||
|
||||
// Close the session socket
|
||||
Socket.Close();
|
||||
|
||||
// Dispose the session socket
|
||||
Socket.Dispose();
|
||||
|
||||
// Dispose event arguments
|
||||
_receiveEventArg.Dispose();
|
||||
_sendEventArg.Dispose();
|
||||
|
||||
// Update the session socket disposed flag
|
||||
IsSocketDisposed = true;
|
||||
}
|
||||
catch (ObjectDisposedException) {}
|
||||
|
||||
// Update the connected flag
|
||||
IsConnected = false;
|
||||
|
||||
// Update sending/receiving flags
|
||||
_receiving = false;
|
||||
_sending = false;
|
||||
|
||||
// Clear send/receive buffers
|
||||
ClearBuffers();
|
||||
|
||||
// Call the session disconnected handler
|
||||
OnDisconnected();
|
||||
|
||||
// Call the session disconnected handler in the server
|
||||
Server.OnDisconnectedInternal(this);
|
||||
|
||||
// Unregister session
|
||||
Server.UnregisterSession(Id);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Send/Recieve data
|
||||
|
||||
// Receive buffer
|
||||
private bool _receiving;
|
||||
private Buffer _receiveBuffer;
|
||||
private SocketAsyncEventArgs _receiveEventArg;
|
||||
// Send buffer
|
||||
private readonly object _sendLock = new object();
|
||||
private bool _sending;
|
||||
private Buffer _sendBufferMain;
|
||||
private Buffer _sendBufferFlush;
|
||||
private SocketAsyncEventArgs _sendEventArg;
|
||||
private long _sendBufferFlushOffset;
|
||||
|
||||
/// <summary>
|
||||
/// Send data to the client (synchronous)
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to send</param>
|
||||
/// <returns>Size of sent data</returns>
|
||||
public virtual long Send(byte[] buffer) { return Send(buffer, 0, buffer.Length); }
|
||||
|
||||
/// <summary>
|
||||
/// Send data to the client (synchronous)
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to send</param>
|
||||
/// <param name="offset">Buffer offset</param>
|
||||
/// <param name="size">Buffer size</param>
|
||||
/// <returns>Size of sent data</returns>
|
||||
public virtual long Send(byte[] buffer, long offset, long size)
|
||||
{
|
||||
if (!IsConnected)
|
||||
return 0;
|
||||
|
||||
if (size == 0)
|
||||
return 0;
|
||||
|
||||
// Sent data to the client
|
||||
long sent = Socket.Send(buffer, (int)offset, (int)size, SocketFlags.None, out SocketError ec);
|
||||
if (sent > 0)
|
||||
{
|
||||
// Update statistic
|
||||
BytesSent += sent;
|
||||
Interlocked.Add(ref Server._bytesSent, size);
|
||||
|
||||
// Call the buffer sent handler
|
||||
OnSent(sent, BytesPending + BytesSending);
|
||||
}
|
||||
|
||||
// Check for socket error
|
||||
if (ec != SocketError.Success)
|
||||
{
|
||||
SendError(ec);
|
||||
Disconnect();
|
||||
}
|
||||
|
||||
return sent;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Send text to the client (synchronous)
|
||||
/// </summary>
|
||||
/// <param name="text">Text string to send</param>
|
||||
/// <returns>Size of sent data</returns>
|
||||
public virtual long Send(string text) { return Send(Encoding.UTF8.GetBytes(text)); }
|
||||
|
||||
/// <summary>
|
||||
/// Send data to the client (asynchronous)
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to send</param>
|
||||
/// <returns>'true' if the data was successfully sent, 'false' if the session is not connected</returns>
|
||||
public virtual bool SendAsync(byte[] buffer) { return SendAsync(buffer, 0, buffer.Length); }
|
||||
|
||||
/// <summary>
|
||||
/// Send data to the client (asynchronous)
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to send</param>
|
||||
/// <param name="offset">Buffer offset</param>
|
||||
/// <param name="size">Buffer size</param>
|
||||
/// <returns>'true' if the data was successfully sent, 'false' if the session is not connected</returns>
|
||||
public virtual bool SendAsync(byte[] buffer, long offset, long size)
|
||||
{
|
||||
if (!IsConnected)
|
||||
return false;
|
||||
|
||||
if (size == 0)
|
||||
return true;
|
||||
|
||||
lock (_sendLock)
|
||||
{
|
||||
// Fill the main send buffer
|
||||
_sendBufferMain.Append(buffer, offset, size);
|
||||
|
||||
// Update statistic
|
||||
BytesPending = _sendBufferMain.Size;
|
||||
|
||||
// Avoid multiple send handlers
|
||||
if (_sending)
|
||||
return true;
|
||||
else
|
||||
_sending = true;
|
||||
|
||||
// Try to send the main buffer
|
||||
Task.Factory.StartNew(TrySend);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Send text to the client (asynchronous)
|
||||
/// </summary>
|
||||
/// <param name="text">Text string to send</param>
|
||||
/// <returns>'true' if the text was successfully sent, 'false' if the session is not connected</returns>
|
||||
public virtual bool SendAsync(string text) { return SendAsync(Encoding.UTF8.GetBytes(text)); }
|
||||
|
||||
/// <summary>
|
||||
/// Receive data from the client (synchronous)
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to receive</param>
|
||||
/// <returns>Size of received data</returns>
|
||||
public virtual long Receive(byte[] buffer) { return Receive(buffer, 0, buffer.Length); }
|
||||
|
||||
/// <summary>
|
||||
/// Receive data from the client (synchronous)
|
||||
/// </summary>
|
||||
/// <param name="buffer">Buffer to receive</param>
|
||||
/// <param name="offset">Buffer offset</param>
|
||||
/// <param name="size">Buffer size</param>
|
||||
/// <returns>Size of received data</returns>
|
||||
public virtual long Receive(byte[] buffer, long offset, long size)
|
||||
{
|
||||
if (!IsConnected)
|
||||
return 0;
|
||||
|
||||
if (size == 0)
|
||||
return 0;
|
||||
|
||||
// Receive data from the client
|
||||
long received = Socket.Receive(buffer, (int)offset, (int)size, SocketFlags.None, out SocketError ec);
|
||||
if (received > 0)
|
||||
{
|
||||
// Update statistic
|
||||
BytesReceived += received;
|
||||
Interlocked.Add(ref Server._bytesReceived, received);
|
||||
|
||||
// Call the buffer received handler
|
||||
OnReceived(buffer, 0, received);
|
||||
}
|
||||
|
||||
// Check for socket error
|
||||
if (ec != SocketError.Success)
|
||||
{
|
||||
SendError(ec);
|
||||
Disconnect();
|
||||
}
|
||||
|
||||
return received;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Receive text from the client (synchronous)
|
||||
/// </summary>
|
||||
/// <param name="size">Text size to receive</param>
|
||||
/// <returns>Received text</returns>
|
||||
public virtual string Receive(long size)
|
||||
{
|
||||
var buffer = new byte[size];
|
||||
var length = Receive(buffer);
|
||||
return Encoding.UTF8.GetString(buffer, 0, (int)length);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Receive data from the client (asynchronous)
|
||||
/// </summary>
|
||||
public virtual void ReceiveAsync()
|
||||
{
|
||||
// Try to receive data from the client
|
||||
TryReceive();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Try to receive new data
|
||||
/// </summary>
|
||||
private void TryReceive()
|
||||
{
|
||||
if (_receiving)
|
||||
return;
|
||||
|
||||
if (!IsConnected)
|
||||
return;
|
||||
|
||||
bool process = true;
|
||||
|
||||
while (process)
|
||||
{
|
||||
process = false;
|
||||
|
||||
try
|
||||
{
|
||||
// Async receive with the receive handler
|
||||
_receiving = true;
|
||||
_receiveEventArg.SetBuffer(_receiveBuffer.Data, 0, (int)_receiveBuffer.Capacity);
|
||||
if (!Socket.ReceiveAsync(_receiveEventArg))
|
||||
process = ProcessReceive(_receiveEventArg);
|
||||
}
|
||||
catch (ObjectDisposedException) {}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Try to send pending data
|
||||
/// </summary>
|
||||
private void TrySend()
|
||||
{
|
||||
if (!IsConnected)
|
||||
return;
|
||||
|
||||
bool empty = false;
|
||||
bool process = true;
|
||||
|
||||
while (process)
|
||||
{
|
||||
process = false;
|
||||
|
||||
lock (_sendLock)
|
||||
{
|
||||
// Is previous socket send in progress?
|
||||
if (_sendBufferFlush.IsEmpty)
|
||||
{
|
||||
// Swap flush and main buffers
|
||||
_sendBufferFlush = Interlocked.Exchange(ref _sendBufferMain, _sendBufferFlush);
|
||||
_sendBufferFlushOffset = 0;
|
||||
|
||||
// Update statistic
|
||||
BytesPending = 0;
|
||||
BytesSending += _sendBufferFlush.Size;
|
||||
|
||||
// Check if the flush buffer is empty
|
||||
if (_sendBufferFlush.IsEmpty)
|
||||
{
|
||||
// Need to call empty send buffer handler
|
||||
empty = true;
|
||||
|
||||
// End sending process
|
||||
_sending = false;
|
||||
}
|
||||
}
|
||||
else
|
||||
return;
|
||||
}
|
||||
|
||||
// Call the empty send buffer handler
|
||||
if (empty)
|
||||
{
|
||||
OnEmpty();
|
||||
return;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
// Async write with the write handler
|
||||
_sendEventArg.SetBuffer(_sendBufferFlush.Data, (int)_sendBufferFlushOffset, (int)(_sendBufferFlush.Size - _sendBufferFlushOffset));
|
||||
if (!Socket.SendAsync(_sendEventArg))
|
||||
process = ProcessSend(_sendEventArg);
|
||||
}
|
||||
catch (ObjectDisposedException) {}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Clear send/receive buffers
|
||||
/// </summary>
|
||||
private void ClearBuffers()
|
||||
{
|
||||
lock (_sendLock)
|
||||
{
|
||||
// Clear send buffers
|
||||
_sendBufferMain.Clear();
|
||||
_sendBufferFlush.Clear();
|
||||
_sendBufferFlushOffset= 0;
|
||||
|
||||
// Update statistic
|
||||
BytesPending = 0;
|
||||
BytesSending = 0;
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region IO processing
|
||||
|
||||
/// <summary>
|
||||
/// This method is called whenever a receive or send operation is completed on a socket
|
||||
/// </summary>
|
||||
private void OnAsyncCompleted(object sender, SocketAsyncEventArgs e)
|
||||
{
|
||||
// Determine which type of operation just completed and call the associated handler
|
||||
switch (e.LastOperation)
|
||||
{
|
||||
case SocketAsyncOperation.Receive:
|
||||
if (ProcessReceive(e))
|
||||
TryReceive();
|
||||
break;
|
||||
case SocketAsyncOperation.Send:
|
||||
if (ProcessSend(e))
|
||||
TrySend();
|
||||
break;
|
||||
default:
|
||||
throw new ArgumentException("The last operation completed on the socket was not a receive or send");
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// This method is invoked when an asynchronous receive operation completes
|
||||
/// </summary>
|
||||
private bool ProcessReceive(SocketAsyncEventArgs e)
|
||||
{
|
||||
if (!IsConnected)
|
||||
return false;
|
||||
|
||||
long size = e.BytesTransferred;
|
||||
|
||||
// Received some data from the client
|
||||
if (size > 0)
|
||||
{
|
||||
// Update statistic
|
||||
BytesReceived += size;
|
||||
Interlocked.Add(ref Server._bytesReceived, size);
|
||||
|
||||
// Call the buffer received handler
|
||||
OnReceived(_receiveBuffer.Data, 0, size);
|
||||
|
||||
// If the receive buffer is full increase its size
|
||||
if (_receiveBuffer.Capacity == size)
|
||||
_receiveBuffer.Reserve(2 * size);
|
||||
}
|
||||
|
||||
_receiving = false;
|
||||
|
||||
// Try to receive again if the session is valid
|
||||
if (e.SocketError == SocketError.Success)
|
||||
{
|
||||
// If zero is returned from a read operation, the remote end has closed the connection
|
||||
if (size > 0)
|
||||
return true;
|
||||
else
|
||||
Disconnect();
|
||||
}
|
||||
else
|
||||
{
|
||||
SendError(e.SocketError);
|
||||
Disconnect();
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// This method is invoked when an asynchronous send operation completes
|
||||
/// </summary>
|
||||
private bool ProcessSend(SocketAsyncEventArgs e)
|
||||
{
|
||||
if (!IsConnected)
|
||||
return false;
|
||||
|
||||
long size = e.BytesTransferred;
|
||||
|
||||
// Send some data to the client
|
||||
if (size > 0)
|
||||
{
|
||||
// Update statistic
|
||||
BytesSending -= size;
|
||||
BytesSent += size;
|
||||
Interlocked.Add(ref Server._bytesSent, size);
|
||||
|
||||
// Increase the flush buffer offset
|
||||
_sendBufferFlushOffset += size;
|
||||
|
||||
// Successfully send the whole flush buffer
|
||||
if (_sendBufferFlushOffset == _sendBufferFlush.Size)
|
||||
{
|
||||
// Clear the flush buffer
|
||||
_sendBufferFlush.Clear();
|
||||
_sendBufferFlushOffset = 0;
|
||||
}
|
||||
|
||||
// Call the buffer sent handler
|
||||
OnSent(size, BytesPending + BytesSending);
|
||||
}
|
||||
|
||||
// Try to send again if the session is valid
|
||||
if (e.SocketError == SocketError.Success)
|
||||
return true;
|
||||
else
|
||||
{
|
||||
SendError(e.SocketError);
|
||||
Disconnect();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Session handlers
|
||||
|
||||
/// <summary>
|
||||
/// Handle client connected notification
|
||||
/// </summary>
|
||||
protected virtual void OnConnected() {}
|
||||
/// <summary>
|
||||
/// Handle client disconnected notification
|
||||
/// </summary>
|
||||
protected virtual void OnDisconnected() {}
|
||||
|
||||
/// <summary>
|
||||
/// Handle buffer received notification
|
||||
/// </summary>
|
||||
/// <param name="buffer">Received buffer</param>
|
||||
/// <param name="offset">Received buffer offset</param>
|
||||
/// <param name="size">Received buffer size</param>
|
||||
/// <remarks>
|
||||
/// Notification is called when another chunk of buffer was received from the client
|
||||
/// </remarks>
|
||||
protected virtual void OnReceived(byte[] buffer, long offset, long size) {}
|
||||
/// <summary>
|
||||
/// Handle buffer sent notification
|
||||
/// </summary>
|
||||
/// <param name="sent">Size of sent buffer</param>
|
||||
/// <param name="pending">Size of pending buffer</param>
|
||||
/// <remarks>
|
||||
/// Notification is called when another chunk of buffer was sent to the client.
|
||||
/// This handler could be used to send another buffer to the client for instance when the pending size is zero.
|
||||
/// </remarks>
|
||||
protected virtual void OnSent(long sent, long pending) {}
|
||||
|
||||
/// <summary>
|
||||
/// Handle empty send buffer notification
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// Notification is called when the send buffer is empty and ready for a new data to send.
|
||||
/// This handler could be used to send another buffer to the client.
|
||||
/// </remarks>
|
||||
protected virtual void OnEmpty() {}
|
||||
|
||||
/// <summary>
|
||||
/// Handle error notification
|
||||
/// </summary>
|
||||
/// <param name="error">Socket error code</param>
|
||||
protected virtual void OnError(SocketError error) {}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Error handling
|
||||
|
||||
/// <summary>
|
||||
/// Send error notification
|
||||
/// </summary>
|
||||
/// <param name="error">Socket error code</param>
|
||||
private void SendError(SocketError error)
|
||||
{
|
||||
// Skip disconnect errors
|
||||
if ((error == SocketError.ConnectionAborted) ||
|
||||
(error == SocketError.ConnectionRefused) ||
|
||||
(error == SocketError.ConnectionReset) ||
|
||||
(error == SocketError.OperationAborted) ||
|
||||
(error == SocketError.Shutdown))
|
||||
return;
|
||||
|
||||
OnError(error);
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region IDisposable implementation
|
||||
|
||||
/// <summary>
|
||||
/// Disposed flag
|
||||
/// </summary>
|
||||
public bool IsDisposed { get; private set; }
|
||||
|
||||
/// <summary>
|
||||
/// Session socket disposed flag
|
||||
/// </summary>
|
||||
public bool IsSocketDisposed { get; private set; } = true;
|
||||
|
||||
// Implement IDisposable.
|
||||
public void Dispose()
|
||||
{
|
||||
Dispose(true);
|
||||
GC.SuppressFinalize(this);
|
||||
}
|
||||
|
||||
protected virtual void Dispose(bool disposingManagedResources)
|
||||
{
|
||||
// The idea here is that Dispose(Boolean) knows whether it is
|
||||
// being called to do explicit cleanup (the Boolean is true)
|
||||
// versus being called due to a garbage collection (the Boolean
|
||||
// is false). This distinction is useful because, when being
|
||||
// disposed explicitly, the Dispose(Boolean) method can safely
|
||||
// execute code using reference type fields that refer to other
|
||||
// objects knowing for sure that these other objects have not been
|
||||
// finalized or disposed of yet. When the Boolean is false,
|
||||
// the Dispose(Boolean) method should not execute code that
|
||||
// refer to reference type fields because those objects may
|
||||
// have already been finalized."
|
||||
|
||||
if (!IsDisposed)
|
||||
{
|
||||
if (disposingManagedResources)
|
||||
{
|
||||
// Dispose managed resources here...
|
||||
Disconnect();
|
||||
}
|
||||
|
||||
// Dispose unmanaged resources here...
|
||||
|
||||
// Set large fields to null here...
|
||||
|
||||
// Mark as disposed.
|
||||
IsDisposed = true;
|
||||
}
|
||||
}
|
||||
|
||||
// Use C# destructor syntax for finalization code.
|
||||
~TcpSession()
|
||||
{
|
||||
// Simply call Dispose(false).
|
||||
Dispose(false);
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user