nagios檢測http和https的外掛程式,python檢測http和https的外掛程式

標籤：

一個檢測http和https狀態和返回時間的程式,nagios可以使用的外掛程式

代碼中飽含注釋, 如果不明之處和認為有錯誤的地方, 請各位指出

#!/bin/env python#coding:utf8###########################################################################File name: check_http_status.py#Description: 利用python的pycurl模組,擷取url的返回狀態和返回時間#Author:pangli#mail:[email protected]##########################################################################import pycurlimport fcntlimport sysimport osimport StringIOimport timefrom optparse  import OptionParserTIME = int(time.time())data_path = os.path.split(os.path.realpath(__file__))[0] + "/http_data/"#判斷資料檔案夾是否存在if not os.path.isdir(data_path):    os.makedirs(data_path)#傳回碼:200-207成功狀態,300-307重新導向狀態,此處也劃分到成功裡code_rule = [200, 201, 202, 203, 204, 205, 206, 207, 300, 301, 302, 303, 304, 305, 306, 307]usage = "python %(script_name)s -u <url|ipaddress> -w <connectTime,totalTime> -c <connectTime,totalTime> -t <http|https>"parser = OptionParser(usage=usage % {"script_name" : sys.argv[0]})parser.add_option("-u", "--url", dest="url", help="url or ipaddress")parser.add_option("-t", "--type", dest="type", help="transport protocols type,http|https")parser.add_option("-w", "--warning", dest="w_value", help="alarm value(warning)")parser.add_option("-c", "--critical", dest="c_value", help="alarm value(critical)")option,args = parser.parse_args()if option.url == -1 or option.w_value == -1 or option.c_value == -1 or option.type == -1:    parser.print_help()    sys.exit(3)def http_url_req(url):    try:        buf = StringIO.StringIO()        status = dict()        #去掉使用者輸入的url頭        format_url = url.replace("http://", "")        req = pycurl.Curl()        #perform返回寫入緩衝忽略掉        req.setopt(req.WRITEFUNCTION, buf.write)        #佈建要求的URL        req.setopt(req.URL,"http://" + format_url)        #設定連線逾時        req.setopt(req.TIMEOUT,5)        #執行請求        req.perform()        status["return_code"] = req.getinfo(pycurl.HTTP_CODE)        status["con_time"] = float("%0.3f" % req.getinfo(pycurl.CONNECT_TIME))        status["tol_time"] = float("%0.3f" % req.getinfo(pycurl.TOTAL_TIME))        req.close()        return status    except pycurl.error,e:        print "The http status  : CRITICAL | connect failed "        sys.exit(2)    except Exception, e:        print str(e)        sys.exit(3)def https_url_req(url):    try:        buf = StringIO.StringIO()        status = dict()        #去掉使用者輸入的url頭        format_url = url.replace("https://", "")        req = pycurl.Curl()        #perform返回寫入緩衝忽略掉        req.setopt(req.WRITEFUNCTION, buf.write)        #忽略認證檢查        req.setopt(req.SSL_VERIFYPEER, 0)        #忽略主機驗證        req.setopt(req.SSL_VERIFYHOST, 0)        #佈建要求的URL        req.setopt(req.URL,"https://" + format_url)        #設定逾時串連        req.setopt(req.TIMEOUT,5)        #執行請求        req.perform()        status["return_code"] = req.getinfo(pycurl.HTTP_CODE)        status["con_time"] = float("%0.3f" % req.getinfo(pycurl.CONNECT_TIME))        status["tol_time"] = float("%0.3f" % req.getinfo(pycurl.TOTAL_TIME))        req.close()        return status    except pycurl.error:        print "The https status  : CRITICAL | connect failed "        sys.exit(2)    except Exception, e:        print str(e)        sys.exit(3)#判斷警示狀態def alarm(**params):    w = dict()    c = dict()    print_result  = "The http status  : %(status)s | URL=%(ip)s http_return_code=%(return_code)s connect_time=%(c_time)s total_time=%(t_time)s"    code = params["return_code"]    con_time = round(params["con_time"],3)    tol_time = round(params["tol_time"],3)    URL = params["url"]    w["cTime"],w["tTime"] = [float(each) for each in (option.w_value).split(",")]    c["cTime"],c["tTime"] = [float(each) for each in (option.c_value).split(",")]    #警示判斷    if cmp(con_time,c["cTime"]) >= 0 or cmp(tol_time,c["tTime"]) >= 0 or code not in code_rule:        print print_result % {"status":"CRITICAL","ip":URL, "return_code":code,"c_time":con_time,"t_time":tol_time}        sys.exit(2)        elif cmp(con_time,w["cTime"]) >= 0 or cmp(tol_time,w["tTime"]) >= 0 or code not in code_rule:        print print_result % {"status":"WARNING","ip":URL, "return_code":code,"c_time":con_time,"t_time":tol_time}         sys.exit(1)    else:        print print_result % {"status":"OK","ip":URL,"return_code":code,"c_time":con_time,"t_time":tol_time}        sys.exit(0)if __name__ == ‘__main__‘:    url = option.url        if option.type == "http":        return_data = http_url_req(url)        alarm(return_code = return_data["return_code"],            url=option.url,            con_time = return_data["con_time"],            tol_time = return_data["tol_time"])    elif option.type == "https":        return_data = https_url_req(url)        alarm(return_code = return_data["return_code"],            url=option.url,            con_time = return_data["con_time"],            tol_time = return_data["tol_time"])    else:        print "ERROR: transport protocols type error"        parser.print_help()        sys.exit(3)

執行結果展示

nagios檢測http和https的外掛程式,python檢測http和https的外掛程式