OpenCL GPU miner
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

441 lines
10 KiB

#include "config.h"
#include <stdio.h>
#include <ccan/tap/tap.h>
#include <setjmp.h>
#include <stdlib.h>
#include <limits.h>
#include "utils.h"
/* We don't actually want it to exit... */
static jmp_buf exited;
#define exit(status) longjmp(exited, (status) + 1)
#define printf saved_printf
static int saved_printf(const char *fmt, ...);
#define fprintf saved_fprintf
static int saved_fprintf(FILE *ignored, const char *fmt, ...);
#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>
#include <ccan/opt/parse.c>
static void reset_options(void)
{
free(opt_table);
opt_table = NULL;
opt_count = opt_num_short = opt_num_short_arg = opt_num_long = 0;
}
static char *output = NULL;
static int saved_vprintf(const char *fmt, va_list ap)
{
char *p;
int ret = vasprintf(&p, fmt, ap);
if (output) {
output = realloc(output, strlen(output) + strlen(p) + 1);
strcat(output, p);
free(p);
} else
output = p;
return ret;
}
static int saved_printf(const char *fmt, ...)
{
va_list ap;
int ret;
va_start(ap, fmt);
ret = saved_vprintf(fmt, ap);
va_end(ap);
return ret;
}
static int saved_fprintf(FILE *ignored, const char *fmt, ...)
{
va_list ap;
int ret;
va_start(ap, fmt);
ret = saved_vprintf(fmt, ap);
va_end(ap);
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[])
{
plan_tests(100);
/* opt_set_bool */
{
bool arg = false;
reset_options();
opt_register_noarg("-a", opt_set_bool, &arg, "");
ok1(parse_args(&argc, &argv, "-a", NULL));
ok1(arg);
opt_register_arg("-b", opt_set_bool_arg, NULL, &arg, "");
ok1(parse_args(&argc, &argv, "-b", "no", NULL));
ok1(!arg);
ok1(parse_args(&argc, &argv, "-b", "yes", NULL));
ok1(arg);
ok1(parse_args(&argc, &argv, "-b", "false", NULL));
ok1(!arg);
ok1(parse_args(&argc, &argv, "-b", "true", NULL));
ok1(arg);
ok1(!parse_args(&argc, &argv, "-b", "unknown", NULL));
ok1(arg);
ok1(strstr(err_output, ": -b: Invalid argument 'unknown'"));
}
/* opt_set_invbool */
{
bool arg = true;
reset_options();
opt_register_noarg("-a", opt_set_invbool, &arg, "");
ok1(parse_args(&argc, &argv, "-a", NULL));
ok1(!arg);
opt_register_arg("-b", opt_set_invbool_arg, NULL,
&arg, "");
ok1(parse_args(&argc, &argv, "-b", "no", NULL));
ok1(arg);
ok1(parse_args(&argc, &argv, "-b", "yes", NULL));
ok1(!arg);
ok1(parse_args(&argc, &argv, "-b", "false", NULL));
ok1(arg);
ok1(parse_args(&argc, &argv, "-b", "true", NULL));
ok1(!arg);
ok1(!parse_args(&argc, &argv, "-b", "unknown", NULL));
ok1(!arg);
ok1(strstr(err_output, ": -b: Invalid argument 'unknown'"));
}
/* opt_set_charp */
{
char *arg = (char *)"wrong";
reset_options();
opt_register_arg("-a", opt_set_charp, NULL, &arg, "All");
ok1(parse_args(&argc, &argv, "-a", "string", NULL));
ok1(strcmp(arg, "string") == 0);
}
/* opt_set_intval */
{
int arg = 1000;
reset_options();
opt_register_arg("-a", opt_set_intval, NULL, &arg, "All");
ok1(parse_args(&argc, &argv, "-a", "9999", NULL));
ok1(arg == 9999);
ok1(parse_args(&argc, &argv, "-a", "-9999", NULL));
ok1(arg == -9999);
ok1(parse_args(&argc, &argv, "-a", "0", NULL));
ok1(arg == 0);
ok1(!parse_args(&argc, &argv, "-a", "100crap", NULL));
if (sizeof(int) == 4)
ok1(!parse_args(&argc, &argv, "-a", "4294967296", NULL));
else
fail("Handle other int sizes");
}
/* opt_set_uintval */
{
unsigned int arg = 1000;
reset_options();
opt_register_arg("-a", opt_set_uintval, NULL, &arg, "All");
ok1(parse_args(&argc, &argv, "-a", "9999", NULL));
ok1(arg == 9999);
ok1(!parse_args(&argc, &argv, "-a", "-9999", NULL));
ok1(parse_args(&argc, &argv, "-a", "0", NULL));
ok1(arg == 0);
ok1(!parse_args(&argc, &argv, "-a", "100crap", NULL));
ok1(!parse_args(&argc, &argv, "-a", "4294967296", NULL));
if (ULONG_MAX == UINT_MAX) {
pass("Can't test overflow");
pass("Can't test error message");
} else {
char buf[30];
sprintf(buf, "%lu", ULONG_MAX);
ok1(!parse_args(&argc, &argv, "-a", buf, NULL));
ok1(strstr(err_output, ": -a: value '")
&& strstr(err_output, buf)
&& strstr(err_output, "' does not fit into an integer"));
}
}
/* opt_set_longval */
{
long int arg = 1000;
reset_options();
opt_register_arg("-a", opt_set_longval, NULL, &arg, "All");
ok1(parse_args(&argc, &argv, "-a", "9999", NULL));
ok1(arg == 9999);
ok1(parse_args(&argc, &argv, "-a", "-9999", NULL));
ok1(arg == -9999);
ok1(parse_args(&argc, &argv, "-a", "0", NULL));
ok1(arg == 0);
ok1(!parse_args(&argc, &argv, "-a", "100crap", NULL));
if (sizeof(long) == 4)
ok1(!parse_args(&argc, &argv, "-a", "4294967296", NULL));
else if (sizeof(long)== 8)
ok1(!parse_args(&argc, &argv, "-a", "18446744073709551616", NULL));
else
fail("FIXME: Handle other long sizes");
}
/* opt_set_ulongval */
{
unsigned long int arg = 1000;
reset_options();
opt_register_arg("-a", opt_set_ulongval, NULL, &arg, "All");
ok1(parse_args(&argc, &argv, "-a", "9999", NULL));
ok1(arg == 9999);
ok1(!parse_args(&argc, &argv, "-a", "-9999", NULL));
ok1(parse_args(&argc, &argv, "-a", "0", NULL));
ok1(arg == 0);
ok1(!parse_args(&argc, &argv, "-a", "100crap", NULL));
if (sizeof(long) == 4)
ok1(!parse_args(&argc, &argv, "-a", "4294967296", NULL));
else if (sizeof(long)== 8)
ok1(!parse_args(&argc, &argv, "-a", "18446744073709551616", NULL));
else
fail("FIXME: Handle other long sizes");
}
/* opt_inc_intval */
{
int arg = 1000;
reset_options();
opt_register_noarg("-a", opt_inc_intval, &arg, "");
ok1(parse_args(&argc, &argv, "-a", NULL));
ok1(arg == 1001);
ok1(parse_args(&argc, &argv, "-a", "-a", NULL));
ok1(arg == 1003);
ok1(parse_args(&argc, &argv, "-aa", NULL));
ok1(arg == 1005);
}
/* opt_show_version_and_exit. */
{
int exitval;
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) {
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;
}
/* opt_usage_and_exit. */
{
int exitval;
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) {
opt_parse(&argc, argv, save_err_output);
fail("opt_usage_and_exit returned?");
} else {
ok1(exitval - 1 == 0);
}
ok1(strstr(output, "[args]"));
ok1(strstr(output, argv[0]));
ok1(strstr(output, "[-a]"));
free(output);
free(argv);
/* It exits without freeing usage string. */
free(last_allocation);
output = NULL;
}
/* opt_show_bool */
{
bool b;
char buf[OPT_SHOW_LEN+2] = { 0 };
buf[OPT_SHOW_LEN] = '!';
b = true;
opt_show_bool(buf, &b);
ok1(strcmp(buf, "true") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
b = false;
opt_show_bool(buf, &b);
ok1(strcmp(buf, "false") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
}
/* opt_show_invbool */
{
bool b;
char buf[OPT_SHOW_LEN+2] = { 0 };
buf[OPT_SHOW_LEN] = '!';
b = true;
opt_show_invbool(buf, &b);
ok1(strcmp(buf, "false") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
b = false;
opt_show_invbool(buf, &b);
ok1(strcmp(buf, "true") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
}
/* opt_show_charp */
{
char str[OPT_SHOW_LEN*2], *p;
char buf[OPT_SHOW_LEN+2] = { 0 };
buf[OPT_SHOW_LEN] = '!';
/* Short test. */
p = str;
strcpy(p, "short");
opt_show_charp(buf, &p);
ok1(strcmp(buf, "\"short\"") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
/* Truncate test. */
memset(p, 'x', OPT_SHOW_LEN*2);
p[OPT_SHOW_LEN*2-1] = '\0';
opt_show_charp(buf, &p);
ok1(buf[0] == '"');
ok1(buf[OPT_SHOW_LEN-1] == '"');
ok1(buf[OPT_SHOW_LEN] == '!');
ok1(strspn(buf+1, "x") == OPT_SHOW_LEN-2);
}
/* opt_show_intval */
{
int i;
char buf[OPT_SHOW_LEN+2] = { 0 };
buf[OPT_SHOW_LEN] = '!';
i = -77;
opt_show_intval(buf, &i);
ok1(strcmp(buf, "-77") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
i = 77;
opt_show_intval(buf, &i);
ok1(strcmp(buf, "77") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
}
/* opt_show_uintval */
{
unsigned int ui;
char buf[OPT_SHOW_LEN+2] = { 0 };
buf[OPT_SHOW_LEN] = '!';
ui = 4294967295U;
opt_show_uintval(buf, &ui);
ok1(strcmp(buf, "4294967295") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
}
/* opt_show_longval */
{
long l;
char buf[OPT_SHOW_LEN+2] = { 0 };
buf[OPT_SHOW_LEN] = '!';
l = 1234567890L;
opt_show_longval(buf, &l);
ok1(strcmp(buf, "1234567890") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
}
/* opt_show_ulongval */
{
unsigned long ul;
char buf[OPT_SHOW_LEN+2] = { 0 };
buf[OPT_SHOW_LEN] = '!';
ul = 4294967295UL;
opt_show_ulongval(buf, &ul);
ok1(strcmp(buf, "4294967295") == 0);
ok1(buf[OPT_SHOW_LEN] == '!');
}
/* opt_log_stderr. */
{
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;
ok1(!opt_parse(&argc, argv, opt_log_stderr));
ok1(!strcmp(output,
"thisprog: --garbage: unrecognized option\n"));
free(output);
free(argv);
output = NULL;
}
/* opt_log_stderr_exit. */
{
int exitval;
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) {
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);
output = NULL;
}
return exit_status();
}