jellyfin/Emby.Common.Implementations/Net/NetAcceptSocket.cs

105 lines
2.5 KiB
C#
Raw Normal View History

2016-11-08 18:44:23 +00:00
using System;
using System.Net;
using System.Net.Sockets;
using System.Threading;
using Emby.Common.Implementations.Networking;
using MediaBrowser.Model.Net;
using MediaBrowser.Model.Logging;
namespace Emby.Common.Implementations.Net
{
2017-03-02 20:50:09 +00:00
public class NetAcceptSocket : IAcceptSocket
2016-11-08 18:44:23 +00:00
{
public Socket Socket { get; private set; }
private readonly ILogger _logger;
2016-12-07 20:02:34 +00:00
public bool DualMode { get; private set; }
2017-03-02 20:50:09 +00:00
public NetAcceptSocket(Socket socket, ILogger logger, bool isDualMode)
2016-11-08 18:44:23 +00:00
{
2016-11-13 21:04:21 +00:00
if (socket == null)
{
throw new ArgumentNullException("socket");
}
if (logger == null)
{
throw new ArgumentNullException("logger");
}
2016-11-08 18:44:23 +00:00
Socket = socket;
_logger = logger;
2016-12-07 20:02:34 +00:00
DualMode = isDualMode;
2016-11-08 18:44:23 +00:00
}
public IpEndPointInfo LocalEndPoint
{
get
{
2016-11-11 07:24:36 +00:00
return NetworkManager.ToIpEndPointInfo((IPEndPoint)Socket.LocalEndPoint);
2016-11-08 18:44:23 +00:00
}
}
public IpEndPointInfo RemoteEndPoint
{
get
{
2016-11-11 07:24:36 +00:00
return NetworkManager.ToIpEndPointInfo((IPEndPoint)Socket.RemoteEndPoint);
2016-11-08 18:44:23 +00:00
}
}
2017-03-03 05:53:21 +00:00
public void Connect(IpEndPointInfo endPoint)
{
var nativeEndpoint = NetworkManager.ToIPEndPoint(endPoint);
Socket.Connect(nativeEndpoint);
}
2016-11-08 18:44:23 +00:00
public void Close()
{
#if NET46
Socket.Close();
#else
Socket.Dispose();
#endif
}
public void Shutdown(bool both)
{
if (both)
{
Socket.Shutdown(SocketShutdown.Both);
}
else
{
// Change interface if ever needed
throw new NotImplementedException();
}
}
public void Listen(int backlog)
{
Socket.Listen(backlog);
}
public void Bind(IpEndPointInfo endpoint)
{
2016-11-11 07:24:36 +00:00
var nativeEndpoint = NetworkManager.ToIPEndPoint(endpoint);
2016-11-08 18:44:23 +00:00
Socket.Bind(nativeEndpoint);
}
private SocketAcceptor _acceptor;
2017-03-02 20:50:09 +00:00
public void StartAccept(Action<IAcceptSocket> onAccept, Func<bool> isClosed)
2016-11-08 18:44:23 +00:00
{
2016-12-07 20:02:34 +00:00
_acceptor = new SocketAcceptor(_logger, Socket, onAccept, isClosed, DualMode);
2016-11-08 18:44:23 +00:00
_acceptor.StartAccept();
}
public void Dispose()
{
Socket.Dispose();
}
}
}