jellyfin/Emby.Server.Implementations/Net/SocketFactory.cs

259 lines
9.6 KiB
C#
Raw Normal View History

using System;
2017-05-25 13:00:14 +00:00
using System.IO;
2016-11-04 08:31:05 +00:00
using System.Net;
using System.Net.Sockets;
using Emby.Server.Implementations.Networking;
2016-11-04 08:31:05 +00:00
using MediaBrowser.Model.Net;
using Microsoft.Extensions.Logging;
2016-11-04 08:31:05 +00:00
namespace Emby.Server.Implementations.Net
2016-11-04 08:31:05 +00:00
{
public class SocketFactory : ISocketFactory
{
2019-01-07 23:24:34 +00:00
// THIS IS A LINKED FILE - SHARED AMONGST MULTIPLE PLATFORMS
2016-11-04 08:31:05 +00:00
// Be careful to check any changes compile and work for all platform projects it is shared in.
// Not entirely happy with this. Would have liked to have done something more generic/reusable,
2019-01-07 23:24:34 +00:00
// but that wasn't really the point so kept to YAGNI principal for now, even if the
2016-11-04 08:31:05 +00:00
// interfaces are a bit ugly, specific and make assumptions.
2017-03-02 20:50:09 +00:00
public ISocket CreateTcpSocket(IpAddressInfo remoteAddress, int remotePort)
{
if (remotePort < 0) throw new ArgumentException("remotePort cannot be less than zero.", nameof(remotePort));
2017-03-02 20:50:09 +00:00
2017-03-29 06:53:26 +00:00
var addressFamily = remoteAddress.AddressFamily == IpAddressFamily.InterNetwork
2017-08-17 20:19:02 +00:00
? AddressFamily.InterNetwork
: AddressFamily.InterNetworkV6;
2017-03-29 06:53:26 +00:00
var retVal = new Socket(addressFamily, System.Net.Sockets.SocketType.Stream, System.Net.Sockets.ProtocolType.Tcp);
2017-03-02 20:50:09 +00:00
try
{
retVal.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
2017-03-29 06:53:26 +00:00
}
catch (SocketException)
{
// This is not supported on all operating systems (qnap)
}
2017-08-17 20:19:02 +00:00
2017-03-29 06:53:26 +00:00
try
{
2017-03-02 20:50:09 +00:00
return new UdpSocket(retVal, new IpEndPointInfo(remoteAddress, remotePort));
}
catch
{
if (retVal != null)
retVal.Dispose();
throw;
}
}
2016-11-04 08:31:05 +00:00
2016-11-04 18:56:47 +00:00
/// <summary>
2017-03-02 20:50:09 +00:00
/// Creates a new UDP acceptSocket and binds it to the specified local port.
2016-11-04 18:56:47 +00:00
/// </summary>
2017-03-02 20:50:09 +00:00
/// <param name="localPort">An integer specifying the local port to bind the acceptSocket to.</param>
public ISocket CreateUdpSocket(int localPort)
2016-11-04 18:56:47 +00:00
{
if (localPort < 0) throw new ArgumentException("localPort cannot be less than zero.", nameof(localPort));
2016-11-04 18:56:47 +00:00
2016-11-08 18:44:23 +00:00
var retVal = new Socket(AddressFamily.InterNetwork, System.Net.Sockets.SocketType.Dgram, System.Net.Sockets.ProtocolType.Udp);
2016-11-04 18:56:47 +00:00
try
{
retVal.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
2016-12-04 21:30:38 +00:00
return new UdpSocket(retVal, localPort, IPAddress.Any);
2016-11-04 18:56:47 +00:00
}
catch
{
if (retVal != null)
retVal.Dispose();
throw;
}
}
2017-03-13 04:08:23 +00:00
public ISocket CreateUdpBroadcastSocket(int localPort)
{
if (localPort < 0) throw new ArgumentException("localPort cannot be less than zero.", nameof(localPort));
2017-03-13 04:08:23 +00:00
var retVal = new Socket(AddressFamily.InterNetwork, System.Net.Sockets.SocketType.Dgram, System.Net.Sockets.ProtocolType.Udp);
try
{
retVal.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
retVal.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.Broadcast, 1);
return new UdpSocket(retVal, localPort, IPAddress.Any);
}
catch
{
if (retVal != null)
retVal.Dispose();
throw;
}
}
2017-08-17 20:19:02 +00:00
2016-11-04 08:31:05 +00:00
/// <summary>
2017-08-17 20:19:02 +00:00
/// Creates a new UDP acceptSocket that is a member of the SSDP multicast local admin group and binds it to the specified local port.
/// </summary>
/// <returns>An implementation of the <see cref="ISocket"/> interface used by RSSDP components to perform acceptSocket operations.</returns>
2017-03-02 20:50:09 +00:00
public ISocket CreateSsdpUdpSocket(IpAddressInfo localIpAddress, int localPort)
2016-11-04 08:31:05 +00:00
{
if (localPort < 0) throw new ArgumentException("localPort cannot be less than zero.", nameof(localPort));
2016-11-04 08:31:05 +00:00
2016-11-08 18:44:23 +00:00
var retVal = new Socket(AddressFamily.InterNetwork, System.Net.Sockets.SocketType.Dgram, System.Net.Sockets.ProtocolType.Udp);
2016-11-04 08:31:05 +00:00
try
{
retVal.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
retVal.SetSocketOption(SocketOptionLevel.IP, SocketOptionName.MulticastTimeToLive, 4);
2016-12-04 21:30:38 +00:00
var localIp = NetworkManager.ToIPAddress(localIpAddress);
retVal.SetSocketOption(SocketOptionLevel.IP, SocketOptionName.AddMembership, new MulticastOption(IPAddress.Parse("239.255.255.250"), localIp));
return new UdpSocket(retVal, localPort, localIp);
2016-11-04 08:31:05 +00:00
}
catch
{
if (retVal != null)
retVal.Dispose();
throw;
}
}
/// <summary>
2017-03-02 20:50:09 +00:00
/// Creates a new UDP acceptSocket that is a member of the specified multicast IP address, and binds it to the specified local port.
2016-11-04 08:31:05 +00:00
/// </summary>
2017-03-02 20:50:09 +00:00
/// <param name="ipAddress">The multicast IP address to make the acceptSocket a member of.</param>
/// <param name="multicastTimeToLive">The multicast time to live value for the acceptSocket.</param>
2016-11-04 08:31:05 +00:00
/// <param name="localPort">The number of the local port to bind to.</param>
/// <returns></returns>
2017-03-02 20:50:09 +00:00
public ISocket CreateUdpMulticastSocket(string ipAddress, int multicastTimeToLive, int localPort)
2016-11-04 08:31:05 +00:00
{
if (ipAddress == null) throw new ArgumentNullException(nameof(ipAddress));
if (ipAddress.Length == 0) throw new ArgumentException("ipAddress cannot be an empty string.", nameof(ipAddress));
if (multicastTimeToLive <= 0) throw new ArgumentException("multicastTimeToLive cannot be zero or less.", nameof(multicastTimeToLive));
if (localPort < 0) throw new ArgumentException("localPort cannot be less than zero.", nameof(localPort));
2016-11-04 08:31:05 +00:00
2016-11-08 18:44:23 +00:00
var retVal = new Socket(AddressFamily.InterNetwork, System.Net.Sockets.SocketType.Dgram, System.Net.Sockets.ProtocolType.Udp);
2016-11-04 08:31:05 +00:00
try
{
2017-08-17 20:19:02 +00:00
// not supported on all platforms. throws on ubuntu with .net core 2.0
2017-05-24 19:12:55 +00:00
retVal.ExclusiveAddressUse = false;
2017-08-17 20:19:02 +00:00
}
catch (SocketException)
{
}
try
{
2017-11-10 21:22:38 +00:00
// seeing occasional exceptions thrown on qnap
// System.Net.Sockets.SocketException (0x80004005): Protocol not available
2016-11-04 08:31:05 +00:00
retVal.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.ReuseAddress, true);
2017-11-10 21:22:38 +00:00
}
catch (SocketException)
{
}
try
{
//retVal.SetSocketOption(SocketOptionLevel.Socket, SocketOptionName.Broadcast, true);
2016-11-04 08:31:05 +00:00
retVal.SetSocketOption(SocketOptionLevel.IP, SocketOptionName.MulticastTimeToLive, multicastTimeToLive);
2016-12-04 21:30:38 +00:00
var localIp = IPAddress.Any;
retVal.SetSocketOption(SocketOptionLevel.IP, SocketOptionName.AddMembership, new MulticastOption(IPAddress.Parse(ipAddress), localIp));
2016-11-04 08:31:05 +00:00
retVal.MulticastLoopback = true;
2016-12-04 21:30:38 +00:00
return new UdpSocket(retVal, localPort, localIp);
2016-11-04 08:31:05 +00:00
}
catch
{
if (retVal != null)
retVal.Dispose();
throw;
}
}
2017-05-25 13:00:14 +00:00
public Stream CreateNetworkStream(ISocket socket, bool ownsSocket)
{
var netSocket = (UdpSocket)socket;
return new SocketStream(netSocket.Socket, ownsSocket);
}
2016-11-04 08:31:05 +00:00
}
2017-05-25 13:00:14 +00:00
public class SocketStream : Stream
{
private readonly Socket _socket;
public SocketStream(Socket socket, bool ownsSocket)
{
_socket = socket;
}
public override void Flush()
{
}
public override bool CanRead => true;
public override bool CanSeek => false;
public override bool CanWrite => true;
public override long Length => throw new NotImplementedException();
2017-05-25 13:00:14 +00:00
public override long Position
{
get => throw new NotImplementedException();
set => throw new NotImplementedException();
2017-05-25 13:00:14 +00:00
}
public override void Write(byte[] buffer, int offset, int count)
{
_socket.Send(buffer, offset, count, SocketFlags.None);
}
public override IAsyncResult BeginWrite(byte[] buffer, int offset, int count, AsyncCallback callback, object state)
{
return _socket.BeginSend(buffer, offset, count, SocketFlags.None, callback, state);
}
public override void EndWrite(IAsyncResult asyncResult)
{
_socket.EndSend(asyncResult);
}
public override void SetLength(long value)
{
throw new NotImplementedException();
}
public override long Seek(long offset, SeekOrigin origin)
{
throw new NotImplementedException();
}
public override int Read(byte[] buffer, int offset, int count)
{
return _socket.Receive(buffer, offset, count, SocketFlags.None);
}
public override IAsyncResult BeginRead(byte[] buffer, int offset, int count, AsyncCallback callback, object state)
{
return _socket.BeginReceive(buffer, offset, count, SocketFlags.None, callback, state);
}
public override int EndRead(IAsyncResult asyncResult)
{
return _socket.EndReceive(asyncResult);
}
}
2016-11-04 08:31:05 +00:00
}