rizin/librz/util/name.c
2024-09-15 19:12:07 +08:00

107 lines
2.2 KiB
C

// 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 = rz_str_dup(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;
}