pcap: Add a packet couter for diagnostics

Signed-off-by: Siddharth Chandrasekaran <[email protected]>
goToMain · Feb 17, 2024 · 06259c4 · 06259c4
1 parent a0a6635
commit 06259c4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/include/utils/pcap_gen.h b/include/utils/pcap_gen.h
@@ -18,6 +18,7 @@ typedef struct {
 	FILE *file;
 	size_t offset;
 	void *cache;
+	size_t num_packets;
 } pcap_t;
 
 pcap_t *pcap_start(char *path, uint32_t max_packet_size, uint32_t link_type);

diff --git a/src/pcap_gen.c b/src/pcap_gen.c
@@ -55,6 +55,7 @@ pcap_t *pcap_start(char *path, uint32_t max_packet_size, uint32_t link_type)
 	header.link_type = link_type;
 
 	cap->offset = 0;
+	cap->num_packets = 0;
 	cap->file = fopen(path, "wb");
 	if(cap->file == NULL) {
 		free(cap->cache);
@@ -86,8 +87,9 @@ int pcap_add(pcap_t *cap, uint8_t *capture_data, uint32_t length)
 	uint32_t sec, usec;
 
 	if (cap->offset + sizeof(header) + length > PCAP_CACHE_SIZE) {
-		if (pcap_flush(cap))
+		if (pcap_flush(cap)) {
 			return -1;
+		}
 	}
 
 	get_time(&sec, &usec);
@@ -100,7 +102,7 @@ int pcap_add(pcap_t *cap, uint8_t *capture_data, uint32_t length)
 
 	memcpy((char *)cap->cache + cap->offset, capture_data, length);
 	cap->offset += length;
-
+	cap->num_packets++;
 	return 0;
 }