X-Git-Url: http://git.ozlabs.org/?p=ccan;a=blobdiff_plain;f=tools%2Frun_tests.c;h=e50d1910a210596f09832a691c36267b492d33a2;hp=b4125424a60859f71bb50e7bfaf9384e54a0e1c2;hb=c1daa044b22fce3ca80d3430e3e1ad9360f8a4f1;hpb=9d724ba2aa3b69cc56d20e5f578b2c1399e965f6

diff --git a/tools/run_tests.c b/tools/run_tests.c
index b4125424..e50d1910 100644
--- a/tools/run_tests.c
+++ b/tools/run_tests.c
@@ -16,23 +16,20 @@ static struct test *tests = NULL;
 static struct obj *objs = NULL;
 static int verbose;
 
-struct test_type
-{
+struct test_type {
 	const char *name;
 	void (*buildfn)(const char *dir, struct test_type *t, const char *name,
-			const char *apiobj);
+			const char *apiobj, const char *libs);
 	void (*runfn)(const char *name);
 };
 
-struct test
-{
+struct test {
 	struct test *next;
 	struct test_type *type;
 	char *name;
 };
 
-struct obj
-{
+struct obj {
 	struct obj *next;
 	bool generate;
 	char *name;
@@ -106,14 +103,14 @@ static void add_obj(const char *testdir, const char *name, bool generate)
 }
 
 static int build(const char *dir, const char *name, const char *apiobj,
-		 int fail)
+		 const char *libs, int fail)
 {
 	const char *cmd;
 	int ret;
 
-	cmd = talloc_asprintf(name, "gcc " CFLAGS " %s -o %s %s %s %s %s",
+	cmd = talloc_asprintf(name, "gcc " CFLAGS " %s -o %s %s %s %s%s %s",
 			      fail ? "-DFAIL" : "",
-			      output_name(name), name, apiobj, obj_list(),
+			      output_name(name), name, apiobj, obj_list(), libs,
 			      verbose ? "" : "> /dev/null 2>&1");
 
 	if (verbose)
@@ -127,25 +124,26 @@ static int build(const char *dir, const char *name, const char *apiobj,
 }
 
 static void compile_ok(const char *dir, struct test_type *t, const char *name,
-		       const char *apiobj)
+		       const char *apiobj, const char *libs)
 {
-	ok(build(dir, name, "", 0) == 0, "%s %s", t->name, name);
+	ok(build(dir, name, "", libs, 0) == 0, "%s %s", t->name, name);
 }
 
 /* api tests get the API obj linked in as well. */
 static void compile_api_ok(const char *dir, struct test_type *t,
-			   const char *name, const char *apiobj)
+			   const char *name, const char *apiobj,
+			   const char *libs)
 {
-	ok(build(dir, name, apiobj, 0) == 0, "%s %s", t->name, name);
+	ok(build(dir, name, apiobj, libs, 0) == 0, "%s %s", t->name, name);
 }
 
 static void compile_fail(const char *dir, struct test_type *t, const char *name,
-			 const char *apiobj)
+			 const char *apiobj, const char *libs)
 {
-	if (build(dir, name, "", 0) != 0)
+	if (build(dir, name, "", libs, 0) != 0)
 		fail("non-FAIL build %s", name);
 	else
-		ok(build(dir, name, "", 1) > 0, "%s %s", t->name, name);
+		ok(build(dir, name, "", libs, 1) > 0, "%s %s", t->name, name);
 }
 
 static void no_run(const char *name)
@@ -176,6 +174,7 @@ int main(int argc, char *argv[])
 	struct dirent *d;
 	char *testdir, *cwd;
 	const char *apiobj = "";
+	char *libs = talloc_strdup(NULL, "");
 	struct test *test;
 	unsigned int num_tests = 0, num_objs = 0, i;
 
@@ -185,6 +184,13 @@ int main(int argc, char *argv[])
 		argv++;
 	}
 
+	while (argc > 1 && strstarts(argv[1], "--lib=")) {
+		libs = talloc_asprintf_append(libs, " -l%s",
+					      argv[1] + strlen("--lib="));
+		argc--;
+		argv++;
+	}
+
 	if (argc > 1 && strstarts(argv[1], "--apiobj=")) {
 		apiobj = argv[1] + strlen("--apiobj=");
 		argc--;
@@ -227,7 +233,8 @@ int main(int argc, char *argv[])
 
 	/* Do all the test compilations. */
 	for (test = tests; test; test = test->next)
-		test->type->buildfn(argv[1], test->type, test->name, apiobj);
+		test->type->buildfn(argv[1], test->type, test->name,
+				    apiobj, libs);
 
 	cleanup_objs();