zilasm/main: Extend wrong_arg routine.
[zilutils.git] / zilasm / main.c
1 /*
2  * main.c
3  *
4  * Copyright (C) 2015 Alexander Andrejevic <theflash AT sdf DOT lonestar DOT org>
5  * Copyright (C) 2015 Jason Self <j@jxself.org>
6  *
7  * This program is free software: you can redistribute it and/or modify
8  * it under the terms of the GNU Affero General Public License as
9  * published by the Free Software Foundation, either version 3 of the
10  * License, or (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU Affero General Public License for more details.
16  *
17  * You should have received a copy of the GNU Affero General Public License
18  * along with this program.  If not, see <http://www.gnu.org/licenses/>
19  */
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <stdarg.h>
24 #include <string.h>
25 #include <getopt.h>
26 #include <time.h>
27 #include <ctype.h>
28
29 #include "config.h"
30
31 enum { ZVERSION = 11, ZORKID, ZSERIAL };
32
33 static struct option const long_options[] =
34 {
35     { "help",     no_argument,       NULL, 'h' },
36     { "version",  no_argument,       NULL, 'V' },
37     { "zversion", required_argument, NULL, ZVERSION },
38     { "zorkid",   required_argument, NULL, ZORKID   },
39     { "serial",   required_argument, NULL, ZSERIAL  },
40     { NULL, 0, NULL, 0 }
41
42 };
43
44 struct
45 {
46     int  zversion;     /* 0 - 8     */
47     int  zorkid;       /* 0 - 65535 */
48     char zserial[7];   /* YYMMDD    */
49 } Config;
50
51 void wrong_arg(const char *err, ...)
52 {
53     if (err)
54     {
55         va_list ap;
56         va_start(ap, err);
57         vfprintf(stderr, err, ap);
58         va_end(ap);
59     }
60     fprintf(stderr, "Try `" PACKAGE_NAME " --help' for more information.\n");
61     exit(1);
62 }
63
64 void print_version()
65 {
66     printf( PACKAGE_STRING "\n"
67             "License AGPLv3+: GNU AGPL version 3 or later\n"
68             "<http://gnu.org/licenses/agpl.html>\n"
69             "This is free software: you are free to change and redistribute it.\n"
70             "There is NO WARRANTY, to the extent permitted by law.\n"
71           );
72     exit(0);
73 }
74
75 void print_usage(int failed)
76 {
77     printf("Usage: " PACKAGE_NAME " [OPTION...] [FILES...]\n"
78            "\n"
79            "--version  Display program version and exit\n"
80            "--help     Display this help\n"
81            "\n"
82            "--zversion (accepts numbers 1 - 8, defaults to 1 if not specified)\n"
83            "--zorkid   (integer between 0 and 65535, defaults to 0 if not specified)\n"
84            "--serial   (six characters of ASCII, defaults to current date\n"
85            "            in the form YYMMDD if not specified)\n"
86           );
87     exit(failed);
88 }
89
90 void fill_zserial(void)
91 {
92     time_t t;
93     struct tm *timeinfo;
94     time (&t);
95     timeinfo = localtime(&t);
96     strftime (Config.zserial, sizeof(Config.zserial), "%y%m%d", timeinfo);
97 }
98
99 void fill_config(void)
100 {
101     Config.zversion = 6;
102     Config.zorkid   = 0;
103     fill_zserial();
104 }
105
106 void parse_intarg(int *dest, const char name[], int min, int max, int defval)
107 {
108     if (!optarg)
109     {
110         *dest = defval;
111         return;
112     }
113     int n = atoi(optarg);
114     if (n >= min && n <= max)
115     {
116         *dest = n;
117         return;
118     }
119     wrong_arg("Wrong %s value %s, must be integer between %d and %d\n",
120               name, optarg, min, max);
121 }
122
123 void parse_zserial(void)
124 {
125     if (!optarg)
126     {
127         fill_zserial();
128         return;
129     }
130     size_t n = strlen(optarg);
131     if (n == sizeof(Config.zserial) - 1)
132     {
133         char *p = optarg;
134         while (*p && isalnum(*p))
135             p++;
136         if (!*p)    /* ..optarg contains alphanumeric only? */
137         {
138             strncpy(Config.zserial, optarg, sizeof(Config.zserial));
139             return;
140         }
141     }
142     wrong_arg("Wrong zserial value %s, must be 6 ascii characters\n", optarg);
143 }
144
145 int main(int argc, char *argv[], char *envp[])
146 {
147     fill_config();
148
149     int opt = 0;
150     while ((opt = getopt_long (argc, argv, "hV", long_options, NULL)) != -1)
151     {
152         switch(opt)
153         {
154         case 'h'     :
155             print_usage(0);
156         case 'V'     :
157             print_version();
158         case ZVERSION:
159             parse_intarg(&Config.zversion, "zversion", 1, 8,      1);
160             break;
161         case ZORKID  :
162             parse_intarg(&Config.zorkid,   "zorkid",   0, 0xFFFF, 0);
163             break;
164         case ZSERIAL :
165             parse_zserial();
166             break;
167         default      :
168             wrong_arg(0);
169         }
170     }
171
172     // TODO: Everything :)
173
174     printf("Config:\n"
175            "- ZVersion: %d\n"
176            "- ZorkID:   %d\n"
177            "- ZSerial:  %s\n",
178            Config.zversion, Config.zorkid, Config.zserial
179           );
180
181     return 0;
182 }