GraFScan.py 文件源码

def main():
    banner()
    results = []
    args = getArguments(sys.argv)
    print "Start the analyze: "
    if args.tor == True:
            socks.setdefaultproxy(proxy_type=socks.PROXY_TYPE_SOCKS5, addr="127.0.0.1", port=9050)
            socket.socket = socks.socksocket
    for ip in args.listIps:
        print "IP: " + ip
        print "-----------------------"
        if args.all:
            dataN = analyzeIP_Neo4j(str(ip),args)
            if dataN:
                results.append(dataN)
            dataO = analyzeIP_Orient(str(ip),args)
            if dataO:
                results.append(dataO)
            dataA = analyzeIP_ArangoDB(str(ip),args)
            if dataA:
                results.append(dataA)
            dataV = analyzeIP_Virtuoso(str(ip),args)
            if dataV:
                results.append(dataV)
            dataAl = analyzeIP_Allegro(str(ip),args)
            if dataAl:
                results.append(dataAl)
        elif args.neo4j:
            data = analyzeIP_Neo4j(str(ip),args)
            if data:
                results.append(data)
        elif args.orient:
            data = analyzeIP_Orient(str(ip),args)
            if data:
                results.append(data)
        elif args.arango:
            data = analyzeIP_ArangoDB(str(ip),args)
            if data:
                results.append(data)
        elif args.virtuoso:
            data = analyzeIP_Virtuoso(str(ip),args)
            if data:
                results.append(data)
        elif args.allegro:
            data = analyzeIP_Allegro(str(ip),args)
            if data:
                results.append(data)    
        else:
            print "Error with arguments"
    if results:
        print "Writting the results in the output file: " + args.output
        fileout = open(args.output, "w")
        json_str = json.dumps(results)
        fileout.write(json_str)
        fileout.close()
    else:
        print "-----------"
        print "No results"