mirror of
https://github.com/mfontanini/libtins
synced 2026-01-23 10:45:57 +01:00
Compare commits
98 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
67c81199fb | ||
|
|
fe22186623 | ||
|
|
bbac2ece52 | ||
|
|
ba9a2155ca | ||
|
|
00619e0500 | ||
|
|
64da67ca56 | ||
|
|
3943700830 | ||
|
|
142b6f62cb | ||
|
|
dfd2701ee4 | ||
|
|
850bb9b642 | ||
|
|
b7e61f4c76 | ||
|
|
e2a14d8898 | ||
|
|
ba0c820852 | ||
|
|
df509e7e36 | ||
|
|
fa87e1b6f6 | ||
|
|
812be7966d | ||
|
|
f89cc9f076 | ||
|
|
638bf9b34b | ||
|
|
eb997f5438 | ||
|
|
18cbab4fc7 | ||
|
|
7cd2b2c396 | ||
|
|
2601493752 | ||
|
|
7204fbd688 | ||
|
|
c302e659d7 | ||
|
|
54e4e4b0f4 | ||
|
|
6a17e59032 | ||
|
|
e3aedc56ed | ||
|
|
3b006c15db | ||
|
|
177d0b4621 | ||
|
|
24ac038c30 | ||
|
|
a619e4ff98 | ||
|
|
14bb185d7a | ||
|
|
137b56d5a7 | ||
|
|
ed2b3c12d5 | ||
|
|
1650b60234 | ||
|
|
c20c82bcb5 | ||
|
|
5858132261 | ||
|
|
16f5795243 | ||
|
|
e90e377b73 | ||
|
|
222611b377 | ||
|
|
b447c664e1 | ||
|
|
468159e6d2 | ||
|
|
cd40b232e7 | ||
|
|
1166094a2f | ||
|
|
b3d874d6a8 | ||
|
|
553b1fb255 | ||
|
|
94939dd0fa | ||
|
|
0774a8dcad | ||
|
|
f46dee9f19 | ||
|
|
5b082a82b2 | ||
|
|
07012648fb | ||
|
|
ce409dbc7e | ||
|
|
16e77146ab | ||
|
|
a87c4a64f5 | ||
|
|
9e61286a59 | ||
|
|
8da102fb48 | ||
|
|
750c3556d9 | ||
|
|
28663b0e93 | ||
|
|
731e36e373 | ||
|
|
608b48f25c | ||
|
|
de247fcbc8 | ||
|
|
7bc4d38470 | ||
|
|
a926b75224 | ||
|
|
064439236c | ||
|
|
0c40a0714b | ||
|
|
d74520768b | ||
|
|
3385df9cc9 | ||
|
|
18c31b20f5 | ||
|
|
7387912ca1 | ||
|
|
0d52763a61 | ||
|
|
86b505f998 | ||
|
|
62a803c55c | ||
|
|
0573808aeb | ||
|
|
22b4435c81 | ||
|
|
b803959e11 | ||
|
|
2f16497bf8 | ||
|
|
78aa7d1787 | ||
|
|
ba2216e6e9 | ||
|
|
74e3d909e6 | ||
|
|
5e52091ecf | ||
|
|
dafb299ea3 | ||
|
|
659bec56d7 | ||
|
|
8c1bea6f84 | ||
|
|
d8d1dc003b | ||
|
|
c2bb52b82f | ||
|
|
1f5456b18b | ||
|
|
dc702f4fd6 | ||
|
|
2a8101eef3 | ||
|
|
20702ddaff | ||
|
|
f4e2701705 | ||
|
|
776ae4e83b | ||
|
|
2158c7a92e | ||
|
|
b53bad7b29 | ||
|
|
6c92bcdad1 | ||
|
|
b949e56d15 | ||
|
|
18ff3e7b6a | ||
|
|
eb7628eca4 | ||
|
|
7e90c8be6a |
28
.github/workflows/tests.yaml
vendored
Normal file
28
.github/workflows/tests.yaml
vendored
Normal file
@@ -0,0 +1,28 @@
|
||||
name: Tests
|
||||
on:
|
||||
push:
|
||||
branches:
|
||||
- master
|
||||
pull_request:
|
||||
branches:
|
||||
- master
|
||||
jobs:
|
||||
Ubuntu-Tests:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v2
|
||||
|
||||
- name: Install libpcap
|
||||
run: sudo apt-get install -y libpcap-dev
|
||||
|
||||
- name: Initialize submodules
|
||||
run: git submodule init && git submodule update
|
||||
|
||||
- name: Initialize build system
|
||||
run: mkdir build && cd build && cmake ..
|
||||
|
||||
- name: Build tests
|
||||
run: cmake --build build --target tests
|
||||
|
||||
- name: Run tests
|
||||
run: ctest build
|
||||
132
CHANGES.md
132
CHANGES.md
@@ -1,3 +1,85 @@
|
||||
##### v4.5 - Sun Aug 20 04:46:53 PM UTC 2023
|
||||
|
||||
- Add VXLAN support (#501)
|
||||
|
||||
- Add `FileSniffer` constructor taking a `FILE*` (#499).
|
||||
|
||||
- Remove use of deprecated `std::iterator` (#481).
|
||||
|
||||
- Add missing `algorithm` header include (#497).
|
||||
|
||||
- Validate high order two bits of first dns label octet (#494).
|
||||
|
||||
- Use `CMAKE_INSTALL_LIBDIR` in `CONF_CMAKE_INSTALL_DIR` (#472).
|
||||
|
||||
- Ignore IPv6 packets with payload after one with no Next Header (#500).
|
||||
|
||||
- Fix unqualified `std::move` call warnings on clang (#488).
|
||||
|
||||
- Fix incorrect IP address range calculation when using /0 prefix (#484) (#486).
|
||||
|
||||
- Fall back to system GTest if available (#473).
|
||||
|
||||
- Fix compilation issue on android (#471).
|
||||
|
||||
- Fix DNS parser reading garbage on misreported packet size (#468).
|
||||
|
||||
- Fix DNS parser misinterpreting bad label size (#466).
|
||||
|
||||
##### v4.4 - Thu Feb 17 14:41:59 UTC 2022
|
||||
|
||||
- Add RFC8335 extended echo types to `ICMP` and `ICMPv6` (#426)
|
||||
|
||||
- Handle loops in DNS name decompression (#444)
|
||||
|
||||
- Fix Windows' `interface` macro colliding with uses of that identifier in the code (#458)
|
||||
|
||||
- Sending IPv6 packets to a link-scope destination address now uses the right interface index (#448)
|
||||
|
||||
- Fix incorrect endian being used for ICMP's `gateway` and `address_mask` (#437)
|
||||
|
||||
- Socket in `PacketSender::open_l3_socket` is now closed if `setsockopt` fails (#433)
|
||||
|
||||
- Fix various incorrect doxygen documentation comments (#439).
|
||||
|
||||
- Fix infinite loop when querying the routing table in \*BSD (#427)
|
||||
|
||||
##### v4.3 - Fri Sep 18 03:08:33 UTC 2020
|
||||
|
||||
- Assign a PDUType to `Dot11ControlTA` (#420)
|
||||
|
||||
- Don't consider IPv6 ESP header a normal extension header (#374)
|
||||
|
||||
- Don't include non-existing headers when installed without libpcap (#382)
|
||||
|
||||
- Add `IPv6Address::is_local_unicast` (#369)
|
||||
|
||||
- Fix memory leak in `PacketWriter` (#343)
|
||||
|
||||
- Fix memory leaks in `OfflinePacketFilter` (#343)
|
||||
|
||||
- Fix detection of new TCP stream (#335)
|
||||
|
||||
- Introduce `TCP::has_flags` (#334)
|
||||
|
||||
- Fix padding calculations in RadioTapWriter (#333)
|
||||
|
||||
##### v4.2 - Fri Mar 8 04:15:13 UTC 2019
|
||||
|
||||
- Updated location of installed CMake files in unix systems (#331)
|
||||
|
||||
- Fix check to detect cross compilation (#330)
|
||||
|
||||
- Allow getting a PDU's advertised size and use it in `PacketWriter` (#324)
|
||||
|
||||
- Install DLLs in right directory (#326)
|
||||
|
||||
- Add missing Dot11 tagged option types (#305)
|
||||
|
||||
- Add support for DLT_RAW (#313)
|
||||
|
||||
- Fix potential invalid memory accesses when parsing RadioTap (#322)
|
||||
|
||||
##### v4.1 - Tue Dec 11 02:08:48 UTC 2018
|
||||
|
||||
- Fix serialization for QinQ (#316)
|
||||
@@ -216,7 +298,7 @@
|
||||
|
||||
- Move implementations on utils.h to utils.cpp
|
||||
|
||||
- Add ICMPv6 Multicast Listener Query Messages support
|
||||
- Add ICMPv6 Multicast Listener Query Messages support
|
||||
|
||||
- Add ICMPv6 Multicast Listener Report Message support
|
||||
|
||||
@@ -360,7 +442,7 @@
|
||||
|
||||
- Added RadioTap channel map type
|
||||
|
||||
- Made rsn_information() a const member function to make Dot11ManagementFrame
|
||||
- Made rsn_information() a const member function to make Dot11ManagementFrame
|
||||
immutable
|
||||
|
||||
- Ensure HAVE_CXX11 is checked when defining TINS_IS_CXX11
|
||||
@@ -410,7 +492,7 @@ its value.
|
||||
|
||||
- Removed obsolete autotools files.
|
||||
|
||||
- Fixed exception thrown when an interface didn't have an IP address
|
||||
- Fixed exception thrown when an interface didn't have an IP address
|
||||
on NetworkInterface.
|
||||
|
||||
- Added NetworkInterface::is_loopback.
|
||||
@@ -454,7 +536,7 @@ conversion on integral constant.
|
||||
|
||||
- Added DataLinkType class.
|
||||
|
||||
- IPv4Address now uses inet_pton when constructing from string.
|
||||
- IPv4Address now uses inet_pton when constructing from string.
|
||||
|
||||
-------------------------------------------------------------------------------
|
||||
|
||||
@@ -475,8 +557,8 @@ conversion on integral constant.
|
||||
- Packets can now be constructed from a PDU pointer and take ownership
|
||||
of it.
|
||||
|
||||
- All protocols now set the next layer protocol flag, regardless if
|
||||
it was already set. This was not done in some protocols,
|
||||
- All protocols now set the next layer protocol flag, regardless if
|
||||
it was already set. This was not done in some protocols,
|
||||
like EthernetII, and as a consequence if the network layer protocol
|
||||
was replaced by other, the packet would be serialized incorrectly.
|
||||
|
||||
@@ -517,7 +599,7 @@ TCPStreamFollower::follow_streams.
|
||||
|
||||
- Added DHCP::hostname to extract the hostname options.
|
||||
|
||||
- Removed extra qualifier on SessionKeys::decrypt_unicast which
|
||||
- Removed extra qualifier on SessionKeys::decrypt_unicast which
|
||||
produced compilation errors on some platforms.
|
||||
|
||||
- PacketSender::send now uses PDU::matches_flag to match specific
|
||||
@@ -533,9 +615,9 @@ PDU types.
|
||||
|
||||
-------------------------------------------------------------------------------
|
||||
|
||||
##### v2.0 - Thu Jan 23 11:09:38 ART 2014
|
||||
##### v2.0 - Thu Jan 23 11:09:38 ART 2014
|
||||
|
||||
- DNSResourceRecord was removed. Now DNS records are added using
|
||||
- DNSResourceRecord was removed. Now DNS records are added using
|
||||
DNS::Resource.
|
||||
|
||||
- tins.h now includes ppi.h.
|
||||
@@ -544,21 +626,21 @@ DNS::Resource.
|
||||
|
||||
- Added PDUOption<>::to<> which converts a PDUOption to a specific type.
|
||||
|
||||
- Layer 3 packets sent using PacketSender::send_recv for which the
|
||||
- Layer 3 packets sent using PacketSender::send_recv for which the
|
||||
answer is a different PDU type.
|
||||
|
||||
- ICMP::gateway now uses IPv4Address.
|
||||
|
||||
- Added support for ICMP address mask request/reply.
|
||||
|
||||
- Fixed bug in PacketSender when using send_recv and a layer 2 PDU. The
|
||||
interface in which the packet was sent was not the default_interface
|
||||
- Fixed bug in PacketSender when using send_recv and a layer 2 PDU. The
|
||||
interface in which the packet was sent was not the default_interface
|
||||
set when the sender was constructed.
|
||||
|
||||
- IP packets sent using PacketSender::send_recv now match ICMP
|
||||
- IP packets sent using PacketSender::send_recv now match ICMP
|
||||
responses.
|
||||
|
||||
- Added support for ICMP timestamp request/reply packets.
|
||||
- Added support for ICMP timestamp request/reply packets.
|
||||
ICMP::matches_response now works with these types of packets as well.
|
||||
|
||||
- Added support for reassembling of fragmented IP packets via the
|
||||
@@ -566,20 +648,20 @@ IPv4Reassembler class.
|
||||
|
||||
- Fragmented IP packet's inner_pdu PDUs are not decoded now.
|
||||
|
||||
- Added 1000ms as the default read timeout used when calling
|
||||
- Added 1000ms as the default read timeout used when calling
|
||||
pcap_open_live. Added BaseSniffer::set_timeout to modify this parameter.
|
||||
|
||||
- Added the --disable-dot11 configure switch.
|
||||
|
||||
- Added support for IPSec.
|
||||
|
||||
- Fixed bug triggered when ifaddrs::ifa_addr was null in
|
||||
- Fixed bug triggered when ifaddrs::ifa_addr was null in
|
||||
NetworkInterface::addresses.
|
||||
|
||||
- Added another overload of Utils::route_entries which returns the
|
||||
result either than storing it in a parameter.
|
||||
|
||||
- Added ARP monitor, WPS detector, DNS queries sniffer and DNS spoofer
|
||||
- Added ARP monitor, WPS detector, DNS queries sniffer and DNS spoofer
|
||||
examples.
|
||||
|
||||
- Added another Sniffer constructor which doesn't expect the maximum
|
||||
@@ -598,10 +680,10 @@ doesn't work well on some linux distributions.
|
||||
|
||||
- Added PPI PDU class.
|
||||
|
||||
- Fixed a bug in EthernetII triggered when the size of the whole frame
|
||||
- Fixed a bug in EthernetII triggered when the size of the whole frame
|
||||
was lower than 60 bytes.
|
||||
|
||||
- Added AddressRange class and IPv4Address, IPv6Address and
|
||||
- Added AddressRange class and IPv4Address, IPv6Address and
|
||||
HWAddress<>::operator/.
|
||||
|
||||
- Added is_broadcast, is_multicast and is_unicast to IPv4, IPv6
|
||||
@@ -611,7 +693,7 @@ and HWAddress.
|
||||
|
||||
- Done some optimizations on TCP's constructor from buffer.
|
||||
|
||||
- Added helper functions to Dot11Data to retrieve the source,
|
||||
- Added helper functions to Dot11Data to retrieve the source,
|
||||
destination and BSSID addresses.
|
||||
|
||||
- Fixed bugs in DNS triggered when parsing MX and unknown records.
|
||||
@@ -631,7 +713,7 @@ types.
|
||||
|
||||
- Added WPA2Decrypter class.
|
||||
|
||||
- IEEE 802.11 frames are not parsed if the RadioTap FAILED_FCS flag
|
||||
- IEEE 802.11 frames are not parsed if the RadioTap FAILED_FCS flag
|
||||
is on.
|
||||
|
||||
- RadioTap now calculates its size everytime it's serialized.
|
||||
@@ -648,13 +730,13 @@ v1.0 - Tue Apr 23 20:40:57 ART 2013
|
||||
- Link layer protocol PDUs now don't hold a NetworkInterface. This led
|
||||
to changes in their constructors.
|
||||
|
||||
- Removed the obsolete PDU* parameter taken by several classes'
|
||||
- Removed the obsolete PDU* parameter taken by several classes'
|
||||
constructors.
|
||||
|
||||
- IP now sets the sender's address automatically when no link layer
|
||||
PDU is used.
|
||||
|
||||
- IP, TCP and UDP now calculate the checksum everytime they're
|
||||
- IP, TCP and UDP now calculate the checksum everytime they're
|
||||
serialized.
|
||||
|
||||
- Added PDU::rfind_pdu.
|
||||
@@ -665,7 +747,7 @@ serialized.
|
||||
|
||||
- PacketSender is now movable.
|
||||
|
||||
- Added an overload of add_option that takes an rvalue-reference in IP,
|
||||
- Added an overload of add_option that takes an rvalue-reference in IP,
|
||||
TCP, DHCP, ICMPv6 and Dot11.
|
||||
|
||||
- Added support for GNU/kFreeBSD.
|
||||
@@ -708,7 +790,7 @@ pseudo protocol.
|
||||
|
||||
##### v0.2 - Sat Oct 20 11:26:40 2012
|
||||
|
||||
- Added support for big endian architectures.
|
||||
- Added support for big endian architectures.
|
||||
|
||||
- Simplified several interfaces.
|
||||
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
CMAKE_MINIMUM_REQUIRED(VERSION 2.8.1)
|
||||
CMAKE_MINIMUM_REQUIRED(VERSION 3.10)
|
||||
PROJECT(libtins)
|
||||
|
||||
OPTION(LIBTINS_BUILD_EXAMPLES "Build examples" ON)
|
||||
@@ -46,7 +46,7 @@ ENDIF(LIBTINS_BUILD_SHARED)
|
||||
|
||||
# The version number.
|
||||
SET(TINS_VERSION_MAJOR 4)
|
||||
SET(TINS_VERSION_MINOR 1)
|
||||
SET(TINS_VERSION_MINOR 6)
|
||||
SET(TINS_VERSION_PATCH 0)
|
||||
SET(LIBTINS_VERSION "${TINS_VERSION_MAJOR}.${TINS_VERSION_MINOR}")
|
||||
|
||||
@@ -234,6 +234,9 @@ CONFIGURE_FILE(
|
||||
IF (NOT CMAKE_INSTALL_LIBDIR)
|
||||
SET(CMAKE_INSTALL_LIBDIR lib)
|
||||
ENDIF()
|
||||
IF (NOT CMAKE_INSTALL_BINDIR)
|
||||
SET(CMAKE_INSTALL_BINDIR bin)
|
||||
ENDIF()
|
||||
# The library output directory
|
||||
SET(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${PROJECT_BINARY_DIR}/${CMAKE_INSTALL_LIBDIR})
|
||||
SET(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${PROJECT_BINARY_DIR}/${CMAKE_INSTALL_LIBDIR})
|
||||
@@ -300,7 +303,13 @@ IF(LIBTINS_BUILD_TESTS)
|
||||
ENABLE_TESTING()
|
||||
ADD_SUBDIRECTORY(tests)
|
||||
ELSE()
|
||||
MESSAGE(STATUS "googletest git submodule is absent. Run `git submodule init && git submodule update` to get it")
|
||||
FIND_PACKAGE(GTest QUIET)
|
||||
IF(${GTest_FOUND})
|
||||
ENABLE_TESTING()
|
||||
ADD_SUBDIRECTORY(tests)
|
||||
ELSE()
|
||||
MESSAGE(STATUS "googletest git submodule is absent. Run `git submodule init && git submodule update` to get it")
|
||||
ENDIF()
|
||||
ENDIF()
|
||||
ENDIF()
|
||||
|
||||
@@ -308,6 +317,12 @@ ENDIF()
|
||||
# CMake project configuration export
|
||||
# **********************************
|
||||
|
||||
if(UNIX)
|
||||
set(CONF_CMAKE_INSTALL_DIR "${CMAKE_INSTALL_LIBDIR}/cmake/libtins")
|
||||
else()
|
||||
set(CONF_CMAKE_INSTALL_DIR CMake)
|
||||
endif()
|
||||
|
||||
# Add all targets to the build-tree export set
|
||||
EXPORT(
|
||||
TARGETS tins
|
||||
@@ -334,13 +349,13 @@ INSTALL(
|
||||
FILES
|
||||
"${PROJECT_BINARY_DIR}/libtinsConfig.cmake"
|
||||
"${PROJECT_BINARY_DIR}/libtinsConfigVersion.cmake"
|
||||
DESTINATION CMake
|
||||
DESTINATION ${CONF_CMAKE_INSTALL_DIR}
|
||||
COMPONENT dev
|
||||
)
|
||||
|
||||
# Install the export set for use with the install-tree
|
||||
INSTALL(
|
||||
EXPORT libtinsTargets
|
||||
DESTINATION CMake
|
||||
DESTINATION ${CONF_CMAKE_INSTALL_DIR}
|
||||
COMPONENT dev
|
||||
)
|
||||
|
||||
@@ -50,7 +50,7 @@ if (NOT CMAKE_CXX_COMPILER_LOADED)
|
||||
message(FATAL_ERROR "CheckCXX11Features modules only works if language CXX is enabled")
|
||||
endif ()
|
||||
|
||||
cmake_minimum_required(VERSION 2.8.3)
|
||||
cmake_minimum_required(VERSION 3.10)
|
||||
|
||||
#
|
||||
### Check for needed compiler flags
|
||||
@@ -79,14 +79,14 @@ function(cxx11_check_feature FEATURE_NAME RESULT_VAR)
|
||||
set(_SRCFILE_FAIL "${_SRCFILE_BASE}_fail.cpp")
|
||||
set(_SRCFILE_FAIL_COMPILE "${_SRCFILE_BASE}_fail_compile.cpp")
|
||||
|
||||
if (CROSS_COMPILING)
|
||||
if (CMAKE_CROSSCOMPILING)
|
||||
try_compile(${RESULT_VAR} "${_bindir}" "${_SRCFILE}"
|
||||
COMPILE_DEFINITIONS "${CXX11_COMPILER_FLAGS}")
|
||||
if (${RESULT_VAR} AND EXISTS ${_SRCFILE_FAIL})
|
||||
try_compile(${RESULT_VAR} "${_bindir}_fail" "${_SRCFILE_FAIL}"
|
||||
COMPILE_DEFINITIONS "${CXX11_COMPILER_FLAGS}")
|
||||
endif (${RESULT_VAR} AND EXISTS ${_SRCFILE_FAIL})
|
||||
else (CROSS_COMPILING)
|
||||
else (CMAKE_CROSSCOMPILING)
|
||||
try_run(_RUN_RESULT_VAR _COMPILE_RESULT_VAR
|
||||
"${_bindir}" "${_SRCFILE}"
|
||||
COMPILE_DEFINITIONS "${CXX11_COMPILER_FLAGS}")
|
||||
@@ -105,7 +105,7 @@ function(cxx11_check_feature FEATURE_NAME RESULT_VAR)
|
||||
set(${RESULT_VAR} FALSE)
|
||||
endif (_COMPILE_RESULT_VAR AND _RUN_RESULT_VAR)
|
||||
endif (${RESULT_VAR} AND EXISTS ${_SRCFILE_FAIL})
|
||||
endif (CROSS_COMPILING)
|
||||
endif (CMAKE_CROSSCOMPILING)
|
||||
if (${RESULT_VAR} AND EXISTS ${_SRCFILE_FAIL_COMPILE})
|
||||
try_compile(_TMP_RESULT "${_bindir}_fail_compile" "${_SRCFILE_FAIL_COMPILE}"
|
||||
COMPILE_DEFINITIONS "${CXX11_COMPILER_FLAGS}")
|
||||
|
||||
@@ -116,7 +116,7 @@ bool Scanner::callback(PDU& pdu) {
|
||||
cout << "Port: " << setw(5) << tcp.sport() << " closed\n";
|
||||
}
|
||||
// Is SYN flag on? Then port is open!
|
||||
else if(tcp.flags() == (TCP::SYN | TCP::ACK)) {
|
||||
else if(tcp.has_flags(TCP::SYN | TCP::ACK)) {
|
||||
cout << "Port: " << setw(5) << tcp.sport() << " open\n";
|
||||
}
|
||||
}
|
||||
|
||||
@@ -68,7 +68,7 @@ private:
|
||||
const IP& ip = pdu.rfind_pdu<IP>();
|
||||
const TCP& tcp = pdu.rfind_pdu<TCP>();
|
||||
// We'll only close a connection when seeing a SYN|ACK
|
||||
if (tcp.flags() == (TCP::SYN | TCP::ACK)) {
|
||||
if (tcp.has_flags(TCP::SYN | TCP::ACK)) {
|
||||
// Create an ethernet header flipping the addresses
|
||||
EthernetII packet(eth.src_addr(), eth.dst_addr());
|
||||
// Do the same for IP
|
||||
|
||||
@@ -40,9 +40,13 @@ namespace Tins {
|
||||
* \brief AddressRange iterator class.
|
||||
*/
|
||||
template<typename Address>
|
||||
class AddressRangeIterator : public std::iterator<std::forward_iterator_tag, const Address> {
|
||||
class AddressRangeIterator {
|
||||
public:
|
||||
typedef typename std::iterator<std::forward_iterator_tag, const Address>::value_type value_type;
|
||||
typedef std::forward_iterator_tag iterator_category;
|
||||
typedef const Address value_type;
|
||||
typedef std::ptrdiff_t difference_type;
|
||||
typedef const Address* pointer;
|
||||
typedef const Address& reference;
|
||||
|
||||
struct end_iterator {
|
||||
|
||||
|
||||
@@ -41,6 +41,7 @@ class Dot3;
|
||||
class SLL;
|
||||
class Loopback;
|
||||
class PPI;
|
||||
class IP;
|
||||
|
||||
/**
|
||||
* \struct DataLinkType
|
||||
@@ -73,6 +74,7 @@ TINS_MAKE_DATA_LINK_TYPE(Loopback, DLT_LOOP)
|
||||
TINS_MAKE_DATA_LINK_TYPE(PPI, DLT_PPI)
|
||||
TINS_MAKE_DATA_LINK_TYPE(Dot11, DLT_IEEE802_11)
|
||||
TINS_MAKE_DATA_LINK_TYPE(RadioTap, DLT_IEEE802_11_RADIO)
|
||||
TINS_MAKE_DATA_LINK_TYPE(IP, DLT_RAW)
|
||||
|
||||
#undef TINS_MAKE_DATA_LINK_TYPE
|
||||
|
||||
|
||||
@@ -652,7 +652,7 @@ public:
|
||||
// Getters
|
||||
|
||||
/**
|
||||
* \brief Setter for the id field.
|
||||
* \brief Getter for the id field.
|
||||
*
|
||||
* \return uint16_t containing the value of the id field.
|
||||
*/
|
||||
@@ -661,7 +661,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the query response field.
|
||||
* \brief Getter for the query response field.
|
||||
*
|
||||
* \return QRType containing the value of the query response
|
||||
* field.
|
||||
@@ -671,7 +671,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the opcode field.
|
||||
* \brief Getter for the opcode field.
|
||||
*
|
||||
* \return uint8_t containing the value of the opcode field.
|
||||
*/
|
||||
@@ -680,7 +680,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the authoritative answer field.
|
||||
* \brief Getter for the authoritative answer field.
|
||||
*
|
||||
* \return uint8_t containing the value of the authoritative
|
||||
* answer field.
|
||||
@@ -690,7 +690,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the truncated field.
|
||||
* \brief Getter for the truncated field.
|
||||
*
|
||||
* \return uint8_t containing the value of the truncated field.
|
||||
*/
|
||||
@@ -699,7 +699,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the recursion desired field.
|
||||
* \brief Getter for the recursion desired field.
|
||||
*
|
||||
* \return uint8_t containing the value of the recursion
|
||||
* desired field.
|
||||
@@ -709,7 +709,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the recursion available field.
|
||||
* \brief Getter for the recursion available field.
|
||||
*
|
||||
* \return uint8_t containing the value of the recursion
|
||||
* available field.
|
||||
@@ -719,7 +719,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the z desired field.
|
||||
* \brief Getter for the z desired field.
|
||||
*
|
||||
* \return uint8_t containing the value of the z field.
|
||||
*/
|
||||
@@ -728,7 +728,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the authenticated data field.
|
||||
* \brief Getter for the authenticated data field.
|
||||
*
|
||||
* \return uint8_t containing the value of the authenticated
|
||||
* data field.
|
||||
@@ -738,7 +738,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the checking disabled field.
|
||||
* \brief Getter for the checking disabled field.
|
||||
*
|
||||
* \return uint8_t containing the value of the checking
|
||||
* disabled field.
|
||||
@@ -748,7 +748,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the rcode field.
|
||||
* \brief Getter for the rcode field.
|
||||
*
|
||||
* \return uint8_t containing the value of the rcode field.
|
||||
*/
|
||||
@@ -757,7 +757,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the questions field.
|
||||
* \brief Getter for the questions field.
|
||||
*
|
||||
* \return uint16_t containing the value of the questions field.
|
||||
*/
|
||||
@@ -766,7 +766,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the answers field.
|
||||
* \brief Getter for the answers field.
|
||||
*
|
||||
* \return uint16_t containing the value of the answers field.
|
||||
*/
|
||||
@@ -775,7 +775,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the authority field.
|
||||
* \brief Getter for the authority field.
|
||||
*
|
||||
* \return uint16_t containing the value of the authority field.
|
||||
*/
|
||||
@@ -784,7 +784,7 @@ public:
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the additional field.
|
||||
* \brief Getter for the additional field.
|
||||
*
|
||||
* \return uint16_t containing the value of the additional field.
|
||||
*/
|
||||
@@ -1034,7 +1034,8 @@ private:
|
||||
uint32_t compose_name(const uint8_t* ptr, char* out_ptr) const;
|
||||
void convert_records(const uint8_t* ptr,
|
||||
const uint8_t* end,
|
||||
resources_type& res) const;
|
||||
resources_type& res,
|
||||
const uint16_t rr_count) const;
|
||||
void skip_to_section_end(Memory::InputMemoryStream& stream,
|
||||
const uint32_t num_records) const;
|
||||
void skip_to_dname_end(Memory::InputMemoryStream& stream) const;
|
||||
|
||||
@@ -127,9 +127,148 @@ public:
|
||||
ERP_INFORMATION,
|
||||
TS_DELAY,
|
||||
TCLAS_PROCESSING,
|
||||
QOS_CAPABILITY = 46,
|
||||
HT_CAPABILITY,
|
||||
QOS_CAPABILITY,
|
||||
RSN = 48,
|
||||
EXT_SUPPORTED_RATES = 50,
|
||||
AP_CHANNEL_REPORT,
|
||||
NEIGHBOR_REPORT,
|
||||
RCPI,
|
||||
MOBILITY_DOMAIN_MDE,
|
||||
FAST_BSS_TRANSITION_FTE,
|
||||
TIMEOUT_INTERVAL,
|
||||
RIC_DATA_RDE,
|
||||
DSE_REG_LOC,
|
||||
SUPPORTED_OP_CLASSES,
|
||||
EXT_CH_SWITCH_ANNOUNCEMENT,
|
||||
HT_OPERATION,
|
||||
SEC_CH_OFFSET,
|
||||
BSS_AVG_ACCESS_DELAY,
|
||||
ANTENNA,
|
||||
RSNI,
|
||||
MEASUREMENT_PILOT_TRANSMISSION,
|
||||
BSS_AVAIL_ADMISSION_CAPACITY,
|
||||
BSS_AC_ACCESS_DELAY,
|
||||
TIME_ADVERTISEMENT,
|
||||
RM_ENABLED_CAP,
|
||||
MULTIPLE_BSSID,
|
||||
BSS_2040_COEX,
|
||||
BSS_2040_INTOLERANT_CH_REPORT,
|
||||
OVERLAPPING_BSS_SCAN_PARAM,
|
||||
RIC_DESCRIPTOR,
|
||||
MGMT_MIC,
|
||||
EVENT_REQ = 78,
|
||||
EVENT_REPORT,
|
||||
DIAG_REQ,
|
||||
DIAG_REPORT,
|
||||
LOCATION_PARAMS,
|
||||
NONTRANSMITTED_BSSID_CAP,
|
||||
SSID_LIST,
|
||||
MULTIPLE_BSSID_INDEX,
|
||||
FMS_DESCRIPTOR,
|
||||
FMS_REQ,
|
||||
FMS_RESP,
|
||||
QOS_TRAFFIC_CAP,
|
||||
BSS_MAX_IDLE_PERIOD,
|
||||
TFS_REQ,
|
||||
TFS_RESP,
|
||||
WNM_SLEEP_MODE,
|
||||
TIM_BROADCAST_REQ,
|
||||
TIM_BROADCAST_RESP,
|
||||
COLLOCATED_INTERFERENCE_REPORT,
|
||||
CH_USAGE,
|
||||
TIME_ZONE,
|
||||
DMS_REQ,
|
||||
DMS_RESP,
|
||||
LINK_ID,
|
||||
WAKEUP_SCHEDULE,
|
||||
CH_SWITCH_TIMING,
|
||||
PTI_CONTROL,
|
||||
TPU_BUFFER_STATUS,
|
||||
INTERWORKING,
|
||||
ADVERTISEMENT_PROTOCOL,
|
||||
EXPEDITED_BANDWIDTH_REQ,
|
||||
QOS_MAP,
|
||||
ROAMING_CONSORTIUM,
|
||||
EMERG_ALERT_ID,
|
||||
MESH_CONFIG,
|
||||
MESH_ID,
|
||||
MESH_LINK_METRIC_REPORT,
|
||||
CONGESTION_NOTIFICATION,
|
||||
MESH_PEERING_MGMT,
|
||||
MESH_CH_SWITCH_PARAMS,
|
||||
MESH_AWAKE_WINDOW,
|
||||
BEACON_TIMING,
|
||||
MCCAOP_SETUP_REQ,
|
||||
MCCAOP_SETUP_REPLY,
|
||||
MCCAOP_ADVERTISEMENT,
|
||||
MCCAOP_TEARDOWN,
|
||||
GANN,
|
||||
RANN,
|
||||
EXT_CAP,
|
||||
PREQ = 130,
|
||||
PREP,
|
||||
PERR,
|
||||
PXU = 137,
|
||||
PXUC,
|
||||
AUTH_MESH_PEER_EX,
|
||||
MIC,
|
||||
DEST_URI,
|
||||
UAPSD_COEX,
|
||||
DMG_WAKEUP_SCHEDULE,
|
||||
EXT_SCHEDULE,
|
||||
STA_AVAIL,
|
||||
DMG_TSPEC,
|
||||
NEXT_DMG_ATI,
|
||||
DMG_CAP,
|
||||
DMG_OP = 151,
|
||||
DMG_BSS_PARAM_CHG,
|
||||
DMG_BEAM_REFINEMENT,
|
||||
CH_MEASUREMENT_FEEDBACK,
|
||||
AWAKE_WINDOW = 157,
|
||||
MULTIBAND,
|
||||
ADDBA_ESXT,
|
||||
NEXTPCP_LIST,
|
||||
PCP_HANDOVER,
|
||||
DMG_LINK_MARGIN,
|
||||
SWITCHING_STREAM,
|
||||
SESSION_TRANSITION,
|
||||
DYNAMIC_TONE_PAIRING_REPORT,
|
||||
CLUSTER_REPORT,
|
||||
RELAY_CAP,
|
||||
RELAY_TRANSFER_PARAM_SET,
|
||||
BEAMLINK_MAINTENANCE,
|
||||
MULTIPLE_MAC_SUBLAYERS,
|
||||
UPID,
|
||||
DMG_LINK_ADAPTATION_ACK,
|
||||
MCCAOP_ADV_OVERVIEW = 174,
|
||||
QUIET_PERIOD_REQ,
|
||||
QUIET_PERIOD_RESP = 177,
|
||||
QMF_POLICY = 181,
|
||||
ECAPC_POLICY = 182,
|
||||
CLUSTER_TIME_OFFSET,
|
||||
INTRA_ACCESS_CAT_PRIORITY,
|
||||
SCS_DESCRIPTOR,
|
||||
QLOAD_REPORT,
|
||||
HCCA_TXOP_UPDATE_COUNT,
|
||||
HIGHER_LAYER_STREAM_ID,
|
||||
GCR_GROUP_ADDR,
|
||||
ANTENNA_SECTOR_ID_PATTERN,
|
||||
VHT_CAP,
|
||||
VHT_OP,
|
||||
EXT_BSS_LOAD,
|
||||
WIDE_BANDWIDTH_CH_SWITCH,
|
||||
TRANSMIT_POWER_ENVELOPE,
|
||||
CH_SWITCH_WRAPPER,
|
||||
AID,
|
||||
QUIET_CHANNEL,
|
||||
OP_MODE_NOTIFY,
|
||||
UPSIM,
|
||||
REDUCED_NEIGHBOR_REPORT,
|
||||
TVHT_OP,
|
||||
DEV_LOC = 204,
|
||||
WHITE_SPACE_MAP,
|
||||
FINE_TUNING_MEASUREMENT_PARAMS,
|
||||
VENDOR_SPECIFIC = 221
|
||||
};
|
||||
|
||||
|
||||
@@ -5,14 +5,14 @@
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are
|
||||
* met:
|
||||
*
|
||||
*
|
||||
* * Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* * Redistributions in binary form must reproduce the above
|
||||
* copyright notice, this list of conditions and the following disclaimer
|
||||
* in the documentation and/or other materials provided with the
|
||||
* distribution.
|
||||
*
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
@@ -46,11 +46,11 @@ public:
|
||||
* \brief This PDU's flag.
|
||||
*/
|
||||
static const PDU::PDUType pdu_flag = PDU::DOT11_CONTROL;
|
||||
|
||||
|
||||
/**
|
||||
* \brief Constructor for creating a 802.11 control frame PDU
|
||||
*
|
||||
* Constructs a 802.11 Control PDU taking the destination and
|
||||
* Constructs a 802.11 Control PDU taking the destination and
|
||||
* source hardware addresses.
|
||||
*
|
||||
* \param dst_addr The destination hardware address.
|
||||
@@ -59,15 +59,15 @@ public:
|
||||
|
||||
/**
|
||||
* \brief Constructs a Dot11Control object from a buffer and
|
||||
* adds all identifiable PDUs found in the buffer as children
|
||||
* adds all identifiable PDUs found in the buffer as children
|
||||
* of this one.
|
||||
*
|
||||
*
|
||||
* If the next PDU is not recognized, then a RawPDU is used.
|
||||
*
|
||||
*
|
||||
* If there is not enough size for the header in the buffer
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* is thrown.
|
||||
*
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
@@ -97,6 +97,11 @@ public:
|
||||
*/
|
||||
class TINS_API Dot11ControlTA : public Dot11Control {
|
||||
public:
|
||||
/**
|
||||
* \brief This PDU's flag.
|
||||
*/
|
||||
static const PDU::PDUType pdu_flag = PDU::DOT11_CONTROL_TA;
|
||||
|
||||
/**
|
||||
* \brief Getter for the target address field.
|
||||
*/
|
||||
@@ -109,30 +114,39 @@ public:
|
||||
* \param addr The new target address.
|
||||
*/
|
||||
void target_addr(const address_type& addr);
|
||||
|
||||
/**
|
||||
* \brief Check whether this PDU matches the specified flag.
|
||||
* \param flag The flag to match
|
||||
* \sa PDU::matches_flag
|
||||
*/
|
||||
bool matches_flag(PDUType flag) const {
|
||||
return flag == pdu_flag || Dot11::matches_flag(flag);
|
||||
}
|
||||
protected:
|
||||
/**
|
||||
* \brief Constructor for creating a 802.11 control frame TA PDU
|
||||
*
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* hardware addresses.
|
||||
*
|
||||
* \param dst_addr The destination hardware address.
|
||||
* \param target_addr The source hardware address.
|
||||
*/
|
||||
Dot11ControlTA(const address_type& dst_addr = address_type(),
|
||||
Dot11ControlTA(const address_type& dst_addr = address_type(),
|
||||
const address_type& target_addr = address_type());
|
||||
|
||||
/**
|
||||
* \brief Constructs a Dot11ControlTA object from a buffer and
|
||||
* adds all identifiable PDUs found in the buffer as children
|
||||
* adds all identifiable PDUs found in the buffer as children
|
||||
* of this one.
|
||||
*
|
||||
*
|
||||
* If the next PDU is not recognized, then a RawPDU is used.
|
||||
*
|
||||
*
|
||||
* If there is not enough size for the header in the buffer
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* is thrown.
|
||||
*
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
@@ -149,8 +163,8 @@ protected:
|
||||
/**
|
||||
* \brief Getter for the control ta additional fields size.
|
||||
*/
|
||||
uint32_t controlta_size() const {
|
||||
return static_cast<uint32_t>(taddr_.size() + sizeof(dot11_header));
|
||||
uint32_t controlta_size() const {
|
||||
return static_cast<uint32_t>(taddr_.size() + sizeof(dot11_header));
|
||||
}
|
||||
|
||||
void write_ext_header(Memory::OutputMemoryStream& stream);
|
||||
@@ -178,19 +192,19 @@ public:
|
||||
* \param dst_addr The destination hardware address.
|
||||
* \param target_addr The source hardware address.
|
||||
*/
|
||||
Dot11RTS(const address_type& dst_addr = address_type(),
|
||||
Dot11RTS(const address_type& dst_addr = address_type(),
|
||||
const address_type& target_addr = address_type());
|
||||
|
||||
|
||||
/**
|
||||
* \brief Constructs a Dot11RTS object from a buffer and adds all
|
||||
* \brief Constructs a Dot11RTS object from a buffer and adds all
|
||||
* identifiable PDUs found in the buffer as children of this one.
|
||||
*
|
||||
*
|
||||
* If the next PDU is not recognized, then a RawPDU is used.
|
||||
*
|
||||
*
|
||||
* If there is not enough size for the header in the buffer
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* is thrown.
|
||||
*
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
@@ -233,26 +247,26 @@ public:
|
||||
/**
|
||||
* \brief Constructor for creating a 802.11 PS-Poll frame PDU
|
||||
*
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* hardware addresses.
|
||||
*
|
||||
* \param dst_addr The destination hardware address.
|
||||
* \param target_addr The source hardware address.
|
||||
*/
|
||||
Dot11PSPoll(const address_type& dst_addr = address_type(),
|
||||
Dot11PSPoll(const address_type& dst_addr = address_type(),
|
||||
const address_type& target_addr = address_type());
|
||||
|
||||
/**
|
||||
* \brief Constructs a Dot11PSPoll object from a buffer and
|
||||
* adds all identifiable PDUs found in the buffer as children of
|
||||
* adds all identifiable PDUs found in the buffer as children of
|
||||
* this one.
|
||||
*
|
||||
*
|
||||
* If the next PDU is not recognized, then a RawPDU is used.
|
||||
*
|
||||
*
|
||||
* If there is not enough size for the header in the buffer
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* is thrown.
|
||||
*
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
@@ -295,26 +309,26 @@ public:
|
||||
/**
|
||||
* \brief Constructor for creating a 802.11 CF-End frame PDU
|
||||
*
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* hardware addresses.
|
||||
*
|
||||
* \param dst_addr The destination hardware address.
|
||||
* \param target_addr The source hardware address.
|
||||
*/
|
||||
Dot11CFEnd(const address_type& dst_addr = address_type(),
|
||||
Dot11CFEnd(const address_type& dst_addr = address_type(),
|
||||
const address_type& target_addr = address_type());
|
||||
|
||||
|
||||
/**
|
||||
* \brief Constructs a Dot11CFEnd object from a buffer and adds
|
||||
* all identifiable PDUs found in the buffer as children of this
|
||||
* \brief Constructs a Dot11CFEnd object from a buffer and adds
|
||||
* all identifiable PDUs found in the buffer as children of this
|
||||
* one.
|
||||
*
|
||||
*
|
||||
* If the next PDU is not recognized, then a RawPDU is used.
|
||||
*
|
||||
*
|
||||
* If there is not enough size for the header in the buffer
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* is thrown.
|
||||
*
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
@@ -357,26 +371,26 @@ public:
|
||||
/**
|
||||
* \brief Constructor for creating a 802.11 End-CF-Ack frame PDU
|
||||
*
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* hardware addresses.
|
||||
*
|
||||
*
|
||||
* \param dst_addr The destination hardware address.
|
||||
* \param target_addr The source hardware address.
|
||||
*/
|
||||
Dot11EndCFAck(const address_type& dst_addr = address_type(),
|
||||
Dot11EndCFAck(const address_type& dst_addr = address_type(),
|
||||
const address_type& target_addr = address_type());
|
||||
|
||||
/**
|
||||
* \brief Constructs a Dot11EndCFAck frame object from a buffer
|
||||
* and adds all identifiable PDUs found in the buffer as children
|
||||
* \brief Constructs a Dot11EndCFAck frame object from a buffer
|
||||
* and adds all identifiable PDUs found in the buffer as children
|
||||
* of this one.
|
||||
*
|
||||
*
|
||||
* If the next PDU is not recognized, then a RawPDU is used.
|
||||
*
|
||||
*
|
||||
* If there is not enough size for the header in the buffer
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* is thrown.
|
||||
*
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
@@ -419,7 +433,7 @@ public:
|
||||
/**
|
||||
* \brief Constructor for creating a 802.11 Ack frame PDU
|
||||
*
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* hardware addresses.
|
||||
*
|
||||
* \param dst_addr The destination hardware address.
|
||||
@@ -428,15 +442,15 @@ public:
|
||||
|
||||
/**
|
||||
* \brief Constructs a Dot11Ack frame object from a buffer and
|
||||
* adds all identifiable PDUs found in the buffer as children of
|
||||
* adds all identifiable PDUs found in the buffer as children of
|
||||
* this one.
|
||||
*
|
||||
*
|
||||
* If the next PDU is not recognized, then a RawPDU is used.
|
||||
*
|
||||
*
|
||||
* If there is not enough size for the header in the buffer
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* is thrown.
|
||||
*
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
@@ -482,26 +496,26 @@ public:
|
||||
/**
|
||||
* \brief Constructor for creating a 802.11 Block Ack request frame PDU
|
||||
*
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* hardware addresses.
|
||||
*
|
||||
*
|
||||
* \param dst_addr The destination hardware address.
|
||||
* \param target_addr The source hardware address.
|
||||
*/
|
||||
Dot11BlockAckRequest(const address_type& dst_addr = address_type(),
|
||||
Dot11BlockAckRequest(const address_type& dst_addr = address_type(),
|
||||
const address_type& target_addr = address_type());
|
||||
|
||||
/**
|
||||
* \brief Constructs a Dot11BlockAckRequest object from a buffer
|
||||
* and adds all identifiable PDUs found in the buffer as children
|
||||
* \brief Constructs a Dot11BlockAckRequest object from a buffer
|
||||
* and adds all identifiable PDUs found in the buffer as children
|
||||
* of this one.
|
||||
*
|
||||
*
|
||||
* If the next PDU is not recognized, then a RawPDU is used.
|
||||
*
|
||||
*
|
||||
* If there is not enough size for the header in the buffer
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* is thrown.
|
||||
*
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
@@ -513,11 +527,11 @@ public:
|
||||
* \brief Getter for the bar control field.
|
||||
* \return The stored bar control field.
|
||||
*/
|
||||
small_uint<4> bar_control() const {
|
||||
small_uint<4> bar_control() const {
|
||||
#if TINS_IS_LITTLE_ENDIAN
|
||||
return bar_control_ & 0xf;
|
||||
return bar_control_ & 0xf;
|
||||
#else
|
||||
return (bar_control_ >> 8) & 0xf;
|
||||
return (bar_control_ >> 8) & 0xf;
|
||||
#endif
|
||||
}
|
||||
|
||||
@@ -525,26 +539,26 @@ public:
|
||||
* \brief Getter for the start sequence field.
|
||||
* \return The stored start sequence.
|
||||
*/
|
||||
small_uint<12> start_sequence() const {
|
||||
small_uint<12> start_sequence() const {
|
||||
#if TINS_IS_LITTLE_ENDIAN
|
||||
return (start_sequence_ >> 4) & 0xfff;
|
||||
return (start_sequence_ >> 4) & 0xfff;
|
||||
#else
|
||||
return (Endian::le_to_host<uint16_t>(start_sequence_) >> 4) & 0xfff;
|
||||
return (Endian::le_to_host<uint16_t>(start_sequence_) >> 4) & 0xfff;
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* \brief Getter for the fragment number field.
|
||||
* \return The stored fragment number field.
|
||||
*/
|
||||
small_uint<4> fragment_number() const {
|
||||
small_uint<4> fragment_number() const {
|
||||
#if TINS_IS_LITTLE_ENDIAN
|
||||
return start_sequence_ & 0xf;
|
||||
return start_sequence_ & 0xf;
|
||||
#else
|
||||
return (start_sequence_ >> 8) & 0xf;
|
||||
return (start_sequence_ >> 8) & 0xf;
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* \brief Returns the 802.11 frame's header length.
|
||||
*
|
||||
@@ -566,7 +580,7 @@ public:
|
||||
* \param bar The start sequence field to be set.
|
||||
*/
|
||||
void start_sequence(small_uint<12> seq);
|
||||
|
||||
|
||||
/**
|
||||
* \brief Setter for the fragment number field.
|
||||
* \param frag The fragment number field to be set.
|
||||
@@ -614,7 +628,7 @@ public:
|
||||
* \brief This PDU's flag.
|
||||
*/
|
||||
static const PDU::PDUType pdu_flag = PDU::DOT11_BLOCK_ACK;
|
||||
|
||||
|
||||
/**
|
||||
* The size of the bitmap field.
|
||||
*/
|
||||
@@ -623,26 +637,26 @@ public:
|
||||
/**
|
||||
* \brief Constructor for creating a 802.11 Block Ack frame PDU
|
||||
*
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* Constructs a 802.11 PDU taking the destination and source
|
||||
* hardware addresses.
|
||||
*
|
||||
*
|
||||
* \param dst_addr The destination hardware address.
|
||||
* \param target_addr The source hardware address.
|
||||
*/
|
||||
Dot11BlockAck(const address_type& dst_addr = address_type(),
|
||||
Dot11BlockAck(const address_type& dst_addr = address_type(),
|
||||
const address_type& target_addr = address_type());
|
||||
|
||||
/**
|
||||
* \brief Constructs a Dot11BlockAck frame object from a buffer
|
||||
* and adds all identifiable PDUs found in the buffer as children
|
||||
* \brief Constructs a Dot11BlockAck frame object from a buffer
|
||||
* and adds all identifiable PDUs found in the buffer as children
|
||||
* of this one.
|
||||
*
|
||||
*
|
||||
* If the next PDU is not recognized, then a RawPDU is used.
|
||||
*
|
||||
*
|
||||
* If there is not enough size for the header in the buffer
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* or the input data is malformed, a malformed_packet exception
|
||||
* is thrown.
|
||||
*
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
@@ -654,11 +668,11 @@ public:
|
||||
* \brief Getter for the bar control field.
|
||||
* \return The stored bar control field.
|
||||
*/
|
||||
small_uint<4> bar_control() const {
|
||||
small_uint<4> bar_control() const {
|
||||
#if TINS_IS_LITTLE_ENDIAN
|
||||
return bar_control_ & 0xf;
|
||||
return bar_control_ & 0xf;
|
||||
#else
|
||||
return (bar_control_ >> 8) & 0xf;
|
||||
return (bar_control_ >> 8) & 0xf;
|
||||
#endif
|
||||
}
|
||||
|
||||
@@ -666,26 +680,26 @@ public:
|
||||
* \brief Getter for the start sequence field.
|
||||
* \return The stored start sequence.
|
||||
*/
|
||||
small_uint<12> start_sequence() const {
|
||||
small_uint<12> start_sequence() const {
|
||||
#if TINS_IS_LITTLE_ENDIAN
|
||||
return (start_sequence_ >> 4) & 0xfff;
|
||||
return (start_sequence_ >> 4) & 0xfff;
|
||||
#else
|
||||
return (Endian::le_to_host<uint16_t>(start_sequence_) >> 4) & 0xfff;
|
||||
return (Endian::le_to_host<uint16_t>(start_sequence_) >> 4) & 0xfff;
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* \brief Getter for the fragment number field.
|
||||
* \return The stored fragment number field.
|
||||
*/
|
||||
small_uint<4> fragment_number() const {
|
||||
small_uint<4> fragment_number() const {
|
||||
#if TINS_IS_LITTLE_ENDIAN
|
||||
return start_sequence_ & 0xf;
|
||||
return start_sequence_ & 0xf;
|
||||
#else
|
||||
return (start_sequence_ >> 8) & 0xf;
|
||||
return (start_sequence_ >> 8) & 0xf;
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* \brief Returns the 802.11 frame's header length.
|
||||
*
|
||||
@@ -707,7 +721,7 @@ public:
|
||||
* \param bar The start sequence field to be set.
|
||||
*/
|
||||
void start_sequence(small_uint<12> seq);
|
||||
|
||||
|
||||
/**
|
||||
* \brief Setter for the fragment number field.
|
||||
* \param frag The fragment number field to be set.
|
||||
@@ -716,9 +730,9 @@ public:
|
||||
|
||||
/**
|
||||
* \brief Getter for the bitmap field.
|
||||
*
|
||||
*
|
||||
* The returned pointer <b>must not</b> be free'd.
|
||||
*
|
||||
*
|
||||
* \return The bitmap field.
|
||||
*/
|
||||
const uint8_t* bitmap() const {
|
||||
|
||||
@@ -64,8 +64,26 @@ public:
|
||||
class malformed_packet : public exception_base {
|
||||
public:
|
||||
malformed_packet() : exception_base("Malformed packet") { }
|
||||
malformed_packet(const std::string& message) : exception_base(message) { }
|
||||
};
|
||||
|
||||
/**
|
||||
* \brief Exception thrown when a DNS decompression pointer is out of bounds.
|
||||
*/
|
||||
class dns_decompression_pointer_out_of_bounds : public malformed_packet {
|
||||
public:
|
||||
dns_decompression_pointer_out_of_bounds() : malformed_packet("DNS decompression: pointer out of bounds") { }
|
||||
};
|
||||
|
||||
/**
|
||||
* \brief Exception thrown when a DNS decompression pointer loops.
|
||||
*/
|
||||
class dns_decompression_pointer_loops : public malformed_packet {
|
||||
public:
|
||||
dns_decompression_pointer_loops() : malformed_packet("DNS decompression: pointer loops") { }
|
||||
};
|
||||
|
||||
|
||||
/**
|
||||
* \brief Exception thrown when serializing a packet fails.
|
||||
*/
|
||||
|
||||
@@ -89,7 +89,9 @@ public:
|
||||
INFO_REQUEST = 15,
|
||||
INFO_REPLY = 16,
|
||||
ADDRESS_MASK_REQUEST = 17,
|
||||
ADDRESS_MASK_REPLY = 18
|
||||
ADDRESS_MASK_REPLY = 18,
|
||||
EXTENDED_ECHO_REQUEST = 42,
|
||||
EXTENDED_ECHO_REPLY = 43
|
||||
};
|
||||
|
||||
/**
|
||||
@@ -318,7 +320,7 @@ public:
|
||||
* \return Returns the gateway field value.
|
||||
*/
|
||||
address_type gateway() const {
|
||||
return address_type(Endian::be_to_host(header_.un.gateway));
|
||||
return address_type(header_.un.gateway);
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -381,7 +383,7 @@ public:
|
||||
* \return Returns the address mask value.
|
||||
*/
|
||||
address_type address_mask() const {
|
||||
return address_type(Endian::be_to_host(orig_timestamp_or_address_mask_));
|
||||
return address_type(orig_timestamp_or_address_mask_);
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -30,6 +30,7 @@
|
||||
#ifndef TINS_ICMPV6_H
|
||||
#define TINS_ICMPV6_H
|
||||
|
||||
#include <algorithm>
|
||||
#include <vector>
|
||||
#include <tins/macros.h>
|
||||
#include <tins/pdu.h>
|
||||
@@ -92,7 +93,9 @@ public:
|
||||
MULTICAST_ROUTER_ADVERT = 151,
|
||||
MULTICAST_ROUTER_SOLICIT = 152,
|
||||
MULTICAST_ROUTER_TERMINATE = 153,
|
||||
RPL_CONTROL_MSG = 155
|
||||
RPL_CONTROL_MSG = 155,
|
||||
EXTENDED_ECHO_REQUEST = 160,
|
||||
EXTENDED_ECHO_REPLY = 161
|
||||
};
|
||||
|
||||
/**
|
||||
@@ -173,7 +176,7 @@ public:
|
||||
|
||||
addr_list_type(const addresses_type& addresses = addresses_type())
|
||||
: addresses(addresses) {
|
||||
std::fill(reserved, reserved + sizeof(reserved), 0);
|
||||
std::fill(reserved, reserved + sizeof(reserved), static_cast<uint8_t>(0));
|
||||
}
|
||||
|
||||
static addr_list_type from_option(const option& opt);
|
||||
@@ -199,7 +202,7 @@ public:
|
||||
|
||||
naack_type(uint8_t code = 0, uint8_t status = 0)
|
||||
: code(code), status(status) {
|
||||
std::fill(reserved, reserved + 4, 0);
|
||||
std::fill(reserved, reserved + 4, static_cast<uint8_t>(0));
|
||||
}
|
||||
|
||||
static naack_type from_option(const option& opt);
|
||||
@@ -323,7 +326,7 @@ public:
|
||||
* The key_hash member will be 0-initialized.
|
||||
*/
|
||||
rsa_sign_type() {
|
||||
std::fill(key_hash, key_hash + sizeof(key_hash), 0);
|
||||
std::fill(key_hash, key_hash + sizeof(key_hash), static_cast<uint8_t>(0));
|
||||
}
|
||||
|
||||
static rsa_sign_type from_option(const option& opt);
|
||||
@@ -489,7 +492,7 @@ public:
|
||||
|
||||
timestamp_type(uint64_t timestamp = 0)
|
||||
: timestamp(timestamp) {
|
||||
std::fill(reserved, reserved + sizeof(reserved), 0);
|
||||
std::fill(reserved, reserved + sizeof(reserved), static_cast<uint8_t>(0));
|
||||
}
|
||||
|
||||
static timestamp_type from_option(const option& opt);
|
||||
|
||||
@@ -174,9 +174,9 @@ public:
|
||||
option_identifier(OptionNumber number, OptionClass op_class,
|
||||
small_uint<1> copied)
|
||||
#if TINS_IS_LITTLE_ENDIAN
|
||||
: number(number), op_class(op_class), copied(copied) {}
|
||||
: number(static_cast<uint8_t>(number)), op_class(static_cast<uint8_t>(op_class)), copied(copied) {}
|
||||
#else
|
||||
: copied(copied), op_class(op_class), number(number) {}
|
||||
: copied(copied), op_class(static_cast<uint8_t>(op_class)), number(static_cast<uint8_t>(number)) {}
|
||||
#endif
|
||||
|
||||
/**
|
||||
@@ -281,6 +281,10 @@ public:
|
||||
|
||||
/* Getters */
|
||||
|
||||
uint32_t advertised_size() const {
|
||||
return static_cast<uint32_t>(tot_len());
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Getter for the header length field.
|
||||
*
|
||||
|
||||
@@ -33,7 +33,7 @@
|
||||
#include <string>
|
||||
#include <iosfwd>
|
||||
#include <functional>
|
||||
#include <stdint.h>
|
||||
#include <cstdint>
|
||||
#include <tins/cxxstd.h>
|
||||
#include <tins/macros.h>
|
||||
|
||||
|
||||
@@ -230,6 +230,14 @@ public:
|
||||
*/
|
||||
bool is_multicast() const;
|
||||
|
||||
/**
|
||||
* \brief Return true if this is a Link-Local unicast IPv6 address.
|
||||
*
|
||||
* This method returns true if this address is in the address range
|
||||
* fe80::/10, false otherwise
|
||||
*/
|
||||
bool is_local_unicast() const;
|
||||
|
||||
/**
|
||||
* \brief Returns the size of an IPv6 Address.
|
||||
*
|
||||
|
||||
@@ -236,7 +236,7 @@ public:
|
||||
* \return The LLC frame format.
|
||||
*/
|
||||
uint8_t type() {
|
||||
return type_;
|
||||
return static_cast<uint8_t>(type_);
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -245,7 +245,7 @@ public:
|
||||
* \return The sender send sequence number if format is INFORMATION else 0.
|
||||
*/
|
||||
uint8_t send_seq_number() {
|
||||
return (type() == INFORMATION) ? (control_field.info.send_seq_num) : 0;
|
||||
return static_cast<uint8_t>((type() == INFORMATION) ? (control_field.info.send_seq_num) : 0);
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -65,7 +65,9 @@ class PDU;
|
||||
* - Those that don't contain a link layer PDU. In this case, the
|
||||
* kernel will be responsible for picking the appropriate network interface
|
||||
* based on the destination address.
|
||||
*
|
||||
* - Exception: <a href="https://datatracker.ietf.org/doc/html/rfc2553#section-3.3">RFC2553</a>
|
||||
* requires IPv6 link-scope address have a interface defined.
|
||||
* .
|
||||
* \par Note for Windows users:
|
||||
* Sending layer 3 PDUs (without a link layer protocol) is very restricted
|
||||
* on Windows (<a href="https://msdn.microsoft.com/en-us/library/windows/desktop/ms740548(v=vs.85).aspx">link</a>).
|
||||
|
||||
@@ -5,14 +5,14 @@
|
||||
* Redistribution and use in source and binary forms, with or without
|
||||
* modification, are permitted provided that the following conditions are
|
||||
* met:
|
||||
*
|
||||
*
|
||||
* * Redistributions of source code must retain the above copyright
|
||||
* notice, this list of conditions and the following disclaimer.
|
||||
* * Redistributions in binary form must reproduce the above
|
||||
* copyright notice, this list of conditions and the following disclaimer
|
||||
* in the documentation and/or other materials provided with the
|
||||
* distribution.
|
||||
*
|
||||
*
|
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
@@ -26,7 +26,7 @@
|
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*
|
||||
*/
|
||||
|
||||
|
||||
#ifndef TINS_PDU_H
|
||||
#define TINS_PDU_H
|
||||
|
||||
@@ -49,22 +49,22 @@ class NetworkInterface;
|
||||
*/
|
||||
typedef std::vector<uint8_t> byte_array;
|
||||
|
||||
/**
|
||||
/**
|
||||
* \class PDU
|
||||
* \brief Base class for protocol data units.
|
||||
*
|
||||
* Every PDU implementation inherits from this class.
|
||||
* Every PDU implementation inherits from this class.
|
||||
*
|
||||
* PDUs can contain 0 or 1 inner PDU. By stacking several PDUs together,
|
||||
* you can construct packets. These are created upwards: upper layers
|
||||
* will be children of the lower ones.
|
||||
* you can construct packets. These are created upwards: upper layers
|
||||
* will be children of the lower ones.
|
||||
*
|
||||
* If you want to find a specific protocol within a PDU chain, you can use
|
||||
* PDU::find_pdu and PDU::rfind_pdu. Both of them take a template parameter
|
||||
* that indicates the PDU type you are looking for. The first one returns a
|
||||
* pointer to the first object of that type, and the second one returns a
|
||||
* reference (and throws if it is not found).
|
||||
*
|
||||
* that indicates the PDU type you are looking for. The first one returns a
|
||||
* pointer to the first object of that type, and the second one returns a
|
||||
* reference (and throws if it is not found).
|
||||
*
|
||||
* For example:
|
||||
*
|
||||
* \code
|
||||
@@ -86,7 +86,7 @@ typedef std::vector<uint8_t> byte_array;
|
||||
* stack into a vector of bytes. This process might modify some parameters
|
||||
* on packets depending on which protocols are used in it. For example:
|
||||
*
|
||||
* - If the lowest protocol layer is IP (this means that there is no
|
||||
* - If the lowest protocol layer is IP (this means that there is no
|
||||
* link layer protocol in the packet), then it calculates the source address
|
||||
* that should be used in that IP PDU. \sa IP
|
||||
* - If a protocol contains a checksum field, its value will be calculated
|
||||
@@ -179,10 +179,13 @@ public:
|
||||
IPSEC_ESP,
|
||||
PKTAP,
|
||||
MPLS,
|
||||
DOT11_CONTROL_TA,
|
||||
VXLAN,
|
||||
RTP,
|
||||
UNKNOWN = 999,
|
||||
USER_DEFINED_PDU = 1000
|
||||
};
|
||||
|
||||
|
||||
/**
|
||||
* The endianness used by this PDU. This can be overriden
|
||||
* by subclasses.
|
||||
@@ -197,10 +200,10 @@ public:
|
||||
* \brief Default constructor
|
||||
*/
|
||||
metadata();
|
||||
|
||||
|
||||
/**
|
||||
* \brief Constructs an instance of metadata using the given values
|
||||
|
||||
|
||||
*/
|
||||
metadata(uint32_t header_size, PDUType current_type, PDUType next_type);
|
||||
|
||||
@@ -220,28 +223,28 @@ public:
|
||||
PDUType next_pdu_type;
|
||||
};
|
||||
|
||||
/**
|
||||
/**
|
||||
* \brief Default constructor.
|
||||
*/
|
||||
PDU();
|
||||
|
||||
|
||||
#if TINS_IS_CXX11
|
||||
/**
|
||||
* \brief Move constructor.
|
||||
*
|
||||
*
|
||||
* \param rhs The PDU to be moved.
|
||||
*/
|
||||
PDU(PDU &&rhs) TINS_NOEXCEPT
|
||||
PDU(PDU &&rhs) TINS_NOEXCEPT
|
||||
: inner_pdu_(0), parent_pdu_(0) {
|
||||
std::swap(inner_pdu_, rhs.inner_pdu_);
|
||||
if (inner_pdu_) {
|
||||
inner_pdu_->parent_pdu(this);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* \brief Move assignment operator.
|
||||
*
|
||||
*
|
||||
* \param rhs The PDU to be moved.
|
||||
*/
|
||||
PDU& operator=(PDU &&rhs) TINS_NOEXCEPT {
|
||||
@@ -255,7 +258,7 @@ public:
|
||||
}
|
||||
#endif
|
||||
|
||||
/**
|
||||
/**
|
||||
* \brief PDU destructor.
|
||||
*
|
||||
* Deletes the inner pdu, as a consequence every child pdu is
|
||||
@@ -281,6 +284,12 @@ public:
|
||||
*/
|
||||
uint32_t size() const;
|
||||
|
||||
/** \brief The whole chain of PDU's advertised size, including this one.
|
||||
*
|
||||
* Returns the sum of this and all children PDU's advertised size.
|
||||
*/
|
||||
virtual uint32_t advertised_size() const;
|
||||
|
||||
/**
|
||||
* \brief Getter for the inner PDU.
|
||||
* \return The current inner PDU. Might be a null pointer.
|
||||
@@ -296,18 +305,18 @@ public:
|
||||
PDU* parent_pdu() const {
|
||||
return parent_pdu_;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* \brief Releases the inner PDU.
|
||||
*
|
||||
*
|
||||
* This method makes this PDU to <b>no longer own</b> the inner
|
||||
* PDU. The current inner PDU is returned, and is <b>not</b>
|
||||
* destroyed. That means after calling this function, you are
|
||||
* destroyed. That means after calling this function, you are
|
||||
* responsible for using operator delete on the returned pointer.
|
||||
*
|
||||
*
|
||||
* Use this method if you want to somehow re-use a PDU that
|
||||
* is already owned by another PDU.
|
||||
*
|
||||
*
|
||||
* \return The current inner PDU. Might be 0.
|
||||
*/
|
||||
PDU* release_inner_pdu();
|
||||
@@ -317,26 +326,26 @@ public:
|
||||
*
|
||||
* When setting a new inner_pdu, the instance takesownership of
|
||||
* the object, therefore deleting it when it's no longer required.
|
||||
*
|
||||
*
|
||||
* \param next_pdu The new child PDU.
|
||||
*/
|
||||
void inner_pdu(PDU* next_pdu);
|
||||
|
||||
|
||||
/**
|
||||
* \brief Sets the child PDU.
|
||||
*
|
||||
* The PDU parameter is cloned using PDU::clone.
|
||||
*
|
||||
*
|
||||
* \param next_pdu The new child PDU.
|
||||
*/
|
||||
void inner_pdu(const PDU& next_pdu);
|
||||
|
||||
/**
|
||||
/**
|
||||
* \brief Serializes the whole chain of PDU's, including this one.
|
||||
*
|
||||
* This allocates a std::vector of size size(), and fills it
|
||||
* with the serialization this PDU, and all of the inner ones'.
|
||||
*
|
||||
*
|
||||
* \return serialization_type containing the serialization
|
||||
* of the whole stack of PDUs.
|
||||
*/
|
||||
@@ -350,7 +359,7 @@ public:
|
||||
* If no PDU matches, 0 is returned.
|
||||
* \param flag The flag which being searched.
|
||||
*/
|
||||
template<typename T>
|
||||
template<typename T>
|
||||
T* find_pdu(PDUType type = T::pdu_flag) {
|
||||
PDU* pdu = this;
|
||||
while (pdu) {
|
||||
@@ -361,24 +370,24 @@ public:
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* \brief Finds and returns the first PDU that matches the given flag.
|
||||
*
|
||||
* \param flag The flag which being searched.
|
||||
*/
|
||||
template<typename T>
|
||||
template<typename T>
|
||||
const T* find_pdu(PDUType type = T::pdu_flag) const {
|
||||
return const_cast<PDU*>(this)->find_pdu<T>(type);
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Finds and returns the first PDU that matches the given flag.
|
||||
*
|
||||
*
|
||||
* If the PDU is not found, a pdu_not_found exception is thrown.
|
||||
*
|
||||
*
|
||||
* \sa PDU::find_pdu
|
||||
*
|
||||
*
|
||||
* \param flag The flag which being searched.
|
||||
*/
|
||||
template<typename T>
|
||||
@@ -395,7 +404,7 @@ public:
|
||||
*
|
||||
* \param flag The flag which being searched.
|
||||
*/
|
||||
template<typename T>
|
||||
template<typename T>
|
||||
const T& rfind_pdu(PDUType type = T::pdu_flag) const {
|
||||
return const_cast<PDU*>(this)->rfind_pdu<T>(type);
|
||||
}
|
||||
@@ -410,36 +419,36 @@ public:
|
||||
*/
|
||||
virtual PDU* clone() const = 0;
|
||||
|
||||
/**
|
||||
/**
|
||||
* \brief Send the stack of PDUs through a PacketSender.
|
||||
*
|
||||
* This method will be called only for the PDU on the bottom of the stack,
|
||||
* therefore it should only implement this method if it can be sent.
|
||||
*
|
||||
*
|
||||
* PacketSender implements specific methods to send packets which start
|
||||
* on every valid TCP/IP stack layer; this should only be a proxy for
|
||||
* those methods.
|
||||
*
|
||||
*
|
||||
* If this PDU does not represent a link layer protocol, then
|
||||
* the interface argument will be ignored.
|
||||
*
|
||||
*
|
||||
* \param sender The PacketSender which will send the packet.
|
||||
* \param iface The network interface in which this packet will
|
||||
* \param iface The network interface in which this packet will
|
||||
* be sent.
|
||||
*/
|
||||
virtual void send(PacketSender& sender, const NetworkInterface& iface);
|
||||
|
||||
/**
|
||||
/**
|
||||
* \brief Receives a matching response for this packet.
|
||||
*
|
||||
* This method should act as a proxy for PacketSender::recv_lX methods.
|
||||
*
|
||||
*
|
||||
* \param sender The packet sender which will receive the packet.
|
||||
* \param iface The interface in which to expect the response.
|
||||
*/
|
||||
virtual PDU* recv_response(PacketSender& sender, const NetworkInterface& iface);
|
||||
|
||||
/**
|
||||
/**
|
||||
* \brief Check whether ptr points to a valid response for this PDU.
|
||||
*
|
||||
* This method must check whether the buffer pointed by ptr is a valid
|
||||
@@ -488,18 +497,18 @@ protected:
|
||||
|
||||
/**
|
||||
* \brief Prepares this PDU for serialization.
|
||||
*
|
||||
*
|
||||
* This method is called before the inner PDUs are serialized.
|
||||
* It's useful in situations such as when serializing IP PDUs,
|
||||
* which don't contain any link layer encapsulation, and therefore
|
||||
* require to set the source IP address before the TCP/UDP checksum
|
||||
* is calculated.
|
||||
*
|
||||
*
|
||||
* By default, this method does nothing
|
||||
*/
|
||||
virtual void prepare_for_serialize();
|
||||
|
||||
/**
|
||||
/**
|
||||
* \brief Serializes this PDU and propagates this action to child PDUs.
|
||||
*
|
||||
* \param buffer The buffer in which to store this PDU's serialization.
|
||||
@@ -507,7 +516,7 @@ protected:
|
||||
*/
|
||||
void serialize(uint8_t* buffer, uint32_t total_sz);
|
||||
|
||||
/**
|
||||
/**
|
||||
* \brief Serializes this TCP PDU.
|
||||
*
|
||||
* Each PDU must override this method and implement it's own
|
||||
@@ -525,18 +534,18 @@ private:
|
||||
|
||||
/**
|
||||
* \brief Concatenation operator.
|
||||
*
|
||||
* This operator concatenates several PDUs. A copy of the right
|
||||
*
|
||||
* This operator concatenates several PDUs. A copy of the right
|
||||
* operand is set at the end of the left one's inner PDU chain.
|
||||
* This means that:
|
||||
*
|
||||
*
|
||||
* IP some_ip = IP("127.0.0.1") / TCP(12, 13) / RawPDU("bleh");
|
||||
*
|
||||
* Works as expected, meaning the output PDU will look like the
|
||||
*
|
||||
* Works as expected, meaning the output PDU will look like the
|
||||
* following:
|
||||
*
|
||||
*
|
||||
* IP - TCP - RawPDU
|
||||
*
|
||||
*
|
||||
* \param lop The left operand, which will be the one modified.
|
||||
* \param rop The right operand, the one which will be appended
|
||||
* to lop.
|
||||
@@ -553,7 +562,7 @@ T& operator/= (T& lop, const PDU& rop) {
|
||||
|
||||
/**
|
||||
* \brief Concatenation operator.
|
||||
*
|
||||
*
|
||||
* \sa operator/=
|
||||
*/
|
||||
template<typename T>
|
||||
@@ -564,7 +573,7 @@ T operator/ (T lop, const PDU& rop) {
|
||||
|
||||
/**
|
||||
* \brief Concatenation operator on PDU pointers.
|
||||
*
|
||||
*
|
||||
* \sa operator/=
|
||||
*/
|
||||
template<typename T>
|
||||
|
||||
@@ -115,7 +115,7 @@ public:
|
||||
* \param data The payload to use.
|
||||
*/
|
||||
RawPDU(payload_type&& data)
|
||||
: payload_(move(data)) { }
|
||||
: payload_(std::move(data)) { }
|
||||
#endif // TINS_IS_CXX11
|
||||
|
||||
/**
|
||||
|
||||
328
include/tins/rtp.h
Normal file
328
include/tins/rtp.h
Normal file
@@ -0,0 +1,328 @@
|
||||
#ifndef TINS_RTP_H
|
||||
#define TINS_RTP_H
|
||||
|
||||
#include <tins/endianness.h>
|
||||
#include <tins/pdu.h>
|
||||
#include <tins/pdu_option.h>
|
||||
#include <tins/small_uint.h>
|
||||
|
||||
namespace Tins {
|
||||
|
||||
/**
|
||||
* \class RTP
|
||||
* \brief Represents a RTP PDU.
|
||||
*
|
||||
* This class represents a RTP PDU.
|
||||
*
|
||||
* \sa RawPDU
|
||||
*/
|
||||
class TINS_API RTP : public PDU {
|
||||
public:
|
||||
/**
|
||||
* \brief This PDU's flag.
|
||||
*/
|
||||
static const PDU::PDUType pdu_flag = PDU::RTP;
|
||||
|
||||
/**
|
||||
* The type used to store CSRC identifiers.
|
||||
*/
|
||||
typedef std::vector<uint32_t> csrc_ids_type;
|
||||
|
||||
/**
|
||||
* The type used to store extension header data.
|
||||
*/
|
||||
typedef std::vector<uint32_t> extension_header_data_type;
|
||||
|
||||
/**
|
||||
* Default constructor.
|
||||
*/
|
||||
RTP();
|
||||
|
||||
/**
|
||||
* \brief Constructs a RTP object from a buffer.
|
||||
*
|
||||
* \param data The buffer from which this PDU will be constructed.
|
||||
* \param size The size of the data buffer.
|
||||
*/
|
||||
RTP(const uint8_t* data, uint32_t size);
|
||||
|
||||
/**
|
||||
* \brief Getter for the version.
|
||||
*/
|
||||
small_uint<2> version() const { return header_.version; }
|
||||
|
||||
/**
|
||||
* \brief Getter for the padding bit.
|
||||
*/
|
||||
small_uint<1> padding_bit() const { return header_.padding; }
|
||||
|
||||
/**
|
||||
* \brief Getter for the extension bit.
|
||||
*/
|
||||
small_uint<1> extension_bit() const { return header_.extension; }
|
||||
|
||||
/**
|
||||
* \brief Getter for the CSRC count.
|
||||
*/
|
||||
small_uint<4> csrc_count() const { return header_.csrc_count; }
|
||||
|
||||
/**
|
||||
* \brief Getter for the marker bit.
|
||||
*/
|
||||
small_uint<1> marker_bit() const { return header_.marker; }
|
||||
|
||||
/**
|
||||
* \brief Getter for the payload type.
|
||||
*/
|
||||
small_uint<7> payload_type() const { return header_.payload_type; }
|
||||
|
||||
/**
|
||||
* \brief Getter for the sequence number.
|
||||
*/
|
||||
uint16_t sequence_number() const { return Endian::be_to_host(header_.seq_num); }
|
||||
|
||||
/**
|
||||
* \brief Getter for the timestamp.
|
||||
*/
|
||||
uint32_t timestamp() const { return Endian::be_to_host(header_.timestamp); }
|
||||
|
||||
/**
|
||||
* \brief Getter for the SSRC identifier.
|
||||
*/
|
||||
uint32_t ssrc_id() const { return Endian::be_to_host(header_.ssrc_id); }
|
||||
|
||||
/**
|
||||
* \brief Getter for the CSRC identifiers.
|
||||
*/
|
||||
const csrc_ids_type& csrc_ids() const {
|
||||
return csrc_ids_;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Getter for the padding size.
|
||||
*/
|
||||
uint8_t padding_size() const { return padding_size_; }
|
||||
|
||||
/**
|
||||
* \brief Getter for the extension header profile.
|
||||
*/
|
||||
uint16_t extension_profile() const { return Endian::be_to_host(ext_header_.profile); }
|
||||
|
||||
/**
|
||||
* \brief Getter for the extension header length.
|
||||
*/
|
||||
uint16_t extension_length() const { return Endian::be_to_host(ext_header_.length); }
|
||||
|
||||
/**
|
||||
* \brief Getter for the extension header data.
|
||||
*/
|
||||
const extension_header_data_type& extension_data() const {
|
||||
return ext_data_;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the version.
|
||||
* \param version The new version.
|
||||
*/
|
||||
void version(small_uint<2> version) { header_.version = version; }
|
||||
|
||||
/**
|
||||
* \brief Setter for the extension bit.
|
||||
* \param extension The new extension bit.
|
||||
*/
|
||||
void extension_bit(small_uint<1> extension) { header_.extension = extension; }
|
||||
|
||||
/**
|
||||
* \brief Setter for the marker bit.
|
||||
* \param marker The new marker bit.
|
||||
*/
|
||||
void marker_bit(small_uint<1> marker) { header_.marker = marker; }
|
||||
|
||||
/**
|
||||
* \brief Setter for the payload type.
|
||||
* \param payload_type The new payload type.
|
||||
*/
|
||||
void payload_type(small_uint<7> payload_type) { header_.payload_type = payload_type; }
|
||||
|
||||
/**
|
||||
* \brief Setter for the sequence number.
|
||||
* \param seq_num The new sequence number.
|
||||
*/
|
||||
void sequence_number(uint16_t seq_num) { header_.seq_num = Endian::host_to_be(seq_num); }
|
||||
|
||||
/**
|
||||
* \brief Setter for the timestamp.
|
||||
* \param timestamp The new timestamp.
|
||||
*/
|
||||
void timestamp(uint32_t timestamp) { header_.timestamp = Endian::host_to_be(timestamp); }
|
||||
|
||||
/**
|
||||
* \brief Setter for the SSRC identifier.
|
||||
* \param ssrc_id The new SSRC identifier.
|
||||
*/
|
||||
void ssrc_id(uint32_t ssrc_id) { header_.ssrc_id = Endian::host_to_be(ssrc_id); }
|
||||
|
||||
/**
|
||||
* \brief Setter for the padding size.
|
||||
* \param size The new padding size.
|
||||
*/
|
||||
void padding_size(uint8_t size) {
|
||||
padding_bit(size > 0);
|
||||
padding_size_ = size;
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Setter for the extension header profile.
|
||||
* \param profile The new extension header profile.
|
||||
*/
|
||||
void extension_profile(uint16_t profile) { ext_header_.profile = Endian::host_to_be(profile); }
|
||||
|
||||
/**
|
||||
* \brief Adds a word of extension header data.
|
||||
*
|
||||
* The word is added after the last word of extension header data.
|
||||
*
|
||||
* \param value The value of the extension header data to be added.
|
||||
*/
|
||||
void add_extension_data(const uint32_t value);
|
||||
|
||||
/**
|
||||
* \brief Removes a word of extension header data.
|
||||
*
|
||||
* If there are multiple words of extension header data of the given value,
|
||||
* only the first one will be removed.
|
||||
*
|
||||
* \param value The value of the extension header data to be removed.
|
||||
* \return true if the extension header data was removed, false otherwise.
|
||||
*/
|
||||
bool remove_extension_data(const uint32_t value);
|
||||
|
||||
/**
|
||||
* \brief Searches for extension header data that matches the given value.
|
||||
* \param value The extension header data to be searched.
|
||||
* \return true if the extension header data was found, false otherwise.
|
||||
*/
|
||||
bool search_extension_data(const uint32_t value);
|
||||
|
||||
/**
|
||||
* \brief Adds a CSRC identifier.
|
||||
*
|
||||
* The CSRC identifier is added after the last CSRC identifier in the extension
|
||||
* header.
|
||||
*
|
||||
* \param csrc_id The CSRC identifier to be added
|
||||
*/
|
||||
void add_csrc_id(const uint32_t csrc_id);
|
||||
|
||||
/**
|
||||
* \brief Removes a CSRC identifier.
|
||||
*
|
||||
* If there are multiple CSRC identifiers of the given value, only the first one
|
||||
* will be removed.
|
||||
*
|
||||
* \param value The value of the CSRC identifier to be removed.
|
||||
* \return true if the CSRC identifier was removed, false otherwise.
|
||||
*/
|
||||
bool remove_csrc_id(const uint32_t value);
|
||||
|
||||
/**
|
||||
* \brief Searches for a CSRC identifier that matches the given value.
|
||||
* \param value The CSRC identifier to be searched.
|
||||
* \return true if the CSRC identifier was found, false otherwise.
|
||||
*/
|
||||
bool search_csrc_id(const uint32_t value);
|
||||
|
||||
/**
|
||||
* \brief Returns the RTP packet's header length.
|
||||
*
|
||||
* This method overrides PDU::header_size.
|
||||
*
|
||||
* \return An uint32_t with the header's size.
|
||||
* \sa PDU::header_size
|
||||
*/
|
||||
uint32_t header_size() const;
|
||||
|
||||
/**
|
||||
* \brief Returns the RTP packet's trailer length.
|
||||
*
|
||||
* This method overrides PDU::trailer_size.
|
||||
*
|
||||
* \return An uint32_t with the trailer's size.
|
||||
* \sa PDU::trailer_size
|
||||
*/
|
||||
uint32_t trailer_size() const { return static_cast<uint32_t>(padding_size_); }
|
||||
|
||||
/**
|
||||
* \brief Getter for the PDU's type.
|
||||
* \sa PDU::pdu_type
|
||||
*/
|
||||
PDUType pdu_type() const { return pdu_flag; }
|
||||
|
||||
/**
|
||||
* \sa PDU::clone
|
||||
*/
|
||||
RTP *clone() const { return new RTP(*this); }
|
||||
|
||||
private:
|
||||
TINS_BEGIN_PACK
|
||||
struct rtp_header {
|
||||
#if TINS_IS_BIG_ENDIAN
|
||||
uint16_t version:2,
|
||||
padding:1,
|
||||
extension:1,
|
||||
csrc_count:4,
|
||||
marker:1,
|
||||
payload_type:7;
|
||||
#elif TINS_IS_LITTLE_ENDIAN
|
||||
uint16_t csrc_count:4,
|
||||
extension:1,
|
||||
padding:1,
|
||||
version:2,
|
||||
payload_type:7,
|
||||
marker:1;
|
||||
#endif
|
||||
uint16_t seq_num;
|
||||
uint32_t timestamp;
|
||||
uint32_t ssrc_id;
|
||||
} TINS_END_PACK;
|
||||
|
||||
TINS_BEGIN_PACK
|
||||
struct rtp_extension_header {
|
||||
uint16_t profile;
|
||||
uint16_t length;
|
||||
} TINS_END_PACK;
|
||||
|
||||
void write_serialization(uint8_t* buffer, uint32_t size);
|
||||
csrc_ids_type::const_iterator search_csrc_id_iterator(const uint32_t csrc_id) const;
|
||||
csrc_ids_type::iterator search_csrc_id_iterator(const uint32_t csrc_id);
|
||||
extension_header_data_type::const_iterator search_extension_data_iterator(const uint32_t data) const;
|
||||
extension_header_data_type::iterator search_extension_data_iterator(const uint32_t data);
|
||||
|
||||
/**
|
||||
* \brief Setter for the padding bit.
|
||||
* \param padding The new padding bit.
|
||||
*/
|
||||
void padding_bit(small_uint<1> padding) { header_.padding = padding; }
|
||||
|
||||
/**
|
||||
* \brief Setter for the CSRC count. Hidden from the public interface.
|
||||
* \param csrc_count The new CSRC count.
|
||||
*/
|
||||
void csrc_count(small_uint<4> csrc_count) { header_.csrc_count = csrc_count; }
|
||||
|
||||
/**
|
||||
* \brief Setter for the extension header length. Hidden from the public interface.
|
||||
* \param length The new extension header length.
|
||||
*/
|
||||
void extension_length(uint16_t length) { ext_header_.length = Endian::host_to_be(length); }
|
||||
|
||||
rtp_header header_;
|
||||
csrc_ids_type csrc_ids_;
|
||||
rtp_extension_header ext_header_;
|
||||
extension_header_data_type ext_data_;
|
||||
uint8_t padding_size_;
|
||||
};
|
||||
|
||||
} // Tins
|
||||
|
||||
#endif // TINS_RTP_H
|
||||
@@ -406,10 +406,17 @@ private:
|
||||
*/
|
||||
class TINS_API FileSniffer : public BaseSniffer {
|
||||
public:
|
||||
/**
|
||||
* \brief Constructs an instance of FileSniffer.
|
||||
* \param fp The pcap file which will be parsed.
|
||||
* \param configuration A SnifferConfiguration to be used on the file.
|
||||
*/
|
||||
FileSniffer(FILE *fp, const SnifferConfiguration& configuration);
|
||||
|
||||
/**
|
||||
* \brief Constructs an instance of FileSniffer.
|
||||
* \param file_name The pcap file which will be parsed.
|
||||
* \param filter A capture filter to be used on the file.(optional);
|
||||
* \param configuration A SnifferConfiguration to be used on the file.
|
||||
*/
|
||||
FileSniffer(const std::string& file_name, const SnifferConfiguration& configuration);
|
||||
|
||||
@@ -418,9 +425,18 @@ public:
|
||||
*
|
||||
* \brief Constructs an instance of FileSniffer.
|
||||
* \param file_name The pcap file which will be parsed.
|
||||
* \param filter A capture filter to be used on the file.(optional);
|
||||
* \param filter A capture filter to be used on the file. (optional)
|
||||
*/
|
||||
FileSniffer(const std::string& file_name, const std::string& filter = "");
|
||||
|
||||
/**
|
||||
* \deprecated Use the constructor that takes a SnifferConfiguration instead.
|
||||
*
|
||||
* \brief Constructs an instance of FileSniffer.
|
||||
* \param fp The pcap file which will be parsed.
|
||||
* \param filter A capture filter to be used on the file. (optional)
|
||||
*/
|
||||
FileSniffer(FILE *fp, const std::string& filter = "");
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
@@ -449,8 +465,14 @@ HandlerProxy<T> make_sniffer_handler(T* ptr,
|
||||
/**
|
||||
* \brief Iterates over packets sniffed by a BaseSniffer.
|
||||
*/
|
||||
class SnifferIterator : public std::iterator<std::forward_iterator_tag, Packet> {
|
||||
class SnifferIterator {
|
||||
public:
|
||||
typedef std::forward_iterator_tag iterator_category;
|
||||
typedef Packet value_type;
|
||||
typedef std::ptrdiff_t difference_type;
|
||||
typedef Packet* pointer;
|
||||
typedef Packet& reference;
|
||||
|
||||
/**
|
||||
* Constructs a SnifferIterator.
|
||||
* \param sniffer The sniffer to iterate.
|
||||
|
||||
@@ -288,13 +288,29 @@ public:
|
||||
* \code
|
||||
* TCP tcp = ...;
|
||||
* if(tcp.flags() == (TCP::SYN | TCP::ACK)) {
|
||||
* // It's a SYN+ACK!
|
||||
* // It's a SYN+ACK, but not SYN+ACK+ECN!
|
||||
* }
|
||||
* \endcode
|
||||
*
|
||||
* \return The value of the flags field.
|
||||
*/
|
||||
small_uint<12> flags() const;
|
||||
|
||||
/**
|
||||
* \brief Check if the given flags are set.
|
||||
*
|
||||
* \code
|
||||
* TCP tcp = ...;
|
||||
* if(tcp.has_flags(TCP::SYN | TCP::ACK)) {
|
||||
* // It's a SYN+ACK, but it also possible that other flags are set!
|
||||
* // it is equivalent to: (tcp.flags() & (TCP::SYN | TCP::ACK)) == (TCP::SYN | TCP::ACK)
|
||||
* }
|
||||
* \endcode
|
||||
*
|
||||
* \param check_flags
|
||||
* \return true if all check_flags are set
|
||||
*/
|
||||
bool has_flags(small_uint<12> check_flags) const;
|
||||
|
||||
/* Setters */
|
||||
|
||||
|
||||
@@ -49,16 +49,19 @@
|
||||
#include <tins/ipv6.h>
|
||||
#include <tins/mpls.h>
|
||||
#include <tins/packet_sender.h>
|
||||
#include <tins/packet_writer.h>
|
||||
#include <tins/pdu.h>
|
||||
#include <tins/radiotap.h>
|
||||
#include <tins/rawpdu.h>
|
||||
#include <tins/snap.h>
|
||||
#include <tins/sniffer.h>
|
||||
#include <tins/tcp.h>
|
||||
#include <tins/udp.h>
|
||||
#include <tins/utils.h>
|
||||
#if defined(TINS_HAVE_PCAP)
|
||||
#include <tins/packet_writer.h>
|
||||
#include <tins/sniffer.h>
|
||||
#include <tins/ppi.h>
|
||||
#include <tins/tcp_stream.h>
|
||||
#endif
|
||||
#include <tins/crypto.h>
|
||||
#include <tins/pdu_cacher.h>
|
||||
#include <tins/rsn_information.h>
|
||||
@@ -75,7 +78,8 @@
|
||||
#include <tins/pdu_allocator.h>
|
||||
#include <tins/ipsec.h>
|
||||
#include <tins/ip_reassembler.h>
|
||||
#include <tins/ppi.h>
|
||||
#include <tins/pdu_iterator.h>
|
||||
#include <tins/vxlan.h>
|
||||
#include <tins/rtp.h>
|
||||
|
||||
#endif // TINS_TINS_H
|
||||
|
||||
@@ -114,7 +114,11 @@ public:
|
||||
RadioTap::option current_option();
|
||||
|
||||
/**
|
||||
* Gets the pointer at which the current option is located
|
||||
* \brief Gets the pointer at which the current option is located
|
||||
*
|
||||
* A past-the-end pointer may be returned in case of malformed input or
|
||||
* end of data. Its validity must be checked (e.g. using
|
||||
* \ref RadioTapParser.has_fields) before dereference.
|
||||
*/
|
||||
const uint8_t* current_option_ptr() const;
|
||||
|
||||
|
||||
98
include/tins/vxlan.h
Normal file
98
include/tins/vxlan.h
Normal file
@@ -0,0 +1,98 @@
|
||||
#ifndef TINS_VXLAN_H
|
||||
#define TINS_VXLAN_H
|
||||
|
||||
#include <tins/pdu.h>
|
||||
#include <tins/small_uint.h>
|
||||
|
||||
namespace Tins {
|
||||
|
||||
/**
|
||||
* \class VXLAN
|
||||
* \brief Represents a VXLAN PDU.
|
||||
*
|
||||
* This class represents a VXLAN PDU.
|
||||
*
|
||||
* \sa RawPDU
|
||||
*/
|
||||
class TINS_API VXLAN : public PDU {
|
||||
public:
|
||||
/**
|
||||
* \brief This PDU's flag.
|
||||
*/
|
||||
static const PDU::PDUType pdu_flag = PDU::VXLAN;
|
||||
|
||||
/**
|
||||
* \brief Constructs a VXLAN PDU.
|
||||
*
|
||||
* \param vni VXLAN Network Identifier.
|
||||
*/
|
||||
VXLAN(const small_uint<24> vni = 0);
|
||||
|
||||
/**
|
||||
* \brief Constructs a VXLAN object from a buffer and adds
|
||||
* the Ethernet II PDU.
|
||||
*
|
||||
* \param buffer The buffer from which this PDU will be constructed.
|
||||
* \param total_sz The total size of the buffer.
|
||||
*/
|
||||
VXLAN(const uint8_t* buffer, uint32_t total_sz);
|
||||
|
||||
/**
|
||||
* \brief Getter for the flags.
|
||||
*/
|
||||
uint8_t get_flags() const { return Endian::be_to_host(header_.flags) >> 24; }
|
||||
|
||||
/**
|
||||
* \brief Getter for the VNI.
|
||||
*/
|
||||
small_uint<24> get_vni() const { return Endian::be_to_host(header_.vni) >> 8; }
|
||||
|
||||
/**
|
||||
* \brief Setter for the flags.
|
||||
* \param new_flags The new flags.
|
||||
*/
|
||||
void set_flags(uint8_t new_flags) { header_.flags = Endian::host_to_be(new_flags << 24); }
|
||||
|
||||
/**
|
||||
* \brief Setter for the VNI.
|
||||
* \param new_vni The new VNI.
|
||||
*/
|
||||
void set_vni(small_uint<24> new_vni) { header_.vni = Endian::host_to_be(new_vni << 8); }
|
||||
|
||||
/**
|
||||
* \brief Returns the VXLAN frame's header length.
|
||||
*
|
||||
* This method overrides PDU::header_size. This size includes the
|
||||
* payload and options size.
|
||||
*
|
||||
* \return An uint32_t with the header's size.
|
||||
* \sa PDU::header_size
|
||||
*/
|
||||
uint32_t header_size() const { return sizeof(header_); }
|
||||
|
||||
/**
|
||||
* \brief Getter for the PDU's type.
|
||||
* \sa PDU::pdu_type
|
||||
*/
|
||||
PDUType pdu_type() const { return pdu_flag; }
|
||||
|
||||
/**
|
||||
* \sa PDU::clone
|
||||
*/
|
||||
VXLAN *clone() const { return new VXLAN(*this); }
|
||||
|
||||
private:
|
||||
TINS_BEGIN_PACK
|
||||
struct vxlan_header {
|
||||
uint32_t flags;
|
||||
uint32_t vni;
|
||||
} TINS_END_PACK;
|
||||
|
||||
void write_serialization(uint8_t* buffer, uint32_t total_sz);
|
||||
|
||||
vxlan_header header_;
|
||||
};
|
||||
|
||||
} // Tins
|
||||
|
||||
#endif // TINS_VXLAN_H
|
||||
@@ -54,6 +54,7 @@ set(SOURCES
|
||||
radiotap.cpp
|
||||
rawpdu.cpp
|
||||
rsn_information.cpp
|
||||
rtp.cpp
|
||||
sll.cpp
|
||||
snap.cpp
|
||||
stp.cpp
|
||||
@@ -73,6 +74,7 @@ set(SOURCES
|
||||
utils/routing_utils.cpp
|
||||
utils/resolve_utils.cpp
|
||||
utils/pdu_utils.cpp
|
||||
vxlan.cpp
|
||||
)
|
||||
|
||||
set(HEADERS
|
||||
@@ -130,6 +132,7 @@ set(HEADERS
|
||||
${LIBTINS_INCLUDE_DIR}/tins/radiotap.h
|
||||
${LIBTINS_INCLUDE_DIR}/tins/rawpdu.h
|
||||
${LIBTINS_INCLUDE_DIR}/tins/rsn_information.h
|
||||
${LIBTINS_INCLUDE_DIR}/tins/rtp.h
|
||||
${LIBTINS_INCLUDE_DIR}/tins/sll.h
|
||||
${LIBTINS_INCLUDE_DIR}/tins/small_uint.h
|
||||
${LIBTINS_INCLUDE_DIR}/tins/snap.h
|
||||
@@ -151,6 +154,7 @@ set(HEADERS
|
||||
${LIBTINS_INCLUDE_DIR}/tins/utils/routing_utils.h
|
||||
${LIBTINS_INCLUDE_DIR}/tins/utils/resolve_utils.h
|
||||
${LIBTINS_INCLUDE_DIR}/tins/utils/pdu_utils.h
|
||||
${LIBTINS_INCLUDE_DIR}/tins/vxlan.h
|
||||
)
|
||||
|
||||
SET(DOT11_DEPENDENT_SOURCES
|
||||
@@ -219,6 +223,7 @@ SET_TARGET_PROPERTIES(tins PROPERTIES VERSION ${LIBTINS_VERSION} SOVERSION ${LIB
|
||||
INSTALL(
|
||||
TARGETS tins
|
||||
EXPORT libtinsTargets
|
||||
RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR}
|
||||
LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
|
||||
ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
|
||||
COMPONENT dev
|
||||
|
||||
@@ -705,8 +705,11 @@ WPA2Decrypter::bssids_map::const_iterator WPA2Decrypter::find_ap(const Dot11Data
|
||||
|
||||
bool WPA2Decrypter::decrypt(PDU& pdu) {
|
||||
if (capturer_.process_packet(pdu)) {
|
||||
try_add_keys(pdu.rfind_pdu<Dot11Data>(), capturer_.handshakes().front());
|
||||
capturer_.clear_handshakes();
|
||||
Dot11Data* data = pdu.find_pdu<Dot11Data>();
|
||||
if (data) {
|
||||
try_add_keys(*data, capturer_.handshakes().front());
|
||||
capturer_.clear_handshakes();
|
||||
}
|
||||
}
|
||||
else if (const Dot11Beacon* beacon = pdu.find_pdu<Dot11Beacon>()) {
|
||||
if (aps_.count(beacon->addr3()) == 0) {
|
||||
|
||||
@@ -169,6 +169,8 @@ Tins::PDU* pdu_from_flag(PDU::PDUType type, const uint8_t* buffer, uint32_t size
|
||||
return new Tins::IEEE802_3(buffer, size);
|
||||
case Tins::PDU::PPPOE:
|
||||
return new Tins::PPPoE(buffer, size);
|
||||
case Tins::PDU::RAW:
|
||||
return new Tins::RawPDU(buffer, size);
|
||||
#ifdef TINS_HAVE_DOT11
|
||||
case Tins::PDU::RADIOTAP:
|
||||
return new Tins::RadioTap(buffer, size);
|
||||
|
||||
32
src/dns.cpp
32
src/dns.cpp
@@ -86,14 +86,18 @@ void DNS::skip_to_dname_end(InputMemoryStream& stream) const {
|
||||
break;
|
||||
}
|
||||
else {
|
||||
if ((value & 0xc0)) {
|
||||
// This is an offset label, skip the second byte and we're done
|
||||
const uint8_t offset_discriminator = value & 0xc0;
|
||||
if (offset_discriminator == 0xc0) {
|
||||
// This is an offset pointer, skip the second byte and we're done
|
||||
stream.skip(1);
|
||||
break;
|
||||
}
|
||||
else {
|
||||
else if (offset_discriminator == 0) {
|
||||
// This is an actual label, skip its contents
|
||||
stream.skip(value);
|
||||
} else {
|
||||
// high order two bits of the first octet of a label must be either 11 or 00
|
||||
throw malformed_packet();
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -336,9 +340,13 @@ uint32_t DNS::compose_name(const uint8_t* ptr, char* out_ptr) const {
|
||||
const uint8_t* end = &records_data_[0] + records_data_.size();
|
||||
const uint8_t* end_ptr = 0;
|
||||
char* current_out_ptr = out_ptr;
|
||||
uint8_t pointer_counter = 0;
|
||||
while (*ptr) {
|
||||
if (pointer_counter++ > 30){
|
||||
throw dns_decompression_pointer_loops();
|
||||
}
|
||||
// It's an offset
|
||||
if ((*ptr & 0xc0)) {
|
||||
if (((*ptr & 0xc0) == 0xc0)) {
|
||||
if (TINS_UNLIKELY(ptr + sizeof(uint16_t) > end)) {
|
||||
throw malformed_packet();
|
||||
}
|
||||
@@ -347,7 +355,7 @@ uint32_t DNS::compose_name(const uint8_t* ptr, char* out_ptr) const {
|
||||
index = Endian::be_to_host(index) & 0x3fff;
|
||||
// Check that the offset is neither too low or too high
|
||||
if (index < 0x0c || (&records_data_[0] + (index - 0x0c)) >= end) {
|
||||
throw malformed_packet();
|
||||
throw dns_decompression_pointer_out_of_bounds();
|
||||
}
|
||||
// We've probably found the end of the original domain name. Save it.
|
||||
if (end_ptr == 0) {
|
||||
@@ -410,10 +418,11 @@ void DNS::inline_convert_v4(uint32_t value, char* output) {
|
||||
// Parses records in some section.
|
||||
void DNS::convert_records(const uint8_t* ptr,
|
||||
const uint8_t* end,
|
||||
resources_type& res) const {
|
||||
resources_type& res,
|
||||
const uint16_t rr_count) const {
|
||||
InputMemoryStream stream(ptr, end - ptr);
|
||||
char dname[256], small_addr_buf[256];
|
||||
while (stream) {
|
||||
while (stream && (res.size() < rr_count)) {
|
||||
string data;
|
||||
bool used_small_buffer = false;
|
||||
// Retrieve the record's domain name.
|
||||
@@ -573,7 +582,8 @@ DNS::resources_type DNS::answers() const {
|
||||
convert_records(
|
||||
&records_data_[0] + answers_idx_,
|
||||
&records_data_[0] + authority_idx_,
|
||||
res
|
||||
res,
|
||||
answers_count()
|
||||
);
|
||||
}
|
||||
return res;
|
||||
@@ -585,7 +595,8 @@ DNS::resources_type DNS::authority() const {
|
||||
convert_records(
|
||||
&records_data_[0] + authority_idx_,
|
||||
&records_data_[0] + additional_idx_,
|
||||
res
|
||||
res,
|
||||
authority_count()
|
||||
);
|
||||
}
|
||||
return res;
|
||||
@@ -597,7 +608,8 @@ DNS::resources_type DNS::additional() const {
|
||||
convert_records(
|
||||
&records_data_[0] + additional_idx_,
|
||||
&records_data_[0] + records_data_.size(),
|
||||
res
|
||||
res,
|
||||
additional_count()
|
||||
);
|
||||
}
|
||||
return res;
|
||||
|
||||
@@ -98,7 +98,7 @@ void ICMP::sequence(uint16_t new_seq) {
|
||||
}
|
||||
|
||||
void ICMP::gateway(address_type new_gw) {
|
||||
header_.un.gateway = Endian::host_to_be(static_cast<uint32_t>(new_gw));
|
||||
header_.un.gateway = static_cast<uint32_t>(new_gw);
|
||||
}
|
||||
|
||||
void ICMP::mtu(uint16_t new_mtu) {
|
||||
@@ -122,7 +122,7 @@ void ICMP::transmit_timestamp(uint32_t new_timestamp) {
|
||||
}
|
||||
|
||||
void ICMP::address_mask(address_type new_mask) {
|
||||
orig_timestamp_or_address_mask_ = Endian::host_to_be(static_cast<uint32_t>(new_mask));
|
||||
orig_timestamp_or_address_mask_ = static_cast<uint32_t>(new_mask);
|
||||
}
|
||||
|
||||
uint32_t ICMP::header_size() const {
|
||||
|
||||
@@ -62,7 +62,7 @@ const AddressRange<IPv4Address> loopback_range = IPv4Address("127.0.0.0") / 8;
|
||||
const AddressRange<IPv4Address> multicast_range = IPv4Address("224.0.0.0") / 4;
|
||||
|
||||
IPv4Address IPv4Address::from_prefix_length(uint32_t prefix_length) {
|
||||
return IPv4Address(Endian::host_to_be(0xffffffff << (32 - prefix_length)));
|
||||
return IPv4Address(prefix_length ? Endian::host_to_be(0xffffffff << (32 - prefix_length)) : 0u);
|
||||
}
|
||||
|
||||
IPv4Address::IPv4Address(uint32_t ip)
|
||||
|
||||
16
src/ipv6.cpp
16
src/ipv6.cpp
@@ -32,6 +32,9 @@
|
||||
#include <netinet/in.h>
|
||||
#include <sys/socket.h>
|
||||
#else
|
||||
#ifndef WIN32_LEAN_AND_MEAN
|
||||
#define WIN32_LEAN_AND_MEAN
|
||||
#endif
|
||||
#include <ws2tcpip.h>
|
||||
#endif
|
||||
#include <tins/ipv6.h>
|
||||
@@ -127,7 +130,7 @@ IPv6::IPv6(const uint8_t* buffer, uint32_t total_sz) {
|
||||
uint32_t actual_payload_length = payload_length();
|
||||
bool is_payload_fragmented = false;
|
||||
while (stream) {
|
||||
if (is_extension_header(current_header)) {
|
||||
if (is_extension_header(current_header) && current_header != NO_NEXT_HEADER) {
|
||||
if (current_header == FRAGMENT) {
|
||||
is_payload_fragmented = true;
|
||||
}
|
||||
@@ -204,8 +207,8 @@ IPv6::IPv6(const uint8_t* buffer, uint32_t total_sz) {
|
||||
bool IPv6::is_extension_header(uint8_t header_id) {
|
||||
return header_id == HOP_BY_HOP || header_id == DESTINATION_ROUTING_OPTIONS
|
||||
|| header_id == ROUTING || header_id == FRAGMENT || header_id == AUTHENTICATION
|
||||
|| header_id == SECURITY_ENCAPSULATION || header_id == DESTINATION_OPTIONS
|
||||
|| header_id == MOBILITY || header_id == NO_NEXT_HEADER;
|
||||
|| header_id == DESTINATION_OPTIONS || header_id == MOBILITY
|
||||
|| header_id == NO_NEXT_HEADER;
|
||||
}
|
||||
|
||||
uint32_t IPv6::get_padding_size(const ext_header& header) {
|
||||
@@ -366,11 +369,16 @@ void IPv6::write_serialization(uint8_t* buffer, uint32_t total_sz) {
|
||||
}
|
||||
|
||||
#ifndef BSD
|
||||
void IPv6::send(PacketSender& sender, const NetworkInterface &) {
|
||||
void IPv6::send(PacketSender& sender, const NetworkInterface& interface) {
|
||||
sockaddr_in6 link_addr;
|
||||
const PacketSender::SocketType type = PacketSender::IPV6_SOCKET;
|
||||
link_addr.sin6_family = AF_INET6;
|
||||
link_addr.sin6_port = 0;
|
||||
// Required to set sin6_scope_id to interface index as stated in RFC2553.
|
||||
// https://datatracker.ietf.org/doc/html/rfc2553#section-3.3
|
||||
if (IPv6Address(header_.dst_addr).is_local_unicast()) {
|
||||
link_addr.sin6_scope_id = interface.id();
|
||||
}
|
||||
memcpy((uint8_t*)&link_addr.sin6_addr, header_.dst_addr, address_type::address_size);
|
||||
sender.send_l3(*this, (struct sockaddr*)&link_addr, sizeof(link_addr), type);
|
||||
}
|
||||
|
||||
@@ -57,6 +57,7 @@ namespace Tins {
|
||||
|
||||
const IPv6Address loopback_address = "::1";
|
||||
const AddressRange<IPv6Address> multicast_range = IPv6Address("ff00::") / 8;
|
||||
const AddressRange<IPv6Address> local_unicast_range = IPv6Address("fe80::") / 10;
|
||||
|
||||
IPv6Address IPv6Address::from_prefix_length(uint32_t prefix_length) {
|
||||
IPv6Address address;
|
||||
@@ -138,6 +139,10 @@ bool IPv6Address::is_multicast() const {
|
||||
return multicast_range.contains(*this);
|
||||
}
|
||||
|
||||
bool IPv6Address::is_local_unicast() const {
|
||||
return local_unicast_range.contains(*this);
|
||||
}
|
||||
|
||||
ostream& operator<<(ostream& os, const IPv6Address& addr) {
|
||||
return os << addr.to_string();
|
||||
}
|
||||
|
||||
@@ -331,7 +331,7 @@ NetworkInterface::Info NetworkInterface::info() const {
|
||||
}
|
||||
|
||||
#else // _WIN32
|
||||
|
||||
#if !defined(ANDROID) || (defined(__ANDROID_API__) && __ANDROID_API__ >= 24)
|
||||
struct ifaddrs* ifaddrs = 0;
|
||||
struct ifaddrs* if_it = 0;
|
||||
getifaddrs(&ifaddrs);
|
||||
@@ -341,7 +341,9 @@ NetworkInterface::Info NetworkInterface::info() const {
|
||||
if (ifaddrs) {
|
||||
freeifaddrs(ifaddrs);
|
||||
}
|
||||
|
||||
#else
|
||||
throw new std::runtime_error("android ifaddr not supported");
|
||||
#endif
|
||||
#endif // _WIN32
|
||||
|
||||
// If we didn't even get the hw address or ip address, this went wrong
|
||||
|
||||
@@ -37,11 +37,14 @@ using std::string;
|
||||
namespace Tins {
|
||||
|
||||
OfflinePacketFilter::OfflinePacketFilter(const OfflinePacketFilter& other) {
|
||||
*this = other;
|
||||
string_filter_ = other.string_filter_;
|
||||
init(string_filter_, pcap_datalink(other.handle_), pcap_snapshot(other.handle_));
|
||||
}
|
||||
|
||||
OfflinePacketFilter& OfflinePacketFilter::operator=(const OfflinePacketFilter& other) {
|
||||
string_filter_ = other.string_filter_;
|
||||
pcap_freecode(&filter_);
|
||||
pcap_close(handle_);
|
||||
init(string_filter_, pcap_datalink(other.handle_), pcap_snapshot(other.handle_));
|
||||
return* this;
|
||||
}
|
||||
@@ -58,8 +61,14 @@ void OfflinePacketFilter::init(const string& pcap_filter,
|
||||
link_type,
|
||||
snap_len
|
||||
);
|
||||
if (!handle_) {
|
||||
throw pcap_open_failed();
|
||||
}
|
||||
if (pcap_compile(handle_, &filter_, pcap_filter.c_str(), 1, 0xffffffff) == -1) {
|
||||
throw invalid_pcap_filter(pcap_geterr(handle_));
|
||||
string error(pcap_geterr(handle_));
|
||||
pcap_freecode(&filter_);
|
||||
pcap_close(handle_);
|
||||
throw invalid_pcap_filter(error.c_str());
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -267,6 +267,11 @@ void PacketSender::open_l3_socket(SocketType type) {
|
||||
#endif
|
||||
const int level = (is_v6) ? IPPROTO_IPV6 : IPPROTO_IP;
|
||||
if (setsockopt(sockfd, level, IP_HDRINCL, (option_ptr)&on, sizeof(on)) != 0) {
|
||||
#ifndef _WIN32
|
||||
::close(sockfd);
|
||||
#else
|
||||
::closesocket(sockfd);
|
||||
#endif
|
||||
throw socket_open_error(make_error_string());
|
||||
}
|
||||
|
||||
|
||||
@@ -70,12 +70,12 @@ void PacketWriter::write(Packet& packet) {
|
||||
}
|
||||
|
||||
void PacketWriter::write(PDU& pdu, const struct timeval& tv) {
|
||||
PDU::serialization_type buffer = pdu.serialize();
|
||||
struct pcap_pkthdr header;
|
||||
memset(&header, 0, sizeof(header));
|
||||
header.ts = tv;
|
||||
header.len = static_cast<bpf_u_int32>(pdu.advertised_size());
|
||||
PDU::serialization_type buffer = pdu.serialize();
|
||||
header.caplen = static_cast<bpf_u_int32>(buffer.size());
|
||||
header.len = static_cast<bpf_u_int32>(buffer.size());
|
||||
pcap_dump((u_char*)dumper_, &header, &buffer[0]);
|
||||
}
|
||||
|
||||
@@ -86,8 +86,9 @@ void PacketWriter::init(const string& file_name, int link_type) {
|
||||
}
|
||||
dumper_ = pcap_dump_open(handle_, file_name.c_str());
|
||||
if (!dumper_) {
|
||||
string error(pcap_geterr(handle_));
|
||||
pcap_close(handle_);
|
||||
throw pcap_error(pcap_geterr(handle_));
|
||||
throw pcap_error(error);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -85,6 +85,14 @@ uint32_t PDU::size() const {
|
||||
return sz;
|
||||
}
|
||||
|
||||
uint32_t PDU::advertised_size() const {
|
||||
uint32_t result = header_size() + trailer_size();
|
||||
if (inner_pdu_) {
|
||||
result += inner_pdu()->advertised_size();
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
void PDU::send(PacketSender &, const NetworkInterface &) {
|
||||
|
||||
}
|
||||
|
||||
195
src/rtp.cpp
Normal file
195
src/rtp.cpp
Normal file
@@ -0,0 +1,195 @@
|
||||
#include <algorithm>
|
||||
#include <tins/exceptions.h>
|
||||
#include <tins/internals.h>
|
||||
#include <tins/memory_helpers.h>
|
||||
#include <tins/rtp.h>
|
||||
|
||||
using std::logic_error;
|
||||
using Tins::Memory::InputMemoryStream;
|
||||
using Tins::Memory::OutputMemoryStream;
|
||||
|
||||
namespace Tins {
|
||||
|
||||
RTP::RTP()
|
||||
: header_(), ext_header_(), padding_size_(0) {
|
||||
version(2);
|
||||
}
|
||||
|
||||
RTP::RTP(const uint8_t* buffer, uint32_t total_sz) {
|
||||
InputMemoryStream stream(buffer, total_sz);
|
||||
stream.read(header_);
|
||||
|
||||
small_uint<4> csrc_count_ = csrc_count();
|
||||
|
||||
for (uint32_t i = 0; i < csrc_count_; ++i) {
|
||||
uint32_t csrc_id;
|
||||
stream.read(csrc_id);
|
||||
csrc_ids_.push_back(csrc_id);
|
||||
}
|
||||
|
||||
if (extension_bit() == 1) {
|
||||
stream.read(ext_header_);
|
||||
for (uint32_t i = 0; i < extension_length(); ++i) {
|
||||
uint32_t data;
|
||||
stream.read(data);
|
||||
ext_data_.push_back(data);
|
||||
}
|
||||
}
|
||||
|
||||
padding_size_ = 0;
|
||||
|
||||
const uint8_t* data_ptr = stream.pointer();
|
||||
const size_t data_size = stream.size();
|
||||
|
||||
if (padding_bit() == 1) {
|
||||
if (data_size > 0) {
|
||||
stream.skip(data_size - sizeof(uint8_t));
|
||||
stream.read(padding_size_);
|
||||
} else {
|
||||
throw malformed_packet();
|
||||
}
|
||||
|
||||
if (padding_size() == 0) {
|
||||
throw malformed_packet();
|
||||
}
|
||||
}
|
||||
|
||||
if (padding_size() > data_size) {
|
||||
throw malformed_packet();
|
||||
}
|
||||
|
||||
if (data_size > padding_size()) {
|
||||
inner_pdu(
|
||||
Internals::pdu_from_flag(
|
||||
PDU::RAW,
|
||||
data_ptr,
|
||||
data_size - padding_size()
|
||||
)
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
uint32_t RTP::header_size() const {
|
||||
uint32_t extension_size = 0;
|
||||
if (extension_bit() == 1) {
|
||||
extension_size = sizeof(ext_header_) + (extension_length() * sizeof(uint32_t));
|
||||
}
|
||||
return static_cast<uint32_t>(sizeof(header_) + (csrc_ids_.size() * sizeof(uint32_t)) + extension_size);
|
||||
}
|
||||
|
||||
void RTP::add_csrc_id(const uint32_t csrc_id) {
|
||||
small_uint<4> csrc_count_ = csrc_count();
|
||||
if (TINS_UNLIKELY(csrc_count_ >= 15)) {
|
||||
throw logic_error("Maximum number of CSRC IDs reached");
|
||||
}
|
||||
|
||||
csrc_ids_.push_back(Endian::host_to_be(csrc_id));
|
||||
csrc_count(csrc_count_ + 1);
|
||||
}
|
||||
|
||||
bool RTP::remove_csrc_id(const uint32_t csrc_id) {
|
||||
small_uint<4> csrc_count_ = csrc_count();
|
||||
if (csrc_count_ == 0) {
|
||||
return false;
|
||||
}
|
||||
|
||||
csrc_ids_type::iterator iter = search_csrc_id_iterator(Endian::host_to_be(csrc_id));
|
||||
if (iter == csrc_ids_.end()) {
|
||||
return false;
|
||||
}
|
||||
|
||||
csrc_ids_.erase(iter);
|
||||
csrc_count(csrc_count_ - 1);
|
||||
return true;
|
||||
}
|
||||
|
||||
bool RTP::search_csrc_id(const uint32_t csrc_id) {
|
||||
csrc_ids_type::const_iterator iter = search_csrc_id_iterator(Endian::host_to_be(csrc_id));
|
||||
return (iter != csrc_ids_.cend());
|
||||
}
|
||||
|
||||
RTP::csrc_ids_type::const_iterator RTP::search_csrc_id_iterator(const uint32_t csrc_id) const {
|
||||
return std::find(csrc_ids_.cbegin(), csrc_ids_.cend(), csrc_id);
|
||||
}
|
||||
|
||||
RTP::csrc_ids_type::iterator RTP::search_csrc_id_iterator(const uint32_t csrc_id) {
|
||||
return std::find(csrc_ids_.begin(), csrc_ids_.end(), csrc_id);
|
||||
}
|
||||
|
||||
void RTP::add_extension_data(const uint32_t value) {
|
||||
if (TINS_UNLIKELY(extension_length() >= 65535)) {
|
||||
throw logic_error("Maximum number of extension data reached");
|
||||
}
|
||||
|
||||
extension_bit(1);
|
||||
ext_data_.push_back(Endian::host_to_be(value));
|
||||
extension_length(extension_length() + 1);
|
||||
}
|
||||
|
||||
bool RTP::remove_extension_data(const uint32_t value) {
|
||||
if (extension_bit() == 0 || extension_length() == 0) {
|
||||
return false;
|
||||
}
|
||||
|
||||
extension_header_data_type::iterator iter = search_extension_data_iterator(Endian::host_to_be(value));
|
||||
if (iter == ext_data_.end()) {
|
||||
return false;
|
||||
}
|
||||
|
||||
ext_data_.erase(iter);
|
||||
|
||||
extension_length(extension_length() - 1);
|
||||
|
||||
if (extension_length() == 0) {
|
||||
extension_bit(0);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool RTP::search_extension_data(const uint32_t value) {
|
||||
if (extension_bit() == 0 || extension_length() == 0) {
|
||||
return false;
|
||||
}
|
||||
|
||||
extension_header_data_type::const_iterator iter = search_extension_data_iterator(Endian::host_to_be(value));
|
||||
return (iter != ext_data_.cend());
|
||||
}
|
||||
|
||||
RTP::extension_header_data_type::const_iterator RTP::search_extension_data_iterator(const uint32_t data) const {
|
||||
return std::find(ext_data_.cbegin(), ext_data_.cend(), data);
|
||||
}
|
||||
|
||||
RTP::extension_header_data_type::iterator RTP::search_extension_data_iterator(const uint32_t data) {
|
||||
return std::find(ext_data_.begin(), ext_data_.end(), data);
|
||||
}
|
||||
|
||||
void RTP::write_serialization(uint8_t* buffer, uint32_t total_sz) {
|
||||
OutputMemoryStream stream(buffer, total_sz);
|
||||
stream.write(header_);
|
||||
|
||||
for (auto csrc_id : csrc_ids_) {
|
||||
stream.write(csrc_id);
|
||||
}
|
||||
|
||||
if (extension_bit() == 1) {
|
||||
stream.write(ext_header_);
|
||||
for (auto data : ext_data_) {
|
||||
stream.write(data);
|
||||
}
|
||||
}
|
||||
|
||||
if (padding_bit() == 1) {
|
||||
if (padding_size() > 0) {
|
||||
if (inner_pdu()) {
|
||||
stream.skip(inner_pdu()->size());
|
||||
}
|
||||
stream.fill(padding_size() - 1, 0);
|
||||
stream.write(padding_size());
|
||||
} else {
|
||||
throw pdu_not_serializable();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
} // Tins
|
||||
@@ -391,6 +391,20 @@ void Sniffer::set_rfmon(bool rfmon_enabled) {
|
||||
|
||||
// **************************** FileSniffer ****************************
|
||||
|
||||
FileSniffer::FileSniffer(FILE *fp,
|
||||
const SnifferConfiguration& configuration) {
|
||||
char error[PCAP_ERRBUF_SIZE];
|
||||
pcap_t* phandle = pcap_fopen_offline(fp, error);
|
||||
if (!phandle) {
|
||||
throw pcap_error(error);
|
||||
}
|
||||
set_pcap_handle(phandle);
|
||||
|
||||
// Configure the sniffer
|
||||
configuration.configure_sniffer_pre_activation(*this);
|
||||
|
||||
}
|
||||
|
||||
FileSniffer::FileSniffer(const string& file_name,
|
||||
const SnifferConfiguration& configuration) {
|
||||
char error[PCAP_ERRBUF_SIZE];
|
||||
@@ -420,6 +434,22 @@ FileSniffer::FileSniffer(const string& file_name, const string& filter) {
|
||||
config.configure_sniffer_pre_activation(*this);
|
||||
}
|
||||
|
||||
FileSniffer::FileSniffer(FILE *fp, const string& filter) {
|
||||
SnifferConfiguration config;
|
||||
config.set_filter(filter);
|
||||
|
||||
char error[PCAP_ERRBUF_SIZE];
|
||||
pcap_t* phandle = pcap_fopen_offline(fp, error);
|
||||
if (!phandle) {
|
||||
throw pcap_error(error);
|
||||
}
|
||||
set_pcap_handle(phandle);
|
||||
|
||||
// Configure the sniffer
|
||||
config.configure_sniffer_pre_activation(*this);
|
||||
}
|
||||
|
||||
|
||||
// ************************ SnifferConfiguration ************************
|
||||
|
||||
const unsigned SnifferConfiguration::DEFAULT_SNAP_LEN = 65535;
|
||||
|
||||
@@ -257,6 +257,10 @@ small_uint<12> TCP::flags() const {
|
||||
return (header_.res1 << 8) | header_.flags_8;
|
||||
}
|
||||
|
||||
bool TCP::has_flags(small_uint<12> check_flags) const {
|
||||
return (flags() & check_flags) == check_flags;
|
||||
}
|
||||
|
||||
void TCP::set_flag(Flags tcp_flag, small_uint<1> value) {
|
||||
switch (tcp_flag) {
|
||||
case FIN:
|
||||
|
||||
@@ -33,8 +33,6 @@
|
||||
|
||||
#include <tins/detail/sequence_number_helpers.h>
|
||||
|
||||
using std::move;
|
||||
|
||||
using Tins::Internals::seq_compare;
|
||||
|
||||
namespace Tins {
|
||||
@@ -67,7 +65,7 @@ bool DataTracker::process_payload(uint32_t seq, payload_type payload) {
|
||||
}
|
||||
bool added_some = false;
|
||||
// Store this payload
|
||||
store_payload(seq, move(payload));
|
||||
store_payload(seq, std::move(payload));
|
||||
// Keep looping while the fragments seq is lower or equal to our seq
|
||||
buffered_payload_type::iterator iter = buffered_payload_.find(seq_number_);
|
||||
while (iter != buffered_payload_.end() && seq_compare(iter->first, seq_number_) <= 0) {
|
||||
@@ -85,7 +83,7 @@ bool DataTracker::process_payload(uint32_t seq, payload_type payload) {
|
||||
payload.begin(),
|
||||
payload.begin() + (seq_number_ - iter->first)
|
||||
);
|
||||
store_payload(seq_number_, move(iter->second));
|
||||
store_payload(seq_number_, std::move(iter->second));
|
||||
iter = erase_iterator(iter);
|
||||
}
|
||||
else {
|
||||
@@ -158,14 +156,14 @@ void DataTracker::store_payload(uint32_t seq, payload_type payload) {
|
||||
// New segment, store it
|
||||
if (iter == buffered_payload_.end()) {
|
||||
total_buffered_bytes_ += payload.size();
|
||||
buffered_payload_.insert(make_pair(seq, move(payload)));
|
||||
buffered_payload_.insert(make_pair(seq, std::move(payload)));
|
||||
}
|
||||
else if (iter->second.size() < payload.size()) {
|
||||
// Increment by the diff between sizes
|
||||
total_buffered_bytes_ += (payload.size() - iter->second.size());
|
||||
// If we already have payload on this position but it's a shorter
|
||||
// chunk than the new one, replace it
|
||||
iter->second = move(payload);
|
||||
iter->second = std::move(payload);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -115,7 +115,7 @@ void Flow::process_packet(PDU& pdu) {
|
||||
}
|
||||
|
||||
// can process either way, since it will abort immediately if not needed
|
||||
if (data_tracker_.process_payload(tcp->seq(), move(raw->payload()))) {
|
||||
if (data_tracker_.process_payload(tcp->seq(), std::move(raw->payload()))) {
|
||||
if (on_data_callback_) {
|
||||
on_data_callback_(*this);
|
||||
}
|
||||
@@ -127,19 +127,19 @@ void Flow::advance_sequence(uint32_t seq) {
|
||||
}
|
||||
|
||||
void Flow::update_state(const TCP& tcp) {
|
||||
if ((tcp.flags() & TCP::FIN) != 0) {
|
||||
if (tcp.has_flags(TCP::FIN)) {
|
||||
state_ = FIN_SENT;
|
||||
}
|
||||
else if ((tcp.flags() & TCP::RST) != 0) {
|
||||
else if (tcp.has_flags(TCP::RST)) {
|
||||
state_ = RST_SENT;
|
||||
}
|
||||
else if (state_ == SYN_SENT && (tcp.flags() & TCP::ACK) != 0) {
|
||||
else if (state_ == SYN_SENT && tcp.has_flags(TCP::ACK)) {
|
||||
#ifdef TINS_HAVE_ACK_TRACKER
|
||||
ack_tracker_ = AckTracker(tcp.ack_seq());
|
||||
#endif // TINS_HAVE_ACK_TRACKER
|
||||
state_ = ESTABLISHED;
|
||||
}
|
||||
else if (state_ == UNKNOWN && (tcp.flags() & TCP::SYN) != 0) {
|
||||
else if (state_ == UNKNOWN && tcp.has_flags(TCP::SYN)) {
|
||||
// This is the server's state, sending it's first SYN|ACK
|
||||
#ifdef TINS_HAVE_ACK_TRACKER
|
||||
ack_tracker_ = AckTracker(tcp.ack_seq());
|
||||
|
||||
@@ -59,9 +59,11 @@ Stream::Stream(PDU& packet, const timestamp_type& ts)
|
||||
client_hw_addr_ = eth->src_addr();
|
||||
server_hw_addr_ = eth->dst_addr();
|
||||
}
|
||||
const TCP& tcp = packet.rfind_pdu<TCP>();
|
||||
// If this is not the first packet of a stream (SYN), then it's a partial stream
|
||||
is_partial_stream_ = tcp.flags() != TCP::SYN;
|
||||
const TCP* tcp = packet.find_pdu<TCP>();
|
||||
if (tcp) {
|
||||
// If this is not the first packet of a stream (SYN), then it's a partial stream
|
||||
is_partial_stream_ = !tcp->has_flags(TCP::SYN);
|
||||
}
|
||||
}
|
||||
|
||||
void Stream::process_packet(PDU& packet, const timestamp_type& ts) {
|
||||
|
||||
@@ -84,7 +84,9 @@ void StreamFollower::process_packet(PDU& packet, const timestamp_type& ts) {
|
||||
if (iter == streams_.end()) {
|
||||
// Start tracking if they're either SYNs or they contain data (attach
|
||||
// to an already running flow).
|
||||
if (tcp->flags() == TCP::SYN || (attach_to_flows_ && tcp->find_pdu<RawPDU>() != 0)) {
|
||||
// Start on client's SYN, not on server's SYN+ACK
|
||||
const bool is_syn = tcp->has_flags(TCP::SYN) && !tcp->has_flags(TCP::ACK);
|
||||
if (is_syn || (attach_to_flows_ && tcp->find_pdu<RawPDU>() != 0)) {
|
||||
iter = streams_.insert(make_pair(identifier, Stream(packet, ts))).first;
|
||||
iter->second.setup_flows_callbacks();
|
||||
if (on_new_connection_) {
|
||||
@@ -93,7 +95,7 @@ void StreamFollower::process_packet(PDU& packet, const timestamp_type& ts) {
|
||||
else {
|
||||
throw callback_not_set();
|
||||
}
|
||||
if (tcp->flags() != TCP::SYN) {
|
||||
if (!is_syn) {
|
||||
// assume the connection is established
|
||||
iter->second.client_flow().state(Flow::ESTABLISHED);
|
||||
iter->second.server_flow().state(Flow::ESTABLISHED);
|
||||
|
||||
@@ -230,7 +230,7 @@ bool TCPStream::generic_process(uint32_t& my_seq,
|
||||
|
||||
bool TCPStream::update(IP* ip, TCP* tcp) {
|
||||
if (!syn_ack_sent_) {
|
||||
if (tcp->flags() == (TCP::SYN | TCP::ACK)) {
|
||||
if (tcp->has_flags(TCP::SYN | TCP::ACK)) {
|
||||
server_seq_ = tcp->seq() + 1;
|
||||
client_seq_ = tcp->ack_seq();
|
||||
syn_ack_sent_ = true;
|
||||
|
||||
@@ -65,7 +65,7 @@ const RadioTapParser::FieldMetadata RadioTapParser::RADIOTAP_METADATA[] = {
|
||||
};
|
||||
|
||||
const uint32_t RadioTapParser::MAX_RADIOTAP_FIELD = sizeof(RADIOTAP_METADATA) /
|
||||
sizeof(FieldMetadata) + 1;
|
||||
sizeof(FieldMetadata);
|
||||
|
||||
#if TINS_IS_LITTLE_ENDIAN
|
||||
TINS_BEGIN_PACK
|
||||
@@ -131,13 +131,10 @@ struct RadioTapFlags {
|
||||
} TINS_END_PACK;
|
||||
#endif
|
||||
|
||||
void align_buffer(const uint8_t* buffer_start, const uint8_t*& buffer, uint32_t size, size_t n) {
|
||||
void align_buffer(const uint8_t* buffer_start, const uint8_t*& buffer, size_t n) {
|
||||
uint32_t offset = (buffer - buffer_start) & (n - 1);
|
||||
if (offset) {
|
||||
offset = n - offset;
|
||||
if (TINS_UNLIKELY(offset > size)) {
|
||||
throw malformed_packet();
|
||||
}
|
||||
buffer += offset;
|
||||
}
|
||||
}
|
||||
@@ -281,7 +278,7 @@ bool RadioTapParser::advance_to_next_field() {
|
||||
if (current_bit_ < MAX_RADIOTAP_FIELD) {
|
||||
const uint8_t* radiotap_start = start_ - sizeof(uint32_t);
|
||||
// Skip and align the buffer
|
||||
align_buffer(radiotap_start, current_ptr_, end_ - radiotap_start,
|
||||
align_buffer(radiotap_start, current_ptr_,
|
||||
RADIOTAP_METADATA[current_bit_].alignment);
|
||||
return true;
|
||||
}
|
||||
|
||||
@@ -41,11 +41,12 @@ namespace Tins {
|
||||
namespace Utils {
|
||||
|
||||
uint32_t calculate_padding(uint32_t alignment, uint32_t offset) {
|
||||
return offset % alignment;
|
||||
uint32_t extra = offset % alignment;
|
||||
return extra == 0 ? 0 : alignment - extra;
|
||||
}
|
||||
|
||||
uint32_t get_bit(uint32_t value) {
|
||||
return log(value) / log(2);
|
||||
return round(log2(value));
|
||||
}
|
||||
|
||||
RadioTapWriter::RadioTapWriter(vector<uint8_t>& buffer)
|
||||
@@ -54,7 +55,7 @@ RadioTapWriter::RadioTapWriter(vector<uint8_t>& buffer)
|
||||
|
||||
void RadioTapWriter::write_option(const RadioTap::option& option) {
|
||||
const uint32_t bit = get_bit(option.option());
|
||||
if (bit > RadioTapParser::MAX_RADIOTAP_FIELD) {
|
||||
if (bit >= RadioTapParser::MAX_RADIOTAP_FIELD) {
|
||||
throw malformed_option();
|
||||
}
|
||||
const bool is_empty = buffer_.empty();
|
||||
|
||||
@@ -154,6 +154,7 @@ vector<char> query_route_table(int family) {
|
||||
throw exception_base("sysctl failed");
|
||||
}
|
||||
|
||||
buf.resize(len);
|
||||
return buf;
|
||||
}
|
||||
|
||||
@@ -412,17 +413,21 @@ set<string> network_interfaces() {
|
||||
}
|
||||
#else
|
||||
set<string> network_interfaces() {
|
||||
set<string> output;
|
||||
struct ifaddrs* ifaddrs = 0;
|
||||
struct ifaddrs* if_it = 0;
|
||||
getifaddrs(&ifaddrs);
|
||||
for (if_it = ifaddrs; if_it; if_it = if_it->ifa_next) {
|
||||
output.insert(if_it->ifa_name);
|
||||
}
|
||||
if (ifaddrs) {
|
||||
freeifaddrs(ifaddrs);
|
||||
}
|
||||
#if !defined(ANDROID) || (defined(__ANDROID_API__) && __ANDROID_API__ >= 24)
|
||||
set<string> output;
|
||||
struct ifaddrs* ifaddrs = 0;
|
||||
struct ifaddrs* if_it = 0;
|
||||
getifaddrs(&ifaddrs);
|
||||
for (if_it = ifaddrs; if_it; if_it = if_it->ifa_next) {
|
||||
output.insert(if_it->ifa_name);
|
||||
}
|
||||
if (ifaddrs) {
|
||||
freeifaddrs(ifaddrs);
|
||||
}
|
||||
return output;
|
||||
#else
|
||||
throw std::runtime_error("android ifaddr not supported");
|
||||
#endif
|
||||
}
|
||||
#endif // _WIN32
|
||||
|
||||
|
||||
36
src/vxlan.cpp
Normal file
36
src/vxlan.cpp
Normal file
@@ -0,0 +1,36 @@
|
||||
#include <tins/internals.h>
|
||||
#include <tins/memory_helpers.h>
|
||||
#include <tins/small_uint.h>
|
||||
#include <tins/vxlan.h>
|
||||
|
||||
using Tins::Memory::InputMemoryStream;
|
||||
using Tins::Memory::OutputMemoryStream;
|
||||
|
||||
namespace Tins {
|
||||
|
||||
VXLAN::VXLAN(const small_uint<24> vni) {
|
||||
set_flags(8);
|
||||
set_vni(vni);
|
||||
}
|
||||
|
||||
VXLAN::VXLAN(const uint8_t* buffer, uint32_t total_sz) {
|
||||
InputMemoryStream stream(buffer, total_sz);
|
||||
stream.read(header_);
|
||||
// If there is any size left
|
||||
if (stream) {
|
||||
inner_pdu(
|
||||
Internals::pdu_from_flag(
|
||||
PDU::ETHERNET_II,
|
||||
stream.pointer(),
|
||||
stream.size()
|
||||
)
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
void VXLAN::write_serialization(uint8_t* buffer, uint32_t total_sz) {
|
||||
OutputMemoryStream stream(buffer, total_sz);
|
||||
stream.write(header_);
|
||||
}
|
||||
|
||||
} // Tins
|
||||
@@ -62,7 +62,7 @@ private:
|
||||
|
||||
template <typename T>
|
||||
void ActiveTestRunner::add_test() {
|
||||
tests_.emplace_back(new T(packet_sender_, configuration_));
|
||||
tests_.emplace_back(std::unique_ptr<T>(new T(packet_sender_, configuration_)));
|
||||
}
|
||||
|
||||
#endif // TINS_ACTIVE_TEST_RUNNER_H
|
||||
|
||||
@@ -63,6 +63,7 @@ CREATE_TEST(pppoe)
|
||||
CREATE_TEST(raw_pdu)
|
||||
CREATE_TEST(rc4_eapol)
|
||||
CREATE_TEST(rsn_eapol)
|
||||
CREATE_TEST(rtp)
|
||||
CREATE_TEST(sll)
|
||||
CREATE_TEST(snap)
|
||||
CREATE_TEST(stp)
|
||||
@@ -70,6 +71,7 @@ CREATE_TEST(tcp)
|
||||
CREATE_TEST(tcp_ip)
|
||||
CREATE_TEST(udp)
|
||||
CREATE_TEST(utils)
|
||||
CREATE_TEST(vxlan)
|
||||
|
||||
IF(LIBTINS_ENABLE_PCAP)
|
||||
CREATE_TEST(offline_packet_filter)
|
||||
|
||||
@@ -12,11 +12,19 @@ using namespace Tins;
|
||||
|
||||
class AddressRangeTest : public testing::Test {
|
||||
public:
|
||||
void contain_tests0(const IPv4Range& range);
|
||||
void contain_tests0(const IPv6Range& range);
|
||||
void contain_tests24(const IPv4Range& range);
|
||||
void contain_tests24(const IPv6Range& range);
|
||||
void contain_tests26(const IPv4Range& range);
|
||||
};
|
||||
|
||||
void AddressRangeTest::contain_tests0(const IPv4Range& range) {
|
||||
EXPECT_TRUE(range.contains("0.0.0.0"));
|
||||
EXPECT_TRUE(range.contains("192.168.1.1"));
|
||||
EXPECT_TRUE(range.contains("255.255.255.255"));
|
||||
}
|
||||
|
||||
void AddressRangeTest::contain_tests24(const IPv4Range& range) {
|
||||
EXPECT_TRUE(range.contains("192.168.0.0"));
|
||||
EXPECT_TRUE(range.contains("192.168.0.1"));
|
||||
@@ -33,6 +41,12 @@ void AddressRangeTest::contain_tests26(const IPv4Range& range) {
|
||||
EXPECT_FALSE(range.contains("192.168.254.191"));
|
||||
}
|
||||
|
||||
void AddressRangeTest::contain_tests0(const IPv6Range& range) {
|
||||
EXPECT_TRUE(range.contains("::"));
|
||||
EXPECT_TRUE(range.contains("dead::1:1"));
|
||||
EXPECT_TRUE(range.contains("ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff"));
|
||||
}
|
||||
|
||||
void AddressRangeTest::contain_tests24(const IPv6Range& range) {
|
||||
EXPECT_TRUE(range.contains("dead::1"));
|
||||
EXPECT_TRUE(range.contains("dead::1fee"));
|
||||
@@ -42,13 +56,22 @@ void AddressRangeTest::contain_tests24(const IPv6Range& range) {
|
||||
}
|
||||
|
||||
TEST_F(AddressRangeTest, Contains) {
|
||||
contain_tests0(IPv4Range("0.0.0.0", "255.255.255.255"));
|
||||
contain_tests0(IPv4Range::from_mask("0.0.0.0", "0.0.0.0"));
|
||||
contain_tests0(IPv4Range::from_mask("0.0.0.0", IPv4Address::from_prefix_length(0)));
|
||||
contain_tests24(IPv4Range("192.168.0.0", "192.168.0.255"));
|
||||
contain_tests24(IPv4Range::from_mask("192.168.0.0", "255.255.255.0"));
|
||||
contain_tests24(IPv4Range::from_mask("192.168.0.0", IPv4Address::from_prefix_length(24)));
|
||||
contain_tests26(IPv4Range("192.168.254.192", "192.168.254.255"));
|
||||
contain_tests26(IPv4Range::from_mask("192.168.254.192", "255.255.255.192"));
|
||||
contain_tests26(IPv4Range::from_mask("192.168.254.192", IPv4Address::from_prefix_length(26)));
|
||||
|
||||
contain_tests0(IPv6Range("::0", "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff"));
|
||||
contain_tests0(IPv6Range::from_mask("::", "::"));
|
||||
contain_tests0(IPv6Range::from_mask("::", IPv6Address::from_prefix_length(0)));
|
||||
contain_tests24(IPv6Range("dead::0", "dead::ffff"));
|
||||
contain_tests24(IPv6Range::from_mask("dead::0", "ffff:ffff:ffff:ffff:ffff:ffff:ffff:0"));
|
||||
contain_tests24(IPv6Range::from_mask("dead::0", IPv6Address::from_prefix_length(112)));
|
||||
|
||||
{
|
||||
AddressRange<HWAddress<6> > range("00:00:00:00:00:00", "00:00:00:00:00:ff");
|
||||
|
||||
@@ -58,7 +58,7 @@ const uint8_t DHCPTest::expected_packet[] = {
|
||||
TEST_F(DHCPTest, DefaultConstructor) {
|
||||
DHCP dhcp;
|
||||
EXPECT_EQ(dhcp.htype(), 1);
|
||||
EXPECT_EQ(dhcp.hlen(), (const size_t)EthernetII::address_type::address_size);
|
||||
EXPECT_EQ(dhcp.hlen(), (size_t)EthernetII::address_type::address_size);
|
||||
}
|
||||
|
||||
TEST_F(DHCPTest, CopyConstructor) {
|
||||
@@ -280,7 +280,7 @@ TEST_F(DHCPTest, ConstructorFromBuffer) {
|
||||
|
||||
EXPECT_EQ(dhcp1.opcode(), DHCP::DISCOVER);
|
||||
EXPECT_EQ(dhcp1.htype(), 1);
|
||||
ASSERT_EQ(dhcp1.hlen(), (const size_t)EthernetII::address_type::address_size);
|
||||
ASSERT_EQ(dhcp1.hlen(), (size_t)EthernetII::address_type::address_size);
|
||||
EXPECT_EQ(dhcp1.hops(), 0x1f);
|
||||
EXPECT_EQ(dhcp1.xid(), 0x3fab23deU);
|
||||
EXPECT_EQ(dhcp1.secs(), 0x9f1a);
|
||||
|
||||
@@ -551,3 +551,113 @@ TEST_F(DNSTest, SOARecordSerialize) {
|
||||
EXPECT_EQ(0x8ad71928U, r2.expire());
|
||||
EXPECT_EQ(0x1ad92871U, r2.minimum_ttl());
|
||||
}
|
||||
|
||||
TEST_F(DNSTest, BadLabelSize) {
|
||||
const uint8_t header[] = {
|
||||
0x45, 0xbc, // ID
|
||||
0x81, 0x80, // response, recursion desired, recursion available, no error
|
||||
0x00, 0x01, // QDCOUNT
|
||||
0x00, 0x00, // ANCOUNT
|
||||
0x00, 0x00, // NSCOUNT
|
||||
0x00, 0x00 // ARCOUNT
|
||||
};
|
||||
size_t payload_sz{sizeof(header)};
|
||||
uint8_t payload[512];
|
||||
|
||||
// copy header
|
||||
std::copy(header,
|
||||
header + payload_sz,
|
||||
payload);
|
||||
|
||||
// add bad length
|
||||
const size_t bad_label_len{0x80};
|
||||
const size_t label_offset = payload_sz;
|
||||
payload[payload_sz++] = bad_label_len;
|
||||
|
||||
// fill label for incorrect length and terminate
|
||||
std::fill(payload + payload_sz,
|
||||
payload + payload_sz + bad_label_len,
|
||||
'a');
|
||||
payload_sz += bad_label_len;
|
||||
payload[payload_sz++] = 0x0;
|
||||
|
||||
// add type and class
|
||||
const uint8_t type_class[] = {
|
||||
0x00, 0x01,
|
||||
0x00, 0x01
|
||||
};
|
||||
std::copy(type_class,
|
||||
type_class + sizeof(type_class),
|
||||
payload + payload_sz);
|
||||
payload_sz += sizeof(type_class);
|
||||
|
||||
// invalid high two bits of label first octest is detected early now
|
||||
try {
|
||||
const DNS packet(payload, payload_sz);
|
||||
FAIL();
|
||||
} catch (malformed_packet& mp) {
|
||||
SUCCEED();
|
||||
}
|
||||
|
||||
// check the other invalid value of high two bits in label size
|
||||
payload[label_offset] = 0x10;
|
||||
try {
|
||||
const DNS packet(payload, payload_sz);
|
||||
FAIL();
|
||||
} catch (malformed_packet& mp) {
|
||||
SUCCEED();
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(DNSTest, BadPacketLength) {
|
||||
|
||||
// valid response packet with RR's in all sections
|
||||
const uint8_t payload[] = {
|
||||
0x74,0xa9,0x85,0x80,0x00,0x01,0x00,0x02,0x00,0x01,0x00,0x04,0x08,0x5f,0x73,0x65,0x72,
|
||||
0x76,0x69,0x63,0x65,0x04,0x5f,0x74,0x63,0x70,0x05,0x77,0x69,0x66,0x69,0x36,0x03,
|
||||
0x6c,0x61,0x6e,0x00,0x00,0x21,0x00,0x01,0xc0,0x0c,0x00,0x21,0x00,0x01,0x00,0x01,
|
||||
0x51,0x80,0x00,0x16,0x00,0x00,0x00,0x03,0x00,0x09,0x04,0x66,0x61,0x73,0x74,0x05,
|
||||
0x77,0x69,0x66,0x69,0x36,0x03,0x6c,0x61,0x6e,0x00,0xc0,0x0c,0x00,0x21,0x00,0x01,
|
||||
0x00,0x01,0x51,0x80,0x00,0x16,0x00,0x00,0x00,0x01,0x00,0x09,0x04,0x73,0x6c,0x6f,
|
||||
0x77,0x05,0x77,0x69,0x66,0x69,0x36,0x03,0x6c,0x61,0x6e,0x00,0xc0,0x62,0x00,0x02,
|
||||
0x00,0x01,0x00,0x01,0x51,0x80,0x00,0x05,0x02,0x70,0x69,0xc0,0x62,0xc0,0x5d,0x00,
|
||||
0x01,0x00,0x01,0x00,0x01,0x51,0x80,0x00,0x04,0x0a,0x18,0x00,0x02,0xc0,0x3b,0x00,
|
||||
0x01,0x00,0x01,0x00,0x01,0x51,0x80,0x00,0x04,0x0a,0x18,0x00,0x02,0xc0,0x79,0x00,
|
||||
0x01,0x00,0x01,0x00,0x01,0x51,0x80,0x00,0x04,0x0a,0x18,0x00,0x02,0x00,0x00,0x29,
|
||||
0x10,0x00,0x00,0x00,0x00,0x00,0x00,0x1c,0x00,0x0a,0x00,0x18,0x86,0x1f,0x14,0x0f,
|
||||
0x41,0xfa,0xf3,0x95,0x48,0x6e,0x79,0x61,0x61,0x78,0x32,0x0f,0x44,0x5d,0x21,0x47,
|
||||
0x85,0x83,0x9a,0x95
|
||||
};
|
||||
|
||||
// valid DNS message but misreport packet size;
|
||||
// before fix, parser headed into uncharted waters on requesting additional section
|
||||
|
||||
// buffer with space for valid packet plus garbage bytes
|
||||
const size_t bigsz{512};
|
||||
uint8_t big_packet[bigsz];
|
||||
|
||||
// copy valid packet
|
||||
std::copy(payload,
|
||||
payload + sizeof(payload),
|
||||
big_packet);
|
||||
|
||||
// fill additional bytes with junk
|
||||
std::fill(big_packet + sizeof(payload),
|
||||
big_packet + bigsz,
|
||||
0x5A);
|
||||
|
||||
// initial packet parse ok
|
||||
const DNS packet(big_packet, bigsz);
|
||||
|
||||
// RR's parse ok now
|
||||
EXPECT_EQ(packet.questions_count(), 1);
|
||||
EXPECT_EQ(packet.answers_count(), 2);
|
||||
EXPECT_EQ(packet.authority_count(), 1);
|
||||
EXPECT_EQ(packet.additional_count(), 4);
|
||||
EXPECT_EQ(packet.queries().size(), 1U);
|
||||
EXPECT_EQ(packet.answers().size(), 2U);
|
||||
EXPECT_EQ(packet.authority().size(), 1U);
|
||||
EXPECT_EQ(packet.additional().size(), 4U);
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -759,4 +759,27 @@ TEST_F(RadioTapTest, RadioTapWritingEmptyBuffer) {
|
||||
|
||||
}
|
||||
|
||||
TEST_F(RadioTapTest, RadioTapWritingInvalidOption) {
|
||||
vector<uint8_t> buffer(4, 0);
|
||||
RadioTapWriter writer(buffer);
|
||||
uint8_t foo = 0;
|
||||
RadioTap::option option((RadioTap::PresentFlags)(1 << RadioTapParser::MAX_RADIOTAP_FIELD), sizeof(foo), &foo);
|
||||
EXPECT_THROW(writer.write_option(option), malformed_option);
|
||||
}
|
||||
|
||||
TEST_F(RadioTapTest, RadioTapWriterAlignment) {
|
||||
vector<uint8_t> buffer(4, 0);
|
||||
RadioTapWriter writer(buffer);
|
||||
uint8_t flags = 10;
|
||||
uint8_t xchannel[sizeof(RadioTap::xchannel_type)] = {
|
||||
1, 2, 3, 4, 5, 6, 7, 8
|
||||
};
|
||||
writer.write_option(RadioTap::option(RadioTap::FLAGS, sizeof(flags), &flags));
|
||||
writer.write_option(RadioTap::option(RadioTap::XCHANNEL, sizeof(xchannel), xchannel));
|
||||
vector<uint8_t> expected = {
|
||||
2, 0, 4, 0, 10, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8
|
||||
};
|
||||
EXPECT_EQ(buffer, expected);
|
||||
}
|
||||
|
||||
#endif // TINS_HAVE_DOT11
|
||||
|
||||
292
tests/src/rtp_test.cpp
Normal file
292
tests/src/rtp_test.cpp
Normal file
@@ -0,0 +1,292 @@
|
||||
#include <gtest/gtest.h>
|
||||
#include <string>
|
||||
#include <tins/endianness.h>
|
||||
#include <tins/ethernetII.h>
|
||||
#include <tins/ip.h>
|
||||
#include <tins/udp.h>
|
||||
#include <tins/pdu.h>
|
||||
#include <tins/rawpdu.h>
|
||||
#include <tins/small_uint.h>
|
||||
#include <tins/rtp.h>
|
||||
|
||||
#define PACKET_SIZE 60ul
|
||||
#define CSRC_COUNT 5
|
||||
#define EXTENSION_LENGTH 2
|
||||
#define PAYLOAD_SIZE 12
|
||||
|
||||
using namespace std;
|
||||
using namespace Tins;
|
||||
|
||||
class RTPTest : public testing::Test {
|
||||
public:
|
||||
static const uint8_t expected_packet[PACKET_SIZE];
|
||||
static const uint8_t invalid_packet_one[];
|
||||
static const uint8_t invalid_packet_two[];
|
||||
static const uint8_t packet_with_zero_padding_value[];
|
||||
static const uint8_t packet_without_data_one[];
|
||||
static const uint8_t packet_without_data_two[];
|
||||
static const uint8_t packet_with_zero_extension_length[];
|
||||
static const small_uint<2> version;
|
||||
static const small_uint<1> padding;
|
||||
static const small_uint<1> extension;
|
||||
static const small_uint<4> csrc_count;
|
||||
static const small_uint<1> marker;
|
||||
static const small_uint<7> payload_type;
|
||||
static const uint16_t sequence_number;
|
||||
static const uint32_t timestamp;
|
||||
static const uint32_t ssrc_id;
|
||||
static const uint32_t csrc_ids[CSRC_COUNT];
|
||||
static const uint16_t profile;
|
||||
static const uint16_t extension_length;
|
||||
static const uint32_t extension_data[EXTENSION_LENGTH];
|
||||
static const uint8_t padding_size;
|
||||
static const uint8_t payload[PAYLOAD_SIZE];
|
||||
static const uint16_t dport, sport;
|
||||
static const IP::address_type dst_ip, src_ip;
|
||||
static const EthernetII::address_type dst_addr, src_addr;
|
||||
};
|
||||
|
||||
const uint8_t RTPTest::expected_packet[PACKET_SIZE] = {
|
||||
0xb5, 0xaa, 0xa4, 0x10,
|
||||
0xde, 0xad, 0xbe, 0xef,
|
||||
0xab, 0xcd, 0xad, 0xbc,
|
||||
0x00, 0x00, 0x00, 0x01,
|
||||
0x00, 0x00, 0x00, 0x02,
|
||||
0x00, 0x00, 0x00, 0x03,
|
||||
0x00, 0x00, 0x00, 0x04,
|
||||
0x00, 0x00, 0x00, 0x05,
|
||||
0x01, 0x01, 0x00, 0x02,
|
||||
0x77, 0x00, 0x00, 0x00,
|
||||
0x88, 0x00, 0x00, 0x00,
|
||||
0x42, 0x42, 0x42, 0x42,
|
||||
0x42, 0x42, 0x42, 0x42,
|
||||
0x42, 0x42, 0x42, 0x42,
|
||||
0x00, 0x00, 0x00, 0x04,
|
||||
};
|
||||
|
||||
const uint8_t RTPTest::invalid_packet_one[] = {
|
||||
160, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 0xff,
|
||||
};
|
||||
|
||||
const uint8_t RTPTest::invalid_packet_two[] = {
|
||||
160, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1,
|
||||
};
|
||||
|
||||
const uint8_t RTPTest::packet_with_zero_padding_value[] = {
|
||||
160, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0,
|
||||
};
|
||||
|
||||
const uint8_t RTPTest::packet_without_data_one[] = {
|
||||
128, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1,
|
||||
};
|
||||
|
||||
const uint8_t RTPTest::packet_without_data_two[] = {
|
||||
160, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 7,
|
||||
};
|
||||
|
||||
const uint8_t RTPTest::packet_with_zero_extension_length[] = {
|
||||
144, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 0x56, 0x97, 0, 0,
|
||||
};
|
||||
|
||||
const small_uint<2> RTPTest::version = 2;
|
||||
const small_uint<1> RTPTest::padding = 1;
|
||||
const small_uint<1> RTPTest::extension = 1;
|
||||
const small_uint<4> RTPTest::csrc_count = CSRC_COUNT;
|
||||
const small_uint<1> RTPTest::marker = 1;
|
||||
const small_uint<7> RTPTest::payload_type = 42;
|
||||
const uint16_t RTPTest::sequence_number = 42000;
|
||||
const uint32_t RTPTest::timestamp = 0xdeadbeef;
|
||||
const uint32_t RTPTest::ssrc_id = 0xabcdadbc;
|
||||
const uint32_t RTPTest::csrc_ids[CSRC_COUNT] = { 1, 2, 3, 4, 5 };
|
||||
const uint16_t RTPTest::profile = 0x0101;
|
||||
const uint16_t RTPTest::extension_length = EXTENSION_LENGTH;
|
||||
const uint32_t RTPTest::extension_data[EXTENSION_LENGTH] = { 0x77000000, 0x88000000 };
|
||||
const uint8_t RTPTest::padding_size = 4;
|
||||
const uint8_t RTPTest::payload[PAYLOAD_SIZE] = {
|
||||
0x42, 0x42, 0x42, 0x42,
|
||||
0x42, 0x42, 0x42, 0x42,
|
||||
0x42, 0x42, 0x42, 0x42,
|
||||
};
|
||||
const uint16_t RTPTest::dport = 5004;
|
||||
const uint16_t RTPTest::sport = 30000;
|
||||
const IP::address_type RTPTest::dst_ip = IP::address_type{"2.2.2.2"};
|
||||
const IP::address_type RTPTest::src_ip = IP::address_type{"1.1.1.1"};
|
||||
const EthernetII::address_type RTPTest::dst_addr = EthernetII::address_type{"aa:bb:cc:dd:ee:ff"};
|
||||
const EthernetII::address_type RTPTest::src_addr = EthernetII::address_type{"8a:8b:8c:8d:8e:8f"};
|
||||
|
||||
TEST_F(RTPTest, DefaultConstructor) {
|
||||
auto const rtp = RTP{};
|
||||
EXPECT_EQ(rtp.version(), version);
|
||||
EXPECT_EQ(rtp.padding_bit(), 0);
|
||||
EXPECT_EQ(rtp.extension_bit(), 0);
|
||||
EXPECT_EQ(rtp.csrc_count(), 0);
|
||||
EXPECT_EQ(rtp.marker_bit(), 0);
|
||||
EXPECT_EQ(rtp.payload_type(), 0);
|
||||
EXPECT_EQ(rtp.sequence_number(), 0);
|
||||
EXPECT_EQ(rtp.timestamp(), 0);
|
||||
EXPECT_EQ(rtp.ssrc_id(), 0);
|
||||
EXPECT_EQ(rtp.csrc_ids().size(), 0);
|
||||
EXPECT_EQ(rtp.extension_profile(), 0);
|
||||
EXPECT_EQ(rtp.extension_length(), 0);
|
||||
EXPECT_EQ(rtp.extension_data().size(), 0);
|
||||
EXPECT_EQ(rtp.padding_size(), 0);
|
||||
EXPECT_EQ(rtp.header_size(), 12);
|
||||
EXPECT_EQ(rtp.trailer_size(), 0);
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, Serialize) {
|
||||
auto rtp = RTP{};
|
||||
rtp.version(version);
|
||||
rtp.padding_size(padding_size);
|
||||
rtp.extension_bit(extension);
|
||||
rtp.marker_bit(marker);
|
||||
rtp.payload_type(payload_type);
|
||||
rtp.sequence_number(sequence_number);
|
||||
rtp.timestamp(timestamp);
|
||||
rtp.ssrc_id(ssrc_id);
|
||||
|
||||
for (auto csrc_id : csrc_ids) {
|
||||
rtp.add_csrc_id(csrc_id);
|
||||
}
|
||||
|
||||
rtp.extension_profile(profile);
|
||||
|
||||
for (auto data : extension_data) {
|
||||
rtp.add_extension_data(data);
|
||||
}
|
||||
|
||||
auto raw_pdu = RawPDU(payload, PAYLOAD_SIZE);
|
||||
rtp.inner_pdu(raw_pdu);
|
||||
|
||||
EXPECT_EQ(rtp.header_size(), PACKET_SIZE - PAYLOAD_SIZE - padding_size);
|
||||
EXPECT_EQ(rtp.trailer_size(), padding_size);
|
||||
|
||||
auto serialized = rtp.serialize();
|
||||
ASSERT_EQ(serialized.size(), PACKET_SIZE);
|
||||
EXPECT_TRUE(std::equal(serialized.begin(), serialized.end(), expected_packet));
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, ConstructorFromBuffer) {
|
||||
auto rtp = RTP{expected_packet, PACKET_SIZE};
|
||||
EXPECT_EQ(rtp.version(), version);
|
||||
EXPECT_EQ(rtp.padding_bit(), padding);
|
||||
EXPECT_EQ(rtp.extension_bit(), extension);
|
||||
EXPECT_EQ(rtp.csrc_count(), csrc_count);
|
||||
EXPECT_EQ(rtp.marker_bit(), marker);
|
||||
EXPECT_EQ(rtp.payload_type(), payload_type);
|
||||
EXPECT_EQ(rtp.sequence_number(), sequence_number);
|
||||
EXPECT_EQ(rtp.timestamp(), timestamp);
|
||||
EXPECT_EQ(rtp.ssrc_id(), ssrc_id);
|
||||
|
||||
auto csrc_id_values = rtp.csrc_ids();
|
||||
for (size_t i = 0; i < csrc_count; ++i) {
|
||||
EXPECT_EQ(csrc_id_values[i], Endian::host_to_be(csrc_ids[i]));
|
||||
}
|
||||
|
||||
EXPECT_EQ(rtp.extension_profile(), profile);
|
||||
EXPECT_EQ(rtp.extension_length(), extension_length);
|
||||
|
||||
auto extension_data_values = rtp.extension_data();
|
||||
for (size_t i = 0; i < extension_length; ++i) {
|
||||
EXPECT_EQ(extension_data_values[i], Endian::host_to_be(extension_data[i]));
|
||||
}
|
||||
|
||||
EXPECT_EQ(rtp.padding_size(), padding_size);
|
||||
EXPECT_EQ(rtp.header_size(), PACKET_SIZE - PAYLOAD_SIZE - padding_size);
|
||||
|
||||
auto inner_pdu_payload = rtp.inner_pdu()->serialize();
|
||||
EXPECT_TRUE(std::equal(inner_pdu_payload.begin(), inner_pdu_payload.end(), payload));
|
||||
|
||||
auto raw_pdu = RawPDU(payload, PAYLOAD_SIZE);
|
||||
auto raw_pdu_payload = raw_pdu.serialize();
|
||||
EXPECT_EQ(rtp.inner_pdu()->size(), raw_pdu.size());
|
||||
EXPECT_EQ(inner_pdu_payload, raw_pdu_payload);
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, SearchAndRemoveCSRCID) {
|
||||
auto rtp = RTP{};
|
||||
|
||||
for (auto csrc_id : csrc_ids) {
|
||||
rtp.add_csrc_id(csrc_id);
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < csrc_count; ++i) {
|
||||
EXPECT_TRUE(rtp.search_csrc_id(csrc_ids[i]));
|
||||
}
|
||||
|
||||
EXPECT_FALSE(rtp.search_csrc_id(0));
|
||||
EXPECT_FALSE(rtp.remove_csrc_id(0));
|
||||
EXPECT_TRUE(rtp.remove_csrc_id(csrc_ids[0]));
|
||||
EXPECT_FALSE(rtp.search_csrc_id(csrc_ids[0]));
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, SearchAndRemoveExtensionData) {
|
||||
auto rtp = RTP{};
|
||||
|
||||
for (auto data : extension_data) {
|
||||
rtp.add_extension_data(data);
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < extension_length; ++i) {
|
||||
EXPECT_TRUE(rtp.search_extension_data(extension_data[i]));
|
||||
}
|
||||
|
||||
EXPECT_FALSE(rtp.search_extension_data(0));
|
||||
EXPECT_FALSE(rtp.remove_extension_data(0));
|
||||
EXPECT_TRUE(rtp.remove_extension_data(extension_data[0]));
|
||||
EXPECT_FALSE(rtp.search_extension_data(extension_data[0]));
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, OuterUDP) {
|
||||
auto pkt = EthernetII{dst_addr, src_addr} / IP{dst_ip, src_ip} / UDP{dport, sport} / RTP{expected_packet, PACKET_SIZE};
|
||||
|
||||
auto udp = pkt.find_pdu<UDP>();
|
||||
ASSERT_TRUE(udp != nullptr);
|
||||
EXPECT_EQ(udp->dport(), dport);
|
||||
EXPECT_EQ(udp->sport(), sport);
|
||||
|
||||
auto rtp = udp->find_pdu<RTP>();
|
||||
ASSERT_TRUE(rtp != nullptr);
|
||||
EXPECT_EQ(rtp->header_size(), PACKET_SIZE - PAYLOAD_SIZE - padding_size);
|
||||
EXPECT_EQ(rtp->trailer_size(), padding_size);
|
||||
EXPECT_EQ(rtp->size(), PACKET_SIZE);
|
||||
EXPECT_EQ(rtp->inner_pdu()->size(), PAYLOAD_SIZE);
|
||||
auto inner_pdu_payload = rtp->inner_pdu()->serialize();
|
||||
EXPECT_TRUE(std::equal(inner_pdu_payload.begin(), inner_pdu_payload.end(), payload));
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, PaddingSizeTooLarge) {
|
||||
EXPECT_THROW((RTP{invalid_packet_one, sizeof(invalid_packet_one)}), malformed_packet);
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, PaddingBitSetWithoutPadding) {
|
||||
EXPECT_THROW((RTP{invalid_packet_two, sizeof(invalid_packet_two)}), malformed_packet);
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, PacketWithInvalidZeroPaddingValue) {
|
||||
EXPECT_THROW((RTP{packet_with_zero_padding_value, sizeof(packet_with_zero_padding_value)}), malformed_packet);
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, PacketWithoutData) {
|
||||
auto rtp = RTP{packet_without_data_one, sizeof(packet_without_data_one)};
|
||||
EXPECT_EQ(rtp.size(), sizeof(packet_without_data_one));
|
||||
EXPECT_EQ(rtp.header_size(), sizeof(packet_without_data_one));
|
||||
EXPECT_EQ(rtp.inner_pdu(), nullptr);
|
||||
EXPECT_EQ(rtp.padding_size(), 0);
|
||||
|
||||
const uint8_t padding_size_ = 7;
|
||||
rtp = RTP{packet_without_data_two, sizeof(packet_without_data_two)};
|
||||
EXPECT_EQ(rtp.size(), sizeof(packet_without_data_two));
|
||||
EXPECT_EQ(rtp.header_size(), sizeof(packet_without_data_two) - padding_size_);
|
||||
EXPECT_EQ(rtp.inner_pdu(), nullptr);
|
||||
EXPECT_EQ(rtp.padding_size(), padding_size_);
|
||||
}
|
||||
|
||||
TEST_F(RTPTest, PacketWithZeroExtensionLength) {
|
||||
auto rtp = RTP{packet_with_zero_extension_length, sizeof(packet_with_zero_extension_length)};
|
||||
EXPECT_EQ(rtp.size(), sizeof(packet_with_zero_extension_length));
|
||||
EXPECT_EQ(rtp.header_size(), sizeof(packet_with_zero_extension_length));
|
||||
EXPECT_EQ(rtp.extension_profile(), 0x5697);
|
||||
EXPECT_EQ(rtp.extension_length(), 0);
|
||||
EXPECT_EQ(rtp.extension_data().size(), 0);
|
||||
}
|
||||
94
tests/src/vxlan_test.cpp
Normal file
94
tests/src/vxlan_test.cpp
Normal file
@@ -0,0 +1,94 @@
|
||||
#include <gtest/gtest.h>
|
||||
#include <string>
|
||||
#include <tins/ethernetII.h>
|
||||
#include <tins/ip.h>
|
||||
#include <tins/tcp.h>
|
||||
#include <tins/udp.h>
|
||||
#include <tins/pdu.h>
|
||||
#include <tins/small_uint.h>
|
||||
#include <tins/vxlan.h>
|
||||
|
||||
#define PACKET_SIZE 68ul
|
||||
|
||||
using namespace std;
|
||||
using namespace Tins;
|
||||
|
||||
class VXLANTest : public testing::Test {
|
||||
public:
|
||||
static const uint8_t expected_packet[PACKET_SIZE];
|
||||
static const uint8_t flags;
|
||||
static const uint16_t dport, sport, p_type;
|
||||
static const small_uint<24> vni;
|
||||
static const IP::address_type dst_ip, src_ip;
|
||||
static const EthernetII::address_type dst_addr, src_addr;
|
||||
};
|
||||
|
||||
const uint8_t VXLANTest::expected_packet[PACKET_SIZE] = {
|
||||
0x08, 0x00, 0x00, 0x00,
|
||||
0xff, 0xff, 0xff, 0x00,
|
||||
0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
|
||||
0x8a, 0x8b, 0x8c, 0x8d, 0x8e, 0x8f,
|
||||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||
};
|
||||
|
||||
const uint8_t VXLANTest::flags = 8;
|
||||
const uint16_t VXLANTest::dport = 4789;
|
||||
const uint16_t VXLANTest::sport = 19627;
|
||||
const uint16_t VXLANTest::p_type = 0xd0ab;
|
||||
const small_uint<24> VXLANTest::vni = 0xffffff;
|
||||
const IP::address_type VXLANTest::dst_ip = IP::address_type{"2.2.2.2"};
|
||||
const IP::address_type VXLANTest::src_ip = IP::address_type{"1.1.1.1"};
|
||||
const EthernetII::address_type VXLANTest::dst_addr = EthernetII::address_type{"aa:bb:cc:dd:ee:ff"};
|
||||
const EthernetII::address_type VXLANTest::src_addr = EthernetII::address_type{"8a:8b:8c:8d:8e:8f"};
|
||||
|
||||
TEST_F(VXLANTest, Flags) {
|
||||
auto const vxlan = VXLAN{};
|
||||
EXPECT_EQ(vxlan.get_flags(), flags);
|
||||
}
|
||||
|
||||
TEST_F(VXLANTest, VNI) {
|
||||
auto const vxlan = VXLAN{vni};
|
||||
EXPECT_EQ(vxlan.get_vni(), vni);
|
||||
}
|
||||
|
||||
TEST_F(VXLANTest, Find) {
|
||||
auto const pdu = VXLAN{} / EthernetII{dst_addr, src_addr};
|
||||
auto const eth = pdu.find_pdu<EthernetII>();
|
||||
ASSERT_TRUE(eth != nullptr);
|
||||
EXPECT_EQ(eth->dst_addr(), dst_addr);
|
||||
EXPECT_EQ(eth->src_addr(), src_addr);
|
||||
}
|
||||
|
||||
TEST_F(VXLANTest, Serialize) {
|
||||
auto eth = EthernetII{dst_addr, src_addr};
|
||||
eth.payload_type(p_type);
|
||||
auto vxlan = VXLAN{vni};
|
||||
vxlan.inner_pdu(eth);
|
||||
auto serialized = vxlan.serialize();
|
||||
ASSERT_EQ(serialized.size(), PACKET_SIZE);
|
||||
EXPECT_TRUE(std::equal(serialized.begin(), serialized.end(), expected_packet));
|
||||
}
|
||||
|
||||
TEST_F(VXLANTest, ConstructorFromBuffer) {
|
||||
auto vxlan = VXLAN{expected_packet, PACKET_SIZE};
|
||||
EXPECT_EQ(vxlan.get_vni(), vni);
|
||||
EXPECT_EQ(vxlan.get_flags(), flags);
|
||||
auto const eth = vxlan.find_pdu<EthernetII>();
|
||||
ASSERT_TRUE(eth != nullptr);
|
||||
EXPECT_EQ(eth->dst_addr(), dst_addr);
|
||||
EXPECT_EQ(eth->src_addr(), src_addr);
|
||||
}
|
||||
|
||||
TEST_F(VXLANTest, OuterUDP) {
|
||||
auto pkt = IP{dst_ip, src_ip} / UDP{dport, sport} / VXLAN{expected_packet, PACKET_SIZE};
|
||||
auto const udp = pkt.find_pdu<UDP>();
|
||||
ASSERT_TRUE(udp != nullptr);
|
||||
EXPECT_EQ(udp->dport(), dport);
|
||||
EXPECT_EQ(udp->sport(), sport);
|
||||
auto const vxlan = udp->find_pdu<VXLAN>();
|
||||
ASSERT_TRUE(vxlan != nullptr);
|
||||
EXPECT_EQ(vxlan->get_flags(), flags);
|
||||
EXPECT_EQ(vxlan->get_vni(), vni);
|
||||
}
|
||||
Reference in New Issue
Block a user