Re-import of fprobe-ulog
[fprobe-ulog.git] / src / my_getopt.c
1 /*
2         Copyright (C) Slava Astashonok <sla@0n.ru>
3
4         This program is free software; you can redistribute it and/or
5         modify it under the terms of the GNU General Public License.
6
7         $Id: my_getopt.c,v 1.2.2.3 2004/02/02 08:06:24 sla Exp $
8 */
9
10 #include <common.h>
11
12 /* getopt() */
13 #include <unistd.h>
14
15 /* fprintf() */
16 #include <stdio.h>
17
18 #include <my_getopt.h>
19
20 extern char *optarg;
21 extern int optind, opterr, optopt;
22
23 int my_getopt(int argc, char * const argv[],
24         struct getopt_parms parms[])
25 {
26         int c, i, p;
27         static int flag = 0, my_opterr = 1;
28         static char optstring[MY_GETOPT_MAX_OPTSTR];
29
30         if (flag++ == 0) {
31                 p = 0; i = 0;
32                 while ((parms[i].name != 0) && (p < (MY_GETOPT_MAX_OPTSTR - 1))) {
33                         optstring[p++] = parms[i].name;
34                         if (parms[i].flag & MY_GETOPT_ARG_REQUIRED) optstring[p++] = ':';
35                         optstring[p] = 0;
36                         i++;
37                 }
38                 flag = 1;
39         }
40
41         c = getopt(argc, argv, optstring);
42         switch (c) {
43                 case '?':
44                         if (my_opterr) fprintf(stderr, "Wrong parameters\n");
45                         break;
46
47                 case -1:
48                         flag = 0;
49                         i = 0;
50                         while (parms[i].name != 0) {
51                                 if ((parms[i].flag & MY_GETOPT_REQUIRED) && (parms[i].count == 0)) {
52                                         if (my_opterr) fprintf(stderr, "Missing required option\n");
53                                         return '?';
54                                 }
55                                 i++;
56                         }
57                         break;
58
59                 default:
60                         i = 0;
61                         while ((parms[i].name != 0) && (parms[i].name != c)) {
62                                 i++;
63                         }
64                         if (parms[i].flag & MY_GETOPT_ARG_REQUIRED) {
65                                 if (optarg == 0) {
66                                         if (my_opterr) fprintf(stderr, "Option `-%c': %s\n",
67                                                         c, "require parameter");
68                                         return '?';
69                                 } else parms[i].arg = optarg;
70                         }
71                         if ((++parms[i].count > 1) && !(parms[i].flag & MY_GETOPT_ALLOW_REPEAT)) {
72                                 if (my_opterr) fprintf(stderr, "Option `-%c': %s\n",
73                                                 c, "repeat not allowed");
74                                 return '?';
75                         }
76                         break;
77         }
78         return c;
79 }