X-Git-Url: http://git.archive.openwrt.org/?p=project%2Fuci.git;a=blobdiff_plain;f=util.c;h=12aec9bd3596103741b564198cd969e97207a165;hp=28cc5df2250c4b7b7d51350b45f6e98b293e52c7;hb=5ad59ad412a784c5c478e31d1701ce39568ffd58;hpb=d3740c6347f1e487719a2139023dc942a3b6a8fa diff --git a/util.c b/util.c index 28cc5df..12aec9b 100644 --- a/util.c +++ b/util.c @@ -9,17 +9,30 @@ * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. + * GNU Lesser General Public License for more details. */ /* - * This file contains wrappers to standard functions, which - * throw exceptions upon failure. + * This file contains misc utility functions and wrappers to standard + * functions, which throw exceptions upon failure. */ +#define _GNU_SOURCE +#include +#include +#include #include +#include #include +#include +#include +#include +#include +#include -static void *uci_malloc(struct uci_context *ctx, size_t size) +#include "uci.h" +#include "uci_internal.h" + +__private void *uci_malloc(struct uci_context *ctx, size_t size) { void *ptr; @@ -31,7 +44,7 @@ static void *uci_malloc(struct uci_context *ctx, size_t size) return ptr; } -static void *uci_realloc(struct uci_context *ctx, void *ptr, size_t size) +__private void *uci_realloc(struct uci_context *ctx, void *ptr, size_t size) { ptr = realloc(ptr, size); if (!ptr) @@ -40,7 +53,7 @@ static void *uci_realloc(struct uci_context *ctx, void *ptr, size_t size) return ptr; } -static char *uci_strdup(struct uci_context *ctx, const char *str) +__private char *uci_strdup(struct uci_context *ctx, const char *str) { char *ptr; @@ -51,66 +64,191 @@ static char *uci_strdup(struct uci_context *ctx, const char *str) return ptr; } -static bool uci_validate_name(char *str) +/* + * validate strings for names and types, reject special characters + * for names, only alphanum and _ is allowed (shell compatibility) + * for types, we allow more characters + */ +__private bool uci_validate_str(const char *str, bool name, bool package) { if (!*str) return false; + for (; *str; str++) { + unsigned char c = *str; + + if (isalnum(c) || c == '_') + continue; + + if (c == '-' && package) + continue; + + if (name || (c < 33) || (c > 126)) + return false; + } + return true; +} + +bool uci_validate_text(const char *str) +{ while (*str) { - if (!isalnum(*str) && (*str != '_')) + unsigned char c = *str; + + if (c < 32 && c != '\t' && c != '\n' && c != '\r') return false; + str++; } return true; } -int uci_parse_tuple(struct uci_context *ctx, char *str, char **package, char **section, char **option, char **value) +__private void uci_alloc_parse_context(struct uci_context *ctx) +{ + ctx->pctx = (struct uci_parse_context *) uci_malloc(ctx, sizeof(struct uci_parse_context)); +} + +int uci_parse_ptr(struct uci_context *ctx, struct uci_ptr *ptr, char *str) { char *last = NULL; + char *tmp; UCI_HANDLE_ERR(ctx); - UCI_ASSERT(ctx, str && package && section && option); + UCI_ASSERT(ctx, str); + UCI_ASSERT(ctx, ptr); - *package = strtok(str, "."); - if (!*package || !uci_validate_name(*package)) + memset(ptr, 0, sizeof(struct uci_ptr)); + + /* value */ + last = strchr(str, '='); + if (last) { + *last = 0; + last++; + ptr->value = last; + } + + ptr->package = strsep(&str, "."); + if (!ptr->package) goto error; - last = *package; - *section = strtok(NULL, "."); - if (!*section) + ptr->section = strsep(&str, "."); + if (!ptr->section) { + ptr->target = UCI_TYPE_PACKAGE; goto lastval; + } - last = *section; - *option = strtok(NULL, "."); - if (!*option) + ptr->option = strsep(&str, "."); + if (!ptr->option) { + ptr->target = UCI_TYPE_SECTION; goto lastval; + } else { + ptr->target = UCI_TYPE_OPTION; + } - last = *option; + tmp = strsep(&str, "."); + if (tmp) + goto error; lastval: - last = strchr(last, '='); - if (last) { - if (!value) - goto error; + if (ptr->package && !uci_validate_package(ptr->package)) + goto error; + if (ptr->section && !uci_validate_name(ptr->section)) + ptr->flags |= UCI_LOOKUP_EXTENDED; + if (ptr->option && !uci_validate_name(ptr->option)) + goto error; + if (ptr->value && !uci_validate_text(ptr->value)) + goto error; - *last = 0; - last++; - if (!*last) - goto error; - *value = last; + return 0; + +error: + memset(ptr, 0, sizeof(struct uci_ptr)); + UCI_THROW(ctx, UCI_ERR_PARSE); +} + + +__private void uci_parse_error(struct uci_context *ctx, char *reason) +{ + struct uci_parse_context *pctx = ctx->pctx; + + pctx->reason = reason; + pctx->byte = pctx_pos(pctx); + UCI_THROW(ctx, UCI_ERR_PARSE); +} + + + +/* + * open a stream and go to the right position + * + * note: when opening for write and seeking to the beginning of + * the stream, truncate the file + */ +__private FILE *uci_open_stream(struct uci_context *ctx, const char *filename, const char *origfilename, int pos, bool write, bool create) +{ + struct stat statbuf; + FILE *file = NULL; + int fd, ret; + int flags = (write ? O_RDWR : O_RDONLY); + mode_t mode = UCI_FILEMODE; + char *name = NULL; + char *filename2 = NULL; + + if (create) { + flags |= O_CREAT; + if (origfilename) { + name = basename((char *) origfilename); + } else { + name = basename((char *) filename); + } + if ((asprintf(&filename2, "%s/%s", ctx->confdir, name) < 0) || !filename2) { + UCI_THROW(ctx, UCI_ERR_MEM); + } else { + if (stat(filename2, &statbuf) == 0) + mode = statbuf.st_mode; + + free(filename2); + } + } + + if (!write && ((stat(filename, &statbuf) < 0) || + ((statbuf.st_mode & S_IFMT) != S_IFREG))) { + UCI_THROW(ctx, UCI_ERR_NOTFOUND); } - if (*section && !uci_validate_name(*section)) + fd = open(filename, flags, mode); + if (fd < 0) goto error; - if (*option && !uci_validate_name(*option)) + + ret = flock(fd, (write ? LOCK_EX : LOCK_SH)); + if ((ret < 0) && (errno != ENOSYS)) goto error; - goto done; + ret = lseek(fd, 0, pos); -error: - UCI_THROW(ctx, UCI_ERR_PARSE); + if (ret < 0) + goto error; + + file = fdopen(fd, (write ? "w+" : "r")); + if (file) + goto done; +error: + UCI_THROW(ctx, UCI_ERR_IO); done: - return 0; + return file; +} + +__private void uci_close_stream(FILE *stream) +{ + int fd; + + if (!stream) + return; + + fflush(stream); + fd = fileno(stream); + flock(fd, LOCK_UN); + fclose(stream); } +