123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963 |
- using System;
- using System.Net;
- using System.Net.Sockets;
- using System.Text;
- using System.Threading;
- namespace NetCoreServer
- {
-
-
-
-
- public class UdsClient : IDisposable
- {
-
-
-
-
- public UdsClient(string path) : this(new UnixDomainSocketEndPoint(path)) {}
-
-
-
-
- public UdsClient(UnixDomainSocketEndPoint endpoint)
- {
- Id = Guid.NewGuid();
- Endpoint = endpoint;
- }
-
-
-
- public Guid Id { get; }
-
-
-
- public EndPoint Endpoint { get; private set; }
-
-
-
- public Socket Socket { get; private set; }
-
-
-
- public long BytesPending { get; private set; }
-
-
-
- public long BytesSending { get; private set; }
-
-
-
- public long BytesSent { get; private set; }
-
-
-
- public long BytesReceived { get; private set; }
-
-
-
- public int OptionReceiveBufferLimit { get; set; } = 0;
-
-
-
- public int OptionReceiveBufferSize { get; set; } = 8192;
-
-
-
- public int OptionSendBufferLimit { get; set; } = 0;
-
-
-
- public int OptionSendBufferSize { get; set; } = 8192;
- #region Connect/Disconnect client
- private SocketAsyncEventArgs _connectEventArg;
-
-
-
- public bool IsConnecting { get; private set; }
-
-
-
- public bool IsConnected { get; private set; }
-
-
-
-
-
-
-
- protected virtual Socket CreateSocket()
- {
- return new Socket(Endpoint.AddressFamily, SocketType.Stream, ProtocolType.IP);
- }
-
-
-
-
-
-
-
-
- public virtual bool Connect()
- {
- if (IsConnected || IsConnecting)
- return false;
-
- _receiveBuffer = new Buffer();
- _sendBufferMain = new Buffer();
- _sendBufferFlush = new Buffer();
-
- _connectEventArg = new SocketAsyncEventArgs();
- _connectEventArg.RemoteEndPoint = Endpoint;
- _connectEventArg.Completed += OnAsyncCompleted;
- _receiveEventArg = new SocketAsyncEventArgs();
- _receiveEventArg.Completed += OnAsyncCompleted;
- _sendEventArg = new SocketAsyncEventArgs();
- _sendEventArg.Completed += OnAsyncCompleted;
-
- Socket = CreateSocket();
-
- IsSocketDisposed = false;
-
- OnConnecting();
- try
- {
-
- Socket.Connect(Endpoint);
- }
- catch (SocketException ex)
- {
-
- SendError(ex.SocketErrorCode);
-
- _connectEventArg.Completed -= OnAsyncCompleted;
- _receiveEventArg.Completed -= OnAsyncCompleted;
- _sendEventArg.Completed -= OnAsyncCompleted;
-
- OnDisconnecting();
-
- Socket.Close();
-
- Socket.Dispose();
-
- _connectEventArg.Dispose();
- _receiveEventArg.Dispose();
- _sendEventArg.Dispose();
-
- OnDisconnected();
- return false;
- }
-
- _receiveBuffer.Reserve(OptionReceiveBufferSize);
- _sendBufferMain.Reserve(OptionSendBufferSize);
- _sendBufferFlush.Reserve(OptionSendBufferSize);
-
- BytesPending = 0;
- BytesSending = 0;
- BytesSent = 0;
- BytesReceived = 0;
-
- IsConnected = true;
-
- OnConnected();
-
- if (_sendBufferMain.IsEmpty)
- OnEmpty();
- return true;
- }
-
-
-
-
- public virtual bool Disconnect()
- {
- if (!IsConnected && !IsConnecting)
- return false;
-
- if (IsConnecting)
- Socket.CancelConnectAsync(_connectEventArg);
-
- _connectEventArg.Completed -= OnAsyncCompleted;
- _receiveEventArg.Completed -= OnAsyncCompleted;
- _sendEventArg.Completed -= OnAsyncCompleted;
-
- OnDisconnecting();
- try
- {
- try
- {
-
- Socket.Shutdown(SocketShutdown.Both);
- }
- catch (SocketException) {}
-
- Socket.Close();
-
- Socket.Dispose();
-
- _connectEventArg.Dispose();
- _receiveEventArg.Dispose();
- _sendEventArg.Dispose();
-
- IsSocketDisposed = true;
- }
- catch (ObjectDisposedException) {}
-
- IsConnected = false;
-
- _receiving = false;
- _sending = false;
-
- ClearBuffers();
-
- OnDisconnected();
- return true;
- }
-
-
-
-
- public virtual bool Reconnect()
- {
- if (!Disconnect())
- return false;
- return Connect();
- }
-
-
-
-
- public virtual bool ConnectAsync()
- {
- if (IsConnected || IsConnecting)
- return false;
-
- _receiveBuffer = new Buffer();
- _sendBufferMain = new Buffer();
- _sendBufferFlush = new Buffer();
-
- _connectEventArg = new SocketAsyncEventArgs();
- _connectEventArg.RemoteEndPoint = Endpoint;
- _connectEventArg.Completed += OnAsyncCompleted;
- _receiveEventArg = new SocketAsyncEventArgs();
- _receiveEventArg.Completed += OnAsyncCompleted;
- _sendEventArg = new SocketAsyncEventArgs();
- _sendEventArg.Completed += OnAsyncCompleted;
-
- Socket = CreateSocket();
-
- IsSocketDisposed = false;
-
- IsConnecting = true;
-
- OnConnecting();
-
- if (!Socket.ConnectAsync(_connectEventArg))
- ProcessConnect(_connectEventArg);
- return true;
- }
-
-
-
-
- public virtual bool DisconnectAsync() => Disconnect();
-
-
-
-
- public virtual bool ReconnectAsync()
- {
- if (!DisconnectAsync())
- return false;
- while (IsConnected)
- Thread.Yield();
- return ConnectAsync();
- }
- #endregion
- #region Send/Receive data
-
- private bool _receiving;
- private Buffer _receiveBuffer;
- private SocketAsyncEventArgs _receiveEventArg;
-
- private readonly object _sendLock = new object();
- private bool _sending;
- private Buffer _sendBufferMain;
- private Buffer _sendBufferFlush;
- private SocketAsyncEventArgs _sendEventArg;
- private long _sendBufferFlushOffset;
-
-
-
-
-
- public virtual long Send(byte[] buffer) => Send(buffer.AsSpan());
-
-
-
-
-
-
-
- public virtual long Send(byte[] buffer, long offset, long size) => Send(buffer.AsSpan((int)offset, (int)size));
-
-
-
-
-
- public virtual long Send(ReadOnlySpan<byte> buffer)
- {
- if (!IsConnected)
- return 0;
- if (buffer.IsEmpty)
- return 0;
-
- long sent = Socket.Send(buffer, SocketFlags.None, out SocketError ec);
- if (sent > 0)
- {
-
- BytesSent += sent;
-
- OnSent(sent, BytesPending + BytesSending);
- }
-
- if (ec != SocketError.Success)
- {
- SendError(ec);
- Disconnect();
- }
- return sent;
- }
-
-
-
-
-
- public virtual long Send(string text) => Send(Encoding.UTF8.GetBytes(text));
-
-
-
-
-
- public virtual long Send(ReadOnlySpan<char> text) => Send(Encoding.UTF8.GetBytes(text.ToArray()));
-
-
-
-
-
- public virtual bool SendAsync(byte[] buffer) => SendAsync(buffer.AsSpan());
-
-
-
-
-
-
-
- public virtual bool SendAsync(byte[] buffer, long offset, long size) => SendAsync(buffer.AsSpan((int)offset, (int)size));
-
-
-
-
-
- public virtual bool SendAsync(ReadOnlySpan<byte> buffer)
- {
- if (!IsConnected)
- return false;
- if (buffer.IsEmpty)
- return true;
- lock (_sendLock)
- {
-
- if (((_sendBufferMain.Size + buffer.Length) > OptionSendBufferLimit) && (OptionSendBufferLimit > 0))
- {
- SendError(SocketError.NoBufferSpaceAvailable);
- return false;
- }
-
- _sendBufferMain.Append(buffer);
-
- BytesPending = _sendBufferMain.Size;
-
- if (_sending)
- return true;
- else
- _sending = true;
-
- TrySend();
- }
- return true;
- }
-
-
-
-
-
- public virtual bool SendAsync(string text) => SendAsync(Encoding.UTF8.GetBytes(text));
-
-
-
-
-
- public virtual bool SendAsync(ReadOnlySpan<char> text) => SendAsync(Encoding.UTF8.GetBytes(text.ToArray()));
-
-
-
-
-
- public virtual long Receive(byte[] buffer) { return Receive(buffer, 0, buffer.Length); }
-
-
-
-
-
-
-
- public virtual long Receive(byte[] buffer, long offset, long size)
- {
- if (!IsConnected)
- return 0;
- if (size == 0)
- return 0;
-
- long received = Socket.Receive(buffer, (int)offset, (int)size, SocketFlags.None, out SocketError ec);
- if (received > 0)
- {
-
- BytesReceived += received;
-
- OnReceived(buffer, 0, received);
- }
-
- if (ec != SocketError.Success)
- {
- SendError(ec);
- Disconnect();
- }
- return received;
- }
-
-
-
-
-
- public virtual string Receive(long size)
- {
- var buffer = new byte[size];
- var length = Receive(buffer);
- return Encoding.UTF8.GetString(buffer, 0, (int)length);
- }
-
-
-
- public virtual void ReceiveAsync()
- {
-
- TryReceive();
- }
-
-
-
- private void TryReceive()
- {
- if (_receiving)
- return;
- if (!IsConnected)
- return;
- bool process = true;
- while (process)
- {
- process = false;
- try
- {
-
- _receiving = true;
- _receiveEventArg.SetBuffer(_receiveBuffer.Data, 0, (int)_receiveBuffer.Capacity);
- if (!Socket.ReceiveAsync(_receiveEventArg))
- process = ProcessReceive(_receiveEventArg);
- }
- catch (ObjectDisposedException) {}
- }
- }
-
-
-
- private void TrySend()
- {
- if (!IsConnected)
- return;
- bool empty = false;
- bool process = true;
- while (process)
- {
- process = false;
- lock (_sendLock)
- {
-
- if (_sendBufferFlush.IsEmpty)
- {
-
- _sendBufferFlush = Interlocked.Exchange(ref _sendBufferMain, _sendBufferFlush);
- _sendBufferFlushOffset = 0;
-
- BytesPending = 0;
- BytesSending += _sendBufferFlush.Size;
-
- if (_sendBufferFlush.IsEmpty)
- {
-
- empty = true;
-
- _sending = false;
- }
- }
- else
- return;
- }
-
- if (empty)
- {
- OnEmpty();
- return;
- }
- try
- {
-
- _sendEventArg.SetBuffer(_sendBufferFlush.Data, (int)_sendBufferFlushOffset, (int)(_sendBufferFlush.Size - _sendBufferFlushOffset));
- if (!Socket.SendAsync(_sendEventArg))
- process = ProcessSend(_sendEventArg);
- }
- catch (ObjectDisposedException) {}
- }
- }
-
-
-
- private void ClearBuffers()
- {
- lock (_sendLock)
- {
-
- _sendBufferMain.Clear();
- _sendBufferFlush.Clear();
- _sendBufferFlushOffset= 0;
-
- BytesPending = 0;
- BytesSending = 0;
- }
- }
- #endregion
- #region IO processing
-
-
-
- private void OnAsyncCompleted(object sender, SocketAsyncEventArgs e)
- {
- if (IsSocketDisposed)
- return;
-
- switch (e.LastOperation)
- {
- case SocketAsyncOperation.Connect:
- ProcessConnect(e);
- break;
- 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");
- }
- }
-
-
-
- private void ProcessConnect(SocketAsyncEventArgs e)
- {
- IsConnecting = false;
- if (e.SocketError == SocketError.Success)
- {
-
- _receiveBuffer.Reserve(OptionReceiveBufferSize);
- _sendBufferMain.Reserve(OptionSendBufferSize);
- _sendBufferFlush.Reserve(OptionSendBufferSize);
-
- BytesPending = 0;
- BytesSending = 0;
- BytesSent = 0;
- BytesReceived = 0;
-
- IsConnected = true;
-
- TryReceive();
-
- if (IsSocketDisposed)
- return;
-
- OnConnected();
-
- if (_sendBufferMain.IsEmpty)
- OnEmpty();
- }
- else
- {
-
- SendError(e.SocketError);
- OnDisconnected();
- }
- }
-
-
-
- private bool ProcessReceive(SocketAsyncEventArgs e)
- {
- if (!IsConnected)
- return false;
- long size = e.BytesTransferred;
-
- if (size > 0)
- {
-
- BytesReceived += size;
-
- OnReceived(_receiveBuffer.Data, 0, size);
-
- if (_receiveBuffer.Capacity == size)
- {
-
- if (((2 * size) > OptionReceiveBufferLimit) && (OptionReceiveBufferLimit > 0))
- {
- SendError(SocketError.NoBufferSpaceAvailable);
- DisconnectAsync();
- return false;
- }
- _receiveBuffer.Reserve(2 * size);
- }
- }
- _receiving = false;
-
- if (e.SocketError == SocketError.Success)
- {
-
- if (size > 0)
- return true;
- else
- DisconnectAsync();
- }
- else
- {
- SendError(e.SocketError);
- DisconnectAsync();
- }
- return false;
- }
-
-
-
- private bool ProcessSend(SocketAsyncEventArgs e)
- {
- if (!IsConnected)
- return false;
- long size = e.BytesTransferred;
-
- if (size > 0)
- {
-
- BytesSending -= size;
- BytesSent += size;
-
- _sendBufferFlushOffset += size;
-
- if (_sendBufferFlushOffset == _sendBufferFlush.Size)
- {
-
- _sendBufferFlush.Clear();
- _sendBufferFlushOffset = 0;
- }
-
- OnSent(size, BytesPending + BytesSending);
- }
-
- if (e.SocketError == SocketError.Success)
- return true;
- else
- {
- SendError(e.SocketError);
- DisconnectAsync();
- return false;
- }
- }
- #endregion
- #region Session handlers
-
-
-
- protected virtual void OnConnecting() {}
-
-
-
- protected virtual void OnConnected() {}
-
-
-
- protected virtual void OnDisconnecting() {}
-
-
-
- protected virtual void OnDisconnected() {}
-
-
-
-
-
-
-
-
-
- protected virtual void OnReceived(byte[] buffer, long offset, long size) {}
-
-
-
-
-
-
-
-
-
- protected virtual void OnSent(long sent, long pending) {}
-
-
-
-
-
-
-
- protected virtual void OnEmpty() {}
-
-
-
-
- protected virtual void OnError(SocketError error) {}
- #endregion
- #region Error handling
-
-
-
-
- private void SendError(SocketError error)
- {
-
- if ((error == SocketError.ConnectionAborted) ||
- (error == SocketError.ConnectionRefused) ||
- (error == SocketError.ConnectionReset) ||
- (error == SocketError.OperationAborted) ||
- (error == SocketError.Shutdown))
- return;
- OnError(error);
- }
- #endregion
- #region IDisposable implementation
-
-
-
- public bool IsDisposed { get; private set; }
-
-
-
- public bool IsSocketDisposed { get; private set; } = true;
-
- public void Dispose()
- {
- Dispose(true);
- GC.SuppressFinalize(this);
- }
- protected virtual void Dispose(bool disposingManagedResources)
- {
-
-
-
-
-
-
-
-
-
-
-
- if (!IsDisposed)
- {
- if (disposingManagedResources)
- {
-
- DisconnectAsync();
- }
-
-
-
- IsDisposed = true;
- }
- }
- #endregion
- }
- }
|