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

This commit is contained in:
Christoph Lohmann 2016-02-22 06:47:46 +01:00 committed by Quentin Rameau
parent 934705c420
commit 581075796a
1 changed files with 27 additions and 20 deletions

47
arg.h
View File

@ -3,39 +3,46 @@
* by 20h * 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[]) */
#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_;\
if (argv[0][1] == '-' && argv[0][2] == '\0') {\ if (argv[0][1] == '-' && argv[0][2] == '\0') {\
argv++;\ argv++;\
argc--;\ argc--;\
break;\ break;\
}\ }\
for (argv[0]++, _argv = argv; argv[0][0];\ for (brk_ = 0, argv[0]++, argv_ = argv;\
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 EARGF(x) ((argv[1] == NULL)? ((x), abort(), (char *)0) :\ #define ARGC() argc_
(argc--, argv++, argv[0]))
#define EARGF(x) ((argv[0][1] == '\0' && argv[1] == NULL)?\
((x), abort(), (char *)0) :\
(brk_ = 1, (argv[0][1] != '\0')?\
(&argv[0][1]) :\
(argc--, argv++, argv[0])))
#define ARGF() ((argv[0][1] == '\0' && argv[1] == NULL)?\
(char *)0 :\
(brk_ = 1, (argv[0][1] != '\0')?\
(&argv[0][1]) :\
(argc--, argv++, argv[0])))
#endif #endif