#!/usr/local/bin/python3.8
# -*- coding:utf-8 -*-
# Author: 上善
# Time: 2019-10-25 13:37
#
import os
import sys
import re
import time
import hashlib
import requests
import datetime
import subprocess
import dns.resolver
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)
class box (object):
"""运维工具"""
def check_ip(self):
iplist = open("/Users/sean/Desktop/Module/cdnetwork/tools_box/module/node.list", encoding="utf-8")
ip_list = iplist.readlines()
for ip in ip_list:
res = os.popen('curl -s http://freeapi.ipip.net/' + ip).read()
ip_address = ip.replace("\n", "") + " " + "" + (res[1:-1]).replace("\"", "").replace(",", " ")
print(ip_address)
def beian(self):
'''
备案查询
:return:
'''
while True:
try:
domain = input("请输入二级域名(eg baidu.com 按q or Q退出) ：")
if re.match('(([\\w-]{1,62})?(\\.[\\w-]{1,62})+)', domain):
# 新网api接口
http_uri = "http://icp.fleacloud.com/api/v1/icp?domain="
http_url = "http://icp.fleacloud.com/api/v1/icp?domain=" + domain
gmt_format = '%a, %d %b %Y %H:%M:%S GMT'
date_gmt = datetime.datetime.utcnow().strftime(gmt_format)
headers = {
'Date': date_gmt,
'Accept': 'application/json',
'Content-type': 'application/json',
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_6) AppleW '
'ebKit/537.36 (KHTML, like Gecko) Chrome/77.0.3865.120 Safari/537.36'
}
res = requests.get(http_url, headers=headers)
if res.json()['status'] == 3:
print('域名 : ' + domain + " 备案号 : " + res.json()['icp'])
elif res.json()['status'] == 2:
print('域名 : ' + domain + " 备案号 : " + res.json()['icp'])
else:
print('查询失败或未知')
elif domain.upper() == 'Q':
break
else:
print('域名输入有误；')
except AttributeError:
sys.exit('系统错误')
def oscp(self):
'''
OCSP Stapling 测试
:return:
'''
with open('/Users/sean/Desktop/Module/cdnetwork/tools_box/module/domain_list', 'r') as domain_list:
for domain_list in domain_list.readlines():
(ip, domain) = domain_list.strip("\n").split()
print(domain + ":\n" + subprocess.Popen(('openssl s_client -connect ' + domain + ':443 -servername '
+ domain + ' -status -tlsextdebug < /dev/null 2>&1 | grep -i '
'"OCSP response"'),
shell=True, stdout=subprocess.PIPE,
stderr=subprocess.PIPE).stdout.read().decode())
def datetime(self):
'''
日期和md5加密
md5加密
:return:
'''
date_time = datetime.datetime.now().strftime('%Y-%m-%d-%H%M%S')
print(date_time)
print(subprocess.Popen(('echo'+date_time+'|md5'), shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE).
stdout.read().decode())
def openssl(self):
'''
证书校验
:return:
'''
taps = '''
domain_cname: images-cn.ssl-images-amazon.com.wtxcdn.com
domain_pad: images-cn.ssl-images-amazon.com
'''
print(taps)
c_domain = input('domain_cname: ')
s_domain = input('domain_pad: ')
print(os.popen('openssl s_client -connect c_domain:443 '
'-servername s_domain< /dev/null 2> /dev/null | openssl x509 -text | grep -C2 Altern'))
def ab_test(self):
'''
apachc ab 压力测试
:return:
'''
url = input('Http_Url: ')
proxy = input('Proxy: ')
port = input('Port: ')
print('ab -X ' + proxy + ':' + port + ' -n 100 -c 50 ' + url)
if (subprocess.Popen(('ab -X ' + proxy + ':' + port + ' -n 100 -c 50 ' + url), shell=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE).stdout.read().decode()):
print('命令执行完成100链接10并发')
def timestamp_datetime(self, value):
'''
时间戳转时间
:param value:
:return:
'''
format = '%Y-%m-%d %H:%M:%S'
value = time.localtime(value)
dt = time.strftime(format, value)
return dt
def datetime_timestamp(self, dt):
'''
时间转时间戳
:param dt:
:return:
'''
# dt为字符串
# 中间过程，一般都需要将字符串转化为时间数组
time.strptime(dt, '%Y-%m-%d %H:%M:%S')
# 10位 (秒)
s = int(time.mktime(time.strptime(dt, '%Y-%m-%d %H:%M:%S')))
# 13位(毫秒)
# s = (time.mktime(time.strptime(dt, '%Y-%m-%d %H:%M:%S')))*1000
print('unix时间戳（10位秒级）：' + str(s))
return int(s)
def md5sum_encryption(self, unix_time, uri, key, encryption):
'''
秘文组合方式需要在变量上修改
:return:
'''
wstime = str(600)
if encryption == 'key+unixtime+wstime':
uri_encryption = bytes((key + str(unix_time) + wstime).encode('utf-8'))
md5encryption = hashlib.md5(uri_encryption).hexdigest()
print('加密方式方式：' + encryption)
print('md5: ' + md5encryption)
elif encryption == 'uri+key+unixtime':
uri_encryption = bytes((uri + key + str(unix_time)).encode('utf-8'))
md5encryption = hashlib.md5(uri_encryption).hexdigest()
print('加密方式方式：' + encryption)
print('md5: ' + md5encryption)
elif encryption == 'key+time+uri':
uri_encryption = bytes((key + str(unix_time)+ uri).encode('utf-8'))
md5encryption = hashlib.md5(uri_encryption).hexdigest()
print('加密方式方式：' + encryption)
print('md5: ' + md5encryption)
elif encryption == 'time+key+uri':
uri_encryption = bytes((str(unix_time) +key+ uri +str(3600)).encode('utf-8'))
md5encryption = hashlib.md5(uri_encryption).hexdigest()
print(unix_time)
print('加密方式方式：' + encryption)
print('md5: ' + md5encryption)
else:
print('暂不支持的加密方式，请检查或修改python')
def unix_timestamp_create(self):
'''
变量获取并用md5 加密
:return:
'''
if os.path.exists('/Users/sean/Desktop/Module/cdnetwork/tools_box/module/uri_information'):
unix_time = box.datetime_timestamp(self, time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
with open('/Users/sean/Desktop/Module/cdnetwork/tools_box/module/uri_information', "r") as f:
for uri_line in f.readlines():
uri_information = uri_line
(uri, key, encryption) = uri_information.strip('\n').split()
box.md5sum_encryption(self,unix_time, uri, key, encryption)
else:
with open('/Users/sean/Desktop/Module/cdnetwork/tools_box/module/uri_information', "w+") as f:
f.write('origin/111 SEd0kljv5M key+unixtime+wstime ')
sys.exit('uri_information文件不存在,将在当前目录生成,请退出后')
def http_requests(self):
'''
request 测试
:return:
'''
print(datetime.datetime.utcnow().strftime('%a, %d %b %Y %H:%M:%S GMT'))
s = requests.session()
s.keep_alive = False # 关闭多余连接
with open('/Users/sean/Desktop/Module/cdnetwork/tools_box/module/url', 'r') as f:
for i in f.readlines():
(http_url, proxy) = i.strip().split()
print(http_url)
proxies = {
'http': proxy + ':80',
'https': proxy + ':443'
}
headers = {
'Date': datetime.datetime.utcnow().strftime('%a, %d %b %Y %H:%M:%S GMT'),
'Accept': 'application/json',
'Content-type': 'application/json',
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_6) '
'AppleWebKit/537.36 (KHTML, like Gecko) Chrome/77.0.3865.120 Safari/537.36',
'cdn-src-ip': proxy
}
res = requests.get(http_url, headers=headers, proxies=proxies, verify=False)
headers_post = dict(res.headers)
tmp_str = "statusCode:{}\nDate:{}\nContent-Length:{}\nConnection:{}\nCache-Control:{}\nVia:{}\nX-px:{}\n".format(
res.status_code,
headers_post.get('Date'),
headers_post.get('Content-Length'),
headers_post.get('Connection'),
headers_post.get('Cache-Control'),
headers_post.get('Via'),
headers_post.get('X-Px'),
)
print(tmp_str)
def hosts(self):
domain = input("Please input an domain: ").strip() # 输入一个域名
try:
A = dns.resolver.query(domain, 'A') # 指定查看类型为A记录
except Exception as e:
print(e)
else:
for i in A.response.answer: # 通过response.answer方法获取查询回应信息
for j in i.items: # 遍历回应信息
hosts = (j.address + " " + domain)
os.system("echo "+hosts+"|sed 's/.whecloud.com//g' | sed 's/.wtxcdn.com//g'>> /etc/hosts")
os.system('cat /etc/hosts')
