fdset: minor modernizations

Let's use more _cleanup_ expressions. Various other modernizations. No
actual code changes, except for maybe a conversion to use heap memory
when generating an array of fds, instead of stack as before. Given that
fdstores are typically user controlled, that should be a wise idea.
This commit is contained in:
Lennart Poettering
2023-03-29 19:09:10 +02:00
parent 5f43c97cd2
commit 08a0ebc6be
+45 -56
View File
@@ -15,6 +15,7 @@
#include "parse-util.h"
#include "path-util.h"
#include "set.h"
#include "stat-util.h"
#define MAKE_SET(s) ((Set*) s)
#define MAKE_FDSET(s) ((FDSet*) s)
@@ -23,27 +24,29 @@ FDSet *fdset_new(void) {
return MAKE_FDSET(set_new(NULL));
}
int fdset_new_array(FDSet **ret, const int *fds, size_t n_fds) {
size_t i;
FDSet *s;
static inline void fdset_shallow_freep(FDSet **s) {
/* Destroys the set, but does not free the fds inside, like fdset_free()! */
set_free(MAKE_SET(*ASSERT_PTR(s)));
}
int fdset_new_array(FDSet **ret, const int fds[], size_t n_fds) {
_cleanup_(fdset_shallow_freep) FDSet *s = NULL;
int r;
assert(ret);
assert(fds || n_fds == 0);
s = fdset_new();
if (!s)
return -ENOMEM;
for (i = 0; i < n_fds; i++) {
for (size_t i = 0; i < n_fds; i++) {
r = fdset_put(s, fds[i]);
if (r < 0) {
set_free(MAKE_SET(s));
if (r < 0)
return r;
}
}
*ret = s;
*ret = TAKE_PTR(s);
return 0;
}
@@ -78,7 +81,8 @@ int fdset_put(FDSet *s, int fd) {
}
int fdset_put_dup(FDSet *s, int fd) {
int copy, r;
_cleanup_close_ int copy = -EBADF;
int r;
assert(s);
assert(fd >= 0);
@@ -88,12 +92,10 @@ int fdset_put_dup(FDSet *s, int fd) {
return -errno;
r = fdset_put(s, copy);
if (r < 0) {
safe_close(copy);
if (r < 0)
return r;
}
return copy;
return TAKE_FD(copy);
}
bool fdset_contains(FDSet *s, int fd) {
@@ -110,53 +112,46 @@ int fdset_remove(FDSet *s, int fd) {
return set_remove(MAKE_SET(s), FD_TO_PTR(fd)) ? fd : -ENOENT;
}
int fdset_new_fill(FDSet **_s) {
int fdset_new_fill(FDSet **ret) {
_cleanup_(fdset_shallow_freep) FDSet *s = NULL;
_cleanup_closedir_ DIR *d = NULL;
int r = 0;
FDSet *s;
int r;
assert(_s);
assert(ret);
/* Creates an fdset and fills in all currently open file
* descriptors. */
/* Creates an fdset and fills in all currently open file descriptors. */
d = opendir("/proc/self/fd");
if (!d)
if (!d) {
if (errno == ENOENT && proc_mounted() == 0)
return -ENOSYS;
return -errno;
}
s = fdset_new();
if (!s) {
r = -ENOMEM;
goto finish;
}
if (!s)
return -ENOMEM;
FOREACH_DIRENT(de, d, return -errno) {
int fd = -EBADF;
r = safe_atoi(de->d_name, &fd);
if (r < 0)
goto finish;
return r;
if (fd < 3)
continue;
if (fd == dirfd(d))
continue;
r = fdset_put(s, fd);
if (r < 0)
goto finish;
return r;
}
r = 0;
*_s = TAKE_PTR(s);
finish:
/* We won't close the fds here! */
if (s)
set_free(MAKE_SET(s));
return r;
*ret = TAKE_PTR(s);
return 0;
}
int fdset_cloexec(FDSet *fds, bool b) {
@@ -174,46 +169,40 @@ int fdset_cloexec(FDSet *fds, bool b) {
return 0;
}
int fdset_new_listen_fds(FDSet **_s, bool unset) {
int fdset_new_listen_fds(FDSet **ret, bool unset) {
_cleanup_(fdset_shallow_freep) FDSet *s = NULL;
int n, fd, r;
FDSet *s;
assert(_s);
assert(ret);
/* Creates an fdset and fills in all passed file descriptors */
s = fdset_new();
if (!s) {
r = -ENOMEM;
goto fail;
}
if (!s)
return -ENOMEM;
n = sd_listen_fds(unset);
for (fd = SD_LISTEN_FDS_START; fd < SD_LISTEN_FDS_START + n; fd ++) {
r = fdset_put(s, fd);
if (r < 0)
goto fail;
return r;
}
*_s = s;
*ret = TAKE_PTR(s);
return 0;
fail:
if (s)
set_free(MAKE_SET(s));
return r;
}
int fdset_close_others(FDSet *fds) {
void *e;
int *a = NULL;
_cleanup_free_ int *a = NULL;
size_t j = 0, m;
void *e;
m = fdset_size(fds);
if (m > 0) {
a = newa(int, m);
a = new(int, m);
if (!a)
return -ENOMEM;
SET_FOREACH(e, MAKE_SET(fds))
a[j++] = PTR_TO_FD(e);
}