From 96c3a7c2459bdafdbd12f307aa5d1e7c61c9bc52 Mon Sep 17 00:00:00 2001
From: Rasmus Villemoes <rv@rasmusvillemoes.dk>
Date: Sat, 7 Feb 2015 01:09:35 +0100
Subject: [PATCH] smatch: Add --enable option

It can be useful, especially when developing a new check, to only run
a subset of checks. The --enable option takes a comma-separated list
of integers or ranges and only enables those checks (the default is of
course that everything is enabled).

For now, anything called register_* is unconditionally called, since lots
of stuff seems to depend on that.
---
 smatch.c | 49 +++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 47 insertions(+), 2 deletions(-)

diff --git a/smatch.c b/smatch.c
index 65f9f65c..ff81469a 100644
--- a/smatch.c
+++ b/smatch.c
@@ -32,6 +32,7 @@ int option_full_path = 0;
 int option_param_mapper = 0;
 int option_call_tree = 0;
 int option_no_db = 0;
+int option_enable = 0;
 int option_debug_related;
 int option_file_output;
 int option_time;
@@ -39,10 +40,11 @@ char *option_datadir_str;
 FILE *sm_outfd;
 
 typedef void (*reg_func) (int id);
-#define CK(_x) {.name = #_x, .func = &_x},
+#define CK(_x) {.name = #_x, .func = &_x, .enabled = 0},
 static struct reg_func_info {
 	const char *name;
 	reg_func func;
+	int enabled;
 } reg_funcs[] = {
 	{NULL, NULL},
 #include "check_list.h"
@@ -79,6 +81,42 @@ static void show_checks(void)
 	}
 	exit(1);
 }
+static void enable_check(int i)
+{
+	if (1 <= i && i < ARRAY_SIZE(reg_funcs))
+		reg_funcs[i].enabled = 1;
+}
+
+static void enable_checks(const char *s)
+{
+	int n = 0, lo = -1, i;
+
+	do {
+		switch (*s) {
+		case ',':
+		case '\0':
+			if (lo < 0)
+				enable_check(n);
+			else
+				for (i = lo; i <= n; ++i)
+					enable_check(i);
+			lo = -1;
+			n = 0;
+			break;
+		case '-':
+			lo = n;
+			n = 0;
+			break;
+		case '0' ... '9':
+			n = 10*n + (*s - '0');
+			break;
+		default:
+			fprintf(stderr, "invalid character '%c'\n", *s);
+			exit(1);
+			
+		}
+	} while (*s++);
+}
 
 static void help(void)
 {
@@ -164,6 +202,12 @@ void parse_args(int *argcp, char ***argvp)
 			(*argvp)[1] = (*argvp)[0];
 			found = 1;
 		}
+		if (!found && strncmp((*argvp)[1], "--enable=", 9) == 0) {
+			enable_checks((*argvp)[1] + 9);
+			option_enable = 1;
+			(*argvp)[1] = (*argvp)[0];
+			found = 1;
+		}
 
 		OPTION(spammy);
 		OPTION(info);
@@ -252,7 +296,8 @@ int main(int argc, char **argv)
 		func = reg_funcs[i].func;
 		/* The script IDs start at 1.
 		   0 is used for internal stuff. */
-		func(i);
+		if (!option_enable || reg_funcs[i].enabled || !strncmp(reg_funcs[i].name, "register_", 9))
+			func(i);
 	}
 
 	smatch(argc, argv);
-- 
2.11.4.GIT