name.c

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2009-2020 pancake <pancake@nopcode.org>
// SPDX-License-Identifier: LGPL-3.0-only
 
#include <rz_util.h>
 
RZ_API bool rz_name_validate_char(const char ch, bool strict) {
    if ((ch >= 'a' && ch <= 'z') || (ch >= 'A' && ch <= 'Z') || (IS_DIGIT(ch))) {
        return true;
    }
    switch (ch) {
    case ':':
    case '.':
    case '_':
        return true;
    case '-':
    case '[':
    case ']':
        return !strict;
    default:
        return false;
    }
}
 
RZ_API bool rz_name_check(const char *name, bool strict) {
    /* Cannot start by number */
    if (!name || !*name || IS_DIGIT(*name)) {
        return false;
    }
    /* Cannot contain non-alphanumeric chars + [:._] */
    for (; *name != '\0'; name++) {
        if (!rz_name_validate_char(*name, strict)) {
            return false;
        }
    }
    return true;
}
 
static inline bool is_special_char(char *name) {
    const char n = *name;
    return (n == 'b' || n == 'f' || n == 'n' || n == 'r' || n == 't' || n == 'v' || n == 'a');
}
 
RZ_API bool rz_name_filter(char *name, int maxlen, bool strict) {
    size_t i, len;
    if (!name) {
        return false;
    }
    if (maxlen < 0) {
        maxlen = strlen(name);
    }
    rz_str_trim(name);
    char *oname = name;
    for (i = 0; *name; name++, i++) {
        if (maxlen && i > maxlen) {
            *name = '\0';
            break;
        }
        if (!rz_name_validate_char(*name, strict) && *name != '\\') {
            *name = '_';
        }
    }
    while (i > 0) {
        if (*(name - 1) == '\\' && is_special_char(name)) {
            *name = '_';
            *(name - 1) = '_';
        }
        if (*name == '\\') {
            *name = '_';
        }
        name--;
        i--;
    }
    if (*name == '\\') {
        *name = '_';
    }
    // trimming trailing and leading underscores
    len = strlen(name);
    for (; len > 0 && *(name + len - 1) == '_'; len--) {
        ;
    }
    if (!len) { // name consists only of underscores
        return rz_name_check(oname, strict);
    }
    for (i = 0; *(name + i) == '_'; i++, len--) {
        ;
    }
    memmove(name, name + i, len);
    *(name + len) = '\0';
    return rz_name_check(oname, strict);
}
 
RZ_API char *rz_name_filter2(const char *name, bool strict) {
    size_t i;
    while (!rz_name_validate_char(*name, strict)) {
        name++;
    }
    char *res = strdup(name);
    for (i = 0; res[i]; i++) {
        if (!rz_name_validate_char(res[i], strict)) {
            res[i] = '_';
        }
    }
    for (i--; i != 0 && res[i] == '_'; i--) {
        res[i] = '\0';
    }
    return res;
}