Python 找出9個連續的空閑端口
項目需求
安裝某軟件,配置時候需要填寫空閑的端口。查看5個平臺的某個端口是否被占用
5個平臺為windows, linux, aix, hp, solaris
實現方案有兩種
利用 python 的 socket 模塊里的
def isInuse(ipList, port):
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
flag=True
for ip in ipList:
try:
s.connect((ip, int(port)))
s.shutdown(2)
print '%d is inuse' % port
flag=True
break
except:
print '%d is free' % port
flag=False
return flag
在try 模塊中 如果 s.connect((ip, int(port))) 如果能成功, 說明端口被占用.
否則, connect 不成功, 會進到except 中, 說明端口不被占用.
但是有個問題, 端口監聽的ip 除了 "127.0.0.1","0.0.0.0" 還有可能是本機的局域網ip 如 222.25.136.17 , 或者與之通信的那臺機器的ip。
可以通過這個方法獲得局域網 ip
def getLocalIp():
localIP = socket.gethostbyname(socket.gethostname())
return localIP
本代碼只針對 ipList = ("127.0.0.1","0.0.0.0",getLocalIp()) 這3個 ip 進行 connect
import sys
import os
import socket
def isInuse(ipList, port):
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
flag=True
for ip in ipList:
try:
s.connect((ip, int(port)))
s.shutdown(2)
print '%d is inuse' % port
flag=True
break
except:
print '%d is free' % port
flag=False
return flag
def getLocalIp():
localIP = socket.gethostbyname(socket.gethostname())
return localIP
def checkNinePort(startPort):
flag = True
ipList = ("127.0.0.1","0.0.0.0",getLocalIp())
for i in range(1, 10):
if (isInuse(ipList, startPort)):
flag = False
break
else:
startPort = startPort + 1
return flag, startPort
def findPort(startPort):
while True:
flag, endPort = checkNinePort(startPort)
if (flag == True): #ninePort is ok
break
else:
startPort = endPort + 1
return startPort
def main():
startPort=51988
# startPort = int(sys.argv[1])
print findPort(startPort)
main()
2. 利用netstat 輸出信息查找端口號匹配
第一種方法的準確性依賴于 connect((ip, int(port))) 中的 ip,到底怎樣的 ip 集合才是完備的, 可以確定這個端口不被占用?
于是, 有下面這個方法
**在 linux 用 netstat -tnpl 可以得到端口監聽信息,
觀察 tcp 0 0 10.173.1.208:3210 0.0.0.0:* LISTEN 55563/repserver
出現了 10.173.1.208:3210 所以 3210 端口是被占用的
對這些信息進行搜索 :5000, 如果存在, 就表示5000端口是LISTEN**.
如果輸出結果中不存在 :5000 的相關字符,表示這個端口不被占用.
netstat - tnpl | grep 321 tcp 0 0 10.173.1.208:3211 0.0.0.0:* LISTEN 55563/*** tcp 0 0 0.0.0.0:3212 0.0.0.0:* LISTEN 55586/*** tcp 0 0 10.173.1.208:3213 0.0.0.0:* LISTEN 55707/*** tcp 0 0 0.0.0.0:3214 0.0.0.0:* LISTEN 54272/java tcp 0 0 0.0.0.0:3215 0.0.0.0:* LISTEN 54272/java tcp 0 0 10.173.1.208:3216 0.0.0.0:* LISTEN 54822/*** tcp 0 0 10.173.1.208:3217 0.0.0.0:* LISTEN 34959/*** tcp 0 0 10.173.1.208:3218 0.0.0.0:* LISTEN 54849/***
依據這個思路, 給出代碼.
AIX 、HP 、WINDOWS、 LINUX、 SOLARIS 這幾個平臺查看端口信息的方式不同,
先進行機器平臺的判斷
然后調用各個平臺的端口占用判斷函數
如果要找出連續端口, 其中只要有一個端口占用, 就跳出循環
__author__ = 'I316736'
import os
import platform
import sys
def isInuseWindow(port):
if os.popen('netstat -an | findstr :' + str(port)).readlines():
portIsUse = True
print '%d is inuse' % port
else:
portIsUse = False
print '%d is free' % port
return portIsUse
def isInuseLinux(port):
#lsof -i:4906
#not show pid to avoid complex
if os.popen('netstat -na | grep :' + str(port)).readlines():
portIsUse = True
print '%d is inuse' % port
else:
portIsUse = False
print '%d is free' % port
return portIsUse
def isInuseAix(port):
if os.popen('netstat -Aan | grep "\.' + str(port) + ' "').readlines():
portIsUse = True
print '%d is inuse' % port
else:
portIsUse = False
print '%d is free' % port
return portIsUse
def isInuseHp(port):
if os.popen('netstat -an | grep "\.' + str(port) + ' "').readlines():
portIsUse = True
print '%d is inuse' % port
else:
portIsUse = False
print '%d is free' % port
return portIsUse
def isInuseSun(port):
if os.popen('netstat -an | grep "\.' + str(port) + ' "').readlines():
portIsUse = True
print '%d is inuse' % port
else:
portIsUse = False
print '%d is free' % port
return portIsUse
def choosePlatform():
#'Windows-7-6.1.7601-SP1'
#'AIX-1-00F739CE4C00-powerpc-32bit'
#'HP-UX-B.11.31-ia64-32bit'
#'Linux-3.0.101-0.35-default-x86_64-with-SuSE-11-x86_64'
#'SunOS-5.10-sun4u-sparc-32bit-ELF'
machine = platform.platform().lower()
if 'windows-' in machine:
return isInuseWindow
elif 'linux-' in machine:
return isInuseLinux
elif 'aix-' in machine:
return isInuseAix
elif 'hp-' in machine:
return isInuseHp
elif 'sunos-' in machine:
return isInuseSun
else:
print 'Error, sorry, platform is unknown'
exit(-1)
def checkNinePort(startPort):
isInuseFun = choosePlatform()
nineIsFree = True
for i in range(1, 10):
if (isInuseFun(startPort)):
nineIsFree = False
break
else:
startPort = startPort + 1
return nineIsFree, startPort
def findPort(startPort):
while True:
flag, endPort = checkNinePort(startPort)
if (flag == True): # ninePort is ok
break
else:
startPort = endPort + 1
return startPort
def main(startPort):
firstPort=findPort(startPort)
print 'First port of nine free ports is ', firstPort
if __name__ == '__main__' :
if len(sys.argv) > 1:
print len(sys.argv)
startPort = int(sys.argv[1])
else:
startPort = 500
main(startPort)
相關知識點總結
os.popen()
可以調用系統的一些shell命令
os.popen().readlines()
讀取調用shell命令后的回顯信息
netstat -tnpl -tnpl 各個參數的含義 -l或--listening 顯示監控中的服務器的Socket。 -n或--numeric 直接使用IP地址,而不通過域名服務器。 -p或--programs 顯示正在使用Socket的程序識別碼和程序名稱。 -t或--tcp 顯示TCP傳輸協議的連線狀況 ---------- tcp 0 0 10.173.1.208:4903 0.0.0.0:* LISTEN 54849/jsagent 最后的54849/jsagent 表示 進程號 54849 進程名 jsagent