Files
systemd/src/basic/glob-util.c
Yu Watanabe 2a5f950e56 glob-util: rework safe_glob()
Currently, callers of safe_glob() set an empty glob_t or glob_t with
opendir func, and all other components are always zero.
So, let's introduce safe_glob_full() which optionally takes opendir
function, rather than glob_t, and returns result strv, rather than
storing results in glob_t.
Also, introduce safe_glob() which is a trivial wrapper of
safe_glob_full() without opendir func.

No functional change, just refactoring.
2025-06-23 23:11:44 +09:00

111 lines
2.8 KiB
C

/* SPDX-License-Identifier: LGPL-2.1-or-later */
#include <dirent.h>
#include <sys/stat.h>
#include "dirent-util.h"
#include "errno-util.h"
#include "glob-util.h"
#include "string-util.h"
#include "strv.h"
DEFINE_TRIVIAL_DESTRUCTOR(closedir_wrapper, void, closedir);
int safe_glob_full(const char *path, int flags, opendir_t opendir_func, char ***ret) {
_cleanup_(globfree) glob_t g = {
.gl_closedir = closedir_wrapper,
.gl_readdir = (struct dirent* (*)(void *)) readdir_no_dot,
.gl_opendir = (void* (*)(const char *)) (opendir_func ?: opendir),
.gl_lstat = lstat,
.gl_stat = stat,
};
int r;
assert(path);
errno = 0;
r = glob(path, flags | GLOB_ALTDIRFUNC, NULL, &g);
if (r == GLOB_NOMATCH)
return -ENOENT;
if (r == GLOB_NOSPACE)
return -ENOMEM;
if (r != 0)
return errno_or_else(EIO);
if (strv_isempty(g.gl_pathv))
return -ENOENT;
if (ret) {
*ret = g.gl_pathv;
TAKE_STRUCT(g); /* To avoid the result being freed. */
}
return 0;
}
int glob_first(const char *path, char **ret) {
_cleanup_strv_free_ char **v = NULL;
int r;
assert(path);
r = safe_glob(path, GLOB_NOSORT|GLOB_BRACE, &v);
if (r == -ENOENT) {
if (ret)
*ret = NULL;
return false;
}
if (r < 0)
return r;
assert(!strv_isempty(v));
if (ret) {
/* Free all results except for the first one. */
STRV_FOREACH(p, strv_skip(v, 1))
*p = mfree(*p);
/* Then, take the first result. */
*ret = TAKE_PTR(*v);
}
return true;
}
int glob_extend(char ***strv, const char *path, int flags) {
char **v;
int r;
assert(path);
r = safe_glob(path, GLOB_NOSORT|GLOB_BRACE|flags, &v);
if (r < 0)
return r;
return strv_extend_strv_consume(strv, v, /* filter_duplicates = */ false);
}
int glob_non_glob_prefix(const char *path, char **ret) {
/* Return the path of the path that has no glob characters. */
size_t n = strcspn(path, GLOB_CHARS);
if (path[n] != '\0')
while (n > 0 && path[n-1] != '/')
n--;
if (n == 0)
return -ENOENT;
char *ans = strndup(path, n);
if (!ans)
return -ENOMEM;
*ret = ans;
return 0;
}
bool string_is_glob(const char *p) {
/* Check if a string contains any glob patterns. */
return !!strpbrk(p, GLOB_CHARS);
}