# -*- coding: utf-8 -*-

# ping for Windows

import os

import sys

import socket

import struct

import select

import time

import ConfigParser

import weixin

import mail

class getcfg(object):

	def __init__(self,filename):

		self.filename=filename

		self.cfg=ConfigParser.ConfigParser()

		self.cfg.read(self.filename)

		#self.allip=self.cfg.items('allip')

		self.retry=self.cfg.get('rule','retry')

		self.timeout=self.cfg.get('rule','timeout')

		self.reload=self.cfg.get('rule','next_check')

		self.scan=self.cfg.items('scan_network')

		self.mailto=self.cfg.items('mail','mail_to')

	#发送邮件报警

	def sendmessage(self,failip):

		weixin.sendMessage('@all','ping %s failed.'% failip)

		for key,value in self.mailto:

			mail.send(value,"ping %s failed." % failip,'python ping fail')

	#判断要扫描ip

	def scan_network(self):

		self.scan.sort()

		allip=self.cfg.items('allip')

		i=0

		while i < len(self.scan):

			ipbegin=int(self.scan[i][1].split('.')[-1])

			ipend=int(self.scan[i+1][1])+1

			for x in range(ipbegin,ipend):

				ipvalue=self.scan[i][1].split('.')[0]+'.'+self.scan[i][1].split('.')[1]+'.'+self.scan[i][1].split('.')[2]+'.'+str(x)

				ip_dict=[]

				for k,v in allip:

					ip_dict.append(v)

				if ipvalue in ip_dict:

					pass

				else:

					#执行扫描

					self.to_ping(ipvalue,int(self.timeout))

			i=i+2

	#执行扫描

	def to_ping(self,ip_addr,timeout):

		print 'scan ip ' + ip_addr,

		try:

			delay = ping(self.filename).ping_once(ip_addr, timeout)

			if delay == None:

				print 'failed. (timeout within %s second.)' % timeout

			else:

				print 'get reply in %0.4f ms' % (delay * 1000)

				self.scan_resule(ip_addr)

		except socket.gaierror, e:

			print "failed. (socket error: '%s')" % e[1]

	#判断扫描结果,如果有新扫描到的ip则保存到配置文件

	def scan_resule(self,ip_addr):

		allip=self.cfg.items('allip')

		ip_dict=[]

		for k,v in allip:

			ip_dict.append(v)

		if ip_addr in ip_dict:

			pass

		else:

			self.cfg_write(ip_addr)

	#保存扫描到的ip到配置文件

	def cfg_write(self,ipvalue):

		ipkey='auto_'+str(time.time())

		self.cfg.set('allip',ipkey,ipvalue)

		self.cfg.write(open('ping_monitor.txt','w'))

	#去执行ping类

	def to_ping(self):

		allip=self.cfg.items('allip')

		for k,v in allip:

			if len(v) >0:

				ping(self.filename).icmp_ping(v,int(self.timeout),int(self.retry))

	#下次执行扫描时间

	def next_check(self):

		while True:

			self.to_ping()

			print('------------------------------------------------------------')

			nextcheck=0

			while nextcheck < int(self.reload):

				sys.stdout.write('next check %s second\r' % (int(self.reload)-nextcheck))

				sys.stdout.flush()

				time.sleep(1)

				nextcheck=nextcheck+1

			self.scan_network()

			print('------------------------------------------------------------')

class ping(object):

	def __init__(self,filename):

		self.ICMP_ECHO_REQUEST = 8

		self.filename=filename

	def receive_ping(self,my_socket, ID, timeout):

		start_time = timeout

		while True:

			start_select = time.clock()

			what_ready = select.select([my_socket], [], [], start_time)

			how_long = (time.clock() - start_select)

			if what_ready[0] == []: #timeout

				return

			time_received = time.clock()

			rec_packet, addr = my_socket.recvfrom(1024)

			icmp_header = rec_packet[20 : 28]

			ip_type, code, checksum, packet_ID, sequence = struct.unpack("bbHHh", icmp_header)

			if ip_type != 8 and packet_ID == ID: # ip_type should be 0

				byte_in_double = struct.calcsize("d")

				time_sent = struct.unpack("d", rec_packet[28 : 28 + byte_in_double])[0]

				return time_received - time_sent

			start_time = start_time - how_long

			if start_time <= 0:

				return

	def get_checksum(self,source):

		checksum = 0

		count = (len(source) / 2) * 2

		i = 0

		while i < count:

			temp = ord(source[i + 1]) * 256 + ord(source[i]) # 256 = 2^8

			checksum = checksum + temp

			checksum = checksum & 0xffffffff # 4,294,967,296 (2^32)

			i = i + 2

		if i < len(source):

			checksum = checksum + ord(source[len(source) - 1])

			checksum = checksum & 0xffffffff

		# 32-bit to 16-bit

		checksum = (checksum >> 16) + (checksum & 0xffff)

		checksum = checksum + (checksum >> 16)

		answer = ~checksum

		answer = answer & 0xffff

		# why? ans[9:16 1:8]

		answer = answer >> 8 | (answer << 8 & 0xff00)

		return answer

	def send_ping(self,my_socket, ip_addr, ID):

		ip = socket.gethostbyname(ip_addr)

		my_checksum = 0

		header = struct.pack('bbHHh', self.ICMP_ECHO_REQUEST, 0, my_checksum, ID, 1)

		byte_in_double = struct.calcsize("d") # C type: double

		data = (192 - byte_in_double) * "P" # any char is OK, any length is OK

		data = struct.pack("d", time.clock()) + data

		my_checksum = self.get_checksum(header + data)

		header = struct.pack("bbHHh", self.ICMP_ECHO_REQUEST, 0, socket.htons(my_checksum), ID, 1)

		packet = header + data

		my_socket.sendto(packet, (ip, 80)) # it seems that 0~65535 is OK (port?)

	def ping_once(self,ip_addr, timeout):

		icmp = socket.getprotobyname('icmp')

		try:

			my_socket = socket.socket(socket.AF_INET, socket.SOCK_RAW, icmp)

		except socket.error:

			raise

		my_ID = os.getpid() & 0xFFFF

		self.send_ping(my_socket, ip_addr, my_ID)

		delay = self.receive_ping(my_socket, my_ID, timeout)

		my_socket.close()

		return delay

	def icmp_ping(self,ip_addr,timeout,count):

		number=1

		for i in range(count):

			print 'ping ' + ip_addr,

			try:

				delay = self.ping_once(ip_addr, timeout)

				if delay == None:

					print 'failed. (timeout within %s second.)' % timeout

					if number==count:

						print('The request has failed %s times,Email alerts are being sent' % count)

						getcfg(self.filename).sendmessage(ip_addr)

				else:

					print 'get reply in %0.4f ms' % (delay * 1000)

					break

			except socket.gaierror, e:

				print "failed. (socket error: '%s')" % e[1]

				break

			number=number+1

if __name__ == '__main__':

	weixin=weixin.WeChat()

	mail=mail.sendmail()

	cfgname='ping_monitor.txt'

	ping(cfgname)

	pingcfg=getcfg(cfgname)

	pingcfg.next_check()

[rule]

retry = 4

timeout = 1

next_check = 300

#单位都是秒

[scan_network]

3paragraph_begin = 10.1.3.198

3paragraph_end = 202

2paragraph_begin = 10.1.2.1

2paragraph_end = 10

[mail]

mailto1=guoyabin@ccln.gov.cn

mailto2=lzt417@126.com

[allip]

#这里会自动把扫描到的ip写入