args_parse.c

#include "args_parse.h"

void how_to_use(char *name)
{

	fprintf(stderr, "\n\t\t\t\tHTTP Packet Dissector\n\n");
	fprintf(stderr, "%s [options] -i=input_file\n\n", name);
	// fprintf(stderr, "\t-b  --binary\t\t\tThe output is binary.\n");
	fprintf(stderr, "\t-c  --capture=<interface>\tCapture from the given interface.\n");
	fprintf(stderr, "\t    --hpcap=<core>\t\tUses HPCAP.\n");
	fprintf(stderr, "\t    --hpcap_ifindex=<ifindex>\tHPCAP ifindex parameter. Default 1.\n");
	fprintf(stderr, "\t    --hpcap_qindex=<qindex>\tHPCAP qindex parameter. Default 0.\n");
	fprintf(stderr, "\t    --filter_or=<filter>\tJoins the default filter with the introduced one with an 'OR'.\n");
	fprintf(stderr, "\t    --filter_and=<filter>\tJoins the default filter with the introduced one with an 'AND'.\n");
	fprintf(stderr, "\t    --filter_ovr=<filter>\tOverwrites the default filter with the introduced one.\n");
	fprintf(stderr, "\t-d  --discards=<file>\t\tPacket discards file. Indicates the path to a file with the packtet numbers to be discarded. File must be sorted.\n");
	fprintf(stderr, "\t-D  --debug=<debug_level>\tActivates debug lines.\n");
	fprintf(stderr, "\t-h  --help\t\t\tShows this message.\n");
	fprintf(stderr, "\t-H  --host=<host>\t\tFilter the request by URL host\n");
	fprintf(stderr, "\t-i  --input=<file>\t\tInput file. This parameter is mandatory.\n");
	fprintf(stderr, "\t-I  --input-files\t\tIndicates that the input file is a list of files, the flag -i is still necesary.\n");
	fprintf(stderr, "\t    --log\t\t\tWrites more debug stuff in the log (httpDissector)\n");
	fprintf(stderr, "\t    --noRtx\t\t\tRemoves Retransmissions. The size of the window is 1 million transactions. Enables --sorted option.\n");
	fprintf(stderr, "\t-o  --output=<file>\t\tOutput file instead of stdout.\n");
	fprintf(stderr, "\t    --gc-output=<file>\t\tOutput file for the garbage collector. Writes the requests without responses in the given file.\n");
	fprintf(stderr, "\t-p  --pcap\t\t\tSets the input file format as pcap. (Set by Default)\n");
	fprintf(stderr, "\t-r  --raw\t\t\tSets the input file format as raw.\n");
	fprintf(stderr, "\t-R  --rrd\t\t\tOnly Prints second and the diff average from that second\n");
	fprintf(stderr, "\t    --sorted\t\t\tSorted output by request timestamp\n");
	fprintf(stderr, "\t    --agent\t\t\tPrint user-agent information\n");
	fprintf(stderr, "\t    --fqdn\t\t\tHost header must be a FQDN, otherwise the IP will be printed as host.\n");
	fprintf(stderr, "\t-u  --url=<url>\t\t\tFilter the request by url\n");
	fprintf(stderr, "\t-v  --verbose\t\t\tVerbose mode. Shows information about the Garbage Collector\n");
	fprintf(stderr, "\t    --version\t\t\tShows the program version\n");
	// fprintf(stderr, "\t-x  --index=<file>\t\tCreate index file. Writes an index in the given file. An entry every 5 minutes of traffic.\n\n");
	fprintf(stderr, "\t-x  --index=<file>\t\tUses an index file to perform a sampling processing of the file jumping to the interesting parts of the file.\n\n");


	fprintf(stderr, "\t\t\t\tOUTPUT FORMAT DETAILS\n");
	fprintf(stderr, "Default output\n");
	fprintf(stderr, "\tSourceIP|SourcePort|DestIP|DestPort|ReqTS|ResTS|Diff|ResponseMSG|ResponseCode|RequestType|Host|URL\n");
	fprintf(stderr, "\t*SourceIP is the requester IP\n");
	fprintf(stderr, "\t*The RequestType: HEAD, POST, GET, PUT, etc.\n");
	fprintf(stderr, "\t*The vertical bar ( | ) has been chosen as the separator character instead of the blank space due to the blank spaces inside the HTTP response message.\n\t It's faster than replace the blank spaces of the messages by low dashes in each message.\n\t It would take a lot of CPU time.\n\n");

	fprintf(stderr, "RRD output\n");
	fprintf(stderr, "\tSEC AVG_DIFF\n");
	// fprintf(stderr, "\t*If both options (--rrd and --two-lines) are enabled the RRD output has the priority\n\n");

	fprintf(stderr, "\t\t\t\tFILE OF FILES FORMAT\n");
	fprintf(stderr, "One file path per line.\n");
	fprintf(stderr, "Instead of \"~/file.pcap\" use the absolute path \"/Users/user/file.pcap\"\n");
	fprintf(stderr, "There is no problem using \"../file.pcap\" paths.\n\n");

	fprintf(stderr, "\n");

	return;
}

struct args_parse parse_args(int argc, char **argv) {

	struct args_parse options;
	options.input       = NULL;
	options.binary      = 0;
	options.output      = NULL;
	options.gcoutput    = NULL;
	options.filter      = NULL;
	options.filter_mode = OR;
	options.url         = NULL;
	options.host        = NULL;
	options.hpcap       = -1;
	options.hpcap_qindex = 0;
	options.hpcap_ifindex = 1;
	options.raw         = -1;
	options.rrd         = 0;
	options.debug       = 0;
	options.log         = 0;
	options.twolines    = 0;
	options.files       = 0;
	options.err         = -1;
	options.interface   = NULL;
	options.collector   = 1;
	options.verbose     = 0;
	options.version     = 0;
	options.sorted      = 0;
	options.agent       = 0;
	options.fqdn        = 0;
	options.noRtx       = 0;
	options.index       = NULL;
	options.discards    = NULL;
	options.skip        = 1;

	strcpy(options.errbuf, "Invalid arguments");

	int next_op;

	/* Una cadena que lista las opciones cortas válidas */
	const char *const short_op = "D:hrIvpbi:d:o:u:H:c:x:" ;

	/* Una estructura de varios arrays describiendo los valores largos */
	const struct option long_op[] = {
		{ "agent",          0,  NULL,   'A'},
		{ "capture",        1,  NULL,   'c'},
		{ "no-collector",   0,  NULL,   'C'},
		{ "discards",       1,  NULL,   'd'},
		{ "debug",          1,  NULL,   'D'},
		{ "filter_or",      1,  NULL,   'f'},
		{ "filter_and",     1,  NULL,   'F'},
		{ "help",           0,  NULL,   'h'},
		{ "host",           1,  NULL,   'H'},
		{ "input",          1,  NULL,   'i'},
		{ "input-files",    0,  NULL,   'I'},
		{ "skip",           1,  NULL,   'K'},
		{ "log",            0,  NULL,   'L'},
		{ "output",         1,  NULL,   'o'},
		{ "gc-output",      1,  NULL,   'O'},
		{ "raw",            0,  NULL,   'r'},
		{ "rrd",            0,  NULL,   'R'},
		{ "sorted",         0,  NULL,   'S'},
		{ "two-lines",      0,  NULL,   'T'},
		{ "pcap",			      0, 	NULL, 	'p'},
		{ "fqdn",           0,  NULL,   'q'},
		{ "url",            1,  NULL,   'u'},
		{ "verbose",        0,  NULL,   'v'},
		{ "version",        0,  NULL,   'V'},
		{ "filter_ovr",     1,  NULL,   'W'},
		{ "hpcap",          1,  NULL,   'Y'},
		{ "index",          1,  NULL,   'x'},
		{ "noRtx",          0,  NULL,   'X'},
		{ "hpcap_ifindex",  1,  NULL,   'z'},
		{ "hpcap_qindex",   1,  NULL,   'Z'},
		{ NULL,             0,  NULL,    0 }
		// { "binary",         0,  NULL,   'b'},
	};

	while (1) {
		/* Llamamos a la función getopt */
		next_op = getopt_long(argc, argv, short_op, long_op, NULL);

		if (next_op == -1) {
			break;    /* No hay más opciones. Rompemos el bucle */
		}

		switch (next_op) {
		case 'h' : /* -h o --help */
			options.err = -3;
			strcpy(options.errbuf, "Help:");
			return options;

		case 'D' :
			options.debug = atoi(optarg);
			break;

		case 'K' :
			options.skip = atoi(optarg);
			break;

		case 'i' : /* -i o --input */
			options.input = optarg;
			options.err = 0;
			break;

		case 'x' :
			options.index = optarg;
			break;

		case 'd' :
			options.discards = optarg;
			break;

		case 'b' :
			options.binary = 1;
			options.sorted = 1;
			break;

		case 'R' : /* -rrd */
			options.rrd = 1;
			break;

		case 'O' :
			options.gcoutput = optarg;
			break;

		case 'S' : /* --sorted */
			options.sorted = 1;
			break;

		case 'A' : /* --agent */
			options.agent = 1;
			break;

		case 'c' : /*-c o --capture */
			if (options.input) {
				options.err = -2;
				strcpy(options.errbuf, "Input from file & interface? Just choose one");
				return options;
			}

			options.interface = optarg;
			options.err = 0;
			break;

		case 'o' : /* -o ó --output */
			options.output = optarg; /* optarg contiene el argumento de -o */
			break;

		case 'r' : /*raw*/
			if (options.raw == 0) {
				options.err = -2;
				strcpy(options.errbuf, "Raw & pcap? Just choose one");
				return options;
			}

			options.raw = 1;
			break;

		case 'p' : /*pcap*/
			if (options.raw == 1) {
				options.err = -2;
				strcpy(options.errbuf, "Raw & pcap? Just choose one");
				return options;
			}

			options.raw = 0;
			break;

		case 'q' : /* --agent */
			options.fqdn = 1;
			break;

		case 'f' : /*filter OR*/
			options.filter = optarg;
			options.filter_mode = OR;
			break;

		case 'F' : /*filter AND*/
			options.filter = optarg;
			options.filter_mode = AND;
			break;

		case 'W' : /*filter overwrite*/
			options.filter = optarg;
			options.filter_mode = OVERWRITE;
			break;

		case 'Y' : /*HPCAP*/
			options.hpcap = atoi(optarg);
			options.err = 0;
			break;

		case 'z' : /*HPCAP*/
			options.hpcap_ifindex = atoi(optarg);
			break;

		case 'Z' : /*HPCAP*/
			options.hpcap_qindex = atoi(optarg);
			break;


		case 'v' :
			options.verbose = 1;
			break;

		case 'I' :
			options.files = 1;
			break;

		case 'T' :
			options.twolines = 1;
			break;

		case 'u' :
			options.url = optarg;

			if (options.host != NULL) {
				fprintf(stderr, "Choose just one filter, URL or HOST\n");
				exit(0);
				return options;
			}

			break;

		case 'H' :
			options.host = optarg;

			if (options.url != NULL) {
				fprintf(stderr, "Choose just one filter, URL or HOST\n");
				exit(0);
				return options;
			}

			break;

		case 'C' :
			options.collector = 0;
			break;

		case 'L' :
			options.log = 1;
			break;

		case 'X' :
			options.noRtx  = 1;
			options.sorted = 1;
			break;

		case 'V' :
			options.err = 0;
			options.version = 1;
			break;

		case '?' : /* opción no valida */
			options.err = -1;
			strcpy(options.errbuf, "Invalid arguments");
			return options;

		case -1 : /* No hay más opciones */
			break;

		default : /* Algo más? No esperado. Abortamos */
			options.err = -1;
			strcpy(options.errbuf, "Invalid arguments");
			return options;
		}
	}


	if (options.raw == -1) {
		options.raw = 0;
	}

	return options;

}