adulau/git-vuln-finder
1
0
Fork 0
mirror of https://github.com/cve-search/git-vuln-finder.git synced 2024-11-22 23:07:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

'# FIXME: Is there a better way?': yes

Browse source
This commit is contained in:
Cédric Bonhomme 2020-01-06 08:08:08 +01:00
parent 335bdff4cb
commit 851d8946fd
Signed by untrusted user who does not match committer: cedric
GPG key ID: A1CB94DE57B7A70D
1 changed files with 11 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
git_vuln_finder/pattern.py
View file

@ -12,6 +12,14 @@
import os import os
import re import re
from collections import defaultdict
def tree():
"""Autovivification.
Call it a tree or call it 'patterns'.
"""
return defaultdict(tree)
PATTERNS_PATH = "./git_vuln_finder/patterns" PATTERNS_PATH = "./git_vuln_finder/patterns"
@ -43,7 +51,7 @@ def build_pattern(pattern_file):
def get_patterns(patterns_path=PATTERNS_PATH): def get_patterns(patterns_path=PATTERNS_PATH):
patterns = {} patterns = tree()
for root, dirs, files in os.walk(patterns_path): for root, dirs, files in os.walk(patterns_path):
path = root.split(os.sep) path = root.split(os.sep)
for f in files: for f in files:
@ -59,17 +67,6 @@ def get_patterns(patterns_path=PATTERNS_PATH):
severity = npath[1] severity = npath[1]
pattern_category = f pattern_category = f
try: # FIXME: Is there a better way?
a = patterns[lang]
except KeyError:
patterns[lang] = {}
try:
a = patterns[lang][severity]
except KeyError:
patterns[lang][severity] = {}
try:
a = patterns[lang][severity][pattern_category]
except KeyError:
rex = build_pattern(root + os.sep + f) rex = build_pattern(root + os.sep + f)
patterns[lang][severity][pattern_category] = re.compile(rex) patterns[lang][severity][pattern_category] = re.compile(rex)