2014-10-03 18 views
6

Così sto facendo un port scanner in python ...Python - Fare un Port Scanner veloce

import socket 
ip = "External IP" 
s = socket.socket(2, 1) #socket.AF_INET, socket.SOCK_STREAM 

def porttry(ip, port): 
    try: 
     s.connect((ip, port)) 
     return True 
    except: 
     return None 

for port in range(0, 10000): 
    value = porttry(ip, port) 
    if value == None: 
     print("Port not opened on %d" % port) 
    else: 
     print("Port opened on %d" % port) 
     break 
raw_input() 

Ma questo è troppo lento, voglio essere in qualche modo in grado in qualche modo chiudere o rompere codice dopo un periodo di tempo di non restituire nulla. Se questo non è possibile, ti prego, aiutami in entrambi i casi

risposta

11

Oltre a impostare il timeout del socket, è possibile applicare anche la tecnica multi-threading turbo potenzia il processo. Nel migliore dei casi, sarà N volte più veloce quando hai N porte da scansionare.

# This script runs on Python 3 
import socket, threading 


def TCP_connect(ip, port_number, delay, output): 
    TCPsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
    TCPsock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) 
    TCPsock.settimeout(delay) 
    try: 
     TCPsock.connect((ip, port_number)) 
     output[port_number] = 'Listening' 
    except: 
     output[port_number] = '' 



def scan_ports(host_ip, delay): 

    threads = []  # To run TCP_connect concurrently 
    output = {}   # For printing purposes 

    # Spawning threads to scan ports 
    for i in range(10000): 
     t = threading.Thread(target=TCP_connect, args=(host_ip, i, delay, output)) 
     threads.append(t) 

    # Starting threads 
    for i in range(10000): 
     threads[i].start() 

    # Locking the script until all threads complete 
    for i in range(10000): 
     threads[i].join() 

    # Printing listening ports from small to large 
    for i in range(10000): 
     if output[i] == 'Listening': 
      print(str(i) + ': ' + output[i]) 



def main(): 
    host_ip = input("Enter host IP: ") 
    delay = int(input("How many seconds the socket is going to wait until timeout: ")) 
    scan_ports(host_ip, delay) 

if __name__ == "__main__": 
    main() 
2

considerare la creazione di un timeout invece di un ciclo for usando socket.setdefaulttimeout(timeout).

0

socket.setdefaulttimeout (0.5) Ciò renderà il programma più veloce!

+0

Questa risposta è un buon inizio, ma penso che sarebbe utile spiegare perché accelera il programma, magari con alcuni riferimenti. – Shepmaster

0

socket.setdefualttimeout (tempo)

viene utilizzato per continuare a cercare di connettersi con porta per il tempo perticular ... quando si invia richiesta e c'è timeout impostato per 2 secondi in modo che cercherà di connettersi con porta per 2 secondi .... se non ci sarà risposta da quella porta in 2 secondi .... sarà conteggiato come porta morta

0

Questo dovrebbe essere un po 'più veloce.

#-*-coding:utf8;-*- 
#qpy:3 
#qpy:console 

import socket 
import os 

# This is used to set a default timeout on socket 
# objects. 
DEFAULT_TIMEOUT = 0.5 

# This is used for checking if a call to socket.connect_ex 
# was successful. 
SUCCESS = 0 

def check_port(*host_port, timeout=DEFAULT_TIMEOUT): 
    ''' Try to connect to a specified host on a specified port. 
    If the connection takes longer then the TIMEOUT we set we assume 
    the host is down. If the connection is a success we can safely assume 
    the host is up and listing on port x. If the connection fails for any 
    other reason we assume the host is down and the port is closed.''' 

    # Create and configure the socket. 
    sock = socket.socket() 
    sock.settimeout(timeout) 

    # the SO_REUSEADDR flag tells the kernel to reuse a local 
    # socket in TIME_WAIT state, without waiting for its natural 
    # timeout to expire. 
    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) 

    # Like connect(address), but return an error indicator instead 
    # of raising an exception for errors returned by the C-level connect()  
    # call (other problems, such as “host not found,” can still raise exceptions). 
    # The error indicator is 0 if the operation succeeded, otherwise the value of 
    # the errnovariable. This is useful to support, for example, asynchronous connects. 
    connected = sock.connect_ex(host_port) is SUCCESS 

    # Mark the socket closed. 
    # The underlying system resource (e.g. a file descriptor) 
    # is also closed when all file objects from makefile() are closed. 
    # Once that happens, all future operations on the socket object will fail. 
    # The remote end will receive no more data (after queued data is flushed). 
    sock.close() 

    # return True if port is open or False if port is closed. 
    return connected 


con = check_port('www.google.com', 83) 
print(con)