diff options
author | Denis Volpato Martins <denisvm@gmail.com> | 2016-12-06 18:48:26 -0200 |
---|---|---|
committer | Denis Volpato Martins <denisvm@gmail.com> | 2016-12-06 18:48:26 -0200 |
commit | 483eb6340809a686b855f8ca1e35c18e2c314b6e (patch) | |
tree | b8a69ea0870b78c409f2e9207783798d802c6e3e /example | |
parent | b019dcd11a0c62bd9eb3f28d37ecc68e99e8c294 (diff) |
Removed some magic numbers
Diffstat (limited to 'example')
-rw-r--r-- | example/ndpiReader.c | 14 | ||||
-rw-r--r-- | example/ndpi_util.c | 24 |
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; } |