static void
test_regSubkeys(void)
{
RegStatus status;
char* value;
status = reg_putString("/subkey/string_key", "string_value");
if (status) {
log_error("test_subkeys(): Couldn't put string");
}
else {
CU_ASSERT_EQUAL(status, 0);
}
status = reg_putString("/subkey/string_key", "string value 2");
if (status) {
log_error("test_subkeys(): Couldn't replace string");
}
else {
CU_ASSERT_EQUAL(status, 0);
}
status = reg_getString("/subkey/string_key", &value);
if (status) {
log_error("test_regString(): Couldn't get string");
}
else {
CU_ASSERT_EQUAL(status, 0);
CU_ASSERT_STRING_EQUAL(value, "string value 2");
free(value);
}
status = reg_putString("/subkey/string_key2", "string_value2");
if (status) {
log_error("test_subkeys(): Couldn't put second string");
}
else {
CU_ASSERT_EQUAL(status, 0);
}
status = reg_getString("/subkey/nonexistant_key", &value);
if (status && ENOENT != status) {
log_error("test_regString(): Couldn't verify non-existant value");
}
else {
CU_ASSERT_EQUAL(status, ENOENT);
}
}
static void
test_keyWithSpace(void)
{
RegStatus status;
char* value;
status = reg_putString("/foo key", "foo value 0");
CU_ASSERT_EQUAL(status, EINVAL);
log_clear();
}
static void
test_regString(void)
{
RegStatus status;
char* value;
status = reg_putString("/foo_key", "foo value 0");
if (status) {
log_error("test_regString(): Couldn't put string");
}
else {
CU_ASSERT_EQUAL(status, 0);
}
status = reg_putString("/foo_key", "foo value");
if (status) {
log_error("test_regString(): Couldn't replace string");
}
else {
CU_ASSERT_EQUAL(status, 0);
}
status = reg_getString("/foo_key", &value);
if (status) {
log_error("test_regString(): Couldn't get string");
}
else {
CU_ASSERT_EQUAL(status, 0);
CU_ASSERT_STRING_EQUAL(value, "foo value");
free(value);
}
status = reg_getString("/bar_key", &value);
if (status && ENOENT != status) {
log_error("test_regString(): Couldn't add second string");
}
else {
CU_ASSERT_EQUAL(status, ENOENT);
}
}
/*
* Returns:
* 0 Success
* 1 Incorrect usage (i.e., command-line syntax error).
* 2 No such parameter or node. Error message written.
* 3 System error. Error message written.
*/
int main(
int argc,
char* argv[])
{
int status;
const char* const progname = basename(argv[0]);
(void)log_init(progname);
if ((status = sb_new(&_valuePath, 80))) {
log_error("Couldn't initialize utility");
status = SYSTEM_ERROR;
}
else {
enum {
UNKNOWN,
CREATE,
PRINT,
PUT_BOOL,
PUT_STRING,
PUT_UINT,
PUT_SIGNATURE,
PUT_TIME,
RESET,
REMOVE
} action = UNKNOWN;
const char* string;
signaturet signature;
timestampt timestamp;
unsigned long uint;
int boolean;
int ch;
int quiet = 0;
opterr = 0; /* supress getopt(3) error messages */
while (0 == status && (ch = getopt(argc, argv, ":b:cd:h:qRrs:t:u:vx"))
!= -1) {
switch (ch) {
case 'b': {
if (strcasecmp(optarg, "TRUE") == 0) {
boolean = 1;
}
else if (strcasecmp(optarg, "FALSE") == 0) {
boolean = 0;
}
else {
log_add("Not a boolean value: \"%s\"", optarg);
status = COMMAND_SYNTAX;
}
if (status == 0) {
if (CREATE == action) {
log_error("Create option ignored");
}
action = PUT_BOOL;
}
break;
}
case 'c': {
if (UNKNOWN != action) {
log_add("Can't mix create action with other actions");
status = COMMAND_SYNTAX;
}
else {
action = CREATE;
}
break;
}
case 'd': {
if ((status = reg_setDirectory(optarg)))
status = SYSTEM_ERROR;
break;
}
case 'h': {
status = sigParse(optarg, &signature);
if (0 > status || 0 != optarg[status]) {
log_add("Not a signature: \"%s\"", optarg);
status = COMMAND_SYNTAX;
}
else {
if (CREATE == action) {
log_info("Create action ignored");
}
action = PUT_SIGNATURE;
status = 0;
}
break;
}
case 'q': {
quiet = 1;
break;
}
case 'R': {
if (UNKNOWN != action) {
log_add("Can't mix reset action with other actions");
status = COMMAND_SYNTAX;
}
else {
action = RESET;
}
break;
}
case 'r': {
if (UNKNOWN != action) {
log_add("Can't mix remove action with other actions");
status = COMMAND_SYNTAX;
}
else {
action = REMOVE;
}
break;
}
case 's': {
if (CREATE == action) {
log_info("Create action ignored");
}
string = optarg;
action = PUT_STRING;
break;
}
case 't': {
status = tsParse(optarg, ×tamp);
if (0 > status || 0 != optarg[status]) {
log_add("Not a timestamp: \"%s\"", optarg);
status = COMMAND_SYNTAX;
}
else {
if (CREATE == action) {
log_info("Create action ignored");
}
action = PUT_TIME;
status = 0;
}
break;
}
case 'u': {
char* end;
errno = 0;
uint = strtoul(optarg, &end, 0);
if (0 != *end || (0 == uint && 0 != errno)) {
log_add("Not an unsigned integer: \"%s\"", optarg);
status = COMMAND_SYNTAX;
}
else {
if (CREATE == action) {
log_info("Create option ignored");
}
action = PUT_UINT;
}
break;
}
case 'v': {
if (!log_is_enabled_info)
(void)log_set_level(LOG_LEVEL_INFO);
break;
}
case 'x': {
(void)log_set_level(LOG_LEVEL_DEBUG);
break;
}
case ':': {
log_add("Option \"-%c\" requires an operand", optopt);
status = COMMAND_SYNTAX;
break;
}
default:
log_add("Unknown option: \"%c\"", optopt);
status = COMMAND_SYNTAX;
/* no break */
}
} /* options loop */
if (status) {
log_flush_error();
if (COMMAND_SYNTAX == status)
usage(progname);
}
else {
const int argCount = argc - optind;
if (UNKNOWN == action)
action = PRINT;
switch (action) {
case CREATE: {
if (0 < argCount) {
log_error("Too many arguments");
usage(progname);
status = COMMAND_SYNTAX;
}
else {
status = createRegistry();
}
break;
}
case RESET: {
if (0 < argCount) {
log_error("Too many arguments");
usage(progname);
status = COMMAND_SYNTAX;
}
else {
status = resetRegistry();
}
break;
}
case REMOVE: {
if (0 == argCount) {
log_error(
"Removal action requires absolute pathname(s)");
usage(progname);
status = COMMAND_SYNTAX;
}
else {
log_debug("Removing registry");
status = actUponPathList(argv + optind, deletePath,
quiet);
}
break;
}
case PRINT: {
log_debug("Printing registry");
status = (0 == argCount)
? printPath("/", quiet)
: actUponPathList(argv + optind, printPath, quiet);
break;
}
default: {
/*
* Must be some kind of "put".
*/
if (0 == argCount) {
log_error("Put action requires value pathname");
usage(progname);
status = COMMAND_SYNTAX;
}
else {
switch (action) {
case PUT_BOOL:
status = reg_putBool(argv[optind], boolean);
break;
case PUT_UINT:
status = reg_putUint(argv[optind], uint);
break;
case PUT_STRING:
status = reg_putString(argv[optind], string);
break;
case PUT_TIME:
status = reg_putTime(argv[optind], ×tamp);
break;
case PUT_SIGNATURE:
status = reg_putSignature(argv[optind], signature);
break;
default:
abort();
}
if (status) {
log_flush_error();
status = SYSTEM_ERROR;
}
}
} /* put switch */
/* no break */
} /* "action" switch */
} /* decoded options */
sb_free(_valuePath);
} /* "_valuePath" allocated */
return status;
}
static void
test_regDelete(void)
{
RegStatus status;
char* string;
int value;
status = reg_putString("/string_key", "string value");
if (status) {
log_error("test_regDelete(): Couldn't put string");
}
else {
CU_ASSERT_EQUAL(status, 0);
}
status = reg_getString("/string_key", &string);
if (status) {
log_error("test_regDelete(): Couldn't get string");
}
else {
CU_ASSERT_EQUAL(status, 0);
CU_ASSERT_STRING_EQUAL(string, "string value");
free(string);
}
status = reg_deleteValue("/string_key");
if (status) {
log_error("test_regDelete(): Couldn't delete string");
}
else {
CU_ASSERT_EQUAL(status, 0);
}
status = reg_getString("/string_key", &string);
if (status && ENOENT != status) {
log_error("test_regDelete(): Couldn't verify string deletion");
}
else {
CU_ASSERT_EQUAL(status, ENOENT);
}
status = reg_putUint("/int_key", -1);
if (status) {
log_error("test_regDelete(): Couldn't put int");
}
else {
CU_ASSERT_EQUAL(status, 0);
}
status = reg_getUint("/int_key", &value);
if (status) {
log_error("test_regDelete(): Couldn't get int");
}
else {
CU_ASSERT_EQUAL(status, 0);
CU_ASSERT_EQUAL(value, -1);
}
status = reg_deleteValue("/int_key");
if (status) {
log_error("test_regDelete(): Couldn't delete int");
}
else {
CU_ASSERT_EQUAL(status, 0);
}
status = reg_getUint("/int_key", &value);
if (status && ENOENT != status) {
log_error("test_regDelete(): Couldn't verify int deletion");
}
else {
CU_ASSERT_EQUAL(status, ENOENT);
}
status = reg_deleteValue("/nosuch_key");
if (status && ENOENT != status) {
log_error("test_regDelete(): Couldn't verify no-such-value deletion");
}
else {
CU_ASSERT_EQUAL(status, ENOENT);
}
}
static void
test_regBool(void)
{
RegStatus status;
int value;
status = reg_putBool("/fooBool_key", 0);
if (status) {
log_error("test_regBool(): Couldn't add boolean");
}
CU_ASSERT_EQUAL(status, 0);
status = reg_getBool("/fooBool_key", &value);
if (status) {
log_error("test_regBool(): Couldn't get boolean");
}
CU_ASSERT_EQUAL(value, 0);
status = reg_putString("/fooBool_key", "FALSE");
if (status) {
log_error("test_regBool(): Couldn't add boolean");
}
CU_ASSERT_EQUAL(status, 0);
status = reg_getBool("/fooBool_key", &value);
if (status) {
log_error("test_regBool(): Couldn't get boolean");
}
CU_ASSERT_EQUAL(value, 0);
status = reg_putBool("/fooBool_key", 1);
if (status) {
log_error("test_regBool(): Couldn't replace boolean");
}
CU_ASSERT_EQUAL(status, 0);
status = reg_getBool("/fooBool_key", &value);
if (status) {
log_error("test_regBool(): Couldn't get boolean");
}
CU_ASSERT_EQUAL(value, 1);
status = reg_putString("/fooBool_key", "TRUE");
if (status) {
log_error("test_regBool(): Couldn't replace boolean");
}
CU_ASSERT_EQUAL(status, 0);
status = reg_getBool("/fooBool_key", &value);
if (status) {
log_error("test_regBool(): Couldn't get boolean");
}
CU_ASSERT_EQUAL(value, 1);
status = reg_getBool("/barBool_key", &value);
if (status && ENOENT != status) {
log_error("test_regBool(): Bad status getting non-existent boolean: %d",
status);
}
CU_ASSERT_EQUAL(status, ENOENT);
}
