aboutsummaryrefslogtreecommitdiff
path: root/example
diff options
context:
space:
mode:
authorDenis Volpato Martins <denisvm@gmail.com>2016-12-06 18:48:26 -0200
committerDenis Volpato Martins <denisvm@gmail.com>2016-12-06 18:48:26 -0200
commit483eb6340809a686b855f8ca1e35c18e2c314b6e (patch)
treeb8a69ea0870b78c409f2e9207783798d802c6e3e /example
parentb019dcd11a0c62bd9eb3f28d37ecc68e99e8c294 (diff)
Removed some magic numbers
Diffstat (limited to 'example')
-rw-r--r--example/ndpiReader.c14
-rw-r--r--example/ndpi_util.c24
2 files changed, 19 insertions, 19 deletions
diff --git a/example/ndpiReader.c b/example/ndpiReader.c
index 76c18537d..e1777d7da 100644
--- a/example/ndpiReader.c
+++ b/example/ndpiReader.c
@@ -675,10 +675,10 @@ char* formatTraffic(float numBits, int bits, char *buf) {
if(numBits < 1024) {
snprintf(buf, 32, "%lu %c", (unsigned long)numBits, unit);
- } else if(numBits < 1048576) {
+ } else if(numBits < (1024*1024)) {
snprintf(buf, 32, "%.2f K%c", (float)(numBits)/1024, unit);
} else {
- float tmpMBits = ((float)numBits)/1048576;
+ float tmpMBits = ((float)numBits)/(1024*1024);
if(tmpMBits < 1024) {
snprintf(buf, 32, "%.2f M%c", tmpMBits, unit);
@@ -704,10 +704,10 @@ char* formatPackets(float numPkts, char *buf) {
if(numPkts < 1000) {
snprintf(buf, 32, "%.2f", numPkts);
- } else if(numPkts < 1000000) {
+ } else if(numPkts < (1000*1000)) {
snprintf(buf, 32, "%.2f K", numPkts/1000);
} else {
- numPkts /= 1000000;
+ numPkts /= (1000*1000);
snprintf(buf, 32, "%.2f M", numPkts);
}
@@ -735,10 +735,10 @@ char* formatBytes(u_int32_t howMuch, char *buf, u_int buf_len) {
if(howMuch < 1024) {
snprintf(buf, buf_len, "%lu %c", (unsigned long)howMuch, unit);
- } else if(howMuch < 1048576) {
+ } else if(howMuch < (1024*1024)) {
snprintf(buf, buf_len, "%.2f K%c", (float)(howMuch)/1024, unit);
} else {
- float tmpGB = ((float)howMuch)/1048576;
+ float tmpGB = ((float)howMuch)/(1024*1024);
if(tmpGB < 1024) {
snprintf(buf, buf_len, "%.2f M%c", tmpGB, unit);
@@ -801,7 +801,7 @@ static void printResults(u_int64_t tot_usec) {
cumulative_stats.pppoe_count += ndpi_thread_info[thread_id].workflow->stats.pppoe_count;
cumulative_stats.vlan_count += ndpi_thread_info[thread_id].workflow->stats.vlan_count;
cumulative_stats.fragmented_count += ndpi_thread_info[thread_id].workflow->stats.fragmented_count;
- for(i = 0; i < 6; i++)
+ for(i = 0; i < sizeof(cumulative_stats.packet_len)/sizeof(cumulative_stats.packet_len[0]); i++)
cumulative_stats.packet_len[i] += ndpi_thread_info[thread_id].workflow->stats.packet_len[i];
cumulative_stats.max_packet_len += ndpi_thread_info[thread_id].workflow->stats.max_packet_len;
}
diff --git a/example/ndpi_util.c b/example/ndpi_util.c
index 4f7d68b94..d121b5841 100644
--- a/example/ndpi_util.c
+++ b/example/ndpi_util.c
@@ -212,7 +212,7 @@ static struct ndpi_flow_info *get_ndpi_flow_info(struct ndpi_workflow * workflow
Note: to keep things simple (ndpiReader is just a demo app)
we handle IPv6 a-la-IPv4.
*/
- if(version == 4) {
+ if(version == IPVERSION) {
if(ipsize < 20)
return NULL;
@@ -254,7 +254,7 @@ static struct ndpi_flow_info *get_ndpi_flow_info(struct ndpi_workflow * workflow
*proto = iph->protocol;
l4 = ((u_int8_t *) l3 + l4_offset);
- if(iph->protocol == 6 && l4_packet_len >= 20) {
+ if(iph->protocol == IPPROTO_TCP && l4_packet_len >= 20) {
u_int tcp_len;
workflow->stats.tcp_count++;
@@ -284,7 +284,7 @@ static struct ndpi_flow_info *get_ndpi_flow_info(struct ndpi_workflow * workflow
tcp_len = ndpi_min(4*(*tcph)->doff, l4_packet_len);
*payload = &l4[tcp_len];
*payload_len = ndpi_max(0, l4_packet_len-4*(*tcph)->doff);
- } else if(iph->protocol == 17 && l4_packet_len >= 8) {
+ } else if(iph->protocol == IPPROTO_UDP && l4_packet_len >= 8) {
// udp
workflow->stats.udp_count++;
@@ -347,7 +347,7 @@ static struct ndpi_flow_info *get_ndpi_flow_info(struct ndpi_workflow * workflow
newflow->lower_port = lower_port, newflow->upper_port = upper_port;
newflow->ip_version = version;
- if(version == 4) {
+ if(version == IPVERSION) {
inet_ntop(AF_INET, &lower_ip, newflow->lower_name, sizeof(newflow->lower_name));
inet_ntop(AF_INET, &upper_ip, newflow->upper_name, sizeof(newflow->upper_name));
} else {
@@ -416,12 +416,12 @@ static struct ndpi_flow_info *get_ndpi_flow_info6(struct ndpi_workflow * workflo
struct ndpi_iphdr iph;
memset(&iph, 0, sizeof(iph));
- iph.version = 4;
+ iph.version = IPVERSION;
iph.saddr = iph6->ip6_src.u6_addr.u6_addr32[2] + iph6->ip6_src.u6_addr.u6_addr32[3];
iph.daddr = iph6->ip6_dst.u6_addr.u6_addr32[2] + iph6->ip6_dst.u6_addr.u6_addr32[3];
iph.protocol = iph6->ip6_ctlun.ip6_un1.ip6_un1_nxt;
- if(iph.protocol == 0x3C /* IPv6 destination option */) {
+ if(iph.protocol == IPPROTO_DSTOPTS /* IPv6 destination option */) {
u_int8_t *options = (u_int8_t*)iph6 + sizeof(const struct ndpi_ipv6hdr);
iph.protocol = options[0];
@@ -461,7 +461,7 @@ static unsigned int packet_processing(struct ndpi_workflow * workflow,
u_int8_t src_to_dst_direction= 1;
if(iph)
- flow = get_ndpi_flow_info(workflow, 4, vlan_id, iph, NULL,
+ flow = get_ndpi_flow_info(workflow, IPVERSION, vlan_id, iph, NULL,
ip_offset, ipsize,
ntohs(iph->tot_len) - (iph->ihl * 4),
&tcph, &udph, &sport, &dport,
@@ -510,7 +510,7 @@ static unsigned int packet_processing(struct ndpi_workflow * workflow,
if(flow->detected_protocol.protocol == NDPI_PROTOCOL_BITTORRENT) {
int i, j, n = 0;
- for(i=0, j = 0; i<20; i++) {
+ for(i=0, j = 0; j < sizeof(flow->bittorent_hash)-1; i++) {
sprintf(&flow->bittorent_hash[j], "%02x", flow->ndpi_flow->bittorent_hash[i]);
j += 2, n += flow->ndpi_flow->bittorent_hash[i];
}
@@ -748,11 +748,11 @@ void ndpi_workflow_process_packet (struct ndpi_workflow * workflow,
}
}
- if(iph->version == 4) {
+ if(iph->version == IPVERSION) {
ip_len = ((u_int16_t)iph->ihl * 4);
iph6 = NULL;
- if(iph->protocol == 41) {
+ if(iph->protocol == IPPROTO_IPV6) {
ip_offset += ip_len;
goto iph_check;
}
@@ -775,7 +775,7 @@ void ndpi_workflow_process_packet (struct ndpi_workflow * workflow,
proto = iph6->ip6_ctlun.ip6_un1.ip6_un1_nxt;
ip_len = sizeof(struct ndpi_ipv6hdr);
- if(proto == 0x3C /* IPv6 destination option */) {
+ if(proto == IPPROTO_DSTOPTS /* IPv6 destination option */) {
u_int8_t *options = (u_int8_t*)&packet[ip_offset+ip_len];
proto = options[0];
@@ -816,7 +816,7 @@ void ndpi_workflow_process_packet (struct ndpi_workflow * workflow,
iph = (struct ndpi_iphdr *) &packet[ip_offset];
- if(iph->version != 4) {
+ if(iph->version != IPVERSION) {
// printf("WARNING: not good (packet_id=%u)!\n", (unsigned int)workflow->stats.raw_packet_count);
goto v4_warning;
}