ouczbs/FastGithub
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

减少CalcChecksums

Browse Source
This commit is contained in:
陈国伟 2021-09-17 17:49:33 +08:00
parent 592ed21e83
commit 3cdefb8553
1 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
FastGithub.Dns/DnsInterceptor.cs
View File

@ -73,26 +73,31 @@ namespace FastGithub.Dns
{ {
try try
{ {
this.ProcessDnsPacket(packetBuffer, ref winDivertAddress, ref packetLength); if (this.ModifyDnsPacket(packetBuffer, ref winDivertAddress, ref packetLength))
{
WinDivert.WinDivertHelperCalcChecksums(winDivertBuffer, packetLength, ref winDivertAddress, WinDivertChecksumHelperParam.All);
}
} }
catch (Exception ex) catch (Exception ex)
{ {
this.logger.LogWarning(ex.Message); this.logger.LogWarning(ex.Message);
} }
finally
WinDivert.WinDivertHelperCalcChecksums(winDivertBuffer, packetLength, ref winDivertAddress, WinDivertChecksumHelperParam.All); {
WinDivert.WinDivertSend(handle, winDivertBuffer, packetLength, ref winDivertAddress); WinDivert.WinDivertSend(handle, winDivertBuffer, packetLength, ref winDivertAddress);
}
} }
} }
} }
/// <summary> /// <summary>
/// 处理DNS数据包 /// 修改DNS数据包
/// </summary> /// </summary>
/// <param name="packetBuffer"></param> /// <param name="packetBuffer"></param>
/// <param name="winDivertAddress"></param> /// <param name="winDivertAddress"></param>
/// <param name="packetLength"></param> /// <param name="packetLength"></param>
private void ProcessDnsPacket(byte[] packetBuffer, ref WinDivertAddress winDivertAddress, ref uint packetLength) /// <returns></returns>
private bool ModifyDnsPacket(byte[] packetBuffer, ref WinDivertAddress winDivertAddress, ref uint packetLength)
{ {
var packetData = packetBuffer.AsSpan(0, (int)packetLength).ToArray(); var packetData = packetBuffer.AsSpan(0, (int)packetLength).ToArray();
var packet = Packet.ParsePacket(LinkLayers.Raw, packetData); var packet = Packet.ParsePacket(LinkLayers.Raw, packetData);
@ -102,19 +107,19 @@ namespace FastGithub.Dns
var request = Request.FromArray(udpPacket.PayloadData); var request = Request.FromArray(udpPacket.PayloadData);
if (request.OperationCode != OperationCode.Query) if (request.OperationCode != OperationCode.Query)
{ {
return; return false;
} }
var question = request.Questions.FirstOrDefault(); var question = request.Questions.FirstOrDefault();
if (question == null || question.Type != RecordType.A) if (question == null || question.Type != RecordType.A)
{ {
return; return false;
} }
var domain = question.Name; var domain = question.Name;
if (this.fastGithubConfig.IsMatch(domain.ToString()) == false) if (this.fastGithubConfig.IsMatch(domain.ToString()) == false)
{ {
return; return false;
} }
// 反转ip // 反转ip
@ -148,6 +153,7 @@ namespace FastGithub.Dns
} }
this.logger.LogInformation($"已拦截dns查询{domain}并伪造响应内容为{IPAddress.Loopback}"); this.logger.LogInformation($"已拦截dns查询{domain}并伪造响应内容为{IPAddress.Loopback}");
return true;
} }
} }
} }