378 lines
9.3 KiB
C#
378 lines
9.3 KiB
C#
|
using System;
|
|||
|
using System.Net;
|
|||
|
using System.Net.Sockets;
|
|||
|
using System.Text;
|
|||
|
using System.Threading;
|
|||
|
using Aitex.Core.RT.Event;
|
|||
|
using Aitex.Core.RT.Log;
|
|||
|
using MECF.Framework.Common.Equipment;
|
|||
|
|
|||
|
namespace MECF.Framework.Common.Communications
|
|||
|
{
|
|||
|
public class AsynSocketClient : IDisposable
|
|||
|
{
|
|||
|
public delegate void ErrorHandler(TCPErrorEventArgs args);
|
|||
|
|
|||
|
public delegate void MessageHandler(string message);
|
|||
|
|
|||
|
public delegate void BinaryMessageHandler(byte[] message);
|
|||
|
|
|||
|
public class ClientStateObject
|
|||
|
{
|
|||
|
public Socket workSocket = null;
|
|||
|
|
|||
|
public static int BufferSize = 256;
|
|||
|
|
|||
|
public byte[] buffer = new byte[BufferSize];
|
|||
|
|
|||
|
public StringBuilder sb = new StringBuilder();
|
|||
|
|
|||
|
public ClientStateObject(int bufferSize = 256)
|
|||
|
{
|
|||
|
BufferSize = bufferSize;
|
|||
|
buffer = new byte[bufferSize];
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
private static object _locker = new object();
|
|||
|
|
|||
|
private Socket _socket;
|
|||
|
|
|||
|
private string _ip;
|
|||
|
|
|||
|
private int _port;
|
|||
|
|
|||
|
private string _address;
|
|||
|
|
|||
|
private int _bufferSize = 256;
|
|||
|
|
|||
|
private bool _isEndConnect = false;
|
|||
|
|
|||
|
private bool _isAsciiMode;
|
|||
|
|
|||
|
public string NewLine { get; set; }
|
|||
|
|
|||
|
public bool NeedLog { get; set; } = true;
|
|||
|
|
|||
|
|
|||
|
public bool IsConnected => _socket != null && _socket.Connected && _isEndConnect;
|
|||
|
|
|||
|
public bool IsHstConnected => _socket != null && IsSocketConnected(_socket);
|
|||
|
|
|||
|
public event ErrorHandler OnErrorHappened;
|
|||
|
|
|||
|
public event MessageHandler OnDataChanged;
|
|||
|
|
|||
|
public event BinaryMessageHandler OnBinaryDataChanged;
|
|||
|
|
|||
|
private bool IsSocketConnected(Socket client)
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
byte[] buffer = new byte[1];
|
|||
|
int num = client.Send(buffer);
|
|||
|
if (num == 1)
|
|||
|
{
|
|||
|
return true;
|
|||
|
}
|
|||
|
return false;
|
|||
|
}
|
|||
|
catch (SocketException ex)
|
|||
|
{
|
|||
|
LOG.Write(ex.Message);
|
|||
|
return false;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
public AsynSocketClient(string address, bool isAsciiMode, string newline = "\r")
|
|||
|
{
|
|||
|
_socket = null;
|
|||
|
NewLine = newline;
|
|||
|
_address = address;
|
|||
|
_isAsciiMode = isAsciiMode;
|
|||
|
}
|
|||
|
|
|||
|
public AsynSocketClient(string address, int bufferSize, string newline = "\r")
|
|||
|
{
|
|||
|
_socket = null;
|
|||
|
NewLine = newline;
|
|||
|
_address = address;
|
|||
|
_bufferSize = bufferSize;
|
|||
|
}
|
|||
|
|
|||
|
~AsynSocketClient()
|
|||
|
{
|
|||
|
Dispose();
|
|||
|
}
|
|||
|
|
|||
|
public void Connect()
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
_ip = _address.Split(':')[0];
|
|||
|
_port = int.Parse(_address.Split(':')[1]);
|
|||
|
IPAddress address = IPAddress.Parse(_ip);
|
|||
|
IPEndPoint remoteEP = new IPEndPoint(address, _port);
|
|||
|
lock (_locker)
|
|||
|
{
|
|||
|
_isEndConnect = false;
|
|||
|
Dispose();
|
|||
|
if (NeedLog)
|
|||
|
{
|
|||
|
LOG.Info($"Start new socket of {_address}.", isTraceOn: false);
|
|||
|
}
|
|||
|
if (_socket == null)
|
|||
|
{
|
|||
|
_socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
|
|||
|
}
|
|||
|
_socket.BeginConnect(remoteEP, ConnectCallback, _socket);
|
|||
|
}
|
|||
|
}
|
|||
|
catch (Exception ex)
|
|||
|
{
|
|||
|
LOG.Write(ex);
|
|||
|
throw new Exception(ex.ToString());
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
public bool Poll()
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
return _socket.Poll(1, SelectMode.SelectRead);
|
|||
|
}
|
|||
|
catch (Exception)
|
|||
|
{
|
|||
|
return true;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
private void ConnectCallback(IAsyncResult ar)
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
if (NeedLog)
|
|||
|
{
|
|||
|
LOG.Info($"ConnectCallback {_address}", isTraceOn: false);
|
|||
|
}
|
|||
|
Socket socket = (Socket)ar.AsyncState;
|
|||
|
if (socket.Connected)
|
|||
|
{
|
|||
|
socket.EndConnect(ar);
|
|||
|
_isEndConnect = true;
|
|||
|
if (NeedLog)
|
|||
|
{
|
|||
|
LOG.Info($"EndConnect", isTraceOn: false);
|
|||
|
}
|
|||
|
EV.PostMessage(ModuleName.Robot.ToString(), EventEnum.TCPConnSucess, _ip, _port.ToString());
|
|||
|
Receive(_socket);
|
|||
|
}
|
|||
|
}
|
|||
|
catch (Exception ex)
|
|||
|
{
|
|||
|
LOG.Write(ex);
|
|||
|
string message = $"Communication {_ip}:{_port:D} {ex}.";
|
|||
|
LOG.Error(message);
|
|||
|
Thread.Sleep(1000);
|
|||
|
Connect();
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
private void Receive(Socket client)
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
ClientStateObject clientStateObject = new ClientStateObject(_bufferSize);
|
|||
|
clientStateObject.workSocket = client;
|
|||
|
client.BeginReceive(clientStateObject.buffer, 0, ClientStateObject.BufferSize, SocketFlags.None, ReceiveCallback, clientStateObject);
|
|||
|
}
|
|||
|
catch (Exception ex)
|
|||
|
{
|
|||
|
LOG.Write(ex);
|
|||
|
string text = $"TCP连接发生错误:{ex.Message}";
|
|||
|
LOG.Error($"Communication {_ip}:{_port:D} {text}.");
|
|||
|
this.OnErrorHappened(new TCPErrorEventArgs(text));
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
private void ReceiveCallback(IAsyncResult ar)
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
if (!IsConnected)
|
|||
|
{
|
|||
|
return;
|
|||
|
}
|
|||
|
ClientStateObject clientStateObject = (ClientStateObject)ar.AsyncState;
|
|||
|
Socket workSocket = clientStateObject.workSocket;
|
|||
|
if (workSocket == null || !workSocket.Connected)
|
|||
|
{
|
|||
|
return;
|
|||
|
}
|
|||
|
int num = workSocket.EndReceive(ar);
|
|||
|
if (num <= 0)
|
|||
|
{
|
|||
|
return;
|
|||
|
}
|
|||
|
clientStateObject.sb.Append(Encoding.ASCII.GetString(clientStateObject.buffer, 0, num));
|
|||
|
string @string = Encoding.ASCII.GetString(clientStateObject.buffer, 0, num);
|
|||
|
if (!_isAsciiMode)
|
|||
|
{
|
|||
|
byte[] array = new byte[num];
|
|||
|
for (int i = 0; i < num; i++)
|
|||
|
{
|
|||
|
array[i] = clientStateObject.buffer[i];
|
|||
|
}
|
|||
|
if (NeedLog)
|
|||
|
{
|
|||
|
LOG.Info(string.Format("Communication {0}:{1:D} receive {2}.", _ip, _port, string.Join(" ", Array.ConvertAll(array, (byte x) => x.ToString("X2")))), isTraceOn: false);
|
|||
|
LOG.Info($"Communication {_ip}:{_port:D} receive {Encoding.ASCII.GetString(array)} in ASCII.", isTraceOn: false);
|
|||
|
}
|
|||
|
this.OnBinaryDataChanged(array);
|
|||
|
clientStateObject.sb.Clear();
|
|||
|
}
|
|||
|
else if (clientStateObject.sb.Length > NewLine.Length && clientStateObject.sb.ToString().Substring(clientStateObject.sb.Length - NewLine.Length).Equals(NewLine))
|
|||
|
{
|
|||
|
string text = clientStateObject.sb.ToString();
|
|||
|
if (NeedLog)
|
|||
|
{
|
|||
|
LOG.Info($"Communication {_ip}:{_port:D} receive {text.TrimEnd('\n').TrimEnd('\r')}.", isTraceOn: false);
|
|||
|
LOG.Info(string.Format("Communication {0}:{1:D} receive {2}. in BIN", _ip, _port, string.Join(" ", Array.ConvertAll(Encoding.ASCII.GetBytes(text), (byte x) => x.ToString("X2")))), isTraceOn: false);
|
|||
|
}
|
|||
|
this.OnDataChanged(clientStateObject.sb.ToString());
|
|||
|
clientStateObject.sb.Clear();
|
|||
|
}
|
|||
|
workSocket.BeginReceive(clientStateObject.buffer, 0, ClientStateObject.BufferSize, SocketFlags.None, ReceiveCallback, clientStateObject);
|
|||
|
}
|
|||
|
catch (Exception ex)
|
|||
|
{
|
|||
|
LOG.Write(ex);
|
|||
|
string text2 = $"TCP Socket recevice data failed:{ex.Message}";
|
|||
|
LOG.Error($"Communication {_ip}:{_port:D} {text2}.");
|
|||
|
this.OnErrorHappened(new TCPErrorEventArgs(text2));
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
public bool Write(string data)
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
lock (_locker)
|
|||
|
{
|
|||
|
byte[] bytes = Encoding.ASCII.GetBytes(data);
|
|||
|
_socket.BeginSend(bytes, 0, bytes.Length, SocketFlags.None, SendCallback, _socket);
|
|||
|
if (NeedLog)
|
|||
|
{
|
|||
|
LOG.Info($"Communication {_ip}:{_port:D} Send {data}.", isTraceOn: false);
|
|||
|
string arg = string.Join(" ", Array.ConvertAll(Encoding.ASCII.GetBytes(data), (byte x) => x.ToString("X2")));
|
|||
|
LOG.Info($"Communication {_ip}:{_port:D} Send {arg} in Bin.", isTraceOn: false);
|
|||
|
}
|
|||
|
}
|
|||
|
return true;
|
|||
|
}
|
|||
|
catch (Exception ex)
|
|||
|
{
|
|||
|
LOG.Write(ex);
|
|||
|
LOG.Info($"Communication {_ip}:{_port:D} Send {data}. failed", isTraceOn: false);
|
|||
|
string reason = $"Send command failed:{ex.Message}";
|
|||
|
this.OnErrorHappened(new TCPErrorEventArgs(reason));
|
|||
|
}
|
|||
|
return false;
|
|||
|
}
|
|||
|
|
|||
|
public bool Write(byte[] byteData)
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
lock (_locker)
|
|||
|
{
|
|||
|
_socket.BeginSend(byteData, 0, byteData.Length, SocketFlags.None, SendCallback, _socket);
|
|||
|
if (NeedLog)
|
|||
|
{
|
|||
|
string arg = string.Join(" ", Array.ConvertAll(byteData, (byte x) => x.ToString("X2")));
|
|||
|
LOG.Info($"Communication {_ip}:{_port:D} Send {arg}.", isTraceOn: false);
|
|||
|
LOG.Info($"Communication {_ip}:{_port:D} Send {Encoding.ASCII.GetString(byteData)} in ASCII.", isTraceOn: false);
|
|||
|
}
|
|||
|
}
|
|||
|
return true;
|
|||
|
}
|
|||
|
catch (Exception ex)
|
|||
|
{
|
|||
|
LOG.Write(ex);
|
|||
|
string arg2 = string.Join(" ", Array.ConvertAll(byteData, (byte x) => x.ToString("X2")));
|
|||
|
LOG.Info($"Communication {_ip}:{_port:D} Send {arg2}. failed", isTraceOn: false);
|
|||
|
string reason = $"Send command failed:{ex.Message}";
|
|||
|
this.OnErrorHappened(new TCPErrorEventArgs(reason));
|
|||
|
}
|
|||
|
return false;
|
|||
|
}
|
|||
|
|
|||
|
private void SendCallback(IAsyncResult ar)
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
Socket socket = (Socket)ar.AsyncState;
|
|||
|
int num = socket.EndSend(ar);
|
|||
|
}
|
|||
|
catch (Exception ex)
|
|||
|
{
|
|||
|
LOG.Write(ex);
|
|||
|
string reason = $"Send command failed:{ex.Message}";
|
|||
|
this.OnErrorHappened(new TCPErrorEventArgs(reason));
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
public void Dispose()
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
if (_socket != null)
|
|||
|
{
|
|||
|
if (NeedLog)
|
|||
|
{
|
|||
|
LOG.Info($"Dispose current socket of {_address}", isTraceOn: false);
|
|||
|
}
|
|||
|
if (IsConnected)
|
|||
|
{
|
|||
|
_socket.Shutdown(SocketShutdown.Both);
|
|||
|
}
|
|||
|
_socket.Close();
|
|||
|
_socket.Dispose();
|
|||
|
_socket = null;
|
|||
|
}
|
|||
|
}
|
|||
|
catch (Exception ex)
|
|||
|
{
|
|||
|
LOG.Write(ex);
|
|||
|
string reason = $"释放socket资源失败:{ex.Message}";
|
|||
|
this.OnErrorHappened(new TCPErrorEventArgs(reason));
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
public void Close()
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
if (_socket != null)
|
|||
|
{
|
|||
|
if (NeedLog)
|
|||
|
{
|
|||
|
LOG.Info($"Close current socket of {_address}", isTraceOn: false);
|
|||
|
}
|
|||
|
if (IsConnected)
|
|||
|
{
|
|||
|
_socket.Shutdown(SocketShutdown.Both);
|
|||
|
}
|
|||
|
_socket.Close();
|
|||
|
}
|
|||
|
}
|
|||
|
catch (Exception ex)
|
|||
|
{
|
|||
|
LOG.Write(ex);
|
|||
|
string reason = $"Close socket资源失败:{ex.Message}";
|
|||
|
this.OnErrorHappened(new TCPErrorEventArgs(reason));
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
}
|