python监控端口_Python监控服务端口并报警

#!/usr/bin/env python

importosimportwmiimporttimeimportsocketimportbase64importsmtplibimportloggingfrom email.mime.text importMIMETextdefGetSrv(designation):"""Get stopped service name and caption,

Filtration ‘designation‘ service whether there is ‘Stopped‘.

:return: service state"""c=wmi.WMI()

ret=dict()for service inc.Win32_Service():

state, caption=service.State, service.Captionif state == ‘Stopped‘:

t=ret.get(state, [])

t.append(caption)

ret[state]=t#If ‘designation‘ service in the ‘Stopped‘, return status is ‘down‘

if designation in ret.get(‘Stopped‘):

logging.error(‘Service [%s] is down, try to restart the service. \r\n‘ %designation)return ‘down‘

returnTruedefMonitor(sname):"""Send the machine IP port 20000 socket request,

If capture the abnormal returns the string ‘ex‘.

:return: string ‘ex‘"""s=socket.socket()

s.settimeout(3) #timeout

host = (‘127.0.0.1‘, 20000)try: #Try connection to the host

s.connect(host)exceptsocket.error as e:

logging.warning(‘[%s] service connection failed: %s \r\n‘ %(sname, e))return ‘ex‘

returnTruedefRestartSocket(rstname, conn, run):"""First check whether the service is stopped,

if stop, start the service directly.

The check whether the zombies,

if a zombie, then restart the service.

:return: flag or True"""flag=Falsetry:#From GetSrv() to obtain the return value, the return value

if run == ‘down‘:

ret= os.system(‘sc start "%s"‘ %rstname)if ret !=0:raise Exception(‘[Errno %s]‘ %ret)

flag=Trueelif conn == ‘ex‘:

retStop= os.system(‘sc stop "%s"‘ %rstname)

retSart= os.system(‘sc start "%s"‘ %rstname)if retSart !=0:raise Exception(‘retStop [Status code %s]‘

‘retSart [Status code %s]‘ %(retStop, retSart))

flag=Trueelse:

logging.info(‘[%s] service running status to normal‘ %rstname)returnTrueexceptException as e:

logging.warning(‘[%s] service restart failed: %s \r\n‘ %(rstname, e))returnflagdefSendMail(to_list, sub, contents):"""Send alarm mail.

:return: flag"""mail_server= ‘mail.stmp.com‘ #STMP Server

mail_user = ‘YouAccount‘ #Mail account

mail_pass = base64.b64decode(‘Password‘) #The encrypted password

mail_postfix = ‘smtp.com‘ #Domain name

me= ‘Monitor alarm‘ %(mail_user, mail_postfix)

message= MIMEText(contents, _subtype=‘html‘, _charset=‘utf-8‘)

message[‘Subject‘] =sub

message[‘From‘] =me

message[‘To‘] = ‘;‘.join(to_list)

flag= False #To determine whether a mail sent successfully

try:

s=smtplib.SMTP()

s.connect(mail_server)

s.login(mail_user, mail_pass)

s.sendmail(me, to_list, message.as_string())

s.close()

flag=TrueexceptException, e:

logging.warning(‘Send mail failed, exception: [%s]. \r\n‘ %e)returnflagdefmain(sname):"""Parameter type in the name of the service need to monitor,

perform functions defined in turn, and the return value is correct.

After the program is running, will test three times,

each time interval to 10 seconds.

:return: retValue"""retry= 3count=0

retValue= False #Used return to the state of the socket

while count

ret=Monitor(sname)if ret != ‘ex‘: #If socket connection is normaol, return retValue

retValue =retreturnretValue

isDown=GetSrv(sname)

RestartSocket(rstname=sname, conn=ret, run=isDown)

host=socket.gethostname()

address=socket.gethostbyname(host)

mailto_list= [‘mail@smtp.com‘, ] #Alarm contacts

SendMail(mailto_list, ‘Alarm‘,‘

Level: ERROR Host name: %s‘

‘IP Address: %s‘

‘Service name:

%s

‘

%(host, address, sname))

count+= 1time.sleep(10)else:

logging.error(‘[%s] service try to restart more than three times \r\n‘ %sname)returnretValueif __name__ == ‘__main__‘:

logging.basicConfig(level=logging.INFO,

format=‘%(asctime)s %(levelname)s %(message)s‘,

datefmt=‘%Y/%m/%d %H:%M:%S‘,

filename=‘D:\\em_logs\\SocketMonitor.log‘,

filemode=‘ab‘)

name= ‘IM_IMConnectorServerWinService‘response=main(name)ifresponse:

logging.info(‘The [%s] service connection is normal \r\n‘ % name)