Import the arg.h from st with the fixed oob bug.

This commit is contained in:
Christoph Lohmann 2016-02-21 17:22:58 +01:00
parent 3c79bcb408
commit 4ccaba4968

44
arg.h
View file

@ -1,52 +1,48 @@
/* See the LICENSE file for copyright and license details. */ /*
* Copy me if you can.
* by 20h
*/
#ifndef __ARG_H__ #ifndef ARG_H__
#define __ARG_H__ #define ARG_H__
extern char *argv0; extern char *argv0;
#define USED(x) ((void)(x))
/* use main(int argc, char *argv[]) */ /* use main(int argc, char *argv[]) */
#define ARGBEGIN for (argv0 = *argv, argv++, argc--;\ #define ARGBEGIN for (argv0 = *argv, argv++, argc--;\
argv[0] && argv[0][1]\ argv[0] && argv[0][0] == '-'\
&& argv[0][0] == '-';\ && argv[0][1];\
argc--, argv++) {\ argc--, argv++) {\
char _argc;\ char argc_;\
char **_argv;\ char **argv_;\
int brk;\ int brk_;\
if (argv[0][1] == '-' && argv[0][2] == '\0') {\ if (argv[0][1] == '-' && argv[0][2] == '\0') {\
argv++;\ argv++;\
argc--;\ argc--;\
break;\ break;\
}\ }\
for (brk = 0, argv[0]++, _argv = argv;\ for (brk_ = 0, argv[0]++, argv_ = argv;\
argv[0][0] && !brk;\ argv[0][0] && !brk_;\
argv[0]++) {\ argv[0]++) {\
if (_argv != argv)\ if (argv_ != argv)\
break;\ break;\
_argc = argv[0][0];\ argc_ = argv[0][0];\
switch (_argc) switch (argc_)
#define ARGEND }\ #define ARGEND }\
USED(_argc);\ }
}\
USED(argv);\
USED(argc);
#define ARGC() _argc #define ARGC() argc_
#define EARGF(x) ((argv[0][1] == '\0' && argv[1] == NULL)?\ #define EARGF(x) ((argv[0][1] == '\0' && argv[1] == NULL)?\
((x), abort(), (char *)0) :\ ((x), abort(), (char *)0) :\
(brk = 1, (argv[0][1] != '\0')?\ (brk_ = 1, (argv[0][1] != '\0')?\
(&argv[0][1]) :\ (&argv[0][1]) :\
(argc--, argv++, argv[0]))) (argc--, argv++, argv[0])))
#define ARGF() ((argv[0][1] == '\0' && argv[1] == NULL)?\ #define ARGF() ((argv[0][1] == '\0' && argv[1] == NULL)?\
(char *)0 :\ (char *)0 :\
(brk = 1, (argv[0][1] != '\0')?\ (brk_ = 1, (argv[0][1] != '\0')?\
(&argv[0][1]) :\ (&argv[0][1]) :\
(argc--, argv++, argv[0]))) (argc--, argv++, argv[0])))
#endif #endif