also change CONFIG_*_MODULE

Signed-off-by: Johannes Berg <johannes@sipsolutions.net>

diff --git a/gentree.py b/gentree.py
index 48684c389a234f959c02f239b73eb11494232a78..53cae5df14da2d2fb7240ca87e8ebafc7bf62b27 100755 (executable)
--- a/gentree.py
+++ b/gentree.py
@@ -127,7 +127,7 @@ def main():
     # XXX Apply patches here!!
 
     # rewrite Makefile and source symbols
-    r = 'CONFIG_((' + '|'.join(symbols) + ')([^A-Za-z0-9_]|$))'
+    r = 'CONFIG_((' + '|'.join([s + '(_MODULE)?' for s in symbols]) + ')([^A-Za-z0-9_]|$))'
     r = re.compile(r, re.MULTILINE)
     for root, dirs, files in os.walk(args.outdir):
         for f in files: