summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--waftools/checks/generic.py11
1 files changed, 7 insertions, 4 deletions
diff --git a/waftools/checks/generic.py b/waftools/checks/generic.py
index efe8bb51ff..9f5f9d6509 100644
--- a/waftools/checks/generic.py
+++ b/waftools/checks/generic.py
@@ -41,10 +41,13 @@ def check_libs(libs, function):
def check_statement(header, statement, **kw_ext):
def fn(ctx, dependency_identifier, **kw):
- fragment = """
- #include <{0}>
- int main(int argc, char **argv)
- {{ {1}; return 0; }} """.format(header, statement)
+ headers = header
+ if not isinstance(headers, list):
+ headers = [header]
+ hs = "\n".join(["#include <{0}>".format(h) for h in headers])
+ fragment = ("{0}\n"
+ "int main(int argc, char **argv)\n"
+ "{{ {1}; return 0; }}").format(hs, statement)
opts = __merge_options__(dependency_identifier,
{'fragment':fragment},
__define_options__(dependency_identifier),