본문 바로가기

컴퓨터/Python

포트 스캐너

import optparse
from socket import *

def connScan(tgtHost, tgtPort):
    try:
        connSkt = socket(AF_INET, SOCK_STREAM)
        connSkt.connect((tgtHost, tgtPort))
        connSkt.send('ViolentPython\r\n')
        results = connSkt.recv(100)
        print '[+]%d/tcp open'% tgtPort
        print '[+] '+str(results)
        connSkt.close()
    except:
        print '[-]%d/tcp closed'% tgtPort
def portScan(tgtHost, tgtPorts):
    try:
        tgtIP = gethostbyname(tgtHost)
    except:
        print "[-] Cannot resolve '%s': Unknown host"%tgtHost
        return
    try:
        tgtName = gethostbyaddr(tgtIP)
        print '\n[+] Scan Result for: ' + tgtName[0]
    except:
        print '\n[+] Scan Result for: ' + tgtIP
    setdefaulttimeout(1)
    for tgtPort in tgtPorts:
        print 'Scanning port ' + tgtPort
        connScan(tgtHost, int(tgtPort))

def main():
    parser = optparse.OptionParser('usage%prog '+\
    '-H <target host> -p <target port>')
    parser.add_option('-H', dest='tgtHost', type='string',\
    help = 'specify target host')
    parser.add_option('-p', dest='tgtPort', type='string',\
    help='specify target port[s] separated by comma')
    (options, args) = parser.parse_args()
    tgtHost = options.tgtHost
    tgtPorts = str(options.tgtPort).split(',')
    if (tgtHost == None) | (tgtPorts[0] == None):
        print '[-] You must specify a target host and port[s].'
        exit(0)
    portScan(tgtHost, tgtPorts)
if __name__ == '__main__':
    main()



쓰레드 사용하기
import optparse
from socket import *
from threading import *
screenLock = Semaphore(value=1)

def connScan(tgtHost, tgtPort):
        try:
                connSkt = socket(AF_INET, SOCK_STREAM)
                connSkt.connect((tgtHost, tgtPort))
                connSkt.send('ViolentPython\r\n')
                results = connSkt.recv(100)
                screenLock.acquire()
                print '[+]%d/tcp open'% tgtPort
                print '[+] '+str(results)
        except:
                screenLock.acquire()
                print '[-]%d/tcp closed'% tgtPort
        finally:
                screenLock.release()
                connSkt.close()
def portScan(tgtHost, tgtPorts):
        try:
                tgtIP = gethostbyname(tgtHost)
        except:
                print "[-] Cannot resolve '%s': Unknown host"%tgtHost
                return
        try:
                tgtName = gethostbyaddr(tgtIP)
                print '\n[+] Scan Result for: ' + tgtName[0]
        except:
                print '\n[+] Scan Result for: ' + tgtIP
        setdefaulttimeout(1)
        for tgtPort in tgtPorts:
                t = Thread(target=connScan, args=(tgtHost, int(tgtPort)))
                t.start()

def main():
        parser = optparse.OptionParser('usage%prog '+\
        '-H <target host> -p <target port>')
        parser.add_option('-H', dest='tgtHost', type='string',\
        help = 'specify target host')
        parser.add_option('-p', dest='tgtPort', type='string',\
        help='specify target port[s] separated by comma')
        (options, args) = parser.parse_args()
        tgtHost = options.tgtHost
        tgtPorts = str(options.tgtPort).split(',')
        if (tgtHost == None) | (tgtPorts[0] == None):
                print '[-] You must specify a target host and port[s].'
                exit(0)
        portScan(tgtHost, tgtPorts)
if __name__ == '__main__':
        main()


root@bt:~/Desktop# python portscanner.py -H 127.0.0.1 -p 7337
[+] Scan Result for: localhost
[+]7337/tcp open


'컴퓨터 > Python' 카테고리의 다른 글

Pexpect로 SSH 연결하기  (0) 2013.08.17
nmap 포트 스캐너 통합하기  (0) 2013.08.13
압축 파일의 패스워드 찾기  (0) 2013.08.10
압축파일에 잘못된 패스워드를 입력  (0) 2013.08.09
압축파일 해제  (0) 2013.08.09