summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorLennart Poettering <lennart@poettering.net>2013-12-02 23:08:25 +0100
committerLennart Poettering <lennart@poettering.net>2013-12-02 23:32:34 +0100
commitcabb78068899232c152f4585f19d023e373aa73d (patch)
tree7c4fa9684386a6b681ef0475315b0d30d6f12e18
parent1ae464e09376853c52075ec4d8a6bfc4b4036d0c (diff)
macro: add a macro to test whether a value is in a specified list
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.
-rw-r--r--src/shared/macro.h13
-rw-r--r--src/test/test-util.c11
2 files changed, 24 insertions, 0 deletions
diff --git a/src/shared/macro.h b/src/shared/macro.h
index 9f5f51cb13..54b641be23 100644
--- a/src/shared/macro.h
+++ b/src/shared/macro.h
@@ -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"
diff --git a/src/test/test-util.c b/src/test/test-util.c
index afc3247ffd..05b2294e24 100644
--- a/src/test/test-util.c
+++ b/src/test/test-util.c
@@ -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;
}