2016-07-24 9 views
0

Ich habe ein Python-Skript geschrieben Zertifikate von einer Liste der IP-Adresse holen eine Domain entsprechen:for-Schleife einmal in Thread läuft in Python 3

#! /usr/bin/env python3 

import ssl 
import socket 
import argparse 
from threading import Thread, Lock 
from itertools import islice 

class scanThread(Thread): 
    def __init__(self,iplist, q, hostname, port): 
     Thread.__init__(self) 
     self.iplist = iplist 
     self.hostname = hostname 
     self.port = port 
     self.queue = q 

    def dummy(self,ip): 
     print("Running dummy") 

    def checkCert(self, ip): 
     print('Processing IP: %s' % ip) 
     ctx = ssl.create_default_context() 
     s = ctx.wrap_socket(socket.socket(), server_hostname=self.hostname) 
     try: 
      s.connect((ip, self.port)) 
      cert = s.getpeercert() 
      if cert['subjectAltName'][0][1].find(hostname) != -1: 
       return ip 
     except (ssl.CertificateError, ssl.SSLError): 
      print('Ignore: %s' % ip) 
     finally: 
      s.close() 
      return 

    def run(self): 
     for ip in self.iplist: 
      returnIP = self.checkCert(ip) 
      if returnIP: 
       self.queue.append(ip) 

def main(l, hostname, port): 
    iplist = [] 
    threads = [] 
    hostPool = [] 
    with open(l,'r') as f: 
     #while True: 
     iplist.extend([f.readline().strip() for x in islice(f, 10000)]) 
     #print(iplist) 
     t = scanThread(iplist, hostPool, hostname, port) 
     t.start() 
     threads.append(t) 
     iplist.clear() 

    for t in threads: 
     t.join() 

    for h in hostPool: 
     print(h) 




if __name__ == '__main__': 
    parser = argparse.ArgumentParser() 
    parser.add_argument("hostname",help="root hostname") 
    parser.add_argument("-l","--list",required=True, help="IP list for scanning") 
    parser.add_argument("-p","--port", nargs='?', const=443, default=443, type=int, help="port to scan") 
    arg = parser.parse_args() 
    main(arg.list,arg.hostname, arg.port) 

ich in kommentieren Sie nur while Schleife main Funktion, Daher erstellt das Skript einen Thread und scannt 10.000 IPs.

Unter 'google.com' zum Beispiel hat es zahlreiche IP-Adressen weltweit:

./google.py -l 443.txt google.com 

Beispielausgabe:

Processing IP: 13.76.139.89 
Ignore: 13.76.139.89 

Nach einigen Tests, ich bin mir ziemlich sicher, dass die for ... in Schleife in scanThread.run() einmal ausgeführt. Habe ich in diesem Snippet-Code etwas Unpassendes getan?

Antwort

0

Dies könnte daran liegen, dass Sie die Liste in der Hauptfunktion löschen.

t = scanThread(iplist, hostPool, hostname, port) 
    t.start() 
    threads.append(t) 
    iplist.clear() // here you are clearing. 

Können Sie versuchen:

class scanThread(Thread): 
    def __init__(self,iplist, q, hostname, port): 
     Thread.__init__(self) 
     self.iplist = list(iplist) 

self.iplist = list(iplist) dies ist eine Kopie der Liste machen, anstatt die Liste verwendet, die übergeben wird.

+0

Ziemlich. Ich habe das nicht verstanden. Vielen Dank für Ihre Antwort. –

+0

@ J.K hat es funktioniert? –

+0

Ja. Es geht um. Übrigens hast du gesagt :) –