/src/PcapPlusPlus/Tests/Fuzzers/ReadParsedPacket.h

Source
#pragma once

#include <Packet.h>
#include <IPv4Layer.h>
#include <TelnetLayer.h>
#include <BgpLayer.h>
#include <DhcpLayer.h>
#include <DhcpV6Layer.h>
#include <DnsLayer.h>
#include <IcmpLayer.h>
#include <NtpLayer.h>
#include <SSLLayer.h>
#include <SSLHandshake.h>
#include <TcpLayer.h>
#include <SdpLayer.h>
#include <VrrpLayer.h>
#include <Sll2Layer.h>
#include <FtpLayer.h>
#include <GreLayer.h>
#include <GtpLayer.h>
#include <SomeIpSdLayer.h>

// Call some pcpp::Packet methods that are not invoked from general virtual methods
// as `pcpp::Packet::toString` or `pcpp::Packet::computeCalculateFields` to trigger possible crashes.
// The general rule is the functions do not modify the `parsedPacket`.
// If you made changes to PcapPlusPlus and the code doesn't compile - fix the method call as any other unit test
static void readParsedPacket(pcpp::Packet parsedPacket, pcpp::Layer* layer)
{
  if (parsedPacket.isPacketOfType(pcpp::Telnet))
  {
    if (auto telnetLayer = dynamic_cast<pcpp::TelnetLayer*>(layer))
    {
      telnetLayer->getFirstCommand();
      telnetLayer->getTotalNumberOfCommands();

      pcpp::TelnetLayer::TelnetCommand commandVal;
      do
      {
        commandVal = telnetLayer->getNextCommand();
        std::cout << "Telnet command is '" << telnetLayer->getTelnetCommandAsString(commandVal) << "'"
                  << std::endl;
        pcpp::TelnetLayer::TelnetOption option = telnetLayer->getOption();
        std::cout << "Telnet option is '" << telnetLayer->getTelnetOptionAsString(option) << "'" << std::endl;

        telnetLayer->getDataAsString(true);
        telnetLayer->getNumberOfCommands(commandVal);
        telnetLayer->getOption(commandVal);
        size_t length = 0;
        telnetLayer->getOptionData(length);
        telnetLayer->getOptionData(commandVal, length);
      } while (commandVal != pcpp::TelnetLayer::TelnetCommand::TelnetCommandEndOfPacket);
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::ARP))
  {
    if (auto arpLayer = dynamic_cast<pcpp::ArpLayer*>(layer))
    {
      arpLayer->isReply();
      arpLayer->isRequest();
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::SomeIP))
  {
    if (auto someipLayer = dynamic_cast<pcpp::SomeIpSdLayer*>(layer))
    {
      auto entries = someipLayer->getEntries();
      if (!entries.empty())
      {
        auto opts = someipLayer->getOptionsFromEntry(0);
        for (auto opt : opts)
          delete opt;
      }

      for (auto entry : entries)
      {
        entry->getNumOptions();
        entry->getServiceId();
        entry->getInstanceId();
        entry->getMajorVersion();
        entry->getMinorVersion();
        entry->getCounter();
        entry->getEventgroupId();
        delete entry;
      }

      someipLayer->getFlags();
      auto opts = someipLayer->getOptions();
      for (auto opt : opts)
      {
        opt->getType();
        if (auto v4opt = dynamic_cast<pcpp::SomeIpSdIPv4Option*>(opt))
        {
          v4opt->getIpAddress();
          v4opt->getPort();
          v4opt->getProtocol();
        }
        else if (auto v6opt = dynamic_cast<pcpp::SomeIpSdIPv6Option*>(opt))
        {
          v6opt->getIpAddress();
          v6opt->getPort();
          v6opt->getProtocol();
        }
        delete opt;
      }
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::GTP))
  {
    if (auto gtpLayer = dynamic_cast<pcpp::GtpV1Layer*>(layer))
    {
      uint16_t value16 = 0;
      gtpLayer->getSequenceNumber(value16);
      uint8_t value8;
      gtpLayer->getNpduNumber(value8);
      gtpLayer->getMessageType();
      gtpLayer->getMessageTypeAsString();
      gtpLayer->isGTPUMessage();
      gtpLayer->isGTPCMessage();
      auto ext = gtpLayer->getNextExtension();
      ext.getExtensionType();
      ext.getContent();
      ext.getContentLength();
      ext.getNextExtension();
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::GRE))
  {
    if (auto greLayer = dynamic_cast<pcpp::GreLayer*>(layer))
    {
      uint32_t value32 = 0;
      greLayer->getSequenceNumber(value32);
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::GREv0))
  {
    if (auto greLayer = dynamic_cast<pcpp::GREv0Layer*>(layer))
    {
      uint16_t value16 = 0;
      greLayer->getChecksum(value16);
      greLayer->getOffset(value16);
      uint32_t value32 = 0;
      greLayer->getKey(value32);
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::GREv1))
  {
    if (auto greLayer = dynamic_cast<pcpp::GREv1Layer*>(layer))
    {
      uint32_t value32 = 0;
      greLayer->getAcknowledgmentNum(value32);
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::FTP))
  {
    if (auto ftpLayer = dynamic_cast<pcpp::FtpRequestLayer*>(layer))
    {
      ftpLayer->getCommandOption(false);
      ftpLayer->getCommandOption(true);
    }
    else if (auto ftpLayer = dynamic_cast<pcpp::FtpResponseLayer*>(layer))
    {
      ftpLayer->getStatusCode();
      ftpLayer->getStatusOption(false);
      ftpLayer->getStatusOption(true);
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::SLL2))
  {
    if (auto sllLayer = dynamic_cast<pcpp::Sll2Layer*>(layer))
    {
      sllLayer->getLinkLayerAsMacAddress();
      sllLayer->getProtocolType();
      sllLayer->getInterfaceIndex();
      sllLayer->getArphrdType();
      sllLayer->getPacketType();
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::VRRP))
  {
    if (auto vrrpLayer = dynamic_cast<pcpp::VrrpLayer*>(layer))
    {
      vrrpLayer->getIPAddresses();
      vrrpLayer->isChecksumCorrect();
      vrrpLayer->getChecksum();
      vrrpLayer->getPriorityAsEnum();
      vrrpLayer->getPriority();
      vrrpLayer->getType();
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::VRRPv2))
  {
    if (auto vrrpLayer = dynamic_cast<pcpp::VrrpV2Layer*>(layer))
    {
      vrrpLayer->getAuthTypeAsEnum();
      vrrpLayer->getAdvInt();
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::VRRPv3))
  {
    if (auto vrrpLayer = dynamic_cast<pcpp::VrrpV3Layer*>(layer))
    {
      vrrpLayer->getMaxAdvInt();
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::TCP))
  {
    if (auto tcpLayer = dynamic_cast<pcpp::TcpLayer*>(layer))
    {
      auto tcpLayer2(*tcpLayer);
      tcpLayer2.insertTcpOptionAfter(pcpp::TcpOptionBuilder(pcpp::TcpOptionBuilder::NopEolOptionEnumType::Nop),
                                     pcpp::TcpOptionEnumType::Nop);
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::SDP))
  {
    if (auto sdpLayer = dynamic_cast<pcpp::SdpLayer*>(layer))
    {
      sdpLayer->getOwnerIPv4Address();
      sdpLayer->getMediaPort("audio");
      sdpLayer->getFieldCount();

      auto sdpLayer2 = *sdpLayer;
      std::vector<std::string> audioAttributes;
      audioAttributes.push_back("rtpmap:8 PCMA/8000");
      sdpLayer2.addMediaDescription("audio", 6010, "RTP/AVP", "8 96", audioAttributes);
      sdpLayer2.addField(PCPP_SDP_PROTOCOL_VERSION_FIELD, "0");
      sdpLayer2.removeField(PCPP_SDP_PROTOCOL_VERSION_FIELD);
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::SSL))
  {
    if (auto handshakeLayer = dynamic_cast<pcpp::SSLHandshakeLayer*>(layer))
    {
      if (auto clientHelloMessage = handshakeLayer->getHandshakeMessageOfType<pcpp::SSLClientHelloMessage>())
      {
        clientHelloMessage->getCompressionMethodsValue();
        clientHelloMessage->getSessionID();
        clientHelloMessage->getHandshakeType();
        clientHelloMessage->getHandshakeVersion();

        pcpp::SSLCipherSuite::getCipherSuiteByName("TLS_RSA_WITH_NULL_MD5");
        for (int i = 0; i < clientHelloMessage->getCipherSuiteCount(); i++)
        {
          clientHelloMessage->getCipherSuite(i);
          bool valid;
          clientHelloMessage->getCipherSuiteID(i, valid);
        }
        if (auto ext = clientHelloMessage->getExtensionOfType<pcpp::SSLServerNameIndicationExtension>())
          ext->getHostName();
        if (auto ext = clientHelloMessage->getExtensionOfType<pcpp::SSLSupportedVersionsExtension>())
          ext->getSupportedVersions();

        clientHelloMessage->getExtensionOfType(pcpp::SSL_EXT_SERVER_NAME);
        clientHelloMessage->getExtensionOfType((uint16_t)0);

        auto fingerprint = clientHelloMessage->generateTLSFingerprint();
        fingerprint.toMD5();
      }
      if (auto serverHelloMessage = handshakeLayer->getHandshakeMessageOfType<pcpp::SSLServerHelloMessage>())
      {
        serverHelloMessage->getCompressionMethodsValue();
        serverHelloMessage->getSessionID();
        serverHelloMessage->getCipherSuite();

        serverHelloMessage->getExtensionOfType<pcpp::SSLServerNameIndicationExtension>();
        serverHelloMessage->getExtensionOfType(pcpp::SSL_EXT_SERVER_NAME);
        serverHelloMessage->getExtensionOfType((uint16_t)0);

        serverHelloMessage->getHandshakeVersion();
        auto fingerprint = serverHelloMessage->generateTLSFingerprint();
        fingerprint.toMD5();
      }
      if (auto handshakeMessage = handshakeLayer->getHandshakeMessageOfType<pcpp::SSLHandshakeMessage>())
      {
        handshakeMessage->isMessageComplete();
      }
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::NTP))
  {
    if (auto ntpLayer = dynamic_cast<pcpp::NtpLayer*>(layer))
    {
      ntpLayer->getLeapIndicator();
      ntpLayer->getMode();
      ntpLayer->getModeString();
      ntpLayer->getStratum();
      ntpLayer->getPollInterval();
      ntpLayer->getPrecision();
      ntpLayer->getRootDelay();
      ntpLayer->getRootDispersion();
      ntpLayer->getReferenceIdentifier();
      ntpLayer->getReferenceIdentifierString();
      ntpLayer->getReferenceTimestamp();
      ntpLayer->getOriginTimestamp();
      ntpLayer->getReceiveTimestamp();
      ntpLayer->getTransmitTimestamp();

      ntpLayer->getDigest();
      ntpLayer->getKeyID();

      ntpLayer->getPollIntervalInSecs();
      ntpLayer->getPrecisionInSecs();
      ntpLayer->getRootDelayInSecs();
      ntpLayer->getRootDispersionInSecs();
      ntpLayer->getReferenceTimestampInSecs();
      ntpLayer->getOriginTimestampInSecs();
      ntpLayer->getReceiveTimestampInSecs();
      ntpLayer->getTransmitTimestampInSecs();

      ntpLayer->getReferenceTimestampAsString();
      ntpLayer->getOriginTimestampAsString();
      ntpLayer->getReceiveTimestampAsString();
      ntpLayer->getTransmitTimestampAsString();

      auto ntpLayer2(*ntpLayer);
      ntpLayer2.setRootDelayInSecs(0.1);
      ntpLayer2.setReferenceTimestampInSecs(0.1);
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::ICMP))
  {
    if (auto icmpLayer = dynamic_cast<pcpp::IcmpLayer*>(layer))
    {
      auto icmpLayer2(*icmpLayer);

      if (icmpLayer->isMessageOfType(pcpp::ICMP_TIMESTAMP_REPLY))
      {
        icmpLayer->getTimestampReplyData();
        timeval orig = { 16131, 171000 };
        timeval recv = { 16133, 474000 };
        timeval tran = { 16133, 474000 };
        icmpLayer2.setTimestampReplyData(14640, 0, orig, recv, tran);
      }
      else if (icmpLayer->isMessageOfType(pcpp::ICMP_ADDRESS_MASK_REQUEST))
      {
        icmpLayer->getAddressMaskRequestData();
        icmpLayer2.setAddressMaskRequestData(45068, 1536, pcpp::IPv4Address::Zero);
      }
      else if (icmpLayer->isMessageOfType(pcpp::ICMP_ADDRESS_MASK_REPLY))
      {
        icmpLayer->getAddressMaskReplyData();
        icmpLayer2.setAddressMaskReplyData(45068, 1536, pcpp::IPv4Address::Zero);
      }
      else if (icmpLayer->isMessageOfType(pcpp::ICMP_DEST_UNREACHABLE))
      {
        icmpLayer->getDestUnreachableData();
        icmpLayer2.setDestUnreachableData(pcpp::IcmpHostUnreachable, 0, nullptr, nullptr);
      }
      else if (icmpLayer->isMessageOfType(pcpp::ICMP_INFO_REPLY))
      {
        auto layerData = icmpLayer->getInfoReplyData();
        icmpLayer2.setInfoReplyData(layerData->id, layerData->sequence);
      }
      else if (icmpLayer->isMessageOfType(pcpp::ICMP_INFO_REQUEST))
      {
        auto layerData = icmpLayer->getInfoRequestData();
        icmpLayer2.setInfoRequestData(layerData->id, layerData->sequence);
      }
      else if (icmpLayer->isMessageOfType(pcpp::ICMP_PARAM_PROBLEM))
      {
        auto layerData = icmpLayer->getParamProblemData();
        icmpLayer2.setParamProblemData(layerData->code, layerData->pointer, nullptr, nullptr);
      }
      else if (icmpLayer->isMessageOfType(pcpp::ICMP_TIME_EXCEEDED))
      {
        icmpLayer->getTimeExceededData();
        icmpLayer2.setTimeExceededData(1, nullptr, nullptr);
      }
      else if (icmpLayer->isMessageOfType(pcpp::ICMP_ROUTER_ADV))
      {
        icmpLayer->getRouterAdvertisementData();
        pcpp::icmp_router_address_structure addr1;
        addr1.setRouterAddress(pcpp::IPv4Address("192.168.144.2"), (uint32_t)0x08000000);
        std::vector<pcpp::icmp_router_address_structure> routerAddresses;
        routerAddresses.push_back(addr1);
        icmpLayer2.setRouterAdvertisementData(16, 200, routerAddresses);
      }
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::DHCPv6))
  {
    if (auto dhcpLayer = dynamic_cast<pcpp::DhcpV6Layer*>(layer))
    {
      dhcpLayer->getTransactionID();
      if (dhcpLayer->getOptionCount() > 0)
      {
        pcpp::DhcpV6Option opt = dhcpLayer->getFirstOptionData();
        opt.getType();
        opt.getTotalSize();
        opt.getValueAsHexString();
        for (size_t i = 0; i < dhcpLayer->getOptionCount(); i++)
        {
          opt = dhcpLayer->getNextOptionData(opt);
        }
        dhcpLayer->getOptionData(pcpp::DHCPV6_OPT_CLIENTID);
      }
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::DHCP))
  {
    if (auto dhcpLayer = dynamic_cast<pcpp::DhcpLayer*>(layer))
    {
      dhcpLayer->getOpCode();
      dhcpLayer->getDhcpHeader();
      dhcpLayer->getClientIpAddress();
      dhcpLayer->getYourIpAddress();
      dhcpLayer->getServerIpAddress();
      dhcpLayer->getGatewayIpAddress();
      dhcpLayer->getClientHardwareAddress();
      if (dhcpLayer->getOptionsCount() > 0)
      {
        pcpp::DhcpOption opt = dhcpLayer->getFirstOptionData();
        opt.getValueAsIpAddr();
        opt.getValueAsString();
        for (size_t i = 0; i < dhcpLayer->getOptionsCount(); i++)
        {
          opt = dhcpLayer->getNextOptionData(opt);
        }
      }
      dhcpLayer->getOptionData(pcpp::DHCPOPT_SUBNET_MASK);
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::BGP))
  {
    if (auto bgpLayer = dynamic_cast<pcpp::BgpLayer*>(layer))
    {
      bgpLayer->getMessageTypeAsString();
      if (auto bgpOpenMsgLayer = dynamic_cast<pcpp::BgpOpenMessageLayer*>(bgpLayer))
      {
        std::vector<pcpp::BgpOpenMessageLayer::optional_parameter> optionalParams;
        bgpOpenMsgLayer->getOptionalParameters(optionalParams);
        std::vector<pcpp::BgpOpenMessageLayer::optional_parameter> optionalParams2(optionalParams);
        optionalParams2.push_back(pcpp::BgpOpenMessageLayer::optional_parameter(2, "010400010001"));
        optionalParams2.push_back(pcpp::BgpOpenMessageLayer::optional_parameter(2, "8000"));
        optionalParams2.push_back(pcpp::BgpOpenMessageLayer::optional_parameter(2, "0200"));
        optionalParams2.push_back(pcpp::BgpOpenMessageLayer::optional_parameter(2, "4600"));
        optionalParams2.push_back(pcpp::BgpOpenMessageLayer::optional_parameter(2, "410400000001"));
        bgpOpenMsgLayer->setOptionalParameters(optionalParams2);
        bgpOpenMsgLayer->clearOptionalParameters();
        bgpOpenMsgLayer->setOptionalParameters(optionalParams);
      }
      else if (auto bgpUpdateMsgLayer = dynamic_cast<pcpp::BgpUpdateMessageLayer*>(bgpLayer))
      {
        std::vector<pcpp::BgpUpdateMessageLayer::prefix_and_ip> withdrawnRoutes;
        bgpUpdateMsgLayer->getWithdrawnRoutes(withdrawnRoutes);
        std::vector<pcpp::BgpUpdateMessageLayer::prefix_and_ip> withdrawnRoutes2(withdrawnRoutes);
        withdrawnRoutes2.push_back(pcpp::BgpUpdateMessageLayer::prefix_and_ip(24, "40.1.1.0"));
        withdrawnRoutes2.push_back(pcpp::BgpUpdateMessageLayer::prefix_and_ip(24, "40.40.40.0"));
        withdrawnRoutes2.push_back(pcpp::BgpUpdateMessageLayer::prefix_and_ip(16, "103.103.0.0"));
        withdrawnRoutes2.push_back(pcpp::BgpUpdateMessageLayer::prefix_and_ip(24, "103.103.40.0"));
        bgpUpdateMsgLayer->setWithdrawnRoutes(withdrawnRoutes2);
        bgpUpdateMsgLayer->clearWithdrawnRoutes();
        bgpUpdateMsgLayer->setWithdrawnRoutes(withdrawnRoutes);

        std::vector<pcpp::BgpUpdateMessageLayer::prefix_and_ip> nlriVec;
        bgpUpdateMsgLayer->getNetworkLayerReachabilityInfo(nlriVec);
        std::vector<pcpp::BgpUpdateMessageLayer::prefix_and_ip> nlriVec2(nlriVec);
        nlriVec2.push_back(pcpp::BgpUpdateMessageLayer::prefix_and_ip(24, "40.1.1.0"));
        bgpUpdateMsgLayer->setNetworkLayerReachabilityInfo(nlriVec2);
        bgpUpdateMsgLayer->clearNetworkLayerReachabilityInfo();
        bgpUpdateMsgLayer->setNetworkLayerReachabilityInfo(nlriVec);

        std::vector<pcpp::BgpUpdateMessageLayer::path_attribute> pathAttributes;
        bgpUpdateMsgLayer->getPathAttributes(pathAttributes);
        std::vector<pcpp::BgpUpdateMessageLayer::path_attribute> pathAttributes2(pathAttributes);
        pathAttributes2.push_back(pcpp::BgpUpdateMessageLayer::path_attribute(0x40, 1, "02"));
        pathAttributes2.push_back(
            pcpp::BgpUpdateMessageLayer::path_attribute(0x40, 2, "02030000000a0000001400000028"));
        pathAttributes2.push_back(pcpp::BgpUpdateMessageLayer::path_attribute(0x40, 3, "1e031e03"));
        bgpUpdateMsgLayer->setPathAttributes(pathAttributes2);
        bgpUpdateMsgLayer->clearPathAttributes();
        bgpUpdateMsgLayer->setPathAttributes(pathAttributes);
      }
      else if (auto bgpNotificationMsgLayer = dynamic_cast<pcpp::BgpNotificationMessageLayer*>(bgpLayer))
      {
        bgpNotificationMsgLayer->getNotificationDataAsHexString();
      }
    }
  }
  if (parsedPacket.isPacketOfType(pcpp::DNS))
  {
    if (auto dnsLayer = dynamic_cast<pcpp::DnsLayer*>(layer))
    {
      dnsLayer->addQuery("mail-attachment.googleusercontent.com", pcpp::DNS_TYPE_A, pcpp::DNS_CLASS_IN);
      dnsLayer->removeQuery("a", true);
      dnsLayer->removeQuery("mail-attachment.googleusercontent.com", false);
      pcpp::IPv4DnsResourceData ipv4DnsData(std::string("151.249.90.217"));
      dnsLayer->addAnswer("assets.pinterest.com.cdngc.net", pcpp::DNS_TYPE_A, pcpp::DNS_CLASS_IN, 3,
                          &ipv4DnsData);
      dnsLayer->removeAnswer("a", true);
      dnsLayer->removeAnswer("assets.pinterest.com.cdngc.net", false);
      dnsLayer->addAuthority("Yaels-iPhone.local", pcpp::DNS_TYPE_A, pcpp::DNS_CLASS_IN, 120, &ipv4DnsData);
      dnsLayer->removeAuthority("a", true);
      dnsLayer->removeAuthority("Yaels-iPhone.local", false);
      pcpp::GenericDnsResourceData genericData("0004000800df581faa4f3f9d");
      dnsLayer->addAdditionalRecord("abc", pcpp::DNS_TYPE_OPT, 0xa005, 0x1194, &genericData);
      dnsLayer->removeAdditionalRecord("a", true);
      dnsLayer->removeAdditionalRecord("abc", false);

      auto add = dnsLayer->getFirstAdditionalRecord();
      while (add != nullptr)
      {
        add = dnsLayer->getNextAdditionalRecord(add);
      }

      auto answer = dnsLayer->getFirstAnswer();
      while (answer != nullptr)
      {
        answer = dnsLayer->getNextAnswer(answer);
      }

      auto auth = dnsLayer->getFirstAuthority();
      while (auth != nullptr)
      {
        auth = dnsLayer->getNextAuthority(auth);
      }

      pcpp::DnsLayer other(*dnsLayer);
      other = *dnsLayer;
    }
  }
}

Coverage Report

Created: 2025-12-14 07:46