andreabont/Project-Riddle
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Opt riddle

Browse source
This commit is contained in:
Andrea Bontempi 2012-10-12 11:43:22 +02:00
parent 65af38e7d4
commit 82e641e8de
1 changed files with 14 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
Riddle.cpp
View file

@ -30,7 +30,6 @@
#include <cstdio>
#include <cstdlib>
#include <string>
#include <limits>
#include <pcap.h>
#include <boost/program_options.hpp>
#include "./commons/libDump.h"
@ -43,6 +42,15 @@
using namespace std;
using namespace boost::program_options;
string ( *dumper ) ( string, uint64_t, uint32_t );
void process_packet ( u_char *useless, const struct pcap_pkthdr* header, const u_char* packet ) {
cout << dumper ( libDump::encodeHexText( packet, header->len ), header->ts.tv_sec, header->ts.tv_usec );
cout.flush();
}
int main ( int argc, char **argv ) {
options_description desc ( "Riddle - Network Sniffer" );
desc.add_options()
@ -215,36 +223,23 @@ int main ( int argc, char **argv ) {
}
}
int maxpacket = numeric_limits<int>::max();
int maxpacket = -1;
if ( vm.count ( "limit" ) ) {
maxpacket=vm["limit"].as<int>();
}
string ( *dumper ) ( string, uint64_t, uint32_t );
if ( vm.count ( "dump" ) ) {
dumper=libDump::classicDump;
} else {
dumper=libDump::riddleDump;
}
const u_char *packet;
pcap_pkthdr header;
for ( ; maxpacket > 0; ) {
packet = pcap_next ( pcap_handle, &header );
if ( packet == NULL ) {
cerr << ">> Flow terminated" << endl;
break;
}
cout << dumper ( libDump::encodeHexText( packet, header.len ), header.ts.tv_sec, header.ts.tv_usec );
cout.flush();
if ( maxpacket!=numeric_limits<int>::max() ) maxpacket--;
}
pcap_loop( pcap_handle , maxpacket , process_packet , NULL );
cerr << ">> I finished the job, goodbye!" << endl;
pcap_close ( pcap_handle );
return EXIT_SUCCESS;
}
}