import requests
from bs4 import BeautifulSoup
from nonebot import on, on_command
from nonebot.rule import to_me
from nonebot.matcher import Matcher
from nonebot.adapters import Message
from nonebot.params import Arg, CommandArg, ArgPlainText
from nonebot.adapters.onebot.v11 import Message, MessageSegment
import asyncio
headers = {
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6',
'Cache-Control': 'max-age=0',
'Connection': 'keep-alive',
'Cookie': 'PHPSESSID=lj0tsuf8poheud74qbbgp8o4c4',
'Sec-Fetch-Dest': 'document',
'Sec-Fetch-Mode': 'navigate',
'Sec-Fetch-Site': 'none',
'Sec-Fetch-User': '?1',
'Upgrade-Insecure-Requests': '1',
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.64 Safari/537.36 Edg/101.0.1210.53',
'sec-ch-ua': '" Not A;Brand";v="99", "Chromium";v="101", "Microsoft Edge";v="101"',
'sec-ch-ua-mobile': '?0',
'sec-ch-ua-platform': '"Windows"'
}
payload = {}
searchfileform = on_command('searchfileform', aliases={'查询文件头', '文件头'}, priority=5)
@searchfileform.handle()
async def searchfileform_(matcher: Matcher, args: Message = CommandArg()):
houzhuiming = args.extract_plain_text()
if houzhuiming:
try:
url = "https://filesignatures.net/index.php?page=search&search={0}&mode=EXT".format(houzhuiming)
response = requests.request("GET", url, headers=headers, data=payload)
soup = BeautifulSoup(response.text, 'html.parser')
selector = soup.select('#results > a')
text = "[+] 查询成功!"+Message('[CQ:face,id=324]')+"\n[+] 文件后缀为:"+selector[0].get_text()+"\n[+] 文件头为:"+selector[1].get_text()
await matcher.send(text)
except Exception as e:
await matcher.send("[-] 查询失败!"+Message('[CQ:face,id=173]')+"\n[-] "+str(e))
re_searchfileform = on_command('re_searchfileform', aliases={'反查文件头'}, priority=5)
@re_searchfileform.handle()
async def re_searchfileform_(matcher: Matcher, args: Message = CommandArg()):
qianzhui = args.extract_plain_text()
url = "https://filesignatures.net/index.php?search={0}&mode=SIG".format(qianzhui.strip(' '))
if qianzhui:
try :
response = requests.request("GET", url, headers=headers, data=payload)
soup = BeautifulSoup(response.text, 'html.parser')
selector = soup.select('#results > a')
#只打印奇数位
text = ''
for i in range(0,len(selector),2):
#text逗号分割
text = text+selector[i].get_text()+","
if text:
text = text[:-1]
text = "[+] 查询成功!"+Message('[CQ:face,id=324]')+"\n[+] 可能的文件类型为:\n"+text
await matcher.send(text)
else:
await matcher.send("[-] 查询失败!"+Message('[CQ:face,id=173]'))
except Exception as e:
await matcher.send("[-] 查询失败!"+Message('[CQ:face,id=173]')+"\n[-] "+str(e))
【CTF机器人】文件头字节查询(Nonebot2+CQHTTP)
最新推荐文章于 2023-09-21 16:35:17 发布