circlecloud/tmux
1
0
Fork 0
mirror of https://github.com/tmux/tmux.git synced 2024-12-12 17:38:48 +00:00
Code Issues Projects Releases Wiki Activity

Free value properly.

Browse Source
This commit is contained in:
nicm 2021-08-21 20:57:52 +00:00
parent 069f5925af
commit 0084cbef5a
1 changed files with 93 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

143
arguments.c
View File

@ -39,9 +39,9 @@ struct args_entry {
}; };
struct args { struct args {
struct args_tree tree; struct args_tree tree;
int argc; u_int count;
char **argv; struct args_value *values;
}; };
static struct args_entry *args_find(struct args *, u_char); static struct args_entry *args_find(struct args *, u_char);
@ -66,17 +66,37 @@ args_find(struct args *args, u_char flag)
return (RB_FIND(args_tree, &args->tree, &entry)); return (RB_FIND(args_tree, &args->tree, &entry));
} }
/* Copy value. */
static void
args_copy_value(struct args_value *to, struct args_value *from)
{
to->type = from->type;
switch (from->type) {
case ARGS_NONE:
break;
case ARGS_COMMANDS:
to->cmdlist = from->cmdlist;
to->cmdlist->references++;
break;
case ARGS_STRING:
to->string = xstrdup(from->string);
break;
}
}
/* Get value as string. */ /* Get value as string. */
static char * static const char *
args_value_as_string(struct args_value *value) args_value_as_string(struct args_value *value)
{ {
switch (value->type) { switch (value->type) {
case ARGS_NONE: case ARGS_NONE:
return (xstrdup("")); return ("");
case ARGS_COMMANDS: case ARGS_COMMANDS:
return (cmd_list_print(value->cmdlist, 0)); if (value->cached == NULL)
value->cached = cmd_list_print(value->cmdlist, 0);
return (value->cached);
case ARGS_STRING: case ARGS_STRING:
return (xstrdup(value->string)); return (value->string);
} }
} }
@ -98,10 +118,9 @@ args_parse(const struct args_parse *parse, struct args_value *values,
{ {
struct args *args; struct args *args;
u_int i; u_int i;
struct args_value *value; struct args_value *value, *new;
u_char flag, argument; u_char flag, argument;
const char *found, *string; const char *found, *string, *s;
char *s;
if (count == 0) if (count == 0)
return (args_create()); return (args_create());
@ -109,11 +128,6 @@ args_parse(const struct args_parse *parse, struct args_value *values,
args = args_create(); args = args_create();
for (i = 1; i < count; /* nothing */) { for (i = 1; i < count; /* nothing */) {
value = &values[i]; value = &values[i];
s = args_value_as_string(value);
log_debug("%s: %u = %s", __func__, i, s);
free(s);
if (value->type != ARGS_STRING) if (value->type != ARGS_STRING)
break; break;
@ -143,18 +157,20 @@ args_parse(const struct args_parse *parse, struct args_value *values,
args_set(args, flag, NULL); args_set(args, flag, NULL);
continue; continue;
} }
if (*string != '\0') new = xcalloc(1, sizeof *value);
s = xstrdup(string); if (*string != '\0') {
else { new->type = ARGS_STRING;
new->string = xstrdup(string);
} else {
if (i == count) { if (i == count) {
args_free(args); args_free(args);
return (NULL); return (NULL);
} }
s = args_value_as_string(&values[i++]); args_copy_value(new, &values[i++]);
} }
s = args_value_as_string(new);
log_debug("%s: add -%c = %s", __func__, flag, s); log_debug("%s: add -%c = %s", __func__, flag, s);
args_set(args, flag, s); args_set(args, flag, new);
free(s);
break; break;
} }
} }
@ -165,13 +181,15 @@ args_parse(const struct args_parse *parse, struct args_value *values,
s = args_value_as_string(value); s = args_value_as_string(value);
log_debug("%s: %u = %s", __func__, i, s); log_debug("%s: %u = %s", __func__, i, s);
cmd_append_argv(&args->argc, &args->argv, s);
free(s); args->values = xrecallocarray(args->values,
args->count, args->count + 1, sizeof *args->values);
args_copy_value(&args->values[args->count++], value);
} }
} }
if ((parse->lower != -1 && args->argc < parse->lower) || if ((parse->lower != -1 && args->count < (u_int)parse->lower) ||
(parse->upper != -1 && args->argc > parse->upper)) { (parse->upper != -1 && args->count > (u_int)parse->upper)) {
args_free(args); args_free(args);
return (NULL); return (NULL);
} }
@ -192,6 +210,7 @@ args_free_value(struct args_value *value)
cmd_list_free(value->cmdlist); cmd_list_free(value->cmdlist);
break; break;
} }
free(value->cached);
} }
/* Free an arguments set. */ /* Free an arguments set. */
@ -202,14 +221,17 @@ args_free(struct args *args)
struct args_entry *entry1; struct args_entry *entry1;
struct args_value *value; struct args_value *value;
struct args_value *value1; struct args_value *value1;
u_int i;
cmd_free_argv(args->argc, args->argv); for (i = 0; i < args->count; i++)
args_free_value(&args->values[i]);
free(args->values);
RB_FOREACH_SAFE(entry, args_tree, &args->tree, entry1) { RB_FOREACH_SAFE(entry, args_tree, &args->tree, entry1) {
RB_REMOVE(args_tree, &args->tree, entry); RB_REMOVE(args_tree, &args->tree, entry);
TAILQ_FOREACH_SAFE(value, &entry->values, entry, value1) { TAILQ_FOREACH_SAFE(value, &entry->values, entry, value1) {
TAILQ_REMOVE(&entry->values, value, entry); TAILQ_REMOVE(&entry->values, value, entry);
free(value->string); args_free_value(value);
free(value); free(value);
} }
free(entry); free(entry);
@ -222,8 +244,16 @@ args_free(struct args *args)
void void
args_vector(struct args *args, int *argc, char ***argv) args_vector(struct args *args, int *argc, char ***argv)
{ {
*argc = args->argc; struct args_value *value;
*argv = cmd_copy_argv(args->argc, args->argv); u_int i;
*argc = 0;
*argv = NULL;
for (i = 0; i < args->count; i++) {
value = &args->values[i];
cmd_append_argv(argc, argv, args_value_as_string(value));
}
} }
/* Add to string. */ /* Add to string. */
@ -245,18 +275,28 @@ args_print_add(char **buf, size_t *len, const char *fmt, ...)
free(s); free(s);
} }
/* Add argument to string. */ /* Add value to string. */
static void static void
args_print_add_argument(char **buf, size_t *len, const char *argument) args_print_add_value(char **buf, size_t *len, struct args_value *value)
{ {
char *escaped; char *expanded = NULL;
if (**buf != '\0') if (**buf != '\0')
args_print_add(buf, len, " "); args_print_add(buf, len, " ");
escaped = args_escape(argument); switch (value->type) {
args_print_add(buf, len, "%s", escaped); case ARGS_NONE:
free(escaped); break;
case ARGS_COMMANDS:
expanded = cmd_list_print(value->cmdlist, 0);
args_print_add(buf, len, "{ %s }", expanded);
break;
case ARGS_STRING:
expanded = args_escape(value->string);
args_print_add(buf, len, "%s", expanded);
break;
}
free(expanded);
} }
/* Print a set of arguments. */ /* Print a set of arguments. */
@ -265,8 +305,7 @@ args_print(struct args *args)
{ {
size_t len; size_t len;
char *buf; char *buf;
int i; u_int i, j;
u_int j;
struct args_entry *entry; struct args_entry *entry;
struct args_value *value; struct args_value *value;
@ -291,13 +330,13 @@ args_print(struct args *args)
args_print_add(&buf, &len, " -%c", entry->flag); args_print_add(&buf, &len, " -%c", entry->flag);
else else
args_print_add(&buf, &len, "-%c", entry->flag); args_print_add(&buf, &len, "-%c", entry->flag);
args_print_add_argument(&buf, &len, value->string); args_print_add_value(&buf, &len, value);
} }
} }
/* And finally the argument vector. */ /* And finally the argument vector. */
for (i = 0; i < args->argc; i++) for (i = 0; i < args->count; i++)
args_print_add_argument(&buf, &len, args->argv[i]); args_print_add_value(&buf, &len, &args->values[i]);
return (buf); return (buf);
} }
@ -363,10 +402,9 @@ args_has(struct args *args, u_char flag)
/* Set argument value in the arguments tree. */ /* Set argument value in the arguments tree. */
void void
args_set(struct args *args, u_char flag, const char *s) args_set(struct args *args, u_char flag, struct args_value *value)
{ {
struct args_entry *entry; struct args_entry *entry;
struct args_value *value;
entry = args_find(args, flag); entry = args_find(args, flag);
if (entry == NULL) { if (entry == NULL) {
@ -377,12 +415,8 @@ args_set(struct args *args, u_char flag, const char *s)
RB_INSERT(args_tree, &args->tree, entry); RB_INSERT(args_tree, &args->tree, entry);
} else } else
entry->count++; entry->count++;
if (value != NULL && value->type != ARGS_NONE)
if (s != NULL) {
value = xcalloc(1, sizeof *value);
value->string = xstrdup(s);
TAILQ_INSERT_TAIL(&entry->values, value, entry); TAILQ_INSERT_TAIL(&entry->values, value, entry);
}
} }
/* Get argument value. Will be NULL if it isn't present. */ /* Get argument value. Will be NULL if it isn't present. */
@ -422,16 +456,25 @@ args_next(struct args_entry **entry)
u_int u_int
args_count(struct args *args) args_count(struct args *args)
{ {
return (args->argc); return (args->count);
}
/* Get argument value. */
struct args_value *
args_value(struct args *args, u_int idx)
{
if (idx >= args->count)
return (NULL);
return (&args->values[idx]);
} }
/* Return argument as string. */ /* Return argument as string. */
const char * const char *
args_string(struct args *args, u_int idx) args_string(struct args *args, u_int idx)
{ {
if (idx >= (u_int)args->argc) if (idx >= args->count)
return (NULL); return (NULL);
return (args->argv[idx]); return (args_value_as_string(&args->values[idx]));
} }
/* Get first value in argument. */ /* Get first value in argument. */