private static void CreateRandomUdpPayload(Random random, UdpLayer udpLayer, List<ILayer> layers) { if (random.NextBool(20)) { // Finish with payload. PayloadLayer payloadLayer = random.NextPayloadLayer(random.Next(100)); layers.Add(payloadLayer); return; } DnsLayer dnsLayer = random.NextDnsLayer(); layers.Add(dnsLayer); ushort specialPort = (ushort)(random.NextBool() ? 53 : 5355); if (dnsLayer.IsQuery) udpLayer.DestinationPort = specialPort; else udpLayer.SourcePort = specialPort; }
public void RandomDnsTest() { EthernetLayer ethernetLayer = new EthernetLayer { Source = new MacAddress("00:01:02:03:04:05"), Destination = new MacAddress("A0:A1:A2:A3:A4:A5") }; int seed = new Random().Next(); Console.WriteLine("Seed: " + seed); Random random = new Random(seed); for (int i = 0; i != 1000; ++i) { IpV4Layer ipV4Layer = random.NextIpV4Layer(null); ipV4Layer.HeaderChecksum = null; Layer ipLayer = random.NextBool() ? (Layer)ipV4Layer : random.NextIpV6Layer(true); UdpLayer udpLayer = random.NextUdpLayer(); udpLayer.Checksum = null; DnsLayer dnsLayer; do { dnsLayer = random.NextDnsLayer(); } while (dnsLayer.Length > 65000 - ipLayer.Length); Packet packet = PacketBuilder.Build(DateTime.Now, ethernetLayer, ipLayer, udpLayer, dnsLayer); Assert.IsTrue(packet.IsValid, "IsValid"); // DNS DnsLayer actualLayer = (DnsLayer)packet.Ethernet.Ip.Udp.Dns.ExtractLayer(); Assert.AreEqual(dnsLayer, actualLayer, "DNS Layer"); Assert.IsTrue(packet.Ethernet.Ip.Udp.Dns.IsValid); DnsDataResourceRecord opt = packet.Ethernet.Ip.Udp.Dns.Additionals.FirstOrDefault(additional => additional.DnsType == DnsType.Opt); Assert.AreEqual(opt, packet.Ethernet.Ip.Udp.Dns.OptionsRecord); foreach (var record in packet.Ethernet.Ip.Udp.Dns.ResourceRecords) { Assert.IsTrue(record.Equals(record)); Assert.IsTrue(record.DomainName.Equals((object)record.DomainName)); Assert.IsTrue(record.DomainName.Equals((object)record.DomainName)); Assert.AreEqual(record.GetHashCode(), record.GetHashCode()); } foreach (var record in packet.Ethernet.Ip.Udp.Dns.DataResourceRecords) { MoreAssert.IsBiggerOrEqual(9, record.ToString().Length); Assert.IsTrue(record.Equals((object)record)); Assert.IsInstanceOfType(record.Data, DnsResourceData.GetDnsResourceDataType(record.DnsType) ?? typeof(DnsResourceDataAnything)); Assert.IsTrue(record.DomainName.Equals((object)record.DomainName)); Assert.IsFalse(record.Data.Equals(null)); } } }