aboutsummaryrefslogtreecommitdiff
path: root/example
diff options
context:
space:
mode:
authorVitaly Lavrov <vel21ripn@gmail.com>2022-03-08 02:20:56 +0300
committerGitHub <noreply@github.com>2022-03-08 00:20:56 +0100
commita1451935b8653adc830ee4cb827def3622fb02d6 (patch)
tree7056ae6059f3a4126afec650420654cba0f44e66 /example
parentc345b3c7af89957ef4bc55e2ccf1b1a4bc724f3a (diff)
Errors fixed (#1482)
Fixed errors for bigendian platforms in ndpiReader. All address and port comparisons and hash calculations are done with endian in mind. The get_ndpi_flow_info() function searched for an existing flow for the forward and reverse direction of the packet. The ndpi_workflow_node_cmp() function looked for a flow regardless of the packet's direction. This is what led to an error in determining the direction of transmission of the packet. Fixed error in "synscan" test: the number of packets in the forward and reverse direction is incorrectly defined (verified via tcpdump). Fixed bug with icmp protocol checksum check for big endian platforms.
Diffstat (limited to 'example')
-rw-r--r--example/reader_util.c63
1 files changed, 23 insertions, 40 deletions
diff --git a/example/reader_util.c b/example/reader_util.c
index a04f009db..49ab00ea3 100644
--- a/example/reader_util.c
+++ b/example/reader_util.c
@@ -544,6 +544,13 @@ void ndpi_workflow_free(struct ndpi_workflow * workflow) {
ndpi_free(workflow);
}
+static inline int cmp_n32(uint32_t a,uint32_t b) {
+ return a == b ? 0 : ntohl(a) < ntohl(b) ? -1:1;
+}
+static inline int cmp_n16(uint16_t a,uint16_t b) {
+ return a == b ? 0 : ntohs(a) < ntohs(b) ? -1:1;
+}
+
/* ***************************************************** */
int ndpi_workflow_node_cmp(const void *a, const void *b) {
@@ -557,29 +564,13 @@ int ndpi_workflow_node_cmp(const void *a, const void *b) {
if(fa->vlan_id < fb->vlan_id ) return(-1); else { if(fa->vlan_id > fb->vlan_id ) return(1); }
if(fa->protocol < fb->protocol ) return(-1); else { if(fa->protocol > fb->protocol ) return(1); }
- if(
- (
- (fa->src_ip == fb->src_ip )
- && (fa->src_port == fb->src_port)
- && (fa->dst_ip == fb->dst_ip )
- && (fa->dst_port == fb->dst_port)
- )
- ||
- (
- (fa->src_ip == fb->dst_ip )
- && (fa->src_port == fb->dst_port)
- && (fa->dst_ip == fb->src_ip )
- && (fa->dst_port == fb->src_port)
- )
- )
- return(0);
-
- if(fa->src_ip < fb->src_ip ) return(-1); else { if(fa->src_ip > fb->src_ip ) return(1); }
- if(fa->src_port < fb->src_port) return(-1); else { if(fa->src_port > fb->src_port) return(1); }
- if(fa->dst_ip < fb->dst_ip ) return(-1); else { if(fa->dst_ip > fb->dst_ip ) return(1); }
- if(fa->dst_port < fb->dst_port) return(-1); else { if(fa->dst_port > fb->dst_port) return(1); }
+ int r;
+ r = cmp_n32(fa->src_ip, fb->src_ip); if(r) return r;
+ r = cmp_n16(fa->src_port, fb->src_port) ; if(r) return r;
+ r = cmp_n32(fa->dst_ip, fb->dst_ip); if(r) return r;
+ r = cmp_n16(fa->dst_port, fb->dst_port);
- return(0); /* notreached */
+ return(r);
}
/* ***************************************************** */
@@ -789,11 +780,17 @@ static struct ndpi_flow_info *get_ndpi_flow_info(struct ndpi_workflow * workflow
flow.protocol = iph->protocol, flow.vlan_id = vlan_id;
flow.src_ip = iph->saddr, flow.dst_ip = iph->daddr;
flow.src_port = htons(*sport), flow.dst_port = htons(*dport);
- flow.hashval = hashval = flow.protocol + flow.src_ip + flow.dst_ip + flow.src_port + flow.dst_port;
+ flow.hashval = hashval = flow.protocol + ntohl(flow.src_ip) + ntohl(flow.dst_ip)
+ + ntohs(flow.src_port) + ntohs(flow.dst_port);
#if 0
- printf("hashval=%u [%u][%u][%u:%u][%u:%u]\n", hashval, flow.protocol, flow.vlan_id,
- flow.src_ip, flow.src_port, ntohs(flow.dst_ip), ntohs(flow.dst_port));
+ {
+ char ip1[48],ip2[48];
+ inet_ntop(AF_INET, &flow.src_ip, ip1, sizeof(ip1));
+ inet_ntop(AF_INET, &flow.dst_ip, ip2, sizeof(ip2));
+ printf("hashval=%u [%u][%u][%s:%u][%s:%u]\n", hashval, flow.protocol, flow.vlan_id,
+ ip1, ntohs(flow.src_port), ip2, ntohs(flow.dst_port));
+ }
#endif
idx = hashval % workflow->prefs.num_roots;
@@ -905,24 +902,10 @@ static struct ndpi_flow_info *get_ndpi_flow_info(struct ndpi_workflow * workflow
struct ndpi_flow_info *rflow = *(struct ndpi_flow_info**)ret;
if(is_changed) {
- if(rflow->src_ip == iph->saddr
- && rflow->dst_ip == iph->daddr
- && rflow->src_port == htons(*sport)
- && rflow->dst_port == htons(*dport)
- )
- *src_to_dst_direction = 0, rflow->bidirectional = 1;
- else
- *src_to_dst_direction = 1;
+ *src_to_dst_direction = 0, rflow->bidirectional |= 1;
}
else {
- if(rflow->src_ip == iph->saddr
- && rflow->dst_ip == iph->daddr
- && rflow->src_port == htons(*sport)
- && rflow->dst_port == htons(*dport)
- )
*src_to_dst_direction = 1;
- else
- *src_to_dst_direction = 0, rflow->bidirectional = 1;
}
if(enable_flow_stats) {
if(src_to_dst_direction) {