Function simple_cli

src/tasks.py:107–394

()

def simple_cli():
    start_time = time.time()

    parser = argparse.ArgumentParser(
                    prog='GPTScan',
                    description='GPTScan is an AI based smart contract vulnerability scanner.')
    parser.add_argument("-s", "--source", help="The source code directory", required=True)
    # not need ast, compile first
    # parser.add_argument("-a", "--ast", help="The AST directory", required=True)
    parser.add_argument("-o", "--output", help="The output file", required=True)
    parser.add_argument("-k", "--gptkey", help="The OpenAI API key", required=True)
    
    
    scan_rules = load_all_rules()
    console.log(f"Loaded [bold green]{len(scan_rules)}[/bold green] rules")
    
    source_dir = parser.parse_args().source

    try:
        falcon_instance = compile_project(source_dir)
    except:
        console.log(traceback.format_exc())
        console.log("Compile [bold red]failed[/bold red].")
        console.log("[yellow]Since the compilation is failed, some static analysis tool may not be enabled, which may cause lower precision and recall.[/yellow]")
        falcon_instance = None
    output_file = parser.parse_args().output
    gptkey = parser.parse_args().gptkey

    os.environ["OPENAI_API_KEY"] = gptkey

    import analyze_pipeline
    import chatgpt_api

    res, cg, meta_data = analyze_pipeline.ask_whether_has_vul_with_scenario_v9(
        source_dir, scan_rules)
    final_result = {}
    # logger.info(res)
    for file in res:
        with open(file) as f:
            source = f.read().splitlines()
            for contract in res[file]:
                for function1 in res[file][contract]:
                    
                    # I think should first ask for function 1
                    # this is a key value map for vul -> result
                    function1_tmp_result = {}

                    for function2 in res[file][contract][function1]:
                        confirmed_vuls = {}
                        for vul in res[file][contract][function1][function2]["data"]:
                            meta_data["rules_types_for_static"].add(vul["name"])
                            # if the rule need static check
                            if "static" in vul:
                                # if need static check

                                # if function1 is not asked yet, ask for function1 first
                                if vul["name"] not in function1_tmp_result:
                                    function1_detail = cg.get_function_detail(file, contract, function1)