#define vfprintf(f, fmt, ap) saved_vprintf(fmt, ap)
static int saved_vprintf(const char *fmt, va_list ap);
+#define malloc(size) saved_malloc(size)
+static void *saved_malloc(size_t size);
+
#include <ccan/opt/helpers.c>
#include <ccan/opt/opt.c>
#include <ccan/opt/usage.c>
return ret;
}
+#undef malloc
+static void *last_allocation;
+static void *saved_malloc(size_t size)
+{
+ return last_allocation = malloc(size);
+}
+
/* Test helpers. */
int main(int argc, char *argv[])
{
reset_options();
opt_register_noarg("-a",
opt_version_and_exit, "1.2.3", "");
+ /* parse_args allocates argv */
+ free(argv);
+
+ argc = 2;
+ argv = malloc(sizeof(argv[0]) * 3);
+ argv[0] = "thisprog";
+ argv[1] = "-a";
+ argv[2] = NULL;
+
exitval = setjmp(exited);
if (exitval == 0) {
- parse_args(&argc, &argv, "-a", NULL);
+ opt_parse(&argc, argv, save_err_output);
fail("opt_show_version_and_exit returned?");
} else {
ok1(exitval - 1 == 0);
}
ok1(strcmp(output, "1.2.3\n") == 0);
free(output);
+ free(argv);
output = NULL;
}
reset_options();
opt_register_noarg("-a",
opt_usage_and_exit, "[args]", "");
+
+ argc = 2;
+ argv = malloc(sizeof(argv[0]) * 3);
+ argv[0] = "thisprog";
+ argv[1] = "-a";
+ argv[2] = NULL;
+
exitval = setjmp(exited);
if (exitval == 0) {
- parse_args(&argc, &argv, "-a", NULL);
+ opt_parse(&argc, argv, save_err_output);
fail("opt_usage_and_exit returned?");
} else {
ok1(exitval - 1 == 0);
ok1(strstr(output, argv[0]));
ok1(strstr(output, "[-a]"));
free(output);
+ free(argv);
+ /* It exits without freeing usage string. */
+ free(last_allocation);
output = NULL;
}
ok1(!strcmp(output,
"thisprog: --garbage: unrecognized option\n"));
free(output);
+ free(argv);
output = NULL;
}
reset_options();
opt_register_noarg("-a",
opt_usage_and_exit, "[args]", "");
+ argc = 2;
+ argv = malloc(sizeof(argv[0]) * 3);
+ argv[0] = "thisprog";
+ argv[1] = "--garbage";
+ argv[2] = NULL;
exitval = setjmp(exited);
if (exitval == 0) {
- argc = 2;
- argv = malloc(sizeof(argv[0]) * 3);
- argv[0] = "thisprog";
- argv[1] = "--garbage";
- argv[2] = NULL;
opt_parse(&argc, argv, opt_log_stderr_exit);
fail("opt_log_stderr_exit returned?");
} else {
ok1(exitval - 1 == 1);
}
+ free(argv);
ok1(!strcmp(output,
"thisprog: --garbage: unrecognized option\n"));
free(output);
char *err_output = NULL;
-static void save_err_output(const char *fmt, ...)
+void save_err_output(const char *fmt, ...)
{
va_list ap;
char *p;
err_output = p;
}
-/* FIXME: This leaks, BTW. */
+static bool allocated = false;
+
bool parse_args(int *argc, char ***argv, ...)
{
char **a;
(*argc)++;
a = realloc(a, sizeof(*a) * (*argc + 1));
}
+
+ if (allocated)
+ free(*argv);
+
*argv = a;
+ allocated = true;
/* Re-set before parsing. */
optind = 0;