Droppy/main.py at Main · blessingcharles/Droppy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
from pprint import pp, pprint
from typing import List
from rich.traceback import install
import webbrowser
from OutputGenerator.general_generator import GeneralOutputGenerator
from OutputGenerator.html_generator import HtmlGenerator
install(show_locals=True)

from DroppyAnalyzer import Token
from prettytable import PrettyTable
from DroppyAnalyzer.Scanner import ControlFlow, VulnScanner
from DroppyAnalyzer.Fuzzer import Fuzzer
from DroppyAnalyzer.Analyzer import DroppyAnalyzer

from utils.DroppyArgs import droopy_args
from DroppyAnalyzer.Tokenizer import Tokenizer
from utils.banner import print_seperator
from utils.colors import *
from utils.utils import dict_to_csv_writer, dir_create, recursive_dir_search

if __name__ == "__main__":

    (file ,directory , thread
        , output , verbose , is_logoutput ,
        generate_json , generate_xml) = droopy_args()

    csv_dir = f"{output}/csv_output"
    dir_create(output)
    dir_create(csv_dir)

    analyzer = DroppyAnalyzer(file , directory , output , thread , verbose)

    # tokenize both file or recursively in directory
    if file:
        analyzer.analyze_file()
    else:
        analyzer.analyze_dir()

    #log to output file
    if is_logoutput : analyzer.save_to_file()
    count_table = {}

    #scanning the project
    v_scanner = VulnScanner(analyzer.analyzed_files , output , csv_dir , verbose)
    v_scanner.scan()
    v_scanner.save_to_file()

    count_table["sinks_sources_count"] = v_scanner.xss_count
    count_table["deprecated_features_count"] = v_scanner.deprecated_features_count

    #Fuzzing the project
    fuzzer = Fuzzer(analyzer.analyzed_files , output ,csv_dir, verbose)
    fuzzer.fuzz()
    fuzzer.brief_detail()
    fuzzer.save_to_file()
    fuzzer.generate_total_results()

    count_table["variables_count"] = fuzzer.var_total_count
    count_table["functions_count"] = fuzzer.func_total_count
    count_table["constants_count"] = fuzzer.var_constants_count

    dict_to_csv_writer(count_table , f"{csv_dir}/Results.csv")

    cfa = ControlFlow(analyzer.analyzed_files , output , csv_dir , verbose)
    cfa.find_dead_code()
    cfa.save_to_file()
    html_gen = HtmlGenerator(output , csv_dir)
    html_gen.generate()

    gen = GeneralOutputGenerator(output,csv_dir)
    if generate_json:
        gen.generate_json()
    if generate_xml:
        gen.generate_xml()

    webbrowser.open(html_gen.html_output)