static int revwalk_parseopts(git_repository *repo, git_revwalk *walk, int nopts, char **opts)
{
int hide, i, error;
unsigned int sorting = GIT_SORT_NONE;
hide = 0;
for (i = 0; i < nopts; i++) {
if (!strcmp(opts[i], "--topo-order")) {
sorting = GIT_SORT_TOPOLOGICAL | (sorting & GIT_SORT_REVERSE);
git_revwalk_sorting(walk, sorting);
} else if (!strcmp(opts[i], "--date-order")) {
sorting = GIT_SORT_TIME | (sorting & GIT_SORT_REVERSE);
git_revwalk_sorting(walk, sorting);
} else if (!strcmp(opts[i], "--reverse")) {
sorting = (sorting & ~GIT_SORT_REVERSE)
| ((sorting & GIT_SORT_REVERSE) ? 0 : GIT_SORT_REVERSE);
git_revwalk_sorting(walk, sorting);
} else if (!strcmp(opts[i], "--not")) {
hide = !hide;
} else if (opts[i][0] == '^') {
if ((error = push_spec(repo, walk, opts[i] + 1, !hide)))
return error;
} else if (strstr(opts[i], "..")) {
if ((error = push_range(repo, walk, opts[i], hide)))
return error;
} else {
if ((error = push_spec(repo, walk, opts[i], hide)))
return error;
}
}
return 0;
}
static bool run_test(
Queue * queue)
{
TEST(void *,
"%p",
((void *)queue),
!=,
NULL);
if (!test_empty(queue))
return false;
if (!push_range(queue, 0, 0, 4000))
return false;
if (!pop_range(queue, 4000, 0, 1000))
return false;
if (!push_range(queue, 3000, 4000, 6000))
return false;
if (!pop_range(queue, 5000, 1000, 6000))
return false;
if (!test_empty(queue))
return false;
if (!push_range(queue, 0, 0, 1000))
return false;
if (!pop_range(queue, 1000, 0, 999))
return false;
if (!push_range(queue, 1, 1000, 6000))
return false;
if (!pop_range(queue, 5001, 999, 6000))
return false;
if (!test_empty(queue))
return false;
if (!push_range(queue, 0, 0, 2000))
return false;
if (!pop_range(queue, 2000, 0, 1998))
return false;
if (!push_range(queue, 2, 2000, 10000))
return false;
if (!pop_range(queue, 8002, 1998, 10000))
return false;
if (!test_empty(queue))
return false;
Queue_free(queue);
return true;
}
int main(int argc, char *argv[]) {
unsigned i, narg;
unsigned sym_stdout = 0;
if (argc < 2) {
fprintf(stderr, "Usage: %s <random-seed> <argument-types>\n", argv[0]);
fprintf(stderr, " If <random-seed> is 0, time(NULL)*getpid() is used as a seed\n");
fprintf(stderr, " <argument-types> are the ones accepted by KLEE: --sym-args, --sym-files etc.\n");
fprintf(stderr, " Ex: %s 100 --sym-args 0 2 2 --sym-files 1 8\n", argv[0]);
exit(1);
}
unsigned seed = atoi(argv[1]);
if (seed)
srandom(seed);
else srandom(time(NULL) * getpid());
KTest b;
b.numArgs = argc;
b.args = argv;
b.symArgvs = 0;
b.symArgvLen = 0;
b.numObjects = 0;
b.objects = (KTestObject *)malloc(MAX * sizeof *b.objects);
for(i = 2; i < (unsigned)argc; i++) {
if(strcmp(argv[i], "--sym-args") == 0) {
unsigned lb = getint(argv[++i]);
unsigned ub = getint(argv[++i]);
unsigned nbytes = getint(argv[++i]);
narg = random() % (ub - lb) + lb;
push_range(&b, "range", narg);
while(narg-- > 0) {
unsigned x = random() % (nbytes + 1);
// A little different than how klee does it but more natural
// for random.
static int total_args;
char arg[1024];
sprintf(arg, "arg%d", total_args++);
push_obj(&b, arg, x, nbytes+1);
}
} else if(strcmp(argv[i], "--sym-stdout") == 0) {
if(!sym_stdout) {
sym_stdout = 1;
push_obj(&b, "stdout", 1024, 1024);
push_obj(&b, "stdout-stat", sizeof(struct stat64),
sizeof(struct stat64));
}
} else if(strcmp(argv[i], "--sym-files") == 0) {
unsigned nfiles = getint(argv[++i]);
unsigned nbytes = getint(argv[++i]);
while(nfiles-- > 0) {
push_obj(&b, "file", nbytes, nbytes);
push_obj(&b, "file-stat", sizeof(struct stat64), sizeof(struct stat64));
}
push_obj(&b, "stdin", nbytes, nbytes);
push_obj(&b, "stdin-stat", sizeof(struct stat64), sizeof(struct stat64));
} else {
fprintf(stderr, "unexpected option <%s>\n", argv[i]);
assert(0);
}
}
if (!kTest_toFile(&b, "file.bout"))
assert(0);
return 0;
}
int main(int argc, char *argv[]) {
unsigned i, argv_copy_idx;
unsigned file_counter = 0;
char *stdout_content_filename = NULL;
char *stdin_content_filename = NULL;
char *content_filenames_list[1024];
char **argv_copy;
if (argc < 2)
print_usage_and_exit(argv[0]);
KTest b;
b.symArgvs = 0;
b.symArgvLen = 0;
b.numObjects = 0;
b.objects = (KTestObject *)malloc(MAX * sizeof *b.objects);
if ((argv_copy = (char **)malloc(sizeof(char *) * argc * 2)) == NULL) {
fprintf(stderr, "Could not allocate more memory\n");
return 1;
}
argv_copy[0] = (char *)malloc(strlen(argv[0]) + 1);
strcpy(argv_copy[0], argv[0]);
argv_copy_idx = 1;
for (i = 1; i < (unsigned)argc; i++) {
if (strcmp(argv[i], "--sym-stdout") == 0 ||
strcmp(argv[i], "-sym-stdout") == 0) {
if (++i == (unsigned)argc || argv[i][0] == '-')
print_usage_and_exit(argv[0]);
if (stdout_content_filename)
print_usage_and_exit(argv[0]);
stdout_content_filename = argv[i];
} else if (strcmp(argv[i], "--sym-stdin") == 0 ||
strcmp(argv[i], "-sym-stdin") == 0) {
if (++i == (unsigned)argc || argv[i][0] == '-')
print_usage_and_exit(argv[0]);
if (stdin_content_filename)
print_usage_and_exit(argv[0]);
stdin_content_filename = argv[i];
} else if (strcmp(argv[i], "--sym-file") == 0 ||
strcmp(argv[i], "-sym-file") == 0) {
if (++i == (unsigned)argc || argv[i][0] == '-')
print_usage_and_exit(argv[0]);
content_filenames_list[file_counter++] = argv[i];
} else {
long nbytes = strlen(argv[i]) + 1;
static int total_args = 0;
char arg[1024];
sprintf(arg, "arg%d", total_args++);
push_obj(&b, (const char *)arg, nbytes, (unsigned char *)argv[i]);
char *buf1 = (char *)malloc(1024);
char *buf2 = (char *)malloc(1024);
strcpy(buf1, "-sym-arg");
sprintf(buf2, "%ld", nbytes - 1);
argv_copy[argv_copy_idx++] = buf1;
argv_copy[argv_copy_idx++] = buf2;
}
}
if (file_counter > 0) {
FILE *fp;
struct stat64 file_stat;
char *content_filename = content_filenames_list[file_counter - 1];
if ((fp = fopen(content_filename, "r")) == NULL ||
stat64(content_filename, &file_stat) < 0) {
fprintf(stderr, "Failure opening %s\n", content_filename);
print_usage_and_exit(argv[0]);
}
long nbytes = file_stat.st_size;
char filename[7] = "A-data";
char statname[12] = "A-data-stat";
unsigned char *file_content, *fptr;
if ((file_content = (unsigned char *)malloc(nbytes)) == NULL) {
fprintf(stderr, "Memory allocation failure\n");
exit(1);
}
int read_char;
fptr = file_content;
while ((read_char = fgetc(fp)) != EOF) {
*fptr = (unsigned char)read_char;
fptr++;
}
push_obj(&b, filename, nbytes, file_content);
push_obj(&b, statname, sizeof(struct stat64), (unsigned char *)&file_stat);
free(file_content);
char *buf1 = (char *)malloc(1024);
char *buf2 = (char *)malloc(1024);
char *buf3 = (char *)malloc(1024);
sprintf(buf1, "-sym-files");
sprintf(buf2, "1");
sprintf(buf3, "%ld", nbytes);
argv_copy[argv_copy_idx++] = buf1;
argv_copy[argv_copy_idx++] = buf2;
argv_copy[argv_copy_idx++] = buf3;
}
if (stdin_content_filename) {
FILE *fp;
struct stat64 file_stat;
char filename[6] = "stdin";
char statname[11] = "stdin-stat";
if ((fp = fopen(stdin_content_filename, "r")) == NULL ||
stat64(stdin_content_filename, &file_stat) < 0) {
fprintf(stderr, "Failure opening %s\n", stdin_content_filename);
print_usage_and_exit(argv[0]);
}
unsigned char *file_content, *fptr;
if ((file_content = (unsigned char *)malloc(file_stat.st_size)) == NULL) {
fprintf(stderr, "Memory allocation failure\n");
exit(1);
}
int read_char;
fptr = file_content;
while ((read_char = fgetc(fp)) != EOF) {
*fptr = (unsigned char)read_char;
fptr++;
}
push_obj(&b, filename, file_stat.st_size, file_content);
push_obj(&b, statname, sizeof(struct stat64), (unsigned char *)&file_stat);
free(file_content);
char *buf1 = (char *)malloc(1024);
char *buf2 = (char *)malloc(1024);
sprintf(buf1, "-sym-stdin");
sprintf(buf2, "%ld", file_stat.st_size);
argv_copy[argv_copy_idx++] = buf1;
argv_copy[argv_copy_idx++] = buf2;
}
if (stdout_content_filename) {
FILE *fp;
struct stat64 file_stat;
unsigned char file_content[1024];
char filename[7] = "stdout";
char statname[12] = "stdout-stat";
if ((fp = fopen(stdout_content_filename, "r")) == NULL ||
stat64(stdout_content_filename, &file_stat) < 0) {
fprintf(stderr, "Failure opening %s\n", stdout_content_filename);
print_usage_and_exit(argv[0]);
}
int read_char;
for (int i = 0; i < file_stat.st_size && i < 1024; ++i) {
read_char = fgetc(fp);
file_content[i] = (unsigned char)read_char;
}
for (int i = file_stat.st_size; i < 1024; ++i) {
file_content[i] = 0;
}
file_stat.st_size = 1024;
push_obj(&b, filename, 1024, file_content);
push_obj(&b, statname, sizeof(struct stat64), (unsigned char *)&file_stat);
char *buf = (char *)malloc(1024);
sprintf(buf, "-sym-stdout");
argv_copy[argv_copy_idx++] = buf;
}
argv_copy[argv_copy_idx] = 0;
b.numArgs = argv_copy_idx;
b.args = argv_copy;
push_range(&b, "model_version", 1);
if (!kTest_toFile(&b, "file.bout"))
assert(0);
for (int i = 0; i < (int)b.numObjects; ++i) {
free(b.objects[i].name);
free(b.objects[i].bytes);
}
free(b.objects);
for (int i = 0; i < (int)argv_copy_idx; ++i) {
free(argv_copy[i]);
}
free(argv_copy);
return 0;
}