Centralize code for fetching pcap devices.

Pcap++/CMakeLists.txt

@@ -1,11 +1,13 @@
 add_library(
  Pcap++
+ src/DeviceUtils.cpp
  $<$<BOOL:${PCAPPP_USE_DPDK}>:src/DpdkDevice.cpp>
  $<$<BOOL:${PCAPPP_USE_DPDK}>:src/DpdkDeviceList.cpp>
  $<$<BOOL:${PCAPPP_USE_DPDK_KNI}>:src/KniDevice.cpp>
  $<$<BOOL:${PCAPPP_USE_DPDK_KNI}>:src/KniDeviceList.cpp>
  $<$<BOOL:${LINUX}>:src/LinuxNicInformationSocket.cpp>
  $<$<BOOL:${PCAPPP_USE_DPDK}>:src/MBufRawPacket.cpp>
+ src/MemoryUtils.cpp
  src/NetworkUtils.cpp
  src/PcapFileDevice.cpp
  src/PcapDevice.cpp
@@ -24,6 +26,8 @@ add_library(
 
 set(public_headers
  header/Device.h
+ header/DeviceUtils.h
+ header/MemoryUtils.h
  header/NetworkUtils.h
  header/PcapDevice.h
  header/PcapFileDevice.h

Pcap++/header/DeviceUtils.h

@@ -0,0 +1,34 @@
+#pragma once
+
+/// @file
+
+#include <memory>
+#include "IpAddress.h"
+#include "MemoryUtils.h"
+
+// Forward declaration
+struct pcap_rmtauth;
+
+namespace pcpp
+{
+ namespace internal
+ {
+ /**
+ * Fetches a list of all network devices on the local machine that LibPcap/WinPcap/NPcap can find.
+ * @return A smart pointer to an interface list structure.
+ * @throws std::runtime_error The system encountered an error fetching the devices.
+ */
+ std::unique_ptr<pcap_if_t, PcapFreeAllDevsDeleter> getAllLocalPcapDevices();
+#ifdef _WIN32
+ /**
+ * Fetches a list of all network devices on a remote machine that WinPcap/NPcap can find.
+ * @param[in] ipAddress IP address of the remote machine.
+ * @param[in] port Port to use when connecting to the remote machine.
+ * @param[in] pRmAuth Pointer to an authentication structure to use when connecting to the remote machine. Nullptr if no authentication is required.
+ * @return A smart pointer to an interface list structure.
+ * @throws std::runtime_error The system encountered an error fetching the devices.
+ */
+ std::unique_ptr<pcap_if_t, PcapFreeAllDevsDeleter> getAllRemotePcapDevices(const IPAddress& ipAddress, uint16_t port, pcap_rmtauth* pRmAuth = nullptr);
+#endif // _WIN32
+ }
+}

Pcap++/header/MemoryUtils.h

@@ -0,0 +1,31 @@
+#pragma once
+
+// Forward declarations
+struct pcap;
+typedef pcap pcap_t;
+struct pcap_if;
+typedef pcap_if pcap_if_t;
+
+namespace pcpp
+{
+ namespace internal
+ {
+ /**
+ * @class PcapCloseDeleter
+ * A deleter that cleans up a pcap_t structure by calling pcap_close.
+ */
+ struct PcapCloseDeleter
+ {
+ void operator()(pcap_t* ptr) const;
+ };
+
+ /**
+ * @class PcapFreeAllDevsDeleter
+ * A deleter that frees an interface list of pcap_if_t ptr by calling 'pcap_freealldevs' function on it.
+ */
+ struct PcapFreeAllDevsDeleter
+ {
+ void operator()(pcap_if_t* ptr) const;
+ };
+ }
+}

Pcap++/src/DeviceUtils.cpp

@@ -0,0 +1,50 @@
+#include "DeviceUtils.h"
+
+#include <array>
+#include <string>
+
+#include "pcap.h"
+#include "Logger.h"
+#include "IpAddress.h"
+
+namespace pcpp
+{
+ namespace internal
+ {
+ std::unique_ptr<pcap_if_t, PcapFreeAllDevsDeleter> getAllLocalPcapDevices()
+ {
+ pcap_if_t* interfaceListRaw;
+ std::array<char, PCAP_ERRBUF_SIZE> errbuf;
+ int err = pcap_findalldevs(&interfaceListRaw, errbuf.data());
+ if (err < 0)
+ {
+ throw std::runtime_error("Error searching for devices: " + std::string(errbuf.begin(), errbuf.end()));
+ }
+ // Assigns the raw pointer to the smart pointer with specialized deleter.
+ return std::unique_ptr<pcap_if_t, internal::PcapFreeAllDevsDeleter>(interfaceListRaw);
+ }
+
+#ifdef _WIN32
+ std::unique_ptr<pcap_if_t, PcapFreeAllDevsDeleter> getAllRemotePcapDevices(const IPAddress& ipAddress, uint16_t port, pcap_rmtauth* pRmAuth)
+ {
+ PCPP_LOG_DEBUG("Searching remote devices on IP: " << ipAddress << " and port: " << port);
+ std::array<char, PCAP_BUF_SIZE> remoteCaptureString;
+ std::array<char, PCAP_ERRBUF_SIZE> errorBuf;
+ if (pcap_createsrcstr(remoteCaptureString.data(), PCAP_SRC_IFREMOTE, ipAddress.toString().c_str(),
+ std::to_string(port).c_str(), nullptr, errorBuf.data()) != 0)
+ {
+ throw std::runtime_error("Error creating the remote connection string. Error: " + std::string(errorBuf.begin(), errorBuf.end()));
+ }
+
+ PCPP_LOG_DEBUG("Remote capture string: " << remoteCaptureString.data());
+
+ pcap_if_t* interfaceListRaw;
+ if (pcap_findalldevs_ex(remoteCaptureString.data(), pRmAuth, &interfaceListRaw, errorBuf.data()) < 0)
+ {
+ throw std::runtime_error("Error retrieving device on remote machine. Error: " + std::string(errorBuf.begin(), errorBuf.end()));
+ }
+ return std::unique_ptr<pcap_if_t, internal::PcapFreeAllDevsDeleter>(interfaceListRaw);
+ }
+#endif // _WIN32
+ }
+}

Pcap++/src/MemoryUtils.cpp

@@ -0,0 +1,13 @@
+#include "MemoryUtils.h"
+
+#include "pcap.h"
+
+namespace pcpp
+{
+ namespace internal
+ {
+ void PcapCloseDeleter::operator()(pcap_t* ptr) const { pcap_close(ptr); }
+
+ void PcapFreeAllDevsDeleter::operator()(pcap_if_t* ptr) const { pcap_freealldevs(ptr); }
+ }
+}

Pcap++/src/PcapLiveDevice.cpp

@@ -1,6 +1,8 @@
 #define LOG_MODULE PcapLogModuleLiveDevice
 
 #include "IpUtils.h"
+#include "DeviceUtils.h"
+#include "MemoryUtils.h"
 #include "PcapLiveDevice.h"
 #include "PcapLiveDeviceList.h"
 #include "Packet.h"
@@ -11,7 +13,7 @@
 #include <thread>
 #include "Logger.h"
 #include "SystemUtils.h"
-#include <string.h>
+#include <cstring>
 #include <iostream>
 #include <fstream>
 #include <chrono>
@@ -405,32 +407,27 @@ void PcapLiveDevice::close()
 
 PcapLiveDevice* PcapLiveDevice::clone() const
 {
- PcapLiveDevice* retval = nullptr;
-
- pcap_if_t* interfaceList;
- char errbuf[PCAP_ERRBUF_SIZE];
- int err = pcap_findalldevs(&interfaceList, errbuf);
- if (err < 0)
+ std::unique_ptr<pcap_if_t, internal::PcapFreeAllDevsDeleter> interfaceList;
+ try
+ {
+  interfaceList = internal::getAllLocalPcapDevices();
+ }
+ catch (const std::exception& e)
  {
- PCPP_LOG_ERROR("Error searching for devices: " << errbuf);
+ PCPP_LOG_ERROR(e.what());
  return nullptr;
  }
 
- pcap_if_t* currInterface = interfaceList;
- while (currInterface != nullptr)
+ for (pcap_if_t* currInterface = interfaceList.get(); currInterface != nullptr; currInterface = currInterface->next)
  {
- if(!strcmp(currInterface->name, getName().c_str()))
- break;
- currInterface = currInterface->next;
+ if (!std::strcmp(currInterface->name, getName().c_str()))
+ {
+ return cloneInternal(*currInterface);
+ }
  }
 
- if(currInterface)
- retval = cloneInternal(*currInterface);
- else
- PCPP_LOG_ERROR("Can't find interface " << getName().c_str());
-
- pcap_freealldevs(interfaceList);
- return retval;
+ PCPP_LOG_ERROR("Can't find interface " << getName().c_str());
+ return nullptr;
 }
 
 PcapLiveDevice* PcapLiveDevice::cloneInternal(pcap_if_t& devInterface) const

Pcap++/src/PcapLiveDeviceList.cpp

@@ -4,6 +4,8 @@
 #include "IpAddressUtils.h"
 #include "PcapLiveDeviceList.h"
 #include "Logger.h"
+#include "MemoryUtils.h"
+#include "DeviceUtils.h"
 #include "SystemUtils.h"
 #include "pcap.h"
 #include <string.h>
@@ -39,32 +41,30 @@ PcapLiveDeviceList::~PcapLiveDeviceList()
 
 void PcapLiveDeviceList::init()
 {
- pcap_if_t* interfaceList;
- char errbuf[PCAP_ERRBUF_SIZE];
- int err = pcap_findalldevs(&interfaceList, errbuf);
- if (err < 0)
+ std::unique_ptr<pcap_if_t, internal::PcapFreeAllDevsDeleter> interfaceList;
+ try
+ {
+ interfaceList = internal::getAllLocalPcapDevices();
+ }
+ catch (const std::exception& e)
  {
- PCPP_LOG_ERROR("Error searching for devices: " << errbuf);
+ PCPP_LOG_ERROR(e.what());
  }
 
  PCPP_LOG_DEBUG("Pcap lib version info: " << IPcapDevice::getPcapLibVersionInfo());
 
- pcap_if_t* currInterface = interfaceList;
- while (currInterface != nullptr)
+
+ for (pcap_if_t* currInterface = interfaceList.get(); currInterface != nullptr; currInterface = currInterface->next)
  {
 #if defined(_WIN32)
  PcapLiveDevice* dev = new WinPcapLiveDevice(currInterface, true, true, true);
 #else //__linux__, __APPLE__, __FreeBSD__
  PcapLiveDevice* dev = new PcapLiveDevice(currInterface, true, true, true);
 #endif
- currInterface = currInterface->next;
  m_LiveDeviceList.insert(m_LiveDeviceList.end(), dev);
  }
 
  setDnsServers();
-
- PCPP_LOG_DEBUG("Freeing live device data");
- pcap_freealldevs(interfaceList);
 }
 
 void PcapLiveDeviceList::setDnsServers()

Pcap++/src/PcapRemoteDeviceList.cpp

@@ -5,6 +5,8 @@
 #include "PcapRemoteDeviceList.h"
 #include "Logger.h"
 #include "IpUtils.h"
+#include "DeviceUtils.h"
+#include "MemoryUtils.h"
 #include "IpAddressUtils.h"
 #include "pcap.h"
 #include <array>
@@ -20,51 +22,39 @@ PcapRemoteDeviceList* PcapRemoteDeviceList::getRemoteDeviceList(const IPAddress&
 
 PcapRemoteDeviceList* PcapRemoteDeviceList::getRemoteDeviceList(const IPAddress& ipAddress, uint16_t port, PcapRemoteAuthentication* remoteAuth)
 {
- PCPP_LOG_DEBUG("Searching remote devices on IP: " << ipAddress << " and port: " << port);
- char remoteCaptureString[PCAP_BUF_SIZE];
- char errbuf[PCAP_ERRBUF_SIZE];
- std::ostringstream portAsString;
- portAsString << port;
- if (pcap_createsrcstr(remoteCaptureString, PCAP_SRC_IFREMOTE, ipAddress.toString().c_str(), portAsString.str().c_str(), NULL, errbuf) != 0)
- {
- PCPP_LOG_ERROR("Error in creating the remote connection string. Error was: " << errbuf);
- return NULL;
- }
-
- PCPP_LOG_DEBUG("Remote capture string: " << remoteCaptureString);
-
- pcap_rmtauth* pRmAuth = NULL;
+ pcap_rmtauth* pRmAuth = nullptr;
  pcap_rmtauth rmAuth;
- if (remoteAuth != NULL)
+ if (remoteAuth != nullptr)
  {
  PCPP_LOG_DEBUG("Authentication requested. Username: " << remoteAuth->userName << ", Password: " << remoteAuth->password);
  rmAuth = remoteAuth->getPcapRmAuth();
  pRmAuth = &rmAuth;
  }
 
- pcap_if_t* interfaceList;
- char errorBuf[PCAP_ERRBUF_SIZE];
- if (pcap_findalldevs_ex(remoteCaptureString, pRmAuth, &interfaceList, errorBuf) < 0)
+ std::unique_ptr<pcap_if_t, internal::PcapFreeAllDevsDeleter> interfaceList;
+ try
  {
- PCPP_LOG_ERROR("Error retrieving device on remote machine. Error string is: " << errorBuf);
- return NULL;
+ interfaceList = internal::getAllRemotePcapDevices(ipAddress, port, pRmAuth);
+ }
+ catch (const std::exception& e)
+ {
+ PCPP_LOG_ERROR(e.what());
+ return nullptr;
  }
 
  PcapRemoteDeviceList* resultList = new PcapRemoteDeviceList();
  resultList->setRemoteMachineIpAddress(ipAddress);
  resultList->setRemoteMachinePort(port);
  resultList->setRemoteAuthentication(remoteAuth);
 
- pcap_if_t* currInterface = interfaceList;
- while (currInterface != NULL)
+
+ for (pcap_if_t* currInterface = interfaceList.get(); currInterface != nullptr; currInterface = currInterface->next)
  {
  PcapRemoteDevice* pNewRemoteDevice = new PcapRemoteDevice(currInterface, resultList->m_RemoteAuthentication,
  resultList->getRemoteMachineIpAddress(), resultList->getRemoteMachinePort());
  resultList->m_RemoteDeviceList.push_back(pNewRemoteDevice);
- currInterface = currInterface->next;
  }
 
- pcap_freealldevs(interfaceList);
  return resultList;
 }