1*0a6a1f1dSLionel Sambuc /* Id */
2eda6f593SDavid van Moolenbroek
3eda6f593SDavid van Moolenbroek /*
4eda6f593SDavid van Moolenbroek * Copyright (c) 2010 Nicholas Marriott <nicm@users.sourceforge.net>
5eda6f593SDavid van Moolenbroek *
6eda6f593SDavid van Moolenbroek * Permission to use, copy, modify, and distribute this software for any
7eda6f593SDavid van Moolenbroek * purpose with or without fee is hereby granted, provided that the above
8eda6f593SDavid van Moolenbroek * copyright notice and this permission notice appear in all copies.
9eda6f593SDavid van Moolenbroek *
10eda6f593SDavid van Moolenbroek * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11eda6f593SDavid van Moolenbroek * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12eda6f593SDavid van Moolenbroek * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13eda6f593SDavid van Moolenbroek * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14eda6f593SDavid van Moolenbroek * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
15eda6f593SDavid van Moolenbroek * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
16eda6f593SDavid van Moolenbroek * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17eda6f593SDavid van Moolenbroek */
18eda6f593SDavid van Moolenbroek
19eda6f593SDavid van Moolenbroek #include <sys/types.h>
20eda6f593SDavid van Moolenbroek
21eda6f593SDavid van Moolenbroek #include <stdlib.h>
22eda6f593SDavid van Moolenbroek #include <string.h>
23*0a6a1f1dSLionel Sambuc #include <unistd.h>
24eda6f593SDavid van Moolenbroek
25eda6f593SDavid van Moolenbroek #include "tmux.h"
26eda6f593SDavid van Moolenbroek
27*0a6a1f1dSLionel Sambuc /*
28*0a6a1f1dSLionel Sambuc * Manipulate command arguments.
29*0a6a1f1dSLionel Sambuc */
30*0a6a1f1dSLionel Sambuc
31*0a6a1f1dSLionel Sambuc struct args_entry *args_find(struct args *, u_char);
32*0a6a1f1dSLionel Sambuc
33*0a6a1f1dSLionel Sambuc RB_GENERATE(args_tree, args_entry, entry, args_cmp);
34*0a6a1f1dSLionel Sambuc
35*0a6a1f1dSLionel Sambuc /* Arguments tree comparison function. */
36*0a6a1f1dSLionel Sambuc int
args_cmp(struct args_entry * a1,struct args_entry * a2)37*0a6a1f1dSLionel Sambuc args_cmp(struct args_entry *a1, struct args_entry *a2)
38*0a6a1f1dSLionel Sambuc {
39*0a6a1f1dSLionel Sambuc return (a1->flag - a2->flag);
40*0a6a1f1dSLionel Sambuc }
41*0a6a1f1dSLionel Sambuc
42eda6f593SDavid van Moolenbroek /* Create an arguments set with no flags. */
43eda6f593SDavid van Moolenbroek struct args *
args_create(int argc,...)44eda6f593SDavid van Moolenbroek args_create(int argc, ...)
45eda6f593SDavid van Moolenbroek {
46eda6f593SDavid van Moolenbroek struct args *args;
47eda6f593SDavid van Moolenbroek va_list ap;
48eda6f593SDavid van Moolenbroek int i;
49eda6f593SDavid van Moolenbroek
50eda6f593SDavid van Moolenbroek args = xcalloc(1, sizeof *args);
51eda6f593SDavid van Moolenbroek
52eda6f593SDavid van Moolenbroek args->argc = argc;
53eda6f593SDavid van Moolenbroek if (argc == 0)
54eda6f593SDavid van Moolenbroek args->argv = NULL;
55eda6f593SDavid van Moolenbroek else
56eda6f593SDavid van Moolenbroek args->argv = xcalloc(argc, sizeof *args->argv);
57eda6f593SDavid van Moolenbroek
58eda6f593SDavid van Moolenbroek va_start(ap, argc);
59eda6f593SDavid van Moolenbroek for (i = 0; i < argc; i++)
60eda6f593SDavid van Moolenbroek args->argv[i] = xstrdup(va_arg(ap, char *));
61eda6f593SDavid van Moolenbroek va_end(ap);
62eda6f593SDavid van Moolenbroek
63eda6f593SDavid van Moolenbroek return (args);
64eda6f593SDavid van Moolenbroek }
65eda6f593SDavid van Moolenbroek
66*0a6a1f1dSLionel Sambuc /* Find a flag in the arguments tree. */
67*0a6a1f1dSLionel Sambuc struct args_entry *
args_find(struct args * args,u_char ch)68*0a6a1f1dSLionel Sambuc args_find(struct args *args, u_char ch)
69*0a6a1f1dSLionel Sambuc {
70*0a6a1f1dSLionel Sambuc struct args_entry entry;
71*0a6a1f1dSLionel Sambuc
72*0a6a1f1dSLionel Sambuc entry.flag = ch;
73*0a6a1f1dSLionel Sambuc return (RB_FIND(args_tree, &args->tree, &entry));
74*0a6a1f1dSLionel Sambuc }
75*0a6a1f1dSLionel Sambuc
76eda6f593SDavid van Moolenbroek /* Parse an argv and argc into a new argument set. */
77eda6f593SDavid van Moolenbroek struct args *
args_parse(const char * template,int argc,char ** argv)78eda6f593SDavid van Moolenbroek args_parse(const char *template, int argc, char **argv)
79eda6f593SDavid van Moolenbroek {
80eda6f593SDavid van Moolenbroek struct args *args;
81eda6f593SDavid van Moolenbroek int opt;
82eda6f593SDavid van Moolenbroek
83eda6f593SDavid van Moolenbroek args = xcalloc(1, sizeof *args);
84eda6f593SDavid van Moolenbroek
85eda6f593SDavid van Moolenbroek optreset = 1;
86eda6f593SDavid van Moolenbroek optind = 1;
87eda6f593SDavid van Moolenbroek
88eda6f593SDavid van Moolenbroek while ((opt = getopt(argc, argv, template)) != -1) {
89*0a6a1f1dSLionel Sambuc if (opt < 0)
90eda6f593SDavid van Moolenbroek continue;
91*0a6a1f1dSLionel Sambuc if (opt == '?' || strchr(template, opt) == NULL) {
92*0a6a1f1dSLionel Sambuc args_free(args);
93eda6f593SDavid van Moolenbroek return (NULL);
94eda6f593SDavid van Moolenbroek }
95*0a6a1f1dSLionel Sambuc args_set(args, opt, optarg);
96eda6f593SDavid van Moolenbroek }
97eda6f593SDavid van Moolenbroek argc -= optind;
98eda6f593SDavid van Moolenbroek argv += optind;
99eda6f593SDavid van Moolenbroek
100eda6f593SDavid van Moolenbroek args->argc = argc;
101eda6f593SDavid van Moolenbroek args->argv = cmd_copy_argv(argc, argv);
102eda6f593SDavid van Moolenbroek
103eda6f593SDavid van Moolenbroek return (args);
104eda6f593SDavid van Moolenbroek }
105eda6f593SDavid van Moolenbroek
106eda6f593SDavid van Moolenbroek /* Free an arguments set. */
107eda6f593SDavid van Moolenbroek void
args_free(struct args * args)108eda6f593SDavid van Moolenbroek args_free(struct args *args)
109eda6f593SDavid van Moolenbroek {
110*0a6a1f1dSLionel Sambuc struct args_entry *entry;
111*0a6a1f1dSLionel Sambuc struct args_entry *entry1;
112eda6f593SDavid van Moolenbroek
113eda6f593SDavid van Moolenbroek cmd_free_argv(args->argc, args->argv);
114eda6f593SDavid van Moolenbroek
115*0a6a1f1dSLionel Sambuc RB_FOREACH_SAFE(entry, args_tree, &args->tree, entry1) {
116*0a6a1f1dSLionel Sambuc RB_REMOVE(args_tree, &args->tree, entry);
117*0a6a1f1dSLionel Sambuc free(entry->value);
118*0a6a1f1dSLionel Sambuc free(entry);
119eda6f593SDavid van Moolenbroek }
120eda6f593SDavid van Moolenbroek
121*0a6a1f1dSLionel Sambuc free(args);
122eda6f593SDavid van Moolenbroek }
123eda6f593SDavid van Moolenbroek
124eda6f593SDavid van Moolenbroek /* Print a set of arguments. */
125eda6f593SDavid van Moolenbroek size_t
args_print(struct args * args,char * buf,size_t len)126eda6f593SDavid van Moolenbroek args_print(struct args *args, char *buf, size_t len)
127eda6f593SDavid van Moolenbroek {
128eda6f593SDavid van Moolenbroek size_t off;
129eda6f593SDavid van Moolenbroek int i;
130eda6f593SDavid van Moolenbroek const char *quotes;
131*0a6a1f1dSLionel Sambuc struct args_entry *entry;
132eda6f593SDavid van Moolenbroek
133eda6f593SDavid van Moolenbroek /* There must be at least one byte at the start. */
134eda6f593SDavid van Moolenbroek if (len == 0)
135eda6f593SDavid van Moolenbroek return (0);
136eda6f593SDavid van Moolenbroek off = 0;
137eda6f593SDavid van Moolenbroek
138eda6f593SDavid van Moolenbroek /* Process the flags first. */
139eda6f593SDavid van Moolenbroek buf[off++] = '-';
140*0a6a1f1dSLionel Sambuc RB_FOREACH(entry, args_tree, &args->tree) {
141*0a6a1f1dSLionel Sambuc if (entry->value != NULL)
142eda6f593SDavid van Moolenbroek continue;
143eda6f593SDavid van Moolenbroek
144eda6f593SDavid van Moolenbroek if (off == len - 1) {
145eda6f593SDavid van Moolenbroek buf[off] = '\0';
146eda6f593SDavid van Moolenbroek return (len);
147eda6f593SDavid van Moolenbroek }
148*0a6a1f1dSLionel Sambuc buf[off++] = entry->flag;
149eda6f593SDavid van Moolenbroek buf[off] = '\0';
150eda6f593SDavid van Moolenbroek }
151eda6f593SDavid van Moolenbroek if (off == 1)
152eda6f593SDavid van Moolenbroek buf[--off] = '\0';
153eda6f593SDavid van Moolenbroek
154eda6f593SDavid van Moolenbroek /* Then the flags with arguments. */
155*0a6a1f1dSLionel Sambuc RB_FOREACH(entry, args_tree, &args->tree) {
156*0a6a1f1dSLionel Sambuc if (entry->value == NULL)
157eda6f593SDavid van Moolenbroek continue;
158eda6f593SDavid van Moolenbroek
159eda6f593SDavid van Moolenbroek if (off >= len) {
160eda6f593SDavid van Moolenbroek /* snprintf will have zero terminated. */
161eda6f593SDavid van Moolenbroek return (len);
162eda6f593SDavid van Moolenbroek }
163eda6f593SDavid van Moolenbroek
164*0a6a1f1dSLionel Sambuc if (strchr(entry->value, ' ') != NULL)
165eda6f593SDavid van Moolenbroek quotes = "\"";
166eda6f593SDavid van Moolenbroek else
167eda6f593SDavid van Moolenbroek quotes = "";
168eda6f593SDavid van Moolenbroek off += xsnprintf(buf + off, len - off, "%s-%c %s%s%s",
169*0a6a1f1dSLionel Sambuc off != 0 ? " " : "", entry->flag, quotes, entry->value,
170*0a6a1f1dSLionel Sambuc quotes);
171eda6f593SDavid van Moolenbroek }
172eda6f593SDavid van Moolenbroek
173eda6f593SDavid van Moolenbroek /* And finally the argument vector. */
174eda6f593SDavid van Moolenbroek for (i = 0; i < args->argc; i++) {
175eda6f593SDavid van Moolenbroek if (off >= len) {
176eda6f593SDavid van Moolenbroek /* snprintf will have zero terminated. */
177eda6f593SDavid van Moolenbroek return (len);
178eda6f593SDavid van Moolenbroek }
179eda6f593SDavid van Moolenbroek
180eda6f593SDavid van Moolenbroek if (strchr(args->argv[i], ' ') != NULL)
181eda6f593SDavid van Moolenbroek quotes = "\"";
182eda6f593SDavid van Moolenbroek else
183eda6f593SDavid van Moolenbroek quotes = "";
184eda6f593SDavid van Moolenbroek off += xsnprintf(buf + off, len - off, "%s%s%s%s",
185eda6f593SDavid van Moolenbroek off != 0 ? " " : "", quotes, args->argv[i], quotes);
186eda6f593SDavid van Moolenbroek }
187eda6f593SDavid van Moolenbroek
188eda6f593SDavid van Moolenbroek return (off);
189eda6f593SDavid van Moolenbroek }
190eda6f593SDavid van Moolenbroek
191eda6f593SDavid van Moolenbroek /* Return if an argument is present. */
192eda6f593SDavid van Moolenbroek int
args_has(struct args * args,u_char ch)193eda6f593SDavid van Moolenbroek args_has(struct args *args, u_char ch)
194eda6f593SDavid van Moolenbroek {
195*0a6a1f1dSLionel Sambuc return (args_find(args, ch) == NULL ? 0 : 1);
196eda6f593SDavid van Moolenbroek }
197eda6f593SDavid van Moolenbroek
198*0a6a1f1dSLionel Sambuc /* Set argument value in the arguments tree. */
199eda6f593SDavid van Moolenbroek void
args_set(struct args * args,u_char ch,const char * value)200eda6f593SDavid van Moolenbroek args_set(struct args *args, u_char ch, const char *value)
201eda6f593SDavid van Moolenbroek {
202*0a6a1f1dSLionel Sambuc struct args_entry *entry;
203*0a6a1f1dSLionel Sambuc
204*0a6a1f1dSLionel Sambuc /* Replace existing argument. */
205*0a6a1f1dSLionel Sambuc if ((entry = args_find(args, ch)) != NULL) {
206*0a6a1f1dSLionel Sambuc free(entry->value);
207*0a6a1f1dSLionel Sambuc entry->value = NULL;
208*0a6a1f1dSLionel Sambuc } else {
209*0a6a1f1dSLionel Sambuc entry = xcalloc(1, sizeof *entry);
210*0a6a1f1dSLionel Sambuc entry->flag = ch;
211*0a6a1f1dSLionel Sambuc RB_INSERT(args_tree, &args->tree, entry);
212*0a6a1f1dSLionel Sambuc }
213*0a6a1f1dSLionel Sambuc
214eda6f593SDavid van Moolenbroek if (value != NULL)
215*0a6a1f1dSLionel Sambuc entry->value = xstrdup(value);
216eda6f593SDavid van Moolenbroek }
217eda6f593SDavid van Moolenbroek
218eda6f593SDavid van Moolenbroek /* Get argument value. Will be NULL if it isn't present. */
219eda6f593SDavid van Moolenbroek const char *
args_get(struct args * args,u_char ch)220eda6f593SDavid van Moolenbroek args_get(struct args *args, u_char ch)
221eda6f593SDavid van Moolenbroek {
222*0a6a1f1dSLionel Sambuc struct args_entry *entry;
223*0a6a1f1dSLionel Sambuc
224*0a6a1f1dSLionel Sambuc if ((entry = args_find(args, ch)) == NULL)
225*0a6a1f1dSLionel Sambuc return (NULL);
226*0a6a1f1dSLionel Sambuc return (entry->value);
227eda6f593SDavid van Moolenbroek }
228eda6f593SDavid van Moolenbroek
229eda6f593SDavid van Moolenbroek /* Convert an argument value to a number. */
230eda6f593SDavid van Moolenbroek long long
args_strtonum(struct args * args,u_char ch,long long minval,long long maxval,char ** cause)231*0a6a1f1dSLionel Sambuc args_strtonum(struct args *args, u_char ch, long long minval, long long maxval,
232*0a6a1f1dSLionel Sambuc char **cause)
233eda6f593SDavid van Moolenbroek {
234eda6f593SDavid van Moolenbroek const char *errstr;
235eda6f593SDavid van Moolenbroek long long ll;
236*0a6a1f1dSLionel Sambuc struct args_entry *entry;
237eda6f593SDavid van Moolenbroek
238*0a6a1f1dSLionel Sambuc if ((entry = args_find(args, ch)) == NULL) {
239eda6f593SDavid van Moolenbroek *cause = xstrdup("missing");
240eda6f593SDavid van Moolenbroek return (0);
241eda6f593SDavid van Moolenbroek }
242eda6f593SDavid van Moolenbroek
243*0a6a1f1dSLionel Sambuc ll = strtonum(entry->value, minval, maxval, &errstr);
244eda6f593SDavid van Moolenbroek if (errstr != NULL) {
245eda6f593SDavid van Moolenbroek *cause = xstrdup(errstr);
246eda6f593SDavid van Moolenbroek return (0);
247eda6f593SDavid van Moolenbroek }
248eda6f593SDavid van Moolenbroek
249eda6f593SDavid van Moolenbroek *cause = NULL;
250eda6f593SDavid van Moolenbroek return (ll);
251eda6f593SDavid van Moolenbroek }
252