python 实现端口连通性检测

代码语言：javascript复制

# -*- coding: utf-8 -*-
#!/bin/env python
#AUTHOR:karl
#DATE:2018-1-19
#VERSION:V1.0
######################
import time
import os
import paramiko
import datetime
import sys
import MySQLdb
import threading 
#######################################
#首次插入数据是REG=0，更新数据REG为其他
#######################################
REG=1
private_key = paramiko.RSAKey.from_private_key_file('/home/appdeploy/.ssh/id_rsa')
def Data_mysql(info):
    try:
        string=info
        Pip=string.split(" ")[0]
        Sip=string.split(" ")[1]
        Dip=string.split(" ")[2]
        Port=string.split(" ")[3]
        Result=string.split(" ")[4].replace('n','')
        con = MySQLdb.connect(host='localhost',user="root",passwd="******",db="zabbix",port=3306,charset="utf8")
        dt = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        cursor=con.cursor()
        valuse=(dt,Result,Pip,Sip,Dip,Port)
        if REG == 0:
            sql="""INSERT INTO Balant_telnet (create_time,telnet_result,telnet_physicalip,telnet_sourceip,telnet_desip,telnet_port) VALUES (%s,%s,%s,%s,%s,%s) """
            result=cursor.execute(sql,valuse)
        else:
            sql="""UPDATE Balant_telnet set create_time='%s',telnet_result=%s where telnet_physicalip='%s' and telnet_sourceip='%s' and telnet_desip='%s' and telnet_port='%s';""" %(valuse)
            result=cursor.execute(sql)
        cursor.close()
        con.commit()
        con.close()
    except MySQLdb.Error, e:
        print "Error %d: %s" % (e.args[0], e.args[1])
        sys.exit(1)
def check_port(value,content):
    ager={}
    ager[value]=content
    ssh=paramiko.SSHClient()
    try:
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(value,port=22,username='appdeploy',pkey=private_key,timeout=5)
        cmd="sh /home/appdeploy/monitor.sh {value}".format(value=ager[ager.keys()[index]]) 
        stdin,stdout,stderr = ssh.exec_command(cmd)
        for file_d in stdout.readlines():  
            Data_mysql(file_d)
    except:
        result=cmd ',' 'failed' 'n'
    ssh.close()
###################################
# 对配置文件进行参数解析
###################################
def get_parameter():
    ager={}
    value=[]
    with open("monitor_config") as context:
        for line in context:
            while not "," in list(line):
                host=line.strip()[1:-1]
                value=[]
                break
            line=line.strip().split(",")
            value=value line
            ager[str(host)]=value
    return ager
if __name__ == '__main__':
    threads=[]
    parm=get_parameter()
    files=range(len(parm.keys()))
    for index,content in parm.items():
        t = threading.Thread(target=check_port,args=(index,content))
        threads.append(t)
    for index in files:
        threads[index].start()
    for index in files:
        threads[index].join()

配置文件：

[10.117.194.23]

10.117.194.77,10.116.41.82,9920

10.117.194.77,10.116.47.12,1080

10.117.194.77,10.116.45.56,1081

[10.117.194.24]

10.117.194.78,10.116.41.82,9920

10.117.194.78,10.116.47.12,1080

10.117.194.78,10.116.45.56,1081

10.117.194.78,10.116.145.33,8001

【】里面的IP 是物理IP，下面分别对应着应用的源IP，目标IP和测试端口号

agent 直接用shell 实现.