1 /*
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <string.h>
14 #include <limits.h>
15 #include "args.h"
16
17 #include "vpx/vpx_integer.h"
18
19 #if defined(__GNUC__)
20 __attribute__((noreturn)) extern void die(const char *fmt, ...);
21 #elif defined(_MSC_VER)
22 __declspec(noreturn) extern void die(const char *fmt, ...);
23 #else
24 extern void die(const char *fmt, ...);
25 #endif
26
arg_init(char ** argv)27 struct arg arg_init(char **argv) {
28 struct arg a;
29
30 a.argv = argv;
31 a.argv_step = 1;
32 a.name = NULL;
33 a.val = NULL;
34 a.def = NULL;
35 return a;
36 }
37
arg_match(struct arg * arg_,const struct arg_def * def,char ** argv)38 int arg_match(struct arg *arg_, const struct arg_def *def, char **argv) {
39 struct arg arg;
40
41 if (!argv[0] || argv[0][0] != '-') return 0;
42
43 arg = arg_init(argv);
44
45 if (def->short_name && strlen(arg.argv[0]) == strlen(def->short_name) + 1 &&
46 !strcmp(arg.argv[0] + 1, def->short_name)) {
47 arg.name = arg.argv[0] + 1;
48 arg.val = def->has_val ? arg.argv[1] : NULL;
49 arg.argv_step = def->has_val ? 2 : 1;
50 } else if (def->long_name) {
51 const size_t name_len = strlen(def->long_name);
52
53 if (strlen(arg.argv[0]) >= name_len + 2 && arg.argv[0][1] == '-' &&
54 !strncmp(arg.argv[0] + 2, def->long_name, name_len) &&
55 (arg.argv[0][name_len + 2] == '=' ||
56 arg.argv[0][name_len + 2] == '\0')) {
57 arg.name = arg.argv[0] + 2;
58 arg.val = arg.name[name_len] == '=' ? arg.name + name_len + 1 : NULL;
59 arg.argv_step = 1;
60 }
61 }
62
63 if (arg.name && !arg.val && def->has_val)
64 die("Error: option %s requires argument.\n", arg.name);
65
66 if (arg.name && arg.val && !def->has_val)
67 die("Error: option %s requires no argument.\n", arg.name);
68
69 if (arg.name && (arg.val || !def->has_val)) {
70 arg.def = def;
71 *arg_ = arg;
72 return 1;
73 }
74
75 return 0;
76 }
77
arg_next(struct arg * arg)78 const char *arg_next(struct arg *arg) {
79 if (arg->argv[0]) arg->argv += arg->argv_step;
80
81 return *arg->argv;
82 }
83
argv_dup(int argc,const char ** argv)84 char **argv_dup(int argc, const char **argv) {
85 char **new_argv = malloc((argc + 1) * sizeof(*argv));
86 if (!new_argv) return NULL;
87
88 memcpy(new_argv, argv, argc * sizeof(*argv));
89 new_argv[argc] = NULL;
90 return new_argv;
91 }
92
arg_show_usage(FILE * fp,const struct arg_def * const * defs)93 void arg_show_usage(FILE *fp, const struct arg_def *const *defs) {
94 char option_text[40] = { 0 };
95
96 for (; *defs; defs++) {
97 const struct arg_def *def = *defs;
98 char *short_val = def->has_val ? " <arg>" : "";
99 char *long_val = def->has_val ? "=<arg>" : "";
100
101 if (def->short_name && def->long_name) {
102 char *comma = def->has_val ? "," : ", ";
103
104 snprintf(option_text, 37, "-%s%s%s --%s%6s", def->short_name, short_val,
105 comma, def->long_name, long_val);
106 } else if (def->short_name)
107 snprintf(option_text, 37, "-%s%s", def->short_name, short_val);
108 else if (def->long_name)
109 snprintf(option_text, 37, " --%s%s", def->long_name, long_val);
110
111 fprintf(fp, " %-37s\t%s\n", option_text, def->desc);
112
113 if (def->enums) {
114 const struct arg_enum_list *listptr;
115
116 fprintf(fp, " %-37s\t ", "");
117
118 for (listptr = def->enums; listptr->name; listptr++)
119 fprintf(fp, "%s%s", listptr->name, listptr[1].name ? ", " : "\n");
120 }
121 }
122 }
123
arg_parse_uint(const struct arg * arg)124 unsigned int arg_parse_uint(const struct arg *arg) {
125 uint32_t rawval;
126 char *endptr;
127
128 rawval = (uint32_t)strtoul(arg->val, &endptr, 10);
129
130 if (arg->val[0] != '\0' && endptr[0] == '\0') {
131 if (rawval <= UINT_MAX) return rawval;
132
133 die("Option %s: Value %ld out of range for unsigned int\n", arg->name,
134 rawval);
135 }
136
137 die("Option %s: Invalid character '%c'\n", arg->name, *endptr);
138 }
139
arg_parse_int(const struct arg * arg)140 int arg_parse_int(const struct arg *arg) {
141 int32_t rawval;
142 char *endptr;
143
144 rawval = (int32_t)strtol(arg->val, &endptr, 10);
145
146 if (arg->val[0] != '\0' && endptr[0] == '\0') {
147 if (rawval >= INT_MIN && rawval <= INT_MAX) return (int)rawval;
148
149 die("Option %s: Value %ld out of range for signed int\n", arg->name,
150 rawval);
151 }
152
153 die("Option %s: Invalid character '%c'\n", arg->name, *endptr);
154 }
155
156 struct vpx_rational {
157 int num; /**< fraction numerator */
158 int den; /**< fraction denominator */
159 };
arg_parse_rational(const struct arg * arg)160 struct vpx_rational arg_parse_rational(const struct arg *arg) {
161 long int rawval;
162 char *endptr;
163 struct vpx_rational rat;
164
165 /* parse numerator */
166 rawval = strtol(arg->val, &endptr, 10);
167
168 if (arg->val[0] != '\0' && endptr[0] == '/') {
169 if (rawval >= INT_MIN && rawval <= INT_MAX)
170 rat.num = (int)rawval;
171 else
172 die("Option %s: Value %ld out of range for signed int\n", arg->name,
173 rawval);
174 } else
175 die("Option %s: Expected / at '%c'\n", arg->name, *endptr);
176
177 /* parse denominator */
178 rawval = strtol(endptr + 1, &endptr, 10);
179
180 if (arg->val[0] != '\0' && endptr[0] == '\0') {
181 if (rawval >= INT_MIN && rawval <= INT_MAX)
182 rat.den = (int)rawval;
183 else
184 die("Option %s: Value %ld out of range for signed int\n", arg->name,
185 rawval);
186 } else
187 die("Option %s: Invalid character '%c'\n", arg->name, *endptr);
188
189 return rat;
190 }
191
arg_parse_enum(const struct arg * arg)192 int arg_parse_enum(const struct arg *arg) {
193 const struct arg_enum_list *listptr;
194 long int rawval;
195 char *endptr;
196
197 /* First see if the value can be parsed as a raw value */
198 rawval = strtol(arg->val, &endptr, 10);
199 if (arg->val[0] != '\0' && endptr[0] == '\0') {
200 /* Got a raw value, make sure it's valid */
201 for (listptr = arg->def->enums; listptr->name; listptr++)
202 if (listptr->val == rawval) return (int)rawval;
203 }
204
205 /* Next see if it can be parsed as a string */
206 for (listptr = arg->def->enums; listptr->name; listptr++)
207 if (!strcmp(arg->val, listptr->name)) return listptr->val;
208
209 die("Option %s: Invalid value '%s'\n", arg->name, arg->val);
210 }
211
arg_parse_enum_or_int(const struct arg * arg)212 int arg_parse_enum_or_int(const struct arg *arg) {
213 if (arg->def->enums) return arg_parse_enum(arg);
214 return arg_parse_int(arg);
215 }
216