mirror of
https://github.com/netchx/netch.git
synced 2026-04-05 19:45:05 +08:00
done
This commit is contained in:
Connection Refused
256
Netch/3rd/DnsClient.NET/DnsRecordFactory.cs
Normal file
256
Netch/3rd/DnsClient.NET/DnsRecordFactory.cs
Normal file
@@ -0,0 +1,256 @@
|
||||
using DnsClient.Protocol;
|
||||
using DnsClient.Protocol.Options;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
|
||||
namespace DnsClient
|
||||
{
|
||||
internal class DnsRecordFactory
|
||||
{
|
||||
private readonly DnsDatagramReader _reader;
|
||||
|
||||
public DnsRecordFactory(DnsDatagramReader reader)
|
||||
{
|
||||
_reader = reader ?? throw new ArgumentNullException(nameof(reader));
|
||||
}
|
||||
|
||||
/*
|
||||
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5
|
||||
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
|
||||
| |
|
||||
/ /
|
||||
/ NAME /
|
||||
| |
|
||||
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
|
||||
| TYPE |
|
||||
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
|
||||
| CLASS |
|
||||
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
|
||||
| TTL |
|
||||
| |
|
||||
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
|
||||
| RDLENGTH |
|
||||
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--|
|
||||
/ RDATA /
|
||||
/ /
|
||||
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
|
||||
* */
|
||||
|
||||
public ResourceRecordInfo ReadRecordInfo()
|
||||
{
|
||||
return new ResourceRecordInfo(
|
||||
_reader.ReadQuestionQueryString(), // name
|
||||
(ResourceRecordType)_reader.ReadUInt16NetworkOrder(), // type
|
||||
(QueryClass)_reader.ReadUInt16NetworkOrder(), // class
|
||||
(int)_reader.ReadUInt32NetworkOrder(), // ttl - 32bit!!
|
||||
_reader.ReadUInt16NetworkOrder()); // RDLength
|
||||
}
|
||||
|
||||
public DnsResourceRecord GetRecord(ResourceRecordInfo info)
|
||||
{
|
||||
if (info == null)
|
||||
{
|
||||
throw new ArgumentNullException(nameof(info));
|
||||
}
|
||||
|
||||
var oldIndex = _reader.Index;
|
||||
DnsResourceRecord result;
|
||||
|
||||
switch (info.RecordType)
|
||||
{
|
||||
case ResourceRecordType.A:
|
||||
result = new ARecord(info, _reader.ReadIPAddress());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.NS:
|
||||
result = new NsRecord(info, _reader.ReadDnsName());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.CNAME:
|
||||
result = new CNameRecord(info, _reader.ReadDnsName());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.SOA:
|
||||
result = ResolveSoaRecord(info);
|
||||
break;
|
||||
|
||||
case ResourceRecordType.MB:
|
||||
result = new MbRecord(info, _reader.ReadDnsName());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.MG:
|
||||
result = new MgRecord(info, _reader.ReadDnsName());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.MR:
|
||||
result = new MrRecord(info, _reader.ReadDnsName());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.NULL:
|
||||
result = new NullRecord(info, _reader.ReadBytes(info.RawDataLength).ToArray());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.WKS:
|
||||
result = ResolveWksRecord(info);
|
||||
break;
|
||||
|
||||
case ResourceRecordType.PTR:
|
||||
result = new PtrRecord(info, _reader.ReadDnsName());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.HINFO:
|
||||
result = new HInfoRecord(info, _reader.ReadString(), _reader.ReadString());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.MINFO:
|
||||
result = new MInfoRecord(info, _reader.ReadDnsName(), _reader.ReadDnsName());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.MX:
|
||||
result = ResolveMXRecord(info);
|
||||
break;
|
||||
|
||||
case ResourceRecordType.TXT:
|
||||
result = ResolveTXTRecord(info);
|
||||
break;
|
||||
|
||||
case ResourceRecordType.RP:
|
||||
result = new RpRecord(info, _reader.ReadDnsName(), _reader.ReadDnsName());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.AFSDB:
|
||||
result = new AfsDbRecord(info, (AfsType)_reader.ReadUInt16NetworkOrder(), _reader.ReadDnsName());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.AAAA:
|
||||
result = new AaaaRecord(info, _reader.ReadIPv6Address());
|
||||
break;
|
||||
|
||||
case ResourceRecordType.SRV:
|
||||
result = ResolveSrvRecord(info);
|
||||
break;
|
||||
|
||||
case ResourceRecordType.OPT:
|
||||
result = ResolveOptRecord(info);
|
||||
break;
|
||||
|
||||
case ResourceRecordType.URI:
|
||||
result = ResolveUriRecord(info);
|
||||
break;
|
||||
|
||||
case ResourceRecordType.CAA:
|
||||
result = ResolveCaaRecord(info);
|
||||
break;
|
||||
|
||||
case ResourceRecordType.SSHFP:
|
||||
result = ResolveSshfpRecord(info);
|
||||
break;
|
||||
|
||||
default:
|
||||
// update reader index because we don't read full data for the empty record
|
||||
_reader.Advance(info.RawDataLength);
|
||||
result = new EmptyRecord(info);
|
||||
break;
|
||||
}
|
||||
|
||||
// sanity check
|
||||
if (_reader.Index != oldIndex + info.RawDataLength)
|
||||
{
|
||||
throw new InvalidOperationException("Record reader index out of sync.");
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
private DnsResourceRecord ResolveUriRecord(ResourceRecordInfo info)
|
||||
{
|
||||
var prio = _reader.ReadUInt16NetworkOrder();
|
||||
var weight = _reader.ReadUInt16NetworkOrder();
|
||||
var target = _reader.ReadString(info.RawDataLength - 4);
|
||||
return new UriRecord(info, prio, weight, target);
|
||||
}
|
||||
|
||||
private DnsResourceRecord ResolveOptRecord(ResourceRecordInfo info)
|
||||
{
|
||||
return new OptRecord((int)info.RecordClass, info.InitialTimeToLive, info.RawDataLength);
|
||||
}
|
||||
|
||||
private DnsResourceRecord ResolveWksRecord(ResourceRecordInfo info)
|
||||
{
|
||||
var address = _reader.ReadIPAddress();
|
||||
var protocol = _reader.ReadByte();
|
||||
var bitmap = _reader.ReadBytes(info.RawDataLength - 5);
|
||||
|
||||
return new WksRecord(info, address, protocol, bitmap.ToArray());
|
||||
}
|
||||
|
||||
private DnsResourceRecord ResolveMXRecord(ResourceRecordInfo info)
|
||||
{
|
||||
var preference = _reader.ReadUInt16NetworkOrder();
|
||||
var domain = _reader.ReadDnsName();
|
||||
|
||||
return new MxRecord(info, preference, domain);
|
||||
}
|
||||
|
||||
private DnsResourceRecord ResolveSoaRecord(ResourceRecordInfo info)
|
||||
{
|
||||
var mName = _reader.ReadDnsName();
|
||||
var rName = _reader.ReadDnsName();
|
||||
var serial = _reader.ReadUInt32NetworkOrder();
|
||||
var refresh = _reader.ReadUInt32NetworkOrder();
|
||||
var retry = _reader.ReadUInt32NetworkOrder();
|
||||
var expire = _reader.ReadUInt32NetworkOrder();
|
||||
var minimum = _reader.ReadUInt32NetworkOrder();
|
||||
|
||||
return new SoaRecord(info, mName, rName, serial, refresh, retry, expire, minimum);
|
||||
}
|
||||
|
||||
private DnsResourceRecord ResolveSrvRecord(ResourceRecordInfo info)
|
||||
{
|
||||
var priority = _reader.ReadUInt16NetworkOrder();
|
||||
var weight = _reader.ReadUInt16NetworkOrder();
|
||||
var port = _reader.ReadUInt16NetworkOrder();
|
||||
var target = _reader.ReadDnsName();
|
||||
|
||||
return new SrvRecord(info, priority, weight, port, target);
|
||||
}
|
||||
|
||||
private DnsResourceRecord ResolveTXTRecord(ResourceRecordInfo info)
|
||||
{
|
||||
int pos = _reader.Index;
|
||||
|
||||
var values = new List<string>();
|
||||
var utf8Values = new List<string>();
|
||||
while ((_reader.Index - pos) < info.RawDataLength)
|
||||
{
|
||||
var length = _reader.ReadByte();
|
||||
var bytes = _reader.ReadBytes(length);
|
||||
var array = new ArraySegment<byte>(bytes.Array, bytes.Offset, length);
|
||||
var escaped = DnsDatagramReader.ParseString(bytes);
|
||||
var utf = DnsDatagramReader.ReadUTF8String(bytes);
|
||||
values.Add(escaped);
|
||||
utf8Values.Add(utf);
|
||||
}
|
||||
|
||||
return new TxtRecord(info, values.ToArray(), utf8Values.ToArray());
|
||||
}
|
||||
|
||||
private DnsResourceRecord ResolveSshfpRecord(ResourceRecordInfo info)
|
||||
{
|
||||
var algorithm = (SshfpAlgorithm)_reader.ReadByte();
|
||||
var fingerprintType = (SshfpFingerprintType)_reader.ReadByte();
|
||||
var fingerprint = _reader.ReadBytes(info.RawDataLength - 2).ToArray();
|
||||
var fingerprintHexString = string.Join(string.Empty, fingerprint.Select(b => b.ToString("X2")));
|
||||
return new SshfpRecord(info, algorithm, fingerprintType, fingerprintHexString);
|
||||
}
|
||||
|
||||
private DnsResourceRecord ResolveCaaRecord(ResourceRecordInfo info)
|
||||
{
|
||||
var flag = _reader.ReadByte();
|
||||
var tag = _reader.ReadString();
|
||||
var stringValue = DnsDatagramReader.ParseString(_reader, info.RawDataLength - 2 - tag.Length);
|
||||
return new CaaRecord(info, flag, tag, stringValue);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user