96 lines
3.4 KiB
C#
96 lines
3.4 KiB
C#
using DNS.Client;
|
||
using Microsoft.Extensions.Caching.Memory;
|
||
using Microsoft.Extensions.Logging;
|
||
using Microsoft.Extensions.Options;
|
||
using System;
|
||
using System.Linq;
|
||
using System.Net;
|
||
using System.Threading;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace FastGithub.ReverseProxy
|
||
{
|
||
/// <summary>
|
||
/// 域名解析器
|
||
/// </summary>
|
||
sealed class DomainResolver
|
||
{
|
||
private DnsClient dnsClient;
|
||
private readonly IMemoryCache memoryCache;
|
||
private readonly IOptionsMonitor<FastGithubOptions> options;
|
||
private readonly ILogger<DomainResolver> logger;
|
||
private readonly TimeSpan cacheTimeSpan = TimeSpan.FromSeconds(10d);
|
||
|
||
/// <summary>
|
||
/// 域名解析器
|
||
/// </summary>
|
||
/// <param name="memoryCache"></param>
|
||
/// <param name="fastGithubConfig"></param>
|
||
public DomainResolver(
|
||
IMemoryCache memoryCache,
|
||
IOptionsMonitor<FastGithubOptions> options,
|
||
ILogger<DomainResolver> logger)
|
||
{
|
||
this.memoryCache = memoryCache;
|
||
this.options = options;
|
||
this.logger = logger;
|
||
|
||
this.dnsClient = new DnsClient(options.CurrentValue.PureDns.ToIPEndPoint());
|
||
options.OnChange(opt => this.dnsClient = new DnsClient(opt.PureDns.ToIPEndPoint()));
|
||
}
|
||
|
||
/// <summary>
|
||
/// 解析指定的域名
|
||
/// </summary>
|
||
/// <param name="domain"></param>
|
||
/// <returns></returns>
|
||
/// <exception cref="FastGithubException"></exception>
|
||
public async Task<IPAddress> ResolveAsync(string domain, CancellationToken cancellationToken)
|
||
{
|
||
// 缓存以避免做不必要的并发查询
|
||
var key = $"domain:{domain}";
|
||
var address = await this.memoryCache.GetOrCreateAsync(key, e =>
|
||
{
|
||
e.SetAbsoluteExpiration(this.cacheTimeSpan);
|
||
return this.LookupAsync(domain, cancellationToken);
|
||
});
|
||
return address;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 查找ip
|
||
/// </summary>
|
||
/// <param name="domain"></param>
|
||
/// <param name="cancellationToken"></param>
|
||
/// <returns></returns>
|
||
/// <exception cref="FastGithubException">
|
||
private async Task<IPAddress> LookupAsync(string domain, CancellationToken cancellationToken)
|
||
{
|
||
try
|
||
{
|
||
var addresses = await dnsClient.Lookup(domain, DNS.Protocol.RecordType.A, cancellationToken);
|
||
var address = addresses?.FirstOrDefault();
|
||
if (address == null)
|
||
{
|
||
throw new Exception($"解析不到{domain}的ip");
|
||
}
|
||
|
||
// 受干扰的dns，常常返回127.0.0.1来阻断请求
|
||
// 如果解析到的ip为本机ip，会产生反向代理请求死循环
|
||
if (address.Equals(IPAddress.Loopback))
|
||
{
|
||
throw new Exception($"dns被污染，解析{domain}为{address}");
|
||
}
|
||
|
||
this.logger.LogInformation($"[{domain}->{address}]");
|
||
return address;
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
var dns = this.options.CurrentValue.PureDns;
|
||
throw new FastGithubException($"dns({dns})服务器异常：{ex.Message}", ex);
|
||
}
|
||
}
|
||
}
|
||
}
|