/**
* @brief Main function logic
*
* Parse command line options and start running threads
*/
int
main(int argc, char* argv[])
{
int opt, idx, limit, only_calls, no_incomplete, i;
const char *device, *outfile;
char bpf[512];
const char *keyfile;
const char *match_expr;
int match_insensitive = 0, match_invert = 0;
int no_interface = 0, quiet = 0, rtp_capture = 0;
vector_t *infiles = vector_create(0, 1);
// Program otptions
static struct option long_options[] = {
{ "help", no_argument, 0, 'h' },
{ "version", no_argument, 0, 'V' },
{ "device", required_argument, 0, 'd' },
{ "input", required_argument, 0, 'I' },
{ "output", required_argument, 0, 'O' },
#if defined(WITH_GNUTLS) || defined(WITH_OPENSSL)
{ "keyfile", required_argument, 0, 'k' },
#endif
{ "calls", no_argument, 0, 'c' },
{ "rtp", no_argument, 0, 'r' },
{ "limit", no_argument, 0, 'l' },
{ "icase", no_argument, 0, 'i' },
{ "invert", no_argument, 0, 'v' },
{ "no-interface", no_argument, 0, 'N' },
{ "dump-config", no_argument, 0, 'D' },
#ifdef USE_EEP
{ "eep-listen", required_argument, 0, 'L' },
{ "eep-send", required_argument, 0, 'H' },
#endif
{ "quiet", no_argument, 0, 'q' },
};
// Initialize configuration options
init_options();
// Get initial values for configurable arguments
device = setting_get_value(SETTING_CAPTURE_DEVICE);
outfile = setting_get_value(SETTING_CAPTURE_OUTFILE);
keyfile = setting_get_value(SETTING_CAPTURE_KEYFILE);
limit = setting_get_intvalue(SETTING_CAPTURE_LIMIT);
only_calls = setting_enabled(SETTING_SIP_CALLS);
no_incomplete = setting_enabled(SETTING_SIP_NOINCOMPLETE);
rtp_capture = setting_enabled(SETTING_CAPTURE_RTP);
// Parse command line arguments
opterr = 0;
char *options = "hVd:I:O:pqtW:k:crl:ivNqDL:H:";
while ((opt = getopt_long(argc, argv, options, long_options, &idx)) != -1) {
switch (opt) {
case 'h':
usage();
return 0;
case 'V':
version();
return 0;
case 'd':
device = optarg;
break;
case 'I':
vector_append(infiles, optarg);
break;
case 'O':
outfile = optarg;
break;
case 'l':
if(!(limit = atoi(optarg))) {
fprintf(stderr, "Invalid limit value.\n");
return 0;
}
break;
#if defined(WITH_GNUTLS) || defined(WITH_OPENSSL)
case 'k':
keyfile = optarg;
break;
#endif
case 'c':
only_calls = 1;
setting_set_value(SETTING_SIP_CALLS, SETTING_ON);
break;
case 'r':
rtp_capture = 1;
setting_set_value(SETTING_CAPTURE_RTP, SETTING_ON);
break;
case 'i':
match_insensitive++;
break;
case 'v':
match_invert++;
break;
case 'N':
no_interface = 1;
setting_set_value(SETTING_CAPTURE_STORAGE, "none");
break;
case 'q':
quiet = 1;
break;
case 'D':
key_bindings_dump();
settings_dump();
return 0;
// Dark options for dummy ones
case 'p':
case 't':
case 'W':
break;
#ifdef USE_EEP
case 'L':
capture_eep_set_server_url(optarg);
break;
case 'H':
capture_eep_set_client_url(optarg);
break;
#endif
case '?':
if (strchr(options, optopt)) {
fprintf(stderr, "-%c option requires an argument.\n", optopt);
} else if (isprint(optopt)) {
fprintf(stderr, "Unknown option -%c.\n", optopt);
} else {
fprintf(stderr, "Unknown option character '\\x%x'.\n", optopt);
}
return 1;
default:
break;
}
}
#if defined(WITH_GNUTLS) || defined(WITH_OPENSSL)
// Set capture decrypt key file
capture_set_keyfile(keyfile);
// Check if we have a keyfile and is valid
if (keyfile && !tls_check_keyfile(keyfile)) {
fprintf(stderr, "%s does not contain a valid RSA private key.\n", keyfile);
return 1;
}
#endif
// Check if given argument is a file
if (argc == 2 && (access(argv[1], F_OK) == 0)) {
// Old legacy option to open pcaps without other arguments
printf("%s seems to be a file: You forgot -I flag?\n", argv[1]);
return 0;
}
// Initialize SIP Messages Storage
sip_init(limit, only_calls, no_incomplete);
// Set capture options
capture_init(limit, rtp_capture);
#ifdef USE_EEP
// Initialize EEP if enabled
capture_eep_init();
#endif
// If we have an input file, load it
if (vector_count(infiles)) {
for (i = 0; i < vector_count(infiles); i++) {
// Try to load file
if (capture_offline(vector_item(infiles, i), outfile) != 0)
return 1;
}
} else {
// Check if all capture data is valid
if (capture_online(device, outfile) != 0)
return 1;
}
// Remove Input files vector
vector_destroy(infiles);
// More arguments pending!
if (argv[optind]) {
// Assume first pending argument is match expression
match_expr = argv[optind++];
// Try to build the bpf filter string with the rest
memset(bpf, 0, sizeof(bpf));
for (i = optind; i < argc; i++)
sprintf(bpf + strlen(bpf), "%s ", argv[i]);
// Check if this BPF filter is valid
if (capture_set_bpf_filter(bpf) != 0) {
// BPF Filter invalid, check incluiding match_expr
match_expr = 0; // There is no need to parse all payload at this point
// Build the bpf filter string
memset(bpf, 0, sizeof(bpf));
for (i = optind - 1; i < argc; i++)
sprintf(bpf + strlen(bpf), "%s ", argv[i]);
// Check bpf filter is valid again
if (capture_set_bpf_filter(bpf) != 0) {
fprintf(stderr, "Couldn't install filter %s: %s\n", bpf, capture_last_error());
return 1;
}
}
// Set the capture filter
if (match_expr)
if (sip_set_match_expression(match_expr, match_insensitive, match_invert)) {
fprintf(stderr, "Unable to parse expression %s\n", match_expr);
return 1;
}
}
// Start a capture thread
if (capture_launch_thread() != 0) {
ncurses_deinit();
fprintf(stderr, "Failed to launch capture thread.\n");
return 1;
}
if (!no_interface) {
// Initialize interface
ncurses_init();
// This is a blocking call.
// Create the first panel and wait for user input
ui_create_panel(PANEL_CALL_LIST);
wait_for_input();
} else {
setbuf(stdout, NULL);
while(capture_get_status() != CAPTURE_OFFLINE) {
if (!quiet)
printf("\rDialog count: %d", sip_calls_count());
usleep(500 * 1000);
}
if (!quiet)
printf("\rDialog count: %d\n", sip_calls_count());
}
// Capture deinit
capture_deinit();
// Deinitialize interface
ncurses_deinit();
// Deinitialize configuration options
deinit_options();
// Deallocate sip stored messages
sip_deinit();
// Leaving!
return 0;
}
int
save_to_file(ui_t *ui)
{
char savepath[256];
char savefile[256];
char fullfile[512];
sip_call_t *call = NULL;
sip_msg_t *msg = NULL;
pcap_dumper_t *pd = NULL;
FILE *f = NULL;
int cur = 0, total = 0;
WINDOW *progress;
vector_iter_t calls, msgs, rtps, packets;
packet_t *packet;
vector_t *sorted;
// Get panel information
save_info_t *info = save_info(ui);
// Get current path field value.
memset(savepath, 0, sizeof(savepath));
strcpy(savepath, field_buffer(info->fields[FLD_SAVE_PATH], 0));
strtrim(savepath);
if (strlen(savepath))
strcat(savepath, "/");
// Get current file field value.
memset(savefile, 0, sizeof(savefile));
strcpy(savefile, field_buffer(info->fields[FLD_SAVE_FILE], 0));
strtrim(savefile);
if (!strlen(savefile)) {
dialog_run("Please enter a valid filename");
return 1;
}
if (info->saveformat == SAVE_PCAP || info->saveformat == SAVE_PCAP_RTP) {
if (!strstr(savefile, ".pcap"))
strcat(savefile, ".pcap");
} else {
if (!strstr(savefile, ".txt"))
strcat(savefile, ".txt");
}
// Absolute filename
sprintf(fullfile, "%s%s", savepath, savefile);
if (access(fullfile, R_OK) == 0) {
if (dialog_confirm("Overwrite confirmation", "Selected file already exits.\n Do you want to overwrite it?", "Yes,No") != 0)
return 1;
}
// Don't allow to save no packets!
if (info->savemode == SAVE_SELECTED && call_group_msg_count(info->group) == 0) {
dialog_run("Unable to save: No selected dialogs.");
return 1;
}
if (info->saveformat == SAVE_PCAP || info->saveformat == SAVE_PCAP_RTP) {
// Open dump file
pd = dump_open(fullfile);
if (access(fullfile, W_OK) != 0) {
dialog_run(capture_last_error());
return 1;
}
} else {
// Open a text file
if (!(f = fopen(fullfile, "w"))) {
dialog_run("Error: %s", strerror(errno));
return 0;
}
}
// Get calls iterator
switch (info->savemode) {
case SAVE_ALL:
// Get calls iterator
calls = sip_calls_iterator();
break;
case SAVE_SELECTED:
// Save selected packets to file
calls = vector_iterator(info->group->calls);
break;
case SAVE_DISPLAYED:
// Set filtering for this iterator
calls = sip_calls_iterator();
vector_iterator_set_filter(&calls, filter_check_call);
break;
default:
break;
}
if (info->savemode == SAVE_MESSAGE) {
if (info->saveformat == SAVE_TXT) {
// Save selected message to file
save_msg_txt(f, info->msg);
} else {
// Save selected message packet to pcap
dump_packet(pd, info->msg->packet);
}
} else if (info->saveformat == SAVE_TXT) {
// Save selected packets to file
while ((call = vector_iterator_next(&calls))) {
msgs = vector_iterator(call->msgs);
// Save SIP message content
while ((msg = vector_iterator_next(&msgs))) {
save_msg_txt(f, msg);
}
}
} else {
// Store all messages in a time sorted vector
sorted = vector_create(100, 50);
vector_set_sorter(sorted, capture_packet_time_sorter);
// Count packages for progress bar
while ((call = vector_iterator_next(&calls))) {
total += vector_count(call->msgs);
if (info->saveformat == SAVE_PCAP_RTP)
total += vector_count(call->rtp_packets);
}
vector_iterator_reset(&calls);
progress = dialog_progress_run("Saving packets...");
dialog_progress_set_value(progress, 0);
// Save selected packets to file
while ((call = vector_iterator_next(&calls))) {
msgs = vector_iterator(call->msgs);
// Save SIP message content
while ((msg = vector_iterator_next(&msgs))) {
// Update progress bar dialog
dialog_progress_set_value(progress, (++cur * 100) / total);
vector_append(sorted, msg->packet);
}
// Save RTP packets
if (info->saveformat == SAVE_PCAP_RTP) {
rtps = vector_iterator(call->rtp_packets);
while ((packet = vector_iterator_next(&rtps))) {
// Update progress bar dialog
dialog_progress_set_value(progress, (++cur * 100) / total);
vector_append(sorted, packet);
}
}
}
// Save sorted packets
packets = vector_iterator(sorted);
while ((packet = vector_iterator_next(&packets))) {
dump_packet(pd, packet);
}
dialog_progress_destroy(progress);
}
// Close saved file
if (info->saveformat == SAVE_PCAP || info->saveformat == SAVE_PCAP_RTP) {
dump_close(pd);
} else {
fclose(f);
}
// Show success popup
if (info->savemode == SAVE_MESSAGE) {
dialog_run("Successfully saved selected SIP message to %s", savefile);
} else {
dialog_run("Successfully saved %d dialogs to %s", vector_iterator_count(&calls), savefile);
}
return 0;
}
