/*
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
* University Research and Technology
* Corporation. All rights reserved.
* Copyright (c) 2004-2005 The University of Tennessee and The University
* of Tennessee Research Foundation. All rights
* reserved.
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "ompi_config.h"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include "support.h"
#include "opal/constants.h"
#include "opal/util/argv.h"
static bool test1(void);
static bool test2(void);
static bool test3(void);
static bool test4(void);
static bool test5(void);
static bool test6(void);
static bool test7(void);
static bool test8(void);
static bool test9(void);
static bool test10(void);
int main(int argc, char* argv[])
{
test_init("opal_argv_t");
if( test1() ) test_success();
else test_failure("test1 argv test failed");
if( test2() ) test_success();
else test_failure("test2 argv test failed");
if( test3() ) test_success();
else test_failure("test3 argv test failed");
if( test4() ) test_success();
else test_failure("test4 argv test failed");
if( test5() ) test_success();
else test_failure("test5 argv test failed");
if( test6() ) test_success();
else test_failure("test6 argv test failed");
if( test7() ) test_success();
else test_failure("test7 argv test failed");
if( test8() ) test_success();
else test_failure("test8 argv test failed");
if (test9()) {
test_success();
} else {
test_failure("test9 argv test failed");
}
if (test10()) {
test_success();
} else {
test_failure("test10 argv test failed");
}
/* All done */
return test_finalize();
}
static bool test1(void)
{
char *a[] = { "argv1", "argv2", "argv3", NULL };
char **argv = NULL;
int i, j, argc = 28;
/* Test basic functionality. Start with a NULL argv and add the
contents of the a array.
Set argc to be an initiall bogus number -- opal_argv_add() should
reset it back to zero after the first iteration.
After adding the a[i], ensure that argv[0 - (i-1)] are the same
as a[0 - (i-1)].
Also ensure that a[i + 1] == NULL and that argc is the proper
value. */
for (i = 0; a[i] != NULL; ++i) {
if (opal_argv_append(&argc, &argv, a[i]) != OPAL_SUCCESS) {
return false;
}
for (j = 0; j <= i; ++j) {
if (strcmp(argv[j], a[j]) != 0) {
return false;
}
}
if (NULL != argv[i + 1]) {
return false;
}
if (argc != i + 1) {
return false;
}
}
opal_argv_free(argv);
return true;
}
static bool test2(void)
{
int i, j;
char **argv = NULL;
int argc;
char *a[] = { "aaa", "bbb", "ccc", NULL };
char *b[4];
/* Similar test to above, but ensure that opal_argv_add is actually
*copying* the string by value, not by reference. So copy the a
array into b, and then opal_argv_add() from b. After that,
scribble in the first character of the b[] string that we just
added, and compare all entries in a to argv -- they should be
identical (even though b is now corrupted). */
for (i = 0; a[i] != NULL; ++i) {
b[i] = strdup(a[i]);
}
b[i] = NULL;
for (i = 0; b[i] != NULL; ++i) {
if (opal_argv_append(&argc, &argv, b[i]) != OPAL_SUCCESS) {
return false;
}
++b[i][0];
for (j = 0; j <= i; ++j) {
if (strcmp(argv[j], a[j]) != 0) {
return false;
}
}
if (NULL != argv[i + 1]) {
return false;
}
if (argc != i + 1) {
return false;
}
}
opal_argv_free(argv);
for (i = 0; b[i] != NULL; ++i) {
free(b[i]);
}
return true;
}
static bool test3(void)
{
int i;
int argc;
char **argv = NULL;
char *a[] = { "aaa", "bbb", "ccc", NULL };
char *b[4];
/* Try to free a null argv -- should be harmless (we'll seg fault if
it's not!) */
opal_argv_free(argv);
/* Now add some stuff and try to free it. We'll seg fault if
anything goes wrong. a is on the stack, so if it mistakenly
tries to free it, we should get a seg fault. */
for (i = 0; a[i] != NULL; ++i) {
if (opal_argv_append(&argc, &argv, a[i]) != OPAL_SUCCESS) {
return false;
}
}
opal_argv_free(argv);
/* Do the same thing but guarantee that the copied array was from
the heap and was freed before we call opal_argv_free(). */
argc = 0;
argv = NULL;
for (i = 0; a[i] != NULL; ++i) {
b[i] = strdup(a[i]);
}
b[i] = NULL;
for (i = 0; b[i] != NULL; ++i) {
if (opal_argv_append(&argc, &argv, b[i]) != OPAL_SUCCESS) {
return false;
}
}
for (i = 0; b[i] != NULL; ++i) {
free(b[i]);
}
opal_argv_free(argv);
/* All done */
return true;
}
static bool test4(void)
{
size_t i, count;
char *a = strdup("the quick brown fox jumped over the lazy dog a_really_long_argument_to_force_a_long_copy_zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz");
char **b;
char *start;
/* split a string into an argv, and compare it against the original
string. Add double spaces into the string; opal_argv_split()
should skip them. */
b = opal_argv_split(a, ' ');
for (count = i = 1; i < strlen(a); ++i) {
if (a[i] != ' ' && a[i - 1] == ' ') {
++count;
}
}
for (i = 0; b[i] != NULL; ++i) {
continue;
}
if (i != count) {
return false;
}
/* now do the same thing and compare each token in b */
for (start = a, count = i = 0; i < strlen(a); ++i) {
if (a[i] == ' ' && a[i - 1] != ' ') {
a[i] = '\0';
if (strcmp(start, b[count]) != 0) {
return false;
}
++count;
a[i] = ' ';
}
if (a[i] == ' ' && a[i + 1] != ' ') {
start = a + i + 1;
}
}
if (strcmp(start, b[count]) != 0) {
return false;
}
/* all done */
opal_argv_free(b);
free(a);
return true;
}
static bool test5(void)
{
char *a[] = { "aaa", "bbb", "ccc", NULL };
return (opal_argv_count(NULL) == 0 && opal_argv_count(a) == 3);
}
static bool test6(void)
{
char *a = "the quick brown fox jumped over the lazy dog a_really_long_argument_to_force_a_long_copy_zzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz";
char **b;
char *c;
/* split the string above and then join it -- the joined version
should be just like the original */
b = opal_argv_split(a, ' ');
c = opal_argv_join(b, ' ');
if (strcmp(a, c) != 0) {
return false;
}
/* All done */
free(c);
opal_argv_free(b);
return true;
}
static bool test7(void)
{
char *a[] = { "a", "b", "c", NULL };
size_t a_len = (1 + 1 + sizeof(char *)) * 3 + sizeof(char **);
/* check a NULL pointer first -- should return 0 */
if (opal_argv_len(NULL) != (size_t) 0) {
return false;
}
/* now check a real string */
/* size should be (sizeof(char **) + (sizeof(char) + sizeof('\0') +
sizeof(char*)) * 3) */
if (opal_argv_len(a) != a_len) {
return false;
}
/* All done */
return true;
}
static bool test8(void)
{
char *a[] = { "aaa", "bbbbbbbb", "cccccccccc", NULL };
int i;
char **b;
/* bozo case */
if (NULL != opal_argv_copy(NULL)) {
return false;
}
/* dup the a array and compare it (array length, contents, etc.) */
b = opal_argv_copy(a);
if (opal_argv_count(a) != opal_argv_count(b)) {
return false;
}
for (i = 0; a[i] != NULL; ++i) {
if (0 != strcmp(a[i], b[i])) {
return false;
}
}
/* All done */
opal_argv_free(b);
return true;
}
static bool test9(void)
{
char **a = NULL;
int argc;
/* bozo cases */
if (OPAL_SUCCESS != opal_argv_delete(NULL, NULL, 0, 0)) {
return false;
}
a = NULL;
argc = 0;
opal_argv_append(&argc, &a, "foo");
if (OPAL_SUCCESS != opal_argv_delete(&argc, &a, 7, 1) ||
1 != opal_argv_count(a)) {
return false;
}
opal_argv_free(a);
a = NULL;
argc = 0;
opal_argv_append(&argc, &a, "foo");
if (OPAL_SUCCESS != opal_argv_delete(&argc, &a, 0, 0) ||
1 != opal_argv_count(a)) {
return false;
}
opal_argv_free(a);
/* now some real tests */
/* delete 1 off the top */
a = NULL;
argc = 0;
opal_argv_append(&argc, &a, "a");
opal_argv_append(&argc, &a, "b");
opal_argv_append(&argc, &a, "c");
opal_argv_append(&argc, &a, "d");
opal_argv_append(&argc, &a, "e");
opal_argv_append(&argc, &a, "f");
if (OPAL_SUCCESS != opal_argv_delete(&argc, &a, 0, 1) ||
5 != opal_argv_count(a) ||
0 != strcmp(a[0], "b") ||
0 != strcmp(a[1], "c") ||
0 != strcmp(a[2], "d") ||
0 != strcmp(a[3], "e") ||
0 != strcmp(a[4], "f")) {
return false;
}
opal_argv_free(a);
/* delete 2 off the top */
a = NULL;
argc = 0;
opal_argv_append(&argc, &a, "a");
opal_argv_append(&argc, &a, "b");
opal_argv_append(&argc, &a, "c");
opal_argv_append(&argc, &a, "d");
opal_argv_append(&argc, &a, "e");
opal_argv_append(&argc, &a, "f");
if (OPAL_SUCCESS != opal_argv_delete(&argc, &a, 0, 2) ||
4 != opal_argv_count(a) ||
0 != strcmp(a[0], "c") ||
0 != strcmp(a[1], "d") ||
0 != strcmp(a[2], "e") ||
0 != strcmp(a[3], "f")) {
return false;
}
opal_argv_free(a);
/* delete 1 in the middle */
a = NULL;
argc = 0;
opal_argv_append(&argc, &a, "a");
opal_argv_append(&argc, &a, "b");
opal_argv_append(&argc, &a, "c");
opal_argv_append(&argc, &a, "d");
opal_argv_append(&argc, &a, "e");
opal_argv_append(&argc, &a, "f");
if (OPAL_SUCCESS != opal_argv_delete(&argc, &a, 1, 1) ||
5 != opal_argv_count(a) ||
0 != strcmp(a[0], "a") ||
0 != strcmp(a[1], "c") ||
0 != strcmp(a[2], "d") ||
0 != strcmp(a[3], "e") ||
0 != strcmp(a[4], "f")) {
return false;
}
opal_argv_free(a);
/* delete 2 in the middle */
a = NULL;
argc = 0;
opal_argv_append(&argc, &a, "a");
opal_argv_append(&argc, &a, "b");
opal_argv_append(&argc, &a, "c");
opal_argv_append(&argc, &a, "d");
opal_argv_append(&argc, &a, "e");
opal_argv_append(&argc, &a, "f");
if (OPAL_SUCCESS != opal_argv_delete(&argc, &a, 1, 2) ||
4 != opal_argv_count(a) ||
0 != strcmp(a[0], "a") ||
0 != strcmp(a[1], "d") ||
0 != strcmp(a[2], "e") ||
0 != strcmp(a[3], "f")) {
return false;
}
opal_argv_free(a);
/* delete everything from the top */
a = NULL;
argc = 0;
opal_argv_append(&argc, &a, "a");
opal_argv_append(&argc, &a, "b");
if (OPAL_SUCCESS != opal_argv_delete(&argc, &a, 0, 99) ||
0 != opal_argv_count(a)) {
return false;
}
opal_argv_free(a);
/* delete everything from the middle */
a = NULL;
argc = 0;
opal_argv_append(&argc, &a, "a");
opal_argv_append(&argc, &a, "b");
if (OPAL_SUCCESS != opal_argv_delete(&argc, &a, 1, 99) ||
1 != opal_argv_count(a) ||
0 != strcmp(a[0], "a")) {
return false;
}
opal_argv_free(a);
/* All done */
return true;
}
static bool test10(void)
{
char **orig;
char **insert;
int o, i;
/* bozo cases */
orig = NULL;
o = 0;
insert = NULL;
i = 0;
opal_argv_append(&i, &insert, "insert a");
if (OPAL_SUCCESS == opal_argv_insert(NULL, 0, insert)) {
return false;
}
opal_argv_append(&o, &orig, "orig a");
if (OPAL_SUCCESS != opal_argv_insert(&orig, 0, NULL)) {
return false;
}
if (OPAL_SUCCESS == opal_argv_insert(&orig, -1, insert)) {
return false;
}
opal_argv_free(orig);
opal_argv_free(insert);
/* append to the end */
orig = NULL;
o = 0;
insert = NULL;
i = 0;
opal_argv_append(&i, &insert, "insert a");
opal_argv_append(&i, &insert, "insert b");
opal_argv_append(&i, &insert, "insert c");
opal_argv_append(&o, &orig, "orig a");
opal_argv_append(&o, &orig, "orig b");
opal_argv_append(&o, &orig, "orig c");
if (OPAL_SUCCESS != opal_argv_insert(&orig, 99, insert) ||
6 != opal_argv_count(orig) ||
0 != strcmp(orig[0], "orig a") ||
0 != strcmp(orig[1], "orig b") ||
0 != strcmp(orig[2], "orig c") ||
0 != strcmp(orig[3], "insert a") ||
0 != strcmp(orig[4], "insert b") ||
0 != strcmp(orig[5], "insert c")) {
return false;
}
opal_argv_free(orig);
opal_argv_free(insert);
/* insert at the beginning */
orig = NULL;
o = 0;
insert = NULL;
i = 0;
opal_argv_append(&i, &insert, "insert a");
opal_argv_append(&i, &insert, "insert b");
opal_argv_append(&i, &insert, "insert c");
opal_argv_append(&o, &orig, "orig a");
opal_argv_append(&o, &orig, "orig b");
opal_argv_append(&o, &orig, "orig c");
if (OPAL_SUCCESS != opal_argv_insert(&orig, 0, insert) ||
6 != opal_argv_count(orig) ||
0 != strcmp(orig[3], "orig a") ||
0 != strcmp(orig[4], "orig b") ||
0 != strcmp(orig[5], "orig c") ||
0 != strcmp(orig[0], "insert a") ||
0 != strcmp(orig[1], "insert b") ||
0 != strcmp(orig[2], "insert c")) {
return false;
}
opal_argv_free(orig);
opal_argv_free(insert);
/* insert in the middle */
orig = NULL;
o = 0;
insert = NULL;
i = 0;
opal_argv_append(&i, &insert, "insert a");
opal_argv_append(&i, &insert, "insert b");
opal_argv_append(&i, &insert, "insert c");
opal_argv_append(&o, &orig, "orig a");
opal_argv_append(&o, &orig, "orig b");
opal_argv_append(&o, &orig, "orig c");
if (OPAL_SUCCESS != opal_argv_insert(&orig, 1, insert) ||
6 != opal_argv_count(orig) ||
0 != strcmp(orig[0], "orig a") ||
0 != strcmp(orig[4], "orig b") ||
0 != strcmp(orig[5], "orig c") ||
0 != strcmp(orig[1], "insert a") ||
0 != strcmp(orig[2], "insert b") ||
0 != strcmp(orig[3], "insert c")) {
return false;
}
opal_argv_free(orig);
opal_argv_free(insert);
/* insert in the middle */
orig = NULL;
o = 0;
insert = NULL;
i = 0;
opal_argv_append(&i, &insert, "insert a");
opal_argv_append(&i, &insert, "insert b");
opal_argv_append(&i, &insert, "insert c");
opal_argv_append(&o, &orig, "orig a");
opal_argv_append(&o, &orig, "orig b");
opal_argv_append(&o, &orig, "orig c");
opal_argv_append(&o, &orig, "orig d");
opal_argv_append(&o, &orig, "orig e");
opal_argv_append(&o, &orig, "orig f");
if (OPAL_SUCCESS != opal_argv_insert(&orig, 1, insert) ||
9 != opal_argv_count(orig) ||
0 != strcmp(orig[0], "orig a") ||
0 != strcmp(orig[4], "orig b") ||
0 != strcmp(orig[5], "orig c") ||
0 != strcmp(orig[6], "orig d") ||
0 != strcmp(orig[7], "orig e") ||
0 != strcmp(orig[8], "orig f") ||
0 != strcmp(orig[1], "insert a") ||
0 != strcmp(orig[2], "insert b") ||
0 != strcmp(orig[3], "insert c")) {
return false;
}
opal_argv_free(orig);
opal_argv_free(insert);
/* All done */
return true;
}