之前項目上線前,上司要求讓寫一個腳本用來判斷端口的占用情況。由于現在python3使用也比較多,基于python2修改了一下,做了個python3版本的,現在做一下總結。
一、python腳本實作掃描端口:
pthon2下代碼如下(當時的環境):
#!/usr/bin/env python
# -*- coding:utf-8 -*-
# used for pthon2.*
import socket, time, thread
socket.setdefaulttimeout(3) #設定預設逾時時間
def socket_port(ip, port):
"""
輸入IP和端口号,掃描判斷端口是否占用
"""
try:
if port >=65535:
print u'端口掃描結束'
s=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
result=s.connect_ex((ip, port))
if result==0:
lock.acquire()
print ip,u':',port,u'端口已占用'
lock.release()
except:
print u'端口掃描異常'
def ip_scan(ip):
"""
輸入IP,掃描IP的0-65534端口情況
"""
try:
print u'開始掃描 %s' % ip
start_time=time.time()
for i in range(0,65534):
thread.start_new_thread(socket_port,(ip, int(i)))
print u'掃描端口完成,總共用時:%.2f' %(time.time()-start_time)
# raw_input("Press Enter to Exit")
except:
print u'掃描ip出錯'
if __name__=='__main__':
url=raw_input('Input the ip you want to scan: ')
lock=thread.allocate_lock()
ip_scan(url)
效果圖:

python3的代碼:
#!/usr/bin/env python
# -*- coding:utf-8 -*-
# used for python3.*
import socket,time,_thread
socket.setdefaulttimeout(3) #設定預設逾時時間
def socket_port(ip, port):
"""
輸入IP和端口号,掃描判斷端口是否占用
"""
try:
if port >=65535:
print (u'端口掃描結束')
s=socket.socket(socket.AF_INET, socket.SOCK_STREAM)
result=s.connect_ex((ip, port))
if result==0:
lock.acquire()
print (ip,u':',port,u'端口已占用')
lock.release()
except:
print (u'端口掃描異常')
def ip_scan(ip):
"""
輸入IP,掃描IP的0-65534端口情況
"""
try:
print (u'開始掃描 %s' % ip)
start_time=time.time()
for i in range(0,65534):
_thread.start_new_thread(socket_port,(ip, int(i)))
print (u'掃描端口完成,總共用時:%.2f' %(time.time()-start_time))
# raw_input("Press Enter to Exit")
except:
print (u'掃描ip出錯')
if __name__=='__main__':
url=input('Input the ip you want to scan: ')
lock=_thread.allocate_lock()
ip_scan(url)
效果:
二、linux指令判斷
1. lsof -i:端口号 用于檢視指定端口号的占用情況,如下檢視80端口的情況。
2.netstat -tunlp |grep 端口号,用于檢視指定的端口号的程序情況,如檢視25端口的情況,netstat -tunlp |grep 25
三、寫python腳本中出現的問題
1.ImportError: No module named 'thread'
說沒有thread這個子產品,python3中沒有了thread子產品,取而代之的是_thread和threading(推薦使用),_thread是為了過渡使用的。
2.IndentationError: unindent does not match any outer indentation level
百度後發現原因是因為混用了空格和tab。
tab鍵設定為4位。vim /etc/vimrc 增加set ts=4