python实现微信hook_graylog通过python实现钉钉/微信/webhook告警
# coding:utf-8
# python3.6
# 只支持HTTP头包含Content-Length的请求
# @创建者：jonnyan404
# #日期：2020-05-09
# 博客地址：www.jonnyan404.top:8088
import socket
import threading
import json
import my_logging
import logging
import requests
my_logging.load_my_logging_cfg()
HTTPHEADER = b"""HTTP/1.1 200 OK
Content-Type: text/html;charset=utf-8
Connection:Keep-Alive
"""
def resolve_http_header(header):
header = header.decode('utf-8')
headers = {}
for i in header.split('\r\n'):
if ':' in i:
k = i.split('[0].strip()
v = i.split('[1].strip()
headers[k] = v
return headers
def dingding(text):
headers = {'Content-Type': 'application/json;charset=utf-8'}
api_url = "此处更改为你自己的钉钉api url"
json_text= {
"msgtype": "text",
"at": {
"atMobiles": [
"你的手机号码"
],
"isAtAll": False
},
"text": {
"content": text
}
}
requests.post(api_url,json.dumps(json_text),headers=headers).content
class React(threading.Thread):
def init(self, conn, addr):
threading.Thread.init(self)
self.setDaemon(True)
self.conn = conn
self.addr = addr
def read_chunk(self):
data = b''
# 第一次接收
try:
self.conn.settimeout(5)
chunk = self.conn.recv(1024)
except Exception as e:
logging.error(e)
return b'', b''
header_len = chunk.index(b'\r\n\r\n') + len(b'\r\n\r\n')
# 获取请求头并转换成字典格式
header = chunk[:header_len]
headers = resolve_http_header(header)
# 将第一次请求获取到的数据部分保存到data
data += chunk[header_len:]
data = data.strip() if data else b''
# print(header)
# print()
# print(data)
# 在while循环中接收剩余数据，当data长度与header中的Content-Length相同时结束循环
try:
print('Contetn-Length:', headers['Content-Length'])
logging.info('Contetn-Length:', headers['Content-Length'])
except:
print('http header error.')
logging.error('http header error.')
return b'', b''
while not int(headers['Content-Length']) == len(data):
try:
chunk = self.conn.recv(4096)
if chunk:
data += chunk
logging.info('read %d bytes.' % len(data))
# 如果客户端没有主动关闭连接则不再接收
else:
break
except Exception as e:
logging.error(e)
break
logging.info('done')
# 数据接收完成，响应客户端请求，并关闭连接
self.conn.send(HTTPHEADER + b'ok ')
self.conn.close()
return header, data
def run(self):
header, payload = self.read_chunk()
if not header or not payload:
return
# 保存接收到的所有数据
logging.info(payload)
data=json.loads(payload.decode("utf-8"))
msg=data['event']['message']
logging.info(msg)
dingding(msg)
def server():
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
address = ('0.0.0.0', 8000)
s.bind(address)
s.listen(5)
logging.info('listening on port %d ...' % 8000)
while True:
conn, addr = s.accept()
React(conn, addr).start()
if name == 'main':
server()