int main(int argc, char* argv[]) {
char c;
int cluster_id = -1, queue_id = -1;
pthread_t my_thread;
int wait_for_packet = 1, verbose = 0, dump_as_sysdig_event = 0;
startTime.tv_sec = 0;
while((c = getopt(argc,argv,"ac:g:hi:svu")) != '?') {
if((c == 255) || (c == -1)) break;
switch(c) {
case 'h':
printHelp();
break;
case 'a':
wait_for_packet = 0;
break;
case 'c':
cluster_id = atoi(optarg);
break;
case 'i':
queue_id = atoi(optarg);
break;
case 'g':
bind_core = atoi(optarg);
break;
case 's':
dump_as_sysdig_event = 1;
break;
case 'v':
verbose = 1;
break;
case 'u':
vm_guest = 1;
break;
}
}
if (cluster_id < 0) printHelp();
if (queue_id < 0) printHelp();
bind2node(bind_core);
/* volatile global variables initialization */
globals = calloc(1, sizeof(*globals));
globals->wait_for_packet = wait_for_packet;
globals->verbose = verbose;
globals->dump_as_sysdig_event = dump_as_sysdig_event;
globals->numPkts = 0;
globals->numBytes = 0;
globals->do_shutdown = 0;
if (vm_guest)
pfring_zc_vm_guest_init(NULL /* auto */);
zq = pfring_zc_ipc_attach_queue(cluster_id, queue_id, rx_only);
if(zq == NULL) {
fprintf(stderr, "pfring_zc_ipc_attach_queue error [%s] Please check that cluster %d is running\n",
strerror(errno), cluster_id);
return -1;
}
zp = pfring_zc_ipc_attach_buffer_pool(cluster_id, queue_id);
if(zp == NULL) {
fprintf(stderr, "pfring_zc_ipc_attach_buffer_pool error [%s] Please check that cluster %d is running\n",
strerror(errno), cluster_id);
return -1;
}
buffer = pfring_zc_get_packet_handle_from_pool(zp);
if (buffer == NULL) {
fprintf(stderr, "pfring_zc_get_packet_handle_from_pool error\n");
return -1;
}
signal(SIGINT, sigproc);
signal(SIGTERM, sigproc);
signal(SIGINT, sigproc);
pthread_create(&my_thread, NULL, packet_consumer_thread, (void*) NULL);
if (!verbose) while (!globals->do_shutdown) {
sleep(ALARM_SLEEP);
print_stats();
}
pthread_join(my_thread, NULL);
sleep(1);
pfring_zc_release_packet_handle_to_pool(zp, buffer);
pfring_zc_ipc_detach_queue(zq);
pfring_zc_ipc_detach_buffer_pool(zp);
return 0;
}
int main(int argc, char* argv[]) {
char c;
int cluster_id = -1, in_queue_id = -1, out_queue_id = -1;
pthread_t my_thread;
startTime.tv_sec = 0;
while((c = getopt(argc,argv,"ac:g:hi:o:fvu")) != '?') {
if((c == 255) || (c == -1)) break;
switch(c) {
case 'h':
printHelp();
break;
case 'a':
wait_for_packet = 0;
break;
case 'f':
flush_packet = 1;
break;
case 'v':
verbose = 1;
break;
case 'c':
cluster_id = atoi(optarg);
break;
case 'i':
in_queue_id = atoi(optarg);
break;
case 'o':
out_queue_id = atoi(optarg);
break;
case 'g':
bind_core = atoi(optarg);
break;
case 'u':
vm_guest = 1;
break;
}
}
if (in_queue_id < 0) printHelp();
if (out_queue_id < 0) printHelp();
if (cluster_id < 0) printHelp();
if (vm_guest)
pfring_zc_vm_guest_init(NULL /* auto */);
inzq = pfring_zc_ipc_attach_queue(cluster_id, in_queue_id, rx_only);
if(inzq == NULL) {
fprintf(stderr, "pfring_zc_ipc_attach_queue error [%s] Please check that cluster %d is running\n",
strerror(errno), cluster_id);
return -1;
}
outzq = pfring_zc_ipc_attach_queue(cluster_id, out_queue_id, tx_only);
if(outzq == NULL) {
fprintf(stderr, "pfring_zc_ipc_attach_queue error [%s] Please check that cluster %d is running\n",
strerror(errno), cluster_id);
return -1;
}
zp = pfring_zc_ipc_attach_buffer_pool(cluster_id, in_queue_id);
if(zp == NULL) {
fprintf(stderr, "pfring_zc_ipc_attach_buffer_pool error [%s] Please check that cluster %d is running\n",
strerror(errno), cluster_id);
return -1;
}
tmpbuff = pfring_zc_get_packet_handle_from_pool(zp);
if (tmpbuff == NULL) {
fprintf(stderr, "pfring_zc_get_packet_handle_from_pool error\n");
return -1;
}
signal(SIGINT, sigproc);
signal(SIGTERM, sigproc);
signal(SIGINT, sigproc);
if (!verbose) { /* periodic stats */
signal(SIGALRM, my_sigalarm);
alarm(ALARM_SLEEP);
}
pthread_create(&my_thread, NULL, packet_consumer_thread, (void*) NULL);
pthread_join(my_thread, NULL);
sleep(1);
pfring_zc_release_packet_handle_to_pool(zp, tmpbuff);
pfring_zc_ipc_detach_queue(inzq);
pfring_zc_ipc_detach_queue(outzq);
pfring_zc_ipc_detach_buffer_pool(zp);
return 0;
}
