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