1 /* $NetBSD: main.c,v 1.4 2023/06/19 21:41:42 christos Exp $ */
2
3 /*
4 * Copyright (c) 1997-2005 Kungliga Tekniska Högskolan
5 * (Royal Institute of Technology, Stockholm, Sweden).
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 *
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 *
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * 3. Neither the name of the Institute nor the names of its contributors
20 * may be used to endorse or promote products derived from this software
21 * without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * SUCH DAMAGE.
34 */
35
36 #include "gen_locl.h"
37 #include "getarg.h"
38 #include "lex.h"
39
40 extern FILE *yyin;
41
42 static getarg_strings preserve;
43 static getarg_strings seq;
44
45 int
preserve_type(const char * p)46 preserve_type(const char *p)
47 {
48 int i;
49 for (i = 0; i < preserve.num_strings; i++)
50 if (strcmp(preserve.strings[i], p) == 0)
51 return 1;
52 return 0;
53 }
54
55 int
seq_type(const char * p)56 seq_type(const char *p)
57 {
58 int i;
59 for (i = 0; i < seq.num_strings; i++)
60 if (strcmp(seq.strings[i], p) == 0)
61 return 1;
62 return 0;
63 }
64
65 const char *fuzzer_string = "";
66 int fuzzer_flag;
67 int support_ber;
68 int template_flag;
69 int rfc1510_bitstring;
70 int one_code_file;
71 char *option_file;
72 int parse_units_flag = 1;
73 char *type_file_string = "krb5-types.h";
74 int version_flag;
75 int help_flag;
76 struct getargs args[] = {
77 { "fuzzer", 0, arg_flag, &fuzzer_flag, NULL, NULL },
78 { "template", 0, arg_flag, &template_flag, NULL, NULL },
79 { "encode-rfc1510-bit-string", 0, arg_flag, &rfc1510_bitstring, NULL, NULL },
80 { "decode-dce-ber", 0, arg_flag, &support_ber, NULL, NULL },
81 { "support-ber", 0, arg_flag, &support_ber, NULL, NULL },
82 { "preserve-binary", 0, arg_strings, &preserve, NULL, NULL },
83 { "sequence", 0, arg_strings, &seq, NULL, NULL },
84 { "one-code-file", 0, arg_flag, &one_code_file, NULL, NULL },
85 { "option-file", 0, arg_string, &option_file, NULL, NULL },
86 { "parse-units", 0, arg_negative_flag, &parse_units_flag, NULL, NULL },
87 { "type-file", 0, arg_string, &type_file_string, NULL, NULL },
88 { "version", 0, arg_flag, &version_flag, NULL, NULL },
89 { "help", 0, arg_flag, &help_flag, NULL, NULL }
90 };
91 int num_args = sizeof(args) / sizeof(args[0]);
92
93 static void
usage(int code)94 usage(int code)
95 {
96 arg_printusage(args, num_args, NULL, "[asn1-file [name]]");
97 exit(code);
98 }
99
100 int error_flag;
101
102 int
main(int argc,char ** argv)103 main(int argc, char **argv)
104 {
105 int ret;
106 const char *file;
107 const char *name = NULL;
108 int optidx = 0;
109 char **arg = NULL;
110 int len = 0, i;
111
112 setprogname(argv[0]);
113 if(getarg(args, num_args, argc, argv, &optidx))
114 usage(1);
115 if(help_flag)
116 usage(0);
117 if(version_flag) {
118 print_version(NULL);
119 exit(0);
120 }
121 if (argc == optidx) {
122 file = "stdin";
123 name = "stdin";
124 yyin = stdin;
125 } else {
126 file = argv[optidx];
127 yyin = fopen (file, "r");
128 if (yyin == NULL)
129 err (1, "open %s", file);
130 if (argc == optidx + 1) {
131 char *p;
132 name = estrdup(file);
133 p = strrchr(name, '.');
134 if (p)
135 *p = '\0';
136 } else
137 name = argv[optidx + 1];
138 }
139
140 /*
141 * Parse extra options file
142 */
143 if (option_file) {
144 char buf[1024];
145 FILE *opt;
146
147 opt = fopen(option_file, "r");
148 if (opt == NULL) {
149 perror("open");
150 exit(1);
151 }
152
153 arg = calloc(2, sizeof(arg[0]));
154 if (arg == NULL) {
155 perror("calloc");
156 exit(1);
157 }
158 arg[0] = option_file;
159 arg[1] = NULL;
160 len = 1;
161
162 while (fgets(buf, sizeof(buf), opt) != NULL) {
163 buf[strcspn(buf, "\n\r")] = '\0';
164
165 arg = realloc(arg, (len + 2) * sizeof(arg[0]));
166 if (arg == NULL) {
167 perror("malloc");
168 exit(1);
169 }
170 arg[len] = strdup(buf);
171 if (arg[len] == NULL) {
172 perror("strdup");
173 exit(1);
174 }
175 arg[len + 1] = NULL;
176 len++;
177 }
178 fclose(opt);
179
180 optidx = 0;
181 if(getarg(args, num_args, len, arg, &optidx))
182 usage(1);
183
184 if (len != optidx) {
185 fprintf(stderr, "extra args");
186 exit(1);
187 }
188 }
189
190 if (fuzzer_flag) {
191 if (!template_flag) {
192 printf("can't do fuzzer w/o --template");
193 exit(1);
194 }
195 #ifdef ASN1_FUZZER
196 fuzzer_string = "_fuzzer";
197 #endif
198 }
199
200
201 init_generate (file, name);
202
203 if (one_code_file)
204 generate_header_of_codefile(name);
205
206 initsym ();
207 ret = yyparse ();
208 if(ret != 0 || error_flag != 0)
209 exit(1);
210 close_generate ();
211 if (argc != optidx)
212 fclose(yyin);
213
214 if (one_code_file)
215 close_codefile();
216
217 if (arg) {
218 for (i = 1; i < len; i++)
219 free(arg[i]);
220 free(arg);
221 }
222
223 return 0;
224 }
225