Context Navigation

parseargs.cfa@ 92bfda0

Visit:

ADT arm-eh ast-experimental enum forall-pointer-decay jacob/cs343-translation new-ast-unique-expr pthread-emulation qualifiedEnum

Last change on this file since 92bfda0 was 6f94958, checked in by Thierry Delisle <tdelisle@…>, 5 years ago
fix usage printing
Property mode set to `100644`
File size: 7.5 KB

Rev	Line
[7f389a5c]	1	#include "parseargs.hfa"
	2
[e699eb6]	3	#include <stdint.h>
[7f389a5c]	4	#include <string.h>
[e699eb6]	5	#include <errno.h>
[3f1d9b5]	6	#include <unistd.h>
[7f389a5c]	7	extern "C" {
	8	#include <getopt.h>
[3f1d9b5]	9	#include <sys/ioctl.h>
[7f389a5c]	10
	11	struct FILE;
	12	extern FILE * stderr;
	13	extern FILE * stdout;
	14
[3f1d9b5]	15	extern int fileno(FILE *stream);
	16
[7f389a5c]	17	extern int fprintf ( FILE * stream, const char * format, ... );
[53e4562]	18
	19	extern long long int strtoll (const char* str, char** endptr, int base);
	20	extern unsigned long long int strtoull(const char* str, char** endptr, int base);
[7f6e9eb]	21	extern double strtod (const char* str, char** endptr);
[7f389a5c]	22	}
	23
[e699eb6]	24	#include "common.hfa"
	25	#include "limits.hfa"
[3f1d9b5]	26
[433d352]	27	extern int cfa_args_argc __attribute__((weak));
	28	extern char ** cfa_args_argv __attribute__((weak));
	29	extern char ** cfa_args_envp __attribute__((weak));
[7874d77]	30
[419c434]	31	static void usage(char * cmd, cfa_option options[], size_t opt_count, const char * usage, FILE * out) __attribute__ ((noreturn));
[80d3b1b]	32	//-----------------------------------------------------------------------------
	33	// checking
	34	static void check_args(cfa_option options[], size_t opt_count) {
	35	for(i; opt_count) {
	36	for(j; opt_count) {
	37	if(i == j) continue;
	38
	39	if( options[i].short_name != '\0'
	40	&& options[i].short_name == options[j].short_name)
[b7664a03]	41	abort("Parse Args error: two options have short name '%c' (%zu & %zu)", options[i].short_name, i, j);
[3f1d9b5]	42
[b7664a03]	43	if(0 == strcmp(options[i].long_name, options[j].long_name)) abort("Parse Args error: two options have long name '%s' (%zu & %zu)", options[i].long_name, i, j);
[80d3b1b]	44	}
	45	}
	46	}
	47
	48
	49	//-----------------------------------------------------------------------------
	50	// Parsing args
[7874d77]	51	void parse_args( cfa_option options[], size_t opt_count, const char * usage, char ** & left ) {
[433d352]	52	if( 0p != &cfa_args_argc ) {
	53	parse_args(cfa_args_argc, cfa_args_argv, options, opt_count, usage, left );
	54	}
	55	else {
	56	char * temp = "";
	57	parse_args(0, &temp, options, opt_count, usage, left );
	58	}
[7874d77]	59	}
	60
[7f389a5c]	61	void parse_args(
	62	int argc,
	63	char * argv[],
	64	cfa_option options[],
	65	size_t opt_count,
	66	const char * usage,
	67	char ** & left
	68	) {
[80d3b1b]	69	check_args(options, opt_count);
	70
	71	int maxv = 'h';
	72	char optstring[opt_count * 3] = { '\0' };
	73	{
	74	int idx = 0;
	75	for(i; opt_count) {
	76	if (options[i].short_name) {
	77	maxv = max(options[i].short_name, maxv);
	78	optstring[idx] = options[i].short_name;
	79	idx++;
	80	if( ((intptr_t)options[i].parse) != ((intptr_t)parse_settrue)
	81	&& ((intptr_t)options[i].parse) != ((intptr_t)parse_setfalse) ) {
	82	optstring[idx] = ':';
	83	idx++;
	84	}
	85	}
	86	}
	87	optstring[idx+0] = 'h';
	88	optstring[idx+1] = '\0';
	89	}
	90
[7f389a5c]	91	struct option optarr[opt_count + 2];
	92	{
	93	int idx = 0;
	94	for(i; opt_count) {
	95	if(options[i].long_name) {
[80d3b1b]	96	options[i].val = (options[i].short_name != '\0') ? ((int)options[i].short_name) : ++maxv;
[7f389a5c]	97	optarr[idx].name = options[i].long_name;
	98	optarr[idx].flag = 0p;
[80d3b1b]	99	optarr[idx].val = options[i].val;
[7f389a5c]	100	if( ((intptr_t)options[i].parse) == ((intptr_t)parse_settrue)
	101	\|\| ((intptr_t)options[i].parse) == ((intptr_t)parse_setfalse) ) {
	102	optarr[idx].has_arg = no_argument;
	103	} else {
	104	optarr[idx].has_arg = required_argument;
	105	}
	106	idx++;
	107	}
	108	}
[e699eb6]	109	optarr[idx+0].[name, has_arg, flag, val] = ["help", no_argument, 0, 'h'];
	110	optarr[idx+1].[name, has_arg, flag, val] = [0, no_argument, 0, 0];
[7f389a5c]	111	}
	112
	113	FILE * out = stderr;
	114	NEXT_ARG:
	115	for() {
	116	int idx = 0;
	117	int opt = getopt_long(argc, argv, optstring, optarr, &idx);
	118	switch(opt) {
	119	case -1:
	120	if(&left != 0p) left = argv + optind;
	121	return;
	122	case 'h':
	123	out = stdout;
	124	case '?':
[419c434]	125	usage(argv[0], options, opt_count, usage, out);
[7f389a5c]	126	default:
	127	for(i; opt_count) {
[80d3b1b]	128	if(opt == options[i].val) {
[7f389a5c]	129	const char * arg = optarg ? optarg : "";
[772b300]	130	if( arg[0] == '=' ) { arg++; }
[7f389a5c]	131	bool success = options[i].parse( arg, options[i].variable );
	132	if(success) continue NEXT_ARG;
	133
	134	fprintf(out, "Argument '%s' for option %c could not be parsed\n\n", arg, (char)opt);
[419c434]	135	usage(argv[0], options, opt_count, usage, out);
[7f389a5c]	136	}
	137	}
	138	abort("Internal parse arg error\n");
	139	}
	140
	141	}
[419c434]	142	}
[7f389a5c]	143
[419c434]	144	//-----------------------------------------------------------------------------
	145	// Print usage
	146	static void printopt(FILE * out, int width, int max, char sn, const char * ln, const char * help) {
	147	int hwidth = max - (11 + width);
	148	if(hwidth <= 0) hwidth = max;
	149
[6f94958]	150	char sname[4] = { ' ', ' ', ' ', '\0' };
	151	if(sn != '\0') {
	152	sname[0] = '-';
	153	sname[1] = sn;
	154	sname[2] = ',';
	155	}
	156
	157	fprintf(out, " %s --%-s %.s\n", sname, width, ln, hwidth, help);
[419c434]	158	for() {
	159	help += min(strlen(help), hwidth);
	160	if('\0' == *help) break;
	161	fprintf(out, "%s%.s\n", width + 11, "", hwidth, help);
	162	}
	163	}
	164
	165	void print_args_usage(cfa_option options[], size_t opt_count, const char * usage, bool error) __attribute__ ((noreturn)) {
	166	usage(cfa_args_argv[0], options, opt_count, usage, error ? stderr : stdout);
	167	}
	168
	169	void print_args_usage(int , char * argv[], cfa_option options[], size_t opt_count, const char * usage, bool error) __attribute__ ((noreturn)) {
	170	usage(argv[0], options, opt_count, usage, error ? stderr : stdout);
	171	}
	172
	173	static void usage(char * cmd, cfa_option options[], size_t opt_count, const char * help, FILE * out) __attribute__((noreturn)) {
	174	int width = 0;
	175	{
	176	for(i; opt_count) {
	177	if(options[i].long_name) {
	178	int w = strlen(options[i].long_name);
	179	if(w > width) width = w;
	180	}
	181	}
	182	}
	183
	184	int max_width = 1_000_000;
[3f1d9b5]	185	int outfd = fileno(out);
	186	if(isatty(outfd)) {
	187	struct winsize size;
	188	int ret = ioctl(outfd, TIOCGWINSZ, &size);
	189	if(ret < 0) abort( "ioctl error: (%d) %s\n", (int)errno, strerror(errno) );
	190	max_width = size.ws_col;
	191	}
	192
[419c434]	193	fprintf(out, "Usage:\n %s %s\n", cmd, help);
[7f389a5c]	194
	195	for(i; opt_count) {
[3f1d9b5]	196	printopt(out, width, max_width, options[i].short_name, options[i].long_name, options[i].help);
[7f389a5c]	197	}
	198	fprintf(out, " -%c, --%-*s %s\n", 'h', width, "help", "print this help message");
	199	exit(out == stdout ? 0 : 1);
	200	}
	201
[419c434]	202	//-----------------------------------------------------------------------------
	203	// Typed argument parsing
[7f389a5c]	204	bool parse_yesno(const char * arg, bool & value ) {
	205	if(strcmp(arg, "yes") == 0) {
	206	value = true;
	207	return true;
	208	}
	209
	210	if(strcmp(arg, "no") == 0) {
	211	value = false;
	212	return true;
	213	}
	214
	215	return false;
	216	}
	217
[7efb322]	218	bool parse_truefalse(const char * arg, bool & value) {
[d411769c]	219	if(strcmp(arg, "true") == 0) {
	220	value = true;
	221	return true;
	222	}
	223
	224	if(strcmp(arg, "false") == 0) {
	225	value = false;
	226	return true;
	227	}
	228
	229	return false;
	230	}
	231
[7f389a5c]	232	bool parse_settrue (const char *, bool & value ) {
	233	value = true;
	234	return true;
	235	}
	236
	237	bool parse_setfalse(const char *, bool & value ) {
	238	value = false;
	239	return true;
	240	}
	241
	242	bool parse(const char * arg, const char * & value ) {
	243	value = arg;
	244	return true;
	245	}
	246
[7f6e9eb]	247	bool parse(const char * arg, int & value) {
	248	char * end;
	249	int r = strtoll(arg, &end, 10);
	250	if(*end != '\0') return false;
	251
	252	value = r;
	253	return true;
	254	}
	255
[53e4562]	256	bool parse(const char * arg, unsigned & value) {
	257	char * end;
	258	unsigned long long int r = strtoull(arg, &end, 10);
	259	if(*end != '\0') return false;
[3f1d9b5]	260	if(r > (unsigned)MAX) return false;
[53e4562]	261
	262	value = r;
	263	return true;
	264	}
	265
[56e8cb3]	266	bool parse(const char * arg, unsigned long & value) {
[53e4562]	267	char * end;
	268	unsigned long long int r = strtoull(arg, &end, 10);
	269	if(*end != '\0') return false;
[56e8cb3]	270	if(r > (unsigned long)MAX) return false;
[53e4562]	271
	272	value = r;
	273	return true;
	274	}
	275
[56e8cb3]	276	bool parse(const char * arg, unsigned long long & value) {
	277	char * end;
	278	unsigned long long int r = strtoull(arg, &end, 10);
	279	if(*end != '\0') return false;
	280	if(r > (unsigned long long)MAX) return false;
	281
	282	value = r;
	283	return true;
	284	}
	285
[7f6e9eb]	286	bool parse(const char * arg, double & value) {
[7f389a5c]	287	char * end;
[7f6e9eb]	288	double r = strtod(arg, &end);
[7f389a5c]	289	if(*end != '\0') return false;
	290
	291	value = r;
	292	return true;
[56e8cb3]	293	}

Note: See TracBrowser for help on using the repository browser.

Download in other formats:

Original Format