Esempio n. 1
0
def program(args):
    # Script cannot be called outside script directory. It contains a lot of os.getcwd().
    if not os.path.dirname(os.path.abspath(__file__)) == os.getcwd():
        Logger("Script cannot be called outside directory", Logger.ERROR)


    # Keep track of execution time
    start_time = time()

    # Read information from config file
    # Todo edit dockerfile with new path for report
    # ToDo create a settings class that parses the ini file with set and get functions

    config = configparser.ConfigParser()
    config.read("config.ini")
    server_disabled = config.getboolean("ProgramConfig", 'server_disabled')
    DRAG_DROP_SERVER_PORT = json.loads(config.get("Server", 'drag_drop_server_port'))

    # Update log level
    if not (args.log_warnings or args.log_errors):
        loglevel = 3
    else:
        loglevel = 1 if args.log_errors else 2
    config.set('ProgramConfig', 'loglevel', str(loglevel))
    with open("config.ini", "w+") as configfile:
        config.write(configfile)

    # Import the searchwords lists
    # Searchwords.searchwords_import(Searchwords())
    SearchLists()


    # Server(args) checks if the server should be run and handles the spawning of the server and control of it
    if not args.project:
        server(args, server_disabled, DRAG_DROP_SERVER_PORT)

    # For each project (read .ipa or .apk file), run the scripts.
    all_project_paths = args.project

    if not all_project_paths:
        sys.exit(0)
    for project_path in all_project_paths:
        try:
            Project.projects[project_path] = Project(project_path)
        except:
            sys.exit(0)

        report_folder = os.path.join(Project.projects[project_path].name, config.get("ProgramConfig", 'report_folder'))
        report_folder_start = os.path.join(os.getcwd(), report_folder, "start.html")

        Logger("Decompiling app...")
        Project.projects[project_path].app_prepper()
        Logger("Decompiling done.")
        Logger("Searching trough files")
        Project.projects[project_path].searchcontroller()
        Logger("Searching done.")
        Logger("start generating report")

        # ToDo: Generate the tree-view + Source code view for each SOURCE file
        all_files = dict()
        all_files.update(Project.projects[project_path].db_files)
        all_files.update(Project.projects[project_path].src_files)
        amount_files = len(all_files)
        for i, file in enumerate(all_files):
            Logger("progress: "+str(format((i/amount_files)*100, '.2f'))+"%", rewriteLine=True)
            hash_object = hashlib.md5(file.encode('utf-8'))
            file_report_file = os.path.join(report_folder, hash_object.hexdigest()+'.html')
            overview_html = Report_html(Project.projects[project_path])
            overview_html.header("tree")
            overview_html.navigation()
            overview_html.tree_view(Project.projects[project_path], file)
            overview_html.footer()
            f = codecs.open(file_report_file, 'w', encoding='utf8')
            f.write(overview_html.gethtml())
            # with open(file_report_file, 'w') as f:
            #     print(overview_html.gethtml(), file=f)
        Logger("progress: 100%  ")

        # Generate the startpage
        file_report_file = os.path.join(report_folder, 'start.html')
        overview_html = Report_html(Project.projects[project_path])
        overview_html.header("tree")
        overview_html.navigation()
        overview_html.tree_view(Project.projects[project_path], "")
        overview_html.footer()
        f = codecs.open(file_report_file, 'w', encoding='utf8')
        f.write(overview_html.gethtml())
        # with open(file_report_file, 'w') as f:
        #     print(overview_html.gethtml(), file=f)

        # Generate words overview html file
        words_overview_html_report_file = os.path.join(report_folder, "wordlist_overview.html")
        words_overview_html = Report_html(Project.projects[project_path])
        words_overview_html.header("words_overview")
        words_overview_html.navigation()
        words_overview_html.html_wordlist(Project.projects[project_path])
        words_overview_html.footer()
        with open(words_overview_html_report_file, 'w') as f:
            print(words_overview_html.gethtml(), file=f)

        # Generate lootbox
        lootbox_html_report_file = os.path.join(report_folder, "lootbox.html")
        lootbox_html_report = Report_html(Project.projects[project_path])
        lootbox_html_report.header("lootbox")
        lootbox_html_report.navigation()
        lootbox_html_report.lootbox()
        lootbox_html_report.footer()
        f = codecs.open(lootbox_html_report_file, 'w', encoding='utf8')
        f.write(lootbox_html_report.gethtml())
        # with open(lootbox_html_report_file, 'w') as f:
        #     print(lootbox_html_report.gethtml(), file=f)

        # Generate the treeview
        tree_js_file_path = os.path.join(report_folder, "tree_js_content.js")
        f = codecs.open(tree_js_file_path, 'w', encoding='utf8')
        f.write(Report_html.Tree_builder.tree_js_file(Project.projects[project_path]))
        # with open(tree_js_file_path, 'w') as f:
        #     print(Report_html.Tree_builder.tree_js_file(Project.projects[project_path]), file=f)

        # Generate looty.js file, for the zip creation process at the lootbox page
        Report_html().make_loot_report_content()

        # Write all log-events to logfile
        Logger.dump()

        # Log some end results
        if loglevel == 3:
            print("\n--------------------\n")
        Logger("Static code analyzer completed succesfully in %fs." % (time() - start_time))
        Logger("HTML report is available at: %s" % report_folder_start)
        if (not args.disable_browser) or (args.disable_server or server_disabled):
            Logger("Now automatically opening the HTML report.")
            # Open the webbrowser to the generated start page.
            if sys.platform == "darwin":  # check if on OSX
                # strip off http:///
                report_folder_start = str(report_folder_start).strip("http:///")
                report_folder_start = "file:///" + report_folder_start
            webbrowser.open(report_folder_start)
    # Exit program
    sys.exit()
Esempio n. 2
0
def program():
    # Script cannot be called outside script directory. It contains a lot of os.getcwd().
    if not os.path.dirname(os.path.abspath(__file__)) == os.getcwd():
        Logger("Script cannot be called outside directory", Logger.ERROR)

    # Read information from config file
    config = configparser.ConfigParser()
    config.read("config.ini")
    report_folder = config.get("ProgramConfig", 'report_folder')
    development = config.getint("Development", 'development')
    # Import the searchwords lists
    Searchwords.searchwords_import(Searchwords())

    # For each project (read .ipa or .apk file), run the scripts.
    all_project_paths = list()
    if len(sys.argv) > 1:
        all_project_paths = sys.argv[1:]
    else:
        # No arguments given.
        Logger("No input file given", Logger.ERROR)
    for project_path in all_project_paths:
        Project.projects[project_path] = Project(project_path)
        Logger("Decompiling app...")
        Project.projects[project_path].app_prepper()
        Logger("Decompiling done.")
        Logger("Searching trough files")
        Project.projects[project_path].searchcontroller()
        Logger("Searching done.")
        Logger("start generating report")

    # To Do: Generate the tree-view + Source code view for each SOURCE file
    all_files = dict()
    all_files.update(Project.projects[project_path].db_files)
    all_files.update(Project.projects[project_path].src_files)
    amount_files = len(all_files)
    i = 0
    for file in all_files:
        #os.system('cls' if os.name == 'nt' else 'clear')   #  This function is making the program 5000% slower
        Logger("progress: "+str(format((i/amount_files)*100, '.2f'))+"%")
        i += 1
        hash_object = hashlib.md5(file.encode('utf-8'))
        file_report_file = os.path.join(report_folder, hash_object.hexdigest()+'.html')
        overview_html = Report_html(Project.projects[project_path])
        overview_html.header("tree")
        overview_html.navigation()
        overview_html.tree_view(Project.projects[project_path], file)
        overview_html.footer()
        f = codecs.open(file_report_file, 'w', encoding='utf8')
        f.write(overview_html.gethtml())
        # with open(file_report_file, 'w') as f:
        #     print(overview_html.gethtml(), file=f)

    # Generate the startpage
    file_report_file = os.path.join(report_folder, 'start.html')
    overview_html = Report_html(Project.projects[project_path])
    overview_html.header("tree")
    overview_html.navigation()
    overview_html.tree_view(Project.projects[project_path], "")
    overview_html.footer()
    f = codecs.open(file_report_file, 'w', encoding='utf8')
    f.write(overview_html.gethtml())
    # with open(file_report_file, 'w') as f:
    #     print(overview_html.gethtml(), file=f)

    # Generate words overview html file
    words_overview_html_report_file = os.path.join(report_folder, "wordlist_overview.html")
    words_overview_html = Report_html(Project.projects[project_path])
    words_overview_html.header("words_overview")
    words_overview_html.navigation()
    words_overview_html.html_wordlist(Project.projects[project_path])
    words_overview_html.footer()
    with open(words_overview_html_report_file, 'w') as f:
        print(words_overview_html.gethtml(), file=f)

    # Generate lootbox
    lootbox_html_report_file = os.path.join(report_folder, "lootbox.html")
    lootbox_html_report = Report_html(Project.projects[project_path])
    lootbox_html_report.header("lootbox")
    lootbox_html_report.navigation()
    lootbox_html_report.lootbox()
    lootbox_html_report.footer()
    f = codecs.open(lootbox_html_report_file, 'w', encoding='utf8')
    f.write(lootbox_html_report.gethtml())
    # with open(lootbox_html_report_file, 'w') as f:
    #     print(lootbox_html_report.gethtml(), file=f)

    # Generate the treeview
    tree_js_file_path = os.path.join(report_folder, "tree_js_content.js")
    f = codecs.open(tree_js_file_path, 'w', encoding='utf8')
    f.write(Report_html.Tree_builder.tree_js_file(Project.projects[project_path]))
    # with open(tree_js_file_path, 'w') as f:
    #     print(Report_html.Tree_builder.tree_js_file(Project.projects[project_path]), file=f)


    # Generate looty.js file, for the zip creation process at the lootbox page
    Report_html().make_loot_report_content()

    # Write all log-events to logfile
    Logger.dump()

    # Open the webbrowser to the generated start page.
    url = os.path.join(report_folder, "start.html")
    webbrowser.open(url)

    # Exit program
    sys.exit()