github_mirrors/vectorscan
3
0
Fork 1
mirror of https://github.com/VectorCamp/vectorscan.git synced 2025-06-28 16:41:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix wrong casts in pcapscan.cc

Browse Source
This commit is contained in:
Konstantinos Margaritis 2024-05-18 14:56:30 +03:00
parent a8373df48b
commit 80812ee5b3
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/pcapscan.cc
View File

@ -106,7 +106,8 @@ struct FiveTuple {
dstAddr = iphdr->ip_dst.s_addr; dstAddr = iphdr->ip_dst.s_addr;
// UDP/TCP ports // UDP/TCP ports
const struct udphdr *uh = reinterpret_cast<const struct udphdr *>(iphdr) + (iphdr->ip_hl * 4); const char * iphdr_base = reinterpret_cast<const char *>(iphdr);
const struct udphdr *uh = reinterpret_cast<const struct udphdr *>(iphdr_base + (iphdr->ip_hl * 4));
srcPort = uh->uh_sport; srcPort = uh->uh_sport;
dstPort = uh->uh_dport; dstPort = uh->uh_dport;
} }
@ -231,7 +232,7 @@ public:
} }
// Valid TCP or UDP packet // Valid TCP or UDP packet
const struct ip *iphdr = reinterpret_cast<const struct ip *>(pktData) + sizeof(struct ether_header); const struct ip *iphdr = reinterpret_cast<const struct ip *>(pktData + sizeof(struct ether_header));
const char *payload = reinterpret_cast<const char *>(pktData) + offset; const char *payload = reinterpret_cast<const char *>(pktData) + offset;
size_t id = stream_map.insert(std::make_pair(FiveTuple(iphdr), size_t id = stream_map.insert(std::make_pair(FiveTuple(iphdr),
@ -572,7 +573,8 @@ int main(int argc, char **argv) {
*/ */
static bool payloadOffset(const unsigned char *pkt_data, unsigned int *offset, static bool payloadOffset(const unsigned char *pkt_data, unsigned int *offset,
unsigned int *length) { unsigned int *length) {
const ip *iph = reinterpret_cast<const ip *>(pkt_data) + sizeof(ether_header); const ip *iph = reinterpret_cast<const ip *>(pkt_data + sizeof(ether_header));
const char *iph_base = reinterpret_cast<const char *>(iph);
const tcphdr *th = nullptr; const tcphdr *th = nullptr;
// Ignore packets that aren't IPv4 // Ignore packets that aren't IPv4
@ -591,7 +593,7 @@ static bool payloadOffset(const unsigned char *pkt_data, unsigned int *offset,
switch (iph->ip_p) { switch (iph->ip_p) {
case IPPROTO_TCP: case IPPROTO_TCP:
th = reinterpret_cast<const tcphdr *>(iph) + ihlen; th = reinterpret_cast<const tcphdr *>(iph_base + ihlen);
thlen = th->th_off * 4; thlen = th->th_off * 4;
break; break;
case IPPROTO_UDP: case IPPROTO_UDP: