mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 11:34:57 +02:00
154 lines
4.4 KiB
C#
154 lines
4.4 KiB
C#
using System;
|
|
using System.Net;
|
|
using System.Net.Sockets;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Emby.Server.Implementations.Networking;
|
|
using MediaBrowser.Model.Logging;
|
|
using MediaBrowser.Model.Net;
|
|
|
|
namespace Emby.Server.Implementations.Net
|
|
{
|
|
public class NetAcceptSocket : IAcceptSocket
|
|
{
|
|
public Socket Socket { get; private set; }
|
|
private readonly ILogger _logger;
|
|
|
|
public bool DualMode { get; private set; }
|
|
|
|
public NetAcceptSocket(Socket socket, ILogger logger, bool isDualMode)
|
|
{
|
|
if (socket == null)
|
|
{
|
|
throw new ArgumentNullException("socket");
|
|
}
|
|
if (logger == null)
|
|
{
|
|
throw new ArgumentNullException("logger");
|
|
}
|
|
|
|
Socket = socket;
|
|
_logger = logger;
|
|
DualMode = isDualMode;
|
|
}
|
|
|
|
public IpEndPointInfo LocalEndPoint
|
|
{
|
|
get
|
|
{
|
|
return NetworkManager.ToIpEndPointInfo((IPEndPoint)Socket.LocalEndPoint);
|
|
}
|
|
}
|
|
|
|
public IpEndPointInfo RemoteEndPoint
|
|
{
|
|
get
|
|
{
|
|
return NetworkManager.ToIpEndPointInfo((IPEndPoint)Socket.RemoteEndPoint);
|
|
}
|
|
}
|
|
|
|
public void Connect(IpEndPointInfo endPoint)
|
|
{
|
|
var nativeEndpoint = NetworkManager.ToIPEndPoint(endPoint);
|
|
|
|
Socket.Connect(nativeEndpoint);
|
|
}
|
|
|
|
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)
|
|
{
|
|
var nativeEndpoint = NetworkManager.ToIPEndPoint(endpoint);
|
|
|
|
Socket.Bind(nativeEndpoint);
|
|
}
|
|
|
|
private SocketAcceptor _acceptor;
|
|
public void StartAccept(Action<IAcceptSocket> onAccept, Func<bool> isClosed)
|
|
{
|
|
_acceptor = new SocketAcceptor(_logger, Socket, onAccept, isClosed, DualMode);
|
|
|
|
_acceptor.StartAccept();
|
|
}
|
|
|
|
public Task SendFile(string path, byte[] preBuffer, byte[] postBuffer, CancellationToken cancellationToken)
|
|
{
|
|
var options = TransmitFileOptions.UseDefaultWorkerThread;
|
|
|
|
var completionSource = new TaskCompletionSource<bool>();
|
|
|
|
var result = Socket.BeginSendFile(path, preBuffer, postBuffer, options, new AsyncCallback(FileSendCallback), new Tuple<Socket, string, TaskCompletionSource<bool>>(Socket, path, completionSource));
|
|
|
|
return completionSource.Task;
|
|
}
|
|
|
|
public IAsyncResult BeginSendFile(string path, byte[] preBuffer, byte[] postBuffer, AsyncCallback callback, object state)
|
|
{
|
|
var options = TransmitFileOptions.UseDefaultWorkerThread;
|
|
|
|
return Socket.BeginSendFile(path, preBuffer, postBuffer, options, new AsyncCallback(FileSendCallback), state);
|
|
}
|
|
|
|
public void EndSendFile(IAsyncResult result)
|
|
{
|
|
Socket.EndSendFile(result);
|
|
}
|
|
|
|
private void FileSendCallback(IAsyncResult ar)
|
|
{
|
|
// Retrieve the socket from the state object.
|
|
Tuple<Socket, string, TaskCompletionSource<bool>> data = (Tuple<Socket, string, TaskCompletionSource<bool>>)ar.AsyncState;
|
|
|
|
var client = data.Item1;
|
|
var path = data.Item2;
|
|
var taskCompletion = data.Item3;
|
|
|
|
// Complete sending the data to the remote device.
|
|
try
|
|
{
|
|
client.EndSendFile(ar);
|
|
taskCompletion.TrySetResult(true);
|
|
}
|
|
catch (SocketException ex)
|
|
{
|
|
_logger.Info("Socket.SendFile failed for {0}. error code {1}", path, ex.SocketErrorCode);
|
|
taskCompletion.TrySetException(ex);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
taskCompletion.TrySetException(ex);
|
|
}
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
Socket.Dispose();
|
|
}
|
|
}
|
|
}
|