jellyfin/Emby.Server.Implementations/Connect/ConnectEntryPoint.cs

219 lines
7.1 KiB
C#
Raw Normal View History

2015-06-11 21:22:44 +00:00
using MediaBrowser.Common;
2015-06-04 20:27:46 +00:00
using MediaBrowser.Common.Configuration;
2014-08-25 03:54:45 +00:00
using MediaBrowser.Common.Net;
using MediaBrowser.Controller.Connect;
using MediaBrowser.Controller.Plugins;
using MediaBrowser.Model.Logging;
2015-06-03 03:16:39 +00:00
using MediaBrowser.Model.Net;
2014-08-25 03:54:45 +00:00
using System;
using System.IO;
using System.Text;
2016-01-02 21:54:37 +00:00
using System.Threading.Tasks;
2017-01-01 20:47:54 +00:00
using MediaBrowser.Controller.Security;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Model.IO;
2016-11-03 22:53:02 +00:00
using MediaBrowser.Model.Threading;
2014-08-25 03:54:45 +00:00
2016-11-04 18:56:47 +00:00
namespace Emby.Server.Implementations.Connect
2014-08-25 03:54:45 +00:00
{
public class ConnectEntryPoint : IServerEntryPoint
{
2016-11-03 22:53:02 +00:00
private ITimer _timer;
private IpAddressInfo _cachedIpAddress;
2014-08-25 03:54:45 +00:00
private readonly IHttpClient _httpClient;
private readonly IApplicationPaths _appPaths;
private readonly ILogger _logger;
private readonly IConnectManager _connectManager;
private readonly INetworkManager _networkManager;
2015-06-11 21:22:44 +00:00
private readonly IApplicationHost _appHost;
2015-09-13 23:07:54 +00:00
private readonly IFileSystem _fileSystem;
2016-11-03 22:53:02 +00:00
private readonly ITimerFactory _timerFactory;
2017-01-01 20:47:54 +00:00
private readonly IEncryptionManager _encryption;
2014-08-25 03:54:45 +00:00
2017-01-01 20:47:54 +00:00
public ConnectEntryPoint(IHttpClient httpClient, IApplicationPaths appPaths, ILogger logger, INetworkManager networkManager, IConnectManager connectManager, IApplicationHost appHost, IFileSystem fileSystem, ITimerFactory timerFactory, IEncryptionManager encryption)
2014-08-25 03:54:45 +00:00
{
_httpClient = httpClient;
_appPaths = appPaths;
_logger = logger;
_networkManager = networkManager;
_connectManager = connectManager;
2015-06-11 21:22:44 +00:00
_appHost = appHost;
2015-09-13 23:07:54 +00:00
_fileSystem = fileSystem;
2016-11-03 22:53:02 +00:00
_timerFactory = timerFactory;
2017-01-01 20:47:54 +00:00
_encryption = encryption;
2014-08-25 03:54:45 +00:00
}
public void Run()
{
2016-06-30 02:38:20 +00:00
LoadCachedAddress();
2014-08-25 03:54:45 +00:00
2016-11-03 22:53:02 +00:00
_timer = _timerFactory.Create(TimerCallback, null, TimeSpan.FromSeconds(5), TimeSpan.FromHours(1));
2016-06-30 04:23:52 +00:00
((ConnectManager)_connectManager).Start();
2014-08-25 03:54:45 +00:00
}
2016-03-25 16:23:48 +00:00
private readonly string[] _ipLookups =
{
2016-06-30 04:23:52 +00:00
"http://bot.whatismyipaddress.com",
2016-03-25 16:23:48 +00:00
"https://connect.emby.media/service/ip"
};
2015-06-04 20:27:46 +00:00
2014-08-25 03:54:45 +00:00
private async void TimerCallback(object state)
{
2016-11-04 18:56:47 +00:00
IpAddressInfo validIpAddress = null;
2015-10-07 21:42:29 +00:00
2015-06-04 20:27:46 +00:00
foreach (var ipLookupUrl in _ipLookups)
2014-08-25 03:54:45 +00:00
{
2015-06-04 20:27:46 +00:00
try
2014-08-25 03:54:45 +00:00
{
2016-03-25 16:23:48 +00:00
validIpAddress = await GetIpAddress(ipLookupUrl).ConfigureAwait(false);
2016-03-25 05:39:49 +00:00
2016-03-25 16:23:48 +00:00
// Try to find the ipv4 address, if present
2016-11-08 18:44:23 +00:00
if (validIpAddress.AddressFamily != IpAddressFamily.InterNetworkV6)
2015-06-04 20:27:46 +00:00
{
2016-03-25 16:23:48 +00:00
break;
2014-08-25 03:54:45 +00:00
}
}
2015-06-04 20:27:46 +00:00
catch (HttpException)
{
}
catch (Exception ex)
{
_logger.ErrorException("Error getting connection info", ex);
}
2016-03-25 16:23:48 +00:00
}
2016-11-08 18:44:23 +00:00
2016-03-25 16:23:48 +00:00
// If this produced an ipv6 address, try again
2016-11-08 18:44:23 +00:00
if (validIpAddress != null && validIpAddress.AddressFamily == IpAddressFamily.InterNetworkV6)
2016-03-25 16:23:48 +00:00
{
foreach (var ipLookupUrl in _ipLookups)
{
try
{
2016-06-30 04:23:52 +00:00
var newAddress = await GetIpAddress(ipLookupUrl, true).ConfigureAwait(false);
2016-03-25 16:23:48 +00:00
// Try to find the ipv4 address, if present
2016-11-08 18:44:23 +00:00
if (newAddress.AddressFamily != IpAddressFamily.InterNetworkV6)
2016-03-25 16:23:48 +00:00
{
2016-06-30 04:23:52 +00:00
validIpAddress = newAddress;
2016-03-25 16:23:48 +00:00
break;
}
}
catch (HttpException)
{
}
catch (Exception ex)
{
_logger.ErrorException("Error getting connection info", ex);
}
}
}
2015-10-07 21:42:29 +00:00
2016-03-25 16:23:48 +00:00
if (validIpAddress != null)
{
((ConnectManager)_connectManager).OnWanAddressResolved(validIpAddress);
CacheAddress(validIpAddress);
}
}
2016-11-04 18:56:47 +00:00
private async Task<IpAddressInfo> GetIpAddress(string lookupUrl, bool preferIpv4 = false)
2016-03-25 16:23:48 +00:00
{
// Sometimes whatismyipaddress might fail, but it won't do us any good having users raise alarms over it.
var logErrors = false;
#if DEBUG
logErrors = true;
#endif
using (var stream = await _httpClient.Get(new HttpRequestOptions
{
Url = lookupUrl,
UserAgent = "Emby/" + _appHost.ApplicationVersion,
LogErrors = logErrors,
// Seeing block length errors with our server
EnableHttpCompression = false,
2016-10-06 18:55:01 +00:00
PreferIpv4 = preferIpv4,
BufferContent = false
2016-03-25 16:23:48 +00:00
}).ConfigureAwait(false))
{
using (var reader = new StreamReader(stream))
{
var addressString = await reader.ReadToEndAsync().ConfigureAwait(false);
2016-11-04 18:56:47 +00:00
return _networkManager.ParseIpAddress(addressString);
2016-03-25 16:23:48 +00:00
}
2015-06-03 03:16:39 +00:00
}
2014-08-25 03:54:45 +00:00
}
private string CacheFilePath
{
2017-01-01 20:47:54 +00:00
get { return Path.Combine(_appPaths.DataPath, "wan.dat"); }
2014-08-25 03:54:45 +00:00
}
2016-11-04 18:56:47 +00:00
private void CacheAddress(IpAddressInfo address)
2014-08-25 03:54:45 +00:00
{
if (_cachedIpAddress != null && _cachedIpAddress.Equals(address))
{
// no need to update the file if the address has not changed
return;
}
2014-08-25 03:54:45 +00:00
var path = CacheFilePath;
try
{
2015-10-07 21:42:29 +00:00
_fileSystem.CreateDirectory(Path.GetDirectoryName(path));
2017-01-01 20:47:54 +00:00
}
catch (Exception ex)
{
}
try
{
_fileSystem.WriteAllText(path, _encryption.EncryptString(address.ToString()), Encoding.UTF8);
_cachedIpAddress = address;
2014-08-25 03:54:45 +00:00
}
catch (Exception ex)
{
_logger.ErrorException("Error saving data", ex);
}
}
private void LoadCachedAddress()
{
var path = CacheFilePath;
_logger.Info("Loading data from {0}", path);
2014-08-25 03:54:45 +00:00
try
{
2017-01-01 20:47:54 +00:00
var endpoint = _encryption.DecryptString(_fileSystem.ReadAllText(path, Encoding.UTF8));
2016-11-04 18:56:47 +00:00
IpAddressInfo ipAddress;
2014-08-25 03:54:45 +00:00
2016-11-04 18:56:47 +00:00
if (_networkManager.TryParseIpAddress(endpoint, out ipAddress))
2014-08-25 03:54:45 +00:00
{
_cachedIpAddress = ipAddress;
2016-03-25 16:23:48 +00:00
((ConnectManager)_connectManager).OnWanAddressResolved(ipAddress);
2014-08-25 03:54:45 +00:00
}
}
catch (IOException)
{
// File isn't there. no biggie
}
catch (Exception ex)
{
_logger.ErrorException("Error loading data", ex);
}
}
public void Dispose()
{
if (_timer != null)
{
_timer.Dispose();
_timer = null;
}
}
}
}