python实现探测socket和web服务示例

操作系统：linux
软件环境：Python 2.7.3

用法：

$ ./MonSocket.py 
# This is check the URI or Socket of the script #
Usage:
 ./MonSocket.py -d URL; This is Http protocol
 ./MonSocket.py -s socket IP or domain; This is Socket protocol
 ./MonSocket.py -p port; This is Socket port
 ./MonSocket.py -n ; Total number of requests
 ./MonSocket.py -c ; Number of concurrent requests
 ./MonSocket.py -t ; Timeout time(s),socket default is 1s,http default is 5s
For exampale: ./MonSocket.py -d www.weibo.com/index.php -n 200 -c 10 -t 2
For exampale: ./MonSocket.py -s 10.210.214.249 -p 80 -n 200 -c 50 -t 3

代码：

#!/usr/bin/env python
# encoding: utf-8

#
# Write by 飞奔的蜗牛-Bob
import os,sys
import getopt,re
import socket,threading,urllib2
def usage():
 print '# This is check the URI or Socket of the script #'
 print 'Usage:'
 print " %s -d URL; This is Http protocol" %sys.argv[0]
print " %s -s socket IP or domain; This is Socket protocol" %sys.argv[0]
print " %s -p port; This is Socket port" %sys.argv[0]
print " %s -n ; Total number of requests" %sys.argv[0]
print " %s -c ; Number of concurrent requests" %sys.argv[0]
print " %s -t ; Timeout time(s),socket default is 1s,http default is 5s" %sys.argv[0]
 print "For exampale: %s -d www.weibo.com/index.php -n 200 -c 10 -t 2" %sys.argv[0]
print "For exampale: %s -s 10.210.214.249 -p 80 -n 200 -c 50 -t 3" %sys.argv[0]
def Detect_url(url,sign):
if timeout:
time = int(timeout)
else:
time = 5
urllib2.socket.setdefaulttimeout(time)
request = urllib2.Request('http://%s' %(url))
try:
ret = urllib2.urlopen(request)
except urllib2.URLError,e:
if hasattr(e,"reason"):
port_timeout.append('1')
elif hasattr(e,"code"):
if re.findall('^3\d*','%s' %e.code):
port_normal.append('1')
if re.findall('^404\d*','%s' %e.code):
port_404.append('1')
 if re.findall('^403\d*','%s' %e.code):
 port_403.append('1')
 if re.findall('^500\d*','%s' %e.code):
 port_500.append('1')
if re.findall('^502\d*','%s' %e.code):
port_502.append('1')
 if re.findall('^503\d*','%s' %e.code):
 port_503.append('1')
else:
port_other.append('1')
else:
 port_normal.append('1')
def Detect_socket(server,port):
sign = 0
 if timeout:
 time = int(timeout)
 else:
 time = 1socket.setdefaulttimeout(time)
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
try:
ret = s.connect((server, int(port)))
except socket.error, e:
if re.findall('^timed\ out*','%s' %e):
socket_timeout.append('1')
sign = 1
else:
print '%s' %e
sys.exit(2)
else:
socket_normal.append('1')
sign = 1
if sign == 0:
s.close()
def print_out():
if url_mod:
print 'URL:'
 print 'timeout:[%s]' %len(port_timeout)
 print 'normal:[%s]' %len(port_normal)
 print '\033[;31mError_403:[%s]\tError_404:[%s]\033[0m' %(len(port_403),len(port_404))
 print '\033[;31mError_500:[%s]\tError_502:[%s]\tError_503:[%s]\033[0m' %(len(port_500),len(port_502),len(port_503))
print '\033[;31mError_other:[%s]\033[0m' %len(port_other)if sock_mod:
print 'Socket:'
 print 'timeout:[%s]' %len(socket_timeout)
  print 'normal:[%s]' %len(socket_normal)

def main():
if sock_mod:
dest_arg1 = sock_mod
dest_arg2 = dport
dest_function = Detect_socket
elif url_mod:
dest_arg1 = url_mod
dest_arg2 = ''
dest_function = Detect_url
else:
sys.exit()total = int(dcount)
concurrent = int(dconcurrent)
 n = 0
 sign = 0
lastnu = total%concurrent

 while 1:
 threads = []
 if n + concurrent > total:
 nloops = range(n,total)
 sign = 1
 else:
 nloops = range(n,n+concurrent)
 for i in nloops:
 t = threading.Thread(target=dest_function,args=(dest_arg1,dest_arg2))
 threads.append(t)
 if sign == 1:
 th_nu = lastnu
 else:
 th_nu = concurrent
 for i in range(th_nu):
 threads[i].start()
 for i in range(th_nu):
 threads[i].join()
 n = n + concurrent
 if sign == 1:
 break
print_out()

if __name__=='__main__':
try:
opts,args=getopt.getopt(sys.argv[1:],"hd:s:p:n:c:t:")
except getopt.GetoptError:
usage()
sys.exit(2)
port_timeout = []
port_normal = []
port_403= []
port_404 = []
port_500 = []
port_502 = []
port_503 = []
port_other = []
socket_normal = []
socket_timeout = []
dcount = 0
url_mod = ''
sock_mod = ''
dport = ''
dconcurrent = 0
timeout = 0
if opts:
for opt,arg in opts:
if opt == '-h':
usage()
sys.exit()
if opt == '-d':
url_mod = arg
if opt == '-s':
sock_mod = arg
if opt == '-p':
dport = arg
if opt == '-n':
dcount = arg
if opt == '-c':
dconcurrent = arg
if opt == '-t':
timeout = arg
if url_mod and dcount and dconcurrent:
main()
elif sock_mod and dport and dcount and dconcurrent:
main()
else:
usage()
 else:
usage()
sys.exit()

本文python实现探测socket和web服务示例到此结束。没有改变，就没有成长，幸福的人都喜欢沉默。一向喋喋不休说自我如何幸福的人必须内心是虚弱的。当一个人内心足够强大时，说与不说，都已无用。最重要的是，选择最适合自我的方向，一意孤行走下去。找到生命中最确定的信息——那些相似的人或事物终会走到一齐，那些不相似的人或事物，终会背道而驰。小编再次感谢大家对我们的支持！