/*
* Copyright (C) 2008-2009 Karel Zak <kzak@redhat.com>
*
* This file may be redistributed under the terms of the
* GNU Lesser General Public License.
*/
#include <string.h>
#include <stdlib.h>
#include <ctype.h>
#include <errno.h>
#include "nls.h"
#include "mountP.h"
/*
* Parses the first option from @optstr. The @optstr pointer is set to begin of
* the next option.
*
* Returns -1 on parse error, 1 at the end of optstr and 0 on success.
*/
static int mnt_optstr_parse_next(char **optstr, char **name, size_t *namesz,
char **value, size_t *valsz)
{
int open_quote = 0;
char *start = NULL, *stop = NULL, *p, *sep = NULL;
char *optstr0;
assert(optstr);
assert(*optstr);
optstr0 = *optstr;
if (name)
*name = NULL;
if (namesz)
*namesz = 0;
if (value)
*value = NULL;
if (valsz)
*valsz = 0;
for (p = optstr0; p && *p; p++) {
if (!start)
start = p; /* begin of the option item */
if (*p == '"')
open_quote ^= 1; /* reverse the status */
if (open_quote)
continue; /* still in quoted block */
if (!sep && *p == '=')
sep = p; /* name and value separator */
if (*p == ',')
stop = p; /* terminate the option item */
else if (*(p + 1) == '\0')
stop = p + 1; /* end of optstr */
if (!start || !stop)
continue;
if (stop <= start)
goto error;
if (name)
*name = start;
if (namesz)
*namesz = sep ? sep - start : stop - start;
*optstr = *stop ? stop + 1 : stop;
if (sep) {
if (value)
*value = sep + 1;
if (valsz)
*valsz = stop - sep - 1;
}
return 0;
}
return 1; /* end of optstr */
error:
DBG(DEBUG_OPTIONS, fprintf(stderr,
"libmount: parse error: \"%s\"\n", optstr0));
return -1;
}
/*
* Locates the first option that match with @name. The @end is set to
* char behind the option (it means ',' or \0).
*
* Returns -1 on parse error, 1 when not found and 0 on success.
*/
static int mnt_optstr_locate_option(char *optstr, const char *name, char **begin,
char **end, char **value, size_t *valsz)
{
char *n;
size_t namesz, nsz;
int rc;
assert(name);
assert(optstr);
namesz = strlen(name);
do {
rc = mnt_optstr_parse_next(&optstr, &n, &nsz, value, valsz);
if (rc)
break;
if (namesz == nsz && strncmp(n, name, nsz) == 0) {
if (begin)
*begin = n;
if (end)
*end = *(optstr - 1) == ',' ?
optstr - 1 : optstr;
return 0;
}
} while(1);
DBG(DEBUG_OPTIONS, fprintf(stderr,
"libmount: can't found '%s' option\n", name));
return rc;
}
/**
* mnt_optstr_next_option:
* @optstr: option string, returns position to next option
* @name: returns option name
* @namesz: returns option name length
* @value: returns option value or NULL
* @valuesz: returns option value length or zero
*
* Parses the first option in @optstr or -1 in case of error.
*
* Returns 0 on success, 1 at the end of @optstr or -1 in case of error.
*/
int mnt_optstr_next_option(char **optstr, char **name, size_t *namesz,
char **value, size_t *valuesz)
{
if (!optstr || !*optstr)
return -1;
return mnt_optstr_parse_next(optstr, name, namesz, value, valuesz);
}
/**
* mnt_optstr_append_option:
* @optstr: option string or NULL
* @name: value name
* @value: value
*
* Returns: reallocated (or newly allocated) @optstr with ,name=value
*/
int mnt_optstr_append_option(char **optstr, const char *name, const char *value)
{
char *p;
size_t sz, vsz, osz, nsz;
if (!name)
return -1;
osz = *optstr ? strlen(*optstr) : 0;
nsz = strlen(name);
vsz = value ? strlen(value) : 0;
sz = osz + nsz + 1; /* 1: '\0' */
if (osz)
sz++; /* ',' options separator */
if (vsz)
sz += vsz + 1; /* 1: '=' */
p = realloc(*optstr, sz);
if (!p)
return -1;
*optstr = p;
if (osz) {
p += osz;
*p++ = ',';
}
memcpy(p, name, nsz);
p += nsz;
if (vsz) {
*p++ = '=';
memcpy(p, value, vsz);
p += vsz;
}
*p = '\0';
return 0;
}
/**
* mnt_optstr_get_option:
* @optstr: string with comma separated list of options
* @name: requested option name
* @value: returns pointer to the begin of the value (e.g. name=VALUE) or NULL
* @valsz: returns size of the value or 0
*
* Returns: 0 on success, 1 when not found the @name or -1 in case of error.
*/
int mnt_optstr_get_option(char *optstr, const char *name,
char **value, size_t *valsz)
{
return mnt_optstr_locate_option(optstr, name, NULL, NULL, value, valsz);
}
/* Removes substring located between @begin and @end from @str
* -- result never starts or ends with comma or contains two commas
* (e.g. ",aaa,bbb" or "aaa,,bbb" or "aaa,")
*/
static void remove_substring(char *str, char *begin, char *end)
{
size_t sz = strlen(end);
if ((begin == str || *(begin - 1) == ',') && *end == ',')
end++;
memmove(begin, end, sz + 1);
if (!*begin && *(begin - 1) == ',')
*(begin - 1) = '\0';
}
/* insert '=substr' to @str on position @pos */
static int insert_substring(char **str, char *pos, const char *substr)
{
char *p;
size_t ssz = strlen(substr); /* substring size */
p = realloc(*str, strlen(*str) + 1 + ssz);
if (!p)
return -1;
*str = p;
memmove(pos + ssz + 1, pos, strlen(pos) + 1);
*pos++ = '=';
memcpy(pos, substr, ssz);
return 0;
}
/**
* mnt_optstr_set_option:
* @optstr: string with comma separated list of options
* @name: requested option
* @value: new value or NULL
*
* Set or unset option @value.
*
* Returns: 0 on success, 1 when not found the @name or -1 in case of error.
*/
int mnt_optstr_set_option(char **optstr, const char *name, const char *value)
{
char *val = NULL, *begin, *end, *nameend;
size_t valsz;
int rc = 1;
if (!optstr)
return -1;
if (*optstr)
rc = mnt_optstr_locate_option(*optstr, name,
&begin, &end, &val, &valsz);
if (rc == -1)
/* parse error */
return -1;
if (rc == 1)
/* not found */
return mnt_optstr_append_option(optstr, name, value);
nameend = begin + strlen(name);
if (value == NULL && val && valsz)
/* remove unwanted "=value" */
remove_substring(*optstr, nameend, end);
else if (value && val == NULL)
/* insert "=value" */
rc = insert_substring(optstr, nameend, value);
else if (value && val && strlen(value) == valsz)
/* simply replace =value */
memcpy(val, value, valsz);
else if (value && val) {
remove_substring(*optstr, nameend, end);
rc = insert_substring(optstr, nameend, value);
}
return 0;
}
/**
* mnt_optstr_remove_option:
* @optstr: string with comma separated list of options
* @name: requested option name
*
* Returns: 0 on success, 1 when not found the @name or -1 in case of error.
*/
int mnt_optstr_remove_option(char **optstr, const char *name)
{
char *begin, *end;
int rc;
rc = mnt_optstr_locate_option(*optstr, name,
&begin, &end, NULL, NULL);
if (rc != 0)
return rc;
remove_substring(*optstr, begin, end);
return 0;
}
#ifdef TEST_PROGRAM
int test_append(struct mtest *ts, int argc, char *argv[])
{
const char *value = NULL, *name;
char *optstr;
if (argc < 3)
goto done;
optstr = strdup(argv[1]);
name = argv[2];
if (argc == 4)
value = argv[3];
if (mnt_optstr_append_option(&optstr, name, value) == 0) {
printf("result: >%s<\n", optstr);
return 0;
}
done:
return -1;
}
int test_set(struct mtest *ts, int argc, char *argv[])
{
const char *value = NULL, *name;
char *optstr;
if (argc < 3)
goto done;
optstr = strdup(argv[1]);
name = argv[2];
if (argc == 4)
value = argv[3];
if (mnt_optstr_set_option(&optstr, name, value) == 0) {
printf("result: >%s<\n", optstr);
return 0;
}
done:
return -1;
}
int test_get(struct mtest *ts, int argc, char *argv[])
{
char *optstr;
const char *name;
char *val = NULL;
size_t sz = 0;
int rc;
if (argc < 2)
goto done;
optstr = argv[1];
name = argv[2];
rc = mnt_optstr_get_option(optstr, name, &val, &sz);
if (rc == 0) {
printf("found; name: %s", name);
if (sz) {
printf(", argument: size=%zd data=", sz);
if (fwrite(val, 1, sz, stdout) != sz)
goto done;
}
printf("\n");
return 0;
} else if (rc == 1)
printf("%s: not found\n", name);
else
printf("parse error: %s\n", optstr);
done:
return -1;
}
int test_remove(struct mtest *ts, int argc, char *argv[])
{
const char *name;
char *optstr;
if (argc < 3)
goto done;
optstr = strdup(argv[1]);
name = argv[2];
if (mnt_optstr_remove_option(&optstr, name) == 0) {
printf("result: >%s<\n", optstr);
return 0;
}
done:
return -1;
}
int main(int argc, char *argv[])
{
struct mtest tss[] = {
{ "--append", test_append, "<optstr> <name> [<value>] append value to optstr" },
{ "--set", test_set, "<optstr> <name> [<value>] (un)set value" },
{ "--get", test_get, "<optstr> <name> search name in optstr" },
{ "--remove", test_remove, "<optstr> <name> remove name in optstr" },
{ NULL }
};
return mnt_run_test(tss, argc, argv);
}
#endif /* TEST_PROGRAM */
