This commit is contained in:
Connection Refused
2019-12-02 19:51:12 +08:00
commit b2ea730984
229 changed files with 86605 additions and 0 deletions

View File

@@ -0,0 +1,97 @@
using DnsClient.Protocol;
namespace DnsClient
{
internal class DnsRequestHeader
{
public const int HeaderLength = 12;
private ushort _flags = 0;
public ushort RawFlags => _flags;
internal DnsHeaderFlag HeaderFlags
{
get
{
return (DnsHeaderFlag)_flags;
}
set
{
_flags &= (ushort)~(DnsHeaderFlag.IsCheckingDisabled);
_flags &= (ushort)~(DnsHeaderFlag.IsAuthenticData);
_flags &= (ushort)~(DnsHeaderFlag.FutureUse);
_flags &= (ushort)~(DnsHeaderFlag.HasQuery);
_flags &= (ushort)~(DnsHeaderFlag.HasAuthorityAnswer);
_flags &= (ushort)~(DnsHeaderFlag.ResultTruncated);
_flags &= (ushort)~(DnsHeaderFlag.RecursionDesired);
_flags &= (ushort)~(DnsHeaderFlag.RecursionAvailable);
_flags |= (ushort)value;
}
}
public int Id { get; set; }
public DnsOpCode OpCode
{
get
{
return (DnsOpCode)((DnsHeader.OPCodeMask & _flags) >> DnsHeader.OPCodeShift);
}
set
{
_flags &= (ushort)~(DnsHeader.OPCodeMask);
_flags |= (ushort)(((ushort)value << DnsHeader.OPCodeShift) & DnsHeader.OPCodeMask);
}
}
public ushort RCode
{
get
{
return (ushort)(DnsHeader.RCodeMask & _flags);
}
set
{
_flags &= (ushort)~(DnsHeader.RCodeMask);
_flags |= (ushort)(value & DnsHeader.RCodeMask);
}
}
public bool UseRecursion
{
get
{
return HeaderFlags.HasFlag(DnsHeaderFlag.RecursionDesired);
}
set
{
if (value)
{
_flags |= (ushort)DnsHeaderFlag.RecursionDesired;
}
else
{
_flags &= (ushort)~(DnsHeaderFlag.RecursionDesired);
}
}
}
public DnsRequestHeader(int id, DnsOpCode queryKind)
: this(id, true, queryKind)
{
}
public DnsRequestHeader(int id, bool useRecursion, DnsOpCode queryKind)
{
Id = id;
OpCode = queryKind;
UseRecursion = useRecursion;
}
public override string ToString()
{
return $"{Id} - Qs: {1} Recursion: {UseRecursion} OpCode: {OpCode}";
}
}
}