diff --git a/snaffcore/classifier.py b/snaffcore/classifier.py index 9699b89..9cb3d93 100644 --- a/snaffcore/classifier.py +++ b/snaffcore/classifier.py @@ -25,6 +25,7 @@ class Rules: # TODO + def is_interest(file): # massive_wordlist = prepare_classifiers() # print(massive_wordlist) diff --git a/snaffcore/go_snaffle.py b/snaffcore/go_snaffle.py index 49fd48b..5fcc96f 100644 --- a/snaffcore/go_snaffle.py +++ b/snaffcore/go_snaffle.py @@ -94,7 +94,7 @@ def begin_snaffle(options): # filelist.append(file) # Ask do they want file sizes? # log.info(f"{target} Found file in {share}: {file.get_longname()}") - naive_classify(share, file) + naive_classify(share, file, prepped_rules) # log.info(f"{target} Found file in {share}: {file}") except FileListError: log.error( @@ -166,11 +166,11 @@ def list_computers(connection: Connection, domain): return None # TODO -def naive_classify(share, file): +def naive_classify(share, file, rules:Rules): log.info(f"{share}: {file.get_longname()}") - if is_interest(file): + if is_interest(file, rules): log.info(f"Found interesting file: {share}/{file}")