xref: /third_party/popt/src/popt.c

51 static void prtcon(const char *msg, poptContext con)
55 	con, con->os,
56 	(con->os->nextCharArg ? con->os->nextCharArg : ""),
57 	(con->os->nextArg ? con->os->nextArg : ""),
58 	con->os->next,
59 	(con->os->argv && con->os->argv[con->os->next]
60 		? con->os->argv[con->os->next] : ""));
64 void poptSetExecPath(poptContext con, const char * path, int allowAbsolute)
66     con->execPath = _free(con->execPath);
67     con->execPath = xstrdup(path);
68     con->execAbsolute = allowAbsolute;
72 static void invokeCallbacksPRE(poptContext con, const struct poptOption * opt)
81 	    invokeCallbacksPRE(con, arg.opt);
86 	    arg.cb(con, POPT_CALLBACK_REASON_PRE, NULL, NULL, opt->descrip);
92 static void invokeCallbacksPOST(poptContext con, const struct poptOption * opt)
101 	    invokeCallbacksPOST(con, arg.opt);
106 	    arg.cb(con, POPT_CALLBACK_REASON_POST, NULL, NULL, opt->descrip);
112 static void invokeCallbacksOPTION(poptContext con,
127 		invokeCallbacksOPTION(con, opt->arg, myOpt, myData, shorty);
143 		cbarg.cb(con, POPT_CALLBACK_REASON_OPTION,
144 			myOpt, con->os->nextArg, cbData);
157     poptContext con = malloc(sizeof(*con));
159     if (con == NULL) return NULL;	/* XXX can't happen */
160     memset(con, 0, sizeof(*con));
162     con->os = con->optionStack;
163     con->os->argc = argc;
164     con->os->argv = argv;
165     con->os->argb = NULL;
168 	con->os->next = 1;		/* skip argv[0] */
170     con->leftovers = calloc( (size_t)(argc + 1), sizeof(*con->leftovers) );
171     con->allocLeftovers = argc + 1;
172     con->options = options;
173     con->aliases = NULL;
174     con->numAliases = 0;
175     con->flags = flags;
176     con->execs = NULL;
177     con->numExecs = 0;
178     con->execFail = NULL;
179     con->finalArgvAlloced = argc * 2;
180     con->finalArgv = calloc( (size_t)con->finalArgvAlloced, sizeof(*con->finalArgv) );
181     con->execAbsolute = 1;
182     con->arg_strip = NULL;
185 	con->flags |= POPT_CONTEXT_POSIXMEHARDER;
188 	con->appName = xstrdup(name);
190     invokeCallbacksPRE(con, con->options);
192     return con;
202 void poptResetContext(poptContext con)
206     if (con == NULL) return;
207     while (con->os > con->optionStack) {
208 	cleanOSE(con->os--);
210     con->os->argb = PBM_FREE(con->os->argb);
211     con->os->currAlias = NULL;
212     con->os->nextCharArg = NULL;
213     con->os->nextArg = _free(con->os->nextArg);
214     if (!(con->flags & POPT_CONTEXT_KEEP_FIRST))
215 	con->os->next = 1;		/* skip argv[0] */
217 	con->os->next = 0;
219     for (i = 0; i < con->numLeftovers; i++) {
220         con->leftovers[i] = _free(con->leftovers[i]);
222     con->numLeftovers = 0;
223     con->nextLeftover = 0;
224     con->restLeftover = 0;
225     con->doExec = NULL;
226     con->execFail = _free(con->execFail);
228     if (con->finalArgv != NULL)
229     for (i = 0; i < con->finalArgvCount; i++) {
230 	con->finalArgv[i] = _free(con->finalArgv[i]);
233     con->finalArgvCount = 0;
234     con->arg_strip = PBM_FREE(con->arg_strip);
239 static int handleExec(poptContext con,
245     if (con->execs == NULL || con->numExecs <= 0) /* XXX can't happen */
248     for (i = con->numExecs - 1; i >= 0; i--) {
249 	item = con->execs + i;
260     if (con->flags & POPT_CONTEXT_NO_EXEC)
263     if (con->doExec == NULL) {
264 	con->doExec = con->execs + i;
270     if ((con->finalArgvCount + 1) >= (con->finalArgvAlloced)) {
271 	con->finalArgvAlloced += 10;
272 	con->finalArgv = realloc(con->finalArgv,
273 			sizeof(*con->finalArgv) * con->finalArgvAlloced);
276     i = con->finalArgvCount++;
277     if (con->finalArgv != NULL)	/* XXX can't happen */
280 	    con->finalArgv[i] = s;
288 	    con->finalArgv[i] = NULL;
333 static int handleAlias(poptContext con,
338     poptItem item = con->os->currAlias;
351     if (con->aliases == NULL || con->numAliases <= 0) /* XXX can't happen */
354     for (i = con->numAliases - 1; i >= 0; i--) {
355 	item = con->aliases + i;
367     if ((con->os - con->optionStack + 1) == POPT_OPTION_DEPTH)
371 	con->os->nextCharArg = nextArg;
373     con->os++;
374     con->os->next = 0;
375     con->os->stuffed = 0;
376     con->os->nextArg = NULL;
377     con->os->nextCharArg = NULL;
378     con->os->currAlias = con->aliases + i;
380 	int ac = con->os->currAlias->argc;
386 		    av[i] = con->os->currAlias->argv[i];
391 		av = con->os->currAlias->argv;
393 	    av = con->os->currAlias->argv;
394 	rc = poptDupArgv(ac, av, &con->os->argc, &con->os->argv);
395 	if (av != NULL && av != con->os->currAlias->argv)
398     con->os->argb = NULL;
448 static int execCommand(poptContext con)
450     poptItem item = con->doExec;
460 	(!con->execAbsolute && strchr(item->argv[0], '/')))
464 			(6 + item->argc + con->numLeftovers + con->finalArgvCount));
467     if (!strchr(item->argv[0], '/') && con->execPath != NULL) {
468 	char *s = malloc(strlen(con->execPath) + strlen(item->argv[0]) + sizeof("/"));
470 	    (void)stpcpy(stpcpy(stpcpy(s, con->execPath), "/"), item->argv[0]);
485     if (con->finalArgv != NULL && con->finalArgvCount > 0) {
486 	memcpy(argv + argc, con->finalArgv,
487 		sizeof(*argv) * con->finalArgvCount);
488 	argc += con->finalArgvCount;
491     if (con->leftovers != NULL && con->numLeftovers > 0) {
492 	memcpy(argv + argc, con->leftovers, sizeof(*argv) * con->numLeftovers);
493 	argc += con->numLeftovers;
541     con->execFail = xstrdup(argv[0]);
615 static const char * findNextArg(poptContext con,
618     struct optionStackEntry * os = con->os;
624 	while (os->next == os->argc && os > con->optionStack) os--;
625 	if (os->next == os->argc && os == con->optionStack) break;
642 	if (os > con->optionStack) os--;
648 expandNextArg(poptContext con, const char * s)
670 		if ((a = findNextArg(con, 1U, 1)) == NULL)
701 static void poptStripArg(poptContext con, int which)
703     if (con->arg_strip == NULL)
704 	con->arg_strip = PBM_ALLOC(con->optionStack[0].argc);
705     if (con->arg_strip != NULL)		/* XXX can't happen */
706     PBM_SET(which, con->arg_strip);
844 int poptBitsArgs(poptContext con, poptBits *ap)
849     if (con == NULL || ap == NULL || _poptBitsNew(ap) ||
850 	con->leftovers == NULL || con->numLeftovers == con->nextLeftover)
854     con->leftovers[con->numLeftovers] = NULL;
856     for (av = con->leftovers + con->nextLeftover; *av != NULL; av++) {
1048  * @param con		context
1052 static unsigned int poptArgInfo(poptContext con, const struct poptOption * opt)
1056     if (con->os->argv != NULL && con->os->next > 0 && opt->longName != NULL)
1058 	const char * longName = con->os->argv[con->os->next-1];
1100  * @param con		context
1104 static int poptSaveArg(poptContext con, const struct poptOption * opt)
1112 	rc = poptSaveBits(arg.ptr, opt->argInfo, con->os->nextArg);
1116 	rc = poptSaveString(arg.ptr, opt->argInfo, con->os->nextArg);
1120 	arg.argv[0] = (con->os->nextArg) ? xstrdup(con->os->nextArg) : NULL;
1127     {	unsigned int argInfo = poptArgInfo(con, opt);
1130 	if ((rc = poptParseInteger(&aNUM, argInfo, con->os->nextArg)) != 0)
1167 	if (con->os->nextArg) {
1170 	    aDouble = strtod(con->os->nextArg, &end);
1197 	con->maincall = opt->arg;
1209 int poptGetNextOpt(poptContext con)
1214     if (con == NULL)
1224 	while (!con->os->nextCharArg && con->os->next == con->os->argc
1225 		&& con->os > con->optionStack) {
1226 	    cleanOSE(con->os--);
1228 	if (!con->os->nextCharArg && con->os->next == con->os->argc) {
1229 	    invokeCallbacksPOST(con, con->options);
1231 	    if (con->maincall) {
1232 		(void) (*con->maincall) (con->finalArgvCount, con->finalArgv);
1236 	    if (con->doExec) return execCommand(con);
1241 	if (!con->os->nextCharArg) {
1246 	    if (con->os->argb && PBM_ISSET(con->os->next, con->os->argb)) {
1247 		con->os->next++;
1250 	    thisopt = con->os->next;
1251 	    if (con->os->argv != NULL)	/* XXX can't happen */
1252 	    origOptString = con->os->argv[con->os->next++];
1257 	    if (con->restLeftover || *origOptString != '-' ||
1260 		if (con->flags & POPT_CONTEXT_POSIXMEHARDER)
1261 		    con->restLeftover = 1;
1262 		if (con->flags & POPT_CONTEXT_ARG_OPTS) {
1263 		    con->os->nextArg = xstrdup(origOptString);
1266 		if (con->leftovers != NULL) {	/* XXX can't happen */
1270 		    if ((con->numLeftovers + 1) >= (con->allocLeftovers)) {
1271 			con->allocLeftovers += 10;
1272 			con->leftovers =
1273 			    realloc(con->leftovers,
1274 				    sizeof(*con->leftovers) * con->allocLeftovers);
1276 		    con->leftovers[con->numLeftovers++]
1291 		con->restLeftover = 1;
1311 		if (handleAlias(con, optString, optStringLen, '\0', longArg)) {
1316 		if (handleExec(con, optString, '\0'))
1319 		opt = findOption(con->options, optString, optStringLen, '\0', &cb, &cbData,
1326 		con->os->nextCharArg = origOptString + 1;
1329 		if (con->os == con->optionStack && F_ISSET(opt, STRIP))
1332 		    poptStripArg(con, thisopt);
1339 	if (con->os->nextCharArg) {
1340 	    const char * nextCharArg = con->os->nextCharArg;
1342 	    con->os->nextCharArg = NULL;
1344 	    if (handleAlias(con, NULL, 0, *nextCharArg, nextCharArg + 1))
1347 	    if (handleExec(con, NULL, *nextCharArg)) {
1351 		    con->os->nextCharArg = nextCharArg;
1355 	    opt = findOption(con->options, NULL, 0, *nextCharArg, &cb,
1363 		con->os->nextCharArg = nextCharArg;
1368 	    if (longArg || (con->os->nextCharArg && con->os->nextCharArg[0] == '='))
1372 		unsigned int argInfo = poptArgInfo(con, opt);
1379 	    con->os->nextArg = _free(con->os->nextArg);
1381 		longArg = expandNextArg(con, longArg);
1382 		con->os->nextArg = (char *) longArg;
1383 	    } else if (con->os->nextCharArg) {
1384 		longArg = expandNextArg(con, con->os->nextCharArg + (int)(*con->os->nextCharArg == '='));
1385 		con->os->nextArg = (char *) longArg;
1386 		con->os->nextCharArg = NULL;
1388 		while (con->os->next == con->os->argc &&
1389 			con->os > con->optionStack)
1391 		    cleanOSE(con->os--);
1393 		if (con->os->next == con->os->argc) {
1396 		    con->os->nextArg = NULL;
1403 		    if (con->os == con->optionStack
1406 			poptStripArg(con, con->os->next);
1409 		    if (con->os->argv != NULL) {	/* XXX can't happen */
1411 			    con->os->argv[con->os->next][0] == '-') {
1412 			    con->os->nextArg = NULL;
1415 			    longArg = con->os->argv[con->os->next++];
1416 			    longArg = expandNextArg(con, longArg);
1417 			    con->os->nextArg = (char *) longArg;
1425 	    if (opt->arg != NULL && (rc = poptSaveArg(con, opt)) != 0)
1430 	    invokeCallbacksOPTION(con, con->options, opt, cbData, shorty);
1434 	if ((con->finalArgvCount + 2) >= (con->finalArgvAlloced)) {
1435 	    con->finalArgvAlloced += 10;
1436 	    con->finalArgv = realloc(con->finalArgv,
1437 			    sizeof(*con->finalArgv) * con->finalArgvAlloced);
1440 	if (con->finalArgv != NULL)
1443 		con->finalArgv[con->finalArgvCount++] = s;
1454 		con->finalArgv[con->finalArgvCount++] = NULL;
1462 	    if (con->finalArgv != NULL && con->os->nextArg != NULL)
1463 	        con->finalArgv[con->finalArgvCount++] =
1464 			xstrdup(con->os->nextArg);
1471 char * poptGetOptArg(poptContext con)
1474     if (con) {
1475 	ret = con->os->nextArg;
1476 	con->os->nextArg = NULL;
1481 const char * poptGetArg(poptContext con)
1484     if (con && con->leftovers != NULL && con->nextLeftover < con->numLeftovers)
1485 	ret = con->leftovers[con->nextLeftover++];
1489 const char * poptPeekArg(poptContext con)
1492     if (con && con->leftovers != NULL && con->nextLeftover < con->numLeftovers)
1493 	ret = con->leftovers[con->nextLeftover];
1497 const char ** poptGetArgs(poptContext con)
1499     if (con == NULL ||
1500 	con->leftovers == NULL || con->numLeftovers == con->nextLeftover)
1504     con->leftovers[con->numLeftovers] = NULL;
1506     return (con->leftovers + con->nextLeftover);
1526 poptContext poptFreeContext(poptContext con)
1530     if (con == NULL) return con;
1531     poptResetContext(con);
1533     con->aliases = poptFreeItems(con->aliases, con->numAliases);
1534     con->numAliases = 0;
1536     con->execs = poptFreeItems(con->execs, con->numExecs);
1537     con->numExecs = 0;
1539     for (i = 0; i < con->numLeftovers; i++) {
1540         con->leftovers[i] = _free(con->leftovers[i]);
1542     con->leftovers = _free(con->leftovers);
1544     con->finalArgv = _free(con->finalArgv);
1545     con->appName = _free(con->appName);
1546     con->otherHelp = _free(con->otherHelp);
1547     con->execPath = _free(con->execPath);
1548     con->arg_strip = PBM_FREE(con->arg_strip);
1550     con = _free(con);
1551     return con;
1554 int poptAddAlias(poptContext con, struct poptAlias alias,
1569     return poptAddItem(con, item, 0);
1572 int poptAddItem(poptContext con, poptItem newItem, int flags)
1579 	items = &con->execs;
1580 	nitems = &con->numExecs;
1583 	items = &con->aliases;
1584 	nitems = &con->numAliases;
1616 const char * poptBadOption(poptContext con, unsigned int flags)
1621     if (con != NULL) {
1623        if (con->execFail) {
1624            badOpt = con->execFail;
1626            os = (flags & POPT_BADOPTION_NOALIAS) ? con->optionStack : con->os;
1666 int poptStuffArgs(poptContext con, const char ** argv)
1671     if ((con->os - con->optionStack) == POPT_OPTION_DEPTH)
1677     con->os++;
1678     con->os->next = 0;
1679     con->os->nextArg = NULL;
1680     con->os->nextCharArg = NULL;
1681     con->os->currAlias = NULL;
1682     rc = poptDupArgv(argc, argv, &con->os->argc, &con->os->argv);
1683     con->os->argb = NULL;
1684     con->os->stuffed = 1;
1689 const char * poptGetInvocationName(poptContext con)
1691     return (con->os->argv ? con->os->argv[0] : "");
1694 int poptStrippedArgv(poptContext con, int argc, char ** argv)
1700     if (con->arg_strip)
1702 	if (PBM_ISSET(i, con->arg_strip))
1707 	if (con->arg_strip && PBM_ISSET(i, con->arg_strip))

Indexes created Thu Nov 07 10:32:03 CST 2024