picnoir/Systemd
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

macro: add a macro to test whether a value is in a specified list

Browse source 
Introduce IN_SET() macro to nicely check whether a value a is one of a
few listed values.

This makes writing this:

        if (a == 1 || a == 7 || a == 8 || a == 9)

nicer, by allowing this:

        if (IN_SET(a, 1, 7, 8, 9))

This is particularly useful for state machine enums.
This commit is contained in:
Lennart Poettering 2013-12-02 23:08:25 +01:00
parent 1ae464e093
commit cabb780688
2 changed files with 24 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/shared/macro.h
View file

@ -285,4 +285,17 @@ do { \
#define SET_FLAG(v, flag, b) \
(v) = (b) ? ((v) | (flag)) : ((v) & ~(flag))
#define IN_SET(x, ...) ({ \
typeof(x) _x = (x); \
unsigned _i; \
bool _found = false; \
for (_i = 0; _i < sizeof((typeof(_x)[]) { __VA_ARGS__ })/sizeof(typeof(_x)); _i++) \
if (((typeof(_x)[]) { __VA_ARGS__ })[_i] == _x) { \
_found = true; \
break; \
} \
_found; \
})
#include "log.h"

11
src/test/test-util.c
View file

@ -568,6 +568,16 @@ static void test_get_files_in_directory(void) {
assert_se(get_files_in_directory(".", NULL) >= 0);
}
static void test_in_set(void) {
assert_se(IN_SET(1, 1));
assert_se(IN_SET(1, 1, 2, 3, 4));
assert_se(IN_SET(2, 1, 2, 3, 4));
assert_se(IN_SET(3, 1, 2, 3, 4));
assert_se(IN_SET(4, 1, 2, 3, 4));
assert_se(!IN_SET(0, 1));
assert_se(!IN_SET(0, 1, 2, 3, 4));
}
int main(int argc, char *argv[]) {
test_streq_ptr();
test_first_word();
@ -604,6 +614,7 @@ int main(int argc, char *argv[]) {
test_split_pair();
test_fstab_node_to_udev_node();
test_get_files_in_directory();
test_in_set();
return 0;
}